fpadovani commited on
Commit
e7731ce
·
verified ·
1 Parent(s): eda8c1c

Training in progress, step 30000, checkpoint

Browse files
checkpoint-30000/config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "gpt2",
3
+ "activation_function": "gelu_new",
4
+ "architectures": [
5
+ "GPT2LMHeadModel"
6
+ ],
7
+ "attn_pdrop": 0.1,
8
+ "bos_token_id": 5,
9
+ "embd_pdrop": 0.1,
10
+ "eos_token_id": 4,
11
+ "initializer_range": 0.02,
12
+ "layer_norm_epsilon": 1e-05,
13
+ "model_type": "gpt2",
14
+ "n_ctx": 128,
15
+ "n_embd": 128,
16
+ "n_head": 4,
17
+ "n_inner": null,
18
+ "n_layer": 4,
19
+ "n_positions": 128,
20
+ "reorder_and_upcast_attn": false,
21
+ "resid_pdrop": 0.1,
22
+ "scale_attn_by_inverse_layer_idx": false,
23
+ "scale_attn_weights": true,
24
+ "summary_activation": null,
25
+ "summary_first_dropout": 0.1,
26
+ "summary_proj_to_labels": true,
27
+ "summary_type": "cls_index",
28
+ "summary_use_proj": true,
29
+ "task_specific_params": {
30
+ "text-generation": {
31
+ "do_sample": true,
32
+ "max_length": 50
33
+ }
34
+ },
35
+ "torch_dtype": "float32",
36
+ "transformers_version": "4.45.2",
37
+ "use_cache": true,
38
+ "vocab_size": 8192
39
+ }
checkpoint-30000/generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 5,
4
+ "eos_token_id": 4,
5
+ "transformers_version": "4.45.2"
6
+ }
checkpoint-30000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:222de254ed8812ccd67278c3847ea7685a7f4f73dedbb8922eb61288824d4a59
3
+ size 7438256
checkpoint-30000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ae33543dfbea80e491fff8279c8b57ba5459cb53f50ca68faeeb9ec8d1465be
3
+ size 14910010
checkpoint-30000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e402c00cc3aa268668eeb0ef2dd968e6761b3333c64a1a93c3cfc83ff8cc3a3
3
+ size 14244
checkpoint-30000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:af0bc228fa60ab22ed4a022a196eac60f0a4f53d1ad88d9174e57a7e22a79674
3
+ size 1192
checkpoint-30000/special_tokens_map.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "[BOS]",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "[EOS]",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": "[EOS]"
17
+ }
checkpoint-30000/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-30000/tokenizer_config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[UNK]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "[CLS]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "[SEP]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "[PAD]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "4": {
36
+ "content": "[EOS]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "5": {
44
+ "content": "[BOS]",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ }
51
+ },
52
+ "bos_token": "[BOS]",
53
+ "clean_up_tokenization_spaces": true,
54
+ "eos_token": "[EOS]",
55
+ "model_max_length": 1000000000000000019884624838656,
56
+ "pad_token": "[EOS]",
57
+ "tokenizer_class": "PreTrainedTokenizerFast"
58
+ }
checkpoint-30000/trainer_state.json ADDED
@@ -0,0 +1,702 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 3.2750871181488037,
3
+ "best_model_checkpoint": "/home/p318482/babyLM_controlled/models_trained/reduce_on_plateau_convergence/age_transcript_conv1/checkpoint-29000",
4
+ "epoch": 0.7618273699179766,
5
+ "eval_steps": 1000,
6
+ "global_step": 30000,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.012697122831966277,
13
+ "grad_norm": 1.3349946737289429,
14
+ "learning_rate": 0.0001,
15
+ "loss": 5.9732,
16
+ "step": 500
17
+ },
18
+ {
19
+ "epoch": 0.025394245663932554,
20
+ "grad_norm": 0.8543657064437866,
21
+ "learning_rate": 0.0001,
22
+ "loss": 4.7007,
23
+ "step": 1000
24
+ },
25
+ {
26
+ "epoch": 0.025394245663932554,
27
+ "eval_loss": 4.492173671722412,
28
+ "eval_runtime": 1.9045,
29
+ "eval_samples_per_second": 918.89,
30
+ "eval_steps_per_second": 28.879,
31
+ "step": 1000
32
+ },
33
+ {
34
+ "epoch": 0.03809136849589883,
35
+ "grad_norm": 0.9668338298797607,
36
+ "learning_rate": 0.0001,
37
+ "loss": 4.4512,
38
+ "step": 1500
39
+ },
40
+ {
41
+ "epoch": 0.05078849132786511,
42
+ "grad_norm": 1.5050448179244995,
43
+ "learning_rate": 0.0001,
44
+ "loss": 4.2819,
45
+ "step": 2000
46
+ },
47
+ {
48
+ "epoch": 0.05078849132786511,
49
+ "eval_loss": 4.212340831756592,
50
+ "eval_runtime": 1.9267,
51
+ "eval_samples_per_second": 908.277,
52
+ "eval_steps_per_second": 28.546,
53
+ "step": 2000
54
+ },
55
+ {
56
+ "epoch": 0.06348561415983138,
57
+ "grad_norm": 1.4784210920333862,
58
+ "learning_rate": 0.0001,
59
+ "loss": 4.2117,
60
+ "step": 2500
61
+ },
62
+ {
63
+ "epoch": 0.07618273699179766,
64
+ "grad_norm": 1.1260796785354614,
65
+ "learning_rate": 0.0001,
66
+ "loss": 4.1092,
67
+ "step": 3000
68
+ },
69
+ {
70
+ "epoch": 0.07618273699179766,
71
+ "eval_loss": 4.032414436340332,
72
+ "eval_runtime": 1.9167,
73
+ "eval_samples_per_second": 913.005,
74
+ "eval_steps_per_second": 28.694,
75
+ "step": 3000
76
+ },
77
+ {
78
+ "epoch": 0.08887985982376394,
79
+ "grad_norm": 1.8558170795440674,
80
+ "learning_rate": 0.0001,
81
+ "loss": 4.0612,
82
+ "step": 3500
83
+ },
84
+ {
85
+ "epoch": 0.10157698265573022,
86
+ "grad_norm": 1.994602918624878,
87
+ "learning_rate": 0.0001,
88
+ "loss": 3.9861,
89
+ "step": 4000
90
+ },
91
+ {
92
+ "epoch": 0.10157698265573022,
93
+ "eval_loss": 3.9267466068267822,
94
+ "eval_runtime": 1.8908,
95
+ "eval_samples_per_second": 925.512,
96
+ "eval_steps_per_second": 29.088,
97
+ "step": 4000
98
+ },
99
+ {
100
+ "epoch": 0.11427410548769648,
101
+ "grad_norm": 1.322039008140564,
102
+ "learning_rate": 0.0001,
103
+ "loss": 3.9785,
104
+ "step": 4500
105
+ },
106
+ {
107
+ "epoch": 0.12697122831966276,
108
+ "grad_norm": 1.4683969020843506,
109
+ "learning_rate": 0.0001,
110
+ "loss": 3.9221,
111
+ "step": 5000
112
+ },
113
+ {
114
+ "epoch": 0.12697122831966276,
115
+ "eval_loss": 3.858163595199585,
116
+ "eval_runtime": 1.9116,
117
+ "eval_samples_per_second": 915.452,
118
+ "eval_steps_per_second": 28.771,
119
+ "step": 5000
120
+ },
121
+ {
122
+ "epoch": 0.13966835115162904,
123
+ "grad_norm": 1.9317216873168945,
124
+ "learning_rate": 0.0001,
125
+ "loss": 3.8523,
126
+ "step": 5500
127
+ },
128
+ {
129
+ "epoch": 0.15236547398359532,
130
+ "grad_norm": 1.6324900388717651,
131
+ "learning_rate": 0.0001,
132
+ "loss": 3.8904,
133
+ "step": 6000
134
+ },
135
+ {
136
+ "epoch": 0.15236547398359532,
137
+ "eval_loss": 3.780858039855957,
138
+ "eval_runtime": 1.895,
139
+ "eval_samples_per_second": 923.489,
140
+ "eval_steps_per_second": 29.024,
141
+ "step": 6000
142
+ },
143
+ {
144
+ "epoch": 0.1650625968155616,
145
+ "grad_norm": 1.8691177368164062,
146
+ "learning_rate": 0.0001,
147
+ "loss": 3.8174,
148
+ "step": 6500
149
+ },
150
+ {
151
+ "epoch": 0.17775971964752788,
152
+ "grad_norm": 2.08821964263916,
153
+ "learning_rate": 0.0001,
154
+ "loss": 3.7526,
155
+ "step": 7000
156
+ },
157
+ {
158
+ "epoch": 0.17775971964752788,
159
+ "eval_loss": 3.7251765727996826,
160
+ "eval_runtime": 1.9041,
161
+ "eval_samples_per_second": 919.09,
162
+ "eval_steps_per_second": 28.886,
163
+ "step": 7000
164
+ },
165
+ {
166
+ "epoch": 0.19045684247949415,
167
+ "grad_norm": 1.4778984785079956,
168
+ "learning_rate": 0.0001,
169
+ "loss": 3.7759,
170
+ "step": 7500
171
+ },
172
+ {
173
+ "epoch": 0.20315396531146043,
174
+ "grad_norm": 1.7713682651519775,
175
+ "learning_rate": 0.0001,
176
+ "loss": 3.7724,
177
+ "step": 8000
178
+ },
179
+ {
180
+ "epoch": 0.20315396531146043,
181
+ "eval_loss": 3.684647798538208,
182
+ "eval_runtime": 1.917,
183
+ "eval_samples_per_second": 912.872,
184
+ "eval_steps_per_second": 28.69,
185
+ "step": 8000
186
+ },
187
+ {
188
+ "epoch": 0.2158510881434267,
189
+ "grad_norm": 1.6588795185089111,
190
+ "learning_rate": 0.0001,
191
+ "loss": 3.6645,
192
+ "step": 8500
193
+ },
194
+ {
195
+ "epoch": 0.22854821097539296,
196
+ "grad_norm": 1.9190706014633179,
197
+ "learning_rate": 0.0001,
198
+ "loss": 3.6967,
199
+ "step": 9000
200
+ },
201
+ {
202
+ "epoch": 0.22854821097539296,
203
+ "eval_loss": 3.6292779445648193,
204
+ "eval_runtime": 1.9099,
205
+ "eval_samples_per_second": 916.302,
206
+ "eval_steps_per_second": 28.798,
207
+ "step": 9000
208
+ },
209
+ {
210
+ "epoch": 0.24124533380735924,
211
+ "grad_norm": 1.7939610481262207,
212
+ "learning_rate": 0.0001,
213
+ "loss": 3.7286,
214
+ "step": 9500
215
+ },
216
+ {
217
+ "epoch": 0.2539424566393255,
218
+ "grad_norm": 1.8331050872802734,
219
+ "learning_rate": 0.0001,
220
+ "loss": 3.5701,
221
+ "step": 10000
222
+ },
223
+ {
224
+ "epoch": 0.2539424566393255,
225
+ "eval_loss": 3.590202569961548,
226
+ "eval_runtime": 1.8859,
227
+ "eval_samples_per_second": 927.93,
228
+ "eval_steps_per_second": 29.163,
229
+ "step": 10000
230
+ },
231
+ {
232
+ "epoch": 0.2666395794712918,
233
+ "grad_norm": 1.7435561418533325,
234
+ "learning_rate": 0.0001,
235
+ "loss": 3.654,
236
+ "step": 10500
237
+ },
238
+ {
239
+ "epoch": 0.2793367023032581,
240
+ "grad_norm": 1.7139521837234497,
241
+ "learning_rate": 0.0001,
242
+ "loss": 3.676,
243
+ "step": 11000
244
+ },
245
+ {
246
+ "epoch": 0.2793367023032581,
247
+ "eval_loss": 3.578721046447754,
248
+ "eval_runtime": 1.9307,
249
+ "eval_samples_per_second": 906.398,
250
+ "eval_steps_per_second": 28.487,
251
+ "step": 11000
252
+ },
253
+ {
254
+ "epoch": 0.29203382513522436,
255
+ "grad_norm": 1.8859118223190308,
256
+ "learning_rate": 0.0001,
257
+ "loss": 3.5154,
258
+ "step": 11500
259
+ },
260
+ {
261
+ "epoch": 0.30473094796719064,
262
+ "grad_norm": 1.6960543394088745,
263
+ "learning_rate": 0.0001,
264
+ "loss": 3.6092,
265
+ "step": 12000
266
+ },
267
+ {
268
+ "epoch": 0.30473094796719064,
269
+ "eval_loss": 3.5332963466644287,
270
+ "eval_runtime": 1.8853,
271
+ "eval_samples_per_second": 928.227,
272
+ "eval_steps_per_second": 29.173,
273
+ "step": 12000
274
+ },
275
+ {
276
+ "epoch": 0.3174280707991569,
277
+ "grad_norm": 1.5386900901794434,
278
+ "learning_rate": 0.0001,
279
+ "loss": 3.6035,
280
+ "step": 12500
281
+ },
282
+ {
283
+ "epoch": 0.3301251936311232,
284
+ "grad_norm": 1.6977068185806274,
285
+ "learning_rate": 0.0001,
286
+ "loss": 3.5105,
287
+ "step": 13000
288
+ },
289
+ {
290
+ "epoch": 0.3301251936311232,
291
+ "eval_loss": 3.5060741901397705,
292
+ "eval_runtime": 1.8975,
293
+ "eval_samples_per_second": 922.266,
294
+ "eval_steps_per_second": 28.985,
295
+ "step": 13000
296
+ },
297
+ {
298
+ "epoch": 0.34282231646308947,
299
+ "grad_norm": 1.6782351732254028,
300
+ "learning_rate": 0.0001,
301
+ "loss": 3.5569,
302
+ "step": 13500
303
+ },
304
+ {
305
+ "epoch": 0.35551943929505575,
306
+ "grad_norm": 1.5853595733642578,
307
+ "learning_rate": 0.0001,
308
+ "loss": 3.5298,
309
+ "step": 14000
310
+ },
311
+ {
312
+ "epoch": 0.35551943929505575,
313
+ "eval_loss": 3.47761607170105,
314
+ "eval_runtime": 1.9587,
315
+ "eval_samples_per_second": 893.455,
316
+ "eval_steps_per_second": 28.08,
317
+ "step": 14000
318
+ },
319
+ {
320
+ "epoch": 0.36821656212702203,
321
+ "grad_norm": 1.7518577575683594,
322
+ "learning_rate": 0.0001,
323
+ "loss": 3.4899,
324
+ "step": 14500
325
+ },
326
+ {
327
+ "epoch": 0.3809136849589883,
328
+ "grad_norm": 1.598228096961975,
329
+ "learning_rate": 0.0001,
330
+ "loss": 3.4857,
331
+ "step": 15000
332
+ },
333
+ {
334
+ "epoch": 0.3809136849589883,
335
+ "eval_loss": 3.4536595344543457,
336
+ "eval_runtime": 1.907,
337
+ "eval_samples_per_second": 917.685,
338
+ "eval_steps_per_second": 28.842,
339
+ "step": 15000
340
+ },
341
+ {
342
+ "epoch": 0.3936108077909546,
343
+ "grad_norm": 1.726940631866455,
344
+ "learning_rate": 0.0001,
345
+ "loss": 3.5381,
346
+ "step": 15500
347
+ },
348
+ {
349
+ "epoch": 0.40630793062292087,
350
+ "grad_norm": 1.8652628660202026,
351
+ "learning_rate": 0.0001,
352
+ "loss": 3.4688,
353
+ "step": 16000
354
+ },
355
+ {
356
+ "epoch": 0.40630793062292087,
357
+ "eval_loss": 3.4489903450012207,
358
+ "eval_runtime": 1.8851,
359
+ "eval_samples_per_second": 928.332,
360
+ "eval_steps_per_second": 29.176,
361
+ "step": 16000
362
+ },
363
+ {
364
+ "epoch": 0.41900505345488714,
365
+ "grad_norm": 1.738990068435669,
366
+ "learning_rate": 0.0001,
367
+ "loss": 3.4371,
368
+ "step": 16500
369
+ },
370
+ {
371
+ "epoch": 0.4317021762868534,
372
+ "grad_norm": 1.6831852197647095,
373
+ "learning_rate": 0.0001,
374
+ "loss": 3.4914,
375
+ "step": 17000
376
+ },
377
+ {
378
+ "epoch": 0.4317021762868534,
379
+ "eval_loss": 3.414088726043701,
380
+ "eval_runtime": 1.8985,
381
+ "eval_samples_per_second": 921.778,
382
+ "eval_steps_per_second": 28.97,
383
+ "step": 17000
384
+ },
385
+ {
386
+ "epoch": 0.4443992991188197,
387
+ "grad_norm": 1.9883878231048584,
388
+ "learning_rate": 0.0001,
389
+ "loss": 3.4688,
390
+ "step": 17500
391
+ },
392
+ {
393
+ "epoch": 0.4570964219507859,
394
+ "grad_norm": 1.8864822387695312,
395
+ "learning_rate": 0.0001,
396
+ "loss": 3.3866,
397
+ "step": 18000
398
+ },
399
+ {
400
+ "epoch": 0.4570964219507859,
401
+ "eval_loss": 3.396988868713379,
402
+ "eval_runtime": 1.9075,
403
+ "eval_samples_per_second": 917.452,
404
+ "eval_steps_per_second": 28.834,
405
+ "step": 18000
406
+ },
407
+ {
408
+ "epoch": 0.4697935447827522,
409
+ "grad_norm": 1.8801751136779785,
410
+ "learning_rate": 0.0001,
411
+ "loss": 3.4386,
412
+ "step": 18500
413
+ },
414
+ {
415
+ "epoch": 0.4824906676147185,
416
+ "grad_norm": 2.172294855117798,
417
+ "learning_rate": 0.0001,
418
+ "loss": 3.484,
419
+ "step": 19000
420
+ },
421
+ {
422
+ "epoch": 0.4824906676147185,
423
+ "eval_loss": 3.3963444232940674,
424
+ "eval_runtime": 1.9167,
425
+ "eval_samples_per_second": 913.043,
426
+ "eval_steps_per_second": 28.696,
427
+ "step": 19000
428
+ },
429
+ {
430
+ "epoch": 0.49518779044668476,
431
+ "grad_norm": 2.3698723316192627,
432
+ "learning_rate": 0.0001,
433
+ "loss": 3.325,
434
+ "step": 19500
435
+ },
436
+ {
437
+ "epoch": 0.507884913278651,
438
+ "grad_norm": 1.6967418193817139,
439
+ "learning_rate": 0.0001,
440
+ "loss": 3.4187,
441
+ "step": 20000
442
+ },
443
+ {
444
+ "epoch": 0.507884913278651,
445
+ "eval_loss": 3.373260021209717,
446
+ "eval_runtime": 1.8987,
447
+ "eval_samples_per_second": 921.684,
448
+ "eval_steps_per_second": 28.967,
449
+ "step": 20000
450
+ },
451
+ {
452
+ "epoch": 0.5205820361106174,
453
+ "grad_norm": 1.7509515285491943,
454
+ "learning_rate": 0.0001,
455
+ "loss": 3.4752,
456
+ "step": 20500
457
+ },
458
+ {
459
+ "epoch": 0.5332791589425836,
460
+ "grad_norm": 1.777363896369934,
461
+ "learning_rate": 0.0001,
462
+ "loss": 3.2706,
463
+ "step": 21000
464
+ },
465
+ {
466
+ "epoch": 0.5332791589425836,
467
+ "eval_loss": 3.3545970916748047,
468
+ "eval_runtime": 1.9049,
469
+ "eval_samples_per_second": 918.682,
470
+ "eval_steps_per_second": 28.873,
471
+ "step": 21000
472
+ },
473
+ {
474
+ "epoch": 0.5459762817745499,
475
+ "grad_norm": 1.583158254623413,
476
+ "learning_rate": 0.0001,
477
+ "loss": 3.4098,
478
+ "step": 21500
479
+ },
480
+ {
481
+ "epoch": 0.5586734046065162,
482
+ "grad_norm": 1.7274887561798096,
483
+ "learning_rate": 0.0001,
484
+ "loss": 3.4344,
485
+ "step": 22000
486
+ },
487
+ {
488
+ "epoch": 0.5586734046065162,
489
+ "eval_loss": 3.363961935043335,
490
+ "eval_runtime": 1.9181,
491
+ "eval_samples_per_second": 912.358,
492
+ "eval_steps_per_second": 28.674,
493
+ "step": 22000
494
+ },
495
+ {
496
+ "epoch": 0.5713705274384825,
497
+ "grad_norm": 1.8767764568328857,
498
+ "learning_rate": 0.0001,
499
+ "loss": 3.2792,
500
+ "step": 22500
501
+ },
502
+ {
503
+ "epoch": 0.5840676502704487,
504
+ "grad_norm": 2.2327566146850586,
505
+ "learning_rate": 0.0001,
506
+ "loss": 3.3577,
507
+ "step": 23000
508
+ },
509
+ {
510
+ "epoch": 0.5840676502704487,
511
+ "eval_loss": 3.333698034286499,
512
+ "eval_runtime": 1.8926,
513
+ "eval_samples_per_second": 924.643,
514
+ "eval_steps_per_second": 29.06,
515
+ "step": 23000
516
+ },
517
+ {
518
+ "epoch": 0.596764773102415,
519
+ "grad_norm": 1.7552568912506104,
520
+ "learning_rate": 0.0001,
521
+ "loss": 3.4057,
522
+ "step": 23500
523
+ },
524
+ {
525
+ "epoch": 0.6094618959343813,
526
+ "grad_norm": 1.8450969457626343,
527
+ "learning_rate": 0.0001,
528
+ "loss": 3.3058,
529
+ "step": 24000
530
+ },
531
+ {
532
+ "epoch": 0.6094618959343813,
533
+ "eval_loss": 3.3363912105560303,
534
+ "eval_runtime": 1.9075,
535
+ "eval_samples_per_second": 917.453,
536
+ "eval_steps_per_second": 28.834,
537
+ "step": 24000
538
+ },
539
+ {
540
+ "epoch": 0.6221590187663475,
541
+ "grad_norm": 1.8642781972885132,
542
+ "learning_rate": 0.0001,
543
+ "loss": 3.3239,
544
+ "step": 24500
545
+ },
546
+ {
547
+ "epoch": 0.6348561415983138,
548
+ "grad_norm": 1.5582977533340454,
549
+ "learning_rate": 0.0001,
550
+ "loss": 3.3558,
551
+ "step": 25000
552
+ },
553
+ {
554
+ "epoch": 0.6348561415983138,
555
+ "eval_loss": 3.3194997310638428,
556
+ "eval_runtime": 1.9141,
557
+ "eval_samples_per_second": 914.259,
558
+ "eval_steps_per_second": 28.734,
559
+ "step": 25000
560
+ },
561
+ {
562
+ "epoch": 0.64755326443028,
563
+ "grad_norm": 2.183225631713867,
564
+ "learning_rate": 0.0001,
565
+ "loss": 3.3304,
566
+ "step": 25500
567
+ },
568
+ {
569
+ "epoch": 0.6602503872622464,
570
+ "grad_norm": 1.7446324825286865,
571
+ "learning_rate": 0.0001,
572
+ "loss": 3.2865,
573
+ "step": 26000
574
+ },
575
+ {
576
+ "epoch": 0.6602503872622464,
577
+ "eval_loss": 3.2987701892852783,
578
+ "eval_runtime": 1.9018,
579
+ "eval_samples_per_second": 920.197,
580
+ "eval_steps_per_second": 28.92,
581
+ "step": 26000
582
+ },
583
+ {
584
+ "epoch": 0.6729475100942126,
585
+ "grad_norm": 1.7698390483856201,
586
+ "learning_rate": 0.0001,
587
+ "loss": 3.3445,
588
+ "step": 26500
589
+ },
590
+ {
591
+ "epoch": 0.6856446329261789,
592
+ "grad_norm": 2.1546449661254883,
593
+ "learning_rate": 0.0001,
594
+ "loss": 3.3295,
595
+ "step": 27000
596
+ },
597
+ {
598
+ "epoch": 0.6856446329261789,
599
+ "eval_loss": 3.305447816848755,
600
+ "eval_runtime": 1.9096,
601
+ "eval_samples_per_second": 916.405,
602
+ "eval_steps_per_second": 28.801,
603
+ "step": 27000
604
+ },
605
+ {
606
+ "epoch": 0.6983417557581452,
607
+ "grad_norm": 1.7213537693023682,
608
+ "learning_rate": 0.0001,
609
+ "loss": 3.2552,
610
+ "step": 27500
611
+ },
612
+ {
613
+ "epoch": 0.7110388785901115,
614
+ "grad_norm": 1.981979250907898,
615
+ "learning_rate": 0.0001,
616
+ "loss": 3.3024,
617
+ "step": 28000
618
+ },
619
+ {
620
+ "epoch": 0.7110388785901115,
621
+ "eval_loss": 3.28672456741333,
622
+ "eval_runtime": 1.9283,
623
+ "eval_samples_per_second": 907.529,
624
+ "eval_steps_per_second": 28.522,
625
+ "step": 28000
626
+ },
627
+ {
628
+ "epoch": 0.7237360014220777,
629
+ "grad_norm": 1.8241162300109863,
630
+ "learning_rate": 0.0001,
631
+ "loss": 3.3452,
632
+ "step": 28500
633
+ },
634
+ {
635
+ "epoch": 0.7364331242540441,
636
+ "grad_norm": 1.6229711771011353,
637
+ "learning_rate": 0.0001,
638
+ "loss": 3.1984,
639
+ "step": 29000
640
+ },
641
+ {
642
+ "epoch": 0.7364331242540441,
643
+ "eval_loss": 3.2750871181488037,
644
+ "eval_runtime": 1.9233,
645
+ "eval_samples_per_second": 909.914,
646
+ "eval_steps_per_second": 28.597,
647
+ "step": 29000
648
+ },
649
+ {
650
+ "epoch": 0.7491302470860103,
651
+ "grad_norm": 1.9197314977645874,
652
+ "learning_rate": 0.0001,
653
+ "loss": 3.3077,
654
+ "step": 29500
655
+ },
656
+ {
657
+ "epoch": 0.7618273699179766,
658
+ "grad_norm": 2.0677168369293213,
659
+ "learning_rate": 0.0001,
660
+ "loss": 3.3467,
661
+ "step": 30000
662
+ },
663
+ {
664
+ "epoch": 0.7618273699179766,
665
+ "eval_loss": 3.279165267944336,
666
+ "eval_runtime": 1.9216,
667
+ "eval_samples_per_second": 910.709,
668
+ "eval_steps_per_second": 28.622,
669
+ "step": 30000
670
+ }
671
+ ],
672
+ "logging_steps": 500,
673
+ "max_steps": 39379,
674
+ "num_input_tokens_seen": 0,
675
+ "num_train_epochs": 1,
676
+ "save_steps": 1000,
677
+ "stateful_callbacks": {
678
+ "EarlyStoppingCallback": {
679
+ "args": {
680
+ "early_stopping_patience": 3,
681
+ "early_stopping_threshold": 0.001
682
+ },
683
+ "attributes": {
684
+ "early_stopping_patience_counter": 1
685
+ }
686
+ },
687
+ "TrainerControl": {
688
+ "args": {
689
+ "should_epoch_stop": false,
690
+ "should_evaluate": false,
691
+ "should_log": false,
692
+ "should_save": true,
693
+ "should_training_stop": false
694
+ },
695
+ "attributes": {}
696
+ }
697
+ },
698
+ "total_flos": 584916664320000.0,
699
+ "train_batch_size": 32,
700
+ "trial_name": null,
701
+ "trial_params": null
702
+ }
checkpoint-30000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2eb0eb50cb3f22510b6b7cd611b9597a3233ab2f8c14c8c592b0be912fd4228e
3
+ size 5432