johko commited on
Commit
67a745a
·
1 Parent(s): 34e20b0

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -0
README.md ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ A classification model trained with <a href='https://arxiv.org/abs/2004.11362' target='_blank'>**Supervised Contrastive Learning**</a> (Prannay Khosla et al.).
2
+ The training procedure was done as seen in the example on <a href='https://keras.io/examples/vision/supervised-contrastive-learning/' target='_blank'>**keras.io**</a> by Khalid Salama.
3
+
4
+ The model was **trained on cifar10**, which includes ten classes: airplane, automobile, bird, cat, deer, dog, frog, horse, ship, truck.
5
+
6
+ The **test accuracy after 50 epochs** of the model with contrastive learning was **81.06%** (opposed to 79.88% without contrastive learning).