Hugging Face's logo

Marcos12886
/
distilhubert-finetuned-cry-detector

Audio Classification
Transformers
TensorBoard
Safetensors
hubert
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
Marcos12886 commited on
Commit
aa34cd9
•
1 Parent(s): 5d2701e

End of training

Browse files
Files changed (27) hide show
  1. README.md +5 -34
  2. config.json +3 -2
  3. model.safetensors +1 -1
  4. runs/Sep05_13-01-54_ubumarcos/events.out.tfevents.1725534121.ubumarcos +3 -0
  5. runs/Sep05_17-04-12_ubumarcos/events.out.tfevents.1725548654.ubumarcos +3 -0
  6. runs/Sep05_17-06-43_ubumarcos/events.out.tfevents.1725548806.ubumarcos +3 -0
  7. runs/Sep05_17-08-32_ubumarcos/events.out.tfevents.1725548916.ubumarcos +3 -0
  8. runs/Sep05_17-09-04_ubumarcos/events.out.tfevents.1725548950.ubumarcos +3 -0
  9. runs/Sep05_17-09-41_ubumarcos/events.out.tfevents.1725548987.ubumarcos +3 -0
  10. runs/Sep05_17-13-25_ubumarcos/events.out.tfevents.1725549211.ubumarcos +3 -0
  11. runs/Sep05_17-14-17_ubumarcos/events.out.tfevents.1725549258.ubumarcos +3 -0
  12. runs/Sep05_17-20-02_ubumarcos/events.out.tfevents.1725549603.ubumarcos +3 -0
  13. runs/Sep05_17-44-32_ubumarcos/events.out.tfevents.1725551074.ubumarcos +3 -0
  14. runs/Sep05_17-52-09_ubumarcos/events.out.tfevents.1725551530.ubumarcos +3 -0
  15. runs/Sep05_18-05-46_ubumarcos/events.out.tfevents.1725552348.ubumarcos +3 -0
  16. runs/Sep05_18-08-28_ubumarcos/events.out.tfevents.1725552509.ubumarcos +3 -0
  17. runs/Sep05_18-12-36_ubumarcos/events.out.tfevents.1725552757.ubumarcos +3 -0
  18. runs/Sep05_18-15-13_ubumarcos/events.out.tfevents.1725552914.ubumarcos +3 -0
  19. runs/Sep05_18-16-38_ubumarcos/events.out.tfevents.1725552999.ubumarcos +3 -0
  20. runs/Sep05_19-02-30_ubumarcos/events.out.tfevents.1725555751.ubumarcos +3 -0
  21. runs/Sep05_19-02-55_ubumarcos/events.out.tfevents.1725555776.ubumarcos +3 -0
  22. runs/Sep05_19-13-46_ubumarcos/events.out.tfevents.1725556427.ubumarcos +3 -0
  23. runs/Sep05_19-30-17_ubumarcos/events.out.tfevents.1725557418.ubumarcos +3 -0
  24. runs/Sep05_19-37-49_ubumarcos/events.out.tfevents.1725557870.ubumarcos +3 -0
  25. runs/Sep05_19-38-16_ubumarcos/events.out.tfevents.1725557897.ubumarcos +3 -0
  26. runs/Sep05_19-43-42_ubumarcos/events.out.tfevents.1725558223.ubumarcos +3 -0
  27. training_args.bin +1 -1
README.md CHANGED
@@ -4,26 +4,9 @@ license: apache-2.0
4
  base_model: ntu-spml/distilhubert
5
  tags:
6
  - generated_from_trainer
7
- datasets:
8
- - audiofolder
9
- metrics:
10
- - accuracy
11
  model-index:
12
  - name: distilhubert-finetuned-cry-detector
13
- results:
14
- - task:
15
- name: Audio Classification
16
- type: audio-classification
17
- dataset:
18
- name: audiofolder
19
- type: audiofolder
20
- config: default
21
- split: train
22
- args: default
23
- metrics:
24
- - name: Accuracy
25
- type: accuracy
26
- value: 0.9926739926739927
27
  ---
28
 
29
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,10 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
31
 
32
  # distilhubert-finetuned-cry-detector
33
 
34
- This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the audiofolder dataset.
35
- It achieves the following results on the evaluation set:
36
- - Loss: 0.0459
37
- - Accuracy: 0.9927
38
 
39
  ## Model description
40
 
@@ -62,26 +42,17 @@ The following hyperparameters were used during training:
62
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
63
  - lr_scheduler_type: cosine
64
  - lr_scheduler_warmup_ratio: 0.001
65
- - num_epochs: 10
66
 
67
  ### Training results
68
 
69
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
70
  |:-------------:|:------:|:----:|:---------------:|:--------:|
71
- | No log | 0.9956 | 85 | 0.0692 | 0.9773 |
72
- | No log | 1.9912 | 170 | 0.0466 | 0.9861 |
73
- | No log | 2.9985 | 256 | 0.0489 | 0.9853 |
74
- | No log | 3.9941 | 341 | 0.0423 | 0.9897 |
75
- | No log | 4.9898 | 426 | 0.0443 | 0.9919 |
76
- | 0.055 | 5.9971 | 512 | 0.0434 | 0.9927 |
77
- | 0.055 | 6.9927 | 597 | 0.0440 | 0.9927 |
78
- | 0.055 | 8.0 | 683 | 0.0460 | 0.9927 |
79
- | 0.055 | 8.9956 | 768 | 0.0459 | 0.9927 |
80
 
81
 
82
  ### Framework versions
83
 
84
  - Transformers 4.44.2
85
- - Pytorch 2.4.0+cu121
86
- - Datasets 2.21.0
87
  - Tokenizers 0.19.1
 
4
  base_model: ntu-spml/distilhubert
5
  tags:
6
  - generated_from_trainer
 
 
 
 
7
  model-index:
8
  - name: distilhubert-finetuned-cry-detector
9
+ results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  ---
11
 
12
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
14
 
15
  # distilhubert-finetuned-cry-detector
16
 
17
+ This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on an unknown dataset.
 
 
 
18
 
19
  ## Model description
20
 
 
42
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
  - lr_scheduler_type: cosine
44
  - lr_scheduler_warmup_ratio: 0.001
45
+ - num_epochs: 1
46
 
47
  ### Training results
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:------:|:----:|:---------------:|:--------:|
51
+ | No log | 0.9956 | 85 | nan | 0.7114 |
 
 
 
 
 
 
 
 
52
 
53
 
54
  ### Framework versions
55
 
56
  - Transformers 4.44.2
57
+ - Pytorch 2.4.1+cu121
 
58
  - Tokenizers 0.19.1
config.json CHANGED
@@ -45,6 +45,7 @@
45
  "feat_proj_dropout": 0.0,
46
  "feat_proj_layer_norm": false,
47
  "final_dropout": 0.0,
 
48
  "hidden_act": "gelu",
49
  "hidden_dropout": 0.1,
50
  "hidden_size": 768,
@@ -55,8 +56,8 @@
55
  "initializer_range": 0.02,
56
  "intermediate_size": 3072,
57
  "label2id": {
58
- "crying": "0",
59
- "no_crying": "1"
60
  },
61
  "layer_norm_eps": 1e-05,
62
  "layerdrop": 0.0,
 
45
  "feat_proj_dropout": 0.0,
46
  "feat_proj_layer_norm": false,
47
  "final_dropout": 0.0,
48
+ "finetuning_task": "audio-classification",
49
  "hidden_act": "gelu",
50
  "hidden_dropout": 0.1,
51
  "hidden_size": 768,
 
56
  "initializer_range": 0.02,
57
  "intermediate_size": 3072,
58
  "label2id": {
59
+ "crying": 0,
60
+ "no_crying": 1
61
  },
62
  "layer_norm_eps": 1e-05,
63
  "layerdrop": 0.0,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3dc7074a340619ed7e6d2c55347500f679c4e5f5a30b51392562feb760a9f1c7
3
  size 94763496
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cd019309b99ca09903e80ceb76fc22c519638fcc4b6eebcb72b2a3ef350c1b52
3
  size 94763496
runs/Sep05_13-01-54_ubumarcos/events.out.tfevents.1725534121.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c3ab1392ad3528b09f2d980d36eba5b4634a31039f5a196fb23593424526e42
3
+ size 5855
runs/Sep05_17-04-12_ubumarcos/events.out.tfevents.1725548654.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d14b6ef52b454973c46c0f92b5e1c611357b8340cfd121fc400dfd4337033fd5
3
+ size 5851
runs/Sep05_17-06-43_ubumarcos/events.out.tfevents.1725548806.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a573d5d1ea50aed05c3bdcb4e5a8adb6e968d8ba02fdc2f391dff8fc82994f7
3
+ size 5851
runs/Sep05_17-08-32_ubumarcos/events.out.tfevents.1725548916.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:55ffc0276842f674ce9cb487788ae648be4b2245858bbaa2b3b02aa4e590c848
3
+ size 5851
runs/Sep05_17-09-04_ubumarcos/events.out.tfevents.1725548950.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f628fc8860a3d3eb778cde5a38c2c7d51c02c1867a8fedb792cb0450f473fe3
3
+ size 5851
runs/Sep05_17-09-41_ubumarcos/events.out.tfevents.1725548987.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:445b5bf83a4c93d0a605c8d1237280678eab314dc1d2849195edf991062bbdea
3
+ size 5851
runs/Sep05_17-13-25_ubumarcos/events.out.tfevents.1725549211.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd910066f9f4f73410724a5256e88731701adcb421f25ccfa13875dea797d3d5
3
+ size 5851
runs/Sep05_17-14-17_ubumarcos/events.out.tfevents.1725549258.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:64824851aa77328a3747d6e127f3d33fa036134b794c53adef07ab057921fcb1
3
+ size 6116
runs/Sep05_17-20-02_ubumarcos/events.out.tfevents.1725549603.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3501a0f6dbc94451e24eef48a0cdc57bb3e45e7e7944b40d8e5b86c274086177
3
+ size 5850
runs/Sep05_17-44-32_ubumarcos/events.out.tfevents.1725551074.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:836bee0014f47fe3039e533509b74a85e9746adc8746d2f6e7f499060438d95b
3
+ size 5850
runs/Sep05_17-52-09_ubumarcos/events.out.tfevents.1725551530.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca77db79d6e05aebc674d60611e3a41be0950a5f495d65eb5605d8b5d774bb04
3
+ size 6116
runs/Sep05_18-05-46_ubumarcos/events.out.tfevents.1725552348.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d939829a328b4d3d78c6eb18e334972c6fc467e767d8a7b6e692e7ca734644b
3
+ size 6116
runs/Sep05_18-08-28_ubumarcos/events.out.tfevents.1725552509.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5cd63b42fa4fb60644792051dbaec1e29db0e3dc459cf18f0185b744a758db2
3
+ size 5850
runs/Sep05_18-12-36_ubumarcos/events.out.tfevents.1725552757.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c67a8915815f527738766bd0766a2c6c406687e7a0c3bb70d97588d9ed5f17c
3
+ size 5850
runs/Sep05_18-15-13_ubumarcos/events.out.tfevents.1725552914.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32254e2e3536f221f6e36183169cb1e62790d7f4a47895118b142ecb78884717
3
+ size 5850
runs/Sep05_18-16-38_ubumarcos/events.out.tfevents.1725552999.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:605f09c5ccded948e707374916dba1441769a010dd5bead921cec35d66e28793
3
+ size 5850
runs/Sep05_19-02-30_ubumarcos/events.out.tfevents.1725555751.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d828bad56bb817c5113a18696ef46e0a8173bf90bd9efd23b35ba2aaab847f3
3
+ size 5850
runs/Sep05_19-02-55_ubumarcos/events.out.tfevents.1725555776.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6cea79005a6bfd43b6e281ac757a70871c5e805b6df568703d4f82362d328a8e
3
+ size 5850
runs/Sep05_19-13-46_ubumarcos/events.out.tfevents.1725556427.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1af408d4ded2f34b8775366a79853d8b11426901ced0dd7cadbba8c794c49cd6
3
+ size 5850
runs/Sep05_19-30-17_ubumarcos/events.out.tfevents.1725557418.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:33878c792398c2d1125dec345e97bf6b8fe32074ea5868234785c263bab87979
3
+ size 5850
runs/Sep05_19-37-49_ubumarcos/events.out.tfevents.1725557870.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:014f6bcb08047ea27940889cf8a7970ff21f57086d08c2ca4ce067414df2c5a4
3
+ size 5850
runs/Sep05_19-38-16_ubumarcos/events.out.tfevents.1725557897.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6cb5473b30ef3822a6292ad298d6a12b001ec7e5a6d03bd04805bd6c50cb401
3
+ size 5850
runs/Sep05_19-43-42_ubumarcos/events.out.tfevents.1725558223.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f31b72d8f41464488c8da9157489e8a288f0d5b695be6c94ca240fb463cf6a9
3
+ size 6515
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:84d763bc52a548d7900e672478766167dea5015bedfc4ad077e5cffda9256cc8
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3bd8a5d913ab65031a96ccd1a869b8fc2c6bee0dae599113f8a9697cb2b8aae
3
  size 5240