End of training

README.md

@@ -1,6 +1,6 @@
 ---
 license: apache-2.0
-base_model: ntu-spml/distilhubert
+base_model: chaouch/distilhubert-finetuned-gtzan
 tags:
 - generated_from_trainer
 datasets:
@@ -8,7 +8,7 @@ datasets:
 metrics:
 - accuracy
 model-index:
-- name: distilhubert-finetuned-gtzan2
+- name: distilhubert-finetuned-gtzan-finetuned-gtzan
   results:
   - task:
       name: Audio Classification
@@ -22,18 +22,18 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.7125
+      value: 0.9666666666666667
 ---
 
 <!-- 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-gtzan2
+# distilhubert-finetuned-gtzan-finetuned-gtzan
 
-This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the GTZAN dataset.
+This model is a fine-tuned version of [chaouch/distilhubert-finetuned-gtzan](https://huggingface.co/chaouch/distilhubert-finetuned-gtzan) on the GTZAN dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.5220
-- Accuracy: 0.7125
+- Loss: 0.1733
+- Accuracy: 0.9667
 
 ## Model description
 
@@ -52,35 +52,27 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.001
-- train_batch_size: 32
-- eval_batch_size: 32
+- learning_rate: 0.0001
+- train_batch_size: 6
+- eval_batch_size: 6
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 15
+- num_epochs: 7
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.7489        | 1.0   | 29   | 1.4959          | 0.3875   |
-| 1.328         | 2.0   | 58   | 2.0243          | 0.35     |
-| 1.2168        | 3.0   | 87   | 1.1332          | 0.5875   |
-| 1.0299        | 4.0   | 116  | 1.4826          | 0.5375   |
-| 0.911         | 5.0   | 145  | 1.2510          | 0.625    |
-| 1.0819        | 6.0   | 174  | 1.7365          | 0.55     |
-| 0.9513        | 7.0   | 203  | 1.3000          | 0.6      |
-| 0.5687        | 8.0   | 232  | 1.0503          | 0.7125   |
-| 0.4684        | 9.0   | 261  | 1.1167          | 0.7125   |
-| 0.2836        | 10.0  | 290  | 1.5990          | 0.65     |
-| 0.138         | 11.0  | 319  | 1.2096          | 0.7375   |
-| 0.0406        | 12.0  | 348  | 1.7311          | 0.6375   |
-| 0.0341        | 13.0  | 377  | 1.7048          | 0.6375   |
-| 0.0059        | 14.0  | 406  | 1.4933          | 0.7      |
-| 0.0034        | 15.0  | 435  | 1.5220          | 0.7125   |
+| 0.026         | 1.0   | 135  | 0.2289          | 0.9444   |
+| 0.1351        | 2.0   | 270  | 0.1379          | 0.9778   |
+| 0.01          | 3.0   | 405  | 0.2310          | 0.9667   |
+| 0.0053        | 4.0   | 540  | 0.1727          | 0.9667   |
+| 0.0002        | 5.0   | 675  | 0.1703          | 0.9667   |
+| 0.0002        | 6.0   | 810  | 0.1722          | 0.9667   |
+| 0.0002        | 7.0   | 945  | 0.1733          | 0.9667   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5e8ffdb667fed855e84a4d45f364d9deacae6daf07cf98f6c3d6c25f66f1d04f
+oid sha256:17396ed3157e82be6b93f4785e57371e45578248b3c0f0e12d10687d0d129f4a
 size 94771728