Ell-M
Results
3
comments of
Ell-M
I got the same error. Did u solve the issue?
found it. I had to change this in test.py set the default parser.add_argument('--vit_name', type=str, default='R50-ViT-B_16', help='select one vit model')
in train.py we used a pretrained model default='R50-ViT-B_16' it should be the same in test.py to find where we saved the model and make the prediction on our test database