marsyas/gtzan
Updated • 1.75k • 17
How to use PawanKrGunjan/whisper-tiny-finetuned-gtzan with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("audio-classification", model="PawanKrGunjan/whisper-tiny-finetuned-gtzan") # Load model directly
from transformers import AutoProcessor, AutoModelForAudioClassification
processor = AutoProcessor.from_pretrained("PawanKrGunjan/whisper-tiny-finetuned-gtzan")
model = AutoModelForAudioClassification.from_pretrained("PawanKrGunjan/whisper-tiny-finetuned-gtzan")This model is a fine-tuned version of openai/whisper-tiny on the GTZAN dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|---|---|---|---|---|
| 2.3484 | 1.0 | 113 | 1.8521 | 0.26 |
| 1.9419 | 2.0 | 226 | 1.9107 | 0.3 |
| 1.8627 | 3.0 | 339 | 1.5300 | 0.49 |
| 1.8178 | 4.0 | 452 | 1.5152 | 0.41 |
| 1.5341 | 5.0 | 565 | 1.3365 | 0.53 |
Base model
openai/whisper-tiny