Downloads · 30 days
10
1% of all-time downloads
supervised-contrastive-learning-cifar10
keras-io/supervised-contrastive-learning-cifar10
supervised-contrastive-learning-cifar10 is a image classification model from keras-io. Use it when you need a label for an image. It is set up for tf-keras. The card lists the license as apache-2.0.
A classification model trained with <a href='https://arxiv.org/abs/2004.11362' target='blank'Supervised Contrastive Learning</a (Prannay Khosla et al.). The training procedure was done as seen in the example on <a hre…
All-time downloads
996
Public
Repo size
227 MB
Likes
3
Public
Repo makeup
Click a slice to open those files.
.data-00000-of-00001107 MB · 94%
At a glance
- Task
- Image Classification
- Library
- tf-keras
- License
- apache-2.0
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jul 5, 2024
- SHA
- 326f3bb0
Datasets
Model card
From the Hugging Face model README
A classification model trained with <a href='https://arxiv.org/abs/2004.11362' target='_blank'>Supervised Contrastive Learning</a> (Prannay Khosla et al.). 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.
The model was trained on cifar10, which includes ten classes: airplane, automobile, bird, cat, deer, dog, frog, horse, ship, truck.
The test accuracy after 50 epochs of the model with contrastive learning was 81.06% (opposed to 79.88% without contrastive learning).
- Task
- Image Classification
- Library
- tf-keras
- License
- apache-2.0
- Created
- Mar 2, 2022
- Updated
- Jul 5, 2024