search

HobbitLong / SupContrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)
BSD 2-Clause "Simplified" License
3.12k stars 537 forks source link

Pre-trained model #143

Open tanushreebanerjee opened 10 months ago

tanushreebanerjee commented 10 months ago

Hi!

Is it possible for you to share the pretrained SupContrast model at all? Would be super useful!

Thanks!

HobbitLong commented 10 months ago

Hi @tanushreebanerjee ,

Do you mean the ImageNet model? If so, you could find it in the update section.

If you meant the CIFAR-10/100 models, sry that I did not keep those ckpts (as I thought it would not be hard to reproduce).