End of training

README.md

@@ -6,9 +6,24 @@ tags:
 - generated_from_trainer
 datasets:
 - audiofolder
+metrics:
+- accuracy
 model-index:
 - name: distilhubert-finetuned-cry-detector
-  results: []
+  results:
+  - task:
+      name: Audio Classification
+      type: audio-classification
+    dataset:
+      name: audiofolder
+      type: audiofolder
+      config: default
+      split: None
+      args: default
+    metrics:
+    - name: Accuracy
+      type: accuracy
+      value: 0.9926739926739927
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -17,6 +32,9 @@ should probably proofread and complete it, then remove this comment. -->
 # distilhubert-finetuned-cry-detector
 
 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.0459
+- Accuracy: 0.9927
 
 ## Model description
 
@@ -44,13 +62,21 @@ 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.001
-- num_epochs: 1
+- num_epochs: 10
 
 ### Training results
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:--------:|
-| No log        | 0.9956 | 85   | 0.1008          | 0.9641   |
+| No log        | 0.9956 | 85   | 0.0692          | 0.9773   |
+| No log        | 1.9912 | 170  | 0.0466          | 0.9861   |
+| No log        | 2.9985 | 256  | 0.0489          | 0.9853   |
+| No log        | 3.9941 | 341  | 0.0423          | 0.9897   |
+| No log        | 4.9898 | 426  | 0.0443          | 0.9919   |
+| 0.055         | 5.9971 | 512  | 0.0434          | 0.9927   |
+| 0.055         | 6.9927 | 597  | 0.0440          | 0.9927   |
+| 0.055         | 8.0    | 683  | 0.0460          | 0.9927   |
+| 0.055         | 8.9956 | 768  | 0.0459          | 0.9927   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d7d7bbba5330741d7209ddb0c09280c4a121ee71f156203bf14f8362c8b4abb0
+oid sha256:3dc7074a340619ed7e6d2c55347500f679c4e5f5a30b51392562feb760a9f1c7
 size 94763496

runs/Sep03_16-11-40_ubumarcos/events.out.tfevents.1725372708.ubumarcos

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

runs/Sep03_16-12-14_ubumarcos/events.out.tfevents.1725372743.ubumarcos

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ab829fea11f8663161811c0f51125bfde969197b6312bba5ec51774dec5f2665
+oid sha256:abb08b4365c154a27d61d8cdfad62c4717bf003430323e6c3916f01c3121d721
 size 5240