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.19k stars 457 forks source link

Linear evaluation #62

Open noah822 opened 1 year ago

noah822 commented 1 year ago

I followed the code provided in the notebook to do my own research. I found that the loaded checkpoint for cifar10 will project image into a 1000 dimension vector, which is not 512 as reported in README. Is this expected or my interpretation is incorrect.

To do linear eval, do we need to append a MLP, which first projects 1000-d vector to 512-d, interpreted as feature, then projects it to 128-d, followed by a 128-to-10 classifier.