HobbitLong / PyContrast

PyTorch implementation of Contrastive Learning methods
1.94k stars 185 forks source link

CMCv2 #28

Open irolaina opened 3 years ago

irolaina commented 3 years ago

Is there a pretrained CMCv2 model available for download somewhere? I would be very interested in that. I am assuming that the CMC.pth model in Dropbox is not that one, though a CMCv2 option does exist in datasets/util.py (aug == 'E').

Thank you for the great repo!