Closed deBrian07 closed 8 months ago
Do you use my provided checkpoints? They are only for ResNet18. If you are retraining the models, pretrain from ImageNet pretrain weights.
Yes, I'm trying to use slfm-pretext.py to retrain the models with resnet50, and the errors occurred.
Have you had the opportunity to troubleshoot by yourself in order to identify the specific line of code causing this issue? I suspect it's the imagenet pretraining loading issue. Please modify it to meet the torchvision package.
I have tested it on my end, I don't have this issue. I assume it's likely because of the pytorch and torchvision environment update. Please try to fix this by yourself or use the provided environment yml file.
Hello, I was trying the preset training with resnet-50. However, the error shows that there's a key not found error, and a size mismatch error. The only thing I changed was in slfm-pretext.py. I've attached the screenshots below. Please take a look. Thank you so much!