Run experiments on ImageNet

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

Run experiments on ImageNet #2

Closed Abyssaledge closed 3 years ago

Abyssaledge commented 4 years ago

Hi, Thanks for your nice work. I am planning to run SimCLR on ImageNet dataset. I wonder if I need to adjust the structure of network or add some tricks. For example increasing the dimension of output used to calculating losses, which in your code is 64. Or directly change the dataset to ImageNet and keep other configuration the same? I'll appreciate any advice.

Jessejx commented 3 years ago

HI, have you finished it? I am planning to SimCLR on ImageNet dataset too.

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.