Supervised-Contrastive-Learning-in-TensorFlow-2 icon indicating copy to clipboard operation
Supervised-Contrastive-Learning-in-TensorFlow-2 copied to clipboard

Implements the ideas presented in https://arxiv.org/pdf/2004.11362v1.pdf by Khosla et al.

Results 2 Supervised-Contrastive-Learning-in-TensorFlow-2 issues
Sort by recently updated
recently updated
newest added

Hi I want first thank you for your good work. I have been modifying the notebook to run on my own dataset. Size is close to 1000 samples, 5 classes....

You didn't use the loss function introduced in the original paper. Do you know how I can use that? Do you have it in a special place?