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.19k stars 458 forks source link

another concise way to implement loss #23

Closed YinAoXiong closed 3 years ago

sthalles commented 3 years ago

Hello YinAoXiong, thanks for PR! I was already about to push the new changes to the project so I couldn't use it.

YinAoXiong commented 3 years ago

Hello YinAoXiong, thanks for PR! I was already about to push the new changes to the project so I couldn't use it.

I have seen the message, thanks for your work 😄