SupCon

Repository for implementation of Supervised Contrastive Learning.

files

• main.py: main procedure of experiment
• utils.py: defines functions and classes for descripted in main pocedure.
• (created automatically) ckpt, data, results: saves checkpoints, dataset, and results

arguments

common hyperparameters are selectable.