diff --git "a/vlmfreeze_loc_h20_10k/trainer_state.json" "b/vlmfreeze_loc_h20_10k/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/vlmfreeze_loc_h20_10k/trainer_state.json"
@@ -0,0 +1,6034 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.5,
+  "eval_steps": 500,
+  "global_step": 10000,
+  "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
+    }
+  ],
+  "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
+}