search

RElbers / info-nce-pytorch

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

about loss parameter #6

Closed hrdxwandg closed 2 years ago

hrdxwandg commented 2 years ago

Is can set The temperature parameter？

hrdxwandg commented 2 years ago

location: https://github.com/RElbers/info-nce-pytorch/blob/main/info_nce/__init__.py