froabera commited on
Commit
0cf58dc
·
verified ·
1 Parent(s): c337b64

Training in progress, step 5500, checkpoint

Browse files
checkpoint-5500/config.json CHANGED
@@ -84,7 +84,7 @@
84
  "speech_decoder_prenet_layers": 2,
85
  "speech_decoder_prenet_units": 256,
86
  "tie_word_embeddings": true,
87
- "transformers_version": "5.19.0.dev0",
88
  "use_cache": false,
89
  "use_guided_attention_loss": false,
90
  "vocab_size": 81
 
84
  "speech_decoder_prenet_layers": 2,
85
  "speech_decoder_prenet_units": 256,
86
  "tie_word_embeddings": true,
87
+ "transformers_version": "5.19.0",
88
  "use_cache": false,
89
  "use_guided_attention_loss": false,
90
  "vocab_size": 81
checkpoint-5500/generation_config.json CHANGED
@@ -6,6 +6,6 @@
6
  "output_attentions": false,
7
  "output_hidden_states": false,
8
  "pad_token_id": 1,
9
- "transformers_version": "5.19.0.dev0",
10
  "use_cache": true
11
  }
 
6
  "output_attentions": false,
7
  "output_hidden_states": false,
8
  "pad_token_id": 1,
9
+ "transformers_version": "5.19.0",
10
  "use_cache": true
11
  }
checkpoint-5500/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:172f85b6a2162e3b50693d064554cd79eed1a8faa36a633cfe07b28dee0c9627
3
  size 577789320
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34d5f2229c3c8779494e770c600d92f26cc38d2a8d8f7499e736e68df48f57fb
3
  size 577789320
checkpoint-5500/optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a807d27c01c0ca34d325e03ac6d25aeeda46e2dc8285672f47d03262fa4da827
3
  size 1155772634
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56676d7852056ada59ae1edf0d1f175789a61f2c304f81557577eabd51b72002
3
  size 1155772634
checkpoint-5500/rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b070c91256cbbb68d3fe1b7d065d27aef1b0912648afe5b0b52669e4e1a3ba42
3
  size 14645
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43ebefdb3f1f49b9dad2350c88c3260e2ca40535907434a9dec38f797589200a
3
  size 14645
checkpoint-5500/scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c3b3cd3371d78986477c4511279a91f3289adfbebdf82f926b83c9c0c7803fe9
3
  size 1465
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a967478652b012ce6d3273fa34dc7816afa3c93ccb2006280d6792ace12c6fe8
3
  size 1465
checkpoint-5500/trainer_state.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "best_global_step": 5500,
3
- "best_metric": 0.8907444477081299,
4
  "best_model_checkpoint": "/kaggle/working/tts_processed/checkpoint-5500",
5
- "epoch": 27.227863777089784,
6
  "eval_steps": 500,
7
  "global_step": 5500,
8
  "is_hyper_param_search": false,
@@ -10,1631 +10,1631 @@
10
  "is_world_process_zero": true,
11
  "log_history": [
12
  {
13
- "epoch": 0.1238390092879257,
14
- "grad_norm": 829.4996337890625,
15
  "learning_rate": 4.8e-08,
16
- "loss": 21.73699951171875,
17
  "step": 25
18
  },
19
  {
20
- "epoch": 0.2476780185758514,
21
- "grad_norm": 701.2330932617188,
22
  "learning_rate": 9.8e-08,
23
- "loss": 21.00220947265625,
24
  "step": 50
25
  },
26
  {
27
- "epoch": 0.3715170278637771,
28
- "grad_norm": 631.0809326171875,
29
  "learning_rate": 1.4799999999999998e-07,
30
- "loss": 19.772623291015623,
31
  "step": 75
32
  },
33
  {
34
- "epoch": 0.4953560371517028,
35
- "grad_norm": 556.35791015625,
36
  "learning_rate": 1.98e-07,
37
- "loss": 18.46118896484375,
38
  "step": 100
39
  },
40
  {
41
- "epoch": 0.6191950464396285,
42
- "grad_norm": 536.2819213867188,
43
  "learning_rate": 2.48e-07,
44
- "loss": 17.51901611328125,
45
  "step": 125
46
  },
47
  {
48
- "epoch": 0.7430340557275542,
49
- "grad_norm": 459.290283203125,
50
  "learning_rate": 2.98e-07,
51
- "loss": 16.43676513671875,
52
  "step": 150
53
  },
54
  {
55
- "epoch": 0.8668730650154799,
56
- "grad_norm": 230.56405639648438,
57
  "learning_rate": 3.4799999999999994e-07,
58
- "loss": 15.75008056640625,
59
  "step": 175
60
  },
61
  {
62
- "epoch": 0.9907120743034056,
63
- "grad_norm": 233.90635681152344,
64
  "learning_rate": 3.98e-07,
65
- "loss": 14.56845947265625,
66
  "step": 200
67
  },
68
  {
69
- "epoch": 1.1139318885448917,
70
- "grad_norm": 107.61080169677734,
71
  "learning_rate": 4.48e-07,
72
- "loss": 13.99685546875,
73
  "step": 225
74
  },
75
  {
76
- "epoch": 1.2377708978328172,
77
- "grad_norm": 115.48291015625,
78
  "learning_rate": 4.979999999999999e-07,
79
- "loss": 13.390751953125,
80
  "step": 250
81
  },
82
  {
83
- "epoch": 1.361609907120743,
84
- "grad_norm": 150.88845825195312,
85
  "learning_rate": 5.48e-07,
86
- "loss": 12.885958251953125,
87
  "step": 275
88
  },
89
  {
90
- "epoch": 1.4854489164086688,
91
- "grad_norm": 102.25688934326172,
92
  "learning_rate": 5.979999999999999e-07,
93
- "loss": 12.281358642578125,
94
  "step": 300
95
  },
96
  {
97
- "epoch": 1.6092879256965944,
98
- "grad_norm": 91.96704864501953,
99
  "learning_rate": 6.48e-07,
100
- "loss": 12.51485107421875,
101
  "step": 325
102
  },
103
  {
104
- "epoch": 1.73312693498452,
105
- "grad_norm": 89.19316864013672,
106
  "learning_rate": 6.979999999999999e-07,
107
- "loss": 11.758416748046875,
108
  "step": 350
109
  },
110
  {
111
- "epoch": 1.8569659442724458,
112
- "grad_norm": 43.06995391845703,
113
  "learning_rate": 7.48e-07,
114
- "loss": 11.7668603515625,
115
  "step": 375
116
  },
117
  {
118
- "epoch": 1.9808049535603716,
119
- "grad_norm": 37.787078857421875,
120
  "learning_rate": 7.98e-07,
121
- "loss": 11.696016845703125,
122
  "step": 400
123
  },
124
  {
125
- "epoch": 2.1040247678018575,
126
- "grad_norm": 52.63007354736328,
127
  "learning_rate": 8.48e-07,
128
- "loss": 11.60255859375,
129
  "step": 425
130
  },
131
  {
132
- "epoch": 2.2278637770897833,
133
- "grad_norm": 37.73246383666992,
134
  "learning_rate": 8.98e-07,
135
- "loss": 11.121099853515625,
136
  "step": 450
137
  },
138
  {
139
- "epoch": 2.351702786377709,
140
- "grad_norm": 50.654396057128906,
141
  "learning_rate": 9.479999999999999e-07,
142
- "loss": 10.981490478515624,
143
  "step": 475
144
  },
145
  {
146
- "epoch": 2.4755417956656345,
147
- "grad_norm": 34.40877151489258,
148
  "learning_rate": 9.98e-07,
149
- "loss": 10.89916015625,
150
  "step": 500
151
  },
152
  {
153
- "epoch": 2.4755417956656345,
154
- "eval_loss": 1.2403377294540405,
155
- "eval_runtime": 78.2214,
156
- "eval_samples_per_second": 20.659,
157
- "eval_steps_per_second": 5.165,
158
  "step": 500
159
  },
160
  {
161
- "epoch": 2.5993808049535603,
162
- "grad_norm": 32.53867721557617,
163
- "learning_rate": 9.986285714285714e-07,
164
- "loss": 10.70981201171875,
165
  "step": 525
166
  },
167
  {
168
- "epoch": 2.723219814241486,
169
- "grad_norm": 37.28243637084961,
170
- "learning_rate": 9.972e-07,
171
- "loss": 10.68586669921875,
172
  "step": 550
173
  },
174
  {
175
- "epoch": 2.847058823529412,
176
- "grad_norm": 33.349178314208984,
177
- "learning_rate": 9.957714285714285e-07,
178
- "loss": 10.5280859375,
179
  "step": 575
180
  },
181
  {
182
- "epoch": 2.9708978328173377,
183
- "grad_norm": 40.658817291259766,
184
- "learning_rate": 9.94342857142857e-07,
185
- "loss": 10.160640258789062,
186
  "step": 600
187
  },
188
  {
189
- "epoch": 3.0941176470588236,
190
- "grad_norm": 35.320701599121094,
191
- "learning_rate": 9.929142857142856e-07,
192
- "loss": 10.215652465820312,
193
  "step": 625
194
  },
195
  {
196
- "epoch": 3.2179566563467494,
197
- "grad_norm": 31.319137573242188,
198
- "learning_rate": 9.914857142857142e-07,
199
- "loss": 10.158941650390625,
200
  "step": 650
201
  },
202
  {
203
- "epoch": 3.341795665634675,
204
- "grad_norm": 28.438013076782227,
205
- "learning_rate": 9.900571428571427e-07,
206
- "loss": 9.933291625976562,
207
  "step": 675
208
  },
209
  {
210
- "epoch": 3.4656346749226006,
211
- "grad_norm": 41.65836715698242,
212
- "learning_rate": 9.886285714285713e-07,
213
- "loss": 9.679817504882813,
214
  "step": 700
215
  },
216
  {
217
- "epoch": 3.5894736842105264,
218
- "grad_norm": 49.5134162902832,
219
- "learning_rate": 9.871999999999998e-07,
220
- "loss": 9.66293212890625,
221
  "step": 725
222
  },
223
  {
224
- "epoch": 3.713312693498452,
225
- "grad_norm": 37.431331634521484,
226
- "learning_rate": 9.857714285714284e-07,
227
- "loss": 9.642633056640625,
228
  "step": 750
229
  },
230
  {
231
- "epoch": 3.837151702786378,
232
- "grad_norm": 39.974544525146484,
233
- "learning_rate": 9.843428571428572e-07,
234
- "loss": 9.296731567382812,
235
  "step": 775
236
  },
237
  {
238
- "epoch": 3.9609907120743033,
239
- "grad_norm": 39.033599853515625,
240
- "learning_rate": 9.829142857142857e-07,
241
- "loss": 9.472877807617188,
242
  "step": 800
243
  },
244
  {
245
- "epoch": 4.08421052631579,
246
- "grad_norm": 29.367778778076172,
247
- "learning_rate": 9.814857142857143e-07,
248
- "loss": 9.286629028320313,
249
  "step": 825
250
  },
251
  {
252
- "epoch": 4.208049535603715,
253
- "grad_norm": 30.283451080322266,
254
- "learning_rate": 9.800571428571428e-07,
255
- "loss": 9.21810546875,
256
  "step": 850
257
  },
258
  {
259
- "epoch": 4.33188854489164,
260
- "grad_norm": 43.98243713378906,
261
- "learning_rate": 9.786285714285714e-07,
262
- "loss": 9.4026904296875,
263
  "step": 875
264
  },
265
  {
266
- "epoch": 4.455727554179567,
267
- "grad_norm": 35.94754409790039,
268
- "learning_rate": 9.772e-07,
269
- "loss": 9.243385009765625,
270
  "step": 900
271
  },
272
  {
273
- "epoch": 4.579566563467492,
274
- "grad_norm": 97.34254455566406,
275
- "learning_rate": 9.757714285714285e-07,
276
- "loss": 9.081926879882813,
277
  "step": 925
278
  },
279
  {
280
- "epoch": 4.703405572755418,
281
- "grad_norm": 35.22529602050781,
282
- "learning_rate": 9.74342857142857e-07,
283
- "loss": 9.177518310546875,
284
  "step": 950
285
  },
286
  {
287
- "epoch": 4.827244582043344,
288
- "grad_norm": 31.761302947998047,
289
- "learning_rate": 9.729142857142856e-07,
290
- "loss": 9.266145629882812,
291
  "step": 975
292
  },
293
  {
294
- "epoch": 4.951083591331269,
295
- "grad_norm": 34.29395294189453,
296
- "learning_rate": 9.714857142857142e-07,
297
- "loss": 9.146566162109375,
298
  "step": 1000
299
  },
300
  {
301
- "epoch": 4.951083591331269,
302
- "eval_loss": 1.01394784450531,
303
- "eval_runtime": 78.015,
304
- "eval_samples_per_second": 20.714,
305
- "eval_steps_per_second": 5.178,
306
  "step": 1000
307
  },
308
  {
309
- "epoch": 5.074303405572755,
310
- "grad_norm": 28.847089767456055,
311
- "learning_rate": 9.700571428571427e-07,
312
- "loss": 9.07594970703125,
313
  "step": 1025
314
  },
315
  {
316
- "epoch": 5.198142414860681,
317
- "grad_norm": 31.509410858154297,
318
- "learning_rate": 9.686285714285713e-07,
319
- "loss": 9.165030517578124,
320
  "step": 1050
321
  },
322
  {
323
- "epoch": 5.321981424148607,
324
- "grad_norm": 28.566282272338867,
325
- "learning_rate": 9.671999999999998e-07,
326
- "loss": 8.919168701171875,
327
  "step": 1075
328
  },
329
  {
330
- "epoch": 5.445820433436532,
331
- "grad_norm": 26.087688446044922,
332
- "learning_rate": 9.657714285714286e-07,
333
- "loss": 8.877406005859376,
334
  "step": 1100
335
  },
336
  {
337
- "epoch": 5.569659442724459,
338
- "grad_norm": 27.453046798706055,
339
- "learning_rate": 9.643428571428571e-07,
340
- "loss": 8.87409912109375,
341
  "step": 1125
342
  },
343
  {
344
- "epoch": 5.693498452012384,
345
- "grad_norm": 26.87615203857422,
346
- "learning_rate": 9.629142857142857e-07,
347
- "loss": 8.98948486328125,
348
  "step": 1150
349
  },
350
  {
351
- "epoch": 5.817337461300309,
352
- "grad_norm": 28.69729232788086,
353
- "learning_rate": 9.614857142857143e-07,
354
- "loss": 8.854622192382813,
355
  "step": 1175
356
  },
357
  {
358
- "epoch": 5.9411764705882355,
359
- "grad_norm": 31.937828063964844,
360
- "learning_rate": 9.600571428571428e-07,
361
- "loss": 9.01401123046875,
362
  "step": 1200
363
  },
364
  {
365
- "epoch": 6.064396284829721,
366
- "grad_norm": 42.471954345703125,
367
- "learning_rate": 9.586285714285714e-07,
368
- "loss": 8.731229858398438,
369
  "step": 1225
370
  },
371
  {
372
- "epoch": 6.188235294117647,
373
- "grad_norm": 23.436538696289062,
374
- "learning_rate": 9.572e-07,
375
- "loss": 8.83049560546875,
376
  "step": 1250
377
  },
378
  {
379
- "epoch": 6.312074303405573,
380
- "grad_norm": 36.96808624267578,
381
- "learning_rate": 9.557714285714285e-07,
382
- "loss": 8.846947631835938,
383
  "step": 1275
384
  },
385
  {
386
- "epoch": 6.435913312693499,
387
- "grad_norm": 34.232627868652344,
388
- "learning_rate": 9.54342857142857e-07,
389
- "loss": 8.8575048828125,
390
  "step": 1300
391
  },
392
  {
393
- "epoch": 6.559752321981424,
394
- "grad_norm": 25.819372177124023,
395
- "learning_rate": 9.529142857142857e-07,
396
- "loss": 8.967839965820312,
397
  "step": 1325
398
  },
399
  {
400
- "epoch": 6.68359133126935,
401
- "grad_norm": 30.442367553710938,
402
- "learning_rate": 9.514857142857142e-07,
403
- "loss": 8.79695556640625,
404
  "step": 1350
405
  },
406
  {
407
- "epoch": 6.807430340557276,
408
- "grad_norm": 25.49448585510254,
409
- "learning_rate": 9.500571428571428e-07,
410
- "loss": 8.629199829101562,
411
  "step": 1375
412
  },
413
  {
414
- "epoch": 6.931269349845201,
415
- "grad_norm": 50.014862060546875,
416
- "learning_rate": 9.486285714285714e-07,
417
- "loss": 8.7688330078125,
418
  "step": 1400
419
  },
420
  {
421
- "epoch": 7.054489164086688,
422
- "grad_norm": 29.534305572509766,
423
- "learning_rate": 9.472e-07,
424
- "loss": 8.635714721679687,
425
  "step": 1425
426
  },
427
  {
428
- "epoch": 7.178328173374613,
429
- "grad_norm": 24.548370361328125,
430
- "learning_rate": 9.457714285714286e-07,
431
- "loss": 8.751875,
432
  "step": 1450
433
  },
434
  {
435
- "epoch": 7.302167182662538,
436
- "grad_norm": 27.02566146850586,
437
- "learning_rate": 9.443428571428571e-07,
438
- "loss": 8.664575805664063,
439
  "step": 1475
440
  },
441
  {
442
- "epoch": 7.4260061919504645,
443
- "grad_norm": 26.448801040649414,
444
- "learning_rate": 9.429142857142857e-07,
445
- "loss": 8.631251831054687,
446
  "step": 1500
447
  },
448
  {
449
- "epoch": 7.4260061919504645,
450
- "eval_loss": 0.9785507321357727,
451
- "eval_runtime": 78.788,
452
- "eval_samples_per_second": 20.511,
453
- "eval_steps_per_second": 5.128,
454
  "step": 1500
455
  },
456
  {
457
- "epoch": 7.54984520123839,
458
- "grad_norm": 26.66581916809082,
459
- "learning_rate": 9.414857142857142e-07,
460
- "loss": 8.60765869140625,
461
  "step": 1525
462
  },
463
  {
464
- "epoch": 7.673684210526316,
465
- "grad_norm": 24.334070205688477,
466
- "learning_rate": 9.400571428571428e-07,
467
- "loss": 8.694551391601562,
468
  "step": 1550
469
  },
470
  {
471
- "epoch": 7.7975232198142415,
472
- "grad_norm": 25.746564865112305,
473
- "learning_rate": 9.386285714285714e-07,
474
- "loss": 8.605336303710937,
475
  "step": 1575
476
  },
477
  {
478
- "epoch": 7.921362229102167,
479
- "grad_norm": 26.221471786499023,
480
- "learning_rate": 9.372e-07,
481
- "loss": 8.670101928710938,
482
  "step": 1600
483
  },
484
  {
485
- "epoch": 8.044582043343654,
486
- "grad_norm": 40.2961311340332,
487
- "learning_rate": 9.357714285714286e-07,
488
- "loss": 8.697719116210937,
489
  "step": 1625
490
  },
491
  {
492
- "epoch": 8.16842105263158,
493
- "grad_norm": 29.41912078857422,
494
- "learning_rate": 9.343428571428571e-07,
495
- "loss": 8.6234228515625,
496
  "step": 1650
497
  },
498
  {
499
- "epoch": 8.292260061919505,
500
- "grad_norm": 23.59851837158203,
501
- "learning_rate": 9.329142857142857e-07,
502
- "loss": 8.555197143554688,
503
  "step": 1675
504
  },
505
  {
506
- "epoch": 8.41609907120743,
507
- "grad_norm": 23.335365295410156,
508
- "learning_rate": 9.314857142857142e-07,
509
- "loss": 8.533560180664063,
510
  "step": 1700
511
  },
512
  {
513
- "epoch": 8.539938080495356,
514
- "grad_norm": 19.63068389892578,
515
- "learning_rate": 9.300571428571428e-07,
516
- "loss": 8.534976196289062,
517
  "step": 1725
518
  },
519
  {
520
- "epoch": 8.66377708978328,
521
- "grad_norm": 26.558069229125977,
522
- "learning_rate": 9.286285714285714e-07,
523
- "loss": 8.529254150390624,
524
  "step": 1750
525
  },
526
  {
527
- "epoch": 8.787616099071208,
528
- "grad_norm": 28.553796768188477,
529
- "learning_rate": 9.272e-07,
530
- "loss": 8.4541943359375,
531
  "step": 1775
532
  },
533
  {
534
- "epoch": 8.911455108359133,
535
- "grad_norm": 26.726491928100586,
536
- "learning_rate": 9.257714285714285e-07,
537
- "loss": 8.505891723632812,
538
  "step": 1800
539
  },
540
  {
541
- "epoch": 9.034674922600619,
542
- "grad_norm": 33.665992736816406,
543
- "learning_rate": 9.243428571428571e-07,
544
- "loss": 8.420493774414062,
545
  "step": 1825
546
  },
547
  {
548
- "epoch": 9.158513931888544,
549
- "grad_norm": 26.62388801574707,
550
- "learning_rate": 9.229142857142857e-07,
551
- "loss": 8.4884375,
552
  "step": 1850
553
  },
554
  {
555
- "epoch": 9.282352941176471,
556
- "grad_norm": 21.369853973388672,
557
- "learning_rate": 9.214857142857142e-07,
558
- "loss": 8.413807983398437,
559
  "step": 1875
560
  },
561
  {
562
- "epoch": 9.406191950464397,
563
- "grad_norm": 23.232192993164062,
564
- "learning_rate": 9.200571428571429e-07,
565
- "loss": 8.473324584960938,
566
  "step": 1900
567
  },
568
  {
569
- "epoch": 9.530030959752322,
570
- "grad_norm": 48.790531158447266,
571
- "learning_rate": 9.186285714285714e-07,
572
- "loss": 8.452188720703125,
573
  "step": 1925
574
  },
575
  {
576
- "epoch": 9.653869969040247,
577
- "grad_norm": 23.798295974731445,
578
- "learning_rate": 9.172e-07,
579
- "loss": 8.495001831054687,
580
  "step": 1950
581
  },
582
  {
583
- "epoch": 9.777708978328173,
584
- "grad_norm": 33.404537200927734,
585
- "learning_rate": 9.157714285714285e-07,
586
- "loss": 8.58928955078125,
587
  "step": 1975
588
  },
589
  {
590
- "epoch": 9.9015479876161,
591
- "grad_norm": 30.602357864379883,
592
- "learning_rate": 9.143428571428571e-07,
593
- "loss": 8.496845703125,
594
  "step": 2000
595
  },
596
  {
597
- "epoch": 9.9015479876161,
598
- "eval_loss": 0.9522278904914856,
599
- "eval_runtime": 78.3957,
600
- "eval_samples_per_second": 20.613,
601
- "eval_steps_per_second": 5.153,
602
  "step": 2000
603
  },
604
  {
605
- "epoch": 10.024767801857585,
606
- "grad_norm": 28.26131248474121,
607
- "learning_rate": 9.129142857142856e-07,
608
- "loss": 8.348336181640626,
609
  "step": 2025
610
  },
611
  {
612
- "epoch": 10.14860681114551,
613
- "grad_norm": 27.35468864440918,
614
- "learning_rate": 9.114857142857142e-07,
615
- "loss": 8.510921020507812,
616
  "step": 2050
617
  },
618
  {
619
- "epoch": 10.272445820433436,
620
- "grad_norm": 21.89419174194336,
621
- "learning_rate": 9.100571428571429e-07,
622
- "loss": 8.4816162109375,
623
  "step": 2075
624
  },
625
  {
626
- "epoch": 10.396284829721361,
627
- "grad_norm": 24.475194931030273,
628
- "learning_rate": 9.086285714285714e-07,
629
- "loss": 8.524483032226563,
630
  "step": 2100
631
  },
632
  {
633
- "epoch": 10.520123839009289,
634
- "grad_norm": 28.319612503051758,
635
- "learning_rate": 9.072e-07,
636
- "loss": 8.434509887695313,
637
  "step": 2125
638
  },
639
  {
640
- "epoch": 10.643962848297214,
641
- "grad_norm": 22.886144638061523,
642
- "learning_rate": 9.057714285714285e-07,
643
- "loss": 8.604608764648438,
644
  "step": 2150
645
  },
646
  {
647
- "epoch": 10.76780185758514,
648
- "grad_norm": 31.52608871459961,
649
- "learning_rate": 9.043428571428571e-07,
650
- "loss": 8.302666015625,
651
  "step": 2175
652
  },
653
  {
654
- "epoch": 10.891640866873065,
655
- "grad_norm": 23.449596405029297,
656
- "learning_rate": 9.029142857142856e-07,
657
- "loss": 8.249719848632813,
658
  "step": 2200
659
  },
660
  {
661
- "epoch": 11.014860681114552,
662
- "grad_norm": 26.614627838134766,
663
- "learning_rate": 9.014857142857143e-07,
664
- "loss": 8.312252807617188,
665
  "step": 2225
666
  },
667
  {
668
- "epoch": 11.138699690402477,
669
- "grad_norm": 25.212867736816406,
670
- "learning_rate": 9.000571428571429e-07,
671
- "loss": 8.25859130859375,
672
  "step": 2250
673
  },
674
  {
675
- "epoch": 11.262538699690403,
676
- "grad_norm": 27.4830265045166,
677
- "learning_rate": 8.986285714285714e-07,
678
- "loss": 8.3023095703125,
679
  "step": 2275
680
  },
681
  {
682
- "epoch": 11.386377708978328,
683
- "grad_norm": 23.062416076660156,
684
- "learning_rate": 8.972e-07,
685
- "loss": 8.250548095703126,
686
  "step": 2300
687
  },
688
  {
689
- "epoch": 11.510216718266253,
690
- "grad_norm": 26.304353713989258,
691
- "learning_rate": 8.957714285714285e-07,
692
- "loss": 8.221282958984375,
693
  "step": 2325
694
  },
695
  {
696
- "epoch": 11.63405572755418,
697
- "grad_norm": 30.5177001953125,
698
- "learning_rate": 8.943428571428571e-07,
699
- "loss": 8.381609497070313,
700
  "step": 2350
701
  },
702
  {
703
- "epoch": 11.757894736842106,
704
- "grad_norm": 23.5266056060791,
705
- "learning_rate": 8.929142857142856e-07,
706
- "loss": 8.32532958984375,
707
  "step": 2375
708
  },
709
  {
710
- "epoch": 11.881733746130031,
711
- "grad_norm": 24.466337203979492,
712
- "learning_rate": 8.914857142857143e-07,
713
- "loss": 8.319666137695313,
714
  "step": 2400
715
  },
716
  {
717
- "epoch": 12.004953560371517,
718
- "grad_norm": 28.647558212280273,
719
- "learning_rate": 8.900571428571428e-07,
720
- "loss": 8.305534057617187,
721
  "step": 2425
722
  },
723
  {
724
- "epoch": 12.128792569659442,
725
- "grad_norm": 25.798559188842773,
726
- "learning_rate": 8.886285714285714e-07,
727
- "loss": 8.325819091796875,
728
  "step": 2450
729
  },
730
  {
731
- "epoch": 12.25263157894737,
732
- "grad_norm": 23.050085067749023,
733
- "learning_rate": 8.872e-07,
734
- "loss": 8.418189086914062,
735
  "step": 2475
736
  },
737
  {
738
- "epoch": 12.376470588235295,
739
- "grad_norm": 45.27325439453125,
740
- "learning_rate": 8.857714285714285e-07,
741
- "loss": 8.232376098632812,
742
  "step": 2500
743
  },
744
  {
745
- "epoch": 12.376470588235295,
746
- "eval_loss": 0.9387099742889404,
747
- "eval_runtime": 78.8561,
748
- "eval_samples_per_second": 20.493,
749
- "eval_steps_per_second": 5.123,
750
  "step": 2500
751
  },
752
  {
753
- "epoch": 12.50030959752322,
754
- "grad_norm": 30.00120735168457,
755
- "learning_rate": 8.843428571428571e-07,
756
- "loss": 8.216619262695312,
757
  "step": 2525
758
  },
759
  {
760
- "epoch": 12.624148606811145,
761
- "grad_norm": 20.330636978149414,
762
- "learning_rate": 8.829142857142857e-07,
763
- "loss": 8.305621948242187,
764
  "step": 2550
765
  },
766
  {
767
- "epoch": 12.74798761609907,
768
- "grad_norm": 23.445192337036133,
769
- "learning_rate": 8.814857142857143e-07,
770
- "loss": 8.283675537109374,
771
  "step": 2575
772
  },
773
  {
774
- "epoch": 12.871826625386998,
775
- "grad_norm": 19.532381057739258,
776
- "learning_rate": 8.800571428571428e-07,
777
- "loss": 8.197587890625,
778
  "step": 2600
779
  },
780
  {
781
- "epoch": 12.995665634674923,
782
- "grad_norm": 24.901432037353516,
783
- "learning_rate": 8.786285714285714e-07,
784
- "loss": 8.2815869140625,
785
  "step": 2625
786
  },
787
  {
788
- "epoch": 13.118885448916409,
789
- "grad_norm": 23.698482513427734,
790
- "learning_rate": 8.771999999999999e-07,
791
- "loss": 8.184017333984375,
792
  "step": 2650
793
  },
794
  {
795
- "epoch": 13.242724458204334,
796
- "grad_norm": 22.537649154663086,
797
- "learning_rate": 8.757714285714285e-07,
798
- "loss": 8.187122802734375,
799
  "step": 2675
800
  },
801
  {
802
- "epoch": 13.36656346749226,
803
- "grad_norm": 23.588315963745117,
804
- "learning_rate": 8.74342857142857e-07,
805
- "loss": 8.233043212890625,
806
  "step": 2700
807
  },
808
  {
809
- "epoch": 13.490402476780186,
810
- "grad_norm": 31.038589477539062,
811
- "learning_rate": 8.729142857142857e-07,
812
- "loss": 8.288897094726563,
813
  "step": 2725
814
  },
815
  {
816
- "epoch": 13.614241486068112,
817
- "grad_norm": 27.92134666442871,
818
- "learning_rate": 8.714857142857143e-07,
819
- "loss": 8.241436767578126,
820
  "step": 2750
821
  },
822
  {
823
- "epoch": 13.738080495356037,
824
- "grad_norm": 22.192489624023438,
825
- "learning_rate": 8.700571428571428e-07,
826
- "loss": 8.224349975585938,
827
  "step": 2775
828
  },
829
  {
830
- "epoch": 13.861919504643962,
831
- "grad_norm": 29.820369720458984,
832
- "learning_rate": 8.686285714285714e-07,
833
- "loss": 8.162620849609375,
834
  "step": 2800
835
  },
836
  {
837
- "epoch": 13.985758513931888,
838
- "grad_norm": 28.781845092773438,
839
- "learning_rate": 8.671999999999999e-07,
840
- "loss": 8.23837890625,
841
  "step": 2825
842
  },
843
  {
844
- "epoch": 14.108978328173375,
845
- "grad_norm": 21.96494483947754,
846
- "learning_rate": 8.657714285714285e-07,
847
- "loss": 8.158751831054687,
848
  "step": 2850
849
  },
850
  {
851
- "epoch": 14.2328173374613,
852
- "grad_norm": 22.800256729125977,
853
- "learning_rate": 8.643428571428571e-07,
854
- "loss": 8.228921508789062,
855
  "step": 2875
856
  },
857
  {
858
- "epoch": 14.356656346749226,
859
- "grad_norm": 23.881921768188477,
860
- "learning_rate": 8.629142857142857e-07,
861
- "loss": 8.183645629882813,
862
  "step": 2900
863
  },
864
  {
865
- "epoch": 14.480495356037151,
866
- "grad_norm": 22.399667739868164,
867
- "learning_rate": 8.614857142857143e-07,
868
- "loss": 8.157160034179688,
869
  "step": 2925
870
  },
871
  {
872
- "epoch": 14.604334365325077,
873
- "grad_norm": 22.374099731445312,
874
- "learning_rate": 8.600571428571428e-07,
875
- "loss": 8.129403076171876,
876
  "step": 2950
877
  },
878
  {
879
- "epoch": 14.728173374613004,
880
- "grad_norm": 22.495939254760742,
881
- "learning_rate": 8.586285714285714e-07,
882
- "loss": 8.19878662109375,
883
  "step": 2975
884
  },
885
  {
886
- "epoch": 14.852012383900929,
887
- "grad_norm": 26.343931198120117,
888
- "learning_rate": 8.571999999999999e-07,
889
- "loss": 8.154603881835937,
890
  "step": 3000
891
  },
892
  {
893
- "epoch": 14.852012383900929,
894
- "eval_loss": 0.9244788289070129,
895
- "eval_runtime": 78.6746,
896
- "eval_samples_per_second": 20.54,
897
- "eval_steps_per_second": 5.135,
898
  "step": 3000
899
  },
900
  {
901
- "epoch": 14.975851393188854,
902
- "grad_norm": 20.63393783569336,
903
- "learning_rate": 8.557714285714285e-07,
904
- "loss": 8.259378051757812,
905
  "step": 3025
906
  },
907
  {
908
- "epoch": 15.09907120743034,
909
- "grad_norm": 28.88397789001465,
910
- "learning_rate": 8.543428571428571e-07,
911
- "loss": 8.186115112304687,
912
  "step": 3050
913
  },
914
  {
915
- "epoch": 15.222910216718267,
916
- "grad_norm": 25.326343536376953,
917
- "learning_rate": 8.529142857142857e-07,
918
- "loss": 8.17370849609375,
919
  "step": 3075
920
  },
921
  {
922
- "epoch": 15.346749226006192,
923
- "grad_norm": 22.252532958984375,
924
- "learning_rate": 8.514857142857142e-07,
925
- "loss": 7.99333984375,
926
  "step": 3100
927
  },
928
  {
929
- "epoch": 15.470588235294118,
930
- "grad_norm": 22.498172760009766,
931
- "learning_rate": 8.500571428571428e-07,
932
- "loss": 8.131607666015626,
933
  "step": 3125
934
  },
935
  {
936
- "epoch": 15.594427244582043,
937
- "grad_norm": 19.141489028930664,
938
- "learning_rate": 8.486285714285714e-07,
939
- "loss": 8.157739868164063,
940
  "step": 3150
941
  },
942
  {
943
- "epoch": 15.718266253869968,
944
- "grad_norm": 24.114906311035156,
945
- "learning_rate": 8.471999999999999e-07,
946
- "loss": 8.153767700195312,
947
  "step": 3175
948
  },
949
  {
950
- "epoch": 15.842105263157894,
951
- "grad_norm": 24.1041259765625,
952
- "learning_rate": 8.457714285714286e-07,
953
- "loss": 8.211907348632813,
954
  "step": 3200
955
  },
956
  {
957
- "epoch": 15.965944272445821,
958
- "grad_norm": 26.138748168945312,
959
- "learning_rate": 8.443428571428571e-07,
960
- "loss": 8.123822631835937,
961
  "step": 3225
962
  },
963
  {
964
- "epoch": 16.08916408668731,
965
- "grad_norm": 19.59873390197754,
966
- "learning_rate": 8.429142857142857e-07,
967
- "loss": 8.09514404296875,
968
  "step": 3250
969
  },
970
  {
971
- "epoch": 16.213003095975232,
972
- "grad_norm": 23.773271560668945,
973
- "learning_rate": 8.414857142857142e-07,
974
- "loss": 8.156911010742187,
975
  "step": 3275
976
  },
977
  {
978
- "epoch": 16.33684210526316,
979
- "grad_norm": 25.273815155029297,
980
- "learning_rate": 8.400571428571428e-07,
981
- "loss": 8.161531982421875,
982
  "step": 3300
983
  },
984
  {
985
- "epoch": 16.460681114551083,
986
- "grad_norm": 22.487651824951172,
987
- "learning_rate": 8.386285714285713e-07,
988
- "loss": 8.00319091796875,
989
  "step": 3325
990
  },
991
  {
992
- "epoch": 16.58452012383901,
993
- "grad_norm": 28.419109344482422,
994
- "learning_rate": 8.372e-07,
995
- "loss": 8.031408081054687,
996
  "step": 3350
997
  },
998
  {
999
- "epoch": 16.708359133126933,
1000
- "grad_norm": 21.66465187072754,
1001
- "learning_rate": 8.357714285714286e-07,
1002
- "loss": 8.159840087890625,
1003
  "step": 3375
1004
  },
1005
  {
1006
- "epoch": 16.83219814241486,
1007
- "grad_norm": 23.416608810424805,
1008
- "learning_rate": 8.343428571428571e-07,
1009
- "loss": 8.243095092773437,
1010
  "step": 3400
1011
  },
1012
  {
1013
- "epoch": 16.956037151702787,
1014
- "grad_norm": 24.853750228881836,
1015
- "learning_rate": 8.329142857142857e-07,
1016
- "loss": 8.11635986328125,
1017
  "step": 3425
1018
  },
1019
  {
1020
- "epoch": 17.07925696594427,
1021
- "grad_norm": 24.341514587402344,
1022
- "learning_rate": 8.314857142857142e-07,
1023
- "loss": 7.91652587890625,
1024
  "step": 3450
1025
  },
1026
  {
1027
- "epoch": 17.2030959752322,
1028
- "grad_norm": 25.844379425048828,
1029
- "learning_rate": 8.300571428571428e-07,
1030
- "loss": 8.211973876953126,
1031
  "step": 3475
1032
  },
1033
  {
1034
- "epoch": 17.326934984520125,
1035
- "grad_norm": 23.98019790649414,
1036
- "learning_rate": 8.286285714285713e-07,
1037
- "loss": 8.040772705078124,
1038
  "step": 3500
1039
  },
1040
  {
1041
- "epoch": 17.326934984520125,
1042
- "eval_loss": 0.9177140593528748,
1043
- "eval_runtime": 79.4513,
1044
- "eval_samples_per_second": 20.34,
1045
- "eval_steps_per_second": 5.085,
1046
  "step": 3500
1047
  },
1048
  {
1049
- "epoch": 17.45077399380805,
1050
- "grad_norm": 27.95872688293457,
1051
- "learning_rate": 8.272e-07,
1052
- "loss": 7.994537963867187,
1053
  "step": 3525
1054
  },
1055
  {
1056
- "epoch": 17.574613003095976,
1057
- "grad_norm": 23.433879852294922,
1058
- "learning_rate": 8.257714285714286e-07,
1059
- "loss": 8.102257080078125,
1060
  "step": 3550
1061
  },
1062
  {
1063
- "epoch": 17.6984520123839,
1064
- "grad_norm": 25.75996208190918,
1065
- "learning_rate": 8.243428571428571e-07,
1066
- "loss": 8.0616259765625,
1067
  "step": 3575
1068
  },
1069
  {
1070
- "epoch": 17.822291021671827,
1071
- "grad_norm": 20.315242767333984,
1072
- "learning_rate": 8.229142857142857e-07,
1073
- "loss": 8.003837280273437,
1074
  "step": 3600
1075
  },
1076
  {
1077
- "epoch": 17.946130030959754,
1078
- "grad_norm": 34.34598922729492,
1079
- "learning_rate": 8.214857142857142e-07,
1080
- "loss": 8.088245239257812,
1081
  "step": 3625
1082
  },
1083
  {
1084
- "epoch": 18.069349845201238,
1085
- "grad_norm": 20.82579231262207,
1086
- "learning_rate": 8.200571428571428e-07,
1087
- "loss": 8.047254638671875,
1088
  "step": 3650
1089
  },
1090
  {
1091
- "epoch": 18.193188854489165,
1092
- "grad_norm": 29.7625732421875,
1093
- "learning_rate": 8.186285714285714e-07,
1094
- "loss": 8.1514794921875,
1095
  "step": 3675
1096
  },
1097
  {
1098
- "epoch": 18.31702786377709,
1099
- "grad_norm": 24.978836059570312,
1100
- "learning_rate": 8.172e-07,
1101
- "loss": 7.944334106445313,
1102
  "step": 3700
1103
  },
1104
  {
1105
- "epoch": 18.440866873065016,
1106
- "grad_norm": 28.676979064941406,
1107
- "learning_rate": 8.157714285714285e-07,
1108
- "loss": 8.004066772460938,
1109
  "step": 3725
1110
  },
1111
  {
1112
- "epoch": 18.564705882352943,
1113
- "grad_norm": 22.64858627319336,
1114
- "learning_rate": 8.143428571428571e-07,
1115
- "loss": 8.025421752929688,
1116
  "step": 3750
1117
  },
1118
  {
1119
- "epoch": 18.688544891640866,
1120
- "grad_norm": 24.17665672302246,
1121
- "learning_rate": 8.129142857142857e-07,
1122
- "loss": 8.024356079101562,
1123
  "step": 3775
1124
  },
1125
  {
1126
- "epoch": 18.812383900928793,
1127
- "grad_norm": 36.6374397277832,
1128
- "learning_rate": 8.114857142857142e-07,
1129
- "loss": 8.029654541015624,
1130
  "step": 3800
1131
  },
1132
  {
1133
- "epoch": 18.936222910216717,
1134
- "grad_norm": 27.306865692138672,
1135
- "learning_rate": 8.100571428571428e-07,
1136
- "loss": 8.086602172851563,
1137
  "step": 3825
1138
  },
1139
  {
1140
- "epoch": 19.059442724458204,
1141
- "grad_norm": 21.691247940063477,
1142
- "learning_rate": 8.086285714285714e-07,
1143
- "loss": 7.949559326171875,
1144
  "step": 3850
1145
  },
1146
  {
1147
- "epoch": 19.18328173374613,
1148
- "grad_norm": 21.185199737548828,
1149
- "learning_rate": 8.072e-07,
1150
- "loss": 7.867513427734375,
1151
  "step": 3875
1152
  },
1153
  {
1154
- "epoch": 19.307120743034055,
1155
- "grad_norm": 20.94526481628418,
1156
- "learning_rate": 8.057714285714285e-07,
1157
- "loss": 7.994710693359375,
1158
  "step": 3900
1159
  },
1160
  {
1161
- "epoch": 19.430959752321982,
1162
- "grad_norm": 20.232423782348633,
1163
- "learning_rate": 8.043428571428571e-07,
1164
- "loss": 8.108115234375,
1165
  "step": 3925
1166
  },
1167
  {
1168
- "epoch": 19.554798761609906,
1169
- "grad_norm": 22.80320930480957,
1170
- "learning_rate": 8.029142857142856e-07,
1171
- "loss": 8.067723388671874,
1172
  "step": 3950
1173
  },
1174
  {
1175
- "epoch": 19.678637770897833,
1176
- "grad_norm": 25.35050392150879,
1177
- "learning_rate": 8.014857142857142e-07,
1178
- "loss": 7.935881958007813,
1179
  "step": 3975
1180
  },
1181
  {
1182
- "epoch": 19.80247678018576,
1183
- "grad_norm": 30.46210479736328,
1184
- "learning_rate": 8.000571428571429e-07,
1185
- "loss": 8.018241577148437,
1186
  "step": 4000
1187
  },
1188
  {
1189
- "epoch": 19.80247678018576,
1190
- "eval_loss": 0.9094438552856445,
1191
- "eval_runtime": 80.6469,
1192
- "eval_samples_per_second": 20.038,
1193
- "eval_steps_per_second": 5.009,
1194
  "step": 4000
1195
  },
1196
  {
1197
- "epoch": 19.926315789473684,
1198
- "grad_norm": 27.09125518798828,
1199
- "learning_rate": 7.986285714285714e-07,
1200
- "loss": 8.072515869140625,
1201
  "step": 4025
1202
  },
1203
  {
1204
- "epoch": 20.04953560371517,
1205
- "grad_norm": 26.138822555541992,
1206
- "learning_rate": 7.972e-07,
1207
- "loss": 7.916986694335938,
1208
  "step": 4050
1209
  },
1210
  {
1211
- "epoch": 20.173374613003094,
1212
- "grad_norm": 23.57080078125,
1213
- "learning_rate": 7.957714285714285e-07,
1214
- "loss": 7.979219970703125,
1215
  "step": 4075
1216
  },
1217
  {
1218
- "epoch": 20.29721362229102,
1219
- "grad_norm": 28.19639778137207,
1220
- "learning_rate": 7.943428571428571e-07,
1221
- "loss": 7.938901977539063,
1222
  "step": 4100
1223
  },
1224
  {
1225
- "epoch": 20.42105263157895,
1226
- "grad_norm": 23.884334564208984,
1227
- "learning_rate": 7.929142857142856e-07,
1228
- "loss": 8.061412353515625,
1229
  "step": 4125
1230
  },
1231
  {
1232
- "epoch": 20.544891640866872,
1233
- "grad_norm": 21.111995697021484,
1234
- "learning_rate": 7.914857142857142e-07,
1235
- "loss": 7.908670043945312,
1236
  "step": 4150
1237
  },
1238
  {
1239
- "epoch": 20.6687306501548,
1240
- "grad_norm": 22.234756469726562,
1241
- "learning_rate": 7.900571428571428e-07,
1242
- "loss": 8.053919067382813,
1243
  "step": 4175
1244
  },
1245
  {
1246
- "epoch": 20.792569659442723,
1247
- "grad_norm": 27.65105438232422,
1248
- "learning_rate": 7.886285714285714e-07,
1249
- "loss": 7.927275390625,
1250
  "step": 4200
1251
  },
1252
  {
1253
- "epoch": 20.91640866873065,
1254
- "grad_norm": 20.67005157470703,
1255
- "learning_rate": 7.872e-07,
1256
- "loss": 8.0426611328125,
1257
  "step": 4225
1258
  },
1259
  {
1260
- "epoch": 21.039628482972137,
1261
- "grad_norm": 21.650495529174805,
1262
- "learning_rate": 7.857714285714285e-07,
1263
- "loss": 7.8834521484375,
1264
  "step": 4250
1265
  },
1266
  {
1267
- "epoch": 21.16346749226006,
1268
- "grad_norm": 24.481433868408203,
1269
- "learning_rate": 7.843428571428571e-07,
1270
- "loss": 7.956741943359375,
1271
  "step": 4275
1272
  },
1273
  {
1274
- "epoch": 21.287306501547988,
1275
- "grad_norm": 25.228164672851562,
1276
- "learning_rate": 7.829142857142856e-07,
1277
- "loss": 7.872012939453125,
1278
  "step": 4300
1279
  },
1280
  {
1281
- "epoch": 21.41114551083591,
1282
- "grad_norm": 23.258808135986328,
1283
- "learning_rate": 7.814857142857143e-07,
1284
- "loss": 7.93873046875,
1285
  "step": 4325
1286
  },
1287
  {
1288
- "epoch": 21.53498452012384,
1289
- "grad_norm": 39.16718292236328,
1290
- "learning_rate": 7.800571428571428e-07,
1291
- "loss": 8.053985595703125,
1292
  "step": 4350
1293
  },
1294
  {
1295
- "epoch": 21.658823529411766,
1296
- "grad_norm": 23.273096084594727,
1297
- "learning_rate": 7.786285714285714e-07,
1298
- "loss": 7.987719116210937,
1299
  "step": 4375
1300
  },
1301
  {
1302
- "epoch": 21.78266253869969,
1303
- "grad_norm": 21.058330535888672,
1304
- "learning_rate": 7.771999999999999e-07,
1305
- "loss": 7.941270141601563,
1306
  "step": 4400
1307
  },
1308
  {
1309
- "epoch": 21.906501547987617,
1310
- "grad_norm": 20.82206153869629,
1311
- "learning_rate": 7.757714285714285e-07,
1312
- "loss": 7.894559936523438,
1313
  "step": 4425
1314
  },
1315
  {
1316
- "epoch": 22.029721362229104,
1317
- "grad_norm": 30.24216651916504,
1318
- "learning_rate": 7.743428571428571e-07,
1319
- "loss": 7.936047973632813,
1320
  "step": 4450
1321
  },
1322
  {
1323
- "epoch": 22.153560371517028,
1324
- "grad_norm": 25.227153778076172,
1325
- "learning_rate": 7.729142857142856e-07,
1326
- "loss": 8.067274169921875,
1327
  "step": 4475
1328
  },
1329
  {
1330
- "epoch": 22.277399380804955,
1331
- "grad_norm": 24.134035110473633,
1332
- "learning_rate": 7.714857142857143e-07,
1333
- "loss": 7.912605590820313,
1334
  "step": 4500
1335
  },
1336
  {
1337
- "epoch": 22.277399380804955,
1338
- "eval_loss": 0.9072220325469971,
1339
- "eval_runtime": 78.7656,
1340
- "eval_samples_per_second": 20.517,
1341
- "eval_steps_per_second": 5.129,
1342
  "step": 4500
1343
  },
1344
  {
1345
- "epoch": 22.401238390092878,
1346
- "grad_norm": 24.902118682861328,
1347
- "learning_rate": 7.700571428571428e-07,
1348
- "loss": 7.941981811523437,
1349
  "step": 4525
1350
  },
1351
  {
1352
- "epoch": 22.525077399380805,
1353
- "grad_norm": 22.811532974243164,
1354
- "learning_rate": 7.686285714285714e-07,
1355
- "loss": 8.003760375976562,
1356
  "step": 4550
1357
  },
1358
  {
1359
- "epoch": 22.648916408668732,
1360
- "grad_norm": 27.126218795776367,
1361
- "learning_rate": 7.671999999999999e-07,
1362
- "loss": 7.954713134765625,
1363
  "step": 4575
1364
  },
1365
  {
1366
- "epoch": 22.772755417956656,
1367
- "grad_norm": 21.63432502746582,
1368
- "learning_rate": 7.657714285714285e-07,
1369
- "loss": 8.02743896484375,
1370
  "step": 4600
1371
  },
1372
  {
1373
- "epoch": 22.896594427244583,
1374
- "grad_norm": 16.372127532958984,
1375
- "learning_rate": 7.64342857142857e-07,
1376
- "loss": 7.892459716796875,
1377
  "step": 4625
1378
  },
1379
  {
1380
- "epoch": 23.019814241486067,
1381
- "grad_norm": 21.199737548828125,
1382
- "learning_rate": 7.629142857142857e-07,
1383
- "loss": 7.839408569335937,
1384
  "step": 4650
1385
  },
1386
  {
1387
- "epoch": 23.143653250773994,
1388
- "grad_norm": 21.877565383911133,
1389
- "learning_rate": 7.614857142857143e-07,
1390
- "loss": 7.9298638916015625,
1391
  "step": 4675
1392
  },
1393
  {
1394
- "epoch": 23.26749226006192,
1395
- "grad_norm": 33.95394515991211,
1396
- "learning_rate": 7.600571428571428e-07,
1397
- "loss": 7.974397583007812,
1398
  "step": 4700
1399
  },
1400
  {
1401
- "epoch": 23.391331269349845,
1402
- "grad_norm": 20.519060134887695,
1403
- "learning_rate": 7.586285714285714e-07,
1404
- "loss": 7.82852783203125,
1405
  "step": 4725
1406
  },
1407
  {
1408
- "epoch": 23.515170278637772,
1409
- "grad_norm": 18.024555206298828,
1410
- "learning_rate": 7.571999999999999e-07,
1411
- "loss": 7.9392626953125,
1412
  "step": 4750
1413
  },
1414
  {
1415
- "epoch": 23.639009287925695,
1416
- "grad_norm": 24.224964141845703,
1417
- "learning_rate": 7.557714285714285e-07,
1418
- "loss": 7.970030517578125,
1419
  "step": 4775
1420
  },
1421
  {
1422
- "epoch": 23.762848297213623,
1423
- "grad_norm": 23.275896072387695,
1424
- "learning_rate": 7.54342857142857e-07,
1425
- "loss": 7.920643310546875,
1426
  "step": 4800
1427
  },
1428
  {
1429
- "epoch": 23.88668730650155,
1430
- "grad_norm": 21.45893096923828,
1431
- "learning_rate": 7.529142857142857e-07,
1432
- "loss": 7.948008422851562,
1433
  "step": 4825
1434
  },
1435
  {
1436
- "epoch": 24.009907120743033,
1437
- "grad_norm": 29.533275604248047,
1438
- "learning_rate": 7.514857142857142e-07,
1439
- "loss": 7.849251098632813,
1440
  "step": 4850
1441
  },
1442
  {
1443
- "epoch": 24.13374613003096,
1444
- "grad_norm": 27.810075759887695,
1445
- "learning_rate": 7.500571428571428e-07,
1446
- "loss": 7.834190673828125,
1447
  "step": 4875
1448
  },
1449
  {
1450
- "epoch": 24.257585139318884,
1451
- "grad_norm": 19.784141540527344,
1452
- "learning_rate": 7.486285714285714e-07,
1453
- "loss": 7.896500244140625,
1454
  "step": 4900
1455
  },
1456
  {
1457
- "epoch": 24.38142414860681,
1458
- "grad_norm": 20.703563690185547,
1459
- "learning_rate": 7.471999999999999e-07,
1460
- "loss": 7.881874389648438,
1461
  "step": 4925
1462
  },
1463
  {
1464
- "epoch": 24.50526315789474,
1465
- "grad_norm": 24.245025634765625,
1466
- "learning_rate": 7.457714285714285e-07,
1467
- "loss": 7.839683227539062,
1468
  "step": 4950
1469
  },
1470
  {
1471
- "epoch": 24.629102167182662,
1472
- "grad_norm": 21.31154441833496,
1473
- "learning_rate": 7.443428571428571e-07,
1474
- "loss": 7.845946655273438,
1475
  "step": 4975
1476
  },
1477
  {
1478
- "epoch": 24.75294117647059,
1479
- "grad_norm": 28.88249969482422,
1480
- "learning_rate": 7.429142857142857e-07,
1481
- "loss": 7.919102172851563,
1482
  "step": 5000
1483
  },
1484
  {
1485
- "epoch": 24.75294117647059,
1486
- "eval_loss": 0.8998559713363647,
1487
- "eval_runtime": 78.9705,
1488
- "eval_samples_per_second": 20.463,
1489
- "eval_steps_per_second": 5.116,
1490
  "step": 5000
1491
  },
1492
  {
1493
- "epoch": 24.876780185758513,
1494
- "grad_norm": 24.55940818786621,
1495
- "learning_rate": 7.414857142857142e-07,
1496
- "loss": 7.9317694091796875,
1497
  "step": 5025
1498
  },
1499
  {
1500
  "epoch": 25.0,
1501
- "grad_norm": 22.524856567382812,
1502
- "learning_rate": 7.400571428571428e-07,
1503
- "loss": 7.979331665039062,
1504
  "step": 5050
1505
  },
1506
  {
1507
- "epoch": 25.123839009287927,
1508
- "grad_norm": 22.78742218017578,
1509
- "learning_rate": 7.386285714285713e-07,
1510
- "loss": 7.89152099609375,
1511
  "step": 5075
1512
  },
1513
  {
1514
- "epoch": 25.24767801857585,
1515
- "grad_norm": 20.438522338867188,
1516
- "learning_rate": 7.371999999999999e-07,
1517
- "loss": 7.915701904296875,
1518
  "step": 5100
1519
  },
1520
  {
1521
- "epoch": 25.371517027863778,
1522
- "grad_norm": 17.50803565979004,
1523
- "learning_rate": 7.357714285714285e-07,
1524
- "loss": 7.80771728515625,
1525
  "step": 5125
1526
  },
1527
  {
1528
- "epoch": 25.4953560371517,
1529
- "grad_norm": 23.66606903076172,
1530
- "learning_rate": 7.343428571428571e-07,
1531
- "loss": 7.9225396728515625,
1532
  "step": 5150
1533
  },
1534
  {
1535
- "epoch": 25.61919504643963,
1536
- "grad_norm": 25.388748168945312,
1537
- "learning_rate": 7.329142857142857e-07,
1538
- "loss": 7.979483642578125,
1539
  "step": 5175
1540
  },
1541
  {
1542
- "epoch": 25.743034055727556,
1543
- "grad_norm": 17.87958335876465,
1544
- "learning_rate": 7.314857142857142e-07,
1545
- "loss": 7.863919067382812,
1546
  "step": 5200
1547
  },
1548
  {
1549
- "epoch": 25.86687306501548,
1550
- "grad_norm": 25.70256996154785,
1551
- "learning_rate": 7.300571428571428e-07,
1552
- "loss": 7.822443237304688,
1553
  "step": 5225
1554
  },
1555
  {
1556
- "epoch": 25.990712074303406,
1557
- "grad_norm": 21.018875122070312,
1558
- "learning_rate": 7.286285714285713e-07,
1559
- "loss": 7.927708740234375,
1560
  "step": 5250
1561
  },
1562
  {
1563
- "epoch": 26.11393188854489,
1564
- "grad_norm": 22.857200622558594,
1565
- "learning_rate": 7.271999999999999e-07,
1566
- "loss": 7.804317626953125,
1567
  "step": 5275
1568
  },
1569
  {
1570
- "epoch": 26.237770897832817,
1571
- "grad_norm": 24.68769073486328,
1572
- "learning_rate": 7.257714285714286e-07,
1573
- "loss": 7.823025512695312,
1574
  "step": 5300
1575
  },
1576
  {
1577
- "epoch": 26.361609907120744,
1578
- "grad_norm": 23.744749069213867,
1579
- "learning_rate": 7.243428571428571e-07,
1580
- "loss": 7.970792236328125,
1581
  "step": 5325
1582
  },
1583
  {
1584
- "epoch": 26.485448916408668,
1585
- "grad_norm": 35.228553771972656,
1586
- "learning_rate": 7.229142857142857e-07,
1587
- "loss": 7.818600463867187,
1588
  "step": 5350
1589
  },
1590
  {
1591
- "epoch": 26.609287925696595,
1592
- "grad_norm": 20.797626495361328,
1593
- "learning_rate": 7.214857142857142e-07,
1594
- "loss": 7.821973266601563,
1595
  "step": 5375
1596
  },
1597
  {
1598
- "epoch": 26.73312693498452,
1599
- "grad_norm": 29.83957290649414,
1600
- "learning_rate": 7.200571428571428e-07,
1601
- "loss": 7.875313720703125,
1602
  "step": 5400
1603
  },
1604
  {
1605
- "epoch": 26.856965944272446,
1606
- "grad_norm": 22.217191696166992,
1607
- "learning_rate": 7.186285714285713e-07,
1608
- "loss": 7.792033081054687,
1609
  "step": 5425
1610
  },
1611
  {
1612
- "epoch": 26.980804953560373,
1613
- "grad_norm": 22.105436325073242,
1614
- "learning_rate": 7.171999999999999e-07,
1615
- "loss": 7.988189086914063,
1616
  "step": 5450
1617
  },
1618
  {
1619
- "epoch": 27.104024767801857,
1620
- "grad_norm": 20.319862365722656,
1621
- "learning_rate": 7.157714285714285e-07,
1622
- "loss": 7.729344482421875,
1623
  "step": 5475
1624
  },
1625
  {
1626
- "epoch": 27.227863777089784,
1627
- "grad_norm": 18.165098190307617,
1628
- "learning_rate": 7.143428571428571e-07,
1629
- "loss": 7.781807250976563,
1630
  "step": 5500
1631
  },
1632
  {
1633
- "epoch": 27.227863777089784,
1634
- "eval_loss": 0.8907444477081299,
1635
- "eval_runtime": 78.9877,
1636
- "eval_samples_per_second": 20.459,
1637
- "eval_steps_per_second": 5.115,
1638
  "step": 5500
1639
  }
1640
  ],
@@ -1664,8 +1664,8 @@
1664
  "attributes": {}
1665
  }
1666
  },
1667
- "total_flos": 2.1419286975434304e+16,
1668
- "train_batch_size": 4,
1669
  "trial_name": null,
1670
  "trial_params": null
1671
  }
 
1
  {
2
  "best_global_step": 5500,
3
+ "best_metric": 0.8911440968513489,
4
  "best_model_checkpoint": "/kaggle/working/tts_processed/checkpoint-5500",
5
+ "epoch": 27.22772277227723,
6
  "eval_steps": 500,
7
  "global_step": 5500,
8
  "is_hyper_param_search": false,
 
10
  "is_world_process_zero": true,
11
  "log_history": [
12
  {
13
+ "epoch": 0.12376237623762376,
14
+ "grad_norm": 363.43377685546875,
15
  "learning_rate": 4.8e-08,
16
+ "loss": 10.89256103515625,
17
  "step": 25
18
  },
19
  {
20
+ "epoch": 0.24752475247524752,
21
+ "grad_norm": 336.6938171386719,
22
  "learning_rate": 9.8e-08,
23
+ "loss": 10.47541259765625,
24
  "step": 50
25
  },
26
  {
27
+ "epoch": 0.3712871287128713,
28
+ "grad_norm": 255.0070343017578,
29
  "learning_rate": 1.4799999999999998e-07,
30
+ "loss": 10.08507568359375,
31
  "step": 75
32
  },
33
  {
34
+ "epoch": 0.49504950495049505,
35
+ "grad_norm": 247.54541015625,
36
  "learning_rate": 1.98e-07,
37
+ "loss": 9.282888793945313,
38
  "step": 100
39
  },
40
  {
41
+ "epoch": 0.6188118811881188,
42
+ "grad_norm": 124.208740234375,
43
  "learning_rate": 2.48e-07,
44
+ "loss": 8.717689208984375,
45
  "step": 125
46
  },
47
  {
48
+ "epoch": 0.7425742574257426,
49
+ "grad_norm": 194.5498809814453,
50
  "learning_rate": 2.98e-07,
51
+ "loss": 8.212156372070313,
52
  "step": 150
53
  },
54
  {
55
+ "epoch": 0.8663366336633663,
56
+ "grad_norm": 143.2777557373047,
57
  "learning_rate": 3.4799999999999994e-07,
58
+ "loss": 7.8293896484375,
59
  "step": 175
60
  },
61
  {
62
+ "epoch": 0.9900990099009901,
63
+ "grad_norm": 55.50076675415039,
64
  "learning_rate": 3.98e-07,
65
+ "loss": 7.3809716796875,
66
  "step": 200
67
  },
68
  {
69
+ "epoch": 1.113861386138614,
70
+ "grad_norm": 69.43047332763672,
71
  "learning_rate": 4.48e-07,
72
+ "loss": 6.937587890625,
73
  "step": 225
74
  },
75
  {
76
+ "epoch": 1.2376237623762376,
77
+ "grad_norm": 49.69321060180664,
78
  "learning_rate": 4.979999999999999e-07,
79
+ "loss": 6.8110888671875,
80
  "step": 250
81
  },
82
  {
83
+ "epoch": 1.3613861386138613,
84
+ "grad_norm": 26.72427749633789,
85
  "learning_rate": 5.48e-07,
86
+ "loss": 6.522671508789062,
87
  "step": 275
88
  },
89
  {
90
+ "epoch": 1.4851485148514851,
91
+ "grad_norm": 51.36345672607422,
92
  "learning_rate": 5.979999999999999e-07,
93
+ "loss": 6.398966064453125,
94
  "step": 300
95
  },
96
  {
97
+ "epoch": 1.608910891089109,
98
+ "grad_norm": 38.85439682006836,
99
  "learning_rate": 6.48e-07,
100
+ "loss": 6.198662109375,
101
  "step": 325
102
  },
103
  {
104
+ "epoch": 1.7326732673267327,
105
+ "grad_norm": 27.508665084838867,
106
  "learning_rate": 6.979999999999999e-07,
107
+ "loss": 6.075740356445312,
108
  "step": 350
109
  },
110
  {
111
+ "epoch": 1.8564356435643563,
112
+ "grad_norm": 36.8642463684082,
113
  "learning_rate": 7.48e-07,
114
+ "loss": 6.0834722900390625,
115
  "step": 375
116
  },
117
  {
118
+ "epoch": 1.9801980198019802,
119
+ "grad_norm": 27.07192039489746,
120
  "learning_rate": 7.98e-07,
121
+ "loss": 5.9166015625,
122
  "step": 400
123
  },
124
  {
125
+ "epoch": 2.103960396039604,
126
+ "grad_norm": 22.465463638305664,
127
  "learning_rate": 8.48e-07,
128
+ "loss": 5.795056762695313,
129
  "step": 425
130
  },
131
  {
132
+ "epoch": 2.227722772277228,
133
+ "grad_norm": 34.86723709106445,
134
  "learning_rate": 8.98e-07,
135
+ "loss": 5.741965942382812,
136
  "step": 450
137
  },
138
  {
139
+ "epoch": 2.3514851485148514,
140
+ "grad_norm": 22.37932777404785,
141
  "learning_rate": 9.479999999999999e-07,
142
+ "loss": 5.716455688476563,
143
  "step": 475
144
  },
145
  {
146
+ "epoch": 2.4752475247524752,
147
+ "grad_norm": 29.54213523864746,
148
  "learning_rate": 9.98e-07,
149
+ "loss": 5.599214477539062,
150
  "step": 500
151
  },
152
  {
153
+ "epoch": 2.4752475247524752,
154
+ "eval_loss": 1.2534884214401245,
155
+ "eval_runtime": 82.4141,
156
+ "eval_samples_per_second": 19.608,
157
+ "eval_steps_per_second": 4.902,
158
  "step": 500
159
  },
160
  {
161
+ "epoch": 2.599009900990099,
162
+ "grad_norm": 21.54619026184082,
163
+ "learning_rate": 9.999953592789052e-07,
164
+ "loss": 5.526549072265625,
165
  "step": 525
166
  },
167
  {
168
+ "epoch": 2.7227722772277225,
169
+ "grad_norm": 31.9982967376709,
170
+ "learning_rate": 9.999806557001091e-07,
171
+ "loss": 5.38673828125,
172
  "step": 550
173
  },
174
  {
175
+ "epoch": 2.8465346534653464,
176
+ "grad_norm": 19.33993148803711,
177
+ "learning_rate": 9.999558815029513e-07,
178
+ "loss": 5.291504516601562,
179
  "step": 575
180
  },
181
  {
182
+ "epoch": 2.9702970297029703,
183
+ "grad_norm": 18.838518142700195,
184
+ "learning_rate": 9.999210371864342e-07,
185
+ "loss": 5.345247192382812,
186
  "step": 600
187
  },
188
  {
189
+ "epoch": 3.094059405940594,
190
+ "grad_norm": 19.941740036010742,
191
+ "learning_rate": 9.998761234523925e-07,
192
+ "loss": 5.081980285644531,
193
  "step": 625
194
  },
195
  {
196
+ "epoch": 3.217821782178218,
197
+ "grad_norm": 24.84575080871582,
198
+ "learning_rate": 9.998211412054793e-07,
199
+ "loss": 5.025005493164063,
200
  "step": 650
201
  },
202
  {
203
+ "epoch": 3.3415841584158414,
204
+ "grad_norm": 24.532045364379883,
205
+ "learning_rate": 9.99756091553148e-07,
206
+ "loss": 5.004798889160156,
207
  "step": 675
208
  },
209
  {
210
+ "epoch": 3.4653465346534653,
211
+ "grad_norm": 21.20513916015625,
212
+ "learning_rate": 9.996809758056298e-07,
213
+ "loss": 4.870877685546875,
214
  "step": 700
215
  },
216
  {
217
+ "epoch": 3.589108910891089,
218
+ "grad_norm": 16.589248657226562,
219
+ "learning_rate": 9.995957954759072e-07,
220
+ "loss": 4.95923095703125,
221
  "step": 725
222
  },
223
  {
224
+ "epoch": 3.7128712871287126,
225
+ "grad_norm": 26.863059997558594,
226
+ "learning_rate": 9.995005522796837e-07,
227
+ "loss": 4.81072265625,
228
  "step": 750
229
  },
230
  {
231
+ "epoch": 3.8366336633663365,
232
+ "grad_norm": 14.973981857299805,
233
+ "learning_rate": 9.993952481353493e-07,
234
+ "loss": 4.8048876953125,
235
  "step": 775
236
  },
237
  {
238
+ "epoch": 3.9603960396039604,
239
+ "grad_norm": 18.018110275268555,
240
+ "learning_rate": 9.992798851639418e-07,
241
+ "loss": 4.770014343261718,
242
  "step": 800
243
  },
244
  {
245
+ "epoch": 4.084158415841584,
246
+ "grad_norm": 15.32919979095459,
247
+ "learning_rate": 9.991544656891043e-07,
248
+ "loss": 4.663901672363282,
249
  "step": 825
250
  },
251
  {
252
+ "epoch": 4.207920792079208,
253
+ "grad_norm": 14.925799369812012,
254
+ "learning_rate": 9.990189922370373e-07,
255
+ "loss": 4.614352111816406,
256
  "step": 850
257
  },
258
  {
259
+ "epoch": 4.3316831683168315,
260
+ "grad_norm": 14.5320405960083,
261
+ "learning_rate": 9.988734675364496e-07,
262
+ "loss": 4.607881774902344,
263
  "step": 875
264
  },
265
  {
266
+ "epoch": 4.455445544554456,
267
+ "grad_norm": 17.06329345703125,
268
+ "learning_rate": 9.987178945185019e-07,
269
+ "loss": 4.582254028320312,
270
  "step": 900
271
  },
272
  {
273
+ "epoch": 4.579207920792079,
274
+ "grad_norm": 16.860803604125977,
275
+ "learning_rate": 9.985522763167486e-07,
276
+ "loss": 4.6025848388671875,
277
  "step": 925
278
  },
279
  {
280
+ "epoch": 4.702970297029703,
281
+ "grad_norm": 12.743483543395996,
282
+ "learning_rate": 9.983766162670739e-07,
283
+ "loss": 4.603981628417968,
284
  "step": 950
285
  },
286
  {
287
+ "epoch": 4.826732673267327,
288
+ "grad_norm": 11.648216247558594,
289
+ "learning_rate": 9.981909179076255e-07,
290
+ "loss": 4.5333544921875,
291
  "step": 975
292
  },
293
  {
294
+ "epoch": 4.9504950495049505,
295
+ "grad_norm": 12.38879108428955,
296
+ "learning_rate": 9.979951849787429e-07,
297
+ "loss": 4.538429565429688,
298
  "step": 1000
299
  },
300
  {
301
+ "epoch": 4.9504950495049505,
302
+ "eval_loss": 1.0112695693969727,
303
+ "eval_runtime": 82.132,
304
+ "eval_samples_per_second": 19.676,
305
+ "eval_steps_per_second": 4.919,
306
  "step": 1000
307
  },
308
  {
309
+ "epoch": 5.074257425742574,
310
+ "grad_norm": 39.850196838378906,
311
+ "learning_rate": 9.977894214228815e-07,
312
+ "loss": 4.55607666015625,
313
  "step": 1025
314
  },
315
  {
316
+ "epoch": 5.198019801980198,
317
+ "grad_norm": 13.423649787902832,
318
+ "learning_rate": 9.975736313845345e-07,
319
+ "loss": 4.43293212890625,
320
  "step": 1050
321
  },
322
  {
323
+ "epoch": 5.321782178217822,
324
+ "grad_norm": 15.34720516204834,
325
+ "learning_rate": 9.97347819210148e-07,
326
+ "loss": 4.4848849487304685,
327
  "step": 1075
328
  },
329
  {
330
+ "epoch": 5.445544554455446,
331
+ "grad_norm": 10.895915031433105,
332
+ "learning_rate": 9.971119894480344e-07,
333
+ "loss": 4.418946228027344,
334
  "step": 1100
335
  },
336
  {
337
+ "epoch": 5.569306930693069,
338
+ "grad_norm": 13.77745532989502,
339
+ "learning_rate": 9.96866146848281e-07,
340
+ "loss": 4.675699768066406,
341
  "step": 1125
342
  },
343
  {
344
+ "epoch": 5.693069306930693,
345
+ "grad_norm": 21.53974723815918,
346
+ "learning_rate": 9.96610296362653e-07,
347
+ "loss": 4.483356628417969,
348
  "step": 1150
349
  },
350
  {
351
+ "epoch": 5.816831683168317,
352
+ "grad_norm": 14.47573184967041,
353
+ "learning_rate": 9.963444431444948e-07,
354
+ "loss": 4.555756530761719,
355
  "step": 1175
356
  },
357
  {
358
+ "epoch": 5.9405940594059405,
359
+ "grad_norm": 11.701032638549805,
360
+ "learning_rate": 9.960685925486265e-07,
361
+ "loss": 4.419072265625,
362
  "step": 1200
363
  },
364
  {
365
+ "epoch": 6.064356435643564,
366
+ "grad_norm": 17.197729110717773,
367
+ "learning_rate": 9.957827501312352e-07,
368
+ "loss": 4.375239562988281,
369
  "step": 1225
370
  },
371
  {
372
+ "epoch": 6.188118811881188,
373
+ "grad_norm": 12.936083793640137,
374
+ "learning_rate": 9.954869216497633e-07,
375
+ "loss": 4.3699853515625,
376
  "step": 1250
377
  },
378
  {
379
+ "epoch": 6.311881188118812,
380
+ "grad_norm": 15.114112854003906,
381
+ "learning_rate": 9.95181113062793e-07,
382
+ "loss": 4.41375732421875,
383
  "step": 1275
384
  },
385
  {
386
+ "epoch": 6.435643564356436,
387
+ "grad_norm": 17.72418785095215,
388
+ "learning_rate": 9.948653305299256e-07,
389
+ "loss": 4.430754089355469,
390
  "step": 1300
391
  },
392
  {
393
+ "epoch": 6.5594059405940595,
394
+ "grad_norm": 18.342178344726562,
395
+ "learning_rate": 9.945395804116578e-07,
396
+ "loss": 4.4040380859375,
397
  "step": 1325
398
  },
399
  {
400
+ "epoch": 6.683168316831683,
401
+ "grad_norm": 12.649271965026855,
402
+ "learning_rate": 9.94203869269254e-07,
403
+ "loss": 4.3536474609375,
404
  "step": 1350
405
  },
406
  {
407
+ "epoch": 6.806930693069307,
408
+ "grad_norm": 32.92485809326172,
409
+ "learning_rate": 9.938582038646131e-07,
410
+ "loss": 4.349139709472656,
411
  "step": 1375
412
  },
413
  {
414
+ "epoch": 6.930693069306931,
415
+ "grad_norm": 11.99206829071045,
416
+ "learning_rate": 9.935025911601334e-07,
417
+ "loss": 4.317200317382812,
418
  "step": 1400
419
  },
420
  {
421
+ "epoch": 7.054455445544554,
422
+ "grad_norm": 15.798395156860352,
423
+ "learning_rate": 9.931370383185716e-07,
424
+ "loss": 4.390576782226563,
425
  "step": 1425
426
  },
427
  {
428
+ "epoch": 7.178217821782178,
429
+ "grad_norm": 12.485000610351562,
430
+ "learning_rate": 9.92761552702899e-07,
431
+ "loss": 4.415244750976562,
432
  "step": 1450
433
  },
434
  {
435
+ "epoch": 7.301980198019802,
436
+ "grad_norm": 14.873079299926758,
437
+ "learning_rate": 9.923761418761526e-07,
438
+ "loss": 4.3524868774414065,
439
  "step": 1475
440
  },
441
  {
442
+ "epoch": 7.425742574257426,
443
+ "grad_norm": 13.601946830749512,
444
+ "learning_rate": 9.919808136012837e-07,
445
+ "loss": 4.294336547851563,
446
  "step": 1500
447
  },
448
  {
449
+ "epoch": 7.425742574257426,
450
+ "eval_loss": 0.9704130291938782,
451
+ "eval_runtime": 86.4979,
452
+ "eval_samples_per_second": 18.683,
453
+ "eval_steps_per_second": 4.671,
454
  "step": 1500
455
  },
456
  {
457
+ "epoch": 7.5495049504950495,
458
+ "grad_norm": 15.53011417388916,
459
+ "learning_rate": 9.915755758410008e-07,
460
+ "loss": 4.254639587402344,
461
  "step": 1525
462
  },
463
  {
464
+ "epoch": 7.673267326732673,
465
+ "grad_norm": 11.795248985290527,
466
+ "learning_rate": 9.91160436757609e-07,
467
+ "loss": 4.307816467285156,
468
  "step": 1550
469
  },
470
  {
471
+ "epoch": 7.797029702970297,
472
+ "grad_norm": 15.00704574584961,
473
+ "learning_rate": 9.907354047128464e-07,
474
+ "loss": 4.349948120117188,
475
  "step": 1575
476
  },
477
  {
478
+ "epoch": 7.920792079207921,
479
+ "grad_norm": 13.17168140411377,
480
+ "learning_rate": 9.903004882677155e-07,
481
+ "loss": 4.2257516479492185,
482
  "step": 1600
483
  },
484
  {
485
+ "epoch": 8.044554455445544,
486
+ "grad_norm": 13.770243644714355,
487
+ "learning_rate": 9.898556961823103e-07,
488
+ "loss": 4.310787048339844,
489
  "step": 1625
490
  },
491
  {
492
+ "epoch": 8.168316831683168,
493
+ "grad_norm": 13.734089851379395,
494
+ "learning_rate": 9.894010374156398e-07,
495
+ "loss": 4.148223266601563,
496
  "step": 1650
497
  },
498
  {
499
+ "epoch": 8.292079207920793,
500
+ "grad_norm": 15.90095329284668,
501
+ "learning_rate": 9.889365211254485e-07,
502
+ "loss": 4.3510107421875,
503
  "step": 1675
504
  },
505
  {
506
+ "epoch": 8.415841584158416,
507
+ "grad_norm": 12.133323669433594,
508
+ "learning_rate": 9.884621566680306e-07,
509
+ "loss": 4.337423400878906,
510
  "step": 1700
511
  },
512
  {
513
+ "epoch": 8.53960396039604,
514
+ "grad_norm": 50.35639190673828,
515
+ "learning_rate": 9.87977953598043e-07,
516
+ "loss": 4.3044869995117185,
517
  "step": 1725
518
  },
519
  {
520
+ "epoch": 8.663366336633663,
521
+ "grad_norm": 12.457736015319824,
522
+ "learning_rate": 9.874839216683112e-07,
523
+ "loss": 4.3085205078125,
524
  "step": 1750
525
  },
526
  {
527
+ "epoch": 8.787128712871286,
528
+ "grad_norm": 14.483731269836426,
529
+ "learning_rate": 9.869800708296346e-07,
530
+ "loss": 4.1831143188476565,
531
  "step": 1775
532
  },
533
  {
534
+ "epoch": 8.910891089108912,
535
+ "grad_norm": 11.174741744995117,
536
+ "learning_rate": 9.864664112305846e-07,
537
+ "loss": 4.162017822265625,
538
  "step": 1800
539
  },
540
  {
541
+ "epoch": 9.034653465346535,
542
+ "grad_norm": 10.312405586242676,
543
+ "learning_rate": 9.859429532173012e-07,
544
+ "loss": 4.320256958007812,
545
  "step": 1825
546
  },
547
  {
548
+ "epoch": 9.158415841584159,
549
+ "grad_norm": 13.814125061035156,
550
+ "learning_rate": 9.85409707333284e-07,
551
+ "loss": 4.314511413574219,
552
  "step": 1850
553
  },
554
  {
555
+ "epoch": 9.282178217821782,
556
+ "grad_norm": 11.276887893676758,
557
+ "learning_rate": 9.848666843191804e-07,
558
+ "loss": 4.327219543457031,
559
  "step": 1875
560
  },
561
  {
562
+ "epoch": 9.405940594059405,
563
+ "grad_norm": 13.48475170135498,
564
+ "learning_rate": 9.84313895112568e-07,
565
+ "loss": 4.2580517578125,
566
  "step": 1900
567
  },
568
  {
569
+ "epoch": 9.52970297029703,
570
+ "grad_norm": 13.70500659942627,
571
+ "learning_rate": 9.837513508477365e-07,
572
+ "loss": 4.269812316894531,
573
  "step": 1925
574
  },
575
  {
576
+ "epoch": 9.653465346534654,
577
+ "grad_norm": 12.701525688171387,
578
+ "learning_rate": 9.831790628554612e-07,
579
+ "loss": 4.257364196777344,
580
  "step": 1950
581
  },
582
  {
583
+ "epoch": 9.777227722772277,
584
+ "grad_norm": 11.086860656738281,
585
+ "learning_rate": 9.82597042662776e-07,
586
+ "loss": 4.120067443847656,
587
  "step": 1975
588
  },
589
  {
590
+ "epoch": 9.900990099009901,
591
+ "grad_norm": 14.811179161071777,
592
+ "learning_rate": 9.820053019927408e-07,
593
+ "loss": 4.13207275390625,
594
  "step": 2000
595
  },
596
  {
597
+ "epoch": 9.900990099009901,
598
+ "eval_loss": 0.9480290412902832,
599
+ "eval_runtime": 81.9519,
600
+ "eval_samples_per_second": 19.719,
601
+ "eval_steps_per_second": 4.93,
602
  "step": 2000
603
  },
604
  {
605
+ "epoch": 10.024752475247524,
606
+ "grad_norm": 13.266257286071777,
607
+ "learning_rate": 9.814038527642059e-07,
608
+ "loss": 4.281947021484375,
609
  "step": 2025
610
  },
611
  {
612
+ "epoch": 10.148514851485148,
613
+ "grad_norm": 13.618546485900879,
614
+ "learning_rate": 9.80792707091571e-07,
615
+ "loss": 4.166450500488281,
616
  "step": 2050
617
  },
618
  {
619
+ "epoch": 10.272277227722773,
620
+ "grad_norm": 12.208344459533691,
621
+ "learning_rate": 9.801718772845422e-07,
622
+ "loss": 4.151670227050781,
623
  "step": 2075
624
  },
625
  {
626
+ "epoch": 10.396039603960396,
627
+ "grad_norm": 12.220300674438477,
628
+ "learning_rate": 9.795413758478835e-07,
629
+ "loss": 4.250581359863281,
630
  "step": 2100
631
  },
632
  {
633
+ "epoch": 10.51980198019802,
634
+ "grad_norm": 10.937910079956055,
635
+ "learning_rate": 9.789012154811646e-07,
636
+ "loss": 4.219160461425782,
637
  "step": 2125
638
  },
639
  {
640
+ "epoch": 10.643564356435643,
641
+ "grad_norm": 12.94577407836914,
642
+ "learning_rate": 9.78251409078506e-07,
643
+ "loss": 4.256142272949218,
644
  "step": 2150
645
  },
646
  {
647
+ "epoch": 10.767326732673267,
648
+ "grad_norm": 12.101743698120117,
649
+ "learning_rate": 9.77591969728319e-07,
650
+ "loss": 4.145942077636719,
651
  "step": 2175
652
  },
653
  {
654
+ "epoch": 10.891089108910892,
655
+ "grad_norm": 15.079996109008789,
656
+ "learning_rate": 9.769229107130408e-07,
657
+ "loss": 4.2266848754882815,
658
  "step": 2200
659
  },
660
  {
661
+ "epoch": 11.014851485148515,
662
+ "grad_norm": 13.785998344421387,
663
+ "learning_rate": 9.762442455088694e-07,
664
+ "loss": 4.173150634765625,
665
  "step": 2225
666
  },
667
  {
668
+ "epoch": 11.138613861386139,
669
+ "grad_norm": 33.18384552001953,
670
+ "learning_rate": 9.7555598778549e-07,
671
+ "loss": 4.193008728027344,
672
  "step": 2250
673
  },
674
  {
675
+ "epoch": 11.262376237623762,
676
+ "grad_norm": 17.419424057006836,
677
+ "learning_rate": 9.748581514058005e-07,
678
+ "loss": 4.254207153320312,
679
  "step": 2275
680
  },
681
  {
682
+ "epoch": 11.386138613861386,
683
+ "grad_norm": 11.025071144104004,
684
+ "learning_rate": 9.741507504256326e-07,
685
+ "loss": 4.130965270996094,
686
  "step": 2300
687
  },
688
  {
689
+ "epoch": 11.509900990099009,
690
+ "grad_norm": 11.500884056091309,
691
+ "learning_rate": 9.734337990934681e-07,
692
+ "loss": 4.12019775390625,
693
  "step": 2325
694
  },
695
  {
696
+ "epoch": 11.633663366336634,
697
+ "grad_norm": 13.27956771850586,
698
+ "learning_rate": 9.727073118501523e-07,
699
+ "loss": 4.216612854003906,
700
  "step": 2350
701
  },
702
  {
703
+ "epoch": 11.757425742574258,
704
+ "grad_norm": 16.143634796142578,
705
+ "learning_rate": 9.719713033286028e-07,
706
+ "loss": 4.163922424316406,
707
  "step": 2375
708
  },
709
  {
710
+ "epoch": 11.881188118811881,
711
+ "grad_norm": 14.004642486572266,
712
+ "learning_rate": 9.712257883535152e-07,
713
+ "loss": 4.1760952758789065,
714
  "step": 2400
715
  },
716
  {
717
+ "epoch": 12.004950495049505,
718
+ "grad_norm": 11.351595878601074,
719
+ "learning_rate": 9.704707819410634e-07,
720
+ "loss": 4.229373168945313,
721
  "step": 2425
722
  },
723
  {
724
+ "epoch": 12.128712871287128,
725
+ "grad_norm": 9.855143547058105,
726
+ "learning_rate": 9.697062992985993e-07,
727
+ "loss": 4.096666259765625,
728
  "step": 2450
729
  },
730
  {
731
+ "epoch": 12.252475247524753,
732
+ "grad_norm": 11.369580268859863,
733
+ "learning_rate": 9.689323558243444e-07,
734
+ "loss": 4.218254089355469,
735
  "step": 2475
736
  },
737
  {
738
+ "epoch": 12.376237623762377,
739
+ "grad_norm": 9.929821014404297,
740
+ "learning_rate": 9.681489671070802e-07,
741
+ "loss": 4.054186401367187,
742
  "step": 2500
743
  },
744
  {
745
+ "epoch": 12.376237623762377,
746
+ "eval_loss": 0.9306696653366089,
747
+ "eval_runtime": 82.6001,
748
+ "eval_samples_per_second": 19.564,
749
+ "eval_steps_per_second": 4.891,
750
  "step": 2500
751
  },
752
  {
753
+ "epoch": 12.5,
754
+ "grad_norm": 23.94037628173828,
755
+ "learning_rate": 9.673561489258352e-07,
756
+ "loss": 4.108751220703125,
757
  "step": 2525
758
  },
759
  {
760
+ "epoch": 12.623762376237623,
761
+ "grad_norm": 10.571653366088867,
762
+ "learning_rate": 9.665539172495656e-07,
763
+ "loss": 4.077666320800781,
764
  "step": 2550
765
  },
766
  {
767
+ "epoch": 12.747524752475247,
768
+ "grad_norm": 12.341602325439453,
769
+ "learning_rate": 9.657422882368347e-07,
770
+ "loss": 4.187429504394531,
771
  "step": 2575
772
  },
773
  {
774
+ "epoch": 12.871287128712872,
775
+ "grad_norm": 9.840188026428223,
776
+ "learning_rate": 9.64921278235487e-07,
777
+ "loss": 4.135216369628906,
778
  "step": 2600
779
  },
780
  {
781
+ "epoch": 12.995049504950495,
782
+ "grad_norm": 11.124446868896484,
783
+ "learning_rate": 9.640909037823194e-07,
784
+ "loss": 4.087601318359375,
785
  "step": 2625
786
  },
787
  {
788
+ "epoch": 13.118811881188119,
789
+ "grad_norm": 11.180383682250977,
790
+ "learning_rate": 9.63251181602747e-07,
791
+ "loss": 4.0938818359375,
792
  "step": 2650
793
  },
794
  {
795
+ "epoch": 13.242574257425742,
796
+ "grad_norm": 12.539691925048828,
797
+ "learning_rate": 9.624021286104674e-07,
798
+ "loss": 4.085841369628906,
799
  "step": 2675
800
  },
801
  {
802
+ "epoch": 13.366336633663366,
803
+ "grad_norm": 11.089624404907227,
804
+ "learning_rate": 9.615437619071198e-07,
805
+ "loss": 4.057616577148438,
806
  "step": 2700
807
  },
808
  {
809
+ "epoch": 13.490099009900991,
810
+ "grad_norm": 10.118108749389648,
811
+ "learning_rate": 9.6067609878194e-07,
812
+ "loss": 4.123738403320313,
813
  "step": 2725
814
  },
815
  {
816
+ "epoch": 13.613861386138614,
817
+ "grad_norm": 13.25112247467041,
818
+ "learning_rate": 9.597991567114124e-07,
819
+ "loss": 4.035686645507813,
820
  "step": 2750
821
  },
822
  {
823
+ "epoch": 13.737623762376238,
824
+ "grad_norm": 10.743168830871582,
825
+ "learning_rate": 9.589129533589185e-07,
826
+ "loss": 4.089259643554687,
827
  "step": 2775
828
  },
829
  {
830
+ "epoch": 13.861386138613861,
831
+ "grad_norm": 13.900023460388184,
832
+ "learning_rate": 9.580175065743798e-07,
833
+ "loss": 4.086239929199219,
834
  "step": 2800
835
  },
836
  {
837
+ "epoch": 13.985148514851485,
838
+ "grad_norm": 14.214149475097656,
839
+ "learning_rate": 9.571128343939005e-07,
840
+ "loss": 4.133528747558594,
841
  "step": 2825
842
  },
843
  {
844
+ "epoch": 14.108910891089108,
845
+ "grad_norm": 13.158830642700195,
846
+ "learning_rate": 9.561989550394016e-07,
847
+ "loss": 4.137518005371094,
848
  "step": 2850
849
  },
850
  {
851
+ "epoch": 14.232673267326733,
852
+ "grad_norm": 13.256293296813965,
853
+ "learning_rate": 9.552758869182557e-07,
854
+ "loss": 4.049294128417968,
855
  "step": 2875
856
  },
857
  {
858
+ "epoch": 14.356435643564357,
859
+ "grad_norm": 10.833374977111816,
860
+ "learning_rate": 9.543436486229161e-07,
861
+ "loss": 4.0441677856445315,
862
  "step": 2900
863
  },
864
  {
865
+ "epoch": 14.48019801980198,
866
+ "grad_norm": 12.035900115966797,
867
+ "learning_rate": 9.534022589305411e-07,
868
+ "loss": 4.099019775390625,
869
  "step": 2925
870
  },
871
  {
872
+ "epoch": 14.603960396039604,
873
+ "grad_norm": 11.711991310119629,
874
+ "learning_rate": 9.524517368026168e-07,
875
+ "loss": 4.064933166503907,
876
  "step": 2950
877
  },
878
  {
879
+ "epoch": 14.727722772277227,
880
+ "grad_norm": 20.470388412475586,
881
+ "learning_rate": 9.514921013845754e-07,
882
+ "loss": 4.110359802246093,
883
  "step": 2975
884
  },
885
  {
886
+ "epoch": 14.851485148514852,
887
+ "grad_norm": 12.352797508239746,
888
+ "learning_rate": 9.505233720054085e-07,
889
+ "loss": 4.122251892089844,
890
  "step": 3000
891
  },
892
  {
893
+ "epoch": 14.851485148514852,
894
+ "eval_loss": 0.920803427696228,
895
+ "eval_runtime": 86.0706,
896
+ "eval_samples_per_second": 18.775,
897
+ "eval_steps_per_second": 4.694,
898
  "step": 3000
899
  },
900
  {
901
+ "epoch": 14.975247524752476,
902
+ "grad_norm": 12.874964714050293,
903
+ "learning_rate": 9.495455681772791e-07,
904
+ "loss": 4.0896240234375,
905
  "step": 3025
906
  },
907
  {
908
+ "epoch": 15.099009900990099,
909
+ "grad_norm": 12.206281661987305,
910
+ "learning_rate": 9.485587095951275e-07,
911
+ "loss": 4.052577819824219,
912
  "step": 3050
913
  },
914
  {
915
+ "epoch": 15.222772277227723,
916
+ "grad_norm": 11.373126983642578,
917
+ "learning_rate": 9.475628161362747e-07,
918
+ "loss": 4.107190551757813,
919
  "step": 3075
920
  },
921
  {
922
+ "epoch": 15.346534653465346,
923
+ "grad_norm": 10.882997512817383,
924
+ "learning_rate": 9.465579078600232e-07,
925
+ "loss": 4.058759155273438,
926
  "step": 3100
927
  },
928
  {
929
+ "epoch": 15.47029702970297,
930
+ "grad_norm": 14.810848236083984,
931
+ "learning_rate": 9.455440050072511e-07,
932
+ "loss": 4.046248168945312,
933
  "step": 3125
934
  },
935
  {
936
+ "epoch": 15.594059405940595,
937
+ "grad_norm": 13.74386215209961,
938
+ "learning_rate": 9.44521128000006e-07,
939
+ "loss": 4.134948120117188,
940
  "step": 3150
941
  },
942
  {
943
+ "epoch": 15.717821782178218,
944
+ "grad_norm": 14.098150253295898,
945
+ "learning_rate": 9.434892974410932e-07,
946
+ "loss": 3.9966055297851564,
947
  "step": 3175
948
  },
949
  {
950
+ "epoch": 15.841584158415841,
951
+ "grad_norm": 10.933466911315918,
952
+ "learning_rate": 9.424485341136599e-07,
953
+ "loss": 4.099876708984375,
954
  "step": 3200
955
  },
956
  {
957
+ "epoch": 15.965346534653465,
958
+ "grad_norm": 9.829030990600586,
959
+ "learning_rate": 9.41398858980778e-07,
960
+ "loss": 3.9940283203125,
961
  "step": 3225
962
  },
963
  {
964
+ "epoch": 16.08910891089109,
965
+ "grad_norm": 10.526725769042969,
966
+ "learning_rate": 9.403402931850207e-07,
967
+ "loss": 4.016290588378906,
968
  "step": 3250
969
  },
970
  {
971
+ "epoch": 16.212871287128714,
972
+ "grad_norm": 9.416884422302246,
973
+ "learning_rate": 9.392728580480371e-07,
974
+ "loss": 4.113368835449219,
975
  "step": 3275
976
  },
977
  {
978
+ "epoch": 16.336633663366335,
979
+ "grad_norm": 11.515923500061035,
980
+ "learning_rate": 9.381965750701228e-07,
981
+ "loss": 4.077259216308594,
982
  "step": 3300
983
  },
984
  {
985
+ "epoch": 16.46039603960396,
986
+ "grad_norm": 12.539885520935059,
987
+ "learning_rate": 9.37111465929787e-07,
988
+ "loss": 3.9970358276367186,
989
  "step": 3325
990
  },
991
  {
992
+ "epoch": 16.584158415841586,
993
+ "grad_norm": 11.099287986755371,
994
+ "learning_rate": 9.360175524833151e-07,
995
+ "loss": 4.0322119140625,
996
  "step": 3350
997
  },
998
  {
999
+ "epoch": 16.707920792079207,
1000
+ "grad_norm": 12.52721118927002,
1001
+ "learning_rate": 9.349148567643294e-07,
1002
+ "loss": 4.074447937011719,
1003
  "step": 3375
1004
  },
1005
  {
1006
+ "epoch": 16.831683168316832,
1007
+ "grad_norm": 9.58317756652832,
1008
+ "learning_rate": 9.338034009833445e-07,
1009
+ "loss": 4.086669921875,
1010
  "step": 3400
1011
  },
1012
  {
1013
+ "epoch": 16.955445544554454,
1014
+ "grad_norm": 11.126392364501953,
1015
+ "learning_rate": 9.326832075273207e-07,
1016
+ "loss": 4.079499206542969,
1017
  "step": 3425
1018
  },
1019
  {
1020
+ "epoch": 17.07920792079208,
1021
+ "grad_norm": 11.195257186889648,
1022
+ "learning_rate": 9.315542989592124e-07,
1023
+ "loss": 4.001406555175781,
1024
  "step": 3450
1025
  },
1026
  {
1027
+ "epoch": 17.202970297029704,
1028
+ "grad_norm": 11.249403953552246,
1029
+ "learning_rate": 9.304166980175141e-07,
1030
+ "loss": 4.042621765136719,
1031
  "step": 3475
1032
  },
1033
  {
1034
+ "epoch": 17.326732673267326,
1035
+ "grad_norm": 22.27759552001953,
1036
+ "learning_rate": 9.292704276158021e-07,
1037
+ "loss": 4.025510559082031,
1038
  "step": 3500
1039
  },
1040
  {
1041
+ "epoch": 17.326732673267326,
1042
+ "eval_loss": 0.9127047657966614,
1043
+ "eval_runtime": 81.0553,
1044
+ "eval_samples_per_second": 19.937,
1045
+ "eval_steps_per_second": 4.984,
1046
  "step": 3500
1047
  },
1048
  {
1049
+ "epoch": 17.45049504950495,
1050
+ "grad_norm": 10.390947341918945,
1051
+ "learning_rate": 9.281155108422731e-07,
1052
+ "loss": 4.0325146484375,
1053
  "step": 3525
1054
  },
1055
  {
1056
+ "epoch": 17.574257425742573,
1057
+ "grad_norm": 11.612289428710938,
1058
+ "learning_rate": 9.269519709592794e-07,
1059
+ "loss": 4.141344909667969,
1060
  "step": 3550
1061
  },
1062
  {
1063
+ "epoch": 17.698019801980198,
1064
+ "grad_norm": 15.810525894165039,
1065
+ "learning_rate": 9.257798314028599e-07,
1066
+ "loss": 4.035428771972656,
1067
  "step": 3575
1068
  },
1069
  {
1070
+ "epoch": 17.821782178217823,
1071
+ "grad_norm": 17.175111770629883,
1072
+ "learning_rate": 9.245991157822681e-07,
1073
+ "loss": 4.041367492675781,
1074
  "step": 3600
1075
  },
1076
  {
1077
+ "epoch": 17.945544554455445,
1078
+ "grad_norm": 10.926711082458496,
1079
+ "learning_rate": 9.234098478794969e-07,
1080
+ "loss": 4.121484069824219,
1081
  "step": 3625
1082
  },
1083
  {
1084
+ "epoch": 18.06930693069307,
1085
+ "grad_norm": 11.38853931427002,
1086
+ "learning_rate": 9.222120516487997e-07,
1087
+ "loss": 3.9692178344726563,
1088
  "step": 3650
1089
  },
1090
  {
1091
+ "epoch": 18.193069306930692,
1092
+ "grad_norm": 11.953936576843262,
1093
+ "learning_rate": 9.210057512162072e-07,
1094
+ "loss": 4.020705261230469,
1095
  "step": 3675
1096
  },
1097
  {
1098
+ "epoch": 18.316831683168317,
1099
+ "grad_norm": 13.49695110321045,
1100
+ "learning_rate": 9.197909708790419e-07,
1101
+ "loss": 4.005714721679688,
1102
  "step": 3700
1103
  },
1104
  {
1105
+ "epoch": 18.440594059405942,
1106
+ "grad_norm": 12.650299072265625,
1107
+ "learning_rate": 9.185677351054289e-07,
1108
+ "loss": 4.006802062988282,
1109
  "step": 3725
1110
  },
1111
  {
1112
+ "epoch": 18.564356435643564,
1113
+ "grad_norm": 14.999792098999023,
1114
+ "learning_rate": 9.173360685338026e-07,
1115
+ "loss": 3.983961486816406,
1116
  "step": 3750
1117
  },
1118
  {
1119
+ "epoch": 18.68811881188119,
1120
+ "grad_norm": 8.753050804138184,
1121
+ "learning_rate": 9.160959959724109e-07,
1122
+ "loss": 4.121242980957032,
1123
  "step": 3775
1124
  },
1125
  {
1126
+ "epoch": 18.81188118811881,
1127
+ "grad_norm": 11.767355918884277,
1128
+ "learning_rate": 9.148475423988149e-07,
1129
+ "loss": 3.988392333984375,
1130
  "step": 3800
1131
  },
1132
  {
1133
+ "epoch": 18.935643564356436,
1134
+ "grad_norm": 13.041923522949219,
1135
+ "learning_rate": 9.135907329593869e-07,
1136
+ "loss": 4.037374877929688,
1137
  "step": 3825
1138
  },
1139
  {
1140
+ "epoch": 19.059405940594058,
1141
+ "grad_norm": 17.08255386352539,
1142
+ "learning_rate": 9.123255929688023e-07,
1143
+ "loss": 4.015922241210937,
1144
  "step": 3850
1145
  },
1146
  {
1147
+ "epoch": 19.183168316831683,
1148
+ "grad_norm": 10.86015796661377,
1149
+ "learning_rate": 9.110521479095312e-07,
1150
+ "loss": 4.031682739257812,
1151
  "step": 3875
1152
  },
1153
  {
1154
+ "epoch": 19.306930693069308,
1155
+ "grad_norm": 14.082168579101562,
1156
+ "learning_rate": 9.097704234313243e-07,
1157
+ "loss": 3.973893737792969,
1158
  "step": 3900
1159
  },
1160
  {
1161
+ "epoch": 19.43069306930693,
1162
+ "grad_norm": 12.722586631774902,
1163
+ "learning_rate": 9.084804453506966e-07,
1164
+ "loss": 4.083378295898438,
1165
  "step": 3925
1166
  },
1167
  {
1168
+ "epoch": 19.554455445544555,
1169
+ "grad_norm": 12.841690063476562,
1170
+ "learning_rate": 9.071822396504072e-07,
1171
+ "loss": 4.002433166503907,
1172
  "step": 3950
1173
  },
1174
  {
1175
+ "epoch": 19.678217821782177,
1176
+ "grad_norm": 11.057418823242188,
1177
+ "learning_rate": 9.058758324789359e-07,
1178
+ "loss": 4.0087548828125,
1179
  "step": 3975
1180
  },
1181
  {
1182
+ "epoch": 19.801980198019802,
1183
+ "grad_norm": 10.621847152709961,
1184
+ "learning_rate": 9.045612501499573e-07,
1185
+ "loss": 3.928231201171875,
1186
  "step": 4000
1187
  },
1188
  {
1189
+ "epoch": 19.801980198019802,
1190
+ "eval_loss": 0.9033195376396179,
1191
+ "eval_runtime": 84.5771,
1192
+ "eval_samples_per_second": 19.107,
1193
+ "eval_steps_per_second": 4.777,
1194
  "step": 4000
1195
  },
1196
  {
1197
+ "epoch": 19.925742574257427,
1198
+ "grad_norm": 10.820324897766113,
1199
+ "learning_rate": 9.032385191418093e-07,
1200
+ "loss": 4.026563720703125,
1201
  "step": 4025
1202
  },
1203
  {
1204
+ "epoch": 20.04950495049505,
1205
+ "grad_norm": 12.08041763305664,
1206
+ "learning_rate": 9.019076660969609e-07,
1207
+ "loss": 3.98918212890625,
1208
  "step": 4050
1209
  },
1210
  {
1211
+ "epoch": 20.173267326732674,
1212
+ "grad_norm": 12.175297737121582,
1213
+ "learning_rate": 9.005687178214758e-07,
1214
+ "loss": 3.962005615234375,
1215
  "step": 4075
1216
  },
1217
  {
1218
+ "epoch": 20.297029702970296,
1219
+ "grad_norm": 15.371024131774902,
1220
+ "learning_rate": 8.992217012844713e-07,
1221
+ "loss": 3.970758972167969,
1222
  "step": 4100
1223
  },
1224
  {
1225
+ "epoch": 20.42079207920792,
1226
+ "grad_norm": 12.102559089660645,
1227
+ "learning_rate": 8.978666436175761e-07,
1228
+ "loss": 3.9590899658203127,
1229
  "step": 4125
1230
  },
1231
  {
1232
+ "epoch": 20.544554455445546,
1233
+ "grad_norm": 11.632902145385742,
1234
+ "learning_rate": 8.965035721143834e-07,
1235
+ "loss": 3.9882553100585936,
1236
  "step": 4150
1237
  },
1238
  {
1239
+ "epoch": 20.668316831683168,
1240
+ "grad_norm": 14.134031295776367,
1241
+ "learning_rate": 8.951325142299014e-07,
1242
+ "loss": 3.931231689453125,
1243
  "step": 4175
1244
  },
1245
  {
1246
+ "epoch": 20.792079207920793,
1247
+ "grad_norm": 11.716127395629883,
1248
+ "learning_rate": 8.937534975799996e-07,
1249
+ "loss": 3.9607308959960936,
1250
  "step": 4200
1251
  },
1252
  {
1253
+ "epoch": 20.915841584158414,
1254
+ "grad_norm": 11.062108993530273,
1255
+ "learning_rate": 8.923665499408535e-07,
1256
+ "loss": 3.955765380859375,
1257
  "step": 4225
1258
  },
1259
  {
1260
+ "epoch": 21.03960396039604,
1261
+ "grad_norm": 10.680591583251953,
1262
+ "learning_rate": 8.909716992483847e-07,
1263
+ "loss": 3.8945526123046874,
1264
  "step": 4250
1265
  },
1266
  {
1267
+ "epoch": 21.163366336633665,
1268
+ "grad_norm": 13.237275123596191,
1269
+ "learning_rate": 8.895689735976983e-07,
1270
+ "loss": 3.956689758300781,
1271
  "step": 4275
1272
  },
1273
  {
1274
+ "epoch": 21.287128712871286,
1275
+ "grad_norm": 17.047996520996094,
1276
+ "learning_rate": 8.881584012425165e-07,
1277
+ "loss": 3.9637225341796873,
1278
  "step": 4300
1279
  },
1280
  {
1281
+ "epoch": 21.41089108910891,
1282
+ "grad_norm": 13.230228424072266,
1283
+ "learning_rate": 8.867400105946103e-07,
1284
+ "loss": 3.951360168457031,
1285
  "step": 4325
1286
  },
1287
  {
1288
+ "epoch": 21.534653465346533,
1289
+ "grad_norm": 12.366623878479004,
1290
+ "learning_rate": 8.853138302232271e-07,
1291
+ "loss": 3.9974639892578123,
1292
  "step": 4350
1293
  },
1294
  {
1295
+ "epoch": 21.65841584158416,
1296
+ "grad_norm": 13.02001667022705,
1297
+ "learning_rate": 8.838798888545143e-07,
1298
+ "loss": 3.9881216430664064,
1299
  "step": 4375
1300
  },
1301
  {
1302
+ "epoch": 21.782178217821784,
1303
+ "grad_norm": 12.130973815917969,
1304
+ "learning_rate": 8.824382153709421e-07,
1305
+ "loss": 3.9237442016601562,
1306
  "step": 4400
1307
  },
1308
  {
1309
+ "epoch": 21.905940594059405,
1310
+ "grad_norm": 11.393623352050781,
1311
+ "learning_rate": 8.809888388107209e-07,
1312
+ "loss": 4.023025207519531,
1313
  "step": 4425
1314
  },
1315
  {
1316
+ "epoch": 22.02970297029703,
1317
+ "grad_norm": 10.824515342712402,
1318
+ "learning_rate": 8.795317883672165e-07,
1319
+ "loss": 3.9387527465820313,
1320
  "step": 4450
1321
  },
1322
  {
1323
+ "epoch": 22.153465346534652,
1324
+ "grad_norm": 20.875307083129883,
1325
+ "learning_rate": 8.780670933883622e-07,
1326
+ "loss": 3.97408203125,
1327
  "step": 4475
1328
  },
1329
  {
1330
+ "epoch": 22.277227722772277,
1331
+ "grad_norm": 17.839189529418945,
1332
+ "learning_rate": 8.765947833760677e-07,
1333
+ "loss": 4.018715209960938,
1334
  "step": 4500
1335
  },
1336
  {
1337
+ "epoch": 22.277227722772277,
1338
+ "eval_loss": 0.905015230178833,
1339
+ "eval_runtime": 84.5389,
1340
+ "eval_samples_per_second": 19.115,
1341
+ "eval_steps_per_second": 4.779,
1342
  "step": 4500
1343
  },
1344
  {
1345
+ "epoch": 22.400990099009903,
1346
+ "grad_norm": 13.415910720825195,
1347
+ "learning_rate": 8.751148879856249e-07,
1348
+ "loss": 3.94878173828125,
1349
  "step": 4525
1350
  },
1351
  {
1352
+ "epoch": 22.524752475247524,
1353
+ "grad_norm": 11.05270767211914,
1354
+ "learning_rate": 8.736274370251103e-07,
1355
+ "loss": 3.9672274780273438,
1356
  "step": 4550
1357
  },
1358
  {
1359
+ "epoch": 22.64851485148515,
1360
+ "grad_norm": 11.800627708435059,
1361
+ "learning_rate": 8.721324604547849e-07,
1362
+ "loss": 3.927026672363281,
1363
  "step": 4575
1364
  },
1365
  {
1366
+ "epoch": 22.77227722772277,
1367
+ "grad_norm": 10.663934707641602,
1368
+ "learning_rate": 8.706299883864909e-07,
1369
+ "loss": 3.9203482055664063,
1370
  "step": 4600
1371
  },
1372
  {
1373
+ "epoch": 22.896039603960396,
1374
+ "grad_norm": 16.535402297973633,
1375
+ "learning_rate": 8.691200510830444e-07,
1376
+ "loss": 3.9108685302734374,
1377
  "step": 4625
1378
  },
1379
  {
1380
+ "epoch": 23.019801980198018,
1381
+ "grad_norm": 9.973243713378906,
1382
+ "learning_rate": 8.676026789576269e-07,
1383
+ "loss": 3.960736083984375,
1384
  "step": 4650
1385
  },
1386
  {
1387
+ "epoch": 23.143564356435643,
1388
+ "grad_norm": 21.756301879882812,
1389
+ "learning_rate": 8.660779025731721e-07,
1390
+ "loss": 3.991618957519531,
1391
  "step": 4675
1392
  },
1393
  {
1394
+ "epoch": 23.26732673267327,
1395
+ "grad_norm": 12.650042533874512,
1396
+ "learning_rate": 8.645457526417503e-07,
1397
+ "loss": 3.9313955688476563,
1398
  "step": 4700
1399
  },
1400
  {
1401
+ "epoch": 23.39108910891089,
1402
+ "grad_norm": 10.731987953186035,
1403
+ "learning_rate": 8.630062600239496e-07,
1404
+ "loss": 3.90642333984375,
1405
  "step": 4725
1406
  },
1407
  {
1408
+ "epoch": 23.514851485148515,
1409
+ "grad_norm": 13.895561218261719,
1410
+ "learning_rate": 8.614594557282553e-07,
1411
+ "loss": 3.871036376953125,
1412
  "step": 4750
1413
  },
1414
  {
1415
+ "epoch": 23.638613861386137,
1416
+ "grad_norm": 15.030277252197266,
1417
+ "learning_rate": 8.599053709104241e-07,
1418
+ "loss": 3.9301937866210936,
1419
  "step": 4775
1420
  },
1421
  {
1422
+ "epoch": 23.762376237623762,
1423
+ "grad_norm": 12.624570846557617,
1424
+ "learning_rate": 8.583440368728572e-07,
1425
+ "loss": 3.915058898925781,
1426
  "step": 4800
1427
  },
1428
  {
1429
+ "epoch": 23.886138613861387,
1430
+ "grad_norm": 11.839984893798828,
1431
+ "learning_rate": 8.567754850639699e-07,
1432
+ "loss": 3.924931640625,
1433
  "step": 4825
1434
  },
1435
  {
1436
+ "epoch": 24.00990099009901,
1437
+ "grad_norm": 13.87601089477539,
1438
+ "learning_rate": 8.551997470775576e-07,
1439
+ "loss": 3.9027645874023436,
1440
  "step": 4850
1441
  },
1442
  {
1443
+ "epoch": 24.133663366336634,
1444
+ "grad_norm": 11.400012969970703,
1445
+ "learning_rate": 8.536168546521602e-07,
1446
+ "loss": 3.9296466064453126,
1447
  "step": 4875
1448
  },
1449
  {
1450
+ "epoch": 24.257425742574256,
1451
+ "grad_norm": 11.829928398132324,
1452
+ "learning_rate": 8.520268396704222e-07,
1453
+ "loss": 3.993326416015625,
1454
  "step": 4900
1455
  },
1456
  {
1457
+ "epoch": 24.38118811881188,
1458
+ "grad_norm": 11.399077415466309,
1459
+ "learning_rate": 8.504297341584508e-07,
1460
+ "loss": 3.891920471191406,
1461
  "step": 4925
1462
  },
1463
  {
1464
+ "epoch": 24.504950495049506,
1465
+ "grad_norm": 13.8989839553833,
1466
+ "learning_rate": 8.488255702851707e-07,
1467
+ "loss": 3.9781057739257815,
1468
  "step": 4950
1469
  },
1470
  {
1471
+ "epoch": 24.628712871287128,
1472
+ "grad_norm": 12.6453218460083,
1473
+ "learning_rate": 8.472143803616764e-07,
1474
+ "loss": 3.9407168579101564,
1475
  "step": 4975
1476
  },
1477
  {
1478
+ "epoch": 24.752475247524753,
1479
+ "grad_norm": 9.551429748535156,
1480
+ "learning_rate": 8.45596196840581e-07,
1481
+ "loss": 3.9068170166015626,
1482
  "step": 5000
1483
  },
1484
  {
1485
+ "epoch": 24.752475247524753,
1486
+ "eval_loss": 0.8924599885940552,
1487
+ "eval_runtime": 85.1496,
1488
+ "eval_samples_per_second": 18.978,
1489
+ "eval_steps_per_second": 4.745,
1490
  "step": 5000
1491
  },
1492
  {
1493
+ "epoch": 24.876237623762375,
1494
+ "grad_norm": 10.968220710754395,
1495
+ "learning_rate": 8.439710523153627e-07,
1496
+ "loss": 3.960721740722656,
1497
  "step": 5025
1498
  },
1499
  {
1500
  "epoch": 25.0,
1501
+ "grad_norm": 16.728450775146484,
1502
+ "learning_rate": 8.423389795197088e-07,
1503
+ "loss": 3.8826409912109376,
1504
  "step": 5050
1505
  },
1506
  {
1507
+ "epoch": 25.123762376237625,
1508
+ "grad_norm": 10.608399391174316,
1509
+ "learning_rate": 8.407000113268555e-07,
1510
+ "loss": 3.933037109375,
1511
  "step": 5075
1512
  },
1513
  {
1514
+ "epoch": 25.247524752475247,
1515
+ "grad_norm": 11.371829986572266,
1516
+ "learning_rate": 8.390541807489264e-07,
1517
+ "loss": 3.9157955932617186,
1518
  "step": 5100
1519
  },
1520
  {
1521
+ "epoch": 25.371287128712872,
1522
+ "grad_norm": 9.11988639831543,
1523
+ "learning_rate": 8.374015209362674e-07,
1524
+ "loss": 3.8903854370117186,
1525
  "step": 5125
1526
  },
1527
  {
1528
+ "epoch": 25.495049504950494,
1529
+ "grad_norm": 16.707979202270508,
1530
+ "learning_rate": 8.357420651767785e-07,
1531
+ "loss": 3.883365478515625,
1532
  "step": 5150
1533
  },
1534
  {
1535
+ "epoch": 25.61881188118812,
1536
+ "grad_norm": 12.343544960021973,
1537
+ "learning_rate": 8.340758468952448e-07,
1538
+ "loss": 3.9449783325195313,
1539
  "step": 5175
1540
  },
1541
  {
1542
+ "epoch": 25.742574257425744,
1543
+ "grad_norm": 13.58958911895752,
1544
+ "learning_rate": 8.32402899652661e-07,
1545
+ "loss": 3.9113555908203126,
1546
  "step": 5200
1547
  },
1548
  {
1549
+ "epoch": 25.866336633663366,
1550
+ "grad_norm": 12.369688987731934,
1551
+ "learning_rate": 8.307232571455573e-07,
1552
+ "loss": 3.8758831787109376,
1553
  "step": 5225
1554
  },
1555
  {
1556
+ "epoch": 25.99009900990099,
1557
+ "grad_norm": 14.925145149230957,
1558
+ "learning_rate": 8.290369532053201e-07,
1559
+ "loss": 3.8888095092773436,
1560
  "step": 5250
1561
  },
1562
  {
1563
+ "epoch": 26.113861386138613,
1564
+ "grad_norm": 12.04305648803711,
1565
+ "learning_rate": 8.273440217975102e-07,
1566
+ "loss": 3.9101412963867186,
1567
  "step": 5275
1568
  },
1569
  {
1570
+ "epoch": 26.237623762376238,
1571
+ "grad_norm": 10.851409912109375,
1572
+ "learning_rate": 8.256444970211792e-07,
1573
+ "loss": 3.9206503295898436,
1574
  "step": 5300
1575
  },
1576
  {
1577
+ "epoch": 26.361386138613863,
1578
+ "grad_norm": 10.527778625488281,
1579
+ "learning_rate": 8.239384131081823e-07,
1580
+ "loss": 3.908270263671875,
1581
  "step": 5325
1582
  },
1583
  {
1584
+ "epoch": 26.485148514851485,
1585
+ "grad_norm": 13.652122497558594,
1586
+ "learning_rate": 8.222258044224888e-07,
1587
+ "loss": 3.8833071899414064,
1588
  "step": 5350
1589
  },
1590
  {
1591
+ "epoch": 26.60891089108911,
1592
+ "grad_norm": 12.982370376586914,
1593
+ "learning_rate": 8.205067054594904e-07,
1594
+ "loss": 3.902032470703125,
1595
  "step": 5375
1596
  },
1597
  {
1598
+ "epoch": 26.73267326732673,
1599
+ "grad_norm": 11.420575141906738,
1600
+ "learning_rate": 8.187811508453059e-07,
1601
+ "loss": 3.8902410888671874,
1602
  "step": 5400
1603
  },
1604
  {
1605
+ "epoch": 26.856435643564357,
1606
+ "grad_norm": 10.537010192871094,
1607
+ "learning_rate": 8.170491753360838e-07,
1608
+ "loss": 3.840379333496094,
1609
  "step": 5425
1610
  },
1611
  {
1612
+ "epoch": 26.980198019801982,
1613
+ "grad_norm": 10.516629219055176,
1614
+ "learning_rate": 8.153108138173026e-07,
1615
+ "loss": 3.8664096069335936,
1616
  "step": 5450
1617
  },
1618
  {
1619
+ "epoch": 27.103960396039604,
1620
+ "grad_norm": 14.301212310791016,
1621
+ "learning_rate": 8.135661013030676e-07,
1622
+ "loss": 4.034673461914062,
1623
  "step": 5475
1624
  },
1625
  {
1626
+ "epoch": 27.22772277227723,
1627
+ "grad_norm": 15.154522895812988,
1628
+ "learning_rate": 8.118150729354067e-07,
1629
+ "loss": 3.874417724609375,
1630
  "step": 5500
1631
  },
1632
  {
1633
+ "epoch": 27.22772277227723,
1634
+ "eval_loss": 0.8911440968513489,
1635
+ "eval_runtime": 85.0731,
1636
+ "eval_samples_per_second": 18.995,
1637
+ "eval_steps_per_second": 4.749,
1638
  "step": 5500
1639
  }
1640
  ],
 
1664
  "attributes": {}
1665
  }
1666
  },
1667
+ "total_flos": 2.3611779625046976e+16,
1668
+ "train_batch_size": 8,
1669
  "trial_name": null,
1670
  "trial_params": null
1671
  }
checkpoint-5500/training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bc3ae1f3188fc78f14b1e24c35f0f902ff13c8475d4503f269d542bb33a09e53
3
  size 5393
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d34645042a62af83117180e7d8c3a51dc32127e71ed6c053e7766610460e0b1
3
  size 5393