diff --git "a/trainer_state.json" "b/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/trainer_state.json"
@@ -0,0 +1,24367 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 5.0,
+  "eval_steps": 500,
+  "global_step": 3475,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0014388489208633094,
+      "grad_norm": 5.232170786665442,
+      "learning_rate": 2.2988505747126437e-07,
+      "loss": 0.8858,
+      "step": 1
+    },
+    {
+      "epoch": 0.0028776978417266188,
+      "grad_norm": 5.189003189324628,
+      "learning_rate": 4.5977011494252875e-07,
+      "loss": 0.8824,
+      "step": 2
+    },
+    {
+      "epoch": 0.004316546762589928,
+      "grad_norm": 5.307893904128877,
+      "learning_rate": 6.896551724137931e-07,
+      "loss": 0.8989,
+      "step": 3
+    },
+    {
+      "epoch": 0.0057553956834532375,
+      "grad_norm": 5.24498684344306,
+      "learning_rate": 9.195402298850575e-07,
+      "loss": 0.8875,
+      "step": 4
+    },
+    {
+      "epoch": 0.007194244604316547,
+      "grad_norm": 5.139649808396179,
+      "learning_rate": 1.1494252873563219e-06,
+      "loss": 0.8715,
+      "step": 5
+    },
+    {
+      "epoch": 0.008633093525179856,
+      "grad_norm": 4.94061782589568,
+      "learning_rate": 1.3793103448275862e-06,
+      "loss": 0.8736,
+      "step": 6
+    },
+    {
+      "epoch": 0.010071942446043165,
+      "grad_norm": 4.757209414249211,
+      "learning_rate": 1.6091954022988506e-06,
+      "loss": 0.8562,
+      "step": 7
+    },
+    {
+      "epoch": 0.011510791366906475,
+      "grad_norm": 3.9956540897026795,
+      "learning_rate": 1.839080459770115e-06,
+      "loss": 0.8307,
+      "step": 8
+    },
+    {
+      "epoch": 0.012949640287769784,
+      "grad_norm": 3.7568289996074418,
+      "learning_rate": 2.0689655172413796e-06,
+      "loss": 0.8259,
+      "step": 9
+    },
+    {
+      "epoch": 0.014388489208633094,
+      "grad_norm": 2.192756078161156,
+      "learning_rate": 2.2988505747126437e-06,
+      "loss": 0.7865,
+      "step": 10
+    },
+    {
+      "epoch": 0.015827338129496403,
+      "grad_norm": 2.1187206799876277,
+      "learning_rate": 2.5287356321839083e-06,
+      "loss": 0.7909,
+      "step": 11
+    },
+    {
+      "epoch": 0.017266187050359712,
+      "grad_norm": 1.8660742674953652,
+      "learning_rate": 2.7586206896551725e-06,
+      "loss": 0.7736,
+      "step": 12
+    },
+    {
+      "epoch": 0.01870503597122302,
+      "grad_norm": 1.5692999774699359,
+      "learning_rate": 2.988505747126437e-06,
+      "loss": 0.7577,
+      "step": 13
+    },
+    {
+      "epoch": 0.02014388489208633,
+      "grad_norm": 3.0415774169771965,
+      "learning_rate": 3.2183908045977012e-06,
+      "loss": 0.7518,
+      "step": 14
+    },
+    {
+      "epoch": 0.02158273381294964,
+      "grad_norm": 3.216180658900517,
+      "learning_rate": 3.448275862068966e-06,
+      "loss": 0.748,
+      "step": 15
+    },
+    {
+      "epoch": 0.02302158273381295,
+      "grad_norm": 3.1235913271233002,
+      "learning_rate": 3.67816091954023e-06,
+      "loss": 0.7518,
+      "step": 16
+    },
+    {
+      "epoch": 0.02446043165467626,
+      "grad_norm": 2.9049126603009823,
+      "learning_rate": 3.908045977011495e-06,
+      "loss": 0.7397,
+      "step": 17
+    },
+    {
+      "epoch": 0.025899280575539568,
+      "grad_norm": 2.1000702530912303,
+      "learning_rate": 4.137931034482759e-06,
+      "loss": 0.6992,
+      "step": 18
+    },
+    {
+      "epoch": 0.027338129496402876,
+      "grad_norm": 1.8334607763736972,
+      "learning_rate": 4.367816091954023e-06,
+      "loss": 0.6987,
+      "step": 19
+    },
+    {
+      "epoch": 0.02877697841726619,
+      "grad_norm": 1.5129650569733848,
+      "learning_rate": 4.5977011494252875e-06,
+      "loss": 0.6819,
+      "step": 20
+    },
+    {
+      "epoch": 0.030215827338129497,
+      "grad_norm": 1.2049975073314985,
+      "learning_rate": 4.8275862068965525e-06,
+      "loss": 0.678,
+      "step": 21
+    },
+    {
+      "epoch": 0.031654676258992806,
+      "grad_norm": 1.1334124431583827,
+      "learning_rate": 5.057471264367817e-06,
+      "loss": 0.6714,
+      "step": 22
+    },
+    {
+      "epoch": 0.033093525179856115,
+      "grad_norm": 1.2261346776502773,
+      "learning_rate": 5.287356321839081e-06,
+      "loss": 0.6587,
+      "step": 23
+    },
+    {
+      "epoch": 0.034532374100719423,
+      "grad_norm": 1.1277895897758023,
+      "learning_rate": 5.517241379310345e-06,
+      "loss": 0.6492,
+      "step": 24
+    },
+    {
+      "epoch": 0.03597122302158273,
+      "grad_norm": 1.1047080413701726,
+      "learning_rate": 5.747126436781609e-06,
+      "loss": 0.639,
+      "step": 25
+    },
+    {
+      "epoch": 0.03741007194244604,
+      "grad_norm": 0.9207307510687444,
+      "learning_rate": 5.977011494252874e-06,
+      "loss": 0.6345,
+      "step": 26
+    },
+    {
+      "epoch": 0.03884892086330935,
+      "grad_norm": 0.7894647228308851,
+      "learning_rate": 6.206896551724138e-06,
+      "loss": 0.6311,
+      "step": 27
+    },
+    {
+      "epoch": 0.04028776978417266,
+      "grad_norm": 0.8631101241424584,
+      "learning_rate": 6.4367816091954025e-06,
+      "loss": 0.6275,
+      "step": 28
+    },
+    {
+      "epoch": 0.041726618705035974,
+      "grad_norm": 0.8182259655769767,
+      "learning_rate": 6.666666666666667e-06,
+      "loss": 0.6156,
+      "step": 29
+    },
+    {
+      "epoch": 0.04316546762589928,
+      "grad_norm": 0.646368696077582,
+      "learning_rate": 6.896551724137932e-06,
+      "loss": 0.6257,
+      "step": 30
+    },
+    {
+      "epoch": 0.04460431654676259,
+      "grad_norm": 0.49976707011608257,
+      "learning_rate": 7.126436781609196e-06,
+      "loss": 0.6176,
+      "step": 31
+    },
+    {
+      "epoch": 0.0460431654676259,
+      "grad_norm": 0.6799614136172442,
+      "learning_rate": 7.35632183908046e-06,
+      "loss": 0.6103,
+      "step": 32
+    },
+    {
+      "epoch": 0.04748201438848921,
+      "grad_norm": 0.6347758272440364,
+      "learning_rate": 7.586206896551724e-06,
+      "loss": 0.5904,
+      "step": 33
+    },
+    {
+      "epoch": 0.04892086330935252,
+      "grad_norm": 0.44059215896413373,
+      "learning_rate": 7.81609195402299e-06,
+      "loss": 0.6114,
+      "step": 34
+    },
+    {
+      "epoch": 0.050359712230215826,
+      "grad_norm": 0.4568865835639603,
+      "learning_rate": 8.045977011494253e-06,
+      "loss": 0.5947,
+      "step": 35
+    },
+    {
+      "epoch": 0.051798561151079135,
+      "grad_norm": 0.5224557606359925,
+      "learning_rate": 8.275862068965518e-06,
+      "loss": 0.5908,
+      "step": 36
+    },
+    {
+      "epoch": 0.053237410071942444,
+      "grad_norm": 0.5031264697052017,
+      "learning_rate": 8.505747126436782e-06,
+      "loss": 0.5873,
+      "step": 37
+    },
+    {
+      "epoch": 0.05467625899280575,
+      "grad_norm": 0.3916284756805459,
+      "learning_rate": 8.735632183908047e-06,
+      "loss": 0.5903,
+      "step": 38
+    },
+    {
+      "epoch": 0.05611510791366906,
+      "grad_norm": 0.3204316996379746,
+      "learning_rate": 8.965517241379312e-06,
+      "loss": 0.5723,
+      "step": 39
+    },
+    {
+      "epoch": 0.05755395683453238,
+      "grad_norm": 0.42456684891862123,
+      "learning_rate": 9.195402298850575e-06,
+      "loss": 0.6004,
+      "step": 40
+    },
+    {
+      "epoch": 0.058992805755395686,
+      "grad_norm": 0.42093007381873226,
+      "learning_rate": 9.42528735632184e-06,
+      "loss": 0.5726,
+      "step": 41
+    },
+    {
+      "epoch": 0.060431654676258995,
+      "grad_norm": 0.32619760323201125,
+      "learning_rate": 9.655172413793105e-06,
+      "loss": 0.5743,
+      "step": 42
+    },
+    {
+      "epoch": 0.0618705035971223,
+      "grad_norm": 0.36955472140670037,
+      "learning_rate": 9.885057471264368e-06,
+      "loss": 0.5657,
+      "step": 43
+    },
+    {
+      "epoch": 0.06330935251798561,
+      "grad_norm": 0.41646173590921515,
+      "learning_rate": 1.0114942528735633e-05,
+      "loss": 0.5826,
+      "step": 44
+    },
+    {
+      "epoch": 0.06474820143884892,
+      "grad_norm": 0.40117142453928417,
+      "learning_rate": 1.0344827586206898e-05,
+      "loss": 0.5789,
+      "step": 45
+    },
+    {
+      "epoch": 0.06618705035971223,
+      "grad_norm": 0.34911727085364064,
+      "learning_rate": 1.0574712643678162e-05,
+      "loss": 0.5632,
+      "step": 46
+    },
+    {
+      "epoch": 0.06762589928057554,
+      "grad_norm": 0.42852395741759985,
+      "learning_rate": 1.0804597701149427e-05,
+      "loss": 0.5581,
+      "step": 47
+    },
+    {
+      "epoch": 0.06906474820143885,
+      "grad_norm": 0.3203905738896688,
+      "learning_rate": 1.103448275862069e-05,
+      "loss": 0.559,
+      "step": 48
+    },
+    {
+      "epoch": 0.07050359712230216,
+      "grad_norm": 0.3396044368467807,
+      "learning_rate": 1.1264367816091955e-05,
+      "loss": 0.563,
+      "step": 49
+    },
+    {
+      "epoch": 0.07194244604316546,
+      "grad_norm": 0.3961432162322185,
+      "learning_rate": 1.1494252873563218e-05,
+      "loss": 0.5598,
+      "step": 50
+    },
+    {
+      "epoch": 0.07338129496402877,
+      "grad_norm": 0.24444535453022656,
+      "learning_rate": 1.1724137931034483e-05,
+      "loss": 0.5568,
+      "step": 51
+    },
+    {
+      "epoch": 0.07482014388489208,
+      "grad_norm": 0.34947999235158883,
+      "learning_rate": 1.1954022988505748e-05,
+      "loss": 0.5533,
+      "step": 52
+    },
+    {
+      "epoch": 0.07625899280575539,
+      "grad_norm": 0.3065382163330684,
+      "learning_rate": 1.2183908045977013e-05,
+      "loss": 0.551,
+      "step": 53
+    },
+    {
+      "epoch": 0.0776978417266187,
+      "grad_norm": 0.304266662513688,
+      "learning_rate": 1.2413793103448277e-05,
+      "loss": 0.5457,
+      "step": 54
+    },
+    {
+      "epoch": 0.07913669064748201,
+      "grad_norm": 0.23181630612110501,
+      "learning_rate": 1.2643678160919542e-05,
+      "loss": 0.5485,
+      "step": 55
+    },
+    {
+      "epoch": 0.08057553956834532,
+      "grad_norm": 0.2608490814351373,
+      "learning_rate": 1.2873563218390805e-05,
+      "loss": 0.5391,
+      "step": 56
+    },
+    {
+      "epoch": 0.08201438848920864,
+      "grad_norm": 0.22861678964733734,
+      "learning_rate": 1.310344827586207e-05,
+      "loss": 0.5488,
+      "step": 57
+    },
+    {
+      "epoch": 0.08345323741007195,
+      "grad_norm": 0.2387132398893568,
+      "learning_rate": 1.3333333333333333e-05,
+      "loss": 0.5497,
+      "step": 58
+    },
+    {
+      "epoch": 0.08489208633093526,
+      "grad_norm": 0.23731641935350856,
+      "learning_rate": 1.3563218390804598e-05,
+      "loss": 0.5351,
+      "step": 59
+    },
+    {
+      "epoch": 0.08633093525179857,
+      "grad_norm": 0.2517381403070896,
+      "learning_rate": 1.3793103448275863e-05,
+      "loss": 0.5499,
+      "step": 60
+    },
+    {
+      "epoch": 0.08776978417266187,
+      "grad_norm": 0.2490594423576649,
+      "learning_rate": 1.4022988505747128e-05,
+      "loss": 0.5339,
+      "step": 61
+    },
+    {
+      "epoch": 0.08920863309352518,
+      "grad_norm": 0.20571335373889837,
+      "learning_rate": 1.4252873563218392e-05,
+      "loss": 0.5518,
+      "step": 62
+    },
+    {
+      "epoch": 0.09064748201438849,
+      "grad_norm": 0.24952993436192078,
+      "learning_rate": 1.4482758620689657e-05,
+      "loss": 0.5457,
+      "step": 63
+    },
+    {
+      "epoch": 0.0920863309352518,
+      "grad_norm": 0.20944495579900255,
+      "learning_rate": 1.471264367816092e-05,
+      "loss": 0.5502,
+      "step": 64
+    },
+    {
+      "epoch": 0.09352517985611511,
+      "grad_norm": 0.23265496702724225,
+      "learning_rate": 1.4942528735632185e-05,
+      "loss": 0.5265,
+      "step": 65
+    },
+    {
+      "epoch": 0.09496402877697842,
+      "grad_norm": 0.2572714284263536,
+      "learning_rate": 1.5172413793103448e-05,
+      "loss": 0.5322,
+      "step": 66
+    },
+    {
+      "epoch": 0.09640287769784173,
+      "grad_norm": 0.2546598700811475,
+      "learning_rate": 1.540229885057471e-05,
+      "loss": 0.5359,
+      "step": 67
+    },
+    {
+      "epoch": 0.09784172661870504,
+      "grad_norm": 0.2670212576850694,
+      "learning_rate": 1.563218390804598e-05,
+      "loss": 0.5339,
+      "step": 68
+    },
+    {
+      "epoch": 0.09928057553956834,
+      "grad_norm": 0.3097272932505348,
+      "learning_rate": 1.586206896551724e-05,
+      "loss": 0.5363,
+      "step": 69
+    },
+    {
+      "epoch": 0.10071942446043165,
+      "grad_norm": 0.22597215512314667,
+      "learning_rate": 1.6091954022988507e-05,
+      "loss": 0.5366,
+      "step": 70
+    },
+    {
+      "epoch": 0.10215827338129496,
+      "grad_norm": 0.24896049028910877,
+      "learning_rate": 1.632183908045977e-05,
+      "loss": 0.5188,
+      "step": 71
+    },
+    {
+      "epoch": 0.10359712230215827,
+      "grad_norm": 0.22038628808523028,
+      "learning_rate": 1.6551724137931037e-05,
+      "loss": 0.5307,
+      "step": 72
+    },
+    {
+      "epoch": 0.10503597122302158,
+      "grad_norm": 0.2642384824187733,
+      "learning_rate": 1.6781609195402298e-05,
+      "loss": 0.517,
+      "step": 73
+    },
+    {
+      "epoch": 0.10647482014388489,
+      "grad_norm": 0.31600117407722406,
+      "learning_rate": 1.7011494252873563e-05,
+      "loss": 0.5289,
+      "step": 74
+    },
+    {
+      "epoch": 0.1079136690647482,
+      "grad_norm": 0.42116471977610626,
+      "learning_rate": 1.7241379310344828e-05,
+      "loss": 0.5234,
+      "step": 75
+    },
+    {
+      "epoch": 0.1093525179856115,
+      "grad_norm": 0.7330810250123087,
+      "learning_rate": 1.7471264367816093e-05,
+      "loss": 0.524,
+      "step": 76
+    },
+    {
+      "epoch": 0.11079136690647481,
+      "grad_norm": 1.1433562597711888,
+      "learning_rate": 1.770114942528736e-05,
+      "loss": 0.5339,
+      "step": 77
+    },
+    {
+      "epoch": 0.11223021582733812,
+      "grad_norm": 1.0151631179784177,
+      "learning_rate": 1.7931034482758623e-05,
+      "loss": 0.5309,
+      "step": 78
+    },
+    {
+      "epoch": 0.11366906474820145,
+      "grad_norm": 0.4383599009940003,
+      "learning_rate": 1.8160919540229885e-05,
+      "loss": 0.5247,
+      "step": 79
+    },
+    {
+      "epoch": 0.11510791366906475,
+      "grad_norm": 0.6380284405492358,
+      "learning_rate": 1.839080459770115e-05,
+      "loss": 0.5286,
+      "step": 80
+    },
+    {
+      "epoch": 0.11654676258992806,
+      "grad_norm": 0.9433211526077079,
+      "learning_rate": 1.8620689655172415e-05,
+      "loss": 0.526,
+      "step": 81
+    },
+    {
+      "epoch": 0.11798561151079137,
+      "grad_norm": 0.6454820817885937,
+      "learning_rate": 1.885057471264368e-05,
+      "loss": 0.5184,
+      "step": 82
+    },
+    {
+      "epoch": 0.11942446043165468,
+      "grad_norm": 0.6186938094843576,
+      "learning_rate": 1.908045977011494e-05,
+      "loss": 0.5373,
+      "step": 83
+    },
+    {
+      "epoch": 0.12086330935251799,
+      "grad_norm": 0.8222985848657414,
+      "learning_rate": 1.931034482758621e-05,
+      "loss": 0.5133,
+      "step": 84
+    },
+    {
+      "epoch": 0.1223021582733813,
+      "grad_norm": 0.6373941213971877,
+      "learning_rate": 1.9540229885057475e-05,
+      "loss": 0.5225,
+      "step": 85
+    },
+    {
+      "epoch": 0.1237410071942446,
+      "grad_norm": 0.5803977667901652,
+      "learning_rate": 1.9770114942528737e-05,
+      "loss": 0.5109,
+      "step": 86
+    },
+    {
+      "epoch": 0.1251798561151079,
+      "grad_norm": 0.7300312340518113,
+      "learning_rate": 2e-05,
+      "loss": 0.5161,
+      "step": 87
+    },
+    {
+      "epoch": 0.12661870503597122,
+      "grad_norm": 0.49245807578355527,
+      "learning_rate": 2.0229885057471267e-05,
+      "loss": 0.5243,
+      "step": 88
+    },
+    {
+      "epoch": 0.12805755395683452,
+      "grad_norm": 0.5783075683156449,
+      "learning_rate": 2.0459770114942528e-05,
+      "loss": 0.5143,
+      "step": 89
+    },
+    {
+      "epoch": 0.12949640287769784,
+      "grad_norm": 0.5790626254010335,
+      "learning_rate": 2.0689655172413797e-05,
+      "loss": 0.5123,
+      "step": 90
+    },
+    {
+      "epoch": 0.13093525179856116,
+      "grad_norm": 0.6617277361345342,
+      "learning_rate": 2.0919540229885058e-05,
+      "loss": 0.5156,
+      "step": 91
+    },
+    {
+      "epoch": 0.13237410071942446,
+      "grad_norm": 1.0316999106593459,
+      "learning_rate": 2.1149425287356323e-05,
+      "loss": 0.5135,
+      "step": 92
+    },
+    {
+      "epoch": 0.13381294964028778,
+      "grad_norm": 1.1145362643699137,
+      "learning_rate": 2.1379310344827585e-05,
+      "loss": 0.5224,
+      "step": 93
+    },
+    {
+      "epoch": 0.13525179856115108,
+      "grad_norm": 0.8549443929767417,
+      "learning_rate": 2.1609195402298853e-05,
+      "loss": 0.5157,
+      "step": 94
+    },
+    {
+      "epoch": 0.1366906474820144,
+      "grad_norm": 1.363906719878284,
+      "learning_rate": 2.183908045977012e-05,
+      "loss": 0.5135,
+      "step": 95
+    },
+    {
+      "epoch": 0.1381294964028777,
+      "grad_norm": 0.909276356364517,
+      "learning_rate": 2.206896551724138e-05,
+      "loss": 0.518,
+      "step": 96
+    },
+    {
+      "epoch": 0.13956834532374102,
+      "grad_norm": 1.306827176055507,
+      "learning_rate": 2.229885057471265e-05,
+      "loss": 0.5133,
+      "step": 97
+    },
+    {
+      "epoch": 0.1410071942446043,
+      "grad_norm": 0.7284051970530677,
+      "learning_rate": 2.252873563218391e-05,
+      "loss": 0.5141,
+      "step": 98
+    },
+    {
+      "epoch": 0.14244604316546763,
+      "grad_norm": 1.4690961991545168,
+      "learning_rate": 2.2758620689655175e-05,
+      "loss": 0.5155,
+      "step": 99
+    },
+    {
+      "epoch": 0.14388489208633093,
+      "grad_norm": 0.8564984095806956,
+      "learning_rate": 2.2988505747126437e-05,
+      "loss": 0.514,
+      "step": 100
+    },
+    {
+      "epoch": 0.14532374100719425,
+      "grad_norm": 1.58868035636525,
+      "learning_rate": 2.3218390804597705e-05,
+      "loss": 0.5214,
+      "step": 101
+    },
+    {
+      "epoch": 0.14676258992805755,
+      "grad_norm": 1.1821772704261104,
+      "learning_rate": 2.3448275862068967e-05,
+      "loss": 0.5068,
+      "step": 102
+    },
+    {
+      "epoch": 0.14820143884892087,
+      "grad_norm": 1.3970183237937182,
+      "learning_rate": 2.367816091954023e-05,
+      "loss": 0.5114,
+      "step": 103
+    },
+    {
+      "epoch": 0.14964028776978416,
+      "grad_norm": 1.3856255061438907,
+      "learning_rate": 2.3908045977011497e-05,
+      "loss": 0.5194,
+      "step": 104
+    },
+    {
+      "epoch": 0.1510791366906475,
+      "grad_norm": 1.0199674097168947,
+      "learning_rate": 2.413793103448276e-05,
+      "loss": 0.5106,
+      "step": 105
+    },
+    {
+      "epoch": 0.15251798561151078,
+      "grad_norm": 1.2300452127883577,
+      "learning_rate": 2.4367816091954027e-05,
+      "loss": 0.5078,
+      "step": 106
+    },
+    {
+      "epoch": 0.1539568345323741,
+      "grad_norm": 0.9037488204666687,
+      "learning_rate": 2.4597701149425288e-05,
+      "loss": 0.5114,
+      "step": 107
+    },
+    {
+      "epoch": 0.1553956834532374,
+      "grad_norm": 0.9619214892056264,
+      "learning_rate": 2.4827586206896553e-05,
+      "loss": 0.5088,
+      "step": 108
+    },
+    {
+      "epoch": 0.15683453237410072,
+      "grad_norm": 0.7461287597995564,
+      "learning_rate": 2.5057471264367815e-05,
+      "loss": 0.4968,
+      "step": 109
+    },
+    {
+      "epoch": 0.15827338129496402,
+      "grad_norm": 0.7472151674844343,
+      "learning_rate": 2.5287356321839083e-05,
+      "loss": 0.5027,
+      "step": 110
+    },
+    {
+      "epoch": 0.15971223021582734,
+      "grad_norm": 0.7111109009469494,
+      "learning_rate": 2.551724137931035e-05,
+      "loss": 0.5089,
+      "step": 111
+    },
+    {
+      "epoch": 0.16115107913669063,
+      "grad_norm": 0.6338520803535057,
+      "learning_rate": 2.574712643678161e-05,
+      "loss": 0.5154,
+      "step": 112
+    },
+    {
+      "epoch": 0.16258992805755396,
+      "grad_norm": 0.675370574100815,
+      "learning_rate": 2.597701149425288e-05,
+      "loss": 0.5195,
+      "step": 113
+    },
+    {
+      "epoch": 0.16402877697841728,
+      "grad_norm": 0.7298286160968676,
+      "learning_rate": 2.620689655172414e-05,
+      "loss": 0.5111,
+      "step": 114
+    },
+    {
+      "epoch": 0.16546762589928057,
+      "grad_norm": 0.4997081048522759,
+      "learning_rate": 2.6436781609195405e-05,
+      "loss": 0.5109,
+      "step": 115
+    },
+    {
+      "epoch": 0.1669064748201439,
+      "grad_norm": 0.5721967528687938,
+      "learning_rate": 2.6666666666666667e-05,
+      "loss": 0.4977,
+      "step": 116
+    },
+    {
+      "epoch": 0.1683453237410072,
+      "grad_norm": 0.621106185128584,
+      "learning_rate": 2.6896551724137935e-05,
+      "loss": 0.4996,
+      "step": 117
+    },
+    {
+      "epoch": 0.1697841726618705,
+      "grad_norm": 0.46702811394454086,
+      "learning_rate": 2.7126436781609197e-05,
+      "loss": 0.5017,
+      "step": 118
+    },
+    {
+      "epoch": 0.1712230215827338,
+      "grad_norm": 0.6367448127200948,
+      "learning_rate": 2.735632183908046e-05,
+      "loss": 0.5095,
+      "step": 119
+    },
+    {
+      "epoch": 0.17266187050359713,
+      "grad_norm": 0.8388646694034495,
+      "learning_rate": 2.7586206896551727e-05,
+      "loss": 0.5082,
+      "step": 120
+    },
+    {
+      "epoch": 0.17410071942446043,
+      "grad_norm": 0.8221549454299817,
+      "learning_rate": 2.781609195402299e-05,
+      "loss": 0.5081,
+      "step": 121
+    },
+    {
+      "epoch": 0.17553956834532375,
+      "grad_norm": 1.0718933842162548,
+      "learning_rate": 2.8045977011494257e-05,
+      "loss": 0.5076,
+      "step": 122
+    },
+    {
+      "epoch": 0.17697841726618704,
+      "grad_norm": 1.1573932937368792,
+      "learning_rate": 2.8275862068965518e-05,
+      "loss": 0.5006,
+      "step": 123
+    },
+    {
+      "epoch": 0.17841726618705037,
+      "grad_norm": 0.8462900690961672,
+      "learning_rate": 2.8505747126436783e-05,
+      "loss": 0.4994,
+      "step": 124
+    },
+    {
+      "epoch": 0.17985611510791366,
+      "grad_norm": 0.7461924145603112,
+      "learning_rate": 2.8735632183908045e-05,
+      "loss": 0.4955,
+      "step": 125
+    },
+    {
+      "epoch": 0.18129496402877698,
+      "grad_norm": 0.903532518218726,
+      "learning_rate": 2.8965517241379313e-05,
+      "loss": 0.4967,
+      "step": 126
+    },
+    {
+      "epoch": 0.18273381294964028,
+      "grad_norm": 1.1408053895067267,
+      "learning_rate": 2.919540229885058e-05,
+      "loss": 0.5088,
+      "step": 127
+    },
+    {
+      "epoch": 0.1841726618705036,
+      "grad_norm": 1.0551621554288855,
+      "learning_rate": 2.942528735632184e-05,
+      "loss": 0.5017,
+      "step": 128
+    },
+    {
+      "epoch": 0.1856115107913669,
+      "grad_norm": 1.0963160001943373,
+      "learning_rate": 2.965517241379311e-05,
+      "loss": 0.5044,
+      "step": 129
+    },
+    {
+      "epoch": 0.18705035971223022,
+      "grad_norm": 0.9073606740947865,
+      "learning_rate": 2.988505747126437e-05,
+      "loss": 0.5071,
+      "step": 130
+    },
+    {
+      "epoch": 0.1884892086330935,
+      "grad_norm": 0.9264114085091071,
+      "learning_rate": 3.0114942528735635e-05,
+      "loss": 0.5064,
+      "step": 131
+    },
+    {
+      "epoch": 0.18992805755395684,
+      "grad_norm": 1.0371785802095712,
+      "learning_rate": 3.0344827586206897e-05,
+      "loss": 0.497,
+      "step": 132
+    },
+    {
+      "epoch": 0.19136690647482013,
+      "grad_norm": 0.9909533495251422,
+      "learning_rate": 3.057471264367816e-05,
+      "loss": 0.4964,
+      "step": 133
+    },
+    {
+      "epoch": 0.19280575539568345,
+      "grad_norm": 1.1174578586845554,
+      "learning_rate": 3.080459770114942e-05,
+      "loss": 0.5089,
+      "step": 134
+    },
+    {
+      "epoch": 0.19424460431654678,
+      "grad_norm": 0.9005843732910414,
+      "learning_rate": 3.103448275862069e-05,
+      "loss": 0.5,
+      "step": 135
+    },
+    {
+      "epoch": 0.19568345323741007,
+      "grad_norm": 0.8643748894738493,
+      "learning_rate": 3.126436781609196e-05,
+      "loss": 0.4979,
+      "step": 136
+    },
+    {
+      "epoch": 0.1971223021582734,
+      "grad_norm": 0.7844564240356566,
+      "learning_rate": 3.149425287356322e-05,
+      "loss": 0.4914,
+      "step": 137
+    },
+    {
+      "epoch": 0.1985611510791367,
+      "grad_norm": 0.8420107305727663,
+      "learning_rate": 3.172413793103448e-05,
+      "loss": 0.4989,
+      "step": 138
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.8010607454745727,
+      "learning_rate": 3.195402298850575e-05,
+      "loss": 0.4972,
+      "step": 139
+    },
+    {
+      "epoch": 0.2014388489208633,
+      "grad_norm": 0.7536572747354274,
+      "learning_rate": 3.218390804597701e-05,
+      "loss": 0.4947,
+      "step": 140
+    },
+    {
+      "epoch": 0.20287769784172663,
+      "grad_norm": 0.7581207765273857,
+      "learning_rate": 3.2413793103448275e-05,
+      "loss": 0.4917,
+      "step": 141
+    },
+    {
+      "epoch": 0.20431654676258992,
+      "grad_norm": 0.9658796512775292,
+      "learning_rate": 3.264367816091954e-05,
+      "loss": 0.491,
+      "step": 142
+    },
+    {
+      "epoch": 0.20575539568345325,
+      "grad_norm": 1.37702441636452,
+      "learning_rate": 3.287356321839081e-05,
+      "loss": 0.493,
+      "step": 143
+    },
+    {
+      "epoch": 0.20719424460431654,
+      "grad_norm": 0.5256931079204877,
+      "learning_rate": 3.310344827586207e-05,
+      "loss": 0.4887,
+      "step": 144
+    },
+    {
+      "epoch": 0.20863309352517986,
+      "grad_norm": 0.9428348881298956,
+      "learning_rate": 3.3333333333333335e-05,
+      "loss": 0.4957,
+      "step": 145
+    },
+    {
+      "epoch": 0.21007194244604316,
+      "grad_norm": 1.3016157308761458,
+      "learning_rate": 3.3563218390804597e-05,
+      "loss": 0.4998,
+      "step": 146
+    },
+    {
+      "epoch": 0.21151079136690648,
+      "grad_norm": 0.613818512646567,
+      "learning_rate": 3.3793103448275865e-05,
+      "loss": 0.4979,
+      "step": 147
+    },
+    {
+      "epoch": 0.21294964028776978,
+      "grad_norm": 1.009800309668571,
+      "learning_rate": 3.4022988505747127e-05,
+      "loss": 0.4947,
+      "step": 148
+    },
+    {
+      "epoch": 0.2143884892086331,
+      "grad_norm": 0.9624925516955802,
+      "learning_rate": 3.4252873563218395e-05,
+      "loss": 0.4969,
+      "step": 149
+    },
+    {
+      "epoch": 0.2158273381294964,
+      "grad_norm": 0.8825107931109423,
+      "learning_rate": 3.4482758620689657e-05,
+      "loss": 0.4992,
+      "step": 150
+    },
+    {
+      "epoch": 0.21726618705035972,
+      "grad_norm": 1.1975991844403961,
+      "learning_rate": 3.4712643678160925e-05,
+      "loss": 0.5014,
+      "step": 151
+    },
+    {
+      "epoch": 0.218705035971223,
+      "grad_norm": 0.9023368561489841,
+      "learning_rate": 3.4942528735632187e-05,
+      "loss": 0.504,
+      "step": 152
+    },
+    {
+      "epoch": 0.22014388489208633,
+      "grad_norm": 1.4831068778936636,
+      "learning_rate": 3.517241379310345e-05,
+      "loss": 0.4937,
+      "step": 153
+    },
+    {
+      "epoch": 0.22158273381294963,
+      "grad_norm": 0.7609441164303112,
+      "learning_rate": 3.540229885057472e-05,
+      "loss": 0.4942,
+      "step": 154
+    },
+    {
+      "epoch": 0.22302158273381295,
+      "grad_norm": 1.199593587875097,
+      "learning_rate": 3.563218390804598e-05,
+      "loss": 0.4845,
+      "step": 155
+    },
+    {
+      "epoch": 0.22446043165467625,
+      "grad_norm": 0.7396075741999119,
+      "learning_rate": 3.586206896551725e-05,
+      "loss": 0.4966,
+      "step": 156
+    },
+    {
+      "epoch": 0.22589928057553957,
+      "grad_norm": 0.9778734659781795,
+      "learning_rate": 3.609195402298851e-05,
+      "loss": 0.498,
+      "step": 157
+    },
+    {
+      "epoch": 0.2273381294964029,
+      "grad_norm": 1.185683620829216,
+      "learning_rate": 3.632183908045977e-05,
+      "loss": 0.4945,
+      "step": 158
+    },
+    {
+      "epoch": 0.22877697841726619,
+      "grad_norm": 0.6786828186846136,
+      "learning_rate": 3.655172413793104e-05,
+      "loss": 0.5011,
+      "step": 159
+    },
+    {
+      "epoch": 0.2302158273381295,
+      "grad_norm": 0.9352379928924268,
+      "learning_rate": 3.67816091954023e-05,
+      "loss": 0.4918,
+      "step": 160
+    },
+    {
+      "epoch": 0.2316546762589928,
+      "grad_norm": 0.8893597905763662,
+      "learning_rate": 3.701149425287357e-05,
+      "loss": 0.4989,
+      "step": 161
+    },
+    {
+      "epoch": 0.23309352517985613,
+      "grad_norm": 0.7629180786629474,
+      "learning_rate": 3.724137931034483e-05,
+      "loss": 0.4937,
+      "step": 162
+    },
+    {
+      "epoch": 0.23453237410071942,
+      "grad_norm": 0.9092757267696973,
+      "learning_rate": 3.74712643678161e-05,
+      "loss": 0.5025,
+      "step": 163
+    },
+    {
+      "epoch": 0.23597122302158274,
+      "grad_norm": 1.0300863068218389,
+      "learning_rate": 3.770114942528736e-05,
+      "loss": 0.4894,
+      "step": 164
+    },
+    {
+      "epoch": 0.23741007194244604,
+      "grad_norm": 1.4373845829394083,
+      "learning_rate": 3.793103448275862e-05,
+      "loss": 0.5019,
+      "step": 165
+    },
+    {
+      "epoch": 0.23884892086330936,
+      "grad_norm": 1.1369279239361678,
+      "learning_rate": 3.816091954022988e-05,
+      "loss": 0.4941,
+      "step": 166
+    },
+    {
+      "epoch": 0.24028776978417266,
+      "grad_norm": 1.2368590685738565,
+      "learning_rate": 3.839080459770115e-05,
+      "loss": 0.5106,
+      "step": 167
+    },
+    {
+      "epoch": 0.24172661870503598,
+      "grad_norm": 1.2470456300859118,
+      "learning_rate": 3.862068965517242e-05,
+      "loss": 0.4916,
+      "step": 168
+    },
+    {
+      "epoch": 0.24316546762589927,
+      "grad_norm": 1.4542567342620654,
+      "learning_rate": 3.885057471264368e-05,
+      "loss": 0.4971,
+      "step": 169
+    },
+    {
+      "epoch": 0.2446043165467626,
+      "grad_norm": 0.778853934850899,
+      "learning_rate": 3.908045977011495e-05,
+      "loss": 0.4923,
+      "step": 170
+    },
+    {
+      "epoch": 0.2460431654676259,
+      "grad_norm": 1.7833309124005472,
+      "learning_rate": 3.931034482758621e-05,
+      "loss": 0.4958,
+      "step": 171
+    },
+    {
+      "epoch": 0.2474820143884892,
+      "grad_norm": 0.93242302095893,
+      "learning_rate": 3.954022988505747e-05,
+      "loss": 0.4912,
+      "step": 172
+    },
+    {
+      "epoch": 0.2489208633093525,
+      "grad_norm": 2.12123455399472,
+      "learning_rate": 3.9770114942528735e-05,
+      "loss": 0.4964,
+      "step": 173
+    },
+    {
+      "epoch": 0.2503597122302158,
+      "grad_norm": 1.7833482803959417,
+      "learning_rate": 4e-05,
+      "loss": 0.4945,
+      "step": 174
+    },
+    {
+      "epoch": 0.2517985611510791,
+      "grad_norm": 1.3701247359533977,
+      "learning_rate": 4.022988505747127e-05,
+      "loss": 0.498,
+      "step": 175
+    },
+    {
+      "epoch": 0.25323741007194245,
+      "grad_norm": 1.5349346547437832,
+      "learning_rate": 4.045977011494253e-05,
+      "loss": 0.4927,
+      "step": 176
+    },
+    {
+      "epoch": 0.25467625899280577,
+      "grad_norm": 1.1074123983244828,
+      "learning_rate": 4.0689655172413795e-05,
+      "loss": 0.502,
+      "step": 177
+    },
+    {
+      "epoch": 0.25611510791366904,
+      "grad_norm": 1.7093858086487201,
+      "learning_rate": 4.0919540229885057e-05,
+      "loss": 0.4954,
+      "step": 178
+    },
+    {
+      "epoch": 0.25755395683453236,
+      "grad_norm": 1.1468166105078421,
+      "learning_rate": 4.1149425287356325e-05,
+      "loss": 0.5036,
+      "step": 179
+    },
+    {
+      "epoch": 0.2589928057553957,
+      "grad_norm": 1.771636805784584,
+      "learning_rate": 4.137931034482759e-05,
+      "loss": 0.4949,
+      "step": 180
+    },
+    {
+      "epoch": 0.260431654676259,
+      "grad_norm": 1.2476683208712824,
+      "learning_rate": 4.160919540229885e-05,
+      "loss": 0.4957,
+      "step": 181
+    },
+    {
+      "epoch": 0.26187050359712233,
+      "grad_norm": 1.8081462170443292,
+      "learning_rate": 4.1839080459770117e-05,
+      "loss": 0.5002,
+      "step": 182
+    },
+    {
+      "epoch": 0.2633093525179856,
+      "grad_norm": 1.1150500115400306,
+      "learning_rate": 4.2068965517241385e-05,
+      "loss": 0.4933,
+      "step": 183
+    },
+    {
+      "epoch": 0.2647482014388489,
+      "grad_norm": 2.093915191575719,
+      "learning_rate": 4.2298850574712647e-05,
+      "loss": 0.5128,
+      "step": 184
+    },
+    {
+      "epoch": 0.26618705035971224,
+      "grad_norm": 1.600106959664035,
+      "learning_rate": 4.2528735632183915e-05,
+      "loss": 0.5009,
+      "step": 185
+    },
+    {
+      "epoch": 0.26762589928057556,
+      "grad_norm": 1.8336012966254829,
+      "learning_rate": 4.275862068965517e-05,
+      "loss": 0.4964,
+      "step": 186
+    },
+    {
+      "epoch": 0.26906474820143883,
+      "grad_norm": 1.5245854964800294,
+      "learning_rate": 4.298850574712644e-05,
+      "loss": 0.5077,
+      "step": 187
+    },
+    {
+      "epoch": 0.27050359712230215,
+      "grad_norm": 1.767595562772753,
+      "learning_rate": 4.321839080459771e-05,
+      "loss": 0.5076,
+      "step": 188
+    },
+    {
+      "epoch": 0.2719424460431655,
+      "grad_norm": 1.5139363631737155,
+      "learning_rate": 4.344827586206897e-05,
+      "loss": 0.5029,
+      "step": 189
+    },
+    {
+      "epoch": 0.2733812949640288,
+      "grad_norm": 1.542045683799105,
+      "learning_rate": 4.367816091954024e-05,
+      "loss": 0.5013,
+      "step": 190
+    },
+    {
+      "epoch": 0.27482014388489207,
+      "grad_norm": 1.3381200993387352,
+      "learning_rate": 4.39080459770115e-05,
+      "loss": 0.4874,
+      "step": 191
+    },
+    {
+      "epoch": 0.2762589928057554,
+      "grad_norm": 1.5390253779260872,
+      "learning_rate": 4.413793103448276e-05,
+      "loss": 0.494,
+      "step": 192
+    },
+    {
+      "epoch": 0.2776978417266187,
+      "grad_norm": 1.0552446418090449,
+      "learning_rate": 4.436781609195403e-05,
+      "loss": 0.4863,
+      "step": 193
+    },
+    {
+      "epoch": 0.27913669064748203,
+      "grad_norm": 1.77100245437121,
+      "learning_rate": 4.45977011494253e-05,
+      "loss": 0.5031,
+      "step": 194
+    },
+    {
+      "epoch": 0.2805755395683453,
+      "grad_norm": 1.4851400864755933,
+      "learning_rate": 4.482758620689655e-05,
+      "loss": 0.4962,
+      "step": 195
+    },
+    {
+      "epoch": 0.2820143884892086,
+      "grad_norm": 1.4291720970659931,
+      "learning_rate": 4.505747126436782e-05,
+      "loss": 0.49,
+      "step": 196
+    },
+    {
+      "epoch": 0.28345323741007195,
+      "grad_norm": 1.1639239859473705,
+      "learning_rate": 4.528735632183908e-05,
+      "loss": 0.5018,
+      "step": 197
+    },
+    {
+      "epoch": 0.28489208633093527,
+      "grad_norm": 1.413061827929404,
+      "learning_rate": 4.551724137931035e-05,
+      "loss": 0.4887,
+      "step": 198
+    },
+    {
+      "epoch": 0.28633093525179854,
+      "grad_norm": 0.998532759658418,
+      "learning_rate": 4.574712643678162e-05,
+      "loss": 0.488,
+      "step": 199
+    },
+    {
+      "epoch": 0.28776978417266186,
+      "grad_norm": 1.7513659096171872,
+      "learning_rate": 4.597701149425287e-05,
+      "loss": 0.4903,
+      "step": 200
+    },
+    {
+      "epoch": 0.2892086330935252,
+      "grad_norm": 1.3216333801978293,
+      "learning_rate": 4.620689655172414e-05,
+      "loss": 0.4855,
+      "step": 201
+    },
+    {
+      "epoch": 0.2906474820143885,
+      "grad_norm": 1.4684021935280545,
+      "learning_rate": 4.643678160919541e-05,
+      "loss": 0.4872,
+      "step": 202
+    },
+    {
+      "epoch": 0.2920863309352518,
+      "grad_norm": 1.2952384457839685,
+      "learning_rate": 4.666666666666667e-05,
+      "loss": 0.4888,
+      "step": 203
+    },
+    {
+      "epoch": 0.2935251798561151,
+      "grad_norm": 1.3990150221114375,
+      "learning_rate": 4.689655172413793e-05,
+      "loss": 0.4974,
+      "step": 204
+    },
+    {
+      "epoch": 0.2949640287769784,
+      "grad_norm": 0.9876775520718227,
+      "learning_rate": 4.7126436781609195e-05,
+      "loss": 0.488,
+      "step": 205
+    },
+    {
+      "epoch": 0.29640287769784174,
+      "grad_norm": 1.36524312409901,
+      "learning_rate": 4.735632183908046e-05,
+      "loss": 0.4879,
+      "step": 206
+    },
+    {
+      "epoch": 0.29784172661870506,
+      "grad_norm": 1.1149550671155792,
+      "learning_rate": 4.758620689655173e-05,
+      "loss": 0.5,
+      "step": 207
+    },
+    {
+      "epoch": 0.2992805755395683,
+      "grad_norm": 1.4692418153023588,
+      "learning_rate": 4.781609195402299e-05,
+      "loss": 0.4978,
+      "step": 208
+    },
+    {
+      "epoch": 0.30071942446043165,
+      "grad_norm": 1.195808176437362,
+      "learning_rate": 4.8045977011494255e-05,
+      "loss": 0.5006,
+      "step": 209
+    },
+    {
+      "epoch": 0.302158273381295,
+      "grad_norm": 1.3168347564869889,
+      "learning_rate": 4.827586206896552e-05,
+      "loss": 0.4878,
+      "step": 210
+    },
+    {
+      "epoch": 0.3035971223021583,
+      "grad_norm": 1.0935788325080322,
+      "learning_rate": 4.8505747126436785e-05,
+      "loss": 0.4933,
+      "step": 211
+    },
+    {
+      "epoch": 0.30503597122302156,
+      "grad_norm": 1.1314020714687931,
+      "learning_rate": 4.873563218390805e-05,
+      "loss": 0.478,
+      "step": 212
+    },
+    {
+      "epoch": 0.3064748201438849,
+      "grad_norm": 0.9368152176738559,
+      "learning_rate": 4.896551724137931e-05,
+      "loss": 0.4969,
+      "step": 213
+    },
+    {
+      "epoch": 0.3079136690647482,
+      "grad_norm": 1.0341123301704138,
+      "learning_rate": 4.9195402298850577e-05,
+      "loss": 0.4874,
+      "step": 214
+    },
+    {
+      "epoch": 0.30935251798561153,
+      "grad_norm": 0.8357582951933543,
+      "learning_rate": 4.9425287356321845e-05,
+      "loss": 0.4847,
+      "step": 215
+    },
+    {
+      "epoch": 0.3107913669064748,
+      "grad_norm": 0.7732084800276675,
+      "learning_rate": 4.9655172413793107e-05,
+      "loss": 0.4867,
+      "step": 216
+    },
+    {
+      "epoch": 0.3122302158273381,
+      "grad_norm": 0.7720204405423883,
+      "learning_rate": 4.9885057471264375e-05,
+      "loss": 0.4852,
+      "step": 217
+    },
+    {
+      "epoch": 0.31366906474820144,
+      "grad_norm": 0.7119262037670632,
+      "learning_rate": 5.011494252873563e-05,
+      "loss": 0.4851,
+      "step": 218
+    },
+    {
+      "epoch": 0.31510791366906477,
+      "grad_norm": 0.7681186415638412,
+      "learning_rate": 5.03448275862069e-05,
+      "loss": 0.4795,
+      "step": 219
+    },
+    {
+      "epoch": 0.31654676258992803,
+      "grad_norm": 1.010887539368242,
+      "learning_rate": 5.057471264367817e-05,
+      "loss": 0.4876,
+      "step": 220
+    },
+    {
+      "epoch": 0.31798561151079136,
+      "grad_norm": 1.1684469481139526,
+      "learning_rate": 5.0804597701149435e-05,
+      "loss": 0.4816,
+      "step": 221
+    },
+    {
+      "epoch": 0.3194244604316547,
+      "grad_norm": 1.104313919572175,
+      "learning_rate": 5.10344827586207e-05,
+      "loss": 0.4789,
+      "step": 222
+    },
+    {
+      "epoch": 0.320863309352518,
+      "grad_norm": 1.259432298938836,
+      "learning_rate": 5.126436781609196e-05,
+      "loss": 0.4852,
+      "step": 223
+    },
+    {
+      "epoch": 0.32230215827338127,
+      "grad_norm": 0.9975875091199469,
+      "learning_rate": 5.149425287356322e-05,
+      "loss": 0.4795,
+      "step": 224
+    },
+    {
+      "epoch": 0.3237410071942446,
+      "grad_norm": 1.179464731042854,
+      "learning_rate": 5.172413793103449e-05,
+      "loss": 0.4797,
+      "step": 225
+    },
+    {
+      "epoch": 0.3251798561151079,
+      "grad_norm": 1.3336680054277188,
+      "learning_rate": 5.195402298850576e-05,
+      "loss": 0.4709,
+      "step": 226
+    },
+    {
+      "epoch": 0.32661870503597124,
+      "grad_norm": 0.6182311428543251,
+      "learning_rate": 5.218390804597701e-05,
+      "loss": 0.48,
+      "step": 227
+    },
+    {
+      "epoch": 0.32805755395683456,
+      "grad_norm": 1.5772180116039807,
+      "learning_rate": 5.241379310344828e-05,
+      "loss": 0.4933,
+      "step": 228
+    },
+    {
+      "epoch": 0.3294964028776978,
+      "grad_norm": 0.8301808960896024,
+      "learning_rate": 5.264367816091954e-05,
+      "loss": 0.4851,
+      "step": 229
+    },
+    {
+      "epoch": 0.33093525179856115,
+      "grad_norm": 1.287879247394346,
+      "learning_rate": 5.287356321839081e-05,
+      "loss": 0.4882,
+      "step": 230
+    },
+    {
+      "epoch": 0.33237410071942447,
+      "grad_norm": 0.7930256285204332,
+      "learning_rate": 5.310344827586208e-05,
+      "loss": 0.4724,
+      "step": 231
+    },
+    {
+      "epoch": 0.3338129496402878,
+      "grad_norm": 0.8685183431577114,
+      "learning_rate": 5.333333333333333e-05,
+      "loss": 0.4913,
+      "step": 232
+    },
+    {
+      "epoch": 0.33525179856115106,
+      "grad_norm": 1.1066482481846536,
+      "learning_rate": 5.35632183908046e-05,
+      "loss": 0.4848,
+      "step": 233
+    },
+    {
+      "epoch": 0.3366906474820144,
+      "grad_norm": 0.8998014785047385,
+      "learning_rate": 5.379310344827587e-05,
+      "loss": 0.4811,
+      "step": 234
+    },
+    {
+      "epoch": 0.3381294964028777,
+      "grad_norm": 1.1251664212166712,
+      "learning_rate": 5.402298850574713e-05,
+      "loss": 0.4744,
+      "step": 235
+    },
+    {
+      "epoch": 0.339568345323741,
+      "grad_norm": 0.797966347020963,
+      "learning_rate": 5.425287356321839e-05,
+      "loss": 0.473,
+      "step": 236
+    },
+    {
+      "epoch": 0.3410071942446043,
+      "grad_norm": 1.4078158818561126,
+      "learning_rate": 5.4482758620689655e-05,
+      "loss": 0.4877,
+      "step": 237
+    },
+    {
+      "epoch": 0.3424460431654676,
+      "grad_norm": 0.5758988831861516,
+      "learning_rate": 5.471264367816092e-05,
+      "loss": 0.4842,
+      "step": 238
+    },
+    {
+      "epoch": 0.34388489208633094,
+      "grad_norm": 1.04982528972906,
+      "learning_rate": 5.494252873563219e-05,
+      "loss": 0.4849,
+      "step": 239
+    },
+    {
+      "epoch": 0.34532374100719426,
+      "grad_norm": 1.0248462844064845,
+      "learning_rate": 5.517241379310345e-05,
+      "loss": 0.4749,
+      "step": 240
+    },
+    {
+      "epoch": 0.34676258992805753,
+      "grad_norm": 0.788697716625114,
+      "learning_rate": 5.5402298850574715e-05,
+      "loss": 0.4837,
+      "step": 241
+    },
+    {
+      "epoch": 0.34820143884892085,
+      "grad_norm": 1.103753220461005,
+      "learning_rate": 5.563218390804598e-05,
+      "loss": 0.4812,
+      "step": 242
+    },
+    {
+      "epoch": 0.3496402877697842,
+      "grad_norm": 1.1125950371481492,
+      "learning_rate": 5.5862068965517245e-05,
+      "loss": 0.4853,
+      "step": 243
+    },
+    {
+      "epoch": 0.3510791366906475,
+      "grad_norm": 0.7511131774634014,
+      "learning_rate": 5.609195402298851e-05,
+      "loss": 0.4708,
+      "step": 244
+    },
+    {
+      "epoch": 0.35251798561151076,
+      "grad_norm": 0.984133060801353,
+      "learning_rate": 5.632183908045977e-05,
+      "loss": 0.4779,
+      "step": 245
+    },
+    {
+      "epoch": 0.3539568345323741,
+      "grad_norm": 1.3331164516977596,
+      "learning_rate": 5.6551724137931037e-05,
+      "loss": 0.4825,
+      "step": 246
+    },
+    {
+      "epoch": 0.3553956834532374,
+      "grad_norm": 1.070100828780393,
+      "learning_rate": 5.6781609195402305e-05,
+      "loss": 0.4829,
+      "step": 247
+    },
+    {
+      "epoch": 0.35683453237410073,
+      "grad_norm": 1.0753477967632834,
+      "learning_rate": 5.7011494252873567e-05,
+      "loss": 0.4806,
+      "step": 248
+    },
+    {
+      "epoch": 0.35827338129496406,
+      "grad_norm": 1.0086916569998965,
+      "learning_rate": 5.7241379310344835e-05,
+      "loss": 0.479,
+      "step": 249
+    },
+    {
+      "epoch": 0.3597122302158273,
+      "grad_norm": 1.000129764716916,
+      "learning_rate": 5.747126436781609e-05,
+      "loss": 0.4776,
+      "step": 250
+    },
+    {
+      "epoch": 0.36115107913669064,
+      "grad_norm": 1.363231863410886,
+      "learning_rate": 5.770114942528736e-05,
+      "loss": 0.4708,
+      "step": 251
+    },
+    {
+      "epoch": 0.36258992805755397,
+      "grad_norm": 0.9754682903695154,
+      "learning_rate": 5.7931034482758627e-05,
+      "loss": 0.4708,
+      "step": 252
+    },
+    {
+      "epoch": 0.3640287769784173,
+      "grad_norm": 1.1626305469845395,
+      "learning_rate": 5.8160919540229895e-05,
+      "loss": 0.4819,
+      "step": 253
+    },
+    {
+      "epoch": 0.36546762589928056,
+      "grad_norm": 1.0281653908967707,
+      "learning_rate": 5.839080459770116e-05,
+      "loss": 0.4862,
+      "step": 254
+    },
+    {
+      "epoch": 0.3669064748201439,
+      "grad_norm": 1.5241614994146868,
+      "learning_rate": 5.862068965517242e-05,
+      "loss": 0.4755,
+      "step": 255
+    },
+    {
+      "epoch": 0.3683453237410072,
+      "grad_norm": 0.6054701794166305,
+      "learning_rate": 5.885057471264368e-05,
+      "loss": 0.4836,
+      "step": 256
+    },
+    {
+      "epoch": 0.3697841726618705,
+      "grad_norm": 1.597009972339469,
+      "learning_rate": 5.908045977011495e-05,
+      "loss": 0.4903,
+      "step": 257
+    },
+    {
+      "epoch": 0.3712230215827338,
+      "grad_norm": 0.898775265650471,
+      "learning_rate": 5.931034482758622e-05,
+      "loss": 0.4752,
+      "step": 258
+    },
+    {
+      "epoch": 0.3726618705035971,
+      "grad_norm": 1.2510005303008964,
+      "learning_rate": 5.954022988505747e-05,
+      "loss": 0.492,
+      "step": 259
+    },
+    {
+      "epoch": 0.37410071942446044,
+      "grad_norm": 1.263106758218865,
+      "learning_rate": 5.977011494252874e-05,
+      "loss": 0.4839,
+      "step": 260
+    },
+    {
+      "epoch": 0.37553956834532376,
+      "grad_norm": 0.8391195279641513,
+      "learning_rate": 6.000000000000001e-05,
+      "loss": 0.4799,
+      "step": 261
+    },
+    {
+      "epoch": 0.376978417266187,
+      "grad_norm": 1.2992976420589124,
+      "learning_rate": 6.022988505747127e-05,
+      "loss": 0.4918,
+      "step": 262
+    },
+    {
+      "epoch": 0.37841726618705035,
+      "grad_norm": 0.9180286078429746,
+      "learning_rate": 6.045977011494254e-05,
+      "loss": 0.4831,
+      "step": 263
+    },
+    {
+      "epoch": 0.37985611510791367,
+      "grad_norm": 1.1975745180339628,
+      "learning_rate": 6.068965517241379e-05,
+      "loss": 0.4739,
+      "step": 264
+    },
+    {
+      "epoch": 0.381294964028777,
+      "grad_norm": 0.8871909141965262,
+      "learning_rate": 6.091954022988506e-05,
+      "loss": 0.4734,
+      "step": 265
+    },
+    {
+      "epoch": 0.38273381294964026,
+      "grad_norm": 0.7935115442549551,
+      "learning_rate": 6.114942528735632e-05,
+      "loss": 0.4723,
+      "step": 266
+    },
+    {
+      "epoch": 0.3841726618705036,
+      "grad_norm": 0.7815236006599882,
+      "learning_rate": 6.137931034482759e-05,
+      "loss": 0.4715,
+      "step": 267
+    },
+    {
+      "epoch": 0.3856115107913669,
+      "grad_norm": 0.9352283862124354,
+      "learning_rate": 6.160919540229885e-05,
+      "loss": 0.4815,
+      "step": 268
+    },
+    {
+      "epoch": 0.38705035971223023,
+      "grad_norm": 1.4559923722177637,
+      "learning_rate": 6.183908045977011e-05,
+      "loss": 0.4815,
+      "step": 269
+    },
+    {
+      "epoch": 0.38848920863309355,
+      "grad_norm": 0.6876046733921063,
+      "learning_rate": 6.206896551724138e-05,
+      "loss": 0.4745,
+      "step": 270
+    },
+    {
+      "epoch": 0.3899280575539568,
+      "grad_norm": 0.833844240162271,
+      "learning_rate": 6.229885057471265e-05,
+      "loss": 0.4763,
+      "step": 271
+    },
+    {
+      "epoch": 0.39136690647482014,
+      "grad_norm": 1.2650849654272054,
+      "learning_rate": 6.252873563218392e-05,
+      "loss": 0.4822,
+      "step": 272
+    },
+    {
+      "epoch": 0.39280575539568346,
+      "grad_norm": 1.0858926640637128,
+      "learning_rate": 6.275862068965517e-05,
+      "loss": 0.4664,
+      "step": 273
+    },
+    {
+      "epoch": 0.3942446043165468,
+      "grad_norm": 1.019011478452704,
+      "learning_rate": 6.298850574712644e-05,
+      "loss": 0.474,
+      "step": 274
+    },
+    {
+      "epoch": 0.39568345323741005,
+      "grad_norm": 1.0938566637945895,
+      "learning_rate": 6.321839080459771e-05,
+      "loss": 0.4774,
+      "step": 275
+    },
+    {
+      "epoch": 0.3971223021582734,
+      "grad_norm": 0.729489325137161,
+      "learning_rate": 6.344827586206897e-05,
+      "loss": 0.4747,
+      "step": 276
+    },
+    {
+      "epoch": 0.3985611510791367,
+      "grad_norm": 0.9699927603048148,
+      "learning_rate": 6.367816091954023e-05,
+      "loss": 0.4725,
+      "step": 277
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.842114037888034,
+      "learning_rate": 6.39080459770115e-05,
+      "loss": 0.4774,
+      "step": 278
+    },
+    {
+      "epoch": 0.4014388489208633,
+      "grad_norm": 1.2681756784337148,
+      "learning_rate": 6.413793103448276e-05,
+      "loss": 0.4776,
+      "step": 279
+    },
+    {
+      "epoch": 0.4028776978417266,
+      "grad_norm": 1.0074650909265712,
+      "learning_rate": 6.436781609195403e-05,
+      "loss": 0.4707,
+      "step": 280
+    },
+    {
+      "epoch": 0.40431654676258993,
+      "grad_norm": 0.8006996946486058,
+      "learning_rate": 6.45977011494253e-05,
+      "loss": 0.4759,
+      "step": 281
+    },
+    {
+      "epoch": 0.40575539568345326,
+      "grad_norm": 0.8574536311516484,
+      "learning_rate": 6.482758620689655e-05,
+      "loss": 0.4737,
+      "step": 282
+    },
+    {
+      "epoch": 0.4071942446043165,
+      "grad_norm": 0.9260422341892245,
+      "learning_rate": 6.505747126436782e-05,
+      "loss": 0.4772,
+      "step": 283
+    },
+    {
+      "epoch": 0.40863309352517985,
+      "grad_norm": 1.0937426163706612,
+      "learning_rate": 6.528735632183909e-05,
+      "loss": 0.4734,
+      "step": 284
+    },
+    {
+      "epoch": 0.41007194244604317,
+      "grad_norm": 1.1852915257823429,
+      "learning_rate": 6.551724137931035e-05,
+      "loss": 0.4743,
+      "step": 285
+    },
+    {
+      "epoch": 0.4115107913669065,
+      "grad_norm": 0.8483014350505912,
+      "learning_rate": 6.574712643678162e-05,
+      "loss": 0.4762,
+      "step": 286
+    },
+    {
+      "epoch": 0.41294964028776976,
+      "grad_norm": 0.7821533734039084,
+      "learning_rate": 6.597701149425288e-05,
+      "loss": 0.4712,
+      "step": 287
+    },
+    {
+      "epoch": 0.4143884892086331,
+      "grad_norm": 0.8098544578006053,
+      "learning_rate": 6.620689655172415e-05,
+      "loss": 0.4704,
+      "step": 288
+    },
+    {
+      "epoch": 0.4158273381294964,
+      "grad_norm": 1.0002003064812475,
+      "learning_rate": 6.643678160919542e-05,
+      "loss": 0.4828,
+      "step": 289
+    },
+    {
+      "epoch": 0.4172661870503597,
+      "grad_norm": 1.0166549231116344,
+      "learning_rate": 6.666666666666667e-05,
+      "loss": 0.473,
+      "step": 290
+    },
+    {
+      "epoch": 0.418705035971223,
+      "grad_norm": 0.8173620976657725,
+      "learning_rate": 6.689655172413794e-05,
+      "loss": 0.4666,
+      "step": 291
+    },
+    {
+      "epoch": 0.4201438848920863,
+      "grad_norm": 0.8176969242130069,
+      "learning_rate": 6.712643678160919e-05,
+      "loss": 0.466,
+      "step": 292
+    },
+    {
+      "epoch": 0.42158273381294964,
+      "grad_norm": 1.067242351763639,
+      "learning_rate": 6.735632183908046e-05,
+      "loss": 0.4743,
+      "step": 293
+    },
+    {
+      "epoch": 0.42302158273381296,
+      "grad_norm": 1.1373720540518368,
+      "learning_rate": 6.758620689655173e-05,
+      "loss": 0.4769,
+      "step": 294
+    },
+    {
+      "epoch": 0.4244604316546763,
+      "grad_norm": 1.2091834939605204,
+      "learning_rate": 6.7816091954023e-05,
+      "loss": 0.4723,
+      "step": 295
+    },
+    {
+      "epoch": 0.42589928057553955,
+      "grad_norm": 0.8553948982776225,
+      "learning_rate": 6.804597701149425e-05,
+      "loss": 0.4688,
+      "step": 296
+    },
+    {
+      "epoch": 0.4273381294964029,
+      "grad_norm": 0.9063903596659111,
+      "learning_rate": 6.827586206896552e-05,
+      "loss": 0.472,
+      "step": 297
+    },
+    {
+      "epoch": 0.4287769784172662,
+      "grad_norm": 0.8888596110092511,
+      "learning_rate": 6.850574712643679e-05,
+      "loss": 0.4724,
+      "step": 298
+    },
+    {
+      "epoch": 0.4302158273381295,
+      "grad_norm": 1.283932813683222,
+      "learning_rate": 6.873563218390806e-05,
+      "loss": 0.4701,
+      "step": 299
+    },
+    {
+      "epoch": 0.4316546762589928,
+      "grad_norm": 1.221149892015976,
+      "learning_rate": 6.896551724137931e-05,
+      "loss": 0.4729,
+      "step": 300
+    },
+    {
+      "epoch": 0.4330935251798561,
+      "grad_norm": 0.9176746013485364,
+      "learning_rate": 6.919540229885058e-05,
+      "loss": 0.4693,
+      "step": 301
+    },
+    {
+      "epoch": 0.43453237410071943,
+      "grad_norm": 0.7027300631739719,
+      "learning_rate": 6.942528735632185e-05,
+      "loss": 0.4664,
+      "step": 302
+    },
+    {
+      "epoch": 0.43597122302158275,
+      "grad_norm": 0.7020387618730545,
+      "learning_rate": 6.96551724137931e-05,
+      "loss": 0.4673,
+      "step": 303
+    },
+    {
+      "epoch": 0.437410071942446,
+      "grad_norm": 0.6808651310429936,
+      "learning_rate": 6.988505747126437e-05,
+      "loss": 0.4683,
+      "step": 304
+    },
+    {
+      "epoch": 0.43884892086330934,
+      "grad_norm": 0.8424811250488698,
+      "learning_rate": 7.011494252873563e-05,
+      "loss": 0.457,
+      "step": 305
+    },
+    {
+      "epoch": 0.44028776978417267,
+      "grad_norm": 0.9438233740449851,
+      "learning_rate": 7.03448275862069e-05,
+      "loss": 0.4771,
+      "step": 306
+    },
+    {
+      "epoch": 0.441726618705036,
+      "grad_norm": 0.9851213333724348,
+      "learning_rate": 7.057471264367816e-05,
+      "loss": 0.4681,
+      "step": 307
+    },
+    {
+      "epoch": 0.44316546762589926,
+      "grad_norm": 1.289642954389415,
+      "learning_rate": 7.080459770114943e-05,
+      "loss": 0.4785,
+      "step": 308
+    },
+    {
+      "epoch": 0.4446043165467626,
+      "grad_norm": 1.0544507950814246,
+      "learning_rate": 7.10344827586207e-05,
+      "loss": 0.4724,
+      "step": 309
+    },
+    {
+      "epoch": 0.4460431654676259,
+      "grad_norm": 1.3215957113890155,
+      "learning_rate": 7.126436781609196e-05,
+      "loss": 0.4757,
+      "step": 310
+    },
+    {
+      "epoch": 0.4474820143884892,
+      "grad_norm": 0.7316608520379191,
+      "learning_rate": 7.149425287356322e-05,
+      "loss": 0.4684,
+      "step": 311
+    },
+    {
+      "epoch": 0.4489208633093525,
+      "grad_norm": 0.969696808950938,
+      "learning_rate": 7.17241379310345e-05,
+      "loss": 0.479,
+      "step": 312
+    },
+    {
+      "epoch": 0.4503597122302158,
+      "grad_norm": 0.9477963717668171,
+      "learning_rate": 7.195402298850576e-05,
+      "loss": 0.4741,
+      "step": 313
+    },
+    {
+      "epoch": 0.45179856115107914,
+      "grad_norm": 0.9206502919147358,
+      "learning_rate": 7.218390804597702e-05,
+      "loss": 0.4748,
+      "step": 314
+    },
+    {
+      "epoch": 0.45323741007194246,
+      "grad_norm": 1.1130093138557773,
+      "learning_rate": 7.241379310344828e-05,
+      "loss": 0.4694,
+      "step": 315
+    },
+    {
+      "epoch": 0.4546762589928058,
+      "grad_norm": 1.1388074164156727,
+      "learning_rate": 7.264367816091954e-05,
+      "loss": 0.4692,
+      "step": 316
+    },
+    {
+      "epoch": 0.45611510791366905,
+      "grad_norm": 1.1164661442567994,
+      "learning_rate": 7.287356321839081e-05,
+      "loss": 0.4741,
+      "step": 317
+    },
+    {
+      "epoch": 0.45755395683453237,
+      "grad_norm": 1.0654010909802574,
+      "learning_rate": 7.310344827586208e-05,
+      "loss": 0.4683,
+      "step": 318
+    },
+    {
+      "epoch": 0.4589928057553957,
+      "grad_norm": 0.8053385009779973,
+      "learning_rate": 7.333333333333333e-05,
+      "loss": 0.4666,
+      "step": 319
+    },
+    {
+      "epoch": 0.460431654676259,
+      "grad_norm": 0.7282412960155227,
+      "learning_rate": 7.35632183908046e-05,
+      "loss": 0.463,
+      "step": 320
+    },
+    {
+      "epoch": 0.4618705035971223,
+      "grad_norm": 0.942774932234221,
+      "learning_rate": 7.379310344827587e-05,
+      "loss": 0.4667,
+      "step": 321
+    },
+    {
+      "epoch": 0.4633093525179856,
+      "grad_norm": 1.2102092661091306,
+      "learning_rate": 7.402298850574714e-05,
+      "loss": 0.4753,
+      "step": 322
+    },
+    {
+      "epoch": 0.46474820143884893,
+      "grad_norm": 1.0870046005139904,
+      "learning_rate": 7.425287356321839e-05,
+      "loss": 0.472,
+      "step": 323
+    },
+    {
+      "epoch": 0.46618705035971225,
+      "grad_norm": 1.1085973189399043,
+      "learning_rate": 7.448275862068966e-05,
+      "loss": 0.4696,
+      "step": 324
+    },
+    {
+      "epoch": 0.4676258992805755,
+      "grad_norm": 1.0038842187549621,
+      "learning_rate": 7.471264367816093e-05,
+      "loss": 0.4615,
+      "step": 325
+    },
+    {
+      "epoch": 0.46906474820143884,
+      "grad_norm": 1.0359164440235704,
+      "learning_rate": 7.49425287356322e-05,
+      "loss": 0.4694,
+      "step": 326
+    },
+    {
+      "epoch": 0.47050359712230216,
+      "grad_norm": 0.9837789064434185,
+      "learning_rate": 7.517241379310345e-05,
+      "loss": 0.4713,
+      "step": 327
+    },
+    {
+      "epoch": 0.4719424460431655,
+      "grad_norm": 0.9620898511324295,
+      "learning_rate": 7.540229885057472e-05,
+      "loss": 0.4688,
+      "step": 328
+    },
+    {
+      "epoch": 0.47338129496402875,
+      "grad_norm": 1.1430319437453274,
+      "learning_rate": 7.563218390804599e-05,
+      "loss": 0.472,
+      "step": 329
+    },
+    {
+      "epoch": 0.4748201438848921,
+      "grad_norm": 1.007227411189436,
+      "learning_rate": 7.586206896551724e-05,
+      "loss": 0.4648,
+      "step": 330
+    },
+    {
+      "epoch": 0.4762589928057554,
+      "grad_norm": 1.0606046089785117,
+      "learning_rate": 7.609195402298851e-05,
+      "loss": 0.4743,
+      "step": 331
+    },
+    {
+      "epoch": 0.4776978417266187,
+      "grad_norm": 0.8960210040501575,
+      "learning_rate": 7.632183908045977e-05,
+      "loss": 0.4628,
+      "step": 332
+    },
+    {
+      "epoch": 0.479136690647482,
+      "grad_norm": 1.1241387696547236,
+      "learning_rate": 7.655172413793103e-05,
+      "loss": 0.4702,
+      "step": 333
+    },
+    {
+      "epoch": 0.4805755395683453,
+      "grad_norm": 1.3163055988137928,
+      "learning_rate": 7.67816091954023e-05,
+      "loss": 0.4706,
+      "step": 334
+    },
+    {
+      "epoch": 0.48201438848920863,
+      "grad_norm": 0.8647862396783775,
+      "learning_rate": 7.701149425287357e-05,
+      "loss": 0.469,
+      "step": 335
+    },
+    {
+      "epoch": 0.48345323741007196,
+      "grad_norm": 0.7643381038686257,
+      "learning_rate": 7.724137931034484e-05,
+      "loss": 0.4728,
+      "step": 336
+    },
+    {
+      "epoch": 0.4848920863309353,
+      "grad_norm": 0.8778154724393922,
+      "learning_rate": 7.74712643678161e-05,
+      "loss": 0.4625,
+      "step": 337
+    },
+    {
+      "epoch": 0.48633093525179855,
+      "grad_norm": 1.114347449968458,
+      "learning_rate": 7.770114942528736e-05,
+      "loss": 0.471,
+      "step": 338
+    },
+    {
+      "epoch": 0.48776978417266187,
+      "grad_norm": 0.7707670307497948,
+      "learning_rate": 7.793103448275863e-05,
+      "loss": 0.4725,
+      "step": 339
+    },
+    {
+      "epoch": 0.4892086330935252,
+      "grad_norm": 0.9295505229318217,
+      "learning_rate": 7.81609195402299e-05,
+      "loss": 0.4583,
+      "step": 340
+    },
+    {
+      "epoch": 0.4906474820143885,
+      "grad_norm": 1.3350454744024955,
+      "learning_rate": 7.839080459770115e-05,
+      "loss": 0.468,
+      "step": 341
+    },
+    {
+      "epoch": 0.4920863309352518,
+      "grad_norm": 0.8539494522690838,
+      "learning_rate": 7.862068965517242e-05,
+      "loss": 0.4654,
+      "step": 342
+    },
+    {
+      "epoch": 0.4935251798561151,
+      "grad_norm": 0.851472197165697,
+      "learning_rate": 7.885057471264368e-05,
+      "loss": 0.4592,
+      "step": 343
+    },
+    {
+      "epoch": 0.4949640287769784,
+      "grad_norm": 0.7949738359780544,
+      "learning_rate": 7.908045977011495e-05,
+      "loss": 0.4674,
+      "step": 344
+    },
+    {
+      "epoch": 0.49640287769784175,
+      "grad_norm": 0.6553751988750726,
+      "learning_rate": 7.931034482758621e-05,
+      "loss": 0.4653,
+      "step": 345
+    },
+    {
+      "epoch": 0.497841726618705,
+      "grad_norm": 0.9279428864744551,
+      "learning_rate": 7.954022988505747e-05,
+      "loss": 0.4661,
+      "step": 346
+    },
+    {
+      "epoch": 0.49928057553956834,
+      "grad_norm": 1.081074146306807,
+      "learning_rate": 7.977011494252874e-05,
+      "loss": 0.4613,
+      "step": 347
+    },
+    {
+      "epoch": 0.5007194244604316,
+      "grad_norm": 0.8701513845004177,
+      "learning_rate": 8e-05,
+      "loss": 0.4624,
+      "step": 348
+    },
+    {
+      "epoch": 0.5021582733812949,
+      "grad_norm": 1.2328271579662156,
+      "learning_rate": 7.999997981289966e-05,
+      "loss": 0.4798,
+      "step": 349
+    },
+    {
+      "epoch": 0.5035971223021583,
+      "grad_norm": 0.8508168481221876,
+      "learning_rate": 7.999991925161896e-05,
+      "loss": 0.4626,
+      "step": 350
+    },
+    {
+      "epoch": 0.5050359712230216,
+      "grad_norm": 0.8463101325573931,
+      "learning_rate": 7.999981831621906e-05,
+      "loss": 0.4758,
+      "step": 351
+    },
+    {
+      "epoch": 0.5064748201438849,
+      "grad_norm": 1.1928759234173607,
+      "learning_rate": 7.999967700680183e-05,
+      "loss": 0.4797,
+      "step": 352
+    },
+    {
+      "epoch": 0.5079136690647482,
+      "grad_norm": 1.4347979499250567,
+      "learning_rate": 7.99994953235099e-05,
+      "loss": 0.4592,
+      "step": 353
+    },
+    {
+      "epoch": 0.5093525179856115,
+      "grad_norm": 0.8032557148584375,
+      "learning_rate": 7.999927326652667e-05,
+      "loss": 0.4683,
+      "step": 354
+    },
+    {
+      "epoch": 0.5107913669064749,
+      "grad_norm": 1.0275144868037598,
+      "learning_rate": 7.999901083607624e-05,
+      "loss": 0.4708,
+      "step": 355
+    },
+    {
+      "epoch": 0.5122302158273381,
+      "grad_norm": 1.3566895504819316,
+      "learning_rate": 7.99987080324235e-05,
+      "loss": 0.4681,
+      "step": 356
+    },
+    {
+      "epoch": 0.5136690647482014,
+      "grad_norm": 1.0626475712542294,
+      "learning_rate": 7.999836485587415e-05,
+      "loss": 0.4619,
+      "step": 357
+    },
+    {
+      "epoch": 0.5151079136690647,
+      "grad_norm": 0.9998118581806145,
+      "learning_rate": 7.99979813067745e-05,
+      "loss": 0.4576,
+      "step": 358
+    },
+    {
+      "epoch": 0.516546762589928,
+      "grad_norm": 0.9557268904806338,
+      "learning_rate": 7.999755738551171e-05,
+      "loss": 0.4668,
+      "step": 359
+    },
+    {
+      "epoch": 0.5179856115107914,
+      "grad_norm": 1.0796775023064598,
+      "learning_rate": 7.999709309251368e-05,
+      "loss": 0.4574,
+      "step": 360
+    },
+    {
+      "epoch": 0.5194244604316547,
+      "grad_norm": 1.0500619364793755,
+      "learning_rate": 7.999658842824904e-05,
+      "loss": 0.4577,
+      "step": 361
+    },
+    {
+      "epoch": 0.520863309352518,
+      "grad_norm": 0.9744110414015235,
+      "learning_rate": 7.999604339322717e-05,
+      "loss": 0.4604,
+      "step": 362
+    },
+    {
+      "epoch": 0.5223021582733813,
+      "grad_norm": 0.6004652084768644,
+      "learning_rate": 7.999545798799823e-05,
+      "loss": 0.4583,
+      "step": 363
+    },
+    {
+      "epoch": 0.5237410071942447,
+      "grad_norm": 0.4987568342275504,
+      "learning_rate": 7.999483221315307e-05,
+      "loss": 0.4656,
+      "step": 364
+    },
+    {
+      "epoch": 0.5251798561151079,
+      "grad_norm": 0.6004131997500541,
+      "learning_rate": 7.999416606932331e-05,
+      "loss": 0.4534,
+      "step": 365
+    },
+    {
+      "epoch": 0.5266187050359712,
+      "grad_norm": 0.6149304879668175,
+      "learning_rate": 7.999345955718136e-05,
+      "loss": 0.4613,
+      "step": 366
+    },
+    {
+      "epoch": 0.5280575539568345,
+      "grad_norm": 0.8500344008908939,
+      "learning_rate": 7.999271267744033e-05,
+      "loss": 0.4652,
+      "step": 367
+    },
+    {
+      "epoch": 0.5294964028776978,
+      "grad_norm": 0.9413054629306632,
+      "learning_rate": 7.999192543085407e-05,
+      "loss": 0.4687,
+      "step": 368
+    },
+    {
+      "epoch": 0.5309352517985612,
+      "grad_norm": 1.2121156729309888,
+      "learning_rate": 7.999109781821722e-05,
+      "loss": 0.4625,
+      "step": 369
+    },
+    {
+      "epoch": 0.5323741007194245,
+      "grad_norm": 0.7421231375424578,
+      "learning_rate": 7.999022984036512e-05,
+      "loss": 0.4623,
+      "step": 370
+    },
+    {
+      "epoch": 0.5338129496402878,
+      "grad_norm": 0.6597146546987727,
+      "learning_rate": 7.998932149817386e-05,
+      "loss": 0.457,
+      "step": 371
+    },
+    {
+      "epoch": 0.5352517985611511,
+      "grad_norm": 0.9338747126537402,
+      "learning_rate": 7.998837279256028e-05,
+      "loss": 0.4741,
+      "step": 372
+    },
+    {
+      "epoch": 0.5366906474820143,
+      "grad_norm": 1.1172196724004102,
+      "learning_rate": 7.998738372448196e-05,
+      "loss": 0.4587,
+      "step": 373
+    },
+    {
+      "epoch": 0.5381294964028777,
+      "grad_norm": 0.7812818117620776,
+      "learning_rate": 7.998635429493726e-05,
+      "loss": 0.4728,
+      "step": 374
+    },
+    {
+      "epoch": 0.539568345323741,
+      "grad_norm": 0.6787927858149476,
+      "learning_rate": 7.998528450496519e-05,
+      "loss": 0.4578,
+      "step": 375
+    },
+    {
+      "epoch": 0.5410071942446043,
+      "grad_norm": 0.714091642204277,
+      "learning_rate": 7.998417435564557e-05,
+      "loss": 0.4722,
+      "step": 376
+    },
+    {
+      "epoch": 0.5424460431654676,
+      "grad_norm": 0.8504914934981742,
+      "learning_rate": 7.998302384809893e-05,
+      "loss": 0.4537,
+      "step": 377
+    },
+    {
+      "epoch": 0.543884892086331,
+      "grad_norm": 0.9898476948452237,
+      "learning_rate": 7.998183298348654e-05,
+      "loss": 0.4664,
+      "step": 378
+    },
+    {
+      "epoch": 0.5453237410071943,
+      "grad_norm": 1.2535233391949137,
+      "learning_rate": 7.998060176301041e-05,
+      "loss": 0.4541,
+      "step": 379
+    },
+    {
+      "epoch": 0.5467625899280576,
+      "grad_norm": 0.6997883291597523,
+      "learning_rate": 7.997933018791327e-05,
+      "loss": 0.4583,
+      "step": 380
+    },
+    {
+      "epoch": 0.5482014388489208,
+      "grad_norm": 0.7736047534054794,
+      "learning_rate": 7.99780182594786e-05,
+      "loss": 0.4607,
+      "step": 381
+    },
+    {
+      "epoch": 0.5496402877697841,
+      "grad_norm": 0.849423061431514,
+      "learning_rate": 7.99766659790306e-05,
+      "loss": 0.4664,
+      "step": 382
+    },
+    {
+      "epoch": 0.5510791366906475,
+      "grad_norm": 0.8499243646515073,
+      "learning_rate": 7.997527334793419e-05,
+      "loss": 0.4532,
+      "step": 383
+    },
+    {
+      "epoch": 0.5525179856115108,
+      "grad_norm": 0.9507351440178784,
+      "learning_rate": 7.997384036759505e-05,
+      "loss": 0.4644,
+      "step": 384
+    },
+    {
+      "epoch": 0.5539568345323741,
+      "grad_norm": 0.7880806752546511,
+      "learning_rate": 7.997236703945955e-05,
+      "loss": 0.4565,
+      "step": 385
+    },
+    {
+      "epoch": 0.5553956834532374,
+      "grad_norm": 0.8235336874532355,
+      "learning_rate": 7.99708533650148e-05,
+      "loss": 0.4602,
+      "step": 386
+    },
+    {
+      "epoch": 0.5568345323741007,
+      "grad_norm": 1.1406064704681915,
+      "learning_rate": 7.996929934578864e-05,
+      "loss": 0.4631,
+      "step": 387
+    },
+    {
+      "epoch": 0.5582733812949641,
+      "grad_norm": 1.0936145617683148,
+      "learning_rate": 7.996770498334963e-05,
+      "loss": 0.4665,
+      "step": 388
+    },
+    {
+      "epoch": 0.5597122302158274,
+      "grad_norm": 0.8845721786169036,
+      "learning_rate": 7.996607027930705e-05,
+      "loss": 0.4534,
+      "step": 389
+    },
+    {
+      "epoch": 0.5611510791366906,
+      "grad_norm": 0.8780922124552242,
+      "learning_rate": 7.996439523531088e-05,
+      "loss": 0.4624,
+      "step": 390
+    },
+    {
+      "epoch": 0.5625899280575539,
+      "grad_norm": 1.004467673049218,
+      "learning_rate": 7.996267985305186e-05,
+      "loss": 0.4589,
+      "step": 391
+    },
+    {
+      "epoch": 0.5640287769784172,
+      "grad_norm": 1.0826878157187065,
+      "learning_rate": 7.99609241342614e-05,
+      "loss": 0.4558,
+      "step": 392
+    },
+    {
+      "epoch": 0.5654676258992806,
+      "grad_norm": 0.8097477147009582,
+      "learning_rate": 7.995912808071164e-05,
+      "loss": 0.4619,
+      "step": 393
+    },
+    {
+      "epoch": 0.5669064748201439,
+      "grad_norm": 0.693965038621037,
+      "learning_rate": 7.995729169421545e-05,
+      "loss": 0.4616,
+      "step": 394
+    },
+    {
+      "epoch": 0.5683453237410072,
+      "grad_norm": 0.9308338290598528,
+      "learning_rate": 7.99554149766264e-05,
+      "loss": 0.4622,
+      "step": 395
+    },
+    {
+      "epoch": 0.5697841726618705,
+      "grad_norm": 1.1817576406269814,
+      "learning_rate": 7.995349792983874e-05,
+      "loss": 0.4614,
+      "step": 396
+    },
+    {
+      "epoch": 0.5712230215827339,
+      "grad_norm": 0.6528770424649769,
+      "learning_rate": 7.995154055578748e-05,
+      "loss": 0.459,
+      "step": 397
+    },
+    {
+      "epoch": 0.5726618705035971,
+      "grad_norm": 0.45150970577996347,
+      "learning_rate": 7.994954285644827e-05,
+      "loss": 0.4591,
+      "step": 398
+    },
+    {
+      "epoch": 0.5741007194244604,
+      "grad_norm": 0.5556005129157767,
+      "learning_rate": 7.994750483383753e-05,
+      "loss": 0.4613,
+      "step": 399
+    },
+    {
+      "epoch": 0.5755395683453237,
+      "grad_norm": 0.6912244983340633,
+      "learning_rate": 7.994542649001235e-05,
+      "loss": 0.4553,
+      "step": 400
+    },
+    {
+      "epoch": 0.576978417266187,
+      "grad_norm": 0.8960022596299475,
+      "learning_rate": 7.994330782707048e-05,
+      "loss": 0.4597,
+      "step": 401
+    },
+    {
+      "epoch": 0.5784172661870504,
+      "grad_norm": 1.025870832226272,
+      "learning_rate": 7.994114884715045e-05,
+      "loss": 0.451,
+      "step": 402
+    },
+    {
+      "epoch": 0.5798561151079137,
+      "grad_norm": 0.8911346100337705,
+      "learning_rate": 7.99389495524314e-05,
+      "loss": 0.4627,
+      "step": 403
+    },
+    {
+      "epoch": 0.581294964028777,
+      "grad_norm": 0.7554954071013955,
+      "learning_rate": 7.993670994513321e-05,
+      "loss": 0.4526,
+      "step": 404
+    },
+    {
+      "epoch": 0.5827338129496403,
+      "grad_norm": 0.6689831752367191,
+      "learning_rate": 7.993443002751646e-05,
+      "loss": 0.4594,
+      "step": 405
+    },
+    {
+      "epoch": 0.5841726618705037,
+      "grad_norm": 0.5578468632676835,
+      "learning_rate": 7.993210980188236e-05,
+      "loss": 0.4526,
+      "step": 406
+    },
+    {
+      "epoch": 0.5856115107913669,
+      "grad_norm": 0.5803040347753762,
+      "learning_rate": 7.992974927057287e-05,
+      "loss": 0.4643,
+      "step": 407
+    },
+    {
+      "epoch": 0.5870503597122302,
+      "grad_norm": 0.6577126675911862,
+      "learning_rate": 7.992734843597058e-05,
+      "loss": 0.4604,
+      "step": 408
+    },
+    {
+      "epoch": 0.5884892086330935,
+      "grad_norm": 0.6586582167853312,
+      "learning_rate": 7.992490730049881e-05,
+      "loss": 0.4506,
+      "step": 409
+    },
+    {
+      "epoch": 0.5899280575539568,
+      "grad_norm": 0.5099795931765497,
+      "learning_rate": 7.992242586662152e-05,
+      "loss": 0.4527,
+      "step": 410
+    },
+    {
+      "epoch": 0.5913669064748202,
+      "grad_norm": 0.43969451590829833,
+      "learning_rate": 7.991990413684336e-05,
+      "loss": 0.4408,
+      "step": 411
+    },
+    {
+      "epoch": 0.5928057553956835,
+      "grad_norm": 0.48368464104887693,
+      "learning_rate": 7.991734211370965e-05,
+      "loss": 0.4543,
+      "step": 412
+    },
+    {
+      "epoch": 0.5942446043165468,
+      "grad_norm": 0.6189368627099892,
+      "learning_rate": 7.991473979980637e-05,
+      "loss": 0.4501,
+      "step": 413
+    },
+    {
+      "epoch": 0.5956834532374101,
+      "grad_norm": 0.7712221623764791,
+      "learning_rate": 7.99120971977602e-05,
+      "loss": 0.4439,
+      "step": 414
+    },
+    {
+      "epoch": 0.5971223021582733,
+      "grad_norm": 1.1017419802815924,
+      "learning_rate": 7.990941431023844e-05,
+      "loss": 0.4692,
+      "step": 415
+    },
+    {
+      "epoch": 0.5985611510791367,
+      "grad_norm": 1.3463982426436771,
+      "learning_rate": 7.990669113994911e-05,
+      "loss": 0.4535,
+      "step": 416
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.5313608173987471,
+      "learning_rate": 7.99039276896408e-05,
+      "loss": 0.4552,
+      "step": 417
+    },
+    {
+      "epoch": 0.6014388489208633,
+      "grad_norm": 0.8384027425355336,
+      "learning_rate": 7.990112396210288e-05,
+      "loss": 0.4498,
+      "step": 418
+    },
+    {
+      "epoch": 0.6028776978417266,
+      "grad_norm": 1.5419005868846274,
+      "learning_rate": 7.989827996016525e-05,
+      "loss": 0.4541,
+      "step": 419
+    },
+    {
+      "epoch": 0.60431654676259,
+      "grad_norm": 0.5236106532851287,
+      "learning_rate": 7.989539568669856e-05,
+      "loss": 0.4547,
+      "step": 420
+    },
+    {
+      "epoch": 0.6057553956834533,
+      "grad_norm": 1.5495871475389187,
+      "learning_rate": 7.989247114461403e-05,
+      "loss": 0.4727,
+      "step": 421
+    },
+    {
+      "epoch": 0.6071942446043166,
+      "grad_norm": 0.6710377853395758,
+      "learning_rate": 7.988950633686358e-05,
+      "loss": 0.4629,
+      "step": 422
+    },
+    {
+      "epoch": 0.6086330935251798,
+      "grad_norm": 1.1495687379614778,
+      "learning_rate": 7.988650126643976e-05,
+      "loss": 0.4643,
+      "step": 423
+    },
+    {
+      "epoch": 0.6100719424460431,
+      "grad_norm": 0.7728837622561019,
+      "learning_rate": 7.988345593637572e-05,
+      "loss": 0.4505,
+      "step": 424
+    },
+    {
+      "epoch": 0.6115107913669064,
+      "grad_norm": 0.7153271121647338,
+      "learning_rate": 7.988037034974532e-05,
+      "loss": 0.454,
+      "step": 425
+    },
+    {
+      "epoch": 0.6129496402877698,
+      "grad_norm": 0.8556877038559139,
+      "learning_rate": 7.9877244509663e-05,
+      "loss": 0.4501,
+      "step": 426
+    },
+    {
+      "epoch": 0.6143884892086331,
+      "grad_norm": 0.5822077080483781,
+      "learning_rate": 7.987407841928384e-05,
+      "loss": 0.4555,
+      "step": 427
+    },
+    {
+      "epoch": 0.6158273381294964,
+      "grad_norm": 0.7502935198956667,
+      "learning_rate": 7.987087208180355e-05,
+      "loss": 0.4597,
+      "step": 428
+    },
+    {
+      "epoch": 0.6172661870503597,
+      "grad_norm": 0.894659880332655,
+      "learning_rate": 7.986762550045844e-05,
+      "loss": 0.4425,
+      "step": 429
+    },
+    {
+      "epoch": 0.6187050359712231,
+      "grad_norm": 0.8428743370334983,
+      "learning_rate": 7.98643386785255e-05,
+      "loss": 0.4614,
+      "step": 430
+    },
+    {
+      "epoch": 0.6201438848920864,
+      "grad_norm": 1.0150305538139375,
+      "learning_rate": 7.986101161932227e-05,
+      "loss": 0.4542,
+      "step": 431
+    },
+    {
+      "epoch": 0.6215827338129496,
+      "grad_norm": 1.1915793912859454,
+      "learning_rate": 7.985764432620695e-05,
+      "loss": 0.4591,
+      "step": 432
+    },
+    {
+      "epoch": 0.6230215827338129,
+      "grad_norm": 0.9500995658899973,
+      "learning_rate": 7.985423680257833e-05,
+      "loss": 0.4557,
+      "step": 433
+    },
+    {
+      "epoch": 0.6244604316546762,
+      "grad_norm": 0.8504572576195092,
+      "learning_rate": 7.985078905187582e-05,
+      "loss": 0.4489,
+      "step": 434
+    },
+    {
+      "epoch": 0.6258992805755396,
+      "grad_norm": 0.6552243752639753,
+      "learning_rate": 7.984730107757942e-05,
+      "loss": 0.4535,
+      "step": 435
+    },
+    {
+      "epoch": 0.6273381294964029,
+      "grad_norm": 0.9486300093548566,
+      "learning_rate": 7.984377288320973e-05,
+      "loss": 0.4709,
+      "step": 436
+    },
+    {
+      "epoch": 0.6287769784172662,
+      "grad_norm": 1.2450786856279674,
+      "learning_rate": 7.984020447232795e-05,
+      "loss": 0.4603,
+      "step": 437
+    },
+    {
+      "epoch": 0.6302158273381295,
+      "grad_norm": 0.7081913514292113,
+      "learning_rate": 7.983659584853586e-05,
+      "loss": 0.4555,
+      "step": 438
+    },
+    {
+      "epoch": 0.6316546762589929,
+      "grad_norm": 0.5133251754462216,
+      "learning_rate": 7.983294701547588e-05,
+      "loss": 0.457,
+      "step": 439
+    },
+    {
+      "epoch": 0.6330935251798561,
+      "grad_norm": 0.6949146616899369,
+      "learning_rate": 7.982925797683095e-05,
+      "loss": 0.4654,
+      "step": 440
+    },
+    {
+      "epoch": 0.6345323741007194,
+      "grad_norm": 0.7996453947268342,
+      "learning_rate": 7.982552873632461e-05,
+      "loss": 0.4503,
+      "step": 441
+    },
+    {
+      "epoch": 0.6359712230215827,
+      "grad_norm": 0.7979003739605092,
+      "learning_rate": 7.982175929772102e-05,
+      "loss": 0.4614,
+      "step": 442
+    },
+    {
+      "epoch": 0.637410071942446,
+      "grad_norm": 0.7773994652608701,
+      "learning_rate": 7.981794966482486e-05,
+      "loss": 0.4603,
+      "step": 443
+    },
+    {
+      "epoch": 0.6388489208633094,
+      "grad_norm": 0.7595360865967253,
+      "learning_rate": 7.98140998414814e-05,
+      "loss": 0.4578,
+      "step": 444
+    },
+    {
+      "epoch": 0.6402877697841727,
+      "grad_norm": 0.7520837510292663,
+      "learning_rate": 7.98102098315765e-05,
+      "loss": 0.4502,
+      "step": 445
+    },
+    {
+      "epoch": 0.641726618705036,
+      "grad_norm": 0.8160841775970298,
+      "learning_rate": 7.980627963903654e-05,
+      "loss": 0.4485,
+      "step": 446
+    },
+    {
+      "epoch": 0.6431654676258993,
+      "grad_norm": 1.0464193362779555,
+      "learning_rate": 7.980230926782848e-05,
+      "loss": 0.4477,
+      "step": 447
+    },
+    {
+      "epoch": 0.6446043165467625,
+      "grad_norm": 1.0091826922037084,
+      "learning_rate": 7.979829872195984e-05,
+      "loss": 0.4538,
+      "step": 448
+    },
+    {
+      "epoch": 0.6460431654676259,
+      "grad_norm": 0.8990157367734588,
+      "learning_rate": 7.979424800547869e-05,
+      "loss": 0.4557,
+      "step": 449
+    },
+    {
+      "epoch": 0.6474820143884892,
+      "grad_norm": 0.8300047632795456,
+      "learning_rate": 7.979015712247365e-05,
+      "loss": 0.458,
+      "step": 450
+    },
+    {
+      "epoch": 0.6489208633093525,
+      "grad_norm": 0.7149509540074934,
+      "learning_rate": 7.978602607707383e-05,
+      "loss": 0.47,
+      "step": 451
+    },
+    {
+      "epoch": 0.6503597122302158,
+      "grad_norm": 0.5970009144007832,
+      "learning_rate": 7.978185487344897e-05,
+      "loss": 0.4519,
+      "step": 452
+    },
+    {
+      "epoch": 0.6517985611510791,
+      "grad_norm": 0.4967763910687283,
+      "learning_rate": 7.977764351580928e-05,
+      "loss": 0.4439,
+      "step": 453
+    },
+    {
+      "epoch": 0.6532374100719425,
+      "grad_norm": 0.5092650465666206,
+      "learning_rate": 7.97733920084055e-05,
+      "loss": 0.4511,
+      "step": 454
+    },
+    {
+      "epoch": 0.6546762589928058,
+      "grad_norm": 0.6185513764677956,
+      "learning_rate": 7.976910035552892e-05,
+      "loss": 0.4513,
+      "step": 455
+    },
+    {
+      "epoch": 0.6561151079136691,
+      "grad_norm": 0.6398423442285093,
+      "learning_rate": 7.976476856151134e-05,
+      "loss": 0.4493,
+      "step": 456
+    },
+    {
+      "epoch": 0.6575539568345323,
+      "grad_norm": 0.7664721097324098,
+      "learning_rate": 7.976039663072509e-05,
+      "loss": 0.4441,
+      "step": 457
+    },
+    {
+      "epoch": 0.6589928057553956,
+      "grad_norm": 0.827523466225571,
+      "learning_rate": 7.975598456758298e-05,
+      "loss": 0.4593,
+      "step": 458
+    },
+    {
+      "epoch": 0.660431654676259,
+      "grad_norm": 0.9182022459615838,
+      "learning_rate": 7.975153237653836e-05,
+      "loss": 0.4487,
+      "step": 459
+    },
+    {
+      "epoch": 0.6618705035971223,
+      "grad_norm": 0.9735790118953446,
+      "learning_rate": 7.974704006208509e-05,
+      "loss": 0.4479,
+      "step": 460
+    },
+    {
+      "epoch": 0.6633093525179856,
+      "grad_norm": 0.9415118087459532,
+      "learning_rate": 7.974250762875747e-05,
+      "loss": 0.4533,
+      "step": 461
+    },
+    {
+      "epoch": 0.6647482014388489,
+      "grad_norm": 0.7848209637914124,
+      "learning_rate": 7.973793508113035e-05,
+      "loss": 0.4423,
+      "step": 462
+    },
+    {
+      "epoch": 0.6661870503597123,
+      "grad_norm": 0.5158906455961678,
+      "learning_rate": 7.973332242381908e-05,
+      "loss": 0.4514,
+      "step": 463
+    },
+    {
+      "epoch": 0.6676258992805756,
+      "grad_norm": 0.41797571499127906,
+      "learning_rate": 7.972866966147942e-05,
+      "loss": 0.4475,
+      "step": 464
+    },
+    {
+      "epoch": 0.6690647482014388,
+      "grad_norm": 0.47340136151923096,
+      "learning_rate": 7.972397679880771e-05,
+      "loss": 0.4517,
+      "step": 465
+    },
+    {
+      "epoch": 0.6705035971223021,
+      "grad_norm": 0.5787418072919882,
+      "learning_rate": 7.971924384054068e-05,
+      "loss": 0.4573,
+      "step": 466
+    },
+    {
+      "epoch": 0.6719424460431654,
+      "grad_norm": 0.6523824054062983,
+      "learning_rate": 7.971447079145557e-05,
+      "loss": 0.4542,
+      "step": 467
+    },
+    {
+      "epoch": 0.6733812949640288,
+      "grad_norm": 0.7105356243378426,
+      "learning_rate": 7.970965765637011e-05,
+      "loss": 0.4577,
+      "step": 468
+    },
+    {
+      "epoch": 0.6748201438848921,
+      "grad_norm": 0.692146495435072,
+      "learning_rate": 7.970480444014244e-05,
+      "loss": 0.4499,
+      "step": 469
+    },
+    {
+      "epoch": 0.6762589928057554,
+      "grad_norm": 0.6645295929196283,
+      "learning_rate": 7.969991114767114e-05,
+      "loss": 0.465,
+      "step": 470
+    },
+    {
+      "epoch": 0.6776978417266187,
+      "grad_norm": 0.6729940725627133,
+      "learning_rate": 7.969497778389534e-05,
+      "loss": 0.4536,
+      "step": 471
+    },
+    {
+      "epoch": 0.679136690647482,
+      "grad_norm": 0.6771506293473587,
+      "learning_rate": 7.969000435379454e-05,
+      "loss": 0.4508,
+      "step": 472
+    },
+    {
+      "epoch": 0.6805755395683454,
+      "grad_norm": 0.6495567644754584,
+      "learning_rate": 7.968499086238867e-05,
+      "loss": 0.443,
+      "step": 473
+    },
+    {
+      "epoch": 0.6820143884892086,
+      "grad_norm": 0.6882929173622521,
+      "learning_rate": 7.967993731473815e-05,
+      "loss": 0.448,
+      "step": 474
+    },
+    {
+      "epoch": 0.6834532374100719,
+      "grad_norm": 0.7237880211855974,
+      "learning_rate": 7.96748437159438e-05,
+      "loss": 0.449,
+      "step": 475
+    },
+    {
+      "epoch": 0.6848920863309352,
+      "grad_norm": 0.8347328822613261,
+      "learning_rate": 7.966971007114686e-05,
+      "loss": 0.4523,
+      "step": 476
+    },
+    {
+      "epoch": 0.6863309352517986,
+      "grad_norm": 0.9719198065761699,
+      "learning_rate": 7.966453638552901e-05,
+      "loss": 0.4591,
+      "step": 477
+    },
+    {
+      "epoch": 0.6877697841726619,
+      "grad_norm": 1.040787964834932,
+      "learning_rate": 7.965932266431232e-05,
+      "loss": 0.4499,
+      "step": 478
+    },
+    {
+      "epoch": 0.6892086330935252,
+      "grad_norm": 0.9334315562534466,
+      "learning_rate": 7.96540689127593e-05,
+      "loss": 0.4545,
+      "step": 479
+    },
+    {
+      "epoch": 0.6906474820143885,
+      "grad_norm": 0.845279321333729,
+      "learning_rate": 7.964877513617285e-05,
+      "loss": 0.4475,
+      "step": 480
+    },
+    {
+      "epoch": 0.6920863309352518,
+      "grad_norm": 0.7979150737125542,
+      "learning_rate": 7.964344133989627e-05,
+      "loss": 0.4508,
+      "step": 481
+    },
+    {
+      "epoch": 0.6935251798561151,
+      "grad_norm": 0.8147584264372341,
+      "learning_rate": 7.963806752931324e-05,
+      "loss": 0.4572,
+      "step": 482
+    },
+    {
+      "epoch": 0.6949640287769784,
+      "grad_norm": 0.7981752284101542,
+      "learning_rate": 7.963265370984786e-05,
+      "loss": 0.4459,
+      "step": 483
+    },
+    {
+      "epoch": 0.6964028776978417,
+      "grad_norm": 0.5970863793585182,
+      "learning_rate": 7.962719988696458e-05,
+      "loss": 0.4515,
+      "step": 484
+    },
+    {
+      "epoch": 0.697841726618705,
+      "grad_norm": 0.4620326784454222,
+      "learning_rate": 7.962170606616826e-05,
+      "loss": 0.4476,
+      "step": 485
+    },
+    {
+      "epoch": 0.6992805755395683,
+      "grad_norm": 0.6131933586653694,
+      "learning_rate": 7.96161722530041e-05,
+      "loss": 0.4539,
+      "step": 486
+    },
+    {
+      "epoch": 0.7007194244604317,
+      "grad_norm": 0.6436228395416139,
+      "learning_rate": 7.96105984530577e-05,
+      "loss": 0.4441,
+      "step": 487
+    },
+    {
+      "epoch": 0.702158273381295,
+      "grad_norm": 0.6941374919640537,
+      "learning_rate": 7.9604984671955e-05,
+      "loss": 0.4598,
+      "step": 488
+    },
+    {
+      "epoch": 0.7035971223021583,
+      "grad_norm": 0.8557240788774572,
+      "learning_rate": 7.959933091536227e-05,
+      "loss": 0.4587,
+      "step": 489
+    },
+    {
+      "epoch": 0.7050359712230215,
+      "grad_norm": 0.8250817581367761,
+      "learning_rate": 7.95936371889862e-05,
+      "loss": 0.4522,
+      "step": 490
+    },
+    {
+      "epoch": 0.7064748201438849,
+      "grad_norm": 0.7243783703842612,
+      "learning_rate": 7.958790349857375e-05,
+      "loss": 0.446,
+      "step": 491
+    },
+    {
+      "epoch": 0.7079136690647482,
+      "grad_norm": 0.6328054932553656,
+      "learning_rate": 7.958212984991226e-05,
+      "loss": 0.4511,
+      "step": 492
+    },
+    {
+      "epoch": 0.7093525179856115,
+      "grad_norm": 0.6050868740033291,
+      "learning_rate": 7.957631624882938e-05,
+      "loss": 0.4478,
+      "step": 493
+    },
+    {
+      "epoch": 0.7107913669064748,
+      "grad_norm": 0.6145153359274808,
+      "learning_rate": 7.957046270119313e-05,
+      "loss": 0.4511,
+      "step": 494
+    },
+    {
+      "epoch": 0.7122302158273381,
+      "grad_norm": 0.4955750148447891,
+      "learning_rate": 7.956456921291178e-05,
+      "loss": 0.4437,
+      "step": 495
+    },
+    {
+      "epoch": 0.7136690647482015,
+      "grad_norm": 0.4658103703985101,
+      "learning_rate": 7.955863578993396e-05,
+      "loss": 0.4485,
+      "step": 496
+    },
+    {
+      "epoch": 0.7151079136690648,
+      "grad_norm": 0.4789906846468974,
+      "learning_rate": 7.955266243824864e-05,
+      "loss": 0.4555,
+      "step": 497
+    },
+    {
+      "epoch": 0.7165467625899281,
+      "grad_norm": 0.3628724330812125,
+      "learning_rate": 7.954664916388499e-05,
+      "loss": 0.4484,
+      "step": 498
+    },
+    {
+      "epoch": 0.7179856115107913,
+      "grad_norm": 0.38832507873774125,
+      "learning_rate": 7.954059597291257e-05,
+      "loss": 0.4471,
+      "step": 499
+    },
+    {
+      "epoch": 0.7194244604316546,
+      "grad_norm": 0.4895740944847101,
+      "learning_rate": 7.953450287144121e-05,
+      "loss": 0.4527,
+      "step": 500
+    },
+    {
+      "epoch": 0.720863309352518,
+      "grad_norm": 0.560521301328584,
+      "learning_rate": 7.952836986562099e-05,
+      "loss": 0.4527,
+      "step": 501
+    },
+    {
+      "epoch": 0.7223021582733813,
+      "grad_norm": 0.5655012201487319,
+      "learning_rate": 7.952219696164231e-05,
+      "loss": 0.446,
+      "step": 502
+    },
+    {
+      "epoch": 0.7237410071942446,
+      "grad_norm": 0.6256450052744676,
+      "learning_rate": 7.95159841657358e-05,
+      "loss": 0.4578,
+      "step": 503
+    },
+    {
+      "epoch": 0.7251798561151079,
+      "grad_norm": 0.843383844614636,
+      "learning_rate": 7.950973148417239e-05,
+      "loss": 0.456,
+      "step": 504
+    },
+    {
+      "epoch": 0.7266187050359713,
+      "grad_norm": 1.2252756875756765,
+      "learning_rate": 7.950343892326327e-05,
+      "loss": 0.4481,
+      "step": 505
+    },
+    {
+      "epoch": 0.7280575539568346,
+      "grad_norm": 0.81689007020895,
+      "learning_rate": 7.949710648935984e-05,
+      "loss": 0.4444,
+      "step": 506
+    },
+    {
+      "epoch": 0.7294964028776978,
+      "grad_norm": 0.5386696162694277,
+      "learning_rate": 7.949073418885378e-05,
+      "loss": 0.439,
+      "step": 507
+    },
+    {
+      "epoch": 0.7309352517985611,
+      "grad_norm": 0.4174766191896591,
+      "learning_rate": 7.948432202817703e-05,
+      "loss": 0.452,
+      "step": 508
+    },
+    {
+      "epoch": 0.7323741007194244,
+      "grad_norm": 0.6394814206881526,
+      "learning_rate": 7.94778700138017e-05,
+      "loss": 0.4457,
+      "step": 509
+    },
+    {
+      "epoch": 0.7338129496402878,
+      "grad_norm": 0.7941586105291152,
+      "learning_rate": 7.947137815224018e-05,
+      "loss": 0.4562,
+      "step": 510
+    },
+    {
+      "epoch": 0.7352517985611511,
+      "grad_norm": 0.8189576364572564,
+      "learning_rate": 7.946484645004508e-05,
+      "loss": 0.4598,
+      "step": 511
+    },
+    {
+      "epoch": 0.7366906474820144,
+      "grad_norm": 0.740740940198565,
+      "learning_rate": 7.945827491380916e-05,
+      "loss": 0.4477,
+      "step": 512
+    },
+    {
+      "epoch": 0.7381294964028777,
+      "grad_norm": 0.6823905757803396,
+      "learning_rate": 7.945166355016548e-05,
+      "loss": 0.4514,
+      "step": 513
+    },
+    {
+      "epoch": 0.739568345323741,
+      "grad_norm": 0.6897531675747028,
+      "learning_rate": 7.944501236578722e-05,
+      "loss": 0.4437,
+      "step": 514
+    },
+    {
+      "epoch": 0.7410071942446043,
+      "grad_norm": 0.6769501230737464,
+      "learning_rate": 7.943832136738783e-05,
+      "loss": 0.4474,
+      "step": 515
+    },
+    {
+      "epoch": 0.7424460431654676,
+      "grad_norm": 0.6940470611142417,
+      "learning_rate": 7.943159056172084e-05,
+      "loss": 0.4568,
+      "step": 516
+    },
+    {
+      "epoch": 0.7438848920863309,
+      "grad_norm": 0.6391637366081594,
+      "learning_rate": 7.942481995558007e-05,
+      "loss": 0.4407,
+      "step": 517
+    },
+    {
+      "epoch": 0.7453237410071942,
+      "grad_norm": 0.7123591035777244,
+      "learning_rate": 7.941800955579946e-05,
+      "loss": 0.4617,
+      "step": 518
+    },
+    {
+      "epoch": 0.7467625899280576,
+      "grad_norm": 0.8389132097956593,
+      "learning_rate": 7.941115936925311e-05,
+      "loss": 0.4465,
+      "step": 519
+    },
+    {
+      "epoch": 0.7482014388489209,
+      "grad_norm": 0.8571734890511532,
+      "learning_rate": 7.940426940285529e-05,
+      "loss": 0.4475,
+      "step": 520
+    },
+    {
+      "epoch": 0.7496402877697842,
+      "grad_norm": 0.8845916711288304,
+      "learning_rate": 7.939733966356042e-05,
+      "loss": 0.4404,
+      "step": 521
+    },
+    {
+      "epoch": 0.7510791366906475,
+      "grad_norm": 0.8603567525751223,
+      "learning_rate": 7.939037015836308e-05,
+      "loss": 0.4549,
+      "step": 522
+    },
+    {
+      "epoch": 0.7525179856115108,
+      "grad_norm": 0.6545784514861386,
+      "learning_rate": 7.938336089429796e-05,
+      "loss": 0.4436,
+      "step": 523
+    },
+    {
+      "epoch": 0.753956834532374,
+      "grad_norm": 0.46746127864891357,
+      "learning_rate": 7.937631187843991e-05,
+      "loss": 0.4346,
+      "step": 524
+    },
+    {
+      "epoch": 0.7553956834532374,
+      "grad_norm": 0.6535704298488452,
+      "learning_rate": 7.936922311790388e-05,
+      "loss": 0.4457,
+      "step": 525
+    },
+    {
+      "epoch": 0.7568345323741007,
+      "grad_norm": 0.8604580580713901,
+      "learning_rate": 7.936209461984495e-05,
+      "loss": 0.4554,
+      "step": 526
+    },
+    {
+      "epoch": 0.758273381294964,
+      "grad_norm": 0.8596757460031823,
+      "learning_rate": 7.935492639145831e-05,
+      "loss": 0.4547,
+      "step": 527
+    },
+    {
+      "epoch": 0.7597122302158273,
+      "grad_norm": 0.6855811230205074,
+      "learning_rate": 7.934771843997922e-05,
+      "loss": 0.4527,
+      "step": 528
+    },
+    {
+      "epoch": 0.7611510791366907,
+      "grad_norm": 0.4247306668156656,
+      "learning_rate": 7.934047077268311e-05,
+      "loss": 0.4529,
+      "step": 529
+    },
+    {
+      "epoch": 0.762589928057554,
+      "grad_norm": 0.5480303698838035,
+      "learning_rate": 7.93331833968854e-05,
+      "loss": 0.4467,
+      "step": 530
+    },
+    {
+      "epoch": 0.7640287769784173,
+      "grad_norm": 0.7585712280153364,
+      "learning_rate": 7.932585631994168e-05,
+      "loss": 0.4381,
+      "step": 531
+    },
+    {
+      "epoch": 0.7654676258992805,
+      "grad_norm": 0.8302009985547316,
+      "learning_rate": 7.931848954924754e-05,
+      "loss": 0.449,
+      "step": 532
+    },
+    {
+      "epoch": 0.7669064748201438,
+      "grad_norm": 0.7819737375461054,
+      "learning_rate": 7.931108309223868e-05,
+      "loss": 0.4474,
+      "step": 533
+    },
+    {
+      "epoch": 0.7683453237410072,
+      "grad_norm": 0.6155414299458252,
+      "learning_rate": 7.930363695639085e-05,
+      "loss": 0.4444,
+      "step": 534
+    },
+    {
+      "epoch": 0.7697841726618705,
+      "grad_norm": 0.6867692350658405,
+      "learning_rate": 7.929615114921984e-05,
+      "loss": 0.4431,
+      "step": 535
+    },
+    {
+      "epoch": 0.7712230215827338,
+      "grad_norm": 0.8520333131629818,
+      "learning_rate": 7.92886256782815e-05,
+      "loss": 0.4495,
+      "step": 536
+    },
+    {
+      "epoch": 0.7726618705035971,
+      "grad_norm": 0.928867857208261,
+      "learning_rate": 7.928106055117168e-05,
+      "loss": 0.4447,
+      "step": 537
+    },
+    {
+      "epoch": 0.7741007194244605,
+      "grad_norm": 0.8792475479227084,
+      "learning_rate": 7.927345577552627e-05,
+      "loss": 0.4361,
+      "step": 538
+    },
+    {
+      "epoch": 0.7755395683453238,
+      "grad_norm": 0.8503296097274649,
+      "learning_rate": 7.926581135902122e-05,
+      "loss": 0.4404,
+      "step": 539
+    },
+    {
+      "epoch": 0.7769784172661871,
+      "grad_norm": 0.8781035666975411,
+      "learning_rate": 7.925812730937245e-05,
+      "loss": 0.4395,
+      "step": 540
+    },
+    {
+      "epoch": 0.7784172661870503,
+      "grad_norm": 0.7640793827684093,
+      "learning_rate": 7.92504036343359e-05,
+      "loss": 0.4418,
+      "step": 541
+    },
+    {
+      "epoch": 0.7798561151079136,
+      "grad_norm": 0.5452920148681172,
+      "learning_rate": 7.924264034170747e-05,
+      "loss": 0.439,
+      "step": 542
+    },
+    {
+      "epoch": 0.781294964028777,
+      "grad_norm": 0.4078470063626378,
+      "learning_rate": 7.923483743932311e-05,
+      "loss": 0.438,
+      "step": 543
+    },
+    {
+      "epoch": 0.7827338129496403,
+      "grad_norm": 0.5268504728636281,
+      "learning_rate": 7.922699493505871e-05,
+      "loss": 0.4445,
+      "step": 544
+    },
+    {
+      "epoch": 0.7841726618705036,
+      "grad_norm": 0.5668670210410232,
+      "learning_rate": 7.921911283683013e-05,
+      "loss": 0.4481,
+      "step": 545
+    },
+    {
+      "epoch": 0.7856115107913669,
+      "grad_norm": 0.5041645002269407,
+      "learning_rate": 7.921119115259322e-05,
+      "loss": 0.4447,
+      "step": 546
+    },
+    {
+      "epoch": 0.7870503597122303,
+      "grad_norm": 0.5076344748214742,
+      "learning_rate": 7.920322989034377e-05,
+      "loss": 0.4565,
+      "step": 547
+    },
+    {
+      "epoch": 0.7884892086330936,
+      "grad_norm": 0.6064613677143496,
+      "learning_rate": 7.919522905811752e-05,
+      "loss": 0.4321,
+      "step": 548
+    },
+    {
+      "epoch": 0.7899280575539568,
+      "grad_norm": 0.7411255790888717,
+      "learning_rate": 7.918718866399012e-05,
+      "loss": 0.4495,
+      "step": 549
+    },
+    {
+      "epoch": 0.7913669064748201,
+      "grad_norm": 0.8797687912923331,
+      "learning_rate": 7.917910871607723e-05,
+      "loss": 0.4509,
+      "step": 550
+    },
+    {
+      "epoch": 0.7928057553956834,
+      "grad_norm": 0.9548767462700447,
+      "learning_rate": 7.917098922253436e-05,
+      "loss": 0.4423,
+      "step": 551
+    },
+    {
+      "epoch": 0.7942446043165468,
+      "grad_norm": 0.9972894572478253,
+      "learning_rate": 7.916283019155696e-05,
+      "loss": 0.4503,
+      "step": 552
+    },
+    {
+      "epoch": 0.7956834532374101,
+      "grad_norm": 0.9893577525224617,
+      "learning_rate": 7.915463163138041e-05,
+      "loss": 0.4595,
+      "step": 553
+    },
+    {
+      "epoch": 0.7971223021582734,
+      "grad_norm": 0.8972701855596109,
+      "learning_rate": 7.914639355027995e-05,
+      "loss": 0.4391,
+      "step": 554
+    },
+    {
+      "epoch": 0.7985611510791367,
+      "grad_norm": 0.7835065881851512,
+      "learning_rate": 7.913811595657072e-05,
+      "loss": 0.4372,
+      "step": 555
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.6394121885285287,
+      "learning_rate": 7.912979885860776e-05,
+      "loss": 0.452,
+      "step": 556
+    },
+    {
+      "epoch": 0.8014388489208633,
+      "grad_norm": 0.6801870197183523,
+      "learning_rate": 7.912144226478598e-05,
+      "loss": 0.447,
+      "step": 557
+    },
+    {
+      "epoch": 0.8028776978417266,
+      "grad_norm": 0.7980152451628971,
+      "learning_rate": 7.911304618354015e-05,
+      "loss": 0.4487,
+      "step": 558
+    },
+    {
+      "epoch": 0.8043165467625899,
+      "grad_norm": 0.7279067435889713,
+      "learning_rate": 7.910461062334488e-05,
+      "loss": 0.4328,
+      "step": 559
+    },
+    {
+      "epoch": 0.8057553956834532,
+      "grad_norm": 0.6438312702522009,
+      "learning_rate": 7.909613559271467e-05,
+      "loss": 0.4527,
+      "step": 560
+    },
+    {
+      "epoch": 0.8071942446043165,
+      "grad_norm": 0.5888117923156623,
+      "learning_rate": 7.908762110020382e-05,
+      "loss": 0.4484,
+      "step": 561
+    },
+    {
+      "epoch": 0.8086330935251799,
+      "grad_norm": 0.5027698438536533,
+      "learning_rate": 7.907906715440649e-05,
+      "loss": 0.441,
+      "step": 562
+    },
+    {
+      "epoch": 0.8100719424460432,
+      "grad_norm": 0.33582201172495263,
+      "learning_rate": 7.907047376395661e-05,
+      "loss": 0.448,
+      "step": 563
+    },
+    {
+      "epoch": 0.8115107913669065,
+      "grad_norm": 0.37382020125050747,
+      "learning_rate": 7.906184093752801e-05,
+      "loss": 0.447,
+      "step": 564
+    },
+    {
+      "epoch": 0.8129496402877698,
+      "grad_norm": 0.535153938679081,
+      "learning_rate": 7.905316868383425e-05,
+      "loss": 0.4549,
+      "step": 565
+    },
+    {
+      "epoch": 0.814388489208633,
+      "grad_norm": 0.6398796518984677,
+      "learning_rate": 7.904445701162872e-05,
+      "loss": 0.4415,
+      "step": 566
+    },
+    {
+      "epoch": 0.8158273381294964,
+      "grad_norm": 0.7020604505602379,
+      "learning_rate": 7.903570592970458e-05,
+      "loss": 0.4401,
+      "step": 567
+    },
+    {
+      "epoch": 0.8172661870503597,
+      "grad_norm": 0.6924476831387525,
+      "learning_rate": 7.902691544689479e-05,
+      "loss": 0.4373,
+      "step": 568
+    },
+    {
+      "epoch": 0.818705035971223,
+      "grad_norm": 0.720414887360444,
+      "learning_rate": 7.901808557207206e-05,
+      "loss": 0.4413,
+      "step": 569
+    },
+    {
+      "epoch": 0.8201438848920863,
+      "grad_norm": 0.795139580874903,
+      "learning_rate": 7.900921631414887e-05,
+      "loss": 0.4531,
+      "step": 570
+    },
+    {
+      "epoch": 0.8215827338129497,
+      "grad_norm": 0.8747103811649327,
+      "learning_rate": 7.900030768207746e-05,
+      "loss": 0.4496,
+      "step": 571
+    },
+    {
+      "epoch": 0.823021582733813,
+      "grad_norm": 0.961793877132737,
+      "learning_rate": 7.899135968484979e-05,
+      "loss": 0.4431,
+      "step": 572
+    },
+    {
+      "epoch": 0.8244604316546763,
+      "grad_norm": 0.9721856253540773,
+      "learning_rate": 7.898237233149758e-05,
+      "loss": 0.4417,
+      "step": 573
+    },
+    {
+      "epoch": 0.8258992805755395,
+      "grad_norm": 0.8357185389253394,
+      "learning_rate": 7.897334563109225e-05,
+      "loss": 0.4418,
+      "step": 574
+    },
+    {
+      "epoch": 0.8273381294964028,
+      "grad_norm": 0.6615605307941537,
+      "learning_rate": 7.896427959274494e-05,
+      "loss": 0.4419,
+      "step": 575
+    },
+    {
+      "epoch": 0.8287769784172662,
+      "grad_norm": 0.48424946567746674,
+      "learning_rate": 7.895517422560651e-05,
+      "loss": 0.4401,
+      "step": 576
+    },
+    {
+      "epoch": 0.8302158273381295,
+      "grad_norm": 0.42481938118484464,
+      "learning_rate": 7.89460295388675e-05,
+      "loss": 0.4467,
+      "step": 577
+    },
+    {
+      "epoch": 0.8316546762589928,
+      "grad_norm": 0.567836520865301,
+      "learning_rate": 7.893684554175817e-05,
+      "loss": 0.4403,
+      "step": 578
+    },
+    {
+      "epoch": 0.8330935251798561,
+      "grad_norm": 0.6493035442382297,
+      "learning_rate": 7.892762224354839e-05,
+      "loss": 0.4487,
+      "step": 579
+    },
+    {
+      "epoch": 0.8345323741007195,
+      "grad_norm": 0.6415333815536135,
+      "learning_rate": 7.891835965354778e-05,
+      "loss": 0.4423,
+      "step": 580
+    },
+    {
+      "epoch": 0.8359712230215828,
+      "grad_norm": 0.5888206024793224,
+      "learning_rate": 7.890905778110557e-05,
+      "loss": 0.4371,
+      "step": 581
+    },
+    {
+      "epoch": 0.837410071942446,
+      "grad_norm": 0.4362844457550126,
+      "learning_rate": 7.889971663561065e-05,
+      "loss": 0.441,
+      "step": 582
+    },
+    {
+      "epoch": 0.8388489208633093,
+      "grad_norm": 0.516045485925217,
+      "learning_rate": 7.889033622649155e-05,
+      "loss": 0.442,
+      "step": 583
+    },
+    {
+      "epoch": 0.8402877697841726,
+      "grad_norm": 0.6938260293020249,
+      "learning_rate": 7.888091656321644e-05,
+      "loss": 0.4439,
+      "step": 584
+    },
+    {
+      "epoch": 0.841726618705036,
+      "grad_norm": 0.8866903216895299,
+      "learning_rate": 7.88714576552931e-05,
+      "loss": 0.4445,
+      "step": 585
+    },
+    {
+      "epoch": 0.8431654676258993,
+      "grad_norm": 0.9571652867714048,
+      "learning_rate": 7.886195951226892e-05,
+      "loss": 0.4561,
+      "step": 586
+    },
+    {
+      "epoch": 0.8446043165467626,
+      "grad_norm": 0.9639686794717786,
+      "learning_rate": 7.885242214373091e-05,
+      "loss": 0.4479,
+      "step": 587
+    },
+    {
+      "epoch": 0.8460431654676259,
+      "grad_norm": 0.9415290789476168,
+      "learning_rate": 7.884284555930564e-05,
+      "loss": 0.433,
+      "step": 588
+    },
+    {
+      "epoch": 0.8474820143884892,
+      "grad_norm": 0.8602482684889639,
+      "learning_rate": 7.883322976865932e-05,
+      "loss": 0.4436,
+      "step": 589
+    },
+    {
+      "epoch": 0.8489208633093526,
+      "grad_norm": 0.7968024067108611,
+      "learning_rate": 7.882357478149767e-05,
+      "loss": 0.443,
+      "step": 590
+    },
+    {
+      "epoch": 0.8503597122302158,
+      "grad_norm": 0.6837876068072184,
+      "learning_rate": 7.8813880607566e-05,
+      "loss": 0.4454,
+      "step": 591
+    },
+    {
+      "epoch": 0.8517985611510791,
+      "grad_norm": 0.6487747345944853,
+      "learning_rate": 7.880414725664918e-05,
+      "loss": 0.439,
+      "step": 592
+    },
+    {
+      "epoch": 0.8532374100719424,
+      "grad_norm": 0.6613059271967316,
+      "learning_rate": 7.879437473857161e-05,
+      "loss": 0.4384,
+      "step": 593
+    },
+    {
+      "epoch": 0.8546762589928057,
+      "grad_norm": 0.7117235558673315,
+      "learning_rate": 7.878456306319723e-05,
+      "loss": 0.434,
+      "step": 594
+    },
+    {
+      "epoch": 0.8561151079136691,
+      "grad_norm": 0.7068550862253427,
+      "learning_rate": 7.877471224042952e-05,
+      "loss": 0.4433,
+      "step": 595
+    },
+    {
+      "epoch": 0.8575539568345324,
+      "grad_norm": 0.5607720123908527,
+      "learning_rate": 7.876482228021144e-05,
+      "loss": 0.4498,
+      "step": 596
+    },
+    {
+      "epoch": 0.8589928057553957,
+      "grad_norm": 0.4156934120702612,
+      "learning_rate": 7.875489319252549e-05,
+      "loss": 0.4403,
+      "step": 597
+    },
+    {
+      "epoch": 0.860431654676259,
+      "grad_norm": 0.5178057413217351,
+      "learning_rate": 7.874492498739362e-05,
+      "loss": 0.4391,
+      "step": 598
+    },
+    {
+      "epoch": 0.8618705035971223,
+      "grad_norm": 0.612546753914525,
+      "learning_rate": 7.87349176748773e-05,
+      "loss": 0.4448,
+      "step": 599
+    },
+    {
+      "epoch": 0.8633093525179856,
+      "grad_norm": 0.554902923548074,
+      "learning_rate": 7.872487126507747e-05,
+      "loss": 0.4418,
+      "step": 600
+    },
+    {
+      "epoch": 0.8647482014388489,
+      "grad_norm": 0.46737793048588583,
+      "learning_rate": 7.87147857681345e-05,
+      "loss": 0.4471,
+      "step": 601
+    },
+    {
+      "epoch": 0.8661870503597122,
+      "grad_norm": 0.5002389683481634,
+      "learning_rate": 7.870466119422826e-05,
+      "loss": 0.4473,
+      "step": 602
+    },
+    {
+      "epoch": 0.8676258992805755,
+      "grad_norm": 0.4864787553709012,
+      "learning_rate": 7.869449755357803e-05,
+      "loss": 0.4337,
+      "step": 603
+    },
+    {
+      "epoch": 0.8690647482014389,
+      "grad_norm": 0.4529141193587548,
+      "learning_rate": 7.868429485644252e-05,
+      "loss": 0.443,
+      "step": 604
+    },
+    {
+      "epoch": 0.8705035971223022,
+      "grad_norm": 0.5213553189981291,
+      "learning_rate": 7.86740531131199e-05,
+      "loss": 0.4453,
+      "step": 605
+    },
+    {
+      "epoch": 0.8719424460431655,
+      "grad_norm": 0.614473071399125,
+      "learning_rate": 7.866377233394771e-05,
+      "loss": 0.4323,
+      "step": 606
+    },
+    {
+      "epoch": 0.8733812949640288,
+      "grad_norm": 0.5923931578535838,
+      "learning_rate": 7.865345252930291e-05,
+      "loss": 0.4405,
+      "step": 607
+    },
+    {
+      "epoch": 0.874820143884892,
+      "grad_norm": 0.4877696955164087,
+      "learning_rate": 7.864309370960184e-05,
+      "loss": 0.4445,
+      "step": 608
+    },
+    {
+      "epoch": 0.8762589928057554,
+      "grad_norm": 0.4759437715506379,
+      "learning_rate": 7.863269588530023e-05,
+      "loss": 0.4424,
+      "step": 609
+    },
+    {
+      "epoch": 0.8776978417266187,
+      "grad_norm": 0.5031051834295212,
+      "learning_rate": 7.862225906689319e-05,
+      "loss": 0.4451,
+      "step": 610
+    },
+    {
+      "epoch": 0.879136690647482,
+      "grad_norm": 0.6237107752983813,
+      "learning_rate": 7.861178326491514e-05,
+      "loss": 0.442,
+      "step": 611
+    },
+    {
+      "epoch": 0.8805755395683453,
+      "grad_norm": 0.8022485843670729,
+      "learning_rate": 7.860126848993992e-05,
+      "loss": 0.434,
+      "step": 612
+    },
+    {
+      "epoch": 0.8820143884892087,
+      "grad_norm": 0.8512996518757874,
+      "learning_rate": 7.859071475258065e-05,
+      "loss": 0.4496,
+      "step": 613
+    },
+    {
+      "epoch": 0.883453237410072,
+      "grad_norm": 0.8482540481368318,
+      "learning_rate": 7.85801220634898e-05,
+      "loss": 0.4328,
+      "step": 614
+    },
+    {
+      "epoch": 0.8848920863309353,
+      "grad_norm": 0.8197679584683998,
+      "learning_rate": 7.856949043335917e-05,
+      "loss": 0.4369,
+      "step": 615
+    },
+    {
+      "epoch": 0.8863309352517985,
+      "grad_norm": 0.9063289810672892,
+      "learning_rate": 7.855881987291983e-05,
+      "loss": 0.4383,
+      "step": 616
+    },
+    {
+      "epoch": 0.8877697841726618,
+      "grad_norm": 1.0126671490109374,
+      "learning_rate": 7.854811039294216e-05,
+      "loss": 0.4439,
+      "step": 617
+    },
+    {
+      "epoch": 0.8892086330935252,
+      "grad_norm": 0.9523707023332784,
+      "learning_rate": 7.853736200423584e-05,
+      "loss": 0.4485,
+      "step": 618
+    },
+    {
+      "epoch": 0.8906474820143885,
+      "grad_norm": 0.726533354436605,
+      "learning_rate": 7.852657471764983e-05,
+      "loss": 0.4396,
+      "step": 619
+    },
+    {
+      "epoch": 0.8920863309352518,
+      "grad_norm": 0.5317978393118742,
+      "learning_rate": 7.851574854407228e-05,
+      "loss": 0.436,
+      "step": 620
+    },
+    {
+      "epoch": 0.8935251798561151,
+      "grad_norm": 0.5176552199820211,
+      "learning_rate": 7.85048834944307e-05,
+      "loss": 0.4426,
+      "step": 621
+    },
+    {
+      "epoch": 0.8949640287769784,
+      "grad_norm": 0.5084489605107904,
+      "learning_rate": 7.849397957969173e-05,
+      "loss": 0.4477,
+      "step": 622
+    },
+    {
+      "epoch": 0.8964028776978418,
+      "grad_norm": 0.581415487862338,
+      "learning_rate": 7.848303681086134e-05,
+      "loss": 0.4391,
+      "step": 623
+    },
+    {
+      "epoch": 0.897841726618705,
+      "grad_norm": 0.5787490361092864,
+      "learning_rate": 7.847205519898461e-05,
+      "loss": 0.4428,
+      "step": 624
+    },
+    {
+      "epoch": 0.8992805755395683,
+      "grad_norm": 0.4641397927785643,
+      "learning_rate": 7.846103475514595e-05,
+      "loss": 0.4428,
+      "step": 625
+    },
+    {
+      "epoch": 0.9007194244604316,
+      "grad_norm": 0.4145065936872774,
+      "learning_rate": 7.844997549046886e-05,
+      "loss": 0.4384,
+      "step": 626
+    },
+    {
+      "epoch": 0.902158273381295,
+      "grad_norm": 0.4596652659908233,
+      "learning_rate": 7.843887741611608e-05,
+      "loss": 0.4437,
+      "step": 627
+    },
+    {
+      "epoch": 0.9035971223021583,
+      "grad_norm": 0.5582975704307552,
+      "learning_rate": 7.842774054328949e-05,
+      "loss": 0.4351,
+      "step": 628
+    },
+    {
+      "epoch": 0.9050359712230216,
+      "grad_norm": 0.6206476700620335,
+      "learning_rate": 7.841656488323017e-05,
+      "loss": 0.4373,
+      "step": 629
+    },
+    {
+      "epoch": 0.9064748201438849,
+      "grad_norm": 0.5875705133755592,
+      "learning_rate": 7.840535044721832e-05,
+      "loss": 0.4406,
+      "step": 630
+    },
+    {
+      "epoch": 0.9079136690647482,
+      "grad_norm": 0.6055808749949936,
+      "learning_rate": 7.839409724657327e-05,
+      "loss": 0.439,
+      "step": 631
+    },
+    {
+      "epoch": 0.9093525179856116,
+      "grad_norm": 0.5157141811047572,
+      "learning_rate": 7.838280529265353e-05,
+      "loss": 0.4503,
+      "step": 632
+    },
+    {
+      "epoch": 0.9107913669064748,
+      "grad_norm": 0.3784451801413552,
+      "learning_rate": 7.837147459685666e-05,
+      "loss": 0.4359,
+      "step": 633
+    },
+    {
+      "epoch": 0.9122302158273381,
+      "grad_norm": 0.42682466417211756,
+      "learning_rate": 7.836010517061937e-05,
+      "loss": 0.4371,
+      "step": 634
+    },
+    {
+      "epoch": 0.9136690647482014,
+      "grad_norm": 0.483018868022784,
+      "learning_rate": 7.834869702541742e-05,
+      "loss": 0.4485,
+      "step": 635
+    },
+    {
+      "epoch": 0.9151079136690647,
+      "grad_norm": 0.373002785791148,
+      "learning_rate": 7.833725017276573e-05,
+      "loss": 0.431,
+      "step": 636
+    },
+    {
+      "epoch": 0.9165467625899281,
+      "grad_norm": 0.5226798778671032,
+      "learning_rate": 7.83257646242182e-05,
+      "loss": 0.4411,
+      "step": 637
+    },
+    {
+      "epoch": 0.9179856115107914,
+      "grad_norm": 0.6245402002062002,
+      "learning_rate": 7.831424039136783e-05,
+      "loss": 0.4352,
+      "step": 638
+    },
+    {
+      "epoch": 0.9194244604316547,
+      "grad_norm": 0.7298061828081333,
+      "learning_rate": 7.830267748584666e-05,
+      "loss": 0.4375,
+      "step": 639
+    },
+    {
+      "epoch": 0.920863309352518,
+      "grad_norm": 0.8633694543394981,
+      "learning_rate": 7.829107591932578e-05,
+      "loss": 0.4351,
+      "step": 640
+    },
+    {
+      "epoch": 0.9223021582733812,
+      "grad_norm": 1.1393206979864197,
+      "learning_rate": 7.82794357035153e-05,
+      "loss": 0.4522,
+      "step": 641
+    },
+    {
+      "epoch": 0.9237410071942446,
+      "grad_norm": 0.6438360260021782,
+      "learning_rate": 7.82677568501643e-05,
+      "loss": 0.4421,
+      "step": 642
+    },
+    {
+      "epoch": 0.9251798561151079,
+      "grad_norm": 0.42681014197527684,
+      "learning_rate": 7.82560393710609e-05,
+      "loss": 0.4405,
+      "step": 643
+    },
+    {
+      "epoch": 0.9266187050359712,
+      "grad_norm": 0.47523960417453875,
+      "learning_rate": 7.824428327803221e-05,
+      "loss": 0.4307,
+      "step": 644
+    },
+    {
+      "epoch": 0.9280575539568345,
+      "grad_norm": 0.5290286468933811,
+      "learning_rate": 7.823248858294428e-05,
+      "loss": 0.4381,
+      "step": 645
+    },
+    {
+      "epoch": 0.9294964028776979,
+      "grad_norm": 0.5520431122422456,
+      "learning_rate": 7.822065529770216e-05,
+      "loss": 0.4359,
+      "step": 646
+    },
+    {
+      "epoch": 0.9309352517985612,
+      "grad_norm": 0.5624293793113554,
+      "learning_rate": 7.820878343424984e-05,
+      "loss": 0.4288,
+      "step": 647
+    },
+    {
+      "epoch": 0.9323741007194245,
+      "grad_norm": 0.5790328927418427,
+      "learning_rate": 7.819687300457021e-05,
+      "loss": 0.4414,
+      "step": 648
+    },
+    {
+      "epoch": 0.9338129496402877,
+      "grad_norm": 0.6663729371991175,
+      "learning_rate": 7.818492402068517e-05,
+      "loss": 0.4552,
+      "step": 649
+    },
+    {
+      "epoch": 0.935251798561151,
+      "grad_norm": 0.7082272101827812,
+      "learning_rate": 7.817293649465546e-05,
+      "loss": 0.4541,
+      "step": 650
+    },
+    {
+      "epoch": 0.9366906474820144,
+      "grad_norm": 0.7769959445586613,
+      "learning_rate": 7.816091043858076e-05,
+      "loss": 0.4321,
+      "step": 651
+    },
+    {
+      "epoch": 0.9381294964028777,
+      "grad_norm": 0.9309288447563863,
+      "learning_rate": 7.814884586459962e-05,
+      "loss": 0.4442,
+      "step": 652
+    },
+    {
+      "epoch": 0.939568345323741,
+      "grad_norm": 1.0193886938186758,
+      "learning_rate": 7.813674278488949e-05,
+      "loss": 0.4408,
+      "step": 653
+    },
+    {
+      "epoch": 0.9410071942446043,
+      "grad_norm": 0.8100948835729215,
+      "learning_rate": 7.812460121166666e-05,
+      "loss": 0.4373,
+      "step": 654
+    },
+    {
+      "epoch": 0.9424460431654677,
+      "grad_norm": 0.8449929536774763,
+      "learning_rate": 7.81124211571863e-05,
+      "loss": 0.4393,
+      "step": 655
+    },
+    {
+      "epoch": 0.943884892086331,
+      "grad_norm": 0.9862265081898043,
+      "learning_rate": 7.810020263374239e-05,
+      "loss": 0.4304,
+      "step": 656
+    },
+    {
+      "epoch": 0.9453237410071943,
+      "grad_norm": 0.8462187956035074,
+      "learning_rate": 7.808794565366778e-05,
+      "loss": 0.4437,
+      "step": 657
+    },
+    {
+      "epoch": 0.9467625899280575,
+      "grad_norm": 0.6678706409869629,
+      "learning_rate": 7.807565022933412e-05,
+      "loss": 0.437,
+      "step": 658
+    },
+    {
+      "epoch": 0.9482014388489208,
+      "grad_norm": 0.542640476887084,
+      "learning_rate": 7.806331637315183e-05,
+      "loss": 0.4426,
+      "step": 659
+    },
+    {
+      "epoch": 0.9496402877697842,
+      "grad_norm": 0.532435359619077,
+      "learning_rate": 7.805094409757017e-05,
+      "loss": 0.4308,
+      "step": 660
+    },
+    {
+      "epoch": 0.9510791366906475,
+      "grad_norm": 0.6982411367119742,
+      "learning_rate": 7.803853341507715e-05,
+      "loss": 0.445,
+      "step": 661
+    },
+    {
+      "epoch": 0.9525179856115108,
+      "grad_norm": 0.6172363235937145,
+      "learning_rate": 7.802608433819957e-05,
+      "loss": 0.4469,
+      "step": 662
+    },
+    {
+      "epoch": 0.9539568345323741,
+      "grad_norm": 0.4425610817255528,
+      "learning_rate": 7.801359687950292e-05,
+      "loss": 0.4443,
+      "step": 663
+    },
+    {
+      "epoch": 0.9553956834532374,
+      "grad_norm": 0.4900095133638887,
+      "learning_rate": 7.800107105159155e-05,
+      "loss": 0.4307,
+      "step": 664
+    },
+    {
+      "epoch": 0.9568345323741008,
+      "grad_norm": 0.4187982557155431,
+      "learning_rate": 7.798850686710841e-05,
+      "loss": 0.444,
+      "step": 665
+    },
+    {
+      "epoch": 0.958273381294964,
+      "grad_norm": 0.4096966639403132,
+      "learning_rate": 7.797590433873526e-05,
+      "loss": 0.4375,
+      "step": 666
+    },
+    {
+      "epoch": 0.9597122302158273,
+      "grad_norm": 0.467720980260205,
+      "learning_rate": 7.79632634791925e-05,
+      "loss": 0.4436,
+      "step": 667
+    },
+    {
+      "epoch": 0.9611510791366906,
+      "grad_norm": 0.47391654471052125,
+      "learning_rate": 7.795058430123925e-05,
+      "loss": 0.4428,
+      "step": 668
+    },
+    {
+      "epoch": 0.962589928057554,
+      "grad_norm": 0.4792805071275589,
+      "learning_rate": 7.793786681767333e-05,
+      "loss": 0.4389,
+      "step": 669
+    },
+    {
+      "epoch": 0.9640287769784173,
+      "grad_norm": 0.36928231507310566,
+      "learning_rate": 7.792511104133117e-05,
+      "loss": 0.4476,
+      "step": 670
+    },
+    {
+      "epoch": 0.9654676258992806,
+      "grad_norm": 0.316373392013736,
+      "learning_rate": 7.791231698508786e-05,
+      "loss": 0.4457,
+      "step": 671
+    },
+    {
+      "epoch": 0.9669064748201439,
+      "grad_norm": 0.345313584756319,
+      "learning_rate": 7.789948466185718e-05,
+      "loss": 0.4406,
+      "step": 672
+    },
+    {
+      "epoch": 0.9683453237410072,
+      "grad_norm": 0.3893150006205188,
+      "learning_rate": 7.788661408459146e-05,
+      "loss": 0.4281,
+      "step": 673
+    },
+    {
+      "epoch": 0.9697841726618706,
+      "grad_norm": 0.3901165937334357,
+      "learning_rate": 7.787370526628173e-05,
+      "loss": 0.4304,
+      "step": 674
+    },
+    {
+      "epoch": 0.9712230215827338,
+      "grad_norm": 0.3949496609169628,
+      "learning_rate": 7.786075821995754e-05,
+      "loss": 0.4421,
+      "step": 675
+    },
+    {
+      "epoch": 0.9726618705035971,
+      "grad_norm": 0.4315872922327187,
+      "learning_rate": 7.784777295868706e-05,
+      "loss": 0.439,
+      "step": 676
+    },
+    {
+      "epoch": 0.9741007194244604,
+      "grad_norm": 0.4446444112401096,
+      "learning_rate": 7.783474949557704e-05,
+      "loss": 0.4344,
+      "step": 677
+    },
+    {
+      "epoch": 0.9755395683453237,
+      "grad_norm": 0.4579936400268822,
+      "learning_rate": 7.782168784377276e-05,
+      "loss": 0.4311,
+      "step": 678
+    },
+    {
+      "epoch": 0.9769784172661871,
+      "grad_norm": 0.5336441400695344,
+      "learning_rate": 7.780858801645806e-05,
+      "loss": 0.4442,
+      "step": 679
+    },
+    {
+      "epoch": 0.9784172661870504,
+      "grad_norm": 0.6129966638581394,
+      "learning_rate": 7.779545002685535e-05,
+      "loss": 0.4348,
+      "step": 680
+    },
+    {
+      "epoch": 0.9798561151079137,
+      "grad_norm": 0.6015585936775667,
+      "learning_rate": 7.778227388822552e-05,
+      "loss": 0.4497,
+      "step": 681
+    },
+    {
+      "epoch": 0.981294964028777,
+      "grad_norm": 0.6238991019639683,
+      "learning_rate": 7.776905961386793e-05,
+      "loss": 0.4394,
+      "step": 682
+    },
+    {
+      "epoch": 0.9827338129496402,
+      "grad_norm": 0.637066953593209,
+      "learning_rate": 7.77558072171205e-05,
+      "loss": 0.4462,
+      "step": 683
+    },
+    {
+      "epoch": 0.9841726618705036,
+      "grad_norm": 0.6920493127521262,
+      "learning_rate": 7.774251671135961e-05,
+      "loss": 0.442,
+      "step": 684
+    },
+    {
+      "epoch": 0.9856115107913669,
+      "grad_norm": 0.902123798026066,
+      "learning_rate": 7.77291881100001e-05,
+      "loss": 0.4422,
+      "step": 685
+    },
+    {
+      "epoch": 0.9870503597122302,
+      "grad_norm": 1.080075310815761,
+      "learning_rate": 7.771582142649523e-05,
+      "loss": 0.4361,
+      "step": 686
+    },
+    {
+      "epoch": 0.9884892086330935,
+      "grad_norm": 0.7868523641587457,
+      "learning_rate": 7.770241667433677e-05,
+      "loss": 0.4437,
+      "step": 687
+    },
+    {
+      "epoch": 0.9899280575539569,
+      "grad_norm": 0.6176988692747292,
+      "learning_rate": 7.768897386705488e-05,
+      "loss": 0.4344,
+      "step": 688
+    },
+    {
+      "epoch": 0.9913669064748202,
+      "grad_norm": 0.6170658867591173,
+      "learning_rate": 7.767549301821807e-05,
+      "loss": 0.4364,
+      "step": 689
+    },
+    {
+      "epoch": 0.9928057553956835,
+      "grad_norm": 0.48431512071645794,
+      "learning_rate": 7.766197414143333e-05,
+      "loss": 0.4316,
+      "step": 690
+    },
+    {
+      "epoch": 0.9942446043165467,
+      "grad_norm": 0.3441788556944156,
+      "learning_rate": 7.764841725034602e-05,
+      "loss": 0.4369,
+      "step": 691
+    },
+    {
+      "epoch": 0.99568345323741,
+      "grad_norm": 0.36446005498861855,
+      "learning_rate": 7.763482235863985e-05,
+      "loss": 0.439,
+      "step": 692
+    },
+    {
+      "epoch": 0.9971223021582734,
+      "grad_norm": 0.45740062246008695,
+      "learning_rate": 7.762118948003688e-05,
+      "loss": 0.4381,
+      "step": 693
+    },
+    {
+      "epoch": 0.9985611510791367,
+      "grad_norm": 0.4857706549122722,
+      "learning_rate": 7.760751862829754e-05,
+      "loss": 0.4279,
+      "step": 694
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.5428761936290282,
+      "learning_rate": 7.759380981722055e-05,
+      "loss": 0.4366,
+      "step": 695
+    },
+    {
+      "epoch": 1.0014388489208632,
+      "grad_norm": 0.5992478587237964,
+      "learning_rate": 7.758006306064301e-05,
+      "loss": 0.4209,
+      "step": 696
+    },
+    {
+      "epoch": 1.0028776978417266,
+      "grad_norm": 0.5650891647222237,
+      "learning_rate": 7.756627837244023e-05,
+      "loss": 0.416,
+      "step": 697
+    },
+    {
+      "epoch": 1.0043165467625899,
+      "grad_norm": 0.568430795185666,
+      "learning_rate": 7.755245576652588e-05,
+      "loss": 0.415,
+      "step": 698
+    },
+    {
+      "epoch": 1.0057553956834533,
+      "grad_norm": 0.6422965686402028,
+      "learning_rate": 7.753859525685187e-05,
+      "loss": 0.4222,
+      "step": 699
+    },
+    {
+      "epoch": 1.0071942446043165,
+      "grad_norm": 0.6090204933635286,
+      "learning_rate": 7.752469685740838e-05,
+      "loss": 0.4233,
+      "step": 700
+    },
+    {
+      "epoch": 1.00863309352518,
+      "grad_norm": 0.6244656974243654,
+      "learning_rate": 7.751076058222381e-05,
+      "loss": 0.4394,
+      "step": 701
+    },
+    {
+      "epoch": 1.0100719424460431,
+      "grad_norm": 0.6321414492283441,
+      "learning_rate": 7.749678644536485e-05,
+      "loss": 0.4154,
+      "step": 702
+    },
+    {
+      "epoch": 1.0115107913669066,
+      "grad_norm": 0.7184162771194458,
+      "learning_rate": 7.748277446093631e-05,
+      "loss": 0.4215,
+      "step": 703
+    },
+    {
+      "epoch": 1.0129496402877698,
+      "grad_norm": 0.7578461572480496,
+      "learning_rate": 7.746872464308131e-05,
+      "loss": 0.4165,
+      "step": 704
+    },
+    {
+      "epoch": 1.014388489208633,
+      "grad_norm": 0.6677024168677679,
+      "learning_rate": 7.745463700598108e-05,
+      "loss": 0.4174,
+      "step": 705
+    },
+    {
+      "epoch": 1.0158273381294964,
+      "grad_norm": 0.6107255829884857,
+      "learning_rate": 7.744051156385503e-05,
+      "loss": 0.4253,
+      "step": 706
+    },
+    {
+      "epoch": 1.0172661870503596,
+      "grad_norm": 0.6250570156008007,
+      "learning_rate": 7.742634833096077e-05,
+      "loss": 0.4188,
+      "step": 707
+    },
+    {
+      "epoch": 1.018705035971223,
+      "grad_norm": 0.6380171143766701,
+      "learning_rate": 7.741214732159403e-05,
+      "loss": 0.4146,
+      "step": 708
+    },
+    {
+      "epoch": 1.0201438848920863,
+      "grad_norm": 0.6168839916378758,
+      "learning_rate": 7.739790855008867e-05,
+      "loss": 0.4248,
+      "step": 709
+    },
+    {
+      "epoch": 1.0215827338129497,
+      "grad_norm": 0.5726456992958739,
+      "learning_rate": 7.738363203081664e-05,
+      "loss": 0.4205,
+      "step": 710
+    },
+    {
+      "epoch": 1.023021582733813,
+      "grad_norm": 0.5481407715702845,
+      "learning_rate": 7.736931777818805e-05,
+      "loss": 0.4219,
+      "step": 711
+    },
+    {
+      "epoch": 1.0244604316546762,
+      "grad_norm": 0.519092810724968,
+      "learning_rate": 7.735496580665105e-05,
+      "loss": 0.4238,
+      "step": 712
+    },
+    {
+      "epoch": 1.0258992805755396,
+      "grad_norm": 0.5406096531900932,
+      "learning_rate": 7.734057613069188e-05,
+      "loss": 0.4135,
+      "step": 713
+    },
+    {
+      "epoch": 1.0273381294964028,
+      "grad_norm": 0.633851534977381,
+      "learning_rate": 7.73261487648348e-05,
+      "loss": 0.4198,
+      "step": 714
+    },
+    {
+      "epoch": 1.0287769784172662,
+      "grad_norm": 0.6490670741748399,
+      "learning_rate": 7.731168372364219e-05,
+      "loss": 0.4226,
+      "step": 715
+    },
+    {
+      "epoch": 1.0302158273381294,
+      "grad_norm": 0.7422510055372743,
+      "learning_rate": 7.729718102171438e-05,
+      "loss": 0.417,
+      "step": 716
+    },
+    {
+      "epoch": 1.0316546762589929,
+      "grad_norm": 0.9312794652767561,
+      "learning_rate": 7.728264067368976e-05,
+      "loss": 0.4163,
+      "step": 717
+    },
+    {
+      "epoch": 1.033093525179856,
+      "grad_norm": 1.3856090088417783,
+      "learning_rate": 7.726806269424469e-05,
+      "loss": 0.4273,
+      "step": 718
+    },
+    {
+      "epoch": 1.0345323741007195,
+      "grad_norm": 0.41077893968222584,
+      "learning_rate": 7.725344709809355e-05,
+      "loss": 0.4127,
+      "step": 719
+    },
+    {
+      "epoch": 1.0359712230215827,
+      "grad_norm": 1.254621053277421,
+      "learning_rate": 7.723879389998864e-05,
+      "loss": 0.4197,
+      "step": 720
+    },
+    {
+      "epoch": 1.037410071942446,
+      "grad_norm": 0.8999150862991432,
+      "learning_rate": 7.722410311472026e-05,
+      "loss": 0.4232,
+      "step": 721
+    },
+    {
+      "epoch": 1.0388489208633094,
+      "grad_norm": 0.6094735034533252,
+      "learning_rate": 7.72093747571166e-05,
+      "loss": 0.4232,
+      "step": 722
+    },
+    {
+      "epoch": 1.0402877697841726,
+      "grad_norm": 0.561578298258745,
+      "learning_rate": 7.719460884204383e-05,
+      "loss": 0.4186,
+      "step": 723
+    },
+    {
+      "epoch": 1.041726618705036,
+      "grad_norm": 0.6265469575259921,
+      "learning_rate": 7.717980538440599e-05,
+      "loss": 0.4242,
+      "step": 724
+    },
+    {
+      "epoch": 1.0431654676258992,
+      "grad_norm": 0.6957086180987356,
+      "learning_rate": 7.716496439914502e-05,
+      "loss": 0.4263,
+      "step": 725
+    },
+    {
+      "epoch": 1.0446043165467627,
+      "grad_norm": 0.6804151719159521,
+      "learning_rate": 7.715008590124076e-05,
+      "loss": 0.4141,
+      "step": 726
+    },
+    {
+      "epoch": 1.0460431654676259,
+      "grad_norm": 0.6133997418818746,
+      "learning_rate": 7.713516990571088e-05,
+      "loss": 0.4212,
+      "step": 727
+    },
+    {
+      "epoch": 1.0474820143884893,
+      "grad_norm": 0.5238363700426827,
+      "learning_rate": 7.71202164276109e-05,
+      "loss": 0.4196,
+      "step": 728
+    },
+    {
+      "epoch": 1.0489208633093525,
+      "grad_norm": 0.3610851256412559,
+      "learning_rate": 7.710522548203424e-05,
+      "loss": 0.4128,
+      "step": 729
+    },
+    {
+      "epoch": 1.0503597122302157,
+      "grad_norm": 0.370798257580522,
+      "learning_rate": 7.709019708411202e-05,
+      "loss": 0.4215,
+      "step": 730
+    },
+    {
+      "epoch": 1.0517985611510792,
+      "grad_norm": 0.45825791549468775,
+      "learning_rate": 7.707513124901327e-05,
+      "loss": 0.4154,
+      "step": 731
+    },
+    {
+      "epoch": 1.0532374100719424,
+      "grad_norm": 0.4493226973620425,
+      "learning_rate": 7.706002799194476e-05,
+      "loss": 0.4295,
+      "step": 732
+    },
+    {
+      "epoch": 1.0546762589928058,
+      "grad_norm": 0.4377297934543004,
+      "learning_rate": 7.704488732815105e-05,
+      "loss": 0.4115,
+      "step": 733
+    },
+    {
+      "epoch": 1.056115107913669,
+      "grad_norm": 0.40999934507501834,
+      "learning_rate": 7.702970927291442e-05,
+      "loss": 0.4174,
+      "step": 734
+    },
+    {
+      "epoch": 1.0575539568345325,
+      "grad_norm": 0.3422549763398815,
+      "learning_rate": 7.701449384155492e-05,
+      "loss": 0.4242,
+      "step": 735
+    },
+    {
+      "epoch": 1.0589928057553957,
+      "grad_norm": 0.3160845755661669,
+      "learning_rate": 7.699924104943033e-05,
+      "loss": 0.422,
+      "step": 736
+    },
+    {
+      "epoch": 1.0604316546762589,
+      "grad_norm": 0.3725745991616709,
+      "learning_rate": 7.698395091193615e-05,
+      "loss": 0.4181,
+      "step": 737
+    },
+    {
+      "epoch": 1.0618705035971223,
+      "grad_norm": 0.3846927289474282,
+      "learning_rate": 7.696862344450553e-05,
+      "loss": 0.4208,
+      "step": 738
+    },
+    {
+      "epoch": 1.0633093525179855,
+      "grad_norm": 0.4433253183936695,
+      "learning_rate": 7.695325866260932e-05,
+      "loss": 0.4079,
+      "step": 739
+    },
+    {
+      "epoch": 1.064748201438849,
+      "grad_norm": 0.6210118491031278,
+      "learning_rate": 7.693785658175607e-05,
+      "loss": 0.4281,
+      "step": 740
+    },
+    {
+      "epoch": 1.0661870503597122,
+      "grad_norm": 0.7680274413581848,
+      "learning_rate": 7.692241721749194e-05,
+      "loss": 0.4148,
+      "step": 741
+    },
+    {
+      "epoch": 1.0676258992805756,
+      "grad_norm": 0.8021919763630236,
+      "learning_rate": 7.69069405854007e-05,
+      "loss": 0.4222,
+      "step": 742
+    },
+    {
+      "epoch": 1.0690647482014388,
+      "grad_norm": 0.7288764113640167,
+      "learning_rate": 7.68914267011038e-05,
+      "loss": 0.4121,
+      "step": 743
+    },
+    {
+      "epoch": 1.0705035971223023,
+      "grad_norm": 0.6378506361039106,
+      "learning_rate": 7.687587558026024e-05,
+      "loss": 0.4138,
+      "step": 744
+    },
+    {
+      "epoch": 1.0719424460431655,
+      "grad_norm": 0.5576514753911082,
+      "learning_rate": 7.686028723856664e-05,
+      "loss": 0.4182,
+      "step": 745
+    },
+    {
+      "epoch": 1.0733812949640287,
+      "grad_norm": 0.4314748972331098,
+      "learning_rate": 7.684466169175714e-05,
+      "loss": 0.4145,
+      "step": 746
+    },
+    {
+      "epoch": 1.074820143884892,
+      "grad_norm": 0.2744023173556825,
+      "learning_rate": 7.68289989556035e-05,
+      "loss": 0.4289,
+      "step": 747
+    },
+    {
+      "epoch": 1.0762589928057553,
+      "grad_norm": 0.432284955906597,
+      "learning_rate": 7.681329904591495e-05,
+      "loss": 0.4116,
+      "step": 748
+    },
+    {
+      "epoch": 1.0776978417266188,
+      "grad_norm": 0.5768063729704934,
+      "learning_rate": 7.67975619785383e-05,
+      "loss": 0.4281,
+      "step": 749
+    },
+    {
+      "epoch": 1.079136690647482,
+      "grad_norm": 0.4962978373730406,
+      "learning_rate": 7.678178776935781e-05,
+      "loss": 0.4094,
+      "step": 750
+    },
+    {
+      "epoch": 1.0805755395683454,
+      "grad_norm": 0.41707422752363793,
+      "learning_rate": 7.676597643429528e-05,
+      "loss": 0.4191,
+      "step": 751
+    },
+    {
+      "epoch": 1.0820143884892086,
+      "grad_norm": 0.5272821722002808,
+      "learning_rate": 7.675012798930994e-05,
+      "loss": 0.4192,
+      "step": 752
+    },
+    {
+      "epoch": 1.083453237410072,
+      "grad_norm": 0.603326470094669,
+      "learning_rate": 7.673424245039852e-05,
+      "loss": 0.4148,
+      "step": 753
+    },
+    {
+      "epoch": 1.0848920863309353,
+      "grad_norm": 0.49183314809179646,
+      "learning_rate": 7.671831983359515e-05,
+      "loss": 0.4258,
+      "step": 754
+    },
+    {
+      "epoch": 1.0863309352517985,
+      "grad_norm": 0.4918040628996872,
+      "learning_rate": 7.670236015497141e-05,
+      "loss": 0.4221,
+      "step": 755
+    },
+    {
+      "epoch": 1.087769784172662,
+      "grad_norm": 0.6446415855495722,
+      "learning_rate": 7.668636343063628e-05,
+      "loss": 0.4122,
+      "step": 756
+    },
+    {
+      "epoch": 1.0892086330935251,
+      "grad_norm": 0.7542892767463729,
+      "learning_rate": 7.667032967673614e-05,
+      "loss": 0.4209,
+      "step": 757
+    },
+    {
+      "epoch": 1.0906474820143885,
+      "grad_norm": 0.8568636634175414,
+      "learning_rate": 7.665425890945474e-05,
+      "loss": 0.4236,
+      "step": 758
+    },
+    {
+      "epoch": 1.0920863309352518,
+      "grad_norm": 0.864668217646376,
+      "learning_rate": 7.663815114501319e-05,
+      "loss": 0.4106,
+      "step": 759
+    },
+    {
+      "epoch": 1.0935251798561152,
+      "grad_norm": 0.8526717179777845,
+      "learning_rate": 7.662200639966992e-05,
+      "loss": 0.4176,
+      "step": 760
+    },
+    {
+      "epoch": 1.0949640287769784,
+      "grad_norm": 0.7336891682731496,
+      "learning_rate": 7.660582468972074e-05,
+      "loss": 0.4224,
+      "step": 761
+    },
+    {
+      "epoch": 1.0964028776978418,
+      "grad_norm": 0.7285535475605878,
+      "learning_rate": 7.658960603149873e-05,
+      "loss": 0.4196,
+      "step": 762
+    },
+    {
+      "epoch": 1.097841726618705,
+      "grad_norm": 0.7078631580302459,
+      "learning_rate": 7.657335044137427e-05,
+      "loss": 0.4161,
+      "step": 763
+    },
+    {
+      "epoch": 1.0992805755395683,
+      "grad_norm": 0.6430803820273636,
+      "learning_rate": 7.655705793575504e-05,
+      "loss": 0.4244,
+      "step": 764
+    },
+    {
+      "epoch": 1.1007194244604317,
+      "grad_norm": 0.5619905072652992,
+      "learning_rate": 7.654072853108592e-05,
+      "loss": 0.4284,
+      "step": 765
+    },
+    {
+      "epoch": 1.102158273381295,
+      "grad_norm": 0.5244169515954566,
+      "learning_rate": 7.652436224384911e-05,
+      "loss": 0.4282,
+      "step": 766
+    },
+    {
+      "epoch": 1.1035971223021583,
+      "grad_norm": 0.3987451516791701,
+      "learning_rate": 7.6507959090564e-05,
+      "loss": 0.4153,
+      "step": 767
+    },
+    {
+      "epoch": 1.1050359712230216,
+      "grad_norm": 0.31705681956288584,
+      "learning_rate": 7.649151908778721e-05,
+      "loss": 0.4207,
+      "step": 768
+    },
+    {
+      "epoch": 1.106474820143885,
+      "grad_norm": 0.45003755116239236,
+      "learning_rate": 7.64750422521125e-05,
+      "loss": 0.4248,
+      "step": 769
+    },
+    {
+      "epoch": 1.1079136690647482,
+      "grad_norm": 0.51467358904595,
+      "learning_rate": 7.645852860017086e-05,
+      "loss": 0.4263,
+      "step": 770
+    },
+    {
+      "epoch": 1.1093525179856114,
+      "grad_norm": 0.43555140409683163,
+      "learning_rate": 7.644197814863045e-05,
+      "loss": 0.4303,
+      "step": 771
+    },
+    {
+      "epoch": 1.1107913669064748,
+      "grad_norm": 0.41737442204483916,
+      "learning_rate": 7.642539091419654e-05,
+      "loss": 0.4257,
+      "step": 772
+    },
+    {
+      "epoch": 1.112230215827338,
+      "grad_norm": 0.4252050622071493,
+      "learning_rate": 7.640876691361152e-05,
+      "loss": 0.4122,
+      "step": 773
+    },
+    {
+      "epoch": 1.1136690647482015,
+      "grad_norm": 0.421240589158121,
+      "learning_rate": 7.639210616365494e-05,
+      "loss": 0.4179,
+      "step": 774
+    },
+    {
+      "epoch": 1.1151079136690647,
+      "grad_norm": 0.41311441086953393,
+      "learning_rate": 7.637540868114338e-05,
+      "loss": 0.4198,
+      "step": 775
+    },
+    {
+      "epoch": 1.1165467625899281,
+      "grad_norm": 0.5074729502685691,
+      "learning_rate": 7.635867448293056e-05,
+      "loss": 0.4237,
+      "step": 776
+    },
+    {
+      "epoch": 1.1179856115107913,
+      "grad_norm": 0.6085863862933002,
+      "learning_rate": 7.63419035859072e-05,
+      "loss": 0.4153,
+      "step": 777
+    },
+    {
+      "epoch": 1.1194244604316548,
+      "grad_norm": 0.49446882458967484,
+      "learning_rate": 7.63250960070011e-05,
+      "loss": 0.4146,
+      "step": 778
+    },
+    {
+      "epoch": 1.120863309352518,
+      "grad_norm": 0.31521512846895183,
+      "learning_rate": 7.630825176317707e-05,
+      "loss": 0.4167,
+      "step": 779
+    },
+    {
+      "epoch": 1.1223021582733812,
+      "grad_norm": 0.5936739993914748,
+      "learning_rate": 7.629137087143693e-05,
+      "loss": 0.4168,
+      "step": 780
+    },
+    {
+      "epoch": 1.1237410071942446,
+      "grad_norm": 0.7538350242414342,
+      "learning_rate": 7.627445334881951e-05,
+      "loss": 0.418,
+      "step": 781
+    },
+    {
+      "epoch": 1.1251798561151078,
+      "grad_norm": 0.75065201309353,
+      "learning_rate": 7.625749921240058e-05,
+      "loss": 0.4153,
+      "step": 782
+    },
+    {
+      "epoch": 1.1266187050359713,
+      "grad_norm": 0.6423560489394579,
+      "learning_rate": 7.62405084792929e-05,
+      "loss": 0.4176,
+      "step": 783
+    },
+    {
+      "epoch": 1.1280575539568345,
+      "grad_norm": 0.5168027732967486,
+      "learning_rate": 7.622348116664611e-05,
+      "loss": 0.4279,
+      "step": 784
+    },
+    {
+      "epoch": 1.129496402877698,
+      "grad_norm": 0.5891227033433204,
+      "learning_rate": 7.620641729164686e-05,
+      "loss": 0.4103,
+      "step": 785
+    },
+    {
+      "epoch": 1.1309352517985611,
+      "grad_norm": 0.7134438221688847,
+      "learning_rate": 7.618931687151863e-05,
+      "loss": 0.4185,
+      "step": 786
+    },
+    {
+      "epoch": 1.1323741007194243,
+      "grad_norm": 0.7213046385941655,
+      "learning_rate": 7.617217992352183e-05,
+      "loss": 0.4252,
+      "step": 787
+    },
+    {
+      "epoch": 1.1338129496402878,
+      "grad_norm": 0.5705780566060814,
+      "learning_rate": 7.615500646495373e-05,
+      "loss": 0.4204,
+      "step": 788
+    },
+    {
+      "epoch": 1.135251798561151,
+      "grad_norm": 0.4302245388899498,
+      "learning_rate": 7.613779651314841e-05,
+      "loss": 0.4154,
+      "step": 789
+    },
+    {
+      "epoch": 1.1366906474820144,
+      "grad_norm": 0.37727734368717947,
+      "learning_rate": 7.612055008547688e-05,
+      "loss": 0.4243,
+      "step": 790
+    },
+    {
+      "epoch": 1.1381294964028776,
+      "grad_norm": 0.3739177488902291,
+      "learning_rate": 7.610326719934685e-05,
+      "loss": 0.4252,
+      "step": 791
+    },
+    {
+      "epoch": 1.139568345323741,
+      "grad_norm": 0.3263271090762003,
+      "learning_rate": 7.608594787220292e-05,
+      "loss": 0.4173,
+      "step": 792
+    },
+    {
+      "epoch": 1.1410071942446043,
+      "grad_norm": 0.37994810884379415,
+      "learning_rate": 7.606859212152644e-05,
+      "loss": 0.4173,
+      "step": 793
+    },
+    {
+      "epoch": 1.1424460431654677,
+      "grad_norm": 0.5773190952778903,
+      "learning_rate": 7.605119996483551e-05,
+      "loss": 0.418,
+      "step": 794
+    },
+    {
+      "epoch": 1.143884892086331,
+      "grad_norm": 0.6208254448365393,
+      "learning_rate": 7.6033771419685e-05,
+      "loss": 0.4177,
+      "step": 795
+    },
+    {
+      "epoch": 1.1453237410071941,
+      "grad_norm": 0.6584362582594144,
+      "learning_rate": 7.601630650366648e-05,
+      "loss": 0.4222,
+      "step": 796
+    },
+    {
+      "epoch": 1.1467625899280576,
+      "grad_norm": 0.6752585542994677,
+      "learning_rate": 7.59988052344083e-05,
+      "loss": 0.415,
+      "step": 797
+    },
+    {
+      "epoch": 1.1482014388489208,
+      "grad_norm": 0.760537285794366,
+      "learning_rate": 7.59812676295754e-05,
+      "loss": 0.4167,
+      "step": 798
+    },
+    {
+      "epoch": 1.1496402877697842,
+      "grad_norm": 0.8781635828536211,
+      "learning_rate": 7.596369370686947e-05,
+      "loss": 0.4351,
+      "step": 799
+    },
+    {
+      "epoch": 1.1510791366906474,
+      "grad_norm": 0.9415873644042007,
+      "learning_rate": 7.594608348402885e-05,
+      "loss": 0.4256,
+      "step": 800
+    },
+    {
+      "epoch": 1.1525179856115109,
+      "grad_norm": 0.8338541290612863,
+      "learning_rate": 7.592843697882848e-05,
+      "loss": 0.4165,
+      "step": 801
+    },
+    {
+      "epoch": 1.153956834532374,
+      "grad_norm": 0.6683626268938538,
+      "learning_rate": 7.591075420907997e-05,
+      "loss": 0.4259,
+      "step": 802
+    },
+    {
+      "epoch": 1.1553956834532375,
+      "grad_norm": 0.4844470964570286,
+      "learning_rate": 7.589303519263151e-05,
+      "loss": 0.423,
+      "step": 803
+    },
+    {
+      "epoch": 1.1568345323741007,
+      "grad_norm": 0.35350010087738093,
+      "learning_rate": 7.587527994736787e-05,
+      "loss": 0.4179,
+      "step": 804
+    },
+    {
+      "epoch": 1.158273381294964,
+      "grad_norm": 0.44582792317039727,
+      "learning_rate": 7.58574884912104e-05,
+      "loss": 0.4072,
+      "step": 805
+    },
+    {
+      "epoch": 1.1597122302158274,
+      "grad_norm": 0.48742184625135226,
+      "learning_rate": 7.5839660842117e-05,
+      "loss": 0.4232,
+      "step": 806
+    },
+    {
+      "epoch": 1.1611510791366906,
+      "grad_norm": 0.504469321517267,
+      "learning_rate": 7.582179701808208e-05,
+      "loss": 0.4104,
+      "step": 807
+    },
+    {
+      "epoch": 1.162589928057554,
+      "grad_norm": 0.4152839795687809,
+      "learning_rate": 7.580389703713661e-05,
+      "loss": 0.4162,
+      "step": 808
+    },
+    {
+      "epoch": 1.1640287769784172,
+      "grad_norm": 0.34815113916988333,
+      "learning_rate": 7.5785960917348e-05,
+      "loss": 0.4226,
+      "step": 809
+    },
+    {
+      "epoch": 1.1654676258992807,
+      "grad_norm": 0.4819757121959254,
+      "learning_rate": 7.576798867682018e-05,
+      "loss": 0.4214,
+      "step": 810
+    },
+    {
+      "epoch": 1.1669064748201439,
+      "grad_norm": 0.5178763620456481,
+      "learning_rate": 7.574998033369349e-05,
+      "loss": 0.4171,
+      "step": 811
+    },
+    {
+      "epoch": 1.1683453237410073,
+      "grad_norm": 0.4154321963183835,
+      "learning_rate": 7.573193590614479e-05,
+      "loss": 0.4215,
+      "step": 812
+    },
+    {
+      "epoch": 1.1697841726618705,
+      "grad_norm": 0.29377641830878887,
+      "learning_rate": 7.571385541238727e-05,
+      "loss": 0.415,
+      "step": 813
+    },
+    {
+      "epoch": 1.1712230215827337,
+      "grad_norm": 0.3873091669226385,
+      "learning_rate": 7.569573887067059e-05,
+      "loss": 0.4301,
+      "step": 814
+    },
+    {
+      "epoch": 1.1726618705035972,
+      "grad_norm": 0.4440531679666098,
+      "learning_rate": 7.567758629928076e-05,
+      "loss": 0.4269,
+      "step": 815
+    },
+    {
+      "epoch": 1.1741007194244604,
+      "grad_norm": 0.4238674797228725,
+      "learning_rate": 7.565939771654018e-05,
+      "loss": 0.412,
+      "step": 816
+    },
+    {
+      "epoch": 1.1755395683453238,
+      "grad_norm": 0.46840086425383703,
+      "learning_rate": 7.564117314080758e-05,
+      "loss": 0.4124,
+      "step": 817
+    },
+    {
+      "epoch": 1.176978417266187,
+      "grad_norm": 0.4391720956518648,
+      "learning_rate": 7.562291259047804e-05,
+      "loss": 0.4301,
+      "step": 818
+    },
+    {
+      "epoch": 1.1784172661870504,
+      "grad_norm": 0.34956265214346327,
+      "learning_rate": 7.560461608398292e-05,
+      "loss": 0.4182,
+      "step": 819
+    },
+    {
+      "epoch": 1.1798561151079137,
+      "grad_norm": 0.36980676294878867,
+      "learning_rate": 7.558628363978991e-05,
+      "loss": 0.4123,
+      "step": 820
+    },
+    {
+      "epoch": 1.181294964028777,
+      "grad_norm": 0.46513685237089797,
+      "learning_rate": 7.556791527640292e-05,
+      "loss": 0.4176,
+      "step": 821
+    },
+    {
+      "epoch": 1.1827338129496403,
+      "grad_norm": 0.5231382758166871,
+      "learning_rate": 7.554951101236219e-05,
+      "loss": 0.4201,
+      "step": 822
+    },
+    {
+      "epoch": 1.1841726618705035,
+      "grad_norm": 0.5160947629933547,
+      "learning_rate": 7.553107086624413e-05,
+      "loss": 0.4147,
+      "step": 823
+    },
+    {
+      "epoch": 1.185611510791367,
+      "grad_norm": 0.45194829521755847,
+      "learning_rate": 7.551259485666141e-05,
+      "loss": 0.4174,
+      "step": 824
+    },
+    {
+      "epoch": 1.1870503597122302,
+      "grad_norm": 0.4063881919431981,
+      "learning_rate": 7.549408300226287e-05,
+      "loss": 0.4161,
+      "step": 825
+    },
+    {
+      "epoch": 1.1884892086330936,
+      "grad_norm": 0.4188773062037776,
+      "learning_rate": 7.547553532173356e-05,
+      "loss": 0.4292,
+      "step": 826
+    },
+    {
+      "epoch": 1.1899280575539568,
+      "grad_norm": 0.43758049905108026,
+      "learning_rate": 7.545695183379465e-05,
+      "loss": 0.4283,
+      "step": 827
+    },
+    {
+      "epoch": 1.19136690647482,
+      "grad_norm": 0.4549818928421618,
+      "learning_rate": 7.54383325572035e-05,
+      "loss": 0.4188,
+      "step": 828
+    },
+    {
+      "epoch": 1.1928057553956835,
+      "grad_norm": 0.4509482466305469,
+      "learning_rate": 7.541967751075354e-05,
+      "loss": 0.4224,
+      "step": 829
+    },
+    {
+      "epoch": 1.1942446043165469,
+      "grad_norm": 0.5444101749600369,
+      "learning_rate": 7.540098671327438e-05,
+      "loss": 0.4249,
+      "step": 830
+    },
+    {
+      "epoch": 1.19568345323741,
+      "grad_norm": 0.626516592937173,
+      "learning_rate": 7.538226018363164e-05,
+      "loss": 0.4171,
+      "step": 831
+    },
+    {
+      "epoch": 1.1971223021582733,
+      "grad_norm": 0.6904870109865596,
+      "learning_rate": 7.536349794072705e-05,
+      "loss": 0.4193,
+      "step": 832
+    },
+    {
+      "epoch": 1.1985611510791367,
+      "grad_norm": 0.7503121033224123,
+      "learning_rate": 7.534470000349835e-05,
+      "loss": 0.4234,
+      "step": 833
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.8111243569118842,
+      "learning_rate": 7.532586639091936e-05,
+      "loss": 0.4173,
+      "step": 834
+    },
+    {
+      "epoch": 1.2014388489208634,
+      "grad_norm": 0.8836389030724152,
+      "learning_rate": 7.530699712199985e-05,
+      "loss": 0.4256,
+      "step": 835
+    },
+    {
+      "epoch": 1.2028776978417266,
+      "grad_norm": 0.8013048465813984,
+      "learning_rate": 7.528809221578565e-05,
+      "loss": 0.4184,
+      "step": 836
+    },
+    {
+      "epoch": 1.2043165467625898,
+      "grad_norm": 0.5653848901027986,
+      "learning_rate": 7.52691516913585e-05,
+      "loss": 0.4318,
+      "step": 837
+    },
+    {
+      "epoch": 1.2057553956834532,
+      "grad_norm": 0.4638018529487915,
+      "learning_rate": 7.525017556783612e-05,
+      "loss": 0.4174,
+      "step": 838
+    },
+    {
+      "epoch": 1.2071942446043165,
+      "grad_norm": 0.6313017471518908,
+      "learning_rate": 7.523116386437216e-05,
+      "loss": 0.4166,
+      "step": 839
+    },
+    {
+      "epoch": 1.20863309352518,
+      "grad_norm": 0.65247948493221,
+      "learning_rate": 7.521211660015615e-05,
+      "loss": 0.4162,
+      "step": 840
+    },
+    {
+      "epoch": 1.210071942446043,
+      "grad_norm": 0.48412170832446494,
+      "learning_rate": 7.519303379441357e-05,
+      "loss": 0.4226,
+      "step": 841
+    },
+    {
+      "epoch": 1.2115107913669065,
+      "grad_norm": 0.3894002356186535,
+      "learning_rate": 7.517391546640573e-05,
+      "loss": 0.4259,
+      "step": 842
+    },
+    {
+      "epoch": 1.2129496402877697,
+      "grad_norm": 0.4743786697065132,
+      "learning_rate": 7.515476163542982e-05,
+      "loss": 0.4189,
+      "step": 843
+    },
+    {
+      "epoch": 1.2143884892086332,
+      "grad_norm": 0.44692095971218454,
+      "learning_rate": 7.513557232081887e-05,
+      "loss": 0.4217,
+      "step": 844
+    },
+    {
+      "epoch": 1.2158273381294964,
+      "grad_norm": 0.30137146063183906,
+      "learning_rate": 7.511634754194168e-05,
+      "loss": 0.4205,
+      "step": 845
+    },
+    {
+      "epoch": 1.2172661870503596,
+      "grad_norm": 0.4219746836134125,
+      "learning_rate": 7.50970873182029e-05,
+      "loss": 0.4186,
+      "step": 846
+    },
+    {
+      "epoch": 1.218705035971223,
+      "grad_norm": 0.47854173151744644,
+      "learning_rate": 7.507779166904292e-05,
+      "loss": 0.4204,
+      "step": 847
+    },
+    {
+      "epoch": 1.2201438848920863,
+      "grad_norm": 0.3758110059976876,
+      "learning_rate": 7.50584606139379e-05,
+      "loss": 0.4228,
+      "step": 848
+    },
+    {
+      "epoch": 1.2215827338129497,
+      "grad_norm": 0.35428654478853616,
+      "learning_rate": 7.503909417239975e-05,
+      "loss": 0.4218,
+      "step": 849
+    },
+    {
+      "epoch": 1.223021582733813,
+      "grad_norm": 0.5034960728391591,
+      "learning_rate": 7.501969236397607e-05,
+      "loss": 0.4128,
+      "step": 850
+    },
+    {
+      "epoch": 1.2244604316546763,
+      "grad_norm": 0.6410823859828029,
+      "learning_rate": 7.500025520825018e-05,
+      "loss": 0.4323,
+      "step": 851
+    },
+    {
+      "epoch": 1.2258992805755395,
+      "grad_norm": 0.8171577583438102,
+      "learning_rate": 7.498078272484108e-05,
+      "loss": 0.4206,
+      "step": 852
+    },
+    {
+      "epoch": 1.227338129496403,
+      "grad_norm": 0.9905863473480702,
+      "learning_rate": 7.496127493340341e-05,
+      "loss": 0.4265,
+      "step": 853
+    },
+    {
+      "epoch": 1.2287769784172662,
+      "grad_norm": 1.1019278429398947,
+      "learning_rate": 7.494173185362745e-05,
+      "loss": 0.4254,
+      "step": 854
+    },
+    {
+      "epoch": 1.2302158273381294,
+      "grad_norm": 0.7783355621185797,
+      "learning_rate": 7.492215350523913e-05,
+      "loss": 0.4335,
+      "step": 855
+    },
+    {
+      "epoch": 1.2316546762589928,
+      "grad_norm": 0.5123283474072974,
+      "learning_rate": 7.490253990799991e-05,
+      "loss": 0.4123,
+      "step": 856
+    },
+    {
+      "epoch": 1.233093525179856,
+      "grad_norm": 0.4051449414915292,
+      "learning_rate": 7.488289108170692e-05,
+      "loss": 0.4207,
+      "step": 857
+    },
+    {
+      "epoch": 1.2345323741007195,
+      "grad_norm": 0.45377464565373454,
+      "learning_rate": 7.486320704619276e-05,
+      "loss": 0.4146,
+      "step": 858
+    },
+    {
+      "epoch": 1.2359712230215827,
+      "grad_norm": 0.5245555215627437,
+      "learning_rate": 7.484348782132565e-05,
+      "loss": 0.4162,
+      "step": 859
+    },
+    {
+      "epoch": 1.2374100719424461,
+      "grad_norm": 0.5327932452184712,
+      "learning_rate": 7.482373342700927e-05,
+      "loss": 0.4075,
+      "step": 860
+    },
+    {
+      "epoch": 1.2388489208633093,
+      "grad_norm": 0.5541213309539225,
+      "learning_rate": 7.48039438831828e-05,
+      "loss": 0.4159,
+      "step": 861
+    },
+    {
+      "epoch": 1.2402877697841728,
+      "grad_norm": 0.5643639432637277,
+      "learning_rate": 7.478411920982095e-05,
+      "loss": 0.4157,
+      "step": 862
+    },
+    {
+      "epoch": 1.241726618705036,
+      "grad_norm": 0.580061380544043,
+      "learning_rate": 7.476425942693382e-05,
+      "loss": 0.4202,
+      "step": 863
+    },
+    {
+      "epoch": 1.2431654676258992,
+      "grad_norm": 0.5220355777656684,
+      "learning_rate": 7.474436455456701e-05,
+      "loss": 0.4204,
+      "step": 864
+    },
+    {
+      "epoch": 1.2446043165467626,
+      "grad_norm": 0.38702531956157415,
+      "learning_rate": 7.472443461280149e-05,
+      "loss": 0.4176,
+      "step": 865
+    },
+    {
+      "epoch": 1.2460431654676258,
+      "grad_norm": 0.41202934508706474,
+      "learning_rate": 7.470446962175367e-05,
+      "loss": 0.4267,
+      "step": 866
+    },
+    {
+      "epoch": 1.2474820143884893,
+      "grad_norm": 0.5035311974061664,
+      "learning_rate": 7.468446960157527e-05,
+      "loss": 0.4265,
+      "step": 867
+    },
+    {
+      "epoch": 1.2489208633093525,
+      "grad_norm": 0.569434393610722,
+      "learning_rate": 7.466443457245344e-05,
+      "loss": 0.4172,
+      "step": 868
+    },
+    {
+      "epoch": 1.2503597122302157,
+      "grad_norm": 0.6006260739620486,
+      "learning_rate": 7.464436455461066e-05,
+      "loss": 0.4218,
+      "step": 869
+    },
+    {
+      "epoch": 1.2517985611510791,
+      "grad_norm": 0.5455962473135081,
+      "learning_rate": 7.462425956830466e-05,
+      "loss": 0.4152,
+      "step": 870
+    },
+    {
+      "epoch": 1.2532374100719426,
+      "grad_norm": 0.3809920362148773,
+      "learning_rate": 7.460411963382853e-05,
+      "loss": 0.4178,
+      "step": 871
+    },
+    {
+      "epoch": 1.2546762589928058,
+      "grad_norm": 0.41925821122076273,
+      "learning_rate": 7.45839447715106e-05,
+      "loss": 0.4165,
+      "step": 872
+    },
+    {
+      "epoch": 1.256115107913669,
+      "grad_norm": 0.4823757639947033,
+      "learning_rate": 7.456373500171449e-05,
+      "loss": 0.4111,
+      "step": 873
+    },
+    {
+      "epoch": 1.2575539568345324,
+      "grad_norm": 0.4346303434108943,
+      "learning_rate": 7.454349034483903e-05,
+      "loss": 0.4185,
+      "step": 874
+    },
+    {
+      "epoch": 1.2589928057553956,
+      "grad_norm": 0.44014311113204413,
+      "learning_rate": 7.452321082131824e-05,
+      "loss": 0.4159,
+      "step": 875
+    },
+    {
+      "epoch": 1.260431654676259,
+      "grad_norm": 0.4565120265001911,
+      "learning_rate": 7.450289645162138e-05,
+      "loss": 0.4185,
+      "step": 876
+    },
+    {
+      "epoch": 1.2618705035971223,
+      "grad_norm": 0.46322718110655864,
+      "learning_rate": 7.448254725625287e-05,
+      "loss": 0.4185,
+      "step": 877
+    },
+    {
+      "epoch": 1.2633093525179855,
+      "grad_norm": 0.4851681984355067,
+      "learning_rate": 7.446216325575225e-05,
+      "loss": 0.4182,
+      "step": 878
+    },
+    {
+      "epoch": 1.264748201438849,
+      "grad_norm": 0.5717465825487871,
+      "learning_rate": 7.444174447069423e-05,
+      "loss": 0.4102,
+      "step": 879
+    },
+    {
+      "epoch": 1.2661870503597124,
+      "grad_norm": 0.5430731629551574,
+      "learning_rate": 7.442129092168859e-05,
+      "loss": 0.4139,
+      "step": 880
+    },
+    {
+      "epoch": 1.2676258992805756,
+      "grad_norm": 0.4706433407685034,
+      "learning_rate": 7.440080262938026e-05,
+      "loss": 0.4198,
+      "step": 881
+    },
+    {
+      "epoch": 1.2690647482014388,
+      "grad_norm": 0.5170788910681468,
+      "learning_rate": 7.438027961444916e-05,
+      "loss": 0.4237,
+      "step": 882
+    },
+    {
+      "epoch": 1.2705035971223022,
+      "grad_norm": 0.5067616291813265,
+      "learning_rate": 7.435972189761033e-05,
+      "loss": 0.4265,
+      "step": 883
+    },
+    {
+      "epoch": 1.2719424460431654,
+      "grad_norm": 0.4597445039030618,
+      "learning_rate": 7.43391294996138e-05,
+      "loss": 0.4171,
+      "step": 884
+    },
+    {
+      "epoch": 1.2733812949640289,
+      "grad_norm": 0.42939256207205173,
+      "learning_rate": 7.431850244124459e-05,
+      "loss": 0.4161,
+      "step": 885
+    },
+    {
+      "epoch": 1.274820143884892,
+      "grad_norm": 0.46502457984089807,
+      "learning_rate": 7.429784074332274e-05,
+      "loss": 0.416,
+      "step": 886
+    },
+    {
+      "epoch": 1.2762589928057553,
+      "grad_norm": 0.5397594990198463,
+      "learning_rate": 7.427714442670324e-05,
+      "loss": 0.4199,
+      "step": 887
+    },
+    {
+      "epoch": 1.2776978417266187,
+      "grad_norm": 0.6916813186652401,
+      "learning_rate": 7.425641351227602e-05,
+      "loss": 0.4127,
+      "step": 888
+    },
+    {
+      "epoch": 1.2791366906474821,
+      "grad_norm": 0.8740038338577955,
+      "learning_rate": 7.423564802096592e-05,
+      "loss": 0.416,
+      "step": 889
+    },
+    {
+      "epoch": 1.2805755395683454,
+      "grad_norm": 1.082921425091471,
+      "learning_rate": 7.42148479737327e-05,
+      "loss": 0.4121,
+      "step": 890
+    },
+    {
+      "epoch": 1.2820143884892086,
+      "grad_norm": 0.8147796616381295,
+      "learning_rate": 7.419401339157099e-05,
+      "loss": 0.4086,
+      "step": 891
+    },
+    {
+      "epoch": 1.283453237410072,
+      "grad_norm": 0.5109372878426125,
+      "learning_rate": 7.41731442955103e-05,
+      "loss": 0.4181,
+      "step": 892
+    },
+    {
+      "epoch": 1.2848920863309352,
+      "grad_norm": 0.5118049349727969,
+      "learning_rate": 7.415224070661492e-05,
+      "loss": 0.4276,
+      "step": 893
+    },
+    {
+      "epoch": 1.2863309352517986,
+      "grad_norm": 0.7298639025862309,
+      "learning_rate": 7.413130264598404e-05,
+      "loss": 0.4234,
+      "step": 894
+    },
+    {
+      "epoch": 1.2877697841726619,
+      "grad_norm": 1.1217691855762015,
+      "learning_rate": 7.411033013475156e-05,
+      "loss": 0.4246,
+      "step": 895
+    },
+    {
+      "epoch": 1.289208633093525,
+      "grad_norm": 0.3163488848249356,
+      "learning_rate": 7.408932319408619e-05,
+      "loss": 0.4061,
+      "step": 896
+    },
+    {
+      "epoch": 1.2906474820143885,
+      "grad_norm": 0.7430713251309599,
+      "learning_rate": 7.406828184519141e-05,
+      "loss": 0.4217,
+      "step": 897
+    },
+    {
+      "epoch": 1.292086330935252,
+      "grad_norm": 0.8800106116979265,
+      "learning_rate": 7.40472061093054e-05,
+      "loss": 0.4162,
+      "step": 898
+    },
+    {
+      "epoch": 1.2935251798561151,
+      "grad_norm": 0.7060110576639796,
+      "learning_rate": 7.402609600770104e-05,
+      "loss": 0.4271,
+      "step": 899
+    },
+    {
+      "epoch": 1.2949640287769784,
+      "grad_norm": 0.5071115772502598,
+      "learning_rate": 7.400495156168596e-05,
+      "loss": 0.4215,
+      "step": 900
+    },
+    {
+      "epoch": 1.2964028776978418,
+      "grad_norm": 0.6015817039371135,
+      "learning_rate": 7.39837727926024e-05,
+      "loss": 0.4132,
+      "step": 901
+    },
+    {
+      "epoch": 1.297841726618705,
+      "grad_norm": 0.6836388340295115,
+      "learning_rate": 7.396255972182723e-05,
+      "loss": 0.4236,
+      "step": 902
+    },
+    {
+      "epoch": 1.2992805755395684,
+      "grad_norm": 0.5118641074100791,
+      "learning_rate": 7.394131237077199e-05,
+      "loss": 0.4148,
+      "step": 903
+    },
+    {
+      "epoch": 1.3007194244604317,
+      "grad_norm": 0.4947960018961066,
+      "learning_rate": 7.39200307608828e-05,
+      "loss": 0.4228,
+      "step": 904
+    },
+    {
+      "epoch": 1.3021582733812949,
+      "grad_norm": 2.809075742801845,
+      "learning_rate": 7.389871491364036e-05,
+      "loss": 0.4854,
+      "step": 905
+    },
+    {
+      "epoch": 1.3035971223021583,
+      "grad_norm": 3.1902703322169135,
+      "learning_rate": 7.387736485055993e-05,
+      "loss": 0.4914,
+      "step": 906
+    },
+    {
+      "epoch": 1.3050359712230215,
+      "grad_norm": 3.169079308937887,
+      "learning_rate": 7.385598059319129e-05,
+      "loss": 0.515,
+      "step": 907
+    },
+    {
+      "epoch": 1.306474820143885,
+      "grad_norm": 2.296508230631993,
+      "learning_rate": 7.383456216311875e-05,
+      "loss": 0.4804,
+      "step": 908
+    },
+    {
+      "epoch": 1.3079136690647482,
+      "grad_norm": 25.565188899774473,
+      "learning_rate": 7.381310958196112e-05,
+      "loss": 0.7105,
+      "step": 909
+    },
+    {
+      "epoch": 1.3093525179856116,
+      "grad_norm": 544.9258123420445,
+      "learning_rate": 7.379162287137167e-05,
+      "loss": 6.2232,
+      "step": 910
+    },
+    {
+      "epoch": 1.3107913669064748,
+      "grad_norm": 20.044170581073192,
+      "learning_rate": 7.37701020530381e-05,
+      "loss": 1.0189,
+      "step": 911
+    },
+    {
+      "epoch": 1.3122302158273382,
+      "grad_norm": 32.03230660800528,
+      "learning_rate": 7.374854714868259e-05,
+      "loss": 1.0518,
+      "step": 912
+    },
+    {
+      "epoch": 1.3136690647482014,
+      "grad_norm": 12.129176440633575,
+      "learning_rate": 7.372695818006167e-05,
+      "loss": 0.8048,
+      "step": 913
+    },
+    {
+      "epoch": 1.3151079136690647,
+      "grad_norm": 20.22774171993354,
+      "learning_rate": 7.370533516896627e-05,
+      "loss": 1.0387,
+      "step": 914
+    },
+    {
+      "epoch": 1.316546762589928,
+      "grad_norm": 4.675031924475641,
+      "learning_rate": 7.368367813722169e-05,
+      "loss": 0.6672,
+      "step": 915
+    },
+    {
+      "epoch": 1.3179856115107913,
+      "grad_norm": 3.3403459553688046,
+      "learning_rate": 7.366198710668755e-05,
+      "loss": 0.6201,
+      "step": 916
+    },
+    {
+      "epoch": 1.3194244604316547,
+      "grad_norm": 0.9943048642973623,
+      "learning_rate": 7.364026209925783e-05,
+      "loss": 0.5459,
+      "step": 917
+    },
+    {
+      "epoch": 1.320863309352518,
+      "grad_norm": 91.42623205140953,
+      "learning_rate": 7.361850313686076e-05,
+      "loss": 0.9093,
+      "step": 918
+    },
+    {
+      "epoch": 1.3223021582733812,
+      "grad_norm": 247.10257252231983,
+      "learning_rate": 7.359671024145886e-05,
+      "loss": 5.0509,
+      "step": 919
+    },
+    {
+      "epoch": 1.3237410071942446,
+      "grad_norm": 18.58663410845094,
+      "learning_rate": 7.35748834350489e-05,
+      "loss": 1.4504,
+      "step": 920
+    },
+    {
+      "epoch": 1.325179856115108,
+      "grad_norm": 5.159800355671047,
+      "learning_rate": 7.355302273966186e-05,
+      "loss": 0.9351,
+      "step": 921
+    },
+    {
+      "epoch": 1.3266187050359712,
+      "grad_norm": 3.3547307250861187,
+      "learning_rate": 7.353112817736295e-05,
+      "loss": 0.7155,
+      "step": 922
+    },
+    {
+      "epoch": 1.3280575539568344,
+      "grad_norm": 1.5664564817666657,
+      "learning_rate": 7.350919977025157e-05,
+      "loss": 0.5998,
+      "step": 923
+    },
+    {
+      "epoch": 1.3294964028776979,
+      "grad_norm": 57.02363649654408,
+      "learning_rate": 7.348723754046127e-05,
+      "loss": 0.8424,
+      "step": 924
+    },
+    {
+      "epoch": 1.330935251798561,
+      "grad_norm": 4.901340781457194,
+      "learning_rate": 7.34652415101597e-05,
+      "loss": 0.6654,
+      "step": 925
+    },
+    {
+      "epoch": 1.3323741007194245,
+      "grad_norm": 2.5937845788317055,
+      "learning_rate": 7.344321170154871e-05,
+      "loss": 0.6026,
+      "step": 926
+    },
+    {
+      "epoch": 1.3338129496402877,
+      "grad_norm": 0.9994241678902555,
+      "learning_rate": 7.342114813686419e-05,
+      "loss": 0.5567,
+      "step": 927
+    },
+    {
+      "epoch": 1.335251798561151,
+      "grad_norm": 1.1829428406464586,
+      "learning_rate": 7.339905083837608e-05,
+      "loss": 0.5524,
+      "step": 928
+    },
+    {
+      "epoch": 1.3366906474820144,
+      "grad_norm": 0.9102676284435843,
+      "learning_rate": 7.337691982838841e-05,
+      "loss": 0.5277,
+      "step": 929
+    },
+    {
+      "epoch": 1.3381294964028778,
+      "grad_norm": 1.1130004246164928,
+      "learning_rate": 7.335475512923924e-05,
+      "loss": 0.525,
+      "step": 930
+    },
+    {
+      "epoch": 1.339568345323741,
+      "grad_norm": 0.6353478530329731,
+      "learning_rate": 7.33325567633006e-05,
+      "loss": 0.5075,
+      "step": 931
+    },
+    {
+      "epoch": 1.3410071942446042,
+      "grad_norm": 0.7154586227162724,
+      "learning_rate": 7.331032475297855e-05,
+      "loss": 0.4959,
+      "step": 932
+    },
+    {
+      "epoch": 1.3424460431654677,
+      "grad_norm": 0.7249413976258889,
+      "learning_rate": 7.328805912071307e-05,
+      "loss": 0.4797,
+      "step": 933
+    },
+    {
+      "epoch": 1.3438848920863309,
+      "grad_norm": 0.517053575901379,
+      "learning_rate": 7.326575988897807e-05,
+      "loss": 0.4764,
+      "step": 934
+    },
+    {
+      "epoch": 1.3453237410071943,
+      "grad_norm": 0.5874905496337369,
+      "learning_rate": 7.324342708028141e-05,
+      "loss": 0.469,
+      "step": 935
+    },
+    {
+      "epoch": 1.3467625899280575,
+      "grad_norm": 0.43744070662731827,
+      "learning_rate": 7.322106071716483e-05,
+      "loss": 0.4717,
+      "step": 936
+    },
+    {
+      "epoch": 1.3482014388489207,
+      "grad_norm": 0.5453893982842335,
+      "learning_rate": 7.319866082220388e-05,
+      "loss": 0.4569,
+      "step": 937
+    },
+    {
+      "epoch": 1.3496402877697842,
+      "grad_norm": 0.5200274639447857,
+      "learning_rate": 7.317622741800808e-05,
+      "loss": 0.4547,
+      "step": 938
+    },
+    {
+      "epoch": 1.3510791366906476,
+      "grad_norm": 0.4821080932076026,
+      "learning_rate": 7.315376052722065e-05,
+      "loss": 0.4615,
+      "step": 939
+    },
+    {
+      "epoch": 1.3525179856115108,
+      "grad_norm": 0.4078836036589005,
+      "learning_rate": 7.313126017251868e-05,
+      "loss": 0.4521,
+      "step": 940
+    },
+    {
+      "epoch": 1.353956834532374,
+      "grad_norm": 0.3473670552596463,
+      "learning_rate": 7.3108726376613e-05,
+      "loss": 0.4379,
+      "step": 941
+    },
+    {
+      "epoch": 1.3553956834532375,
+      "grad_norm": 0.3806100761112652,
+      "learning_rate": 7.308615916224823e-05,
+      "loss": 0.4441,
+      "step": 942
+    },
+    {
+      "epoch": 1.3568345323741007,
+      "grad_norm": 0.410225859601859,
+      "learning_rate": 7.306355855220267e-05,
+      "loss": 0.4445,
+      "step": 943
+    },
+    {
+      "epoch": 1.358273381294964,
+      "grad_norm": 0.31067952080530287,
+      "learning_rate": 7.30409245692884e-05,
+      "loss": 0.4376,
+      "step": 944
+    },
+    {
+      "epoch": 1.3597122302158273,
+      "grad_norm": 0.36147055056886784,
+      "learning_rate": 7.301825723635111e-05,
+      "loss": 0.4518,
+      "step": 945
+    },
+    {
+      "epoch": 1.3611510791366905,
+      "grad_norm": 0.4580320309729214,
+      "learning_rate": 7.299555657627021e-05,
+      "loss": 0.4368,
+      "step": 946
+    },
+    {
+      "epoch": 1.362589928057554,
+      "grad_norm": 0.4679993897648468,
+      "learning_rate": 7.29728226119587e-05,
+      "loss": 0.4509,
+      "step": 947
+    },
+    {
+      "epoch": 1.3640287769784174,
+      "grad_norm": 0.6932528279206558,
+      "learning_rate": 7.295005536636325e-05,
+      "loss": 0.4418,
+      "step": 948
+    },
+    {
+      "epoch": 1.3654676258992806,
+      "grad_norm": 0.7804774459738667,
+      "learning_rate": 7.292725486246407e-05,
+      "loss": 0.4427,
+      "step": 949
+    },
+    {
+      "epoch": 1.3669064748201438,
+      "grad_norm": 0.49179197252689205,
+      "learning_rate": 7.290442112327498e-05,
+      "loss": 0.4367,
+      "step": 950
+    },
+    {
+      "epoch": 1.3683453237410073,
+      "grad_norm": 0.5982458676214871,
+      "learning_rate": 7.288155417184331e-05,
+      "loss": 0.4351,
+      "step": 951
+    },
+    {
+      "epoch": 1.3697841726618705,
+      "grad_norm": 0.6881110302017649,
+      "learning_rate": 7.285865403124995e-05,
+      "loss": 0.4361,
+      "step": 952
+    },
+    {
+      "epoch": 1.371223021582734,
+      "grad_norm": 0.9825511136783236,
+      "learning_rate": 7.283572072460927e-05,
+      "loss": 0.4451,
+      "step": 953
+    },
+    {
+      "epoch": 1.3726618705035971,
+      "grad_norm": 1.1179334698800347,
+      "learning_rate": 7.28127542750691e-05,
+      "loss": 0.4385,
+      "step": 954
+    },
+    {
+      "epoch": 1.3741007194244603,
+      "grad_norm": 0.43264156063929776,
+      "learning_rate": 7.278975470581076e-05,
+      "loss": 0.433,
+      "step": 955
+    },
+    {
+      "epoch": 1.3755395683453238,
+      "grad_norm": 0.7286896948512599,
+      "learning_rate": 7.276672204004898e-05,
+      "loss": 0.4349,
+      "step": 956
+    },
+    {
+      "epoch": 1.376978417266187,
+      "grad_norm": 0.8825376334064328,
+      "learning_rate": 7.274365630103189e-05,
+      "loss": 0.4338,
+      "step": 957
+    },
+    {
+      "epoch": 1.3784172661870504,
+      "grad_norm": 0.6639218424283293,
+      "learning_rate": 7.2720557512041e-05,
+      "loss": 0.4345,
+      "step": 958
+    },
+    {
+      "epoch": 1.3798561151079136,
+      "grad_norm": 0.43365041814785904,
+      "learning_rate": 7.269742569639121e-05,
+      "loss": 0.4357,
+      "step": 959
+    },
+    {
+      "epoch": 1.381294964028777,
+      "grad_norm": 0.6429578440689165,
+      "learning_rate": 7.267426087743073e-05,
+      "loss": 0.4226,
+      "step": 960
+    },
+    {
+      "epoch": 1.3827338129496403,
+      "grad_norm": 0.6408192107115253,
+      "learning_rate": 7.265106307854107e-05,
+      "loss": 0.4308,
+      "step": 961
+    },
+    {
+      "epoch": 1.3841726618705037,
+      "grad_norm": 0.6000714524229904,
+      "learning_rate": 7.262783232313706e-05,
+      "loss": 0.4336,
+      "step": 962
+    },
+    {
+      "epoch": 1.385611510791367,
+      "grad_norm": 0.5423233672454112,
+      "learning_rate": 7.260456863466676e-05,
+      "loss": 0.4234,
+      "step": 963
+    },
+    {
+      "epoch": 1.3870503597122301,
+      "grad_norm": 0.623914979251738,
+      "learning_rate": 7.258127203661153e-05,
+      "loss": 0.4255,
+      "step": 964
+    },
+    {
+      "epoch": 1.3884892086330936,
+      "grad_norm": 0.529835304290532,
+      "learning_rate": 7.255794255248587e-05,
+      "loss": 0.431,
+      "step": 965
+    },
+    {
+      "epoch": 1.3899280575539568,
+      "grad_norm": 0.4873371283273517,
+      "learning_rate": 7.253458020583752e-05,
+      "loss": 0.4352,
+      "step": 966
+    },
+    {
+      "epoch": 1.3913669064748202,
+      "grad_norm": 0.47874618876157593,
+      "learning_rate": 7.25111850202474e-05,
+      "loss": 0.423,
+      "step": 967
+    },
+    {
+      "epoch": 1.3928057553956834,
+      "grad_norm": 0.44584845046178967,
+      "learning_rate": 7.248775701932953e-05,
+      "loss": 0.421,
+      "step": 968
+    },
+    {
+      "epoch": 1.3942446043165468,
+      "grad_norm": 0.44820331929916596,
+      "learning_rate": 7.246429622673111e-05,
+      "loss": 0.4259,
+      "step": 969
+    },
+    {
+      "epoch": 1.39568345323741,
+      "grad_norm": 0.3470987235749387,
+      "learning_rate": 7.244080266613238e-05,
+      "loss": 0.4237,
+      "step": 970
+    },
+    {
+      "epoch": 1.3971223021582735,
+      "grad_norm": 0.4200217431351872,
+      "learning_rate": 7.241727636124671e-05,
+      "loss": 0.4169,
+      "step": 971
+    },
+    {
+      "epoch": 1.3985611510791367,
+      "grad_norm": 0.47456172127286683,
+      "learning_rate": 7.239371733582047e-05,
+      "loss": 0.4245,
+      "step": 972
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.37151128289895746,
+      "learning_rate": 7.23701256136331e-05,
+      "loss": 0.4287,
+      "step": 973
+    },
+    {
+      "epoch": 1.4014388489208633,
+      "grad_norm": 0.4077295931874118,
+      "learning_rate": 7.2346501218497e-05,
+      "loss": 0.4256,
+      "step": 974
+    },
+    {
+      "epoch": 1.4028776978417266,
+      "grad_norm": 0.29297584698969287,
+      "learning_rate": 7.23228441742576e-05,
+      "loss": 0.4262,
+      "step": 975
+    },
+    {
+      "epoch": 1.40431654676259,
+      "grad_norm": 0.38241491031733416,
+      "learning_rate": 7.229915450479324e-05,
+      "loss": 0.4242,
+      "step": 976
+    },
+    {
+      "epoch": 1.4057553956834532,
+      "grad_norm": 0.37225834692829823,
+      "learning_rate": 7.227543223401522e-05,
+      "loss": 0.4224,
+      "step": 977
+    },
+    {
+      "epoch": 1.4071942446043164,
+      "grad_norm": 0.2942808197297915,
+      "learning_rate": 7.225167738586772e-05,
+      "loss": 0.4209,
+      "step": 978
+    },
+    {
+      "epoch": 1.4086330935251798,
+      "grad_norm": 0.4121809107133147,
+      "learning_rate": 7.22278899843278e-05,
+      "loss": 0.4283,
+      "step": 979
+    },
+    {
+      "epoch": 1.4100719424460433,
+      "grad_norm": 0.35849148544830317,
+      "learning_rate": 7.220407005340542e-05,
+      "loss": 0.4249,
+      "step": 980
+    },
+    {
+      "epoch": 1.4115107913669065,
+      "grad_norm": 0.37478659814507403,
+      "learning_rate": 7.218021761714336e-05,
+      "loss": 0.4251,
+      "step": 981
+    },
+    {
+      "epoch": 1.4129496402877697,
+      "grad_norm": 0.34362680740178464,
+      "learning_rate": 7.215633269961714e-05,
+      "loss": 0.4192,
+      "step": 982
+    },
+    {
+      "epoch": 1.4143884892086331,
+      "grad_norm": 0.4142071899637734,
+      "learning_rate": 7.213241532493516e-05,
+      "loss": 0.4252,
+      "step": 983
+    },
+    {
+      "epoch": 1.4158273381294963,
+      "grad_norm": 0.3842983010726522,
+      "learning_rate": 7.210846551723855e-05,
+      "loss": 0.4182,
+      "step": 984
+    },
+    {
+      "epoch": 1.4172661870503598,
+      "grad_norm": 0.303550811942089,
+      "learning_rate": 7.208448330070116e-05,
+      "loss": 0.4207,
+      "step": 985
+    },
+    {
+      "epoch": 1.418705035971223,
+      "grad_norm": 0.31739587700139277,
+      "learning_rate": 7.206046869952954e-05,
+      "loss": 0.4225,
+      "step": 986
+    },
+    {
+      "epoch": 1.4201438848920862,
+      "grad_norm": 0.33225743068915725,
+      "learning_rate": 7.203642173796298e-05,
+      "loss": 0.42,
+      "step": 987
+    },
+    {
+      "epoch": 1.4215827338129496,
+      "grad_norm": 0.23677363095088855,
+      "learning_rate": 7.201234244027338e-05,
+      "loss": 0.4161,
+      "step": 988
+    },
+    {
+      "epoch": 1.423021582733813,
+      "grad_norm": 0.26592553747289,
+      "learning_rate": 7.19882308307653e-05,
+      "loss": 0.4144,
+      "step": 989
+    },
+    {
+      "epoch": 1.4244604316546763,
+      "grad_norm": 0.3655310161495709,
+      "learning_rate": 7.196408693377594e-05,
+      "loss": 0.4075,
+      "step": 990
+    },
+    {
+      "epoch": 1.4258992805755395,
+      "grad_norm": 0.3103060704669727,
+      "learning_rate": 7.193991077367501e-05,
+      "loss": 0.4206,
+      "step": 991
+    },
+    {
+      "epoch": 1.427338129496403,
+      "grad_norm": 0.30395533893193416,
+      "learning_rate": 7.19157023748649e-05,
+      "loss": 0.4205,
+      "step": 992
+    },
+    {
+      "epoch": 1.4287769784172661,
+      "grad_norm": 0.29152986731845987,
+      "learning_rate": 7.189146176178044e-05,
+      "loss": 0.4339,
+      "step": 993
+    },
+    {
+      "epoch": 1.4302158273381296,
+      "grad_norm": 0.2859730011717553,
+      "learning_rate": 7.186718895888904e-05,
+      "loss": 0.4176,
+      "step": 994
+    },
+    {
+      "epoch": 1.4316546762589928,
+      "grad_norm": 0.39314266580090007,
+      "learning_rate": 7.184288399069054e-05,
+      "loss": 0.4129,
+      "step": 995
+    },
+    {
+      "epoch": 1.433093525179856,
+      "grad_norm": 0.37407860657904163,
+      "learning_rate": 7.181854688171732e-05,
+      "loss": 0.4226,
+      "step": 996
+    },
+    {
+      "epoch": 1.4345323741007194,
+      "grad_norm": 0.33658296177937513,
+      "learning_rate": 7.179417765653413e-05,
+      "loss": 0.4248,
+      "step": 997
+    },
+    {
+      "epoch": 1.4359712230215829,
+      "grad_norm": 0.3651105251554801,
+      "learning_rate": 7.17697763397382e-05,
+      "loss": 0.4183,
+      "step": 998
+    },
+    {
+      "epoch": 1.437410071942446,
+      "grad_norm": 0.37810065568343026,
+      "learning_rate": 7.174534295595911e-05,
+      "loss": 0.4219,
+      "step": 999
+    },
+    {
+      "epoch": 1.4388489208633093,
+      "grad_norm": 0.35778324901029257,
+      "learning_rate": 7.17208775298588e-05,
+      "loss": 0.4241,
+      "step": 1000
+    },
+    {
+      "epoch": 1.4402877697841727,
+      "grad_norm": 0.296505444912392,
+      "learning_rate": 7.169638008613158e-05,
+      "loss": 0.4189,
+      "step": 1001
+    },
+    {
+      "epoch": 1.441726618705036,
+      "grad_norm": 0.29440226960781385,
+      "learning_rate": 7.16718506495041e-05,
+      "loss": 0.4264,
+      "step": 1002
+    },
+    {
+      "epoch": 1.4431654676258994,
+      "grad_norm": 0.37142554533913175,
+      "learning_rate": 7.164728924473522e-05,
+      "loss": 0.4132,
+      "step": 1003
+    },
+    {
+      "epoch": 1.4446043165467626,
+      "grad_norm": 0.47661380552179405,
+      "learning_rate": 7.162269589661614e-05,
+      "loss": 0.4065,
+      "step": 1004
+    },
+    {
+      "epoch": 1.4460431654676258,
+      "grad_norm": 0.593874208360848,
+      "learning_rate": 7.15980706299703e-05,
+      "loss": 0.4155,
+      "step": 1005
+    },
+    {
+      "epoch": 1.4474820143884892,
+      "grad_norm": 0.7149350346081539,
+      "learning_rate": 7.15734134696533e-05,
+      "loss": 0.4255,
+      "step": 1006
+    },
+    {
+      "epoch": 1.4489208633093524,
+      "grad_norm": 0.7287849487118199,
+      "learning_rate": 7.1548724440553e-05,
+      "loss": 0.4238,
+      "step": 1007
+    },
+    {
+      "epoch": 1.4503597122302159,
+      "grad_norm": 0.5985591154146628,
+      "learning_rate": 7.152400356758937e-05,
+      "loss": 0.422,
+      "step": 1008
+    },
+    {
+      "epoch": 1.451798561151079,
+      "grad_norm": 0.5260381396690487,
+      "learning_rate": 7.149925087571456e-05,
+      "loss": 0.42,
+      "step": 1009
+    },
+    {
+      "epoch": 1.4532374100719425,
+      "grad_norm": 0.5131418767923352,
+      "learning_rate": 7.147446638991283e-05,
+      "loss": 0.4221,
+      "step": 1010
+    },
+    {
+      "epoch": 1.4546762589928057,
+      "grad_norm": 0.43675175574186137,
+      "learning_rate": 7.14496501352005e-05,
+      "loss": 0.4243,
+      "step": 1011
+    },
+    {
+      "epoch": 1.4561151079136692,
+      "grad_norm": 0.3857976039087925,
+      "learning_rate": 7.1424802136626e-05,
+      "loss": 0.4235,
+      "step": 1012
+    },
+    {
+      "epoch": 1.4575539568345324,
+      "grad_norm": 0.40581206323597546,
+      "learning_rate": 7.139992241926978e-05,
+      "loss": 0.4161,
+      "step": 1013
+    },
+    {
+      "epoch": 1.4589928057553956,
+      "grad_norm": 0.4172114133159447,
+      "learning_rate": 7.137501100824432e-05,
+      "loss": 0.4157,
+      "step": 1014
+    },
+    {
+      "epoch": 1.460431654676259,
+      "grad_norm": 0.3775952175793599,
+      "learning_rate": 7.135006792869405e-05,
+      "loss": 0.4154,
+      "step": 1015
+    },
+    {
+      "epoch": 1.4618705035971222,
+      "grad_norm": 0.3341118295254195,
+      "learning_rate": 7.132509320579542e-05,
+      "loss": 0.4262,
+      "step": 1016
+    },
+    {
+      "epoch": 1.4633093525179857,
+      "grad_norm": 0.3204508808850505,
+      "learning_rate": 7.130008686475677e-05,
+      "loss": 0.411,
+      "step": 1017
+    },
+    {
+      "epoch": 1.4647482014388489,
+      "grad_norm": 0.4070976049771881,
+      "learning_rate": 7.127504893081839e-05,
+      "loss": 0.4189,
+      "step": 1018
+    },
+    {
+      "epoch": 1.4661870503597123,
+      "grad_norm": 0.4921448353070135,
+      "learning_rate": 7.124997942925244e-05,
+      "loss": 0.4171,
+      "step": 1019
+    },
+    {
+      "epoch": 1.4676258992805755,
+      "grad_norm": 0.545224514417452,
+      "learning_rate": 7.122487838536295e-05,
+      "loss": 0.4181,
+      "step": 1020
+    },
+    {
+      "epoch": 1.469064748201439,
+      "grad_norm": 0.48394482335755673,
+      "learning_rate": 7.119974582448577e-05,
+      "loss": 0.423,
+      "step": 1021
+    },
+    {
+      "epoch": 1.4705035971223022,
+      "grad_norm": 0.43772370368974317,
+      "learning_rate": 7.11745817719886e-05,
+      "loss": 0.4181,
+      "step": 1022
+    },
+    {
+      "epoch": 1.4719424460431654,
+      "grad_norm": 0.3427269934601975,
+      "learning_rate": 7.114938625327088e-05,
+      "loss": 0.4063,
+      "step": 1023
+    },
+    {
+      "epoch": 1.4733812949640288,
+      "grad_norm": 0.2924781802740147,
+      "learning_rate": 7.112415929376385e-05,
+      "loss": 0.4113,
+      "step": 1024
+    },
+    {
+      "epoch": 1.474820143884892,
+      "grad_norm": 0.32198837101413197,
+      "learning_rate": 7.109890091893047e-05,
+      "loss": 0.4263,
+      "step": 1025
+    },
+    {
+      "epoch": 1.4762589928057555,
+      "grad_norm": 0.282277936977763,
+      "learning_rate": 7.107361115426537e-05,
+      "loss": 0.4194,
+      "step": 1026
+    },
+    {
+      "epoch": 1.4776978417266187,
+      "grad_norm": 0.25104791660310116,
+      "learning_rate": 7.104829002529496e-05,
+      "loss": 0.4229,
+      "step": 1027
+    },
+    {
+      "epoch": 1.4791366906474819,
+      "grad_norm": 0.27270860838716215,
+      "learning_rate": 7.102293755757721e-05,
+      "loss": 0.4223,
+      "step": 1028
+    },
+    {
+      "epoch": 1.4805755395683453,
+      "grad_norm": 0.26112130161175445,
+      "learning_rate": 7.099755377670177e-05,
+      "loss": 0.4196,
+      "step": 1029
+    },
+    {
+      "epoch": 1.4820143884892087,
+      "grad_norm": 0.2616174582828157,
+      "learning_rate": 7.097213870828989e-05,
+      "loss": 0.4147,
+      "step": 1030
+    },
+    {
+      "epoch": 1.483453237410072,
+      "grad_norm": 0.26475123111466153,
+      "learning_rate": 7.094669237799437e-05,
+      "loss": 0.4068,
+      "step": 1031
+    },
+    {
+      "epoch": 1.4848920863309352,
+      "grad_norm": 0.30854399100120633,
+      "learning_rate": 7.092121481149964e-05,
+      "loss": 0.4211,
+      "step": 1032
+    },
+    {
+      "epoch": 1.4863309352517986,
+      "grad_norm": 0.4252541911845257,
+      "learning_rate": 7.089570603452157e-05,
+      "loss": 0.4093,
+      "step": 1033
+    },
+    {
+      "epoch": 1.4877697841726618,
+      "grad_norm": 0.41391174764328076,
+      "learning_rate": 7.087016607280758e-05,
+      "loss": 0.422,
+      "step": 1034
+    },
+    {
+      "epoch": 1.4892086330935252,
+      "grad_norm": 0.31922865280493273,
+      "learning_rate": 7.084459495213658e-05,
+      "loss": 0.4265,
+      "step": 1035
+    },
+    {
+      "epoch": 1.4906474820143885,
+      "grad_norm": 0.2818316267301246,
+      "learning_rate": 7.081899269831888e-05,
+      "loss": 0.4161,
+      "step": 1036
+    },
+    {
+      "epoch": 1.4920863309352517,
+      "grad_norm": 0.2884472885297439,
+      "learning_rate": 7.079335933719625e-05,
+      "loss": 0.4057,
+      "step": 1037
+    },
+    {
+      "epoch": 1.493525179856115,
+      "grad_norm": 0.2823559617033364,
+      "learning_rate": 7.076769489464188e-05,
+      "loss": 0.4228,
+      "step": 1038
+    },
+    {
+      "epoch": 1.4949640287769785,
+      "grad_norm": 0.27447245904460427,
+      "learning_rate": 7.074199939656027e-05,
+      "loss": 0.4281,
+      "step": 1039
+    },
+    {
+      "epoch": 1.4964028776978417,
+      "grad_norm": 0.30292815899900255,
+      "learning_rate": 7.071627286888731e-05,
+      "loss": 0.4293,
+      "step": 1040
+    },
+    {
+      "epoch": 1.497841726618705,
+      "grad_norm": 0.3060320514171228,
+      "learning_rate": 7.06905153375902e-05,
+      "loss": 0.4198,
+      "step": 1041
+    },
+    {
+      "epoch": 1.4992805755395684,
+      "grad_norm": 0.27322000525441553,
+      "learning_rate": 7.066472682866744e-05,
+      "loss": 0.42,
+      "step": 1042
+    },
+    {
+      "epoch": 1.5007194244604316,
+      "grad_norm": 0.23469857234845679,
+      "learning_rate": 7.063890736814878e-05,
+      "loss": 0.4274,
+      "step": 1043
+    },
+    {
+      "epoch": 1.502158273381295,
+      "grad_norm": 0.2623497692808711,
+      "learning_rate": 7.061305698209524e-05,
+      "loss": 0.4152,
+      "step": 1044
+    },
+    {
+      "epoch": 1.5035971223021583,
+      "grad_norm": 0.3600783635419875,
+      "learning_rate": 7.058717569659901e-05,
+      "loss": 0.4085,
+      "step": 1045
+    },
+    {
+      "epoch": 1.5050359712230215,
+      "grad_norm": 0.43471456079737636,
+      "learning_rate": 7.05612635377835e-05,
+      "loss": 0.4191,
+      "step": 1046
+    },
+    {
+      "epoch": 1.506474820143885,
+      "grad_norm": 0.5194537867825783,
+      "learning_rate": 7.053532053180332e-05,
+      "loss": 0.4097,
+      "step": 1047
+    },
+    {
+      "epoch": 1.5079136690647483,
+      "grad_norm": 0.5496556685894735,
+      "learning_rate": 7.050934670484413e-05,
+      "loss": 0.4121,
+      "step": 1048
+    },
+    {
+      "epoch": 1.5093525179856115,
+      "grad_norm": 0.673626835157798,
+      "learning_rate": 7.048334208312273e-05,
+      "loss": 0.414,
+      "step": 1049
+    },
+    {
+      "epoch": 1.5107913669064748,
+      "grad_norm": 0.8602181463655494,
+      "learning_rate": 7.045730669288706e-05,
+      "loss": 0.4102,
+      "step": 1050
+    },
+    {
+      "epoch": 1.512230215827338,
+      "grad_norm": 0.8986188065364306,
+      "learning_rate": 7.043124056041606e-05,
+      "loss": 0.4251,
+      "step": 1051
+    },
+    {
+      "epoch": 1.5136690647482014,
+      "grad_norm": 0.907388183546702,
+      "learning_rate": 7.040514371201969e-05,
+      "loss": 0.4181,
+      "step": 1052
+    },
+    {
+      "epoch": 1.5151079136690648,
+      "grad_norm": 0.8893936713510535,
+      "learning_rate": 7.037901617403894e-05,
+      "loss": 0.4218,
+      "step": 1053
+    },
+    {
+      "epoch": 1.516546762589928,
+      "grad_norm": 0.8161310640612155,
+      "learning_rate": 7.035285797284578e-05,
+      "loss": 0.4195,
+      "step": 1054
+    },
+    {
+      "epoch": 1.5179856115107913,
+      "grad_norm": 0.7155314514536278,
+      "learning_rate": 7.032666913484313e-05,
+      "loss": 0.4168,
+      "step": 1055
+    },
+    {
+      "epoch": 1.5194244604316547,
+      "grad_norm": 0.5447772835854228,
+      "learning_rate": 7.030044968646481e-05,
+      "loss": 0.4081,
+      "step": 1056
+    },
+    {
+      "epoch": 1.5208633093525181,
+      "grad_norm": 0.30566260478398155,
+      "learning_rate": 7.027419965417556e-05,
+      "loss": 0.413,
+      "step": 1057
+    },
+    {
+      "epoch": 1.5223021582733813,
+      "grad_norm": 0.3168915642410363,
+      "learning_rate": 7.024791906447098e-05,
+      "loss": 0.4231,
+      "step": 1058
+    },
+    {
+      "epoch": 1.5237410071942445,
+      "grad_norm": 0.47337167935734187,
+      "learning_rate": 7.022160794387751e-05,
+      "loss": 0.4106,
+      "step": 1059
+    },
+    {
+      "epoch": 1.5251798561151078,
+      "grad_norm": 0.5600565641871665,
+      "learning_rate": 7.019526631895242e-05,
+      "loss": 0.4145,
+      "step": 1060
+    },
+    {
+      "epoch": 1.5266187050359712,
+      "grad_norm": 0.5347195703673441,
+      "learning_rate": 7.016889421628374e-05,
+      "loss": 0.414,
+      "step": 1061
+    },
+    {
+      "epoch": 1.5280575539568346,
+      "grad_norm": 0.5109593840142974,
+      "learning_rate": 7.014249166249032e-05,
+      "loss": 0.429,
+      "step": 1062
+    },
+    {
+      "epoch": 1.5294964028776978,
+      "grad_norm": 0.3632853891620355,
+      "learning_rate": 7.011605868422168e-05,
+      "loss": 0.4174,
+      "step": 1063
+    },
+    {
+      "epoch": 1.530935251798561,
+      "grad_norm": 0.43564028553592854,
+      "learning_rate": 7.00895953081581e-05,
+      "loss": 0.4224,
+      "step": 1064
+    },
+    {
+      "epoch": 1.5323741007194245,
+      "grad_norm": 0.4735520908422807,
+      "learning_rate": 7.00631015610105e-05,
+      "loss": 0.4201,
+      "step": 1065
+    },
+    {
+      "epoch": 1.533812949640288,
+      "grad_norm": 0.4633117691574936,
+      "learning_rate": 7.00365774695205e-05,
+      "loss": 0.4219,
+      "step": 1066
+    },
+    {
+      "epoch": 1.5352517985611511,
+      "grad_norm": 0.3744894751669038,
+      "learning_rate": 7.001002306046031e-05,
+      "loss": 0.4277,
+      "step": 1067
+    },
+    {
+      "epoch": 1.5366906474820143,
+      "grad_norm": 0.3909096065367158,
+      "learning_rate": 6.998343836063276e-05,
+      "loss": 0.42,
+      "step": 1068
+    },
+    {
+      "epoch": 1.5381294964028775,
+      "grad_norm": 0.35574327706934766,
+      "learning_rate": 6.995682339687125e-05,
+      "loss": 0.4203,
+      "step": 1069
+    },
+    {
+      "epoch": 1.539568345323741,
+      "grad_norm": 0.475660686372296,
+      "learning_rate": 6.993017819603973e-05,
+      "loss": 0.4198,
+      "step": 1070
+    },
+    {
+      "epoch": 1.5410071942446044,
+      "grad_norm": 0.5436500042351103,
+      "learning_rate": 6.990350278503267e-05,
+      "loss": 0.4149,
+      "step": 1071
+    },
+    {
+      "epoch": 1.5424460431654676,
+      "grad_norm": 0.5620520336725127,
+      "learning_rate": 6.9876797190775e-05,
+      "loss": 0.413,
+      "step": 1072
+    },
+    {
+      "epoch": 1.5438848920863308,
+      "grad_norm": 0.5336709442212036,
+      "learning_rate": 6.985006144022219e-05,
+      "loss": 0.4277,
+      "step": 1073
+    },
+    {
+      "epoch": 1.5453237410071943,
+      "grad_norm": 0.5000331369952123,
+      "learning_rate": 6.982329556036007e-05,
+      "loss": 0.4179,
+      "step": 1074
+    },
+    {
+      "epoch": 1.5467625899280577,
+      "grad_norm": 0.4747773119476601,
+      "learning_rate": 6.979649957820494e-05,
+      "loss": 0.4215,
+      "step": 1075
+    },
+    {
+      "epoch": 1.548201438848921,
+      "grad_norm": 0.3942932302309888,
+      "learning_rate": 6.976967352080345e-05,
+      "loss": 0.417,
+      "step": 1076
+    },
+    {
+      "epoch": 1.5496402877697841,
+      "grad_norm": 0.3524259398548682,
+      "learning_rate": 6.974281741523259e-05,
+      "loss": 0.4082,
+      "step": 1077
+    },
+    {
+      "epoch": 1.5510791366906473,
+      "grad_norm": 0.33503081944705626,
+      "learning_rate": 6.971593128859974e-05,
+      "loss": 0.4135,
+      "step": 1078
+    },
+    {
+      "epoch": 1.5525179856115108,
+      "grad_norm": 0.294686683623946,
+      "learning_rate": 6.968901516804254e-05,
+      "loss": 0.4099,
+      "step": 1079
+    },
+    {
+      "epoch": 1.5539568345323742,
+      "grad_norm": 0.3048950261303681,
+      "learning_rate": 6.966206908072891e-05,
+      "loss": 0.4171,
+      "step": 1080
+    },
+    {
+      "epoch": 1.5553956834532374,
+      "grad_norm": 0.40859414851177817,
+      "learning_rate": 6.963509305385701e-05,
+      "loss": 0.4209,
+      "step": 1081
+    },
+    {
+      "epoch": 1.5568345323741006,
+      "grad_norm": 0.4126086910978001,
+      "learning_rate": 6.960808711465524e-05,
+      "loss": 0.4225,
+      "step": 1082
+    },
+    {
+      "epoch": 1.558273381294964,
+      "grad_norm": 0.3863251061315781,
+      "learning_rate": 6.958105129038216e-05,
+      "loss": 0.4119,
+      "step": 1083
+    },
+    {
+      "epoch": 1.5597122302158275,
+      "grad_norm": 0.32697010940880766,
+      "learning_rate": 6.955398560832654e-05,
+      "loss": 0.4126,
+      "step": 1084
+    },
+    {
+      "epoch": 1.5611510791366907,
+      "grad_norm": 0.30920484763692224,
+      "learning_rate": 6.952689009580724e-05,
+      "loss": 0.4145,
+      "step": 1085
+    },
+    {
+      "epoch": 1.562589928057554,
+      "grad_norm": 0.3355958190960155,
+      "learning_rate": 6.949976478017327e-05,
+      "loss": 0.429,
+      "step": 1086
+    },
+    {
+      "epoch": 1.5640287769784171,
+      "grad_norm": 0.33720997246138995,
+      "learning_rate": 6.947260968880369e-05,
+      "loss": 0.4061,
+      "step": 1087
+    },
+    {
+      "epoch": 1.5654676258992806,
+      "grad_norm": 0.365706540668333,
+      "learning_rate": 6.944542484910763e-05,
+      "loss": 0.4196,
+      "step": 1088
+    },
+    {
+      "epoch": 1.566906474820144,
+      "grad_norm": 0.3191029511156109,
+      "learning_rate": 6.941821028852424e-05,
+      "loss": 0.4143,
+      "step": 1089
+    },
+    {
+      "epoch": 1.5683453237410072,
+      "grad_norm": 0.3337584496700208,
+      "learning_rate": 6.939096603452269e-05,
+      "loss": 0.4056,
+      "step": 1090
+    },
+    {
+      "epoch": 1.5697841726618704,
+      "grad_norm": 0.382575244198341,
+      "learning_rate": 6.93636921146021e-05,
+      "loss": 0.421,
+      "step": 1091
+    },
+    {
+      "epoch": 1.5712230215827339,
+      "grad_norm": 0.3603814261628815,
+      "learning_rate": 6.933638855629153e-05,
+      "loss": 0.4229,
+      "step": 1092
+    },
+    {
+      "epoch": 1.572661870503597,
+      "grad_norm": 0.33198688932968207,
+      "learning_rate": 6.930905538714995e-05,
+      "loss": 0.4063,
+      "step": 1093
+    },
+    {
+      "epoch": 1.5741007194244605,
+      "grad_norm": 0.3319118654900316,
+      "learning_rate": 6.928169263476628e-05,
+      "loss": 0.4122,
+      "step": 1094
+    },
+    {
+      "epoch": 1.5755395683453237,
+      "grad_norm": 0.24612081962950533,
+      "learning_rate": 6.92543003267592e-05,
+      "loss": 0.4256,
+      "step": 1095
+    },
+    {
+      "epoch": 1.576978417266187,
+      "grad_norm": 0.24578792236490904,
+      "learning_rate": 6.922687849077729e-05,
+      "loss": 0.4127,
+      "step": 1096
+    },
+    {
+      "epoch": 1.5784172661870504,
+      "grad_norm": 0.3620980186429764,
+      "learning_rate": 6.919942715449893e-05,
+      "loss": 0.417,
+      "step": 1097
+    },
+    {
+      "epoch": 1.5798561151079138,
+      "grad_norm": 0.4056281865573301,
+      "learning_rate": 6.917194634563225e-05,
+      "loss": 0.4171,
+      "step": 1098
+    },
+    {
+      "epoch": 1.581294964028777,
+      "grad_norm": 0.4151741231993419,
+      "learning_rate": 6.914443609191514e-05,
+      "loss": 0.4202,
+      "step": 1099
+    },
+    {
+      "epoch": 1.5827338129496402,
+      "grad_norm": 0.42372551750913784,
+      "learning_rate": 6.911689642111523e-05,
+      "loss": 0.4258,
+      "step": 1100
+    },
+    {
+      "epoch": 1.5841726618705037,
+      "grad_norm": 0.33954014694508594,
+      "learning_rate": 6.90893273610298e-05,
+      "loss": 0.4084,
+      "step": 1101
+    },
+    {
+      "epoch": 1.5856115107913669,
+      "grad_norm": 0.3472411886216252,
+      "learning_rate": 6.906172893948585e-05,
+      "loss": 0.4279,
+      "step": 1102
+    },
+    {
+      "epoch": 1.5870503597122303,
+      "grad_norm": 0.28252822327431176,
+      "learning_rate": 6.903410118433996e-05,
+      "loss": 0.4113,
+      "step": 1103
+    },
+    {
+      "epoch": 1.5884892086330935,
+      "grad_norm": 0.21511848522747434,
+      "learning_rate": 6.900644412347836e-05,
+      "loss": 0.4162,
+      "step": 1104
+    },
+    {
+      "epoch": 1.5899280575539567,
+      "grad_norm": 0.3377842110574489,
+      "learning_rate": 6.897875778481682e-05,
+      "loss": 0.4183,
+      "step": 1105
+    },
+    {
+      "epoch": 1.5913669064748202,
+      "grad_norm": 0.413068015031045,
+      "learning_rate": 6.89510421963007e-05,
+      "loss": 0.418,
+      "step": 1106
+    },
+    {
+      "epoch": 1.5928057553956836,
+      "grad_norm": 0.4520641357977242,
+      "learning_rate": 6.892329738590489e-05,
+      "loss": 0.4089,
+      "step": 1107
+    },
+    {
+      "epoch": 1.5942446043165468,
+      "grad_norm": 0.5190681917869406,
+      "learning_rate": 6.889552338163372e-05,
+      "loss": 0.4165,
+      "step": 1108
+    },
+    {
+      "epoch": 1.59568345323741,
+      "grad_norm": 0.5658167151358489,
+      "learning_rate": 6.886772021152104e-05,
+      "loss": 0.4077,
+      "step": 1109
+    },
+    {
+      "epoch": 1.5971223021582732,
+      "grad_norm": 0.6251081781208279,
+      "learning_rate": 6.883988790363009e-05,
+      "loss": 0.4187,
+      "step": 1110
+    },
+    {
+      "epoch": 1.5985611510791367,
+      "grad_norm": 0.6977729292257839,
+      "learning_rate": 6.881202648605359e-05,
+      "loss": 0.4192,
+      "step": 1111
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.6784368627526026,
+      "learning_rate": 6.878413598691358e-05,
+      "loss": 0.4081,
+      "step": 1112
+    },
+    {
+      "epoch": 1.6014388489208633,
+      "grad_norm": 0.611175666696122,
+      "learning_rate": 6.875621643436147e-05,
+      "loss": 0.4271,
+      "step": 1113
+    },
+    {
+      "epoch": 1.6028776978417265,
+      "grad_norm": 0.5275664308141039,
+      "learning_rate": 6.872826785657802e-05,
+      "loss": 0.4148,
+      "step": 1114
+    },
+    {
+      "epoch": 1.60431654676259,
+      "grad_norm": 0.44921141035285606,
+      "learning_rate": 6.870029028177324e-05,
+      "loss": 0.4087,
+      "step": 1115
+    },
+    {
+      "epoch": 1.6057553956834534,
+      "grad_norm": 0.4219492984920089,
+      "learning_rate": 6.867228373818648e-05,
+      "loss": 0.4096,
+      "step": 1116
+    },
+    {
+      "epoch": 1.6071942446043166,
+      "grad_norm": 0.4661276903722583,
+      "learning_rate": 6.864424825408624e-05,
+      "loss": 0.4155,
+      "step": 1117
+    },
+    {
+      "epoch": 1.6086330935251798,
+      "grad_norm": 0.45069842468613974,
+      "learning_rate": 6.861618385777028e-05,
+      "loss": 0.4221,
+      "step": 1118
+    },
+    {
+      "epoch": 1.610071942446043,
+      "grad_norm": 0.4113236965059679,
+      "learning_rate": 6.858809057756558e-05,
+      "loss": 0.4117,
+      "step": 1119
+    },
+    {
+      "epoch": 1.6115107913669064,
+      "grad_norm": 0.4019565134513667,
+      "learning_rate": 6.855996844182819e-05,
+      "loss": 0.4102,
+      "step": 1120
+    },
+    {
+      "epoch": 1.6129496402877699,
+      "grad_norm": 0.46925268431368816,
+      "learning_rate": 6.853181747894334e-05,
+      "loss": 0.4158,
+      "step": 1121
+    },
+    {
+      "epoch": 1.614388489208633,
+      "grad_norm": 0.5308935521837395,
+      "learning_rate": 6.850363771732536e-05,
+      "loss": 0.411,
+      "step": 1122
+    },
+    {
+      "epoch": 1.6158273381294963,
+      "grad_norm": 0.5644359644344518,
+      "learning_rate": 6.847542918541762e-05,
+      "loss": 0.4185,
+      "step": 1123
+    },
+    {
+      "epoch": 1.6172661870503597,
+      "grad_norm": 0.5485119854230217,
+      "learning_rate": 6.844719191169254e-05,
+      "loss": 0.4187,
+      "step": 1124
+    },
+    {
+      "epoch": 1.6187050359712232,
+      "grad_norm": 0.46044082739927455,
+      "learning_rate": 6.841892592465158e-05,
+      "loss": 0.4059,
+      "step": 1125
+    },
+    {
+      "epoch": 1.6201438848920864,
+      "grad_norm": 0.30868193509334346,
+      "learning_rate": 6.839063125282512e-05,
+      "loss": 0.4138,
+      "step": 1126
+    },
+    {
+      "epoch": 1.6215827338129496,
+      "grad_norm": 0.28786051831720627,
+      "learning_rate": 6.836230792477256e-05,
+      "loss": 0.4069,
+      "step": 1127
+    },
+    {
+      "epoch": 1.6230215827338128,
+      "grad_norm": 0.4336899650080154,
+      "learning_rate": 6.833395596908217e-05,
+      "loss": 0.4225,
+      "step": 1128
+    },
+    {
+      "epoch": 1.6244604316546762,
+      "grad_norm": 0.4552106798230225,
+      "learning_rate": 6.830557541437114e-05,
+      "loss": 0.3948,
+      "step": 1129
+    },
+    {
+      "epoch": 1.6258992805755397,
+      "grad_norm": 0.3344228802154637,
+      "learning_rate": 6.827716628928556e-05,
+      "loss": 0.4183,
+      "step": 1130
+    },
+    {
+      "epoch": 1.6273381294964029,
+      "grad_norm": 0.2719259797015168,
+      "learning_rate": 6.824872862250028e-05,
+      "loss": 0.4107,
+      "step": 1131
+    },
+    {
+      "epoch": 1.628776978417266,
+      "grad_norm": 0.38746826794473005,
+      "learning_rate": 6.822026244271903e-05,
+      "loss": 0.4127,
+      "step": 1132
+    },
+    {
+      "epoch": 1.6302158273381295,
+      "grad_norm": 0.40749598860469377,
+      "learning_rate": 6.819176777867425e-05,
+      "loss": 0.4161,
+      "step": 1133
+    },
+    {
+      "epoch": 1.631654676258993,
+      "grad_norm": 0.28357078247401757,
+      "learning_rate": 6.816324465912723e-05,
+      "loss": 0.4173,
+      "step": 1134
+    },
+    {
+      "epoch": 1.6330935251798562,
+      "grad_norm": 0.2497149857828995,
+      "learning_rate": 6.813469311286789e-05,
+      "loss": 0.4053,
+      "step": 1135
+    },
+    {
+      "epoch": 1.6345323741007194,
+      "grad_norm": 0.32384723529170906,
+      "learning_rate": 6.810611316871488e-05,
+      "loss": 0.4203,
+      "step": 1136
+    },
+    {
+      "epoch": 1.6359712230215826,
+      "grad_norm": 0.4087164952132661,
+      "learning_rate": 6.80775048555155e-05,
+      "loss": 0.4228,
+      "step": 1137
+    },
+    {
+      "epoch": 1.637410071942446,
+      "grad_norm": 0.36329025160015227,
+      "learning_rate": 6.804886820214572e-05,
+      "loss": 0.4136,
+      "step": 1138
+    },
+    {
+      "epoch": 1.6388489208633095,
+      "grad_norm": 0.32032361616204535,
+      "learning_rate": 6.802020323751008e-05,
+      "loss": 0.4148,
+      "step": 1139
+    },
+    {
+      "epoch": 1.6402877697841727,
+      "grad_norm": 0.3441846476219566,
+      "learning_rate": 6.799150999054169e-05,
+      "loss": 0.4168,
+      "step": 1140
+    },
+    {
+      "epoch": 1.641726618705036,
+      "grad_norm": 0.76682242723555,
+      "learning_rate": 6.796278849020225e-05,
+      "loss": 0.4102,
+      "step": 1141
+    },
+    {
+      "epoch": 1.6431654676258993,
+      "grad_norm": 0.33620190524925814,
+      "learning_rate": 6.79340387654819e-05,
+      "loss": 0.4211,
+      "step": 1142
+    },
+    {
+      "epoch": 1.6446043165467625,
+      "grad_norm": 0.43533432543301903,
+      "learning_rate": 6.790526084539939e-05,
+      "loss": 0.4143,
+      "step": 1143
+    },
+    {
+      "epoch": 1.646043165467626,
+      "grad_norm": 0.503682323862886,
+      "learning_rate": 6.787645475900182e-05,
+      "loss": 0.4121,
+      "step": 1144
+    },
+    {
+      "epoch": 1.6474820143884892,
+      "grad_norm": 0.4407492323966843,
+      "learning_rate": 6.784762053536475e-05,
+      "loss": 0.4061,
+      "step": 1145
+    },
+    {
+      "epoch": 1.6489208633093524,
+      "grad_norm": 0.3870577172781102,
+      "learning_rate": 6.781875820359216e-05,
+      "loss": 0.4261,
+      "step": 1146
+    },
+    {
+      "epoch": 1.6503597122302158,
+      "grad_norm": 0.34174235825815463,
+      "learning_rate": 6.778986779281639e-05,
+      "loss": 0.4285,
+      "step": 1147
+    },
+    {
+      "epoch": 1.6517985611510793,
+      "grad_norm": 0.34649866044879724,
+      "learning_rate": 6.776094933219811e-05,
+      "loss": 0.4176,
+      "step": 1148
+    },
+    {
+      "epoch": 1.6532374100719425,
+      "grad_norm": 0.33122534094477013,
+      "learning_rate": 6.773200285092633e-05,
+      "loss": 0.4269,
+      "step": 1149
+    },
+    {
+      "epoch": 1.6546762589928057,
+      "grad_norm": 0.38469081599419475,
+      "learning_rate": 6.770302837821833e-05,
+      "loss": 0.4193,
+      "step": 1150
+    },
+    {
+      "epoch": 1.6561151079136691,
+      "grad_norm": 0.40768423839818896,
+      "learning_rate": 6.767402594331961e-05,
+      "loss": 0.4228,
+      "step": 1151
+    },
+    {
+      "epoch": 1.6575539568345323,
+      "grad_norm": 0.45279080893685925,
+      "learning_rate": 6.764499557550396e-05,
+      "loss": 0.4164,
+      "step": 1152
+    },
+    {
+      "epoch": 1.6589928057553958,
+      "grad_norm": 0.4903719398809837,
+      "learning_rate": 6.761593730407329e-05,
+      "loss": 0.4085,
+      "step": 1153
+    },
+    {
+      "epoch": 1.660431654676259,
+      "grad_norm": 0.487492750001549,
+      "learning_rate": 6.758685115835776e-05,
+      "loss": 0.4168,
+      "step": 1154
+    },
+    {
+      "epoch": 1.6618705035971222,
+      "grad_norm": 0.5583506134292617,
+      "learning_rate": 6.755773716771555e-05,
+      "loss": 0.4145,
+      "step": 1155
+    },
+    {
+      "epoch": 1.6633093525179856,
+      "grad_norm": 0.5956179425579616,
+      "learning_rate": 6.752859536153306e-05,
+      "loss": 0.4223,
+      "step": 1156
+    },
+    {
+      "epoch": 1.664748201438849,
+      "grad_norm": 0.6033217069177361,
+      "learning_rate": 6.749942576922473e-05,
+      "loss": 0.4039,
+      "step": 1157
+    },
+    {
+      "epoch": 1.6661870503597123,
+      "grad_norm": 0.6583228089216405,
+      "learning_rate": 6.7470228420233e-05,
+      "loss": 0.4217,
+      "step": 1158
+    },
+    {
+      "epoch": 1.6676258992805755,
+      "grad_norm": 0.6680499004987813,
+      "learning_rate": 6.744100334402836e-05,
+      "loss": 0.4115,
+      "step": 1159
+    },
+    {
+      "epoch": 1.6690647482014387,
+      "grad_norm": 0.5380725649234539,
+      "learning_rate": 6.741175057010932e-05,
+      "loss": 0.4228,
+      "step": 1160
+    },
+    {
+      "epoch": 1.6705035971223021,
+      "grad_norm": 0.5426590107448707,
+      "learning_rate": 6.738247012800228e-05,
+      "loss": 0.4263,
+      "step": 1161
+    },
+    {
+      "epoch": 1.6719424460431656,
+      "grad_norm": 0.3446250027390038,
+      "learning_rate": 6.735316204726163e-05,
+      "loss": 0.406,
+      "step": 1162
+    },
+    {
+      "epoch": 1.6733812949640288,
+      "grad_norm": 0.363875910280198,
+      "learning_rate": 6.732382635746961e-05,
+      "loss": 0.4086,
+      "step": 1163
+    },
+    {
+      "epoch": 1.674820143884892,
+      "grad_norm": 0.3644991778739921,
+      "learning_rate": 6.729446308823635e-05,
+      "loss": 0.4203,
+      "step": 1164
+    },
+    {
+      "epoch": 1.6762589928057554,
+      "grad_norm": 0.349716535674518,
+      "learning_rate": 6.72650722691998e-05,
+      "loss": 0.4159,
+      "step": 1165
+    },
+    {
+      "epoch": 1.6776978417266188,
+      "grad_norm": 0.3514867098428374,
+      "learning_rate": 6.723565393002576e-05,
+      "loss": 0.405,
+      "step": 1166
+    },
+    {
+      "epoch": 1.679136690647482,
+      "grad_norm": 0.4165407906592057,
+      "learning_rate": 6.720620810040776e-05,
+      "loss": 0.4055,
+      "step": 1167
+    },
+    {
+      "epoch": 1.6805755395683453,
+      "grad_norm": 0.5052046649138455,
+      "learning_rate": 6.717673481006709e-05,
+      "loss": 0.4189,
+      "step": 1168
+    },
+    {
+      "epoch": 1.6820143884892085,
+      "grad_norm": 0.5142496765109894,
+      "learning_rate": 6.714723408875279e-05,
+      "loss": 0.4089,
+      "step": 1169
+    },
+    {
+      "epoch": 1.683453237410072,
+      "grad_norm": 0.4553648017174242,
+      "learning_rate": 6.711770596624153e-05,
+      "loss": 0.417,
+      "step": 1170
+    },
+    {
+      "epoch": 1.6848920863309353,
+      "grad_norm": 0.3730268976826378,
+      "learning_rate": 6.708815047233768e-05,
+      "loss": 0.4156,
+      "step": 1171
+    },
+    {
+      "epoch": 1.6863309352517986,
+      "grad_norm": 0.37168971826069924,
+      "learning_rate": 6.705856763687324e-05,
+      "loss": 0.4131,
+      "step": 1172
+    },
+    {
+      "epoch": 1.6877697841726618,
+      "grad_norm": 0.3092228093700726,
+      "learning_rate": 6.702895748970776e-05,
+      "loss": 0.4152,
+      "step": 1173
+    },
+    {
+      "epoch": 1.6892086330935252,
+      "grad_norm": 0.38955015639100576,
+      "learning_rate": 6.699932006072842e-05,
+      "loss": 0.4162,
+      "step": 1174
+    },
+    {
+      "epoch": 1.6906474820143886,
+      "grad_norm": 0.5080735895636728,
+      "learning_rate": 6.69696553798499e-05,
+      "loss": 0.4114,
+      "step": 1175
+    },
+    {
+      "epoch": 1.6920863309352518,
+      "grad_norm": 0.4620152095782709,
+      "learning_rate": 6.693996347701442e-05,
+      "loss": 0.4149,
+      "step": 1176
+    },
+    {
+      "epoch": 1.693525179856115,
+      "grad_norm": 0.47857199036868725,
+      "learning_rate": 6.691024438219159e-05,
+      "loss": 0.4141,
+      "step": 1177
+    },
+    {
+      "epoch": 1.6949640287769783,
+      "grad_norm": 0.48308800583112116,
+      "learning_rate": 6.688049812537857e-05,
+      "loss": 0.4176,
+      "step": 1178
+    },
+    {
+      "epoch": 1.6964028776978417,
+      "grad_norm": 0.48259699395769545,
+      "learning_rate": 6.685072473659989e-05,
+      "loss": 0.4157,
+      "step": 1179
+    },
+    {
+      "epoch": 1.6978417266187051,
+      "grad_norm": 0.36203361838439907,
+      "learning_rate": 6.682092424590747e-05,
+      "loss": 0.4083,
+      "step": 1180
+    },
+    {
+      "epoch": 1.6992805755395683,
+      "grad_norm": 0.31495479128819703,
+      "learning_rate": 6.679109668338057e-05,
+      "loss": 0.4087,
+      "step": 1181
+    },
+    {
+      "epoch": 1.7007194244604316,
+      "grad_norm": 0.3318572949891459,
+      "learning_rate": 6.676124207912582e-05,
+      "loss": 0.4079,
+      "step": 1182
+    },
+    {
+      "epoch": 1.702158273381295,
+      "grad_norm": 0.41650781673916587,
+      "learning_rate": 6.673136046327707e-05,
+      "loss": 0.4184,
+      "step": 1183
+    },
+    {
+      "epoch": 1.7035971223021584,
+      "grad_norm": 0.5589325931977038,
+      "learning_rate": 6.670145186599552e-05,
+      "loss": 0.4135,
+      "step": 1184
+    },
+    {
+      "epoch": 1.7050359712230216,
+      "grad_norm": 0.6913999906311661,
+      "learning_rate": 6.667151631746953e-05,
+      "loss": 0.413,
+      "step": 1185
+    },
+    {
+      "epoch": 1.7064748201438849,
+      "grad_norm": 0.6701838285934596,
+      "learning_rate": 6.664155384791473e-05,
+      "loss": 0.4095,
+      "step": 1186
+    },
+    {
+      "epoch": 1.707913669064748,
+      "grad_norm": 0.5579192511025082,
+      "learning_rate": 6.661156448757386e-05,
+      "loss": 0.4234,
+      "step": 1187
+    },
+    {
+      "epoch": 1.7093525179856115,
+      "grad_norm": 0.57084696911057,
+      "learning_rate": 6.658154826671685e-05,
+      "loss": 0.4214,
+      "step": 1188
+    },
+    {
+      "epoch": 1.710791366906475,
+      "grad_norm": 0.538521220072237,
+      "learning_rate": 6.655150521564072e-05,
+      "loss": 0.4177,
+      "step": 1189
+    },
+    {
+      "epoch": 1.7122302158273381,
+      "grad_norm": 0.4459896651146677,
+      "learning_rate": 6.652143536466955e-05,
+      "loss": 0.4151,
+      "step": 1190
+    },
+    {
+      "epoch": 1.7136690647482014,
+      "grad_norm": 0.4425563287727174,
+      "learning_rate": 6.649133874415454e-05,
+      "loss": 0.4261,
+      "step": 1191
+    },
+    {
+      "epoch": 1.7151079136690648,
+      "grad_norm": 0.5044467565268774,
+      "learning_rate": 6.646121538447382e-05,
+      "loss": 0.4216,
+      "step": 1192
+    },
+    {
+      "epoch": 1.7165467625899282,
+      "grad_norm": 0.3354306043962459,
+      "learning_rate": 6.643106531603259e-05,
+      "loss": 0.4219,
+      "step": 1193
+    },
+    {
+      "epoch": 1.7179856115107914,
+      "grad_norm": 0.39357004259393435,
+      "learning_rate": 6.640088856926294e-05,
+      "loss": 0.4076,
+      "step": 1194
+    },
+    {
+      "epoch": 1.7194244604316546,
+      "grad_norm": 0.5273217051181752,
+      "learning_rate": 6.637068517462395e-05,
+      "loss": 0.4218,
+      "step": 1195
+    },
+    {
+      "epoch": 1.7208633093525179,
+      "grad_norm": 0.6087745983365078,
+      "learning_rate": 6.634045516260156e-05,
+      "loss": 0.4126,
+      "step": 1196
+    },
+    {
+      "epoch": 1.7223021582733813,
+      "grad_norm": 0.47047396262886537,
+      "learning_rate": 6.631019856370856e-05,
+      "loss": 0.4115,
+      "step": 1197
+    },
+    {
+      "epoch": 1.7237410071942447,
+      "grad_norm": 0.33245740608012014,
+      "learning_rate": 6.627991540848464e-05,
+      "loss": 0.4201,
+      "step": 1198
+    },
+    {
+      "epoch": 1.725179856115108,
+      "grad_norm": 0.41922774163862975,
+      "learning_rate": 6.624960572749622e-05,
+      "loss": 0.4126,
+      "step": 1199
+    },
+    {
+      "epoch": 1.7266187050359711,
+      "grad_norm": 0.4384331345863592,
+      "learning_rate": 6.621926955133657e-05,
+      "loss": 0.4251,
+      "step": 1200
+    },
+    {
+      "epoch": 1.7280575539568346,
+      "grad_norm": 0.4079618292679597,
+      "learning_rate": 6.618890691062561e-05,
+      "loss": 0.4146,
+      "step": 1201
+    },
+    {
+      "epoch": 1.7294964028776978,
+      "grad_norm": 0.36251729181268255,
+      "learning_rate": 6.615851783601006e-05,
+      "loss": 0.4125,
+      "step": 1202
+    },
+    {
+      "epoch": 1.7309352517985612,
+      "grad_norm": 0.3201921174826245,
+      "learning_rate": 6.612810235816326e-05,
+      "loss": 0.4177,
+      "step": 1203
+    },
+    {
+      "epoch": 1.7323741007194244,
+      "grad_norm": 0.30717496698971825,
+      "learning_rate": 6.609766050778525e-05,
+      "loss": 0.4134,
+      "step": 1204
+    },
+    {
+      "epoch": 1.7338129496402876,
+      "grad_norm": 0.34144771508129956,
+      "learning_rate": 6.606719231560265e-05,
+      "loss": 0.4103,
+      "step": 1205
+    },
+    {
+      "epoch": 1.735251798561151,
+      "grad_norm": 0.3399546603031398,
+      "learning_rate": 6.60366978123687e-05,
+      "loss": 0.4168,
+      "step": 1206
+    },
+    {
+      "epoch": 1.7366906474820145,
+      "grad_norm": 0.3136693953734484,
+      "learning_rate": 6.600617702886314e-05,
+      "loss": 0.4127,
+      "step": 1207
+    },
+    {
+      "epoch": 1.7381294964028777,
+      "grad_norm": 0.31926857746325077,
+      "learning_rate": 6.597562999589233e-05,
+      "loss": 0.4101,
+      "step": 1208
+    },
+    {
+      "epoch": 1.739568345323741,
+      "grad_norm": 0.358660556486896,
+      "learning_rate": 6.594505674428903e-05,
+      "loss": 0.4203,
+      "step": 1209
+    },
+    {
+      "epoch": 1.7410071942446042,
+      "grad_norm": 0.3659716126309705,
+      "learning_rate": 6.59144573049125e-05,
+      "loss": 0.4167,
+      "step": 1210
+    },
+    {
+      "epoch": 1.7424460431654676,
+      "grad_norm": 0.4052981700072071,
+      "learning_rate": 6.588383170864849e-05,
+      "loss": 0.4084,
+      "step": 1211
+    },
+    {
+      "epoch": 1.743884892086331,
+      "grad_norm": 0.37102819892247557,
+      "learning_rate": 6.585317998640903e-05,
+      "loss": 0.4114,
+      "step": 1212
+    },
+    {
+      "epoch": 1.7453237410071942,
+      "grad_norm": 0.37890137130457685,
+      "learning_rate": 6.582250216913265e-05,
+      "loss": 0.4049,
+      "step": 1213
+    },
+    {
+      "epoch": 1.7467625899280574,
+      "grad_norm": 0.47660635938468554,
+      "learning_rate": 6.579179828778414e-05,
+      "loss": 0.4317,
+      "step": 1214
+    },
+    {
+      "epoch": 1.7482014388489209,
+      "grad_norm": 0.5403075083072826,
+      "learning_rate": 6.576106837335458e-05,
+      "loss": 0.4058,
+      "step": 1215
+    },
+    {
+      "epoch": 1.7496402877697843,
+      "grad_norm": 0.5866191248034582,
+      "learning_rate": 6.573031245686142e-05,
+      "loss": 0.4115,
+      "step": 1216
+    },
+    {
+      "epoch": 1.7510791366906475,
+      "grad_norm": 0.504537377126103,
+      "learning_rate": 6.569953056934826e-05,
+      "loss": 0.4177,
+      "step": 1217
+    },
+    {
+      "epoch": 1.7525179856115107,
+      "grad_norm": 0.43781511223917,
+      "learning_rate": 6.566872274188496e-05,
+      "loss": 0.4208,
+      "step": 1218
+    },
+    {
+      "epoch": 1.753956834532374,
+      "grad_norm": 0.4636740890078483,
+      "learning_rate": 6.563788900556756e-05,
+      "loss": 0.4162,
+      "step": 1219
+    },
+    {
+      "epoch": 1.7553956834532374,
+      "grad_norm": 0.4134745031820477,
+      "learning_rate": 6.560702939151826e-05,
+      "loss": 0.4032,
+      "step": 1220
+    },
+    {
+      "epoch": 1.7568345323741008,
+      "grad_norm": 0.44431900170972155,
+      "learning_rate": 6.557614393088534e-05,
+      "loss": 0.4132,
+      "step": 1221
+    },
+    {
+      "epoch": 1.758273381294964,
+      "grad_norm": 0.43696209712924794,
+      "learning_rate": 6.554523265484321e-05,
+      "loss": 0.4109,
+      "step": 1222
+    },
+    {
+      "epoch": 1.7597122302158272,
+      "grad_norm": 0.3936897209104171,
+      "learning_rate": 6.551429559459231e-05,
+      "loss": 0.4185,
+      "step": 1223
+    },
+    {
+      "epoch": 1.7611510791366907,
+      "grad_norm": 0.3723578226039111,
+      "learning_rate": 6.548333278135915e-05,
+      "loss": 0.4089,
+      "step": 1224
+    },
+    {
+      "epoch": 1.762589928057554,
+      "grad_norm": 0.3569593718648685,
+      "learning_rate": 6.545234424639616e-05,
+      "loss": 0.3998,
+      "step": 1225
+    },
+    {
+      "epoch": 1.7640287769784173,
+      "grad_norm": 0.3178602355572999,
+      "learning_rate": 6.542133002098178e-05,
+      "loss": 0.4038,
+      "step": 1226
+    },
+    {
+      "epoch": 1.7654676258992805,
+      "grad_norm": 0.3429199290938974,
+      "learning_rate": 6.53902901364204e-05,
+      "loss": 0.4046,
+      "step": 1227
+    },
+    {
+      "epoch": 1.7669064748201437,
+      "grad_norm": 0.35686322312114677,
+      "learning_rate": 6.535922462404226e-05,
+      "loss": 0.409,
+      "step": 1228
+    },
+    {
+      "epoch": 1.7683453237410072,
+      "grad_norm": 0.33994382692535474,
+      "learning_rate": 6.53281335152035e-05,
+      "loss": 0.4175,
+      "step": 1229
+    },
+    {
+      "epoch": 1.7697841726618706,
+      "grad_norm": 0.27286996566385857,
+      "learning_rate": 6.529701684128608e-05,
+      "loss": 0.4114,
+      "step": 1230
+    },
+    {
+      "epoch": 1.7712230215827338,
+      "grad_norm": 0.3375280762047218,
+      "learning_rate": 6.526587463369779e-05,
+      "loss": 0.4111,
+      "step": 1231
+    },
+    {
+      "epoch": 1.772661870503597,
+      "grad_norm": 0.32409851741711043,
+      "learning_rate": 6.523470692387215e-05,
+      "loss": 0.4067,
+      "step": 1232
+    },
+    {
+      "epoch": 1.7741007194244605,
+      "grad_norm": 0.32465639257449525,
+      "learning_rate": 6.520351374326846e-05,
+      "loss": 0.425,
+      "step": 1233
+    },
+    {
+      "epoch": 1.775539568345324,
+      "grad_norm": 0.3072756505982714,
+      "learning_rate": 6.51722951233717e-05,
+      "loss": 0.4048,
+      "step": 1234
+    },
+    {
+      "epoch": 1.776978417266187,
+      "grad_norm": 0.296270829699046,
+      "learning_rate": 6.514105109569254e-05,
+      "loss": 0.4134,
+      "step": 1235
+    },
+    {
+      "epoch": 1.7784172661870503,
+      "grad_norm": 0.3815748221438414,
+      "learning_rate": 6.510978169176731e-05,
+      "loss": 0.4147,
+      "step": 1236
+    },
+    {
+      "epoch": 1.7798561151079135,
+      "grad_norm": 0.4231476871569944,
+      "learning_rate": 6.507848694315794e-05,
+      "loss": 0.4042,
+      "step": 1237
+    },
+    {
+      "epoch": 1.781294964028777,
+      "grad_norm": 0.38830569275115767,
+      "learning_rate": 6.504716688145192e-05,
+      "loss": 0.4198,
+      "step": 1238
+    },
+    {
+      "epoch": 1.7827338129496404,
+      "grad_norm": 0.4021387335238997,
+      "learning_rate": 6.501582153826235e-05,
+      "loss": 0.4188,
+      "step": 1239
+    },
+    {
+      "epoch": 1.7841726618705036,
+      "grad_norm": 0.33591783933274055,
+      "learning_rate": 6.498445094522776e-05,
+      "loss": 0.4141,
+      "step": 1240
+    },
+    {
+      "epoch": 1.7856115107913668,
+      "grad_norm": 0.3400651749476804,
+      "learning_rate": 6.495305513401226e-05,
+      "loss": 0.4161,
+      "step": 1241
+    },
+    {
+      "epoch": 1.7870503597122303,
+      "grad_norm": 0.420865484735002,
+      "learning_rate": 6.492163413630534e-05,
+      "loss": 0.4099,
+      "step": 1242
+    },
+    {
+      "epoch": 1.7884892086330937,
+      "grad_norm": 0.4085444891570115,
+      "learning_rate": 6.489018798382195e-05,
+      "loss": 0.4144,
+      "step": 1243
+    },
+    {
+      "epoch": 1.789928057553957,
+      "grad_norm": 0.45628227242989433,
+      "learning_rate": 6.485871670830243e-05,
+      "loss": 0.4097,
+      "step": 1244
+    },
+    {
+      "epoch": 1.79136690647482,
+      "grad_norm": 0.43906513069837294,
+      "learning_rate": 6.482722034151247e-05,
+      "loss": 0.4063,
+      "step": 1245
+    },
+    {
+      "epoch": 1.7928057553956833,
+      "grad_norm": 0.3609264245556682,
+      "learning_rate": 6.479569891524307e-05,
+      "loss": 0.4124,
+      "step": 1246
+    },
+    {
+      "epoch": 1.7942446043165468,
+      "grad_norm": 0.3209187946231822,
+      "learning_rate": 6.476415246131056e-05,
+      "loss": 0.4152,
+      "step": 1247
+    },
+    {
+      "epoch": 1.7956834532374102,
+      "grad_norm": 0.3604096044404803,
+      "learning_rate": 6.47325810115565e-05,
+      "loss": 0.4058,
+      "step": 1248
+    },
+    {
+      "epoch": 1.7971223021582734,
+      "grad_norm": 0.4160391262347793,
+      "learning_rate": 6.470098459784768e-05,
+      "loss": 0.4142,
+      "step": 1249
+    },
+    {
+      "epoch": 1.7985611510791366,
+      "grad_norm": 0.33336614192768277,
+      "learning_rate": 6.466936325207612e-05,
+      "loss": 0.4195,
+      "step": 1250
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.2750037566170555,
+      "learning_rate": 6.463771700615898e-05,
+      "loss": 0.4019,
+      "step": 1251
+    },
+    {
+      "epoch": 1.8014388489208633,
+      "grad_norm": 0.28172172333368006,
+      "learning_rate": 6.460604589203854e-05,
+      "loss": 0.41,
+      "step": 1252
+    },
+    {
+      "epoch": 1.8028776978417267,
+      "grad_norm": 0.29998294888495797,
+      "learning_rate": 6.457434994168224e-05,
+      "loss": 0.4124,
+      "step": 1253
+    },
+    {
+      "epoch": 1.80431654676259,
+      "grad_norm": 0.3017781858392867,
+      "learning_rate": 6.454262918708247e-05,
+      "loss": 0.4159,
+      "step": 1254
+    },
+    {
+      "epoch": 1.8057553956834531,
+      "grad_norm": 0.39356494672954967,
+      "learning_rate": 6.451088366025682e-05,
+      "loss": 0.4126,
+      "step": 1255
+    },
+    {
+      "epoch": 1.8071942446043165,
+      "grad_norm": 0.40728281276855766,
+      "learning_rate": 6.447911339324773e-05,
+      "loss": 0.4142,
+      "step": 1256
+    },
+    {
+      "epoch": 1.80863309352518,
+      "grad_norm": 0.35974633943904694,
+      "learning_rate": 6.444731841812274e-05,
+      "loss": 0.4118,
+      "step": 1257
+    },
+    {
+      "epoch": 1.8100719424460432,
+      "grad_norm": 0.35993558626760885,
+      "learning_rate": 6.44154987669742e-05,
+      "loss": 0.401,
+      "step": 1258
+    },
+    {
+      "epoch": 1.8115107913669064,
+      "grad_norm": 0.36454535373259755,
+      "learning_rate": 6.438365447191947e-05,
+      "loss": 0.412,
+      "step": 1259
+    },
+    {
+      "epoch": 1.8129496402877698,
+      "grad_norm": 0.4577783792211193,
+      "learning_rate": 6.435178556510076e-05,
+      "loss": 0.4172,
+      "step": 1260
+    },
+    {
+      "epoch": 1.814388489208633,
+      "grad_norm": 0.4624565308209469,
+      "learning_rate": 6.431989207868508e-05,
+      "loss": 0.4134,
+      "step": 1261
+    },
+    {
+      "epoch": 1.8158273381294965,
+      "grad_norm": 0.5594542203277918,
+      "learning_rate": 6.428797404486431e-05,
+      "loss": 0.4183,
+      "step": 1262
+    },
+    {
+      "epoch": 1.8172661870503597,
+      "grad_norm": 0.7594171209916067,
+      "learning_rate": 6.425603149585507e-05,
+      "loss": 0.4208,
+      "step": 1263
+    },
+    {
+      "epoch": 1.818705035971223,
+      "grad_norm": 0.8261431524291456,
+      "learning_rate": 6.422406446389872e-05,
+      "loss": 0.4088,
+      "step": 1264
+    },
+    {
+      "epoch": 1.8201438848920863,
+      "grad_norm": 0.8291601036807382,
+      "learning_rate": 6.419207298126135e-05,
+      "loss": 0.427,
+      "step": 1265
+    },
+    {
+      "epoch": 1.8215827338129498,
+      "grad_norm": 0.8404790471917353,
+      "learning_rate": 6.416005708023372e-05,
+      "loss": 0.4177,
+      "step": 1266
+    },
+    {
+      "epoch": 1.823021582733813,
+      "grad_norm": 0.6143417437150636,
+      "learning_rate": 6.412801679313125e-05,
+      "loss": 0.4157,
+      "step": 1267
+    },
+    {
+      "epoch": 1.8244604316546762,
+      "grad_norm": 0.34204692176509743,
+      "learning_rate": 6.409595215229397e-05,
+      "loss": 0.4191,
+      "step": 1268
+    },
+    {
+      "epoch": 1.8258992805755394,
+      "grad_norm": 0.40360132906785,
+      "learning_rate": 6.406386319008647e-05,
+      "loss": 0.4093,
+      "step": 1269
+    },
+    {
+      "epoch": 1.8273381294964028,
+      "grad_norm": 0.5509190671616278,
+      "learning_rate": 6.403174993889791e-05,
+      "loss": 0.4149,
+      "step": 1270
+    },
+    {
+      "epoch": 1.8287769784172663,
+      "grad_norm": 0.6010830771352358,
+      "learning_rate": 6.399961243114197e-05,
+      "loss": 0.4054,
+      "step": 1271
+    },
+    {
+      "epoch": 1.8302158273381295,
+      "grad_norm": 0.4913176518711791,
+      "learning_rate": 6.39674506992568e-05,
+      "loss": 0.4156,
+      "step": 1272
+    },
+    {
+      "epoch": 1.8316546762589927,
+      "grad_norm": 0.41656157572707764,
+      "learning_rate": 6.393526477570499e-05,
+      "loss": 0.4214,
+      "step": 1273
+    },
+    {
+      "epoch": 1.8330935251798561,
+      "grad_norm": 0.2891390865807047,
+      "learning_rate": 6.390305469297357e-05,
+      "loss": 0.4125,
+      "step": 1274
+    },
+    {
+      "epoch": 1.8345323741007196,
+      "grad_norm": 0.30863018534875003,
+      "learning_rate": 6.387082048357397e-05,
+      "loss": 0.415,
+      "step": 1275
+    },
+    {
+      "epoch": 1.8359712230215828,
+      "grad_norm": 0.37043296322044944,
+      "learning_rate": 6.383856218004193e-05,
+      "loss": 0.4004,
+      "step": 1276
+    },
+    {
+      "epoch": 1.837410071942446,
+      "grad_norm": 0.3284467353673348,
+      "learning_rate": 6.380627981493753e-05,
+      "loss": 0.4065,
+      "step": 1277
+    },
+    {
+      "epoch": 1.8388489208633092,
+      "grad_norm": 0.26346469726495136,
+      "learning_rate": 6.377397342084514e-05,
+      "loss": 0.4137,
+      "step": 1278
+    },
+    {
+      "epoch": 1.8402877697841726,
+      "grad_norm": 0.275682686605177,
+      "learning_rate": 6.37416430303734e-05,
+      "loss": 0.4062,
+      "step": 1279
+    },
+    {
+      "epoch": 1.841726618705036,
+      "grad_norm": 0.3188723729080881,
+      "learning_rate": 6.370928867615513e-05,
+      "loss": 0.4074,
+      "step": 1280
+    },
+    {
+      "epoch": 1.8431654676258993,
+      "grad_norm": 0.34449166633354666,
+      "learning_rate": 6.367691039084736e-05,
+      "loss": 0.4129,
+      "step": 1281
+    },
+    {
+      "epoch": 1.8446043165467625,
+      "grad_norm": 0.32534814011808155,
+      "learning_rate": 6.36445082071313e-05,
+      "loss": 0.4045,
+      "step": 1282
+    },
+    {
+      "epoch": 1.846043165467626,
+      "grad_norm": 0.3034647114568148,
+      "learning_rate": 6.361208215771222e-05,
+      "loss": 0.4122,
+      "step": 1283
+    },
+    {
+      "epoch": 1.8474820143884894,
+      "grad_norm": 0.3144141267128642,
+      "learning_rate": 6.357963227531954e-05,
+      "loss": 0.415,
+      "step": 1284
+    },
+    {
+      "epoch": 1.8489208633093526,
+      "grad_norm": 0.3560605725734302,
+      "learning_rate": 6.35471585927067e-05,
+      "loss": 0.4146,
+      "step": 1285
+    },
+    {
+      "epoch": 1.8503597122302158,
+      "grad_norm": 0.31395779215893044,
+      "learning_rate": 6.351466114265118e-05,
+      "loss": 0.4107,
+      "step": 1286
+    },
+    {
+      "epoch": 1.851798561151079,
+      "grad_norm": 0.2759705546226354,
+      "learning_rate": 6.348213995795445e-05,
+      "loss": 0.414,
+      "step": 1287
+    },
+    {
+      "epoch": 1.8532374100719424,
+      "grad_norm": 0.3899655847777587,
+      "learning_rate": 6.344959507144192e-05,
+      "loss": 0.4116,
+      "step": 1288
+    },
+    {
+      "epoch": 1.8546762589928059,
+      "grad_norm": 0.3814047970832645,
+      "learning_rate": 6.341702651596293e-05,
+      "loss": 0.4173,
+      "step": 1289
+    },
+    {
+      "epoch": 1.856115107913669,
+      "grad_norm": 0.37854424120396857,
+      "learning_rate": 6.338443432439074e-05,
+      "loss": 0.4188,
+      "step": 1290
+    },
+    {
+      "epoch": 1.8575539568345323,
+      "grad_norm": 0.39368141971937126,
+      "learning_rate": 6.335181852962242e-05,
+      "loss": 0.4162,
+      "step": 1291
+    },
+    {
+      "epoch": 1.8589928057553957,
+      "grad_norm": 0.8658673346387007,
+      "learning_rate": 6.331917916457889e-05,
+      "loss": 0.4206,
+      "step": 1292
+    },
+    {
+      "epoch": 1.8604316546762591,
+      "grad_norm": 0.49994881402323965,
+      "learning_rate": 6.328651626220485e-05,
+      "loss": 0.4059,
+      "step": 1293
+    },
+    {
+      "epoch": 1.8618705035971224,
+      "grad_norm": 0.5250618927656368,
+      "learning_rate": 6.325382985546879e-05,
+      "loss": 0.4082,
+      "step": 1294
+    },
+    {
+      "epoch": 1.8633093525179856,
+      "grad_norm": 0.6803949679147394,
+      "learning_rate": 6.322111997736288e-05,
+      "loss": 0.4158,
+      "step": 1295
+    },
+    {
+      "epoch": 1.8647482014388488,
+      "grad_norm": 0.8362950445425676,
+      "learning_rate": 6.3188386660903e-05,
+      "loss": 0.419,
+      "step": 1296
+    },
+    {
+      "epoch": 1.8661870503597122,
+      "grad_norm": 0.823545918156688,
+      "learning_rate": 6.315562993912869e-05,
+      "loss": 0.4124,
+      "step": 1297
+    },
+    {
+      "epoch": 1.8676258992805757,
+      "grad_norm": 0.801474049829953,
+      "learning_rate": 6.31228498451031e-05,
+      "loss": 0.4289,
+      "step": 1298
+    },
+    {
+      "epoch": 1.8690647482014389,
+      "grad_norm": 0.6485961538539604,
+      "learning_rate": 6.309004641191299e-05,
+      "loss": 0.416,
+      "step": 1299
+    },
+    {
+      "epoch": 1.870503597122302,
+      "grad_norm": 0.4779049998944919,
+      "learning_rate": 6.305721967266869e-05,
+      "loss": 0.4081,
+      "step": 1300
+    },
+    {
+      "epoch": 1.8719424460431655,
+      "grad_norm": 0.40011164454277504,
+      "learning_rate": 6.302436966050401e-05,
+      "loss": 0.4164,
+      "step": 1301
+    },
+    {
+      "epoch": 1.873381294964029,
+      "grad_norm": 0.41896670092009536,
+      "learning_rate": 6.29914964085763e-05,
+      "loss": 0.4083,
+      "step": 1302
+    },
+    {
+      "epoch": 1.8748201438848922,
+      "grad_norm": 0.49824582360359393,
+      "learning_rate": 6.295859995006629e-05,
+      "loss": 0.4031,
+      "step": 1303
+    },
+    {
+      "epoch": 1.8762589928057554,
+      "grad_norm": 0.41962534284540975,
+      "learning_rate": 6.292568031817823e-05,
+      "loss": 0.417,
+      "step": 1304
+    },
+    {
+      "epoch": 1.8776978417266186,
+      "grad_norm": 0.34665690081232475,
+      "learning_rate": 6.28927375461397e-05,
+      "loss": 0.4083,
+      "step": 1305
+    },
+    {
+      "epoch": 1.879136690647482,
+      "grad_norm": 0.44765233076906147,
+      "learning_rate": 6.285977166720166e-05,
+      "loss": 0.4172,
+      "step": 1306
+    },
+    {
+      "epoch": 1.8805755395683454,
+      "grad_norm": 0.46476012579703596,
+      "learning_rate": 6.28267827146384e-05,
+      "loss": 0.4137,
+      "step": 1307
+    },
+    {
+      "epoch": 1.8820143884892087,
+      "grad_norm": 0.38037507101938134,
+      "learning_rate": 6.279377072174744e-05,
+      "loss": 0.4109,
+      "step": 1308
+    },
+    {
+      "epoch": 1.8834532374100719,
+      "grad_norm": 0.32169622312544566,
+      "learning_rate": 6.276073572184964e-05,
+      "loss": 0.4055,
+      "step": 1309
+    },
+    {
+      "epoch": 1.8848920863309353,
+      "grad_norm": 0.36816131402540186,
+      "learning_rate": 6.272767774828903e-05,
+      "loss": 0.4077,
+      "step": 1310
+    },
+    {
+      "epoch": 1.8863309352517985,
+      "grad_norm": 0.3700201083539994,
+      "learning_rate": 6.269459683443283e-05,
+      "loss": 0.4069,
+      "step": 1311
+    },
+    {
+      "epoch": 1.887769784172662,
+      "grad_norm": 0.35744457741260083,
+      "learning_rate": 6.266149301367146e-05,
+      "loss": 0.4161,
+      "step": 1312
+    },
+    {
+      "epoch": 1.8892086330935252,
+      "grad_norm": 0.3581449937273382,
+      "learning_rate": 6.262836631941839e-05,
+      "loss": 0.415,
+      "step": 1313
+    },
+    {
+      "epoch": 1.8906474820143884,
+      "grad_norm": 0.364350376748174,
+      "learning_rate": 6.259521678511023e-05,
+      "loss": 0.4088,
+      "step": 1314
+    },
+    {
+      "epoch": 1.8920863309352518,
+      "grad_norm": 0.3099762299417227,
+      "learning_rate": 6.256204444420663e-05,
+      "loss": 0.4036,
+      "step": 1315
+    },
+    {
+      "epoch": 1.8935251798561152,
+      "grad_norm": 0.34894540444737093,
+      "learning_rate": 6.252884933019028e-05,
+      "loss": 0.4089,
+      "step": 1316
+    },
+    {
+      "epoch": 1.8949640287769784,
+      "grad_norm": 0.44353539498483013,
+      "learning_rate": 6.249563147656679e-05,
+      "loss": 0.4061,
+      "step": 1317
+    },
+    {
+      "epoch": 1.8964028776978417,
+      "grad_norm": 0.4498233764462982,
+      "learning_rate": 6.24623909168648e-05,
+      "loss": 0.4165,
+      "step": 1318
+    },
+    {
+      "epoch": 1.8978417266187049,
+      "grad_norm": 0.41749822238022266,
+      "learning_rate": 6.242912768463581e-05,
+      "loss": 0.3984,
+      "step": 1319
+    },
+    {
+      "epoch": 1.8992805755395683,
+      "grad_norm": 0.34484560899709454,
+      "learning_rate": 6.239584181345426e-05,
+      "loss": 0.4125,
+      "step": 1320
+    },
+    {
+      "epoch": 1.9007194244604317,
+      "grad_norm": 0.427987432108927,
+      "learning_rate": 6.236253333691739e-05,
+      "loss": 0.4023,
+      "step": 1321
+    },
+    {
+      "epoch": 1.902158273381295,
+      "grad_norm": 0.4823641444906322,
+      "learning_rate": 6.23292022886453e-05,
+      "loss": 0.4042,
+      "step": 1322
+    },
+    {
+      "epoch": 1.9035971223021582,
+      "grad_norm": 0.3651976602578544,
+      "learning_rate": 6.229584870228083e-05,
+      "loss": 0.4121,
+      "step": 1323
+    },
+    {
+      "epoch": 1.9050359712230216,
+      "grad_norm": 0.2263944366922672,
+      "learning_rate": 6.226247261148958e-05,
+      "loss": 0.4203,
+      "step": 1324
+    },
+    {
+      "epoch": 1.906474820143885,
+      "grad_norm": 0.33145983822020075,
+      "learning_rate": 6.22290740499599e-05,
+      "loss": 0.4184,
+      "step": 1325
+    },
+    {
+      "epoch": 1.9079136690647482,
+      "grad_norm": 0.35964447853254644,
+      "learning_rate": 6.21956530514028e-05,
+      "loss": 0.4169,
+      "step": 1326
+    },
+    {
+      "epoch": 1.9093525179856115,
+      "grad_norm": 0.3066675728373096,
+      "learning_rate": 6.216220964955192e-05,
+      "loss": 0.4094,
+      "step": 1327
+    },
+    {
+      "epoch": 1.9107913669064747,
+      "grad_norm": 0.32985730307035266,
+      "learning_rate": 6.21287438781635e-05,
+      "loss": 0.4087,
+      "step": 1328
+    },
+    {
+      "epoch": 1.912230215827338,
+      "grad_norm": 0.31748231267418137,
+      "learning_rate": 6.209525577101642e-05,
+      "loss": 0.4121,
+      "step": 1329
+    },
+    {
+      "epoch": 1.9136690647482015,
+      "grad_norm": 0.333443294448134,
+      "learning_rate": 6.206174536191207e-05,
+      "loss": 0.4144,
+      "step": 1330
+    },
+    {
+      "epoch": 1.9151079136690647,
+      "grad_norm": 0.30500839535193813,
+      "learning_rate": 6.202821268467433e-05,
+      "loss": 0.3967,
+      "step": 1331
+    },
+    {
+      "epoch": 1.916546762589928,
+      "grad_norm": 0.3146013067290611,
+      "learning_rate": 6.199465777314958e-05,
+      "loss": 0.4176,
+      "step": 1332
+    },
+    {
+      "epoch": 1.9179856115107914,
+      "grad_norm": 0.2769375621970634,
+      "learning_rate": 6.196108066120663e-05,
+      "loss": 0.4074,
+      "step": 1333
+    },
+    {
+      "epoch": 1.9194244604316548,
+      "grad_norm": 0.27574004683235964,
+      "learning_rate": 6.192748138273674e-05,
+      "loss": 0.4035,
+      "step": 1334
+    },
+    {
+      "epoch": 1.920863309352518,
+      "grad_norm": 0.2912381975508216,
+      "learning_rate": 6.189385997165348e-05,
+      "loss": 0.4105,
+      "step": 1335
+    },
+    {
+      "epoch": 1.9223021582733812,
+      "grad_norm": 0.23670494436860218,
+      "learning_rate": 6.186021646189281e-05,
+      "loss": 0.4124,
+      "step": 1336
+    },
+    {
+      "epoch": 1.9237410071942445,
+      "grad_norm": 0.20344126310126784,
+      "learning_rate": 6.182655088741294e-05,
+      "loss": 0.4027,
+      "step": 1337
+    },
+    {
+      "epoch": 1.925179856115108,
+      "grad_norm": 0.23852188447780012,
+      "learning_rate": 6.179286328219442e-05,
+      "loss": 0.4148,
+      "step": 1338
+    },
+    {
+      "epoch": 1.9266187050359713,
+      "grad_norm": 0.2920784841505559,
+      "learning_rate": 6.175915368024e-05,
+      "loss": 0.4069,
+      "step": 1339
+    },
+    {
+      "epoch": 1.9280575539568345,
+      "grad_norm": 0.31369463060257147,
+      "learning_rate": 6.172542211557463e-05,
+      "loss": 0.4124,
+      "step": 1340
+    },
+    {
+      "epoch": 1.9294964028776977,
+      "grad_norm": 0.37988212667817745,
+      "learning_rate": 6.169166862224542e-05,
+      "loss": 0.4189,
+      "step": 1341
+    },
+    {
+      "epoch": 1.9309352517985612,
+      "grad_norm": 0.40533514154650313,
+      "learning_rate": 6.165789323432166e-05,
+      "loss": 0.4003,
+      "step": 1342
+    },
+    {
+      "epoch": 1.9323741007194246,
+      "grad_norm": 0.38002459364623914,
+      "learning_rate": 6.162409598589467e-05,
+      "loss": 0.4098,
+      "step": 1343
+    },
+    {
+      "epoch": 1.9338129496402878,
+      "grad_norm": 0.5121102181780426,
+      "learning_rate": 6.159027691107791e-05,
+      "loss": 0.4208,
+      "step": 1344
+    },
+    {
+      "epoch": 1.935251798561151,
+      "grad_norm": 0.6506184503662952,
+      "learning_rate": 6.15564360440068e-05,
+      "loss": 0.4082,
+      "step": 1345
+    },
+    {
+      "epoch": 1.9366906474820142,
+      "grad_norm": 0.5736472588450692,
+      "learning_rate": 6.15225734188388e-05,
+      "loss": 0.4098,
+      "step": 1346
+    },
+    {
+      "epoch": 1.9381294964028777,
+      "grad_norm": 0.35704831088633465,
+      "learning_rate": 6.148868906975334e-05,
+      "loss": 0.4118,
+      "step": 1347
+    },
+    {
+      "epoch": 1.9395683453237411,
+      "grad_norm": 0.33181184996259766,
+      "learning_rate": 6.145478303095174e-05,
+      "loss": 0.4067,
+      "step": 1348
+    },
+    {
+      "epoch": 1.9410071942446043,
+      "grad_norm": 0.4069463361425132,
+      "learning_rate": 6.142085533665722e-05,
+      "loss": 0.4072,
+      "step": 1349
+    },
+    {
+      "epoch": 1.9424460431654675,
+      "grad_norm": 0.37388758221848784,
+      "learning_rate": 6.138690602111487e-05,
+      "loss": 0.4173,
+      "step": 1350
+    },
+    {
+      "epoch": 1.943884892086331,
+      "grad_norm": 0.3542153370672022,
+      "learning_rate": 6.135293511859164e-05,
+      "loss": 0.4215,
+      "step": 1351
+    },
+    {
+      "epoch": 1.9453237410071944,
+      "grad_norm": 0.3454628285076372,
+      "learning_rate": 6.131894266337618e-05,
+      "loss": 0.4132,
+      "step": 1352
+    },
+    {
+      "epoch": 1.9467625899280576,
+      "grad_norm": 0.2864504777225727,
+      "learning_rate": 6.128492868977897e-05,
+      "loss": 0.4063,
+      "step": 1353
+    },
+    {
+      "epoch": 1.9482014388489208,
+      "grad_norm": 0.25629228963906786,
+      "learning_rate": 6.12508932321322e-05,
+      "loss": 0.4064,
+      "step": 1354
+    },
+    {
+      "epoch": 1.949640287769784,
+      "grad_norm": 0.3641256393384321,
+      "learning_rate": 6.12168363247897e-05,
+      "loss": 0.4021,
+      "step": 1355
+    },
+    {
+      "epoch": 1.9510791366906475,
+      "grad_norm": 0.395733173150441,
+      "learning_rate": 6.1182758002127e-05,
+      "loss": 0.4067,
+      "step": 1356
+    },
+    {
+      "epoch": 1.952517985611511,
+      "grad_norm": 0.3740525490367782,
+      "learning_rate": 6.114865829854123e-05,
+      "loss": 0.4062,
+      "step": 1357
+    },
+    {
+      "epoch": 1.9539568345323741,
+      "grad_norm": 0.3948848350037882,
+      "learning_rate": 6.111453724845106e-05,
+      "loss": 0.4035,
+      "step": 1358
+    },
+    {
+      "epoch": 1.9553956834532373,
+      "grad_norm": 0.3025584688762319,
+      "learning_rate": 6.108039488629679e-05,
+      "loss": 0.4088,
+      "step": 1359
+    },
+    {
+      "epoch": 1.9568345323741008,
+      "grad_norm": 0.3029961956579855,
+      "learning_rate": 6.104623124654016e-05,
+      "loss": 0.4088,
+      "step": 1360
+    },
+    {
+      "epoch": 1.958273381294964,
+      "grad_norm": 0.39285369871901754,
+      "learning_rate": 6.101204636366441e-05,
+      "loss": 0.4041,
+      "step": 1361
+    },
+    {
+      "epoch": 1.9597122302158274,
+      "grad_norm": 0.2922226995225387,
+      "learning_rate": 6.0977840272174224e-05,
+      "loss": 0.3992,
+      "step": 1362
+    },
+    {
+      "epoch": 1.9611510791366906,
+      "grad_norm": 0.26267833177425903,
+      "learning_rate": 6.094361300659571e-05,
+      "loss": 0.4107,
+      "step": 1363
+    },
+    {
+      "epoch": 1.9625899280575538,
+      "grad_norm": 0.2617780818706576,
+      "learning_rate": 6.090936460147632e-05,
+      "loss": 0.4123,
+      "step": 1364
+    },
+    {
+      "epoch": 1.9640287769784173,
+      "grad_norm": 0.19966697740668615,
+      "learning_rate": 6.087509509138483e-05,
+      "loss": 0.3975,
+      "step": 1365
+    },
+    {
+      "epoch": 1.9654676258992807,
+      "grad_norm": 0.2453492732586692,
+      "learning_rate": 6.0840804510911374e-05,
+      "loss": 0.4101,
+      "step": 1366
+    },
+    {
+      "epoch": 1.966906474820144,
+      "grad_norm": 0.2776952871988876,
+      "learning_rate": 6.0806492894667315e-05,
+      "loss": 0.4095,
+      "step": 1367
+    },
+    {
+      "epoch": 1.9683453237410071,
+      "grad_norm": 0.2837398029906124,
+      "learning_rate": 6.077216027728524e-05,
+      "loss": 0.4064,
+      "step": 1368
+    },
+    {
+      "epoch": 1.9697841726618706,
+      "grad_norm": 0.2943359821815254,
+      "learning_rate": 6.073780669341896e-05,
+      "loss": 0.4111,
+      "step": 1369
+    },
+    {
+      "epoch": 1.9712230215827338,
+      "grad_norm": 0.3090899888025719,
+      "learning_rate": 6.070343217774343e-05,
+      "loss": 0.4088,
+      "step": 1370
+    },
+    {
+      "epoch": 1.9726618705035972,
+      "grad_norm": 0.3297382622058121,
+      "learning_rate": 6.066903676495477e-05,
+      "loss": 0.4038,
+      "step": 1371
+    },
+    {
+      "epoch": 1.9741007194244604,
+      "grad_norm": 0.32365191273040594,
+      "learning_rate": 6.063462048977011e-05,
+      "loss": 0.4175,
+      "step": 1372
+    },
+    {
+      "epoch": 1.9755395683453236,
+      "grad_norm": 0.36847282941476367,
+      "learning_rate": 6.060018338692774e-05,
+      "loss": 0.4206,
+      "step": 1373
+    },
+    {
+      "epoch": 1.976978417266187,
+      "grad_norm": 0.38276412372212654,
+      "learning_rate": 6.056572549118688e-05,
+      "loss": 0.4063,
+      "step": 1374
+    },
+    {
+      "epoch": 1.9784172661870505,
+      "grad_norm": 0.36046848840628315,
+      "learning_rate": 6.053124683732781e-05,
+      "loss": 0.4059,
+      "step": 1375
+    },
+    {
+      "epoch": 1.9798561151079137,
+      "grad_norm": 0.3460277018852855,
+      "learning_rate": 6.049674746015172e-05,
+      "loss": 0.4052,
+      "step": 1376
+    },
+    {
+      "epoch": 1.981294964028777,
+      "grad_norm": 0.4554861210413628,
+      "learning_rate": 6.046222739448075e-05,
+      "loss": 0.4091,
+      "step": 1377
+    },
+    {
+      "epoch": 1.9827338129496401,
+      "grad_norm": 0.6046812360447316,
+      "learning_rate": 6.042768667515786e-05,
+      "loss": 0.4126,
+      "step": 1378
+    },
+    {
+      "epoch": 1.9841726618705036,
+      "grad_norm": 0.6805215423686101,
+      "learning_rate": 6.039312533704692e-05,
+      "loss": 0.4138,
+      "step": 1379
+    },
+    {
+      "epoch": 1.985611510791367,
+      "grad_norm": 0.6521568279627475,
+      "learning_rate": 6.0358543415032625e-05,
+      "loss": 0.4138,
+      "step": 1380
+    },
+    {
+      "epoch": 1.9870503597122302,
+      "grad_norm": 0.590040105509639,
+      "learning_rate": 6.032394094402035e-05,
+      "loss": 0.4109,
+      "step": 1381
+    },
+    {
+      "epoch": 1.9884892086330934,
+      "grad_norm": 0.5507782563828758,
+      "learning_rate": 6.0289317958936305e-05,
+      "loss": 0.4056,
+      "step": 1382
+    },
+    {
+      "epoch": 1.9899280575539569,
+      "grad_norm": 0.46989797761590174,
+      "learning_rate": 6.0254674494727374e-05,
+      "loss": 0.4054,
+      "step": 1383
+    },
+    {
+      "epoch": 1.9913669064748203,
+      "grad_norm": 0.3689791897607179,
+      "learning_rate": 6.022001058636111e-05,
+      "loss": 0.4014,
+      "step": 1384
+    },
+    {
+      "epoch": 1.9928057553956835,
+      "grad_norm": 0.23556075043882688,
+      "learning_rate": 6.01853262688257e-05,
+      "loss": 0.4125,
+      "step": 1385
+    },
+    {
+      "epoch": 1.9942446043165467,
+      "grad_norm": 0.27317519191530143,
+      "learning_rate": 6.0150621577129934e-05,
+      "loss": 0.4248,
+      "step": 1386
+    },
+    {
+      "epoch": 1.99568345323741,
+      "grad_norm": 0.32185748101615674,
+      "learning_rate": 6.011589654630318e-05,
+      "loss": 0.3986,
+      "step": 1387
+    },
+    {
+      "epoch": 1.9971223021582734,
+      "grad_norm": 0.329612866758165,
+      "learning_rate": 6.008115121139528e-05,
+      "loss": 0.4133,
+      "step": 1388
+    },
+    {
+      "epoch": 1.9985611510791368,
+      "grad_norm": 0.32417942052255694,
+      "learning_rate": 6.0046385607476655e-05,
+      "loss": 0.4109,
+      "step": 1389
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 0.29654712665293514,
+      "learning_rate": 6.001159976963814e-05,
+      "loss": 0.4193,
+      "step": 1390
+    },
+    {
+      "epoch": 2.001438848920863,
+      "grad_norm": 0.3459667004954266,
+      "learning_rate": 5.9976793732990965e-05,
+      "loss": 0.3824,
+      "step": 1391
+    },
+    {
+      "epoch": 2.0028776978417264,
+      "grad_norm": 0.37518647892071594,
+      "learning_rate": 5.9941967532666806e-05,
+      "loss": 0.3865,
+      "step": 1392
+    },
+    {
+      "epoch": 2.00431654676259,
+      "grad_norm": 0.3279409839377487,
+      "learning_rate": 5.990712120381766e-05,
+      "loss": 0.3785,
+      "step": 1393
+    },
+    {
+      "epoch": 2.0057553956834533,
+      "grad_norm": 0.37998707272526827,
+      "learning_rate": 5.987225478161583e-05,
+      "loss": 0.3886,
+      "step": 1394
+    },
+    {
+      "epoch": 2.0071942446043165,
+      "grad_norm": 0.33163850726507343,
+      "learning_rate": 5.9837368301253905e-05,
+      "loss": 0.3825,
+      "step": 1395
+    },
+    {
+      "epoch": 2.0086330935251797,
+      "grad_norm": 0.3461474484939308,
+      "learning_rate": 5.980246179794476e-05,
+      "loss": 0.384,
+      "step": 1396
+    },
+    {
+      "epoch": 2.0100719424460434,
+      "grad_norm": 0.3302823572490198,
+      "learning_rate": 5.976753530692144e-05,
+      "loss": 0.3927,
+      "step": 1397
+    },
+    {
+      "epoch": 2.0115107913669066,
+      "grad_norm": 0.5977889055298922,
+      "learning_rate": 5.9732588863437155e-05,
+      "loss": 0.3819,
+      "step": 1398
+    },
+    {
+      "epoch": 2.01294964028777,
+      "grad_norm": 0.3635705965151427,
+      "learning_rate": 5.96976225027653e-05,
+      "loss": 0.3849,
+      "step": 1399
+    },
+    {
+      "epoch": 2.014388489208633,
+      "grad_norm": 0.395481424837781,
+      "learning_rate": 5.966263626019932e-05,
+      "loss": 0.3861,
+      "step": 1400
+    },
+    {
+      "epoch": 2.015827338129496,
+      "grad_norm": 0.42318411480225626,
+      "learning_rate": 5.9627630171052774e-05,
+      "loss": 0.3916,
+      "step": 1401
+    },
+    {
+      "epoch": 2.01726618705036,
+      "grad_norm": 0.36316339747330917,
+      "learning_rate": 5.9592604270659234e-05,
+      "loss": 0.3827,
+      "step": 1402
+    },
+    {
+      "epoch": 2.018705035971223,
+      "grad_norm": 0.37143452419933953,
+      "learning_rate": 5.955755859437225e-05,
+      "loss": 0.3793,
+      "step": 1403
+    },
+    {
+      "epoch": 2.0201438848920863,
+      "grad_norm": 0.43755692064098567,
+      "learning_rate": 5.9522493177565366e-05,
+      "loss": 0.3887,
+      "step": 1404
+    },
+    {
+      "epoch": 2.0215827338129495,
+      "grad_norm": 0.35195856639014117,
+      "learning_rate": 5.948740805563203e-05,
+      "loss": 0.3867,
+      "step": 1405
+    },
+    {
+      "epoch": 2.023021582733813,
+      "grad_norm": 0.2924243151297277,
+      "learning_rate": 5.94523032639856e-05,
+      "loss": 0.3853,
+      "step": 1406
+    },
+    {
+      "epoch": 2.0244604316546764,
+      "grad_norm": 0.27236408606934676,
+      "learning_rate": 5.9417178838059254e-05,
+      "loss": 0.3844,
+      "step": 1407
+    },
+    {
+      "epoch": 2.0258992805755396,
+      "grad_norm": 0.2808506053100754,
+      "learning_rate": 5.9382034813306014e-05,
+      "loss": 0.3933,
+      "step": 1408
+    },
+    {
+      "epoch": 2.027338129496403,
+      "grad_norm": 0.3249046985300833,
+      "learning_rate": 5.934687122519868e-05,
+      "loss": 0.3879,
+      "step": 1409
+    },
+    {
+      "epoch": 2.028776978417266,
+      "grad_norm": 0.30255528141566884,
+      "learning_rate": 5.93116881092298e-05,
+      "loss": 0.3933,
+      "step": 1410
+    },
+    {
+      "epoch": 2.0302158273381297,
+      "grad_norm": 0.25697615522420125,
+      "learning_rate": 5.927648550091162e-05,
+      "loss": 0.385,
+      "step": 1411
+    },
+    {
+      "epoch": 2.031654676258993,
+      "grad_norm": 0.2933321125736272,
+      "learning_rate": 5.9241263435776087e-05,
+      "loss": 0.3742,
+      "step": 1412
+    },
+    {
+      "epoch": 2.033093525179856,
+      "grad_norm": 0.2871683640914002,
+      "learning_rate": 5.920602194937474e-05,
+      "loss": 0.382,
+      "step": 1413
+    },
+    {
+      "epoch": 2.0345323741007193,
+      "grad_norm": 0.274468606150787,
+      "learning_rate": 5.9170761077278766e-05,
+      "loss": 0.3927,
+      "step": 1414
+    },
+    {
+      "epoch": 2.0359712230215825,
+      "grad_norm": 0.22310469914268924,
+      "learning_rate": 5.9135480855078915e-05,
+      "loss": 0.3952,
+      "step": 1415
+    },
+    {
+      "epoch": 2.037410071942446,
+      "grad_norm": 0.27311522093486645,
+      "learning_rate": 5.910018131838544e-05,
+      "loss": 0.3837,
+      "step": 1416
+    },
+    {
+      "epoch": 2.0388489208633094,
+      "grad_norm": 0.2930255558309253,
+      "learning_rate": 5.906486250282811e-05,
+      "loss": 0.3927,
+      "step": 1417
+    },
+    {
+      "epoch": 2.0402877697841726,
+      "grad_norm": 0.2949923450998013,
+      "learning_rate": 5.902952444405615e-05,
+      "loss": 0.3865,
+      "step": 1418
+    },
+    {
+      "epoch": 2.041726618705036,
+      "grad_norm": 0.367148447428638,
+      "learning_rate": 5.899416717773822e-05,
+      "loss": 0.3823,
+      "step": 1419
+    },
+    {
+      "epoch": 2.0431654676258995,
+      "grad_norm": 0.40861101175486775,
+      "learning_rate": 5.8958790739562316e-05,
+      "loss": 0.3777,
+      "step": 1420
+    },
+    {
+      "epoch": 2.0446043165467627,
+      "grad_norm": 0.3336076746347345,
+      "learning_rate": 5.892339516523586e-05,
+      "loss": 0.3872,
+      "step": 1421
+    },
+    {
+      "epoch": 2.046043165467626,
+      "grad_norm": 0.31298154003525624,
+      "learning_rate": 5.8887980490485536e-05,
+      "loss": 0.3924,
+      "step": 1422
+    },
+    {
+      "epoch": 2.047482014388489,
+      "grad_norm": 0.24082912268403683,
+      "learning_rate": 5.8852546751057337e-05,
+      "loss": 0.3827,
+      "step": 1423
+    },
+    {
+      "epoch": 2.0489208633093523,
+      "grad_norm": 0.2438905924001371,
+      "learning_rate": 5.8817093982716455e-05,
+      "loss": 0.38,
+      "step": 1424
+    },
+    {
+      "epoch": 2.050359712230216,
+      "grad_norm": 0.2625708850715305,
+      "learning_rate": 5.878162222124735e-05,
+      "loss": 0.38,
+      "step": 1425
+    },
+    {
+      "epoch": 2.051798561151079,
+      "grad_norm": 0.24961055276235888,
+      "learning_rate": 5.8746131502453623e-05,
+      "loss": 0.3878,
+      "step": 1426
+    },
+    {
+      "epoch": 2.0532374100719424,
+      "grad_norm": 0.22308771158231927,
+      "learning_rate": 5.871062186215799e-05,
+      "loss": 0.3759,
+      "step": 1427
+    },
+    {
+      "epoch": 2.0546762589928056,
+      "grad_norm": 0.2840986003311144,
+      "learning_rate": 5.867509333620231e-05,
+      "loss": 0.3843,
+      "step": 1428
+    },
+    {
+      "epoch": 2.0561151079136692,
+      "grad_norm": 0.3127898412044105,
+      "learning_rate": 5.863954596044744e-05,
+      "loss": 0.3793,
+      "step": 1429
+    },
+    {
+      "epoch": 2.0575539568345325,
+      "grad_norm": 0.3313591176294113,
+      "learning_rate": 5.8603979770773344e-05,
+      "loss": 0.3825,
+      "step": 1430
+    },
+    {
+      "epoch": 2.0589928057553957,
+      "grad_norm": 0.3385757526260623,
+      "learning_rate": 5.85683948030789e-05,
+      "loss": 0.3995,
+      "step": 1431
+    },
+    {
+      "epoch": 2.060431654676259,
+      "grad_norm": 0.2887325144460221,
+      "learning_rate": 5.8532791093282e-05,
+      "loss": 0.3879,
+      "step": 1432
+    },
+    {
+      "epoch": 2.061870503597122,
+      "grad_norm": 0.2978522332390748,
+      "learning_rate": 5.849716867731941e-05,
+      "loss": 0.3917,
+      "step": 1433
+    },
+    {
+      "epoch": 2.0633093525179858,
+      "grad_norm": 0.2716509667278568,
+      "learning_rate": 5.84615275911468e-05,
+      "loss": 0.3819,
+      "step": 1434
+    },
+    {
+      "epoch": 2.064748201438849,
+      "grad_norm": 0.24739912138242007,
+      "learning_rate": 5.8425867870738684e-05,
+      "loss": 0.3844,
+      "step": 1435
+    },
+    {
+      "epoch": 2.066187050359712,
+      "grad_norm": 0.2742167550579385,
+      "learning_rate": 5.839018955208838e-05,
+      "loss": 0.3743,
+      "step": 1436
+    },
+    {
+      "epoch": 2.0676258992805754,
+      "grad_norm": 0.3253999143230146,
+      "learning_rate": 5.835449267120796e-05,
+      "loss": 0.3867,
+      "step": 1437
+    },
+    {
+      "epoch": 2.069064748201439,
+      "grad_norm": 0.2619192587544168,
+      "learning_rate": 5.831877726412827e-05,
+      "loss": 0.3811,
+      "step": 1438
+    },
+    {
+      "epoch": 2.0705035971223023,
+      "grad_norm": 0.23391498232949504,
+      "learning_rate": 5.828304336689883e-05,
+      "loss": 0.3841,
+      "step": 1439
+    },
+    {
+      "epoch": 2.0719424460431655,
+      "grad_norm": 0.27588999427877353,
+      "learning_rate": 5.824729101558781e-05,
+      "loss": 0.3887,
+      "step": 1440
+    },
+    {
+      "epoch": 2.0733812949640287,
+      "grad_norm": 0.20580107415618934,
+      "learning_rate": 5.821152024628207e-05,
+      "loss": 0.3952,
+      "step": 1441
+    },
+    {
+      "epoch": 2.074820143884892,
+      "grad_norm": 0.25656023537508604,
+      "learning_rate": 5.8175731095086974e-05,
+      "loss": 0.3859,
+      "step": 1442
+    },
+    {
+      "epoch": 2.0762589928057555,
+      "grad_norm": 0.3102205957999117,
+      "learning_rate": 5.813992359812649e-05,
+      "loss": 0.3808,
+      "step": 1443
+    },
+    {
+      "epoch": 2.0776978417266188,
+      "grad_norm": 0.3170056077742382,
+      "learning_rate": 5.8104097791543104e-05,
+      "loss": 0.3843,
+      "step": 1444
+    },
+    {
+      "epoch": 2.079136690647482,
+      "grad_norm": 0.264046971425198,
+      "learning_rate": 5.806825371149778e-05,
+      "loss": 0.3891,
+      "step": 1445
+    },
+    {
+      "epoch": 2.080575539568345,
+      "grad_norm": 0.27391192188464647,
+      "learning_rate": 5.803239139416989e-05,
+      "loss": 0.3876,
+      "step": 1446
+    },
+    {
+      "epoch": 2.082014388489209,
+      "grad_norm": 0.34098754315012436,
+      "learning_rate": 5.799651087575728e-05,
+      "loss": 0.3914,
+      "step": 1447
+    },
+    {
+      "epoch": 2.083453237410072,
+      "grad_norm": 0.35068952798880154,
+      "learning_rate": 5.7960612192476096e-05,
+      "loss": 0.3879,
+      "step": 1448
+    },
+    {
+      "epoch": 2.0848920863309353,
+      "grad_norm": 0.2949671254441889,
+      "learning_rate": 5.792469538056089e-05,
+      "loss": 0.3943,
+      "step": 1449
+    },
+    {
+      "epoch": 2.0863309352517985,
+      "grad_norm": 0.30766133986982047,
+      "learning_rate": 5.7888760476264445e-05,
+      "loss": 0.3803,
+      "step": 1450
+    },
+    {
+      "epoch": 2.0877697841726617,
+      "grad_norm": 1.0803950671766844,
+      "learning_rate": 5.785280751585785e-05,
+      "loss": 0.4023,
+      "step": 1451
+    },
+    {
+      "epoch": 2.0892086330935253,
+      "grad_norm": 0.39452949530176423,
+      "learning_rate": 5.7816836535630436e-05,
+      "loss": 0.3894,
+      "step": 1452
+    },
+    {
+      "epoch": 2.0906474820143885,
+      "grad_norm": 0.34118233214173016,
+      "learning_rate": 5.7780847571889625e-05,
+      "loss": 0.3901,
+      "step": 1453
+    },
+    {
+      "epoch": 2.0920863309352518,
+      "grad_norm": 0.3018886581079717,
+      "learning_rate": 5.7744840660961126e-05,
+      "loss": 0.39,
+      "step": 1454
+    },
+    {
+      "epoch": 2.093525179856115,
+      "grad_norm": 0.4524051858154562,
+      "learning_rate": 5.770881583918865e-05,
+      "loss": 0.3915,
+      "step": 1455
+    },
+    {
+      "epoch": 2.0949640287769786,
+      "grad_norm": 0.42544552361250226,
+      "learning_rate": 5.767277314293404e-05,
+      "loss": 0.39,
+      "step": 1456
+    },
+    {
+      "epoch": 2.096402877697842,
+      "grad_norm": 0.37257336494331905,
+      "learning_rate": 5.76367126085772e-05,
+      "loss": 0.3922,
+      "step": 1457
+    },
+    {
+      "epoch": 2.097841726618705,
+      "grad_norm": 0.4693117389719594,
+      "learning_rate": 5.760063427251599e-05,
+      "loss": 0.3943,
+      "step": 1458
+    },
+    {
+      "epoch": 2.0992805755395683,
+      "grad_norm": 0.45114686143866667,
+      "learning_rate": 5.756453817116624e-05,
+      "loss": 0.3939,
+      "step": 1459
+    },
+    {
+      "epoch": 2.1007194244604315,
+      "grad_norm": 0.4839684373567707,
+      "learning_rate": 5.752842434096176e-05,
+      "loss": 0.3882,
+      "step": 1460
+    },
+    {
+      "epoch": 2.102158273381295,
+      "grad_norm": 0.5213357084405652,
+      "learning_rate": 5.7492292818354224e-05,
+      "loss": 0.3889,
+      "step": 1461
+    },
+    {
+      "epoch": 2.1035971223021583,
+      "grad_norm": 0.4748272524025722,
+      "learning_rate": 5.745614363981316e-05,
+      "loss": 0.3935,
+      "step": 1462
+    },
+    {
+      "epoch": 2.1050359712230216,
+      "grad_norm": 0.36349426117437394,
+      "learning_rate": 5.741997684182591e-05,
+      "loss": 0.3856,
+      "step": 1463
+    },
+    {
+      "epoch": 2.1064748201438848,
+      "grad_norm": 0.30830762113025034,
+      "learning_rate": 5.7383792460897626e-05,
+      "loss": 0.3944,
+      "step": 1464
+    },
+    {
+      "epoch": 2.1079136690647484,
+      "grad_norm": 0.3019268807772761,
+      "learning_rate": 5.73475905335512e-05,
+      "loss": 0.3865,
+      "step": 1465
+    },
+    {
+      "epoch": 2.1093525179856116,
+      "grad_norm": 0.3514724192590477,
+      "learning_rate": 5.731137109632722e-05,
+      "loss": 0.3863,
+      "step": 1466
+    },
+    {
+      "epoch": 2.110791366906475,
+      "grad_norm": 0.3490486941049694,
+      "learning_rate": 5.727513418578397e-05,
+      "loss": 0.3872,
+      "step": 1467
+    },
+    {
+      "epoch": 2.112230215827338,
+      "grad_norm": 0.3145147820518404,
+      "learning_rate": 5.723887983849732e-05,
+      "loss": 0.3891,
+      "step": 1468
+    },
+    {
+      "epoch": 2.1136690647482013,
+      "grad_norm": 0.2514407045031779,
+      "learning_rate": 5.720260809106083e-05,
+      "loss": 0.3917,
+      "step": 1469
+    },
+    {
+      "epoch": 2.115107913669065,
+      "grad_norm": 0.28621965565971635,
+      "learning_rate": 5.716631898008553e-05,
+      "loss": 0.385,
+      "step": 1470
+    },
+    {
+      "epoch": 2.116546762589928,
+      "grad_norm": 0.29628756710605814,
+      "learning_rate": 5.713001254220002e-05,
+      "loss": 0.3846,
+      "step": 1471
+    },
+    {
+      "epoch": 2.1179856115107913,
+      "grad_norm": 0.2442164832622535,
+      "learning_rate": 5.7093688814050425e-05,
+      "loss": 0.3889,
+      "step": 1472
+    },
+    {
+      "epoch": 2.1194244604316546,
+      "grad_norm": 0.2412959429087779,
+      "learning_rate": 5.705734783230022e-05,
+      "loss": 0.3826,
+      "step": 1473
+    },
+    {
+      "epoch": 2.1208633093525178,
+      "grad_norm": 0.2810101099676454,
+      "learning_rate": 5.7020989633630414e-05,
+      "loss": 0.3905,
+      "step": 1474
+    },
+    {
+      "epoch": 2.1223021582733814,
+      "grad_norm": 0.25445474818311004,
+      "learning_rate": 5.6984614254739306e-05,
+      "loss": 0.3873,
+      "step": 1475
+    },
+    {
+      "epoch": 2.1237410071942446,
+      "grad_norm": 0.25557649490833445,
+      "learning_rate": 5.694822173234257e-05,
+      "loss": 0.3844,
+      "step": 1476
+    },
+    {
+      "epoch": 2.125179856115108,
+      "grad_norm": 0.2777684812774492,
+      "learning_rate": 5.691181210317319e-05,
+      "loss": 0.3857,
+      "step": 1477
+    },
+    {
+      "epoch": 2.126618705035971,
+      "grad_norm": 0.22345516855286213,
+      "learning_rate": 5.687538540398141e-05,
+      "loss": 0.3872,
+      "step": 1478
+    },
+    {
+      "epoch": 2.1280575539568347,
+      "grad_norm": 0.21060976344773377,
+      "learning_rate": 5.683894167153468e-05,
+      "loss": 0.3795,
+      "step": 1479
+    },
+    {
+      "epoch": 2.129496402877698,
+      "grad_norm": 0.2861691465626582,
+      "learning_rate": 5.680248094261769e-05,
+      "loss": 0.3922,
+      "step": 1480
+    },
+    {
+      "epoch": 2.130935251798561,
+      "grad_norm": 0.3107365211538534,
+      "learning_rate": 5.676600325403224e-05,
+      "loss": 0.3793,
+      "step": 1481
+    },
+    {
+      "epoch": 2.1323741007194243,
+      "grad_norm": 0.2748030294623659,
+      "learning_rate": 5.672950864259729e-05,
+      "loss": 0.3832,
+      "step": 1482
+    },
+    {
+      "epoch": 2.133812949640288,
+      "grad_norm": 0.27990029682683215,
+      "learning_rate": 5.669299714514884e-05,
+      "loss": 0.388,
+      "step": 1483
+    },
+    {
+      "epoch": 2.135251798561151,
+      "grad_norm": 0.22121964580171558,
+      "learning_rate": 5.665646879853995e-05,
+      "loss": 0.388,
+      "step": 1484
+    },
+    {
+      "epoch": 2.1366906474820144,
+      "grad_norm": 0.21626914568347483,
+      "learning_rate": 5.661992363964072e-05,
+      "loss": 0.3831,
+      "step": 1485
+    },
+    {
+      "epoch": 2.1381294964028776,
+      "grad_norm": 0.29919711192071885,
+      "learning_rate": 5.658336170533814e-05,
+      "loss": 0.3812,
+      "step": 1486
+    },
+    {
+      "epoch": 2.139568345323741,
+      "grad_norm": 0.3158627972498642,
+      "learning_rate": 5.654678303253624e-05,
+      "loss": 0.3876,
+      "step": 1487
+    },
+    {
+      "epoch": 2.1410071942446045,
+      "grad_norm": 0.28145261964361495,
+      "learning_rate": 5.6510187658155846e-05,
+      "loss": 0.3907,
+      "step": 1488
+    },
+    {
+      "epoch": 2.1424460431654677,
+      "grad_norm": 0.342743346803727,
+      "learning_rate": 5.6473575619134686e-05,
+      "loss": 0.3812,
+      "step": 1489
+    },
+    {
+      "epoch": 2.143884892086331,
+      "grad_norm": 0.374531993468505,
+      "learning_rate": 5.643694695242731e-05,
+      "loss": 0.3818,
+      "step": 1490
+    },
+    {
+      "epoch": 2.145323741007194,
+      "grad_norm": 0.27133829796780834,
+      "learning_rate": 5.640030169500508e-05,
+      "loss": 0.3836,
+      "step": 1491
+    },
+    {
+      "epoch": 2.1467625899280574,
+      "grad_norm": 0.2351914169210999,
+      "learning_rate": 5.636363988385601e-05,
+      "loss": 0.396,
+      "step": 1492
+    },
+    {
+      "epoch": 2.148201438848921,
+      "grad_norm": 0.24933413811162175,
+      "learning_rate": 5.632696155598493e-05,
+      "loss": 0.3902,
+      "step": 1493
+    },
+    {
+      "epoch": 2.149640287769784,
+      "grad_norm": 0.22425796164616213,
+      "learning_rate": 5.6290266748413266e-05,
+      "loss": 0.3867,
+      "step": 1494
+    },
+    {
+      "epoch": 2.1510791366906474,
+      "grad_norm": 0.2305010005410816,
+      "learning_rate": 5.6253555498179124e-05,
+      "loss": 0.382,
+      "step": 1495
+    },
+    {
+      "epoch": 2.1525179856115106,
+      "grad_norm": 0.23705255037178416,
+      "learning_rate": 5.621682784233718e-05,
+      "loss": 0.3878,
+      "step": 1496
+    },
+    {
+      "epoch": 2.1539568345323743,
+      "grad_norm": 0.1768755293978743,
+      "learning_rate": 5.618008381795868e-05,
+      "loss": 0.3903,
+      "step": 1497
+    },
+    {
+      "epoch": 2.1553956834532375,
+      "grad_norm": 0.20575095776656013,
+      "learning_rate": 5.61433234621314e-05,
+      "loss": 0.393,
+      "step": 1498
+    },
+    {
+      "epoch": 2.1568345323741007,
+      "grad_norm": 0.19699655048659204,
+      "learning_rate": 5.610654681195957e-05,
+      "loss": 0.3875,
+      "step": 1499
+    },
+    {
+      "epoch": 2.158273381294964,
+      "grad_norm": 0.19827610098763424,
+      "learning_rate": 5.606975390456391e-05,
+      "loss": 0.3909,
+      "step": 1500
+    },
+    {
+      "epoch": 2.159712230215827,
+      "grad_norm": 0.21305747803672276,
+      "learning_rate": 5.603294477708149e-05,
+      "loss": 0.3893,
+      "step": 1501
+    },
+    {
+      "epoch": 2.161151079136691,
+      "grad_norm": 0.2459941053050217,
+      "learning_rate": 5.599611946666581e-05,
+      "loss": 0.3942,
+      "step": 1502
+    },
+    {
+      "epoch": 2.162589928057554,
+      "grad_norm": 0.23071666194309015,
+      "learning_rate": 5.595927801048669e-05,
+      "loss": 0.3834,
+      "step": 1503
+    },
+    {
+      "epoch": 2.1640287769784172,
+      "grad_norm": 0.2305452784564125,
+      "learning_rate": 5.5922420445730245e-05,
+      "loss": 0.3885,
+      "step": 1504
+    },
+    {
+      "epoch": 2.1654676258992804,
+      "grad_norm": 0.3170557603384182,
+      "learning_rate": 5.5885546809598805e-05,
+      "loss": 0.3857,
+      "step": 1505
+    },
+    {
+      "epoch": 2.166906474820144,
+      "grad_norm": 0.3885810828081808,
+      "learning_rate": 5.584865713931098e-05,
+      "loss": 0.3892,
+      "step": 1506
+    },
+    {
+      "epoch": 2.1683453237410073,
+      "grad_norm": 0.3686007865701777,
+      "learning_rate": 5.5811751472101564e-05,
+      "loss": 0.3856,
+      "step": 1507
+    },
+    {
+      "epoch": 2.1697841726618705,
+      "grad_norm": 0.3356733847553759,
+      "learning_rate": 5.577482984522145e-05,
+      "loss": 0.3844,
+      "step": 1508
+    },
+    {
+      "epoch": 2.1712230215827337,
+      "grad_norm": 0.3917999468495738,
+      "learning_rate": 5.573789229593767e-05,
+      "loss": 0.3839,
+      "step": 1509
+    },
+    {
+      "epoch": 2.172661870503597,
+      "grad_norm": 0.47448224083035506,
+      "learning_rate": 5.570093886153334e-05,
+      "loss": 0.3857,
+      "step": 1510
+    },
+    {
+      "epoch": 2.1741007194244606,
+      "grad_norm": 0.5088636398067714,
+      "learning_rate": 5.5663969579307594e-05,
+      "loss": 0.3913,
+      "step": 1511
+    },
+    {
+      "epoch": 2.175539568345324,
+      "grad_norm": 0.5098767681725841,
+      "learning_rate": 5.562698448657553e-05,
+      "loss": 0.3743,
+      "step": 1512
+    },
+    {
+      "epoch": 2.176978417266187,
+      "grad_norm": 0.44226504165942554,
+      "learning_rate": 5.5589983620668286e-05,
+      "loss": 0.3868,
+      "step": 1513
+    },
+    {
+      "epoch": 2.1784172661870502,
+      "grad_norm": 0.3701485993906192,
+      "learning_rate": 5.555296701893284e-05,
+      "loss": 0.3891,
+      "step": 1514
+    },
+    {
+      "epoch": 2.1798561151079134,
+      "grad_norm": 0.32659810006018575,
+      "learning_rate": 5.551593471873208e-05,
+      "loss": 0.3865,
+      "step": 1515
+    },
+    {
+      "epoch": 2.181294964028777,
+      "grad_norm": 0.2739322729567986,
+      "learning_rate": 5.547888675744476e-05,
+      "loss": 0.3904,
+      "step": 1516
+    },
+    {
+      "epoch": 2.1827338129496403,
+      "grad_norm": 1.1514120414314388,
+      "learning_rate": 5.5441823172465427e-05,
+      "loss": 0.3816,
+      "step": 1517
+    },
+    {
+      "epoch": 2.1841726618705035,
+      "grad_norm": 0.5094085249815662,
+      "learning_rate": 5.540474400120438e-05,
+      "loss": 0.3961,
+      "step": 1518
+    },
+    {
+      "epoch": 2.1856115107913667,
+      "grad_norm": 0.3689605263551239,
+      "learning_rate": 5.536764928108769e-05,
+      "loss": 0.3925,
+      "step": 1519
+    },
+    {
+      "epoch": 2.1870503597122304,
+      "grad_norm": 0.6710103294342745,
+      "learning_rate": 5.533053904955709e-05,
+      "loss": 0.4006,
+      "step": 1520
+    },
+    {
+      "epoch": 2.1884892086330936,
+      "grad_norm": 0.589844121232794,
+      "learning_rate": 5.5293413344069964e-05,
+      "loss": 0.4039,
+      "step": 1521
+    },
+    {
+      "epoch": 2.189928057553957,
+      "grad_norm": 0.4465240610476147,
+      "learning_rate": 5.525627220209934e-05,
+      "loss": 0.4048,
+      "step": 1522
+    },
+    {
+      "epoch": 2.19136690647482,
+      "grad_norm": 0.5119028970967803,
+      "learning_rate": 5.5219115661133815e-05,
+      "loss": 0.3916,
+      "step": 1523
+    },
+    {
+      "epoch": 2.1928057553956837,
+      "grad_norm": 0.5617122367035737,
+      "learning_rate": 5.518194375867754e-05,
+      "loss": 0.3961,
+      "step": 1524
+    },
+    {
+      "epoch": 2.194244604316547,
+      "grad_norm": 0.4690244242416867,
+      "learning_rate": 5.514475653225014e-05,
+      "loss": 0.3922,
+      "step": 1525
+    },
+    {
+      "epoch": 2.19568345323741,
+      "grad_norm": 0.4015231614518023,
+      "learning_rate": 5.510755401938676e-05,
+      "loss": 0.392,
+      "step": 1526
+    },
+    {
+      "epoch": 2.1971223021582733,
+      "grad_norm": 0.5097283725286479,
+      "learning_rate": 5.5070336257637904e-05,
+      "loss": 0.3964,
+      "step": 1527
+    },
+    {
+      "epoch": 2.1985611510791365,
+      "grad_norm": 0.435515690531513,
+      "learning_rate": 5.503310328456953e-05,
+      "loss": 0.3974,
+      "step": 1528
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 0.3806838811922357,
+      "learning_rate": 5.4995855137762926e-05,
+      "loss": 0.3954,
+      "step": 1529
+    },
+    {
+      "epoch": 2.2014388489208634,
+      "grad_norm": 0.39291846470544906,
+      "learning_rate": 5.4958591854814695e-05,
+      "loss": 0.3892,
+      "step": 1530
+    },
+    {
+      "epoch": 2.2028776978417266,
+      "grad_norm": 0.397331489863752,
+      "learning_rate": 5.492131347333671e-05,
+      "loss": 0.3892,
+      "step": 1531
+    },
+    {
+      "epoch": 2.20431654676259,
+      "grad_norm": 0.3429131255958608,
+      "learning_rate": 5.48840200309561e-05,
+      "loss": 0.4013,
+      "step": 1532
+    },
+    {
+      "epoch": 2.205755395683453,
+      "grad_norm": 0.34498430285961723,
+      "learning_rate": 5.484671156531519e-05,
+      "loss": 0.3905,
+      "step": 1533
+    },
+    {
+      "epoch": 2.2071942446043167,
+      "grad_norm": 0.3627701636779315,
+      "learning_rate": 5.480938811407146e-05,
+      "loss": 0.3866,
+      "step": 1534
+    },
+    {
+      "epoch": 2.20863309352518,
+      "grad_norm": 0.32607719730695583,
+      "learning_rate": 5.477204971489753e-05,
+      "loss": 0.3926,
+      "step": 1535
+    },
+    {
+      "epoch": 2.210071942446043,
+      "grad_norm": 0.32683828106525564,
+      "learning_rate": 5.473469640548109e-05,
+      "loss": 0.4002,
+      "step": 1536
+    },
+    {
+      "epoch": 2.2115107913669063,
+      "grad_norm": 0.3622966949282371,
+      "learning_rate": 5.469732822352491e-05,
+      "loss": 0.3828,
+      "step": 1537
+    },
+    {
+      "epoch": 2.21294964028777,
+      "grad_norm": 0.2366794753501399,
+      "learning_rate": 5.465994520674672e-05,
+      "loss": 0.3907,
+      "step": 1538
+    },
+    {
+      "epoch": 2.214388489208633,
+      "grad_norm": 0.28809503041902335,
+      "learning_rate": 5.4622547392879295e-05,
+      "loss": 0.3873,
+      "step": 1539
+    },
+    {
+      "epoch": 2.2158273381294964,
+      "grad_norm": 0.3237451617454751,
+      "learning_rate": 5.458513481967027e-05,
+      "loss": 0.3916,
+      "step": 1540
+    },
+    {
+      "epoch": 2.2172661870503596,
+      "grad_norm": 0.2730893600150766,
+      "learning_rate": 5.454770752488223e-05,
+      "loss": 0.3941,
+      "step": 1541
+    },
+    {
+      "epoch": 2.218705035971223,
+      "grad_norm": 0.25974636320336986,
+      "learning_rate": 5.4510265546292615e-05,
+      "loss": 0.3931,
+      "step": 1542
+    },
+    {
+      "epoch": 2.2201438848920865,
+      "grad_norm": 0.34671179580477074,
+      "learning_rate": 5.4472808921693657e-05,
+      "loss": 0.3913,
+      "step": 1543
+    },
+    {
+      "epoch": 2.2215827338129497,
+      "grad_norm": 0.3535548881559468,
+      "learning_rate": 5.4435337688892396e-05,
+      "loss": 0.3968,
+      "step": 1544
+    },
+    {
+      "epoch": 2.223021582733813,
+      "grad_norm": 0.2636642875839242,
+      "learning_rate": 5.4397851885710595e-05,
+      "loss": 0.3848,
+      "step": 1545
+    },
+    {
+      "epoch": 2.224460431654676,
+      "grad_norm": 0.3133545904069258,
+      "learning_rate": 5.4360351549984755e-05,
+      "loss": 0.3912,
+      "step": 1546
+    },
+    {
+      "epoch": 2.2258992805755398,
+      "grad_norm": 0.3808123551337556,
+      "learning_rate": 5.432283671956601e-05,
+      "loss": 0.3888,
+      "step": 1547
+    },
+    {
+      "epoch": 2.227338129496403,
+      "grad_norm": 0.35583536953867234,
+      "learning_rate": 5.428530743232016e-05,
+      "loss": 0.3838,
+      "step": 1548
+    },
+    {
+      "epoch": 2.228776978417266,
+      "grad_norm": 0.219318701997273,
+      "learning_rate": 5.4247763726127564e-05,
+      "loss": 0.3751,
+      "step": 1549
+    },
+    {
+      "epoch": 2.2302158273381294,
+      "grad_norm": 0.22318535359522693,
+      "learning_rate": 5.421020563888317e-05,
+      "loss": 0.3909,
+      "step": 1550
+    },
+    {
+      "epoch": 2.2316546762589926,
+      "grad_norm": 0.24960667305651438,
+      "learning_rate": 5.417263320849641e-05,
+      "loss": 0.3893,
+      "step": 1551
+    },
+    {
+      "epoch": 2.2330935251798563,
+      "grad_norm": 0.2615472186059933,
+      "learning_rate": 5.4135046472891205e-05,
+      "loss": 0.3965,
+      "step": 1552
+    },
+    {
+      "epoch": 2.2345323741007195,
+      "grad_norm": 0.20597246340467404,
+      "learning_rate": 5.409744547000591e-05,
+      "loss": 0.3926,
+      "step": 1553
+    },
+    {
+      "epoch": 2.2359712230215827,
+      "grad_norm": 0.23479074496220995,
+      "learning_rate": 5.405983023779328e-05,
+      "loss": 0.3975,
+      "step": 1554
+    },
+    {
+      "epoch": 2.237410071942446,
+      "grad_norm": 0.30908876136663427,
+      "learning_rate": 5.402220081422048e-05,
+      "loss": 0.3794,
+      "step": 1555
+    },
+    {
+      "epoch": 2.2388489208633096,
+      "grad_norm": 0.23958569016002487,
+      "learning_rate": 5.3984557237268905e-05,
+      "loss": 0.397,
+      "step": 1556
+    },
+    {
+      "epoch": 2.2402877697841728,
+      "grad_norm": 0.231453892673959,
+      "learning_rate": 5.394689954493432e-05,
+      "loss": 0.3874,
+      "step": 1557
+    },
+    {
+      "epoch": 2.241726618705036,
+      "grad_norm": 0.2836005114568376,
+      "learning_rate": 5.390922777522669e-05,
+      "loss": 0.3897,
+      "step": 1558
+    },
+    {
+      "epoch": 2.243165467625899,
+      "grad_norm": 0.3062651354015222,
+      "learning_rate": 5.3871541966170225e-05,
+      "loss": 0.3971,
+      "step": 1559
+    },
+    {
+      "epoch": 2.2446043165467624,
+      "grad_norm": 0.2636891794944903,
+      "learning_rate": 5.383384215580326e-05,
+      "loss": 0.3882,
+      "step": 1560
+    },
+    {
+      "epoch": 2.246043165467626,
+      "grad_norm": 0.22758009767252638,
+      "learning_rate": 5.37961283821783e-05,
+      "loss": 0.3924,
+      "step": 1561
+    },
+    {
+      "epoch": 2.2474820143884893,
+      "grad_norm": 0.25066092725129,
+      "learning_rate": 5.3758400683361926e-05,
+      "loss": 0.3997,
+      "step": 1562
+    },
+    {
+      "epoch": 2.2489208633093525,
+      "grad_norm": 0.2424680271461611,
+      "learning_rate": 5.372065909743479e-05,
+      "loss": 0.383,
+      "step": 1563
+    },
+    {
+      "epoch": 2.2503597122302157,
+      "grad_norm": 0.20431411327869473,
+      "learning_rate": 5.368290366249155e-05,
+      "loss": 0.3752,
+      "step": 1564
+    },
+    {
+      "epoch": 2.2517985611510793,
+      "grad_norm": 0.20934677671259003,
+      "learning_rate": 5.364513441664084e-05,
+      "loss": 0.3886,
+      "step": 1565
+    },
+    {
+      "epoch": 2.2532374100719426,
+      "grad_norm": 0.23967597244584293,
+      "learning_rate": 5.3607351398005234e-05,
+      "loss": 0.39,
+      "step": 1566
+    },
+    {
+      "epoch": 2.2546762589928058,
+      "grad_norm": 0.2657640992169594,
+      "learning_rate": 5.356955464472121e-05,
+      "loss": 0.3994,
+      "step": 1567
+    },
+    {
+      "epoch": 2.256115107913669,
+      "grad_norm": 0.2618534462802217,
+      "learning_rate": 5.353174419493913e-05,
+      "loss": 0.377,
+      "step": 1568
+    },
+    {
+      "epoch": 2.257553956834532,
+      "grad_norm": 0.24253080028381546,
+      "learning_rate": 5.349392008682314e-05,
+      "loss": 0.3921,
+      "step": 1569
+    },
+    {
+      "epoch": 2.258992805755396,
+      "grad_norm": 0.34307249380050653,
+      "learning_rate": 5.3456082358551204e-05,
+      "loss": 0.3899,
+      "step": 1570
+    },
+    {
+      "epoch": 2.260431654676259,
+      "grad_norm": 0.3409052578002414,
+      "learning_rate": 5.341823104831501e-05,
+      "loss": 0.3903,
+      "step": 1571
+    },
+    {
+      "epoch": 2.2618705035971223,
+      "grad_norm": 0.2339187186412542,
+      "learning_rate": 5.338036619431999e-05,
+      "loss": 0.3909,
+      "step": 1572
+    },
+    {
+      "epoch": 2.2633093525179855,
+      "grad_norm": 0.21239871731333118,
+      "learning_rate": 5.33424878347852e-05,
+      "loss": 0.3823,
+      "step": 1573
+    },
+    {
+      "epoch": 2.2647482014388487,
+      "grad_norm": 0.2929779081662266,
+      "learning_rate": 5.330459600794337e-05,
+      "loss": 0.3921,
+      "step": 1574
+    },
+    {
+      "epoch": 2.2661870503597124,
+      "grad_norm": 0.22381918976927967,
+      "learning_rate": 5.32666907520408e-05,
+      "loss": 0.3781,
+      "step": 1575
+    },
+    {
+      "epoch": 2.2676258992805756,
+      "grad_norm": 0.23426147068490208,
+      "learning_rate": 5.322877210533735e-05,
+      "loss": 0.3852,
+      "step": 1576
+    },
+    {
+      "epoch": 2.2690647482014388,
+      "grad_norm": 0.2719058724374197,
+      "learning_rate": 5.319084010610638e-05,
+      "loss": 0.387,
+      "step": 1577
+    },
+    {
+      "epoch": 2.270503597122302,
+      "grad_norm": 0.26897588687119994,
+      "learning_rate": 5.3152894792634785e-05,
+      "loss": 0.3866,
+      "step": 1578
+    },
+    {
+      "epoch": 2.2719424460431656,
+      "grad_norm": 0.32395317237830024,
+      "learning_rate": 5.311493620322282e-05,
+      "loss": 0.3819,
+      "step": 1579
+    },
+    {
+      "epoch": 2.273381294964029,
+      "grad_norm": 0.33028302802465415,
+      "learning_rate": 5.3076964376184186e-05,
+      "loss": 0.3881,
+      "step": 1580
+    },
+    {
+      "epoch": 2.274820143884892,
+      "grad_norm": 0.35374219168145615,
+      "learning_rate": 5.303897934984595e-05,
+      "loss": 0.3833,
+      "step": 1581
+    },
+    {
+      "epoch": 2.2762589928057553,
+      "grad_norm": 0.3192560554107421,
+      "learning_rate": 5.300098116254848e-05,
+      "loss": 0.3933,
+      "step": 1582
+    },
+    {
+      "epoch": 2.277697841726619,
+      "grad_norm": 0.3020262943029512,
+      "learning_rate": 5.296296985264543e-05,
+      "loss": 0.389,
+      "step": 1583
+    },
+    {
+      "epoch": 2.279136690647482,
+      "grad_norm": 0.3224059430495137,
+      "learning_rate": 5.2924945458503713e-05,
+      "loss": 0.3992,
+      "step": 1584
+    },
+    {
+      "epoch": 2.2805755395683454,
+      "grad_norm": 0.25799963699770995,
+      "learning_rate": 5.2886908018503454e-05,
+      "loss": 0.3862,
+      "step": 1585
+    },
+    {
+      "epoch": 2.2820143884892086,
+      "grad_norm": 0.21127475643898685,
+      "learning_rate": 5.284885757103792e-05,
+      "loss": 0.3994,
+      "step": 1586
+    },
+    {
+      "epoch": 2.283453237410072,
+      "grad_norm": 0.2333998104861042,
+      "learning_rate": 5.2810794154513503e-05,
+      "loss": 0.3894,
+      "step": 1587
+    },
+    {
+      "epoch": 2.2848920863309354,
+      "grad_norm": 0.26324442876196735,
+      "learning_rate": 5.277271780734975e-05,
+      "loss": 0.3855,
+      "step": 1588
+    },
+    {
+      "epoch": 2.2863309352517986,
+      "grad_norm": 0.22322544371503758,
+      "learning_rate": 5.273462856797918e-05,
+      "loss": 0.3931,
+      "step": 1589
+    },
+    {
+      "epoch": 2.287769784172662,
+      "grad_norm": 0.22903560510895377,
+      "learning_rate": 5.269652647484735e-05,
+      "loss": 0.3834,
+      "step": 1590
+    },
+    {
+      "epoch": 2.289208633093525,
+      "grad_norm": 0.24076772553209855,
+      "learning_rate": 5.2658411566412837e-05,
+      "loss": 0.3968,
+      "step": 1591
+    },
+    {
+      "epoch": 2.2906474820143883,
+      "grad_norm": 0.2496121201066887,
+      "learning_rate": 5.262028388114708e-05,
+      "loss": 0.3942,
+      "step": 1592
+    },
+    {
+      "epoch": 2.292086330935252,
+      "grad_norm": 0.22160483679343942,
+      "learning_rate": 5.258214345753446e-05,
+      "loss": 0.3858,
+      "step": 1593
+    },
+    {
+      "epoch": 2.293525179856115,
+      "grad_norm": 0.2633714450095719,
+      "learning_rate": 5.254399033407221e-05,
+      "loss": 0.3869,
+      "step": 1594
+    },
+    {
+      "epoch": 2.2949640287769784,
+      "grad_norm": 0.2665266751171906,
+      "learning_rate": 5.250582454927037e-05,
+      "loss": 0.3863,
+      "step": 1595
+    },
+    {
+      "epoch": 2.2964028776978416,
+      "grad_norm": 0.30336682615969185,
+      "learning_rate": 5.2467646141651764e-05,
+      "loss": 0.3944,
+      "step": 1596
+    },
+    {
+      "epoch": 2.2978417266187052,
+      "grad_norm": 0.28219159817093825,
+      "learning_rate": 5.2429455149751976e-05,
+      "loss": 0.3869,
+      "step": 1597
+    },
+    {
+      "epoch": 2.2992805755395684,
+      "grad_norm": 0.315586024026505,
+      "learning_rate": 5.2391251612119256e-05,
+      "loss": 0.3891,
+      "step": 1598
+    },
+    {
+      "epoch": 2.3007194244604317,
+      "grad_norm": 0.34645587224096863,
+      "learning_rate": 5.235303556731456e-05,
+      "loss": 0.3815,
+      "step": 1599
+    },
+    {
+      "epoch": 2.302158273381295,
+      "grad_norm": 0.32416702210472587,
+      "learning_rate": 5.23148070539114e-05,
+      "loss": 0.3943,
+      "step": 1600
+    },
+    {
+      "epoch": 2.3035971223021585,
+      "grad_norm": 0.34359351373970476,
+      "learning_rate": 5.227656611049598e-05,
+      "loss": 0.3797,
+      "step": 1601
+    },
+    {
+      "epoch": 2.3050359712230217,
+      "grad_norm": 0.28629848486179393,
+      "learning_rate": 5.2238312775666935e-05,
+      "loss": 0.3944,
+      "step": 1602
+    },
+    {
+      "epoch": 2.306474820143885,
+      "grad_norm": 0.19655450800377178,
+      "learning_rate": 5.220004708803548e-05,
+      "loss": 0.3869,
+      "step": 1603
+    },
+    {
+      "epoch": 2.307913669064748,
+      "grad_norm": 0.31199316853355963,
+      "learning_rate": 5.216176908622528e-05,
+      "loss": 0.392,
+      "step": 1604
+    },
+    {
+      "epoch": 2.3093525179856114,
+      "grad_norm": 0.31184479529560166,
+      "learning_rate": 5.2123478808872436e-05,
+      "loss": 0.3934,
+      "step": 1605
+    },
+    {
+      "epoch": 2.310791366906475,
+      "grad_norm": 0.2672326157880456,
+      "learning_rate": 5.208517629462541e-05,
+      "loss": 0.3896,
+      "step": 1606
+    },
+    {
+      "epoch": 2.3122302158273382,
+      "grad_norm": 0.21122181565761514,
+      "learning_rate": 5.204686158214507e-05,
+      "loss": 0.3923,
+      "step": 1607
+    },
+    {
+      "epoch": 2.3136690647482014,
+      "grad_norm": 0.2278103420446257,
+      "learning_rate": 5.200853471010453e-05,
+      "loss": 0.3894,
+      "step": 1608
+    },
+    {
+      "epoch": 2.3151079136690647,
+      "grad_norm": 0.24873073464908046,
+      "learning_rate": 5.197019571718921e-05,
+      "loss": 0.376,
+      "step": 1609
+    },
+    {
+      "epoch": 2.316546762589928,
+      "grad_norm": 0.21658729968501075,
+      "learning_rate": 5.19318446420968e-05,
+      "loss": 0.3855,
+      "step": 1610
+    },
+    {
+      "epoch": 2.3179856115107915,
+      "grad_norm": 0.2529075081894275,
+      "learning_rate": 5.189348152353712e-05,
+      "loss": 0.3783,
+      "step": 1611
+    },
+    {
+      "epoch": 2.3194244604316547,
+      "grad_norm": 0.2153775315337548,
+      "learning_rate": 5.1855106400232196e-05,
+      "loss": 0.3905,
+      "step": 1612
+    },
+    {
+      "epoch": 2.320863309352518,
+      "grad_norm": 0.269495226521631,
+      "learning_rate": 5.181671931091612e-05,
+      "loss": 0.3879,
+      "step": 1613
+    },
+    {
+      "epoch": 2.322302158273381,
+      "grad_norm": 0.2707722450348019,
+      "learning_rate": 5.1778320294335126e-05,
+      "loss": 0.3759,
+      "step": 1614
+    },
+    {
+      "epoch": 2.3237410071942444,
+      "grad_norm": 0.28244783707110455,
+      "learning_rate": 5.1739909389247445e-05,
+      "loss": 0.3916,
+      "step": 1615
+    },
+    {
+      "epoch": 2.325179856115108,
+      "grad_norm": 0.28701297443418267,
+      "learning_rate": 5.17014866344233e-05,
+      "loss": 0.3859,
+      "step": 1616
+    },
+    {
+      "epoch": 2.3266187050359712,
+      "grad_norm": 0.18644053030705382,
+      "learning_rate": 5.166305206864492e-05,
+      "loss": 0.3904,
+      "step": 1617
+    },
+    {
+      "epoch": 2.3280575539568344,
+      "grad_norm": 0.2709891480835394,
+      "learning_rate": 5.162460573070642e-05,
+      "loss": 0.3871,
+      "step": 1618
+    },
+    {
+      "epoch": 2.3294964028776977,
+      "grad_norm": 0.2617322592172387,
+      "learning_rate": 5.158614765941376e-05,
+      "loss": 0.3841,
+      "step": 1619
+    },
+    {
+      "epoch": 2.3309352517985613,
+      "grad_norm": 0.26174736986859515,
+      "learning_rate": 5.1547677893584846e-05,
+      "loss": 0.3859,
+      "step": 1620
+    },
+    {
+      "epoch": 2.3323741007194245,
+      "grad_norm": 0.2307642654355334,
+      "learning_rate": 5.15091964720493e-05,
+      "loss": 0.3916,
+      "step": 1621
+    },
+    {
+      "epoch": 2.3338129496402877,
+      "grad_norm": 0.21977774824301788,
+      "learning_rate": 5.1470703433648556e-05,
+      "loss": 0.3842,
+      "step": 1622
+    },
+    {
+      "epoch": 2.335251798561151,
+      "grad_norm": 0.22971288228127282,
+      "learning_rate": 5.143219881723573e-05,
+      "loss": 0.4007,
+      "step": 1623
+    },
+    {
+      "epoch": 2.3366906474820146,
+      "grad_norm": 0.28261283623252514,
+      "learning_rate": 5.139368266167567e-05,
+      "loss": 0.3981,
+      "step": 1624
+    },
+    {
+      "epoch": 2.338129496402878,
+      "grad_norm": 0.28051381072839177,
+      "learning_rate": 5.135515500584484e-05,
+      "loss": 0.3906,
+      "step": 1625
+    },
+    {
+      "epoch": 2.339568345323741,
+      "grad_norm": 0.2813911820987458,
+      "learning_rate": 5.131661588863132e-05,
+      "loss": 0.3875,
+      "step": 1626
+    },
+    {
+      "epoch": 2.3410071942446042,
+      "grad_norm": 0.2604237435520954,
+      "learning_rate": 5.1278065348934786e-05,
+      "loss": 0.3827,
+      "step": 1627
+    },
+    {
+      "epoch": 2.3424460431654675,
+      "grad_norm": 0.24605378870914793,
+      "learning_rate": 5.123950342566639e-05,
+      "loss": 0.3857,
+      "step": 1628
+    },
+    {
+      "epoch": 2.343884892086331,
+      "grad_norm": 0.2820459068205453,
+      "learning_rate": 5.120093015774882e-05,
+      "loss": 0.3891,
+      "step": 1629
+    },
+    {
+      "epoch": 2.3453237410071943,
+      "grad_norm": 0.29832490331226674,
+      "learning_rate": 5.116234558411618e-05,
+      "loss": 0.3954,
+      "step": 1630
+    },
+    {
+      "epoch": 2.3467625899280575,
+      "grad_norm": 0.27682515377856187,
+      "learning_rate": 5.1123749743714024e-05,
+      "loss": 0.3845,
+      "step": 1631
+    },
+    {
+      "epoch": 2.3482014388489207,
+      "grad_norm": 0.27644616429449503,
+      "learning_rate": 5.1085142675499246e-05,
+      "loss": 0.3836,
+      "step": 1632
+    },
+    {
+      "epoch": 2.349640287769784,
+      "grad_norm": 0.2686334074934592,
+      "learning_rate": 5.1046524418440075e-05,
+      "loss": 0.3899,
+      "step": 1633
+    },
+    {
+      "epoch": 2.3510791366906476,
+      "grad_norm": 0.25101187988966867,
+      "learning_rate": 5.100789501151607e-05,
+      "loss": 0.3919,
+      "step": 1634
+    },
+    {
+      "epoch": 2.352517985611511,
+      "grad_norm": 0.25963436453622124,
+      "learning_rate": 5.0969254493717996e-05,
+      "loss": 0.3893,
+      "step": 1635
+    },
+    {
+      "epoch": 2.353956834532374,
+      "grad_norm": 0.2857525462609916,
+      "learning_rate": 5.093060290404785e-05,
+      "loss": 0.3838,
+      "step": 1636
+    },
+    {
+      "epoch": 2.3553956834532372,
+      "grad_norm": 0.30757391496232317,
+      "learning_rate": 5.089194028151882e-05,
+      "loss": 0.3804,
+      "step": 1637
+    },
+    {
+      "epoch": 2.356834532374101,
+      "grad_norm": 0.38990252907497586,
+      "learning_rate": 5.085326666515521e-05,
+      "loss": 0.3904,
+      "step": 1638
+    },
+    {
+      "epoch": 2.358273381294964,
+      "grad_norm": 0.3608392668172516,
+      "learning_rate": 5.081458209399243e-05,
+      "loss": 0.3841,
+      "step": 1639
+    },
+    {
+      "epoch": 2.3597122302158273,
+      "grad_norm": 0.2955583789377134,
+      "learning_rate": 5.0775886607076954e-05,
+      "loss": 0.3895,
+      "step": 1640
+    },
+    {
+      "epoch": 2.3611510791366905,
+      "grad_norm": 0.2586577242218363,
+      "learning_rate": 5.073718024346626e-05,
+      "loss": 0.3761,
+      "step": 1641
+    },
+    {
+      "epoch": 2.362589928057554,
+      "grad_norm": 0.2615000083548353,
+      "learning_rate": 5.06984630422288e-05,
+      "loss": 0.3837,
+      "step": 1642
+    },
+    {
+      "epoch": 2.3640287769784174,
+      "grad_norm": 0.3326260309023068,
+      "learning_rate": 5.065973504244399e-05,
+      "loss": 0.3815,
+      "step": 1643
+    },
+    {
+      "epoch": 2.3654676258992806,
+      "grad_norm": 0.3193365286232002,
+      "learning_rate": 5.062099628320213e-05,
+      "loss": 0.3911,
+      "step": 1644
+    },
+    {
+      "epoch": 2.366906474820144,
+      "grad_norm": 0.2704723034935209,
+      "learning_rate": 5.058224680360438e-05,
+      "loss": 0.3902,
+      "step": 1645
+    },
+    {
+      "epoch": 2.368345323741007,
+      "grad_norm": 0.28067033155134913,
+      "learning_rate": 5.054348664276271e-05,
+      "loss": 0.397,
+      "step": 1646
+    },
+    {
+      "epoch": 2.3697841726618707,
+      "grad_norm": 0.3350444247767652,
+      "learning_rate": 5.05047158397999e-05,
+      "loss": 0.3905,
+      "step": 1647
+    },
+    {
+      "epoch": 2.371223021582734,
+      "grad_norm": 0.2931220556257705,
+      "learning_rate": 5.046593443384945e-05,
+      "loss": 0.3839,
+      "step": 1648
+    },
+    {
+      "epoch": 2.372661870503597,
+      "grad_norm": 0.2540958503700094,
+      "learning_rate": 5.042714246405555e-05,
+      "loss": 0.3862,
+      "step": 1649
+    },
+    {
+      "epoch": 2.3741007194244603,
+      "grad_norm": 0.23977184211113461,
+      "learning_rate": 5.038833996957309e-05,
+      "loss": 0.3906,
+      "step": 1650
+    },
+    {
+      "epoch": 2.3755395683453235,
+      "grad_norm": 0.25432728108805436,
+      "learning_rate": 5.0349526989567546e-05,
+      "loss": 0.3853,
+      "step": 1651
+    },
+    {
+      "epoch": 2.376978417266187,
+      "grad_norm": 0.2561709827869189,
+      "learning_rate": 5.0310703563215016e-05,
+      "loss": 0.3846,
+      "step": 1652
+    },
+    {
+      "epoch": 2.3784172661870504,
+      "grad_norm": 0.21704652844193725,
+      "learning_rate": 5.027186972970211e-05,
+      "loss": 0.388,
+      "step": 1653
+    },
+    {
+      "epoch": 2.3798561151079136,
+      "grad_norm": 0.17061332702750404,
+      "learning_rate": 5.0233025528225934e-05,
+      "loss": 0.3789,
+      "step": 1654
+    },
+    {
+      "epoch": 2.381294964028777,
+      "grad_norm": 0.18399449260481843,
+      "learning_rate": 5.01941709979941e-05,
+      "loss": 0.3799,
+      "step": 1655
+    },
+    {
+      "epoch": 2.38273381294964,
+      "grad_norm": 0.22162020684974135,
+      "learning_rate": 5.015530617822462e-05,
+      "loss": 0.3837,
+      "step": 1656
+    },
+    {
+      "epoch": 2.3841726618705037,
+      "grad_norm": 0.20671573781841068,
+      "learning_rate": 5.011643110814589e-05,
+      "loss": 0.3917,
+      "step": 1657
+    },
+    {
+      "epoch": 2.385611510791367,
+      "grad_norm": 0.16593354428047088,
+      "learning_rate": 5.007754582699666e-05,
+      "loss": 0.3845,
+      "step": 1658
+    },
+    {
+      "epoch": 2.38705035971223,
+      "grad_norm": 0.22254692306957252,
+      "learning_rate": 5.003865037402598e-05,
+      "loss": 0.394,
+      "step": 1659
+    },
+    {
+      "epoch": 2.3884892086330938,
+      "grad_norm": 0.22310726078403112,
+      "learning_rate": 4.999974478849319e-05,
+      "loss": 0.3804,
+      "step": 1660
+    },
+    {
+      "epoch": 2.389928057553957,
+      "grad_norm": 0.2619630555973278,
+      "learning_rate": 4.99608291096678e-05,
+      "loss": 0.3868,
+      "step": 1661
+    },
+    {
+      "epoch": 2.39136690647482,
+      "grad_norm": 0.2458818523957843,
+      "learning_rate": 4.9921903376829565e-05,
+      "loss": 0.3775,
+      "step": 1662
+    },
+    {
+      "epoch": 2.3928057553956834,
+      "grad_norm": 0.2579910384042537,
+      "learning_rate": 4.988296762926838e-05,
+      "loss": 0.3845,
+      "step": 1663
+    },
+    {
+      "epoch": 2.3942446043165466,
+      "grad_norm": 0.22595778256110546,
+      "learning_rate": 4.984402190628422e-05,
+      "loss": 0.3779,
+      "step": 1664
+    },
+    {
+      "epoch": 2.3956834532374103,
+      "grad_norm": 0.2365438089684538,
+      "learning_rate": 4.980506624718716e-05,
+      "loss": 0.3825,
+      "step": 1665
+    },
+    {
+      "epoch": 2.3971223021582735,
+      "grad_norm": 0.2854766742388604,
+      "learning_rate": 4.9766100691297284e-05,
+      "loss": 0.3939,
+      "step": 1666
+    },
+    {
+      "epoch": 2.3985611510791367,
+      "grad_norm": 0.362537703997184,
+      "learning_rate": 4.9727125277944675e-05,
+      "loss": 0.378,
+      "step": 1667
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 0.31454121000332186,
+      "learning_rate": 4.968814004646934e-05,
+      "loss": 0.399,
+      "step": 1668
+    },
+    {
+      "epoch": 2.401438848920863,
+      "grad_norm": 0.28207607370827553,
+      "learning_rate": 4.964914503622126e-05,
+      "loss": 0.3851,
+      "step": 1669
+    },
+    {
+      "epoch": 2.402877697841727,
+      "grad_norm": 0.25651958753351733,
+      "learning_rate": 4.961014028656021e-05,
+      "loss": 0.3913,
+      "step": 1670
+    },
+    {
+      "epoch": 2.40431654676259,
+      "grad_norm": 0.26643824821221995,
+      "learning_rate": 4.9571125836855825e-05,
+      "loss": 0.3855,
+      "step": 1671
+    },
+    {
+      "epoch": 2.405755395683453,
+      "grad_norm": 0.38308916207841903,
+      "learning_rate": 4.9532101726487564e-05,
+      "loss": 0.3787,
+      "step": 1672
+    },
+    {
+      "epoch": 2.4071942446043164,
+      "grad_norm": 0.4127926515912552,
+      "learning_rate": 4.9493067994844606e-05,
+      "loss": 0.3911,
+      "step": 1673
+    },
+    {
+      "epoch": 2.4086330935251796,
+      "grad_norm": 0.40012187170633984,
+      "learning_rate": 4.9454024681325815e-05,
+      "loss": 0.3862,
+      "step": 1674
+    },
+    {
+      "epoch": 2.4100719424460433,
+      "grad_norm": 0.3244702493554365,
+      "learning_rate": 4.941497182533978e-05,
+      "loss": 0.386,
+      "step": 1675
+    },
+    {
+      "epoch": 2.4115107913669065,
+      "grad_norm": 0.2840241844161442,
+      "learning_rate": 4.937590946630469e-05,
+      "loss": 0.3921,
+      "step": 1676
+    },
+    {
+      "epoch": 2.4129496402877697,
+      "grad_norm": 0.313971585605015,
+      "learning_rate": 4.9336837643648335e-05,
+      "loss": 0.3821,
+      "step": 1677
+    },
+    {
+      "epoch": 2.414388489208633,
+      "grad_norm": 0.4288719828415872,
+      "learning_rate": 4.929775639680805e-05,
+      "loss": 0.3888,
+      "step": 1678
+    },
+    {
+      "epoch": 2.4158273381294966,
+      "grad_norm": 0.44916133701646954,
+      "learning_rate": 4.925866576523069e-05,
+      "loss": 0.3868,
+      "step": 1679
+    },
+    {
+      "epoch": 2.41726618705036,
+      "grad_norm": 0.3259177807565223,
+      "learning_rate": 4.921956578837259e-05,
+      "loss": 0.3899,
+      "step": 1680
+    },
+    {
+      "epoch": 2.418705035971223,
+      "grad_norm": 0.26965941651128966,
+      "learning_rate": 4.918045650569949e-05,
+      "loss": 0.3851,
+      "step": 1681
+    },
+    {
+      "epoch": 2.420143884892086,
+      "grad_norm": 0.29360546025499556,
+      "learning_rate": 4.9141337956686564e-05,
+      "loss": 0.3882,
+      "step": 1682
+    },
+    {
+      "epoch": 2.42158273381295,
+      "grad_norm": 0.2907702864259603,
+      "learning_rate": 4.91022101808183e-05,
+      "loss": 0.3859,
+      "step": 1683
+    },
+    {
+      "epoch": 2.423021582733813,
+      "grad_norm": 0.26413262410779004,
+      "learning_rate": 4.90630732175885e-05,
+      "loss": 0.3816,
+      "step": 1684
+    },
+    {
+      "epoch": 2.4244604316546763,
+      "grad_norm": 0.2031957652912718,
+      "learning_rate": 4.902392710650028e-05,
+      "loss": 0.3852,
+      "step": 1685
+    },
+    {
+      "epoch": 2.4258992805755395,
+      "grad_norm": 0.21883823433372213,
+      "learning_rate": 4.898477188706596e-05,
+      "loss": 0.3844,
+      "step": 1686
+    },
+    {
+      "epoch": 2.4273381294964027,
+      "grad_norm": 0.24813905857170207,
+      "learning_rate": 4.894560759880705e-05,
+      "loss": 0.386,
+      "step": 1687
+    },
+    {
+      "epoch": 2.4287769784172664,
+      "grad_norm": 0.29886426314455694,
+      "learning_rate": 4.8906434281254223e-05,
+      "loss": 0.3849,
+      "step": 1688
+    },
+    {
+      "epoch": 2.4302158273381296,
+      "grad_norm": 0.2462432057626903,
+      "learning_rate": 4.886725197394726e-05,
+      "loss": 0.3709,
+      "step": 1689
+    },
+    {
+      "epoch": 2.431654676258993,
+      "grad_norm": 0.22984391832958947,
+      "learning_rate": 4.882806071643503e-05,
+      "loss": 0.3844,
+      "step": 1690
+    },
+    {
+      "epoch": 2.433093525179856,
+      "grad_norm": 0.20851399760817524,
+      "learning_rate": 4.878886054827541e-05,
+      "loss": 0.3885,
+      "step": 1691
+    },
+    {
+      "epoch": 2.434532374100719,
+      "grad_norm": 0.24732415430598192,
+      "learning_rate": 4.874965150903529e-05,
+      "loss": 0.3817,
+      "step": 1692
+    },
+    {
+      "epoch": 2.435971223021583,
+      "grad_norm": 0.31324080974370105,
+      "learning_rate": 4.871043363829053e-05,
+      "loss": 0.3915,
+      "step": 1693
+    },
+    {
+      "epoch": 2.437410071942446,
+      "grad_norm": 0.2797132571446861,
+      "learning_rate": 4.8671206975625856e-05,
+      "loss": 0.3811,
+      "step": 1694
+    },
+    {
+      "epoch": 2.4388489208633093,
+      "grad_norm": 0.26446108791806094,
+      "learning_rate": 4.863197156063492e-05,
+      "loss": 0.3806,
+      "step": 1695
+    },
+    {
+      "epoch": 2.4402877697841725,
+      "grad_norm": 0.2231399345001575,
+      "learning_rate": 4.859272743292017e-05,
+      "loss": 0.3875,
+      "step": 1696
+    },
+    {
+      "epoch": 2.441726618705036,
+      "grad_norm": 0.28628230910642666,
+      "learning_rate": 4.855347463209287e-05,
+      "loss": 0.4024,
+      "step": 1697
+    },
+    {
+      "epoch": 2.4431654676258994,
+      "grad_norm": 0.3078005652030596,
+      "learning_rate": 4.851421319777304e-05,
+      "loss": 0.3926,
+      "step": 1698
+    },
+    {
+      "epoch": 2.4446043165467626,
+      "grad_norm": 0.26622692722444935,
+      "learning_rate": 4.847494316958939e-05,
+      "loss": 0.3916,
+      "step": 1699
+    },
+    {
+      "epoch": 2.446043165467626,
+      "grad_norm": 0.2158880712975389,
+      "learning_rate": 4.8435664587179315e-05,
+      "loss": 0.3827,
+      "step": 1700
+    },
+    {
+      "epoch": 2.4474820143884894,
+      "grad_norm": 0.31400694325498657,
+      "learning_rate": 4.839637749018887e-05,
+      "loss": 0.3931,
+      "step": 1701
+    },
+    {
+      "epoch": 2.4489208633093527,
+      "grad_norm": 0.32557854065890673,
+      "learning_rate": 4.835708191827268e-05,
+      "loss": 0.3863,
+      "step": 1702
+    },
+    {
+      "epoch": 2.450359712230216,
+      "grad_norm": 0.21232550658716104,
+      "learning_rate": 4.831777791109392e-05,
+      "loss": 0.3868,
+      "step": 1703
+    },
+    {
+      "epoch": 2.451798561151079,
+      "grad_norm": 0.3361666495557492,
+      "learning_rate": 4.827846550832428e-05,
+      "loss": 0.3865,
+      "step": 1704
+    },
+    {
+      "epoch": 2.4532374100719423,
+      "grad_norm": 0.4299216376230312,
+      "learning_rate": 4.8239144749643936e-05,
+      "loss": 0.3796,
+      "step": 1705
+    },
+    {
+      "epoch": 2.454676258992806,
+      "grad_norm": 0.30048479764175423,
+      "learning_rate": 4.819981567474152e-05,
+      "loss": 0.3799,
+      "step": 1706
+    },
+    {
+      "epoch": 2.456115107913669,
+      "grad_norm": 0.21691420801560146,
+      "learning_rate": 4.8160478323313974e-05,
+      "loss": 0.3896,
+      "step": 1707
+    },
+    {
+      "epoch": 2.4575539568345324,
+      "grad_norm": 0.1941356571231044,
+      "learning_rate": 4.812113273506671e-05,
+      "loss": 0.3776,
+      "step": 1708
+    },
+    {
+      "epoch": 2.4589928057553956,
+      "grad_norm": 0.22030413531456672,
+      "learning_rate": 4.808177894971336e-05,
+      "loss": 0.3906,
+      "step": 1709
+    },
+    {
+      "epoch": 2.460431654676259,
+      "grad_norm": 0.2471830041613509,
+      "learning_rate": 4.804241700697588e-05,
+      "loss": 0.382,
+      "step": 1710
+    },
+    {
+      "epoch": 2.4618705035971225,
+      "grad_norm": 0.2632273217637152,
+      "learning_rate": 4.800304694658443e-05,
+      "loss": 0.3955,
+      "step": 1711
+    },
+    {
+      "epoch": 2.4633093525179857,
+      "grad_norm": 0.28868079552476417,
+      "learning_rate": 4.796366880827739e-05,
+      "loss": 0.3771,
+      "step": 1712
+    },
+    {
+      "epoch": 2.464748201438849,
+      "grad_norm": 0.21361752591660416,
+      "learning_rate": 4.792428263180128e-05,
+      "loss": 0.3871,
+      "step": 1713
+    },
+    {
+      "epoch": 2.466187050359712,
+      "grad_norm": 0.2507688064265132,
+      "learning_rate": 4.7884888456910734e-05,
+      "loss": 0.3901,
+      "step": 1714
+    },
+    {
+      "epoch": 2.4676258992805753,
+      "grad_norm": 0.2598408449312428,
+      "learning_rate": 4.784548632336846e-05,
+      "loss": 0.3861,
+      "step": 1715
+    },
+    {
+      "epoch": 2.469064748201439,
+      "grad_norm": 0.2296367717457473,
+      "learning_rate": 4.7806076270945197e-05,
+      "loss": 0.3857,
+      "step": 1716
+    },
+    {
+      "epoch": 2.470503597122302,
+      "grad_norm": 0.1940649865490585,
+      "learning_rate": 4.776665833941968e-05,
+      "loss": 0.3885,
+      "step": 1717
+    },
+    {
+      "epoch": 2.4719424460431654,
+      "grad_norm": 0.19295142640898422,
+      "learning_rate": 4.772723256857859e-05,
+      "loss": 0.3814,
+      "step": 1718
+    },
+    {
+      "epoch": 2.4733812949640286,
+      "grad_norm": 0.20467746234339637,
+      "learning_rate": 4.768779899821655e-05,
+      "loss": 0.3819,
+      "step": 1719
+    },
+    {
+      "epoch": 2.4748201438848922,
+      "grad_norm": 0.20585922948625396,
+      "learning_rate": 4.7648357668135996e-05,
+      "loss": 0.3956,
+      "step": 1720
+    },
+    {
+      "epoch": 2.4762589928057555,
+      "grad_norm": 0.2612823747600825,
+      "learning_rate": 4.760890861814726e-05,
+      "loss": 0.3779,
+      "step": 1721
+    },
+    {
+      "epoch": 2.4776978417266187,
+      "grad_norm": 0.2665043843681745,
+      "learning_rate": 4.756945188806843e-05,
+      "loss": 0.3887,
+      "step": 1722
+    },
+    {
+      "epoch": 2.479136690647482,
+      "grad_norm": 0.23432511588086502,
+      "learning_rate": 4.752998751772536e-05,
+      "loss": 0.3888,
+      "step": 1723
+    },
+    {
+      "epoch": 2.4805755395683455,
+      "grad_norm": 0.2565009983326827,
+      "learning_rate": 4.749051554695159e-05,
+      "loss": 0.3918,
+      "step": 1724
+    },
+    {
+      "epoch": 2.4820143884892087,
+      "grad_norm": 0.25325968492801615,
+      "learning_rate": 4.745103601558838e-05,
+      "loss": 0.388,
+      "step": 1725
+    },
+    {
+      "epoch": 2.483453237410072,
+      "grad_norm": 0.2182418705061013,
+      "learning_rate": 4.741154896348458e-05,
+      "loss": 0.3837,
+      "step": 1726
+    },
+    {
+      "epoch": 2.484892086330935,
+      "grad_norm": 0.236976744544838,
+      "learning_rate": 4.7372054430496636e-05,
+      "loss": 0.3875,
+      "step": 1727
+    },
+    {
+      "epoch": 2.4863309352517984,
+      "grad_norm": 0.22964815996595087,
+      "learning_rate": 4.733255245648857e-05,
+      "loss": 0.3829,
+      "step": 1728
+    },
+    {
+      "epoch": 2.487769784172662,
+      "grad_norm": 0.1651653371026665,
+      "learning_rate": 4.729304308133189e-05,
+      "loss": 0.3886,
+      "step": 1729
+    },
+    {
+      "epoch": 2.4892086330935252,
+      "grad_norm": 0.2218187872751732,
+      "learning_rate": 4.725352634490557e-05,
+      "loss": 0.3959,
+      "step": 1730
+    },
+    {
+      "epoch": 2.4906474820143885,
+      "grad_norm": 0.21094480244595293,
+      "learning_rate": 4.7214002287096035e-05,
+      "loss": 0.3888,
+      "step": 1731
+    },
+    {
+      "epoch": 2.4920863309352517,
+      "grad_norm": 0.4148620177860389,
+      "learning_rate": 4.7174470947797117e-05,
+      "loss": 0.3932,
+      "step": 1732
+    },
+    {
+      "epoch": 2.493525179856115,
+      "grad_norm": 0.30534444944427797,
+      "learning_rate": 4.7134932366909915e-05,
+      "loss": 0.3902,
+      "step": 1733
+    },
+    {
+      "epoch": 2.4949640287769785,
+      "grad_norm": 0.2879398340647878,
+      "learning_rate": 4.709538658434294e-05,
+      "loss": 0.3931,
+      "step": 1734
+    },
+    {
+      "epoch": 2.4964028776978417,
+      "grad_norm": 0.23892029069087156,
+      "learning_rate": 4.705583364001192e-05,
+      "loss": 0.3841,
+      "step": 1735
+    },
+    {
+      "epoch": 2.497841726618705,
+      "grad_norm": 0.19158560944302708,
+      "learning_rate": 4.701627357383981e-05,
+      "loss": 0.3881,
+      "step": 1736
+    },
+    {
+      "epoch": 2.499280575539568,
+      "grad_norm": 0.2319804244744238,
+      "learning_rate": 4.697670642575675e-05,
+      "loss": 0.3812,
+      "step": 1737
+    },
+    {
+      "epoch": 2.5007194244604314,
+      "grad_norm": 0.23479997824854995,
+      "learning_rate": 4.693713223570006e-05,
+      "loss": 0.3945,
+      "step": 1738
+    },
+    {
+      "epoch": 2.502158273381295,
+      "grad_norm": 0.2106359414153762,
+      "learning_rate": 4.689755104361414e-05,
+      "loss": 0.3925,
+      "step": 1739
+    },
+    {
+      "epoch": 2.5035971223021583,
+      "grad_norm": 0.22985112128848637,
+      "learning_rate": 4.685796288945046e-05,
+      "loss": 0.3846,
+      "step": 1740
+    },
+    {
+      "epoch": 2.5050359712230215,
+      "grad_norm": 0.2506365207244284,
+      "learning_rate": 4.6818367813167535e-05,
+      "loss": 0.3913,
+      "step": 1741
+    },
+    {
+      "epoch": 2.506474820143885,
+      "grad_norm": 0.2711726587566118,
+      "learning_rate": 4.6778765854730835e-05,
+      "loss": 0.39,
+      "step": 1742
+    },
+    {
+      "epoch": 2.5079136690647483,
+      "grad_norm": 0.2345839460545374,
+      "learning_rate": 4.673915705411281e-05,
+      "loss": 0.3734,
+      "step": 1743
+    },
+    {
+      "epoch": 2.5093525179856115,
+      "grad_norm": 0.17855830514222112,
+      "learning_rate": 4.6699541451292786e-05,
+      "loss": 0.3916,
+      "step": 1744
+    },
+    {
+      "epoch": 2.5107913669064748,
+      "grad_norm": 0.29951869445533486,
+      "learning_rate": 4.665991908625699e-05,
+      "loss": 0.3902,
+      "step": 1745
+    },
+    {
+      "epoch": 2.512230215827338,
+      "grad_norm": 0.36775107518074845,
+      "learning_rate": 4.6620289998998445e-05,
+      "loss": 0.3904,
+      "step": 1746
+    },
+    {
+      "epoch": 2.5136690647482016,
+      "grad_norm": 0.3677315970085526,
+      "learning_rate": 4.658065422951697e-05,
+      "loss": 0.391,
+      "step": 1747
+    },
+    {
+      "epoch": 2.515107913669065,
+      "grad_norm": 0.302362529995463,
+      "learning_rate": 4.654101181781913e-05,
+      "loss": 0.3822,
+      "step": 1748
+    },
+    {
+      "epoch": 2.516546762589928,
+      "grad_norm": 0.21053438895006088,
+      "learning_rate": 4.650136280391818e-05,
+      "loss": 0.3924,
+      "step": 1749
+    },
+    {
+      "epoch": 2.5179856115107913,
+      "grad_norm": 0.23268040489983013,
+      "learning_rate": 4.646170722783408e-05,
+      "loss": 0.3914,
+      "step": 1750
+    },
+    {
+      "epoch": 2.5194244604316545,
+      "grad_norm": 0.41306004698242743,
+      "learning_rate": 4.6422045129593344e-05,
+      "loss": 0.3776,
+      "step": 1751
+    },
+    {
+      "epoch": 2.520863309352518,
+      "grad_norm": 0.33919670732582624,
+      "learning_rate": 4.6382376549229146e-05,
+      "loss": 0.3866,
+      "step": 1752
+    },
+    {
+      "epoch": 2.5223021582733813,
+      "grad_norm": 0.24703474471833506,
+      "learning_rate": 4.634270152678115e-05,
+      "loss": 0.3815,
+      "step": 1753
+    },
+    {
+      "epoch": 2.5237410071942445,
+      "grad_norm": 0.20434553641822384,
+      "learning_rate": 4.630302010229555e-05,
+      "loss": 0.3827,
+      "step": 1754
+    },
+    {
+      "epoch": 2.5251798561151078,
+      "grad_norm": 0.24072794662768396,
+      "learning_rate": 4.6263332315824964e-05,
+      "loss": 0.3894,
+      "step": 1755
+    },
+    {
+      "epoch": 2.526618705035971,
+      "grad_norm": 0.2786157444020619,
+      "learning_rate": 4.622363820742848e-05,
+      "loss": 0.3916,
+      "step": 1756
+    },
+    {
+      "epoch": 2.5280575539568346,
+      "grad_norm": 0.29326672351755495,
+      "learning_rate": 4.618393781717156e-05,
+      "loss": 0.3824,
+      "step": 1757
+    },
+    {
+      "epoch": 2.529496402877698,
+      "grad_norm": 0.2739519540372521,
+      "learning_rate": 4.614423118512595e-05,
+      "loss": 0.3963,
+      "step": 1758
+    },
+    {
+      "epoch": 2.530935251798561,
+      "grad_norm": 0.27324386407840145,
+      "learning_rate": 4.610451835136978e-05,
+      "loss": 0.3893,
+      "step": 1759
+    },
+    {
+      "epoch": 2.5323741007194247,
+      "grad_norm": 0.2379716174002137,
+      "learning_rate": 4.606479935598738e-05,
+      "loss": 0.3815,
+      "step": 1760
+    },
+    {
+      "epoch": 2.533812949640288,
+      "grad_norm": 0.24029502092392346,
+      "learning_rate": 4.602507423906931e-05,
+      "loss": 0.3888,
+      "step": 1761
+    },
+    {
+      "epoch": 2.535251798561151,
+      "grad_norm": 0.20765957903574359,
+      "learning_rate": 4.598534304071233e-05,
+      "loss": 0.3935,
+      "step": 1762
+    },
+    {
+      "epoch": 2.5366906474820143,
+      "grad_norm": 0.21187887496187346,
+      "learning_rate": 4.5945605801019315e-05,
+      "loss": 0.3865,
+      "step": 1763
+    },
+    {
+      "epoch": 2.5381294964028775,
+      "grad_norm": 0.24969578617451754,
+      "learning_rate": 4.5905862560099255e-05,
+      "loss": 0.3842,
+      "step": 1764
+    },
+    {
+      "epoch": 2.539568345323741,
+      "grad_norm": 0.2132089086991246,
+      "learning_rate": 4.5866113358067187e-05,
+      "loss": 0.3834,
+      "step": 1765
+    },
+    {
+      "epoch": 2.5410071942446044,
+      "grad_norm": 0.23531078200450925,
+      "learning_rate": 4.582635823504416e-05,
+      "loss": 0.3836,
+      "step": 1766
+    },
+    {
+      "epoch": 2.5424460431654676,
+      "grad_norm": 0.18302970671707344,
+      "learning_rate": 4.5786597231157214e-05,
+      "loss": 0.3833,
+      "step": 1767
+    },
+    {
+      "epoch": 2.543884892086331,
+      "grad_norm": 0.19255100995567193,
+      "learning_rate": 4.574683038653932e-05,
+      "loss": 0.401,
+      "step": 1768
+    },
+    {
+      "epoch": 2.545323741007194,
+      "grad_norm": 0.23597284237746652,
+      "learning_rate": 4.5707057741329324e-05,
+      "loss": 0.376,
+      "step": 1769
+    },
+    {
+      "epoch": 2.5467625899280577,
+      "grad_norm": 0.2007552121621161,
+      "learning_rate": 4.5667279335671986e-05,
+      "loss": 0.3896,
+      "step": 1770
+    },
+    {
+      "epoch": 2.548201438848921,
+      "grad_norm": 0.18058634807160767,
+      "learning_rate": 4.56274952097178e-05,
+      "loss": 0.388,
+      "step": 1771
+    },
+    {
+      "epoch": 2.549640287769784,
+      "grad_norm": 0.15739369997263258,
+      "learning_rate": 4.558770540362308e-05,
+      "loss": 0.3872,
+      "step": 1772
+    },
+    {
+      "epoch": 2.5510791366906473,
+      "grad_norm": 0.1848617608599081,
+      "learning_rate": 4.554790995754988e-05,
+      "loss": 0.3854,
+      "step": 1773
+    },
+    {
+      "epoch": 2.5525179856115106,
+      "grad_norm": 0.24291941314215762,
+      "learning_rate": 4.5508108911665926e-05,
+      "loss": 0.3858,
+      "step": 1774
+    },
+    {
+      "epoch": 2.553956834532374,
+      "grad_norm": 0.21020910584334165,
+      "learning_rate": 4.5468302306144594e-05,
+      "loss": 0.3824,
+      "step": 1775
+    },
+    {
+      "epoch": 2.5553956834532374,
+      "grad_norm": 0.18853792035606604,
+      "learning_rate": 4.542849018116491e-05,
+      "loss": 0.3861,
+      "step": 1776
+    },
+    {
+      "epoch": 2.5568345323741006,
+      "grad_norm": 0.21209069463952884,
+      "learning_rate": 4.538867257691141e-05,
+      "loss": 0.3868,
+      "step": 1777
+    },
+    {
+      "epoch": 2.5582733812949643,
+      "grad_norm": 0.24596054016837202,
+      "learning_rate": 4.53488495335742e-05,
+      "loss": 0.3934,
+      "step": 1778
+    },
+    {
+      "epoch": 2.5597122302158275,
+      "grad_norm": 0.23147225203726496,
+      "learning_rate": 4.5309021091348885e-05,
+      "loss": 0.3864,
+      "step": 1779
+    },
+    {
+      "epoch": 2.5611510791366907,
+      "grad_norm": 0.20286262887801743,
+      "learning_rate": 4.5269187290436486e-05,
+      "loss": 0.3847,
+      "step": 1780
+    },
+    {
+      "epoch": 2.562589928057554,
+      "grad_norm": 0.20862433491865667,
+      "learning_rate": 4.5229348171043466e-05,
+      "loss": 0.3844,
+      "step": 1781
+    },
+    {
+      "epoch": 2.564028776978417,
+      "grad_norm": 0.2627683575084139,
+      "learning_rate": 4.51895037733816e-05,
+      "loss": 0.3868,
+      "step": 1782
+    },
+    {
+      "epoch": 2.565467625899281,
+      "grad_norm": 0.32422091188574925,
+      "learning_rate": 4.5149654137668095e-05,
+      "loss": 0.3817,
+      "step": 1783
+    },
+    {
+      "epoch": 2.566906474820144,
+      "grad_norm": 0.3165741282069741,
+      "learning_rate": 4.5109799304125333e-05,
+      "loss": 0.3892,
+      "step": 1784
+    },
+    {
+      "epoch": 2.568345323741007,
+      "grad_norm": 0.2567119926854359,
+      "learning_rate": 4.5069939312981e-05,
+      "loss": 0.3878,
+      "step": 1785
+    },
+    {
+      "epoch": 2.5697841726618704,
+      "grad_norm": 0.2267977012510537,
+      "learning_rate": 4.503007420446798e-05,
+      "loss": 0.3749,
+      "step": 1786
+    },
+    {
+      "epoch": 2.5712230215827336,
+      "grad_norm": 0.34461047558661684,
+      "learning_rate": 4.499020401882433e-05,
+      "loss": 0.3943,
+      "step": 1787
+    },
+    {
+      "epoch": 2.5726618705035973,
+      "grad_norm": 0.32960505460472705,
+      "learning_rate": 4.49503287962932e-05,
+      "loss": 0.3867,
+      "step": 1788
+    },
+    {
+      "epoch": 2.5741007194244605,
+      "grad_norm": 0.22921860434176516,
+      "learning_rate": 4.491044857712288e-05,
+      "loss": 0.3884,
+      "step": 1789
+    },
+    {
+      "epoch": 2.5755395683453237,
+      "grad_norm": 0.22852165220912105,
+      "learning_rate": 4.4870563401566634e-05,
+      "loss": 0.3958,
+      "step": 1790
+    },
+    {
+      "epoch": 2.576978417266187,
+      "grad_norm": 0.24099158446341248,
+      "learning_rate": 4.483067330988278e-05,
+      "loss": 0.3795,
+      "step": 1791
+    },
+    {
+      "epoch": 2.57841726618705,
+      "grad_norm": 0.23129817552242166,
+      "learning_rate": 4.479077834233458e-05,
+      "loss": 0.3877,
+      "step": 1792
+    },
+    {
+      "epoch": 2.579856115107914,
+      "grad_norm": 0.23630958464372778,
+      "learning_rate": 4.475087853919023e-05,
+      "loss": 0.3833,
+      "step": 1793
+    },
+    {
+      "epoch": 2.581294964028777,
+      "grad_norm": 0.1994916154967065,
+      "learning_rate": 4.4710973940722786e-05,
+      "loss": 0.389,
+      "step": 1794
+    },
+    {
+      "epoch": 2.58273381294964,
+      "grad_norm": 0.25994331315702757,
+      "learning_rate": 4.4671064587210146e-05,
+      "loss": 0.3822,
+      "step": 1795
+    },
+    {
+      "epoch": 2.584172661870504,
+      "grad_norm": 0.21802541448223417,
+      "learning_rate": 4.4631150518935044e-05,
+      "loss": 0.3842,
+      "step": 1796
+    },
+    {
+      "epoch": 2.5856115107913666,
+      "grad_norm": 0.19238044186831893,
+      "learning_rate": 4.459123177618491e-05,
+      "loss": 0.3781,
+      "step": 1797
+    },
+    {
+      "epoch": 2.5870503597122303,
+      "grad_norm": 0.18801893052710708,
+      "learning_rate": 4.455130839925195e-05,
+      "loss": 0.3818,
+      "step": 1798
+    },
+    {
+      "epoch": 2.5884892086330935,
+      "grad_norm": 0.21243905334812005,
+      "learning_rate": 4.451138042843302e-05,
+      "loss": 0.3956,
+      "step": 1799
+    },
+    {
+      "epoch": 2.5899280575539567,
+      "grad_norm": 0.214458072394087,
+      "learning_rate": 4.447144790402963e-05,
+      "loss": 0.3813,
+      "step": 1800
+    },
+    {
+      "epoch": 2.5913669064748204,
+      "grad_norm": 0.2653829259754617,
+      "learning_rate": 4.4431510866347837e-05,
+      "loss": 0.3857,
+      "step": 1801
+    },
+    {
+      "epoch": 2.5928057553956836,
+      "grad_norm": 0.26311399803344365,
+      "learning_rate": 4.439156935569833e-05,
+      "loss": 0.3837,
+      "step": 1802
+    },
+    {
+      "epoch": 2.594244604316547,
+      "grad_norm": 0.24357443397451756,
+      "learning_rate": 4.435162341239625e-05,
+      "loss": 0.3826,
+      "step": 1803
+    },
+    {
+      "epoch": 2.59568345323741,
+      "grad_norm": 0.2329412189965746,
+      "learning_rate": 4.4311673076761254e-05,
+      "loss": 0.3874,
+      "step": 1804
+    },
+    {
+      "epoch": 2.597122302158273,
+      "grad_norm": 0.20926413357202217,
+      "learning_rate": 4.42717183891174e-05,
+      "loss": 0.3842,
+      "step": 1805
+    },
+    {
+      "epoch": 2.598561151079137,
+      "grad_norm": 0.17769587040566737,
+      "learning_rate": 4.4231759389793144e-05,
+      "loss": 0.3868,
+      "step": 1806
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 0.21004670713763612,
+      "learning_rate": 4.4191796119121335e-05,
+      "loss": 0.3813,
+      "step": 1807
+    },
+    {
+      "epoch": 2.6014388489208633,
+      "grad_norm": 0.2235053760814713,
+      "learning_rate": 4.415182861743906e-05,
+      "loss": 0.3765,
+      "step": 1808
+    },
+    {
+      "epoch": 2.6028776978417265,
+      "grad_norm": 0.3040060715703648,
+      "learning_rate": 4.411185692508774e-05,
+      "loss": 0.3812,
+      "step": 1809
+    },
+    {
+      "epoch": 2.6043165467625897,
+      "grad_norm": 0.26147948047551334,
+      "learning_rate": 4.4071881082413e-05,
+      "loss": 0.3842,
+      "step": 1810
+    },
+    {
+      "epoch": 2.6057553956834534,
+      "grad_norm": 0.1745707951688262,
+      "learning_rate": 4.4031901129764665e-05,
+      "loss": 0.3847,
+      "step": 1811
+    },
+    {
+      "epoch": 2.6071942446043166,
+      "grad_norm": 0.20669080261916384,
+      "learning_rate": 4.3991917107496695e-05,
+      "loss": 0.3898,
+      "step": 1812
+    },
+    {
+      "epoch": 2.60863309352518,
+      "grad_norm": 0.23227452863695033,
+      "learning_rate": 4.395192905596716e-05,
+      "loss": 0.3717,
+      "step": 1813
+    },
+    {
+      "epoch": 2.610071942446043,
+      "grad_norm": 0.22790458417236623,
+      "learning_rate": 4.3911937015538186e-05,
+      "loss": 0.3846,
+      "step": 1814
+    },
+    {
+      "epoch": 2.6115107913669062,
+      "grad_norm": 0.18928946784415096,
+      "learning_rate": 4.3871941026575965e-05,
+      "loss": 0.3827,
+      "step": 1815
+    },
+    {
+      "epoch": 2.61294964028777,
+      "grad_norm": 0.21757958462658672,
+      "learning_rate": 4.383194112945066e-05,
+      "loss": 0.3935,
+      "step": 1816
+    },
+    {
+      "epoch": 2.614388489208633,
+      "grad_norm": 0.2827530676409471,
+      "learning_rate": 4.379193736453633e-05,
+      "loss": 0.3868,
+      "step": 1817
+    },
+    {
+      "epoch": 2.6158273381294963,
+      "grad_norm": 0.2630615627721376,
+      "learning_rate": 4.375192977221099e-05,
+      "loss": 0.3888,
+      "step": 1818
+    },
+    {
+      "epoch": 2.61726618705036,
+      "grad_norm": 0.2168400391998586,
+      "learning_rate": 4.371191839285651e-05,
+      "loss": 0.3856,
+      "step": 1819
+    },
+    {
+      "epoch": 2.618705035971223,
+      "grad_norm": 0.22607905125918135,
+      "learning_rate": 4.367190326685858e-05,
+      "loss": 0.3909,
+      "step": 1820
+    },
+    {
+      "epoch": 2.6201438848920864,
+      "grad_norm": 0.19453063079707447,
+      "learning_rate": 4.363188443460666e-05,
+      "loss": 0.3864,
+      "step": 1821
+    },
+    {
+      "epoch": 2.6215827338129496,
+      "grad_norm": 0.2657942478306949,
+      "learning_rate": 4.3591861936493964e-05,
+      "loss": 0.379,
+      "step": 1822
+    },
+    {
+      "epoch": 2.623021582733813,
+      "grad_norm": 0.27540316986659147,
+      "learning_rate": 4.3551835812917395e-05,
+      "loss": 0.3739,
+      "step": 1823
+    },
+    {
+      "epoch": 2.6244604316546765,
+      "grad_norm": 0.24557361664055938,
+      "learning_rate": 4.351180610427754e-05,
+      "loss": 0.3875,
+      "step": 1824
+    },
+    {
+      "epoch": 2.6258992805755397,
+      "grad_norm": 0.24206244637264343,
+      "learning_rate": 4.347177285097855e-05,
+      "loss": 0.3835,
+      "step": 1825
+    },
+    {
+      "epoch": 2.627338129496403,
+      "grad_norm": 0.19725770112484106,
+      "learning_rate": 4.343173609342822e-05,
+      "loss": 0.3881,
+      "step": 1826
+    },
+    {
+      "epoch": 2.628776978417266,
+      "grad_norm": 0.2608118908692465,
+      "learning_rate": 4.339169587203785e-05,
+      "loss": 0.3812,
+      "step": 1827
+    },
+    {
+      "epoch": 2.6302158273381293,
+      "grad_norm": 0.29338714121904846,
+      "learning_rate": 4.335165222722222e-05,
+      "loss": 0.389,
+      "step": 1828
+    },
+    {
+      "epoch": 2.631654676258993,
+      "grad_norm": 0.2562460674118761,
+      "learning_rate": 4.331160519939962e-05,
+      "loss": 0.3902,
+      "step": 1829
+    },
+    {
+      "epoch": 2.633093525179856,
+      "grad_norm": 0.2136119726581698,
+      "learning_rate": 4.327155482899168e-05,
+      "loss": 0.3937,
+      "step": 1830
+    },
+    {
+      "epoch": 2.6345323741007194,
+      "grad_norm": 0.18708751197016005,
+      "learning_rate": 4.323150115642346e-05,
+      "loss": 0.3867,
+      "step": 1831
+    },
+    {
+      "epoch": 2.6359712230215826,
+      "grad_norm": 0.1915683028324215,
+      "learning_rate": 4.3191444222123326e-05,
+      "loss": 0.382,
+      "step": 1832
+    },
+    {
+      "epoch": 2.637410071942446,
+      "grad_norm": 0.19480422569077002,
+      "learning_rate": 4.3151384066522964e-05,
+      "loss": 0.3799,
+      "step": 1833
+    },
+    {
+      "epoch": 2.6388489208633095,
+      "grad_norm": 0.21530948884558096,
+      "learning_rate": 4.311132073005727e-05,
+      "loss": 0.3857,
+      "step": 1834
+    },
+    {
+      "epoch": 2.6402877697841727,
+      "grad_norm": 0.1864530172996806,
+      "learning_rate": 4.3071254253164395e-05,
+      "loss": 0.3879,
+      "step": 1835
+    },
+    {
+      "epoch": 2.641726618705036,
+      "grad_norm": 0.17590300501204492,
+      "learning_rate": 4.3031184676285625e-05,
+      "loss": 0.3983,
+      "step": 1836
+    },
+    {
+      "epoch": 2.6431654676258995,
+      "grad_norm": 0.20243039026627202,
+      "learning_rate": 4.299111203986539e-05,
+      "loss": 0.3869,
+      "step": 1837
+    },
+    {
+      "epoch": 2.6446043165467623,
+      "grad_norm": 0.3099275802406128,
+      "learning_rate": 4.29510363843512e-05,
+      "loss": 0.3842,
+      "step": 1838
+    },
+    {
+      "epoch": 2.646043165467626,
+      "grad_norm": 0.20685838531850764,
+      "learning_rate": 4.291095775019364e-05,
+      "loss": 0.3934,
+      "step": 1839
+    },
+    {
+      "epoch": 2.647482014388489,
+      "grad_norm": 0.18001002469842584,
+      "learning_rate": 4.287087617784627e-05,
+      "loss": 0.3838,
+      "step": 1840
+    },
+    {
+      "epoch": 2.6489208633093524,
+      "grad_norm": 0.21630775495621132,
+      "learning_rate": 4.283079170776561e-05,
+      "loss": 0.3822,
+      "step": 1841
+    },
+    {
+      "epoch": 2.650359712230216,
+      "grad_norm": 0.1882221069360396,
+      "learning_rate": 4.279070438041116e-05,
+      "loss": 0.39,
+      "step": 1842
+    },
+    {
+      "epoch": 2.6517985611510793,
+      "grad_norm": 0.23126448519474344,
+      "learning_rate": 4.275061423624522e-05,
+      "loss": 0.3866,
+      "step": 1843
+    },
+    {
+      "epoch": 2.6532374100719425,
+      "grad_norm": 0.2024574208546783,
+      "learning_rate": 4.2710521315733e-05,
+      "loss": 0.3862,
+      "step": 1844
+    },
+    {
+      "epoch": 2.6546762589928057,
+      "grad_norm": 0.24465296829058134,
+      "learning_rate": 4.26704256593425e-05,
+      "loss": 0.3915,
+      "step": 1845
+    },
+    {
+      "epoch": 2.656115107913669,
+      "grad_norm": 0.20156038024269043,
+      "learning_rate": 4.2630327307544454e-05,
+      "loss": 0.382,
+      "step": 1846
+    },
+    {
+      "epoch": 2.6575539568345325,
+      "grad_norm": 0.20093878292170558,
+      "learning_rate": 4.2590226300812335e-05,
+      "loss": 0.3937,
+      "step": 1847
+    },
+    {
+      "epoch": 2.6589928057553958,
+      "grad_norm": 0.20992811409452886,
+      "learning_rate": 4.255012267962232e-05,
+      "loss": 0.389,
+      "step": 1848
+    },
+    {
+      "epoch": 2.660431654676259,
+      "grad_norm": 0.19443184008420034,
+      "learning_rate": 4.251001648445317e-05,
+      "loss": 0.3774,
+      "step": 1849
+    },
+    {
+      "epoch": 2.661870503597122,
+      "grad_norm": 0.20508834552588467,
+      "learning_rate": 4.246990775578628e-05,
+      "loss": 0.3916,
+      "step": 1850
+    },
+    {
+      "epoch": 2.6633093525179854,
+      "grad_norm": 0.21597683207543966,
+      "learning_rate": 4.242979653410562e-05,
+      "loss": 0.3857,
+      "step": 1851
+    },
+    {
+      "epoch": 2.664748201438849,
+      "grad_norm": 0.2314417209320147,
+      "learning_rate": 4.238968285989762e-05,
+      "loss": 0.3874,
+      "step": 1852
+    },
+    {
+      "epoch": 2.6661870503597123,
+      "grad_norm": 0.21639164495492177,
+      "learning_rate": 4.2349566773651236e-05,
+      "loss": 0.3811,
+      "step": 1853
+    },
+    {
+      "epoch": 2.6676258992805755,
+      "grad_norm": 0.21936968244618701,
+      "learning_rate": 4.2309448315857844e-05,
+      "loss": 0.3876,
+      "step": 1854
+    },
+    {
+      "epoch": 2.6690647482014387,
+      "grad_norm": 0.23102840657098456,
+      "learning_rate": 4.226932752701122e-05,
+      "loss": 0.3925,
+      "step": 1855
+    },
+    {
+      "epoch": 2.670503597122302,
+      "grad_norm": 0.19722110868434983,
+      "learning_rate": 4.2229204447607456e-05,
+      "loss": 0.383,
+      "step": 1856
+    },
+    {
+      "epoch": 2.6719424460431656,
+      "grad_norm": 0.2195835985460399,
+      "learning_rate": 4.2189079118145e-05,
+      "loss": 0.386,
+      "step": 1857
+    },
+    {
+      "epoch": 2.6733812949640288,
+      "grad_norm": 0.22270094722650063,
+      "learning_rate": 4.214895157912454e-05,
+      "loss": 0.3807,
+      "step": 1858
+    },
+    {
+      "epoch": 2.674820143884892,
+      "grad_norm": 0.19377785620489504,
+      "learning_rate": 4.210882187104904e-05,
+      "loss": 0.3869,
+      "step": 1859
+    },
+    {
+      "epoch": 2.6762589928057556,
+      "grad_norm": 0.254485241645623,
+      "learning_rate": 4.206869003442358e-05,
+      "loss": 0.3846,
+      "step": 1860
+    },
+    {
+      "epoch": 2.677697841726619,
+      "grad_norm": 0.22236843005196488,
+      "learning_rate": 4.2028556109755465e-05,
+      "loss": 0.3839,
+      "step": 1861
+    },
+    {
+      "epoch": 2.679136690647482,
+      "grad_norm": 0.1969134763507888,
+      "learning_rate": 4.198842013755408e-05,
+      "loss": 0.3868,
+      "step": 1862
+    },
+    {
+      "epoch": 2.6805755395683453,
+      "grad_norm": 0.26206145325810626,
+      "learning_rate": 4.194828215833082e-05,
+      "loss": 0.3842,
+      "step": 1863
+    },
+    {
+      "epoch": 2.6820143884892085,
+      "grad_norm": 0.26348235581142704,
+      "learning_rate": 4.1908142212599206e-05,
+      "loss": 0.3962,
+      "step": 1864
+    },
+    {
+      "epoch": 2.683453237410072,
+      "grad_norm": 0.27050117521450223,
+      "learning_rate": 4.1868000340874674e-05,
+      "loss": 0.3947,
+      "step": 1865
+    },
+    {
+      "epoch": 2.6848920863309353,
+      "grad_norm": 0.24475766021828424,
+      "learning_rate": 4.182785658367462e-05,
+      "loss": 0.3861,
+      "step": 1866
+    },
+    {
+      "epoch": 2.6863309352517986,
+      "grad_norm": 0.22801076642145454,
+      "learning_rate": 4.178771098151835e-05,
+      "loss": 0.3888,
+      "step": 1867
+    },
+    {
+      "epoch": 2.6877697841726618,
+      "grad_norm": 0.2739226492353372,
+      "learning_rate": 4.1747563574927034e-05,
+      "loss": 0.3846,
+      "step": 1868
+    },
+    {
+      "epoch": 2.689208633093525,
+      "grad_norm": 0.21666453088785986,
+      "learning_rate": 4.170741440442366e-05,
+      "loss": 0.3885,
+      "step": 1869
+    },
+    {
+      "epoch": 2.6906474820143886,
+      "grad_norm": 0.1935920292066028,
+      "learning_rate": 4.166726351053299e-05,
+      "loss": 0.3842,
+      "step": 1870
+    },
+    {
+      "epoch": 2.692086330935252,
+      "grad_norm": 0.2554502113776128,
+      "learning_rate": 4.1627110933781515e-05,
+      "loss": 0.3802,
+      "step": 1871
+    },
+    {
+      "epoch": 2.693525179856115,
+      "grad_norm": 0.25847156484503014,
+      "learning_rate": 4.158695671469746e-05,
+      "loss": 0.3892,
+      "step": 1872
+    },
+    {
+      "epoch": 2.6949640287769783,
+      "grad_norm": 0.21120440672036592,
+      "learning_rate": 4.154680089381068e-05,
+      "loss": 0.3948,
+      "step": 1873
+    },
+    {
+      "epoch": 2.6964028776978415,
+      "grad_norm": 0.23296719621704934,
+      "learning_rate": 4.150664351165266e-05,
+      "loss": 0.3887,
+      "step": 1874
+    },
+    {
+      "epoch": 2.697841726618705,
+      "grad_norm": 0.19948198703925238,
+      "learning_rate": 4.146648460875646e-05,
+      "loss": 0.3841,
+      "step": 1875
+    },
+    {
+      "epoch": 2.6992805755395683,
+      "grad_norm": 0.23759199087716618,
+      "learning_rate": 4.1426324225656644e-05,
+      "loss": 0.3865,
+      "step": 1876
+    },
+    {
+      "epoch": 2.7007194244604316,
+      "grad_norm": 0.22701961869252213,
+      "learning_rate": 4.138616240288934e-05,
+      "loss": 0.3916,
+      "step": 1877
+    },
+    {
+      "epoch": 2.702158273381295,
+      "grad_norm": 0.20588673256139395,
+      "learning_rate": 4.134599918099204e-05,
+      "loss": 0.3875,
+      "step": 1878
+    },
+    {
+      "epoch": 2.7035971223021584,
+      "grad_norm": 0.21325481866450907,
+      "learning_rate": 4.130583460050371e-05,
+      "loss": 0.3704,
+      "step": 1879
+    },
+    {
+      "epoch": 2.7050359712230216,
+      "grad_norm": 0.16806017671260462,
+      "learning_rate": 4.126566870196468e-05,
+      "loss": 0.389,
+      "step": 1880
+    },
+    {
+      "epoch": 2.706474820143885,
+      "grad_norm": 0.22813748694019656,
+      "learning_rate": 4.12255015259166e-05,
+      "loss": 0.3863,
+      "step": 1881
+    },
+    {
+      "epoch": 2.707913669064748,
+      "grad_norm": 0.18398201430847969,
+      "learning_rate": 4.1185333112902394e-05,
+      "loss": 0.3843,
+      "step": 1882
+    },
+    {
+      "epoch": 2.7093525179856117,
+      "grad_norm": 0.20900623881404212,
+      "learning_rate": 4.114516350346626e-05,
+      "loss": 0.3843,
+      "step": 1883
+    },
+    {
+      "epoch": 2.710791366906475,
+      "grad_norm": 0.19209389526656,
+      "learning_rate": 4.1104992738153616e-05,
+      "loss": 0.3899,
+      "step": 1884
+    },
+    {
+      "epoch": 2.712230215827338,
+      "grad_norm": 0.17657244957473936,
+      "learning_rate": 4.1064820857511e-05,
+      "loss": 0.3904,
+      "step": 1885
+    },
+    {
+      "epoch": 2.7136690647482014,
+      "grad_norm": 0.18002473184963683,
+      "learning_rate": 4.1024647902086107e-05,
+      "loss": 0.3788,
+      "step": 1886
+    },
+    {
+      "epoch": 2.7151079136690646,
+      "grad_norm": 0.16759575946391514,
+      "learning_rate": 4.0984473912427706e-05,
+      "loss": 0.3938,
+      "step": 1887
+    },
+    {
+      "epoch": 2.716546762589928,
+      "grad_norm": 0.18222342273573017,
+      "learning_rate": 4.0944298929085633e-05,
+      "loss": 0.3858,
+      "step": 1888
+    },
+    {
+      "epoch": 2.7179856115107914,
+      "grad_norm": 0.21787089769791332,
+      "learning_rate": 4.090412299261068e-05,
+      "loss": 0.3782,
+      "step": 1889
+    },
+    {
+      "epoch": 2.7194244604316546,
+      "grad_norm": 0.2055194052955941,
+      "learning_rate": 4.086394614355467e-05,
+      "loss": 0.3806,
+      "step": 1890
+    },
+    {
+      "epoch": 2.720863309352518,
+      "grad_norm": 0.17948724453157378,
+      "learning_rate": 4.082376842247027e-05,
+      "loss": 0.3855,
+      "step": 1891
+    },
+    {
+      "epoch": 2.722302158273381,
+      "grad_norm": 0.15385133013252994,
+      "learning_rate": 4.0783589869911074e-05,
+      "loss": 0.3838,
+      "step": 1892
+    },
+    {
+      "epoch": 2.7237410071942447,
+      "grad_norm": 0.209723402063529,
+      "learning_rate": 4.074341052643152e-05,
+      "loss": 0.3801,
+      "step": 1893
+    },
+    {
+      "epoch": 2.725179856115108,
+      "grad_norm": 0.19691672015146391,
+      "learning_rate": 4.070323043258683e-05,
+      "loss": 0.3846,
+      "step": 1894
+    },
+    {
+      "epoch": 2.726618705035971,
+      "grad_norm": 0.16722335973554103,
+      "learning_rate": 4.066304962893297e-05,
+      "loss": 0.3821,
+      "step": 1895
+    },
+    {
+      "epoch": 2.728057553956835,
+      "grad_norm": 0.19678366933227714,
+      "learning_rate": 4.062286815602661e-05,
+      "loss": 0.385,
+      "step": 1896
+    },
+    {
+      "epoch": 2.7294964028776976,
+      "grad_norm": 0.23986977143401303,
+      "learning_rate": 4.0582686054425196e-05,
+      "loss": 0.3815,
+      "step": 1897
+    },
+    {
+      "epoch": 2.7309352517985612,
+      "grad_norm": 0.17856562550021124,
+      "learning_rate": 4.054250336468666e-05,
+      "loss": 0.386,
+      "step": 1898
+    },
+    {
+      "epoch": 2.7323741007194244,
+      "grad_norm": 0.24355770904784735,
+      "learning_rate": 4.050232012736964e-05,
+      "loss": 0.3833,
+      "step": 1899
+    },
+    {
+      "epoch": 2.7338129496402876,
+      "grad_norm": 0.25168778926362656,
+      "learning_rate": 4.0462136383033285e-05,
+      "loss": 0.3914,
+      "step": 1900
+    },
+    {
+      "epoch": 2.7352517985611513,
+      "grad_norm": 0.19058711164039246,
+      "learning_rate": 4.0421952172237254e-05,
+      "loss": 0.393,
+      "step": 1901
+    },
+    {
+      "epoch": 2.7366906474820145,
+      "grad_norm": 0.19474399024123062,
+      "learning_rate": 4.038176753554166e-05,
+      "loss": 0.3947,
+      "step": 1902
+    },
+    {
+      "epoch": 2.7381294964028777,
+      "grad_norm": 0.18578664144744955,
+      "learning_rate": 4.034158251350711e-05,
+      "loss": 0.392,
+      "step": 1903
+    },
+    {
+      "epoch": 2.739568345323741,
+      "grad_norm": 0.18928528480319595,
+      "learning_rate": 4.030139714669453e-05,
+      "loss": 0.3836,
+      "step": 1904
+    },
+    {
+      "epoch": 2.741007194244604,
+      "grad_norm": 0.19240981764999657,
+      "learning_rate": 4.026121147566522e-05,
+      "loss": 0.3977,
+      "step": 1905
+    },
+    {
+      "epoch": 2.742446043165468,
+      "grad_norm": 0.15538158480519618,
+      "learning_rate": 4.02210255409808e-05,
+      "loss": 0.3806,
+      "step": 1906
+    },
+    {
+      "epoch": 2.743884892086331,
+      "grad_norm": 0.19774743810262413,
+      "learning_rate": 4.018083938320314e-05,
+      "loss": 0.3853,
+      "step": 1907
+    },
+    {
+      "epoch": 2.7453237410071942,
+      "grad_norm": 0.18394363944089592,
+      "learning_rate": 4.014065304289435e-05,
+      "loss": 0.3868,
+      "step": 1908
+    },
+    {
+      "epoch": 2.7467625899280574,
+      "grad_norm": 0.20010680409203171,
+      "learning_rate": 4.010046656061669e-05,
+      "loss": 0.3814,
+      "step": 1909
+    },
+    {
+      "epoch": 2.7482014388489207,
+      "grad_norm": 0.27375663615626805,
+      "learning_rate": 4.006027997693262e-05,
+      "loss": 0.3852,
+      "step": 1910
+    },
+    {
+      "epoch": 2.7496402877697843,
+      "grad_norm": 0.16225804971406108,
+      "learning_rate": 4.002009333240465e-05,
+      "loss": 0.3888,
+      "step": 1911
+    },
+    {
+      "epoch": 2.7510791366906475,
+      "grad_norm": 0.21776413666241295,
+      "learning_rate": 3.997990666759536e-05,
+      "loss": 0.3799,
+      "step": 1912
+    },
+    {
+      "epoch": 2.7525179856115107,
+      "grad_norm": 0.24881360037822242,
+      "learning_rate": 3.99397200230674e-05,
+      "loss": 0.3808,
+      "step": 1913
+    },
+    {
+      "epoch": 2.753956834532374,
+      "grad_norm": 0.19109611424998146,
+      "learning_rate": 3.989953343938331e-05,
+      "loss": 0.3893,
+      "step": 1914
+    },
+    {
+      "epoch": 2.755395683453237,
+      "grad_norm": 0.20445928132697555,
+      "learning_rate": 3.985934695710566e-05,
+      "loss": 0.3828,
+      "step": 1915
+    },
+    {
+      "epoch": 2.756834532374101,
+      "grad_norm": 0.23828172071503892,
+      "learning_rate": 3.9819160616796873e-05,
+      "loss": 0.383,
+      "step": 1916
+    },
+    {
+      "epoch": 2.758273381294964,
+      "grad_norm": 0.2197593560082593,
+      "learning_rate": 3.977897445901922e-05,
+      "loss": 0.4037,
+      "step": 1917
+    },
+    {
+      "epoch": 2.7597122302158272,
+      "grad_norm": 0.1553271737678793,
+      "learning_rate": 3.9738788524334794e-05,
+      "loss": 0.3832,
+      "step": 1918
+    },
+    {
+      "epoch": 2.761151079136691,
+      "grad_norm": 0.16500774605695115,
+      "learning_rate": 3.969860285330549e-05,
+      "loss": 0.3746,
+      "step": 1919
+    },
+    {
+      "epoch": 2.762589928057554,
+      "grad_norm": 0.19886829293428393,
+      "learning_rate": 3.965841748649291e-05,
+      "loss": 0.3916,
+      "step": 1920
+    },
+    {
+      "epoch": 2.7640287769784173,
+      "grad_norm": 0.20717357076093512,
+      "learning_rate": 3.961823246445834e-05,
+      "loss": 0.3855,
+      "step": 1921
+    },
+    {
+      "epoch": 2.7654676258992805,
+      "grad_norm": 0.1790361182179321,
+      "learning_rate": 3.957804782776276e-05,
+      "loss": 0.3862,
+      "step": 1922
+    },
+    {
+      "epoch": 2.7669064748201437,
+      "grad_norm": 0.19813084858714128,
+      "learning_rate": 3.953786361696673e-05,
+      "loss": 0.3753,
+      "step": 1923
+    },
+    {
+      "epoch": 2.7683453237410074,
+      "grad_norm": 0.2068279472551279,
+      "learning_rate": 3.9497679872630366e-05,
+      "loss": 0.3838,
+      "step": 1924
+    },
+    {
+      "epoch": 2.7697841726618706,
+      "grad_norm": 0.1831003922469856,
+      "learning_rate": 3.945749663531334e-05,
+      "loss": 0.3796,
+      "step": 1925
+    },
+    {
+      "epoch": 2.771223021582734,
+      "grad_norm": 0.1818167482893324,
+      "learning_rate": 3.941731394557482e-05,
+      "loss": 0.387,
+      "step": 1926
+    },
+    {
+      "epoch": 2.772661870503597,
+      "grad_norm": 0.20082357390537708,
+      "learning_rate": 3.9377131843973394e-05,
+      "loss": 0.3874,
+      "step": 1927
+    },
+    {
+      "epoch": 2.7741007194244602,
+      "grad_norm": 0.20408518807963416,
+      "learning_rate": 3.933695037106705e-05,
+      "loss": 0.3863,
+      "step": 1928
+    },
+    {
+      "epoch": 2.775539568345324,
+      "grad_norm": 0.2066669950154263,
+      "learning_rate": 3.9296769567413177e-05,
+      "loss": 0.3892,
+      "step": 1929
+    },
+    {
+      "epoch": 2.776978417266187,
+      "grad_norm": 0.19174472735428247,
+      "learning_rate": 3.925658947356849e-05,
+      "loss": 0.3856,
+      "step": 1930
+    },
+    {
+      "epoch": 2.7784172661870503,
+      "grad_norm": 0.1987859128921994,
+      "learning_rate": 3.921641013008893e-05,
+      "loss": 0.3874,
+      "step": 1931
+    },
+    {
+      "epoch": 2.7798561151079135,
+      "grad_norm": 0.22291685372004874,
+      "learning_rate": 3.9176231577529734e-05,
+      "loss": 0.3887,
+      "step": 1932
+    },
+    {
+      "epoch": 2.7812949640287767,
+      "grad_norm": 0.22024270869600057,
+      "learning_rate": 3.913605385644535e-05,
+      "loss": 0.3986,
+      "step": 1933
+    },
+    {
+      "epoch": 2.7827338129496404,
+      "grad_norm": 0.1721122392744245,
+      "learning_rate": 3.909587700738933e-05,
+      "loss": 0.3811,
+      "step": 1934
+    },
+    {
+      "epoch": 2.7841726618705036,
+      "grad_norm": 0.21030490272757463,
+      "learning_rate": 3.9055701070914393e-05,
+      "loss": 0.3843,
+      "step": 1935
+    },
+    {
+      "epoch": 2.785611510791367,
+      "grad_norm": 0.20704087889653908,
+      "learning_rate": 3.90155260875723e-05,
+      "loss": 0.3802,
+      "step": 1936
+    },
+    {
+      "epoch": 2.7870503597122305,
+      "grad_norm": 0.17365781856632703,
+      "learning_rate": 3.8975352097913914e-05,
+      "loss": 0.3855,
+      "step": 1937
+    },
+    {
+      "epoch": 2.7884892086330937,
+      "grad_norm": 0.17770995901551442,
+      "learning_rate": 3.8935179142489016e-05,
+      "loss": 0.3833,
+      "step": 1938
+    },
+    {
+      "epoch": 2.789928057553957,
+      "grad_norm": 0.21779087822273127,
+      "learning_rate": 3.8895007261846404e-05,
+      "loss": 0.3904,
+      "step": 1939
+    },
+    {
+      "epoch": 2.79136690647482,
+      "grad_norm": 0.24277937539167369,
+      "learning_rate": 3.885483649653374e-05,
+      "loss": 0.3851,
+      "step": 1940
+    },
+    {
+      "epoch": 2.7928057553956833,
+      "grad_norm": 0.2341004136712482,
+      "learning_rate": 3.881466688709761e-05,
+      "loss": 0.3942,
+      "step": 1941
+    },
+    {
+      "epoch": 2.794244604316547,
+      "grad_norm": 0.18027649424590186,
+      "learning_rate": 3.877449847408342e-05,
+      "loss": 0.3781,
+      "step": 1942
+    },
+    {
+      "epoch": 2.79568345323741,
+      "grad_norm": 0.18717571945485895,
+      "learning_rate": 3.873433129803532e-05,
+      "loss": 0.3884,
+      "step": 1943
+    },
+    {
+      "epoch": 2.7971223021582734,
+      "grad_norm": 0.1858550625658424,
+      "learning_rate": 3.86941653994963e-05,
+      "loss": 0.3785,
+      "step": 1944
+    },
+    {
+      "epoch": 2.7985611510791366,
+      "grad_norm": 0.2026505538974637,
+      "learning_rate": 3.8654000819007974e-05,
+      "loss": 0.3797,
+      "step": 1945
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 0.19196092906122655,
+      "learning_rate": 3.8613837597110686e-05,
+      "loss": 0.3867,
+      "step": 1946
+    },
+    {
+      "epoch": 2.8014388489208635,
+      "grad_norm": 0.1670074465104803,
+      "learning_rate": 3.8573675774343356e-05,
+      "loss": 0.3961,
+      "step": 1947
+    },
+    {
+      "epoch": 2.8028776978417267,
+      "grad_norm": 0.17526592934078594,
+      "learning_rate": 3.853351539124355e-05,
+      "loss": 0.3798,
+      "step": 1948
+    },
+    {
+      "epoch": 2.80431654676259,
+      "grad_norm": 0.2017938976837373,
+      "learning_rate": 3.8493356488347345e-05,
+      "loss": 0.3852,
+      "step": 1949
+    },
+    {
+      "epoch": 2.805755395683453,
+      "grad_norm": 0.16141891543912168,
+      "learning_rate": 3.845319910618933e-05,
+      "loss": 0.3764,
+      "step": 1950
+    },
+    {
+      "epoch": 2.8071942446043163,
+      "grad_norm": 0.19869539544738132,
+      "learning_rate": 3.841304328530254e-05,
+      "loss": 0.3862,
+      "step": 1951
+    },
+    {
+      "epoch": 2.80863309352518,
+      "grad_norm": 0.1928193491860682,
+      "learning_rate": 3.83728890662185e-05,
+      "loss": 0.3918,
+      "step": 1952
+    },
+    {
+      "epoch": 2.810071942446043,
+      "grad_norm": 0.15205698302835358,
+      "learning_rate": 3.833273648946704e-05,
+      "loss": 0.386,
+      "step": 1953
+    },
+    {
+      "epoch": 2.8115107913669064,
+      "grad_norm": 0.16265067065993183,
+      "learning_rate": 3.829258559557635e-05,
+      "loss": 0.3862,
+      "step": 1954
+    },
+    {
+      "epoch": 2.81294964028777,
+      "grad_norm": 0.17480379497791992,
+      "learning_rate": 3.825243642507297e-05,
+      "loss": 0.383,
+      "step": 1955
+    },
+    {
+      "epoch": 2.814388489208633,
+      "grad_norm": 0.21289773047432814,
+      "learning_rate": 3.8212289018481666e-05,
+      "loss": 0.3857,
+      "step": 1956
+    },
+    {
+      "epoch": 2.8158273381294965,
+      "grad_norm": 0.17765426769618362,
+      "learning_rate": 3.817214341632539e-05,
+      "loss": 0.3904,
+      "step": 1957
+    },
+    {
+      "epoch": 2.8172661870503597,
+      "grad_norm": 0.16439226286825281,
+      "learning_rate": 3.813199965912533e-05,
+      "loss": 0.3848,
+      "step": 1958
+    },
+    {
+      "epoch": 2.818705035971223,
+      "grad_norm": 0.16251635630418615,
+      "learning_rate": 3.80918577874008e-05,
+      "loss": 0.3811,
+      "step": 1959
+    },
+    {
+      "epoch": 2.8201438848920866,
+      "grad_norm": 0.18480166519435853,
+      "learning_rate": 3.8051717841669196e-05,
+      "loss": 0.3879,
+      "step": 1960
+    },
+    {
+      "epoch": 2.8215827338129498,
+      "grad_norm": 0.1805350012707336,
+      "learning_rate": 3.801157986244595e-05,
+      "loss": 0.384,
+      "step": 1961
+    },
+    {
+      "epoch": 2.823021582733813,
+      "grad_norm": 0.21508103500088852,
+      "learning_rate": 3.7971443890244534e-05,
+      "loss": 0.3894,
+      "step": 1962
+    },
+    {
+      "epoch": 2.824460431654676,
+      "grad_norm": 0.1903700633733518,
+      "learning_rate": 3.7931309965576426e-05,
+      "loss": 0.3844,
+      "step": 1963
+    },
+    {
+      "epoch": 2.8258992805755394,
+      "grad_norm": 0.1894381997836098,
+      "learning_rate": 3.7891178128950975e-05,
+      "loss": 0.3955,
+      "step": 1964
+    },
+    {
+      "epoch": 2.827338129496403,
+      "grad_norm": 0.18150584566748393,
+      "learning_rate": 3.785104842087546e-05,
+      "loss": 0.3854,
+      "step": 1965
+    },
+    {
+      "epoch": 2.8287769784172663,
+      "grad_norm": 0.1891095814148094,
+      "learning_rate": 3.7810920881855016e-05,
+      "loss": 0.3872,
+      "step": 1966
+    },
+    {
+      "epoch": 2.8302158273381295,
+      "grad_norm": 0.23099345204149493,
+      "learning_rate": 3.777079555239255e-05,
+      "loss": 0.3761,
+      "step": 1967
+    },
+    {
+      "epoch": 2.8316546762589927,
+      "grad_norm": 0.20368419578794905,
+      "learning_rate": 3.77306724729888e-05,
+      "loss": 0.3952,
+      "step": 1968
+    },
+    {
+      "epoch": 2.833093525179856,
+      "grad_norm": 0.2013250517032747,
+      "learning_rate": 3.769055168414215e-05,
+      "loss": 0.3802,
+      "step": 1969
+    },
+    {
+      "epoch": 2.8345323741007196,
+      "grad_norm": 0.22040252433501764,
+      "learning_rate": 3.765043322634877e-05,
+      "loss": 0.385,
+      "step": 1970
+    },
+    {
+      "epoch": 2.8359712230215828,
+      "grad_norm": 0.22321698921110333,
+      "learning_rate": 3.761031714010239e-05,
+      "loss": 0.3928,
+      "step": 1971
+    },
+    {
+      "epoch": 2.837410071942446,
+      "grad_norm": 0.17808466214916618,
+      "learning_rate": 3.75702034658944e-05,
+      "loss": 0.394,
+      "step": 1972
+    },
+    {
+      "epoch": 2.838848920863309,
+      "grad_norm": 0.16906371940273435,
+      "learning_rate": 3.753009224421373e-05,
+      "loss": 0.3765,
+      "step": 1973
+    },
+    {
+      "epoch": 2.8402877697841724,
+      "grad_norm": 0.18308010487641513,
+      "learning_rate": 3.748998351554684e-05,
+      "loss": 0.3903,
+      "step": 1974
+    },
+    {
+      "epoch": 2.841726618705036,
+      "grad_norm": 0.2225548415238196,
+      "learning_rate": 3.74498773203777e-05,
+      "loss": 0.3846,
+      "step": 1975
+    },
+    {
+      "epoch": 2.8431654676258993,
+      "grad_norm": 0.16580518917297782,
+      "learning_rate": 3.7409773699187664e-05,
+      "loss": 0.3866,
+      "step": 1976
+    },
+    {
+      "epoch": 2.8446043165467625,
+      "grad_norm": 0.18060687743856188,
+      "learning_rate": 3.736967269245555e-05,
+      "loss": 0.3752,
+      "step": 1977
+    },
+    {
+      "epoch": 2.846043165467626,
+      "grad_norm": 0.2442856647276992,
+      "learning_rate": 3.732957434065751e-05,
+      "loss": 0.3825,
+      "step": 1978
+    },
+    {
+      "epoch": 2.8474820143884894,
+      "grad_norm": 0.17377603588060617,
+      "learning_rate": 3.728947868426701e-05,
+      "loss": 0.3918,
+      "step": 1979
+    },
+    {
+      "epoch": 2.8489208633093526,
+      "grad_norm": 0.17366991770930496,
+      "learning_rate": 3.724938576375479e-05,
+      "loss": 0.3872,
+      "step": 1980
+    },
+    {
+      "epoch": 2.850359712230216,
+      "grad_norm": 0.16520872352908733,
+      "learning_rate": 3.7209295619588856e-05,
+      "loss": 0.3831,
+      "step": 1981
+    },
+    {
+      "epoch": 2.851798561151079,
+      "grad_norm": 0.17358900130450544,
+      "learning_rate": 3.7169208292234395e-05,
+      "loss": 0.3847,
+      "step": 1982
+    },
+    {
+      "epoch": 2.8532374100719426,
+      "grad_norm": 0.16055853887610544,
+      "learning_rate": 3.7129123822153746e-05,
+      "loss": 0.3761,
+      "step": 1983
+    },
+    {
+      "epoch": 2.854676258992806,
+      "grad_norm": 0.16978866572029805,
+      "learning_rate": 3.708904224980636e-05,
+      "loss": 0.3797,
+      "step": 1984
+    },
+    {
+      "epoch": 2.856115107913669,
+      "grad_norm": 0.16168212577426672,
+      "learning_rate": 3.704896361564881e-05,
+      "loss": 0.3857,
+      "step": 1985
+    },
+    {
+      "epoch": 2.8575539568345323,
+      "grad_norm": 0.21795965698751302,
+      "learning_rate": 3.700888796013462e-05,
+      "loss": 0.387,
+      "step": 1986
+    },
+    {
+      "epoch": 2.8589928057553955,
+      "grad_norm": 0.2072773046432437,
+      "learning_rate": 3.696881532371439e-05,
+      "loss": 0.3751,
+      "step": 1987
+    },
+    {
+      "epoch": 2.860431654676259,
+      "grad_norm": 0.17241642786348224,
+      "learning_rate": 3.692874574683562e-05,
+      "loss": 0.3925,
+      "step": 1988
+    },
+    {
+      "epoch": 2.8618705035971224,
+      "grad_norm": 0.24022281114246952,
+      "learning_rate": 3.688867926994274e-05,
+      "loss": 0.3833,
+      "step": 1989
+    },
+    {
+      "epoch": 2.8633093525179856,
+      "grad_norm": 0.2016307515118801,
+      "learning_rate": 3.684861593347705e-05,
+      "loss": 0.3823,
+      "step": 1990
+    },
+    {
+      "epoch": 2.864748201438849,
+      "grad_norm": 0.23297438393509676,
+      "learning_rate": 3.6808555777876673e-05,
+      "loss": 0.3765,
+      "step": 1991
+    },
+    {
+      "epoch": 2.866187050359712,
+      "grad_norm": 0.24468893457129012,
+      "learning_rate": 3.676849884357655e-05,
+      "loss": 0.3758,
+      "step": 1992
+    },
+    {
+      "epoch": 2.8676258992805757,
+      "grad_norm": 0.19354597620274588,
+      "learning_rate": 3.672844517100833e-05,
+      "loss": 0.3831,
+      "step": 1993
+    },
+    {
+      "epoch": 2.869064748201439,
+      "grad_norm": 0.20311587736464948,
+      "learning_rate": 3.66883948006004e-05,
+      "loss": 0.3817,
+      "step": 1994
+    },
+    {
+      "epoch": 2.870503597122302,
+      "grad_norm": 0.21481338071322306,
+      "learning_rate": 3.664834777277777e-05,
+      "loss": 0.3875,
+      "step": 1995
+    },
+    {
+      "epoch": 2.8719424460431657,
+      "grad_norm": 0.22454309712498646,
+      "learning_rate": 3.6608304127962166e-05,
+      "loss": 0.3809,
+      "step": 1996
+    },
+    {
+      "epoch": 2.873381294964029,
+      "grad_norm": 0.21854088471309374,
+      "learning_rate": 3.656826390657179e-05,
+      "loss": 0.3895,
+      "step": 1997
+    },
+    {
+      "epoch": 2.874820143884892,
+      "grad_norm": 0.18912763371226035,
+      "learning_rate": 3.6528227149021455e-05,
+      "loss": 0.3736,
+      "step": 1998
+    },
+    {
+      "epoch": 2.8762589928057554,
+      "grad_norm": 0.24831039293665552,
+      "learning_rate": 3.648819389572248e-05,
+      "loss": 0.3767,
+      "step": 1999
+    },
+    {
+      "epoch": 2.8776978417266186,
+      "grad_norm": 0.21648203780760272,
+      "learning_rate": 3.644816418708261e-05,
+      "loss": 0.3798,
+      "step": 2000
+    },
+    {
+      "epoch": 2.8791366906474822,
+      "grad_norm": 0.20509193910389217,
+      "learning_rate": 3.6408138063506057e-05,
+      "loss": 0.389,
+      "step": 2001
+    },
+    {
+      "epoch": 2.8805755395683454,
+      "grad_norm": 0.21791850596163106,
+      "learning_rate": 3.636811556539335e-05,
+      "loss": 0.3792,
+      "step": 2002
+    },
+    {
+      "epoch": 2.8820143884892087,
+      "grad_norm": 0.2554805546344398,
+      "learning_rate": 3.6328096733141423e-05,
+      "loss": 0.3803,
+      "step": 2003
+    },
+    {
+      "epoch": 2.883453237410072,
+      "grad_norm": 0.19046409853189405,
+      "learning_rate": 3.6288081607143496e-05,
+      "loss": 0.3833,
+      "step": 2004
+    },
+    {
+      "epoch": 2.884892086330935,
+      "grad_norm": 0.23598509312791277,
+      "learning_rate": 3.6248070227789034e-05,
+      "loss": 0.3741,
+      "step": 2005
+    },
+    {
+      "epoch": 2.8863309352517987,
+      "grad_norm": 0.1999608065119113,
+      "learning_rate": 3.620806263546369e-05,
+      "loss": 0.3837,
+      "step": 2006
+    },
+    {
+      "epoch": 2.887769784172662,
+      "grad_norm": 0.17771440852964768,
+      "learning_rate": 3.6168058870549355e-05,
+      "loss": 0.3804,
+      "step": 2007
+    },
+    {
+      "epoch": 2.889208633093525,
+      "grad_norm": 0.24204378684346625,
+      "learning_rate": 3.612805897342405e-05,
+      "loss": 0.3786,
+      "step": 2008
+    },
+    {
+      "epoch": 2.8906474820143884,
+      "grad_norm": 0.1980316580190834,
+      "learning_rate": 3.608806298446182e-05,
+      "loss": 0.3836,
+      "step": 2009
+    },
+    {
+      "epoch": 2.8920863309352516,
+      "grad_norm": 0.17968345202327035,
+      "learning_rate": 3.604807094403286e-05,
+      "loss": 0.3846,
+      "step": 2010
+    },
+    {
+      "epoch": 2.8935251798561152,
+      "grad_norm": 0.1877683648006211,
+      "learning_rate": 3.6008082892503325e-05,
+      "loss": 0.3847,
+      "step": 2011
+    },
+    {
+      "epoch": 2.8949640287769784,
+      "grad_norm": 0.20175307646308607,
+      "learning_rate": 3.596809887023534e-05,
+      "loss": 0.386,
+      "step": 2012
+    },
+    {
+      "epoch": 2.8964028776978417,
+      "grad_norm": 0.16055923442945932,
+      "learning_rate": 3.5928118917587e-05,
+      "loss": 0.3871,
+      "step": 2013
+    },
+    {
+      "epoch": 2.897841726618705,
+      "grad_norm": 0.19637652624877666,
+      "learning_rate": 3.588814307491227e-05,
+      "loss": 0.3857,
+      "step": 2014
+    },
+    {
+      "epoch": 2.899280575539568,
+      "grad_norm": 0.19818332483671994,
+      "learning_rate": 3.584817138256096e-05,
+      "loss": 0.3836,
+      "step": 2015
+    },
+    {
+      "epoch": 2.9007194244604317,
+      "grad_norm": 0.1451205820964952,
+      "learning_rate": 3.580820388087869e-05,
+      "loss": 0.3933,
+      "step": 2016
+    },
+    {
+      "epoch": 2.902158273381295,
+      "grad_norm": 0.19786506703629175,
+      "learning_rate": 3.5768240610206855e-05,
+      "loss": 0.393,
+      "step": 2017
+    },
+    {
+      "epoch": 2.903597122302158,
+      "grad_norm": 0.18083530222742925,
+      "learning_rate": 3.572828161088262e-05,
+      "loss": 0.3824,
+      "step": 2018
+    },
+    {
+      "epoch": 2.905035971223022,
+      "grad_norm": 0.181636151918574,
+      "learning_rate": 3.568832692323876e-05,
+      "loss": 0.3828,
+      "step": 2019
+    },
+    {
+      "epoch": 2.906474820143885,
+      "grad_norm": 0.23088256110171856,
+      "learning_rate": 3.564837658760376e-05,
+      "loss": 0.3884,
+      "step": 2020
+    },
+    {
+      "epoch": 2.9079136690647482,
+      "grad_norm": 0.19929833520067272,
+      "learning_rate": 3.560843064430168e-05,
+      "loss": 0.384,
+      "step": 2021
+    },
+    {
+      "epoch": 2.9093525179856115,
+      "grad_norm": 0.19123040805220842,
+      "learning_rate": 3.556848913365218e-05,
+      "loss": 0.3761,
+      "step": 2022
+    },
+    {
+      "epoch": 2.9107913669064747,
+      "grad_norm": 0.22589457826263576,
+      "learning_rate": 3.552855209597039e-05,
+      "loss": 0.3931,
+      "step": 2023
+    },
+    {
+      "epoch": 2.9122302158273383,
+      "grad_norm": 0.21781504032045054,
+      "learning_rate": 3.548861957156698e-05,
+      "loss": 0.3865,
+      "step": 2024
+    },
+    {
+      "epoch": 2.9136690647482015,
+      "grad_norm": 0.1850197179194681,
+      "learning_rate": 3.544869160074806e-05,
+      "loss": 0.3864,
+      "step": 2025
+    },
+    {
+      "epoch": 2.9151079136690647,
+      "grad_norm": 0.24238459049549935,
+      "learning_rate": 3.5408768223815105e-05,
+      "loss": 0.3801,
+      "step": 2026
+    },
+    {
+      "epoch": 2.916546762589928,
+      "grad_norm": 0.21673031475617144,
+      "learning_rate": 3.536884948106498e-05,
+      "loss": 0.3855,
+      "step": 2027
+    },
+    {
+      "epoch": 2.917985611510791,
+      "grad_norm": 0.5998072090981956,
+      "learning_rate": 3.532893541278986e-05,
+      "loss": 0.3929,
+      "step": 2028
+    },
+    {
+      "epoch": 2.919424460431655,
+      "grad_norm": 0.2621668968319146,
+      "learning_rate": 3.528902605927722e-05,
+      "loss": 0.3911,
+      "step": 2029
+    },
+    {
+      "epoch": 2.920863309352518,
+      "grad_norm": 0.22450894990011058,
+      "learning_rate": 3.524912146080978e-05,
+      "loss": 0.3798,
+      "step": 2030
+    },
+    {
+      "epoch": 2.9223021582733812,
+      "grad_norm": 0.20134239652713007,
+      "learning_rate": 3.5209221657665436e-05,
+      "loss": 0.3831,
+      "step": 2031
+    },
+    {
+      "epoch": 2.9237410071942445,
+      "grad_norm": 0.186279994472004,
+      "learning_rate": 3.516932669011723e-05,
+      "loss": 0.3839,
+      "step": 2032
+    },
+    {
+      "epoch": 2.9251798561151077,
+      "grad_norm": 0.22831410659162382,
+      "learning_rate": 3.512943659843337e-05,
+      "loss": 0.3809,
+      "step": 2033
+    },
+    {
+      "epoch": 2.9266187050359713,
+      "grad_norm": 0.20425913332091689,
+      "learning_rate": 3.508955142287714e-05,
+      "loss": 0.3891,
+      "step": 2034
+    },
+    {
+      "epoch": 2.9280575539568345,
+      "grad_norm": 0.20480759469192697,
+      "learning_rate": 3.50496712037068e-05,
+      "loss": 0.3879,
+      "step": 2035
+    },
+    {
+      "epoch": 2.9294964028776977,
+      "grad_norm": 0.2038244372869383,
+      "learning_rate": 3.5009795981175676e-05,
+      "loss": 0.3919,
+      "step": 2036
+    },
+    {
+      "epoch": 2.9309352517985614,
+      "grad_norm": 0.22707319489735453,
+      "learning_rate": 3.496992579553203e-05,
+      "loss": 0.3924,
+      "step": 2037
+    },
+    {
+      "epoch": 2.9323741007194246,
+      "grad_norm": 0.245562565118133,
+      "learning_rate": 3.4930060687019015e-05,
+      "loss": 0.3888,
+      "step": 2038
+    },
+    {
+      "epoch": 2.933812949640288,
+      "grad_norm": 0.2441632626394093,
+      "learning_rate": 3.489020069587467e-05,
+      "loss": 0.3855,
+      "step": 2039
+    },
+    {
+      "epoch": 2.935251798561151,
+      "grad_norm": 0.17745251314180563,
+      "learning_rate": 3.485034586233192e-05,
+      "loss": 0.39,
+      "step": 2040
+    },
+    {
+      "epoch": 2.9366906474820142,
+      "grad_norm": 0.22487949386773248,
+      "learning_rate": 3.4810496226618404e-05,
+      "loss": 0.3975,
+      "step": 2041
+    },
+    {
+      "epoch": 2.938129496402878,
+      "grad_norm": 0.20002018533834687,
+      "learning_rate": 3.477065182895656e-05,
+      "loss": 0.3837,
+      "step": 2042
+    },
+    {
+      "epoch": 2.939568345323741,
+      "grad_norm": 0.230180258512151,
+      "learning_rate": 3.473081270956352e-05,
+      "loss": 0.3877,
+      "step": 2043
+    },
+    {
+      "epoch": 2.9410071942446043,
+      "grad_norm": 0.1756588002638614,
+      "learning_rate": 3.469097890865113e-05,
+      "loss": 0.3783,
+      "step": 2044
+    },
+    {
+      "epoch": 2.9424460431654675,
+      "grad_norm": 0.26835782252608176,
+      "learning_rate": 3.465115046642581e-05,
+      "loss": 0.3815,
+      "step": 2045
+    },
+    {
+      "epoch": 2.9438848920863308,
+      "grad_norm": 0.23997963562866945,
+      "learning_rate": 3.461132742308859e-05,
+      "loss": 0.3771,
+      "step": 2046
+    },
+    {
+      "epoch": 2.9453237410071944,
+      "grad_norm": 0.20888633375860216,
+      "learning_rate": 3.45715098188351e-05,
+      "loss": 0.3899,
+      "step": 2047
+    },
+    {
+      "epoch": 2.9467625899280576,
+      "grad_norm": 0.20176189346580645,
+      "learning_rate": 3.453169769385541e-05,
+      "loss": 0.3809,
+      "step": 2048
+    },
+    {
+      "epoch": 2.948201438848921,
+      "grad_norm": 0.17608840392320604,
+      "learning_rate": 3.449189108833409e-05,
+      "loss": 0.3832,
+      "step": 2049
+    },
+    {
+      "epoch": 2.949640287769784,
+      "grad_norm": 0.1854169732719609,
+      "learning_rate": 3.445209004245012e-05,
+      "loss": 0.3821,
+      "step": 2050
+    },
+    {
+      "epoch": 2.9510791366906473,
+      "grad_norm": 0.1619286806631231,
+      "learning_rate": 3.441229459637693e-05,
+      "loss": 0.3879,
+      "step": 2051
+    },
+    {
+      "epoch": 2.952517985611511,
+      "grad_norm": 0.19130379361566077,
+      "learning_rate": 3.4372504790282215e-05,
+      "loss": 0.3875,
+      "step": 2052
+    },
+    {
+      "epoch": 2.953956834532374,
+      "grad_norm": 0.19380190248276075,
+      "learning_rate": 3.4332720664328034e-05,
+      "loss": 0.3839,
+      "step": 2053
+    },
+    {
+      "epoch": 2.9553956834532373,
+      "grad_norm": 0.22132673629007027,
+      "learning_rate": 3.4292942258670675e-05,
+      "loss": 0.387,
+      "step": 2054
+    },
+    {
+      "epoch": 2.956834532374101,
+      "grad_norm": 0.18534009294543696,
+      "learning_rate": 3.425316961346069e-05,
+      "loss": 0.3953,
+      "step": 2055
+    },
+    {
+      "epoch": 2.9582733812949638,
+      "grad_norm": 0.17548642072351006,
+      "learning_rate": 3.42134027688428e-05,
+      "loss": 0.3848,
+      "step": 2056
+    },
+    {
+      "epoch": 2.9597122302158274,
+      "grad_norm": 0.20392081088111974,
+      "learning_rate": 3.417364176495585e-05,
+      "loss": 0.3897,
+      "step": 2057
+    },
+    {
+      "epoch": 2.9611510791366906,
+      "grad_norm": 0.21337179838526357,
+      "learning_rate": 3.4133886641932834e-05,
+      "loss": 0.3823,
+      "step": 2058
+    },
+    {
+      "epoch": 2.962589928057554,
+      "grad_norm": 0.17007119429337325,
+      "learning_rate": 3.409413743990076e-05,
+      "loss": 0.3929,
+      "step": 2059
+    },
+    {
+      "epoch": 2.9640287769784175,
+      "grad_norm": 0.16472519586281026,
+      "learning_rate": 3.4054394198980705e-05,
+      "loss": 0.3787,
+      "step": 2060
+    },
+    {
+      "epoch": 2.9654676258992807,
+      "grad_norm": 0.1708056234057997,
+      "learning_rate": 3.401465695928768e-05,
+      "loss": 0.3772,
+      "step": 2061
+    },
+    {
+      "epoch": 2.966906474820144,
+      "grad_norm": 0.15953734302553965,
+      "learning_rate": 3.3974925760930694e-05,
+      "loss": 0.3886,
+      "step": 2062
+    },
+    {
+      "epoch": 2.968345323741007,
+      "grad_norm": 0.15527382940345988,
+      "learning_rate": 3.393520064401264e-05,
+      "loss": 0.3803,
+      "step": 2063
+    },
+    {
+      "epoch": 2.9697841726618703,
+      "grad_norm": 0.17300060715215387,
+      "learning_rate": 3.3895481648630234e-05,
+      "loss": 0.3851,
+      "step": 2064
+    },
+    {
+      "epoch": 2.971223021582734,
+      "grad_norm": 0.16797967283810863,
+      "learning_rate": 3.385576881487405e-05,
+      "loss": 0.3794,
+      "step": 2065
+    },
+    {
+      "epoch": 2.972661870503597,
+      "grad_norm": 0.1754748771386963,
+      "learning_rate": 3.381606218282846e-05,
+      "loss": 0.3903,
+      "step": 2066
+    },
+    {
+      "epoch": 2.9741007194244604,
+      "grad_norm": 0.14154374320062357,
+      "learning_rate": 3.377636179257153e-05,
+      "loss": 0.3847,
+      "step": 2067
+    },
+    {
+      "epoch": 2.9755395683453236,
+      "grad_norm": 0.19033944623921709,
+      "learning_rate": 3.373666768417505e-05,
+      "loss": 0.3765,
+      "step": 2068
+    },
+    {
+      "epoch": 2.976978417266187,
+      "grad_norm": 0.20493110424364452,
+      "learning_rate": 3.3696979897704466e-05,
+      "loss": 0.3899,
+      "step": 2069
+    },
+    {
+      "epoch": 2.9784172661870505,
+      "grad_norm": 0.17833206085806985,
+      "learning_rate": 3.3657298473218864e-05,
+      "loss": 0.3952,
+      "step": 2070
+    },
+    {
+      "epoch": 2.9798561151079137,
+      "grad_norm": 0.1686821102041315,
+      "learning_rate": 3.361762345077087e-05,
+      "loss": 0.4012,
+      "step": 2071
+    },
+    {
+      "epoch": 2.981294964028777,
+      "grad_norm": 0.2436521168045105,
+      "learning_rate": 3.3577954870406656e-05,
+      "loss": 0.3909,
+      "step": 2072
+    },
+    {
+      "epoch": 2.98273381294964,
+      "grad_norm": 0.22817825742264522,
+      "learning_rate": 3.3538292772165936e-05,
+      "loss": 0.379,
+      "step": 2073
+    },
+    {
+      "epoch": 2.9841726618705033,
+      "grad_norm": 0.19740041368304065,
+      "learning_rate": 3.3498637196081825e-05,
+      "loss": 0.3829,
+      "step": 2074
+    },
+    {
+      "epoch": 2.985611510791367,
+      "grad_norm": 0.19622604138848831,
+      "learning_rate": 3.345898818218089e-05,
+      "loss": 0.3927,
+      "step": 2075
+    },
+    {
+      "epoch": 2.98705035971223,
+      "grad_norm": 0.150841854712471,
+      "learning_rate": 3.341934577048304e-05,
+      "loss": 0.3838,
+      "step": 2076
+    },
+    {
+      "epoch": 2.9884892086330934,
+      "grad_norm": 0.1678703869895141,
+      "learning_rate": 3.337971000100157e-05,
+      "loss": 0.3818,
+      "step": 2077
+    },
+    {
+      "epoch": 2.989928057553957,
+      "grad_norm": 0.21269206206655902,
+      "learning_rate": 3.334008091374303e-05,
+      "loss": 0.3919,
+      "step": 2078
+    },
+    {
+      "epoch": 2.9913669064748203,
+      "grad_norm": 0.18124149976349824,
+      "learning_rate": 3.3300458548707214e-05,
+      "loss": 0.386,
+      "step": 2079
+    },
+    {
+      "epoch": 2.9928057553956835,
+      "grad_norm": 0.1672719747664083,
+      "learning_rate": 3.326084294588721e-05,
+      "loss": 0.389,
+      "step": 2080
+    },
+    {
+      "epoch": 2.9942446043165467,
+      "grad_norm": 0.2155189730344991,
+      "learning_rate": 3.322123414526917e-05,
+      "loss": 0.3852,
+      "step": 2081
+    },
+    {
+      "epoch": 2.99568345323741,
+      "grad_norm": 0.20505929598971243,
+      "learning_rate": 3.3181632186832485e-05,
+      "loss": 0.3914,
+      "step": 2082
+    },
+    {
+      "epoch": 2.9971223021582736,
+      "grad_norm": 0.1771213290261762,
+      "learning_rate": 3.3142037110549546e-05,
+      "loss": 0.3786,
+      "step": 2083
+    },
+    {
+      "epoch": 2.998561151079137,
+      "grad_norm": 0.20191658002270133,
+      "learning_rate": 3.310244895638587e-05,
+      "loss": 0.3809,
+      "step": 2084
+    },
+    {
+      "epoch": 3.0,
+      "grad_norm": 0.20225356044957576,
+      "learning_rate": 3.306286776429995e-05,
+      "loss": 0.3747,
+      "step": 2085
+    },
+    {
+      "epoch": 3.001438848920863,
+      "grad_norm": 0.20569754631248985,
+      "learning_rate": 3.302329357424326e-05,
+      "loss": 0.3707,
+      "step": 2086
+    },
+    {
+      "epoch": 3.0028776978417264,
+      "grad_norm": 0.20114942842678957,
+      "learning_rate": 3.2983726426160204e-05,
+      "loss": 0.364,
+      "step": 2087
+    },
+    {
+      "epoch": 3.00431654676259,
+      "grad_norm": 0.18723446351048248,
+      "learning_rate": 3.2944166359988083e-05,
+      "loss": 0.3681,
+      "step": 2088
+    },
+    {
+      "epoch": 3.0057553956834533,
+      "grad_norm": 0.17220903321630057,
+      "learning_rate": 3.290461341565707e-05,
+      "loss": 0.359,
+      "step": 2089
+    },
+    {
+      "epoch": 3.0071942446043165,
+      "grad_norm": 0.23430100610978008,
+      "learning_rate": 3.286506763309009e-05,
+      "loss": 0.3547,
+      "step": 2090
+    },
+    {
+      "epoch": 3.0086330935251797,
+      "grad_norm": 0.19688002346513944,
+      "learning_rate": 3.2825529052202904e-05,
+      "loss": 0.3652,
+      "step": 2091
+    },
+    {
+      "epoch": 3.0100719424460434,
+      "grad_norm": 0.20321597432510047,
+      "learning_rate": 3.278599771290397e-05,
+      "loss": 0.3519,
+      "step": 2092
+    },
+    {
+      "epoch": 3.0115107913669066,
+      "grad_norm": 0.19081182863087742,
+      "learning_rate": 3.274647365509445e-05,
+      "loss": 0.3637,
+      "step": 2093
+    },
+    {
+      "epoch": 3.01294964028777,
+      "grad_norm": 0.2328943681815441,
+      "learning_rate": 3.2706956918668126e-05,
+      "loss": 0.3582,
+      "step": 2094
+    },
+    {
+      "epoch": 3.014388489208633,
+      "grad_norm": 0.18125767772085483,
+      "learning_rate": 3.266744754351144e-05,
+      "loss": 0.3603,
+      "step": 2095
+    },
+    {
+      "epoch": 3.015827338129496,
+      "grad_norm": 0.20943968943457905,
+      "learning_rate": 3.262794556950338e-05,
+      "loss": 0.3665,
+      "step": 2096
+    },
+    {
+      "epoch": 3.01726618705036,
+      "grad_norm": 0.2016966526436745,
+      "learning_rate": 3.2588451036515435e-05,
+      "loss": 0.362,
+      "step": 2097
+    },
+    {
+      "epoch": 3.018705035971223,
+      "grad_norm": 0.18929307021417738,
+      "learning_rate": 3.2548963984411623e-05,
+      "loss": 0.3628,
+      "step": 2098
+    },
+    {
+      "epoch": 3.0201438848920863,
+      "grad_norm": 0.2079651623363538,
+      "learning_rate": 3.2509484453048413e-05,
+      "loss": 0.3718,
+      "step": 2099
+    },
+    {
+      "epoch": 3.0215827338129495,
+      "grad_norm": 0.17061447600345955,
+      "learning_rate": 3.247001248227465e-05,
+      "loss": 0.3559,
+      "step": 2100
+    },
+    {
+      "epoch": 3.023021582733813,
+      "grad_norm": 0.18547696640719988,
+      "learning_rate": 3.2430548111931574e-05,
+      "loss": 0.3596,
+      "step": 2101
+    },
+    {
+      "epoch": 3.0244604316546764,
+      "grad_norm": 0.1653250126196465,
+      "learning_rate": 3.239109138185275e-05,
+      "loss": 0.3607,
+      "step": 2102
+    },
+    {
+      "epoch": 3.0258992805755396,
+      "grad_norm": 0.18282052325980813,
+      "learning_rate": 3.2351642331864024e-05,
+      "loss": 0.3606,
+      "step": 2103
+    },
+    {
+      "epoch": 3.027338129496403,
+      "grad_norm": 0.17101471749049846,
+      "learning_rate": 3.2312201001783473e-05,
+      "loss": 0.3638,
+      "step": 2104
+    },
+    {
+      "epoch": 3.028776978417266,
+      "grad_norm": 0.16413900278688257,
+      "learning_rate": 3.2272767431421416e-05,
+      "loss": 0.3632,
+      "step": 2105
+    },
+    {
+      "epoch": 3.0302158273381297,
+      "grad_norm": 0.16625703759551594,
+      "learning_rate": 3.2233341660580335e-05,
+      "loss": 0.3532,
+      "step": 2106
+    },
+    {
+      "epoch": 3.031654676258993,
+      "grad_norm": 0.2144615795712448,
+      "learning_rate": 3.219392372905482e-05,
+      "loss": 0.3674,
+      "step": 2107
+    },
+    {
+      "epoch": 3.033093525179856,
+      "grad_norm": 0.14936298512309384,
+      "learning_rate": 3.215451367663156e-05,
+      "loss": 0.3636,
+      "step": 2108
+    },
+    {
+      "epoch": 3.0345323741007193,
+      "grad_norm": 0.20451775844635806,
+      "learning_rate": 3.211511154308927e-05,
+      "loss": 0.3578,
+      "step": 2109
+    },
+    {
+      "epoch": 3.0359712230215825,
+      "grad_norm": 0.16421410694291483,
+      "learning_rate": 3.207571736819873e-05,
+      "loss": 0.3619,
+      "step": 2110
+    },
+    {
+      "epoch": 3.037410071942446,
+      "grad_norm": 0.16014427605999473,
+      "learning_rate": 3.203633119172262e-05,
+      "loss": 0.3672,
+      "step": 2111
+    },
+    {
+      "epoch": 3.0388489208633094,
+      "grad_norm": 0.16802997831800656,
+      "learning_rate": 3.1996953053415575e-05,
+      "loss": 0.3634,
+      "step": 2112
+    },
+    {
+      "epoch": 3.0402877697841726,
+      "grad_norm": 0.19318307562131506,
+      "learning_rate": 3.1957582993024135e-05,
+      "loss": 0.3665,
+      "step": 2113
+    },
+    {
+      "epoch": 3.041726618705036,
+      "grad_norm": 0.17388976938204928,
+      "learning_rate": 3.191822105028665e-05,
+      "loss": 0.3561,
+      "step": 2114
+    },
+    {
+      "epoch": 3.0431654676258995,
+      "grad_norm": 0.1645764090504961,
+      "learning_rate": 3.1878867264933305e-05,
+      "loss": 0.3542,
+      "step": 2115
+    },
+    {
+      "epoch": 3.0446043165467627,
+      "grad_norm": 0.18244817741104855,
+      "learning_rate": 3.1839521676686026e-05,
+      "loss": 0.3561,
+      "step": 2116
+    },
+    {
+      "epoch": 3.046043165467626,
+      "grad_norm": 0.16508563331234616,
+      "learning_rate": 3.1800184325258494e-05,
+      "loss": 0.3685,
+      "step": 2117
+    },
+    {
+      "epoch": 3.047482014388489,
+      "grad_norm": 0.16473841098624484,
+      "learning_rate": 3.176085525035607e-05,
+      "loss": 0.362,
+      "step": 2118
+    },
+    {
+      "epoch": 3.0489208633093523,
+      "grad_norm": 0.15113713787417427,
+      "learning_rate": 3.172153449167574e-05,
+      "loss": 0.3584,
+      "step": 2119
+    },
+    {
+      "epoch": 3.050359712230216,
+      "grad_norm": 0.19044107514465825,
+      "learning_rate": 3.1682222088906096e-05,
+      "loss": 0.3588,
+      "step": 2120
+    },
+    {
+      "epoch": 3.051798561151079,
+      "grad_norm": 0.1478330796372751,
+      "learning_rate": 3.1642918081727327e-05,
+      "loss": 0.3642,
+      "step": 2121
+    },
+    {
+      "epoch": 3.0532374100719424,
+      "grad_norm": 0.19689036313293712,
+      "learning_rate": 3.1603622509811144e-05,
+      "loss": 0.3609,
+      "step": 2122
+    },
+    {
+      "epoch": 3.0546762589928056,
+      "grad_norm": 0.20486526790795795,
+      "learning_rate": 3.156433541282069e-05,
+      "loss": 0.3623,
+      "step": 2123
+    },
+    {
+      "epoch": 3.0561151079136692,
+      "grad_norm": 0.15617141094534115,
+      "learning_rate": 3.152505683041062e-05,
+      "loss": 0.3593,
+      "step": 2124
+    },
+    {
+      "epoch": 3.0575539568345325,
+      "grad_norm": 0.2027148810745762,
+      "learning_rate": 3.1485786802226976e-05,
+      "loss": 0.3646,
+      "step": 2125
+    },
+    {
+      "epoch": 3.0589928057553957,
+      "grad_norm": 0.21447667121894384,
+      "learning_rate": 3.1446525367907134e-05,
+      "loss": 0.3602,
+      "step": 2126
+    },
+    {
+      "epoch": 3.060431654676259,
+      "grad_norm": 0.15381484589567843,
+      "learning_rate": 3.1407272567079834e-05,
+      "loss": 0.3639,
+      "step": 2127
+    },
+    {
+      "epoch": 3.061870503597122,
+      "grad_norm": 0.1872412094336588,
+      "learning_rate": 3.136802843936509e-05,
+      "loss": 0.3581,
+      "step": 2128
+    },
+    {
+      "epoch": 3.0633093525179858,
+      "grad_norm": 0.2037870768861435,
+      "learning_rate": 3.132879302437416e-05,
+      "loss": 0.3593,
+      "step": 2129
+    },
+    {
+      "epoch": 3.064748201438849,
+      "grad_norm": 0.16557314944365134,
+      "learning_rate": 3.128956636170949e-05,
+      "loss": 0.3676,
+      "step": 2130
+    },
+    {
+      "epoch": 3.066187050359712,
+      "grad_norm": 0.21739010073156842,
+      "learning_rate": 3.125034849096471e-05,
+      "loss": 0.3616,
+      "step": 2131
+    },
+    {
+      "epoch": 3.0676258992805754,
+      "grad_norm": 0.23829079888615076,
+      "learning_rate": 3.1211139451724605e-05,
+      "loss": 0.3666,
+      "step": 2132
+    },
+    {
+      "epoch": 3.069064748201439,
+      "grad_norm": 0.19003741526013926,
+      "learning_rate": 3.1171939283564986e-05,
+      "loss": 0.3624,
+      "step": 2133
+    },
+    {
+      "epoch": 3.0705035971223023,
+      "grad_norm": 0.19280640697509915,
+      "learning_rate": 3.113274802605276e-05,
+      "loss": 0.3649,
+      "step": 2134
+    },
+    {
+      "epoch": 3.0719424460431655,
+      "grad_norm": 0.17465491343281253,
+      "learning_rate": 3.109356571874579e-05,
+      "loss": 0.3597,
+      "step": 2135
+    },
+    {
+      "epoch": 3.0733812949640287,
+      "grad_norm": 0.1575551077356398,
+      "learning_rate": 3.105439240119296e-05,
+      "loss": 0.3601,
+      "step": 2136
+    },
+    {
+      "epoch": 3.074820143884892,
+      "grad_norm": 0.1582159815146858,
+      "learning_rate": 3.101522811293405e-05,
+      "loss": 0.354,
+      "step": 2137
+    },
+    {
+      "epoch": 3.0762589928057555,
+      "grad_norm": 0.1532665643566935,
+      "learning_rate": 3.0976072893499724e-05,
+      "loss": 0.3493,
+      "step": 2138
+    },
+    {
+      "epoch": 3.0776978417266188,
+      "grad_norm": 0.1268319703596362,
+      "learning_rate": 3.093692678241151e-05,
+      "loss": 0.3533,
+      "step": 2139
+    },
+    {
+      "epoch": 3.079136690647482,
+      "grad_norm": 0.17721432868865453,
+      "learning_rate": 3.0897789819181715e-05,
+      "loss": 0.3524,
+      "step": 2140
+    },
+    {
+      "epoch": 3.080575539568345,
+      "grad_norm": 0.17209923810182967,
+      "learning_rate": 3.0858662043313456e-05,
+      "loss": 0.3543,
+      "step": 2141
+    },
+    {
+      "epoch": 3.082014388489209,
+      "grad_norm": 0.1642312010716747,
+      "learning_rate": 3.081954349430051e-05,
+      "loss": 0.3562,
+      "step": 2142
+    },
+    {
+      "epoch": 3.083453237410072,
+      "grad_norm": 0.17503639457937903,
+      "learning_rate": 3.0780434211627415e-05,
+      "loss": 0.361,
+      "step": 2143
+    },
+    {
+      "epoch": 3.0848920863309353,
+      "grad_norm": 0.16118245417268506,
+      "learning_rate": 3.074133423476932e-05,
+      "loss": 0.3644,
+      "step": 2144
+    },
+    {
+      "epoch": 3.0863309352517985,
+      "grad_norm": 0.14723476541237263,
+      "learning_rate": 3.070224360319197e-05,
+      "loss": 0.362,
+      "step": 2145
+    },
+    {
+      "epoch": 3.0877697841726617,
+      "grad_norm": 0.14714634472098959,
+      "learning_rate": 3.066316235635168e-05,
+      "loss": 0.3614,
+      "step": 2146
+    },
+    {
+      "epoch": 3.0892086330935253,
+      "grad_norm": 0.15193247445136177,
+      "learning_rate": 3.0624090533695324e-05,
+      "loss": 0.352,
+      "step": 2147
+    },
+    {
+      "epoch": 3.0906474820143885,
+      "grad_norm": 0.18749179624902063,
+      "learning_rate": 3.0585028174660236e-05,
+      "loss": 0.3642,
+      "step": 2148
+    },
+    {
+      "epoch": 3.0920863309352518,
+      "grad_norm": 0.1392666647698655,
+      "learning_rate": 3.054597531867419e-05,
+      "loss": 0.3649,
+      "step": 2149
+    },
+    {
+      "epoch": 3.093525179856115,
+      "grad_norm": 0.16501078769019958,
+      "learning_rate": 3.0506932005155407e-05,
+      "loss": 0.3628,
+      "step": 2150
+    },
+    {
+      "epoch": 3.0949640287769786,
+      "grad_norm": 0.17065492296996415,
+      "learning_rate": 3.0467898273512446e-05,
+      "loss": 0.3673,
+      "step": 2151
+    },
+    {
+      "epoch": 3.096402877697842,
+      "grad_norm": 0.15350854835529376,
+      "learning_rate": 3.042887416314418e-05,
+      "loss": 0.3567,
+      "step": 2152
+    },
+    {
+      "epoch": 3.097841726618705,
+      "grad_norm": 0.3506708121375363,
+      "learning_rate": 3.03898597134398e-05,
+      "loss": 0.3618,
+      "step": 2153
+    },
+    {
+      "epoch": 3.0992805755395683,
+      "grad_norm": 0.1294663959546018,
+      "learning_rate": 3.0350854963778755e-05,
+      "loss": 0.3574,
+      "step": 2154
+    },
+    {
+      "epoch": 3.1007194244604315,
+      "grad_norm": 0.14377101612817697,
+      "learning_rate": 3.0311859953530672e-05,
+      "loss": 0.3537,
+      "step": 2155
+    },
+    {
+      "epoch": 3.102158273381295,
+      "grad_norm": 0.13514133077016605,
+      "learning_rate": 3.027287472205535e-05,
+      "loss": 0.3686,
+      "step": 2156
+    },
+    {
+      "epoch": 3.1035971223021583,
+      "grad_norm": 0.19111383611662044,
+      "learning_rate": 3.0233899308702722e-05,
+      "loss": 0.36,
+      "step": 2157
+    },
+    {
+      "epoch": 3.1050359712230216,
+      "grad_norm": 0.14677733546118896,
+      "learning_rate": 3.0194933752812853e-05,
+      "loss": 0.3648,
+      "step": 2158
+    },
+    {
+      "epoch": 3.1064748201438848,
+      "grad_norm": 0.1485521726666662,
+      "learning_rate": 3.0155978093715787e-05,
+      "loss": 0.3561,
+      "step": 2159
+    },
+    {
+      "epoch": 3.1079136690647484,
+      "grad_norm": 0.1627671669920901,
+      "learning_rate": 3.011703237073162e-05,
+      "loss": 0.3623,
+      "step": 2160
+    },
+    {
+      "epoch": 3.1093525179856116,
+      "grad_norm": 0.16502848496230754,
+      "learning_rate": 3.0078096623170442e-05,
+      "loss": 0.3664,
+      "step": 2161
+    },
+    {
+      "epoch": 3.110791366906475,
+      "grad_norm": 0.13162514219343133,
+      "learning_rate": 3.0039170890332214e-05,
+      "loss": 0.3602,
+      "step": 2162
+    },
+    {
+      "epoch": 3.112230215827338,
+      "grad_norm": 0.15660829701380588,
+      "learning_rate": 3.0000255211506836e-05,
+      "loss": 0.3653,
+      "step": 2163
+    },
+    {
+      "epoch": 3.1136690647482013,
+      "grad_norm": 0.17539034318923397,
+      "learning_rate": 2.9961349625974022e-05,
+      "loss": 0.3618,
+      "step": 2164
+    },
+    {
+      "epoch": 3.115107913669065,
+      "grad_norm": 0.14793606029669693,
+      "learning_rate": 2.992245417300335e-05,
+      "loss": 0.355,
+      "step": 2165
+    },
+    {
+      "epoch": 3.116546762589928,
+      "grad_norm": 0.15900645500930594,
+      "learning_rate": 2.9883568891854118e-05,
+      "loss": 0.3638,
+      "step": 2166
+    },
+    {
+      "epoch": 3.1179856115107913,
+      "grad_norm": 0.1612124230990281,
+      "learning_rate": 2.9844693821775394e-05,
+      "loss": 0.3668,
+      "step": 2167
+    },
+    {
+      "epoch": 3.1194244604316546,
+      "grad_norm": 0.1511141171936181,
+      "learning_rate": 2.9805829002005907e-05,
+      "loss": 0.3589,
+      "step": 2168
+    },
+    {
+      "epoch": 3.1208633093525178,
+      "grad_norm": 0.16026881604644838,
+      "learning_rate": 2.9766974471774072e-05,
+      "loss": 0.3597,
+      "step": 2169
+    },
+    {
+      "epoch": 3.1223021582733814,
+      "grad_norm": 0.1595382668139755,
+      "learning_rate": 2.9728130270297913e-05,
+      "loss": 0.3675,
+      "step": 2170
+    },
+    {
+      "epoch": 3.1237410071942446,
+      "grad_norm": 0.15483819300457385,
+      "learning_rate": 2.968929643678499e-05,
+      "loss": 0.3665,
+      "step": 2171
+    },
+    {
+      "epoch": 3.125179856115108,
+      "grad_norm": 0.1747356909354723,
+      "learning_rate": 2.965047301043246e-05,
+      "loss": 0.3625,
+      "step": 2172
+    },
+    {
+      "epoch": 3.126618705035971,
+      "grad_norm": 0.1577084537740128,
+      "learning_rate": 2.961166003042692e-05,
+      "loss": 0.3563,
+      "step": 2173
+    },
+    {
+      "epoch": 3.1280575539568347,
+      "grad_norm": 0.16341455776658428,
+      "learning_rate": 2.9572857535944473e-05,
+      "loss": 0.3745,
+      "step": 2174
+    },
+    {
+      "epoch": 3.129496402877698,
+      "grad_norm": 0.1478147632598593,
+      "learning_rate": 2.9534065566150567e-05,
+      "loss": 0.3629,
+      "step": 2175
+    },
+    {
+      "epoch": 3.130935251798561,
+      "grad_norm": 0.13978710000013367,
+      "learning_rate": 2.9495284160200105e-05,
+      "loss": 0.3634,
+      "step": 2176
+    },
+    {
+      "epoch": 3.1323741007194243,
+      "grad_norm": 0.1372616987172764,
+      "learning_rate": 2.9456513357237305e-05,
+      "loss": 0.3619,
+      "step": 2177
+    },
+    {
+      "epoch": 3.133812949640288,
+      "grad_norm": 0.13634600501027924,
+      "learning_rate": 2.9417753196395637e-05,
+      "loss": 0.3569,
+      "step": 2178
+    },
+    {
+      "epoch": 3.135251798561151,
+      "grad_norm": 0.13330859445128854,
+      "learning_rate": 2.9379003716797877e-05,
+      "loss": 0.3638,
+      "step": 2179
+    },
+    {
+      "epoch": 3.1366906474820144,
+      "grad_norm": 0.1387344689681127,
+      "learning_rate": 2.9340264957556018e-05,
+      "loss": 0.3651,
+      "step": 2180
+    },
+    {
+      "epoch": 3.1381294964028776,
+      "grad_norm": 0.11495602812743524,
+      "learning_rate": 2.9301536957771218e-05,
+      "loss": 0.3673,
+      "step": 2181
+    },
+    {
+      "epoch": 3.139568345323741,
+      "grad_norm": 0.14156198330485406,
+      "learning_rate": 2.9262819756533754e-05,
+      "loss": 0.3601,
+      "step": 2182
+    },
+    {
+      "epoch": 3.1410071942446045,
+      "grad_norm": 0.13409357944157357,
+      "learning_rate": 2.922411339292306e-05,
+      "loss": 0.3566,
+      "step": 2183
+    },
+    {
+      "epoch": 3.1424460431654677,
+      "grad_norm": 0.1378045890509274,
+      "learning_rate": 2.9185417906007586e-05,
+      "loss": 0.364,
+      "step": 2184
+    },
+    {
+      "epoch": 3.143884892086331,
+      "grad_norm": 0.13291284842051967,
+      "learning_rate": 2.914673333484481e-05,
+      "loss": 0.3656,
+      "step": 2185
+    },
+    {
+      "epoch": 3.145323741007194,
+      "grad_norm": 0.13332793776054366,
+      "learning_rate": 2.9108059718481184e-05,
+      "loss": 0.3554,
+      "step": 2186
+    },
+    {
+      "epoch": 3.1467625899280574,
+      "grad_norm": 0.15458068216679197,
+      "learning_rate": 2.906939709595216e-05,
+      "loss": 0.3652,
+      "step": 2187
+    },
+    {
+      "epoch": 3.148201438848921,
+      "grad_norm": 0.15967056211424555,
+      "learning_rate": 2.9030745506282017e-05,
+      "loss": 0.3563,
+      "step": 2188
+    },
+    {
+      "epoch": 3.149640287769784,
+      "grad_norm": 0.1834537720451152,
+      "learning_rate": 2.8992104988483943e-05,
+      "loss": 0.3622,
+      "step": 2189
+    },
+    {
+      "epoch": 3.1510791366906474,
+      "grad_norm": 0.15455959102393504,
+      "learning_rate": 2.895347558155992e-05,
+      "loss": 0.3581,
+      "step": 2190
+    },
+    {
+      "epoch": 3.1525179856115106,
+      "grad_norm": 0.18274074548645566,
+      "learning_rate": 2.8914857324500767e-05,
+      "loss": 0.3619,
+      "step": 2191
+    },
+    {
+      "epoch": 3.1539568345323743,
+      "grad_norm": 0.19042272209457872,
+      "learning_rate": 2.887625025628599e-05,
+      "loss": 0.3605,
+      "step": 2192
+    },
+    {
+      "epoch": 3.1553956834532375,
+      "grad_norm": 0.178393476055343,
+      "learning_rate": 2.8837654415883817e-05,
+      "loss": 0.362,
+      "step": 2193
+    },
+    {
+      "epoch": 3.1568345323741007,
+      "grad_norm": 0.17452940444183646,
+      "learning_rate": 2.879906984225119e-05,
+      "loss": 0.364,
+      "step": 2194
+    },
+    {
+      "epoch": 3.158273381294964,
+      "grad_norm": 0.19848713272935012,
+      "learning_rate": 2.8760496574333613e-05,
+      "loss": 0.3637,
+      "step": 2195
+    },
+    {
+      "epoch": 3.159712230215827,
+      "grad_norm": 0.22150541825308392,
+      "learning_rate": 2.8721934651065227e-05,
+      "loss": 0.3602,
+      "step": 2196
+    },
+    {
+      "epoch": 3.161151079136691,
+      "grad_norm": 0.1564675995472657,
+      "learning_rate": 2.8683384111368675e-05,
+      "loss": 0.3656,
+      "step": 2197
+    },
+    {
+      "epoch": 3.162589928057554,
+      "grad_norm": 0.17609019863597677,
+      "learning_rate": 2.864484499415517e-05,
+      "loss": 0.3566,
+      "step": 2198
+    },
+    {
+      "epoch": 3.1640287769784172,
+      "grad_norm": 0.1619486377977057,
+      "learning_rate": 2.8606317338324347e-05,
+      "loss": 0.3605,
+      "step": 2199
+    },
+    {
+      "epoch": 3.1654676258992804,
+      "grad_norm": 0.14723851731572016,
+      "learning_rate": 2.856780118276429e-05,
+      "loss": 0.3631,
+      "step": 2200
+    },
+    {
+      "epoch": 3.166906474820144,
+      "grad_norm": 0.16034339510914788,
+      "learning_rate": 2.852929656635146e-05,
+      "loss": 0.3668,
+      "step": 2201
+    },
+    {
+      "epoch": 3.1683453237410073,
+      "grad_norm": 0.14324836151156242,
+      "learning_rate": 2.8490803527950706e-05,
+      "loss": 0.3573,
+      "step": 2202
+    },
+    {
+      "epoch": 3.1697841726618705,
+      "grad_norm": 0.13913473975969853,
+      "learning_rate": 2.845232210641517e-05,
+      "loss": 0.3702,
+      "step": 2203
+    },
+    {
+      "epoch": 3.1712230215827337,
+      "grad_norm": 0.13827107183418555,
+      "learning_rate": 2.841385234058624e-05,
+      "loss": 0.3655,
+      "step": 2204
+    },
+    {
+      "epoch": 3.172661870503597,
+      "grad_norm": 0.14026787843847674,
+      "learning_rate": 2.83753942692936e-05,
+      "loss": 0.351,
+      "step": 2205
+    },
+    {
+      "epoch": 3.1741007194244606,
+      "grad_norm": 0.1392484108958197,
+      "learning_rate": 2.8336947931355096e-05,
+      "loss": 0.36,
+      "step": 2206
+    },
+    {
+      "epoch": 3.175539568345324,
+      "grad_norm": 0.15614689262731513,
+      "learning_rate": 2.8298513365576715e-05,
+      "loss": 0.3602,
+      "step": 2207
+    },
+    {
+      "epoch": 3.176978417266187,
+      "grad_norm": 0.1614798507847841,
+      "learning_rate": 2.826009061075257e-05,
+      "loss": 0.3571,
+      "step": 2208
+    },
+    {
+      "epoch": 3.1784172661870502,
+      "grad_norm": 0.14968532541531165,
+      "learning_rate": 2.822167970566488e-05,
+      "loss": 0.3592,
+      "step": 2209
+    },
+    {
+      "epoch": 3.1798561151079134,
+      "grad_norm": 0.14943552936276022,
+      "learning_rate": 2.8183280689083895e-05,
+      "loss": 0.3552,
+      "step": 2210
+    },
+    {
+      "epoch": 3.181294964028777,
+      "grad_norm": 0.13942741347585594,
+      "learning_rate": 2.8144893599767828e-05,
+      "loss": 0.3539,
+      "step": 2211
+    },
+    {
+      "epoch": 3.1827338129496403,
+      "grad_norm": 0.14109944425966342,
+      "learning_rate": 2.8106518476462886e-05,
+      "loss": 0.3633,
+      "step": 2212
+    },
+    {
+      "epoch": 3.1841726618705035,
+      "grad_norm": 0.12957269945337271,
+      "learning_rate": 2.806815535790321e-05,
+      "loss": 0.3558,
+      "step": 2213
+    },
+    {
+      "epoch": 3.1856115107913667,
+      "grad_norm": 0.17172783398095212,
+      "learning_rate": 2.8029804282810794e-05,
+      "loss": 0.3644,
+      "step": 2214
+    },
+    {
+      "epoch": 3.1870503597122304,
+      "grad_norm": 0.1695904394556663,
+      "learning_rate": 2.7991465289895497e-05,
+      "loss": 0.3689,
+      "step": 2215
+    },
+    {
+      "epoch": 3.1884892086330936,
+      "grad_norm": 0.15800796596879413,
+      "learning_rate": 2.7953138417854952e-05,
+      "loss": 0.3641,
+      "step": 2216
+    },
+    {
+      "epoch": 3.189928057553957,
+      "grad_norm": 0.13255286416329495,
+      "learning_rate": 2.79148237053746e-05,
+      "loss": 0.3644,
+      "step": 2217
+    },
+    {
+      "epoch": 3.19136690647482,
+      "grad_norm": 0.16659802965723186,
+      "learning_rate": 2.787652119112758e-05,
+      "loss": 0.3613,
+      "step": 2218
+    },
+    {
+      "epoch": 3.1928057553956837,
+      "grad_norm": 0.13041920241377952,
+      "learning_rate": 2.783823091377472e-05,
+      "loss": 0.3581,
+      "step": 2219
+    },
+    {
+      "epoch": 3.194244604316547,
+      "grad_norm": 0.14796305058514558,
+      "learning_rate": 2.7799952911964535e-05,
+      "loss": 0.3626,
+      "step": 2220
+    },
+    {
+      "epoch": 3.19568345323741,
+      "grad_norm": 0.15254320307560715,
+      "learning_rate": 2.776168722433308e-05,
+      "loss": 0.3697,
+      "step": 2221
+    },
+    {
+      "epoch": 3.1971223021582733,
+      "grad_norm": 0.14452204938283672,
+      "learning_rate": 2.7723433889504046e-05,
+      "loss": 0.3625,
+      "step": 2222
+    },
+    {
+      "epoch": 3.1985611510791365,
+      "grad_norm": 0.16913298343868277,
+      "learning_rate": 2.7685192946088597e-05,
+      "loss": 0.355,
+      "step": 2223
+    },
+    {
+      "epoch": 3.2,
+      "grad_norm": 0.14654307628150673,
+      "learning_rate": 2.7646964432685456e-05,
+      "loss": 0.3628,
+      "step": 2224
+    },
+    {
+      "epoch": 3.2014388489208634,
+      "grad_norm": 0.16711073485909242,
+      "learning_rate": 2.7608748387880754e-05,
+      "loss": 0.3677,
+      "step": 2225
+    },
+    {
+      "epoch": 3.2028776978417266,
+      "grad_norm": 0.1553248559515571,
+      "learning_rate": 2.7570544850248047e-05,
+      "loss": 0.3607,
+      "step": 2226
+    },
+    {
+      "epoch": 3.20431654676259,
+      "grad_norm": 0.14502630303023836,
+      "learning_rate": 2.753235385834824e-05,
+      "loss": 0.3706,
+      "step": 2227
+    },
+    {
+      "epoch": 3.205755395683453,
+      "grad_norm": 0.1722575599110508,
+      "learning_rate": 2.749417545072964e-05,
+      "loss": 0.3615,
+      "step": 2228
+    },
+    {
+      "epoch": 3.2071942446043167,
+      "grad_norm": 0.14424325687418108,
+      "learning_rate": 2.7456009665927807e-05,
+      "loss": 0.3625,
+      "step": 2229
+    },
+    {
+      "epoch": 3.20863309352518,
+      "grad_norm": 0.14399380358513653,
+      "learning_rate": 2.741785654246555e-05,
+      "loss": 0.3693,
+      "step": 2230
+    },
+    {
+      "epoch": 3.210071942446043,
+      "grad_norm": 0.15737465739846349,
+      "learning_rate": 2.7379716118852927e-05,
+      "loss": 0.3584,
+      "step": 2231
+    },
+    {
+      "epoch": 3.2115107913669063,
+      "grad_norm": 0.18840367516428955,
+      "learning_rate": 2.734158843358718e-05,
+      "loss": 0.3555,
+      "step": 2232
+    },
+    {
+      "epoch": 3.21294964028777,
+      "grad_norm": 0.20377731475847594,
+      "learning_rate": 2.730347352515266e-05,
+      "loss": 0.3488,
+      "step": 2233
+    },
+    {
+      "epoch": 3.214388489208633,
+      "grad_norm": 0.13565411926982154,
+      "learning_rate": 2.7265371432020836e-05,
+      "loss": 0.3682,
+      "step": 2234
+    },
+    {
+      "epoch": 3.2158273381294964,
+      "grad_norm": 0.2139869802667484,
+      "learning_rate": 2.7227282192650258e-05,
+      "loss": 0.3581,
+      "step": 2235
+    },
+    {
+      "epoch": 3.2172661870503596,
+      "grad_norm": 0.17260179079279206,
+      "learning_rate": 2.7189205845486503e-05,
+      "loss": 0.3596,
+      "step": 2236
+    },
+    {
+      "epoch": 3.218705035971223,
+      "grad_norm": 0.14636440232485984,
+      "learning_rate": 2.7151142428962103e-05,
+      "loss": 0.3656,
+      "step": 2237
+    },
+    {
+      "epoch": 3.2201438848920865,
+      "grad_norm": 0.18774944778882494,
+      "learning_rate": 2.711309198149655e-05,
+      "loss": 0.3597,
+      "step": 2238
+    },
+    {
+      "epoch": 3.2215827338129497,
+      "grad_norm": 0.16589078085953027,
+      "learning_rate": 2.7075054541496296e-05,
+      "loss": 0.3607,
+      "step": 2239
+    },
+    {
+      "epoch": 3.223021582733813,
+      "grad_norm": 0.12980125257144026,
+      "learning_rate": 2.7037030147354582e-05,
+      "loss": 0.3569,
+      "step": 2240
+    },
+    {
+      "epoch": 3.224460431654676,
+      "grad_norm": 0.12139736644831225,
+      "learning_rate": 2.6999018837451523e-05,
+      "loss": 0.3647,
+      "step": 2241
+    },
+    {
+      "epoch": 3.2258992805755398,
+      "grad_norm": 0.1492330721671925,
+      "learning_rate": 2.6961020650154057e-05,
+      "loss": 0.3647,
+      "step": 2242
+    },
+    {
+      "epoch": 3.227338129496403,
+      "grad_norm": 1.35098016235772,
+      "learning_rate": 2.6923035623815824e-05,
+      "loss": 0.3719,
+      "step": 2243
+    },
+    {
+      "epoch": 3.228776978417266,
+      "grad_norm": 0.1585641258903242,
+      "learning_rate": 2.6885063796777195e-05,
+      "loss": 0.3579,
+      "step": 2244
+    },
+    {
+      "epoch": 3.2302158273381294,
+      "grad_norm": 0.14264742791391744,
+      "learning_rate": 2.6847105207365225e-05,
+      "loss": 0.3533,
+      "step": 2245
+    },
+    {
+      "epoch": 3.2316546762589926,
+      "grad_norm": 0.14495436282817886,
+      "learning_rate": 2.6809159893893624e-05,
+      "loss": 0.361,
+      "step": 2246
+    },
+    {
+      "epoch": 3.2330935251798563,
+      "grad_norm": 0.15342747948375662,
+      "learning_rate": 2.6771227894662666e-05,
+      "loss": 0.3547,
+      "step": 2247
+    },
+    {
+      "epoch": 3.2345323741007195,
+      "grad_norm": 0.12183175364710477,
+      "learning_rate": 2.6733309247959217e-05,
+      "loss": 0.3525,
+      "step": 2248
+    },
+    {
+      "epoch": 3.2359712230215827,
+      "grad_norm": 0.15689188085640454,
+      "learning_rate": 2.669540399205664e-05,
+      "loss": 0.3593,
+      "step": 2249
+    },
+    {
+      "epoch": 3.237410071942446,
+      "grad_norm": 0.1332782811429741,
+      "learning_rate": 2.6657512165214806e-05,
+      "loss": 0.3537,
+      "step": 2250
+    },
+    {
+      "epoch": 3.2388489208633096,
+      "grad_norm": 0.16238962487909972,
+      "learning_rate": 2.6619633805680028e-05,
+      "loss": 0.3579,
+      "step": 2251
+    },
+    {
+      "epoch": 3.2402877697841728,
+      "grad_norm": 0.1522823914204186,
+      "learning_rate": 2.6581768951684992e-05,
+      "loss": 0.364,
+      "step": 2252
+    },
+    {
+      "epoch": 3.241726618705036,
+      "grad_norm": 0.14345631035638734,
+      "learning_rate": 2.6543917641448813e-05,
+      "loss": 0.3644,
+      "step": 2253
+    },
+    {
+      "epoch": 3.243165467625899,
+      "grad_norm": 0.13918719422262607,
+      "learning_rate": 2.650607991317687e-05,
+      "loss": 0.362,
+      "step": 2254
+    },
+    {
+      "epoch": 3.2446043165467624,
+      "grad_norm": 0.1367299718289178,
+      "learning_rate": 2.6468255805060885e-05,
+      "loss": 0.3542,
+      "step": 2255
+    },
+    {
+      "epoch": 3.246043165467626,
+      "grad_norm": 0.14431389291588131,
+      "learning_rate": 2.6430445355278788e-05,
+      "loss": 0.3688,
+      "step": 2256
+    },
+    {
+      "epoch": 3.2474820143884893,
+      "grad_norm": 0.41771378248386287,
+      "learning_rate": 2.639264860199477e-05,
+      "loss": 0.3616,
+      "step": 2257
+    },
+    {
+      "epoch": 3.2489208633093525,
+      "grad_norm": 0.14462195721294618,
+      "learning_rate": 2.6354865583359175e-05,
+      "loss": 0.3607,
+      "step": 2258
+    },
+    {
+      "epoch": 3.2503597122302157,
+      "grad_norm": 0.11415729967552433,
+      "learning_rate": 2.631709633750847e-05,
+      "loss": 0.3632,
+      "step": 2259
+    },
+    {
+      "epoch": 3.2517985611510793,
+      "grad_norm": 0.15255190313869024,
+      "learning_rate": 2.6279340902565217e-05,
+      "loss": 0.3578,
+      "step": 2260
+    },
+    {
+      "epoch": 3.2532374100719426,
+      "grad_norm": 0.13805446285540282,
+      "learning_rate": 2.6241599316638084e-05,
+      "loss": 0.3563,
+      "step": 2261
+    },
+    {
+      "epoch": 3.2546762589928058,
+      "grad_norm": 0.1418527357355985,
+      "learning_rate": 2.6203871617821717e-05,
+      "loss": 0.3612,
+      "step": 2262
+    },
+    {
+      "epoch": 3.256115107913669,
+      "grad_norm": 0.14013824159648403,
+      "learning_rate": 2.6166157844196755e-05,
+      "loss": 0.3538,
+      "step": 2263
+    },
+    {
+      "epoch": 3.257553956834532,
+      "grad_norm": 0.13935395781290988,
+      "learning_rate": 2.6128458033829792e-05,
+      "loss": 0.3582,
+      "step": 2264
+    },
+    {
+      "epoch": 3.258992805755396,
+      "grad_norm": 0.14360309972844257,
+      "learning_rate": 2.609077222477332e-05,
+      "loss": 0.361,
+      "step": 2265
+    },
+    {
+      "epoch": 3.260431654676259,
+      "grad_norm": 0.17150198840354863,
+      "learning_rate": 2.6053100455065693e-05,
+      "loss": 0.3585,
+      "step": 2266
+    },
+    {
+      "epoch": 3.2618705035971223,
+      "grad_norm": 0.13678134536281703,
+      "learning_rate": 2.6015442762731095e-05,
+      "loss": 0.3621,
+      "step": 2267
+    },
+    {
+      "epoch": 3.2633093525179855,
+      "grad_norm": 0.18600744624660287,
+      "learning_rate": 2.5977799185779534e-05,
+      "loss": 0.3648,
+      "step": 2268
+    },
+    {
+      "epoch": 3.2647482014388487,
+      "grad_norm": 0.20982476736184863,
+      "learning_rate": 2.5940169762206722e-05,
+      "loss": 0.3559,
+      "step": 2269
+    },
+    {
+      "epoch": 3.2661870503597124,
+      "grad_norm": 0.21102441705807895,
+      "learning_rate": 2.5902554529994105e-05,
+      "loss": 0.3649,
+      "step": 2270
+    },
+    {
+      "epoch": 3.2676258992805756,
+      "grad_norm": 0.15796192676257098,
+      "learning_rate": 2.5864953527108805e-05,
+      "loss": 0.3669,
+      "step": 2271
+    },
+    {
+      "epoch": 3.2690647482014388,
+      "grad_norm": 0.2025325960295383,
+      "learning_rate": 2.58273667915036e-05,
+      "loss": 0.3578,
+      "step": 2272
+    },
+    {
+      "epoch": 3.270503597122302,
+      "grad_norm": 0.20369588142188552,
+      "learning_rate": 2.578979436111684e-05,
+      "loss": 0.3594,
+      "step": 2273
+    },
+    {
+      "epoch": 3.2719424460431656,
+      "grad_norm": 0.15584996538262325,
+      "learning_rate": 2.5752236273872432e-05,
+      "loss": 0.3693,
+      "step": 2274
+    },
+    {
+      "epoch": 3.273381294964029,
+      "grad_norm": 0.29022040048565395,
+      "learning_rate": 2.5714692567679853e-05,
+      "loss": 0.358,
+      "step": 2275
+    },
+    {
+      "epoch": 3.274820143884892,
+      "grad_norm": 0.22458395920921975,
+      "learning_rate": 2.5677163280433995e-05,
+      "loss": 0.3566,
+      "step": 2276
+    },
+    {
+      "epoch": 3.2762589928057553,
+      "grad_norm": 0.18734340823549248,
+      "learning_rate": 2.5639648450015268e-05,
+      "loss": 0.3582,
+      "step": 2277
+    },
+    {
+      "epoch": 3.277697841726619,
+      "grad_norm": 0.26602126473058135,
+      "learning_rate": 2.5602148114289415e-05,
+      "loss": 0.3626,
+      "step": 2278
+    },
+    {
+      "epoch": 3.279136690647482,
+      "grad_norm": 0.1681330470232265,
+      "learning_rate": 2.556466231110762e-05,
+      "loss": 0.3646,
+      "step": 2279
+    },
+    {
+      "epoch": 3.2805755395683454,
+      "grad_norm": 0.1992067827528513,
+      "learning_rate": 2.552719107830635e-05,
+      "loss": 0.3509,
+      "step": 2280
+    },
+    {
+      "epoch": 3.2820143884892086,
+      "grad_norm": 0.18565700988937925,
+      "learning_rate": 2.54897344537074e-05,
+      "loss": 0.3724,
+      "step": 2281
+    },
+    {
+      "epoch": 3.283453237410072,
+      "grad_norm": 0.1512645675478467,
+      "learning_rate": 2.5452292475117767e-05,
+      "loss": 0.3633,
+      "step": 2282
+    },
+    {
+      "epoch": 3.2848920863309354,
+      "grad_norm": 0.1904658977836513,
+      "learning_rate": 2.541486518032973e-05,
+      "loss": 0.3627,
+      "step": 2283
+    },
+    {
+      "epoch": 3.2863309352517986,
+      "grad_norm": 0.1931450539981554,
+      "learning_rate": 2.5377452607120722e-05,
+      "loss": 0.3556,
+      "step": 2284
+    },
+    {
+      "epoch": 3.287769784172662,
+      "grad_norm": 0.1856206254349257,
+      "learning_rate": 2.5340054793253276e-05,
+      "loss": 0.3728,
+      "step": 2285
+    },
+    {
+      "epoch": 3.289208633093525,
+      "grad_norm": 0.13019591341177691,
+      "learning_rate": 2.5302671776475098e-05,
+      "loss": 0.3558,
+      "step": 2286
+    },
+    {
+      "epoch": 3.2906474820143883,
+      "grad_norm": 0.21782000590473927,
+      "learning_rate": 2.526530359451892e-05,
+      "loss": 0.3662,
+      "step": 2287
+    },
+    {
+      "epoch": 3.292086330935252,
+      "grad_norm": 0.1734671650120193,
+      "learning_rate": 2.522795028510249e-05,
+      "loss": 0.3546,
+      "step": 2288
+    },
+    {
+      "epoch": 3.293525179856115,
+      "grad_norm": 0.15642074578337598,
+      "learning_rate": 2.5190611885928547e-05,
+      "loss": 0.3673,
+      "step": 2289
+    },
+    {
+      "epoch": 3.2949640287769784,
+      "grad_norm": 0.22968584304270215,
+      "learning_rate": 2.5153288434684816e-05,
+      "loss": 0.3634,
+      "step": 2290
+    },
+    {
+      "epoch": 3.2964028776978416,
+      "grad_norm": 0.14230543830002376,
+      "learning_rate": 2.5115979969043914e-05,
+      "loss": 0.3618,
+      "step": 2291
+    },
+    {
+      "epoch": 3.2978417266187052,
+      "grad_norm": 0.26943377434593496,
+      "learning_rate": 2.5078686526663304e-05,
+      "loss": 0.3717,
+      "step": 2292
+    },
+    {
+      "epoch": 3.2992805755395684,
+      "grad_norm": 0.6120181192293852,
+      "learning_rate": 2.5041408145185312e-05,
+      "loss": 0.3615,
+      "step": 2293
+    },
+    {
+      "epoch": 3.3007194244604317,
+      "grad_norm": 0.19122352513352453,
+      "learning_rate": 2.5004144862237084e-05,
+      "loss": 0.3679,
+      "step": 2294
+    },
+    {
+      "epoch": 3.302158273381295,
+      "grad_norm": 0.23795571722273903,
+      "learning_rate": 2.4966896715430484e-05,
+      "loss": 0.3607,
+      "step": 2295
+    },
+    {
+      "epoch": 3.3035971223021585,
+      "grad_norm": 0.1896858659606159,
+      "learning_rate": 2.4929663742362103e-05,
+      "loss": 0.3653,
+      "step": 2296
+    },
+    {
+      "epoch": 3.3050359712230217,
+      "grad_norm": 0.22376447671270214,
+      "learning_rate": 2.4892445980613254e-05,
+      "loss": 0.3627,
+      "step": 2297
+    },
+    {
+      "epoch": 3.306474820143885,
+      "grad_norm": 0.18157008784299794,
+      "learning_rate": 2.4855243467749865e-05,
+      "loss": 0.3713,
+      "step": 2298
+    },
+    {
+      "epoch": 3.307913669064748,
+      "grad_norm": 0.27676117774412173,
+      "learning_rate": 2.481805624132247e-05,
+      "loss": 0.3618,
+      "step": 2299
+    },
+    {
+      "epoch": 3.3093525179856114,
+      "grad_norm": 0.25296348485621617,
+      "learning_rate": 2.478088433886618e-05,
+      "loss": 0.3599,
+      "step": 2300
+    },
+    {
+      "epoch": 3.310791366906475,
+      "grad_norm": 0.19493846157171688,
+      "learning_rate": 2.4743727797900668e-05,
+      "loss": 0.3652,
+      "step": 2301
+    },
+    {
+      "epoch": 3.3122302158273382,
+      "grad_norm": 0.23870314122074515,
+      "learning_rate": 2.4706586655930042e-05,
+      "loss": 0.3593,
+      "step": 2302
+    },
+    {
+      "epoch": 3.3136690647482014,
+      "grad_norm": 0.1788363419814252,
+      "learning_rate": 2.4669460950442926e-05,
+      "loss": 0.3556,
+      "step": 2303
+    },
+    {
+      "epoch": 3.3151079136690647,
+      "grad_norm": 0.19801080360218584,
+      "learning_rate": 2.463235071891231e-05,
+      "loss": 0.3579,
+      "step": 2304
+    },
+    {
+      "epoch": 3.316546762589928,
+      "grad_norm": 0.16986415708063274,
+      "learning_rate": 2.4595255998795625e-05,
+      "loss": 0.3628,
+      "step": 2305
+    },
+    {
+      "epoch": 3.3179856115107915,
+      "grad_norm": 0.14381662338937498,
+      "learning_rate": 2.4558176827534587e-05,
+      "loss": 0.3633,
+      "step": 2306
+    },
+    {
+      "epoch": 3.3194244604316547,
+      "grad_norm": 0.14437074326676055,
+      "learning_rate": 2.452111324255524e-05,
+      "loss": 0.3597,
+      "step": 2307
+    },
+    {
+      "epoch": 3.320863309352518,
+      "grad_norm": 0.17139115163412483,
+      "learning_rate": 2.448406528126793e-05,
+      "loss": 0.3608,
+      "step": 2308
+    },
+    {
+      "epoch": 3.322302158273381,
+      "grad_norm": 0.13817997375385707,
+      "learning_rate": 2.444703298106718e-05,
+      "loss": 0.3694,
+      "step": 2309
+    },
+    {
+      "epoch": 3.3237410071942444,
+      "grad_norm": 0.1530764392910559,
+      "learning_rate": 2.441001637933173e-05,
+      "loss": 0.3609,
+      "step": 2310
+    },
+    {
+      "epoch": 3.325179856115108,
+      "grad_norm": 0.12399449512004619,
+      "learning_rate": 2.437301551342447e-05,
+      "loss": 0.374,
+      "step": 2311
+    },
+    {
+      "epoch": 3.3266187050359712,
+      "grad_norm": 0.1709783821443115,
+      "learning_rate": 2.433603042069242e-05,
+      "loss": 0.3634,
+      "step": 2312
+    },
+    {
+      "epoch": 3.3280575539568344,
+      "grad_norm": 0.15145193550435615,
+      "learning_rate": 2.4299061138466667e-05,
+      "loss": 0.3705,
+      "step": 2313
+    },
+    {
+      "epoch": 3.3294964028776977,
+      "grad_norm": 0.14118503963320508,
+      "learning_rate": 2.4262107704062343e-05,
+      "loss": 0.3584,
+      "step": 2314
+    },
+    {
+      "epoch": 3.3309352517985613,
+      "grad_norm": 0.14488814930761457,
+      "learning_rate": 2.4225170154778562e-05,
+      "loss": 0.364,
+      "step": 2315
+    },
+    {
+      "epoch": 3.3323741007194245,
+      "grad_norm": 0.14630912827988185,
+      "learning_rate": 2.4188248527898446e-05,
+      "loss": 0.3607,
+      "step": 2316
+    },
+    {
+      "epoch": 3.3338129496402877,
+      "grad_norm": 0.1577597818350434,
+      "learning_rate": 2.415134286068903e-05,
+      "loss": 0.357,
+      "step": 2317
+    },
+    {
+      "epoch": 3.335251798561151,
+      "grad_norm": 0.14694587128564138,
+      "learning_rate": 2.411445319040121e-05,
+      "loss": 0.3633,
+      "step": 2318
+    },
+    {
+      "epoch": 3.3366906474820146,
+      "grad_norm": 0.14965947001787325,
+      "learning_rate": 2.407757955426977e-05,
+      "loss": 0.3621,
+      "step": 2319
+    },
+    {
+      "epoch": 3.338129496402878,
+      "grad_norm": 0.13259739344131605,
+      "learning_rate": 2.4040721989513314e-05,
+      "loss": 0.3648,
+      "step": 2320
+    },
+    {
+      "epoch": 3.339568345323741,
+      "grad_norm": 0.13523920153875157,
+      "learning_rate": 2.40038805333342e-05,
+      "loss": 0.3618,
+      "step": 2321
+    },
+    {
+      "epoch": 3.3410071942446042,
+      "grad_norm": 0.1311362813996723,
+      "learning_rate": 2.396705522291852e-05,
+      "loss": 0.3586,
+      "step": 2322
+    },
+    {
+      "epoch": 3.3424460431654675,
+      "grad_norm": 0.12998381952826785,
+      "learning_rate": 2.393024609543611e-05,
+      "loss": 0.3583,
+      "step": 2323
+    },
+    {
+      "epoch": 3.343884892086331,
+      "grad_norm": 0.135675371615169,
+      "learning_rate": 2.3893453188040442e-05,
+      "loss": 0.3611,
+      "step": 2324
+    },
+    {
+      "epoch": 3.3453237410071943,
+      "grad_norm": 0.14144602755827798,
+      "learning_rate": 2.3856676537868614e-05,
+      "loss": 0.3582,
+      "step": 2325
+    },
+    {
+      "epoch": 3.3467625899280575,
+      "grad_norm": 0.13529652876471696,
+      "learning_rate": 2.3819916182041318e-05,
+      "loss": 0.3544,
+      "step": 2326
+    },
+    {
+      "epoch": 3.3482014388489207,
+      "grad_norm": 0.13419199847959343,
+      "learning_rate": 2.378317215766283e-05,
+      "loss": 0.3661,
+      "step": 2327
+    },
+    {
+      "epoch": 3.349640287769784,
+      "grad_norm": 0.148272002046631,
+      "learning_rate": 2.3746444501820886e-05,
+      "loss": 0.3547,
+      "step": 2328
+    },
+    {
+      "epoch": 3.3510791366906476,
+      "grad_norm": 0.14830889739587663,
+      "learning_rate": 2.370973325158675e-05,
+      "loss": 0.3614,
+      "step": 2329
+    },
+    {
+      "epoch": 3.352517985611511,
+      "grad_norm": 0.1300783152214282,
+      "learning_rate": 2.3673038444015087e-05,
+      "loss": 0.3635,
+      "step": 2330
+    },
+    {
+      "epoch": 3.353956834532374,
+      "grad_norm": 0.16703248529218837,
+      "learning_rate": 2.363636011614401e-05,
+      "loss": 0.3564,
+      "step": 2331
+    },
+    {
+      "epoch": 3.3553956834532372,
+      "grad_norm": 0.13588325260397235,
+      "learning_rate": 2.3599698304994946e-05,
+      "loss": 0.3645,
+      "step": 2332
+    },
+    {
+      "epoch": 3.356834532374101,
+      "grad_norm": 0.16336185339863962,
+      "learning_rate": 2.3563053047572683e-05,
+      "loss": 0.3629,
+      "step": 2333
+    },
+    {
+      "epoch": 3.358273381294964,
+      "grad_norm": 0.14998556339025992,
+      "learning_rate": 2.352642438086533e-05,
+      "loss": 0.3611,
+      "step": 2334
+    },
+    {
+      "epoch": 3.3597122302158273,
+      "grad_norm": 0.13681888822607777,
+      "learning_rate": 2.348981234184417e-05,
+      "loss": 0.3595,
+      "step": 2335
+    },
+    {
+      "epoch": 3.3611510791366905,
+      "grad_norm": 0.13572005828339215,
+      "learning_rate": 2.3453216967463785e-05,
+      "loss": 0.3669,
+      "step": 2336
+    },
+    {
+      "epoch": 3.362589928057554,
+      "grad_norm": 0.15313958123322324,
+      "learning_rate": 2.3416638294661864e-05,
+      "loss": 0.366,
+      "step": 2337
+    },
+    {
+      "epoch": 3.3640287769784174,
+      "grad_norm": 0.14478875784069747,
+      "learning_rate": 2.3380076360359293e-05,
+      "loss": 0.361,
+      "step": 2338
+    },
+    {
+      "epoch": 3.3654676258992806,
+      "grad_norm": 0.1277587655150165,
+      "learning_rate": 2.3343531201460067e-05,
+      "loss": 0.3597,
+      "step": 2339
+    },
+    {
+      "epoch": 3.366906474820144,
+      "grad_norm": 0.1357502343423544,
+      "learning_rate": 2.3307002854851188e-05,
+      "loss": 0.355,
+      "step": 2340
+    },
+    {
+      "epoch": 3.368345323741007,
+      "grad_norm": 0.15983854814101542,
+      "learning_rate": 2.3270491357402715e-05,
+      "loss": 0.3621,
+      "step": 2341
+    },
+    {
+      "epoch": 3.3697841726618707,
+      "grad_norm": 0.1421995801195904,
+      "learning_rate": 2.3233996745967772e-05,
+      "loss": 0.3575,
+      "step": 2342
+    },
+    {
+      "epoch": 3.371223021582734,
+      "grad_norm": 0.1352167104197795,
+      "learning_rate": 2.3197519057382326e-05,
+      "loss": 0.3626,
+      "step": 2343
+    },
+    {
+      "epoch": 3.372661870503597,
+      "grad_norm": 0.12458809922761538,
+      "learning_rate": 2.316105832846532e-05,
+      "loss": 0.3634,
+      "step": 2344
+    },
+    {
+      "epoch": 3.3741007194244603,
+      "grad_norm": 0.13950285735057272,
+      "learning_rate": 2.3124614596018606e-05,
+      "loss": 0.3621,
+      "step": 2345
+    },
+    {
+      "epoch": 3.3755395683453235,
+      "grad_norm": 0.15166983091892183,
+      "learning_rate": 2.308818789682682e-05,
+      "loss": 0.3568,
+      "step": 2346
+    },
+    {
+      "epoch": 3.376978417266187,
+      "grad_norm": 0.13848819446565608,
+      "learning_rate": 2.3051778267657436e-05,
+      "loss": 0.3662,
+      "step": 2347
+    },
+    {
+      "epoch": 3.3784172661870504,
+      "grad_norm": 0.17810830096812302,
+      "learning_rate": 2.3015385745260704e-05,
+      "loss": 0.375,
+      "step": 2348
+    },
+    {
+      "epoch": 3.3798561151079136,
+      "grad_norm": 0.1250981158644372,
+      "learning_rate": 2.2979010366369595e-05,
+      "loss": 0.3536,
+      "step": 2349
+    },
+    {
+      "epoch": 3.381294964028777,
+      "grad_norm": 0.16379848518712722,
+      "learning_rate": 2.294265216769978e-05,
+      "loss": 0.3608,
+      "step": 2350
+    },
+    {
+      "epoch": 3.38273381294964,
+      "grad_norm": 0.12056498068075343,
+      "learning_rate": 2.2906311185949605e-05,
+      "loss": 0.363,
+      "step": 2351
+    },
+    {
+      "epoch": 3.3841726618705037,
+      "grad_norm": 0.16456533418318436,
+      "learning_rate": 2.2869987457799977e-05,
+      "loss": 0.3605,
+      "step": 2352
+    },
+    {
+      "epoch": 3.385611510791367,
+      "grad_norm": 0.12932181191485873,
+      "learning_rate": 2.283368101991448e-05,
+      "loss": 0.3615,
+      "step": 2353
+    },
+    {
+      "epoch": 3.38705035971223,
+      "grad_norm": 0.1281926781896969,
+      "learning_rate": 2.2797391908939196e-05,
+      "loss": 0.3522,
+      "step": 2354
+    },
+    {
+      "epoch": 3.3884892086330938,
+      "grad_norm": 0.13911342135131524,
+      "learning_rate": 2.2761120161502674e-05,
+      "loss": 0.3695,
+      "step": 2355
+    },
+    {
+      "epoch": 3.389928057553957,
+      "grad_norm": 0.1397088332645429,
+      "learning_rate": 2.2724865814216042e-05,
+      "loss": 0.36,
+      "step": 2356
+    },
+    {
+      "epoch": 3.39136690647482,
+      "grad_norm": 0.1201278802417132,
+      "learning_rate": 2.2688628903672792e-05,
+      "loss": 0.3578,
+      "step": 2357
+    },
+    {
+      "epoch": 3.3928057553956834,
+      "grad_norm": 0.1234892259171623,
+      "learning_rate": 2.265240946644881e-05,
+      "loss": 0.3553,
+      "step": 2358
+    },
+    {
+      "epoch": 3.3942446043165466,
+      "grad_norm": 0.11589818426007034,
+      "learning_rate": 2.261620753910238e-05,
+      "loss": 0.3524,
+      "step": 2359
+    },
+    {
+      "epoch": 3.3956834532374103,
+      "grad_norm": 0.11636042290314506,
+      "learning_rate": 2.25800231581741e-05,
+      "loss": 0.3572,
+      "step": 2360
+    },
+    {
+      "epoch": 3.3971223021582735,
+      "grad_norm": 0.1358301033153014,
+      "learning_rate": 2.254385636018686e-05,
+      "loss": 0.3633,
+      "step": 2361
+    },
+    {
+      "epoch": 3.3985611510791367,
+      "grad_norm": 0.11423064523067442,
+      "learning_rate": 2.250770718164579e-05,
+      "loss": 0.3648,
+      "step": 2362
+    },
+    {
+      "epoch": 3.4,
+      "grad_norm": 0.13761042159576356,
+      "learning_rate": 2.247157565903825e-05,
+      "loss": 0.3539,
+      "step": 2363
+    },
+    {
+      "epoch": 3.401438848920863,
+      "grad_norm": 0.13704512777709393,
+      "learning_rate": 2.243546182883377e-05,
+      "loss": 0.3571,
+      "step": 2364
+    },
+    {
+      "epoch": 3.402877697841727,
+      "grad_norm": 0.1600033975665896,
+      "learning_rate": 2.2399365727484047e-05,
+      "loss": 0.3756,
+      "step": 2365
+    },
+    {
+      "epoch": 3.40431654676259,
+      "grad_norm": 0.13893941806247376,
+      "learning_rate": 2.2363287391422806e-05,
+      "loss": 0.3534,
+      "step": 2366
+    },
+    {
+      "epoch": 3.405755395683453,
+      "grad_norm": 0.17822170749507893,
+      "learning_rate": 2.2327226857065954e-05,
+      "loss": 0.357,
+      "step": 2367
+    },
+    {
+      "epoch": 3.4071942446043164,
+      "grad_norm": 0.13088365758010723,
+      "learning_rate": 2.2291184160811374e-05,
+      "loss": 0.3592,
+      "step": 2368
+    },
+    {
+      "epoch": 3.4086330935251796,
+      "grad_norm": 0.13233694762747514,
+      "learning_rate": 2.22551593390389e-05,
+      "loss": 0.3701,
+      "step": 2369
+    },
+    {
+      "epoch": 3.4100719424460433,
+      "grad_norm": 0.1395773832633985,
+      "learning_rate": 2.2219152428110368e-05,
+      "loss": 0.361,
+      "step": 2370
+    },
+    {
+      "epoch": 3.4115107913669065,
+      "grad_norm": 0.13488363597766204,
+      "learning_rate": 2.218316346436959e-05,
+      "loss": 0.3654,
+      "step": 2371
+    },
+    {
+      "epoch": 3.4129496402877697,
+      "grad_norm": 0.13112663562211102,
+      "learning_rate": 2.2147192484142154e-05,
+      "loss": 0.3655,
+      "step": 2372
+    },
+    {
+      "epoch": 3.414388489208633,
+      "grad_norm": 0.15989586902235292,
+      "learning_rate": 2.2111239523735568e-05,
+      "loss": 0.3633,
+      "step": 2373
+    },
+    {
+      "epoch": 3.4158273381294966,
+      "grad_norm": 0.12999622257849563,
+      "learning_rate": 2.2075304619439127e-05,
+      "loss": 0.3603,
+      "step": 2374
+    },
+    {
+      "epoch": 3.41726618705036,
+      "grad_norm": 0.17371115054607864,
+      "learning_rate": 2.2039387807523914e-05,
+      "loss": 0.3713,
+      "step": 2375
+    },
+    {
+      "epoch": 3.418705035971223,
+      "grad_norm": 0.13595918088936446,
+      "learning_rate": 2.2003489124242742e-05,
+      "loss": 0.3545,
+      "step": 2376
+    },
+    {
+      "epoch": 3.420143884892086,
+      "grad_norm": 0.1667334733541783,
+      "learning_rate": 2.1967608605830115e-05,
+      "loss": 0.364,
+      "step": 2377
+    },
+    {
+      "epoch": 3.42158273381295,
+      "grad_norm": 0.17163831435716062,
+      "learning_rate": 2.1931746288502235e-05,
+      "loss": 0.367,
+      "step": 2378
+    },
+    {
+      "epoch": 3.423021582733813,
+      "grad_norm": 0.1660405582847028,
+      "learning_rate": 2.1895902208456903e-05,
+      "loss": 0.3719,
+      "step": 2379
+    },
+    {
+      "epoch": 3.4244604316546763,
+      "grad_norm": 0.17241463487185166,
+      "learning_rate": 2.186007640187353e-05,
+      "loss": 0.3548,
+      "step": 2380
+    },
+    {
+      "epoch": 3.4258992805755395,
+      "grad_norm": 0.14327969394563383,
+      "learning_rate": 2.1824268904913036e-05,
+      "loss": 0.3528,
+      "step": 2381
+    },
+    {
+      "epoch": 3.4273381294964027,
+      "grad_norm": 0.1685271143607362,
+      "learning_rate": 2.1788479753717935e-05,
+      "loss": 0.3724,
+      "step": 2382
+    },
+    {
+      "epoch": 3.4287769784172664,
+      "grad_norm": 0.126245263397103,
+      "learning_rate": 2.1752708984412196e-05,
+      "loss": 0.364,
+      "step": 2383
+    },
+    {
+      "epoch": 3.4302158273381296,
+      "grad_norm": 0.1350036452412881,
+      "learning_rate": 2.171695663310119e-05,
+      "loss": 0.3616,
+      "step": 2384
+    },
+    {
+      "epoch": 3.431654676258993,
+      "grad_norm": 0.12641814167693138,
+      "learning_rate": 2.1681222735871747e-05,
+      "loss": 0.3587,
+      "step": 2385
+    },
+    {
+      "epoch": 3.433093525179856,
+      "grad_norm": 0.14682365408094103,
+      "learning_rate": 2.1645507328792058e-05,
+      "loss": 0.364,
+      "step": 2386
+    },
+    {
+      "epoch": 3.434532374100719,
+      "grad_norm": 0.14666110922925735,
+      "learning_rate": 2.1609810447911637e-05,
+      "loss": 0.3561,
+      "step": 2387
+    },
+    {
+      "epoch": 3.435971223021583,
+      "grad_norm": 0.14493166271009153,
+      "learning_rate": 2.157413212926133e-05,
+      "loss": 0.3618,
+      "step": 2388
+    },
+    {
+      "epoch": 3.437410071942446,
+      "grad_norm": 0.1460329551895372,
+      "learning_rate": 2.1538472408853206e-05,
+      "loss": 0.3577,
+      "step": 2389
+    },
+    {
+      "epoch": 3.4388489208633093,
+      "grad_norm": 0.1416819326100094,
+      "learning_rate": 2.1502831322680598e-05,
+      "loss": 0.3596,
+      "step": 2390
+    },
+    {
+      "epoch": 3.4402877697841725,
+      "grad_norm": 0.13223788299368044,
+      "learning_rate": 2.1467208906718008e-05,
+      "loss": 0.3607,
+      "step": 2391
+    },
+    {
+      "epoch": 3.441726618705036,
+      "grad_norm": 0.13332809542290672,
+      "learning_rate": 2.1431605196921103e-05,
+      "loss": 0.3563,
+      "step": 2392
+    },
+    {
+      "epoch": 3.4431654676258994,
+      "grad_norm": 0.1447394560751939,
+      "learning_rate": 2.1396020229226666e-05,
+      "loss": 0.3587,
+      "step": 2393
+    },
+    {
+      "epoch": 3.4446043165467626,
+      "grad_norm": 0.13384779722057508,
+      "learning_rate": 2.1360454039552577e-05,
+      "loss": 0.3587,
+      "step": 2394
+    },
+    {
+      "epoch": 3.446043165467626,
+      "grad_norm": 0.14334642905409703,
+      "learning_rate": 2.1324906663797718e-05,
+      "loss": 0.3749,
+      "step": 2395
+    },
+    {
+      "epoch": 3.4474820143884894,
+      "grad_norm": 0.1321639897240863,
+      "learning_rate": 2.1289378137842008e-05,
+      "loss": 0.3655,
+      "step": 2396
+    },
+    {
+      "epoch": 3.4489208633093527,
+      "grad_norm": 0.14159494357555968,
+      "learning_rate": 2.125386849754639e-05,
+      "loss": 0.3661,
+      "step": 2397
+    },
+    {
+      "epoch": 3.450359712230216,
+      "grad_norm": 0.14423307154734125,
+      "learning_rate": 2.121837777875266e-05,
+      "loss": 0.3546,
+      "step": 2398
+    },
+    {
+      "epoch": 3.451798561151079,
+      "grad_norm": 0.1384862291131786,
+      "learning_rate": 2.118290601728354e-05,
+      "loss": 0.3591,
+      "step": 2399
+    },
+    {
+      "epoch": 3.4532374100719423,
+      "grad_norm": 0.1378806022213264,
+      "learning_rate": 2.1147453248942687e-05,
+      "loss": 0.3638,
+      "step": 2400
+    },
+    {
+      "epoch": 3.454676258992806,
+      "grad_norm": 0.1397248975920691,
+      "learning_rate": 2.1112019509514478e-05,
+      "loss": 0.3662,
+      "step": 2401
+    },
+    {
+      "epoch": 3.456115107913669,
+      "grad_norm": 0.14987395038178755,
+      "learning_rate": 2.1076604834764154e-05,
+      "loss": 0.3622,
+      "step": 2402
+    },
+    {
+      "epoch": 3.4575539568345324,
+      "grad_norm": 0.14653004538413536,
+      "learning_rate": 2.1041209260437694e-05,
+      "loss": 0.3614,
+      "step": 2403
+    },
+    {
+      "epoch": 3.4589928057553956,
+      "grad_norm": 0.14247884835751787,
+      "learning_rate": 2.10058328222618e-05,
+      "loss": 0.3576,
+      "step": 2404
+    },
+    {
+      "epoch": 3.460431654676259,
+      "grad_norm": 0.16784949026039206,
+      "learning_rate": 2.097047555594385e-05,
+      "loss": 0.3718,
+      "step": 2405
+    },
+    {
+      "epoch": 3.4618705035971225,
+      "grad_norm": 0.13598534286178163,
+      "learning_rate": 2.0935137497171904e-05,
+      "loss": 0.3585,
+      "step": 2406
+    },
+    {
+      "epoch": 3.4633093525179857,
+      "grad_norm": 0.15804531633554808,
+      "learning_rate": 2.0899818681614557e-05,
+      "loss": 0.3662,
+      "step": 2407
+    },
+    {
+      "epoch": 3.464748201438849,
+      "grad_norm": 0.13091743193933542,
+      "learning_rate": 2.086451914492108e-05,
+      "loss": 0.3475,
+      "step": 2408
+    },
+    {
+      "epoch": 3.466187050359712,
+      "grad_norm": 0.1468398185670848,
+      "learning_rate": 2.082923892272124e-05,
+      "loss": 0.3636,
+      "step": 2409
+    },
+    {
+      "epoch": 3.4676258992805753,
+      "grad_norm": 0.12952582836962553,
+      "learning_rate": 2.079397805062526e-05,
+      "loss": 0.3576,
+      "step": 2410
+    },
+    {
+      "epoch": 3.469064748201439,
+      "grad_norm": 0.1618036519607178,
+      "learning_rate": 2.0758736564223937e-05,
+      "loss": 0.36,
+      "step": 2411
+    },
+    {
+      "epoch": 3.470503597122302,
+      "grad_norm": 0.13052504625988254,
+      "learning_rate": 2.0723514499088388e-05,
+      "loss": 0.357,
+      "step": 2412
+    },
+    {
+      "epoch": 3.4719424460431654,
+      "grad_norm": 0.14981628312349463,
+      "learning_rate": 2.068831189077021e-05,
+      "loss": 0.3569,
+      "step": 2413
+    },
+    {
+      "epoch": 3.4733812949640286,
+      "grad_norm": 0.14676353418467492,
+      "learning_rate": 2.065312877480133e-05,
+      "loss": 0.3642,
+      "step": 2414
+    },
+    {
+      "epoch": 3.4748201438848922,
+      "grad_norm": 0.12580317862821208,
+      "learning_rate": 2.0617965186694e-05,
+      "loss": 0.3619,
+      "step": 2415
+    },
+    {
+      "epoch": 3.4762589928057555,
+      "grad_norm": 0.16397422943318538,
+      "learning_rate": 2.058282116194076e-05,
+      "loss": 0.3579,
+      "step": 2416
+    },
+    {
+      "epoch": 3.4776978417266187,
+      "grad_norm": 0.12879245245884377,
+      "learning_rate": 2.0547696736014415e-05,
+      "loss": 0.3616,
+      "step": 2417
+    },
+    {
+      "epoch": 3.479136690647482,
+      "grad_norm": 0.16143321846998793,
+      "learning_rate": 2.0512591944367976e-05,
+      "loss": 0.3607,
+      "step": 2418
+    },
+    {
+      "epoch": 3.4805755395683455,
+      "grad_norm": 0.14541572120664673,
+      "learning_rate": 2.0477506822434644e-05,
+      "loss": 0.3632,
+      "step": 2419
+    },
+    {
+      "epoch": 3.4820143884892087,
+      "grad_norm": 0.140853997077851,
+      "learning_rate": 2.0442441405627776e-05,
+      "loss": 0.3644,
+      "step": 2420
+    },
+    {
+      "epoch": 3.483453237410072,
+      "grad_norm": 0.16358711152968886,
+      "learning_rate": 2.0407395729340792e-05,
+      "loss": 0.3561,
+      "step": 2421
+    },
+    {
+      "epoch": 3.484892086330935,
+      "grad_norm": 0.12691895951240845,
+      "learning_rate": 2.037236982894723e-05,
+      "loss": 0.3694,
+      "step": 2422
+    },
+    {
+      "epoch": 3.4863309352517984,
+      "grad_norm": 0.12280266266571434,
+      "learning_rate": 2.0337363739800695e-05,
+      "loss": 0.36,
+      "step": 2423
+    },
+    {
+      "epoch": 3.487769784172662,
+      "grad_norm": 0.15523018091402865,
+      "learning_rate": 2.030237749723472e-05,
+      "loss": 0.3594,
+      "step": 2424
+    },
+    {
+      "epoch": 3.4892086330935252,
+      "grad_norm": 0.12599233392046144,
+      "learning_rate": 2.026741113656284e-05,
+      "loss": 0.3625,
+      "step": 2425
+    },
+    {
+      "epoch": 3.4906474820143885,
+      "grad_norm": 0.15043753377349986,
+      "learning_rate": 2.0232464693078578e-05,
+      "loss": 0.3579,
+      "step": 2426
+    },
+    {
+      "epoch": 3.4920863309352517,
+      "grad_norm": 0.12535904059882044,
+      "learning_rate": 2.0197538202055246e-05,
+      "loss": 0.3543,
+      "step": 2427
+    },
+    {
+      "epoch": 3.493525179856115,
+      "grad_norm": 0.14685558418663916,
+      "learning_rate": 2.01626316987461e-05,
+      "loss": 0.357,
+      "step": 2428
+    },
+    {
+      "epoch": 3.4949640287769785,
+      "grad_norm": 0.17985442805567875,
+      "learning_rate": 2.0127745218384193e-05,
+      "loss": 0.3668,
+      "step": 2429
+    },
+    {
+      "epoch": 3.4964028776978417,
+      "grad_norm": 0.13826582257846703,
+      "learning_rate": 2.009287879618236e-05,
+      "loss": 0.3613,
+      "step": 2430
+    },
+    {
+      "epoch": 3.497841726618705,
+      "grad_norm": 0.14923428690400006,
+      "learning_rate": 2.0058032467333204e-05,
+      "loss": 0.3682,
+      "step": 2431
+    },
+    {
+      "epoch": 3.499280575539568,
+      "grad_norm": 0.13411374381163862,
+      "learning_rate": 2.0023206267009056e-05,
+      "loss": 0.3585,
+      "step": 2432
+    },
+    {
+      "epoch": 3.5007194244604314,
+      "grad_norm": 0.18295002413954828,
+      "learning_rate": 1.9988400230361872e-05,
+      "loss": 0.3618,
+      "step": 2433
+    },
+    {
+      "epoch": 3.502158273381295,
+      "grad_norm": 0.15045896811357695,
+      "learning_rate": 1.9953614392523345e-05,
+      "loss": 0.3583,
+      "step": 2434
+    },
+    {
+      "epoch": 3.5035971223021583,
+      "grad_norm": 0.1807016161535834,
+      "learning_rate": 1.9918848788604738e-05,
+      "loss": 0.3594,
+      "step": 2435
+    },
+    {
+      "epoch": 3.5050359712230215,
+      "grad_norm": 0.13053791305177803,
+      "learning_rate": 1.9884103453696837e-05,
+      "loss": 0.3605,
+      "step": 2436
+    },
+    {
+      "epoch": 3.506474820143885,
+      "grad_norm": 0.15640401309251897,
+      "learning_rate": 1.9849378422870082e-05,
+      "loss": 0.3607,
+      "step": 2437
+    },
+    {
+      "epoch": 3.5079136690647483,
+      "grad_norm": 0.13300910014485748,
+      "learning_rate": 1.9814673731174315e-05,
+      "loss": 0.3586,
+      "step": 2438
+    },
+    {
+      "epoch": 3.5093525179856115,
+      "grad_norm": 0.14174317411297754,
+      "learning_rate": 1.97799894136389e-05,
+      "loss": 0.3605,
+      "step": 2439
+    },
+    {
+      "epoch": 3.5107913669064748,
+      "grad_norm": 0.12256497862548782,
+      "learning_rate": 1.9745325505272633e-05,
+      "loss": 0.3627,
+      "step": 2440
+    },
+    {
+      "epoch": 3.512230215827338,
+      "grad_norm": 0.14283792503707346,
+      "learning_rate": 1.9710682041063705e-05,
+      "loss": 0.3602,
+      "step": 2441
+    },
+    {
+      "epoch": 3.5136690647482016,
+      "grad_norm": 0.1268181159687369,
+      "learning_rate": 1.9676059055979663e-05,
+      "loss": 0.3605,
+      "step": 2442
+    },
+    {
+      "epoch": 3.515107913669065,
+      "grad_norm": 0.14306929986034367,
+      "learning_rate": 1.9641456584967392e-05,
+      "loss": 0.3775,
+      "step": 2443
+    },
+    {
+      "epoch": 3.516546762589928,
+      "grad_norm": 0.14205093798260596,
+      "learning_rate": 1.9606874662953076e-05,
+      "loss": 0.3631,
+      "step": 2444
+    },
+    {
+      "epoch": 3.5179856115107913,
+      "grad_norm": 0.13463326265482048,
+      "learning_rate": 1.9572313324842148e-05,
+      "loss": 0.3555,
+      "step": 2445
+    },
+    {
+      "epoch": 3.5194244604316545,
+      "grad_norm": 0.11819326222058735,
+      "learning_rate": 1.9537772605519285e-05,
+      "loss": 0.3703,
+      "step": 2446
+    },
+    {
+      "epoch": 3.520863309352518,
+      "grad_norm": 0.22688031401747902,
+      "learning_rate": 1.950325253984828e-05,
+      "loss": 0.37,
+      "step": 2447
+    },
+    {
+      "epoch": 3.5223021582733813,
+      "grad_norm": 0.135085886606042,
+      "learning_rate": 1.946875316267219e-05,
+      "loss": 0.3554,
+      "step": 2448
+    },
+    {
+      "epoch": 3.5237410071942445,
+      "grad_norm": 0.12712478217185483,
+      "learning_rate": 1.9434274508813135e-05,
+      "loss": 0.3596,
+      "step": 2449
+    },
+    {
+      "epoch": 3.5251798561151078,
+      "grad_norm": 0.12909468816223446,
+      "learning_rate": 1.9399816613072287e-05,
+      "loss": 0.3515,
+      "step": 2450
+    },
+    {
+      "epoch": 3.526618705035971,
+      "grad_norm": 0.14655535569434083,
+      "learning_rate": 1.9365379510229888e-05,
+      "loss": 0.3649,
+      "step": 2451
+    },
+    {
+      "epoch": 3.5280575539568346,
+      "grad_norm": 0.12981482793178992,
+      "learning_rate": 1.9330963235045253e-05,
+      "loss": 0.371,
+      "step": 2452
+    },
+    {
+      "epoch": 3.529496402877698,
+      "grad_norm": 0.16364626318779124,
+      "learning_rate": 1.9296567822256577e-05,
+      "loss": 0.3742,
+      "step": 2453
+    },
+    {
+      "epoch": 3.530935251798561,
+      "grad_norm": 0.11319185092004506,
+      "learning_rate": 1.9262193306581052e-05,
+      "loss": 0.3622,
+      "step": 2454
+    },
+    {
+      "epoch": 3.5323741007194247,
+      "grad_norm": 0.13702614039758187,
+      "learning_rate": 1.922783972271477e-05,
+      "loss": 0.3515,
+      "step": 2455
+    },
+    {
+      "epoch": 3.533812949640288,
+      "grad_norm": 0.12785770585094197,
+      "learning_rate": 1.9193507105332702e-05,
+      "loss": 0.3592,
+      "step": 2456
+    },
+    {
+      "epoch": 3.535251798561151,
+      "grad_norm": 0.14364796412529102,
+      "learning_rate": 1.9159195489088636e-05,
+      "loss": 0.3678,
+      "step": 2457
+    },
+    {
+      "epoch": 3.5366906474820143,
+      "grad_norm": 0.1311224626798062,
+      "learning_rate": 1.9124904908615178e-05,
+      "loss": 0.3638,
+      "step": 2458
+    },
+    {
+      "epoch": 3.5381294964028775,
+      "grad_norm": 0.1143751908426819,
+      "learning_rate": 1.9090635398523698e-05,
+      "loss": 0.3542,
+      "step": 2459
+    },
+    {
+      "epoch": 3.539568345323741,
+      "grad_norm": 0.11960495133083206,
+      "learning_rate": 1.9056386993404294e-05,
+      "loss": 0.3513,
+      "step": 2460
+    },
+    {
+      "epoch": 3.5410071942446044,
+      "grad_norm": 0.1242060469217888,
+      "learning_rate": 1.902215972782579e-05,
+      "loss": 0.3584,
+      "step": 2461
+    },
+    {
+      "epoch": 3.5424460431654676,
+      "grad_norm": 0.11815072560041254,
+      "learning_rate": 1.8987953636335595e-05,
+      "loss": 0.3661,
+      "step": 2462
+    },
+    {
+      "epoch": 3.543884892086331,
+      "grad_norm": 0.12024541003788661,
+      "learning_rate": 1.8953768753459863e-05,
+      "loss": 0.365,
+      "step": 2463
+    },
+    {
+      "epoch": 3.545323741007194,
+      "grad_norm": 0.11095016672614343,
+      "learning_rate": 1.8919605113703227e-05,
+      "loss": 0.3611,
+      "step": 2464
+    },
+    {
+      "epoch": 3.5467625899280577,
+      "grad_norm": 0.12991405641396,
+      "learning_rate": 1.888546275154895e-05,
+      "loss": 0.3682,
+      "step": 2465
+    },
+    {
+      "epoch": 3.548201438848921,
+      "grad_norm": 0.1119292554235309,
+      "learning_rate": 1.885134170145879e-05,
+      "loss": 0.3538,
+      "step": 2466
+    },
+    {
+      "epoch": 3.549640287769784,
+      "grad_norm": 0.12423616963521807,
+      "learning_rate": 1.8817241997873007e-05,
+      "loss": 0.3617,
+      "step": 2467
+    },
+    {
+      "epoch": 3.5510791366906473,
+      "grad_norm": 0.11604277374417908,
+      "learning_rate": 1.8783163675210307e-05,
+      "loss": 0.3602,
+      "step": 2468
+    },
+    {
+      "epoch": 3.5525179856115106,
+      "grad_norm": 0.1372858234476605,
+      "learning_rate": 1.8749106767867808e-05,
+      "loss": 0.3678,
+      "step": 2469
+    },
+    {
+      "epoch": 3.553956834532374,
+      "grad_norm": 0.10744657188136675,
+      "learning_rate": 1.871507131022103e-05,
+      "loss": 0.364,
+      "step": 2470
+    },
+    {
+      "epoch": 3.5553956834532374,
+      "grad_norm": 0.12982426605734915,
+      "learning_rate": 1.8681057336623825e-05,
+      "loss": 0.3592,
+      "step": 2471
+    },
+    {
+      "epoch": 3.5568345323741006,
+      "grad_norm": 0.11708731330532512,
+      "learning_rate": 1.864706488140839e-05,
+      "loss": 0.3629,
+      "step": 2472
+    },
+    {
+      "epoch": 3.5582733812949643,
+      "grad_norm": 0.12605064899795568,
+      "learning_rate": 1.861309397888513e-05,
+      "loss": 0.3608,
+      "step": 2473
+    },
+    {
+      "epoch": 3.5597122302158275,
+      "grad_norm": 0.12874151067946632,
+      "learning_rate": 1.857914466334279e-05,
+      "loss": 0.3584,
+      "step": 2474
+    },
+    {
+      "epoch": 3.5611510791366907,
+      "grad_norm": 0.13231097084358348,
+      "learning_rate": 1.8545216969048288e-05,
+      "loss": 0.369,
+      "step": 2475
+    },
+    {
+      "epoch": 3.562589928057554,
+      "grad_norm": 0.12529594305664066,
+      "learning_rate": 1.851131093024668e-05,
+      "loss": 0.3591,
+      "step": 2476
+    },
+    {
+      "epoch": 3.564028776978417,
+      "grad_norm": 0.11728122094530871,
+      "learning_rate": 1.8477426581161192e-05,
+      "loss": 0.3605,
+      "step": 2477
+    },
+    {
+      "epoch": 3.565467625899281,
+      "grad_norm": 0.13442675436700735,
+      "learning_rate": 1.844356395599322e-05,
+      "loss": 0.3633,
+      "step": 2478
+    },
+    {
+      "epoch": 3.566906474820144,
+      "grad_norm": 0.12863775394917398,
+      "learning_rate": 1.840972308892211e-05,
+      "loss": 0.3738,
+      "step": 2479
+    },
+    {
+      "epoch": 3.568345323741007,
+      "grad_norm": 0.13669933094304723,
+      "learning_rate": 1.837590401410532e-05,
+      "loss": 0.3646,
+      "step": 2480
+    },
+    {
+      "epoch": 3.5697841726618704,
+      "grad_norm": 0.13848020004952785,
+      "learning_rate": 1.8342106765678358e-05,
+      "loss": 0.3647,
+      "step": 2481
+    },
+    {
+      "epoch": 3.5712230215827336,
+      "grad_norm": 0.11722366973269575,
+      "learning_rate": 1.8308331377754584e-05,
+      "loss": 0.3578,
+      "step": 2482
+    },
+    {
+      "epoch": 3.5726618705035973,
+      "grad_norm": 0.11214821744928535,
+      "learning_rate": 1.8274577884425383e-05,
+      "loss": 0.3624,
+      "step": 2483
+    },
+    {
+      "epoch": 3.5741007194244605,
+      "grad_norm": 0.1342441684020052,
+      "learning_rate": 1.8240846319760012e-05,
+      "loss": 0.3604,
+      "step": 2484
+    },
+    {
+      "epoch": 3.5755395683453237,
+      "grad_norm": 0.12462232422906017,
+      "learning_rate": 1.8207136717805585e-05,
+      "loss": 0.3511,
+      "step": 2485
+    },
+    {
+      "epoch": 3.576978417266187,
+      "grad_norm": 0.12892567568478816,
+      "learning_rate": 1.8173449112587062e-05,
+      "loss": 0.3539,
+      "step": 2486
+    },
+    {
+      "epoch": 3.57841726618705,
+      "grad_norm": 0.13031483613689354,
+      "learning_rate": 1.813978353810722e-05,
+      "loss": 0.3647,
+      "step": 2487
+    },
+    {
+      "epoch": 3.579856115107914,
+      "grad_norm": 0.14172223735283349,
+      "learning_rate": 1.8106140028346526e-05,
+      "loss": 0.3711,
+      "step": 2488
+    },
+    {
+      "epoch": 3.581294964028777,
+      "grad_norm": 0.13642062629442642,
+      "learning_rate": 1.8072518617263276e-05,
+      "loss": 0.3596,
+      "step": 2489
+    },
+    {
+      "epoch": 3.58273381294964,
+      "grad_norm": 0.14056365127860532,
+      "learning_rate": 1.803891933879338e-05,
+      "loss": 0.3683,
+      "step": 2490
+    },
+    {
+      "epoch": 3.584172661870504,
+      "grad_norm": 0.13568708568140467,
+      "learning_rate": 1.8005342226850423e-05,
+      "loss": 0.3634,
+      "step": 2491
+    },
+    {
+      "epoch": 3.5856115107913666,
+      "grad_norm": 0.11816877259119497,
+      "learning_rate": 1.7971787315325684e-05,
+      "loss": 0.3504,
+      "step": 2492
+    },
+    {
+      "epoch": 3.5870503597122303,
+      "grad_norm": 0.12781359007090007,
+      "learning_rate": 1.7938254638087946e-05,
+      "loss": 0.3685,
+      "step": 2493
+    },
+    {
+      "epoch": 3.5884892086330935,
+      "grad_norm": 0.11390141299480655,
+      "learning_rate": 1.7904744228983585e-05,
+      "loss": 0.3696,
+      "step": 2494
+    },
+    {
+      "epoch": 3.5899280575539567,
+      "grad_norm": 0.11336779470702313,
+      "learning_rate": 1.7871256121836507e-05,
+      "loss": 0.3541,
+      "step": 2495
+    },
+    {
+      "epoch": 3.5913669064748204,
+      "grad_norm": 0.1138319923084464,
+      "learning_rate": 1.7837790350448098e-05,
+      "loss": 0.3604,
+      "step": 2496
+    },
+    {
+      "epoch": 3.5928057553956836,
+      "grad_norm": 0.11728097483001036,
+      "learning_rate": 1.7804346948597206e-05,
+      "loss": 0.3589,
+      "step": 2497
+    },
+    {
+      "epoch": 3.594244604316547,
+      "grad_norm": 0.10378891310250349,
+      "learning_rate": 1.7770925950040114e-05,
+      "loss": 0.3463,
+      "step": 2498
+    },
+    {
+      "epoch": 3.59568345323741,
+      "grad_norm": 0.11345324723403843,
+      "learning_rate": 1.773752738851042e-05,
+      "loss": 0.3596,
+      "step": 2499
+    },
+    {
+      "epoch": 3.597122302158273,
+      "grad_norm": 0.11309436317467,
+      "learning_rate": 1.770415129771918e-05,
+      "loss": 0.3695,
+      "step": 2500
+    },
+    {
+      "epoch": 3.598561151079137,
+      "grad_norm": 0.10853114682496083,
+      "learning_rate": 1.7670797711354724e-05,
+      "loss": 0.3643,
+      "step": 2501
+    },
+    {
+      "epoch": 3.6,
+      "grad_norm": 0.13250831651816233,
+      "learning_rate": 1.763746666308261e-05,
+      "loss": 0.3549,
+      "step": 2502
+    },
+    {
+      "epoch": 3.6014388489208633,
+      "grad_norm": 0.12205829710881429,
+      "learning_rate": 1.760415818654574e-05,
+      "loss": 0.3704,
+      "step": 2503
+    },
+    {
+      "epoch": 3.6028776978417265,
+      "grad_norm": 0.12321635496401367,
+      "learning_rate": 1.75708723153642e-05,
+      "loss": 0.3627,
+      "step": 2504
+    },
+    {
+      "epoch": 3.6043165467625897,
+      "grad_norm": 0.12800767751956713,
+      "learning_rate": 1.7537609083135224e-05,
+      "loss": 0.3597,
+      "step": 2505
+    },
+    {
+      "epoch": 3.6057553956834534,
+      "grad_norm": 0.13900200317563696,
+      "learning_rate": 1.7504368523433216e-05,
+      "loss": 0.353,
+      "step": 2506
+    },
+    {
+      "epoch": 3.6071942446043166,
+      "grad_norm": 0.12316202576985372,
+      "learning_rate": 1.747115066980974e-05,
+      "loss": 0.3586,
+      "step": 2507
+    },
+    {
+      "epoch": 3.60863309352518,
+      "grad_norm": 0.13760818995581628,
+      "learning_rate": 1.7437955555793372e-05,
+      "loss": 0.3555,
+      "step": 2508
+    },
+    {
+      "epoch": 3.610071942446043,
+      "grad_norm": 0.11355210115229894,
+      "learning_rate": 1.740478321488978e-05,
+      "loss": 0.3555,
+      "step": 2509
+    },
+    {
+      "epoch": 3.6115107913669062,
+      "grad_norm": 0.15416602094829648,
+      "learning_rate": 1.737163368058162e-05,
+      "loss": 0.3603,
+      "step": 2510
+    },
+    {
+      "epoch": 3.61294964028777,
+      "grad_norm": 0.11189006173598844,
+      "learning_rate": 1.7338506986328552e-05,
+      "loss": 0.359,
+      "step": 2511
+    },
+    {
+      "epoch": 3.614388489208633,
+      "grad_norm": 0.1468371487818277,
+      "learning_rate": 1.730540316556717e-05,
+      "loss": 0.3626,
+      "step": 2512
+    },
+    {
+      "epoch": 3.6158273381294963,
+      "grad_norm": 0.11321436813944175,
+      "learning_rate": 1.727232225171098e-05,
+      "loss": 0.3627,
+      "step": 2513
+    },
+    {
+      "epoch": 3.61726618705036,
+      "grad_norm": 0.11380782275839862,
+      "learning_rate": 1.7239264278150364e-05,
+      "loss": 0.3588,
+      "step": 2514
+    },
+    {
+      "epoch": 3.618705035971223,
+      "grad_norm": 0.12040023848313533,
+      "learning_rate": 1.7206229278252577e-05,
+      "loss": 0.3609,
+      "step": 2515
+    },
+    {
+      "epoch": 3.6201438848920864,
+      "grad_norm": 0.11452473435933457,
+      "learning_rate": 1.717321728536163e-05,
+      "loss": 0.3637,
+      "step": 2516
+    },
+    {
+      "epoch": 3.6215827338129496,
+      "grad_norm": 0.11317997303110827,
+      "learning_rate": 1.7140228332798336e-05,
+      "loss": 0.3622,
+      "step": 2517
+    },
+    {
+      "epoch": 3.623021582733813,
+      "grad_norm": 0.12973695947641795,
+      "learning_rate": 1.7107262453860308e-05,
+      "loss": 0.3674,
+      "step": 2518
+    },
+    {
+      "epoch": 3.6244604316546765,
+      "grad_norm": 0.11362680022756308,
+      "learning_rate": 1.707431968182179e-05,
+      "loss": 0.3581,
+      "step": 2519
+    },
+    {
+      "epoch": 3.6258992805755397,
+      "grad_norm": 0.13445202472189502,
+      "learning_rate": 1.7041400049933726e-05,
+      "loss": 0.3585,
+      "step": 2520
+    },
+    {
+      "epoch": 3.627338129496403,
+      "grad_norm": 0.12766773295452266,
+      "learning_rate": 1.700850359142373e-05,
+      "loss": 0.369,
+      "step": 2521
+    },
+    {
+      "epoch": 3.628776978417266,
+      "grad_norm": 0.1281623527561033,
+      "learning_rate": 1.6975630339496e-05,
+      "loss": 0.3586,
+      "step": 2522
+    },
+    {
+      "epoch": 3.6302158273381293,
+      "grad_norm": 0.12272089410013572,
+      "learning_rate": 1.6942780327331317e-05,
+      "loss": 0.3667,
+      "step": 2523
+    },
+    {
+      "epoch": 3.631654676258993,
+      "grad_norm": 0.12798702385340094,
+      "learning_rate": 1.6909953588087024e-05,
+      "loss": 0.3597,
+      "step": 2524
+    },
+    {
+      "epoch": 3.633093525179856,
+      "grad_norm": 0.11597463713977145,
+      "learning_rate": 1.687715015489691e-05,
+      "loss": 0.3597,
+      "step": 2525
+    },
+    {
+      "epoch": 3.6345323741007194,
+      "grad_norm": 0.11928967616784174,
+      "learning_rate": 1.6844370060871324e-05,
+      "loss": 0.3619,
+      "step": 2526
+    },
+    {
+      "epoch": 3.6359712230215826,
+      "grad_norm": 0.12846214158720146,
+      "learning_rate": 1.6811613339097022e-05,
+      "loss": 0.3572,
+      "step": 2527
+    },
+    {
+      "epoch": 3.637410071942446,
+      "grad_norm": 0.11488451532344186,
+      "learning_rate": 1.6778880022637123e-05,
+      "loss": 0.3549,
+      "step": 2528
+    },
+    {
+      "epoch": 3.6388489208633095,
+      "grad_norm": 0.13407144008798366,
+      "learning_rate": 1.674617014453121e-05,
+      "loss": 0.365,
+      "step": 2529
+    },
+    {
+      "epoch": 3.6402877697841727,
+      "grad_norm": 0.1355485639637215,
+      "learning_rate": 1.6713483737795155e-05,
+      "loss": 0.3632,
+      "step": 2530
+    },
+    {
+      "epoch": 3.641726618705036,
+      "grad_norm": 0.10844701831092295,
+      "learning_rate": 1.6680820835421124e-05,
+      "loss": 0.3573,
+      "step": 2531
+    },
+    {
+      "epoch": 3.6431654676258995,
+      "grad_norm": 0.16354547575824532,
+      "learning_rate": 1.664818147037758e-05,
+      "loss": 0.36,
+      "step": 2532
+    },
+    {
+      "epoch": 3.6446043165467623,
+      "grad_norm": 0.11764460107006486,
+      "learning_rate": 1.6615565675609272e-05,
+      "loss": 0.3566,
+      "step": 2533
+    },
+    {
+      "epoch": 3.646043165467626,
+      "grad_norm": 0.14988060309989518,
+      "learning_rate": 1.6582973484037076e-05,
+      "loss": 0.3577,
+      "step": 2534
+    },
+    {
+      "epoch": 3.647482014388489,
+      "grad_norm": 0.11848140017555707,
+      "learning_rate": 1.6550404928558094e-05,
+      "loss": 0.3649,
+      "step": 2535
+    },
+    {
+      "epoch": 3.6489208633093524,
+      "grad_norm": 0.15544540214821592,
+      "learning_rate": 1.6517860042045564e-05,
+      "loss": 0.3635,
+      "step": 2536
+    },
+    {
+      "epoch": 3.650359712230216,
+      "grad_norm": 0.11932637817742124,
+      "learning_rate": 1.6485338857348826e-05,
+      "loss": 0.3666,
+      "step": 2537
+    },
+    {
+      "epoch": 3.6517985611510793,
+      "grad_norm": 0.10945585272046333,
+      "learning_rate": 1.6452841407293307e-05,
+      "loss": 0.352,
+      "step": 2538
+    },
+    {
+      "epoch": 3.6532374100719425,
+      "grad_norm": 0.1602187491110357,
+      "learning_rate": 1.642036772468047e-05,
+      "loss": 0.3668,
+      "step": 2539
+    },
+    {
+      "epoch": 3.6546762589928057,
+      "grad_norm": 0.11601223141768494,
+      "learning_rate": 1.6387917842287783e-05,
+      "loss": 0.3571,
+      "step": 2540
+    },
+    {
+      "epoch": 3.656115107913669,
+      "grad_norm": 0.14922969910297568,
+      "learning_rate": 1.635549179286871e-05,
+      "loss": 0.3669,
+      "step": 2541
+    },
+    {
+      "epoch": 3.6575539568345325,
+      "grad_norm": 0.11373684076883384,
+      "learning_rate": 1.6323089609152648e-05,
+      "loss": 0.3696,
+      "step": 2542
+    },
+    {
+      "epoch": 3.6589928057553958,
+      "grad_norm": 0.14063225178210415,
+      "learning_rate": 1.6290711323844866e-05,
+      "loss": 0.3708,
+      "step": 2543
+    },
+    {
+      "epoch": 3.660431654676259,
+      "grad_norm": 0.11673634471305658,
+      "learning_rate": 1.6258356969626614e-05,
+      "loss": 0.3691,
+      "step": 2544
+    },
+    {
+      "epoch": 3.661870503597122,
+      "grad_norm": 0.14829695978497334,
+      "learning_rate": 1.622602657915487e-05,
+      "loss": 0.3722,
+      "step": 2545
+    },
+    {
+      "epoch": 3.6633093525179854,
+      "grad_norm": 0.1394931984719001,
+      "learning_rate": 1.6193720185062484e-05,
+      "loss": 0.3681,
+      "step": 2546
+    },
+    {
+      "epoch": 3.664748201438849,
+      "grad_norm": 0.13122276548358858,
+      "learning_rate": 1.6161437819958087e-05,
+      "loss": 0.3655,
+      "step": 2547
+    },
+    {
+      "epoch": 3.6661870503597123,
+      "grad_norm": 0.14332210065033338,
+      "learning_rate": 1.6129179516426048e-05,
+      "loss": 0.3634,
+      "step": 2548
+    },
+    {
+      "epoch": 3.6676258992805755,
+      "grad_norm": 0.1492095853609529,
+      "learning_rate": 1.609694530702644e-05,
+      "loss": 0.3653,
+      "step": 2549
+    },
+    {
+      "epoch": 3.6690647482014387,
+      "grad_norm": 0.1253180629490774,
+      "learning_rate": 1.6064735224295027e-05,
+      "loss": 0.3606,
+      "step": 2550
+    },
+    {
+      "epoch": 3.670503597122302,
+      "grad_norm": 0.14174121524182898,
+      "learning_rate": 1.603254930074322e-05,
+      "loss": 0.3599,
+      "step": 2551
+    },
+    {
+      "epoch": 3.6719424460431656,
+      "grad_norm": 0.13329892305991034,
+      "learning_rate": 1.6000387568858042e-05,
+      "loss": 0.3552,
+      "step": 2552
+    },
+    {
+      "epoch": 3.6733812949640288,
+      "grad_norm": 0.14095030488587826,
+      "learning_rate": 1.5968250061102105e-05,
+      "loss": 0.3622,
+      "step": 2553
+    },
+    {
+      "epoch": 3.674820143884892,
+      "grad_norm": 0.130119087262895,
+      "learning_rate": 1.593613680991353e-05,
+      "loss": 0.3616,
+      "step": 2554
+    },
+    {
+      "epoch": 3.6762589928057556,
+      "grad_norm": 0.10573893978496751,
+      "learning_rate": 1.590404784770603e-05,
+      "loss": 0.3562,
+      "step": 2555
+    },
+    {
+      "epoch": 3.677697841726619,
+      "grad_norm": 0.1425703024959088,
+      "learning_rate": 1.5871983206868756e-05,
+      "loss": 0.3598,
+      "step": 2556
+    },
+    {
+      "epoch": 3.679136690647482,
+      "grad_norm": 0.14003753245179645,
+      "learning_rate": 1.583994291976629e-05,
+      "loss": 0.3733,
+      "step": 2557
+    },
+    {
+      "epoch": 3.6805755395683453,
+      "grad_norm": 0.10566309600002534,
+      "learning_rate": 1.580792701873865e-05,
+      "loss": 0.3595,
+      "step": 2558
+    },
+    {
+      "epoch": 3.6820143884892085,
+      "grad_norm": 0.1337458720644581,
+      "learning_rate": 1.5775935536101296e-05,
+      "loss": 0.359,
+      "step": 2559
+    },
+    {
+      "epoch": 3.683453237410072,
+      "grad_norm": 0.13452100949385928,
+      "learning_rate": 1.5743968504144946e-05,
+      "loss": 0.3602,
+      "step": 2560
+    },
+    {
+      "epoch": 3.6848920863309353,
+      "grad_norm": 0.1141164300059276,
+      "learning_rate": 1.57120259551357e-05,
+      "loss": 0.3588,
+      "step": 2561
+    },
+    {
+      "epoch": 3.6863309352517986,
+      "grad_norm": 0.12200071347526566,
+      "learning_rate": 1.5680107921314926e-05,
+      "loss": 0.3641,
+      "step": 2562
+    },
+    {
+      "epoch": 3.6877697841726618,
+      "grad_norm": 0.12161005020780033,
+      "learning_rate": 1.5648214434899257e-05,
+      "loss": 0.3629,
+      "step": 2563
+    },
+    {
+      "epoch": 3.689208633093525,
+      "grad_norm": 0.10535673795518845,
+      "learning_rate": 1.5616345528080537e-05,
+      "loss": 0.3532,
+      "step": 2564
+    },
+    {
+      "epoch": 3.6906474820143886,
+      "grad_norm": 0.13843402653199047,
+      "learning_rate": 1.5584501233025813e-05,
+      "loss": 0.3618,
+      "step": 2565
+    },
+    {
+      "epoch": 3.692086330935252,
+      "grad_norm": 0.10102346363401019,
+      "learning_rate": 1.555268158187728e-05,
+      "loss": 0.3546,
+      "step": 2566
+    },
+    {
+      "epoch": 3.693525179856115,
+      "grad_norm": 0.13534669321488643,
+      "learning_rate": 1.552088660675227e-05,
+      "loss": 0.3673,
+      "step": 2567
+    },
+    {
+      "epoch": 3.6949640287769783,
+      "grad_norm": 0.1037905989181797,
+      "learning_rate": 1.54891163397432e-05,
+      "loss": 0.3578,
+      "step": 2568
+    },
+    {
+      "epoch": 3.6964028776978415,
+      "grad_norm": 0.12973135784699819,
+      "learning_rate": 1.5457370812917526e-05,
+      "loss": 0.3578,
+      "step": 2569
+    },
+    {
+      "epoch": 3.697841726618705,
+      "grad_norm": 0.10545431149666043,
+      "learning_rate": 1.5425650058317795e-05,
+      "loss": 0.3618,
+      "step": 2570
+    },
+    {
+      "epoch": 3.6992805755395683,
+      "grad_norm": 0.11620996412674596,
+      "learning_rate": 1.5393954107961467e-05,
+      "loss": 0.3628,
+      "step": 2571
+    },
+    {
+      "epoch": 3.7007194244604316,
+      "grad_norm": 0.12248263561229705,
+      "learning_rate": 1.536228299384102e-05,
+      "loss": 0.3535,
+      "step": 2572
+    },
+    {
+      "epoch": 3.702158273381295,
+      "grad_norm": 0.12492464754252347,
+      "learning_rate": 1.533063674792389e-05,
+      "loss": 0.3652,
+      "step": 2573
+    },
+    {
+      "epoch": 3.7035971223021584,
+      "grad_norm": 0.1374768174174054,
+      "learning_rate": 1.529901540215233e-05,
+      "loss": 0.353,
+      "step": 2574
+    },
+    {
+      "epoch": 3.7050359712230216,
+      "grad_norm": 0.11331164182765775,
+      "learning_rate": 1.5267418988443517e-05,
+      "loss": 0.3717,
+      "step": 2575
+    },
+    {
+      "epoch": 3.706474820143885,
+      "grad_norm": 0.12210755308569196,
+      "learning_rate": 1.5235847538689452e-05,
+      "loss": 0.3628,
+      "step": 2576
+    },
+    {
+      "epoch": 3.707913669064748,
+      "grad_norm": 0.13691028227148266,
+      "learning_rate": 1.5204301084756936e-05,
+      "loss": 0.3664,
+      "step": 2577
+    },
+    {
+      "epoch": 3.7093525179856117,
+      "grad_norm": 0.12456712506114138,
+      "learning_rate": 1.5172779658487539e-05,
+      "loss": 0.3611,
+      "step": 2578
+    },
+    {
+      "epoch": 3.710791366906475,
+      "grad_norm": 0.11450658605211136,
+      "learning_rate": 1.5141283291697587e-05,
+      "loss": 0.3589,
+      "step": 2579
+    },
+    {
+      "epoch": 3.712230215827338,
+      "grad_norm": 0.12082326327275167,
+      "learning_rate": 1.5109812016178053e-05,
+      "loss": 0.355,
+      "step": 2580
+    },
+    {
+      "epoch": 3.7136690647482014,
+      "grad_norm": 0.12871886244640798,
+      "learning_rate": 1.5078365863694667e-05,
+      "loss": 0.3593,
+      "step": 2581
+    },
+    {
+      "epoch": 3.7151079136690646,
+      "grad_norm": 0.12454464902255565,
+      "learning_rate": 1.5046944865987763e-05,
+      "loss": 0.36,
+      "step": 2582
+    },
+    {
+      "epoch": 3.716546762589928,
+      "grad_norm": 0.11084581787132927,
+      "learning_rate": 1.501554905477224e-05,
+      "loss": 0.3556,
+      "step": 2583
+    },
+    {
+      "epoch": 3.7179856115107914,
+      "grad_norm": 0.1085932755865407,
+      "learning_rate": 1.4984178461737663e-05,
+      "loss": 0.3735,
+      "step": 2584
+    },
+    {
+      "epoch": 3.7194244604316546,
+      "grad_norm": 0.11993697365057852,
+      "learning_rate": 1.4952833118548094e-05,
+      "loss": 0.3567,
+      "step": 2585
+    },
+    {
+      "epoch": 3.720863309352518,
+      "grad_norm": 0.13108234909015168,
+      "learning_rate": 1.492151305684208e-05,
+      "loss": 0.3612,
+      "step": 2586
+    },
+    {
+      "epoch": 3.722302158273381,
+      "grad_norm": 0.12257848253539193,
+      "learning_rate": 1.4890218308232704e-05,
+      "loss": 0.3657,
+      "step": 2587
+    },
+    {
+      "epoch": 3.7237410071942447,
+      "grad_norm": 0.10565647987771688,
+      "learning_rate": 1.4858948904307476e-05,
+      "loss": 0.3644,
+      "step": 2588
+    },
+    {
+      "epoch": 3.725179856115108,
+      "grad_norm": 0.15214811899217365,
+      "learning_rate": 1.4827704876628319e-05,
+      "loss": 0.3663,
+      "step": 2589
+    },
+    {
+      "epoch": 3.726618705035971,
+      "grad_norm": 0.11821078712737504,
+      "learning_rate": 1.4796486256731561e-05,
+      "loss": 0.3595,
+      "step": 2590
+    },
+    {
+      "epoch": 3.728057553956835,
+      "grad_norm": 0.12834242190669048,
+      "learning_rate": 1.4765293076127862e-05,
+      "loss": 0.3578,
+      "step": 2591
+    },
+    {
+      "epoch": 3.7294964028776976,
+      "grad_norm": 0.15115163524953665,
+      "learning_rate": 1.4734125366302224e-05,
+      "loss": 0.3616,
+      "step": 2592
+    },
+    {
+      "epoch": 3.7309352517985612,
+      "grad_norm": 0.13582216648067696,
+      "learning_rate": 1.470298315871392e-05,
+      "loss": 0.3618,
+      "step": 2593
+    },
+    {
+      "epoch": 3.7323741007194244,
+      "grad_norm": 0.14893276799316646,
+      "learning_rate": 1.4671866484796505e-05,
+      "loss": 0.3603,
+      "step": 2594
+    },
+    {
+      "epoch": 3.7338129496402876,
+      "grad_norm": 0.11681439057036415,
+      "learning_rate": 1.4640775375957742e-05,
+      "loss": 0.3623,
+      "step": 2595
+    },
+    {
+      "epoch": 3.7352517985611513,
+      "grad_norm": 0.14681579919735874,
+      "learning_rate": 1.4609709863579622e-05,
+      "loss": 0.3579,
+      "step": 2596
+    },
+    {
+      "epoch": 3.7366906474820145,
+      "grad_norm": 0.10968043209668547,
+      "learning_rate": 1.4578669979018231e-05,
+      "loss": 0.3595,
+      "step": 2597
+    },
+    {
+      "epoch": 3.7381294964028777,
+      "grad_norm": 0.12582209444001602,
+      "learning_rate": 1.454765575360385e-05,
+      "loss": 0.361,
+      "step": 2598
+    },
+    {
+      "epoch": 3.739568345323741,
+      "grad_norm": 0.13210425620212404,
+      "learning_rate": 1.4516667218640877e-05,
+      "loss": 0.3605,
+      "step": 2599
+    },
+    {
+      "epoch": 3.741007194244604,
+      "grad_norm": 0.11481028562948814,
+      "learning_rate": 1.4485704405407699e-05,
+      "loss": 0.36,
+      "step": 2600
+    },
+    {
+      "epoch": 3.742446043165468,
+      "grad_norm": 0.13825468448413022,
+      "learning_rate": 1.4454767345156806e-05,
+      "loss": 0.3611,
+      "step": 2601
+    },
+    {
+      "epoch": 3.743884892086331,
+      "grad_norm": 0.09926469784063446,
+      "learning_rate": 1.4423856069114677e-05,
+      "loss": 0.3552,
+      "step": 2602
+    },
+    {
+      "epoch": 3.7453237410071942,
+      "grad_norm": 0.15498737636073653,
+      "learning_rate": 1.4392970608481758e-05,
+      "loss": 0.3638,
+      "step": 2603
+    },
+    {
+      "epoch": 3.7467625899280574,
+      "grad_norm": 0.11990393870595022,
+      "learning_rate": 1.4362110994432445e-05,
+      "loss": 0.3591,
+      "step": 2604
+    },
+    {
+      "epoch": 3.7482014388489207,
+      "grad_norm": 0.12322270345908738,
+      "learning_rate": 1.433127725811505e-05,
+      "loss": 0.3567,
+      "step": 2605
+    },
+    {
+      "epoch": 3.7496402877697843,
+      "grad_norm": 0.12819082916741287,
+      "learning_rate": 1.4300469430651754e-05,
+      "loss": 0.3596,
+      "step": 2606
+    },
+    {
+      "epoch": 3.7510791366906475,
+      "grad_norm": 0.1212969449865843,
+      "learning_rate": 1.4269687543138594e-05,
+      "loss": 0.3578,
+      "step": 2607
+    },
+    {
+      "epoch": 3.7525179856115107,
+      "grad_norm": 0.15757106796906958,
+      "learning_rate": 1.4238931626645434e-05,
+      "loss": 0.3615,
+      "step": 2608
+    },
+    {
+      "epoch": 3.753956834532374,
+      "grad_norm": 0.14413627556955358,
+      "learning_rate": 1.4208201712215871e-05,
+      "loss": 0.3692,
+      "step": 2609
+    },
+    {
+      "epoch": 3.755395683453237,
+      "grad_norm": 0.1501816207712942,
+      "learning_rate": 1.4177497830867348e-05,
+      "loss": 0.366,
+      "step": 2610
+    },
+    {
+      "epoch": 3.756834532374101,
+      "grad_norm": 0.1311310678361702,
+      "learning_rate": 1.4146820013590973e-05,
+      "loss": 0.3541,
+      "step": 2611
+    },
+    {
+      "epoch": 3.758273381294964,
+      "grad_norm": 0.14674841924402438,
+      "learning_rate": 1.411616829135153e-05,
+      "loss": 0.36,
+      "step": 2612
+    },
+    {
+      "epoch": 3.7597122302158272,
+      "grad_norm": 0.13296654783018239,
+      "learning_rate": 1.4085542695087502e-05,
+      "loss": 0.358,
+      "step": 2613
+    },
+    {
+      "epoch": 3.761151079136691,
+      "grad_norm": 0.1358015049583282,
+      "learning_rate": 1.4054943255710987e-05,
+      "loss": 0.3608,
+      "step": 2614
+    },
+    {
+      "epoch": 3.762589928057554,
+      "grad_norm": 0.12831505931112983,
+      "learning_rate": 1.4024370004107683e-05,
+      "loss": 0.3697,
+      "step": 2615
+    },
+    {
+      "epoch": 3.7640287769784173,
+      "grad_norm": 0.1323113967521606,
+      "learning_rate": 1.3993822971136859e-05,
+      "loss": 0.3609,
+      "step": 2616
+    },
+    {
+      "epoch": 3.7654676258992805,
+      "grad_norm": 0.12769123081942618,
+      "learning_rate": 1.3963302187631316e-05,
+      "loss": 0.357,
+      "step": 2617
+    },
+    {
+      "epoch": 3.7669064748201437,
+      "grad_norm": 0.1407961080621416,
+      "learning_rate": 1.3932807684397348e-05,
+      "loss": 0.3538,
+      "step": 2618
+    },
+    {
+      "epoch": 3.7683453237410074,
+      "grad_norm": 0.12449058240087635,
+      "learning_rate": 1.3902339492214751e-05,
+      "loss": 0.3536,
+      "step": 2619
+    },
+    {
+      "epoch": 3.7697841726618706,
+      "grad_norm": 0.1326906224890556,
+      "learning_rate": 1.387189764183674e-05,
+      "loss": 0.3567,
+      "step": 2620
+    },
+    {
+      "epoch": 3.771223021582734,
+      "grad_norm": 0.11401026260483377,
+      "learning_rate": 1.384148216398995e-05,
+      "loss": 0.3654,
+      "step": 2621
+    },
+    {
+      "epoch": 3.772661870503597,
+      "grad_norm": 0.12168061074136714,
+      "learning_rate": 1.381109308937441e-05,
+      "loss": 0.3625,
+      "step": 2622
+    },
+    {
+      "epoch": 3.7741007194244602,
+      "grad_norm": 0.13387443422805737,
+      "learning_rate": 1.3780730448663456e-05,
+      "loss": 0.3636,
+      "step": 2623
+    },
+    {
+      "epoch": 3.775539568345324,
+      "grad_norm": 0.11349669555243698,
+      "learning_rate": 1.3750394272503775e-05,
+      "loss": 0.3608,
+      "step": 2624
+    },
+    {
+      "epoch": 3.776978417266187,
+      "grad_norm": 0.1181487230287458,
+      "learning_rate": 1.3720084591515374e-05,
+      "loss": 0.3607,
+      "step": 2625
+    },
+    {
+      "epoch": 3.7784172661870503,
+      "grad_norm": 0.13030452340281176,
+      "learning_rate": 1.3689801436291448e-05,
+      "loss": 0.3616,
+      "step": 2626
+    },
+    {
+      "epoch": 3.7798561151079135,
+      "grad_norm": 0.11183480055976384,
+      "learning_rate": 1.365954483739846e-05,
+      "loss": 0.3728,
+      "step": 2627
+    },
+    {
+      "epoch": 3.7812949640287767,
+      "grad_norm": 0.140740320543905,
+      "learning_rate": 1.3629314825376061e-05,
+      "loss": 0.3572,
+      "step": 2628
+    },
+    {
+      "epoch": 3.7827338129496404,
+      "grad_norm": 0.10512997696773077,
+      "learning_rate": 1.359911143073707e-05,
+      "loss": 0.3569,
+      "step": 2629
+    },
+    {
+      "epoch": 3.7841726618705036,
+      "grad_norm": 0.11841566025187138,
+      "learning_rate": 1.3568934683967427e-05,
+      "loss": 0.3542,
+      "step": 2630
+    },
+    {
+      "epoch": 3.785611510791367,
+      "grad_norm": 0.13508715727642095,
+      "learning_rate": 1.3538784615526188e-05,
+      "loss": 0.3584,
+      "step": 2631
+    },
+    {
+      "epoch": 3.7870503597122305,
+      "grad_norm": 0.12569299187321786,
+      "learning_rate": 1.3508661255845477e-05,
+      "loss": 0.3709,
+      "step": 2632
+    },
+    {
+      "epoch": 3.7884892086330937,
+      "grad_norm": 0.10732638862230746,
+      "learning_rate": 1.3478564635330455e-05,
+      "loss": 0.3503,
+      "step": 2633
+    },
+    {
+      "epoch": 3.789928057553957,
+      "grad_norm": 0.1251059736980278,
+      "learning_rate": 1.344849478435931e-05,
+      "loss": 0.3595,
+      "step": 2634
+    },
+    {
+      "epoch": 3.79136690647482,
+      "grad_norm": 0.11947824093913274,
+      "learning_rate": 1.3418451733283156e-05,
+      "loss": 0.3598,
+      "step": 2635
+    },
+    {
+      "epoch": 3.7928057553956833,
+      "grad_norm": 0.11577844202764537,
+      "learning_rate": 1.3388435512426142e-05,
+      "loss": 0.3652,
+      "step": 2636
+    },
+    {
+      "epoch": 3.794244604316547,
+      "grad_norm": 0.1337616812891817,
+      "learning_rate": 1.3358446152085289e-05,
+      "loss": 0.3497,
+      "step": 2637
+    },
+    {
+      "epoch": 3.79568345323741,
+      "grad_norm": 0.10072223135771624,
+      "learning_rate": 1.332848368253048e-05,
+      "loss": 0.3523,
+      "step": 2638
+    },
+    {
+      "epoch": 3.7971223021582734,
+      "grad_norm": 0.1236485929567498,
+      "learning_rate": 1.3298548134004498e-05,
+      "loss": 0.3573,
+      "step": 2639
+    },
+    {
+      "epoch": 3.7985611510791366,
+      "grad_norm": 0.10987681896024151,
+      "learning_rate": 1.326863953672294e-05,
+      "loss": 0.3684,
+      "step": 2640
+    },
+    {
+      "epoch": 3.8,
+      "grad_norm": 0.11930130321558292,
+      "learning_rate": 1.3238757920874203e-05,
+      "loss": 0.3551,
+      "step": 2641
+    },
+    {
+      "epoch": 3.8014388489208635,
+      "grad_norm": 0.10748542105822156,
+      "learning_rate": 1.3208903316619436e-05,
+      "loss": 0.361,
+      "step": 2642
+    },
+    {
+      "epoch": 3.8028776978417267,
+      "grad_norm": 0.09818385057623447,
+      "learning_rate": 1.317907575409254e-05,
+      "loss": 0.3631,
+      "step": 2643
+    },
+    {
+      "epoch": 3.80431654676259,
+      "grad_norm": 0.09617813037382426,
+      "learning_rate": 1.3149275263400116e-05,
+      "loss": 0.3627,
+      "step": 2644
+    },
+    {
+      "epoch": 3.805755395683453,
+      "grad_norm": 0.10794047897019735,
+      "learning_rate": 1.3119501874621437e-05,
+      "loss": 0.3626,
+      "step": 2645
+    },
+    {
+      "epoch": 3.8071942446043163,
+      "grad_norm": 0.1028120508008396,
+      "learning_rate": 1.3089755617808417e-05,
+      "loss": 0.3498,
+      "step": 2646
+    },
+    {
+      "epoch": 3.80863309352518,
+      "grad_norm": 0.11536134670582235,
+      "learning_rate": 1.3060036522985598e-05,
+      "loss": 0.3583,
+      "step": 2647
+    },
+    {
+      "epoch": 3.810071942446043,
+      "grad_norm": 0.1014443276620562,
+      "learning_rate": 1.3030344620150105e-05,
+      "loss": 0.358,
+      "step": 2648
+    },
+    {
+      "epoch": 3.8115107913669064,
+      "grad_norm": 0.10555213738641042,
+      "learning_rate": 1.3000679939271588e-05,
+      "loss": 0.3623,
+      "step": 2649
+    },
+    {
+      "epoch": 3.81294964028777,
+      "grad_norm": 0.10616309893328842,
+      "learning_rate": 1.2971042510292238e-05,
+      "loss": 0.3629,
+      "step": 2650
+    },
+    {
+      "epoch": 3.814388489208633,
+      "grad_norm": 0.11041867870782239,
+      "learning_rate": 1.2941432363126784e-05,
+      "loss": 0.3633,
+      "step": 2651
+    },
+    {
+      "epoch": 3.8158273381294965,
+      "grad_norm": 0.1190498763283357,
+      "learning_rate": 1.2911849527662335e-05,
+      "loss": 0.3671,
+      "step": 2652
+    },
+    {
+      "epoch": 3.8172661870503597,
+      "grad_norm": 0.12022477616666787,
+      "learning_rate": 1.2882294033758473e-05,
+      "loss": 0.3663,
+      "step": 2653
+    },
+    {
+      "epoch": 3.818705035971223,
+      "grad_norm": 0.10685794435909542,
+      "learning_rate": 1.2852765911247227e-05,
+      "loss": 0.3571,
+      "step": 2654
+    },
+    {
+      "epoch": 3.8201438848920866,
+      "grad_norm": 0.10794695689107299,
+      "learning_rate": 1.2823265189932914e-05,
+      "loss": 0.3617,
+      "step": 2655
+    },
+    {
+      "epoch": 3.8215827338129498,
+      "grad_norm": 0.11693638820113349,
+      "learning_rate": 1.2793791899592254e-05,
+      "loss": 0.3691,
+      "step": 2656
+    },
+    {
+      "epoch": 3.823021582733813,
+      "grad_norm": 0.09592170560544878,
+      "learning_rate": 1.2764346069974249e-05,
+      "loss": 0.3548,
+      "step": 2657
+    },
+    {
+      "epoch": 3.824460431654676,
+      "grad_norm": 0.09943996269770372,
+      "learning_rate": 1.2734927730800206e-05,
+      "loss": 0.3541,
+      "step": 2658
+    },
+    {
+      "epoch": 3.8258992805755394,
+      "grad_norm": 0.13401545776494028,
+      "learning_rate": 1.2705536911763665e-05,
+      "loss": 0.3666,
+      "step": 2659
+    },
+    {
+      "epoch": 3.827338129496403,
+      "grad_norm": 0.10808324164765093,
+      "learning_rate": 1.2676173642530417e-05,
+      "loss": 0.3661,
+      "step": 2660
+    },
+    {
+      "epoch": 3.8287769784172663,
+      "grad_norm": 0.0985569274615787,
+      "learning_rate": 1.2646837952738382e-05,
+      "loss": 0.3521,
+      "step": 2661
+    },
+    {
+      "epoch": 3.8302158273381295,
+      "grad_norm": 0.10177724691359441,
+      "learning_rate": 1.2617529871997727e-05,
+      "loss": 0.3687,
+      "step": 2662
+    },
+    {
+      "epoch": 3.8316546762589927,
+      "grad_norm": 0.11808499539692265,
+      "learning_rate": 1.2588249429890706e-05,
+      "loss": 0.3532,
+      "step": 2663
+    },
+    {
+      "epoch": 3.833093525179856,
+      "grad_norm": 0.1111604668473334,
+      "learning_rate": 1.2558996655971644e-05,
+      "loss": 0.3558,
+      "step": 2664
+    },
+    {
+      "epoch": 3.8345323741007196,
+      "grad_norm": 0.1279783316430316,
+      "learning_rate": 1.2529771579767024e-05,
+      "loss": 0.359,
+      "step": 2665
+    },
+    {
+      "epoch": 3.8359712230215828,
+      "grad_norm": 0.09848527975004501,
+      "learning_rate": 1.2500574230775294e-05,
+      "loss": 0.3663,
+      "step": 2666
+    },
+    {
+      "epoch": 3.837410071942446,
+      "grad_norm": 0.12029030117458071,
+      "learning_rate": 1.2471404638466949e-05,
+      "loss": 0.362,
+      "step": 2667
+    },
+    {
+      "epoch": 3.838848920863309,
+      "grad_norm": 0.11000558152107098,
+      "learning_rate": 1.2442262832284464e-05,
+      "loss": 0.3515,
+      "step": 2668
+    },
+    {
+      "epoch": 3.8402877697841724,
+      "grad_norm": 0.11723113093826766,
+      "learning_rate": 1.2413148841642268e-05,
+      "loss": 0.3615,
+      "step": 2669
+    },
+    {
+      "epoch": 3.841726618705036,
+      "grad_norm": 0.11302251660022884,
+      "learning_rate": 1.2384062695926713e-05,
+      "loss": 0.3592,
+      "step": 2670
+    },
+    {
+      "epoch": 3.8431654676258993,
+      "grad_norm": 0.12412774957255451,
+      "learning_rate": 1.235500442449605e-05,
+      "loss": 0.3618,
+      "step": 2671
+    },
+    {
+      "epoch": 3.8446043165467625,
+      "grad_norm": 0.09488577240409062,
+      "learning_rate": 1.232597405668039e-05,
+      "loss": 0.3611,
+      "step": 2672
+    },
+    {
+      "epoch": 3.846043165467626,
+      "grad_norm": 0.11443172578323646,
+      "learning_rate": 1.2296971621781677e-05,
+      "loss": 0.3581,
+      "step": 2673
+    },
+    {
+      "epoch": 3.8474820143884894,
+      "grad_norm": 0.10661096580146355,
+      "learning_rate": 1.2267997149073679e-05,
+      "loss": 0.366,
+      "step": 2674
+    },
+    {
+      "epoch": 3.8489208633093526,
+      "grad_norm": 0.10099669801061048,
+      "learning_rate": 1.2239050667801885e-05,
+      "loss": 0.3566,
+      "step": 2675
+    },
+    {
+      "epoch": 3.850359712230216,
+      "grad_norm": 0.11442540074877293,
+      "learning_rate": 1.2210132207183611e-05,
+      "loss": 0.3663,
+      "step": 2676
+    },
+    {
+      "epoch": 3.851798561151079,
+      "grad_norm": 0.10326919646973422,
+      "learning_rate": 1.2181241796407855e-05,
+      "loss": 0.3655,
+      "step": 2677
+    },
+    {
+      "epoch": 3.8532374100719426,
+      "grad_norm": 0.11407317713325474,
+      "learning_rate": 1.2152379464635264e-05,
+      "loss": 0.36,
+      "step": 2678
+    },
+    {
+      "epoch": 3.854676258992806,
+      "grad_norm": 0.10025218616264969,
+      "learning_rate": 1.2123545240998182e-05,
+      "loss": 0.355,
+      "step": 2679
+    },
+    {
+      "epoch": 3.856115107913669,
+      "grad_norm": 0.09564519109975264,
+      "learning_rate": 1.2094739154600616e-05,
+      "loss": 0.3531,
+      "step": 2680
+    },
+    {
+      "epoch": 3.8575539568345323,
+      "grad_norm": 0.1259784386469854,
+      "learning_rate": 1.2065961234518096e-05,
+      "loss": 0.3579,
+      "step": 2681
+    },
+    {
+      "epoch": 3.8589928057553955,
+      "grad_norm": 0.10632725682300842,
+      "learning_rate": 1.2037211509797771e-05,
+      "loss": 0.3613,
+      "step": 2682
+    },
+    {
+      "epoch": 3.860431654676259,
+      "grad_norm": 0.10146620175603975,
+      "learning_rate": 1.2008490009458322e-05,
+      "loss": 0.3625,
+      "step": 2683
+    },
+    {
+      "epoch": 3.8618705035971224,
+      "grad_norm": 0.09500020982820409,
+      "learning_rate": 1.1979796762489934e-05,
+      "loss": 0.3581,
+      "step": 2684
+    },
+    {
+      "epoch": 3.8633093525179856,
+      "grad_norm": 0.10521972710858278,
+      "learning_rate": 1.195113179785429e-05,
+      "loss": 0.3454,
+      "step": 2685
+    },
+    {
+      "epoch": 3.864748201438849,
+      "grad_norm": 0.09582255281437967,
+      "learning_rate": 1.1922495144484504e-05,
+      "loss": 0.3614,
+      "step": 2686
+    },
+    {
+      "epoch": 3.866187050359712,
+      "grad_norm": 0.10135225534705446,
+      "learning_rate": 1.1893886831285136e-05,
+      "loss": 0.3635,
+      "step": 2687
+    },
+    {
+      "epoch": 3.8676258992805757,
+      "grad_norm": 0.09511353365433746,
+      "learning_rate": 1.1865306887132122e-05,
+      "loss": 0.3566,
+      "step": 2688
+    },
+    {
+      "epoch": 3.869064748201439,
+      "grad_norm": 0.10025779281801901,
+      "learning_rate": 1.183675534087279e-05,
+      "loss": 0.3619,
+      "step": 2689
+    },
+    {
+      "epoch": 3.870503597122302,
+      "grad_norm": 0.08858656791121877,
+      "learning_rate": 1.1808232221325749e-05,
+      "loss": 0.3628,
+      "step": 2690
+    },
+    {
+      "epoch": 3.8719424460431657,
+      "grad_norm": 0.09260597242456182,
+      "learning_rate": 1.1779737557280985e-05,
+      "loss": 0.3608,
+      "step": 2691
+    },
+    {
+      "epoch": 3.873381294964029,
+      "grad_norm": 0.10298075180631532,
+      "learning_rate": 1.1751271377499736e-05,
+      "loss": 0.3587,
+      "step": 2692
+    },
+    {
+      "epoch": 3.874820143884892,
+      "grad_norm": 0.09065564279621754,
+      "learning_rate": 1.1722833710714454e-05,
+      "loss": 0.3672,
+      "step": 2693
+    },
+    {
+      "epoch": 3.8762589928057554,
+      "grad_norm": 0.1028082028401011,
+      "learning_rate": 1.1694424585628861e-05,
+      "loss": 0.3635,
+      "step": 2694
+    },
+    {
+      "epoch": 3.8776978417266186,
+      "grad_norm": 0.0930008537084772,
+      "learning_rate": 1.166604403091784e-05,
+      "loss": 0.362,
+      "step": 2695
+    },
+    {
+      "epoch": 3.8791366906474822,
+      "grad_norm": 0.10968089516367259,
+      "learning_rate": 1.1637692075227451e-05,
+      "loss": 0.3643,
+      "step": 2696
+    },
+    {
+      "epoch": 3.8805755395683454,
+      "grad_norm": 0.09167057693074489,
+      "learning_rate": 1.1609368747174883e-05,
+      "loss": 0.3631,
+      "step": 2697
+    },
+    {
+      "epoch": 3.8820143884892087,
+      "grad_norm": 0.09608290565096088,
+      "learning_rate": 1.1581074075348431e-05,
+      "loss": 0.3558,
+      "step": 2698
+    },
+    {
+      "epoch": 3.883453237410072,
+      "grad_norm": 0.11203789648383493,
+      "learning_rate": 1.155280808830746e-05,
+      "loss": 0.3637,
+      "step": 2699
+    },
+    {
+      "epoch": 3.884892086330935,
+      "grad_norm": 0.09942035822295522,
+      "learning_rate": 1.15245708145824e-05,
+      "loss": 0.3676,
+      "step": 2700
+    },
+    {
+      "epoch": 3.8863309352517987,
+      "grad_norm": 0.10350023938493974,
+      "learning_rate": 1.1496362282674647e-05,
+      "loss": 0.3614,
+      "step": 2701
+    },
+    {
+      "epoch": 3.887769784172662,
+      "grad_norm": 0.09981408944424823,
+      "learning_rate": 1.1468182521056663e-05,
+      "loss": 0.359,
+      "step": 2702
+    },
+    {
+      "epoch": 3.889208633093525,
+      "grad_norm": 0.11511236345501889,
+      "learning_rate": 1.1440031558171834e-05,
+      "loss": 0.3608,
+      "step": 2703
+    },
+    {
+      "epoch": 3.8906474820143884,
+      "grad_norm": 0.10090422597951822,
+      "learning_rate": 1.1411909422434441e-05,
+      "loss": 0.3637,
+      "step": 2704
+    },
+    {
+      "epoch": 3.8920863309352516,
+      "grad_norm": 0.10735672779609931,
+      "learning_rate": 1.1383816142229715e-05,
+      "loss": 0.3629,
+      "step": 2705
+    },
+    {
+      "epoch": 3.8935251798561152,
+      "grad_norm": 0.11333440737627762,
+      "learning_rate": 1.1355751745913781e-05,
+      "loss": 0.3596,
+      "step": 2706
+    },
+    {
+      "epoch": 3.8949640287769784,
+      "grad_norm": 0.0950438089714313,
+      "learning_rate": 1.1327716261813539e-05,
+      "loss": 0.3573,
+      "step": 2707
+    },
+    {
+      "epoch": 3.8964028776978417,
+      "grad_norm": 0.10940745545202456,
+      "learning_rate": 1.1299709718226745e-05,
+      "loss": 0.3588,
+      "step": 2708
+    },
+    {
+      "epoch": 3.897841726618705,
+      "grad_norm": 0.12759866682775797,
+      "learning_rate": 1.1271732143421992e-05,
+      "loss": 0.3525,
+      "step": 2709
+    },
+    {
+      "epoch": 3.899280575539568,
+      "grad_norm": 0.11431046414492715,
+      "learning_rate": 1.1243783565638533e-05,
+      "loss": 0.3651,
+      "step": 2710
+    },
+    {
+      "epoch": 3.9007194244604317,
+      "grad_norm": 0.9564132684060368,
+      "learning_rate": 1.121586401308643e-05,
+      "loss": 0.3685,
+      "step": 2711
+    },
+    {
+      "epoch": 3.902158273381295,
+      "grad_norm": 0.11077573358899574,
+      "learning_rate": 1.1187973513946417e-05,
+      "loss": 0.3634,
+      "step": 2712
+    },
+    {
+      "epoch": 3.903597122302158,
+      "grad_norm": 0.10778792389149551,
+      "learning_rate": 1.1160112096369913e-05,
+      "loss": 0.3593,
+      "step": 2713
+    },
+    {
+      "epoch": 3.905035971223022,
+      "grad_norm": 0.11189896505320317,
+      "learning_rate": 1.1132279788478977e-05,
+      "loss": 0.3585,
+      "step": 2714
+    },
+    {
+      "epoch": 3.906474820143885,
+      "grad_norm": 0.1052821688406415,
+      "learning_rate": 1.1104476618366298e-05,
+      "loss": 0.3558,
+      "step": 2715
+    },
+    {
+      "epoch": 3.9079136690647482,
+      "grad_norm": 0.11120022223037514,
+      "learning_rate": 1.1076702614095116e-05,
+      "loss": 0.3627,
+      "step": 2716
+    },
+    {
+      "epoch": 3.9093525179856115,
+      "grad_norm": 0.11008204754639704,
+      "learning_rate": 1.1048957803699292e-05,
+      "loss": 0.3629,
+      "step": 2717
+    },
+    {
+      "epoch": 3.9107913669064747,
+      "grad_norm": 0.12393216658264912,
+      "learning_rate": 1.1021242215183193e-05,
+      "loss": 0.3694,
+      "step": 2718
+    },
+    {
+      "epoch": 3.9122302158273383,
+      "grad_norm": 0.4970887948932858,
+      "learning_rate": 1.0993555876521658e-05,
+      "loss": 0.3655,
+      "step": 2719
+    },
+    {
+      "epoch": 3.9136690647482015,
+      "grad_norm": 0.14163532032564988,
+      "learning_rate": 1.096589881566005e-05,
+      "loss": 0.3663,
+      "step": 2720
+    },
+    {
+      "epoch": 3.9151079136690647,
+      "grad_norm": 0.09692858784116108,
+      "learning_rate": 1.0938271060514162e-05,
+      "loss": 0.3583,
+      "step": 2721
+    },
+    {
+      "epoch": 3.916546762589928,
+      "grad_norm": 0.12425912973919774,
+      "learning_rate": 1.0910672638970206e-05,
+      "loss": 0.357,
+      "step": 2722
+    },
+    {
+      "epoch": 3.917985611510791,
+      "grad_norm": 0.11842405300151483,
+      "learning_rate": 1.0883103578884784e-05,
+      "loss": 0.3635,
+      "step": 2723
+    },
+    {
+      "epoch": 3.919424460431655,
+      "grad_norm": 0.09541247158438779,
+      "learning_rate": 1.085556390808487e-05,
+      "loss": 0.3595,
+      "step": 2724
+    },
+    {
+      "epoch": 3.920863309352518,
+      "grad_norm": 0.11200656304991745,
+      "learning_rate": 1.082805365436776e-05,
+      "loss": 0.3679,
+      "step": 2725
+    },
+    {
+      "epoch": 3.9223021582733812,
+      "grad_norm": 0.11181453719163843,
+      "learning_rate": 1.0800572845501095e-05,
+      "loss": 0.3567,
+      "step": 2726
+    },
+    {
+      "epoch": 3.9237410071942445,
+      "grad_norm": 0.09861364204229336,
+      "learning_rate": 1.0773121509222712e-05,
+      "loss": 0.3672,
+      "step": 2727
+    },
+    {
+      "epoch": 3.9251798561151077,
+      "grad_norm": 0.1057757176136241,
+      "learning_rate": 1.0745699673240808e-05,
+      "loss": 0.36,
+      "step": 2728
+    },
+    {
+      "epoch": 3.9266187050359713,
+      "grad_norm": 0.1019318459420541,
+      "learning_rate": 1.0718307365233737e-05,
+      "loss": 0.3543,
+      "step": 2729
+    },
+    {
+      "epoch": 3.9280575539568345,
+      "grad_norm": 0.10306055253627819,
+      "learning_rate": 1.0690944612850052e-05,
+      "loss": 0.3577,
+      "step": 2730
+    },
+    {
+      "epoch": 3.9294964028776977,
+      "grad_norm": 0.11664140763494153,
+      "learning_rate": 1.0663611443708471e-05,
+      "loss": 0.3594,
+      "step": 2731
+    },
+    {
+      "epoch": 3.9309352517985614,
+      "grad_norm": 0.11073226114226936,
+      "learning_rate": 1.0636307885397911e-05,
+      "loss": 0.3689,
+      "step": 2732
+    },
+    {
+      "epoch": 3.9323741007194246,
+      "grad_norm": 0.10681885965436957,
+      "learning_rate": 1.0609033965477318e-05,
+      "loss": 0.3624,
+      "step": 2733
+    },
+    {
+      "epoch": 3.933812949640288,
+      "grad_norm": 0.12500736336387938,
+      "learning_rate": 1.0581789711475752e-05,
+      "loss": 0.3665,
+      "step": 2734
+    },
+    {
+      "epoch": 3.935251798561151,
+      "grad_norm": 0.09930500365361575,
+      "learning_rate": 1.0554575150892386e-05,
+      "loss": 0.3548,
+      "step": 2735
+    },
+    {
+      "epoch": 3.9366906474820142,
+      "grad_norm": 0.1021624596019113,
+      "learning_rate": 1.0527390311196326e-05,
+      "loss": 0.3545,
+      "step": 2736
+    },
+    {
+      "epoch": 3.938129496402878,
+      "grad_norm": 0.10628437462092273,
+      "learning_rate": 1.0500235219826748e-05,
+      "loss": 0.362,
+      "step": 2737
+    },
+    {
+      "epoch": 3.939568345323741,
+      "grad_norm": 0.11071067883850721,
+      "learning_rate": 1.0473109904192773e-05,
+      "loss": 0.3587,
+      "step": 2738
+    },
+    {
+      "epoch": 3.9410071942446043,
+      "grad_norm": 0.1068522520290516,
+      "learning_rate": 1.0446014391673476e-05,
+      "loss": 0.3657,
+      "step": 2739
+    },
+    {
+      "epoch": 3.9424460431654675,
+      "grad_norm": 0.09185630866760745,
+      "learning_rate": 1.0418948709617846e-05,
+      "loss": 0.3657,
+      "step": 2740
+    },
+    {
+      "epoch": 3.9438848920863308,
+      "grad_norm": 0.12260735019479156,
+      "learning_rate": 1.0391912885344784e-05,
+      "loss": 0.3527,
+      "step": 2741
+    },
+    {
+      "epoch": 3.9453237410071944,
+      "grad_norm": 0.10653215957239656,
+      "learning_rate": 1.0364906946142996e-05,
+      "loss": 0.3583,
+      "step": 2742
+    },
+    {
+      "epoch": 3.9467625899280576,
+      "grad_norm": 0.1042669525207971,
+      "learning_rate": 1.0337930919271094e-05,
+      "loss": 0.365,
+      "step": 2743
+    },
+    {
+      "epoch": 3.948201438848921,
+      "grad_norm": 0.1121419412537695,
+      "learning_rate": 1.0310984831957471e-05,
+      "loss": 0.3572,
+      "step": 2744
+    },
+    {
+      "epoch": 3.949640287769784,
+      "grad_norm": 0.10209132317781502,
+      "learning_rate": 1.0284068711400254e-05,
+      "loss": 0.3573,
+      "step": 2745
+    },
+    {
+      "epoch": 3.9510791366906473,
+      "grad_norm": 0.0996760617468679,
+      "learning_rate": 1.0257182584767423e-05,
+      "loss": 0.3623,
+      "step": 2746
+    },
+    {
+      "epoch": 3.952517985611511,
+      "grad_norm": 0.11716774537978202,
+      "learning_rate": 1.0230326479196573e-05,
+      "loss": 0.3556,
+      "step": 2747
+    },
+    {
+      "epoch": 3.953956834532374,
+      "grad_norm": 0.09557787024252977,
+      "learning_rate": 1.0203500421795075e-05,
+      "loss": 0.362,
+      "step": 2748
+    },
+    {
+      "epoch": 3.9553956834532373,
+      "grad_norm": 0.1148898935436786,
+      "learning_rate": 1.017670443963994e-05,
+      "loss": 0.3663,
+      "step": 2749
+    },
+    {
+      "epoch": 3.956834532374101,
+      "grad_norm": 0.10351097078129262,
+      "learning_rate": 1.0149938559777825e-05,
+      "loss": 0.363,
+      "step": 2750
+    },
+    {
+      "epoch": 3.9582733812949638,
+      "grad_norm": 0.0980401694645122,
+      "learning_rate": 1.0123202809225009e-05,
+      "loss": 0.3575,
+      "step": 2751
+    },
+    {
+      "epoch": 3.9597122302158274,
+      "grad_norm": 0.09750837058182665,
+      "learning_rate": 1.0096497214967349e-05,
+      "loss": 0.3614,
+      "step": 2752
+    },
+    {
+      "epoch": 3.9611510791366906,
+      "grad_norm": 0.11031556302217374,
+      "learning_rate": 1.0069821803960277e-05,
+      "loss": 0.3541,
+      "step": 2753
+    },
+    {
+      "epoch": 3.962589928057554,
+      "grad_norm": 0.09977694465125021,
+      "learning_rate": 1.0043176603128755e-05,
+      "loss": 0.3623,
+      "step": 2754
+    },
+    {
+      "epoch": 3.9640287769784175,
+      "grad_norm": 0.11347225388698515,
+      "learning_rate": 1.0016561639367253e-05,
+      "loss": 0.3611,
+      "step": 2755
+    },
+    {
+      "epoch": 3.9654676258992807,
+      "grad_norm": 0.11182600990952397,
+      "learning_rate": 9.989976939539687e-06,
+      "loss": 0.3589,
+      "step": 2756
+    },
+    {
+      "epoch": 3.966906474820144,
+      "grad_norm": 0.09250923238264663,
+      "learning_rate": 9.963422530479496e-06,
+      "loss": 0.3565,
+      "step": 2757
+    },
+    {
+      "epoch": 3.968345323741007,
+      "grad_norm": 0.09920114232935916,
+      "learning_rate": 9.936898438989507e-06,
+      "loss": 0.351,
+      "step": 2758
+    },
+    {
+      "epoch": 3.9697841726618703,
+      "grad_norm": 0.10377812097578386,
+      "learning_rate": 9.910404691841915e-06,
+      "loss": 0.3604,
+      "step": 2759
+    },
+    {
+      "epoch": 3.971223021582734,
+      "grad_norm": 0.09473159919254981,
+      "learning_rate": 9.883941315778319e-06,
+      "loss": 0.3551,
+      "step": 2760
+    },
+    {
+      "epoch": 3.972661870503597,
+      "grad_norm": 0.12287944307641946,
+      "learning_rate": 9.857508337509692e-06,
+      "loss": 0.3622,
+      "step": 2761
+    },
+    {
+      "epoch": 3.9741007194244604,
+      "grad_norm": 0.09875799435187659,
+      "learning_rate": 9.831105783716266e-06,
+      "loss": 0.3619,
+      "step": 2762
+    },
+    {
+      "epoch": 3.9755395683453236,
+      "grad_norm": 0.10299870999836316,
+      "learning_rate": 9.8047336810476e-06,
+      "loss": 0.3603,
+      "step": 2763
+    },
+    {
+      "epoch": 3.976978417266187,
+      "grad_norm": 0.10514063205962414,
+      "learning_rate": 9.778392056122503e-06,
+      "loss": 0.3573,
+      "step": 2764
+    },
+    {
+      "epoch": 3.9784172661870505,
+      "grad_norm": 0.08862739370425651,
+      "learning_rate": 9.752080935529037e-06,
+      "loss": 0.3566,
+      "step": 2765
+    },
+    {
+      "epoch": 3.9798561151079137,
+      "grad_norm": 0.10420965063882555,
+      "learning_rate": 9.725800345824453e-06,
+      "loss": 0.3601,
+      "step": 2766
+    },
+    {
+      "epoch": 3.981294964028777,
+      "grad_norm": 0.11741476939494742,
+      "learning_rate": 9.699550313535196e-06,
+      "loss": 0.3626,
+      "step": 2767
+    },
+    {
+      "epoch": 3.98273381294964,
+      "grad_norm": 0.1051511996966301,
+      "learning_rate": 9.673330865156875e-06,
+      "loss": 0.3697,
+      "step": 2768
+    },
+    {
+      "epoch": 3.9841726618705033,
+      "grad_norm": 0.10984037723207088,
+      "learning_rate": 9.647142027154222e-06,
+      "loss": 0.362,
+      "step": 2769
+    },
+    {
+      "epoch": 3.985611510791367,
+      "grad_norm": 0.10093858750504218,
+      "learning_rate": 9.620983825961078e-06,
+      "loss": 0.3552,
+      "step": 2770
+    },
+    {
+      "epoch": 3.98705035971223,
+      "grad_norm": 0.09746498465302932,
+      "learning_rate": 9.594856287980323e-06,
+      "loss": 0.3524,
+      "step": 2771
+    },
+    {
+      "epoch": 3.9884892086330934,
+      "grad_norm": 0.10533187198362187,
+      "learning_rate": 9.56875943958396e-06,
+      "loss": 0.3623,
+      "step": 2772
+    },
+    {
+      "epoch": 3.989928057553957,
+      "grad_norm": 0.09715565692470797,
+      "learning_rate": 9.542693307112949e-06,
+      "loss": 0.3556,
+      "step": 2773
+    },
+    {
+      "epoch": 3.9913669064748203,
+      "grad_norm": 0.09676232181036293,
+      "learning_rate": 9.516657916877272e-06,
+      "loss": 0.3578,
+      "step": 2774
+    },
+    {
+      "epoch": 3.9928057553956835,
+      "grad_norm": 0.12015577947176528,
+      "learning_rate": 9.490653295155891e-06,
+      "loss": 0.3594,
+      "step": 2775
+    },
+    {
+      "epoch": 3.9942446043165467,
+      "grad_norm": 0.09481055578760383,
+      "learning_rate": 9.464679468196696e-06,
+      "loss": 0.3558,
+      "step": 2776
+    },
+    {
+      "epoch": 3.99568345323741,
+      "grad_norm": 0.09389360547241736,
+      "learning_rate": 9.438736462216496e-06,
+      "loss": 0.3605,
+      "step": 2777
+    },
+    {
+      "epoch": 3.9971223021582736,
+      "grad_norm": 0.10665313888423696,
+      "learning_rate": 9.412824303401003e-06,
+      "loss": 0.362,
+      "step": 2778
+    },
+    {
+      "epoch": 3.998561151079137,
+      "grad_norm": 0.09496625781759228,
+      "learning_rate": 9.38694301790478e-06,
+      "loss": 0.374,
+      "step": 2779
+    },
+    {
+      "epoch": 4.0,
+      "grad_norm": 0.10124169992214632,
+      "learning_rate": 9.361092631851228e-06,
+      "loss": 0.3679,
+      "step": 2780
+    },
+    {
+      "epoch": 4.001438848920864,
+      "grad_norm": 0.13559259159060802,
+      "learning_rate": 9.335273171332581e-06,
+      "loss": 0.3454,
+      "step": 2781
+    },
+    {
+      "epoch": 4.002877697841726,
+      "grad_norm": 0.10035380451874652,
+      "learning_rate": 9.30948466240981e-06,
+      "loss": 0.3419,
+      "step": 2782
+    },
+    {
+      "epoch": 4.00431654676259,
+      "grad_norm": 0.09733148277656034,
+      "learning_rate": 9.2837271311127e-06,
+      "loss": 0.3421,
+      "step": 2783
+    },
+    {
+      "epoch": 4.005755395683453,
+      "grad_norm": 0.10382884714560332,
+      "learning_rate": 9.25800060343975e-06,
+      "loss": 0.3401,
+      "step": 2784
+    },
+    {
+      "epoch": 4.0071942446043165,
+      "grad_norm": 0.11033527019094946,
+      "learning_rate": 9.232305105358139e-06,
+      "loss": 0.3463,
+      "step": 2785
+    },
+    {
+      "epoch": 4.00863309352518,
+      "grad_norm": 0.12516376026251724,
+      "learning_rate": 9.206640662803746e-06,
+      "loss": 0.3374,
+      "step": 2786
+    },
+    {
+      "epoch": 4.010071942446043,
+      "grad_norm": 0.1094679305415975,
+      "learning_rate": 9.181007301681135e-06,
+      "loss": 0.3488,
+      "step": 2787
+    },
+    {
+      "epoch": 4.011510791366907,
+      "grad_norm": 0.11467050689120065,
+      "learning_rate": 9.155405047863439e-06,
+      "loss": 0.3389,
+      "step": 2788
+    },
+    {
+      "epoch": 4.012949640287769,
+      "grad_norm": 0.11131158738810967,
+      "learning_rate": 9.12983392719243e-06,
+      "loss": 0.3412,
+      "step": 2789
+    },
+    {
+      "epoch": 4.014388489208633,
+      "grad_norm": 0.11992712129415503,
+      "learning_rate": 9.104293965478446e-06,
+      "loss": 0.3379,
+      "step": 2790
+    },
+    {
+      "epoch": 4.015827338129497,
+      "grad_norm": 0.10721680297702718,
+      "learning_rate": 9.078785188500378e-06,
+      "loss": 0.3454,
+      "step": 2791
+    },
+    {
+      "epoch": 4.017266187050359,
+      "grad_norm": 0.11297504942643836,
+      "learning_rate": 9.053307622005639e-06,
+      "loss": 0.3383,
+      "step": 2792
+    },
+    {
+      "epoch": 4.018705035971223,
+      "grad_norm": 0.10600194033425428,
+      "learning_rate": 9.02786129171013e-06,
+      "loss": 0.3315,
+      "step": 2793
+    },
+    {
+      "epoch": 4.020143884892087,
+      "grad_norm": 0.10334289899927786,
+      "learning_rate": 9.002446223298244e-06,
+      "loss": 0.3448,
+      "step": 2794
+    },
+    {
+      "epoch": 4.0215827338129495,
+      "grad_norm": 0.11321397054770597,
+      "learning_rate": 8.977062442422796e-06,
+      "loss": 0.3368,
+      "step": 2795
+    },
+    {
+      "epoch": 4.023021582733813,
+      "grad_norm": 0.1035332486262864,
+      "learning_rate": 8.951709974705057e-06,
+      "loss": 0.3514,
+      "step": 2796
+    },
+    {
+      "epoch": 4.024460431654676,
+      "grad_norm": 0.09504472096495813,
+      "learning_rate": 8.926388845734624e-06,
+      "loss": 0.3445,
+      "step": 2797
+    },
+    {
+      "epoch": 4.02589928057554,
+      "grad_norm": 0.10885943183780947,
+      "learning_rate": 8.901099081069553e-06,
+      "loss": 0.337,
+      "step": 2798
+    },
+    {
+      "epoch": 4.027338129496403,
+      "grad_norm": 0.09986907982849921,
+      "learning_rate": 8.875840706236163e-06,
+      "loss": 0.3352,
+      "step": 2799
+    },
+    {
+      "epoch": 4.028776978417266,
+      "grad_norm": 0.10310959624014754,
+      "learning_rate": 8.850613746729117e-06,
+      "loss": 0.3403,
+      "step": 2800
+    },
+    {
+      "epoch": 4.03021582733813,
+      "grad_norm": 0.10520753354202317,
+      "learning_rate": 8.825418228011413e-06,
+      "loss": 0.3388,
+      "step": 2801
+    },
+    {
+      "epoch": 4.031654676258992,
+      "grad_norm": 0.09357355812436322,
+      "learning_rate": 8.80025417551424e-06,
+      "loss": 0.3393,
+      "step": 2802
+    },
+    {
+      "epoch": 4.033093525179856,
+      "grad_norm": 0.09943262643204327,
+      "learning_rate": 8.775121614637064e-06,
+      "loss": 0.3445,
+      "step": 2803
+    },
+    {
+      "epoch": 4.03453237410072,
+      "grad_norm": 0.0931000734423195,
+      "learning_rate": 8.750020570747568e-06,
+      "loss": 0.3434,
+      "step": 2804
+    },
+    {
+      "epoch": 4.0359712230215825,
+      "grad_norm": 0.10407012795568063,
+      "learning_rate": 8.724951069181617e-06,
+      "loss": 0.3408,
+      "step": 2805
+    },
+    {
+      "epoch": 4.037410071942446,
+      "grad_norm": 0.09234876336430461,
+      "learning_rate": 8.699913135243237e-06,
+      "loss": 0.346,
+      "step": 2806
+    },
+    {
+      "epoch": 4.038848920863309,
+      "grad_norm": 0.0948259182251536,
+      "learning_rate": 8.6749067942046e-06,
+      "loss": 0.3477,
+      "step": 2807
+    },
+    {
+      "epoch": 4.040287769784173,
+      "grad_norm": 0.09581875981707894,
+      "learning_rate": 8.649932071305952e-06,
+      "loss": 0.3444,
+      "step": 2808
+    },
+    {
+      "epoch": 4.041726618705036,
+      "grad_norm": 0.09404360483482738,
+      "learning_rate": 8.624988991755687e-06,
+      "loss": 0.3484,
+      "step": 2809
+    },
+    {
+      "epoch": 4.043165467625899,
+      "grad_norm": 0.09424538079465061,
+      "learning_rate": 8.60007758073023e-06,
+      "loss": 0.3422,
+      "step": 2810
+    },
+    {
+      "epoch": 4.044604316546763,
+      "grad_norm": 0.09028279429198212,
+      "learning_rate": 8.575197863374006e-06,
+      "loss": 0.3389,
+      "step": 2811
+    },
+    {
+      "epoch": 4.046043165467626,
+      "grad_norm": 0.09737949864105801,
+      "learning_rate": 8.550349864799505e-06,
+      "loss": 0.3464,
+      "step": 2812
+    },
+    {
+      "epoch": 4.047482014388489,
+      "grad_norm": 0.09569267256554466,
+      "learning_rate": 8.525533610087193e-06,
+      "loss": 0.3433,
+      "step": 2813
+    },
+    {
+      "epoch": 4.048920863309353,
+      "grad_norm": 0.816965116942467,
+      "learning_rate": 8.500749124285455e-06,
+      "loss": 0.3512,
+      "step": 2814
+    },
+    {
+      "epoch": 4.0503597122302155,
+      "grad_norm": 0.08126219538723158,
+      "learning_rate": 8.475996432410642e-06,
+      "loss": 0.3271,
+      "step": 2815
+    },
+    {
+      "epoch": 4.051798561151079,
+      "grad_norm": 0.08746993619788168,
+      "learning_rate": 8.451275559447011e-06,
+      "loss": 0.3399,
+      "step": 2816
+    },
+    {
+      "epoch": 4.053237410071943,
+      "grad_norm": 0.0902846287819087,
+      "learning_rate": 8.426586530346705e-06,
+      "loss": 0.3496,
+      "step": 2817
+    },
+    {
+      "epoch": 4.054676258992806,
+      "grad_norm": 0.10193643857742915,
+      "learning_rate": 8.401929370029708e-06,
+      "loss": 0.3486,
+      "step": 2818
+    },
+    {
+      "epoch": 4.056115107913669,
+      "grad_norm": 0.09145226631320796,
+      "learning_rate": 8.377304103383857e-06,
+      "loss": 0.3482,
+      "step": 2819
+    },
+    {
+      "epoch": 4.057553956834532,
+      "grad_norm": 0.09814500261065982,
+      "learning_rate": 8.352710755264786e-06,
+      "loss": 0.336,
+      "step": 2820
+    },
+    {
+      "epoch": 4.058992805755396,
+      "grad_norm": 0.09982457529970826,
+      "learning_rate": 8.328149350495916e-06,
+      "loss": 0.3435,
+      "step": 2821
+    },
+    {
+      "epoch": 4.060431654676259,
+      "grad_norm": 0.08801249772630416,
+      "learning_rate": 8.303619913868427e-06,
+      "loss": 0.3424,
+      "step": 2822
+    },
+    {
+      "epoch": 4.061870503597122,
+      "grad_norm": 0.09319954228614029,
+      "learning_rate": 8.279122470141208e-06,
+      "loss": 0.3392,
+      "step": 2823
+    },
+    {
+      "epoch": 4.063309352517986,
+      "grad_norm": 0.0960504249714296,
+      "learning_rate": 8.254657044040914e-06,
+      "loss": 0.3507,
+      "step": 2824
+    },
+    {
+      "epoch": 4.0647482014388485,
+      "grad_norm": 0.08805181094143594,
+      "learning_rate": 8.230223660261814e-06,
+      "loss": 0.3441,
+      "step": 2825
+    },
+    {
+      "epoch": 4.066187050359712,
+      "grad_norm": 0.08993959326068168,
+      "learning_rate": 8.205822343465865e-06,
+      "loss": 0.3438,
+      "step": 2826
+    },
+    {
+      "epoch": 4.067625899280576,
+      "grad_norm": 0.09462892477326966,
+      "learning_rate": 8.181453118282694e-06,
+      "loss": 0.3461,
+      "step": 2827
+    },
+    {
+      "epoch": 4.069064748201439,
+      "grad_norm": 0.09223113763588198,
+      "learning_rate": 8.157116009309467e-06,
+      "loss": 0.3357,
+      "step": 2828
+    },
+    {
+      "epoch": 4.070503597122302,
+      "grad_norm": 0.0947373152478922,
+      "learning_rate": 8.132811041110976e-06,
+      "loss": 0.3385,
+      "step": 2829
+    },
+    {
+      "epoch": 4.071942446043165,
+      "grad_norm": 0.0911691108813414,
+      "learning_rate": 8.108538238219564e-06,
+      "loss": 0.3482,
+      "step": 2830
+    },
+    {
+      "epoch": 4.073381294964029,
+      "grad_norm": 0.09481325760507595,
+      "learning_rate": 8.084297625135104e-06,
+      "loss": 0.3484,
+      "step": 2831
+    },
+    {
+      "epoch": 4.074820143884892,
+      "grad_norm": 0.09280800444964485,
+      "learning_rate": 8.060089226324987e-06,
+      "loss": 0.3396,
+      "step": 2832
+    },
+    {
+      "epoch": 4.076258992805755,
+      "grad_norm": 0.09206834340203833,
+      "learning_rate": 8.035913066224088e-06,
+      "loss": 0.3402,
+      "step": 2833
+    },
+    {
+      "epoch": 4.077697841726619,
+      "grad_norm": 0.09786690966344266,
+      "learning_rate": 8.0117691692347e-06,
+      "loss": 0.3448,
+      "step": 2834
+    },
+    {
+      "epoch": 4.079136690647482,
+      "grad_norm": 0.09435932024235821,
+      "learning_rate": 7.987657559726628e-06,
+      "loss": 0.3394,
+      "step": 2835
+    },
+    {
+      "epoch": 4.080575539568345,
+      "grad_norm": 0.10264188177189495,
+      "learning_rate": 7.963578262037038e-06,
+      "loss": 0.343,
+      "step": 2836
+    },
+    {
+      "epoch": 4.082014388489209,
+      "grad_norm": 0.08142038934543594,
+      "learning_rate": 7.939531300470458e-06,
+      "loss": 0.3377,
+      "step": 2837
+    },
+    {
+      "epoch": 4.083453237410072,
+      "grad_norm": 0.10294910680683012,
+      "learning_rate": 7.915516699298847e-06,
+      "loss": 0.3332,
+      "step": 2838
+    },
+    {
+      "epoch": 4.084892086330935,
+      "grad_norm": 0.09467645237728697,
+      "learning_rate": 7.891534482761463e-06,
+      "loss": 0.3411,
+      "step": 2839
+    },
+    {
+      "epoch": 4.086330935251799,
+      "grad_norm": 0.09106635791743539,
+      "learning_rate": 7.867584675064846e-06,
+      "loss": 0.342,
+      "step": 2840
+    },
+    {
+      "epoch": 4.087769784172662,
+      "grad_norm": 0.10286179689172759,
+      "learning_rate": 7.843667300382863e-06,
+      "loss": 0.3383,
+      "step": 2841
+    },
+    {
+      "epoch": 4.089208633093525,
+      "grad_norm": 0.09020293381361195,
+      "learning_rate": 7.81978238285667e-06,
+      "loss": 0.3416,
+      "step": 2842
+    },
+    {
+      "epoch": 4.090647482014388,
+      "grad_norm": 0.09102358622946989,
+      "learning_rate": 7.795929946594584e-06,
+      "loss": 0.3543,
+      "step": 2843
+    },
+    {
+      "epoch": 4.092086330935252,
+      "grad_norm": 0.09068947018842027,
+      "learning_rate": 7.772110015672209e-06,
+      "loss": 0.3471,
+      "step": 2844
+    },
+    {
+      "epoch": 4.093525179856115,
+      "grad_norm": 0.1039169415430003,
+      "learning_rate": 7.748322614132297e-06,
+      "loss": 0.3429,
+      "step": 2845
+    },
+    {
+      "epoch": 4.094964028776978,
+      "grad_norm": 0.09358126911792906,
+      "learning_rate": 7.72456776598479e-06,
+      "loss": 0.3385,
+      "step": 2846
+    },
+    {
+      "epoch": 4.096402877697842,
+      "grad_norm": 0.0917379650216092,
+      "learning_rate": 7.70084549520676e-06,
+      "loss": 0.3536,
+      "step": 2847
+    },
+    {
+      "epoch": 4.097841726618705,
+      "grad_norm": 0.09264993647622498,
+      "learning_rate": 7.6771558257424e-06,
+      "loss": 0.3459,
+      "step": 2848
+    },
+    {
+      "epoch": 4.099280575539568,
+      "grad_norm": 0.1618684592841564,
+      "learning_rate": 7.653498781502997e-06,
+      "loss": 0.3381,
+      "step": 2849
+    },
+    {
+      "epoch": 4.100719424460432,
+      "grad_norm": 0.09029043245153334,
+      "learning_rate": 7.629874386366918e-06,
+      "loss": 0.3446,
+      "step": 2850
+    },
+    {
+      "epoch": 4.102158273381295,
+      "grad_norm": 0.09763948580556599,
+      "learning_rate": 7.606282664179545e-06,
+      "loss": 0.3391,
+      "step": 2851
+    },
+    {
+      "epoch": 4.103597122302158,
+      "grad_norm": 0.08585510341128091,
+      "learning_rate": 7.5827236387532976e-06,
+      "loss": 0.3444,
+      "step": 2852
+    },
+    {
+      "epoch": 4.105035971223022,
+      "grad_norm": 0.08803938625909877,
+      "learning_rate": 7.559197333867629e-06,
+      "loss": 0.3463,
+      "step": 2853
+    },
+    {
+      "epoch": 4.106474820143885,
+      "grad_norm": 0.08232335896419818,
+      "learning_rate": 7.53570377326891e-06,
+      "loss": 0.3366,
+      "step": 2854
+    },
+    {
+      "epoch": 4.107913669064748,
+      "grad_norm": 0.09020494999270168,
+      "learning_rate": 7.512242980670481e-06,
+      "loss": 0.3452,
+      "step": 2855
+    },
+    {
+      "epoch": 4.109352517985611,
+      "grad_norm": 0.08940659670696266,
+      "learning_rate": 7.488814979752615e-06,
+      "loss": 0.3468,
+      "step": 2856
+    },
+    {
+      "epoch": 4.110791366906475,
+      "grad_norm": 0.0849791874102727,
+      "learning_rate": 7.465419794162487e-06,
+      "loss": 0.3388,
+      "step": 2857
+    },
+    {
+      "epoch": 4.1122302158273385,
+      "grad_norm": 0.09211763625939283,
+      "learning_rate": 7.442057447514144e-06,
+      "loss": 0.336,
+      "step": 2858
+    },
+    {
+      "epoch": 4.113669064748201,
+      "grad_norm": 0.09220375553797318,
+      "learning_rate": 7.418727963388481e-06,
+      "loss": 0.3368,
+      "step": 2859
+    },
+    {
+      "epoch": 4.115107913669065,
+      "grad_norm": 0.08853290729700264,
+      "learning_rate": 7.395431365333241e-06,
+      "loss": 0.3392,
+      "step": 2860
+    },
+    {
+      "epoch": 4.116546762589928,
+      "grad_norm": 0.09169049439407151,
+      "learning_rate": 7.372167676862952e-06,
+      "loss": 0.3507,
+      "step": 2861
+    },
+    {
+      "epoch": 4.117985611510791,
+      "grad_norm": 0.09536571699126575,
+      "learning_rate": 7.348936921458949e-06,
+      "loss": 0.3428,
+      "step": 2862
+    },
+    {
+      "epoch": 4.119424460431655,
+      "grad_norm": 0.08894051229026097,
+      "learning_rate": 7.325739122569282e-06,
+      "loss": 0.3465,
+      "step": 2863
+    },
+    {
+      "epoch": 4.120863309352518,
+      "grad_norm": 0.10288922380997478,
+      "learning_rate": 7.302574303608794e-06,
+      "loss": 0.3428,
+      "step": 2864
+    },
+    {
+      "epoch": 4.122302158273381,
+      "grad_norm": 0.09496950439862678,
+      "learning_rate": 7.279442487959012e-06,
+      "loss": 0.3409,
+      "step": 2865
+    },
+    {
+      "epoch": 4.123741007194244,
+      "grad_norm": 0.0900851360203878,
+      "learning_rate": 7.256343698968131e-06,
+      "loss": 0.339,
+      "step": 2866
+    },
+    {
+      "epoch": 4.125179856115108,
+      "grad_norm": 0.10725679388307317,
+      "learning_rate": 7.233277959951026e-06,
+      "loss": 0.3503,
+      "step": 2867
+    },
+    {
+      "epoch": 4.1266187050359715,
+      "grad_norm": 0.09041396138224692,
+      "learning_rate": 7.210245294189251e-06,
+      "loss": 0.3356,
+      "step": 2868
+    },
+    {
+      "epoch": 4.128057553956834,
+      "grad_norm": 0.08832255033104126,
+      "learning_rate": 7.187245724930911e-06,
+      "loss": 0.3417,
+      "step": 2869
+    },
+    {
+      "epoch": 4.129496402877698,
+      "grad_norm": 0.09162282354223023,
+      "learning_rate": 7.164279275390749e-06,
+      "loss": 0.3523,
+      "step": 2870
+    },
+    {
+      "epoch": 4.130935251798562,
+      "grad_norm": 0.09564916976117638,
+      "learning_rate": 7.14134596875006e-06,
+      "loss": 0.3455,
+      "step": 2871
+    },
+    {
+      "epoch": 4.132374100719424,
+      "grad_norm": 0.08867960402246636,
+      "learning_rate": 7.118445828156697e-06,
+      "loss": 0.3403,
+      "step": 2872
+    },
+    {
+      "epoch": 4.133812949640288,
+      "grad_norm": 0.0935344041429054,
+      "learning_rate": 7.0955788767250334e-06,
+      "loss": 0.3447,
+      "step": 2873
+    },
+    {
+      "epoch": 4.135251798561151,
+      "grad_norm": 0.10034740439432684,
+      "learning_rate": 7.0727451375359345e-06,
+      "loss": 0.3431,
+      "step": 2874
+    },
+    {
+      "epoch": 4.136690647482014,
+      "grad_norm": 0.08893466563609016,
+      "learning_rate": 7.049944633636756e-06,
+      "loss": 0.3402,
+      "step": 2875
+    },
+    {
+      "epoch": 4.138129496402878,
+      "grad_norm": 0.09170257237482983,
+      "learning_rate": 7.027177388041311e-06,
+      "loss": 0.344,
+      "step": 2876
+    },
+    {
+      "epoch": 4.139568345323741,
+      "grad_norm": 0.09811187802598147,
+      "learning_rate": 7.004443423729808e-06,
+      "loss": 0.3527,
+      "step": 2877
+    },
+    {
+      "epoch": 4.1410071942446045,
+      "grad_norm": 0.08494697223687862,
+      "learning_rate": 6.981742763648891e-06,
+      "loss": 0.3307,
+      "step": 2878
+    },
+    {
+      "epoch": 4.142446043165467,
+      "grad_norm": 0.0892102471859085,
+      "learning_rate": 6.959075430711614e-06,
+      "loss": 0.3417,
+      "step": 2879
+    },
+    {
+      "epoch": 4.143884892086331,
+      "grad_norm": 0.08902351209775089,
+      "learning_rate": 6.936441447797335e-06,
+      "loss": 0.3433,
+      "step": 2880
+    },
+    {
+      "epoch": 4.145323741007195,
+      "grad_norm": 0.09282861304930493,
+      "learning_rate": 6.913840837751778e-06,
+      "loss": 0.3512,
+      "step": 2881
+    },
+    {
+      "epoch": 4.146762589928057,
+      "grad_norm": 0.08886461511735246,
+      "learning_rate": 6.8912736233870095e-06,
+      "loss": 0.3457,
+      "step": 2882
+    },
+    {
+      "epoch": 4.148201438848921,
+      "grad_norm": 0.08739195560940712,
+      "learning_rate": 6.868739827481335e-06,
+      "loss": 0.3505,
+      "step": 2883
+    },
+    {
+      "epoch": 4.149640287769784,
+      "grad_norm": 0.08097011756110467,
+      "learning_rate": 6.846239472779359e-06,
+      "loss": 0.3376,
+      "step": 2884
+    },
+    {
+      "epoch": 4.151079136690647,
+      "grad_norm": 0.08747604945027315,
+      "learning_rate": 6.82377258199193e-06,
+      "loss": 0.3412,
+      "step": 2885
+    },
+    {
+      "epoch": 4.152517985611511,
+      "grad_norm": 0.08555350508939398,
+      "learning_rate": 6.80133917779612e-06,
+      "loss": 0.3374,
+      "step": 2886
+    },
+    {
+      "epoch": 4.153956834532374,
+      "grad_norm": 0.0938565644787564,
+      "learning_rate": 6.778939282835195e-06,
+      "loss": 0.3465,
+      "step": 2887
+    },
+    {
+      "epoch": 4.1553956834532375,
+      "grad_norm": 0.08617666533820252,
+      "learning_rate": 6.756572919718611e-06,
+      "loss": 0.3406,
+      "step": 2888
+    },
+    {
+      "epoch": 4.1568345323741,
+      "grad_norm": 0.09348935554855656,
+      "learning_rate": 6.734240111021937e-06,
+      "loss": 0.3354,
+      "step": 2889
+    },
+    {
+      "epoch": 4.158273381294964,
+      "grad_norm": 0.08692755170714953,
+      "learning_rate": 6.711940879286944e-06,
+      "loss": 0.3536,
+      "step": 2890
+    },
+    {
+      "epoch": 4.159712230215828,
+      "grad_norm": 0.09515313442434982,
+      "learning_rate": 6.689675247021461e-06,
+      "loss": 0.3456,
+      "step": 2891
+    },
+    {
+      "epoch": 4.16115107913669,
+      "grad_norm": 0.10434344890434036,
+      "learning_rate": 6.667443236699398e-06,
+      "loss": 0.3529,
+      "step": 2892
+    },
+    {
+      "epoch": 4.162589928057554,
+      "grad_norm": 0.08869915921598391,
+      "learning_rate": 6.64524487076077e-06,
+      "loss": 0.3383,
+      "step": 2893
+    },
+    {
+      "epoch": 4.164028776978418,
+      "grad_norm": 0.0920159140031608,
+      "learning_rate": 6.623080171611605e-06,
+      "loss": 0.3447,
+      "step": 2894
+    },
+    {
+      "epoch": 4.16546762589928,
+      "grad_norm": 0.09685356082379706,
+      "learning_rate": 6.600949161623939e-06,
+      "loss": 0.3511,
+      "step": 2895
+    },
+    {
+      "epoch": 4.166906474820144,
+      "grad_norm": 0.10215445669580586,
+      "learning_rate": 6.578851863135831e-06,
+      "loss": 0.3424,
+      "step": 2896
+    },
+    {
+      "epoch": 4.168345323741007,
+      "grad_norm": 0.09084195527806635,
+      "learning_rate": 6.556788298451291e-06,
+      "loss": 0.3358,
+      "step": 2897
+    },
+    {
+      "epoch": 4.1697841726618705,
+      "grad_norm": 0.09877614034532968,
+      "learning_rate": 6.534758489840296e-06,
+      "loss": 0.3506,
+      "step": 2898
+    },
+    {
+      "epoch": 4.171223021582734,
+      "grad_norm": 0.09779949648074235,
+      "learning_rate": 6.512762459538744e-06,
+      "loss": 0.3439,
+      "step": 2899
+    },
+    {
+      "epoch": 4.172661870503597,
+      "grad_norm": 0.08404080007595911,
+      "learning_rate": 6.49080022974843e-06,
+      "loss": 0.3417,
+      "step": 2900
+    },
+    {
+      "epoch": 4.174100719424461,
+      "grad_norm": 0.0964919782724387,
+      "learning_rate": 6.468871822637051e-06,
+      "loss": 0.3412,
+      "step": 2901
+    },
+    {
+      "epoch": 4.175539568345323,
+      "grad_norm": 0.12962737131413704,
+      "learning_rate": 6.446977260338152e-06,
+      "loss": 0.3445,
+      "step": 2902
+    },
+    {
+      "epoch": 4.176978417266187,
+      "grad_norm": 0.08778057787721012,
+      "learning_rate": 6.425116564951115e-06,
+      "loss": 0.3392,
+      "step": 2903
+    },
+    {
+      "epoch": 4.178417266187051,
+      "grad_norm": 0.09106657357644896,
+      "learning_rate": 6.403289758541143e-06,
+      "loss": 0.338,
+      "step": 2904
+    },
+    {
+      "epoch": 4.179856115107913,
+      "grad_norm": 0.09364878729531954,
+      "learning_rate": 6.381496863139247e-06,
+      "loss": 0.3487,
+      "step": 2905
+    },
+    {
+      "epoch": 4.181294964028777,
+      "grad_norm": 0.08472370937562462,
+      "learning_rate": 6.3597379007421755e-06,
+      "loss": 0.3533,
+      "step": 2906
+    },
+    {
+      "epoch": 4.18273381294964,
+      "grad_norm": 0.0931818196092673,
+      "learning_rate": 6.338012893312444e-06,
+      "loss": 0.3544,
+      "step": 2907
+    },
+    {
+      "epoch": 4.1841726618705035,
+      "grad_norm": 0.08969813506885384,
+      "learning_rate": 6.31632186277833e-06,
+      "loss": 0.3588,
+      "step": 2908
+    },
+    {
+      "epoch": 4.185611510791367,
+      "grad_norm": 0.08392440444302791,
+      "learning_rate": 6.294664831033746e-06,
+      "loss": 0.3434,
+      "step": 2909
+    },
+    {
+      "epoch": 4.18705035971223,
+      "grad_norm": 0.08789028884788187,
+      "learning_rate": 6.273041819938343e-06,
+      "loss": 0.353,
+      "step": 2910
+    },
+    {
+      "epoch": 4.188489208633094,
+      "grad_norm": 0.089880166974654,
+      "learning_rate": 6.251452851317421e-06,
+      "loss": 0.3456,
+      "step": 2911
+    },
+    {
+      "epoch": 4.189928057553957,
+      "grad_norm": 0.09249524397394611,
+      "learning_rate": 6.229897946961903e-06,
+      "loss": 0.3412,
+      "step": 2912
+    },
+    {
+      "epoch": 4.19136690647482,
+      "grad_norm": 0.0858512697385683,
+      "learning_rate": 6.20837712862834e-06,
+      "loss": 0.3415,
+      "step": 2913
+    },
+    {
+      "epoch": 4.192805755395684,
+      "grad_norm": 0.08636326475073688,
+      "learning_rate": 6.186890418038887e-06,
+      "loss": 0.3407,
+      "step": 2914
+    },
+    {
+      "epoch": 4.194244604316546,
+      "grad_norm": 0.08556622288993193,
+      "learning_rate": 6.165437836881256e-06,
+      "loss": 0.3357,
+      "step": 2915
+    },
+    {
+      "epoch": 4.19568345323741,
+      "grad_norm": 0.08470488394732564,
+      "learning_rate": 6.144019406808724e-06,
+      "loss": 0.3398,
+      "step": 2916
+    },
+    {
+      "epoch": 4.197122302158274,
+      "grad_norm": 0.0880734728867388,
+      "learning_rate": 6.122635149440093e-06,
+      "loss": 0.3415,
+      "step": 2917
+    },
+    {
+      "epoch": 4.1985611510791365,
+      "grad_norm": 0.11309470200267255,
+      "learning_rate": 6.101285086359645e-06,
+      "loss": 0.3442,
+      "step": 2918
+    },
+    {
+      "epoch": 4.2,
+      "grad_norm": 0.08545897562989867,
+      "learning_rate": 6.079969239117201e-06,
+      "loss": 0.3596,
+      "step": 2919
+    },
+    {
+      "epoch": 4.201438848920863,
+      "grad_norm": 0.08692145679669436,
+      "learning_rate": 6.05868762922802e-06,
+      "loss": 0.3499,
+      "step": 2920
+    },
+    {
+      "epoch": 4.202877697841727,
+      "grad_norm": 0.08901956079471854,
+      "learning_rate": 6.037440278172782e-06,
+      "loss": 0.344,
+      "step": 2921
+    },
+    {
+      "epoch": 4.20431654676259,
+      "grad_norm": 0.08386206223360473,
+      "learning_rate": 6.016227207397616e-06,
+      "loss": 0.3456,
+      "step": 2922
+    },
+    {
+      "epoch": 4.205755395683453,
+      "grad_norm": 0.08630288027571373,
+      "learning_rate": 5.995048438314044e-06,
+      "loss": 0.3433,
+      "step": 2923
+    },
+    {
+      "epoch": 4.207194244604317,
+      "grad_norm": 0.08400093611736381,
+      "learning_rate": 5.973903992298962e-06,
+      "loss": 0.3497,
+      "step": 2924
+    },
+    {
+      "epoch": 4.2086330935251794,
+      "grad_norm": 0.08297635184757658,
+      "learning_rate": 5.952793890694617e-06,
+      "loss": 0.3466,
+      "step": 2925
+    },
+    {
+      "epoch": 4.210071942446043,
+      "grad_norm": 0.08475397330983742,
+      "learning_rate": 5.9317181548086055e-06,
+      "loss": 0.344,
+      "step": 2926
+    },
+    {
+      "epoch": 4.211510791366907,
+      "grad_norm": 0.08413977347488912,
+      "learning_rate": 5.910676805913822e-06,
+      "loss": 0.3411,
+      "step": 2927
+    },
+    {
+      "epoch": 4.2129496402877695,
+      "grad_norm": 0.07947451939078741,
+      "learning_rate": 5.889669865248455e-06,
+      "loss": 0.3418,
+      "step": 2928
+    },
+    {
+      "epoch": 4.214388489208633,
+      "grad_norm": 0.08340989022991739,
+      "learning_rate": 5.8686973540159706e-06,
+      "loss": 0.3362,
+      "step": 2929
+    },
+    {
+      "epoch": 4.215827338129497,
+      "grad_norm": 0.0850053530095417,
+      "learning_rate": 5.847759293385075e-06,
+      "loss": 0.341,
+      "step": 2930
+    },
+    {
+      "epoch": 4.21726618705036,
+      "grad_norm": 0.0840802055462811,
+      "learning_rate": 5.8268557044897175e-06,
+      "loss": 0.3498,
+      "step": 2931
+    },
+    {
+      "epoch": 4.218705035971223,
+      "grad_norm": 0.08122030264526585,
+      "learning_rate": 5.805986608429019e-06,
+      "loss": 0.3404,
+      "step": 2932
+    },
+    {
+      "epoch": 4.220143884892086,
+      "grad_norm": 0.0871425920990511,
+      "learning_rate": 5.785152026267309e-06,
+      "loss": 0.3367,
+      "step": 2933
+    },
+    {
+      "epoch": 4.22158273381295,
+      "grad_norm": 0.08484976538740269,
+      "learning_rate": 5.764351979034102e-06,
+      "loss": 0.3473,
+      "step": 2934
+    },
+    {
+      "epoch": 4.223021582733813,
+      "grad_norm": 0.08530919098709946,
+      "learning_rate": 5.743586487724e-06,
+      "loss": 0.3478,
+      "step": 2935
+    },
+    {
+      "epoch": 4.224460431654676,
+      "grad_norm": 0.08269622733550601,
+      "learning_rate": 5.722855573296775e-06,
+      "loss": 0.3425,
+      "step": 2936
+    },
+    {
+      "epoch": 4.22589928057554,
+      "grad_norm": 0.08307626370675829,
+      "learning_rate": 5.702159256677266e-06,
+      "loss": 0.3475,
+      "step": 2937
+    },
+    {
+      "epoch": 4.2273381294964025,
+      "grad_norm": 0.08473481837112164,
+      "learning_rate": 5.681497558755417e-06,
+      "loss": 0.3501,
+      "step": 2938
+    },
+    {
+      "epoch": 4.228776978417266,
+      "grad_norm": 0.08407206284132669,
+      "learning_rate": 5.6608705003862085e-06,
+      "loss": 0.3466,
+      "step": 2939
+    },
+    {
+      "epoch": 4.23021582733813,
+      "grad_norm": 0.08111194604984087,
+      "learning_rate": 5.6402781023896695e-06,
+      "loss": 0.3487,
+      "step": 2940
+    },
+    {
+      "epoch": 4.231654676258993,
+      "grad_norm": 0.08693074502656296,
+      "learning_rate": 5.619720385550835e-06,
+      "loss": 0.3408,
+      "step": 2941
+    },
+    {
+      "epoch": 4.233093525179856,
+      "grad_norm": 0.0815234523575802,
+      "learning_rate": 5.5991973706197445e-06,
+      "loss": 0.337,
+      "step": 2942
+    },
+    {
+      "epoch": 4.234532374100719,
+      "grad_norm": 0.08087147008035375,
+      "learning_rate": 5.578709078311417e-06,
+      "loss": 0.3371,
+      "step": 2943
+    },
+    {
+      "epoch": 4.235971223021583,
+      "grad_norm": 0.08160489185806802,
+      "learning_rate": 5.558255529305779e-06,
+      "loss": 0.3375,
+      "step": 2944
+    },
+    {
+      "epoch": 4.237410071942446,
+      "grad_norm": 0.08593421266525207,
+      "learning_rate": 5.537836744247753e-06,
+      "loss": 0.3487,
+      "step": 2945
+    },
+    {
+      "epoch": 4.238848920863309,
+      "grad_norm": 0.08061252299367003,
+      "learning_rate": 5.517452743747145e-06,
+      "loss": 0.3447,
+      "step": 2946
+    },
+    {
+      "epoch": 4.240287769784173,
+      "grad_norm": 0.07808709488841185,
+      "learning_rate": 5.497103548378628e-06,
+      "loss": 0.3324,
+      "step": 2947
+    },
+    {
+      "epoch": 4.2417266187050355,
+      "grad_norm": 0.08486904781829617,
+      "learning_rate": 5.476789178681769e-06,
+      "loss": 0.3476,
+      "step": 2948
+    },
+    {
+      "epoch": 4.243165467625899,
+      "grad_norm": 0.08844928626365854,
+      "learning_rate": 5.456509655160989e-06,
+      "loss": 0.341,
+      "step": 2949
+    },
+    {
+      "epoch": 4.244604316546763,
+      "grad_norm": 0.07812810108986816,
+      "learning_rate": 5.436264998285516e-06,
+      "loss": 0.3549,
+      "step": 2950
+    },
+    {
+      "epoch": 4.246043165467626,
+      "grad_norm": 0.08236018914944991,
+      "learning_rate": 5.4160552284894075e-06,
+      "loss": 0.3423,
+      "step": 2951
+    },
+    {
+      "epoch": 4.247482014388489,
+      "grad_norm": 0.08402920013679564,
+      "learning_rate": 5.3958803661714865e-06,
+      "loss": 0.3536,
+      "step": 2952
+    },
+    {
+      "epoch": 4.248920863309353,
+      "grad_norm": 0.0802633227777653,
+      "learning_rate": 5.375740431695353e-06,
+      "loss": 0.3425,
+      "step": 2953
+    },
+    {
+      "epoch": 4.250359712230216,
+      "grad_norm": 0.0831045914244016,
+      "learning_rate": 5.355635445389355e-06,
+      "loss": 0.3546,
+      "step": 2954
+    },
+    {
+      "epoch": 4.251798561151079,
+      "grad_norm": 0.08031457033414124,
+      "learning_rate": 5.3355654275465584e-06,
+      "loss": 0.3471,
+      "step": 2955
+    },
+    {
+      "epoch": 4.253237410071942,
+      "grad_norm": 0.0892289236594425,
+      "learning_rate": 5.315530398424735e-06,
+      "loss": 0.3427,
+      "step": 2956
+    },
+    {
+      "epoch": 4.254676258992806,
+      "grad_norm": 0.08371451371942847,
+      "learning_rate": 5.295530378246354e-06,
+      "loss": 0.3448,
+      "step": 2957
+    },
+    {
+      "epoch": 4.256115107913669,
+      "grad_norm": 0.08107850094332589,
+      "learning_rate": 5.27556538719852e-06,
+      "loss": 0.3457,
+      "step": 2958
+    },
+    {
+      "epoch": 4.257553956834532,
+      "grad_norm": 0.08153789081196773,
+      "learning_rate": 5.2556354454329895e-06,
+      "loss": 0.3417,
+      "step": 2959
+    },
+    {
+      "epoch": 4.258992805755396,
+      "grad_norm": 0.07830203977660653,
+      "learning_rate": 5.235740573066186e-06,
+      "loss": 0.3346,
+      "step": 2960
+    },
+    {
+      "epoch": 4.260431654676259,
+      "grad_norm": 0.08268550399709272,
+      "learning_rate": 5.21588079017906e-06,
+      "loss": 0.3457,
+      "step": 2961
+    },
+    {
+      "epoch": 4.261870503597122,
+      "grad_norm": 0.08568080562957445,
+      "learning_rate": 5.196056116817194e-06,
+      "loss": 0.3517,
+      "step": 2962
+    },
+    {
+      "epoch": 4.263309352517986,
+      "grad_norm": 0.07714225270430754,
+      "learning_rate": 5.1762665729907424e-06,
+      "loss": 0.3364,
+      "step": 2963
+    },
+    {
+      "epoch": 4.264748201438849,
+      "grad_norm": 0.08884219846371594,
+      "learning_rate": 5.156512178674358e-06,
+      "loss": 0.3482,
+      "step": 2964
+    },
+    {
+      "epoch": 4.266187050359712,
+      "grad_norm": 0.08076451805775184,
+      "learning_rate": 5.136792953807242e-06,
+      "loss": 0.3411,
+      "step": 2965
+    },
+    {
+      "epoch": 4.267625899280576,
+      "grad_norm": 0.08119686193415565,
+      "learning_rate": 5.117108918293095e-06,
+      "loss": 0.3431,
+      "step": 2966
+    },
+    {
+      "epoch": 4.269064748201439,
+      "grad_norm": 0.08696701935451825,
+      "learning_rate": 5.097460092000095e-06,
+      "loss": 0.3456,
+      "step": 2967
+    },
+    {
+      "epoch": 4.270503597122302,
+      "grad_norm": 0.08407445072592444,
+      "learning_rate": 5.07784649476089e-06,
+      "loss": 0.342,
+      "step": 2968
+    },
+    {
+      "epoch": 4.271942446043165,
+      "grad_norm": 0.07878122354523241,
+      "learning_rate": 5.058268146372562e-06,
+      "loss": 0.3453,
+      "step": 2969
+    },
+    {
+      "epoch": 4.273381294964029,
+      "grad_norm": 0.08790159343290388,
+      "learning_rate": 5.038725066596595e-06,
+      "loss": 0.349,
+      "step": 2970
+    },
+    {
+      "epoch": 4.274820143884892,
+      "grad_norm": 0.0857332703437221,
+      "learning_rate": 5.019217275158923e-06,
+      "loss": 0.3445,
+      "step": 2971
+    },
+    {
+      "epoch": 4.276258992805755,
+      "grad_norm": 0.08472709552697828,
+      "learning_rate": 4.9997447917498276e-06,
+      "loss": 0.3394,
+      "step": 2972
+    },
+    {
+      "epoch": 4.277697841726619,
+      "grad_norm": 0.08659514612357604,
+      "learning_rate": 4.9803076360239335e-06,
+      "loss": 0.3386,
+      "step": 2973
+    },
+    {
+      "epoch": 4.279136690647482,
+      "grad_norm": 0.08682369026157778,
+      "learning_rate": 4.960905827600266e-06,
+      "loss": 0.3478,
+      "step": 2974
+    },
+    {
+      "epoch": 4.280575539568345,
+      "grad_norm": 0.08332643444853381,
+      "learning_rate": 4.941539386062113e-06,
+      "loss": 0.3358,
+      "step": 2975
+    },
+    {
+      "epoch": 4.282014388489209,
+      "grad_norm": 0.08330925158192157,
+      "learning_rate": 4.922208330957094e-06,
+      "loss": 0.3414,
+      "step": 2976
+    },
+    {
+      "epoch": 4.283453237410072,
+      "grad_norm": 0.09450486650057416,
+      "learning_rate": 4.902912681797114e-06,
+      "loss": 0.3451,
+      "step": 2977
+    },
+    {
+      "epoch": 4.284892086330935,
+      "grad_norm": 0.08335715797784177,
+      "learning_rate": 4.88365245805833e-06,
+      "loss": 0.3445,
+      "step": 2978
+    },
+    {
+      "epoch": 4.286330935251798,
+      "grad_norm": 0.08214126141522422,
+      "learning_rate": 4.864427679181143e-06,
+      "loss": 0.3459,
+      "step": 2979
+    },
+    {
+      "epoch": 4.287769784172662,
+      "grad_norm": 0.08242959504595915,
+      "learning_rate": 4.8452383645701815e-06,
+      "loss": 0.3442,
+      "step": 2980
+    },
+    {
+      "epoch": 4.2892086330935255,
+      "grad_norm": 0.08389674770581983,
+      "learning_rate": 4.826084533594277e-06,
+      "loss": 0.346,
+      "step": 2981
+    },
+    {
+      "epoch": 4.290647482014388,
+      "grad_norm": 0.09101394194088418,
+      "learning_rate": 4.806966205586441e-06,
+      "loss": 0.3565,
+      "step": 2982
+    },
+    {
+      "epoch": 4.292086330935252,
+      "grad_norm": 0.0833235081630495,
+      "learning_rate": 4.787883399843871e-06,
+      "loss": 0.346,
+      "step": 2983
+    },
+    {
+      "epoch": 4.293525179856115,
+      "grad_norm": 0.08064060682534645,
+      "learning_rate": 4.768836135627859e-06,
+      "loss": 0.3468,
+      "step": 2984
+    },
+    {
+      "epoch": 4.294964028776978,
+      "grad_norm": 0.07966757656484869,
+      "learning_rate": 4.749824432163888e-06,
+      "loss": 0.3482,
+      "step": 2985
+    },
+    {
+      "epoch": 4.296402877697842,
+      "grad_norm": 0.08182163847871617,
+      "learning_rate": 4.730848308641509e-06,
+      "loss": 0.337,
+      "step": 2986
+    },
+    {
+      "epoch": 4.297841726618705,
+      "grad_norm": 0.08060734717765031,
+      "learning_rate": 4.711907784214358e-06,
+      "loss": 0.3477,
+      "step": 2987
+    },
+    {
+      "epoch": 4.299280575539568,
+      "grad_norm": 0.07935103021292428,
+      "learning_rate": 4.693002878000146e-06,
+      "loss": 0.3371,
+      "step": 2988
+    },
+    {
+      "epoch": 4.300719424460432,
+      "grad_norm": 0.0858043873763291,
+      "learning_rate": 4.674133609080658e-06,
+      "loss": 0.3458,
+      "step": 2989
+    },
+    {
+      "epoch": 4.302158273381295,
+      "grad_norm": 0.08259003310366053,
+      "learning_rate": 4.6552999965016634e-06,
+      "loss": 0.3442,
+      "step": 2990
+    },
+    {
+      "epoch": 4.3035971223021585,
+      "grad_norm": 0.08425042343414377,
+      "learning_rate": 4.6365020592729694e-06,
+      "loss": 0.354,
+      "step": 2991
+    },
+    {
+      "epoch": 4.305035971223021,
+      "grad_norm": 0.0776034386383685,
+      "learning_rate": 4.617739816368367e-06,
+      "loss": 0.3486,
+      "step": 2992
+    },
+    {
+      "epoch": 4.306474820143885,
+      "grad_norm": 0.08175188536756343,
+      "learning_rate": 4.599013286725624e-06,
+      "loss": 0.344,
+      "step": 2993
+    },
+    {
+      "epoch": 4.307913669064749,
+      "grad_norm": 0.0859993200464421,
+      "learning_rate": 4.580322489246456e-06,
+      "loss": 0.351,
+      "step": 2994
+    },
+    {
+      "epoch": 4.309352517985611,
+      "grad_norm": 0.08306294500700052,
+      "learning_rate": 4.5616674427965135e-06,
+      "loss": 0.3469,
+      "step": 2995
+    },
+    {
+      "epoch": 4.310791366906475,
+      "grad_norm": 0.08738722292533195,
+      "learning_rate": 4.543048166205357e-06,
+      "loss": 0.3454,
+      "step": 2996
+    },
+    {
+      "epoch": 4.312230215827338,
+      "grad_norm": 0.09862492416042193,
+      "learning_rate": 4.524464678266452e-06,
+      "loss": 0.339,
+      "step": 2997
+    },
+    {
+      "epoch": 4.313669064748201,
+      "grad_norm": 0.07988019638011482,
+      "learning_rate": 4.505916997737143e-06,
+      "loss": 0.3451,
+      "step": 2998
+    },
+    {
+      "epoch": 4.315107913669065,
+      "grad_norm": 0.39842864982525744,
+      "learning_rate": 4.487405143338599e-06,
+      "loss": 0.3492,
+      "step": 2999
+    },
+    {
+      "epoch": 4.316546762589928,
+      "grad_norm": 0.09590906047464921,
+      "learning_rate": 4.468929133755881e-06,
+      "loss": 0.3412,
+      "step": 3000
+    },
+    {
+      "epoch": 4.3179856115107915,
+      "grad_norm": 0.08344721370744118,
+      "learning_rate": 4.450488987637824e-06,
+      "loss": 0.3529,
+      "step": 3001
+    },
+    {
+      "epoch": 4.319424460431654,
+      "grad_norm": 0.08400604144374915,
+      "learning_rate": 4.43208472359709e-06,
+      "loss": 0.3411,
+      "step": 3002
+    },
+    {
+      "epoch": 4.320863309352518,
+      "grad_norm": 0.08687667408588712,
+      "learning_rate": 4.4137163602101114e-06,
+      "loss": 0.3423,
+      "step": 3003
+    },
+    {
+      "epoch": 4.322302158273382,
+      "grad_norm": 0.0936330058653016,
+      "learning_rate": 4.3953839160170906e-06,
+      "loss": 0.3397,
+      "step": 3004
+    },
+    {
+      "epoch": 4.323741007194244,
+      "grad_norm": 0.08504338924868987,
+      "learning_rate": 4.377087409521972e-06,
+      "loss": 0.3473,
+      "step": 3005
+    },
+    {
+      "epoch": 4.325179856115108,
+      "grad_norm": 0.08688979011045339,
+      "learning_rate": 4.358826859192422e-06,
+      "loss": 0.3449,
+      "step": 3006
+    },
+    {
+      "epoch": 4.326618705035971,
+      "grad_norm": 0.08143198876697011,
+      "learning_rate": 4.340602283459827e-06,
+      "loss": 0.3433,
+      "step": 3007
+    },
+    {
+      "epoch": 4.3280575539568344,
+      "grad_norm": 0.08829899153966095,
+      "learning_rate": 4.322413700719246e-06,
+      "loss": 0.34,
+      "step": 3008
+    },
+    {
+      "epoch": 4.329496402877698,
+      "grad_norm": 0.08095509235938958,
+      "learning_rate": 4.3042611293294276e-06,
+      "loss": 0.3403,
+      "step": 3009
+    },
+    {
+      "epoch": 4.330935251798561,
+      "grad_norm": 0.08489746893790455,
+      "learning_rate": 4.28614458761274e-06,
+      "loss": 0.346,
+      "step": 3010
+    },
+    {
+      "epoch": 4.3323741007194245,
+      "grad_norm": 0.08758068155942195,
+      "learning_rate": 4.2680640938552245e-06,
+      "loss": 0.3519,
+      "step": 3011
+    },
+    {
+      "epoch": 4.333812949640288,
+      "grad_norm": 0.07758055235191116,
+      "learning_rate": 4.250019666306515e-06,
+      "loss": 0.3374,
+      "step": 3012
+    },
+    {
+      "epoch": 4.335251798561151,
+      "grad_norm": 0.08417028239751985,
+      "learning_rate": 4.232011323179839e-06,
+      "loss": 0.3434,
+      "step": 3013
+    },
+    {
+      "epoch": 4.336690647482015,
+      "grad_norm": 0.08546010877698863,
+      "learning_rate": 4.214039082652002e-06,
+      "loss": 0.3504,
+      "step": 3014
+    },
+    {
+      "epoch": 4.338129496402877,
+      "grad_norm": 0.08671087781325834,
+      "learning_rate": 4.1961029628634e-06,
+      "loss": 0.3502,
+      "step": 3015
+    },
+    {
+      "epoch": 4.339568345323741,
+      "grad_norm": 0.08271817851173717,
+      "learning_rate": 4.17820298191792e-06,
+      "loss": 0.3431,
+      "step": 3016
+    },
+    {
+      "epoch": 4.341007194244605,
+      "grad_norm": 0.07408603517895467,
+      "learning_rate": 4.160339157883e-06,
+      "loss": 0.3379,
+      "step": 3017
+    },
+    {
+      "epoch": 4.3424460431654675,
+      "grad_norm": 0.07970953911087428,
+      "learning_rate": 4.142511508789606e-06,
+      "loss": 0.3442,
+      "step": 3018
+    },
+    {
+      "epoch": 4.343884892086331,
+      "grad_norm": 0.07987015124686353,
+      "learning_rate": 4.1247200526321364e-06,
+      "loss": 0.3473,
+      "step": 3019
+    },
+    {
+      "epoch": 4.345323741007194,
+      "grad_norm": 0.08194500782890717,
+      "learning_rate": 4.106964807368496e-06,
+      "loss": 0.3364,
+      "step": 3020
+    },
+    {
+      "epoch": 4.3467625899280575,
+      "grad_norm": 0.08249522124831325,
+      "learning_rate": 4.089245790920031e-06,
+      "loss": 0.354,
+      "step": 3021
+    },
+    {
+      "epoch": 4.348201438848921,
+      "grad_norm": 0.08249382730822125,
+      "learning_rate": 4.071563021171523e-06,
+      "loss": 0.3328,
+      "step": 3022
+    },
+    {
+      "epoch": 4.349640287769784,
+      "grad_norm": 0.07931275073284706,
+      "learning_rate": 4.0539165159711615e-06,
+      "loss": 0.3464,
+      "step": 3023
+    },
+    {
+      "epoch": 4.351079136690648,
+      "grad_norm": 0.08131517415986211,
+      "learning_rate": 4.036306293130543e-06,
+      "loss": 0.345,
+      "step": 3024
+    },
+    {
+      "epoch": 4.35251798561151,
+      "grad_norm": 0.07903812308544889,
+      "learning_rate": 4.01873237042461e-06,
+      "loss": 0.3397,
+      "step": 3025
+    },
+    {
+      "epoch": 4.353956834532374,
+      "grad_norm": 0.07840845669837926,
+      "learning_rate": 4.001194765591723e-06,
+      "loss": 0.3396,
+      "step": 3026
+    },
+    {
+      "epoch": 4.355395683453238,
+      "grad_norm": 0.0830296898055647,
+      "learning_rate": 3.983693496333522e-06,
+      "loss": 0.3519,
+      "step": 3027
+    },
+    {
+      "epoch": 4.3568345323741005,
+      "grad_norm": 0.07885795958132294,
+      "learning_rate": 3.966228580315017e-06,
+      "loss": 0.3445,
+      "step": 3028
+    },
+    {
+      "epoch": 4.358273381294964,
+      "grad_norm": 0.08003001337742997,
+      "learning_rate": 3.9488000351645036e-06,
+      "loss": 0.3417,
+      "step": 3029
+    },
+    {
+      "epoch": 4.359712230215827,
+      "grad_norm": 0.08063767668536838,
+      "learning_rate": 3.931407878473575e-06,
+      "loss": 0.3324,
+      "step": 3030
+    },
+    {
+      "epoch": 4.3611510791366905,
+      "grad_norm": 0.07926094066180016,
+      "learning_rate": 3.914052127797088e-06,
+      "loss": 0.3459,
+      "step": 3031
+    },
+    {
+      "epoch": 4.362589928057554,
+      "grad_norm": 0.07902322917345408,
+      "learning_rate": 3.8967328006531605e-06,
+      "loss": 0.342,
+      "step": 3032
+    },
+    {
+      "epoch": 4.364028776978417,
+      "grad_norm": 0.08061229625077315,
+      "learning_rate": 3.879449914523137e-06,
+      "loss": 0.339,
+      "step": 3033
+    },
+    {
+      "epoch": 4.365467625899281,
+      "grad_norm": 0.08700057193394455,
+      "learning_rate": 3.862203486851588e-06,
+      "loss": 0.3408,
+      "step": 3034
+    },
+    {
+      "epoch": 4.366906474820144,
+      "grad_norm": 0.07834873633345833,
+      "learning_rate": 3.844993535046291e-06,
+      "loss": 0.3486,
+      "step": 3035
+    },
+    {
+      "epoch": 4.368345323741007,
+      "grad_norm": 0.07885514080870147,
+      "learning_rate": 3.8278200764781725e-06,
+      "loss": 0.3461,
+      "step": 3036
+    },
+    {
+      "epoch": 4.369784172661871,
+      "grad_norm": 0.07559422550427315,
+      "learning_rate": 3.8106831284813718e-06,
+      "loss": 0.3372,
+      "step": 3037
+    },
+    {
+      "epoch": 4.3712230215827335,
+      "grad_norm": 0.0870947644704936,
+      "learning_rate": 3.7935827083531585e-06,
+      "loss": 0.3332,
+      "step": 3038
+    },
+    {
+      "epoch": 4.372661870503597,
+      "grad_norm": 0.08258021084009133,
+      "learning_rate": 3.7765188333539037e-06,
+      "loss": 0.3389,
+      "step": 3039
+    },
+    {
+      "epoch": 4.374100719424461,
+      "grad_norm": 0.07932424220275947,
+      "learning_rate": 3.759491520707119e-06,
+      "loss": 0.3467,
+      "step": 3040
+    },
+    {
+      "epoch": 4.3755395683453235,
+      "grad_norm": 0.0784928567073004,
+      "learning_rate": 3.74250078759943e-06,
+      "loss": 0.3442,
+      "step": 3041
+    },
+    {
+      "epoch": 4.376978417266187,
+      "grad_norm": 0.07612824084672545,
+      "learning_rate": 3.7255466511805007e-06,
+      "loss": 0.3378,
+      "step": 3042
+    },
+    {
+      "epoch": 4.37841726618705,
+      "grad_norm": 0.07607197089248201,
+      "learning_rate": 3.7086291285630683e-06,
+      "loss": 0.3394,
+      "step": 3043
+    },
+    {
+      "epoch": 4.379856115107914,
+      "grad_norm": 0.08569927997685474,
+      "learning_rate": 3.6917482368229406e-06,
+      "loss": 0.3445,
+      "step": 3044
+    },
+    {
+      "epoch": 4.381294964028777,
+      "grad_norm": 0.08379836100508248,
+      "learning_rate": 3.674903992998915e-06,
+      "loss": 0.3549,
+      "step": 3045
+    },
+    {
+      "epoch": 4.38273381294964,
+      "grad_norm": 0.08145298713314272,
+      "learning_rate": 3.6580964140928133e-06,
+      "loss": 0.3391,
+      "step": 3046
+    },
+    {
+      "epoch": 4.384172661870504,
+      "grad_norm": 0.07745765869883456,
+      "learning_rate": 3.6413255170694515e-06,
+      "loss": 0.3468,
+      "step": 3047
+    },
+    {
+      "epoch": 4.385611510791367,
+      "grad_norm": 0.0753590285626336,
+      "learning_rate": 3.6245913188566227e-06,
+      "loss": 0.3466,
+      "step": 3048
+    },
+    {
+      "epoch": 4.38705035971223,
+      "grad_norm": 0.08003272434455065,
+      "learning_rate": 3.607893836345069e-06,
+      "loss": 0.3397,
+      "step": 3049
+    },
+    {
+      "epoch": 4.388489208633094,
+      "grad_norm": 0.08303712301744344,
+      "learning_rate": 3.5912330863884904e-06,
+      "loss": 0.3358,
+      "step": 3050
+    },
+    {
+      "epoch": 4.3899280575539565,
+      "grad_norm": 0.07970129873064605,
+      "learning_rate": 3.574609085803471e-06,
+      "loss": 0.3373,
+      "step": 3051
+    },
+    {
+      "epoch": 4.39136690647482,
+      "grad_norm": 0.08026216266485611,
+      "learning_rate": 3.5580218513695573e-06,
+      "loss": 0.3391,
+      "step": 3052
+    },
+    {
+      "epoch": 4.392805755395684,
+      "grad_norm": 0.08038839081273219,
+      "learning_rate": 3.5414713998291483e-06,
+      "loss": 0.3467,
+      "step": 3053
+    },
+    {
+      "epoch": 4.394244604316547,
+      "grad_norm": 0.0837473501114694,
+      "learning_rate": 3.524957747887512e-06,
+      "loss": 0.3497,
+      "step": 3054
+    },
+    {
+      "epoch": 4.39568345323741,
+      "grad_norm": 0.08117837642280494,
+      "learning_rate": 3.5084809122128125e-06,
+      "loss": 0.344,
+      "step": 3055
+    },
+    {
+      "epoch": 4.397122302158273,
+      "grad_norm": 0.08677566517858053,
+      "learning_rate": 3.4920409094360054e-06,
+      "loss": 0.347,
+      "step": 3056
+    },
+    {
+      "epoch": 4.398561151079137,
+      "grad_norm": 0.07741552711244623,
+      "learning_rate": 3.475637756150896e-06,
+      "loss": 0.3412,
+      "step": 3057
+    },
+    {
+      "epoch": 4.4,
+      "grad_norm": 0.07796545704124043,
+      "learning_rate": 3.4592714689140895e-06,
+      "loss": 0.3459,
+      "step": 3058
+    },
+    {
+      "epoch": 4.401438848920863,
+      "grad_norm": 0.08349721210960026,
+      "learning_rate": 3.442942064244981e-06,
+      "loss": 0.3448,
+      "step": 3059
+    },
+    {
+      "epoch": 4.402877697841727,
+      "grad_norm": 0.08567701444737574,
+      "learning_rate": 3.426649558625732e-06,
+      "loss": 0.3384,
+      "step": 3060
+    },
+    {
+      "epoch": 4.4043165467625895,
+      "grad_norm": 0.08528488543691852,
+      "learning_rate": 3.4103939685012823e-06,
+      "loss": 0.3406,
+      "step": 3061
+    },
+    {
+      "epoch": 4.405755395683453,
+      "grad_norm": 0.0812158856847987,
+      "learning_rate": 3.3941753102792617e-06,
+      "loss": 0.3482,
+      "step": 3062
+    },
+    {
+      "epoch": 4.407194244604317,
+      "grad_norm": 0.0763936300963798,
+      "learning_rate": 3.377993600330083e-06,
+      "loss": 0.342,
+      "step": 3063
+    },
+    {
+      "epoch": 4.40863309352518,
+      "grad_norm": 0.07925828017399017,
+      "learning_rate": 3.361848854986831e-06,
+      "loss": 0.3398,
+      "step": 3064
+    },
+    {
+      "epoch": 4.410071942446043,
+      "grad_norm": 0.0808463931849389,
+      "learning_rate": 3.3457410905452624e-06,
+      "loss": 0.349,
+      "step": 3065
+    },
+    {
+      "epoch": 4.411510791366906,
+      "grad_norm": 0.07916967431613016,
+      "learning_rate": 3.3296703232638606e-06,
+      "loss": 0.3402,
+      "step": 3066
+    },
+    {
+      "epoch": 4.41294964028777,
+      "grad_norm": 0.08007145961041624,
+      "learning_rate": 3.3136365693637294e-06,
+      "loss": 0.3496,
+      "step": 3067
+    },
+    {
+      "epoch": 4.414388489208633,
+      "grad_norm": 0.07548848994529554,
+      "learning_rate": 3.297639845028604e-06,
+      "loss": 0.3462,
+      "step": 3068
+    },
+    {
+      "epoch": 4.415827338129496,
+      "grad_norm": 0.07812403206266526,
+      "learning_rate": 3.281680166404857e-06,
+      "loss": 0.3497,
+      "step": 3069
+    },
+    {
+      "epoch": 4.41726618705036,
+      "grad_norm": 0.07731420504538403,
+      "learning_rate": 3.265757549601496e-06,
+      "loss": 0.3546,
+      "step": 3070
+    },
+    {
+      "epoch": 4.418705035971223,
+      "grad_norm": 0.07795361207341998,
+      "learning_rate": 3.249872010690074e-06,
+      "loss": 0.3426,
+      "step": 3071
+    },
+    {
+      "epoch": 4.420143884892086,
+      "grad_norm": 0.07931639958151648,
+      "learning_rate": 3.234023565704738e-06,
+      "loss": 0.347,
+      "step": 3072
+    },
+    {
+      "epoch": 4.42158273381295,
+      "grad_norm": 0.07836346779831599,
+      "learning_rate": 3.2182122306422035e-06,
+      "loss": 0.353,
+      "step": 3073
+    },
+    {
+      "epoch": 4.423021582733813,
+      "grad_norm": 0.08262669100761946,
+      "learning_rate": 3.2024380214617136e-06,
+      "loss": 0.342,
+      "step": 3074
+    },
+    {
+      "epoch": 4.424460431654676,
+      "grad_norm": 0.07827073218996666,
+      "learning_rate": 3.186700954085056e-06,
+      "loss": 0.3428,
+      "step": 3075
+    },
+    {
+      "epoch": 4.42589928057554,
+      "grad_norm": 0.07531361706208638,
+      "learning_rate": 3.1710010443965065e-06,
+      "loss": 0.3419,
+      "step": 3076
+    },
+    {
+      "epoch": 4.427338129496403,
+      "grad_norm": 0.08151124013280812,
+      "learning_rate": 3.1553383082428568e-06,
+      "loss": 0.3401,
+      "step": 3077
+    },
+    {
+      "epoch": 4.428776978417266,
+      "grad_norm": 0.07240958763049264,
+      "learning_rate": 3.139712761433367e-06,
+      "loss": 0.3452,
+      "step": 3078
+    },
+    {
+      "epoch": 4.430215827338129,
+      "grad_norm": 1.719996588847683,
+      "learning_rate": 3.1241244197397626e-06,
+      "loss": 0.369,
+      "step": 3079
+    },
+    {
+      "epoch": 4.431654676258993,
+      "grad_norm": 0.07852556430797913,
+      "learning_rate": 3.1085732988962003e-06,
+      "loss": 0.3488,
+      "step": 3080
+    },
+    {
+      "epoch": 4.433093525179856,
+      "grad_norm": 0.077465495792956,
+      "learning_rate": 3.0930594145993063e-06,
+      "loss": 0.3451,
+      "step": 3081
+    },
+    {
+      "epoch": 4.434532374100719,
+      "grad_norm": 0.07838966292674063,
+      "learning_rate": 3.077582782508075e-06,
+      "loss": 0.3499,
+      "step": 3082
+    },
+    {
+      "epoch": 4.435971223021583,
+      "grad_norm": 0.07511031589145918,
+      "learning_rate": 3.0621434182439345e-06,
+      "loss": 0.3461,
+      "step": 3083
+    },
+    {
+      "epoch": 4.437410071942446,
+      "grad_norm": 0.07451080337790358,
+      "learning_rate": 3.0467413373906773e-06,
+      "loss": 0.3388,
+      "step": 3084
+    },
+    {
+      "epoch": 4.438848920863309,
+      "grad_norm": 0.07774781165516859,
+      "learning_rate": 3.0313765554944806e-06,
+      "loss": 0.3393,
+      "step": 3085
+    },
+    {
+      "epoch": 4.440287769784173,
+      "grad_norm": 0.07923173316894326,
+      "learning_rate": 3.0160490880638593e-06,
+      "loss": 0.3399,
+      "step": 3086
+    },
+    {
+      "epoch": 4.441726618705036,
+      "grad_norm": 0.07684985118117568,
+      "learning_rate": 3.0007589505696645e-06,
+      "loss": 0.3415,
+      "step": 3087
+    },
+    {
+      "epoch": 4.443165467625899,
+      "grad_norm": 0.07590897552642471,
+      "learning_rate": 2.9855061584450795e-06,
+      "loss": 0.3457,
+      "step": 3088
+    },
+    {
+      "epoch": 4.444604316546762,
+      "grad_norm": 0.07207050368652652,
+      "learning_rate": 2.97029072708559e-06,
+      "loss": 0.3371,
+      "step": 3089
+    },
+    {
+      "epoch": 4.446043165467626,
+      "grad_norm": 0.08091106041280295,
+      "learning_rate": 2.955112671848963e-06,
+      "loss": 0.3382,
+      "step": 3090
+    },
+    {
+      "epoch": 4.4474820143884894,
+      "grad_norm": 0.07479526898230791,
+      "learning_rate": 2.9399720080552383e-06,
+      "loss": 0.3379,
+      "step": 3091
+    },
+    {
+      "epoch": 4.448920863309352,
+      "grad_norm": 0.07771800650294812,
+      "learning_rate": 2.924868750986729e-06,
+      "loss": 0.3473,
+      "step": 3092
+    },
+    {
+      "epoch": 4.450359712230216,
+      "grad_norm": 0.07650163090027194,
+      "learning_rate": 2.9098029158879914e-06,
+      "loss": 0.341,
+      "step": 3093
+    },
+    {
+      "epoch": 4.4517985611510795,
+      "grad_norm": 0.07242688225084594,
+      "learning_rate": 2.8947745179657815e-06,
+      "loss": 0.3392,
+      "step": 3094
+    },
+    {
+      "epoch": 4.453237410071942,
+      "grad_norm": 0.07827079698152796,
+      "learning_rate": 2.8797835723890944e-06,
+      "loss": 0.3479,
+      "step": 3095
+    },
+    {
+      "epoch": 4.454676258992806,
+      "grad_norm": 0.08222727514776726,
+      "learning_rate": 2.864830094289137e-06,
+      "loss": 0.3433,
+      "step": 3096
+    },
+    {
+      "epoch": 4.456115107913669,
+      "grad_norm": 0.07515734856204817,
+      "learning_rate": 2.84991409875925e-06,
+      "loss": 0.3405,
+      "step": 3097
+    },
+    {
+      "epoch": 4.457553956834532,
+      "grad_norm": 0.0804660226850685,
+      "learning_rate": 2.8350356008549806e-06,
+      "loss": 0.3345,
+      "step": 3098
+    },
+    {
+      "epoch": 4.458992805755396,
+      "grad_norm": 0.07612451962872864,
+      "learning_rate": 2.8201946155940142e-06,
+      "loss": 0.3448,
+      "step": 3099
+    },
+    {
+      "epoch": 4.460431654676259,
+      "grad_norm": 0.0769348032440111,
+      "learning_rate": 2.8053911579561764e-06,
+      "loss": 0.3494,
+      "step": 3100
+    },
+    {
+      "epoch": 4.4618705035971225,
+      "grad_norm": 0.08222538513305827,
+      "learning_rate": 2.7906252428834044e-06,
+      "loss": 0.3485,
+      "step": 3101
+    },
+    {
+      "epoch": 4.463309352517985,
+      "grad_norm": 0.08580367255632801,
+      "learning_rate": 2.7758968852797542e-06,
+      "loss": 0.3451,
+      "step": 3102
+    },
+    {
+      "epoch": 4.464748201438849,
+      "grad_norm": 0.07748436073576172,
+      "learning_rate": 2.761206100011369e-06,
+      "loss": 0.3366,
+      "step": 3103
+    },
+    {
+      "epoch": 4.4661870503597125,
+      "grad_norm": 0.07354963767301814,
+      "learning_rate": 2.746552901906463e-06,
+      "loss": 0.3456,
+      "step": 3104
+    },
+    {
+      "epoch": 4.467625899280575,
+      "grad_norm": 0.0868966731070377,
+      "learning_rate": 2.731937305755321e-06,
+      "loss": 0.3377,
+      "step": 3105
+    },
+    {
+      "epoch": 4.469064748201439,
+      "grad_norm": 0.07788029701225116,
+      "learning_rate": 2.717359326310249e-06,
+      "loss": 0.3415,
+      "step": 3106
+    },
+    {
+      "epoch": 4.470503597122303,
+      "grad_norm": 0.08433311420417798,
+      "learning_rate": 2.702818978285633e-06,
+      "loss": 0.35,
+      "step": 3107
+    },
+    {
+      "epoch": 4.471942446043165,
+      "grad_norm": 0.07443447534638524,
+      "learning_rate": 2.688316276357825e-06,
+      "loss": 0.3479,
+      "step": 3108
+    },
+    {
+      "epoch": 4.473381294964029,
+      "grad_norm": 0.08059610410020289,
+      "learning_rate": 2.6738512351652012e-06,
+      "loss": 0.34,
+      "step": 3109
+    },
+    {
+      "epoch": 4.474820143884892,
+      "grad_norm": 0.08193043694761587,
+      "learning_rate": 2.65942386930814e-06,
+      "loss": 0.3406,
+      "step": 3110
+    },
+    {
+      "epoch": 4.4762589928057555,
+      "grad_norm": 0.08286360992951043,
+      "learning_rate": 2.645034193348961e-06,
+      "loss": 0.3405,
+      "step": 3111
+    },
+    {
+      "epoch": 4.477697841726619,
+      "grad_norm": 0.07424441864030382,
+      "learning_rate": 2.6306822218119533e-06,
+      "loss": 0.3431,
+      "step": 3112
+    },
+    {
+      "epoch": 4.479136690647482,
+      "grad_norm": 0.07416878253769449,
+      "learning_rate": 2.61636796918336e-06,
+      "loss": 0.3339,
+      "step": 3113
+    },
+    {
+      "epoch": 4.4805755395683455,
+      "grad_norm": 0.07750644969509364,
+      "learning_rate": 2.6020914499113438e-06,
+      "loss": 0.3406,
+      "step": 3114
+    },
+    {
+      "epoch": 4.482014388489208,
+      "grad_norm": 0.07582412223932053,
+      "learning_rate": 2.587852678405973e-06,
+      "loss": 0.3448,
+      "step": 3115
+    },
+    {
+      "epoch": 4.483453237410072,
+      "grad_norm": 0.07739574975156975,
+      "learning_rate": 2.5736516690392366e-06,
+      "loss": 0.3453,
+      "step": 3116
+    },
+    {
+      "epoch": 4.484892086330936,
+      "grad_norm": 0.07794678185547059,
+      "learning_rate": 2.5594884361449746e-06,
+      "loss": 0.3467,
+      "step": 3117
+    },
+    {
+      "epoch": 4.486330935251798,
+      "grad_norm": 0.07718586965445183,
+      "learning_rate": 2.5453629940189338e-06,
+      "loss": 0.3433,
+      "step": 3118
+    },
+    {
+      "epoch": 4.487769784172662,
+      "grad_norm": 0.07665613157929926,
+      "learning_rate": 2.531275356918701e-06,
+      "loss": 0.3421,
+      "step": 3119
+    },
+    {
+      "epoch": 4.489208633093525,
+      "grad_norm": 0.07599974393918184,
+      "learning_rate": 2.5172255390636878e-06,
+      "loss": 0.3463,
+      "step": 3120
+    },
+    {
+      "epoch": 4.4906474820143885,
+      "grad_norm": 0.07487563325987932,
+      "learning_rate": 2.5032135546351644e-06,
+      "loss": 0.3385,
+      "step": 3121
+    },
+    {
+      "epoch": 4.492086330935252,
+      "grad_norm": 0.07483111999350718,
+      "learning_rate": 2.4892394177761947e-06,
+      "loss": 0.3364,
+      "step": 3122
+    },
+    {
+      "epoch": 4.493525179856115,
+      "grad_norm": 0.07789042210389767,
+      "learning_rate": 2.475303142591634e-06,
+      "loss": 0.3547,
+      "step": 3123
+    },
+    {
+      "epoch": 4.4949640287769785,
+      "grad_norm": 0.07604161844796199,
+      "learning_rate": 2.461404743148141e-06,
+      "loss": 0.341,
+      "step": 3124
+    },
+    {
+      "epoch": 4.496402877697841,
+      "grad_norm": 0.07830740061499633,
+      "learning_rate": 2.4475442334741306e-06,
+      "loss": 0.3412,
+      "step": 3125
+    },
+    {
+      "epoch": 4.497841726618705,
+      "grad_norm": 0.07520493007992193,
+      "learning_rate": 2.43372162755978e-06,
+      "loss": 0.347,
+      "step": 3126
+    },
+    {
+      "epoch": 4.499280575539569,
+      "grad_norm": 0.07341117177491838,
+      "learning_rate": 2.419936939357004e-06,
+      "loss": 0.3445,
+      "step": 3127
+    },
+    {
+      "epoch": 4.500719424460431,
+      "grad_norm": 0.07534906304450155,
+      "learning_rate": 2.4061901827794466e-06,
+      "loss": 0.3414,
+      "step": 3128
+    },
+    {
+      "epoch": 4.502158273381295,
+      "grad_norm": 0.07719835961500789,
+      "learning_rate": 2.3924813717024663e-06,
+      "loss": 0.3431,
+      "step": 3129
+    },
+    {
+      "epoch": 4.503597122302159,
+      "grad_norm": 0.07607486825470737,
+      "learning_rate": 2.378810519963124e-06,
+      "loss": 0.3566,
+      "step": 3130
+    },
+    {
+      "epoch": 4.5050359712230215,
+      "grad_norm": 0.07706948930854154,
+      "learning_rate": 2.3651776413601634e-06,
+      "loss": 0.3472,
+      "step": 3131
+    },
+    {
+      "epoch": 4.506474820143885,
+      "grad_norm": 0.07022086982451177,
+      "learning_rate": 2.3515827496539823e-06,
+      "loss": 0.3338,
+      "step": 3132
+    },
+    {
+      "epoch": 4.507913669064748,
+      "grad_norm": 0.07459929303125831,
+      "learning_rate": 2.3380258585666793e-06,
+      "loss": 0.3506,
+      "step": 3133
+    },
+    {
+      "epoch": 4.5093525179856115,
+      "grad_norm": 0.07260413687930173,
+      "learning_rate": 2.324506981781949e-06,
+      "loss": 0.3476,
+      "step": 3134
+    },
+    {
+      "epoch": 4.510791366906475,
+      "grad_norm": 0.07566417402073569,
+      "learning_rate": 2.311026132945138e-06,
+      "loss": 0.3458,
+      "step": 3135
+    },
+    {
+      "epoch": 4.512230215827338,
+      "grad_norm": 0.08908274283781233,
+      "learning_rate": 2.297583325663233e-06,
+      "loss": 0.3413,
+      "step": 3136
+    },
+    {
+      "epoch": 4.513669064748202,
+      "grad_norm": 0.07587416190007623,
+      "learning_rate": 2.2841785735047717e-06,
+      "loss": 0.3432,
+      "step": 3137
+    },
+    {
+      "epoch": 4.515107913669064,
+      "grad_norm": 0.07235969865275474,
+      "learning_rate": 2.2708118899999175e-06,
+      "loss": 0.3403,
+      "step": 3138
+    },
+    {
+      "epoch": 4.516546762589928,
+      "grad_norm": 0.07571752220304737,
+      "learning_rate": 2.2574832886403988e-06,
+      "loss": 0.3382,
+      "step": 3139
+    },
+    {
+      "epoch": 4.517985611510792,
+      "grad_norm": 0.2794684253861891,
+      "learning_rate": 2.2441927828795106e-06,
+      "loss": 0.3579,
+      "step": 3140
+    },
+    {
+      "epoch": 4.5194244604316545,
+      "grad_norm": 0.08188168452046989,
+      "learning_rate": 2.230940386132088e-06,
+      "loss": 0.3467,
+      "step": 3141
+    },
+    {
+      "epoch": 4.520863309352518,
+      "grad_norm": 0.07617472768450836,
+      "learning_rate": 2.21772611177451e-06,
+      "loss": 0.337,
+      "step": 3142
+    },
+    {
+      "epoch": 4.522302158273382,
+      "grad_norm": 0.07322485314476528,
+      "learning_rate": 2.204549973144654e-06,
+      "loss": 0.3377,
+      "step": 3143
+    },
+    {
+      "epoch": 4.5237410071942445,
+      "grad_norm": 0.07232542143510301,
+      "learning_rate": 2.1914119835419358e-06,
+      "loss": 0.3335,
+      "step": 3144
+    },
+    {
+      "epoch": 4.525179856115108,
+      "grad_norm": 0.07279597533953248,
+      "learning_rate": 2.178312156227258e-06,
+      "loss": 0.3464,
+      "step": 3145
+    },
+    {
+      "epoch": 4.526618705035971,
+      "grad_norm": 0.07658839683786779,
+      "learning_rate": 2.1652505044229734e-06,
+      "loss": 0.3374,
+      "step": 3146
+    },
+    {
+      "epoch": 4.528057553956835,
+      "grad_norm": 0.07706757501700515,
+      "learning_rate": 2.1522270413129444e-06,
+      "loss": 0.3329,
+      "step": 3147
+    },
+    {
+      "epoch": 4.529496402877697,
+      "grad_norm": 0.07503480245602073,
+      "learning_rate": 2.1392417800424738e-06,
+      "loss": 0.3372,
+      "step": 3148
+    },
+    {
+      "epoch": 4.530935251798561,
+      "grad_norm": 0.07445954513335917,
+      "learning_rate": 2.1262947337182815e-06,
+      "loss": 0.3449,
+      "step": 3149
+    },
+    {
+      "epoch": 4.532374100719425,
+      "grad_norm": 0.0762378365425642,
+      "learning_rate": 2.113385915408546e-06,
+      "loss": 0.3463,
+      "step": 3150
+    },
+    {
+      "epoch": 4.5338129496402875,
+      "grad_norm": 0.07408879638626946,
+      "learning_rate": 2.100515338142839e-06,
+      "loss": 0.3396,
+      "step": 3151
+    },
+    {
+      "epoch": 4.535251798561151,
+      "grad_norm": 0.07776095998479204,
+      "learning_rate": 2.087683014912152e-06,
+      "loss": 0.3448,
+      "step": 3152
+    },
+    {
+      "epoch": 4.536690647482015,
+      "grad_norm": 0.07329928007429921,
+      "learning_rate": 2.0748889586688526e-06,
+      "loss": 0.3456,
+      "step": 3153
+    },
+    {
+      "epoch": 4.5381294964028775,
+      "grad_norm": 0.0716971153654081,
+      "learning_rate": 2.0621331823266777e-06,
+      "loss": 0.3434,
+      "step": 3154
+    },
+    {
+      "epoch": 4.539568345323741,
+      "grad_norm": 0.07432013950757492,
+      "learning_rate": 2.049415698760746e-06,
+      "loss": 0.3365,
+      "step": 3155
+    },
+    {
+      "epoch": 4.541007194244604,
+      "grad_norm": 0.07400174544490337,
+      "learning_rate": 2.036736520807505e-06,
+      "loss": 0.3405,
+      "step": 3156
+    },
+    {
+      "epoch": 4.542446043165468,
+      "grad_norm": 0.07618606779242666,
+      "learning_rate": 2.0240956612647487e-06,
+      "loss": 0.3431,
+      "step": 3157
+    },
+    {
+      "epoch": 4.543884892086331,
+      "grad_norm": 0.08163048401173585,
+      "learning_rate": 2.011493132891591e-06,
+      "loss": 0.3449,
+      "step": 3158
+    },
+    {
+      "epoch": 4.545323741007194,
+      "grad_norm": 0.07680418104811734,
+      "learning_rate": 1.998928948408465e-06,
+      "loss": 0.3411,
+      "step": 3159
+    },
+    {
+      "epoch": 4.546762589928058,
+      "grad_norm": 0.07883169821475698,
+      "learning_rate": 1.9864031204970847e-06,
+      "loss": 0.3514,
+      "step": 3160
+    },
+    {
+      "epoch": 4.5482014388489205,
+      "grad_norm": 0.07376418720795792,
+      "learning_rate": 1.973915661800452e-06,
+      "loss": 0.3449,
+      "step": 3161
+    },
+    {
+      "epoch": 4.549640287769784,
+      "grad_norm": 0.07628001522921406,
+      "learning_rate": 1.9614665849228666e-06,
+      "loss": 0.3412,
+      "step": 3162
+    },
+    {
+      "epoch": 4.551079136690648,
+      "grad_norm": 0.07268400084313503,
+      "learning_rate": 1.949055902429846e-06,
+      "loss": 0.3429,
+      "step": 3163
+    },
+    {
+      "epoch": 4.5525179856115106,
+      "grad_norm": 0.07535215186043386,
+      "learning_rate": 1.936683626848179e-06,
+      "loss": 0.3418,
+      "step": 3164
+    },
+    {
+      "epoch": 4.553956834532374,
+      "grad_norm": 0.08101213007294282,
+      "learning_rate": 1.9243497706658944e-06,
+      "loss": 0.3359,
+      "step": 3165
+    },
+    {
+      "epoch": 4.555395683453238,
+      "grad_norm": 0.07516027494800719,
+      "learning_rate": 1.9120543463322238e-06,
+      "loss": 0.3362,
+      "step": 3166
+    },
+    {
+      "epoch": 4.556834532374101,
+      "grad_norm": 0.07281658610061269,
+      "learning_rate": 1.899797366257614e-06,
+      "loss": 0.3506,
+      "step": 3167
+    },
+    {
+      "epoch": 4.558273381294964,
+      "grad_norm": 0.0749458781232783,
+      "learning_rate": 1.887578842813711e-06,
+      "loss": 0.3408,
+      "step": 3168
+    },
+    {
+      "epoch": 4.559712230215827,
+      "grad_norm": 0.07758718799261118,
+      "learning_rate": 1.875398788333347e-06,
+      "loss": 0.3445,
+      "step": 3169
+    },
+    {
+      "epoch": 4.561151079136691,
+      "grad_norm": 0.07215043406316093,
+      "learning_rate": 1.8632572151105189e-06,
+      "loss": 0.3311,
+      "step": 3170
+    },
+    {
+      "epoch": 4.5625899280575535,
+      "grad_norm": 0.07224962203751209,
+      "learning_rate": 1.8511541354003882e-06,
+      "loss": 0.3345,
+      "step": 3171
+    },
+    {
+      "epoch": 4.564028776978417,
+      "grad_norm": 0.07432208385081143,
+      "learning_rate": 1.8390895614192405e-06,
+      "loss": 0.3437,
+      "step": 3172
+    },
+    {
+      "epoch": 4.565467625899281,
+      "grad_norm": 0.07270811076997219,
+      "learning_rate": 1.8270635053445352e-06,
+      "loss": 0.345,
+      "step": 3173
+    },
+    {
+      "epoch": 4.566906474820144,
+      "grad_norm": 0.07275907743997405,
+      "learning_rate": 1.8150759793148332e-06,
+      "loss": 0.3376,
+      "step": 3174
+    },
+    {
+      "epoch": 4.568345323741007,
+      "grad_norm": 0.07308875094806778,
+      "learning_rate": 1.803126995429789e-06,
+      "loss": 0.3438,
+      "step": 3175
+    },
+    {
+      "epoch": 4.569784172661871,
+      "grad_norm": 0.0737266373370186,
+      "learning_rate": 1.7912165657501779e-06,
+      "loss": 0.346,
+      "step": 3176
+    },
+    {
+      "epoch": 4.571223021582734,
+      "grad_norm": 0.0741984037891141,
+      "learning_rate": 1.779344702297845e-06,
+      "loss": 0.3408,
+      "step": 3177
+    },
+    {
+      "epoch": 4.572661870503597,
+      "grad_norm": 0.07262141217459457,
+      "learning_rate": 1.767511417055725e-06,
+      "loss": 0.3423,
+      "step": 3178
+    },
+    {
+      "epoch": 4.57410071942446,
+      "grad_norm": 0.07357686840899615,
+      "learning_rate": 1.7557167219678018e-06,
+      "loss": 0.3402,
+      "step": 3179
+    },
+    {
+      "epoch": 4.575539568345324,
+      "grad_norm": 0.07504841556503612,
+      "learning_rate": 1.7439606289391032e-06,
+      "loss": 0.3416,
+      "step": 3180
+    },
+    {
+      "epoch": 4.576978417266187,
+      "grad_norm": 0.07086931022268723,
+      "learning_rate": 1.7322431498357063e-06,
+      "loss": 0.3462,
+      "step": 3181
+    },
+    {
+      "epoch": 4.57841726618705,
+      "grad_norm": 0.07113355151124676,
+      "learning_rate": 1.7205642964847103e-06,
+      "loss": 0.3435,
+      "step": 3182
+    },
+    {
+      "epoch": 4.579856115107914,
+      "grad_norm": 0.0739077808553021,
+      "learning_rate": 1.7089240806742147e-06,
+      "loss": 0.3555,
+      "step": 3183
+    },
+    {
+      "epoch": 4.581294964028777,
+      "grad_norm": 0.07502195655337876,
+      "learning_rate": 1.697322514153341e-06,
+      "loss": 0.3361,
+      "step": 3184
+    },
+    {
+      "epoch": 4.58273381294964,
+      "grad_norm": 0.07176624941980808,
+      "learning_rate": 1.6857596086321848e-06,
+      "loss": 0.3377,
+      "step": 3185
+    },
+    {
+      "epoch": 4.584172661870504,
+      "grad_norm": 0.0747136618152716,
+      "learning_rate": 1.6742353757818187e-06,
+      "loss": 0.3387,
+      "step": 3186
+    },
+    {
+      "epoch": 4.585611510791367,
+      "grad_norm": 0.07512691471483383,
+      "learning_rate": 1.6627498272342802e-06,
+      "loss": 0.3419,
+      "step": 3187
+    },
+    {
+      "epoch": 4.58705035971223,
+      "grad_norm": 0.07190887116303142,
+      "learning_rate": 1.6513029745825803e-06,
+      "loss": 0.3382,
+      "step": 3188
+    },
+    {
+      "epoch": 4.588489208633094,
+      "grad_norm": 0.0692314368608551,
+      "learning_rate": 1.6398948293806504e-06,
+      "loss": 0.3397,
+      "step": 3189
+    },
+    {
+      "epoch": 4.589928057553957,
+      "grad_norm": 0.07877068717631153,
+      "learning_rate": 1.6285254031433462e-06,
+      "loss": 0.3426,
+      "step": 3190
+    },
+    {
+      "epoch": 4.59136690647482,
+      "grad_norm": 0.07065417528656995,
+      "learning_rate": 1.6171947073464834e-06,
+      "loss": 0.3461,
+      "step": 3191
+    },
+    {
+      "epoch": 4.592805755395683,
+      "grad_norm": 0.07350813952071268,
+      "learning_rate": 1.6059027534267313e-06,
+      "loss": 0.348,
+      "step": 3192
+    },
+    {
+      "epoch": 4.594244604316547,
+      "grad_norm": 0.07717010230209637,
+      "learning_rate": 1.594649552781693e-06,
+      "loss": 0.3494,
+      "step": 3193
+    },
+    {
+      "epoch": 4.5956834532374105,
+      "grad_norm": 0.07926012384009389,
+      "learning_rate": 1.5834351167698336e-06,
+      "loss": 0.3477,
+      "step": 3194
+    },
+    {
+      "epoch": 4.597122302158273,
+      "grad_norm": 0.07283523499323487,
+      "learning_rate": 1.572259456710512e-06,
+      "loss": 0.3403,
+      "step": 3195
+    },
+    {
+      "epoch": 4.598561151079137,
+      "grad_norm": 0.07370832275629358,
+      "learning_rate": 1.5611225838839272e-06,
+      "loss": 0.3447,
+      "step": 3196
+    },
+    {
+      "epoch": 4.6,
+      "grad_norm": 0.07672318443808299,
+      "learning_rate": 1.550024509531145e-06,
+      "loss": 0.3416,
+      "step": 3197
+    },
+    {
+      "epoch": 4.601438848920863,
+      "grad_norm": 0.0784881040441822,
+      "learning_rate": 1.5389652448540537e-06,
+      "loss": 0.3463,
+      "step": 3198
+    },
+    {
+      "epoch": 4.602877697841727,
+      "grad_norm": 0.07560071813185676,
+      "learning_rate": 1.527944801015382e-06,
+      "loss": 0.3484,
+      "step": 3199
+    },
+    {
+      "epoch": 4.60431654676259,
+      "grad_norm": 0.07470938049897778,
+      "learning_rate": 1.5169631891386805e-06,
+      "loss": 0.3421,
+      "step": 3200
+    },
+    {
+      "epoch": 4.605755395683453,
+      "grad_norm": 0.07310946501465287,
+      "learning_rate": 1.506020420308274e-06,
+      "loss": 0.3449,
+      "step": 3201
+    },
+    {
+      "epoch": 4.607194244604317,
+      "grad_norm": 0.07419465325977942,
+      "learning_rate": 1.495116505569314e-06,
+      "loss": 0.34,
+      "step": 3202
+    },
+    {
+      "epoch": 4.60863309352518,
+      "grad_norm": 0.07401016996922452,
+      "learning_rate": 1.4842514559277254e-06,
+      "loss": 0.3319,
+      "step": 3203
+    },
+    {
+      "epoch": 4.6100719424460435,
+      "grad_norm": 0.07823238049951006,
+      "learning_rate": 1.4734252823501894e-06,
+      "loss": 0.3431,
+      "step": 3204
+    },
+    {
+      "epoch": 4.611510791366906,
+      "grad_norm": 0.07590369101605819,
+      "learning_rate": 1.4626379957641646e-06,
+      "loss": 0.3414,
+      "step": 3205
+    },
+    {
+      "epoch": 4.61294964028777,
+      "grad_norm": 0.07282825632276907,
+      "learning_rate": 1.451889607057848e-06,
+      "loss": 0.341,
+      "step": 3206
+    },
+    {
+      "epoch": 4.614388489208633,
+      "grad_norm": 0.0758640363462003,
+      "learning_rate": 1.4411801270801885e-06,
+      "loss": 0.3458,
+      "step": 3207
+    },
+    {
+      "epoch": 4.615827338129496,
+      "grad_norm": 0.07657141865565384,
+      "learning_rate": 1.4305095666408453e-06,
+      "loss": 0.3351,
+      "step": 3208
+    },
+    {
+      "epoch": 4.61726618705036,
+      "grad_norm": 0.07517839689819975,
+      "learning_rate": 1.4198779365102077e-06,
+      "loss": 0.3478,
+      "step": 3209
+    },
+    {
+      "epoch": 4.618705035971223,
+      "grad_norm": 0.07344136344657473,
+      "learning_rate": 1.409285247419363e-06,
+      "loss": 0.3438,
+      "step": 3210
+    },
+    {
+      "epoch": 4.620143884892086,
+      "grad_norm": 0.0732225156045941,
+      "learning_rate": 1.3987315100600961e-06,
+      "loss": 0.3462,
+      "step": 3211
+    },
+    {
+      "epoch": 4.62158273381295,
+      "grad_norm": 0.07564182080034411,
+      "learning_rate": 1.3882167350848686e-06,
+      "loss": 0.3422,
+      "step": 3212
+    },
+    {
+      "epoch": 4.623021582733813,
+      "grad_norm": 0.07356043598482831,
+      "learning_rate": 1.3777409331068258e-06,
+      "loss": 0.3463,
+      "step": 3213
+    },
+    {
+      "epoch": 4.6244604316546765,
+      "grad_norm": 0.07247769492428899,
+      "learning_rate": 1.3673041146997768e-06,
+      "loss": 0.3425,
+      "step": 3214
+    },
+    {
+      "epoch": 4.625899280575539,
+      "grad_norm": 0.0747090378206456,
+      "learning_rate": 1.35690629039817e-06,
+      "loss": 0.3555,
+      "step": 3215
+    },
+    {
+      "epoch": 4.627338129496403,
+      "grad_norm": 0.07456576634752349,
+      "learning_rate": 1.346547470697095e-06,
+      "loss": 0.3447,
+      "step": 3216
+    },
+    {
+      "epoch": 4.6287769784172665,
+      "grad_norm": 0.07166479986017334,
+      "learning_rate": 1.3362276660522943e-06,
+      "loss": 0.3423,
+      "step": 3217
+    },
+    {
+      "epoch": 4.630215827338129,
+      "grad_norm": 0.07296834092957366,
+      "learning_rate": 1.325946886880103e-06,
+      "loss": 0.3335,
+      "step": 3218
+    },
+    {
+      "epoch": 4.631654676258993,
+      "grad_norm": 0.07689047760758543,
+      "learning_rate": 1.315705143557482e-06,
+      "loss": 0.3356,
+      "step": 3219
+    },
+    {
+      "epoch": 4.633093525179856,
+      "grad_norm": 0.07306585852255368,
+      "learning_rate": 1.3055024464219846e-06,
+      "loss": 0.3409,
+      "step": 3220
+    },
+    {
+      "epoch": 4.634532374100719,
+      "grad_norm": 0.07244722903402687,
+      "learning_rate": 1.295338805771751e-06,
+      "loss": 0.3395,
+      "step": 3221
+    },
+    {
+      "epoch": 4.635971223021583,
+      "grad_norm": 0.07044884383448612,
+      "learning_rate": 1.285214231865508e-06,
+      "loss": 0.3345,
+      "step": 3222
+    },
+    {
+      "epoch": 4.637410071942446,
+      "grad_norm": 0.07340245842043562,
+      "learning_rate": 1.2751287349225484e-06,
+      "loss": 0.3507,
+      "step": 3223
+    },
+    {
+      "epoch": 4.6388489208633095,
+      "grad_norm": 0.078882411830014,
+      "learning_rate": 1.2650823251227062e-06,
+      "loss": 0.3507,
+      "step": 3224
+    },
+    {
+      "epoch": 4.640287769784173,
+      "grad_norm": 0.07770979722776154,
+      "learning_rate": 1.255075012606386e-06,
+      "loss": 0.3442,
+      "step": 3225
+    },
+    {
+      "epoch": 4.641726618705036,
+      "grad_norm": 0.07589427534198123,
+      "learning_rate": 1.2451068074745254e-06,
+      "loss": 0.3472,
+      "step": 3226
+    },
+    {
+      "epoch": 4.6431654676258995,
+      "grad_norm": 0.07696166151922781,
+      "learning_rate": 1.2351777197885606e-06,
+      "loss": 0.349,
+      "step": 3227
+    },
+    {
+      "epoch": 4.644604316546762,
+      "grad_norm": 0.07219860964692072,
+      "learning_rate": 1.2252877595704838e-06,
+      "loss": 0.3421,
+      "step": 3228
+    },
+    {
+      "epoch": 4.646043165467626,
+      "grad_norm": 0.07242363649293503,
+      "learning_rate": 1.2154369368027763e-06,
+      "loss": 0.3407,
+      "step": 3229
+    },
+    {
+      "epoch": 4.647482014388489,
+      "grad_norm": 0.17023544957477652,
+      "learning_rate": 1.2056252614284047e-06,
+      "loss": 0.3479,
+      "step": 3230
+    },
+    {
+      "epoch": 4.648920863309352,
+      "grad_norm": 0.07162113454038929,
+      "learning_rate": 1.1958527433508381e-06,
+      "loss": 0.3381,
+      "step": 3231
+    },
+    {
+      "epoch": 4.650359712230216,
+      "grad_norm": 0.0728067529289856,
+      "learning_rate": 1.1861193924340176e-06,
+      "loss": 0.3432,
+      "step": 3232
+    },
+    {
+      "epoch": 4.651798561151079,
+      "grad_norm": 0.07059212351903399,
+      "learning_rate": 1.176425218502346e-06,
+      "loss": 0.3426,
+      "step": 3233
+    },
+    {
+      "epoch": 4.6532374100719425,
+      "grad_norm": 0.07235953337046205,
+      "learning_rate": 1.1667702313406903e-06,
+      "loss": 0.3432,
+      "step": 3234
+    },
+    {
+      "epoch": 4.654676258992806,
+      "grad_norm": 0.0696798547491888,
+      "learning_rate": 1.1571544406943614e-06,
+      "loss": 0.3386,
+      "step": 3235
+    },
+    {
+      "epoch": 4.656115107913669,
+      "grad_norm": 0.07107800574432338,
+      "learning_rate": 1.147577856269102e-06,
+      "loss": 0.3393,
+      "step": 3236
+    },
+    {
+      "epoch": 4.6575539568345325,
+      "grad_norm": 0.07468832447428979,
+      "learning_rate": 1.1380404877310957e-06,
+      "loss": 0.3376,
+      "step": 3237
+    },
+    {
+      "epoch": 4.658992805755395,
+      "grad_norm": 0.07634705223343236,
+      "learning_rate": 1.1285423447069133e-06,
+      "loss": 0.343,
+      "step": 3238
+    },
+    {
+      "epoch": 4.660431654676259,
+      "grad_norm": 0.07352989842580739,
+      "learning_rate": 1.1190834367835701e-06,
+      "loss": 0.3483,
+      "step": 3239
+    },
+    {
+      "epoch": 4.661870503597123,
+      "grad_norm": 0.07129723426655685,
+      "learning_rate": 1.1096637735084602e-06,
+      "loss": 0.345,
+      "step": 3240
+    },
+    {
+      "epoch": 4.663309352517985,
+      "grad_norm": 0.07292570415856726,
+      "learning_rate": 1.1002833643893606e-06,
+      "loss": 0.3434,
+      "step": 3241
+    },
+    {
+      "epoch": 4.664748201438849,
+      "grad_norm": 0.07098015505406921,
+      "learning_rate": 1.0909422188944308e-06,
+      "loss": 0.3455,
+      "step": 3242
+    },
+    {
+      "epoch": 4.666187050359712,
+      "grad_norm": 0.07307054276226224,
+      "learning_rate": 1.0816403464522262e-06,
+      "loss": 0.3491,
+      "step": 3243
+    },
+    {
+      "epoch": 4.6676258992805755,
+      "grad_norm": 0.07755083661084325,
+      "learning_rate": 1.0723777564516148e-06,
+      "loss": 0.3445,
+      "step": 3244
+    },
+    {
+      "epoch": 4.669064748201439,
+      "grad_norm": 0.0718690723346127,
+      "learning_rate": 1.0631544582418463e-06,
+      "loss": 0.3432,
+      "step": 3245
+    },
+    {
+      "epoch": 4.670503597122302,
+      "grad_norm": 0.07259792487389534,
+      "learning_rate": 1.0539704611325008e-06,
+      "loss": 0.3474,
+      "step": 3246
+    },
+    {
+      "epoch": 4.6719424460431656,
+      "grad_norm": 0.070609052250591,
+      "learning_rate": 1.0448257743934964e-06,
+      "loss": 0.3369,
+      "step": 3247
+    },
+    {
+      "epoch": 4.673381294964029,
+      "grad_norm": 0.07081140150355565,
+      "learning_rate": 1.0357204072550676e-06,
+      "loss": 0.3382,
+      "step": 3248
+    },
+    {
+      "epoch": 4.674820143884892,
+      "grad_norm": 0.07256789962790246,
+      "learning_rate": 1.0266543689077602e-06,
+      "loss": 0.3412,
+      "step": 3249
+    },
+    {
+      "epoch": 4.676258992805756,
+      "grad_norm": 0.07373465051135351,
+      "learning_rate": 1.0176276685024233e-06,
+      "loss": 0.3509,
+      "step": 3250
+    },
+    {
+      "epoch": 4.677697841726618,
+      "grad_norm": 0.07107883300929509,
+      "learning_rate": 1.0086403151502088e-06,
+      "loss": 0.351,
+      "step": 3251
+    },
+    {
+      "epoch": 4.679136690647482,
+      "grad_norm": 0.07329045248109889,
+      "learning_rate": 9.996923179225448e-07,
+      "loss": 0.3352,
+      "step": 3252
+    },
+    {
+      "epoch": 4.680575539568346,
+      "grad_norm": 0.0722249679325728,
+      "learning_rate": 9.90783685851131e-07,
+      "loss": 0.3493,
+      "step": 3253
+    },
+    {
+      "epoch": 4.6820143884892085,
+      "grad_norm": 0.07246260082915003,
+      "learning_rate": 9.81914427927948e-07,
+      "loss": 0.3423,
+      "step": 3254
+    },
+    {
+      "epoch": 4.683453237410072,
+      "grad_norm": 0.0713126062283184,
+      "learning_rate": 9.730845531052214e-07,
+      "loss": 0.3433,
+      "step": 3255
+    },
+    {
+      "epoch": 4.684892086330935,
+      "grad_norm": 0.07355877347085475,
+      "learning_rate": 9.642940702954306e-07,
+      "loss": 0.3408,
+      "step": 3256
+    },
+    {
+      "epoch": 4.686330935251799,
+      "grad_norm": 0.07052278467158665,
+      "learning_rate": 9.555429883712963e-07,
+      "loss": 0.3356,
+      "step": 3257
+    },
+    {
+      "epoch": 4.687769784172662,
+      "grad_norm": 0.0735207329605086,
+      "learning_rate": 9.468313161657617e-07,
+      "loss": 0.3402,
+      "step": 3258
+    },
+    {
+      "epoch": 4.689208633093525,
+      "grad_norm": 0.06859602773813163,
+      "learning_rate": 9.381590624719972e-07,
+      "loss": 0.3428,
+      "step": 3259
+    },
+    {
+      "epoch": 4.690647482014389,
+      "grad_norm": 0.0692500493142813,
+      "learning_rate": 9.295262360433921e-07,
+      "loss": 0.3373,
+      "step": 3260
+    },
+    {
+      "epoch": 4.692086330935252,
+      "grad_norm": 0.07026107245791961,
+      "learning_rate": 9.209328455935274e-07,
+      "loss": 0.3446,
+      "step": 3261
+    },
+    {
+      "epoch": 4.693525179856115,
+      "grad_norm": 0.07121245958557837,
+      "learning_rate": 9.123788997961847e-07,
+      "loss": 0.3498,
+      "step": 3262
+    },
+    {
+      "epoch": 4.694964028776979,
+      "grad_norm": 0.07428187585450076,
+      "learning_rate": 9.038644072853331e-07,
+      "loss": 0.3481,
+      "step": 3263
+    },
+    {
+      "epoch": 4.6964028776978415,
+      "grad_norm": 0.07209399796559547,
+      "learning_rate": 8.953893766551203e-07,
+      "loss": 0.3438,
+      "step": 3264
+    },
+    {
+      "epoch": 4.697841726618705,
+      "grad_norm": 0.07119356925873069,
+      "learning_rate": 8.86953816459859e-07,
+      "loss": 0.3447,
+      "step": 3265
+    },
+    {
+      "epoch": 4.699280575539568,
+      "grad_norm": 0.07518530283351295,
+      "learning_rate": 8.785577352140317e-07,
+      "loss": 0.3493,
+      "step": 3266
+    },
+    {
+      "epoch": 4.700719424460432,
+      "grad_norm": 0.0746073054119646,
+      "learning_rate": 8.702011413922506e-07,
+      "loss": 0.3531,
+      "step": 3267
+    },
+    {
+      "epoch": 4.702158273381295,
+      "grad_norm": 0.07090028442671885,
+      "learning_rate": 8.61884043429293e-07,
+      "loss": 0.3446,
+      "step": 3268
+    },
+    {
+      "epoch": 4.703597122302158,
+      "grad_norm": 0.0712249508346909,
+      "learning_rate": 8.536064497200702e-07,
+      "loss": 0.3454,
+      "step": 3269
+    },
+    {
+      "epoch": 4.705035971223022,
+      "grad_norm": 0.07349652446599979,
+      "learning_rate": 8.453683686196012e-07,
+      "loss": 0.3562,
+      "step": 3270
+    },
+    {
+      "epoch": 4.706474820143885,
+      "grad_norm": 0.07385176044106828,
+      "learning_rate": 8.371698084430346e-07,
+      "loss": 0.3471,
+      "step": 3271
+    },
+    {
+      "epoch": 4.707913669064748,
+      "grad_norm": 0.07317079825374315,
+      "learning_rate": 8.290107774656441e-07,
+      "loss": 0.3449,
+      "step": 3272
+    },
+    {
+      "epoch": 4.709352517985612,
+      "grad_norm": 0.07568657649877363,
+      "learning_rate": 8.208912839227712e-07,
+      "loss": 0.3467,
+      "step": 3273
+    },
+    {
+      "epoch": 4.7107913669064745,
+      "grad_norm": 0.06993763923796827,
+      "learning_rate": 8.128113360098777e-07,
+      "loss": 0.3431,
+      "step": 3274
+    },
+    {
+      "epoch": 4.712230215827338,
+      "grad_norm": 0.06839755659991058,
+      "learning_rate": 8.047709418824934e-07,
+      "loss": 0.3421,
+      "step": 3275
+    },
+    {
+      "epoch": 4.713669064748202,
+      "grad_norm": 0.07195142770477857,
+      "learning_rate": 7.96770109656233e-07,
+      "loss": 0.3354,
+      "step": 3276
+    },
+    {
+      "epoch": 4.715107913669065,
+      "grad_norm": 0.0725067551374577,
+      "learning_rate": 7.88808847406779e-07,
+      "loss": 0.3469,
+      "step": 3277
+    },
+    {
+      "epoch": 4.716546762589928,
+      "grad_norm": 0.07219832598972031,
+      "learning_rate": 7.808871631698723e-07,
+      "loss": 0.3534,
+      "step": 3278
+    },
+    {
+      "epoch": 4.717985611510791,
+      "grad_norm": 0.0721936519889365,
+      "learning_rate": 7.730050649412946e-07,
+      "loss": 0.3439,
+      "step": 3279
+    },
+    {
+      "epoch": 4.719424460431655,
+      "grad_norm": 0.07203963159496164,
+      "learning_rate": 7.651625606768908e-07,
+      "loss": 0.3487,
+      "step": 3280
+    },
+    {
+      "epoch": 4.720863309352518,
+      "grad_norm": 0.06976106999488406,
+      "learning_rate": 7.573596582925291e-07,
+      "loss": 0.3392,
+      "step": 3281
+    },
+    {
+      "epoch": 4.722302158273381,
+      "grad_norm": 0.0702046647598994,
+      "learning_rate": 7.495963656641048e-07,
+      "loss": 0.3503,
+      "step": 3282
+    },
+    {
+      "epoch": 4.723741007194245,
+      "grad_norm": 0.06946908302821556,
+      "learning_rate": 7.418726906275497e-07,
+      "loss": 0.3442,
+      "step": 3283
+    },
+    {
+      "epoch": 4.725179856115108,
+      "grad_norm": 0.06859861930076236,
+      "learning_rate": 7.341886409787746e-07,
+      "loss": 0.3383,
+      "step": 3284
+    },
+    {
+      "epoch": 4.726618705035971,
+      "grad_norm": 0.07028895278734329,
+      "learning_rate": 7.265442244737264e-07,
+      "loss": 0.3466,
+      "step": 3285
+    },
+    {
+      "epoch": 4.728057553956835,
+      "grad_norm": 0.07010202037342096,
+      "learning_rate": 7.189394488283307e-07,
+      "loss": 0.34,
+      "step": 3286
+    },
+    {
+      "epoch": 4.729496402877698,
+      "grad_norm": 0.07159689438722366,
+      "learning_rate": 7.113743217185099e-07,
+      "loss": 0.3539,
+      "step": 3287
+    },
+    {
+      "epoch": 4.730935251798561,
+      "grad_norm": 0.07020760800692664,
+      "learning_rate": 7.0384885078016e-07,
+      "loss": 0.3365,
+      "step": 3288
+    },
+    {
+      "epoch": 4.732374100719424,
+      "grad_norm": 0.0703843916403367,
+      "learning_rate": 6.963630436091518e-07,
+      "loss": 0.3467,
+      "step": 3289
+    },
+    {
+      "epoch": 4.733812949640288,
+      "grad_norm": 0.07092803734791567,
+      "learning_rate": 6.889169077613212e-07,
+      "loss": 0.3364,
+      "step": 3290
+    },
+    {
+      "epoch": 4.735251798561151,
+      "grad_norm": 0.07035847540003927,
+      "learning_rate": 6.815104507524695e-07,
+      "loss": 0.3396,
+      "step": 3291
+    },
+    {
+      "epoch": 4.736690647482014,
+      "grad_norm": 0.06895173835205612,
+      "learning_rate": 6.741436800583367e-07,
+      "loss": 0.3433,
+      "step": 3292
+    },
+    {
+      "epoch": 4.738129496402878,
+      "grad_norm": 0.07197375726645833,
+      "learning_rate": 6.668166031146062e-07,
+      "loss": 0.3472,
+      "step": 3293
+    },
+    {
+      "epoch": 4.739568345323741,
+      "grad_norm": 0.06847000902765593,
+      "learning_rate": 6.595292273169041e-07,
+      "loss": 0.3392,
+      "step": 3294
+    },
+    {
+      "epoch": 4.741007194244604,
+      "grad_norm": 0.0704583019496255,
+      "learning_rate": 6.522815600207866e-07,
+      "loss": 0.3406,
+      "step": 3295
+    },
+    {
+      "epoch": 4.742446043165468,
+      "grad_norm": 0.07086617229826032,
+      "learning_rate": 6.450736085417086e-07,
+      "loss": 0.3398,
+      "step": 3296
+    },
+    {
+      "epoch": 4.743884892086331,
+      "grad_norm": 0.07247846141002307,
+      "learning_rate": 6.379053801550594e-07,
+      "loss": 0.3474,
+      "step": 3297
+    },
+    {
+      "epoch": 4.745323741007194,
+      "grad_norm": 0.07146186302062525,
+      "learning_rate": 6.307768820961269e-07,
+      "loss": 0.3457,
+      "step": 3298
+    },
+    {
+      "epoch": 4.746762589928058,
+      "grad_norm": 0.07267016536214223,
+      "learning_rate": 6.236881215600976e-07,
+      "loss": 0.344,
+      "step": 3299
+    },
+    {
+      "epoch": 4.748201438848921,
+      "grad_norm": 0.07158713318428622,
+      "learning_rate": 6.166391057020438e-07,
+      "loss": 0.3386,
+      "step": 3300
+    },
+    {
+      "epoch": 4.749640287769784,
+      "grad_norm": 0.07202452509042864,
+      "learning_rate": 6.096298416369273e-07,
+      "loss": 0.343,
+      "step": 3301
+    },
+    {
+      "epoch": 4.751079136690647,
+      "grad_norm": 0.07242217820098035,
+      "learning_rate": 6.026603364395867e-07,
+      "loss": 0.347,
+      "step": 3302
+    },
+    {
+      "epoch": 4.752517985611511,
+      "grad_norm": 0.07090701666068121,
+      "learning_rate": 5.957305971447192e-07,
+      "loss": 0.3409,
+      "step": 3303
+    },
+    {
+      "epoch": 4.753956834532374,
+      "grad_norm": 0.07061662022993838,
+      "learning_rate": 5.888406307468986e-07,
+      "loss": 0.3496,
+      "step": 3304
+    },
+    {
+      "epoch": 4.755395683453237,
+      "grad_norm": 0.07367594232626853,
+      "learning_rate": 5.819904442005442e-07,
+      "loss": 0.3467,
+      "step": 3305
+    },
+    {
+      "epoch": 4.756834532374101,
+      "grad_norm": 0.07097064998410584,
+      "learning_rate": 5.751800444199295e-07,
+      "loss": 0.343,
+      "step": 3306
+    },
+    {
+      "epoch": 4.7582733812949645,
+      "grad_norm": 0.07204400418422385,
+      "learning_rate": 5.684094382791605e-07,
+      "loss": 0.3424,
+      "step": 3307
+    },
+    {
+      "epoch": 4.759712230215827,
+      "grad_norm": 0.07319040042214421,
+      "learning_rate": 5.616786326121837e-07,
+      "loss": 0.3463,
+      "step": 3308
+    },
+    {
+      "epoch": 4.761151079136691,
+      "grad_norm": 0.06954107897437739,
+      "learning_rate": 5.54987634212778e-07,
+      "loss": 0.3447,
+      "step": 3309
+    },
+    {
+      "epoch": 4.762589928057554,
+      "grad_norm": 0.0712861956177736,
+      "learning_rate": 5.483364498345279e-07,
+      "loss": 0.3414,
+      "step": 3310
+    },
+    {
+      "epoch": 4.764028776978417,
+      "grad_norm": 0.06919069782996472,
+      "learning_rate": 5.417250861908452e-07,
+      "loss": 0.342,
+      "step": 3311
+    },
+    {
+      "epoch": 4.76546762589928,
+      "grad_norm": 0.06769359574302063,
+      "learning_rate": 5.351535499549387e-07,
+      "loss": 0.3364,
+      "step": 3312
+    },
+    {
+      "epoch": 4.766906474820144,
+      "grad_norm": 0.06966426302621624,
+      "learning_rate": 5.286218477598226e-07,
+      "loss": 0.3407,
+      "step": 3313
+    },
+    {
+      "epoch": 4.768345323741007,
+      "grad_norm": 0.06988763571315693,
+      "learning_rate": 5.221299861983075e-07,
+      "loss": 0.3346,
+      "step": 3314
+    },
+    {
+      "epoch": 4.76978417266187,
+      "grad_norm": 0.07286917998653716,
+      "learning_rate": 5.156779718229787e-07,
+      "loss": 0.3508,
+      "step": 3315
+    },
+    {
+      "epoch": 4.771223021582734,
+      "grad_norm": 0.07050658311773654,
+      "learning_rate": 5.092658111462179e-07,
+      "loss": 0.341,
+      "step": 3316
+    },
+    {
+      "epoch": 4.7726618705035975,
+      "grad_norm": 0.07028255172120178,
+      "learning_rate": 5.028935106401678e-07,
+      "loss": 0.3403,
+      "step": 3317
+    },
+    {
+      "epoch": 4.77410071942446,
+      "grad_norm": 0.07289048655011893,
+      "learning_rate": 4.965610767367413e-07,
+      "loss": 0.3457,
+      "step": 3318
+    },
+    {
+      "epoch": 4.775539568345324,
+      "grad_norm": 0.06787022097041073,
+      "learning_rate": 4.902685158276078e-07,
+      "loss": 0.3374,
+      "step": 3319
+    },
+    {
+      "epoch": 4.7769784172661875,
+      "grad_norm": 0.07081096261177364,
+      "learning_rate": 4.840158342642021e-07,
+      "loss": 0.3392,
+      "step": 3320
+    },
+    {
+      "epoch": 4.77841726618705,
+      "grad_norm": 0.06997345301677844,
+      "learning_rate": 4.778030383577026e-07,
+      "loss": 0.3399,
+      "step": 3321
+    },
+    {
+      "epoch": 4.779856115107914,
+      "grad_norm": 0.06859614733030588,
+      "learning_rate": 4.716301343790175e-07,
+      "loss": 0.3428,
+      "step": 3322
+    },
+    {
+      "epoch": 4.781294964028777,
+      "grad_norm": 0.07171319473830083,
+      "learning_rate": 4.6549712855879837e-07,
+      "loss": 0.3371,
+      "step": 3323
+    },
+    {
+      "epoch": 4.78273381294964,
+      "grad_norm": 0.07370327698272955,
+      "learning_rate": 4.5940402708744005e-07,
+      "loss": 0.3527,
+      "step": 3324
+    },
+    {
+      "epoch": 4.784172661870503,
+      "grad_norm": 0.07050434053534654,
+      "learning_rate": 4.5335083611502293e-07,
+      "loss": 0.3415,
+      "step": 3325
+    },
+    {
+      "epoch": 4.785611510791367,
+      "grad_norm": 0.11270684077356648,
+      "learning_rate": 4.473375617513842e-07,
+      "loss": 0.3456,
+      "step": 3326
+    },
+    {
+      "epoch": 4.7870503597122305,
+      "grad_norm": 0.0714616413391725,
+      "learning_rate": 4.41364210066042e-07,
+      "loss": 0.3505,
+      "step": 3327
+    },
+    {
+      "epoch": 4.788489208633093,
+      "grad_norm": 0.07188375038968106,
+      "learning_rate": 4.3543078708823126e-07,
+      "loss": 0.3507,
+      "step": 3328
+    },
+    {
+      "epoch": 4.789928057553957,
+      "grad_norm": 0.06887456261390511,
+      "learning_rate": 4.295372988068813e-07,
+      "loss": 0.3351,
+      "step": 3329
+    },
+    {
+      "epoch": 4.7913669064748206,
+      "grad_norm": 0.07098726200078284,
+      "learning_rate": 4.2368375117062043e-07,
+      "loss": 0.3491,
+      "step": 3330
+    },
+    {
+      "epoch": 4.792805755395683,
+      "grad_norm": 0.07395405969237014,
+      "learning_rate": 4.178701500877491e-07,
+      "loss": 0.3505,
+      "step": 3331
+    },
+    {
+      "epoch": 4.794244604316547,
+      "grad_norm": 0.0705628304126942,
+      "learning_rate": 4.120965014262579e-07,
+      "loss": 0.3376,
+      "step": 3332
+    },
+    {
+      "epoch": 4.79568345323741,
+      "grad_norm": 0.0687397453264519,
+      "learning_rate": 4.063628110138096e-07,
+      "loss": 0.34,
+      "step": 3333
+    },
+    {
+      "epoch": 4.797122302158273,
+      "grad_norm": 0.07100196280713694,
+      "learning_rate": 4.0066908463772593e-07,
+      "loss": 0.3426,
+      "step": 3334
+    },
+    {
+      "epoch": 4.798561151079137,
+      "grad_norm": 0.07064818367164927,
+      "learning_rate": 3.9501532804500974e-07,
+      "loss": 0.3355,
+      "step": 3335
+    },
+    {
+      "epoch": 4.8,
+      "grad_norm": 0.07034955191773087,
+      "learning_rate": 3.894015469423007e-07,
+      "loss": 0.3495,
+      "step": 3336
+    },
+    {
+      "epoch": 4.8014388489208635,
+      "grad_norm": 0.07098980654888197,
+      "learning_rate": 3.838277469958973e-07,
+      "loss": 0.3418,
+      "step": 3337
+    },
+    {
+      "epoch": 4.802877697841726,
+      "grad_norm": 0.0693087014278541,
+      "learning_rate": 3.7829393383174375e-07,
+      "loss": 0.3465,
+      "step": 3338
+    },
+    {
+      "epoch": 4.80431654676259,
+      "grad_norm": 0.06885712962710298,
+      "learning_rate": 3.7280011303542084e-07,
+      "loss": 0.3389,
+      "step": 3339
+    },
+    {
+      "epoch": 4.805755395683454,
+      "grad_norm": 0.07252644943272837,
+      "learning_rate": 3.673462901521463e-07,
+      "loss": 0.3401,
+      "step": 3340
+    },
+    {
+      "epoch": 4.807194244604316,
+      "grad_norm": 0.07000126262931973,
+      "learning_rate": 3.619324706867655e-07,
+      "loss": 0.3403,
+      "step": 3341
+    },
+    {
+      "epoch": 4.80863309352518,
+      "grad_norm": 0.07092740269571697,
+      "learning_rate": 3.5655866010373853e-07,
+      "loss": 0.3434,
+      "step": 3342
+    },
+    {
+      "epoch": 4.810071942446044,
+      "grad_norm": 0.07381551346368283,
+      "learning_rate": 3.5122486382715314e-07,
+      "loss": 0.3504,
+      "step": 3343
+    },
+    {
+      "epoch": 4.811510791366906,
+      "grad_norm": 0.0667687378931654,
+      "learning_rate": 3.459310872407029e-07,
+      "loss": 0.3369,
+      "step": 3344
+    },
+    {
+      "epoch": 4.81294964028777,
+      "grad_norm": 0.06986551611735359,
+      "learning_rate": 3.4067733568768246e-07,
+      "loss": 0.3455,
+      "step": 3345
+    },
+    {
+      "epoch": 4.814388489208633,
+      "grad_norm": 0.0755061097725317,
+      "learning_rate": 3.3546361447099664e-07,
+      "loss": 0.3499,
+      "step": 3346
+    },
+    {
+      "epoch": 4.8158273381294965,
+      "grad_norm": 0.07154098628724741,
+      "learning_rate": 3.3028992885314247e-07,
+      "loss": 0.3452,
+      "step": 3347
+    },
+    {
+      "epoch": 4.817266187050359,
+      "grad_norm": 0.07282192416556188,
+      "learning_rate": 3.2515628405620503e-07,
+      "loss": 0.3455,
+      "step": 3348
+    },
+    {
+      "epoch": 4.818705035971223,
+      "grad_norm": 0.06891706028658195,
+      "learning_rate": 3.2006268526184824e-07,
+      "loss": 0.3455,
+      "step": 3349
+    },
+    {
+      "epoch": 4.820143884892087,
+      "grad_norm": 0.07276800257251627,
+      "learning_rate": 3.150091376113329e-07,
+      "loss": 0.3364,
+      "step": 3350
+    },
+    {
+      "epoch": 4.821582733812949,
+      "grad_norm": 0.07407440886093652,
+      "learning_rate": 3.0999564620547207e-07,
+      "loss": 0.3434,
+      "step": 3351
+    },
+    {
+      "epoch": 4.823021582733813,
+      "grad_norm": 0.06941563104967682,
+      "learning_rate": 3.0502221610465786e-07,
+      "loss": 0.3356,
+      "step": 3352
+    },
+    {
+      "epoch": 4.824460431654677,
+      "grad_norm": 0.06863053348683697,
+      "learning_rate": 3.0008885232886144e-07,
+      "loss": 0.3415,
+      "step": 3353
+    },
+    {
+      "epoch": 4.825899280575539,
+      "grad_norm": 0.07364843527397481,
+      "learning_rate": 2.95195559857584e-07,
+      "loss": 0.3369,
+      "step": 3354
+    },
+    {
+      "epoch": 4.827338129496403,
+      "grad_norm": 0.06804047164493178,
+      "learning_rate": 2.9034234362989687e-07,
+      "loss": 0.3425,
+      "step": 3355
+    },
+    {
+      "epoch": 4.828776978417266,
+      "grad_norm": 0.07216054806563041,
+      "learning_rate": 2.855292085444239e-07,
+      "loss": 0.3507,
+      "step": 3356
+    },
+    {
+      "epoch": 4.8302158273381295,
+      "grad_norm": 0.06982074106557941,
+      "learning_rate": 2.8075615945932333e-07,
+      "loss": 0.3442,
+      "step": 3357
+    },
+    {
+      "epoch": 4.831654676258993,
+      "grad_norm": 0.06827818079322905,
+      "learning_rate": 2.7602320119229254e-07,
+      "loss": 0.3385,
+      "step": 3358
+    },
+    {
+      "epoch": 4.833093525179856,
+      "grad_norm": 0.06923145674977711,
+      "learning_rate": 2.7133033852057675e-07,
+      "loss": 0.3498,
+      "step": 3359
+    },
+    {
+      "epoch": 4.83453237410072,
+      "grad_norm": 0.06746504159607383,
+      "learning_rate": 2.666775761809337e-07,
+      "loss": 0.3371,
+      "step": 3360
+    },
+    {
+      "epoch": 4.835971223021582,
+      "grad_norm": 0.07075266970183086,
+      "learning_rate": 2.620649188696511e-07,
+      "loss": 0.3439,
+      "step": 3361
+    },
+    {
+      "epoch": 4.837410071942446,
+      "grad_norm": 0.06885614287105779,
+      "learning_rate": 2.574923712425426e-07,
+      "loss": 0.3447,
+      "step": 3362
+    },
+    {
+      "epoch": 4.83884892086331,
+      "grad_norm": 0.06863579601971304,
+      "learning_rate": 2.52959937914925e-07,
+      "loss": 0.3417,
+      "step": 3363
+    },
+    {
+      "epoch": 4.840287769784172,
+      "grad_norm": 0.07207973539335823,
+      "learning_rate": 2.484676234616412e-07,
+      "loss": 0.3408,
+      "step": 3364
+    },
+    {
+      "epoch": 4.841726618705036,
+      "grad_norm": 0.06925055589688599,
+      "learning_rate": 2.440154324170285e-07,
+      "loss": 0.3525,
+      "step": 3365
+    },
+    {
+      "epoch": 4.8431654676259,
+      "grad_norm": 0.06858531426032595,
+      "learning_rate": 2.3960336927492333e-07,
+      "loss": 0.3462,
+      "step": 3366
+    },
+    {
+      "epoch": 4.8446043165467625,
+      "grad_norm": 0.07021885567068153,
+      "learning_rate": 2.3523143848867003e-07,
+      "loss": 0.3438,
+      "step": 3367
+    },
+    {
+      "epoch": 4.846043165467626,
+      "grad_norm": 0.06985375092417792,
+      "learning_rate": 2.3089964447109425e-07,
+      "loss": 0.342,
+      "step": 3368
+    },
+    {
+      "epoch": 4.847482014388489,
+      "grad_norm": 0.070370726460504,
+      "learning_rate": 2.2660799159451629e-07,
+      "loss": 0.3477,
+      "step": 3369
+    },
+    {
+      "epoch": 4.848920863309353,
+      "grad_norm": 0.06937979912952354,
+      "learning_rate": 2.2235648419073773e-07,
+      "loss": 0.3442,
+      "step": 3370
+    },
+    {
+      "epoch": 4.850359712230215,
+      "grad_norm": 0.06921640426519425,
+      "learning_rate": 2.1814512655103703e-07,
+      "loss": 0.3427,
+      "step": 3371
+    },
+    {
+      "epoch": 4.851798561151079,
+      "grad_norm": 0.07121538254960598,
+      "learning_rate": 2.1397392292617392e-07,
+      "loss": 0.3408,
+      "step": 3372
+    },
+    {
+      "epoch": 4.853237410071943,
+      "grad_norm": 0.07054649424777172,
+      "learning_rate": 2.0984287752636722e-07,
+      "loss": 0.3567,
+      "step": 3373
+    },
+    {
+      "epoch": 4.854676258992805,
+      "grad_norm": 0.06859174423295826,
+      "learning_rate": 2.0575199452131268e-07,
+      "loss": 0.3452,
+      "step": 3374
+    },
+    {
+      "epoch": 4.856115107913669,
+      "grad_norm": 0.06924411163154238,
+      "learning_rate": 2.017012780401606e-07,
+      "loss": 0.3433,
+      "step": 3375
+    },
+    {
+      "epoch": 4.857553956834533,
+      "grad_norm": 0.07349884644632264,
+      "learning_rate": 1.9769073217152933e-07,
+      "loss": 0.345,
+      "step": 3376
+    },
+    {
+      "epoch": 4.8589928057553955,
+      "grad_norm": 0.07081026265598288,
+      "learning_rate": 1.9372036096347414e-07,
+      "loss": 0.3453,
+      "step": 3377
+    },
+    {
+      "epoch": 4.860431654676259,
+      "grad_norm": 0.06952943020219723,
+      "learning_rate": 1.8979016842350928e-07,
+      "loss": 0.3492,
+      "step": 3378
+    },
+    {
+      "epoch": 4.861870503597122,
+      "grad_norm": 0.07076935581770977,
+      "learning_rate": 1.8590015851860376e-07,
+      "loss": 0.3526,
+      "step": 3379
+    },
+    {
+      "epoch": 4.863309352517986,
+      "grad_norm": 0.06881864526271664,
+      "learning_rate": 1.8205033517515015e-07,
+      "loss": 0.3423,
+      "step": 3380
+    },
+    {
+      "epoch": 4.864748201438849,
+      "grad_norm": 0.06788157979408772,
+      "learning_rate": 1.7824070227899115e-07,
+      "loss": 0.3454,
+      "step": 3381
+    },
+    {
+      "epoch": 4.866187050359712,
+      "grad_norm": 0.07196156739195902,
+      "learning_rate": 1.7447126367539313e-07,
+      "loss": 0.344,
+      "step": 3382
+    },
+    {
+      "epoch": 4.867625899280576,
+      "grad_norm": 0.06738594219189462,
+      "learning_rate": 1.7074202316906374e-07,
+      "loss": 0.331,
+      "step": 3383
+    },
+    {
+      "epoch": 4.869064748201438,
+      "grad_norm": 0.06930807246944674,
+      "learning_rate": 1.6705298452412978e-07,
+      "loss": 0.3403,
+      "step": 3384
+    },
+    {
+      "epoch": 4.870503597122302,
+      "grad_norm": 0.06932608247336144,
+      "learning_rate": 1.6340415146414157e-07,
+      "loss": 0.3424,
+      "step": 3385
+    },
+    {
+      "epoch": 4.871942446043166,
+      "grad_norm": 0.06815944754953394,
+      "learning_rate": 1.597955276720642e-07,
+      "loss": 0.3421,
+      "step": 3386
+    },
+    {
+      "epoch": 4.8733812949640285,
+      "grad_norm": 0.07345784152517565,
+      "learning_rate": 1.562271167902818e-07,
+      "loss": 0.3379,
+      "step": 3387
+    },
+    {
+      "epoch": 4.874820143884892,
+      "grad_norm": 0.06893531752585413,
+      "learning_rate": 1.526989224205888e-07,
+      "loss": 0.3348,
+      "step": 3388
+    },
+    {
+      "epoch": 4.876258992805756,
+      "grad_norm": 0.06931265398781795,
+      "learning_rate": 1.4921094812418103e-07,
+      "loss": 0.3437,
+      "step": 3389
+    },
+    {
+      "epoch": 4.877697841726619,
+      "grad_norm": 0.06913324074740719,
+      "learning_rate": 1.457631974216689e-07,
+      "loss": 0.3357,
+      "step": 3390
+    },
+    {
+      "epoch": 4.879136690647482,
+      "grad_norm": 0.06999009405527448,
+      "learning_rate": 1.4235567379305536e-07,
+      "loss": 0.342,
+      "step": 3391
+    },
+    {
+      "epoch": 4.880575539568345,
+      "grad_norm": 0.06704443213036577,
+      "learning_rate": 1.389883806777359e-07,
+      "loss": 0.3381,
+      "step": 3392
+    },
+    {
+      "epoch": 4.882014388489209,
+      "grad_norm": 0.06954696835327542,
+      "learning_rate": 1.356613214745117e-07,
+      "loss": 0.3555,
+      "step": 3393
+    },
+    {
+      "epoch": 4.883453237410072,
+      "grad_norm": 0.07257864704658673,
+      "learning_rate": 1.3237449954156767e-07,
+      "loss": 0.3497,
+      "step": 3394
+    },
+    {
+      "epoch": 4.884892086330935,
+      "grad_norm": 0.06835497987647865,
+      "learning_rate": 1.2912791819646774e-07,
+      "loss": 0.3472,
+      "step": 3395
+    },
+    {
+      "epoch": 4.886330935251799,
+      "grad_norm": 0.07115444901613881,
+      "learning_rate": 1.2592158071616844e-07,
+      "loss": 0.3474,
+      "step": 3396
+    },
+    {
+      "epoch": 4.8877697841726615,
+      "grad_norm": 0.07128270297209283,
+      "learning_rate": 1.2275549033700097e-07,
+      "loss": 0.3553,
+      "step": 3397
+    },
+    {
+      "epoch": 4.889208633093525,
+      "grad_norm": 0.06944010179810285,
+      "learning_rate": 1.1962965025467564e-07,
+      "loss": 0.3525,
+      "step": 3398
+    },
+    {
+      "epoch": 4.890647482014389,
+      "grad_norm": 0.06802836608159324,
+      "learning_rate": 1.1654406362427762e-07,
+      "loss": 0.3418,
+      "step": 3399
+    },
+    {
+      "epoch": 4.892086330935252,
+      "grad_norm": 0.07014184850529115,
+      "learning_rate": 1.1349873356025332e-07,
+      "loss": 0.3503,
+      "step": 3400
+    },
+    {
+      "epoch": 4.893525179856115,
+      "grad_norm": 0.06877137343946575,
+      "learning_rate": 1.1049366313642395e-07,
+      "loss": 0.3432,
+      "step": 3401
+    },
+    {
+      "epoch": 4.894964028776979,
+      "grad_norm": 0.07090780883347099,
+      "learning_rate": 1.0752885538598102e-07,
+      "loss": 0.3453,
+      "step": 3402
+    },
+    {
+      "epoch": 4.896402877697842,
+      "grad_norm": 0.06937322732548545,
+      "learning_rate": 1.0460431330145515e-07,
+      "loss": 0.3429,
+      "step": 3403
+    },
+    {
+      "epoch": 4.897841726618705,
+      "grad_norm": 0.07059332959477979,
+      "learning_rate": 1.0172003983475176e-07,
+      "loss": 0.3442,
+      "step": 3404
+    },
+    {
+      "epoch": 4.899280575539568,
+      "grad_norm": 0.0691006213917355,
+      "learning_rate": 9.887603789712875e-08,
+      "loss": 0.3456,
+      "step": 3405
+    },
+    {
+      "epoch": 4.900719424460432,
+      "grad_norm": 0.07090172393618893,
+      "learning_rate": 9.607231035919651e-08,
+      "loss": 0.3423,
+      "step": 3406
+    },
+    {
+      "epoch": 4.9021582733812945,
+      "grad_norm": 0.07005456018551516,
+      "learning_rate": 9.330886005090467e-08,
+      "loss": 0.3493,
+      "step": 3407
+    },
+    {
+      "epoch": 4.903597122302158,
+      "grad_norm": 0.0718711843822888,
+      "learning_rate": 9.058568976155979e-08,
+      "loss": 0.3389,
+      "step": 3408
+    },
+    {
+      "epoch": 4.905035971223022,
+      "grad_norm": 0.06994140125733872,
+      "learning_rate": 8.790280223980763e-08,
+      "loss": 0.343,
+      "step": 3409
+    },
+    {
+      "epoch": 4.906474820143885,
+      "grad_norm": 0.07023114151718875,
+      "learning_rate": 8.526020019363313e-08,
+      "loss": 0.349,
+      "step": 3410
+    },
+    {
+      "epoch": 4.907913669064748,
+      "grad_norm": 0.07040002767426613,
+      "learning_rate": 8.265788629036043e-08,
+      "loss": 0.3284,
+      "step": 3411
+    },
+    {
+      "epoch": 4.909352517985612,
+      "grad_norm": 0.0703481353829406,
+      "learning_rate": 8.009586315664842e-08,
+      "loss": 0.354,
+      "step": 3412
+    },
+    {
+      "epoch": 4.910791366906475,
+      "grad_norm": 0.07351480314016241,
+      "learning_rate": 7.757413337848629e-08,
+      "loss": 0.3385,
+      "step": 3413
+    },
+    {
+      "epoch": 4.912230215827338,
+      "grad_norm": 0.07161459246687522,
+      "learning_rate": 7.509269950119358e-08,
+      "loss": 0.3539,
+      "step": 3414
+    },
+    {
+      "epoch": 4.913669064748201,
+      "grad_norm": 0.07032513023391523,
+      "learning_rate": 7.265156402942452e-08,
+      "loss": 0.3483,
+      "step": 3415
+    },
+    {
+      "epoch": 4.915107913669065,
+      "grad_norm": 0.06914653953914185,
+      "learning_rate": 7.025072942714595e-08,
+      "loss": 0.3499,
+      "step": 3416
+    },
+    {
+      "epoch": 4.916546762589928,
+      "grad_norm": 0.07079020521232746,
+      "learning_rate": 6.789019811765052e-08,
+      "loss": 0.3469,
+      "step": 3417
+    },
+    {
+      "epoch": 4.917985611510791,
+      "grad_norm": 0.06854317738438706,
+      "learning_rate": 6.556997248355679e-08,
+      "loss": 0.3461,
+      "step": 3418
+    },
+    {
+      "epoch": 4.919424460431655,
+      "grad_norm": 0.07083775956631229,
+      "learning_rate": 6.329005486679584e-08,
+      "loss": 0.3447,
+      "step": 3419
+    },
+    {
+      "epoch": 4.920863309352518,
+      "grad_norm": 0.06983250569314274,
+      "learning_rate": 6.105044756861134e-08,
+      "loss": 0.3417,
+      "step": 3420
+    },
+    {
+      "epoch": 4.922302158273381,
+      "grad_norm": 0.07146295606304731,
+      "learning_rate": 5.8851152849563886e-08,
+      "loss": 0.3529,
+      "step": 3421
+    },
+    {
+      "epoch": 4.923741007194245,
+      "grad_norm": 0.06983646641591423,
+      "learning_rate": 5.669217292952223e-08,
+      "loss": 0.3381,
+      "step": 3422
+    },
+    {
+      "epoch": 4.925179856115108,
+      "grad_norm": 0.06950858504583642,
+      "learning_rate": 5.4573509987663196e-08,
+      "loss": 0.3496,
+      "step": 3423
+    },
+    {
+      "epoch": 4.926618705035971,
+      "grad_norm": 0.0673192526049635,
+      "learning_rate": 5.2495166162471747e-08,
+      "loss": 0.3429,
+      "step": 3424
+    },
+    {
+      "epoch": 4.928057553956835,
+      "grad_norm": 0.06773038018463526,
+      "learning_rate": 5.045714355173203e-08,
+      "loss": 0.3414,
+      "step": 3425
+    },
+    {
+      "epoch": 4.929496402877698,
+      "grad_norm": 0.06898430126079236,
+      "learning_rate": 4.845944421253634e-08,
+      "loss": 0.3416,
+      "step": 3426
+    },
+    {
+      "epoch": 4.930935251798561,
+      "grad_norm": 0.06814436446211519,
+      "learning_rate": 4.650207016126729e-08,
+      "loss": 0.3429,
+      "step": 3427
+    },
+    {
+      "epoch": 4.932374100719424,
+      "grad_norm": 0.06859985130717335,
+      "learning_rate": 4.458502337361115e-08,
+      "loss": 0.3524,
+      "step": 3428
+    },
+    {
+      "epoch": 4.933812949640288,
+      "grad_norm": 0.06996981696239825,
+      "learning_rate": 4.270830578455343e-08,
+      "loss": 0.3434,
+      "step": 3429
+    },
+    {
+      "epoch": 4.935251798561151,
+      "grad_norm": 0.10234678520498625,
+      "learning_rate": 4.087191928836554e-08,
+      "loss": 0.3342,
+      "step": 3430
+    },
+    {
+      "epoch": 4.936690647482014,
+      "grad_norm": 0.0711256466785614,
+      "learning_rate": 3.907586573860922e-08,
+      "loss": 0.3442,
+      "step": 3431
+    },
+    {
+      "epoch": 4.938129496402878,
+      "grad_norm": 0.06915931117632583,
+      "learning_rate": 3.7320146948149894e-08,
+      "loss": 0.3387,
+      "step": 3432
+    },
+    {
+      "epoch": 4.939568345323741,
+      "grad_norm": 0.07217472866366731,
+      "learning_rate": 3.560476468912111e-08,
+      "loss": 0.3416,
+      "step": 3433
+    },
+    {
+      "epoch": 4.941007194244604,
+      "grad_norm": 0.06985373626638046,
+      "learning_rate": 3.392972069295564e-08,
+      "loss": 0.3426,
+      "step": 3434
+    },
+    {
+      "epoch": 4.942446043165468,
+      "grad_norm": 0.06908507990367821,
+      "learning_rate": 3.229501665037216e-08,
+      "loss": 0.3446,
+      "step": 3435
+    },
+    {
+      "epoch": 4.943884892086331,
+      "grad_norm": 0.06921946977253772,
+      "learning_rate": 3.0700654211361925e-08,
+      "loss": 0.3395,
+      "step": 3436
+    },
+    {
+      "epoch": 4.945323741007194,
+      "grad_norm": 0.06958416841465791,
+      "learning_rate": 2.9146634985206535e-08,
+      "loss": 0.3433,
+      "step": 3437
+    },
+    {
+      "epoch": 4.946762589928057,
+      "grad_norm": 0.06839442541200638,
+      "learning_rate": 2.7632960540460162e-08,
+      "loss": 0.3373,
+      "step": 3438
+    },
+    {
+      "epoch": 4.948201438848921,
+      "grad_norm": 0.06973767335219232,
+      "learning_rate": 2.6159632404958447e-08,
+      "loss": 0.3406,
+      "step": 3439
+    },
+    {
+      "epoch": 4.9496402877697845,
+      "grad_norm": 0.07048415571233173,
+      "learning_rate": 2.472665206581404e-08,
+      "loss": 0.3494,
+      "step": 3440
+    },
+    {
+      "epoch": 4.951079136690647,
+      "grad_norm": 0.071024230979393,
+      "learning_rate": 2.3334020969407733e-08,
+      "loss": 0.3482,
+      "step": 3441
+    },
+    {
+      "epoch": 4.952517985611511,
+      "grad_norm": 0.06840496393214036,
+      "learning_rate": 2.1981740521406226e-08,
+      "loss": 0.3396,
+      "step": 3442
+    },
+    {
+      "epoch": 4.953956834532374,
+      "grad_norm": 0.06957410661638072,
+      "learning_rate": 2.0669812086735464e-08,
+      "loss": 0.339,
+      "step": 3443
+    },
+    {
+      "epoch": 4.955395683453237,
+      "grad_norm": 0.07314235710964967,
+      "learning_rate": 1.9398236989598418e-08,
+      "loss": 0.3453,
+      "step": 3444
+    },
+    {
+      "epoch": 4.956834532374101,
+      "grad_norm": 0.07084760603623917,
+      "learning_rate": 1.8167016513470636e-08,
+      "loss": 0.3497,
+      "step": 3445
+    },
+    {
+      "epoch": 4.958273381294964,
+      "grad_norm": 0.0676189923280852,
+      "learning_rate": 1.697615190107804e-08,
+      "loss": 0.3475,
+      "step": 3446
+    },
+    {
+      "epoch": 4.959712230215827,
+      "grad_norm": 0.0708687898940551,
+      "learning_rate": 1.582564435444134e-08,
+      "loss": 0.341,
+      "step": 3447
+    },
+    {
+      "epoch": 4.961151079136691,
+      "grad_norm": 0.0703179408620645,
+      "learning_rate": 1.4715495034818284e-08,
+      "loss": 0.334,
+      "step": 3448
+    },
+    {
+      "epoch": 4.962589928057554,
+      "grad_norm": 0.07113636099874597,
+      "learning_rate": 1.3645705062748094e-08,
+      "loss": 0.3441,
+      "step": 3449
+    },
+    {
+      "epoch": 4.9640287769784175,
+      "grad_norm": 0.07345379436488886,
+      "learning_rate": 1.2616275518033683e-08,
+      "loss": 0.3394,
+      "step": 3450
+    },
+    {
+      "epoch": 4.96546762589928,
+      "grad_norm": 0.07026785929363381,
+      "learning_rate": 1.1627207439728339e-08,
+      "loss": 0.3398,
+      "step": 3451
+    },
+    {
+      "epoch": 4.966906474820144,
+      "grad_norm": 0.06962592911490988,
+      "learning_rate": 1.0678501826153486e-08,
+      "loss": 0.3519,
+      "step": 3452
+    },
+    {
+      "epoch": 4.968345323741008,
+      "grad_norm": 0.06548377636152011,
+      "learning_rate": 9.770159634894249e-09,
+      "loss": 0.3486,
+      "step": 3453
+    },
+    {
+      "epoch": 4.96978417266187,
+      "grad_norm": 0.06879843007364607,
+      "learning_rate": 8.902181782786124e-09,
+      "loss": 0.3346,
+      "step": 3454
+    },
+    {
+      "epoch": 4.971223021582734,
+      "grad_norm": 0.06919510418451531,
+      "learning_rate": 8.074569145928301e-09,
+      "loss": 0.3444,
+      "step": 3455
+    },
+    {
+      "epoch": 4.972661870503597,
+      "grad_norm": 0.06857966949274884,
+      "learning_rate": 7.287322559679233e-09,
+      "loss": 0.3378,
+      "step": 3456
+    },
+    {
+      "epoch": 4.97410071942446,
+      "grad_norm": 0.06709999989660448,
+      "learning_rate": 6.5404428186433e-09,
+      "loss": 0.3387,
+      "step": 3457
+    },
+    {
+      "epoch": 4.975539568345324,
+      "grad_norm": 0.06821727148625614,
+      "learning_rate": 5.833930676693023e-09,
+      "loss": 0.3292,
+      "step": 3458
+    },
+    {
+      "epoch": 4.976978417266187,
+      "grad_norm": 0.06616684632878932,
+      "learning_rate": 5.167786846946854e-09,
+      "loss": 0.3421,
+      "step": 3459
+    },
+    {
+      "epoch": 4.9784172661870505,
+      "grad_norm": 0.07530907256547428,
+      "learning_rate": 4.542012001778062e-09,
+      "loss": 0.3454,
+      "step": 3460
+    },
+    {
+      "epoch": 4.979856115107914,
+      "grad_norm": 0.06900234637341345,
+      "learning_rate": 3.956606772823613e-09,
+      "loss": 0.3465,
+      "step": 3461
+    },
+    {
+      "epoch": 4.981294964028777,
+      "grad_norm": 0.07441430381886469,
+      "learning_rate": 3.4115717509619616e-09,
+      "loss": 0.3556,
+      "step": 3462
+    },
+    {
+      "epoch": 4.982733812949641,
+      "grad_norm": 0.39321687521638127,
+      "learning_rate": 2.9069074863219414e-09,
+      "loss": 0.3508,
+      "step": 3463
+    },
+    {
+      "epoch": 4.984172661870503,
+      "grad_norm": 0.0708930817455364,
+      "learning_rate": 2.4426144882916392e-09,
+      "loss": 0.3458,
+      "step": 3464
+    },
+    {
+      "epoch": 4.985611510791367,
+      "grad_norm": 0.06812053265762419,
+      "learning_rate": 2.018693225509516e-09,
+      "loss": 0.3433,
+      "step": 3465
+    },
+    {
+      "epoch": 4.98705035971223,
+      "grad_norm": 0.06911482196637857,
+      "learning_rate": 1.6351441258644073e-09,
+      "loss": 0.3396,
+      "step": 3466
+    },
+    {
+      "epoch": 4.988489208633093,
+      "grad_norm": 0.07016639908358226,
+      "learning_rate": 1.2919675764910823e-09,
+      "loss": 0.3405,
+      "step": 3467
+    },
+    {
+      "epoch": 4.989928057553957,
+      "grad_norm": 0.0688260652987113,
+      "learning_rate": 9.89163923770242e-10,
+      "loss": 0.3394,
+      "step": 3468
+    },
+    {
+      "epoch": 4.99136690647482,
+      "grad_norm": 0.06872232929291658,
+      "learning_rate": 7.26733473350727e-10,
+      "loss": 0.3366,
+      "step": 3469
+    },
+    {
+      "epoch": 4.9928057553956835,
+      "grad_norm": 0.07188593839486318,
+      "learning_rate": 5.046764901095457e-10,
+      "loss": 0.3463,
+      "step": 3470
+    },
+    {
+      "epoch": 4.994244604316547,
+      "grad_norm": 0.06941468398555707,
+      "learning_rate": 3.2299319817852283e-10,
+      "loss": 0.3409,
+      "step": 3471
+    },
+    {
+      "epoch": 4.99568345323741,
+      "grad_norm": 0.06822491198591221,
+      "learning_rate": 1.816837809487382e-10,
+      "loss": 0.3453,
+      "step": 3472
+    },
+    {
+      "epoch": 4.997122302158274,
+      "grad_norm": 0.07047016856587555,
+      "learning_rate": 8.074838104832338e-11,
+      "loss": 0.3421,
+      "step": 3473
+    },
+    {
+      "epoch": 4.998561151079136,
+      "grad_norm": 0.06851680587861847,
+      "learning_rate": 2.0187100355784085e-11,
+      "loss": 0.3459,
+      "step": 3474
+    },
+    {
+      "epoch": 5.0,
+      "grad_norm": 0.06834187171522385,
+      "learning_rate": 0.0,
+      "loss": 0.3406,
+      "step": 3475
+    },
+    {
+      "epoch": 5.0,
+      "step": 3475,
+      "total_flos": 5.82970943471616e+16,
+      "train_loss": 0.3069635923474813,
+      "train_runtime": 45586.6789,
+      "train_samples_per_second": 39.019,
+      "train_steps_per_second": 0.076
+    }
+  ],
+  "logging_steps": 1,
+  "max_steps": 3475,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 5,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 5.82970943471616e+16,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}