File size: 780 Bytes
1c9704f 412f2de 31f01cf 412f2de 9993d7c 412f2de |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
---
base_model:
- openai/clip-vit-base-patch16
---
# Fine-Tuned ViT on FERPlus Dataset
This model is a fine-tuned version of ViT on the [FERPlus dataset](https://www.kaggle.com/datasets/arnabkumarroy02/ferplus), which is more balanced ferplus dataset as owner claimed.
## Model Details
- **Base Model**: [Clip-ViT-Base-Patch16](https://huggingface.co/openai/clip-vit-base-patch16)
- **Dataset**: FERPlus, which contains grayscale images of faces labeled with emotion categories.
- **Task**: Emotion Classification
- **Accuracy** %82.4
- **Labels**:
- 0: Angry
- 1: Contempt
- 2: Disgust
- 3: Fear
- 4: Happy
- 5: Neutral
- 6: Sad
- 7: Surprise
### Training Hyperparameters
- **Batch Size**: 16
- **Epochs**: 1
- **Learning Rate**: 2e-5
- **Weight Decay**: 0.01 |