gweltou commited on
Commit
7e79252
1 Parent(s): 94dec05

End of training

Browse files
README.md ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: openai/whisper-small
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - wer
8
+ model-index:
9
+ - name: whisper-small-br
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
+ # whisper-small-br
17
+
18
+ This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the None dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.5767
21
+ - Wer: 39.9748
22
+ - Cer: 15.0329
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - learning_rate: 2e-05
42
+ - train_batch_size: 8
43
+ - eval_batch_size: 8
44
+ - seed: 42
45
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
+ - lr_scheduler_type: linear
47
+ - lr_scheduler_warmup_steps: 200
48
+ - training_steps: 4000
49
+ - mixed_precision_training: Native AMP
50
+
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Validation Loss | Wer | Cer |
54
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
55
+ | 0.782 | 0.58 | 500 | 0.7847 | 61.4497 | 24.5285 |
56
+ | 0.3209 | 1.16 | 1000 | 0.6244 | 47.0028 | 17.7797 |
57
+ | 0.3041 | 1.74 | 1500 | 0.5578 | 45.1182 | 18.4874 |
58
+ | 0.1177 | 2.33 | 2000 | 0.5479 | 42.1620 | 16.4081 |
59
+ | 0.1234 | 2.91 | 2500 | 0.5353 | 41.6136 | 15.9008 |
60
+ | 0.0371 | 3.49 | 3000 | 0.5593 | 39.1428 | 14.7689 |
61
+ | 0.02 | 4.07 | 3500 | 0.5714 | 38.8591 | 14.7176 |
62
+ | 0.0115 | 4.65 | 4000 | 0.5767 | 39.9748 | 15.0329 |
63
+
64
+
65
+ ### Framework versions
66
+
67
+ - Transformers 4.39.1
68
+ - Pytorch 2.0.1+cu117
69
+ - Datasets 2.18.0
70
+ - Tokenizers 0.15.2
bck-checkpoint-3500/config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "openai/whisper-small",
3
+ "activation_dropout": 0.0,
4
+ "activation_function": "gelu",
5
+ "apply_spec_augment": false,
6
+ "architectures": [
7
+ "WhisperForConditionalGeneration"
8
+ ],
9
+ "attention_dropout": 0.0,
10
+ "begin_suppress_tokens": [
11
+ 220,
12
+ 50257
13
+ ],
14
+ "bos_token_id": 50257,
15
+ "classifier_proj_size": 256,
16
+ "d_model": 768,
17
+ "decoder_attention_heads": 12,
18
+ "decoder_ffn_dim": 3072,
19
+ "decoder_layerdrop": 0.0,
20
+ "decoder_layers": 12,
21
+ "decoder_start_token_id": 50258,
22
+ "dropout": 0.0,
23
+ "encoder_attention_heads": 12,
24
+ "encoder_ffn_dim": 3072,
25
+ "encoder_layerdrop": 0.0,
26
+ "encoder_layers": 12,
27
+ "eos_token_id": 50257,
28
+ "forced_decoder_ids": null,
29
+ "init_std": 0.02,
30
+ "is_encoder_decoder": true,
31
+ "mask_feature_length": 10,
32
+ "mask_feature_min_masks": 0,
33
+ "mask_feature_prob": 0.0,
34
+ "mask_time_length": 10,
35
+ "mask_time_min_masks": 2,
36
+ "mask_time_prob": 0.05,
37
+ "max_length": 448,
38
+ "max_source_positions": 1500,
39
+ "max_target_positions": 448,
40
+ "median_filter_width": 7,
41
+ "model_type": "whisper",
42
+ "num_hidden_layers": 12,
43
+ "num_mel_bins": 80,
44
+ "pad_token_id": 50257,
45
+ "scale_embedding": false,
46
+ "suppress_tokens": [],
47
+ "torch_dtype": "float32",
48
+ "transformers_version": "4.39.1",
49
+ "use_cache": false,
50
+ "use_weighted_layer_sum": false,
51
+ "vocab_size": 51865
52
+ }
bck-checkpoint-3500/generation_config.json ADDED
@@ -0,0 +1,266 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alignment_heads": [
3
+ [
4
+ 5,
5
+ 3
6
+ ],
7
+ [
8
+ 5,
9
+ 9
10
+ ],
11
+ [
12
+ 8,
13
+ 0
14
+ ],
15
+ [
16
+ 8,
17
+ 4
18
+ ],
19
+ [
20
+ 8,
21
+ 7
22
+ ],
23
+ [
24
+ 8,
25
+ 8
26
+ ],
27
+ [
28
+ 9,
29
+ 0
30
+ ],
31
+ [
32
+ 9,
33
+ 7
34
+ ],
35
+ [
36
+ 9,
37
+ 9
38
+ ],
39
+ [
40
+ 10,
41
+ 5
42
+ ]
43
+ ],
44
+ "begin_suppress_tokens": [
45
+ 220,
46
+ 50257
47
+ ],
48
+ "bos_token_id": 50257,
49
+ "decoder_start_token_id": 50258,
50
+ "eos_token_id": 50257,
51
+ "forced_decoder_ids": [
52
+ [
53
+ 1,
54
+ null
55
+ ],
56
+ [
57
+ 2,
58
+ 50359
59
+ ]
60
+ ],
61
+ "is_multilingual": true,
62
+ "lang_to_id": {
63
+ "<|af|>": 50327,
64
+ "<|am|>": 50334,
65
+ "<|ar|>": 50272,
66
+ "<|as|>": 50350,
67
+ "<|az|>": 50304,
68
+ "<|ba|>": 50355,
69
+ "<|be|>": 50330,
70
+ "<|bg|>": 50292,
71
+ "<|bn|>": 50302,
72
+ "<|bo|>": 50347,
73
+ "<|br|>": 50309,
74
+ "<|bs|>": 50315,
75
+ "<|ca|>": 50270,
76
+ "<|cs|>": 50283,
77
+ "<|cy|>": 50297,
78
+ "<|da|>": 50285,
79
+ "<|de|>": 50261,
80
+ "<|el|>": 50281,
81
+ "<|en|>": 50259,
82
+ "<|es|>": 50262,
83
+ "<|et|>": 50307,
84
+ "<|eu|>": 50310,
85
+ "<|fa|>": 50300,
86
+ "<|fi|>": 50277,
87
+ "<|fo|>": 50338,
88
+ "<|fr|>": 50265,
89
+ "<|gl|>": 50319,
90
+ "<|gu|>": 50333,
91
+ "<|haw|>": 50352,
92
+ "<|ha|>": 50354,
93
+ "<|he|>": 50279,
94
+ "<|hi|>": 50276,
95
+ "<|hr|>": 50291,
96
+ "<|ht|>": 50339,
97
+ "<|hu|>": 50286,
98
+ "<|hy|>": 50312,
99
+ "<|id|>": 50275,
100
+ "<|is|>": 50311,
101
+ "<|it|>": 50274,
102
+ "<|ja|>": 50266,
103
+ "<|jw|>": 50356,
104
+ "<|ka|>": 50329,
105
+ "<|kk|>": 50316,
106
+ "<|km|>": 50323,
107
+ "<|kn|>": 50306,
108
+ "<|ko|>": 50264,
109
+ "<|la|>": 50294,
110
+ "<|lb|>": 50345,
111
+ "<|ln|>": 50353,
112
+ "<|lo|>": 50336,
113
+ "<|lt|>": 50293,
114
+ "<|lv|>": 50301,
115
+ "<|mg|>": 50349,
116
+ "<|mi|>": 50295,
117
+ "<|mk|>": 50308,
118
+ "<|ml|>": 50296,
119
+ "<|mn|>": 50314,
120
+ "<|mr|>": 50320,
121
+ "<|ms|>": 50282,
122
+ "<|mt|>": 50343,
123
+ "<|my|>": 50346,
124
+ "<|ne|>": 50313,
125
+ "<|nl|>": 50271,
126
+ "<|nn|>": 50342,
127
+ "<|no|>": 50288,
128
+ "<|oc|>": 50328,
129
+ "<|pa|>": 50321,
130
+ "<|pl|>": 50269,
131
+ "<|ps|>": 50340,
132
+ "<|pt|>": 50267,
133
+ "<|ro|>": 50284,
134
+ "<|ru|>": 50263,
135
+ "<|sa|>": 50344,
136
+ "<|sd|>": 50332,
137
+ "<|si|>": 50322,
138
+ "<|sk|>": 50298,
139
+ "<|sl|>": 50305,
140
+ "<|sn|>": 50324,
141
+ "<|so|>": 50326,
142
+ "<|sq|>": 50317,
143
+ "<|sr|>": 50303,
144
+ "<|su|>": 50357,
145
+ "<|sv|>": 50273,
146
+ "<|sw|>": 50318,
147
+ "<|ta|>": 50287,
148
+ "<|te|>": 50299,
149
+ "<|tg|>": 50331,
150
+ "<|th|>": 50289,
151
+ "<|tk|>": 50341,
152
+ "<|tl|>": 50348,
153
+ "<|tr|>": 50268,
154
+ "<|tt|>": 50351,
155
+ "<|uk|>": 50280,
156
+ "<|ur|>": 50290,
157
+ "<|uz|>": 50337,
158
+ "<|vi|>": 50278,
159
+ "<|yi|>": 50335,
160
+ "<|yo|>": 50325,
161
+ "<|zh|>": 50260
162
+ },
163
+ "language": "breton",
164
+ "max_initial_timestamp_index": 50,
165
+ "max_length": 448,
166
+ "no_timestamps_token_id": 50363,
167
+ "pad_token_id": 50257,
168
+ "prev_sot_token_id": 50361,
169
+ "return_timestamps": false,
170
+ "suppress_tokens": [
171
+ 1,
172
+ 2,
173
+ 7,
174
+ 8,
175
+ 9,
176
+ 10,
177
+ 14,
178
+ 25,
179
+ 26,
180
+ 27,
181
+ 28,
182
+ 29,
183
+ 31,
184
+ 58,
185
+ 59,
186
+ 60,
187
+ 61,
188
+ 62,
189
+ 63,
190
+ 90,
191
+ 91,
192
+ 92,
193
+ 93,
194
+ 359,
195
+ 503,
196
+ 522,
197
+ 542,
198
+ 873,
199
+ 893,
200
+ 902,
201
+ 918,
202
+ 922,
203
+ 931,
204
+ 1350,
205
+ 1853,
206
+ 1982,
207
+ 2460,
208
+ 2627,
209
+ 3246,
210
+ 3253,
211
+ 3268,
212
+ 3536,
213
+ 3846,
214
+ 3961,
215
+ 4183,
216
+ 4667,
217
+ 6585,
218
+ 6647,
219
+ 7273,
220
+ 9061,
221
+ 9383,
222
+ 10428,
223
+ 10929,
224
+ 11938,
225
+ 12033,
226
+ 12331,
227
+ 12562,
228
+ 13793,
229
+ 14157,
230
+ 14635,
231
+ 15265,
232
+ 15618,
233
+ 16553,
234
+ 16604,
235
+ 18362,
236
+ 18956,
237
+ 20075,
238
+ 21675,
239
+ 22520,
240
+ 26130,
241
+ 26161,
242
+ 26435,
243
+ 28279,
244
+ 29464,
245
+ 31650,
246
+ 32302,
247
+ 32470,
248
+ 36865,
249
+ 42863,
250
+ 47425,
251
+ 49870,
252
+ 50254,
253
+ 50258,
254
+ 50358,
255
+ 50359,
256
+ 50360,
257
+ 50361,
258
+ 50362
259
+ ],
260
+ "task": "transcribe",
261
+ "task_to_id": {
262
+ "transcribe": 50359,
263
+ "translate": 50358
264
+ },
265
+ "transformers_version": "4.39.1"
266
+ }
bck-checkpoint-3500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df77e3bda800c7d644d885a1d1ed62588d30ce79964f489153557d0f3bc22bc4
3
+ size 966995080
bck-checkpoint-3500/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b442570c163e08d67c2ac084ad1a650a8c4dcbc23c52037ba2a924c26015f53
3
+ size 1925063607
bck-checkpoint-3500/preprocessor_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "chunk_length": 30,
3
+ "feature_extractor_type": "WhisperFeatureExtractor",
4
+ "feature_size": 80,
5
+ "hop_length": 160,
6
+ "n_fft": 400,
7
+ "n_samples": 480000,
8
+ "nb_max_frames": 3000,
9
+ "padding_side": "right",
10
+ "padding_value": 0.0,
11
+ "processor_class": "WhisperProcessor",
12
+ "return_attention_mask": false,
13
+ "sampling_rate": 16000
14
+ }
bck-checkpoint-3500/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c3a77fa271e382167ede871e69b26f480319be2b7af14900871ec0a2e39e43a
3
+ size 14575
bck-checkpoint-3500/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:52db84bc37411d48adce43568dbf28a2c86b3c5a86cdc3dd9cd4593f0b5c7689
3
+ size 627
bck-checkpoint-3500/trainer_state.json ADDED
@@ -0,0 +1,336 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 43.19571383548692,
3
+ "best_model_checkpoint": "whisper-small-br/checkpoint-3500",
4
+ "epoch": 4.069767441860465,
5
+ "eval_steps": 500,
6
+ "global_step": 3500,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.12,
13
+ "grad_norm": 23.873233795166016,
14
+ "learning_rate": 4.800000000000001e-06,
15
+ "loss": 4.4977,
16
+ "step": 100
17
+ },
18
+ {
19
+ "epoch": 0.23,
20
+ "grad_norm": 18.380311965942383,
21
+ "learning_rate": 9.75e-06,
22
+ "loss": 2.0428,
23
+ "step": 200
24
+ },
25
+ {
26
+ "epoch": 0.35,
27
+ "grad_norm": 15.277495384216309,
28
+ "learning_rate": 9.75e-06,
29
+ "loss": 1.282,
30
+ "step": 300
31
+ },
32
+ {
33
+ "epoch": 0.47,
34
+ "grad_norm": 16.010156631469727,
35
+ "learning_rate": 9.486842105263158e-06,
36
+ "loss": 0.9714,
37
+ "step": 400
38
+ },
39
+ {
40
+ "epoch": 0.58,
41
+ "grad_norm": 12.95919418334961,
42
+ "learning_rate": 9.223684210526316e-06,
43
+ "loss": 0.8721,
44
+ "step": 500
45
+ },
46
+ {
47
+ "epoch": 0.58,
48
+ "eval_cer": 23.931937364773187,
49
+ "eval_loss": 0.847813606262207,
50
+ "eval_runtime": 707.7025,
51
+ "eval_samples_per_second": 3.613,
52
+ "eval_steps_per_second": 0.452,
53
+ "eval_wer": 62.74188465174914,
54
+ "step": 500
55
+ },
56
+ {
57
+ "epoch": 0.7,
58
+ "grad_norm": 15.317465782165527,
59
+ "learning_rate": 8.960526315789474e-06,
60
+ "loss": 0.8022,
61
+ "step": 600
62
+ },
63
+ {
64
+ "epoch": 0.81,
65
+ "grad_norm": 14.4519624710083,
66
+ "learning_rate": 8.697368421052633e-06,
67
+ "loss": 0.7294,
68
+ "step": 700
69
+ },
70
+ {
71
+ "epoch": 0.93,
72
+ "grad_norm": 13.281527519226074,
73
+ "learning_rate": 8.43421052631579e-06,
74
+ "loss": 0.7037,
75
+ "step": 800
76
+ },
77
+ {
78
+ "epoch": 1.05,
79
+ "grad_norm": 11.638937950134277,
80
+ "learning_rate": 8.171052631578949e-06,
81
+ "loss": 0.5775,
82
+ "step": 900
83
+ },
84
+ {
85
+ "epoch": 1.16,
86
+ "grad_norm": 10.528143882751465,
87
+ "learning_rate": 7.907894736842105e-06,
88
+ "loss": 0.413,
89
+ "step": 1000
90
+ },
91
+ {
92
+ "epoch": 1.16,
93
+ "eval_cer": 20.239099343576957,
94
+ "eval_loss": 0.6627262830734253,
95
+ "eval_runtime": 694.5097,
96
+ "eval_samples_per_second": 3.682,
97
+ "eval_steps_per_second": 0.461,
98
+ "eval_wer": 51.78695241096754,
99
+ "step": 1000
100
+ },
101
+ {
102
+ "epoch": 1.28,
103
+ "grad_norm": 11.562024116516113,
104
+ "learning_rate": 7.644736842105264e-06,
105
+ "loss": 0.4373,
106
+ "step": 1100
107
+ },
108
+ {
109
+ "epoch": 1.4,
110
+ "grad_norm": 11.719500541687012,
111
+ "learning_rate": 7.381578947368422e-06,
112
+ "loss": 0.4028,
113
+ "step": 1200
114
+ },
115
+ {
116
+ "epoch": 1.51,
117
+ "grad_norm": 11.709799766540527,
118
+ "learning_rate": 7.11842105263158e-06,
119
+ "loss": 0.4085,
120
+ "step": 1300
121
+ },
122
+ {
123
+ "epoch": 1.63,
124
+ "grad_norm": 10.174458503723145,
125
+ "learning_rate": 6.855263157894737e-06,
126
+ "loss": 0.394,
127
+ "step": 1400
128
+ },
129
+ {
130
+ "epoch": 1.74,
131
+ "grad_norm": 10.892149925231934,
132
+ "learning_rate": 6.5921052631578955e-06,
133
+ "loss": 0.3822,
134
+ "step": 1500
135
+ },
136
+ {
137
+ "epoch": 1.74,
138
+ "eval_cer": 17.72097742246996,
139
+ "eval_loss": 0.6052409410476685,
140
+ "eval_runtime": 688.7859,
141
+ "eval_samples_per_second": 3.712,
142
+ "eval_steps_per_second": 0.465,
143
+ "eval_wer": 46.448156318941066,
144
+ "step": 1500
145
+ },
146
+ {
147
+ "epoch": 1.86,
148
+ "grad_norm": 12.372304916381836,
149
+ "learning_rate": 6.328947368421054e-06,
150
+ "loss": 0.3767,
151
+ "step": 1600
152
+ },
153
+ {
154
+ "epoch": 1.98,
155
+ "grad_norm": 13.72421646118164,
156
+ "learning_rate": 6.065789473684212e-06,
157
+ "loss": 0.3938,
158
+ "step": 1700
159
+ },
160
+ {
161
+ "epoch": 2.09,
162
+ "grad_norm": 10.771753311157227,
163
+ "learning_rate": 5.802631578947368e-06,
164
+ "loss": 0.2385,
165
+ "step": 1800
166
+ },
167
+ {
168
+ "epoch": 2.21,
169
+ "grad_norm": 5.737514495849609,
170
+ "learning_rate": 5.5394736842105266e-06,
171
+ "loss": 0.1885,
172
+ "step": 1900
173
+ },
174
+ {
175
+ "epoch": 2.33,
176
+ "grad_norm": 6.4449920654296875,
177
+ "learning_rate": 5.276315789473685e-06,
178
+ "loss": 0.1868,
179
+ "step": 2000
180
+ },
181
+ {
182
+ "epoch": 2.33,
183
+ "eval_cer": 17.650078844108695,
184
+ "eval_loss": 0.5885388255119324,
185
+ "eval_runtime": 693.9092,
186
+ "eval_samples_per_second": 3.685,
187
+ "eval_steps_per_second": 0.461,
188
+ "eval_wer": 45.30728017648913,
189
+ "step": 2000
190
+ },
191
+ {
192
+ "epoch": 2.44,
193
+ "grad_norm": 7.215250015258789,
194
+ "learning_rate": 5.013157894736842e-06,
195
+ "loss": 0.1873,
196
+ "step": 2100
197
+ },
198
+ {
199
+ "epoch": 2.56,
200
+ "grad_norm": 6.823380947113037,
201
+ "learning_rate": 4.75e-06,
202
+ "loss": 0.1898,
203
+ "step": 2200
204
+ },
205
+ {
206
+ "epoch": 2.67,
207
+ "grad_norm": 7.433701992034912,
208
+ "learning_rate": 4.4868421052631584e-06,
209
+ "loss": 0.1801,
210
+ "step": 2300
211
+ },
212
+ {
213
+ "epoch": 2.79,
214
+ "grad_norm": 8.947402000427246,
215
+ "learning_rate": 4.223684210526316e-06,
216
+ "loss": 0.1911,
217
+ "step": 2400
218
+ },
219
+ {
220
+ "epoch": 2.91,
221
+ "grad_norm": 4.656549453735352,
222
+ "learning_rate": 3.960526315789474e-06,
223
+ "loss": 0.1956,
224
+ "step": 2500
225
+ },
226
+ {
227
+ "epoch": 2.91,
228
+ "eval_cer": 16.69661520407789,
229
+ "eval_loss": 0.5721948146820068,
230
+ "eval_runtime": 688.2161,
231
+ "eval_samples_per_second": 3.715,
232
+ "eval_steps_per_second": 0.465,
233
+ "eval_wer": 43.98361172392058,
234
+ "step": 2500
235
+ },
236
+ {
237
+ "epoch": 3.02,
238
+ "grad_norm": 5.275825023651123,
239
+ "learning_rate": 3.6973684210526317e-06,
240
+ "loss": 0.1568,
241
+ "step": 2600
242
+ },
243
+ {
244
+ "epoch": 3.14,
245
+ "grad_norm": 3.9346418380737305,
246
+ "learning_rate": 3.43421052631579e-06,
247
+ "loss": 0.0884,
248
+ "step": 2700
249
+ },
250
+ {
251
+ "epoch": 3.26,
252
+ "grad_norm": 8.16520881652832,
253
+ "learning_rate": 3.1710526315789477e-06,
254
+ "loss": 0.0893,
255
+ "step": 2800
256
+ },
257
+ {
258
+ "epoch": 3.37,
259
+ "grad_norm": 9.721732139587402,
260
+ "learning_rate": 2.907894736842106e-06,
261
+ "loss": 0.0893,
262
+ "step": 2900
263
+ },
264
+ {
265
+ "epoch": 3.49,
266
+ "grad_norm": 4.916592597961426,
267
+ "learning_rate": 2.644736842105263e-06,
268
+ "loss": 0.0814,
269
+ "step": 3000
270
+ },
271
+ {
272
+ "epoch": 3.49,
273
+ "eval_cer": 17.04132898162749,
274
+ "eval_loss": 0.5933119058609009,
275
+ "eval_runtime": 688.3586,
276
+ "eval_samples_per_second": 3.715,
277
+ "eval_steps_per_second": 0.465,
278
+ "eval_wer": 44.28616451307911,
279
+ "step": 3000
280
+ },
281
+ {
282
+ "epoch": 3.6,
283
+ "grad_norm": 5.402080535888672,
284
+ "learning_rate": 2.381578947368421e-06,
285
+ "loss": 0.0847,
286
+ "step": 3100
287
+ },
288
+ {
289
+ "epoch": 3.72,
290
+ "grad_norm": 4.390013694763184,
291
+ "learning_rate": 2.118421052631579e-06,
292
+ "loss": 0.0902,
293
+ "step": 3200
294
+ },
295
+ {
296
+ "epoch": 3.84,
297
+ "grad_norm": 3.571608543395996,
298
+ "learning_rate": 1.855263157894737e-06,
299
+ "loss": 0.0757,
300
+ "step": 3300
301
+ },
302
+ {
303
+ "epoch": 3.95,
304
+ "grad_norm": 6.823545455932617,
305
+ "learning_rate": 1.5921052631578949e-06,
306
+ "loss": 0.0811,
307
+ "step": 3400
308
+ },
309
+ {
310
+ "epoch": 4.07,
311
+ "grad_norm": 2.958061933517456,
312
+ "learning_rate": 1.3289473684210526e-06,
313
+ "loss": 0.0554,
314
+ "step": 3500
315
+ },
316
+ {
317
+ "epoch": 4.07,
318
+ "eval_cer": 16.441135844120918,
319
+ "eval_loss": 0.6083739399909973,
320
+ "eval_runtime": 690.4893,
321
+ "eval_samples_per_second": 3.703,
322
+ "eval_steps_per_second": 0.463,
323
+ "eval_wer": 43.19571383548692,
324
+ "step": 3500
325
+ }
326
+ ],
327
+ "logging_steps": 100,
328
+ "max_steps": 4000,
329
+ "num_input_tokens_seen": 0,
330
+ "num_train_epochs": 5,
331
+ "save_steps": 500,
332
+ "total_flos": 8.0746195009536e+18,
333
+ "train_batch_size": 8,
334
+ "trial_name": null,
335
+ "trial_params": null
336
+ }
bck-checkpoint-3500/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c6b96e02caa3728fdbe69b1b53c129d42b7cf7db2b7fa3d6c5dece5dcba33f8
3
+ size 4667
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "openai/whisper-base",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "apply_spec_augment": false,
@@ -13,17 +13,17 @@
13
  ],
14
  "bos_token_id": 50257,
15
  "classifier_proj_size": 256,
16
- "d_model": 512,
17
- "decoder_attention_heads": 8,
18
- "decoder_ffn_dim": 2048,
19
  "decoder_layerdrop": 0.0,
20
- "decoder_layers": 6,
21
  "decoder_start_token_id": 50258,
22
  "dropout": 0.0,
23
- "encoder_attention_heads": 8,
24
- "encoder_ffn_dim": 2048,
25
  "encoder_layerdrop": 0.0,
26
- "encoder_layers": 6,
27
  "eos_token_id": 50257,
28
  "forced_decoder_ids": null,
29
  "init_std": 0.02,
@@ -39,14 +39,14 @@
39
  "max_target_positions": 448,
40
  "median_filter_width": 7,
41
  "model_type": "whisper",
42
- "num_hidden_layers": 6,
43
  "num_mel_bins": 80,
44
  "pad_token_id": 50257,
45
  "scale_embedding": false,
46
  "suppress_tokens": [],
47
  "torch_dtype": "float32",
48
- "transformers_version": "4.35.2",
49
- "use_cache": true,
50
  "use_weighted_layer_sum": false,
51
  "vocab_size": 51865
52
  }
 
1
  {
2
+ "_name_or_path": "openai/whisper-small",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "apply_spec_augment": false,
 
13
  ],
14
  "bos_token_id": 50257,
15
  "classifier_proj_size": 256,
16
+ "d_model": 768,
17
+ "decoder_attention_heads": 12,
18
+ "decoder_ffn_dim": 3072,
19
  "decoder_layerdrop": 0.0,
20
+ "decoder_layers": 12,
21
  "decoder_start_token_id": 50258,
22
  "dropout": 0.0,
23
+ "encoder_attention_heads": 12,
24
+ "encoder_ffn_dim": 3072,
25
  "encoder_layerdrop": 0.0,
26
+ "encoder_layers": 12,
27
  "eos_token_id": 50257,
28
  "forced_decoder_ids": null,
29
  "init_std": 0.02,
 
39
  "max_target_positions": 448,
40
  "median_filter_width": 7,
41
  "model_type": "whisper",
42
+ "num_hidden_layers": 12,
43
  "num_mel_bins": 80,
44
  "pad_token_id": 50257,
45
  "scale_embedding": false,
46
  "suppress_tokens": [],
47
  "torch_dtype": "float32",
48
+ "transformers_version": "4.39.1",
49
+ "use_cache": false,
50
  "use_weighted_layer_sum": false,
51
  "vocab_size": 51865
52
  }
generation_config.json ADDED
@@ -0,0 +1,266 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alignment_heads": [
3
+ [
4
+ 5,
5
+ 3
6
+ ],
7
+ [
8
+ 5,
9
+ 9
10
+ ],
11
+ [
12
+ 8,
13
+ 0
14
+ ],
15
+ [
16
+ 8,
17
+ 4
18
+ ],
19
+ [
20
+ 8,
21
+ 7
22
+ ],
23
+ [
24
+ 8,
25
+ 8
26
+ ],
27
+ [
28
+ 9,
29
+ 0
30
+ ],
31
+ [
32
+ 9,
33
+ 7
34
+ ],
35
+ [
36
+ 9,
37
+ 9
38
+ ],
39
+ [
40
+ 10,
41
+ 5
42
+ ]
43
+ ],
44
+ "begin_suppress_tokens": [
45
+ 220,
46
+ 50257
47
+ ],
48
+ "bos_token_id": 50257,
49
+ "decoder_start_token_id": 50258,
50
+ "eos_token_id": 50257,
51
+ "forced_decoder_ids": [
52
+ [
53
+ 1,
54
+ null
55
+ ],
56
+ [
57
+ 2,
58
+ 50359
59
+ ]
60
+ ],
61
+ "is_multilingual": true,
62
+ "lang_to_id": {
63
+ "<|af|>": 50327,
64
+ "<|am|>": 50334,
65
+ "<|ar|>": 50272,
66
+ "<|as|>": 50350,
67
+ "<|az|>": 50304,
68
+ "<|ba|>": 50355,
69
+ "<|be|>": 50330,
70
+ "<|bg|>": 50292,
71
+ "<|bn|>": 50302,
72
+ "<|bo|>": 50347,
73
+ "<|br|>": 50309,
74
+ "<|bs|>": 50315,
75
+ "<|ca|>": 50270,
76
+ "<|cs|>": 50283,
77
+ "<|cy|>": 50297,
78
+ "<|da|>": 50285,
79
+ "<|de|>": 50261,
80
+ "<|el|>": 50281,
81
+ "<|en|>": 50259,
82
+ "<|es|>": 50262,
83
+ "<|et|>": 50307,
84
+ "<|eu|>": 50310,
85
+ "<|fa|>": 50300,
86
+ "<|fi|>": 50277,
87
+ "<|fo|>": 50338,
88
+ "<|fr|>": 50265,
89
+ "<|gl|>": 50319,
90
+ "<|gu|>": 50333,
91
+ "<|haw|>": 50352,
92
+ "<|ha|>": 50354,
93
+ "<|he|>": 50279,
94
+ "<|hi|>": 50276,
95
+ "<|hr|>": 50291,
96
+ "<|ht|>": 50339,
97
+ "<|hu|>": 50286,
98
+ "<|hy|>": 50312,
99
+ "<|id|>": 50275,
100
+ "<|is|>": 50311,
101
+ "<|it|>": 50274,
102
+ "<|ja|>": 50266,
103
+ "<|jw|>": 50356,
104
+ "<|ka|>": 50329,
105
+ "<|kk|>": 50316,
106
+ "<|km|>": 50323,
107
+ "<|kn|>": 50306,
108
+ "<|ko|>": 50264,
109
+ "<|la|>": 50294,
110
+ "<|lb|>": 50345,
111
+ "<|ln|>": 50353,
112
+ "<|lo|>": 50336,
113
+ "<|lt|>": 50293,
114
+ "<|lv|>": 50301,
115
+ "<|mg|>": 50349,
116
+ "<|mi|>": 50295,
117
+ "<|mk|>": 50308,
118
+ "<|ml|>": 50296,
119
+ "<|mn|>": 50314,
120
+ "<|mr|>": 50320,
121
+ "<|ms|>": 50282,
122
+ "<|mt|>": 50343,
123
+ "<|my|>": 50346,
124
+ "<|ne|>": 50313,
125
+ "<|nl|>": 50271,
126
+ "<|nn|>": 50342,
127
+ "<|no|>": 50288,
128
+ "<|oc|>": 50328,
129
+ "<|pa|>": 50321,
130
+ "<|pl|>": 50269,
131
+ "<|ps|>": 50340,
132
+ "<|pt|>": 50267,
133
+ "<|ro|>": 50284,
134
+ "<|ru|>": 50263,
135
+ "<|sa|>": 50344,
136
+ "<|sd|>": 50332,
137
+ "<|si|>": 50322,
138
+ "<|sk|>": 50298,
139
+ "<|sl|>": 50305,
140
+ "<|sn|>": 50324,
141
+ "<|so|>": 50326,
142
+ "<|sq|>": 50317,
143
+ "<|sr|>": 50303,
144
+ "<|su|>": 50357,
145
+ "<|sv|>": 50273,
146
+ "<|sw|>": 50318,
147
+ "<|ta|>": 50287,
148
+ "<|te|>": 50299,
149
+ "<|tg|>": 50331,
150
+ "<|th|>": 50289,
151
+ "<|tk|>": 50341,
152
+ "<|tl|>": 50348,
153
+ "<|tr|>": 50268,
154
+ "<|tt|>": 50351,
155
+ "<|uk|>": 50280,
156
+ "<|ur|>": 50290,
157
+ "<|uz|>": 50337,
158
+ "<|vi|>": 50278,
159
+ "<|yi|>": 50335,
160
+ "<|yo|>": 50325,
161
+ "<|zh|>": 50260
162
+ },
163
+ "language": "breton",
164
+ "max_initial_timestamp_index": 50,
165
+ "max_length": 448,
166
+ "no_timestamps_token_id": 50363,
167
+ "pad_token_id": 50257,
168
+ "prev_sot_token_id": 50361,
169
+ "return_timestamps": false,
170
+ "suppress_tokens": [
171
+ 1,
172
+ 2,
173
+ 7,
174
+ 8,
175
+ 9,
176
+ 10,
177
+ 14,
178
+ 25,
179
+ 26,
180
+ 27,
181
+ 28,
182
+ 29,
183
+ 31,
184
+ 58,
185
+ 59,
186
+ 60,
187
+ 61,
188
+ 62,
189
+ 63,
190
+ 90,
191
+ 91,
192
+ 92,
193
+ 93,
194
+ 359,
195
+ 503,
196
+ 522,
197
+ 542,
198
+ 873,
199
+ 893,
200
+ 902,
201
+ 918,
202
+ 922,
203
+ 931,
204
+ 1350,
205
+ 1853,
206
+ 1982,
207
+ 2460,
208
+ 2627,
209
+ 3246,
210
+ 3253,
211
+ 3268,
212
+ 3536,
213
+ 3846,
214
+ 3961,
215
+ 4183,
216
+ 4667,
217
+ 6585,
218
+ 6647,
219
+ 7273,
220
+ 9061,
221
+ 9383,
222
+ 10428,
223
+ 10929,
224
+ 11938,
225
+ 12033,
226
+ 12331,
227
+ 12562,
228
+ 13793,
229
+ 14157,
230
+ 14635,
231
+ 15265,
232
+ 15618,
233
+ 16553,
234
+ 16604,
235
+ 18362,
236
+ 18956,
237
+ 20075,
238
+ 21675,
239
+ 22520,
240
+ 26130,
241
+ 26161,
242
+ 26435,
243
+ 28279,
244
+ 29464,
245
+ 31650,
246
+ 32302,
247
+ 32470,
248
+ 36865,
249
+ 42863,
250
+ 47425,
251
+ 49870,
252
+ 50254,
253
+ 50258,
254
+ 50358,
255
+ 50359,
256
+ 50360,
257
+ 50361,
258
+ 50362
259
+ ],
260
+ "task": "transcribe",
261
+ "task_to_id": {
262
+ "transcribe": 50359,
263
+ "translate": 50358
264
+ },
265
+ "transformers_version": "4.39.1"
266
+ }
merges.txt CHANGED
@@ -1,4 +1,5 @@
1
  #version: 0.2
 
2
  Ġ a
3
  Ġt h
4
  i n
 
1
  #version: 0.2
2
+ Ġ t
3
  Ġ a
4
  Ġt h
5
  i n
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1f241bc8c2b1789113b10be95fda8d4090543330e87fed0bb66bac16a607d8cb
3
- size 290403936
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aeeef5fbd5d40b26b8e943b19f39f76e4bb8a455d0e7f642ac1ba51dc5e4bd69
3
+ size 966995080
runs/Aug20_21-41-26_gweltaz-NUC10i7FNK/events.out.tfevents.1724182888.gweltaz-NUC10i7FNK.3502.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:620ac5e92d2fc446c799558e8026bc5e456bebd2365e90ff6c658f7fc4e7089e
3
+ size 17166
runs/Aug21_09-51-18_gweltaz-NUC10i7FNK/events.out.tfevents.1724226679.gweltaz-NUC10i7FNK.7651.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71625f72c3cfa7a64e459283b1dbee14c60402a91bd9fbf689c99cc9b9f81663
3
+ size 17166
special_tokens_map.json CHANGED
@@ -111,14 +111,14 @@
111
  "bos_token": {
112
  "content": "<|endoftext|>",
113
  "lstrip": false,
114
- "normalized": true,
115
  "rstrip": false,
116
  "single_word": false
117
  },
118
  "eos_token": {
119
  "content": "<|endoftext|>",
120
  "lstrip": false,
121
- "normalized": true,
122
  "rstrip": false,
123
  "single_word": false
124
  },
@@ -132,7 +132,7 @@
132
  "unk_token": {
133
  "content": "<|endoftext|>",
134
  "lstrip": false,
135
- "normalized": true,
136
  "rstrip": false,
137
  "single_word": false
138
  }
 
111
  "bos_token": {
112
  "content": "<|endoftext|>",
113
  "lstrip": false,
114
+ "normalized": false,
115
  "rstrip": false,
116
  "single_word": false
117
  },
118
  "eos_token": {
119
  "content": "<|endoftext|>",
120
  "lstrip": false,
121
+ "normalized": false,
122
  "rstrip": false,
123
  "single_word": false
124
  },
 
132
  "unk_token": {
133
  "content": "<|endoftext|>",
134
  "lstrip": false,
135
+ "normalized": false,
136
  "rstrip": false,
137
  "single_word": false
138
  }
tokenizer_config.json CHANGED
@@ -5,7 +5,7 @@
5
  "50257": {
6
  "content": "<|endoftext|>",
7
  "lstrip": false,
8
- "normalized": true,
9
  "rstrip": false,
10
  "single_word": false,
11
  "special": true
@@ -12980,7 +12980,7 @@
12980
  "clean_up_tokenization_spaces": true,
12981
  "eos_token": "<|endoftext|>",
12982
  "errors": "replace",
12983
- "model_max_length": 448,
12984
  "pad_token": "<|endoftext|>",
12985
  "processor_class": "WhisperProcessor",
12986
  "return_attention_mask": false,
 
5
  "50257": {
6
  "content": "<|endoftext|>",
7
  "lstrip": false,
8
+ "normalized": false,
9
  "rstrip": false,
10
  "single_word": false,
11
  "special": true
 
12980
  "clean_up_tokenization_spaces": true,
12981
  "eos_token": "<|endoftext|>",
12982
  "errors": "replace",
12983
+ "model_max_length": 1024,
12984
  "pad_token": "<|endoftext|>",
12985
  "processor_class": "WhisperProcessor",
12986
  "return_attention_mask": false,
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fc2e73cdda8257333a1ff484734cbec89e0eac555048409375d15fcadc3075bf
3
- size 4728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61e6ce06127f678cc499816e943dbf33660aa9806bdbf3007ca58d9efda9f324
3
+ size 4667