End of training

README.md

@@ -1,6 +1,6 @@
 ---
 license: apache-2.0
-base_model: lighteternal/wav2vec2-large-xlsr-53-greek
+base_model: ntu-spml/distilhubert
 tags:
 - generated_from_trainer
 datasets:
@@ -30,9 +30,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # wav2vec2-xlsr-finetuned-gtzan
 
-This model is a fine-tuned version of [lighteternal/wav2vec2-large-xlsr-53-greek](https://huggingface.co/lighteternal/wav2vec2-large-xlsr-53-greek) on the GTZAN dataset.
+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.6511
+- Loss: 0.4910
 - Accuracy: 0.88
 
 ## Model description
@@ -61,23 +61,21 @@ 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: 10
+- num_epochs: 8
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 2.1788        | 1.0   | 112  | 1.9921          | 0.29     |
-| 1.4318        | 2.0   | 225  | 1.3264          | 0.49     |
-| 1.422         | 3.0   | 337  | 0.9604          | 0.65     |
-| 0.7665        | 4.0   | 450  | 0.7403          | 0.76     |
-| 0.6839        | 5.0   | 562  | 0.5957          | 0.83     |
-| 0.459         | 6.0   | 675  | 0.5525          | 0.83     |
-| 0.139         | 7.0   | 787  | 0.6561          | 0.86     |
-| 0.4184        | 8.0   | 900  | 0.6185          | 0.85     |
-| 0.1972        | 9.0   | 1012 | 0.5794          | 0.89     |
-| 0.2296        | 9.96  | 1120 | 0.6511          | 0.88     |
+| 2.1164        | 1.0   | 112  | 1.8109          | 0.41     |
+| 1.4074        | 2.0   | 225  | 1.3693          | 0.52     |
+| 1.3318        | 3.0   | 337  | 1.0215          | 0.67     |
+| 0.753         | 4.0   | 450  | 0.7795          | 0.76     |
+| 0.6883        | 5.0   | 562  | 0.7275          | 0.8      |
+| 0.3683        | 6.0   | 675  | 0.5863          | 0.82     |
+| 0.2497        | 7.0   | 787  | 0.4621          | 0.89     |
+| 0.4483        | 7.96  | 896  | 0.4910          | 0.88     |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:aedbaa4a2130e5ba9d636ee678fcf7036d5676fb46062d3e81a1faee0ed9092b
+oid sha256:a46f6d95f564694fa4f42abc4c0137251a67ae87d1df24749a11be48dc9be744
 size 1266047104

runs/Jan14_17-22-04_f65dd54c3f1e/events.out.tfevents.1705252925.f65dd54c3f1e.526.0

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ba3892dee01b2842d29aba802c4ca5747f7e63a7b668f10038490b68215e88b2
+oid sha256:2c6d0640066272dd87649e2d3f420351373fdc4e01d98f0fa539add5dd2d3cc1
 size 4728