Proxy-Anchor-CVPR2020

Proxy-Anchor-CVPR2020 copied to clipboard

if args.loss == 'Proxy_Anchor': param_groups.append({'params': criterion.parameters(), 'lr':float(args.lr) * 100}) elif args.loss == 'Proxy_NCA': param_groups.append({'params': criterion.parameters(), 'lr':float(args.lr)}) Why do you multiply yours lr by 100, but not the other method? In...

grypesc

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? | Method...

junwei-h

CUDA out of memory

3

python train.py --gpu-id 0 --loss Proxy_Anchor --model r esnet50 --embedding-size 512 --batch-size 180 --lr 6e-4 --dataset SOP --warm 1 --bn-freeze 0 --lr-decay-step 20 --lr-dec ay-gamma 0.25 wandb: Currently logged in...

ShuteLee