diff --git "a/vlmfreeze_loc_h20_15k/trainer_state.json" "b/vlmfreeze_loc_h20_15k/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/vlmfreeze_loc_h20_15k/trainer_state.json"
@@ -0,0 +1,9034 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.75,
+  "eval_steps": 500,
+  "global_step": 15000,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "grad_norm": 0.36992350220680237,
+      "learning_rate": 9e-07,
+      "loss": 0.3833,
+      "step": 10
+    },
+    {
+      "grad_norm": 0.5050746202468872,
+      "learning_rate": 1.9e-06,
+      "loss": 0.3909,
+      "step": 20
+    },
+    {
+      "grad_norm": 0.650155246257782,
+      "learning_rate": 2.9e-06,
+      "loss": 0.3794,
+      "step": 30
+    },
+    {
+      "grad_norm": 0.4079556167125702,
+      "learning_rate": 3.9e-06,
+      "loss": 0.3782,
+      "step": 40
+    },
+    {
+      "grad_norm": 0.3647061884403229,
+      "learning_rate": 4.9000000000000005e-06,
+      "loss": 0.3768,
+      "step": 50
+    },
+    {
+      "grad_norm": 0.4319082200527191,
+      "learning_rate": 5.9e-06,
+      "loss": 0.3634,
+      "step": 60
+    },
+    {
+      "grad_norm": 0.38906344771385193,
+      "learning_rate": 6.900000000000001e-06,
+      "loss": 0.3529,
+      "step": 70
+    },
+    {
+      "grad_norm": 0.29221007227897644,
+      "learning_rate": 7.9e-06,
+      "loss": 0.3452,
+      "step": 80
+    },
+    {
+      "grad_norm": 0.3150082230567932,
+      "learning_rate": 8.9e-06,
+      "loss": 0.3296,
+      "step": 90
+    },
+    {
+      "grad_norm": 0.3151991665363312,
+      "learning_rate": 9.900000000000002e-06,
+      "loss": 0.3252,
+      "step": 100
+    },
+    {
+      "grad_norm": 0.3357653021812439,
+      "learning_rate": 1.09e-05,
+      "loss": 0.3282,
+      "step": 110
+    },
+    {
+      "grad_norm": 0.3207228481769562,
+      "learning_rate": 1.19e-05,
+      "loss": 0.3208,
+      "step": 120
+    },
+    {
+      "grad_norm": 0.39490988850593567,
+      "learning_rate": 1.29e-05,
+      "loss": 0.3231,
+      "step": 130
+    },
+    {
+      "grad_norm": 0.45401591062545776,
+      "learning_rate": 1.3900000000000002e-05,
+      "loss": 0.3247,
+      "step": 140
+    },
+    {
+      "grad_norm": 0.2971478998661041,
+      "learning_rate": 1.49e-05,
+      "loss": 0.3105,
+      "step": 150
+    },
+    {
+      "grad_norm": 0.270562082529068,
+      "learning_rate": 1.59e-05,
+      "loss": 0.2872,
+      "step": 160
+    },
+    {
+      "grad_norm": 0.3511049151420593,
+      "learning_rate": 1.69e-05,
+      "loss": 0.2909,
+      "step": 170
+    },
+    {
+      "grad_norm": 0.32625672221183777,
+      "learning_rate": 1.79e-05,
+      "loss": 0.2811,
+      "step": 180
+    },
+    {
+      "grad_norm": 0.37837666273117065,
+      "learning_rate": 1.8900000000000002e-05,
+      "loss": 0.2827,
+      "step": 190
+    },
+    {
+      "grad_norm": 0.34683090448379517,
+      "learning_rate": 1.9900000000000003e-05,
+      "loss": 0.2699,
+      "step": 200
+    },
+    {
+      "grad_norm": 0.3021489977836609,
+      "learning_rate": 2.09e-05,
+      "loss": 0.2709,
+      "step": 210
+    },
+    {
+      "grad_norm": 0.3923785388469696,
+      "learning_rate": 2.19e-05,
+      "loss": 0.2569,
+      "step": 220
+    },
+    {
+      "grad_norm": 0.32955387234687805,
+      "learning_rate": 2.29e-05,
+      "loss": 0.2733,
+      "step": 230
+    },
+    {
+      "grad_norm": 0.2845437526702881,
+      "learning_rate": 2.39e-05,
+      "loss": 0.2532,
+      "step": 240
+    },
+    {
+      "grad_norm": 0.3498857915401459,
+      "learning_rate": 2.4900000000000002e-05,
+      "loss": 0.2563,
+      "step": 250
+    },
+    {
+      "grad_norm": 0.3556017279624939,
+      "learning_rate": 2.5900000000000003e-05,
+      "loss": 0.2421,
+      "step": 260
+    },
+    {
+      "grad_norm": 0.3323434293270111,
+      "learning_rate": 2.6900000000000003e-05,
+      "loss": 0.244,
+      "step": 270
+    },
+    {
+      "grad_norm": 0.34393027424812317,
+      "learning_rate": 2.7900000000000004e-05,
+      "loss": 0.2291,
+      "step": 280
+    },
+    {
+      "grad_norm": 0.3392191231250763,
+      "learning_rate": 2.8899999999999998e-05,
+      "loss": 0.2431,
+      "step": 290
+    },
+    {
+      "grad_norm": 0.625434160232544,
+      "learning_rate": 2.9900000000000002e-05,
+      "loss": 0.2493,
+      "step": 300
+    },
+    {
+      "grad_norm": 0.45169949531555176,
+      "learning_rate": 3.09e-05,
+      "loss": 0.2478,
+      "step": 310
+    },
+    {
+      "grad_norm": 0.4545411467552185,
+      "learning_rate": 3.19e-05,
+      "loss": 0.2443,
+      "step": 320
+    },
+    {
+      "grad_norm": 0.35845717787742615,
+      "learning_rate": 3.29e-05,
+      "loss": 0.229,
+      "step": 330
+    },
+    {
+      "grad_norm": 0.5311976671218872,
+      "learning_rate": 3.3900000000000004e-05,
+      "loss": 0.2333,
+      "step": 340
+    },
+    {
+      "grad_norm": 0.36608144640922546,
+      "learning_rate": 3.49e-05,
+      "loss": 0.2334,
+      "step": 350
+    },
+    {
+      "grad_norm": 0.44500136375427246,
+      "learning_rate": 3.59e-05,
+      "loss": 0.2351,
+      "step": 360
+    },
+    {
+      "grad_norm": 0.36625224351882935,
+      "learning_rate": 3.69e-05,
+      "loss": 0.2229,
+      "step": 370
+    },
+    {
+      "grad_norm": 0.4702928066253662,
+      "learning_rate": 3.79e-05,
+      "loss": 0.226,
+      "step": 380
+    },
+    {
+      "grad_norm": 0.4553168714046478,
+      "learning_rate": 3.8900000000000004e-05,
+      "loss": 0.2502,
+      "step": 390
+    },
+    {
+      "grad_norm": 0.4030553102493286,
+      "learning_rate": 3.99e-05,
+      "loss": 0.2387,
+      "step": 400
+    },
+    {
+      "grad_norm": 0.32008203864097595,
+      "learning_rate": 4.09e-05,
+      "loss": 0.2077,
+      "step": 410
+    },
+    {
+      "grad_norm": 0.4638312757015228,
+      "learning_rate": 4.19e-05,
+      "loss": 0.2273,
+      "step": 420
+    },
+    {
+      "grad_norm": 0.437203586101532,
+      "learning_rate": 4.29e-05,
+      "loss": 0.2376,
+      "step": 430
+    },
+    {
+      "grad_norm": 0.5474429726600647,
+      "learning_rate": 4.39e-05,
+      "loss": 0.2435,
+      "step": 440
+    },
+    {
+      "grad_norm": 0.42206066846847534,
+      "learning_rate": 4.49e-05,
+      "loss": 0.2388,
+      "step": 450
+    },
+    {
+      "grad_norm": 0.38861826062202454,
+      "learning_rate": 4.5900000000000004e-05,
+      "loss": 0.2176,
+      "step": 460
+    },
+    {
+      "grad_norm": 0.5863853693008423,
+      "learning_rate": 4.69e-05,
+      "loss": 0.2487,
+      "step": 470
+    },
+    {
+      "grad_norm": 0.5604053139686584,
+      "learning_rate": 4.79e-05,
+      "loss": 0.2251,
+      "step": 480
+    },
+    {
+      "grad_norm": 0.4201134145259857,
+      "learning_rate": 4.89e-05,
+      "loss": 0.2118,
+      "step": 490
+    },
+    {
+      "grad_norm": 0.45899316668510437,
+      "learning_rate": 4.99e-05,
+      "loss": 0.2266,
+      "step": 500
+    },
+    {
+      "grad_norm": 0.49020540714263916,
+      "learning_rate": 5.0900000000000004e-05,
+      "loss": 0.2303,
+      "step": 510
+    },
+    {
+      "grad_norm": 0.3729408383369446,
+      "learning_rate": 5.19e-05,
+      "loss": 0.2381,
+      "step": 520
+    },
+    {
+      "grad_norm": 0.3140546381473541,
+      "learning_rate": 5.2900000000000005e-05,
+      "loss": 0.2252,
+      "step": 530
+    },
+    {
+      "grad_norm": 0.4602254331111908,
+      "learning_rate": 5.390000000000001e-05,
+      "loss": 0.2185,
+      "step": 540
+    },
+    {
+      "grad_norm": 0.4507838785648346,
+      "learning_rate": 5.4900000000000006e-05,
+      "loss": 0.2381,
+      "step": 550
+    },
+    {
+      "grad_norm": 0.414305716753006,
+      "learning_rate": 5.590000000000001e-05,
+      "loss": 0.2384,
+      "step": 560
+    },
+    {
+      "grad_norm": 0.42185264825820923,
+      "learning_rate": 5.69e-05,
+      "loss": 0.2249,
+      "step": 570
+    },
+    {
+      "grad_norm": 0.6396268606185913,
+      "learning_rate": 5.79e-05,
+      "loss": 0.2417,
+      "step": 580
+    },
+    {
+      "grad_norm": 0.6472272276878357,
+      "learning_rate": 5.89e-05,
+      "loss": 0.2314,
+      "step": 590
+    },
+    {
+      "grad_norm": 0.45745864510536194,
+      "learning_rate": 5.99e-05,
+      "loss": 0.2072,
+      "step": 600
+    },
+    {
+      "grad_norm": 0.48093411326408386,
+      "learning_rate": 6.09e-05,
+      "loss": 0.2268,
+      "step": 610
+    },
+    {
+      "grad_norm": 0.40112951397895813,
+      "learning_rate": 6.19e-05,
+      "loss": 0.2202,
+      "step": 620
+    },
+    {
+      "grad_norm": 0.6858670115470886,
+      "learning_rate": 6.29e-05,
+      "loss": 0.2323,
+      "step": 630
+    },
+    {
+      "grad_norm": 0.9508777856826782,
+      "learning_rate": 6.390000000000001e-05,
+      "loss": 0.21,
+      "step": 640
+    },
+    {
+      "grad_norm": 0.3538168668746948,
+      "learning_rate": 6.49e-05,
+      "loss": 0.2313,
+      "step": 650
+    },
+    {
+      "grad_norm": 0.5191048383712769,
+      "learning_rate": 6.59e-05,
+      "loss": 0.2161,
+      "step": 660
+    },
+    {
+      "grad_norm": 0.40369874238967896,
+      "learning_rate": 6.690000000000001e-05,
+      "loss": 0.2158,
+      "step": 670
+    },
+    {
+      "grad_norm": 0.6382988691329956,
+      "learning_rate": 6.790000000000001e-05,
+      "loss": 0.2177,
+      "step": 680
+    },
+    {
+      "grad_norm": 0.4162931740283966,
+      "learning_rate": 6.89e-05,
+      "loss": 0.2315,
+      "step": 690
+    },
+    {
+      "grad_norm": 0.4648604393005371,
+      "learning_rate": 6.99e-05,
+      "loss": 0.2054,
+      "step": 700
+    },
+    {
+      "grad_norm": 0.46259018778800964,
+      "learning_rate": 7.09e-05,
+      "loss": 0.2041,
+      "step": 710
+    },
+    {
+      "grad_norm": 0.600862443447113,
+      "learning_rate": 7.19e-05,
+      "loss": 0.2315,
+      "step": 720
+    },
+    {
+      "grad_norm": 0.5207156538963318,
+      "learning_rate": 7.29e-05,
+      "loss": 0.2324,
+      "step": 730
+    },
+    {
+      "grad_norm": 0.42644572257995605,
+      "learning_rate": 7.390000000000001e-05,
+      "loss": 0.2205,
+      "step": 740
+    },
+    {
+      "grad_norm": 0.6184375286102295,
+      "learning_rate": 7.49e-05,
+      "loss": 0.2203,
+      "step": 750
+    },
+    {
+      "grad_norm": 0.5230656266212463,
+      "learning_rate": 7.59e-05,
+      "loss": 0.2258,
+      "step": 760
+    },
+    {
+      "grad_norm": 0.4643101692199707,
+      "learning_rate": 7.69e-05,
+      "loss": 0.2182,
+      "step": 770
+    },
+    {
+      "grad_norm": 0.4660038948059082,
+      "learning_rate": 7.790000000000001e-05,
+      "loss": 0.2382,
+      "step": 780
+    },
+    {
+      "grad_norm": 0.5577147006988525,
+      "learning_rate": 7.890000000000001e-05,
+      "loss": 0.196,
+      "step": 790
+    },
+    {
+      "grad_norm": 0.4699309766292572,
+      "learning_rate": 7.99e-05,
+      "loss": 0.219,
+      "step": 800
+    },
+    {
+      "grad_norm": 0.49427250027656555,
+      "learning_rate": 8.090000000000001e-05,
+      "loss": 0.2277,
+      "step": 810
+    },
+    {
+      "grad_norm": 0.6862667202949524,
+      "learning_rate": 8.19e-05,
+      "loss": 0.2333,
+      "step": 820
+    },
+    {
+      "grad_norm": 0.4058024287223816,
+      "learning_rate": 8.29e-05,
+      "loss": 0.2244,
+      "step": 830
+    },
+    {
+      "grad_norm": 0.558210015296936,
+      "learning_rate": 8.39e-05,
+      "loss": 0.2326,
+      "step": 840
+    },
+    {
+      "grad_norm": 0.3579351007938385,
+      "learning_rate": 8.49e-05,
+      "loss": 0.2187,
+      "step": 850
+    },
+    {
+      "grad_norm": 0.6476197838783264,
+      "learning_rate": 8.59e-05,
+      "loss": 0.2217,
+      "step": 860
+    },
+    {
+      "grad_norm": 0.3715510666370392,
+      "learning_rate": 8.69e-05,
+      "loss": 0.2129,
+      "step": 870
+    },
+    {
+      "grad_norm": 0.3069533705711365,
+      "learning_rate": 8.790000000000001e-05,
+      "loss": 0.2268,
+      "step": 880
+    },
+    {
+      "grad_norm": 0.5044841170310974,
+      "learning_rate": 8.89e-05,
+      "loss": 0.2075,
+      "step": 890
+    },
+    {
+      "grad_norm": 0.761902928352356,
+      "learning_rate": 8.99e-05,
+      "loss": 0.2332,
+      "step": 900
+    },
+    {
+      "grad_norm": 0.4373076558113098,
+      "learning_rate": 9.090000000000001e-05,
+      "loss": 0.2046,
+      "step": 910
+    },
+    {
+      "grad_norm": 0.4297371506690979,
+      "learning_rate": 9.190000000000001e-05,
+      "loss": 0.2065,
+      "step": 920
+    },
+    {
+      "grad_norm": 0.4776003956794739,
+      "learning_rate": 9.290000000000001e-05,
+      "loss": 0.2426,
+      "step": 930
+    },
+    {
+      "grad_norm": 0.5716440677642822,
+      "learning_rate": 9.39e-05,
+      "loss": 0.2242,
+      "step": 940
+    },
+    {
+      "grad_norm": 0.41989845037460327,
+      "learning_rate": 9.49e-05,
+      "loss": 0.2225,
+      "step": 950
+    },
+    {
+      "grad_norm": 0.45383840799331665,
+      "learning_rate": 9.59e-05,
+      "loss": 0.2156,
+      "step": 960
+    },
+    {
+      "grad_norm": 0.4710844159126282,
+      "learning_rate": 9.69e-05,
+      "loss": 0.2195,
+      "step": 970
+    },
+    {
+      "grad_norm": 0.5963785648345947,
+      "learning_rate": 9.790000000000001e-05,
+      "loss": 0.2226,
+      "step": 980
+    },
+    {
+      "grad_norm": 0.762100100517273,
+      "learning_rate": 9.89e-05,
+      "loss": 0.2135,
+      "step": 990
+    },
+    {
+      "grad_norm": 0.6671667695045471,
+      "learning_rate": 9.99e-05,
+      "loss": 0.2379,
+      "step": 1000
+    },
+    {
+      "grad_norm": 0.43393734097480774,
+      "learning_rate": 9.999994463727085e-05,
+      "loss": 0.2208,
+      "step": 1010
+    },
+    {
+      "grad_norm": 0.40694981813430786,
+      "learning_rate": 9.999975326009292e-05,
+      "loss": 0.2122,
+      "step": 1020
+    },
+    {
+      "grad_norm": 0.4984284043312073,
+      "learning_rate": 9.999942518549879e-05,
+      "loss": 0.2191,
+      "step": 1030
+    },
+    {
+      "grad_norm": 0.8161082863807678,
+      "learning_rate": 9.999896041438544e-05,
+      "loss": 0.2265,
+      "step": 1040
+    },
+    {
+      "grad_norm": 0.5866460800170898,
+      "learning_rate": 9.999835894802353e-05,
+      "loss": 0.2576,
+      "step": 1050
+    },
+    {
+      "grad_norm": 0.3424144983291626,
+      "learning_rate": 9.999762078805743e-05,
+      "loss": 0.2356,
+      "step": 1060
+    },
+    {
+      "grad_norm": 0.45800429582595825,
+      "learning_rate": 9.999674593650526e-05,
+      "loss": 0.2292,
+      "step": 1070
+    },
+    {
+      "grad_norm": 0.40440768003463745,
+      "learning_rate": 9.99957343957588e-05,
+      "loss": 0.2153,
+      "step": 1080
+    },
+    {
+      "grad_norm": 0.5812469124794006,
+      "learning_rate": 9.99945861685836e-05,
+      "loss": 0.2324,
+      "step": 1090
+    },
+    {
+      "grad_norm": 0.3851524889469147,
+      "learning_rate": 9.999330125811884e-05,
+      "loss": 0.2233,
+      "step": 1100
+    },
+    {
+      "grad_norm": 0.402841717004776,
+      "learning_rate": 9.999187966787744e-05,
+      "loss": 0.2217,
+      "step": 1110
+    },
+    {
+      "grad_norm": 0.4815830886363983,
+      "learning_rate": 9.999032140174595e-05,
+      "loss": 0.2146,
+      "step": 1120
+    },
+    {
+      "grad_norm": 0.47242680191993713,
+      "learning_rate": 9.998862646398464e-05,
+      "loss": 0.2098,
+      "step": 1130
+    },
+    {
+      "grad_norm": 0.24558864533901215,
+      "learning_rate": 9.998679485922739e-05,
+      "loss": 0.2215,
+      "step": 1140
+    },
+    {
+      "grad_norm": 0.5753674507141113,
+      "learning_rate": 9.998482659248174e-05,
+      "loss": 0.2228,
+      "step": 1150
+    },
+    {
+      "grad_norm": 0.5498942136764526,
+      "learning_rate": 9.998272166912883e-05,
+      "loss": 0.2165,
+      "step": 1160
+    },
+    {
+      "grad_norm": 0.49823030829429626,
+      "learning_rate": 9.998048009492347e-05,
+      "loss": 0.2473,
+      "step": 1170
+    },
+    {
+      "grad_norm": 0.39590612053871155,
+      "learning_rate": 9.997810187599403e-05,
+      "loss": 0.2236,
+      "step": 1180
+    },
+    {
+      "grad_norm": 0.48530957102775574,
+      "learning_rate": 9.997558701884249e-05,
+      "loss": 0.2234,
+      "step": 1190
+    },
+    {
+      "grad_norm": 0.4040852189064026,
+      "learning_rate": 9.997293553034433e-05,
+      "loss": 0.2198,
+      "step": 1200
+    },
+    {
+      "grad_norm": 0.5381982326507568,
+      "learning_rate": 9.997014741774866e-05,
+      "loss": 0.2221,
+      "step": 1210
+    },
+    {
+      "grad_norm": 0.7528003454208374,
+      "learning_rate": 9.996722268867803e-05,
+      "loss": 0.2182,
+      "step": 1220
+    },
+    {
+      "grad_norm": 0.4129217863082886,
+      "learning_rate": 9.996416135112858e-05,
+      "loss": 0.2157,
+      "step": 1230
+    },
+    {
+      "grad_norm": 0.869448184967041,
+      "learning_rate": 9.996096341346988e-05,
+      "loss": 0.2144,
+      "step": 1240
+    },
+    {
+      "grad_norm": 0.5117060542106628,
+      "learning_rate": 9.995762888444495e-05,
+      "loss": 0.2344,
+      "step": 1250
+    },
+    {
+      "grad_norm": 0.7513025999069214,
+      "learning_rate": 9.995415777317027e-05,
+      "loss": 0.2504,
+      "step": 1260
+    },
+    {
+      "grad_norm": 0.3775970935821533,
+      "learning_rate": 9.995055008913574e-05,
+      "loss": 0.2313,
+      "step": 1270
+    },
+    {
+      "grad_norm": 0.43620729446411133,
+      "learning_rate": 9.994680584220463e-05,
+      "loss": 0.2088,
+      "step": 1280
+    },
+    {
+      "grad_norm": 0.514728844165802,
+      "learning_rate": 9.994292504261355e-05,
+      "loss": 0.2399,
+      "step": 1290
+    },
+    {
+      "grad_norm": 0.5275346040725708,
+      "learning_rate": 9.993890770097247e-05,
+      "loss": 0.2232,
+      "step": 1300
+    },
+    {
+      "grad_norm": 0.4375029504299164,
+      "learning_rate": 9.993475382826467e-05,
+      "loss": 0.2188,
+      "step": 1310
+    },
+    {
+      "grad_norm": 0.4474101960659027,
+      "learning_rate": 9.993046343584664e-05,
+      "loss": 0.2244,
+      "step": 1320
+    },
+    {
+      "grad_norm": 0.521884560585022,
+      "learning_rate": 9.992603653544816e-05,
+      "loss": 0.2135,
+      "step": 1330
+    },
+    {
+      "grad_norm": 0.3240484595298767,
+      "learning_rate": 9.992147313917222e-05,
+      "loss": 0.2226,
+      "step": 1340
+    },
+    {
+      "grad_norm": 0.3958008885383606,
+      "learning_rate": 9.991677325949497e-05,
+      "loss": 0.2303,
+      "step": 1350
+    },
+    {
+      "grad_norm": 0.4697926342487335,
+      "learning_rate": 9.991193690926568e-05,
+      "loss": 0.2188,
+      "step": 1360
+    },
+    {
+      "grad_norm": 0.46560364961624146,
+      "learning_rate": 9.990696410170678e-05,
+      "loss": 0.2214,
+      "step": 1370
+    },
+    {
+      "grad_norm": 0.4131556749343872,
+      "learning_rate": 9.990185485041371e-05,
+      "loss": 0.2171,
+      "step": 1380
+    },
+    {
+      "grad_norm": 0.4153015911579132,
+      "learning_rate": 9.989660916935498e-05,
+      "loss": 0.2131,
+      "step": 1390
+    },
+    {
+      "grad_norm": 0.6420057415962219,
+      "learning_rate": 9.989122707287208e-05,
+      "loss": 0.2078,
+      "step": 1400
+    },
+    {
+      "grad_norm": 0.8191539645195007,
+      "learning_rate": 9.988570857567945e-05,
+      "loss": 0.2108,
+      "step": 1410
+    },
+    {
+      "grad_norm": 0.5542505979537964,
+      "learning_rate": 9.988005369286446e-05,
+      "loss": 0.2205,
+      "step": 1420
+    },
+    {
+      "grad_norm": 0.6949684023857117,
+      "learning_rate": 9.987426243988734e-05,
+      "loss": 0.2274,
+      "step": 1430
+    },
+    {
+      "grad_norm": 0.6960007548332214,
+      "learning_rate": 9.986833483258114e-05,
+      "loss": 0.232,
+      "step": 1440
+    },
+    {
+      "grad_norm": 0.4363483786582947,
+      "learning_rate": 9.986227088715173e-05,
+      "loss": 0.2112,
+      "step": 1450
+    },
+    {
+      "grad_norm": 0.41103672981262207,
+      "learning_rate": 9.98560706201777e-05,
+      "loss": 0.2297,
+      "step": 1460
+    },
+    {
+      "grad_norm": 0.6309000849723816,
+      "learning_rate": 9.984973404861036e-05,
+      "loss": 0.2268,
+      "step": 1470
+    },
+    {
+      "grad_norm": 0.36002808809280396,
+      "learning_rate": 9.984326118977361e-05,
+      "loss": 0.2151,
+      "step": 1480
+    },
+    {
+      "grad_norm": 0.5909473896026611,
+      "learning_rate": 9.983665206136406e-05,
+      "loss": 0.2281,
+      "step": 1490
+    },
+    {
+      "grad_norm": 0.4377276301383972,
+      "learning_rate": 9.982990668145075e-05,
+      "loss": 0.2335,
+      "step": 1500
+    },
+    {
+      "grad_norm": 0.37812647223472595,
+      "learning_rate": 9.982302506847534e-05,
+      "loss": 0.2027,
+      "step": 1510
+    },
+    {
+      "grad_norm": 0.431948721408844,
+      "learning_rate": 9.981600724125189e-05,
+      "loss": 0.2162,
+      "step": 1520
+    },
+    {
+      "grad_norm": 0.37573811411857605,
+      "learning_rate": 9.980885321896685e-05,
+      "loss": 0.2184,
+      "step": 1530
+    },
+    {
+      "grad_norm": 0.5530624985694885,
+      "learning_rate": 9.980156302117905e-05,
+      "loss": 0.2185,
+      "step": 1540
+    },
+    {
+      "grad_norm": 0.40344762802124023,
+      "learning_rate": 9.979413666781963e-05,
+      "loss": 0.2258,
+      "step": 1550
+    },
+    {
+      "grad_norm": 0.487790584564209,
+      "learning_rate": 9.978657417919193e-05,
+      "loss": 0.2345,
+      "step": 1560
+    },
+    {
+      "grad_norm": 0.6412849426269531,
+      "learning_rate": 9.977887557597153e-05,
+      "loss": 0.2199,
+      "step": 1570
+    },
+    {
+      "grad_norm": 0.6398754715919495,
+      "learning_rate": 9.97710408792061e-05,
+      "loss": 0.2154,
+      "step": 1580
+    },
+    {
+      "grad_norm": 0.43206343054771423,
+      "learning_rate": 9.976307011031542e-05,
+      "loss": 0.2331,
+      "step": 1590
+    },
+    {
+      "grad_norm": 0.4423554241657257,
+      "learning_rate": 9.975496329109126e-05,
+      "loss": 0.222,
+      "step": 1600
+    },
+    {
+      "grad_norm": 0.38466179370880127,
+      "learning_rate": 9.974672044369732e-05,
+      "loss": 0.2074,
+      "step": 1610
+    },
+    {
+      "grad_norm": 0.45003658533096313,
+      "learning_rate": 9.97383415906693e-05,
+      "loss": 0.2135,
+      "step": 1620
+    },
+    {
+      "grad_norm": 0.5214859843254089,
+      "learning_rate": 9.97298267549146e-05,
+      "loss": 0.2285,
+      "step": 1630
+    },
+    {
+      "grad_norm": 0.4215741753578186,
+      "learning_rate": 9.972117595971249e-05,
+      "loss": 0.2129,
+      "step": 1640
+    },
+    {
+      "grad_norm": 0.5960712432861328,
+      "learning_rate": 9.971238922871391e-05,
+      "loss": 0.2346,
+      "step": 1650
+    },
+    {
+      "grad_norm": 0.5126683712005615,
+      "learning_rate": 9.970346658594142e-05,
+      "loss": 0.2147,
+      "step": 1660
+    },
+    {
+      "grad_norm": 0.4562714397907257,
+      "learning_rate": 9.969440805578923e-05,
+      "loss": 0.21,
+      "step": 1670
+    },
+    {
+      "grad_norm": 0.5866053700447083,
+      "learning_rate": 9.968521366302298e-05,
+      "loss": 0.2376,
+      "step": 1680
+    },
+    {
+      "grad_norm": 0.6102207899093628,
+      "learning_rate": 9.967588343277981e-05,
+      "loss": 0.2395,
+      "step": 1690
+    },
+    {
+      "grad_norm": 0.4125145971775055,
+      "learning_rate": 9.966641739056818e-05,
+      "loss": 0.2202,
+      "step": 1700
+    },
+    {
+      "grad_norm": 0.5642959475517273,
+      "learning_rate": 9.965681556226793e-05,
+      "loss": 0.2169,
+      "step": 1710
+    },
+    {
+      "grad_norm": 0.38978514075279236,
+      "learning_rate": 9.964707797413006e-05,
+      "loss": 0.2175,
+      "step": 1720
+    },
+    {
+      "grad_norm": 0.5113131403923035,
+      "learning_rate": 9.963720465277679e-05,
+      "loss": 0.1976,
+      "step": 1730
+    },
+    {
+      "grad_norm": 0.44819653034210205,
+      "learning_rate": 9.96271956252014e-05,
+      "loss": 0.2075,
+      "step": 1740
+    },
+    {
+      "grad_norm": 0.3951730728149414,
+      "learning_rate": 9.961705091876816e-05,
+      "loss": 0.2215,
+      "step": 1750
+    },
+    {
+      "grad_norm": 0.36689838767051697,
+      "learning_rate": 9.960677056121235e-05,
+      "loss": 0.2196,
+      "step": 1760
+    },
+    {
+      "grad_norm": 0.45175766944885254,
+      "learning_rate": 9.959635458064005e-05,
+      "loss": 0.2074,
+      "step": 1770
+    },
+    {
+      "grad_norm": 0.5387119054794312,
+      "learning_rate": 9.958580300552815e-05,
+      "loss": 0.2303,
+      "step": 1780
+    },
+    {
+      "grad_norm": 0.6717667579650879,
+      "learning_rate": 9.957511586472426e-05,
+      "loss": 0.2263,
+      "step": 1790
+    },
+    {
+      "grad_norm": 0.4420381486415863,
+      "learning_rate": 9.956429318744662e-05,
+      "loss": 0.2392,
+      "step": 1800
+    },
+    {
+      "grad_norm": 0.471440851688385,
+      "learning_rate": 9.955333500328404e-05,
+      "loss": 0.2154,
+      "step": 1810
+    },
+    {
+      "grad_norm": 0.5590283274650574,
+      "learning_rate": 9.95422413421957e-05,
+      "loss": 0.2199,
+      "step": 1820
+    },
+    {
+      "grad_norm": 0.37907642126083374,
+      "learning_rate": 9.953101223451133e-05,
+      "loss": 0.2228,
+      "step": 1830
+    },
+    {
+      "grad_norm": 0.4260089695453644,
+      "learning_rate": 9.951964771093085e-05,
+      "loss": 0.2181,
+      "step": 1840
+    },
+    {
+      "grad_norm": 0.5218323469161987,
+      "learning_rate": 9.950814780252442e-05,
+      "loss": 0.2051,
+      "step": 1850
+    },
+    {
+      "grad_norm": 0.35399192571640015,
+      "learning_rate": 9.949651254073236e-05,
+      "loss": 0.2182,
+      "step": 1860
+    },
+    {
+      "grad_norm": 0.7958611845970154,
+      "learning_rate": 9.948474195736504e-05,
+      "loss": 0.2195,
+      "step": 1870
+    },
+    {
+      "grad_norm": 0.38490650057792664,
+      "learning_rate": 9.947283608460277e-05,
+      "loss": 0.2319,
+      "step": 1880
+    },
+    {
+      "grad_norm": 0.5937964916229248,
+      "learning_rate": 9.946079495499577e-05,
+      "loss": 0.2225,
+      "step": 1890
+    },
+    {
+      "grad_norm": 0.4649060368537903,
+      "learning_rate": 9.944861860146401e-05,
+      "loss": 0.205,
+      "step": 1900
+    },
+    {
+      "grad_norm": 0.4207160472869873,
+      "learning_rate": 9.943630705729719e-05,
+      "loss": 0.2059,
+      "step": 1910
+    },
+    {
+      "grad_norm": 0.6125879287719727,
+      "learning_rate": 9.942386035615459e-05,
+      "loss": 0.2023,
+      "step": 1920
+    },
+    {
+      "grad_norm": 0.6017894148826599,
+      "learning_rate": 9.941127853206503e-05,
+      "loss": 0.2264,
+      "step": 1930
+    },
+    {
+      "grad_norm": 0.7823480367660522,
+      "learning_rate": 9.939856161942673e-05,
+      "loss": 0.2267,
+      "step": 1940
+    },
+    {
+      "grad_norm": 0.4436163902282715,
+      "learning_rate": 9.938570965300724e-05,
+      "loss": 0.233,
+      "step": 1950
+    },
+    {
+      "grad_norm": 0.42437097430229187,
+      "learning_rate": 9.937272266794335e-05,
+      "loss": 0.2243,
+      "step": 1960
+    },
+    {
+      "grad_norm": 0.5850749015808105,
+      "learning_rate": 9.935960069974096e-05,
+      "loss": 0.2013,
+      "step": 1970
+    },
+    {
+      "grad_norm": 0.4582327902317047,
+      "learning_rate": 9.934634378427506e-05,
+      "loss": 0.2239,
+      "step": 1980
+    },
+    {
+      "grad_norm": 0.3929188549518585,
+      "learning_rate": 9.933295195778954e-05,
+      "loss": 0.2537,
+      "step": 1990
+    },
+    {
+      "grad_norm": 0.4952102303504944,
+      "learning_rate": 9.931942525689715e-05,
+      "loss": 0.1979,
+      "step": 2000
+    },
+    {
+      "grad_norm": 0.34466275572776794,
+      "learning_rate": 9.930576371857936e-05,
+      "loss": 0.2244,
+      "step": 2010
+    },
+    {
+      "grad_norm": 0.5722326040267944,
+      "learning_rate": 9.929196738018629e-05,
+      "loss": 0.207,
+      "step": 2020
+    },
+    {
+      "grad_norm": 0.5094285607337952,
+      "learning_rate": 9.927803627943662e-05,
+      "loss": 0.193,
+      "step": 2030
+    },
+    {
+      "grad_norm": 0.557445228099823,
+      "learning_rate": 9.926397045441744e-05,
+      "loss": 0.2226,
+      "step": 2040
+    },
+    {
+      "grad_norm": 0.5022002458572388,
+      "learning_rate": 9.924976994358417e-05,
+      "loss": 0.2255,
+      "step": 2050
+    },
+    {
+      "grad_norm": 0.3470063805580139,
+      "learning_rate": 9.923543478576048e-05,
+      "loss": 0.2005,
+      "step": 2060
+    },
+    {
+      "grad_norm": 0.3966708779335022,
+      "learning_rate": 9.922096502013813e-05,
+      "loss": 0.1954,
+      "step": 2070
+    },
+    {
+      "grad_norm": 0.6764780879020691,
+      "learning_rate": 9.92063606862769e-05,
+      "loss": 0.2096,
+      "step": 2080
+    },
+    {
+      "grad_norm": 0.5607836246490479,
+      "learning_rate": 9.919162182410453e-05,
+      "loss": 0.2009,
+      "step": 2090
+    },
+    {
+      "grad_norm": 0.5560227036476135,
+      "learning_rate": 9.917674847391645e-05,
+      "loss": 0.2005,
+      "step": 2100
+    },
+    {
+      "grad_norm": 0.6449229717254639,
+      "learning_rate": 9.916174067637584e-05,
+      "loss": 0.2081,
+      "step": 2110
+    },
+    {
+      "grad_norm": 0.39691615104675293,
+      "learning_rate": 9.914659847251348e-05,
+      "loss": 0.2014,
+      "step": 2120
+    },
+    {
+      "grad_norm": 0.4174331724643707,
+      "learning_rate": 9.913132190372753e-05,
+      "loss": 0.1991,
+      "step": 2130
+    },
+    {
+      "grad_norm": 0.6411559581756592,
+      "learning_rate": 9.911591101178359e-05,
+      "loss": 0.1905,
+      "step": 2140
+    },
+    {
+      "grad_norm": 0.4073953628540039,
+      "learning_rate": 9.910036583881443e-05,
+      "loss": 0.1997,
+      "step": 2150
+    },
+    {
+      "grad_norm": 0.38011106848716736,
+      "learning_rate": 9.908468642731995e-05,
+      "loss": 0.2012,
+      "step": 2160
+    },
+    {
+      "grad_norm": 0.4752521514892578,
+      "learning_rate": 9.906887282016707e-05,
+      "loss": 0.2082,
+      "step": 2170
+    },
+    {
+      "grad_norm": 0.6735978126525879,
+      "learning_rate": 9.90529250605896e-05,
+      "loss": 0.1988,
+      "step": 2180
+    },
+    {
+      "grad_norm": 0.46789517998695374,
+      "learning_rate": 9.903684319218809e-05,
+      "loss": 0.2031,
+      "step": 2190
+    },
+    {
+      "grad_norm": 0.592193603515625,
+      "learning_rate": 9.902062725892976e-05,
+      "loss": 0.2021,
+      "step": 2200
+    },
+    {
+      "grad_norm": 0.5898789167404175,
+      "learning_rate": 9.900427730514834e-05,
+      "loss": 0.1925,
+      "step": 2210
+    },
+    {
+      "grad_norm": 0.5945219993591309,
+      "learning_rate": 9.8987793375544e-05,
+      "loss": 0.1993,
+      "step": 2220
+    },
+    {
+      "grad_norm": 0.7195473313331604,
+      "learning_rate": 9.897117551518318e-05,
+      "loss": 0.2181,
+      "step": 2230
+    },
+    {
+      "grad_norm": 0.47695791721343994,
+      "learning_rate": 9.895442376949844e-05,
+      "loss": 0.2021,
+      "step": 2240
+    },
+    {
+      "grad_norm": 0.2558879554271698,
+      "learning_rate": 9.893753818428845e-05,
+      "loss": 0.2029,
+      "step": 2250
+    },
+    {
+      "grad_norm": 0.3444175124168396,
+      "learning_rate": 9.892051880571773e-05,
+      "loss": 0.2106,
+      "step": 2260
+    },
+    {
+      "grad_norm": 0.5179827213287354,
+      "learning_rate": 9.890336568031663e-05,
+      "loss": 0.2034,
+      "step": 2270
+    },
+    {
+      "grad_norm": 0.4513013958930969,
+      "learning_rate": 9.888607885498113e-05,
+      "loss": 0.2035,
+      "step": 2280
+    },
+    {
+      "grad_norm": 0.38986438512802124,
+      "learning_rate": 9.886865837697275e-05,
+      "loss": 0.2165,
+      "step": 2290
+    },
+    {
+      "grad_norm": 0.45096898078918457,
+      "learning_rate": 9.88511042939184e-05,
+      "loss": 0.2064,
+      "step": 2300
+    },
+    {
+      "grad_norm": 0.38840264081954956,
+      "learning_rate": 9.883341665381028e-05,
+      "loss": 0.2177,
+      "step": 2310
+    },
+    {
+      "grad_norm": 0.599179744720459,
+      "learning_rate": 9.881559550500575e-05,
+      "loss": 0.1895,
+      "step": 2320
+    },
+    {
+      "grad_norm": 0.5268388390541077,
+      "learning_rate": 9.879764089622712e-05,
+      "loss": 0.1903,
+      "step": 2330
+    },
+    {
+      "grad_norm": 0.5076361298561096,
+      "learning_rate": 9.87795528765616e-05,
+      "loss": 0.2202,
+      "step": 2340
+    },
+    {
+      "grad_norm": 0.4006635248661041,
+      "learning_rate": 9.876133149546118e-05,
+      "loss": 0.1996,
+      "step": 2350
+    },
+    {
+      "grad_norm": 0.44850823283195496,
+      "learning_rate": 9.874297680274238e-05,
+      "loss": 0.198,
+      "step": 2360
+    },
+    {
+      "grad_norm": 0.4126204550266266,
+      "learning_rate": 9.872448884858624e-05,
+      "loss": 0.2165,
+      "step": 2370
+    },
+    {
+      "grad_norm": 0.3829819858074188,
+      "learning_rate": 9.870586768353815e-05,
+      "loss": 0.2057,
+      "step": 2380
+    },
+    {
+      "grad_norm": 0.4593510925769806,
+      "learning_rate": 9.868711335850764e-05,
+      "loss": 0.2036,
+      "step": 2390
+    },
+    {
+      "grad_norm": 0.4643329679965973,
+      "learning_rate": 9.866822592476833e-05,
+      "loss": 0.1958,
+      "step": 2400
+    },
+    {
+      "grad_norm": 0.47348296642303467,
+      "learning_rate": 9.86492054339577e-05,
+      "loss": 0.1981,
+      "step": 2410
+    },
+    {
+      "grad_norm": 0.5830390453338623,
+      "learning_rate": 9.863005193807711e-05,
+      "loss": 0.1938,
+      "step": 2420
+    },
+    {
+      "grad_norm": 0.6716923117637634,
+      "learning_rate": 9.861076548949143e-05,
+      "loss": 0.1965,
+      "step": 2430
+    },
+    {
+      "grad_norm": 0.4246164858341217,
+      "learning_rate": 9.859134614092912e-05,
+      "loss": 0.1928,
+      "step": 2440
+    },
+    {
+      "grad_norm": 0.36640089750289917,
+      "learning_rate": 9.857179394548191e-05,
+      "loss": 0.2014,
+      "step": 2450
+    },
+    {
+      "grad_norm": 0.6031244993209839,
+      "learning_rate": 9.855210895660477e-05,
+      "loss": 0.1964,
+      "step": 2460
+    },
+    {
+      "grad_norm": 0.6716413497924805,
+      "learning_rate": 9.853229122811568e-05,
+      "loss": 0.2057,
+      "step": 2470
+    },
+    {
+      "grad_norm": 0.5295430421829224,
+      "learning_rate": 9.851234081419559e-05,
+      "loss": 0.213,
+      "step": 2480
+    },
+    {
+      "grad_norm": 0.47871553897857666,
+      "learning_rate": 9.849225776938814e-05,
+      "loss": 0.1958,
+      "step": 2490
+    },
+    {
+      "grad_norm": 0.6119757890701294,
+      "learning_rate": 9.847204214859964e-05,
+      "loss": 0.2167,
+      "step": 2500
+    },
+    {
+      "grad_norm": 0.5226811766624451,
+      "learning_rate": 9.845169400709879e-05,
+      "loss": 0.2072,
+      "step": 2510
+    },
+    {
+      "grad_norm": 0.45799878239631653,
+      "learning_rate": 9.843121340051664e-05,
+      "loss": 0.1966,
+      "step": 2520
+    },
+    {
+      "grad_norm": 0.4275935888290405,
+      "learning_rate": 9.841060038484641e-05,
+      "loss": 0.1941,
+      "step": 2530
+    },
+    {
+      "grad_norm": 0.45139697194099426,
+      "learning_rate": 9.838985501644328e-05,
+      "loss": 0.2009,
+      "step": 2540
+    },
+    {
+      "grad_norm": 0.47737398743629456,
+      "learning_rate": 9.83689773520243e-05,
+      "loss": 0.207,
+      "step": 2550
+    },
+    {
+      "grad_norm": 0.4446268379688263,
+      "learning_rate": 9.834796744866819e-05,
+      "loss": 0.1939,
+      "step": 2560
+    },
+    {
+      "grad_norm": 0.5285122394561768,
+      "learning_rate": 9.832682536381525e-05,
+      "loss": 0.2191,
+      "step": 2570
+    },
+    {
+      "grad_norm": 0.5420625805854797,
+      "learning_rate": 9.830555115526711e-05,
+      "loss": 0.2002,
+      "step": 2580
+    },
+    {
+      "grad_norm": 0.45661666989326477,
+      "learning_rate": 9.828414488118667e-05,
+      "loss": 0.2096,
+      "step": 2590
+    },
+    {
+      "grad_norm": 0.573259174823761,
+      "learning_rate": 9.826260660009785e-05,
+      "loss": 0.2037,
+      "step": 2600
+    },
+    {
+      "grad_norm": 0.45775482058525085,
+      "learning_rate": 9.824093637088547e-05,
+      "loss": 0.2039,
+      "step": 2610
+    },
+    {
+      "grad_norm": 0.3507033884525299,
+      "learning_rate": 9.821913425279514e-05,
+      "loss": 0.2014,
+      "step": 2620
+    },
+    {
+      "grad_norm": 0.7480464577674866,
+      "learning_rate": 9.8197200305433e-05,
+      "loss": 0.2149,
+      "step": 2630
+    },
+    {
+      "grad_norm": 0.5780835747718811,
+      "learning_rate": 9.817513458876564e-05,
+      "loss": 0.2085,
+      "step": 2640
+    },
+    {
+      "grad_norm": 0.4085920751094818,
+      "learning_rate": 9.815293716311987e-05,
+      "loss": 0.2021,
+      "step": 2650
+    },
+    {
+      "grad_norm": 0.56276535987854,
+      "learning_rate": 9.813060808918262e-05,
+      "loss": 0.1884,
+      "step": 2660
+    },
+    {
+      "grad_norm": 0.49423038959503174,
+      "learning_rate": 9.810814742800069e-05,
+      "loss": 0.211,
+      "step": 2670
+    },
+    {
+      "grad_norm": 0.7306650876998901,
+      "learning_rate": 9.808555524098074e-05,
+      "loss": 0.2181,
+      "step": 2680
+    },
+    {
+      "grad_norm": 0.41126078367233276,
+      "learning_rate": 9.806283158988887e-05,
+      "loss": 0.2156,
+      "step": 2690
+    },
+    {
+      "grad_norm": 0.5092565417289734,
+      "learning_rate": 9.803997653685072e-05,
+      "loss": 0.2128,
+      "step": 2700
+    },
+    {
+      "grad_norm": 0.4337257146835327,
+      "learning_rate": 9.801699014435112e-05,
+      "loss": 0.2097,
+      "step": 2710
+    },
+    {
+      "grad_norm": 0.43089592456817627,
+      "learning_rate": 9.799387247523398e-05,
+      "loss": 0.222,
+      "step": 2720
+    },
+    {
+      "grad_norm": 0.4961455762386322,
+      "learning_rate": 9.797062359270215e-05,
+      "loss": 0.2109,
+      "step": 2730
+    },
+    {
+      "grad_norm": 0.44819968938827515,
+      "learning_rate": 9.794724356031715e-05,
+      "loss": 0.2181,
+      "step": 2740
+    },
+    {
+      "grad_norm": 0.3907797336578369,
+      "learning_rate": 9.792373244199913e-05,
+      "loss": 0.2,
+      "step": 2750
+    },
+    {
+      "grad_norm": 0.6559931635856628,
+      "learning_rate": 9.790009030202658e-05,
+      "loss": 0.1949,
+      "step": 2760
+    },
+    {
+      "grad_norm": 0.31975728273391724,
+      "learning_rate": 9.78763172050362e-05,
+      "loss": 0.2052,
+      "step": 2770
+    },
+    {
+      "grad_norm": 0.5366892218589783,
+      "learning_rate": 9.785241321602274e-05,
+      "loss": 0.2109,
+      "step": 2780
+    },
+    {
+      "grad_norm": 0.45059722661972046,
+      "learning_rate": 9.782837840033879e-05,
+      "loss": 0.2047,
+      "step": 2790
+    },
+    {
+      "grad_norm": 0.5156622529029846,
+      "learning_rate": 9.780421282369461e-05,
+      "loss": 0.1995,
+      "step": 2800
+    },
+    {
+      "grad_norm": 0.4442014694213867,
+      "learning_rate": 9.777991655215797e-05,
+      "loss": 0.2123,
+      "step": 2810
+    },
+    {
+      "grad_norm": 0.4781392514705658,
+      "learning_rate": 9.775548965215394e-05,
+      "loss": 0.2126,
+      "step": 2820
+    },
+    {
+      "grad_norm": 0.33649227023124695,
+      "learning_rate": 9.773093219046474e-05,
+      "loss": 0.1962,
+      "step": 2830
+    },
+    {
+      "grad_norm": 0.4778091609477997,
+      "learning_rate": 9.770624423422954e-05,
+      "loss": 0.2091,
+      "step": 2840
+    },
+    {
+      "grad_norm": 0.414142370223999,
+      "learning_rate": 9.768142585094426e-05,
+      "loss": 0.2118,
+      "step": 2850
+    },
+    {
+      "grad_norm": 0.5731590390205383,
+      "learning_rate": 9.765647710846142e-05,
+      "loss": 0.2037,
+      "step": 2860
+    },
+    {
+      "grad_norm": 0.4652884602546692,
+      "learning_rate": 9.763139807498991e-05,
+      "loss": 0.1994,
+      "step": 2870
+    },
+    {
+      "grad_norm": 0.5806286334991455,
+      "learning_rate": 9.760618881909487e-05,
+      "loss": 0.1992,
+      "step": 2880
+    },
+    {
+      "grad_norm": 0.49678346514701843,
+      "learning_rate": 9.758084940969744e-05,
+      "loss": 0.1932,
+      "step": 2890
+    },
+    {
+      "grad_norm": 0.6259763240814209,
+      "learning_rate": 9.755537991607459e-05,
+      "loss": 0.2219,
+      "step": 2900
+    },
+    {
+      "grad_norm": 0.4482049345970154,
+      "learning_rate": 9.752978040785895e-05,
+      "loss": 0.2097,
+      "step": 2910
+    },
+    {
+      "grad_norm": 0.3813600540161133,
+      "learning_rate": 9.750405095503859e-05,
+      "loss": 0.2175,
+      "step": 2920
+    },
+    {
+      "grad_norm": 0.6804177165031433,
+      "learning_rate": 9.747819162795686e-05,
+      "loss": 0.1995,
+      "step": 2930
+    },
+    {
+      "grad_norm": 0.5395722389221191,
+      "learning_rate": 9.745220249731217e-05,
+      "loss": 0.2116,
+      "step": 2940
+    },
+    {
+      "grad_norm": 0.4733133614063263,
+      "learning_rate": 9.742608363415781e-05,
+      "loss": 0.2311,
+      "step": 2950
+    },
+    {
+      "grad_norm": 0.41697046160697937,
+      "learning_rate": 9.739983510990176e-05,
+      "loss": 0.2012,
+      "step": 2960
+    },
+    {
+      "grad_norm": 0.49018508195877075,
+      "learning_rate": 9.737345699630647e-05,
+      "loss": 0.2187,
+      "step": 2970
+    },
+    {
+      "grad_norm": 0.4407442808151245,
+      "learning_rate": 9.734694936548869e-05,
+      "loss": 0.2095,
+      "step": 2980
+    },
+    {
+      "grad_norm": 0.4653457701206207,
+      "learning_rate": 9.732031228991932e-05,
+      "loss": 0.2283,
+      "step": 2990
+    },
+    {
+      "grad_norm": 0.4447305202484131,
+      "learning_rate": 9.729354584242302e-05,
+      "loss": 0.1974,
+      "step": 3000
+    },
+    {
+      "grad_norm": 0.520149827003479,
+      "learning_rate": 9.726665009617832e-05,
+      "loss": 0.199,
+      "step": 3010
+    },
+    {
+      "grad_norm": 0.4517776668071747,
+      "learning_rate": 9.723962512471714e-05,
+      "loss": 0.1913,
+      "step": 3020
+    },
+    {
+      "grad_norm": 0.48857182264328003,
+      "learning_rate": 9.72124710019247e-05,
+      "loss": 0.205,
+      "step": 3030
+    },
+    {
+      "grad_norm": 0.571943461894989,
+      "learning_rate": 9.718518780203934e-05,
+      "loss": 0.1986,
+      "step": 3040
+    },
+    {
+      "grad_norm": 0.5149146914482117,
+      "learning_rate": 9.715777559965228e-05,
+      "loss": 0.2035,
+      "step": 3050
+    },
+    {
+      "grad_norm": 0.4136259853839874,
+      "learning_rate": 9.713023446970746e-05,
+      "loss": 0.2228,
+      "step": 3060
+    },
+    {
+      "grad_norm": 0.5972459316253662,
+      "learning_rate": 9.710256448750126e-05,
+      "loss": 0.2128,
+      "step": 3070
+    },
+    {
+      "grad_norm": 0.43982112407684326,
+      "learning_rate": 9.707476572868235e-05,
+      "loss": 0.195,
+      "step": 3080
+    },
+    {
+      "grad_norm": 0.6201239824295044,
+      "learning_rate": 9.704683826925149e-05,
+      "loss": 0.2098,
+      "step": 3090
+    },
+    {
+      "grad_norm": 0.46517086029052734,
+      "learning_rate": 9.701878218556129e-05,
+      "loss": 0.1971,
+      "step": 3100
+    },
+    {
+      "grad_norm": 0.47265520691871643,
+      "learning_rate": 9.699059755431598e-05,
+      "loss": 0.2088,
+      "step": 3110
+    },
+    {
+      "grad_norm": 0.5491688251495361,
+      "learning_rate": 9.696228445257132e-05,
+      "loss": 0.2178,
+      "step": 3120
+    },
+    {
+      "grad_norm": 0.7150092124938965,
+      "learning_rate": 9.693384295773419e-05,
+      "loss": 0.2151,
+      "step": 3130
+    },
+    {
+      "grad_norm": 0.39649122953414917,
+      "learning_rate": 9.690527314756259e-05,
+      "loss": 0.2162,
+      "step": 3140
+    },
+    {
+      "grad_norm": 0.4652166962623596,
+      "learning_rate": 9.687657510016527e-05,
+      "loss": 0.2088,
+      "step": 3150
+    },
+    {
+      "grad_norm": 0.5147058963775635,
+      "learning_rate": 9.684774889400161e-05,
+      "loss": 0.2165,
+      "step": 3160
+    },
+    {
+      "grad_norm": 0.5441020131111145,
+      "learning_rate": 9.681879460788135e-05,
+      "loss": 0.2298,
+      "step": 3170
+    },
+    {
+      "grad_norm": 0.5349255800247192,
+      "learning_rate": 9.67897123209644e-05,
+      "loss": 0.2231,
+      "step": 3180
+    },
+    {
+      "grad_norm": 0.4025271236896515,
+      "learning_rate": 9.676050211276062e-05,
+      "loss": 0.219,
+      "step": 3190
+    },
+    {
+      "grad_norm": 0.3355466425418854,
+      "learning_rate": 9.673116406312962e-05,
+      "loss": 0.1937,
+      "step": 3200
+    },
+    {
+      "grad_norm": 0.5339361429214478,
+      "learning_rate": 9.67016982522805e-05,
+      "loss": 0.2147,
+      "step": 3210
+    },
+    {
+      "grad_norm": 0.5363774299621582,
+      "learning_rate": 9.667210476077164e-05,
+      "loss": 0.2099,
+      "step": 3220
+    },
+    {
+      "grad_norm": 0.48728325963020325,
+      "learning_rate": 9.664238366951055e-05,
+      "loss": 0.202,
+      "step": 3230
+    },
+    {
+      "grad_norm": 0.7211222648620605,
+      "learning_rate": 9.661253505975355e-05,
+      "loss": 0.2182,
+      "step": 3240
+    },
+    {
+      "grad_norm": 0.7391670942306519,
+      "learning_rate": 9.658255901310557e-05,
+      "loss": 0.2476,
+      "step": 3250
+    },
+    {
+      "grad_norm": 0.47313016653060913,
+      "learning_rate": 9.655245561152e-05,
+      "loss": 0.2104,
+      "step": 3260
+    },
+    {
+      "grad_norm": 0.45255759358406067,
+      "learning_rate": 9.65222249372984e-05,
+      "loss": 0.2076,
+      "step": 3270
+    },
+    {
+      "grad_norm": 0.42268306016921997,
+      "learning_rate": 9.649186707309026e-05,
+      "loss": 0.2177,
+      "step": 3280
+    },
+    {
+      "grad_norm": 0.7118562459945679,
+      "learning_rate": 9.646138210189283e-05,
+      "loss": 0.2487,
+      "step": 3290
+    },
+    {
+      "grad_norm": 0.5355364680290222,
+      "learning_rate": 9.643077010705087e-05,
+      "loss": 0.2105,
+      "step": 3300
+    },
+    {
+      "grad_norm": 0.4892050325870514,
+      "learning_rate": 9.640003117225637e-05,
+      "loss": 0.1919,
+      "step": 3310
+    },
+    {
+      "grad_norm": 0.41247648000717163,
+      "learning_rate": 9.636916538154846e-05,
+      "loss": 0.2103,
+      "step": 3320
+    },
+    {
+      "grad_norm": 0.5025883913040161,
+      "learning_rate": 9.633817281931296e-05,
+      "loss": 0.2159,
+      "step": 3330
+    },
+    {
+      "grad_norm": 0.5106640458106995,
+      "learning_rate": 9.630705357028242e-05,
+      "loss": 0.2091,
+      "step": 3340
+    },
+    {
+      "grad_norm": 0.5605065822601318,
+      "learning_rate": 9.627580771953563e-05,
+      "loss": 0.2095,
+      "step": 3350
+    },
+    {
+      "grad_norm": 0.3837064206600189,
+      "learning_rate": 9.624443535249759e-05,
+      "loss": 0.2047,
+      "step": 3360
+    },
+    {
+      "grad_norm": 0.574596643447876,
+      "learning_rate": 9.621293655493913e-05,
+      "loss": 0.207,
+      "step": 3370
+    },
+    {
+      "grad_norm": 0.41673874855041504,
+      "learning_rate": 9.618131141297675e-05,
+      "loss": 0.2206,
+      "step": 3380
+    },
+    {
+      "grad_norm": 0.40438029170036316,
+      "learning_rate": 9.614956001307242e-05,
+      "loss": 0.2101,
+      "step": 3390
+    },
+    {
+      "grad_norm": 0.4186011254787445,
+      "learning_rate": 9.611768244203321e-05,
+      "loss": 0.181,
+      "step": 3400
+    },
+    {
+      "grad_norm": 0.7165257334709167,
+      "learning_rate": 9.60856787870112e-05,
+      "loss": 0.2236,
+      "step": 3410
+    },
+    {
+      "grad_norm": 0.3978814482688904,
+      "learning_rate": 9.605354913550318e-05,
+      "loss": 0.2081,
+      "step": 3420
+    },
+    {
+      "grad_norm": 0.3733712434768677,
+      "learning_rate": 9.602129357535037e-05,
+      "loss": 0.2076,
+      "step": 3430
+    },
+    {
+      "grad_norm": 0.43590813875198364,
+      "learning_rate": 9.598891219473825e-05,
+      "loss": 0.2227,
+      "step": 3440
+    },
+    {
+      "grad_norm": 0.6133702397346497,
+      "learning_rate": 9.595640508219625e-05,
+      "loss": 0.2141,
+      "step": 3450
+    },
+    {
+      "grad_norm": 0.39810365438461304,
+      "learning_rate": 9.592377232659761e-05,
+      "loss": 0.2059,
+      "step": 3460
+    },
+    {
+      "grad_norm": 0.4794234037399292,
+      "learning_rate": 9.589101401715904e-05,
+      "loss": 0.2013,
+      "step": 3470
+    },
+    {
+      "grad_norm": 0.4503355920314789,
+      "learning_rate": 9.585813024344045e-05,
+      "loss": 0.2165,
+      "step": 3480
+    },
+    {
+      "grad_norm": 0.4410129487514496,
+      "learning_rate": 9.58251210953449e-05,
+      "loss": 0.2146,
+      "step": 3490
+    },
+    {
+      "grad_norm": 0.5044501423835754,
+      "learning_rate": 9.579198666311809e-05,
+      "loss": 0.2017,
+      "step": 3500
+    },
+    {
+      "grad_norm": 0.38198474049568176,
+      "learning_rate": 9.575872703734832e-05,
+      "loss": 0.1937,
+      "step": 3510
+    },
+    {
+      "grad_norm": 0.42876899242401123,
+      "learning_rate": 9.572534230896611e-05,
+      "loss": 0.2094,
+      "step": 3520
+    },
+    {
+      "grad_norm": 0.5578532814979553,
+      "learning_rate": 9.569183256924403e-05,
+      "loss": 0.2024,
+      "step": 3530
+    },
+    {
+      "grad_norm": 0.48812222480773926,
+      "learning_rate": 9.565819790979646e-05,
+      "loss": 0.2122,
+      "step": 3540
+    },
+    {
+      "grad_norm": 0.48869720101356506,
+      "learning_rate": 9.562443842257925e-05,
+      "loss": 0.2025,
+      "step": 3550
+    },
+    {
+      "grad_norm": 0.46242064237594604,
+      "learning_rate": 9.559055419988956e-05,
+      "loss": 0.2132,
+      "step": 3560
+    },
+    {
+      "grad_norm": 0.7380343675613403,
+      "learning_rate": 9.555654533436557e-05,
+      "loss": 0.2052,
+      "step": 3570
+    },
+    {
+      "grad_norm": 0.6011578440666199,
+      "learning_rate": 9.552241191898621e-05,
+      "loss": 0.2034,
+      "step": 3580
+    },
+    {
+      "grad_norm": 0.6868306994438171,
+      "learning_rate": 9.548815404707092e-05,
+      "loss": 0.1924,
+      "step": 3590
+    },
+    {
+      "grad_norm": 0.52702397108078,
+      "learning_rate": 9.545377181227942e-05,
+      "loss": 0.1951,
+      "step": 3600
+    },
+    {
+      "grad_norm": 0.5616778135299683,
+      "learning_rate": 9.541926530861145e-05,
+      "loss": 0.1958,
+      "step": 3610
+    },
+    {
+      "grad_norm": 0.48593631386756897,
+      "learning_rate": 9.538463463040645e-05,
+      "loss": 0.2107,
+      "step": 3620
+    },
+    {
+      "grad_norm": 0.49697476625442505,
+      "learning_rate": 9.534987987234337e-05,
+      "loss": 0.217,
+      "step": 3630
+    },
+    {
+      "grad_norm": 0.4382655918598175,
+      "learning_rate": 9.53150011294404e-05,
+      "loss": 0.2058,
+      "step": 3640
+    },
+    {
+      "grad_norm": 0.5205913186073303,
+      "learning_rate": 9.527999849705471e-05,
+      "loss": 0.199,
+      "step": 3650
+    },
+    {
+      "grad_norm": 0.4751761853694916,
+      "learning_rate": 9.524487207088213e-05,
+      "loss": 0.2151,
+      "step": 3660
+    },
+    {
+      "grad_norm": 0.43685925006866455,
+      "learning_rate": 9.520962194695698e-05,
+      "loss": 0.2024,
+      "step": 3670
+    },
+    {
+      "grad_norm": 0.44975072145462036,
+      "learning_rate": 9.517424822165175e-05,
+      "loss": 0.1803,
+      "step": 3680
+    },
+    {
+      "grad_norm": 0.49048271775245667,
+      "learning_rate": 9.513875099167685e-05,
+      "loss": 0.1951,
+      "step": 3690
+    },
+    {
+      "grad_norm": 0.5175989866256714,
+      "learning_rate": 9.510313035408035e-05,
+      "loss": 0.1885,
+      "step": 3700
+    },
+    {
+      "grad_norm": 0.4474627673625946,
+      "learning_rate": 9.506738640624775e-05,
+      "loss": 0.1904,
+      "step": 3710
+    },
+    {
+      "grad_norm": 0.6010826230049133,
+      "learning_rate": 9.50315192459016e-05,
+      "loss": 0.2063,
+      "step": 3720
+    },
+    {
+      "grad_norm": 0.5450491905212402,
+      "learning_rate": 9.499552897110136e-05,
+      "loss": 0.201,
+      "step": 3730
+    },
+    {
+      "grad_norm": 0.8840779662132263,
+      "learning_rate": 9.495941568024304e-05,
+      "loss": 0.2108,
+      "step": 3740
+    },
+    {
+      "grad_norm": 0.5438401103019714,
+      "learning_rate": 9.492317947205904e-05,
+      "loss": 0.197,
+      "step": 3750
+    },
+    {
+      "grad_norm": 0.3539958596229553,
+      "learning_rate": 9.488682044561775e-05,
+      "loss": 0.1951,
+      "step": 3760
+    },
+    {
+      "grad_norm": 0.6255428791046143,
+      "learning_rate": 9.485033870032335e-05,
+      "loss": 0.2105,
+      "step": 3770
+    },
+    {
+      "grad_norm": 0.5884273648262024,
+      "learning_rate": 9.481373433591556e-05,
+      "loss": 0.2219,
+      "step": 3780
+    },
+    {
+      "grad_norm": 0.7482819557189941,
+      "learning_rate": 9.47770074524693e-05,
+      "loss": 0.1974,
+      "step": 3790
+    },
+    {
+      "grad_norm": 0.41895976662635803,
+      "learning_rate": 9.474015815039446e-05,
+      "loss": 0.1975,
+      "step": 3800
+    },
+    {
+      "grad_norm": 0.4787296950817108,
+      "learning_rate": 9.470318653043565e-05,
+      "loss": 0.2319,
+      "step": 3810
+    },
+    {
+      "grad_norm": 0.4994959235191345,
+      "learning_rate": 9.466609269367185e-05,
+      "loss": 0.2139,
+      "step": 3820
+    },
+    {
+      "grad_norm": 0.6430701017379761,
+      "learning_rate": 9.46288767415162e-05,
+      "loss": 0.2024,
+      "step": 3830
+    },
+    {
+      "grad_norm": 0.47564437985420227,
+      "learning_rate": 9.459153877571567e-05,
+      "loss": 0.2118,
+      "step": 3840
+    },
+    {
+      "grad_norm": 0.6205402612686157,
+      "learning_rate": 9.455407889835087e-05,
+      "loss": 0.1879,
+      "step": 3850
+    },
+    {
+      "grad_norm": 0.3639780580997467,
+      "learning_rate": 9.451649721183564e-05,
+      "loss": 0.2123,
+      "step": 3860
+    },
+    {
+      "grad_norm": 0.5286294221878052,
+      "learning_rate": 9.447879381891692e-05,
+      "loss": 0.2078,
+      "step": 3870
+    },
+    {
+      "grad_norm": 0.7133933901786804,
+      "learning_rate": 9.444096882267428e-05,
+      "loss": 0.2069,
+      "step": 3880
+    },
+    {
+      "grad_norm": 0.45813655853271484,
+      "learning_rate": 9.440302232651988e-05,
+      "loss": 0.1907,
+      "step": 3890
+    },
+    {
+      "grad_norm": 0.5061445236206055,
+      "learning_rate": 9.436495443419795e-05,
+      "loss": 0.2012,
+      "step": 3900
+    },
+    {
+      "grad_norm": 0.516703188419342,
+      "learning_rate": 9.432676524978466e-05,
+      "loss": 0.2063,
+      "step": 3910
+    },
+    {
+      "grad_norm": 0.6524009704589844,
+      "learning_rate": 9.42884548776878e-05,
+      "loss": 0.1765,
+      "step": 3920
+    },
+    {
+      "grad_norm": 0.49970391392707825,
+      "learning_rate": 9.425002342264646e-05,
+      "loss": 0.212,
+      "step": 3930
+    },
+    {
+      "grad_norm": 0.4697237014770508,
+      "learning_rate": 9.421147098973077e-05,
+      "loss": 0.2065,
+      "step": 3940
+    },
+    {
+      "grad_norm": 0.6069733500480652,
+      "learning_rate": 9.41727976843416e-05,
+      "loss": 0.2045,
+      "step": 3950
+    },
+    {
+      "grad_norm": 0.49133363366127014,
+      "learning_rate": 9.413400361221029e-05,
+      "loss": 0.1947,
+      "step": 3960
+    },
+    {
+      "grad_norm": 0.5003812909126282,
+      "learning_rate": 9.409508887939835e-05,
+      "loss": 0.2038,
+      "step": 3970
+    },
+    {
+      "grad_norm": 0.480617880821228,
+      "learning_rate": 9.40560535922972e-05,
+      "loss": 0.2074,
+      "step": 3980
+    },
+    {
+      "grad_norm": 0.5644674301147461,
+      "learning_rate": 9.40168978576278e-05,
+      "loss": 0.2177,
+      "step": 3990
+    },
+    {
+      "grad_norm": 0.7343176603317261,
+      "learning_rate": 9.397762178244043e-05,
+      "loss": 0.2132,
+      "step": 4000
+    },
+    {
+      "grad_norm": 0.6299076080322266,
+      "learning_rate": 9.393822547411439e-05,
+      "loss": 0.1908,
+      "step": 4010
+    },
+    {
+      "grad_norm": 0.605165421962738,
+      "learning_rate": 9.389870904035769e-05,
+      "loss": 0.2052,
+      "step": 4020
+    },
+    {
+      "grad_norm": 0.5718099474906921,
+      "learning_rate": 9.385907258920672e-05,
+      "loss": 0.1922,
+      "step": 4030
+    },
+    {
+      "grad_norm": 0.5534369349479675,
+      "learning_rate": 9.381931622902607e-05,
+      "loss": 0.2062,
+      "step": 4040
+    },
+    {
+      "grad_norm": 0.45373842120170593,
+      "learning_rate": 9.377944006850807e-05,
+      "loss": 0.2052,
+      "step": 4050
+    },
+    {
+      "grad_norm": 0.7358400225639343,
+      "learning_rate": 9.373944421667265e-05,
+      "loss": 0.1901,
+      "step": 4060
+    },
+    {
+      "grad_norm": 0.6396151781082153,
+      "learning_rate": 9.369932878286691e-05,
+      "loss": 0.2006,
+      "step": 4070
+    },
+    {
+      "grad_norm": 0.562699556350708,
+      "learning_rate": 9.365909387676494e-05,
+      "loss": 0.1928,
+      "step": 4080
+    },
+    {
+      "grad_norm": 0.5175834894180298,
+      "learning_rate": 9.361873960836744e-05,
+      "loss": 0.1951,
+      "step": 4090
+    },
+    {
+      "grad_norm": 0.48984676599502563,
+      "learning_rate": 9.357826608800142e-05,
+      "loss": 0.2024,
+      "step": 4100
+    },
+    {
+      "grad_norm": 0.5194165706634521,
+      "learning_rate": 9.353767342631994e-05,
+      "loss": 0.2088,
+      "step": 4110
+    },
+    {
+      "grad_norm": 0.4917297661304474,
+      "learning_rate": 9.34969617343018e-05,
+      "loss": 0.2051,
+      "step": 4120
+    },
+    {
+      "grad_norm": 0.64734947681427,
+      "learning_rate": 9.345613112325122e-05,
+      "loss": 0.1893,
+      "step": 4130
+    },
+    {
+      "grad_norm": 0.5278508067131042,
+      "learning_rate": 9.34151817047975e-05,
+      "loss": 0.2062,
+      "step": 4140
+    },
+    {
+      "grad_norm": 0.5165694355964661,
+      "learning_rate": 9.33741135908948e-05,
+      "loss": 0.2118,
+      "step": 4150
+    },
+    {
+      "grad_norm": 0.43432632088661194,
+      "learning_rate": 9.33329268938218e-05,
+      "loss": 0.1982,
+      "step": 4160
+    },
+    {
+      "grad_norm": 0.7682166695594788,
+      "learning_rate": 9.329162172618132e-05,
+      "loss": 0.1994,
+      "step": 4170
+    },
+    {
+      "grad_norm": 0.44178205728530884,
+      "learning_rate": 9.325019820090013e-05,
+      "loss": 0.2128,
+      "step": 4180
+    },
+    {
+      "grad_norm": 0.5214252471923828,
+      "learning_rate": 9.320865643122855e-05,
+      "loss": 0.1991,
+      "step": 4190
+    },
+    {
+      "grad_norm": 0.40373316407203674,
+      "learning_rate": 9.316699653074023e-05,
+      "loss": 0.2036,
+      "step": 4200
+    },
+    {
+      "grad_norm": 0.5378400683403015,
+      "learning_rate": 9.312521861333172e-05,
+      "loss": 0.2026,
+      "step": 4210
+    },
+    {
+      "grad_norm": 0.628427267074585,
+      "learning_rate": 9.308332279322224e-05,
+      "loss": 0.1845,
+      "step": 4220
+    },
+    {
+      "grad_norm": 0.5630612969398499,
+      "learning_rate": 9.304130918495338e-05,
+      "loss": 0.2088,
+      "step": 4230
+    },
+    {
+      "grad_norm": 0.48832347989082336,
+      "learning_rate": 9.299917790338874e-05,
+      "loss": 0.2117,
+      "step": 4240
+    },
+    {
+      "grad_norm": 0.6269424557685852,
+      "learning_rate": 9.295692906371363e-05,
+      "loss": 0.1972,
+      "step": 4250
+    },
+    {
+      "grad_norm": 0.518443763256073,
+      "learning_rate": 9.291456278143476e-05,
+      "loss": 0.1904,
+      "step": 4260
+    },
+    {
+      "grad_norm": 0.40958812832832336,
+      "learning_rate": 9.287207917237994e-05,
+      "loss": 0.2005,
+      "step": 4270
+    },
+    {
+      "grad_norm": 0.513816237449646,
+      "learning_rate": 9.282947835269773e-05,
+      "loss": 0.2025,
+      "step": 4280
+    },
+    {
+      "grad_norm": 0.3779807388782501,
+      "learning_rate": 9.278676043885715e-05,
+      "loss": 0.1925,
+      "step": 4290
+    },
+    {
+      "grad_norm": 0.3904816508293152,
+      "learning_rate": 9.274392554764733e-05,
+      "loss": 0.1935,
+      "step": 4300
+    },
+    {
+      "grad_norm": 0.6148681640625,
+      "learning_rate": 9.270097379617723e-05,
+      "loss": 0.2036,
+      "step": 4310
+    },
+    {
+      "grad_norm": 0.554248034954071,
+      "learning_rate": 9.26579053018753e-05,
+      "loss": 0.1922,
+      "step": 4320
+    },
+    {
+      "grad_norm": 0.6833897233009338,
+      "learning_rate": 9.261472018248918e-05,
+      "loss": 0.1953,
+      "step": 4330
+    },
+    {
+      "grad_norm": 0.4256086051464081,
+      "learning_rate": 9.25714185560853e-05,
+      "loss": 0.2019,
+      "step": 4340
+    },
+    {
+      "grad_norm": 0.48727989196777344,
+      "learning_rate": 9.252800054104868e-05,
+      "loss": 0.2112,
+      "step": 4350
+    },
+    {
+      "grad_norm": 0.37245839834213257,
+      "learning_rate": 9.248446625608252e-05,
+      "loss": 0.2065,
+      "step": 4360
+    },
+    {
+      "grad_norm": 0.48356667160987854,
+      "learning_rate": 9.244081582020789e-05,
+      "loss": 0.1892,
+      "step": 4370
+    },
+    {
+      "grad_norm": 0.4223763942718506,
+      "learning_rate": 9.239704935276339e-05,
+      "loss": 0.1842,
+      "step": 4380
+    },
+    {
+      "grad_norm": 0.6383532881736755,
+      "learning_rate": 9.235316697340489e-05,
+      "loss": 0.1919,
+      "step": 4390
+    },
+    {
+      "grad_norm": 0.5077792406082153,
+      "learning_rate": 9.230916880210512e-05,
+      "loss": 0.2108,
+      "step": 4400
+    },
+    {
+      "grad_norm": 0.41724324226379395,
+      "learning_rate": 9.226505495915342e-05,
+      "loss": 0.1944,
+      "step": 4410
+    },
+    {
+      "grad_norm": 0.4914439916610718,
+      "learning_rate": 9.222082556515536e-05,
+      "loss": 0.1875,
+      "step": 4420
+    },
+    {
+      "grad_norm": 0.4365803003311157,
+      "learning_rate": 9.217648074103242e-05,
+      "loss": 0.2013,
+      "step": 4430
+    },
+    {
+      "grad_norm": 0.3719128370285034,
+      "learning_rate": 9.213202060802161e-05,
+      "loss": 0.1797,
+      "step": 4440
+    },
+    {
+      "grad_norm": 0.5573856830596924,
+      "learning_rate": 9.208744528767528e-05,
+      "loss": 0.1889,
+      "step": 4450
+    },
+    {
+      "grad_norm": 0.5137471556663513,
+      "learning_rate": 9.204275490186064e-05,
+      "loss": 0.1991,
+      "step": 4460
+    },
+    {
+      "grad_norm": 0.44392815232276917,
+      "learning_rate": 9.199794957275949e-05,
+      "loss": 0.1934,
+      "step": 4470
+    },
+    {
+      "grad_norm": 0.46090516448020935,
+      "learning_rate": 9.19530294228679e-05,
+      "loss": 0.1901,
+      "step": 4480
+    },
+    {
+      "grad_norm": 0.5551798939704895,
+      "learning_rate": 9.190799457499583e-05,
+      "loss": 0.2018,
+      "step": 4490
+    },
+    {
+      "grad_norm": 0.6372601389884949,
+      "learning_rate": 9.186284515226686e-05,
+      "loss": 0.1941,
+      "step": 4500
+    },
+    {
+      "grad_norm": 0.49452638626098633,
+      "learning_rate": 9.181758127811777e-05,
+      "loss": 0.1925,
+      "step": 4510
+    },
+    {
+      "grad_norm": 0.5816018581390381,
+      "learning_rate": 9.177220307629825e-05,
+      "loss": 0.185,
+      "step": 4520
+    },
+    {
+      "grad_norm": 0.5436152219772339,
+      "learning_rate": 9.172671067087059e-05,
+      "loss": 0.1865,
+      "step": 4530
+    },
+    {
+      "grad_norm": 0.5821353793144226,
+      "learning_rate": 9.16811041862093e-05,
+      "loss": 0.1959,
+      "step": 4540
+    },
+    {
+      "grad_norm": 0.6327117085456848,
+      "learning_rate": 9.163538374700076e-05,
+      "loss": 0.1968,
+      "step": 4550
+    },
+    {
+      "grad_norm": 0.40687280893325806,
+      "learning_rate": 9.158954947824287e-05,
+      "loss": 0.1896,
+      "step": 4560
+    },
+    {
+      "grad_norm": 0.5784546136856079,
+      "learning_rate": 9.154360150524482e-05,
+      "loss": 0.2084,
+      "step": 4570
+    },
+    {
+      "grad_norm": 0.5444551110267639,
+      "learning_rate": 9.14975399536266e-05,
+      "loss": 0.1996,
+      "step": 4580
+    },
+    {
+      "grad_norm": 0.7478448152542114,
+      "learning_rate": 9.14513649493187e-05,
+      "loss": 0.1934,
+      "step": 4590
+    },
+    {
+      "grad_norm": 0.4143761694431305,
+      "learning_rate": 9.140507661856187e-05,
+      "loss": 0.1869,
+      "step": 4600
+    },
+    {
+      "grad_norm": 0.6463064551353455,
+      "learning_rate": 9.135867508790661e-05,
+      "loss": 0.19,
+      "step": 4610
+    },
+    {
+      "grad_norm": 0.45436468720436096,
+      "learning_rate": 9.131216048421291e-05,
+      "loss": 0.2062,
+      "step": 4620
+    },
+    {
+      "grad_norm": 0.46877533197402954,
+      "learning_rate": 9.126553293464998e-05,
+      "loss": 0.2125,
+      "step": 4630
+    },
+    {
+      "grad_norm": 0.5249183773994446,
+      "learning_rate": 9.121879256669572e-05,
+      "loss": 0.2041,
+      "step": 4640
+    },
+    {
+      "grad_norm": 0.40862980484962463,
+      "learning_rate": 9.117193950813652e-05,
+      "loss": 0.203,
+      "step": 4650
+    },
+    {
+      "grad_norm": 0.43272387981414795,
+      "learning_rate": 9.112497388706685e-05,
+      "loss": 0.1874,
+      "step": 4660
+    },
+    {
+      "grad_norm": 0.5737069845199585,
+      "learning_rate": 9.10778958318889e-05,
+      "loss": 0.2055,
+      "step": 4670
+    },
+    {
+      "grad_norm": 0.36695626378059387,
+      "learning_rate": 9.103070547131232e-05,
+      "loss": 0.1737,
+      "step": 4680
+    },
+    {
+      "grad_norm": 0.43480822443962097,
+      "learning_rate": 9.098340293435375e-05,
+      "loss": 0.1819,
+      "step": 4690
+    },
+    {
+      "grad_norm": 0.5157934427261353,
+      "learning_rate": 9.093598835033649e-05,
+      "loss": 0.1806,
+      "step": 4700
+    },
+    {
+      "grad_norm": 0.46655765175819397,
+      "learning_rate": 9.088846184889021e-05,
+      "loss": 0.1936,
+      "step": 4710
+    },
+    {
+      "grad_norm": 0.34897446632385254,
+      "learning_rate": 9.084082355995057e-05,
+      "loss": 0.1951,
+      "step": 4720
+    },
+    {
+      "grad_norm": 0.622456967830658,
+      "learning_rate": 9.079307361375882e-05,
+      "loss": 0.2091,
+      "step": 4730
+    },
+    {
+      "grad_norm": 0.3473908305168152,
+      "learning_rate": 9.074521214086149e-05,
+      "loss": 0.1888,
+      "step": 4740
+    },
+    {
+      "grad_norm": 0.5895974636077881,
+      "learning_rate": 9.069723927211001e-05,
+      "loss": 0.2093,
+      "step": 4750
+    },
+    {
+      "grad_norm": 0.40156781673431396,
+      "learning_rate": 9.064915513866037e-05,
+      "loss": 0.1974,
+      "step": 4760
+    },
+    {
+      "grad_norm": 0.5745791792869568,
+      "learning_rate": 9.060095987197279e-05,
+      "loss": 0.1961,
+      "step": 4770
+    },
+    {
+      "grad_norm": 0.746356725692749,
+      "learning_rate": 9.055265360381126e-05,
+      "loss": 0.1869,
+      "step": 4780
+    },
+    {
+      "grad_norm": 0.8018269538879395,
+      "learning_rate": 9.050423646624326e-05,
+      "loss": 0.2196,
+      "step": 4790
+    },
+    {
+      "grad_norm": 0.48450300097465515,
+      "learning_rate": 9.045570859163943e-05,
+      "loss": 0.1985,
+      "step": 4800
+    },
+    {
+      "grad_norm": 0.4796624481678009,
+      "learning_rate": 9.04070701126731e-05,
+      "loss": 0.1957,
+      "step": 4810
+    },
+    {
+      "grad_norm": 0.47574514150619507,
+      "learning_rate": 9.035832116232001e-05,
+      "loss": 0.2004,
+      "step": 4820
+    },
+    {
+      "grad_norm": 0.4350902736186981,
+      "learning_rate": 9.030946187385796e-05,
+      "loss": 0.1889,
+      "step": 4830
+    },
+    {
+      "grad_norm": 0.5604615211486816,
+      "learning_rate": 9.026049238086635e-05,
+      "loss": 0.2024,
+      "step": 4840
+    },
+    {
+      "grad_norm": 0.6135896444320679,
+      "learning_rate": 9.021141281722591e-05,
+      "loss": 0.1977,
+      "step": 4850
+    },
+    {
+      "grad_norm": 0.41801658272743225,
+      "learning_rate": 9.01622233171183e-05,
+      "loss": 0.1778,
+      "step": 4860
+    },
+    {
+      "grad_norm": 0.49779269099235535,
+      "learning_rate": 9.011292401502574e-05,
+      "loss": 0.2078,
+      "step": 4870
+    },
+    {
+      "grad_norm": 0.5378538370132446,
+      "learning_rate": 9.006351504573063e-05,
+      "loss": 0.1992,
+      "step": 4880
+    },
+    {
+      "grad_norm": 0.41355544328689575,
+      "learning_rate": 9.001399654431519e-05,
+      "loss": 0.1787,
+      "step": 4890
+    },
+    {
+      "grad_norm": 0.5710010528564453,
+      "learning_rate": 8.996436864616116e-05,
+      "loss": 0.2008,
+      "step": 4900
+    },
+    {
+      "grad_norm": 1.0215823650360107,
+      "learning_rate": 8.991463148694925e-05,
+      "loss": 0.2187,
+      "step": 4910
+    },
+    {
+      "grad_norm": 0.5636600852012634,
+      "learning_rate": 8.986478520265902e-05,
+      "loss": 0.1964,
+      "step": 4920
+    },
+    {
+      "grad_norm": 0.4414118528366089,
+      "learning_rate": 8.981482992956827e-05,
+      "loss": 0.1962,
+      "step": 4930
+    },
+    {
+      "grad_norm": 0.602918803691864,
+      "learning_rate": 8.976476580425282e-05,
+      "loss": 0.203,
+      "step": 4940
+    },
+    {
+      "grad_norm": 0.5081776976585388,
+      "learning_rate": 8.971459296358606e-05,
+      "loss": 0.2296,
+      "step": 4950
+    },
+    {
+      "grad_norm": 0.5749547481536865,
+      "learning_rate": 8.966431154473864e-05,
+      "loss": 0.1993,
+      "step": 4960
+    },
+    {
+      "grad_norm": 0.556530237197876,
+      "learning_rate": 8.961392168517803e-05,
+      "loss": 0.1828,
+      "step": 4970
+    },
+    {
+      "grad_norm": 0.5410858988761902,
+      "learning_rate": 8.956342352266821e-05,
+      "loss": 0.2187,
+      "step": 4980
+    },
+    {
+      "grad_norm": 0.422293096780777,
+      "learning_rate": 8.95128171952692e-05,
+      "loss": 0.1799,
+      "step": 4990
+    },
+    {
+      "grad_norm": 0.37484434247016907,
+      "learning_rate": 8.946210284133676e-05,
+      "loss": 0.1897,
+      "step": 5000
+    },
+    {
+      "grad_norm": 0.5707630515098572,
+      "learning_rate": 8.941128059952201e-05,
+      "loss": 0.202,
+      "step": 5010
+    },
+    {
+      "grad_norm": 0.5029783248901367,
+      "learning_rate": 8.936035060877102e-05,
+      "loss": 0.1957,
+      "step": 5020
+    },
+    {
+      "grad_norm": 0.7676258683204651,
+      "learning_rate": 8.930931300832443e-05,
+      "loss": 0.1779,
+      "step": 5030
+    },
+    {
+      "grad_norm": 0.6760695576667786,
+      "learning_rate": 8.925816793771711e-05,
+      "loss": 0.2165,
+      "step": 5040
+    },
+    {
+      "grad_norm": 0.44495904445648193,
+      "learning_rate": 8.92069155367777e-05,
+      "loss": 0.2082,
+      "step": 5050
+    },
+    {
+      "grad_norm": 0.45802170038223267,
+      "learning_rate": 8.915555594562834e-05,
+      "loss": 0.1899,
+      "step": 5060
+    },
+    {
+      "grad_norm": 0.545892059803009,
+      "learning_rate": 8.910408930468416e-05,
+      "loss": 0.2007,
+      "step": 5070
+    },
+    {
+      "grad_norm": 0.5191405415534973,
+      "learning_rate": 8.905251575465303e-05,
+      "loss": 0.225,
+      "step": 5080
+    },
+    {
+      "grad_norm": 0.40124520659446716,
+      "learning_rate": 8.900083543653502e-05,
+      "loss": 0.2086,
+      "step": 5090
+    },
+    {
+      "grad_norm": 0.8918811678886414,
+      "learning_rate": 8.894904849162218e-05,
+      "loss": 0.2106,
+      "step": 5100
+    },
+    {
+      "grad_norm": 0.47879716753959656,
+      "learning_rate": 8.889715506149802e-05,
+      "loss": 0.2169,
+      "step": 5110
+    },
+    {
+      "grad_norm": 0.4863725006580353,
+      "learning_rate": 8.884515528803722e-05,
+      "loss": 0.2015,
+      "step": 5120
+    },
+    {
+      "grad_norm": 0.4220241904258728,
+      "learning_rate": 8.879304931340517e-05,
+      "loss": 0.2108,
+      "step": 5130
+    },
+    {
+      "grad_norm": 0.518157958984375,
+      "learning_rate": 8.874083728005759e-05,
+      "loss": 0.1964,
+      "step": 5140
+    },
+    {
+      "grad_norm": 0.5109302401542664,
+      "learning_rate": 8.868851933074021e-05,
+      "loss": 0.2183,
+      "step": 5150
+    },
+    {
+      "grad_norm": 0.4269387722015381,
+      "learning_rate": 8.863609560848829e-05,
+      "loss": 0.2003,
+      "step": 5160
+    },
+    {
+      "grad_norm": 0.395548015832901,
+      "learning_rate": 8.85835662566263e-05,
+      "loss": 0.2046,
+      "step": 5170
+    },
+    {
+      "grad_norm": 0.4899046719074249,
+      "learning_rate": 8.853093141876747e-05,
+      "loss": 0.2081,
+      "step": 5180
+    },
+    {
+      "grad_norm": 0.6365777850151062,
+      "learning_rate": 8.847819123881343e-05,
+      "loss": 0.2131,
+      "step": 5190
+    },
+    {
+      "grad_norm": 0.5763066411018372,
+      "learning_rate": 8.842534586095383e-05,
+      "loss": 0.2004,
+      "step": 5200
+    },
+    {
+      "grad_norm": 0.5714003443717957,
+      "learning_rate": 8.837239542966593e-05,
+      "loss": 0.2027,
+      "step": 5210
+    },
+    {
+      "grad_norm": 0.673712432384491,
+      "learning_rate": 8.831934008971417e-05,
+      "loss": 0.2124,
+      "step": 5220
+    },
+    {
+      "grad_norm": 0.46018970012664795,
+      "learning_rate": 8.826617998614982e-05,
+      "loss": 0.21,
+      "step": 5230
+    },
+    {
+      "grad_norm": 0.725900411605835,
+      "learning_rate": 8.821291526431056e-05,
+      "loss": 0.202,
+      "step": 5240
+    },
+    {
+      "grad_norm": 0.441424697637558,
+      "learning_rate": 8.815954606982015e-05,
+      "loss": 0.1979,
+      "step": 5250
+    },
+    {
+      "grad_norm": 0.5047786235809326,
+      "learning_rate": 8.810607254858789e-05,
+      "loss": 0.2185,
+      "step": 5260
+    },
+    {
+      "grad_norm": 0.4661320745944977,
+      "learning_rate": 8.805249484680838e-05,
+      "loss": 0.1939,
+      "step": 5270
+    },
+    {
+      "grad_norm": 0.5545666813850403,
+      "learning_rate": 8.799881311096096e-05,
+      "loss": 0.2299,
+      "step": 5280
+    },
+    {
+      "grad_norm": 0.54994136095047,
+      "learning_rate": 8.794502748780949e-05,
+      "loss": 0.2238,
+      "step": 5290
+    },
+    {
+      "grad_norm": 0.3950221836566925,
+      "learning_rate": 8.78911381244018e-05,
+      "loss": 0.199,
+      "step": 5300
+    },
+    {
+      "grad_norm": 0.5033600926399231,
+      "learning_rate": 8.783714516806933e-05,
+      "loss": 0.1942,
+      "step": 5310
+    },
+    {
+      "grad_norm": 0.6238504648208618,
+      "learning_rate": 8.77830487664268e-05,
+      "loss": 0.1966,
+      "step": 5320
+    },
+    {
+      "grad_norm": 0.5857625007629395,
+      "learning_rate": 8.772884906737167e-05,
+      "loss": 0.2307,
+      "step": 5330
+    },
+    {
+      "grad_norm": 0.4198428690433502,
+      "learning_rate": 8.767454621908387e-05,
+      "loss": 0.2006,
+      "step": 5340
+    },
+    {
+      "grad_norm": 0.6497552990913391,
+      "learning_rate": 8.76201403700253e-05,
+      "loss": 0.193,
+      "step": 5350
+    },
+    {
+      "grad_norm": 0.44059914350509644,
+      "learning_rate": 8.756563166893949e-05,
+      "loss": 0.1998,
+      "step": 5360
+    },
+    {
+      "grad_norm": 0.5056824684143066,
+      "learning_rate": 8.751102026485113e-05,
+      "loss": 0.2068,
+      "step": 5370
+    },
+    {
+      "grad_norm": 0.6478718519210815,
+      "learning_rate": 8.745630630706571e-05,
+      "loss": 0.2054,
+      "step": 5380
+    },
+    {
+      "grad_norm": 0.42337265610694885,
+      "learning_rate": 8.740148994516912e-05,
+      "loss": 0.1973,
+      "step": 5390
+    },
+    {
+      "grad_norm": 0.5339463949203491,
+      "learning_rate": 8.73465713290272e-05,
+      "loss": 0.2085,
+      "step": 5400
+    },
+    {
+      "grad_norm": 0.5710785984992981,
+      "learning_rate": 8.729155060878533e-05,
+      "loss": 0.1961,
+      "step": 5410
+    },
+    {
+      "grad_norm": 0.4668373465538025,
+      "learning_rate": 8.723642793486809e-05,
+      "loss": 0.2027,
+      "step": 5420
+    },
+    {
+      "grad_norm": 0.4218912422657013,
+      "learning_rate": 8.718120345797873e-05,
+      "loss": 0.2124,
+      "step": 5430
+    },
+    {
+      "grad_norm": 0.5655269026756287,
+      "learning_rate": 8.712587732909889e-05,
+      "loss": 0.2155,
+      "step": 5440
+    },
+    {
+      "grad_norm": 0.4211824834346771,
+      "learning_rate": 8.707044969948806e-05,
+      "loss": 0.2161,
+      "step": 5450
+    },
+    {
+      "grad_norm": 0.45340895652770996,
+      "learning_rate": 8.701492072068329e-05,
+      "loss": 0.1837,
+      "step": 5460
+    },
+    {
+      "grad_norm": 0.49008607864379883,
+      "learning_rate": 8.695929054449869e-05,
+      "loss": 0.1994,
+      "step": 5470
+    },
+    {
+      "grad_norm": 0.5747946500778198,
+      "learning_rate": 8.690355932302501e-05,
+      "loss": 0.1986,
+      "step": 5480
+    },
+    {
+      "grad_norm": 0.7699674367904663,
+      "learning_rate": 8.684772720862931e-05,
+      "loss": 0.2106,
+      "step": 5490
+    },
+    {
+      "grad_norm": 0.5051481127738953,
+      "learning_rate": 8.679179435395446e-05,
+      "loss": 0.1897,
+      "step": 5500
+    },
+    {
+      "grad_norm": 0.5317304134368896,
+      "learning_rate": 8.673576091191874e-05,
+      "loss": 0.1956,
+      "step": 5510
+    },
+    {
+      "grad_norm": 0.5544655919075012,
+      "learning_rate": 8.667962703571541e-05,
+      "loss": 0.1978,
+      "step": 5520
+    },
+    {
+      "grad_norm": 0.46938788890838623,
+      "learning_rate": 8.662339287881238e-05,
+      "loss": 0.201,
+      "step": 5530
+    },
+    {
+      "grad_norm": 0.6373662352561951,
+      "learning_rate": 8.656705859495169e-05,
+      "loss": 0.2102,
+      "step": 5540
+    },
+    {
+      "grad_norm": 0.9178352952003479,
+      "learning_rate": 8.651062433814912e-05,
+      "loss": 0.2106,
+      "step": 5550
+    },
+    {
+      "grad_norm": 0.6245887279510498,
+      "learning_rate": 8.645409026269375e-05,
+      "loss": 0.1913,
+      "step": 5560
+    },
+    {
+      "grad_norm": 0.4351654350757599,
+      "learning_rate": 8.639745652314759e-05,
+      "loss": 0.2018,
+      "step": 5570
+    },
+    {
+      "grad_norm": 0.6059114933013916,
+      "learning_rate": 8.634072327434515e-05,
+      "loss": 0.1972,
+      "step": 5580
+    },
+    {
+      "grad_norm": 0.43515288829803467,
+      "learning_rate": 8.628389067139294e-05,
+      "loss": 0.2014,
+      "step": 5590
+    },
+    {
+      "grad_norm": 0.47767817974090576,
+      "learning_rate": 8.622695886966911e-05,
+      "loss": 0.2088,
+      "step": 5600
+    },
+    {
+      "grad_norm": 0.3646154999732971,
+      "learning_rate": 8.616992802482308e-05,
+      "loss": 0.1897,
+      "step": 5610
+    },
+    {
+      "grad_norm": 0.5689325928688049,
+      "learning_rate": 8.611279829277496e-05,
+      "loss": 0.1909,
+      "step": 5620
+    },
+    {
+      "grad_norm": 1.5341813564300537,
+      "learning_rate": 8.605556982971528e-05,
+      "loss": 0.233,
+      "step": 5630
+    },
+    {
+      "grad_norm": 0.39875251054763794,
+      "learning_rate": 8.599824279210447e-05,
+      "loss": 0.199,
+      "step": 5640
+    },
+    {
+      "grad_norm": 0.3836033046245575,
+      "learning_rate": 8.594081733667243e-05,
+      "loss": 0.1948,
+      "step": 5650
+    },
+    {
+      "grad_norm": 0.45160359144210815,
+      "learning_rate": 8.58832936204182e-05,
+      "loss": 0.1858,
+      "step": 5660
+    },
+    {
+      "grad_norm": 0.44377389550209045,
+      "learning_rate": 8.582567180060942e-05,
+      "loss": 0.2095,
+      "step": 5670
+    },
+    {
+      "grad_norm": 0.6009142994880676,
+      "learning_rate": 8.576795203478194e-05,
+      "loss": 0.2089,
+      "step": 5680
+    },
+    {
+      "grad_norm": 0.500217854976654,
+      "learning_rate": 8.571013448073939e-05,
+      "loss": 0.2008,
+      "step": 5690
+    },
+    {
+      "grad_norm": 0.4467677175998688,
+      "learning_rate": 8.565221929655275e-05,
+      "loss": 0.2164,
+      "step": 5700
+    },
+    {
+      "grad_norm": 0.4804268777370453,
+      "learning_rate": 8.559420664055992e-05,
+      "loss": 0.205,
+      "step": 5710
+    },
+    {
+      "grad_norm": 0.5384911298751831,
+      "learning_rate": 8.553609667136532e-05,
+      "loss": 0.2001,
+      "step": 5720
+    },
+    {
+      "grad_norm": 0.49594050645828247,
+      "learning_rate": 8.547788954783936e-05,
+      "loss": 0.1906,
+      "step": 5730
+    },
+    {
+      "grad_norm": 0.581782877445221,
+      "learning_rate": 8.541958542911808e-05,
+      "loss": 0.1979,
+      "step": 5740
+    },
+    {
+      "grad_norm": 0.46274513006210327,
+      "learning_rate": 8.536118447460275e-05,
+      "loss": 0.212,
+      "step": 5750
+    },
+    {
+      "grad_norm": 0.5406103730201721,
+      "learning_rate": 8.530268684395932e-05,
+      "loss": 0.2024,
+      "step": 5760
+    },
+    {
+      "grad_norm": 0.5763973593711853,
+      "learning_rate": 8.524409269711807e-05,
+      "loss": 0.2216,
+      "step": 5770
+    },
+    {
+      "grad_norm": 0.530036449432373,
+      "learning_rate": 8.51854021942732e-05,
+      "loss": 0.1944,
+      "step": 5780
+    },
+    {
+      "grad_norm": 0.43865275382995605,
+      "learning_rate": 8.512661549588227e-05,
+      "loss": 0.1957,
+      "step": 5790
+    },
+    {
+      "grad_norm": 0.5761095881462097,
+      "learning_rate": 8.506773276266588e-05,
+      "loss": 0.1992,
+      "step": 5800
+    },
+    {
+      "grad_norm": 0.4627121090888977,
+      "learning_rate": 8.500875415560721e-05,
+      "loss": 0.2166,
+      "step": 5810
+    },
+    {
+      "grad_norm": 0.4544637203216553,
+      "learning_rate": 8.494967983595144e-05,
+      "loss": 0.2039,
+      "step": 5820
+    },
+    {
+      "grad_norm": 0.6750137209892273,
+      "learning_rate": 8.489050996520558e-05,
+      "loss": 0.1995,
+      "step": 5830
+    },
+    {
+      "grad_norm": 0.5190580487251282,
+      "learning_rate": 8.483124470513775e-05,
+      "loss": 0.1964,
+      "step": 5840
+    },
+    {
+      "grad_norm": 0.3574702739715576,
+      "learning_rate": 8.477188421777692e-05,
+      "loss": 0.2048,
+      "step": 5850
+    },
+    {
+      "grad_norm": 0.598330020904541,
+      "learning_rate": 8.47124286654124e-05,
+      "loss": 0.1962,
+      "step": 5860
+    },
+    {
+      "grad_norm": 0.3662823736667633,
+      "learning_rate": 8.465287821059341e-05,
+      "loss": 0.1854,
+      "step": 5870
+    },
+    {
+      "grad_norm": 0.4132087826728821,
+      "learning_rate": 8.45932330161286e-05,
+      "loss": 0.1975,
+      "step": 5880
+    },
+    {
+      "grad_norm": 0.49038878083229065,
+      "learning_rate": 8.453349324508567e-05,
+      "loss": 0.1872,
+      "step": 5890
+    },
+    {
+      "grad_norm": 0.8267894983291626,
+      "learning_rate": 8.447365906079088e-05,
+      "loss": 0.198,
+      "step": 5900
+    },
+    {
+      "grad_norm": 0.5228694081306458,
+      "learning_rate": 8.441373062682856e-05,
+      "loss": 0.221,
+      "step": 5910
+    },
+    {
+      "grad_norm": 0.6964914202690125,
+      "learning_rate": 8.43537081070408e-05,
+      "loss": 0.1986,
+      "step": 5920
+    },
+    {
+      "grad_norm": 0.6608155369758606,
+      "learning_rate": 8.429359166552689e-05,
+      "loss": 0.1952,
+      "step": 5930
+    },
+    {
+      "grad_norm": 0.4782712459564209,
+      "learning_rate": 8.423338146664284e-05,
+      "loss": 0.1961,
+      "step": 5940
+    },
+    {
+      "grad_norm": 0.7551800608634949,
+      "learning_rate": 8.417307767500107e-05,
+      "loss": 0.2109,
+      "step": 5950
+    },
+    {
+      "grad_norm": 0.45815548300743103,
+      "learning_rate": 8.411268045546983e-05,
+      "loss": 0.1974,
+      "step": 5960
+    },
+    {
+      "grad_norm": 0.5458553433418274,
+      "learning_rate": 8.405218997317281e-05,
+      "loss": 0.2006,
+      "step": 5970
+    },
+    {
+      "grad_norm": 0.5553959012031555,
+      "learning_rate": 8.399160639348869e-05,
+      "loss": 0.2117,
+      "step": 5980
+    },
+    {
+      "grad_norm": 0.5195659399032593,
+      "learning_rate": 8.393092988205065e-05,
+      "loss": 0.1997,
+      "step": 5990
+    },
+    {
+      "grad_norm": 0.5967625379562378,
+      "learning_rate": 8.387016060474597e-05,
+      "loss": 0.196,
+      "step": 6000
+    },
+    {
+      "grad_norm": 0.5088404417037964,
+      "learning_rate": 8.380929872771551e-05,
+      "loss": 0.1904,
+      "step": 6010
+    },
+    {
+      "grad_norm": 0.5641393065452576,
+      "learning_rate": 8.374834441735335e-05,
+      "loss": 0.2037,
+      "step": 6020
+    },
+    {
+      "grad_norm": 0.4933110773563385,
+      "learning_rate": 8.368729784030622e-05,
+      "loss": 0.1922,
+      "step": 6030
+    },
+    {
+      "grad_norm": 0.41249117255210876,
+      "learning_rate": 8.362615916347315e-05,
+      "loss": 0.1868,
+      "step": 6040
+    },
+    {
+      "grad_norm": 0.6308994889259338,
+      "learning_rate": 8.356492855400493e-05,
+      "loss": 0.1998,
+      "step": 6050
+    },
+    {
+      "grad_norm": 0.5501776933670044,
+      "learning_rate": 8.350360617930371e-05,
+      "loss": 0.173,
+      "step": 6060
+    },
+    {
+      "grad_norm": 0.46614089608192444,
+      "learning_rate": 8.344219220702255e-05,
+      "loss": 0.2206,
+      "step": 6070
+    },
+    {
+      "grad_norm": 0.6035048961639404,
+      "learning_rate": 8.338068680506485e-05,
+      "loss": 0.1996,
+      "step": 6080
+    },
+    {
+      "grad_norm": 0.4649960696697235,
+      "learning_rate": 8.33190901415841e-05,
+      "loss": 0.188,
+      "step": 6090
+    },
+    {
+      "grad_norm": 0.46320343017578125,
+      "learning_rate": 8.325740238498317e-05,
+      "loss": 0.1923,
+      "step": 6100
+    },
+    {
+      "grad_norm": 0.5669880509376526,
+      "learning_rate": 8.319562370391406e-05,
+      "loss": 0.1853,
+      "step": 6110
+    },
+    {
+      "grad_norm": 0.6290435194969177,
+      "learning_rate": 8.31337542672773e-05,
+      "loss": 0.2109,
+      "step": 6120
+    },
+    {
+      "grad_norm": 0.49063247442245483,
+      "learning_rate": 8.307179424422158e-05,
+      "loss": 0.2081,
+      "step": 6130
+    },
+    {
+      "grad_norm": 1.261579155921936,
+      "learning_rate": 8.300974380414327e-05,
+      "loss": 0.2088,
+      "step": 6140
+    },
+    {
+      "grad_norm": 0.45252230763435364,
+      "learning_rate": 8.294760311668586e-05,
+      "loss": 0.1957,
+      "step": 6150
+    },
+    {
+      "grad_norm": 0.45189493894577026,
+      "learning_rate": 8.288537235173961e-05,
+      "loss": 0.2242,
+      "step": 6160
+    },
+    {
+      "grad_norm": 0.5580816268920898,
+      "learning_rate": 8.282305167944108e-05,
+      "loss": 0.2166,
+      "step": 6170
+    },
+    {
+      "grad_norm": 0.591963529586792,
+      "learning_rate": 8.276064127017262e-05,
+      "loss": 0.2042,
+      "step": 6180
+    },
+    {
+      "grad_norm": 0.36994874477386475,
+      "learning_rate": 8.269814129456189e-05,
+      "loss": 0.1992,
+      "step": 6190
+    },
+    {
+      "grad_norm": 0.32167744636535645,
+      "learning_rate": 8.263555192348143e-05,
+      "loss": 0.203,
+      "step": 6200
+    },
+    {
+      "grad_norm": 0.734040379524231,
+      "learning_rate": 8.257287332804819e-05,
+      "loss": 0.2047,
+      "step": 6210
+    },
+    {
+      "grad_norm": 0.5754173398017883,
+      "learning_rate": 8.251010567962307e-05,
+      "loss": 0.2084,
+      "step": 6220
+    },
+    {
+      "grad_norm": 0.5412654876708984,
+      "learning_rate": 8.244724914981041e-05,
+      "loss": 0.2276,
+      "step": 6230
+    },
+    {
+      "grad_norm": 0.4913170039653778,
+      "learning_rate": 8.238430391045757e-05,
+      "loss": 0.1902,
+      "step": 6240
+    },
+    {
+      "grad_norm": 0.8557559847831726,
+      "learning_rate": 8.232127013365445e-05,
+      "loss": 0.2244,
+      "step": 6250
+    },
+    {
+      "grad_norm": 0.48343992233276367,
+      "learning_rate": 8.225814799173295e-05,
+      "loss": 0.2051,
+      "step": 6260
+    },
+    {
+      "grad_norm": 0.4791530668735504,
+      "learning_rate": 8.219493765726663e-05,
+      "loss": 0.198,
+      "step": 6270
+    },
+    {
+      "grad_norm": 0.5592235922813416,
+      "learning_rate": 8.21316393030701e-05,
+      "loss": 0.2,
+      "step": 6280
+    },
+    {
+      "grad_norm": 0.5867027640342712,
+      "learning_rate": 8.206825310219865e-05,
+      "loss": 0.2124,
+      "step": 6290
+    },
+    {
+      "grad_norm": 0.6531945466995239,
+      "learning_rate": 8.200477922794776e-05,
+      "loss": 0.2144,
+      "step": 6300
+    },
+    {
+      "grad_norm": 0.5493502020835876,
+      "learning_rate": 8.194121785385256e-05,
+      "loss": 0.2028,
+      "step": 6310
+    },
+    {
+      "grad_norm": 0.5517439246177673,
+      "learning_rate": 8.187756915368741e-05,
+      "loss": 0.2045,
+      "step": 6320
+    },
+    {
+      "grad_norm": 0.5531640648841858,
+      "learning_rate": 8.181383330146544e-05,
+      "loss": 0.2157,
+      "step": 6330
+    },
+    {
+      "grad_norm": 0.4700435698032379,
+      "learning_rate": 8.175001047143804e-05,
+      "loss": 0.1942,
+      "step": 6340
+    },
+    {
+      "grad_norm": 0.544187068939209,
+      "learning_rate": 8.168610083809438e-05,
+      "loss": 0.2223,
+      "step": 6350
+    },
+    {
+      "grad_norm": 1.0072758197784424,
+      "learning_rate": 8.162210457616095e-05,
+      "loss": 0.2243,
+      "step": 6360
+    },
+    {
+      "grad_norm": 0.7184207439422607,
+      "learning_rate": 8.155802186060109e-05,
+      "loss": 0.1985,
+      "step": 6370
+    },
+    {
+      "grad_norm": 0.47824904322624207,
+      "learning_rate": 8.149385286661453e-05,
+      "loss": 0.2103,
+      "step": 6380
+    },
+    {
+      "grad_norm": 0.564469039440155,
+      "learning_rate": 8.14295977696368e-05,
+      "loss": 0.2024,
+      "step": 6390
+    },
+    {
+      "grad_norm": 0.36813125014305115,
+      "learning_rate": 8.13652567453389e-05,
+      "loss": 0.2045,
+      "step": 6400
+    },
+    {
+      "grad_norm": 0.5051242113113403,
+      "learning_rate": 8.130082996962676e-05,
+      "loss": 0.2102,
+      "step": 6410
+    },
+    {
+      "grad_norm": 0.5781051516532898,
+      "learning_rate": 8.123631761864068e-05,
+      "loss": 0.2059,
+      "step": 6420
+    },
+    {
+      "grad_norm": 0.5968287587165833,
+      "learning_rate": 8.1171719868755e-05,
+      "loss": 0.2009,
+      "step": 6430
+    },
+    {
+      "grad_norm": 0.530032753944397,
+      "learning_rate": 8.110703689657748e-05,
+      "loss": 0.2019,
+      "step": 6440
+    },
+    {
+      "grad_norm": 0.5196192860603333,
+      "learning_rate": 8.104226887894892e-05,
+      "loss": 0.1898,
+      "step": 6450
+    },
+    {
+      "grad_norm": 0.5619369149208069,
+      "learning_rate": 8.097741599294257e-05,
+      "loss": 0.2024,
+      "step": 6460
+    },
+    {
+      "grad_norm": 0.4498804211616516,
+      "learning_rate": 8.091247841586378e-05,
+      "loss": 0.2018,
+      "step": 6470
+    },
+    {
+      "grad_norm": 0.5622605681419373,
+      "learning_rate": 8.084745632524939e-05,
+      "loss": 0.2042,
+      "step": 6480
+    },
+    {
+      "grad_norm": 0.621238648891449,
+      "learning_rate": 8.07823498988673e-05,
+      "loss": 0.2155,
+      "step": 6490
+    },
+    {
+      "grad_norm": 0.369157999753952,
+      "learning_rate": 8.071715931471602e-05,
+      "loss": 0.2006,
+      "step": 6500
+    },
+    {
+      "grad_norm": 0.6119118332862854,
+      "learning_rate": 8.06518847510241e-05,
+      "loss": 0.2183,
+      "step": 6510
+    },
+    {
+      "grad_norm": 0.45959708094596863,
+      "learning_rate": 8.058652638624971e-05,
+      "loss": 0.1975,
+      "step": 6520
+    },
+    {
+      "grad_norm": 0.5493879914283752,
+      "learning_rate": 8.052108439908013e-05,
+      "loss": 0.1919,
+      "step": 6530
+    },
+    {
+      "grad_norm": 0.3636343777179718,
+      "learning_rate": 8.045555896843125e-05,
+      "loss": 0.2002,
+      "step": 6540
+    },
+    {
+      "grad_norm": 0.5918999314308167,
+      "learning_rate": 8.03899502734471e-05,
+      "loss": 0.2082,
+      "step": 6550
+    },
+    {
+      "grad_norm": 0.5676548480987549,
+      "learning_rate": 8.032425849349931e-05,
+      "loss": 0.1992,
+      "step": 6560
+    },
+    {
+      "grad_norm": 0.6158645749092102,
+      "learning_rate": 8.025848380818674e-05,
+      "loss": 0.2085,
+      "step": 6570
+    },
+    {
+      "grad_norm": 0.5648828744888306,
+      "learning_rate": 8.019262639733487e-05,
+      "loss": 0.2102,
+      "step": 6580
+    },
+    {
+      "grad_norm": 0.6536079049110413,
+      "learning_rate": 8.012668644099531e-05,
+      "loss": 0.2048,
+      "step": 6590
+    },
+    {
+      "grad_norm": 0.5085182785987854,
+      "learning_rate": 8.006066411944542e-05,
+      "loss": 0.2078,
+      "step": 6600
+    },
+    {
+      "grad_norm": 0.4610349237918854,
+      "learning_rate": 7.999455961318769e-05,
+      "loss": 0.2064,
+      "step": 6610
+    },
+    {
+      "grad_norm": 0.4900440275669098,
+      "learning_rate": 7.992837310294932e-05,
+      "loss": 0.2084,
+      "step": 6620
+    },
+    {
+      "grad_norm": 0.5451095700263977,
+      "learning_rate": 7.986210476968167e-05,
+      "loss": 0.1938,
+      "step": 6630
+    },
+    {
+      "grad_norm": 0.5785449743270874,
+      "learning_rate": 7.97957547945599e-05,
+      "loss": 0.1816,
+      "step": 6640
+    },
+    {
+      "grad_norm": 0.5497417449951172,
+      "learning_rate": 7.972932335898226e-05,
+      "loss": 0.2015,
+      "step": 6650
+    },
+    {
+      "grad_norm": 0.45251306891441345,
+      "learning_rate": 7.966281064456975e-05,
+      "loss": 0.191,
+      "step": 6660
+    },
+    {
+      "grad_norm": 0.4512160122394562,
+      "learning_rate": 7.959621683316563e-05,
+      "loss": 0.208,
+      "step": 6670
+    },
+    {
+      "grad_norm": 0.46106255054473877,
+      "learning_rate": 7.952954210683481e-05,
+      "loss": 0.1961,
+      "step": 6680
+    },
+    {
+      "grad_norm": 0.450971394777298,
+      "learning_rate": 7.946278664786345e-05,
+      "loss": 0.1872,
+      "step": 6690
+    },
+    {
+      "grad_norm": 0.5406850576400757,
+      "learning_rate": 7.939595063875842e-05,
+      "loss": 0.1776,
+      "step": 6700
+    },
+    {
+      "grad_norm": 0.5689697861671448,
+      "learning_rate": 7.932903426224683e-05,
+      "loss": 0.2034,
+      "step": 6710
+    },
+    {
+      "grad_norm": 0.4658198058605194,
+      "learning_rate": 7.926203770127552e-05,
+      "loss": 0.1722,
+      "step": 6720
+    },
+    {
+      "grad_norm": 0.757060706615448,
+      "learning_rate": 7.919496113901046e-05,
+      "loss": 0.1889,
+      "step": 6730
+    },
+    {
+      "grad_norm": 0.49916157126426697,
+      "learning_rate": 7.912780475883649e-05,
+      "loss": 0.1843,
+      "step": 6740
+    },
+    {
+      "grad_norm": 0.48528656363487244,
+      "learning_rate": 7.906056874435652e-05,
+      "loss": 0.1989,
+      "step": 6750
+    },
+    {
+      "grad_norm": 0.4794119894504547,
+      "learning_rate": 7.899325327939131e-05,
+      "loss": 0.1879,
+      "step": 6760
+    },
+    {
+      "grad_norm": 0.5329006910324097,
+      "learning_rate": 7.892585854797872e-05,
+      "loss": 0.1935,
+      "step": 6770
+    },
+    {
+      "grad_norm": 0.5803582668304443,
+      "learning_rate": 7.88583847343734e-05,
+      "loss": 0.1764,
+      "step": 6780
+    },
+    {
+      "grad_norm": 0.5917492508888245,
+      "learning_rate": 7.879083202304616e-05,
+      "loss": 0.1836,
+      "step": 6790
+    },
+    {
+      "grad_norm": 0.5864911079406738,
+      "learning_rate": 7.872320059868355e-05,
+      "loss": 0.1922,
+      "step": 6800
+    },
+    {
+      "grad_norm": 0.682766854763031,
+      "learning_rate": 7.865549064618729e-05,
+      "loss": 0.2001,
+      "step": 6810
+    },
+    {
+      "grad_norm": 0.579669177532196,
+      "learning_rate": 7.858770235067381e-05,
+      "loss": 0.1962,
+      "step": 6820
+    },
+    {
+      "grad_norm": 0.49040183424949646,
+      "learning_rate": 7.851983589747374e-05,
+      "loss": 0.2011,
+      "step": 6830
+    },
+    {
+      "grad_norm": 0.6946597695350647,
+      "learning_rate": 7.845189147213133e-05,
+      "loss": 0.1844,
+      "step": 6840
+    },
+    {
+      "grad_norm": 0.7063781023025513,
+      "learning_rate": 7.838386926040407e-05,
+      "loss": 0.2036,
+      "step": 6850
+    },
+    {
+      "grad_norm": 0.6515227556228638,
+      "learning_rate": 7.83157694482621e-05,
+      "loss": 0.1967,
+      "step": 6860
+    },
+    {
+      "grad_norm": 0.5737599730491638,
+      "learning_rate": 7.824759222188768e-05,
+      "loss": 0.2021,
+      "step": 6870
+    },
+    {
+      "grad_norm": 0.4795118570327759,
+      "learning_rate": 7.817933776767478e-05,
+      "loss": 0.1891,
+      "step": 6880
+    },
+    {
+      "grad_norm": 0.4484804570674896,
+      "learning_rate": 7.811100627222842e-05,
+      "loss": 0.2025,
+      "step": 6890
+    },
+    {
+      "grad_norm": 0.4386521577835083,
+      "learning_rate": 7.804259792236435e-05,
+      "loss": 0.1817,
+      "step": 6900
+    },
+    {
+      "grad_norm": 0.45415475964546204,
+      "learning_rate": 7.797411290510835e-05,
+      "loss": 0.1793,
+      "step": 6910
+    },
+    {
+      "grad_norm": 0.47173362970352173,
+      "learning_rate": 7.790555140769586e-05,
+      "loss": 0.1826,
+      "step": 6920
+    },
+    {
+      "grad_norm": 0.9042406678199768,
+      "learning_rate": 7.78369136175714e-05,
+      "loss": 0.2089,
+      "step": 6930
+    },
+    {
+      "grad_norm": 0.902664840221405,
+      "learning_rate": 7.776819972238806e-05,
+      "loss": 0.2213,
+      "step": 6940
+    },
+    {
+      "grad_norm": 0.4913310408592224,
+      "learning_rate": 7.7699409910007e-05,
+      "loss": 0.1998,
+      "step": 6950
+    },
+    {
+      "grad_norm": 0.6368779540061951,
+      "learning_rate": 7.763054436849694e-05,
+      "loss": 0.1913,
+      "step": 6960
+    },
+    {
+      "grad_norm": 0.4483477771282196,
+      "learning_rate": 7.756160328613364e-05,
+      "loss": 0.1922,
+      "step": 6970
+    },
+    {
+      "grad_norm": 0.5121405720710754,
+      "learning_rate": 7.749258685139942e-05,
+      "loss": 0.1913,
+      "step": 6980
+    },
+    {
+      "grad_norm": 0.5345188975334167,
+      "learning_rate": 7.742349525298253e-05,
+      "loss": 0.1877,
+      "step": 6990
+    },
+    {
+      "grad_norm": 0.5706868767738342,
+      "learning_rate": 7.735432867977679e-05,
+      "loss": 0.1879,
+      "step": 7000
+    },
+    {
+      "grad_norm": 0.536948561668396,
+      "learning_rate": 7.728508732088096e-05,
+      "loss": 0.1958,
+      "step": 7010
+    },
+    {
+      "grad_norm": 0.5245667695999146,
+      "learning_rate": 7.721577136559825e-05,
+      "loss": 0.2019,
+      "step": 7020
+    },
+    {
+      "grad_norm": 0.6903913617134094,
+      "learning_rate": 7.714638100343588e-05,
+      "loss": 0.1969,
+      "step": 7030
+    },
+    {
+      "grad_norm": 0.46299195289611816,
+      "learning_rate": 7.707691642410444e-05,
+      "loss": 0.19,
+      "step": 7040
+    },
+    {
+      "grad_norm": 0.4258784055709839,
+      "learning_rate": 7.70073778175174e-05,
+      "loss": 0.187,
+      "step": 7050
+    },
+    {
+      "grad_norm": 0.6074666976928711,
+      "learning_rate": 7.69377653737907e-05,
+      "loss": 0.1821,
+      "step": 7060
+    },
+    {
+      "grad_norm": 0.4395882487297058,
+      "learning_rate": 7.686807928324209e-05,
+      "loss": 0.1997,
+      "step": 7070
+    },
+    {
+      "grad_norm": 0.4473331570625305,
+      "learning_rate": 7.679831973639065e-05,
+      "loss": 0.1846,
+      "step": 7080
+    },
+    {
+      "grad_norm": 0.4719548225402832,
+      "learning_rate": 7.672848692395637e-05,
+      "loss": 0.1953,
+      "step": 7090
+    },
+    {
+      "grad_norm": 0.5652232766151428,
+      "learning_rate": 7.665858103685944e-05,
+      "loss": 0.2167,
+      "step": 7100
+    },
+    {
+      "grad_norm": 0.5881660580635071,
+      "learning_rate": 7.658860226621991e-05,
+      "loss": 0.1804,
+      "step": 7110
+    },
+    {
+      "grad_norm": 0.5621788501739502,
+      "learning_rate": 7.651855080335708e-05,
+      "loss": 0.1869,
+      "step": 7120
+    },
+    {
+      "grad_norm": 0.5403567552566528,
+      "learning_rate": 7.644842683978896e-05,
+      "loss": 0.1771,
+      "step": 7130
+    },
+    {
+      "grad_norm": 0.6547488570213318,
+      "learning_rate": 7.63782305672318e-05,
+      "loss": 0.1775,
+      "step": 7140
+    },
+    {
+      "grad_norm": 0.7151280045509338,
+      "learning_rate": 7.63079621775995e-05,
+      "loss": 0.1909,
+      "step": 7150
+    },
+    {
+      "grad_norm": 0.5887133479118347,
+      "learning_rate": 7.623762186300319e-05,
+      "loss": 0.21,
+      "step": 7160
+    },
+    {
+      "grad_norm": 0.6549092531204224,
+      "learning_rate": 7.616720981575057e-05,
+      "loss": 0.1937,
+      "step": 7170
+    },
+    {
+      "grad_norm": 0.5851017832756042,
+      "learning_rate": 7.609672622834552e-05,
+      "loss": 0.1982,
+      "step": 7180
+    },
+    {
+      "grad_norm": 0.40296027064323425,
+      "learning_rate": 7.602617129348747e-05,
+      "loss": 0.1893,
+      "step": 7190
+    },
+    {
+      "grad_norm": 0.6152846217155457,
+      "learning_rate": 7.595554520407088e-05,
+      "loss": 0.1837,
+      "step": 7200
+    },
+    {
+      "grad_norm": 0.6107895970344543,
+      "learning_rate": 7.588484815318484e-05,
+      "loss": 0.1827,
+      "step": 7210
+    },
+    {
+      "grad_norm": 0.573887050151825,
+      "learning_rate": 7.581408033411234e-05,
+      "loss": 0.2156,
+      "step": 7220
+    },
+    {
+      "grad_norm": 0.484561562538147,
+      "learning_rate": 7.574324194032995e-05,
+      "loss": 0.1975,
+      "step": 7230
+    },
+    {
+      "grad_norm": 0.714483380317688,
+      "learning_rate": 7.567233316550705e-05,
+      "loss": 0.2041,
+      "step": 7240
+    },
+    {
+      "grad_norm": 0.656585693359375,
+      "learning_rate": 7.560135420350562e-05,
+      "loss": 0.1869,
+      "step": 7250
+    },
+    {
+      "grad_norm": 0.7101415991783142,
+      "learning_rate": 7.553030524837935e-05,
+      "loss": 0.1836,
+      "step": 7260
+    },
+    {
+      "grad_norm": 0.7503987550735474,
+      "learning_rate": 7.545918649437341e-05,
+      "loss": 0.201,
+      "step": 7270
+    },
+    {
+      "grad_norm": 0.5086443424224854,
+      "learning_rate": 7.538799813592377e-05,
+      "loss": 0.1857,
+      "step": 7280
+    },
+    {
+      "grad_norm": 0.4963659942150116,
+      "learning_rate": 7.531674036765662e-05,
+      "loss": 0.1819,
+      "step": 7290
+    },
+    {
+      "grad_norm": 0.4735184609889984,
+      "learning_rate": 7.524541338438807e-05,
+      "loss": 0.2,
+      "step": 7300
+    },
+    {
+      "grad_norm": 0.44475656747817993,
+      "learning_rate": 7.517401738112328e-05,
+      "loss": 0.1793,
+      "step": 7310
+    },
+    {
+      "grad_norm": 0.4195130467414856,
+      "learning_rate": 7.510255255305628e-05,
+      "loss": 0.1825,
+      "step": 7320
+    },
+    {
+      "grad_norm": 0.6032595634460449,
+      "learning_rate": 7.503101909556911e-05,
+      "loss": 0.1919,
+      "step": 7330
+    },
+    {
+      "grad_norm": 0.7640248537063599,
+      "learning_rate": 7.495941720423154e-05,
+      "loss": 0.1923,
+      "step": 7340
+    },
+    {
+      "grad_norm": 0.5643214583396912,
+      "learning_rate": 7.488774707480042e-05,
+      "loss": 0.206,
+      "step": 7350
+    },
+    {
+      "grad_norm": 0.6764867305755615,
+      "learning_rate": 7.481600890321911e-05,
+      "loss": 0.1942,
+      "step": 7360
+    },
+    {
+      "grad_norm": 0.6418370604515076,
+      "learning_rate": 7.474420288561708e-05,
+      "loss": 0.1846,
+      "step": 7370
+    },
+    {
+      "grad_norm": 0.8879308104515076,
+      "learning_rate": 7.467232921830921e-05,
+      "loss": 0.1991,
+      "step": 7380
+    },
+    {
+      "grad_norm": 0.419623464345932,
+      "learning_rate": 7.460038809779537e-05,
+      "loss": 0.1877,
+      "step": 7390
+    },
+    {
+      "grad_norm": 0.5669676065444946,
+      "learning_rate": 7.452837972075983e-05,
+      "loss": 0.2002,
+      "step": 7400
+    },
+    {
+      "grad_norm": 0.6196750998497009,
+      "learning_rate": 7.445630428407074e-05,
+      "loss": 0.173,
+      "step": 7410
+    },
+    {
+      "grad_norm": 0.46565520763397217,
+      "learning_rate": 7.43841619847796e-05,
+      "loss": 0.1916,
+      "step": 7420
+    },
+    {
+      "grad_norm": 0.42450425028800964,
+      "learning_rate": 7.431195302012072e-05,
+      "loss": 0.1868,
+      "step": 7430
+    },
+    {
+      "grad_norm": 0.44390377402305603,
+      "learning_rate": 7.423967758751061e-05,
+      "loss": 0.1859,
+      "step": 7440
+    },
+    {
+      "grad_norm": 0.8113678693771362,
+      "learning_rate": 7.416733588454758e-05,
+      "loss": 0.1874,
+      "step": 7450
+    },
+    {
+      "grad_norm": 0.6981191635131836,
+      "learning_rate": 7.409492810901106e-05,
+      "loss": 0.194,
+      "step": 7460
+    },
+    {
+      "grad_norm": 0.42460256814956665,
+      "learning_rate": 7.402245445886116e-05,
+      "loss": 0.2041,
+      "step": 7470
+    },
+    {
+      "grad_norm": 0.49199631810188293,
+      "learning_rate": 7.394991513223806e-05,
+      "loss": 0.1909,
+      "step": 7480
+    },
+    {
+      "grad_norm": 0.5836703181266785,
+      "learning_rate": 7.38773103274615e-05,
+      "loss": 0.1937,
+      "step": 7490
+    },
+    {
+      "grad_norm": 0.561226487159729,
+      "learning_rate": 7.380464024303028e-05,
+      "loss": 0.1758,
+      "step": 7500
+    },
+    {
+      "grad_norm": 0.45330142974853516,
+      "learning_rate": 7.373190507762162e-05,
+      "loss": 0.1824,
+      "step": 7510
+    },
+    {
+      "grad_norm": 0.42888009548187256,
+      "learning_rate": 7.365910503009066e-05,
+      "loss": 0.1978,
+      "step": 7520
+    },
+    {
+      "grad_norm": 0.5273244380950928,
+      "learning_rate": 7.358624029946996e-05,
+      "loss": 0.1708,
+      "step": 7530
+    },
+    {
+      "grad_norm": 0.6099250912666321,
+      "learning_rate": 7.351331108496893e-05,
+      "loss": 0.2093,
+      "step": 7540
+    },
+    {
+      "grad_norm": 0.5093023180961609,
+      "learning_rate": 7.344031758597325e-05,
+      "loss": 0.1939,
+      "step": 7550
+    },
+    {
+      "grad_norm": 0.675226628780365,
+      "learning_rate": 7.336726000204435e-05,
+      "loss": 0.191,
+      "step": 7560
+    },
+    {
+      "grad_norm": 0.559049129486084,
+      "learning_rate": 7.32941385329189e-05,
+      "loss": 0.1668,
+      "step": 7570
+    },
+    {
+      "grad_norm": 0.6432634592056274,
+      "learning_rate": 7.322095337850816e-05,
+      "loss": 0.1941,
+      "step": 7580
+    },
+    {
+      "grad_norm": 0.5262665748596191,
+      "learning_rate": 7.314770473889758e-05,
+      "loss": 0.1882,
+      "step": 7590
+    },
+    {
+      "grad_norm": 0.466363787651062,
+      "learning_rate": 7.307439281434615e-05,
+      "loss": 0.2045,
+      "step": 7600
+    },
+    {
+      "grad_norm": 0.5623495578765869,
+      "learning_rate": 7.300101780528585e-05,
+      "loss": 0.1916,
+      "step": 7610
+    },
+    {
+      "grad_norm": 0.5473973155021667,
+      "learning_rate": 7.292757991232117e-05,
+      "loss": 0.188,
+      "step": 7620
+    },
+    {
+      "grad_norm": 0.9303495287895203,
+      "learning_rate": 7.285407933622848e-05,
+      "loss": 0.1963,
+      "step": 7630
+    },
+    {
+      "grad_norm": 0.4040043354034424,
+      "learning_rate": 7.278051627795557e-05,
+      "loss": 0.1819,
+      "step": 7640
+    },
+    {
+      "grad_norm": 0.649565577507019,
+      "learning_rate": 7.270689093862105e-05,
+      "loss": 0.1809,
+      "step": 7650
+    },
+    {
+      "grad_norm": 0.5760002732276917,
+      "learning_rate": 7.263320351951374e-05,
+      "loss": 0.2185,
+      "step": 7660
+    },
+    {
+      "grad_norm": 0.4970298409461975,
+      "learning_rate": 7.255945422209227e-05,
+      "loss": 0.1944,
+      "step": 7670
+    },
+    {
+      "grad_norm": 0.6643884778022766,
+      "learning_rate": 7.248564324798437e-05,
+      "loss": 0.1933,
+      "step": 7680
+    },
+    {
+      "grad_norm": 0.40995460748672485,
+      "learning_rate": 7.241177079898644e-05,
+      "loss": 0.1909,
+      "step": 7690
+    },
+    {
+      "grad_norm": 0.6403678059577942,
+      "learning_rate": 7.233783707706295e-05,
+      "loss": 0.1928,
+      "step": 7700
+    },
+    {
+      "grad_norm": 0.7296698093414307,
+      "learning_rate": 7.226384228434586e-05,
+      "loss": 0.2017,
+      "step": 7710
+    },
+    {
+      "grad_norm": 0.6717272996902466,
+      "learning_rate": 7.21897866231341e-05,
+      "loss": 0.2003,
+      "step": 7720
+    },
+    {
+      "grad_norm": 0.6107869148254395,
+      "learning_rate": 7.211567029589303e-05,
+      "loss": 0.1858,
+      "step": 7730
+    },
+    {
+      "grad_norm": 0.5806393623352051,
+      "learning_rate": 7.204149350525387e-05,
+      "loss": 0.2147,
+      "step": 7740
+    },
+    {
+      "grad_norm": 0.5226749777793884,
+      "learning_rate": 7.196725645401309e-05,
+      "loss": 0.1976,
+      "step": 7750
+    },
+    {
+      "grad_norm": 0.47211989760398865,
+      "learning_rate": 7.1892959345132e-05,
+      "loss": 0.2006,
+      "step": 7760
+    },
+    {
+      "grad_norm": 0.8552802801132202,
+      "learning_rate": 7.181860238173605e-05,
+      "loss": 0.1887,
+      "step": 7770
+    },
+    {
+      "grad_norm": 0.4356684982776642,
+      "learning_rate": 7.174418576711432e-05,
+      "loss": 0.187,
+      "step": 7780
+    },
+    {
+      "grad_norm": 0.45009759068489075,
+      "learning_rate": 7.1669709704719e-05,
+      "loss": 0.1961,
+      "step": 7790
+    },
+    {
+      "grad_norm": 0.5037819743156433,
+      "learning_rate": 7.159517439816481e-05,
+      "loss": 0.1935,
+      "step": 7800
+    },
+    {
+      "grad_norm": 0.49052193760871887,
+      "learning_rate": 7.152058005122842e-05,
+      "loss": 0.1973,
+      "step": 7810
+    },
+    {
+      "grad_norm": 0.4813680350780487,
+      "learning_rate": 7.144592686784793e-05,
+      "loss": 0.2141,
+      "step": 7820
+    },
+    {
+      "grad_norm": 0.48662835359573364,
+      "learning_rate": 7.137121505212229e-05,
+      "loss": 0.1821,
+      "step": 7830
+    },
+    {
+      "grad_norm": 0.7366966605186462,
+      "learning_rate": 7.129644480831077e-05,
+      "loss": 0.1825,
+      "step": 7840
+    },
+    {
+      "grad_norm": 0.4875495731830597,
+      "learning_rate": 7.122161634083234e-05,
+      "loss": 0.1837,
+      "step": 7850
+    },
+    {
+      "grad_norm": 0.6581618785858154,
+      "learning_rate": 7.114672985426516e-05,
+      "loss": 0.1872,
+      "step": 7860
+    },
+    {
+      "grad_norm": 0.41009217500686646,
+      "learning_rate": 7.107178555334606e-05,
+      "loss": 0.1848,
+      "step": 7870
+    },
+    {
+      "grad_norm": 0.6551119685173035,
+      "learning_rate": 7.099678364296989e-05,
+      "loss": 0.1931,
+      "step": 7880
+    },
+    {
+      "grad_norm": 0.71089106798172,
+      "learning_rate": 7.0921724328189e-05,
+      "loss": 0.1932,
+      "step": 7890
+    },
+    {
+      "grad_norm": 0.6458206176757812,
+      "learning_rate": 7.084660781421268e-05,
+      "loss": 0.1851,
+      "step": 7900
+    },
+    {
+      "grad_norm": 0.4673345386981964,
+      "learning_rate": 7.077143430640662e-05,
+      "loss": 0.2037,
+      "step": 7910
+    },
+    {
+      "grad_norm": 0.6755942702293396,
+      "learning_rate": 7.069620401029232e-05,
+      "loss": 0.2038,
+      "step": 7920
+    },
+    {
+      "grad_norm": 0.7360473871231079,
+      "learning_rate": 7.062091713154655e-05,
+      "loss": 0.217,
+      "step": 7930
+    },
+    {
+      "grad_norm": 0.5232375860214233,
+      "learning_rate": 7.054557387600075e-05,
+      "loss": 0.1918,
+      "step": 7940
+    },
+    {
+      "grad_norm": 0.6279832720756531,
+      "learning_rate": 7.04701744496405e-05,
+      "loss": 0.2036,
+      "step": 7950
+    },
+    {
+      "grad_norm": 0.5670412182807922,
+      "learning_rate": 7.039471905860495e-05,
+      "loss": 0.2001,
+      "step": 7960
+    },
+    {
+      "grad_norm": 0.7871286869049072,
+      "learning_rate": 7.031920790918628e-05,
+      "loss": 0.1816,
+      "step": 7970
+    },
+    {
+      "grad_norm": 0.5510205030441284,
+      "learning_rate": 7.024364120782906e-05,
+      "loss": 0.2027,
+      "step": 7980
+    },
+    {
+      "grad_norm": 0.5103322267532349,
+      "learning_rate": 7.016801916112978e-05,
+      "loss": 0.192,
+      "step": 7990
+    },
+    {
+      "grad_norm": 0.7340911030769348,
+      "learning_rate": 7.009234197583623e-05,
+      "loss": 0.2208,
+      "step": 8000
+    },
+    {
+      "grad_norm": 0.5420516133308411,
+      "learning_rate": 7.001660985884692e-05,
+      "loss": 0.1907,
+      "step": 8010
+    },
+    {
+      "grad_norm": 0.6190202236175537,
+      "learning_rate": 6.994082301721063e-05,
+      "loss": 0.1885,
+      "step": 8020
+    },
+    {
+      "grad_norm": 0.589179515838623,
+      "learning_rate": 6.986498165812563e-05,
+      "loss": 0.1928,
+      "step": 8030
+    },
+    {
+      "grad_norm": 0.6170777082443237,
+      "learning_rate": 6.978908598893932e-05,
+      "loss": 0.1997,
+      "step": 8040
+    },
+    {
+      "grad_norm": 0.437487930059433,
+      "learning_rate": 6.971313621714756e-05,
+      "loss": 0.185,
+      "step": 8050
+    },
+    {
+      "grad_norm": 0.457376092672348,
+      "learning_rate": 6.96371325503941e-05,
+      "loss": 0.2039,
+      "step": 8060
+    },
+    {
+      "grad_norm": 0.6549373269081116,
+      "learning_rate": 6.956107519647014e-05,
+      "loss": 0.2021,
+      "step": 8070
+    },
+    {
+      "grad_norm": 0.570849597454071,
+      "learning_rate": 6.94849643633135e-05,
+      "loss": 0.194,
+      "step": 8080
+    },
+    {
+      "grad_norm": 0.7480877637863159,
+      "learning_rate": 6.940880025900834e-05,
+      "loss": 0.2012,
+      "step": 8090
+    },
+    {
+      "grad_norm": 0.48173055052757263,
+      "learning_rate": 6.933258309178438e-05,
+      "loss": 0.1875,
+      "step": 8100
+    },
+    {
+      "grad_norm": 0.7457817792892456,
+      "learning_rate": 6.925631307001646e-05,
+      "loss": 0.1981,
+      "step": 8110
+    },
+    {
+      "grad_norm": 0.4331342279911041,
+      "learning_rate": 6.91799904022239e-05,
+      "loss": 0.2018,
+      "step": 8120
+    },
+    {
+      "grad_norm": 0.46949392557144165,
+      "learning_rate": 6.910361529706997e-05,
+      "loss": 0.2038,
+      "step": 8130
+    },
+    {
+      "grad_norm": 0.6023057103157043,
+      "learning_rate": 6.902718796336131e-05,
+      "loss": 0.1898,
+      "step": 8140
+    },
+    {
+      "grad_norm": 0.5556503534317017,
+      "learning_rate": 6.895070861004729e-05,
+      "loss": 0.1998,
+      "step": 8150
+    },
+    {
+      "grad_norm": 0.6252777576446533,
+      "learning_rate": 6.887417744621956e-05,
+      "loss": 0.2131,
+      "step": 8160
+    },
+    {
+      "grad_norm": 0.7829018235206604,
+      "learning_rate": 6.87975946811114e-05,
+      "loss": 0.2037,
+      "step": 8170
+    },
+    {
+      "grad_norm": 0.48886922001838684,
+      "learning_rate": 6.872096052409718e-05,
+      "loss": 0.2024,
+      "step": 8180
+    },
+    {
+      "grad_norm": 0.5436359643936157,
+      "learning_rate": 6.864427518469174e-05,
+      "loss": 0.1906,
+      "step": 8190
+    },
+    {
+      "grad_norm": 0.8032960295677185,
+      "learning_rate": 6.856753887254986e-05,
+      "loss": 0.2056,
+      "step": 8200
+    },
+    {
+      "grad_norm": 0.8376327753067017,
+      "learning_rate": 6.849075179746572e-05,
+      "loss": 0.201,
+      "step": 8210
+    },
+    {
+      "grad_norm": 0.6409190893173218,
+      "learning_rate": 6.841391416937221e-05,
+      "loss": 0.1882,
+      "step": 8220
+    },
+    {
+      "grad_norm": 0.8630953431129456,
+      "learning_rate": 6.833702619834053e-05,
+      "loss": 0.2017,
+      "step": 8230
+    },
+    {
+      "grad_norm": 0.5896700620651245,
+      "learning_rate": 6.82600880945794e-05,
+      "loss": 0.1935,
+      "step": 8240
+    },
+    {
+      "grad_norm": 0.5692768096923828,
+      "learning_rate": 6.818310006843468e-05,
+      "loss": 0.2064,
+      "step": 8250
+    },
+    {
+      "grad_norm": 0.47074583172798157,
+      "learning_rate": 6.810606233038868e-05,
+      "loss": 0.1876,
+      "step": 8260
+    },
+    {
+      "grad_norm": 0.5413876175880432,
+      "learning_rate": 6.802897509105966e-05,
+      "loss": 0.1906,
+      "step": 8270
+    },
+    {
+      "grad_norm": 0.7367281317710876,
+      "learning_rate": 6.79518385612012e-05,
+      "loss": 0.2136,
+      "step": 8280
+    },
+    {
+      "grad_norm": 0.5562236905097961,
+      "learning_rate": 6.787465295170157e-05,
+      "loss": 0.195,
+      "step": 8290
+    },
+    {
+      "grad_norm": 0.8434953093528748,
+      "learning_rate": 6.779741847358332e-05,
+      "loss": 0.2073,
+      "step": 8300
+    },
+    {
+      "grad_norm": 0.645828902721405,
+      "learning_rate": 6.772013533800256e-05,
+      "loss": 0.1993,
+      "step": 8310
+    },
+    {
+      "grad_norm": 0.6425758004188538,
+      "learning_rate": 6.764280375624843e-05,
+      "loss": 0.1929,
+      "step": 8320
+    },
+    {
+      "grad_norm": 0.5786496996879578,
+      "learning_rate": 6.756542393974252e-05,
+      "loss": 0.1876,
+      "step": 8330
+    },
+    {
+      "grad_norm": 0.4930208921432495,
+      "learning_rate": 6.748799610003828e-05,
+      "loss": 0.2019,
+      "step": 8340
+    },
+    {
+      "grad_norm": 0.5453881025314331,
+      "learning_rate": 6.741052044882048e-05,
+      "loss": 0.1938,
+      "step": 8350
+    },
+    {
+      "grad_norm": 0.6413447260856628,
+      "learning_rate": 6.73329971979046e-05,
+      "loss": 0.193,
+      "step": 8360
+    },
+    {
+      "grad_norm": 0.5339936017990112,
+      "learning_rate": 6.725542655923625e-05,
+      "loss": 0.2266,
+      "step": 8370
+    },
+    {
+      "grad_norm": 0.6366740465164185,
+      "learning_rate": 6.717780874489057e-05,
+      "loss": 0.2094,
+      "step": 8380
+    },
+    {
+      "grad_norm": 0.5143768191337585,
+      "learning_rate": 6.710014396707172e-05,
+      "loss": 0.195,
+      "step": 8390
+    },
+    {
+      "grad_norm": 0.5888283252716064,
+      "learning_rate": 6.702243243811221e-05,
+      "loss": 0.1917,
+      "step": 8400
+    },
+    {
+      "grad_norm": 0.6242719292640686,
+      "learning_rate": 6.694467437047244e-05,
+      "loss": 0.1927,
+      "step": 8410
+    },
+    {
+      "grad_norm": 0.4434303045272827,
+      "learning_rate": 6.686686997673997e-05,
+      "loss": 0.2231,
+      "step": 8420
+    },
+    {
+      "grad_norm": 0.4299023747444153,
+      "learning_rate": 6.678901946962903e-05,
+      "loss": 0.1842,
+      "step": 8430
+    },
+    {
+      "grad_norm": 0.5216169953346252,
+      "learning_rate": 6.671112306197996e-05,
+      "loss": 0.1983,
+      "step": 8440
+    },
+    {
+      "grad_norm": 0.5768060684204102,
+      "learning_rate": 6.663318096675854e-05,
+      "loss": 0.1864,
+      "step": 8450
+    },
+    {
+      "grad_norm": 0.7971596121788025,
+      "learning_rate": 6.655519339705552e-05,
+      "loss": 0.2107,
+      "step": 8460
+    },
+    {
+      "grad_norm": 0.5996659398078918,
+      "learning_rate": 6.647716056608588e-05,
+      "loss": 0.1962,
+      "step": 8470
+    },
+    {
+      "grad_norm": 0.5041460990905762,
+      "learning_rate": 6.639908268718843e-05,
+      "loss": 0.1784,
+      "step": 8480
+    },
+    {
+      "grad_norm": 0.6551742553710938,
+      "learning_rate": 6.632095997382514e-05,
+      "loss": 0.2006,
+      "step": 8490
+    },
+    {
+      "grad_norm": 0.38995882868766785,
+      "learning_rate": 6.624279263958047e-05,
+      "loss": 0.1815,
+      "step": 8500
+    },
+    {
+      "grad_norm": 0.5085399150848389,
+      "learning_rate": 6.616458089816097e-05,
+      "loss": 0.1832,
+      "step": 8510
+    },
+    {
+      "grad_norm": 0.5540745854377747,
+      "learning_rate": 6.608632496339454e-05,
+      "loss": 0.1951,
+      "step": 8520
+    },
+    {
+      "grad_norm": 0.620813250541687,
+      "learning_rate": 6.600802504922988e-05,
+      "loss": 0.1976,
+      "step": 8530
+    },
+    {
+      "grad_norm": 0.481110543012619,
+      "learning_rate": 6.592968136973604e-05,
+      "loss": 0.2014,
+      "step": 8540
+    },
+    {
+      "grad_norm": 0.5640373826026917,
+      "learning_rate": 6.585129413910159e-05,
+      "loss": 0.1916,
+      "step": 8550
+    },
+    {
+      "grad_norm": 0.4579847455024719,
+      "learning_rate": 6.577286357163424e-05,
+      "loss": 0.1849,
+      "step": 8560
+    },
+    {
+      "grad_norm": 0.5077273845672607,
+      "learning_rate": 6.569438988176018e-05,
+      "loss": 0.2129,
+      "step": 8570
+    },
+    {
+      "grad_norm": 0.5341709852218628,
+      "learning_rate": 6.561587328402347e-05,
+      "loss": 0.2095,
+      "step": 8580
+    },
+    {
+      "grad_norm": 0.7573632597923279,
+      "learning_rate": 6.553731399308549e-05,
+      "loss": 0.1989,
+      "step": 8590
+    },
+    {
+      "grad_norm": 0.6411945819854736,
+      "learning_rate": 6.545871222372436e-05,
+      "loss": 0.2021,
+      "step": 8600
+    },
+    {
+      "grad_norm": 0.6982321739196777,
+      "learning_rate": 6.538006819083426e-05,
+      "loss": 0.2067,
+      "step": 8610
+    },
+    {
+      "grad_norm": 0.5399815440177917,
+      "learning_rate": 6.530138210942505e-05,
+      "loss": 0.2028,
+      "step": 8620
+    },
+    {
+      "grad_norm": 0.5343695878982544,
+      "learning_rate": 6.522265419462141e-05,
+      "loss": 0.1972,
+      "step": 8630
+    },
+    {
+      "grad_norm": 0.5076909065246582,
+      "learning_rate": 6.514388466166248e-05,
+      "loss": 0.1847,
+      "step": 8640
+    },
+    {
+      "grad_norm": 0.6630603075027466,
+      "learning_rate": 6.506507372590119e-05,
+      "loss": 0.1869,
+      "step": 8650
+    },
+    {
+      "grad_norm": 0.5100239515304565,
+      "learning_rate": 6.498622160280355e-05,
+      "loss": 0.1827,
+      "step": 8660
+    },
+    {
+      "grad_norm": 0.5570880174636841,
+      "learning_rate": 6.490732850794832e-05,
+      "loss": 0.1877,
+      "step": 8670
+    },
+    {
+      "grad_norm": 0.552374005317688,
+      "learning_rate": 6.482839465702616e-05,
+      "loss": 0.2099,
+      "step": 8680
+    },
+    {
+      "grad_norm": 0.7463684678077698,
+      "learning_rate": 6.474942026583923e-05,
+      "loss": 0.1891,
+      "step": 8690
+    },
+    {
+      "grad_norm": 0.4179556369781494,
+      "learning_rate": 6.467040555030052e-05,
+      "loss": 0.1933,
+      "step": 8700
+    },
+    {
+      "grad_norm": 0.4803868532180786,
+      "learning_rate": 6.459135072643321e-05,
+      "loss": 0.207,
+      "step": 8710
+    },
+    {
+      "grad_norm": 0.4781976044178009,
+      "learning_rate": 6.451225601037019e-05,
+      "loss": 0.1716,
+      "step": 8720
+    },
+    {
+      "grad_norm": 0.5252571105957031,
+      "learning_rate": 6.443312161835338e-05,
+      "loss": 0.1895,
+      "step": 8730
+    },
+    {
+      "grad_norm": 0.9479270577430725,
+      "learning_rate": 6.43539477667332e-05,
+      "loss": 0.1886,
+      "step": 8740
+    },
+    {
+      "grad_norm": 0.6369916796684265,
+      "learning_rate": 6.427473467196793e-05,
+      "loss": 0.1802,
+      "step": 8750
+    },
+    {
+      "grad_norm": 0.9778500199317932,
+      "learning_rate": 6.419548255062315e-05,
+      "loss": 0.1907,
+      "step": 8760
+    },
+    {
+      "grad_norm": 0.46750280261039734,
+      "learning_rate": 6.411619161937112e-05,
+      "loss": 0.1898,
+      "step": 8770
+    },
+    {
+      "grad_norm": 0.7409607768058777,
+      "learning_rate": 6.403686209499022e-05,
+      "loss": 0.2104,
+      "step": 8780
+    },
+    {
+      "grad_norm": 0.5607050061225891,
+      "learning_rate": 6.395749419436437e-05,
+      "loss": 0.1842,
+      "step": 8790
+    },
+    {
+      "grad_norm": 0.5214778184890747,
+      "learning_rate": 6.387808813448234e-05,
+      "loss": 0.1702,
+      "step": 8800
+    },
+    {
+      "grad_norm": 0.6756618022918701,
+      "learning_rate": 6.37986441324373e-05,
+      "loss": 0.1874,
+      "step": 8810
+    },
+    {
+      "grad_norm": 0.5906814336776733,
+      "learning_rate": 6.37191624054261e-05,
+      "loss": 0.2149,
+      "step": 8820
+    },
+    {
+      "grad_norm": 0.6167814135551453,
+      "learning_rate": 6.363964317074872e-05,
+      "loss": 0.1958,
+      "step": 8830
+    },
+    {
+      "grad_norm": 0.623216450214386,
+      "learning_rate": 6.356008664580776e-05,
+      "loss": 0.1959,
+      "step": 8840
+    },
+    {
+      "grad_norm": 0.464743971824646,
+      "learning_rate": 6.348049304810771e-05,
+      "loss": 0.1961,
+      "step": 8850
+    },
+    {
+      "grad_norm": 0.8087256550788879,
+      "learning_rate": 6.340086259525442e-05,
+      "loss": 0.1844,
+      "step": 8860
+    },
+    {
+      "grad_norm": 0.6259470582008362,
+      "learning_rate": 6.332119550495448e-05,
+      "loss": 0.1924,
+      "step": 8870
+    },
+    {
+      "grad_norm": 0.5108415484428406,
+      "learning_rate": 6.324149199501473e-05,
+      "loss": 0.1987,
+      "step": 8880
+    },
+    {
+      "grad_norm": 0.8204335570335388,
+      "learning_rate": 6.316175228334146e-05,
+      "loss": 0.1871,
+      "step": 8890
+    },
+    {
+      "grad_norm": 0.3855741620063782,
+      "learning_rate": 6.308197658794003e-05,
+      "loss": 0.1899,
+      "step": 8900
+    },
+    {
+      "grad_norm": 0.937233030796051,
+      "learning_rate": 6.300216512691417e-05,
+      "loss": 0.207,
+      "step": 8910
+    },
+    {
+      "grad_norm": 0.5823239684104919,
+      "learning_rate": 6.292231811846532e-05,
+      "loss": 0.1873,
+      "step": 8920
+    },
+    {
+      "grad_norm": 0.6572949886322021,
+      "learning_rate": 6.284243578089217e-05,
+      "loss": 0.1903,
+      "step": 8930
+    },
+    {
+      "grad_norm": 0.7032896876335144,
+      "learning_rate": 6.276251833258999e-05,
+      "loss": 0.1807,
+      "step": 8940
+    },
+    {
+      "grad_norm": 0.7452958226203918,
+      "learning_rate": 6.268256599205003e-05,
+      "loss": 0.1795,
+      "step": 8950
+    },
+    {
+      "grad_norm": 0.4749715328216553,
+      "learning_rate": 6.260257897785892e-05,
+      "loss": 0.1989,
+      "step": 8960
+    },
+    {
+      "grad_norm": 0.6590916514396667,
+      "learning_rate": 6.252255750869811e-05,
+      "loss": 0.1889,
+      "step": 8970
+    },
+    {
+      "grad_norm": 0.5558798313140869,
+      "learning_rate": 6.244250180334325e-05,
+      "loss": 0.1924,
+      "step": 8980
+    },
+    {
+      "grad_norm": 0.5769075751304626,
+      "learning_rate": 6.236241208066356e-05,
+      "loss": 0.1768,
+      "step": 8990
+    },
+    {
+      "grad_norm": 0.896496057510376,
+      "learning_rate": 6.228228855962133e-05,
+      "loss": 0.1934,
+      "step": 9000
+    },
+    {
+      "grad_norm": 0.606447696685791,
+      "learning_rate": 6.220213145927115e-05,
+      "loss": 0.198,
+      "step": 9010
+    },
+    {
+      "grad_norm": 0.6868760585784912,
+      "learning_rate": 6.212194099875951e-05,
+      "loss": 0.1932,
+      "step": 9020
+    },
+    {
+      "grad_norm": 0.5239525437355042,
+      "learning_rate": 6.204171739732405e-05,
+      "loss": 0.1893,
+      "step": 9030
+    },
+    {
+      "grad_norm": 0.5544044375419617,
+      "learning_rate": 6.196146087429303e-05,
+      "loss": 0.1958,
+      "step": 9040
+    },
+    {
+      "grad_norm": 0.5353380441665649,
+      "learning_rate": 6.188117164908474e-05,
+      "loss": 0.195,
+      "step": 9050
+    },
+    {
+      "grad_norm": 0.43021804094314575,
+      "learning_rate": 6.180084994120684e-05,
+      "loss": 0.199,
+      "step": 9060
+    },
+    {
+      "grad_norm": 0.7091668248176575,
+      "learning_rate": 6.17204959702558e-05,
+      "loss": 0.1958,
+      "step": 9070
+    },
+    {
+      "grad_norm": 0.8248500227928162,
+      "learning_rate": 6.164010995591635e-05,
+      "loss": 0.1965,
+      "step": 9080
+    },
+    {
+      "grad_norm": 0.5080732703208923,
+      "learning_rate": 6.155969211796076e-05,
+      "loss": 0.1949,
+      "step": 9090
+    },
+    {
+      "grad_norm": 0.6272768378257751,
+      "learning_rate": 6.147924267624829e-05,
+      "loss": 0.1902,
+      "step": 9100
+    },
+    {
+      "grad_norm": 0.57228684425354,
+      "learning_rate": 6.13987618507247e-05,
+      "loss": 0.1863,
+      "step": 9110
+    },
+    {
+      "grad_norm": 0.5205716490745544,
+      "learning_rate": 6.131824986142147e-05,
+      "loss": 0.1955,
+      "step": 9120
+    },
+    {
+      "grad_norm": 0.40080127120018005,
+      "learning_rate": 6.123770692845529e-05,
+      "loss": 0.1699,
+      "step": 9130
+    },
+    {
+      "grad_norm": 0.5248435735702515,
+      "learning_rate": 6.11571332720275e-05,
+      "loss": 0.2024,
+      "step": 9140
+    },
+    {
+      "grad_norm": 0.5605234503746033,
+      "learning_rate": 6.107652911242336e-05,
+      "loss": 0.1891,
+      "step": 9150
+    },
+    {
+      "grad_norm": 0.4158274531364441,
+      "learning_rate": 6.0995894670011586e-05,
+      "loss": 0.166,
+      "step": 9160
+    },
+    {
+      "grad_norm": 0.499946653842926,
+      "learning_rate": 6.091523016524368e-05,
+      "loss": 0.1863,
+      "step": 9170
+    },
+    {
+      "grad_norm": 0.5058438181877136,
+      "learning_rate": 6.083453581865328e-05,
+      "loss": 0.1933,
+      "step": 9180
+    },
+    {
+      "grad_norm": 0.5873285531997681,
+      "learning_rate": 6.075381185085568e-05,
+      "loss": 0.1874,
+      "step": 9190
+    },
+    {
+      "grad_norm": 0.40993040800094604,
+      "learning_rate": 6.067305848254709e-05,
+      "loss": 0.195,
+      "step": 9200
+    },
+    {
+      "grad_norm": 0.5749062299728394,
+      "learning_rate": 6.059227593450418e-05,
+      "loss": 0.1878,
+      "step": 9210
+    },
+    {
+      "grad_norm": 0.5144668817520142,
+      "learning_rate": 6.051146442758333e-05,
+      "loss": 0.2018,
+      "step": 9220
+    },
+    {
+      "grad_norm": 0.5539340376853943,
+      "learning_rate": 6.043062418272012e-05,
+      "loss": 0.1805,
+      "step": 9230
+    },
+    {
+      "grad_norm": 0.5033758878707886,
+      "learning_rate": 6.0349755420928666e-05,
+      "loss": 0.1784,
+      "step": 9240
+    },
+    {
+      "grad_norm": 0.6414024829864502,
+      "learning_rate": 6.0268858363301105e-05,
+      "loss": 0.1865,
+      "step": 9250
+    },
+    {
+      "grad_norm": 0.6080801486968994,
+      "learning_rate": 6.018793323100689e-05,
+      "loss": 0.1759,
+      "step": 9260
+    },
+    {
+      "grad_norm": 0.6421687602996826,
+      "learning_rate": 6.0106980245292255e-05,
+      "loss": 0.2027,
+      "step": 9270
+    },
+    {
+      "grad_norm": 0.47517043352127075,
+      "learning_rate": 6.002599962747957e-05,
+      "loss": 0.1627,
+      "step": 9280
+    },
+    {
+      "grad_norm": 0.43102124333381653,
+      "learning_rate": 5.994499159896673e-05,
+      "loss": 0.1912,
+      "step": 9290
+    },
+    {
+      "grad_norm": 0.7118813395500183,
+      "learning_rate": 5.9863956381226607e-05,
+      "loss": 0.1826,
+      "step": 9300
+    },
+    {
+      "grad_norm": 0.6441318988800049,
+      "learning_rate": 5.9782894195806394e-05,
+      "loss": 0.1726,
+      "step": 9310
+    },
+    {
+      "grad_norm": 0.47374090552330017,
+      "learning_rate": 5.9701805264327004e-05,
+      "loss": 0.1785,
+      "step": 9320
+    },
+    {
+      "grad_norm": 0.5784379243850708,
+      "learning_rate": 5.96206898084825e-05,
+      "loss": 0.1859,
+      "step": 9330
+    },
+    {
+      "grad_norm": 0.604211688041687,
+      "learning_rate": 5.953954805003942e-05,
+      "loss": 0.1985,
+      "step": 9340
+    },
+    {
+      "grad_norm": 0.826299250125885,
+      "learning_rate": 5.945838021083623e-05,
+      "loss": 0.1861,
+      "step": 9350
+    },
+    {
+      "grad_norm": 0.5101220607757568,
+      "learning_rate": 5.9377186512782714e-05,
+      "loss": 0.2099,
+      "step": 9360
+    },
+    {
+      "grad_norm": 0.5706347227096558,
+      "learning_rate": 5.929596717785935e-05,
+      "loss": 0.1885,
+      "step": 9370
+    },
+    {
+      "grad_norm": 0.4635280668735504,
+      "learning_rate": 5.921472242811668e-05,
+      "loss": 0.1761,
+      "step": 9380
+    },
+    {
+      "grad_norm": 0.512699544429779,
+      "learning_rate": 5.913345248567475e-05,
+      "loss": 0.1883,
+      "step": 9390
+    },
+    {
+      "grad_norm": 0.6179343461990356,
+      "learning_rate": 5.905215757272248e-05,
+      "loss": 0.1964,
+      "step": 9400
+    },
+    {
+      "grad_norm": 0.6252385973930359,
+      "learning_rate": 5.897083791151706e-05,
+      "loss": 0.183,
+      "step": 9410
+    },
+    {
+      "grad_norm": 0.5846800804138184,
+      "learning_rate": 5.888949372438336e-05,
+      "loss": 0.1781,
+      "step": 9420
+    },
+    {
+      "grad_norm": 0.5272970795631409,
+      "learning_rate": 5.8808125233713255e-05,
+      "loss": 0.1844,
+      "step": 9430
+    },
+    {
+      "grad_norm": 0.6292622089385986,
+      "learning_rate": 5.872673266196509e-05,
+      "loss": 0.1726,
+      "step": 9440
+    },
+    {
+      "grad_norm": 0.5844361186027527,
+      "learning_rate": 5.864531623166305e-05,
+      "loss": 0.1786,
+      "step": 9450
+    },
+    {
+      "grad_norm": 0.5066063404083252,
+      "learning_rate": 5.856387616539656e-05,
+      "loss": 0.1705,
+      "step": 9460
+    },
+    {
+      "grad_norm": 0.36124861240386963,
+      "learning_rate": 5.848241268581967e-05,
+      "loss": 0.1839,
+      "step": 9470
+    },
+    {
+      "grad_norm": 0.6957190036773682,
+      "learning_rate": 5.840092601565037e-05,
+      "loss": 0.1757,
+      "step": 9480
+    },
+    {
+      "grad_norm": 0.3795112371444702,
+      "learning_rate": 5.8319416377670144e-05,
+      "loss": 0.1914,
+      "step": 9490
+    },
+    {
+      "grad_norm": 0.5808974504470825,
+      "learning_rate": 5.82378839947232e-05,
+      "loss": 0.1937,
+      "step": 9500
+    },
+    {
+      "grad_norm": 0.7069080471992493,
+      "learning_rate": 5.815632908971599e-05,
+      "loss": 0.1834,
+      "step": 9510
+    },
+    {
+      "grad_norm": 0.5776515007019043,
+      "learning_rate": 5.80747518856165e-05,
+      "loss": 0.1814,
+      "step": 9520
+    },
+    {
+      "grad_norm": 0.5001153349876404,
+      "learning_rate": 5.799315260545367e-05,
+      "loss": 0.1898,
+      "step": 9530
+    },
+    {
+      "grad_norm": 0.4933784604072571,
+      "learning_rate": 5.791153147231686e-05,
+      "loss": 0.1891,
+      "step": 9540
+    },
+    {
+      "grad_norm": 0.6418437957763672,
+      "learning_rate": 5.782988870935509e-05,
+      "loss": 0.185,
+      "step": 9550
+    },
+    {
+      "grad_norm": 0.4385838508605957,
+      "learning_rate": 5.774822453977657e-05,
+      "loss": 0.1804,
+      "step": 9560
+    },
+    {
+      "grad_norm": 0.4515121579170227,
+      "learning_rate": 5.7666539186848036e-05,
+      "loss": 0.1762,
+      "step": 9570
+    },
+    {
+      "grad_norm": 0.7937777042388916,
+      "learning_rate": 5.758483287389411e-05,
+      "loss": 0.1898,
+      "step": 9580
+    },
+    {
+      "grad_norm": 0.5393437147140503,
+      "learning_rate": 5.7503105824296735e-05,
+      "loss": 0.1747,
+      "step": 9590
+    },
+    {
+      "grad_norm": 0.42680469155311584,
+      "learning_rate": 5.742135826149453e-05,
+      "loss": 0.1796,
+      "step": 9600
+    },
+    {
+      "grad_norm": 0.6199423670768738,
+      "learning_rate": 5.7339590408982223e-05,
+      "loss": 0.1804,
+      "step": 9610
+    },
+    {
+      "grad_norm": 0.685000479221344,
+      "learning_rate": 5.725780249031e-05,
+      "loss": 0.1784,
+      "step": 9620
+    },
+    {
+      "grad_norm": 0.5748093724250793,
+      "learning_rate": 5.717599472908292e-05,
+      "loss": 0.1879,
+      "step": 9630
+    },
+    {
+      "grad_norm": 0.4606155753135681,
+      "learning_rate": 5.7094167348960237e-05,
+      "loss": 0.1741,
+      "step": 9640
+    },
+    {
+      "grad_norm": 0.5204547643661499,
+      "learning_rate": 5.7012320573654945e-05,
+      "loss": 0.1855,
+      "step": 9650
+    },
+    {
+      "grad_norm": 0.496120810508728,
+      "learning_rate": 5.693045462693295e-05,
+      "loss": 0.1543,
+      "step": 9660
+    },
+    {
+      "grad_norm": 0.5643748044967651,
+      "learning_rate": 5.684856973261266e-05,
+      "loss": 0.1689,
+      "step": 9670
+    },
+    {
+      "grad_norm": 0.5129379630088806,
+      "learning_rate": 5.6766666114564215e-05,
+      "loss": 0.195,
+      "step": 9680
+    },
+    {
+      "grad_norm": 0.5942407846450806,
+      "learning_rate": 5.668474399670899e-05,
+      "loss": 0.1647,
+      "step": 9690
+    },
+    {
+      "grad_norm": 0.4243850111961365,
+      "learning_rate": 5.660280360301896e-05,
+      "loss": 0.1914,
+      "step": 9700
+    },
+    {
+      "grad_norm": 0.5172005295753479,
+      "learning_rate": 5.652084515751599e-05,
+      "loss": 0.1936,
+      "step": 9710
+    },
+    {
+      "grad_norm": 0.615800142288208,
+      "learning_rate": 5.643886888427137e-05,
+      "loss": 0.2001,
+      "step": 9720
+    },
+    {
+      "grad_norm": 0.520674467086792,
+      "learning_rate": 5.6356875007405074e-05,
+      "loss": 0.1923,
+      "step": 9730
+    },
+    {
+      "grad_norm": 0.5037645101547241,
+      "learning_rate": 5.627486375108525e-05,
+      "loss": 0.1796,
+      "step": 9740
+    },
+    {
+      "grad_norm": 0.5950884222984314,
+      "learning_rate": 5.619283533952754e-05,
+      "loss": 0.1966,
+      "step": 9750
+    },
+    {
+      "grad_norm": 0.556892991065979,
+      "learning_rate": 5.6110789996994474e-05,
+      "loss": 0.1868,
+      "step": 9760
+    },
+    {
+      "grad_norm": 0.45335066318511963,
+      "learning_rate": 5.602872794779491e-05,
+      "loss": 0.1894,
+      "step": 9770
+    },
+    {
+      "grad_norm": 0.5126853585243225,
+      "learning_rate": 5.594664941628334e-05,
+      "loss": 0.2173,
+      "step": 9780
+    },
+    {
+      "grad_norm": 0.4223232567310333,
+      "learning_rate": 5.5864554626859324e-05,
+      "loss": 0.2007,
+      "step": 9790
+    },
+    {
+      "grad_norm": 0.6520665884017944,
+      "learning_rate": 5.578244380396691e-05,
+      "loss": 0.2025,
+      "step": 9800
+    },
+    {
+      "grad_norm": 0.5002579092979431,
+      "learning_rate": 5.570031717209394e-05,
+      "loss": 0.1844,
+      "step": 9810
+    },
+    {
+      "grad_norm": 0.6791883707046509,
+      "learning_rate": 5.561817495577147e-05,
+      "loss": 0.1909,
+      "step": 9820
+    },
+    {
+      "grad_norm": 0.4473389685153961,
+      "learning_rate": 5.5536017379573215e-05,
+      "loss": 0.1968,
+      "step": 9830
+    },
+    {
+      "grad_norm": 0.5177410244941711,
+      "learning_rate": 5.545384466811483e-05,
+      "loss": 0.1843,
+      "step": 9840
+    },
+    {
+      "grad_norm": 0.7230008244514465,
+      "learning_rate": 5.5371657046053384e-05,
+      "loss": 0.1848,
+      "step": 9850
+    },
+    {
+      "grad_norm": 0.44801655411720276,
+      "learning_rate": 5.528945473808669e-05,
+      "loss": 0.1757,
+      "step": 9860
+    },
+    {
+      "grad_norm": 0.47381478548049927,
+      "learning_rate": 5.520723796895272e-05,
+      "loss": 0.1889,
+      "step": 9870
+    },
+    {
+      "grad_norm": 0.5601184964179993,
+      "learning_rate": 5.512500696342897e-05,
+      "loss": 0.1935,
+      "step": 9880
+    },
+    {
+      "grad_norm": 0.4680139422416687,
+      "learning_rate": 5.504276194633188e-05,
+      "loss": 0.1777,
+      "step": 9890
+    },
+    {
+      "grad_norm": 0.4131275713443756,
+      "learning_rate": 5.49605031425162e-05,
+      "loss": 0.1853,
+      "step": 9900
+    },
+    {
+      "grad_norm": 0.4846619665622711,
+      "learning_rate": 5.487823077687434e-05,
+      "loss": 0.2056,
+      "step": 9910
+    },
+    {
+      "grad_norm": 0.5987968444824219,
+      "learning_rate": 5.4795945074335806e-05,
+      "loss": 0.1765,
+      "step": 9920
+    },
+    {
+      "grad_norm": 0.459291011095047,
+      "learning_rate": 5.471364625986657e-05,
+      "loss": 0.1892,
+      "step": 9930
+    },
+    {
+      "grad_norm": 0.5296525359153748,
+      "learning_rate": 5.463133455846845e-05,
+      "loss": 0.2103,
+      "step": 9940
+    },
+    {
+      "grad_norm": 0.6497468948364258,
+      "learning_rate": 5.4549010195178505e-05,
+      "loss": 0.2064,
+      "step": 9950
+    },
+    {
+      "grad_norm": 0.5304078459739685,
+      "learning_rate": 5.446667339506838e-05,
+      "loss": 0.1814,
+      "step": 9960
+    },
+    {
+      "grad_norm": 0.4897553622722626,
+      "learning_rate": 5.4384324383243756e-05,
+      "loss": 0.1854,
+      "step": 9970
+    },
+    {
+      "grad_norm": 0.5768560171127319,
+      "learning_rate": 5.430196338484368e-05,
+      "loss": 0.1995,
+      "step": 9980
+    },
+    {
+      "grad_norm": 0.5253862142562866,
+      "learning_rate": 5.4219590625039975e-05,
+      "loss": 0.1982,
+      "step": 9990
+    },
+    {
+      "grad_norm": 0.48000842332839966,
+      "learning_rate": 5.413720632903664e-05,
+      "loss": 0.2043,
+      "step": 10000
+    },
+    {
+      "grad_norm": 0.5921430587768555,
+      "learning_rate": 5.405481072206917e-05,
+      "loss": 0.1833,
+      "step": 10010
+    },
+    {
+      "grad_norm": 0.5838000774383545,
+      "learning_rate": 5.397240402940402e-05,
+      "loss": 0.1975,
+      "step": 10020
+    },
+    {
+      "grad_norm": 0.5746766924858093,
+      "learning_rate": 5.388998647633794e-05,
+      "loss": 0.19,
+      "step": 10030
+    },
+    {
+      "grad_norm": 0.6367522478103638,
+      "learning_rate": 5.380755828819737e-05,
+      "loss": 0.182,
+      "step": 10040
+    },
+    {
+      "grad_norm": 0.550219714641571,
+      "learning_rate": 5.3725119690337846e-05,
+      "loss": 0.1828,
+      "step": 10050
+    },
+    {
+      "grad_norm": 0.5789723992347717,
+      "learning_rate": 5.3642670908143324e-05,
+      "loss": 0.1812,
+      "step": 10060
+    },
+    {
+      "grad_norm": 0.4615694582462311,
+      "learning_rate": 5.356021216702562e-05,
+      "loss": 0.1853,
+      "step": 10070
+    },
+    {
+      "grad_norm": 0.5290621519088745,
+      "learning_rate": 5.347774369242381e-05,
+      "loss": 0.1962,
+      "step": 10080
+    },
+    {
+      "grad_norm": 0.6132157444953918,
+      "learning_rate": 5.3395265709803545e-05,
+      "loss": 0.1839,
+      "step": 10090
+    },
+    {
+      "grad_norm": 0.8560709357261658,
+      "learning_rate": 5.331277844465647e-05,
+      "loss": 0.2109,
+      "step": 10100
+    },
+    {
+      "grad_norm": 0.5495644807815552,
+      "learning_rate": 5.323028212249963e-05,
+      "loss": 0.1938,
+      "step": 10110
+    },
+    {
+      "grad_norm": 0.46827682852745056,
+      "learning_rate": 5.314777696887481e-05,
+      "loss": 0.1846,
+      "step": 10120
+    },
+    {
+      "grad_norm": 0.48011574149131775,
+      "learning_rate": 5.306526320934796e-05,
+      "loss": 0.1926,
+      "step": 10130
+    },
+    {
+      "grad_norm": 0.5753622651100159,
+      "learning_rate": 5.298274106950854e-05,
+      "loss": 0.1992,
+      "step": 10140
+    },
+    {
+      "grad_norm": 0.6128261685371399,
+      "learning_rate": 5.290021077496893e-05,
+      "loss": 0.1948,
+      "step": 10150
+    },
+    {
+      "grad_norm": 0.4646427035331726,
+      "learning_rate": 5.2817672551363816e-05,
+      "loss": 0.1917,
+      "step": 10160
+    },
+    {
+      "grad_norm": 0.689041793346405,
+      "learning_rate": 5.273512662434952e-05,
+      "loss": 0.1905,
+      "step": 10170
+    },
+    {
+      "grad_norm": 0.781278669834137,
+      "learning_rate": 5.265257321960349e-05,
+      "loss": 0.2027,
+      "step": 10180
+    },
+    {
+      "grad_norm": 0.7293897867202759,
+      "learning_rate": 5.257001256282357e-05,
+      "loss": 0.1994,
+      "step": 10190
+    },
+    {
+      "grad_norm": 0.7376835346221924,
+      "learning_rate": 5.248744487972742e-05,
+      "loss": 0.2133,
+      "step": 10200
+    },
+    {
+      "grad_norm": 0.7664139866828918,
+      "learning_rate": 5.240487039605196e-05,
+      "loss": 0.193,
+      "step": 10210
+    },
+    {
+      "grad_norm": 0.6295002698898315,
+      "learning_rate": 5.232228933755267e-05,
+      "loss": 0.2021,
+      "step": 10220
+    },
+    {
+      "grad_norm": 0.5366851687431335,
+      "learning_rate": 5.2239701930003006e-05,
+      "loss": 0.1972,
+      "step": 10230
+    },
+    {
+      "grad_norm": 0.612837016582489,
+      "learning_rate": 5.215710839919379e-05,
+      "loss": 0.2017,
+      "step": 10240
+    },
+    {
+      "grad_norm": 0.3923555314540863,
+      "learning_rate": 5.207450897093257e-05,
+      "loss": 0.1834,
+      "step": 10250
+    },
+    {
+      "grad_norm": 0.7040473818778992,
+      "learning_rate": 5.1991903871043046e-05,
+      "loss": 0.189,
+      "step": 10260
+    },
+    {
+      "grad_norm": 0.47470343112945557,
+      "learning_rate": 5.190929332536439e-05,
+      "loss": 0.1801,
+      "step": 10270
+    },
+    {
+      "grad_norm": 0.5682361125946045,
+      "learning_rate": 5.182667755975071e-05,
+      "loss": 0.2056,
+      "step": 10280
+    },
+    {
+      "grad_norm": 0.6745263934135437,
+      "learning_rate": 5.1744056800070315e-05,
+      "loss": 0.1923,
+      "step": 10290
+    },
+    {
+      "grad_norm": 0.4591670632362366,
+      "learning_rate": 5.166143127220524e-05,
+      "loss": 0.1983,
+      "step": 10300
+    },
+    {
+      "grad_norm": 0.8794211745262146,
+      "learning_rate": 5.1578801202050485e-05,
+      "loss": 0.177,
+      "step": 10310
+    },
+    {
+      "grad_norm": 0.5333230495452881,
+      "learning_rate": 5.149616681551355e-05,
+      "loss": 0.1881,
+      "step": 10320
+    },
+    {
+      "grad_norm": 0.6851953268051147,
+      "learning_rate": 5.141352833851367e-05,
+      "loss": 0.1823,
+      "step": 10330
+    },
+    {
+      "grad_norm": 0.5733579397201538,
+      "learning_rate": 5.1330885996981285e-05,
+      "loss": 0.186,
+      "step": 10340
+    },
+    {
+      "grad_norm": 0.6254924535751343,
+      "learning_rate": 5.124824001685741e-05,
+      "loss": 0.1858,
+      "step": 10350
+    },
+    {
+      "grad_norm": 0.6448032855987549,
+      "learning_rate": 5.116559062409298e-05,
+      "loss": 0.212,
+      "step": 10360
+    },
+    {
+      "grad_norm": 0.6888636946678162,
+      "learning_rate": 5.10829380446483e-05,
+      "loss": 0.1964,
+      "step": 10370
+    },
+    {
+      "grad_norm": 0.4768984317779541,
+      "learning_rate": 5.100028250449235e-05,
+      "loss": 0.2231,
+      "step": 10380
+    },
+    {
+      "grad_norm": 0.6630909442901611,
+      "learning_rate": 5.0917624229602234e-05,
+      "loss": 0.1938,
+      "step": 10390
+    },
+    {
+      "grad_norm": 0.565851092338562,
+      "learning_rate": 5.0834963445962524e-05,
+      "loss": 0.1772,
+      "step": 10400
+    },
+    {
+      "grad_norm": 0.5517013072967529,
+      "learning_rate": 5.075230037956461e-05,
+      "loss": 0.1816,
+      "step": 10410
+    },
+    {
+      "grad_norm": 0.452099084854126,
+      "learning_rate": 5.0669635256406213e-05,
+      "loss": 0.1995,
+      "step": 10420
+    },
+    {
+      "grad_norm": 0.5043447613716125,
+      "learning_rate": 5.058696830249058e-05,
+      "loss": 0.1827,
+      "step": 10430
+    },
+    {
+      "grad_norm": 0.8682702779769897,
+      "learning_rate": 5.050429974382602e-05,
+      "loss": 0.1987,
+      "step": 10440
+    },
+    {
+      "grad_norm": 0.5646085739135742,
+      "learning_rate": 5.042162980642523e-05,
+      "loss": 0.1862,
+      "step": 10450
+    },
+    {
+      "grad_norm": 0.5915936231613159,
+      "learning_rate": 5.033895871630462e-05,
+      "loss": 0.2001,
+      "step": 10460
+    },
+    {
+      "grad_norm": 0.5350385308265686,
+      "learning_rate": 5.025628669948386e-05,
+      "loss": 0.1899,
+      "step": 10470
+    },
+    {
+      "grad_norm": 0.44209498167037964,
+      "learning_rate": 5.017361398198502e-05,
+      "loss": 0.1873,
+      "step": 10480
+    },
+    {
+      "grad_norm": 0.662107527256012,
+      "learning_rate": 5.009094078983221e-05,
+      "loss": 0.1894,
+      "step": 10490
+    },
+    {
+      "grad_norm": 0.6172156929969788,
+      "learning_rate": 5.000826734905073e-05,
+      "loss": 0.1805,
+      "step": 10500
+    },
+    {
+      "grad_norm": 0.3793625235557556,
+      "learning_rate": 4.9925593885666645e-05,
+      "loss": 0.1775,
+      "step": 10510
+    },
+    {
+      "grad_norm": 0.574526846408844,
+      "learning_rate": 4.984292062570602e-05,
+      "loss": 0.204,
+      "step": 10520
+    },
+    {
+      "grad_norm": 0.46478793025016785,
+      "learning_rate": 4.976024779519442e-05,
+      "loss": 0.1871,
+      "step": 10530
+    },
+    {
+      "grad_norm": 0.6187533736228943,
+      "learning_rate": 4.9677575620156194e-05,
+      "loss": 0.196,
+      "step": 10540
+    },
+    {
+      "grad_norm": 0.534498929977417,
+      "learning_rate": 4.959490432661391e-05,
+      "loss": 0.1758,
+      "step": 10550
+    },
+    {
+      "grad_norm": 0.53885817527771,
+      "learning_rate": 4.9512234140587726e-05,
+      "loss": 0.1797,
+      "step": 10560
+    },
+    {
+      "grad_norm": 0.7067932486534119,
+      "learning_rate": 4.942956528809477e-05,
+      "loss": 0.1915,
+      "step": 10570
+    },
+    {
+      "grad_norm": 0.47702959179878235,
+      "learning_rate": 4.934689799514854e-05,
+      "loss": 0.1903,
+      "step": 10580
+    },
+    {
+      "grad_norm": 0.5818909406661987,
+      "learning_rate": 4.926423248775827e-05,
+      "loss": 0.1854,
+      "step": 10590
+    },
+    {
+      "grad_norm": 0.538126528263092,
+      "learning_rate": 4.918156899192826e-05,
+      "loss": 0.2099,
+      "step": 10600
+    },
+    {
+      "grad_norm": 0.5049379467964172,
+      "learning_rate": 4.909890773365738e-05,
+      "loss": 0.1888,
+      "step": 10610
+    },
+    {
+      "grad_norm": 0.5011106729507446,
+      "learning_rate": 4.9016248938938344e-05,
+      "loss": 0.1895,
+      "step": 10620
+    },
+    {
+      "grad_norm": 0.5258385539054871,
+      "learning_rate": 4.8933592833757156e-05,
+      "loss": 0.189,
+      "step": 10630
+    },
+    {
+      "grad_norm": 0.7272977828979492,
+      "learning_rate": 4.8850939644092435e-05,
+      "loss": 0.1831,
+      "step": 10640
+    },
+    {
+      "grad_norm": 0.6661507487297058,
+      "learning_rate": 4.876828959591485e-05,
+      "loss": 0.2,
+      "step": 10650
+    },
+    {
+      "grad_norm": 0.5348215103149414,
+      "learning_rate": 4.8685642915186474e-05,
+      "loss": 0.2009,
+      "step": 10660
+    },
+    {
+      "grad_norm": 0.5352081060409546,
+      "learning_rate": 4.860299982786018e-05,
+      "loss": 0.1784,
+      "step": 10670
+    },
+    {
+      "grad_norm": 0.4694165885448456,
+      "learning_rate": 4.852036055987901e-05,
+      "loss": 0.196,
+      "step": 10680
+    },
+    {
+      "grad_norm": 0.45586293935775757,
+      "learning_rate": 4.843772533717558e-05,
+      "loss": 0.1589,
+      "step": 10690
+    },
+    {
+      "grad_norm": 0.8393077850341797,
+      "learning_rate": 4.835509438567142e-05,
+      "loss": 0.1861,
+      "step": 10700
+    },
+    {
+      "grad_norm": 0.7372359037399292,
+      "learning_rate": 4.827246793127639e-05,
+      "loss": 0.1734,
+      "step": 10710
+    },
+    {
+      "grad_norm": 0.4734967052936554,
+      "learning_rate": 4.818984619988807e-05,
+      "loss": 0.1946,
+      "step": 10720
+    },
+    {
+      "grad_norm": 0.5520846247673035,
+      "learning_rate": 4.810722941739115e-05,
+      "loss": 0.1847,
+      "step": 10730
+    },
+    {
+      "grad_norm": 0.7998974323272705,
+      "learning_rate": 4.8024617809656684e-05,
+      "loss": 0.1901,
+      "step": 10740
+    },
+    {
+      "grad_norm": 0.5827769041061401,
+      "learning_rate": 4.794201160254171e-05,
+      "loss": 0.2006,
+      "step": 10750
+    },
+    {
+      "grad_norm": 0.4608442783355713,
+      "learning_rate": 4.785941102188844e-05,
+      "loss": 0.1893,
+      "step": 10760
+    },
+    {
+      "grad_norm": 0.5592003464698792,
+      "learning_rate": 4.7776816293523686e-05,
+      "loss": 0.1829,
+      "step": 10770
+    },
+    {
+      "grad_norm": 0.7013199925422668,
+      "learning_rate": 4.769422764325832e-05,
+      "loss": 0.1931,
+      "step": 10780
+    },
+    {
+      "grad_norm": 0.8320543169975281,
+      "learning_rate": 4.76116452968865e-05,
+      "loss": 0.1727,
+      "step": 10790
+    },
+    {
+      "grad_norm": 0.5147238969802856,
+      "learning_rate": 4.752906948018525e-05,
+      "loss": 0.1816,
+      "step": 10800
+    },
+    {
+      "grad_norm": 0.4769739508628845,
+      "learning_rate": 4.7446500418913684e-05,
+      "loss": 0.1937,
+      "step": 10810
+    },
+    {
+      "grad_norm": 0.7855521440505981,
+      "learning_rate": 4.736393833881247e-05,
+      "loss": 0.1962,
+      "step": 10820
+    },
+    {
+      "grad_norm": 0.43348100781440735,
+      "learning_rate": 4.7281383465603194e-05,
+      "loss": 0.1833,
+      "step": 10830
+    },
+    {
+      "grad_norm": 0.9891307353973389,
+      "learning_rate": 4.71988360249877e-05,
+      "loss": 0.2012,
+      "step": 10840
+    },
+    {
+      "grad_norm": 0.7483139038085938,
+      "learning_rate": 4.7116296242647554e-05,
+      "loss": 0.1952,
+      "step": 10850
+    },
+    {
+      "grad_norm": 0.5029531121253967,
+      "learning_rate": 4.703376434424336e-05,
+      "loss": 0.1856,
+      "step": 10860
+    },
+    {
+      "grad_norm": 0.5491126179695129,
+      "learning_rate": 4.695124055541421e-05,
+      "loss": 0.2092,
+      "step": 10870
+    },
+    {
+      "grad_norm": 0.5565106868743896,
+      "learning_rate": 4.6868725101776934e-05,
+      "loss": 0.1925,
+      "step": 10880
+    },
+    {
+      "grad_norm": 0.5181046724319458,
+      "learning_rate": 4.678621820892567e-05,
+      "loss": 0.2,
+      "step": 10890
+    },
+    {
+      "grad_norm": 0.5177876949310303,
+      "learning_rate": 4.670372010243111e-05,
+      "loss": 0.1882,
+      "step": 10900
+    },
+    {
+      "grad_norm": 0.46149250864982605,
+      "learning_rate": 4.662123100783992e-05,
+      "loss": 0.1968,
+      "step": 10910
+    },
+    {
+      "grad_norm": 0.5935540199279785,
+      "learning_rate": 4.653875115067415e-05,
+      "loss": 0.1849,
+      "step": 10920
+    },
+    {
+      "grad_norm": 0.6780995726585388,
+      "learning_rate": 4.6456280756430545e-05,
+      "loss": 0.1983,
+      "step": 10930
+    },
+    {
+      "grad_norm": 0.7507543563842773,
+      "learning_rate": 4.637382005058004e-05,
+      "loss": 0.1805,
+      "step": 10940
+    },
+    {
+      "grad_norm": 0.5642117261886597,
+      "learning_rate": 4.629136925856705e-05,
+      "loss": 0.1878,
+      "step": 10950
+    },
+    {
+      "grad_norm": 0.39535418152809143,
+      "learning_rate": 4.6208928605808895e-05,
+      "loss": 0.1817,
+      "step": 10960
+    },
+    {
+      "grad_norm": 0.5035179257392883,
+      "learning_rate": 4.612649831769519e-05,
+      "loss": 0.2001,
+      "step": 10970
+    },
+    {
+      "grad_norm": 0.6236174702644348,
+      "learning_rate": 4.604407861958715e-05,
+      "loss": 0.1864,
+      "step": 10980
+    },
+    {
+      "grad_norm": 0.5145127177238464,
+      "learning_rate": 4.5961669736817114e-05,
+      "loss": 0.1991,
+      "step": 10990
+    },
+    {
+      "grad_norm": 0.43264245986938477,
+      "learning_rate": 4.5879271894687814e-05,
+      "loss": 0.1819,
+      "step": 11000
+    },
+    {
+      "grad_norm": 0.6213241815567017,
+      "learning_rate": 4.5796885318471826e-05,
+      "loss": 0.1871,
+      "step": 11010
+    },
+    {
+      "grad_norm": 0.6162437200546265,
+      "learning_rate": 4.571451023341086e-05,
+      "loss": 0.1862,
+      "step": 11020
+    },
+    {
+      "grad_norm": 0.649620771408081,
+      "learning_rate": 4.563214686471527e-05,
+      "loss": 0.1758,
+      "step": 11030
+    },
+    {
+      "grad_norm": 0.5043689012527466,
+      "learning_rate": 4.5549795437563365e-05,
+      "loss": 0.2102,
+      "step": 11040
+    },
+    {
+      "grad_norm": 0.8355112671852112,
+      "learning_rate": 4.546745617710081e-05,
+      "loss": 0.1869,
+      "step": 11050
+    },
+    {
+      "grad_norm": 0.8619815707206726,
+      "learning_rate": 4.5385129308440014e-05,
+      "loss": 0.2023,
+      "step": 11060
+    },
+    {
+      "grad_norm": 0.7279309630393982,
+      "learning_rate": 4.530281505665944e-05,
+      "loss": 0.1798,
+      "step": 11070
+    },
+    {
+      "grad_norm": 0.532270610332489,
+      "learning_rate": 4.5220513646803134e-05,
+      "loss": 0.1913,
+      "step": 11080
+    },
+    {
+      "grad_norm": 0.6851012706756592,
+      "learning_rate": 4.513822530388003e-05,
+      "loss": 0.1931,
+      "step": 11090
+    },
+    {
+      "grad_norm": 0.4894924759864807,
+      "learning_rate": 4.5055950252863296e-05,
+      "loss": 0.1948,
+      "step": 11100
+    },
+    {
+      "grad_norm": 0.6766902804374695,
+      "learning_rate": 4.4973688718689803e-05,
+      "loss": 0.212,
+      "step": 11110
+    },
+    {
+      "grad_norm": 0.5810624957084656,
+      "learning_rate": 4.4891440926259406e-05,
+      "loss": 0.2032,
+      "step": 11120
+    },
+    {
+      "grad_norm": 0.6164013743400574,
+      "learning_rate": 4.480920710043443e-05,
+      "loss": 0.1856,
+      "step": 11130
+    },
+    {
+      "grad_norm": 0.487947553396225,
+      "learning_rate": 4.4726987466039044e-05,
+      "loss": 0.2001,
+      "step": 11140
+    },
+    {
+      "grad_norm": 0.4141201078891754,
+      "learning_rate": 4.46447822478586e-05,
+      "loss": 0.1891,
+      "step": 11150
+    },
+    {
+      "grad_norm": 0.5276414752006531,
+      "learning_rate": 4.4562591670638974e-05,
+      "loss": 0.1933,
+      "step": 11160
+    },
+    {
+      "grad_norm": 0.5534875392913818,
+      "learning_rate": 4.4480415959086105e-05,
+      "loss": 0.1758,
+      "step": 11170
+    },
+    {
+      "grad_norm": 0.6195520758628845,
+      "learning_rate": 4.439825533786522e-05,
+      "loss": 0.1843,
+      "step": 11180
+    },
+    {
+      "grad_norm": 0.6805815100669861,
+      "learning_rate": 4.431611003160035e-05,
+      "loss": 0.1847,
+      "step": 11190
+    },
+    {
+      "grad_norm": 0.5071933269500732,
+      "learning_rate": 4.4233980264873636e-05,
+      "loss": 0.1692,
+      "step": 11200
+    },
+    {
+      "grad_norm": 0.47736048698425293,
+      "learning_rate": 4.4151866262224684e-05,
+      "loss": 0.1891,
+      "step": 11210
+    },
+    {
+      "grad_norm": 1.2811380624771118,
+      "learning_rate": 4.406976824815006e-05,
+      "loss": 0.2043,
+      "step": 11220
+    },
+    {
+      "grad_norm": 0.3782571256160736,
+      "learning_rate": 4.3987686447102595e-05,
+      "loss": 0.1825,
+      "step": 11230
+    },
+    {
+      "grad_norm": 0.4263767898082733,
+      "learning_rate": 4.3905621083490804e-05,
+      "loss": 0.1962,
+      "step": 11240
+    },
+    {
+      "grad_norm": 0.5812981128692627,
+      "learning_rate": 4.3823572381678286e-05,
+      "loss": 0.1778,
+      "step": 11250
+    },
+    {
+      "grad_norm": 0.5018457174301147,
+      "learning_rate": 4.374154056598301e-05,
+      "loss": 0.1919,
+      "step": 11260
+    },
+    {
+      "grad_norm": 0.47254663705825806,
+      "learning_rate": 4.3659525860676845e-05,
+      "loss": 0.1787,
+      "step": 11270
+    },
+    {
+      "grad_norm": 0.6569596529006958,
+      "learning_rate": 4.3577528489984854e-05,
+      "loss": 0.2003,
+      "step": 11280
+    },
+    {
+      "grad_norm": 0.46944305300712585,
+      "learning_rate": 4.349554867808476e-05,
+      "loss": 0.1997,
+      "step": 11290
+    },
+    {
+      "grad_norm": 0.5730595588684082,
+      "learning_rate": 4.34135866491062e-05,
+      "loss": 0.1988,
+      "step": 11300
+    },
+    {
+      "grad_norm": 0.4969821572303772,
+      "learning_rate": 4.333164262713022e-05,
+      "loss": 0.1869,
+      "step": 11310
+    },
+    {
+      "grad_norm": 0.4558808505535126,
+      "learning_rate": 4.324971683618868e-05,
+      "loss": 0.1783,
+      "step": 11320
+    },
+    {
+      "grad_norm": 0.5664698481559753,
+      "learning_rate": 4.316780950026354e-05,
+      "loss": 0.1934,
+      "step": 11330
+    },
+    {
+      "grad_norm": 0.6081880331039429,
+      "learning_rate": 4.308592084328637e-05,
+      "loss": 0.1794,
+      "step": 11340
+    },
+    {
+      "grad_norm": 0.6394439339637756,
+      "learning_rate": 4.3004051089137576e-05,
+      "loss": 0.1824,
+      "step": 11350
+    },
+    {
+      "grad_norm": 0.6486170887947083,
+      "learning_rate": 4.292220046164597e-05,
+      "loss": 0.1826,
+      "step": 11360
+    },
+    {
+      "grad_norm": 0.4712790548801422,
+      "learning_rate": 4.2840369184588035e-05,
+      "loss": 0.1824,
+      "step": 11370
+    },
+    {
+      "grad_norm": 0.5029439330101013,
+      "learning_rate": 4.2758557481687345e-05,
+      "loss": 0.1827,
+      "step": 11380
+    },
+    {
+      "grad_norm": 0.40326592326164246,
+      "learning_rate": 4.267676557661403e-05,
+      "loss": 0.1904,
+      "step": 11390
+    },
+    {
+      "grad_norm": 0.5952796936035156,
+      "learning_rate": 4.2594993692983955e-05,
+      "loss": 0.1831,
+      "step": 11400
+    },
+    {
+      "grad_norm": 0.725638210773468,
+      "learning_rate": 4.251324205435837e-05,
+      "loss": 0.1719,
+      "step": 11410
+    },
+    {
+      "grad_norm": 0.48128288984298706,
+      "learning_rate": 4.243151088424312e-05,
+      "loss": 0.1746,
+      "step": 11420
+    },
+    {
+      "grad_norm": 0.6406065225601196,
+      "learning_rate": 4.234980040608813e-05,
+      "loss": 0.2005,
+      "step": 11430
+    },
+    {
+      "grad_norm": 0.48028433322906494,
+      "learning_rate": 4.22681108432867e-05,
+      "loss": 0.1826,
+      "step": 11440
+    },
+    {
+      "grad_norm": 0.5963494181632996,
+      "learning_rate": 4.2186442419174984e-05,
+      "loss": 0.1766,
+      "step": 11450
+    },
+    {
+      "grad_norm": 0.5826793909072876,
+      "learning_rate": 4.210479535703133e-05,
+      "loss": 0.1967,
+      "step": 11460
+    },
+    {
+      "grad_norm": 0.6523909568786621,
+      "learning_rate": 4.202316988007567e-05,
+      "loss": 0.1838,
+      "step": 11470
+    },
+    {
+      "grad_norm": 0.6941064596176147,
+      "learning_rate": 4.194156621146901e-05,
+      "loss": 0.1868,
+      "step": 11480
+    },
+    {
+      "grad_norm": 0.41902777552604675,
+      "learning_rate": 4.1859984574312596e-05,
+      "loss": 0.1717,
+      "step": 11490
+    },
+    {
+      "grad_norm": 0.6556402444839478,
+      "learning_rate": 4.177842519164752e-05,
+      "loss": 0.1827,
+      "step": 11500
+    },
+    {
+      "grad_norm": 0.38877320289611816,
+      "learning_rate": 4.169688828645404e-05,
+      "loss": 0.1829,
+      "step": 11510
+    },
+    {
+      "grad_norm": 0.6213542222976685,
+      "learning_rate": 4.161537408165092e-05,
+      "loss": 0.1867,
+      "step": 11520
+    },
+    {
+      "grad_norm": 0.6299763917922974,
+      "learning_rate": 4.1533882800094924e-05,
+      "loss": 0.1852,
+      "step": 11530
+    },
+    {
+      "grad_norm": 0.45979878306388855,
+      "learning_rate": 4.145241466458005e-05,
+      "loss": 0.1832,
+      "step": 11540
+    },
+    {
+      "grad_norm": 0.5702639222145081,
+      "learning_rate": 4.13709698978371e-05,
+      "loss": 0.1873,
+      "step": 11550
+    },
+    {
+      "grad_norm": 0.399537056684494,
+      "learning_rate": 4.1289548722532944e-05,
+      "loss": 0.1805,
+      "step": 11560
+    },
+    {
+      "grad_norm": 0.4895675480365753,
+      "learning_rate": 4.120815136126999e-05,
+      "loss": 0.1796,
+      "step": 11570
+    },
+    {
+      "grad_norm": 0.6637800335884094,
+      "learning_rate": 4.112677803658548e-05,
+      "loss": 0.1858,
+      "step": 11580
+    },
+    {
+      "grad_norm": 0.6842101812362671,
+      "learning_rate": 4.1045428970951e-05,
+      "loss": 0.1732,
+      "step": 11590
+    },
+    {
+      "grad_norm": 0.45011740922927856,
+      "learning_rate": 4.0964104386771785e-05,
+      "loss": 0.2007,
+      "step": 11600
+    },
+    {
+      "grad_norm": 0.675012469291687,
+      "learning_rate": 4.0882804506386144e-05,
+      "loss": 0.1878,
+      "step": 11610
+    },
+    {
+      "grad_norm": 0.7779334187507629,
+      "learning_rate": 4.080152955206485e-05,
+      "loss": 0.1885,
+      "step": 11620
+    },
+    {
+      "grad_norm": 0.629752516746521,
+      "learning_rate": 4.0720279746010505e-05,
+      "loss": 0.1687,
+      "step": 11630
+    },
+    {
+      "grad_norm": 0.6045376062393188,
+      "learning_rate": 4.063905531035699e-05,
+      "loss": 0.1897,
+      "step": 11640
+    },
+    {
+      "grad_norm": 0.60813969373703,
+      "learning_rate": 4.055785646716882e-05,
+      "loss": 0.183,
+      "step": 11650
+    },
+    {
+      "grad_norm": 0.708234429359436,
+      "learning_rate": 4.047668343844051e-05,
+      "loss": 0.1979,
+      "step": 11660
+    },
+    {
+      "grad_norm": 0.45979952812194824,
+      "learning_rate": 4.039553644609604e-05,
+      "loss": 0.1789,
+      "step": 11670
+    },
+    {
+      "grad_norm": 0.5996567606925964,
+      "learning_rate": 4.0314415711988176e-05,
+      "loss": 0.1822,
+      "step": 11680
+    },
+    {
+      "grad_norm": 0.5695363283157349,
+      "learning_rate": 4.023332145789792e-05,
+      "loss": 0.1678,
+      "step": 11690
+    },
+    {
+      "grad_norm": 0.5068591237068176,
+      "learning_rate": 4.015225390553385e-05,
+      "loss": 0.1942,
+      "step": 11700
+    },
+    {
+      "grad_norm": 0.6761585474014282,
+      "learning_rate": 4.007121327653158e-05,
+      "loss": 0.192,
+      "step": 11710
+    },
+    {
+      "grad_norm": 0.5368741154670715,
+      "learning_rate": 3.9990199792453064e-05,
+      "loss": 0.1807,
+      "step": 11720
+    },
+    {
+      "grad_norm": 0.5772503614425659,
+      "learning_rate": 3.9909213674786103e-05,
+      "loss": 0.1857,
+      "step": 11730
+    },
+    {
+      "grad_norm": 0.5463266372680664,
+      "learning_rate": 3.982825514494363e-05,
+      "loss": 0.182,
+      "step": 11740
+    },
+    {
+      "grad_norm": 0.46924707293510437,
+      "learning_rate": 3.974732442426319e-05,
+      "loss": 0.1776,
+      "step": 11750
+    },
+    {
+      "grad_norm": 0.560883641242981,
+      "learning_rate": 3.966642173400629e-05,
+      "loss": 0.176,
+      "step": 11760
+    },
+    {
+      "grad_norm": 0.5194474458694458,
+      "learning_rate": 3.9585547295357764e-05,
+      "loss": 0.1829,
+      "step": 11770
+    },
+    {
+      "grad_norm": 0.5968332290649414,
+      "learning_rate": 3.950470132942526e-05,
+      "loss": 0.1875,
+      "step": 11780
+    },
+    {
+      "grad_norm": 0.46108946204185486,
+      "learning_rate": 3.942388405723856e-05,
+      "loss": 0.1708,
+      "step": 11790
+    },
+    {
+      "grad_norm": 0.6097173094749451,
+      "learning_rate": 3.9343095699749e-05,
+      "loss": 0.1812,
+      "step": 11800
+    },
+    {
+      "grad_norm": 0.4924684464931488,
+      "learning_rate": 3.9262336477828874e-05,
+      "loss": 0.1715,
+      "step": 11810
+    },
+    {
+      "grad_norm": 0.4776810109615326,
+      "learning_rate": 3.9181606612270794e-05,
+      "loss": 0.2045,
+      "step": 11820
+    },
+    {
+      "grad_norm": 0.851047933101654,
+      "learning_rate": 3.910090632378713e-05,
+      "loss": 0.1913,
+      "step": 11830
+    },
+    {
+      "grad_norm": 0.4778156876564026,
+      "learning_rate": 3.90202358330094e-05,
+      "loss": 0.2092,
+      "step": 11840
+    },
+    {
+      "grad_norm": 0.5308887362480164,
+      "learning_rate": 3.8939595360487656e-05,
+      "loss": 0.1873,
+      "step": 11850
+    },
+    {
+      "grad_norm": 0.671119749546051,
+      "learning_rate": 3.885898512668984e-05,
+      "loss": 0.192,
+      "step": 11860
+    },
+    {
+      "grad_norm": 0.4529836177825928,
+      "learning_rate": 3.877840535200127e-05,
+      "loss": 0.1816,
+      "step": 11870
+    },
+    {
+      "grad_norm": 0.5047312378883362,
+      "learning_rate": 3.869785625672397e-05,
+      "loss": 0.1852,
+      "step": 11880
+    },
+    {
+      "grad_norm": 0.44674035906791687,
+      "learning_rate": 3.8617338061076094e-05,
+      "loss": 0.1806,
+      "step": 11890
+    },
+    {
+      "grad_norm": 0.7303107380867004,
+      "learning_rate": 3.853685098519132e-05,
+      "loss": 0.2045,
+      "step": 11900
+    },
+    {
+      "grad_norm": 0.3832497298717499,
+      "learning_rate": 3.845639524911823e-05,
+      "loss": 0.1837,
+      "step": 11910
+    },
+    {
+      "grad_norm": 0.6022061109542847,
+      "learning_rate": 3.837597107281974e-05,
+      "loss": 0.1949,
+      "step": 11920
+    },
+    {
+      "grad_norm": 0.48971471190452576,
+      "learning_rate": 3.829557867617247e-05,
+      "loss": 0.1862,
+      "step": 11930
+    },
+    {
+      "grad_norm": 0.5695327520370483,
+      "learning_rate": 3.821521827896618e-05,
+      "loss": 0.184,
+      "step": 11940
+    },
+    {
+      "grad_norm": 0.5861474275588989,
+      "learning_rate": 3.81348901009031e-05,
+      "loss": 0.182,
+      "step": 11950
+    },
+    {
+      "grad_norm": 0.5904955267906189,
+      "learning_rate": 3.805459436159741e-05,
+      "loss": 0.1801,
+      "step": 11960
+    },
+    {
+      "grad_norm": 0.49573782086372375,
+      "learning_rate": 3.797433128057461e-05,
+      "loss": 0.2006,
+      "step": 11970
+    },
+    {
+      "grad_norm": 0.7443907856941223,
+      "learning_rate": 3.789410107727089e-05,
+      "loss": 0.181,
+      "step": 11980
+    },
+    {
+      "grad_norm": 0.6684632897377014,
+      "learning_rate": 3.781390397103257e-05,
+      "loss": 0.1792,
+      "step": 11990
+    },
+    {
+      "grad_norm": 0.7872788310050964,
+      "learning_rate": 3.7733740181115455e-05,
+      "loss": 0.1921,
+      "step": 12000
+    },
+    {
+      "grad_norm": 0.6138386726379395,
+      "learning_rate": 3.7653609926684306e-05,
+      "loss": 0.1762,
+      "step": 12010
+    },
+    {
+      "grad_norm": 0.6283270716667175,
+      "learning_rate": 3.757351342681217e-05,
+      "loss": 0.21,
+      "step": 12020
+    },
+    {
+      "grad_norm": 0.8601664900779724,
+      "learning_rate": 3.749345090047982e-05,
+      "loss": 0.1986,
+      "step": 12030
+    },
+    {
+      "grad_norm": 0.4480983316898346,
+      "learning_rate": 3.741342256657515e-05,
+      "loss": 0.1781,
+      "step": 12040
+    },
+    {
+      "grad_norm": 0.523077130317688,
+      "learning_rate": 3.7333428643892567e-05,
+      "loss": 0.1842,
+      "step": 12050
+    },
+    {
+      "grad_norm": 0.5474849939346313,
+      "learning_rate": 3.725346935113239e-05,
+      "loss": 0.1951,
+      "step": 12060
+    },
+    {
+      "grad_norm": 0.5857459902763367,
+      "learning_rate": 3.717354490690029e-05,
+      "loss": 0.1863,
+      "step": 12070
+    },
+    {
+      "grad_norm": 0.5759073495864868,
+      "learning_rate": 3.709365552970664e-05,
+      "loss": 0.1803,
+      "step": 12080
+    },
+    {
+      "grad_norm": 0.6624888181686401,
+      "learning_rate": 3.7013801437965945e-05,
+      "loss": 0.188,
+      "step": 12090
+    },
+    {
+      "grad_norm": 0.40444815158843994,
+      "learning_rate": 3.693398284999623e-05,
+      "loss": 0.1851,
+      "step": 12100
+    },
+    {
+      "grad_norm": 0.5732042193412781,
+      "learning_rate": 3.6854199984018484e-05,
+      "loss": 0.1674,
+      "step": 12110
+    },
+    {
+      "grad_norm": 0.5677944421768188,
+      "learning_rate": 3.677445305815601e-05,
+      "loss": 0.1828,
+      "step": 12120
+    },
+    {
+      "grad_norm": 0.599403440952301,
+      "learning_rate": 3.669474229043387e-05,
+      "loss": 0.1981,
+      "step": 12130
+    },
+    {
+      "grad_norm": 0.7384734749794006,
+      "learning_rate": 3.6615067898778235e-05,
+      "loss": 0.1864,
+      "step": 12140
+    },
+    {
+      "grad_norm": 0.5349586009979248,
+      "learning_rate": 3.6535430101015866e-05,
+      "loss": 0.1747,
+      "step": 12150
+    },
+    {
+      "grad_norm": 0.596298098564148,
+      "learning_rate": 3.645582911487345e-05,
+      "loss": 0.2003,
+      "step": 12160
+    },
+    {
+      "grad_norm": 0.5655982494354248,
+      "learning_rate": 3.637626515797706e-05,
+      "loss": 0.1789,
+      "step": 12170
+    },
+    {
+      "grad_norm": 0.46314722299575806,
+      "learning_rate": 3.629673844785152e-05,
+      "loss": 0.1994,
+      "step": 12180
+    },
+    {
+      "grad_norm": 0.723676860332489,
+      "learning_rate": 3.621724920191979e-05,
+      "loss": 0.1756,
+      "step": 12190
+    },
+    {
+      "grad_norm": 0.616674542427063,
+      "learning_rate": 3.6137797637502444e-05,
+      "loss": 0.1778,
+      "step": 12200
+    },
+    {
+      "grad_norm": 0.5770185589790344,
+      "learning_rate": 3.6058383971817035e-05,
+      "loss": 0.1913,
+      "step": 12210
+    },
+    {
+      "grad_norm": 0.6013627648353577,
+      "learning_rate": 3.59790084219775e-05,
+      "loss": 0.1852,
+      "step": 12220
+    },
+    {
+      "grad_norm": 0.5854552388191223,
+      "learning_rate": 3.589967120499353e-05,
+      "loss": 0.1713,
+      "step": 12230
+    },
+    {
+      "grad_norm": 0.6339314579963684,
+      "learning_rate": 3.5820372537770075e-05,
+      "loss": 0.1975,
+      "step": 12240
+    },
+    {
+      "grad_norm": 0.4986847937107086,
+      "learning_rate": 3.5741112637106655e-05,
+      "loss": 0.1878,
+      "step": 12250
+    },
+    {
+      "grad_norm": 0.7464695572853088,
+      "learning_rate": 3.5661891719696804e-05,
+      "loss": 0.1896,
+      "step": 12260
+    },
+    {
+      "grad_norm": 0.4946253001689911,
+      "learning_rate": 3.5582710002127504e-05,
+      "loss": 0.185,
+      "step": 12270
+    },
+    {
+      "grad_norm": 0.5305449962615967,
+      "learning_rate": 3.550356770087853e-05,
+      "loss": 0.1892,
+      "step": 12280
+    },
+    {
+      "grad_norm": 0.6125974059104919,
+      "learning_rate": 3.5424465032321914e-05,
+      "loss": 0.1959,
+      "step": 12290
+    },
+    {
+      "grad_norm": 0.41510531306266785,
+      "learning_rate": 3.5345402212721335e-05,
+      "loss": 0.1973,
+      "step": 12300
+    },
+    {
+      "grad_norm": 0.5443097352981567,
+      "learning_rate": 3.526637945823152e-05,
+      "loss": 0.1823,
+      "step": 12310
+    },
+    {
+      "grad_norm": 0.5749171376228333,
+      "learning_rate": 3.518739698489767e-05,
+      "loss": 0.1806,
+      "step": 12320
+    },
+    {
+      "grad_norm": 0.5086957812309265,
+      "learning_rate": 3.510845500865485e-05,
+      "loss": 0.1811,
+      "step": 12330
+    },
+    {
+      "grad_norm": 0.4774307310581207,
+      "learning_rate": 3.502955374532739e-05,
+      "loss": 0.1957,
+      "step": 12340
+    },
+    {
+      "grad_norm": 0.5163477063179016,
+      "learning_rate": 3.495069341062836e-05,
+      "loss": 0.1941,
+      "step": 12350
+    },
+    {
+      "grad_norm": 0.7605944275856018,
+      "learning_rate": 3.4871874220158896e-05,
+      "loss": 0.1979,
+      "step": 12360
+    },
+    {
+      "grad_norm": 0.5929067134857178,
+      "learning_rate": 3.479309638940762e-05,
+      "loss": 0.1883,
+      "step": 12370
+    },
+    {
+      "grad_norm": 0.7660536170005798,
+      "learning_rate": 3.4714360133750146e-05,
+      "loss": 0.1913,
+      "step": 12380
+    },
+    {
+      "grad_norm": 0.42936426401138306,
+      "learning_rate": 3.463566566844839e-05,
+      "loss": 0.1819,
+      "step": 12390
+    },
+    {
+      "grad_norm": 0.6468411087989807,
+      "learning_rate": 3.4557013208650016e-05,
+      "loss": 0.1821,
+      "step": 12400
+    },
+    {
+      "grad_norm": 0.5163009762763977,
+      "learning_rate": 3.4478402969387857e-05,
+      "loss": 0.1875,
+      "step": 12410
+    },
+    {
+      "grad_norm": 0.696492075920105,
+      "learning_rate": 3.4399835165579266e-05,
+      "loss": 0.1912,
+      "step": 12420
+    },
+    {
+      "grad_norm": 0.5546051263809204,
+      "learning_rate": 3.4321310012025645e-05,
+      "loss": 0.1989,
+      "step": 12430
+    },
+    {
+      "grad_norm": 0.7930641770362854,
+      "learning_rate": 3.424282772341176e-05,
+      "loss": 0.1866,
+      "step": 12440
+    },
+    {
+      "grad_norm": 0.9020024538040161,
+      "learning_rate": 3.416438851430519e-05,
+      "loss": 0.1839,
+      "step": 12450
+    },
+    {
+      "grad_norm": 0.6731504201889038,
+      "learning_rate": 3.408599259915577e-05,
+      "loss": 0.1888,
+      "step": 12460
+    },
+    {
+      "grad_norm": 0.6925093531608582,
+      "learning_rate": 3.400764019229487e-05,
+      "loss": 0.1758,
+      "step": 12470
+    },
+    {
+      "grad_norm": 0.6492420434951782,
+      "learning_rate": 3.3929331507935035e-05,
+      "loss": 0.168,
+      "step": 12480
+    },
+    {
+      "grad_norm": 0.6595091223716736,
+      "learning_rate": 3.3851066760169196e-05,
+      "loss": 0.2031,
+      "step": 12490
+    },
+    {
+      "grad_norm": 0.4686817526817322,
+      "learning_rate": 3.377284616297021e-05,
+      "loss": 0.1932,
+      "step": 12500
+    },
+    {
+      "grad_norm": 0.47329333424568176,
+      "learning_rate": 3.3694669930190166e-05,
+      "loss": 0.1873,
+      "step": 12510
+    },
+    {
+      "grad_norm": 0.5107689499855042,
+      "learning_rate": 3.36165382755599e-05,
+      "loss": 0.181,
+      "step": 12520
+    },
+    {
+      "grad_norm": 0.5841307044029236,
+      "learning_rate": 3.35384514126884e-05,
+      "loss": 0.1917,
+      "step": 12530
+    },
+    {
+      "grad_norm": 0.5091886520385742,
+      "learning_rate": 3.3460409555062154e-05,
+      "loss": 0.1824,
+      "step": 12540
+    },
+    {
+      "grad_norm": 0.4074237048625946,
+      "learning_rate": 3.3382412916044645e-05,
+      "loss": 0.1851,
+      "step": 12550
+    },
+    {
+      "grad_norm": 0.6309996843338013,
+      "learning_rate": 3.330446170887566e-05,
+      "loss": 0.191,
+      "step": 12560
+    },
+    {
+      "grad_norm": 0.9555942416191101,
+      "learning_rate": 3.3226556146670834e-05,
+      "loss": 0.2006,
+      "step": 12570
+    },
+    {
+      "grad_norm": 0.7077241539955139,
+      "learning_rate": 3.314869644242102e-05,
+      "loss": 0.1769,
+      "step": 12580
+    },
+    {
+      "grad_norm": 0.5194014310836792,
+      "learning_rate": 3.3070882808991674e-05,
+      "loss": 0.1863,
+      "step": 12590
+    },
+    {
+      "grad_norm": 0.5001497864723206,
+      "learning_rate": 3.2993115459122305e-05,
+      "loss": 0.1947,
+      "step": 12600
+    },
+    {
+      "grad_norm": 0.7991508841514587,
+      "learning_rate": 3.2915394605425835e-05,
+      "loss": 0.1813,
+      "step": 12610
+    },
+    {
+      "grad_norm": 0.5455057621002197,
+      "learning_rate": 3.283772046038816e-05,
+      "loss": 0.1901,
+      "step": 12620
+    },
+    {
+      "grad_norm": 0.6000956296920776,
+      "learning_rate": 3.276009323636739e-05,
+      "loss": 0.192,
+      "step": 12630
+    },
+    {
+      "grad_norm": 0.4884570240974426,
+      "learning_rate": 3.268251314559344e-05,
+      "loss": 0.1849,
+      "step": 12640
+    },
+    {
+      "grad_norm": 0.5747490525245667,
+      "learning_rate": 3.2604980400167254e-05,
+      "loss": 0.181,
+      "step": 12650
+    },
+    {
+      "grad_norm": 0.6687218546867371,
+      "learning_rate": 3.252749521206042e-05,
+      "loss": 0.2065,
+      "step": 12660
+    },
+    {
+      "grad_norm": 0.587972104549408,
+      "learning_rate": 3.2450057793114494e-05,
+      "loss": 0.1876,
+      "step": 12670
+    },
+    {
+      "grad_norm": 0.4784529507160187,
+      "learning_rate": 3.2372668355040435e-05,
+      "loss": 0.1818,
+      "step": 12680
+    },
+    {
+      "grad_norm": 0.4098474383354187,
+      "learning_rate": 3.2295327109418005e-05,
+      "loss": 0.1784,
+      "step": 12690
+    },
+    {
+      "grad_norm": 0.6101776361465454,
+      "learning_rate": 3.221803426769518e-05,
+      "loss": 0.189,
+      "step": 12700
+    },
+    {
+      "grad_norm": 0.6486833691596985,
+      "learning_rate": 3.214079004118768e-05,
+      "loss": 0.1789,
+      "step": 12710
+    },
+    {
+      "grad_norm": 0.6838169693946838,
+      "learning_rate": 3.2063594641078234e-05,
+      "loss": 0.1983,
+      "step": 12720
+    },
+    {
+      "grad_norm": 0.9375953674316406,
+      "learning_rate": 3.198644827841616e-05,
+      "loss": 0.193,
+      "step": 12730
+    },
+    {
+      "grad_norm": 0.6022742390632629,
+      "learning_rate": 3.1909351164116654e-05,
+      "loss": 0.1789,
+      "step": 12740
+    },
+    {
+      "grad_norm": 0.7463772892951965,
+      "learning_rate": 3.183230350896026e-05,
+      "loss": 0.1901,
+      "step": 12750
+    },
+    {
+      "grad_norm": 0.4773140251636505,
+      "learning_rate": 3.1755305523592337e-05,
+      "loss": 0.1828,
+      "step": 12760
+    },
+    {
+      "grad_norm": 0.43412017822265625,
+      "learning_rate": 3.167835741852245e-05,
+      "loss": 0.195,
+      "step": 12770
+    },
+    {
+      "grad_norm": 0.575477659702301,
+      "learning_rate": 3.160145940412378e-05,
+      "loss": 0.1976,
+      "step": 12780
+    },
+    {
+      "grad_norm": 0.5075719952583313,
+      "learning_rate": 3.1524611690632545e-05,
+      "loss": 0.1977,
+      "step": 12790
+    },
+    {
+      "grad_norm": 0.7830810546875,
+      "learning_rate": 3.144781448814746e-05,
+      "loss": 0.1934,
+      "step": 12800
+    },
+    {
+      "grad_norm": 0.3786950707435608,
+      "learning_rate": 3.1371068006629145e-05,
+      "loss": 0.186,
+      "step": 12810
+    },
+    {
+      "grad_norm": 0.3866654336452484,
+      "learning_rate": 3.129437245589956e-05,
+      "loss": 0.1873,
+      "step": 12820
+    },
+    {
+      "grad_norm": 0.6159698963165283,
+      "learning_rate": 3.121772804564143e-05,
+      "loss": 0.1935,
+      "step": 12830
+    },
+    {
+      "grad_norm": 0.5301638841629028,
+      "learning_rate": 3.11411349853976e-05,
+      "loss": 0.1761,
+      "step": 12840
+    },
+    {
+      "grad_norm": 0.6358177065849304,
+      "learning_rate": 3.10645934845706e-05,
+      "loss": 0.1927,
+      "step": 12850
+    },
+    {
+      "grad_norm": 0.5462133884429932,
+      "learning_rate": 3.098810375242196e-05,
+      "loss": 0.1998,
+      "step": 12860
+    },
+    {
+      "grad_norm": 0.5117841958999634,
+      "learning_rate": 3.0911665998071704e-05,
+      "loss": 0.1893,
+      "step": 12870
+    },
+    {
+      "grad_norm": 0.6373854875564575,
+      "learning_rate": 3.083528043049774e-05,
+      "loss": 0.1897,
+      "step": 12880
+    },
+    {
+      "grad_norm": 0.5433614253997803,
+      "learning_rate": 3.0758947258535255e-05,
+      "loss": 0.1851,
+      "step": 12890
+    },
+    {
+      "grad_norm": 0.5387536883354187,
+      "learning_rate": 3.068266669087625e-05,
+      "loss": 0.1818,
+      "step": 12900
+    },
+    {
+      "grad_norm": 0.8177366852760315,
+      "learning_rate": 3.060643893606887e-05,
+      "loss": 0.2066,
+      "step": 12910
+    },
+    {
+      "grad_norm": 0.48732468485832214,
+      "learning_rate": 3.053026420251693e-05,
+      "loss": 0.1801,
+      "step": 12920
+    },
+    {
+      "grad_norm": 0.5326870083808899,
+      "learning_rate": 3.0454142698479183e-05,
+      "loss": 0.1972,
+      "step": 12930
+    },
+    {
+      "grad_norm": 0.5280678868293762,
+      "learning_rate": 3.0378074632068954e-05,
+      "loss": 0.1927,
+      "step": 12940
+    },
+    {
+      "grad_norm": 0.5140218734741211,
+      "learning_rate": 3.0302060211253408e-05,
+      "loss": 0.1756,
+      "step": 12950
+    },
+    {
+      "grad_norm": 0.49693265557289124,
+      "learning_rate": 3.0226099643853073e-05,
+      "loss": 0.1858,
+      "step": 12960
+    },
+    {
+      "grad_norm": 0.5547195076942444,
+      "learning_rate": 3.0150193137541283e-05,
+      "loss": 0.1917,
+      "step": 12970
+    },
+    {
+      "grad_norm": 0.4737677574157715,
+      "learning_rate": 3.0074340899843467e-05,
+      "loss": 0.1781,
+      "step": 12980
+    },
+    {
+      "grad_norm": 0.5268831253051758,
+      "learning_rate": 2.999854313813677e-05,
+      "loss": 0.1935,
+      "step": 12990
+    },
+    {
+      "grad_norm": 0.4259425103664398,
+      "learning_rate": 2.9922800059649382e-05,
+      "loss": 0.1766,
+      "step": 13000
+    },
+    {
+      "grad_norm": 0.6276829838752747,
+      "learning_rate": 2.9847111871459976e-05,
+      "loss": 0.19,
+      "step": 13010
+    },
+    {
+      "grad_norm": 0.6113032102584839,
+      "learning_rate": 2.977147878049721e-05,
+      "loss": 0.1819,
+      "step": 13020
+    },
+    {
+      "grad_norm": 0.765427827835083,
+      "learning_rate": 2.9695900993539006e-05,
+      "loss": 0.195,
+      "step": 13030
+    },
+    {
+      "grad_norm": 0.503204882144928,
+      "learning_rate": 2.9620378717212183e-05,
+      "loss": 0.1953,
+      "step": 13040
+    },
+    {
+      "grad_norm": 0.5533180236816406,
+      "learning_rate": 2.9544912157991745e-05,
+      "loss": 0.1884,
+      "step": 13050
+    },
+    {
+      "grad_norm": 0.643366277217865,
+      "learning_rate": 2.9469501522200405e-05,
+      "loss": 0.1882,
+      "step": 13060
+    },
+    {
+      "grad_norm": 0.4859360158443451,
+      "learning_rate": 2.9394147016007946e-05,
+      "loss": 0.171,
+      "step": 13070
+    },
+    {
+      "grad_norm": 0.4193473756313324,
+      "learning_rate": 2.9318848845430702e-05,
+      "loss": 0.1797,
+      "step": 13080
+    },
+    {
+      "grad_norm": 0.6340864300727844,
+      "learning_rate": 2.9243607216331013e-05,
+      "loss": 0.1866,
+      "step": 13090
+    },
+    {
+      "grad_norm": 0.7535688281059265,
+      "learning_rate": 2.916842233441661e-05,
+      "loss": 0.1886,
+      "step": 13100
+    },
+    {
+      "grad_norm": 0.5250940918922424,
+      "learning_rate": 2.90932944052401e-05,
+      "loss": 0.1701,
+      "step": 13110
+    },
+    {
+      "grad_norm": 0.6251844167709351,
+      "learning_rate": 2.9018223634198354e-05,
+      "loss": 0.2057,
+      "step": 13120
+    },
+    {
+      "grad_norm": 0.5567381978034973,
+      "learning_rate": 2.8943210226532025e-05,
+      "loss": 0.1915,
+      "step": 13130
+    },
+    {
+      "grad_norm": 0.6369378566741943,
+      "learning_rate": 2.8868254387324857e-05,
+      "loss": 0.18,
+      "step": 13140
+    },
+    {
+      "grad_norm": 0.557335615158081,
+      "learning_rate": 2.8793356321503306e-05,
+      "loss": 0.176,
+      "step": 13150
+    },
+    {
+      "grad_norm": 0.48559805750846863,
+      "learning_rate": 2.87185162338358e-05,
+      "loss": 0.1813,
+      "step": 13160
+    },
+    {
+      "grad_norm": 0.8626813292503357,
+      "learning_rate": 2.8643734328932253e-05,
+      "loss": 0.186,
+      "step": 13170
+    },
+    {
+      "grad_norm": 0.6032774448394775,
+      "learning_rate": 2.856901081124359e-05,
+      "loss": 0.1829,
+      "step": 13180
+    },
+    {
+      "grad_norm": 0.6123090982437134,
+      "learning_rate": 2.8494345885061002e-05,
+      "loss": 0.1833,
+      "step": 13190
+    },
+    {
+      "grad_norm": 0.6044758558273315,
+      "learning_rate": 2.8419739754515616e-05,
+      "loss": 0.1859,
+      "step": 13200
+    },
+    {
+      "grad_norm": 0.6611677408218384,
+      "learning_rate": 2.8345192623577666e-05,
+      "loss": 0.1842,
+      "step": 13210
+    },
+    {
+      "grad_norm": 0.5389902591705322,
+      "learning_rate": 2.8270704696056193e-05,
+      "loss": 0.1928,
+      "step": 13220
+    },
+    {
+      "grad_norm": 0.5804538726806641,
+      "learning_rate": 2.8196276175598367e-05,
+      "loss": 0.1809,
+      "step": 13230
+    },
+    {
+      "grad_norm": 0.6737906336784363,
+      "learning_rate": 2.8121907265688884e-05,
+      "loss": 0.1835,
+      "step": 13240
+    },
+    {
+      "grad_norm": 0.8530315160751343,
+      "learning_rate": 2.804759816964957e-05,
+      "loss": 0.2078,
+      "step": 13250
+    },
+    {
+      "grad_norm": 0.4701538681983948,
+      "learning_rate": 2.797334909063857e-05,
+      "loss": 0.1907,
+      "step": 13260
+    },
+    {
+      "grad_norm": 0.5288217663764954,
+      "learning_rate": 2.7899160231650056e-05,
+      "loss": 0.1745,
+      "step": 13270
+    },
+    {
+      "grad_norm": 0.6637155413627625,
+      "learning_rate": 2.7825031795513585e-05,
+      "loss": 0.1951,
+      "step": 13280
+    },
+    {
+      "grad_norm": 0.5395070910453796,
+      "learning_rate": 2.775096398489341e-05,
+      "loss": 0.178,
+      "step": 13290
+    },
+    {
+      "grad_norm": 0.7923184037208557,
+      "learning_rate": 2.7676957002288163e-05,
+      "loss": 0.1799,
+      "step": 13300
+    },
+    {
+      "grad_norm": 0.6776189804077148,
+      "learning_rate": 2.760301105003003e-05,
+      "loss": 0.1802,
+      "step": 13310
+    },
+    {
+      "grad_norm": 0.4402746558189392,
+      "learning_rate": 2.752912633028446e-05,
+      "loss": 0.1814,
+      "step": 13320
+    },
+    {
+      "grad_norm": 0.7362489700317383,
+      "learning_rate": 2.7455303045049474e-05,
+      "loss": 0.185,
+      "step": 13330
+    },
+    {
+      "grad_norm": 0.6385077834129333,
+      "learning_rate": 2.7381541396155098e-05,
+      "loss": 0.1862,
+      "step": 13340
+    },
+    {
+      "grad_norm": 0.491832435131073,
+      "learning_rate": 2.730784158526286e-05,
+      "loss": 0.1802,
+      "step": 13350
+    },
+    {
+      "grad_norm": 0.6515607237815857,
+      "learning_rate": 2.723420381386521e-05,
+      "loss": 0.1816,
+      "step": 13360
+    },
+    {
+      "grad_norm": 0.44346171617507935,
+      "learning_rate": 2.7160628283285018e-05,
+      "loss": 0.1862,
+      "step": 13370
+    },
+    {
+      "grad_norm": 0.6440896987915039,
+      "learning_rate": 2.7087115194675007e-05,
+      "loss": 0.186,
+      "step": 13380
+    },
+    {
+      "grad_norm": 0.5582860112190247,
+      "learning_rate": 2.701366474901712e-05,
+      "loss": 0.1878,
+      "step": 13390
+    },
+    {
+      "grad_norm": 0.6565811038017273,
+      "learning_rate": 2.6940277147122085e-05,
+      "loss": 0.1895,
+      "step": 13400
+    },
+    {
+      "grad_norm": 0.5691128373146057,
+      "learning_rate": 2.686695258962878e-05,
+      "loss": 0.2011,
+      "step": 13410
+    },
+    {
+      "grad_norm": 0.6165726184844971,
+      "learning_rate": 2.679369127700375e-05,
+      "loss": 0.2022,
+      "step": 13420
+    },
+    {
+      "grad_norm": 0.4249313473701477,
+      "learning_rate": 2.672049340954067e-05,
+      "loss": 0.182,
+      "step": 13430
+    },
+    {
+      "grad_norm": 0.5584747791290283,
+      "learning_rate": 2.6647359187359676e-05,
+      "loss": 0.2049,
+      "step": 13440
+    },
+    {
+      "grad_norm": 0.641289472579956,
+      "learning_rate": 2.6574288810406946e-05,
+      "loss": 0.1749,
+      "step": 13450
+    },
+    {
+      "grad_norm": 0.8388134837150574,
+      "learning_rate": 2.6501282478454083e-05,
+      "loss": 0.1979,
+      "step": 13460
+    },
+    {
+      "grad_norm": 0.5440933704376221,
+      "learning_rate": 2.6428340391097618e-05,
+      "loss": 0.1763,
+      "step": 13470
+    },
+    {
+      "grad_norm": 0.46771925687789917,
+      "learning_rate": 2.6355462747758485e-05,
+      "loss": 0.1778,
+      "step": 13480
+    },
+    {
+      "grad_norm": 0.6563947796821594,
+      "learning_rate": 2.6282649747681304e-05,
+      "loss": 0.1755,
+      "step": 13490
+    },
+    {
+      "grad_norm": 0.5098826289176941,
+      "learning_rate": 2.620990158993406e-05,
+      "loss": 0.1819,
+      "step": 13500
+    },
+    {
+      "grad_norm": 0.7139420509338379,
+      "learning_rate": 2.6137218473407477e-05,
+      "loss": 0.2023,
+      "step": 13510
+    },
+    {
+      "grad_norm": 0.5638613700866699,
+      "learning_rate": 2.606460059681436e-05,
+      "loss": 0.1907,
+      "step": 13520
+    },
+    {
+      "grad_norm": 0.73574298620224,
+      "learning_rate": 2.599204815868928e-05,
+      "loss": 0.203,
+      "step": 13530
+    },
+    {
+      "grad_norm": 0.5603978037834167,
+      "learning_rate": 2.5919561357387756e-05,
+      "loss": 0.1797,
+      "step": 13540
+    },
+    {
+      "grad_norm": 0.6000884175300598,
+      "learning_rate": 2.5847140391085972e-05,
+      "loss": 0.1707,
+      "step": 13550
+    },
+    {
+      "grad_norm": 0.5105131268501282,
+      "learning_rate": 2.5774785457780103e-05,
+      "loss": 0.1812,
+      "step": 13560
+    },
+    {
+      "grad_norm": 0.5263512134552002,
+      "learning_rate": 2.5702496755285753e-05,
+      "loss": 0.1923,
+      "step": 13570
+    },
+    {
+      "grad_norm": 0.542470395565033,
+      "learning_rate": 2.5630274481237483e-05,
+      "loss": 0.181,
+      "step": 13580
+    },
+    {
+      "grad_norm": 0.5612111687660217,
+      "learning_rate": 2.5558118833088197e-05,
+      "loss": 0.1846,
+      "step": 13590
+    },
+    {
+      "grad_norm": 0.7328010201454163,
+      "learning_rate": 2.548603000810872e-05,
+      "loss": 0.2031,
+      "step": 13600
+    },
+    {
+      "grad_norm": 0.5741540193557739,
+      "learning_rate": 2.5414008203387152e-05,
+      "loss": 0.1794,
+      "step": 13610
+    },
+    {
+      "grad_norm": 0.7677403092384338,
+      "learning_rate": 2.534205361582834e-05,
+      "loss": 0.1825,
+      "step": 13620
+    },
+    {
+      "grad_norm": 0.7265116572380066,
+      "learning_rate": 2.527016644215338e-05,
+      "loss": 0.1925,
+      "step": 13630
+    },
+    {
+      "grad_norm": 0.48603546619415283,
+      "learning_rate": 2.519834687889905e-05,
+      "loss": 0.1735,
+      "step": 13640
+    },
+    {
+      "grad_norm": 0.7419706583023071,
+      "learning_rate": 2.5126595122417295e-05,
+      "loss": 0.1762,
+      "step": 13650
+    },
+    {
+      "grad_norm": 0.8526235818862915,
+      "learning_rate": 2.5054911368874713e-05,
+      "loss": 0.1886,
+      "step": 13660
+    },
+    {
+      "grad_norm": 0.7336394786834717,
+      "learning_rate": 2.4983295814251916e-05,
+      "loss": 0.1752,
+      "step": 13670
+    },
+    {
+      "grad_norm": 0.4712144136428833,
+      "learning_rate": 2.4911748654343105e-05,
+      "loss": 0.1834,
+      "step": 13680
+    },
+    {
+      "grad_norm": 0.5053937435150146,
+      "learning_rate": 2.4840270084755463e-05,
+      "loss": 0.2051,
+      "step": 13690
+    },
+    {
+      "grad_norm": 0.6134051084518433,
+      "learning_rate": 2.4768860300908685e-05,
+      "loss": 0.166,
+      "step": 13700
+    },
+    {
+      "grad_norm": 0.6703867316246033,
+      "learning_rate": 2.469751949803443e-05,
+      "loss": 0.1843,
+      "step": 13710
+    },
+    {
+      "grad_norm": 0.5169312357902527,
+      "learning_rate": 2.4626247871175666e-05,
+      "loss": 0.1859,
+      "step": 13720
+    },
+    {
+      "grad_norm": 0.5225235819816589,
+      "learning_rate": 2.4555045615186346e-05,
+      "loss": 0.1685,
+      "step": 13730
+    },
+    {
+      "grad_norm": 0.700053334236145,
+      "learning_rate": 2.4483912924730677e-05,
+      "loss": 0.1821,
+      "step": 13740
+    },
+    {
+      "grad_norm": 0.45239967107772827,
+      "learning_rate": 2.4412849994282742e-05,
+      "loss": 0.1681,
+      "step": 13750
+    },
+    {
+      "grad_norm": 0.5646662712097168,
+      "learning_rate": 2.434185701812592e-05,
+      "loss": 0.1844,
+      "step": 13760
+    },
+    {
+      "grad_norm": 0.6244551539421082,
+      "learning_rate": 2.4270934190352218e-05,
+      "loss": 0.1937,
+      "step": 13770
+    },
+    {
+      "grad_norm": 0.5019373297691345,
+      "learning_rate": 2.4200081704861998e-05,
+      "loss": 0.188,
+      "step": 13780
+    },
+    {
+      "grad_norm": 0.8081969618797302,
+      "learning_rate": 2.412929975536321e-05,
+      "loss": 0.1858,
+      "step": 13790
+    },
+    {
+      "grad_norm": 0.6091670989990234,
+      "learning_rate": 2.4058588535371017e-05,
+      "loss": 0.1746,
+      "step": 13800
+    },
+    {
+      "grad_norm": 0.6191215515136719,
+      "learning_rate": 2.3987948238207243e-05,
+      "loss": 0.1853,
+      "step": 13810
+    },
+    {
+      "grad_norm": 0.5293408632278442,
+      "learning_rate": 2.3917379056999678e-05,
+      "loss": 0.1755,
+      "step": 13820
+    },
+    {
+      "grad_norm": 0.6145422458648682,
+      "learning_rate": 2.3846881184681824e-05,
+      "loss": 0.1844,
+      "step": 13830
+    },
+    {
+      "grad_norm": 0.5462968349456787,
+      "learning_rate": 2.377645481399214e-05,
+      "loss": 0.1864,
+      "step": 13840
+    },
+    {
+      "grad_norm": 0.5131819844245911,
+      "learning_rate": 2.3706100137473667e-05,
+      "loss": 0.1874,
+      "step": 13850
+    },
+    {
+      "grad_norm": 0.5527155995368958,
+      "learning_rate": 2.3635817347473394e-05,
+      "loss": 0.1773,
+      "step": 13860
+    },
+    {
+      "grad_norm": 0.7643164992332458,
+      "learning_rate": 2.3565606636141757e-05,
+      "loss": 0.1781,
+      "step": 13870
+    },
+    {
+      "grad_norm": 0.4871480464935303,
+      "learning_rate": 2.3495468195432203e-05,
+      "loss": 0.1603,
+      "step": 13880
+    },
+    {
+      "grad_norm": 0.5684931874275208,
+      "learning_rate": 2.3425402217100507e-05,
+      "loss": 0.1672,
+      "step": 13890
+    },
+    {
+      "grad_norm": 0.7298163175582886,
+      "learning_rate": 2.3355408892704424e-05,
+      "loss": 0.1787,
+      "step": 13900
+    },
+    {
+      "grad_norm": 0.7406446933746338,
+      "learning_rate": 2.3285488413603003e-05,
+      "loss": 0.1774,
+      "step": 13910
+    },
+    {
+      "grad_norm": 0.9271367788314819,
+      "learning_rate": 2.321564097095615e-05,
+      "loss": 0.1873,
+      "step": 13920
+    },
+    {
+      "grad_norm": 0.42193230986595154,
+      "learning_rate": 2.3145866755724142e-05,
+      "loss": 0.1818,
+      "step": 13930
+    },
+    {
+      "grad_norm": 0.7870957851409912,
+      "learning_rate": 2.307616595866699e-05,
+      "loss": 0.1818,
+      "step": 13940
+    },
+    {
+      "grad_norm": 0.5509008765220642,
+      "learning_rate": 2.3006538770344032e-05,
+      "loss": 0.1599,
+      "step": 13950
+    },
+    {
+      "grad_norm": 0.6330592632293701,
+      "learning_rate": 2.293698538111334e-05,
+      "loss": 0.1809,
+      "step": 13960
+    },
+    {
+      "grad_norm": 0.4337199330329895,
+      "learning_rate": 2.28675059811312e-05,
+      "loss": 0.1981,
+      "step": 13970
+    },
+    {
+      "grad_norm": 0.6920579075813293,
+      "learning_rate": 2.279810076035167e-05,
+      "loss": 0.1906,
+      "step": 13980
+    },
+    {
+      "grad_norm": 0.5556825399398804,
+      "learning_rate": 2.272876990852596e-05,
+      "loss": 0.1803,
+      "step": 13990
+    },
+    {
+      "grad_norm": 0.4891877770423889,
+      "learning_rate": 2.265951361520195e-05,
+      "loss": 0.1841,
+      "step": 14000
+    },
+    {
+      "grad_norm": 0.5031809210777283,
+      "learning_rate": 2.2590332069723748e-05,
+      "loss": 0.1753,
+      "step": 14010
+    },
+    {
+      "grad_norm": 0.5527625679969788,
+      "learning_rate": 2.2521225461231004e-05,
+      "loss": 0.1756,
+      "step": 14020
+    },
+    {
+      "grad_norm": 0.6153786182403564,
+      "learning_rate": 2.2452193978658597e-05,
+      "loss": 0.186,
+      "step": 14030
+    },
+    {
+      "grad_norm": 0.8035627603530884,
+      "learning_rate": 2.238323781073594e-05,
+      "loss": 0.1603,
+      "step": 14040
+    },
+    {
+      "grad_norm": 0.8184167742729187,
+      "learning_rate": 2.2314357145986552e-05,
+      "loss": 0.2026,
+      "step": 14050
+    },
+    {
+      "grad_norm": 0.7528314590454102,
+      "learning_rate": 2.224555217272757e-05,
+      "loss": 0.1837,
+      "step": 14060
+    },
+    {
+      "grad_norm": 0.45660513639450073,
+      "learning_rate": 2.2176823079069127e-05,
+      "loss": 0.1638,
+      "step": 14070
+    },
+    {
+      "grad_norm": 0.7419241666793823,
+      "learning_rate": 2.210817005291398e-05,
+      "loss": 0.1798,
+      "step": 14080
+    },
+    {
+      "grad_norm": 0.8710065484046936,
+      "learning_rate": 2.203959328195686e-05,
+      "loss": 0.1835,
+      "step": 14090
+    },
+    {
+      "grad_norm": 0.5627427697181702,
+      "learning_rate": 2.1971092953684026e-05,
+      "loss": 0.1773,
+      "step": 14100
+    },
+    {
+      "grad_norm": 0.6897695660591125,
+      "learning_rate": 2.1902669255372788e-05,
+      "loss": 0.1737,
+      "step": 14110
+    },
+    {
+      "grad_norm": 0.676856517791748,
+      "learning_rate": 2.1834322374090897e-05,
+      "loss": 0.18,
+      "step": 14120
+    },
+    {
+      "grad_norm": 0.6322209239006042,
+      "learning_rate": 2.1766052496696153e-05,
+      "loss": 0.2059,
+      "step": 14130
+    },
+    {
+      "grad_norm": 0.61594557762146,
+      "learning_rate": 2.169785980983577e-05,
+      "loss": 0.1853,
+      "step": 14140
+    },
+    {
+      "grad_norm": 0.5965731143951416,
+      "learning_rate": 2.162974449994593e-05,
+      "loss": 0.1721,
+      "step": 14150
+    },
+    {
+      "grad_norm": 0.6333994269371033,
+      "learning_rate": 2.1561706753251337e-05,
+      "loss": 0.1879,
+      "step": 14160
+    },
+    {
+      "grad_norm": 0.598121702671051,
+      "learning_rate": 2.1493746755764544e-05,
+      "loss": 0.1784,
+      "step": 14170
+    },
+    {
+      "grad_norm": 0.9111278057098389,
+      "learning_rate": 2.1425864693285635e-05,
+      "loss": 0.1776,
+      "step": 14180
+    },
+    {
+      "grad_norm": 0.6412689685821533,
+      "learning_rate": 2.1358060751401547e-05,
+      "loss": 0.176,
+      "step": 14190
+    },
+    {
+      "grad_norm": 0.7198978066444397,
+      "learning_rate": 2.129033511548566e-05,
+      "loss": 0.1839,
+      "step": 14200
+    },
+    {
+      "grad_norm": 0.7167001366615295,
+      "learning_rate": 2.1222687970697315e-05,
+      "loss": 0.1989,
+      "step": 14210
+    },
+    {
+      "grad_norm": 0.5145686864852905,
+      "learning_rate": 2.1155119501981173e-05,
+      "loss": 0.186,
+      "step": 14220
+    },
+    {
+      "grad_norm": 0.5235252380371094,
+      "learning_rate": 2.1087629894066895e-05,
+      "loss": 0.1816,
+      "step": 14230
+    },
+    {
+      "grad_norm": 0.8080321550369263,
+      "learning_rate": 2.1020219331468473e-05,
+      "loss": 0.1855,
+      "step": 14240
+    },
+    {
+      "grad_norm": 0.5882876515388489,
+      "learning_rate": 2.095288799848379e-05,
+      "loss": 0.1707,
+      "step": 14250
+    },
+    {
+      "grad_norm": 0.9200838208198547,
+      "learning_rate": 2.088563607919417e-05,
+      "loss": 0.1881,
+      "step": 14260
+    },
+    {
+      "grad_norm": 0.5685015320777893,
+      "learning_rate": 2.0818463757463786e-05,
+      "loss": 0.1777,
+      "step": 14270
+    },
+    {
+      "grad_norm": 0.6687608361244202,
+      "learning_rate": 2.0751371216939175e-05,
+      "loss": 0.1796,
+      "step": 14280
+    },
+    {
+      "grad_norm": 0.7200865745544434,
+      "learning_rate": 2.068435864104882e-05,
+      "loss": 0.169,
+      "step": 14290
+    },
+    {
+      "grad_norm": 0.9801043272018433,
+      "learning_rate": 2.0617426213002506e-05,
+      "loss": 0.1711,
+      "step": 14300
+    },
+    {
+      "grad_norm": 0.5257559418678284,
+      "learning_rate": 2.055057411579097e-05,
+      "loss": 0.1862,
+      "step": 14310
+    },
+    {
+      "grad_norm": 0.5687223076820374,
+      "learning_rate": 2.0483802532185286e-05,
+      "loss": 0.1772,
+      "step": 14320
+    },
+    {
+      "grad_norm": 0.614574134349823,
+      "learning_rate": 2.041711164473638e-05,
+      "loss": 0.201,
+      "step": 14330
+    },
+    {
+      "grad_norm": 0.91933673620224,
+      "learning_rate": 2.0350501635774637e-05,
+      "loss": 0.1871,
+      "step": 14340
+    },
+    {
+      "grad_norm": 0.5663885474205017,
+      "learning_rate": 2.0283972687409247e-05,
+      "loss": 0.171,
+      "step": 14350
+    },
+    {
+      "grad_norm": 0.8177073001861572,
+      "learning_rate": 2.021752498152784e-05,
+      "loss": 0.1855,
+      "step": 14360
+    },
+    {
+      "grad_norm": 0.6501665711402893,
+      "learning_rate": 2.015115869979589e-05,
+      "loss": 0.1749,
+      "step": 14370
+    },
+    {
+      "grad_norm": 0.5722736716270447,
+      "learning_rate": 2.0084874023656265e-05,
+      "loss": 0.1817,
+      "step": 14380
+    },
+    {
+      "grad_norm": 0.9081240892410278,
+      "learning_rate": 2.001867113432877e-05,
+      "loss": 0.1943,
+      "step": 14390
+    },
+    {
+      "grad_norm": 0.6183856129646301,
+      "learning_rate": 1.995255021280954e-05,
+      "loss": 0.1709,
+      "step": 14400
+    },
+    {
+      "grad_norm": 0.6205171942710876,
+      "learning_rate": 1.9886511439870688e-05,
+      "loss": 0.188,
+      "step": 14410
+    },
+    {
+      "grad_norm": 0.56046062707901,
+      "learning_rate": 1.9820554996059675e-05,
+      "loss": 0.177,
+      "step": 14420
+    },
+    {
+      "grad_norm": 0.46052810549736023,
+      "learning_rate": 1.9754681061698893e-05,
+      "loss": 0.1769,
+      "step": 14430
+    },
+    {
+      "grad_norm": 0.6785239577293396,
+      "learning_rate": 1.9688889816885185e-05,
+      "loss": 0.1759,
+      "step": 14440
+    },
+    {
+      "grad_norm": 0.9971427917480469,
+      "learning_rate": 1.962318144148928e-05,
+      "loss": 0.1994,
+      "step": 14450
+    },
+    {
+      "grad_norm": 0.6560341715812683,
+      "learning_rate": 1.955755611515539e-05,
+      "loss": 0.172,
+      "step": 14460
+    },
+    {
+      "grad_norm": 0.49518224596977234,
+      "learning_rate": 1.9492014017300642e-05,
+      "loss": 0.1996,
+      "step": 14470
+    },
+    {
+      "grad_norm": 0.6369688510894775,
+      "learning_rate": 1.942655532711461e-05,
+      "loss": 0.1757,
+      "step": 14480
+    },
+    {
+      "grad_norm": 1.2675223350524902,
+      "learning_rate": 1.9361180223558882e-05,
+      "loss": 0.1894,
+      "step": 14490
+    },
+    {
+      "grad_norm": 0.7055156230926514,
+      "learning_rate": 1.929588888536647e-05,
+      "loss": 0.1843,
+      "step": 14500
+    },
+    {
+      "grad_norm": 0.9152706861495972,
+      "learning_rate": 1.9230681491041425e-05,
+      "loss": 0.2015,
+      "step": 14510
+    },
+    {
+      "grad_norm": 0.48137137293815613,
+      "learning_rate": 1.9165558218858264e-05,
+      "loss": 0.189,
+      "step": 14520
+    },
+    {
+      "grad_norm": 0.6425464749336243,
+      "learning_rate": 1.9100519246861505e-05,
+      "loss": 0.1737,
+      "step": 14530
+    },
+    {
+      "grad_norm": 0.5238012671470642,
+      "learning_rate": 1.9035564752865248e-05,
+      "loss": 0.184,
+      "step": 14540
+    },
+    {
+      "grad_norm": 0.5861884355545044,
+      "learning_rate": 1.897069491445258e-05,
+      "loss": 0.1713,
+      "step": 14550
+    },
+    {
+      "grad_norm": 0.618438184261322,
+      "learning_rate": 1.890590990897515e-05,
+      "loss": 0.1742,
+      "step": 14560
+    },
+    {
+      "grad_norm": 0.8278201222419739,
+      "learning_rate": 1.884120991355272e-05,
+      "loss": 0.188,
+      "step": 14570
+    },
+    {
+      "grad_norm": 0.6770163774490356,
+      "learning_rate": 1.8776595105072576e-05,
+      "loss": 0.1862,
+      "step": 14580
+    },
+    {
+      "grad_norm": 0.718325138092041,
+      "learning_rate": 1.8712065660189166e-05,
+      "loss": 0.1795,
+      "step": 14590
+    },
+    {
+      "grad_norm": 1.1929357051849365,
+      "learning_rate": 1.8647621755323513e-05,
+      "loss": 0.181,
+      "step": 14600
+    },
+    {
+      "grad_norm": 0.6007441878318787,
+      "learning_rate": 1.858326356666278e-05,
+      "loss": 0.1721,
+      "step": 14610
+    },
+    {
+      "grad_norm": 0.63893061876297,
+      "learning_rate": 1.851899127015983e-05,
+      "loss": 0.1878,
+      "step": 14620
+    },
+    {
+      "grad_norm": 0.6796618103981018,
+      "learning_rate": 1.8454805041532626e-05,
+      "loss": 0.2001,
+      "step": 14630
+    },
+    {
+      "grad_norm": 0.4488603174686432,
+      "learning_rate": 1.8390705056263906e-05,
+      "loss": 0.1766,
+      "step": 14640
+    },
+    {
+      "grad_norm": 0.49013105034828186,
+      "learning_rate": 1.832669148960057e-05,
+      "loss": 0.1745,
+      "step": 14650
+    },
+    {
+      "grad_norm": 0.6219348907470703,
+      "learning_rate": 1.8262764516553233e-05,
+      "loss": 0.1846,
+      "step": 14660
+    },
+    {
+      "grad_norm": 0.5187748074531555,
+      "learning_rate": 1.8198924311895843e-05,
+      "loss": 0.2036,
+      "step": 14670
+    },
+    {
+      "grad_norm": 0.629365086555481,
+      "learning_rate": 1.813517105016505e-05,
+      "loss": 0.1893,
+      "step": 14680
+    },
+    {
+      "grad_norm": 0.6857553720474243,
+      "learning_rate": 1.8071504905659888e-05,
+      "loss": 0.1849,
+      "step": 14690
+    },
+    {
+      "grad_norm": 0.62723308801651,
+      "learning_rate": 1.800792605244109e-05,
+      "loss": 0.1709,
+      "step": 14700
+    },
+    {
+      "grad_norm": 0.6837912201881409,
+      "learning_rate": 1.7944434664330844e-05,
+      "loss": 0.1813,
+      "step": 14710
+    },
+    {
+      "grad_norm": 0.7823312878608704,
+      "learning_rate": 1.7881030914912212e-05,
+      "loss": 0.193,
+      "step": 14720
+    },
+    {
+      "grad_norm": 0.6905558705329895,
+      "learning_rate": 1.7817714977528577e-05,
+      "loss": 0.1753,
+      "step": 14730
+    },
+    {
+      "grad_norm": 0.7947484254837036,
+      "learning_rate": 1.7754487025283332e-05,
+      "loss": 0.187,
+      "step": 14740
+    },
+    {
+      "grad_norm": 0.6798580288887024,
+      "learning_rate": 1.7691347231039275e-05,
+      "loss": 0.1755,
+      "step": 14750
+    },
+    {
+      "grad_norm": 0.927318811416626,
+      "learning_rate": 1.7628295767418164e-05,
+      "loss": 0.1936,
+      "step": 14760
+    },
+    {
+      "grad_norm": 0.4793388247489929,
+      "learning_rate": 1.7565332806800333e-05,
+      "loss": 0.1845,
+      "step": 14770
+    },
+    {
+      "grad_norm": 0.7225732803344727,
+      "learning_rate": 1.750245852132408e-05,
+      "loss": 0.1989,
+      "step": 14780
+    },
+    {
+      "grad_norm": 0.5416461825370789,
+      "learning_rate": 1.7439673082885323e-05,
+      "loss": 0.1739,
+      "step": 14790
+    },
+    {
+      "grad_norm": 0.6384958624839783,
+      "learning_rate": 1.7376976663137047e-05,
+      "loss": 0.1883,
+      "step": 14800
+    },
+    {
+      "grad_norm": 0.5407596826553345,
+      "learning_rate": 1.7314369433488853e-05,
+      "loss": 0.1862,
+      "step": 14810
+    },
+    {
+      "grad_norm": 0.7871677279472351,
+      "learning_rate": 1.7251851565106548e-05,
+      "loss": 0.1737,
+      "step": 14820
+    },
+    {
+      "grad_norm": 0.482360303401947,
+      "learning_rate": 1.7189423228911574e-05,
+      "loss": 0.183,
+      "step": 14830
+    },
+    {
+      "grad_norm": 0.770272970199585,
+      "learning_rate": 1.7127084595580606e-05,
+      "loss": 0.181,
+      "step": 14840
+    },
+    {
+      "grad_norm": 0.563123345375061,
+      "learning_rate": 1.706483583554513e-05,
+      "loss": 0.1773,
+      "step": 14850
+    },
+    {
+      "grad_norm": 0.6416179537773132,
+      "learning_rate": 1.700267711899083e-05,
+      "loss": 0.1863,
+      "step": 14860
+    },
+    {
+      "grad_norm": 0.7249609231948853,
+      "learning_rate": 1.69406086158573e-05,
+      "loss": 0.1983,
+      "step": 14870
+    },
+    {
+      "grad_norm": 0.7140301465988159,
+      "learning_rate": 1.6878630495837455e-05,
+      "loss": 0.1714,
+      "step": 14880
+    },
+    {
+      "grad_norm": 0.46881842613220215,
+      "learning_rate": 1.681674292837707e-05,
+      "loss": 0.1789,
+      "step": 14890
+    },
+    {
+      "grad_norm": 0.544331967830658,
+      "learning_rate": 1.6754946082674444e-05,
+      "loss": 0.172,
+      "step": 14900
+    },
+    {
+      "grad_norm": 1.763749361038208,
+      "learning_rate": 1.6693240127679748e-05,
+      "loss": 0.1937,
+      "step": 14910
+    },
+    {
+      "grad_norm": 0.6070051193237305,
+      "learning_rate": 1.663162523209475e-05,
+      "loss": 0.183,
+      "step": 14920
+    },
+    {
+      "grad_norm": 0.6405426263809204,
+      "learning_rate": 1.6570101564372193e-05,
+      "loss": 0.1872,
+      "step": 14930
+    },
+    {
+      "grad_norm": 0.49250295758247375,
+      "learning_rate": 1.650866929271543e-05,
+      "loss": 0.1876,
+      "step": 14940
+    },
+    {
+      "grad_norm": 0.6298572421073914,
+      "learning_rate": 1.644732858507797e-05,
+      "loss": 0.1816,
+      "step": 14950
+    },
+    {
+      "grad_norm": 0.8196234107017517,
+      "learning_rate": 1.6386079609162943e-05,
+      "loss": 0.2043,
+      "step": 14960
+    },
+    {
+      "grad_norm": 0.4392532706260681,
+      "learning_rate": 1.6324922532422742e-05,
+      "loss": 0.1927,
+      "step": 14970
+    },
+    {
+      "grad_norm": 0.679341733455658,
+      "learning_rate": 1.6263857522058434e-05,
+      "loss": 0.1792,
+      "step": 14980
+    },
+    {
+      "grad_norm": 0.4882771372795105,
+      "learning_rate": 1.6202884745019443e-05,
+      "loss": 0.1933,
+      "step": 14990
+    },
+    {
+      "grad_norm": 0.592680811882019,
+      "learning_rate": 1.614200436800304e-05,
+      "loss": 0.1734,
+      "step": 15000
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 20000,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 9223372036854775807,
+  "save_steps": 1000,
+  "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": 16,
+  "trial_name": null,
+  "trial_params": null
+}