End of training

README.md

@@ -22,7 +22,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.81
+      value: 0.83
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +32,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: 1.6711
-- Accuracy: 0.81
+- Loss: 1.4965
+- Accuracy: 0.83
 
 ## Model description
 
@@ -59,13 +59,14 @@ 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: 1
+- training_steps: 10
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.0001        | 1.0   | 113  | 1.6711          | 0.81     |
+| 0.0           | 0.04  | 5    | 1.3686          | 0.81     |
+| 0.0           | 0.09  | 10   | 1.4965          | 0.83     |
 
 
 ### Framework versions

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4a29b0ea3ddc8a51e0229fe689406cda3fa88e473a9f640d4b0a0a2d520a031e
+oid sha256:92e16fc020beda7a3b5fdb66a9385ff35b5b927335f8e0dac37b5f8e44c71874
 size 94783376

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f904483a7086bcc20971463183792c9ebd1da55ff71fe44a4a490e9232542b51
+oid sha256:e5d124c5fa2f7850d7306365a6e391afaa9b1209e0a56dd31f9c84894548fc0b
 size 4091