How to import info-nce-pytorch

RElbers / info-nce-pytorch

PyTorch implementation of the InfoNCE loss for self-supervised learning.

MIT License

445 stars 37 forks source link

How to import info-nce-pytorch #14

Closed Skywalker-Harrison closed 9 months ago

Skywalker-Harrison commented 10 months ago

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?

pulkitkumar95 commented 9 months ago

from info_nce import InfoNCE

Skywalker-Harrison commented 9 months ago

from info_nce import InfoNCE

Thank you!