Model save

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.0867
+- Loss: 0.0001
 - Accuracy: 1.0
 
 ## Model description
@@ -59,13 +59,16 @@ 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
+- num_epochs: 4
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.6464        | 1.0   | 8    | 0.0867          | 1.0      |
+| 0.8112        | 1.0   | 8    | 0.0346          | 1.0      |
+| 0.0108        | 2.0   | 16   | 0.0004          | 1.0      |
+| 0.0003        | 3.0   | 24   | 0.0001          | 1.0      |
+| 0.0001        | 4.0   | 32   | 0.0001          | 1.0      |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:76ede263deb51ddb571cb939ee22869881a5e53c79e113bd55d86b3b0b3ba9f3
+oid sha256:8d2a868a837699d288a34caa95cd130bfad367131798f68413b2ae72217ba0f2
 size 94766588