Downloads · 30 days
131
6% of all-time downloads
GenreVim-Music-Classification-DistilHuBERT
MarekCech/GenreVim-Music-Classification-DistilHuBERT
GenreVim-Music-Classification-DistilHuBERT is a audio classification model from MarekCech. Use it for the audio classification task on the model card, and read the license before you ship it in a product. It is set up for transformers.
This model is finetuned version of ntu-spml/distilhubert for music genre classification. - Blues - Classical music - Country music - Drum & Bass - Dubstep - Folk - Future Bass - Hardstyle - House - Jazz - Latin music…
All-time downloads
2.1K
Public
Parameters
23.7M
94.8 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors94.8 MB · 100%
At a glance
- Task
- Audio Classification
- Library
- transformers
- Model type
- hubert
- Access
- Public
- Created
- Jul 1, 2024
- Updated
- Aug 12, 2024
- SHA
- dd58d766
Model card
From the Hugging Face model README
This model is finetuned version of ntu-spml/distilhubert for music genre classification.
This model was finetuned to distinguish between 18 music genres including:
- Blues
- Classical music
- Country music
- Drum & Bass
- Dubstep
- Folk
- Future Bass
- Hardstyle
- House
- Jazz
- Latin music
- Metal
- Pop
- R&B, Soul
- Rap
- Reagge
- Rock
- Trap
Training hyperparameters
The following hyperparameters were used during training:
- metric_for_best_model = accuracy
- learning_rate = 5e-5
- seed = 42
- per_device_train_batch_size = 4
- per_device_eval_batch_size = 4
- gradient_accumulation_steps = 1
- warmup_ratio = 0.1
- fp16 = True
- adam_epsilon = 1e-08
- adam_beta1 = 0.9
- adam_beta2 = 0.999
- Task
- Audio Classification
- Library
- transformers
- Type
- hubert
- Created
- Jul 1, 2024
- Updated
- Aug 12, 2024