comodoro commited on
Commit
4f0c63f
1 Parent(s): 13d24c4

Added Polish model

Browse files
added_tokens.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"<s>": 37, "</s>": 38}
config.json ADDED
@@ -0,0 +1,108 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.1,
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
+ "gradient_checkpointing": false,
56
+ "hidden_act": "gelu",
57
+ "hidden_dropout": 0.1,
58
+ "hidden_size": 1024,
59
+ "initializer_range": 0.02,
60
+ "intermediate_size": 4096,
61
+ "layer_norm_eps": 1e-05,
62
+ "layerdrop": 0.1,
63
+ "mask_feature_length": 10,
64
+ "mask_feature_min_masks": 0,
65
+ "mask_feature_prob": 0.0,
66
+ "mask_time_length": 10,
67
+ "mask_time_min_masks": 2,
68
+ "mask_time_prob": 0.05,
69
+ "model_type": "wav2vec2",
70
+ "num_adapter_layers": 3,
71
+ "num_attention_heads": 16,
72
+ "num_codevector_groups": 2,
73
+ "num_codevectors_per_group": 320,
74
+ "num_conv_pos_embedding_groups": 16,
75
+ "num_conv_pos_embeddings": 128,
76
+ "num_feat_extract_layers": 7,
77
+ "num_hidden_layers": 24,
78
+ "num_negatives": 100,
79
+ "output_hidden_size": 1024,
80
+ "pad_token_id": 36,
81
+ "proj_codevector_dim": 768,
82
+ "tdnn_dilation": [
83
+ 1,
84
+ 2,
85
+ 3,
86
+ 1,
87
+ 1
88
+ ],
89
+ "tdnn_dim": [
90
+ 512,
91
+ 512,
92
+ 512,
93
+ 512,
94
+ 1500
95
+ ],
96
+ "tdnn_kernel": [
97
+ 5,
98
+ 3,
99
+ 3,
100
+ 1,
101
+ 1
102
+ ],
103
+ "torch_dtype": "float32",
104
+ "transformers_version": "4.16.0.dev0",
105
+ "use_weighted_layer_sum": false,
106
+ "vocab_size": 39,
107
+ "xvector_output_dim": 512
108
+ }
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.0,
7
+ "processor_class": "Wav2Vec2Processor",
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:9207e40dd155719fa2417d1c9f6b98fb40c2034307e3325e7c429cf8cf219184
3
+ size 1262083569
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}]}
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": "./", "tokenizer_class": "Wav2Vec2CTCTokenizer", "processor_class": "Wav2Vec2Processor"}
trainer_state.json ADDED
@@ -0,0 +1,224 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 4.669540229885057,
5
+ "global_step": 3250,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.36,
12
+ "learning_rate": 4.94e-05,
13
+ "loss": 7.315,
14
+ "step": 250
15
+ },
16
+ {
17
+ "epoch": 0.36,
18
+ "eval_cer": 1.0,
19
+ "eval_loss": 3.364396810531616,
20
+ "eval_runtime": 411.3291,
21
+ "eval_samples_per_second": 18.839,
22
+ "eval_steps_per_second": 2.356,
23
+ "eval_wer": 1.0,
24
+ "step": 250
25
+ },
26
+ {
27
+ "epoch": 0.72,
28
+ "learning_rate": 9.94e-05,
29
+ "loss": 3.2126,
30
+ "step": 500
31
+ },
32
+ {
33
+ "epoch": 0.72,
34
+ "eval_cer": 1.0,
35
+ "eval_loss": 3.178248405456543,
36
+ "eval_runtime": 409.0228,
37
+ "eval_samples_per_second": 18.945,
38
+ "eval_steps_per_second": 2.369,
39
+ "eval_wer": 1.0,
40
+ "step": 500
41
+ },
42
+ {
43
+ "epoch": 1.08,
44
+ "learning_rate": 9.992892086330936e-05,
45
+ "loss": 3.1414,
46
+ "step": 750
47
+ },
48
+ {
49
+ "epoch": 1.08,
50
+ "eval_cer": 1.0,
51
+ "eval_loss": 3.1066977977752686,
52
+ "eval_runtime": 414.126,
53
+ "eval_samples_per_second": 18.712,
54
+ "eval_steps_per_second": 2.34,
55
+ "eval_wer": 1.0,
56
+ "step": 750
57
+ },
58
+ {
59
+ "epoch": 1.44,
60
+ "learning_rate": 9.985697841726619e-05,
61
+ "loss": 1.9013,
62
+ "step": 1000
63
+ },
64
+ {
65
+ "epoch": 1.44,
66
+ "eval_cer": 0.12787848954973915,
67
+ "eval_loss": 0.5506780743598938,
68
+ "eval_runtime": 412.7461,
69
+ "eval_samples_per_second": 18.774,
70
+ "eval_steps_per_second": 2.348,
71
+ "eval_wer": 0.5514430132072395,
72
+ "step": 1000
73
+ },
74
+ {
75
+ "epoch": 1.8,
76
+ "learning_rate": 9.978503597122302e-05,
77
+ "loss": 0.5632,
78
+ "step": 1250
79
+ },
80
+ {
81
+ "epoch": 1.8,
82
+ "eval_cer": 0.06747688721413134,
83
+ "eval_loss": 0.28180211782455444,
84
+ "eval_runtime": 412.6671,
85
+ "eval_samples_per_second": 18.778,
86
+ "eval_steps_per_second": 2.348,
87
+ "eval_wer": 0.30760729749805243,
88
+ "step": 1250
89
+ },
90
+ {
91
+ "epoch": 2.16,
92
+ "learning_rate": 9.971309352517985e-05,
93
+ "loss": 0.388,
94
+ "step": 1500
95
+ },
96
+ {
97
+ "epoch": 2.16,
98
+ "eval_cer": 0.053711059057835714,
99
+ "eval_loss": 0.2308780699968338,
100
+ "eval_runtime": 524.2327,
101
+ "eval_samples_per_second": 14.782,
102
+ "eval_steps_per_second": 1.848,
103
+ "eval_wer": 0.24573799300686633,
104
+ "step": 1500
105
+ },
106
+ {
107
+ "epoch": 2.51,
108
+ "learning_rate": 9.96411510791367e-05,
109
+ "loss": 0.3175,
110
+ "step": 1750
111
+ },
112
+ {
113
+ "epoch": 2.51,
114
+ "eval_cer": 0.0491677133901393,
115
+ "eval_loss": 0.2099951058626175,
116
+ "eval_runtime": 529.8067,
117
+ "eval_samples_per_second": 14.626,
118
+ "eval_steps_per_second": 1.829,
119
+ "eval_wer": 0.22468612424588294,
120
+ "step": 1750
121
+ },
122
+ {
123
+ "epoch": 2.87,
124
+ "learning_rate": 9.956920863309353e-05,
125
+ "loss": 0.294,
126
+ "step": 2000
127
+ },
128
+ {
129
+ "epoch": 2.87,
130
+ "eval_cer": 0.046197281913749985,
131
+ "eval_loss": 0.19443732500076294,
132
+ "eval_runtime": 533.3896,
133
+ "eval_samples_per_second": 14.528,
134
+ "eval_steps_per_second": 1.817,
135
+ "eval_wer": 0.21001141366378606,
136
+ "step": 2000
137
+ },
138
+ {
139
+ "epoch": 3.23,
140
+ "learning_rate": 9.949726618705036e-05,
141
+ "loss": 0.2543,
142
+ "step": 2250
143
+ },
144
+ {
145
+ "epoch": 3.23,
146
+ "eval_cer": 0.04413495377442826,
147
+ "eval_loss": 0.19428253173828125,
148
+ "eval_runtime": 534.1775,
149
+ "eval_samples_per_second": 14.506,
150
+ "eval_steps_per_second": 1.814,
151
+ "eval_wer": 0.19475696142906318,
152
+ "step": 2250
153
+ },
154
+ {
155
+ "epoch": 3.59,
156
+ "learning_rate": 9.94253237410072e-05,
157
+ "loss": 0.2493,
158
+ "step": 2500
159
+ },
160
+ {
161
+ "epoch": 3.59,
162
+ "eval_cer": 0.04232723404736848,
163
+ "eval_loss": 0.1892683357000351,
164
+ "eval_runtime": 528.5117,
165
+ "eval_samples_per_second": 14.662,
166
+ "eval_steps_per_second": 1.833,
167
+ "eval_wer": 0.18792688008406255,
168
+ "step": 2500
169
+ },
170
+ {
171
+ "epoch": 3.95,
172
+ "learning_rate": 9.935338129496404e-05,
173
+ "loss": 0.2333,
174
+ "step": 2750
175
+ },
176
+ {
177
+ "epoch": 3.95,
178
+ "eval_cer": 0.04126919464530219,
179
+ "eval_loss": 0.1779049038887024,
180
+ "eval_runtime": 417.6945,
181
+ "eval_samples_per_second": 18.552,
182
+ "eval_steps_per_second": 2.32,
183
+ "eval_wer": 0.18323459608312045,
184
+ "step": 2750
185
+ },
186
+ {
187
+ "epoch": 4.31,
188
+ "learning_rate": 9.928143884892087e-05,
189
+ "loss": 0.2131,
190
+ "step": 3000
191
+ },
192
+ {
193
+ "epoch": 4.31,
194
+ "eval_cer": 0.040952348621153996,
195
+ "eval_loss": 0.18109484016895294,
196
+ "eval_runtime": 409.5311,
197
+ "eval_samples_per_second": 18.922,
198
+ "eval_steps_per_second": 2.366,
199
+ "eval_wer": 0.17970179538743047,
200
+ "step": 3000
201
+ },
202
+ {
203
+ "epoch": 4.67,
204
+ "learning_rate": 9.92094964028777e-05,
205
+ "loss": 0.1992,
206
+ "step": 3250
207
+ },
208
+ {
209
+ "epoch": 4.67,
210
+ "eval_cer": 0.03845435719862851,
211
+ "eval_loss": 0.17126505076885223,
212
+ "eval_runtime": 422.692,
213
+ "eval_samples_per_second": 18.332,
214
+ "eval_steps_per_second": 2.292,
215
+ "eval_wer": 0.169665018026342,
216
+ "step": 3250
217
+ }
218
+ ],
219
+ "max_steps": 348000,
220
+ "num_train_epochs": 500,
221
+ "total_flos": 1.643409928882224e+19,
222
+ "trial_name": null,
223
+ "trial_params": null
224
+ }
vocab.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"a": 1, "b": 2, "c": 3, "d": 4, "e": 5, "f": 6, "g": 7, "h": 8, "i": 9, "j": 10, "k": 11, "l": 12, "m": 13, "n": 14, "o": 15, "p": 16, "q": 17, "r": 18, "s": 19, "t": 20, "u": 21, "v": 22, "w": 23, "x": 24, "y": 25, "z": 26, "ó": 27, "č": 28, "ł": 29, "ń": 30, "ś": 31, "š": 32, "ź": 33, "ż": 34, "|": 0, "[UNK]": 35, "[PAD]": 36}