fontadie commited on
Commit
0c7bd2a
1 Parent(s): e273afa

Model save

Browse files
README.md ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: openai/whisper-tiny.en
4
+ tags:
5
+ - generated_from_trainer
6
+ datasets:
7
+ - tedlium
8
+ metrics:
9
+ - wer
10
+ model-index:
11
+ - name: whisper-tiny-openslrdev
12
+ results:
13
+ - task:
14
+ name: Automatic Speech Recognition
15
+ type: automatic-speech-recognition
16
+ dataset:
17
+ name: tedlium
18
+ type: tedlium
19
+ config: release1
20
+ split: test
21
+ args: release1
22
+ metrics:
23
+ - name: Wer
24
+ type: wer
25
+ value: 90.4153910381297
26
+ ---
27
+
28
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
29
+ should probably proofread and complete it, then remove this comment. -->
30
+
31
+ # whisper-tiny-openslrdev
32
+
33
+ This model is a fine-tuned version of [openai/whisper-tiny.en](https://huggingface.co/openai/whisper-tiny.en) on the tedlium dataset.
34
+ It achieves the following results on the evaluation set:
35
+ - Loss: 2.0820
36
+ - Wer: 90.4154
37
+
38
+ ## Model description
39
+
40
+ More information needed
41
+
42
+ ## Intended uses & limitations
43
+
44
+ More information needed
45
+
46
+ ## Training and evaluation data
47
+
48
+ More information needed
49
+
50
+ ## Training procedure
51
+
52
+ ### Training hyperparameters
53
+
54
+ The following hyperparameters were used during training:
55
+ - learning_rate: 1e-05
56
+ - train_batch_size: 16
57
+ - eval_batch_size: 16
58
+ - seed: 42
59
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
60
+ - lr_scheduler_type: linear
61
+ - lr_scheduler_warmup_steps: 200
62
+ - training_steps: 300
63
+ - mixed_precision_training: Native AMP
64
+
65
+ ### Training results
66
+
67
+ | Training Loss | Epoch | Step | Validation Loss | Wer |
68
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|
69
+ | No log | 0.06 | 20 | 3.7027 | 35.3291 |
70
+ | 3.9098 | 0.13 | 40 | 3.3264 | 35.0647 |
71
+ | 3.0852 | 0.19 | 60 | 2.9769 | 34.0871 |
72
+ | 2.2682 | 0.26 | 80 | 2.7802 | 31.6309 |
73
+ | 1.6662 | 0.32 | 100 | 2.5284 | 27.7728 |
74
+ | 1.6662 | 0.38 | 120 | 2.4481 | 24.3668 |
75
+ | 1.2505 | 0.45 | 140 | 2.4118 | 21.6532 |
76
+ | 1.0859 | 0.51 | 160 | 2.3687 | 20.9087 |
77
+ | 0.9491 | 0.58 | 180 | 2.1924 | 19.6493 |
78
+ | 0.8746 | 0.64 | 200 | 2.1752 | 22.1229 |
79
+ | 0.8746 | 0.7 | 220 | 2.2546 | 29.7245 |
80
+ | 0.8064 | 0.77 | 240 | 2.1611 | 39.6326 |
81
+ | 0.733 | 0.83 | 260 | 2.1281 | 55.7334 |
82
+ | 0.7135 | 0.89 | 280 | 2.0406 | 75.1705 |
83
+ | 0.6806 | 0.96 | 300 | 2.0820 | 90.4154 |
84
+
85
+
86
+ ### Framework versions
87
+
88
+ - Transformers 4.39.2
89
+ - Pytorch 2.2.2+cu121
90
+ - Datasets 2.18.0
91
+ - Tokenizers 0.15.2
generation_config.json ADDED
@@ -0,0 +1,149 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alignment_heads": [
3
+ [
4
+ 1,
5
+ 0
6
+ ],
7
+ [
8
+ 2,
9
+ 0
10
+ ],
11
+ [
12
+ 2,
13
+ 5
14
+ ],
15
+ [
16
+ 3,
17
+ 0
18
+ ],
19
+ [
20
+ 3,
21
+ 1
22
+ ],
23
+ [
24
+ 3,
25
+ 2
26
+ ],
27
+ [
28
+ 3,
29
+ 3
30
+ ],
31
+ [
32
+ 3,
33
+ 4
34
+ ]
35
+ ],
36
+ "begin_suppress_tokens": [
37
+ 220,
38
+ 50256
39
+ ],
40
+ "bos_token_id": 50257,
41
+ "decoder_start_token_id": 50257,
42
+ "eos_token_id": 50256,
43
+ "forced_decoder_ids": [
44
+ [
45
+ 1,
46
+ 50362
47
+ ]
48
+ ],
49
+ "is_multilingual": false,
50
+ "max_initial_timestamp_index": 50,
51
+ "max_length": 448,
52
+ "no_timestamps_token_id": 50362,
53
+ "pad_token_id": 50256,
54
+ "prev_sot_token_id": 50360,
55
+ "return_timestamps": false,
56
+ "suppress_tokens": [
57
+ 1,
58
+ 2,
59
+ 7,
60
+ 8,
61
+ 9,
62
+ 10,
63
+ 14,
64
+ 25,
65
+ 26,
66
+ 27,
67
+ 28,
68
+ 29,
69
+ 31,
70
+ 58,
71
+ 59,
72
+ 60,
73
+ 61,
74
+ 62,
75
+ 63,
76
+ 90,
77
+ 91,
78
+ 92,
79
+ 93,
80
+ 357,
81
+ 366,
82
+ 438,
83
+ 532,
84
+ 685,
85
+ 705,
86
+ 796,
87
+ 930,
88
+ 1058,
89
+ 1220,
90
+ 1267,
91
+ 1279,
92
+ 1303,
93
+ 1343,
94
+ 1377,
95
+ 1391,
96
+ 1635,
97
+ 1782,
98
+ 1875,
99
+ 2162,
100
+ 2361,
101
+ 2488,
102
+ 3467,
103
+ 4008,
104
+ 4211,
105
+ 4600,
106
+ 4808,
107
+ 5299,
108
+ 5855,
109
+ 6329,
110
+ 7203,
111
+ 9609,
112
+ 9959,
113
+ 10563,
114
+ 10786,
115
+ 11420,
116
+ 11709,
117
+ 11907,
118
+ 13163,
119
+ 13697,
120
+ 13700,
121
+ 14808,
122
+ 15306,
123
+ 16410,
124
+ 16791,
125
+ 17992,
126
+ 19203,
127
+ 19510,
128
+ 20724,
129
+ 22305,
130
+ 22935,
131
+ 27007,
132
+ 30109,
133
+ 30420,
134
+ 33409,
135
+ 34949,
136
+ 40283,
137
+ 40493,
138
+ 40549,
139
+ 47282,
140
+ 49146,
141
+ 50257,
142
+ 50357,
143
+ 50358,
144
+ 50359,
145
+ 50360,
146
+ 50361
147
+ ],
148
+ "transformers_version": "4.39.2"
149
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e4e4d404dd3d57de8b0d0944a550cea3342812e8551c36f6878eb16fdd5eb52c
3
  size 151060136
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a6e9c37be358f522da249e821aadbc27c0b9fed47a093a1a2af2661d400182e
3
  size 151060136
runs/May05_18-43-42_DESKTOP-OHECI79/events.out.tfevents.1714927422.DESKTOP-OHECI79.26784.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4cd1b6f52be5fd96fadebb0f7992aca39183f8e939fc2590539853567878a418
3
- size 11106
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c54e2af128dce1f641618f7a064141f916e1e2e0421fd2a634b887f806f424c
3
+ size 13047