Revert back to best(last epoch) model.

README.md

@@ -21,7 +21,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.84
+      value: 0.87
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,8 +31,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.6100
-- Accuracy: 0.84
+- Loss: 0.5542
+- Accuracy: 0.87
 
 ## Model description
 
@@ -60,23 +60,24 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
 - lr_scheduler_warmup_ratio: 0.1
-- training_steps: 1237
+- num_epochs: 12
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 2.0099        | 1.0   | 112  | 1.8954          | 0.42     |
-| 1.4882        | 2.0   | 225  | 1.2842          | 0.68     |
-| 1.0753        | 3.0   | 337  | 0.9071          | 0.76     |
-| 0.7429        | 4.0   | 450  | 0.7317          | 0.81     |
-| 0.3772        | 5.0   | 562  | 0.7185          | 0.81     |
-| 0.3127        | 6.0   | 675  | 0.5813          | 0.84     |
-| 0.1909        | 7.0   | 787  | 0.6122          | 0.84     |
-| 0.1504        | 8.0   | 900  | 0.5366          | 0.86     |
-| 0.1015        | 9.0   | 1012 | 0.5833          | 0.85     |
-| 0.0612        | 10.0  | 1125 | 0.6030          | 0.84     |
-| 0.0772        | 11.0  | 1237 | 0.6100          | 0.84     |
+| 2.0241        | 1.0   | 112  | 1.9155          | 0.4      |
+| 1.5443        | 2.0   | 225  | 1.2937          | 0.65     |
+| 1.1134        | 3.0   | 337  | 0.9665          | 0.71     |
+| 0.7215        | 4.0   | 450  | 0.8201          | 0.74     |
+| 0.4679        | 5.0   | 562  | 0.7616          | 0.75     |
+| 0.3626        | 6.0   | 675  | 0.5217          | 0.85     |
+| 0.1775        | 7.0   | 787  | 0.6748          | 0.81     |
+| 0.1642        | 8.0   | 900  | 0.5287          | 0.86     |
+| 0.0772        | 9.0   | 1012 | 0.5632          | 0.84     |
+| 0.0478        | 10.0  | 1125 | 0.5576          | 0.85     |
+| 0.0662        | 11.0  | 1237 | 0.5455          | 0.88     |
+| 0.0446        | 11.95 | 1344 | 0.5542          | 0.87     |
 
 
 ### Framework versions

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c1638e252eb860d325f86393c0227b9b7380b28a3c527f992033065ad7e4f781
+oid sha256:a3dca215a079301e837cec4f5ceaa940c480013f12a5d7cb281e5b872c5f29f1
 size 94783376

runs/Jul06_12-46-19_susnato-desktop/events.out.tfevents.1688627787.susnato-desktop.74225.0

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4a200c20be1f2ec741c1f0e4341d551a391fcc47bb734368cc9bd2c6114cc233
-size 25271

runs/Jul06_14-12-36_susnato-desktop/events.out.tfevents.1688632965.susnato-desktop.82180.0

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5ec1a327e1a62364cfe8620c7304c0543703ddea49bc05a7a852a4de67d2bba1
-size 26651

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e48b5b7eb6f69d4410061924708717040a58b11ee676284ced0fe5354c5e257c
+oid sha256:7244a13234e6e833bf672f6950c63ea493de41482cc99a4b169eac19350a6d3e
 size 4027