PyTorch implementation of the InfoNCE loss for self-supervised learning.
Project description
PyTorch implementation of the InfoNCE loss from “Representation Learning with Contrastive Predictive Coding”.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
info-nce-pytorch-0.1.1.tar.gz
(3.1 kB
view hashes)