Issues with reproducing the results on STL10, ImageNet

AidenDurrant / SimCLR-Pytorch

An implementation of "A Simple Framework for Contrastive Learning of Visual Representatoins" SimCLR

31 stars 9 forks source link

Issues with reproducing the results on STL10, ImageNet #6

Open stoneMo opened 3 years ago

stoneMo commented 3 years ago

Hello Aiden,

I am using your code to reproduce the result on the STL10 dataset, but the performance does not make sense. I got 11.87% and 49.79% for top-1 and top-5 accuracy. It shows that the model does not learn anything. Did you reproduce the performance results on the STL10 dataset? Thank you.

Best, Shentong