search

HobbitLong / SupContrast

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

Execution of code for ImageNet code (Moco trick) #134

Open Angelina1996 opened 1 year ago

Angelina1996 commented 1 year ago

Dear author, thanks so much for the brilliant work. Could you guide me in the execution of code for the Imagenet dataset?

Cheers

HobbitLong commented 1 year ago

Hi, you may just need to combine the momentum encoder idea from the MoCo code, with the SupCon loss here. Our paper did not use the MoCo trick, tho.