Wav2vec2 Base 100k Gtzan Music Genres Finetuned Wav2vec2 Ivan

🚀 wav2vec2-base-100k-gtzan-music-genres-finetuned-wav2vec2-ivan

This model is a fine - tuned version of m3hrdadfi/wav2vec2-base-100k-gtzan-music-genres on the GTZAN dataset. It offers high - accuracy audio classification, achieving an accuracy of 0.98 on the evaluation set.

🚀 Quick Start

This model is designed for audio classification tasks. You can use it to classify music genres based on the GTZAN dataset.

📚 Documentation

Model Information

Model Performance

This model achieves the following results on the evaluation set:

• Loss: 1.2147
• Accuracy: 0.98

Training Procedure

Training Hyperparameters

The following hyperparameters were used during training:

• learning_rate: 3e - 05
• train_batch_size: 8
• eval_batch_size: 8
• seed: 42
• optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon = 1e - 08 and optimizer_args = No additional optimizer arguments
• lr_scheduler_type: linear
• lr_scheduler_warmup_ratio: 0.1
• num_epochs: 5
• mixed_precision_training: Native AMP

Training Results

Framework Versions

• Transformers 4.47.1
• Pytorch 2.5.1+cu121
• Datasets 3.2.0
• Tokenizers 0.21.0