The implementation using the trick of the momentum encoder

HobbitLong / SupContrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)

BSD 2-Clause "Simplified" License

3.08k stars 528 forks source link

The implementation using the trick of the momentum encoder #105

Open yinguoweiOvO opened 2 years ago

yinguoweiOvO commented 2 years ago

Thanks for your great paper and code. But I need to set a small batch due to the limitation of memory. Could you release the implementation which uses the trick of the momentum encoder? Thanks for your help.

sherlockers commented 2 years ago

Did you get the code？

yinguoweiOvO commented 2 years ago

Sorry, I haven't got the code.