mikhail-panzo commited on
Commit
a39a1c0
1 Parent(s): 677f8a0

Training in progress, step 4000, checkpoint

Browse files
checkpoint-4000/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c78cb19314ba41236d30680320db29050da6646ac0e2823349ec74ebe569a3b7
3
  size 577789320
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbbcce497ef06f90320ebed4b498fbb96448b523d20b75988ad244c6889d0008
3
  size 577789320
checkpoint-4000/optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8491f751df97d435d985acbaf79117e8cad36b8dc384c6d2738db7224a2d27bf
3
  size 1155772233
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:92ff288f4e644ba764b1e18b666604dd6290a820a0c276a3efd7e7b9756431e7
3
  size 1155772233
checkpoint-4000/rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f2c3c83504595130a5e8f93508e5c1913ab34b307d61060fdfb487c4a7bad148
3
  size 14244
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9713ed540d72f5d0a8e9b24860426ea48f9708bb17413984b51517b05967645e
3
  size 14244
checkpoint-4000/scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0bd01f1a543bd9ecb58737837f10f865ac053c6c89fa1a7cdbd04da0c2caf558
3
  size 1064
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab2be22f42104529ffd5988129734d57784719df2fd001493474c00c55fafb29
3
  size 1064
checkpoint-4000/trainer_state.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
- "best_metric": 0.41223829984664917,
3
- "best_model_checkpoint": "mikhail-panzo/zlm-fil-ceb_b64_le5_s8000/checkpoint-1500",
4
- "epoch": 173.91304347826087,
5
  "eval_steps": 500,
6
  "global_step": 4000,
7
  "is_hyper_param_search": false,
@@ -9,634 +9,634 @@
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
- "epoch": 2.1739130434782608,
13
- "grad_norm": 0.7357823252677917,
14
  "learning_rate": 2.5000000000000004e-07,
15
- "loss": 0.4055,
16
  "step": 50
17
  },
18
  {
19
- "epoch": 4.3478260869565215,
20
- "grad_norm": 0.7944665551185608,
21
  "learning_rate": 5.000000000000001e-07,
22
- "loss": 0.3988,
23
  "step": 100
24
  },
25
  {
26
- "epoch": 6.521739130434782,
27
- "grad_norm": 0.7226569652557373,
28
  "learning_rate": 7.5e-07,
29
- "loss": 0.4072,
30
  "step": 150
31
  },
32
  {
33
- "epoch": 8.695652173913043,
34
- "grad_norm": 0.7845238447189331,
35
  "learning_rate": 1.0000000000000002e-06,
36
- "loss": 0.4028,
37
  "step": 200
38
  },
39
  {
40
- "epoch": 10.869565217391305,
41
- "grad_norm": 0.9769566655158997,
42
  "learning_rate": 1.25e-06,
43
- "loss": 0.4066,
44
  "step": 250
45
  },
46
  {
47
- "epoch": 13.043478260869565,
48
- "grad_norm": 0.8634074330329895,
49
  "learning_rate": 1.5e-06,
50
- "loss": 0.405,
51
  "step": 300
52
  },
53
  {
54
- "epoch": 15.217391304347826,
55
- "grad_norm": 0.9029327034950256,
56
  "learning_rate": 1.75e-06,
57
- "loss": 0.4074,
58
  "step": 350
59
  },
60
  {
61
- "epoch": 17.391304347826086,
62
- "grad_norm": 0.7624074220657349,
63
  "learning_rate": 2.0000000000000003e-06,
64
- "loss": 0.406,
65
  "step": 400
66
  },
67
  {
68
- "epoch": 19.565217391304348,
69
- "grad_norm": 1.005196213722229,
70
  "learning_rate": 2.25e-06,
71
- "loss": 0.4052,
72
  "step": 450
73
  },
74
  {
75
- "epoch": 21.73913043478261,
76
- "grad_norm": 1.0276380777359009,
77
  "learning_rate": 2.5e-06,
78
- "loss": 0.4019,
79
  "step": 500
80
  },
81
  {
82
- "epoch": 21.73913043478261,
83
- "eval_loss": 0.4143177568912506,
84
- "eval_runtime": 6.4074,
85
- "eval_samples_per_second": 24.815,
86
- "eval_steps_per_second": 3.121,
87
  "step": 500
88
  },
89
  {
90
- "epoch": 23.91304347826087,
91
- "grad_norm": 0.9067140817642212,
92
  "learning_rate": 2.7500000000000004e-06,
93
- "loss": 0.4021,
94
  "step": 550
95
  },
96
  {
97
- "epoch": 26.08695652173913,
98
- "grad_norm": 0.8799042701721191,
99
  "learning_rate": 3e-06,
100
- "loss": 0.4071,
101
  "step": 600
102
  },
103
  {
104
- "epoch": 28.26086956521739,
105
- "grad_norm": 1.0451533794403076,
106
  "learning_rate": 3.2500000000000002e-06,
107
- "loss": 0.4146,
108
  "step": 650
109
  },
110
  {
111
- "epoch": 30.434782608695652,
112
- "grad_norm": 0.7998800277709961,
113
  "learning_rate": 3.5e-06,
114
- "loss": 0.4023,
115
  "step": 700
116
  },
117
  {
118
- "epoch": 32.608695652173914,
119
- "grad_norm": 1.0301792621612549,
120
  "learning_rate": 3.7500000000000005e-06,
121
- "loss": 0.4045,
122
  "step": 750
123
  },
124
  {
125
- "epoch": 34.78260869565217,
126
- "grad_norm": 0.6388185620307922,
127
  "learning_rate": 4.000000000000001e-06,
128
- "loss": 0.4045,
129
  "step": 800
130
  },
131
  {
132
- "epoch": 36.95652173913044,
133
- "grad_norm": 0.9676840901374817,
134
  "learning_rate": 4.25e-06,
135
- "loss": 0.4116,
136
  "step": 850
137
  },
138
  {
139
- "epoch": 39.130434782608695,
140
- "grad_norm": 0.8355514407157898,
141
  "learning_rate": 4.5e-06,
142
- "loss": 0.4013,
143
  "step": 900
144
  },
145
  {
146
- "epoch": 41.30434782608695,
147
- "grad_norm": 0.7534067630767822,
148
  "learning_rate": 4.75e-06,
149
- "loss": 0.4019,
150
  "step": 950
151
  },
152
  {
153
- "epoch": 43.47826086956522,
154
- "grad_norm": 0.9366117119789124,
155
  "learning_rate": 5e-06,
156
- "loss": 0.4064,
157
  "step": 1000
158
  },
159
  {
160
- "epoch": 43.47826086956522,
161
- "eval_loss": 0.41316837072372437,
162
- "eval_runtime": 6.5616,
163
- "eval_samples_per_second": 24.232,
164
- "eval_steps_per_second": 3.048,
165
  "step": 1000
166
  },
167
  {
168
- "epoch": 45.65217391304348,
169
- "grad_norm": 0.7025179266929626,
170
  "learning_rate": 5.2500000000000006e-06,
171
- "loss": 0.4061,
172
  "step": 1050
173
  },
174
  {
175
- "epoch": 47.82608695652174,
176
- "grad_norm": 0.9240646958351135,
177
  "learning_rate": 5.500000000000001e-06,
178
- "loss": 0.4061,
179
  "step": 1100
180
  },
181
  {
182
- "epoch": 50.0,
183
- "grad_norm": 0.9685391187667847,
184
  "learning_rate": 5.75e-06,
185
- "loss": 0.4063,
186
  "step": 1150
187
  },
188
  {
189
- "epoch": 52.17391304347826,
190
- "grad_norm": 0.8591094017028809,
191
  "learning_rate": 6e-06,
192
- "loss": 0.4013,
193
  "step": 1200
194
  },
195
  {
196
- "epoch": 54.34782608695652,
197
- "grad_norm": 0.9348228573799133,
198
  "learning_rate": 6.25e-06,
199
- "loss": 0.402,
200
  "step": 1250
201
  },
202
  {
203
- "epoch": 56.52173913043478,
204
- "grad_norm": 0.7908409237861633,
205
  "learning_rate": 6.5000000000000004e-06,
206
- "loss": 0.4046,
207
  "step": 1300
208
  },
209
  {
210
- "epoch": 58.69565217391305,
211
- "grad_norm": 1.109017014503479,
212
  "learning_rate": 6.750000000000001e-06,
213
- "loss": 0.3995,
214
  "step": 1350
215
  },
216
  {
217
- "epoch": 60.869565217391305,
218
- "grad_norm": 0.8529478311538696,
219
  "learning_rate": 7e-06,
220
- "loss": 0.4018,
221
  "step": 1400
222
  },
223
  {
224
- "epoch": 63.04347826086956,
225
- "grad_norm": 0.9947476387023926,
226
  "learning_rate": 7.25e-06,
227
- "loss": 0.401,
228
  "step": 1450
229
  },
230
  {
231
- "epoch": 65.21739130434783,
232
- "grad_norm": 1.2476813793182373,
233
  "learning_rate": 7.500000000000001e-06,
234
- "loss": 0.4034,
235
  "step": 1500
236
  },
237
  {
238
- "epoch": 65.21739130434783,
239
- "eval_loss": 0.41223829984664917,
240
- "eval_runtime": 6.5846,
241
- "eval_samples_per_second": 24.147,
242
- "eval_steps_per_second": 3.037,
243
  "step": 1500
244
  },
245
  {
246
- "epoch": 67.3913043478261,
247
- "grad_norm": 0.8658013343811035,
248
  "learning_rate": 7.75e-06,
249
- "loss": 0.4007,
250
  "step": 1550
251
  },
252
  {
253
- "epoch": 69.56521739130434,
254
- "grad_norm": 0.7932788133621216,
255
  "learning_rate": 8.000000000000001e-06,
256
- "loss": 0.4054,
257
  "step": 1600
258
  },
259
  {
260
- "epoch": 71.73913043478261,
261
- "grad_norm": 2.33347225189209,
262
  "learning_rate": 8.25e-06,
263
- "loss": 0.4075,
264
  "step": 1650
265
  },
266
  {
267
- "epoch": 73.91304347826087,
268
- "grad_norm": 1.0095164775848389,
269
  "learning_rate": 8.5e-06,
270
- "loss": 0.3982,
271
  "step": 1700
272
  },
273
  {
274
- "epoch": 76.08695652173913,
275
- "grad_norm": 0.8895902633666992,
276
  "learning_rate": 8.750000000000001e-06,
277
- "loss": 0.4018,
278
  "step": 1750
279
  },
280
  {
281
- "epoch": 78.26086956521739,
282
- "grad_norm": 1.2252330780029297,
283
  "learning_rate": 9e-06,
284
- "loss": 0.4024,
285
  "step": 1800
286
  },
287
  {
288
- "epoch": 80.43478260869566,
289
- "grad_norm": 0.7731898427009583,
290
  "learning_rate": 9.250000000000001e-06,
291
- "loss": 0.3994,
292
  "step": 1850
293
  },
294
  {
295
- "epoch": 82.6086956521739,
296
- "grad_norm": 1.047607660293579,
297
  "learning_rate": 9.5e-06,
298
- "loss": 0.4013,
299
  "step": 1900
300
  },
301
  {
302
- "epoch": 84.78260869565217,
303
- "grad_norm": 0.9284490346908569,
304
  "learning_rate": 9.75e-06,
305
- "loss": 0.3986,
306
  "step": 1950
307
  },
308
  {
309
- "epoch": 86.95652173913044,
310
- "grad_norm": 4.725491046905518,
311
  "learning_rate": 1e-05,
312
- "loss": 0.3985,
313
  "step": 2000
314
  },
315
  {
316
- "epoch": 86.95652173913044,
317
- "eval_loss": 0.4147759974002838,
318
- "eval_runtime": 6.5475,
319
- "eval_samples_per_second": 24.284,
320
- "eval_steps_per_second": 3.055,
321
  "step": 2000
322
  },
323
  {
324
- "epoch": 89.1304347826087,
325
- "grad_norm": 1.1678683757781982,
326
  "learning_rate": 9.916666666666668e-06,
327
- "loss": 0.4066,
328
  "step": 2050
329
  },
330
  {
331
- "epoch": 91.30434782608695,
332
- "grad_norm": 1.988890528678894,
333
- "learning_rate": 9.835000000000002e-06,
334
- "loss": 0.3997,
335
  "step": 2100
336
  },
337
  {
338
- "epoch": 93.47826086956522,
339
- "grad_norm": 1.0278427600860596,
340
- "learning_rate": 9.751666666666667e-06,
341
- "loss": 0.3948,
342
  "step": 2150
343
  },
344
  {
345
- "epoch": 95.65217391304348,
346
- "grad_norm": 0.9606735110282898,
347
- "learning_rate": 9.67e-06,
348
- "loss": 0.42,
349
  "step": 2200
350
  },
351
  {
352
- "epoch": 97.82608695652173,
353
- "grad_norm": 1.0861903429031372,
354
- "learning_rate": 9.586666666666667e-06,
355
- "loss": 0.398,
356
  "step": 2250
357
  },
358
  {
359
- "epoch": 100.0,
360
- "grad_norm": 1.4872643947601318,
361
- "learning_rate": 9.503333333333334e-06,
362
- "loss": 0.4059,
363
  "step": 2300
364
  },
365
  {
366
- "epoch": 102.17391304347827,
367
- "grad_norm": 1.0460981130599976,
368
- "learning_rate": 9.42e-06,
369
- "loss": 0.4014,
370
  "step": 2350
371
  },
372
  {
373
- "epoch": 104.34782608695652,
374
- "grad_norm": 1.906476378440857,
375
- "learning_rate": 9.336666666666666e-06,
376
- "loss": 0.4015,
377
  "step": 2400
378
  },
379
  {
380
- "epoch": 106.52173913043478,
381
- "grad_norm": 1.3965051174163818,
382
- "learning_rate": 9.253333333333333e-06,
383
- "loss": 0.4063,
384
  "step": 2450
385
  },
386
  {
387
- "epoch": 108.69565217391305,
388
- "grad_norm": 1.7410107851028442,
389
- "learning_rate": 9.17e-06,
390
- "loss": 0.396,
391
  "step": 2500
392
  },
393
  {
394
- "epoch": 108.69565217391305,
395
- "eval_loss": 0.41233959794044495,
396
- "eval_runtime": 6.6031,
397
- "eval_samples_per_second": 24.08,
398
- "eval_steps_per_second": 3.029,
399
  "step": 2500
400
  },
401
  {
402
- "epoch": 110.8695652173913,
403
- "grad_norm": 1.4562053680419922,
404
- "learning_rate": 9.088333333333334e-06,
405
- "loss": 0.4042,
406
  "step": 2550
407
  },
408
  {
409
- "epoch": 113.04347826086956,
410
- "grad_norm": 0.8741047978401184,
411
- "learning_rate": 9.005000000000001e-06,
412
- "loss": 0.3985,
413
  "step": 2600
414
  },
415
  {
416
- "epoch": 115.21739130434783,
417
- "grad_norm": 1.1817028522491455,
418
- "learning_rate": 8.921666666666668e-06,
419
- "loss": 0.4036,
420
  "step": 2650
421
  },
422
  {
423
- "epoch": 117.3913043478261,
424
- "grad_norm": 0.8515534996986389,
425
- "learning_rate": 8.838333333333335e-06,
426
- "loss": 0.3986,
427
  "step": 2700
428
  },
429
  {
430
- "epoch": 119.56521739130434,
431
- "grad_norm": 0.9545428156852722,
432
- "learning_rate": 8.755e-06,
433
- "loss": 0.3946,
434
  "step": 2750
435
  },
436
  {
437
- "epoch": 121.73913043478261,
438
- "grad_norm": 0.6653321385383606,
439
- "learning_rate": 8.671666666666667e-06,
440
- "loss": 0.3957,
441
  "step": 2800
442
  },
443
  {
444
- "epoch": 123.91304347826087,
445
- "grad_norm": 0.8100114464759827,
446
- "learning_rate": 8.588333333333334e-06,
447
- "loss": 0.3953,
448
  "step": 2850
449
  },
450
  {
451
- "epoch": 126.08695652173913,
452
- "grad_norm": 1.4174237251281738,
453
- "learning_rate": 8.505e-06,
454
- "loss": 0.3933,
455
  "step": 2900
456
  },
457
  {
458
- "epoch": 128.2608695652174,
459
- "grad_norm": 1.6101123094558716,
460
- "learning_rate": 8.421666666666668e-06,
461
- "loss": 0.4027,
462
  "step": 2950
463
  },
464
  {
465
- "epoch": 130.43478260869566,
466
- "grad_norm": 0.7453967332839966,
467
- "learning_rate": 8.338333333333335e-06,
468
- "loss": 0.3958,
469
  "step": 3000
470
  },
471
  {
472
- "epoch": 130.43478260869566,
473
- "eval_loss": 0.4138609766960144,
474
- "eval_runtime": 6.6093,
475
- "eval_samples_per_second": 24.057,
476
- "eval_steps_per_second": 3.026,
477
  "step": 3000
478
  },
479
  {
480
- "epoch": 132.6086956521739,
481
- "grad_norm": 1.061684012413025,
482
- "learning_rate": 8.255000000000001e-06,
483
- "loss": 0.3966,
484
  "step": 3050
485
  },
486
  {
487
- "epoch": 134.7826086956522,
488
- "grad_norm": 1.0820105075836182,
489
- "learning_rate": 8.171666666666668e-06,
490
- "loss": 0.3937,
491
  "step": 3100
492
  },
493
  {
494
- "epoch": 136.95652173913044,
495
- "grad_norm": 0.8147062063217163,
496
- "learning_rate": 8.088333333333334e-06,
497
- "loss": 0.3961,
498
  "step": 3150
499
  },
500
  {
501
- "epoch": 139.1304347826087,
502
- "grad_norm": 0.7450351715087891,
503
- "learning_rate": 8.005e-06,
504
- "loss": 0.3923,
505
  "step": 3200
506
  },
507
  {
508
- "epoch": 141.30434782608697,
509
- "grad_norm": 1.05226731300354,
510
- "learning_rate": 7.921666666666667e-06,
511
- "loss": 0.3951,
512
  "step": 3250
513
  },
514
  {
515
- "epoch": 143.47826086956522,
516
- "grad_norm": 1.5119163990020752,
517
- "learning_rate": 7.838333333333334e-06,
518
- "loss": 0.3957,
519
  "step": 3300
520
  },
521
  {
522
- "epoch": 145.65217391304347,
523
- "grad_norm": 1.0422383546829224,
524
- "learning_rate": 7.755000000000001e-06,
525
- "loss": 0.3942,
526
  "step": 3350
527
  },
528
  {
529
- "epoch": 147.82608695652175,
530
- "grad_norm": 1.2120935916900635,
531
- "learning_rate": 7.671666666666668e-06,
532
- "loss": 0.3946,
533
  "step": 3400
534
  },
535
  {
536
- "epoch": 150.0,
537
- "grad_norm": 1.3127976655960083,
538
- "learning_rate": 7.588333333333334e-06,
539
- "loss": 0.3933,
540
  "step": 3450
541
  },
542
  {
543
- "epoch": 152.17391304347825,
544
- "grad_norm": 0.7704339027404785,
545
- "learning_rate": 7.505e-06,
546
- "loss": 0.3928,
547
  "step": 3500
548
  },
549
  {
550
- "epoch": 152.17391304347825,
551
- "eval_loss": 0.4132215976715088,
552
- "eval_runtime": 6.6162,
553
- "eval_samples_per_second": 24.032,
554
- "eval_steps_per_second": 3.023,
555
  "step": 3500
556
  },
557
  {
558
- "epoch": 154.34782608695653,
559
- "grad_norm": 1.4827122688293457,
560
- "learning_rate": 7.421666666666667e-06,
561
- "loss": 0.3908,
562
  "step": 3550
563
  },
564
  {
565
- "epoch": 156.52173913043478,
566
- "grad_norm": 1.0464049577713013,
567
- "learning_rate": 7.338333333333334e-06,
568
- "loss": 0.3993,
569
  "step": 3600
570
  },
571
  {
572
- "epoch": 158.69565217391303,
573
- "grad_norm": 0.8279913663864136,
574
- "learning_rate": 7.255000000000001e-06,
575
- "loss": 0.3967,
576
  "step": 3650
577
  },
578
  {
579
- "epoch": 160.8695652173913,
580
- "grad_norm": 0.9175034165382385,
581
- "learning_rate": 7.171666666666667e-06,
582
- "loss": 0.3911,
583
  "step": 3700
584
  },
585
  {
586
- "epoch": 163.04347826086956,
587
- "grad_norm": 1.0080233812332153,
588
- "learning_rate": 7.088333333333334e-06,
589
- "loss": 0.3898,
590
  "step": 3750
591
  },
592
  {
593
- "epoch": 165.2173913043478,
594
- "grad_norm": 0.9944823980331421,
595
- "learning_rate": 7.005000000000001e-06,
596
- "loss": 0.3901,
597
  "step": 3800
598
  },
599
  {
600
- "epoch": 167.3913043478261,
601
- "grad_norm": 1.8222392797470093,
602
- "learning_rate": 6.921666666666668e-06,
603
- "loss": 0.3901,
604
  "step": 3850
605
  },
606
  {
607
- "epoch": 169.56521739130434,
608
- "grad_norm": 0.9506675004959106,
609
- "learning_rate": 6.838333333333334e-06,
610
- "loss": 0.3945,
611
  "step": 3900
612
  },
613
  {
614
- "epoch": 171.7391304347826,
615
- "grad_norm": 4.512270450592041,
616
- "learning_rate": 6.7550000000000005e-06,
617
- "loss": 0.3913,
618
  "step": 3950
619
  },
620
  {
621
- "epoch": 173.91304347826087,
622
- "grad_norm": 0.74826580286026,
623
- "learning_rate": 6.6716666666666674e-06,
624
- "loss": 0.3914,
625
  "step": 4000
626
  },
627
  {
628
- "epoch": 173.91304347826087,
629
- "eval_loss": 0.4138501286506653,
630
- "eval_runtime": 6.5399,
631
- "eval_samples_per_second": 24.312,
632
- "eval_steps_per_second": 3.058,
633
  "step": 4000
634
  }
635
  ],
636
  "logging_steps": 50,
637
  "max_steps": 8000,
638
  "num_input_tokens_seen": 0,
639
- "num_train_epochs": 348,
640
  "save_steps": 500,
641
  "stateful_callbacks": {
642
  "TrainerControl": {
@@ -650,8 +650,8 @@
650
  "attributes": {}
651
  }
652
  },
653
- "total_flos": 5.923192104795082e+16,
654
- "train_batch_size": 64,
655
  "trial_name": null,
656
  "trial_params": null
657
  }
 
1
  {
2
+ "best_metric": 0.3931116759777069,
3
+ "best_model_checkpoint": "mikhail-panzo/zlm-fil-ceb_b64_le5_s8000/checkpoint-3500",
4
+ "epoch": 158.41584158415841,
5
  "eval_steps": 500,
6
  "global_step": 4000,
7
  "is_hyper_param_search": false,
 
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
+ "epoch": 1.9801980198019802,
13
+ "grad_norm": 1.2683665752410889,
14
  "learning_rate": 2.5000000000000004e-07,
15
+ "loss": 0.4932,
16
  "step": 50
17
  },
18
  {
19
+ "epoch": 3.9603960396039604,
20
+ "grad_norm": 1.1266463994979858,
21
  "learning_rate": 5.000000000000001e-07,
22
+ "loss": 0.4898,
23
  "step": 100
24
  },
25
  {
26
+ "epoch": 5.9405940594059405,
27
+ "grad_norm": 1.1800968647003174,
28
  "learning_rate": 7.5e-07,
29
+ "loss": 0.4839,
30
  "step": 150
31
  },
32
  {
33
+ "epoch": 7.920792079207921,
34
+ "grad_norm": 0.8235568404197693,
35
  "learning_rate": 1.0000000000000002e-06,
36
+ "loss": 0.4785,
37
  "step": 200
38
  },
39
  {
40
+ "epoch": 9.900990099009901,
41
+ "grad_norm": 1.313211441040039,
42
  "learning_rate": 1.25e-06,
43
+ "loss": 0.4767,
44
  "step": 250
45
  },
46
  {
47
+ "epoch": 11.881188118811881,
48
+ "grad_norm": 0.7831560373306274,
49
  "learning_rate": 1.5e-06,
50
+ "loss": 0.4681,
51
  "step": 300
52
  },
53
  {
54
+ "epoch": 13.861386138613861,
55
+ "grad_norm": 0.7987237572669983,
56
  "learning_rate": 1.75e-06,
57
+ "loss": 0.4658,
58
  "step": 350
59
  },
60
  {
61
+ "epoch": 15.841584158415841,
62
+ "grad_norm": 0.7143813371658325,
63
  "learning_rate": 2.0000000000000003e-06,
64
+ "loss": 0.4627,
65
  "step": 400
66
  },
67
  {
68
+ "epoch": 17.821782178217823,
69
+ "grad_norm": 0.8037531971931458,
70
  "learning_rate": 2.25e-06,
71
+ "loss": 0.461,
72
  "step": 450
73
  },
74
  {
75
+ "epoch": 19.801980198019802,
76
+ "grad_norm": 0.79031902551651,
77
  "learning_rate": 2.5e-06,
78
+ "loss": 0.4592,
79
  "step": 500
80
  },
81
  {
82
+ "epoch": 19.801980198019802,
83
+ "eval_loss": 0.42528408765792847,
84
+ "eval_runtime": 7.8982,
85
+ "eval_samples_per_second": 22.79,
86
+ "eval_steps_per_second": 2.912,
87
  "step": 500
88
  },
89
  {
90
+ "epoch": 21.782178217821784,
91
+ "grad_norm": 0.7966288328170776,
92
  "learning_rate": 2.7500000000000004e-06,
93
+ "loss": 0.4551,
94
  "step": 550
95
  },
96
  {
97
+ "epoch": 23.762376237623762,
98
+ "grad_norm": 0.8126183152198792,
99
  "learning_rate": 3e-06,
100
+ "loss": 0.4542,
101
  "step": 600
102
  },
103
  {
104
+ "epoch": 25.742574257425744,
105
+ "grad_norm": 0.6602805852890015,
106
  "learning_rate": 3.2500000000000002e-06,
107
+ "loss": 0.4535,
108
  "step": 650
109
  },
110
  {
111
+ "epoch": 27.722772277227723,
112
+ "grad_norm": 0.7177000641822815,
113
  "learning_rate": 3.5e-06,
114
+ "loss": 0.4484,
115
  "step": 700
116
  },
117
  {
118
+ "epoch": 29.702970297029704,
119
+ "grad_norm": 0.7912996411323547,
120
  "learning_rate": 3.7500000000000005e-06,
121
+ "loss": 0.4486,
122
  "step": 750
123
  },
124
  {
125
+ "epoch": 31.683168316831683,
126
+ "grad_norm": 0.6251747608184814,
127
  "learning_rate": 4.000000000000001e-06,
128
+ "loss": 0.4463,
129
  "step": 800
130
  },
131
  {
132
+ "epoch": 33.663366336633665,
133
+ "grad_norm": 1.003780722618103,
134
  "learning_rate": 4.25e-06,
135
+ "loss": 0.4474,
136
  "step": 850
137
  },
138
  {
139
+ "epoch": 35.64356435643565,
140
+ "grad_norm": 0.742171585559845,
141
  "learning_rate": 4.5e-06,
142
+ "loss": 0.4434,
143
  "step": 900
144
  },
145
  {
146
+ "epoch": 37.62376237623762,
147
+ "grad_norm": 0.6034647822380066,
148
  "learning_rate": 4.75e-06,
149
+ "loss": 0.4425,
150
  "step": 950
151
  },
152
  {
153
+ "epoch": 39.603960396039604,
154
+ "grad_norm": 0.6990635395050049,
155
  "learning_rate": 5e-06,
156
+ "loss": 0.4381,
157
  "step": 1000
158
  },
159
  {
160
+ "epoch": 39.603960396039604,
161
+ "eval_loss": 0.41002073884010315,
162
+ "eval_runtime": 6.7782,
163
+ "eval_samples_per_second": 26.556,
164
+ "eval_steps_per_second": 3.393,
165
  "step": 1000
166
  },
167
  {
168
+ "epoch": 41.584158415841586,
169
+ "grad_norm": 0.9420008659362793,
170
  "learning_rate": 5.2500000000000006e-06,
171
+ "loss": 0.4421,
172
  "step": 1050
173
  },
174
  {
175
+ "epoch": 43.56435643564357,
176
+ "grad_norm": 0.7143107652664185,
177
  "learning_rate": 5.500000000000001e-06,
178
+ "loss": 0.4389,
179
  "step": 1100
180
  },
181
  {
182
+ "epoch": 45.54455445544554,
183
+ "grad_norm": 0.7934616804122925,
184
  "learning_rate": 5.75e-06,
185
+ "loss": 0.4369,
186
  "step": 1150
187
  },
188
  {
189
+ "epoch": 47.524752475247524,
190
+ "grad_norm": 0.9682563543319702,
191
  "learning_rate": 6e-06,
192
+ "loss": 0.4355,
193
  "step": 1200
194
  },
195
  {
196
+ "epoch": 49.504950495049506,
197
+ "grad_norm": 0.7880070209503174,
198
  "learning_rate": 6.25e-06,
199
+ "loss": 0.4359,
200
  "step": 1250
201
  },
202
  {
203
+ "epoch": 51.48514851485149,
204
+ "grad_norm": 0.8443828821182251,
205
  "learning_rate": 6.5000000000000004e-06,
206
+ "loss": 0.4324,
207
  "step": 1300
208
  },
209
  {
210
+ "epoch": 53.46534653465346,
211
+ "grad_norm": 0.724743127822876,
212
  "learning_rate": 6.750000000000001e-06,
213
+ "loss": 0.4348,
214
  "step": 1350
215
  },
216
  {
217
+ "epoch": 55.445544554455445,
218
+ "grad_norm": 0.7557791471481323,
219
  "learning_rate": 7e-06,
220
+ "loss": 0.433,
221
  "step": 1400
222
  },
223
  {
224
+ "epoch": 57.42574257425743,
225
+ "grad_norm": 0.7936097383499146,
226
  "learning_rate": 7.25e-06,
227
+ "loss": 0.4273,
228
  "step": 1450
229
  },
230
  {
231
+ "epoch": 59.40594059405941,
232
+ "grad_norm": 0.7041624784469604,
233
  "learning_rate": 7.500000000000001e-06,
234
+ "loss": 0.4281,
235
  "step": 1500
236
  },
237
  {
238
+ "epoch": 59.40594059405941,
239
+ "eval_loss": 0.40218469500541687,
240
+ "eval_runtime": 7.4309,
241
+ "eval_samples_per_second": 24.223,
242
+ "eval_steps_per_second": 3.095,
243
  "step": 1500
244
  },
245
  {
246
+ "epoch": 61.386138613861384,
247
+ "grad_norm": 0.9194286465644836,
248
  "learning_rate": 7.75e-06,
249
+ "loss": 0.4275,
250
  "step": 1550
251
  },
252
  {
253
+ "epoch": 63.366336633663366,
254
+ "grad_norm": 0.7239477634429932,
255
  "learning_rate": 8.000000000000001e-06,
256
+ "loss": 0.4245,
257
  "step": 1600
258
  },
259
  {
260
+ "epoch": 65.34653465346534,
261
+ "grad_norm": 0.7235706448554993,
262
  "learning_rate": 8.25e-06,
263
+ "loss": 0.4248,
264
  "step": 1650
265
  },
266
  {
267
+ "epoch": 67.32673267326733,
268
+ "grad_norm": 0.7281640768051147,
269
  "learning_rate": 8.5e-06,
270
+ "loss": 0.425,
271
  "step": 1700
272
  },
273
  {
274
+ "epoch": 69.3069306930693,
275
+ "grad_norm": 1.011294960975647,
276
  "learning_rate": 8.750000000000001e-06,
277
+ "loss": 0.4238,
278
  "step": 1750
279
  },
280
  {
281
+ "epoch": 71.2871287128713,
282
+ "grad_norm": 1.1249107122421265,
283
  "learning_rate": 9e-06,
284
+ "loss": 0.4203,
285
  "step": 1800
286
  },
287
  {
288
+ "epoch": 73.26732673267327,
289
+ "grad_norm": 0.7369679808616638,
290
  "learning_rate": 9.250000000000001e-06,
291
+ "loss": 0.422,
292
  "step": 1850
293
  },
294
  {
295
+ "epoch": 75.24752475247524,
296
+ "grad_norm": 0.704367995262146,
297
  "learning_rate": 9.5e-06,
298
+ "loss": 0.4201,
299
  "step": 1900
300
  },
301
  {
302
+ "epoch": 77.22772277227723,
303
+ "grad_norm": 0.8307437896728516,
304
  "learning_rate": 9.75e-06,
305
+ "loss": 0.4203,
306
  "step": 1950
307
  },
308
  {
309
+ "epoch": 79.20792079207921,
310
+ "grad_norm": 0.7827133536338806,
311
  "learning_rate": 1e-05,
312
+ "loss": 0.4195,
313
  "step": 2000
314
  },
315
  {
316
+ "epoch": 79.20792079207921,
317
+ "eval_loss": 0.39878609776496887,
318
+ "eval_runtime": 8.1579,
319
+ "eval_samples_per_second": 22.064,
320
+ "eval_steps_per_second": 2.819,
321
  "step": 2000
322
  },
323
  {
324
+ "epoch": 81.18811881188118,
325
+ "grad_norm": 0.8909313082695007,
326
  "learning_rate": 9.916666666666668e-06,
327
+ "loss": 0.4203,
328
  "step": 2050
329
  },
330
  {
331
+ "epoch": 83.16831683168317,
332
+ "grad_norm": 0.7147483825683594,
333
+ "learning_rate": 9.833333333333333e-06,
334
+ "loss": 0.418,
335
  "step": 2100
336
  },
337
  {
338
+ "epoch": 85.14851485148515,
339
+ "grad_norm": 0.6844545006752014,
340
+ "learning_rate": 9.75e-06,
341
+ "loss": 0.4167,
342
  "step": 2150
343
  },
344
  {
345
+ "epoch": 87.12871287128714,
346
+ "grad_norm": 0.88512122631073,
347
+ "learning_rate": 9.666666666666667e-06,
348
+ "loss": 0.4108,
349
  "step": 2200
350
  },
351
  {
352
+ "epoch": 89.10891089108911,
353
+ "grad_norm": 0.7097252607345581,
354
+ "learning_rate": 9.583333333333335e-06,
355
+ "loss": 0.4135,
356
  "step": 2250
357
  },
358
  {
359
+ "epoch": 91.08910891089108,
360
+ "grad_norm": 0.8400272727012634,
361
+ "learning_rate": 9.5e-06,
362
+ "loss": 0.4147,
363
  "step": 2300
364
  },
365
  {
366
+ "epoch": 93.06930693069307,
367
+ "grad_norm": 0.7614939212799072,
368
+ "learning_rate": 9.416666666666667e-06,
369
+ "loss": 0.4119,
370
  "step": 2350
371
  },
372
  {
373
+ "epoch": 95.04950495049505,
374
+ "grad_norm": 0.8099732995033264,
375
+ "learning_rate": 9.333333333333334e-06,
376
+ "loss": 0.4108,
377
  "step": 2400
378
  },
379
  {
380
+ "epoch": 97.02970297029702,
381
+ "grad_norm": 1.0697815418243408,
382
+ "learning_rate": 9.250000000000001e-06,
383
+ "loss": 0.4095,
384
  "step": 2450
385
  },
386
  {
387
+ "epoch": 99.00990099009901,
388
+ "grad_norm": 0.7235235571861267,
389
+ "learning_rate": 9.166666666666666e-06,
390
+ "loss": 0.4134,
391
  "step": 2500
392
  },
393
  {
394
+ "epoch": 99.00990099009901,
395
+ "eval_loss": 0.39545077085494995,
396
+ "eval_runtime": 7.724,
397
+ "eval_samples_per_second": 23.304,
398
+ "eval_steps_per_second": 2.978,
399
  "step": 2500
400
  },
401
  {
402
+ "epoch": 100.99009900990099,
403
+ "grad_norm": 0.6166990399360657,
404
+ "learning_rate": 9.083333333333333e-06,
405
+ "loss": 0.4097,
406
  "step": 2550
407
  },
408
  {
409
+ "epoch": 102.97029702970298,
410
+ "grad_norm": 0.6944295763969421,
411
+ "learning_rate": 9e-06,
412
+ "loss": 0.4107,
413
  "step": 2600
414
  },
415
  {
416
+ "epoch": 104.95049504950495,
417
+ "grad_norm": 0.8469858765602112,
418
+ "learning_rate": 8.916666666666667e-06,
419
+ "loss": 0.4056,
420
  "step": 2650
421
  },
422
  {
423
+ "epoch": 106.93069306930693,
424
+ "grad_norm": 0.6261239647865295,
425
+ "learning_rate": 8.833333333333334e-06,
426
+ "loss": 0.409,
427
  "step": 2700
428
  },
429
  {
430
+ "epoch": 108.91089108910892,
431
+ "grad_norm": 0.6304221749305725,
432
+ "learning_rate": 8.750000000000001e-06,
433
+ "loss": 0.4081,
434
  "step": 2750
435
  },
436
  {
437
+ "epoch": 110.89108910891089,
438
+ "grad_norm": 0.6513387560844421,
439
+ "learning_rate": 8.666666666666668e-06,
440
+ "loss": 0.4071,
441
  "step": 2800
442
  },
443
  {
444
+ "epoch": 112.87128712871286,
445
+ "grad_norm": 0.9311261177062988,
446
+ "learning_rate": 8.583333333333333e-06,
447
+ "loss": 0.4091,
448
  "step": 2850
449
  },
450
  {
451
+ "epoch": 114.85148514851485,
452
+ "grad_norm": 0.7296612858772278,
453
+ "learning_rate": 8.5e-06,
454
+ "loss": 0.4055,
455
  "step": 2900
456
  },
457
  {
458
+ "epoch": 116.83168316831683,
459
+ "grad_norm": 0.8149676322937012,
460
+ "learning_rate": 8.416666666666667e-06,
461
+ "loss": 0.4052,
462
  "step": 2950
463
  },
464
  {
465
+ "epoch": 118.81188118811882,
466
+ "grad_norm": 0.8266366124153137,
467
+ "learning_rate": 8.333333333333334e-06,
468
+ "loss": 0.4049,
469
  "step": 3000
470
  },
471
  {
472
+ "epoch": 118.81188118811882,
473
+ "eval_loss": 0.39345207810401917,
474
+ "eval_runtime": 7.2151,
475
+ "eval_samples_per_second": 24.948,
476
+ "eval_steps_per_second": 3.188,
477
  "step": 3000
478
  },
479
  {
480
+ "epoch": 120.79207920792079,
481
+ "grad_norm": 0.6681864261627197,
482
+ "learning_rate": 8.25e-06,
483
+ "loss": 0.4022,
484
  "step": 3050
485
  },
486
  {
487
+ "epoch": 122.77227722772277,
488
+ "grad_norm": 0.5798916220664978,
489
+ "learning_rate": 8.166666666666668e-06,
490
+ "loss": 0.4026,
491
  "step": 3100
492
  },
493
  {
494
+ "epoch": 124.75247524752476,
495
+ "grad_norm": 0.8516507744789124,
496
+ "learning_rate": 8.083333333333334e-06,
497
+ "loss": 0.4064,
498
  "step": 3150
499
  },
500
  {
501
+ "epoch": 126.73267326732673,
502
+ "grad_norm": 0.6938304901123047,
503
+ "learning_rate": 8.000000000000001e-06,
504
+ "loss": 0.4057,
505
  "step": 3200
506
  },
507
  {
508
+ "epoch": 128.7128712871287,
509
+ "grad_norm": 0.6889196038246155,
510
+ "learning_rate": 7.916666666666667e-06,
511
+ "loss": 0.4022,
512
  "step": 3250
513
  },
514
  {
515
+ "epoch": 130.69306930693068,
516
+ "grad_norm": 0.735680103302002,
517
+ "learning_rate": 7.833333333333333e-06,
518
+ "loss": 0.4019,
519
  "step": 3300
520
  },
521
  {
522
+ "epoch": 132.67326732673268,
523
+ "grad_norm": 1.0390517711639404,
524
+ "learning_rate": 7.75e-06,
525
+ "loss": 0.4014,
526
  "step": 3350
527
  },
528
  {
529
+ "epoch": 134.65346534653466,
530
+ "grad_norm": 0.807412326335907,
531
+ "learning_rate": 7.666666666666667e-06,
532
+ "loss": 0.3989,
533
  "step": 3400
534
  },
535
  {
536
+ "epoch": 136.63366336633663,
537
+ "grad_norm": 0.6928243637084961,
538
+ "learning_rate": 7.583333333333333e-06,
539
+ "loss": 0.4011,
540
  "step": 3450
541
  },
542
  {
543
+ "epoch": 138.6138613861386,
544
+ "grad_norm": 0.6894844770431519,
545
+ "learning_rate": 7.500000000000001e-06,
546
+ "loss": 0.4016,
547
  "step": 3500
548
  },
549
  {
550
+ "epoch": 138.6138613861386,
551
+ "eval_loss": 0.3931116759777069,
552
+ "eval_runtime": 7.7493,
553
+ "eval_samples_per_second": 23.228,
554
+ "eval_steps_per_second": 2.968,
555
  "step": 3500
556
  },
557
  {
558
+ "epoch": 140.59405940594058,
559
+ "grad_norm": 0.9431400895118713,
560
+ "learning_rate": 7.416666666666668e-06,
561
+ "loss": 0.4028,
562
  "step": 3550
563
  },
564
  {
565
+ "epoch": 142.5742574257426,
566
+ "grad_norm": 0.6703277230262756,
567
+ "learning_rate": 7.333333333333333e-06,
568
+ "loss": 0.3992,
569
  "step": 3600
570
  },
571
  {
572
+ "epoch": 144.55445544554456,
573
+ "grad_norm": 0.8334361910820007,
574
+ "learning_rate": 7.25e-06,
575
+ "loss": 0.3979,
576
  "step": 3650
577
  },
578
  {
579
+ "epoch": 146.53465346534654,
580
+ "grad_norm": 0.8266313672065735,
581
+ "learning_rate": 7.166666666666667e-06,
582
+ "loss": 0.3964,
583
  "step": 3700
584
  },
585
  {
586
+ "epoch": 148.5148514851485,
587
+ "grad_norm": 0.7688102126121521,
588
+ "learning_rate": 7.083333333333335e-06,
589
+ "loss": 0.3968,
590
  "step": 3750
591
  },
592
  {
593
+ "epoch": 150.4950495049505,
594
+ "grad_norm": 0.7516873478889465,
595
+ "learning_rate": 7e-06,
596
+ "loss": 0.3971,
597
  "step": 3800
598
  },
599
  {
600
+ "epoch": 152.47524752475246,
601
+ "grad_norm": 0.8034970760345459,
602
+ "learning_rate": 6.916666666666667e-06,
603
+ "loss": 0.3926,
604
  "step": 3850
605
  },
606
  {
607
+ "epoch": 154.45544554455446,
608
+ "grad_norm": 0.681676983833313,
609
+ "learning_rate": 6.833333333333334e-06,
610
+ "loss": 0.3975,
611
  "step": 3900
612
  },
613
  {
614
+ "epoch": 156.43564356435644,
615
+ "grad_norm": 0.6939652562141418,
616
+ "learning_rate": 6.750000000000001e-06,
617
+ "loss": 0.4005,
618
  "step": 3950
619
  },
620
  {
621
+ "epoch": 158.41584158415841,
622
+ "grad_norm": 0.6093599200248718,
623
+ "learning_rate": 6.666666666666667e-06,
624
+ "loss": 0.3937,
625
  "step": 4000
626
  },
627
  {
628
+ "epoch": 158.41584158415841,
629
+ "eval_loss": 0.39361944794654846,
630
+ "eval_runtime": 7.0786,
631
+ "eval_samples_per_second": 25.429,
632
+ "eval_steps_per_second": 3.249,
633
  "step": 4000
634
  }
635
  ],
636
  "logging_steps": 50,
637
  "max_steps": 8000,
638
  "num_input_tokens_seen": 0,
639
+ "num_train_epochs": 320,
640
  "save_steps": 500,
641
  "stateful_callbacks": {
642
  "TrainerControl": {
 
650
  "attributes": {}
651
  }
652
  },
653
+ "total_flos": 4.322114324975938e+16,
654
+ "train_batch_size": 16,
655
  "trial_name": null,
656
  "trial_params": null
657
  }
checkpoint-4000/training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f95de2b194fe7f16ec10fbf1b93d83e04c25498d479793b9a5748c1d338bfbd0
3
  size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e97537a0fb344678f807f30d02bd561fb4b5c2162a68338251550dd3ef50d20
3
  size 5304