Chillarmo commited on
Commit
858d1c6
·
verified ·
1 Parent(s): a0f2870

Training complete - Strict Armenian WER: 42%

Browse files
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ language:
4
+ - hy
5
+ tags:
6
+ - asr
7
+ - audio
8
+ - speech
9
+ - whisper
10
+ - low-resource
11
+ - generated_from_trainer
12
+ datasets:
13
+ - Chillarmo/common_voice_20_armenian
14
+ - mozilla-foundation/common_voice_20_0
15
+ model-index:
16
+ - name: checkpoint_9000
17
+ results: []
18
+ ---
19
+
20
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
21
+ should probably proofread and complete it, then remove this comment. -->
22
+
23
+ # checkpoint_9000
24
+
25
+ This model was trained from scratch on the Common Voice 20.0 dataset.
26
+
27
+ ## Model description
28
+
29
+ More information needed
30
+
31
+ ## Intended uses & limitations
32
+
33
+ More information needed
34
+
35
+ ## Training and evaluation data
36
+
37
+ More information needed
38
+
39
+ ## Training procedure
40
+
41
+ ### Training hyperparameters
42
+
43
+ The following hyperparameters were used during training:
44
+ - learning_rate: 5e-05
45
+ - train_batch_size: 8
46
+ - eval_batch_size: 16
47
+ - seed: 42
48
+ - optimizer: Use adamw_torch_fused with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
49
+ - lr_scheduler_type: linear
50
+ - num_epochs: 3.0
51
+ - mixed_precision_training: Native AMP
52
+
53
+ ### Framework versions
54
+
55
+ - Transformers 4.56.2
56
+ - Pytorch 2.8.0+cu129
57
+ - Datasets 3.5.0
58
+ - Tokenizers 0.22.1
config.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_dropout": 0.0,
3
+ "activation_function": "gelu",
4
+ "apply_spec_augment": false,
5
+ "architectures": [
6
+ "WhisperForConditionalGeneration"
7
+ ],
8
+ "attention_dropout": 0.0,
9
+ "begin_suppress_tokens": null,
10
+ "bos_token_id": 1,
11
+ "classifier_proj_size": 256,
12
+ "d_model": 384,
13
+ "decoder_attention_heads": 6,
14
+ "decoder_ffn_dim": 1024,
15
+ "decoder_layerdrop": 0.0,
16
+ "decoder_layers": 2,
17
+ "decoder_start_token_id": 1,
18
+ "dropout": 0.0,
19
+ "dtype": "float16",
20
+ "encoder_attention_heads": 6,
21
+ "encoder_ffn_dim": 1536,
22
+ "encoder_layerdrop": 0.0,
23
+ "encoder_layers": 4,
24
+ "eos_token_id": 2,
25
+ "forced_decoder_ids": null,
26
+ "init_std": 0.02,
27
+ "is_encoder_decoder": true,
28
+ "mask_feature_length": 10,
29
+ "mask_feature_min_masks": 0,
30
+ "mask_feature_prob": 0.0,
31
+ "mask_time_length": 10,
32
+ "mask_time_min_masks": 2,
33
+ "mask_time_prob": 0.05,
34
+ "max_length": null,
35
+ "max_source_positions": 1500,
36
+ "max_target_positions": 448,
37
+ "median_filter_width": 7,
38
+ "model_type": "whisper",
39
+ "num_hidden_layers": 4,
40
+ "num_mel_bins": 80,
41
+ "pad_token_id": 0,
42
+ "scale_embedding": false,
43
+ "transformers_version": "4.56.2",
44
+ "use_cache": true,
45
+ "use_weighted_layer_sum": false,
46
+ "vocab_size": 5000
47
+ }
generation_config.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "begin_suppress_tokens": [],
3
+ "bos_token_id": 1,
4
+ "decoder_start_token_id": 1,
5
+ "eos_token_id": 2,
6
+ "forced_decoder_ids": null,
7
+ "max_length": 448,
8
+ "pad_token_id": 0,
9
+ "suppress_tokens": [],
10
+ "transformers_version": "4.56.2",
11
+ "vocab_size": 5000
12
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:26f33a614a6e03eb2afbe7277fd9511446ce241dce7f1a70e23756b7b0fe7cbd
3
+ size 28504056
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:522cb173c1edaed339d0d2a44d3d383f4981a84159179948b3cad975dccebc86
3
+ size 48340299
preprocessor_config.json ADDED
The diff for this file is too large to render. See raw diff
 
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5ac7846d433abea5c16b7eeb3755a7faf76de4592a9b9d134ab5685daa27b433
3
+ size 14645
scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56410013d5a6584db199c4b19fe4268f15b8b3cb219a0f37341404083055bc16
3
+ size 1383
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b11393457f05d4fbe28bb082192a3c5fbb59c96c0a7126f55f48cbb3b145578
3
+ size 1465
special_tokens_map.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": "<pad>",
17
+ "unk_token": {
18
+ "content": "<unk>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ }
24
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<pad>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<s>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<unk>",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ }
36
+ },
37
+ "bos_token": "<s>",
38
+ "clean_up_tokenization_spaces": false,
39
+ "eos_token": "</s>",
40
+ "extra_special_tokens": {},
41
+ "model_max_length": 448,
42
+ "pad_token": "<pad>",
43
+ "tokenizer_class": "WhisperTokenizer",
44
+ "unk_token": "<unk>"
45
+ }
trainer_state.json ADDED
@@ -0,0 +1,664 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 11.734028683181226,
6
+ "eval_steps": 500,
7
+ "global_step": 9000,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.1303780964797914,
14
+ "grad_norm": 2.199315071105957,
15
+ "learning_rate": 1.9800000000000004e-05,
16
+ "loss": 8.1376,
17
+ "step": 100
18
+ },
19
+ {
20
+ "epoch": 0.2607561929595828,
21
+ "grad_norm": 1.6398614645004272,
22
+ "learning_rate": 3.9800000000000005e-05,
23
+ "loss": 7.2341,
24
+ "step": 200
25
+ },
26
+ {
27
+ "epoch": 0.39113428943937417,
28
+ "grad_norm": 1.6936060190200806,
29
+ "learning_rate": 5.9800000000000003e-05,
30
+ "loss": 6.5612,
31
+ "step": 300
32
+ },
33
+ {
34
+ "epoch": 0.5215123859191656,
35
+ "grad_norm": 2.0208520889282227,
36
+ "learning_rate": 7.98e-05,
37
+ "loss": 6.2796,
38
+ "step": 400
39
+ },
40
+ {
41
+ "epoch": 0.651890482398957,
42
+ "grad_norm": 2.6271910667419434,
43
+ "learning_rate": 9.98e-05,
44
+ "loss": 6.0939,
45
+ "step": 500
46
+ },
47
+ {
48
+ "epoch": 0.7822685788787483,
49
+ "grad_norm": 3.186422348022461,
50
+ "learning_rate": 9.913913043478262e-05,
51
+ "loss": 5.8614,
52
+ "step": 600
53
+ },
54
+ {
55
+ "epoch": 0.9126466753585397,
56
+ "grad_norm": 2.3998301029205322,
57
+ "learning_rate": 9.826956521739131e-05,
58
+ "loss": 5.6819,
59
+ "step": 700
60
+ },
61
+ {
62
+ "epoch": 1.0430247718383312,
63
+ "grad_norm": 2.232499837875366,
64
+ "learning_rate": 9.74e-05,
65
+ "loss": 5.568,
66
+ "step": 800
67
+ },
68
+ {
69
+ "epoch": 1.1734028683181226,
70
+ "grad_norm": 3.5190041065216064,
71
+ "learning_rate": 9.65304347826087e-05,
72
+ "loss": 5.4076,
73
+ "step": 900
74
+ },
75
+ {
76
+ "epoch": 1.303780964797914,
77
+ "grad_norm": 3.0163352489471436,
78
+ "learning_rate": 9.566086956521739e-05,
79
+ "loss": 5.3031,
80
+ "step": 1000
81
+ },
82
+ {
83
+ "epoch": 1.4341590612777053,
84
+ "grad_norm": 2.686789035797119,
85
+ "learning_rate": 9.47913043478261e-05,
86
+ "loss": 5.1789,
87
+ "step": 1100
88
+ },
89
+ {
90
+ "epoch": 1.5645371577574967,
91
+ "grad_norm": 2.6344940662384033,
92
+ "learning_rate": 9.392173913043478e-05,
93
+ "loss": 5.0844,
94
+ "step": 1200
95
+ },
96
+ {
97
+ "epoch": 1.694915254237288,
98
+ "grad_norm": 3.4910781383514404,
99
+ "learning_rate": 9.305217391304349e-05,
100
+ "loss": 4.9929,
101
+ "step": 1300
102
+ },
103
+ {
104
+ "epoch": 1.8252933507170797,
105
+ "grad_norm": 3.5532472133636475,
106
+ "learning_rate": 9.218260869565217e-05,
107
+ "loss": 4.8998,
108
+ "step": 1400
109
+ },
110
+ {
111
+ "epoch": 1.9556714471968708,
112
+ "grad_norm": 3.246469497680664,
113
+ "learning_rate": 9.131304347826088e-05,
114
+ "loss": 4.7656,
115
+ "step": 1500
116
+ },
117
+ {
118
+ "epoch": 2.0860495436766624,
119
+ "grad_norm": 3.618093729019165,
120
+ "learning_rate": 9.044347826086958e-05,
121
+ "loss": 4.5793,
122
+ "step": 1600
123
+ },
124
+ {
125
+ "epoch": 2.2164276401564535,
126
+ "grad_norm": 4.264451026916504,
127
+ "learning_rate": 8.957391304347826e-05,
128
+ "loss": 4.4459,
129
+ "step": 1700
130
+ },
131
+ {
132
+ "epoch": 2.346805736636245,
133
+ "grad_norm": 4.198596477508545,
134
+ "learning_rate": 8.870434782608697e-05,
135
+ "loss": 4.3207,
136
+ "step": 1800
137
+ },
138
+ {
139
+ "epoch": 2.4771838331160367,
140
+ "grad_norm": 4.508902549743652,
141
+ "learning_rate": 8.783478260869565e-05,
142
+ "loss": 4.235,
143
+ "step": 1900
144
+ },
145
+ {
146
+ "epoch": 2.607561929595828,
147
+ "grad_norm": 4.511523723602295,
148
+ "learning_rate": 8.696521739130436e-05,
149
+ "loss": 4.1349,
150
+ "step": 2000
151
+ },
152
+ {
153
+ "epoch": 2.737940026075619,
154
+ "grad_norm": 3.903628349304199,
155
+ "learning_rate": 8.609565217391304e-05,
156
+ "loss": 4.0358,
157
+ "step": 2100
158
+ },
159
+ {
160
+ "epoch": 2.8683181225554106,
161
+ "grad_norm": 4.530259609222412,
162
+ "learning_rate": 8.523478260869565e-05,
163
+ "loss": 3.9168,
164
+ "step": 2200
165
+ },
166
+ {
167
+ "epoch": 2.9986962190352022,
168
+ "grad_norm": 4.071807861328125,
169
+ "learning_rate": 8.436521739130435e-05,
170
+ "loss": 3.8121,
171
+ "step": 2300
172
+ },
173
+ {
174
+ "epoch": 3.1290743155149934,
175
+ "grad_norm": 4.118851661682129,
176
+ "learning_rate": 8.349565217391306e-05,
177
+ "loss": 3.6089,
178
+ "step": 2400
179
+ },
180
+ {
181
+ "epoch": 3.259452411994785,
182
+ "grad_norm": 4.954625606536865,
183
+ "learning_rate": 8.262608695652174e-05,
184
+ "loss": 3.5162,
185
+ "step": 2500
186
+ },
187
+ {
188
+ "epoch": 3.389830508474576,
189
+ "grad_norm": 4.958930492401123,
190
+ "learning_rate": 8.175652173913044e-05,
191
+ "loss": 3.427,
192
+ "step": 2600
193
+ },
194
+ {
195
+ "epoch": 3.5202086049543677,
196
+ "grad_norm": 4.683836460113525,
197
+ "learning_rate": 8.088695652173913e-05,
198
+ "loss": 3.3498,
199
+ "step": 2700
200
+ },
201
+ {
202
+ "epoch": 3.6505867014341593,
203
+ "grad_norm": 5.2733259201049805,
204
+ "learning_rate": 8.001739130434783e-05,
205
+ "loss": 3.2661,
206
+ "step": 2800
207
+ },
208
+ {
209
+ "epoch": 3.7809647979139505,
210
+ "grad_norm": 5.038583278656006,
211
+ "learning_rate": 7.914782608695652e-05,
212
+ "loss": 3.1859,
213
+ "step": 2900
214
+ },
215
+ {
216
+ "epoch": 3.9113428943937416,
217
+ "grad_norm": 5.139358043670654,
218
+ "learning_rate": 7.827826086956522e-05,
219
+ "loss": 3.125,
220
+ "step": 3000
221
+ },
222
+ {
223
+ "epoch": 4.041720990873533,
224
+ "grad_norm": 5.076649188995361,
225
+ "learning_rate": 7.740869565217391e-05,
226
+ "loss": 2.9774,
227
+ "step": 3100
228
+ },
229
+ {
230
+ "epoch": 4.172099087353325,
231
+ "grad_norm": 7.2738165855407715,
232
+ "learning_rate": 7.653913043478261e-05,
233
+ "loss": 2.8133,
234
+ "step": 3200
235
+ },
236
+ {
237
+ "epoch": 4.302477183833116,
238
+ "grad_norm": 4.856014728546143,
239
+ "learning_rate": 7.56695652173913e-05,
240
+ "loss": 2.7578,
241
+ "step": 3300
242
+ },
243
+ {
244
+ "epoch": 4.432855280312907,
245
+ "grad_norm": 4.625192165374756,
246
+ "learning_rate": 7.48e-05,
247
+ "loss": 2.706,
248
+ "step": 3400
249
+ },
250
+ {
251
+ "epoch": 4.563233376792699,
252
+ "grad_norm": 5.898316860198975,
253
+ "learning_rate": 7.39304347826087e-05,
254
+ "loss": 2.6505,
255
+ "step": 3500
256
+ },
257
+ {
258
+ "epoch": 4.69361147327249,
259
+ "grad_norm": 5.6338324546813965,
260
+ "learning_rate": 7.30608695652174e-05,
261
+ "loss": 2.5716,
262
+ "step": 3600
263
+ },
264
+ {
265
+ "epoch": 4.823989569752282,
266
+ "grad_norm": 4.824554443359375,
267
+ "learning_rate": 7.219130434782609e-05,
268
+ "loss": 2.4845,
269
+ "step": 3700
270
+ },
271
+ {
272
+ "epoch": 4.9543676662320735,
273
+ "grad_norm": 6.3988037109375,
274
+ "learning_rate": 7.132173913043479e-05,
275
+ "loss": 2.4531,
276
+ "step": 3800
277
+ },
278
+ {
279
+ "epoch": 5.084745762711864,
280
+ "grad_norm": 4.477128982543945,
281
+ "learning_rate": 7.045217391304348e-05,
282
+ "loss": 2.3122,
283
+ "step": 3900
284
+ },
285
+ {
286
+ "epoch": 5.215123859191656,
287
+ "grad_norm": 5.06989049911499,
288
+ "learning_rate": 6.958260869565218e-05,
289
+ "loss": 2.2015,
290
+ "step": 4000
291
+ },
292
+ {
293
+ "epoch": 5.345501955671447,
294
+ "grad_norm": 4.576299667358398,
295
+ "learning_rate": 6.871304347826087e-05,
296
+ "loss": 2.1651,
297
+ "step": 4100
298
+ },
299
+ {
300
+ "epoch": 5.475880052151239,
301
+ "grad_norm": 4.288216590881348,
302
+ "learning_rate": 6.785217391304349e-05,
303
+ "loss": 2.1132,
304
+ "step": 4200
305
+ },
306
+ {
307
+ "epoch": 5.60625814863103,
308
+ "grad_norm": 6.912166118621826,
309
+ "learning_rate": 6.698260869565218e-05,
310
+ "loss": 2.0712,
311
+ "step": 4300
312
+ },
313
+ {
314
+ "epoch": 5.736636245110821,
315
+ "grad_norm": 5.21943998336792,
316
+ "learning_rate": 6.611304347826088e-05,
317
+ "loss": 2.0318,
318
+ "step": 4400
319
+ },
320
+ {
321
+ "epoch": 5.867014341590613,
322
+ "grad_norm": 6.814871311187744,
323
+ "learning_rate": 6.524347826086957e-05,
324
+ "loss": 1.996,
325
+ "step": 4500
326
+ },
327
+ {
328
+ "epoch": 5.9973924380704045,
329
+ "grad_norm": 6.987309455871582,
330
+ "learning_rate": 6.437391304347827e-05,
331
+ "loss": 1.9435,
332
+ "step": 4600
333
+ },
334
+ {
335
+ "epoch": 6.127770534550195,
336
+ "grad_norm": 6.552413463592529,
337
+ "learning_rate": 6.350434782608696e-05,
338
+ "loss": 1.7934,
339
+ "step": 4700
340
+ },
341
+ {
342
+ "epoch": 6.258148631029987,
343
+ "grad_norm": 4.565166473388672,
344
+ "learning_rate": 6.263478260869565e-05,
345
+ "loss": 1.7635,
346
+ "step": 4800
347
+ },
348
+ {
349
+ "epoch": 6.388526727509778,
350
+ "grad_norm": 6.433986186981201,
351
+ "learning_rate": 6.176521739130436e-05,
352
+ "loss": 1.7403,
353
+ "step": 4900
354
+ },
355
+ {
356
+ "epoch": 6.51890482398957,
357
+ "grad_norm": 4.71969747543335,
358
+ "learning_rate": 6.0895652173913044e-05,
359
+ "loss": 1.7153,
360
+ "step": 5000
361
+ },
362
+ {
363
+ "epoch": 6.6492829204693615,
364
+ "grad_norm": 5.415853023529053,
365
+ "learning_rate": 6.002608695652174e-05,
366
+ "loss": 1.6563,
367
+ "step": 5100
368
+ },
369
+ {
370
+ "epoch": 6.779661016949152,
371
+ "grad_norm": 4.742384910583496,
372
+ "learning_rate": 5.9156521739130436e-05,
373
+ "loss": 1.6431,
374
+ "step": 5200
375
+ },
376
+ {
377
+ "epoch": 6.910039113428944,
378
+ "grad_norm": 4.704698085784912,
379
+ "learning_rate": 5.828695652173913e-05,
380
+ "loss": 1.6071,
381
+ "step": 5300
382
+ },
383
+ {
384
+ "epoch": 7.040417209908735,
385
+ "grad_norm": 4.532726764678955,
386
+ "learning_rate": 5.7417391304347834e-05,
387
+ "loss": 1.561,
388
+ "step": 5400
389
+ },
390
+ {
391
+ "epoch": 7.170795306388527,
392
+ "grad_norm": 4.465181827545166,
393
+ "learning_rate": 5.654782608695652e-05,
394
+ "loss": 1.4572,
395
+ "step": 5500
396
+ },
397
+ {
398
+ "epoch": 7.301173402868318,
399
+ "grad_norm": 5.061060428619385,
400
+ "learning_rate": 5.568695652173913e-05,
401
+ "loss": 1.4313,
402
+ "step": 5600
403
+ },
404
+ {
405
+ "epoch": 7.431551499348109,
406
+ "grad_norm": 4.221780300140381,
407
+ "learning_rate": 5.481739130434783e-05,
408
+ "loss": 1.4099,
409
+ "step": 5700
410
+ },
411
+ {
412
+ "epoch": 7.561929595827901,
413
+ "grad_norm": 4.012993812561035,
414
+ "learning_rate": 5.394782608695652e-05,
415
+ "loss": 1.4044,
416
+ "step": 5800
417
+ },
418
+ {
419
+ "epoch": 7.6923076923076925,
420
+ "grad_norm": 5.935675621032715,
421
+ "learning_rate": 5.307826086956522e-05,
422
+ "loss": 1.3888,
423
+ "step": 5900
424
+ },
425
+ {
426
+ "epoch": 7.822685788787483,
427
+ "grad_norm": 5.962758541107178,
428
+ "learning_rate": 5.220869565217391e-05,
429
+ "loss": 1.3509,
430
+ "step": 6000
431
+ },
432
+ {
433
+ "epoch": 7.953063885267275,
434
+ "grad_norm": 5.295892238616943,
435
+ "learning_rate": 5.1339130434782615e-05,
436
+ "loss": 1.3376,
437
+ "step": 6100
438
+ },
439
+ {
440
+ "epoch": 8.083441981747066,
441
+ "grad_norm": 4.8437347412109375,
442
+ "learning_rate": 5.0469565217391304e-05,
443
+ "loss": 1.2682,
444
+ "step": 6200
445
+ },
446
+ {
447
+ "epoch": 8.213820078226858,
448
+ "grad_norm": 6.373332500457764,
449
+ "learning_rate": 4.96e-05,
450
+ "loss": 1.2201,
451
+ "step": 6300
452
+ },
453
+ {
454
+ "epoch": 8.34419817470665,
455
+ "grad_norm": 4.604928493499756,
456
+ "learning_rate": 4.8730434782608695e-05,
457
+ "loss": 1.2068,
458
+ "step": 6400
459
+ },
460
+ {
461
+ "epoch": 8.474576271186441,
462
+ "grad_norm": 6.433839797973633,
463
+ "learning_rate": 4.786086956521739e-05,
464
+ "loss": 1.197,
465
+ "step": 6500
466
+ },
467
+ {
468
+ "epoch": 8.604954367666233,
469
+ "grad_norm": 3.5552725791931152,
470
+ "learning_rate": 4.699130434782609e-05,
471
+ "loss": 1.1726,
472
+ "step": 6600
473
+ },
474
+ {
475
+ "epoch": 8.735332464146023,
476
+ "grad_norm": 8.83135986328125,
477
+ "learning_rate": 4.612173913043478e-05,
478
+ "loss": 1.1484,
479
+ "step": 6700
480
+ },
481
+ {
482
+ "epoch": 8.865710560625814,
483
+ "grad_norm": 4.828709602355957,
484
+ "learning_rate": 4.5252173913043485e-05,
485
+ "loss": 1.1583,
486
+ "step": 6800
487
+ },
488
+ {
489
+ "epoch": 8.996088657105606,
490
+ "grad_norm": 4.650803089141846,
491
+ "learning_rate": 4.438260869565218e-05,
492
+ "loss": 1.1481,
493
+ "step": 6900
494
+ },
495
+ {
496
+ "epoch": 9.126466753585397,
497
+ "grad_norm": 3.8928608894348145,
498
+ "learning_rate": 4.3513043478260876e-05,
499
+ "loss": 1.0397,
500
+ "step": 7000
501
+ },
502
+ {
503
+ "epoch": 9.256844850065189,
504
+ "grad_norm": 3.941554307937622,
505
+ "learning_rate": 4.2643478260869565e-05,
506
+ "loss": 1.0332,
507
+ "step": 7100
508
+ },
509
+ {
510
+ "epoch": 9.38722294654498,
511
+ "grad_norm": 5.031588077545166,
512
+ "learning_rate": 4.177391304347826e-05,
513
+ "loss": 1.0444,
514
+ "step": 7200
515
+ },
516
+ {
517
+ "epoch": 9.517601043024772,
518
+ "grad_norm": 5.682355880737305,
519
+ "learning_rate": 4.090434782608696e-05,
520
+ "loss": 1.0123,
521
+ "step": 7300
522
+ },
523
+ {
524
+ "epoch": 9.647979139504564,
525
+ "grad_norm": 4.108279705047607,
526
+ "learning_rate": 4.003478260869565e-05,
527
+ "loss": 1.0194,
528
+ "step": 7400
529
+ },
530
+ {
531
+ "epoch": 9.778357235984355,
532
+ "grad_norm": 3.3550074100494385,
533
+ "learning_rate": 3.916521739130435e-05,
534
+ "loss": 1.007,
535
+ "step": 7500
536
+ },
537
+ {
538
+ "epoch": 9.908735332464147,
539
+ "grad_norm": 4.076298713684082,
540
+ "learning_rate": 3.8295652173913044e-05,
541
+ "loss": 0.9973,
542
+ "step": 7600
543
+ },
544
+ {
545
+ "epoch": 10.039113428943937,
546
+ "grad_norm": 4.534595012664795,
547
+ "learning_rate": 3.742608695652174e-05,
548
+ "loss": 0.9646,
549
+ "step": 7700
550
+ },
551
+ {
552
+ "epoch": 10.169491525423728,
553
+ "grad_norm": 3.5908210277557373,
554
+ "learning_rate": 3.6556521739130435e-05,
555
+ "loss": 0.8984,
556
+ "step": 7800
557
+ },
558
+ {
559
+ "epoch": 10.29986962190352,
560
+ "grad_norm": 4.6488213539123535,
561
+ "learning_rate": 3.568695652173913e-05,
562
+ "loss": 0.8942,
563
+ "step": 7900
564
+ },
565
+ {
566
+ "epoch": 10.430247718383312,
567
+ "grad_norm": 4.129245758056641,
568
+ "learning_rate": 3.481739130434783e-05,
569
+ "loss": 0.8929,
570
+ "step": 8000
571
+ },
572
+ {
573
+ "epoch": 10.560625814863103,
574
+ "grad_norm": 3.645109176635742,
575
+ "learning_rate": 3.394782608695652e-05,
576
+ "loss": 0.8892,
577
+ "step": 8100
578
+ },
579
+ {
580
+ "epoch": 10.691003911342895,
581
+ "grad_norm": 4.7509565353393555,
582
+ "learning_rate": 3.307826086956522e-05,
583
+ "loss": 0.8937,
584
+ "step": 8200
585
+ },
586
+ {
587
+ "epoch": 10.821382007822686,
588
+ "grad_norm": 4.302902698516846,
589
+ "learning_rate": 3.2208695652173914e-05,
590
+ "loss": 0.8836,
591
+ "step": 8300
592
+ },
593
+ {
594
+ "epoch": 10.951760104302478,
595
+ "grad_norm": 4.266739845275879,
596
+ "learning_rate": 3.133913043478261e-05,
597
+ "loss": 0.8875,
598
+ "step": 8400
599
+ },
600
+ {
601
+ "epoch": 11.082138200782268,
602
+ "grad_norm": 3.6782920360565186,
603
+ "learning_rate": 3.0469565217391305e-05,
604
+ "loss": 0.8411,
605
+ "step": 8500
606
+ },
607
+ {
608
+ "epoch": 11.21251629726206,
609
+ "grad_norm": 3.9318060874938965,
610
+ "learning_rate": 2.96e-05,
611
+ "loss": 0.8063,
612
+ "step": 8600
613
+ },
614
+ {
615
+ "epoch": 11.342894393741851,
616
+ "grad_norm": 4.048793315887451,
617
+ "learning_rate": 2.8730434782608694e-05,
618
+ "loss": 0.7968,
619
+ "step": 8700
620
+ },
621
+ {
622
+ "epoch": 11.473272490221643,
623
+ "grad_norm": 4.4102888107299805,
624
+ "learning_rate": 2.786086956521739e-05,
625
+ "loss": 0.7983,
626
+ "step": 8800
627
+ },
628
+ {
629
+ "epoch": 11.603650586701434,
630
+ "grad_norm": 4.744139671325684,
631
+ "learning_rate": 2.6991304347826092e-05,
632
+ "loss": 0.7903,
633
+ "step": 8900
634
+ },
635
+ {
636
+ "epoch": 11.734028683181226,
637
+ "grad_norm": 4.147999286651611,
638
+ "learning_rate": 2.6121739130434788e-05,
639
+ "loss": 0.7997,
640
+ "step": 9000
641
+ }
642
+ ],
643
+ "logging_steps": 100,
644
+ "max_steps": 12000,
645
+ "num_input_tokens_seen": 0,
646
+ "num_train_epochs": 16,
647
+ "save_steps": 3000,
648
+ "stateful_callbacks": {
649
+ "TrainerControl": {
650
+ "args": {
651
+ "should_epoch_stop": false,
652
+ "should_evaluate": false,
653
+ "should_log": false,
654
+ "should_save": true,
655
+ "should_training_stop": false
656
+ },
657
+ "attributes": {}
658
+ }
659
+ },
660
+ "total_flos": 4.7976884011008e+18,
661
+ "train_batch_size": 32,
662
+ "trial_name": null,
663
+ "trial_params": null
664
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50472477cc62d84b2d93e339afda70651261ba26b46389224cbac1258329a178
3
+ size 5969
vocab.json ADDED
The diff for this file is too large to render. See raw diff