search

giakoumoglou / pyssl

Self-Supervised Learning in PyTorch
MIT License
126 stars 9 forks source link

SimCLRv2 memory bank #2

Open sfandres opened 1 year ago

sfandres commented 1 year ago

I looked at the SimCRLv2 implementation and I don't see the memory bank that you implement in the version given in the paper, which is also explained in your post on Medium. Is that correct? I wanted to use your library but I need the full implementation of that method.

giakoumoglou commented 1 year ago

Hi and thank you for your comment. Indeed, the implementation does not include a memory bank, rather it calculates the loss using the mini batch.