benfoley commited on
Commit
285ae81
1 Parent(s): b3bab15
.gitignore ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ checkpoint-*
2
+ *.txt
README.md ADDED
@@ -0,0 +1 @@
 
 
1
+ Model page for FYTM_cv_all
all_results.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 40.0,
3
+ "eval_loss": 0.38463133573532104,
4
+ "eval_mem_cpu_alloc_delta": 36864,
5
+ "eval_mem_cpu_peaked_delta": 0,
6
+ "eval_mem_gpu_alloc_delta": 0,
7
+ "eval_mem_gpu_peaked_delta": 807446528,
8
+ "eval_runtime": 34.3132,
9
+ "eval_samples": 405,
10
+ "eval_samples_per_second": 11.803,
11
+ "eval_wer": 0.3688760806916426,
12
+ "init_mem_cpu_alloc_delta": 0,
13
+ "init_mem_cpu_peaked_delta": 0,
14
+ "init_mem_gpu_alloc_delta": 0,
15
+ "init_mem_gpu_peaked_delta": 0,
16
+ "train_mem_cpu_alloc_delta": 429268992,
17
+ "train_mem_cpu_peaked_delta": 10952704,
18
+ "train_mem_gpu_alloc_delta": 3826897920,
19
+ "train_mem_gpu_peaked_delta": 1210447360,
20
+ "train_runtime": 7468.243,
21
+ "train_samples": 1618,
22
+ "train_samples_per_second": 1.082
23
+ }
config.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/wav2vec2-large-xlsr-indonesian",
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": true,
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": 28,
74
+ "transformers_version": "4.6.0",
75
+ "vocab_size": 29
76
+ }
eval_results.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 40.0,
3
+ "eval_loss": 0.38463133573532104,
4
+ "eval_mem_cpu_alloc_delta": 36864,
5
+ "eval_mem_cpu_peaked_delta": 0,
6
+ "eval_mem_gpu_alloc_delta": 0,
7
+ "eval_mem_gpu_peaked_delta": 807446528,
8
+ "eval_runtime": 34.3132,
9
+ "eval_samples": 405,
10
+ "eval_samples_per_second": 11.803,
11
+ "eval_wer": 0.3688760806916426
12
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "feature_extractor_type": "Wav2Vec2FeatureExtractor",
4
+ "feature_size": 1,
5
+ "padding_side": "right",
6
+ "padding_value": 0.0,
7
+ "return_attention_mask": true,
8
+ "sampling_rate": 16000
9
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cdc09460276b6fd1915194cfb7ac6ac58a47e380fff6ede7e842d45037e7e08d
3
+ size 1262042545
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": 40.0,
3
+ "init_mem_cpu_alloc_delta": 0,
4
+ "init_mem_cpu_peaked_delta": 0,
5
+ "init_mem_gpu_alloc_delta": 0,
6
+ "init_mem_gpu_peaked_delta": 0,
7
+ "train_mem_cpu_alloc_delta": 429268992,
8
+ "train_mem_cpu_peaked_delta": 10952704,
9
+ "train_mem_gpu_alloc_delta": 3826897920,
10
+ "train_mem_gpu_peaked_delta": 1210447360,
11
+ "train_runtime": 7468.243,
12
+ "train_samples": 1618,
13
+ "train_samples_per_second": 1.082
14
+ }
trainer_state.json ADDED
@@ -0,0 +1,247 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 39.99753086419753,
5
+ "global_step": 8080,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 2.47,
12
+ "learning_rate": 4.96e-05,
13
+ "loss": 6.1701,
14
+ "step": 500
15
+ },
16
+ {
17
+ "epoch": 2.47,
18
+ "eval_loss": 2.8719067573547363,
19
+ "eval_runtime": 34.3427,
20
+ "eval_samples_per_second": 11.793,
21
+ "eval_wer": 1.0,
22
+ "step": 500
23
+ },
24
+ {
25
+ "epoch": 4.95,
26
+ "learning_rate": 9.960000000000001e-05,
27
+ "loss": 1.9379,
28
+ "step": 1000
29
+ },
30
+ {
31
+ "epoch": 4.95,
32
+ "eval_loss": 0.8804072737693787,
33
+ "eval_runtime": 34.3595,
34
+ "eval_samples_per_second": 11.787,
35
+ "eval_wer": 0.6778097982708934,
36
+ "step": 1000
37
+ },
38
+ {
39
+ "epoch": 7.42,
40
+ "learning_rate": 9.299435028248588e-05,
41
+ "loss": 1.0455,
42
+ "step": 1500
43
+ },
44
+ {
45
+ "epoch": 7.42,
46
+ "eval_loss": 0.6408620476722717,
47
+ "eval_runtime": 34.2855,
48
+ "eval_samples_per_second": 11.813,
49
+ "eval_wer": 0.5227665706051873,
50
+ "step": 1500
51
+ },
52
+ {
53
+ "epoch": 9.9,
54
+ "learning_rate": 8.593220338983051e-05,
55
+ "loss": 0.7456,
56
+ "step": 2000
57
+ },
58
+ {
59
+ "epoch": 9.9,
60
+ "eval_loss": 0.6179625391960144,
61
+ "eval_runtime": 34.3686,
62
+ "eval_samples_per_second": 11.784,
63
+ "eval_wer": 0.4755043227665706,
64
+ "step": 2000
65
+ },
66
+ {
67
+ "epoch": 12.38,
68
+ "learning_rate": 7.887005649717515e-05,
69
+ "loss": 0.5543,
70
+ "step": 2500
71
+ },
72
+ {
73
+ "epoch": 12.38,
74
+ "eval_loss": 0.4895203411579132,
75
+ "eval_runtime": 34.1759,
76
+ "eval_samples_per_second": 11.85,
77
+ "eval_wer": 0.4340057636887608,
78
+ "step": 2500
79
+ },
80
+ {
81
+ "epoch": 14.85,
82
+ "learning_rate": 7.180790960451977e-05,
83
+ "loss": 0.4526,
84
+ "step": 3000
85
+ },
86
+ {
87
+ "epoch": 14.85,
88
+ "eval_loss": 0.4277503192424774,
89
+ "eval_runtime": 34.18,
90
+ "eval_samples_per_second": 11.849,
91
+ "eval_wer": 0.42536023054755046,
92
+ "step": 3000
93
+ },
94
+ {
95
+ "epoch": 17.33,
96
+ "learning_rate": 6.474576271186441e-05,
97
+ "loss": 0.3827,
98
+ "step": 3500
99
+ },
100
+ {
101
+ "epoch": 17.33,
102
+ "eval_loss": 0.3866088092327118,
103
+ "eval_runtime": 34.3584,
104
+ "eval_samples_per_second": 11.788,
105
+ "eval_wer": 0.414985590778098,
106
+ "step": 3500
107
+ },
108
+ {
109
+ "epoch": 19.8,
110
+ "learning_rate": 5.7683615819209045e-05,
111
+ "loss": 0.3226,
112
+ "step": 4000
113
+ },
114
+ {
115
+ "epoch": 19.8,
116
+ "eval_loss": 0.38195741176605225,
117
+ "eval_runtime": 34.2414,
118
+ "eval_samples_per_second": 11.828,
119
+ "eval_wer": 0.392507204610951,
120
+ "step": 4000
121
+ },
122
+ {
123
+ "epoch": 22.28,
124
+ "learning_rate": 5.0635593220338987e-05,
125
+ "loss": 0.295,
126
+ "step": 4500
127
+ },
128
+ {
129
+ "epoch": 22.28,
130
+ "eval_loss": 0.41771677136421204,
131
+ "eval_runtime": 34.2045,
132
+ "eval_samples_per_second": 11.841,
133
+ "eval_wer": 0.3919308357348703,
134
+ "step": 4500
135
+ },
136
+ {
137
+ "epoch": 24.75,
138
+ "learning_rate": 4.357344632768362e-05,
139
+ "loss": 0.2613,
140
+ "step": 5000
141
+ },
142
+ {
143
+ "epoch": 24.75,
144
+ "eval_loss": 0.41437995433807373,
145
+ "eval_runtime": 34.2585,
146
+ "eval_samples_per_second": 11.822,
147
+ "eval_wer": 0.3855907780979827,
148
+ "step": 5000
149
+ },
150
+ {
151
+ "epoch": 27.23,
152
+ "learning_rate": 3.651129943502825e-05,
153
+ "loss": 0.228,
154
+ "step": 5500
155
+ },
156
+ {
157
+ "epoch": 27.23,
158
+ "eval_loss": 0.4071010947227478,
159
+ "eval_runtime": 34.2741,
160
+ "eval_samples_per_second": 11.817,
161
+ "eval_wer": 0.38962536023054756,
162
+ "step": 5500
163
+ },
164
+ {
165
+ "epoch": 29.7,
166
+ "learning_rate": 2.9449152542372883e-05,
167
+ "loss": 0.2081,
168
+ "step": 6000
169
+ },
170
+ {
171
+ "epoch": 29.7,
172
+ "eval_loss": 0.3746974468231201,
173
+ "eval_runtime": 34.1866,
174
+ "eval_samples_per_second": 11.847,
175
+ "eval_wer": 0.38213256484149855,
176
+ "step": 6000
177
+ },
178
+ {
179
+ "epoch": 32.18,
180
+ "learning_rate": 2.2387005649717515e-05,
181
+ "loss": 0.1906,
182
+ "step": 6500
183
+ },
184
+ {
185
+ "epoch": 32.18,
186
+ "eval_loss": 0.396076500415802,
187
+ "eval_runtime": 34.2512,
188
+ "eval_samples_per_second": 11.824,
189
+ "eval_wer": 0.37175792507204614,
190
+ "step": 6500
191
+ },
192
+ {
193
+ "epoch": 34.65,
194
+ "learning_rate": 1.535310734463277e-05,
195
+ "loss": 0.1752,
196
+ "step": 7000
197
+ },
198
+ {
199
+ "epoch": 34.65,
200
+ "eval_loss": 0.37449249625205994,
201
+ "eval_runtime": 34.2395,
202
+ "eval_samples_per_second": 11.828,
203
+ "eval_wer": 0.3729106628242075,
204
+ "step": 7000
205
+ },
206
+ {
207
+ "epoch": 37.13,
208
+ "learning_rate": 8.290960451977402e-06,
209
+ "loss": 0.176,
210
+ "step": 7500
211
+ },
212
+ {
213
+ "epoch": 37.13,
214
+ "eval_loss": 0.3883863687515259,
215
+ "eval_runtime": 34.2865,
216
+ "eval_samples_per_second": 11.812,
217
+ "eval_wer": 0.37118155619596543,
218
+ "step": 7500
219
+ },
220
+ {
221
+ "epoch": 39.6,
222
+ "learning_rate": 1.228813559322034e-06,
223
+ "loss": 0.1531,
224
+ "step": 8000
225
+ },
226
+ {
227
+ "epoch": 39.6,
228
+ "eval_loss": 0.3848457336425781,
229
+ "eval_runtime": 34.1853,
230
+ "eval_samples_per_second": 11.847,
231
+ "eval_wer": 0.368299711815562,
232
+ "step": 8000
233
+ },
234
+ {
235
+ "epoch": 40.0,
236
+ "step": 8080,
237
+ "total_flos": 3.820109864655936e+16,
238
+ "train_runtime": 7468.243,
239
+ "train_samples_per_second": 1.082
240
+ }
241
+ ],
242
+ "max_steps": 8080,
243
+ "num_train_epochs": 40,
244
+ "total_flos": 3.820109864655936e+16,
245
+ "trial_name": null,
246
+ "trial_params": null
247
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a5f6240899039c0c218f434b65298bd074553c3b87d7b78557f45211b4bae09
3
+ size 2479
vocab.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"c": 0, " ": 1, "q": 2, "v": 3, "m": 4, "e": 5, "j": 6, "y": 7, "u": 8, "h": 9, "a": 10, "k": 11, "b": 12, "d": 13, "r": 14, "p": 15, "t": 16, "n": 17, "s": 18, "w": 19, "g": 20, "z": 21, "f": 22, "o": 23, "i": 24, "l": 25, "x": 26, "[UNK]": 27, "[PAD]": 28}