HobbitLong / SupContrast

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

Make network weights available #77

Closed jnboehm closed 3 years ago

jnboehm commented 3 years ago

Would it be possible to make the trained network weights (on ImageNet or CIFAR, ideally both) available or are they available somewhere ? That'd make it easier to compare the own training process or just generally to reproduce the results.