search

HobbitLong / CMC

[arXiv 2019] "Contrastive Multiview Coding", also contains implementations for MoCo and InstDis
BSD 2-Clause "Simplified" License
1.3k stars 179 forks source link

Pre-trained weights for linear classifier available? #47

Closed rgeirhos closed 4 years ago

rgeirhos commented 4 years ago

Hey there, thanks for the well-documented code!

Quick question: Am I correctly assuming that in order to evaluate the model on the 1,000-class ImageNet validation dataset one has to train the linear classifier first (using LinearProbing.py)? If so, would it be possible to release pre-trained weights for the classifier as well, such that one can use classifier.load_state_dict(checkpoint['classifier'])?

HobbitLong commented 4 years ago

@rgeirhos, currently not available. It will take several hours to train a new one.

rgeirhos commented 4 years ago

Thanks for your reply.