You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I guess the README reported training result, not evaluation for SOP? I used your pre-trained model, ran the evaluate.py and got much lower result. Is this expected?
Here is what I did python Proxy-Anchor-CVPR2020/code/evaluate.py --gpu-id -1 --batch-size 120 --model bn_inception --embedding-size 512 --dataset SOP --resume ../pretrained/SOP_bn_inception_best.pth --workers 4
To get it run on CPU, Ubuntu 20.04 (WSL), torch==1.13.1, I changed code related to cuda:
# model = model.cuda() ifargs.gpu_id!=-1:
model=model.cuda()
torch.load(args.resume, map_location=torch.device('cpu'))
Hello,
I guess the README reported training result, not evaluation for SOP? I used your pre-trained model, ran the evaluate.py and got much lower result. Is this expected?
Here is what I did
python Proxy-Anchor-CVPR2020/code/evaluate.py --gpu-id -1 --batch-size 120 --model bn_inception --embedding-size 512 --dataset SOP --resume ../pretrained/SOP_bn_inception_best.pth --workers 4
To get it run on CPU, Ubuntu 20.04 (WSL), torch==1.13.1, I changed code related to cuda:
Also fixed an error by adding
strict=False
model.load_state_dict(checkpoint['model_state_dict'], strict=False)
Otherwise, I will hit errors raised by torch shown below
The text was updated successfully, but these errors were encountered: