renato145 / ContrastiveLoss

Supervised Constrastive Loss implementation using fastai+pytorch

Supervised Constrastive Loss

• Paper: https://arxiv.org/abs/2004.11362
• Original code: https://github.com/google-research/google-research/tree/master/supcon
• Official pytorch code: https://github.com/HobbitLong/SupContrast

Implementation and visualizations using fastai+pytorch.

Notes:

• An explanation for the loss function can be found on cifar10.ipynb