End of training

README.md

@@ -23,7 +23,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.717948717948718
+      value: 0.7435897435897436
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -33,8 +33,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 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: 0.8872
-- Accuracy: 0.7179
+- Loss: 0.8173
+- Accuracy: 0.7436
 
 ## Model description
 
@@ -53,12 +53,12 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 4e-05
-- train_batch_size: 10
-- eval_batch_size: 10
+- learning_rate: 5e-05
+- train_batch_size: 8
+- eval_batch_size: 8
 - seed: 42
 - gradient_accumulation_steps: 2
-- total_train_batch_size: 20
+- total_train_batch_size: 16
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
@@ -69,21 +69,17 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 2.2117        | 1.0   | 35   | 2.1969          | 0.1923   |
-| 1.9698        | 2.0   | 70   | 1.9327          | 0.3846   |
-| 1.6629        | 3.0   | 105  | 1.5580          | 0.5      |
-| 1.2324        | 4.0   | 140  | 1.3368          | 0.6154   |
-| 1.0466        | 5.0   | 175  | 1.1638          | 0.6538   |
-| 0.8969        | 6.0   | 210  | 1.0416          | 0.6923   |
-| 0.7626        | 7.0   | 245  | 0.9258          | 0.7436   |
-| 0.6015        | 8.0   | 280  | 1.0475          | 0.6667   |
-| 0.5003        | 9.0   | 315  | 0.8890          | 0.7308   |
-| 0.3956        | 10.0  | 350  | 0.8396          | 0.7564   |
-| 0.3228        | 11.0  | 385  | 0.8072          | 0.6795   |
-| 0.2558        | 12.0  | 420  | 0.7788          | 0.7308   |
-| 0.1901        | 13.0  | 455  | 0.8432          | 0.7308   |
-| 0.1251        | 14.0  | 490  | 0.8287          | 0.7051   |
-| 0.1185        | 15.0  | 525  | 0.8872          | 0.7179   |
+| 2.1874        | 1.0   | 44   | 2.1429          | 0.3974   |
+| 1.8257        | 2.0   | 88   | 1.7390          | 0.4872   |
+| 1.4881        | 3.0   | 132  | 1.3711          | 0.6026   |
+| 1.0373        | 4.0   | 176  | 1.1632          | 0.6667   |
+| 0.7621        | 5.0   | 220  | 1.0026          | 0.7308   |
+| 0.6114        | 6.0   | 264  | 0.8857          | 0.7436   |
+| 0.5642        | 7.0   | 308  | 0.8796          | 0.7179   |
+| 0.3386        | 8.0   | 352  | 1.0714          | 0.6923   |
+| 0.3364        | 9.0   | 396  | 0.8363          | 0.7308   |
+| 0.1678        | 10.0  | 440  | 0.7834          | 0.7436   |
+| 0.1154        | 11.0  | 484  | 0.8173          | 0.7436   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:78b2024302e467190689d1630354b7351b009b9235da6a2cd7eb2e8bd7c07e2b
+oid sha256:44ac79fc98a21d10107f6fbdc7c27ce428c1b8689f3eb61ba15765937174964a
 size 94771728