FreeZ3e

Same problem, I tried torchvision==0.8.2, torch==1.7.1, timm==0.3.2 and CUDA11.0. But didn't work in evaluate mode.

I solved this problem by using "--resume" to load pretrained model instead of "--pretrain_path". If you want use "--pretrain_path" to evaluate model, plz use "torch.save()" to save model first.