salesforce / PCL

PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"
MIT License
570 stars 83 forks source link

about cifar10 #4

Closed Hiiragi0107 closed 3 years ago

Hiiragi0107 commented 4 years ago

Thank you for providing this code. My question is: what is the optimal number of clusters to train on cifar10?

LiJunnan1992 commented 3 years ago

Thanks for your interest.

I haven't experimented with cifar10, but I would assume over-clustering to be helpful, I would try 100<k<1000.

Hiiragi0107 commented 3 years ago

Thanks for the reply. :)

aakash-saboo commented 3 years ago

@Hiiragi0107 @LiJunnan1992

Did you get the results for CIFAR10?