gchhablani commited on
Commit
3aa590f
1 Parent(s): d1d0762

Add Model Files

Browse files
all_results.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 30.0,
3
+ "eval_loss": 0.3495674431324005,
4
+ "eval_mem_cpu_alloc_delta": 443444278,
5
+ "eval_mem_cpu_peaked_delta": 34247898,
6
+ "eval_mem_gpu_alloc_delta": 0,
7
+ "eval_mem_gpu_peaked_delta": 2035233792,
8
+ "eval_runtime": 473.1711,
9
+ "eval_samples": 4641,
10
+ "eval_samples_per_second": 9.808,
11
+ "eval_wer": 0.28846481029887144,
12
+ "init_mem_cpu_alloc_delta": 10140165,
13
+ "init_mem_cpu_peaked_delta": 18306,
14
+ "init_mem_gpu_alloc_delta": 1261939712,
15
+ "init_mem_gpu_peaked_delta": 0,
16
+ "train_mem_cpu_alloc_delta": 1543641,
17
+ "train_mem_cpu_peaked_delta": 478977252,
18
+ "train_mem_gpu_alloc_delta": 3779933184,
19
+ "train_mem_gpu_peaked_delta": 7771228672,
20
+ "train_runtime": 43862.7197,
21
+ "train_samples": 11106,
22
+ "train_samples_per_second": 0.238
23
+ }
config.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/wav2vec2-large-xlsr-53",
3
+ "activation_dropout": 0.1,
4
+ "apply_spec_augment": true,
5
+ "architectures": [
6
+ "Wav2Vec2ForCTC"
7
+ ],
8
+ "attention_dropout": 0.1,
9
+ "bos_token_id": 1,
10
+ "conv_bias": true,
11
+ "conv_dim": [
12
+ 512,
13
+ 512,
14
+ 512,
15
+ 512,
16
+ 512,
17
+ 512,
18
+ 512
19
+ ],
20
+ "conv_kernel": [
21
+ 10,
22
+ 3,
23
+ 3,
24
+ 3,
25
+ 3,
26
+ 2,
27
+ 2
28
+ ],
29
+ "conv_stride": [
30
+ 5,
31
+ 2,
32
+ 2,
33
+ 2,
34
+ 2,
35
+ 2,
36
+ 2
37
+ ],
38
+ "ctc_loss_reduction": "mean",
39
+ "ctc_zero_infinity": false,
40
+ "do_stable_layer_norm": true,
41
+ "eos_token_id": 2,
42
+ "feat_extract_activation": "gelu",
43
+ "feat_extract_dropout": 0.0,
44
+ "feat_extract_norm": "layer",
45
+ "feat_proj_dropout": 0.0,
46
+ "final_dropout": 0.0,
47
+ "gradient_checkpointing": true,
48
+ "hidden_act": "gelu",
49
+ "hidden_dropout": 0.1,
50
+ "hidden_size": 1024,
51
+ "initializer_range": 0.02,
52
+ "intermediate_size": 4096,
53
+ "layer_norm_eps": 1e-05,
54
+ "layerdrop": 0.1,
55
+ "mask_channel_length": 10,
56
+ "mask_channel_min_space": 1,
57
+ "mask_channel_other": 0.0,
58
+ "mask_channel_prob": 0.0,
59
+ "mask_channel_selection": "static",
60
+ "mask_feature_length": 10,
61
+ "mask_feature_prob": 0.0,
62
+ "mask_time_length": 10,
63
+ "mask_time_min_space": 1,
64
+ "mask_time_other": 0.0,
65
+ "mask_time_prob": 0.05,
66
+ "mask_time_selection": "static",
67
+ "model_type": "wav2vec2",
68
+ "num_attention_heads": 16,
69
+ "num_conv_pos_embedding_groups": 16,
70
+ "num_conv_pos_embeddings": 128,
71
+ "num_feat_extract_layers": 7,
72
+ "num_hidden_layers": 24,
73
+ "pad_token_id": 44,
74
+ "transformers_version": "4.5.0.dev0",
75
+ "vocab_size": 45
76
+ }
eval_results.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 30.0,
3
+ "eval_loss": 0.3495674431324005,
4
+ "eval_mem_cpu_alloc_delta": 443444278,
5
+ "eval_mem_cpu_peaked_delta": 34247898,
6
+ "eval_mem_gpu_alloc_delta": 0,
7
+ "eval_mem_gpu_peaked_delta": 2035233792,
8
+ "eval_runtime": 473.1711,
9
+ "eval_samples": 4641,
10
+ "eval_samples_per_second": 9.808,
11
+ "eval_wer": 0.28846481029887144
12
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "feature_size": 1,
4
+ "padding_side": "right",
5
+ "padding_value": 0.0,
6
+ "return_attention_mask": true,
7
+ "sampling_rate": 16000
8
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bfbf97987832b6f75ea8a8a917ace6e33a29e65413e85df5ea0bc683793da1b3
3
+ size 1262118359
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
1
+ {"bos_token": "<s>", "eos_token": "</s>", "unk_token": "[UNK]", "pad_token": "[PAD]"}
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
1
+ {"unk_token": "[UNK]", "bos_token": "<s>", "eos_token": "</s>", "pad_token": "[PAD]", "do_lower_case": false, "word_delimiter_token": "|"}
train_results.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 30.0,
3
+ "init_mem_cpu_alloc_delta": 10140165,
4
+ "init_mem_cpu_peaked_delta": 18306,
5
+ "init_mem_gpu_alloc_delta": 1261939712,
6
+ "init_mem_gpu_peaked_delta": 0,
7
+ "train_mem_cpu_alloc_delta": 1543641,
8
+ "train_mem_cpu_peaked_delta": 478977252,
9
+ "train_mem_gpu_alloc_delta": 3779933184,
10
+ "train_mem_gpu_peaked_delta": 7771228672,
11
+ "train_runtime": 43862.7197,
12
+ "train_samples": 11106,
13
+ "train_samples_per_second": 0.238
14
+ }
trainer_state.json ADDED
@@ -0,0 +1,303 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 30.0,
5
+ "global_step": 10440,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 1.44,
12
+ "learning_rate": 0.0003,
13
+ "loss": NaN,
14
+ "step": 500
15
+ },
16
+ {
17
+ "epoch": 1.44,
18
+ "eval_loss": 2.3965017795562744,
19
+ "eval_runtime": 464.2782,
20
+ "eval_samples_per_second": 9.996,
21
+ "eval_wer": 0.9993478137582941,
22
+ "step": 500
23
+ },
24
+ {
25
+ "epoch": 2.87,
26
+ "learning_rate": 0.00028490945674044263,
27
+ "loss": NaN,
28
+ "step": 1000
29
+ },
30
+ {
31
+ "epoch": 2.87,
32
+ "eval_loss": 0.40979766845703125,
33
+ "eval_runtime": 469.7691,
34
+ "eval_samples_per_second": 9.879,
35
+ "eval_wer": 0.4245448874269835,
36
+ "step": 1000
37
+ },
38
+ {
39
+ "epoch": 4.31,
40
+ "learning_rate": 0.0002698189134808853,
41
+ "loss": NaN,
42
+ "step": 1500
43
+ },
44
+ {
45
+ "epoch": 4.31,
46
+ "eval_loss": 0.32825562357902527,
47
+ "eval_runtime": 475.331,
48
+ "eval_samples_per_second": 9.764,
49
+ "eval_wer": 0.37052685305960414,
50
+ "step": 1500
51
+ },
52
+ {
53
+ "epoch": 5.75,
54
+ "learning_rate": 0.00025472837022132795,
55
+ "loss": NaN,
56
+ "step": 2000
57
+ },
58
+ {
59
+ "epoch": 5.75,
60
+ "eval_loss": 0.30649319291114807,
61
+ "eval_runtime": 474.8166,
62
+ "eval_samples_per_second": 9.774,
63
+ "eval_wer": 0.35002552033119716,
64
+ "step": 2000
65
+ },
66
+ {
67
+ "epoch": 7.18,
68
+ "learning_rate": 0.0002396378269617706,
69
+ "loss": NaN,
70
+ "step": 2500
71
+ },
72
+ {
73
+ "epoch": 7.18,
74
+ "eval_loss": 0.3162720799446106,
75
+ "eval_runtime": 475.8373,
76
+ "eval_samples_per_second": 9.753,
77
+ "eval_wer": 0.3463392502693813,
78
+ "step": 2500
79
+ },
80
+ {
81
+ "epoch": 8.62,
82
+ "learning_rate": 0.00022454728370221327,
83
+ "loss": NaN,
84
+ "step": 3000
85
+ },
86
+ {
87
+ "epoch": 8.62,
88
+ "eval_loss": 0.30734783411026,
89
+ "eval_runtime": 475.1293,
90
+ "eval_samples_per_second": 9.768,
91
+ "eval_wer": 0.3295808994498951,
92
+ "step": 3000
93
+ },
94
+ {
95
+ "epoch": 10.06,
96
+ "learning_rate": 0.00020945674044265593,
97
+ "loss": NaN,
98
+ "step": 3500
99
+ },
100
+ {
101
+ "epoch": 10.06,
102
+ "eval_loss": 0.32298463582992554,
103
+ "eval_runtime": 478.2831,
104
+ "eval_samples_per_second": 9.703,
105
+ "eval_wer": 0.32586627346452673,
106
+ "step": 3500
107
+ },
108
+ {
109
+ "epoch": 11.49,
110
+ "learning_rate": 0.0001943661971830986,
111
+ "loss": NaN,
112
+ "step": 4000
113
+ },
114
+ {
115
+ "epoch": 11.49,
116
+ "eval_loss": 0.3256703317165375,
117
+ "eval_runtime": 476.0024,
118
+ "eval_samples_per_second": 9.75,
119
+ "eval_wer": 0.3196279702829921,
120
+ "step": 4000
121
+ },
122
+ {
123
+ "epoch": 12.93,
124
+ "learning_rate": 0.00017927565392354125,
125
+ "loss": NaN,
126
+ "step": 4500
127
+ },
128
+ {
129
+ "epoch": 12.93,
130
+ "eval_loss": 0.32864469289779663,
131
+ "eval_runtime": 473.6083,
132
+ "eval_samples_per_second": 9.799,
133
+ "eval_wer": 0.32019508875404074,
134
+ "step": 4500
135
+ },
136
+ {
137
+ "epoch": 14.37,
138
+ "learning_rate": 0.0001641851106639839,
139
+ "loss": NaN,
140
+ "step": 5000
141
+ },
142
+ {
143
+ "epoch": 14.37,
144
+ "eval_loss": 0.32181525230407715,
145
+ "eval_runtime": 474.4211,
146
+ "eval_samples_per_second": 9.782,
147
+ "eval_wer": 0.31662224238643455,
148
+ "step": 5000
149
+ },
150
+ {
151
+ "epoch": 15.8,
152
+ "learning_rate": 0.00014909456740442654,
153
+ "loss": NaN,
154
+ "step": 5500
155
+ },
156
+ {
157
+ "epoch": 15.8,
158
+ "eval_loss": 0.33631598949432373,
159
+ "eval_runtime": 477.6284,
160
+ "eval_samples_per_second": 9.717,
161
+ "eval_wer": 0.3121136505415981,
162
+ "step": 5500
163
+ },
164
+ {
165
+ "epoch": 17.24,
166
+ "learning_rate": 0.0001340040241448692,
167
+ "loss": NaN,
168
+ "step": 6000
169
+ },
170
+ {
171
+ "epoch": 17.24,
172
+ "eval_loss": 0.3525121510028839,
173
+ "eval_runtime": 482.4872,
174
+ "eval_samples_per_second": 9.619,
175
+ "eval_wer": 0.30533658481256737,
176
+ "step": 6000
177
+ },
178
+ {
179
+ "epoch": 18.68,
180
+ "learning_rate": 0.00011891348088531185,
181
+ "loss": NaN,
182
+ "step": 6500
183
+ },
184
+ {
185
+ "epoch": 18.68,
186
+ "eval_loss": 0.34829291701316833,
187
+ "eval_runtime": 477.1506,
188
+ "eval_samples_per_second": 9.726,
189
+ "eval_wer": 0.3055634322009868,
190
+ "step": 6500
191
+ },
192
+ {
193
+ "epoch": 20.11,
194
+ "learning_rate": 0.00010382293762575451,
195
+ "loss": NaN,
196
+ "step": 7000
197
+ },
198
+ {
199
+ "epoch": 20.11,
200
+ "eval_loss": 0.35680249333381653,
201
+ "eval_runtime": 474.1059,
202
+ "eval_samples_per_second": 9.789,
203
+ "eval_wer": 0.30780355016162875,
204
+ "step": 7000
205
+ },
206
+ {
207
+ "epoch": 21.55,
208
+ "learning_rate": 8.873239436619717e-05,
209
+ "loss": NaN,
210
+ "step": 7500
211
+ },
212
+ {
213
+ "epoch": 21.55,
214
+ "eval_loss": 0.36808809638023376,
215
+ "eval_runtime": 477.9995,
216
+ "eval_samples_per_second": 9.709,
217
+ "eval_wer": 0.3064424658311121,
218
+ "step": 7500
219
+ },
220
+ {
221
+ "epoch": 22.99,
222
+ "learning_rate": 7.364185110663983e-05,
223
+ "loss": NaN,
224
+ "step": 8000
225
+ },
226
+ {
227
+ "epoch": 22.99,
228
+ "eval_loss": 0.36075320839881897,
229
+ "eval_runtime": 479.4594,
230
+ "eval_samples_per_second": 9.68,
231
+ "eval_wer": 0.29960868825497644,
232
+ "step": 8000
233
+ },
234
+ {
235
+ "epoch": 24.43,
236
+ "learning_rate": 5.855130784708249e-05,
237
+ "loss": NaN,
238
+ "step": 8500
239
+ },
240
+ {
241
+ "epoch": 24.43,
242
+ "eval_loss": 0.36022624373435974,
243
+ "eval_runtime": 482.5145,
244
+ "eval_samples_per_second": 9.618,
245
+ "eval_wer": 0.29745363806499175,
246
+ "step": 8500
247
+ },
248
+ {
249
+ "epoch": 25.86,
250
+ "learning_rate": 4.346076458752515e-05,
251
+ "loss": NaN,
252
+ "step": 9000
253
+ },
254
+ {
255
+ "epoch": 25.86,
256
+ "eval_loss": 0.35646718740463257,
257
+ "eval_runtime": 476.5808,
258
+ "eval_samples_per_second": 9.738,
259
+ "eval_wer": 0.2930301139908127,
260
+ "step": 9000
261
+ },
262
+ {
263
+ "epoch": 27.3,
264
+ "learning_rate": 2.8370221327967802e-05,
265
+ "loss": NaN,
266
+ "step": 9500
267
+ },
268
+ {
269
+ "epoch": 27.3,
270
+ "eval_loss": 0.3554363548755646,
271
+ "eval_runtime": 478.8286,
272
+ "eval_samples_per_second": 9.692,
273
+ "eval_wer": 0.29130040265411444,
274
+ "step": 9500
275
+ },
276
+ {
277
+ "epoch": 28.74,
278
+ "learning_rate": 1.3279678068410461e-05,
279
+ "loss": NaN,
280
+ "step": 10000
281
+ },
282
+ {
283
+ "epoch": 28.74,
284
+ "eval_loss": 0.3562660217285156,
285
+ "eval_runtime": 478.4327,
286
+ "eval_samples_per_second": 9.7,
287
+ "eval_wer": 0.29016616571201725,
288
+ "step": 10000
289
+ },
290
+ {
291
+ "epoch": 30.0,
292
+ "step": 10440,
293
+ "total_flos": 4.532712593948488e+19,
294
+ "train_runtime": 43862.7197,
295
+ "train_samples_per_second": 0.238
296
+ }
297
+ ],
298
+ "max_steps": 10440,
299
+ "num_train_epochs": 30,
300
+ "total_flos": 4.532712593948488e+19,
301
+ "trial_name": null,
302
+ "trial_params": null
303
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4770aa94d94300eefadfa2e8c000ddedbf2959ebf582123b6b38386ff4156145
3
+ size 2351
vocab.json ADDED
@@ -0,0 +1 @@
 
1
+ {"ó": 0, "q": 1, "á": 2, "ê": 3, "ô": 4, "c": 5, "”": 6, "ú": 7, "r": 8, "f": 9, "a": 10, "é": 11, "ü": 12, "l": 13, "ã": 14, "k": 15, "p": 16, "n": 17, "b": 18, "v": 19, "j": 20, "o": 21, "à": 22, "e": 23, "õ": 24, "t": 25, "y": 26, "ç": 27, "“": 28, "â": 29, "í": 30, "h": 31, "z": 32, "d": 33, "s": 34, "u": 35, "i": 36, "g": 37, "w": 38, "x": 39, "ñ": 41, "m": 42, "|": 40, "[UNK]": 43, "[PAD]": 44}