Open enesmsahin opened 2 years ago
Hi, Thanks for noticing that. Indeed, there is a problem in loading the pre-trained model for the TResNet-L model. We will fix that soon. In the meantime, you can use TResNet-M.
Any update on this? im still running into the same problem
I am still getting an error while loading the TResNet-L model. Could you please give the updated checkpoint? @ebenbaruch @mrT23 Thank you!
Runtime Error occurs while loading state_dict for provided TResNet-L model. I think provided pretrained model and TResnetL class differ. Selective (CSL) - TResNet-M works without any problems.
Console Output
Environment
OS: Ubuntu 18.04 PyTorch: 1.10.1 CUDA: 10.2
Command to Reproduce
python infer.py --dataset_type=OpenImages --model_name=tresnet_l --model_path=ltresnet_v2_opim_87.34.pth --pic_path=test_img.jpg