Downloads · 30 days
6
7% of all-time downloads
vit-base-vocalsound
andrei-saceleanu/vit-base-vocalsound
vit-base-vocalsound is a image feature extraction model from andrei-saceleanu. Use it for the image feature extraction task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
This model is a fine-tuned version of google/vit-base-patch16-224 on VocalSound dataset. It achieves the following results on the evaluation set:
All-time downloads
88
Public
Repo size
346 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.h5346 MB · 100%
At a glance
- Task
- Image Feature Extraction
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- May 13, 2023
- Updated
- May 13, 2023
- SHA
- ecccc935
Model card
From the Hugging Face model README
vit-base-vocalsound
This model is a fine-tuned version of google/vit-base-patch16-224 on VocalSound dataset. It achieves the following results on the evaluation set:
- accuracy: 81.5
- precision (micro): 86.9
- recall (micro): 76.4
- f1 score (micro): 81.3
- f1 score (macro): 81.2
Training and evaluation data
Training: VocalSound training split (#samples = 15570)
Evaluation: VocalSound test split(#samples = 3594)
Training hyperparameters
The following hyperparameters were used during training:
- optimizer: AdamW
- weight_decay: 0
- learning_rate: 5e-5
- batch_size: 32
- training_precision: float32
Framework versions
- Transformers 4.27.4
- TensorFlow 2.12.0
- Tokenizers 0.13.3
- Task
- Image Feature Extraction
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- May 13, 2023
- Updated
- May 13, 2023