cstorm125 commited on
Commit
c9452a5
1 Parent(s): a690ae6

add checkpoint 135k

Browse files

config.json ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/wav2vec2-large-xlsr-53",
3
+ "activation_dropout": 0.0,
4
+ "apply_spec_augment": true,
5
+ "architectures": [
6
+ "Wav2Vec2ForCTC"
7
+ ],
8
+ "attention_dropout": 0.1,
9
+ "bos_token_id": 1,
10
+ "codevector_dim": 768,
11
+ "contrastive_logits_temperature": 0.1,
12
+ "conv_bias": true,
13
+ "conv_dim": [
14
+ 512,
15
+ 512,
16
+ 512,
17
+ 512,
18
+ 512,
19
+ 512,
20
+ 512
21
+ ],
22
+ "conv_kernel": [
23
+ 10,
24
+ 3,
25
+ 3,
26
+ 3,
27
+ 3,
28
+ 2,
29
+ 2
30
+ ],
31
+ "conv_stride": [
32
+ 5,
33
+ 2,
34
+ 2,
35
+ 2,
36
+ 2,
37
+ 2,
38
+ 2
39
+ ],
40
+ "ctc_loss_reduction": "mean",
41
+ "ctc_zero_infinity": false,
42
+ "diversity_loss_weight": 0.1,
43
+ "do_stable_layer_norm": true,
44
+ "eos_token_id": 2,
45
+ "feat_extract_activation": "gelu",
46
+ "feat_extract_dropout": 0.0,
47
+ "feat_extract_norm": "layer",
48
+ "feat_proj_dropout": 0.0,
49
+ "feat_quantizer_dropout": 0.0,
50
+ "final_dropout": 0.0,
51
+ "gradient_checkpointing": true,
52
+ "hidden_act": "gelu",
53
+ "hidden_dropout": 0.1,
54
+ "hidden_size": 1024,
55
+ "initializer_range": 0.02,
56
+ "intermediate_size": 4096,
57
+ "layer_norm_eps": 1e-05,
58
+ "layerdrop": 0.1,
59
+ "mask_channel_length": 10,
60
+ "mask_channel_min_space": 1,
61
+ "mask_channel_other": 0.0,
62
+ "mask_channel_prob": 0.0,
63
+ "mask_channel_selection": "static",
64
+ "mask_feature_length": 10,
65
+ "mask_feature_prob": 0.0,
66
+ "mask_time_length": 10,
67
+ "mask_time_min_space": 1,
68
+ "mask_time_other": 0.0,
69
+ "mask_time_prob": 0.05,
70
+ "mask_time_selection": "static",
71
+ "model_type": "wav2vec2",
72
+ "num_attention_heads": 16,
73
+ "num_codevector_groups": 2,
74
+ "num_codevectors_per_group": 320,
75
+ "num_conv_pos_embedding_groups": 16,
76
+ "num_conv_pos_embeddings": 128,
77
+ "num_feat_extract_layers": 7,
78
+ "num_hidden_layers": 24,
79
+ "num_negatives": 100,
80
+ "pad_token_id": 69,
81
+ "proj_codevector_dim": 768,
82
+ "torch_dtype": "float32",
83
+ "transformers_version": "4.9.1",
84
+ "vocab_size": 70
85
+ }
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": false,
8
+ "sampling_rate": 16000
9
+ }
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:040892c8367ed2663b43590614fcaee20e683cc064af55f65e3baf94db0473d8
3
+ size 15587
scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eaad0a550aad796c00c2715fb3dd07c74e5ca1520507a15a505e4faf0123239f
3
+ size 559
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb118fee7b3cd82205b90d8e4ffc01eee7dfb1f3f32f5210cf6b36e69150bc49
3
+ size 623
trainer_state.json ADDED
@@ -0,0 +1,2041 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 1.0,
3
+ "best_model_checkpoint": "../data/wav2vec2-large-xlsr-53-thai/checkpoint-1000",
4
+ "epoch": 99.77827050997783,
5
+ "global_step": 135000,
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.74,
12
+ "learning_rate": 9.970000000000001e-05,
13
+ "loss": 7.1725,
14
+ "step": 1000
15
+ },
16
+ {
17
+ "epoch": 0.74,
18
+ "eval_loss": 3.039940595626831,
19
+ "eval_runtime": 185.7874,
20
+ "eval_samples_per_second": 16.293,
21
+ "eval_steps_per_second": 0.511,
22
+ "eval_wer": 1.0,
23
+ "step": 1000
24
+ },
25
+ {
26
+ "epoch": 1.48,
27
+ "learning_rate": 9.925763216679077e-05,
28
+ "loss": 1.0788,
29
+ "step": 2000
30
+ },
31
+ {
32
+ "epoch": 1.48,
33
+ "eval_loss": 0.6130112409591675,
34
+ "eval_runtime": 176.9194,
35
+ "eval_samples_per_second": 17.109,
36
+ "eval_steps_per_second": 0.537,
37
+ "eval_wer": 0.6820256330103157,
38
+ "step": 2000
39
+ },
40
+ {
41
+ "epoch": 2.22,
42
+ "learning_rate": 9.851303052866717e-05,
43
+ "loss": 0.6339,
44
+ "step": 3000
45
+ },
46
+ {
47
+ "epoch": 2.22,
48
+ "eval_loss": 0.49624550342559814,
49
+ "eval_runtime": 175.6715,
50
+ "eval_samples_per_second": 17.231,
51
+ "eval_steps_per_second": 0.541,
52
+ "eval_wer": 0.6168802750859643,
53
+ "step": 3000
54
+ },
55
+ {
56
+ "epoch": 2.96,
57
+ "learning_rate": 9.776842889054357e-05,
58
+ "loss": 0.5295,
59
+ "step": 4000
60
+ },
61
+ {
62
+ "epoch": 2.96,
63
+ "eval_loss": 0.4389719069004059,
64
+ "eval_runtime": 176.6072,
65
+ "eval_samples_per_second": 17.14,
66
+ "eval_steps_per_second": 0.538,
67
+ "eval_wer": 0.577117849327915,
68
+ "step": 4000
69
+ },
70
+ {
71
+ "epoch": 3.7,
72
+ "learning_rate": 9.702382725241996e-05,
73
+ "loss": 0.4616,
74
+ "step": 5000
75
+ },
76
+ {
77
+ "epoch": 3.7,
78
+ "eval_loss": 0.4208119809627533,
79
+ "eval_runtime": 196.7481,
80
+ "eval_samples_per_second": 15.385,
81
+ "eval_steps_per_second": 0.483,
82
+ "eval_wer": 0.5618005626758362,
83
+ "step": 5000
84
+ },
85
+ {
86
+ "epoch": 4.43,
87
+ "learning_rate": 9.627922561429635e-05,
88
+ "loss": 0.4206,
89
+ "step": 6000
90
+ },
91
+ {
92
+ "epoch": 4.43,
93
+ "eval_loss": 0.4009915888309479,
94
+ "eval_runtime": 180.4803,
95
+ "eval_samples_per_second": 16.772,
96
+ "eval_steps_per_second": 0.526,
97
+ "eval_wer": 0.5411691153485464,
98
+ "step": 6000
99
+ },
100
+ {
101
+ "epoch": 5.17,
102
+ "learning_rate": 9.553462397617276e-05,
103
+ "loss": 0.3908,
104
+ "step": 7000
105
+ },
106
+ {
107
+ "epoch": 5.17,
108
+ "eval_loss": 0.399889200925827,
109
+ "eval_runtime": 189.7255,
110
+ "eval_samples_per_second": 15.955,
111
+ "eval_steps_per_second": 0.501,
112
+ "eval_wer": 0.5384807752422632,
113
+ "step": 7000
114
+ },
115
+ {
116
+ "epoch": 5.91,
117
+ "learning_rate": 9.479002233804914e-05,
118
+ "loss": 0.3659,
119
+ "step": 8000
120
+ },
121
+ {
122
+ "epoch": 5.91,
123
+ "eval_loss": 0.3719017505645752,
124
+ "eval_runtime": 184.5994,
125
+ "eval_samples_per_second": 16.398,
126
+ "eval_steps_per_second": 0.515,
127
+ "eval_wer": 0.5141606752110034,
128
+ "step": 8000
129
+ },
130
+ {
131
+ "epoch": 6.65,
132
+ "learning_rate": 9.404542069992555e-05,
133
+ "loss": 0.338,
134
+ "step": 9000
135
+ },
136
+ {
137
+ "epoch": 6.65,
138
+ "eval_loss": 0.35533592104911804,
139
+ "eval_runtime": 188.5325,
140
+ "eval_samples_per_second": 16.056,
141
+ "eval_steps_per_second": 0.504,
142
+ "eval_wer": 0.5101594248202563,
143
+ "step": 9000
144
+ },
145
+ {
146
+ "epoch": 7.39,
147
+ "learning_rate": 9.330156366344007e-05,
148
+ "loss": 0.3222,
149
+ "step": 10000
150
+ },
151
+ {
152
+ "epoch": 7.39,
153
+ "eval_loss": 0.40722423791885376,
154
+ "eval_runtime": 195.2723,
155
+ "eval_samples_per_second": 15.501,
156
+ "eval_steps_per_second": 0.487,
157
+ "eval_wer": 0.49309159112222567,
158
+ "step": 10000
159
+ },
160
+ {
161
+ "epoch": 8.13,
162
+ "learning_rate": 9.255696202531646e-05,
163
+ "loss": 0.31,
164
+ "step": 11000
165
+ },
166
+ {
167
+ "epoch": 8.13,
168
+ "eval_loss": 0.37148234248161316,
169
+ "eval_runtime": 182.6314,
170
+ "eval_samples_per_second": 16.574,
171
+ "eval_steps_per_second": 0.52,
172
+ "eval_wer": 0.48715223507346045,
173
+ "step": 11000
174
+ },