search

sthalles / SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations
https://sthalles.github.io/simple-self-supervised-learning/
MIT License
2.21k stars 458 forks source link

About loss function #9

Closed leeh43 closed 3 years ago

leeh43 commented 4 years ago

Hi,

Thank you for the great work and I am trying to use your code in 3D patches. I separately input two paired datasets, which contain the domain difference and didn't use the data augmentation. I have extracted the representation using an encoder. However, the contrastive loss calculation is zero. Is there any steps that I haven't done to run the code successfully?

Thanks a lot!

sthalles commented 3 years ago

Hello guys, that is a general message to say that I have refactored the whole project. I believe the project is much easier to understand now. Please have a look at the new impl and free to submit PR if you find any bugs. Thanks.