diff --git "a/checkpoint-30000/trainer_state.json" "b/checkpoint-30000/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/checkpoint-30000/trainer_state.json"
@@ -0,0 +1,18034 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.6,
+  "eval_steps": 500,
+  "global_step": 30000,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "grad_norm": 0.3760893940925598,
+      "learning_rate": 3.6e-07,
+      "loss": 1.1971,
+      "step": 10
+    },
+    {
+      "grad_norm": 0.38699445128440857,
+      "learning_rate": 7.6e-07,
+      "loss": 1.1941,
+      "step": 20
+    },
+    {
+      "grad_norm": 0.3882782757282257,
+      "learning_rate": 1.16e-06,
+      "loss": 1.1982,
+      "step": 30
+    },
+    {
+      "grad_norm": 0.34431272745132446,
+      "learning_rate": 1.56e-06,
+      "loss": 1.1873,
+      "step": 40
+    },
+    {
+      "grad_norm": 0.31745338439941406,
+      "learning_rate": 1.96e-06,
+      "loss": 1.1648,
+      "step": 50
+    },
+    {
+      "grad_norm": 0.20698370039463043,
+      "learning_rate": 2.36e-06,
+      "loss": 1.1459,
+      "step": 60
+    },
+    {
+      "grad_norm": 0.14155547320842743,
+      "learning_rate": 2.7600000000000003e-06,
+      "loss": 1.1412,
+      "step": 70
+    },
+    {
+      "grad_norm": 0.1603740155696869,
+      "learning_rate": 3.1600000000000007e-06,
+      "loss": 1.142,
+      "step": 80
+    },
+    {
+      "grad_norm": 0.19656726717948914,
+      "learning_rate": 3.5600000000000002e-06,
+      "loss": 1.1303,
+      "step": 90
+    },
+    {
+      "grad_norm": 0.22255265712738037,
+      "learning_rate": 3.96e-06,
+      "loss": 1.1246,
+      "step": 100
+    },
+    {
+      "grad_norm": 0.24070441722869873,
+      "learning_rate": 4.360000000000001e-06,
+      "loss": 1.115,
+      "step": 110
+    },
+    {
+      "grad_norm": 0.23834212124347687,
+      "learning_rate": 4.76e-06,
+      "loss": 1.1133,
+      "step": 120
+    },
+    {
+      "grad_norm": 0.23702768981456757,
+      "learning_rate": 5.1600000000000006e-06,
+      "loss": 1.1057,
+      "step": 130
+    },
+    {
+      "grad_norm": 0.2990497946739197,
+      "learning_rate": 5.56e-06,
+      "loss": 1.0996,
+      "step": 140
+    },
+    {
+      "grad_norm": 0.3145226836204529,
+      "learning_rate": 5.9600000000000005e-06,
+      "loss": 1.0859,
+      "step": 150
+    },
+    {
+      "grad_norm": 0.39154860377311707,
+      "learning_rate": 6.360000000000001e-06,
+      "loss": 1.059,
+      "step": 160
+    },
+    {
+      "grad_norm": 0.47389188408851624,
+      "learning_rate": 6.76e-06,
+      "loss": 1.0044,
+      "step": 170
+    },
+    {
+      "grad_norm": 0.8026126027107239,
+      "learning_rate": 7.16e-06,
+      "loss": 0.9363,
+      "step": 180
+    },
+    {
+      "grad_norm": 0.7675435543060303,
+      "learning_rate": 7.5600000000000005e-06,
+      "loss": 0.8721,
+      "step": 190
+    },
+    {
+      "grad_norm": 0.6525933742523193,
+      "learning_rate": 7.96e-06,
+      "loss": 0.8031,
+      "step": 200
+    },
+    {
+      "grad_norm": 0.8948182463645935,
+      "learning_rate": 8.36e-06,
+      "loss": 0.745,
+      "step": 210
+    },
+    {
+      "grad_norm": 0.7966125011444092,
+      "learning_rate": 8.76e-06,
+      "loss": 0.678,
+      "step": 220
+    },
+    {
+      "grad_norm": 1.0041399002075195,
+      "learning_rate": 9.16e-06,
+      "loss": 0.6237,
+      "step": 230
+    },
+    {
+      "grad_norm": 1.2987180948257446,
+      "learning_rate": 9.560000000000002e-06,
+      "loss": 0.5734,
+      "step": 240
+    },
+    {
+      "grad_norm": 1.1447937488555908,
+      "learning_rate": 9.96e-06,
+      "loss": 0.5191,
+      "step": 250
+    },
+    {
+      "grad_norm": 1.0339360237121582,
+      "learning_rate": 1.036e-05,
+      "loss": 0.4831,
+      "step": 260
+    },
+    {
+      "grad_norm": 1.3291385173797607,
+      "learning_rate": 1.076e-05,
+      "loss": 0.447,
+      "step": 270
+    },
+    {
+      "grad_norm": 1.5241327285766602,
+      "learning_rate": 1.1160000000000002e-05,
+      "loss": 0.4102,
+      "step": 280
+    },
+    {
+      "grad_norm": 1.558066725730896,
+      "learning_rate": 1.156e-05,
+      "loss": 0.3814,
+      "step": 290
+    },
+    {
+      "grad_norm": 1.5226203203201294,
+      "learning_rate": 1.196e-05,
+      "loss": 0.3633,
+      "step": 300
+    },
+    {
+      "grad_norm": 1.2429163455963135,
+      "learning_rate": 1.236e-05,
+      "loss": 0.3403,
+      "step": 310
+    },
+    {
+      "grad_norm": 1.3791176080703735,
+      "learning_rate": 1.276e-05,
+      "loss": 0.3,
+      "step": 320
+    },
+    {
+      "grad_norm": 1.2396610975265503,
+      "learning_rate": 1.316e-05,
+      "loss": 0.2863,
+      "step": 330
+    },
+    {
+      "grad_norm": 1.3340177536010742,
+      "learning_rate": 1.356e-05,
+      "loss": 0.2813,
+      "step": 340
+    },
+    {
+      "grad_norm": 1.2155733108520508,
+      "learning_rate": 1.396e-05,
+      "loss": 0.2558,
+      "step": 350
+    },
+    {
+      "grad_norm": 1.574377417564392,
+      "learning_rate": 1.4360000000000001e-05,
+      "loss": 0.2503,
+      "step": 360
+    },
+    {
+      "grad_norm": 1.214402198791504,
+      "learning_rate": 1.4760000000000001e-05,
+      "loss": 0.2249,
+      "step": 370
+    },
+    {
+      "grad_norm": 1.3006868362426758,
+      "learning_rate": 1.5160000000000002e-05,
+      "loss": 0.2171,
+      "step": 380
+    },
+    {
+      "grad_norm": 1.563675880432129,
+      "learning_rate": 1.556e-05,
+      "loss": 0.1999,
+      "step": 390
+    },
+    {
+      "grad_norm": 1.0877236127853394,
+      "learning_rate": 1.596e-05,
+      "loss": 0.1908,
+      "step": 400
+    },
+    {
+      "grad_norm": 1.5446659326553345,
+      "learning_rate": 1.636e-05,
+      "loss": 0.1898,
+      "step": 410
+    },
+    {
+      "grad_norm": 1.1931387186050415,
+      "learning_rate": 1.6760000000000002e-05,
+      "loss": 0.1808,
+      "step": 420
+    },
+    {
+      "grad_norm": 1.5018916130065918,
+      "learning_rate": 1.7160000000000002e-05,
+      "loss": 0.1838,
+      "step": 430
+    },
+    {
+      "grad_norm": 1.3344367742538452,
+      "learning_rate": 1.756e-05,
+      "loss": 0.1712,
+      "step": 440
+    },
+    {
+      "grad_norm": 1.1495717763900757,
+      "learning_rate": 1.796e-05,
+      "loss": 0.173,
+      "step": 450
+    },
+    {
+      "grad_norm": 1.7898554801940918,
+      "learning_rate": 1.8360000000000004e-05,
+      "loss": 0.1674,
+      "step": 460
+    },
+    {
+      "grad_norm": 1.2987682819366455,
+      "learning_rate": 1.876e-05,
+      "loss": 0.1584,
+      "step": 470
+    },
+    {
+      "grad_norm": 1.2640180587768555,
+      "learning_rate": 1.916e-05,
+      "loss": 0.1612,
+      "step": 480
+    },
+    {
+      "grad_norm": 1.5086727142333984,
+      "learning_rate": 1.956e-05,
+      "loss": 0.159,
+      "step": 490
+    },
+    {
+      "grad_norm": 1.2820016145706177,
+      "learning_rate": 1.9960000000000002e-05,
+      "loss": 0.161,
+      "step": 500
+    },
+    {
+      "grad_norm": 1.6688973903656006,
+      "learning_rate": 2.036e-05,
+      "loss": 0.1508,
+      "step": 510
+    },
+    {
+      "grad_norm": 1.4324454069137573,
+      "learning_rate": 2.076e-05,
+      "loss": 0.1511,
+      "step": 520
+    },
+    {
+      "grad_norm": 1.5076853036880493,
+      "learning_rate": 2.116e-05,
+      "loss": 0.1522,
+      "step": 530
+    },
+    {
+      "grad_norm": 1.374239444732666,
+      "learning_rate": 2.1560000000000004e-05,
+      "loss": 0.1512,
+      "step": 540
+    },
+    {
+      "grad_norm": 1.3314145803451538,
+      "learning_rate": 2.196e-05,
+      "loss": 0.1371,
+      "step": 550
+    },
+    {
+      "grad_norm": 1.3286856412887573,
+      "learning_rate": 2.236e-05,
+      "loss": 0.1492,
+      "step": 560
+    },
+    {
+      "grad_norm": 1.523295521736145,
+      "learning_rate": 2.2760000000000002e-05,
+      "loss": 0.1529,
+      "step": 570
+    },
+    {
+      "grad_norm": 1.5259977579116821,
+      "learning_rate": 2.3160000000000002e-05,
+      "loss": 0.1485,
+      "step": 580
+    },
+    {
+      "grad_norm": 1.1062870025634766,
+      "learning_rate": 2.356e-05,
+      "loss": 0.1407,
+      "step": 590
+    },
+    {
+      "grad_norm": 1.5761271715164185,
+      "learning_rate": 2.396e-05,
+      "loss": 0.1454,
+      "step": 600
+    },
+    {
+      "grad_norm": 1.458147644996643,
+      "learning_rate": 2.4360000000000004e-05,
+      "loss": 0.1388,
+      "step": 610
+    },
+    {
+      "grad_norm": 1.248154878616333,
+      "learning_rate": 2.476e-05,
+      "loss": 0.1382,
+      "step": 620
+    },
+    {
+      "grad_norm": 1.2746696472167969,
+      "learning_rate": 2.516e-05,
+      "loss": 0.1391,
+      "step": 630
+    },
+    {
+      "grad_norm": 1.3740546703338623,
+      "learning_rate": 2.556e-05,
+      "loss": 0.1401,
+      "step": 640
+    },
+    {
+      "grad_norm": 1.3423243761062622,
+      "learning_rate": 2.5960000000000002e-05,
+      "loss": 0.1365,
+      "step": 650
+    },
+    {
+      "grad_norm": 1.3729647397994995,
+      "learning_rate": 2.6360000000000002e-05,
+      "loss": 0.1369,
+      "step": 660
+    },
+    {
+      "grad_norm": 1.210018277168274,
+      "learning_rate": 2.676e-05,
+      "loss": 0.1304,
+      "step": 670
+    },
+    {
+      "grad_norm": 1.2842462062835693,
+      "learning_rate": 2.716e-05,
+      "loss": 0.1279,
+      "step": 680
+    },
+    {
+      "grad_norm": 1.2189555168151855,
+      "learning_rate": 2.7560000000000004e-05,
+      "loss": 0.1209,
+      "step": 690
+    },
+    {
+      "grad_norm": 1.3705235719680786,
+      "learning_rate": 2.7960000000000003e-05,
+      "loss": 0.127,
+      "step": 700
+    },
+    {
+      "grad_norm": 1.3271591663360596,
+      "learning_rate": 2.8360000000000003e-05,
+      "loss": 0.1254,
+      "step": 710
+    },
+    {
+      "grad_norm": 1.1703218221664429,
+      "learning_rate": 2.8760000000000002e-05,
+      "loss": 0.1245,
+      "step": 720
+    },
+    {
+      "grad_norm": 1.1060250997543335,
+      "learning_rate": 2.9160000000000005e-05,
+      "loss": 0.1282,
+      "step": 730
+    },
+    {
+      "grad_norm": 1.2592685222625732,
+      "learning_rate": 2.9559999999999998e-05,
+      "loss": 0.1298,
+      "step": 740
+    },
+    {
+      "grad_norm": 1.33540678024292,
+      "learning_rate": 2.9959999999999998e-05,
+      "loss": 0.128,
+      "step": 750
+    },
+    {
+      "grad_norm": 1.4297021627426147,
+      "learning_rate": 3.036e-05,
+      "loss": 0.1224,
+      "step": 760
+    },
+    {
+      "grad_norm": 1.2536511421203613,
+      "learning_rate": 3.076e-05,
+      "loss": 0.1271,
+      "step": 770
+    },
+    {
+      "grad_norm": 1.2478418350219727,
+      "learning_rate": 3.116e-05,
+      "loss": 0.1219,
+      "step": 780
+    },
+    {
+      "grad_norm": 1.1744983196258545,
+      "learning_rate": 3.156e-05,
+      "loss": 0.1131,
+      "step": 790
+    },
+    {
+      "grad_norm": 1.2983360290527344,
+      "learning_rate": 3.196e-05,
+      "loss": 0.119,
+      "step": 800
+    },
+    {
+      "grad_norm": 1.1186442375183105,
+      "learning_rate": 3.236e-05,
+      "loss": 0.1116,
+      "step": 810
+    },
+    {
+      "grad_norm": 1.0257961750030518,
+      "learning_rate": 3.2760000000000005e-05,
+      "loss": 0.1146,
+      "step": 820
+    },
+    {
+      "grad_norm": 1.4574180841445923,
+      "learning_rate": 3.316e-05,
+      "loss": 0.1125,
+      "step": 830
+    },
+    {
+      "grad_norm": 1.1860361099243164,
+      "learning_rate": 3.3560000000000004e-05,
+      "loss": 0.1179,
+      "step": 840
+    },
+    {
+      "grad_norm": 1.0476770401000977,
+      "learning_rate": 3.396e-05,
+      "loss": 0.1128,
+      "step": 850
+    },
+    {
+      "grad_norm": 1.2919471263885498,
+      "learning_rate": 3.436e-05,
+      "loss": 0.1127,
+      "step": 860
+    },
+    {
+      "grad_norm": 1.3819019794464111,
+      "learning_rate": 3.4760000000000006e-05,
+      "loss": 0.1137,
+      "step": 870
+    },
+    {
+      "grad_norm": 1.2365814447402954,
+      "learning_rate": 3.516e-05,
+      "loss": 0.1142,
+      "step": 880
+    },
+    {
+      "grad_norm": 1.203905463218689,
+      "learning_rate": 3.5560000000000005e-05,
+      "loss": 0.1095,
+      "step": 890
+    },
+    {
+      "grad_norm": 0.930099368095398,
+      "learning_rate": 3.596e-05,
+      "loss": 0.1143,
+      "step": 900
+    },
+    {
+      "grad_norm": 1.1405550241470337,
+      "learning_rate": 3.636e-05,
+      "loss": 0.1051,
+      "step": 910
+    },
+    {
+      "grad_norm": 1.299981713294983,
+      "learning_rate": 3.676e-05,
+      "loss": 0.1045,
+      "step": 920
+    },
+    {
+      "grad_norm": 1.0480676889419556,
+      "learning_rate": 3.716e-05,
+      "loss": 0.1124,
+      "step": 930
+    },
+    {
+      "grad_norm": 1.0490866899490356,
+      "learning_rate": 3.756e-05,
+      "loss": 0.1053,
+      "step": 940
+    },
+    {
+      "grad_norm": 1.1781903505325317,
+      "learning_rate": 3.796e-05,
+      "loss": 0.1078,
+      "step": 950
+    },
+    {
+      "grad_norm": 1.3801507949829102,
+      "learning_rate": 3.836e-05,
+      "loss": 0.107,
+      "step": 960
+    },
+    {
+      "grad_norm": 1.212050437927246,
+      "learning_rate": 3.876e-05,
+      "loss": 0.107,
+      "step": 970
+    },
+    {
+      "grad_norm": 1.0979703664779663,
+      "learning_rate": 3.9160000000000005e-05,
+      "loss": 0.1019,
+      "step": 980
+    },
+    {
+      "grad_norm": 0.9218549132347107,
+      "learning_rate": 3.956e-05,
+      "loss": 0.1104,
+      "step": 990
+    },
+    {
+      "grad_norm": 1.1156573295593262,
+      "learning_rate": 3.9960000000000004e-05,
+      "loss": 0.0971,
+      "step": 1000
+    },
+    {
+      "grad_norm": 1.1480847597122192,
+      "learning_rate": 4.0360000000000007e-05,
+      "loss": 0.0978,
+      "step": 1010
+    },
+    {
+      "grad_norm": 1.1778688430786133,
+      "learning_rate": 4.076e-05,
+      "loss": 0.1051,
+      "step": 1020
+    },
+    {
+      "grad_norm": 1.0225567817687988,
+      "learning_rate": 4.1160000000000006e-05,
+      "loss": 0.0968,
+      "step": 1030
+    },
+    {
+      "grad_norm": 1.0937094688415527,
+      "learning_rate": 4.156e-05,
+      "loss": 0.093,
+      "step": 1040
+    },
+    {
+      "grad_norm": 1.1233748197555542,
+      "learning_rate": 4.196e-05,
+      "loss": 0.0936,
+      "step": 1050
+    },
+    {
+      "grad_norm": 1.2596155405044556,
+      "learning_rate": 4.236e-05,
+      "loss": 0.0925,
+      "step": 1060
+    },
+    {
+      "grad_norm": 1.0893105268478394,
+      "learning_rate": 4.276e-05,
+      "loss": 0.0975,
+      "step": 1070
+    },
+    {
+      "grad_norm": 1.1965258121490479,
+      "learning_rate": 4.316e-05,
+      "loss": 0.0995,
+      "step": 1080
+    },
+    {
+      "grad_norm": 0.9918234944343567,
+      "learning_rate": 4.356e-05,
+      "loss": 0.0945,
+      "step": 1090
+    },
+    {
+      "grad_norm": 0.9043128490447998,
+      "learning_rate": 4.396e-05,
+      "loss": 0.0897,
+      "step": 1100
+    },
+    {
+      "grad_norm": 0.9621319770812988,
+      "learning_rate": 4.436e-05,
+      "loss": 0.0895,
+      "step": 1110
+    },
+    {
+      "grad_norm": 1.032353401184082,
+      "learning_rate": 4.4760000000000005e-05,
+      "loss": 0.0909,
+      "step": 1120
+    },
+    {
+      "grad_norm": 0.7686614990234375,
+      "learning_rate": 4.516e-05,
+      "loss": 0.0948,
+      "step": 1130
+    },
+    {
+      "grad_norm": 1.1762384176254272,
+      "learning_rate": 4.5560000000000004e-05,
+      "loss": 0.09,
+      "step": 1140
+    },
+    {
+      "grad_norm": 1.1499114036560059,
+      "learning_rate": 4.596e-05,
+      "loss": 0.0865,
+      "step": 1150
+    },
+    {
+      "grad_norm": 1.1761037111282349,
+      "learning_rate": 4.636e-05,
+      "loss": 0.0921,
+      "step": 1160
+    },
+    {
+      "grad_norm": 0.9816262125968933,
+      "learning_rate": 4.6760000000000006e-05,
+      "loss": 0.0903,
+      "step": 1170
+    },
+    {
+      "grad_norm": 1.0206578969955444,
+      "learning_rate": 4.716e-05,
+      "loss": 0.0891,
+      "step": 1180
+    },
+    {
+      "grad_norm": 1.1863044500350952,
+      "learning_rate": 4.7560000000000005e-05,
+      "loss": 0.0856,
+      "step": 1190
+    },
+    {
+      "grad_norm": 1.156340479850769,
+      "learning_rate": 4.796e-05,
+      "loss": 0.0912,
+      "step": 1200
+    },
+    {
+      "grad_norm": 0.9611001014709473,
+      "learning_rate": 4.836e-05,
+      "loss": 0.0863,
+      "step": 1210
+    },
+    {
+      "grad_norm": 0.9392980933189392,
+      "learning_rate": 4.876e-05,
+      "loss": 0.0885,
+      "step": 1220
+    },
+    {
+      "grad_norm": 0.9977777004241943,
+      "learning_rate": 4.9160000000000004e-05,
+      "loss": 0.0885,
+      "step": 1230
+    },
+    {
+      "grad_norm": 1.022810459136963,
+      "learning_rate": 4.956e-05,
+      "loss": 0.0866,
+      "step": 1240
+    },
+    {
+      "grad_norm": 0.9722366333007812,
+      "learning_rate": 4.996e-05,
+      "loss": 0.0809,
+      "step": 1250
+    },
+    {
+      "grad_norm": 1.203736424446106,
+      "learning_rate": 5.0360000000000006e-05,
+      "loss": 0.0833,
+      "step": 1260
+    },
+    {
+      "grad_norm": 1.088273525238037,
+      "learning_rate": 5.076000000000001e-05,
+      "loss": 0.0825,
+      "step": 1270
+    },
+    {
+      "grad_norm": 0.9986542463302612,
+      "learning_rate": 5.1160000000000005e-05,
+      "loss": 0.088,
+      "step": 1280
+    },
+    {
+      "grad_norm": 0.9373347759246826,
+      "learning_rate": 5.1559999999999994e-05,
+      "loss": 0.0811,
+      "step": 1290
+    },
+    {
+      "grad_norm": 0.9269506335258484,
+      "learning_rate": 5.196e-05,
+      "loss": 0.081,
+      "step": 1300
+    },
+    {
+      "grad_norm": 0.7714357972145081,
+      "learning_rate": 5.236e-05,
+      "loss": 0.0785,
+      "step": 1310
+    },
+    {
+      "grad_norm": 0.8493567109107971,
+      "learning_rate": 5.2759999999999996e-05,
+      "loss": 0.0736,
+      "step": 1320
+    },
+    {
+      "grad_norm": 1.1910462379455566,
+      "learning_rate": 5.316e-05,
+      "loss": 0.0866,
+      "step": 1330
+    },
+    {
+      "grad_norm": 0.9215689897537231,
+      "learning_rate": 5.356e-05,
+      "loss": 0.0824,
+      "step": 1340
+    },
+    {
+      "grad_norm": 0.8018513321876526,
+      "learning_rate": 5.396e-05,
+      "loss": 0.0814,
+      "step": 1350
+    },
+    {
+      "grad_norm": 0.9150856137275696,
+      "learning_rate": 5.436e-05,
+      "loss": 0.08,
+      "step": 1360
+    },
+    {
+      "grad_norm": 0.8493928909301758,
+      "learning_rate": 5.476e-05,
+      "loss": 0.0788,
+      "step": 1370
+    },
+    {
+      "grad_norm": 0.9720738530158997,
+      "learning_rate": 5.516e-05,
+      "loss": 0.0738,
+      "step": 1380
+    },
+    {
+      "grad_norm": 0.9904879927635193,
+      "learning_rate": 5.556e-05,
+      "loss": 0.0757,
+      "step": 1390
+    },
+    {
+      "grad_norm": 0.9153587818145752,
+      "learning_rate": 5.596e-05,
+      "loss": 0.0753,
+      "step": 1400
+    },
+    {
+      "grad_norm": 1.0268296003341675,
+      "learning_rate": 5.636e-05,
+      "loss": 0.0751,
+      "step": 1410
+    },
+    {
+      "grad_norm": 0.9284419417381287,
+      "learning_rate": 5.6760000000000005e-05,
+      "loss": 0.0736,
+      "step": 1420
+    },
+    {
+      "grad_norm": 0.9678248167037964,
+      "learning_rate": 5.716e-05,
+      "loss": 0.0748,
+      "step": 1430
+    },
+    {
+      "grad_norm": 0.8807039856910706,
+      "learning_rate": 5.7560000000000005e-05,
+      "loss": 0.0717,
+      "step": 1440
+    },
+    {
+      "grad_norm": 0.9213088750839233,
+      "learning_rate": 5.796e-05,
+      "loss": 0.0727,
+      "step": 1450
+    },
+    {
+      "grad_norm": 1.0568040609359741,
+      "learning_rate": 5.8360000000000004e-05,
+      "loss": 0.0833,
+      "step": 1460
+    },
+    {
+      "grad_norm": 0.7340942621231079,
+      "learning_rate": 5.876000000000001e-05,
+      "loss": 0.0733,
+      "step": 1470
+    },
+    {
+      "grad_norm": 0.7920780777931213,
+      "learning_rate": 5.916e-05,
+      "loss": 0.0761,
+      "step": 1480
+    },
+    {
+      "grad_norm": 0.8226042985916138,
+      "learning_rate": 5.9560000000000006e-05,
+      "loss": 0.072,
+      "step": 1490
+    },
+    {
+      "grad_norm": 0.9058213233947754,
+      "learning_rate": 5.996e-05,
+      "loss": 0.0712,
+      "step": 1500
+    },
+    {
+      "grad_norm": 0.9899120330810547,
+      "learning_rate": 6.0360000000000005e-05,
+      "loss": 0.0745,
+      "step": 1510
+    },
+    {
+      "grad_norm": 1.0195763111114502,
+      "learning_rate": 6.076000000000001e-05,
+      "loss": 0.0651,
+      "step": 1520
+    },
+    {
+      "grad_norm": 0.7832097411155701,
+      "learning_rate": 6.116e-05,
+      "loss": 0.0722,
+      "step": 1530
+    },
+    {
+      "grad_norm": 0.9364311695098877,
+      "learning_rate": 6.156e-05,
+      "loss": 0.0736,
+      "step": 1540
+    },
+    {
+      "grad_norm": 0.8307925462722778,
+      "learning_rate": 6.196000000000001e-05,
+      "loss": 0.0724,
+      "step": 1550
+    },
+    {
+      "grad_norm": 0.9692476987838745,
+      "learning_rate": 6.236e-05,
+      "loss": 0.0715,
+      "step": 1560
+    },
+    {
+      "grad_norm": 0.9711029529571533,
+      "learning_rate": 6.276e-05,
+      "loss": 0.0731,
+      "step": 1570
+    },
+    {
+      "grad_norm": 0.8757683634757996,
+      "learning_rate": 6.316000000000001e-05,
+      "loss": 0.0695,
+      "step": 1580
+    },
+    {
+      "grad_norm": 0.8889736533164978,
+      "learning_rate": 6.356000000000001e-05,
+      "loss": 0.069,
+      "step": 1590
+    },
+    {
+      "grad_norm": 0.7749250531196594,
+      "learning_rate": 6.396e-05,
+      "loss": 0.073,
+      "step": 1600
+    },
+    {
+      "grad_norm": 0.8744382858276367,
+      "learning_rate": 6.436e-05,
+      "loss": 0.0678,
+      "step": 1610
+    },
+    {
+      "grad_norm": 0.8584306836128235,
+      "learning_rate": 6.476e-05,
+      "loss": 0.0634,
+      "step": 1620
+    },
+    {
+      "grad_norm": 0.7049430012702942,
+      "learning_rate": 6.515999999999999e-05,
+      "loss": 0.0699,
+      "step": 1630
+    },
+    {
+      "grad_norm": 0.7322341799736023,
+      "learning_rate": 6.556e-05,
+      "loss": 0.0647,
+      "step": 1640
+    },
+    {
+      "grad_norm": 0.7489832639694214,
+      "learning_rate": 6.596e-05,
+      "loss": 0.0663,
+      "step": 1650
+    },
+    {
+      "grad_norm": 0.6740495562553406,
+      "learning_rate": 6.636e-05,
+      "loss": 0.0616,
+      "step": 1660
+    },
+    {
+      "grad_norm": 0.981742799282074,
+      "learning_rate": 6.676e-05,
+      "loss": 0.0658,
+      "step": 1670
+    },
+    {
+      "grad_norm": 0.9348631501197815,
+      "learning_rate": 6.716e-05,
+      "loss": 0.0685,
+      "step": 1680
+    },
+    {
+      "grad_norm": 0.6122395396232605,
+      "learning_rate": 6.756e-05,
+      "loss": 0.0736,
+      "step": 1690
+    },
+    {
+      "grad_norm": 0.85508131980896,
+      "learning_rate": 6.796e-05,
+      "loss": 0.0722,
+      "step": 1700
+    },
+    {
+      "grad_norm": 0.8538367748260498,
+      "learning_rate": 6.836e-05,
+      "loss": 0.0666,
+      "step": 1710
+    },
+    {
+      "grad_norm": 0.6837155222892761,
+      "learning_rate": 6.876e-05,
+      "loss": 0.0652,
+      "step": 1720
+    },
+    {
+      "grad_norm": 0.7249764204025269,
+      "learning_rate": 6.916000000000001e-05,
+      "loss": 0.0634,
+      "step": 1730
+    },
+    {
+      "grad_norm": 0.6477764248847961,
+      "learning_rate": 6.956e-05,
+      "loss": 0.0614,
+      "step": 1740
+    },
+    {
+      "grad_norm": 0.995432436466217,
+      "learning_rate": 6.996e-05,
+      "loss": 0.0665,
+      "step": 1750
+    },
+    {
+      "grad_norm": 0.8936553001403809,
+      "learning_rate": 7.036e-05,
+      "loss": 0.0619,
+      "step": 1760
+    },
+    {
+      "grad_norm": 0.6701622605323792,
+      "learning_rate": 7.076000000000001e-05,
+      "loss": 0.0632,
+      "step": 1770
+    },
+    {
+      "grad_norm": 0.984387218952179,
+      "learning_rate": 7.116e-05,
+      "loss": 0.0627,
+      "step": 1780
+    },
+    {
+      "grad_norm": 0.7273459434509277,
+      "learning_rate": 7.156e-05,
+      "loss": 0.0651,
+      "step": 1790
+    },
+    {
+      "grad_norm": 0.7154653668403625,
+      "learning_rate": 7.196000000000001e-05,
+      "loss": 0.0632,
+      "step": 1800
+    },
+    {
+      "grad_norm": 0.8628573417663574,
+      "learning_rate": 7.236e-05,
+      "loss": 0.0593,
+      "step": 1810
+    },
+    {
+      "grad_norm": 0.8995165228843689,
+      "learning_rate": 7.276e-05,
+      "loss": 0.0621,
+      "step": 1820
+    },
+    {
+      "grad_norm": 0.7969207167625427,
+      "learning_rate": 7.316000000000001e-05,
+      "loss": 0.0584,
+      "step": 1830
+    },
+    {
+      "grad_norm": 0.7371864318847656,
+      "learning_rate": 7.356000000000001e-05,
+      "loss": 0.0611,
+      "step": 1840
+    },
+    {
+      "grad_norm": 0.8195193409919739,
+      "learning_rate": 7.396e-05,
+      "loss": 0.0604,
+      "step": 1850
+    },
+    {
+      "grad_norm": 0.704009473323822,
+      "learning_rate": 7.436000000000001e-05,
+      "loss": 0.062,
+      "step": 1860
+    },
+    {
+      "grad_norm": 0.8253884315490723,
+      "learning_rate": 7.476000000000001e-05,
+      "loss": 0.0586,
+      "step": 1870
+    },
+    {
+      "grad_norm": 0.7015737891197205,
+      "learning_rate": 7.516e-05,
+      "loss": 0.0581,
+      "step": 1880
+    },
+    {
+      "grad_norm": 0.618018388748169,
+      "learning_rate": 7.556000000000002e-05,
+      "loss": 0.0627,
+      "step": 1890
+    },
+    {
+      "grad_norm": 0.7612485885620117,
+      "learning_rate": 7.596000000000001e-05,
+      "loss": 0.0572,
+      "step": 1900
+    },
+    {
+      "grad_norm": 0.9041772484779358,
+      "learning_rate": 7.636e-05,
+      "loss": 0.0577,
+      "step": 1910
+    },
+    {
+      "grad_norm": 0.8144323229789734,
+      "learning_rate": 7.676e-05,
+      "loss": 0.0584,
+      "step": 1920
+    },
+    {
+      "grad_norm": 0.8304576873779297,
+      "learning_rate": 7.716e-05,
+      "loss": 0.0602,
+      "step": 1930
+    },
+    {
+      "grad_norm": 0.7301172018051147,
+      "learning_rate": 7.756e-05,
+      "loss": 0.0587,
+      "step": 1940
+    },
+    {
+      "grad_norm": 0.6217229962348938,
+      "learning_rate": 7.796e-05,
+      "loss": 0.0565,
+      "step": 1950
+    },
+    {
+      "grad_norm": 0.6973743438720703,
+      "learning_rate": 7.836e-05,
+      "loss": 0.0566,
+      "step": 1960
+    },
+    {
+      "grad_norm": 0.781914472579956,
+      "learning_rate": 7.876e-05,
+      "loss": 0.0579,
+      "step": 1970
+    },
+    {
+      "grad_norm": 0.7222826480865479,
+      "learning_rate": 7.916e-05,
+      "loss": 0.0569,
+      "step": 1980
+    },
+    {
+      "grad_norm": 0.9131620526313782,
+      "learning_rate": 7.956e-05,
+      "loss": 0.0562,
+      "step": 1990
+    },
+    {
+      "grad_norm": 0.9919409155845642,
+      "learning_rate": 7.996e-05,
+      "loss": 0.0568,
+      "step": 2000
+    },
+    {
+      "grad_norm": 0.924788236618042,
+      "learning_rate": 8.036e-05,
+      "loss": 0.054,
+      "step": 2010
+    },
+    {
+      "grad_norm": 0.7026146650314331,
+      "learning_rate": 8.076e-05,
+      "loss": 0.0591,
+      "step": 2020
+    },
+    {
+      "grad_norm": 0.7011987566947937,
+      "learning_rate": 8.116e-05,
+      "loss": 0.0558,
+      "step": 2030
+    },
+    {
+      "grad_norm": 0.8663361072540283,
+      "learning_rate": 8.156e-05,
+      "loss": 0.056,
+      "step": 2040
+    },
+    {
+      "grad_norm": 0.8171712756156921,
+      "learning_rate": 8.196000000000001e-05,
+      "loss": 0.0595,
+      "step": 2050
+    },
+    {
+      "grad_norm": 0.5253852009773254,
+      "learning_rate": 8.236e-05,
+      "loss": 0.0576,
+      "step": 2060
+    },
+    {
+      "grad_norm": 0.7613351941108704,
+      "learning_rate": 8.276e-05,
+      "loss": 0.0568,
+      "step": 2070
+    },
+    {
+      "grad_norm": 0.820510745048523,
+      "learning_rate": 8.316000000000001e-05,
+      "loss": 0.0502,
+      "step": 2080
+    },
+    {
+      "grad_norm": 0.8240765929222107,
+      "learning_rate": 8.356e-05,
+      "loss": 0.0573,
+      "step": 2090
+    },
+    {
+      "grad_norm": 0.6421526670455933,
+      "learning_rate": 8.396e-05,
+      "loss": 0.0526,
+      "step": 2100
+    },
+    {
+      "grad_norm": 0.7394821643829346,
+      "learning_rate": 8.436000000000001e-05,
+      "loss": 0.0497,
+      "step": 2110
+    },
+    {
+      "grad_norm": 0.6699467897415161,
+      "learning_rate": 8.476000000000001e-05,
+      "loss": 0.0521,
+      "step": 2120
+    },
+    {
+      "grad_norm": 0.5605759024620056,
+      "learning_rate": 8.516e-05,
+      "loss": 0.0536,
+      "step": 2130
+    },
+    {
+      "grad_norm": 0.9242166876792908,
+      "learning_rate": 8.556e-05,
+      "loss": 0.0502,
+      "step": 2140
+    },
+    {
+      "grad_norm": 0.6284075379371643,
+      "learning_rate": 8.596000000000001e-05,
+      "loss": 0.0507,
+      "step": 2150
+    },
+    {
+      "grad_norm": 0.7128509283065796,
+      "learning_rate": 8.636e-05,
+      "loss": 0.0477,
+      "step": 2160
+    },
+    {
+      "grad_norm": 0.6592251062393188,
+      "learning_rate": 8.676e-05,
+      "loss": 0.0471,
+      "step": 2170
+    },
+    {
+      "grad_norm": 0.7484652996063232,
+      "learning_rate": 8.716000000000001e-05,
+      "loss": 0.0484,
+      "step": 2180
+    },
+    {
+      "grad_norm": 0.7146239280700684,
+      "learning_rate": 8.756000000000001e-05,
+      "loss": 0.0478,
+      "step": 2190
+    },
+    {
+      "grad_norm": 0.6576828956604004,
+      "learning_rate": 8.796e-05,
+      "loss": 0.0553,
+      "step": 2200
+    },
+    {
+      "grad_norm": 0.6931664943695068,
+      "learning_rate": 8.836000000000001e-05,
+      "loss": 0.0521,
+      "step": 2210
+    },
+    {
+      "grad_norm": 0.7046792507171631,
+      "learning_rate": 8.876e-05,
+      "loss": 0.0537,
+      "step": 2220
+    },
+    {
+      "grad_norm": 0.7148909568786621,
+      "learning_rate": 8.916e-05,
+      "loss": 0.0467,
+      "step": 2230
+    },
+    {
+      "grad_norm": 0.5783629417419434,
+      "learning_rate": 8.956e-05,
+      "loss": 0.0463,
+      "step": 2240
+    },
+    {
+      "grad_norm": 0.6429301500320435,
+      "learning_rate": 8.996e-05,
+      "loss": 0.048,
+      "step": 2250
+    },
+    {
+      "grad_norm": 0.8260622024536133,
+      "learning_rate": 9.036e-05,
+      "loss": 0.0505,
+      "step": 2260
+    },
+    {
+      "grad_norm": 0.5829892158508301,
+      "learning_rate": 9.076e-05,
+      "loss": 0.0467,
+      "step": 2270
+    },
+    {
+      "grad_norm": 0.7291595935821533,
+      "learning_rate": 9.116e-05,
+      "loss": 0.0442,
+      "step": 2280
+    },
+    {
+      "grad_norm": 0.6018039584159851,
+      "learning_rate": 9.156e-05,
+      "loss": 0.0465,
+      "step": 2290
+    },
+    {
+      "grad_norm": 0.6948915123939514,
+      "learning_rate": 9.196000000000001e-05,
+      "loss": 0.0461,
+      "step": 2300
+    },
+    {
+      "grad_norm": 0.6171465516090393,
+      "learning_rate": 9.236e-05,
+      "loss": 0.0488,
+      "step": 2310
+    },
+    {
+      "grad_norm": 0.699181079864502,
+      "learning_rate": 9.276e-05,
+      "loss": 0.0475,
+      "step": 2320
+    },
+    {
+      "grad_norm": 0.8079102635383606,
+      "learning_rate": 9.316000000000001e-05,
+      "loss": 0.0523,
+      "step": 2330
+    },
+    {
+      "grad_norm": 0.740074098110199,
+      "learning_rate": 9.356e-05,
+      "loss": 0.0465,
+      "step": 2340
+    },
+    {
+      "grad_norm": 0.682598352432251,
+      "learning_rate": 9.396e-05,
+      "loss": 0.0534,
+      "step": 2350
+    },
+    {
+      "grad_norm": 0.6481727361679077,
+      "learning_rate": 9.436e-05,
+      "loss": 0.0485,
+      "step": 2360
+    },
+    {
+      "grad_norm": 0.7049840092658997,
+      "learning_rate": 9.476000000000001e-05,
+      "loss": 0.0451,
+      "step": 2370
+    },
+    {
+      "grad_norm": 0.5933404564857483,
+      "learning_rate": 9.516e-05,
+      "loss": 0.0457,
+      "step": 2380
+    },
+    {
+      "grad_norm": 0.6148690581321716,
+      "learning_rate": 9.556e-05,
+      "loss": 0.0461,
+      "step": 2390
+    },
+    {
+      "grad_norm": 0.7480904459953308,
+      "learning_rate": 9.596000000000001e-05,
+      "loss": 0.0444,
+      "step": 2400
+    },
+    {
+      "grad_norm": 0.602479875087738,
+      "learning_rate": 9.636e-05,
+      "loss": 0.0411,
+      "step": 2410
+    },
+    {
+      "grad_norm": 0.6820979118347168,
+      "learning_rate": 9.676e-05,
+      "loss": 0.0514,
+      "step": 2420
+    },
+    {
+      "grad_norm": 0.6606934070587158,
+      "learning_rate": 9.716000000000001e-05,
+      "loss": 0.0458,
+      "step": 2430
+    },
+    {
+      "grad_norm": 0.5188480019569397,
+      "learning_rate": 9.756000000000001e-05,
+      "loss": 0.0471,
+      "step": 2440
+    },
+    {
+      "grad_norm": 0.5802158713340759,
+      "learning_rate": 9.796e-05,
+      "loss": 0.0456,
+      "step": 2450
+    },
+    {
+      "grad_norm": 0.6156823635101318,
+      "learning_rate": 9.836000000000001e-05,
+      "loss": 0.0411,
+      "step": 2460
+    },
+    {
+      "grad_norm": 0.7152356505393982,
+      "learning_rate": 9.876000000000001e-05,
+      "loss": 0.0447,
+      "step": 2470
+    },
+    {
+      "grad_norm": 0.48657292127609253,
+      "learning_rate": 9.916e-05,
+      "loss": 0.0409,
+      "step": 2480
+    },
+    {
+      "grad_norm": 0.7265614867210388,
+      "learning_rate": 9.956e-05,
+      "loss": 0.0489,
+      "step": 2490
+    },
+    {
+      "grad_norm": 0.5641865730285645,
+      "learning_rate": 9.996000000000001e-05,
+      "loss": 0.0427,
+      "step": 2500
+    },
+    {
+      "grad_norm": 0.5341804027557373,
+      "learning_rate": 9.999999114196196e-05,
+      "loss": 0.0423,
+      "step": 2510
+    },
+    {
+      "grad_norm": 0.4816654324531555,
+      "learning_rate": 9.99999605215876e-05,
+      "loss": 0.0401,
+      "step": 2520
+    },
+    {
+      "grad_norm": 0.7432696223258972,
+      "learning_rate": 9.999990802953179e-05,
+      "loss": 0.043,
+      "step": 2530
+    },
+    {
+      "grad_norm": 0.6123446226119995,
+      "learning_rate": 9.99998336658175e-05,
+      "loss": 0.0401,
+      "step": 2540
+    },
+    {
+      "grad_norm": 0.6642088294029236,
+      "learning_rate": 9.999973743047727e-05,
+      "loss": 0.0432,
+      "step": 2550
+    },
+    {
+      "grad_norm": 0.5572113394737244,
+      "learning_rate": 9.999961932355319e-05,
+      "loss": 0.0462,
+      "step": 2560
+    },
+    {
+      "grad_norm": 0.598061740398407,
+      "learning_rate": 9.999947934509693e-05,
+      "loss": 0.0382,
+      "step": 2570
+    },
+    {
+      "grad_norm": 0.49145016074180603,
+      "learning_rate": 9.999931749516971e-05,
+      "loss": 0.0412,
+      "step": 2580
+    },
+    {
+      "grad_norm": 0.6124422550201416,
+      "learning_rate": 9.999913377384233e-05,
+      "loss": 0.0371,
+      "step": 2590
+    },
+    {
+      "grad_norm": 0.48855262994766235,
+      "learning_rate": 9.999892818119517e-05,
+      "loss": 0.0448,
+      "step": 2600
+    },
+    {
+      "grad_norm": 0.620868980884552,
+      "learning_rate": 9.999870071731814e-05,
+      "loss": 0.0425,
+      "step": 2610
+    },
+    {
+      "grad_norm": 0.5371754169464111,
+      "learning_rate": 9.999845138231076e-05,
+      "loss": 0.0406,
+      "step": 2620
+    },
+    {
+      "grad_norm": 0.5338670015335083,
+      "learning_rate": 9.999818017628208e-05,
+      "loss": 0.0396,
+      "step": 2630
+    },
+    {
+      "grad_norm": 0.5457975268363953,
+      "learning_rate": 9.999788709935078e-05,
+      "loss": 0.0417,
+      "step": 2640
+    },
+    {
+      "grad_norm": 0.7097445130348206,
+      "learning_rate": 9.9997572151645e-05,
+      "loss": 0.0404,
+      "step": 2650
+    },
+    {
+      "grad_norm": 0.5693011283874512,
+      "learning_rate": 9.999723533330254e-05,
+      "loss": 0.0392,
+      "step": 2660
+    },
+    {
+      "grad_norm": 0.5302420854568481,
+      "learning_rate": 9.999687664447074e-05,
+      "loss": 0.0393,
+      "step": 2670
+    },
+    {
+      "grad_norm": 0.4600958526134491,
+      "learning_rate": 9.99964960853065e-05,
+      "loss": 0.0384,
+      "step": 2680
+    },
+    {
+      "grad_norm": 0.46369120478630066,
+      "learning_rate": 9.999609365597627e-05,
+      "loss": 0.0401,
+      "step": 2690
+    },
+    {
+      "grad_norm": 0.5673149824142456,
+      "learning_rate": 9.99956693566561e-05,
+      "loss": 0.0366,
+      "step": 2700
+    },
+    {
+      "grad_norm": 0.7230439782142639,
+      "learning_rate": 9.99952231875316e-05,
+      "loss": 0.0368,
+      "step": 2710
+    },
+    {
+      "grad_norm": 0.629014253616333,
+      "learning_rate": 9.999475514879795e-05,
+      "loss": 0.0386,
+      "step": 2720
+    },
+    {
+      "grad_norm": 0.6077961325645447,
+      "learning_rate": 9.999426524065984e-05,
+      "loss": 0.0396,
+      "step": 2730
+    },
+    {
+      "grad_norm": 0.5505583882331848,
+      "learning_rate": 9.999375346333162e-05,
+      "loss": 0.0364,
+      "step": 2740
+    },
+    {
+      "grad_norm": 0.6141350865364075,
+      "learning_rate": 9.999321981703715e-05,
+      "loss": 0.0345,
+      "step": 2750
+    },
+    {
+      "grad_norm": 0.4496050179004669,
+      "learning_rate": 9.999266430200985e-05,
+      "loss": 0.0356,
+      "step": 2760
+    },
+    {
+      "grad_norm": 0.6305267810821533,
+      "learning_rate": 9.999208691849271e-05,
+      "loss": 0.0384,
+      "step": 2770
+    },
+    {
+      "grad_norm": 0.4410249590873718,
+      "learning_rate": 9.999148766673832e-05,
+      "loss": 0.0336,
+      "step": 2780
+    },
+    {
+      "grad_norm": 0.4349486827850342,
+      "learning_rate": 9.999086654700881e-05,
+      "loss": 0.033,
+      "step": 2790
+    },
+    {
+      "grad_norm": 0.49041080474853516,
+      "learning_rate": 9.999022355957588e-05,
+      "loss": 0.0358,
+      "step": 2800
+    },
+    {
+      "grad_norm": 0.4800823926925659,
+      "learning_rate": 9.998955870472079e-05,
+      "loss": 0.037,
+      "step": 2810
+    },
+    {
+      "grad_norm": 0.524013340473175,
+      "learning_rate": 9.998887198273437e-05,
+      "loss": 0.0355,
+      "step": 2820
+    },
+    {
+      "grad_norm": 0.593246579170227,
+      "learning_rate": 9.998816339391701e-05,
+      "loss": 0.0321,
+      "step": 2830
+    },
+    {
+      "grad_norm": 0.42245444655418396,
+      "learning_rate": 9.998743293857868e-05,
+      "loss": 0.0335,
+      "step": 2840
+    },
+    {
+      "grad_norm": 0.4688955843448639,
+      "learning_rate": 9.998668061703891e-05,
+      "loss": 0.0342,
+      "step": 2850
+    },
+    {
+      "grad_norm": 0.603750467300415,
+      "learning_rate": 9.998590642962679e-05,
+      "loss": 0.039,
+      "step": 2860
+    },
+    {
+      "grad_norm": 0.37352561950683594,
+      "learning_rate": 9.998511037668095e-05,
+      "loss": 0.0366,
+      "step": 2870
+    },
+    {
+      "grad_norm": 0.5140548944473267,
+      "learning_rate": 9.998429245854964e-05,
+      "loss": 0.0328,
+      "step": 2880
+    },
+    {
+      "grad_norm": 0.5762473940849304,
+      "learning_rate": 9.998345267559064e-05,
+      "loss": 0.0356,
+      "step": 2890
+    },
+    {
+      "grad_norm": 0.4723564088344574,
+      "learning_rate": 9.998259102817129e-05,
+      "loss": 0.0322,
+      "step": 2900
+    },
+    {
+      "grad_norm": 0.45071810483932495,
+      "learning_rate": 9.99817075166685e-05,
+      "loss": 0.0312,
+      "step": 2910
+    },
+    {
+      "grad_norm": 0.545811653137207,
+      "learning_rate": 9.998080214146878e-05,
+      "loss": 0.0351,
+      "step": 2920
+    },
+    {
+      "grad_norm": 0.4992588460445404,
+      "learning_rate": 9.997987490296813e-05,
+      "loss": 0.0364,
+      "step": 2930
+    },
+    {
+      "grad_norm": 0.5212619304656982,
+      "learning_rate": 9.99789258015722e-05,
+      "loss": 0.0378,
+      "step": 2940
+    },
+    {
+      "grad_norm": 0.5206537246704102,
+      "learning_rate": 9.997795483769611e-05,
+      "loss": 0.0332,
+      "step": 2950
+    },
+    {
+      "grad_norm": 0.4801543653011322,
+      "learning_rate": 9.997696201176462e-05,
+      "loss": 0.0343,
+      "step": 2960
+    },
+    {
+      "grad_norm": 0.46418094635009766,
+      "learning_rate": 9.997594732421203e-05,
+      "loss": 0.0309,
+      "step": 2970
+    },
+    {
+      "grad_norm": 0.6558750867843628,
+      "learning_rate": 9.997491077548217e-05,
+      "loss": 0.0295,
+      "step": 2980
+    },
+    {
+      "grad_norm": 0.3945139944553375,
+      "learning_rate": 9.997385236602851e-05,
+      "loss": 0.0308,
+      "step": 2990
+    },
+    {
+      "grad_norm": 0.4074639081954956,
+      "learning_rate": 9.997277209631399e-05,
+      "loss": 0.031,
+      "step": 3000
+    },
+    {
+      "grad_norm": 0.31962695717811584,
+      "learning_rate": 9.997166996681118e-05,
+      "loss": 0.0294,
+      "step": 3010
+    },
+    {
+      "grad_norm": 0.5200231075286865,
+      "learning_rate": 9.997054597800218e-05,
+      "loss": 0.0308,
+      "step": 3020
+    },
+    {
+      "grad_norm": 0.6230039000511169,
+      "learning_rate": 9.996940013037866e-05,
+      "loss": 0.0292,
+      "step": 3030
+    },
+    {
+      "grad_norm": 0.4674220085144043,
+      "learning_rate": 9.996823242444186e-05,
+      "loss": 0.0326,
+      "step": 3040
+    },
+    {
+      "grad_norm": 0.49110841751098633,
+      "learning_rate": 9.996704286070258e-05,
+      "loss": 0.029,
+      "step": 3050
+    },
+    {
+      "grad_norm": 0.4353705644607544,
+      "learning_rate": 9.996583143968115e-05,
+      "loss": 0.0299,
+      "step": 3060
+    },
+    {
+      "grad_norm": 0.4817028045654297,
+      "learning_rate": 9.99645981619075e-05,
+      "loss": 0.0298,
+      "step": 3070
+    },
+    {
+      "grad_norm": 0.5148823261260986,
+      "learning_rate": 9.996334302792114e-05,
+      "loss": 0.0313,
+      "step": 3080
+    },
+    {
+      "grad_norm": 0.5206035375595093,
+      "learning_rate": 9.996206603827105e-05,
+      "loss": 0.032,
+      "step": 3090
+    },
+    {
+      "grad_norm": 0.398294597864151,
+      "learning_rate": 9.996076719351587e-05,
+      "loss": 0.0324,
+      "step": 3100
+    },
+    {
+      "grad_norm": 0.5408644676208496,
+      "learning_rate": 9.995944649422374e-05,
+      "loss": 0.0299,
+      "step": 3110
+    },
+    {
+      "grad_norm": 0.6455337405204773,
+      "learning_rate": 9.995810394097239e-05,
+      "loss": 0.0312,
+      "step": 3120
+    },
+    {
+      "grad_norm": 0.5284717082977295,
+      "learning_rate": 9.995673953434909e-05,
+      "loss": 0.0332,
+      "step": 3130
+    },
+    {
+      "grad_norm": 0.5388599038124084,
+      "learning_rate": 9.995535327495068e-05,
+      "loss": 0.0315,
+      "step": 3140
+    },
+    {
+      "grad_norm": 0.6507496237754822,
+      "learning_rate": 9.995394516338355e-05,
+      "loss": 0.0301,
+      "step": 3150
+    },
+    {
+      "grad_norm": 0.5397722125053406,
+      "learning_rate": 9.995251520026367e-05,
+      "loss": 0.0289,
+      "step": 3160
+    },
+    {
+      "grad_norm": 0.3745012879371643,
+      "learning_rate": 9.995106338621656e-05,
+      "loss": 0.0292,
+      "step": 3170
+    },
+    {
+      "grad_norm": 0.6118871569633484,
+      "learning_rate": 9.994958972187726e-05,
+      "loss": 0.028,
+      "step": 3180
+    },
+    {
+      "grad_norm": 0.4704694151878357,
+      "learning_rate": 9.994809420789044e-05,
+      "loss": 0.0299,
+      "step": 3190
+    },
+    {
+      "grad_norm": 0.5749391913414001,
+      "learning_rate": 9.994657684491027e-05,
+      "loss": 0.0301,
+      "step": 3200
+    },
+    {
+      "grad_norm": 0.3724566400051117,
+      "learning_rate": 9.994503763360048e-05,
+      "loss": 0.0308,
+      "step": 3210
+    },
+    {
+      "grad_norm": 0.4805312156677246,
+      "learning_rate": 9.99434765746344e-05,
+      "loss": 0.0281,
+      "step": 3220
+    },
+    {
+      "grad_norm": 0.43801966309547424,
+      "learning_rate": 9.994189366869488e-05,
+      "loss": 0.0261,
+      "step": 3230
+    },
+    {
+      "grad_norm": 0.4741677939891815,
+      "learning_rate": 9.994028891647433e-05,
+      "loss": 0.0302,
+      "step": 3240
+    },
+    {
+      "grad_norm": 0.4064734876155853,
+      "learning_rate": 9.993866231867475e-05,
+      "loss": 0.0278,
+      "step": 3250
+    },
+    {
+      "grad_norm": 0.41089650988578796,
+      "learning_rate": 9.993701387600762e-05,
+      "loss": 0.0322,
+      "step": 3260
+    },
+    {
+      "grad_norm": 0.4482851028442383,
+      "learning_rate": 9.993534358919408e-05,
+      "loss": 0.0299,
+      "step": 3270
+    },
+    {
+      "grad_norm": 0.5010986328125,
+      "learning_rate": 9.993365145896473e-05,
+      "loss": 0.0303,
+      "step": 3280
+    },
+    {
+      "grad_norm": 0.4836841821670532,
+      "learning_rate": 9.993193748605977e-05,
+      "loss": 0.0355,
+      "step": 3290
+    },
+    {
+      "grad_norm": 0.45244765281677246,
+      "learning_rate": 9.993020167122898e-05,
+      "loss": 0.0287,
+      "step": 3300
+    },
+    {
+      "grad_norm": 0.4268089234828949,
+      "learning_rate": 9.992844401523164e-05,
+      "loss": 0.0273,
+      "step": 3310
+    },
+    {
+      "grad_norm": 0.4934121072292328,
+      "learning_rate": 9.992666451883661e-05,
+      "loss": 0.0303,
+      "step": 3320
+    },
+    {
+      "grad_norm": 0.5121903419494629,
+      "learning_rate": 9.99248631828223e-05,
+      "loss": 0.0275,
+      "step": 3330
+    },
+    {
+      "grad_norm": 0.5032432675361633,
+      "learning_rate": 9.99230400079767e-05,
+      "loss": 0.0297,
+      "step": 3340
+    },
+    {
+      "grad_norm": 0.47000205516815186,
+      "learning_rate": 9.992119499509728e-05,
+      "loss": 0.0287,
+      "step": 3350
+    },
+    {
+      "grad_norm": 0.4018782079219818,
+      "learning_rate": 9.991932814499114e-05,
+      "loss": 0.0289,
+      "step": 3360
+    },
+    {
+      "grad_norm": 0.5098743438720703,
+      "learning_rate": 9.991743945847493e-05,
+      "loss": 0.032,
+      "step": 3370
+    },
+    {
+      "grad_norm": 0.529468834400177,
+      "learning_rate": 9.991552893637478e-05,
+      "loss": 0.0274,
+      "step": 3380
+    },
+    {
+      "grad_norm": 0.4693675935268402,
+      "learning_rate": 9.991359657952644e-05,
+      "loss": 0.0264,
+      "step": 3390
+    },
+    {
+      "grad_norm": 0.637904167175293,
+      "learning_rate": 9.991164238877519e-05,
+      "loss": 0.0326,
+      "step": 3400
+    },
+    {
+      "grad_norm": 0.43420007824897766,
+      "learning_rate": 9.990966636497585e-05,
+      "loss": 0.0331,
+      "step": 3410
+    },
+    {
+      "grad_norm": 0.4679301977157593,
+      "learning_rate": 9.99076685089928e-05,
+      "loss": 0.0312,
+      "step": 3420
+    },
+    {
+      "grad_norm": 0.4944802522659302,
+      "learning_rate": 9.990564882169998e-05,
+      "loss": 0.0318,
+      "step": 3430
+    },
+    {
+      "grad_norm": 0.43934884667396545,
+      "learning_rate": 9.990360730398088e-05,
+      "loss": 0.0352,
+      "step": 3440
+    },
+    {
+      "grad_norm": 0.4320874810218811,
+      "learning_rate": 9.990154395672849e-05,
+      "loss": 0.0293,
+      "step": 3450
+    },
+    {
+      "grad_norm": 0.4531351923942566,
+      "learning_rate": 9.989945878084541e-05,
+      "loss": 0.0326,
+      "step": 3460
+    },
+    {
+      "grad_norm": 0.45614591240882874,
+      "learning_rate": 9.989735177724378e-05,
+      "loss": 0.0257,
+      "step": 3470
+    },
+    {
+      "grad_norm": 0.4763658940792084,
+      "learning_rate": 9.989522294684526e-05,
+      "loss": 0.032,
+      "step": 3480
+    },
+    {
+      "grad_norm": 0.4897955358028412,
+      "learning_rate": 9.989307229058107e-05,
+      "loss": 0.0278,
+      "step": 3490
+    },
+    {
+      "grad_norm": 0.4496730864048004,
+      "learning_rate": 9.989089980939202e-05,
+      "loss": 0.0288,
+      "step": 3500
+    },
+    {
+      "grad_norm": 0.416805237531662,
+      "learning_rate": 9.988870550422835e-05,
+      "loss": 0.0266,
+      "step": 3510
+    },
+    {
+      "grad_norm": 0.45189645886421204,
+      "learning_rate": 9.988648937604999e-05,
+      "loss": 0.0299,
+      "step": 3520
+    },
+    {
+      "grad_norm": 0.5110573768615723,
+      "learning_rate": 9.988425142582632e-05,
+      "loss": 0.0318,
+      "step": 3530
+    },
+    {
+      "grad_norm": 0.4993090033531189,
+      "learning_rate": 9.98819916545363e-05,
+      "loss": 0.0281,
+      "step": 3540
+    },
+    {
+      "grad_norm": 0.5170537233352661,
+      "learning_rate": 9.987971006316844e-05,
+      "loss": 0.0272,
+      "step": 3550
+    },
+    {
+      "grad_norm": 0.46066486835479736,
+      "learning_rate": 9.987740665272077e-05,
+      "loss": 0.0289,
+      "step": 3560
+    },
+    {
+      "grad_norm": 0.49337151646614075,
+      "learning_rate": 9.98750814242009e-05,
+      "loss": 0.0281,
+      "step": 3570
+    },
+    {
+      "grad_norm": 0.4190320670604706,
+      "learning_rate": 9.987273437862594e-05,
+      "loss": 0.0303,
+      "step": 3580
+    },
+    {
+      "grad_norm": 0.3774312138557434,
+      "learning_rate": 9.987036551702259e-05,
+      "loss": 0.0268,
+      "step": 3590
+    },
+    {
+      "grad_norm": 0.4794900715351105,
+      "learning_rate": 9.986797484042706e-05,
+      "loss": 0.0291,
+      "step": 3600
+    },
+    {
+      "grad_norm": 0.5011126399040222,
+      "learning_rate": 9.986556234988512e-05,
+      "loss": 0.0269,
+      "step": 3610
+    },
+    {
+      "grad_norm": 0.38879141211509705,
+      "learning_rate": 9.986312804645205e-05,
+      "loss": 0.025,
+      "step": 3620
+    },
+    {
+      "grad_norm": 0.5543105006217957,
+      "learning_rate": 9.986067193119273e-05,
+      "loss": 0.0263,
+      "step": 3630
+    },
+    {
+      "grad_norm": 0.37508076429367065,
+      "learning_rate": 9.985819400518153e-05,
+      "loss": 0.0257,
+      "step": 3640
+    },
+    {
+      "grad_norm": 0.4521786868572235,
+      "learning_rate": 9.985569426950239e-05,
+      "loss": 0.027,
+      "step": 3650
+    },
+    {
+      "grad_norm": 0.505350649356842,
+      "learning_rate": 9.985317272524876e-05,
+      "loss": 0.0274,
+      "step": 3660
+    },
+    {
+      "grad_norm": 0.4197271168231964,
+      "learning_rate": 9.985062937352366e-05,
+      "loss": 0.0269,
+      "step": 3670
+    },
+    {
+      "grad_norm": 0.3423755466938019,
+      "learning_rate": 9.984806421543966e-05,
+      "loss": 0.0239,
+      "step": 3680
+    },
+    {
+      "grad_norm": 0.45793643593788147,
+      "learning_rate": 9.984547725211881e-05,
+      "loss": 0.0255,
+      "step": 3690
+    },
+    {
+      "grad_norm": 0.5920965075492859,
+      "learning_rate": 9.984286848469276e-05,
+      "loss": 0.0271,
+      "step": 3700
+    },
+    {
+      "grad_norm": 0.4791988432407379,
+      "learning_rate": 9.984023791430266e-05,
+      "loss": 0.0259,
+      "step": 3710
+    },
+    {
+      "grad_norm": 0.4176543354988098,
+      "learning_rate": 9.983758554209924e-05,
+      "loss": 0.0271,
+      "step": 3720
+    },
+    {
+      "grad_norm": 0.498963326215744,
+      "learning_rate": 9.983491136924268e-05,
+      "loss": 0.0261,
+      "step": 3730
+    },
+    {
+      "grad_norm": 0.28769829869270325,
+      "learning_rate": 9.983221539690282e-05,
+      "loss": 0.0264,
+      "step": 3740
+    },
+    {
+      "grad_norm": 0.4772230088710785,
+      "learning_rate": 9.982949762625892e-05,
+      "loss": 0.0269,
+      "step": 3750
+    },
+    {
+      "grad_norm": 0.48311951756477356,
+      "learning_rate": 9.982675805849986e-05,
+      "loss": 0.026,
+      "step": 3760
+    },
+    {
+      "grad_norm": 0.4668444097042084,
+      "learning_rate": 9.982399669482399e-05,
+      "loss": 0.0261,
+      "step": 3770
+    },
+    {
+      "grad_norm": 0.34021511673927307,
+      "learning_rate": 9.982121353643924e-05,
+      "loss": 0.0229,
+      "step": 3780
+    },
+    {
+      "grad_norm": 0.4771108627319336,
+      "learning_rate": 9.981840858456306e-05,
+      "loss": 0.0265,
+      "step": 3790
+    },
+    {
+      "grad_norm": 0.4224259555339813,
+      "learning_rate": 9.981558184042243e-05,
+      "loss": 0.0271,
+      "step": 3800
+    },
+    {
+      "grad_norm": 0.41765841841697693,
+      "learning_rate": 9.981273330525387e-05,
+      "loss": 0.0233,
+      "step": 3810
+    },
+    {
+      "grad_norm": 0.3940288722515106,
+      "learning_rate": 9.980986298030341e-05,
+      "loss": 0.0243,
+      "step": 3820
+    },
+    {
+      "grad_norm": 0.30766552686691284,
+      "learning_rate": 9.980697086682662e-05,
+      "loss": 0.0264,
+      "step": 3830
+    },
+    {
+      "grad_norm": 0.4462141692638397,
+      "learning_rate": 9.980405696608866e-05,
+      "loss": 0.0248,
+      "step": 3840
+    },
+    {
+      "grad_norm": 0.4133457541465759,
+      "learning_rate": 9.980112127936409e-05,
+      "loss": 0.0248,
+      "step": 3850
+    },
+    {
+      "grad_norm": 0.4097680449485779,
+      "learning_rate": 9.979816380793717e-05,
+      "loss": 0.0251,
+      "step": 3860
+    },
+    {
+      "grad_norm": 0.39771568775177,
+      "learning_rate": 9.979518455310151e-05,
+      "loss": 0.0259,
+      "step": 3870
+    },
+    {
+      "grad_norm": 0.4057018458843231,
+      "learning_rate": 9.97921835161604e-05,
+      "loss": 0.0247,
+      "step": 3880
+    },
+    {
+      "grad_norm": 0.43093815445899963,
+      "learning_rate": 9.978916069842656e-05,
+      "loss": 0.0236,
+      "step": 3890
+    },
+    {
+      "grad_norm": 0.39105674624443054,
+      "learning_rate": 9.97861161012223e-05,
+      "loss": 0.0255,
+      "step": 3900
+    },
+    {
+      "grad_norm": 0.396752268075943,
+      "learning_rate": 9.978304972587942e-05,
+      "loss": 0.0293,
+      "step": 3910
+    },
+    {
+      "grad_norm": 0.4894019663333893,
+      "learning_rate": 9.977996157373925e-05,
+      "loss": 0.0307,
+      "step": 3920
+    },
+    {
+      "grad_norm": 0.32372811436653137,
+      "learning_rate": 9.977685164615265e-05,
+      "loss": 0.024,
+      "step": 3930
+    },
+    {
+      "grad_norm": 0.5265254974365234,
+      "learning_rate": 9.977371994448002e-05,
+      "loss": 0.026,
+      "step": 3940
+    },
+    {
+      "grad_norm": 0.47224506735801697,
+      "learning_rate": 9.977056647009127e-05,
+      "loss": 0.0246,
+      "step": 3950
+    },
+    {
+      "grad_norm": 0.40714237093925476,
+      "learning_rate": 9.976739122436582e-05,
+      "loss": 0.0284,
+      "step": 3960
+    },
+    {
+      "grad_norm": 0.439060777425766,
+      "learning_rate": 9.976419420869265e-05,
+      "loss": 0.0265,
+      "step": 3970
+    },
+    {
+      "grad_norm": 0.39264538884162903,
+      "learning_rate": 9.976097542447025e-05,
+      "loss": 0.033,
+      "step": 3980
+    },
+    {
+      "grad_norm": 0.4365125894546509,
+      "learning_rate": 9.97577348731066e-05,
+      "loss": 0.0315,
+      "step": 3990
+    },
+    {
+      "grad_norm": 0.3404941260814667,
+      "learning_rate": 9.975447255601927e-05,
+      "loss": 0.0259,
+      "step": 4000
+    },
+    {
+      "grad_norm": 0.38505586981773376,
+      "learning_rate": 9.975118847463525e-05,
+      "loss": 0.0266,
+      "step": 4010
+    },
+    {
+      "grad_norm": 0.3946306109428406,
+      "learning_rate": 9.974788263039114e-05,
+      "loss": 0.0251,
+      "step": 4020
+    },
+    {
+      "grad_norm": 0.2998659908771515,
+      "learning_rate": 9.974455502473303e-05,
+      "loss": 0.0249,
+      "step": 4030
+    },
+    {
+      "grad_norm": 0.38948115706443787,
+      "learning_rate": 9.974120565911652e-05,
+      "loss": 0.0242,
+      "step": 4040
+    },
+    {
+      "grad_norm": 0.31953033804893494,
+      "learning_rate": 9.973783453500674e-05,
+      "loss": 0.0228,
+      "step": 4050
+    },
+    {
+      "grad_norm": 0.4850027561187744,
+      "learning_rate": 9.973444165387835e-05,
+      "loss": 0.0253,
+      "step": 4060
+    },
+    {
+      "grad_norm": 0.5223883390426636,
+      "learning_rate": 9.973102701721549e-05,
+      "loss": 0.0236,
+      "step": 4070
+    },
+    {
+      "grad_norm": 0.3267204463481903,
+      "learning_rate": 9.972759062651184e-05,
+      "loss": 0.0248,
+      "step": 4080
+    },
+    {
+      "grad_norm": 0.3633793294429779,
+      "learning_rate": 9.972413248327059e-05,
+      "loss": 0.0259,
+      "step": 4090
+    },
+    {
+      "grad_norm": 0.4617072343826294,
+      "learning_rate": 9.972065258900447e-05,
+      "loss": 0.0242,
+      "step": 4100
+    },
+    {
+      "grad_norm": 0.40263453125953674,
+      "learning_rate": 9.971715094523569e-05,
+      "loss": 0.0245,
+      "step": 4110
+    },
+    {
+      "grad_norm": 0.37316393852233887,
+      "learning_rate": 9.971362755349598e-05,
+      "loss": 0.0236,
+      "step": 4120
+    },
+    {
+      "grad_norm": 0.3545263409614563,
+      "learning_rate": 9.971008241532662e-05,
+      "loss": 0.0224,
+      "step": 4130
+    },
+    {
+      "grad_norm": 0.4703061580657959,
+      "learning_rate": 9.970651553227835e-05,
+      "loss": 0.0274,
+      "step": 4140
+    },
+    {
+      "grad_norm": 0.325410395860672,
+      "learning_rate": 9.970292690591143e-05,
+      "loss": 0.023,
+      "step": 4150
+    },
+    {
+      "grad_norm": 0.380282998085022,
+      "learning_rate": 9.969931653779569e-05,
+      "loss": 0.0258,
+      "step": 4160
+    },
+    {
+      "grad_norm": 0.3761638104915619,
+      "learning_rate": 9.969568442951038e-05,
+      "loss": 0.023,
+      "step": 4170
+    },
+    {
+      "grad_norm": 0.4377315938472748,
+      "learning_rate": 9.969203058264436e-05,
+      "loss": 0.0234,
+      "step": 4180
+    },
+    {
+      "grad_norm": 0.44106921553611755,
+      "learning_rate": 9.96883549987959e-05,
+      "loss": 0.0211,
+      "step": 4190
+    },
+    {
+      "grad_norm": 0.33367282152175903,
+      "learning_rate": 9.968465767957287e-05,
+      "loss": 0.0222,
+      "step": 4200
+    },
+    {
+      "grad_norm": 0.5206263661384583,
+      "learning_rate": 9.968093862659256e-05,
+      "loss": 0.0224,
+      "step": 4210
+    },
+    {
+      "grad_norm": 0.437534898519516,
+      "learning_rate": 9.967719784148182e-05,
+      "loss": 0.0225,
+      "step": 4220
+    },
+    {
+      "grad_norm": 0.3489209711551666,
+      "learning_rate": 9.967343532587702e-05,
+      "loss": 0.0217,
+      "step": 4230
+    },
+    {
+      "grad_norm": 0.43395212292671204,
+      "learning_rate": 9.966965108142399e-05,
+      "loss": 0.0253,
+      "step": 4240
+    },
+    {
+      "grad_norm": 0.4697989821434021,
+      "learning_rate": 9.96658451097781e-05,
+      "loss": 0.0241,
+      "step": 4250
+    },
+    {
+      "grad_norm": 0.4404914379119873,
+      "learning_rate": 9.966201741260419e-05,
+      "loss": 0.0252,
+      "step": 4260
+    },
+    {
+      "grad_norm": 0.38549482822418213,
+      "learning_rate": 9.965816799157665e-05,
+      "loss": 0.0225,
+      "step": 4270
+    },
+    {
+      "grad_norm": 0.35402536392211914,
+      "learning_rate": 9.965429684837935e-05,
+      "loss": 0.0243,
+      "step": 4280
+    },
+    {
+      "grad_norm": 0.38654282689094543,
+      "learning_rate": 9.965040398470562e-05,
+      "loss": 0.0239,
+      "step": 4290
+    },
+    {
+      "grad_norm": 0.4091793894767761,
+      "learning_rate": 9.964648940225838e-05,
+      "loss": 0.0234,
+      "step": 4300
+    },
+    {
+      "grad_norm": 0.36413222551345825,
+      "learning_rate": 9.964255310274997e-05,
+      "loss": 0.0247,
+      "step": 4310
+    },
+    {
+      "grad_norm": 0.421766072511673,
+      "learning_rate": 9.963859508790228e-05,
+      "loss": 0.022,
+      "step": 4320
+    },
+    {
+      "grad_norm": 0.432120144367218,
+      "learning_rate": 9.963461535944664e-05,
+      "loss": 0.0248,
+      "step": 4330
+    },
+    {
+      "grad_norm": 0.5248905420303345,
+      "learning_rate": 9.963061391912399e-05,
+      "loss": 0.0258,
+      "step": 4340
+    },
+    {
+      "grad_norm": 0.4155650734901428,
+      "learning_rate": 9.962659076868463e-05,
+      "loss": 0.0232,
+      "step": 4350
+    },
+    {
+      "grad_norm": 0.35478758811950684,
+      "learning_rate": 9.962254590988846e-05,
+      "loss": 0.021,
+      "step": 4360
+    },
+    {
+      "grad_norm": 0.4828854203224182,
+      "learning_rate": 9.961847934450481e-05,
+      "loss": 0.0234,
+      "step": 4370
+    },
+    {
+      "grad_norm": 0.3365130126476288,
+      "learning_rate": 9.961439107431257e-05,
+      "loss": 0.0242,
+      "step": 4380
+    },
+    {
+      "grad_norm": 0.4112580716609955,
+      "learning_rate": 9.961028110110006e-05,
+      "loss": 0.0227,
+      "step": 4390
+    },
+    {
+      "grad_norm": 0.47759896516799927,
+      "learning_rate": 9.960614942666513e-05,
+      "loss": 0.0249,
+      "step": 4400
+    },
+    {
+      "grad_norm": 0.5083246231079102,
+      "learning_rate": 9.960199605281511e-05,
+      "loss": 0.0269,
+      "step": 4410
+    },
+    {
+      "grad_norm": 0.39654260873794556,
+      "learning_rate": 9.959782098136683e-05,
+      "loss": 0.0219,
+      "step": 4420
+    },
+    {
+      "grad_norm": 0.40617844462394714,
+      "learning_rate": 9.959362421414662e-05,
+      "loss": 0.0247,
+      "step": 4430
+    },
+    {
+      "grad_norm": 0.3405906558036804,
+      "learning_rate": 9.958940575299027e-05,
+      "loss": 0.0217,
+      "step": 4440
+    },
+    {
+      "grad_norm": 0.33479437232017517,
+      "learning_rate": 9.95851655997431e-05,
+      "loss": 0.0236,
+      "step": 4450
+    },
+    {
+      "grad_norm": 0.3753277361392975,
+      "learning_rate": 9.958090375625986e-05,
+      "loss": 0.0201,
+      "step": 4460
+    },
+    {
+      "grad_norm": 0.316630482673645,
+      "learning_rate": 9.957662022440486e-05,
+      "loss": 0.0196,
+      "step": 4470
+    },
+    {
+      "grad_norm": 0.33641859889030457,
+      "learning_rate": 9.957231500605187e-05,
+      "loss": 0.0233,
+      "step": 4480
+    },
+    {
+      "grad_norm": 0.4224380552768707,
+      "learning_rate": 9.95679881030841e-05,
+      "loss": 0.021,
+      "step": 4490
+    },
+    {
+      "grad_norm": 0.42875245213508606,
+      "learning_rate": 9.95636395173943e-05,
+      "loss": 0.0244,
+      "step": 4500
+    },
+    {
+      "grad_norm": 0.4839310646057129,
+      "learning_rate": 9.95592692508847e-05,
+      "loss": 0.0206,
+      "step": 4510
+    },
+    {
+      "grad_norm": 0.3010939657688141,
+      "learning_rate": 9.9554877305467e-05,
+      "loss": 0.0203,
+      "step": 4520
+    },
+    {
+      "grad_norm": 0.31267791986465454,
+      "learning_rate": 9.955046368306237e-05,
+      "loss": 0.021,
+      "step": 4530
+    },
+    {
+      "grad_norm": 0.2783145010471344,
+      "learning_rate": 9.954602838560153e-05,
+      "loss": 0.0211,
+      "step": 4540
+    },
+    {
+      "grad_norm": 0.32712653279304504,
+      "learning_rate": 9.954157141502456e-05,
+      "loss": 0.0213,
+      "step": 4550
+    },
+    {
+      "grad_norm": 0.3944172263145447,
+      "learning_rate": 9.953709277328112e-05,
+      "loss": 0.0233,
+      "step": 4560
+    },
+    {
+      "grad_norm": 0.3710629642009735,
+      "learning_rate": 9.953259246233032e-05,
+      "loss": 0.0186,
+      "step": 4570
+    },
+    {
+      "grad_norm": 0.3196753263473511,
+      "learning_rate": 9.952807048414077e-05,
+      "loss": 0.0225,
+      "step": 4580
+    },
+    {
+      "grad_norm": 0.4878087341785431,
+      "learning_rate": 9.95235268406905e-05,
+      "loss": 0.0241,
+      "step": 4590
+    },
+    {
+      "grad_norm": 0.2757819890975952,
+      "learning_rate": 9.951896153396708e-05,
+      "loss": 0.0214,
+      "step": 4600
+    },
+    {
+      "grad_norm": 0.3346608281135559,
+      "learning_rate": 9.95143745659675e-05,
+      "loss": 0.0231,
+      "step": 4610
+    },
+    {
+      "grad_norm": 0.4076615273952484,
+      "learning_rate": 9.95097659386983e-05,
+      "loss": 0.0209,
+      "step": 4620
+    },
+    {
+      "grad_norm": 0.383935809135437,
+      "learning_rate": 9.950513565417542e-05,
+      "loss": 0.0225,
+      "step": 4630
+    },
+    {
+      "grad_norm": 0.34777235984802246,
+      "learning_rate": 9.95004837144243e-05,
+      "loss": 0.0205,
+      "step": 4640
+    },
+    {
+      "grad_norm": 0.3899579346179962,
+      "learning_rate": 9.949581012147988e-05,
+      "loss": 0.0216,
+      "step": 4650
+    },
+    {
+      "grad_norm": 0.451011598110199,
+      "learning_rate": 9.949111487738653e-05,
+      "loss": 0.0215,
+      "step": 4660
+    },
+    {
+      "grad_norm": 0.35339757800102234,
+      "learning_rate": 9.948639798419813e-05,
+      "loss": 0.0221,
+      "step": 4670
+    },
+    {
+      "grad_norm": 0.3547573387622833,
+      "learning_rate": 9.948165944397799e-05,
+      "loss": 0.0206,
+      "step": 4680
+    },
+    {
+      "grad_norm": 0.2891366481781006,
+      "learning_rate": 9.94768992587989e-05,
+      "loss": 0.0244,
+      "step": 4690
+    },
+    {
+      "grad_norm": 0.35341107845306396,
+      "learning_rate": 9.947211743074313e-05,
+      "loss": 0.0201,
+      "step": 4700
+    },
+    {
+      "grad_norm": 0.35320010781288147,
+      "learning_rate": 9.946731396190246e-05,
+      "loss": 0.0203,
+      "step": 4710
+    },
+    {
+      "grad_norm": 0.41338032484054565,
+      "learning_rate": 9.946248885437803e-05,
+      "loss": 0.0225,
+      "step": 4720
+    },
+    {
+      "grad_norm": 0.403791218996048,
+      "learning_rate": 9.945764211028053e-05,
+      "loss": 0.0216,
+      "step": 4730
+    },
+    {
+      "grad_norm": 0.3111172914505005,
+      "learning_rate": 9.94527737317301e-05,
+      "loss": 0.0203,
+      "step": 4740
+    },
+    {
+      "grad_norm": 0.37261950969696045,
+      "learning_rate": 9.944788372085631e-05,
+      "loss": 0.0223,
+      "step": 4750
+    },
+    {
+      "grad_norm": 0.31674131751060486,
+      "learning_rate": 9.944297207979825e-05,
+      "loss": 0.022,
+      "step": 4760
+    },
+    {
+      "grad_norm": 0.2943297326564789,
+      "learning_rate": 9.943803881070441e-05,
+      "loss": 0.0222,
+      "step": 4770
+    },
+    {
+      "grad_norm": 0.32278957962989807,
+      "learning_rate": 9.943308391573278e-05,
+      "loss": 0.0229,
+      "step": 4780
+    },
+    {
+      "grad_norm": 0.34058910608291626,
+      "learning_rate": 9.942810739705079e-05,
+      "loss": 0.0228,
+      "step": 4790
+    },
+    {
+      "grad_norm": 0.38450783491134644,
+      "learning_rate": 9.942310925683538e-05,
+      "loss": 0.0201,
+      "step": 4800
+    },
+    {
+      "grad_norm": 0.3569624125957489,
+      "learning_rate": 9.941808949727285e-05,
+      "loss": 0.0242,
+      "step": 4810
+    },
+    {
+      "grad_norm": 0.3103424608707428,
+      "learning_rate": 9.941304812055903e-05,
+      "loss": 0.0235,
+      "step": 4820
+    },
+    {
+      "grad_norm": 0.455632746219635,
+      "learning_rate": 9.940798512889921e-05,
+      "loss": 0.0241,
+      "step": 4830
+    },
+    {
+      "grad_norm": 0.42037898302078247,
+      "learning_rate": 9.94029005245081e-05,
+      "loss": 0.0233,
+      "step": 4840
+    },
+    {
+      "grad_norm": 0.2653869390487671,
+      "learning_rate": 9.939779430960988e-05,
+      "loss": 0.0196,
+      "step": 4850
+    },
+    {
+      "grad_norm": 0.2665639817714691,
+      "learning_rate": 9.939266648643817e-05,
+      "loss": 0.0231,
+      "step": 4860
+    },
+    {
+      "grad_norm": 0.4474591910839081,
+      "learning_rate": 9.938751705723607e-05,
+      "loss": 0.0214,
+      "step": 4870
+    },
+    {
+      "grad_norm": 0.37019091844558716,
+      "learning_rate": 9.938234602425613e-05,
+      "loss": 0.02,
+      "step": 4880
+    },
+    {
+      "grad_norm": 0.4162817895412445,
+      "learning_rate": 9.93771533897603e-05,
+      "loss": 0.0203,
+      "step": 4890
+    },
+    {
+      "grad_norm": 0.2828487753868103,
+      "learning_rate": 9.937193915602004e-05,
+      "loss": 0.0202,
+      "step": 4900
+    },
+    {
+      "grad_norm": 0.3802485466003418,
+      "learning_rate": 9.936670332531621e-05,
+      "loss": 0.0197,
+      "step": 4910
+    },
+    {
+      "grad_norm": 0.4110012650489807,
+      "learning_rate": 9.936144589993916e-05,
+      "loss": 0.0239,
+      "step": 4920
+    },
+    {
+      "grad_norm": 0.4395545721054077,
+      "learning_rate": 9.935616688218867e-05,
+      "loss": 0.0244,
+      "step": 4930
+    },
+    {
+      "grad_norm": 0.31748923659324646,
+      "learning_rate": 9.935086627437395e-05,
+      "loss": 0.0213,
+      "step": 4940
+    },
+    {
+      "grad_norm": 0.2936075031757355,
+      "learning_rate": 9.934554407881366e-05,
+      "loss": 0.0246,
+      "step": 4950
+    },
+    {
+      "grad_norm": 0.3678935468196869,
+      "learning_rate": 9.934020029783593e-05,
+      "loss": 0.0235,
+      "step": 4960
+    },
+    {
+      "grad_norm": 0.3598695397377014,
+      "learning_rate": 9.933483493377829e-05,
+      "loss": 0.0212,
+      "step": 4970
+    },
+    {
+      "grad_norm": 0.3960917592048645,
+      "learning_rate": 9.932944798898774e-05,
+      "loss": 0.0247,
+      "step": 4980
+    },
+    {
+      "grad_norm": 0.44564226269721985,
+      "learning_rate": 9.932403946582072e-05,
+      "loss": 0.0238,
+      "step": 4990
+    },
+    {
+      "grad_norm": 0.3851899206638336,
+      "learning_rate": 9.93186093666431e-05,
+      "loss": 0.0251,
+      "step": 5000
+    },
+    {
+      "grad_norm": 0.4190506935119629,
+      "learning_rate": 9.931315769383018e-05,
+      "loss": 0.0227,
+      "step": 5010
+    },
+    {
+      "grad_norm": 0.3737106919288635,
+      "learning_rate": 9.930768444976672e-05,
+      "loss": 0.0246,
+      "step": 5020
+    },
+    {
+      "grad_norm": 0.3981052339076996,
+      "learning_rate": 9.93021896368469e-05,
+      "loss": 0.0223,
+      "step": 5030
+    },
+    {
+      "grad_norm": 0.42110228538513184,
+      "learning_rate": 9.929667325747431e-05,
+      "loss": 0.0225,
+      "step": 5040
+    },
+    {
+      "grad_norm": 0.3276391327381134,
+      "learning_rate": 9.929113531406205e-05,
+      "loss": 0.0206,
+      "step": 5050
+    },
+    {
+      "grad_norm": 0.33060380816459656,
+      "learning_rate": 9.928557580903257e-05,
+      "loss": 0.0217,
+      "step": 5060
+    },
+    {
+      "grad_norm": 0.28709059953689575,
+      "learning_rate": 9.927999474481779e-05,
+      "loss": 0.0196,
+      "step": 5070
+    },
+    {
+      "grad_norm": 0.393949419260025,
+      "learning_rate": 9.927439212385907e-05,
+      "loss": 0.0196,
+      "step": 5080
+    },
+    {
+      "grad_norm": 0.33341261744499207,
+      "learning_rate": 9.926876794860718e-05,
+      "loss": 0.0228,
+      "step": 5090
+    },
+    {
+      "grad_norm": 0.38900285959243774,
+      "learning_rate": 9.926312222152235e-05,
+      "loss": 0.0217,
+      "step": 5100
+    },
+    {
+      "grad_norm": 0.4549901485443115,
+      "learning_rate": 9.925745494507414e-05,
+      "loss": 0.0219,
+      "step": 5110
+    },
+    {
+      "grad_norm": 0.3871377110481262,
+      "learning_rate": 9.925176612174169e-05,
+      "loss": 0.022,
+      "step": 5120
+    },
+    {
+      "grad_norm": 0.46740689873695374,
+      "learning_rate": 9.924605575401346e-05,
+      "loss": 0.0226,
+      "step": 5130
+    },
+    {
+      "grad_norm": 0.417953222990036,
+      "learning_rate": 9.924032384438733e-05,
+      "loss": 0.0249,
+      "step": 5140
+    },
+    {
+      "grad_norm": 0.3330901265144348,
+      "learning_rate": 9.923457039537066e-05,
+      "loss": 0.0214,
+      "step": 5150
+    },
+    {
+      "grad_norm": 0.34471163153648376,
+      "learning_rate": 9.92287954094802e-05,
+      "loss": 0.0224,
+      "step": 5160
+    },
+    {
+      "grad_norm": 0.34382888674736023,
+      "learning_rate": 9.922299888924212e-05,
+      "loss": 0.0197,
+      "step": 5170
+    },
+    {
+      "grad_norm": 0.2844744026660919,
+      "learning_rate": 9.921718083719203e-05,
+      "loss": 0.0195,
+      "step": 5180
+    },
+    {
+      "grad_norm": 0.40530309081077576,
+      "learning_rate": 9.921134125587491e-05,
+      "loss": 0.0229,
+      "step": 5190
+    },
+    {
+      "grad_norm": 0.4107360541820526,
+      "learning_rate": 9.920548014784523e-05,
+      "loss": 0.0209,
+      "step": 5200
+    },
+    {
+      "grad_norm": 0.41395023465156555,
+      "learning_rate": 9.919959751566681e-05,
+      "loss": 0.0212,
+      "step": 5210
+    },
+    {
+      "grad_norm": 0.3510611355304718,
+      "learning_rate": 9.919369336191291e-05,
+      "loss": 0.0213,
+      "step": 5220
+    },
+    {
+      "grad_norm": 0.38708657026290894,
+      "learning_rate": 9.918776768916625e-05,
+      "loss": 0.02,
+      "step": 5230
+    },
+    {
+      "grad_norm": 0.3202907145023346,
+      "learning_rate": 9.918182050001888e-05,
+      "loss": 0.0217,
+      "step": 5240
+    },
+    {
+      "grad_norm": 0.323348730802536,
+      "learning_rate": 9.91758517970723e-05,
+      "loss": 0.0203,
+      "step": 5250
+    },
+    {
+      "grad_norm": 0.3412604331970215,
+      "learning_rate": 9.916986158293744e-05,
+      "loss": 0.023,
+      "step": 5260
+    },
+    {
+      "grad_norm": 0.34787967801094055,
+      "learning_rate": 9.916384986023463e-05,
+      "loss": 0.0205,
+      "step": 5270
+    },
+    {
+      "grad_norm": 0.3832896053791046,
+      "learning_rate": 9.915781663159359e-05,
+      "loss": 0.0192,
+      "step": 5280
+    },
+    {
+      "grad_norm": 0.34575024247169495,
+      "learning_rate": 9.915176189965346e-05,
+      "loss": 0.0193,
+      "step": 5290
+    },
+    {
+      "grad_norm": 0.29879263043403625,
+      "learning_rate": 9.914568566706279e-05,
+      "loss": 0.0209,
+      "step": 5300
+    },
+    {
+      "grad_norm": 0.32127827405929565,
+      "learning_rate": 9.913958793647953e-05,
+      "loss": 0.0199,
+      "step": 5310
+    },
+    {
+      "grad_norm": 0.30529698729515076,
+      "learning_rate": 9.913346871057102e-05,
+      "loss": 0.0203,
+      "step": 5320
+    },
+    {
+      "grad_norm": 0.26579582691192627,
+      "learning_rate": 9.912732799201403e-05,
+      "loss": 0.0201,
+      "step": 5330
+    },
+    {
+      "grad_norm": 0.3353303372859955,
+      "learning_rate": 9.912116578349474e-05,
+      "loss": 0.0211,
+      "step": 5340
+    },
+    {
+      "grad_norm": 0.3893202245235443,
+      "learning_rate": 9.911498208770867e-05,
+      "loss": 0.0189,
+      "step": 5350
+    },
+    {
+      "grad_norm": 0.4526613652706146,
+      "learning_rate": 9.910877690736078e-05,
+      "loss": 0.0214,
+      "step": 5360
+    },
+    {
+      "grad_norm": 0.3455732464790344,
+      "learning_rate": 9.910255024516546e-05,
+      "loss": 0.0209,
+      "step": 5370
+    },
+    {
+      "grad_norm": 0.46266719698905945,
+      "learning_rate": 9.909630210384644e-05,
+      "loss": 0.0205,
+      "step": 5380
+    },
+    {
+      "grad_norm": 0.3487505614757538,
+      "learning_rate": 9.909003248613688e-05,
+      "loss": 0.0208,
+      "step": 5390
+    },
+    {
+      "grad_norm": 0.2546757459640503,
+      "learning_rate": 9.908374139477934e-05,
+      "loss": 0.0212,
+      "step": 5400
+    },
+    {
+      "grad_norm": 0.30674245953559875,
+      "learning_rate": 9.907742883252571e-05,
+      "loss": 0.0217,
+      "step": 5410
+    },
+    {
+      "grad_norm": 0.4218088984489441,
+      "learning_rate": 9.907109480213736e-05,
+      "loss": 0.0235,
+      "step": 5420
+    },
+    {
+      "grad_norm": 0.32780128717422485,
+      "learning_rate": 9.906473930638498e-05,
+      "loss": 0.0213,
+      "step": 5430
+    },
+    {
+      "grad_norm": 0.34243783354759216,
+      "learning_rate": 9.90583623480487e-05,
+      "loss": 0.0198,
+      "step": 5440
+    },
+    {
+      "grad_norm": 0.29020997881889343,
+      "learning_rate": 9.905196392991802e-05,
+      "loss": 0.0196,
+      "step": 5450
+    },
+    {
+      "grad_norm": 0.33883586525917053,
+      "learning_rate": 9.904554405479183e-05,
+      "loss": 0.0181,
+      "step": 5460
+    },
+    {
+      "grad_norm": 0.39794132113456726,
+      "learning_rate": 9.903910272547838e-05,
+      "loss": 0.02,
+      "step": 5470
+    },
+    {
+      "grad_norm": 0.4234314262866974,
+      "learning_rate": 9.903263994479533e-05,
+      "loss": 0.0197,
+      "step": 5480
+    },
+    {
+      "grad_norm": 0.33708757162094116,
+      "learning_rate": 9.902615571556972e-05,
+      "loss": 0.02,
+      "step": 5490
+    },
+    {
+      "grad_norm": 0.2579928934574127,
+      "learning_rate": 9.9019650040638e-05,
+      "loss": 0.0226,
+      "step": 5500
+    },
+    {
+      "grad_norm": 0.38343164324760437,
+      "learning_rate": 9.901312292284594e-05,
+      "loss": 0.018,
+      "step": 5510
+    },
+    {
+      "grad_norm": 0.32312721014022827,
+      "learning_rate": 9.900657436504875e-05,
+      "loss": 0.0208,
+      "step": 5520
+    },
+    {
+      "grad_norm": 0.29825785756111145,
+      "learning_rate": 9.900000437011097e-05,
+      "loss": 0.0195,
+      "step": 5530
+    },
+    {
+      "grad_norm": 0.37522104382514954,
+      "learning_rate": 9.899341294090653e-05,
+      "loss": 0.0221,
+      "step": 5540
+    },
+    {
+      "grad_norm": 0.31435421109199524,
+      "learning_rate": 9.898680008031877e-05,
+      "loss": 0.0203,
+      "step": 5550
+    },
+    {
+      "grad_norm": 0.3178274631500244,
+      "learning_rate": 9.898016579124037e-05,
+      "loss": 0.0222,
+      "step": 5560
+    },
+    {
+      "grad_norm": 0.33011436462402344,
+      "learning_rate": 9.89735100765734e-05,
+      "loss": 0.0218,
+      "step": 5570
+    },
+    {
+      "grad_norm": 0.26615071296691895,
+      "learning_rate": 9.89668329392293e-05,
+      "loss": 0.0263,
+      "step": 5580
+    },
+    {
+      "grad_norm": 0.31268659234046936,
+      "learning_rate": 9.896013438212885e-05,
+      "loss": 0.0222,
+      "step": 5590
+    },
+    {
+      "grad_norm": 0.3553520441055298,
+      "learning_rate": 9.895341440820225e-05,
+      "loss": 0.0258,
+      "step": 5600
+    },
+    {
+      "grad_norm": 0.39487162232398987,
+      "learning_rate": 9.894667302038902e-05,
+      "loss": 0.0243,
+      "step": 5610
+    },
+    {
+      "grad_norm": 0.33653688430786133,
+      "learning_rate": 9.893991022163812e-05,
+      "loss": 0.0223,
+      "step": 5620
+    },
+    {
+      "grad_norm": 0.3505474925041199,
+      "learning_rate": 9.893312601490779e-05,
+      "loss": 0.0222,
+      "step": 5630
+    },
+    {
+      "grad_norm": 0.31886622309684753,
+      "learning_rate": 9.892632040316568e-05,
+      "loss": 0.0189,
+      "step": 5640
+    },
+    {
+      "grad_norm": 0.39699551463127136,
+      "learning_rate": 9.891949338938877e-05,
+      "loss": 0.0203,
+      "step": 5650
+    },
+    {
+      "grad_norm": 0.35268929600715637,
+      "learning_rate": 9.89126449765635e-05,
+      "loss": 0.0193,
+      "step": 5660
+    },
+    {
+      "grad_norm": 0.3256807029247284,
+      "learning_rate": 9.890577516768551e-05,
+      "loss": 0.0233,
+      "step": 5670
+    },
+    {
+      "grad_norm": 0.28616154193878174,
+      "learning_rate": 9.889888396575995e-05,
+      "loss": 0.0187,
+      "step": 5680
+    },
+    {
+      "grad_norm": 0.27966874837875366,
+      "learning_rate": 9.889197137380125e-05,
+      "loss": 0.0214,
+      "step": 5690
+    },
+    {
+      "grad_norm": 0.19445021450519562,
+      "learning_rate": 9.888503739483321e-05,
+      "loss": 0.0174,
+      "step": 5700
+    },
+    {
+      "grad_norm": 0.3724949061870575,
+      "learning_rate": 9.887808203188897e-05,
+      "loss": 0.0214,
+      "step": 5710
+    },
+    {
+      "grad_norm": 0.3301282525062561,
+      "learning_rate": 9.887110528801108e-05,
+      "loss": 0.0189,
+      "step": 5720
+    },
+    {
+      "grad_norm": 0.32831037044525146,
+      "learning_rate": 9.886410716625137e-05,
+      "loss": 0.023,
+      "step": 5730
+    },
+    {
+      "grad_norm": 0.4092217981815338,
+      "learning_rate": 9.885708766967108e-05,
+      "loss": 0.0206,
+      "step": 5740
+    },
+    {
+      "grad_norm": 0.31922006607055664,
+      "learning_rate": 9.885004680134076e-05,
+      "loss": 0.0185,
+      "step": 5750
+    },
+    {
+      "grad_norm": 0.3902971148490906,
+      "learning_rate": 9.884298456434033e-05,
+      "loss": 0.019,
+      "step": 5760
+    },
+    {
+      "grad_norm": 0.31563547253608704,
+      "learning_rate": 9.883590096175905e-05,
+      "loss": 0.0195,
+      "step": 5770
+    },
+    {
+      "grad_norm": 0.3342783451080322,
+      "learning_rate": 9.882879599669554e-05,
+      "loss": 0.0209,
+      "step": 5780
+    },
+    {
+      "grad_norm": 0.4186359643936157,
+      "learning_rate": 9.882166967225775e-05,
+      "loss": 0.0188,
+      "step": 5790
+    },
+    {
+      "grad_norm": 0.31970587372779846,
+      "learning_rate": 9.881452199156296e-05,
+      "loss": 0.0201,
+      "step": 5800
+    },
+    {
+      "grad_norm": 0.3520440459251404,
+      "learning_rate": 9.880735295773782e-05,
+      "loss": 0.0205,
+      "step": 5810
+    },
+    {
+      "grad_norm": 0.4234348237514496,
+      "learning_rate": 9.88001625739183e-05,
+      "loss": 0.0194,
+      "step": 5820
+    },
+    {
+      "grad_norm": 0.33897021412849426,
+      "learning_rate": 9.879295084324976e-05,
+      "loss": 0.0188,
+      "step": 5830
+    },
+    {
+      "grad_norm": 0.32725125551223755,
+      "learning_rate": 9.87857177688868e-05,
+      "loss": 0.0207,
+      "step": 5840
+    },
+    {
+      "grad_norm": 0.21404729783535004,
+      "learning_rate": 9.877846335399343e-05,
+      "loss": 0.0205,
+      "step": 5850
+    },
+    {
+      "grad_norm": 0.27704718708992004,
+      "learning_rate": 9.8771187601743e-05,
+      "loss": 0.0203,
+      "step": 5860
+    },
+    {
+      "grad_norm": 0.3404671549797058,
+      "learning_rate": 9.876389051531813e-05,
+      "loss": 0.0178,
+      "step": 5870
+    },
+    {
+      "grad_norm": 0.3518414795398712,
+      "learning_rate": 9.875657209791088e-05,
+      "loss": 0.0189,
+      "step": 5880
+    },
+    {
+      "grad_norm": 0.3724534511566162,
+      "learning_rate": 9.874923235272248e-05,
+      "loss": 0.0161,
+      "step": 5890
+    },
+    {
+      "grad_norm": 0.4224037826061249,
+      "learning_rate": 9.874187128296369e-05,
+      "loss": 0.0189,
+      "step": 5900
+    },
+    {
+      "grad_norm": 0.4166416525840759,
+      "learning_rate": 9.873448889185439e-05,
+      "loss": 0.0196,
+      "step": 5910
+    },
+    {
+      "grad_norm": 0.25864657759666443,
+      "learning_rate": 9.872708518262397e-05,
+      "loss": 0.0184,
+      "step": 5920
+    },
+    {
+      "grad_norm": 0.37809693813323975,
+      "learning_rate": 9.871966015851101e-05,
+      "loss": 0.0218,
+      "step": 5930
+    },
+    {
+      "grad_norm": 0.3720448911190033,
+      "learning_rate": 9.87122138227635e-05,
+      "loss": 0.0206,
+      "step": 5940
+    },
+    {
+      "grad_norm": 0.311397522687912,
+      "learning_rate": 9.87047461786387e-05,
+      "loss": 0.0234,
+      "step": 5950
+    },
+    {
+      "grad_norm": 0.3835497200489044,
+      "learning_rate": 9.869725722940323e-05,
+      "loss": 0.0214,
+      "step": 5960
+    },
+    {
+      "grad_norm": 0.253582239151001,
+      "learning_rate": 9.868974697833299e-05,
+      "loss": 0.019,
+      "step": 5970
+    },
+    {
+      "grad_norm": 0.2803143858909607,
+      "learning_rate": 9.868221542871324e-05,
+      "loss": 0.0194,
+      "step": 5980
+    },
+    {
+      "grad_norm": 0.4318469762802124,
+      "learning_rate": 9.867466258383853e-05,
+      "loss": 0.0205,
+      "step": 5990
+    },
+    {
+      "grad_norm": 0.321250855922699,
+      "learning_rate": 9.866708844701272e-05,
+      "loss": 0.0182,
+      "step": 6000
+    },
+    {
+      "grad_norm": 0.37409961223602295,
+      "learning_rate": 9.865949302154899e-05,
+      "loss": 0.0226,
+      "step": 6010
+    },
+    {
+      "grad_norm": 0.3186977207660675,
+      "learning_rate": 9.865187631076987e-05,
+      "loss": 0.0196,
+      "step": 6020
+    },
+    {
+      "grad_norm": 0.3711942732334137,
+      "learning_rate": 9.864423831800712e-05,
+      "loss": 0.0223,
+      "step": 6030
+    },
+    {
+      "grad_norm": 0.3160367012023926,
+      "learning_rate": 9.863657904660191e-05,
+      "loss": 0.0194,
+      "step": 6040
+    },
+    {
+      "grad_norm": 0.33964473009109497,
+      "learning_rate": 9.862889849990462e-05,
+      "loss": 0.0205,
+      "step": 6050
+    },
+    {
+      "grad_norm": 0.4114055633544922,
+      "learning_rate": 9.8621196681275e-05,
+      "loss": 0.0216,
+      "step": 6060
+    },
+    {
+      "grad_norm": 0.33214277029037476,
+      "learning_rate": 9.861347359408211e-05,
+      "loss": 0.0205,
+      "step": 6070
+    },
+    {
+      "grad_norm": 0.3627830147743225,
+      "learning_rate": 9.860572924170426e-05,
+      "loss": 0.024,
+      "step": 6080
+    },
+    {
+      "grad_norm": 0.33737629652023315,
+      "learning_rate": 9.85979636275291e-05,
+      "loss": 0.0216,
+      "step": 6090
+    },
+    {
+      "grad_norm": 0.3489571213722229,
+      "learning_rate": 9.859017675495357e-05,
+      "loss": 0.0192,
+      "step": 6100
+    },
+    {
+      "grad_norm": 0.3351980447769165,
+      "learning_rate": 9.858236862738392e-05,
+      "loss": 0.0214,
+      "step": 6110
+    },
+    {
+      "grad_norm": 0.3590877652168274,
+      "learning_rate": 9.85745392482357e-05,
+      "loss": 0.0205,
+      "step": 6120
+    },
+    {
+      "grad_norm": 0.3898162245750427,
+      "learning_rate": 9.856668862093372e-05,
+      "loss": 0.0197,
+      "step": 6130
+    },
+    {
+      "grad_norm": 0.3354169428348541,
+      "learning_rate": 9.855881674891214e-05,
+      "loss": 0.0203,
+      "step": 6140
+    },
+    {
+      "grad_norm": 0.3553580939769745,
+      "learning_rate": 9.855092363561437e-05,
+      "loss": 0.017,
+      "step": 6150
+    },
+    {
+      "grad_norm": 0.3047468364238739,
+      "learning_rate": 9.854300928449312e-05,
+      "loss": 0.0176,
+      "step": 6160
+    },
+    {
+      "grad_norm": 0.3227238953113556,
+      "learning_rate": 9.85350736990104e-05,
+      "loss": 0.0167,
+      "step": 6170
+    },
+    {
+      "grad_norm": 0.33069944381713867,
+      "learning_rate": 9.852711688263752e-05,
+      "loss": 0.0174,
+      "step": 6180
+    },
+    {
+      "grad_norm": 0.365953654050827,
+      "learning_rate": 9.851913883885503e-05,
+      "loss": 0.0192,
+      "step": 6190
+    },
+    {
+      "grad_norm": 0.40460461378097534,
+      "learning_rate": 9.851113957115283e-05,
+      "loss": 0.0198,
+      "step": 6200
+    },
+    {
+      "grad_norm": 0.3165339231491089,
+      "learning_rate": 9.850311908303005e-05,
+      "loss": 0.0196,
+      "step": 6210
+    },
+    {
+      "grad_norm": 0.3397218585014343,
+      "learning_rate": 9.849507737799514e-05,
+      "loss": 0.0188,
+      "step": 6220
+    },
+    {
+      "grad_norm": 0.2602338194847107,
+      "learning_rate": 9.84870144595658e-05,
+      "loss": 0.0177,
+      "step": 6230
+    },
+    {
+      "grad_norm": 0.31266888976097107,
+      "learning_rate": 9.847893033126903e-05,
+      "loss": 0.0188,
+      "step": 6240
+    },
+    {
+      "grad_norm": 0.36433306336402893,
+      "learning_rate": 9.847082499664111e-05,
+      "loss": 0.0178,
+      "step": 6250
+    },
+    {
+      "grad_norm": 0.30065828561782837,
+      "learning_rate": 9.846269845922758e-05,
+      "loss": 0.0169,
+      "step": 6260
+    },
+    {
+      "grad_norm": 0.35674798488616943,
+      "learning_rate": 9.845455072258326e-05,
+      "loss": 0.0178,
+      "step": 6270
+    },
+    {
+      "grad_norm": 0.2833085358142853,
+      "learning_rate": 9.844638179027226e-05,
+      "loss": 0.0192,
+      "step": 6280
+    },
+    {
+      "grad_norm": 0.3853893280029297,
+      "learning_rate": 9.843819166586795e-05,
+      "loss": 0.0213,
+      "step": 6290
+    },
+    {
+      "grad_norm": 0.33940932154655457,
+      "learning_rate": 9.842998035295294e-05,
+      "loss": 0.0197,
+      "step": 6300
+    },
+    {
+      "grad_norm": 0.30166372656822205,
+      "learning_rate": 9.842174785511918e-05,
+      "loss": 0.0193,
+      "step": 6310
+    },
+    {
+      "grad_norm": 0.3530243933200836,
+      "learning_rate": 9.841349417596779e-05,
+      "loss": 0.0191,
+      "step": 6320
+    },
+    {
+      "grad_norm": 0.31284666061401367,
+      "learning_rate": 9.840521931910926e-05,
+      "loss": 0.019,
+      "step": 6330
+    },
+    {
+      "grad_norm": 0.2872731387615204,
+      "learning_rate": 9.839692328816327e-05,
+      "loss": 0.0183,
+      "step": 6340
+    },
+    {
+      "grad_norm": 0.31504446268081665,
+      "learning_rate": 9.838860608675879e-05,
+      "loss": 0.0235,
+      "step": 6350
+    },
+    {
+      "grad_norm": 0.316715806722641,
+      "learning_rate": 9.838026771853406e-05,
+      "loss": 0.0189,
+      "step": 6360
+    },
+    {
+      "grad_norm": 0.34900811314582825,
+      "learning_rate": 9.837190818713655e-05,
+      "loss": 0.0199,
+      "step": 6370
+    },
+    {
+      "grad_norm": 0.3429293930530548,
+      "learning_rate": 9.836352749622299e-05,
+      "loss": 0.0198,
+      "step": 6380
+    },
+    {
+      "grad_norm": 0.31453776359558105,
+      "learning_rate": 9.835512564945941e-05,
+      "loss": 0.0206,
+      "step": 6390
+    },
+    {
+      "grad_norm": 0.2731575071811676,
+      "learning_rate": 9.834670265052104e-05,
+      "loss": 0.0214,
+      "step": 6400
+    },
+    {
+      "grad_norm": 0.3495926856994629,
+      "learning_rate": 9.833825850309239e-05,
+      "loss": 0.0205,
+      "step": 6410
+    },
+    {
+      "grad_norm": 0.3381836414337158,
+      "learning_rate": 9.832979321086723e-05,
+      "loss": 0.0187,
+      "step": 6420
+    },
+    {
+      "grad_norm": 0.4271707236766815,
+      "learning_rate": 9.832130677754854e-05,
+      "loss": 0.017,
+      "step": 6430
+    },
+    {
+      "grad_norm": 0.3237077593803406,
+      "learning_rate": 9.831279920684861e-05,
+      "loss": 0.0192,
+      "step": 6440
+    },
+    {
+      "grad_norm": 0.28291055560112,
+      "learning_rate": 9.830427050248891e-05,
+      "loss": 0.0168,
+      "step": 6450
+    },
+    {
+      "grad_norm": 0.382289320230484,
+      "learning_rate": 9.829572066820021e-05,
+      "loss": 0.0185,
+      "step": 6460
+    },
+    {
+      "grad_norm": 0.3358441889286041,
+      "learning_rate": 9.828714970772247e-05,
+      "loss": 0.0157,
+      "step": 6470
+    },
+    {
+      "grad_norm": 0.21419993042945862,
+      "learning_rate": 9.827855762480493e-05,
+      "loss": 0.0171,
+      "step": 6480
+    },
+    {
+      "grad_norm": 0.2983962893486023,
+      "learning_rate": 9.826994442320608e-05,
+      "loss": 0.019,
+      "step": 6490
+    },
+    {
+      "grad_norm": 0.25639960169792175,
+      "learning_rate": 9.82613101066936e-05,
+      "loss": 0.0181,
+      "step": 6500
+    },
+    {
+      "grad_norm": 0.2753413915634155,
+      "learning_rate": 9.825265467904445e-05,
+      "loss": 0.0191,
+      "step": 6510
+    },
+    {
+      "grad_norm": 0.25280821323394775,
+      "learning_rate": 9.82439781440448e-05,
+      "loss": 0.017,
+      "step": 6520
+    },
+    {
+      "grad_norm": 0.3761622905731201,
+      "learning_rate": 9.823528050549006e-05,
+      "loss": 0.0179,
+      "step": 6530
+    },
+    {
+      "grad_norm": 0.3847559988498688,
+      "learning_rate": 9.822656176718487e-05,
+      "loss": 0.0149,
+      "step": 6540
+    },
+    {
+      "grad_norm": 0.2999807596206665,
+      "learning_rate": 9.82178219329431e-05,
+      "loss": 0.0184,
+      "step": 6550
+    },
+    {
+      "grad_norm": 0.3590312600135803,
+      "learning_rate": 9.820906100658789e-05,
+      "loss": 0.0181,
+      "step": 6560
+    },
+    {
+      "grad_norm": 0.2895771861076355,
+      "learning_rate": 9.82002789919515e-05,
+      "loss": 0.0183,
+      "step": 6570
+    },
+    {
+      "grad_norm": 0.23312729597091675,
+      "learning_rate": 9.819147589287554e-05,
+      "loss": 0.0155,
+      "step": 6580
+    },
+    {
+      "grad_norm": 0.2551847994327545,
+      "learning_rate": 9.818265171321074e-05,
+      "loss": 0.0173,
+      "step": 6590
+    },
+    {
+      "grad_norm": 0.3760150372982025,
+      "learning_rate": 9.817380645681714e-05,
+      "loss": 0.0182,
+      "step": 6600
+    },
+    {
+      "grad_norm": 0.3413234353065491,
+      "learning_rate": 9.816494012756392e-05,
+      "loss": 0.0178,
+      "step": 6610
+    },
+    {
+      "grad_norm": 0.28231438994407654,
+      "learning_rate": 9.815605272932953e-05,
+      "loss": 0.0202,
+      "step": 6620
+    },
+    {
+      "grad_norm": 0.35070836544036865,
+      "learning_rate": 9.814714426600162e-05,
+      "loss": 0.0212,
+      "step": 6630
+    },
+    {
+      "grad_norm": 0.25691017508506775,
+      "learning_rate": 9.813821474147704e-05,
+      "loss": 0.0201,
+      "step": 6640
+    },
+    {
+      "grad_norm": 0.3082495927810669,
+      "learning_rate": 9.812926415966189e-05,
+      "loss": 0.0212,
+      "step": 6650
+    },
+    {
+      "grad_norm": 0.32203003764152527,
+      "learning_rate": 9.812029252447143e-05,
+      "loss": 0.0168,
+      "step": 6660
+    },
+    {
+      "grad_norm": 0.2846328020095825,
+      "learning_rate": 9.811129983983018e-05,
+      "loss": 0.0189,
+      "step": 6670
+    },
+    {
+      "grad_norm": 0.386328786611557,
+      "learning_rate": 9.810228610967185e-05,
+      "loss": 0.0168,
+      "step": 6680
+    },
+    {
+      "grad_norm": 0.34293895959854126,
+      "learning_rate": 9.809325133793934e-05,
+      "loss": 0.0177,
+      "step": 6690
+    },
+    {
+      "grad_norm": 0.3505919873714447,
+      "learning_rate": 9.808419552858477e-05,
+      "loss": 0.0179,
+      "step": 6700
+    },
+    {
+      "grad_norm": 0.32018762826919556,
+      "learning_rate": 9.807511868556944e-05,
+      "loss": 0.0181,
+      "step": 6710
+    },
+    {
+      "grad_norm": 0.28522124886512756,
+      "learning_rate": 9.806602081286392e-05,
+      "loss": 0.0209,
+      "step": 6720
+    },
+    {
+      "grad_norm": 0.3764745891094208,
+      "learning_rate": 9.805690191444785e-05,
+      "loss": 0.0185,
+      "step": 6730
+    },
+    {
+      "grad_norm": 0.26050126552581787,
+      "learning_rate": 9.804776199431022e-05,
+      "loss": 0.0172,
+      "step": 6740
+    },
+    {
+      "grad_norm": 0.2483110874891281,
+      "learning_rate": 9.80386010564491e-05,
+      "loss": 0.0224,
+      "step": 6750
+    },
+    {
+      "grad_norm": 0.2636786997318268,
+      "learning_rate": 9.802941910487183e-05,
+      "loss": 0.0188,
+      "step": 6760
+    },
+    {
+      "grad_norm": 0.34675920009613037,
+      "learning_rate": 9.802021614359487e-05,
+      "loss": 0.0193,
+      "step": 6770
+    },
+    {
+      "grad_norm": 0.35509130358695984,
+      "learning_rate": 9.801099217664394e-05,
+      "loss": 0.0213,
+      "step": 6780
+    },
+    {
+      "grad_norm": 0.3069632947444916,
+      "learning_rate": 9.800174720805388e-05,
+      "loss": 0.0205,
+      "step": 6790
+    },
+    {
+      "grad_norm": 0.2569907307624817,
+      "learning_rate": 9.799248124186878e-05,
+      "loss": 0.0192,
+      "step": 6800
+    },
+    {
+      "grad_norm": 0.30510249733924866,
+      "learning_rate": 9.798319428214187e-05,
+      "loss": 0.019,
+      "step": 6810
+    },
+    {
+      "grad_norm": 0.3681212067604065,
+      "learning_rate": 9.79738863329356e-05,
+      "loss": 0.0204,
+      "step": 6820
+    },
+    {
+      "grad_norm": 0.3518400490283966,
+      "learning_rate": 9.79645573983216e-05,
+      "loss": 0.0185,
+      "step": 6830
+    },
+    {
+      "grad_norm": 0.33140143752098083,
+      "learning_rate": 9.795520748238062e-05,
+      "loss": 0.0185,
+      "step": 6840
+    },
+    {
+      "grad_norm": 0.31977716088294983,
+      "learning_rate": 9.794583658920264e-05,
+      "loss": 0.0168,
+      "step": 6850
+    },
+    {
+      "grad_norm": 0.347068190574646,
+      "learning_rate": 9.793644472288683e-05,
+      "loss": 0.0174,
+      "step": 6860
+    },
+    {
+      "grad_norm": 0.2709243595600128,
+      "learning_rate": 9.79270318875415e-05,
+      "loss": 0.0174,
+      "step": 6870
+    },
+    {
+      "grad_norm": 0.2900465130805969,
+      "learning_rate": 9.791759808728416e-05,
+      "loss": 0.0193,
+      "step": 6880
+    },
+    {
+      "grad_norm": 0.3605364263057709,
+      "learning_rate": 9.790814332624143e-05,
+      "loss": 0.0174,
+      "step": 6890
+    },
+    {
+      "grad_norm": 0.27951502799987793,
+      "learning_rate": 9.789866760854919e-05,
+      "loss": 0.0203,
+      "step": 6900
+    },
+    {
+      "grad_norm": 0.4382677972316742,
+      "learning_rate": 9.788917093835242e-05,
+      "loss": 0.0203,
+      "step": 6910
+    },
+    {
+      "grad_norm": 0.38475170731544495,
+      "learning_rate": 9.787965331980528e-05,
+      "loss": 0.0197,
+      "step": 6920
+    },
+    {
+      "grad_norm": 0.26169997453689575,
+      "learning_rate": 9.787011475707111e-05,
+      "loss": 0.0206,
+      "step": 6930
+    },
+    {
+      "grad_norm": 0.3852261006832123,
+      "learning_rate": 9.786055525432241e-05,
+      "loss": 0.02,
+      "step": 6940
+    },
+    {
+      "grad_norm": 0.3383841812610626,
+      "learning_rate": 9.78509748157408e-05,
+      "loss": 0.0241,
+      "step": 6950
+    },
+    {
+      "grad_norm": 0.3337130546569824,
+      "learning_rate": 9.784137344551713e-05,
+      "loss": 0.0221,
+      "step": 6960
+    },
+    {
+      "grad_norm": 0.31400367617607117,
+      "learning_rate": 9.783175114785134e-05,
+      "loss": 0.0196,
+      "step": 6970
+    },
+    {
+      "grad_norm": 0.35225582122802734,
+      "learning_rate": 9.782210792695254e-05,
+      "loss": 0.0214,
+      "step": 6980
+    },
+    {
+      "grad_norm": 0.3338993787765503,
+      "learning_rate": 9.781244378703904e-05,
+      "loss": 0.0176,
+      "step": 6990
+    },
+    {
+      "grad_norm": 0.2713209390640259,
+      "learning_rate": 9.780275873233824e-05,
+      "loss": 0.0171,
+      "step": 7000
+    },
+    {
+      "grad_norm": 0.3477729856967926,
+      "learning_rate": 9.77930527670867e-05,
+      "loss": 0.0193,
+      "step": 7010
+    },
+    {
+      "grad_norm": 0.2875981628894806,
+      "learning_rate": 9.778332589553016e-05,
+      "loss": 0.0194,
+      "step": 7020
+    },
+    {
+      "grad_norm": 0.2606404423713684,
+      "learning_rate": 9.777357812192349e-05,
+      "loss": 0.0177,
+      "step": 7030
+    },
+    {
+      "grad_norm": 0.3439493179321289,
+      "learning_rate": 9.776380945053066e-05,
+      "loss": 0.0188,
+      "step": 7040
+    },
+    {
+      "grad_norm": 0.33768367767333984,
+      "learning_rate": 9.775401988562486e-05,
+      "loss": 0.0196,
+      "step": 7050
+    },
+    {
+      "grad_norm": 0.2906929850578308,
+      "learning_rate": 9.774420943148835e-05,
+      "loss": 0.0177,
+      "step": 7060
+    },
+    {
+      "grad_norm": 0.29560762643814087,
+      "learning_rate": 9.773437809241257e-05,
+      "loss": 0.0218,
+      "step": 7070
+    },
+    {
+      "grad_norm": 0.3206325173377991,
+      "learning_rate": 9.772452587269808e-05,
+      "loss": 0.0182,
+      "step": 7080
+    },
+    {
+      "grad_norm": 0.2930667996406555,
+      "learning_rate": 9.771465277665457e-05,
+      "loss": 0.0187,
+      "step": 7090
+    },
+    {
+      "grad_norm": 0.36127686500549316,
+      "learning_rate": 9.770475880860089e-05,
+      "loss": 0.0166,
+      "step": 7100
+    },
+    {
+      "grad_norm": 0.30213460326194763,
+      "learning_rate": 9.769484397286495e-05,
+      "loss": 0.0185,
+      "step": 7110
+    },
+    {
+      "grad_norm": 0.27768614888191223,
+      "learning_rate": 9.768490827378388e-05,
+      "loss": 0.0181,
+      "step": 7120
+    },
+    {
+      "grad_norm": 0.29009509086608887,
+      "learning_rate": 9.767495171570387e-05,
+      "loss": 0.0201,
+      "step": 7130
+    },
+    {
+      "grad_norm": 0.34916937351226807,
+      "learning_rate": 9.766497430298027e-05,
+      "loss": 0.0167,
+      "step": 7140
+    },
+    {
+      "grad_norm": 0.2706291675567627,
+      "learning_rate": 9.765497603997754e-05,
+      "loss": 0.0162,
+      "step": 7150
+    },
+    {
+      "grad_norm": 0.2542431354522705,
+      "learning_rate": 9.764495693106926e-05,
+      "loss": 0.0188,
+      "step": 7160
+    },
+    {
+      "grad_norm": 0.46041643619537354,
+      "learning_rate": 9.76349169806381e-05,
+      "loss": 0.0193,
+      "step": 7170
+    },
+    {
+      "grad_norm": 0.34627753496170044,
+      "learning_rate": 9.762485619307591e-05,
+      "loss": 0.0188,
+      "step": 7180
+    },
+    {
+      "grad_norm": 0.302895724773407,
+      "learning_rate": 9.76147745727836e-05,
+      "loss": 0.0199,
+      "step": 7190
+    },
+    {
+      "grad_norm": 0.3211849629878998,
+      "learning_rate": 9.760467212417124e-05,
+      "loss": 0.0177,
+      "step": 7200
+    },
+    {
+      "grad_norm": 0.3034629225730896,
+      "learning_rate": 9.759454885165796e-05,
+      "loss": 0.0178,
+      "step": 7210
+    },
+    {
+      "grad_norm": 0.2492295354604721,
+      "learning_rate": 9.7584404759672e-05,
+      "loss": 0.0149,
+      "step": 7220
+    },
+    {
+      "grad_norm": 0.3764285147190094,
+      "learning_rate": 9.75742398526508e-05,
+      "loss": 0.0199,
+      "step": 7230
+    },
+    {
+      "grad_norm": 0.29855966567993164,
+      "learning_rate": 9.756405413504077e-05,
+      "loss": 0.0205,
+      "step": 7240
+    },
+    {
+      "grad_norm": 0.302048921585083,
+      "learning_rate": 9.755384761129752e-05,
+      "loss": 0.0161,
+      "step": 7250
+    },
+    {
+      "grad_norm": 0.2954849600791931,
+      "learning_rate": 9.754362028588572e-05,
+      "loss": 0.0156,
+      "step": 7260
+    },
+    {
+      "grad_norm": 0.2673201560974121,
+      "learning_rate": 9.753337216327917e-05,
+      "loss": 0.0179,
+      "step": 7270
+    },
+    {
+      "grad_norm": 0.2569791078567505,
+      "learning_rate": 9.752310324796073e-05,
+      "loss": 0.0176,
+      "step": 7280
+    },
+    {
+      "grad_norm": 0.31080785393714905,
+      "learning_rate": 9.751281354442235e-05,
+      "loss": 0.0195,
+      "step": 7290
+    },
+    {
+      "grad_norm": 0.24717780947685242,
+      "learning_rate": 9.750250305716513e-05,
+      "loss": 0.0179,
+      "step": 7300
+    },
+    {
+      "grad_norm": 0.3685244023799896,
+      "learning_rate": 9.749217179069923e-05,
+      "loss": 0.0182,
+      "step": 7310
+    },
+    {
+      "grad_norm": 0.347946435213089,
+      "learning_rate": 9.748181974954386e-05,
+      "loss": 0.0205,
+      "step": 7320
+    },
+    {
+      "grad_norm": 0.26662352681159973,
+      "learning_rate": 9.74714469382274e-05,
+      "loss": 0.0181,
+      "step": 7330
+    },
+    {
+      "grad_norm": 0.2533002495765686,
+      "learning_rate": 9.746105336128724e-05,
+      "loss": 0.0164,
+      "step": 7340
+    },
+    {
+      "grad_norm": 0.3013625741004944,
+      "learning_rate": 9.745063902326988e-05,
+      "loss": 0.0165,
+      "step": 7350
+    },
+    {
+      "grad_norm": 0.26680493354797363,
+      "learning_rate": 9.744020392873093e-05,
+      "loss": 0.0227,
+      "step": 7360
+    },
+    {
+      "grad_norm": 0.2611214220523834,
+      "learning_rate": 9.742974808223504e-05,
+      "loss": 0.0204,
+      "step": 7370
+    },
+    {
+      "grad_norm": 0.21558502316474915,
+      "learning_rate": 9.741927148835596e-05,
+      "loss": 0.0186,
+      "step": 7380
+    },
+    {
+      "grad_norm": 0.295011430978775,
+      "learning_rate": 9.740877415167648e-05,
+      "loss": 0.0191,
+      "step": 7390
+    },
+    {
+      "grad_norm": 0.30721810460090637,
+      "learning_rate": 9.739825607678854e-05,
+      "loss": 0.0163,
+      "step": 7400
+    },
+    {
+      "grad_norm": 0.37634387612342834,
+      "learning_rate": 9.738771726829308e-05,
+      "loss": 0.016,
+      "step": 7410
+    },
+    {
+      "grad_norm": 0.208169624209404,
+      "learning_rate": 9.737715773080011e-05,
+      "loss": 0.0161,
+      "step": 7420
+    },
+    {
+      "grad_norm": 0.2882494330406189,
+      "learning_rate": 9.736657746892877e-05,
+      "loss": 0.0164,
+      "step": 7430
+    },
+    {
+      "grad_norm": 0.3299116790294647,
+      "learning_rate": 9.735597648730718e-05,
+      "loss": 0.0167,
+      "step": 7440
+    },
+    {
+      "grad_norm": 0.3128003180027008,
+      "learning_rate": 9.734535479057262e-05,
+      "loss": 0.017,
+      "step": 7450
+    },
+    {
+      "grad_norm": 0.25450965762138367,
+      "learning_rate": 9.733471238337136e-05,
+      "loss": 0.0194,
+      "step": 7460
+    },
+    {
+      "grad_norm": 0.2447034865617752,
+      "learning_rate": 9.732404927035871e-05,
+      "loss": 0.017,
+      "step": 7470
+    },
+    {
+      "grad_norm": 0.2228989452123642,
+      "learning_rate": 9.731336545619915e-05,
+      "loss": 0.016,
+      "step": 7480
+    },
+    {
+      "grad_norm": 0.22852838039398193,
+      "learning_rate": 9.73026609455661e-05,
+      "loss": 0.0147,
+      "step": 7490
+    },
+    {
+      "grad_norm": 0.2357887327671051,
+      "learning_rate": 9.72919357431421e-05,
+      "loss": 0.0183,
+      "step": 7500
+    },
+    {
+      "grad_norm": 0.27929630875587463,
+      "learning_rate": 9.72811898536187e-05,
+      "loss": 0.0156,
+      "step": 7510
+    },
+    {
+      "grad_norm": 0.26032036542892456,
+      "learning_rate": 9.72704232816965e-05,
+      "loss": 0.0178,
+      "step": 7520
+    },
+    {
+      "grad_norm": 0.32722219824790955,
+      "learning_rate": 9.725963603208519e-05,
+      "loss": 0.0179,
+      "step": 7530
+    },
+    {
+      "grad_norm": 0.2755366563796997,
+      "learning_rate": 9.724882810950348e-05,
+      "loss": 0.0183,
+      "step": 7540
+    },
+    {
+      "grad_norm": 0.28937822580337524,
+      "learning_rate": 9.723799951867913e-05,
+      "loss": 0.0176,
+      "step": 7550
+    },
+    {
+      "grad_norm": 0.37954676151275635,
+      "learning_rate": 9.722715026434889e-05,
+      "loss": 0.0215,
+      "step": 7560
+    },
+    {
+      "grad_norm": 0.30485105514526367,
+      "learning_rate": 9.721628035125866e-05,
+      "loss": 0.0181,
+      "step": 7570
+    },
+    {
+      "grad_norm": 0.3920777142047882,
+      "learning_rate": 9.720538978416325e-05,
+      "loss": 0.0168,
+      "step": 7580
+    },
+    {
+      "grad_norm": 0.2998286187648773,
+      "learning_rate": 9.71944785678266e-05,
+      "loss": 0.0169,
+      "step": 7590
+    },
+    {
+      "grad_norm": 0.31956079602241516,
+      "learning_rate": 9.718354670702161e-05,
+      "loss": 0.0182,
+      "step": 7600
+    },
+    {
+      "grad_norm": 0.25753054022789,
+      "learning_rate": 9.717259420653029e-05,
+      "loss": 0.0158,
+      "step": 7610
+    },
+    {
+      "grad_norm": 0.36010029911994934,
+      "learning_rate": 9.716162107114361e-05,
+      "loss": 0.0171,
+      "step": 7620
+    },
+    {
+      "grad_norm": 0.26551494002342224,
+      "learning_rate": 9.71506273056616e-05,
+      "loss": 0.0176,
+      "step": 7630
+    },
+    {
+      "grad_norm": 0.3344331383705139,
+      "learning_rate": 9.713961291489331e-05,
+      "loss": 0.0171,
+      "step": 7640
+    },
+    {
+      "grad_norm": 0.2667110562324524,
+      "learning_rate": 9.71285779036568e-05,
+      "loss": 0.0177,
+      "step": 7650
+    },
+    {
+      "grad_norm": 0.3104725480079651,
+      "learning_rate": 9.711752227677916e-05,
+      "loss": 0.0202,
+      "step": 7660
+    },
+    {
+      "grad_norm": 0.3035726547241211,
+      "learning_rate": 9.710644603909649e-05,
+      "loss": 0.0184,
+      "step": 7670
+    },
+    {
+      "grad_norm": 0.313547819852829,
+      "learning_rate": 9.709534919545393e-05,
+      "loss": 0.0165,
+      "step": 7680
+    },
+    {
+      "grad_norm": 0.26152196526527405,
+      "learning_rate": 9.708423175070563e-05,
+      "loss": 0.015,
+      "step": 7690
+    },
+    {
+      "grad_norm": 0.30663272738456726,
+      "learning_rate": 9.707309370971468e-05,
+      "loss": 0.0193,
+      "step": 7700
+    },
+    {
+      "grad_norm": 0.24621541798114777,
+      "learning_rate": 9.70619350773533e-05,
+      "loss": 0.0179,
+      "step": 7710
+    },
+    {
+      "grad_norm": 0.2608759105205536,
+      "learning_rate": 9.70507558585026e-05,
+      "loss": 0.0151,
+      "step": 7720
+    },
+    {
+      "grad_norm": 0.3009045720100403,
+      "learning_rate": 9.703955605805281e-05,
+      "loss": 0.0197,
+      "step": 7730
+    },
+    {
+      "grad_norm": 0.24260351061820984,
+      "learning_rate": 9.702833568090306e-05,
+      "loss": 0.0175,
+      "step": 7740
+    },
+    {
+      "grad_norm": 0.3245747685432434,
+      "learning_rate": 9.701709473196154e-05,
+      "loss": 0.0198,
+      "step": 7750
+    },
+    {
+      "grad_norm": 0.3041107654571533,
+      "learning_rate": 9.700583321614541e-05,
+      "loss": 0.0174,
+      "step": 7760
+    },
+    {
+      "grad_norm": 0.2914700508117676,
+      "learning_rate": 9.699455113838085e-05,
+      "loss": 0.0236,
+      "step": 7770
+    },
+    {
+      "grad_norm": 0.23055170476436615,
+      "learning_rate": 9.698324850360303e-05,
+      "loss": 0.0186,
+      "step": 7780
+    },
+    {
+      "grad_norm": 0.35021236538887024,
+      "learning_rate": 9.69719253167561e-05,
+      "loss": 0.0166,
+      "step": 7790
+    },
+    {
+      "grad_norm": 0.3389432430267334,
+      "learning_rate": 9.69605815827932e-05,
+      "loss": 0.0173,
+      "step": 7800
+    },
+    {
+      "grad_norm": 0.2455502152442932,
+      "learning_rate": 9.694921730667647e-05,
+      "loss": 0.0172,
+      "step": 7810
+    },
+    {
+      "grad_norm": 0.30869391560554504,
+      "learning_rate": 9.693783249337705e-05,
+      "loss": 0.0152,
+      "step": 7820
+    },
+    {
+      "grad_norm": 0.25111138820648193,
+      "learning_rate": 9.692642714787501e-05,
+      "loss": 0.0192,
+      "step": 7830
+    },
+    {
+      "grad_norm": 0.3202841579914093,
+      "learning_rate": 9.691500127515945e-05,
+      "loss": 0.0168,
+      "step": 7840
+    },
+    {
+      "grad_norm": 0.23745019733905792,
+      "learning_rate": 9.690355488022844e-05,
+      "loss": 0.0167,
+      "step": 7850
+    },
+    {
+      "grad_norm": 0.24982476234436035,
+      "learning_rate": 9.689208796808902e-05,
+      "loss": 0.0174,
+      "step": 7860
+    },
+    {
+      "grad_norm": 0.3621152937412262,
+      "learning_rate": 9.68806005437572e-05,
+      "loss": 0.0163,
+      "step": 7870
+    },
+    {
+      "grad_norm": 0.27913394570350647,
+      "learning_rate": 9.686909261225796e-05,
+      "loss": 0.0163,
+      "step": 7880
+    },
+    {
+      "grad_norm": 0.3351999521255493,
+      "learning_rate": 9.685756417862531e-05,
+      "loss": 0.0179,
+      "step": 7890
+    },
+    {
+      "grad_norm": 0.29304763674736023,
+      "learning_rate": 9.684601524790212e-05,
+      "loss": 0.0177,
+      "step": 7900
+    },
+    {
+      "grad_norm": 0.3573150634765625,
+      "learning_rate": 9.68344458251403e-05,
+      "loss": 0.0177,
+      "step": 7910
+    },
+    {
+      "grad_norm": 0.362509161233902,
+      "learning_rate": 9.682285591540072e-05,
+      "loss": 0.0162,
+      "step": 7920
+    },
+    {
+      "grad_norm": 0.3151058256626129,
+      "learning_rate": 9.681124552375322e-05,
+      "loss": 0.0174,
+      "step": 7930
+    },
+    {
+      "grad_norm": 0.26832640171051025,
+      "learning_rate": 9.679961465527654e-05,
+      "loss": 0.0179,
+      "step": 7940
+    },
+    {
+      "grad_norm": 0.3299589157104492,
+      "learning_rate": 9.678796331505843e-05,
+      "loss": 0.0163,
+      "step": 7950
+    },
+    {
+      "grad_norm": 0.28243011236190796,
+      "learning_rate": 9.677629150819558e-05,
+      "loss": 0.018,
+      "step": 7960
+    },
+    {
+      "grad_norm": 0.2872242033481598,
+      "learning_rate": 9.676459923979366e-05,
+      "loss": 0.019,
+      "step": 7970
+    },
+    {
+      "grad_norm": 0.23864701390266418,
+      "learning_rate": 9.675288651496723e-05,
+      "loss": 0.0176,
+      "step": 7980
+    },
+    {
+      "grad_norm": 0.2671719491481781,
+      "learning_rate": 9.674115333883986e-05,
+      "loss": 0.0159,
+      "step": 7990
+    },
+    {
+      "grad_norm": 0.20232592523097992,
+      "learning_rate": 9.672939971654402e-05,
+      "loss": 0.014,
+      "step": 8000
+    },
+    {
+      "grad_norm": 0.2715405523777008,
+      "learning_rate": 9.671762565322117e-05,
+      "loss": 0.0128,
+      "step": 8010
+    },
+    {
+      "grad_norm": 0.3833991587162018,
+      "learning_rate": 9.670583115402166e-05,
+      "loss": 0.0152,
+      "step": 8020
+    },
+    {
+      "grad_norm": 0.3070376515388489,
+      "learning_rate": 9.669401622410482e-05,
+      "loss": 0.0158,
+      "step": 8030
+    },
+    {
+      "grad_norm": 0.31053975224494934,
+      "learning_rate": 9.668218086863887e-05,
+      "loss": 0.0169,
+      "step": 8040
+    },
+    {
+      "grad_norm": 0.2504522204399109,
+      "learning_rate": 9.667032509280103e-05,
+      "loss": 0.0162,
+      "step": 8050
+    },
+    {
+      "grad_norm": 0.2611943185329437,
+      "learning_rate": 9.665844890177743e-05,
+      "loss": 0.0155,
+      "step": 8060
+    },
+    {
+      "grad_norm": 0.29758018255233765,
+      "learning_rate": 9.664655230076309e-05,
+      "loss": 0.0158,
+      "step": 8070
+    },
+    {
+      "grad_norm": 0.26844164729118347,
+      "learning_rate": 9.663463529496199e-05,
+      "loss": 0.0165,
+      "step": 8080
+    },
+    {
+      "grad_norm": 0.3141821026802063,
+      "learning_rate": 9.662269788958705e-05,
+      "loss": 0.0168,
+      "step": 8090
+    },
+    {
+      "grad_norm": 0.33833351731300354,
+      "learning_rate": 9.661074008986008e-05,
+      "loss": 0.0172,
+      "step": 8100
+    },
+    {
+      "grad_norm": 0.26897549629211426,
+      "learning_rate": 9.659876190101184e-05,
+      "loss": 0.0145,
+      "step": 8110
+    },
+    {
+      "grad_norm": 0.3424350917339325,
+      "learning_rate": 9.6586763328282e-05,
+      "loss": 0.0164,
+      "step": 8120
+    },
+    {
+      "grad_norm": 0.3773399889469147,
+      "learning_rate": 9.657474437691913e-05,
+      "loss": 0.0168,
+      "step": 8130
+    },
+    {
+      "grad_norm": 0.3302556574344635,
+      "learning_rate": 9.656270505218073e-05,
+      "loss": 0.0153,
+      "step": 8140
+    },
+    {
+      "grad_norm": 0.27595070004463196,
+      "learning_rate": 9.655064535933321e-05,
+      "loss": 0.017,
+      "step": 8150
+    },
+    {
+      "grad_norm": 0.24360628426074982,
+      "learning_rate": 9.653856530365189e-05,
+      "loss": 0.0166,
+      "step": 8160
+    },
+    {
+      "grad_norm": 0.2539483308792114,
+      "learning_rate": 9.6526464890421e-05,
+      "loss": 0.0167,
+      "step": 8170
+    },
+    {
+      "grad_norm": 0.253546804189682,
+      "learning_rate": 9.651434412493367e-05,
+      "loss": 0.0162,
+      "step": 8180
+    },
+    {
+      "grad_norm": 0.26443150639533997,
+      "learning_rate": 9.650220301249195e-05,
+      "loss": 0.0171,
+      "step": 8190
+    },
+    {
+      "grad_norm": 0.32957175374031067,
+      "learning_rate": 9.649004155840675e-05,
+      "loss": 0.0158,
+      "step": 8200
+    },
+    {
+      "grad_norm": 0.30958208441734314,
+      "learning_rate": 9.647785976799792e-05,
+      "loss": 0.015,
+      "step": 8210
+    },
+    {
+      "grad_norm": 0.281331866979599,
+      "learning_rate": 9.646565764659417e-05,
+      "loss": 0.0138,
+      "step": 8220
+    },
+    {
+      "grad_norm": 0.36112746596336365,
+      "learning_rate": 9.645343519953314e-05,
+      "loss": 0.0143,
+      "step": 8230
+    },
+    {
+      "grad_norm": 0.31710800528526306,
+      "learning_rate": 9.644119243216135e-05,
+      "loss": 0.0156,
+      "step": 8240
+    },
+    {
+      "grad_norm": 0.3625369369983673,
+      "learning_rate": 9.642892934983417e-05,
+      "loss": 0.017,
+      "step": 8250
+    },
+    {
+      "grad_norm": 0.3713219463825226,
+      "learning_rate": 9.641664595791593e-05,
+      "loss": 0.0141,
+      "step": 8260
+    },
+    {
+      "grad_norm": 0.2783694267272949,
+      "learning_rate": 9.640434226177977e-05,
+      "loss": 0.0153,
+      "step": 8270
+    },
+    {
+      "grad_norm": 0.2924513518810272,
+      "learning_rate": 9.639201826680777e-05,
+      "loss": 0.0156,
+      "step": 8280
+    },
+    {
+      "grad_norm": 0.3321302533149719,
+      "learning_rate": 9.637967397839083e-05,
+      "loss": 0.0166,
+      "step": 8290
+    },
+    {
+      "grad_norm": 0.2749614715576172,
+      "learning_rate": 9.63673094019288e-05,
+      "loss": 0.0148,
+      "step": 8300
+    },
+    {
+      "grad_norm": 0.1880040019750595,
+      "learning_rate": 9.635492454283035e-05,
+      "loss": 0.0164,
+      "step": 8310
+    },
+    {
+      "grad_norm": 0.2796301543712616,
+      "learning_rate": 9.634251940651302e-05,
+      "loss": 0.0137,
+      "step": 8320
+    },
+    {
+      "grad_norm": 0.27801117300987244,
+      "learning_rate": 9.633009399840327e-05,
+      "loss": 0.015,
+      "step": 8330
+    },
+    {
+      "grad_norm": 0.2667204439640045,
+      "learning_rate": 9.631764832393639e-05,
+      "loss": 0.0143,
+      "step": 8340
+    },
+    {
+      "grad_norm": 0.2765606641769409,
+      "learning_rate": 9.630518238855651e-05,
+      "loss": 0.0165,
+      "step": 8350
+    },
+    {
+      "grad_norm": 0.21531830728054047,
+      "learning_rate": 9.629269619771668e-05,
+      "loss": 0.0148,
+      "step": 8360
+    },
+    {
+      "grad_norm": 0.20877371728420258,
+      "learning_rate": 9.628018975687878e-05,
+      "loss": 0.0128,
+      "step": 8370
+    },
+    {
+      "grad_norm": 0.291494220495224,
+      "learning_rate": 9.626766307151355e-05,
+      "loss": 0.0142,
+      "step": 8380
+    },
+    {
+      "grad_norm": 0.17771676182746887,
+      "learning_rate": 9.62551161471006e-05,
+      "loss": 0.0152,
+      "step": 8390
+    },
+    {
+      "grad_norm": 0.22908322513103485,
+      "learning_rate": 9.624254898912837e-05,
+      "loss": 0.0153,
+      "step": 8400
+    },
+    {
+      "grad_norm": 0.3158961236476898,
+      "learning_rate": 9.622996160309414e-05,
+      "loss": 0.0173,
+      "step": 8410
+    },
+    {
+      "grad_norm": 0.3140466511249542,
+      "learning_rate": 9.62173539945041e-05,
+      "loss": 0.0145,
+      "step": 8420
+    },
+    {
+      "grad_norm": 0.20852278172969818,
+      "learning_rate": 9.620472616887323e-05,
+      "loss": 0.0133,
+      "step": 8430
+    },
+    {
+      "grad_norm": 0.3100578486919403,
+      "learning_rate": 9.619207813172536e-05,
+      "loss": 0.0162,
+      "step": 8440
+    },
+    {
+      "grad_norm": 0.2964276969432831,
+      "learning_rate": 9.617940988859318e-05,
+      "loss": 0.0204,
+      "step": 8450
+    },
+    {
+      "grad_norm": 0.24366337060928345,
+      "learning_rate": 9.616672144501821e-05,
+      "loss": 0.0157,
+      "step": 8460
+    },
+    {
+      "grad_norm": 0.25211432576179504,
+      "learning_rate": 9.615401280655081e-05,
+      "loss": 0.0147,
+      "step": 8470
+    },
+    {
+      "grad_norm": 0.2324327677488327,
+      "learning_rate": 9.614128397875017e-05,
+      "loss": 0.0154,
+      "step": 8480
+    },
+    {
+      "grad_norm": 0.3534395396709442,
+      "learning_rate": 9.612853496718429e-05,
+      "loss": 0.0153,
+      "step": 8490
+    },
+    {
+      "grad_norm": 0.28008079528808594,
+      "learning_rate": 9.611576577743004e-05,
+      "loss": 0.0181,
+      "step": 8500
+    },
+    {
+      "grad_norm": 0.37532371282577515,
+      "learning_rate": 9.610297641507313e-05,
+      "loss": 0.0151,
+      "step": 8510
+    },
+    {
+      "grad_norm": 0.31317541003227234,
+      "learning_rate": 9.6090166885708e-05,
+      "loss": 0.015,
+      "step": 8520
+    },
+    {
+      "grad_norm": 0.33534374833106995,
+      "learning_rate": 9.607733719493798e-05,
+      "loss": 0.0145,
+      "step": 8530
+    },
+    {
+      "grad_norm": 0.2918477952480316,
+      "learning_rate": 9.606448734837527e-05,
+      "loss": 0.0173,
+      "step": 8540
+    },
+    {
+      "grad_norm": 0.17789967358112335,
+      "learning_rate": 9.605161735164079e-05,
+      "loss": 0.0137,
+      "step": 8550
+    },
+    {
+      "grad_norm": 0.2251691371202469,
+      "learning_rate": 9.60387272103643e-05,
+      "loss": 0.016,
+      "step": 8560
+    },
+    {
+      "grad_norm": 0.3063332438468933,
+      "learning_rate": 9.60258169301844e-05,
+      "loss": 0.0195,
+      "step": 8570
+    },
+    {
+      "grad_norm": 0.4514167010784149,
+      "learning_rate": 9.601288651674851e-05,
+      "loss": 0.0154,
+      "step": 8580
+    },
+    {
+      "grad_norm": 0.2603996694087982,
+      "learning_rate": 9.59999359757128e-05,
+      "loss": 0.0149,
+      "step": 8590
+    },
+    {
+      "grad_norm": 0.2523835301399231,
+      "learning_rate": 9.598696531274227e-05,
+      "loss": 0.0143,
+      "step": 8600
+    },
+    {
+      "grad_norm": 0.2896885275840759,
+      "learning_rate": 9.597397453351076e-05,
+      "loss": 0.0158,
+      "step": 8610
+    },
+    {
+      "grad_norm": 0.2776479721069336,
+      "learning_rate": 9.596096364370087e-05,
+      "loss": 0.0144,
+      "step": 8620
+    },
+    {
+      "grad_norm": 0.28963086009025574,
+      "learning_rate": 9.594793264900399e-05,
+      "loss": 0.0167,
+      "step": 8630
+    },
+    {
+      "grad_norm": 0.3241257071495056,
+      "learning_rate": 9.593488155512032e-05,
+      "loss": 0.0166,
+      "step": 8640
+    },
+    {
+      "grad_norm": 0.31735771894454956,
+      "learning_rate": 9.592181036775886e-05,
+      "loss": 0.0163,
+      "step": 8650
+    },
+    {
+      "grad_norm": 0.2794424295425415,
+      "learning_rate": 9.590871909263741e-05,
+      "loss": 0.016,
+      "step": 8660
+    },
+    {
+      "grad_norm": 0.308666855096817,
+      "learning_rate": 9.589560773548252e-05,
+      "loss": 0.0172,
+      "step": 8670
+    },
+    {
+      "grad_norm": 0.337235689163208,
+      "learning_rate": 9.588247630202952e-05,
+      "loss": 0.0169,
+      "step": 8680
+    },
+    {
+      "grad_norm": 0.21625541150569916,
+      "learning_rate": 9.586932479802258e-05,
+      "loss": 0.0163,
+      "step": 8690
+    },
+    {
+      "grad_norm": 0.22229833900928497,
+      "learning_rate": 9.585615322921462e-05,
+      "loss": 0.0145,
+      "step": 8700
+    },
+    {
+      "grad_norm": 0.28545504808425903,
+      "learning_rate": 9.58429616013673e-05,
+      "loss": 0.0172,
+      "step": 8710
+    },
+    {
+      "grad_norm": 0.2615208327770233,
+      "learning_rate": 9.58297499202511e-05,
+      "loss": 0.0164,
+      "step": 8720
+    },
+    {
+      "grad_norm": 0.3007999360561371,
+      "learning_rate": 9.581651819164525e-05,
+      "loss": 0.0151,
+      "step": 8730
+    },
+    {
+      "grad_norm": 0.27854955196380615,
+      "learning_rate": 9.580326642133777e-05,
+      "loss": 0.0163,
+      "step": 8740
+    },
+    {
+      "grad_norm": 0.256217360496521,
+      "learning_rate": 9.578999461512544e-05,
+      "loss": 0.0135,
+      "step": 8750
+    },
+    {
+      "grad_norm": 0.2539404034614563,
+      "learning_rate": 9.577670277881378e-05,
+      "loss": 0.0141,
+      "step": 8760
+    },
+    {
+      "grad_norm": 0.34555208683013916,
+      "learning_rate": 9.57633909182171e-05,
+      "loss": 0.0172,
+      "step": 8770
+    },
+    {
+      "grad_norm": 0.3139221668243408,
+      "learning_rate": 9.575005903915847e-05,
+      "loss": 0.0173,
+      "step": 8780
+    },
+    {
+      "grad_norm": 0.2815629839897156,
+      "learning_rate": 9.573670714746972e-05,
+      "loss": 0.0171,
+      "step": 8790
+    },
+    {
+      "grad_norm": 0.20435239374637604,
+      "learning_rate": 9.572333524899136e-05,
+      "loss": 0.0134,
+      "step": 8800
+    },
+    {
+      "grad_norm": 0.30334368348121643,
+      "learning_rate": 9.570994334957278e-05,
+      "loss": 0.0145,
+      "step": 8810
+    },
+    {
+      "grad_norm": 0.3218114674091339,
+      "learning_rate": 9.569653145507201e-05,
+      "loss": 0.0162,
+      "step": 8820
+    },
+    {
+      "grad_norm": 0.206560879945755,
+      "learning_rate": 9.568309957135587e-05,
+      "loss": 0.0158,
+      "step": 8830
+    },
+    {
+      "grad_norm": 0.3363226056098938,
+      "learning_rate": 9.566964770429995e-05,
+      "loss": 0.0188,
+      "step": 8840
+    },
+    {
+      "grad_norm": 0.3671949803829193,
+      "learning_rate": 9.565617585978853e-05,
+      "loss": 0.0183,
+      "step": 8850
+    },
+    {
+      "grad_norm": 0.23831966519355774,
+      "learning_rate": 9.564268404371466e-05,
+      "loss": 0.0167,
+      "step": 8860
+    },
+    {
+      "grad_norm": 0.2799449563026428,
+      "learning_rate": 9.562917226198013e-05,
+      "loss": 0.018,
+      "step": 8870
+    },
+    {
+      "grad_norm": 0.26393261551856995,
+      "learning_rate": 9.561564052049543e-05,
+      "loss": 0.0152,
+      "step": 8880
+    },
+    {
+      "grad_norm": 0.2501900792121887,
+      "learning_rate": 9.560208882517982e-05,
+      "loss": 0.0178,
+      "step": 8890
+    },
+    {
+      "grad_norm": 0.2461346536874771,
+      "learning_rate": 9.558851718196127e-05,
+      "loss": 0.0177,
+      "step": 8900
+    },
+    {
+      "grad_norm": 0.24427275359630585,
+      "learning_rate": 9.557492559677646e-05,
+      "loss": 0.0153,
+      "step": 8910
+    },
+    {
+      "grad_norm": 0.30132240056991577,
+      "learning_rate": 9.556131407557082e-05,
+      "loss": 0.0178,
+      "step": 8920
+    },
+    {
+      "grad_norm": 0.29414570331573486,
+      "learning_rate": 9.554768262429853e-05,
+      "loss": 0.0201,
+      "step": 8930
+    },
+    {
+      "grad_norm": 0.3106103241443634,
+      "learning_rate": 9.553403124892239e-05,
+      "loss": 0.0164,
+      "step": 8940
+    },
+    {
+      "grad_norm": 0.2973262071609497,
+      "learning_rate": 9.552035995541402e-05,
+      "loss": 0.0155,
+      "step": 8950
+    },
+    {
+      "grad_norm": 0.3418499231338501,
+      "learning_rate": 9.550666874975368e-05,
+      "loss": 0.0161,
+      "step": 8960
+    },
+    {
+      "grad_norm": 0.29414600133895874,
+      "learning_rate": 9.549295763793038e-05,
+      "loss": 0.0164,
+      "step": 8970
+    },
+    {
+      "grad_norm": 0.30453795194625854,
+      "learning_rate": 9.547922662594183e-05,
+      "loss": 0.0149,
+      "step": 8980
+    },
+    {
+      "grad_norm": 0.2804470658302307,
+      "learning_rate": 9.546547571979443e-05,
+      "loss": 0.0175,
+      "step": 8990
+    },
+    {
+      "grad_norm": 0.3709416389465332,
+      "learning_rate": 9.545170492550331e-05,
+      "loss": 0.0178,
+      "step": 9000
+    },
+    {
+      "grad_norm": 0.25225183367729187,
+      "learning_rate": 9.543791424909226e-05,
+      "loss": 0.0176,
+      "step": 9010
+    },
+    {
+      "grad_norm": 0.26969945430755615,
+      "learning_rate": 9.542410369659382e-05,
+      "loss": 0.0179,
+      "step": 9020
+    },
+    {
+      "grad_norm": 0.2743229269981384,
+      "learning_rate": 9.541027327404916e-05,
+      "loss": 0.0185,
+      "step": 9030
+    },
+    {
+      "grad_norm": 0.2960783541202545,
+      "learning_rate": 9.539642298750821e-05,
+      "loss": 0.0149,
+      "step": 9040
+    },
+    {
+      "grad_norm": 0.2864016890525818,
+      "learning_rate": 9.538255284302954e-05,
+      "loss": 0.0148,
+      "step": 9050
+    },
+    {
+      "grad_norm": 0.2913634181022644,
+      "learning_rate": 9.536866284668043e-05,
+      "loss": 0.0161,
+      "step": 9060
+    },
+    {
+      "grad_norm": 0.2844342291355133,
+      "learning_rate": 9.53547530045368e-05,
+      "loss": 0.0155,
+      "step": 9070
+    },
+    {
+      "grad_norm": 0.34117022156715393,
+      "learning_rate": 9.534082332268335e-05,
+      "loss": 0.0151,
+      "step": 9080
+    },
+    {
+      "grad_norm": 0.253232479095459,
+      "learning_rate": 9.532687380721334e-05,
+      "loss": 0.0151,
+      "step": 9090
+    },
+    {
+      "grad_norm": 0.2041856348514557,
+      "learning_rate": 9.53129044642288e-05,
+      "loss": 0.0153,
+      "step": 9100
+    },
+    {
+      "grad_norm": 0.2712419331073761,
+      "learning_rate": 9.529891529984039e-05,
+      "loss": 0.0143,
+      "step": 9110
+    },
+    {
+      "grad_norm": 0.2583564221858978,
+      "learning_rate": 9.528490632016743e-05,
+      "loss": 0.0159,
+      "step": 9120
+    },
+    {
+      "grad_norm": 0.258567750453949,
+      "learning_rate": 9.52708775313379e-05,
+      "loss": 0.015,
+      "step": 9130
+    },
+    {
+      "grad_norm": 0.22506636381149292,
+      "learning_rate": 9.525682893948853e-05,
+      "loss": 0.0135,
+      "step": 9140
+    },
+    {
+      "grad_norm": 0.3068717122077942,
+      "learning_rate": 9.524276055076463e-05,
+      "loss": 0.0167,
+      "step": 9150
+    },
+    {
+      "grad_norm": 0.2987333834171295,
+      "learning_rate": 9.522867237132016e-05,
+      "loss": 0.0132,
+      "step": 9160
+    },
+    {
+      "grad_norm": 0.21940146386623383,
+      "learning_rate": 9.52145644073178e-05,
+      "loss": 0.0146,
+      "step": 9170
+    },
+    {
+      "grad_norm": 0.2841590940952301,
+      "learning_rate": 9.520043666492884e-05,
+      "loss": 0.0138,
+      "step": 9180
+    },
+    {
+      "grad_norm": 0.29749977588653564,
+      "learning_rate": 9.518628915033323e-05,
+      "loss": 0.0163,
+      "step": 9190
+    },
+    {
+      "grad_norm": 0.2661469280719757,
+      "learning_rate": 9.517212186971957e-05,
+      "loss": 0.0154,
+      "step": 9200
+    },
+    {
+      "grad_norm": 0.1948835551738739,
+      "learning_rate": 9.515793482928515e-05,
+      "loss": 0.0165,
+      "step": 9210
+    },
+    {
+      "grad_norm": 0.2540862560272217,
+      "learning_rate": 9.514372803523582e-05,
+      "loss": 0.015,
+      "step": 9220
+    },
+    {
+      "grad_norm": 0.2740703821182251,
+      "learning_rate": 9.512950149378613e-05,
+      "loss": 0.0154,
+      "step": 9230
+    },
+    {
+      "grad_norm": 0.23598672449588776,
+      "learning_rate": 9.511525521115925e-05,
+      "loss": 0.0129,
+      "step": 9240
+    },
+    {
+      "grad_norm": 0.26309236884117126,
+      "learning_rate": 9.510098919358698e-05,
+      "loss": 0.0149,
+      "step": 9250
+    },
+    {
+      "grad_norm": 0.23764029145240784,
+      "learning_rate": 9.508670344730979e-05,
+      "loss": 0.0171,
+      "step": 9260
+    },
+    {
+      "grad_norm": 0.2780969738960266,
+      "learning_rate": 9.507239797857672e-05,
+      "loss": 0.0183,
+      "step": 9270
+    },
+    {
+      "grad_norm": 0.3561362326145172,
+      "learning_rate": 9.505807279364548e-05,
+      "loss": 0.0159,
+      "step": 9280
+    },
+    {
+      "grad_norm": 0.32771429419517517,
+      "learning_rate": 9.504372789878239e-05,
+      "loss": 0.0149,
+      "step": 9290
+    },
+    {
+      "grad_norm": 0.30190345644950867,
+      "learning_rate": 9.502936330026239e-05,
+      "loss": 0.0163,
+      "step": 9300
+    },
+    {
+      "grad_norm": 0.15261222422122955,
+      "learning_rate": 9.501497900436906e-05,
+      "loss": 0.0142,
+      "step": 9310
+    },
+    {
+      "grad_norm": 0.3154904246330261,
+      "learning_rate": 9.500057501739457e-05,
+      "loss": 0.0127,
+      "step": 9320
+    },
+    {
+      "grad_norm": 0.23921480774879456,
+      "learning_rate": 9.49861513456397e-05,
+      "loss": 0.0139,
+      "step": 9330
+    },
+    {
+      "grad_norm": 0.24217601120471954,
+      "learning_rate": 9.497170799541388e-05,
+      "loss": 0.0177,
+      "step": 9340
+    },
+    {
+      "grad_norm": 0.22302724421024323,
+      "learning_rate": 9.495724497303508e-05,
+      "loss": 0.016,
+      "step": 9350
+    },
+    {
+      "grad_norm": 0.2498113512992859,
+      "learning_rate": 9.494276228482998e-05,
+      "loss": 0.0137,
+      "step": 9360
+    },
+    {
+      "grad_norm": 0.18652917444705963,
+      "learning_rate": 9.492825993713374e-05,
+      "loss": 0.0136,
+      "step": 9370
+    },
+    {
+      "grad_norm": 0.24613749980926514,
+      "learning_rate": 9.491373793629023e-05,
+      "loss": 0.0125,
+      "step": 9380
+    },
+    {
+      "grad_norm": 0.29064521193504333,
+      "learning_rate": 9.489919628865182e-05,
+      "loss": 0.014,
+      "step": 9390
+    },
+    {
+      "grad_norm": 0.2801339030265808,
+      "learning_rate": 9.488463500057955e-05,
+      "loss": 0.0152,
+      "step": 9400
+    },
+    {
+      "grad_norm": 0.3251652717590332,
+      "learning_rate": 9.487005407844302e-05,
+      "loss": 0.0154,
+      "step": 9410
+    },
+    {
+      "grad_norm": 0.2969908118247986,
+      "learning_rate": 9.485545352862039e-05,
+      "loss": 0.013,
+      "step": 9420
+    },
+    {
+      "grad_norm": 0.23512700200080872,
+      "learning_rate": 9.48408333574985e-05,
+      "loss": 0.0134,
+      "step": 9430
+    },
+    {
+      "grad_norm": 0.2688489854335785,
+      "learning_rate": 9.482619357147264e-05,
+      "loss": 0.014,
+      "step": 9440
+    },
+    {
+      "grad_norm": 0.29437384009361267,
+      "learning_rate": 9.481153417694679e-05,
+      "loss": 0.0141,
+      "step": 9450
+    },
+    {
+      "grad_norm": 0.22547294199466705,
+      "learning_rate": 9.479685518033347e-05,
+      "loss": 0.0155,
+      "step": 9460
+    },
+    {
+      "grad_norm": 0.2829169034957886,
+      "learning_rate": 9.478215658805375e-05,
+      "loss": 0.0143,
+      "step": 9470
+    },
+    {
+      "grad_norm": 0.27697646617889404,
+      "learning_rate": 9.47674384065373e-05,
+      "loss": 0.0138,
+      "step": 9480
+    },
+    {
+      "grad_norm": 0.213151752948761,
+      "learning_rate": 9.475270064222237e-05,
+      "loss": 0.015,
+      "step": 9490
+    },
+    {
+      "grad_norm": 0.20435881614685059,
+      "learning_rate": 9.473794330155572e-05,
+      "loss": 0.0123,
+      "step": 9500
+    },
+    {
+      "grad_norm": 0.25518450140953064,
+      "learning_rate": 9.472316639099276e-05,
+      "loss": 0.0154,
+      "step": 9510
+    },
+    {
+      "grad_norm": 0.295439749956131,
+      "learning_rate": 9.470836991699739e-05,
+      "loss": 0.0153,
+      "step": 9520
+    },
+    {
+      "grad_norm": 0.28492432832717896,
+      "learning_rate": 9.469355388604208e-05,
+      "loss": 0.0133,
+      "step": 9530
+    },
+    {
+      "grad_norm": 0.27607154846191406,
+      "learning_rate": 9.467871830460787e-05,
+      "loss": 0.0142,
+      "step": 9540
+    },
+    {
+      "grad_norm": 0.29151293635368347,
+      "learning_rate": 9.466386317918436e-05,
+      "loss": 0.0135,
+      "step": 9550
+    },
+    {
+      "grad_norm": 0.2986663281917572,
+      "learning_rate": 9.464898851626969e-05,
+      "loss": 0.013,
+      "step": 9560
+    },
+    {
+      "grad_norm": 0.24253949522972107,
+      "learning_rate": 9.463409432237051e-05,
+      "loss": 0.0144,
+      "step": 9570
+    },
+    {
+      "grad_norm": 0.20058345794677734,
+      "learning_rate": 9.461918060400209e-05,
+      "loss": 0.0147,
+      "step": 9580
+    },
+    {
+      "grad_norm": 0.3138510584831238,
+      "learning_rate": 9.460424736768816e-05,
+      "loss": 0.0144,
+      "step": 9590
+    },
+    {
+      "grad_norm": 0.32172515988349915,
+      "learning_rate": 9.458929461996105e-05,
+      "loss": 0.0142,
+      "step": 9600
+    },
+    {
+      "grad_norm": 0.26138147711753845,
+      "learning_rate": 9.457432236736158e-05,
+      "loss": 0.0125,
+      "step": 9610
+    },
+    {
+      "grad_norm": 0.25788918137550354,
+      "learning_rate": 9.455933061643916e-05,
+      "loss": 0.0145,
+      "step": 9620
+    },
+    {
+      "grad_norm": 0.22983905673027039,
+      "learning_rate": 9.454431937375164e-05,
+      "loss": 0.014,
+      "step": 9630
+    },
+    {
+      "grad_norm": 0.28231507539749146,
+      "learning_rate": 9.452928864586547e-05,
+      "loss": 0.0141,
+      "step": 9640
+    },
+    {
+      "grad_norm": 0.28121674060821533,
+      "learning_rate": 9.451423843935563e-05,
+      "loss": 0.0136,
+      "step": 9650
+    },
+    {
+      "grad_norm": 0.24544131755828857,
+      "learning_rate": 9.449916876080553e-05,
+      "loss": 0.0123,
+      "step": 9660
+    },
+    {
+      "grad_norm": 0.214189350605011,
+      "learning_rate": 9.44840796168072e-05,
+      "loss": 0.0141,
+      "step": 9670
+    },
+    {
+      "grad_norm": 0.15472345054149628,
+      "learning_rate": 9.446897101396115e-05,
+      "loss": 0.0141,
+      "step": 9680
+    },
+    {
+      "grad_norm": 0.29675644636154175,
+      "learning_rate": 9.445384295887638e-05,
+      "loss": 0.0133,
+      "step": 9690
+    },
+    {
+      "grad_norm": 0.26749470829963684,
+      "learning_rate": 9.443869545817043e-05,
+      "loss": 0.0134,
+      "step": 9700
+    },
+    {
+      "grad_norm": 0.269154816865921,
+      "learning_rate": 9.442352851846929e-05,
+      "loss": 0.0143,
+      "step": 9710
+    },
+    {
+      "grad_norm": 0.2850525677204132,
+      "learning_rate": 9.440834214640755e-05,
+      "loss": 0.0137,
+      "step": 9720
+    },
+    {
+      "grad_norm": 0.2736743688583374,
+      "learning_rate": 9.439313634862823e-05,
+      "loss": 0.0126,
+      "step": 9730
+    },
+    {
+      "grad_norm": 0.29700276255607605,
+      "learning_rate": 9.437791113178282e-05,
+      "loss": 0.017,
+      "step": 9740
+    },
+    {
+      "grad_norm": 0.24303904175758362,
+      "learning_rate": 9.43626665025314e-05,
+      "loss": 0.0162,
+      "step": 9750
+    },
+    {
+      "grad_norm": 0.26754263043403625,
+      "learning_rate": 9.434740246754248e-05,
+      "loss": 0.0145,
+      "step": 9760
+    },
+    {
+      "grad_norm": 0.2721576988697052,
+      "learning_rate": 9.433211903349304e-05,
+      "loss": 0.0144,
+      "step": 9770
+    },
+    {
+      "grad_norm": 0.30301353335380554,
+      "learning_rate": 9.431681620706858e-05,
+      "loss": 0.0179,
+      "step": 9780
+    },
+    {
+      "grad_norm": 0.3498634994029999,
+      "learning_rate": 9.43014939949631e-05,
+      "loss": 0.0161,
+      "step": 9790
+    },
+    {
+      "grad_norm": 0.30888277292251587,
+      "learning_rate": 9.428615240387904e-05,
+      "loss": 0.0129,
+      "step": 9800
+    },
+    {
+      "grad_norm": 0.27331918478012085,
+      "learning_rate": 9.427079144052732e-05,
+      "loss": 0.0124,
+      "step": 9810
+    },
+    {
+      "grad_norm": 0.29742905497550964,
+      "learning_rate": 9.425541111162739e-05,
+      "loss": 0.0125,
+      "step": 9820
+    },
+    {
+      "grad_norm": 0.2915572226047516,
+      "learning_rate": 9.424001142390709e-05,
+      "loss": 0.0131,
+      "step": 9830
+    },
+    {
+      "grad_norm": 0.1951620727777481,
+      "learning_rate": 9.422459238410277e-05,
+      "loss": 0.0139,
+      "step": 9840
+    },
+    {
+      "grad_norm": 0.22789807617664337,
+      "learning_rate": 9.420915399895926e-05,
+      "loss": 0.0163,
+      "step": 9850
+    },
+    {
+      "grad_norm": 0.28540289402008057,
+      "learning_rate": 9.419369627522981e-05,
+      "loss": 0.0133,
+      "step": 9860
+    },
+    {
+      "grad_norm": 0.28515905141830444,
+      "learning_rate": 9.417821921967618e-05,
+      "loss": 0.0155,
+      "step": 9870
+    },
+    {
+      "grad_norm": 0.2680476903915405,
+      "learning_rate": 9.416272283906855e-05,
+      "loss": 0.0159,
+      "step": 9880
+    },
+    {
+      "grad_norm": 0.31304964423179626,
+      "learning_rate": 9.414720714018554e-05,
+      "loss": 0.0169,
+      "step": 9890
+    },
+    {
+      "grad_norm": 0.3319147825241089,
+      "learning_rate": 9.413167212981427e-05,
+      "loss": 0.0165,
+      "step": 9900
+    },
+    {
+      "grad_norm": 0.27870187163352966,
+      "learning_rate": 9.41161178147503e-05,
+      "loss": 0.015,
+      "step": 9910
+    },
+    {
+      "grad_norm": 0.2566712200641632,
+      "learning_rate": 9.410054420179755e-05,
+      "loss": 0.019,
+      "step": 9920
+    },
+    {
+      "grad_norm": 0.27133211493492126,
+      "learning_rate": 9.408495129776851e-05,
+      "loss": 0.0184,
+      "step": 9930
+    },
+    {
+      "grad_norm": 0.25104138255119324,
+      "learning_rate": 9.406933910948403e-05,
+      "loss": 0.0153,
+      "step": 9940
+    },
+    {
+      "grad_norm": 0.28210002183914185,
+      "learning_rate": 9.40537076437734e-05,
+      "loss": 0.0161,
+      "step": 9950
+    },
+    {
+      "grad_norm": 0.21537400782108307,
+      "learning_rate": 9.403805690747436e-05,
+      "loss": 0.0142,
+      "step": 9960
+    },
+    {
+      "grad_norm": 0.32900547981262207,
+      "learning_rate": 9.402238690743308e-05,
+      "loss": 0.0137,
+      "step": 9970
+    },
+    {
+      "grad_norm": 0.25202223658561707,
+      "learning_rate": 9.400669765050413e-05,
+      "loss": 0.016,
+      "step": 9980
+    },
+    {
+      "grad_norm": 0.27597740292549133,
+      "learning_rate": 9.399098914355055e-05,
+      "loss": 0.0147,
+      "step": 9990
+    },
+    {
+      "grad_norm": 0.24091102182865143,
+      "learning_rate": 9.397526139344378e-05,
+      "loss": 0.014,
+      "step": 10000
+    },
+    {
+      "grad_norm": 0.35163599252700806,
+      "learning_rate": 9.395951440706362e-05,
+      "loss": 0.013,
+      "step": 10010
+    },
+    {
+      "grad_norm": 0.31008997559547424,
+      "learning_rate": 9.394374819129839e-05,
+      "loss": 0.014,
+      "step": 10020
+    },
+    {
+      "grad_norm": 0.2910120189189911,
+      "learning_rate": 9.392796275304474e-05,
+      "loss": 0.014,
+      "step": 10030
+    },
+    {
+      "grad_norm": 0.24214985966682434,
+      "learning_rate": 9.391215809920778e-05,
+      "loss": 0.0134,
+      "step": 10040
+    },
+    {
+      "grad_norm": 0.2688451111316681,
+      "learning_rate": 9.389633423670099e-05,
+      "loss": 0.0137,
+      "step": 10050
+    },
+    {
+      "grad_norm": 0.17426806688308716,
+      "learning_rate": 9.388049117244626e-05,
+      "loss": 0.0129,
+      "step": 10060
+    },
+    {
+      "grad_norm": 0.32833048701286316,
+      "learning_rate": 9.386462891337389e-05,
+      "loss": 0.0139,
+      "step": 10070
+    },
+    {
+      "grad_norm": 0.2213020622730255,
+      "learning_rate": 9.384874746642257e-05,
+      "loss": 0.0162,
+      "step": 10080
+    },
+    {
+      "grad_norm": 0.27719980478286743,
+      "learning_rate": 9.383284683853937e-05,
+      "loss": 0.0148,
+      "step": 10090
+    },
+    {
+      "grad_norm": 0.2506389319896698,
+      "learning_rate": 9.381692703667981e-05,
+      "loss": 0.017,
+      "step": 10100
+    },
+    {
+      "grad_norm": 0.3613995313644409,
+      "learning_rate": 9.380098806780771e-05,
+      "loss": 0.0153,
+      "step": 10110
+    },
+    {
+      "grad_norm": 0.3200376033782959,
+      "learning_rate": 9.378502993889533e-05,
+      "loss": 0.0145,
+      "step": 10120
+    },
+    {
+      "grad_norm": 0.2464158684015274,
+      "learning_rate": 9.376905265692329e-05,
+      "loss": 0.0139,
+      "step": 10130
+    },
+    {
+      "grad_norm": 0.26729506254196167,
+      "learning_rate": 9.37530562288806e-05,
+      "loss": 0.0141,
+      "step": 10140
+    },
+    {
+      "grad_norm": 0.22871346771717072,
+      "learning_rate": 9.373704066176465e-05,
+      "loss": 0.0126,
+      "step": 10150
+    },
+    {
+      "grad_norm": 0.2288282960653305,
+      "learning_rate": 9.372100596258118e-05,
+      "loss": 0.0139,
+      "step": 10160
+    },
+    {
+      "grad_norm": 0.27285706996917725,
+      "learning_rate": 9.370495213834433e-05,
+      "loss": 0.0174,
+      "step": 10170
+    },
+    {
+      "grad_norm": 0.27901002764701843,
+      "learning_rate": 9.368887919607657e-05,
+      "loss": 0.0143,
+      "step": 10180
+    },
+    {
+      "grad_norm": 0.2604979872703552,
+      "learning_rate": 9.367278714280876e-05,
+      "loss": 0.0127,
+      "step": 10190
+    },
+    {
+      "grad_norm": 0.25182247161865234,
+      "learning_rate": 9.36566759855801e-05,
+      "loss": 0.0139,
+      "step": 10200
+    },
+    {
+      "grad_norm": 0.22035740315914154,
+      "learning_rate": 9.36405457314382e-05,
+      "loss": 0.014,
+      "step": 10210
+    },
+    {
+      "grad_norm": 0.25344154238700867,
+      "learning_rate": 9.36243963874389e-05,
+      "loss": 0.0149,
+      "step": 10220
+    },
+    {
+      "grad_norm": 0.20783038437366486,
+      "learning_rate": 9.360822796064655e-05,
+      "loss": 0.0132,
+      "step": 10230
+    },
+    {
+      "grad_norm": 0.22307534515857697,
+      "learning_rate": 9.359204045813372e-05,
+      "loss": 0.0136,
+      "step": 10240
+    },
+    {
+      "grad_norm": 0.2457592934370041,
+      "learning_rate": 9.357583388698141e-05,
+      "loss": 0.012,
+      "step": 10250
+    },
+    {
+      "grad_norm": 0.19754335284233093,
+      "learning_rate": 9.35596082542789e-05,
+      "loss": 0.0149,
+      "step": 10260
+    },
+    {
+      "grad_norm": 0.2506454885005951,
+      "learning_rate": 9.354336356712383e-05,
+      "loss": 0.0143,
+      "step": 10270
+    },
+    {
+      "grad_norm": 0.2664726972579956,
+      "learning_rate": 9.35270998326222e-05,
+      "loss": 0.0165,
+      "step": 10280
+    },
+    {
+      "grad_norm": 0.25301432609558105,
+      "learning_rate": 9.351081705788831e-05,
+      "loss": 0.0163,
+      "step": 10290
+    },
+    {
+      "grad_norm": 0.2662853002548218,
+      "learning_rate": 9.349451525004477e-05,
+      "loss": 0.0158,
+      "step": 10300
+    },
+    {
+      "grad_norm": 0.2716609537601471,
+      "learning_rate": 9.347819441622261e-05,
+      "loss": 0.0146,
+      "step": 10310
+    },
+    {
+      "grad_norm": 0.26468801498413086,
+      "learning_rate": 9.346185456356105e-05,
+      "loss": 0.0128,
+      "step": 10320
+    },
+    {
+      "grad_norm": 0.349569171667099,
+      "learning_rate": 9.344549569920774e-05,
+      "loss": 0.0167,
+      "step": 10330
+    },
+    {
+      "grad_norm": 0.2427150160074234,
+      "learning_rate": 9.342911783031858e-05,
+      "loss": 0.0137,
+      "step": 10340
+    },
+    {
+      "grad_norm": 0.30224916338920593,
+      "learning_rate": 9.341272096405782e-05,
+      "loss": 0.0141,
+      "step": 10350
+    },
+    {
+      "grad_norm": 0.27465322613716125,
+      "learning_rate": 9.3396305107598e-05,
+      "loss": 0.0133,
+      "step": 10360
+    },
+    {
+      "grad_norm": 0.25351932644844055,
+      "learning_rate": 9.337987026811998e-05,
+      "loss": 0.014,
+      "step": 10370
+    },
+    {
+      "grad_norm": 0.2038041651248932,
+      "learning_rate": 9.33634164528129e-05,
+      "loss": 0.0123,
+      "step": 10380
+    },
+    {
+      "grad_norm": 0.22355180978775024,
+      "learning_rate": 9.334694366887424e-05,
+      "loss": 0.015,
+      "step": 10390
+    },
+    {
+      "grad_norm": 0.288185715675354,
+      "learning_rate": 9.333045192350973e-05,
+      "loss": 0.0154,
+      "step": 10400
+    },
+    {
+      "grad_norm": 0.2925601601600647,
+      "learning_rate": 9.331394122393345e-05,
+      "loss": 0.0141,
+      "step": 10410
+    },
+    {
+      "grad_norm": 0.333363801240921,
+      "learning_rate": 9.329741157736771e-05,
+      "loss": 0.0139,
+      "step": 10420
+    },
+    {
+      "grad_norm": 0.1867290884256363,
+      "learning_rate": 9.328086299104317e-05,
+      "loss": 0.0124,
+      "step": 10430
+    },
+    {
+      "grad_norm": 0.23186808824539185,
+      "learning_rate": 9.326429547219872e-05,
+      "loss": 0.0126,
+      "step": 10440
+    },
+    {
+      "grad_norm": 0.23887453973293304,
+      "learning_rate": 9.324770902808155e-05,
+      "loss": 0.0129,
+      "step": 10450
+    },
+    {
+      "grad_norm": 0.23415963351726532,
+      "learning_rate": 9.323110366594717e-05,
+      "loss": 0.0146,
+      "step": 10460
+    },
+    {
+      "grad_norm": 0.23775692284107208,
+      "learning_rate": 9.32144793930593e-05,
+      "loss": 0.0135,
+      "step": 10470
+    },
+    {
+      "grad_norm": 0.31388434767723083,
+      "learning_rate": 9.319783621668996e-05,
+      "loss": 0.0133,
+      "step": 10480
+    },
+    {
+      "grad_norm": 0.21948568522930145,
+      "learning_rate": 9.318117414411947e-05,
+      "loss": 0.0144,
+      "step": 10490
+    },
+    {
+      "grad_norm": 0.19564461708068848,
+      "learning_rate": 9.316449318263635e-05,
+      "loss": 0.0173,
+      "step": 10500
+    },
+    {
+      "grad_norm": 0.15975500643253326,
+      "learning_rate": 9.314779333953744e-05,
+      "loss": 0.0112,
+      "step": 10510
+    },
+    {
+      "grad_norm": 0.23147901892662048,
+      "learning_rate": 9.313107462212781e-05,
+      "loss": 0.0124,
+      "step": 10520
+    },
+    {
+      "grad_norm": 0.29941844940185547,
+      "learning_rate": 9.311433703772082e-05,
+      "loss": 0.0145,
+      "step": 10530
+    },
+    {
+      "grad_norm": 0.27438974380493164,
+      "learning_rate": 9.3097580593638e-05,
+      "loss": 0.0155,
+      "step": 10540
+    },
+    {
+      "grad_norm": 0.27903836965560913,
+      "learning_rate": 9.308080529720926e-05,
+      "loss": 0.0142,
+      "step": 10550
+    },
+    {
+      "grad_norm": 0.2277524620294571,
+      "learning_rate": 9.306401115577264e-05,
+      "loss": 0.0135,
+      "step": 10560
+    },
+    {
+      "grad_norm": 0.27457812428474426,
+      "learning_rate": 9.304719817667447e-05,
+      "loss": 0.0137,
+      "step": 10570
+    },
+    {
+      "grad_norm": 0.24671700596809387,
+      "learning_rate": 9.303036636726934e-05,
+      "loss": 0.0129,
+      "step": 10580
+    },
+    {
+      "grad_norm": 0.21143518388271332,
+      "learning_rate": 9.301351573492003e-05,
+      "loss": 0.0131,
+      "step": 10590
+    },
+    {
+      "grad_norm": 0.24725928902626038,
+      "learning_rate": 9.299664628699758e-05,
+      "loss": 0.0126,
+      "step": 10600
+    },
+    {
+      "grad_norm": 0.24132879078388214,
+      "learning_rate": 9.297975803088129e-05,
+      "loss": 0.0121,
+      "step": 10610
+    },
+    {
+      "grad_norm": 0.23140601813793182,
+      "learning_rate": 9.296285097395864e-05,
+      "loss": 0.0153,
+      "step": 10620
+    },
+    {
+      "grad_norm": 0.25162768363952637,
+      "learning_rate": 9.294592512362533e-05,
+      "loss": 0.0148,
+      "step": 10630
+    },
+    {
+      "grad_norm": 0.2920922338962555,
+      "learning_rate": 9.292898048728533e-05,
+      "loss": 0.0159,
+      "step": 10640
+    },
+    {
+      "grad_norm": 0.28566068410873413,
+      "learning_rate": 9.29120170723508e-05,
+      "loss": 0.0154,
+      "step": 10650
+    },
+    {
+      "grad_norm": 0.22458794713020325,
+      "learning_rate": 9.28950348862421e-05,
+      "loss": 0.0121,
+      "step": 10660
+    },
+    {
+      "grad_norm": 0.3077460825443268,
+      "learning_rate": 9.287803393638781e-05,
+      "loss": 0.0128,
+      "step": 10670
+    },
+    {
+      "grad_norm": 0.24980658292770386,
+      "learning_rate": 9.286101423022474e-05,
+      "loss": 0.0147,
+      "step": 10680
+    },
+    {
+      "grad_norm": 0.23122701048851013,
+      "learning_rate": 9.284397577519788e-05,
+      "loss": 0.0141,
+      "step": 10690
+    },
+    {
+      "grad_norm": 0.22616548836231232,
+      "learning_rate": 9.282691857876043e-05,
+      "loss": 0.0132,
+      "step": 10700
+    },
+    {
+      "grad_norm": 0.2817981243133545,
+      "learning_rate": 9.280984264837377e-05,
+      "loss": 0.0142,
+      "step": 10710
+    },
+    {
+      "grad_norm": 0.2066621035337448,
+      "learning_rate": 9.279274799150752e-05,
+      "loss": 0.0146,
+      "step": 10720
+    },
+    {
+      "grad_norm": 0.2782624065876007,
+      "learning_rate": 9.277563461563945e-05,
+      "loss": 0.0144,
+      "step": 10730
+    },
+    {
+      "grad_norm": 0.24995705485343933,
+      "learning_rate": 9.275850252825555e-05,
+      "loss": 0.0146,
+      "step": 10740
+    },
+    {
+      "grad_norm": 0.30658167600631714,
+      "learning_rate": 9.274135173684994e-05,
+      "loss": 0.014,
+      "step": 10750
+    },
+    {
+      "grad_norm": 0.2611781060695648,
+      "learning_rate": 9.272418224892498e-05,
+      "loss": 0.0148,
+      "step": 10760
+    },
+    {
+      "grad_norm": 0.25174257159233093,
+      "learning_rate": 9.27069940719912e-05,
+      "loss": 0.013,
+      "step": 10770
+    },
+    {
+      "grad_norm": 0.2635747790336609,
+      "learning_rate": 9.268978721356727e-05,
+      "loss": 0.0114,
+      "step": 10780
+    },
+    {
+      "grad_norm": 0.32819390296936035,
+      "learning_rate": 9.267256168118008e-05,
+      "loss": 0.0139,
+      "step": 10790
+    },
+    {
+      "grad_norm": 0.29211723804473877,
+      "learning_rate": 9.265531748236463e-05,
+      "loss": 0.0146,
+      "step": 10800
+    },
+    {
+      "grad_norm": 0.2344823032617569,
+      "learning_rate": 9.263805462466416e-05,
+      "loss": 0.0147,
+      "step": 10810
+    },
+    {
+      "grad_norm": 0.22849930822849274,
+      "learning_rate": 9.262077311562998e-05,
+      "loss": 0.0133,
+      "step": 10820
+    },
+    {
+      "grad_norm": 0.23765994608402252,
+      "learning_rate": 9.260347296282165e-05,
+      "loss": 0.0133,
+      "step": 10830
+    },
+    {
+      "grad_norm": 0.31116771697998047,
+      "learning_rate": 9.258615417380683e-05,
+      "loss": 0.018,
+      "step": 10840
+    },
+    {
+      "grad_norm": 0.3743302524089813,
+      "learning_rate": 9.256881675616133e-05,
+      "loss": 0.0188,
+      "step": 10850
+    },
+    {
+      "grad_norm": 0.2636663317680359,
+      "learning_rate": 9.255146071746917e-05,
+      "loss": 0.0158,
+      "step": 10860
+    },
+    {
+      "grad_norm": 0.2500058114528656,
+      "learning_rate": 9.253408606532241e-05,
+      "loss": 0.0151,
+      "step": 10870
+    },
+    {
+      "grad_norm": 0.33764392137527466,
+      "learning_rate": 9.251669280732137e-05,
+      "loss": 0.0157,
+      "step": 10880
+    },
+    {
+      "grad_norm": 0.25541290640830994,
+      "learning_rate": 9.249928095107441e-05,
+      "loss": 0.0142,
+      "step": 10890
+    },
+    {
+      "grad_norm": 0.25813060998916626,
+      "learning_rate": 9.248185050419811e-05,
+      "loss": 0.0139,
+      "step": 10900
+    },
+    {
+      "grad_norm": 0.26618775725364685,
+      "learning_rate": 9.24644014743171e-05,
+      "loss": 0.0148,
+      "step": 10910
+    },
+    {
+      "grad_norm": 0.19412410259246826,
+      "learning_rate": 9.24469338690642e-05,
+      "loss": 0.0148,
+      "step": 10920
+    },
+    {
+      "grad_norm": 0.25289398431777954,
+      "learning_rate": 9.242944769608033e-05,
+      "loss": 0.0151,
+      "step": 10930
+    },
+    {
+      "grad_norm": 0.23862771689891815,
+      "learning_rate": 9.241194296301454e-05,
+      "loss": 0.0139,
+      "step": 10940
+    },
+    {
+      "grad_norm": 0.23038651049137115,
+      "learning_rate": 9.239441967752397e-05,
+      "loss": 0.0133,
+      "step": 10950
+    },
+    {
+      "grad_norm": 0.23498328030109406,
+      "learning_rate": 9.237687784727393e-05,
+      "loss": 0.0124,
+      "step": 10960
+    },
+    {
+      "grad_norm": 0.31601256132125854,
+      "learning_rate": 9.235931747993781e-05,
+      "loss": 0.0155,
+      "step": 10970
+    },
+    {
+      "grad_norm": 0.2308618128299713,
+      "learning_rate": 9.234173858319707e-05,
+      "loss": 0.0133,
+      "step": 10980
+    },
+    {
+      "grad_norm": 0.2721408009529114,
+      "learning_rate": 9.23241411647414e-05,
+      "loss": 0.0153,
+      "step": 10990
+    },
+    {
+      "grad_norm": 0.2536720931529999,
+      "learning_rate": 9.230652523226841e-05,
+      "loss": 0.0134,
+      "step": 11000
+    },
+    {
+      "grad_norm": 0.22098755836486816,
+      "learning_rate": 9.2288890793484e-05,
+      "loss": 0.0134,
+      "step": 11010
+    },
+    {
+      "grad_norm": 0.2314929962158203,
+      "learning_rate": 9.227123785610199e-05,
+      "loss": 0.0135,
+      "step": 11020
+    },
+    {
+      "grad_norm": 0.21171818673610687,
+      "learning_rate": 9.225356642784443e-05,
+      "loss": 0.0121,
+      "step": 11030
+    },
+    {
+      "grad_norm": 0.21902430057525635,
+      "learning_rate": 9.223587651644139e-05,
+      "loss": 0.0123,
+      "step": 11040
+    },
+    {
+      "grad_norm": 0.31117936968803406,
+      "learning_rate": 9.221816812963104e-05,
+      "loss": 0.0136,
+      "step": 11050
+    },
+    {
+      "grad_norm": 0.24061807990074158,
+      "learning_rate": 9.22004412751596e-05,
+      "loss": 0.0149,
+      "step": 11060
+    },
+    {
+      "grad_norm": 0.2613106966018677,
+      "learning_rate": 9.218269596078146e-05,
+      "loss": 0.0123,
+      "step": 11070
+    },
+    {
+      "grad_norm": 0.30412352085113525,
+      "learning_rate": 9.216493219425895e-05,
+      "loss": 0.0139,
+      "step": 11080
+    },
+    {
+      "grad_norm": 0.252042680978775,
+      "learning_rate": 9.21471499833626e-05,
+      "loss": 0.0124,
+      "step": 11090
+    },
+    {
+      "grad_norm": 0.26979923248291016,
+      "learning_rate": 9.212934933587094e-05,
+      "loss": 0.0134,
+      "step": 11100
+    },
+    {
+      "grad_norm": 0.3091415464878082,
+      "learning_rate": 9.211153025957056e-05,
+      "loss": 0.0126,
+      "step": 11110
+    },
+    {
+      "grad_norm": 0.2822752892971039,
+      "learning_rate": 9.209369276225614e-05,
+      "loss": 0.014,
+      "step": 11120
+    },
+    {
+      "grad_norm": 0.28876379132270813,
+      "learning_rate": 9.207583685173042e-05,
+      "loss": 0.0142,
+      "step": 11130
+    },
+    {
+      "grad_norm": 0.2675466239452362,
+      "learning_rate": 9.205796253580417e-05,
+      "loss": 0.0125,
+      "step": 11140
+    },
+    {
+      "grad_norm": 0.2900436818599701,
+      "learning_rate": 9.204006982229621e-05,
+      "loss": 0.0119,
+      "step": 11150
+    },
+    {
+      "grad_norm": 0.22882400453090668,
+      "learning_rate": 9.202215871903346e-05,
+      "loss": 0.0142,
+      "step": 11160
+    },
+    {
+      "grad_norm": 0.28485071659088135,
+      "learning_rate": 9.20042292338508e-05,
+      "loss": 0.0148,
+      "step": 11170
+    },
+    {
+      "grad_norm": 0.2479930967092514,
+      "learning_rate": 9.198628137459123e-05,
+      "loss": 0.0132,
+      "step": 11180
+    },
+    {
+      "grad_norm": 0.2547415494918823,
+      "learning_rate": 9.196831514910572e-05,
+      "loss": 0.0156,
+      "step": 11190
+    },
+    {
+      "grad_norm": 0.22205187380313873,
+      "learning_rate": 9.195033056525332e-05,
+      "loss": 0.0146,
+      "step": 11200
+    },
+    {
+      "grad_norm": 0.25888702273368835,
+      "learning_rate": 9.193232763090114e-05,
+      "loss": 0.0149,
+      "step": 11210
+    },
+    {
+      "grad_norm": 0.26368752121925354,
+      "learning_rate": 9.191430635392422e-05,
+      "loss": 0.0137,
+      "step": 11220
+    },
+    {
+      "grad_norm": 0.3016478717327118,
+      "learning_rate": 9.18962667422057e-05,
+      "loss": 0.0193,
+      "step": 11230
+    },
+    {
+      "grad_norm": 0.27413469552993774,
+      "learning_rate": 9.187820880363671e-05,
+      "loss": 0.0138,
+      "step": 11240
+    },
+    {
+      "grad_norm": 0.2510729134082794,
+      "learning_rate": 9.186013254611643e-05,
+      "loss": 0.0145,
+      "step": 11250
+    },
+    {
+      "grad_norm": 0.22679252922534943,
+      "learning_rate": 9.1842037977552e-05,
+      "loss": 0.013,
+      "step": 11260
+    },
+    {
+      "grad_norm": 0.22027374804019928,
+      "learning_rate": 9.182392510585862e-05,
+      "loss": 0.0131,
+      "step": 11270
+    },
+    {
+      "grad_norm": 0.28450873494148254,
+      "learning_rate": 9.180579393895946e-05,
+      "loss": 0.0112,
+      "step": 11280
+    },
+    {
+      "grad_norm": 0.26339906454086304,
+      "learning_rate": 9.178764448478572e-05,
+      "loss": 0.0119,
+      "step": 11290
+    },
+    {
+      "grad_norm": 0.24024119973182678,
+      "learning_rate": 9.176947675127658e-05,
+      "loss": 0.0113,
+      "step": 11300
+    },
+    {
+      "grad_norm": 0.27253109216690063,
+      "learning_rate": 9.175129074637924e-05,
+      "loss": 0.0124,
+      "step": 11310
+    },
+    {
+      "grad_norm": 0.41042640805244446,
+      "learning_rate": 9.173308647804884e-05,
+      "loss": 0.0118,
+      "step": 11320
+    },
+    {
+      "grad_norm": 0.23225905001163483,
+      "learning_rate": 9.171486395424859e-05,
+      "loss": 0.0121,
+      "step": 11330
+    },
+    {
+      "grad_norm": 0.2629808783531189,
+      "learning_rate": 9.16966231829496e-05,
+      "loss": 0.0135,
+      "step": 11340
+    },
+    {
+      "grad_norm": 0.23633228242397308,
+      "learning_rate": 9.167836417213105e-05,
+      "loss": 0.0144,
+      "step": 11350
+    },
+    {
+      "grad_norm": 0.2297479510307312,
+      "learning_rate": 9.166008692978001e-05,
+      "loss": 0.0131,
+      "step": 11360
+    },
+    {
+      "grad_norm": 0.22152197360992432,
+      "learning_rate": 9.164179146389158e-05,
+      "loss": 0.0145,
+      "step": 11370
+    },
+    {
+      "grad_norm": 0.24759851396083832,
+      "learning_rate": 9.162347778246882e-05,
+      "loss": 0.0145,
+      "step": 11380
+    },
+    {
+      "grad_norm": 0.27114227414131165,
+      "learning_rate": 9.160514589352276e-05,
+      "loss": 0.0141,
+      "step": 11390
+    },
+    {
+      "grad_norm": 0.3248614966869354,
+      "learning_rate": 9.15867958050724e-05,
+      "loss": 0.016,
+      "step": 11400
+    },
+    {
+      "grad_norm": 0.2459639608860016,
+      "learning_rate": 9.156842752514466e-05,
+      "loss": 0.0148,
+      "step": 11410
+    },
+    {
+      "grad_norm": 0.2511340081691742,
+      "learning_rate": 9.155004106177447e-05,
+      "loss": 0.013,
+      "step": 11420
+    },
+    {
+      "grad_norm": 0.23085638880729675,
+      "learning_rate": 9.153163642300471e-05,
+      "loss": 0.0128,
+      "step": 11430
+    },
+    {
+      "grad_norm": 0.20289812982082367,
+      "learning_rate": 9.151321361688616e-05,
+      "loss": 0.0124,
+      "step": 11440
+    },
+    {
+      "grad_norm": 0.20715966820716858,
+      "learning_rate": 9.149477265147762e-05,
+      "loss": 0.0134,
+      "step": 11450
+    },
+    {
+      "grad_norm": 0.2055046409368515,
+      "learning_rate": 9.147631353484574e-05,
+      "loss": 0.0144,
+      "step": 11460
+    },
+    {
+      "grad_norm": 0.23201784491539001,
+      "learning_rate": 9.145783627506522e-05,
+      "loss": 0.0121,
+      "step": 11470
+    },
+    {
+      "grad_norm": 0.2076769322156906,
+      "learning_rate": 9.143934088021861e-05,
+      "loss": 0.0123,
+      "step": 11480
+    },
+    {
+      "grad_norm": 0.21445706486701965,
+      "learning_rate": 9.142082735839645e-05,
+      "loss": 0.0134,
+      "step": 11490
+    },
+    {
+      "grad_norm": 0.26283207535743713,
+      "learning_rate": 9.140229571769715e-05,
+      "loss": 0.0127,
+      "step": 11500
+    },
+    {
+      "grad_norm": 0.2866324782371521,
+      "learning_rate": 9.138374596622709e-05,
+      "loss": 0.0136,
+      "step": 11510
+    },
+    {
+      "grad_norm": 0.2560141086578369,
+      "learning_rate": 9.136517811210059e-05,
+      "loss": 0.0139,
+      "step": 11520
+    },
+    {
+      "grad_norm": 0.292280912399292,
+      "learning_rate": 9.134659216343984e-05,
+      "loss": 0.0158,
+      "step": 11530
+    },
+    {
+      "grad_norm": 0.3215332627296448,
+      "learning_rate": 9.132798812837494e-05,
+      "loss": 0.0157,
+      "step": 11540
+    },
+    {
+      "grad_norm": 0.28316575288772583,
+      "learning_rate": 9.130936601504396e-05,
+      "loss": 0.0156,
+      "step": 11550
+    },
+    {
+      "grad_norm": 0.24701006710529327,
+      "learning_rate": 9.129072583159284e-05,
+      "loss": 0.0152,
+      "step": 11560
+    },
+    {
+      "grad_norm": 0.26256465911865234,
+      "learning_rate": 9.127206758617542e-05,
+      "loss": 0.0134,
+      "step": 11570
+    },
+    {
+      "grad_norm": 0.24059419333934784,
+      "learning_rate": 9.125339128695346e-05,
+      "loss": 0.0148,
+      "step": 11580
+    },
+    {
+      "grad_norm": 0.17729520797729492,
+      "learning_rate": 9.123469694209659e-05,
+      "loss": 0.0137,
+      "step": 11590
+    },
+    {
+      "grad_norm": 0.19218367338180542,
+      "learning_rate": 9.121598455978239e-05,
+      "loss": 0.0123,
+      "step": 11600
+    },
+    {
+      "grad_norm": 0.24817945063114166,
+      "learning_rate": 9.119725414819624e-05,
+      "loss": 0.0148,
+      "step": 11610
+    },
+    {
+      "grad_norm": 0.22954469919204712,
+      "learning_rate": 9.117850571553149e-05,
+      "loss": 0.0171,
+      "step": 11620
+    },
+    {
+      "grad_norm": 0.28983625769615173,
+      "learning_rate": 9.115973926998935e-05,
+      "loss": 0.0165,
+      "step": 11630
+    },
+    {
+      "grad_norm": 0.2506774365901947,
+      "learning_rate": 9.114095481977888e-05,
+      "loss": 0.0132,
+      "step": 11640
+    },
+    {
+      "grad_norm": 0.27042248845100403,
+      "learning_rate": 9.112215237311703e-05,
+      "loss": 0.0131,
+      "step": 11650
+    },
+    {
+      "grad_norm": 0.21814094483852386,
+      "learning_rate": 9.110333193822867e-05,
+      "loss": 0.0121,
+      "step": 11660
+    },
+    {
+      "grad_norm": 0.3850848078727722,
+      "learning_rate": 9.108449352334645e-05,
+      "loss": 0.0135,
+      "step": 11670
+    },
+    {
+      "grad_norm": 0.26791301369667053,
+      "learning_rate": 9.106563713671094e-05,
+      "loss": 0.012,
+      "step": 11680
+    },
+    {
+      "grad_norm": 0.26859238743782043,
+      "learning_rate": 9.104676278657061e-05,
+      "loss": 0.013,
+      "step": 11690
+    },
+    {
+      "grad_norm": 0.2259884774684906,
+      "learning_rate": 9.102787048118169e-05,
+      "loss": 0.0139,
+      "step": 11700
+    },
+    {
+      "grad_norm": 0.24317844212055206,
+      "learning_rate": 9.100896022880834e-05,
+      "loss": 0.013,
+      "step": 11710
+    },
+    {
+      "grad_norm": 0.21118058264255524,
+      "learning_rate": 9.099003203772254e-05,
+      "loss": 0.0128,
+      "step": 11720
+    },
+    {
+      "grad_norm": 0.1887577474117279,
+      "learning_rate": 9.097108591620413e-05,
+      "loss": 0.013,
+      "step": 11730
+    },
+    {
+      "grad_norm": 0.18447475135326385,
+      "learning_rate": 9.095212187254078e-05,
+      "loss": 0.0141,
+      "step": 11740
+    },
+    {
+      "grad_norm": 0.21144814789295197,
+      "learning_rate": 9.093313991502801e-05,
+      "loss": 0.0127,
+      "step": 11750
+    },
+    {
+      "grad_norm": 0.2570485770702362,
+      "learning_rate": 9.091414005196917e-05,
+      "loss": 0.0127,
+      "step": 11760
+    },
+    {
+      "grad_norm": 0.29918333888053894,
+      "learning_rate": 9.089512229167545e-05,
+      "loss": 0.0123,
+      "step": 11770
+    },
+    {
+      "grad_norm": 0.26613110303878784,
+      "learning_rate": 9.087608664246587e-05,
+      "loss": 0.0108,
+      "step": 11780
+    },
+    {
+      "grad_norm": 0.21688684821128845,
+      "learning_rate": 9.085703311266727e-05,
+      "loss": 0.0128,
+      "step": 11790
+    },
+    {
+      "grad_norm": 0.21538616716861725,
+      "learning_rate": 9.083796171061429e-05,
+      "loss": 0.0128,
+      "step": 11800
+    },
+    {
+      "grad_norm": 0.2209499031305313,
+      "learning_rate": 9.081887244464941e-05,
+      "loss": 0.0126,
+      "step": 11810
+    },
+    {
+      "grad_norm": 0.2372194528579712,
+      "learning_rate": 9.079976532312297e-05,
+      "loss": 0.0139,
+      "step": 11820
+    },
+    {
+      "grad_norm": 0.21154478192329407,
+      "learning_rate": 9.078064035439301e-05,
+      "loss": 0.014,
+      "step": 11830
+    },
+    {
+      "grad_norm": 0.25482606887817383,
+      "learning_rate": 9.07614975468255e-05,
+      "loss": 0.0138,
+      "step": 11840
+    },
+    {
+      "grad_norm": 0.18858803808689117,
+      "learning_rate": 9.074233690879412e-05,
+      "loss": 0.0136,
+      "step": 11850
+    },
+    {
+      "grad_norm": 0.28423166275024414,
+      "learning_rate": 9.072315844868038e-05,
+      "loss": 0.0127,
+      "step": 11860
+    },
+    {
+      "grad_norm": 0.21768033504486084,
+      "learning_rate": 9.07039621748736e-05,
+      "loss": 0.0154,
+      "step": 11870
+    },
+    {
+      "grad_norm": 0.23686261475086212,
+      "learning_rate": 9.06847480957709e-05,
+      "loss": 0.0147,
+      "step": 11880
+    },
+    {
+      "grad_norm": 0.290964275598526,
+      "learning_rate": 9.066551621977713e-05,
+      "loss": 0.0135,
+      "step": 11890
+    },
+    {
+      "grad_norm": 0.19140270352363586,
+      "learning_rate": 9.064626655530501e-05,
+      "loss": 0.0136,
+      "step": 11900
+    },
+    {
+      "grad_norm": 0.2873893976211548,
+      "learning_rate": 9.062699911077497e-05,
+      "loss": 0.0129,
+      "step": 11910
+    },
+    {
+      "grad_norm": 0.1849132627248764,
+      "learning_rate": 9.060771389461524e-05,
+      "loss": 0.0134,
+      "step": 11920
+    },
+    {
+      "grad_norm": 0.27869924902915955,
+      "learning_rate": 9.058841091526187e-05,
+      "loss": 0.0155,
+      "step": 11930
+    },
+    {
+      "grad_norm": 0.24860349297523499,
+      "learning_rate": 9.056909018115858e-05,
+      "loss": 0.0164,
+      "step": 11940
+    },
+    {
+      "grad_norm": 0.1981237679719925,
+      "learning_rate": 9.054975170075697e-05,
+      "loss": 0.0121,
+      "step": 11950
+    },
+    {
+      "grad_norm": 0.24947912991046906,
+      "learning_rate": 9.053039548251631e-05,
+      "loss": 0.0151,
+      "step": 11960
+    },
+    {
+      "grad_norm": 0.19595862925052643,
+      "learning_rate": 9.051102153490368e-05,
+      "loss": 0.0125,
+      "step": 11970
+    },
+    {
+      "grad_norm": 0.25750216841697693,
+      "learning_rate": 9.04916298663939e-05,
+      "loss": 0.0125,
+      "step": 11980
+    },
+    {
+      "grad_norm": 0.19051556289196014,
+      "learning_rate": 9.047222048546955e-05,
+      "loss": 0.011,
+      "step": 11990
+    },
+    {
+      "grad_norm": 0.19021835923194885,
+      "learning_rate": 9.045279340062097e-05,
+      "loss": 0.0142,
+      "step": 12000
+    },
+    {
+      "grad_norm": 0.22078993916511536,
+      "learning_rate": 9.043334862034618e-05,
+      "loss": 0.0141,
+      "step": 12010
+    },
+    {
+      "grad_norm": 0.2381676584482193,
+      "learning_rate": 9.041388615315102e-05,
+      "loss": 0.0126,
+      "step": 12020
+    },
+    {
+      "grad_norm": 0.22673667967319489,
+      "learning_rate": 9.039440600754905e-05,
+      "loss": 0.0118,
+      "step": 12030
+    },
+    {
+      "grad_norm": 0.21367882192134857,
+      "learning_rate": 9.037490819206151e-05,
+      "loss": 0.0117,
+      "step": 12040
+    },
+    {
+      "grad_norm": 0.281891405582428,
+      "learning_rate": 9.035539271521744e-05,
+      "loss": 0.0126,
+      "step": 12050
+    },
+    {
+      "grad_norm": 0.1749451607465744,
+      "learning_rate": 9.033585958555356e-05,
+      "loss": 0.0127,
+      "step": 12060
+    },
+    {
+      "grad_norm": 0.24947386980056763,
+      "learning_rate": 9.031630881161431e-05,
+      "loss": 0.0108,
+      "step": 12070
+    },
+    {
+      "grad_norm": 0.16668656468391418,
+      "learning_rate": 9.029674040195186e-05,
+      "loss": 0.0118,
+      "step": 12080
+    },
+    {
+      "grad_norm": 0.2863198518753052,
+      "learning_rate": 9.027715436512613e-05,
+      "loss": 0.0113,
+      "step": 12090
+    },
+    {
+      "grad_norm": 0.22714634239673615,
+      "learning_rate": 9.02575507097047e-05,
+      "loss": 0.0124,
+      "step": 12100
+    },
+    {
+      "grad_norm": 0.30033570528030396,
+      "learning_rate": 9.023792944426286e-05,
+      "loss": 0.0127,
+      "step": 12110
+    },
+    {
+      "grad_norm": 0.2730199694633484,
+      "learning_rate": 9.021829057738364e-05,
+      "loss": 0.0126,
+      "step": 12120
+    },
+    {
+      "grad_norm": 0.2726667523384094,
+      "learning_rate": 9.019863411765775e-05,
+      "loss": 0.0125,
+      "step": 12130
+    },
+    {
+      "grad_norm": 0.24281176924705505,
+      "learning_rate": 9.017896007368357e-05,
+      "loss": 0.0122,
+      "step": 12140
+    },
+    {
+      "grad_norm": 0.2433428019285202,
+      "learning_rate": 9.015926845406722e-05,
+      "loss": 0.0117,
+      "step": 12150
+    },
+    {
+      "grad_norm": 0.3382183015346527,
+      "learning_rate": 9.013955926742245e-05,
+      "loss": 0.0133,
+      "step": 12160
+    },
+    {
+      "grad_norm": 0.28900811076164246,
+      "learning_rate": 9.011983252237077e-05,
+      "loss": 0.0118,
+      "step": 12170
+    },
+    {
+      "grad_norm": 0.30618301033973694,
+      "learning_rate": 9.01000882275413e-05,
+      "loss": 0.0132,
+      "step": 12180
+    },
+    {
+      "grad_norm": 0.2641192078590393,
+      "learning_rate": 9.008032639157088e-05,
+      "loss": 0.011,
+      "step": 12190
+    },
+    {
+      "grad_norm": 0.21677494049072266,
+      "learning_rate": 9.006054702310401e-05,
+      "loss": 0.014,
+      "step": 12200
+    },
+    {
+      "grad_norm": 0.2295880764722824,
+      "learning_rate": 9.004075013079283e-05,
+      "loss": 0.0143,
+      "step": 12210
+    },
+    {
+      "grad_norm": 0.20400449633598328,
+      "learning_rate": 9.00209357232972e-05,
+      "loss": 0.0106,
+      "step": 12220
+    },
+    {
+      "grad_norm": 0.20092745125293732,
+      "learning_rate": 9.000110380928461e-05,
+      "loss": 0.0133,
+      "step": 12230
+    },
+    {
+      "grad_norm": 0.23945334553718567,
+      "learning_rate": 8.998125439743021e-05,
+      "loss": 0.0135,
+      "step": 12240
+    },
+    {
+      "grad_norm": 0.1785716414451599,
+      "learning_rate": 8.996138749641682e-05,
+      "loss": 0.0108,
+      "step": 12250
+    },
+    {
+      "grad_norm": 0.21711979806423187,
+      "learning_rate": 8.994150311493488e-05,
+      "loss": 0.0111,
+      "step": 12260
+    },
+    {
+      "grad_norm": 0.3127778172492981,
+      "learning_rate": 8.992160126168247e-05,
+      "loss": 0.0145,
+      "step": 12270
+    },
+    {
+      "grad_norm": 0.25316670536994934,
+      "learning_rate": 8.99016819453654e-05,
+      "loss": 0.011,
+      "step": 12280
+    },
+    {
+      "grad_norm": 0.19786453247070312,
+      "learning_rate": 8.988174517469702e-05,
+      "loss": 0.01,
+      "step": 12290
+    },
+    {
+      "grad_norm": 0.20878389477729797,
+      "learning_rate": 8.986179095839835e-05,
+      "loss": 0.015,
+      "step": 12300
+    },
+    {
+      "grad_norm": 0.24965909123420715,
+      "learning_rate": 8.984181930519804e-05,
+      "loss": 0.0132,
+      "step": 12310
+    },
+    {
+      "grad_norm": 0.22923129796981812,
+      "learning_rate": 8.982183022383237e-05,
+      "loss": 0.0117,
+      "step": 12320
+    },
+    {
+      "grad_norm": 0.22293172776699066,
+      "learning_rate": 8.980182372304525e-05,
+      "loss": 0.011,
+      "step": 12330
+    },
+    {
+      "grad_norm": 0.21359023451805115,
+      "learning_rate": 8.97817998115882e-05,
+      "loss": 0.0123,
+      "step": 12340
+    },
+    {
+      "grad_norm": 0.2526428997516632,
+      "learning_rate": 8.976175849822038e-05,
+      "loss": 0.0137,
+      "step": 12350
+    },
+    {
+      "grad_norm": 0.18424206972122192,
+      "learning_rate": 8.97416997917085e-05,
+      "loss": 0.0115,
+      "step": 12360
+    },
+    {
+      "grad_norm": 0.18353477120399475,
+      "learning_rate": 8.972162370082695e-05,
+      "loss": 0.0108,
+      "step": 12370
+    },
+    {
+      "grad_norm": 0.18540769815444946,
+      "learning_rate": 8.97015302343577e-05,
+      "loss": 0.0105,
+      "step": 12380
+    },
+    {
+      "grad_norm": 0.2159658521413803,
+      "learning_rate": 8.968141940109027e-05,
+      "loss": 0.0116,
+      "step": 12390
+    },
+    {
+      "grad_norm": 0.19713394343852997,
+      "learning_rate": 8.966129120982188e-05,
+      "loss": 0.012,
+      "step": 12400
+    },
+    {
+      "grad_norm": 0.24771293997764587,
+      "learning_rate": 8.964114566935724e-05,
+      "loss": 0.0101,
+      "step": 12410
+    },
+    {
+      "grad_norm": 0.2450655847787857,
+      "learning_rate": 8.962098278850871e-05,
+      "loss": 0.0118,
+      "step": 12420
+    },
+    {
+      "grad_norm": 0.2380683571100235,
+      "learning_rate": 8.960080257609622e-05,
+      "loss": 0.0134,
+      "step": 12430
+    },
+    {
+      "grad_norm": 0.2411668300628662,
+      "learning_rate": 8.95806050409473e-05,
+      "loss": 0.0106,
+      "step": 12440
+    },
+    {
+      "grad_norm": 0.24851037561893463,
+      "learning_rate": 8.9560390191897e-05,
+      "loss": 0.0114,
+      "step": 12450
+    },
+    {
+      "grad_norm": 0.17735406756401062,
+      "learning_rate": 8.9540158037788e-05,
+      "loss": 0.0103,
+      "step": 12460
+    },
+    {
+      "grad_norm": 0.2563645839691162,
+      "learning_rate": 8.951990858747054e-05,
+      "loss": 0.0142,
+      "step": 12470
+    },
+    {
+      "grad_norm": 0.20154759287834167,
+      "learning_rate": 8.94996418498024e-05,
+      "loss": 0.011,
+      "step": 12480
+    },
+    {
+      "grad_norm": 0.16451139748096466,
+      "learning_rate": 8.947935783364896e-05,
+      "loss": 0.0153,
+      "step": 12490
+    },
+    {
+      "grad_norm": 0.28778183460235596,
+      "learning_rate": 8.945905654788311e-05,
+      "loss": 0.0112,
+      "step": 12500
+    },
+    {
+      "grad_norm": 0.17721152305603027,
+      "learning_rate": 8.943873800138535e-05,
+      "loss": 0.0141,
+      "step": 12510
+    },
+    {
+      "grad_norm": 0.1580948531627655,
+      "learning_rate": 8.94184022030437e-05,
+      "loss": 0.0104,
+      "step": 12520
+    },
+    {
+      "grad_norm": 0.23480086028575897,
+      "learning_rate": 8.939804916175372e-05,
+      "loss": 0.0127,
+      "step": 12530
+    },
+    {
+      "grad_norm": 0.24772533774375916,
+      "learning_rate": 8.93776788864185e-05,
+      "loss": 0.0107,
+      "step": 12540
+    },
+    {
+      "grad_norm": 0.2690843939781189,
+      "learning_rate": 8.935729138594873e-05,
+      "loss": 0.0118,
+      "step": 12550
+    },
+    {
+      "grad_norm": 0.29743632674217224,
+      "learning_rate": 8.933688666926258e-05,
+      "loss": 0.0148,
+      "step": 12560
+    },
+    {
+      "grad_norm": 0.23219773173332214,
+      "learning_rate": 8.931646474528575e-05,
+      "loss": 0.0127,
+      "step": 12570
+    },
+    {
+      "grad_norm": 0.2770198583602905,
+      "learning_rate": 8.929602562295151e-05,
+      "loss": 0.0141,
+      "step": 12580
+    },
+    {
+      "grad_norm": 0.2357957661151886,
+      "learning_rate": 8.92755693112006e-05,
+      "loss": 0.0114,
+      "step": 12590
+    },
+    {
+      "grad_norm": 0.2321712076663971,
+      "learning_rate": 8.925509581898136e-05,
+      "loss": 0.0115,
+      "step": 12600
+    },
+    {
+      "grad_norm": 0.16979417204856873,
+      "learning_rate": 8.923460515524951e-05,
+      "loss": 0.0096,
+      "step": 12610
+    },
+    {
+      "grad_norm": 0.16053420305252075,
+      "learning_rate": 8.921409732896842e-05,
+      "loss": 0.01,
+      "step": 12620
+    },
+    {
+      "grad_norm": 0.2007959932088852,
+      "learning_rate": 8.919357234910887e-05,
+      "loss": 0.0125,
+      "step": 12630
+    },
+    {
+      "grad_norm": 0.21310065686702728,
+      "learning_rate": 8.917303022464923e-05,
+      "loss": 0.0132,
+      "step": 12640
+    },
+    {
+      "grad_norm": 0.16244758665561676,
+      "learning_rate": 8.915247096457531e-05,
+      "loss": 0.011,
+      "step": 12650
+    },
+    {
+      "grad_norm": 0.25555679202079773,
+      "learning_rate": 8.91318945778804e-05,
+      "loss": 0.0121,
+      "step": 12660
+    },
+    {
+      "grad_norm": 0.23894864320755005,
+      "learning_rate": 8.911130107356534e-05,
+      "loss": 0.0118,
+      "step": 12670
+    },
+    {
+      "grad_norm": 0.29733794927597046,
+      "learning_rate": 8.90906904606384e-05,
+      "loss": 0.0114,
+      "step": 12680
+    },
+    {
+      "grad_norm": 0.2364368438720703,
+      "learning_rate": 8.90700627481154e-05,
+      "loss": 0.0145,
+      "step": 12690
+    },
+    {
+      "grad_norm": 0.2178865522146225,
+      "learning_rate": 8.904941794501957e-05,
+      "loss": 0.0108,
+      "step": 12700
+    },
+    {
+      "grad_norm": 0.2216091752052307,
+      "learning_rate": 8.902875606038166e-05,
+      "loss": 0.012,
+      "step": 12710
+    },
+    {
+      "grad_norm": 0.22113750874996185,
+      "learning_rate": 8.900807710323989e-05,
+      "loss": 0.0146,
+      "step": 12720
+    },
+    {
+      "grad_norm": 0.20707811415195465,
+      "learning_rate": 8.898738108263993e-05,
+      "loss": 0.0139,
+      "step": 12730
+    },
+    {
+      "grad_norm": 0.2477085292339325,
+      "learning_rate": 8.896666800763491e-05,
+      "loss": 0.0103,
+      "step": 12740
+    },
+    {
+      "grad_norm": 0.1809002012014389,
+      "learning_rate": 8.894593788728546e-05,
+      "loss": 0.012,
+      "step": 12750
+    },
+    {
+      "grad_norm": 0.2794867753982544,
+      "learning_rate": 8.892519073065961e-05,
+      "loss": 0.0143,
+      "step": 12760
+    },
+    {
+      "grad_norm": 0.28247302770614624,
+      "learning_rate": 8.89044265468329e-05,
+      "loss": 0.014,
+      "step": 12770
+    },
+    {
+      "grad_norm": 0.23065543174743652,
+      "learning_rate": 8.888364534488827e-05,
+      "loss": 0.0127,
+      "step": 12780
+    },
+    {
+      "grad_norm": 0.21015337109565735,
+      "learning_rate": 8.886284713391613e-05,
+      "loss": 0.0137,
+      "step": 12790
+    },
+    {
+      "grad_norm": 0.26602649688720703,
+      "learning_rate": 8.884203192301431e-05,
+      "loss": 0.0135,
+      "step": 12800
+    },
+    {
+      "grad_norm": 0.2146083265542984,
+      "learning_rate": 8.88211997212881e-05,
+      "loss": 0.0115,
+      "step": 12810
+    },
+    {
+      "grad_norm": 0.2697013020515442,
+      "learning_rate": 8.880035053785023e-05,
+      "loss": 0.0127,
+      "step": 12820
+    },
+    {
+      "grad_norm": 0.2579370439052582,
+      "learning_rate": 8.877948438182083e-05,
+      "loss": 0.0119,
+      "step": 12830
+    },
+    {
+      "grad_norm": 0.23603367805480957,
+      "learning_rate": 8.875860126232745e-05,
+      "loss": 0.0114,
+      "step": 12840
+    },
+    {
+      "grad_norm": 0.24330082535743713,
+      "learning_rate": 8.87377011885051e-05,
+      "loss": 0.0132,
+      "step": 12850
+    },
+    {
+      "grad_norm": 0.18809005618095398,
+      "learning_rate": 8.871678416949617e-05,
+      "loss": 0.0119,
+      "step": 12860
+    },
+    {
+      "grad_norm": 0.2491377741098404,
+      "learning_rate": 8.869585021445046e-05,
+      "loss": 0.0126,
+      "step": 12870
+    },
+    {
+      "grad_norm": 0.24637944996356964,
+      "learning_rate": 8.867489933252521e-05,
+      "loss": 0.0106,
+      "step": 12880
+    },
+    {
+      "grad_norm": 0.2899917662143707,
+      "learning_rate": 8.865393153288504e-05,
+      "loss": 0.0126,
+      "step": 12890
+    },
+    {
+      "grad_norm": 0.23213069140911102,
+      "learning_rate": 8.8632946824702e-05,
+      "loss": 0.0107,
+      "step": 12900
+    },
+    {
+      "grad_norm": 0.2424149215221405,
+      "learning_rate": 8.86119452171555e-05,
+      "loss": 0.0098,
+      "step": 12910
+    },
+    {
+      "grad_norm": 0.2826816737651825,
+      "learning_rate": 8.859092671943234e-05,
+      "loss": 0.0126,
+      "step": 12920
+    },
+    {
+      "grad_norm": 0.2566341459751129,
+      "learning_rate": 8.856989134072676e-05,
+      "loss": 0.0138,
+      "step": 12930
+    },
+    {
+      "grad_norm": 0.2674964964389801,
+      "learning_rate": 8.85488390902403e-05,
+      "loss": 0.0118,
+      "step": 12940
+    },
+    {
+      "grad_norm": 0.27530139684677124,
+      "learning_rate": 8.852776997718199e-05,
+      "loss": 0.0131,
+      "step": 12950
+    },
+    {
+      "grad_norm": 0.25569185614585876,
+      "learning_rate": 8.850668401076812e-05,
+      "loss": 0.0112,
+      "step": 12960
+    },
+    {
+      "grad_norm": 0.24537570774555206,
+      "learning_rate": 8.848558120022246e-05,
+      "loss": 0.0109,
+      "step": 12970
+    },
+    {
+      "grad_norm": 0.3018043339252472,
+      "learning_rate": 8.846446155477603e-05,
+      "loss": 0.012,
+      "step": 12980
+    },
+    {
+      "grad_norm": 0.2121061384677887,
+      "learning_rate": 8.844332508366735e-05,
+      "loss": 0.0115,
+      "step": 12990
+    },
+    {
+      "grad_norm": 0.25486233830451965,
+      "learning_rate": 8.84221717961422e-05,
+      "loss": 0.013,
+      "step": 13000
+    },
+    {
+      "grad_norm": 0.20938284695148468,
+      "learning_rate": 8.840100170145374e-05,
+      "loss": 0.0109,
+      "step": 13010
+    },
+    {
+      "grad_norm": 0.20211170613765717,
+      "learning_rate": 8.837981480886249e-05,
+      "loss": 0.0122,
+      "step": 13020
+    },
+    {
+      "grad_norm": 0.26173096895217896,
+      "learning_rate": 8.835861112763633e-05,
+      "loss": 0.0119,
+      "step": 13030
+    },
+    {
+      "grad_norm": 0.1877232789993286,
+      "learning_rate": 8.833739066705044e-05,
+      "loss": 0.0105,
+      "step": 13040
+    },
+    {
+      "grad_norm": 0.23357494175434113,
+      "learning_rate": 8.831615343638742e-05,
+      "loss": 0.0126,
+      "step": 13050
+    },
+    {
+      "grad_norm": 0.23743999004364014,
+      "learning_rate": 8.829489944493711e-05,
+      "loss": 0.0116,
+      "step": 13060
+    },
+    {
+      "grad_norm": 0.2846521735191345,
+      "learning_rate": 8.827362870199675e-05,
+      "loss": 0.0098,
+      "step": 13070
+    },
+    {
+      "grad_norm": 0.20547032356262207,
+      "learning_rate": 8.825234121687089e-05,
+      "loss": 0.0131,
+      "step": 13080
+    },
+    {
+      "grad_norm": 0.23087230324745178,
+      "learning_rate": 8.823103699887139e-05,
+      "loss": 0.0128,
+      "step": 13090
+    },
+    {
+      "grad_norm": 0.22372552752494812,
+      "learning_rate": 8.820971605731745e-05,
+      "loss": 0.011,
+      "step": 13100
+    },
+    {
+      "grad_norm": 0.26860445737838745,
+      "learning_rate": 8.818837840153556e-05,
+      "loss": 0.0117,
+      "step": 13110
+    },
+    {
+      "grad_norm": 0.21150100231170654,
+      "learning_rate": 8.816702404085952e-05,
+      "loss": 0.0112,
+      "step": 13120
+    },
+    {
+      "grad_norm": 0.23763838410377502,
+      "learning_rate": 8.814565298463048e-05,
+      "loss": 0.0121,
+      "step": 13130
+    },
+    {
+      "grad_norm": 0.2159801721572876,
+      "learning_rate": 8.812426524219688e-05,
+      "loss": 0.01,
+      "step": 13140
+    },
+    {
+      "grad_norm": 0.17293375730514526,
+      "learning_rate": 8.81028608229144e-05,
+      "loss": 0.0112,
+      "step": 13150
+    },
+    {
+      "grad_norm": 0.2553454637527466,
+      "learning_rate": 8.808143973614611e-05,
+      "loss": 0.0103,
+      "step": 13160
+    },
+    {
+      "grad_norm": 0.2682110071182251,
+      "learning_rate": 8.806000199126228e-05,
+      "loss": 0.0117,
+      "step": 13170
+    },
+    {
+      "grad_norm": 0.2560890316963196,
+      "learning_rate": 8.803854759764052e-05,
+      "loss": 0.0114,
+      "step": 13180
+    },
+    {
+      "grad_norm": 0.20107702910900116,
+      "learning_rate": 8.801707656466572e-05,
+      "loss": 0.0134,
+      "step": 13190
+    },
+    {
+      "grad_norm": 0.2863423228263855,
+      "learning_rate": 8.799558890173003e-05,
+      "loss": 0.0099,
+      "step": 13200
+    },
+    {
+      "grad_norm": 0.2272268384695053,
+      "learning_rate": 8.79740846182329e-05,
+      "loss": 0.0113,
+      "step": 13210
+    },
+    {
+      "grad_norm": 0.20370031893253326,
+      "learning_rate": 8.7952563723581e-05,
+      "loss": 0.0113,
+      "step": 13220
+    },
+    {
+      "grad_norm": 0.2593918740749359,
+      "learning_rate": 8.793102622718834e-05,
+      "loss": 0.0129,
+      "step": 13230
+    },
+    {
+      "grad_norm": 0.17372731864452362,
+      "learning_rate": 8.790947213847613e-05,
+      "loss": 0.0095,
+      "step": 13240
+    },
+    {
+      "grad_norm": 0.20547451078891754,
+      "learning_rate": 8.788790146687286e-05,
+      "loss": 0.011,
+      "step": 13250
+    },
+    {
+      "grad_norm": 0.21150174736976624,
+      "learning_rate": 8.786631422181429e-05,
+      "loss": 0.0124,
+      "step": 13260
+    },
+    {
+      "grad_norm": 0.20662793517112732,
+      "learning_rate": 8.78447104127434e-05,
+      "loss": 0.0115,
+      "step": 13270
+    },
+    {
+      "grad_norm": 0.232677161693573,
+      "learning_rate": 8.782309004911042e-05,
+      "loss": 0.0116,
+      "step": 13280
+    },
+    {
+      "grad_norm": 0.22701185941696167,
+      "learning_rate": 8.780145314037286e-05,
+      "loss": 0.0128,
+      "step": 13290
+    },
+    {
+      "grad_norm": 0.23137614130973816,
+      "learning_rate": 8.777979969599542e-05,
+      "loss": 0.0126,
+      "step": 13300
+    },
+    {
+      "grad_norm": 0.20675139129161835,
+      "learning_rate": 8.775812972545006e-05,
+      "loss": 0.0113,
+      "step": 13310
+    },
+    {
+      "grad_norm": 0.25976312160491943,
+      "learning_rate": 8.773644323821596e-05,
+      "loss": 0.015,
+      "step": 13320
+    },
+    {
+      "grad_norm": 0.2075965255498886,
+      "learning_rate": 8.771474024377953e-05,
+      "loss": 0.0126,
+      "step": 13330
+    },
+    {
+      "grad_norm": 0.2551446855068207,
+      "learning_rate": 8.769302075163438e-05,
+      "loss": 0.0115,
+      "step": 13340
+    },
+    {
+      "grad_norm": 0.2741076648235321,
+      "learning_rate": 8.767128477128137e-05,
+      "loss": 0.0113,
+      "step": 13350
+    },
+    {
+      "grad_norm": 0.314983606338501,
+      "learning_rate": 8.764953231222854e-05,
+      "loss": 0.014,
+      "step": 13360
+    },
+    {
+      "grad_norm": 0.26172617077827454,
+      "learning_rate": 8.762776338399119e-05,
+      "loss": 0.0114,
+      "step": 13370
+    },
+    {
+      "grad_norm": 0.19639045000076294,
+      "learning_rate": 8.760597799609176e-05,
+      "loss": 0.0115,
+      "step": 13380
+    },
+    {
+      "grad_norm": 0.21348191797733307,
+      "learning_rate": 8.758417615805992e-05,
+      "loss": 0.0113,
+      "step": 13390
+    },
+    {
+      "grad_norm": 0.22860169410705566,
+      "learning_rate": 8.756235787943254e-05,
+      "loss": 0.0124,
+      "step": 13400
+    },
+    {
+      "grad_norm": 0.274227499961853,
+      "learning_rate": 8.754052316975367e-05,
+      "loss": 0.0124,
+      "step": 13410
+    },
+    {
+      "grad_norm": 0.21839605271816254,
+      "learning_rate": 8.751867203857455e-05,
+      "loss": 0.0133,
+      "step": 13420
+    },
+    {
+      "grad_norm": 0.24039305746555328,
+      "learning_rate": 8.749680449545363e-05,
+      "loss": 0.014,
+      "step": 13430
+    },
+    {
+      "grad_norm": 0.3030259311199188,
+      "learning_rate": 8.747492054995649e-05,
+      "loss": 0.0134,
+      "step": 13440
+    },
+    {
+      "grad_norm": 0.1759081482887268,
+      "learning_rate": 8.745302021165595e-05,
+      "loss": 0.0112,
+      "step": 13450
+    },
+    {
+      "grad_norm": 0.1870216727256775,
+      "learning_rate": 8.743110349013192e-05,
+      "loss": 0.0114,
+      "step": 13460
+    },
+    {
+      "grad_norm": 0.2526007890701294,
+      "learning_rate": 8.740917039497153e-05,
+      "loss": 0.0138,
+      "step": 13470
+    },
+    {
+      "grad_norm": 0.19725747406482697,
+      "learning_rate": 8.738722093576906e-05,
+      "loss": 0.0126,
+      "step": 13480
+    },
+    {
+      "grad_norm": 0.21281969547271729,
+      "learning_rate": 8.736525512212597e-05,
+      "loss": 0.0114,
+      "step": 13490
+    },
+    {
+      "grad_norm": 0.2567230761051178,
+      "learning_rate": 8.734327296365084e-05,
+      "loss": 0.013,
+      "step": 13500
+    },
+    {
+      "grad_norm": 0.2794579267501831,
+      "learning_rate": 8.732127446995939e-05,
+      "loss": 0.0131,
+      "step": 13510
+    },
+    {
+      "grad_norm": 0.28424158692359924,
+      "learning_rate": 8.729925965067454e-05,
+      "loss": 0.0115,
+      "step": 13520
+    },
+    {
+      "grad_norm": 0.1328752636909485,
+      "learning_rate": 8.72772285154263e-05,
+      "loss": 0.0114,
+      "step": 13530
+    },
+    {
+      "grad_norm": 0.26082611083984375,
+      "learning_rate": 8.725518107385187e-05,
+      "loss": 0.0121,
+      "step": 13540
+    },
+    {
+      "grad_norm": 0.13311143219470978,
+      "learning_rate": 8.72331173355955e-05,
+      "loss": 0.0146,
+      "step": 13550
+    },
+    {
+      "grad_norm": 0.25326257944107056,
+      "learning_rate": 8.721103731030867e-05,
+      "loss": 0.0118,
+      "step": 13560
+    },
+    {
+      "grad_norm": 0.2298624962568283,
+      "learning_rate": 8.718894100764989e-05,
+      "loss": 0.0126,
+      "step": 13570
+    },
+    {
+      "grad_norm": 0.27702900767326355,
+      "learning_rate": 8.716682843728485e-05,
+      "loss": 0.0124,
+      "step": 13580
+    },
+    {
+      "grad_norm": 0.2645373046398163,
+      "learning_rate": 8.714469960888634e-05,
+      "loss": 0.0145,
+      "step": 13590
+    },
+    {
+      "grad_norm": 0.22337238490581512,
+      "learning_rate": 8.712255453213427e-05,
+      "loss": 0.0121,
+      "step": 13600
+    },
+    {
+      "grad_norm": 0.2726132869720459,
+      "learning_rate": 8.710039321671563e-05,
+      "loss": 0.0106,
+      "step": 13610
+    },
+    {
+      "grad_norm": 0.2808859646320343,
+      "learning_rate": 8.707821567232456e-05,
+      "loss": 0.0111,
+      "step": 13620
+    },
+    {
+      "grad_norm": 0.26834359765052795,
+      "learning_rate": 8.705602190866225e-05,
+      "loss": 0.0123,
+      "step": 13630
+    },
+    {
+      "grad_norm": 0.19009670615196228,
+      "learning_rate": 8.703381193543701e-05,
+      "loss": 0.0118,
+      "step": 13640
+    },
+    {
+      "grad_norm": 0.21243135631084442,
+      "learning_rate": 8.701158576236423e-05,
+      "loss": 0.0134,
+      "step": 13650
+    },
+    {
+      "grad_norm": 0.23044738173484802,
+      "learning_rate": 8.69893433991664e-05,
+      "loss": 0.0116,
+      "step": 13660
+    },
+    {
+      "grad_norm": 0.22317096590995789,
+      "learning_rate": 8.69670848555731e-05,
+      "loss": 0.0117,
+      "step": 13670
+    },
+    {
+      "grad_norm": 0.29898056387901306,
+      "learning_rate": 8.694481014132096e-05,
+      "loss": 0.0141,
+      "step": 13680
+    },
+    {
+      "grad_norm": 0.24750326573848724,
+      "learning_rate": 8.69225192661537e-05,
+      "loss": 0.0122,
+      "step": 13690
+    },
+    {
+      "grad_norm": 0.2141142338514328,
+      "learning_rate": 8.690021223982208e-05,
+      "loss": 0.0136,
+      "step": 13700
+    },
+    {
+      "grad_norm": 0.30433347821235657,
+      "learning_rate": 8.687788907208398e-05,
+      "loss": 0.0124,
+      "step": 13710
+    },
+    {
+      "grad_norm": 0.23960022628307343,
+      "learning_rate": 8.685554977270431e-05,
+      "loss": 0.0105,
+      "step": 13720
+    },
+    {
+      "grad_norm": 0.23475642502307892,
+      "learning_rate": 8.683319435145503e-05,
+      "loss": 0.0147,
+      "step": 13730
+    },
+    {
+      "grad_norm": 0.2551279366016388,
+      "learning_rate": 8.681082281811517e-05,
+      "loss": 0.0109,
+      "step": 13740
+    },
+    {
+      "grad_norm": 0.30843374133110046,
+      "learning_rate": 8.67884351824708e-05,
+      "loss": 0.014,
+      "step": 13750
+    },
+    {
+      "grad_norm": 0.32532355189323425,
+      "learning_rate": 8.676603145431501e-05,
+      "loss": 0.0122,
+      "step": 13760
+    },
+    {
+      "grad_norm": 0.23043671250343323,
+      "learning_rate": 8.674361164344799e-05,
+      "loss": 0.0106,
+      "step": 13770
+    },
+    {
+      "grad_norm": 0.2534899413585663,
+      "learning_rate": 8.672117575967688e-05,
+      "loss": 0.0102,
+      "step": 13780
+    },
+    {
+      "grad_norm": 0.2037641704082489,
+      "learning_rate": 8.669872381281595e-05,
+      "loss": 0.0109,
+      "step": 13790
+    },
+    {
+      "grad_norm": 0.24862174689769745,
+      "learning_rate": 8.667625581268639e-05,
+      "loss": 0.0104,
+      "step": 13800
+    },
+    {
+      "grad_norm": 0.19678638875484467,
+      "learning_rate": 8.665377176911651e-05,
+      "loss": 0.0103,
+      "step": 13810
+    },
+    {
+      "grad_norm": 0.19367928802967072,
+      "learning_rate": 8.663127169194159e-05,
+      "loss": 0.0113,
+      "step": 13820
+    },
+    {
+      "grad_norm": 0.222662553191185,
+      "learning_rate": 8.660875559100389e-05,
+      "loss": 0.0126,
+      "step": 13830
+    },
+    {
+      "grad_norm": 0.22500859200954437,
+      "learning_rate": 8.658622347615274e-05,
+      "loss": 0.0118,
+      "step": 13840
+    },
+    {
+      "grad_norm": 0.19104662537574768,
+      "learning_rate": 8.656367535724448e-05,
+      "loss": 0.0099,
+      "step": 13850
+    },
+    {
+      "grad_norm": 0.1921711415052414,
+      "learning_rate": 8.65411112441424e-05,
+      "loss": 0.0091,
+      "step": 13860
+    },
+    {
+      "grad_norm": 0.22772419452667236,
+      "learning_rate": 8.651853114671679e-05,
+      "loss": 0.0129,
+      "step": 13870
+    },
+    {
+      "grad_norm": 0.2777252793312073,
+      "learning_rate": 8.649593507484499e-05,
+      "loss": 0.0113,
+      "step": 13880
+    },
+    {
+      "grad_norm": 0.281438946723938,
+      "learning_rate": 8.647332303841126e-05,
+      "loss": 0.0112,
+      "step": 13890
+    },
+    {
+      "grad_norm": 0.2142777293920517,
+      "learning_rate": 8.645069504730689e-05,
+      "loss": 0.0103,
+      "step": 13900
+    },
+    {
+      "grad_norm": 0.16825096309185028,
+      "learning_rate": 8.64280511114301e-05,
+      "loss": 0.0132,
+      "step": 13910
+    },
+    {
+      "grad_norm": 0.2333957999944687,
+      "learning_rate": 8.640539124068617e-05,
+      "loss": 0.0111,
+      "step": 13920
+    },
+    {
+      "grad_norm": 0.17758134007453918,
+      "learning_rate": 8.638271544498727e-05,
+      "loss": 0.0096,
+      "step": 13930
+    },
+    {
+      "grad_norm": 0.17744183540344238,
+      "learning_rate": 8.636002373425257e-05,
+      "loss": 0.0102,
+      "step": 13940
+    },
+    {
+      "grad_norm": 0.26167476177215576,
+      "learning_rate": 8.633731611840817e-05,
+      "loss": 0.0105,
+      "step": 13950
+    },
+    {
+      "grad_norm": 0.2435753345489502,
+      "learning_rate": 8.631459260738717e-05,
+      "loss": 0.0116,
+      "step": 13960
+    },
+    {
+      "grad_norm": 0.2256001979112625,
+      "learning_rate": 8.62918532111296e-05,
+      "loss": 0.013,
+      "step": 13970
+    },
+    {
+      "grad_norm": 0.22290624678134918,
+      "learning_rate": 8.626909793958248e-05,
+      "loss": 0.0112,
+      "step": 13980
+    },
+    {
+      "grad_norm": 0.16998325288295746,
+      "learning_rate": 8.624632680269969e-05,
+      "loss": 0.011,
+      "step": 13990
+    },
+    {
+      "grad_norm": 0.2434377372264862,
+      "learning_rate": 8.622353981044212e-05,
+      "loss": 0.0093,
+      "step": 14000
+    },
+    {
+      "grad_norm": 0.2171565741300583,
+      "learning_rate": 8.620073697277757e-05,
+      "loss": 0.0121,
+      "step": 14010
+    },
+    {
+      "grad_norm": 0.23160746693611145,
+      "learning_rate": 8.617791829968079e-05,
+      "loss": 0.0119,
+      "step": 14020
+    },
+    {
+      "grad_norm": 0.19179892539978027,
+      "learning_rate": 8.615508380113344e-05,
+      "loss": 0.0102,
+      "step": 14030
+    },
+    {
+      "grad_norm": 0.26850423216819763,
+      "learning_rate": 8.613223348712408e-05,
+      "loss": 0.0105,
+      "step": 14040
+    },
+    {
+      "grad_norm": 0.20151951909065247,
+      "learning_rate": 8.610936736764824e-05,
+      "loss": 0.0132,
+      "step": 14050
+    },
+    {
+      "grad_norm": 0.2043834626674652,
+      "learning_rate": 8.608648545270833e-05,
+      "loss": 0.011,
+      "step": 14060
+    },
+    {
+      "grad_norm": 0.2035376876592636,
+      "learning_rate": 8.606358775231366e-05,
+      "loss": 0.0094,
+      "step": 14070
+    },
+    {
+      "grad_norm": 0.12958520650863647,
+      "learning_rate": 8.60406742764805e-05,
+      "loss": 0.0113,
+      "step": 14080
+    },
+    {
+      "grad_norm": 0.2147422432899475,
+      "learning_rate": 8.601774503523195e-05,
+      "loss": 0.011,
+      "step": 14090
+    },
+    {
+      "grad_norm": 0.27389365434646606,
+      "learning_rate": 8.599480003859805e-05,
+      "loss": 0.0114,
+      "step": 14100
+    },
+    {
+      "grad_norm": 0.24447356164455414,
+      "learning_rate": 8.597183929661573e-05,
+      "loss": 0.0116,
+      "step": 14110
+    },
+    {
+      "grad_norm": 0.17310236394405365,
+      "learning_rate": 8.594886281932879e-05,
+      "loss": 0.0123,
+      "step": 14120
+    },
+    {
+      "grad_norm": 0.24971838295459747,
+      "learning_rate": 8.59258706167879e-05,
+      "loss": 0.0116,
+      "step": 14130
+    },
+    {
+      "grad_norm": 0.22574636340141296,
+      "learning_rate": 8.590286269905068e-05,
+      "loss": 0.0111,
+      "step": 14140
+    },
+    {
+      "grad_norm": 0.2921409606933594,
+      "learning_rate": 8.587983907618154e-05,
+      "loss": 0.0097,
+      "step": 14150
+    },
+    {
+      "grad_norm": 0.215134859085083,
+      "learning_rate": 8.585679975825178e-05,
+      "loss": 0.0116,
+      "step": 14160
+    },
+    {
+      "grad_norm": 0.2180074006319046,
+      "learning_rate": 8.583374475533962e-05,
+      "loss": 0.0122,
+      "step": 14170
+    },
+    {
+      "grad_norm": 0.2697533071041107,
+      "learning_rate": 8.581067407753009e-05,
+      "loss": 0.0114,
+      "step": 14180
+    },
+    {
+      "grad_norm": 0.2665674388408661,
+      "learning_rate": 8.578758773491507e-05,
+      "loss": 0.0126,
+      "step": 14190
+    },
+    {
+      "grad_norm": 0.19359563291072845,
+      "learning_rate": 8.576448573759332e-05,
+      "loss": 0.0113,
+      "step": 14200
+    },
+    {
+      "grad_norm": 0.2548215985298157,
+      "learning_rate": 8.574136809567044e-05,
+      "loss": 0.0123,
+      "step": 14210
+    },
+    {
+      "grad_norm": 0.16669930517673492,
+      "learning_rate": 8.57182348192589e-05,
+      "loss": 0.0118,
+      "step": 14220
+    },
+    {
+      "grad_norm": 0.2175881266593933,
+      "learning_rate": 8.569508591847792e-05,
+      "loss": 0.0117,
+      "step": 14230
+    },
+    {
+      "grad_norm": 0.3053397834300995,
+      "learning_rate": 8.567192140345367e-05,
+      "loss": 0.0112,
+      "step": 14240
+    },
+    {
+      "grad_norm": 0.27136504650115967,
+      "learning_rate": 8.564874128431906e-05,
+      "loss": 0.0125,
+      "step": 14250
+    },
+    {
+      "grad_norm": 0.1882592886686325,
+      "learning_rate": 8.562554557121389e-05,
+      "loss": 0.0116,
+      "step": 14260
+    },
+    {
+      "grad_norm": 0.25154924392700195,
+      "learning_rate": 8.560233427428475e-05,
+      "loss": 0.0121,
+      "step": 14270
+    },
+    {
+      "grad_norm": 0.21361149847507477,
+      "learning_rate": 8.557910740368503e-05,
+      "loss": 0.0099,
+      "step": 14280
+    },
+    {
+      "grad_norm": 0.17490749061107635,
+      "learning_rate": 8.555586496957495e-05,
+      "loss": 0.0099,
+      "step": 14290
+    },
+    {
+      "grad_norm": 0.23400843143463135,
+      "learning_rate": 8.553260698212155e-05,
+      "loss": 0.011,
+      "step": 14300
+    },
+    {
+      "grad_norm": 0.23231184482574463,
+      "learning_rate": 8.550933345149868e-05,
+      "loss": 0.0143,
+      "step": 14310
+    },
+    {
+      "grad_norm": 0.18553456664085388,
+      "learning_rate": 8.548604438788696e-05,
+      "loss": 0.0136,
+      "step": 14320
+    },
+    {
+      "grad_norm": 0.1935059279203415,
+      "learning_rate": 8.546273980147383e-05,
+      "loss": 0.0102,
+      "step": 14330
+    },
+    {
+      "grad_norm": 0.21481703221797943,
+      "learning_rate": 8.543941970245348e-05,
+      "loss": 0.0125,
+      "step": 14340
+    },
+    {
+      "grad_norm": 0.2463425248861313,
+      "learning_rate": 8.541608410102693e-05,
+      "loss": 0.0101,
+      "step": 14350
+    },
+    {
+      "grad_norm": 0.2453492432832718,
+      "learning_rate": 8.539273300740195e-05,
+      "loss": 0.0111,
+      "step": 14360
+    },
+    {
+      "grad_norm": 0.1969238668680191,
+      "learning_rate": 8.536936643179313e-05,
+      "loss": 0.0098,
+      "step": 14370
+    },
+    {
+      "grad_norm": 0.24944691359996796,
+      "learning_rate": 8.534598438442179e-05,
+      "loss": 0.0116,
+      "step": 14380
+    },
+    {
+      "grad_norm": 0.24247871339321136,
+      "learning_rate": 8.532258687551603e-05,
+      "loss": 0.0123,
+      "step": 14390
+    },
+    {
+      "grad_norm": 0.21719405055046082,
+      "learning_rate": 8.529917391531071e-05,
+      "loss": 0.0104,
+      "step": 14400
+    },
+    {
+      "grad_norm": 0.24251297116279602,
+      "learning_rate": 8.527574551404747e-05,
+      "loss": 0.0097,
+      "step": 14410
+    },
+    {
+      "grad_norm": 0.15803556144237518,
+      "learning_rate": 8.525230168197468e-05,
+      "loss": 0.0099,
+      "step": 14420
+    },
+    {
+      "grad_norm": 0.20316939055919647,
+      "learning_rate": 8.522884242934745e-05,
+      "loss": 0.0114,
+      "step": 14430
+    },
+    {
+      "grad_norm": 0.20054218173027039,
+      "learning_rate": 8.520536776642768e-05,
+      "loss": 0.0123,
+      "step": 14440
+    },
+    {
+      "grad_norm": 0.193473681807518,
+      "learning_rate": 8.5181877703484e-05,
+      "loss": 0.0107,
+      "step": 14450
+    },
+    {
+      "grad_norm": 0.25915220379829407,
+      "learning_rate": 8.51583722507917e-05,
+      "loss": 0.0117,
+      "step": 14460
+    },
+    {
+      "grad_norm": 0.24090810120105743,
+      "learning_rate": 8.513485141863293e-05,
+      "loss": 0.0107,
+      "step": 14470
+    },
+    {
+      "grad_norm": 0.1982126235961914,
+      "learning_rate": 8.511131521729647e-05,
+      "loss": 0.0093,
+      "step": 14480
+    },
+    {
+      "grad_norm": 0.20103752613067627,
+      "learning_rate": 8.508776365707787e-05,
+      "loss": 0.0104,
+      "step": 14490
+    },
+    {
+      "grad_norm": 0.19555769860744476,
+      "learning_rate": 8.506419674827934e-05,
+      "loss": 0.0126,
+      "step": 14500
+    },
+    {
+      "grad_norm": 0.19296760857105255,
+      "learning_rate": 8.50406145012099e-05,
+      "loss": 0.012,
+      "step": 14510
+    },
+    {
+      "grad_norm": 0.18178395926952362,
+      "learning_rate": 8.501701692618519e-05,
+      "loss": 0.0104,
+      "step": 14520
+    },
+    {
+      "grad_norm": 0.20316867530345917,
+      "learning_rate": 8.499340403352761e-05,
+      "loss": 0.0107,
+      "step": 14530
+    },
+    {
+      "grad_norm": 0.17796030640602112,
+      "learning_rate": 8.496977583356623e-05,
+      "loss": 0.0108,
+      "step": 14540
+    },
+    {
+      "grad_norm": 0.31354615092277527,
+      "learning_rate": 8.494613233663684e-05,
+      "loss": 0.0139,
+      "step": 14550
+    },
+    {
+      "grad_norm": 0.2433123141527176,
+      "learning_rate": 8.492247355308189e-05,
+      "loss": 0.0139,
+      "step": 14560
+    },
+    {
+      "grad_norm": 0.2009527087211609,
+      "learning_rate": 8.489879949325056e-05,
+      "loss": 0.0105,
+      "step": 14570
+    },
+    {
+      "grad_norm": 0.2242869734764099,
+      "learning_rate": 8.487511016749868e-05,
+      "loss": 0.0105,
+      "step": 14580
+    },
+    {
+      "grad_norm": 0.2081242799758911,
+      "learning_rate": 8.485140558618874e-05,
+      "loss": 0.0113,
+      "step": 14590
+    },
+    {
+      "grad_norm": 0.19855578243732452,
+      "learning_rate": 8.482768575968995e-05,
+      "loss": 0.0111,
+      "step": 14600
+    },
+    {
+      "grad_norm": 0.18686425685882568,
+      "learning_rate": 8.480395069837818e-05,
+      "loss": 0.0122,
+      "step": 14610
+    },
+    {
+      "grad_norm": 0.22768664360046387,
+      "learning_rate": 8.478020041263595e-05,
+      "loss": 0.0099,
+      "step": 14620
+    },
+    {
+      "grad_norm": 0.1900821030139923,
+      "learning_rate": 8.475643491285242e-05,
+      "loss": 0.0105,
+      "step": 14630
+    },
+    {
+      "grad_norm": 0.21737751364707947,
+      "learning_rate": 8.473265420942345e-05,
+      "loss": 0.0107,
+      "step": 14640
+    },
+    {
+      "grad_norm": 0.1899714171886444,
+      "learning_rate": 8.470885831275151e-05,
+      "loss": 0.0111,
+      "step": 14650
+    },
+    {
+      "grad_norm": 0.2258838266134262,
+      "learning_rate": 8.468504723324574e-05,
+      "loss": 0.0111,
+      "step": 14660
+    },
+    {
+      "grad_norm": 0.21849079430103302,
+      "learning_rate": 8.466122098132193e-05,
+      "loss": 0.012,
+      "step": 14670
+    },
+    {
+      "grad_norm": 0.15592177212238312,
+      "learning_rate": 8.463737956740245e-05,
+      "loss": 0.0114,
+      "step": 14680
+    },
+    {
+      "grad_norm": 0.17583948373794556,
+      "learning_rate": 8.461352300191639e-05,
+      "loss": 0.0101,
+      "step": 14690
+    },
+    {
+      "grad_norm": 0.2512693703174591,
+      "learning_rate": 8.45896512952994e-05,
+      "loss": 0.012,
+      "step": 14700
+    },
+    {
+      "grad_norm": 0.2562675476074219,
+      "learning_rate": 8.456576445799377e-05,
+      "loss": 0.0102,
+      "step": 14710
+    },
+    {
+      "grad_norm": 0.27093446254730225,
+      "learning_rate": 8.454186250044844e-05,
+      "loss": 0.0109,
+      "step": 14720
+    },
+    {
+      "grad_norm": 0.3213208019733429,
+      "learning_rate": 8.451794543311892e-05,
+      "loss": 0.0122,
+      "step": 14730
+    },
+    {
+      "grad_norm": 0.2291439175605774,
+      "learning_rate": 8.449401326646736e-05,
+      "loss": 0.0111,
+      "step": 14740
+    },
+    {
+      "grad_norm": 0.196000874042511,
+      "learning_rate": 8.447006601096248e-05,
+      "loss": 0.0098,
+      "step": 14750
+    },
+    {
+      "grad_norm": 0.24527637660503387,
+      "learning_rate": 8.444610367707964e-05,
+      "loss": 0.0132,
+      "step": 14760
+    },
+    {
+      "grad_norm": 0.1923275738954544,
+      "learning_rate": 8.442212627530078e-05,
+      "loss": 0.0098,
+      "step": 14770
+    },
+    {
+      "grad_norm": 0.17491698265075684,
+      "learning_rate": 8.439813381611441e-05,
+      "loss": 0.0109,
+      "step": 14780
+    },
+    {
+      "grad_norm": 0.27568820118904114,
+      "learning_rate": 8.437412631001567e-05,
+      "loss": 0.0102,
+      "step": 14790
+    },
+    {
+      "grad_norm": 0.23504909873008728,
+      "learning_rate": 8.435010376750626e-05,
+      "loss": 0.0144,
+      "step": 14800
+    },
+    {
+      "grad_norm": 0.20148515701293945,
+      "learning_rate": 8.432606619909442e-05,
+      "loss": 0.0133,
+      "step": 14810
+    },
+    {
+      "grad_norm": 0.2624683082103729,
+      "learning_rate": 8.430201361529506e-05,
+      "loss": 0.0128,
+      "step": 14820
+    },
+    {
+      "grad_norm": 0.2520950734615326,
+      "learning_rate": 8.427794602662954e-05,
+      "loss": 0.0135,
+      "step": 14830
+    },
+    {
+      "grad_norm": 0.2051675021648407,
+      "learning_rate": 8.425386344362586e-05,
+      "loss": 0.0133,
+      "step": 14840
+    },
+    {
+      "grad_norm": 0.19407731294631958,
+      "learning_rate": 8.422976587681859e-05,
+      "loss": 0.0123,
+      "step": 14850
+    },
+    {
+      "grad_norm": 0.24473409354686737,
+      "learning_rate": 8.42056533367488e-05,
+      "loss": 0.0144,
+      "step": 14860
+    },
+    {
+      "grad_norm": 0.21676215529441833,
+      "learning_rate": 8.41815258339641e-05,
+      "loss": 0.0145,
+      "step": 14870
+    },
+    {
+      "grad_norm": 0.27058276534080505,
+      "learning_rate": 8.415738337901874e-05,
+      "loss": 0.0153,
+      "step": 14880
+    },
+    {
+      "grad_norm": 0.2530534863471985,
+      "learning_rate": 8.413322598247342e-05,
+      "loss": 0.0147,
+      "step": 14890
+    },
+    {
+      "grad_norm": 0.25308144092559814,
+      "learning_rate": 8.41090536548954e-05,
+      "loss": 0.0152,
+      "step": 14900
+    },
+    {
+      "grad_norm": 0.13350895047187805,
+      "learning_rate": 8.408486640685849e-05,
+      "loss": 0.0148,
+      "step": 14910
+    },
+    {
+      "grad_norm": 0.22288718819618225,
+      "learning_rate": 8.4060664248943e-05,
+      "loss": 0.0135,
+      "step": 14920
+    },
+    {
+      "grad_norm": 0.2785625159740448,
+      "learning_rate": 8.40364471917358e-05,
+      "loss": 0.0146,
+      "step": 14930
+    },
+    {
+      "grad_norm": 0.26409482955932617,
+      "learning_rate": 8.401221524583024e-05,
+      "loss": 0.0133,
+      "step": 14940
+    },
+    {
+      "grad_norm": 0.25046905875205994,
+      "learning_rate": 8.398796842182619e-05,
+      "loss": 0.0124,
+      "step": 14950
+    },
+    {
+      "grad_norm": 0.2723139524459839,
+      "learning_rate": 8.396370673033006e-05,
+      "loss": 0.0108,
+      "step": 14960
+    },
+    {
+      "grad_norm": 0.22922638058662415,
+      "learning_rate": 8.39394301819547e-05,
+      "loss": 0.0124,
+      "step": 14970
+    },
+    {
+      "grad_norm": 0.23909199237823486,
+      "learning_rate": 8.391513878731949e-05,
+      "loss": 0.0128,
+      "step": 14980
+    },
+    {
+      "grad_norm": 0.24711565673351288,
+      "learning_rate": 8.389083255705037e-05,
+      "loss": 0.0122,
+      "step": 14990
+    },
+    {
+      "grad_norm": 0.29241040349006653,
+      "learning_rate": 8.386651150177968e-05,
+      "loss": 0.0123,
+      "step": 15000
+    },
+    {
+      "grad_norm": 0.3048204183578491,
+      "learning_rate": 8.384217563214627e-05,
+      "loss": 0.0158,
+      "step": 15010
+    },
+    {
+      "grad_norm": 0.18890880048274994,
+      "learning_rate": 8.381782495879546e-05,
+      "loss": 0.0149,
+      "step": 15020
+    },
+    {
+      "grad_norm": 0.274234414100647,
+      "learning_rate": 8.37934594923791e-05,
+      "loss": 0.0125,
+      "step": 15030
+    },
+    {
+      "grad_norm": 0.22186169028282166,
+      "learning_rate": 8.376907924355546e-05,
+      "loss": 0.0114,
+      "step": 15040
+    },
+    {
+      "grad_norm": 0.21267488598823547,
+      "learning_rate": 8.374468422298926e-05,
+      "loss": 0.0133,
+      "step": 15050
+    },
+    {
+      "grad_norm": 0.21168114244937897,
+      "learning_rate": 8.372027444135176e-05,
+      "loss": 0.0122,
+      "step": 15060
+    },
+    {
+      "grad_norm": 0.19336692988872528,
+      "learning_rate": 8.36958499093206e-05,
+      "loss": 0.0142,
+      "step": 15070
+    },
+    {
+      "grad_norm": 0.33816802501678467,
+      "learning_rate": 8.367141063757988e-05,
+      "loss": 0.0147,
+      "step": 15080
+    },
+    {
+      "grad_norm": 0.2203822284936905,
+      "learning_rate": 8.364695663682022e-05,
+      "loss": 0.015,
+      "step": 15090
+    },
+    {
+      "grad_norm": 0.19696050882339478,
+      "learning_rate": 8.362248791773858e-05,
+      "loss": 0.013,
+      "step": 15100
+    },
+    {
+      "grad_norm": 0.20641343295574188,
+      "learning_rate": 8.359800449103842e-05,
+      "loss": 0.0127,
+      "step": 15110
+    },
+    {
+      "grad_norm": 0.1746257245540619,
+      "learning_rate": 8.357350636742967e-05,
+      "loss": 0.0119,
+      "step": 15120
+    },
+    {
+      "grad_norm": 0.21023130416870117,
+      "learning_rate": 8.354899355762858e-05,
+      "loss": 0.0101,
+      "step": 15130
+    },
+    {
+      "grad_norm": 0.28233930468559265,
+      "learning_rate": 8.352446607235791e-05,
+      "loss": 0.0121,
+      "step": 15140
+    },
+    {
+      "grad_norm": 0.21881423890590668,
+      "learning_rate": 8.349992392234683e-05,
+      "loss": 0.0108,
+      "step": 15150
+    },
+    {
+      "grad_norm": 0.24860574305057526,
+      "learning_rate": 8.347536711833088e-05,
+      "loss": 0.0114,
+      "step": 15160
+    },
+    {
+      "grad_norm": 0.15644940733909607,
+      "learning_rate": 8.345079567105206e-05,
+      "loss": 0.0121,
+      "step": 15170
+    },
+    {
+      "grad_norm": 0.21559150516986847,
+      "learning_rate": 8.342620959125875e-05,
+      "loss": 0.0101,
+      "step": 15180
+    },
+    {
+      "grad_norm": 0.19659461081027985,
+      "learning_rate": 8.340160888970574e-05,
+      "loss": 0.0126,
+      "step": 15190
+    },
+    {
+      "grad_norm": 0.1744043231010437,
+      "learning_rate": 8.337699357715422e-05,
+      "loss": 0.0111,
+      "step": 15200
+    },
+    {
+      "grad_norm": 0.22091403603553772,
+      "learning_rate": 8.335236366437174e-05,
+      "loss": 0.0102,
+      "step": 15210
+    },
+    {
+      "grad_norm": 0.1661144495010376,
+      "learning_rate": 8.332771916213227e-05,
+      "loss": 0.0103,
+      "step": 15220
+    },
+    {
+      "grad_norm": 0.23233731091022491,
+      "learning_rate": 8.330306008121618e-05,
+      "loss": 0.0116,
+      "step": 15230
+    },
+    {
+      "grad_norm": 0.25890713930130005,
+      "learning_rate": 8.327838643241014e-05,
+      "loss": 0.0111,
+      "step": 15240
+    },
+    {
+      "grad_norm": 0.21791304647922516,
+      "learning_rate": 8.32536982265073e-05,
+      "loss": 0.0106,
+      "step": 15250
+    },
+    {
+      "grad_norm": 0.20258378982543945,
+      "learning_rate": 8.32289954743071e-05,
+      "loss": 0.0105,
+      "step": 15260
+    },
+    {
+      "grad_norm": 0.2594950795173645,
+      "learning_rate": 8.320427818661532e-05,
+      "loss": 0.0124,
+      "step": 15270
+    },
+    {
+      "grad_norm": 0.2818995714187622,
+      "learning_rate": 8.317954637424419e-05,
+      "loss": 0.0117,
+      "step": 15280
+    },
+    {
+      "grad_norm": 0.24744687974452972,
+      "learning_rate": 8.315480004801221e-05,
+      "loss": 0.0122,
+      "step": 15290
+    },
+    {
+      "grad_norm": 0.19593751430511475,
+      "learning_rate": 8.31300392187443e-05,
+      "loss": 0.0126,
+      "step": 15300
+    },
+    {
+      "grad_norm": 0.17994940280914307,
+      "learning_rate": 8.31052638972717e-05,
+      "loss": 0.0102,
+      "step": 15310
+    },
+    {
+      "grad_norm": 0.1972566694021225,
+      "learning_rate": 8.30804740944319e-05,
+      "loss": 0.0076,
+      "step": 15320
+    },
+    {
+      "grad_norm": 0.18683426082134247,
+      "learning_rate": 8.305566982106887e-05,
+      "loss": 0.0102,
+      "step": 15330
+    },
+    {
+      "grad_norm": 0.23529189825057983,
+      "learning_rate": 8.303085108803282e-05,
+      "loss": 0.0097,
+      "step": 15340
+    },
+    {
+      "grad_norm": 0.24034328758716583,
+      "learning_rate": 8.300601790618032e-05,
+      "loss": 0.0099,
+      "step": 15350
+    },
+    {
+      "grad_norm": 0.2309717983007431,
+      "learning_rate": 8.298117028637423e-05,
+      "loss": 0.0114,
+      "step": 15360
+    },
+    {
+      "grad_norm": 0.17429903149604797,
+      "learning_rate": 8.295630823948375e-05,
+      "loss": 0.0121,
+      "step": 15370
+    },
+    {
+      "grad_norm": 0.19188131392002106,
+      "learning_rate": 8.293143177638438e-05,
+      "loss": 0.0098,
+      "step": 15380
+    },
+    {
+      "grad_norm": 0.25611576437950134,
+      "learning_rate": 8.290654090795795e-05,
+      "loss": 0.0127,
+      "step": 15390
+    },
+    {
+      "grad_norm": 0.24860849976539612,
+      "learning_rate": 8.288163564509253e-05,
+      "loss": 0.0119,
+      "step": 15400
+    },
+    {
+      "grad_norm": 0.17722712457180023,
+      "learning_rate": 8.285671599868257e-05,
+      "loss": 0.0087,
+      "step": 15410
+    },
+    {
+      "grad_norm": 0.21528954803943634,
+      "learning_rate": 8.283178197962875e-05,
+      "loss": 0.0134,
+      "step": 15420
+    },
+    {
+      "grad_norm": 0.14519406855106354,
+      "learning_rate": 8.280683359883808e-05,
+      "loss": 0.0102,
+      "step": 15430
+    },
+    {
+      "grad_norm": 0.1940581500530243,
+      "learning_rate": 8.278187086722377e-05,
+      "loss": 0.0125,
+      "step": 15440
+    },
+    {
+      "grad_norm": 0.27391213178634644,
+      "learning_rate": 8.275689379570544e-05,
+      "loss": 0.0092,
+      "step": 15450
+    },
+    {
+      "grad_norm": 0.20294760167598724,
+      "learning_rate": 8.273190239520887e-05,
+      "loss": 0.0094,
+      "step": 15460
+    },
+    {
+      "grad_norm": 0.19369636476039886,
+      "learning_rate": 8.270689667666616e-05,
+      "loss": 0.0087,
+      "step": 15470
+    },
+    {
+      "grad_norm": 0.228814497590065,
+      "learning_rate": 8.268187665101563e-05,
+      "loss": 0.0127,
+      "step": 15480
+    },
+    {
+      "grad_norm": 0.2010326236486435,
+      "learning_rate": 8.265684232920194e-05,
+      "loss": 0.0112,
+      "step": 15490
+    },
+    {
+      "grad_norm": 0.2771534323692322,
+      "learning_rate": 8.263179372217592e-05,
+      "loss": 0.0131,
+      "step": 15500
+    },
+    {
+      "grad_norm": 0.17047835886478424,
+      "learning_rate": 8.260673084089468e-05,
+      "loss": 0.0101,
+      "step": 15510
+    },
+    {
+      "grad_norm": 0.24358083307743073,
+      "learning_rate": 8.25816536963216e-05,
+      "loss": 0.012,
+      "step": 15520
+    },
+    {
+      "grad_norm": 0.17219646275043488,
+      "learning_rate": 8.255656229942626e-05,
+      "loss": 0.0096,
+      "step": 15530
+    },
+    {
+      "grad_norm": 0.16456501185894012,
+      "learning_rate": 8.253145666118447e-05,
+      "loss": 0.0106,
+      "step": 15540
+    },
+    {
+      "grad_norm": 0.2208227515220642,
+      "learning_rate": 8.250633679257834e-05,
+      "loss": 0.0106,
+      "step": 15550
+    },
+    {
+      "grad_norm": 0.24964696168899536,
+      "learning_rate": 8.24812027045961e-05,
+      "loss": 0.0119,
+      "step": 15560
+    },
+    {
+      "grad_norm": 0.21451212465763092,
+      "learning_rate": 8.245605440823226e-05,
+      "loss": 0.011,
+      "step": 15570
+    },
+    {
+      "grad_norm": 0.21919575333595276,
+      "learning_rate": 8.243089191448759e-05,
+      "loss": 0.0117,
+      "step": 15580
+    },
+    {
+      "grad_norm": 0.18338501453399658,
+      "learning_rate": 8.240571523436896e-05,
+      "loss": 0.0088,
+      "step": 15590
+    },
+    {
+      "grad_norm": 0.22029252350330353,
+      "learning_rate": 8.238052437888955e-05,
+      "loss": 0.0099,
+      "step": 15600
+    },
+    {
+      "grad_norm": 0.23749051988124847,
+      "learning_rate": 8.235531935906866e-05,
+      "loss": 0.01,
+      "step": 15610
+    },
+    {
+      "grad_norm": 0.20248886942863464,
+      "learning_rate": 8.233010018593185e-05,
+      "loss": 0.0097,
+      "step": 15620
+    },
+    {
+      "grad_norm": 0.21900875866413116,
+      "learning_rate": 8.230486687051082e-05,
+      "loss": 0.0091,
+      "step": 15630
+    },
+    {
+      "grad_norm": 0.263130247592926,
+      "learning_rate": 8.22796194238435e-05,
+      "loss": 0.0098,
+      "step": 15640
+    },
+    {
+      "grad_norm": 0.18036025762557983,
+      "learning_rate": 8.225435785697397e-05,
+      "loss": 0.01,
+      "step": 15650
+    },
+    {
+      "grad_norm": 0.16856826841831207,
+      "learning_rate": 8.222908218095252e-05,
+      "loss": 0.0101,
+      "step": 15660
+    },
+    {
+      "grad_norm": 0.19429641962051392,
+      "learning_rate": 8.220379240683553e-05,
+      "loss": 0.0088,
+      "step": 15670
+    },
+    {
+      "grad_norm": 0.1724906861782074,
+      "learning_rate": 8.217848854568566e-05,
+      "loss": 0.0093,
+      "step": 15680
+    },
+    {
+      "grad_norm": 0.1849764734506607,
+      "learning_rate": 8.215317060857167e-05,
+      "loss": 0.0093,
+      "step": 15690
+    },
+    {
+      "grad_norm": 0.16924457252025604,
+      "learning_rate": 8.212783860656847e-05,
+      "loss": 0.0092,
+      "step": 15700
+    },
+    {
+      "grad_norm": 0.22601939737796783,
+      "learning_rate": 8.210249255075717e-05,
+      "loss": 0.0111,
+      "step": 15710
+    },
+    {
+      "grad_norm": 0.2651759088039398,
+      "learning_rate": 8.207713245222496e-05,
+      "loss": 0.0113,
+      "step": 15720
+    },
+    {
+      "grad_norm": 0.2038411647081375,
+      "learning_rate": 8.205175832206522e-05,
+      "loss": 0.0112,
+      "step": 15730
+    },
+    {
+      "grad_norm": 0.22682759165763855,
+      "learning_rate": 8.202637017137747e-05,
+      "loss": 0.0105,
+      "step": 15740
+    },
+    {
+      "grad_norm": 0.2212003767490387,
+      "learning_rate": 8.200096801126734e-05,
+      "loss": 0.0122,
+      "step": 15750
+    },
+    {
+      "grad_norm": 0.17527377605438232,
+      "learning_rate": 8.197555185284658e-05,
+      "loss": 0.0102,
+      "step": 15760
+    },
+    {
+      "grad_norm": 0.26345452666282654,
+      "learning_rate": 8.19501217072331e-05,
+      "loss": 0.0117,
+      "step": 15770
+    },
+    {
+      "grad_norm": 0.25839823484420776,
+      "learning_rate": 8.192467758555091e-05,
+      "loss": 0.0114,
+      "step": 15780
+    },
+    {
+      "grad_norm": 0.2349306046962738,
+      "learning_rate": 8.189921949893014e-05,
+      "loss": 0.0122,
+      "step": 15790
+    },
+    {
+      "grad_norm": 0.3000774681568146,
+      "learning_rate": 8.187374745850702e-05,
+      "loss": 0.0122,
+      "step": 15800
+    },
+    {
+      "grad_norm": 0.21562987565994263,
+      "learning_rate": 8.184826147542387e-05,
+      "loss": 0.011,
+      "step": 15810
+    },
+    {
+      "grad_norm": 0.17892439663410187,
+      "learning_rate": 8.18227615608291e-05,
+      "loss": 0.0109,
+      "step": 15820
+    },
+    {
+      "grad_norm": 0.19729353487491608,
+      "learning_rate": 8.179724772587732e-05,
+      "loss": 0.0107,
+      "step": 15830
+    },
+    {
+      "grad_norm": 0.2428709715604782,
+      "learning_rate": 8.177171998172908e-05,
+      "loss": 0.0105,
+      "step": 15840
+    },
+    {
+      "grad_norm": 0.21920756995677948,
+      "learning_rate": 8.174617833955112e-05,
+      "loss": 0.0111,
+      "step": 15850
+    },
+    {
+      "grad_norm": 0.21037311851978302,
+      "learning_rate": 8.172062281051618e-05,
+      "loss": 0.0128,
+      "step": 15860
+    },
+    {
+      "grad_norm": 0.2477322816848755,
+      "learning_rate": 8.169505340580316e-05,
+      "loss": 0.0104,
+      "step": 15870
+    },
+    {
+      "grad_norm": 0.18319985270500183,
+      "learning_rate": 8.166947013659696e-05,
+      "loss": 0.0111,
+      "step": 15880
+    },
+    {
+      "grad_norm": 0.25348401069641113,
+      "learning_rate": 8.164387301408858e-05,
+      "loss": 0.0117,
+      "step": 15890
+    },
+    {
+      "grad_norm": 0.21175603568553925,
+      "learning_rate": 8.161826204947507e-05,
+      "loss": 0.0106,
+      "step": 15900
+    },
+    {
+      "grad_norm": 0.2912689745426178,
+      "learning_rate": 8.159263725395954e-05,
+      "loss": 0.0101,
+      "step": 15910
+    },
+    {
+      "grad_norm": 0.1741502434015274,
+      "learning_rate": 8.156699863875115e-05,
+      "loss": 0.0103,
+      "step": 15920
+    },
+    {
+      "grad_norm": 0.180027037858963,
+      "learning_rate": 8.154134621506508e-05,
+      "loss": 0.0107,
+      "step": 15930
+    },
+    {
+      "grad_norm": 0.20615144073963165,
+      "learning_rate": 8.15156799941226e-05,
+      "loss": 0.0093,
+      "step": 15940
+    },
+    {
+      "grad_norm": 0.3746310770511627,
+      "learning_rate": 8.148999998715096e-05,
+      "loss": 0.0114,
+      "step": 15950
+    },
+    {
+      "grad_norm": 0.19320674240589142,
+      "learning_rate": 8.146430620538348e-05,
+      "loss": 0.0094,
+      "step": 15960
+    },
+    {
+      "grad_norm": 0.24865910410881042,
+      "learning_rate": 8.143859866005949e-05,
+      "loss": 0.0109,
+      "step": 15970
+    },
+    {
+      "grad_norm": 0.2946818172931671,
+      "learning_rate": 8.141287736242438e-05,
+      "loss": 0.0112,
+      "step": 15980
+    },
+    {
+      "grad_norm": 0.21331259608268738,
+      "learning_rate": 8.138714232372945e-05,
+      "loss": 0.0109,
+      "step": 15990
+    },
+    {
+      "grad_norm": 0.19751663506031036,
+      "learning_rate": 8.136139355523213e-05,
+      "loss": 0.01,
+      "step": 16000
+    },
+    {
+      "grad_norm": 0.23256437480449677,
+      "learning_rate": 8.13356310681958e-05,
+      "loss": 0.0094,
+      "step": 16010
+    },
+    {
+      "grad_norm": 0.20419779419898987,
+      "learning_rate": 8.130985487388982e-05,
+      "loss": 0.011,
+      "step": 16020
+    },
+    {
+      "grad_norm": 0.22953154146671295,
+      "learning_rate": 8.12840649835896e-05,
+      "loss": 0.0102,
+      "step": 16030
+    },
+    {
+      "grad_norm": 0.207570880651474,
+      "learning_rate": 8.125826140857653e-05,
+      "loss": 0.0117,
+      "step": 16040
+    },
+    {
+      "grad_norm": 0.25653305649757385,
+      "learning_rate": 8.123244416013792e-05,
+      "loss": 0.0111,
+      "step": 16050
+    },
+    {
+      "grad_norm": 0.22336165606975555,
+      "learning_rate": 8.120661324956716e-05,
+      "loss": 0.0123,
+      "step": 16060
+    },
+    {
+      "grad_norm": 0.2260415107011795,
+      "learning_rate": 8.118076868816353e-05,
+      "loss": 0.01,
+      "step": 16070
+    },
+    {
+      "grad_norm": 0.22283132374286652,
+      "learning_rate": 8.115491048723234e-05,
+      "loss": 0.011,
+      "step": 16080
+    },
+    {
+      "grad_norm": 0.2176639586687088,
+      "learning_rate": 8.112903865808486e-05,
+      "loss": 0.013,
+      "step": 16090
+    },
+    {
+      "grad_norm": 0.25164854526519775,
+      "learning_rate": 8.110315321203826e-05,
+      "loss": 0.0081,
+      "step": 16100
+    },
+    {
+      "grad_norm": 0.23929466307163239,
+      "learning_rate": 8.107725416041575e-05,
+      "loss": 0.0098,
+      "step": 16110
+    },
+    {
+      "grad_norm": 0.15176145732402802,
+      "learning_rate": 8.105134151454645e-05,
+      "loss": 0.0101,
+      "step": 16120
+    },
+    {
+      "grad_norm": 0.24426348507404327,
+      "learning_rate": 8.102541528576544e-05,
+      "loss": 0.0102,
+      "step": 16130
+    },
+    {
+      "grad_norm": 0.23256120085716248,
+      "learning_rate": 8.099947548541371e-05,
+      "loss": 0.0101,
+      "step": 16140
+    },
+    {
+      "grad_norm": 0.17810161411762238,
+      "learning_rate": 8.097352212483824e-05,
+      "loss": 0.0106,
+      "step": 16150
+    },
+    {
+      "grad_norm": 0.1843309998512268,
+      "learning_rate": 8.094755521539187e-05,
+      "loss": 0.0101,
+      "step": 16160
+    },
+    {
+      "grad_norm": 0.16482743620872498,
+      "learning_rate": 8.092157476843346e-05,
+      "loss": 0.01,
+      "step": 16170
+    },
+    {
+      "grad_norm": 0.2661721408367157,
+      "learning_rate": 8.089558079532769e-05,
+      "loss": 0.0102,
+      "step": 16180
+    },
+    {
+      "grad_norm": 0.25680652260780334,
+      "learning_rate": 8.086957330744522e-05,
+      "loss": 0.0109,
+      "step": 16190
+    },
+    {
+      "grad_norm": 0.18935170769691467,
+      "learning_rate": 8.084355231616265e-05,
+      "loss": 0.0139,
+      "step": 16200
+    },
+    {
+      "grad_norm": 0.30074724555015564,
+      "learning_rate": 8.08175178328624e-05,
+      "loss": 0.0112,
+      "step": 16210
+    },
+    {
+      "grad_norm": 0.23491744697093964,
+      "learning_rate": 8.079146986893284e-05,
+      "loss": 0.0121,
+      "step": 16220
+    },
+    {
+      "grad_norm": 0.13554248213768005,
+      "learning_rate": 8.076540843576826e-05,
+      "loss": 0.0105,
+      "step": 16230
+    },
+    {
+      "grad_norm": 0.20584659278392792,
+      "learning_rate": 8.073933354476879e-05,
+      "loss": 0.0088,
+      "step": 16240
+    },
+    {
+      "grad_norm": 0.33377841114997864,
+      "learning_rate": 8.07132452073405e-05,
+      "loss": 0.0134,
+      "step": 16250
+    },
+    {
+      "grad_norm": 0.23628592491149902,
+      "learning_rate": 8.068714343489529e-05,
+      "loss": 0.0116,
+      "step": 16260
+    },
+    {
+      "grad_norm": 0.231981560587883,
+      "learning_rate": 8.066102823885097e-05,
+      "loss": 0.0122,
+      "step": 16270
+    },
+    {
+      "grad_norm": 0.24148425459861755,
+      "learning_rate": 8.063489963063125e-05,
+      "loss": 0.0106,
+      "step": 16280
+    },
+    {
+      "grad_norm": 0.3146168887615204,
+      "learning_rate": 8.06087576216656e-05,
+      "loss": 0.0144,
+      "step": 16290
+    },
+    {
+      "grad_norm": 0.2446994036436081,
+      "learning_rate": 8.058260222338946e-05,
+      "loss": 0.0126,
+      "step": 16300
+    },
+    {
+      "grad_norm": 0.258609414100647,
+      "learning_rate": 8.05564334472441e-05,
+      "loss": 0.0106,
+      "step": 16310
+    },
+    {
+      "grad_norm": 0.19223931431770325,
+      "learning_rate": 8.053025130467664e-05,
+      "loss": 0.0105,
+      "step": 16320
+    },
+    {
+      "grad_norm": 0.128304123878479,
+      "learning_rate": 8.050405580714e-05,
+      "loss": 0.011,
+      "step": 16330
+    },
+    {
+      "grad_norm": 0.24637804925441742,
+      "learning_rate": 8.047784696609301e-05,
+      "loss": 0.0109,
+      "step": 16340
+    },
+    {
+      "grad_norm": 0.24292460083961487,
+      "learning_rate": 8.045162479300029e-05,
+      "loss": 0.0098,
+      "step": 16350
+    },
+    {
+      "grad_norm": 0.21124234795570374,
+      "learning_rate": 8.042538929933232e-05,
+      "loss": 0.0107,
+      "step": 16360
+    },
+    {
+      "grad_norm": 0.20458976924419403,
+      "learning_rate": 8.039914049656539e-05,
+      "loss": 0.0094,
+      "step": 16370
+    },
+    {
+      "grad_norm": 0.23097732663154602,
+      "learning_rate": 8.037287839618163e-05,
+      "loss": 0.0094,
+      "step": 16380
+    },
+    {
+      "grad_norm": 0.18364477157592773,
+      "learning_rate": 8.034660300966898e-05,
+      "loss": 0.0118,
+      "step": 16390
+    },
+    {
+      "grad_norm": 0.22090548276901245,
+      "learning_rate": 8.032031434852115e-05,
+      "loss": 0.0096,
+      "step": 16400
+    },
+    {
+      "grad_norm": 0.17480236291885376,
+      "learning_rate": 8.029401242423773e-05,
+      "loss": 0.0089,
+      "step": 16410
+    },
+    {
+      "grad_norm": 0.19305691123008728,
+      "learning_rate": 8.026769724832406e-05,
+      "loss": 0.0088,
+      "step": 16420
+    },
+    {
+      "grad_norm": 0.18178844451904297,
+      "learning_rate": 8.02413688322913e-05,
+      "loss": 0.0104,
+      "step": 16430
+    },
+    {
+      "grad_norm": 0.17978742718696594,
+      "learning_rate": 8.021502718765639e-05,
+      "loss": 0.0092,
+      "step": 16440
+    },
+    {
+      "grad_norm": 0.24330134689807892,
+      "learning_rate": 8.018867232594204e-05,
+      "loss": 0.0106,
+      "step": 16450
+    },
+    {
+      "grad_norm": 0.1695333868265152,
+      "learning_rate": 8.016230425867679e-05,
+      "loss": 0.0087,
+      "step": 16460
+    },
+    {
+      "grad_norm": 0.18313801288604736,
+      "learning_rate": 8.013592299739494e-05,
+      "loss": 0.0087,
+      "step": 16470
+    },
+    {
+      "grad_norm": 0.193218395113945,
+      "learning_rate": 8.01095285536365e-05,
+      "loss": 0.0111,
+      "step": 16480
+    },
+    {
+      "grad_norm": 0.2135763019323349,
+      "learning_rate": 8.008312093894735e-05,
+      "loss": 0.0118,
+      "step": 16490
+    },
+    {
+      "grad_norm": 0.24383479356765747,
+      "learning_rate": 8.005670016487903e-05,
+      "loss": 0.0082,
+      "step": 16500
+    },
+    {
+      "grad_norm": 0.2030351459980011,
+      "learning_rate": 8.003026624298892e-05,
+      "loss": 0.0101,
+      "step": 16510
+    },
+    {
+      "grad_norm": 0.18509720265865326,
+      "learning_rate": 8.00038191848401e-05,
+      "loss": 0.0101,
+      "step": 16520
+    },
+    {
+      "grad_norm": 0.23461537063121796,
+      "learning_rate": 7.997735900200142e-05,
+      "loss": 0.0106,
+      "step": 16530
+    },
+    {
+      "grad_norm": 0.22303499281406403,
+      "learning_rate": 7.995088570604745e-05,
+      "loss": 0.0091,
+      "step": 16540
+    },
+    {
+      "grad_norm": 0.19179308414459229,
+      "learning_rate": 7.992439930855852e-05,
+      "loss": 0.0102,
+      "step": 16550
+    },
+    {
+      "grad_norm": 0.20868606865406036,
+      "learning_rate": 7.989789982112067e-05,
+      "loss": 0.0104,
+      "step": 16560
+    },
+    {
+      "grad_norm": 0.21514029800891876,
+      "learning_rate": 7.987138725532572e-05,
+      "loss": 0.0103,
+      "step": 16570
+    },
+    {
+      "grad_norm": 0.23649436235427856,
+      "learning_rate": 7.98448616227711e-05,
+      "loss": 0.012,
+      "step": 16580
+    },
+    {
+      "grad_norm": 0.1540248543024063,
+      "learning_rate": 7.981832293506004e-05,
+      "loss": 0.0106,
+      "step": 16590
+    },
+    {
+      "grad_norm": 0.2536598742008209,
+      "learning_rate": 7.97917712038015e-05,
+      "loss": 0.013,
+      "step": 16600
+    },
+    {
+      "grad_norm": 0.23496247828006744,
+      "learning_rate": 7.976520644061007e-05,
+      "loss": 0.0099,
+      "step": 16610
+    },
+    {
+      "grad_norm": 0.1858106404542923,
+      "learning_rate": 7.973862865710611e-05,
+      "loss": 0.0093,
+      "step": 16620
+    },
+    {
+      "grad_norm": 0.2595023214817047,
+      "learning_rate": 7.971203786491563e-05,
+      "loss": 0.0094,
+      "step": 16630
+    },
+    {
+      "grad_norm": 0.2205297201871872,
+      "learning_rate": 7.968543407567034e-05,
+      "loss": 0.012,
+      "step": 16640
+    },
+    {
+      "grad_norm": 0.23119135200977325,
+      "learning_rate": 7.965881730100765e-05,
+      "loss": 0.0099,
+      "step": 16650
+    },
+    {
+      "grad_norm": 0.2440512776374817,
+      "learning_rate": 7.963218755257064e-05,
+      "loss": 0.0104,
+      "step": 16660
+    },
+    {
+      "grad_norm": 0.1835424304008484,
+      "learning_rate": 7.960554484200807e-05,
+      "loss": 0.011,
+      "step": 16670
+    },
+    {
+      "grad_norm": 0.24261899292469025,
+      "learning_rate": 7.957888918097436e-05,
+      "loss": 0.0103,
+      "step": 16680
+    },
+    {
+      "grad_norm": 0.18076327443122864,
+      "learning_rate": 7.955222058112962e-05,
+      "loss": 0.0142,
+      "step": 16690
+    },
+    {
+      "grad_norm": 0.2230982631444931,
+      "learning_rate": 7.952553905413955e-05,
+      "loss": 0.0108,
+      "step": 16700
+    },
+    {
+      "grad_norm": 0.19099460542201996,
+      "learning_rate": 7.949884461167563e-05,
+      "loss": 0.0101,
+      "step": 16710
+    },
+    {
+      "grad_norm": 0.23034214973449707,
+      "learning_rate": 7.947213726541484e-05,
+      "loss": 0.0107,
+      "step": 16720
+    },
+    {
+      "grad_norm": 0.21275366842746735,
+      "learning_rate": 7.944541702703996e-05,
+      "loss": 0.0093,
+      "step": 16730
+    },
+    {
+      "grad_norm": 0.18964658677577972,
+      "learning_rate": 7.941868390823928e-05,
+      "loss": 0.0089,
+      "step": 16740
+    },
+    {
+      "grad_norm": 0.21806354820728302,
+      "learning_rate": 7.939193792070677e-05,
+      "loss": 0.0118,
+      "step": 16750
+    },
+    {
+      "grad_norm": 0.16798759996891022,
+      "learning_rate": 7.936517907614207e-05,
+      "loss": 0.0094,
+      "step": 16760
+    },
+    {
+      "grad_norm": 0.19628089666366577,
+      "learning_rate": 7.933840738625036e-05,
+      "loss": 0.0102,
+      "step": 16770
+    },
+    {
+      "grad_norm": 0.224085733294487,
+      "learning_rate": 7.931162286274252e-05,
+      "loss": 0.0094,
+      "step": 16780
+    },
+    {
+      "grad_norm": 0.2057165652513504,
+      "learning_rate": 7.928482551733503e-05,
+      "loss": 0.0097,
+      "step": 16790
+    },
+    {
+      "grad_norm": 0.2671113908290863,
+      "learning_rate": 7.925801536174991e-05,
+      "loss": 0.0116,
+      "step": 16800
+    },
+    {
+      "grad_norm": 0.19563789665699005,
+      "learning_rate": 7.923119240771484e-05,
+      "loss": 0.0116,
+      "step": 16810
+    },
+    {
+      "grad_norm": 0.16877222061157227,
+      "learning_rate": 7.920435666696313e-05,
+      "loss": 0.0108,
+      "step": 16820
+    },
+    {
+      "grad_norm": 0.23126815259456635,
+      "learning_rate": 7.917750815123362e-05,
+      "loss": 0.0108,
+      "step": 16830
+    },
+    {
+      "grad_norm": 0.16751179099082947,
+      "learning_rate": 7.915064687227075e-05,
+      "loss": 0.0092,
+      "step": 16840
+    },
+    {
+      "grad_norm": 0.19327817857265472,
+      "learning_rate": 7.912377284182458e-05,
+      "loss": 0.0096,
+      "step": 16850
+    },
+    {
+      "grad_norm": 0.296217143535614,
+      "learning_rate": 7.90968860716507e-05,
+      "loss": 0.0094,
+      "step": 16860
+    },
+    {
+      "grad_norm": 0.1779392510652542,
+      "learning_rate": 7.906998657351033e-05,
+      "loss": 0.0105,
+      "step": 16870
+    },
+    {
+      "grad_norm": 0.19061607122421265,
+      "learning_rate": 7.90430743591702e-05,
+      "loss": 0.0088,
+      "step": 16880
+    },
+    {
+      "grad_norm": 0.21069100499153137,
+      "learning_rate": 7.901614944040263e-05,
+      "loss": 0.0099,
+      "step": 16890
+    },
+    {
+      "grad_norm": 0.19733178615570068,
+      "learning_rate": 7.89892118289855e-05,
+      "loss": 0.012,
+      "step": 16900
+    },
+    {
+      "grad_norm": 0.19836901128292084,
+      "learning_rate": 7.896226153670223e-05,
+      "loss": 0.0096,
+      "step": 16910
+    },
+    {
+      "grad_norm": 0.22530724108219147,
+      "learning_rate": 7.89352985753418e-05,
+      "loss": 0.0099,
+      "step": 16920
+    },
+    {
+      "grad_norm": 0.19396522641181946,
+      "learning_rate": 7.890832295669871e-05,
+      "loss": 0.0095,
+      "step": 16930
+    },
+    {
+      "grad_norm": 0.27126461267471313,
+      "learning_rate": 7.888133469257303e-05,
+      "loss": 0.0091,
+      "step": 16940
+    },
+    {
+      "grad_norm": 0.21758265793323517,
+      "learning_rate": 7.885433379477034e-05,
+      "loss": 0.0093,
+      "step": 16950
+    },
+    {
+      "grad_norm": 0.19670170545578003,
+      "learning_rate": 7.882732027510174e-05,
+      "loss": 0.01,
+      "step": 16960
+    },
+    {
+      "grad_norm": 0.19757317006587982,
+      "learning_rate": 7.880029414538386e-05,
+      "loss": 0.009,
+      "step": 16970
+    },
+    {
+      "grad_norm": 0.21345657110214233,
+      "learning_rate": 7.877325541743888e-05,
+      "loss": 0.011,
+      "step": 16980
+    },
+    {
+      "grad_norm": 0.21299876272678375,
+      "learning_rate": 7.874620410309443e-05,
+      "loss": 0.0126,
+      "step": 16990
+    },
+    {
+      "grad_norm": 0.24101951718330383,
+      "learning_rate": 7.871914021418368e-05,
+      "loss": 0.0117,
+      "step": 17000
+    },
+    {
+      "grad_norm": 0.2447722852230072,
+      "learning_rate": 7.869206376254528e-05,
+      "loss": 0.0114,
+      "step": 17010
+    },
+    {
+      "grad_norm": 0.20511581003665924,
+      "learning_rate": 7.86649747600234e-05,
+      "loss": 0.0112,
+      "step": 17020
+    },
+    {
+      "grad_norm": 0.16365543007850647,
+      "learning_rate": 7.86378732184677e-05,
+      "loss": 0.009,
+      "step": 17030
+    },
+    {
+      "grad_norm": 0.2212393432855606,
+      "learning_rate": 7.86107591497333e-05,
+      "loss": 0.0107,
+      "step": 17040
+    },
+    {
+      "grad_norm": 0.18850743770599365,
+      "learning_rate": 7.858363256568083e-05,
+      "loss": 0.0087,
+      "step": 17050
+    },
+    {
+      "grad_norm": 0.19118143618106842,
+      "learning_rate": 7.855649347817637e-05,
+      "loss": 0.0113,
+      "step": 17060
+    },
+    {
+      "grad_norm": 0.18384180963039398,
+      "learning_rate": 7.85293418990915e-05,
+      "loss": 0.0086,
+      "step": 17070
+    },
+    {
+      "grad_norm": 0.17348003387451172,
+      "learning_rate": 7.85021778403032e-05,
+      "loss": 0.0104,
+      "step": 17080
+    },
+    {
+      "grad_norm": 0.2571563422679901,
+      "learning_rate": 7.847500131369397e-05,
+      "loss": 0.0096,
+      "step": 17090
+    },
+    {
+      "grad_norm": 0.20301920175552368,
+      "learning_rate": 7.844781233115176e-05,
+      "loss": 0.0098,
+      "step": 17100
+    },
+    {
+      "grad_norm": 0.19150643050670624,
+      "learning_rate": 7.842061090456996e-05,
+      "loss": 0.0118,
+      "step": 17110
+    },
+    {
+      "grad_norm": 0.2576228082180023,
+      "learning_rate": 7.839339704584736e-05,
+      "loss": 0.0092,
+      "step": 17120
+    },
+    {
+      "grad_norm": 0.1443592756986618,
+      "learning_rate": 7.836617076688828e-05,
+      "loss": 0.0098,
+      "step": 17130
+    },
+    {
+      "grad_norm": 0.2071804702281952,
+      "learning_rate": 7.833893207960236e-05,
+      "loss": 0.0109,
+      "step": 17140
+    },
+    {
+      "grad_norm": 0.150458425283432,
+      "learning_rate": 7.831168099590478e-05,
+      "loss": 0.0095,
+      "step": 17150
+    },
+    {
+      "grad_norm": 0.1679334044456482,
+      "learning_rate": 7.828441752771607e-05,
+      "loss": 0.0106,
+      "step": 17160
+    },
+    {
+      "grad_norm": 0.16198943555355072,
+      "learning_rate": 7.825714168696219e-05,
+      "loss": 0.0095,
+      "step": 17170
+    },
+    {
+      "grad_norm": 0.1980520337820053,
+      "learning_rate": 7.82298534855745e-05,
+      "loss": 0.0105,
+      "step": 17180
+    },
+    {
+      "grad_norm": 0.24578545987606049,
+      "learning_rate": 7.820255293548984e-05,
+      "loss": 0.0094,
+      "step": 17190
+    },
+    {
+      "grad_norm": 0.26301172375679016,
+      "learning_rate": 7.817524004865035e-05,
+      "loss": 0.0093,
+      "step": 17200
+    },
+    {
+      "grad_norm": 0.2370380163192749,
+      "learning_rate": 7.814791483700366e-05,
+      "loss": 0.0097,
+      "step": 17210
+    },
+    {
+      "grad_norm": 0.20842283964157104,
+      "learning_rate": 7.81205773125027e-05,
+      "loss": 0.0115,
+      "step": 17220
+    },
+    {
+      "grad_norm": 0.20149433612823486,
+      "learning_rate": 7.809322748710585e-05,
+      "loss": 0.0096,
+      "step": 17230
+    },
+    {
+      "grad_norm": 0.19841772317886353,
+      "learning_rate": 7.806586537277685e-05,
+      "loss": 0.0089,
+      "step": 17240
+    },
+    {
+      "grad_norm": 0.2789178490638733,
+      "learning_rate": 7.803849098148484e-05,
+      "loss": 0.0096,
+      "step": 17250
+    },
+    {
+      "grad_norm": 0.22148409485816956,
+      "learning_rate": 7.801110432520427e-05,
+      "loss": 0.0094,
+      "step": 17260
+    },
+    {
+      "grad_norm": 0.28113579750061035,
+      "learning_rate": 7.798370541591504e-05,
+      "loss": 0.0089,
+      "step": 17270
+    },
+    {
+      "grad_norm": 0.21594908833503723,
+      "learning_rate": 7.795629426560232e-05,
+      "loss": 0.0098,
+      "step": 17280
+    },
+    {
+      "grad_norm": 0.17025430500507355,
+      "learning_rate": 7.792887088625671e-05,
+      "loss": 0.0086,
+      "step": 17290
+    },
+    {
+      "grad_norm": 0.18512755632400513,
+      "learning_rate": 7.790143528987412e-05,
+      "loss": 0.0111,
+      "step": 17300
+    },
+    {
+      "grad_norm": 0.2314087450504303,
+      "learning_rate": 7.787398748845578e-05,
+      "loss": 0.011,
+      "step": 17310
+    },
+    {
+      "grad_norm": 0.1936812698841095,
+      "learning_rate": 7.784652749400834e-05,
+      "loss": 0.0092,
+      "step": 17320
+    },
+    {
+      "grad_norm": 0.26718634366989136,
+      "learning_rate": 7.78190553185437e-05,
+      "loss": 0.0107,
+      "step": 17330
+    },
+    {
+      "grad_norm": 0.16288772225379944,
+      "learning_rate": 7.779157097407915e-05,
+      "loss": 0.0081,
+      "step": 17340
+    },
+    {
+      "grad_norm": 0.15022024512290955,
+      "learning_rate": 7.776407447263725e-05,
+      "loss": 0.0093,
+      "step": 17350
+    },
+    {
+      "grad_norm": 0.2377467155456543,
+      "learning_rate": 7.773656582624593e-05,
+      "loss": 0.0085,
+      "step": 17360
+    },
+    {
+      "grad_norm": 0.26467546820640564,
+      "learning_rate": 7.770904504693837e-05,
+      "loss": 0.01,
+      "step": 17370
+    },
+    {
+      "grad_norm": 0.21886354684829712,
+      "learning_rate": 7.768151214675314e-05,
+      "loss": 0.0098,
+      "step": 17380
+    },
+    {
+      "grad_norm": 0.2448604851961136,
+      "learning_rate": 7.765396713773402e-05,
+      "loss": 0.0103,
+      "step": 17390
+    },
+    {
+      "grad_norm": 0.20872262120246887,
+      "learning_rate": 7.762641003193017e-05,
+      "loss": 0.0128,
+      "step": 17400
+    },
+    {
+      "grad_norm": 0.2037310153245926,
+      "learning_rate": 7.759884084139597e-05,
+      "loss": 0.0088,
+      "step": 17410
+    },
+    {
+      "grad_norm": 0.2071525603532791,
+      "learning_rate": 7.757125957819116e-05,
+      "loss": 0.0087,
+      "step": 17420
+    },
+    {
+      "grad_norm": 0.20220065116882324,
+      "learning_rate": 7.754366625438069e-05,
+      "loss": 0.0097,
+      "step": 17430
+    },
+    {
+      "grad_norm": 0.1751326471567154,
+      "learning_rate": 7.751606088203483e-05,
+      "loss": 0.0077,
+      "step": 17440
+    },
+    {
+      "grad_norm": 0.16065779328346252,
+      "learning_rate": 7.74884434732291e-05,
+      "loss": 0.0078,
+      "step": 17450
+    },
+    {
+      "grad_norm": 0.2173767387866974,
+      "learning_rate": 7.746081404004431e-05,
+      "loss": 0.0109,
+      "step": 17460
+    },
+    {
+      "grad_norm": 0.2863403856754303,
+      "learning_rate": 7.743317259456649e-05,
+      "loss": 0.0104,
+      "step": 17470
+    },
+    {
+      "grad_norm": 0.18642546236515045,
+      "learning_rate": 7.740551914888696e-05,
+      "loss": 0.0091,
+      "step": 17480
+    },
+    {
+      "grad_norm": 0.20981918275356293,
+      "learning_rate": 7.737785371510229e-05,
+      "loss": 0.0105,
+      "step": 17490
+    },
+    {
+      "grad_norm": 0.18988960981369019,
+      "learning_rate": 7.735017630531426e-05,
+      "loss": 0.0084,
+      "step": 17500
+    },
+    {
+      "grad_norm": 0.18471072614192963,
+      "learning_rate": 7.732248693162991e-05,
+      "loss": 0.0082,
+      "step": 17510
+    },
+    {
+      "grad_norm": 0.13725532591342926,
+      "learning_rate": 7.729478560616153e-05,
+      "loss": 0.011,
+      "step": 17520
+    },
+    {
+      "grad_norm": 0.2293591946363449,
+      "learning_rate": 7.726707234102659e-05,
+      "loss": 0.0113,
+      "step": 17530
+    },
+    {
+      "grad_norm": 0.287235826253891,
+      "learning_rate": 7.723934714834785e-05,
+      "loss": 0.0106,
+      "step": 17540
+    },
+    {
+      "grad_norm": 0.255228728055954,
+      "learning_rate": 7.721161004025323e-05,
+      "loss": 0.0109,
+      "step": 17550
+    },
+    {
+      "grad_norm": 0.21882838010787964,
+      "learning_rate": 7.718386102887588e-05,
+      "loss": 0.0123,
+      "step": 17560
+    },
+    {
+      "grad_norm": 0.18785786628723145,
+      "learning_rate": 7.715610012635418e-05,
+      "loss": 0.0131,
+      "step": 17570
+    },
+    {
+      "grad_norm": 0.25786688923835754,
+      "learning_rate": 7.712832734483166e-05,
+      "loss": 0.0116,
+      "step": 17580
+    },
+    {
+      "grad_norm": 0.2684156596660614,
+      "learning_rate": 7.710054269645713e-05,
+      "loss": 0.0115,
+      "step": 17590
+    },
+    {
+      "grad_norm": 0.286653995513916,
+      "learning_rate": 7.707274619338449e-05,
+      "loss": 0.0115,
+      "step": 17600
+    },
+    {
+      "grad_norm": 0.22645710408687592,
+      "learning_rate": 7.704493784777289e-05,
+      "loss": 0.01,
+      "step": 17610
+    },
+    {
+      "grad_norm": 0.19340653717517853,
+      "learning_rate": 7.701711767178665e-05,
+      "loss": 0.0081,
+      "step": 17620
+    },
+    {
+      "grad_norm": 0.23091170191764832,
+      "learning_rate": 7.698928567759525e-05,
+      "loss": 0.0116,
+      "step": 17630
+    },
+    {
+      "grad_norm": 0.18861931562423706,
+      "learning_rate": 7.696144187737335e-05,
+      "loss": 0.0109,
+      "step": 17640
+    },
+    {
+      "grad_norm": 0.16499902307987213,
+      "learning_rate": 7.69335862833008e-05,
+      "loss": 0.0105,
+      "step": 17650
+    },
+    {
+      "grad_norm": 0.1859208345413208,
+      "learning_rate": 7.690571890756256e-05,
+      "loss": 0.0105,
+      "step": 17660
+    },
+    {
+      "grad_norm": 0.1855270117521286,
+      "learning_rate": 7.687783976234877e-05,
+      "loss": 0.0086,
+      "step": 17670
+    },
+    {
+      "grad_norm": 0.1457616686820984,
+      "learning_rate": 7.684994885985472e-05,
+      "loss": 0.0097,
+      "step": 17680
+    },
+    {
+      "grad_norm": 0.18069764971733093,
+      "learning_rate": 7.682204621228083e-05,
+      "loss": 0.0129,
+      "step": 17690
+    },
+    {
+      "grad_norm": 0.19273017346858978,
+      "learning_rate": 7.679413183183267e-05,
+      "loss": 0.0095,
+      "step": 17700
+    },
+    {
+      "grad_norm": 0.12686210870742798,
+      "learning_rate": 7.676620573072093e-05,
+      "loss": 0.0089,
+      "step": 17710
+    },
+    {
+      "grad_norm": 0.19961248338222504,
+      "learning_rate": 7.673826792116145e-05,
+      "loss": 0.0087,
+      "step": 17720
+    },
+    {
+      "grad_norm": 0.2253817468881607,
+      "learning_rate": 7.671031841537519e-05,
+      "loss": 0.0091,
+      "step": 17730
+    },
+    {
+      "grad_norm": 0.17011620104312897,
+      "learning_rate": 7.668235722558819e-05,
+      "loss": 0.0084,
+      "step": 17740
+    },
+    {
+      "grad_norm": 0.23978161811828613,
+      "learning_rate": 7.665438436403161e-05,
+      "loss": 0.009,
+      "step": 17750
+    },
+    {
+      "grad_norm": 0.20312705636024475,
+      "learning_rate": 7.662639984294178e-05,
+      "loss": 0.0088,
+      "step": 17760
+    },
+    {
+      "grad_norm": 0.20985554158687592,
+      "learning_rate": 7.659840367456002e-05,
+      "loss": 0.0087,
+      "step": 17770
+    },
+    {
+      "grad_norm": 0.18335188925266266,
+      "learning_rate": 7.657039587113287e-05,
+      "loss": 0.009,
+      "step": 17780
+    },
+    {
+      "grad_norm": 0.23580825328826904,
+      "learning_rate": 7.654237644491182e-05,
+      "loss": 0.01,
+      "step": 17790
+    },
+    {
+      "grad_norm": 0.19935454428195953,
+      "learning_rate": 7.651434540815358e-05,
+      "loss": 0.009,
+      "step": 17800
+    },
+    {
+      "grad_norm": 0.20525969564914703,
+      "learning_rate": 7.648630277311986e-05,
+      "loss": 0.0104,
+      "step": 17810
+    },
+    {
+      "grad_norm": 0.16337430477142334,
+      "learning_rate": 7.645824855207744e-05,
+      "loss": 0.01,
+      "step": 17820
+    },
+    {
+      "grad_norm": 0.16541628539562225,
+      "learning_rate": 7.643018275729821e-05,
+      "loss": 0.0085,
+      "step": 17830
+    },
+    {
+      "grad_norm": 0.18167831003665924,
+      "learning_rate": 7.640210540105911e-05,
+      "loss": 0.0099,
+      "step": 17840
+    },
+    {
+      "grad_norm": 0.20917296409606934,
+      "learning_rate": 7.637401649564213e-05,
+      "loss": 0.0103,
+      "step": 17850
+    },
+    {
+      "grad_norm": 0.24071529507637024,
+      "learning_rate": 7.63459160533343e-05,
+      "loss": 0.0092,
+      "step": 17860
+    },
+    {
+      "grad_norm": 0.22303034365177155,
+      "learning_rate": 7.631780408642769e-05,
+      "loss": 0.01,
+      "step": 17870
+    },
+    {
+      "grad_norm": 0.13877159357070923,
+      "learning_rate": 7.628968060721946e-05,
+      "loss": 0.0095,
+      "step": 17880
+    },
+    {
+      "grad_norm": 0.17621733248233795,
+      "learning_rate": 7.626154562801176e-05,
+      "loss": 0.0089,
+      "step": 17890
+    },
+    {
+      "grad_norm": 0.1659928262233734,
+      "learning_rate": 7.623339916111177e-05,
+      "loss": 0.0093,
+      "step": 17900
+    },
+    {
+      "grad_norm": 0.13569113612174988,
+      "learning_rate": 7.620524121883174e-05,
+      "loss": 0.009,
+      "step": 17910
+    },
+    {
+      "grad_norm": 0.20344896614551544,
+      "learning_rate": 7.617707181348892e-05,
+      "loss": 0.0104,
+      "step": 17920
+    },
+    {
+      "grad_norm": 0.22364017367362976,
+      "learning_rate": 7.614889095740554e-05,
+      "loss": 0.0095,
+      "step": 17930
+    },
+    {
+      "grad_norm": 0.17400340735912323,
+      "learning_rate": 7.612069866290885e-05,
+      "loss": 0.0091,
+      "step": 17940
+    },
+    {
+      "grad_norm": 0.17388799786567688,
+      "learning_rate": 7.609249494233115e-05,
+      "loss": 0.0089,
+      "step": 17950
+    },
+    {
+      "grad_norm": 0.13706806302070618,
+      "learning_rate": 7.60642798080097e-05,
+      "loss": 0.0084,
+      "step": 17960
+    },
+    {
+      "grad_norm": 0.20318521559238434,
+      "learning_rate": 7.603605327228673e-05,
+      "loss": 0.0105,
+      "step": 17970
+    },
+    {
+      "grad_norm": 0.21220310032367706,
+      "learning_rate": 7.600781534750953e-05,
+      "loss": 0.0081,
+      "step": 17980
+    },
+    {
+      "grad_norm": 0.19932200014591217,
+      "learning_rate": 7.597956604603029e-05,
+      "loss": 0.0082,
+      "step": 17990
+    },
+    {
+      "grad_norm": 0.1688593178987503,
+      "learning_rate": 7.595130538020622e-05,
+      "loss": 0.0089,
+      "step": 18000
+    },
+    {
+      "grad_norm": 0.19705483317375183,
+      "learning_rate": 7.592303336239952e-05,
+      "loss": 0.0095,
+      "step": 18010
+    },
+    {
+      "grad_norm": 0.15600675344467163,
+      "learning_rate": 7.589475000497733e-05,
+      "loss": 0.0088,
+      "step": 18020
+    },
+    {
+      "grad_norm": 0.16848938167095184,
+      "learning_rate": 7.586645532031171e-05,
+      "loss": 0.0095,
+      "step": 18030
+    },
+    {
+      "grad_norm": 0.18032389879226685,
+      "learning_rate": 7.583814932077975e-05,
+      "loss": 0.0098,
+      "step": 18040
+    },
+    {
+      "grad_norm": 0.14692123234272003,
+      "learning_rate": 7.580983201876343e-05,
+      "loss": 0.0102,
+      "step": 18050
+    },
+    {
+      "grad_norm": 0.18763332068920135,
+      "learning_rate": 7.578150342664973e-05,
+      "loss": 0.008,
+      "step": 18060
+    },
+    {
+      "grad_norm": 0.27320897579193115,
+      "learning_rate": 7.575316355683055e-05,
+      "loss": 0.0097,
+      "step": 18070
+    },
+    {
+      "grad_norm": 0.29713210463523865,
+      "learning_rate": 7.572481242170266e-05,
+      "loss": 0.0091,
+      "step": 18080
+    },
+    {
+      "grad_norm": 0.2296714186668396,
+      "learning_rate": 7.569645003366786e-05,
+      "loss": 0.0103,
+      "step": 18090
+    },
+    {
+      "grad_norm": 0.16609564423561096,
+      "learning_rate": 7.566807640513278e-05,
+      "loss": 0.0082,
+      "step": 18100
+    },
+    {
+      "grad_norm": 0.2364702671766281,
+      "learning_rate": 7.563969154850905e-05,
+      "loss": 0.0094,
+      "step": 18110
+    },
+    {
+      "grad_norm": 0.19467990100383759,
+      "learning_rate": 7.561129547621314e-05,
+      "loss": 0.0086,
+      "step": 18120
+    },
+    {
+      "grad_norm": 0.15774765610694885,
+      "learning_rate": 7.558288820066648e-05,
+      "loss": 0.008,
+      "step": 18130
+    },
+    {
+      "grad_norm": 0.28333306312561035,
+      "learning_rate": 7.555446973429534e-05,
+      "loss": 0.0118,
+      "step": 18140
+    },
+    {
+      "grad_norm": 0.23875311017036438,
+      "learning_rate": 7.552604008953097e-05,
+      "loss": 0.0078,
+      "step": 18150
+    },
+    {
+      "grad_norm": 0.18431846797466278,
+      "learning_rate": 7.549759927880942e-05,
+      "loss": 0.0086,
+      "step": 18160
+    },
+    {
+      "grad_norm": 0.2181759625673294,
+      "learning_rate": 7.54691473145717e-05,
+      "loss": 0.0121,
+      "step": 18170
+    },
+    {
+      "grad_norm": 0.21860943734645844,
+      "learning_rate": 7.544068420926365e-05,
+      "loss": 0.0095,
+      "step": 18180
+    },
+    {
+      "grad_norm": 0.18036718666553497,
+      "learning_rate": 7.5412209975336e-05,
+      "loss": 0.0094,
+      "step": 18190
+    },
+    {
+      "grad_norm": 0.16099198162555695,
+      "learning_rate": 7.538372462524435e-05,
+      "loss": 0.0081,
+      "step": 18200
+    },
+    {
+      "grad_norm": 0.1523062288761139,
+      "learning_rate": 7.535522817144915e-05,
+      "loss": 0.0086,
+      "step": 18210
+    },
+    {
+      "grad_norm": 0.2006613314151764,
+      "learning_rate": 7.532672062641575e-05,
+      "loss": 0.0109,
+      "step": 18220
+    },
+    {
+      "grad_norm": 0.15750207006931305,
+      "learning_rate": 7.529820200261427e-05,
+      "loss": 0.0098,
+      "step": 18230
+    },
+    {
+      "grad_norm": 0.23123487830162048,
+      "learning_rate": 7.526967231251977e-05,
+      "loss": 0.01,
+      "step": 18240
+    },
+    {
+      "grad_norm": 0.22517843544483185,
+      "learning_rate": 7.524113156861206e-05,
+      "loss": 0.0101,
+      "step": 18250
+    },
+    {
+      "grad_norm": 0.20890910923480988,
+      "learning_rate": 7.521257978337587e-05,
+      "loss": 0.01,
+      "step": 18260
+    },
+    {
+      "grad_norm": 0.1938062310218811,
+      "learning_rate": 7.518401696930072e-05,
+      "loss": 0.0104,
+      "step": 18270
+    },
+    {
+      "grad_norm": 0.17722143232822418,
+      "learning_rate": 7.515544313888092e-05,
+      "loss": 0.0102,
+      "step": 18280
+    },
+    {
+      "grad_norm": 0.21166454255580902,
+      "learning_rate": 7.512685830461568e-05,
+      "loss": 0.0095,
+      "step": 18290
+    },
+    {
+      "grad_norm": 0.20936179161071777,
+      "learning_rate": 7.509826247900892e-05,
+      "loss": 0.0098,
+      "step": 18300
+    },
+    {
+      "grad_norm": 0.17647689580917358,
+      "learning_rate": 7.50696556745695e-05,
+      "loss": 0.0103,
+      "step": 18310
+    },
+    {
+      "grad_norm": 0.19695976376533508,
+      "learning_rate": 7.504103790381095e-05,
+      "loss": 0.009,
+      "step": 18320
+    },
+    {
+      "grad_norm": 0.25444430112838745,
+      "learning_rate": 7.501240917925167e-05,
+      "loss": 0.0118,
+      "step": 18330
+    },
+    {
+      "grad_norm": 0.23156842589378357,
+      "learning_rate": 7.498376951341483e-05,
+      "loss": 0.0101,
+      "step": 18340
+    },
+    {
+      "grad_norm": 0.2137606143951416,
+      "learning_rate": 7.495511891882844e-05,
+      "loss": 0.0088,
+      "step": 18350
+    },
+    {
+      "grad_norm": 0.22060750424861908,
+      "learning_rate": 7.492645740802517e-05,
+      "loss": 0.0094,
+      "step": 18360
+    },
+    {
+      "grad_norm": 0.2914900779724121,
+      "learning_rate": 7.489778499354259e-05,
+      "loss": 0.0101,
+      "step": 18370
+    },
+    {
+      "grad_norm": 0.2450907677412033,
+      "learning_rate": 7.486910168792298e-05,
+      "loss": 0.0114,
+      "step": 18380
+    },
+    {
+      "grad_norm": 0.21861331164836884,
+      "learning_rate": 7.484040750371337e-05,
+      "loss": 0.0099,
+      "step": 18390
+    },
+    {
+      "grad_norm": 0.16906283795833588,
+      "learning_rate": 7.48117024534656e-05,
+      "loss": 0.0098,
+      "step": 18400
+    },
+    {
+      "grad_norm": 0.21408022940158844,
+      "learning_rate": 7.478298654973621e-05,
+      "loss": 0.0121,
+      "step": 18410
+    },
+    {
+      "grad_norm": 0.25162598490715027,
+      "learning_rate": 7.475425980508654e-05,
+      "loss": 0.0092,
+      "step": 18420
+    },
+    {
+      "grad_norm": 0.22749198973178864,
+      "learning_rate": 7.47255222320826e-05,
+      "loss": 0.0117,
+      "step": 18430
+    },
+    {
+      "grad_norm": 0.24719804525375366,
+      "learning_rate": 7.469677384329522e-05,
+      "loss": 0.0082,
+      "step": 18440
+    },
+    {
+      "grad_norm": 0.1902557611465454,
+      "learning_rate": 7.46680146512999e-05,
+      "loss": 0.0096,
+      "step": 18450
+    },
+    {
+      "grad_norm": 0.19236113131046295,
+      "learning_rate": 7.463924466867692e-05,
+      "loss": 0.0095,
+      "step": 18460
+    },
+    {
+      "grad_norm": 0.23020486533641815,
+      "learning_rate": 7.461046390801121e-05,
+      "loss": 0.0088,
+      "step": 18470
+    },
+    {
+      "grad_norm": 0.22496554255485535,
+      "learning_rate": 7.458167238189248e-05,
+      "loss": 0.0086,
+      "step": 18480
+    },
+    {
+      "grad_norm": 0.19402681291103363,
+      "learning_rate": 7.455287010291509e-05,
+      "loss": 0.0087,
+      "step": 18490
+    },
+    {
+      "grad_norm": 0.17703750729560852,
+      "learning_rate": 7.452405708367818e-05,
+      "loss": 0.0087,
+      "step": 18500
+    },
+    {
+      "grad_norm": 0.19068461656570435,
+      "learning_rate": 7.449523333678549e-05,
+      "loss": 0.0101,
+      "step": 18510
+    },
+    {
+      "grad_norm": 0.2292453795671463,
+      "learning_rate": 7.446639887484555e-05,
+      "loss": 0.0107,
+      "step": 18520
+    },
+    {
+      "grad_norm": 0.15654070675373077,
+      "learning_rate": 7.443755371047152e-05,
+      "loss": 0.0081,
+      "step": 18530
+    },
+    {
+      "grad_norm": 0.18637987971305847,
+      "learning_rate": 7.440869785628124e-05,
+      "loss": 0.0104,
+      "step": 18540
+    },
+    {
+      "grad_norm": 0.23593656718730927,
+      "learning_rate": 7.437983132489727e-05,
+      "loss": 0.0086,
+      "step": 18550
+    },
+    {
+      "grad_norm": 0.19026848673820496,
+      "learning_rate": 7.435095412894678e-05,
+      "loss": 0.0087,
+      "step": 18560
+    },
+    {
+      "grad_norm": 0.19923445582389832,
+      "learning_rate": 7.432206628106165e-05,
+      "loss": 0.0113,
+      "step": 18570
+    },
+    {
+      "grad_norm": 0.19138678908348083,
+      "learning_rate": 7.429316779387842e-05,
+      "loss": 0.0078,
+      "step": 18580
+    },
+    {
+      "grad_norm": 0.17612694203853607,
+      "learning_rate": 7.426425868003825e-05,
+      "loss": 0.0096,
+      "step": 18590
+    },
+    {
+      "grad_norm": 0.2178489863872528,
+      "learning_rate": 7.423533895218698e-05,
+      "loss": 0.0091,
+      "step": 18600
+    },
+    {
+      "grad_norm": 0.1792500615119934,
+      "learning_rate": 7.420640862297507e-05,
+      "loss": 0.0098,
+      "step": 18610
+    },
+    {
+      "grad_norm": 0.1661064624786377,
+      "learning_rate": 7.417746770505766e-05,
+      "loss": 0.008,
+      "step": 18620
+    },
+    {
+      "grad_norm": 0.16117529571056366,
+      "learning_rate": 7.414851621109444e-05,
+      "loss": 0.0084,
+      "step": 18630
+    },
+    {
+      "grad_norm": 0.17144891619682312,
+      "learning_rate": 7.411955415374982e-05,
+      "loss": 0.0086,
+      "step": 18640
+    },
+    {
+      "grad_norm": 0.22562438249588013,
+      "learning_rate": 7.409058154569278e-05,
+      "loss": 0.0072,
+      "step": 18650
+    },
+    {
+      "grad_norm": 0.14842039346694946,
+      "learning_rate": 7.406159839959689e-05,
+      "loss": 0.008,
+      "step": 18660
+    },
+    {
+      "grad_norm": 0.18330974876880646,
+      "learning_rate": 7.403260472814039e-05,
+      "loss": 0.0064,
+      "step": 18670
+    },
+    {
+      "grad_norm": 0.24091827869415283,
+      "learning_rate": 7.400360054400611e-05,
+      "loss": 0.0092,
+      "step": 18680
+    },
+    {
+      "grad_norm": 0.17302671074867249,
+      "learning_rate": 7.397458585988143e-05,
+      "loss": 0.0097,
+      "step": 18690
+    },
+    {
+      "grad_norm": 0.19687789678573608,
+      "learning_rate": 7.394556068845841e-05,
+      "loss": 0.0089,
+      "step": 18700
+    },
+    {
+      "grad_norm": 0.18158932030200958,
+      "learning_rate": 7.39165250424336e-05,
+      "loss": 0.0081,
+      "step": 18710
+    },
+    {
+      "grad_norm": 0.21895216405391693,
+      "learning_rate": 7.38874789345082e-05,
+      "loss": 0.0088,
+      "step": 18720
+    },
+    {
+      "grad_norm": 0.1984478086233139,
+      "learning_rate": 7.385842237738794e-05,
+      "loss": 0.0073,
+      "step": 18730
+    },
+    {
+      "grad_norm": 0.18452002108097076,
+      "learning_rate": 7.382935538378318e-05,
+      "loss": 0.0104,
+      "step": 18740
+    },
+    {
+      "grad_norm": 0.2282298356294632,
+      "learning_rate": 7.38002779664088e-05,
+      "loss": 0.0093,
+      "step": 18750
+    },
+    {
+      "grad_norm": 0.19490157067775726,
+      "learning_rate": 7.377119013798422e-05,
+      "loss": 0.0077,
+      "step": 18760
+    },
+    {
+      "grad_norm": 0.1713787466287613,
+      "learning_rate": 7.374209191123349e-05,
+      "loss": 0.0088,
+      "step": 18770
+    },
+    {
+      "grad_norm": 0.18283410370349884,
+      "learning_rate": 7.371298329888512e-05,
+      "loss": 0.0091,
+      "step": 18780
+    },
+    {
+      "grad_norm": 0.1485232710838318,
+      "learning_rate": 7.368386431367223e-05,
+      "loss": 0.0089,
+      "step": 18790
+    },
+    {
+      "grad_norm": 0.1702098548412323,
+      "learning_rate": 7.365473496833246e-05,
+      "loss": 0.0098,
+      "step": 18800
+    },
+    {
+      "grad_norm": 0.16187717020511627,
+      "learning_rate": 7.362559527560797e-05,
+      "loss": 0.0096,
+      "step": 18810
+    },
+    {
+      "grad_norm": 0.20418283343315125,
+      "learning_rate": 7.359644524824545e-05,
+      "loss": 0.009,
+      "step": 18820
+    },
+    {
+      "grad_norm": 0.2295207381248474,
+      "learning_rate": 7.35672848989961e-05,
+      "loss": 0.0088,
+      "step": 18830
+    },
+    {
+      "grad_norm": 0.18402773141860962,
+      "learning_rate": 7.353811424061565e-05,
+      "loss": 0.0097,
+      "step": 18840
+    },
+    {
+      "grad_norm": 0.20955674350261688,
+      "learning_rate": 7.350893328586435e-05,
+      "loss": 0.0103,
+      "step": 18850
+    },
+    {
+      "grad_norm": 0.1717154085636139,
+      "learning_rate": 7.347974204750696e-05,
+      "loss": 0.0086,
+      "step": 18860
+    },
+    {
+      "grad_norm": 0.21817705035209656,
+      "learning_rate": 7.345054053831266e-05,
+      "loss": 0.008,
+      "step": 18870
+    },
+    {
+      "grad_norm": 0.15987132489681244,
+      "learning_rate": 7.342132877105523e-05,
+      "loss": 0.0084,
+      "step": 18880
+    },
+    {
+      "grad_norm": 0.18847990036010742,
+      "learning_rate": 7.339210675851288e-05,
+      "loss": 0.0083,
+      "step": 18890
+    },
+    {
+      "grad_norm": 0.13651633262634277,
+      "learning_rate": 7.336287451346831e-05,
+      "loss": 0.008,
+      "step": 18900
+    },
+    {
+      "grad_norm": 0.22881603240966797,
+      "learning_rate": 7.333363204870868e-05,
+      "loss": 0.0081,
+      "step": 18910
+    },
+    {
+      "grad_norm": 0.2114374190568924,
+      "learning_rate": 7.330437937702566e-05,
+      "loss": 0.0106,
+      "step": 18920
+    },
+    {
+      "grad_norm": 0.21556219458580017,
+      "learning_rate": 7.327511651121534e-05,
+      "loss": 0.0087,
+      "step": 18930
+    },
+    {
+      "grad_norm": 0.1949467658996582,
+      "learning_rate": 7.324584346407833e-05,
+      "loss": 0.0072,
+      "step": 18940
+    },
+    {
+      "grad_norm": 0.19549560546875,
+      "learning_rate": 7.321656024841962e-05,
+      "loss": 0.0101,
+      "step": 18950
+    },
+    {
+      "grad_norm": 0.1671113520860672,
+      "learning_rate": 7.31872668770487e-05,
+      "loss": 0.0081,
+      "step": 18960
+    },
+    {
+      "grad_norm": 0.21298089623451233,
+      "learning_rate": 7.315796336277949e-05,
+      "loss": 0.0106,
+      "step": 18970
+    },
+    {
+      "grad_norm": 0.1920555979013443,
+      "learning_rate": 7.312864971843031e-05,
+      "loss": 0.0098,
+      "step": 18980
+    },
+    {
+      "grad_norm": 0.21354404091835022,
+      "learning_rate": 7.309932595682398e-05,
+      "loss": 0.0099,
+      "step": 18990
+    },
+    {
+      "grad_norm": 0.1886931210756302,
+      "learning_rate": 7.30699920907877e-05,
+      "loss": 0.0082,
+      "step": 19000
+    },
+    {
+      "grad_norm": 0.20513315498828888,
+      "learning_rate": 7.304064813315308e-05,
+      "loss": 0.0074,
+      "step": 19010
+    },
+    {
+      "grad_norm": 0.24543631076812744,
+      "learning_rate": 7.30112940967562e-05,
+      "loss": 0.0077,
+      "step": 19020
+    },
+    {
+      "grad_norm": 0.23267577588558197,
+      "learning_rate": 7.298192999443747e-05,
+      "loss": 0.0099,
+      "step": 19030
+    },
+    {
+      "grad_norm": 0.17945744097232819,
+      "learning_rate": 7.295255583904179e-05,
+      "loss": 0.0106,
+      "step": 19040
+    },
+    {
+      "grad_norm": 0.2371988296508789,
+      "learning_rate": 7.29231716434184e-05,
+      "loss": 0.0094,
+      "step": 19050
+    },
+    {
+      "grad_norm": 0.1694527417421341,
+      "learning_rate": 7.289377742042092e-05,
+      "loss": 0.012,
+      "step": 19060
+    },
+    {
+      "grad_norm": 0.17784927785396576,
+      "learning_rate": 7.286437318290741e-05,
+      "loss": 0.0096,
+      "step": 19070
+    },
+    {
+      "grad_norm": 0.23809543251991272,
+      "learning_rate": 7.283495894374027e-05,
+      "loss": 0.0102,
+      "step": 19080
+    },
+    {
+      "grad_norm": 0.1874508410692215,
+      "learning_rate": 7.280553471578629e-05,
+      "loss": 0.0094,
+      "step": 19090
+    },
+    {
+      "grad_norm": 0.27973076701164246,
+      "learning_rate": 7.277610051191663e-05,
+      "loss": 0.0108,
+      "step": 19100
+    },
+    {
+      "grad_norm": 0.16957123577594757,
+      "learning_rate": 7.27466563450068e-05,
+      "loss": 0.0124,
+      "step": 19110
+    },
+    {
+      "grad_norm": 0.23982161283493042,
+      "learning_rate": 7.271720222793668e-05,
+      "loss": 0.0105,
+      "step": 19120
+    },
+    {
+      "grad_norm": 0.21413855254650116,
+      "learning_rate": 7.268773817359054e-05,
+      "loss": 0.0103,
+      "step": 19130
+    },
+    {
+      "grad_norm": 0.22344987094402313,
+      "learning_rate": 7.265826419485689e-05,
+      "loss": 0.0101,
+      "step": 19140
+    },
+    {
+      "grad_norm": 0.16102223098278046,
+      "learning_rate": 7.26287803046287e-05,
+      "loss": 0.0098,
+      "step": 19150
+    },
+    {
+      "grad_norm": 0.19559969007968903,
+      "learning_rate": 7.259928651580322e-05,
+      "loss": 0.0085,
+      "step": 19160
+    },
+    {
+      "grad_norm": 0.1873466521501541,
+      "learning_rate": 7.256978284128202e-05,
+      "loss": 0.0073,
+      "step": 19170
+    },
+    {
+      "grad_norm": 0.22023162245750427,
+      "learning_rate": 7.254026929397101e-05,
+      "loss": 0.0092,
+      "step": 19180
+    },
+    {
+      "grad_norm": 0.20659440755844116,
+      "learning_rate": 7.251074588678044e-05,
+      "loss": 0.01,
+      "step": 19190
+    },
+    {
+      "grad_norm": 0.2216113805770874,
+      "learning_rate": 7.248121263262482e-05,
+      "loss": 0.0107,
+      "step": 19200
+    },
+    {
+      "grad_norm": 0.15481005609035492,
+      "learning_rate": 7.245166954442304e-05,
+      "loss": 0.0084,
+      "step": 19210
+    },
+    {
+      "grad_norm": 0.18321958184242249,
+      "learning_rate": 7.24221166350982e-05,
+      "loss": 0.009,
+      "step": 19220
+    },
+    {
+      "grad_norm": 0.18242613971233368,
+      "learning_rate": 7.239255391757777e-05,
+      "loss": 0.0083,
+      "step": 19230
+    },
+    {
+      "grad_norm": 0.20943383872509003,
+      "learning_rate": 7.236298140479351e-05,
+      "loss": 0.009,
+      "step": 19240
+    },
+    {
+      "grad_norm": 0.23244524002075195,
+      "learning_rate": 7.233339910968141e-05,
+      "loss": 0.011,
+      "step": 19250
+    },
+    {
+      "grad_norm": 0.2505984902381897,
+      "learning_rate": 7.230380704518178e-05,
+      "loss": 0.0097,
+      "step": 19260
+    },
+    {
+      "grad_norm": 0.19085128605365753,
+      "learning_rate": 7.227420522423919e-05,
+      "loss": 0.0073,
+      "step": 19270
+    },
+    {
+      "grad_norm": 0.26885250210762024,
+      "learning_rate": 7.224459365980249e-05,
+      "loss": 0.0112,
+      "step": 19280
+    },
+    {
+      "grad_norm": 0.2730563282966614,
+      "learning_rate": 7.221497236482479e-05,
+      "loss": 0.0087,
+      "step": 19290
+    },
+    {
+      "grad_norm": 0.20856527984142303,
+      "learning_rate": 7.218534135226345e-05,
+      "loss": 0.01,
+      "step": 19300
+    },
+    {
+      "grad_norm": 0.17688988149166107,
+      "learning_rate": 7.215570063508005e-05,
+      "loss": 0.0094,
+      "step": 19310
+    },
+    {
+      "grad_norm": 0.33428528904914856,
+      "learning_rate": 7.21260502262405e-05,
+      "loss": 0.0114,
+      "step": 19320
+    },
+    {
+      "grad_norm": 0.21561039984226227,
+      "learning_rate": 7.209639013871487e-05,
+      "loss": 0.0088,
+      "step": 19330
+    },
+    {
+      "grad_norm": 0.16564978659152985,
+      "learning_rate": 7.206672038547746e-05,
+      "loss": 0.0092,
+      "step": 19340
+    },
+    {
+      "grad_norm": 0.18800126016139984,
+      "learning_rate": 7.203704097950687e-05,
+      "loss": 0.0083,
+      "step": 19350
+    },
+    {
+      "grad_norm": 0.18122081458568573,
+      "learning_rate": 7.200735193378587e-05,
+      "loss": 0.0089,
+      "step": 19360
+    },
+    {
+      "grad_norm": 0.1787225753068924,
+      "learning_rate": 7.197765326130145e-05,
+      "loss": 0.0122,
+      "step": 19370
+    },
+    {
+      "grad_norm": 0.18536554276943207,
+      "learning_rate": 7.194794497504481e-05,
+      "loss": 0.0089,
+      "step": 19380
+    },
+    {
+      "grad_norm": 0.16178591549396515,
+      "learning_rate": 7.19182270880114e-05,
+      "loss": 0.0109,
+      "step": 19390
+    },
+    {
+      "grad_norm": 0.15338575839996338,
+      "learning_rate": 7.188849961320079e-05,
+      "loss": 0.0094,
+      "step": 19400
+    },
+    {
+      "grad_norm": 0.26180270314216614,
+      "learning_rate": 7.185876256361681e-05,
+      "loss": 0.0106,
+      "step": 19410
+    },
+    {
+      "grad_norm": 0.15976957976818085,
+      "learning_rate": 7.182901595226745e-05,
+      "loss": 0.0091,
+      "step": 19420
+    },
+    {
+      "grad_norm": 0.20363107323646545,
+      "learning_rate": 7.179925979216491e-05,
+      "loss": 0.0085,
+      "step": 19430
+    },
+    {
+      "grad_norm": 0.22463840246200562,
+      "learning_rate": 7.17694940963255e-05,
+      "loss": 0.0107,
+      "step": 19440
+    },
+    {
+      "grad_norm": 0.1879836767911911,
+      "learning_rate": 7.17397188777698e-05,
+      "loss": 0.007,
+      "step": 19450
+    },
+    {
+      "grad_norm": 0.2521689832210541,
+      "learning_rate": 7.170993414952245e-05,
+      "loss": 0.0082,
+      "step": 19460
+    },
+    {
+      "grad_norm": 0.22171242535114288,
+      "learning_rate": 7.168013992461233e-05,
+      "loss": 0.008,
+      "step": 19470
+    },
+    {
+      "grad_norm": 0.18471956253051758,
+      "learning_rate": 7.165033621607245e-05,
+      "loss": 0.0076,
+      "step": 19480
+    },
+    {
+      "grad_norm": 0.21105968952178955,
+      "learning_rate": 7.162052303693995e-05,
+      "loss": 0.0073,
+      "step": 19490
+    },
+    {
+      "grad_norm": 0.20363396406173706,
+      "learning_rate": 7.159070040025614e-05,
+      "loss": 0.0072,
+      "step": 19500
+    },
+    {
+      "grad_norm": 0.22160229086875916,
+      "learning_rate": 7.156086831906646e-05,
+      "loss": 0.0109,
+      "step": 19510
+    },
+    {
+      "grad_norm": 0.2196601778268814,
+      "learning_rate": 7.153102680642044e-05,
+      "loss": 0.0085,
+      "step": 19520
+    },
+    {
+      "grad_norm": 0.2489532232284546,
+      "learning_rate": 7.150117587537182e-05,
+      "loss": 0.0096,
+      "step": 19530
+    },
+    {
+      "grad_norm": 0.23267734050750732,
+      "learning_rate": 7.147131553897838e-05,
+      "loss": 0.0072,
+      "step": 19540
+    },
+    {
+      "grad_norm": 0.21101802587509155,
+      "learning_rate": 7.144144581030205e-05,
+      "loss": 0.0087,
+      "step": 19550
+    },
+    {
+      "grad_norm": 0.1772487461566925,
+      "learning_rate": 7.141156670240889e-05,
+      "loss": 0.0098,
+      "step": 19560
+    },
+    {
+      "grad_norm": 0.18808630108833313,
+      "learning_rate": 7.138167822836899e-05,
+      "loss": 0.01,
+      "step": 19570
+    },
+    {
+      "grad_norm": 0.18067283928394318,
+      "learning_rate": 7.135178040125661e-05,
+      "loss": 0.0084,
+      "step": 19580
+    },
+    {
+      "grad_norm": 0.2179444581270218,
+      "learning_rate": 7.13218732341501e-05,
+      "loss": 0.0074,
+      "step": 19590
+    },
+    {
+      "grad_norm": 0.22896398603916168,
+      "learning_rate": 7.129195674013181e-05,
+      "loss": 0.0087,
+      "step": 19600
+    },
+    {
+      "grad_norm": 0.1923496574163437,
+      "learning_rate": 7.126203093228829e-05,
+      "loss": 0.0091,
+      "step": 19610
+    },
+    {
+      "grad_norm": 0.20971621572971344,
+      "learning_rate": 7.123209582371006e-05,
+      "loss": 0.0079,
+      "step": 19620
+    },
+    {
+      "grad_norm": 0.17289583384990692,
+      "learning_rate": 7.120215142749179e-05,
+      "loss": 0.0095,
+      "step": 19630
+    },
+    {
+      "grad_norm": 0.1917956918478012,
+      "learning_rate": 7.117219775673218e-05,
+      "loss": 0.0068,
+      "step": 19640
+    },
+    {
+      "grad_norm": 0.1956562101840973,
+      "learning_rate": 7.114223482453394e-05,
+      "loss": 0.0095,
+      "step": 19650
+    },
+    {
+      "grad_norm": 0.15418881177902222,
+      "learning_rate": 7.11122626440039e-05,
+      "loss": 0.0107,
+      "step": 19660
+    },
+    {
+      "grad_norm": 0.18911288678646088,
+      "learning_rate": 7.10822812282529e-05,
+      "loss": 0.0106,
+      "step": 19670
+    },
+    {
+      "grad_norm": 0.12856753170490265,
+      "learning_rate": 7.105229059039586e-05,
+      "loss": 0.0078,
+      "step": 19680
+    },
+    {
+      "grad_norm": 0.14657682180404663,
+      "learning_rate": 7.102229074355168e-05,
+      "loss": 0.0079,
+      "step": 19690
+    },
+    {
+      "grad_norm": 0.23357370495796204,
+      "learning_rate": 7.099228170084332e-05,
+      "loss": 0.0078,
+      "step": 19700
+    },
+    {
+      "grad_norm": 0.16776712238788605,
+      "learning_rate": 7.096226347539771e-05,
+      "loss": 0.0075,
+      "step": 19710
+    },
+    {
+      "grad_norm": 0.2352246642112732,
+      "learning_rate": 7.093223608034592e-05,
+      "loss": 0.0081,
+      "step": 19720
+    },
+    {
+      "grad_norm": 0.18510738015174866,
+      "learning_rate": 7.090219952882291e-05,
+      "loss": 0.0084,
+      "step": 19730
+    },
+    {
+      "grad_norm": 0.16822762787342072,
+      "learning_rate": 7.087215383396768e-05,
+      "loss": 0.0086,
+      "step": 19740
+    },
+    {
+      "grad_norm": 0.17882409691810608,
+      "learning_rate": 7.084209900892325e-05,
+      "loss": 0.009,
+      "step": 19750
+    },
+    {
+      "grad_norm": 0.14525890350341797,
+      "learning_rate": 7.081203506683663e-05,
+      "loss": 0.0114,
+      "step": 19760
+    },
+    {
+      "grad_norm": 0.17319807410240173,
+      "learning_rate": 7.078196202085879e-05,
+      "loss": 0.0087,
+      "step": 19770
+    },
+    {
+      "grad_norm": 0.17516838014125824,
+      "learning_rate": 7.075187988414473e-05,
+      "loss": 0.0079,
+      "step": 19780
+    },
+    {
+      "grad_norm": 0.18354351818561554,
+      "learning_rate": 7.072178866985334e-05,
+      "loss": 0.0104,
+      "step": 19790
+    },
+    {
+      "grad_norm": 0.20608901977539062,
+      "learning_rate": 7.069168839114764e-05,
+      "loss": 0.01,
+      "step": 19800
+    },
+    {
+      "grad_norm": 0.1840846836566925,
+      "learning_rate": 7.066157906119441e-05,
+      "loss": 0.0102,
+      "step": 19810
+    },
+    {
+      "grad_norm": 0.17110197246074677,
+      "learning_rate": 7.063146069316456e-05,
+      "loss": 0.0082,
+      "step": 19820
+    },
+    {
+      "grad_norm": 0.15817180275917053,
+      "learning_rate": 7.060133330023287e-05,
+      "loss": 0.0099,
+      "step": 19830
+    },
+    {
+      "grad_norm": 0.15849851071834564,
+      "learning_rate": 7.057119689557807e-05,
+      "loss": 0.009,
+      "step": 19840
+    },
+    {
+      "grad_norm": 0.1933971494436264,
+      "learning_rate": 7.054105149238287e-05,
+      "loss": 0.0083,
+      "step": 19850
+    },
+    {
+      "grad_norm": 0.24208998680114746,
+      "learning_rate": 7.051089710383387e-05,
+      "loss": 0.0088,
+      "step": 19860
+    },
+    {
+      "grad_norm": 0.1932469606399536,
+      "learning_rate": 7.048073374312166e-05,
+      "loss": 0.0093,
+      "step": 19870
+    },
+    {
+      "grad_norm": 0.17818807065486908,
+      "learning_rate": 7.045056142344067e-05,
+      "loss": 0.0089,
+      "step": 19880
+    },
+    {
+      "grad_norm": 0.16237163543701172,
+      "learning_rate": 7.042038015798934e-05,
+      "loss": 0.0089,
+      "step": 19890
+    },
+    {
+      "grad_norm": 0.17797110974788666,
+      "learning_rate": 7.039018995996995e-05,
+      "loss": 0.0083,
+      "step": 19900
+    },
+    {
+      "grad_norm": 0.1501874476671219,
+      "learning_rate": 7.035999084258876e-05,
+      "loss": 0.0104,
+      "step": 19910
+    },
+    {
+      "grad_norm": 0.18486055731773376,
+      "learning_rate": 7.032978281905583e-05,
+      "loss": 0.0097,
+      "step": 19920
+    },
+    {
+      "grad_norm": 0.17435358464717865,
+      "learning_rate": 7.029956590258522e-05,
+      "loss": 0.0121,
+      "step": 19930
+    },
+    {
+      "grad_norm": 0.14628104865550995,
+      "learning_rate": 7.026934010639483e-05,
+      "loss": 0.0081,
+      "step": 19940
+    },
+    {
+      "grad_norm": 0.14955681562423706,
+      "learning_rate": 7.023910544370645e-05,
+      "loss": 0.0082,
+      "step": 19950
+    },
+    {
+      "grad_norm": 0.19796057045459747,
+      "learning_rate": 7.020886192774573e-05,
+      "loss": 0.0117,
+      "step": 19960
+    },
+    {
+      "grad_norm": 0.19740599393844604,
+      "learning_rate": 7.017860957174222e-05,
+      "loss": 0.008,
+      "step": 19970
+    },
+    {
+      "grad_norm": 0.1770530492067337,
+      "learning_rate": 7.014834838892934e-05,
+      "loss": 0.0087,
+      "step": 19980
+    },
+    {
+      "grad_norm": 0.2630986273288727,
+      "learning_rate": 7.011807839254435e-05,
+      "loss": 0.008,
+      "step": 19990
+    },
+    {
+      "grad_norm": 0.11868008971214294,
+      "learning_rate": 7.008779959582837e-05,
+      "loss": 0.0093,
+      "step": 20000
+    },
+    {
+      "grad_norm": 0.15368297696113586,
+      "learning_rate": 7.005751201202639e-05,
+      "loss": 0.0072,
+      "step": 20010
+    },
+    {
+      "grad_norm": 0.1543615311384201,
+      "learning_rate": 7.002721565438724e-05,
+      "loss": 0.0076,
+      "step": 20020
+    },
+    {
+      "grad_norm": 0.16618846356868744,
+      "learning_rate": 6.999691053616353e-05,
+      "loss": 0.0085,
+      "step": 20030
+    },
+    {
+      "grad_norm": 0.16898736357688904,
+      "learning_rate": 6.996659667061178e-05,
+      "loss": 0.0086,
+      "step": 20040
+    },
+    {
+      "grad_norm": 0.19759555160999298,
+      "learning_rate": 6.993627407099229e-05,
+      "loss": 0.0075,
+      "step": 20050
+    },
+    {
+      "grad_norm": 0.16140277683734894,
+      "learning_rate": 6.990594275056921e-05,
+      "loss": 0.0075,
+      "step": 20060
+    },
+    {
+      "grad_norm": 0.21803422272205353,
+      "learning_rate": 6.987560272261049e-05,
+      "loss": 0.0088,
+      "step": 20070
+    },
+    {
+      "grad_norm": 0.2184513360261917,
+      "learning_rate": 6.984525400038788e-05,
+      "loss": 0.0085,
+      "step": 20080
+    },
+    {
+      "grad_norm": 0.14639346301555634,
+      "learning_rate": 6.981489659717693e-05,
+      "loss": 0.0079,
+      "step": 20090
+    },
+    {
+      "grad_norm": 0.14114396274089813,
+      "learning_rate": 6.978453052625703e-05,
+      "loss": 0.0079,
+      "step": 20100
+    },
+    {
+      "grad_norm": 0.17181788384914398,
+      "learning_rate": 6.975415580091131e-05,
+      "loss": 0.0096,
+      "step": 20110
+    },
+    {
+      "grad_norm": 0.20748434960842133,
+      "learning_rate": 6.972377243442672e-05,
+      "loss": 0.008,
+      "step": 20120
+    },
+    {
+      "grad_norm": 0.22041316330432892,
+      "learning_rate": 6.969338044009396e-05,
+      "loss": 0.0094,
+      "step": 20130
+    },
+    {
+      "grad_norm": 0.20815472304821014,
+      "learning_rate": 6.966297983120753e-05,
+      "loss": 0.0094,
+      "step": 20140
+    },
+    {
+      "grad_norm": 0.200233593583107,
+      "learning_rate": 6.963257062106571e-05,
+      "loss": 0.0092,
+      "step": 20150
+    },
+    {
+      "grad_norm": 0.21200434863567352,
+      "learning_rate": 6.960215282297048e-05,
+      "loss": 0.0099,
+      "step": 20160
+    },
+    {
+      "grad_norm": 0.16643591225147247,
+      "learning_rate": 6.957172645022765e-05,
+      "loss": 0.0101,
+      "step": 20170
+    },
+    {
+      "grad_norm": 0.17127715051174164,
+      "learning_rate": 6.954129151614673e-05,
+      "loss": 0.0087,
+      "step": 20180
+    },
+    {
+      "grad_norm": 0.19519075751304626,
+      "learning_rate": 6.9510848034041e-05,
+      "loss": 0.0087,
+      "step": 20190
+    },
+    {
+      "grad_norm": 0.1941581815481186,
+      "learning_rate": 6.94803960172275e-05,
+      "loss": 0.0084,
+      "step": 20200
+    },
+    {
+      "grad_norm": 0.2202366590499878,
+      "learning_rate": 6.944993547902692e-05,
+      "loss": 0.0102,
+      "step": 20210
+    },
+    {
+      "grad_norm": 0.2276659607887268,
+      "learning_rate": 6.941946643276379e-05,
+      "loss": 0.0098,
+      "step": 20220
+    },
+    {
+      "grad_norm": 0.2717202603816986,
+      "learning_rate": 6.938898889176626e-05,
+      "loss": 0.0085,
+      "step": 20230
+    },
+    {
+      "grad_norm": 0.22065070271492004,
+      "learning_rate": 6.935850286936627e-05,
+      "loss": 0.0092,
+      "step": 20240
+    },
+    {
+      "grad_norm": 0.1922864466905594,
+      "learning_rate": 6.932800837889943e-05,
+      "loss": 0.0086,
+      "step": 20250
+    },
+    {
+      "grad_norm": 0.2155875712633133,
+      "learning_rate": 6.92975054337051e-05,
+      "loss": 0.01,
+      "step": 20260
+    },
+    {
+      "grad_norm": 0.14461876451969147,
+      "learning_rate": 6.926699404712627e-05,
+      "loss": 0.0088,
+      "step": 20270
+    },
+    {
+      "grad_norm": 0.1875162273645401,
+      "learning_rate": 6.923647423250967e-05,
+      "loss": 0.0086,
+      "step": 20280
+    },
+    {
+      "grad_norm": 0.15187287330627441,
+      "learning_rate": 6.920594600320567e-05,
+      "loss": 0.0083,
+      "step": 20290
+    },
+    {
+      "grad_norm": 0.1684095859527588,
+      "learning_rate": 6.91754093725684e-05,
+      "loss": 0.0101,
+      "step": 20300
+    },
+    {
+      "grad_norm": 0.19519184529781342,
+      "learning_rate": 6.914486435395561e-05,
+      "loss": 0.0088,
+      "step": 20310
+    },
+    {
+      "grad_norm": 0.14055907726287842,
+      "learning_rate": 6.911431096072871e-05,
+      "loss": 0.0067,
+      "step": 20320
+    },
+    {
+      "grad_norm": 0.23026931285858154,
+      "learning_rate": 6.90837492062528e-05,
+      "loss": 0.0076,
+      "step": 20330
+    },
+    {
+      "grad_norm": 0.1730712503194809,
+      "learning_rate": 6.905317910389664e-05,
+      "loss": 0.0081,
+      "step": 20340
+    },
+    {
+      "grad_norm": 0.15802161395549774,
+      "learning_rate": 6.902260066703262e-05,
+      "loss": 0.0075,
+      "step": 20350
+    },
+    {
+      "grad_norm": 0.2252558320760727,
+      "learning_rate": 6.89920139090368e-05,
+      "loss": 0.0093,
+      "step": 20360
+    },
+    {
+      "grad_norm": 0.20474527776241302,
+      "learning_rate": 6.896141884328886e-05,
+      "loss": 0.0097,
+      "step": 20370
+    },
+    {
+      "grad_norm": 0.19520209729671478,
+      "learning_rate": 6.893081548317211e-05,
+      "loss": 0.0114,
+      "step": 20380
+    },
+    {
+      "grad_norm": 0.22648867964744568,
+      "learning_rate": 6.890020384207353e-05,
+      "loss": 0.008,
+      "step": 20390
+    },
+    {
+      "grad_norm": 0.17430368065834045,
+      "learning_rate": 6.886958393338366e-05,
+      "loss": 0.0092,
+      "step": 20400
+    },
+    {
+      "grad_norm": 0.1453242301940918,
+      "learning_rate": 6.883895577049668e-05,
+      "loss": 0.0082,
+      "step": 20410
+    },
+    {
+      "grad_norm": 0.18319907784461975,
+      "learning_rate": 6.880831936681045e-05,
+      "loss": 0.0078,
+      "step": 20420
+    },
+    {
+      "grad_norm": 0.16413946449756622,
+      "learning_rate": 6.87776747357263e-05,
+      "loss": 0.0076,
+      "step": 20430
+    },
+    {
+      "grad_norm": 0.1620844006538391,
+      "learning_rate": 6.874702189064927e-05,
+      "loss": 0.009,
+      "step": 20440
+    },
+    {
+      "grad_norm": 0.20575520396232605,
+      "learning_rate": 6.871636084498796e-05,
+      "loss": 0.009,
+      "step": 20450
+    },
+    {
+      "grad_norm": 0.11809684336185455,
+      "learning_rate": 6.868569161215453e-05,
+      "loss": 0.0081,
+      "step": 20460
+    },
+    {
+      "grad_norm": 0.18768206238746643,
+      "learning_rate": 6.865501420556477e-05,
+      "loss": 0.0087,
+      "step": 20470
+    },
+    {
+      "grad_norm": 0.1674424409866333,
+      "learning_rate": 6.862432863863802e-05,
+      "loss": 0.0106,
+      "step": 20480
+    },
+    {
+      "grad_norm": 0.14596623182296753,
+      "learning_rate": 6.859363492479716e-05,
+      "loss": 0.0077,
+      "step": 20490
+    },
+    {
+      "grad_norm": 0.1980106383562088,
+      "learning_rate": 6.856293307746868e-05,
+      "loss": 0.0069,
+      "step": 20500
+    },
+    {
+      "grad_norm": 0.20554815232753754,
+      "learning_rate": 6.853222311008263e-05,
+      "loss": 0.0076,
+      "step": 20510
+    },
+    {
+      "grad_norm": 0.16388605535030365,
+      "learning_rate": 6.850150503607256e-05,
+      "loss": 0.0068,
+      "step": 20520
+    },
+    {
+      "grad_norm": 0.20418515801429749,
+      "learning_rate": 6.847077886887562e-05,
+      "loss": 0.0084,
+      "step": 20530
+    },
+    {
+      "grad_norm": 0.2008107453584671,
+      "learning_rate": 6.844004462193248e-05,
+      "loss": 0.0091,
+      "step": 20540
+    },
+    {
+      "grad_norm": 0.186850443482399,
+      "learning_rate": 6.840930230868734e-05,
+      "loss": 0.0075,
+      "step": 20550
+    },
+    {
+      "grad_norm": 0.1821938008069992,
+      "learning_rate": 6.837855194258791e-05,
+      "loss": 0.0084,
+      "step": 20560
+    },
+    {
+      "grad_norm": 0.20323234796524048,
+      "learning_rate": 6.834779353708548e-05,
+      "loss": 0.0084,
+      "step": 20570
+    },
+    {
+      "grad_norm": 0.22351416945457458,
+      "learning_rate": 6.831702710563478e-05,
+      "loss": 0.0095,
+      "step": 20580
+    },
+    {
+      "grad_norm": 0.20560984313488007,
+      "learning_rate": 6.828625266169416e-05,
+      "loss": 0.0089,
+      "step": 20590
+    },
+    {
+      "grad_norm": 0.24732643365859985,
+      "learning_rate": 6.825547021872534e-05,
+      "loss": 0.0088,
+      "step": 20600
+    },
+    {
+      "grad_norm": 0.15750546753406525,
+      "learning_rate": 6.822467979019365e-05,
+      "loss": 0.0092,
+      "step": 20610
+    },
+    {
+      "grad_norm": 0.20463517308235168,
+      "learning_rate": 6.819388138956783e-05,
+      "loss": 0.0072,
+      "step": 20620
+    },
+    {
+      "grad_norm": 0.19608420133590698,
+      "learning_rate": 6.816307503032019e-05,
+      "loss": 0.0094,
+      "step": 20630
+    },
+    {
+      "grad_norm": 0.19831207394599915,
+      "learning_rate": 6.813226072592645e-05,
+      "loss": 0.0077,
+      "step": 20640
+    },
+    {
+      "grad_norm": 0.19512102007865906,
+      "learning_rate": 6.810143848986583e-05,
+      "loss": 0.0074,
+      "step": 20650
+    },
+    {
+      "grad_norm": 0.16293589770793915,
+      "learning_rate": 6.807060833562106e-05,
+      "loss": 0.0079,
+      "step": 20660
+    },
+    {
+      "grad_norm": 0.1599176824092865,
+      "learning_rate": 6.803977027667825e-05,
+      "loss": 0.0074,
+      "step": 20670
+    },
+    {
+      "grad_norm": 0.16561561822891235,
+      "learning_rate": 6.800892432652704e-05,
+      "loss": 0.0085,
+      "step": 20680
+    },
+    {
+      "grad_norm": 0.21010246872901917,
+      "learning_rate": 6.797807049866049e-05,
+      "loss": 0.0092,
+      "step": 20690
+    },
+    {
+      "grad_norm": 0.1441507637500763,
+      "learning_rate": 6.794720880657512e-05,
+      "loss": 0.0093,
+      "step": 20700
+    },
+    {
+      "grad_norm": 0.14083261787891388,
+      "learning_rate": 6.791633926377085e-05,
+      "loss": 0.0098,
+      "step": 20710
+    },
+    {
+      "grad_norm": 0.193366140127182,
+      "learning_rate": 6.788546188375114e-05,
+      "loss": 0.011,
+      "step": 20720
+    },
+    {
+      "grad_norm": 0.27323678135871887,
+      "learning_rate": 6.785457668002274e-05,
+      "loss": 0.0088,
+      "step": 20730
+    },
+    {
+      "grad_norm": 0.19671867787837982,
+      "learning_rate": 6.78236836660959e-05,
+      "loss": 0.0079,
+      "step": 20740
+    },
+    {
+      "grad_norm": 0.1765950471162796,
+      "learning_rate": 6.779278285548427e-05,
+      "loss": 0.0083,
+      "step": 20750
+    },
+    {
+      "grad_norm": 0.17430013418197632,
+      "learning_rate": 6.776187426170493e-05,
+      "loss": 0.0072,
+      "step": 20760
+    },
+    {
+      "grad_norm": 0.17995867133140564,
+      "learning_rate": 6.773095789827837e-05,
+      "loss": 0.0088,
+      "step": 20770
+    },
+    {
+      "grad_norm": 0.14751912653446198,
+      "learning_rate": 6.770003377872841e-05,
+      "loss": 0.0086,
+      "step": 20780
+    },
+    {
+      "grad_norm": 0.13990157842636108,
+      "learning_rate": 6.766910191658233e-05,
+      "loss": 0.0071,
+      "step": 20790
+    },
+    {
+      "grad_norm": 0.13160467147827148,
+      "learning_rate": 6.763816232537079e-05,
+      "loss": 0.0074,
+      "step": 20800
+    },
+    {
+      "grad_norm": 0.1913018673658371,
+      "learning_rate": 6.76072150186278e-05,
+      "loss": 0.0084,
+      "step": 20810
+    },
+    {
+      "grad_norm": 0.22480560839176178,
+      "learning_rate": 6.757626000989077e-05,
+      "loss": 0.0079,
+      "step": 20820
+    },
+    {
+      "grad_norm": 0.17047463357448578,
+      "learning_rate": 6.754529731270048e-05,
+      "loss": 0.0092,
+      "step": 20830
+    },
+    {
+      "grad_norm": 0.1392960101366043,
+      "learning_rate": 6.751432694060105e-05,
+      "loss": 0.0077,
+      "step": 20840
+    },
+    {
+      "grad_norm": 0.17773088812828064,
+      "learning_rate": 6.748334890713999e-05,
+      "loss": 0.0089,
+      "step": 20850
+    },
+    {
+      "grad_norm": 0.22224949300289154,
+      "learning_rate": 6.745236322586813e-05,
+      "loss": 0.0085,
+      "step": 20860
+    },
+    {
+      "grad_norm": 0.1734849363565445,
+      "learning_rate": 6.742136991033967e-05,
+      "loss": 0.009,
+      "step": 20870
+    },
+    {
+      "grad_norm": 0.17989642918109894,
+      "learning_rate": 6.739036897411215e-05,
+      "loss": 0.007,
+      "step": 20880
+    },
+    {
+      "grad_norm": 0.1841350942850113,
+      "learning_rate": 6.73593604307464e-05,
+      "loss": 0.0082,
+      "step": 20890
+    },
+    {
+      "grad_norm": 0.2612465023994446,
+      "learning_rate": 6.732834429380663e-05,
+      "loss": 0.0082,
+      "step": 20900
+    },
+    {
+      "grad_norm": 0.21569222211837769,
+      "learning_rate": 6.729732057686036e-05,
+      "loss": 0.0086,
+      "step": 20910
+    },
+    {
+      "grad_norm": 0.1664339005947113,
+      "learning_rate": 6.726628929347839e-05,
+      "loss": 0.0091,
+      "step": 20920
+    },
+    {
+      "grad_norm": 0.21332502365112305,
+      "learning_rate": 6.723525045723488e-05,
+      "loss": 0.0081,
+      "step": 20930
+    },
+    {
+      "grad_norm": 0.16544444859027863,
+      "learning_rate": 6.720420408170728e-05,
+      "loss": 0.0094,
+      "step": 20940
+    },
+    {
+      "grad_norm": 0.16560447216033936,
+      "learning_rate": 6.71731501804763e-05,
+      "loss": 0.009,
+      "step": 20950
+    },
+    {
+      "grad_norm": 0.2706261873245239,
+      "learning_rate": 6.714208876712602e-05,
+      "loss": 0.0082,
+      "step": 20960
+    },
+    {
+      "grad_norm": 0.29477423429489136,
+      "learning_rate": 6.71110198552437e-05,
+      "loss": 0.0088,
+      "step": 20970
+    },
+    {
+      "grad_norm": 0.16278548538684845,
+      "learning_rate": 6.707994345841996e-05,
+      "loss": 0.0088,
+      "step": 20980
+    },
+    {
+      "grad_norm": 0.20247964560985565,
+      "learning_rate": 6.70488595902487e-05,
+      "loss": 0.009,
+      "step": 20990
+    },
+    {
+      "grad_norm": 0.24162603914737701,
+      "learning_rate": 6.7017768264327e-05,
+      "loss": 0.0077,
+      "step": 21000
+    },
+    {
+      "grad_norm": 0.2265254259109497,
+      "learning_rate": 6.698666949425532e-05,
+      "loss": 0.0086,
+      "step": 21010
+    },
+    {
+      "grad_norm": 0.20495611429214478,
+      "learning_rate": 6.69555632936373e-05,
+      "loss": 0.0083,
+      "step": 21020
+    },
+    {
+      "grad_norm": 0.12093034386634827,
+      "learning_rate": 6.692444967607983e-05,
+      "loss": 0.0075,
+      "step": 21030
+    },
+    {
+      "grad_norm": 0.1555117815732956,
+      "learning_rate": 6.689332865519309e-05,
+      "loss": 0.0093,
+      "step": 21040
+    },
+    {
+      "grad_norm": 0.17745444178581238,
+      "learning_rate": 6.686220024459045e-05,
+      "loss": 0.0071,
+      "step": 21050
+    },
+    {
+      "grad_norm": 0.14771749079227448,
+      "learning_rate": 6.683106445788856e-05,
+      "loss": 0.0073,
+      "step": 21060
+    },
+    {
+      "grad_norm": 0.15259036421775818,
+      "learning_rate": 6.679992130870723e-05,
+      "loss": 0.0066,
+      "step": 21070
+    },
+    {
+      "grad_norm": 0.17564910650253296,
+      "learning_rate": 6.676877081066957e-05,
+      "loss": 0.0083,
+      "step": 21080
+    },
+    {
+      "grad_norm": 0.20000912249088287,
+      "learning_rate": 6.673761297740186e-05,
+      "loss": 0.0077,
+      "step": 21090
+    },
+    {
+      "grad_norm": 0.2559061050415039,
+      "learning_rate": 6.670644782253358e-05,
+      "loss": 0.0086,
+      "step": 21100
+    },
+    {
+      "grad_norm": 0.15884077548980713,
+      "learning_rate": 6.667527535969744e-05,
+      "loss": 0.0073,
+      "step": 21110
+    },
+    {
+      "grad_norm": 0.1619189977645874,
+      "learning_rate": 6.664409560252933e-05,
+      "loss": 0.0084,
+      "step": 21120
+    },
+    {
+      "grad_norm": 0.2263210564851761,
+      "learning_rate": 6.661290856466833e-05,
+      "loss": 0.0097,
+      "step": 21130
+    },
+    {
+      "grad_norm": 0.16018706560134888,
+      "learning_rate": 6.658171425975672e-05,
+      "loss": 0.0078,
+      "step": 21140
+    },
+    {
+      "grad_norm": 0.17209242284297943,
+      "learning_rate": 6.655051270143994e-05,
+      "loss": 0.0087,
+      "step": 21150
+    },
+    {
+      "grad_norm": 0.17304588854312897,
+      "learning_rate": 6.651930390336662e-05,
+      "loss": 0.0083,
+      "step": 21160
+    },
+    {
+      "grad_norm": 0.22782668471336365,
+      "learning_rate": 6.648808787918854e-05,
+      "loss": 0.0082,
+      "step": 21170
+    },
+    {
+      "grad_norm": 0.16768617928028107,
+      "learning_rate": 6.645686464256065e-05,
+      "loss": 0.0076,
+      "step": 21180
+    },
+    {
+      "grad_norm": 0.20528170466423035,
+      "learning_rate": 6.642563420714106e-05,
+      "loss": 0.0083,
+      "step": 21190
+    },
+    {
+      "grad_norm": 0.15487273037433624,
+      "learning_rate": 6.639439658659102e-05,
+      "loss": 0.0076,
+      "step": 21200
+    },
+    {
+      "grad_norm": 0.13553407788276672,
+      "learning_rate": 6.636315179457492e-05,
+      "loss": 0.008,
+      "step": 21210
+    },
+    {
+      "grad_norm": 0.15661659836769104,
+      "learning_rate": 6.633189984476031e-05,
+      "loss": 0.0074,
+      "step": 21220
+    },
+    {
+      "grad_norm": 0.16726450622081757,
+      "learning_rate": 6.630064075081785e-05,
+      "loss": 0.0087,
+      "step": 21230
+    },
+    {
+      "grad_norm": 0.16675405204296112,
+      "learning_rate": 6.626937452642132e-05,
+      "loss": 0.0089,
+      "step": 21240
+    },
+    {
+      "grad_norm": 0.21543926000595093,
+      "learning_rate": 6.623810118524765e-05,
+      "loss": 0.0072,
+      "step": 21250
+    },
+    {
+      "grad_norm": 0.22170165181159973,
+      "learning_rate": 6.620682074097682e-05,
+      "loss": 0.0068,
+      "step": 21260
+    },
+    {
+      "grad_norm": 0.1545175164937973,
+      "learning_rate": 6.6175533207292e-05,
+      "loss": 0.0069,
+      "step": 21270
+    },
+    {
+      "grad_norm": 0.20090757310390472,
+      "learning_rate": 6.61442385978794e-05,
+      "loss": 0.0072,
+      "step": 21280
+    },
+    {
+      "grad_norm": 0.17319464683532715,
+      "learning_rate": 6.611293692642836e-05,
+      "loss": 0.0076,
+      "step": 21290
+    },
+    {
+      "grad_norm": 0.13765788078308105,
+      "learning_rate": 6.608162820663128e-05,
+      "loss": 0.0076,
+      "step": 21300
+    },
+    {
+      "grad_norm": 0.197941854596138,
+      "learning_rate": 6.605031245218365e-05,
+      "loss": 0.0102,
+      "step": 21310
+    },
+    {
+      "grad_norm": 0.14281494915485382,
+      "learning_rate": 6.601898967678408e-05,
+      "loss": 0.0078,
+      "step": 21320
+    },
+    {
+      "grad_norm": 0.1631798893213272,
+      "learning_rate": 6.598765989413419e-05,
+      "loss": 0.0085,
+      "step": 21330
+    },
+    {
+      "grad_norm": 0.26238471269607544,
+      "learning_rate": 6.595632311793867e-05,
+      "loss": 0.008,
+      "step": 21340
+    },
+    {
+      "grad_norm": 0.2500929832458496,
+      "learning_rate": 6.592497936190534e-05,
+      "loss": 0.0096,
+      "step": 21350
+    },
+    {
+      "grad_norm": 0.24582616984844208,
+      "learning_rate": 6.5893628639745e-05,
+      "loss": 0.0086,
+      "step": 21360
+    },
+    {
+      "grad_norm": 0.16432541608810425,
+      "learning_rate": 6.586227096517151e-05,
+      "loss": 0.0074,
+      "step": 21370
+    },
+    {
+      "grad_norm": 0.1399301290512085,
+      "learning_rate": 6.58309063519018e-05,
+      "loss": 0.0074,
+      "step": 21380
+    },
+    {
+      "grad_norm": 0.12356159836053848,
+      "learning_rate": 6.579953481365582e-05,
+      "loss": 0.007,
+      "step": 21390
+    },
+    {
+      "grad_norm": 0.173086479306221,
+      "learning_rate": 6.576815636415652e-05,
+      "loss": 0.0086,
+      "step": 21400
+    },
+    {
+      "grad_norm": 0.18782921135425568,
+      "learning_rate": 6.573677101712991e-05,
+      "loss": 0.0081,
+      "step": 21410
+    },
+    {
+      "grad_norm": 0.19510743021965027,
+      "learning_rate": 6.570537878630502e-05,
+      "loss": 0.0082,
+      "step": 21420
+    },
+    {
+      "grad_norm": 0.1880255937576294,
+      "learning_rate": 6.56739796854139e-05,
+      "loss": 0.008,
+      "step": 21430
+    },
+    {
+      "grad_norm": 0.21270625293254852,
+      "learning_rate": 6.564257372819153e-05,
+      "loss": 0.0076,
+      "step": 21440
+    },
+    {
+      "grad_norm": 0.15757794678211212,
+      "learning_rate": 6.561116092837596e-05,
+      "loss": 0.0071,
+      "step": 21450
+    },
+    {
+      "grad_norm": 0.20739102363586426,
+      "learning_rate": 6.557974129970825e-05,
+      "loss": 0.0082,
+      "step": 21460
+    },
+    {
+      "grad_norm": 0.16153584420681,
+      "learning_rate": 6.554831485593239e-05,
+      "loss": 0.0091,
+      "step": 21470
+    },
+    {
+      "grad_norm": 0.1779981553554535,
+      "learning_rate": 6.551688161079534e-05,
+      "loss": 0.0067,
+      "step": 21480
+    },
+    {
+      "grad_norm": 0.16173596680164337,
+      "learning_rate": 6.548544157804712e-05,
+      "loss": 0.0073,
+      "step": 21490
+    },
+    {
+      "grad_norm": 0.14316408336162567,
+      "learning_rate": 6.545399477144065e-05,
+      "loss": 0.007,
+      "step": 21500
+    },
+    {
+      "grad_norm": 0.20533978939056396,
+      "learning_rate": 6.542254120473184e-05,
+      "loss": 0.0076,
+      "step": 21510
+    },
+    {
+      "grad_norm": 0.19837647676467896,
+      "learning_rate": 6.539108089167953e-05,
+      "loss": 0.0092,
+      "step": 21520
+    },
+    {
+      "grad_norm": 0.14996473491191864,
+      "learning_rate": 6.535961384604554e-05,
+      "loss": 0.0073,
+      "step": 21530
+    },
+    {
+      "grad_norm": 0.1711873561143875,
+      "learning_rate": 6.532814008159461e-05,
+      "loss": 0.007,
+      "step": 21540
+    },
+    {
+      "grad_norm": 0.13709023594856262,
+      "learning_rate": 6.529665961209446e-05,
+      "loss": 0.008,
+      "step": 21550
+    },
+    {
+      "grad_norm": 0.1889423429965973,
+      "learning_rate": 6.52651724513157e-05,
+      "loss": 0.0081,
+      "step": 21560
+    },
+    {
+      "grad_norm": 0.16440258920192719,
+      "learning_rate": 6.523367861303189e-05,
+      "loss": 0.0079,
+      "step": 21570
+    },
+    {
+      "grad_norm": 0.18061894178390503,
+      "learning_rate": 6.520217811101951e-05,
+      "loss": 0.0084,
+      "step": 21580
+    },
+    {
+      "grad_norm": 0.18335594236850739,
+      "learning_rate": 6.517067095905793e-05,
+      "loss": 0.0085,
+      "step": 21590
+    },
+    {
+      "grad_norm": 0.18441559374332428,
+      "learning_rate": 6.513915717092948e-05,
+      "loss": 0.0084,
+      "step": 21600
+    },
+    {
+      "grad_norm": 0.2098967581987381,
+      "learning_rate": 6.510763676041935e-05,
+      "loss": 0.0104,
+      "step": 21610
+    },
+    {
+      "grad_norm": 0.13140545785427094,
+      "learning_rate": 6.50761097413156e-05,
+      "loss": 0.0088,
+      "step": 21620
+    },
+    {
+      "grad_norm": 0.20761102437973022,
+      "learning_rate": 6.504457612740928e-05,
+      "loss": 0.0091,
+      "step": 21630
+    },
+    {
+      "grad_norm": 0.21739500761032104,
+      "learning_rate": 6.501303593249423e-05,
+      "loss": 0.008,
+      "step": 21640
+    },
+    {
+      "grad_norm": 0.19806040823459625,
+      "learning_rate": 6.49814891703672e-05,
+      "loss": 0.0082,
+      "step": 21650
+    },
+    {
+      "grad_norm": 0.17266607284545898,
+      "learning_rate": 6.494993585482783e-05,
+      "loss": 0.0079,
+      "step": 21660
+    },
+    {
+      "grad_norm": 0.22815746068954468,
+      "learning_rate": 6.49183759996786e-05,
+      "loss": 0.0077,
+      "step": 21670
+    },
+    {
+      "grad_norm": 0.16621984541416168,
+      "learning_rate": 6.488680961872486e-05,
+      "loss": 0.0075,
+      "step": 21680
+    },
+    {
+      "grad_norm": 0.13100789487361908,
+      "learning_rate": 6.485523672577485e-05,
+      "loss": 0.0064,
+      "step": 21690
+    },
+    {
+      "grad_norm": 0.16127553582191467,
+      "learning_rate": 6.482365733463959e-05,
+      "loss": 0.0091,
+      "step": 21700
+    },
+    {
+      "grad_norm": 0.2073376625776291,
+      "learning_rate": 6.4792071459133e-05,
+      "loss": 0.0071,
+      "step": 21710
+    },
+    {
+      "grad_norm": 0.17863644659519196,
+      "learning_rate": 6.476047911307179e-05,
+      "loss": 0.0093,
+      "step": 21720
+    },
+    {
+      "grad_norm": 0.2460300326347351,
+      "learning_rate": 6.472888031027556e-05,
+      "loss": 0.0076,
+      "step": 21730
+    },
+    {
+      "grad_norm": 0.1525702029466629,
+      "learning_rate": 6.469727506456666e-05,
+      "loss": 0.0075,
+      "step": 21740
+    },
+    {
+      "grad_norm": 0.19396649301052094,
+      "learning_rate": 6.466566338977034e-05,
+      "loss": 0.0093,
+      "step": 21750
+    },
+    {
+      "grad_norm": 0.1811201125383377,
+      "learning_rate": 6.46340452997146e-05,
+      "loss": 0.0089,
+      "step": 21760
+    },
+    {
+      "grad_norm": 0.13575956225395203,
+      "learning_rate": 6.460242080823025e-05,
+      "loss": 0.0072,
+      "step": 21770
+    },
+    {
+      "grad_norm": 0.19311784207820892,
+      "learning_rate": 6.457078992915093e-05,
+      "loss": 0.0093,
+      "step": 21780
+    },
+    {
+      "grad_norm": 0.13242587447166443,
+      "learning_rate": 6.453915267631308e-05,
+      "loss": 0.0063,
+      "step": 21790
+    },
+    {
+      "grad_norm": 0.17359067499637604,
+      "learning_rate": 6.450750906355589e-05,
+      "loss": 0.0076,
+      "step": 21800
+    },
+    {
+      "grad_norm": 0.16984082758426666,
+      "learning_rate": 6.447585910472134e-05,
+      "loss": 0.0081,
+      "step": 21810
+    },
+    {
+      "grad_norm": 0.2051703780889511,
+      "learning_rate": 6.444420281365424e-05,
+      "loss": 0.0077,
+      "step": 21820
+    },
+    {
+      "grad_norm": 0.2328711897134781,
+      "learning_rate": 6.441254020420208e-05,
+      "loss": 0.0082,
+      "step": 21830
+    },
+    {
+      "grad_norm": 0.2496817409992218,
+      "learning_rate": 6.438087129021522e-05,
+      "loss": 0.0075,
+      "step": 21840
+    },
+    {
+      "grad_norm": 0.1866501420736313,
+      "learning_rate": 6.434919608554663e-05,
+      "loss": 0.0097,
+      "step": 21850
+    },
+    {
+      "grad_norm": 0.20890018343925476,
+      "learning_rate": 6.43175146040522e-05,
+      "loss": 0.0079,
+      "step": 21860
+    },
+    {
+      "grad_norm": 0.18984006345272064,
+      "learning_rate": 6.428582685959045e-05,
+      "loss": 0.0092,
+      "step": 21870
+    },
+    {
+      "grad_norm": 0.15755140781402588,
+      "learning_rate": 6.42541328660227e-05,
+      "loss": 0.0079,
+      "step": 21880
+    },
+    {
+      "grad_norm": 0.1998271495103836,
+      "learning_rate": 6.422243263721293e-05,
+      "loss": 0.0079,
+      "step": 21890
+    },
+    {
+      "grad_norm": 0.12858223915100098,
+      "learning_rate": 6.419072618702794e-05,
+      "loss": 0.0068,
+      "step": 21900
+    },
+    {
+      "grad_norm": 0.16867828369140625,
+      "learning_rate": 6.41590135293372e-05,
+      "loss": 0.0076,
+      "step": 21910
+    },
+    {
+      "grad_norm": 0.2631497383117676,
+      "learning_rate": 6.41272946780129e-05,
+      "loss": 0.0072,
+      "step": 21920
+    },
+    {
+      "grad_norm": 0.17161472141742706,
+      "learning_rate": 6.409556964692996e-05,
+      "loss": 0.0097,
+      "step": 21930
+    },
+    {
+      "grad_norm": 0.21658587455749512,
+      "learning_rate": 6.406383844996595e-05,
+      "loss": 0.0083,
+      "step": 21940
+    },
+    {
+      "grad_norm": 0.17420735955238342,
+      "learning_rate": 6.403210110100121e-05,
+      "loss": 0.0079,
+      "step": 21950
+    },
+    {
+      "grad_norm": 0.23190806806087494,
+      "learning_rate": 6.400035761391871e-05,
+      "loss": 0.007,
+      "step": 21960
+    },
+    {
+      "grad_norm": 0.14282362163066864,
+      "learning_rate": 6.396860800260416e-05,
+      "loss": 0.0086,
+      "step": 21970
+    },
+    {
+      "grad_norm": 0.20140042901039124,
+      "learning_rate": 6.393685228094588e-05,
+      "loss": 0.0081,
+      "step": 21980
+    },
+    {
+      "grad_norm": 0.1742752641439438,
+      "learning_rate": 6.390509046283493e-05,
+      "loss": 0.0083,
+      "step": 21990
+    },
+    {
+      "grad_norm": 0.13578283786773682,
+      "learning_rate": 6.387332256216498e-05,
+      "loss": 0.0086,
+      "step": 22000
+    },
+    {
+      "grad_norm": 0.1931917518377304,
+      "learning_rate": 6.384154859283243e-05,
+      "loss": 0.0072,
+      "step": 22010
+    },
+    {
+      "grad_norm": 0.21150480210781097,
+      "learning_rate": 6.380976856873627e-05,
+      "loss": 0.008,
+      "step": 22020
+    },
+    {
+      "grad_norm": 0.19361571967601776,
+      "learning_rate": 6.377798250377815e-05,
+      "loss": 0.0078,
+      "step": 22030
+    },
+    {
+      "grad_norm": 0.23764784634113312,
+      "learning_rate": 6.374619041186238e-05,
+      "loss": 0.0073,
+      "step": 22040
+    },
+    {
+      "grad_norm": 0.16223658621311188,
+      "learning_rate": 6.371439230689592e-05,
+      "loss": 0.0075,
+      "step": 22050
+    },
+    {
+      "grad_norm": 0.12752285599708557,
+      "learning_rate": 6.368258820278833e-05,
+      "loss": 0.0061,
+      "step": 22060
+    },
+    {
+      "grad_norm": 0.17574229836463928,
+      "learning_rate": 6.365077811345182e-05,
+      "loss": 0.0074,
+      "step": 22070
+    },
+    {
+      "grad_norm": 0.1768060326576233,
+      "learning_rate": 6.361896205280117e-05,
+      "loss": 0.0082,
+      "step": 22080
+    },
+    {
+      "grad_norm": 0.1692928969860077,
+      "learning_rate": 6.358714003475384e-05,
+      "loss": 0.0078,
+      "step": 22090
+    },
+    {
+      "grad_norm": 0.17516246438026428,
+      "learning_rate": 6.355531207322983e-05,
+      "loss": 0.0072,
+      "step": 22100
+    },
+    {
+      "grad_norm": 0.17466239631175995,
+      "learning_rate": 6.35234781821518e-05,
+      "loss": 0.0076,
+      "step": 22110
+    },
+    {
+      "grad_norm": 0.12647207081317902,
+      "learning_rate": 6.349163837544497e-05,
+      "loss": 0.0061,
+      "step": 22120
+    },
+    {
+      "grad_norm": 0.12662583589553833,
+      "learning_rate": 6.345979266703714e-05,
+      "loss": 0.0085,
+      "step": 22130
+    },
+    {
+      "grad_norm": 0.20686693489551544,
+      "learning_rate": 6.342794107085872e-05,
+      "loss": 0.0081,
+      "step": 22140
+    },
+    {
+      "grad_norm": 0.26733195781707764,
+      "learning_rate": 6.339608360084265e-05,
+      "loss": 0.0075,
+      "step": 22150
+    },
+    {
+      "grad_norm": 0.17465122044086456,
+      "learning_rate": 6.336422027092452e-05,
+      "loss": 0.0093,
+      "step": 22160
+    },
+    {
+      "grad_norm": 0.1662677377462387,
+      "learning_rate": 6.333235109504241e-05,
+      "loss": 0.0078,
+      "step": 22170
+    },
+    {
+      "grad_norm": 0.17957432568073273,
+      "learning_rate": 6.330047608713695e-05,
+      "loss": 0.0072,
+      "step": 22180
+    },
+    {
+      "grad_norm": 0.11625088006258011,
+      "learning_rate": 6.32685952611514e-05,
+      "loss": 0.0065,
+      "step": 22190
+    },
+    {
+      "grad_norm": 0.18465179204940796,
+      "learning_rate": 6.323670863103149e-05,
+      "loss": 0.0079,
+      "step": 22200
+    },
+    {
+      "grad_norm": 0.230351522564888,
+      "learning_rate": 6.32048162107255e-05,
+      "loss": 0.0094,
+      "step": 22210
+    },
+    {
+      "grad_norm": 0.24016420543193817,
+      "learning_rate": 6.317291801418431e-05,
+      "loss": 0.0078,
+      "step": 22220
+    },
+    {
+      "grad_norm": 0.1330106556415558,
+      "learning_rate": 6.31410140553612e-05,
+      "loss": 0.0069,
+      "step": 22230
+    },
+    {
+      "grad_norm": 0.191716268658638,
+      "learning_rate": 6.310910434821211e-05,
+      "loss": 0.0069,
+      "step": 22240
+    },
+    {
+      "grad_norm": 0.21580512821674347,
+      "learning_rate": 6.307718890669541e-05,
+      "loss": 0.0071,
+      "step": 22250
+    },
+    {
+      "grad_norm": 0.139646515250206,
+      "learning_rate": 6.304526774477198e-05,
+      "loss": 0.0062,
+      "step": 22260
+    },
+    {
+      "grad_norm": 0.14491006731987,
+      "learning_rate": 6.301334087640526e-05,
+      "loss": 0.0072,
+      "step": 22270
+    },
+    {
+      "grad_norm": 0.169067844748497,
+      "learning_rate": 6.298140831556112e-05,
+      "loss": 0.0082,
+      "step": 22280
+    },
+    {
+      "grad_norm": 0.15833471715450287,
+      "learning_rate": 6.294947007620794e-05,
+      "loss": 0.0069,
+      "step": 22290
+    },
+    {
+      "grad_norm": 0.19135919213294983,
+      "learning_rate": 6.291752617231661e-05,
+      "loss": 0.0099,
+      "step": 22300
+    },
+    {
+      "grad_norm": 0.23401159048080444,
+      "learning_rate": 6.288557661786047e-05,
+      "loss": 0.0083,
+      "step": 22310
+    },
+    {
+      "grad_norm": 0.16646526753902435,
+      "learning_rate": 6.285362142681534e-05,
+      "loss": 0.0082,
+      "step": 22320
+    },
+    {
+      "grad_norm": 0.25629332661628723,
+      "learning_rate": 6.28216606131595e-05,
+      "loss": 0.0094,
+      "step": 22330
+    },
+    {
+      "grad_norm": 0.2123817354440689,
+      "learning_rate": 6.27896941908737e-05,
+      "loss": 0.0081,
+      "step": 22340
+    },
+    {
+      "grad_norm": 0.18276949226856232,
+      "learning_rate": 6.275772217394115e-05,
+      "loss": 0.0095,
+      "step": 22350
+    },
+    {
+      "grad_norm": 0.20569971203804016,
+      "learning_rate": 6.27257445763475e-05,
+      "loss": 0.0064,
+      "step": 22360
+    },
+    {
+      "grad_norm": 0.21384049952030182,
+      "learning_rate": 6.269376141208082e-05,
+      "loss": 0.0082,
+      "step": 22370
+    },
+    {
+      "grad_norm": 0.17653244733810425,
+      "learning_rate": 6.266177269513162e-05,
+      "loss": 0.0072,
+      "step": 22380
+    },
+    {
+      "grad_norm": 0.22334878146648407,
+      "learning_rate": 6.262977843949289e-05,
+      "loss": 0.0091,
+      "step": 22390
+    },
+    {
+      "grad_norm": 0.17125572264194489,
+      "learning_rate": 6.259777865915997e-05,
+      "loss": 0.0084,
+      "step": 22400
+    },
+    {
+      "grad_norm": 0.21589328348636627,
+      "learning_rate": 6.256577336813069e-05,
+      "loss": 0.0091,
+      "step": 22410
+    },
+    {
+      "grad_norm": 0.1783471554517746,
+      "learning_rate": 6.253376258040521e-05,
+      "loss": 0.0089,
+      "step": 22420
+    },
+    {
+      "grad_norm": 0.2308972030878067,
+      "learning_rate": 6.250174630998616e-05,
+      "loss": 0.0076,
+      "step": 22430
+    },
+    {
+      "grad_norm": 0.1409658044576645,
+      "learning_rate": 6.246972457087854e-05,
+      "loss": 0.0093,
+      "step": 22440
+    },
+    {
+      "grad_norm": 0.19626200199127197,
+      "learning_rate": 6.243769737708974e-05,
+      "loss": 0.0098,
+      "step": 22450
+    },
+    {
+      "grad_norm": 0.1862274408340454,
+      "learning_rate": 6.240566474262955e-05,
+      "loss": 0.0093,
+      "step": 22460
+    },
+    {
+      "grad_norm": 0.16315199434757233,
+      "learning_rate": 6.237362668151012e-05,
+      "loss": 0.0073,
+      "step": 22470
+    },
+    {
+      "grad_norm": 0.1666225790977478,
+      "learning_rate": 6.2341583207746e-05,
+      "loss": 0.007,
+      "step": 22480
+    },
+    {
+      "grad_norm": 0.16889432072639465,
+      "learning_rate": 6.23095343353541e-05,
+      "loss": 0.006,
+      "step": 22490
+    },
+    {
+      "grad_norm": 0.14835582673549652,
+      "learning_rate": 6.227748007835366e-05,
+      "loss": 0.0077,
+      "step": 22500
+    },
+    {
+      "grad_norm": 0.15847952663898468,
+      "learning_rate": 6.224542045076632e-05,
+      "loss": 0.0078,
+      "step": 22510
+    },
+    {
+      "grad_norm": 0.18088752031326294,
+      "learning_rate": 6.221335546661606e-05,
+      "loss": 0.0101,
+      "step": 22520
+    },
+    {
+      "grad_norm": 0.1811734288930893,
+      "learning_rate": 6.218128513992917e-05,
+      "loss": 0.0068,
+      "step": 22530
+    },
+    {
+      "grad_norm": 0.16454529762268066,
+      "learning_rate": 6.214920948473433e-05,
+      "loss": 0.0084,
+      "step": 22540
+    },
+    {
+      "grad_norm": 0.15164248645305634,
+      "learning_rate": 6.211712851506249e-05,
+      "loss": 0.0085,
+      "step": 22550
+    },
+    {
+      "grad_norm": 0.16459867358207703,
+      "learning_rate": 6.208504224494696e-05,
+      "loss": 0.007,
+      "step": 22560
+    },
+    {
+      "grad_norm": 0.1781519055366516,
+      "learning_rate": 6.205295068842338e-05,
+      "loss": 0.0089,
+      "step": 22570
+    },
+    {
+      "grad_norm": 0.1769600212574005,
+      "learning_rate": 6.202085385952968e-05,
+      "loss": 0.0081,
+      "step": 22580
+    },
+    {
+      "grad_norm": 0.14857007563114166,
+      "learning_rate": 6.19887517723061e-05,
+      "loss": 0.0073,
+      "step": 22590
+    },
+    {
+      "grad_norm": 0.18305015563964844,
+      "learning_rate": 6.19566444407952e-05,
+      "loss": 0.0071,
+      "step": 22600
+    },
+    {
+      "grad_norm": 0.1491784304380417,
+      "learning_rate": 6.192453187904178e-05,
+      "loss": 0.0097,
+      "step": 22610
+    },
+    {
+      "grad_norm": 0.28676852583885193,
+      "learning_rate": 6.189241410109303e-05,
+      "loss": 0.0086,
+      "step": 22620
+    },
+    {
+      "grad_norm": 0.10222618281841278,
+      "learning_rate": 6.186029112099829e-05,
+      "loss": 0.0081,
+      "step": 22630
+    },
+    {
+      "grad_norm": 0.1382959932088852,
+      "learning_rate": 6.182816295280926e-05,
+      "loss": 0.0072,
+      "step": 22640
+    },
+    {
+      "grad_norm": 0.2582602798938751,
+      "learning_rate": 6.179602961057993e-05,
+      "loss": 0.0087,
+      "step": 22650
+    },
+    {
+      "grad_norm": 0.19047607481479645,
+      "learning_rate": 6.176389110836647e-05,
+      "loss": 0.0081,
+      "step": 22660
+    },
+    {
+      "grad_norm": 0.17479892075061798,
+      "learning_rate": 6.173174746022738e-05,
+      "loss": 0.0088,
+      "step": 22670
+    },
+    {
+      "grad_norm": 0.24772889912128448,
+      "learning_rate": 6.169959868022338e-05,
+      "loss": 0.0074,
+      "step": 22680
+    },
+    {
+      "grad_norm": 0.14253035187721252,
+      "learning_rate": 6.166744478241742e-05,
+      "loss": 0.0091,
+      "step": 22690
+    },
+    {
+      "grad_norm": 0.21535921096801758,
+      "learning_rate": 6.163528578087473e-05,
+      "loss": 0.0094,
+      "step": 22700
+    },
+    {
+      "grad_norm": 0.20373694598674774,
+      "learning_rate": 6.160312168966274e-05,
+      "loss": 0.0098,
+      "step": 22710
+    },
+    {
+      "grad_norm": 0.20838889479637146,
+      "learning_rate": 6.157095252285112e-05,
+      "loss": 0.0086,
+      "step": 22720
+    },
+    {
+      "grad_norm": 0.18441857397556305,
+      "learning_rate": 6.153877829451174e-05,
+      "loss": 0.0071,
+      "step": 22730
+    },
+    {
+      "grad_norm": 0.18355795741081238,
+      "learning_rate": 6.150659901871872e-05,
+      "loss": 0.0088,
+      "step": 22740
+    },
+    {
+      "grad_norm": 0.1631142646074295,
+      "learning_rate": 6.147441470954836e-05,
+      "loss": 0.008,
+      "step": 22750
+    },
+    {
+      "grad_norm": 0.17655935883522034,
+      "learning_rate": 6.144222538107918e-05,
+      "loss": 0.0082,
+      "step": 22760
+    },
+    {
+      "grad_norm": 0.16474385559558868,
+      "learning_rate": 6.141003104739188e-05,
+      "loss": 0.0089,
+      "step": 22770
+    },
+    {
+      "grad_norm": 0.18977071344852448,
+      "learning_rate": 6.137783172256935e-05,
+      "loss": 0.0081,
+      "step": 22780
+    },
+    {
+      "grad_norm": 0.12454688549041748,
+      "learning_rate": 6.134562742069669e-05,
+      "loss": 0.0072,
+      "step": 22790
+    },
+    {
+      "grad_norm": 0.16954810917377472,
+      "learning_rate": 6.131341815586111e-05,
+      "loss": 0.0067,
+      "step": 22800
+    },
+    {
+      "grad_norm": 0.16513033211231232,
+      "learning_rate": 6.128120394215209e-05,
+      "loss": 0.0075,
+      "step": 22810
+    },
+    {
+      "grad_norm": 0.1710788905620575,
+      "learning_rate": 6.124898479366116e-05,
+      "loss": 0.0068,
+      "step": 22820
+    },
+    {
+      "grad_norm": 0.1888769418001175,
+      "learning_rate": 6.121676072448214e-05,
+      "loss": 0.006,
+      "step": 22830
+    },
+    {
+      "grad_norm": 0.20105259120464325,
+      "learning_rate": 6.118453174871089e-05,
+      "loss": 0.007,
+      "step": 22840
+    },
+    {
+      "grad_norm": 0.24393461644649506,
+      "learning_rate": 6.115229788044548e-05,
+      "loss": 0.0075,
+      "step": 22850
+    },
+    {
+      "grad_norm": 0.1934169977903366,
+      "learning_rate": 6.112005913378607e-05,
+      "loss": 0.0069,
+      "step": 22860
+    },
+    {
+      "grad_norm": 0.18673866987228394,
+      "learning_rate": 6.1087815522835e-05,
+      "loss": 0.0081,
+      "step": 22870
+    },
+    {
+      "grad_norm": 0.14310894906520844,
+      "learning_rate": 6.105556706169673e-05,
+      "loss": 0.0085,
+      "step": 22880
+    },
+    {
+      "grad_norm": 0.17685267329216003,
+      "learning_rate": 6.102331376447782e-05,
+      "loss": 0.0076,
+      "step": 22890
+    },
+    {
+      "grad_norm": 0.11734090000391006,
+      "learning_rate": 6.099105564528697e-05,
+      "loss": 0.008,
+      "step": 22900
+    },
+    {
+      "grad_norm": 0.16577933728694916,
+      "learning_rate": 6.095879271823495e-05,
+      "loss": 0.0074,
+      "step": 22910
+    },
+    {
+      "grad_norm": 0.23455731570720673,
+      "learning_rate": 6.09265249974347e-05,
+      "loss": 0.0089,
+      "step": 22920
+    },
+    {
+      "grad_norm": 0.19860807061195374,
+      "learning_rate": 6.089425249700118e-05,
+      "loss": 0.0104,
+      "step": 22930
+    },
+    {
+      "grad_norm": 0.15269480645656586,
+      "learning_rate": 6.0861975231051505e-05,
+      "loss": 0.0083,
+      "step": 22940
+    },
+    {
+      "grad_norm": 0.211292564868927,
+      "learning_rate": 6.082969321370484e-05,
+      "loss": 0.0069,
+      "step": 22950
+    },
+    {
+      "grad_norm": 0.17815667390823364,
+      "learning_rate": 6.079740645908243e-05,
+      "loss": 0.0077,
+      "step": 22960
+    },
+    {
+      "grad_norm": 0.19667363166809082,
+      "learning_rate": 6.07651149813076e-05,
+      "loss": 0.0076,
+      "step": 22970
+    },
+    {
+      "grad_norm": 0.22455249726772308,
+      "learning_rate": 6.073281879450574e-05,
+      "loss": 0.0082,
+      "step": 22980
+    },
+    {
+      "grad_norm": 0.1967453956604004,
+      "learning_rate": 6.070051791280431e-05,
+      "loss": 0.0079,
+      "step": 22990
+    },
+    {
+      "grad_norm": 0.25115442276000977,
+      "learning_rate": 6.066821235033281e-05,
+      "loss": 0.007,
+      "step": 23000
+    },
+    {
+      "grad_norm": 0.20721101760864258,
+      "learning_rate": 6.063590212122277e-05,
+      "loss": 0.0071,
+      "step": 23010
+    },
+    {
+      "grad_norm": 0.1786475032567978,
+      "learning_rate": 6.0603587239607794e-05,
+      "loss": 0.0072,
+      "step": 23020
+    },
+    {
+      "grad_norm": 0.1611877679824829,
+      "learning_rate": 6.057126771962353e-05,
+      "loss": 0.0075,
+      "step": 23030
+    },
+    {
+      "grad_norm": 0.14453038573265076,
+      "learning_rate": 6.05389435754076e-05,
+      "loss": 0.0075,
+      "step": 23040
+    },
+    {
+      "grad_norm": 0.1955098807811737,
+      "learning_rate": 6.0506614821099705e-05,
+      "loss": 0.0088,
+      "step": 23050
+    },
+    {
+      "grad_norm": 0.14700065553188324,
+      "learning_rate": 6.047428147084153e-05,
+      "loss": 0.0072,
+      "step": 23060
+    },
+    {
+      "grad_norm": 0.16171975433826447,
+      "learning_rate": 6.044194353877679e-05,
+      "loss": 0.0067,
+      "step": 23070
+    },
+    {
+      "grad_norm": 0.13628309965133667,
+      "learning_rate": 6.04096010390512e-05,
+      "loss": 0.0077,
+      "step": 23080
+    },
+    {
+      "grad_norm": 0.13651815056800842,
+      "learning_rate": 6.0377253985812445e-05,
+      "loss": 0.0075,
+      "step": 23090
+    },
+    {
+      "grad_norm": 0.17627166211605072,
+      "learning_rate": 6.0344902393210235e-05,
+      "loss": 0.0087,
+      "step": 23100
+    },
+    {
+      "grad_norm": 0.15148314833641052,
+      "learning_rate": 6.031254627539625e-05,
+      "loss": 0.0079,
+      "step": 23110
+    },
+    {
+      "grad_norm": 0.16996552050113678,
+      "learning_rate": 6.028018564652418e-05,
+      "loss": 0.0079,
+      "step": 23120
+    },
+    {
+      "grad_norm": 0.12153834104537964,
+      "learning_rate": 6.024782052074964e-05,
+      "loss": 0.0079,
+      "step": 23130
+    },
+    {
+      "grad_norm": 0.15507830679416656,
+      "learning_rate": 6.021545091223026e-05,
+      "loss": 0.0062,
+      "step": 23140
+    },
+    {
+      "grad_norm": 0.19956301152706146,
+      "learning_rate": 6.0183076835125586e-05,
+      "loss": 0.0104,
+      "step": 23150
+    },
+    {
+      "grad_norm": 0.19108371436595917,
+      "learning_rate": 6.015069830359714e-05,
+      "loss": 0.0085,
+      "step": 23160
+    },
+    {
+      "grad_norm": 0.19581642746925354,
+      "learning_rate": 6.0118315331808384e-05,
+      "loss": 0.0066,
+      "step": 23170
+    },
+    {
+      "grad_norm": 0.13546869158744812,
+      "learning_rate": 6.0085927933924745e-05,
+      "loss": 0.0074,
+      "step": 23180
+    },
+    {
+      "grad_norm": 0.18051974475383759,
+      "learning_rate": 6.0053536124113585e-05,
+      "loss": 0.0075,
+      "step": 23190
+    },
+    {
+      "grad_norm": 0.23201806843280792,
+      "learning_rate": 6.0021139916544144e-05,
+      "loss": 0.0072,
+      "step": 23200
+    },
+    {
+      "grad_norm": 0.16089293360710144,
+      "learning_rate": 5.998873932538764e-05,
+      "loss": 0.006,
+      "step": 23210
+    },
+    {
+      "grad_norm": 0.1441987156867981,
+      "learning_rate": 5.995633436481721e-05,
+      "loss": 0.0074,
+      "step": 23220
+    },
+    {
+      "grad_norm": 0.21817168593406677,
+      "learning_rate": 5.9923925049007855e-05,
+      "loss": 0.0065,
+      "step": 23230
+    },
+    {
+      "grad_norm": 0.20185299217700958,
+      "learning_rate": 5.989151139213656e-05,
+      "loss": 0.0076,
+      "step": 23240
+    },
+    {
+      "grad_norm": 0.20150071382522583,
+      "learning_rate": 5.9859093408382094e-05,
+      "loss": 0.0081,
+      "step": 23250
+    },
+    {
+      "grad_norm": 0.15181978046894073,
+      "learning_rate": 5.982667111192523e-05,
+      "loss": 0.0087,
+      "step": 23260
+    },
+    {
+      "grad_norm": 0.16457250714302063,
+      "learning_rate": 5.979424451694857e-05,
+      "loss": 0.008,
+      "step": 23270
+    },
+    {
+      "grad_norm": 0.13352569937705994,
+      "learning_rate": 5.97618136376366e-05,
+      "loss": 0.0094,
+      "step": 23280
+    },
+    {
+      "grad_norm": 0.12231193482875824,
+      "learning_rate": 5.97293784881757e-05,
+      "loss": 0.0074,
+      "step": 23290
+    },
+    {
+      "grad_norm": 0.1471714824438095,
+      "learning_rate": 5.969693908275411e-05,
+      "loss": 0.008,
+      "step": 23300
+    },
+    {
+      "grad_norm": 0.1607133150100708,
+      "learning_rate": 5.9664495435561916e-05,
+      "loss": 0.0078,
+      "step": 23310
+    },
+    {
+      "grad_norm": 0.19618867337703705,
+      "learning_rate": 5.963204756079108e-05,
+      "loss": 0.0071,
+      "step": 23320
+    },
+    {
+      "grad_norm": 0.13446524739265442,
+      "learning_rate": 5.959959547263538e-05,
+      "loss": 0.0066,
+      "step": 23330
+    },
+    {
+      "grad_norm": 0.186651349067688,
+      "learning_rate": 5.95671391852905e-05,
+      "loss": 0.0088,
+      "step": 23340
+    },
+    {
+      "grad_norm": 0.17093466222286224,
+      "learning_rate": 5.95346787129539e-05,
+      "loss": 0.0067,
+      "step": 23350
+    },
+    {
+      "grad_norm": 0.17361219227313995,
+      "learning_rate": 5.950221406982489e-05,
+      "loss": 0.0072,
+      "step": 23360
+    },
+    {
+      "grad_norm": 0.14821816980838776,
+      "learning_rate": 5.9469745270104625e-05,
+      "loss": 0.0082,
+      "step": 23370
+    },
+    {
+      "grad_norm": 0.2205081582069397,
+      "learning_rate": 5.9437272327996064e-05,
+      "loss": 0.0068,
+      "step": 23380
+    },
+    {
+      "grad_norm": 0.16671016812324524,
+      "learning_rate": 5.940479525770395e-05,
+      "loss": 0.0075,
+      "step": 23390
+    },
+    {
+      "grad_norm": 0.20130111277103424,
+      "learning_rate": 5.9372314073434876e-05,
+      "loss": 0.0063,
+      "step": 23400
+    },
+    {
+      "grad_norm": 0.1607874035835266,
+      "learning_rate": 5.93398287893972e-05,
+      "loss": 0.0071,
+      "step": 23410
+    },
+    {
+      "grad_norm": 0.1984701156616211,
+      "learning_rate": 5.930733941980111e-05,
+      "loss": 0.0092,
+      "step": 23420
+    },
+    {
+      "grad_norm": 0.1658877432346344,
+      "learning_rate": 5.927484597885855e-05,
+      "loss": 0.0089,
+      "step": 23430
+    },
+    {
+      "grad_norm": 0.16369958221912384,
+      "learning_rate": 5.924234848078325e-05,
+      "loss": 0.0077,
+      "step": 23440
+    },
+    {
+      "grad_norm": 0.166787788271904,
+      "learning_rate": 5.9209846939790715e-05,
+      "loss": 0.0071,
+      "step": 23450
+    },
+    {
+      "grad_norm": 0.22670699656009674,
+      "learning_rate": 5.917734137009824e-05,
+      "loss": 0.0087,
+      "step": 23460
+    },
+    {
+      "grad_norm": 0.24219302833080292,
+      "learning_rate": 5.914483178592485e-05,
+      "loss": 0.0095,
+      "step": 23470
+    },
+    {
+      "grad_norm": 0.19043977558612823,
+      "learning_rate": 5.911231820149136e-05,
+      "loss": 0.0077,
+      "step": 23480
+    },
+    {
+      "grad_norm": 0.1633724719285965,
+      "learning_rate": 5.907980063102031e-05,
+      "loss": 0.0062,
+      "step": 23490
+    },
+    {
+      "grad_norm": 0.17301912605762482,
+      "learning_rate": 5.904727908873597e-05,
+      "loss": 0.008,
+      "step": 23500
+    },
+    {
+      "grad_norm": 0.15818122029304504,
+      "learning_rate": 5.90147535888644e-05,
+      "loss": 0.0076,
+      "step": 23510
+    },
+    {
+      "grad_norm": 0.17122580111026764,
+      "learning_rate": 5.898222414563333e-05,
+      "loss": 0.0079,
+      "step": 23520
+    },
+    {
+      "grad_norm": 0.1849246621131897,
+      "learning_rate": 5.8949690773272256e-05,
+      "loss": 0.008,
+      "step": 23530
+    },
+    {
+      "grad_norm": 0.21565380692481995,
+      "learning_rate": 5.891715348601239e-05,
+      "loss": 0.009,
+      "step": 23540
+    },
+    {
+      "grad_norm": 0.15154613554477692,
+      "learning_rate": 5.888461229808663e-05,
+      "loss": 0.008,
+      "step": 23550
+    },
+    {
+      "grad_norm": 0.1491202563047409,
+      "learning_rate": 5.885206722372959e-05,
+      "loss": 0.0078,
+      "step": 23560
+    },
+    {
+      "grad_norm": 0.17586304247379303,
+      "learning_rate": 5.881951827717763e-05,
+      "loss": 0.0087,
+      "step": 23570
+    },
+    {
+      "grad_norm": 0.13815636932849884,
+      "learning_rate": 5.878696547266872e-05,
+      "loss": 0.0069,
+      "step": 23580
+    },
+    {
+      "grad_norm": 0.1776828169822693,
+      "learning_rate": 5.875440882444259e-05,
+      "loss": 0.0064,
+      "step": 23590
+    },
+    {
+      "grad_norm": 0.18223091959953308,
+      "learning_rate": 5.8721848346740617e-05,
+      "loss": 0.0078,
+      "step": 23600
+    },
+    {
+      "grad_norm": 0.14766332507133484,
+      "learning_rate": 5.868928405380585e-05,
+      "loss": 0.0073,
+      "step": 23610
+    },
+    {
+      "grad_norm": 0.15314190089702606,
+      "learning_rate": 5.865671595988304e-05,
+      "loss": 0.0074,
+      "step": 23620
+    },
+    {
+      "grad_norm": 0.18261218070983887,
+      "learning_rate": 5.862414407921857e-05,
+      "loss": 0.0075,
+      "step": 23630
+    },
+    {
+      "grad_norm": 0.16365991532802582,
+      "learning_rate": 5.8591568426060475e-05,
+      "loss": 0.0064,
+      "step": 23640
+    },
+    {
+      "grad_norm": 0.21958576142787933,
+      "learning_rate": 5.8558989014658464e-05,
+      "loss": 0.0078,
+      "step": 23650
+    },
+    {
+      "grad_norm": 0.17736117541790009,
+      "learning_rate": 5.8526405859263855e-05,
+      "loss": 0.0096,
+      "step": 23660
+    },
+    {
+      "grad_norm": 0.21416521072387695,
+      "learning_rate": 5.8493818974129667e-05,
+      "loss": 0.0078,
+      "step": 23670
+    },
+    {
+      "grad_norm": 0.17005442082881927,
+      "learning_rate": 5.846122837351048e-05,
+      "loss": 0.0062,
+      "step": 23680
+    },
+    {
+      "grad_norm": 0.16373682022094727,
+      "learning_rate": 5.842863407166255e-05,
+      "loss": 0.0064,
+      "step": 23690
+    },
+    {
+      "grad_norm": 0.17922191321849823,
+      "learning_rate": 5.83960360828437e-05,
+      "loss": 0.0069,
+      "step": 23700
+    },
+    {
+      "grad_norm": 0.16627110540866852,
+      "learning_rate": 5.836343442131343e-05,
+      "loss": 0.0066,
+      "step": 23710
+    },
+    {
+      "grad_norm": 0.10587631911039352,
+      "learning_rate": 5.833082910133281e-05,
+      "loss": 0.0079,
+      "step": 23720
+    },
+    {
+      "grad_norm": 0.16076789796352386,
+      "learning_rate": 5.82982201371645e-05,
+      "loss": 0.0077,
+      "step": 23730
+    },
+    {
+      "grad_norm": 0.17550235986709595,
+      "learning_rate": 5.826560754307279e-05,
+      "loss": 0.007,
+      "step": 23740
+    },
+    {
+      "grad_norm": 0.16617868840694427,
+      "learning_rate": 5.8232991333323516e-05,
+      "loss": 0.0083,
+      "step": 23750
+    },
+    {
+      "grad_norm": 0.1555328369140625,
+      "learning_rate": 5.8200371522184116e-05,
+      "loss": 0.009,
+      "step": 23760
+    },
+    {
+      "grad_norm": 0.18178623914718628,
+      "learning_rate": 5.816774812392362e-05,
+      "loss": 0.0065,
+      "step": 23770
+    },
+    {
+      "grad_norm": 0.15610867738723755,
+      "learning_rate": 5.8135121152812614e-05,
+      "loss": 0.0072,
+      "step": 23780
+    },
+    {
+      "grad_norm": 0.2291097790002823,
+      "learning_rate": 5.8102490623123204e-05,
+      "loss": 0.0063,
+      "step": 23790
+    },
+    {
+      "grad_norm": 0.22358888387680054,
+      "learning_rate": 5.806985654912914e-05,
+      "loss": 0.007,
+      "step": 23800
+    },
+    {
+      "grad_norm": 0.15256108343601227,
+      "learning_rate": 5.8037218945105644e-05,
+      "loss": 0.0078,
+      "step": 23810
+    },
+    {
+      "grad_norm": 0.2089097499847412,
+      "learning_rate": 5.800457782532953e-05,
+      "loss": 0.0068,
+      "step": 23820
+    },
+    {
+      "grad_norm": 0.25234970450401306,
+      "learning_rate": 5.797193320407912e-05,
+      "loss": 0.0076,
+      "step": 23830
+    },
+    {
+      "grad_norm": 0.2533577084541321,
+      "learning_rate": 5.793928509563431e-05,
+      "loss": 0.0072,
+      "step": 23840
+    },
+    {
+      "grad_norm": 0.18778306245803833,
+      "learning_rate": 5.790663351427644e-05,
+      "loss": 0.0081,
+      "step": 23850
+    },
+    {
+      "grad_norm": 0.20340430736541748,
+      "learning_rate": 5.7873978474288484e-05,
+      "loss": 0.0076,
+      "step": 23860
+    },
+    {
+      "grad_norm": 0.17989139258861542,
+      "learning_rate": 5.78413199899548e-05,
+      "loss": 0.007,
+      "step": 23870
+    },
+    {
+      "grad_norm": 0.20962738990783691,
+      "learning_rate": 5.780865807556136e-05,
+      "loss": 0.0064,
+      "step": 23880
+    },
+    {
+      "grad_norm": 0.15355485677719116,
+      "learning_rate": 5.777599274539559e-05,
+      "loss": 0.0062,
+      "step": 23890
+    },
+    {
+      "grad_norm": 0.17155227065086365,
+      "learning_rate": 5.77433240137464e-05,
+      "loss": 0.0072,
+      "step": 23900
+    },
+    {
+      "grad_norm": 0.14429792761802673,
+      "learning_rate": 5.771065189490422e-05,
+      "loss": 0.0064,
+      "step": 23910
+    },
+    {
+      "grad_norm": 0.15667051076889038,
+      "learning_rate": 5.767797640316095e-05,
+      "loss": 0.0059,
+      "step": 23920
+    },
+    {
+      "grad_norm": 0.174618199467659,
+      "learning_rate": 5.764529755280994e-05,
+      "loss": 0.007,
+      "step": 23930
+    },
+    {
+      "grad_norm": 0.16174370050430298,
+      "learning_rate": 5.761261535814603e-05,
+      "loss": 0.0066,
+      "step": 23940
+    },
+    {
+      "grad_norm": 0.19393038749694824,
+      "learning_rate": 5.757992983346554e-05,
+      "loss": 0.0073,
+      "step": 23950
+    },
+    {
+      "grad_norm": 0.16276024281978607,
+      "learning_rate": 5.754724099306621e-05,
+      "loss": 0.0079,
+      "step": 23960
+    },
+    {
+      "grad_norm": 0.24538612365722656,
+      "learning_rate": 5.751454885124727e-05,
+      "loss": 0.0093,
+      "step": 23970
+    },
+    {
+      "grad_norm": 0.1797083020210266,
+      "learning_rate": 5.748185342230935e-05,
+      "loss": 0.0071,
+      "step": 23980
+    },
+    {
+      "grad_norm": 0.14447303116321564,
+      "learning_rate": 5.744915472055456e-05,
+      "loss": 0.0074,
+      "step": 23990
+    },
+    {
+      "grad_norm": 0.13524650037288666,
+      "learning_rate": 5.741645276028642e-05,
+      "loss": 0.0068,
+      "step": 24000
+    },
+    {
+      "grad_norm": 0.21271362900733948,
+      "learning_rate": 5.7383747555809873e-05,
+      "loss": 0.0083,
+      "step": 24010
+    },
+    {
+      "grad_norm": 0.246890589594841,
+      "learning_rate": 5.7351039121431294e-05,
+      "loss": 0.0065,
+      "step": 24020
+    },
+    {
+      "grad_norm": 0.15719853341579437,
+      "learning_rate": 5.7318327471458454e-05,
+      "loss": 0.0065,
+      "step": 24030
+    },
+    {
+      "grad_norm": 0.13013042509555817,
+      "learning_rate": 5.728561262020054e-05,
+      "loss": 0.0059,
+      "step": 24040
+    },
+    {
+      "grad_norm": 0.1957017332315445,
+      "learning_rate": 5.725289458196814e-05,
+      "loss": 0.0069,
+      "step": 24050
+    },
+    {
+      "grad_norm": 0.18127702176570892,
+      "learning_rate": 5.722017337107324e-05,
+      "loss": 0.0076,
+      "step": 24060
+    },
+    {
+      "grad_norm": 0.12906430661678314,
+      "learning_rate": 5.7187449001829206e-05,
+      "loss": 0.0073,
+      "step": 24070
+    },
+    {
+      "grad_norm": 0.15885525941848755,
+      "learning_rate": 5.715472148855081e-05,
+      "loss": 0.0074,
+      "step": 24080
+    },
+    {
+      "grad_norm": 0.1468164622783661,
+      "learning_rate": 5.712199084555414e-05,
+      "loss": 0.0066,
+      "step": 24090
+    },
+    {
+      "grad_norm": 0.16547293961048126,
+      "learning_rate": 5.7089257087156734e-05,
+      "loss": 0.0079,
+      "step": 24100
+    },
+    {
+      "grad_norm": 0.19798710942268372,
+      "learning_rate": 5.70565202276774e-05,
+      "loss": 0.0079,
+      "step": 24110
+    },
+    {
+      "grad_norm": 0.17495644092559814,
+      "learning_rate": 5.70237802814364e-05,
+      "loss": 0.0059,
+      "step": 24120
+    },
+    {
+      "grad_norm": 0.15358901023864746,
+      "learning_rate": 5.6991037262755277e-05,
+      "loss": 0.0067,
+      "step": 24130
+    },
+    {
+      "grad_norm": 0.13903512060642242,
+      "learning_rate": 5.6958291185956924e-05,
+      "loss": 0.0061,
+      "step": 24140
+    },
+    {
+      "grad_norm": 0.17674005031585693,
+      "learning_rate": 5.692554206536561e-05,
+      "loss": 0.0058,
+      "step": 24150
+    },
+    {
+      "grad_norm": 0.15553773939609528,
+      "learning_rate": 5.689278991530692e-05,
+      "loss": 0.0094,
+      "step": 24160
+    },
+    {
+      "grad_norm": 0.1602412313222885,
+      "learning_rate": 5.686003475010773e-05,
+      "loss": 0.0073,
+      "step": 24170
+    },
+    {
+      "grad_norm": 0.200641468167305,
+      "learning_rate": 5.682727658409628e-05,
+      "loss": 0.0072,
+      "step": 24180
+    },
+    {
+      "grad_norm": 0.11190202087163925,
+      "learning_rate": 5.67945154316021e-05,
+      "loss": 0.0058,
+      "step": 24190
+    },
+    {
+      "grad_norm": 0.10504284501075745,
+      "learning_rate": 5.676175130695602e-05,
+      "loss": 0.0094,
+      "step": 24200
+    },
+    {
+      "grad_norm": 0.18198174238204956,
+      "learning_rate": 5.672898422449022e-05,
+      "loss": 0.0076,
+      "step": 24210
+    },
+    {
+      "grad_norm": 0.15979549288749695,
+      "learning_rate": 5.66962141985381e-05,
+      "loss": 0.008,
+      "step": 24220
+    },
+    {
+      "grad_norm": 0.13591469824314117,
+      "learning_rate": 5.666344124343439e-05,
+      "loss": 0.0064,
+      "step": 24230
+    },
+    {
+      "grad_norm": 0.13755783438682556,
+      "learning_rate": 5.66306653735151e-05,
+      "loss": 0.0058,
+      "step": 24240
+    },
+    {
+      "grad_norm": 0.14406374096870422,
+      "learning_rate": 5.659788660311749e-05,
+      "loss": 0.0064,
+      "step": 24250
+    },
+    {
+      "grad_norm": 0.11100298911333084,
+      "learning_rate": 5.656510494658014e-05,
+      "loss": 0.0063,
+      "step": 24260
+    },
+    {
+      "grad_norm": 0.12320985645055771,
+      "learning_rate": 5.6532320418242835e-05,
+      "loss": 0.0069,
+      "step": 24270
+    },
+    {
+      "grad_norm": 0.12480558454990387,
+      "learning_rate": 5.649953303244665e-05,
+      "loss": 0.0063,
+      "step": 24280
+    },
+    {
+      "grad_norm": 0.1463974118232727,
+      "learning_rate": 5.646674280353389e-05,
+      "loss": 0.0069,
+      "step": 24290
+    },
+    {
+      "grad_norm": 0.1874552071094513,
+      "learning_rate": 5.6433949745848135e-05,
+      "loss": 0.0087,
+      "step": 24300
+    },
+    {
+      "grad_norm": 0.1899162381887436,
+      "learning_rate": 5.6401153873734166e-05,
+      "loss": 0.0091,
+      "step": 24310
+    },
+    {
+      "grad_norm": 0.17803268134593964,
+      "learning_rate": 5.6368355201538036e-05,
+      "loss": 0.0072,
+      "step": 24320
+    },
+    {
+      "grad_norm": 0.17859311401844025,
+      "learning_rate": 5.633555374360697e-05,
+      "loss": 0.0065,
+      "step": 24330
+    },
+    {
+      "grad_norm": 0.15151937305927277,
+      "learning_rate": 5.630274951428944e-05,
+      "loss": 0.0075,
+      "step": 24340
+    },
+    {
+      "grad_norm": 0.16927455365657806,
+      "learning_rate": 5.626994252793517e-05,
+      "loss": 0.0079,
+      "step": 24350
+    },
+    {
+      "grad_norm": 0.14722636342048645,
+      "learning_rate": 5.6237132798895006e-05,
+      "loss": 0.0076,
+      "step": 24360
+    },
+    {
+      "grad_norm": 0.178540900349617,
+      "learning_rate": 5.620432034152106e-05,
+      "loss": 0.0066,
+      "step": 24370
+    },
+    {
+      "grad_norm": 0.12089332938194275,
+      "learning_rate": 5.6171505170166604e-05,
+      "loss": 0.006,
+      "step": 24380
+    },
+    {
+      "grad_norm": 0.1521543562412262,
+      "learning_rate": 5.613868729918612e-05,
+      "loss": 0.0071,
+      "step": 24390
+    },
+    {
+      "grad_norm": 0.18245062232017517,
+      "learning_rate": 5.6105866742935256e-05,
+      "loss": 0.0081,
+      "step": 24400
+    },
+    {
+      "grad_norm": 0.1957232505083084,
+      "learning_rate": 5.607304351577083e-05,
+      "loss": 0.0084,
+      "step": 24410
+    },
+    {
+      "grad_norm": 0.16492673754692078,
+      "learning_rate": 5.604021763205085e-05,
+      "loss": 0.0066,
+      "step": 24420
+    },
+    {
+      "grad_norm": 0.2060045301914215,
+      "learning_rate": 5.600738910613445e-05,
+      "loss": 0.0075,
+      "step": 24430
+    },
+    {
+      "grad_norm": 0.19383934140205383,
+      "learning_rate": 5.597455795238198e-05,
+      "loss": 0.0087,
+      "step": 24440
+    },
+    {
+      "grad_norm": 0.19557106494903564,
+      "learning_rate": 5.5941724185154876e-05,
+      "loss": 0.01,
+      "step": 24450
+    },
+    {
+      "grad_norm": 0.16713592410087585,
+      "learning_rate": 5.5908887818815746e-05,
+      "loss": 0.0056,
+      "step": 24460
+    },
+    {
+      "grad_norm": 0.13703852891921997,
+      "learning_rate": 5.587604886772834e-05,
+      "loss": 0.0058,
+      "step": 24470
+    },
+    {
+      "grad_norm": 0.13655143976211548,
+      "learning_rate": 5.584320734625752e-05,
+      "loss": 0.0072,
+      "step": 24480
+    },
+    {
+      "grad_norm": 0.13014283776283264,
+      "learning_rate": 5.581036326876928e-05,
+      "loss": 0.0055,
+      "step": 24490
+    },
+    {
+      "grad_norm": 0.13620811700820923,
+      "learning_rate": 5.5777516649630756e-05,
+      "loss": 0.0076,
+      "step": 24500
+    },
+    {
+      "grad_norm": 0.21399259567260742,
+      "learning_rate": 5.574466750321017e-05,
+      "loss": 0.008,
+      "step": 24510
+    },
+    {
+      "grad_norm": 0.11930834501981735,
+      "learning_rate": 5.571181584387683e-05,
+      "loss": 0.0068,
+      "step": 24520
+    },
+    {
+      "grad_norm": 0.1614333689212799,
+      "learning_rate": 5.5678961686001196e-05,
+      "loss": 0.0066,
+      "step": 24530
+    },
+    {
+      "grad_norm": 0.12719382345676422,
+      "learning_rate": 5.5646105043954785e-05,
+      "loss": 0.0058,
+      "step": 24540
+    },
+    {
+      "grad_norm": 0.1688818633556366,
+      "learning_rate": 5.5613245932110205e-05,
+      "loss": 0.007,
+      "step": 24550
+    },
+    {
+      "grad_norm": 0.1624341458082199,
+      "learning_rate": 5.5580384364841157e-05,
+      "loss": 0.0069,
+      "step": 24560
+    },
+    {
+      "grad_norm": 0.1821034997701645,
+      "learning_rate": 5.554752035652237e-05,
+      "loss": 0.0062,
+      "step": 24570
+    },
+    {
+      "grad_norm": 0.1863766461610794,
+      "learning_rate": 5.5514653921529736e-05,
+      "loss": 0.0077,
+      "step": 24580
+    },
+    {
+      "grad_norm": 0.1719273179769516,
+      "learning_rate": 5.5481785074240114e-05,
+      "loss": 0.006,
+      "step": 24590
+    },
+    {
+      "grad_norm": 0.18013523519039154,
+      "learning_rate": 5.544891382903146e-05,
+      "loss": 0.0068,
+      "step": 24600
+    },
+    {
+      "grad_norm": 0.17401951551437378,
+      "learning_rate": 5.541604020028277e-05,
+      "loss": 0.0063,
+      "step": 24610
+    },
+    {
+      "grad_norm": 0.1296333223581314,
+      "learning_rate": 5.5383164202374105e-05,
+      "loss": 0.0074,
+      "step": 24620
+    },
+    {
+      "grad_norm": 0.16221222281455994,
+      "learning_rate": 5.5350285849686515e-05,
+      "loss": 0.0065,
+      "step": 24630
+    },
+    {
+      "grad_norm": 0.14296604692935944,
+      "learning_rate": 5.531740515660213e-05,
+      "loss": 0.0062,
+      "step": 24640
+    },
+    {
+      "grad_norm": 0.16340278089046478,
+      "learning_rate": 5.528452213750408e-05,
+      "loss": 0.0058,
+      "step": 24650
+    },
+    {
+      "grad_norm": 0.15513882040977478,
+      "learning_rate": 5.52516368067765e-05,
+      "loss": 0.0077,
+      "step": 24660
+    },
+    {
+      "grad_norm": 0.10874688625335693,
+      "learning_rate": 5.5218749178804563e-05,
+      "loss": 0.0059,
+      "step": 24670
+    },
+    {
+      "grad_norm": 0.16584277153015137,
+      "learning_rate": 5.518585926797443e-05,
+      "loss": 0.0063,
+      "step": 24680
+    },
+    {
+      "grad_norm": 0.13687729835510254,
+      "learning_rate": 5.5152967088673265e-05,
+      "loss": 0.0069,
+      "step": 24690
+    },
+    {
+      "grad_norm": 0.15806390345096588,
+      "learning_rate": 5.512007265528924e-05,
+      "loss": 0.0063,
+      "step": 24700
+    },
+    {
+      "grad_norm": 0.18036741018295288,
+      "learning_rate": 5.508717598221148e-05,
+      "loss": 0.0087,
+      "step": 24710
+    },
+    {
+      "grad_norm": 0.12424280494451523,
+      "learning_rate": 5.505427708383012e-05,
+      "loss": 0.0066,
+      "step": 24720
+    },
+    {
+      "grad_norm": 0.14004452526569366,
+      "learning_rate": 5.502137597453624e-05,
+      "loss": 0.0062,
+      "step": 24730
+    },
+    {
+      "grad_norm": 0.15052810311317444,
+      "learning_rate": 5.4988472668721914e-05,
+      "loss": 0.0071,
+      "step": 24740
+    },
+    {
+      "grad_norm": 0.15550321340560913,
+      "learning_rate": 5.4955567180780164e-05,
+      "loss": 0.007,
+      "step": 24750
+    },
+    {
+      "grad_norm": 0.1557753086090088,
+      "learning_rate": 5.492265952510497e-05,
+      "loss": 0.0064,
+      "step": 24760
+    },
+    {
+      "grad_norm": 0.14320296049118042,
+      "learning_rate": 5.4889749716091254e-05,
+      "loss": 0.008,
+      "step": 24770
+    },
+    {
+      "grad_norm": 0.16342711448669434,
+      "learning_rate": 5.4856837768134885e-05,
+      "loss": 0.0071,
+      "step": 24780
+    },
+    {
+      "grad_norm": 0.14770549535751343,
+      "learning_rate": 5.482392369563267e-05,
+      "loss": 0.0076,
+      "step": 24790
+    },
+    {
+      "grad_norm": 0.12295711040496826,
+      "learning_rate": 5.4791007512982354e-05,
+      "loss": 0.0068,
+      "step": 24800
+    },
+    {
+      "grad_norm": 0.18345165252685547,
+      "learning_rate": 5.4758089234582566e-05,
+      "loss": 0.0074,
+      "step": 24810
+    },
+    {
+      "grad_norm": 0.21627971529960632,
+      "learning_rate": 5.47251688748329e-05,
+      "loss": 0.0087,
+      "step": 24820
+    },
+    {
+      "grad_norm": 0.1900508999824524,
+      "learning_rate": 5.4692246448133843e-05,
+      "loss": 0.0072,
+      "step": 24830
+    },
+    {
+      "grad_norm": 0.1458715796470642,
+      "learning_rate": 5.465932196888676e-05,
+      "loss": 0.0063,
+      "step": 24840
+    },
+    {
+      "grad_norm": 0.15152527391910553,
+      "learning_rate": 5.462639545149395e-05,
+      "loss": 0.0067,
+      "step": 24850
+    },
+    {
+      "grad_norm": 0.2525458335876465,
+      "learning_rate": 5.459346691035861e-05,
+      "loss": 0.0099,
+      "step": 24860
+    },
+    {
+      "grad_norm": 0.1214364767074585,
+      "learning_rate": 5.4560536359884776e-05,
+      "loss": 0.0072,
+      "step": 24870
+    },
+    {
+      "grad_norm": 0.1853283941745758,
+      "learning_rate": 5.45276038144774e-05,
+      "loss": 0.0085,
+      "step": 24880
+    },
+    {
+      "grad_norm": 0.1570974588394165,
+      "learning_rate": 5.449466928854229e-05,
+      "loss": 0.0074,
+      "step": 24890
+    },
+    {
+      "grad_norm": 0.10555271059274673,
+      "learning_rate": 5.446173279648613e-05,
+      "loss": 0.0065,
+      "step": 24900
+    },
+    {
+      "grad_norm": 0.1739118993282318,
+      "learning_rate": 5.442879435271646e-05,
+      "loss": 0.0085,
+      "step": 24910
+    },
+    {
+      "grad_norm": 0.14852970838546753,
+      "learning_rate": 5.4395853971641675e-05,
+      "loss": 0.0074,
+      "step": 24920
+    },
+    {
+      "grad_norm": 0.1766151636838913,
+      "learning_rate": 5.4362911667671015e-05,
+      "loss": 0.0066,
+      "step": 24930
+    },
+    {
+      "grad_norm": 0.2383168786764145,
+      "learning_rate": 5.432996745521458e-05,
+      "loss": 0.0089,
+      "step": 24940
+    },
+    {
+      "grad_norm": 0.17214009165763855,
+      "learning_rate": 5.4297021348683264e-05,
+      "loss": 0.0058,
+      "step": 24950
+    },
+    {
+      "grad_norm": 0.17735984921455383,
+      "learning_rate": 5.426407336248882e-05,
+      "loss": 0.0087,
+      "step": 24960
+    },
+    {
+      "grad_norm": 0.1778302788734436,
+      "learning_rate": 5.423112351104382e-05,
+      "loss": 0.007,
+      "step": 24970
+    },
+    {
+      "grad_norm": 0.1492597609758377,
+      "learning_rate": 5.4198171808761654e-05,
+      "loss": 0.0066,
+      "step": 24980
+    },
+    {
+      "grad_norm": 0.15815657377243042,
+      "learning_rate": 5.41652182700565e-05,
+      "loss": 0.0062,
+      "step": 24990
+    },
+    {
+      "grad_norm": 0.2175200879573822,
+      "learning_rate": 5.413226290934338e-05,
+      "loss": 0.0092,
+      "step": 25000
+    },
+    {
+      "grad_norm": 0.1794508546590805,
+      "learning_rate": 5.4099305741038056e-05,
+      "loss": 0.006,
+      "step": 25010
+    },
+    {
+      "grad_norm": 0.14141304790973663,
+      "learning_rate": 5.406634677955713e-05,
+      "loss": 0.0065,
+      "step": 25020
+    },
+    {
+      "grad_norm": 0.1738635003566742,
+      "learning_rate": 5.403338603931798e-05,
+      "loss": 0.0069,
+      "step": 25030
+    },
+    {
+      "grad_norm": 0.17221927642822266,
+      "learning_rate": 5.4000423534738745e-05,
+      "loss": 0.008,
+      "step": 25040
+    },
+    {
+      "grad_norm": 0.15490365028381348,
+      "learning_rate": 5.396745928023835e-05,
+      "loss": 0.0066,
+      "step": 25050
+    },
+    {
+      "grad_norm": 0.1719140261411667,
+      "learning_rate": 5.3934493290236474e-05,
+      "loss": 0.0076,
+      "step": 25060
+    },
+    {
+      "grad_norm": 0.18049362301826477,
+      "learning_rate": 5.390152557915357e-05,
+      "loss": 0.0065,
+      "step": 25070
+    },
+    {
+      "grad_norm": 0.22681918740272522,
+      "learning_rate": 5.3868556161410796e-05,
+      "loss": 0.0103,
+      "step": 25080
+    },
+    {
+      "grad_norm": 0.16085097193717957,
+      "learning_rate": 5.383558505143015e-05,
+      "loss": 0.0066,
+      "step": 25090
+    },
+    {
+      "grad_norm": 0.16711580753326416,
+      "learning_rate": 5.38026122636343e-05,
+      "loss": 0.0075,
+      "step": 25100
+    },
+    {
+      "grad_norm": 0.2535240650177002,
+      "learning_rate": 5.3769637812446636e-05,
+      "loss": 0.0084,
+      "step": 25110
+    },
+    {
+      "grad_norm": 0.14821471273899078,
+      "learning_rate": 5.373666171229133e-05,
+      "loss": 0.0064,
+      "step": 25120
+    },
+    {
+      "grad_norm": 0.13103604316711426,
+      "learning_rate": 5.370368397759324e-05,
+      "loss": 0.0074,
+      "step": 25130
+    },
+    {
+      "grad_norm": 0.18176057934761047,
+      "learning_rate": 5.3670704622777944e-05,
+      "loss": 0.0074,
+      "step": 25140
+    },
+    {
+      "grad_norm": 0.1321435272693634,
+      "learning_rate": 5.3637723662271744e-05,
+      "loss": 0.0078,
+      "step": 25150
+    },
+    {
+      "grad_norm": 0.155270516872406,
+      "learning_rate": 5.360474111050162e-05,
+      "loss": 0.0071,
+      "step": 25160
+    },
+    {
+      "grad_norm": 0.14102475345134735,
+      "learning_rate": 5.357175698189527e-05,
+      "loss": 0.0067,
+      "step": 25170
+    },
+    {
+      "grad_norm": 0.12524829804897308,
+      "learning_rate": 5.353877129088107e-05,
+      "loss": 0.0065,
+      "step": 25180
+    },
+    {
+      "grad_norm": 0.1284032166004181,
+      "learning_rate": 5.350578405188809e-05,
+      "loss": 0.0066,
+      "step": 25190
+    },
+    {
+      "grad_norm": 0.16433344781398773,
+      "learning_rate": 5.347279527934603e-05,
+      "loss": 0.0064,
+      "step": 25200
+    },
+    {
+      "grad_norm": 0.21402862668037415,
+      "learning_rate": 5.343980498768536e-05,
+      "loss": 0.0075,
+      "step": 25210
+    },
+    {
+      "grad_norm": 0.1542135775089264,
+      "learning_rate": 5.340681319133711e-05,
+      "loss": 0.0072,
+      "step": 25220
+    },
+    {
+      "grad_norm": 0.2302778959274292,
+      "learning_rate": 5.337381990473305e-05,
+      "loss": 0.006,
+      "step": 25230
+    },
+    {
+      "grad_norm": 0.14822626113891602,
+      "learning_rate": 5.3340825142305504e-05,
+      "loss": 0.007,
+      "step": 25240
+    },
+    {
+      "grad_norm": 0.16254059970378876,
+      "learning_rate": 5.330782891848756e-05,
+      "loss": 0.0075,
+      "step": 25250
+    },
+    {
+      "grad_norm": 0.10788800567388535,
+      "learning_rate": 5.327483124771285e-05,
+      "loss": 0.006,
+      "step": 25260
+    },
+    {
+      "grad_norm": 0.1708451211452484,
+      "learning_rate": 5.3241832144415706e-05,
+      "loss": 0.0079,
+      "step": 25270
+    },
+    {
+      "grad_norm": 0.18555140495300293,
+      "learning_rate": 5.320883162303104e-05,
+      "loss": 0.0057,
+      "step": 25280
+    },
+    {
+      "grad_norm": 0.19584108889102936,
+      "learning_rate": 5.3175829697994416e-05,
+      "loss": 0.0072,
+      "step": 25290
+    },
+    {
+      "grad_norm": 0.1611512154340744,
+      "learning_rate": 5.3142826383741975e-05,
+      "loss": 0.0059,
+      "step": 25300
+    },
+    {
+      "grad_norm": 0.11129899322986603,
+      "learning_rate": 5.3109821694710506e-05,
+      "loss": 0.0067,
+      "step": 25310
+    },
+    {
+      "grad_norm": 0.10932119935750961,
+      "learning_rate": 5.307681564533736e-05,
+      "loss": 0.0058,
+      "step": 25320
+    },
+    {
+      "grad_norm": 0.1800096184015274,
+      "learning_rate": 5.3043808250060535e-05,
+      "loss": 0.0069,
+      "step": 25330
+    },
+    {
+      "grad_norm": 0.21970100700855255,
+      "learning_rate": 5.3010799523318566e-05,
+      "loss": 0.006,
+      "step": 25340
+    },
+    {
+      "grad_norm": 0.18059051036834717,
+      "learning_rate": 5.29777894795506e-05,
+      "loss": 0.0075,
+      "step": 25350
+    },
+    {
+      "grad_norm": 0.18768922984600067,
+      "learning_rate": 5.294477813319634e-05,
+      "loss": 0.0065,
+      "step": 25360
+    },
+    {
+      "grad_norm": 0.19060800969600677,
+      "learning_rate": 5.291176549869608e-05,
+      "loss": 0.0068,
+      "step": 25370
+    },
+    {
+      "grad_norm": 0.15788216888904572,
+      "learning_rate": 5.287875159049065e-05,
+      "loss": 0.0063,
+      "step": 25380
+    },
+    {
+      "grad_norm": 0.17700673639774323,
+      "learning_rate": 5.284573642302149e-05,
+      "loss": 0.0067,
+      "step": 25390
+    },
+    {
+      "grad_norm": 0.170267254114151,
+      "learning_rate": 5.2812720010730543e-05,
+      "loss": 0.0055,
+      "step": 25400
+    },
+    {
+      "grad_norm": 0.17600862681865692,
+      "learning_rate": 5.277970236806029e-05,
+      "loss": 0.0065,
+      "step": 25410
+    },
+    {
+      "grad_norm": 0.14608055353164673,
+      "learning_rate": 5.2746683509453795e-05,
+      "loss": 0.0074,
+      "step": 25420
+    },
+    {
+      "grad_norm": 0.17069856822490692,
+      "learning_rate": 5.271366344935461e-05,
+      "loss": 0.0066,
+      "step": 25430
+    },
+    {
+      "grad_norm": 0.12357627600431442,
+      "learning_rate": 5.268064220220683e-05,
+      "loss": 0.0059,
+      "step": 25440
+    },
+    {
+      "grad_norm": 0.15186940133571625,
+      "learning_rate": 5.2647619782455094e-05,
+      "loss": 0.0081,
+      "step": 25450
+    },
+    {
+      "grad_norm": 0.1337217539548874,
+      "learning_rate": 5.261459620454451e-05,
+      "loss": 0.0057,
+      "step": 25460
+    },
+    {
+      "grad_norm": 0.15506835281848907,
+      "learning_rate": 5.258157148292071e-05,
+      "loss": 0.0071,
+      "step": 25470
+    },
+    {
+      "grad_norm": 0.15926700830459595,
+      "learning_rate": 5.2548545632029846e-05,
+      "loss": 0.0081,
+      "step": 25480
+    },
+    {
+      "grad_norm": 0.15714184939861298,
+      "learning_rate": 5.2515518666318534e-05,
+      "loss": 0.0071,
+      "step": 25490
+    },
+    {
+      "grad_norm": 0.1460959017276764,
+      "learning_rate": 5.248249060023389e-05,
+      "loss": 0.0085,
+      "step": 25500
+    },
+    {
+      "grad_norm": 0.19611872732639313,
+      "learning_rate": 5.2449461448223517e-05,
+      "loss": 0.0058,
+      "step": 25510
+    },
+    {
+      "grad_norm": 0.23220933973789215,
+      "learning_rate": 5.2416431224735486e-05,
+      "loss": 0.0071,
+      "step": 25520
+    },
+    {
+      "grad_norm": 0.14269185066223145,
+      "learning_rate": 5.238339994421836e-05,
+      "loss": 0.0083,
+      "step": 25530
+    },
+    {
+      "grad_norm": 0.18891027569770813,
+      "learning_rate": 5.235036762112111e-05,
+      "loss": 0.0065,
+      "step": 25540
+    },
+    {
+      "grad_norm": 0.16727878153324127,
+      "learning_rate": 5.2317334269893206e-05,
+      "loss": 0.0074,
+      "step": 25550
+    },
+    {
+      "grad_norm": 0.1406354457139969,
+      "learning_rate": 5.228429990498458e-05,
+      "loss": 0.0073,
+      "step": 25560
+    },
+    {
+      "grad_norm": 0.17169837653636932,
+      "learning_rate": 5.225126454084556e-05,
+      "loss": 0.0068,
+      "step": 25570
+    },
+    {
+      "grad_norm": 0.2224496454000473,
+      "learning_rate": 5.221822819192694e-05,
+      "loss": 0.0061,
+      "step": 25580
+    },
+    {
+      "grad_norm": 0.16147352755069733,
+      "learning_rate": 5.218519087267995e-05,
+      "loss": 0.0085,
+      "step": 25590
+    },
+    {
+      "grad_norm": 0.18976305425167084,
+      "learning_rate": 5.2152152597556225e-05,
+      "loss": 0.0072,
+      "step": 25600
+    },
+    {
+      "grad_norm": 0.17545996606349945,
+      "learning_rate": 5.211911338100784e-05,
+      "loss": 0.007,
+      "step": 25610
+    },
+    {
+      "grad_norm": 0.15801173448562622,
+      "learning_rate": 5.2086073237487264e-05,
+      "loss": 0.0059,
+      "step": 25620
+    },
+    {
+      "grad_norm": 0.15710054337978363,
+      "learning_rate": 5.205303218144739e-05,
+      "loss": 0.0072,
+      "step": 25630
+    },
+    {
+      "grad_norm": 0.10817389190196991,
+      "learning_rate": 5.2019990227341494e-05,
+      "loss": 0.0055,
+      "step": 25640
+    },
+    {
+      "grad_norm": 0.19574813544750214,
+      "learning_rate": 5.198694738962324e-05,
+      "loss": 0.0058,
+      "step": 25650
+    },
+    {
+      "grad_norm": 0.19225330650806427,
+      "learning_rate": 5.19539036827467e-05,
+      "loss": 0.0056,
+      "step": 25660
+    },
+    {
+      "grad_norm": 0.17006218433380127,
+      "learning_rate": 5.192085912116629e-05,
+      "loss": 0.0064,
+      "step": 25670
+    },
+    {
+      "grad_norm": 0.15731649100780487,
+      "learning_rate": 5.188781371933685e-05,
+      "loss": 0.0075,
+      "step": 25680
+    },
+    {
+      "grad_norm": 0.17119547724723816,
+      "learning_rate": 5.185476749171356e-05,
+      "loss": 0.0065,
+      "step": 25690
+    },
+    {
+      "grad_norm": 0.18865446746349335,
+      "learning_rate": 5.1821720452751945e-05,
+      "loss": 0.0071,
+      "step": 25700
+    },
+    {
+      "grad_norm": 0.17776601016521454,
+      "learning_rate": 5.178867261690791e-05,
+      "loss": 0.0076,
+      "step": 25710
+    },
+    {
+      "grad_norm": 0.16401617228984833,
+      "learning_rate": 5.1755623998637715e-05,
+      "loss": 0.0064,
+      "step": 25720
+    },
+    {
+      "grad_norm": 0.13625378906726837,
+      "learning_rate": 5.172257461239794e-05,
+      "loss": 0.0056,
+      "step": 25730
+    },
+    {
+      "grad_norm": 0.1432926207780838,
+      "learning_rate": 5.168952447264548e-05,
+      "loss": 0.007,
+      "step": 25740
+    },
+    {
+      "grad_norm": 0.14246174693107605,
+      "learning_rate": 5.165647359383763e-05,
+      "loss": 0.0063,
+      "step": 25750
+    },
+    {
+      "grad_norm": 0.12528719007968903,
+      "learning_rate": 5.1623421990431954e-05,
+      "loss": 0.0067,
+      "step": 25760
+    },
+    {
+      "grad_norm": 0.14926491677761078,
+      "learning_rate": 5.159036967688635e-05,
+      "loss": 0.0068,
+      "step": 25770
+    },
+    {
+      "grad_norm": 0.16560791432857513,
+      "learning_rate": 5.1557316667659015e-05,
+      "loss": 0.0059,
+      "step": 25780
+    },
+    {
+      "grad_norm": 0.14463017880916595,
+      "learning_rate": 5.152426297720846e-05,
+      "loss": 0.009,
+      "step": 25790
+    },
+    {
+      "grad_norm": 0.1735028326511383,
+      "learning_rate": 5.1491208619993505e-05,
+      "loss": 0.0062,
+      "step": 25800
+    },
+    {
+      "grad_norm": 0.16969357430934906,
+      "learning_rate": 5.145815361047321e-05,
+      "loss": 0.0057,
+      "step": 25810
+    },
+    {
+      "grad_norm": 0.15688154101371765,
+      "learning_rate": 5.1425097963107003e-05,
+      "loss": 0.0063,
+      "step": 25820
+    },
+    {
+      "grad_norm": 0.17411458492279053,
+      "learning_rate": 5.1392041692354544e-05,
+      "loss": 0.0054,
+      "step": 25830
+    },
+    {
+      "grad_norm": 0.1602076143026352,
+      "learning_rate": 5.1358984812675736e-05,
+      "loss": 0.0069,
+      "step": 25840
+    },
+    {
+      "grad_norm": 0.13085627555847168,
+      "learning_rate": 5.13259273385308e-05,
+      "loss": 0.006,
+      "step": 25850
+    },
+    {
+      "grad_norm": 0.1737157702445984,
+      "learning_rate": 5.12928692843802e-05,
+      "loss": 0.0063,
+      "step": 25860
+    },
+    {
+      "grad_norm": 0.15454013645648956,
+      "learning_rate": 5.125981066468465e-05,
+      "loss": 0.0064,
+      "step": 25870
+    },
+    {
+      "grad_norm": 0.16939125955104828,
+      "learning_rate": 5.122675149390514e-05,
+      "loss": 0.0062,
+      "step": 25880
+    },
+    {
+      "grad_norm": 0.14058345556259155,
+      "learning_rate": 5.119369178650282e-05,
+      "loss": 0.0071,
+      "step": 25890
+    },
+    {
+      "grad_norm": 0.24220606684684753,
+      "learning_rate": 5.1160631556939156e-05,
+      "loss": 0.007,
+      "step": 25900
+    },
+    {
+      "grad_norm": 0.13441194593906403,
+      "learning_rate": 5.112757081967584e-05,
+      "loss": 0.0063,
+      "step": 25910
+    },
+    {
+      "grad_norm": 0.2573894262313843,
+      "learning_rate": 5.109450958917472e-05,
+      "loss": 0.0063,
+      "step": 25920
+    },
+    {
+      "grad_norm": 0.14964747428894043,
+      "learning_rate": 5.106144787989794e-05,
+      "loss": 0.0069,
+      "step": 25930
+    },
+    {
+      "grad_norm": 0.13486990332603455,
+      "learning_rate": 5.102838570630778e-05,
+      "loss": 0.0075,
+      "step": 25940
+    },
+    {
+      "grad_norm": 0.07799253612756729,
+      "learning_rate": 5.099532308286678e-05,
+      "loss": 0.0058,
+      "step": 25950
+    },
+    {
+      "grad_norm": 0.16573242843151093,
+      "learning_rate": 5.0962260024037645e-05,
+      "loss": 0.0072,
+      "step": 25960
+    },
+    {
+      "grad_norm": 0.17470288276672363,
+      "learning_rate": 5.0929196544283276e-05,
+      "loss": 0.0066,
+      "step": 25970
+    },
+    {
+      "grad_norm": 0.16659271717071533,
+      "learning_rate": 5.089613265806678e-05,
+      "loss": 0.0084,
+      "step": 25980
+    },
+    {
+      "grad_norm": 0.17127497494220734,
+      "learning_rate": 5.086306837985141e-05,
+      "loss": 0.0085,
+      "step": 25990
+    },
+    {
+      "grad_norm": 0.140870600938797,
+      "learning_rate": 5.0830003724100604e-05,
+      "loss": 0.006,
+      "step": 26000
+    },
+    {
+      "grad_norm": 0.16184291243553162,
+      "learning_rate": 5.0796938705277984e-05,
+      "loss": 0.0068,
+      "step": 26010
+    },
+    {
+      "grad_norm": 0.16134357452392578,
+      "learning_rate": 5.076387333784729e-05,
+      "loss": 0.006,
+      "step": 26020
+    },
+    {
+      "grad_norm": 0.13644742965698242,
+      "learning_rate": 5.0730807636272445e-05,
+      "loss": 0.0066,
+      "step": 26030
+    },
+    {
+      "grad_norm": 0.1706143319606781,
+      "learning_rate": 5.0697741615017524e-05,
+      "loss": 0.0075,
+      "step": 26040
+    },
+    {
+      "grad_norm": 0.13527065515518188,
+      "learning_rate": 5.06646752885467e-05,
+      "loss": 0.0055,
+      "step": 26050
+    },
+    {
+      "grad_norm": 0.21148478984832764,
+      "learning_rate": 5.0631608671324324e-05,
+      "loss": 0.0087,
+      "step": 26060
+    },
+    {
+      "grad_norm": 0.16156083345413208,
+      "learning_rate": 5.059854177781487e-05,
+      "loss": 0.0059,
+      "step": 26070
+    },
+    {
+      "grad_norm": 0.15618406236171722,
+      "learning_rate": 5.05654746224829e-05,
+      "loss": 0.0062,
+      "step": 26080
+    },
+    {
+      "grad_norm": 0.19268229603767395,
+      "learning_rate": 5.053240721979311e-05,
+      "loss": 0.0056,
+      "step": 26090
+    },
+    {
+      "grad_norm": 0.1683729887008667,
+      "learning_rate": 5.049933958421031e-05,
+      "loss": 0.0058,
+      "step": 26100
+    },
+    {
+      "grad_norm": 0.18132859468460083,
+      "learning_rate": 5.046627173019943e-05,
+      "loss": 0.007,
+      "step": 26110
+    },
+    {
+      "grad_norm": 0.19777502119541168,
+      "learning_rate": 5.043320367222545e-05,
+      "loss": 0.0071,
+      "step": 26120
+    },
+    {
+      "grad_norm": 0.16235019266605377,
+      "learning_rate": 5.040013542475346e-05,
+      "loss": 0.0057,
+      "step": 26130
+    },
+    {
+      "grad_norm": 0.13332867622375488,
+      "learning_rate": 5.0367067002248646e-05,
+      "loss": 0.0054,
+      "step": 26140
+    },
+    {
+      "grad_norm": 0.10432196408510208,
+      "learning_rate": 5.0333998419176254e-05,
+      "loss": 0.0064,
+      "step": 26150
+    },
+    {
+      "grad_norm": 0.14084380865097046,
+      "learning_rate": 5.030092969000161e-05,
+      "loss": 0.0063,
+      "step": 26160
+    },
+    {
+      "grad_norm": 0.17615759372711182,
+      "learning_rate": 5.026786082919008e-05,
+      "loss": 0.0071,
+      "step": 26170
+    },
+    {
+      "grad_norm": 0.1512862741947174,
+      "learning_rate": 5.0234791851207144e-05,
+      "loss": 0.0059,
+      "step": 26180
+    },
+    {
+      "grad_norm": 0.16053873300552368,
+      "learning_rate": 5.0201722770518256e-05,
+      "loss": 0.006,
+      "step": 26190
+    },
+    {
+      "grad_norm": 0.16982480883598328,
+      "learning_rate": 5.0168653601589e-05,
+      "loss": 0.0077,
+      "step": 26200
+    },
+    {
+      "grad_norm": 0.14041326940059662,
+      "learning_rate": 5.01355843588849e-05,
+      "loss": 0.0056,
+      "step": 26210
+    },
+    {
+      "grad_norm": 0.12147341668605804,
+      "learning_rate": 5.010251505687161e-05,
+      "loss": 0.0062,
+      "step": 26220
+    },
+    {
+      "grad_norm": 0.10925732553005219,
+      "learning_rate": 5.006944571001474e-05,
+      "loss": 0.0073,
+      "step": 26230
+    },
+    {
+      "grad_norm": 0.1466297209262848,
+      "learning_rate": 5.003637633277997e-05,
+      "loss": 0.0065,
+      "step": 26240
+    },
+    {
+      "grad_norm": 0.11616086214780807,
+      "learning_rate": 5.0003306939632946e-05,
+      "loss": 0.0059,
+      "step": 26250
+    },
+    {
+      "grad_norm": 0.18093860149383545,
+      "learning_rate": 4.997023754503937e-05,
+      "loss": 0.0103,
+      "step": 26260
+    },
+    {
+      "grad_norm": 0.16750606894493103,
+      "learning_rate": 4.9937168163464897e-05,
+      "loss": 0.0063,
+      "step": 26270
+    },
+    {
+      "grad_norm": 0.16685305535793304,
+      "learning_rate": 4.990409880937519e-05,
+      "loss": 0.007,
+      "step": 26280
+    },
+    {
+      "grad_norm": 0.14823304116725922,
+      "learning_rate": 4.9871029497235936e-05,
+      "loss": 0.0061,
+      "step": 26290
+    },
+    {
+      "grad_norm": 0.14584434032440186,
+      "learning_rate": 4.9837960241512766e-05,
+      "loss": 0.0051,
+      "step": 26300
+    },
+    {
+      "grad_norm": 0.14362552762031555,
+      "learning_rate": 4.9804891056671265e-05,
+      "loss": 0.01,
+      "step": 26310
+    },
+    {
+      "grad_norm": 0.13646885752677917,
+      "learning_rate": 4.9771821957177074e-05,
+      "loss": 0.007,
+      "step": 26320
+    },
+    {
+      "grad_norm": 0.16100139915943146,
+      "learning_rate": 4.9738752957495694e-05,
+      "loss": 0.0069,
+      "step": 26330
+    },
+    {
+      "grad_norm": 0.18663838505744934,
+      "learning_rate": 4.970568407209265e-05,
+      "loss": 0.0094,
+      "step": 26340
+    },
+    {
+      "grad_norm": 0.12992233037948608,
+      "learning_rate": 4.967261531543339e-05,
+      "loss": 0.0065,
+      "step": 26350
+    },
+    {
+      "grad_norm": 0.09844040870666504,
+      "learning_rate": 4.963954670198332e-05,
+      "loss": 0.005,
+      "step": 26360
+    },
+    {
+      "grad_norm": 0.15050379931926727,
+      "learning_rate": 4.960647824620776e-05,
+      "loss": 0.0064,
+      "step": 26370
+    },
+    {
+      "grad_norm": 0.17685900628566742,
+      "learning_rate": 4.9573409962571995e-05,
+      "loss": 0.0066,
+      "step": 26380
+    },
+    {
+      "grad_norm": 0.18092231452465057,
+      "learning_rate": 4.954034186554121e-05,
+      "loss": 0.0089,
+      "step": 26390
+    },
+    {
+      "grad_norm": 0.16215774416923523,
+      "learning_rate": 4.950727396958051e-05,
+      "loss": 0.0085,
+      "step": 26400
+    },
+    {
+      "grad_norm": 0.13883137702941895,
+      "learning_rate": 4.947420628915489e-05,
+      "loss": 0.0064,
+      "step": 26410
+    },
+    {
+      "grad_norm": 0.11414729058742523,
+      "learning_rate": 4.944113883872932e-05,
+      "loss": 0.0063,
+      "step": 26420
+    },
+    {
+      "grad_norm": 0.15884439647197723,
+      "learning_rate": 4.9408071632768624e-05,
+      "loss": 0.0063,
+      "step": 26430
+    },
+    {
+      "grad_norm": 0.15553578734397888,
+      "learning_rate": 4.937500468573747e-05,
+      "loss": 0.0073,
+      "step": 26440
+    },
+    {
+      "grad_norm": 0.15340955555438995,
+      "learning_rate": 4.934193801210053e-05,
+      "loss": 0.0052,
+      "step": 26450
+    },
+    {
+      "grad_norm": 0.13266998529434204,
+      "learning_rate": 4.930887162632224e-05,
+      "loss": 0.0054,
+      "step": 26460
+    },
+    {
+      "grad_norm": 0.15985818207263947,
+      "learning_rate": 4.9275805542866994e-05,
+      "loss": 0.0067,
+      "step": 26470
+    },
+    {
+      "grad_norm": 0.16829422116279602,
+      "learning_rate": 4.9242739776199006e-05,
+      "loss": 0.0067,
+      "step": 26480
+    },
+    {
+      "grad_norm": 0.16595429182052612,
+      "learning_rate": 4.920967434078237e-05,
+      "loss": 0.0068,
+      "step": 26490
+    },
+    {
+      "grad_norm": 0.17019698023796082,
+      "learning_rate": 4.917660925108101e-05,
+      "loss": 0.0082,
+      "step": 26500
+    },
+    {
+      "grad_norm": 0.13594266772270203,
+      "learning_rate": 4.914354452155876e-05,
+      "loss": 0.0084,
+      "step": 26510
+    },
+    {
+      "grad_norm": 0.11783600598573685,
+      "learning_rate": 4.911048016667923e-05,
+      "loss": 0.0061,
+      "step": 26520
+    },
+    {
+      "grad_norm": 0.15677933394908905,
+      "learning_rate": 4.907741620090588e-05,
+      "loss": 0.009,
+      "step": 26530
+    },
+    {
+      "grad_norm": 0.13114793598651886,
+      "learning_rate": 4.9044352638702046e-05,
+      "loss": 0.0066,
+      "step": 26540
+    },
+    {
+      "grad_norm": 0.15701310336589813,
+      "learning_rate": 4.9011289494530834e-05,
+      "loss": 0.0066,
+      "step": 26550
+    },
+    {
+      "grad_norm": 0.2280014455318451,
+      "learning_rate": 4.897822678285516e-05,
+      "loss": 0.0064,
+      "step": 26560
+    },
+    {
+      "grad_norm": 0.1317746788263321,
+      "learning_rate": 4.8945164518137835e-05,
+      "loss": 0.008,
+      "step": 26570
+    },
+    {
+      "grad_norm": 0.21588997542858124,
+      "learning_rate": 4.891210271484138e-05,
+      "loss": 0.0089,
+      "step": 26580
+    },
+    {
+      "grad_norm": 0.14511559903621674,
+      "learning_rate": 4.8879041387428164e-05,
+      "loss": 0.0061,
+      "step": 26590
+    },
+    {
+      "grad_norm": 0.19094480574131012,
+      "learning_rate": 4.8845980550360306e-05,
+      "loss": 0.0091,
+      "step": 26600
+    },
+    {
+      "grad_norm": 0.1594228744506836,
+      "learning_rate": 4.8812920218099786e-05,
+      "loss": 0.0068,
+      "step": 26610
+    },
+    {
+      "grad_norm": 0.13017764687538147,
+      "learning_rate": 4.8779860405108294e-05,
+      "loss": 0.0063,
+      "step": 26620
+    },
+    {
+      "grad_norm": 0.15065419673919678,
+      "learning_rate": 4.8746801125847285e-05,
+      "loss": 0.0075,
+      "step": 26630
+    },
+    {
+      "grad_norm": 0.12885192036628723,
+      "learning_rate": 4.8713742394778064e-05,
+      "loss": 0.0069,
+      "step": 26640
+    },
+    {
+      "grad_norm": 0.14686627686023712,
+      "learning_rate": 4.868068422636162e-05,
+      "loss": 0.0068,
+      "step": 26650
+    },
+    {
+      "grad_norm": 0.17857609689235687,
+      "learning_rate": 4.86476266350587e-05,
+      "loss": 0.0066,
+      "step": 26660
+    },
+    {
+      "grad_norm": 0.1036638468503952,
+      "learning_rate": 4.861456963532985e-05,
+      "loss": 0.0062,
+      "step": 26670
+    },
+    {
+      "grad_norm": 0.15028482675552368,
+      "learning_rate": 4.8581513241635316e-05,
+      "loss": 0.0063,
+      "step": 26680
+    },
+    {
+      "grad_norm": 0.13248305022716522,
+      "learning_rate": 4.8548457468435065e-05,
+      "loss": 0.0063,
+      "step": 26690
+    },
+    {
+      "grad_norm": 0.1583397537469864,
+      "learning_rate": 4.851540233018884e-05,
+      "loss": 0.0062,
+      "step": 26700
+    },
+    {
+      "grad_norm": 0.14838658273220062,
+      "learning_rate": 4.848234784135608e-05,
+      "loss": 0.0072,
+      "step": 26710
+    },
+    {
+      "grad_norm": 0.156615749001503,
+      "learning_rate": 4.8449294016395916e-05,
+      "loss": 0.0063,
+      "step": 26720
+    },
+    {
+      "grad_norm": 0.20161956548690796,
+      "learning_rate": 4.8416240869767246e-05,
+      "loss": 0.0095,
+      "step": 26730
+    },
+    {
+      "grad_norm": 0.21392270922660828,
+      "learning_rate": 4.838318841592864e-05,
+      "loss": 0.0062,
+      "step": 26740
+    },
+    {
+      "grad_norm": 0.1743893176317215,
+      "learning_rate": 4.835013666933834e-05,
+      "loss": 0.0059,
+      "step": 26750
+    },
+    {
+      "grad_norm": 0.18253424763679504,
+      "learning_rate": 4.83170856444543e-05,
+      "loss": 0.0072,
+      "step": 26760
+    },
+    {
+      "grad_norm": 0.14783994853496552,
+      "learning_rate": 4.828403535573419e-05,
+      "loss": 0.0074,
+      "step": 26770
+    },
+    {
+      "grad_norm": 0.1475859135389328,
+      "learning_rate": 4.825098581763532e-05,
+      "loss": 0.0061,
+      "step": 26780
+    },
+    {
+      "grad_norm": 0.19664110243320465,
+      "learning_rate": 4.821793704461465e-05,
+      "loss": 0.0095,
+      "step": 26790
+    },
+    {
+      "grad_norm": 0.17041297256946564,
+      "learning_rate": 4.8184889051128884e-05,
+      "loss": 0.0074,
+      "step": 26800
+    },
+    {
+      "grad_norm": 0.10668256878852844,
+      "learning_rate": 4.81518418516343e-05,
+      "loss": 0.0057,
+      "step": 26810
+    },
+    {
+      "grad_norm": 0.15636016428470612,
+      "learning_rate": 4.811879546058688e-05,
+      "loss": 0.0062,
+      "step": 26820
+    },
+    {
+      "grad_norm": 0.17856693267822266,
+      "learning_rate": 4.808574989244224e-05,
+      "loss": 0.0071,
+      "step": 26830
+    },
+    {
+      "grad_norm": 0.14647427201271057,
+      "learning_rate": 4.8052705161655644e-05,
+      "loss": 0.0056,
+      "step": 26840
+    },
+    {
+      "grad_norm": 0.1294129639863968,
+      "learning_rate": 4.8019661282681936e-05,
+      "loss": 0.0067,
+      "step": 26850
+    },
+    {
+      "grad_norm": 0.15451762080192566,
+      "learning_rate": 4.79866182699757e-05,
+      "loss": 0.0075,
+      "step": 26860
+    },
+    {
+      "grad_norm": 0.174496129155159,
+      "learning_rate": 4.795357613799103e-05,
+      "loss": 0.0062,
+      "step": 26870
+    },
+    {
+      "grad_norm": 0.17193371057510376,
+      "learning_rate": 4.792053490118166e-05,
+      "loss": 0.0059,
+      "step": 26880
+    },
+    {
+      "grad_norm": 0.15428949892520905,
+      "learning_rate": 4.788749457400099e-05,
+      "loss": 0.0058,
+      "step": 26890
+    },
+    {
+      "grad_norm": 0.10243827104568481,
+      "learning_rate": 4.7854455170901966e-05,
+      "loss": 0.0059,
+      "step": 26900
+    },
+    {
+      "grad_norm": 0.1588166356086731,
+      "learning_rate": 4.7821416706337136e-05,
+      "loss": 0.0065,
+      "step": 26910
+    },
+    {
+      "grad_norm": 0.16216854751110077,
+      "learning_rate": 4.7788379194758645e-05,
+      "loss": 0.0056,
+      "step": 26920
+    },
+    {
+      "grad_norm": 0.13596296310424805,
+      "learning_rate": 4.7755342650618246e-05,
+      "loss": 0.0068,
+      "step": 26930
+    },
+    {
+      "grad_norm": 0.17935620248317719,
+      "learning_rate": 4.7722307088367235e-05,
+      "loss": 0.0057,
+      "step": 26940
+    },
+    {
+      "grad_norm": 0.11294172704219818,
+      "learning_rate": 4.768927252245646e-05,
+      "loss": 0.0058,
+      "step": 26950
+    },
+    {
+      "grad_norm": 0.14291492104530334,
+      "learning_rate": 4.765623896733642e-05,
+      "loss": 0.0062,
+      "step": 26960
+    },
+    {
+      "grad_norm": 0.1529158353805542,
+      "learning_rate": 4.762320643745708e-05,
+      "loss": 0.0054,
+      "step": 26970
+    },
+    {
+      "grad_norm": 0.2189231812953949,
+      "learning_rate": 4.7590174947267966e-05,
+      "loss": 0.0077,
+      "step": 26980
+    },
+    {
+      "grad_norm": 0.22403566539287567,
+      "learning_rate": 4.755714451121823e-05,
+      "loss": 0.0072,
+      "step": 26990
+    },
+    {
+      "grad_norm": 0.16163156926631927,
+      "learning_rate": 4.7524115143756476e-05,
+      "loss": 0.0075,
+      "step": 27000
+    },
+    {
+      "grad_norm": 0.15197168290615082,
+      "learning_rate": 4.7491086859330854e-05,
+      "loss": 0.0064,
+      "step": 27010
+    },
+    {
+      "grad_norm": 0.17928281426429749,
+      "learning_rate": 4.74580596723891e-05,
+      "loss": 0.0071,
+      "step": 27020
+    },
+    {
+      "grad_norm": 0.13206763565540314,
+      "learning_rate": 4.742503359737841e-05,
+      "loss": 0.0084,
+      "step": 27030
+    },
+    {
+      "grad_norm": 0.17218123376369476,
+      "learning_rate": 4.739200864874547e-05,
+      "loss": 0.0072,
+      "step": 27040
+    },
+    {
+      "grad_norm": 0.147601917386055,
+      "learning_rate": 4.735898484093658e-05,
+      "loss": 0.0062,
+      "step": 27050
+    },
+    {
+      "grad_norm": 0.20331603288650513,
+      "learning_rate": 4.7325962188397444e-05,
+      "loss": 0.0066,
+      "step": 27060
+    },
+    {
+      "grad_norm": 0.1489640772342682,
+      "learning_rate": 4.729294070557329e-05,
+      "loss": 0.0077,
+      "step": 27070
+    },
+    {
+      "grad_norm": 0.19359008967876434,
+      "learning_rate": 4.7259920406908833e-05,
+      "loss": 0.0076,
+      "step": 27080
+    },
+    {
+      "grad_norm": 0.18380579352378845,
+      "learning_rate": 4.7226901306848306e-05,
+      "loss": 0.006,
+      "step": 27090
+    },
+    {
+      "grad_norm": 0.1048455610871315,
+      "learning_rate": 4.719388341983535e-05,
+      "loss": 0.0058,
+      "step": 27100
+    },
+    {
+      "grad_norm": 0.12091352790594101,
+      "learning_rate": 4.71608667603131e-05,
+      "loss": 0.0054,
+      "step": 27110
+    },
+    {
+      "grad_norm": 0.12144492566585541,
+      "learning_rate": 4.712785134272422e-05,
+      "loss": 0.0057,
+      "step": 27120
+    },
+    {
+      "grad_norm": 0.16054564714431763,
+      "learning_rate": 4.7094837181510734e-05,
+      "loss": 0.0069,
+      "step": 27130
+    },
+    {
+      "grad_norm": 0.14296454191207886,
+      "learning_rate": 4.706182429111414e-05,
+      "loss": 0.0056,
+      "step": 27140
+    },
+    {
+      "grad_norm": 0.15150395035743713,
+      "learning_rate": 4.702881268597545e-05,
+      "loss": 0.0054,
+      "step": 27150
+    },
+    {
+      "grad_norm": 0.15006963908672333,
+      "learning_rate": 4.699580238053503e-05,
+      "loss": 0.0049,
+      "step": 27160
+    },
+    {
+      "grad_norm": 0.20197927951812744,
+      "learning_rate": 4.69627933892327e-05,
+      "loss": 0.0059,
+      "step": 27170
+    },
+    {
+      "grad_norm": 0.20977924764156342,
+      "learning_rate": 4.692978572650773e-05,
+      "loss": 0.0077,
+      "step": 27180
+    },
+    {
+      "grad_norm": 0.17916904389858246,
+      "learning_rate": 4.689677940679881e-05,
+      "loss": 0.0059,
+      "step": 27190
+    },
+    {
+      "grad_norm": 0.17385835945606232,
+      "learning_rate": 4.6863774444543984e-05,
+      "loss": 0.0063,
+      "step": 27200
+    },
+    {
+      "grad_norm": 0.12588851153850555,
+      "learning_rate": 4.6830770854180784e-05,
+      "loss": 0.0066,
+      "step": 27210
+    },
+    {
+      "grad_norm": 0.17916767299175262,
+      "learning_rate": 4.679776865014608e-05,
+      "loss": 0.006,
+      "step": 27220
+    },
+    {
+      "grad_norm": 0.12838995456695557,
+      "learning_rate": 4.676476784687615e-05,
+      "loss": 0.0053,
+      "step": 27230
+    },
+    {
+      "grad_norm": 0.1505282074213028,
+      "learning_rate": 4.673176845880669e-05,
+      "loss": 0.0054,
+      "step": 27240
+    },
+    {
+      "grad_norm": 0.1675380915403366,
+      "learning_rate": 4.669877050037275e-05,
+      "loss": 0.0071,
+      "step": 27250
+    },
+    {
+      "grad_norm": 0.1202409490942955,
+      "learning_rate": 4.666577398600874e-05,
+      "loss": 0.006,
+      "step": 27260
+    },
+    {
+      "grad_norm": 0.12660051882266998,
+      "learning_rate": 4.663277893014846e-05,
+      "loss": 0.0055,
+      "step": 27270
+    },
+    {
+      "grad_norm": 0.19774045050144196,
+      "learning_rate": 4.659978534722508e-05,
+      "loss": 0.0067,
+      "step": 27280
+    },
+    {
+      "grad_norm": 0.16269466280937195,
+      "learning_rate": 4.6566793251671104e-05,
+      "loss": 0.0061,
+      "step": 27290
+    },
+    {
+      "grad_norm": 0.14417198300361633,
+      "learning_rate": 4.65338026579184e-05,
+      "loss": 0.0074,
+      "step": 27300
+    },
+    {
+      "grad_norm": 0.16276110708713531,
+      "learning_rate": 4.650081358039817e-05,
+      "loss": 0.0058,
+      "step": 27310
+    },
+    {
+      "grad_norm": 0.18586868047714233,
+      "learning_rate": 4.646782603354098e-05,
+      "loss": 0.006,
+      "step": 27320
+    },
+    {
+      "grad_norm": 0.11495117843151093,
+      "learning_rate": 4.643484003177666e-05,
+      "loss": 0.0069,
+      "step": 27330
+    },
+    {
+      "grad_norm": 0.13397309184074402,
+      "learning_rate": 4.6401855589534454e-05,
+      "loss": 0.0062,
+      "step": 27340
+    },
+    {
+      "grad_norm": 0.11913822591304779,
+      "learning_rate": 4.636887272124286e-05,
+      "loss": 0.0058,
+      "step": 27350
+    },
+    {
+      "grad_norm": 0.11111169308423996,
+      "learning_rate": 4.6335891441329685e-05,
+      "loss": 0.0054,
+      "step": 27360
+    },
+    {
+      "grad_norm": 0.17166897654533386,
+      "learning_rate": 4.630291176422209e-05,
+      "loss": 0.0052,
+      "step": 27370
+    },
+    {
+      "grad_norm": 0.16911393404006958,
+      "learning_rate": 4.62699337043465e-05,
+      "loss": 0.0054,
+      "step": 27380
+    },
+    {
+      "grad_norm": 0.09430872648954391,
+      "learning_rate": 4.623695727612861e-05,
+      "loss": 0.0044,
+      "step": 27390
+    },
+    {
+      "grad_norm": 0.1213165894150734,
+      "learning_rate": 4.620398249399347e-05,
+      "loss": 0.0057,
+      "step": 27400
+    },
+    {
+      "grad_norm": 0.15220476686954498,
+      "learning_rate": 4.617100937236535e-05,
+      "loss": 0.0053,
+      "step": 27410
+    },
+    {
+      "grad_norm": 0.10825729370117188,
+      "learning_rate": 4.613803792566781e-05,
+      "loss": 0.0054,
+      "step": 27420
+    },
+    {
+      "grad_norm": 0.1373918205499649,
+      "learning_rate": 4.61050681683237e-05,
+      "loss": 0.0068,
+      "step": 27430
+    },
+    {
+      "grad_norm": 0.11251986026763916,
+      "learning_rate": 4.6072100114755104e-05,
+      "loss": 0.007,
+      "step": 27440
+    },
+    {
+      "grad_norm": 0.18032753467559814,
+      "learning_rate": 4.603913377938336e-05,
+      "loss": 0.0078,
+      "step": 27450
+    },
+    {
+      "grad_norm": 0.1447577029466629,
+      "learning_rate": 4.600616917662905e-05,
+      "loss": 0.0074,
+      "step": 27460
+    },
+    {
+      "grad_norm": 0.12773725390434265,
+      "learning_rate": 4.597320632091206e-05,
+      "loss": 0.0063,
+      "step": 27470
+    },
+    {
+      "grad_norm": 0.13714884221553802,
+      "learning_rate": 4.594024522665142e-05,
+      "loss": 0.0056,
+      "step": 27480
+    },
+    {
+      "grad_norm": 0.13713540136814117,
+      "learning_rate": 4.590728590826543e-05,
+      "loss": 0.0065,
+      "step": 27490
+    },
+    {
+      "grad_norm": 0.10262228548526764,
+      "learning_rate": 4.587432838017165e-05,
+      "loss": 0.005,
+      "step": 27500
+    },
+    {
+      "grad_norm": 0.16094517707824707,
+      "learning_rate": 4.58413726567868e-05,
+      "loss": 0.006,
+      "step": 27510
+    },
+    {
+      "grad_norm": 0.17002801597118378,
+      "learning_rate": 4.5808418752526814e-05,
+      "loss": 0.0074,
+      "step": 27520
+    },
+    {
+      "grad_norm": 0.1544531136751175,
+      "learning_rate": 4.5775466681806895e-05,
+      "loss": 0.0057,
+      "step": 27530
+    },
+    {
+      "grad_norm": 0.1377790868282318,
+      "learning_rate": 4.574251645904136e-05,
+      "loss": 0.0063,
+      "step": 27540
+    },
+    {
+      "grad_norm": 0.12394864112138748,
+      "learning_rate": 4.5709568098643776e-05,
+      "loss": 0.0053,
+      "step": 27550
+    },
+    {
+      "grad_norm": 0.1612556278705597,
+      "learning_rate": 4.567662161502687e-05,
+      "loss": 0.0077,
+      "step": 27560
+    },
+    {
+      "grad_norm": 0.16595953702926636,
+      "learning_rate": 4.5643677022602554e-05,
+      "loss": 0.0068,
+      "step": 27570
+    },
+    {
+      "grad_norm": 0.11988869309425354,
+      "learning_rate": 4.56107343357819e-05,
+      "loss": 0.0047,
+      "step": 27580
+    },
+    {
+      "grad_norm": 0.14439725875854492,
+      "learning_rate": 4.557779356897518e-05,
+      "loss": 0.006,
+      "step": 27590
+    },
+    {
+      "grad_norm": 0.15019118785858154,
+      "learning_rate": 4.55448547365918e-05,
+      "loss": 0.0067,
+      "step": 27600
+    },
+    {
+      "grad_norm": 0.1113421767950058,
+      "learning_rate": 4.5511917853040315e-05,
+      "loss": 0.0058,
+      "step": 27610
+    },
+    {
+      "grad_norm": 0.1581684947013855,
+      "learning_rate": 4.547898293272842e-05,
+      "loss": 0.0071,
+      "step": 27620
+    },
+    {
+      "grad_norm": 0.15211492776870728,
+      "learning_rate": 4.544604999006301e-05,
+      "loss": 0.006,
+      "step": 27630
+    },
+    {
+      "grad_norm": 0.2394832968711853,
+      "learning_rate": 4.541311903945004e-05,
+      "loss": 0.0072,
+      "step": 27640
+    },
+    {
+      "grad_norm": 0.19549426436424255,
+      "learning_rate": 4.538019009529463e-05,
+      "loss": 0.0062,
+      "step": 27650
+    },
+    {
+      "grad_norm": 0.1871912181377411,
+      "learning_rate": 4.5347263172001035e-05,
+      "loss": 0.0064,
+      "step": 27660
+    },
+    {
+      "grad_norm": 0.13701000809669495,
+      "learning_rate": 4.531433828397261e-05,
+      "loss": 0.0052,
+      "step": 27670
+    },
+    {
+      "grad_norm": 0.13966816663742065,
+      "learning_rate": 4.528141544561179e-05,
+      "loss": 0.0062,
+      "step": 27680
+    },
+    {
+      "grad_norm": 0.14909961819648743,
+      "learning_rate": 4.5248494671320186e-05,
+      "loss": 0.008,
+      "step": 27690
+    },
+    {
+      "grad_norm": 0.14133793115615845,
+      "learning_rate": 4.521557597549844e-05,
+      "loss": 0.0054,
+      "step": 27700
+    },
+    {
+      "grad_norm": 0.15147075057029724,
+      "learning_rate": 4.5182659372546284e-05,
+      "loss": 0.0082,
+      "step": 27710
+    },
+    {
+      "grad_norm": 0.1687939465045929,
+      "learning_rate": 4.514974487686261e-05,
+      "loss": 0.0094,
+      "step": 27720
+    },
+    {
+      "grad_norm": 0.1581992655992508,
+      "learning_rate": 4.5116832502845306e-05,
+      "loss": 0.0064,
+      "step": 27730
+    },
+    {
+      "grad_norm": 0.16090218722820282,
+      "learning_rate": 4.508392226489136e-05,
+      "loss": 0.0061,
+      "step": 27740
+    },
+    {
+      "grad_norm": 0.11756483465433121,
+      "learning_rate": 4.505101417739683e-05,
+      "loss": 0.0068,
+      "step": 27750
+    },
+    {
+      "grad_norm": 0.1574968695640564,
+      "learning_rate": 4.501810825475685e-05,
+      "loss": 0.0063,
+      "step": 27760
+    },
+    {
+      "grad_norm": 0.19129633903503418,
+      "learning_rate": 4.4985204511365556e-05,
+      "loss": 0.0053,
+      "step": 27770
+    },
+    {
+      "grad_norm": 0.1916586011648178,
+      "learning_rate": 4.49523029616162e-05,
+      "loss": 0.0075,
+      "step": 27780
+    },
+    {
+      "grad_norm": 0.1456073671579361,
+      "learning_rate": 4.491940361990101e-05,
+      "loss": 0.0065,
+      "step": 27790
+    },
+    {
+      "grad_norm": 0.14971190690994263,
+      "learning_rate": 4.48865065006113e-05,
+      "loss": 0.0075,
+      "step": 27800
+    },
+    {
+      "grad_norm": 0.13580740988254547,
+      "learning_rate": 4.485361161813735e-05,
+      "loss": 0.0054,
+      "step": 27810
+    },
+    {
+      "grad_norm": 0.14781257510185242,
+      "learning_rate": 4.482071898686854e-05,
+      "loss": 0.0066,
+      "step": 27820
+    },
+    {
+      "grad_norm": 0.18319515883922577,
+      "learning_rate": 4.478782862119321e-05,
+      "loss": 0.0065,
+      "step": 27830
+    },
+    {
+      "grad_norm": 0.11033320426940918,
+      "learning_rate": 4.475494053549869e-05,
+      "loss": 0.0045,
+      "step": 27840
+    },
+    {
+      "grad_norm": 0.15768934786319733,
+      "learning_rate": 4.4722054744171405e-05,
+      "loss": 0.005,
+      "step": 27850
+    },
+    {
+      "grad_norm": 0.15933315455913544,
+      "learning_rate": 4.468917126159669e-05,
+      "loss": 0.0054,
+      "step": 27860
+    },
+    {
+      "grad_norm": 0.1657339334487915,
+      "learning_rate": 4.465629010215887e-05,
+      "loss": 0.0059,
+      "step": 27870
+    },
+    {
+      "grad_norm": 0.08506260067224503,
+      "learning_rate": 4.462341128024133e-05,
+      "loss": 0.0079,
+      "step": 27880
+    },
+    {
+      "grad_norm": 0.12905411422252655,
+      "learning_rate": 4.4590534810226356e-05,
+      "loss": 0.0066,
+      "step": 27890
+    },
+    {
+      "grad_norm": 0.18295733630657196,
+      "learning_rate": 4.455766070649523e-05,
+      "loss": 0.0071,
+      "step": 27900
+    },
+    {
+      "grad_norm": 0.16971974074840546,
+      "learning_rate": 4.4524788983428215e-05,
+      "loss": 0.0082,
+      "step": 27910
+    },
+    {
+      "grad_norm": 0.1897502988576889,
+      "learning_rate": 4.4491919655404515e-05,
+      "loss": 0.0083,
+      "step": 27920
+    },
+    {
+      "grad_norm": 0.16306188702583313,
+      "learning_rate": 4.445905273680228e-05,
+      "loss": 0.0053,
+      "step": 27930
+    },
+    {
+      "grad_norm": 0.1332821398973465,
+      "learning_rate": 4.442618824199865e-05,
+      "loss": 0.0064,
+      "step": 27940
+    },
+    {
+      "grad_norm": 0.16654619574546814,
+      "learning_rate": 4.4393326185369635e-05,
+      "loss": 0.0055,
+      "step": 27950
+    },
+    {
+      "grad_norm": 0.16213950514793396,
+      "learning_rate": 4.436046658129024e-05,
+      "loss": 0.0068,
+      "step": 27960
+    },
+    {
+      "grad_norm": 0.2130662500858307,
+      "learning_rate": 4.432760944413433e-05,
+      "loss": 0.0049,
+      "step": 27970
+    },
+    {
+      "grad_norm": 0.1736857295036316,
+      "learning_rate": 4.4294754788274796e-05,
+      "loss": 0.0058,
+      "step": 27980
+    },
+    {
+      "grad_norm": 0.16682590544223785,
+      "learning_rate": 4.426190262808334e-05,
+      "loss": 0.0085,
+      "step": 27990
+    },
+    {
+      "grad_norm": 0.13973276317119598,
+      "learning_rate": 4.42290529779306e-05,
+      "loss": 0.0069,
+      "step": 28000
+    },
+    {
+      "grad_norm": 0.17921367287635803,
+      "learning_rate": 4.4196205852186176e-05,
+      "loss": 0.0068,
+      "step": 28010
+    },
+    {
+      "grad_norm": 0.13637299835681915,
+      "learning_rate": 4.416336126521848e-05,
+      "loss": 0.0071,
+      "step": 28020
+    },
+    {
+      "grad_norm": 0.14907142519950867,
+      "learning_rate": 4.4130519231394865e-05,
+      "loss": 0.0055,
+      "step": 28030
+    },
+    {
+      "grad_norm": 0.17658200860023499,
+      "learning_rate": 4.409767976508154e-05,
+      "loss": 0.0074,
+      "step": 28040
+    },
+    {
+      "grad_norm": 0.17082786560058594,
+      "learning_rate": 4.406484288064363e-05,
+      "loss": 0.0062,
+      "step": 28050
+    },
+    {
+      "grad_norm": 0.15755711495876312,
+      "learning_rate": 4.4032008592445055e-05,
+      "loss": 0.0076,
+      "step": 28060
+    },
+    {
+      "grad_norm": 0.13810834288597107,
+      "learning_rate": 4.3999176914848714e-05,
+      "loss": 0.0054,
+      "step": 28070
+    },
+    {
+      "grad_norm": 0.16267800331115723,
+      "learning_rate": 4.396634786221625e-05,
+      "loss": 0.0074,
+      "step": 28080
+    },
+    {
+      "grad_norm": 0.1485196202993393,
+      "learning_rate": 4.3933521448908203e-05,
+      "loss": 0.007,
+      "step": 28090
+    },
+    {
+      "grad_norm": 0.1359751969575882,
+      "learning_rate": 4.3900697689284e-05,
+      "loss": 0.0053,
+      "step": 28100
+    },
+    {
+      "grad_norm": 0.1434536725282669,
+      "learning_rate": 4.386787659770185e-05,
+      "loss": 0.0078,
+      "step": 28110
+    },
+    {
+      "grad_norm": 0.1730109453201294,
+      "learning_rate": 4.383505818851878e-05,
+      "loss": 0.0064,
+      "step": 28120
+    },
+    {
+      "grad_norm": 0.12611594796180725,
+      "learning_rate": 4.380224247609073e-05,
+      "loss": 0.0069,
+      "step": 28130
+    },
+    {
+      "grad_norm": 0.13033224642276764,
+      "learning_rate": 4.3769429474772376e-05,
+      "loss": 0.0071,
+      "step": 28140
+    },
+    {
+      "grad_norm": 0.19640392065048218,
+      "learning_rate": 4.373661919891726e-05,
+      "loss": 0.007,
+      "step": 28150
+    },
+    {
+      "grad_norm": 0.1639072448015213,
+      "learning_rate": 4.3703811662877674e-05,
+      "loss": 0.0058,
+      "step": 28160
+    },
+    {
+      "grad_norm": 0.1846356987953186,
+      "learning_rate": 4.36710068810048e-05,
+      "loss": 0.0059,
+      "step": 28170
+    },
+    {
+      "grad_norm": 0.17013250291347504,
+      "learning_rate": 4.363820486764853e-05,
+      "loss": 0.007,
+      "step": 28180
+    },
+    {
+      "grad_norm": 0.19539472460746765,
+      "learning_rate": 4.360540563715758e-05,
+      "loss": 0.0062,
+      "step": 28190
+    },
+    {
+      "grad_norm": 0.09512146562337875,
+      "learning_rate": 4.357260920387946e-05,
+      "loss": 0.0044,
+      "step": 28200
+    },
+    {
+      "grad_norm": 0.12211081385612488,
+      "learning_rate": 4.353981558216045e-05,
+      "loss": 0.0053,
+      "step": 28210
+    },
+    {
+      "grad_norm": 0.11235443502664566,
+      "learning_rate": 4.3507024786345565e-05,
+      "loss": 0.0051,
+      "step": 28220
+    },
+    {
+      "grad_norm": 0.12191634625196457,
+      "learning_rate": 4.3474236830778646e-05,
+      "loss": 0.0047,
+      "step": 28230
+    },
+    {
+      "grad_norm": 0.12450134009122849,
+      "learning_rate": 4.344145172980223e-05,
+      "loss": 0.0064,
+      "step": 28240
+    },
+    {
+      "grad_norm": 0.11292971670627594,
+      "learning_rate": 4.3408669497757656e-05,
+      "loss": 0.0075,
+      "step": 28250
+    },
+    {
+      "grad_norm": 0.24386554956436157,
+      "learning_rate": 4.3375890148984976e-05,
+      "loss": 0.0081,
+      "step": 28260
+    },
+    {
+      "grad_norm": 0.11927863955497742,
+      "learning_rate": 4.334311369782301e-05,
+      "loss": 0.0071,
+      "step": 28270
+    },
+    {
+      "grad_norm": 0.12093808501958847,
+      "learning_rate": 4.3310340158609244e-05,
+      "loss": 0.005,
+      "step": 28280
+    },
+    {
+      "grad_norm": 0.13399194180965424,
+      "learning_rate": 4.327756954568e-05,
+      "loss": 0.0058,
+      "step": 28290
+    },
+    {
+      "grad_norm": 0.1336631178855896,
+      "learning_rate": 4.324480187337021e-05,
+      "loss": 0.0048,
+      "step": 28300
+    },
+    {
+      "grad_norm": 0.185091033577919,
+      "learning_rate": 4.3212037156013596e-05,
+      "loss": 0.0067,
+      "step": 28310
+    },
+    {
+      "grad_norm": 0.14208205044269562,
+      "learning_rate": 4.317927540794251e-05,
+      "loss": 0.0056,
+      "step": 28320
+    },
+    {
+      "grad_norm": 0.13566148281097412,
+      "learning_rate": 4.314651664348812e-05,
+      "loss": 0.005,
+      "step": 28330
+    },
+    {
+      "grad_norm": 0.17807099223136902,
+      "learning_rate": 4.3113760876980195e-05,
+      "loss": 0.0062,
+      "step": 28340
+    },
+    {
+      "grad_norm": 0.15549102425575256,
+      "learning_rate": 4.308100812274719e-05,
+      "loss": 0.0051,
+      "step": 28350
+    },
+    {
+      "grad_norm": 0.1781071424484253,
+      "learning_rate": 4.304825839511632e-05,
+      "loss": 0.0063,
+      "step": 28360
+    },
+    {
+      "grad_norm": 0.15796102583408356,
+      "learning_rate": 4.3015511708413395e-05,
+      "loss": 0.0053,
+      "step": 28370
+    },
+    {
+      "grad_norm": 0.1707073301076889,
+      "learning_rate": 4.298276807696294e-05,
+      "loss": 0.0058,
+      "step": 28380
+    },
+    {
+      "grad_norm": 0.15881045162677765,
+      "learning_rate": 4.2950027515088136e-05,
+      "loss": 0.0069,
+      "step": 28390
+    },
+    {
+      "grad_norm": 0.07840579003095627,
+      "learning_rate": 4.291729003711083e-05,
+      "loss": 0.0049,
+      "step": 28400
+    },
+    {
+      "grad_norm": 0.1541912853717804,
+      "learning_rate": 4.2884555657351455e-05,
+      "loss": 0.0051,
+      "step": 28410
+    },
+    {
+      "grad_norm": 0.1011442020535469,
+      "learning_rate": 4.285182439012921e-05,
+      "loss": 0.005,
+      "step": 28420
+    },
+    {
+      "grad_norm": 0.12476082146167755,
+      "learning_rate": 4.281909624976183e-05,
+      "loss": 0.0052,
+      "step": 28430
+    },
+    {
+      "grad_norm": 0.14587357640266418,
+      "learning_rate": 4.278637125056568e-05,
+      "loss": 0.0047,
+      "step": 28440
+    },
+    {
+      "grad_norm": 0.17515668272972107,
+      "learning_rate": 4.2753649406855855e-05,
+      "loss": 0.0053,
+      "step": 28450
+    },
+    {
+      "grad_norm": 0.15305617451667786,
+      "learning_rate": 4.2720930732945966e-05,
+      "loss": 0.0047,
+      "step": 28460
+    },
+    {
+      "grad_norm": 0.13397659361362457,
+      "learning_rate": 4.268821524314826e-05,
+      "loss": 0.0063,
+      "step": 28470
+    },
+    {
+      "grad_norm": 0.153980553150177,
+      "learning_rate": 4.2655502951773623e-05,
+      "loss": 0.006,
+      "step": 28480
+    },
+    {
+      "grad_norm": 0.13988228142261505,
+      "learning_rate": 4.262279387313152e-05,
+      "loss": 0.005,
+      "step": 28490
+    },
+    {
+      "grad_norm": 0.17870433628559113,
+      "learning_rate": 4.259008802153001e-05,
+      "loss": 0.0071,
+      "step": 28500
+    },
+    {
+      "grad_norm": 0.12915830314159393,
+      "learning_rate": 4.2557385411275735e-05,
+      "loss": 0.0057,
+      "step": 28510
+    },
+    {
+      "grad_norm": 0.14032547175884247,
+      "learning_rate": 4.252468605667395e-05,
+      "loss": 0.0056,
+      "step": 28520
+    },
+    {
+      "grad_norm": 0.09860539436340332,
+      "learning_rate": 4.2491989972028456e-05,
+      "loss": 0.0051,
+      "step": 28530
+    },
+    {
+      "grad_norm": 0.11329696327447891,
+      "learning_rate": 4.245929717164161e-05,
+      "loss": 0.0058,
+      "step": 28540
+    },
+    {
+      "grad_norm": 0.09999416023492813,
+      "learning_rate": 4.242660766981438e-05,
+      "loss": 0.0048,
+      "step": 28550
+    },
+    {
+      "grad_norm": 0.14410723745822906,
+      "learning_rate": 4.2393921480846264e-05,
+      "loss": 0.0058,
+      "step": 28560
+    },
+    {
+      "grad_norm": 0.15088805556297302,
+      "learning_rate": 4.236123861903528e-05,
+      "loss": 0.0071,
+      "step": 28570
+    },
+    {
+      "grad_norm": 0.26417627930641174,
+      "learning_rate": 4.232855909867806e-05,
+      "loss": 0.0082,
+      "step": 28580
+    },
+    {
+      "grad_norm": 0.2080899029970169,
+      "learning_rate": 4.229588293406972e-05,
+      "loss": 0.0054,
+      "step": 28590
+    },
+    {
+      "grad_norm": 0.16379490494728088,
+      "learning_rate": 4.226321013950392e-05,
+      "loss": 0.0071,
+      "step": 28600
+    },
+    {
+      "grad_norm": 0.1575908064842224,
+      "learning_rate": 4.223054072927286e-05,
+      "loss": 0.0049,
+      "step": 28610
+    },
+    {
+      "grad_norm": 0.1031169444322586,
+      "learning_rate": 4.219787471766724e-05,
+      "loss": 0.0058,
+      "step": 28620
+    },
+    {
+      "grad_norm": 0.1768019050359726,
+      "learning_rate": 4.216521211897629e-05,
+      "loss": 0.0067,
+      "step": 28630
+    },
+    {
+      "grad_norm": 0.1587243229150772,
+      "learning_rate": 4.2132552947487734e-05,
+      "loss": 0.0056,
+      "step": 28640
+    },
+    {
+      "grad_norm": 0.15130484104156494,
+      "learning_rate": 4.209989721748783e-05,
+      "loss": 0.0052,
+      "step": 28650
+    },
+    {
+      "grad_norm": 0.11827655136585236,
+      "learning_rate": 4.206724494326127e-05,
+      "loss": 0.0054,
+      "step": 28660
+    },
+    {
+      "grad_norm": 0.25572308897972107,
+      "learning_rate": 4.203459613909126e-05,
+      "loss": 0.0074,
+      "step": 28670
+    },
+    {
+      "grad_norm": 0.15429086983203888,
+      "learning_rate": 4.200195081925955e-05,
+      "loss": 0.0048,
+      "step": 28680
+    },
+    {
+      "grad_norm": 0.14072246849536896,
+      "learning_rate": 4.196930899804627e-05,
+      "loss": 0.0067,
+      "step": 28690
+    },
+    {
+      "grad_norm": 0.15622468292713165,
+      "learning_rate": 4.1936670689730044e-05,
+      "loss": 0.0052,
+      "step": 28700
+    },
+    {
+      "grad_norm": 0.17125050723552704,
+      "learning_rate": 4.190403590858804e-05,
+      "loss": 0.0054,
+      "step": 28710
+    },
+    {
+      "grad_norm": 0.1721969097852707,
+      "learning_rate": 4.187140466889575e-05,
+      "loss": 0.0056,
+      "step": 28720
+    },
+    {
+      "grad_norm": 0.1251361221075058,
+      "learning_rate": 4.183877698492724e-05,
+      "loss": 0.005,
+      "step": 28730
+    },
+    {
+      "grad_norm": 0.1574040651321411,
+      "learning_rate": 4.1806152870954935e-05,
+      "loss": 0.0061,
+      "step": 28740
+    },
+    {
+      "grad_norm": 0.15340422093868256,
+      "learning_rate": 4.177353234124976e-05,
+      "loss": 0.007,
+      "step": 28750
+    },
+    {
+      "grad_norm": 0.1287044733762741,
+      "learning_rate": 4.1740915410081e-05,
+      "loss": 0.0074,
+      "step": 28760
+    },
+    {
+      "grad_norm": 0.13760371506214142,
+      "learning_rate": 4.170830209171646e-05,
+      "loss": 0.0054,
+      "step": 28770
+    },
+    {
+      "grad_norm": 0.15677613019943237,
+      "learning_rate": 4.1675692400422286e-05,
+      "loss": 0.0062,
+      "step": 28780
+    },
+    {
+      "grad_norm": 0.13382942974567413,
+      "learning_rate": 4.164308635046304e-05,
+      "loss": 0.0053,
+      "step": 28790
+    },
+    {
+      "grad_norm": 0.1288585215806961,
+      "learning_rate": 4.161048395610177e-05,
+      "loss": 0.0052,
+      "step": 28800
+    },
+    {
+      "grad_norm": 0.14930295944213867,
+      "learning_rate": 4.157788523159985e-05,
+      "loss": 0.007,
+      "step": 28810
+    },
+    {
+      "grad_norm": 0.1510496437549591,
+      "learning_rate": 4.154529019121706e-05,
+      "loss": 0.007,
+      "step": 28820
+    },
+    {
+      "grad_norm": 0.17506738007068634,
+      "learning_rate": 4.1512698849211564e-05,
+      "loss": 0.0053,
+      "step": 28830
+    },
+    {
+      "grad_norm": 0.14016678929328918,
+      "learning_rate": 4.1480111219839954e-05,
+      "loss": 0.0058,
+      "step": 28840
+    },
+    {
+      "grad_norm": 0.14207033812999725,
+      "learning_rate": 4.144752731735714e-05,
+      "loss": 0.0068,
+      "step": 28850
+    },
+    {
+      "grad_norm": 0.17177295684814453,
+      "learning_rate": 4.1414947156016446e-05,
+      "loss": 0.0064,
+      "step": 28860
+    },
+    {
+      "grad_norm": 0.17018017172813416,
+      "learning_rate": 4.138237075006953e-05,
+      "loss": 0.0079,
+      "step": 28870
+    },
+    {
+      "grad_norm": 0.17467808723449707,
+      "learning_rate": 4.134979811376645e-05,
+      "loss": 0.0073,
+      "step": 28880
+    },
+    {
+      "grad_norm": 0.12267519533634186,
+      "learning_rate": 4.131722926135552e-05,
+      "loss": 0.0047,
+      "step": 28890
+    },
+    {
+      "grad_norm": 0.1595081388950348,
+      "learning_rate": 4.1284664207083526e-05,
+      "loss": 0.0064,
+      "step": 28900
+    },
+    {
+      "grad_norm": 0.16115501523017883,
+      "learning_rate": 4.125210296519551e-05,
+      "loss": 0.0079,
+      "step": 28910
+    },
+    {
+      "grad_norm": 0.12851892411708832,
+      "learning_rate": 4.121954554993484e-05,
+      "loss": 0.0054,
+      "step": 28920
+    },
+    {
+      "grad_norm": 0.16341900825500488,
+      "learning_rate": 4.118699197554327e-05,
+      "loss": 0.0061,
+      "step": 28930
+    },
+    {
+      "grad_norm": 0.13245634734630585,
+      "learning_rate": 4.1154442256260814e-05,
+      "loss": 0.0052,
+      "step": 28940
+    },
+    {
+      "grad_norm": 0.16142085194587708,
+      "learning_rate": 4.1121896406325824e-05,
+      "loss": 0.0069,
+      "step": 28950
+    },
+    {
+      "grad_norm": 0.13512933254241943,
+      "learning_rate": 4.108935443997499e-05,
+      "loss": 0.0073,
+      "step": 28960
+    },
+    {
+      "grad_norm": 0.16432897746562958,
+      "learning_rate": 4.105681637144325e-05,
+      "loss": 0.0062,
+      "step": 28970
+    },
+    {
+      "grad_norm": 0.15161867439746857,
+      "learning_rate": 4.102428221496387e-05,
+      "loss": 0.0054,
+      "step": 28980
+    },
+    {
+      "grad_norm": 0.14184993505477905,
+      "learning_rate": 4.099175198476838e-05,
+      "loss": 0.006,
+      "step": 28990
+    },
+    {
+      "grad_norm": 0.15722866356372833,
+      "learning_rate": 4.0959225695086635e-05,
+      "loss": 0.0053,
+      "step": 29000
+    },
+    {
+      "grad_norm": 0.11333426833152771,
+      "learning_rate": 4.0926703360146724e-05,
+      "loss": 0.0063,
+      "step": 29010
+    },
+    {
+      "grad_norm": 0.16640397906303406,
+      "learning_rate": 4.0894184994175e-05,
+      "loss": 0.0056,
+      "step": 29020
+    },
+    {
+      "grad_norm": 0.1331067532300949,
+      "learning_rate": 4.086167061139614e-05,
+      "loss": 0.0054,
+      "step": 29030
+    },
+    {
+      "grad_norm": 0.17902441322803497,
+      "learning_rate": 4.082916022603303e-05,
+      "loss": 0.0061,
+      "step": 29040
+    },
+    {
+      "grad_norm": 0.12817035615444183,
+      "learning_rate": 4.079665385230678e-05,
+      "loss": 0.0068,
+      "step": 29050
+    },
+    {
+      "grad_norm": 0.16196392476558685,
+      "learning_rate": 4.076415150443683e-05,
+      "loss": 0.0069,
+      "step": 29060
+    },
+    {
+      "grad_norm": 0.15724577009677887,
+      "learning_rate": 4.073165319664079e-05,
+      "loss": 0.0061,
+      "step": 29070
+    },
+    {
+      "grad_norm": 0.14044761657714844,
+      "learning_rate": 4.069915894313451e-05,
+      "loss": 0.0069,
+      "step": 29080
+    },
+    {
+      "grad_norm": 0.1378507763147354,
+      "learning_rate": 4.066666875813212e-05,
+      "loss": 0.005,
+      "step": 29090
+    },
+    {
+      "grad_norm": 0.11229481548070908,
+      "learning_rate": 4.063418265584589e-05,
+      "loss": 0.0059,
+      "step": 29100
+    },
+    {
+      "grad_norm": 0.12323877960443497,
+      "learning_rate": 4.060170065048636e-05,
+      "loss": 0.0059,
+      "step": 29110
+    },
+    {
+      "grad_norm": 0.164375439286232,
+      "learning_rate": 4.056922275626227e-05,
+      "loss": 0.0065,
+      "step": 29120
+    },
+    {
+      "grad_norm": 0.09407272189855576,
+      "learning_rate": 4.053674898738056e-05,
+      "loss": 0.006,
+      "step": 29130
+    },
+    {
+      "grad_norm": 0.13087135553359985,
+      "learning_rate": 4.050427935804633e-05,
+      "loss": 0.0068,
+      "step": 29140
+    },
+    {
+      "grad_norm": 0.15250743925571442,
+      "learning_rate": 4.047181388246293e-05,
+      "loss": 0.0068,
+      "step": 29150
+    },
+    {
+      "grad_norm": 0.250069797039032,
+      "learning_rate": 4.043935257483185e-05,
+      "loss": 0.0068,
+      "step": 29160
+    },
+    {
+      "grad_norm": 0.17607218027114868,
+      "learning_rate": 4.0406895449352775e-05,
+      "loss": 0.0065,
+      "step": 29170
+    },
+    {
+      "grad_norm": 0.1528383195400238,
+      "learning_rate": 4.0374442520223525e-05,
+      "loss": 0.0067,
+      "step": 29180
+    },
+    {
+      "grad_norm": 0.15601244568824768,
+      "learning_rate": 4.0341993801640155e-05,
+      "loss": 0.0056,
+      "step": 29190
+    },
+    {
+      "grad_norm": 0.1280871480703354,
+      "learning_rate": 4.030954930779681e-05,
+      "loss": 0.0047,
+      "step": 29200
+    },
+    {
+      "grad_norm": 0.139277845621109,
+      "learning_rate": 4.027710905288582e-05,
+      "loss": 0.0072,
+      "step": 29210
+    },
+    {
+      "grad_norm": 0.18452656269073486,
+      "learning_rate": 4.0244673051097666e-05,
+      "loss": 0.0056,
+      "step": 29220
+    },
+    {
+      "grad_norm": 0.16151019930839539,
+      "learning_rate": 4.021224131662095e-05,
+      "loss": 0.0056,
+      "step": 29230
+    },
+    {
+      "grad_norm": 0.11940456181764603,
+      "learning_rate": 4.0179813863642396e-05,
+      "loss": 0.0066,
+      "step": 29240
+    },
+    {
+      "grad_norm": 0.14150172472000122,
+      "learning_rate": 4.014739070634691e-05,
+      "loss": 0.005,
+      "step": 29250
+    },
+    {
+      "grad_norm": 0.17309872806072235,
+      "learning_rate": 4.011497185891746e-05,
+      "loss": 0.0067,
+      "step": 29260
+    },
+    {
+      "grad_norm": 0.1398734748363495,
+      "learning_rate": 4.008255733553513e-05,
+      "loss": 0.0063,
+      "step": 29270
+    },
+    {
+      "grad_norm": 0.14527831971645355,
+      "learning_rate": 4.0050147150379194e-05,
+      "loss": 0.0062,
+      "step": 29280
+    },
+    {
+      "grad_norm": 0.10920097678899765,
+      "learning_rate": 4.001774131762692e-05,
+      "loss": 0.0065,
+      "step": 29290
+    },
+    {
+      "grad_norm": 0.1077660620212555,
+      "learning_rate": 3.9985339851453705e-05,
+      "loss": 0.0056,
+      "step": 29300
+    },
+    {
+      "grad_norm": 0.14082522690296173,
+      "learning_rate": 3.9952942766033114e-05,
+      "loss": 0.005,
+      "step": 29310
+    },
+    {
+      "grad_norm": 0.13790066540241241,
+      "learning_rate": 3.9920550075536686e-05,
+      "loss": 0.0067,
+      "step": 29320
+    },
+    {
+      "grad_norm": 0.0982794463634491,
+      "learning_rate": 3.988816179413408e-05,
+      "loss": 0.0049,
+      "step": 29330
+    },
+    {
+      "grad_norm": 0.10597933828830719,
+      "learning_rate": 3.985577793599307e-05,
+      "loss": 0.007,
+      "step": 29340
+    },
+    {
+      "grad_norm": 0.12322115898132324,
+      "learning_rate": 3.982339851527942e-05,
+      "loss": 0.007,
+      "step": 29350
+    },
+    {
+      "grad_norm": 0.1776578426361084,
+      "learning_rate": 3.979102354615701e-05,
+      "loss": 0.0053,
+      "step": 29360
+    },
+    {
+      "grad_norm": 0.1358364373445511,
+      "learning_rate": 3.975865304278772e-05,
+      "loss": 0.005,
+      "step": 29370
+    },
+    {
+      "grad_norm": 0.19222579896450043,
+      "learning_rate": 3.9726287019331546e-05,
+      "loss": 0.0049,
+      "step": 29380
+    },
+    {
+      "grad_norm": 0.12444644421339035,
+      "learning_rate": 3.9693925489946474e-05,
+      "loss": 0.0061,
+      "step": 29390
+    },
+    {
+      "grad_norm": 0.1582004725933075,
+      "learning_rate": 3.966156846878852e-05,
+      "loss": 0.0051,
+      "step": 29400
+    },
+    {
+      "grad_norm": 0.1372964233160019,
+      "learning_rate": 3.962921597001177e-05,
+      "loss": 0.0058,
+      "step": 29410
+    },
+    {
+      "grad_norm": 0.13483117520809174,
+      "learning_rate": 3.9596868007768294e-05,
+      "loss": 0.0057,
+      "step": 29420
+    },
+    {
+      "grad_norm": 0.16220125555992126,
+      "learning_rate": 3.9564524596208165e-05,
+      "loss": 0.0082,
+      "step": 29430
+    },
+    {
+      "grad_norm": 0.0958242267370224,
+      "learning_rate": 3.953218574947954e-05,
+      "loss": 0.0042,
+      "step": 29440
+    },
+    {
+      "grad_norm": 0.1599251627922058,
+      "learning_rate": 3.949985148172849e-05,
+      "loss": 0.0048,
+      "step": 29450
+    },
+    {
+      "grad_norm": 0.1284315586090088,
+      "learning_rate": 3.946752180709914e-05,
+      "loss": 0.0072,
+      "step": 29460
+    },
+    {
+      "grad_norm": 0.10999341309070587,
+      "learning_rate": 3.943519673973357e-05,
+      "loss": 0.0066,
+      "step": 29470
+    },
+    {
+      "grad_norm": 0.1229429841041565,
+      "learning_rate": 3.940287629377189e-05,
+      "loss": 0.0055,
+      "step": 29480
+    },
+    {
+      "grad_norm": 0.18364420533180237,
+      "learning_rate": 3.937056048335214e-05,
+      "loss": 0.0061,
+      "step": 29490
+    },
+    {
+      "grad_norm": 0.12222758680582047,
+      "learning_rate": 3.933824932261037e-05,
+      "loss": 0.0066,
+      "step": 29500
+    },
+    {
+      "grad_norm": 0.16719166934490204,
+      "learning_rate": 3.9305942825680576e-05,
+      "loss": 0.0057,
+      "step": 29510
+    },
+    {
+      "grad_norm": 0.136664479970932,
+      "learning_rate": 3.9273641006694706e-05,
+      "loss": 0.0063,
+      "step": 29520
+    },
+    {
+      "grad_norm": 0.0817146897315979,
+      "learning_rate": 3.924134387978266e-05,
+      "loss": 0.0048,
+      "step": 29530
+    },
+    {
+      "grad_norm": 0.1398315727710724,
+      "learning_rate": 3.920905145907234e-05,
+      "loss": 0.0053,
+      "step": 29540
+    },
+    {
+      "grad_norm": 0.12767712771892548,
+      "learning_rate": 3.9176763758689525e-05,
+      "loss": 0.0048,
+      "step": 29550
+    },
+    {
+      "grad_norm": 0.158310666680336,
+      "learning_rate": 3.9144480792757925e-05,
+      "loss": 0.0053,
+      "step": 29560
+    },
+    {
+      "grad_norm": 0.15345335006713867,
+      "learning_rate": 3.911220257539926e-05,
+      "loss": 0.0051,
+      "step": 29570
+    },
+    {
+      "grad_norm": 0.11150381714105606,
+      "learning_rate": 3.907992912073308e-05,
+      "loss": 0.0078,
+      "step": 29580
+    },
+    {
+      "grad_norm": 0.11185961961746216,
+      "learning_rate": 3.90476604428769e-05,
+      "loss": 0.0051,
+      "step": 29590
+    },
+    {
+      "grad_norm": 0.18218621611595154,
+      "learning_rate": 3.901539655594614e-05,
+      "loss": 0.0052,
+      "step": 29600
+    },
+    {
+      "grad_norm": 0.10677412152290344,
+      "learning_rate": 3.898313747405413e-05,
+      "loss": 0.0051,
+      "step": 29610
+    },
+    {
+      "grad_norm": 0.13702310621738434,
+      "learning_rate": 3.895088321131205e-05,
+      "loss": 0.0056,
+      "step": 29620
+    },
+    {
+      "grad_norm": 0.1844206303358078,
+      "learning_rate": 3.891863378182907e-05,
+      "loss": 0.0077,
+      "step": 29630
+    },
+    {
+      "grad_norm": 0.14507457613945007,
+      "learning_rate": 3.8886389199712145e-05,
+      "loss": 0.0066,
+      "step": 29640
+    },
+    {
+      "grad_norm": 0.17167168855667114,
+      "learning_rate": 3.885414947906614e-05,
+      "loss": 0.0052,
+      "step": 29650
+    },
+    {
+      "grad_norm": 0.13174046576023102,
+      "learning_rate": 3.882191463399385e-05,
+      "loss": 0.0051,
+      "step": 29660
+    },
+    {
+      "grad_norm": 0.1265619695186615,
+      "learning_rate": 3.878968467859585e-05,
+      "loss": 0.0046,
+      "step": 29670
+    },
+    {
+      "grad_norm": 0.14203760027885437,
+      "learning_rate": 3.875745962697063e-05,
+      "loss": 0.0051,
+      "step": 29680
+    },
+    {
+      "grad_norm": 0.12139276415109634,
+      "learning_rate": 3.872523949321454e-05,
+      "loss": 0.0057,
+      "step": 29690
+    },
+    {
+      "grad_norm": 0.12147907167673111,
+      "learning_rate": 3.869302429142173e-05,
+      "loss": 0.0066,
+      "step": 29700
+    },
+    {
+      "grad_norm": 0.17265987396240234,
+      "learning_rate": 3.866081403568425e-05,
+      "loss": 0.0059,
+      "step": 29710
+    },
+    {
+      "grad_norm": 0.1543305218219757,
+      "learning_rate": 3.8628608740091934e-05,
+      "loss": 0.0048,
+      "step": 29720
+    },
+    {
+      "grad_norm": 0.12410735338926315,
+      "learning_rate": 3.85964084187325e-05,
+      "loss": 0.0051,
+      "step": 29730
+    },
+    {
+      "grad_norm": 0.11922281980514526,
+      "learning_rate": 3.856421308569146e-05,
+      "loss": 0.0046,
+      "step": 29740
+    },
+    {
+      "grad_norm": 0.1856105774641037,
+      "learning_rate": 3.853202275505212e-05,
+      "loss": 0.0059,
+      "step": 29750
+    },
+    {
+      "grad_norm": 0.14741401374340057,
+      "learning_rate": 3.849983744089565e-05,
+      "loss": 0.0055,
+      "step": 29760
+    },
+    {
+      "grad_norm": 0.14657950401306152,
+      "learning_rate": 3.8467657157301e-05,
+      "loss": 0.0055,
+      "step": 29770
+    },
+    {
+      "grad_norm": 0.12059198319911957,
+      "learning_rate": 3.8435481918344884e-05,
+      "loss": 0.0067,
+      "step": 29780
+    },
+    {
+      "grad_norm": 0.13468307256698608,
+      "learning_rate": 3.84033117381019e-05,
+      "loss": 0.0051,
+      "step": 29790
+    },
+    {
+      "grad_norm": 0.11327297985553741,
+      "learning_rate": 3.8371146630644335e-05,
+      "loss": 0.0061,
+      "step": 29800
+    },
+    {
+      "grad_norm": 0.12961934506893158,
+      "learning_rate": 3.833898661004231e-05,
+      "loss": 0.0071,
+      "step": 29810
+    },
+    {
+      "grad_norm": 0.19903548061847687,
+      "learning_rate": 3.830683169036372e-05,
+      "loss": 0.0053,
+      "step": 29820
+    },
+    {
+      "grad_norm": 0.14746278524398804,
+      "learning_rate": 3.827468188567422e-05,
+      "loss": 0.0051,
+      "step": 29830
+    },
+    {
+      "grad_norm": 0.12346802651882172,
+      "learning_rate": 3.82425372100372e-05,
+      "loss": 0.0061,
+      "step": 29840
+    },
+    {
+      "grad_norm": 0.12555962800979614,
+      "learning_rate": 3.821039767751387e-05,
+      "loss": 0.0066,
+      "step": 29850
+    },
+    {
+      "grad_norm": 0.16442446410655975,
+      "learning_rate": 3.817826330216315e-05,
+      "loss": 0.0062,
+      "step": 29860
+    },
+    {
+      "grad_norm": 0.1252536177635193,
+      "learning_rate": 3.814613409804169e-05,
+      "loss": 0.0057,
+      "step": 29870
+    },
+    {
+      "grad_norm": 0.12736551463603973,
+      "learning_rate": 3.811401007920388e-05,
+      "loss": 0.0063,
+      "step": 29880
+    },
+    {
+      "grad_norm": 0.14350371062755585,
+      "learning_rate": 3.8081891259701895e-05,
+      "loss": 0.007,
+      "step": 29890
+    },
+    {
+      "grad_norm": 0.11305675655603409,
+      "learning_rate": 3.8049777653585584e-05,
+      "loss": 0.005,
+      "step": 29900
+    },
+    {
+      "grad_norm": 0.13388659060001373,
+      "learning_rate": 3.80176692749025e-05,
+      "loss": 0.005,
+      "step": 29910
+    },
+    {
+      "grad_norm": 0.156807541847229,
+      "learning_rate": 3.798556613769799e-05,
+      "loss": 0.0069,
+      "step": 29920
+    },
+    {
+      "grad_norm": 0.19448842108249664,
+      "learning_rate": 3.795346825601502e-05,
+      "loss": 0.0071,
+      "step": 29930
+    },
+    {
+      "grad_norm": 0.1369461715221405,
+      "learning_rate": 3.7921375643894305e-05,
+      "loss": 0.0061,
+      "step": 29940
+    },
+    {
+      "grad_norm": 0.14745786786079407,
+      "learning_rate": 3.788928831537424e-05,
+      "loss": 0.0071,
+      "step": 29950
+    },
+    {
+      "grad_norm": 0.12970054149627686,
+      "learning_rate": 3.7857206284490934e-05,
+      "loss": 0.0045,
+      "step": 29960
+    },
+    {
+      "grad_norm": 0.2510617673397064,
+      "learning_rate": 3.7825129565278114e-05,
+      "loss": 0.0059,
+      "step": 29970
+    },
+    {
+      "grad_norm": 0.14364565908908844,
+      "learning_rate": 3.779305817176728e-05,
+      "loss": 0.0054,
+      "step": 29980
+    },
+    {
+      "grad_norm": 0.11153639107942581,
+      "learning_rate": 3.776099211798751e-05,
+      "loss": 0.0045,
+      "step": 29990
+    },
+    {
+      "grad_norm": 0.10727117210626602,
+      "learning_rate": 3.7728931417965584e-05,
+      "loss": 0.0046,
+      "step": 30000
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 50000,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 9223372036854775807,
+  "save_steps": 10000,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 0.0,
+  "train_batch_size": 64,
+  "trial_name": null,
+  "trial_params": null
+}