search

RElbers / info-nce-pytorch

PyTorch implementation of the InfoNCE loss for self-supervised learning.
MIT License
445 stars 37 forks source link

Do you have infoNCE loss function cpde #5

Closed jiangxinke closed 2 years ago

RElbers commented 2 years ago

The code is in the init.py file.