Supervised-Contrastive-Learning-in-TensorFlow-2
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?