search

HobbitLong / SupContrast

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

Pre-trained model #143

Open tanushreebanerjee opened 5 months ago

tanushreebanerjee commented 5 months ago

Hi!

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

Thanks!

HobbitLong commented 5 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).