diff --git "a/checkpoint-18312/trainer_state.json" "b/checkpoint-18312/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/checkpoint-18312/trainer_state.json"
@@ -0,0 +1,4163 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.45876340314660785,
+  "eval_steps": 500,
+  "global_step": 18312,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.000776630924942379,
+      "grad_norm": 23.55321502685547,
+      "learning_rate": 1.0157273918741808e-06,
+      "loss": 8.3443,
+      "step": 31
+    },
+    {
+      "epoch": 0.001553261849884758,
+      "grad_norm": 14.414477348327637,
+      "learning_rate": 2.0314547837483616e-06,
+      "loss": 7.198,
+      "step": 62
+    },
+    {
+      "epoch": 0.002329892774827137,
+      "grad_norm": 10.739773750305176,
+      "learning_rate": 3.0471821756225426e-06,
+      "loss": 5.8639,
+      "step": 93
+    },
+    {
+      "epoch": 0.003106523699769516,
+      "grad_norm": 5.378795623779297,
+      "learning_rate": 4.062909567496723e-06,
+      "loss": 4.8222,
+      "step": 124
+    },
+    {
+      "epoch": 0.003883154624711895,
+      "grad_norm": 8.51831340789795,
+      "learning_rate": 5.078636959370905e-06,
+      "loss": 4.1983,
+      "step": 155
+    },
+    {
+      "epoch": 0.004659785549654274,
+      "grad_norm": 6.32632303237915,
+      "learning_rate": 6.094364351245085e-06,
+      "loss": 3.8598,
+      "step": 186
+    },
+    {
+      "epoch": 0.005436416474596653,
+      "grad_norm": 5.590794563293457,
+      "learning_rate": 7.110091743119267e-06,
+      "loss": 3.6646,
+      "step": 217
+    },
+    {
+      "epoch": 0.006213047399539032,
+      "grad_norm": 5.9275383949279785,
+      "learning_rate": 8.125819134993446e-06,
+      "loss": 3.5212,
+      "step": 248
+    },
+    {
+      "epoch": 0.006989678324481411,
+      "grad_norm": 5.041291236877441,
+      "learning_rate": 9.141546526867629e-06,
+      "loss": 3.3936,
+      "step": 279
+    },
+    {
+      "epoch": 0.00776630924942379,
+      "grad_norm": 4.923459529876709,
+      "learning_rate": 1.015727391874181e-05,
+      "loss": 3.2713,
+      "step": 310
+    },
+    {
+      "epoch": 0.00854294017436617,
+      "grad_norm": 8.011459350585938,
+      "learning_rate": 1.117300131061599e-05,
+      "loss": 3.2,
+      "step": 341
+    },
+    {
+      "epoch": 0.009319571099308548,
+      "grad_norm": 5.261505603790283,
+      "learning_rate": 1.218872870249017e-05,
+      "loss": 3.1255,
+      "step": 372
+    },
+    {
+      "epoch": 0.010096202024250926,
+      "grad_norm": 5.004406452178955,
+      "learning_rate": 1.3204456094364351e-05,
+      "loss": 3.0726,
+      "step": 403
+    },
+    {
+      "epoch": 0.010872832949193306,
+      "grad_norm": 4.326263427734375,
+      "learning_rate": 1.4220183486238533e-05,
+      "loss": 2.9682,
+      "step": 434
+    },
+    {
+      "epoch": 0.011649463874135685,
+      "grad_norm": 4.251937389373779,
+      "learning_rate": 1.5235910878112714e-05,
+      "loss": 2.9215,
+      "step": 465
+    },
+    {
+      "epoch": 0.012426094799078065,
+      "grad_norm": 4.35053825378418,
+      "learning_rate": 1.6251638269986893e-05,
+      "loss": 2.8902,
+      "step": 496
+    },
+    {
+      "epoch": 0.013202725724020443,
+      "grad_norm": 4.316098690032959,
+      "learning_rate": 1.7267365661861077e-05,
+      "loss": 2.8521,
+      "step": 527
+    },
+    {
+      "epoch": 0.013979356648962823,
+      "grad_norm": 5.536881446838379,
+      "learning_rate": 1.8283093053735257e-05,
+      "loss": 2.8058,
+      "step": 558
+    },
+    {
+      "epoch": 0.014755987573905201,
+      "grad_norm": 3.759843111038208,
+      "learning_rate": 1.9298820445609438e-05,
+      "loss": 2.7438,
+      "step": 589
+    },
+    {
+      "epoch": 0.01553261849884758,
+      "grad_norm": 4.057703018188477,
+      "learning_rate": 2.031454783748362e-05,
+      "loss": 2.7508,
+      "step": 620
+    },
+    {
+      "epoch": 0.01630924942378996,
+      "grad_norm": 3.427708625793457,
+      "learning_rate": 2.13302752293578e-05,
+      "loss": 2.7046,
+      "step": 651
+    },
+    {
+      "epoch": 0.01708588034873234,
+      "grad_norm": 3.9012484550476074,
+      "learning_rate": 2.234600262123198e-05,
+      "loss": 2.6706,
+      "step": 682
+    },
+    {
+      "epoch": 0.017862511273674716,
+      "grad_norm": 3.388504981994629,
+      "learning_rate": 2.336173001310616e-05,
+      "loss": 2.6377,
+      "step": 713
+    },
+    {
+      "epoch": 0.018639142198617096,
+      "grad_norm": 3.2011327743530273,
+      "learning_rate": 2.437745740498034e-05,
+      "loss": 2.6214,
+      "step": 744
+    },
+    {
+      "epoch": 0.019415773123559476,
+      "grad_norm": 3.5897209644317627,
+      "learning_rate": 2.5393184796854525e-05,
+      "loss": 2.6045,
+      "step": 775
+    },
+    {
+      "epoch": 0.020192404048501852,
+      "grad_norm": 3.386697769165039,
+      "learning_rate": 2.6408912188728702e-05,
+      "loss": 2.6066,
+      "step": 806
+    },
+    {
+      "epoch": 0.020969034973444232,
+      "grad_norm": 2.9135866165161133,
+      "learning_rate": 2.7424639580602886e-05,
+      "loss": 2.5697,
+      "step": 837
+    },
+    {
+      "epoch": 0.021745665898386612,
+      "grad_norm": 9.212065696716309,
+      "learning_rate": 2.8440366972477066e-05,
+      "loss": 2.5773,
+      "step": 868
+    },
+    {
+      "epoch": 0.022522296823328992,
+      "grad_norm": 3.0339765548706055,
+      "learning_rate": 2.9456094364351244e-05,
+      "loss": 2.5453,
+      "step": 899
+    },
+    {
+      "epoch": 0.02329892774827137,
+      "grad_norm": 2.8007607460021973,
+      "learning_rate": 3.0471821756225428e-05,
+      "loss": 2.5297,
+      "step": 930
+    },
+    {
+      "epoch": 0.02407555867321375,
+      "grad_norm": 3.251768112182617,
+      "learning_rate": 3.148754914809961e-05,
+      "loss": 2.5273,
+      "step": 961
+    },
+    {
+      "epoch": 0.02485218959815613,
+      "grad_norm": 3.357011079788208,
+      "learning_rate": 3.2503276539973785e-05,
+      "loss": 2.4896,
+      "step": 992
+    },
+    {
+      "epoch": 0.025628820523098506,
+      "grad_norm": 2.6069107055664062,
+      "learning_rate": 3.351900393184797e-05,
+      "loss": 2.4718,
+      "step": 1023
+    },
+    {
+      "epoch": 0.026405451448040886,
+      "grad_norm": 4.663620948791504,
+      "learning_rate": 3.453473132372215e-05,
+      "loss": 2.4338,
+      "step": 1054
+    },
+    {
+      "epoch": 0.027182082372983266,
+      "grad_norm": 3.819638252258301,
+      "learning_rate": 3.555045871559633e-05,
+      "loss": 2.4848,
+      "step": 1085
+    },
+    {
+      "epoch": 0.027958713297925646,
+      "grad_norm": 2.6367456912994385,
+      "learning_rate": 3.6566186107470514e-05,
+      "loss": 2.4415,
+      "step": 1116
+    },
+    {
+      "epoch": 0.028735344222868022,
+      "grad_norm": 2.6118173599243164,
+      "learning_rate": 3.7581913499344695e-05,
+      "loss": 2.4341,
+      "step": 1147
+    },
+    {
+      "epoch": 0.029511975147810402,
+      "grad_norm": 9.820366859436035,
+      "learning_rate": 3.8597640891218876e-05,
+      "loss": 2.3773,
+      "step": 1178
+    },
+    {
+      "epoch": 0.030288606072752782,
+      "grad_norm": 38.890506744384766,
+      "learning_rate": 3.9613368283093056e-05,
+      "loss": 2.3966,
+      "step": 1209
+    },
+    {
+      "epoch": 0.03106523699769516,
+      "grad_norm": 2.5037872791290283,
+      "learning_rate": 4.062909567496724e-05,
+      "loss": 2.3421,
+      "step": 1240
+    },
+    {
+      "epoch": 0.03184186792263754,
+      "grad_norm": 2.426544189453125,
+      "learning_rate": 4.164482306684142e-05,
+      "loss": 2.3568,
+      "step": 1271
+    },
+    {
+      "epoch": 0.03261849884757992,
+      "grad_norm": 2.5375614166259766,
+      "learning_rate": 4.26605504587156e-05,
+      "loss": 2.3314,
+      "step": 1302
+    },
+    {
+      "epoch": 0.0333951297725223,
+      "grad_norm": 3.555330276489258,
+      "learning_rate": 4.367627785058978e-05,
+      "loss": 2.3262,
+      "step": 1333
+    },
+    {
+      "epoch": 0.03417176069746468,
+      "grad_norm": 2.2571263313293457,
+      "learning_rate": 4.469200524246396e-05,
+      "loss": 2.3167,
+      "step": 1364
+    },
+    {
+      "epoch": 0.03494839162240705,
+      "grad_norm": 2.184305429458618,
+      "learning_rate": 4.570773263433814e-05,
+      "loss": 2.2936,
+      "step": 1395
+    },
+    {
+      "epoch": 0.03572502254734943,
+      "grad_norm": 2.8295061588287354,
+      "learning_rate": 4.672346002621232e-05,
+      "loss": 2.3033,
+      "step": 1426
+    },
+    {
+      "epoch": 0.03650165347229181,
+      "grad_norm": 2.0241498947143555,
+      "learning_rate": 4.77391874180865e-05,
+      "loss": 2.2833,
+      "step": 1457
+    },
+    {
+      "epoch": 0.03727828439723419,
+      "grad_norm": 2.0029919147491455,
+      "learning_rate": 4.875491480996068e-05,
+      "loss": 2.2663,
+      "step": 1488
+    },
+    {
+      "epoch": 0.03805491532217657,
+      "grad_norm": 2.033773899078369,
+      "learning_rate": 4.977064220183487e-05,
+      "loss": 2.2597,
+      "step": 1519
+    },
+    {
+      "epoch": 0.03883154624711895,
+      "grad_norm": 2.047210216522217,
+      "learning_rate": 4.9999915451558777e-05,
+      "loss": 2.2231,
+      "step": 1550
+    },
+    {
+      "epoch": 0.03960817717206133,
+      "grad_norm": 2.032099485397339,
+      "learning_rate": 4.999955597496219e-05,
+      "loss": 2.2411,
+      "step": 1581
+    },
+    {
+      "epoch": 0.040384808097003705,
+      "grad_norm": 1.8384301662445068,
+      "learning_rate": 4.9998914381774255e-05,
+      "loss": 2.2409,
+      "step": 1612
+    },
+    {
+      "epoch": 0.041161439021946085,
+      "grad_norm": 1.721187710762024,
+      "learning_rate": 4.999799067923527e-05,
+      "loss": 2.193,
+      "step": 1643
+    },
+    {
+      "epoch": 0.041938069946888465,
+      "grad_norm": 1.8869743347167969,
+      "learning_rate": 4.999678487776908e-05,
+      "loss": 2.1917,
+      "step": 1674
+    },
+    {
+      "epoch": 0.042714700871830845,
+      "grad_norm": 1.796583652496338,
+      "learning_rate": 4.9995296990983006e-05,
+      "loss": 2.1794,
+      "step": 1705
+    },
+    {
+      "epoch": 0.043491331796773225,
+      "grad_norm": 1.6308804750442505,
+      "learning_rate": 4.999352703566763e-05,
+      "loss": 2.1854,
+      "step": 1736
+    },
+    {
+      "epoch": 0.044267962721715605,
+      "grad_norm": 1.683474063873291,
+      "learning_rate": 4.999147503179668e-05,
+      "loss": 2.1797,
+      "step": 1767
+    },
+    {
+      "epoch": 0.045044593646657985,
+      "grad_norm": 1.7442835569381714,
+      "learning_rate": 4.998914100252672e-05,
+      "loss": 2.1695,
+      "step": 1798
+    },
+    {
+      "epoch": 0.04582122457160036,
+      "grad_norm": 1.704970359802246,
+      "learning_rate": 4.998652497419696e-05,
+      "loss": 2.1365,
+      "step": 1829
+    },
+    {
+      "epoch": 0.04659785549654274,
+      "grad_norm": 1.7213023900985718,
+      "learning_rate": 4.9983626976328927e-05,
+      "loss": 2.1524,
+      "step": 1860
+    },
+    {
+      "epoch": 0.04737448642148512,
+      "grad_norm": 4.01851224899292,
+      "learning_rate": 4.998044704162613e-05,
+      "loss": 2.1171,
+      "step": 1891
+    },
+    {
+      "epoch": 0.0481511173464275,
+      "grad_norm": 1.5619858503341675,
+      "learning_rate": 4.9976985205973705e-05,
+      "loss": 2.1608,
+      "step": 1922
+    },
+    {
+      "epoch": 0.04892774827136988,
+      "grad_norm": 1.684493064880371,
+      "learning_rate": 4.997324150843799e-05,
+      "loss": 2.1278,
+      "step": 1953
+    },
+    {
+      "epoch": 0.04970437919631226,
+      "grad_norm": 1.660992980003357,
+      "learning_rate": 4.99692159912661e-05,
+      "loss": 2.1136,
+      "step": 1984
+    },
+    {
+      "epoch": 0.05048101012125464,
+      "grad_norm": 1.5491065979003906,
+      "learning_rate": 4.996490869988546e-05,
+      "loss": 2.1086,
+      "step": 2015
+    },
+    {
+      "epoch": 0.05125764104619701,
+      "grad_norm": 1.5400840044021606,
+      "learning_rate": 4.996031968290326e-05,
+      "loss": 2.1069,
+      "step": 2046
+    },
+    {
+      "epoch": 0.05203427197113939,
+      "grad_norm": 1.5687416791915894,
+      "learning_rate": 4.995544899210594e-05,
+      "loss": 2.0888,
+      "step": 2077
+    },
+    {
+      "epoch": 0.05281090289608177,
+      "grad_norm": 1.5218935012817383,
+      "learning_rate": 4.9950296682458583e-05,
+      "loss": 2.1032,
+      "step": 2108
+    },
+    {
+      "epoch": 0.05358753382102415,
+      "grad_norm": 1.763666033744812,
+      "learning_rate": 4.994486281210429e-05,
+      "loss": 2.0799,
+      "step": 2139
+    },
+    {
+      "epoch": 0.05436416474596653,
+      "grad_norm": 1.5606274604797363,
+      "learning_rate": 4.9939147442363566e-05,
+      "loss": 2.0614,
+      "step": 2170
+    },
+    {
+      "epoch": 0.05514079567090891,
+      "grad_norm": 1.4900213479995728,
+      "learning_rate": 4.9933150637733574e-05,
+      "loss": 2.0565,
+      "step": 2201
+    },
+    {
+      "epoch": 0.05591742659585129,
+      "grad_norm": 1.4454214572906494,
+      "learning_rate": 4.992687246588743e-05,
+      "loss": 2.0381,
+      "step": 2232
+    },
+    {
+      "epoch": 0.056694057520793664,
+      "grad_norm": 1.4872742891311646,
+      "learning_rate": 4.992031299767347e-05,
+      "loss": 2.0227,
+      "step": 2263
+    },
+    {
+      "epoch": 0.057470688445736044,
+      "grad_norm": 1.4538527727127075,
+      "learning_rate": 4.9913472307114386e-05,
+      "loss": 2.0395,
+      "step": 2294
+    },
+    {
+      "epoch": 0.058247319370678424,
+      "grad_norm": 1.432769536972046,
+      "learning_rate": 4.9906350471406446e-05,
+      "loss": 2.0352,
+      "step": 2325
+    },
+    {
+      "epoch": 0.059023950295620804,
+      "grad_norm": 1.4535977840423584,
+      "learning_rate": 4.989894757091861e-05,
+      "loss": 2.0341,
+      "step": 2356
+    },
+    {
+      "epoch": 0.059800581220563184,
+      "grad_norm": 1.4004398584365845,
+      "learning_rate": 4.989126368919158e-05,
+      "loss": 2.0185,
+      "step": 2387
+    },
+    {
+      "epoch": 0.060577212145505564,
+      "grad_norm": 1.433273434638977,
+      "learning_rate": 4.988329891293693e-05,
+      "loss": 2.036,
+      "step": 2418
+    },
+    {
+      "epoch": 0.06135384307044794,
+      "grad_norm": 1.3791290521621704,
+      "learning_rate": 4.987505333203608e-05,
+      "loss": 2.0421,
+      "step": 2449
+    },
+    {
+      "epoch": 0.06213047399539032,
+      "grad_norm": 2.008694648742676,
+      "learning_rate": 4.9866527039539276e-05,
+      "loss": 1.9948,
+      "step": 2480
+    },
+    {
+      "epoch": 0.0629071049203327,
+      "grad_norm": 1.3773164749145508,
+      "learning_rate": 4.9857720131664594e-05,
+      "loss": 2.0158,
+      "step": 2511
+    },
+    {
+      "epoch": 0.06368373584527508,
+      "grad_norm": 1.4489777088165283,
+      "learning_rate": 4.9848632707796773e-05,
+      "loss": 2.0216,
+      "step": 2542
+    },
+    {
+      "epoch": 0.06446036677021745,
+      "grad_norm": 2.1266908645629883,
+      "learning_rate": 4.9839264870486155e-05,
+      "loss": 1.9962,
+      "step": 2573
+    },
+    {
+      "epoch": 0.06523699769515984,
+      "grad_norm": 1.3704332113265991,
+      "learning_rate": 4.9829616725447526e-05,
+      "loss": 1.9801,
+      "step": 2604
+    },
+    {
+      "epoch": 0.06601362862010221,
+      "grad_norm": 1.3643279075622559,
+      "learning_rate": 4.981968838155888e-05,
+      "loss": 1.9902,
+      "step": 2635
+    },
+    {
+      "epoch": 0.0667902595450446,
+      "grad_norm": 1.3495409488677979,
+      "learning_rate": 4.980947995086024e-05,
+      "loss": 1.993,
+      "step": 2666
+    },
+    {
+      "epoch": 0.06756689046998697,
+      "grad_norm": 1.3918874263763428,
+      "learning_rate": 4.979899154855234e-05,
+      "loss": 1.9782,
+      "step": 2697
+    },
+    {
+      "epoch": 0.06834352139492936,
+      "grad_norm": 1.3205535411834717,
+      "learning_rate": 4.9788223292995386e-05,
+      "loss": 1.9638,
+      "step": 2728
+    },
+    {
+      "epoch": 0.06912015231987173,
+      "grad_norm": 1.3123114109039307,
+      "learning_rate": 4.977717530570768e-05,
+      "loss": 1.9826,
+      "step": 2759
+    },
+    {
+      "epoch": 0.0698967832448141,
+      "grad_norm": 1.3138995170593262,
+      "learning_rate": 4.976584771136425e-05,
+      "loss": 1.9938,
+      "step": 2790
+    },
+    {
+      "epoch": 0.07067341416975649,
+      "grad_norm": 1.2964009046554565,
+      "learning_rate": 4.975424063779547e-05,
+      "loss": 1.9924,
+      "step": 2821
+    },
+    {
+      "epoch": 0.07145004509469886,
+      "grad_norm": 1.3010739088058472,
+      "learning_rate": 4.974235421598557e-05,
+      "loss": 1.9592,
+      "step": 2852
+    },
+    {
+      "epoch": 0.07222667601964125,
+      "grad_norm": 1.2549103498458862,
+      "learning_rate": 4.973018858007122e-05,
+      "loss": 1.9394,
+      "step": 2883
+    },
+    {
+      "epoch": 0.07300330694458362,
+      "grad_norm": 1.3198585510253906,
+      "learning_rate": 4.9717743867339963e-05,
+      "loss": 1.989,
+      "step": 2914
+    },
+    {
+      "epoch": 0.07377993786952601,
+      "grad_norm": 1.3653398752212524,
+      "learning_rate": 4.9705020218228695e-05,
+      "loss": 1.9499,
+      "step": 2945
+    },
+    {
+      "epoch": 0.07455656879446838,
+      "grad_norm": 4.606905937194824,
+      "learning_rate": 4.969201777632205e-05,
+      "loss": 1.9504,
+      "step": 2976
+    },
+    {
+      "epoch": 0.07533319971941076,
+      "grad_norm": 1.2823129892349243,
+      "learning_rate": 4.9678736688350846e-05,
+      "loss": 1.9483,
+      "step": 3007
+    },
+    {
+      "epoch": 0.07610983064435314,
+      "grad_norm": 1.2803308963775635,
+      "learning_rate": 4.966517710419033e-05,
+      "loss": 1.9604,
+      "step": 3038
+    },
+    {
+      "epoch": 0.07688646156929552,
+      "grad_norm": 1.2765967845916748,
+      "learning_rate": 4.965133917685858e-05,
+      "loss": 1.941,
+      "step": 3069
+    },
+    {
+      "epoch": 0.0776630924942379,
+      "grad_norm": 2.622286796569824,
+      "learning_rate": 4.9637223062514714e-05,
+      "loss": 1.9426,
+      "step": 3100
+    },
+    {
+      "epoch": 0.07843972341918028,
+      "grad_norm": 1.231387972831726,
+      "learning_rate": 4.962282892045718e-05,
+      "loss": 1.9416,
+      "step": 3131
+    },
+    {
+      "epoch": 0.07921635434412266,
+      "grad_norm": 1.2499909400939941,
+      "learning_rate": 4.9608156913121904e-05,
+      "loss": 1.9226,
+      "step": 3162
+    },
+    {
+      "epoch": 0.07999298526906504,
+      "grad_norm": 1.296276330947876,
+      "learning_rate": 4.959320720608049e-05,
+      "loss": 1.935,
+      "step": 3193
+    },
+    {
+      "epoch": 0.08076961619400741,
+      "grad_norm": 1.2642258405685425,
+      "learning_rate": 4.9577979968038354e-05,
+      "loss": 1.9185,
+      "step": 3224
+    },
+    {
+      "epoch": 0.0815462471189498,
+      "grad_norm": 1.2396386861801147,
+      "learning_rate": 4.956247537083282e-05,
+      "loss": 1.9361,
+      "step": 3255
+    },
+    {
+      "epoch": 0.08232287804389217,
+      "grad_norm": 1.1811271905899048,
+      "learning_rate": 4.9546693589431145e-05,
+      "loss": 1.9276,
+      "step": 3286
+    },
+    {
+      "epoch": 0.08309950896883456,
+      "grad_norm": 1.2897577285766602,
+      "learning_rate": 4.9530634801928595e-05,
+      "loss": 1.9277,
+      "step": 3317
+    },
+    {
+      "epoch": 0.08387613989377693,
+      "grad_norm": 1.1453903913497925,
+      "learning_rate": 4.9514299189546395e-05,
+      "loss": 1.9109,
+      "step": 3348
+    },
+    {
+      "epoch": 0.08465277081871932,
+      "grad_norm": 1.1600992679595947,
+      "learning_rate": 4.949768693662973e-05,
+      "loss": 1.9248,
+      "step": 3379
+    },
+    {
+      "epoch": 0.08542940174366169,
+      "grad_norm": 1.234217882156372,
+      "learning_rate": 4.948079823064559e-05,
+      "loss": 1.9236,
+      "step": 3410
+    },
+    {
+      "epoch": 0.08620603266860406,
+      "grad_norm": 1.1933766603469849,
+      "learning_rate": 4.946363326218074e-05,
+      "loss": 1.9284,
+      "step": 3441
+    },
+    {
+      "epoch": 0.08698266359354645,
+      "grad_norm": 1.2087618112564087,
+      "learning_rate": 4.9446192224939525e-05,
+      "loss": 1.8971,
+      "step": 3472
+    },
+    {
+      "epoch": 0.08775929451848882,
+      "grad_norm": 1.227864146232605,
+      "learning_rate": 4.942847531574167e-05,
+      "loss": 1.9353,
+      "step": 3503
+    },
+    {
+      "epoch": 0.08853592544343121,
+      "grad_norm": 1.2828229665756226,
+      "learning_rate": 4.941048273452008e-05,
+      "loss": 1.9085,
+      "step": 3534
+    },
+    {
+      "epoch": 0.08931255636837358,
+      "grad_norm": 1.2086588144302368,
+      "learning_rate": 4.9392214684318605e-05,
+      "loss": 1.9044,
+      "step": 3565
+    },
+    {
+      "epoch": 0.09008918729331597,
+      "grad_norm": 1.1942152976989746,
+      "learning_rate": 4.93736713712897e-05,
+      "loss": 1.9182,
+      "step": 3596
+    },
+    {
+      "epoch": 0.09086581821825834,
+      "grad_norm": 1.2399394512176514,
+      "learning_rate": 4.9354853004692124e-05,
+      "loss": 1.8943,
+      "step": 3627
+    },
+    {
+      "epoch": 0.09164244914320072,
+      "grad_norm": 1.227376937866211,
+      "learning_rate": 4.93357597968886e-05,
+      "loss": 1.8881,
+      "step": 3658
+    },
+    {
+      "epoch": 0.0924190800681431,
+      "grad_norm": 1.1314102411270142,
+      "learning_rate": 4.931639196334338e-05,
+      "loss": 1.8878,
+      "step": 3689
+    },
+    {
+      "epoch": 0.09319571099308548,
+      "grad_norm": 1.192134141921997,
+      "learning_rate": 4.9296749722619826e-05,
+      "loss": 1.8857,
+      "step": 3720
+    },
+    {
+      "epoch": 0.09397234191802786,
+      "grad_norm": 1.1905503273010254,
+      "learning_rate": 4.9276833296377966e-05,
+      "loss": 1.8701,
+      "step": 3751
+    },
+    {
+      "epoch": 0.09474897284297024,
+      "grad_norm": 1.2335184812545776,
+      "learning_rate": 4.925664290937196e-05,
+      "loss": 1.8741,
+      "step": 3782
+    },
+    {
+      "epoch": 0.09552560376791262,
+      "grad_norm": 1.1403234004974365,
+      "learning_rate": 4.9236178789447576e-05,
+      "loss": 1.8658,
+      "step": 3813
+    },
+    {
+      "epoch": 0.096302234692855,
+      "grad_norm": 1.1734315156936646,
+      "learning_rate": 4.921544116753962e-05,
+      "loss": 1.8724,
+      "step": 3844
+    },
+    {
+      "epoch": 0.09707886561779737,
+      "grad_norm": 1.2637231349945068,
+      "learning_rate": 4.919443027766935e-05,
+      "loss": 1.867,
+      "step": 3875
+    },
+    {
+      "epoch": 0.09785549654273976,
+      "grad_norm": 1.1608805656433105,
+      "learning_rate": 4.91731463569418e-05,
+      "loss": 1.8764,
+      "step": 3906
+    },
+    {
+      "epoch": 0.09863212746768213,
+      "grad_norm": 1.1562029123306274,
+      "learning_rate": 4.915158964554312e-05,
+      "loss": 1.8612,
+      "step": 3937
+    },
+    {
+      "epoch": 0.09940875839262452,
+      "grad_norm": 1.1918668746948242,
+      "learning_rate": 4.912976038673786e-05,
+      "loss": 1.866,
+      "step": 3968
+    },
+    {
+      "epoch": 0.10018538931756689,
+      "grad_norm": 1.180206537246704,
+      "learning_rate": 4.9107658826866254e-05,
+      "loss": 1.849,
+      "step": 3999
+    },
+    {
+      "epoch": 0.10096202024250928,
+      "grad_norm": 1.0747199058532715,
+      "learning_rate": 4.908528521534139e-05,
+      "loss": 1.8547,
+      "step": 4030
+    },
+    {
+      "epoch": 0.10173865116745165,
+      "grad_norm": 1.103786826133728,
+      "learning_rate": 4.906263980464644e-05,
+      "loss": 1.864,
+      "step": 4061
+    },
+    {
+      "epoch": 0.10251528209239402,
+      "grad_norm": 1.1390912532806396,
+      "learning_rate": 4.903972285033178e-05,
+      "loss": 1.8669,
+      "step": 4092
+    },
+    {
+      "epoch": 0.10329191301733641,
+      "grad_norm": 1.1488468647003174,
+      "learning_rate": 4.901653461101213e-05,
+      "loss": 1.8445,
+      "step": 4123
+    },
+    {
+      "epoch": 0.10406854394227878,
+      "grad_norm": 1.1589927673339844,
+      "learning_rate": 4.8993075348363626e-05,
+      "loss": 1.8529,
+      "step": 4154
+    },
+    {
+      "epoch": 0.10484517486722117,
+      "grad_norm": 1.1793478727340698,
+      "learning_rate": 4.896934532712084e-05,
+      "loss": 1.8818,
+      "step": 4185
+    },
+    {
+      "epoch": 0.10562180579216354,
+      "grad_norm": 1.165292739868164,
+      "learning_rate": 4.8945344815073846e-05,
+      "loss": 1.8425,
+      "step": 4216
+    },
+    {
+      "epoch": 0.10639843671710593,
+      "grad_norm": 1.105946660041809,
+      "learning_rate": 4.892107408306516e-05,
+      "loss": 1.8327,
+      "step": 4247
+    },
+    {
+      "epoch": 0.1071750676420483,
+      "grad_norm": 1.1690527200698853,
+      "learning_rate": 4.889653340498669e-05,
+      "loss": 1.8428,
+      "step": 4278
+    },
+    {
+      "epoch": 0.10795169856699068,
+      "grad_norm": 1.1106913089752197,
+      "learning_rate": 4.8871723057776664e-05,
+      "loss": 1.839,
+      "step": 4309
+    },
+    {
+      "epoch": 0.10872832949193306,
+      "grad_norm": 1.1590090990066528,
+      "learning_rate": 4.8846643321416476e-05,
+      "loss": 1.8658,
+      "step": 4340
+    },
+    {
+      "epoch": 0.10950496041687544,
+      "grad_norm": 1.1315875053405762,
+      "learning_rate": 4.882129447892753e-05,
+      "loss": 1.8282,
+      "step": 4371
+    },
+    {
+      "epoch": 0.11028159134181782,
+      "grad_norm": 1.6055009365081787,
+      "learning_rate": 4.8795676816368076e-05,
+      "loss": 1.8294,
+      "step": 4402
+    },
+    {
+      "epoch": 0.1110582222667602,
+      "grad_norm": 1.1236201524734497,
+      "learning_rate": 4.876979062282995e-05,
+      "loss": 1.8325,
+      "step": 4433
+    },
+    {
+      "epoch": 0.11183485319170258,
+      "grad_norm": 1.0830161571502686,
+      "learning_rate": 4.8743636190435325e-05,
+      "loss": 1.8451,
+      "step": 4464
+    },
+    {
+      "epoch": 0.11261148411664496,
+      "grad_norm": 1.1663005352020264,
+      "learning_rate": 4.871721381433344e-05,
+      "loss": 1.8349,
+      "step": 4495
+    },
+    {
+      "epoch": 0.11338811504158733,
+      "grad_norm": 1.2522406578063965,
+      "learning_rate": 4.869052379269719e-05,
+      "loss": 1.8162,
+      "step": 4526
+    },
+    {
+      "epoch": 0.11416474596652972,
+      "grad_norm": 1.1271142959594727,
+      "learning_rate": 4.866356642671985e-05,
+      "loss": 1.8348,
+      "step": 4557
+    },
+    {
+      "epoch": 0.11494137689147209,
+      "grad_norm": 1.053753137588501,
+      "learning_rate": 4.8636342020611634e-05,
+      "loss": 1.8376,
+      "step": 4588
+    },
+    {
+      "epoch": 0.11571800781641448,
+      "grad_norm": 1.1469056606292725,
+      "learning_rate": 4.860885088159626e-05,
+      "loss": 1.8248,
+      "step": 4619
+    },
+    {
+      "epoch": 0.11649463874135685,
+      "grad_norm": 1.116128921508789,
+      "learning_rate": 4.858109331990751e-05,
+      "loss": 1.8454,
+      "step": 4650
+    },
+    {
+      "epoch": 0.11727126966629922,
+      "grad_norm": 1.0742651224136353,
+      "learning_rate": 4.855306964878567e-05,
+      "loss": 1.8231,
+      "step": 4681
+    },
+    {
+      "epoch": 0.11804790059124161,
+      "grad_norm": 1.101657748222351,
+      "learning_rate": 4.8524780184474084e-05,
+      "loss": 1.8421,
+      "step": 4712
+    },
+    {
+      "epoch": 0.11882453151618398,
+      "grad_norm": 1.0891473293304443,
+      "learning_rate": 4.8496225246215496e-05,
+      "loss": 1.8053,
+      "step": 4743
+    },
+    {
+      "epoch": 0.11960116244112637,
+      "grad_norm": 1.1044034957885742,
+      "learning_rate": 4.8467405156248505e-05,
+      "loss": 1.8184,
+      "step": 4774
+    },
+    {
+      "epoch": 0.12037779336606874,
+      "grad_norm": 1.0644783973693848,
+      "learning_rate": 4.843832023980392e-05,
+      "loss": 1.8237,
+      "step": 4805
+    },
+    {
+      "epoch": 0.12115442429101113,
+      "grad_norm": 1.0427252054214478,
+      "learning_rate": 4.840897082510106e-05,
+      "loss": 1.8189,
+      "step": 4836
+    },
+    {
+      "epoch": 0.1219310552159535,
+      "grad_norm": 1.0806516408920288,
+      "learning_rate": 4.8379357243344084e-05,
+      "loss": 1.7958,
+      "step": 4867
+    },
+    {
+      "epoch": 0.12270768614089587,
+      "grad_norm": 1.092581868171692,
+      "learning_rate": 4.8349479828718236e-05,
+      "loss": 1.8229,
+      "step": 4898
+    },
+    {
+      "epoch": 0.12348431706583826,
+      "grad_norm": 1.0858657360076904,
+      "learning_rate": 4.8319338918386075e-05,
+      "loss": 1.8165,
+      "step": 4929
+    },
+    {
+      "epoch": 0.12426094799078063,
+      "grad_norm": 1.073818325996399,
+      "learning_rate": 4.828893485248369e-05,
+      "loss": 1.8352,
+      "step": 4960
+    },
+    {
+      "epoch": 0.125037578915723,
+      "grad_norm": 1.069009780883789,
+      "learning_rate": 4.825826797411682e-05,
+      "loss": 1.8,
+      "step": 4991
+    },
+    {
+      "epoch": 0.1258142098406654,
+      "grad_norm": 1.0416266918182373,
+      "learning_rate": 4.822733862935702e-05,
+      "loss": 1.8224,
+      "step": 5022
+    },
+    {
+      "epoch": 0.12659084076560778,
+      "grad_norm": 1.093551516532898,
+      "learning_rate": 4.819614716723775e-05,
+      "loss": 1.8005,
+      "step": 5053
+    },
+    {
+      "epoch": 0.12736747169055015,
+      "grad_norm": 1.026353359222412,
+      "learning_rate": 4.8164693939750425e-05,
+      "loss": 1.844,
+      "step": 5084
+    },
+    {
+      "epoch": 0.12814410261549253,
+      "grad_norm": 1.102858304977417,
+      "learning_rate": 4.813297930184042e-05,
+      "loss": 1.8045,
+      "step": 5115
+    },
+    {
+      "epoch": 0.1289207335404349,
+      "grad_norm": 1.0537272691726685,
+      "learning_rate": 4.810100361140314e-05,
+      "loss": 1.8054,
+      "step": 5146
+    },
+    {
+      "epoch": 0.1296973644653773,
+      "grad_norm": 1.0389387607574463,
+      "learning_rate": 4.8068767229279885e-05,
+      "loss": 1.7734,
+      "step": 5177
+    },
+    {
+      "epoch": 0.13047399539031967,
+      "grad_norm": 1.0274319648742676,
+      "learning_rate": 4.8036270519253854e-05,
+      "loss": 1.788,
+      "step": 5208
+    },
+    {
+      "epoch": 0.13125062631526205,
+      "grad_norm": 1.0932128429412842,
+      "learning_rate": 4.8003513848046e-05,
+      "loss": 1.7813,
+      "step": 5239
+    },
+    {
+      "epoch": 0.13202725724020442,
+      "grad_norm": 1.0294640064239502,
+      "learning_rate": 4.79704975853109e-05,
+      "loss": 1.81,
+      "step": 5270
+    },
+    {
+      "epoch": 0.13280388816514682,
+      "grad_norm": 1.0617210865020752,
+      "learning_rate": 4.793722210363262e-05,
+      "loss": 1.8017,
+      "step": 5301
+    },
+    {
+      "epoch": 0.1335805190900892,
+      "grad_norm": 1.0657798051834106,
+      "learning_rate": 4.7903687778520414e-05,
+      "loss": 1.7884,
+      "step": 5332
+    },
+    {
+      "epoch": 0.13435715001503157,
+      "grad_norm": 0.9954230189323425,
+      "learning_rate": 4.7869894988404593e-05,
+      "loss": 1.7848,
+      "step": 5363
+    },
+    {
+      "epoch": 0.13513378093997394,
+      "grad_norm": 1.117066740989685,
+      "learning_rate": 4.783584411463221e-05,
+      "loss": 1.7984,
+      "step": 5394
+    },
+    {
+      "epoch": 0.13591041186491631,
+      "grad_norm": 1.0706870555877686,
+      "learning_rate": 4.780153554146274e-05,
+      "loss": 1.8008,
+      "step": 5425
+    },
+    {
+      "epoch": 0.13668704278985871,
+      "grad_norm": 1.063238501548767,
+      "learning_rate": 4.7766969656063766e-05,
+      "loss": 1.7754,
+      "step": 5456
+    },
+    {
+      "epoch": 0.1374636737148011,
+      "grad_norm": 1.0702152252197266,
+      "learning_rate": 4.773214684850662e-05,
+      "loss": 1.8037,
+      "step": 5487
+    },
+    {
+      "epoch": 0.13824030463974346,
+      "grad_norm": 1.0318788290023804,
+      "learning_rate": 4.769706751176193e-05,
+      "loss": 1.8174,
+      "step": 5518
+    },
+    {
+      "epoch": 0.13901693556468583,
+      "grad_norm": 1.0097755193710327,
+      "learning_rate": 4.7661732041695264e-05,
+      "loss": 1.7648,
+      "step": 5549
+    },
+    {
+      "epoch": 0.1397935664896282,
+      "grad_norm": 1.063873291015625,
+      "learning_rate": 4.762614083706258e-05,
+      "loss": 1.809,
+      "step": 5580
+    },
+    {
+      "epoch": 0.1405701974145706,
+      "grad_norm": 1.0409623384475708,
+      "learning_rate": 4.759029429950581e-05,
+      "loss": 1.7853,
+      "step": 5611
+    },
+    {
+      "epoch": 0.14134682833951298,
+      "grad_norm": 1.0131454467773438,
+      "learning_rate": 4.7554192833548235e-05,
+      "loss": 1.7759,
+      "step": 5642
+    },
+    {
+      "epoch": 0.14212345926445535,
+      "grad_norm": 1.1056182384490967,
+      "learning_rate": 4.751783684659e-05,
+      "loss": 1.7959,
+      "step": 5673
+    },
+    {
+      "epoch": 0.14290009018939773,
+      "grad_norm": 1.0572400093078613,
+      "learning_rate": 4.748122674890348e-05,
+      "loss": 1.7597,
+      "step": 5704
+    },
+    {
+      "epoch": 0.14367672111434013,
+      "grad_norm": 1.0375362634658813,
+      "learning_rate": 4.7444362953628654e-05,
+      "loss": 1.7719,
+      "step": 5735
+    },
+    {
+      "epoch": 0.1444533520392825,
+      "grad_norm": 1.0238806009292603,
+      "learning_rate": 4.7407245876768424e-05,
+      "loss": 1.7966,
+      "step": 5766
+    },
+    {
+      "epoch": 0.14522998296422487,
+      "grad_norm": 1.0335805416107178,
+      "learning_rate": 4.736987593718397e-05,
+      "loss": 1.7805,
+      "step": 5797
+    },
+    {
+      "epoch": 0.14600661388916725,
+      "grad_norm": 1.08907151222229,
+      "learning_rate": 4.733225355658999e-05,
+      "loss": 1.7905,
+      "step": 5828
+    },
+    {
+      "epoch": 0.14678324481410962,
+      "grad_norm": 1.0151680707931519,
+      "learning_rate": 4.7294379159549926e-05,
+      "loss": 1.7766,
+      "step": 5859
+    },
+    {
+      "epoch": 0.14755987573905202,
+      "grad_norm": 0.9963664412498474,
+      "learning_rate": 4.725625317347119e-05,
+      "loss": 1.7708,
+      "step": 5890
+    },
+    {
+      "epoch": 0.1483365066639944,
+      "grad_norm": 1.0762684345245361,
+      "learning_rate": 4.7217876028600374e-05,
+      "loss": 1.7998,
+      "step": 5921
+    },
+    {
+      "epoch": 0.14911313758893677,
+      "grad_norm": 1.1151126623153687,
+      "learning_rate": 4.717924815801832e-05,
+      "loss": 1.782,
+      "step": 5952
+    },
+    {
+      "epoch": 0.14988976851387914,
+      "grad_norm": 0.9970519542694092,
+      "learning_rate": 4.714036999763532e-05,
+      "loss": 1.7779,
+      "step": 5983
+    },
+    {
+      "epoch": 0.1506663994388215,
+      "grad_norm": 1.0146524906158447,
+      "learning_rate": 4.7101241986186116e-05,
+      "loss": 1.7631,
+      "step": 6014
+    },
+    {
+      "epoch": 0.15144303036376391,
+      "grad_norm": 1.0554558038711548,
+      "learning_rate": 4.7061864565225e-05,
+      "loss": 1.7729,
+      "step": 6045
+    },
+    {
+      "epoch": 0.1522196612887063,
+      "grad_norm": 1.0413624048233032,
+      "learning_rate": 4.702223817912081e-05,
+      "loss": 1.79,
+      "step": 6076
+    },
+    {
+      "epoch": 0.15299629221364866,
+      "grad_norm": 1.07778000831604,
+      "learning_rate": 4.698236327505195e-05,
+      "loss": 1.787,
+      "step": 6107
+    },
+    {
+      "epoch": 0.15377292313859103,
+      "grad_norm": 1.0084989070892334,
+      "learning_rate": 4.694224030300127e-05,
+      "loss": 1.7695,
+      "step": 6138
+    },
+    {
+      "epoch": 0.1545495540635334,
+      "grad_norm": 1.0067577362060547,
+      "learning_rate": 4.690186971575107e-05,
+      "loss": 1.7611,
+      "step": 6169
+    },
+    {
+      "epoch": 0.1553261849884758,
+      "grad_norm": 1.0149894952774048,
+      "learning_rate": 4.6861251968877916e-05,
+      "loss": 1.7734,
+      "step": 6200
+    },
+    {
+      "epoch": 0.15610281591341818,
+      "grad_norm": 0.996447741985321,
+      "learning_rate": 4.68203875207476e-05,
+      "loss": 1.7609,
+      "step": 6231
+    },
+    {
+      "epoch": 0.15687944683836055,
+      "grad_norm": 1.0334714651107788,
+      "learning_rate": 4.677927683250983e-05,
+      "loss": 1.7725,
+      "step": 6262
+    },
+    {
+      "epoch": 0.15765607776330293,
+      "grad_norm": 0.949685275554657,
+      "learning_rate": 4.6737920368093156e-05,
+      "loss": 1.7597,
+      "step": 6293
+    },
+    {
+      "epoch": 0.15843270868824533,
+      "grad_norm": 1.0946801900863647,
+      "learning_rate": 4.669631859419965e-05,
+      "loss": 1.755,
+      "step": 6324
+    },
+    {
+      "epoch": 0.1592093396131877,
+      "grad_norm": 1.0390102863311768,
+      "learning_rate": 4.6654471980299676e-05,
+      "loss": 1.7755,
+      "step": 6355
+    },
+    {
+      "epoch": 0.15998597053813007,
+      "grad_norm": 0.974733293056488,
+      "learning_rate": 4.661238099862658e-05,
+      "loss": 1.7619,
+      "step": 6386
+    },
+    {
+      "epoch": 0.16076260146307245,
+      "grad_norm": 1.0192877054214478,
+      "learning_rate": 4.657004612417138e-05,
+      "loss": 1.7676,
+      "step": 6417
+    },
+    {
+      "epoch": 0.16153923238801482,
+      "grad_norm": 1.0147426128387451,
+      "learning_rate": 4.6527467834677374e-05,
+      "loss": 1.7473,
+      "step": 6448
+    },
+    {
+      "epoch": 0.16231586331295722,
+      "grad_norm": 1.0231157541275024,
+      "learning_rate": 4.648464661063478e-05,
+      "loss": 1.766,
+      "step": 6479
+    },
+    {
+      "epoch": 0.1630924942378996,
+      "grad_norm": 0.9995896816253662,
+      "learning_rate": 4.6441582935275264e-05,
+      "loss": 1.7594,
+      "step": 6510
+    },
+    {
+      "epoch": 0.16386912516284197,
+      "grad_norm": 0.9674972295761108,
+      "learning_rate": 4.6398277294566586e-05,
+      "loss": 1.7347,
+      "step": 6541
+    },
+    {
+      "epoch": 0.16464575608778434,
+      "grad_norm": 0.9876355528831482,
+      "learning_rate": 4.6354730177207e-05,
+      "loss": 1.748,
+      "step": 6572
+    },
+    {
+      "epoch": 0.1654223870127267,
+      "grad_norm": 1.0107293128967285,
+      "learning_rate": 4.6310942074619787e-05,
+      "loss": 1.7743,
+      "step": 6603
+    },
+    {
+      "epoch": 0.1661990179376691,
+      "grad_norm": 0.9731600284576416,
+      "learning_rate": 4.626691348094777e-05,
+      "loss": 1.7515,
+      "step": 6634
+    },
+    {
+      "epoch": 0.1669756488626115,
+      "grad_norm": 0.9843081831932068,
+      "learning_rate": 4.622264489304762e-05,
+      "loss": 1.7498,
+      "step": 6665
+    },
+    {
+      "epoch": 0.16775227978755386,
+      "grad_norm": 0.9850636720657349,
+      "learning_rate": 4.617813681048434e-05,
+      "loss": 1.7563,
+      "step": 6696
+    },
+    {
+      "epoch": 0.16852891071249623,
+      "grad_norm": 0.964599609375,
+      "learning_rate": 4.61333897355256e-05,
+      "loss": 1.7699,
+      "step": 6727
+    },
+    {
+      "epoch": 0.16930554163743863,
+      "grad_norm": 0.9415149688720703,
+      "learning_rate": 4.608840417313604e-05,
+      "loss": 1.745,
+      "step": 6758
+    },
+    {
+      "epoch": 0.170082172562381,
+      "grad_norm": 0.9608363509178162,
+      "learning_rate": 4.6043180630971646e-05,
+      "loss": 1.7519,
+      "step": 6789
+    },
+    {
+      "epoch": 0.17085880348732338,
+      "grad_norm": 0.9133123159408569,
+      "learning_rate": 4.599771961937391e-05,
+      "loss": 1.7432,
+      "step": 6820
+    },
+    {
+      "epoch": 0.17163543441226575,
+      "grad_norm": 0.943702220916748,
+      "learning_rate": 4.5952021651364204e-05,
+      "loss": 1.7231,
+      "step": 6851
+    },
+    {
+      "epoch": 0.17241206533720813,
+      "grad_norm": 0.9676947593688965,
+      "learning_rate": 4.590608724263786e-05,
+      "loss": 1.7563,
+      "step": 6882
+    },
+    {
+      "epoch": 0.17318869626215053,
+      "grad_norm": 0.964959442615509,
+      "learning_rate": 4.585991691155845e-05,
+      "loss": 1.7514,
+      "step": 6913
+    },
+    {
+      "epoch": 0.1739653271870929,
+      "grad_norm": 1.0078297853469849,
+      "learning_rate": 4.581351117915188e-05,
+      "loss": 1.7538,
+      "step": 6944
+    },
+    {
+      "epoch": 0.17474195811203527,
+      "grad_norm": 0.9631285071372986,
+      "learning_rate": 4.5766870569100534e-05,
+      "loss": 1.7363,
+      "step": 6975
+    },
+    {
+      "epoch": 0.17551858903697765,
+      "grad_norm": 0.9542151093482971,
+      "learning_rate": 4.571999560773736e-05,
+      "loss": 1.753,
+      "step": 7006
+    },
+    {
+      "epoch": 0.17629521996192002,
+      "grad_norm": 0.9770635366439819,
+      "learning_rate": 4.5672886824039915e-05,
+      "loss": 1.7435,
+      "step": 7037
+    },
+    {
+      "epoch": 0.17707185088686242,
+      "grad_norm": 0.979009211063385,
+      "learning_rate": 4.5625544749624435e-05,
+      "loss": 1.7591,
+      "step": 7068
+    },
+    {
+      "epoch": 0.1778484818118048,
+      "grad_norm": 0.9663967490196228,
+      "learning_rate": 4.5577969918739794e-05,
+      "loss": 1.7129,
+      "step": 7099
+    },
+    {
+      "epoch": 0.17862511273674717,
+      "grad_norm": 1.0181857347488403,
+      "learning_rate": 4.5530162868261486e-05,
+      "loss": 1.7396,
+      "step": 7130
+    },
+    {
+      "epoch": 0.17940174366168954,
+      "grad_norm": 1.0090553760528564,
+      "learning_rate": 4.548212413768558e-05,
+      "loss": 1.7488,
+      "step": 7161
+    },
+    {
+      "epoch": 0.18017837458663194,
+      "grad_norm": 1.0095964670181274,
+      "learning_rate": 4.543385426912261e-05,
+      "loss": 1.7342,
+      "step": 7192
+    },
+    {
+      "epoch": 0.1809550055115743,
+      "grad_norm": 0.994108259677887,
+      "learning_rate": 4.53853538072915e-05,
+      "loss": 1.7445,
+      "step": 7223
+    },
+    {
+      "epoch": 0.18173163643651669,
+      "grad_norm": 0.9566254019737244,
+      "learning_rate": 4.533662329951336e-05,
+      "loss": 1.7272,
+      "step": 7254
+    },
+    {
+      "epoch": 0.18250826736145906,
+      "grad_norm": 0.9655300974845886,
+      "learning_rate": 4.528766329570536e-05,
+      "loss": 1.7112,
+      "step": 7285
+    },
+    {
+      "epoch": 0.18328489828640143,
+      "grad_norm": 0.9693655371665955,
+      "learning_rate": 4.523847434837447e-05,
+      "loss": 1.74,
+      "step": 7316
+    },
+    {
+      "epoch": 0.18406152921134383,
+      "grad_norm": 0.9668208956718445,
+      "learning_rate": 4.518905701261128e-05,
+      "loss": 1.7239,
+      "step": 7347
+    },
+    {
+      "epoch": 0.1848381601362862,
+      "grad_norm": 1.0193735361099243,
+      "learning_rate": 4.5139411846083715e-05,
+      "loss": 1.7273,
+      "step": 7378
+    },
+    {
+      "epoch": 0.18561479106122858,
+      "grad_norm": 0.9525153636932373,
+      "learning_rate": 4.508953940903073e-05,
+      "loss": 1.7356,
+      "step": 7409
+    },
+    {
+      "epoch": 0.18639142198617095,
+      "grad_norm": 0.9590099453926086,
+      "learning_rate": 4.5039440264255994e-05,
+      "loss": 1.7296,
+      "step": 7440
+    },
+    {
+      "epoch": 0.18716805291111333,
+      "grad_norm": 0.9519559741020203,
+      "learning_rate": 4.498911497712155e-05,
+      "loss": 1.7183,
+      "step": 7471
+    },
+    {
+      "epoch": 0.18794468383605573,
+      "grad_norm": 0.9692674279212952,
+      "learning_rate": 4.493856411554142e-05,
+      "loss": 1.7329,
+      "step": 7502
+    },
+    {
+      "epoch": 0.1887213147609981,
+      "grad_norm": 0.9676815271377563,
+      "learning_rate": 4.4887788249975206e-05,
+      "loss": 1.7194,
+      "step": 7533
+    },
+    {
+      "epoch": 0.18949794568594047,
+      "grad_norm": 0.9612167477607727,
+      "learning_rate": 4.4836787953421656e-05,
+      "loss": 1.7297,
+      "step": 7564
+    },
+    {
+      "epoch": 0.19027457661088285,
+      "grad_norm": 0.9661709666252136,
+      "learning_rate": 4.478556380141218e-05,
+      "loss": 1.728,
+      "step": 7595
+    },
+    {
+      "epoch": 0.19105120753582525,
+      "grad_norm": 0.95663982629776,
+      "learning_rate": 4.4734116372004375e-05,
+      "loss": 1.7387,
+      "step": 7626
+    },
+    {
+      "epoch": 0.19182783846076762,
+      "grad_norm": 0.9469471573829651,
+      "learning_rate": 4.4682446245775477e-05,
+      "loss": 1.7034,
+      "step": 7657
+    },
+    {
+      "epoch": 0.19260446938571,
+      "grad_norm": 1.03744637966156,
+      "learning_rate": 4.463055400581586e-05,
+      "loss": 1.7394,
+      "step": 7688
+    },
+    {
+      "epoch": 0.19338110031065237,
+      "grad_norm": 0.932919979095459,
+      "learning_rate": 4.4578440237722374e-05,
+      "loss": 1.7113,
+      "step": 7719
+    },
+    {
+      "epoch": 0.19415773123559474,
+      "grad_norm": 0.9527629017829895,
+      "learning_rate": 4.452610552959183e-05,
+      "loss": 1.7228,
+      "step": 7750
+    },
+    {
+      "epoch": 0.19493436216053714,
+      "grad_norm": 0.9649306535720825,
+      "learning_rate": 4.447355047201428e-05,
+      "loss": 1.7434,
+      "step": 7781
+    },
+    {
+      "epoch": 0.1957109930854795,
+      "grad_norm": 0.9027903079986572,
+      "learning_rate": 4.4420775658066414e-05,
+      "loss": 1.7022,
+      "step": 7812
+    },
+    {
+      "epoch": 0.19648762401042189,
+      "grad_norm": 0.9486429691314697,
+      "learning_rate": 4.436778168330484e-05,
+      "loss": 1.7277,
+      "step": 7843
+    },
+    {
+      "epoch": 0.19726425493536426,
+      "grad_norm": 0.9951580762863159,
+      "learning_rate": 4.4314569145759353e-05,
+      "loss": 1.7354,
+      "step": 7874
+    },
+    {
+      "epoch": 0.19804088586030663,
+      "grad_norm": 0.984990656375885,
+      "learning_rate": 4.42611386459262e-05,
+      "loss": 1.7247,
+      "step": 7905
+    },
+    {
+      "epoch": 0.19881751678524903,
+      "grad_norm": 1.0146019458770752,
+      "learning_rate": 4.420749078676133e-05,
+      "loss": 1.7048,
+      "step": 7936
+    },
+    {
+      "epoch": 0.1995941477101914,
+      "grad_norm": 0.9386274814605713,
+      "learning_rate": 4.4153626173673516e-05,
+      "loss": 1.7035,
+      "step": 7967
+    },
+    {
+      "epoch": 0.20037077863513378,
+      "grad_norm": 0.9656661152839661,
+      "learning_rate": 4.409954541451762e-05,
+      "loss": 1.7138,
+      "step": 7998
+    },
+    {
+      "epoch": 0.20114740956007615,
+      "grad_norm": 0.9721409678459167,
+      "learning_rate": 4.404524911958764e-05,
+      "loss": 1.7174,
+      "step": 8029
+    },
+    {
+      "epoch": 0.20192404048501855,
+      "grad_norm": 0.9685981869697571,
+      "learning_rate": 4.399073790160989e-05,
+      "loss": 1.7162,
+      "step": 8060
+    },
+    {
+      "epoch": 0.20270067140996093,
+      "grad_norm": 0.9472204446792603,
+      "learning_rate": 4.393601237573607e-05,
+      "loss": 1.7068,
+      "step": 8091
+    },
+    {
+      "epoch": 0.2034773023349033,
+      "grad_norm": 0.9472411870956421,
+      "learning_rate": 4.388107315953628e-05,
+      "loss": 1.7244,
+      "step": 8122
+    },
+    {
+      "epoch": 0.20425393325984567,
+      "grad_norm": 0.9380378723144531,
+      "learning_rate": 4.382592087299212e-05,
+      "loss": 1.7169,
+      "step": 8153
+    },
+    {
+      "epoch": 0.20503056418478804,
+      "grad_norm": 0.9613087177276611,
+      "learning_rate": 4.377055613848964e-05,
+      "loss": 1.7108,
+      "step": 8184
+    },
+    {
+      "epoch": 0.20580719510973045,
+      "grad_norm": 0.9507735967636108,
+      "learning_rate": 4.3714979580812355e-05,
+      "loss": 1.7147,
+      "step": 8215
+    },
+    {
+      "epoch": 0.20658382603467282,
+      "grad_norm": 0.9669816493988037,
+      "learning_rate": 4.365919182713416e-05,
+      "loss": 1.7058,
+      "step": 8246
+    },
+    {
+      "epoch": 0.2073604569596152,
+      "grad_norm": 0.8880124688148499,
+      "learning_rate": 4.360319350701226e-05,
+      "loss": 1.7049,
+      "step": 8277
+    },
+    {
+      "epoch": 0.20813708788455756,
+      "grad_norm": 4.802671909332275,
+      "learning_rate": 4.3546985252380115e-05,
+      "loss": 1.7349,
+      "step": 8308
+    },
+    {
+      "epoch": 0.20891371880949994,
+      "grad_norm": 0.9394221901893616,
+      "learning_rate": 4.349056769754021e-05,
+      "loss": 1.6932,
+      "step": 8339
+    },
+    {
+      "epoch": 0.20969034973444234,
+      "grad_norm": 0.9521833062171936,
+      "learning_rate": 4.3433941479156994e-05,
+      "loss": 1.695,
+      "step": 8370
+    },
+    {
+      "epoch": 0.2104669806593847,
+      "grad_norm": 0.9627339839935303,
+      "learning_rate": 4.3377107236249647e-05,
+      "loss": 1.7329,
+      "step": 8401
+    },
+    {
+      "epoch": 0.21124361158432708,
+      "grad_norm": 0.9096025228500366,
+      "learning_rate": 4.332006561018488e-05,
+      "loss": 1.6885,
+      "step": 8432
+    },
+    {
+      "epoch": 0.21202024250926946,
+      "grad_norm": 0.9301928877830505,
+      "learning_rate": 4.3262817244669683e-05,
+      "loss": 1.7112,
+      "step": 8463
+    },
+    {
+      "epoch": 0.21279687343421186,
+      "grad_norm": 1.0362213850021362,
+      "learning_rate": 4.3205362785744083e-05,
+      "loss": 1.7061,
+      "step": 8494
+    },
+    {
+      "epoch": 0.21357350435915423,
+      "grad_norm": 1.986802339553833,
+      "learning_rate": 4.314770288177384e-05,
+      "loss": 1.7095,
+      "step": 8525
+    },
+    {
+      "epoch": 0.2143501352840966,
+      "grad_norm": 0.9070325493812561,
+      "learning_rate": 4.308983818344313e-05,
+      "loss": 1.7087,
+      "step": 8556
+    },
+    {
+      "epoch": 0.21512676620903898,
+      "grad_norm": 0.985433042049408,
+      "learning_rate": 4.3031769343747206e-05,
+      "loss": 1.7176,
+      "step": 8587
+    },
+    {
+      "epoch": 0.21590339713398135,
+      "grad_norm": 0.9339021444320679,
+      "learning_rate": 4.297349701798505e-05,
+      "loss": 1.6823,
+      "step": 8618
+    },
+    {
+      "epoch": 0.21668002805892375,
+      "grad_norm": 0.9413474798202515,
+      "learning_rate": 4.2915021863751916e-05,
+      "loss": 1.7205,
+      "step": 8649
+    },
+    {
+      "epoch": 0.21745665898386612,
+      "grad_norm": 1.023187279701233,
+      "learning_rate": 4.285634454093198e-05,
+      "loss": 1.6943,
+      "step": 8680
+    },
+    {
+      "epoch": 0.2182332899088085,
+      "grad_norm": 0.9431496262550354,
+      "learning_rate": 4.279746571169086e-05,
+      "loss": 1.7012,
+      "step": 8711
+    },
+    {
+      "epoch": 0.21900992083375087,
+      "grad_norm": 0.9386807680130005,
+      "learning_rate": 4.2738386040468136e-05,
+      "loss": 1.7069,
+      "step": 8742
+    },
+    {
+      "epoch": 0.21978655175869324,
+      "grad_norm": 0.9307882785797119,
+      "learning_rate": 4.2679106193969866e-05,
+      "loss": 1.6914,
+      "step": 8773
+    },
+    {
+      "epoch": 0.22056318268363564,
+      "grad_norm": 1.0440980195999146,
+      "learning_rate": 4.261962684116106e-05,
+      "loss": 1.6892,
+      "step": 8804
+    },
+    {
+      "epoch": 0.22133981360857802,
+      "grad_norm": 0.9271411895751953,
+      "learning_rate": 4.2559948653258145e-05,
+      "loss": 1.7117,
+      "step": 8835
+    },
+    {
+      "epoch": 0.2221164445335204,
+      "grad_norm": 0.9182149767875671,
+      "learning_rate": 4.250007230372134e-05,
+      "loss": 1.7092,
+      "step": 8866
+    },
+    {
+      "epoch": 0.22289307545846276,
+      "grad_norm": 0.9707378149032593,
+      "learning_rate": 4.2439998468247126e-05,
+      "loss": 1.7055,
+      "step": 8897
+    },
+    {
+      "epoch": 0.22366970638340516,
+      "grad_norm": 0.9413692951202393,
+      "learning_rate": 4.2379727824760566e-05,
+      "loss": 1.6968,
+      "step": 8928
+    },
+    {
+      "epoch": 0.22444633730834754,
+      "grad_norm": 1.1943906545639038,
+      "learning_rate": 4.231926105340768e-05,
+      "loss": 1.7174,
+      "step": 8959
+    },
+    {
+      "epoch": 0.2252229682332899,
+      "grad_norm": 0.9508203268051147,
+      "learning_rate": 4.225859883654776e-05,
+      "loss": 1.6941,
+      "step": 8990
+    },
+    {
+      "epoch": 0.22599959915823228,
+      "grad_norm": 0.9774237275123596,
+      "learning_rate": 4.219774185874569e-05,
+      "loss": 1.6996,
+      "step": 9021
+    },
+    {
+      "epoch": 0.22677623008317466,
+      "grad_norm": 0.9906953573226929,
+      "learning_rate": 4.213669080676418e-05,
+      "loss": 1.7186,
+      "step": 9052
+    },
+    {
+      "epoch": 0.22755286100811706,
+      "grad_norm": 0.966076135635376,
+      "learning_rate": 4.2075446369556056e-05,
+      "loss": 1.6828,
+      "step": 9083
+    },
+    {
+      "epoch": 0.22832949193305943,
+      "grad_norm": 0.9266401529312134,
+      "learning_rate": 4.201400923825648e-05,
+      "loss": 1.6969,
+      "step": 9114
+    },
+    {
+      "epoch": 0.2291061228580018,
+      "grad_norm": 0.8797683119773865,
+      "learning_rate": 4.195238010617511e-05,
+      "loss": 1.6835,
+      "step": 9145
+    },
+    {
+      "epoch": 0.22988275378294418,
+      "grad_norm": 0.92194664478302,
+      "learning_rate": 4.1890559668788344e-05,
+      "loss": 1.7028,
+      "step": 9176
+    },
+    {
+      "epoch": 0.23065938470788655,
+      "grad_norm": 0.9278755187988281,
+      "learning_rate": 4.1828548623731405e-05,
+      "loss": 1.6853,
+      "step": 9207
+    },
+    {
+      "epoch": 0.23143601563282895,
+      "grad_norm": 0.9345482587814331,
+      "learning_rate": 4.1766347670790506e-05,
+      "loss": 1.6879,
+      "step": 9238
+    },
+    {
+      "epoch": 0.23221264655777132,
+      "grad_norm": 0.9217596054077148,
+      "learning_rate": 4.170395751189495e-05,
+      "loss": 1.679,
+      "step": 9269
+    },
+    {
+      "epoch": 0.2329892774827137,
+      "grad_norm": 0.9330849647521973,
+      "learning_rate": 4.164137885110921e-05,
+      "loss": 1.6891,
+      "step": 9300
+    },
+    {
+      "epoch": 0.23376590840765607,
+      "grad_norm": 0.923973023891449,
+      "learning_rate": 4.157861239462495e-05,
+      "loss": 1.6929,
+      "step": 9331
+    },
+    {
+      "epoch": 0.23454253933259844,
+      "grad_norm": 0.9245807528495789,
+      "learning_rate": 4.1515658850753114e-05,
+      "loss": 1.7162,
+      "step": 9362
+    },
+    {
+      "epoch": 0.23531917025754084,
+      "grad_norm": 0.9212767481803894,
+      "learning_rate": 4.145251892991588e-05,
+      "loss": 1.7169,
+      "step": 9393
+    },
+    {
+      "epoch": 0.23609580118248322,
+      "grad_norm": 0.9439975023269653,
+      "learning_rate": 4.138919334463868e-05,
+      "loss": 1.6869,
+      "step": 9424
+    },
+    {
+      "epoch": 0.2368724321074256,
+      "grad_norm": 0.9876207709312439,
+      "learning_rate": 4.1325682809542124e-05,
+      "loss": 1.6989,
+      "step": 9455
+    },
+    {
+      "epoch": 0.23764906303236796,
+      "grad_norm": 0.8833792805671692,
+      "learning_rate": 4.126198804133398e-05,
+      "loss": 1.6917,
+      "step": 9486
+    },
+    {
+      "epoch": 0.23842569395731036,
+      "grad_norm": 0.9133450388908386,
+      "learning_rate": 4.1198109758801055e-05,
+      "loss": 1.7038,
+      "step": 9517
+    },
+    {
+      "epoch": 0.23920232488225274,
+      "grad_norm": 0.8729329705238342,
+      "learning_rate": 4.113404868280107e-05,
+      "loss": 1.672,
+      "step": 9548
+    },
+    {
+      "epoch": 0.2399789558071951,
+      "grad_norm": 0.892375111579895,
+      "learning_rate": 4.106980553625457e-05,
+      "loss": 1.6829,
+      "step": 9579
+    },
+    {
+      "epoch": 0.24075558673213748,
+      "grad_norm": 0.8939517140388489,
+      "learning_rate": 4.100538104413674e-05,
+      "loss": 1.6965,
+      "step": 9610
+    },
+    {
+      "epoch": 0.24153221765707986,
+      "grad_norm": 0.8656035661697388,
+      "learning_rate": 4.09407759334692e-05,
+      "loss": 1.7063,
+      "step": 9641
+    },
+    {
+      "epoch": 0.24230884858202226,
+      "grad_norm": 0.8904181122779846,
+      "learning_rate": 4.087599093331186e-05,
+      "loss": 1.6719,
+      "step": 9672
+    },
+    {
+      "epoch": 0.24308547950696463,
+      "grad_norm": 0.9176281690597534,
+      "learning_rate": 4.081102677475462e-05,
+      "loss": 1.6648,
+      "step": 9703
+    },
+    {
+      "epoch": 0.243862110431907,
+      "grad_norm": 0.9038643836975098,
+      "learning_rate": 4.0745884190909194e-05,
+      "loss": 1.6756,
+      "step": 9734
+    },
+    {
+      "epoch": 0.24463874135684938,
+      "grad_norm": 0.9255871772766113,
+      "learning_rate": 4.0680563916900796e-05,
+      "loss": 1.6752,
+      "step": 9765
+    },
+    {
+      "epoch": 0.24541537228179175,
+      "grad_norm": 0.9124768972396851,
+      "learning_rate": 4.0615066689859815e-05,
+      "loss": 1.6982,
+      "step": 9796
+    },
+    {
+      "epoch": 0.24619200320673415,
+      "grad_norm": 0.9149096608161926,
+      "learning_rate": 4.0549393248913584e-05,
+      "loss": 1.6692,
+      "step": 9827
+    },
+    {
+      "epoch": 0.24696863413167652,
+      "grad_norm": 0.9468510150909424,
+      "learning_rate": 4.048354433517794e-05,
+      "loss": 1.6775,
+      "step": 9858
+    },
+    {
+      "epoch": 0.2477452650566189,
+      "grad_norm": 1.2210197448730469,
+      "learning_rate": 4.0417520691748916e-05,
+      "loss": 1.692,
+      "step": 9889
+    },
+    {
+      "epoch": 0.24852189598156127,
+      "grad_norm": 0.9717695116996765,
+      "learning_rate": 4.035132306369438e-05,
+      "loss": 1.6583,
+      "step": 9920
+    },
+    {
+      "epoch": 0.24929852690650367,
+      "grad_norm": 0.9524528980255127,
+      "learning_rate": 4.028495219804555e-05,
+      "loss": 1.6606,
+      "step": 9951
+    },
+    {
+      "epoch": 0.250075157831446,
+      "grad_norm": 0.937414824962616,
+      "learning_rate": 4.021840884378864e-05,
+      "loss": 1.7063,
+      "step": 9982
+    },
+    {
+      "epoch": 0.2508517887563884,
+      "grad_norm": 0.9392083883285522,
+      "learning_rate": 4.015169375185633e-05,
+      "loss": 1.6813,
+      "step": 10013
+    },
+    {
+      "epoch": 0.2516284196813308,
+      "grad_norm": 0.9055363535881042,
+      "learning_rate": 4.0084807675119396e-05,
+      "loss": 1.6622,
+      "step": 10044
+    },
+    {
+      "epoch": 0.25240505060627316,
+      "grad_norm": 0.8967134356498718,
+      "learning_rate": 4.0017751368378106e-05,
+      "loss": 1.7052,
+      "step": 10075
+    },
+    {
+      "epoch": 0.25318168153121556,
+      "grad_norm": 0.9304991364479065,
+      "learning_rate": 3.995052558835377e-05,
+      "loss": 1.6779,
+      "step": 10106
+    },
+    {
+      "epoch": 0.2539583124561579,
+      "grad_norm": 0.9946010708808899,
+      "learning_rate": 3.988313109368017e-05,
+      "loss": 1.6727,
+      "step": 10137
+    },
+    {
+      "epoch": 0.2547349433811003,
+      "grad_norm": 0.936072051525116,
+      "learning_rate": 3.981556864489504e-05,
+      "loss": 1.6861,
+      "step": 10168
+    },
+    {
+      "epoch": 0.2555115743060427,
+      "grad_norm": 0.9157513380050659,
+      "learning_rate": 3.974783900443142e-05,
+      "loss": 1.6871,
+      "step": 10199
+    },
+    {
+      "epoch": 0.25628820523098506,
+      "grad_norm": 0.8980918526649475,
+      "learning_rate": 3.9679942936609095e-05,
+      "loss": 1.674,
+      "step": 10230
+    },
+    {
+      "epoch": 0.25706483615592746,
+      "grad_norm": 0.9256001114845276,
+      "learning_rate": 3.961188120762596e-05,
+      "loss": 1.6858,
+      "step": 10261
+    },
+    {
+      "epoch": 0.2578414670808698,
+      "grad_norm": 0.8827298879623413,
+      "learning_rate": 3.954365458554938e-05,
+      "loss": 1.6734,
+      "step": 10292
+    },
+    {
+      "epoch": 0.2586180980058122,
+      "grad_norm": 0.9297879338264465,
+      "learning_rate": 3.947526384030751e-05,
+      "loss": 1.6648,
+      "step": 10323
+    },
+    {
+      "epoch": 0.2593947289307546,
+      "grad_norm": 0.91744065284729,
+      "learning_rate": 3.9406709743680624e-05,
+      "loss": 1.6713,
+      "step": 10354
+    },
+    {
+      "epoch": 0.26017135985569695,
+      "grad_norm": 0.949213445186615,
+      "learning_rate": 3.9337993069292366e-05,
+      "loss": 1.6633,
+      "step": 10385
+    },
+    {
+      "epoch": 0.26094799078063935,
+      "grad_norm": 0.9231769442558289,
+      "learning_rate": 3.926911459260109e-05,
+      "loss": 1.6815,
+      "step": 10416
+    },
+    {
+      "epoch": 0.2617246217055817,
+      "grad_norm": 0.9055944681167603,
+      "learning_rate": 3.920007509089102e-05,
+      "loss": 1.6621,
+      "step": 10447
+    },
+    {
+      "epoch": 0.2625012526305241,
+      "grad_norm": 1.7065776586532593,
+      "learning_rate": 3.913087534326357e-05,
+      "loss": 1.6832,
+      "step": 10478
+    },
+    {
+      "epoch": 0.2632778835554665,
+      "grad_norm": 0.9098711013793945,
+      "learning_rate": 3.9061516130628475e-05,
+      "loss": 1.7106,
+      "step": 10509
+    },
+    {
+      "epoch": 0.26405451448040884,
+      "grad_norm": 0.9403277635574341,
+      "learning_rate": 3.8991998235695025e-05,
+      "loss": 1.708,
+      "step": 10540
+    },
+    {
+      "epoch": 0.26483114540535124,
+      "grad_norm": 0.9076827168464661,
+      "learning_rate": 3.8922322442963224e-05,
+      "loss": 1.6571,
+      "step": 10571
+    },
+    {
+      "epoch": 0.26560777633029364,
+      "grad_norm": 0.9140092730522156,
+      "learning_rate": 3.885248953871491e-05,
+      "loss": 1.6626,
+      "step": 10602
+    },
+    {
+      "epoch": 0.266384407255236,
+      "grad_norm": 0.9277907013893127,
+      "learning_rate": 3.8782500311004915e-05,
+      "loss": 1.6625,
+      "step": 10633
+    },
+    {
+      "epoch": 0.2671610381801784,
+      "grad_norm": 0.9199085831642151,
+      "learning_rate": 3.871235554965218e-05,
+      "loss": 1.663,
+      "step": 10664
+    },
+    {
+      "epoch": 0.26793766910512073,
+      "grad_norm": 0.9358407855033875,
+      "learning_rate": 3.864205604623078e-05,
+      "loss": 1.6743,
+      "step": 10695
+    },
+    {
+      "epoch": 0.26871430003006314,
+      "grad_norm": 0.8855702877044678,
+      "learning_rate": 3.857160259406107e-05,
+      "loss": 1.6619,
+      "step": 10726
+    },
+    {
+      "epoch": 0.26949093095500554,
+      "grad_norm": 0.9157874584197998,
+      "learning_rate": 3.8500995988200674e-05,
+      "loss": 1.6772,
+      "step": 10757
+    },
+    {
+      "epoch": 0.2702675618799479,
+      "grad_norm": 0.9023963809013367,
+      "learning_rate": 3.843023702543556e-05,
+      "loss": 1.6559,
+      "step": 10788
+    },
+    {
+      "epoch": 0.2710441928048903,
+      "grad_norm": 0.9174203872680664,
+      "learning_rate": 3.8359326504270984e-05,
+      "loss": 1.6393,
+      "step": 10819
+    },
+    {
+      "epoch": 0.27182082372983263,
+      "grad_norm": 0.9019688963890076,
+      "learning_rate": 3.828826522492255e-05,
+      "loss": 1.6763,
+      "step": 10850
+    },
+    {
+      "epoch": 0.27259745465477503,
+      "grad_norm": 0.8955567479133606,
+      "learning_rate": 3.821705398930713e-05,
+      "loss": 1.6679,
+      "step": 10881
+    },
+    {
+      "epoch": 0.27337408557971743,
+      "grad_norm": 0.9326659440994263,
+      "learning_rate": 3.814569360103385e-05,
+      "loss": 1.6765,
+      "step": 10912
+    },
+    {
+      "epoch": 0.2741507165046598,
+      "grad_norm": 0.8885530233383179,
+      "learning_rate": 3.807418486539499e-05,
+      "loss": 1.674,
+      "step": 10943
+    },
+    {
+      "epoch": 0.2749273474296022,
+      "grad_norm": 0.909267783164978,
+      "learning_rate": 3.80025285893569e-05,
+      "loss": 1.6602,
+      "step": 10974
+    },
+    {
+      "epoch": 0.2757039783545445,
+      "grad_norm": 0.9005350470542908,
+      "learning_rate": 3.793072558155093e-05,
+      "loss": 1.6628,
+      "step": 11005
+    },
+    {
+      "epoch": 0.2764806092794869,
+      "grad_norm": 0.919683575630188,
+      "learning_rate": 3.785877665226426e-05,
+      "loss": 1.6647,
+      "step": 11036
+    },
+    {
+      "epoch": 0.2772572402044293,
+      "grad_norm": 0.8954905867576599,
+      "learning_rate": 3.778668261343079e-05,
+      "loss": 1.6763,
+      "step": 11067
+    },
+    {
+      "epoch": 0.27803387112937167,
+      "grad_norm": 0.9120632410049438,
+      "learning_rate": 3.771444427862192e-05,
+      "loss": 1.6552,
+      "step": 11098
+    },
+    {
+      "epoch": 0.27881050205431407,
+      "grad_norm": 0.9041403532028198,
+      "learning_rate": 3.7642062463037465e-05,
+      "loss": 1.6604,
+      "step": 11129
+    },
+    {
+      "epoch": 0.2795871329792564,
+      "grad_norm": 0.9149596095085144,
+      "learning_rate": 3.7569537983496373e-05,
+      "loss": 1.6666,
+      "step": 11160
+    },
+    {
+      "epoch": 0.2803637639041988,
+      "grad_norm": 0.9379541277885437,
+      "learning_rate": 3.749687165842753e-05,
+      "loss": 1.654,
+      "step": 11191
+    },
+    {
+      "epoch": 0.2811403948291412,
+      "grad_norm": 0.8942285180091858,
+      "learning_rate": 3.7424064307860536e-05,
+      "loss": 1.6558,
+      "step": 11222
+    },
+    {
+      "epoch": 0.28191702575408356,
+      "grad_norm": 0.9407718181610107,
+      "learning_rate": 3.735111675341645e-05,
+      "loss": 1.6574,
+      "step": 11253
+    },
+    {
+      "epoch": 0.28269365667902596,
+      "grad_norm": 0.9167685508728027,
+      "learning_rate": 3.7278029818298524e-05,
+      "loss": 1.6767,
+      "step": 11284
+    },
+    {
+      "epoch": 0.2834702876039683,
+      "grad_norm": 0.8957439661026001,
+      "learning_rate": 3.720480432728287e-05,
+      "loss": 1.6846,
+      "step": 11315
+    },
+    {
+      "epoch": 0.2842469185289107,
+      "grad_norm": 0.9044497609138489,
+      "learning_rate": 3.71314411067092e-05,
+      "loss": 1.6323,
+      "step": 11346
+    },
+    {
+      "epoch": 0.2850235494538531,
+      "grad_norm": 0.9024799466133118,
+      "learning_rate": 3.70579409844715e-05,
+      "loss": 1.682,
+      "step": 11377
+    },
+    {
+      "epoch": 0.28580018037879545,
+      "grad_norm": 0.9126555323600769,
+      "learning_rate": 3.698430479000865e-05,
+      "loss": 1.6802,
+      "step": 11408
+    },
+    {
+      "epoch": 0.28657681130373786,
+      "grad_norm": 0.924310564994812,
+      "learning_rate": 3.691053335429509e-05,
+      "loss": 1.6568,
+      "step": 11439
+    },
+    {
+      "epoch": 0.28735344222868026,
+      "grad_norm": 0.8566572666168213,
+      "learning_rate": 3.683662750983147e-05,
+      "loss": 1.6682,
+      "step": 11470
+    },
+    {
+      "epoch": 0.2881300731536226,
+      "grad_norm": 0.8287222981452942,
+      "learning_rate": 3.676258809063518e-05,
+      "loss": 1.6484,
+      "step": 11501
+    },
+    {
+      "epoch": 0.288906704078565,
+      "grad_norm": 1.0163201093673706,
+      "learning_rate": 3.6688415932231004e-05,
+      "loss": 1.635,
+      "step": 11532
+    },
+    {
+      "epoch": 0.28968333500350735,
+      "grad_norm": 0.8813551068305969,
+      "learning_rate": 3.661411187164166e-05,
+      "loss": 1.6654,
+      "step": 11563
+    },
+    {
+      "epoch": 0.29045996592844975,
+      "grad_norm": 0.8688541650772095,
+      "learning_rate": 3.65396767473784e-05,
+      "loss": 1.6787,
+      "step": 11594
+    },
+    {
+      "epoch": 0.29123659685339215,
+      "grad_norm": 0.8820333480834961,
+      "learning_rate": 3.6465111399431465e-05,
+      "loss": 1.6418,
+      "step": 11625
+    },
+    {
+      "epoch": 0.2920132277783345,
+      "grad_norm": 0.9284527897834778,
+      "learning_rate": 3.6390416669260674e-05,
+      "loss": 1.6727,
+      "step": 11656
+    },
+    {
+      "epoch": 0.2927898587032769,
+      "grad_norm": 0.8665077090263367,
+      "learning_rate": 3.63155933997859e-05,
+      "loss": 1.6377,
+      "step": 11687
+    },
+    {
+      "epoch": 0.29356648962821924,
+      "grad_norm": 0.9927760362625122,
+      "learning_rate": 3.624064243537758e-05,
+      "loss": 1.6628,
+      "step": 11718
+    },
+    {
+      "epoch": 0.29434312055316164,
+      "grad_norm": 0.9006340503692627,
+      "learning_rate": 3.616556462184716e-05,
+      "loss": 1.6589,
+      "step": 11749
+    },
+    {
+      "epoch": 0.29511975147810404,
+      "grad_norm": 0.9278168082237244,
+      "learning_rate": 3.609036080643755e-05,
+      "loss": 1.6519,
+      "step": 11780
+    },
+    {
+      "epoch": 0.2958963824030464,
+      "grad_norm": 0.9146036505699158,
+      "learning_rate": 3.60150318378136e-05,
+      "loss": 1.6541,
+      "step": 11811
+    },
+    {
+      "epoch": 0.2966730133279888,
+      "grad_norm": 0.8698804378509521,
+      "learning_rate": 3.5939578566052465e-05,
+      "loss": 1.6722,
+      "step": 11842
+    },
+    {
+      "epoch": 0.29744964425293113,
+      "grad_norm": 0.9304215312004089,
+      "learning_rate": 3.586400184263408e-05,
+      "loss": 1.6392,
+      "step": 11873
+    },
+    {
+      "epoch": 0.29822627517787353,
+      "grad_norm": 0.8917405605316162,
+      "learning_rate": 3.578830252043148e-05,
+      "loss": 1.6373,
+      "step": 11904
+    },
+    {
+      "epoch": 0.29900290610281594,
+      "grad_norm": 0.9075546860694885,
+      "learning_rate": 3.571248145370125e-05,
+      "loss": 1.6401,
+      "step": 11935
+    },
+    {
+      "epoch": 0.2997795370277583,
+      "grad_norm": 0.9235963225364685,
+      "learning_rate": 3.5636539498073794e-05,
+      "loss": 1.667,
+      "step": 11966
+    },
+    {
+      "epoch": 0.3005561679527007,
+      "grad_norm": 0.9514668583869934,
+      "learning_rate": 3.556047751054378e-05,
+      "loss": 1.6654,
+      "step": 11997
+    },
+    {
+      "epoch": 0.301332798877643,
+      "grad_norm": 0.8843172192573547,
+      "learning_rate": 3.548429634946039e-05,
+      "loss": 1.6609,
+      "step": 12028
+    },
+    {
+      "epoch": 0.3021094298025854,
+      "grad_norm": 0.9325070977210999,
+      "learning_rate": 3.540799687451768e-05,
+      "loss": 1.6392,
+      "step": 12059
+    },
+    {
+      "epoch": 0.30288606072752783,
+      "grad_norm": 0.8855581879615784,
+      "learning_rate": 3.533157994674485e-05,
+      "loss": 1.6607,
+      "step": 12090
+    },
+    {
+      "epoch": 0.3036626916524702,
+      "grad_norm": 0.9044367671012878,
+      "learning_rate": 3.5255046428496546e-05,
+      "loss": 1.6417,
+      "step": 12121
+    },
+    {
+      "epoch": 0.3044393225774126,
+      "grad_norm": 0.9446975588798523,
+      "learning_rate": 3.517839718344311e-05,
+      "loss": 1.6279,
+      "step": 12152
+    },
+    {
+      "epoch": 0.3052159535023549,
+      "grad_norm": 0.8764945864677429,
+      "learning_rate": 3.510163307656086e-05,
+      "loss": 1.6297,
+      "step": 12183
+    },
+    {
+      "epoch": 0.3059925844272973,
+      "grad_norm": 0.8633154034614563,
+      "learning_rate": 3.5024754974122324e-05,
+      "loss": 1.648,
+      "step": 12214
+    },
+    {
+      "epoch": 0.3067692153522397,
+      "grad_norm": 0.8891461491584778,
+      "learning_rate": 3.494776374368643e-05,
+      "loss": 1.6492,
+      "step": 12245
+    },
+    {
+      "epoch": 0.30754584627718207,
+      "grad_norm": 0.8888112306594849,
+      "learning_rate": 3.4870660254088724e-05,
+      "loss": 1.6507,
+      "step": 12276
+    },
+    {
+      "epoch": 0.30832247720212447,
+      "grad_norm": 0.8980075120925903,
+      "learning_rate": 3.479344537543164e-05,
+      "loss": 1.6571,
+      "step": 12307
+    },
+    {
+      "epoch": 0.3090991081270668,
+      "grad_norm": 0.86537104845047,
+      "learning_rate": 3.4716119979074565e-05,
+      "loss": 1.6331,
+      "step": 12338
+    },
+    {
+      "epoch": 0.3098757390520092,
+      "grad_norm": 0.9001527428627014,
+      "learning_rate": 3.463868493762412e-05,
+      "loss": 1.6453,
+      "step": 12369
+    },
+    {
+      "epoch": 0.3106523699769516,
+      "grad_norm": 0.8868582248687744,
+      "learning_rate": 3.456114112492418e-05,
+      "loss": 1.6623,
+      "step": 12400
+    },
+    {
+      "epoch": 0.31142900090189396,
+      "grad_norm": 0.899174690246582,
+      "learning_rate": 3.4483489416046164e-05,
+      "loss": 1.6451,
+      "step": 12431
+    },
+    {
+      "epoch": 0.31220563182683636,
+      "grad_norm": 0.9003099203109741,
+      "learning_rate": 3.440573068727905e-05,
+      "loss": 1.6544,
+      "step": 12462
+    },
+    {
+      "epoch": 0.31298226275177876,
+      "grad_norm": 0.8912012577056885,
+      "learning_rate": 3.4327865816119495e-05,
+      "loss": 1.6579,
+      "step": 12493
+    },
+    {
+      "epoch": 0.3137588936767211,
+      "grad_norm": 0.8915206789970398,
+      "learning_rate": 3.4249895681262025e-05,
+      "loss": 1.65,
+      "step": 12524
+    },
+    {
+      "epoch": 0.3145355246016635,
+      "grad_norm": 0.9975365400314331,
+      "learning_rate": 3.417182116258899e-05,
+      "loss": 1.6506,
+      "step": 12555
+    },
+    {
+      "epoch": 0.31531215552660585,
+      "grad_norm": 0.8904858827590942,
+      "learning_rate": 3.409364314116074e-05,
+      "loss": 1.6402,
+      "step": 12586
+    },
+    {
+      "epoch": 0.31608878645154825,
+      "grad_norm": 0.8596284985542297,
+      "learning_rate": 3.401536249920559e-05,
+      "loss": 1.6473,
+      "step": 12617
+    },
+    {
+      "epoch": 0.31686541737649065,
+      "grad_norm": 0.8970181345939636,
+      "learning_rate": 3.393698012010998e-05,
+      "loss": 1.6347,
+      "step": 12648
+    },
+    {
+      "epoch": 0.317642048301433,
+      "grad_norm": 0.890839695930481,
+      "learning_rate": 3.385849688840839e-05,
+      "loss": 1.6323,
+      "step": 12679
+    },
+    {
+      "epoch": 0.3184186792263754,
+      "grad_norm": 0.914010226726532,
+      "learning_rate": 3.3779913689773414e-05,
+      "loss": 1.6396,
+      "step": 12710
+    },
+    {
+      "epoch": 0.31919531015131775,
+      "grad_norm": 0.8762277960777283,
+      "learning_rate": 3.370123141100578e-05,
+      "loss": 1.6057,
+      "step": 12741
+    },
+    {
+      "epoch": 0.31997194107626015,
+      "grad_norm": 0.9001442193984985,
+      "learning_rate": 3.3622450940024305e-05,
+      "loss": 1.647,
+      "step": 12772
+    },
+    {
+      "epoch": 0.32074857200120255,
+      "grad_norm": 0.8792625069618225,
+      "learning_rate": 3.35435731658559e-05,
+      "loss": 1.6441,
+      "step": 12803
+    },
+    {
+      "epoch": 0.3215252029261449,
+      "grad_norm": 0.889738142490387,
+      "learning_rate": 3.346459897862552e-05,
+      "loss": 1.6433,
+      "step": 12834
+    },
+    {
+      "epoch": 0.3223018338510873,
+      "grad_norm": 0.9831507205963135,
+      "learning_rate": 3.338552926954613e-05,
+      "loss": 1.6635,
+      "step": 12865
+    },
+    {
+      "epoch": 0.32307846477602964,
+      "grad_norm": 0.8929276466369629,
+      "learning_rate": 3.330636493090868e-05,
+      "loss": 1.6439,
+      "step": 12896
+    },
+    {
+      "epoch": 0.32385509570097204,
+      "grad_norm": 0.8983376026153564,
+      "learning_rate": 3.322710685607193e-05,
+      "loss": 1.641,
+      "step": 12927
+    },
+    {
+      "epoch": 0.32463172662591444,
+      "grad_norm": 0.8629191517829895,
+      "learning_rate": 3.314775593945251e-05,
+      "loss": 1.6336,
+      "step": 12958
+    },
+    {
+      "epoch": 0.3254083575508568,
+      "grad_norm": 0.9323459267616272,
+      "learning_rate": 3.3068313076514714e-05,
+      "loss": 1.6304,
+      "step": 12989
+    },
+    {
+      "epoch": 0.3261849884757992,
+      "grad_norm": 0.8520583510398865,
+      "learning_rate": 3.298877916376047e-05,
+      "loss": 1.6277,
+      "step": 13020
+    },
+    {
+      "epoch": 0.32696161940074153,
+      "grad_norm": 0.8893365263938904,
+      "learning_rate": 3.290915509871915e-05,
+      "loss": 1.6508,
+      "step": 13051
+    },
+    {
+      "epoch": 0.32773825032568393,
+      "grad_norm": 0.8612290024757385,
+      "learning_rate": 3.282944177993753e-05,
+      "loss": 1.6319,
+      "step": 13082
+    },
+    {
+      "epoch": 0.32851488125062633,
+      "grad_norm": 0.9378547072410583,
+      "learning_rate": 3.274964010696957e-05,
+      "loss": 1.637,
+      "step": 13113
+    },
+    {
+      "epoch": 0.3292915121755687,
+      "grad_norm": 0.8851338028907776,
+      "learning_rate": 3.266975098036629e-05,
+      "loss": 1.6326,
+      "step": 13144
+    },
+    {
+      "epoch": 0.3300681431005111,
+      "grad_norm": 0.9052163362503052,
+      "learning_rate": 3.258977530166562e-05,
+      "loss": 1.6507,
+      "step": 13175
+    },
+    {
+      "epoch": 0.3308447740254534,
+      "grad_norm": 0.9593476057052612,
+      "learning_rate": 3.250971397338227e-05,
+      "loss": 1.6497,
+      "step": 13206
+    },
+    {
+      "epoch": 0.3316214049503958,
+      "grad_norm": 0.8861746788024902,
+      "learning_rate": 3.2429567898997404e-05,
+      "loss": 1.6487,
+      "step": 13237
+    },
+    {
+      "epoch": 0.3323980358753382,
+      "grad_norm": 0.8835756778717041,
+      "learning_rate": 3.234933798294859e-05,
+      "loss": 1.6328,
+      "step": 13268
+    },
+    {
+      "epoch": 0.33317466680028057,
+      "grad_norm": 0.9133350849151611,
+      "learning_rate": 3.2269025130619535e-05,
+      "loss": 1.6326,
+      "step": 13299
+    },
+    {
+      "epoch": 0.333951297725223,
+      "grad_norm": 0.8866952657699585,
+      "learning_rate": 3.218863024832985e-05,
+      "loss": 1.6418,
+      "step": 13330
+    },
+    {
+      "epoch": 0.3347279286501654,
+      "grad_norm": 0.8961243033409119,
+      "learning_rate": 3.2108154243324864e-05,
+      "loss": 1.6436,
+      "step": 13361
+    },
+    {
+      "epoch": 0.3355045595751077,
+      "grad_norm": 0.871108889579773,
+      "learning_rate": 3.2027598023765345e-05,
+      "loss": 1.6205,
+      "step": 13392
+    },
+    {
+      "epoch": 0.3362811905000501,
+      "grad_norm": 0.8666812777519226,
+      "learning_rate": 3.194696249871729e-05,
+      "loss": 1.6478,
+      "step": 13423
+    },
+    {
+      "epoch": 0.33705782142499247,
+      "grad_norm": 0.8760045766830444,
+      "learning_rate": 3.186624857814164e-05,
+      "loss": 1.63,
+      "step": 13454
+    },
+    {
+      "epoch": 0.33783445234993487,
+      "grad_norm": 0.9416362047195435,
+      "learning_rate": 3.178545717288401e-05,
+      "loss": 1.6268,
+      "step": 13485
+    },
+    {
+      "epoch": 0.33861108327487727,
+      "grad_norm": 0.8591843843460083,
+      "learning_rate": 3.170458919466444e-05,
+      "loss": 1.6206,
+      "step": 13516
+    },
+    {
+      "epoch": 0.3393877141998196,
+      "grad_norm": 0.9452426433563232,
+      "learning_rate": 3.1623645556067063e-05,
+      "loss": 1.6351,
+      "step": 13547
+    },
+    {
+      "epoch": 0.340164345124762,
+      "grad_norm": 0.8680084943771362,
+      "learning_rate": 3.154262717052985e-05,
+      "loss": 1.6389,
+      "step": 13578
+    },
+    {
+      "epoch": 0.34094097604970436,
+      "grad_norm": 0.8631142973899841,
+      "learning_rate": 3.146153495233426e-05,
+      "loss": 1.6198,
+      "step": 13609
+    },
+    {
+      "epoch": 0.34171760697464676,
+      "grad_norm": 0.905168354511261,
+      "learning_rate": 3.1380369816594944e-05,
+      "loss": 1.6395,
+      "step": 13640
+    },
+    {
+      "epoch": 0.34249423789958916,
+      "grad_norm": 0.8644616007804871,
+      "learning_rate": 3.129913267924946e-05,
+      "loss": 1.6282,
+      "step": 13671
+    },
+    {
+      "epoch": 0.3432708688245315,
+      "grad_norm": 0.8989911675453186,
+      "learning_rate": 3.121782445704782e-05,
+      "loss": 1.6378,
+      "step": 13702
+    },
+    {
+      "epoch": 0.3440474997494739,
+      "grad_norm": 0.908371090888977,
+      "learning_rate": 3.11364460675423e-05,
+      "loss": 1.6152,
+      "step": 13733
+    },
+    {
+      "epoch": 0.34482413067441625,
+      "grad_norm": 0.876471221446991,
+      "learning_rate": 3.1054998429076934e-05,
+      "loss": 1.6533,
+      "step": 13764
+    },
+    {
+      "epoch": 0.34560076159935865,
+      "grad_norm": 0.879152238368988,
+      "learning_rate": 3.097348246077728e-05,
+      "loss": 1.6196,
+      "step": 13795
+    },
+    {
+      "epoch": 0.34637739252430105,
+      "grad_norm": 0.8737320303916931,
+      "learning_rate": 3.0891899082539924e-05,
+      "loss": 1.6122,
+      "step": 13826
+    },
+    {
+      "epoch": 0.3471540234492434,
+      "grad_norm": 0.894914448261261,
+      "learning_rate": 3.0810249215022233e-05,
+      "loss": 1.6231,
+      "step": 13857
+    },
+    {
+      "epoch": 0.3479306543741858,
+      "grad_norm": 0.8699216842651367,
+      "learning_rate": 3.0728533779631865e-05,
+      "loss": 1.6374,
+      "step": 13888
+    },
+    {
+      "epoch": 0.34870728529912814,
+      "grad_norm": 0.8811262249946594,
+      "learning_rate": 3.064675369851637e-05,
+      "loss": 1.635,
+      "step": 13919
+    },
+    {
+      "epoch": 0.34948391622407055,
+      "grad_norm": 0.9017398953437805,
+      "learning_rate": 3.056490989455289e-05,
+      "loss": 1.6377,
+      "step": 13950
+    },
+    {
+      "epoch": 0.35026054714901295,
+      "grad_norm": 0.8944412469863892,
+      "learning_rate": 3.0483003291337596e-05,
+      "loss": 1.6401,
+      "step": 13981
+    },
+    {
+      "epoch": 0.3510371780739553,
+      "grad_norm": 0.9429489970207214,
+      "learning_rate": 3.040103481317539e-05,
+      "loss": 1.6313,
+      "step": 14012
+    },
+    {
+      "epoch": 0.3518138089988977,
+      "grad_norm": 0.8750442862510681,
+      "learning_rate": 3.03190053850694e-05,
+      "loss": 1.6305,
+      "step": 14043
+    },
+    {
+      "epoch": 0.35259043992384004,
+      "grad_norm": 0.8219168782234192,
+      "learning_rate": 3.0236915932710573e-05,
+      "loss": 1.6289,
+      "step": 14074
+    },
+    {
+      "epoch": 0.35336707084878244,
+      "grad_norm": 0.9778533577919006,
+      "learning_rate": 3.0154767382467232e-05,
+      "loss": 1.6221,
+      "step": 14105
+    },
+    {
+      "epoch": 0.35414370177372484,
+      "grad_norm": 0.8655532598495483,
+      "learning_rate": 3.0072560661374582e-05,
+      "loss": 1.6352,
+      "step": 14136
+    },
+    {
+      "epoch": 0.3549203326986672,
+      "grad_norm": 0.8914876580238342,
+      "learning_rate": 2.999029669712431e-05,
+      "loss": 1.6283,
+      "step": 14167
+    },
+    {
+      "epoch": 0.3556969636236096,
+      "grad_norm": 0.9823268055915833,
+      "learning_rate": 2.990797641805408e-05,
+      "loss": 1.6037,
+      "step": 14198
+    },
+    {
+      "epoch": 0.356473594548552,
+      "grad_norm": 0.8770225048065186,
+      "learning_rate": 2.982560075313704e-05,
+      "loss": 1.6199,
+      "step": 14229
+    },
+    {
+      "epoch": 0.35725022547349433,
+      "grad_norm": 0.8294642567634583,
+      "learning_rate": 2.9743170631971368e-05,
+      "loss": 1.587,
+      "step": 14260
+    },
+    {
+      "epoch": 0.35802685639843673,
+      "grad_norm": 0.8678600192070007,
+      "learning_rate": 2.9660686984769792e-05,
+      "loss": 1.6006,
+      "step": 14291
+    },
+    {
+      "epoch": 0.3588034873233791,
+      "grad_norm": 0.8829056620597839,
+      "learning_rate": 2.9578150742349047e-05,
+      "loss": 1.6354,
+      "step": 14322
+    },
+    {
+      "epoch": 0.3595801182483215,
+      "grad_norm": 0.8785590529441833,
+      "learning_rate": 2.949556283611942e-05,
+      "loss": 1.6293,
+      "step": 14353
+    },
+    {
+      "epoch": 0.3603567491732639,
+      "grad_norm": 0.8895497918128967,
+      "learning_rate": 2.9412924198074206e-05,
+      "loss": 1.6121,
+      "step": 14384
+    },
+    {
+      "epoch": 0.3611333800982062,
+      "grad_norm": 0.8808982372283936,
+      "learning_rate": 2.9330235760779208e-05,
+      "loss": 1.6251,
+      "step": 14415
+    },
+    {
+      "epoch": 0.3619100110231486,
+      "grad_norm": 0.8854517936706543,
+      "learning_rate": 2.9247498457362188e-05,
+      "loss": 1.617,
+      "step": 14446
+    },
+    {
+      "epoch": 0.36268664194809097,
+      "grad_norm": 0.9169306755065918,
+      "learning_rate": 2.9164713221502373e-05,
+      "loss": 1.632,
+      "step": 14477
+    },
+    {
+      "epoch": 0.36346327287303337,
+      "grad_norm": 0.9230756163597107,
+      "learning_rate": 2.9081880987419912e-05,
+      "loss": 1.6227,
+      "step": 14508
+    },
+    {
+      "epoch": 0.3642399037979758,
+      "grad_norm": 0.8728978037834167,
+      "learning_rate": 2.8999002689865296e-05,
+      "loss": 1.6175,
+      "step": 14539
+    },
+    {
+      "epoch": 0.3650165347229181,
+      "grad_norm": 0.9083303213119507,
+      "learning_rate": 2.8916079264108852e-05,
+      "loss": 1.6147,
+      "step": 14570
+    },
+    {
+      "epoch": 0.3657931656478605,
+      "grad_norm": 0.8577157855033875,
+      "learning_rate": 2.883311164593017e-05,
+      "loss": 1.6357,
+      "step": 14601
+    },
+    {
+      "epoch": 0.36656979657280286,
+      "grad_norm": 0.9495333433151245,
+      "learning_rate": 2.875010077160754e-05,
+      "loss": 1.6192,
+      "step": 14632
+    },
+    {
+      "epoch": 0.36734642749774526,
+      "grad_norm": 0.9099650979042053,
+      "learning_rate": 2.866704757790741e-05,
+      "loss": 1.6417,
+      "step": 14663
+    },
+    {
+      "epoch": 0.36812305842268767,
+      "grad_norm": 0.8684884905815125,
+      "learning_rate": 2.858395300207376e-05,
+      "loss": 1.6171,
+      "step": 14694
+    },
+    {
+      "epoch": 0.36889968934763,
+      "grad_norm": 0.8524884581565857,
+      "learning_rate": 2.8500817981817607e-05,
+      "loss": 1.6153,
+      "step": 14725
+    },
+    {
+      "epoch": 0.3696763202725724,
+      "grad_norm": 0.9122047424316406,
+      "learning_rate": 2.8417643455306336e-05,
+      "loss": 1.6223,
+      "step": 14756
+    },
+    {
+      "epoch": 0.37045295119751476,
+      "grad_norm": 0.875884473323822,
+      "learning_rate": 2.8334430361153185e-05,
+      "loss": 1.6364,
+      "step": 14787
+    },
+    {
+      "epoch": 0.37122958212245716,
+      "grad_norm": 0.8689435720443726,
+      "learning_rate": 2.8251179638406612e-05,
+      "loss": 1.639,
+      "step": 14818
+    },
+    {
+      "epoch": 0.37200621304739956,
+      "grad_norm": 0.8948261141777039,
+      "learning_rate": 2.8167892226539704e-05,
+      "loss": 1.6091,
+      "step": 14849
+    },
+    {
+      "epoch": 0.3727828439723419,
+      "grad_norm": 0.8971095085144043,
+      "learning_rate": 2.8084569065439588e-05,
+      "loss": 1.6527,
+      "step": 14880
+    },
+    {
+      "epoch": 0.3735594748972843,
+      "grad_norm": 0.9517078995704651,
+      "learning_rate": 2.8001211095396807e-05,
+      "loss": 1.6201,
+      "step": 14911
+    },
+    {
+      "epoch": 0.37433610582222665,
+      "grad_norm": 0.8965676426887512,
+      "learning_rate": 2.791781925709473e-05,
+      "loss": 1.6104,
+      "step": 14942
+    },
+    {
+      "epoch": 0.37511273674716905,
+      "grad_norm": 0.9771518707275391,
+      "learning_rate": 2.7834394491598908e-05,
+      "loss": 1.6329,
+      "step": 14973
+    },
+    {
+      "epoch": 0.37588936767211145,
+      "grad_norm": 0.8746607899665833,
+      "learning_rate": 2.7750937740346485e-05,
+      "loss": 1.6135,
+      "step": 15004
+    },
+    {
+      "epoch": 0.3766659985970538,
+      "grad_norm": 0.860730767250061,
+      "learning_rate": 2.7667449945135564e-05,
+      "loss": 1.6163,
+      "step": 15035
+    },
+    {
+      "epoch": 0.3774426295219962,
+      "grad_norm": 0.8879541158676147,
+      "learning_rate": 2.7583932048114557e-05,
+      "loss": 1.6036,
+      "step": 15066
+    },
+    {
+      "epoch": 0.37821926044693854,
+      "grad_norm": 0.936347246170044,
+      "learning_rate": 2.7500384991771587e-05,
+      "loss": 1.5914,
+      "step": 15097
+    },
+    {
+      "epoch": 0.37899589137188094,
+      "grad_norm": 0.8772497773170471,
+      "learning_rate": 2.7416809718923825e-05,
+      "loss": 1.5994,
+      "step": 15128
+    },
+    {
+      "epoch": 0.37977252229682334,
+      "grad_norm": 0.8785439729690552,
+      "learning_rate": 2.7333207172706864e-05,
+      "loss": 1.6057,
+      "step": 15159
+    },
+    {
+      "epoch": 0.3805491532217657,
+      "grad_norm": 1.1660430431365967,
+      "learning_rate": 2.7249578296564088e-05,
+      "loss": 1.6257,
+      "step": 15190
+    },
+    {
+      "epoch": 0.3813257841467081,
+      "grad_norm": 0.8710660934448242,
+      "learning_rate": 2.7165924034235973e-05,
+      "loss": 1.6223,
+      "step": 15221
+    },
+    {
+      "epoch": 0.3821024150716505,
+      "grad_norm": 0.9040597081184387,
+      "learning_rate": 2.708224532974953e-05,
+      "loss": 1.6054,
+      "step": 15252
+    },
+    {
+      "epoch": 0.38287904599659284,
+      "grad_norm": 0.8942875862121582,
+      "learning_rate": 2.6998543127407538e-05,
+      "loss": 1.6307,
+      "step": 15283
+    },
+    {
+      "epoch": 0.38365567692153524,
+      "grad_norm": 0.8733716011047363,
+      "learning_rate": 2.6914818371777988e-05,
+      "loss": 1.6298,
+      "step": 15314
+    },
+    {
+      "epoch": 0.3844323078464776,
+      "grad_norm": 0.9183559417724609,
+      "learning_rate": 2.6831072007683373e-05,
+      "loss": 1.6056,
+      "step": 15345
+    },
+    {
+      "epoch": 0.38520893877142,
+      "grad_norm": 0.8594227433204651,
+      "learning_rate": 2.6747304980190018e-05,
+      "loss": 1.6092,
+      "step": 15376
+    },
+    {
+      "epoch": 0.3859855696963624,
+      "grad_norm": 0.8866741061210632,
+      "learning_rate": 2.6663518234597453e-05,
+      "loss": 1.6013,
+      "step": 15407
+    },
+    {
+      "epoch": 0.38676220062130473,
+      "grad_norm": 0.8940442204475403,
+      "learning_rate": 2.6579712716427696e-05,
+      "loss": 1.6203,
+      "step": 15438
+    },
+    {
+      "epoch": 0.38753883154624713,
+      "grad_norm": 0.8408336639404297,
+      "learning_rate": 2.6495889371414652e-05,
+      "loss": 1.5997,
+      "step": 15469
+    },
+    {
+      "epoch": 0.3883154624711895,
+      "grad_norm": 0.8949300646781921,
+      "learning_rate": 2.6412049145493367e-05,
+      "loss": 1.6062,
+      "step": 15500
+    },
+    {
+      "epoch": 0.3890920933961319,
+      "grad_norm": 0.8919069766998291,
+      "learning_rate": 2.632819298478939e-05,
+      "loss": 1.6197,
+      "step": 15531
+    },
+    {
+      "epoch": 0.3898687243210743,
+      "grad_norm": 0.9069515466690063,
+      "learning_rate": 2.6244321835608105e-05,
+      "loss": 1.6147,
+      "step": 15562
+    },
+    {
+      "epoch": 0.3906453552460166,
+      "grad_norm": 0.909631073474884,
+      "learning_rate": 2.6160436644424024e-05,
+      "loss": 1.6182,
+      "step": 15593
+    },
+    {
+      "epoch": 0.391421986170959,
+      "grad_norm": 0.8957046866416931,
+      "learning_rate": 2.6076538357870133e-05,
+      "loss": 1.5959,
+      "step": 15624
+    },
+    {
+      "epoch": 0.39219861709590137,
+      "grad_norm": 0.8709155321121216,
+      "learning_rate": 2.5992627922727196e-05,
+      "loss": 1.6211,
+      "step": 15655
+    },
+    {
+      "epoch": 0.39297524802084377,
+      "grad_norm": 0.8602601885795593,
+      "learning_rate": 2.5908706285913066e-05,
+      "loss": 1.6032,
+      "step": 15686
+    },
+    {
+      "epoch": 0.39375187894578617,
+      "grad_norm": 0.853210985660553,
+      "learning_rate": 2.5824774394472008e-05,
+      "loss": 1.5994,
+      "step": 15717
+    },
+    {
+      "epoch": 0.3945285098707285,
+      "grad_norm": 0.8873618245124817,
+      "learning_rate": 2.5740833195563996e-05,
+      "loss": 1.5949,
+      "step": 15748
+    },
+    {
+      "epoch": 0.3953051407956709,
+      "grad_norm": 0.8576481342315674,
+      "learning_rate": 2.5656883636454067e-05,
+      "loss": 1.6029,
+      "step": 15779
+    },
+    {
+      "epoch": 0.39608177172061326,
+      "grad_norm": 0.8468286395072937,
+      "learning_rate": 2.557292666450159e-05,
+      "loss": 1.6108,
+      "step": 15810
+    },
+    {
+      "epoch": 0.39685840264555566,
+      "grad_norm": 0.9147023558616638,
+      "learning_rate": 2.5488963227149566e-05,
+      "loss": 1.6175,
+      "step": 15841
+    },
+    {
+      "epoch": 0.39763503357049806,
+      "grad_norm": 0.8911499977111816,
+      "learning_rate": 2.5404994271913983e-05,
+      "loss": 1.6232,
+      "step": 15872
+    },
+    {
+      "epoch": 0.3984116644954404,
+      "grad_norm": 0.8820773363113403,
+      "learning_rate": 2.5321020746373085e-05,
+      "loss": 1.6064,
+      "step": 15903
+    },
+    {
+      "epoch": 0.3991882954203828,
+      "grad_norm": 0.8574590086936951,
+      "learning_rate": 2.52370435981567e-05,
+      "loss": 1.6107,
+      "step": 15934
+    },
+    {
+      "epoch": 0.39996492634532516,
+      "grad_norm": 0.8728383183479309,
+      "learning_rate": 2.5153063774935533e-05,
+      "loss": 1.5737,
+      "step": 15965
+    },
+    {
+      "epoch": 0.40074155727026756,
+      "grad_norm": 0.9165883660316467,
+      "learning_rate": 2.506908222441045e-05,
+      "loss": 1.601,
+      "step": 15996
+    },
+    {
+      "epoch": 0.40151818819520996,
+      "grad_norm": 0.8354183435440063,
+      "learning_rate": 2.498509989430187e-05,
+      "loss": 1.5931,
+      "step": 16027
+    },
+    {
+      "epoch": 0.4022948191201523,
+      "grad_norm": 0.881255567073822,
+      "learning_rate": 2.4901117732338958e-05,
+      "loss": 1.6065,
+      "step": 16058
+    },
+    {
+      "epoch": 0.4030714500450947,
+      "grad_norm": 0.8858495950698853,
+      "learning_rate": 2.481713668624899e-05,
+      "loss": 1.6148,
+      "step": 16089
+    },
+    {
+      "epoch": 0.4038480809700371,
+      "grad_norm": 0.8628421425819397,
+      "learning_rate": 2.4733157703746663e-05,
+      "loss": 1.6098,
+      "step": 16120
+    },
+    {
+      "epoch": 0.40462471189497945,
+      "grad_norm": 0.8520597815513611,
+      "learning_rate": 2.4649181732523392e-05,
+      "loss": 1.6127,
+      "step": 16151
+    },
+    {
+      "epoch": 0.40540134281992185,
+      "grad_norm": 0.8664495348930359,
+      "learning_rate": 2.4565209720236582e-05,
+      "loss": 1.6039,
+      "step": 16182
+    },
+    {
+      "epoch": 0.4061779737448642,
+      "grad_norm": 0.8572731614112854,
+      "learning_rate": 2.4481242614498975e-05,
+      "loss": 1.6023,
+      "step": 16213
+    },
+    {
+      "epoch": 0.4069546046698066,
+      "grad_norm": 0.8819786906242371,
+      "learning_rate": 2.439728136286796e-05,
+      "loss": 1.6105,
+      "step": 16244
+    },
+    {
+      "epoch": 0.407731235594749,
+      "grad_norm": 0.8858864903450012,
+      "learning_rate": 2.4313326912834852e-05,
+      "loss": 1.5995,
+      "step": 16275
+    },
+    {
+      "epoch": 0.40850786651969134,
+      "grad_norm": 0.8981665372848511,
+      "learning_rate": 2.4229380211814206e-05,
+      "loss": 1.6041,
+      "step": 16306
+    },
+    {
+      "epoch": 0.40928449744463374,
+      "grad_norm": 0.8813167214393616,
+      "learning_rate": 2.4145442207133124e-05,
+      "loss": 1.5976,
+      "step": 16337
+    },
+    {
+      "epoch": 0.4100611283695761,
+      "grad_norm": 0.8797750473022461,
+      "learning_rate": 2.406151384602059e-05,
+      "loss": 1.602,
+      "step": 16368
+    },
+    {
+      "epoch": 0.4108377592945185,
+      "grad_norm": 0.8421767950057983,
+      "learning_rate": 2.3977596075596747e-05,
+      "loss": 1.5781,
+      "step": 16399
+    },
+    {
+      "epoch": 0.4116143902194609,
+      "grad_norm": 0.856469452381134,
+      "learning_rate": 2.3893689842862223e-05,
+      "loss": 1.588,
+      "step": 16430
+    },
+    {
+      "epoch": 0.41239102114440324,
+      "grad_norm": 0.8485891819000244,
+      "learning_rate": 2.3809796094687475e-05,
+      "loss": 1.589,
+      "step": 16461
+    },
+    {
+      "epoch": 0.41316765206934564,
+      "grad_norm": 0.8968520760536194,
+      "learning_rate": 2.372591577780202e-05,
+      "loss": 1.6012,
+      "step": 16492
+    },
+    {
+      "epoch": 0.413944282994288,
+      "grad_norm": 0.875661313533783,
+      "learning_rate": 2.3642049838783838e-05,
+      "loss": 1.6067,
+      "step": 16523
+    },
+    {
+      "epoch": 0.4147209139192304,
+      "grad_norm": 0.9094383716583252,
+      "learning_rate": 2.3558199224048666e-05,
+      "loss": 1.5836,
+      "step": 16554
+    },
+    {
+      "epoch": 0.4154975448441728,
+      "grad_norm": 0.8707971572875977,
+      "learning_rate": 2.347436487983929e-05,
+      "loss": 1.6115,
+      "step": 16585
+    },
+    {
+      "epoch": 0.41627417576911513,
+      "grad_norm": 0.8473120331764221,
+      "learning_rate": 2.3390547752214888e-05,
+      "loss": 1.6227,
+      "step": 16616
+    },
+    {
+      "epoch": 0.41705080669405753,
+      "grad_norm": 0.9022457003593445,
+      "learning_rate": 2.330674878704035e-05,
+      "loss": 1.5926,
+      "step": 16647
+    },
+    {
+      "epoch": 0.4178274376189999,
+      "grad_norm": 0.8699711561203003,
+      "learning_rate": 2.322296892997561e-05,
+      "loss": 1.6033,
+      "step": 16678
+    },
+    {
+      "epoch": 0.4186040685439423,
+      "grad_norm": 0.8574140667915344,
+      "learning_rate": 2.313920912646497e-05,
+      "loss": 1.5861,
+      "step": 16709
+    },
+    {
+      "epoch": 0.4193806994688847,
+      "grad_norm": 0.9694622159004211,
+      "learning_rate": 2.305547032172643e-05,
+      "loss": 1.5801,
+      "step": 16740
+    },
+    {
+      "epoch": 0.420157330393827,
+      "grad_norm": 0.8770648837089539,
+      "learning_rate": 2.2971753460741014e-05,
+      "loss": 1.5877,
+      "step": 16771
+    },
+    {
+      "epoch": 0.4209339613187694,
+      "grad_norm": 0.8674799799919128,
+      "learning_rate": 2.288805948824212e-05,
+      "loss": 1.6172,
+      "step": 16802
+    },
+    {
+      "epoch": 0.42171059224371177,
+      "grad_norm": 0.9614238142967224,
+      "learning_rate": 2.2804389348704858e-05,
+      "loss": 1.6031,
+      "step": 16833
+    },
+    {
+      "epoch": 0.42248722316865417,
+      "grad_norm": 0.8952891826629639,
+      "learning_rate": 2.2720743986335374e-05,
+      "loss": 1.5921,
+      "step": 16864
+    },
+    {
+      "epoch": 0.42326385409359657,
+      "grad_norm": 0.8816654682159424,
+      "learning_rate": 2.2637124345060233e-05,
+      "loss": 1.5838,
+      "step": 16895
+    },
+    {
+      "epoch": 0.4240404850185389,
+      "grad_norm": 0.8586190938949585,
+      "learning_rate": 2.2553531368515695e-05,
+      "loss": 1.5965,
+      "step": 16926
+    },
+    {
+      "epoch": 0.4248171159434813,
+      "grad_norm": 0.9344453811645508,
+      "learning_rate": 2.2469966000037144e-05,
+      "loss": 1.6041,
+      "step": 16957
+    },
+    {
+      "epoch": 0.4255937468684237,
+      "grad_norm": 0.924696683883667,
+      "learning_rate": 2.2386429182648417e-05,
+      "loss": 1.5944,
+      "step": 16988
+    },
+    {
+      "epoch": 0.42637037779336606,
+      "grad_norm": 0.8722828030586243,
+      "learning_rate": 2.230292185905114e-05,
+      "loss": 1.6213,
+      "step": 17019
+    },
+    {
+      "epoch": 0.42714700871830846,
+      "grad_norm": 0.8846574425697327,
+      "learning_rate": 2.2219444971614116e-05,
+      "loss": 1.6087,
+      "step": 17050
+    },
+    {
+      "epoch": 0.4279236396432508,
+      "grad_norm": 0.8602490425109863,
+      "learning_rate": 2.2135999462362655e-05,
+      "loss": 1.5915,
+      "step": 17081
+    },
+    {
+      "epoch": 0.4287002705681932,
+      "grad_norm": 0.8821462988853455,
+      "learning_rate": 2.2052586272968003e-05,
+      "loss": 1.6104,
+      "step": 17112
+    },
+    {
+      "epoch": 0.4294769014931356,
+      "grad_norm": 0.8978081345558167,
+      "learning_rate": 2.196920634473666e-05,
+      "loss": 1.5932,
+      "step": 17143
+    },
+    {
+      "epoch": 0.43025353241807796,
+      "grad_norm": 0.9008092284202576,
+      "learning_rate": 2.1885860618599787e-05,
+      "loss": 1.5915,
+      "step": 17174
+    },
+    {
+      "epoch": 0.43103016334302036,
+      "grad_norm": 0.9174037575721741,
+      "learning_rate": 2.1802550035102577e-05,
+      "loss": 1.6062,
+      "step": 17205
+    },
+    {
+      "epoch": 0.4318067942679627,
+      "grad_norm": 0.8990745544433594,
+      "learning_rate": 2.171927553439363e-05,
+      "loss": 1.6019,
+      "step": 17236
+    },
+    {
+      "epoch": 0.4325834251929051,
+      "grad_norm": 0.867262601852417,
+      "learning_rate": 2.1636038056214376e-05,
+      "loss": 1.5862,
+      "step": 17267
+    },
+    {
+      "epoch": 0.4333600561178475,
+      "grad_norm": 0.8440841436386108,
+      "learning_rate": 2.155283853988844e-05,
+      "loss": 1.5698,
+      "step": 17298
+    },
+    {
+      "epoch": 0.43413668704278985,
+      "grad_norm": 0.9019381999969482,
+      "learning_rate": 2.146967792431106e-05,
+      "loss": 1.6102,
+      "step": 17329
+    },
+    {
+      "epoch": 0.43491331796773225,
+      "grad_norm": 0.8731846213340759,
+      "learning_rate": 2.138655714793849e-05,
+      "loss": 1.5867,
+      "step": 17360
+    },
+    {
+      "epoch": 0.4356899488926746,
+      "grad_norm": 0.8628113269805908,
+      "learning_rate": 2.1303477148777367e-05,
+      "loss": 1.5992,
+      "step": 17391
+    },
+    {
+      "epoch": 0.436466579817617,
+      "grad_norm": 0.8760020732879639,
+      "learning_rate": 2.122043886437421e-05,
+      "loss": 1.5897,
+      "step": 17422
+    },
+    {
+      "epoch": 0.4372432107425594,
+      "grad_norm": 0.8777160048484802,
+      "learning_rate": 2.1137443231804765e-05,
+      "loss": 1.5809,
+      "step": 17453
+    },
+    {
+      "epoch": 0.43801984166750174,
+      "grad_norm": 0.8788965344429016,
+      "learning_rate": 2.105449118766347e-05,
+      "loss": 1.6006,
+      "step": 17484
+    },
+    {
+      "epoch": 0.43879647259244414,
+      "grad_norm": 0.9438194036483765,
+      "learning_rate": 2.097158366805287e-05,
+      "loss": 1.5916,
+      "step": 17515
+    },
+    {
+      "epoch": 0.4395731035173865,
+      "grad_norm": 0.9142041206359863,
+      "learning_rate": 2.0888721608573047e-05,
+      "loss": 1.5874,
+      "step": 17546
+    },
+    {
+      "epoch": 0.4403497344423289,
+      "grad_norm": 0.8893091082572937,
+      "learning_rate": 2.0805905944311087e-05,
+      "loss": 1.6032,
+      "step": 17577
+    },
+    {
+      "epoch": 0.4411263653672713,
+      "grad_norm": 0.8822838068008423,
+      "learning_rate": 2.0723137609830497e-05,
+      "loss": 1.6018,
+      "step": 17608
+    },
+    {
+      "epoch": 0.44190299629221363,
+      "grad_norm": 0.8874905109405518,
+      "learning_rate": 2.0640417539160686e-05,
+      "loss": 1.5894,
+      "step": 17639
+    },
+    {
+      "epoch": 0.44267962721715604,
+      "grad_norm": 0.8892821669578552,
+      "learning_rate": 2.0557746665786427e-05,
+      "loss": 1.5873,
+      "step": 17670
+    },
+    {
+      "epoch": 0.4434562581420984,
+      "grad_norm": 0.8921502232551575,
+      "learning_rate": 2.0475125922637256e-05,
+      "loss": 1.6111,
+      "step": 17701
+    },
+    {
+      "epoch": 0.4442328890670408,
+      "grad_norm": 0.908928394317627,
+      "learning_rate": 2.0392556242077047e-05,
+      "loss": 1.5754,
+      "step": 17732
+    },
+    {
+      "epoch": 0.4450095199919832,
+      "grad_norm": 0.9088190793991089,
+      "learning_rate": 2.031003855589343e-05,
+      "loss": 1.589,
+      "step": 17763
+    },
+    {
+      "epoch": 0.4457861509169255,
+      "grad_norm": 0.9038158655166626,
+      "learning_rate": 2.022757379528727e-05,
+      "loss": 1.6003,
+      "step": 17794
+    },
+    {
+      "epoch": 0.44656278184186793,
+      "grad_norm": 0.9049093127250671,
+      "learning_rate": 2.0145162890862184e-05,
+      "loss": 1.5752,
+      "step": 17825
+    },
+    {
+      "epoch": 0.44733941276681033,
+      "grad_norm": 0.827835202217102,
+      "learning_rate": 2.0062806772614022e-05,
+      "loss": 1.584,
+      "step": 17856
+    },
+    {
+      "epoch": 0.4481160436917527,
+      "grad_norm": 0.951187789440155,
+      "learning_rate": 1.9980506369920392e-05,
+      "loss": 1.5783,
+      "step": 17887
+    },
+    {
+      "epoch": 0.4488926746166951,
+      "grad_norm": 0.8784689903259277,
+      "learning_rate": 1.989826261153015e-05,
+      "loss": 1.5744,
+      "step": 17918
+    },
+    {
+      "epoch": 0.4496693055416374,
+      "grad_norm": 0.8900132179260254,
+      "learning_rate": 1.9816076425552923e-05,
+      "loss": 1.5695,
+      "step": 17949
+    },
+    {
+      "epoch": 0.4504459364665798,
+      "grad_norm": 0.8698625564575195,
+      "learning_rate": 1.9733948739448676e-05,
+      "loss": 1.5884,
+      "step": 17980
+    },
+    {
+      "epoch": 0.4512225673915222,
+      "grad_norm": 0.8933484554290771,
+      "learning_rate": 1.9651880480017155e-05,
+      "loss": 1.5761,
+      "step": 18011
+    },
+    {
+      "epoch": 0.45199919831646457,
+      "grad_norm": 0.8772753477096558,
+      "learning_rate": 1.9569872573387516e-05,
+      "loss": 1.5765,
+      "step": 18042
+    },
+    {
+      "epoch": 0.45277582924140697,
+      "grad_norm": 0.8958901762962341,
+      "learning_rate": 1.9487925945007854e-05,
+      "loss": 1.5815,
+      "step": 18073
+    },
+    {
+      "epoch": 0.4535524601663493,
+      "grad_norm": 0.8793728947639465,
+      "learning_rate": 1.9406041519634726e-05,
+      "loss": 1.5662,
+      "step": 18104
+    },
+    {
+      "epoch": 0.4543290910912917,
+      "grad_norm": 0.8501383066177368,
+      "learning_rate": 1.932422022132275e-05,
+      "loss": 1.5755,
+      "step": 18135
+    },
+    {
+      "epoch": 0.4551057220162341,
+      "grad_norm": 0.8817576766014099,
+      "learning_rate": 1.924246297341414e-05,
+      "loss": 1.6053,
+      "step": 18166
+    },
+    {
+      "epoch": 0.45588235294117646,
+      "grad_norm": 0.8871498703956604,
+      "learning_rate": 1.9160770698528338e-05,
+      "loss": 1.5758,
+      "step": 18197
+    },
+    {
+      "epoch": 0.45665898386611886,
+      "grad_norm": 0.8694120049476624,
+      "learning_rate": 1.907914431855156e-05,
+      "loss": 1.5969,
+      "step": 18228
+    },
+    {
+      "epoch": 0.4574356147910612,
+      "grad_norm": 0.8998878598213196,
+      "learning_rate": 1.8997584754626412e-05,
+      "loss": 1.5888,
+      "step": 18259
+    },
+    {
+      "epoch": 0.4582122457160036,
+      "grad_norm": 0.90102618932724,
+      "learning_rate": 1.8916092927141486e-05,
+      "loss": 1.589,
+      "step": 18290
+    }
+  ],
+  "logging_steps": 31,
+  "max_steps": 30517,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "save_steps": 3052,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 1.2790112517183504e+19,
+  "train_batch_size": 16,
+  "trial_name": null,
+  "trial_params": null
+}