cantillation commited on
Commit
24b9c8f
1 Parent(s): f10e3dd

Model save

Browse files
Files changed (3) hide show
  1. README.md +102 -0
  2. generation_config.json +250 -0
  3. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: openai/whisper-tiny
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - wer
8
+ model-index:
9
+ - name: Teamim-tiny_Random-True_DropOut-0.1_WeightDecay-1e-06_Augmented_date-27-06-2024_11-10-07
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # Teamim-tiny_Random-True_DropOut-0.1_WeightDecay-1e-06_Augmented_date-27-06-2024_11-10-07
17
+
18
+ This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on an unknown dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.2552
21
+ - Wer: 16.1604
22
+ - Avg Precision Exact: 0.8115
23
+ - Avg Recall Exact: 0.8113
24
+ - Avg F1 Exact: 0.8110
25
+ - Avg Precision Letter Shift: 0.8369
26
+ - Avg Recall Letter Shift: 0.8369
27
+ - Avg F1 Letter Shift: 0.8365
28
+ - Avg Precision Word Level: 0.8414
29
+ - Avg Recall Word Level: 0.8413
30
+ - Avg F1 Word Level: 0.8409
31
+ - Avg Precision Word Shift: 0.9648
32
+ - Avg Recall Word Shift: 0.9659
33
+ - Avg F1 Word Shift: 0.9648
34
+ - Precision Median Exact: 0.9286
35
+ - Recall Median Exact: 0.9286
36
+ - F1 Median Exact: 0.9286
37
+ - Precision Max Exact: 1.0
38
+ - Recall Max Exact: 1.0
39
+ - F1 Max Exact: 1.0
40
+ - Precision Min Exact: 0.0
41
+ - Recall Min Exact: 0.0
42
+ - F1 Min Exact: 0.0
43
+ - Precision Min Letter Shift: 0.0
44
+ - Recall Min Letter Shift: 0.0
45
+ - F1 Min Letter Shift: 0.0
46
+ - Precision Min Word Level: 0.0
47
+ - Recall Min Word Level: 0.0
48
+ - F1 Min Word Level: 0.0
49
+ - Precision Min Word Shift: 0.0
50
+ - Recall Min Word Shift: 0.0
51
+ - F1 Min Word Shift: 0.0
52
+
53
+ ## Model description
54
+
55
+ More information needed
56
+
57
+ ## Intended uses & limitations
58
+
59
+ More information needed
60
+
61
+ ## Training and evaluation data
62
+
63
+ More information needed
64
+
65
+ ## Training procedure
66
+
67
+ ### Training hyperparameters
68
+
69
+ The following hyperparameters were used during training:
70
+ - learning_rate: 0.0001
71
+ - train_batch_size: 8
72
+ - eval_batch_size: 32
73
+ - seed: 42
74
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
75
+ - lr_scheduler_type: linear
76
+ - lr_scheduler_warmup_steps: 50
77
+ - training_steps: 100000
78
+ - mixed_precision_training: Native AMP
79
+
80
+ ### Training results
81
+
82
+ | Training Loss | Epoch | Step | Validation Loss | Wer | Avg Precision Exact | Avg Recall Exact | Avg F1 Exact | Avg Precision Letter Shift | Avg Recall Letter Shift | Avg F1 Letter Shift | Avg Precision Word Level | Avg Recall Word Level | Avg F1 Word Level | Avg Precision Word Shift | Avg Recall Word Shift | Avg F1 Word Shift | Precision Median Exact | Recall Median Exact | F1 Median Exact | Precision Max Exact | Recall Max Exact | F1 Max Exact | Precision Min Exact | Recall Min Exact | F1 Min Exact | Precision Min Letter Shift | Recall Min Letter Shift | F1 Min Letter Shift | Precision Min Word Level | Recall Min Word Level | F1 Min Word Level | Precision Min Word Shift | Recall Min Word Shift | F1 Min Word Shift |
83
+ |:-------------:|:-----:|:------:|:---------------:|:--------:|:-------------------:|:----------------:|:------------:|:--------------------------:|:-----------------------:|:-------------------:|:------------------------:|:---------------------:|:-----------------:|:------------------------:|:---------------------:|:-----------------:|:----------------------:|:-------------------:|:---------------:|:-------------------:|:----------------:|:------------:|:-------------------:|:----------------:|:------------:|:--------------------------:|:-----------------------:|:-------------------:|:------------------------:|:---------------------:|:-----------------:|:------------------------:|:---------------------:|:-----------------:|
84
+ | No log | 8e-05 | 1 | 7.9394 | 100.8130 | 0.0002 | 0.0009 | 0.0004 | 0.0043 | 0.0034 | 0.0034 | 0.0036 | 0.0204 | 0.0059 | 0.0398 | 0.0360 | 0.0353 | 0.0 | 0.0 | 0.0 | 0.1111 | 0.5 | 0.1538 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 |
85
+ | 0.0573 | 0.8 | 10000 | 0.1857 | 22.9823 | 0.7811 | 0.7770 | 0.7785 | 0.8163 | 0.8120 | 0.8135 | 0.8236 | 0.8193 | 0.8208 | 0.9437 | 0.9406 | 0.9414 | 0.9 | 0.8889 | 0.8889 | 1.0 | 1.0 | 1.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 |
86
+ | 0.0276 | 1.6 | 20000 | 0.2057 | 19.9224 | 0.8052 | 0.8105 | 0.8073 | 0.8336 | 0.8391 | 0.8357 | 0.8390 | 0.8441 | 0.8410 | 0.9451 | 0.9521 | 0.9478 | 0.9091 | 0.9167 | 0.9167 | 1.0 | 1.0 | 1.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 |
87
+ | 0.0238 | 2.4 | 30000 | 0.2099 | 19.0096 | 0.8151 | 0.8152 | 0.8147 | 0.8437 | 0.8440 | 0.8434 | 0.8491 | 0.8495 | 0.8488 | 0.9557 | 0.9579 | 0.9562 | 0.9167 | 0.9167 | 0.9167 | 1.0 | 1.0 | 1.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.1429 | 0.125 | 0.1333 |
88
+ | 0.0157 | 3.2 | 40000 | 0.2171 | 18.0414 | 0.8332 | 0.8294 | 0.8309 | 0.8616 | 0.8580 | 0.8593 | 0.8671 | 0.8641 | 0.8651 | 0.9607 | 0.9587 | 0.9591 | 0.9231 | 0.9231 | 0.9231 | 1.0 | 1.0 | 1.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0769 | 0.0714 | 0.0741 |
89
+ | 0.0061 | 4.0 | 50000 | 0.2365 | 17.8123 | 0.8326 | 0.8322 | 0.8320 | 0.8597 | 0.8592 | 0.8590 | 0.8658 | 0.8651 | 0.8650 | 0.9598 | 0.9604 | 0.9595 | 0.9286 | 0.9286 | 0.9286 | 1.0 | 1.0 | 1.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0833 | 0.0769 | 0.08 |
90
+ | 0.0038 | 4.8 | 60000 | 0.2350 | 17.3651 | 0.8243 | 0.8251 | 0.8242 | 0.8520 | 0.8529 | 0.8520 | 0.8568 | 0.8578 | 0.8568 | 0.9571 | 0.9600 | 0.9579 | 0.9231 | 0.9231 | 0.9286 | 1.0 | 1.0 | 1.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 |
91
+ | 0.002 | 5.6 | 70000 | 0.2404 | 17.0288 | 0.8319 | 0.8323 | 0.8317 | 0.8592 | 0.8595 | 0.8589 | 0.8649 | 0.8652 | 0.8645 | 0.9609 | 0.9620 | 0.9608 | 0.9286 | 0.9286 | 0.9310 | 1.0 | 1.0 | 1.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 |
92
+ | 0.002 | 6.4 | 80000 | 0.2439 | 17.0251 | 0.8116 | 0.8127 | 0.8117 | 0.8368 | 0.8379 | 0.8369 | 0.8420 | 0.8429 | 0.8420 | 0.9567 | 0.9592 | 0.9573 | 0.9231 | 0.9258 | 0.9286 | 1.0 | 1.0 | 1.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 |
93
+ | 0.001 | 7.2 | 90000 | 0.2548 | 16.8995 | 0.8106 | 0.8102 | 0.8100 | 0.8368 | 0.8364 | 0.8362 | 0.8421 | 0.8414 | 0.8413 | 0.9575 | 0.9579 | 0.9571 | 0.9231 | 0.9231 | 0.9286 | 1.0 | 1.0 | 1.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 |
94
+ | 0.0001 | 8.0 | 100000 | 0.2552 | 16.1604 | 0.8115 | 0.8113 | 0.8110 | 0.8369 | 0.8369 | 0.8365 | 0.8414 | 0.8413 | 0.8409 | 0.9648 | 0.9659 | 0.9648 | 0.9286 | 0.9286 | 0.9286 | 1.0 | 1.0 | 1.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 |
95
+
96
+
97
+ ### Framework versions
98
+
99
+ - Transformers 4.41.2
100
+ - Pytorch 2.2.1
101
+ - Datasets 2.20.0
102
+ - Tokenizers 0.19.1
generation_config.json ADDED
@@ -0,0 +1,250 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alignment_heads": [
3
+ [
4
+ 2,
5
+ 2
6
+ ],
7
+ [
8
+ 3,
9
+ 0
10
+ ],
11
+ [
12
+ 3,
13
+ 2
14
+ ],
15
+ [
16
+ 3,
17
+ 3
18
+ ],
19
+ [
20
+ 3,
21
+ 4
22
+ ],
23
+ [
24
+ 3,
25
+ 5
26
+ ]
27
+ ],
28
+ "begin_suppress_tokens": [
29
+ 220,
30
+ 50257
31
+ ],
32
+ "bos_token_id": 50257,
33
+ "decoder_start_token_id": 50258,
34
+ "eos_token_id": 50257,
35
+ "forced_decoder_ids": [
36
+ [
37
+ 1,
38
+ null
39
+ ],
40
+ [
41
+ 2,
42
+ 50359
43
+ ]
44
+ ],
45
+ "is_multilingual": true,
46
+ "lang_to_id": {
47
+ "<|af|>": 50327,
48
+ "<|am|>": 50334,
49
+ "<|ar|>": 50272,
50
+ "<|as|>": 50350,
51
+ "<|az|>": 50304,
52
+ "<|ba|>": 50355,
53
+ "<|be|>": 50330,
54
+ "<|bg|>": 50292,
55
+ "<|bn|>": 50302,
56
+ "<|bo|>": 50347,
57
+ "<|br|>": 50309,
58
+ "<|bs|>": 50315,
59
+ "<|ca|>": 50270,
60
+ "<|cs|>": 50283,
61
+ "<|cy|>": 50297,
62
+ "<|da|>": 50285,
63
+ "<|de|>": 50261,
64
+ "<|el|>": 50281,
65
+ "<|en|>": 50259,
66
+ "<|es|>": 50262,
67
+ "<|et|>": 50307,
68
+ "<|eu|>": 50310,
69
+ "<|fa|>": 50300,
70
+ "<|fi|>": 50277,
71
+ "<|fo|>": 50338,
72
+ "<|fr|>": 50265,
73
+ "<|gl|>": 50319,
74
+ "<|gu|>": 50333,
75
+ "<|haw|>": 50352,
76
+ "<|ha|>": 50354,
77
+ "<|he|>": 50279,
78
+ "<|hi|>": 50276,
79
+ "<|hr|>": 50291,
80
+ "<|ht|>": 50339,
81
+ "<|hu|>": 50286,
82
+ "<|hy|>": 50312,
83
+ "<|id|>": 50275,
84
+ "<|is|>": 50311,
85
+ "<|it|>": 50274,
86
+ "<|ja|>": 50266,
87
+ "<|jw|>": 50356,
88
+ "<|ka|>": 50329,
89
+ "<|kk|>": 50316,
90
+ "<|km|>": 50323,
91
+ "<|kn|>": 50306,
92
+ "<|ko|>": 50264,
93
+ "<|la|>": 50294,
94
+ "<|lb|>": 50345,
95
+ "<|ln|>": 50353,
96
+ "<|lo|>": 50336,
97
+ "<|lt|>": 50293,
98
+ "<|lv|>": 50301,
99
+ "<|mg|>": 50349,
100
+ "<|mi|>": 50295,
101
+ "<|mk|>": 50308,
102
+ "<|ml|>": 50296,
103
+ "<|mn|>": 50314,
104
+ "<|mr|>": 50320,
105
+ "<|ms|>": 50282,
106
+ "<|mt|>": 50343,
107
+ "<|my|>": 50346,
108
+ "<|ne|>": 50313,
109
+ "<|nl|>": 50271,
110
+ "<|nn|>": 50342,
111
+ "<|no|>": 50288,
112
+ "<|oc|>": 50328,
113
+ "<|pa|>": 50321,
114
+ "<|pl|>": 50269,
115
+ "<|ps|>": 50340,
116
+ "<|pt|>": 50267,
117
+ "<|ro|>": 50284,
118
+ "<|ru|>": 50263,
119
+ "<|sa|>": 50344,
120
+ "<|sd|>": 50332,
121
+ "<|si|>": 50322,
122
+ "<|sk|>": 50298,
123
+ "<|sl|>": 50305,
124
+ "<|sn|>": 50324,
125
+ "<|so|>": 50326,
126
+ "<|sq|>": 50317,
127
+ "<|sr|>": 50303,
128
+ "<|su|>": 50357,
129
+ "<|sv|>": 50273,
130
+ "<|sw|>": 50318,
131
+ "<|ta|>": 50287,
132
+ "<|te|>": 50299,
133
+ "<|tg|>": 50331,
134
+ "<|th|>": 50289,
135
+ "<|tk|>": 50341,
136
+ "<|tl|>": 50348,
137
+ "<|tr|>": 50268,
138
+ "<|tt|>": 50351,
139
+ "<|uk|>": 50280,
140
+ "<|ur|>": 50290,
141
+ "<|uz|>": 50337,
142
+ "<|vi|>": 50278,
143
+ "<|yi|>": 50335,
144
+ "<|yo|>": 50325,
145
+ "<|zh|>": 50260
146
+ },
147
+ "language": "he",
148
+ "max_initial_timestamp_index": 50,
149
+ "max_length": 448,
150
+ "no_timestamps_token_id": 50363,
151
+ "pad_token_id": 50257,
152
+ "prev_sot_token_id": 50361,
153
+ "return_timestamps": false,
154
+ "suppress_tokens": [
155
+ 1,
156
+ 2,
157
+ 7,
158
+ 8,
159
+ 9,
160
+ 10,
161
+ 14,
162
+ 25,
163
+ 26,
164
+ 27,
165
+ 28,
166
+ 29,
167
+ 31,
168
+ 58,
169
+ 59,
170
+ 60,
171
+ 61,
172
+ 62,
173
+ 63,
174
+ 90,
175
+ 91,
176
+ 92,
177
+ 93,
178
+ 359,
179
+ 503,
180
+ 522,
181
+ 542,
182
+ 873,
183
+ 893,
184
+ 902,
185
+ 918,
186
+ 922,
187
+ 931,
188
+ 1350,
189
+ 1853,
190
+ 1982,
191
+ 2460,
192
+ 2627,
193
+ 3246,
194
+ 3253,
195
+ 3268,
196
+ 3536,
197
+ 3846,
198
+ 3961,
199
+ 4183,
200
+ 4667,
201
+ 6585,
202
+ 6647,
203
+ 7273,
204
+ 9061,
205
+ 9383,
206
+ 10428,
207
+ 10929,
208
+ 11938,
209
+ 12033,
210
+ 12331,
211
+ 12562,
212
+ 13793,
213
+ 14157,
214
+ 14635,
215
+ 15265,
216
+ 15618,
217
+ 16553,
218
+ 16604,
219
+ 18362,
220
+ 18956,
221
+ 20075,
222
+ 21675,
223
+ 22520,
224
+ 26130,
225
+ 26161,
226
+ 26435,
227
+ 28279,
228
+ 29464,
229
+ 31650,
230
+ 32302,
231
+ 32470,
232
+ 36865,
233
+ 42863,
234
+ 47425,
235
+ 49870,
236
+ 50254,
237
+ 50258,
238
+ 50358,
239
+ 50359,
240
+ 50360,
241
+ 50361,
242
+ 50362
243
+ ],
244
+ "task_to_id": {
245
+ "transcribe": 50359,
246
+ "translate": 50358
247
+ },
248
+ "transformers_version": "4.41.2",
249
+ "use_cache": false
250
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4a5bce2560f83f2d55071db2045440c94c20c3080fa5d70b09bb56569e98fdbb
3
  size 151109288
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8eec63da2a465bb5b0841a6107cccc8da5e63d03a2a6705f1545f8930f56236f
3
  size 151109288