diff --git "a/trainer_state.json" "b/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/trainer_state.json"
@@ -0,0 +1,9428 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 2.0,
+  "eval_steps": 200,
+  "global_step": 1334,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0014992503748125937,
+      "grad_norm": 2.632088334378145,
+      "learning_rate": 9.999986134743949e-06,
+      "loss": 0.2577,
+      "step": 1
+    },
+    {
+      "epoch": 0.0029985007496251873,
+      "grad_norm": 1.3635681301372573,
+      "learning_rate": 9.99994453905269e-06,
+      "loss": 0.1951,
+      "step": 2
+    },
+    {
+      "epoch": 0.004497751124437781,
+      "grad_norm": 1.90775245744519,
+      "learning_rate": 9.999875213156919e-06,
+      "loss": 0.2031,
+      "step": 3
+    },
+    {
+      "epoch": 0.005997001499250375,
+      "grad_norm": 1.5458352422094463,
+      "learning_rate": 9.999778157441126e-06,
+      "loss": 0.1826,
+      "step": 4
+    },
+    {
+      "epoch": 0.0074962518740629685,
+      "grad_norm": 1.477676859024172,
+      "learning_rate": 9.99965337244359e-06,
+      "loss": 0.1897,
+      "step": 5
+    },
+    {
+      "epoch": 0.008995502248875561,
+      "grad_norm": 1.335244916792951,
+      "learning_rate": 9.999500858856382e-06,
+      "loss": 0.1857,
+      "step": 6
+    },
+    {
+      "epoch": 0.010494752623688156,
+      "grad_norm": 1.4545201222875872,
+      "learning_rate": 9.999320617525356e-06,
+      "loss": 0.1764,
+      "step": 7
+    },
+    {
+      "epoch": 0.01199400299850075,
+      "grad_norm": 1.302494952619536,
+      "learning_rate": 9.999112649450154e-06,
+      "loss": 0.167,
+      "step": 8
+    },
+    {
+      "epoch": 0.013493253373313344,
+      "grad_norm": 0.906640890873059,
+      "learning_rate": 9.998876955784183e-06,
+      "loss": 0.1184,
+      "step": 9
+    },
+    {
+      "epoch": 0.014992503748125937,
+      "grad_norm": 1.0531933563032372,
+      "learning_rate": 9.998613537834625e-06,
+      "loss": 0.1693,
+      "step": 10
+    },
+    {
+      "epoch": 0.01649175412293853,
+      "grad_norm": 1.0398325661387315,
+      "learning_rate": 9.998322397062426e-06,
+      "loss": 0.1584,
+      "step": 11
+    },
+    {
+      "epoch": 0.017991004497751123,
+      "grad_norm": 0.771550183184724,
+      "learning_rate": 9.99800353508228e-06,
+      "loss": 0.1267,
+      "step": 12
+    },
+    {
+      "epoch": 0.019490254872563718,
+      "grad_norm": 0.8855376599261632,
+      "learning_rate": 9.997656953662627e-06,
+      "loss": 0.1249,
+      "step": 13
+    },
+    {
+      "epoch": 0.020989505247376312,
+      "grad_norm": 0.9880624702968375,
+      "learning_rate": 9.997282654725645e-06,
+      "loss": 0.1469,
+      "step": 14
+    },
+    {
+      "epoch": 0.022488755622188907,
+      "grad_norm": 0.7504115239691771,
+      "learning_rate": 9.996880640347234e-06,
+      "loss": 0.1135,
+      "step": 15
+    },
+    {
+      "epoch": 0.0239880059970015,
+      "grad_norm": 0.8424284590580469,
+      "learning_rate": 9.99645091275701e-06,
+      "loss": 0.1286,
+      "step": 16
+    },
+    {
+      "epoch": 0.025487256371814093,
+      "grad_norm": 0.8828165810590822,
+      "learning_rate": 9.99599347433828e-06,
+      "loss": 0.1359,
+      "step": 17
+    },
+    {
+      "epoch": 0.026986506746626688,
+      "grad_norm": 0.9159909272226728,
+      "learning_rate": 9.99550832762805e-06,
+      "loss": 0.1557,
+      "step": 18
+    },
+    {
+      "epoch": 0.02848575712143928,
+      "grad_norm": 0.8339059979186626,
+      "learning_rate": 9.99499547531699e-06,
+      "loss": 0.1232,
+      "step": 19
+    },
+    {
+      "epoch": 0.029985007496251874,
+      "grad_norm": 0.9299036988640467,
+      "learning_rate": 9.994454920249433e-06,
+      "loss": 0.1371,
+      "step": 20
+    },
+    {
+      "epoch": 0.031484257871064465,
+      "grad_norm": 0.692840828521263,
+      "learning_rate": 9.993886665423348e-06,
+      "loss": 0.1084,
+      "step": 21
+    },
+    {
+      "epoch": 0.03298350824587706,
+      "grad_norm": 0.7329684426755911,
+      "learning_rate": 9.993290713990343e-06,
+      "loss": 0.1353,
+      "step": 22
+    },
+    {
+      "epoch": 0.034482758620689655,
+      "grad_norm": 0.6861117531500907,
+      "learning_rate": 9.99266706925562e-06,
+      "loss": 0.1032,
+      "step": 23
+    },
+    {
+      "epoch": 0.035982008995502246,
+      "grad_norm": 0.9652589761077365,
+      "learning_rate": 9.992015734677979e-06,
+      "loss": 0.1441,
+      "step": 24
+    },
+    {
+      "epoch": 0.037481259370314844,
+      "grad_norm": 0.7670383995613768,
+      "learning_rate": 9.991336713869785e-06,
+      "loss": 0.1177,
+      "step": 25
+    },
+    {
+      "epoch": 0.038980509745127435,
+      "grad_norm": 0.6468490356392375,
+      "learning_rate": 9.99063001059696e-06,
+      "loss": 0.1048,
+      "step": 26
+    },
+    {
+      "epoch": 0.04047976011994003,
+      "grad_norm": 0.8891919977168528,
+      "learning_rate": 9.989895628778952e-06,
+      "loss": 0.1426,
+      "step": 27
+    },
+    {
+      "epoch": 0.041979010494752625,
+      "grad_norm": 0.7067059093450255,
+      "learning_rate": 9.989133572488716e-06,
+      "loss": 0.1068,
+      "step": 28
+    },
+    {
+      "epoch": 0.043478260869565216,
+      "grad_norm": 0.9397651142602369,
+      "learning_rate": 9.988343845952697e-06,
+      "loss": 0.1177,
+      "step": 29
+    },
+    {
+      "epoch": 0.044977511244377814,
+      "grad_norm": 0.7317702054885957,
+      "learning_rate": 9.987526453550798e-06,
+      "loss": 0.1079,
+      "step": 30
+    },
+    {
+      "epoch": 0.046476761619190406,
+      "grad_norm": 0.7134993287598351,
+      "learning_rate": 9.98668139981636e-06,
+      "loss": 0.0978,
+      "step": 31
+    },
+    {
+      "epoch": 0.047976011994003,
+      "grad_norm": 0.8850671176118551,
+      "learning_rate": 9.98580868943614e-06,
+      "loss": 0.1406,
+      "step": 32
+    },
+    {
+      "epoch": 0.049475262368815595,
+      "grad_norm": 0.6360435306544596,
+      "learning_rate": 9.984908327250278e-06,
+      "loss": 0.0905,
+      "step": 33
+    },
+    {
+      "epoch": 0.050974512743628186,
+      "grad_norm": 0.6631905653595115,
+      "learning_rate": 9.983980318252274e-06,
+      "loss": 0.0832,
+      "step": 34
+    },
+    {
+      "epoch": 0.05247376311844078,
+      "grad_norm": 0.7420240319247169,
+      "learning_rate": 9.983024667588961e-06,
+      "loss": 0.1148,
+      "step": 35
+    },
+    {
+      "epoch": 0.053973013493253376,
+      "grad_norm": 0.8540431304398725,
+      "learning_rate": 9.982041380560476e-06,
+      "loss": 0.1337,
+      "step": 36
+    },
+    {
+      "epoch": 0.05547226386806597,
+      "grad_norm": 0.8889427619908081,
+      "learning_rate": 9.98103046262023e-06,
+      "loss": 0.1159,
+      "step": 37
+    },
+    {
+      "epoch": 0.05697151424287856,
+      "grad_norm": 0.9685249407418689,
+      "learning_rate": 9.979991919374877e-06,
+      "loss": 0.1559,
+      "step": 38
+    },
+    {
+      "epoch": 0.05847076461769116,
+      "grad_norm": 0.7842906189439244,
+      "learning_rate": 9.978925756584284e-06,
+      "loss": 0.1281,
+      "step": 39
+    },
+    {
+      "epoch": 0.05997001499250375,
+      "grad_norm": 0.9078641893869315,
+      "learning_rate": 9.9778319801615e-06,
+      "loss": 0.1031,
+      "step": 40
+    },
+    {
+      "epoch": 0.06146926536731634,
+      "grad_norm": 0.753631640815483,
+      "learning_rate": 9.976710596172721e-06,
+      "loss": 0.1145,
+      "step": 41
+    },
+    {
+      "epoch": 0.06296851574212893,
+      "grad_norm": 0.7866503923292721,
+      "learning_rate": 9.975561610837254e-06,
+      "loss": 0.1128,
+      "step": 42
+    },
+    {
+      "epoch": 0.06446776611694154,
+      "grad_norm": 0.747180587762589,
+      "learning_rate": 9.974385030527496e-06,
+      "loss": 0.1057,
+      "step": 43
+    },
+    {
+      "epoch": 0.06596701649175413,
+      "grad_norm": 0.779548717087657,
+      "learning_rate": 9.973180861768874e-06,
+      "loss": 0.1278,
+      "step": 44
+    },
+    {
+      "epoch": 0.06746626686656672,
+      "grad_norm": 0.9050214055632173,
+      "learning_rate": 9.971949111239838e-06,
+      "loss": 0.1412,
+      "step": 45
+    },
+    {
+      "epoch": 0.06896551724137931,
+      "grad_norm": 0.6598084992720458,
+      "learning_rate": 9.970689785771798e-06,
+      "loss": 0.0923,
+      "step": 46
+    },
+    {
+      "epoch": 0.0704647676161919,
+      "grad_norm": 0.5973051401187608,
+      "learning_rate": 9.969402892349105e-06,
+      "loss": 0.0819,
+      "step": 47
+    },
+    {
+      "epoch": 0.07196401799100449,
+      "grad_norm": 0.7024131560567182,
+      "learning_rate": 9.968088438109002e-06,
+      "loss": 0.0896,
+      "step": 48
+    },
+    {
+      "epoch": 0.0734632683658171,
+      "grad_norm": 0.6895312320801684,
+      "learning_rate": 9.966746430341584e-06,
+      "loss": 0.0901,
+      "step": 49
+    },
+    {
+      "epoch": 0.07496251874062969,
+      "grad_norm": 0.9623184880111413,
+      "learning_rate": 9.965376876489765e-06,
+      "loss": 0.1342,
+      "step": 50
+    },
+    {
+      "epoch": 0.07646176911544228,
+      "grad_norm": 0.759221938945548,
+      "learning_rate": 9.963979784149232e-06,
+      "loss": 0.1186,
+      "step": 51
+    },
+    {
+      "epoch": 0.07796101949025487,
+      "grad_norm": 1.0115053412457504,
+      "learning_rate": 9.962555161068401e-06,
+      "loss": 0.132,
+      "step": 52
+    },
+    {
+      "epoch": 0.07946026986506746,
+      "grad_norm": 0.7823875649500759,
+      "learning_rate": 9.961103015148376e-06,
+      "loss": 0.118,
+      "step": 53
+    },
+    {
+      "epoch": 0.08095952023988005,
+      "grad_norm": 0.7844869185787823,
+      "learning_rate": 9.95962335444291e-06,
+      "loss": 0.1144,
+      "step": 54
+    },
+    {
+      "epoch": 0.08245877061469266,
+      "grad_norm": 0.6650778734275005,
+      "learning_rate": 9.958116187158351e-06,
+      "loss": 0.0926,
+      "step": 55
+    },
+    {
+      "epoch": 0.08395802098950525,
+      "grad_norm": 0.8514977717725526,
+      "learning_rate": 9.956581521653604e-06,
+      "loss": 0.1448,
+      "step": 56
+    },
+    {
+      "epoch": 0.08545727136431784,
+      "grad_norm": 0.8881056871709447,
+      "learning_rate": 9.955019366440082e-06,
+      "loss": 0.1246,
+      "step": 57
+    },
+    {
+      "epoch": 0.08695652173913043,
+      "grad_norm": 0.9002342431951453,
+      "learning_rate": 9.953429730181653e-06,
+      "loss": 0.1224,
+      "step": 58
+    },
+    {
+      "epoch": 0.08845577211394302,
+      "grad_norm": 0.6435030888113159,
+      "learning_rate": 9.95181262169461e-06,
+      "loss": 0.0937,
+      "step": 59
+    },
+    {
+      "epoch": 0.08995502248875563,
+      "grad_norm": 0.7046884560969475,
+      "learning_rate": 9.950168049947597e-06,
+      "loss": 0.1079,
+      "step": 60
+    },
+    {
+      "epoch": 0.09145427286356822,
+      "grad_norm": 0.6001293137264938,
+      "learning_rate": 9.948496024061577e-06,
+      "loss": 0.0784,
+      "step": 61
+    },
+    {
+      "epoch": 0.09295352323838081,
+      "grad_norm": 0.6994714077935421,
+      "learning_rate": 9.94679655330978e-06,
+      "loss": 0.1203,
+      "step": 62
+    },
+    {
+      "epoch": 0.0944527736131934,
+      "grad_norm": 1.0514885534041585,
+      "learning_rate": 9.945069647117645e-06,
+      "loss": 0.1188,
+      "step": 63
+    },
+    {
+      "epoch": 0.095952023988006,
+      "grad_norm": 0.8262054285569036,
+      "learning_rate": 9.943315315062766e-06,
+      "loss": 0.1107,
+      "step": 64
+    },
+    {
+      "epoch": 0.09745127436281859,
+      "grad_norm": 0.61175212034468,
+      "learning_rate": 9.941533566874852e-06,
+      "loss": 0.0713,
+      "step": 65
+    },
+    {
+      "epoch": 0.09895052473763119,
+      "grad_norm": 1.0053501902309385,
+      "learning_rate": 9.939724412435661e-06,
+      "loss": 0.1407,
+      "step": 66
+    },
+    {
+      "epoch": 0.10044977511244378,
+      "grad_norm": 0.6423717001707471,
+      "learning_rate": 9.937887861778947e-06,
+      "loss": 0.0986,
+      "step": 67
+    },
+    {
+      "epoch": 0.10194902548725637,
+      "grad_norm": 1.0329709039675985,
+      "learning_rate": 9.93602392509041e-06,
+      "loss": 0.1244,
+      "step": 68
+    },
+    {
+      "epoch": 0.10344827586206896,
+      "grad_norm": 0.7921131911875811,
+      "learning_rate": 9.934132612707631e-06,
+      "loss": 0.102,
+      "step": 69
+    },
+    {
+      "epoch": 0.10494752623688156,
+      "grad_norm": 0.7646726651152976,
+      "learning_rate": 9.932213935120025e-06,
+      "loss": 0.0925,
+      "step": 70
+    },
+    {
+      "epoch": 0.10644677661169415,
+      "grad_norm": 0.534780309707607,
+      "learning_rate": 9.930267902968774e-06,
+      "loss": 0.0773,
+      "step": 71
+    },
+    {
+      "epoch": 0.10794602698650675,
+      "grad_norm": 0.9112093407251666,
+      "learning_rate": 9.928294527046771e-06,
+      "loss": 0.1255,
+      "step": 72
+    },
+    {
+      "epoch": 0.10944527736131934,
+      "grad_norm": 0.9941740808876021,
+      "learning_rate": 9.92629381829856e-06,
+      "loss": 0.131,
+      "step": 73
+    },
+    {
+      "epoch": 0.11094452773613193,
+      "grad_norm": 1.0286965765757703,
+      "learning_rate": 9.924265787820279e-06,
+      "loss": 0.1532,
+      "step": 74
+    },
+    {
+      "epoch": 0.11244377811094453,
+      "grad_norm": 0.7825416617370475,
+      "learning_rate": 9.92221044685959e-06,
+      "loss": 0.1138,
+      "step": 75
+    },
+    {
+      "epoch": 0.11394302848575712,
+      "grad_norm": 0.7612169911413965,
+      "learning_rate": 9.920127806815627e-06,
+      "loss": 0.092,
+      "step": 76
+    },
+    {
+      "epoch": 0.11544227886056972,
+      "grad_norm": 0.6470931781730698,
+      "learning_rate": 9.918017879238922e-06,
+      "loss": 0.1033,
+      "step": 77
+    },
+    {
+      "epoch": 0.11694152923538231,
+      "grad_norm": 0.5880849008953836,
+      "learning_rate": 9.915880675831352e-06,
+      "loss": 0.0657,
+      "step": 78
+    },
+    {
+      "epoch": 0.1184407796101949,
+      "grad_norm": 0.8338571922615028,
+      "learning_rate": 9.913716208446067e-06,
+      "loss": 0.1024,
+      "step": 79
+    },
+    {
+      "epoch": 0.1199400299850075,
+      "grad_norm": 0.7716670982559944,
+      "learning_rate": 9.91152448908742e-06,
+      "loss": 0.101,
+      "step": 80
+    },
+    {
+      "epoch": 0.12143928035982009,
+      "grad_norm": 0.796506638272956,
+      "learning_rate": 9.909305529910917e-06,
+      "loss": 0.1146,
+      "step": 81
+    },
+    {
+      "epoch": 0.12293853073463268,
+      "grad_norm": 0.7448990091828431,
+      "learning_rate": 9.907059343223129e-06,
+      "loss": 0.1077,
+      "step": 82
+    },
+    {
+      "epoch": 0.12443778110944528,
+      "grad_norm": 0.5799271272535026,
+      "learning_rate": 9.904785941481638e-06,
+      "loss": 0.0716,
+      "step": 83
+    },
+    {
+      "epoch": 0.12593703148425786,
+      "grad_norm": 0.5359775769156907,
+      "learning_rate": 9.902485337294965e-06,
+      "loss": 0.0618,
+      "step": 84
+    },
+    {
+      "epoch": 0.12743628185907047,
+      "grad_norm": 0.7323691864981551,
+      "learning_rate": 9.900157543422493e-06,
+      "loss": 0.0929,
+      "step": 85
+    },
+    {
+      "epoch": 0.12893553223388307,
+      "grad_norm": 0.7955723162180953,
+      "learning_rate": 9.897802572774407e-06,
+      "loss": 0.0991,
+      "step": 86
+    },
+    {
+      "epoch": 0.13043478260869565,
+      "grad_norm": 0.6503586438203615,
+      "learning_rate": 9.895420438411616e-06,
+      "loss": 0.082,
+      "step": 87
+    },
+    {
+      "epoch": 0.13193403298350825,
+      "grad_norm": 0.584188860221843,
+      "learning_rate": 9.893011153545679e-06,
+      "loss": 0.0665,
+      "step": 88
+    },
+    {
+      "epoch": 0.13343328335832083,
+      "grad_norm": 0.8874834609506581,
+      "learning_rate": 9.89057473153874e-06,
+      "loss": 0.1181,
+      "step": 89
+    },
+    {
+      "epoch": 0.13493253373313344,
+      "grad_norm": 0.7437673272160883,
+      "learning_rate": 9.888111185903442e-06,
+      "loss": 0.0862,
+      "step": 90
+    },
+    {
+      "epoch": 0.136431784107946,
+      "grad_norm": 0.7405040593359462,
+      "learning_rate": 9.885620530302865e-06,
+      "loss": 0.0889,
+      "step": 91
+    },
+    {
+      "epoch": 0.13793103448275862,
+      "grad_norm": 0.8798427142907458,
+      "learning_rate": 9.883102778550434e-06,
+      "loss": 0.1086,
+      "step": 92
+    },
+    {
+      "epoch": 0.13943028485757122,
+      "grad_norm": 0.6625922272645881,
+      "learning_rate": 9.880557944609863e-06,
+      "loss": 0.08,
+      "step": 93
+    },
+    {
+      "epoch": 0.1409295352323838,
+      "grad_norm": 0.834105622032754,
+      "learning_rate": 9.877986042595062e-06,
+      "loss": 0.1069,
+      "step": 94
+    },
+    {
+      "epoch": 0.1424287856071964,
+      "grad_norm": 0.9792098493264564,
+      "learning_rate": 9.87538708677006e-06,
+      "loss": 0.1263,
+      "step": 95
+    },
+    {
+      "epoch": 0.14392803598200898,
+      "grad_norm": 0.8324842699218236,
+      "learning_rate": 9.872761091548933e-06,
+      "loss": 0.1023,
+      "step": 96
+    },
+    {
+      "epoch": 0.1454272863568216,
+      "grad_norm": 0.9162821748521435,
+      "learning_rate": 9.870108071495721e-06,
+      "loss": 0.1278,
+      "step": 97
+    },
+    {
+      "epoch": 0.1469265367316342,
+      "grad_norm": 0.7263564343687559,
+      "learning_rate": 9.867428041324345e-06,
+      "loss": 0.0955,
+      "step": 98
+    },
+    {
+      "epoch": 0.14842578710644677,
+      "grad_norm": 0.7191386071900421,
+      "learning_rate": 9.864721015898524e-06,
+      "loss": 0.0851,
+      "step": 99
+    },
+    {
+      "epoch": 0.14992503748125938,
+      "grad_norm": 0.8761709641231086,
+      "learning_rate": 9.861987010231701e-06,
+      "loss": 0.1042,
+      "step": 100
+    },
+    {
+      "epoch": 0.15142428785607195,
+      "grad_norm": 0.8885535342463606,
+      "learning_rate": 9.85922603948695e-06,
+      "loss": 0.1155,
+      "step": 101
+    },
+    {
+      "epoch": 0.15292353823088456,
+      "grad_norm": 0.7668515166413616,
+      "learning_rate": 9.856438118976899e-06,
+      "loss": 0.097,
+      "step": 102
+    },
+    {
+      "epoch": 0.15442278860569716,
+      "grad_norm": 0.7206887436138562,
+      "learning_rate": 9.853623264163638e-06,
+      "loss": 0.0926,
+      "step": 103
+    },
+    {
+      "epoch": 0.15592203898050974,
+      "grad_norm": 0.7057304078999682,
+      "learning_rate": 9.850781490658643e-06,
+      "loss": 0.0908,
+      "step": 104
+    },
+    {
+      "epoch": 0.15742128935532235,
+      "grad_norm": 0.7536636574090124,
+      "learning_rate": 9.84791281422268e-06,
+      "loss": 0.1093,
+      "step": 105
+    },
+    {
+      "epoch": 0.15892053973013492,
+      "grad_norm": 0.8159727509323544,
+      "learning_rate": 9.845017250765721e-06,
+      "loss": 0.1147,
+      "step": 106
+    },
+    {
+      "epoch": 0.16041979010494753,
+      "grad_norm": 0.6797446755365417,
+      "learning_rate": 9.84209481634686e-06,
+      "loss": 0.093,
+      "step": 107
+    },
+    {
+      "epoch": 0.1619190404797601,
+      "grad_norm": 0.9525448512663907,
+      "learning_rate": 9.839145527174216e-06,
+      "loss": 0.1208,
+      "step": 108
+    },
+    {
+      "epoch": 0.1634182908545727,
+      "grad_norm": 0.7199422195823577,
+      "learning_rate": 9.836169399604846e-06,
+      "loss": 0.1029,
+      "step": 109
+    },
+    {
+      "epoch": 0.16491754122938532,
+      "grad_norm": 0.7632660473151701,
+      "learning_rate": 9.833166450144665e-06,
+      "loss": 0.0829,
+      "step": 110
+    },
+    {
+      "epoch": 0.1664167916041979,
+      "grad_norm": 0.8061963846987128,
+      "learning_rate": 9.830136695448334e-06,
+      "loss": 0.0942,
+      "step": 111
+    },
+    {
+      "epoch": 0.1679160419790105,
+      "grad_norm": 0.6855649770366048,
+      "learning_rate": 9.827080152319182e-06,
+      "loss": 0.0944,
+      "step": 112
+    },
+    {
+      "epoch": 0.16941529235382308,
+      "grad_norm": 1.0054017293416877,
+      "learning_rate": 9.823996837709114e-06,
+      "loss": 0.098,
+      "step": 113
+    },
+    {
+      "epoch": 0.17091454272863568,
+      "grad_norm": 0.7835519636199206,
+      "learning_rate": 9.820886768718503e-06,
+      "loss": 0.1128,
+      "step": 114
+    },
+    {
+      "epoch": 0.1724137931034483,
+      "grad_norm": 0.6085473870543324,
+      "learning_rate": 9.817749962596115e-06,
+      "loss": 0.0807,
+      "step": 115
+    },
+    {
+      "epoch": 0.17391304347826086,
+      "grad_norm": 1.0178758047955008,
+      "learning_rate": 9.814586436738998e-06,
+      "loss": 0.185,
+      "step": 116
+    },
+    {
+      "epoch": 0.17541229385307347,
+      "grad_norm": 0.7969621247225069,
+      "learning_rate": 9.811396208692387e-06,
+      "loss": 0.1018,
+      "step": 117
+    },
+    {
+      "epoch": 0.17691154422788605,
+      "grad_norm": 0.9290558112614182,
+      "learning_rate": 9.808179296149616e-06,
+      "loss": 0.1221,
+      "step": 118
+    },
+    {
+      "epoch": 0.17841079460269865,
+      "grad_norm": 0.9688131141393559,
+      "learning_rate": 9.804935716952011e-06,
+      "loss": 0.1179,
+      "step": 119
+    },
+    {
+      "epoch": 0.17991004497751126,
+      "grad_norm": 0.7093096888669723,
+      "learning_rate": 9.801665489088795e-06,
+      "loss": 0.0906,
+      "step": 120
+    },
+    {
+      "epoch": 0.18140929535232383,
+      "grad_norm": 0.977726115293873,
+      "learning_rate": 9.798368630696984e-06,
+      "loss": 0.124,
+      "step": 121
+    },
+    {
+      "epoch": 0.18290854572713644,
+      "grad_norm": 0.8182312725457105,
+      "learning_rate": 9.795045160061295e-06,
+      "loss": 0.1087,
+      "step": 122
+    },
+    {
+      "epoch": 0.18440779610194902,
+      "grad_norm": 0.7976350668565922,
+      "learning_rate": 9.791695095614036e-06,
+      "loss": 0.1084,
+      "step": 123
+    },
+    {
+      "epoch": 0.18590704647676162,
+      "grad_norm": 0.6927813045187238,
+      "learning_rate": 9.788318455935008e-06,
+      "loss": 0.0775,
+      "step": 124
+    },
+    {
+      "epoch": 0.1874062968515742,
+      "grad_norm": 0.7019465819797817,
+      "learning_rate": 9.7849152597514e-06,
+      "loss": 0.0764,
+      "step": 125
+    },
+    {
+      "epoch": 0.1889055472263868,
+      "grad_norm": 0.7344133583074559,
+      "learning_rate": 9.781485525937683e-06,
+      "loss": 0.078,
+      "step": 126
+    },
+    {
+      "epoch": 0.1904047976011994,
+      "grad_norm": 0.6299514820101718,
+      "learning_rate": 9.778029273515519e-06,
+      "loss": 0.0816,
+      "step": 127
+    },
+    {
+      "epoch": 0.191904047976012,
+      "grad_norm": 0.8056389963680768,
+      "learning_rate": 9.774546521653633e-06,
+      "loss": 0.1109,
+      "step": 128
+    },
+    {
+      "epoch": 0.1934032983508246,
+      "grad_norm": 0.7378058116170683,
+      "learning_rate": 9.771037289667726e-06,
+      "loss": 0.0859,
+      "step": 129
+    },
+    {
+      "epoch": 0.19490254872563717,
+      "grad_norm": 0.8039895362497604,
+      "learning_rate": 9.767501597020357e-06,
+      "loss": 0.0976,
+      "step": 130
+    },
+    {
+      "epoch": 0.19640179910044978,
+      "grad_norm": 0.8139779904717225,
+      "learning_rate": 9.76393946332084e-06,
+      "loss": 0.0984,
+      "step": 131
+    },
+    {
+      "epoch": 0.19790104947526238,
+      "grad_norm": 1.0354153568551883,
+      "learning_rate": 9.760350908325131e-06,
+      "loss": 0.1068,
+      "step": 132
+    },
+    {
+      "epoch": 0.19940029985007496,
+      "grad_norm": 0.6790454601590706,
+      "learning_rate": 9.756735951935725e-06,
+      "loss": 0.1013,
+      "step": 133
+    },
+    {
+      "epoch": 0.20089955022488756,
+      "grad_norm": 0.6806177378342914,
+      "learning_rate": 9.753094614201542e-06,
+      "loss": 0.0746,
+      "step": 134
+    },
+    {
+      "epoch": 0.20239880059970014,
+      "grad_norm": 0.7472605096084765,
+      "learning_rate": 9.749426915317812e-06,
+      "loss": 0.0964,
+      "step": 135
+    },
+    {
+      "epoch": 0.20389805097451275,
+      "grad_norm": 0.6678064078439137,
+      "learning_rate": 9.74573287562597e-06,
+      "loss": 0.073,
+      "step": 136
+    },
+    {
+      "epoch": 0.20539730134932535,
+      "grad_norm": 0.7574052365835597,
+      "learning_rate": 9.742012515613536e-06,
+      "loss": 0.081,
+      "step": 137
+    },
+    {
+      "epoch": 0.20689655172413793,
+      "grad_norm": 0.6701862766653289,
+      "learning_rate": 9.738265855914014e-06,
+      "loss": 0.0933,
+      "step": 138
+    },
+    {
+      "epoch": 0.20839580209895053,
+      "grad_norm": 0.6916002955350535,
+      "learning_rate": 9.734492917306754e-06,
+      "loss": 0.0855,
+      "step": 139
+    },
+    {
+      "epoch": 0.2098950524737631,
+      "grad_norm": 0.7925854843946869,
+      "learning_rate": 9.730693720716866e-06,
+      "loss": 0.1171,
+      "step": 140
+    },
+    {
+      "epoch": 0.21139430284857572,
+      "grad_norm": 0.7301751527453835,
+      "learning_rate": 9.72686828721508e-06,
+      "loss": 0.0916,
+      "step": 141
+    },
+    {
+      "epoch": 0.2128935532233883,
+      "grad_norm": 0.9841223650233895,
+      "learning_rate": 9.723016638017644e-06,
+      "loss": 0.1301,
+      "step": 142
+    },
+    {
+      "epoch": 0.2143928035982009,
+      "grad_norm": 0.8838683071964671,
+      "learning_rate": 9.719138794486198e-06,
+      "loss": 0.0864,
+      "step": 143
+    },
+    {
+      "epoch": 0.2158920539730135,
+      "grad_norm": 0.6585131439336515,
+      "learning_rate": 9.715234778127658e-06,
+      "loss": 0.0715,
+      "step": 144
+    },
+    {
+      "epoch": 0.21739130434782608,
+      "grad_norm": 0.7757417054538653,
+      "learning_rate": 9.711304610594104e-06,
+      "loss": 0.0929,
+      "step": 145
+    },
+    {
+      "epoch": 0.21889055472263869,
+      "grad_norm": 0.9216646252686296,
+      "learning_rate": 9.70734831368264e-06,
+      "loss": 0.126,
+      "step": 146
+    },
+    {
+      "epoch": 0.22038980509745126,
+      "grad_norm": 0.7488728520571991,
+      "learning_rate": 9.7033659093353e-06,
+      "loss": 0.0835,
+      "step": 147
+    },
+    {
+      "epoch": 0.22188905547226387,
+      "grad_norm": 0.8379122569804117,
+      "learning_rate": 9.699357419638904e-06,
+      "loss": 0.0828,
+      "step": 148
+    },
+    {
+      "epoch": 0.22338830584707647,
+      "grad_norm": 0.8470724919462431,
+      "learning_rate": 9.695322866824948e-06,
+      "loss": 0.0949,
+      "step": 149
+    },
+    {
+      "epoch": 0.22488755622188905,
+      "grad_norm": 0.807806265120067,
+      "learning_rate": 9.691262273269472e-06,
+      "loss": 0.0931,
+      "step": 150
+    },
+    {
+      "epoch": 0.22638680659670166,
+      "grad_norm": 0.5662774413259967,
+      "learning_rate": 9.687175661492944e-06,
+      "loss": 0.0632,
+      "step": 151
+    },
+    {
+      "epoch": 0.22788605697151423,
+      "grad_norm": 0.6386404607110884,
+      "learning_rate": 9.683063054160136e-06,
+      "loss": 0.0619,
+      "step": 152
+    },
+    {
+      "epoch": 0.22938530734632684,
+      "grad_norm": 0.8064984888757815,
+      "learning_rate": 9.678924474079986e-06,
+      "loss": 0.0763,
+      "step": 153
+    },
+    {
+      "epoch": 0.23088455772113944,
+      "grad_norm": 0.6514506065268959,
+      "learning_rate": 9.67475994420548e-06,
+      "loss": 0.0715,
+      "step": 154
+    },
+    {
+      "epoch": 0.23238380809595202,
+      "grad_norm": 0.7128526826549882,
+      "learning_rate": 9.670569487633534e-06,
+      "loss": 0.0868,
+      "step": 155
+    },
+    {
+      "epoch": 0.23388305847076463,
+      "grad_norm": 0.8578956553417076,
+      "learning_rate": 9.666353127604845e-06,
+      "loss": 0.1128,
+      "step": 156
+    },
+    {
+      "epoch": 0.2353823088455772,
+      "grad_norm": 0.620642400394444,
+      "learning_rate": 9.66211088750378e-06,
+      "loss": 0.0792,
+      "step": 157
+    },
+    {
+      "epoch": 0.2368815592203898,
+      "grad_norm": 0.7451228704562336,
+      "learning_rate": 9.657842790858235e-06,
+      "loss": 0.0843,
+      "step": 158
+    },
+    {
+      "epoch": 0.2383808095952024,
+      "grad_norm": 1.1186380375552898,
+      "learning_rate": 9.65354886133951e-06,
+      "loss": 0.1491,
+      "step": 159
+    },
+    {
+      "epoch": 0.239880059970015,
+      "grad_norm": 0.6339221643120537,
+      "learning_rate": 9.64922912276218e-06,
+      "loss": 0.0664,
+      "step": 160
+    },
+    {
+      "epoch": 0.2413793103448276,
+      "grad_norm": 0.7837470134115554,
+      "learning_rate": 9.644883599083959e-06,
+      "loss": 0.0808,
+      "step": 161
+    },
+    {
+      "epoch": 0.24287856071964017,
+      "grad_norm": 0.8659114249010933,
+      "learning_rate": 9.640512314405563e-06,
+      "loss": 0.0952,
+      "step": 162
+    },
+    {
+      "epoch": 0.24437781109445278,
+      "grad_norm": 0.7228235095293009,
+      "learning_rate": 9.636115292970587e-06,
+      "loss": 0.0773,
+      "step": 163
+    },
+    {
+      "epoch": 0.24587706146926536,
+      "grad_norm": 0.7248808663600435,
+      "learning_rate": 9.63169255916536e-06,
+      "loss": 0.0754,
+      "step": 164
+    },
+    {
+      "epoch": 0.24737631184407796,
+      "grad_norm": 0.7208612157568969,
+      "learning_rate": 9.627244137518821e-06,
+      "loss": 0.0779,
+      "step": 165
+    },
+    {
+      "epoch": 0.24887556221889057,
+      "grad_norm": 0.7322360677400702,
+      "learning_rate": 9.622770052702366e-06,
+      "loss": 0.093,
+      "step": 166
+    },
+    {
+      "epoch": 0.25037481259370314,
+      "grad_norm": 0.6526914130973983,
+      "learning_rate": 9.618270329529734e-06,
+      "loss": 0.0691,
+      "step": 167
+    },
+    {
+      "epoch": 0.2518740629685157,
+      "grad_norm": 0.5263588508018929,
+      "learning_rate": 9.613744992956844e-06,
+      "loss": 0.0623,
+      "step": 168
+    },
+    {
+      "epoch": 0.25337331334332835,
+      "grad_norm": 0.5847646431706183,
+      "learning_rate": 9.609194068081682e-06,
+      "loss": 0.0761,
+      "step": 169
+    },
+    {
+      "epoch": 0.25487256371814093,
+      "grad_norm": 0.6967173973529336,
+      "learning_rate": 9.60461758014414e-06,
+      "loss": 0.0728,
+      "step": 170
+    },
+    {
+      "epoch": 0.2563718140929535,
+      "grad_norm": 0.793569211245774,
+      "learning_rate": 9.60001555452589e-06,
+      "loss": 0.081,
+      "step": 171
+    },
+    {
+      "epoch": 0.25787106446776614,
+      "grad_norm": 0.7751492174187726,
+      "learning_rate": 9.595388016750236e-06,
+      "loss": 0.0776,
+      "step": 172
+    },
+    {
+      "epoch": 0.2593703148425787,
+      "grad_norm": 0.7068790423385978,
+      "learning_rate": 9.590734992481978e-06,
+      "loss": 0.0833,
+      "step": 173
+    },
+    {
+      "epoch": 0.2608695652173913,
+      "grad_norm": 0.751693811689117,
+      "learning_rate": 9.586056507527266e-06,
+      "loss": 0.0859,
+      "step": 174
+    },
+    {
+      "epoch": 0.2623688155922039,
+      "grad_norm": 0.7797404668865464,
+      "learning_rate": 9.581352587833455e-06,
+      "loss": 0.0839,
+      "step": 175
+    },
+    {
+      "epoch": 0.2638680659670165,
+      "grad_norm": 0.9946709623644058,
+      "learning_rate": 9.576623259488966e-06,
+      "loss": 0.1219,
+      "step": 176
+    },
+    {
+      "epoch": 0.2653673163418291,
+      "grad_norm": 0.8361957496035568,
+      "learning_rate": 9.571868548723137e-06,
+      "loss": 0.1321,
+      "step": 177
+    },
+    {
+      "epoch": 0.26686656671664166,
+      "grad_norm": 0.7527791009536994,
+      "learning_rate": 9.567088481906084e-06,
+      "loss": 0.0746,
+      "step": 178
+    },
+    {
+      "epoch": 0.2683658170914543,
+      "grad_norm": 0.5791974630775953,
+      "learning_rate": 9.562283085548546e-06,
+      "loss": 0.0686,
+      "step": 179
+    },
+    {
+      "epoch": 0.2698650674662669,
+      "grad_norm": 0.9942553886306139,
+      "learning_rate": 9.55745238630174e-06,
+      "loss": 0.1077,
+      "step": 180
+    },
+    {
+      "epoch": 0.27136431784107945,
+      "grad_norm": 0.7745308633245181,
+      "learning_rate": 9.552596410957224e-06,
+      "loss": 0.1056,
+      "step": 181
+    },
+    {
+      "epoch": 0.272863568215892,
+      "grad_norm": 0.7460441747163276,
+      "learning_rate": 9.547715186446732e-06,
+      "loss": 0.0804,
+      "step": 182
+    },
+    {
+      "epoch": 0.27436281859070466,
+      "grad_norm": 0.7107636120600744,
+      "learning_rate": 9.542808739842034e-06,
+      "loss": 0.0918,
+      "step": 183
+    },
+    {
+      "epoch": 0.27586206896551724,
+      "grad_norm": 0.7159265848574122,
+      "learning_rate": 9.537877098354787e-06,
+      "loss": 0.0757,
+      "step": 184
+    },
+    {
+      "epoch": 0.2773613193403298,
+      "grad_norm": 0.6594934658254209,
+      "learning_rate": 9.532920289336378e-06,
+      "loss": 0.0825,
+      "step": 185
+    },
+    {
+      "epoch": 0.27886056971514245,
+      "grad_norm": 0.6626853201126666,
+      "learning_rate": 9.52793834027778e-06,
+      "loss": 0.0787,
+      "step": 186
+    },
+    {
+      "epoch": 0.280359820089955,
+      "grad_norm": 0.6686860991243766,
+      "learning_rate": 9.522931278809393e-06,
+      "loss": 0.0864,
+      "step": 187
+    },
+    {
+      "epoch": 0.2818590704647676,
+      "grad_norm": 0.6488411122605924,
+      "learning_rate": 9.517899132700889e-06,
+      "loss": 0.0807,
+      "step": 188
+    },
+    {
+      "epoch": 0.28335832083958024,
+      "grad_norm": 0.6572379446987092,
+      "learning_rate": 9.512841929861069e-06,
+      "loss": 0.0764,
+      "step": 189
+    },
+    {
+      "epoch": 0.2848575712143928,
+      "grad_norm": 0.7098027417918196,
+      "learning_rate": 9.507759698337698e-06,
+      "loss": 0.0798,
+      "step": 190
+    },
+    {
+      "epoch": 0.2863568215892054,
+      "grad_norm": 0.7399239773866507,
+      "learning_rate": 9.50265246631735e-06,
+      "loss": 0.0866,
+      "step": 191
+    },
+    {
+      "epoch": 0.28785607196401797,
+      "grad_norm": 0.6508139143056388,
+      "learning_rate": 9.49752026212526e-06,
+      "loss": 0.0814,
+      "step": 192
+    },
+    {
+      "epoch": 0.2893553223388306,
+      "grad_norm": 0.7551831276534148,
+      "learning_rate": 9.492363114225156e-06,
+      "loss": 0.0926,
+      "step": 193
+    },
+    {
+      "epoch": 0.2908545727136432,
+      "grad_norm": 0.5143278878586518,
+      "learning_rate": 9.487181051219107e-06,
+      "loss": 0.0651,
+      "step": 194
+    },
+    {
+      "epoch": 0.29235382308845576,
+      "grad_norm": 0.6392113826551745,
+      "learning_rate": 9.481974101847371e-06,
+      "loss": 0.0768,
+      "step": 195
+    },
+    {
+      "epoch": 0.2938530734632684,
+      "grad_norm": 0.6390873968028099,
+      "learning_rate": 9.476742294988214e-06,
+      "loss": 0.0576,
+      "step": 196
+    },
+    {
+      "epoch": 0.29535232383808097,
+      "grad_norm": 0.7948718677378915,
+      "learning_rate": 9.471485659657782e-06,
+      "loss": 0.0967,
+      "step": 197
+    },
+    {
+      "epoch": 0.29685157421289354,
+      "grad_norm": 0.696686797618622,
+      "learning_rate": 9.466204225009905e-06,
+      "loss": 0.072,
+      "step": 198
+    },
+    {
+      "epoch": 0.2983508245877061,
+      "grad_norm": 0.7185647801279773,
+      "learning_rate": 9.460898020335964e-06,
+      "loss": 0.0862,
+      "step": 199
+    },
+    {
+      "epoch": 0.29985007496251875,
+      "grad_norm": 0.5605663921640599,
+      "learning_rate": 9.455567075064715e-06,
+      "loss": 0.0571,
+      "step": 200
+    },
+    {
+      "epoch": 0.29985007496251875,
+      "eval_loss": 0.09941808879375458,
+      "eval_runtime": 2.231,
+      "eval_samples_per_second": 24.204,
+      "eval_steps_per_second": 6.275,
+      "step": 200
+    },
+    {
+      "epoch": 0.30134932533733133,
+      "grad_norm": 0.5319414844042836,
+      "learning_rate": 9.450211418762123e-06,
+      "loss": 0.0547,
+      "step": 201
+    },
+    {
+      "epoch": 0.3028485757121439,
+      "grad_norm": 0.6882255101906093,
+      "learning_rate": 9.444831081131209e-06,
+      "loss": 0.069,
+      "step": 202
+    },
+    {
+      "epoch": 0.30434782608695654,
+      "grad_norm": 0.9277051600723664,
+      "learning_rate": 9.439426092011877e-06,
+      "loss": 0.0893,
+      "step": 203
+    },
+    {
+      "epoch": 0.3058470764617691,
+      "grad_norm": 0.625153783820989,
+      "learning_rate": 9.433996481380747e-06,
+      "loss": 0.0593,
+      "step": 204
+    },
+    {
+      "epoch": 0.3073463268365817,
+      "grad_norm": 0.8515364202107966,
+      "learning_rate": 9.428542279351e-06,
+      "loss": 0.0996,
+      "step": 205
+    },
+    {
+      "epoch": 0.30884557721139433,
+      "grad_norm": 0.816563223887244,
+      "learning_rate": 9.423063516172195e-06,
+      "loss": 0.0878,
+      "step": 206
+    },
+    {
+      "epoch": 0.3103448275862069,
+      "grad_norm": 0.6273694161546717,
+      "learning_rate": 9.417560222230115e-06,
+      "loss": 0.0797,
+      "step": 207
+    },
+    {
+      "epoch": 0.3118440779610195,
+      "grad_norm": 0.7372900009455298,
+      "learning_rate": 9.412032428046594e-06,
+      "loss": 0.0799,
+      "step": 208
+    },
+    {
+      "epoch": 0.31334332833583206,
+      "grad_norm": 0.8105467012454688,
+      "learning_rate": 9.40648016427934e-06,
+      "loss": 0.1015,
+      "step": 209
+    },
+    {
+      "epoch": 0.3148425787106447,
+      "grad_norm": 0.7081568689121497,
+      "learning_rate": 9.400903461721783e-06,
+      "loss": 0.076,
+      "step": 210
+    },
+    {
+      "epoch": 0.31634182908545727,
+      "grad_norm": 0.6648883085980242,
+      "learning_rate": 9.395302351302881e-06,
+      "loss": 0.0779,
+      "step": 211
+    },
+    {
+      "epoch": 0.31784107946026985,
+      "grad_norm": 0.8400398618297222,
+      "learning_rate": 9.38967686408697e-06,
+      "loss": 0.1013,
+      "step": 212
+    },
+    {
+      "epoch": 0.3193403298350825,
+      "grad_norm": 0.8403072330905041,
+      "learning_rate": 9.384027031273575e-06,
+      "loss": 0.0942,
+      "step": 213
+    },
+    {
+      "epoch": 0.32083958020989506,
+      "grad_norm": 0.757837155309957,
+      "learning_rate": 9.37835288419725e-06,
+      "loss": 0.0882,
+      "step": 214
+    },
+    {
+      "epoch": 0.32233883058470764,
+      "grad_norm": 0.7671237414041886,
+      "learning_rate": 9.372654454327394e-06,
+      "loss": 0.1011,
+      "step": 215
+    },
+    {
+      "epoch": 0.3238380809595202,
+      "grad_norm": 0.6323415415360393,
+      "learning_rate": 9.366931773268083e-06,
+      "loss": 0.0856,
+      "step": 216
+    },
+    {
+      "epoch": 0.32533733133433285,
+      "grad_norm": 0.807810803714817,
+      "learning_rate": 9.361184872757894e-06,
+      "loss": 0.0974,
+      "step": 217
+    },
+    {
+      "epoch": 0.3268365817091454,
+      "grad_norm": 0.7776694823574857,
+      "learning_rate": 9.355413784669722e-06,
+      "loss": 0.0798,
+      "step": 218
+    },
+    {
+      "epoch": 0.328335832083958,
+      "grad_norm": 0.7615863245056118,
+      "learning_rate": 9.349618541010616e-06,
+      "loss": 0.084,
+      "step": 219
+    },
+    {
+      "epoch": 0.32983508245877063,
+      "grad_norm": 0.8382761413272126,
+      "learning_rate": 9.343799173921591e-06,
+      "loss": 0.1022,
+      "step": 220
+    },
+    {
+      "epoch": 0.3313343328335832,
+      "grad_norm": 0.6374742060038271,
+      "learning_rate": 9.337955715677452e-06,
+      "loss": 0.0775,
+      "step": 221
+    },
+    {
+      "epoch": 0.3328335832083958,
+      "grad_norm": 0.813595194051477,
+      "learning_rate": 9.332088198686618e-06,
+      "loss": 0.1072,
+      "step": 222
+    },
+    {
+      "epoch": 0.3343328335832084,
+      "grad_norm": 0.64609425671912,
+      "learning_rate": 9.326196655490935e-06,
+      "loss": 0.0734,
+      "step": 223
+    },
+    {
+      "epoch": 0.335832083958021,
+      "grad_norm": 0.7746757097937663,
+      "learning_rate": 9.32028111876551e-06,
+      "loss": 0.0948,
+      "step": 224
+    },
+    {
+      "epoch": 0.3373313343328336,
+      "grad_norm": 0.6264589655144736,
+      "learning_rate": 9.314341621318512e-06,
+      "loss": 0.0896,
+      "step": 225
+    },
+    {
+      "epoch": 0.33883058470764615,
+      "grad_norm": 0.8325668261333805,
+      "learning_rate": 9.308378196091006e-06,
+      "loss": 0.1209,
+      "step": 226
+    },
+    {
+      "epoch": 0.3403298350824588,
+      "grad_norm": 0.7416981316222859,
+      "learning_rate": 9.302390876156756e-06,
+      "loss": 0.0842,
+      "step": 227
+    },
+    {
+      "epoch": 0.34182908545727136,
+      "grad_norm": 0.7701585471629608,
+      "learning_rate": 9.296379694722051e-06,
+      "loss": 0.1028,
+      "step": 228
+    },
+    {
+      "epoch": 0.34332833583208394,
+      "grad_norm": 0.6880273274491111,
+      "learning_rate": 9.29034468512552e-06,
+      "loss": 0.0853,
+      "step": 229
+    },
+    {
+      "epoch": 0.3448275862068966,
+      "grad_norm": 0.6039613454989393,
+      "learning_rate": 9.284285880837947e-06,
+      "loss": 0.0749,
+      "step": 230
+    },
+    {
+      "epoch": 0.34632683658170915,
+      "grad_norm": 0.6185853540786429,
+      "learning_rate": 9.278203315462078e-06,
+      "loss": 0.0653,
+      "step": 231
+    },
+    {
+      "epoch": 0.34782608695652173,
+      "grad_norm": 0.581883633370886,
+      "learning_rate": 9.272097022732444e-06,
+      "loss": 0.0742,
+      "step": 232
+    },
+    {
+      "epoch": 0.3493253373313343,
+      "grad_norm": 0.6601678906970294,
+      "learning_rate": 9.26596703651517e-06,
+      "loss": 0.0709,
+      "step": 233
+    },
+    {
+      "epoch": 0.35082458770614694,
+      "grad_norm": 0.7684821756901647,
+      "learning_rate": 9.259813390807788e-06,
+      "loss": 0.1046,
+      "step": 234
+    },
+    {
+      "epoch": 0.3523238380809595,
+      "grad_norm": 0.7897880232335965,
+      "learning_rate": 9.253636119739046e-06,
+      "loss": 0.0846,
+      "step": 235
+    },
+    {
+      "epoch": 0.3538230884557721,
+      "grad_norm": 0.819836464447015,
+      "learning_rate": 9.247435257568724e-06,
+      "loss": 0.0846,
+      "step": 236
+    },
+    {
+      "epoch": 0.3553223388305847,
+      "grad_norm": 0.7153799211795385,
+      "learning_rate": 9.241210838687438e-06,
+      "loss": 0.0901,
+      "step": 237
+    },
+    {
+      "epoch": 0.3568215892053973,
+      "grad_norm": 0.733346924602415,
+      "learning_rate": 9.23496289761645e-06,
+      "loss": 0.0843,
+      "step": 238
+    },
+    {
+      "epoch": 0.3583208395802099,
+      "grad_norm": 0.8716584377918819,
+      "learning_rate": 9.228691469007487e-06,
+      "loss": 0.1079,
+      "step": 239
+    },
+    {
+      "epoch": 0.3598200899550225,
+      "grad_norm": 0.612071074291601,
+      "learning_rate": 9.222396587642528e-06,
+      "loss": 0.0656,
+      "step": 240
+    },
+    {
+      "epoch": 0.3613193403298351,
+      "grad_norm": 0.6823708799913535,
+      "learning_rate": 9.216078288433632e-06,
+      "loss": 0.08,
+      "step": 241
+    },
+    {
+      "epoch": 0.36281859070464767,
+      "grad_norm": 0.7526037269390174,
+      "learning_rate": 9.209736606422736e-06,
+      "loss": 0.0944,
+      "step": 242
+    },
+    {
+      "epoch": 0.36431784107946025,
+      "grad_norm": 0.7558677633807892,
+      "learning_rate": 9.203371576781457e-06,
+      "loss": 0.0872,
+      "step": 243
+    },
+    {
+      "epoch": 0.3658170914542729,
+      "grad_norm": 0.6274047042699632,
+      "learning_rate": 9.1969832348109e-06,
+      "loss": 0.0647,
+      "step": 244
+    },
+    {
+      "epoch": 0.36731634182908546,
+      "grad_norm": 0.6832170192735554,
+      "learning_rate": 9.190571615941462e-06,
+      "loss": 0.0837,
+      "step": 245
+    },
+    {
+      "epoch": 0.36881559220389803,
+      "grad_norm": 0.7127052599608549,
+      "learning_rate": 9.18413675573264e-06,
+      "loss": 0.066,
+      "step": 246
+    },
+    {
+      "epoch": 0.37031484257871067,
+      "grad_norm": 0.866742285459009,
+      "learning_rate": 9.177678689872831e-06,
+      "loss": 0.1212,
+      "step": 247
+    },
+    {
+      "epoch": 0.37181409295352325,
+      "grad_norm": 0.5451843494101154,
+      "learning_rate": 9.171197454179124e-06,
+      "loss": 0.0566,
+      "step": 248
+    },
+    {
+      "epoch": 0.3733133433283358,
+      "grad_norm": 0.7014078594513409,
+      "learning_rate": 9.16469308459712e-06,
+      "loss": 0.077,
+      "step": 249
+    },
+    {
+      "epoch": 0.3748125937031484,
+      "grad_norm": 0.7021596013101258,
+      "learning_rate": 9.158165617200717e-06,
+      "loss": 0.0702,
+      "step": 250
+    },
+    {
+      "epoch": 0.37631184407796103,
+      "grad_norm": 0.8175650166397985,
+      "learning_rate": 9.151615088191918e-06,
+      "loss": 0.0822,
+      "step": 251
+    },
+    {
+      "epoch": 0.3778110944527736,
+      "grad_norm": 0.5553959160218326,
+      "learning_rate": 9.14504153390063e-06,
+      "loss": 0.0665,
+      "step": 252
+    },
+    {
+      "epoch": 0.3793103448275862,
+      "grad_norm": 0.8294031742111282,
+      "learning_rate": 9.138444990784455e-06,
+      "loss": 0.1102,
+      "step": 253
+    },
+    {
+      "epoch": 0.3808095952023988,
+      "grad_norm": 0.6306623344677106,
+      "learning_rate": 9.131825495428496e-06,
+      "loss": 0.0657,
+      "step": 254
+    },
+    {
+      "epoch": 0.3823088455772114,
+      "grad_norm": 0.7246447806402195,
+      "learning_rate": 9.125183084545158e-06,
+      "loss": 0.0969,
+      "step": 255
+    },
+    {
+      "epoch": 0.383808095952024,
+      "grad_norm": 0.7820118693350296,
+      "learning_rate": 9.118517794973925e-06,
+      "loss": 0.0915,
+      "step": 256
+    },
+    {
+      "epoch": 0.3853073463268366,
+      "grad_norm": 0.6885709054524785,
+      "learning_rate": 9.111829663681182e-06,
+      "loss": 0.0855,
+      "step": 257
+    },
+    {
+      "epoch": 0.3868065967016492,
+      "grad_norm": 0.8124154715450759,
+      "learning_rate": 9.105118727759984e-06,
+      "loss": 0.0878,
+      "step": 258
+    },
+    {
+      "epoch": 0.38830584707646176,
+      "grad_norm": 0.9290657038381838,
+      "learning_rate": 9.098385024429875e-06,
+      "loss": 0.1068,
+      "step": 259
+    },
+    {
+      "epoch": 0.38980509745127434,
+      "grad_norm": 0.5580177285258425,
+      "learning_rate": 9.09162859103666e-06,
+      "loss": 0.0676,
+      "step": 260
+    },
+    {
+      "epoch": 0.391304347826087,
+      "grad_norm": 0.7973273992760974,
+      "learning_rate": 9.08484946505221e-06,
+      "loss": 0.0914,
+      "step": 261
+    },
+    {
+      "epoch": 0.39280359820089955,
+      "grad_norm": 0.7937433167816241,
+      "learning_rate": 9.078047684074254e-06,
+      "loss": 0.0925,
+      "step": 262
+    },
+    {
+      "epoch": 0.39430284857571213,
+      "grad_norm": 0.7426726613360578,
+      "learning_rate": 9.071223285826166e-06,
+      "loss": 0.089,
+      "step": 263
+    },
+    {
+      "epoch": 0.39580209895052476,
+      "grad_norm": 0.6982766818727655,
+      "learning_rate": 9.064376308156754e-06,
+      "loss": 0.0834,
+      "step": 264
+    },
+    {
+      "epoch": 0.39730134932533734,
+      "grad_norm": 0.7641272310656788,
+      "learning_rate": 9.057506789040063e-06,
+      "loss": 0.0969,
+      "step": 265
+    },
+    {
+      "epoch": 0.3988005997001499,
+      "grad_norm": 0.7517598568650622,
+      "learning_rate": 9.050614766575147e-06,
+      "loss": 0.1022,
+      "step": 266
+    },
+    {
+      "epoch": 0.4002998500749625,
+      "grad_norm": 0.641243618473385,
+      "learning_rate": 9.043700278985867e-06,
+      "loss": 0.0669,
+      "step": 267
+    },
+    {
+      "epoch": 0.4017991004497751,
+      "grad_norm": 0.7544361510591047,
+      "learning_rate": 9.03676336462068e-06,
+      "loss": 0.096,
+      "step": 268
+    },
+    {
+      "epoch": 0.4032983508245877,
+      "grad_norm": 0.7113079357840938,
+      "learning_rate": 9.029804061952426e-06,
+      "loss": 0.0914,
+      "step": 269
+    },
+    {
+      "epoch": 0.4047976011994003,
+      "grad_norm": 0.544112439965787,
+      "learning_rate": 9.022822409578106e-06,
+      "loss": 0.0636,
+      "step": 270
+    },
+    {
+      "epoch": 0.4062968515742129,
+      "grad_norm": 0.6209733506425709,
+      "learning_rate": 9.015818446218683e-06,
+      "loss": 0.0609,
+      "step": 271
+    },
+    {
+      "epoch": 0.4077961019490255,
+      "grad_norm": 0.7904704158968802,
+      "learning_rate": 9.008792210718854e-06,
+      "loss": 0.0962,
+      "step": 272
+    },
+    {
+      "epoch": 0.40929535232383807,
+      "grad_norm": 1.0107032974852084,
+      "learning_rate": 9.00174374204684e-06,
+      "loss": 0.1386,
+      "step": 273
+    },
+    {
+      "epoch": 0.4107946026986507,
+      "grad_norm": 0.702177433513673,
+      "learning_rate": 8.994673079294171e-06,
+      "loss": 0.0828,
+      "step": 274
+    },
+    {
+      "epoch": 0.4122938530734633,
+      "grad_norm": 0.8021176692957329,
+      "learning_rate": 8.987580261675466e-06,
+      "loss": 0.0779,
+      "step": 275
+    },
+    {
+      "epoch": 0.41379310344827586,
+      "grad_norm": 0.7631108629568449,
+      "learning_rate": 8.98046532852822e-06,
+      "loss": 0.0886,
+      "step": 276
+    },
+    {
+      "epoch": 0.41529235382308843,
+      "grad_norm": 0.6715586012610685,
+      "learning_rate": 8.973328319312577e-06,
+      "loss": 0.0671,
+      "step": 277
+    },
+    {
+      "epoch": 0.41679160419790107,
+      "grad_norm": 0.6443845697349361,
+      "learning_rate": 8.966169273611125e-06,
+      "loss": 0.0782,
+      "step": 278
+    },
+    {
+      "epoch": 0.41829085457271364,
+      "grad_norm": 0.8412230748162437,
+      "learning_rate": 8.958988231128665e-06,
+      "loss": 0.1046,
+      "step": 279
+    },
+    {
+      "epoch": 0.4197901049475262,
+      "grad_norm": 0.654408392682262,
+      "learning_rate": 8.95178523169199e-06,
+      "loss": 0.0828,
+      "step": 280
+    },
+    {
+      "epoch": 0.42128935532233885,
+      "grad_norm": 0.6376862130454409,
+      "learning_rate": 8.944560315249676e-06,
+      "loss": 0.0709,
+      "step": 281
+    },
+    {
+      "epoch": 0.42278860569715143,
+      "grad_norm": 0.6705286412295577,
+      "learning_rate": 8.937313521871846e-06,
+      "loss": 0.0774,
+      "step": 282
+    },
+    {
+      "epoch": 0.424287856071964,
+      "grad_norm": 0.721456724432131,
+      "learning_rate": 8.930044891749962e-06,
+      "loss": 0.0854,
+      "step": 283
+    },
+    {
+      "epoch": 0.4257871064467766,
+      "grad_norm": 0.7860743382070283,
+      "learning_rate": 8.922754465196591e-06,
+      "loss": 0.1073,
+      "step": 284
+    },
+    {
+      "epoch": 0.4272863568215892,
+      "grad_norm": 0.6216348039053096,
+      "learning_rate": 8.915442282645183e-06,
+      "loss": 0.0722,
+      "step": 285
+    },
+    {
+      "epoch": 0.4287856071964018,
+      "grad_norm": 0.6732204352920048,
+      "learning_rate": 8.908108384649856e-06,
+      "loss": 0.0858,
+      "step": 286
+    },
+    {
+      "epoch": 0.4302848575712144,
+      "grad_norm": 0.7239344626732085,
+      "learning_rate": 8.900752811885152e-06,
+      "loss": 0.0919,
+      "step": 287
+    },
+    {
+      "epoch": 0.431784107946027,
+      "grad_norm": 0.5123785637295392,
+      "learning_rate": 8.893375605145837e-06,
+      "loss": 0.0559,
+      "step": 288
+    },
+    {
+      "epoch": 0.4332833583208396,
+      "grad_norm": 0.7870109472188238,
+      "learning_rate": 8.885976805346651e-06,
+      "loss": 0.1105,
+      "step": 289
+    },
+    {
+      "epoch": 0.43478260869565216,
+      "grad_norm": 0.7241713492731523,
+      "learning_rate": 8.8785564535221e-06,
+      "loss": 0.0787,
+      "step": 290
+    },
+    {
+      "epoch": 0.4362818590704648,
+      "grad_norm": 0.6456174195311722,
+      "learning_rate": 8.871114590826211e-06,
+      "loss": 0.0768,
+      "step": 291
+    },
+    {
+      "epoch": 0.43778110944527737,
+      "grad_norm": 0.9882934444731084,
+      "learning_rate": 8.86365125853232e-06,
+      "loss": 0.1054,
+      "step": 292
+    },
+    {
+      "epoch": 0.43928035982008995,
+      "grad_norm": 0.6621389434921209,
+      "learning_rate": 8.85616649803283e-06,
+      "loss": 0.0791,
+      "step": 293
+    },
+    {
+      "epoch": 0.4407796101949025,
+      "grad_norm": 0.5689995982245669,
+      "learning_rate": 8.84866035083899e-06,
+      "loss": 0.0681,
+      "step": 294
+    },
+    {
+      "epoch": 0.44227886056971516,
+      "grad_norm": 0.8712796273000644,
+      "learning_rate": 8.841132858580661e-06,
+      "loss": 0.1102,
+      "step": 295
+    },
+    {
+      "epoch": 0.44377811094452774,
+      "grad_norm": 0.6556874685658192,
+      "learning_rate": 8.833584063006088e-06,
+      "loss": 0.0844,
+      "step": 296
+    },
+    {
+      "epoch": 0.4452773613193403,
+      "grad_norm": 0.7017079868048285,
+      "learning_rate": 8.826014005981662e-06,
+      "loss": 0.0789,
+      "step": 297
+    },
+    {
+      "epoch": 0.44677661169415295,
+      "grad_norm": 0.7540851094370302,
+      "learning_rate": 8.818422729491693e-06,
+      "loss": 0.0879,
+      "step": 298
+    },
+    {
+      "epoch": 0.4482758620689655,
+      "grad_norm": 0.6958967451530746,
+      "learning_rate": 8.810810275638183e-06,
+      "loss": 0.0931,
+      "step": 299
+    },
+    {
+      "epoch": 0.4497751124437781,
+      "grad_norm": 0.5976089373113932,
+      "learning_rate": 8.803176686640577e-06,
+      "loss": 0.0762,
+      "step": 300
+    },
+    {
+      "epoch": 0.4512743628185907,
+      "grad_norm": 0.8933202221454002,
+      "learning_rate": 8.795522004835543e-06,
+      "loss": 0.1106,
+      "step": 301
+    },
+    {
+      "epoch": 0.4527736131934033,
+      "grad_norm": 0.7155681547353292,
+      "learning_rate": 8.787846272676728e-06,
+      "loss": 0.0729,
+      "step": 302
+    },
+    {
+      "epoch": 0.4542728635682159,
+      "grad_norm": 0.5628737933338819,
+      "learning_rate": 8.780149532734531e-06,
+      "loss": 0.0637,
+      "step": 303
+    },
+    {
+      "epoch": 0.45577211394302847,
+      "grad_norm": 0.7719431997317473,
+      "learning_rate": 8.772431827695862e-06,
+      "loss": 0.0983,
+      "step": 304
+    },
+    {
+      "epoch": 0.4572713643178411,
+      "grad_norm": 0.7693722068533134,
+      "learning_rate": 8.764693200363897e-06,
+      "loss": 0.0816,
+      "step": 305
+    },
+    {
+      "epoch": 0.4587706146926537,
+      "grad_norm": 0.6392153458887491,
+      "learning_rate": 8.756933693657863e-06,
+      "loss": 0.0659,
+      "step": 306
+    },
+    {
+      "epoch": 0.46026986506746626,
+      "grad_norm": 0.7138968858661271,
+      "learning_rate": 8.749153350612774e-06,
+      "loss": 0.0828,
+      "step": 307
+    },
+    {
+      "epoch": 0.4617691154422789,
+      "grad_norm": 0.7139334588578117,
+      "learning_rate": 8.74135221437921e-06,
+      "loss": 0.0918,
+      "step": 308
+    },
+    {
+      "epoch": 0.46326836581709147,
+      "grad_norm": 0.6191357712222109,
+      "learning_rate": 8.733530328223076e-06,
+      "loss": 0.065,
+      "step": 309
+    },
+    {
+      "epoch": 0.46476761619190404,
+      "grad_norm": 0.6105810674146535,
+      "learning_rate": 8.725687735525347e-06,
+      "loss": 0.0696,
+      "step": 310
+    },
+    {
+      "epoch": 0.4662668665667166,
+      "grad_norm": 0.8479721960167917,
+      "learning_rate": 8.71782447978185e-06,
+      "loss": 0.1061,
+      "step": 311
+    },
+    {
+      "epoch": 0.46776611694152925,
+      "grad_norm": 0.7875057454316355,
+      "learning_rate": 8.709940604603006e-06,
+      "loss": 0.1064,
+      "step": 312
+    },
+    {
+      "epoch": 0.46926536731634183,
+      "grad_norm": 0.7039132365453881,
+      "learning_rate": 8.702036153713594e-06,
+      "loss": 0.0779,
+      "step": 313
+    },
+    {
+      "epoch": 0.4707646176911544,
+      "grad_norm": 0.8509629823362098,
+      "learning_rate": 8.694111170952508e-06,
+      "loss": 0.1029,
+      "step": 314
+    },
+    {
+      "epoch": 0.47226386806596704,
+      "grad_norm": 0.5737385398809762,
+      "learning_rate": 8.686165700272513e-06,
+      "loss": 0.0788,
+      "step": 315
+    },
+    {
+      "epoch": 0.4737631184407796,
+      "grad_norm": 0.7084130225431673,
+      "learning_rate": 8.678199785740003e-06,
+      "loss": 0.0912,
+      "step": 316
+    },
+    {
+      "epoch": 0.4752623688155922,
+      "grad_norm": 0.6732650251289916,
+      "learning_rate": 8.670213471534759e-06,
+      "loss": 0.0916,
+      "step": 317
+    },
+    {
+      "epoch": 0.4767616191904048,
+      "grad_norm": 0.685666929346874,
+      "learning_rate": 8.662206801949694e-06,
+      "loss": 0.0749,
+      "step": 318
+    },
+    {
+      "epoch": 0.4782608695652174,
+      "grad_norm": 0.6283538299835257,
+      "learning_rate": 8.65417982139062e-06,
+      "loss": 0.0636,
+      "step": 319
+    },
+    {
+      "epoch": 0.47976011994003,
+      "grad_norm": 0.7957412875676163,
+      "learning_rate": 8.646132574375994e-06,
+      "loss": 0.0869,
+      "step": 320
+    },
+    {
+      "epoch": 0.48125937031484256,
+      "grad_norm": 0.8905872971177238,
+      "learning_rate": 8.638065105536669e-06,
+      "loss": 0.11,
+      "step": 321
+    },
+    {
+      "epoch": 0.4827586206896552,
+      "grad_norm": 0.7341726654057851,
+      "learning_rate": 8.629977459615655e-06,
+      "loss": 0.0746,
+      "step": 322
+    },
+    {
+      "epoch": 0.48425787106446777,
+      "grad_norm": 0.7803873943039987,
+      "learning_rate": 8.621869681467865e-06,
+      "loss": 0.0913,
+      "step": 323
+    },
+    {
+      "epoch": 0.48575712143928035,
+      "grad_norm": 0.6742758560842088,
+      "learning_rate": 8.613741816059867e-06,
+      "loss": 0.0737,
+      "step": 324
+    },
+    {
+      "epoch": 0.487256371814093,
+      "grad_norm": 0.9203687614991732,
+      "learning_rate": 8.605593908469635e-06,
+      "loss": 0.0889,
+      "step": 325
+    },
+    {
+      "epoch": 0.48875562218890556,
+      "grad_norm": 0.6314601984023058,
+      "learning_rate": 8.597426003886295e-06,
+      "loss": 0.0806,
+      "step": 326
+    },
+    {
+      "epoch": 0.49025487256371814,
+      "grad_norm": 0.7682962414081259,
+      "learning_rate": 8.58923814760989e-06,
+      "loss": 0.0876,
+      "step": 327
+    },
+    {
+      "epoch": 0.4917541229385307,
+      "grad_norm": 0.6534556866502667,
+      "learning_rate": 8.581030385051105e-06,
+      "loss": 0.0749,
+      "step": 328
+    },
+    {
+      "epoch": 0.49325337331334335,
+      "grad_norm": 0.6996418840014252,
+      "learning_rate": 8.572802761731031e-06,
+      "loss": 0.0867,
+      "step": 329
+    },
+    {
+      "epoch": 0.4947526236881559,
+      "grad_norm": 0.9718795540341122,
+      "learning_rate": 8.564555323280913e-06,
+      "loss": 0.1155,
+      "step": 330
+    },
+    {
+      "epoch": 0.4962518740629685,
+      "grad_norm": 0.7574643941024478,
+      "learning_rate": 8.556288115441887e-06,
+      "loss": 0.0831,
+      "step": 331
+    },
+    {
+      "epoch": 0.49775112443778113,
+      "grad_norm": 0.6604828521528302,
+      "learning_rate": 8.548001184064733e-06,
+      "loss": 0.0772,
+      "step": 332
+    },
+    {
+      "epoch": 0.4992503748125937,
+      "grad_norm": 0.637270018943202,
+      "learning_rate": 8.539694575109626e-06,
+      "loss": 0.0648,
+      "step": 333
+    },
+    {
+      "epoch": 0.5007496251874063,
+      "grad_norm": 0.6092105980447031,
+      "learning_rate": 8.531368334645865e-06,
+      "loss": 0.0719,
+      "step": 334
+    },
+    {
+      "epoch": 0.5022488755622189,
+      "grad_norm": 0.6338533462614352,
+      "learning_rate": 8.523022508851634e-06,
+      "loss": 0.1002,
+      "step": 335
+    },
+    {
+      "epoch": 0.5037481259370314,
+      "grad_norm": 0.6385259498287165,
+      "learning_rate": 8.514657144013738e-06,
+      "loss": 0.0753,
+      "step": 336
+    },
+    {
+      "epoch": 0.5052473763118441,
+      "grad_norm": 0.6095346995719176,
+      "learning_rate": 8.506272286527346e-06,
+      "loss": 0.0708,
+      "step": 337
+    },
+    {
+      "epoch": 0.5067466266866567,
+      "grad_norm": 0.6533362794203948,
+      "learning_rate": 8.497867982895741e-06,
+      "loss": 0.0669,
+      "step": 338
+    },
+    {
+      "epoch": 0.5082458770614693,
+      "grad_norm": 0.603685348246502,
+      "learning_rate": 8.489444279730046e-06,
+      "loss": 0.0728,
+      "step": 339
+    },
+    {
+      "epoch": 0.5097451274362819,
+      "grad_norm": 0.7077982391462756,
+      "learning_rate": 8.481001223748986e-06,
+      "loss": 0.0579,
+      "step": 340
+    },
+    {
+      "epoch": 0.5112443778110944,
+      "grad_norm": 0.8106023014639854,
+      "learning_rate": 8.47253886177861e-06,
+      "loss": 0.1063,
+      "step": 341
+    },
+    {
+      "epoch": 0.512743628185907,
+      "grad_norm": 0.763919936558653,
+      "learning_rate": 8.464057240752046e-06,
+      "loss": 0.0922,
+      "step": 342
+    },
+    {
+      "epoch": 0.5142428785607196,
+      "grad_norm": 0.6733042591119659,
+      "learning_rate": 8.455556407709235e-06,
+      "loss": 0.0758,
+      "step": 343
+    },
+    {
+      "epoch": 0.5157421289355323,
+      "grad_norm": 0.6862095966257666,
+      "learning_rate": 8.447036409796663e-06,
+      "loss": 0.0861,
+      "step": 344
+    },
+    {
+      "epoch": 0.5172413793103449,
+      "grad_norm": 0.6412493505015119,
+      "learning_rate": 8.438497294267117e-06,
+      "loss": 0.0827,
+      "step": 345
+    },
+    {
+      "epoch": 0.5187406296851574,
+      "grad_norm": 0.7992397432269214,
+      "learning_rate": 8.429939108479403e-06,
+      "loss": 0.085,
+      "step": 346
+    },
+    {
+      "epoch": 0.52023988005997,
+      "grad_norm": 0.7403088906656962,
+      "learning_rate": 8.421361899898095e-06,
+      "loss": 0.0774,
+      "step": 347
+    },
+    {
+      "epoch": 0.5217391304347826,
+      "grad_norm": 0.6159479758139523,
+      "learning_rate": 8.412765716093273e-06,
+      "loss": 0.0762,
+      "step": 348
+    },
+    {
+      "epoch": 0.5232383808095952,
+      "grad_norm": 0.8346530553605869,
+      "learning_rate": 8.404150604740248e-06,
+      "loss": 0.1093,
+      "step": 349
+    },
+    {
+      "epoch": 0.5247376311844077,
+      "grad_norm": 0.752428974679364,
+      "learning_rate": 8.395516613619315e-06,
+      "loss": 0.0725,
+      "step": 350
+    },
+    {
+      "epoch": 0.5262368815592204,
+      "grad_norm": 0.5568442783108936,
+      "learning_rate": 8.386863790615472e-06,
+      "loss": 0.0631,
+      "step": 351
+    },
+    {
+      "epoch": 0.527736131934033,
+      "grad_norm": 0.7919019337979545,
+      "learning_rate": 8.378192183718158e-06,
+      "loss": 0.0905,
+      "step": 352
+    },
+    {
+      "epoch": 0.5292353823088456,
+      "grad_norm": 0.8522334607487108,
+      "learning_rate": 8.369501841021e-06,
+      "loss": 0.1085,
+      "step": 353
+    },
+    {
+      "epoch": 0.5307346326836582,
+      "grad_norm": 0.570426536418451,
+      "learning_rate": 8.360792810721522e-06,
+      "loss": 0.0706,
+      "step": 354
+    },
+    {
+      "epoch": 0.5322338830584707,
+      "grad_norm": 0.7812210753254766,
+      "learning_rate": 8.352065141120902e-06,
+      "loss": 0.0964,
+      "step": 355
+    },
+    {
+      "epoch": 0.5337331334332833,
+      "grad_norm": 0.8508687786296654,
+      "learning_rate": 8.343318880623688e-06,
+      "loss": 0.0913,
+      "step": 356
+    },
+    {
+      "epoch": 0.5352323838080959,
+      "grad_norm": 0.8125781595021724,
+      "learning_rate": 8.334554077737535e-06,
+      "loss": 0.0944,
+      "step": 357
+    },
+    {
+      "epoch": 0.5367316341829086,
+      "grad_norm": 0.7260839802058119,
+      "learning_rate": 8.325770781072939e-06,
+      "loss": 0.0786,
+      "step": 358
+    },
+    {
+      "epoch": 0.5382308845577212,
+      "grad_norm": 0.56201930435168,
+      "learning_rate": 8.316969039342963e-06,
+      "loss": 0.0746,
+      "step": 359
+    },
+    {
+      "epoch": 0.5397301349325337,
+      "grad_norm": 0.735205727543574,
+      "learning_rate": 8.30814890136297e-06,
+      "loss": 0.1063,
+      "step": 360
+    },
+    {
+      "epoch": 0.5412293853073463,
+      "grad_norm": 0.8176541078895385,
+      "learning_rate": 8.299310416050345e-06,
+      "loss": 0.1218,
+      "step": 361
+    },
+    {
+      "epoch": 0.5427286356821589,
+      "grad_norm": 0.6706958198392852,
+      "learning_rate": 8.290453632424236e-06,
+      "loss": 0.0861,
+      "step": 362
+    },
+    {
+      "epoch": 0.5442278860569715,
+      "grad_norm": 0.792353784855582,
+      "learning_rate": 8.281578599605269e-06,
+      "loss": 0.0838,
+      "step": 363
+    },
+    {
+      "epoch": 0.545727136431784,
+      "grad_norm": 0.8053107274693725,
+      "learning_rate": 8.272685366815287e-06,
+      "loss": 0.1099,
+      "step": 364
+    },
+    {
+      "epoch": 0.5472263868065967,
+      "grad_norm": 0.7187424928818651,
+      "learning_rate": 8.26377398337707e-06,
+      "loss": 0.0737,
+      "step": 365
+    },
+    {
+      "epoch": 0.5487256371814093,
+      "grad_norm": 0.6705572702606731,
+      "learning_rate": 8.254844498714063e-06,
+      "loss": 0.0856,
+      "step": 366
+    },
+    {
+      "epoch": 0.5502248875562219,
+      "grad_norm": 0.71907936286629,
+      "learning_rate": 8.2458969623501e-06,
+      "loss": 0.0997,
+      "step": 367
+    },
+    {
+      "epoch": 0.5517241379310345,
+      "grad_norm": 0.651133279357392,
+      "learning_rate": 8.23693142390914e-06,
+      "loss": 0.083,
+      "step": 368
+    },
+    {
+      "epoch": 0.553223388305847,
+      "grad_norm": 0.72531870041849,
+      "learning_rate": 8.227947933114971e-06,
+      "loss": 0.0959,
+      "step": 369
+    },
+    {
+      "epoch": 0.5547226386806596,
+      "grad_norm": 0.750110589830587,
+      "learning_rate": 8.218946539790957e-06,
+      "loss": 0.0912,
+      "step": 370
+    },
+    {
+      "epoch": 0.5562218890554723,
+      "grad_norm": 0.6136292367423635,
+      "learning_rate": 8.209927293859746e-06,
+      "loss": 0.0752,
+      "step": 371
+    },
+    {
+      "epoch": 0.5577211394302849,
+      "grad_norm": 0.6780437110129847,
+      "learning_rate": 8.200890245342999e-06,
+      "loss": 0.0807,
+      "step": 372
+    },
+    {
+      "epoch": 0.5592203898050975,
+      "grad_norm": 0.8734735205271278,
+      "learning_rate": 8.191835444361113e-06,
+      "loss": 0.1285,
+      "step": 373
+    },
+    {
+      "epoch": 0.56071964017991,
+      "grad_norm": 0.681944814527203,
+      "learning_rate": 8.182762941132944e-06,
+      "loss": 0.0687,
+      "step": 374
+    },
+    {
+      "epoch": 0.5622188905547226,
+      "grad_norm": 0.7210238017211748,
+      "learning_rate": 8.173672785975522e-06,
+      "loss": 0.0694,
+      "step": 375
+    },
+    {
+      "epoch": 0.5637181409295352,
+      "grad_norm": 0.7186060209489308,
+      "learning_rate": 8.16456502930378e-06,
+      "loss": 0.0869,
+      "step": 376
+    },
+    {
+      "epoch": 0.5652173913043478,
+      "grad_norm": 0.7624428391221892,
+      "learning_rate": 8.155439721630265e-06,
+      "loss": 0.0979,
+      "step": 377
+    },
+    {
+      "epoch": 0.5667166416791605,
+      "grad_norm": 0.6660588778669224,
+      "learning_rate": 8.146296913564872e-06,
+      "loss": 0.0777,
+      "step": 378
+    },
+    {
+      "epoch": 0.568215892053973,
+      "grad_norm": 0.6925135170577529,
+      "learning_rate": 8.13713665581455e-06,
+      "loss": 0.0791,
+      "step": 379
+    },
+    {
+      "epoch": 0.5697151424287856,
+      "grad_norm": 0.6016766898192409,
+      "learning_rate": 8.127958999183027e-06,
+      "loss": 0.0707,
+      "step": 380
+    },
+    {
+      "epoch": 0.5712143928035982,
+      "grad_norm": 0.6365680145834911,
+      "learning_rate": 8.118763994570528e-06,
+      "loss": 0.0629,
+      "step": 381
+    },
+    {
+      "epoch": 0.5727136431784108,
+      "grad_norm": 0.6970245313825577,
+      "learning_rate": 8.109551692973487e-06,
+      "loss": 0.08,
+      "step": 382
+    },
+    {
+      "epoch": 0.5742128935532234,
+      "grad_norm": 0.8744350704734656,
+      "learning_rate": 8.100322145484275e-06,
+      "loss": 0.1046,
+      "step": 383
+    },
+    {
+      "epoch": 0.5757121439280359,
+      "grad_norm": 0.7256834242872856,
+      "learning_rate": 8.091075403290905e-06,
+      "loss": 0.0962,
+      "step": 384
+    },
+    {
+      "epoch": 0.5772113943028486,
+      "grad_norm": 0.719283715522556,
+      "learning_rate": 8.081811517676759e-06,
+      "loss": 0.0888,
+      "step": 385
+    },
+    {
+      "epoch": 0.5787106446776612,
+      "grad_norm": 0.6444612488851637,
+      "learning_rate": 8.072530540020294e-06,
+      "loss": 0.0817,
+      "step": 386
+    },
+    {
+      "epoch": 0.5802098950524738,
+      "grad_norm": 0.7421522730157731,
+      "learning_rate": 8.063232521794762e-06,
+      "loss": 0.094,
+      "step": 387
+    },
+    {
+      "epoch": 0.5817091454272864,
+      "grad_norm": 0.5370944777746403,
+      "learning_rate": 8.053917514567927e-06,
+      "loss": 0.0724,
+      "step": 388
+    },
+    {
+      "epoch": 0.5832083958020989,
+      "grad_norm": 0.8659241763387276,
+      "learning_rate": 8.04458557000177e-06,
+      "loss": 0.0868,
+      "step": 389
+    },
+    {
+      "epoch": 0.5847076461769115,
+      "grad_norm": 0.8305657541991728,
+      "learning_rate": 8.035236739852214e-06,
+      "loss": 0.1155,
+      "step": 390
+    },
+    {
+      "epoch": 0.5862068965517241,
+      "grad_norm": 0.6099385984800269,
+      "learning_rate": 8.025871075968828e-06,
+      "loss": 0.0563,
+      "step": 391
+    },
+    {
+      "epoch": 0.5877061469265368,
+      "grad_norm": 0.6239697556570968,
+      "learning_rate": 8.016488630294539e-06,
+      "loss": 0.0706,
+      "step": 392
+    },
+    {
+      "epoch": 0.5892053973013494,
+      "grad_norm": 0.7426407361018603,
+      "learning_rate": 8.007089454865358e-06,
+      "loss": 0.0799,
+      "step": 393
+    },
+    {
+      "epoch": 0.5907046476761619,
+      "grad_norm": 0.7539535961515794,
+      "learning_rate": 7.997673601810071e-06,
+      "loss": 0.0845,
+      "step": 394
+    },
+    {
+      "epoch": 0.5922038980509745,
+      "grad_norm": 0.5725321243656698,
+      "learning_rate": 7.988241123349965e-06,
+      "loss": 0.0658,
+      "step": 395
+    },
+    {
+      "epoch": 0.5937031484257871,
+      "grad_norm": 0.7906282311685945,
+      "learning_rate": 7.97879207179853e-06,
+      "loss": 0.0978,
+      "step": 396
+    },
+    {
+      "epoch": 0.5952023988005997,
+      "grad_norm": 0.6562300353370503,
+      "learning_rate": 7.969326499561173e-06,
+      "loss": 0.0823,
+      "step": 397
+    },
+    {
+      "epoch": 0.5967016491754122,
+      "grad_norm": 0.6450372366485762,
+      "learning_rate": 7.95984445913493e-06,
+      "loss": 0.0703,
+      "step": 398
+    },
+    {
+      "epoch": 0.5982008995502249,
+      "grad_norm": 0.617328048741293,
+      "learning_rate": 7.950346003108167e-06,
+      "loss": 0.0747,
+      "step": 399
+    },
+    {
+      "epoch": 0.5997001499250375,
+      "grad_norm": 0.6531430525483682,
+      "learning_rate": 7.940831184160294e-06,
+      "loss": 0.0721,
+      "step": 400
+    },
+    {
+      "epoch": 0.5997001499250375,
+      "eval_loss": 0.09513034671545029,
+      "eval_runtime": 2.1832,
+      "eval_samples_per_second": 24.734,
+      "eval_steps_per_second": 6.412,
+      "step": 400
+    },
+    {
+      "epoch": 0.6011994002998501,
+      "grad_norm": 0.5627628114523238,
+      "learning_rate": 7.93130005506147e-06,
+      "loss": 0.0623,
+      "step": 401
+    },
+    {
+      "epoch": 0.6026986506746627,
+      "grad_norm": 0.6721027213234425,
+      "learning_rate": 7.921752668672316e-06,
+      "loss": 0.0953,
+      "step": 402
+    },
+    {
+      "epoch": 0.6041979010494752,
+      "grad_norm": 0.6977577391656965,
+      "learning_rate": 7.912189077943613e-06,
+      "loss": 0.0782,
+      "step": 403
+    },
+    {
+      "epoch": 0.6056971514242878,
+      "grad_norm": 0.7115212376944359,
+      "learning_rate": 7.902609335916015e-06,
+      "loss": 0.0751,
+      "step": 404
+    },
+    {
+      "epoch": 0.6071964017991005,
+      "grad_norm": 0.7234888194062179,
+      "learning_rate": 7.893013495719752e-06,
+      "loss": 0.0894,
+      "step": 405
+    },
+    {
+      "epoch": 0.6086956521739131,
+      "grad_norm": 0.7924783788425257,
+      "learning_rate": 7.883401610574338e-06,
+      "loss": 0.0939,
+      "step": 406
+    },
+    {
+      "epoch": 0.6101949025487257,
+      "grad_norm": 0.7525238212020555,
+      "learning_rate": 7.873773733788268e-06,
+      "loss": 0.0986,
+      "step": 407
+    },
+    {
+      "epoch": 0.6116941529235382,
+      "grad_norm": 0.9558203015106649,
+      "learning_rate": 7.864129918758738e-06,
+      "loss": 0.1115,
+      "step": 408
+    },
+    {
+      "epoch": 0.6131934032983508,
+      "grad_norm": 0.6657696775699985,
+      "learning_rate": 7.854470218971333e-06,
+      "loss": 0.0853,
+      "step": 409
+    },
+    {
+      "epoch": 0.6146926536731634,
+      "grad_norm": 0.7170328249961648,
+      "learning_rate": 7.844794687999737e-06,
+      "loss": 0.0838,
+      "step": 410
+    },
+    {
+      "epoch": 0.616191904047976,
+      "grad_norm": 0.8203303978531875,
+      "learning_rate": 7.835103379505433e-06,
+      "loss": 0.101,
+      "step": 411
+    },
+    {
+      "epoch": 0.6176911544227887,
+      "grad_norm": 0.5503778329351654,
+      "learning_rate": 7.825396347237413e-06,
+      "loss": 0.0507,
+      "step": 412
+    },
+    {
+      "epoch": 0.6191904047976012,
+      "grad_norm": 0.6815564580393113,
+      "learning_rate": 7.815673645031871e-06,
+      "loss": 0.081,
+      "step": 413
+    },
+    {
+      "epoch": 0.6206896551724138,
+      "grad_norm": 0.619758344863062,
+      "learning_rate": 7.805935326811913e-06,
+      "loss": 0.0675,
+      "step": 414
+    },
+    {
+      "epoch": 0.6221889055472264,
+      "grad_norm": 0.6008221363787319,
+      "learning_rate": 7.796181446587244e-06,
+      "loss": 0.0601,
+      "step": 415
+    },
+    {
+      "epoch": 0.623688155922039,
+      "grad_norm": 0.8162472954922391,
+      "learning_rate": 7.786412058453886e-06,
+      "loss": 0.1002,
+      "step": 416
+    },
+    {
+      "epoch": 0.6251874062968515,
+      "grad_norm": 0.684229822204971,
+      "learning_rate": 7.776627216593863e-06,
+      "loss": 0.0729,
+      "step": 417
+    },
+    {
+      "epoch": 0.6266866566716641,
+      "grad_norm": 0.5670346215910291,
+      "learning_rate": 7.766826975274916e-06,
+      "loss": 0.0717,
+      "step": 418
+    },
+    {
+      "epoch": 0.6281859070464768,
+      "grad_norm": 1.2900577630397891,
+      "learning_rate": 7.75701138885018e-06,
+      "loss": 0.1409,
+      "step": 419
+    },
+    {
+      "epoch": 0.6296851574212894,
+      "grad_norm": 0.8187675769895846,
+      "learning_rate": 7.747180511757908e-06,
+      "loss": 0.0854,
+      "step": 420
+    },
+    {
+      "epoch": 0.631184407796102,
+      "grad_norm": 0.6264234704171173,
+      "learning_rate": 7.737334398521149e-06,
+      "loss": 0.0712,
+      "step": 421
+    },
+    {
+      "epoch": 0.6326836581709145,
+      "grad_norm": 0.8230857558856897,
+      "learning_rate": 7.727473103747456e-06,
+      "loss": 0.1285,
+      "step": 422
+    },
+    {
+      "epoch": 0.6341829085457271,
+      "grad_norm": 0.683588046332461,
+      "learning_rate": 7.717596682128578e-06,
+      "loss": 0.0734,
+      "step": 423
+    },
+    {
+      "epoch": 0.6356821589205397,
+      "grad_norm": 0.8231030993673684,
+      "learning_rate": 7.707705188440165e-06,
+      "loss": 0.0754,
+      "step": 424
+    },
+    {
+      "epoch": 0.6371814092953523,
+      "grad_norm": 0.8884280306253548,
+      "learning_rate": 7.697798677541448e-06,
+      "loss": 0.1186,
+      "step": 425
+    },
+    {
+      "epoch": 0.638680659670165,
+      "grad_norm": 0.7565853871152396,
+      "learning_rate": 7.687877204374957e-06,
+      "loss": 0.0917,
+      "step": 426
+    },
+    {
+      "epoch": 0.6401799100449775,
+      "grad_norm": 0.6899866384867913,
+      "learning_rate": 7.677940823966196e-06,
+      "loss": 0.077,
+      "step": 427
+    },
+    {
+      "epoch": 0.6416791604197901,
+      "grad_norm": 0.6881121015705602,
+      "learning_rate": 7.667989591423349e-06,
+      "loss": 0.082,
+      "step": 428
+    },
+    {
+      "epoch": 0.6431784107946027,
+      "grad_norm": 0.5783761630377401,
+      "learning_rate": 7.658023561936966e-06,
+      "loss": 0.0689,
+      "step": 429
+    },
+    {
+      "epoch": 0.6446776611694153,
+      "grad_norm": 0.6357287807928849,
+      "learning_rate": 7.648042790779677e-06,
+      "loss": 0.0759,
+      "step": 430
+    },
+    {
+      "epoch": 0.6461769115442278,
+      "grad_norm": 0.7718638768370127,
+      "learning_rate": 7.638047333305853e-06,
+      "loss": 0.0867,
+      "step": 431
+    },
+    {
+      "epoch": 0.6476761619190404,
+      "grad_norm": 0.7345828299788837,
+      "learning_rate": 7.628037244951328e-06,
+      "loss": 0.0972,
+      "step": 432
+    },
+    {
+      "epoch": 0.6491754122938531,
+      "grad_norm": 0.7462001820047025,
+      "learning_rate": 7.618012581233076e-06,
+      "loss": 0.0699,
+      "step": 433
+    },
+    {
+      "epoch": 0.6506746626686657,
+      "grad_norm": 0.5806221518021646,
+      "learning_rate": 7.607973397748909e-06,
+      "loss": 0.0656,
+      "step": 434
+    },
+    {
+      "epoch": 0.6521739130434783,
+      "grad_norm": 0.6996982147926565,
+      "learning_rate": 7.597919750177168e-06,
+      "loss": 0.0899,
+      "step": 435
+    },
+    {
+      "epoch": 0.6536731634182908,
+      "grad_norm": 0.7291854132295439,
+      "learning_rate": 7.587851694276412e-06,
+      "loss": 0.0771,
+      "step": 436
+    },
+    {
+      "epoch": 0.6551724137931034,
+      "grad_norm": 0.6957131208222146,
+      "learning_rate": 7.57776928588511e-06,
+      "loss": 0.0714,
+      "step": 437
+    },
+    {
+      "epoch": 0.656671664167916,
+      "grad_norm": 0.5205672692864283,
+      "learning_rate": 7.56767258092133e-06,
+      "loss": 0.0572,
+      "step": 438
+    },
+    {
+      "epoch": 0.6581709145427287,
+      "grad_norm": 0.7587806663590021,
+      "learning_rate": 7.557561635382433e-06,
+      "loss": 0.0828,
+      "step": 439
+    },
+    {
+      "epoch": 0.6596701649175413,
+      "grad_norm": 0.5806890280296683,
+      "learning_rate": 7.54743650534476e-06,
+      "loss": 0.0686,
+      "step": 440
+    },
+    {
+      "epoch": 0.6611694152923538,
+      "grad_norm": 0.817529922174908,
+      "learning_rate": 7.537297246963316e-06,
+      "loss": 0.105,
+      "step": 441
+    },
+    {
+      "epoch": 0.6626686656671664,
+      "grad_norm": 0.7167533902063026,
+      "learning_rate": 7.5271439164714695e-06,
+      "loss": 0.0844,
+      "step": 442
+    },
+    {
+      "epoch": 0.664167916041979,
+      "grad_norm": 0.5809897789705553,
+      "learning_rate": 7.5169765701806295e-06,
+      "loss": 0.0668,
+      "step": 443
+    },
+    {
+      "epoch": 0.6656671664167916,
+      "grad_norm": 0.4993158816048489,
+      "learning_rate": 7.506795264479941e-06,
+      "loss": 0.059,
+      "step": 444
+    },
+    {
+      "epoch": 0.6671664167916042,
+      "grad_norm": 0.696127380566264,
+      "learning_rate": 7.4966000558359675e-06,
+      "loss": 0.0809,
+      "step": 445
+    },
+    {
+      "epoch": 0.6686656671664168,
+      "grad_norm": 0.8048314127852231,
+      "learning_rate": 7.486391000792379e-06,
+      "loss": 0.0864,
+      "step": 446
+    },
+    {
+      "epoch": 0.6701649175412294,
+      "grad_norm": 0.7563631305422318,
+      "learning_rate": 7.476168155969643e-06,
+      "loss": 0.1041,
+      "step": 447
+    },
+    {
+      "epoch": 0.671664167916042,
+      "grad_norm": 0.7064816275898936,
+      "learning_rate": 7.465931578064703e-06,
+      "loss": 0.0858,
+      "step": 448
+    },
+    {
+      "epoch": 0.6731634182908546,
+      "grad_norm": 0.6830186435960813,
+      "learning_rate": 7.455681323850669e-06,
+      "loss": 0.0698,
+      "step": 449
+    },
+    {
+      "epoch": 0.6746626686656672,
+      "grad_norm": 0.4517358850843116,
+      "learning_rate": 7.4454174501765e-06,
+      "loss": 0.0506,
+      "step": 450
+    },
+    {
+      "epoch": 0.6761619190404797,
+      "grad_norm": 0.649745983834923,
+      "learning_rate": 7.4351400139666894e-06,
+      "loss": 0.0761,
+      "step": 451
+    },
+    {
+      "epoch": 0.6776611694152923,
+      "grad_norm": 0.6714325361519483,
+      "learning_rate": 7.424849072220953e-06,
+      "loss": 0.0916,
+      "step": 452
+    },
+    {
+      "epoch": 0.679160419790105,
+      "grad_norm": 0.6856370806568042,
+      "learning_rate": 7.414544682013907e-06,
+      "loss": 0.0657,
+      "step": 453
+    },
+    {
+      "epoch": 0.6806596701649176,
+      "grad_norm": 0.6689115960346489,
+      "learning_rate": 7.404226900494753e-06,
+      "loss": 0.0663,
+      "step": 454
+    },
+    {
+      "epoch": 0.6821589205397302,
+      "grad_norm": 0.5846743430250805,
+      "learning_rate": 7.3938957848869684e-06,
+      "loss": 0.0716,
+      "step": 455
+    },
+    {
+      "epoch": 0.6836581709145427,
+      "grad_norm": 0.5423898799806454,
+      "learning_rate": 7.3835513924879755e-06,
+      "loss": 0.0532,
+      "step": 456
+    },
+    {
+      "epoch": 0.6851574212893553,
+      "grad_norm": 0.542434585578821,
+      "learning_rate": 7.373193780668835e-06,
+      "loss": 0.0569,
+      "step": 457
+    },
+    {
+      "epoch": 0.6866566716641679,
+      "grad_norm": 0.8488093586278542,
+      "learning_rate": 7.36282300687392e-06,
+      "loss": 0.0993,
+      "step": 458
+    },
+    {
+      "epoch": 0.6881559220389805,
+      "grad_norm": 0.734794594228449,
+      "learning_rate": 7.35243912862061e-06,
+      "loss": 0.0787,
+      "step": 459
+    },
+    {
+      "epoch": 0.6896551724137931,
+      "grad_norm": 0.8492289745923112,
+      "learning_rate": 7.342042203498952e-06,
+      "loss": 0.0835,
+      "step": 460
+    },
+    {
+      "epoch": 0.6911544227886057,
+      "grad_norm": 0.8615943107447342,
+      "learning_rate": 7.33163228917136e-06,
+      "loss": 0.082,
+      "step": 461
+    },
+    {
+      "epoch": 0.6926536731634183,
+      "grad_norm": 0.6397624792462956,
+      "learning_rate": 7.321209443372284e-06,
+      "loss": 0.0824,
+      "step": 462
+    },
+    {
+      "epoch": 0.6941529235382309,
+      "grad_norm": 0.6764139827195026,
+      "learning_rate": 7.310773723907895e-06,
+      "loss": 0.0891,
+      "step": 463
+    },
+    {
+      "epoch": 0.6956521739130435,
+      "grad_norm": 0.8333441378755534,
+      "learning_rate": 7.300325188655762e-06,
+      "loss": 0.0916,
+      "step": 464
+    },
+    {
+      "epoch": 0.697151424287856,
+      "grad_norm": 0.7600607703245819,
+      "learning_rate": 7.289863895564531e-06,
+      "loss": 0.0995,
+      "step": 465
+    },
+    {
+      "epoch": 0.6986506746626686,
+      "grad_norm": 0.6365526669843394,
+      "learning_rate": 7.279389902653606e-06,
+      "loss": 0.0771,
+      "step": 466
+    },
+    {
+      "epoch": 0.7001499250374813,
+      "grad_norm": 0.5669329498520707,
+      "learning_rate": 7.268903268012823e-06,
+      "loss": 0.0668,
+      "step": 467
+    },
+    {
+      "epoch": 0.7016491754122939,
+      "grad_norm": 0.5894429748355027,
+      "learning_rate": 7.258404049802135e-06,
+      "loss": 0.0702,
+      "step": 468
+    },
+    {
+      "epoch": 0.7031484257871065,
+      "grad_norm": 0.5860134337068711,
+      "learning_rate": 7.247892306251276e-06,
+      "loss": 0.0581,
+      "step": 469
+    },
+    {
+      "epoch": 0.704647676161919,
+      "grad_norm": 0.6088493208245382,
+      "learning_rate": 7.237368095659459e-06,
+      "loss": 0.0799,
+      "step": 470
+    },
+    {
+      "epoch": 0.7061469265367316,
+      "grad_norm": 0.708480354655168,
+      "learning_rate": 7.226831476395028e-06,
+      "loss": 0.0805,
+      "step": 471
+    },
+    {
+      "epoch": 0.7076461769115442,
+      "grad_norm": 0.7133007331918387,
+      "learning_rate": 7.216282506895155e-06,
+      "loss": 0.0685,
+      "step": 472
+    },
+    {
+      "epoch": 0.7091454272863568,
+      "grad_norm": 0.6622372909157326,
+      "learning_rate": 7.2057212456655055e-06,
+      "loss": 0.0738,
+      "step": 473
+    },
+    {
+      "epoch": 0.7106446776611695,
+      "grad_norm": 0.6844987129159301,
+      "learning_rate": 7.195147751279915e-06,
+      "loss": 0.0915,
+      "step": 474
+    },
+    {
+      "epoch": 0.712143928035982,
+      "grad_norm": 0.8933495441728474,
+      "learning_rate": 7.184562082380069e-06,
+      "loss": 0.1271,
+      "step": 475
+    },
+    {
+      "epoch": 0.7136431784107946,
+      "grad_norm": 0.8043235117595697,
+      "learning_rate": 7.173964297675168e-06,
+      "loss": 0.0877,
+      "step": 476
+    },
+    {
+      "epoch": 0.7151424287856072,
+      "grad_norm": 0.7772354331988821,
+      "learning_rate": 7.163354455941614e-06,
+      "loss": 0.0858,
+      "step": 477
+    },
+    {
+      "epoch": 0.7166416791604198,
+      "grad_norm": 0.6872696085047594,
+      "learning_rate": 7.152732616022675e-06,
+      "loss": 0.0814,
+      "step": 478
+    },
+    {
+      "epoch": 0.7181409295352323,
+      "grad_norm": 0.8001607420967578,
+      "learning_rate": 7.142098836828162e-06,
+      "loss": 0.1092,
+      "step": 479
+    },
+    {
+      "epoch": 0.719640179910045,
+      "grad_norm": 0.8859960839166726,
+      "learning_rate": 7.131453177334103e-06,
+      "loss": 0.0999,
+      "step": 480
+    },
+    {
+      "epoch": 0.7211394302848576,
+      "grad_norm": 0.5747414093252482,
+      "learning_rate": 7.120795696582419e-06,
+      "loss": 0.0651,
+      "step": 481
+    },
+    {
+      "epoch": 0.7226386806596702,
+      "grad_norm": 0.779227387734831,
+      "learning_rate": 7.1101264536805885e-06,
+      "loss": 0.0892,
+      "step": 482
+    },
+    {
+      "epoch": 0.7241379310344828,
+      "grad_norm": 0.5742129756218621,
+      "learning_rate": 7.099445507801324e-06,
+      "loss": 0.0618,
+      "step": 483
+    },
+    {
+      "epoch": 0.7256371814092953,
+      "grad_norm": 0.6580027931228973,
+      "learning_rate": 7.088752918182247e-06,
+      "loss": 0.0714,
+      "step": 484
+    },
+    {
+      "epoch": 0.7271364317841079,
+      "grad_norm": 0.8053457971859354,
+      "learning_rate": 7.078048744125553e-06,
+      "loss": 0.1103,
+      "step": 485
+    },
+    {
+      "epoch": 0.7286356821589205,
+      "grad_norm": 0.6514553745184477,
+      "learning_rate": 7.067333044997689e-06,
+      "loss": 0.0933,
+      "step": 486
+    },
+    {
+      "epoch": 0.7301349325337332,
+      "grad_norm": 0.8905126052992827,
+      "learning_rate": 7.0566058802290196e-06,
+      "loss": 0.1156,
+      "step": 487
+    },
+    {
+      "epoch": 0.7316341829085458,
+      "grad_norm": 0.6240727338068806,
+      "learning_rate": 7.045867309313499e-06,
+      "loss": 0.0789,
+      "step": 488
+    },
+    {
+      "epoch": 0.7331334332833583,
+      "grad_norm": 0.6154540467861818,
+      "learning_rate": 7.035117391808341e-06,
+      "loss": 0.0734,
+      "step": 489
+    },
+    {
+      "epoch": 0.7346326836581709,
+      "grad_norm": 0.7020206642079881,
+      "learning_rate": 7.024356187333692e-06,
+      "loss": 0.0969,
+      "step": 490
+    },
+    {
+      "epoch": 0.7361319340329835,
+      "grad_norm": 0.8443215430193017,
+      "learning_rate": 7.01358375557229e-06,
+      "loss": 0.1225,
+      "step": 491
+    },
+    {
+      "epoch": 0.7376311844077961,
+      "grad_norm": 0.5710515558311333,
+      "learning_rate": 7.0028001562691475e-06,
+      "loss": 0.0728,
+      "step": 492
+    },
+    {
+      "epoch": 0.7391304347826086,
+      "grad_norm": 0.6079621515159543,
+      "learning_rate": 6.9920054492312086e-06,
+      "loss": 0.0829,
+      "step": 493
+    },
+    {
+      "epoch": 0.7406296851574213,
+      "grad_norm": 0.8968542739489341,
+      "learning_rate": 6.981199694327024e-06,
+      "loss": 0.0899,
+      "step": 494
+    },
+    {
+      "epoch": 0.7421289355322339,
+      "grad_norm": 0.6432390251721881,
+      "learning_rate": 6.97038295148642e-06,
+      "loss": 0.077,
+      "step": 495
+    },
+    {
+      "epoch": 0.7436281859070465,
+      "grad_norm": 0.5337787016628947,
+      "learning_rate": 6.959555280700162e-06,
+      "loss": 0.0612,
+      "step": 496
+    },
+    {
+      "epoch": 0.7451274362818591,
+      "grad_norm": 0.802822726831146,
+      "learning_rate": 6.948716742019616e-06,
+      "loss": 0.0828,
+      "step": 497
+    },
+    {
+      "epoch": 0.7466266866566716,
+      "grad_norm": 0.8779320312466281,
+      "learning_rate": 6.937867395556428e-06,
+      "loss": 0.0955,
+      "step": 498
+    },
+    {
+      "epoch": 0.7481259370314842,
+      "grad_norm": 0.506616031664386,
+      "learning_rate": 6.927007301482187e-06,
+      "loss": 0.0508,
+      "step": 499
+    },
+    {
+      "epoch": 0.7496251874062968,
+      "grad_norm": 0.89053063662153,
+      "learning_rate": 6.916136520028087e-06,
+      "loss": 0.1178,
+      "step": 500
+    },
+    {
+      "epoch": 0.7511244377811095,
+      "grad_norm": 0.6489416035031975,
+      "learning_rate": 6.905255111484592e-06,
+      "loss": 0.074,
+      "step": 501
+    },
+    {
+      "epoch": 0.7526236881559221,
+      "grad_norm": 0.8057480229792926,
+      "learning_rate": 6.894363136201114e-06,
+      "loss": 0.1009,
+      "step": 502
+    },
+    {
+      "epoch": 0.7541229385307346,
+      "grad_norm": 0.7488315314670658,
+      "learning_rate": 6.88346065458566e-06,
+      "loss": 0.1025,
+      "step": 503
+    },
+    {
+      "epoch": 0.7556221889055472,
+      "grad_norm": 0.7274507633494838,
+      "learning_rate": 6.8725477271045085e-06,
+      "loss": 0.0827,
+      "step": 504
+    },
+    {
+      "epoch": 0.7571214392803598,
+      "grad_norm": 0.691163897193729,
+      "learning_rate": 6.861624414281875e-06,
+      "loss": 0.0727,
+      "step": 505
+    },
+    {
+      "epoch": 0.7586206896551724,
+      "grad_norm": 0.6754199576694153,
+      "learning_rate": 6.850690776699574e-06,
+      "loss": 0.0758,
+      "step": 506
+    },
+    {
+      "epoch": 0.760119940029985,
+      "grad_norm": 0.6504569805376379,
+      "learning_rate": 6.8397468749966735e-06,
+      "loss": 0.0751,
+      "step": 507
+    },
+    {
+      "epoch": 0.7616191904047976,
+      "grad_norm": 0.6779867136197483,
+      "learning_rate": 6.8287927698691745e-06,
+      "loss": 0.0716,
+      "step": 508
+    },
+    {
+      "epoch": 0.7631184407796102,
+      "grad_norm": 0.6477016359944904,
+      "learning_rate": 6.8178285220696686e-06,
+      "loss": 0.0848,
+      "step": 509
+    },
+    {
+      "epoch": 0.7646176911544228,
+      "grad_norm": 0.5749000825279154,
+      "learning_rate": 6.806854192406995e-06,
+      "loss": 0.0585,
+      "step": 510
+    },
+    {
+      "epoch": 0.7661169415292354,
+      "grad_norm": 0.7083668466880977,
+      "learning_rate": 6.795869841745912e-06,
+      "loss": 0.0883,
+      "step": 511
+    },
+    {
+      "epoch": 0.767616191904048,
+      "grad_norm": 0.5884106233263021,
+      "learning_rate": 6.784875531006751e-06,
+      "loss": 0.0633,
+      "step": 512
+    },
+    {
+      "epoch": 0.7691154422788605,
+      "grad_norm": 0.9778090071062603,
+      "learning_rate": 6.7738713211650885e-06,
+      "loss": 0.1183,
+      "step": 513
+    },
+    {
+      "epoch": 0.7706146926536732,
+      "grad_norm": 0.611285417087513,
+      "learning_rate": 6.762857273251396e-06,
+      "loss": 0.0697,
+      "step": 514
+    },
+    {
+      "epoch": 0.7721139430284858,
+      "grad_norm": 0.7891211118958112,
+      "learning_rate": 6.751833448350713e-06,
+      "loss": 0.0814,
+      "step": 515
+    },
+    {
+      "epoch": 0.7736131934032984,
+      "grad_norm": 0.768467656930431,
+      "learning_rate": 6.740799907602302e-06,
+      "loss": 0.0893,
+      "step": 516
+    },
+    {
+      "epoch": 0.775112443778111,
+      "grad_norm": 0.6295692906699851,
+      "learning_rate": 6.729756712199309e-06,
+      "loss": 0.0644,
+      "step": 517
+    },
+    {
+      "epoch": 0.7766116941529235,
+      "grad_norm": 0.5704667463291235,
+      "learning_rate": 6.718703923388427e-06,
+      "loss": 0.0615,
+      "step": 518
+    },
+    {
+      "epoch": 0.7781109445277361,
+      "grad_norm": 0.6233161845509494,
+      "learning_rate": 6.707641602469554e-06,
+      "loss": 0.0703,
+      "step": 519
+    },
+    {
+      "epoch": 0.7796101949025487,
+      "grad_norm": 0.7758494248523509,
+      "learning_rate": 6.696569810795455e-06,
+      "loss": 0.09,
+      "step": 520
+    },
+    {
+      "epoch": 0.7811094452773614,
+      "grad_norm": 0.5710786664316516,
+      "learning_rate": 6.685488609771422e-06,
+      "loss": 0.0569,
+      "step": 521
+    },
+    {
+      "epoch": 0.782608695652174,
+      "grad_norm": 0.7482009050063121,
+      "learning_rate": 6.674398060854931e-06,
+      "loss": 0.0772,
+      "step": 522
+    },
+    {
+      "epoch": 0.7841079460269865,
+      "grad_norm": 0.7454265219864258,
+      "learning_rate": 6.6632982255553004e-06,
+      "loss": 0.0886,
+      "step": 523
+    },
+    {
+      "epoch": 0.7856071964017991,
+      "grad_norm": 1.0010203144852925,
+      "learning_rate": 6.652189165433356e-06,
+      "loss": 0.1022,
+      "step": 524
+    },
+    {
+      "epoch": 0.7871064467766117,
+      "grad_norm": 0.6024887434224161,
+      "learning_rate": 6.64107094210108e-06,
+      "loss": 0.0781,
+      "step": 525
+    },
+    {
+      "epoch": 0.7886056971514243,
+      "grad_norm": 0.7160590143614867,
+      "learning_rate": 6.62994361722128e-06,
+      "loss": 0.076,
+      "step": 526
+    },
+    {
+      "epoch": 0.7901049475262368,
+      "grad_norm": 0.6540050249944035,
+      "learning_rate": 6.618807252507238e-06,
+      "loss": 0.0681,
+      "step": 527
+    },
+    {
+      "epoch": 0.7916041979010495,
+      "grad_norm": 0.9969956488281648,
+      "learning_rate": 6.6076619097223735e-06,
+      "loss": 0.1255,
+      "step": 528
+    },
+    {
+      "epoch": 0.7931034482758621,
+      "grad_norm": 0.9483819300756055,
+      "learning_rate": 6.5965076506799e-06,
+      "loss": 0.1109,
+      "step": 529
+    },
+    {
+      "epoch": 0.7946026986506747,
+      "grad_norm": 0.7319107535113389,
+      "learning_rate": 6.5853445372424805e-06,
+      "loss": 0.0864,
+      "step": 530
+    },
+    {
+      "epoch": 0.7961019490254873,
+      "grad_norm": 0.5870685263719585,
+      "learning_rate": 6.574172631321885e-06,
+      "loss": 0.0638,
+      "step": 531
+    },
+    {
+      "epoch": 0.7976011994002998,
+      "grad_norm": 0.7448237646881719,
+      "learning_rate": 6.562991994878649e-06,
+      "loss": 0.0883,
+      "step": 532
+    },
+    {
+      "epoch": 0.7991004497751124,
+      "grad_norm": 0.6787278322929973,
+      "learning_rate": 6.551802689921726e-06,
+      "loss": 0.0854,
+      "step": 533
+    },
+    {
+      "epoch": 0.800599700149925,
+      "grad_norm": 0.7499814786763364,
+      "learning_rate": 6.5406047785081485e-06,
+      "loss": 0.0971,
+      "step": 534
+    },
+    {
+      "epoch": 0.8020989505247377,
+      "grad_norm": 0.597541243617075,
+      "learning_rate": 6.529398322742677e-06,
+      "loss": 0.0676,
+      "step": 535
+    },
+    {
+      "epoch": 0.8035982008995503,
+      "grad_norm": 0.5632184085750767,
+      "learning_rate": 6.518183384777468e-06,
+      "loss": 0.0636,
+      "step": 536
+    },
+    {
+      "epoch": 0.8050974512743628,
+      "grad_norm": 0.945825532665182,
+      "learning_rate": 6.506960026811712e-06,
+      "loss": 0.1115,
+      "step": 537
+    },
+    {
+      "epoch": 0.8065967016491754,
+      "grad_norm": 0.7513917704841447,
+      "learning_rate": 6.495728311091303e-06,
+      "loss": 0.1052,
+      "step": 538
+    },
+    {
+      "epoch": 0.808095952023988,
+      "grad_norm": 0.6390618653643154,
+      "learning_rate": 6.484488299908487e-06,
+      "loss": 0.0756,
+      "step": 539
+    },
+    {
+      "epoch": 0.8095952023988006,
+      "grad_norm": 0.6620934778014838,
+      "learning_rate": 6.473240055601517e-06,
+      "loss": 0.0751,
+      "step": 540
+    },
+    {
+      "epoch": 0.8110944527736131,
+      "grad_norm": 0.6867795697936318,
+      "learning_rate": 6.46198364055431e-06,
+      "loss": 0.0839,
+      "step": 541
+    },
+    {
+      "epoch": 0.8125937031484258,
+      "grad_norm": 0.9854050745309445,
+      "learning_rate": 6.450719117196094e-06,
+      "loss": 0.1019,
+      "step": 542
+    },
+    {
+      "epoch": 0.8140929535232384,
+      "grad_norm": 0.8682458065772393,
+      "learning_rate": 6.439446548001069e-06,
+      "loss": 0.0838,
+      "step": 543
+    },
+    {
+      "epoch": 0.815592203898051,
+      "grad_norm": 0.8752993640352996,
+      "learning_rate": 6.4281659954880605e-06,
+      "loss": 0.0948,
+      "step": 544
+    },
+    {
+      "epoch": 0.8170914542728636,
+      "grad_norm": 0.6736131964889589,
+      "learning_rate": 6.416877522220167e-06,
+      "loss": 0.0715,
+      "step": 545
+    },
+    {
+      "epoch": 0.8185907046476761,
+      "grad_norm": 0.7948474649268304,
+      "learning_rate": 6.405581190804418e-06,
+      "loss": 0.0758,
+      "step": 546
+    },
+    {
+      "epoch": 0.8200899550224887,
+      "grad_norm": 0.5657943857212395,
+      "learning_rate": 6.394277063891422e-06,
+      "loss": 0.0645,
+      "step": 547
+    },
+    {
+      "epoch": 0.8215892053973014,
+      "grad_norm": 0.7713725074040843,
+      "learning_rate": 6.382965204175027e-06,
+      "loss": 0.1122,
+      "step": 548
+    },
+    {
+      "epoch": 0.823088455772114,
+      "grad_norm": 0.5687770456935312,
+      "learning_rate": 6.371645674391967e-06,
+      "loss": 0.0648,
+      "step": 549
+    },
+    {
+      "epoch": 0.8245877061469266,
+      "grad_norm": 0.8323906665610467,
+      "learning_rate": 6.3603185373215105e-06,
+      "loss": 0.0854,
+      "step": 550
+    },
+    {
+      "epoch": 0.8260869565217391,
+      "grad_norm": 0.7932167136921248,
+      "learning_rate": 6.348983855785122e-06,
+      "loss": 0.117,
+      "step": 551
+    },
+    {
+      "epoch": 0.8275862068965517,
+      "grad_norm": 0.8478509837694573,
+      "learning_rate": 6.337641692646106e-06,
+      "loss": 0.1025,
+      "step": 552
+    },
+    {
+      "epoch": 0.8290854572713643,
+      "grad_norm": 0.5363237152718322,
+      "learning_rate": 6.326292110809258e-06,
+      "loss": 0.0637,
+      "step": 553
+    },
+    {
+      "epoch": 0.8305847076461769,
+      "grad_norm": 0.5111961148421332,
+      "learning_rate": 6.314935173220524e-06,
+      "loss": 0.0653,
+      "step": 554
+    },
+    {
+      "epoch": 0.8320839580209896,
+      "grad_norm": 0.821927205108887,
+      "learning_rate": 6.303570942866643e-06,
+      "loss": 0.12,
+      "step": 555
+    },
+    {
+      "epoch": 0.8335832083958021,
+      "grad_norm": 0.6354394835908234,
+      "learning_rate": 6.2921994827748e-06,
+      "loss": 0.0707,
+      "step": 556
+    },
+    {
+      "epoch": 0.8350824587706147,
+      "grad_norm": 0.7629782665456358,
+      "learning_rate": 6.280820856012277e-06,
+      "loss": 0.0748,
+      "step": 557
+    },
+    {
+      "epoch": 0.8365817091454273,
+      "grad_norm": 0.5320707357341737,
+      "learning_rate": 6.269435125686105e-06,
+      "loss": 0.0578,
+      "step": 558
+    },
+    {
+      "epoch": 0.8380809595202399,
+      "grad_norm": 0.742759171308037,
+      "learning_rate": 6.258042354942708e-06,
+      "loss": 0.0895,
+      "step": 559
+    },
+    {
+      "epoch": 0.8395802098950524,
+      "grad_norm": 0.6784392526119097,
+      "learning_rate": 6.2466426069675626e-06,
+      "loss": 0.0767,
+      "step": 560
+    },
+    {
+      "epoch": 0.841079460269865,
+      "grad_norm": 0.6571467309932431,
+      "learning_rate": 6.235235944984835e-06,
+      "loss": 0.085,
+      "step": 561
+    },
+    {
+      "epoch": 0.8425787106446777,
+      "grad_norm": 0.649485221081861,
+      "learning_rate": 6.223822432257043e-06,
+      "loss": 0.0682,
+      "step": 562
+    },
+    {
+      "epoch": 0.8440779610194903,
+      "grad_norm": 0.7640512287466666,
+      "learning_rate": 6.212402132084697e-06,
+      "loss": 0.1042,
+      "step": 563
+    },
+    {
+      "epoch": 0.8455772113943029,
+      "grad_norm": 0.6962742258808327,
+      "learning_rate": 6.200975107805951e-06,
+      "loss": 0.0629,
+      "step": 564
+    },
+    {
+      "epoch": 0.8470764617691154,
+      "grad_norm": 0.6559738141028493,
+      "learning_rate": 6.189541422796254e-06,
+      "loss": 0.0768,
+      "step": 565
+    },
+    {
+      "epoch": 0.848575712143928,
+      "grad_norm": 0.6174256379639276,
+      "learning_rate": 6.1781011404679905e-06,
+      "loss": 0.0768,
+      "step": 566
+    },
+    {
+      "epoch": 0.8500749625187406,
+      "grad_norm": 0.6887950754826893,
+      "learning_rate": 6.16665432427014e-06,
+      "loss": 0.0708,
+      "step": 567
+    },
+    {
+      "epoch": 0.8515742128935532,
+      "grad_norm": 0.712410525138826,
+      "learning_rate": 6.155201037687917e-06,
+      "loss": 0.1126,
+      "step": 568
+    },
+    {
+      "epoch": 0.8530734632683659,
+      "grad_norm": 0.6268821116654263,
+      "learning_rate": 6.1437413442424236e-06,
+      "loss": 0.0638,
+      "step": 569
+    },
+    {
+      "epoch": 0.8545727136431784,
+      "grad_norm": 0.6144458046503305,
+      "learning_rate": 6.132275307490291e-06,
+      "loss": 0.0732,
+      "step": 570
+    },
+    {
+      "epoch": 0.856071964017991,
+      "grad_norm": 0.6845932379680061,
+      "learning_rate": 6.120802991023334e-06,
+      "loss": 0.06,
+      "step": 571
+    },
+    {
+      "epoch": 0.8575712143928036,
+      "grad_norm": 0.6866927872231734,
+      "learning_rate": 6.109324458468198e-06,
+      "loss": 0.0866,
+      "step": 572
+    },
+    {
+      "epoch": 0.8590704647676162,
+      "grad_norm": 0.6897846850153246,
+      "learning_rate": 6.097839773485995e-06,
+      "loss": 0.0749,
+      "step": 573
+    },
+    {
+      "epoch": 0.8605697151424287,
+      "grad_norm": 0.7598489640675004,
+      "learning_rate": 6.086348999771967e-06,
+      "loss": 0.0896,
+      "step": 574
+    },
+    {
+      "epoch": 0.8620689655172413,
+      "grad_norm": 0.6476107261358266,
+      "learning_rate": 6.074852201055121e-06,
+      "loss": 0.0728,
+      "step": 575
+    },
+    {
+      "epoch": 0.863568215892054,
+      "grad_norm": 0.6804670552084989,
+      "learning_rate": 6.063349441097881e-06,
+      "loss": 0.0727,
+      "step": 576
+    },
+    {
+      "epoch": 0.8650674662668666,
+      "grad_norm": 0.6535133077062418,
+      "learning_rate": 6.051840783695731e-06,
+      "loss": 0.0733,
+      "step": 577
+    },
+    {
+      "epoch": 0.8665667166416792,
+      "grad_norm": 0.6793326950983349,
+      "learning_rate": 6.040326292676865e-06,
+      "loss": 0.0789,
+      "step": 578
+    },
+    {
+      "epoch": 0.8680659670164917,
+      "grad_norm": 0.9639330786361555,
+      "learning_rate": 6.028806031901829e-06,
+      "loss": 0.1026,
+      "step": 579
+    },
+    {
+      "epoch": 0.8695652173913043,
+      "grad_norm": 0.6768838781634493,
+      "learning_rate": 6.0172800652631706e-06,
+      "loss": 0.0897,
+      "step": 580
+    },
+    {
+      "epoch": 0.8710644677661169,
+      "grad_norm": 0.7968682114195774,
+      "learning_rate": 6.005748456685077e-06,
+      "loss": 0.073,
+      "step": 581
+    },
+    {
+      "epoch": 0.8725637181409296,
+      "grad_norm": 0.7558580666824537,
+      "learning_rate": 5.994211270123034e-06,
+      "loss": 0.0987,
+      "step": 582
+    },
+    {
+      "epoch": 0.8740629685157422,
+      "grad_norm": 0.7663939761279863,
+      "learning_rate": 5.9826685695634575e-06,
+      "loss": 0.098,
+      "step": 583
+    },
+    {
+      "epoch": 0.8755622188905547,
+      "grad_norm": 0.8078531531870057,
+      "learning_rate": 5.971120419023349e-06,
+      "loss": 0.0681,
+      "step": 584
+    },
+    {
+      "epoch": 0.8770614692653673,
+      "grad_norm": 0.5504922120597476,
+      "learning_rate": 5.959566882549936e-06,
+      "loss": 0.0888,
+      "step": 585
+    },
+    {
+      "epoch": 0.8785607196401799,
+      "grad_norm": 0.6432980549107314,
+      "learning_rate": 5.948008024220311e-06,
+      "loss": 0.0722,
+      "step": 586
+    },
+    {
+      "epoch": 0.8800599700149925,
+      "grad_norm": 0.5904147674697975,
+      "learning_rate": 5.936443908141088e-06,
+      "loss": 0.0656,
+      "step": 587
+    },
+    {
+      "epoch": 0.881559220389805,
+      "grad_norm": 0.7216477261436639,
+      "learning_rate": 5.924874598448038e-06,
+      "loss": 0.0857,
+      "step": 588
+    },
+    {
+      "epoch": 0.8830584707646177,
+      "grad_norm": 0.5888738614799112,
+      "learning_rate": 5.913300159305741e-06,
+      "loss": 0.0765,
+      "step": 589
+    },
+    {
+      "epoch": 0.8845577211394303,
+      "grad_norm": 0.7011668500615866,
+      "learning_rate": 5.901720654907217e-06,
+      "loss": 0.0718,
+      "step": 590
+    },
+    {
+      "epoch": 0.8860569715142429,
+      "grad_norm": 0.6937449785289002,
+      "learning_rate": 5.8901361494735874e-06,
+      "loss": 0.0791,
+      "step": 591
+    },
+    {
+      "epoch": 0.8875562218890555,
+      "grad_norm": 0.7059187065394823,
+      "learning_rate": 5.878546707253704e-06,
+      "loss": 0.0824,
+      "step": 592
+    },
+    {
+      "epoch": 0.889055472263868,
+      "grad_norm": 0.7588583487828472,
+      "learning_rate": 5.8669523925238e-06,
+      "loss": 0.0921,
+      "step": 593
+    },
+    {
+      "epoch": 0.8905547226386806,
+      "grad_norm": 0.5662023150399038,
+      "learning_rate": 5.855353269587134e-06,
+      "loss": 0.067,
+      "step": 594
+    },
+    {
+      "epoch": 0.8920539730134932,
+      "grad_norm": 0.5600644840339529,
+      "learning_rate": 5.843749402773629e-06,
+      "loss": 0.0576,
+      "step": 595
+    },
+    {
+      "epoch": 0.8935532233883059,
+      "grad_norm": 0.6033216090934186,
+      "learning_rate": 5.8321408564395165e-06,
+      "loss": 0.0607,
+      "step": 596
+    },
+    {
+      "epoch": 0.8950524737631185,
+      "grad_norm": 0.6894574816602232,
+      "learning_rate": 5.820527694966988e-06,
+      "loss": 0.0699,
+      "step": 597
+    },
+    {
+      "epoch": 0.896551724137931,
+      "grad_norm": 0.544027815240822,
+      "learning_rate": 5.808909982763825e-06,
+      "loss": 0.0492,
+      "step": 598
+    },
+    {
+      "epoch": 0.8980509745127436,
+      "grad_norm": 0.5845567660541914,
+      "learning_rate": 5.797287784263047e-06,
+      "loss": 0.0848,
+      "step": 599
+    },
+    {
+      "epoch": 0.8995502248875562,
+      "grad_norm": 0.7799064021683438,
+      "learning_rate": 5.785661163922558e-06,
+      "loss": 0.076,
+      "step": 600
+    },
+    {
+      "epoch": 0.8995502248875562,
+      "eval_loss": 0.0919097512960434,
+      "eval_runtime": 2.1856,
+      "eval_samples_per_second": 24.707,
+      "eval_steps_per_second": 6.406,
+      "step": 600
+    },
+    {
+      "epoch": 0.9010494752623688,
+      "grad_norm": 0.7879473947014103,
+      "learning_rate": 5.774030186224786e-06,
+      "loss": 0.077,
+      "step": 601
+    },
+    {
+      "epoch": 0.9025487256371814,
+      "grad_norm": 0.5726905573414383,
+      "learning_rate": 5.762394915676325e-06,
+      "loss": 0.0589,
+      "step": 602
+    },
+    {
+      "epoch": 0.904047976011994,
+      "grad_norm": 0.7983720297574975,
+      "learning_rate": 5.750755416807575e-06,
+      "loss": 0.1051,
+      "step": 603
+    },
+    {
+      "epoch": 0.9055472263868066,
+      "grad_norm": 0.655710139152802,
+      "learning_rate": 5.7391117541723914e-06,
+      "loss": 0.071,
+      "step": 604
+    },
+    {
+      "epoch": 0.9070464767616192,
+      "grad_norm": 0.7204629020480462,
+      "learning_rate": 5.727463992347719e-06,
+      "loss": 0.0815,
+      "step": 605
+    },
+    {
+      "epoch": 0.9085457271364318,
+      "grad_norm": 0.7008106499373852,
+      "learning_rate": 5.715812195933238e-06,
+      "loss": 0.0686,
+      "step": 606
+    },
+    {
+      "epoch": 0.9100449775112444,
+      "grad_norm": 0.703285499491494,
+      "learning_rate": 5.704156429551004e-06,
+      "loss": 0.0861,
+      "step": 607
+    },
+    {
+      "epoch": 0.9115442278860569,
+      "grad_norm": 0.7395585330013787,
+      "learning_rate": 5.692496757845092e-06,
+      "loss": 0.0824,
+      "step": 608
+    },
+    {
+      "epoch": 0.9130434782608695,
+      "grad_norm": 0.744820512049887,
+      "learning_rate": 5.680833245481234e-06,
+      "loss": 0.0795,
+      "step": 609
+    },
+    {
+      "epoch": 0.9145427286356822,
+      "grad_norm": 0.8642131575214492,
+      "learning_rate": 5.6691659571464655e-06,
+      "loss": 0.1286,
+      "step": 610
+    },
+    {
+      "epoch": 0.9160419790104948,
+      "grad_norm": 0.5654143134098867,
+      "learning_rate": 5.657494957548761e-06,
+      "loss": 0.0719,
+      "step": 611
+    },
+    {
+      "epoch": 0.9175412293853074,
+      "grad_norm": 0.7131574195846725,
+      "learning_rate": 5.645820311416681e-06,
+      "loss": 0.0677,
+      "step": 612
+    },
+    {
+      "epoch": 0.9190404797601199,
+      "grad_norm": 0.7784274891253757,
+      "learning_rate": 5.63414208349901e-06,
+      "loss": 0.0938,
+      "step": 613
+    },
+    {
+      "epoch": 0.9205397301349325,
+      "grad_norm": 0.6540969924306904,
+      "learning_rate": 5.622460338564393e-06,
+      "loss": 0.08,
+      "step": 614
+    },
+    {
+      "epoch": 0.9220389805097451,
+      "grad_norm": 0.5922112443554233,
+      "learning_rate": 5.610775141400986e-06,
+      "loss": 0.0607,
+      "step": 615
+    },
+    {
+      "epoch": 0.9235382308845578,
+      "grad_norm": 0.7546429070422588,
+      "learning_rate": 5.599086556816089e-06,
+      "loss": 0.0843,
+      "step": 616
+    },
+    {
+      "epoch": 0.9250374812593704,
+      "grad_norm": 0.7720396964913429,
+      "learning_rate": 5.587394649635789e-06,
+      "loss": 0.1006,
+      "step": 617
+    },
+    {
+      "epoch": 0.9265367316341829,
+      "grad_norm": 0.7660967459970913,
+      "learning_rate": 5.575699484704599e-06,
+      "loss": 0.0718,
+      "step": 618
+    },
+    {
+      "epoch": 0.9280359820089955,
+      "grad_norm": 0.7640361293965884,
+      "learning_rate": 5.564001126885106e-06,
+      "loss": 0.0819,
+      "step": 619
+    },
+    {
+      "epoch": 0.9295352323838081,
+      "grad_norm": 0.8130048470770087,
+      "learning_rate": 5.552299641057596e-06,
+      "loss": 0.0986,
+      "step": 620
+    },
+    {
+      "epoch": 0.9310344827586207,
+      "grad_norm": 0.7637373241590193,
+      "learning_rate": 5.540595092119709e-06,
+      "loss": 0.0909,
+      "step": 621
+    },
+    {
+      "epoch": 0.9325337331334332,
+      "grad_norm": 0.7480143407150595,
+      "learning_rate": 5.5288875449860745e-06,
+      "loss": 0.0789,
+      "step": 622
+    },
+    {
+      "epoch": 0.9340329835082459,
+      "grad_norm": 0.6027852741396966,
+      "learning_rate": 5.517177064587945e-06,
+      "loss": 0.0744,
+      "step": 623
+    },
+    {
+      "epoch": 0.9355322338830585,
+      "grad_norm": 0.5354586130700105,
+      "learning_rate": 5.505463715872846e-06,
+      "loss": 0.0647,
+      "step": 624
+    },
+    {
+      "epoch": 0.9370314842578711,
+      "grad_norm": 0.6732647593810134,
+      "learning_rate": 5.493747563804211e-06,
+      "loss": 0.0731,
+      "step": 625
+    },
+    {
+      "epoch": 0.9385307346326837,
+      "grad_norm": 0.4727691175844128,
+      "learning_rate": 5.482028673361015e-06,
+      "loss": 0.0589,
+      "step": 626
+    },
+    {
+      "epoch": 0.9400299850074962,
+      "grad_norm": 0.6940440036365666,
+      "learning_rate": 5.470307109537427e-06,
+      "loss": 0.0808,
+      "step": 627
+    },
+    {
+      "epoch": 0.9415292353823088,
+      "grad_norm": 0.7638226634174295,
+      "learning_rate": 5.45858293734244e-06,
+      "loss": 0.0831,
+      "step": 628
+    },
+    {
+      "epoch": 0.9430284857571214,
+      "grad_norm": 0.9215561230136569,
+      "learning_rate": 5.446856221799515e-06,
+      "loss": 0.0977,
+      "step": 629
+    },
+    {
+      "epoch": 0.9445277361319341,
+      "grad_norm": 0.6792831296529718,
+      "learning_rate": 5.435127027946215e-06,
+      "loss": 0.0885,
+      "step": 630
+    },
+    {
+      "epoch": 0.9460269865067467,
+      "grad_norm": 0.8841076948757384,
+      "learning_rate": 5.423395420833853e-06,
+      "loss": 0.095,
+      "step": 631
+    },
+    {
+      "epoch": 0.9475262368815592,
+      "grad_norm": 0.7470239600415952,
+      "learning_rate": 5.411661465527123e-06,
+      "loss": 0.0785,
+      "step": 632
+    },
+    {
+      "epoch": 0.9490254872563718,
+      "grad_norm": 0.6686001720174529,
+      "learning_rate": 5.39992522710374e-06,
+      "loss": 0.0788,
+      "step": 633
+    },
+    {
+      "epoch": 0.9505247376311844,
+      "grad_norm": 0.6406631423357055,
+      "learning_rate": 5.38818677065409e-06,
+      "loss": 0.0568,
+      "step": 634
+    },
+    {
+      "epoch": 0.952023988005997,
+      "grad_norm": 0.5902773953009008,
+      "learning_rate": 5.376446161280851e-06,
+      "loss": 0.0666,
+      "step": 635
+    },
+    {
+      "epoch": 0.9535232383808095,
+      "grad_norm": 0.7499221441325681,
+      "learning_rate": 5.364703464098645e-06,
+      "loss": 0.1043,
+      "step": 636
+    },
+    {
+      "epoch": 0.9550224887556222,
+      "grad_norm": 0.6581619181795593,
+      "learning_rate": 5.352958744233673e-06,
+      "loss": 0.0665,
+      "step": 637
+    },
+    {
+      "epoch": 0.9565217391304348,
+      "grad_norm": 0.6702476378862521,
+      "learning_rate": 5.341212066823356e-06,
+      "loss": 0.0736,
+      "step": 638
+    },
+    {
+      "epoch": 0.9580209895052474,
+      "grad_norm": 0.6646789416784814,
+      "learning_rate": 5.329463497015969e-06,
+      "loss": 0.0805,
+      "step": 639
+    },
+    {
+      "epoch": 0.95952023988006,
+      "grad_norm": 0.6208063778556249,
+      "learning_rate": 5.317713099970283e-06,
+      "loss": 0.0687,
+      "step": 640
+    },
+    {
+      "epoch": 0.9610194902548725,
+      "grad_norm": 0.6646664867114013,
+      "learning_rate": 5.305960940855205e-06,
+      "loss": 0.0931,
+      "step": 641
+    },
+    {
+      "epoch": 0.9625187406296851,
+      "grad_norm": 0.773673377739417,
+      "learning_rate": 5.294207084849412e-06,
+      "loss": 0.0749,
+      "step": 642
+    },
+    {
+      "epoch": 0.9640179910044977,
+      "grad_norm": 0.5691156743493143,
+      "learning_rate": 5.282451597140994e-06,
+      "loss": 0.0612,
+      "step": 643
+    },
+    {
+      "epoch": 0.9655172413793104,
+      "grad_norm": 0.8210759321579921,
+      "learning_rate": 5.270694542927089e-06,
+      "loss": 0.094,
+      "step": 644
+    },
+    {
+      "epoch": 0.967016491754123,
+      "grad_norm": 1.0351037745586606,
+      "learning_rate": 5.258935987413524e-06,
+      "loss": 0.1127,
+      "step": 645
+    },
+    {
+      "epoch": 0.9685157421289355,
+      "grad_norm": 0.6419698201188782,
+      "learning_rate": 5.247175995814452e-06,
+      "loss": 0.0687,
+      "step": 646
+    },
+    {
+      "epoch": 0.9700149925037481,
+      "grad_norm": 0.8251466068510769,
+      "learning_rate": 5.235414633351992e-06,
+      "loss": 0.0742,
+      "step": 647
+    },
+    {
+      "epoch": 0.9715142428785607,
+      "grad_norm": 0.7318516788114338,
+      "learning_rate": 5.223651965255864e-06,
+      "loss": 0.1016,
+      "step": 648
+    },
+    {
+      "epoch": 0.9730134932533733,
+      "grad_norm": 0.6111101224797182,
+      "learning_rate": 5.211888056763029e-06,
+      "loss": 0.0594,
+      "step": 649
+    },
+    {
+      "epoch": 0.974512743628186,
+      "grad_norm": 0.8731300188694648,
+      "learning_rate": 5.20012297311733e-06,
+      "loss": 0.0978,
+      "step": 650
+    },
+    {
+      "epoch": 0.9760119940029985,
+      "grad_norm": 0.5552994145502074,
+      "learning_rate": 5.188356779569125e-06,
+      "loss": 0.0695,
+      "step": 651
+    },
+    {
+      "epoch": 0.9775112443778111,
+      "grad_norm": 0.8733148754539058,
+      "learning_rate": 5.176589541374929e-06,
+      "loss": 0.0839,
+      "step": 652
+    },
+    {
+      "epoch": 0.9790104947526237,
+      "grad_norm": 0.8216213053755895,
+      "learning_rate": 5.164821323797051e-06,
+      "loss": 0.0815,
+      "step": 653
+    },
+    {
+      "epoch": 0.9805097451274363,
+      "grad_norm": 0.769268596588045,
+      "learning_rate": 5.1530521921032305e-06,
+      "loss": 0.0751,
+      "step": 654
+    },
+    {
+      "epoch": 0.9820089955022488,
+      "grad_norm": 0.7686970290394918,
+      "learning_rate": 5.141282211566276e-06,
+      "loss": 0.0909,
+      "step": 655
+    },
+    {
+      "epoch": 0.9835082458770614,
+      "grad_norm": 0.7361545924589357,
+      "learning_rate": 5.129511447463705e-06,
+      "loss": 0.0641,
+      "step": 656
+    },
+    {
+      "epoch": 0.9850074962518741,
+      "grad_norm": 0.8497922349082441,
+      "learning_rate": 5.117739965077382e-06,
+      "loss": 0.0965,
+      "step": 657
+    },
+    {
+      "epoch": 0.9865067466266867,
+      "grad_norm": 0.5597631676095164,
+      "learning_rate": 5.105967829693155e-06,
+      "loss": 0.057,
+      "step": 658
+    },
+    {
+      "epoch": 0.9880059970014993,
+      "grad_norm": 0.7515200230450869,
+      "learning_rate": 5.0941951066004906e-06,
+      "loss": 0.0787,
+      "step": 659
+    },
+    {
+      "epoch": 0.9895052473763118,
+      "grad_norm": 0.7426483907036393,
+      "learning_rate": 5.082421861092116e-06,
+      "loss": 0.0766,
+      "step": 660
+    },
+    {
+      "epoch": 0.9910044977511244,
+      "grad_norm": 0.6370069569769923,
+      "learning_rate": 5.0706481584636605e-06,
+      "loss": 0.082,
+      "step": 661
+    },
+    {
+      "epoch": 0.992503748125937,
+      "grad_norm": 0.7018894258304225,
+      "learning_rate": 5.0588740640132805e-06,
+      "loss": 0.0669,
+      "step": 662
+    },
+    {
+      "epoch": 0.9940029985007496,
+      "grad_norm": 0.6896676555016269,
+      "learning_rate": 5.047099643041312e-06,
+      "loss": 0.0652,
+      "step": 663
+    },
+    {
+      "epoch": 0.9955022488755623,
+      "grad_norm": 0.6384344355358029,
+      "learning_rate": 5.0353249608499e-06,
+      "loss": 0.0812,
+      "step": 664
+    },
+    {
+      "epoch": 0.9970014992503748,
+      "grad_norm": 0.8819867254034709,
+      "learning_rate": 5.023550082742637e-06,
+      "loss": 0.1187,
+      "step": 665
+    },
+    {
+      "epoch": 0.9985007496251874,
+      "grad_norm": 0.8208911641644168,
+      "learning_rate": 5.011775074024202e-06,
+      "loss": 0.0775,
+      "step": 666
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.5439240968842436,
+      "learning_rate": 5e-06,
+      "loss": 0.043,
+      "step": 667
+    },
+    {
+      "epoch": 1.0014992503748126,
+      "grad_norm": 0.44871006604988073,
+      "learning_rate": 4.988224925975799e-06,
+      "loss": 0.0377,
+      "step": 668
+    },
+    {
+      "epoch": 1.0029985007496252,
+      "grad_norm": 0.4626367231252376,
+      "learning_rate": 4.976449917257365e-06,
+      "loss": 0.032,
+      "step": 669
+    },
+    {
+      "epoch": 1.0044977511244377,
+      "grad_norm": 0.4682691377197609,
+      "learning_rate": 4.964675039150102e-06,
+      "loss": 0.0384,
+      "step": 670
+    },
+    {
+      "epoch": 1.0059970014992503,
+      "grad_norm": 0.5889398997612483,
+      "learning_rate": 4.952900356958689e-06,
+      "loss": 0.0482,
+      "step": 671
+    },
+    {
+      "epoch": 1.0074962518740629,
+      "grad_norm": 0.47732664641910144,
+      "learning_rate": 4.941125935986721e-06,
+      "loss": 0.0349,
+      "step": 672
+    },
+    {
+      "epoch": 1.0089955022488755,
+      "grad_norm": 0.4226617325738613,
+      "learning_rate": 4.929351841536342e-06,
+      "loss": 0.0362,
+      "step": 673
+    },
+    {
+      "epoch": 1.0104947526236883,
+      "grad_norm": 0.3772154933809679,
+      "learning_rate": 4.917578138907884e-06,
+      "loss": 0.0347,
+      "step": 674
+    },
+    {
+      "epoch": 1.0119940029985008,
+      "grad_norm": 0.5463837719659405,
+      "learning_rate": 4.90580489339951e-06,
+      "loss": 0.0392,
+      "step": 675
+    },
+    {
+      "epoch": 1.0134932533733134,
+      "grad_norm": 0.4425217497627023,
+      "learning_rate": 4.894032170306846e-06,
+      "loss": 0.0294,
+      "step": 676
+    },
+    {
+      "epoch": 1.014992503748126,
+      "grad_norm": 0.32715715078421687,
+      "learning_rate": 4.882260034922618e-06,
+      "loss": 0.0233,
+      "step": 677
+    },
+    {
+      "epoch": 1.0164917541229386,
+      "grad_norm": 0.31749476859189457,
+      "learning_rate": 4.870488552536296e-06,
+      "loss": 0.0223,
+      "step": 678
+    },
+    {
+      "epoch": 1.0179910044977512,
+      "grad_norm": 0.5928620511432154,
+      "learning_rate": 4.858717788433725e-06,
+      "loss": 0.048,
+      "step": 679
+    },
+    {
+      "epoch": 1.0194902548725637,
+      "grad_norm": 0.5276116656343903,
+      "learning_rate": 4.846947807896771e-06,
+      "loss": 0.0512,
+      "step": 680
+    },
+    {
+      "epoch": 1.0209895052473763,
+      "grad_norm": 0.5151493544053687,
+      "learning_rate": 4.83517867620295e-06,
+      "loss": 0.0301,
+      "step": 681
+    },
+    {
+      "epoch": 1.0224887556221889,
+      "grad_norm": 0.3267540780111297,
+      "learning_rate": 4.823410458625072e-06,
+      "loss": 0.0245,
+      "step": 682
+    },
+    {
+      "epoch": 1.0239880059970015,
+      "grad_norm": 0.4016619386617154,
+      "learning_rate": 4.811643220430877e-06,
+      "loss": 0.0293,
+      "step": 683
+    },
+    {
+      "epoch": 1.025487256371814,
+      "grad_norm": 0.5754443252275956,
+      "learning_rate": 4.7998770268826726e-06,
+      "loss": 0.0431,
+      "step": 684
+    },
+    {
+      "epoch": 1.0269865067466266,
+      "grad_norm": 0.3728301832266742,
+      "learning_rate": 4.788111943236973e-06,
+      "loss": 0.0245,
+      "step": 685
+    },
+    {
+      "epoch": 1.0284857571214392,
+      "grad_norm": 0.4564670136993856,
+      "learning_rate": 4.7763480347441395e-06,
+      "loss": 0.0335,
+      "step": 686
+    },
+    {
+      "epoch": 1.0299850074962518,
+      "grad_norm": 0.31809191754805405,
+      "learning_rate": 4.7645853666480104e-06,
+      "loss": 0.0262,
+      "step": 687
+    },
+    {
+      "epoch": 1.0314842578710646,
+      "grad_norm": 0.4968173690649472,
+      "learning_rate": 4.752824004185548e-06,
+      "loss": 0.0343,
+      "step": 688
+    },
+    {
+      "epoch": 1.0329835082458771,
+      "grad_norm": 0.4535204343497408,
+      "learning_rate": 4.7410640125864785e-06,
+      "loss": 0.0247,
+      "step": 689
+    },
+    {
+      "epoch": 1.0344827586206897,
+      "grad_norm": 0.5887833496782148,
+      "learning_rate": 4.729305457072913e-06,
+      "loss": 0.039,
+      "step": 690
+    },
+    {
+      "epoch": 1.0359820089955023,
+      "grad_norm": 0.34817772612954323,
+      "learning_rate": 4.717548402859008e-06,
+      "loss": 0.0257,
+      "step": 691
+    },
+    {
+      "epoch": 1.0374812593703149,
+      "grad_norm": 0.5296453175128691,
+      "learning_rate": 4.7057929151505895e-06,
+      "loss": 0.0466,
+      "step": 692
+    },
+    {
+      "epoch": 1.0389805097451275,
+      "grad_norm": 0.4938928150122724,
+      "learning_rate": 4.694039059144797e-06,
+      "loss": 0.0286,
+      "step": 693
+    },
+    {
+      "epoch": 1.04047976011994,
+      "grad_norm": 0.45296821233260565,
+      "learning_rate": 4.6822869000297185e-06,
+      "loss": 0.0323,
+      "step": 694
+    },
+    {
+      "epoch": 1.0419790104947526,
+      "grad_norm": 0.7099928503078203,
+      "learning_rate": 4.670536502984033e-06,
+      "loss": 0.0365,
+      "step": 695
+    },
+    {
+      "epoch": 1.0434782608695652,
+      "grad_norm": 0.5297011518522096,
+      "learning_rate": 4.6587879331766465e-06,
+      "loss": 0.028,
+      "step": 696
+    },
+    {
+      "epoch": 1.0449775112443778,
+      "grad_norm": 0.39600114167240236,
+      "learning_rate": 4.647041255766329e-06,
+      "loss": 0.0225,
+      "step": 697
+    },
+    {
+      "epoch": 1.0464767616191903,
+      "grad_norm": 0.4047188519293251,
+      "learning_rate": 4.6352965359013576e-06,
+      "loss": 0.0227,
+      "step": 698
+    },
+    {
+      "epoch": 1.047976011994003,
+      "grad_norm": 0.575812363197737,
+      "learning_rate": 4.623553838719151e-06,
+      "loss": 0.0349,
+      "step": 699
+    },
+    {
+      "epoch": 1.0494752623688155,
+      "grad_norm": 0.47122699195261175,
+      "learning_rate": 4.611813229345911e-06,
+      "loss": 0.0281,
+      "step": 700
+    },
+    {
+      "epoch": 1.050974512743628,
+      "grad_norm": 0.39019358461173803,
+      "learning_rate": 4.6000747728962606e-06,
+      "loss": 0.0248,
+      "step": 701
+    },
+    {
+      "epoch": 1.0524737631184409,
+      "grad_norm": 0.5322897929473427,
+      "learning_rate": 4.588338534472878e-06,
+      "loss": 0.0204,
+      "step": 702
+    },
+    {
+      "epoch": 1.0539730134932535,
+      "grad_norm": 0.8348431975997683,
+      "learning_rate": 4.576604579166147e-06,
+      "loss": 0.0481,
+      "step": 703
+    },
+    {
+      "epoch": 1.055472263868066,
+      "grad_norm": 0.6794818028037216,
+      "learning_rate": 4.564872972053786e-06,
+      "loss": 0.0427,
+      "step": 704
+    },
+    {
+      "epoch": 1.0569715142428786,
+      "grad_norm": 0.663147027131612,
+      "learning_rate": 4.553143778200486e-06,
+      "loss": 0.0499,
+      "step": 705
+    },
+    {
+      "epoch": 1.0584707646176912,
+      "grad_norm": 0.5115389470080361,
+      "learning_rate": 4.541417062657561e-06,
+      "loss": 0.0343,
+      "step": 706
+    },
+    {
+      "epoch": 1.0599700149925038,
+      "grad_norm": 0.45483139918208904,
+      "learning_rate": 4.529692890462574e-06,
+      "loss": 0.0217,
+      "step": 707
+    },
+    {
+      "epoch": 1.0614692653673163,
+      "grad_norm": 0.5274630261976188,
+      "learning_rate": 4.5179713266389866e-06,
+      "loss": 0.0301,
+      "step": 708
+    },
+    {
+      "epoch": 1.062968515742129,
+      "grad_norm": 0.4539841246070867,
+      "learning_rate": 4.50625243619579e-06,
+      "loss": 0.0255,
+      "step": 709
+    },
+    {
+      "epoch": 1.0644677661169415,
+      "grad_norm": 0.6572446536503337,
+      "learning_rate": 4.494536284127155e-06,
+      "loss": 0.0151,
+      "step": 710
+    },
+    {
+      "epoch": 1.065967016491754,
+      "grad_norm": 0.43025641832941247,
+      "learning_rate": 4.4828229354120565e-06,
+      "loss": 0.0163,
+      "step": 711
+    },
+    {
+      "epoch": 1.0674662668665666,
+      "grad_norm": 0.6945621646079666,
+      "learning_rate": 4.471112455013928e-06,
+      "loss": 0.0314,
+      "step": 712
+    },
+    {
+      "epoch": 1.0689655172413792,
+      "grad_norm": 0.6574562756042209,
+      "learning_rate": 4.459404907880293e-06,
+      "loss": 0.0302,
+      "step": 713
+    },
+    {
+      "epoch": 1.0704647676161918,
+      "grad_norm": 0.5794298932450466,
+      "learning_rate": 4.447700358942407e-06,
+      "loss": 0.0405,
+      "step": 714
+    },
+    {
+      "epoch": 1.0719640179910046,
+      "grad_norm": 0.5092056079830732,
+      "learning_rate": 4.435998873114895e-06,
+      "loss": 0.0254,
+      "step": 715
+    },
+    {
+      "epoch": 1.0734632683658172,
+      "grad_norm": 0.547423211920376,
+      "learning_rate": 4.424300515295401e-06,
+      "loss": 0.0289,
+      "step": 716
+    },
+    {
+      "epoch": 1.0749625187406298,
+      "grad_norm": 0.4601404267422463,
+      "learning_rate": 4.412605350364213e-06,
+      "loss": 0.02,
+      "step": 717
+    },
+    {
+      "epoch": 1.0764617691154423,
+      "grad_norm": 0.5429181061267581,
+      "learning_rate": 4.400913443183913e-06,
+      "loss": 0.0231,
+      "step": 718
+    },
+    {
+      "epoch": 1.077961019490255,
+      "grad_norm": 0.6555338093023101,
+      "learning_rate": 4.389224858599015e-06,
+      "loss": 0.036,
+      "step": 719
+    },
+    {
+      "epoch": 1.0794602698650675,
+      "grad_norm": 0.5331402601175965,
+      "learning_rate": 4.377539661435608e-06,
+      "loss": 0.0217,
+      "step": 720
+    },
+    {
+      "epoch": 1.08095952023988,
+      "grad_norm": 0.6898518867222653,
+      "learning_rate": 4.365857916500991e-06,
+      "loss": 0.0409,
+      "step": 721
+    },
+    {
+      "epoch": 1.0824587706146926,
+      "grad_norm": 0.6344143085382405,
+      "learning_rate": 4.35417968858332e-06,
+      "loss": 0.0297,
+      "step": 722
+    },
+    {
+      "epoch": 1.0839580209895052,
+      "grad_norm": 0.5960991105782771,
+      "learning_rate": 4.3425050424512405e-06,
+      "loss": 0.0274,
+      "step": 723
+    },
+    {
+      "epoch": 1.0854572713643178,
+      "grad_norm": 0.7905843374349434,
+      "learning_rate": 4.330834042853537e-06,
+      "loss": 0.0303,
+      "step": 724
+    },
+    {
+      "epoch": 1.0869565217391304,
+      "grad_norm": 0.8612311207352599,
+      "learning_rate": 4.319166754518768e-06,
+      "loss": 0.0316,
+      "step": 725
+    },
+    {
+      "epoch": 1.088455772113943,
+      "grad_norm": 0.5149608722666438,
+      "learning_rate": 4.30750324215491e-06,
+      "loss": 0.0207,
+      "step": 726
+    },
+    {
+      "epoch": 1.0899550224887555,
+      "grad_norm": 0.5007357381458054,
+      "learning_rate": 4.295843570448998e-06,
+      "loss": 0.0245,
+      "step": 727
+    },
+    {
+      "epoch": 1.0914542728635683,
+      "grad_norm": 0.5376125508528267,
+      "learning_rate": 4.284187804066764e-06,
+      "loss": 0.0267,
+      "step": 728
+    },
+    {
+      "epoch": 1.092953523238381,
+      "grad_norm": 0.7020291216542087,
+      "learning_rate": 4.272536007652281e-06,
+      "loss": 0.0246,
+      "step": 729
+    },
+    {
+      "epoch": 1.0944527736131935,
+      "grad_norm": 0.5221959881869965,
+      "learning_rate": 4.260888245827608e-06,
+      "loss": 0.0218,
+      "step": 730
+    },
+    {
+      "epoch": 1.095952023988006,
+      "grad_norm": 0.4776342371360563,
+      "learning_rate": 4.249244583192425e-06,
+      "loss": 0.0263,
+      "step": 731
+    },
+    {
+      "epoch": 1.0974512743628186,
+      "grad_norm": 0.5093197148751731,
+      "learning_rate": 4.237605084323676e-06,
+      "loss": 0.019,
+      "step": 732
+    },
+    {
+      "epoch": 1.0989505247376312,
+      "grad_norm": 0.7474665552429989,
+      "learning_rate": 4.225969813775215e-06,
+      "loss": 0.0306,
+      "step": 733
+    },
+    {
+      "epoch": 1.1004497751124438,
+      "grad_norm": 0.6671071411127204,
+      "learning_rate": 4.214338836077444e-06,
+      "loss": 0.0284,
+      "step": 734
+    },
+    {
+      "epoch": 1.1019490254872564,
+      "grad_norm": 0.6001838653602947,
+      "learning_rate": 4.202712215736955e-06,
+      "loss": 0.022,
+      "step": 735
+    },
+    {
+      "epoch": 1.103448275862069,
+      "grad_norm": 0.6969543846110181,
+      "learning_rate": 4.191090017236177e-06,
+      "loss": 0.0291,
+      "step": 736
+    },
+    {
+      "epoch": 1.1049475262368815,
+      "grad_norm": 0.5339193895071973,
+      "learning_rate": 4.1794723050330125e-06,
+      "loss": 0.0216,
+      "step": 737
+    },
+    {
+      "epoch": 1.106446776611694,
+      "grad_norm": 0.5828083458283416,
+      "learning_rate": 4.167859143560484e-06,
+      "loss": 0.0236,
+      "step": 738
+    },
+    {
+      "epoch": 1.1079460269865067,
+      "grad_norm": 0.5285796176790518,
+      "learning_rate": 4.1562505972263735e-06,
+      "loss": 0.0241,
+      "step": 739
+    },
+    {
+      "epoch": 1.1094452773613193,
+      "grad_norm": 0.6635891666388966,
+      "learning_rate": 4.144646730412868e-06,
+      "loss": 0.0321,
+      "step": 740
+    },
+    {
+      "epoch": 1.1109445277361318,
+      "grad_norm": 0.5498494988966571,
+      "learning_rate": 4.133047607476202e-06,
+      "loss": 0.0302,
+      "step": 741
+    },
+    {
+      "epoch": 1.1124437781109444,
+      "grad_norm": 0.5015652827490643,
+      "learning_rate": 4.121453292746297e-06,
+      "loss": 0.0257,
+      "step": 742
+    },
+    {
+      "epoch": 1.1139430284857572,
+      "grad_norm": 0.7908629236157562,
+      "learning_rate": 4.109863850526413e-06,
+      "loss": 0.034,
+      "step": 743
+    },
+    {
+      "epoch": 1.1154422788605698,
+      "grad_norm": 0.5250286115109797,
+      "learning_rate": 4.098279345092783e-06,
+      "loss": 0.0217,
+      "step": 744
+    },
+    {
+      "epoch": 1.1169415292353824,
+      "grad_norm": 0.5410427092574731,
+      "learning_rate": 4.086699840694262e-06,
+      "loss": 0.0198,
+      "step": 745
+    },
+    {
+      "epoch": 1.118440779610195,
+      "grad_norm": 0.541600639581417,
+      "learning_rate": 4.075125401551963e-06,
+      "loss": 0.0209,
+      "step": 746
+    },
+    {
+      "epoch": 1.1199400299850075,
+      "grad_norm": 0.6504489381634044,
+      "learning_rate": 4.063556091858914e-06,
+      "loss": 0.0315,
+      "step": 747
+    },
+    {
+      "epoch": 1.12143928035982,
+      "grad_norm": 0.7144892730830451,
+      "learning_rate": 4.051991975779691e-06,
+      "loss": 0.0348,
+      "step": 748
+    },
+    {
+      "epoch": 1.1229385307346327,
+      "grad_norm": 0.6723002864100808,
+      "learning_rate": 4.040433117450066e-06,
+      "loss": 0.0289,
+      "step": 749
+    },
+    {
+      "epoch": 1.1244377811094453,
+      "grad_norm": 0.7068510871953166,
+      "learning_rate": 4.0288795809766516e-06,
+      "loss": 0.0354,
+      "step": 750
+    },
+    {
+      "epoch": 1.1259370314842578,
+      "grad_norm": 0.6303198709965698,
+      "learning_rate": 4.017331430436543e-06,
+      "loss": 0.034,
+      "step": 751
+    },
+    {
+      "epoch": 1.1274362818590704,
+      "grad_norm": 0.8881767708416839,
+      "learning_rate": 4.005788729876968e-06,
+      "loss": 0.0302,
+      "step": 752
+    },
+    {
+      "epoch": 1.128935532233883,
+      "grad_norm": 0.6438890353901902,
+      "learning_rate": 3.994251543314925e-06,
+      "loss": 0.0286,
+      "step": 753
+    },
+    {
+      "epoch": 1.1304347826086956,
+      "grad_norm": 0.6425012182285248,
+      "learning_rate": 3.982719934736832e-06,
+      "loss": 0.0269,
+      "step": 754
+    },
+    {
+      "epoch": 1.1319340329835081,
+      "grad_norm": 0.6004107335560911,
+      "learning_rate": 3.971193968098172e-06,
+      "loss": 0.0368,
+      "step": 755
+    },
+    {
+      "epoch": 1.133433283358321,
+      "grad_norm": 0.5868499917306065,
+      "learning_rate": 3.959673707323135e-06,
+      "loss": 0.0351,
+      "step": 756
+    },
+    {
+      "epoch": 1.1349325337331335,
+      "grad_norm": 0.6559634762147885,
+      "learning_rate": 3.948159216304269e-06,
+      "loss": 0.0335,
+      "step": 757
+    },
+    {
+      "epoch": 1.136431784107946,
+      "grad_norm": 0.6141306569280756,
+      "learning_rate": 3.93665055890212e-06,
+      "loss": 0.025,
+      "step": 758
+    },
+    {
+      "epoch": 1.1379310344827587,
+      "grad_norm": 0.7327156027824521,
+      "learning_rate": 3.92514779894488e-06,
+      "loss": 0.0376,
+      "step": 759
+    },
+    {
+      "epoch": 1.1394302848575713,
+      "grad_norm": 0.46590906246414887,
+      "learning_rate": 3.9136510002280344e-06,
+      "loss": 0.0238,
+      "step": 760
+    },
+    {
+      "epoch": 1.1409295352323838,
+      "grad_norm": 0.5379387831227852,
+      "learning_rate": 3.902160226514007e-06,
+      "loss": 0.0242,
+      "step": 761
+    },
+    {
+      "epoch": 1.1424287856071964,
+      "grad_norm": 0.5985403483040879,
+      "learning_rate": 3.8906755415318045e-06,
+      "loss": 0.0314,
+      "step": 762
+    },
+    {
+      "epoch": 1.143928035982009,
+      "grad_norm": 0.6212919005903046,
+      "learning_rate": 3.8791970089766665e-06,
+      "loss": 0.0327,
+      "step": 763
+    },
+    {
+      "epoch": 1.1454272863568216,
+      "grad_norm": 0.6306348449395626,
+      "learning_rate": 3.86772469250971e-06,
+      "loss": 0.0254,
+      "step": 764
+    },
+    {
+      "epoch": 1.1469265367316341,
+      "grad_norm": 0.5024312665813245,
+      "learning_rate": 3.856258655757578e-06,
+      "loss": 0.018,
+      "step": 765
+    },
+    {
+      "epoch": 1.1484257871064467,
+      "grad_norm": 0.6607146962676326,
+      "learning_rate": 3.844798962312085e-06,
+      "loss": 0.0392,
+      "step": 766
+    },
+    {
+      "epoch": 1.1499250374812593,
+      "grad_norm": 0.5866244770338371,
+      "learning_rate": 3.833345675729863e-06,
+      "loss": 0.0321,
+      "step": 767
+    },
+    {
+      "epoch": 1.1514242878560719,
+      "grad_norm": 0.6970517527551673,
+      "learning_rate": 3.821898859532013e-06,
+      "loss": 0.0325,
+      "step": 768
+    },
+    {
+      "epoch": 1.1529235382308847,
+      "grad_norm": 0.6195406213104157,
+      "learning_rate": 3.8104585772037493e-06,
+      "loss": 0.0217,
+      "step": 769
+    },
+    {
+      "epoch": 1.1544227886056972,
+      "grad_norm": 0.6335792115599321,
+      "learning_rate": 3.7990248921940485e-06,
+      "loss": 0.0394,
+      "step": 770
+    },
+    {
+      "epoch": 1.1559220389805098,
+      "grad_norm": 0.7058753861870792,
+      "learning_rate": 3.787597867915303e-06,
+      "loss": 0.0318,
+      "step": 771
+    },
+    {
+      "epoch": 1.1574212893553224,
+      "grad_norm": 0.7410976098936325,
+      "learning_rate": 3.7761775677429567e-06,
+      "loss": 0.0361,
+      "step": 772
+    },
+    {
+      "epoch": 1.158920539730135,
+      "grad_norm": 0.5556080752912006,
+      "learning_rate": 3.7647640550151666e-06,
+      "loss": 0.0272,
+      "step": 773
+    },
+    {
+      "epoch": 1.1604197901049476,
+      "grad_norm": 0.6365404910925648,
+      "learning_rate": 3.7533573930324395e-06,
+      "loss": 0.0275,
+      "step": 774
+    },
+    {
+      "epoch": 1.1619190404797601,
+      "grad_norm": 0.5667593677001239,
+      "learning_rate": 3.7419576450572924e-06,
+      "loss": 0.0301,
+      "step": 775
+    },
+    {
+      "epoch": 1.1634182908545727,
+      "grad_norm": 0.6336069871786666,
+      "learning_rate": 3.7305648743138966e-06,
+      "loss": 0.0268,
+      "step": 776
+    },
+    {
+      "epoch": 1.1649175412293853,
+      "grad_norm": 0.5706824349393088,
+      "learning_rate": 3.7191791439877236e-06,
+      "loss": 0.0237,
+      "step": 777
+    },
+    {
+      "epoch": 1.1664167916041979,
+      "grad_norm": 0.6073680675699474,
+      "learning_rate": 3.7078005172252015e-06,
+      "loss": 0.023,
+      "step": 778
+    },
+    {
+      "epoch": 1.1679160419790104,
+      "grad_norm": 0.573002363285938,
+      "learning_rate": 3.6964290571333583e-06,
+      "loss": 0.0314,
+      "step": 779
+    },
+    {
+      "epoch": 1.169415292353823,
+      "grad_norm": 0.6274077343463547,
+      "learning_rate": 3.6850648267794776e-06,
+      "loss": 0.0275,
+      "step": 780
+    },
+    {
+      "epoch": 1.1709145427286356,
+      "grad_norm": 0.5245270347789064,
+      "learning_rate": 3.673707889190744e-06,
+      "loss": 0.0189,
+      "step": 781
+    },
+    {
+      "epoch": 1.1724137931034484,
+      "grad_norm": 0.5635782873594429,
+      "learning_rate": 3.662358307353897e-06,
+      "loss": 0.0258,
+      "step": 782
+    },
+    {
+      "epoch": 1.1739130434782608,
+      "grad_norm": 0.566053898640951,
+      "learning_rate": 3.6510161442148783e-06,
+      "loss": 0.0181,
+      "step": 783
+    },
+    {
+      "epoch": 1.1754122938530736,
+      "grad_norm": 0.6181889728730517,
+      "learning_rate": 3.63968146267849e-06,
+      "loss": 0.0342,
+      "step": 784
+    },
+    {
+      "epoch": 1.1769115442278861,
+      "grad_norm": 0.7044152890341429,
+      "learning_rate": 3.6283543256080334e-06,
+      "loss": 0.036,
+      "step": 785
+    },
+    {
+      "epoch": 1.1784107946026987,
+      "grad_norm": 0.6543557274392708,
+      "learning_rate": 3.6170347958249728e-06,
+      "loss": 0.0304,
+      "step": 786
+    },
+    {
+      "epoch": 1.1799100449775113,
+      "grad_norm": 0.629610081448567,
+      "learning_rate": 3.605722936108579e-06,
+      "loss": 0.0241,
+      "step": 787
+    },
+    {
+      "epoch": 1.1814092953523239,
+      "grad_norm": 0.7168773850747961,
+      "learning_rate": 3.5944188091955843e-06,
+      "loss": 0.0299,
+      "step": 788
+    },
+    {
+      "epoch": 1.1829085457271364,
+      "grad_norm": 0.40958425442826546,
+      "learning_rate": 3.5831224777798346e-06,
+      "loss": 0.0161,
+      "step": 789
+    },
+    {
+      "epoch": 1.184407796101949,
+      "grad_norm": 0.42238113998660237,
+      "learning_rate": 3.5718340045119416e-06,
+      "loss": 0.0175,
+      "step": 790
+    },
+    {
+      "epoch": 1.1859070464767616,
+      "grad_norm": 0.5101963062313535,
+      "learning_rate": 3.5605534519989327e-06,
+      "loss": 0.0215,
+      "step": 791
+    },
+    {
+      "epoch": 1.1874062968515742,
+      "grad_norm": 0.4384357069865695,
+      "learning_rate": 3.5492808828039083e-06,
+      "loss": 0.0188,
+      "step": 792
+    },
+    {
+      "epoch": 1.1889055472263867,
+      "grad_norm": 0.47400009716341124,
+      "learning_rate": 3.538016359445692e-06,
+      "loss": 0.0266,
+      "step": 793
+    },
+    {
+      "epoch": 1.1904047976011993,
+      "grad_norm": 0.6387504081320909,
+      "learning_rate": 3.5267599443984848e-06,
+      "loss": 0.029,
+      "step": 794
+    },
+    {
+      "epoch": 1.191904047976012,
+      "grad_norm": 0.616428547454195,
+      "learning_rate": 3.5155117000915153e-06,
+      "loss": 0.0415,
+      "step": 795
+    },
+    {
+      "epoch": 1.1934032983508245,
+      "grad_norm": 0.5447730710568164,
+      "learning_rate": 3.5042716889086998e-06,
+      "loss": 0.0235,
+      "step": 796
+    },
+    {
+      "epoch": 1.1949025487256373,
+      "grad_norm": 0.6459604094136135,
+      "learning_rate": 3.493039973188289e-06,
+      "loss": 0.0252,
+      "step": 797
+    },
+    {
+      "epoch": 1.1964017991004499,
+      "grad_norm": 0.5229156252300454,
+      "learning_rate": 3.481816615222533e-06,
+      "loss": 0.0233,
+      "step": 798
+    },
+    {
+      "epoch": 1.1979010494752624,
+      "grad_norm": 0.5370806510477363,
+      "learning_rate": 3.470601677257323e-06,
+      "loss": 0.0291,
+      "step": 799
+    },
+    {
+      "epoch": 1.199400299850075,
+      "grad_norm": 0.590949291947999,
+      "learning_rate": 3.459395221491853e-06,
+      "loss": 0.0234,
+      "step": 800
+    },
+    {
+      "epoch": 1.199400299850075,
+      "eval_loss": 0.10691406577825546,
+      "eval_runtime": 2.1934,
+      "eval_samples_per_second": 24.62,
+      "eval_steps_per_second": 6.383,
+      "step": 800
+    },
+    {
+      "epoch": 1.2008995502248876,
+      "grad_norm": 0.48913487713175335,
+      "learning_rate": 3.4481973100782756e-06,
+      "loss": 0.0236,
+      "step": 801
+    },
+    {
+      "epoch": 1.2023988005997002,
+      "grad_norm": 0.6213005498554974,
+      "learning_rate": 3.4370080051213527e-06,
+      "loss": 0.0272,
+      "step": 802
+    },
+    {
+      "epoch": 1.2038980509745127,
+      "grad_norm": 0.5370116190590234,
+      "learning_rate": 3.4258273686781156e-06,
+      "loss": 0.0323,
+      "step": 803
+    },
+    {
+      "epoch": 1.2053973013493253,
+      "grad_norm": 0.6877183944384336,
+      "learning_rate": 3.4146554627575207e-06,
+      "loss": 0.0286,
+      "step": 804
+    },
+    {
+      "epoch": 1.206896551724138,
+      "grad_norm": 0.7164406993755414,
+      "learning_rate": 3.403492349320101e-06,
+      "loss": 0.0325,
+      "step": 805
+    },
+    {
+      "epoch": 1.2083958020989505,
+      "grad_norm": 0.42138066335378277,
+      "learning_rate": 3.392338090277628e-06,
+      "loss": 0.0158,
+      "step": 806
+    },
+    {
+      "epoch": 1.209895052473763,
+      "grad_norm": 0.5397858340721489,
+      "learning_rate": 3.3811927474927644e-06,
+      "loss": 0.0272,
+      "step": 807
+    },
+    {
+      "epoch": 1.2113943028485756,
+      "grad_norm": 0.47108631578390237,
+      "learning_rate": 3.3700563827787224e-06,
+      "loss": 0.0206,
+      "step": 808
+    },
+    {
+      "epoch": 1.2128935532233882,
+      "grad_norm": 0.477503017335954,
+      "learning_rate": 3.358929057898922e-06,
+      "loss": 0.0203,
+      "step": 809
+    },
+    {
+      "epoch": 1.214392803598201,
+      "grad_norm": 0.5500464252262549,
+      "learning_rate": 3.3478108345666456e-06,
+      "loss": 0.0249,
+      "step": 810
+    },
+    {
+      "epoch": 1.2158920539730136,
+      "grad_norm": 0.5817214885483699,
+      "learning_rate": 3.3367017744446995e-06,
+      "loss": 0.0321,
+      "step": 811
+    },
+    {
+      "epoch": 1.2173913043478262,
+      "grad_norm": 0.7082883249119462,
+      "learning_rate": 3.3256019391450696e-06,
+      "loss": 0.0329,
+      "step": 812
+    },
+    {
+      "epoch": 1.2188905547226387,
+      "grad_norm": 0.5664155057693675,
+      "learning_rate": 3.314511390228578e-06,
+      "loss": 0.026,
+      "step": 813
+    },
+    {
+      "epoch": 1.2203898050974513,
+      "grad_norm": 0.6345428071980368,
+      "learning_rate": 3.303430189204545e-06,
+      "loss": 0.0278,
+      "step": 814
+    },
+    {
+      "epoch": 1.221889055472264,
+      "grad_norm": 0.8125206925143831,
+      "learning_rate": 3.2923583975304474e-06,
+      "loss": 0.0429,
+      "step": 815
+    },
+    {
+      "epoch": 1.2233883058470765,
+      "grad_norm": 0.6246315633583271,
+      "learning_rate": 3.2812960766115747e-06,
+      "loss": 0.0255,
+      "step": 816
+    },
+    {
+      "epoch": 1.224887556221889,
+      "grad_norm": 0.7911694912504367,
+      "learning_rate": 3.270243287800693e-06,
+      "loss": 0.0377,
+      "step": 817
+    },
+    {
+      "epoch": 1.2263868065967016,
+      "grad_norm": 0.4924658502471017,
+      "learning_rate": 3.2592000923976997e-06,
+      "loss": 0.0205,
+      "step": 818
+    },
+    {
+      "epoch": 1.2278860569715142,
+      "grad_norm": 0.596876506224773,
+      "learning_rate": 3.2481665516492876e-06,
+      "loss": 0.0289,
+      "step": 819
+    },
+    {
+      "epoch": 1.2293853073463268,
+      "grad_norm": 0.5206204039479833,
+      "learning_rate": 3.2371427267486044e-06,
+      "loss": 0.0273,
+      "step": 820
+    },
+    {
+      "epoch": 1.2308845577211394,
+      "grad_norm": 0.5444304311733692,
+      "learning_rate": 3.2261286788349127e-06,
+      "loss": 0.0211,
+      "step": 821
+    },
+    {
+      "epoch": 1.232383808095952,
+      "grad_norm": 0.7170413630409371,
+      "learning_rate": 3.2151244689932505e-06,
+      "loss": 0.032,
+      "step": 822
+    },
+    {
+      "epoch": 1.2338830584707647,
+      "grad_norm": 0.6771012193958201,
+      "learning_rate": 3.2041301582540903e-06,
+      "loss": 0.0212,
+      "step": 823
+    },
+    {
+      "epoch": 1.235382308845577,
+      "grad_norm": 0.6380137368121873,
+      "learning_rate": 3.1931458075930046e-06,
+      "loss": 0.0286,
+      "step": 824
+    },
+    {
+      "epoch": 1.23688155922039,
+      "grad_norm": 0.5288132607900348,
+      "learning_rate": 3.182171477930332e-06,
+      "loss": 0.0226,
+      "step": 825
+    },
+    {
+      "epoch": 1.2383808095952025,
+      "grad_norm": 0.5252350610398268,
+      "learning_rate": 3.171207230130826e-06,
+      "loss": 0.0267,
+      "step": 826
+    },
+    {
+      "epoch": 1.239880059970015,
+      "grad_norm": 0.7030553880028503,
+      "learning_rate": 3.1602531250033286e-06,
+      "loss": 0.0283,
+      "step": 827
+    },
+    {
+      "epoch": 1.2413793103448276,
+      "grad_norm": 0.6414943971286622,
+      "learning_rate": 3.149309223300428e-06,
+      "loss": 0.0204,
+      "step": 828
+    },
+    {
+      "epoch": 1.2428785607196402,
+      "grad_norm": 0.7226296944360978,
+      "learning_rate": 3.1383755857181253e-06,
+      "loss": 0.032,
+      "step": 829
+    },
+    {
+      "epoch": 1.2443778110944528,
+      "grad_norm": 0.5330807842937307,
+      "learning_rate": 3.1274522728954928e-06,
+      "loss": 0.0259,
+      "step": 830
+    },
+    {
+      "epoch": 1.2458770614692654,
+      "grad_norm": 0.6645373620541577,
+      "learning_rate": 3.1165393454143423e-06,
+      "loss": 0.0303,
+      "step": 831
+    },
+    {
+      "epoch": 1.247376311844078,
+      "grad_norm": 0.7927136910325479,
+      "learning_rate": 3.1056368637988876e-06,
+      "loss": 0.0357,
+      "step": 832
+    },
+    {
+      "epoch": 1.2488755622188905,
+      "grad_norm": 0.5156357521632016,
+      "learning_rate": 3.0947448885154085e-06,
+      "loss": 0.0242,
+      "step": 833
+    },
+    {
+      "epoch": 1.250374812593703,
+      "grad_norm": 0.633468681663236,
+      "learning_rate": 3.0838634799719157e-06,
+      "loss": 0.0325,
+      "step": 834
+    },
+    {
+      "epoch": 1.2518740629685157,
+      "grad_norm": 0.41200736824648765,
+      "learning_rate": 3.072992698517815e-06,
+      "loss": 0.0215,
+      "step": 835
+    },
+    {
+      "epoch": 1.2533733133433285,
+      "grad_norm": 0.6746773867472416,
+      "learning_rate": 3.0621326044435738e-06,
+      "loss": 0.0328,
+      "step": 836
+    },
+    {
+      "epoch": 1.2548725637181408,
+      "grad_norm": 0.5475538559387787,
+      "learning_rate": 3.0512832579803873e-06,
+      "loss": 0.0329,
+      "step": 837
+    },
+    {
+      "epoch": 1.2563718140929536,
+      "grad_norm": 0.5743885209321663,
+      "learning_rate": 3.0404447192998398e-06,
+      "loss": 0.0235,
+      "step": 838
+    },
+    {
+      "epoch": 1.2578710644677662,
+      "grad_norm": 0.6310283968999627,
+      "learning_rate": 3.029617048513579e-06,
+      "loss": 0.0255,
+      "step": 839
+    },
+    {
+      "epoch": 1.2593703148425788,
+      "grad_norm": 0.46262697455126667,
+      "learning_rate": 3.0188003056729752e-06,
+      "loss": 0.0249,
+      "step": 840
+    },
+    {
+      "epoch": 1.2608695652173914,
+      "grad_norm": 0.6178739408907017,
+      "learning_rate": 3.007994550768793e-06,
+      "loss": 0.0319,
+      "step": 841
+    },
+    {
+      "epoch": 1.262368815592204,
+      "grad_norm": 0.6878475664459268,
+      "learning_rate": 2.9971998437308546e-06,
+      "loss": 0.0283,
+      "step": 842
+    },
+    {
+      "epoch": 1.2638680659670165,
+      "grad_norm": 0.72880044374734,
+      "learning_rate": 2.9864162444277118e-06,
+      "loss": 0.0433,
+      "step": 843
+    },
+    {
+      "epoch": 1.265367316341829,
+      "grad_norm": 0.555149056035918,
+      "learning_rate": 2.97564381266631e-06,
+      "loss": 0.0265,
+      "step": 844
+    },
+    {
+      "epoch": 1.2668665667166417,
+      "grad_norm": 0.47857228095767157,
+      "learning_rate": 2.964882608191659e-06,
+      "loss": 0.0192,
+      "step": 845
+    },
+    {
+      "epoch": 1.2683658170914542,
+      "grad_norm": 0.5679840184107534,
+      "learning_rate": 2.954132690686502e-06,
+      "loss": 0.0287,
+      "step": 846
+    },
+    {
+      "epoch": 1.2698650674662668,
+      "grad_norm": 0.5533483774832688,
+      "learning_rate": 2.9433941197709813e-06,
+      "loss": 0.0307,
+      "step": 847
+    },
+    {
+      "epoch": 1.2713643178410794,
+      "grad_norm": 0.5532010252971621,
+      "learning_rate": 2.9326669550023124e-06,
+      "loss": 0.0257,
+      "step": 848
+    },
+    {
+      "epoch": 1.272863568215892,
+      "grad_norm": 0.5091349939960508,
+      "learning_rate": 2.921951255874449e-06,
+      "loss": 0.0225,
+      "step": 849
+    },
+    {
+      "epoch": 1.2743628185907045,
+      "grad_norm": 0.5898877856296405,
+      "learning_rate": 2.9112470818177563e-06,
+      "loss": 0.0264,
+      "step": 850
+    },
+    {
+      "epoch": 1.2758620689655173,
+      "grad_norm": 0.5338686256235579,
+      "learning_rate": 2.9005544921986774e-06,
+      "loss": 0.0319,
+      "step": 851
+    },
+    {
+      "epoch": 1.2773613193403297,
+      "grad_norm": 0.5457938093823834,
+      "learning_rate": 2.8898735463194128e-06,
+      "loss": 0.0267,
+      "step": 852
+    },
+    {
+      "epoch": 1.2788605697151425,
+      "grad_norm": 0.6516449034189343,
+      "learning_rate": 2.8792043034175817e-06,
+      "loss": 0.0314,
+      "step": 853
+    },
+    {
+      "epoch": 1.280359820089955,
+      "grad_norm": 0.48597171410578927,
+      "learning_rate": 2.8685468226658974e-06,
+      "loss": 0.0278,
+      "step": 854
+    },
+    {
+      "epoch": 1.2818590704647677,
+      "grad_norm": 0.44948825133371684,
+      "learning_rate": 2.85790116317184e-06,
+      "loss": 0.0203,
+      "step": 855
+    },
+    {
+      "epoch": 1.2833583208395802,
+      "grad_norm": 0.5647847600225118,
+      "learning_rate": 2.8472673839773267e-06,
+      "loss": 0.028,
+      "step": 856
+    },
+    {
+      "epoch": 1.2848575712143928,
+      "grad_norm": 0.5662583080927529,
+      "learning_rate": 2.8366455440583874e-06,
+      "loss": 0.0275,
+      "step": 857
+    },
+    {
+      "epoch": 1.2863568215892054,
+      "grad_norm": 0.8401238011208961,
+      "learning_rate": 2.8260357023248323e-06,
+      "loss": 0.0417,
+      "step": 858
+    },
+    {
+      "epoch": 1.287856071964018,
+      "grad_norm": 0.6060119070806742,
+      "learning_rate": 2.815437917619932e-06,
+      "loss": 0.0297,
+      "step": 859
+    },
+    {
+      "epoch": 1.2893553223388305,
+      "grad_norm": 0.6392536844013035,
+      "learning_rate": 2.804852248720085e-06,
+      "loss": 0.0315,
+      "step": 860
+    },
+    {
+      "epoch": 1.2908545727136431,
+      "grad_norm": 0.4467812049817291,
+      "learning_rate": 2.7942787543344957e-06,
+      "loss": 0.0194,
+      "step": 861
+    },
+    {
+      "epoch": 1.2923538230884557,
+      "grad_norm": 0.46369070407506763,
+      "learning_rate": 2.783717493104846e-06,
+      "loss": 0.0151,
+      "step": 862
+    },
+    {
+      "epoch": 1.2938530734632683,
+      "grad_norm": 0.6175395799744641,
+      "learning_rate": 2.7731685236049745e-06,
+      "loss": 0.034,
+      "step": 863
+    },
+    {
+      "epoch": 1.295352323838081,
+      "grad_norm": 0.5445470815358577,
+      "learning_rate": 2.762631904340546e-06,
+      "loss": 0.0201,
+      "step": 864
+    },
+    {
+      "epoch": 1.2968515742128934,
+      "grad_norm": 0.6572098104465212,
+      "learning_rate": 2.7521076937487248e-06,
+      "loss": 0.025,
+      "step": 865
+    },
+    {
+      "epoch": 1.2983508245877062,
+      "grad_norm": 0.4379811822530108,
+      "learning_rate": 2.7415959501978674e-06,
+      "loss": 0.0138,
+      "step": 866
+    },
+    {
+      "epoch": 1.2998500749625188,
+      "grad_norm": 0.5181066360720411,
+      "learning_rate": 2.731096731987177e-06,
+      "loss": 0.0249,
+      "step": 867
+    },
+    {
+      "epoch": 1.3013493253373314,
+      "grad_norm": 0.7244558844223741,
+      "learning_rate": 2.7206100973463958e-06,
+      "loss": 0.0261,
+      "step": 868
+    },
+    {
+      "epoch": 1.302848575712144,
+      "grad_norm": 0.6222089297981163,
+      "learning_rate": 2.71013610443547e-06,
+      "loss": 0.0299,
+      "step": 869
+    },
+    {
+      "epoch": 1.3043478260869565,
+      "grad_norm": 0.5764439926719009,
+      "learning_rate": 2.6996748113442397e-06,
+      "loss": 0.0255,
+      "step": 870
+    },
+    {
+      "epoch": 1.3058470764617691,
+      "grad_norm": 0.4963085590209963,
+      "learning_rate": 2.689226276092107e-06,
+      "loss": 0.0195,
+      "step": 871
+    },
+    {
+      "epoch": 1.3073463268365817,
+      "grad_norm": 0.7172823878914694,
+      "learning_rate": 2.6787905566277185e-06,
+      "loss": 0.0365,
+      "step": 872
+    },
+    {
+      "epoch": 1.3088455772113943,
+      "grad_norm": 0.5369322171294056,
+      "learning_rate": 2.6683677108286423e-06,
+      "loss": 0.0258,
+      "step": 873
+    },
+    {
+      "epoch": 1.3103448275862069,
+      "grad_norm": 0.6630072691811933,
+      "learning_rate": 2.65795779650105e-06,
+      "loss": 0.0281,
+      "step": 874
+    },
+    {
+      "epoch": 1.3118440779610194,
+      "grad_norm": 0.6437274563904017,
+      "learning_rate": 2.6475608713793923e-06,
+      "loss": 0.0285,
+      "step": 875
+    },
+    {
+      "epoch": 1.313343328335832,
+      "grad_norm": 0.62025344306196,
+      "learning_rate": 2.6371769931260806e-06,
+      "loss": 0.0285,
+      "step": 876
+    },
+    {
+      "epoch": 1.3148425787106448,
+      "grad_norm": 0.6106179563309604,
+      "learning_rate": 2.6268062193311672e-06,
+      "loss": 0.0247,
+      "step": 877
+    },
+    {
+      "epoch": 1.3163418290854572,
+      "grad_norm": 0.5981230397704538,
+      "learning_rate": 2.6164486075120245e-06,
+      "loss": 0.0238,
+      "step": 878
+    },
+    {
+      "epoch": 1.31784107946027,
+      "grad_norm": 0.8346024787903114,
+      "learning_rate": 2.606104215113033e-06,
+      "loss": 0.0348,
+      "step": 879
+    },
+    {
+      "epoch": 1.3193403298350825,
+      "grad_norm": 0.5055843677070551,
+      "learning_rate": 2.5957730995052477e-06,
+      "loss": 0.021,
+      "step": 880
+    },
+    {
+      "epoch": 1.3208395802098951,
+      "grad_norm": 0.6719643647882164,
+      "learning_rate": 2.585455317986095e-06,
+      "loss": 0.0349,
+      "step": 881
+    },
+    {
+      "epoch": 1.3223388305847077,
+      "grad_norm": 0.5480134623127872,
+      "learning_rate": 2.5751509277790487e-06,
+      "loss": 0.0241,
+      "step": 882
+    },
+    {
+      "epoch": 1.3238380809595203,
+      "grad_norm": 0.49059448891569707,
+      "learning_rate": 2.5648599860333122e-06,
+      "loss": 0.0175,
+      "step": 883
+    },
+    {
+      "epoch": 1.3253373313343328,
+      "grad_norm": 0.5591155703517027,
+      "learning_rate": 2.554582549823502e-06,
+      "loss": 0.021,
+      "step": 884
+    },
+    {
+      "epoch": 1.3268365817091454,
+      "grad_norm": 0.9032101054084104,
+      "learning_rate": 2.5443186761493327e-06,
+      "loss": 0.0416,
+      "step": 885
+    },
+    {
+      "epoch": 1.328335832083958,
+      "grad_norm": 0.5476734241848452,
+      "learning_rate": 2.5340684219352977e-06,
+      "loss": 0.0208,
+      "step": 886
+    },
+    {
+      "epoch": 1.3298350824587706,
+      "grad_norm": 0.7277406175411532,
+      "learning_rate": 2.523831844030358e-06,
+      "loss": 0.0273,
+      "step": 887
+    },
+    {
+      "epoch": 1.3313343328335832,
+      "grad_norm": 0.5857920623257898,
+      "learning_rate": 2.513608999207622e-06,
+      "loss": 0.0251,
+      "step": 888
+    },
+    {
+      "epoch": 1.3328335832083957,
+      "grad_norm": 0.6938139296214967,
+      "learning_rate": 2.503399944164035e-06,
+      "loss": 0.0277,
+      "step": 889
+    },
+    {
+      "epoch": 1.3343328335832085,
+      "grad_norm": 0.6074733610539251,
+      "learning_rate": 2.4932047355200613e-06,
+      "loss": 0.0274,
+      "step": 890
+    },
+    {
+      "epoch": 1.3358320839580209,
+      "grad_norm": 0.8118901031338822,
+      "learning_rate": 2.483023429819372e-06,
+      "loss": 0.0338,
+      "step": 891
+    },
+    {
+      "epoch": 1.3373313343328337,
+      "grad_norm": 0.6281320267674445,
+      "learning_rate": 2.472856083528531e-06,
+      "loss": 0.0324,
+      "step": 892
+    },
+    {
+      "epoch": 1.338830584707646,
+      "grad_norm": 0.5270101200321455,
+      "learning_rate": 2.4627027530366836e-06,
+      "loss": 0.0196,
+      "step": 893
+    },
+    {
+      "epoch": 1.3403298350824588,
+      "grad_norm": 0.5062761253117802,
+      "learning_rate": 2.4525634946552405e-06,
+      "loss": 0.0329,
+      "step": 894
+    },
+    {
+      "epoch": 1.3418290854572714,
+      "grad_norm": 0.8042988652712693,
+      "learning_rate": 2.442438364617567e-06,
+      "loss": 0.036,
+      "step": 895
+    },
+    {
+      "epoch": 1.343328335832084,
+      "grad_norm": 0.7232552255256411,
+      "learning_rate": 2.4323274190786703e-06,
+      "loss": 0.0337,
+      "step": 896
+    },
+    {
+      "epoch": 1.3448275862068966,
+      "grad_norm": 0.6128414179239623,
+      "learning_rate": 2.422230714114891e-06,
+      "loss": 0.0261,
+      "step": 897
+    },
+    {
+      "epoch": 1.3463268365817092,
+      "grad_norm": 0.7908562061799264,
+      "learning_rate": 2.4121483057235884e-06,
+      "loss": 0.0428,
+      "step": 898
+    },
+    {
+      "epoch": 1.3478260869565217,
+      "grad_norm": 0.5620792177438926,
+      "learning_rate": 2.4020802498228333e-06,
+      "loss": 0.0204,
+      "step": 899
+    },
+    {
+      "epoch": 1.3493253373313343,
+      "grad_norm": 0.4408896888948946,
+      "learning_rate": 2.392026602251093e-06,
+      "loss": 0.0176,
+      "step": 900
+    },
+    {
+      "epoch": 1.3508245877061469,
+      "grad_norm": 0.8120237639020174,
+      "learning_rate": 2.3819874187669266e-06,
+      "loss": 0.0366,
+      "step": 901
+    },
+    {
+      "epoch": 1.3523238380809595,
+      "grad_norm": 0.6756222683848588,
+      "learning_rate": 2.371962755048675e-06,
+      "loss": 0.0333,
+      "step": 902
+    },
+    {
+      "epoch": 1.353823088455772,
+      "grad_norm": 0.6336967198398308,
+      "learning_rate": 2.36195266669415e-06,
+      "loss": 0.0322,
+      "step": 903
+    },
+    {
+      "epoch": 1.3553223388305846,
+      "grad_norm": 0.5675344157468191,
+      "learning_rate": 2.351957209220326e-06,
+      "loss": 0.0228,
+      "step": 904
+    },
+    {
+      "epoch": 1.3568215892053974,
+      "grad_norm": 0.6729107213966883,
+      "learning_rate": 2.341976438063035e-06,
+      "loss": 0.0284,
+      "step": 905
+    },
+    {
+      "epoch": 1.3583208395802098,
+      "grad_norm": 0.6396819588675353,
+      "learning_rate": 2.332010408576653e-06,
+      "loss": 0.0278,
+      "step": 906
+    },
+    {
+      "epoch": 1.3598200899550226,
+      "grad_norm": 0.6079749793405103,
+      "learning_rate": 2.3220591760338046e-06,
+      "loss": 0.0223,
+      "step": 907
+    },
+    {
+      "epoch": 1.3613193403298351,
+      "grad_norm": 0.6503832366211555,
+      "learning_rate": 2.3121227956250435e-06,
+      "loss": 0.0279,
+      "step": 908
+    },
+    {
+      "epoch": 1.3628185907046477,
+      "grad_norm": 0.6160895503930218,
+      "learning_rate": 2.302201322458552e-06,
+      "loss": 0.023,
+      "step": 909
+    },
+    {
+      "epoch": 1.3643178410794603,
+      "grad_norm": 0.4290766865581791,
+      "learning_rate": 2.292294811559837e-06,
+      "loss": 0.0161,
+      "step": 910
+    },
+    {
+      "epoch": 1.3658170914542729,
+      "grad_norm": 0.6563496576472319,
+      "learning_rate": 2.282403317871422e-06,
+      "loss": 0.0365,
+      "step": 911
+    },
+    {
+      "epoch": 1.3673163418290855,
+      "grad_norm": 0.6935523981062507,
+      "learning_rate": 2.2725268962525454e-06,
+      "loss": 0.0364,
+      "step": 912
+    },
+    {
+      "epoch": 1.368815592203898,
+      "grad_norm": 0.5942134363256111,
+      "learning_rate": 2.262665601478852e-06,
+      "loss": 0.0298,
+      "step": 913
+    },
+    {
+      "epoch": 1.3703148425787106,
+      "grad_norm": 0.5197364638220527,
+      "learning_rate": 2.252819488242093e-06,
+      "loss": 0.0234,
+      "step": 914
+    },
+    {
+      "epoch": 1.3718140929535232,
+      "grad_norm": 0.5685973871646581,
+      "learning_rate": 2.24298861114982e-06,
+      "loss": 0.0194,
+      "step": 915
+    },
+    {
+      "epoch": 1.3733133433283358,
+      "grad_norm": 0.5457205011537143,
+      "learning_rate": 2.2331730247250857e-06,
+      "loss": 0.0239,
+      "step": 916
+    },
+    {
+      "epoch": 1.3748125937031483,
+      "grad_norm": 0.665478396710653,
+      "learning_rate": 2.223372783406137e-06,
+      "loss": 0.0261,
+      "step": 917
+    },
+    {
+      "epoch": 1.3763118440779611,
+      "grad_norm": 0.5679854646745142,
+      "learning_rate": 2.2135879415461152e-06,
+      "loss": 0.0249,
+      "step": 918
+    },
+    {
+      "epoch": 1.3778110944527735,
+      "grad_norm": 0.7878314269859837,
+      "learning_rate": 2.203818553412757e-06,
+      "loss": 0.0301,
+      "step": 919
+    },
+    {
+      "epoch": 1.3793103448275863,
+      "grad_norm": 0.5218012327141217,
+      "learning_rate": 2.1940646731880887e-06,
+      "loss": 0.0231,
+      "step": 920
+    },
+    {
+      "epoch": 1.3808095952023989,
+      "grad_norm": 0.534420649310201,
+      "learning_rate": 2.1843263549681287e-06,
+      "loss": 0.0216,
+      "step": 921
+    },
+    {
+      "epoch": 1.3823088455772115,
+      "grad_norm": 0.4769151383477972,
+      "learning_rate": 2.174603652762588e-06,
+      "loss": 0.0225,
+      "step": 922
+    },
+    {
+      "epoch": 1.383808095952024,
+      "grad_norm": 0.7752218719157404,
+      "learning_rate": 2.164896620494569e-06,
+      "loss": 0.041,
+      "step": 923
+    },
+    {
+      "epoch": 1.3853073463268366,
+      "grad_norm": 0.6264041671698231,
+      "learning_rate": 2.1552053120002655e-06,
+      "loss": 0.0236,
+      "step": 924
+    },
+    {
+      "epoch": 1.3868065967016492,
+      "grad_norm": 0.6174352167870553,
+      "learning_rate": 2.145529781028668e-06,
+      "loss": 0.0273,
+      "step": 925
+    },
+    {
+      "epoch": 1.3883058470764618,
+      "grad_norm": 0.4991658071473942,
+      "learning_rate": 2.1358700812412625e-06,
+      "loss": 0.0228,
+      "step": 926
+    },
+    {
+      "epoch": 1.3898050974512743,
+      "grad_norm": 0.5248469289781732,
+      "learning_rate": 2.1262262662117327e-06,
+      "loss": 0.0211,
+      "step": 927
+    },
+    {
+      "epoch": 1.391304347826087,
+      "grad_norm": 0.7342596906495891,
+      "learning_rate": 2.1165983894256647e-06,
+      "loss": 0.0284,
+      "step": 928
+    },
+    {
+      "epoch": 1.3928035982008995,
+      "grad_norm": 0.6664655449189273,
+      "learning_rate": 2.1069865042802502e-06,
+      "loss": 0.0321,
+      "step": 929
+    },
+    {
+      "epoch": 1.394302848575712,
+      "grad_norm": 0.7028687860304457,
+      "learning_rate": 2.0973906640839867e-06,
+      "loss": 0.0266,
+      "step": 930
+    },
+    {
+      "epoch": 1.3958020989505249,
+      "grad_norm": 0.44031840128598276,
+      "learning_rate": 2.0878109220563884e-06,
+      "loss": 0.0202,
+      "step": 931
+    },
+    {
+      "epoch": 1.3973013493253372,
+      "grad_norm": 0.7319043265784508,
+      "learning_rate": 2.078247331327685e-06,
+      "loss": 0.0267,
+      "step": 932
+    },
+    {
+      "epoch": 1.39880059970015,
+      "grad_norm": 0.5321544886719682,
+      "learning_rate": 2.0686999449385286e-06,
+      "loss": 0.0212,
+      "step": 933
+    },
+    {
+      "epoch": 1.4002998500749624,
+      "grad_norm": 0.4632383372797399,
+      "learning_rate": 2.0591688158397054e-06,
+      "loss": 0.0216,
+      "step": 934
+    },
+    {
+      "epoch": 1.4017991004497752,
+      "grad_norm": 0.507317499307577,
+      "learning_rate": 2.0496539968918342e-06,
+      "loss": 0.0155,
+      "step": 935
+    },
+    {
+      "epoch": 1.4032983508245878,
+      "grad_norm": 0.4255578248744823,
+      "learning_rate": 2.0401555408650714e-06,
+      "loss": 0.0178,
+      "step": 936
+    },
+    {
+      "epoch": 1.4047976011994003,
+      "grad_norm": 0.5277470245764748,
+      "learning_rate": 2.030673500438828e-06,
+      "loss": 0.0176,
+      "step": 937
+    },
+    {
+      "epoch": 1.406296851574213,
+      "grad_norm": 0.5213200277597092,
+      "learning_rate": 2.0212079282014725e-06,
+      "loss": 0.0222,
+      "step": 938
+    },
+    {
+      "epoch": 1.4077961019490255,
+      "grad_norm": 0.5473626872276902,
+      "learning_rate": 2.0117588766500375e-06,
+      "loss": 0.0262,
+      "step": 939
+    },
+    {
+      "epoch": 1.409295352323838,
+      "grad_norm": 0.7002348585230302,
+      "learning_rate": 2.002326398189931e-06,
+      "loss": 0.038,
+      "step": 940
+    },
+    {
+      "epoch": 1.4107946026986506,
+      "grad_norm": 0.5667417252103428,
+      "learning_rate": 1.9929105451346436e-06,
+      "loss": 0.0197,
+      "step": 941
+    },
+    {
+      "epoch": 1.4122938530734632,
+      "grad_norm": 0.4652306698389131,
+      "learning_rate": 1.983511369705462e-06,
+      "loss": 0.0168,
+      "step": 942
+    },
+    {
+      "epoch": 1.4137931034482758,
+      "grad_norm": 0.5016915896415785,
+      "learning_rate": 1.9741289240311757e-06,
+      "loss": 0.0149,
+      "step": 943
+    },
+    {
+      "epoch": 1.4152923538230884,
+      "grad_norm": 0.5928191131365009,
+      "learning_rate": 1.9647632601477877e-06,
+      "loss": 0.0234,
+      "step": 944
+    },
+    {
+      "epoch": 1.416791604197901,
+      "grad_norm": 1.057104882412813,
+      "learning_rate": 1.9554144299982314e-06,
+      "loss": 0.0401,
+      "step": 945
+    },
+    {
+      "epoch": 1.4182908545727138,
+      "grad_norm": 0.6331928853753662,
+      "learning_rate": 1.9460824854320755e-06,
+      "loss": 0.0255,
+      "step": 946
+    },
+    {
+      "epoch": 1.419790104947526,
+      "grad_norm": 0.6569165027816367,
+      "learning_rate": 1.9367674782052376e-06,
+      "loss": 0.0332,
+      "step": 947
+    },
+    {
+      "epoch": 1.421289355322339,
+      "grad_norm": 0.6305158628775781,
+      "learning_rate": 1.9274694599797067e-06,
+      "loss": 0.0275,
+      "step": 948
+    },
+    {
+      "epoch": 1.4227886056971515,
+      "grad_norm": 0.5825690998425075,
+      "learning_rate": 1.918188482323242e-06,
+      "loss": 0.0262,
+      "step": 949
+    },
+    {
+      "epoch": 1.424287856071964,
+      "grad_norm": 0.7283554829049703,
+      "learning_rate": 1.9089245967090952e-06,
+      "loss": 0.0286,
+      "step": 950
+    },
+    {
+      "epoch": 1.4257871064467766,
+      "grad_norm": 0.6755328411941485,
+      "learning_rate": 1.8996778545157263e-06,
+      "loss": 0.0262,
+      "step": 951
+    },
+    {
+      "epoch": 1.4272863568215892,
+      "grad_norm": 0.5754676460056062,
+      "learning_rate": 1.8904483070265133e-06,
+      "loss": 0.0285,
+      "step": 952
+    },
+    {
+      "epoch": 1.4287856071964018,
+      "grad_norm": 0.7774073548002273,
+      "learning_rate": 1.8812360054294725e-06,
+      "loss": 0.0321,
+      "step": 953
+    },
+    {
+      "epoch": 1.4302848575712144,
+      "grad_norm": 0.6966367579564179,
+      "learning_rate": 1.8720410008169727e-06,
+      "loss": 0.0331,
+      "step": 954
+    },
+    {
+      "epoch": 1.431784107946027,
+      "grad_norm": 0.7439849235326366,
+      "learning_rate": 1.8628633441854515e-06,
+      "loss": 0.043,
+      "step": 955
+    },
+    {
+      "epoch": 1.4332833583208395,
+      "grad_norm": 0.5169888756826231,
+      "learning_rate": 1.8537030864351303e-06,
+      "loss": 0.0282,
+      "step": 956
+    },
+    {
+      "epoch": 1.434782608695652,
+      "grad_norm": 0.7118356198480503,
+      "learning_rate": 1.8445602783697375e-06,
+      "loss": 0.0288,
+      "step": 957
+    },
+    {
+      "epoch": 1.4362818590704647,
+      "grad_norm": 0.7123485807397486,
+      "learning_rate": 1.8354349706962243e-06,
+      "loss": 0.0328,
+      "step": 958
+    },
+    {
+      "epoch": 1.4377811094452775,
+      "grad_norm": 0.44365067038580586,
+      "learning_rate": 1.8263272140244803e-06,
+      "loss": 0.0174,
+      "step": 959
+    },
+    {
+      "epoch": 1.4392803598200898,
+      "grad_norm": 0.5373739844421722,
+      "learning_rate": 1.8172370588670563e-06,
+      "loss": 0.0174,
+      "step": 960
+    },
+    {
+      "epoch": 1.4407796101949026,
+      "grad_norm": 0.5829369570233167,
+      "learning_rate": 1.8081645556388866e-06,
+      "loss": 0.0289,
+      "step": 961
+    },
+    {
+      "epoch": 1.4422788605697152,
+      "grad_norm": 0.428062178513796,
+      "learning_rate": 1.7991097546570018e-06,
+      "loss": 0.015,
+      "step": 962
+    },
+    {
+      "epoch": 1.4437781109445278,
+      "grad_norm": 0.6311967260732321,
+      "learning_rate": 1.7900727061402556e-06,
+      "loss": 0.0333,
+      "step": 963
+    },
+    {
+      "epoch": 1.4452773613193404,
+      "grad_norm": 0.4922316235849401,
+      "learning_rate": 1.7810534602090445e-06,
+      "loss": 0.0155,
+      "step": 964
+    },
+    {
+      "epoch": 1.446776611694153,
+      "grad_norm": 0.7282727373456914,
+      "learning_rate": 1.77205206688503e-06,
+      "loss": 0.0354,
+      "step": 965
+    },
+    {
+      "epoch": 1.4482758620689655,
+      "grad_norm": 0.6474567594894216,
+      "learning_rate": 1.7630685760908623e-06,
+      "loss": 0.0375,
+      "step": 966
+    },
+    {
+      "epoch": 1.449775112443778,
+      "grad_norm": 0.48357928666893746,
+      "learning_rate": 1.7541030376499002e-06,
+      "loss": 0.0203,
+      "step": 967
+    },
+    {
+      "epoch": 1.4512743628185907,
+      "grad_norm": 0.6071318474872953,
+      "learning_rate": 1.745155501285939e-06,
+      "loss": 0.0248,
+      "step": 968
+    },
+    {
+      "epoch": 1.4527736131934033,
+      "grad_norm": 0.6275550585585258,
+      "learning_rate": 1.736226016622931e-06,
+      "loss": 0.0294,
+      "step": 969
+    },
+    {
+      "epoch": 1.4542728635682158,
+      "grad_norm": 0.7167921221438877,
+      "learning_rate": 1.727314633184714e-06,
+      "loss": 0.0294,
+      "step": 970
+    },
+    {
+      "epoch": 1.4557721139430284,
+      "grad_norm": 0.5814235023924477,
+      "learning_rate": 1.718421400394732e-06,
+      "loss": 0.0237,
+      "step": 971
+    },
+    {
+      "epoch": 1.4572713643178412,
+      "grad_norm": 0.6220841265766438,
+      "learning_rate": 1.7095463675757656e-06,
+      "loss": 0.0273,
+      "step": 972
+    },
+    {
+      "epoch": 1.4587706146926536,
+      "grad_norm": 0.5938405193042682,
+      "learning_rate": 1.7006895839496557e-06,
+      "loss": 0.0332,
+      "step": 973
+    },
+    {
+      "epoch": 1.4602698650674664,
+      "grad_norm": 0.5084457781199155,
+      "learning_rate": 1.6918510986370312e-06,
+      "loss": 0.0203,
+      "step": 974
+    },
+    {
+      "epoch": 1.461769115442279,
+      "grad_norm": 0.5882691616001134,
+      "learning_rate": 1.6830309606570372e-06,
+      "loss": 0.0275,
+      "step": 975
+    },
+    {
+      "epoch": 1.4632683658170915,
+      "grad_norm": 0.4724227172235087,
+      "learning_rate": 1.674229218927062e-06,
+      "loss": 0.0175,
+      "step": 976
+    },
+    {
+      "epoch": 1.464767616191904,
+      "grad_norm": 0.7541912271923925,
+      "learning_rate": 1.665445922262467e-06,
+      "loss": 0.0393,
+      "step": 977
+    },
+    {
+      "epoch": 1.4662668665667167,
+      "grad_norm": 0.7834507238807119,
+      "learning_rate": 1.6566811193763149e-06,
+      "loss": 0.0272,
+      "step": 978
+    },
+    {
+      "epoch": 1.4677661169415293,
+      "grad_norm": 0.7846338504751385,
+      "learning_rate": 1.6479348588791e-06,
+      "loss": 0.0361,
+      "step": 979
+    },
+    {
+      "epoch": 1.4692653673163418,
+      "grad_norm": 0.6956447175018466,
+      "learning_rate": 1.6392071892784789e-06,
+      "loss": 0.0309,
+      "step": 980
+    },
+    {
+      "epoch": 1.4707646176911544,
+      "grad_norm": 0.5193225073879243,
+      "learning_rate": 1.6304981589790015e-06,
+      "loss": 0.0233,
+      "step": 981
+    },
+    {
+      "epoch": 1.472263868065967,
+      "grad_norm": 0.7131091449029804,
+      "learning_rate": 1.6218078162818418e-06,
+      "loss": 0.0276,
+      "step": 982
+    },
+    {
+      "epoch": 1.4737631184407796,
+      "grad_norm": 0.7414766746493929,
+      "learning_rate": 1.6131362093845299e-06,
+      "loss": 0.0352,
+      "step": 983
+    },
+    {
+      "epoch": 1.4752623688155921,
+      "grad_norm": 0.6289779341874668,
+      "learning_rate": 1.6044833863806864e-06,
+      "loss": 0.0324,
+      "step": 984
+    },
+    {
+      "epoch": 1.4767616191904047,
+      "grad_norm": 0.5405410798259783,
+      "learning_rate": 1.5958493952597536e-06,
+      "loss": 0.0294,
+      "step": 985
+    },
+    {
+      "epoch": 1.4782608695652173,
+      "grad_norm": 0.6535884950698414,
+      "learning_rate": 1.5872342839067305e-06,
+      "loss": 0.0339,
+      "step": 986
+    },
+    {
+      "epoch": 1.47976011994003,
+      "grad_norm": 0.6121575580886064,
+      "learning_rate": 1.5786381001019052e-06,
+      "loss": 0.0247,
+      "step": 987
+    },
+    {
+      "epoch": 1.4812593703148424,
+      "grad_norm": 0.592625619020157,
+      "learning_rate": 1.5700608915205978e-06,
+      "loss": 0.027,
+      "step": 988
+    },
+    {
+      "epoch": 1.4827586206896552,
+      "grad_norm": 0.8834588521398952,
+      "learning_rate": 1.561502705732883e-06,
+      "loss": 0.0341,
+      "step": 989
+    },
+    {
+      "epoch": 1.4842578710644678,
+      "grad_norm": 0.5109224427382247,
+      "learning_rate": 1.5529635902033358e-06,
+      "loss": 0.0199,
+      "step": 990
+    },
+    {
+      "epoch": 1.4857571214392804,
+      "grad_norm": 0.8320696401550205,
+      "learning_rate": 1.5444435922907669e-06,
+      "loss": 0.0302,
+      "step": 991
+    },
+    {
+      "epoch": 1.487256371814093,
+      "grad_norm": 0.7855617286306584,
+      "learning_rate": 1.5359427592479553e-06,
+      "loss": 0.0364,
+      "step": 992
+    },
+    {
+      "epoch": 1.4887556221889056,
+      "grad_norm": 0.4880380914129996,
+      "learning_rate": 1.5274611382213922e-06,
+      "loss": 0.0203,
+      "step": 993
+    },
+    {
+      "epoch": 1.4902548725637181,
+      "grad_norm": 0.7473799043588297,
+      "learning_rate": 1.5189987762510167e-06,
+      "loss": 0.0286,
+      "step": 994
+    },
+    {
+      "epoch": 1.4917541229385307,
+      "grad_norm": 0.6288012310640947,
+      "learning_rate": 1.510555720269955e-06,
+      "loss": 0.0236,
+      "step": 995
+    },
+    {
+      "epoch": 1.4932533733133433,
+      "grad_norm": 0.590016461826813,
+      "learning_rate": 1.5021320171042608e-06,
+      "loss": 0.0259,
+      "step": 996
+    },
+    {
+      "epoch": 1.4947526236881559,
+      "grad_norm": 0.859428484803135,
+      "learning_rate": 1.4937277134726542e-06,
+      "loss": 0.0414,
+      "step": 997
+    },
+    {
+      "epoch": 1.4962518740629684,
+      "grad_norm": 0.4977733278607171,
+      "learning_rate": 1.4853428559862637e-06,
+      "loss": 0.019,
+      "step": 998
+    },
+    {
+      "epoch": 1.497751124437781,
+      "grad_norm": 0.6649734653379189,
+      "learning_rate": 1.4769774911483686e-06,
+      "loss": 0.0384,
+      "step": 999
+    },
+    {
+      "epoch": 1.4992503748125938,
+      "grad_norm": 0.5702040704897402,
+      "learning_rate": 1.4686316653541377e-06,
+      "loss": 0.0167,
+      "step": 1000
+    },
+    {
+      "epoch": 1.4992503748125938,
+      "eval_loss": 0.10759080946445465,
+      "eval_runtime": 2.1857,
+      "eval_samples_per_second": 24.706,
+      "eval_steps_per_second": 6.405,
+      "step": 1000
+    },
+    {
+      "epoch": 1.5007496251874062,
+      "grad_norm": 0.8159083822915767,
+      "learning_rate": 1.4603054248903752e-06,
+      "loss": 0.0458,
+      "step": 1001
+    },
+    {
+      "epoch": 1.502248875562219,
+      "grad_norm": 0.5813507283096027,
+      "learning_rate": 1.4519988159352665e-06,
+      "loss": 0.0195,
+      "step": 1002
+    },
+    {
+      "epoch": 1.5037481259370313,
+      "grad_norm": 0.8290518394336848,
+      "learning_rate": 1.4437118845581138e-06,
+      "loss": 0.0449,
+      "step": 1003
+    },
+    {
+      "epoch": 1.5052473763118441,
+      "grad_norm": 0.6707314194208763,
+      "learning_rate": 1.4354446767190873e-06,
+      "loss": 0.0314,
+      "step": 1004
+    },
+    {
+      "epoch": 1.5067466266866567,
+      "grad_norm": 1.0493270985925962,
+      "learning_rate": 1.4271972382689685e-06,
+      "loss": 0.0377,
+      "step": 1005
+    },
+    {
+      "epoch": 1.5082458770614693,
+      "grad_norm": 0.649467449695993,
+      "learning_rate": 1.4189696149488956e-06,
+      "loss": 0.0289,
+      "step": 1006
+    },
+    {
+      "epoch": 1.5097451274362819,
+      "grad_norm": 0.49767436288228745,
+      "learning_rate": 1.4107618523901101e-06,
+      "loss": 0.0196,
+      "step": 1007
+    },
+    {
+      "epoch": 1.5112443778110944,
+      "grad_norm": 0.5352987873275364,
+      "learning_rate": 1.4025739961137043e-06,
+      "loss": 0.0245,
+      "step": 1008
+    },
+    {
+      "epoch": 1.512743628185907,
+      "grad_norm": 0.5231699733876537,
+      "learning_rate": 1.394406091530367e-06,
+      "loss": 0.0183,
+      "step": 1009
+    },
+    {
+      "epoch": 1.5142428785607196,
+      "grad_norm": 0.6161373434179515,
+      "learning_rate": 1.3862581839401346e-06,
+      "loss": 0.0257,
+      "step": 1010
+    },
+    {
+      "epoch": 1.5157421289355324,
+      "grad_norm": 0.7498691818631599,
+      "learning_rate": 1.3781303185321377e-06,
+      "loss": 0.0272,
+      "step": 1011
+    },
+    {
+      "epoch": 1.5172413793103448,
+      "grad_norm": 0.6017438437045454,
+      "learning_rate": 1.370022540384347e-06,
+      "loss": 0.0202,
+      "step": 1012
+    },
+    {
+      "epoch": 1.5187406296851576,
+      "grad_norm": 0.5111789393629014,
+      "learning_rate": 1.3619348944633331e-06,
+      "loss": 0.0241,
+      "step": 1013
+    },
+    {
+      "epoch": 1.52023988005997,
+      "grad_norm": 0.619065447493321,
+      "learning_rate": 1.3538674256240087e-06,
+      "loss": 0.0278,
+      "step": 1014
+    },
+    {
+      "epoch": 1.5217391304347827,
+      "grad_norm": 0.46161830048563396,
+      "learning_rate": 1.3458201786093795e-06,
+      "loss": 0.0155,
+      "step": 1015
+    },
+    {
+      "epoch": 1.523238380809595,
+      "grad_norm": 1.0496166864739152,
+      "learning_rate": 1.3377931980503055e-06,
+      "loss": 0.0315,
+      "step": 1016
+    },
+    {
+      "epoch": 1.5247376311844079,
+      "grad_norm": 0.4433156266784478,
+      "learning_rate": 1.3297865284652417e-06,
+      "loss": 0.0179,
+      "step": 1017
+    },
+    {
+      "epoch": 1.5262368815592204,
+      "grad_norm": 0.6201000140827553,
+      "learning_rate": 1.3218002142599973e-06,
+      "loss": 0.0319,
+      "step": 1018
+    },
+    {
+      "epoch": 1.527736131934033,
+      "grad_norm": 0.5350897101980913,
+      "learning_rate": 1.3138342997274883e-06,
+      "loss": 0.0238,
+      "step": 1019
+    },
+    {
+      "epoch": 1.5292353823088456,
+      "grad_norm": 0.5766817255549758,
+      "learning_rate": 1.3058888290474937e-06,
+      "loss": 0.0266,
+      "step": 1020
+    },
+    {
+      "epoch": 1.5307346326836582,
+      "grad_norm": 0.6785549481843606,
+      "learning_rate": 1.2979638462864069e-06,
+      "loss": 0.0263,
+      "step": 1021
+    },
+    {
+      "epoch": 1.5322338830584707,
+      "grad_norm": 0.6066629719862847,
+      "learning_rate": 1.2900593953969947e-06,
+      "loss": 0.0198,
+      "step": 1022
+    },
+    {
+      "epoch": 1.5337331334332833,
+      "grad_norm": 0.9401837843231554,
+      "learning_rate": 1.2821755202181503e-06,
+      "loss": 0.0321,
+      "step": 1023
+    },
+    {
+      "epoch": 1.535232383808096,
+      "grad_norm": 0.5424154145312716,
+      "learning_rate": 1.2743122644746536e-06,
+      "loss": 0.0174,
+      "step": 1024
+    },
+    {
+      "epoch": 1.5367316341829085,
+      "grad_norm": 0.7427898710493821,
+      "learning_rate": 1.266469671776926e-06,
+      "loss": 0.0387,
+      "step": 1025
+    },
+    {
+      "epoch": 1.5382308845577213,
+      "grad_norm": 0.5921183291604994,
+      "learning_rate": 1.2586477856207902e-06,
+      "loss": 0.0246,
+      "step": 1026
+    },
+    {
+      "epoch": 1.5397301349325336,
+      "grad_norm": 0.5406203427392713,
+      "learning_rate": 1.2508466493872273e-06,
+      "loss": 0.0223,
+      "step": 1027
+    },
+    {
+      "epoch": 1.5412293853073464,
+      "grad_norm": 1.78672861624451,
+      "learning_rate": 1.2430663063421388e-06,
+      "loss": 0.0422,
+      "step": 1028
+    },
+    {
+      "epoch": 1.5427286356821588,
+      "grad_norm": 0.701007243564336,
+      "learning_rate": 1.2353067996361034e-06,
+      "loss": 0.034,
+      "step": 1029
+    },
+    {
+      "epoch": 1.5442278860569716,
+      "grad_norm": 0.8302277439378078,
+      "learning_rate": 1.2275681723041406e-06,
+      "loss": 0.0376,
+      "step": 1030
+    },
+    {
+      "epoch": 1.545727136431784,
+      "grad_norm": 0.6628848240740699,
+      "learning_rate": 1.2198504672654694e-06,
+      "loss": 0.0243,
+      "step": 1031
+    },
+    {
+      "epoch": 1.5472263868065967,
+      "grad_norm": 0.553681031297978,
+      "learning_rate": 1.212153727323273e-06,
+      "loss": 0.022,
+      "step": 1032
+    },
+    {
+      "epoch": 1.5487256371814093,
+      "grad_norm": 0.5716918940454803,
+      "learning_rate": 1.2044779951644586e-06,
+      "loss": 0.0303,
+      "step": 1033
+    },
+    {
+      "epoch": 1.550224887556222,
+      "grad_norm": 0.6360126033680119,
+      "learning_rate": 1.1968233133594243e-06,
+      "loss": 0.0272,
+      "step": 1034
+    },
+    {
+      "epoch": 1.5517241379310345,
+      "grad_norm": 0.49613470291877665,
+      "learning_rate": 1.1891897243618184e-06,
+      "loss": 0.0194,
+      "step": 1035
+    },
+    {
+      "epoch": 1.553223388305847,
+      "grad_norm": 0.5057143375807001,
+      "learning_rate": 1.1815772705083072e-06,
+      "loss": 0.0209,
+      "step": 1036
+    },
+    {
+      "epoch": 1.5547226386806596,
+      "grad_norm": 0.5674540678677971,
+      "learning_rate": 1.17398599401834e-06,
+      "loss": 0.0209,
+      "step": 1037
+    },
+    {
+      "epoch": 1.5562218890554722,
+      "grad_norm": 0.7449689732137141,
+      "learning_rate": 1.1664159369939137e-06,
+      "loss": 0.032,
+      "step": 1038
+    },
+    {
+      "epoch": 1.557721139430285,
+      "grad_norm": 0.709645902889105,
+      "learning_rate": 1.1588671414193397e-06,
+      "loss": 0.0329,
+      "step": 1039
+    },
+    {
+      "epoch": 1.5592203898050974,
+      "grad_norm": 0.5168378926515985,
+      "learning_rate": 1.1513396491610113e-06,
+      "loss": 0.0259,
+      "step": 1040
+    },
+    {
+      "epoch": 1.5607196401799102,
+      "grad_norm": 0.5722586536457223,
+      "learning_rate": 1.1438335019671715e-06,
+      "loss": 0.0203,
+      "step": 1041
+    },
+    {
+      "epoch": 1.5622188905547225,
+      "grad_norm": 0.8088843299068137,
+      "learning_rate": 1.1363487414676805e-06,
+      "loss": 0.0333,
+      "step": 1042
+    },
+    {
+      "epoch": 1.5637181409295353,
+      "grad_norm": 0.6371578861329852,
+      "learning_rate": 1.128885409173789e-06,
+      "loss": 0.0266,
+      "step": 1043
+    },
+    {
+      "epoch": 1.5652173913043477,
+      "grad_norm": 0.5833861970480987,
+      "learning_rate": 1.1214435464779006e-06,
+      "loss": 0.0231,
+      "step": 1044
+    },
+    {
+      "epoch": 1.5667166416791605,
+      "grad_norm": 0.5131253805020719,
+      "learning_rate": 1.1140231946533486e-06,
+      "loss": 0.0206,
+      "step": 1045
+    },
+    {
+      "epoch": 1.568215892053973,
+      "grad_norm": 1.2418569712259722,
+      "learning_rate": 1.1066243948541638e-06,
+      "loss": 0.0323,
+      "step": 1046
+    },
+    {
+      "epoch": 1.5697151424287856,
+      "grad_norm": 0.46290616099325926,
+      "learning_rate": 1.0992471881148497e-06,
+      "loss": 0.0163,
+      "step": 1047
+    },
+    {
+      "epoch": 1.5712143928035982,
+      "grad_norm": 0.6775692308562952,
+      "learning_rate": 1.091891615350147e-06,
+      "loss": 0.0255,
+      "step": 1048
+    },
+    {
+      "epoch": 1.5727136431784108,
+      "grad_norm": 0.7995673291471764,
+      "learning_rate": 1.0845577173548172e-06,
+      "loss": 0.0397,
+      "step": 1049
+    },
+    {
+      "epoch": 1.5742128935532234,
+      "grad_norm": 0.5549445402032995,
+      "learning_rate": 1.07724553480341e-06,
+      "loss": 0.0201,
+      "step": 1050
+    },
+    {
+      "epoch": 1.575712143928036,
+      "grad_norm": 0.7096047337383095,
+      "learning_rate": 1.0699551082500387e-06,
+      "loss": 0.0241,
+      "step": 1051
+    },
+    {
+      "epoch": 1.5772113943028487,
+      "grad_norm": 0.6236759720906369,
+      "learning_rate": 1.0626864781281553e-06,
+      "loss": 0.0239,
+      "step": 1052
+    },
+    {
+      "epoch": 1.578710644677661,
+      "grad_norm": 0.4989579095877573,
+      "learning_rate": 1.0554396847503272e-06,
+      "loss": 0.0172,
+      "step": 1053
+    },
+    {
+      "epoch": 1.580209895052474,
+      "grad_norm": 0.5196594677960036,
+      "learning_rate": 1.0482147683080125e-06,
+      "loss": 0.0189,
+      "step": 1054
+    },
+    {
+      "epoch": 1.5817091454272862,
+      "grad_norm": 0.5765106550947825,
+      "learning_rate": 1.0410117688713366e-06,
+      "loss": 0.0248,
+      "step": 1055
+    },
+    {
+      "epoch": 1.583208395802099,
+      "grad_norm": 0.6524523626352632,
+      "learning_rate": 1.0338307263888748e-06,
+      "loss": 0.0234,
+      "step": 1056
+    },
+    {
+      "epoch": 1.5847076461769114,
+      "grad_norm": 0.5773920025453051,
+      "learning_rate": 1.0266716806874227e-06,
+      "loss": 0.0262,
+      "step": 1057
+    },
+    {
+      "epoch": 1.5862068965517242,
+      "grad_norm": 0.4533208407898587,
+      "learning_rate": 1.0195346714717813e-06,
+      "loss": 0.0217,
+      "step": 1058
+    },
+    {
+      "epoch": 1.5877061469265368,
+      "grad_norm": 0.7174984026219365,
+      "learning_rate": 1.0124197383245344e-06,
+      "loss": 0.0231,
+      "step": 1059
+    },
+    {
+      "epoch": 1.5892053973013494,
+      "grad_norm": 0.49171914769171754,
+      "learning_rate": 1.0053269207058298e-06,
+      "loss": 0.0206,
+      "step": 1060
+    },
+    {
+      "epoch": 1.590704647676162,
+      "grad_norm": 0.5138153338109913,
+      "learning_rate": 9.982562579531607e-07,
+      "loss": 0.0203,
+      "step": 1061
+    },
+    {
+      "epoch": 1.5922038980509745,
+      "grad_norm": 0.4612531169814587,
+      "learning_rate": 9.912077892811473e-07,
+      "loss": 0.0161,
+      "step": 1062
+    },
+    {
+      "epoch": 1.593703148425787,
+      "grad_norm": 0.578811698024907,
+      "learning_rate": 9.841815537813177e-07,
+      "loss": 0.0243,
+      "step": 1063
+    },
+    {
+      "epoch": 1.5952023988005997,
+      "grad_norm": 0.6580298632652755,
+      "learning_rate": 9.77177590421895e-07,
+      "loss": 0.0263,
+      "step": 1064
+    },
+    {
+      "epoch": 1.5967016491754122,
+      "grad_norm": 0.7174306682988506,
+      "learning_rate": 9.70195938047576e-07,
+      "loss": 0.0292,
+      "step": 1065
+    },
+    {
+      "epoch": 1.5982008995502248,
+      "grad_norm": 0.5893819921973688,
+      "learning_rate": 9.63236635379321e-07,
+      "loss": 0.0198,
+      "step": 1066
+    },
+    {
+      "epoch": 1.5997001499250376,
+      "grad_norm": 0.6421467270341864,
+      "learning_rate": 9.562997210141355e-07,
+      "loss": 0.0317,
+      "step": 1067
+    },
+    {
+      "epoch": 1.60119940029985,
+      "grad_norm": 0.6574534499308227,
+      "learning_rate": 9.49385233424856e-07,
+      "loss": 0.0309,
+      "step": 1068
+    },
+    {
+      "epoch": 1.6026986506746628,
+      "grad_norm": 0.6080218452052112,
+      "learning_rate": 9.424932109599372e-07,
+      "loss": 0.0264,
+      "step": 1069
+    },
+    {
+      "epoch": 1.6041979010494751,
+      "grad_norm": 0.58306051653843,
+      "learning_rate": 9.356236918432454e-07,
+      "loss": 0.0266,
+      "step": 1070
+    },
+    {
+      "epoch": 1.605697151424288,
+      "grad_norm": 0.7615996089937827,
+      "learning_rate": 9.287767141738352e-07,
+      "loss": 0.03,
+      "step": 1071
+    },
+    {
+      "epoch": 1.6071964017991005,
+      "grad_norm": 0.5973420038613207,
+      "learning_rate": 9.21952315925746e-07,
+      "loss": 0.0282,
+      "step": 1072
+    },
+    {
+      "epoch": 1.608695652173913,
+      "grad_norm": 0.4965452792272044,
+      "learning_rate": 9.151505349477901e-07,
+      "loss": 0.0243,
+      "step": 1073
+    },
+    {
+      "epoch": 1.6101949025487257,
+      "grad_norm": 0.9359302956967643,
+      "learning_rate": 9.08371408963341e-07,
+      "loss": 0.0454,
+      "step": 1074
+    },
+    {
+      "epoch": 1.6116941529235382,
+      "grad_norm": 0.7990675277791618,
+      "learning_rate": 9.016149755701259e-07,
+      "loss": 0.0394,
+      "step": 1075
+    },
+    {
+      "epoch": 1.6131934032983508,
+      "grad_norm": 0.6173484934040682,
+      "learning_rate": 8.948812722400157e-07,
+      "loss": 0.0265,
+      "step": 1076
+    },
+    {
+      "epoch": 1.6146926536731634,
+      "grad_norm": 0.8557356812832911,
+      "learning_rate": 8.881703363188199e-07,
+      "loss": 0.0477,
+      "step": 1077
+    },
+    {
+      "epoch": 1.616191904047976,
+      "grad_norm": 0.43048615417074343,
+      "learning_rate": 8.814822050260758e-07,
+      "loss": 0.0168,
+      "step": 1078
+    },
+    {
+      "epoch": 1.6176911544227885,
+      "grad_norm": 0.6057071442636979,
+      "learning_rate": 8.748169154548448e-07,
+      "loss": 0.0301,
+      "step": 1079
+    },
+    {
+      "epoch": 1.6191904047976013,
+      "grad_norm": 0.5362403007688328,
+      "learning_rate": 8.681745045715045e-07,
+      "loss": 0.0301,
+      "step": 1080
+    },
+    {
+      "epoch": 1.6206896551724137,
+      "grad_norm": 0.4842364522855512,
+      "learning_rate": 8.615550092155478e-07,
+      "loss": 0.0261,
+      "step": 1081
+    },
+    {
+      "epoch": 1.6221889055472265,
+      "grad_norm": 0.5486019862177269,
+      "learning_rate": 8.549584660993726e-07,
+      "loss": 0.0227,
+      "step": 1082
+    },
+    {
+      "epoch": 1.6236881559220389,
+      "grad_norm": 0.6495217079413514,
+      "learning_rate": 8.483849118080828e-07,
+      "loss": 0.0252,
+      "step": 1083
+    },
+    {
+      "epoch": 1.6251874062968517,
+      "grad_norm": 0.5843494174258468,
+      "learning_rate": 8.418343827992842e-07,
+      "loss": 0.0265,
+      "step": 1084
+    },
+    {
+      "epoch": 1.626686656671664,
+      "grad_norm": 0.7346266327663906,
+      "learning_rate": 8.353069154028814e-07,
+      "loss": 0.0368,
+      "step": 1085
+    },
+    {
+      "epoch": 1.6281859070464768,
+      "grad_norm": 0.5966854157767114,
+      "learning_rate": 8.28802545820877e-07,
+      "loss": 0.035,
+      "step": 1086
+    },
+    {
+      "epoch": 1.6296851574212894,
+      "grad_norm": 0.7719200043978971,
+      "learning_rate": 8.223213101271709e-07,
+      "loss": 0.0323,
+      "step": 1087
+    },
+    {
+      "epoch": 1.631184407796102,
+      "grad_norm": 0.6106854575869424,
+      "learning_rate": 8.158632442673603e-07,
+      "loss": 0.0294,
+      "step": 1088
+    },
+    {
+      "epoch": 1.6326836581709145,
+      "grad_norm": 0.5594423099210338,
+      "learning_rate": 8.094283840585398e-07,
+      "loss": 0.0268,
+      "step": 1089
+    },
+    {
+      "epoch": 1.6341829085457271,
+      "grad_norm": 0.6108168538137474,
+      "learning_rate": 8.03016765189103e-07,
+      "loss": 0.0273,
+      "step": 1090
+    },
+    {
+      "epoch": 1.6356821589205397,
+      "grad_norm": 0.6033527368655265,
+      "learning_rate": 7.966284232185451e-07,
+      "loss": 0.023,
+      "step": 1091
+    },
+    {
+      "epoch": 1.6371814092953523,
+      "grad_norm": 0.5863831430897024,
+      "learning_rate": 7.902633935772647e-07,
+      "loss": 0.025,
+      "step": 1092
+    },
+    {
+      "epoch": 1.638680659670165,
+      "grad_norm": 0.7468220324687744,
+      "learning_rate": 7.839217115663683e-07,
+      "loss": 0.0416,
+      "step": 1093
+    },
+    {
+      "epoch": 1.6401799100449774,
+      "grad_norm": 0.5556577429677079,
+      "learning_rate": 7.776034123574738e-07,
+      "loss": 0.0311,
+      "step": 1094
+    },
+    {
+      "epoch": 1.6416791604197902,
+      "grad_norm": 0.6057204490568195,
+      "learning_rate": 7.713085309925156e-07,
+      "loss": 0.0225,
+      "step": 1095
+    },
+    {
+      "epoch": 1.6431784107946026,
+      "grad_norm": 0.7582339116699343,
+      "learning_rate": 7.650371023835495e-07,
+      "loss": 0.0313,
+      "step": 1096
+    },
+    {
+      "epoch": 1.6446776611694154,
+      "grad_norm": 0.6487699034198455,
+      "learning_rate": 7.587891613125631e-07,
+      "loss": 0.0319,
+      "step": 1097
+    },
+    {
+      "epoch": 1.6461769115442277,
+      "grad_norm": 0.6222721677738255,
+      "learning_rate": 7.525647424312766e-07,
+      "loss": 0.0252,
+      "step": 1098
+    },
+    {
+      "epoch": 1.6476761619190405,
+      "grad_norm": 0.7029188305481112,
+      "learning_rate": 7.46363880260954e-07,
+      "loss": 0.0288,
+      "step": 1099
+    },
+    {
+      "epoch": 1.6491754122938531,
+      "grad_norm": 0.6545711837038333,
+      "learning_rate": 7.401866091922133e-07,
+      "loss": 0.0278,
+      "step": 1100
+    },
+    {
+      "epoch": 1.6506746626686657,
+      "grad_norm": 0.866461347003497,
+      "learning_rate": 7.340329634848309e-07,
+      "loss": 0.048,
+      "step": 1101
+    },
+    {
+      "epoch": 1.6521739130434783,
+      "grad_norm": 0.9131504675830925,
+      "learning_rate": 7.279029772675572e-07,
+      "loss": 0.0365,
+      "step": 1102
+    },
+    {
+      "epoch": 1.6536731634182908,
+      "grad_norm": 0.5540944140955992,
+      "learning_rate": 7.217966845379243e-07,
+      "loss": 0.022,
+      "step": 1103
+    },
+    {
+      "epoch": 1.6551724137931034,
+      "grad_norm": 0.4310501728939062,
+      "learning_rate": 7.157141191620548e-07,
+      "loss": 0.0131,
+      "step": 1104
+    },
+    {
+      "epoch": 1.656671664167916,
+      "grad_norm": 0.4414382897972463,
+      "learning_rate": 7.096553148744806e-07,
+      "loss": 0.0141,
+      "step": 1105
+    },
+    {
+      "epoch": 1.6581709145427288,
+      "grad_norm": 0.6222509027956581,
+      "learning_rate": 7.036203052779506e-07,
+      "loss": 0.0299,
+      "step": 1106
+    },
+    {
+      "epoch": 1.6596701649175412,
+      "grad_norm": 0.5224851882851093,
+      "learning_rate": 6.97609123843247e-07,
+      "loss": 0.021,
+      "step": 1107
+    },
+    {
+      "epoch": 1.661169415292354,
+      "grad_norm": 0.679577304262763,
+      "learning_rate": 6.916218039089961e-07,
+      "loss": 0.0304,
+      "step": 1108
+    },
+    {
+      "epoch": 1.6626686656671663,
+      "grad_norm": 0.7017320039898048,
+      "learning_rate": 6.856583786814891e-07,
+      "loss": 0.0336,
+      "step": 1109
+    },
+    {
+      "epoch": 1.6641679160419791,
+      "grad_norm": 0.7190674760121308,
+      "learning_rate": 6.797188812344907e-07,
+      "loss": 0.023,
+      "step": 1110
+    },
+    {
+      "epoch": 1.6656671664167915,
+      "grad_norm": 0.6161265420033984,
+      "learning_rate": 6.738033445090653e-07,
+      "loss": 0.0342,
+      "step": 1111
+    },
+    {
+      "epoch": 1.6671664167916043,
+      "grad_norm": 0.608315679504456,
+      "learning_rate": 6.67911801313384e-07,
+      "loss": 0.0285,
+      "step": 1112
+    },
+    {
+      "epoch": 1.6686656671664168,
+      "grad_norm": 0.5747957485349098,
+      "learning_rate": 6.620442843225483e-07,
+      "loss": 0.03,
+      "step": 1113
+    },
+    {
+      "epoch": 1.6701649175412294,
+      "grad_norm": 0.5136984313578205,
+      "learning_rate": 6.562008260784092e-07,
+      "loss": 0.0156,
+      "step": 1114
+    },
+    {
+      "epoch": 1.671664167916042,
+      "grad_norm": 0.4741666182970135,
+      "learning_rate": 6.503814589893836e-07,
+      "loss": 0.0163,
+      "step": 1115
+    },
+    {
+      "epoch": 1.6731634182908546,
+      "grad_norm": 0.5173348908670526,
+      "learning_rate": 6.445862153302784e-07,
+      "loss": 0.0217,
+      "step": 1116
+    },
+    {
+      "epoch": 1.6746626686656672,
+      "grad_norm": 0.6696069749101599,
+      "learning_rate": 6.388151272421078e-07,
+      "loss": 0.0297,
+      "step": 1117
+    },
+    {
+      "epoch": 1.6761619190404797,
+      "grad_norm": 0.4868384629819061,
+      "learning_rate": 6.330682267319177e-07,
+      "loss": 0.0182,
+      "step": 1118
+    },
+    {
+      "epoch": 1.6776611694152923,
+      "grad_norm": 0.6296740321969375,
+      "learning_rate": 6.273455456726074e-07,
+      "loss": 0.0215,
+      "step": 1119
+    },
+    {
+      "epoch": 1.6791604197901049,
+      "grad_norm": 0.585822115325922,
+      "learning_rate": 6.216471158027515e-07,
+      "loss": 0.0314,
+      "step": 1120
+    },
+    {
+      "epoch": 1.6806596701649177,
+      "grad_norm": 0.5114788294546149,
+      "learning_rate": 6.159729687264254e-07,
+      "loss": 0.0166,
+      "step": 1121
+    },
+    {
+      "epoch": 1.68215892053973,
+      "grad_norm": 0.91667505757614,
+      "learning_rate": 6.103231359130308e-07,
+      "loss": 0.0329,
+      "step": 1122
+    },
+    {
+      "epoch": 1.6836581709145428,
+      "grad_norm": 0.7221532809526754,
+      "learning_rate": 6.046976486971201e-07,
+      "loss": 0.0307,
+      "step": 1123
+    },
+    {
+      "epoch": 1.6851574212893552,
+      "grad_norm": 0.6702879869078131,
+      "learning_rate": 5.990965382782177e-07,
+      "loss": 0.0266,
+      "step": 1124
+    },
+    {
+      "epoch": 1.686656671664168,
+      "grad_norm": 0.6171695966878039,
+      "learning_rate": 5.935198357206595e-07,
+      "loss": 0.025,
+      "step": 1125
+    },
+    {
+      "epoch": 1.6881559220389803,
+      "grad_norm": 0.5267858538997529,
+      "learning_rate": 5.879675719534078e-07,
+      "loss": 0.0283,
+      "step": 1126
+    },
+    {
+      "epoch": 1.6896551724137931,
+      "grad_norm": 0.586200195013088,
+      "learning_rate": 5.824397777698859e-07,
+      "loss": 0.0273,
+      "step": 1127
+    },
+    {
+      "epoch": 1.6911544227886057,
+      "grad_norm": 0.6418672782506534,
+      "learning_rate": 5.769364838278063e-07,
+      "loss": 0.021,
+      "step": 1128
+    },
+    {
+      "epoch": 1.6926536731634183,
+      "grad_norm": 0.5380974051719066,
+      "learning_rate": 5.714577206490018e-07,
+      "loss": 0.0241,
+      "step": 1129
+    },
+    {
+      "epoch": 1.6941529235382309,
+      "grad_norm": 0.48809976003281486,
+      "learning_rate": 5.660035186192531e-07,
+      "loss": 0.0187,
+      "step": 1130
+    },
+    {
+      "epoch": 1.6956521739130435,
+      "grad_norm": 0.557803596562225,
+      "learning_rate": 5.60573907988124e-07,
+      "loss": 0.026,
+      "step": 1131
+    },
+    {
+      "epoch": 1.697151424287856,
+      "grad_norm": 0.6774225620525076,
+      "learning_rate": 5.551689188687909e-07,
+      "loss": 0.0336,
+      "step": 1132
+    },
+    {
+      "epoch": 1.6986506746626686,
+      "grad_norm": 0.6021993724745683,
+      "learning_rate": 5.497885812378772e-07,
+      "loss": 0.0252,
+      "step": 1133
+    },
+    {
+      "epoch": 1.7001499250374814,
+      "grad_norm": 0.6988442811975228,
+      "learning_rate": 5.444329249352859e-07,
+      "loss": 0.0383,
+      "step": 1134
+    },
+    {
+      "epoch": 1.7016491754122938,
+      "grad_norm": 0.6140914821939075,
+      "learning_rate": 5.391019796640362e-07,
+      "loss": 0.0253,
+      "step": 1135
+    },
+    {
+      "epoch": 1.7031484257871066,
+      "grad_norm": 0.6199587672293568,
+      "learning_rate": 5.337957749900958e-07,
+      "loss": 0.0226,
+      "step": 1136
+    },
+    {
+      "epoch": 1.704647676161919,
+      "grad_norm": 0.5371199700799781,
+      "learning_rate": 5.285143403422188e-07,
+      "loss": 0.0209,
+      "step": 1137
+    },
+    {
+      "epoch": 1.7061469265367317,
+      "grad_norm": 0.4971525387739954,
+      "learning_rate": 5.23257705011786e-07,
+      "loss": 0.0195,
+      "step": 1138
+    },
+    {
+      "epoch": 1.707646176911544,
+      "grad_norm": 0.5798784358090088,
+      "learning_rate": 5.18025898152631e-07,
+      "loss": 0.0201,
+      "step": 1139
+    },
+    {
+      "epoch": 1.7091454272863569,
+      "grad_norm": 0.4351744256946392,
+      "learning_rate": 5.128189487808927e-07,
+      "loss": 0.0147,
+      "step": 1140
+    },
+    {
+      "epoch": 1.7106446776611695,
+      "grad_norm": 0.806659381591431,
+      "learning_rate": 5.076368857748454e-07,
+      "loss": 0.0374,
+      "step": 1141
+    },
+    {
+      "epoch": 1.712143928035982,
+      "grad_norm": 0.7255030834147108,
+      "learning_rate": 5.024797378747414e-07,
+      "loss": 0.0323,
+      "step": 1142
+    },
+    {
+      "epoch": 1.7136431784107946,
+      "grad_norm": 0.5552798055375805,
+      "learning_rate": 4.973475336826506e-07,
+      "loss": 0.0224,
+      "step": 1143
+    },
+    {
+      "epoch": 1.7151424287856072,
+      "grad_norm": 0.4920343185383511,
+      "learning_rate": 4.922403016623034e-07,
+      "loss": 0.02,
+      "step": 1144
+    },
+    {
+      "epoch": 1.7166416791604198,
+      "grad_norm": 0.6315724715123963,
+      "learning_rate": 4.871580701389316e-07,
+      "loss": 0.0226,
+      "step": 1145
+    },
+    {
+      "epoch": 1.7181409295352323,
+      "grad_norm": 0.6352741641543308,
+      "learning_rate": 4.821008672991118e-07,
+      "loss": 0.0307,
+      "step": 1146
+    },
+    {
+      "epoch": 1.7196401799100451,
+      "grad_norm": 0.6208909962478606,
+      "learning_rate": 4.770687211906089e-07,
+      "loss": 0.0261,
+      "step": 1147
+    },
+    {
+      "epoch": 1.7211394302848575,
+      "grad_norm": 0.5995800649761945,
+      "learning_rate": 4.720616597222205e-07,
+      "loss": 0.0289,
+      "step": 1148
+    },
+    {
+      "epoch": 1.7226386806596703,
+      "grad_norm": 0.4077879295148252,
+      "learning_rate": 4.6707971066362324e-07,
+      "loss": 0.0149,
+      "step": 1149
+    },
+    {
+      "epoch": 1.7241379310344827,
+      "grad_norm": 0.7170338147857434,
+      "learning_rate": 4.6212290164521554e-07,
+      "loss": 0.0299,
+      "step": 1150
+    },
+    {
+      "epoch": 1.7256371814092955,
+      "grad_norm": 0.4283550290277706,
+      "learning_rate": 4.5719126015796757e-07,
+      "loss": 0.0161,
+      "step": 1151
+    },
+    {
+      "epoch": 1.7271364317841078,
+      "grad_norm": 0.5754740014590574,
+      "learning_rate": 4.522848135532698e-07,
+      "loss": 0.0242,
+      "step": 1152
+    },
+    {
+      "epoch": 1.7286356821589206,
+      "grad_norm": 0.7619322945086459,
+      "learning_rate": 4.474035890427769e-07,
+      "loss": 0.0321,
+      "step": 1153
+    },
+    {
+      "epoch": 1.7301349325337332,
+      "grad_norm": 0.5971046163170837,
+      "learning_rate": 4.4254761369825984e-07,
+      "loss": 0.0228,
+      "step": 1154
+    },
+    {
+      "epoch": 1.7316341829085458,
+      "grad_norm": 0.40318806137764734,
+      "learning_rate": 4.377169144514554e-07,
+      "loss": 0.0161,
+      "step": 1155
+    },
+    {
+      "epoch": 1.7331334332833583,
+      "grad_norm": 0.6458966138955453,
+      "learning_rate": 4.329115180939164e-07,
+      "loss": 0.0286,
+      "step": 1156
+    },
+    {
+      "epoch": 1.734632683658171,
+      "grad_norm": 0.6509779237643041,
+      "learning_rate": 4.281314512768625e-07,
+      "loss": 0.0242,
+      "step": 1157
+    },
+    {
+      "epoch": 1.7361319340329835,
+      "grad_norm": 0.755074251594893,
+      "learning_rate": 4.2337674051103504e-07,
+      "loss": 0.0349,
+      "step": 1158
+    },
+    {
+      "epoch": 1.737631184407796,
+      "grad_norm": 0.5725313638530071,
+      "learning_rate": 4.186474121665468e-07,
+      "loss": 0.0278,
+      "step": 1159
+    },
+    {
+      "epoch": 1.7391304347826086,
+      "grad_norm": 0.5075255327064533,
+      "learning_rate": 4.139434924727359e-07,
+      "loss": 0.0212,
+      "step": 1160
+    },
+    {
+      "epoch": 1.7406296851574212,
+      "grad_norm": 0.5518391176352521,
+      "learning_rate": 4.092650075180232e-07,
+      "loss": 0.0241,
+      "step": 1161
+    },
+    {
+      "epoch": 1.742128935532234,
+      "grad_norm": 0.4941209636784168,
+      "learning_rate": 4.046119832497658e-07,
+      "loss": 0.0211,
+      "step": 1162
+    },
+    {
+      "epoch": 1.7436281859070464,
+      "grad_norm": 0.6003642986466252,
+      "learning_rate": 3.9998444547411255e-07,
+      "loss": 0.0255,
+      "step": 1163
+    },
+    {
+      "epoch": 1.7451274362818592,
+      "grad_norm": 0.5594669903212188,
+      "learning_rate": 3.9538241985586144e-07,
+      "loss": 0.0221,
+      "step": 1164
+    },
+    {
+      "epoch": 1.7466266866566715,
+      "grad_norm": 0.6581986399705584,
+      "learning_rate": 3.908059319183194e-07,
+      "loss": 0.0352,
+      "step": 1165
+    },
+    {
+      "epoch": 1.7481259370314843,
+      "grad_norm": 0.472655086994358,
+      "learning_rate": 3.8625500704315645e-07,
+      "loss": 0.017,
+      "step": 1166
+    },
+    {
+      "epoch": 1.7496251874062967,
+      "grad_norm": 0.521890986372451,
+      "learning_rate": 3.8172967047026834e-07,
+      "loss": 0.0257,
+      "step": 1167
+    },
+    {
+      "epoch": 1.7511244377811095,
+      "grad_norm": 0.5477231660945511,
+      "learning_rate": 3.7722994729763427e-07,
+      "loss": 0.0246,
+      "step": 1168
+    },
+    {
+      "epoch": 1.752623688155922,
+      "grad_norm": 0.591973946291394,
+      "learning_rate": 3.7275586248118114e-07,
+      "loss": 0.0338,
+      "step": 1169
+    },
+    {
+      "epoch": 1.7541229385307346,
+      "grad_norm": 0.5173649766154534,
+      "learning_rate": 3.683074408346404e-07,
+      "loss": 0.0216,
+      "step": 1170
+    },
+    {
+      "epoch": 1.7556221889055472,
+      "grad_norm": 0.52748486193648,
+      "learning_rate": 3.6388470702941436e-07,
+      "loss": 0.0261,
+      "step": 1171
+    },
+    {
+      "epoch": 1.7571214392803598,
+      "grad_norm": 0.7881512232217596,
+      "learning_rate": 3.594876855944385e-07,
+      "loss": 0.0381,
+      "step": 1172
+    },
+    {
+      "epoch": 1.7586206896551724,
+      "grad_norm": 0.5071093958983124,
+      "learning_rate": 3.5511640091604293e-07,
+      "loss": 0.0195,
+      "step": 1173
+    },
+    {
+      "epoch": 1.760119940029985,
+      "grad_norm": 0.45701973135991986,
+      "learning_rate": 3.50770877237821e-07,
+      "loss": 0.0184,
+      "step": 1174
+    },
+    {
+      "epoch": 1.7616191904047978,
+      "grad_norm": 0.6164588077330094,
+      "learning_rate": 3.4645113866049187e-07,
+      "loss": 0.0253,
+      "step": 1175
+    },
+    {
+      "epoch": 1.76311844077961,
+      "grad_norm": 0.46739210143555815,
+      "learning_rate": 3.42157209141768e-07,
+      "loss": 0.0192,
+      "step": 1176
+    },
+    {
+      "epoch": 1.764617691154423,
+      "grad_norm": 0.6966939996938556,
+      "learning_rate": 3.3788911249622194e-07,
+      "loss": 0.0303,
+      "step": 1177
+    },
+    {
+      "epoch": 1.7661169415292353,
+      "grad_norm": 0.5794470559999699,
+      "learning_rate": 3.336468723951558e-07,
+      "loss": 0.0228,
+      "step": 1178
+    },
+    {
+      "epoch": 1.767616191904048,
+      "grad_norm": 0.616562632869892,
+      "learning_rate": 3.294305123664665e-07,
+      "loss": 0.0249,
+      "step": 1179
+    },
+    {
+      "epoch": 1.7691154422788604,
+      "grad_norm": 0.6730348102218151,
+      "learning_rate": 3.2524005579452014e-07,
+      "loss": 0.0262,
+      "step": 1180
+    },
+    {
+      "epoch": 1.7706146926536732,
+      "grad_norm": 0.5776091703964031,
+      "learning_rate": 3.2107552592001657e-07,
+      "loss": 0.026,
+      "step": 1181
+    },
+    {
+      "epoch": 1.7721139430284858,
+      "grad_norm": 0.478529148255476,
+      "learning_rate": 3.169369458398652e-07,
+      "loss": 0.0167,
+      "step": 1182
+    },
+    {
+      "epoch": 1.7736131934032984,
+      "grad_norm": 0.5075951030224909,
+      "learning_rate": 3.128243385070562e-07,
+      "loss": 0.0221,
+      "step": 1183
+    },
+    {
+      "epoch": 1.775112443778111,
+      "grad_norm": 0.5515928433132049,
+      "learning_rate": 3.087377267305297e-07,
+      "loss": 0.0307,
+      "step": 1184
+    },
+    {
+      "epoch": 1.7766116941529235,
+      "grad_norm": 0.5178251901742074,
+      "learning_rate": 3.0467713317505363e-07,
+      "loss": 0.0237,
+      "step": 1185
+    },
+    {
+      "epoch": 1.778110944527736,
+      "grad_norm": 0.5615234791783714,
+      "learning_rate": 3.006425803610963e-07,
+      "loss": 0.0295,
+      "step": 1186
+    },
+    {
+      "epoch": 1.7796101949025487,
+      "grad_norm": 0.7141294229980446,
+      "learning_rate": 2.9663409066470025e-07,
+      "loss": 0.0275,
+      "step": 1187
+    },
+    {
+      "epoch": 1.7811094452773615,
+      "grad_norm": 0.5627321524477661,
+      "learning_rate": 2.9265168631736005e-07,
+      "loss": 0.0257,
+      "step": 1188
+    },
+    {
+      "epoch": 1.7826086956521738,
+      "grad_norm": 0.669549235386019,
+      "learning_rate": 2.88695389405898e-07,
+      "loss": 0.026,
+      "step": 1189
+    },
+    {
+      "epoch": 1.7841079460269866,
+      "grad_norm": 0.6046853402932771,
+      "learning_rate": 2.8476522187234177e-07,
+      "loss": 0.0253,
+      "step": 1190
+    },
+    {
+      "epoch": 1.785607196401799,
+      "grad_norm": 0.5795117857109351,
+      "learning_rate": 2.808612055138038e-07,
+      "loss": 0.0354,
+      "step": 1191
+    },
+    {
+      "epoch": 1.7871064467766118,
+      "grad_norm": 0.5680272552891236,
+      "learning_rate": 2.76983361982357e-07,
+      "loss": 0.022,
+      "step": 1192
+    },
+    {
+      "epoch": 1.7886056971514241,
+      "grad_norm": 0.4367472067289223,
+      "learning_rate": 2.731317127849209e-07,
+      "loss": 0.0197,
+      "step": 1193
+    },
+    {
+      "epoch": 1.790104947526237,
+      "grad_norm": 0.6273382792037528,
+      "learning_rate": 2.693062792831358e-07,
+      "loss": 0.0271,
+      "step": 1194
+    },
+    {
+      "epoch": 1.7916041979010495,
+      "grad_norm": 0.6697978088290386,
+      "learning_rate": 2.655070826932471e-07,
+      "loss": 0.0247,
+      "step": 1195
+    },
+    {
+      "epoch": 1.793103448275862,
+      "grad_norm": 0.5862269400695345,
+      "learning_rate": 2.617341440859883e-07,
+      "loss": 0.0239,
+      "step": 1196
+    },
+    {
+      "epoch": 1.7946026986506747,
+      "grad_norm": 0.4790274821761689,
+      "learning_rate": 2.5798748438646326e-07,
+      "loss": 0.0217,
+      "step": 1197
+    },
+    {
+      "epoch": 1.7961019490254873,
+      "grad_norm": 0.4676781880807092,
+      "learning_rate": 2.5426712437403134e-07,
+      "loss": 0.0247,
+      "step": 1198
+    },
+    {
+      "epoch": 1.7976011994002998,
+      "grad_norm": 0.5186078852697914,
+      "learning_rate": 2.5057308468218913e-07,
+      "loss": 0.0188,
+      "step": 1199
+    },
+    {
+      "epoch": 1.7991004497751124,
+      "grad_norm": 0.5120669322026774,
+      "learning_rate": 2.4690538579845933e-07,
+      "loss": 0.0254,
+      "step": 1200
+    },
+    {
+      "epoch": 1.7991004497751124,
+      "eval_loss": 0.10704679042100906,
+      "eval_runtime": 2.1938,
+      "eval_samples_per_second": 24.614,
+      "eval_steps_per_second": 6.382,
+      "step": 1200
+    },
+    {
+      "epoch": 1.800599700149925,
+      "grad_norm": 0.7137875375334901,
+      "learning_rate": 2.432640480642756e-07,
+      "loss": 0.0327,
+      "step": 1201
+    },
+    {
+      "epoch": 1.8020989505247376,
+      "grad_norm": 0.5141795120607272,
+      "learning_rate": 2.396490916748706e-07,
+      "loss": 0.0228,
+      "step": 1202
+    },
+    {
+      "epoch": 1.8035982008995504,
+      "grad_norm": 1.2933185909376461,
+      "learning_rate": 2.360605366791624e-07,
+      "loss": 0.0451,
+      "step": 1203
+    },
+    {
+      "epoch": 1.8050974512743627,
+      "grad_norm": 0.5934434537451942,
+      "learning_rate": 2.32498402979644e-07,
+      "loss": 0.0272,
+      "step": 1204
+    },
+    {
+      "epoch": 1.8065967016491755,
+      "grad_norm": 0.541371434587485,
+      "learning_rate": 2.2896271033227392e-07,
+      "loss": 0.0305,
+      "step": 1205
+    },
+    {
+      "epoch": 1.8080959520239879,
+      "grad_norm": 0.8909500515180906,
+      "learning_rate": 2.2545347834636632e-07,
+      "loss": 0.0335,
+      "step": 1206
+    },
+    {
+      "epoch": 1.8095952023988007,
+      "grad_norm": 0.49683029922261046,
+      "learning_rate": 2.219707264844806e-07,
+      "loss": 0.0191,
+      "step": 1207
+    },
+    {
+      "epoch": 1.811094452773613,
+      "grad_norm": 0.6765862680834843,
+      "learning_rate": 2.1851447406231573e-07,
+      "loss": 0.0285,
+      "step": 1208
+    },
+    {
+      "epoch": 1.8125937031484258,
+      "grad_norm": 0.658690672930186,
+      "learning_rate": 2.1508474024860171e-07,
+      "loss": 0.0423,
+      "step": 1209
+    },
+    {
+      "epoch": 1.8140929535232384,
+      "grad_norm": 0.43679648913307667,
+      "learning_rate": 2.1168154406499275e-07,
+      "loss": 0.0178,
+      "step": 1210
+    },
+    {
+      "epoch": 1.815592203898051,
+      "grad_norm": 0.6038864636959431,
+      "learning_rate": 2.0830490438596418e-07,
+      "loss": 0.0263,
+      "step": 1211
+    },
+    {
+      "epoch": 1.8170914542728636,
+      "grad_norm": 0.44390459184629255,
+      "learning_rate": 2.0495483993870578e-07,
+      "loss": 0.0218,
+      "step": 1212
+    },
+    {
+      "epoch": 1.8185907046476761,
+      "grad_norm": 0.605006088632554,
+      "learning_rate": 2.0163136930301696e-07,
+      "loss": 0.026,
+      "step": 1213
+    },
+    {
+      "epoch": 1.8200899550224887,
+      "grad_norm": 0.6622382426888629,
+      "learning_rate": 1.9833451091120727e-07,
+      "loss": 0.0261,
+      "step": 1214
+    },
+    {
+      "epoch": 1.8215892053973013,
+      "grad_norm": 0.7266522707476083,
+      "learning_rate": 1.9506428304799095e-07,
+      "loss": 0.0331,
+      "step": 1215
+    },
+    {
+      "epoch": 1.823088455772114,
+      "grad_norm": 0.5780129495004762,
+      "learning_rate": 1.9182070385038555e-07,
+      "loss": 0.0241,
+      "step": 1216
+    },
+    {
+      "epoch": 1.8245877061469264,
+      "grad_norm": 0.4632801516931901,
+      "learning_rate": 1.886037913076144e-07,
+      "loss": 0.0156,
+      "step": 1217
+    },
+    {
+      "epoch": 1.8260869565217392,
+      "grad_norm": 0.607272800609473,
+      "learning_rate": 1.8541356326100436e-07,
+      "loss": 0.0255,
+      "step": 1218
+    },
+    {
+      "epoch": 1.8275862068965516,
+      "grad_norm": 0.6191376123438204,
+      "learning_rate": 1.8225003740388546e-07,
+      "loss": 0.0301,
+      "step": 1219
+    },
+    {
+      "epoch": 1.8290854572713644,
+      "grad_norm": 0.4856818758813461,
+      "learning_rate": 1.791132312814975e-07,
+      "loss": 0.0196,
+      "step": 1220
+    },
+    {
+      "epoch": 1.8305847076461768,
+      "grad_norm": 0.638579342854401,
+      "learning_rate": 1.760031622908881e-07,
+      "loss": 0.0263,
+      "step": 1221
+    },
+    {
+      "epoch": 1.8320839580209896,
+      "grad_norm": 0.4924740365046096,
+      "learning_rate": 1.729198476808186e-07,
+      "loss": 0.0201,
+      "step": 1222
+    },
+    {
+      "epoch": 1.8335832083958021,
+      "grad_norm": 0.4704306637890718,
+      "learning_rate": 1.6986330455166733e-07,
+      "loss": 0.0196,
+      "step": 1223
+    },
+    {
+      "epoch": 1.8350824587706147,
+      "grad_norm": 0.7011294042836334,
+      "learning_rate": 1.6683354985533583e-07,
+      "loss": 0.0251,
+      "step": 1224
+    },
+    {
+      "epoch": 1.8365817091454273,
+      "grad_norm": 0.5249744934392139,
+      "learning_rate": 1.6383060039515343e-07,
+      "loss": 0.02,
+      "step": 1225
+    },
+    {
+      "epoch": 1.8380809595202399,
+      "grad_norm": 0.6505849657851941,
+      "learning_rate": 1.6085447282578548e-07,
+      "loss": 0.0297,
+      "step": 1226
+    },
+    {
+      "epoch": 1.8395802098950524,
+      "grad_norm": 0.6232703761779411,
+      "learning_rate": 1.579051836531409e-07,
+      "loss": 0.0248,
+      "step": 1227
+    },
+    {
+      "epoch": 1.841079460269865,
+      "grad_norm": 0.6967483760689221,
+      "learning_rate": 1.5498274923427925e-07,
+      "loss": 0.0309,
+      "step": 1228
+    },
+    {
+      "epoch": 1.8425787106446778,
+      "grad_norm": 0.7315438231021295,
+      "learning_rate": 1.5208718577732096e-07,
+      "loss": 0.0352,
+      "step": 1229
+    },
+    {
+      "epoch": 1.8440779610194902,
+      "grad_norm": 0.4705673724149769,
+      "learning_rate": 1.4921850934135785e-07,
+      "loss": 0.0219,
+      "step": 1230
+    },
+    {
+      "epoch": 1.845577211394303,
+      "grad_norm": 0.7950138994902668,
+      "learning_rate": 1.463767358363627e-07,
+      "loss": 0.0444,
+      "step": 1231
+    },
+    {
+      "epoch": 1.8470764617691153,
+      "grad_norm": 0.6505613406984727,
+      "learning_rate": 1.4356188102310266e-07,
+      "loss": 0.0344,
+      "step": 1232
+    },
+    {
+      "epoch": 1.8485757121439281,
+      "grad_norm": 0.6105544736338776,
+      "learning_rate": 1.4077396051305093e-07,
+      "loss": 0.0257,
+      "step": 1233
+    },
+    {
+      "epoch": 1.8500749625187405,
+      "grad_norm": 0.52842618545126,
+      "learning_rate": 1.3801298976830025e-07,
+      "loss": 0.0261,
+      "step": 1234
+    },
+    {
+      "epoch": 1.8515742128935533,
+      "grad_norm": 0.5053114213466128,
+      "learning_rate": 1.3527898410147677e-07,
+      "loss": 0.0198,
+      "step": 1235
+    },
+    {
+      "epoch": 1.8530734632683659,
+      "grad_norm": 0.6144363177426994,
+      "learning_rate": 1.325719586756563e-07,
+      "loss": 0.03,
+      "step": 1236
+    },
+    {
+      "epoch": 1.8545727136431784,
+      "grad_norm": 0.5692689601801861,
+      "learning_rate": 1.2989192850427933e-07,
+      "loss": 0.0285,
+      "step": 1237
+    },
+    {
+      "epoch": 1.856071964017991,
+      "grad_norm": 0.7663000569195995,
+      "learning_rate": 1.2723890845106723e-07,
+      "loss": 0.0317,
+      "step": 1238
+    },
+    {
+      "epoch": 1.8575712143928036,
+      "grad_norm": 0.5413833016273539,
+      "learning_rate": 1.2461291322994118e-07,
+      "loss": 0.0281,
+      "step": 1239
+    },
+    {
+      "epoch": 1.8590704647676162,
+      "grad_norm": 0.5754685249960622,
+      "learning_rate": 1.2201395740493948e-07,
+      "loss": 0.0252,
+      "step": 1240
+    },
+    {
+      "epoch": 1.8605697151424287,
+      "grad_norm": 0.6670912525939089,
+      "learning_rate": 1.1944205539013708e-07,
+      "loss": 0.029,
+      "step": 1241
+    },
+    {
+      "epoch": 1.8620689655172413,
+      "grad_norm": 0.4175640747819078,
+      "learning_rate": 1.1689722144956672e-07,
+      "loss": 0.0202,
+      "step": 1242
+    },
+    {
+      "epoch": 1.863568215892054,
+      "grad_norm": 0.563422656631543,
+      "learning_rate": 1.1437946969713731e-07,
+      "loss": 0.0313,
+      "step": 1243
+    },
+    {
+      "epoch": 1.8650674662668667,
+      "grad_norm": 0.7289878932422447,
+      "learning_rate": 1.1188881409655849e-07,
+      "loss": 0.0342,
+      "step": 1244
+    },
+    {
+      "epoch": 1.866566716641679,
+      "grad_norm": 0.5742843496130039,
+      "learning_rate": 1.0942526846126122e-07,
+      "loss": 0.0295,
+      "step": 1245
+    },
+    {
+      "epoch": 1.8680659670164919,
+      "grad_norm": 0.6136590186975005,
+      "learning_rate": 1.0698884645432117e-07,
+      "loss": 0.0208,
+      "step": 1246
+    },
+    {
+      "epoch": 1.8695652173913042,
+      "grad_norm": 0.45250357412457126,
+      "learning_rate": 1.0457956158838545e-07,
+      "loss": 0.0192,
+      "step": 1247
+    },
+    {
+      "epoch": 1.871064467766117,
+      "grad_norm": 0.5750821599320451,
+      "learning_rate": 1.0219742722559433e-07,
+      "loss": 0.023,
+      "step": 1248
+    },
+    {
+      "epoch": 1.8725637181409296,
+      "grad_norm": 0.594463780416723,
+      "learning_rate": 9.984245657750857e-08,
+      "loss": 0.0306,
+      "step": 1249
+    },
+    {
+      "epoch": 1.8740629685157422,
+      "grad_norm": 0.8404959977273015,
+      "learning_rate": 9.751466270503718e-08,
+      "loss": 0.0355,
+      "step": 1250
+    },
+    {
+      "epoch": 1.8755622188905547,
+      "grad_norm": 0.7269826500830991,
+      "learning_rate": 9.521405851836252e-08,
+      "loss": 0.0343,
+      "step": 1251
+    },
+    {
+      "epoch": 1.8770614692653673,
+      "grad_norm": 0.5919616221138111,
+      "learning_rate": 9.294065677687202e-08,
+      "loss": 0.0251,
+      "step": 1252
+    },
+    {
+      "epoch": 1.87856071964018,
+      "grad_norm": 0.6841986166507333,
+      "learning_rate": 9.069447008908383e-08,
+      "loss": 0.044,
+      "step": 1253
+    },
+    {
+      "epoch": 1.8800599700149925,
+      "grad_norm": 0.43106682098606064,
+      "learning_rate": 8.847551091257956e-08,
+      "loss": 0.0166,
+      "step": 1254
+    },
+    {
+      "epoch": 1.881559220389805,
+      "grad_norm": 0.5473614898025452,
+      "learning_rate": 8.62837915539344e-08,
+      "loss": 0.0274,
+      "step": 1255
+    },
+    {
+      "epoch": 1.8830584707646176,
+      "grad_norm": 0.4759967675756749,
+      "learning_rate": 8.411932416864832e-08,
+      "loss": 0.0163,
+      "step": 1256
+    },
+    {
+      "epoch": 1.8845577211394304,
+      "grad_norm": 0.6093682280360255,
+      "learning_rate": 8.198212076107881e-08,
+      "loss": 0.0241,
+      "step": 1257
+    },
+    {
+      "epoch": 1.8860569715142428,
+      "grad_norm": 0.3562122751108543,
+      "learning_rate": 7.987219318437489e-08,
+      "loss": 0.0137,
+      "step": 1258
+    },
+    {
+      "epoch": 1.8875562218890556,
+      "grad_norm": 0.485196204176868,
+      "learning_rate": 7.778955314041103e-08,
+      "loss": 0.0159,
+      "step": 1259
+    },
+    {
+      "epoch": 1.889055472263868,
+      "grad_norm": 0.5237569977041334,
+      "learning_rate": 7.573421217972222e-08,
+      "loss": 0.02,
+      "step": 1260
+    },
+    {
+      "epoch": 1.8905547226386807,
+      "grad_norm": 0.8131695192861765,
+      "learning_rate": 7.370618170144062e-08,
+      "loss": 0.0488,
+      "step": 1261
+    },
+    {
+      "epoch": 1.892053973013493,
+      "grad_norm": 0.6256127335186865,
+      "learning_rate": 7.170547295323016e-08,
+      "loss": 0.042,
+      "step": 1262
+    },
+    {
+      "epoch": 1.893553223388306,
+      "grad_norm": 0.5142620635479186,
+      "learning_rate": 6.973209703122652e-08,
+      "loss": 0.0214,
+      "step": 1263
+    },
+    {
+      "epoch": 1.8950524737631185,
+      "grad_norm": 0.620943881677787,
+      "learning_rate": 6.778606487997496e-08,
+      "loss": 0.0291,
+      "step": 1264
+    },
+    {
+      "epoch": 1.896551724137931,
+      "grad_norm": 0.6249915440791796,
+      "learning_rate": 6.58673872923693e-08,
+      "loss": 0.0296,
+      "step": 1265
+    },
+    {
+      "epoch": 1.8980509745127436,
+      "grad_norm": 0.6539146916161402,
+      "learning_rate": 6.397607490959134e-08,
+      "loss": 0.0265,
+      "step": 1266
+    },
+    {
+      "epoch": 1.8995502248875562,
+      "grad_norm": 0.7807718588357045,
+      "learning_rate": 6.211213822105378e-08,
+      "loss": 0.0417,
+      "step": 1267
+    },
+    {
+      "epoch": 1.9010494752623688,
+      "grad_norm": 0.5918332727126081,
+      "learning_rate": 6.027558756434015e-08,
+      "loss": 0.0249,
+      "step": 1268
+    },
+    {
+      "epoch": 1.9025487256371814,
+      "grad_norm": 0.48653789208077103,
+      "learning_rate": 5.846643312514888e-08,
+      "loss": 0.0223,
+      "step": 1269
+    },
+    {
+      "epoch": 1.9040479760119942,
+      "grad_norm": 0.4752827906058468,
+      "learning_rate": 5.668468493723489e-08,
+      "loss": 0.0167,
+      "step": 1270
+    },
+    {
+      "epoch": 1.9055472263868065,
+      "grad_norm": 0.5741843381247418,
+      "learning_rate": 5.4930352882357486e-08,
+      "loss": 0.0192,
+      "step": 1271
+    },
+    {
+      "epoch": 1.9070464767616193,
+      "grad_norm": 0.7121489592528197,
+      "learning_rate": 5.3203446690220374e-08,
+      "loss": 0.032,
+      "step": 1272
+    },
+    {
+      "epoch": 1.9085457271364317,
+      "grad_norm": 0.587996305836649,
+      "learning_rate": 5.1503975938422824e-08,
+      "loss": 0.021,
+      "step": 1273
+    },
+    {
+      "epoch": 1.9100449775112445,
+      "grad_norm": 0.5356180461605957,
+      "learning_rate": 4.983195005240415e-08,
+      "loss": 0.0163,
+      "step": 1274
+    },
+    {
+      "epoch": 1.9115442278860568,
+      "grad_norm": 0.5921571834238943,
+      "learning_rate": 4.8187378305390994e-08,
+      "loss": 0.0257,
+      "step": 1275
+    },
+    {
+      "epoch": 1.9130434782608696,
+      "grad_norm": 0.47409142652527236,
+      "learning_rate": 4.657026981834623e-08,
+      "loss": 0.0186,
+      "step": 1276
+    },
+    {
+      "epoch": 1.9145427286356822,
+      "grad_norm": 0.6068263347529701,
+      "learning_rate": 4.498063355991955e-08,
+      "loss": 0.0303,
+      "step": 1277
+    },
+    {
+      "epoch": 1.9160419790104948,
+      "grad_norm": 0.6086885887988437,
+      "learning_rate": 4.341847834639645e-08,
+      "loss": 0.0263,
+      "step": 1278
+    },
+    {
+      "epoch": 1.9175412293853074,
+      "grad_norm": 0.5315711547588107,
+      "learning_rate": 4.188381284164933e-08,
+      "loss": 0.0165,
+      "step": 1279
+    },
+    {
+      "epoch": 1.91904047976012,
+      "grad_norm": 0.5965893424305508,
+      "learning_rate": 4.0376645557090864e-08,
+      "loss": 0.0366,
+      "step": 1280
+    },
+    {
+      "epoch": 1.9205397301349325,
+      "grad_norm": 0.7966171911348032,
+      "learning_rate": 3.889698485162463e-08,
+      "loss": 0.0375,
+      "step": 1281
+    },
+    {
+      "epoch": 1.922038980509745,
+      "grad_norm": 0.6578451468432552,
+      "learning_rate": 3.744483893160067e-08,
+      "loss": 0.0322,
+      "step": 1282
+    },
+    {
+      "epoch": 1.9235382308845579,
+      "grad_norm": 0.6099046078677133,
+      "learning_rate": 3.602021585076942e-08,
+      "loss": 0.026,
+      "step": 1283
+    },
+    {
+      "epoch": 1.9250374812593702,
+      "grad_norm": 0.7197439041888303,
+      "learning_rate": 3.462312351023567e-08,
+      "loss": 0.0317,
+      "step": 1284
+    },
+    {
+      "epoch": 1.926536731634183,
+      "grad_norm": 0.703524009085726,
+      "learning_rate": 3.325356965841686e-08,
+      "loss": 0.0201,
+      "step": 1285
+    },
+    {
+      "epoch": 1.9280359820089954,
+      "grad_norm": 0.5038196386859377,
+      "learning_rate": 3.191156189099931e-08,
+      "loss": 0.0204,
+      "step": 1286
+    },
+    {
+      "epoch": 1.9295352323838082,
+      "grad_norm": 0.5518482974417659,
+      "learning_rate": 3.0597107650894855e-08,
+      "loss": 0.0244,
+      "step": 1287
+    },
+    {
+      "epoch": 1.9310344827586206,
+      "grad_norm": 0.5423867894643277,
+      "learning_rate": 2.9310214228202016e-08,
+      "loss": 0.0211,
+      "step": 1288
+    },
+    {
+      "epoch": 1.9325337331334334,
+      "grad_norm": 0.5658656084029214,
+      "learning_rate": 2.8050888760163265e-08,
+      "loss": 0.0237,
+      "step": 1289
+    },
+    {
+      "epoch": 1.934032983508246,
+      "grad_norm": 0.6527052339972337,
+      "learning_rate": 2.6819138231126695e-08,
+      "loss": 0.0366,
+      "step": 1290
+    },
+    {
+      "epoch": 1.9355322338830585,
+      "grad_norm": 0.5151970620828237,
+      "learning_rate": 2.5614969472506634e-08,
+      "loss": 0.021,
+      "step": 1291
+    },
+    {
+      "epoch": 1.937031484257871,
+      "grad_norm": 0.611780294225939,
+      "learning_rate": 2.4438389162746434e-08,
+      "loss": 0.0249,
+      "step": 1292
+    },
+    {
+      "epoch": 1.9385307346326837,
+      "grad_norm": 0.5146260847116298,
+      "learning_rate": 2.3289403827281287e-08,
+      "loss": 0.021,
+      "step": 1293
+    },
+    {
+      "epoch": 1.9400299850074962,
+      "grad_norm": 0.6142402064993351,
+      "learning_rate": 2.2168019838501032e-08,
+      "loss": 0.0304,
+      "step": 1294
+    },
+    {
+      "epoch": 1.9415292353823088,
+      "grad_norm": 0.546467632797473,
+      "learning_rate": 2.1074243415716288e-08,
+      "loss": 0.0232,
+      "step": 1295
+    },
+    {
+      "epoch": 1.9430284857571214,
+      "grad_norm": 0.5136495781373439,
+      "learning_rate": 2.0008080625124048e-08,
+      "loss": 0.0223,
+      "step": 1296
+    },
+    {
+      "epoch": 1.944527736131934,
+      "grad_norm": 0.5688098443265278,
+      "learning_rate": 1.896953737977103e-08,
+      "loss": 0.0317,
+      "step": 1297
+    },
+    {
+      "epoch": 1.9460269865067468,
+      "grad_norm": 0.5068500397906253,
+      "learning_rate": 1.7958619439524817e-08,
+      "loss": 0.026,
+      "step": 1298
+    },
+    {
+      "epoch": 1.9475262368815591,
+      "grad_norm": 0.6003154229822061,
+      "learning_rate": 1.6975332411040547e-08,
+      "loss": 0.0214,
+      "step": 1299
+    },
+    {
+      "epoch": 1.949025487256372,
+      "grad_norm": 0.5896087715043447,
+      "learning_rate": 1.601968174772761e-08,
+      "loss": 0.0204,
+      "step": 1300
+    },
+    {
+      "epoch": 1.9505247376311843,
+      "grad_norm": 0.5814533710403692,
+      "learning_rate": 1.5091672749723564e-08,
+      "loss": 0.0254,
+      "step": 1301
+    },
+    {
+      "epoch": 1.952023988005997,
+      "grad_norm": 0.5089209084795773,
+      "learning_rate": 1.4191310563860806e-08,
+      "loss": 0.0207,
+      "step": 1302
+    },
+    {
+      "epoch": 1.9535232383808094,
+      "grad_norm": 0.498698470094266,
+      "learning_rate": 1.331860018363995e-08,
+      "loss": 0.0237,
+      "step": 1303
+    },
+    {
+      "epoch": 1.9550224887556222,
+      "grad_norm": 0.5480657491115088,
+      "learning_rate": 1.2473546449203178e-08,
+      "loss": 0.0295,
+      "step": 1304
+    },
+    {
+      "epoch": 1.9565217391304348,
+      "grad_norm": 0.5170654578835844,
+      "learning_rate": 1.1656154047303691e-08,
+      "loss": 0.0243,
+      "step": 1305
+    },
+    {
+      "epoch": 1.9580209895052474,
+      "grad_norm": 0.4771355706396715,
+      "learning_rate": 1.0866427511285194e-08,
+      "loss": 0.0174,
+      "step": 1306
+    },
+    {
+      "epoch": 1.95952023988006,
+      "grad_norm": 0.4518660199672331,
+      "learning_rate": 1.0104371221050236e-08,
+      "loss": 0.0172,
+      "step": 1307
+    },
+    {
+      "epoch": 1.9610194902548725,
+      "grad_norm": 0.5579430726945314,
+      "learning_rate": 9.369989403041347e-09,
+      "loss": 0.0241,
+      "step": 1308
+    },
+    {
+      "epoch": 1.9625187406296851,
+      "grad_norm": 0.48004132638089153,
+      "learning_rate": 8.663286130216608e-09,
+      "loss": 0.0204,
+      "step": 1309
+    },
+    {
+      "epoch": 1.9640179910044977,
+      "grad_norm": 0.6454238315255135,
+      "learning_rate": 7.984265322023011e-09,
+      "loss": 0.022,
+      "step": 1310
+    },
+    {
+      "epoch": 1.9655172413793105,
+      "grad_norm": 0.7461742626412402,
+      "learning_rate": 7.332930744380906e-09,
+      "loss": 0.0466,
+      "step": 1311
+    },
+    {
+      "epoch": 1.9670164917541229,
+      "grad_norm": 0.5018617045456025,
+      "learning_rate": 6.709286009657368e-09,
+      "loss": 0.0186,
+      "step": 1312
+    },
+    {
+      "epoch": 1.9685157421289357,
+      "grad_norm": 0.8454654105529692,
+      "learning_rate": 6.1133345766511975e-09,
+      "loss": 0.0421,
+      "step": 1313
+    },
+    {
+      "epoch": 1.970014992503748,
+      "grad_norm": 0.6101310880218547,
+      "learning_rate": 5.5450797505690605e-09,
+      "loss": 0.0319,
+      "step": 1314
+    },
+    {
+      "epoch": 1.9715142428785608,
+      "grad_norm": 0.5091411615500949,
+      "learning_rate": 5.004524683011048e-09,
+      "loss": 0.0174,
+      "step": 1315
+    },
+    {
+      "epoch": 1.9730134932533732,
+      "grad_norm": 0.5429441137843405,
+      "learning_rate": 4.491672371950695e-09,
+      "loss": 0.0262,
+      "step": 1316
+    },
+    {
+      "epoch": 1.974512743628186,
+      "grad_norm": 0.5115361433528054,
+      "learning_rate": 4.0065256617199954e-09,
+      "loss": 0.0199,
+      "step": 1317
+    },
+    {
+      "epoch": 1.9760119940029985,
+      "grad_norm": 0.642760240384209,
+      "learning_rate": 3.5490872429910784e-09,
+      "loss": 0.0282,
+      "step": 1318
+    },
+    {
+      "epoch": 1.9775112443778111,
+      "grad_norm": 0.568367677658078,
+      "learning_rate": 3.119359652765108e-09,
+      "loss": 0.025,
+      "step": 1319
+    },
+    {
+      "epoch": 1.9790104947526237,
+      "grad_norm": 0.645328394447015,
+      "learning_rate": 2.7173452743550767e-09,
+      "loss": 0.0411,
+      "step": 1320
+    },
+    {
+      "epoch": 1.9805097451274363,
+      "grad_norm": 0.6301534537086467,
+      "learning_rate": 2.343046337374144e-09,
+      "loss": 0.0257,
+      "step": 1321
+    },
+    {
+      "epoch": 1.9820089955022488,
+      "grad_norm": 0.7090862636827612,
+      "learning_rate": 1.9964649177223184e-09,
+      "loss": 0.032,
+      "step": 1322
+    },
+    {
+      "epoch": 1.9835082458770614,
+      "grad_norm": 0.48695652165031617,
+      "learning_rate": 1.6776029375759062e-09,
+      "loss": 0.0247,
+      "step": 1323
+    },
+    {
+      "epoch": 1.9850074962518742,
+      "grad_norm": 0.8555752105920974,
+      "learning_rate": 1.386462165375857e-09,
+      "loss": 0.0425,
+      "step": 1324
+    },
+    {
+      "epoch": 1.9865067466266866,
+      "grad_norm": 0.5543979673104775,
+      "learning_rate": 1.1230442158188804e-09,
+      "loss": 0.0276,
+      "step": 1325
+    },
+    {
+      "epoch": 1.9880059970014994,
+      "grad_norm": 0.8902772466932038,
+      "learning_rate": 8.873505498474544e-10,
+      "loss": 0.035,
+      "step": 1326
+    },
+    {
+      "epoch": 1.9895052473763117,
+      "grad_norm": 0.6672747426600261,
+      "learning_rate": 6.793824746437194e-10,
+      "loss": 0.033,
+      "step": 1327
+    },
+    {
+      "epoch": 1.9910044977511245,
+      "grad_norm": 0.5024813582561877,
+      "learning_rate": 4.991411436189308e-10,
+      "loss": 0.0178,
+      "step": 1328
+    },
+    {
+      "epoch": 1.992503748125937,
+      "grad_norm": 0.6167215476991119,
+      "learning_rate": 3.466275564101285e-10,
+      "loss": 0.0282,
+      "step": 1329
+    },
+    {
+      "epoch": 1.9940029985007497,
+      "grad_norm": 0.6365546653903166,
+      "learning_rate": 2.2184255887403028e-10,
+      "loss": 0.0291,
+      "step": 1330
+    },
+    {
+      "epoch": 1.9955022488755623,
+      "grad_norm": 0.6660568471067422,
+      "learning_rate": 1.2478684308037115e-10,
+      "loss": 0.0299,
+      "step": 1331
+    },
+    {
+      "epoch": 1.9970014992503748,
+      "grad_norm": 0.629648261141815,
+      "learning_rate": 5.5460947310237435e-11,
+      "loss": 0.032,
+      "step": 1332
+    },
+    {
+      "epoch": 1.9985007496251874,
+      "grad_norm": 0.5095018130246789,
+      "learning_rate": 1.3865256052181252e-11,
+      "loss": 0.0191,
+      "step": 1333
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 0.40654188792460844,
+      "learning_rate": 0.0,
+      "loss": 0.0157,
+      "step": 1334
+    },
+    {
+      "epoch": 2.0,
+      "step": 1334,
+      "total_flos": 5978663976960.0,
+      "train_loss": 0.058373368264914585,
+      "train_runtime": 1018.1878,
+      "train_samples_per_second": 10.468,
+      "train_steps_per_second": 1.31
+    }
+  ],
+  "logging_steps": 1,
+  "max_steps": 1334,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2,
+  "save_steps": 5000,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 5978663976960.0,
+  "train_batch_size": 2,
+  "trial_name": null,
+  "trial_params": null
+}