Update README.md

README.md

@@ -22,18 +22,19 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.83
+      value: 0.87
 ---
 
-<!-- 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
 
 This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the GTZAN dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.1893
-- Accuracy: 0.83
+- Loss: 0.8249
+- Accuracy: 0.87
+
+The initial model trained for 20 epochs and overfit, so I recovered the best epoch (10) and pushed to hub. The metrics above reflect the latest model from epoch 10/checkpoint 2250.
 
 ## Model description
 
@@ -75,16 +76,7 @@ The following hyperparameters were used during training:
 | 0.09          | 8.0   | 1800 | 1.0860          | 0.82     |
 | 0.0031        | 9.0   | 2025 | 0.7744          | 0.84     |
 | 0.0026        | 10.0  | 2250 | 0.8249          | 0.87     |
-| 0.0032        | 11.0  | 2475 | 1.0680          | 0.84     |
-| 0.0012        | 12.0  | 2700 | 1.0724          | 0.83     |
-| 0.0011        | 13.0  | 2925 | 1.1407          | 0.83     |
-| 0.0009        | 14.0  | 3150 | 1.0395          | 0.85     |
-| 0.0007        | 15.0  | 3375 | 1.2991          | 0.83     |
-| 0.0006        | 16.0  | 3600 | 1.1403          | 0.83     |
-| 0.0007        | 17.0  | 3825 | 1.0837          | 0.83     |
-| 0.0005        | 18.0  | 4050 | 1.1463          | 0.83     |
-| 0.0005        | 19.0  | 4275 | 1.1987          | 0.83     |
-| 0.0005        | 20.0  | 4500 | 1.1893          | 0.83     |
+
 
 
 ### Framework versions