End of training

README.md

@@ -17,13 +17,13 @@ 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:
-- eval_loss: 2.3098
-- eval_accuracy: 0.22
-- eval_runtime: 49.5115
-- eval_samples_per_second: 2.02
-- eval_steps_per_second: 0.263
-- epoch: 0.98
-- step: 37
+- eval_loss: 0.5861
+- eval_accuracy: 0.87
+- eval_runtime: 56.9025
+- eval_samples_per_second: 1.757
+- eval_steps_per_second: 0.228
+- epoch: 12.0
+- step: 1356
 
 ## Model description
 
@@ -42,21 +42,19 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 7e-05
+- learning_rate: 5e-05
 - train_batch_size: 8
 - eval_batch_size: 8
 - seed: 42
-- gradient_accumulation_steps: 1.5
-- total_train_batch_size: 12.0
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 13
+- num_epochs: 15
 - mixed_precision_training: Native AMP
 
 ### Framework versions
 
-- Transformers 4.38.0.dev0
+- Transformers 4.35.2
 - Pytorch 2.1.0+cu121
 - Datasets 2.16.1
 - Tokenizers 0.15.1

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:48a4ed474c7841ee1ce5940339a3055e885e6c00e8e36a93351da63430aa949c
+oid sha256:73c34882471e2fdcebcb346c835dd1ffcab1a3deeb861af007dcae0ca8df3514
 size 94771728

runs/Jan30_14-25-21_1a33fb1dc461/events.out.tfevents.1706624823.1a33fb1dc461.373.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f656cac10d7f1447cda799813d5df6c643fba2f6c99cdd6c7d1895cb5895556b
-size 51773
+oid sha256:973a8e233334099e88d7280470f940ac5e7623c27bd2ad762c25b97ecfbf2cf3
+size 53657