End of training

README.md

@@ -8,7 +8,7 @@ datasets:
 metrics:
 - accuracy
 model-index:
-- name: distilhubert-finetuned-gtzan-finetuned-gtzan
+- name: distilhubert-finetuned-gtzan
   results:
   - task:
       name: Audio Classification
@@ -22,18 +22,18 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.84
+      value: 0.83
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# distilhubert-finetuned-gtzan-finetuned-gtzan
+# distilhubert-finetuned-gtzan
 
 This model is a fine-tuned version of [gnuevo/distilhubert-finetuned-gtzan](https://huggingface.co/gnuevo/distilhubert-finetuned-gtzan) on the GTZAN dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6298
-- Accuracy: 0.84
+- Loss: 0.6408
+- Accuracy: 0.83
 
 ## Model description
 
@@ -52,13 +52,13 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 3e-05
+- learning_rate: 5e-05
 - train_batch_size: 16
 - eval_batch_size: 16
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- lr_scheduler_warmup_ratio: 0.3
+- lr_scheduler_warmup_ratio: 0.15
 - num_epochs: 10
 - mixed_precision_training: Native AMP
 
@@ -66,16 +66,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.4932        | 1.0   | 57   | 0.8500          | 0.86     |
-| 0.4537        | 2.0   | 114  | 0.7913          | 0.85     |
-| 0.3266        | 3.0   | 171  | 0.7242          | 0.85     |
-| 0.2155        | 4.0   | 228  | 0.6745          | 0.84     |
-| 0.1989        | 5.0   | 285  | 0.6484          | 0.84     |
-| 0.1271        | 6.0   | 342  | 0.6374          | 0.84     |
-| 0.0848        | 7.0   | 399  | 0.6300          | 0.84     |
-| 0.1622        | 8.0   | 456  | 0.6306          | 0.84     |
-| 0.1984        | 9.0   | 513  | 0.6298          | 0.83     |
-| 0.2795        | 10.0  | 570  | 0.6298          | 0.84     |
+| 2.2638        | 1.0   | 57   | 2.2585          | 0.17     |
+| 1.9343        | 2.0   | 114  | 1.9475          | 0.6      |
+| 1.2679        | 3.0   | 171  | 1.4426          | 0.7      |
+| 0.6692        | 4.0   | 228  | 1.0634          | 0.74     |
+| 0.4132        | 5.0   | 285  | 0.8203          | 0.82     |
+| 0.188         | 6.0   | 342  | 0.6990          | 0.85     |
+| 0.1295        | 7.0   | 399  | 0.6518          | 0.83     |
+| 0.2153        | 8.0   | 456  | 0.6430          | 0.84     |
+| 0.1812        | 9.0   | 513  | 0.6403          | 0.83     |
+| 0.3722        | 10.0  | 570  | 0.6408          | 0.83     |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b080fcdb639adf86247aafc9ef32fb9511b698ab576cfd7d2d15a33bb3997981
+oid sha256:0f1ebd589f5a0f5776fd8eb82ba34bc4c814e4c335b057750a62d4be6659134f
 size 95463244