colab运行fvr模型注意点

登陆googledrive-挂载数据集-建立snapshots文件夹
然后开始运行
**本次fvr模型设置：**数据集是SDMULA,损失函数是softmax，使用的网络是lenet，最多训练60个epoch

batch_size=4, dataset='SDMULA', feature_dim=128, gamma_cos=1.0, gamma_tri=1.0, hard_margin=0.2, k=4, log_interval=10, loss='softmax', lr=0.01, lr_decay=0.1, lr_fc=0.1, m=0.2, max_epoch=60, momentum=0.9, network='lenet++', openset=False, p=8, plot=False, pretrained=False, s=15.0, seed=1, wd=0.0005

训练结果：
准确率最高的是在第43轮：seed=1_dataset=SDMULA_network=lenet++_loss=softmax_BestCMC=98.93_Epoch=43
但是一共训练了60轮：
snapshots输出为：seed=1_dataset=SDMULA_network=lenet++_loss=softmax_FinalCMC=98.84_Epoch=59

遇到的bug及解决方法：

1. 需要注意的是在datasets.py和train.py文件里得把文件路径都改成挂载在colab上的数据集的路径，否则cv2.imread会找不到路径，从而image为空值，就会报出“NoneType” object has no attribute ‘array_interface’。
2. RuntimeError: Given groups=1, weight of size 32 1 5 5, expected input[4, 3, 64, 144] to have 1 channels, but got 3 channels instead