发布 VIT + 损失函数

你好， 我使用了 VIT 来处理具有 2 个类别的图像。['horses', 'humans']

model= ViT( image_size = 256, patch_size = 32, num_classes = 2, dim = 1024, depth = 6, heads = 16, mlp_dim = 2048, dropout = 0.1, emb_dropout = 0.1 )