ld76
/

wav2vec2-base-finetuned-gtzan-2

Audio Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

ld76 commited on Jul 27, 2023

Commit

3c5e176

•

1 Parent(s): 743c34f

Update README.md

Files changed (1) hide show

README.md +2 -5

README.md CHANGED Viewed

@@ -33,7 +33,7 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the GTZAN dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.7770
-- Accuracy: 0.82
 ## Model description
@@ -61,7 +61,7 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 14
 ### Training results
@@ -78,9 +78,6 @@ The following hyperparameters were used during training:
 | 0.3863        | 9.0   | 1012 | 0.5786          | 0.85     |
 | 0.4061        | 10.0  | 1125 | 0.7070          | 0.81     |
 | 0.1302        | 11.0  | 1237 | 0.5829          | 0.88     |
-| 0.0326        | 12.0  | 1350 | 0.7896          | 0.8      |
-| 0.0222        | 13.0  | 1462 | 0.8512          | 0.8      |
-| 0.2248        | 13.94 | 1568 | 0.7770          | 0.82     |
 ### Framework versions

 This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the GTZAN dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.7770
+- Accuracy: 0.88
 ## Model description
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
+- num_epochs: 11
 ### Training results
 | 0.3863        | 9.0   | 1012 | 0.5786          | 0.85     |
 | 0.4061        | 10.0  | 1125 | 0.7070          | 0.81     |
 | 0.1302        | 11.0  | 1237 | 0.5829          | 0.88     |
 ### Framework versions