marsyas/gtzan
Updated • 3.82k • 17
How to use susmitabhatt/my_awesome_mind_model with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("audio-classification", model="susmitabhatt/my_awesome_mind_model") # Load model directly
from transformers import AutoProcessor, AutoModelForAudioClassification
processor = AutoProcessor.from_pretrained("susmitabhatt/my_awesome_mind_model")
model = AutoModelForAudioClassification.from_pretrained("susmitabhatt/my_awesome_mind_model")This model is a fine-tuned version of facebook/wav2vec2-base 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 |
|---|---|---|---|---|
| No log | 0.96 | 6 | 2.2804 | 0.22 |
| 2.2742 | 1.92 | 12 | 2.1887 | 0.25 |
| 2.2742 | 2.88 | 18 | 2.1220 | 0.33 |
| 2.1487 | 4.0 | 25 | 2.0666 | 0.355 |
| 2.0354 | 4.96 | 31 | 1.9963 | 0.325 |
| 2.0354 | 5.92 | 37 | 1.9090 | 0.375 |
| 1.9246 | 6.88 | 43 | 1.8679 | 0.405 |
| 1.8416 | 8.0 | 50 | 1.8185 | 0.44 |
| 1.8416 | 8.96 | 56 | 1.7970 | 0.43 |
| 1.7858 | 9.6 | 60 | 1.7894 | 0.46 |