DrishtiSharma commited on
Commit
afba1ec
1 Parent(s): 16902be

Upload lm-boosted decoder

Browse files
added_tokens.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"<s>": 41, "</s>": 42}
all_results.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 149.98,
3
+ "eval_loss": 0.7148692607879639,
4
+ "eval_runtime": 14.0171,
5
+ "eval_samples": 336,
6
+ "eval_samples_per_second": 23.971,
7
+ "eval_steps_per_second": 2.996,
8
+ "eval_wer": 0.451,
9
+ "train_loss": 0.9359948108413003,
10
+ "train_runtime": 8030.991,
11
+ "train_samples": 717,
12
+ "train_samples_per_second": 13.392,
13
+ "train_steps_per_second": 0.411
14
+ }
alphabet.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"labels": [" ", "a", "c", "i", "k", "\u0430", "\u0431", "\u0433", "\u0434", "\u0435", "\u0436", "\u0437", "\u0438", "\u0439", "\u043a", "\u043b", "\u043c", "\u043d", "\u043e", "\u043f", "\u0440", "\u0441", "\u0442", "\u0443", "\u0445", "\u0448", "\u0449", "\u044b", "\u044e", "\u044f", "\u0456", "\u0493", "\u049b", "\u04a3", "\u04af", "\u04b1", "\u04bb", "\u04d9", "\u04e9", "\u2047", "", "<s>", "</s>"], "is_bpe": false}
config.json ADDED
@@ -0,0 +1,107 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/wav2vec2-xls-r-300m",
3
+ "activation_dropout": 0.0,
4
+ "adapter_kernel_size": 3,
5
+ "adapter_stride": 2,
6
+ "add_adapter": false,
7
+ "apply_spec_augment": true,
8
+ "architectures": [
9
+ "Wav2Vec2ForCTC"
10
+ ],
11
+ "attention_dropout": 0.0,
12
+ "bos_token_id": 1,
13
+ "classifier_proj_size": 256,
14
+ "codevector_dim": 768,
15
+ "contrastive_logits_temperature": 0.1,
16
+ "conv_bias": true,
17
+ "conv_dim": [
18
+ 512,
19
+ 512,
20
+ 512,
21
+ 512,
22
+ 512,
23
+ 512,
24
+ 512
25
+ ],
26
+ "conv_kernel": [
27
+ 10,
28
+ 3,
29
+ 3,
30
+ 3,
31
+ 3,
32
+ 2,
33
+ 2
34
+ ],
35
+ "conv_stride": [
36
+ 5,
37
+ 2,
38
+ 2,
39
+ 2,
40
+ 2,
41
+ 2,
42
+ 2
43
+ ],
44
+ "ctc_loss_reduction": "mean",
45
+ "ctc_zero_infinity": false,
46
+ "diversity_loss_weight": 0.1,
47
+ "do_stable_layer_norm": true,
48
+ "eos_token_id": 2,
49
+ "feat_extract_activation": "gelu",
50
+ "feat_extract_dropout": 0.0,
51
+ "feat_extract_norm": "layer",
52
+ "feat_proj_dropout": 0.0,
53
+ "feat_quantizer_dropout": 0.0,
54
+ "final_dropout": 0.0,
55
+ "hidden_act": "gelu",
56
+ "hidden_dropout": 0.0,
57
+ "hidden_size": 1024,
58
+ "initializer_range": 0.02,
59
+ "intermediate_size": 4096,
60
+ "layer_norm_eps": 1e-05,
61
+ "layerdrop": 0.0,
62
+ "mask_feature_length": 10,
63
+ "mask_feature_min_masks": 0,
64
+ "mask_feature_prob": 0.0,
65
+ "mask_time_length": 10,
66
+ "mask_time_min_masks": 2,
67
+ "mask_time_prob": 0.25,
68
+ "model_type": "wav2vec2",
69
+ "num_adapter_layers": 3,
70
+ "num_attention_heads": 16,
71
+ "num_codevector_groups": 2,
72
+ "num_codevectors_per_group": 320,
73
+ "num_conv_pos_embedding_groups": 16,
74
+ "num_conv_pos_embeddings": 128,
75
+ "num_feat_extract_layers": 7,
76
+ "num_hidden_layers": 24,
77
+ "num_negatives": 100,
78
+ "output_hidden_size": 1024,
79
+ "pad_token_id": 40,
80
+ "proj_codevector_dim": 768,
81
+ "tdnn_dilation": [
82
+ 1,
83
+ 2,
84
+ 3,
85
+ 1,
86
+ 1
87
+ ],
88
+ "tdnn_dim": [
89
+ 512,
90
+ 512,
91
+ 512,
92
+ 512,
93
+ 1500
94
+ ],
95
+ "tdnn_kernel": [
96
+ 5,
97
+ 3,
98
+ 3,
99
+ 1,
100
+ 1
101
+ ],
102
+ "torch_dtype": "float32",
103
+ "transformers_version": "4.17.0.dev0",
104
+ "use_weighted_layer_sum": false,
105
+ "vocab_size": 43,
106
+ "xvector_output_dim": 512
107
+ }
eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 149.98,
3
+ "eval_loss": 0.7148692607879639,
4
+ "eval_runtime": 14.0171,
5
+ "eval_samples": 336,
6
+ "eval_samples_per_second": 23.971,
7
+ "eval_steps_per_second": 2.996,
8
+ "eval_wer": 0.451
9
+ }
language_model/5gram.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a1b4e76899c23765f8045a49b0fdd216ac89e687ffde64f3dc4fead2d84b337c
3
+ size 26726842
language_model/attrs.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"alpha": 0.5, "beta": 1.5, "unk_score_offset": -10.0, "score_boundary": true}
language_model/unigrams.txt ADDED
The diff for this file is too large to render. See raw diff
 
preprocessor_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "feature_extractor_type": "Wav2Vec2FeatureExtractor",
4
+ "feature_size": 1,
5
+ "padding_side": "right",
6
+ "padding_value": 0,
7
+ "processor_class": "Wav2Vec2ProcessorWithLM",
8
+ "return_attention_mask": true,
9
+ "sampling_rate": 16000
10
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:242eeb153035874f0ec5f10fee8fd7581625ea7916125d721673df269fa204ad
3
+ size 1262099953
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"bos_token": "<s>", "eos_token": "</s>", "unk_token": "[UNK]", "pad_token": "[PAD]", "additional_special_tokens": [{"content": "<s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, {"content": "</s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, {"content": "<s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, {"content": "</s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}]}
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": "|", "special_tokens_map_file": null, "tokenizer_file": null, "name_or_path": "DrishtiSharma/wav2vec2-xls-r-300m-kk-n2", "tokenizer_class": "Wav2Vec2CTCTokenizer", "processor_class": "Wav2Vec2ProcessorWithLM"}
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 149.98,
3
+ "train_loss": 0.9359948108413003,
4
+ "train_runtime": 8030.991,
5
+ "train_samples": 717,
6
+ "train_samples_per_second": 13.392,
7
+ "train_steps_per_second": 0.411
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,265 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 149.9777777777778,
5
+ "global_step": 3300,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 9.09,
12
+ "learning_rate": 4.3956000000000004e-05,
13
+ "loss": 9.6799,
14
+ "step": 200
15
+ },
16
+ {
17
+ "epoch": 9.09,
18
+ "eval_loss": 3.611884593963623,
19
+ "eval_runtime": 14.8755,
20
+ "eval_samples_per_second": 22.588,
21
+ "eval_steps_per_second": 2.823,
22
+ "eval_wer": 1.0,
23
+ "step": 200
24
+ },
25
+ {
26
+ "epoch": 18.18,
27
+ "learning_rate": 8.8356e-05,
28
+ "loss": 3.1332,
29
+ "step": 400
30
+ },
31
+ {
32
+ "epoch": 18.18,
33
+ "eval_loss": 2.535223960876465,
34
+ "eval_runtime": 14.5917,
35
+ "eval_samples_per_second": 23.027,
36
+ "eval_steps_per_second": 2.878,
37
+ "eval_wer": 1.005,
38
+ "step": 400
39
+ },
40
+ {
41
+ "epoch": 27.27,
42
+ "learning_rate": 0.000132756,
43
+ "loss": 1.0465,
44
+ "step": 600
45
+ },
46
+ {
47
+ "epoch": 27.27,
48
+ "eval_loss": 0.6169007420539856,
49
+ "eval_runtime": 14.7692,
50
+ "eval_samples_per_second": 22.75,
51
+ "eval_steps_per_second": 2.844,
52
+ "eval_wer": 0.682,
53
+ "step": 600
54
+ },
55
+ {
56
+ "epoch": 36.36,
57
+ "learning_rate": 0.000177156,
58
+ "loss": 0.3452,
59
+ "step": 800
60
+ },
61
+ {
62
+ "epoch": 36.36,
63
+ "eval_loss": 0.6572489142417908,
64
+ "eval_runtime": 14.4275,
65
+ "eval_samples_per_second": 23.289,
66
+ "eval_steps_per_second": 2.911,
67
+ "eval_wer": 0.607,
68
+ "step": 800
69
+ },
70
+ {
71
+ "epoch": 45.44,
72
+ "learning_rate": 0.000221556,
73
+ "loss": 0.2575,
74
+ "step": 1000
75
+ },
76
+ {
77
+ "epoch": 45.44,
78
+ "eval_loss": 0.6526808142662048,
79
+ "eval_runtime": 14.9602,
80
+ "eval_samples_per_second": 22.46,
81
+ "eval_steps_per_second": 2.807,
82
+ "eval_wer": 0.578,
83
+ "step": 1000
84
+ },
85
+ {
86
+ "epoch": 54.53,
87
+ "learning_rate": 0.0002028886956521739,
88
+ "loss": 0.2088,
89
+ "step": 1200
90
+ },
91
+ {
92
+ "epoch": 54.53,
93
+ "eval_loss": 0.6828188300132751,
94
+ "eval_runtime": 14.4501,
95
+ "eval_samples_per_second": 23.252,
96
+ "eval_steps_per_second": 2.907,
97
+ "eval_wer": 0.551,
98
+ "step": 1200
99
+ },
100
+ {
101
+ "epoch": 63.62,
102
+ "learning_rate": 0.00018358434782608698,
103
+ "loss": 0.158,
104
+ "step": 1400
105
+ },
106
+ {
107
+ "epoch": 63.62,
108
+ "eval_loss": 0.707419753074646,
109
+ "eval_runtime": 14.2371,
110
+ "eval_samples_per_second": 23.6,
111
+ "eval_steps_per_second": 2.95,
112
+ "eval_wer": 0.5575,
113
+ "step": 1400
114
+ },
115
+ {
116
+ "epoch": 72.71,
117
+ "learning_rate": 0.00016428,
118
+ "loss": 0.1309,
119
+ "step": 1600
120
+ },
121
+ {
122
+ "epoch": 72.71,
123
+ "eval_loss": 0.6523335576057434,
124
+ "eval_runtime": 14.2381,
125
+ "eval_samples_per_second": 23.599,
126
+ "eval_steps_per_second": 2.95,
127
+ "eval_wer": 0.5595,
128
+ "step": 1600
129
+ },
130
+ {
131
+ "epoch": 81.8,
132
+ "learning_rate": 0.00014497565217391306,
133
+ "loss": 0.1074,
134
+ "step": 1800
135
+ },
136
+ {
137
+ "epoch": 81.8,
138
+ "eval_loss": 0.7262120842933655,
139
+ "eval_runtime": 14.2676,
140
+ "eval_samples_per_second": 23.55,
141
+ "eval_steps_per_second": 2.944,
142
+ "eval_wer": 0.5415,
143
+ "step": 1800
144
+ },
145
+ {
146
+ "epoch": 90.89,
147
+ "learning_rate": 0.0001256713043478261,
148
+ "loss": 0.087,
149
+ "step": 2000
150
+ },
151
+ {
152
+ "epoch": 90.89,
153
+ "eval_loss": 0.7198619246482849,
154
+ "eval_runtime": 14.0317,
155
+ "eval_samples_per_second": 23.946,
156
+ "eval_steps_per_second": 2.993,
157
+ "eval_wer": 0.521,
158
+ "step": 2000
159
+ },
160
+ {
161
+ "epoch": 99.98,
162
+ "learning_rate": 0.00010636695652173913,
163
+ "loss": 0.0711,
164
+ "step": 2200
165
+ },
166
+ {
167
+ "epoch": 99.98,
168
+ "eval_loss": 0.7113156318664551,
169
+ "eval_runtime": 14.2007,
170
+ "eval_samples_per_second": 23.661,
171
+ "eval_steps_per_second": 2.958,
172
+ "eval_wer": 0.523,
173
+ "step": 2200
174
+ },
175
+ {
176
+ "epoch": 109.09,
177
+ "learning_rate": 8.706260869565218e-05,
178
+ "loss": 0.0601,
179
+ "step": 2400
180
+ },
181
+ {
182
+ "epoch": 109.09,
183
+ "eval_loss": 0.6862908601760864,
184
+ "eval_runtime": 13.9453,
185
+ "eval_samples_per_second": 24.094,
186
+ "eval_steps_per_second": 3.012,
187
+ "eval_wer": 0.496,
188
+ "step": 2400
189
+ },
190
+ {
191
+ "epoch": 118.18,
192
+ "learning_rate": 6.775826086956522e-05,
193
+ "loss": 0.0451,
194
+ "step": 2600
195
+ },
196
+ {
197
+ "epoch": 118.18,
198
+ "eval_loss": 0.6998496055603027,
199
+ "eval_runtime": 14.078,
200
+ "eval_samples_per_second": 23.867,
201
+ "eval_steps_per_second": 2.983,
202
+ "eval_wer": 0.483,
203
+ "step": 2600
204
+ },
205
+ {
206
+ "epoch": 127.27,
207
+ "learning_rate": 4.8453913043478265e-05,
208
+ "loss": 0.0378,
209
+ "step": 2800
210
+ },
211
+ {
212
+ "epoch": 127.27,
213
+ "eval_loss": 0.6970936059951782,
214
+ "eval_runtime": 14.3534,
215
+ "eval_samples_per_second": 23.409,
216
+ "eval_steps_per_second": 2.926,
217
+ "eval_wer": 0.4615,
218
+ "step": 2800
219
+ },
220
+ {
221
+ "epoch": 136.36,
222
+ "learning_rate": 2.9149565217391308e-05,
223
+ "loss": 0.0319,
224
+ "step": 3000
225
+ },
226
+ {
227
+ "epoch": 136.36,
228
+ "eval_loss": 0.7119385600090027,
229
+ "eval_runtime": 14.2911,
230
+ "eval_samples_per_second": 23.511,
231
+ "eval_steps_per_second": 2.939,
232
+ "eval_wer": 0.4475,
233
+ "step": 3000
234
+ },
235
+ {
236
+ "epoch": 145.44,
237
+ "learning_rate": 9.845217391304348e-06,
238
+ "loss": 0.0305,
239
+ "step": 3200
240
+ },
241
+ {
242
+ "epoch": 145.44,
243
+ "eval_loss": 0.7181466221809387,
244
+ "eval_runtime": 14.0233,
245
+ "eval_samples_per_second": 23.96,
246
+ "eval_steps_per_second": 2.995,
247
+ "eval_wer": 0.459,
248
+ "step": 3200
249
+ },
250
+ {
251
+ "epoch": 149.98,
252
+ "step": 3300,
253
+ "total_flos": 1.7623169369275333e+19,
254
+ "train_loss": 0.9359948108413003,
255
+ "train_runtime": 8030.991,
256
+ "train_samples_per_second": 13.392,
257
+ "train_steps_per_second": 0.411
258
+ }
259
+ ],
260
+ "max_steps": 3300,
261
+ "num_train_epochs": 150,
262
+ "total_flos": 1.7623169369275333e+19,
263
+ "trial_name": null,
264
+ "trial_params": null
265
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:adb0105863dcce00203582f0ef97e1325605ebb083f887a9da4eb3e96f1c91bb
3
+ size 2991
vocab.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"a": 1, "c": 2, "i": 3, "k": 4, "а": 5, "б": 6, "г": 7, "д": 8, "е": 9, "ж": 10, "з": 11, "и": 12, "й": 13, "к": 14, "л": 15, "м": 16, "н": 17, "о": 18, "п": 19, "р": 20, "с": 21, "т": 22, "у": 23, "х": 24, "ш": 25, "щ": 26, "ы": 27, "ю": 28, "я": 29, "і": 30, "ғ": 31, "қ": 32, "ң": 33, "ү": 34, "ұ": 35, "һ": 36, "ә": 37, "ө": 38, "|": 0, "[UNK]": 39, "[PAD]": 40}