解决方法:
在train.py中,作者使用的vit_name 是
parser.add_argument('--vit_name', type=str,default='R50-ViT-B_16', help='select one vit model')
生成的文件夹的名称为:
但在test.py中,作者写的vit_name是:
parser.add_argument('--vit_name', type=str, default='ViT-B_16', help='select one vit model')
解决方法:
test.py中的vit_name 改为和train.py中一样的R50-ViT-B_16
修改后的test.py 中,vit_name的值为
parser.add_argument('--vit_name', type=str, default='R50-ViT-B_16', help='select one vit model')