End of training

README.md

@@ -32,7 +32,7 @@ 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 audiofolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.9121
+- Loss: 1.1038
 - Accuracy: 0.6966
 
 ## Model description
@@ -53,10 +53,10 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 0.001
-- train_batch_size: 8
-- eval_batch_size: 8
+- train_batch_size: 4
+- eval_batch_size: 4
 - seed: 123
-- gradient_accumulation_steps: 8
+- gradient_accumulation_steps: 16
 - total_train_batch_size: 64
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
@@ -66,7 +66,7 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:--------:|
-| No log        | 0.9888 | 11   | 0.9121          | 0.6966   |
+| No log        | 0.9888 | 11   | 1.1038          | 0.6966   |
 
 
 ### Framework versions

runs/Jul03_17-26-22_DesMar/events.out.tfevents.1720020383.DesMar.14444.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7848477f4d3a6d693cb79e17a5115d2e78ce11bc3271822c9f7677a24f98c68c
-size 6073
+oid sha256:81a131d3e9c5076734fdd7bc4cf6f691759ba10fd13cb407a88934402cf59a5f
+size 6738

runs/Jul03_17-26-22_DesMar/events.out.tfevents.1720020506.DesMar.14444.1

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