search

sthalles / SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations
https://sthalles.github.io/simple-self-supervised-learning/
MIT License
2.21k stars 458 forks source link

Loading pretrained model weights #17

Closed kunal-savvy closed 3 years ago

kunal-savvy commented 3 years ago

The code uses state_dict = torch.loadfor pretrained model but I was not able to get it to use pretrained weights for resnets. Any suggestions?

sthalles commented 3 years ago

Hello guys, that is a general message to say that I have refactored the whole project. I believe the project is much easier to understand now. Please have a look at the new impl and free to submit PR if you find any bugs. Thanks.