Wiam commited on
Commit
74680d5
1 Parent(s): 6b71174

End of training

Browse files
README.md ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: ntu-spml/distilhubert
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ - f1
10
+ - precision
11
+ - recall
12
+ model-index:
13
+ - name: distilhubert-finetuned-babycry-v7
14
+ results: []
15
+ ---
16
+
17
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
18
+ should probably proofread and complete it, then remove this comment. -->
19
+
20
+ # distilhubert-finetuned-babycry-v7
21
+
22
+ This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on an unknown dataset.
23
+ It achieves the following results on the evaluation set:
24
+ - Loss: 0.5860
25
+ - Accuracy: {'accuracy': 0.8695652173913043}
26
+ - F1: 0.8089
27
+ - Precision: 0.7561
28
+ - Recall: 0.8696
29
+
30
+ ## Model description
31
+
32
+ More information needed
33
+
34
+ ## Intended uses & limitations
35
+
36
+ More information needed
37
+
38
+ ## Training and evaluation data
39
+
40
+ More information needed
41
+
42
+ ## Training procedure
43
+
44
+ ### Training hyperparameters
45
+
46
+ The following hyperparameters were used during training:
47
+ - learning_rate: 0.001
48
+ - train_batch_size: 4
49
+ - eval_batch_size: 4
50
+ - seed: 42
51
+ - gradient_accumulation_steps: 2
52
+ - total_train_batch_size: 8
53
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
54
+ - lr_scheduler_type: cosine
55
+ - lr_scheduler_warmup_ratio: 0.03
56
+ - num_epochs: 8
57
+
58
+ ### Training results
59
+
60
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
61
+ |:-------------:|:------:|:----:|:---------------:|:--------------------------------:|:------:|:---------:|:------:|
62
+ | 1.005 | 0.5435 | 25 | 0.6526 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
63
+ | 0.7736 | 1.0870 | 50 | 0.6396 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
64
+ | 0.5618 | 1.6304 | 75 | 0.6990 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
65
+ | 0.8623 | 2.1739 | 100 | 0.5802 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
66
+ | 0.6532 | 2.7174 | 125 | 0.6205 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
67
+ | 0.8076 | 3.2609 | 150 | 0.6168 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
68
+ | 0.7581 | 3.8043 | 175 | 0.5917 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
69
+ | 0.7525 | 4.3478 | 200 | 0.5988 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
70
+ | 0.6566 | 4.8913 | 225 | 0.5997 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
71
+ | 0.6845 | 5.4348 | 250 | 0.5815 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
72
+ | 0.6812 | 5.9783 | 275 | 0.5830 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
73
+ | 0.6548 | 6.5217 | 300 | 0.5855 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
74
+ | 0.6555 | 7.0652 | 325 | 0.5859 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
75
+ | 0.7294 | 7.6087 | 350 | 0.5861 | {'accuracy': 0.8695652173913043} | 0.8089 | 0.7561 | 0.8696 |
76
+
77
+
78
+ ### Framework versions
79
+
80
+ - Transformers 4.44.2
81
+ - Pytorch 2.4.1+cu121
82
+ - Tokenizers 0.19.1
runs/Oct02_17-13-51_315dc4c4c032/events.out.tfevents.1727889368.315dc4c4c032.673.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7043e713ebf75f33b9c14d30ca62a883706d43ef8cf529b79429ea4645cb0e34
3
+ size 508