RElbers/info-nce-pytorch

How to import info-nce-pytorch

Skywalker-Harrison opened this issue · 2 comments

Skywalker-Harrison commented

Hi, I am a new to contrastive learning, when I installed the package by using pip install info-nce-pytorch on my server, I could not import the package by using from info-nce-pytorch import InfoNCE(), could you tell me the name to import the package?
image

pulkitkumar95 commented

from info_nce import InfoNCE

Skywalker-Harrison commented

from info_nce import InfoNCE

Thank you!