fpadovani commited on
Commit
2d21de7
·
verified ·
1 Parent(s): 8051387

Training in progress, step 100000, checkpoint

Browse files
checkpoint-100000/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fd72c9b44ae0d5cdae58ebd673076b11b94ffc9f03e54cf93d5ff95cd1f87073
3
  size 51007160
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ed9362952b4fd87e379a57f57f1be931144492d1af09c3d76139f60bb946e69
3
  size 51007160
checkpoint-100000/optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2f67d291f29a655135c26d130495b93bb28a879980013ef9cf7867041b87ac36
3
  size 102078202
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2bc76a1f89eb1f131e0d31c61897d2b3af6d5bbb476dcfc8f2d42bec7a809ebb
3
  size 102078202
checkpoint-100000/rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c3b9085b50c19f9ac60c90400a9c1e9d625894bc0cec5d57cba56af004f51e3e
3
  size 14308
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e4af4adaf699c8fb88a64f00706b102e2f46f8cf32a083c3f6add1ead4b5e19
3
  size 14308
checkpoint-100000/scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:06aecbd3f5ff905306913a3e00527d5fa2c1773d23de5143e3eb9e01b973eb0f
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a5006c3420b0fd8d364af0ca32eb23c67b0cd2ef9e5167273a11d19175a43511
3
  size 1000
checkpoint-100000/tokenizer.json CHANGED
The diff for this file is too large to render. See raw diff
 
checkpoint-100000/trainer_state.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
- "best_metric": 4.224654674530029,
3
- "best_model_checkpoint": "/home/p318482/babyLM_controlled/models_trained/en_clm/wikipedia_30/checkpoint-72000",
4
- "epoch": 52.59006047856955,
5
  "eval_steps": 2000,
6
  "global_step": 100000,
7
  "is_hyper_param_search": false,
@@ -9,585 +9,585 @@
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
- "epoch": 1.0518012095713911,
13
- "eval_loss": 7.518920421600342,
14
- "eval_runtime": 2.2399,
15
- "eval_samples_per_second": 1422.808,
16
- "eval_steps_per_second": 89.288,
17
  "step": 2000
18
  },
19
  {
20
- "epoch": 2.1036024191427822,
21
- "grad_norm": 1.2947496175765991,
22
  "learning_rate": 1e-05,
23
- "loss": 7.6141,
24
  "step": 4000
25
  },
26
  {
27
- "epoch": 2.1036024191427822,
28
- "eval_loss": 6.568221092224121,
29
- "eval_runtime": 2.1236,
30
- "eval_samples_per_second": 1500.783,
31
- "eval_steps_per_second": 94.182,
32
  "step": 4000
33
  },
34
  {
35
- "epoch": 3.155403628714173,
36
- "eval_loss": 6.16925573348999,
37
- "eval_runtime": 2.1624,
38
- "eval_samples_per_second": 1473.825,
39
- "eval_steps_per_second": 92.49,
40
  "step": 6000
41
  },
42
  {
43
- "epoch": 4.2072048382855645,
44
- "grad_norm": 2.330080270767212,
45
- "learning_rate": 2e-05,
46
- "loss": 6.2396,
47
  "step": 8000
48
  },
49
  {
50
- "epoch": 4.2072048382855645,
51
- "eval_loss": 5.901575088500977,
52
- "eval_runtime": 2.1326,
53
- "eval_samples_per_second": 1494.388,
54
- "eval_steps_per_second": 93.78,
55
  "step": 8000
56
  },
57
  {
58
- "epoch": 5.259006047856955,
59
- "eval_loss": 5.67369270324707,
60
- "eval_runtime": 2.1364,
61
- "eval_samples_per_second": 1491.788,
62
- "eval_steps_per_second": 93.617,
63
  "step": 10000
64
  },
65
  {
66
- "epoch": 6.310807257428346,
67
- "grad_norm": 2.565220594406128,
68
- "learning_rate": 2.99925e-05,
69
- "loss": 5.7217,
70
  "step": 12000
71
  },
72
  {
73
- "epoch": 6.310807257428346,
74
- "eval_loss": 5.460005760192871,
75
- "eval_runtime": 2.1835,
76
- "eval_samples_per_second": 1459.563,
77
- "eval_steps_per_second": 91.595,
78
  "step": 12000
79
  },
80
  {
81
- "epoch": 7.362608466999737,
82
- "eval_loss": 5.272489547729492,
83
- "eval_runtime": 2.1833,
84
- "eval_samples_per_second": 1459.691,
85
- "eval_steps_per_second": 91.603,
86
  "step": 14000
87
  },
88
  {
89
- "epoch": 8.414409676571129,
90
- "grad_norm": 2.7334115505218506,
91
- "learning_rate": 3.99875e-05,
92
- "loss": 5.3064,
93
  "step": 16000
94
  },
95
  {
96
- "epoch": 8.414409676571129,
97
- "eval_loss": 5.1074652671813965,
98
- "eval_runtime": 2.2402,
99
- "eval_samples_per_second": 1422.652,
100
- "eval_steps_per_second": 89.278,
101
  "step": 16000
102
  },
103
  {
104
- "epoch": 9.46621088614252,
105
- "eval_loss": 4.9716596603393555,
106
- "eval_runtime": 2.1736,
107
- "eval_samples_per_second": 1466.258,
108
- "eval_steps_per_second": 92.015,
109
  "step": 18000
110
  },
111
  {
112
- "epoch": 10.51801209571391,
113
- "grad_norm": 2.6722376346588135,
114
- "learning_rate": 4.9985e-05,
115
- "loss": 4.9744,
116
  "step": 20000
117
  },
118
  {
119
- "epoch": 10.51801209571391,
120
- "eval_loss": 4.864916801452637,
121
- "eval_runtime": 2.2031,
122
- "eval_samples_per_second": 1446.585,
123
- "eval_steps_per_second": 90.78,
124
  "step": 20000
125
  },
126
  {
127
- "epoch": 11.569813305285301,
128
- "eval_loss": 4.763542652130127,
129
- "eval_runtime": 2.4382,
130
- "eval_samples_per_second": 1307.131,
131
- "eval_steps_per_second": 82.029,
132
  "step": 22000
133
  },
134
  {
135
- "epoch": 12.621614514856692,
136
- "grad_norm": 2.4539408683776855,
137
- "learning_rate": 5.9985e-05,
138
- "loss": 4.7273,
139
  "step": 24000
140
  },
141
  {
142
- "epoch": 12.621614514856692,
143
- "eval_loss": 4.683297634124756,
144
- "eval_runtime": 2.452,
145
- "eval_samples_per_second": 1299.733,
146
- "eval_steps_per_second": 81.565,
147
  "step": 24000
148
  },
149
  {
150
- "epoch": 13.673415724428082,
151
- "eval_loss": 4.621574401855469,
152
- "eval_runtime": 2.1825,
153
- "eval_samples_per_second": 1460.226,
154
- "eval_steps_per_second": 91.636,
155
  "step": 26000
156
  },
157
  {
158
- "epoch": 14.725216933999475,
159
- "grad_norm": 2.4913346767425537,
160
- "learning_rate": 6.998e-05,
161
- "loss": 4.5397,
162
  "step": 28000
163
  },
164
  {
165
- "epoch": 14.725216933999475,
166
- "eval_loss": 4.562623977661133,
167
- "eval_runtime": 2.201,
168
- "eval_samples_per_second": 1447.965,
169
- "eval_steps_per_second": 90.867,
170
  "step": 28000
171
  },
172
  {
173
- "epoch": 15.777018143570865,
174
- "eval_loss": 4.506168842315674,
175
- "eval_runtime": 2.2491,
176
- "eval_samples_per_second": 1416.988,
177
- "eval_steps_per_second": 88.923,
178
  "step": 30000
179
  },
180
  {
181
- "epoch": 16.828819353142258,
182
- "grad_norm": 2.2661523818969727,
183
- "learning_rate": 7.99775e-05,
184
- "loss": 4.3839,
185
  "step": 32000
186
  },
187
  {
188
- "epoch": 16.828819353142258,
189
- "eval_loss": 4.464449882507324,
190
- "eval_runtime": 2.253,
191
- "eval_samples_per_second": 1414.55,
192
- "eval_steps_per_second": 88.77,
193
  "step": 32000
194
  },
195
  {
196
- "epoch": 17.88062056271365,
197
- "eval_loss": 4.426318168640137,
198
- "eval_runtime": 2.2688,
199
- "eval_samples_per_second": 1404.709,
200
- "eval_steps_per_second": 88.152,
201
  "step": 34000
202
  },
203
  {
204
- "epoch": 18.93242177228504,
205
- "grad_norm": 2.226025104522705,
206
- "learning_rate": 8.9975e-05,
207
- "loss": 4.2529,
208
  "step": 36000
209
  },
210
  {
211
- "epoch": 18.93242177228504,
212
- "eval_loss": 4.3944478034973145,
213
- "eval_runtime": 2.294,
214
- "eval_samples_per_second": 1389.273,
215
- "eval_steps_per_second": 87.184,
216
  "step": 36000
217
  },
218
  {
219
- "epoch": 19.98422298185643,
220
- "eval_loss": 4.365135669708252,
221
- "eval_runtime": 2.2176,
222
- "eval_samples_per_second": 1437.165,
223
- "eval_steps_per_second": 90.189,
224
  "step": 38000
225
  },
226
  {
227
- "epoch": 21.03602419142782,
228
- "grad_norm": 2.1669206619262695,
229
- "learning_rate": 9.997e-05,
230
- "loss": 4.1409,
231
  "step": 40000
232
  },
233
  {
234
- "epoch": 21.03602419142782,
235
- "eval_loss": 4.351284503936768,
236
- "eval_runtime": 2.1934,
237
- "eval_samples_per_second": 1453.028,
238
- "eval_steps_per_second": 91.185,
239
  "step": 40000
240
  },
241
  {
242
- "epoch": 22.08782540099921,
243
- "eval_loss": 4.321634292602539,
244
- "eval_runtime": 2.2273,
245
- "eval_samples_per_second": 1430.895,
246
- "eval_steps_per_second": 89.796,
247
  "step": 42000
248
  },
249
  {
250
- "epoch": 23.139626610570602,
251
- "grad_norm": 2.0360896587371826,
252
- "learning_rate": 9.335333333333333e-05,
253
- "loss": 4.0334,
254
  "step": 44000
255
  },
256
  {
257
- "epoch": 23.139626610570602,
258
- "eval_loss": 4.300807476043701,
259
- "eval_runtime": 2.3226,
260
- "eval_samples_per_second": 1372.142,
261
- "eval_steps_per_second": 86.109,
262
  "step": 44000
263
  },
264
  {
265
- "epoch": 24.191427820141993,
266
- "eval_loss": 4.279116153717041,
267
- "eval_runtime": 2.2177,
268
- "eval_samples_per_second": 1437.082,
269
- "eval_steps_per_second": 90.184,
270
  "step": 46000
271
  },
272
  {
273
- "epoch": 25.243229029713383,
274
- "grad_norm": 2.137535572052002,
275
- "learning_rate": 8.668833333333334e-05,
276
- "loss": 3.9378,
277
  "step": 48000
278
  },
279
  {
280
- "epoch": 25.243229029713383,
281
- "eval_loss": 4.2718939781188965,
282
- "eval_runtime": 2.2019,
283
- "eval_samples_per_second": 1447.39,
284
- "eval_steps_per_second": 90.831,
285
  "step": 48000
286
  },
287
  {
288
- "epoch": 26.295030239284774,
289
- "eval_loss": 4.26237678527832,
290
- "eval_runtime": 2.2417,
291
- "eval_samples_per_second": 1421.698,
292
- "eval_steps_per_second": 89.219,
293
  "step": 50000
294
  },
295
  {
296
- "epoch": 27.346831448856165,
297
- "grad_norm": 2.1131036281585693,
298
- "learning_rate": 8.002333333333334e-05,
299
- "loss": 3.8595,
300
  "step": 52000
301
  },
302
  {
303
- "epoch": 27.346831448856165,
304
- "eval_loss": 4.253220558166504,
305
- "eval_runtime": 2.2045,
306
- "eval_samples_per_second": 1445.692,
307
- "eval_steps_per_second": 90.724,
308
  "step": 52000
309
  },
310
  {
311
- "epoch": 28.39863265842756,
312
- "eval_loss": 4.243188381195068,
313
- "eval_runtime": 2.1654,
314
- "eval_samples_per_second": 1471.751,
315
- "eval_steps_per_second": 92.36,
316
  "step": 54000
317
  },
318
  {
319
- "epoch": 29.45043386799895,
320
- "grad_norm": 2.1626858711242676,
321
- "learning_rate": 7.335833333333335e-05,
322
- "loss": 3.7927,
323
  "step": 56000
324
  },
325
  {
326
- "epoch": 29.45043386799895,
327
- "eval_loss": 4.237761974334717,
328
- "eval_runtime": 2.3384,
329
- "eval_samples_per_second": 1362.919,
330
- "eval_steps_per_second": 85.53,
331
  "step": 56000
332
  },
333
  {
334
- "epoch": 30.50223507757034,
335
- "eval_loss": 4.237727165222168,
336
- "eval_runtime": 2.2527,
337
- "eval_samples_per_second": 1414.746,
338
- "eval_steps_per_second": 88.782,
339
  "step": 58000
340
  },
341
  {
342
- "epoch": 31.55403628714173,
343
- "grad_norm": 2.1217854022979736,
344
- "learning_rate": 6.669333333333334e-05,
345
- "loss": 3.7367,
346
  "step": 60000
347
  },
348
  {
349
- "epoch": 31.55403628714173,
350
- "eval_loss": 4.233393669128418,
351
- "eval_runtime": 2.2874,
352
- "eval_samples_per_second": 1393.302,
353
- "eval_steps_per_second": 87.437,
354
  "step": 60000
355
  },
356
  {
357
- "epoch": 32.60583749671312,
358
- "eval_loss": 4.229984283447266,
359
- "eval_runtime": 2.3158,
360
- "eval_samples_per_second": 1376.171,
361
- "eval_steps_per_second": 86.362,
362
  "step": 62000
363
  },
364
  {
365
- "epoch": 33.657638706284516,
366
- "grad_norm": 2.3173110485076904,
367
- "learning_rate": 6.003e-05,
368
- "loss": 3.6869,
369
  "step": 64000
370
  },
371
  {
372
- "epoch": 33.657638706284516,
373
- "eval_loss": 4.226880073547363,
374
- "eval_runtime": 2.3034,
375
- "eval_samples_per_second": 1383.631,
376
- "eval_steps_per_second": 86.83,
377
  "step": 64000
378
  },
379
  {
380
- "epoch": 34.7094399158559,
381
- "eval_loss": 4.224526405334473,
382
- "eval_runtime": 2.2795,
383
- "eval_samples_per_second": 1398.138,
384
- "eval_steps_per_second": 87.74,
385
  "step": 66000
386
  },
387
  {
388
- "epoch": 35.7612411254273,
389
- "grad_norm": 2.3296120166778564,
390
- "learning_rate": 5.336333333333333e-05,
391
- "loss": 3.6416,
392
  "step": 68000
393
  },
394
  {
395
- "epoch": 35.7612411254273,
396
- "eval_loss": 4.225712299346924,
397
- "eval_runtime": 2.2719,
398
- "eval_samples_per_second": 1402.762,
399
- "eval_steps_per_second": 88.03,
400
  "step": 68000
401
  },
402
  {
403
- "epoch": 36.813042334998684,
404
- "eval_loss": 4.220585346221924,
405
- "eval_runtime": 2.2631,
406
- "eval_samples_per_second": 1408.243,
407
- "eval_steps_per_second": 88.374,
408
  "step": 70000
409
  },
410
  {
411
- "epoch": 37.86484354457008,
412
- "grad_norm": 2.309408187866211,
413
- "learning_rate": 4.669833333333333e-05,
414
- "loss": 3.6032,
415
  "step": 72000
416
  },
417
  {
418
- "epoch": 37.86484354457008,
419
- "eval_loss": 4.224654674530029,
420
- "eval_runtime": 2.272,
421
- "eval_samples_per_second": 1402.745,
422
- "eval_steps_per_second": 88.029,
423
  "step": 72000
424
  },
425
  {
426
- "epoch": 38.916644754141466,
427
- "eval_loss": 4.227168083190918,
428
- "eval_runtime": 2.2483,
429
- "eval_samples_per_second": 1417.498,
430
- "eval_steps_per_second": 88.955,
431
  "step": 74000
432
  },
433
  {
434
- "epoch": 39.96844596371286,
435
- "grad_norm": 2.425579309463501,
436
- "learning_rate": 4.0033333333333335e-05,
437
- "loss": 3.5669,
438
  "step": 76000
439
  },
440
  {
441
- "epoch": 39.96844596371286,
442
- "eval_loss": 4.226228713989258,
443
- "eval_runtime": 2.2145,
444
- "eval_samples_per_second": 1439.153,
445
- "eval_steps_per_second": 90.314,
446
  "step": 76000
447
  },
448
  {
449
- "epoch": 41.02024717328425,
450
- "eval_loss": 4.230894565582275,
451
- "eval_runtime": 2.3349,
452
- "eval_samples_per_second": 1364.932,
453
- "eval_steps_per_second": 85.656,
454
  "step": 78000
455
  },
456
  {
457
- "epoch": 42.07204838285564,
458
- "grad_norm": 2.4452333450317383,
459
- "learning_rate": 3.337e-05,
460
- "loss": 3.533,
461
  "step": 80000
462
  },
463
  {
464
- "epoch": 42.07204838285564,
465
- "eval_loss": 4.230439186096191,
466
- "eval_runtime": 2.2462,
467
- "eval_samples_per_second": 1418.848,
468
- "eval_steps_per_second": 89.04,
469
  "step": 80000
470
  },
471
  {
472
- "epoch": 43.12384959242703,
473
- "eval_loss": 4.2289557456970215,
474
- "eval_runtime": 2.404,
475
- "eval_samples_per_second": 1325.692,
476
- "eval_steps_per_second": 83.194,
477
  "step": 82000
478
  },
479
  {
480
- "epoch": 44.17565080199842,
481
- "grad_norm": 2.6046276092529297,
482
- "learning_rate": 2.6705000000000003e-05,
483
- "loss": 3.5033,
484
  "step": 84000
485
  },
486
  {
487
- "epoch": 44.17565080199842,
488
- "eval_loss": 4.233131408691406,
489
- "eval_runtime": 2.3346,
490
- "eval_samples_per_second": 1365.098,
491
- "eval_steps_per_second": 85.667,
492
  "step": 84000
493
  },
494
  {
495
- "epoch": 45.22745201156982,
496
- "eval_loss": 4.232224941253662,
497
- "eval_runtime": 2.2625,
498
- "eval_samples_per_second": 1408.62,
499
- "eval_steps_per_second": 88.398,
500
  "step": 86000
501
  },
502
  {
503
- "epoch": 46.279253221141204,
504
- "grad_norm": 2.675853967666626,
505
- "learning_rate": 2.004e-05,
506
- "loss": 3.4759,
507
  "step": 88000
508
  },
509
  {
510
- "epoch": 46.279253221141204,
511
- "eval_loss": 4.231053352355957,
512
- "eval_runtime": 2.273,
513
- "eval_samples_per_second": 1402.13,
514
- "eval_steps_per_second": 87.991,
515
  "step": 88000
516
  },
517
  {
518
- "epoch": 47.3310544307126,
519
- "eval_loss": 4.234147071838379,
520
- "eval_runtime": 2.4703,
521
- "eval_samples_per_second": 1290.135,
522
- "eval_steps_per_second": 80.962,
523
  "step": 90000
524
  },
525
  {
526
- "epoch": 48.382855640283985,
527
- "grad_norm": 2.581390142440796,
528
- "learning_rate": 1.3375000000000002e-05,
529
- "loss": 3.4528,
530
  "step": 92000
531
  },
532
  {
533
- "epoch": 48.382855640283985,
534
- "eval_loss": 4.23477840423584,
535
- "eval_runtime": 2.2586,
536
- "eval_samples_per_second": 1411.035,
537
- "eval_steps_per_second": 88.549,
538
  "step": 92000
539
  },
540
  {
541
- "epoch": 49.43465684985538,
542
- "eval_loss": 4.232587814331055,
543
- "eval_runtime": 2.2225,
544
- "eval_samples_per_second": 1433.971,
545
- "eval_steps_per_second": 89.989,
546
  "step": 94000
547
  },
548
  {
549
- "epoch": 50.48645805942677,
550
- "grad_norm": 2.7233216762542725,
551
- "learning_rate": 6.710000000000001e-06,
552
- "loss": 3.4319,
553
  "step": 96000
554
  },
555
  {
556
- "epoch": 50.48645805942677,
557
- "eval_loss": 4.2330780029296875,
558
- "eval_runtime": 2.2298,
559
- "eval_samples_per_second": 1429.262,
560
- "eval_steps_per_second": 89.693,
561
  "step": 96000
562
  },
563
  {
564
- "epoch": 51.53825926899816,
565
- "eval_loss": 4.232585906982422,
566
- "eval_runtime": 2.9048,
567
- "eval_samples_per_second": 1097.167,
568
- "eval_steps_per_second": 68.853,
569
  "step": 98000
570
  },
571
  {
572
- "epoch": 52.59006047856955,
573
- "grad_norm": 2.644059181213379,
574
- "learning_rate": 4.5e-08,
575
- "loss": 3.4136,
576
  "step": 100000
577
  },
578
  {
579
- "epoch": 52.59006047856955,
580
- "eval_loss": 4.232517242431641,
581
- "eval_runtime": 2.2133,
582
- "eval_samples_per_second": 1439.961,
583
- "eval_steps_per_second": 90.365,
584
  "step": 100000
585
  }
586
  ],
587
  "logging_steps": 4000,
588
  "max_steps": 100000,
589
  "num_input_tokens_seen": 0,
590
- "num_train_epochs": 53,
591
  "save_steps": 4000,
592
  "stateful_callbacks": {
593
  "TrainerControl": {
@@ -601,7 +601,7 @@
601
  "attributes": {}
602
  }
603
  },
604
- "total_flos": 2.5850315930440704e+16,
605
  "train_batch_size": 16,
606
  "trial_name": null,
607
  "trial_params": null
 
1
  {
2
+ "best_metric": 3.5649592876434326,
3
+ "best_model_checkpoint": "/home/p318482/babyLM_controlled/models_trained/fr_clm/wikipedia_30/checkpoint-36000",
4
+ "epoch": 107.35373054213633,
5
  "eval_steps": 2000,
6
  "global_step": 100000,
7
  "is_hyper_param_search": false,
 
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
+ "epoch": 2.1470746108427265,
13
+ "eval_loss": 7.138043403625488,
14
+ "eval_runtime": 0.8415,
15
+ "eval_samples_per_second": 1273.93,
16
+ "eval_steps_per_second": 79.621,
17
  "step": 2000
18
  },
19
  {
20
+ "epoch": 4.294149221685453,
21
+ "grad_norm": 1.455039381980896,
22
  "learning_rate": 1e-05,
23
+ "loss": 7.213,
24
  "step": 4000
25
  },
26
  {
27
+ "epoch": 4.294149221685453,
28
+ "eval_loss": 5.8541717529296875,
29
+ "eval_runtime": 0.7753,
30
+ "eval_samples_per_second": 1382.676,
31
+ "eval_steps_per_second": 86.417,
32
  "step": 4000
33
  },
34
  {
35
+ "epoch": 6.4412238325281805,
36
+ "eval_loss": 5.403579235076904,
37
+ "eval_runtime": 0.7542,
38
+ "eval_samples_per_second": 1421.437,
39
+ "eval_steps_per_second": 88.84,
40
  "step": 6000
41
  },
42
  {
43
+ "epoch": 8.588298443370906,
44
+ "grad_norm": 3.117489814758301,
45
+ "learning_rate": 1.9997500000000003e-05,
46
+ "loss": 5.4304,
47
  "step": 8000
48
  },
49
  {
50
+ "epoch": 8.588298443370906,
51
+ "eval_loss": 5.049880504608154,
52
+ "eval_runtime": 0.7597,
53
+ "eval_samples_per_second": 1411.127,
54
+ "eval_steps_per_second": 88.195,
55
  "step": 8000
56
  },
57
  {
58
+ "epoch": 10.735373054213634,
59
+ "eval_loss": 4.760603427886963,
60
+ "eval_runtime": 0.7624,
61
+ "eval_samples_per_second": 1406.053,
62
+ "eval_steps_per_second": 87.878,
63
  "step": 10000
64
  },
65
  {
66
+ "epoch": 12.882447665056361,
67
+ "grad_norm": 2.9511141777038574,
68
+ "learning_rate": 2.9995e-05,
69
+ "loss": 4.771,
70
  "step": 12000
71
  },
72
  {
73
+ "epoch": 12.882447665056361,
74
+ "eval_loss": 4.517208576202393,
75
+ "eval_runtime": 0.764,
76
+ "eval_samples_per_second": 1403.097,
77
+ "eval_steps_per_second": 87.694,
78
  "step": 12000
79
  },
80
  {
81
+ "epoch": 15.029522275899087,
82
+ "eval_loss": 4.3206024169921875,
83
+ "eval_runtime": 0.7595,
84
+ "eval_samples_per_second": 1411.435,
85
+ "eval_steps_per_second": 88.215,
86
  "step": 14000
87
  },
88
  {
89
+ "epoch": 17.176596886741816,
90
+ "grad_norm": 2.955515146255493,
91
+ "learning_rate": 3.999e-05,
92
+ "loss": 4.2888,
93
  "step": 16000
94
  },
95
  {
96
+ "epoch": 17.176596886741816,
97
+ "eval_loss": 4.152973175048828,
98
+ "eval_runtime": 0.8247,
99
+ "eval_samples_per_second": 1299.91,
100
+ "eval_steps_per_second": 81.244,
101
  "step": 16000
102
  },
103
  {
104
+ "epoch": 19.32367149758454,
105
+ "eval_loss": 4.015476703643799,
106
+ "eval_runtime": 0.7501,
107
+ "eval_samples_per_second": 1429.237,
108
+ "eval_steps_per_second": 89.327,
109
  "step": 18000
110
  },
111
  {
112
+ "epoch": 21.470746108427267,
113
+ "grad_norm": 3.0531320571899414,
114
+ "learning_rate": 4.99875e-05,
115
+ "loss": 3.9141,
116
  "step": 20000
117
  },
118
  {
119
+ "epoch": 21.470746108427267,
120
+ "eval_loss": 3.896575927734375,
121
+ "eval_runtime": 0.768,
122
+ "eval_samples_per_second": 1395.843,
123
+ "eval_steps_per_second": 87.24,
124
  "step": 20000
125
  },
126
  {
127
+ "epoch": 23.617820719269993,
128
+ "eval_loss": 3.8046905994415283,
129
+ "eval_runtime": 0.7601,
130
+ "eval_samples_per_second": 1410.272,
131
+ "eval_steps_per_second": 88.142,
132
  "step": 22000
133
  },
134
  {
135
+ "epoch": 25.764895330112722,
136
+ "grad_norm": 3.089989185333252,
137
+ "learning_rate": 5.998250000000001e-05,
138
+ "loss": 3.6154,
139
  "step": 24000
140
  },
141
  {
142
+ "epoch": 25.764895330112722,
143
+ "eval_loss": 3.7359092235565186,
144
+ "eval_runtime": 0.7778,
145
+ "eval_samples_per_second": 1378.238,
146
+ "eval_steps_per_second": 86.14,
147
  "step": 24000
148
  },
149
  {
150
+ "epoch": 27.911969940955448,
151
+ "eval_loss": 3.6784231662750244,
152
+ "eval_runtime": 0.7541,
153
+ "eval_samples_per_second": 1421.491,
154
+ "eval_steps_per_second": 88.843,
155
  "step": 26000
156
  },
157
  {
158
+ "epoch": 30.059044551798173,
159
+ "grad_norm": 3.3495922088623047,
160
+ "learning_rate": 6.997500000000001e-05,
161
+ "loss": 3.3661,
162
  "step": 28000
163
  },
164
  {
165
+ "epoch": 30.059044551798173,
166
+ "eval_loss": 3.6360318660736084,
167
+ "eval_runtime": 0.7603,
168
+ "eval_samples_per_second": 1409.913,
169
+ "eval_steps_per_second": 88.12,
170
  "step": 28000
171
  },
172
  {
173
+ "epoch": 32.2061191626409,
174
+ "eval_loss": 3.6019041538238525,
175
+ "eval_runtime": 0.7638,
176
+ "eval_samples_per_second": 1403.547,
177
+ "eval_steps_per_second": 87.722,
178
  "step": 30000
179
  },
180
  {
181
+ "epoch": 34.35319377348363,
182
+ "grad_norm": 3.1256051063537598,
183
+ "learning_rate": 7.997250000000001e-05,
184
+ "loss": 3.1473,
185
  "step": 32000
186
  },
187
  {
188
+ "epoch": 34.35319377348363,
189
+ "eval_loss": 3.5816054344177246,
190
+ "eval_runtime": 0.7628,
191
+ "eval_samples_per_second": 1405.391,
192
+ "eval_steps_per_second": 87.837,
193
  "step": 32000
194
  },
195
  {
196
+ "epoch": 36.500268384326354,
197
+ "eval_loss": 3.5698702335357666,
198
+ "eval_runtime": 0.7617,
199
+ "eval_samples_per_second": 1407.381,
200
+ "eval_steps_per_second": 87.961,
201
  "step": 34000
202
  },
203
  {
204
+ "epoch": 38.64734299516908,
205
+ "grad_norm": 2.921318769454956,
206
+ "learning_rate": 8.997000000000001e-05,
207
+ "loss": 2.9533,
208
  "step": 36000
209
  },
210
  {
211
+ "epoch": 38.64734299516908,
212
+ "eval_loss": 3.5649592876434326,
213
+ "eval_runtime": 0.7607,
214
+ "eval_samples_per_second": 1409.149,
215
+ "eval_steps_per_second": 88.072,
216
  "step": 36000
217
  },
218
  {
219
+ "epoch": 40.794417606011805,
220
+ "eval_loss": 3.5666539669036865,
221
+ "eval_runtime": 0.7558,
222
+ "eval_samples_per_second": 1418.297,
223
+ "eval_steps_per_second": 88.644,
224
  "step": 38000
225
  },
226
  {
227
+ "epoch": 42.941492216854535,
228
+ "grad_norm": 3.202237129211426,
229
+ "learning_rate": 9.996500000000001e-05,
230
+ "loss": 2.777,
231
  "step": 40000
232
  },
233
  {
234
+ "epoch": 42.941492216854535,
235
+ "eval_loss": 3.5746841430664062,
236
+ "eval_runtime": 0.7722,
237
+ "eval_samples_per_second": 1388.315,
238
+ "eval_steps_per_second": 86.77,
239
  "step": 40000
240
  },
241
  {
242
+ "epoch": 45.088566827697264,
243
+ "eval_loss": 3.5878281593322754,
244
+ "eval_runtime": 0.7639,
245
+ "eval_samples_per_second": 1403.298,
246
+ "eval_steps_per_second": 87.706,
247
  "step": 42000
248
  },
249
  {
250
+ "epoch": 47.235641438539986,
251
+ "grad_norm": 2.9902379512786865,
252
+ "learning_rate": 9.336e-05,
253
+ "loss": 2.6015,
254
  "step": 44000
255
  },
256
  {
257
+ "epoch": 47.235641438539986,
258
+ "eval_loss": 3.6107263565063477,
259
+ "eval_runtime": 0.7601,
260
+ "eval_samples_per_second": 1410.262,
261
+ "eval_steps_per_second": 88.141,
262
  "step": 44000
263
  },
264
  {
265
+ "epoch": 49.382716049382715,
266
+ "eval_loss": 3.6260793209075928,
267
+ "eval_runtime": 0.7605,
268
+ "eval_samples_per_second": 1409.681,
269
+ "eval_steps_per_second": 88.105,
270
  "step": 46000
271
  },
272
  {
273
+ "epoch": 51.529790660225444,
274
+ "grad_norm": 3.1626901626586914,
275
+ "learning_rate": 8.669833333333334e-05,
276
+ "loss": 2.4429,
277
  "step": 48000
278
  },
279
  {
280
+ "epoch": 51.529790660225444,
281
+ "eval_loss": 3.6414248943328857,
282
+ "eval_runtime": 0.7596,
283
+ "eval_samples_per_second": 1411.254,
284
+ "eval_steps_per_second": 88.203,
285
  "step": 48000
286
  },
287
  {
288
+ "epoch": 53.676865271068166,
289
+ "eval_loss": 3.6636786460876465,
290
+ "eval_runtime": 0.7637,
291
+ "eval_samples_per_second": 1403.705,
292
+ "eval_steps_per_second": 87.732,
293
  "step": 50000
294
  },
295
  {
296
+ "epoch": 55.823939881910896,
297
+ "grad_norm": 3.162830114364624,
298
+ "learning_rate": 8.003333333333333e-05,
299
+ "loss": 2.3125,
300
  "step": 52000
301
  },
302
  {
303
+ "epoch": 55.823939881910896,
304
+ "eval_loss": 3.6777544021606445,
305
+ "eval_runtime": 0.7597,
306
+ "eval_samples_per_second": 1411.137,
307
+ "eval_steps_per_second": 88.196,
308
  "step": 52000
309
  },
310
  {
311
+ "epoch": 57.971014492753625,
312
+ "eval_loss": 3.703284502029419,
313
+ "eval_runtime": 0.757,
314
+ "eval_samples_per_second": 1416.089,
315
+ "eval_steps_per_second": 88.506,
316
  "step": 54000
317
  },
318
  {
319
+ "epoch": 60.11808910359635,
320
+ "grad_norm": 3.4666566848754883,
321
+ "learning_rate": 7.337000000000001e-05,
322
+ "loss": 2.1989,
323
  "step": 56000
324
  },
325
  {
326
+ "epoch": 60.11808910359635,
327
+ "eval_loss": 3.7409818172454834,
328
+ "eval_runtime": 0.7476,
329
+ "eval_samples_per_second": 1433.993,
330
+ "eval_steps_per_second": 89.625,
331
  "step": 56000
332
  },
333
  {
334
+ "epoch": 62.265163714439076,
335
+ "eval_loss": 3.7754786014556885,
336
+ "eval_runtime": 0.7506,
337
+ "eval_samples_per_second": 1428.237,
338
+ "eval_steps_per_second": 89.265,
339
  "step": 58000
340
  },
341
  {
342
+ "epoch": 64.4122383252818,
343
+ "grad_norm": 3.3885388374328613,
344
+ "learning_rate": 6.670666666666668e-05,
345
+ "loss": 2.1044,
346
  "step": 60000
347
  },
348
  {
349
+ "epoch": 64.4122383252818,
350
+ "eval_loss": 3.7875912189483643,
351
+ "eval_runtime": 0.7543,
352
+ "eval_samples_per_second": 1421.139,
353
+ "eval_steps_per_second": 88.821,
354
  "step": 60000
355
  },
356
  {
357
+ "epoch": 66.55931293612453,
358
+ "eval_loss": 3.8081138134002686,
359
+ "eval_runtime": 0.7633,
360
+ "eval_samples_per_second": 1404.466,
361
+ "eval_steps_per_second": 87.779,
362
  "step": 62000
363
  },
364
  {
365
+ "epoch": 68.70638754696726,
366
+ "grad_norm": 3.763136148452759,
367
+ "learning_rate": 6.004333333333334e-05,
368
+ "loss": 2.0257,
369
  "step": 64000
370
  },
371
  {
372
+ "epoch": 68.70638754696726,
373
+ "eval_loss": 3.822213649749756,
374
+ "eval_runtime": 0.7598,
375
+ "eval_samples_per_second": 1410.912,
376
+ "eval_steps_per_second": 88.182,
377
  "step": 64000
378
  },
379
  {
380
+ "epoch": 70.85346215780999,
381
+ "eval_loss": 3.8411173820495605,
382
+ "eval_runtime": 0.7613,
383
+ "eval_samples_per_second": 1408.036,
384
+ "eval_steps_per_second": 88.002,
385
  "step": 66000
386
  },
387
  {
388
+ "epoch": 73.00053676865271,
389
+ "grad_norm": 4.504343509674072,
390
+ "learning_rate": 5.338000000000001e-05,
391
+ "loss": 1.9563,
392
  "step": 68000
393
  },
394
  {
395
+ "epoch": 73.00053676865271,
396
+ "eval_loss": 3.848768949508667,
397
+ "eval_runtime": 0.7688,
398
+ "eval_samples_per_second": 1394.369,
399
+ "eval_steps_per_second": 87.148,
400
  "step": 68000
401
  },
402
  {
403
+ "epoch": 75.14761137949544,
404
+ "eval_loss": 3.8914690017700195,
405
+ "eval_runtime": 0.7542,
406
+ "eval_samples_per_second": 1421.304,
407
+ "eval_steps_per_second": 88.831,
408
  "step": 70000
409
  },
410
  {
411
+ "epoch": 77.29468599033817,
412
+ "grad_norm": 3.6598613262176514,
413
+ "learning_rate": 4.671666666666667e-05,
414
+ "loss": 1.8905,
415
  "step": 72000
416
  },
417
  {
418
+ "epoch": 77.29468599033817,
419
+ "eval_loss": 3.9079344272613525,
420
+ "eval_runtime": 0.7509,
421
+ "eval_samples_per_second": 1427.665,
422
+ "eval_steps_per_second": 89.229,
423
  "step": 72000
424
  },
425
  {
426
+ "epoch": 79.44176060118089,
427
+ "eval_loss": 3.9169275760650635,
428
+ "eval_runtime": 0.7546,
429
+ "eval_samples_per_second": 1420.536,
430
+ "eval_steps_per_second": 88.783,
431
  "step": 74000
432
  },
433
  {
434
+ "epoch": 81.58883521202361,
435
+ "grad_norm": 3.5242605209350586,
436
+ "learning_rate": 4.005333333333334e-05,
437
+ "loss": 1.836,
438
  "step": 76000
439
  },
440
  {
441
+ "epoch": 81.58883521202361,
442
+ "eval_loss": 3.9381563663482666,
443
+ "eval_runtime": 0.7535,
444
+ "eval_samples_per_second": 1422.783,
445
+ "eval_steps_per_second": 88.924,
446
  "step": 76000
447
  },
448
  {
449
+ "epoch": 83.73590982286635,
450
+ "eval_loss": 3.9429502487182617,
451
+ "eval_runtime": 0.7517,
452
+ "eval_samples_per_second": 1426.084,
453
+ "eval_steps_per_second": 89.13,
454
  "step": 78000
455
  },
456
  {
457
+ "epoch": 85.88298443370907,
458
+ "grad_norm": 3.842487096786499,
459
+ "learning_rate": 3.339e-05,
460
+ "loss": 1.7885,
461
  "step": 80000
462
  },
463
  {
464
+ "epoch": 85.88298443370907,
465
+ "eval_loss": 3.9470765590667725,
466
+ "eval_runtime": 0.7546,
467
+ "eval_samples_per_second": 1420.557,
468
+ "eval_steps_per_second": 88.785,
469
  "step": 80000
470
  },
471
  {
472
+ "epoch": 88.03005904455179,
473
+ "eval_loss": 3.9667794704437256,
474
+ "eval_runtime": 0.7733,
475
+ "eval_samples_per_second": 1386.292,
476
+ "eval_steps_per_second": 86.643,
477
  "step": 82000
478
  },
479
  {
480
+ "epoch": 90.17713365539453,
481
+ "grad_norm": 3.9361636638641357,
482
+ "learning_rate": 2.6726666666666666e-05,
483
+ "loss": 1.7431,
484
  "step": 84000
485
  },
486
  {
487
+ "epoch": 90.17713365539453,
488
+ "eval_loss": 3.986008882522583,
489
+ "eval_runtime": 0.7467,
490
+ "eval_samples_per_second": 1435.654,
491
+ "eval_steps_per_second": 89.728,
492
  "step": 84000
493
  },
494
  {
495
+ "epoch": 92.32420826623725,
496
+ "eval_loss": 4.008753299713135,
497
+ "eval_runtime": 0.7576,
498
+ "eval_samples_per_second": 1415.003,
499
+ "eval_steps_per_second": 88.438,
500
  "step": 86000
501
  },
502
  {
503
+ "epoch": 94.47128287707997,
504
+ "grad_norm": 4.01881742477417,
505
+ "learning_rate": 2.0063333333333335e-05,
506
+ "loss": 1.7024,
507
  "step": 88000
508
  },
509
  {
510
+ "epoch": 94.47128287707997,
511
+ "eval_loss": 4.013205528259277,
512
+ "eval_runtime": 0.7488,
513
+ "eval_samples_per_second": 1431.63,
514
+ "eval_steps_per_second": 89.477,
515
  "step": 88000
516
  },
517
  {
518
+ "epoch": 96.61835748792271,
519
+ "eval_loss": 4.0260210037231445,
520
+ "eval_runtime": 0.7603,
521
+ "eval_samples_per_second": 1410.015,
522
+ "eval_steps_per_second": 88.126,
523
  "step": 90000
524
  },
525
  {
526
+ "epoch": 98.76543209876543,
527
+ "grad_norm": 3.9451382160186768,
528
+ "learning_rate": 1.3400000000000002e-05,
529
+ "loss": 1.6687,
530
  "step": 92000
531
  },
532
  {
533
+ "epoch": 98.76543209876543,
534
+ "eval_loss": 4.035834789276123,
535
+ "eval_runtime": 0.7619,
536
+ "eval_samples_per_second": 1406.935,
537
+ "eval_steps_per_second": 87.933,
538
  "step": 92000
539
  },
540
  {
541
+ "epoch": 100.91250670960815,
542
+ "eval_loss": 4.029040813446045,
543
+ "eval_runtime": 0.7506,
544
+ "eval_samples_per_second": 1428.162,
545
+ "eval_steps_per_second": 89.26,
546
  "step": 94000
547
  },
548
  {
549
+ "epoch": 103.05958132045089,
550
+ "grad_norm": 4.011170387268066,
551
+ "learning_rate": 6.736666666666668e-06,
552
+ "loss": 1.6369,
553
  "step": 96000
554
  },
555
  {
556
+ "epoch": 103.05958132045089,
557
+ "eval_loss": 4.04220724105835,
558
+ "eval_runtime": 0.7565,
559
+ "eval_samples_per_second": 1417.059,
560
+ "eval_steps_per_second": 88.566,
561
  "step": 96000
562
  },
563
  {
564
+ "epoch": 105.20665593129361,
565
+ "eval_loss": 4.044515609741211,
566
+ "eval_runtime": 0.7518,
567
+ "eval_samples_per_second": 1425.874,
568
+ "eval_steps_per_second": 89.117,
569
  "step": 98000
570
  },
571
  {
572
+ "epoch": 107.35373054213633,
573
+ "grad_norm": 3.8686037063598633,
574
+ "learning_rate": 7.333333333333334e-08,
575
+ "loss": 1.6129,
576
  "step": 100000
577
  },
578
  {
579
+ "epoch": 107.35373054213633,
580
+ "eval_loss": 4.045902252197266,
581
+ "eval_runtime": 0.7523,
582
+ "eval_samples_per_second": 1424.957,
583
+ "eval_steps_per_second": 89.06,
584
  "step": 100000
585
  }
586
  ],
587
  "logging_steps": 4000,
588
  "max_steps": 100000,
589
  "num_input_tokens_seen": 0,
590
+ "num_train_epochs": 108,
591
  "save_steps": 4000,
592
  "stateful_callbacks": {
593
  "TrainerControl": {
 
601
  "attributes": {}
602
  }
603
  },
604
+ "total_flos": 2.5849709221380096e+16,
605
  "train_batch_size": 16,
606
  "trial_name": null,
607
  "trial_params": null
checkpoint-100000/training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ab68434ec7156d3a63e4783f824871a48d62c7c2fdcc831fdd5471c5f4aec7fe
3
  size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:26102d0ac750ce531ebda9d7577ccea6c8ecbc93b04bdff23226e376dd5609a2
3
  size 5368