├── LICENSE ├── README.md ├── data_augmentation ├── __init__.py ├── auto_augment.py └── duplicate_sample_transform.py ├── images └── representation_explained.gif ├── loss ├── __init__.py └── spc.py ├── models ├── __init__.py └── resnet_contrastive.py ├── requirements.txt ├── test └── test_loss.py ├── train.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuillaumeErhard/Supervised_contrastive_loss_pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuillaumeErhard/Supervised_contrastive_loss_pytorch/HEAD/README.md -------------------------------------------------------------------------------- /data_augmentation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_augmentation/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuillaumeErhard/Supervised_contrastive_loss_pytorch/HEAD/data_augmentation/auto_augment.py -------------------------------------------------------------------------------- /data_augmentation/duplicate_sample_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuillaumeErhard/Supervised_contrastive_loss_pytorch/HEAD/data_augmentation/duplicate_sample_transform.py -------------------------------------------------------------------------------- /images/representation_explained.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuillaumeErhard/Supervised_contrastive_loss_pytorch/HEAD/images/representation_explained.gif -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /loss/spc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuillaumeErhard/Supervised_contrastive_loss_pytorch/HEAD/loss/spc.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/resnet_contrastive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuillaumeErhard/Supervised_contrastive_loss_pytorch/HEAD/models/resnet_contrastive.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuillaumeErhard/Supervised_contrastive_loss_pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /test/test_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuillaumeErhard/Supervised_contrastive_loss_pytorch/HEAD/test/test_loss.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuillaumeErhard/Supervised_contrastive_loss_pytorch/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuillaumeErhard/Supervised_contrastive_loss_pytorch/HEAD/utils.py --------------------------------------------------------------------------------