diff --git "a/trainer_state.json" "b/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/trainer_state.json"
@@ -0,0 +1,4588 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.9997305930800908,
+  "eval_steps": 500,
+  "global_step": 3247,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0003078936227533387,
+      "grad_norm": 42.473334342863225,
+      "learning_rate": 6.153846153846154e-07,
+      "loss": 1.3715,
+      "step": 1
+    },
+    {
+      "epoch": 0.0015394681137666935,
+      "grad_norm": 39.584233835268385,
+      "learning_rate": 3.0769230769230774e-06,
+      "loss": 1.3252,
+      "step": 5
+    },
+    {
+      "epoch": 0.003078936227533387,
+      "grad_norm": 38.124176994124966,
+      "learning_rate": 6.153846153846155e-06,
+      "loss": 1.3472,
+      "step": 10
+    },
+    {
+      "epoch": 0.004618404341300081,
+      "grad_norm": 33.949332930378546,
+      "learning_rate": 9.230769230769232e-06,
+      "loss": 1.3191,
+      "step": 15
+    },
+    {
+      "epoch": 0.006157872455066774,
+      "grad_norm": 13.263339730204837,
+      "learning_rate": 1.230769230769231e-05,
+      "loss": 1.3273,
+      "step": 20
+    },
+    {
+      "epoch": 0.007697340568833468,
+      "grad_norm": 10.03432034420483,
+      "learning_rate": 1.5384615384615387e-05,
+      "loss": 1.2756,
+      "step": 25
+    },
+    {
+      "epoch": 0.009236808682600161,
+      "grad_norm": 10.404219206269893,
+      "learning_rate": 1.8461538461538465e-05,
+      "loss": 1.2,
+      "step": 30
+    },
+    {
+      "epoch": 0.010776276796366856,
+      "grad_norm": 4.904063566508077,
+      "learning_rate": 2.1538461538461542e-05,
+      "loss": 1.2194,
+      "step": 35
+    },
+    {
+      "epoch": 0.012315744910133548,
+      "grad_norm": 2.625318230724936,
+      "learning_rate": 2.461538461538462e-05,
+      "loss": 1.1508,
+      "step": 40
+    },
+    {
+      "epoch": 0.013855213023900243,
+      "grad_norm": 1.554813766822268,
+      "learning_rate": 2.7692307692307694e-05,
+      "loss": 1.1786,
+      "step": 45
+    },
+    {
+      "epoch": 0.015394681137666935,
+      "grad_norm": 1.1809148162862095,
+      "learning_rate": 3.0769230769230774e-05,
+      "loss": 1.1277,
+      "step": 50
+    },
+    {
+      "epoch": 0.01693414925143363,
+      "grad_norm": 1.203464354806506,
+      "learning_rate": 3.384615384615385e-05,
+      "loss": 1.1463,
+      "step": 55
+    },
+    {
+      "epoch": 0.018473617365200323,
+      "grad_norm": 1.1383524365220588,
+      "learning_rate": 3.692307692307693e-05,
+      "loss": 1.1188,
+      "step": 60
+    },
+    {
+      "epoch": 0.020013085478967015,
+      "grad_norm": 0.9857720112400227,
+      "learning_rate": 4e-05,
+      "loss": 1.136,
+      "step": 65
+    },
+    {
+      "epoch": 0.02155255359273371,
+      "grad_norm": 0.8036894302534616,
+      "learning_rate": 4.3076923076923084e-05,
+      "loss": 1.0966,
+      "step": 70
+    },
+    {
+      "epoch": 0.023092021706500404,
+      "grad_norm": 1.3013628925636782,
+      "learning_rate": 4.615384615384616e-05,
+      "loss": 1.092,
+      "step": 75
+    },
+    {
+      "epoch": 0.024631489820267097,
+      "grad_norm": 1.2268436983757836,
+      "learning_rate": 4.923076923076924e-05,
+      "loss": 1.1211,
+      "step": 80
+    },
+    {
+      "epoch": 0.026170957934033793,
+      "grad_norm": 1.1230597630357015,
+      "learning_rate": 5.230769230769231e-05,
+      "loss": 1.1099,
+      "step": 85
+    },
+    {
+      "epoch": 0.027710426047800486,
+      "grad_norm": 1.1223875634038776,
+      "learning_rate": 5.538461538461539e-05,
+      "loss": 1.051,
+      "step": 90
+    },
+    {
+      "epoch": 0.02924989416156718,
+      "grad_norm": 1.3366126407920835,
+      "learning_rate": 5.846153846153847e-05,
+      "loss": 1.0889,
+      "step": 95
+    },
+    {
+      "epoch": 0.03078936227533387,
+      "grad_norm": 0.8876804012745447,
+      "learning_rate": 6.153846153846155e-05,
+      "loss": 1.0635,
+      "step": 100
+    },
+    {
+      "epoch": 0.03232883038910057,
+      "grad_norm": 0.985567117532096,
+      "learning_rate": 6.461538461538462e-05,
+      "loss": 1.0729,
+      "step": 105
+    },
+    {
+      "epoch": 0.03386829850286726,
+      "grad_norm": 1.3857701900950476,
+      "learning_rate": 6.76923076923077e-05,
+      "loss": 1.0687,
+      "step": 110
+    },
+    {
+      "epoch": 0.03540776661663395,
+      "grad_norm": 1.0307077006740535,
+      "learning_rate": 7.076923076923078e-05,
+      "loss": 1.0756,
+      "step": 115
+    },
+    {
+      "epoch": 0.036947234730400645,
+      "grad_norm": 0.915309617477138,
+      "learning_rate": 7.384615384615386e-05,
+      "loss": 1.0578,
+      "step": 120
+    },
+    {
+      "epoch": 0.03848670284416734,
+      "grad_norm": 1.1416205921187619,
+      "learning_rate": 7.692307692307693e-05,
+      "loss": 1.0722,
+      "step": 125
+    },
+    {
+      "epoch": 0.04002617095793403,
+      "grad_norm": 0.7855309627413886,
+      "learning_rate": 8e-05,
+      "loss": 1.1091,
+      "step": 130
+    },
+    {
+      "epoch": 0.04156563907170073,
+      "grad_norm": 1.027162314274905,
+      "learning_rate": 8.307692307692309e-05,
+      "loss": 1.0632,
+      "step": 135
+    },
+    {
+      "epoch": 0.04310510718546742,
+      "grad_norm": 1.1386693099418572,
+      "learning_rate": 8.615384615384617e-05,
+      "loss": 1.0736,
+      "step": 140
+    },
+    {
+      "epoch": 0.044644575299234115,
+      "grad_norm": 0.7717462620986457,
+      "learning_rate": 8.923076923076924e-05,
+      "loss": 1.0523,
+      "step": 145
+    },
+    {
+      "epoch": 0.04618404341300081,
+      "grad_norm": 0.7657994075849279,
+      "learning_rate": 9.230769230769232e-05,
+      "loss": 1.0557,
+      "step": 150
+    },
+    {
+      "epoch": 0.0477235115267675,
+      "grad_norm": 1.1680871949004308,
+      "learning_rate": 9.53846153846154e-05,
+      "loss": 1.0548,
+      "step": 155
+    },
+    {
+      "epoch": 0.049262979640534194,
+      "grad_norm": 0.8996216400147768,
+      "learning_rate": 9.846153846153848e-05,
+      "loss": 1.0665,
+      "step": 160
+    },
+    {
+      "epoch": 0.050802447754300886,
+      "grad_norm": 0.700328592160364,
+      "learning_rate": 0.00010153846153846153,
+      "loss": 1.0496,
+      "step": 165
+    },
+    {
+      "epoch": 0.052341915868067586,
+      "grad_norm": 1.0558734188354055,
+      "learning_rate": 0.00010461538461538463,
+      "loss": 1.065,
+      "step": 170
+    },
+    {
+      "epoch": 0.05388138398183428,
+      "grad_norm": 0.8578829636967044,
+      "learning_rate": 0.0001076923076923077,
+      "loss": 1.0923,
+      "step": 175
+    },
+    {
+      "epoch": 0.05542085209560097,
+      "grad_norm": 0.8831349656869951,
+      "learning_rate": 0.00011076923076923077,
+      "loss": 1.0259,
+      "step": 180
+    },
+    {
+      "epoch": 0.056960320209367664,
+      "grad_norm": 0.9573775567280769,
+      "learning_rate": 0.00011384615384615384,
+      "loss": 1.0661,
+      "step": 185
+    },
+    {
+      "epoch": 0.05849978832313436,
+      "grad_norm": 0.9495008375470986,
+      "learning_rate": 0.00011692307692307694,
+      "loss": 1.0811,
+      "step": 190
+    },
+    {
+      "epoch": 0.06003925643690105,
+      "grad_norm": 1.1061782824597675,
+      "learning_rate": 0.00012,
+      "loss": 1.0466,
+      "step": 195
+    },
+    {
+      "epoch": 0.06157872455066774,
+      "grad_norm": 1.1202640249643756,
+      "learning_rate": 0.0001230769230769231,
+      "loss": 1.0589,
+      "step": 200
+    },
+    {
+      "epoch": 0.06311819266443444,
+      "grad_norm": 1.0125076015545247,
+      "learning_rate": 0.00012615384615384615,
+      "loss": 1.0407,
+      "step": 205
+    },
+    {
+      "epoch": 0.06465766077820113,
+      "grad_norm": 1.0060078228523262,
+      "learning_rate": 0.00012923076923076923,
+      "loss": 1.0688,
+      "step": 210
+    },
+    {
+      "epoch": 0.06619712889196783,
+      "grad_norm": 0.7760894033200619,
+      "learning_rate": 0.0001323076923076923,
+      "loss": 1.0549,
+      "step": 215
+    },
+    {
+      "epoch": 0.06773659700573452,
+      "grad_norm": 0.8990085149589899,
+      "learning_rate": 0.0001353846153846154,
+      "loss": 1.0537,
+      "step": 220
+    },
+    {
+      "epoch": 0.06927606511950121,
+      "grad_norm": 0.8549275834093875,
+      "learning_rate": 0.00013846153846153847,
+      "loss": 1.0251,
+      "step": 225
+    },
+    {
+      "epoch": 0.0708155332332679,
+      "grad_norm": 0.8075081750437678,
+      "learning_rate": 0.00014153846153846156,
+      "loss": 1.0443,
+      "step": 230
+    },
+    {
+      "epoch": 0.0723550013470346,
+      "grad_norm": 0.9259388698536392,
+      "learning_rate": 0.0001446153846153846,
+      "loss": 1.0134,
+      "step": 235
+    },
+    {
+      "epoch": 0.07389446946080129,
+      "grad_norm": 1.7409427946151614,
+      "learning_rate": 0.00014769230769230772,
+      "loss": 1.073,
+      "step": 240
+    },
+    {
+      "epoch": 0.07543393757456798,
+      "grad_norm": 0.9864593980213632,
+      "learning_rate": 0.00015076923076923077,
+      "loss": 1.0735,
+      "step": 245
+    },
+    {
+      "epoch": 0.07697340568833468,
+      "grad_norm": 0.7479355733124677,
+      "learning_rate": 0.00015384615384615385,
+      "loss": 1.0642,
+      "step": 250
+    },
+    {
+      "epoch": 0.07851287380210137,
+      "grad_norm": 0.9485409237720248,
+      "learning_rate": 0.00015692307692307693,
+      "loss": 1.0,
+      "step": 255
+    },
+    {
+      "epoch": 0.08005234191586806,
+      "grad_norm": 1.0842298527942793,
+      "learning_rate": 0.00016,
+      "loss": 1.0726,
+      "step": 260
+    },
+    {
+      "epoch": 0.08159181002963477,
+      "grad_norm": 1.4490484618864288,
+      "learning_rate": 0.0001630769230769231,
+      "loss": 1.0506,
+      "step": 265
+    },
+    {
+      "epoch": 0.08313127814340146,
+      "grad_norm": 0.7322203679384404,
+      "learning_rate": 0.00016615384615384617,
+      "loss": 1.0934,
+      "step": 270
+    },
+    {
+      "epoch": 0.08467074625716815,
+      "grad_norm": 1.0463212758166829,
+      "learning_rate": 0.00016923076923076923,
+      "loss": 1.0414,
+      "step": 275
+    },
+    {
+      "epoch": 0.08621021437093485,
+      "grad_norm": 0.7534147115742311,
+      "learning_rate": 0.00017230769230769234,
+      "loss": 1.0582,
+      "step": 280
+    },
+    {
+      "epoch": 0.08774968248470154,
+      "grad_norm": 0.8938192794309777,
+      "learning_rate": 0.0001753846153846154,
+      "loss": 1.0676,
+      "step": 285
+    },
+    {
+      "epoch": 0.08928915059846823,
+      "grad_norm": 0.7350477942859435,
+      "learning_rate": 0.00017846153846153847,
+      "loss": 1.0727,
+      "step": 290
+    },
+    {
+      "epoch": 0.09082861871223492,
+      "grad_norm": 0.7547431024386564,
+      "learning_rate": 0.00018153846153846155,
+      "loss": 1.069,
+      "step": 295
+    },
+    {
+      "epoch": 0.09236808682600162,
+      "grad_norm": 1.0017027549794293,
+      "learning_rate": 0.00018461538461538463,
+      "loss": 1.0586,
+      "step": 300
+    },
+    {
+      "epoch": 0.09390755493976831,
+      "grad_norm": 0.7906132895743292,
+      "learning_rate": 0.0001876923076923077,
+      "loss": 1.0546,
+      "step": 305
+    },
+    {
+      "epoch": 0.095447023053535,
+      "grad_norm": 1.401442713529287,
+      "learning_rate": 0.0001907692307692308,
+      "loss": 1.0698,
+      "step": 310
+    },
+    {
+      "epoch": 0.0969864911673017,
+      "grad_norm": 0.5922407946201297,
+      "learning_rate": 0.00019384615384615385,
+      "loss": 1.0901,
+      "step": 315
+    },
+    {
+      "epoch": 0.09852595928106839,
+      "grad_norm": 0.6300597452052293,
+      "learning_rate": 0.00019692307692307696,
+      "loss": 1.0429,
+      "step": 320
+    },
+    {
+      "epoch": 0.10006542739483508,
+      "grad_norm": 0.8806436669088872,
+      "learning_rate": 0.0002,
+      "loss": 1.0552,
+      "step": 325
+    },
+    {
+      "epoch": 0.10160489550860177,
+      "grad_norm": 1.0983869005820532,
+      "learning_rate": 0.00019999855506507185,
+      "loss": 1.0851,
+      "step": 330
+    },
+    {
+      "epoch": 0.10314436362236847,
+      "grad_norm": 0.9815804026010979,
+      "learning_rate": 0.00019999422030204418,
+      "loss": 1.0673,
+      "step": 335
+    },
+    {
+      "epoch": 0.10468383173613517,
+      "grad_norm": 0.7301517719864844,
+      "learning_rate": 0.00019998699583618593,
+      "loss": 1.0516,
+      "step": 340
+    },
+    {
+      "epoch": 0.10622329984990186,
+      "grad_norm": 0.7786167269205654,
+      "learning_rate": 0.00019997688187627482,
+      "loss": 1.064,
+      "step": 345
+    },
+    {
+      "epoch": 0.10776276796366856,
+      "grad_norm": 0.7511697290569181,
+      "learning_rate": 0.0001999638787145911,
+      "loss": 1.0836,
+      "step": 350
+    },
+    {
+      "epoch": 0.10930223607743525,
+      "grad_norm": 0.8111968485885569,
+      "learning_rate": 0.0001999479867269092,
+      "loss": 1.0681,
+      "step": 355
+    },
+    {
+      "epoch": 0.11084170419120194,
+      "grad_norm": 1.1816993244319927,
+      "learning_rate": 0.00019992920637248697,
+      "loss": 1.0571,
+      "step": 360
+    },
+    {
+      "epoch": 0.11238117230496864,
+      "grad_norm": 0.7021483137095024,
+      "learning_rate": 0.00019990753819405213,
+      "loss": 1.0366,
+      "step": 365
+    },
+    {
+      "epoch": 0.11392064041873533,
+      "grad_norm": 0.786152904921158,
+      "learning_rate": 0.00019988298281778684,
+      "loss": 1.0745,
+      "step": 370
+    },
+    {
+      "epoch": 0.11546010853250202,
+      "grad_norm": 0.8625505891808654,
+      "learning_rate": 0.00019985554095330955,
+      "loss": 1.0309,
+      "step": 375
+    },
+    {
+      "epoch": 0.11699957664626871,
+      "grad_norm": 0.7101475350198745,
+      "learning_rate": 0.0001998252133936544,
+      "loss": 1.0623,
+      "step": 380
+    },
+    {
+      "epoch": 0.1185390447600354,
+      "grad_norm": 0.7943463609732394,
+      "learning_rate": 0.00019979200101524845,
+      "loss": 1.0798,
+      "step": 385
+    },
+    {
+      "epoch": 0.1200785128738021,
+      "grad_norm": 1.1298857584764006,
+      "learning_rate": 0.00019975590477788613,
+      "loss": 1.0619,
+      "step": 390
+    },
+    {
+      "epoch": 0.12161798098756879,
+      "grad_norm": 0.6519227994726357,
+      "learning_rate": 0.0001997169257247018,
+      "loss": 1.0198,
+      "step": 395
+    },
+    {
+      "epoch": 0.12315744910133548,
+      "grad_norm": 0.7712100700155368,
+      "learning_rate": 0.00019967506498213931,
+      "loss": 1.08,
+      "step": 400
+    },
+    {
+      "epoch": 0.12469691721510218,
+      "grad_norm": 0.785301798573464,
+      "learning_rate": 0.00019963032375991966,
+      "loss": 1.0889,
+      "step": 405
+    },
+    {
+      "epoch": 0.12623638532886888,
+      "grad_norm": 0.7950130962836747,
+      "learning_rate": 0.00019958270335100595,
+      "loss": 1.0349,
+      "step": 410
+    },
+    {
+      "epoch": 0.12777585344263556,
+      "grad_norm": 0.6411989099588487,
+      "learning_rate": 0.00019953220513156602,
+      "loss": 1.1014,
+      "step": 415
+    },
+    {
+      "epoch": 0.12931532155640227,
+      "grad_norm": 0.6881556293517902,
+      "learning_rate": 0.0001994788305609327,
+      "loss": 1.0433,
+      "step": 420
+    },
+    {
+      "epoch": 0.13085478967016895,
+      "grad_norm": 0.8822396645970323,
+      "learning_rate": 0.00019942258118156163,
+      "loss": 1.0555,
+      "step": 425
+    },
+    {
+      "epoch": 0.13239425778393565,
+      "grad_norm": 0.8566501292249419,
+      "learning_rate": 0.00019936345861898663,
+      "loss": 1.0791,
+      "step": 430
+    },
+    {
+      "epoch": 0.13393372589770233,
+      "grad_norm": 0.6000374004935582,
+      "learning_rate": 0.0001993014645817728,
+      "loss": 1.0644,
+      "step": 435
+    },
+    {
+      "epoch": 0.13547319401146904,
+      "grad_norm": 0.7094811645126697,
+      "learning_rate": 0.00019923660086146723,
+      "loss": 1.0588,
+      "step": 440
+    },
+    {
+      "epoch": 0.13701266212523572,
+      "grad_norm": 0.7299783423452573,
+      "learning_rate": 0.0001991688693325469,
+      "loss": 1.0606,
+      "step": 445
+    },
+    {
+      "epoch": 0.13855213023900242,
+      "grad_norm": 0.7793849224097605,
+      "learning_rate": 0.00019909827195236493,
+      "loss": 1.0535,
+      "step": 450
+    },
+    {
+      "epoch": 0.14009159835276913,
+      "grad_norm": 0.8123995612581718,
+      "learning_rate": 0.00019902481076109372,
+      "loss": 1.0903,
+      "step": 455
+    },
+    {
+      "epoch": 0.1416310664665358,
+      "grad_norm": 0.8269490106548164,
+      "learning_rate": 0.00019894848788166604,
+      "loss": 1.0682,
+      "step": 460
+    },
+    {
+      "epoch": 0.14317053458030252,
+      "grad_norm": 0.6945769324601003,
+      "learning_rate": 0.00019886930551971387,
+      "loss": 1.0918,
+      "step": 465
+    },
+    {
+      "epoch": 0.1447100026940692,
+      "grad_norm": 1.0743052082703004,
+      "learning_rate": 0.0001987872659635043,
+      "loss": 1.0512,
+      "step": 470
+    },
+    {
+      "epoch": 0.1462494708078359,
+      "grad_norm": 0.7274051186012023,
+      "learning_rate": 0.00019870237158387384,
+      "loss": 1.0609,
+      "step": 475
+    },
+    {
+      "epoch": 0.14778893892160258,
+      "grad_norm": 0.8436743181685329,
+      "learning_rate": 0.00019861462483415952,
+      "loss": 1.0534,
+      "step": 480
+    },
+    {
+      "epoch": 0.1493284070353693,
+      "grad_norm": 0.5539587236696554,
+      "learning_rate": 0.0001985240282501282,
+      "loss": 1.0562,
+      "step": 485
+    },
+    {
+      "epoch": 0.15086787514913597,
+      "grad_norm": 0.6673041088797687,
+      "learning_rate": 0.0001984305844499033,
+      "loss": 1.0343,
+      "step": 490
+    },
+    {
+      "epoch": 0.15240734326290267,
+      "grad_norm": 0.6860635549461462,
+      "learning_rate": 0.00019833429613388902,
+      "loss": 1.0602,
+      "step": 495
+    },
+    {
+      "epoch": 0.15394681137666935,
+      "grad_norm": 1.0779978346564654,
+      "learning_rate": 0.0001982351660846924,
+      "loss": 1.0539,
+      "step": 500
+    },
+    {
+      "epoch": 0.15548627949043606,
+      "grad_norm": 0.6737414679607289,
+      "learning_rate": 0.00019813319716704278,
+      "loss": 1.0559,
+      "step": 505
+    },
+    {
+      "epoch": 0.15702574760420274,
+      "grad_norm": 0.7252765201034201,
+      "learning_rate": 0.00019802839232770921,
+      "loss": 1.0986,
+      "step": 510
+    },
+    {
+      "epoch": 0.15856521571796944,
+      "grad_norm": 1.3250469121293895,
+      "learning_rate": 0.00019792075459541518,
+      "loss": 1.0327,
+      "step": 515
+    },
+    {
+      "epoch": 0.16010468383173612,
+      "grad_norm": 0.6408679533216792,
+      "learning_rate": 0.00019781028708075102,
+      "loss": 1.0415,
+      "step": 520
+    },
+    {
+      "epoch": 0.16164415194550283,
+      "grad_norm": 0.9808936518837947,
+      "learning_rate": 0.00019769699297608417,
+      "loss": 1.0386,
+      "step": 525
+    },
+    {
+      "epoch": 0.16318362005926954,
+      "grad_norm": 0.7892295118178237,
+      "learning_rate": 0.00019758087555546682,
+      "loss": 1.0866,
+      "step": 530
+    },
+    {
+      "epoch": 0.16472308817303621,
+      "grad_norm": 0.6508163256108649,
+      "learning_rate": 0.0001974619381745413,
+      "loss": 1.0804,
+      "step": 535
+    },
+    {
+      "epoch": 0.16626255628680292,
+      "grad_norm": 0.7619370139690409,
+      "learning_rate": 0.00019734018427044307,
+      "loss": 1.0471,
+      "step": 540
+    },
+    {
+      "epoch": 0.1678020244005696,
+      "grad_norm": 1.3124132163956248,
+      "learning_rate": 0.0001972156173617016,
+      "loss": 1.0517,
+      "step": 545
+    },
+    {
+      "epoch": 0.1693414925143363,
+      "grad_norm": 0.969795077947931,
+      "learning_rate": 0.00019708824104813837,
+      "loss": 1.0518,
+      "step": 550
+    },
+    {
+      "epoch": 0.17088096062810298,
+      "grad_norm": 0.7315912579640327,
+      "learning_rate": 0.00019695805901076308,
+      "loss": 1.0784,
+      "step": 555
+    },
+    {
+      "epoch": 0.1724204287418697,
+      "grad_norm": 0.7357250477683509,
+      "learning_rate": 0.00019682507501166718,
+      "loss": 1.1008,
+      "step": 560
+    },
+    {
+      "epoch": 0.17395989685563637,
+      "grad_norm": 0.96782095170125,
+      "learning_rate": 0.00019668929289391523,
+      "loss": 1.0444,
+      "step": 565
+    },
+    {
+      "epoch": 0.17549936496940308,
+      "grad_norm": 0.7600182365968516,
+      "learning_rate": 0.00019655071658143366,
+      "loss": 1.0625,
+      "step": 570
+    },
+    {
+      "epoch": 0.17703883308316976,
+      "grad_norm": 0.6161742360584987,
+      "learning_rate": 0.00019640935007889755,
+      "loss": 1.091,
+      "step": 575
+    },
+    {
+      "epoch": 0.17857830119693646,
+      "grad_norm": 0.5777577463068309,
+      "learning_rate": 0.0001962651974716149,
+      "loss": 1.072,
+      "step": 580
+    },
+    {
+      "epoch": 0.18011776931070314,
+      "grad_norm": 0.644094996294978,
+      "learning_rate": 0.0001961182629254084,
+      "loss": 1.0275,
+      "step": 585
+    },
+    {
+      "epoch": 0.18165723742446985,
+      "grad_norm": 0.8589876413667922,
+      "learning_rate": 0.00019596855068649522,
+      "loss": 1.0605,
+      "step": 590
+    },
+    {
+      "epoch": 0.18319670553823653,
+      "grad_norm": 0.7740247212495535,
+      "learning_rate": 0.00019581606508136426,
+      "loss": 1.0764,
+      "step": 595
+    },
+    {
+      "epoch": 0.18473617365200323,
+      "grad_norm": 0.6540083386801736,
+      "learning_rate": 0.00019566081051665098,
+      "loss": 1.0174,
+      "step": 600
+    },
+    {
+      "epoch": 0.18627564176576994,
+      "grad_norm": 0.7815089251045078,
+      "learning_rate": 0.00019550279147901036,
+      "loss": 1.0952,
+      "step": 605
+    },
+    {
+      "epoch": 0.18781510987953662,
+      "grad_norm": 0.6670730533500357,
+      "learning_rate": 0.00019534201253498682,
+      "loss": 1.0484,
+      "step": 610
+    },
+    {
+      "epoch": 0.18935457799330332,
+      "grad_norm": 0.7552112484078745,
+      "learning_rate": 0.0001951784783308827,
+      "loss": 1.0104,
+      "step": 615
+    },
+    {
+      "epoch": 0.19089404610707,
+      "grad_norm": 0.6343514261370218,
+      "learning_rate": 0.0001950121935926236,
+      "loss": 1.0525,
+      "step": 620
+    },
+    {
+      "epoch": 0.1924335142208367,
+      "grad_norm": 1.4473777235442007,
+      "learning_rate": 0.00019484316312562205,
+      "loss": 1.033,
+      "step": 625
+    },
+    {
+      "epoch": 0.1939729823346034,
+      "grad_norm": 0.5653781957358222,
+      "learning_rate": 0.00019467139181463862,
+      "loss": 1.078,
+      "step": 630
+    },
+    {
+      "epoch": 0.1955124504483701,
+      "grad_norm": 0.6315352985318332,
+      "learning_rate": 0.00019449688462364056,
+      "loss": 1.042,
+      "step": 635
+    },
+    {
+      "epoch": 0.19705191856213677,
+      "grad_norm": 0.8308310132176421,
+      "learning_rate": 0.00019431964659565867,
+      "loss": 1.0604,
+      "step": 640
+    },
+    {
+      "epoch": 0.19859138667590348,
+      "grad_norm": 0.5915623011666878,
+      "learning_rate": 0.0001941396828526412,
+      "loss": 1.058,
+      "step": 645
+    },
+    {
+      "epoch": 0.20013085478967016,
+      "grad_norm": 0.9623739180076736,
+      "learning_rate": 0.00019395699859530623,
+      "loss": 1.0314,
+      "step": 650
+    },
+    {
+      "epoch": 0.20167032290343687,
+      "grad_norm": 0.7770406923743924,
+      "learning_rate": 0.00019377159910299093,
+      "loss": 1.0418,
+      "step": 655
+    },
+    {
+      "epoch": 0.20320979101720354,
+      "grad_norm": 0.582853981181253,
+      "learning_rate": 0.00019358348973349943,
+      "loss": 1.0228,
+      "step": 660
+    },
+    {
+      "epoch": 0.20474925913097025,
+      "grad_norm": 0.6672404317903096,
+      "learning_rate": 0.00019339267592294763,
+      "loss": 1.0602,
+      "step": 665
+    },
+    {
+      "epoch": 0.20628872724473693,
+      "grad_norm": 0.7929111958910604,
+      "learning_rate": 0.00019319916318560635,
+      "loss": 1.0359,
+      "step": 670
+    },
+    {
+      "epoch": 0.20782819535850364,
+      "grad_norm": 0.6853640952957439,
+      "learning_rate": 0.00019300295711374187,
+      "loss": 1.0519,
+      "step": 675
+    },
+    {
+      "epoch": 0.20936766347227034,
+      "grad_norm": 0.6773550735079633,
+      "learning_rate": 0.00019280406337745428,
+      "loss": 1.0295,
+      "step": 680
+    },
+    {
+      "epoch": 0.21090713158603702,
+      "grad_norm": 0.7456356464144906,
+      "learning_rate": 0.00019260248772451377,
+      "loss": 1.0614,
+      "step": 685
+    },
+    {
+      "epoch": 0.21244659969980373,
+      "grad_norm": 0.7051829760237421,
+      "learning_rate": 0.0001923982359801943,
+      "loss": 1.0556,
+      "step": 690
+    },
+    {
+      "epoch": 0.2139860678135704,
+      "grad_norm": 0.6567867383663277,
+      "learning_rate": 0.00019219131404710552,
+      "loss": 1.0854,
+      "step": 695
+    },
+    {
+      "epoch": 0.21552553592733711,
+      "grad_norm": 0.6369210418492607,
+      "learning_rate": 0.00019198172790502196,
+      "loss": 1.0368,
+      "step": 700
+    },
+    {
+      "epoch": 0.2170650040411038,
+      "grad_norm": 0.6342900260792401,
+      "learning_rate": 0.0001917694836107104,
+      "loss": 1.0484,
+      "step": 705
+    },
+    {
+      "epoch": 0.2186044721548705,
+      "grad_norm": 0.5622570224326242,
+      "learning_rate": 0.00019155458729775467,
+      "loss": 1.0595,
+      "step": 710
+    },
+    {
+      "epoch": 0.22014394026863718,
+      "grad_norm": 0.809911467681193,
+      "learning_rate": 0.0001913370451763786,
+      "loss": 1.0278,
+      "step": 715
+    },
+    {
+      "epoch": 0.22168340838240388,
+      "grad_norm": 0.6267743698514823,
+      "learning_rate": 0.00019111686353326631,
+      "loss": 1.0309,
+      "step": 720
+    },
+    {
+      "epoch": 0.22322287649617056,
+      "grad_norm": 0.5838744010816593,
+      "learning_rate": 0.00019089404873138082,
+      "loss": 1.0637,
+      "step": 725
+    },
+    {
+      "epoch": 0.22476234460993727,
+      "grad_norm": 0.5837090263106021,
+      "learning_rate": 0.00019066860720977986,
+      "loss": 1.059,
+      "step": 730
+    },
+    {
+      "epoch": 0.22630181272370395,
+      "grad_norm": 0.7696380706231972,
+      "learning_rate": 0.00019044054548343002,
+      "loss": 1.0403,
+      "step": 735
+    },
+    {
+      "epoch": 0.22784128083747066,
+      "grad_norm": 0.5699826758505713,
+      "learning_rate": 0.0001902098701430184,
+      "loss": 1.0799,
+      "step": 740
+    },
+    {
+      "epoch": 0.22938074895123733,
+      "grad_norm": 0.6868169075444956,
+      "learning_rate": 0.00018997658785476214,
+      "loss": 1.0781,
+      "step": 745
+    },
+    {
+      "epoch": 0.23092021706500404,
+      "grad_norm": 0.583006345831051,
+      "learning_rate": 0.00018974070536021572,
+      "loss": 1.0814,
+      "step": 750
+    },
+    {
+      "epoch": 0.23245968517877075,
+      "grad_norm": 0.6006905084344202,
+      "learning_rate": 0.00018950222947607625,
+      "loss": 1.0624,
+      "step": 755
+    },
+    {
+      "epoch": 0.23399915329253743,
+      "grad_norm": 0.5697764314148909,
+      "learning_rate": 0.0001892611670939865,
+      "loss": 1.008,
+      "step": 760
+    },
+    {
+      "epoch": 0.23553862140630413,
+      "grad_norm": 0.7413320445202513,
+      "learning_rate": 0.00018901752518033548,
+      "loss": 1.0612,
+      "step": 765
+    },
+    {
+      "epoch": 0.2370780895200708,
+      "grad_norm": 0.5363151044675402,
+      "learning_rate": 0.0001887713107760575,
+      "loss": 1.0605,
+      "step": 770
+    },
+    {
+      "epoch": 0.23861755763383752,
+      "grad_norm": 0.8682202097559629,
+      "learning_rate": 0.00018852253099642833,
+      "loss": 1.031,
+      "step": 775
+    },
+    {
+      "epoch": 0.2401570257476042,
+      "grad_norm": 0.5831373382248602,
+      "learning_rate": 0.0001882711930308599,
+      "loss": 1.06,
+      "step": 780
+    },
+    {
+      "epoch": 0.2416964938613709,
+      "grad_norm": 0.5590693033565884,
+      "learning_rate": 0.00018801730414269225,
+      "loss": 1.0533,
+      "step": 785
+    },
+    {
+      "epoch": 0.24323596197513758,
+      "grad_norm": 0.6205945960602542,
+      "learning_rate": 0.0001877608716689839,
+      "loss": 1.0757,
+      "step": 790
+    },
+    {
+      "epoch": 0.2447754300889043,
+      "grad_norm": 0.7900938502871934,
+      "learning_rate": 0.00018750190302029956,
+      "loss": 1.0301,
+      "step": 795
+    },
+    {
+      "epoch": 0.24631489820267097,
+      "grad_norm": 1.0979738402724484,
+      "learning_rate": 0.00018724040568049612,
+      "loss": 1.0547,
+      "step": 800
+    },
+    {
+      "epoch": 0.24785436631643767,
+      "grad_norm": 0.7255960650743382,
+      "learning_rate": 0.00018697638720650646,
+      "loss": 1.0454,
+      "step": 805
+    },
+    {
+      "epoch": 0.24939383443020435,
+      "grad_norm": 0.6620553781722808,
+      "learning_rate": 0.00018670985522812084,
+      "loss": 1.0219,
+      "step": 810
+    },
+    {
+      "epoch": 0.25093330254397106,
+      "grad_norm": 0.6854793724956825,
+      "learning_rate": 0.0001864408174477665,
+      "loss": 1.0509,
+      "step": 815
+    },
+    {
+      "epoch": 0.25247277065773777,
+      "grad_norm": 0.5492317189334932,
+      "learning_rate": 0.00018616928164028523,
+      "loss": 1.0159,
+      "step": 820
+    },
+    {
+      "epoch": 0.2540122387715045,
+      "grad_norm": 0.8404726657300783,
+      "learning_rate": 0.00018589525565270844,
+      "loss": 1.0411,
+      "step": 825
+    },
+    {
+      "epoch": 0.2555517068852711,
+      "grad_norm": 0.6623737864891829,
+      "learning_rate": 0.0001856187474040306,
+      "loss": 1.0602,
+      "step": 830
+    },
+    {
+      "epoch": 0.25709117499903783,
+      "grad_norm": 0.5418949047693957,
+      "learning_rate": 0.00018533976488498016,
+      "loss": 1.037,
+      "step": 835
+    },
+    {
+      "epoch": 0.25863064311280454,
+      "grad_norm": 0.7761449291018582,
+      "learning_rate": 0.0001850583161577889,
+      "loss": 1.0738,
+      "step": 840
+    },
+    {
+      "epoch": 0.26017011122657124,
+      "grad_norm": 0.5847223837062915,
+      "learning_rate": 0.00018477440935595873,
+      "loss": 1.0875,
+      "step": 845
+    },
+    {
+      "epoch": 0.2617095793403379,
+      "grad_norm": 0.5106913164341648,
+      "learning_rate": 0.00018448805268402672,
+      "loss": 1.0685,
+      "step": 850
+    },
+    {
+      "epoch": 0.2632490474541046,
+      "grad_norm": 0.7444912623137734,
+      "learning_rate": 0.00018419925441732804,
+      "loss": 1.0407,
+      "step": 855
+    },
+    {
+      "epoch": 0.2647885155678713,
+      "grad_norm": 0.5183580101131431,
+      "learning_rate": 0.00018390802290175673,
+      "loss": 1.0572,
+      "step": 860
+    },
+    {
+      "epoch": 0.266327983681638,
+      "grad_norm": 0.6844054776521512,
+      "learning_rate": 0.00018361436655352456,
+      "loss": 1.0383,
+      "step": 865
+    },
+    {
+      "epoch": 0.26786745179540467,
+      "grad_norm": 1.0067719874632055,
+      "learning_rate": 0.00018331829385891783,
+      "loss": 1.031,
+      "step": 870
+    },
+    {
+      "epoch": 0.26940691990917137,
+      "grad_norm": 0.5484350200540726,
+      "learning_rate": 0.00018301981337405212,
+      "loss": 1.0585,
+      "step": 875
+    },
+    {
+      "epoch": 0.2709463880229381,
+      "grad_norm": 0.6056460900910511,
+      "learning_rate": 0.00018271893372462497,
+      "loss": 1.0585,
+      "step": 880
+    },
+    {
+      "epoch": 0.2724858561367048,
+      "grad_norm": 0.48416296519569474,
+      "learning_rate": 0.00018241566360566665,
+      "loss": 1.039,
+      "step": 885
+    },
+    {
+      "epoch": 0.27402532425047144,
+      "grad_norm": 0.6050871964307601,
+      "learning_rate": 0.00018211001178128892,
+      "loss": 1.0571,
+      "step": 890
+    },
+    {
+      "epoch": 0.27556479236423814,
+      "grad_norm": 0.5969541614482573,
+      "learning_rate": 0.00018180198708443173,
+      "loss": 1.0653,
+      "step": 895
+    },
+    {
+      "epoch": 0.27710426047800485,
+      "grad_norm": 0.7541885423123031,
+      "learning_rate": 0.00018149159841660795,
+      "loss": 1.0521,
+      "step": 900
+    },
+    {
+      "epoch": 0.27864372859177156,
+      "grad_norm": 0.49528417781944445,
+      "learning_rate": 0.00018117885474764613,
+      "loss": 1.0943,
+      "step": 905
+    },
+    {
+      "epoch": 0.28018319670553826,
+      "grad_norm": 0.6918705895372237,
+      "learning_rate": 0.00018086376511543126,
+      "loss": 1.0747,
+      "step": 910
+    },
+    {
+      "epoch": 0.2817226648193049,
+      "grad_norm": 0.7195296276082791,
+      "learning_rate": 0.00018054633862564368,
+      "loss": 1.0507,
+      "step": 915
+    },
+    {
+      "epoch": 0.2832621329330716,
+      "grad_norm": 0.5709551372136472,
+      "learning_rate": 0.0001802265844514958,
+      "loss": 1.0094,
+      "step": 920
+    },
+    {
+      "epoch": 0.2848016010468383,
+      "grad_norm": 0.43690572218401663,
+      "learning_rate": 0.0001799045118334671,
+      "loss": 1.064,
+      "step": 925
+    },
+    {
+      "epoch": 0.28634106916060503,
+      "grad_norm": 0.5329066679717666,
+      "learning_rate": 0.00017958013007903713,
+      "loss": 1.067,
+      "step": 930
+    },
+    {
+      "epoch": 0.2878805372743717,
+      "grad_norm": 0.53800354086698,
+      "learning_rate": 0.0001792534485624164,
+      "loss": 1.0491,
+      "step": 935
+    },
+    {
+      "epoch": 0.2894200053881384,
+      "grad_norm": 0.6415976681925623,
+      "learning_rate": 0.00017892447672427563,
+      "loss": 1.0496,
+      "step": 940
+    },
+    {
+      "epoch": 0.2909594735019051,
+      "grad_norm": 0.5366936452063369,
+      "learning_rate": 0.00017859322407147272,
+      "loss": 1.0657,
+      "step": 945
+    },
+    {
+      "epoch": 0.2924989416156718,
+      "grad_norm": 0.5700114756555418,
+      "learning_rate": 0.00017825970017677832,
+      "loss": 1.0808,
+      "step": 950
+    },
+    {
+      "epoch": 0.29403840972943845,
+      "grad_norm": 0.5602975678360493,
+      "learning_rate": 0.00017792391467859886,
+      "loss": 1.0255,
+      "step": 955
+    },
+    {
+      "epoch": 0.29557787784320516,
+      "grad_norm": 0.5222669096259323,
+      "learning_rate": 0.0001775858772806983,
+      "loss": 1.0762,
+      "step": 960
+    },
+    {
+      "epoch": 0.29711734595697187,
+      "grad_norm": 0.6950722858238849,
+      "learning_rate": 0.00017724559775191744,
+      "loss": 1.0382,
+      "step": 965
+    },
+    {
+      "epoch": 0.2986568140707386,
+      "grad_norm": 1.0357933657324396,
+      "learning_rate": 0.00017690308592589182,
+      "loss": 1.0541,
+      "step": 970
+    },
+    {
+      "epoch": 0.3001962821845053,
+      "grad_norm": 0.5871650345246843,
+      "learning_rate": 0.0001765583517007675,
+      "loss": 1.0035,
+      "step": 975
+    },
+    {
+      "epoch": 0.30173575029827193,
+      "grad_norm": 0.5209052634359257,
+      "learning_rate": 0.00017621140503891488,
+      "loss": 1.0206,
+      "step": 980
+    },
+    {
+      "epoch": 0.30327521841203864,
+      "grad_norm": 0.6487211604010458,
+      "learning_rate": 0.00017586225596664102,
+      "loss": 1.0381,
+      "step": 985
+    },
+    {
+      "epoch": 0.30481468652580535,
+      "grad_norm": 0.5629565123905481,
+      "learning_rate": 0.00017551091457389966,
+      "loss": 1.0434,
+      "step": 990
+    },
+    {
+      "epoch": 0.30635415463957205,
+      "grad_norm": 0.8112166900825726,
+      "learning_rate": 0.00017515739101399983,
+      "loss": 1.0287,
+      "step": 995
+    },
+    {
+      "epoch": 0.3078936227533387,
+      "grad_norm": 0.6366943000992968,
+      "learning_rate": 0.00017480169550331231,
+      "loss": 1.0193,
+      "step": 1000
+    },
+    {
+      "epoch": 0.3094330908671054,
+      "grad_norm": 0.5040648629606902,
+      "learning_rate": 0.00017444383832097442,
+      "loss": 1.0411,
+      "step": 1005
+    },
+    {
+      "epoch": 0.3109725589808721,
+      "grad_norm": 0.8283443113345634,
+      "learning_rate": 0.00017408382980859305,
+      "loss": 1.045,
+      "step": 1010
+    },
+    {
+      "epoch": 0.3125120270946388,
+      "grad_norm": 0.6777410855607223,
+      "learning_rate": 0.00017372168036994566,
+      "loss": 1.0263,
+      "step": 1015
+    },
+    {
+      "epoch": 0.3140514952084055,
+      "grad_norm": 0.5530544091806374,
+      "learning_rate": 0.00017335740047067972,
+      "loss": 1.0307,
+      "step": 1020
+    },
+    {
+      "epoch": 0.3155909633221722,
+      "grad_norm": 0.45671110148384275,
+      "learning_rate": 0.0001729910006380102,
+      "loss": 1.055,
+      "step": 1025
+    },
+    {
+      "epoch": 0.3171304314359389,
+      "grad_norm": 0.6812995390238387,
+      "learning_rate": 0.00017262249146041546,
+      "loss": 1.0838,
+      "step": 1030
+    },
+    {
+      "epoch": 0.3186698995497056,
+      "grad_norm": 0.568521042510064,
+      "learning_rate": 0.00017225188358733107,
+      "loss": 1.0265,
+      "step": 1035
+    },
+    {
+      "epoch": 0.32020936766347224,
+      "grad_norm": 0.8025629235004524,
+      "learning_rate": 0.00017187918772884232,
+      "loss": 1.0721,
+      "step": 1040
+    },
+    {
+      "epoch": 0.32174883577723895,
+      "grad_norm": 0.6004044742833335,
+      "learning_rate": 0.00017150441465537447,
+      "loss": 1.0655,
+      "step": 1045
+    },
+    {
+      "epoch": 0.32328830389100566,
+      "grad_norm": 0.5635765966408569,
+      "learning_rate": 0.00017112757519738154,
+      "loss": 1.0396,
+      "step": 1050
+    },
+    {
+      "epoch": 0.32482777200477236,
+      "grad_norm": 0.5043399412628455,
+      "learning_rate": 0.0001707486802450335,
+      "loss": 1.0536,
+      "step": 1055
+    },
+    {
+      "epoch": 0.32636724011853907,
+      "grad_norm": 0.5106615244506089,
+      "learning_rate": 0.00017036774074790132,
+      "loss": 1.0135,
+      "step": 1060
+    },
+    {
+      "epoch": 0.3279067082323057,
+      "grad_norm": 0.6069038358467338,
+      "learning_rate": 0.00016998476771464072,
+      "loss": 1.0488,
+      "step": 1065
+    },
+    {
+      "epoch": 0.32944617634607243,
+      "grad_norm": 0.5481314050474333,
+      "learning_rate": 0.00016959977221267392,
+      "loss": 1.0226,
+      "step": 1070
+    },
+    {
+      "epoch": 0.33098564445983913,
+      "grad_norm": 0.5692030442740773,
+      "learning_rate": 0.0001692127653678699,
+      "loss": 1.0545,
+      "step": 1075
+    },
+    {
+      "epoch": 0.33252511257360584,
+      "grad_norm": 0.5422422595812953,
+      "learning_rate": 0.00016882375836422284,
+      "loss": 1.04,
+      "step": 1080
+    },
+    {
+      "epoch": 0.3340645806873725,
+      "grad_norm": 0.8241400248029749,
+      "learning_rate": 0.00016843276244352885,
+      "loss": 1.0667,
+      "step": 1085
+    },
+    {
+      "epoch": 0.3356040488011392,
+      "grad_norm": 0.5205208851186217,
+      "learning_rate": 0.00016803978890506113,
+      "loss": 1.0783,
+      "step": 1090
+    },
+    {
+      "epoch": 0.3371435169149059,
+      "grad_norm": 0.6117364442740371,
+      "learning_rate": 0.00016764484910524358,
+      "loss": 1.0339,
+      "step": 1095
+    },
+    {
+      "epoch": 0.3386829850286726,
+      "grad_norm": 0.6490741063813862,
+      "learning_rate": 0.00016724795445732243,
+      "loss": 1.0068,
+      "step": 1100
+    },
+    {
+      "epoch": 0.34022245314243926,
+      "grad_norm": 0.6013736527570045,
+      "learning_rate": 0.00016684911643103642,
+      "loss": 1.0665,
+      "step": 1105
+    },
+    {
+      "epoch": 0.34176192125620597,
+      "grad_norm": 0.6118721687785333,
+      "learning_rate": 0.0001664483465522855,
+      "loss": 1.0423,
+      "step": 1110
+    },
+    {
+      "epoch": 0.3433013893699727,
+      "grad_norm": 0.5419797734028425,
+      "learning_rate": 0.00016604565640279754,
+      "loss": 1.0564,
+      "step": 1115
+    },
+    {
+      "epoch": 0.3448408574837394,
+      "grad_norm": 0.5495541176613125,
+      "learning_rate": 0.0001656410576197938,
+      "loss": 1.0636,
+      "step": 1120
+    },
+    {
+      "epoch": 0.3463803255975061,
+      "grad_norm": 0.4477184453911253,
+      "learning_rate": 0.0001652345618956526,
+      "loss": 1.0797,
+      "step": 1125
+    },
+    {
+      "epoch": 0.34791979371127274,
+      "grad_norm": 0.5912654457171957,
+      "learning_rate": 0.00016482618097757122,
+      "loss": 1.0553,
+      "step": 1130
+    },
+    {
+      "epoch": 0.34945926182503945,
+      "grad_norm": 0.5561325327054591,
+      "learning_rate": 0.00016441592666722684,
+      "loss": 1.0459,
+      "step": 1135
+    },
+    {
+      "epoch": 0.35099872993880615,
+      "grad_norm": 0.6091265051455126,
+      "learning_rate": 0.00016400381082043507,
+      "loss": 1.0916,
+      "step": 1140
+    },
+    {
+      "epoch": 0.35253819805257286,
+      "grad_norm": 0.7050858387246796,
+      "learning_rate": 0.00016358984534680748,
+      "loss": 1.0605,
+      "step": 1145
+    },
+    {
+      "epoch": 0.3540776661663395,
+      "grad_norm": 0.592579238142578,
+      "learning_rate": 0.00016317404220940758,
+      "loss": 1.0319,
+      "step": 1150
+    },
+    {
+      "epoch": 0.3556171342801062,
+      "grad_norm": 0.4991456345649348,
+      "learning_rate": 0.00016275641342440483,
+      "loss": 1.0671,
+      "step": 1155
+    },
+    {
+      "epoch": 0.3571566023938729,
+      "grad_norm": 0.5675708757017103,
+      "learning_rate": 0.0001623369710607277,
+      "loss": 1.0136,
+      "step": 1160
+    },
+    {
+      "epoch": 0.35869607050763963,
+      "grad_norm": 0.45324587662080057,
+      "learning_rate": 0.00016191572723971455,
+      "loss": 1.0752,
+      "step": 1165
+    },
+    {
+      "epoch": 0.3602355386214063,
+      "grad_norm": 0.4578223763710829,
+      "learning_rate": 0.00016149269413476353,
+      "loss": 1.0162,
+      "step": 1170
+    },
+    {
+      "epoch": 0.361775006735173,
+      "grad_norm": 0.4787021798925449,
+      "learning_rate": 0.00016106788397098095,
+      "loss": 1.0031,
+      "step": 1175
+    },
+    {
+      "epoch": 0.3633144748489397,
+      "grad_norm": 0.49337377711536057,
+      "learning_rate": 0.0001606413090248276,
+      "loss": 1.0042,
+      "step": 1180
+    },
+    {
+      "epoch": 0.3648539429627064,
+      "grad_norm": 0.549765500927778,
+      "learning_rate": 0.00016021298162376428,
+      "loss": 1.0301,
+      "step": 1185
+    },
+    {
+      "epoch": 0.36639341107647305,
+      "grad_norm": 0.6910911024000683,
+      "learning_rate": 0.00015978291414589542,
+      "loss": 1.049,
+      "step": 1190
+    },
+    {
+      "epoch": 0.36793287919023976,
+      "grad_norm": 0.5099057113980399,
+      "learning_rate": 0.0001593511190196115,
+      "loss": 1.0613,
+      "step": 1195
+    },
+    {
+      "epoch": 0.36947234730400647,
+      "grad_norm": 0.5326314131616275,
+      "learning_rate": 0.00015891760872322963,
+      "loss": 1.0177,
+      "step": 1200
+    },
+    {
+      "epoch": 0.37101181541777317,
+      "grad_norm": 0.44876418883849256,
+      "learning_rate": 0.00015848239578463325,
+      "loss": 1.0594,
+      "step": 1205
+    },
+    {
+      "epoch": 0.3725512835315399,
+      "grad_norm": 0.518140271684293,
+      "learning_rate": 0.00015804549278090982,
+      "loss": 1.0228,
+      "step": 1210
+    },
+    {
+      "epoch": 0.37409075164530653,
+      "grad_norm": 0.5085879960253272,
+      "learning_rate": 0.00015760691233798757,
+      "loss": 1.0187,
+      "step": 1215
+    },
+    {
+      "epoch": 0.37563021975907324,
+      "grad_norm": 0.5007291726473487,
+      "learning_rate": 0.00015716666713027055,
+      "loss": 1.0433,
+      "step": 1220
+    },
+    {
+      "epoch": 0.37716968787283994,
+      "grad_norm": 0.739316075498731,
+      "learning_rate": 0.00015672476988027228,
+      "loss": 1.0478,
+      "step": 1225
+    },
+    {
+      "epoch": 0.37870915598660665,
+      "grad_norm": 0.7639785434460002,
+      "learning_rate": 0.0001562812333582482,
+      "loss": 1.0117,
+      "step": 1230
+    },
+    {
+      "epoch": 0.3802486241003733,
+      "grad_norm": 0.5823718950453733,
+      "learning_rate": 0.00015583607038182655,
+      "loss": 1.0366,
+      "step": 1235
+    },
+    {
+      "epoch": 0.38178809221414,
+      "grad_norm": 0.6893848055474053,
+      "learning_rate": 0.000155389293815638,
+      "loss": 1.0381,
+      "step": 1240
+    },
+    {
+      "epoch": 0.3833275603279067,
+      "grad_norm": 0.5063333022122084,
+      "learning_rate": 0.00015494091657094385,
+      "loss": 1.0419,
+      "step": 1245
+    },
+    {
+      "epoch": 0.3848670284416734,
+      "grad_norm": 0.44024803486149566,
+      "learning_rate": 0.00015449095160526292,
+      "loss": 1.0645,
+      "step": 1250
+    },
+    {
+      "epoch": 0.38640649655544007,
+      "grad_norm": 0.5599556807381372,
+      "learning_rate": 0.00015403941192199718,
+      "loss": 1.0443,
+      "step": 1255
+    },
+    {
+      "epoch": 0.3879459646692068,
+      "grad_norm": 0.4886207710717129,
+      "learning_rate": 0.0001535863105700558,
+      "loss": 1.0556,
+      "step": 1260
+    },
+    {
+      "epoch": 0.3894854327829735,
+      "grad_norm": 0.5246689722873303,
+      "learning_rate": 0.00015313166064347814,
+      "loss": 1.0256,
+      "step": 1265
+    },
+    {
+      "epoch": 0.3910249008967402,
+      "grad_norm": 0.4204231295176996,
+      "learning_rate": 0.00015267547528105538,
+      "loss": 1.0423,
+      "step": 1270
+    },
+    {
+      "epoch": 0.3925643690105069,
+      "grad_norm": 0.43365441716522,
+      "learning_rate": 0.0001522177676659508,
+      "loss": 1.0706,
+      "step": 1275
+    },
+    {
+      "epoch": 0.39410383712427355,
+      "grad_norm": 0.5576831403247644,
+      "learning_rate": 0.00015175855102531887,
+      "loss": 1.0199,
+      "step": 1280
+    },
+    {
+      "epoch": 0.39564330523804025,
+      "grad_norm": 0.4746796767312794,
+      "learning_rate": 0.00015129783862992283,
+      "loss": 1.029,
+      "step": 1285
+    },
+    {
+      "epoch": 0.39718277335180696,
+      "grad_norm": 0.5397433381233686,
+      "learning_rate": 0.0001508356437937512,
+      "loss": 1.0527,
+      "step": 1290
+    },
+    {
+      "epoch": 0.39872224146557367,
+      "grad_norm": 0.7141429909374478,
+      "learning_rate": 0.00015037197987363338,
+      "loss": 1.0348,
+      "step": 1295
+    },
+    {
+      "epoch": 0.4002617095793403,
+      "grad_norm": 0.4435301474074775,
+      "learning_rate": 0.0001499068602688532,
+      "loss": 1.0702,
+      "step": 1300
+    },
+    {
+      "epoch": 0.401801177693107,
+      "grad_norm": 0.5168064407110803,
+      "learning_rate": 0.00014944029842076185,
+      "loss": 1.0367,
+      "step": 1305
+    },
+    {
+      "epoch": 0.40334064580687373,
+      "grad_norm": 0.5215356272036379,
+      "learning_rate": 0.0001489723078123896,
+      "loss": 1.0458,
+      "step": 1310
+    },
+    {
+      "epoch": 0.40488011392064044,
+      "grad_norm": 0.47530585507691386,
+      "learning_rate": 0.00014850290196805594,
+      "loss": 1.0496,
+      "step": 1315
+    },
+    {
+      "epoch": 0.4064195820344071,
+      "grad_norm": 0.4940892436164985,
+      "learning_rate": 0.00014803209445297887,
+      "loss": 1.0142,
+      "step": 1320
+    },
+    {
+      "epoch": 0.4079590501481738,
+      "grad_norm": 0.5452384723470196,
+      "learning_rate": 0.00014755989887288285,
+      "loss": 1.0494,
+      "step": 1325
+    },
+    {
+      "epoch": 0.4094985182619405,
+      "grad_norm": 0.6302906153858793,
+      "learning_rate": 0.00014708632887360564,
+      "loss": 1.0474,
+      "step": 1330
+    },
+    {
+      "epoch": 0.4110379863757072,
+      "grad_norm": 0.5314488438447993,
+      "learning_rate": 0.0001466113981407039,
+      "loss": 1.0539,
+      "step": 1335
+    },
+    {
+      "epoch": 0.41257745448947386,
+      "grad_norm": 0.5583239486533321,
+      "learning_rate": 0.00014613512039905765,
+      "loss": 1.0425,
+      "step": 1340
+    },
+    {
+      "epoch": 0.41411692260324057,
+      "grad_norm": 0.5284692971206317,
+      "learning_rate": 0.00014565750941247386,
+      "loss": 1.02,
+      "step": 1345
+    },
+    {
+      "epoch": 0.4156563907170073,
+      "grad_norm": 0.5504329155738187,
+      "learning_rate": 0.0001451785789832884,
+      "loss": 1.0266,
+      "step": 1350
+    },
+    {
+      "epoch": 0.417195858830774,
+      "grad_norm": 0.7377959516300412,
+      "learning_rate": 0.00014469834295196743,
+      "loss": 1.0567,
+      "step": 1355
+    },
+    {
+      "epoch": 0.4187353269445407,
+      "grad_norm": 0.5133810725829329,
+      "learning_rate": 0.00014421681519670722,
+      "loss": 1.0516,
+      "step": 1360
+    },
+    {
+      "epoch": 0.42027479505830734,
+      "grad_norm": 0.5092592418186027,
+      "learning_rate": 0.0001437340096330332,
+      "loss": 1.058,
+      "step": 1365
+    },
+    {
+      "epoch": 0.42181426317207404,
+      "grad_norm": 0.6808151766449716,
+      "learning_rate": 0.0001432499402133979,
+      "loss": 1.046,
+      "step": 1370
+    },
+    {
+      "epoch": 0.42335373128584075,
+      "grad_norm": 0.49715289560889164,
+      "learning_rate": 0.0001427646209267775,
+      "loss": 1.0384,
+      "step": 1375
+    },
+    {
+      "epoch": 0.42489319939960746,
+      "grad_norm": 0.4073699217300379,
+      "learning_rate": 0.00014227806579826774,
+      "loss": 1.0181,
+      "step": 1380
+    },
+    {
+      "epoch": 0.4264326675133741,
+      "grad_norm": 0.5386638584236078,
+      "learning_rate": 0.00014179028888867867,
+      "loss": 1.0816,
+      "step": 1385
+    },
+    {
+      "epoch": 0.4279721356271408,
+      "grad_norm": 0.7007555607650736,
+      "learning_rate": 0.00014130130429412815,
+      "loss": 1.0364,
+      "step": 1390
+    },
+    {
+      "epoch": 0.4295116037409075,
+      "grad_norm": 0.6291522129011135,
+      "learning_rate": 0.0001408111261456346,
+      "loss": 1.018,
+      "step": 1395
+    },
+    {
+      "epoch": 0.43105107185467423,
+      "grad_norm": 0.4183183608173097,
+      "learning_rate": 0.00014031976860870855,
+      "loss": 1.0159,
+      "step": 1400
+    },
+    {
+      "epoch": 0.4325905399684409,
+      "grad_norm": 0.42462992648226483,
+      "learning_rate": 0.00013982724588294335,
+      "loss": 1.026,
+      "step": 1405
+    },
+    {
+      "epoch": 0.4341300080822076,
+      "grad_norm": 0.5342525321314037,
+      "learning_rate": 0.00013933357220160476,
+      "loss": 1.0668,
+      "step": 1410
+    },
+    {
+      "epoch": 0.4356694761959743,
+      "grad_norm": 0.5175732551515752,
+      "learning_rate": 0.00013883876183121973,
+      "loss": 1.0613,
+      "step": 1415
+    },
+    {
+      "epoch": 0.437208944309741,
+      "grad_norm": 0.5236448642390229,
+      "learning_rate": 0.000138342829071164,
+      "loss": 1.0505,
+      "step": 1420
+    },
+    {
+      "epoch": 0.4387484124235077,
+      "grad_norm": 0.47885357545447377,
+      "learning_rate": 0.00013784578825324885,
+      "loss": 1.0181,
+      "step": 1425
+    },
+    {
+      "epoch": 0.44028788053727436,
+      "grad_norm": 0.567561554601715,
+      "learning_rate": 0.00013734765374130717,
+      "loss": 1.0337,
+      "step": 1430
+    },
+    {
+      "epoch": 0.44182734865104106,
+      "grad_norm": 0.4640525016441619,
+      "learning_rate": 0.00013684843993077788,
+      "loss": 1.0195,
+      "step": 1435
+    },
+    {
+      "epoch": 0.44336681676480777,
+      "grad_norm": 0.6957136092069967,
+      "learning_rate": 0.00013634816124829063,
+      "loss": 1.0252,
+      "step": 1440
+    },
+    {
+      "epoch": 0.4449062848785745,
+      "grad_norm": 0.414100639900688,
+      "learning_rate": 0.0001358468321512481,
+      "loss": 1.0625,
+      "step": 1445
+    },
+    {
+      "epoch": 0.4464457529923411,
+      "grad_norm": 0.4358981186515927,
+      "learning_rate": 0.00013534446712740877,
+      "loss": 1.0336,
+      "step": 1450
+    },
+    {
+      "epoch": 0.44798522110610783,
+      "grad_norm": 0.7000263701281964,
+      "learning_rate": 0.0001348410806944681,
+      "loss": 1.0247,
+      "step": 1455
+    },
+    {
+      "epoch": 0.44952468921987454,
+      "grad_norm": 0.48487218580357255,
+      "learning_rate": 0.00013433668739963882,
+      "loss": 1.0306,
+      "step": 1460
+    },
+    {
+      "epoch": 0.45106415733364125,
+      "grad_norm": 0.48692168534098523,
+      "learning_rate": 0.00013383130181923071,
+      "loss": 1.0311,
+      "step": 1465
+    },
+    {
+      "epoch": 0.4526036254474079,
+      "grad_norm": 0.6087744044400206,
+      "learning_rate": 0.00013332493855822936,
+      "loss": 1.0138,
+      "step": 1470
+    },
+    {
+      "epoch": 0.4541430935611746,
+      "grad_norm": 0.5570901501341506,
+      "learning_rate": 0.00013281761224987398,
+      "loss": 1.0121,
+      "step": 1475
+    },
+    {
+      "epoch": 0.4556825616749413,
+      "grad_norm": 0.49929406825113404,
+      "learning_rate": 0.00013230933755523466,
+      "loss": 1.0345,
+      "step": 1480
+    },
+    {
+      "epoch": 0.457222029788708,
+      "grad_norm": 0.7306093657667501,
+      "learning_rate": 0.00013180012916278854,
+      "loss": 1.0472,
+      "step": 1485
+    },
+    {
+      "epoch": 0.45876149790247467,
+      "grad_norm": 0.46118596282419816,
+      "learning_rate": 0.00013129000178799548,
+      "loss": 1.0441,
+      "step": 1490
+    },
+    {
+      "epoch": 0.4603009660162414,
+      "grad_norm": 0.4542565239229393,
+      "learning_rate": 0.00013077897017287272,
+      "loss": 1.0092,
+      "step": 1495
+    },
+    {
+      "epoch": 0.4618404341300081,
+      "grad_norm": 0.4921710657691845,
+      "learning_rate": 0.00013026704908556888,
+      "loss": 1.0637,
+      "step": 1500
+    },
+    {
+      "epoch": 0.4633799022437748,
+      "grad_norm": 0.5001776822700446,
+      "learning_rate": 0.0001297542533199371,
+      "loss": 1.0483,
+      "step": 1505
+    },
+    {
+      "epoch": 0.4649193703575415,
+      "grad_norm": 0.463427010769576,
+      "learning_rate": 0.00012924059769510768,
+      "loss": 1.0388,
+      "step": 1510
+    },
+    {
+      "epoch": 0.46645883847130815,
+      "grad_norm": 0.5055392500122334,
+      "learning_rate": 0.00012872609705505964,
+      "loss": 1.0576,
+      "step": 1515
+    },
+    {
+      "epoch": 0.46799830658507485,
+      "grad_norm": 0.4980518480504977,
+      "learning_rate": 0.00012821076626819196,
+      "loss": 1.0485,
+      "step": 1520
+    },
+    {
+      "epoch": 0.46953777469884156,
+      "grad_norm": 0.4390801115545415,
+      "learning_rate": 0.00012769462022689363,
+      "loss": 1.029,
+      "step": 1525
+    },
+    {
+      "epoch": 0.47107724281260827,
+      "grad_norm": 0.48684411734981153,
+      "learning_rate": 0.0001271776738471136,
+      "loss": 1.0353,
+      "step": 1530
+    },
+    {
+      "epoch": 0.4726167109263749,
+      "grad_norm": 0.6254296920632078,
+      "learning_rate": 0.00012665994206792938,
+      "loss": 1.0662,
+      "step": 1535
+    },
+    {
+      "epoch": 0.4741561790401416,
+      "grad_norm": 0.4828153968193146,
+      "learning_rate": 0.00012614143985111565,
+      "loss": 1.1062,
+      "step": 1540
+    },
+    {
+      "epoch": 0.47569564715390833,
+      "grad_norm": 0.6105347585674029,
+      "learning_rate": 0.00012562218218071164,
+      "loss": 1.0285,
+      "step": 1545
+    },
+    {
+      "epoch": 0.47723511526767504,
+      "grad_norm": 0.4580317847852175,
+      "learning_rate": 0.0001251021840625883,
+      "loss": 1.0257,
+      "step": 1550
+    },
+    {
+      "epoch": 0.4787745833814417,
+      "grad_norm": 0.49671748534126925,
+      "learning_rate": 0.00012458146052401442,
+      "loss": 1.0347,
+      "step": 1555
+    },
+    {
+      "epoch": 0.4803140514952084,
+      "grad_norm": 0.5504358336120402,
+      "learning_rate": 0.00012406002661322264,
+      "loss": 1.0246,
+      "step": 1560
+    },
+    {
+      "epoch": 0.4818535196089751,
+      "grad_norm": 0.4998876160614867,
+      "learning_rate": 0.00012353789739897437,
+      "loss": 1.0574,
+      "step": 1565
+    },
+    {
+      "epoch": 0.4833929877227418,
+      "grad_norm": 0.5729717160065313,
+      "learning_rate": 0.00012301508797012432,
+      "loss": 1.0365,
+      "step": 1570
+    },
+    {
+      "epoch": 0.4849324558365085,
+      "grad_norm": 0.47052381273276284,
+      "learning_rate": 0.00012249161343518466,
+      "loss": 1.0173,
+      "step": 1575
+    },
+    {
+      "epoch": 0.48647192395027516,
+      "grad_norm": 0.46095882989505704,
+      "learning_rate": 0.00012196748892188816,
+      "loss": 1.0497,
+      "step": 1580
+    },
+    {
+      "epoch": 0.48801139206404187,
+      "grad_norm": 0.654430908611628,
+      "learning_rate": 0.00012144272957675108,
+      "loss": 1.0312,
+      "step": 1585
+    },
+    {
+      "epoch": 0.4895508601778086,
+      "grad_norm": 0.45228923273407706,
+      "learning_rate": 0.00012091735056463562,
+      "loss": 1.0093,
+      "step": 1590
+    },
+    {
+      "epoch": 0.4910903282915753,
+      "grad_norm": 0.6127718568028115,
+      "learning_rate": 0.00012039136706831145,
+      "loss": 1.0671,
+      "step": 1595
+    },
+    {
+      "epoch": 0.49262979640534194,
+      "grad_norm": 0.6260979638098746,
+      "learning_rate": 0.00011986479428801709,
+      "loss": 1.0436,
+      "step": 1600
+    },
+    {
+      "epoch": 0.49416926451910864,
+      "grad_norm": 0.4570968434668229,
+      "learning_rate": 0.00011933764744102058,
+      "loss": 1.0229,
+      "step": 1605
+    },
+    {
+      "epoch": 0.49570873263287535,
+      "grad_norm": 0.5515514688898091,
+      "learning_rate": 0.00011880994176117976,
+      "loss": 1.0449,
+      "step": 1610
+    },
+    {
+      "epoch": 0.49724820074664206,
+      "grad_norm": 0.42355848666393986,
+      "learning_rate": 0.00011828169249850201,
+      "loss": 1.0453,
+      "step": 1615
+    },
+    {
+      "epoch": 0.4987876688604087,
+      "grad_norm": 0.5463051906159438,
+      "learning_rate": 0.00011775291491870351,
+      "loss": 1.0271,
+      "step": 1620
+    },
+    {
+      "epoch": 0.5003271369741754,
+      "grad_norm": 0.4043160515062089,
+      "learning_rate": 0.00011722362430276816,
+      "loss": 1.0377,
+      "step": 1625
+    },
+    {
+      "epoch": 0.5018666050879421,
+      "grad_norm": 0.8243302494042759,
+      "learning_rate": 0.00011669383594650593,
+      "loss": 1.0663,
+      "step": 1630
+    },
+    {
+      "epoch": 0.5034060732017088,
+      "grad_norm": 0.4547900229781827,
+      "learning_rate": 0.00011616356516011083,
+      "loss": 1.0159,
+      "step": 1635
+    },
+    {
+      "epoch": 0.5049455413154755,
+      "grad_norm": 0.5227895926154218,
+      "learning_rate": 0.00011563282726771847,
+      "loss": 1.0515,
+      "step": 1640
+    },
+    {
+      "epoch": 0.5064850094292422,
+      "grad_norm": 0.5635603323984416,
+      "learning_rate": 0.0001151016376069632,
+      "loss": 1.0701,
+      "step": 1645
+    },
+    {
+      "epoch": 0.508024477543009,
+      "grad_norm": 0.47093920101316833,
+      "learning_rate": 0.00011457001152853493,
+      "loss": 1.0199,
+      "step": 1650
+    },
+    {
+      "epoch": 0.5095639456567755,
+      "grad_norm": 0.4344516158624125,
+      "learning_rate": 0.00011403796439573544,
+      "loss": 1.038,
+      "step": 1655
+    },
+    {
+      "epoch": 0.5111034137705422,
+      "grad_norm": 0.49922332593006863,
+      "learning_rate": 0.00011350551158403442,
+      "loss": 1.0593,
+      "step": 1660
+    },
+    {
+      "epoch": 0.512642881884309,
+      "grad_norm": 0.42072128316053065,
+      "learning_rate": 0.0001129726684806252,
+      "loss": 1.0147,
+      "step": 1665
+    },
+    {
+      "epoch": 0.5141823499980757,
+      "grad_norm": 0.4531404971038023,
+      "learning_rate": 0.00011243945048398003,
+      "loss": 1.0215,
+      "step": 1670
+    },
+    {
+      "epoch": 0.5157218181118424,
+      "grad_norm": 0.4470522441927525,
+      "learning_rate": 0.000111905873003405,
+      "loss": 1.0323,
+      "step": 1675
+    },
+    {
+      "epoch": 0.5172612862256091,
+      "grad_norm": 0.5867123309400825,
+      "learning_rate": 0.00011137195145859494,
+      "loss": 1.0046,
+      "step": 1680
+    },
+    {
+      "epoch": 0.5188007543393758,
+      "grad_norm": 0.3962300365929646,
+      "learning_rate": 0.00011083770127918762,
+      "loss": 1.0043,
+      "step": 1685
+    },
+    {
+      "epoch": 0.5203402224531425,
+      "grad_norm": 0.7925921747531262,
+      "learning_rate": 0.00011030313790431788,
+      "loss": 1.049,
+      "step": 1690
+    },
+    {
+      "epoch": 0.5218796905669091,
+      "grad_norm": 0.5078998609582865,
+      "learning_rate": 0.00010976827678217161,
+      "loss": 1.0109,
+      "step": 1695
+    },
+    {
+      "epoch": 0.5234191586806758,
+      "grad_norm": 0.4109138516526712,
+      "learning_rate": 0.00010923313336953913,
+      "loss": 1.0172,
+      "step": 1700
+    },
+    {
+      "epoch": 0.5249586267944425,
+      "grad_norm": 0.5367265822698736,
+      "learning_rate": 0.00010869772313136861,
+      "loss": 1.0285,
+      "step": 1705
+    },
+    {
+      "epoch": 0.5264980949082092,
+      "grad_norm": 0.45736224715965773,
+      "learning_rate": 0.00010816206154031916,
+      "loss": 1.0146,
+      "step": 1710
+    },
+    {
+      "epoch": 0.5280375630219759,
+      "grad_norm": 0.46461493826686345,
+      "learning_rate": 0.00010762616407631356,
+      "loss": 1.0783,
+      "step": 1715
+    },
+    {
+      "epoch": 0.5295770311357426,
+      "grad_norm": 0.6140868832303202,
+      "learning_rate": 0.00010709004622609116,
+      "loss": 1.0748,
+      "step": 1720
+    },
+    {
+      "epoch": 0.5311164992495093,
+      "grad_norm": 0.6800071721536182,
+      "learning_rate": 0.00010655372348276006,
+      "loss": 1.0272,
+      "step": 1725
+    },
+    {
+      "epoch": 0.532655967363276,
+      "grad_norm": 0.5202497163301416,
+      "learning_rate": 0.00010601721134534959,
+      "loss": 1.0379,
+      "step": 1730
+    },
+    {
+      "epoch": 0.5341954354770427,
+      "grad_norm": 0.4826223398674556,
+      "learning_rate": 0.00010548052531836223,
+      "loss": 1.0357,
+      "step": 1735
+    },
+    {
+      "epoch": 0.5357349035908093,
+      "grad_norm": 1.0784079166978915,
+      "learning_rate": 0.00010494368091132576,
+      "loss": 1.0378,
+      "step": 1740
+    },
+    {
+      "epoch": 0.537274371704576,
+      "grad_norm": 0.4567747436804465,
+      "learning_rate": 0.00010440669363834483,
+      "loss": 1.0188,
+      "step": 1745
+    },
+    {
+      "epoch": 0.5388138398183427,
+      "grad_norm": 0.48032495281062815,
+      "learning_rate": 0.00010386957901765277,
+      "loss": 1.034,
+      "step": 1750
+    },
+    {
+      "epoch": 0.5403533079321095,
+      "grad_norm": 0.5026624164429033,
+      "learning_rate": 0.00010333235257116313,
+      "loss": 0.9786,
+      "step": 1755
+    },
+    {
+      "epoch": 0.5418927760458762,
+      "grad_norm": 0.48003816426131585,
+      "learning_rate": 0.00010279502982402103,
+      "loss": 1.0486,
+      "step": 1760
+    },
+    {
+      "epoch": 0.5434322441596429,
+      "grad_norm": 0.5389477916705987,
+      "learning_rate": 0.00010225762630415457,
+      "loss": 1.017,
+      "step": 1765
+    },
+    {
+      "epoch": 0.5449717122734096,
+      "grad_norm": 0.537004045922141,
+      "learning_rate": 0.00010172015754182607,
+      "loss": 1.0444,
+      "step": 1770
+    },
+    {
+      "epoch": 0.5465111803871763,
+      "grad_norm": 1.929543100464396,
+      "learning_rate": 0.00010118263906918331,
+      "loss": 1.0451,
+      "step": 1775
+    },
+    {
+      "epoch": 0.5480506485009429,
+      "grad_norm": 0.4940854946649758,
+      "learning_rate": 0.00010064508641981054,
+      "loss": 1.0013,
+      "step": 1780
+    },
+    {
+      "epoch": 0.5495901166147096,
+      "grad_norm": 0.4816239235355272,
+      "learning_rate": 0.0001001075151282798,
+      "loss": 1.0568,
+      "step": 1785
+    },
+    {
+      "epoch": 0.5511295847284763,
+      "grad_norm": 0.4644540038274215,
+      "learning_rate": 9.956994072970179e-05,
+      "loss": 1.0332,
+      "step": 1790
+    },
+    {
+      "epoch": 0.552669052842243,
+      "grad_norm": 0.39735664952249866,
+      "learning_rate": 9.903237875927698e-05,
+      "loss": 1.0716,
+      "step": 1795
+    },
+    {
+      "epoch": 0.5542085209560097,
+      "grad_norm": 0.42204715257283004,
+      "learning_rate": 9.849484475184672e-05,
+      "loss": 1.0204,
+      "step": 1800
+    },
+    {
+      "epoch": 0.5557479890697764,
+      "grad_norm": 0.5226725559414837,
+      "learning_rate": 9.795735424144428e-05,
+      "loss": 1.0156,
+      "step": 1805
+    },
+    {
+      "epoch": 0.5572874571835431,
+      "grad_norm": 0.5721667039150645,
+      "learning_rate": 9.74199227608459e-05,
+      "loss": 1.0381,
+      "step": 1810
+    },
+    {
+      "epoch": 0.5588269252973098,
+      "grad_norm": 0.6792294345813789,
+      "learning_rate": 9.688256584112192e-05,
+      "loss": 1.0216,
+      "step": 1815
+    },
+    {
+      "epoch": 0.5603663934110765,
+      "grad_norm": 0.5038747704164618,
+      "learning_rate": 9.634529901118799e-05,
+      "loss": 1.0309,
+      "step": 1820
+    },
+    {
+      "epoch": 0.5619058615248431,
+      "grad_norm": 1.0905768705043635,
+      "learning_rate": 9.580813779735624e-05,
+      "loss": 1.0417,
+      "step": 1825
+    },
+    {
+      "epoch": 0.5634453296386098,
+      "grad_norm": 0.7259613249848627,
+      "learning_rate": 9.52710977228867e-05,
+      "loss": 1.0348,
+      "step": 1830
+    },
+    {
+      "epoch": 0.5649847977523765,
+      "grad_norm": 0.48026375265768084,
+      "learning_rate": 9.473419430753864e-05,
+      "loss": 0.979,
+      "step": 1835
+    },
+    {
+      "epoch": 0.5665242658661432,
+      "grad_norm": 0.5445245479851468,
+      "learning_rate": 9.419744306712197e-05,
+      "loss": 1.0099,
+      "step": 1840
+    },
+    {
+      "epoch": 0.56806373397991,
+      "grad_norm": 0.4377954309966568,
+      "learning_rate": 9.3660859513049e-05,
+      "loss": 1.0681,
+      "step": 1845
+    },
+    {
+      "epoch": 0.5696032020936767,
+      "grad_norm": 0.7151357125545018,
+      "learning_rate": 9.312445915188609e-05,
+      "loss": 1.033,
+      "step": 1850
+    },
+    {
+      "epoch": 0.5711426702074434,
+      "grad_norm": 0.5258015505597993,
+      "learning_rate": 9.258825748490558e-05,
+      "loss": 1.0492,
+      "step": 1855
+    },
+    {
+      "epoch": 0.5726821383212101,
+      "grad_norm": 0.5535249863969953,
+      "learning_rate": 9.205227000763788e-05,
+      "loss": 1.0444,
+      "step": 1860
+    },
+    {
+      "epoch": 0.5742216064349767,
+      "grad_norm": 0.5852809671349769,
+      "learning_rate": 9.151651220942349e-05,
+      "loss": 1.0532,
+      "step": 1865
+    },
+    {
+      "epoch": 0.5757610745487434,
+      "grad_norm": 0.5927067468229194,
+      "learning_rate": 9.098099957296552e-05,
+      "loss": 1.0419,
+      "step": 1870
+    },
+    {
+      "epoch": 0.5773005426625101,
+      "grad_norm": 0.5038197549382673,
+      "learning_rate": 9.044574757388224e-05,
+      "loss": 1.0342,
+      "step": 1875
+    },
+    {
+      "epoch": 0.5788400107762768,
+      "grad_norm": 0.45976069318212914,
+      "learning_rate": 8.991077168025976e-05,
+      "loss": 1.0346,
+      "step": 1880
+    },
+    {
+      "epoch": 0.5803794788900435,
+      "grad_norm": 0.6004992124353027,
+      "learning_rate": 8.937608735220527e-05,
+      "loss": 1.0457,
+      "step": 1885
+    },
+    {
+      "epoch": 0.5819189470038102,
+      "grad_norm": 0.6603686161055189,
+      "learning_rate": 8.884171004139996e-05,
+      "loss": 1.0233,
+      "step": 1890
+    },
+    {
+      "epoch": 0.5834584151175769,
+      "grad_norm": 0.502662627180952,
+      "learning_rate": 8.830765519065262e-05,
+      "loss": 0.9875,
+      "step": 1895
+    },
+    {
+      "epoch": 0.5849978832313436,
+      "grad_norm": 0.7865449254332316,
+      "learning_rate": 8.777393823345343e-05,
+      "loss": 1.0493,
+      "step": 1900
+    },
+    {
+      "epoch": 0.5865373513451103,
+      "grad_norm": 0.4226972216471069,
+      "learning_rate": 8.724057459352784e-05,
+      "loss": 0.9935,
+      "step": 1905
+    },
+    {
+      "epoch": 0.5880768194588769,
+      "grad_norm": 0.5958711278623102,
+      "learning_rate": 8.670757968439086e-05,
+      "loss": 1.0631,
+      "step": 1910
+    },
+    {
+      "epoch": 0.5896162875726436,
+      "grad_norm": 0.5199697414979809,
+      "learning_rate": 8.617496890890179e-05,
+      "loss": 1.0343,
+      "step": 1915
+    },
+    {
+      "epoch": 0.5911557556864103,
+      "grad_norm": 0.6008327984030076,
+      "learning_rate": 8.564275765881887e-05,
+      "loss": 1.0423,
+      "step": 1920
+    },
+    {
+      "epoch": 0.592695223800177,
+      "grad_norm": 0.44604940325807296,
+      "learning_rate": 8.511096131435454e-05,
+      "loss": 1.0175,
+      "step": 1925
+    },
+    {
+      "epoch": 0.5942346919139437,
+      "grad_norm": 0.42675758280608317,
+      "learning_rate": 8.457959524373109e-05,
+      "loss": 1.0307,
+      "step": 1930
+    },
+    {
+      "epoch": 0.5957741600277104,
+      "grad_norm": 0.5597345800622264,
+      "learning_rate": 8.404867480273636e-05,
+      "loss": 1.058,
+      "step": 1935
+    },
+    {
+      "epoch": 0.5973136281414771,
+      "grad_norm": 0.4451211247529823,
+      "learning_rate": 8.351821533428023e-05,
+      "loss": 1.0313,
+      "step": 1940
+    },
+    {
+      "epoch": 0.5988530962552439,
+      "grad_norm": 0.407092060921353,
+      "learning_rate": 8.298823216795093e-05,
+      "loss": 1.0506,
+      "step": 1945
+    },
+    {
+      "epoch": 0.6003925643690106,
+      "grad_norm": 0.43348547367002177,
+      "learning_rate": 8.245874061957224e-05,
+      "loss": 1.0402,
+      "step": 1950
+    },
+    {
+      "epoch": 0.6019320324827772,
+      "grad_norm": 0.4472151967667229,
+      "learning_rate": 8.192975599076078e-05,
+      "loss": 1.0168,
+      "step": 1955
+    },
+    {
+      "epoch": 0.6034715005965439,
+      "grad_norm": 0.3853019563837129,
+      "learning_rate": 8.140129356848387e-05,
+      "loss": 1.0206,
+      "step": 1960
+    },
+    {
+      "epoch": 0.6050109687103106,
+      "grad_norm": 0.4721720358418525,
+      "learning_rate": 8.087336862461783e-05,
+      "loss": 1.0112,
+      "step": 1965
+    },
+    {
+      "epoch": 0.6065504368240773,
+      "grad_norm": 0.46617796491326624,
+      "learning_rate": 8.034599641550642e-05,
+      "loss": 1.0484,
+      "step": 1970
+    },
+    {
+      "epoch": 0.608089904937844,
+      "grad_norm": 0.5930150370566495,
+      "learning_rate": 7.981919218152016e-05,
+      "loss": 1.0019,
+      "step": 1975
+    },
+    {
+      "epoch": 0.6096293730516107,
+      "grad_norm": 0.515110909779576,
+      "learning_rate": 7.929297114661581e-05,
+      "loss": 1.0182,
+      "step": 1980
+    },
+    {
+      "epoch": 0.6111688411653774,
+      "grad_norm": 0.45067927700399973,
+      "learning_rate": 7.876734851789643e-05,
+      "loss": 1.048,
+      "step": 1985
+    },
+    {
+      "epoch": 0.6127083092791441,
+      "grad_norm": 0.48025135310003386,
+      "learning_rate": 7.824233948517185e-05,
+      "loss": 1.0499,
+      "step": 1990
+    },
+    {
+      "epoch": 0.6142477773929107,
+      "grad_norm": 0.3688608602701041,
+      "learning_rate": 7.771795922051999e-05,
+      "loss": 1.0493,
+      "step": 1995
+    },
+    {
+      "epoch": 0.6157872455066774,
+      "grad_norm": 0.47692666795291627,
+      "learning_rate": 7.719422287784798e-05,
+      "loss": 1.018,
+      "step": 2000
+    },
+    {
+      "epoch": 0.6173267136204441,
+      "grad_norm": 0.43950677355297135,
+      "learning_rate": 7.667114559245451e-05,
+      "loss": 1.0086,
+      "step": 2005
+    },
+    {
+      "epoch": 0.6188661817342108,
+      "grad_norm": 0.5992589739511112,
+      "learning_rate": 7.614874248059238e-05,
+      "loss": 1.0286,
+      "step": 2010
+    },
+    {
+      "epoch": 0.6204056498479775,
+      "grad_norm": 0.5113558280370261,
+      "learning_rate": 7.56270286390316e-05,
+      "loss": 1.0196,
+      "step": 2015
+    },
+    {
+      "epoch": 0.6219451179617442,
+      "grad_norm": 0.48971205943506146,
+      "learning_rate": 7.510601914462331e-05,
+      "loss": 1.0021,
+      "step": 2020
+    },
+    {
+      "epoch": 0.6234845860755109,
+      "grad_norm": 2.031201004984545,
+      "learning_rate": 7.458572905386381e-05,
+      "loss": 1.0068,
+      "step": 2025
+    },
+    {
+      "epoch": 0.6250240541892776,
+      "grad_norm": 0.43332008516546355,
+      "learning_rate": 7.406617340245957e-05,
+      "loss": 1.0573,
+      "step": 2030
+    },
+    {
+      "epoch": 0.6265635223030444,
+      "grad_norm": 0.40359150277813904,
+      "learning_rate": 7.354736720489273e-05,
+      "loss": 1.0114,
+      "step": 2035
+    },
+    {
+      "epoch": 0.628102990416811,
+      "grad_norm": 0.5109830787118577,
+      "learning_rate": 7.302932545398721e-05,
+      "loss": 1.0325,
+      "step": 2040
+    },
+    {
+      "epoch": 0.6296424585305777,
+      "grad_norm": 0.33695770975399714,
+      "learning_rate": 7.251206312047547e-05,
+      "loss": 0.9971,
+      "step": 2045
+    },
+    {
+      "epoch": 0.6311819266443444,
+      "grad_norm": 0.4478834681612594,
+      "learning_rate": 7.199559515256573e-05,
+      "loss": 1.0551,
+      "step": 2050
+    },
+    {
+      "epoch": 0.6327213947581111,
+      "grad_norm": 0.49179933979909946,
+      "learning_rate": 7.14799364755101e-05,
+      "loss": 1.0181,
+      "step": 2055
+    },
+    {
+      "epoch": 0.6342608628718778,
+      "grad_norm": 0.476264555556864,
+      "learning_rate": 7.096510199117327e-05,
+      "loss": 1.0346,
+      "step": 2060
+    },
+    {
+      "epoch": 0.6358003309856445,
+      "grad_norm": 0.45265771641362673,
+      "learning_rate": 7.045110657760179e-05,
+      "loss": 1.0442,
+      "step": 2065
+    },
+    {
+      "epoch": 0.6373397990994112,
+      "grad_norm": 0.4988192869036288,
+      "learning_rate": 6.993796508859418e-05,
+      "loss": 1.029,
+      "step": 2070
+    },
+    {
+      "epoch": 0.6388792672131779,
+      "grad_norm": 0.4886571640531873,
+      "learning_rate": 6.942569235327167e-05,
+      "loss": 1.0275,
+      "step": 2075
+    },
+    {
+      "epoch": 0.6404187353269445,
+      "grad_norm": 0.43831855733672936,
+      "learning_rate": 6.891430317564964e-05,
+      "loss": 1.0307,
+      "step": 2080
+    },
+    {
+      "epoch": 0.6419582034407112,
+      "grad_norm": 0.5860619060156503,
+      "learning_rate": 6.840381233420973e-05,
+      "loss": 1.0265,
+      "step": 2085
+    },
+    {
+      "epoch": 0.6434976715544779,
+      "grad_norm": 0.44236844600518777,
+      "learning_rate": 6.789423458147292e-05,
+      "loss": 1.0283,
+      "step": 2090
+    },
+    {
+      "epoch": 0.6450371396682446,
+      "grad_norm": 0.4876716811387609,
+      "learning_rate": 6.738558464357305e-05,
+      "loss": 1.0504,
+      "step": 2095
+    },
+    {
+      "epoch": 0.6465766077820113,
+      "grad_norm": 0.41217817376284266,
+      "learning_rate": 6.687787721983136e-05,
+      "loss": 1.0725,
+      "step": 2100
+    },
+    {
+      "epoch": 0.648116075895778,
+      "grad_norm": 0.5066902107865301,
+      "learning_rate": 6.63711269823317e-05,
+      "loss": 1.0168,
+      "step": 2105
+    },
+    {
+      "epoch": 0.6496555440095447,
+      "grad_norm": 0.8959396144411056,
+      "learning_rate": 6.586534857549638e-05,
+      "loss": 1.0367,
+      "step": 2110
+    },
+    {
+      "epoch": 0.6511950121233114,
+      "grad_norm": 0.41740196707008226,
+      "learning_rate": 6.536055661566312e-05,
+      "loss": 1.0603,
+      "step": 2115
+    },
+    {
+      "epoch": 0.6527344802370781,
+      "grad_norm": 0.4285108933692639,
+      "learning_rate": 6.485676569066258e-05,
+      "loss": 1.0701,
+      "step": 2120
+    },
+    {
+      "epoch": 0.6542739483508447,
+      "grad_norm": 0.490878565864062,
+      "learning_rate": 6.43539903593969e-05,
+      "loss": 1.0274,
+      "step": 2125
+    },
+    {
+      "epoch": 0.6558134164646114,
+      "grad_norm": 0.4955255588450408,
+      "learning_rate": 6.385224515141879e-05,
+      "loss": 1.0213,
+      "step": 2130
+    },
+    {
+      "epoch": 0.6573528845783781,
+      "grad_norm": 0.4943547579484523,
+      "learning_rate": 6.335154456651178e-05,
+      "loss": 1.0424,
+      "step": 2135
+    },
+    {
+      "epoch": 0.6588923526921449,
+      "grad_norm": 0.7266332089943804,
+      "learning_rate": 6.285190307427114e-05,
+      "loss": 1.0646,
+      "step": 2140
+    },
+    {
+      "epoch": 0.6604318208059116,
+      "grad_norm": 0.4121244632336502,
+      "learning_rate": 6.235333511368573e-05,
+      "loss": 1.0555,
+      "step": 2145
+    },
+    {
+      "epoch": 0.6619712889196783,
+      "grad_norm": 0.5185060524362,
+      "learning_rate": 6.185585509272078e-05,
+      "loss": 1.0469,
+      "step": 2150
+    },
+    {
+      "epoch": 0.663510757033445,
+      "grad_norm": 0.4631669349204426,
+      "learning_rate": 6.135947738790145e-05,
+      "loss": 1.0567,
+      "step": 2155
+    },
+    {
+      "epoch": 0.6650502251472117,
+      "grad_norm": 0.5057075839366066,
+      "learning_rate": 6.0864216343897365e-05,
+      "loss": 1.0298,
+      "step": 2160
+    },
+    {
+      "epoch": 0.6665896932609783,
+      "grad_norm": 0.5634732740375421,
+      "learning_rate": 6.0370086273108205e-05,
+      "loss": 1.0629,
+      "step": 2165
+    },
+    {
+      "epoch": 0.668129161374745,
+      "grad_norm": 0.5200825511831939,
+      "learning_rate": 5.987710145524992e-05,
+      "loss": 1.0519,
+      "step": 2170
+    },
+    {
+      "epoch": 0.6696686294885117,
+      "grad_norm": 0.4345129610373104,
+      "learning_rate": 5.938527613694214e-05,
+      "loss": 1.0321,
+      "step": 2175
+    },
+    {
+      "epoch": 0.6712080976022784,
+      "grad_norm": 0.7454797877228255,
+      "learning_rate": 5.8894624531296486e-05,
+      "loss": 0.9871,
+      "step": 2180
+    },
+    {
+      "epoch": 0.6727475657160451,
+      "grad_norm": 0.38956096747028734,
+      "learning_rate": 5.840516081750583e-05,
+      "loss": 1.0224,
+      "step": 2185
+    },
+    {
+      "epoch": 0.6742870338298118,
+      "grad_norm": 0.4637705509612328,
+      "learning_rate": 5.791689914043447e-05,
+      "loss": 1.066,
+      "step": 2190
+    },
+    {
+      "epoch": 0.6758265019435785,
+      "grad_norm": 0.4296286070627721,
+      "learning_rate": 5.742985361020945e-05,
+      "loss": 1.0286,
+      "step": 2195
+    },
+    {
+      "epoch": 0.6773659700573452,
+      "grad_norm": 0.4514049815518666,
+      "learning_rate": 5.69440383018127e-05,
+      "loss": 1.0443,
+      "step": 2200
+    },
+    {
+      "epoch": 0.6789054381711119,
+      "grad_norm": 0.4614392154051119,
+      "learning_rate": 5.6459467254674435e-05,
+      "loss": 0.9877,
+      "step": 2205
+    },
+    {
+      "epoch": 0.6804449062848785,
+      "grad_norm": 0.5704518491798019,
+      "learning_rate": 5.597615447226724e-05,
+      "loss": 1.0337,
+      "step": 2210
+    },
+    {
+      "epoch": 0.6819843743986452,
+      "grad_norm": 0.39999150350123325,
+      "learning_rate": 5.549411392170154e-05,
+      "loss": 1.0088,
+      "step": 2215
+    },
+    {
+      "epoch": 0.6835238425124119,
+      "grad_norm": 0.42287100632286095,
+      "learning_rate": 5.501335953332187e-05,
+      "loss": 1.0358,
+      "step": 2220
+    },
+    {
+      "epoch": 0.6850633106261786,
+      "grad_norm": 0.4444385205318582,
+      "learning_rate": 5.453390520030439e-05,
+      "loss": 1.0274,
+      "step": 2225
+    },
+    {
+      "epoch": 0.6866027787399454,
+      "grad_norm": 0.4681868371661994,
+      "learning_rate": 5.405576477825538e-05,
+      "loss": 1.0135,
+      "step": 2230
+    },
+    {
+      "epoch": 0.6881422468537121,
+      "grad_norm": 0.3647614554162627,
+      "learning_rate": 5.3578952084810765e-05,
+      "loss": 1.0201,
+      "step": 2235
+    },
+    {
+      "epoch": 0.6896817149674788,
+      "grad_norm": 0.4675279093656959,
+      "learning_rate": 5.310348089923681e-05,
+      "loss": 1.0641,
+      "step": 2240
+    },
+    {
+      "epoch": 0.6912211830812455,
+      "grad_norm": 0.5708390560359091,
+      "learning_rate": 5.2629364962032004e-05,
+      "loss": 1.0365,
+      "step": 2245
+    },
+    {
+      "epoch": 0.6927606511950122,
+      "grad_norm": 5.18192671698193,
+      "learning_rate": 5.2156617974529886e-05,
+      "loss": 1.0407,
+      "step": 2250
+    },
+    {
+      "epoch": 0.6943001193087788,
+      "grad_norm": 0.4187190501217835,
+      "learning_rate": 5.1685253598503116e-05,
+      "loss": 1.0471,
+      "step": 2255
+    },
+    {
+      "epoch": 0.6958395874225455,
+      "grad_norm": 0.6110758500884396,
+      "learning_rate": 5.1215285455768794e-05,
+      "loss": 1.0067,
+      "step": 2260
+    },
+    {
+      "epoch": 0.6973790555363122,
+      "grad_norm": 1.2232875510104708,
+      "learning_rate": 5.074672712779456e-05,
+      "loss": 1.0186,
+      "step": 2265
+    },
+    {
+      "epoch": 0.6989185236500789,
+      "grad_norm": 0.4924706604640269,
+      "learning_rate": 5.0279592155306286e-05,
+      "loss": 1.013,
+      "step": 2270
+    },
+    {
+      "epoch": 0.7004579917638456,
+      "grad_norm": 0.5351327628863634,
+      "learning_rate": 4.9813894037896747e-05,
+      "loss": 1.0147,
+      "step": 2275
+    },
+    {
+      "epoch": 0.7019974598776123,
+      "grad_norm": 0.5691385515488996,
+      "learning_rate": 4.93496462336354e-05,
+      "loss": 1.0543,
+      "step": 2280
+    },
+    {
+      "epoch": 0.703536927991379,
+      "grad_norm": 0.45370659861849166,
+      "learning_rate": 4.8886862158679714e-05,
+      "loss": 1.0502,
+      "step": 2285
+    },
+    {
+      "epoch": 0.7050763961051457,
+      "grad_norm": 0.39906039100573537,
+      "learning_rate": 4.8425555186887096e-05,
+      "loss": 1.027,
+      "step": 2290
+    },
+    {
+      "epoch": 0.7066158642189123,
+      "grad_norm": 0.48115067130308187,
+      "learning_rate": 4.796573864942868e-05,
+      "loss": 1.0515,
+      "step": 2295
+    },
+    {
+      "epoch": 0.708155332332679,
+      "grad_norm": 0.43162848352113553,
+      "learning_rate": 4.750742583440397e-05,
+      "loss": 1.0054,
+      "step": 2300
+    },
+    {
+      "epoch": 0.7096948004464457,
+      "grad_norm": 0.40152697201685444,
+      "learning_rate": 4.7050629986456873e-05,
+      "loss": 1.0279,
+      "step": 2305
+    },
+    {
+      "epoch": 0.7112342685602124,
+      "grad_norm": 0.43021780150976113,
+      "learning_rate": 4.65953643063929e-05,
+      "loss": 1.0391,
+      "step": 2310
+    },
+    {
+      "epoch": 0.7127737366739791,
+      "grad_norm": 0.42088619664607274,
+      "learning_rate": 4.6141641950797645e-05,
+      "loss": 0.988,
+      "step": 2315
+    },
+    {
+      "epoch": 0.7143132047877458,
+      "grad_norm": 0.4851450074591337,
+      "learning_rate": 4.5689476031656784e-05,
+      "loss": 1.0581,
+      "step": 2320
+    },
+    {
+      "epoch": 0.7158526729015126,
+      "grad_norm": 0.5755620737371696,
+      "learning_rate": 4.523887961597688e-05,
+      "loss": 1.0462,
+      "step": 2325
+    },
+    {
+      "epoch": 0.7173921410152793,
+      "grad_norm": 0.41663849058693325,
+      "learning_rate": 4.4789865725407934e-05,
+      "loss": 1.0329,
+      "step": 2330
+    },
+    {
+      "epoch": 0.718931609129046,
+      "grad_norm": 0.47615081033464657,
+      "learning_rate": 4.434244733586699e-05,
+      "loss": 1.0076,
+      "step": 2335
+    },
+    {
+      "epoch": 0.7204710772428126,
+      "grad_norm": 0.534330404158238,
+      "learning_rate": 4.389663737716324e-05,
+      "loss": 1.0224,
+      "step": 2340
+    },
+    {
+      "epoch": 0.7220105453565793,
+      "grad_norm": 0.439745239024214,
+      "learning_rate": 4.3452448732624264e-05,
+      "loss": 1.0034,
+      "step": 2345
+    },
+    {
+      "epoch": 0.723550013470346,
+      "grad_norm": 0.46055944759568124,
+      "learning_rate": 4.3009894238723856e-05,
+      "loss": 1.0089,
+      "step": 2350
+    },
+    {
+      "epoch": 0.7250894815841127,
+      "grad_norm": 0.5413958198748791,
+      "learning_rate": 4.256898668471092e-05,
+      "loss": 1.0401,
+      "step": 2355
+    },
+    {
+      "epoch": 0.7266289496978794,
+      "grad_norm": 0.460713247825251,
+      "learning_rate": 4.212973881223994e-05,
+      "loss": 1.0205,
+      "step": 2360
+    },
+    {
+      "epoch": 0.7281684178116461,
+      "grad_norm": 0.4721070044498834,
+      "learning_rate": 4.1692163315002784e-05,
+      "loss": 1.0292,
+      "step": 2365
+    },
+    {
+      "epoch": 0.7297078859254128,
+      "grad_norm": 0.448843218109221,
+      "learning_rate": 4.125627283836184e-05,
+      "loss": 1.0274,
+      "step": 2370
+    },
+    {
+      "epoch": 0.7312473540391795,
+      "grad_norm": 0.4152647845453433,
+      "learning_rate": 4.082207997898457e-05,
+      "loss": 0.9903,
+      "step": 2375
+    },
+    {
+      "epoch": 0.7327868221529461,
+      "grad_norm": 0.386277666412621,
+      "learning_rate": 4.0389597284479595e-05,
+      "loss": 1.0103,
+      "step": 2380
+    },
+    {
+      "epoch": 0.7343262902667128,
+      "grad_norm": 0.4141845583933384,
+      "learning_rate": 3.995883725303392e-05,
+      "loss": 1.0356,
+      "step": 2385
+    },
+    {
+      "epoch": 0.7358657583804795,
+      "grad_norm": 0.4038330356544352,
+      "learning_rate": 3.952981233305183e-05,
+      "loss": 1.0368,
+      "step": 2390
+    },
+    {
+      "epoch": 0.7374052264942462,
+      "grad_norm": 0.4159685288158201,
+      "learning_rate": 3.9102534922795166e-05,
+      "loss": 1.0223,
+      "step": 2395
+    },
+    {
+      "epoch": 0.7389446946080129,
+      "grad_norm": 0.44365052943948013,
+      "learning_rate": 3.867701737002502e-05,
+      "loss": 1.0125,
+      "step": 2400
+    },
+    {
+      "epoch": 0.7404841627217796,
+      "grad_norm": 0.40830415825064736,
+      "learning_rate": 3.825327197164483e-05,
+      "loss": 1.0027,
+      "step": 2405
+    },
+    {
+      "epoch": 0.7420236308355463,
+      "grad_norm": 0.6118151945433313,
+      "learning_rate": 3.7831310973345216e-05,
+      "loss": 1.014,
+      "step": 2410
+    },
+    {
+      "epoch": 0.743563098949313,
+      "grad_norm": 0.43923835412327034,
+      "learning_rate": 3.741114656924983e-05,
+      "loss": 1.0389,
+      "step": 2415
+    },
+    {
+      "epoch": 0.7451025670630798,
+      "grad_norm": 0.4961592786809209,
+      "learning_rate": 3.699279090156315e-05,
+      "loss": 1.0488,
+      "step": 2420
+    },
+    {
+      "epoch": 0.7466420351768464,
+      "grad_norm": 0.5382708745735688,
+      "learning_rate": 3.6576256060219486e-05,
+      "loss": 1.0083,
+      "step": 2425
+    },
+    {
+      "epoch": 0.7481815032906131,
+      "grad_norm": 0.44692774623072656,
+      "learning_rate": 3.616155408253367e-05,
+      "loss": 1.0106,
+      "step": 2430
+    },
+    {
+      "epoch": 0.7497209714043798,
+      "grad_norm": 0.42729189430033954,
+      "learning_rate": 3.574869695285315e-05,
+      "loss": 1.0088,
+      "step": 2435
+    },
+    {
+      "epoch": 0.7512604395181465,
+      "grad_norm": 0.4272983775590473,
+      "learning_rate": 3.5337696602211614e-05,
+      "loss": 0.9977,
+      "step": 2440
+    },
+    {
+      "epoch": 0.7527999076319132,
+      "grad_norm": 0.41113864383958393,
+      "learning_rate": 3.492856490798439e-05,
+      "loss": 1.0322,
+      "step": 2445
+    },
+    {
+      "epoch": 0.7543393757456799,
+      "grad_norm": 0.44309109594960205,
+      "learning_rate": 3.4521313693544966e-05,
+      "loss": 1.0556,
+      "step": 2450
+    },
+    {
+      "epoch": 0.7558788438594466,
+      "grad_norm": 0.4196493142920476,
+      "learning_rate": 3.4115954727923395e-05,
+      "loss": 1.0295,
+      "step": 2455
+    },
+    {
+      "epoch": 0.7574183119732133,
+      "grad_norm": 0.512441618937211,
+      "learning_rate": 3.371249972546624e-05,
+      "loss": 1.0165,
+      "step": 2460
+    },
+    {
+      "epoch": 0.7589577800869799,
+      "grad_norm": 1.9609700442905864,
+      "learning_rate": 3.3310960345497974e-05,
+      "loss": 1.0584,
+      "step": 2465
+    },
+    {
+      "epoch": 0.7604972482007466,
+      "grad_norm": 0.4558093754389953,
+      "learning_rate": 3.291134819198417e-05,
+      "loss": 1.0328,
+      "step": 2470
+    },
+    {
+      "epoch": 0.7620367163145133,
+      "grad_norm": 0.39208217735355977,
+      "learning_rate": 3.251367481319596e-05,
+      "loss": 1.039,
+      "step": 2475
+    },
+    {
+      "epoch": 0.76357618442828,
+      "grad_norm": 0.4433705457387255,
+      "learning_rate": 3.2117951701376436e-05,
+      "loss": 1.0065,
+      "step": 2480
+    },
+    {
+      "epoch": 0.7651156525420467,
+      "grad_norm": 0.3825862067647675,
+      "learning_rate": 3.172419029240853e-05,
+      "loss": 1.0403,
+      "step": 2485
+    },
+    {
+      "epoch": 0.7666551206558134,
+      "grad_norm": 0.4572258364515658,
+      "learning_rate": 3.133240196548447e-05,
+      "loss": 1.0303,
+      "step": 2490
+    },
+    {
+      "epoch": 0.7681945887695801,
+      "grad_norm": 0.84283447972151,
+      "learning_rate": 3.0942598042777073e-05,
+      "loss": 1.0669,
+      "step": 2495
+    },
+    {
+      "epoch": 0.7697340568833468,
+      "grad_norm": 0.4610127698483277,
+      "learning_rate": 3.0554789789112385e-05,
+      "loss": 0.9836,
+      "step": 2500
+    },
+    {
+      "epoch": 0.7712735249971135,
+      "grad_norm": 0.4325409044446896,
+      "learning_rate": 3.0168988411644205e-05,
+      "loss": 1.0065,
+      "step": 2505
+    },
+    {
+      "epoch": 0.7728129931108801,
+      "grad_norm": 0.45733195048796693,
+      "learning_rate": 2.9785205059530263e-05,
+      "loss": 1.0286,
+      "step": 2510
+    },
+    {
+      "epoch": 0.7743524612246468,
+      "grad_norm": 0.4692226922974537,
+      "learning_rate": 2.940345082360997e-05,
+      "loss": 1.0233,
+      "step": 2515
+    },
+    {
+      "epoch": 0.7758919293384136,
+      "grad_norm": 0.40703032879120454,
+      "learning_rate": 2.9023736736083872e-05,
+      "loss": 1.0503,
+      "step": 2520
+    },
+    {
+      "epoch": 0.7774313974521803,
+      "grad_norm": 0.603733310574897,
+      "learning_rate": 2.864607377019498e-05,
+      "loss": 1.0381,
+      "step": 2525
+    },
+    {
+      "epoch": 0.778970865565947,
+      "grad_norm": 0.44557037962518165,
+      "learning_rate": 2.82704728399115e-05,
+      "loss": 1.0297,
+      "step": 2530
+    },
+    {
+      "epoch": 0.7805103336797137,
+      "grad_norm": 0.4138521931758361,
+      "learning_rate": 2.789694479961147e-05,
+      "loss": 1.0195,
+      "step": 2535
+    },
+    {
+      "epoch": 0.7820498017934804,
+      "grad_norm": 0.494990799259698,
+      "learning_rate": 2.7525500443769136e-05,
+      "loss": 1.0257,
+      "step": 2540
+    },
+    {
+      "epoch": 0.7835892699072471,
+      "grad_norm": 0.4425172941594563,
+      "learning_rate": 2.715615050664294e-05,
+      "loss": 1.0242,
+      "step": 2545
+    },
+    {
+      "epoch": 0.7851287380210138,
+      "grad_norm": 0.3752870060929472,
+      "learning_rate": 2.6788905661965458e-05,
+      "loss": 1.0265,
+      "step": 2550
+    },
+    {
+      "epoch": 0.7866682061347804,
+      "grad_norm": 0.4116609510103765,
+      "learning_rate": 2.64237765226347e-05,
+      "loss": 1.0158,
+      "step": 2555
+    },
+    {
+      "epoch": 0.7882076742485471,
+      "grad_norm": 0.46202655193696285,
+      "learning_rate": 2.606077364040762e-05,
+      "loss": 1.0167,
+      "step": 2560
+    },
+    {
+      "epoch": 0.7897471423623138,
+      "grad_norm": 0.6202191711943372,
+      "learning_rate": 2.5699907505595068e-05,
+      "loss": 0.9672,
+      "step": 2565
+    },
+    {
+      "epoch": 0.7912866104760805,
+      "grad_norm": 0.480489604669568,
+      "learning_rate": 2.5341188546758688e-05,
+      "loss": 0.9945,
+      "step": 2570
+    },
+    {
+      "epoch": 0.7928260785898472,
+      "grad_norm": 0.4305543882648283,
+      "learning_rate": 2.4984627130409577e-05,
+      "loss": 1.0108,
+      "step": 2575
+    },
+    {
+      "epoch": 0.7943655467036139,
+      "grad_norm": 0.37712645766524694,
+      "learning_rate": 2.4630233560708615e-05,
+      "loss": 0.9991,
+      "step": 2580
+    },
+    {
+      "epoch": 0.7959050148173806,
+      "grad_norm": 0.4457882679562566,
+      "learning_rate": 2.427801807916874e-05,
+      "loss": 1.0353,
+      "step": 2585
+    },
+    {
+      "epoch": 0.7974444829311473,
+      "grad_norm": 0.43372798791794065,
+      "learning_rate": 2.3927990864358984e-05,
+      "loss": 1.016,
+      "step": 2590
+    },
+    {
+      "epoch": 0.7989839510449139,
+      "grad_norm": 0.5674499912471999,
+      "learning_rate": 2.358016203161031e-05,
+      "loss": 1.035,
+      "step": 2595
+    },
+    {
+      "epoch": 0.8005234191586806,
+      "grad_norm": 0.5654406491640428,
+      "learning_rate": 2.3234541632723272e-05,
+      "loss": 1.018,
+      "step": 2600
+    },
+    {
+      "epoch": 0.8020628872724473,
+      "grad_norm": 0.3806698162491372,
+      "learning_rate": 2.2891139655677673e-05,
+      "loss": 1.0353,
+      "step": 2605
+    },
+    {
+      "epoch": 0.803602355386214,
+      "grad_norm": 0.41395000872395227,
+      "learning_rate": 2.2549966024343682e-05,
+      "loss": 1.0198,
+      "step": 2610
+    },
+    {
+      "epoch": 0.8051418234999808,
+      "grad_norm": 1.0022245903452656,
+      "learning_rate": 2.2211030598195247e-05,
+      "loss": 1.0366,
+      "step": 2615
+    },
+    {
+      "epoch": 0.8066812916137475,
+      "grad_norm": 0.37470132686684005,
+      "learning_rate": 2.187434317202508e-05,
+      "loss": 1.0154,
+      "step": 2620
+    },
+    {
+      "epoch": 0.8082207597275142,
+      "grad_norm": 0.5250486396448188,
+      "learning_rate": 2.1539913475661576e-05,
+      "loss": 1.0053,
+      "step": 2625
+    },
+    {
+      "epoch": 0.8097602278412809,
+      "grad_norm": 0.4181640130676048,
+      "learning_rate": 2.1207751173687785e-05,
+      "loss": 1.0723,
+      "step": 2630
+    },
+    {
+      "epoch": 0.8112996959550476,
+      "grad_norm": 0.455626609677034,
+      "learning_rate": 2.0877865865161915e-05,
+      "loss": 1.0264,
+      "step": 2635
+    },
+    {
+      "epoch": 0.8128391640688142,
+      "grad_norm": 0.6557137988990834,
+      "learning_rate": 2.0550267083340068e-05,
+      "loss": 1.0192,
+      "step": 2640
+    },
+    {
+      "epoch": 0.8143786321825809,
+      "grad_norm": 0.3947380563744468,
+      "learning_rate": 2.0224964295400682e-05,
+      "loss": 1.008,
+      "step": 2645
+    },
+    {
+      "epoch": 0.8159181002963476,
+      "grad_norm": 0.4190129497380854,
+      "learning_rate": 1.9901966902170944e-05,
+      "loss": 1.0024,
+      "step": 2650
+    },
+    {
+      "epoch": 0.8174575684101143,
+      "grad_norm": 0.4733124256368494,
+      "learning_rate": 1.95812842378552e-05,
+      "loss": 1.0224,
+      "step": 2655
+    },
+    {
+      "epoch": 0.818997036523881,
+      "grad_norm": 0.4565926045173295,
+      "learning_rate": 1.9262925569765087e-05,
+      "loss": 1.0503,
+      "step": 2660
+    },
+    {
+      "epoch": 0.8205365046376477,
+      "grad_norm": 0.4253658988474481,
+      "learning_rate": 1.8946900098051778e-05,
+      "loss": 1.0738,
+      "step": 2665
+    },
+    {
+      "epoch": 0.8220759727514144,
+      "grad_norm": 0.43002381136104756,
+      "learning_rate": 1.8633216955440137e-05,
+      "loss": 1.0504,
+      "step": 2670
+    },
+    {
+      "epoch": 0.8236154408651811,
+      "grad_norm": 0.4015805803027289,
+      "learning_rate": 1.832188520696472e-05,
+      "loss": 1.0397,
+      "step": 2675
+    },
+    {
+      "epoch": 0.8251549089789477,
+      "grad_norm": 0.46899175056681264,
+      "learning_rate": 1.8012913849707868e-05,
+      "loss": 1.0154,
+      "step": 2680
+    },
+    {
+      "epoch": 0.8266943770927144,
+      "grad_norm": 0.4232662691268248,
+      "learning_rate": 1.7706311812539757e-05,
+      "loss": 1.0188,
+      "step": 2685
+    },
+    {
+      "epoch": 0.8282338452064811,
+      "grad_norm": 0.3588739754281993,
+      "learning_rate": 1.7402087955860193e-05,
+      "loss": 1.0213,
+      "step": 2690
+    },
+    {
+      "epoch": 0.8297733133202478,
+      "grad_norm": 0.4423853417849867,
+      "learning_rate": 1.710025107134272e-05,
+      "loss": 1.0223,
+      "step": 2695
+    },
+    {
+      "epoch": 0.8313127814340145,
+      "grad_norm": 0.41804031171029893,
+      "learning_rate": 1.680080988168049e-05,
+      "loss": 1.0171,
+      "step": 2700
+    },
+    {
+      "epoch": 0.8328522495477813,
+      "grad_norm": 0.42006267775596445,
+      "learning_rate": 1.6503773040334126e-05,
+      "loss": 0.9947,
+      "step": 2705
+    },
+    {
+      "epoch": 0.834391717661548,
+      "grad_norm": 0.43670340739758357,
+      "learning_rate": 1.620914913128184e-05,
+      "loss": 1.0432,
+      "step": 2710
+    },
+    {
+      "epoch": 0.8359311857753147,
+      "grad_norm": 0.4821381027661066,
+      "learning_rate": 1.591694666877114e-05,
+      "loss": 1.004,
+      "step": 2715
+    },
+    {
+      "epoch": 0.8374706538890814,
+      "grad_norm": 0.3935605580571529,
+      "learning_rate": 1.5627174097072904e-05,
+      "loss": 0.9947,
+      "step": 2720
+    },
+    {
+      "epoch": 0.839010122002848,
+      "grad_norm": 0.5080429213031332,
+      "learning_rate": 1.533983979023733e-05,
+      "loss": 1.0572,
+      "step": 2725
+    },
+    {
+      "epoch": 0.8405495901166147,
+      "grad_norm": 0.4159778200626771,
+      "learning_rate": 1.5054952051851934e-05,
+      "loss": 1.0207,
+      "step": 2730
+    },
+    {
+      "epoch": 0.8420890582303814,
+      "grad_norm": 0.4033999184219861,
+      "learning_rate": 1.477251911480162e-05,
+      "loss": 1.0353,
+      "step": 2735
+    },
+    {
+      "epoch": 0.8436285263441481,
+      "grad_norm": 0.4355088544418655,
+      "learning_rate": 1.4492549141030687e-05,
+      "loss": 1.0475,
+      "step": 2740
+    },
+    {
+      "epoch": 0.8451679944579148,
+      "grad_norm": 0.48537370762792753,
+      "learning_rate": 1.4215050221307002e-05,
+      "loss": 1.0206,
+      "step": 2745
+    },
+    {
+      "epoch": 0.8467074625716815,
+      "grad_norm": 0.37095462049211186,
+      "learning_rate": 1.394003037498821e-05,
+      "loss": 1.025,
+      "step": 2750
+    },
+    {
+      "epoch": 0.8482469306854482,
+      "grad_norm": 0.4437528225326038,
+      "learning_rate": 1.3667497549789932e-05,
+      "loss": 1.0128,
+      "step": 2755
+    },
+    {
+      "epoch": 0.8497863987992149,
+      "grad_norm": 0.4192896798172941,
+      "learning_rate": 1.339745962155613e-05,
+      "loss": 1.0333,
+      "step": 2760
+    },
+    {
+      "epoch": 0.8513258669129815,
+      "grad_norm": 0.4323329881502678,
+      "learning_rate": 1.3129924394031535e-05,
+      "loss": 0.999,
+      "step": 2765
+    },
+    {
+      "epoch": 0.8528653350267482,
+      "grad_norm": 0.5312159218939685,
+      "learning_rate": 1.2864899598636004e-05,
+      "loss": 1.019,
+      "step": 2770
+    },
+    {
+      "epoch": 0.8544048031405149,
+      "grad_norm": 0.37079351875448874,
+      "learning_rate": 1.2602392894241222e-05,
+      "loss": 1.0349,
+      "step": 2775
+    },
+    {
+      "epoch": 0.8559442712542816,
+      "grad_norm": 0.4245080748700514,
+      "learning_rate": 1.234241186694931e-05,
+      "loss": 1.0131,
+      "step": 2780
+    },
+    {
+      "epoch": 0.8574837393680483,
+      "grad_norm": 0.4340531910995576,
+      "learning_rate": 1.208496402987358e-05,
+      "loss": 1.0245,
+      "step": 2785
+    },
+    {
+      "epoch": 0.859023207481815,
+      "grad_norm": 0.37626279842019295,
+      "learning_rate": 1.1830056822921521e-05,
+      "loss": 1.0043,
+      "step": 2790
+    },
+    {
+      "epoch": 0.8605626755955817,
+      "grad_norm": 0.5175737600246689,
+      "learning_rate": 1.1577697612579641e-05,
+      "loss": 1.0232,
+      "step": 2795
+    },
+    {
+      "epoch": 0.8621021437093485,
+      "grad_norm": 0.44974494562352096,
+      "learning_rate": 1.1327893691700698e-05,
+      "loss": 0.9889,
+      "step": 2800
+    },
+    {
+      "epoch": 0.8636416118231152,
+      "grad_norm": 0.430436147584834,
+      "learning_rate": 1.1080652279292891e-05,
+      "loss": 1.0236,
+      "step": 2805
+    },
+    {
+      "epoch": 0.8651810799368818,
+      "grad_norm": 0.4728221888964366,
+      "learning_rate": 1.0835980520311251e-05,
+      "loss": 1.0155,
+      "step": 2810
+    },
+    {
+      "epoch": 0.8667205480506485,
+      "grad_norm": 0.397220073379638,
+      "learning_rate": 1.0593885485451237e-05,
+      "loss": 1.0099,
+      "step": 2815
+    },
+    {
+      "epoch": 0.8682600161644152,
+      "grad_norm": 0.5348645223247931,
+      "learning_rate": 1.0354374170944258e-05,
+      "loss": 1.0416,
+      "step": 2820
+    },
+    {
+      "epoch": 0.8697994842781819,
+      "grad_norm": 0.4043554078765482,
+      "learning_rate": 1.011745349835559e-05,
+      "loss": 1.0121,
+      "step": 2825
+    },
+    {
+      "epoch": 0.8713389523919486,
+      "grad_norm": 0.468593340879125,
+      "learning_rate": 9.883130314384348e-06,
+      "loss": 1.0231,
+      "step": 2830
+    },
+    {
+      "epoch": 0.8728784205057153,
+      "grad_norm": 0.46465265214459056,
+      "learning_rate": 9.651411390665577e-06,
+      "loss": 0.9918,
+      "step": 2835
+    },
+    {
+      "epoch": 0.874417888619482,
+      "grad_norm": 0.5605586645440735,
+      "learning_rate": 9.422303423574596e-06,
+      "loss": 1.0176,
+      "step": 2840
+    },
+    {
+      "epoch": 0.8759573567332487,
+      "grad_norm": 1.775692528658347,
+      "learning_rate": 9.195813034033508e-06,
+      "loss": 1.0134,
+      "step": 2845
+    },
+    {
+      "epoch": 0.8774968248470154,
+      "grad_norm": 0.44922271095321187,
+      "learning_rate": 8.971946767319805e-06,
+      "loss": 0.9981,
+      "step": 2850
+    },
+    {
+      "epoch": 0.879036292960782,
+      "grad_norm": 0.5098548100487027,
+      "learning_rate": 8.75071109287724e-06,
+      "loss": 1.0052,
+      "step": 2855
+    },
+    {
+      "epoch": 0.8805757610745487,
+      "grad_norm": 0.41578211906898727,
+      "learning_rate": 8.532112404128877e-06,
+      "loss": 1.0128,
+      "step": 2860
+    },
+    {
+      "epoch": 0.8821152291883154,
+      "grad_norm": 0.3496924486384235,
+      "learning_rate": 8.316157018292326e-06,
+      "loss": 1.0479,
+      "step": 2865
+    },
+    {
+      "epoch": 0.8836546973020821,
+      "grad_norm": 0.3816021309698796,
+      "learning_rate": 8.102851176197201e-06,
+      "loss": 1.0103,
+      "step": 2870
+    },
+    {
+      "epoch": 0.8851941654158488,
+      "grad_norm": 0.3838865693994203,
+      "learning_rate": 7.892201042104718e-06,
+      "loss": 1.0326,
+      "step": 2875
+    },
+    {
+      "epoch": 0.8867336335296155,
+      "grad_norm": 0.4294597636569908,
+      "learning_rate": 7.684212703529624e-06,
+      "loss": 1.0323,
+      "step": 2880
+    },
+    {
+      "epoch": 0.8882731016433822,
+      "grad_norm": 0.43018290128127384,
+      "learning_rate": 7.4788921710642e-06,
+      "loss": 1.0188,
+      "step": 2885
+    },
+    {
+      "epoch": 0.889812569757149,
+      "grad_norm": 0.6183016933222353,
+      "learning_rate": 7.276245378204616e-06,
+      "loss": 1.0259,
+      "step": 2890
+    },
+    {
+      "epoch": 0.8913520378709155,
+      "grad_norm": 0.46596709036144063,
+      "learning_rate": 7.076278181179485e-06,
+      "loss": 1.0316,
+      "step": 2895
+    },
+    {
+      "epoch": 0.8928915059846823,
+      "grad_norm": 0.45497909906909056,
+      "learning_rate": 6.878996358780532e-06,
+      "loss": 1.0326,
+      "step": 2900
+    },
+    {
+      "epoch": 0.894430974098449,
+      "grad_norm": 0.4605589314545624,
+      "learning_rate": 6.684405612195688e-06,
+      "loss": 1.0184,
+      "step": 2905
+    },
+    {
+      "epoch": 0.8959704422122157,
+      "grad_norm": 0.35212400219047424,
+      "learning_rate": 6.492511564844273e-06,
+      "loss": 0.9976,
+      "step": 2910
+    },
+    {
+      "epoch": 0.8975099103259824,
+      "grad_norm": 0.431075612216501,
+      "learning_rate": 6.303319762214499e-06,
+      "loss": 1.0282,
+      "step": 2915
+    },
+    {
+      "epoch": 0.8990493784397491,
+      "grad_norm": 0.49886462139074744,
+      "learning_rate": 6.11683567170328e-06,
+      "loss": 1.0453,
+      "step": 2920
+    },
+    {
+      "epoch": 0.9005888465535158,
+      "grad_norm": 0.5001791356037993,
+      "learning_rate": 5.933064682458122e-06,
+      "loss": 0.9974,
+      "step": 2925
+    },
+    {
+      "epoch": 0.9021283146672825,
+      "grad_norm": 0.4334035137279178,
+      "learning_rate": 5.7520121052214275e-06,
+      "loss": 1.002,
+      "step": 2930
+    },
+    {
+      "epoch": 0.9036677827810492,
+      "grad_norm": 0.45835892877048534,
+      "learning_rate": 5.57368317217708e-06,
+      "loss": 1.0524,
+      "step": 2935
+    },
+    {
+      "epoch": 0.9052072508948158,
+      "grad_norm": 0.3752767008874508,
+      "learning_rate": 5.398083036799129e-06,
+      "loss": 0.9691,
+      "step": 2940
+    },
+    {
+      "epoch": 0.9067467190085825,
+      "grad_norm": 0.4935328531459081,
+      "learning_rate": 5.225216773702968e-06,
+      "loss": 1.0252,
+      "step": 2945
+    },
+    {
+      "epoch": 0.9082861871223492,
+      "grad_norm": 0.5210304589005964,
+      "learning_rate": 5.055089378498634e-06,
+      "loss": 1.0199,
+      "step": 2950
+    },
+    {
+      "epoch": 0.9098256552361159,
+      "grad_norm": 0.4254020055130796,
+      "learning_rate": 4.887705767646434e-06,
+      "loss": 0.9989,
+      "step": 2955
+    },
+    {
+      "epoch": 0.9113651233498826,
+      "grad_norm": 0.4643941731342843,
+      "learning_rate": 4.7230707783148864e-06,
+      "loss": 1.0074,
+      "step": 2960
+    },
+    {
+      "epoch": 0.9129045914636493,
+      "grad_norm": 0.3715160286597917,
+      "learning_rate": 4.561189168240909e-06,
+      "loss": 1.0682,
+      "step": 2965
+    },
+    {
+      "epoch": 0.914444059577416,
+      "grad_norm": 0.36108361076071477,
+      "learning_rate": 4.402065615592344e-06,
+      "loss": 0.9939,
+      "step": 2970
+    },
+    {
+      "epoch": 0.9159835276911827,
+      "grad_norm": 0.460057449616597,
+      "learning_rate": 4.245704718832811e-06,
+      "loss": 1.029,
+      "step": 2975
+    },
+    {
+      "epoch": 0.9175229958049493,
+      "grad_norm": 0.42151771172138136,
+      "learning_rate": 4.092110996588705e-06,
+      "loss": 1.0416,
+      "step": 2980
+    },
+    {
+      "epoch": 0.919062463918716,
+      "grad_norm": 0.4208982471816418,
+      "learning_rate": 3.941288887518713e-06,
+      "loss": 1.0083,
+      "step": 2985
+    },
+    {
+      "epoch": 0.9206019320324828,
+      "grad_norm": 0.3754941722458392,
+      "learning_rate": 3.7932427501854996e-06,
+      "loss": 1.0317,
+      "step": 2990
+    },
+    {
+      "epoch": 0.9221414001462495,
+      "grad_norm": 0.5370873955203277,
+      "learning_rate": 3.647976862929747e-06,
+      "loss": 1.054,
+      "step": 2995
+    },
+    {
+      "epoch": 0.9236808682600162,
+      "grad_norm": 0.4426665180986752,
+      "learning_rate": 3.505495423746574e-06,
+      "loss": 1.0287,
+      "step": 3000
+    },
+    {
+      "epoch": 0.9252203363737829,
+      "grad_norm": 0.39355763646149944,
+      "learning_rate": 3.3658025501641323e-06,
+      "loss": 1.0538,
+      "step": 3005
+    },
+    {
+      "epoch": 0.9267598044875496,
+      "grad_norm": 0.3926350659174442,
+      "learning_rate": 3.228902279124657e-06,
+      "loss": 1.0269,
+      "step": 3010
+    },
+    {
+      "epoch": 0.9282992726013163,
+      "grad_norm": 0.5307688367772885,
+      "learning_rate": 3.094798566867818e-06,
+      "loss": 1.0098,
+      "step": 3015
+    },
+    {
+      "epoch": 0.929838740715083,
+      "grad_norm": 0.5463010628237303,
+      "learning_rate": 2.963495288816376e-06,
+      "loss": 1.0698,
+      "step": 3020
+    },
+    {
+      "epoch": 0.9313782088288496,
+      "grad_norm": 0.4767217195493804,
+      "learning_rate": 2.8349962394641605e-06,
+      "loss": 1.0652,
+      "step": 3025
+    },
+    {
+      "epoch": 0.9329176769426163,
+      "grad_norm": 0.46383675807120084,
+      "learning_rate": 2.709305132266493e-06,
+      "loss": 1.0566,
+      "step": 3030
+    },
+    {
+      "epoch": 0.934457145056383,
+      "grad_norm": 0.406764613161873,
+      "learning_rate": 2.5864255995327936e-06,
+      "loss": 1.02,
+      "step": 3035
+    },
+    {
+      "epoch": 0.9359966131701497,
+      "grad_norm": 0.5667072139952383,
+      "learning_rate": 2.46636119232162e-06,
+      "loss": 1.0293,
+      "step": 3040
+    },
+    {
+      "epoch": 0.9375360812839164,
+      "grad_norm": 0.4123820408065487,
+      "learning_rate": 2.349115380338096e-06,
+      "loss": 1.0528,
+      "step": 3045
+    },
+    {
+      "epoch": 0.9390755493976831,
+      "grad_norm": 0.42212082973648146,
+      "learning_rate": 2.2346915518335786e-06,
+      "loss": 0.9751,
+      "step": 3050
+    },
+    {
+      "epoch": 0.9406150175114498,
+      "grad_norm": 0.44633003035441593,
+      "learning_rate": 2.1230930135078373e-06,
+      "loss": 1.0071,
+      "step": 3055
+    },
+    {
+      "epoch": 0.9421544856252165,
+      "grad_norm": 0.6009896163488582,
+      "learning_rate": 2.014322990413353e-06,
+      "loss": 1.0275,
+      "step": 3060
+    },
+    {
+      "epoch": 0.9436939537389831,
+      "grad_norm": 0.3529405684211488,
+      "learning_rate": 1.9083846258622586e-06,
+      "loss": 1.0176,
+      "step": 3065
+    },
+    {
+      "epoch": 0.9452334218527498,
+      "grad_norm": 0.7896994457794821,
+      "learning_rate": 1.8052809813354111e-06,
+      "loss": 1.0063,
+      "step": 3070
+    },
+    {
+      "epoch": 0.9467728899665165,
+      "grad_norm": 0.48377625576699135,
+      "learning_rate": 1.705015036393931e-06,
+      "loss": 1.057,
+      "step": 3075
+    },
+    {
+      "epoch": 0.9483123580802832,
+      "grad_norm": 0.43904259295920206,
+      "learning_rate": 1.6075896885931807e-06,
+      "loss": 1.0227,
+      "step": 3080
+    },
+    {
+      "epoch": 0.94985182619405,
+      "grad_norm": 0.4480774273472164,
+      "learning_rate": 1.5130077533988873e-06,
+      "loss": 1.0369,
+      "step": 3085
+    },
+    {
+      "epoch": 0.9513912943078167,
+      "grad_norm": 0.4679465739342283,
+      "learning_rate": 1.421271964105908e-06,
+      "loss": 1.0623,
+      "step": 3090
+    },
+    {
+      "epoch": 0.9529307624215834,
+      "grad_norm": 0.44630980531648395,
+      "learning_rate": 1.3323849717591376e-06,
+      "loss": 1.0166,
+      "step": 3095
+    },
+    {
+      "epoch": 0.9544702305353501,
+      "grad_norm": 0.5287384335577812,
+      "learning_rate": 1.2463493450769915e-06,
+      "loss": 1.0425,
+      "step": 3100
+    },
+    {
+      "epoch": 0.9560096986491168,
+      "grad_norm": 0.40230009571762976,
+      "learning_rate": 1.1631675703771105e-06,
+      "loss": 1.0385,
+      "step": 3105
+    },
+    {
+      "epoch": 0.9575491667628834,
+      "grad_norm": 0.49211013441821094,
+      "learning_rate": 1.0828420515045178e-06,
+      "loss": 1.0234,
+      "step": 3110
+    },
+    {
+      "epoch": 0.9590886348766501,
+      "grad_norm": 0.48292762219822727,
+      "learning_rate": 1.0053751097621856e-06,
+      "loss": 0.9869,
+      "step": 3115
+    },
+    {
+      "epoch": 0.9606281029904168,
+      "grad_norm": 0.4431002985495294,
+      "learning_rate": 9.307689838439104e-07,
+      "loss": 1.0363,
+      "step": 3120
+    },
+    {
+      "epoch": 0.9621675711041835,
+      "grad_norm": 0.3771405426388914,
+      "learning_rate": 8.590258297696108e-07,
+      "loss": 0.978,
+      "step": 3125
+    },
+    {
+      "epoch": 0.9637070392179502,
+      "grad_norm": 0.4242241866712476,
+      "learning_rate": 7.901477208230979e-07,
+      "loss": 1.0173,
+      "step": 3130
+    },
+    {
+      "epoch": 0.9652465073317169,
+      "grad_norm": 0.42765029098432733,
+      "learning_rate": 7.241366474920797e-07,
+      "loss": 1.0021,
+      "step": 3135
+    },
+    {
+      "epoch": 0.9667859754454836,
+      "grad_norm": 0.4026446330368183,
+      "learning_rate": 6.609945174106402e-07,
+      "loss": 1.0352,
+      "step": 3140
+    },
+    {
+      "epoch": 0.9683254435592503,
+      "grad_norm": 0.4586738339921203,
+      "learning_rate": 6.007231553041837e-07,
+      "loss": 1.0087,
+      "step": 3145
+    },
+    {
+      "epoch": 0.969864911673017,
+      "grad_norm": 0.41482262786645374,
+      "learning_rate": 5.43324302936643e-07,
+      "loss": 1.0914,
+      "step": 3150
+    },
+    {
+      "epoch": 0.9714043797867836,
+      "grad_norm": 0.42202537226167186,
+      "learning_rate": 4.887996190601318e-07,
+      "loss": 1.0419,
+      "step": 3155
+    },
+    {
+      "epoch": 0.9729438479005503,
+      "grad_norm": 0.37960996848907147,
+      "learning_rate": 4.3715067936705987e-07,
+      "loss": 1.0171,
+      "step": 3160
+    },
+    {
+      "epoch": 0.974483316014317,
+      "grad_norm": 0.7530238458576849,
+      "learning_rate": 3.8837897644457e-07,
+      "loss": 1.0446,
+      "step": 3165
+    },
+    {
+      "epoch": 0.9760227841280837,
+      "grad_norm": 0.3574506914494524,
+      "learning_rate": 3.4248591973140566e-07,
+      "loss": 1.0402,
+      "step": 3170
+    },
+    {
+      "epoch": 0.9775622522418504,
+      "grad_norm": 0.4403626626769491,
+      "learning_rate": 2.994728354771659e-07,
+      "loss": 1.0618,
+      "step": 3175
+    },
+    {
+      "epoch": 0.9791017203556172,
+      "grad_norm": 0.43643836321504664,
+      "learning_rate": 2.593409667040247e-07,
+      "loss": 0.9829,
+      "step": 3180
+    },
+    {
+      "epoch": 0.9806411884693839,
+      "grad_norm": 0.44879821672356546,
+      "learning_rate": 2.2209147317074908e-07,
+      "loss": 0.9989,
+      "step": 3185
+    },
+    {
+      "epoch": 0.9821806565831506,
+      "grad_norm": 0.464478069097993,
+      "learning_rate": 1.8772543133922515e-07,
+      "loss": 1.0269,
+      "step": 3190
+    },
+    {
+      "epoch": 0.9837201246969172,
+      "grad_norm": 0.3840629750152777,
+      "learning_rate": 1.5624383434333923e-07,
+      "loss": 1.0007,
+      "step": 3195
+    },
+    {
+      "epoch": 0.9852595928106839,
+      "grad_norm": 0.4819476738232411,
+      "learning_rate": 1.276475919602671e-07,
+      "loss": 1.0183,
+      "step": 3200
+    },
+    {
+      "epoch": 0.9867990609244506,
+      "grad_norm": 0.4679445982996664,
+      "learning_rate": 1.019375305842063e-07,
+      "loss": 1.0093,
+      "step": 3205
+    },
+    {
+      "epoch": 0.9883385290382173,
+      "grad_norm": 0.608255287884667,
+      "learning_rate": 7.911439320247294e-08,
+      "loss": 1.0133,
+      "step": 3210
+    },
+    {
+      "epoch": 0.989877997151984,
+      "grad_norm": 0.49744148279623623,
+      "learning_rate": 5.9178839374018914e-08,
+      "loss": 1.0323,
+      "step": 3215
+    },
+    {
+      "epoch": 0.9914174652657507,
+      "grad_norm": 0.4874179505714964,
+      "learning_rate": 4.213144521042489e-08,
+      "loss": 1.0355,
+      "step": 3220
+    },
+    {
+      "epoch": 0.9929569333795174,
+      "grad_norm": 0.42907270925432495,
+      "learning_rate": 2.797270335916924e-08,
+      "loss": 1.0004,
+      "step": 3225
+    },
+    {
+      "epoch": 0.9944964014932841,
+      "grad_norm": 0.36896334930948343,
+      "learning_rate": 1.6703022989494887e-08,
+      "loss": 1.0378,
+      "step": 3230
+    },
+    {
+      "epoch": 0.9960358696070508,
+      "grad_norm": 0.369776577000758,
+      "learning_rate": 8.322729780474436e-09,
+      "loss": 1.0181,
+      "step": 3235
+    },
+    {
+      "epoch": 0.9975753377208174,
+      "grad_norm": 0.381178979843484,
+      "learning_rate": 2.8320659116953806e-09,
+      "loss": 1.016,
+      "step": 3240
+    },
+    {
+      "epoch": 0.9991148058345841,
+      "grad_norm": 1.063968939156568,
+      "learning_rate": 2.311900561768887e-10,
+      "loss": 1.008,
+      "step": 3245
+    },
+    {
+      "epoch": 0.9997305930800908,
+      "eval_loss": 1.0235061645507812,
+      "eval_runtime": 3775.83,
+      "eval_samples_per_second": 6.12,
+      "eval_steps_per_second": 0.383,
+      "step": 3247
+    },
+    {
+      "epoch": 0.9997305930800908,
+      "step": 3247,
+      "total_flos": 4.15615026158633e+16,
+      "train_loss": 1.0426253444200522,
+      "train_runtime": 139956.5138,
+      "train_samples_per_second": 1.485,
+      "train_steps_per_second": 0.023
+    }
+  ],
+  "logging_steps": 5,
+  "max_steps": 3247,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "save_steps": 100,
+  "total_flos": 4.15615026158633e+16,
+  "train_batch_size": 2,
+  "trial_name": null,
+  "trial_params": null
+}