diff --git "a/vlmfreeze_loc_h15_15k/trainer_state.json" "b/vlmfreeze_loc_h15_15k/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/vlmfreeze_loc_h15_15k/trainer_state.json"
@@ -0,0 +1,9034 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.75,
+  "eval_steps": 500,
+  "global_step": 15000,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "grad_norm": 0.3745940327644348,
+      "learning_rate": 9e-07,
+      "loss": 0.3345,
+      "step": 10
+    },
+    {
+      "grad_norm": 0.43203699588775635,
+      "learning_rate": 1.9e-06,
+      "loss": 0.3569,
+      "step": 20
+    },
+    {
+      "grad_norm": 0.41796186566352844,
+      "learning_rate": 2.9e-06,
+      "loss": 0.3457,
+      "step": 30
+    },
+    {
+      "grad_norm": 0.33292850852012634,
+      "learning_rate": 3.9e-06,
+      "loss": 0.3314,
+      "step": 40
+    },
+    {
+      "grad_norm": 0.38360536098480225,
+      "learning_rate": 4.9000000000000005e-06,
+      "loss": 0.3359,
+      "step": 50
+    },
+    {
+      "grad_norm": 0.37815409898757935,
+      "learning_rate": 5.9e-06,
+      "loss": 0.3204,
+      "step": 60
+    },
+    {
+      "grad_norm": 0.32865551114082336,
+      "learning_rate": 6.900000000000001e-06,
+      "loss": 0.3141,
+      "step": 70
+    },
+    {
+      "grad_norm": 0.3118378520011902,
+      "learning_rate": 7.9e-06,
+      "loss": 0.3338,
+      "step": 80
+    },
+    {
+      "grad_norm": 0.2776610255241394,
+      "learning_rate": 8.9e-06,
+      "loss": 0.302,
+      "step": 90
+    },
+    {
+      "grad_norm": 0.35227689146995544,
+      "learning_rate": 9.900000000000002e-06,
+      "loss": 0.2941,
+      "step": 100
+    },
+    {
+      "grad_norm": 0.4729832410812378,
+      "learning_rate": 1.09e-05,
+      "loss": 0.3014,
+      "step": 110
+    },
+    {
+      "grad_norm": 0.38645246624946594,
+      "learning_rate": 1.19e-05,
+      "loss": 0.2861,
+      "step": 120
+    },
+    {
+      "grad_norm": 0.29405367374420166,
+      "learning_rate": 1.29e-05,
+      "loss": 0.2884,
+      "step": 130
+    },
+    {
+      "grad_norm": 0.29601994156837463,
+      "learning_rate": 1.3900000000000002e-05,
+      "loss": 0.3018,
+      "step": 140
+    },
+    {
+      "grad_norm": 0.34279611706733704,
+      "learning_rate": 1.49e-05,
+      "loss": 0.2787,
+      "step": 150
+    },
+    {
+      "grad_norm": 0.26968783140182495,
+      "learning_rate": 1.59e-05,
+      "loss": 0.266,
+      "step": 160
+    },
+    {
+      "grad_norm": 0.2368963062763214,
+      "learning_rate": 1.69e-05,
+      "loss": 0.2564,
+      "step": 170
+    },
+    {
+      "grad_norm": 0.3810964822769165,
+      "learning_rate": 1.79e-05,
+      "loss": 0.2564,
+      "step": 180
+    },
+    {
+      "grad_norm": 0.3497999906539917,
+      "learning_rate": 1.8900000000000002e-05,
+      "loss": 0.2537,
+      "step": 190
+    },
+    {
+      "grad_norm": 0.2594382166862488,
+      "learning_rate": 1.9900000000000003e-05,
+      "loss": 0.2303,
+      "step": 200
+    },
+    {
+      "grad_norm": 0.31402787566185,
+      "learning_rate": 2.09e-05,
+      "loss": 0.2359,
+      "step": 210
+    },
+    {
+      "grad_norm": 0.37911003828048706,
+      "learning_rate": 2.19e-05,
+      "loss": 0.2485,
+      "step": 220
+    },
+    {
+      "grad_norm": 0.5291225910186768,
+      "learning_rate": 2.29e-05,
+      "loss": 0.2525,
+      "step": 230
+    },
+    {
+      "grad_norm": 0.31603312492370605,
+      "learning_rate": 2.39e-05,
+      "loss": 0.2371,
+      "step": 240
+    },
+    {
+      "grad_norm": 0.3666847050189972,
+      "learning_rate": 2.4900000000000002e-05,
+      "loss": 0.2331,
+      "step": 250
+    },
+    {
+      "grad_norm": 0.44824081659317017,
+      "learning_rate": 2.5900000000000003e-05,
+      "loss": 0.241,
+      "step": 260
+    },
+    {
+      "grad_norm": 0.3277708888053894,
+      "learning_rate": 2.6900000000000003e-05,
+      "loss": 0.2275,
+      "step": 270
+    },
+    {
+      "grad_norm": 0.5131637454032898,
+      "learning_rate": 2.7900000000000004e-05,
+      "loss": 0.2322,
+      "step": 280
+    },
+    {
+      "grad_norm": 0.31480810046195984,
+      "learning_rate": 2.8899999999999998e-05,
+      "loss": 0.2262,
+      "step": 290
+    },
+    {
+      "grad_norm": 0.3878740966320038,
+      "learning_rate": 2.9900000000000002e-05,
+      "loss": 0.2201,
+      "step": 300
+    },
+    {
+      "grad_norm": 0.3417105972766876,
+      "learning_rate": 3.09e-05,
+      "loss": 0.2108,
+      "step": 310
+    },
+    {
+      "grad_norm": 0.5033106207847595,
+      "learning_rate": 3.19e-05,
+      "loss": 0.231,
+      "step": 320
+    },
+    {
+      "grad_norm": 0.4744783043861389,
+      "learning_rate": 3.29e-05,
+      "loss": 0.2188,
+      "step": 330
+    },
+    {
+      "grad_norm": 0.274917334318161,
+      "learning_rate": 3.3900000000000004e-05,
+      "loss": 0.2143,
+      "step": 340
+    },
+    {
+      "grad_norm": 0.41807204484939575,
+      "learning_rate": 3.49e-05,
+      "loss": 0.2116,
+      "step": 350
+    },
+    {
+      "grad_norm": 0.40757644176483154,
+      "learning_rate": 3.59e-05,
+      "loss": 0.2017,
+      "step": 360
+    },
+    {
+      "grad_norm": 0.4659121632575989,
+      "learning_rate": 3.69e-05,
+      "loss": 0.2121,
+      "step": 370
+    },
+    {
+      "grad_norm": 0.3957243263721466,
+      "learning_rate": 3.79e-05,
+      "loss": 0.2189,
+      "step": 380
+    },
+    {
+      "grad_norm": 0.44388189911842346,
+      "learning_rate": 3.8900000000000004e-05,
+      "loss": 0.2026,
+      "step": 390
+    },
+    {
+      "grad_norm": 0.7040401101112366,
+      "learning_rate": 3.99e-05,
+      "loss": 0.2184,
+      "step": 400
+    },
+    {
+      "grad_norm": 0.3569212555885315,
+      "learning_rate": 4.09e-05,
+      "loss": 0.1977,
+      "step": 410
+    },
+    {
+      "grad_norm": 0.5557554960250854,
+      "learning_rate": 4.19e-05,
+      "loss": 0.23,
+      "step": 420
+    },
+    {
+      "grad_norm": 0.35263705253601074,
+      "learning_rate": 4.29e-05,
+      "loss": 0.2094,
+      "step": 430
+    },
+    {
+      "grad_norm": 0.5555665493011475,
+      "learning_rate": 4.39e-05,
+      "loss": 0.1925,
+      "step": 440
+    },
+    {
+      "grad_norm": 0.47788968682289124,
+      "learning_rate": 4.49e-05,
+      "loss": 0.218,
+      "step": 450
+    },
+    {
+      "grad_norm": 0.5905026793479919,
+      "learning_rate": 4.5900000000000004e-05,
+      "loss": 0.1997,
+      "step": 460
+    },
+    {
+      "grad_norm": 0.4385456442832947,
+      "learning_rate": 4.69e-05,
+      "loss": 0.1921,
+      "step": 470
+    },
+    {
+      "grad_norm": 0.5633069276809692,
+      "learning_rate": 4.79e-05,
+      "loss": 0.1935,
+      "step": 480
+    },
+    {
+      "grad_norm": 0.4809187054634094,
+      "learning_rate": 4.89e-05,
+      "loss": 0.2003,
+      "step": 490
+    },
+    {
+      "grad_norm": 0.47468334436416626,
+      "learning_rate": 4.99e-05,
+      "loss": 0.1873,
+      "step": 500
+    },
+    {
+      "grad_norm": 0.5676734447479248,
+      "learning_rate": 5.0900000000000004e-05,
+      "loss": 0.2106,
+      "step": 510
+    },
+    {
+      "grad_norm": 0.4051642119884491,
+      "learning_rate": 5.19e-05,
+      "loss": 0.2015,
+      "step": 520
+    },
+    {
+      "grad_norm": 0.504147469997406,
+      "learning_rate": 5.2900000000000005e-05,
+      "loss": 0.1983,
+      "step": 530
+    },
+    {
+      "grad_norm": 0.5737192034721375,
+      "learning_rate": 5.390000000000001e-05,
+      "loss": 0.1873,
+      "step": 540
+    },
+    {
+      "grad_norm": 0.34671539068222046,
+      "learning_rate": 5.4900000000000006e-05,
+      "loss": 0.1917,
+      "step": 550
+    },
+    {
+      "grad_norm": 0.5830896496772766,
+      "learning_rate": 5.590000000000001e-05,
+      "loss": 0.1975,
+      "step": 560
+    },
+    {
+      "grad_norm": 0.4211214482784271,
+      "learning_rate": 5.69e-05,
+      "loss": 0.1886,
+      "step": 570
+    },
+    {
+      "grad_norm": 0.5540159344673157,
+      "learning_rate": 5.79e-05,
+      "loss": 0.2035,
+      "step": 580
+    },
+    {
+      "grad_norm": 0.35927361249923706,
+      "learning_rate": 5.89e-05,
+      "loss": 0.2008,
+      "step": 590
+    },
+    {
+      "grad_norm": 0.5210414528846741,
+      "learning_rate": 5.99e-05,
+      "loss": 0.1949,
+      "step": 600
+    },
+    {
+      "grad_norm": 0.4096316695213318,
+      "learning_rate": 6.09e-05,
+      "loss": 0.2003,
+      "step": 610
+    },
+    {
+      "grad_norm": 0.412835955619812,
+      "learning_rate": 6.19e-05,
+      "loss": 0.2034,
+      "step": 620
+    },
+    {
+      "grad_norm": 0.5024417042732239,
+      "learning_rate": 6.29e-05,
+      "loss": 0.1928,
+      "step": 630
+    },
+    {
+      "grad_norm": 0.41292136907577515,
+      "learning_rate": 6.390000000000001e-05,
+      "loss": 0.19,
+      "step": 640
+    },
+    {
+      "grad_norm": 0.5432610511779785,
+      "learning_rate": 6.49e-05,
+      "loss": 0.1779,
+      "step": 650
+    },
+    {
+      "grad_norm": 0.5518460273742676,
+      "learning_rate": 6.59e-05,
+      "loss": 0.1673,
+      "step": 660
+    },
+    {
+      "grad_norm": 0.45131826400756836,
+      "learning_rate": 6.690000000000001e-05,
+      "loss": 0.1944,
+      "step": 670
+    },
+    {
+      "grad_norm": 0.4321049153804779,
+      "learning_rate": 6.790000000000001e-05,
+      "loss": 0.1866,
+      "step": 680
+    },
+    {
+      "grad_norm": 0.3838651478290558,
+      "learning_rate": 6.89e-05,
+      "loss": 0.1953,
+      "step": 690
+    },
+    {
+      "grad_norm": 0.5376997590065002,
+      "learning_rate": 6.99e-05,
+      "loss": 0.1759,
+      "step": 700
+    },
+    {
+      "grad_norm": 0.5151509046554565,
+      "learning_rate": 7.09e-05,
+      "loss": 0.1802,
+      "step": 710
+    },
+    {
+      "grad_norm": 0.38976597785949707,
+      "learning_rate": 7.19e-05,
+      "loss": 0.1944,
+      "step": 720
+    },
+    {
+      "grad_norm": 0.4345996081829071,
+      "learning_rate": 7.29e-05,
+      "loss": 0.1995,
+      "step": 730
+    },
+    {
+      "grad_norm": 0.4249369502067566,
+      "learning_rate": 7.390000000000001e-05,
+      "loss": 0.1864,
+      "step": 740
+    },
+    {
+      "grad_norm": 0.4485539197921753,
+      "learning_rate": 7.49e-05,
+      "loss": 0.2085,
+      "step": 750
+    },
+    {
+      "grad_norm": 0.5289148688316345,
+      "learning_rate": 7.59e-05,
+      "loss": 0.1886,
+      "step": 760
+    },
+    {
+      "grad_norm": 0.42257630825042725,
+      "learning_rate": 7.69e-05,
+      "loss": 0.1902,
+      "step": 770
+    },
+    {
+      "grad_norm": 0.6824215650558472,
+      "learning_rate": 7.790000000000001e-05,
+      "loss": 0.1855,
+      "step": 780
+    },
+    {
+      "grad_norm": 0.42699098587036133,
+      "learning_rate": 7.890000000000001e-05,
+      "loss": 0.1897,
+      "step": 790
+    },
+    {
+      "grad_norm": 0.3677796721458435,
+      "learning_rate": 7.99e-05,
+      "loss": 0.1766,
+      "step": 800
+    },
+    {
+      "grad_norm": 0.6163636445999146,
+      "learning_rate": 8.090000000000001e-05,
+      "loss": 0.1863,
+      "step": 810
+    },
+    {
+      "grad_norm": 0.5741963386535645,
+      "learning_rate": 8.19e-05,
+      "loss": 0.195,
+      "step": 820
+    },
+    {
+      "grad_norm": 0.5593239068984985,
+      "learning_rate": 8.29e-05,
+      "loss": 0.1835,
+      "step": 830
+    },
+    {
+      "grad_norm": 0.5202434659004211,
+      "learning_rate": 8.39e-05,
+      "loss": 0.1861,
+      "step": 840
+    },
+    {
+      "grad_norm": 0.38479533791542053,
+      "learning_rate": 8.49e-05,
+      "loss": 0.194,
+      "step": 850
+    },
+    {
+      "grad_norm": 0.363511860370636,
+      "learning_rate": 8.59e-05,
+      "loss": 0.1838,
+      "step": 860
+    },
+    {
+      "grad_norm": 0.3553257882595062,
+      "learning_rate": 8.69e-05,
+      "loss": 0.197,
+      "step": 870
+    },
+    {
+      "grad_norm": 0.4694828689098358,
+      "learning_rate": 8.790000000000001e-05,
+      "loss": 0.1922,
+      "step": 880
+    },
+    {
+      "grad_norm": 0.3800056576728821,
+      "learning_rate": 8.89e-05,
+      "loss": 0.1965,
+      "step": 890
+    },
+    {
+      "grad_norm": 0.3647955358028412,
+      "learning_rate": 8.99e-05,
+      "loss": 0.1977,
+      "step": 900
+    },
+    {
+      "grad_norm": 0.5004047155380249,
+      "learning_rate": 9.090000000000001e-05,
+      "loss": 0.1982,
+      "step": 910
+    },
+    {
+      "grad_norm": 0.4323939383029938,
+      "learning_rate": 9.190000000000001e-05,
+      "loss": 0.1788,
+      "step": 920
+    },
+    {
+      "grad_norm": 0.4356240928173065,
+      "learning_rate": 9.290000000000001e-05,
+      "loss": 0.1702,
+      "step": 930
+    },
+    {
+      "grad_norm": 0.3859714865684509,
+      "learning_rate": 9.39e-05,
+      "loss": 0.1587,
+      "step": 940
+    },
+    {
+      "grad_norm": 0.4192478060722351,
+      "learning_rate": 9.49e-05,
+      "loss": 0.2001,
+      "step": 950
+    },
+    {
+      "grad_norm": 0.7999675869941711,
+      "learning_rate": 9.59e-05,
+      "loss": 0.1827,
+      "step": 960
+    },
+    {
+      "grad_norm": 0.471138596534729,
+      "learning_rate": 9.69e-05,
+      "loss": 0.186,
+      "step": 970
+    },
+    {
+      "grad_norm": 0.475413978099823,
+      "learning_rate": 9.790000000000001e-05,
+      "loss": 0.2027,
+      "step": 980
+    },
+    {
+      "grad_norm": 0.7430187463760376,
+      "learning_rate": 9.89e-05,
+      "loss": 0.1899,
+      "step": 990
+    },
+    {
+      "grad_norm": 0.47628703713417053,
+      "learning_rate": 9.99e-05,
+      "loss": 0.1769,
+      "step": 1000
+    },
+    {
+      "grad_norm": 0.4999978840351105,
+      "learning_rate": 9.999994463727085e-05,
+      "loss": 0.1913,
+      "step": 1010
+    },
+    {
+      "grad_norm": 0.499382883310318,
+      "learning_rate": 9.999975326009292e-05,
+      "loss": 0.1918,
+      "step": 1020
+    },
+    {
+      "grad_norm": 0.4502788186073303,
+      "learning_rate": 9.999942518549879e-05,
+      "loss": 0.1842,
+      "step": 1030
+    },
+    {
+      "grad_norm": 0.7677947282791138,
+      "learning_rate": 9.999896041438544e-05,
+      "loss": 0.1874,
+      "step": 1040
+    },
+    {
+      "grad_norm": 0.3715583384037018,
+      "learning_rate": 9.999835894802353e-05,
+      "loss": 0.1987,
+      "step": 1050
+    },
+    {
+      "grad_norm": 0.5732327103614807,
+      "learning_rate": 9.999762078805743e-05,
+      "loss": 0.2126,
+      "step": 1060
+    },
+    {
+      "grad_norm": 0.4857303202152252,
+      "learning_rate": 9.999674593650526e-05,
+      "loss": 0.218,
+      "step": 1070
+    },
+    {
+      "grad_norm": 0.6034067869186401,
+      "learning_rate": 9.99957343957588e-05,
+      "loss": 0.2083,
+      "step": 1080
+    },
+    {
+      "grad_norm": 0.5574386715888977,
+      "learning_rate": 9.99945861685836e-05,
+      "loss": 0.2127,
+      "step": 1090
+    },
+    {
+      "grad_norm": 0.4072793424129486,
+      "learning_rate": 9.999330125811884e-05,
+      "loss": 0.2081,
+      "step": 1100
+    },
+    {
+      "grad_norm": 0.37719443440437317,
+      "learning_rate": 9.999187966787744e-05,
+      "loss": 0.1937,
+      "step": 1110
+    },
+    {
+      "grad_norm": 0.46286049485206604,
+      "learning_rate": 9.999032140174595e-05,
+      "loss": 0.2011,
+      "step": 1120
+    },
+    {
+      "grad_norm": 0.4277225434780121,
+      "learning_rate": 9.998862646398464e-05,
+      "loss": 0.1918,
+      "step": 1130
+    },
+    {
+      "grad_norm": 0.4028959274291992,
+      "learning_rate": 9.998679485922739e-05,
+      "loss": 0.1767,
+      "step": 1140
+    },
+    {
+      "grad_norm": 0.60634446144104,
+      "learning_rate": 9.998482659248174e-05,
+      "loss": 0.1906,
+      "step": 1150
+    },
+    {
+      "grad_norm": 0.4854017496109009,
+      "learning_rate": 9.998272166912883e-05,
+      "loss": 0.2101,
+      "step": 1160
+    },
+    {
+      "grad_norm": 0.49438342452049255,
+      "learning_rate": 9.998048009492347e-05,
+      "loss": 0.2282,
+      "step": 1170
+    },
+    {
+      "grad_norm": 0.47291848063468933,
+      "learning_rate": 9.997810187599403e-05,
+      "loss": 0.1818,
+      "step": 1180
+    },
+    {
+      "grad_norm": 0.49231886863708496,
+      "learning_rate": 9.997558701884249e-05,
+      "loss": 0.2062,
+      "step": 1190
+    },
+    {
+      "grad_norm": 0.5292176604270935,
+      "learning_rate": 9.997293553034433e-05,
+      "loss": 0.1792,
+      "step": 1200
+    },
+    {
+      "grad_norm": 0.5177812576293945,
+      "learning_rate": 9.997014741774866e-05,
+      "loss": 0.1912,
+      "step": 1210
+    },
+    {
+      "grad_norm": 0.4155665636062622,
+      "learning_rate": 9.996722268867803e-05,
+      "loss": 0.2102,
+      "step": 1220
+    },
+    {
+      "grad_norm": 0.3383396863937378,
+      "learning_rate": 9.996416135112858e-05,
+      "loss": 0.195,
+      "step": 1230
+    },
+    {
+      "grad_norm": 0.4092361629009247,
+      "learning_rate": 9.996096341346988e-05,
+      "loss": 0.1837,
+      "step": 1240
+    },
+    {
+      "grad_norm": 0.5595426559448242,
+      "learning_rate": 9.995762888444495e-05,
+      "loss": 0.1861,
+      "step": 1250
+    },
+    {
+      "grad_norm": 0.44696924090385437,
+      "learning_rate": 9.995415777317027e-05,
+      "loss": 0.2106,
+      "step": 1260
+    },
+    {
+      "grad_norm": 0.5123691558837891,
+      "learning_rate": 9.995055008913574e-05,
+      "loss": 0.1896,
+      "step": 1270
+    },
+    {
+      "grad_norm": 0.5934571027755737,
+      "learning_rate": 9.994680584220463e-05,
+      "loss": 0.225,
+      "step": 1280
+    },
+    {
+      "grad_norm": 0.37691208720207214,
+      "learning_rate": 9.994292504261355e-05,
+      "loss": 0.2159,
+      "step": 1290
+    },
+    {
+      "grad_norm": 0.3848681151866913,
+      "learning_rate": 9.993890770097247e-05,
+      "loss": 0.1981,
+      "step": 1300
+    },
+    {
+      "grad_norm": 0.4785479009151459,
+      "learning_rate": 9.993475382826467e-05,
+      "loss": 0.2185,
+      "step": 1310
+    },
+    {
+      "grad_norm": 0.38651376962661743,
+      "learning_rate": 9.993046343584664e-05,
+      "loss": 0.1962,
+      "step": 1320
+    },
+    {
+      "grad_norm": 0.6333630084991455,
+      "learning_rate": 9.992603653544816e-05,
+      "loss": 0.1832,
+      "step": 1330
+    },
+    {
+      "grad_norm": 0.36634382605552673,
+      "learning_rate": 9.992147313917222e-05,
+      "loss": 0.1968,
+      "step": 1340
+    },
+    {
+      "grad_norm": 0.5570665001869202,
+      "learning_rate": 9.991677325949497e-05,
+      "loss": 0.1964,
+      "step": 1350
+    },
+    {
+      "grad_norm": 0.6064546704292297,
+      "learning_rate": 9.991193690926568e-05,
+      "loss": 0.2193,
+      "step": 1360
+    },
+    {
+      "grad_norm": 0.4859426021575928,
+      "learning_rate": 9.990696410170678e-05,
+      "loss": 0.2057,
+      "step": 1370
+    },
+    {
+      "grad_norm": 0.5011364817619324,
+      "learning_rate": 9.990185485041371e-05,
+      "loss": 0.2046,
+      "step": 1380
+    },
+    {
+      "grad_norm": 0.5596189498901367,
+      "learning_rate": 9.989660916935498e-05,
+      "loss": 0.2019,
+      "step": 1390
+    },
+    {
+      "grad_norm": 0.360697478055954,
+      "learning_rate": 9.989122707287208e-05,
+      "loss": 0.19,
+      "step": 1400
+    },
+    {
+      "grad_norm": 0.5550692081451416,
+      "learning_rate": 9.988570857567945e-05,
+      "loss": 0.2164,
+      "step": 1410
+    },
+    {
+      "grad_norm": 0.3348056375980377,
+      "learning_rate": 9.988005369286446e-05,
+      "loss": 0.1913,
+      "step": 1420
+    },
+    {
+      "grad_norm": 0.41162458062171936,
+      "learning_rate": 9.987426243988734e-05,
+      "loss": 0.1925,
+      "step": 1430
+    },
+    {
+      "grad_norm": 0.4569873809814453,
+      "learning_rate": 9.986833483258114e-05,
+      "loss": 0.1956,
+      "step": 1440
+    },
+    {
+      "grad_norm": 0.38820594549179077,
+      "learning_rate": 9.986227088715173e-05,
+      "loss": 0.2025,
+      "step": 1450
+    },
+    {
+      "grad_norm": 0.36603453755378723,
+      "learning_rate": 9.98560706201777e-05,
+      "loss": 0.205,
+      "step": 1460
+    },
+    {
+      "grad_norm": 0.46345266699790955,
+      "learning_rate": 9.984973404861036e-05,
+      "loss": 0.2039,
+      "step": 1470
+    },
+    {
+      "grad_norm": 0.3976169526576996,
+      "learning_rate": 9.984326118977361e-05,
+      "loss": 0.1851,
+      "step": 1480
+    },
+    {
+      "grad_norm": 0.4065582752227783,
+      "learning_rate": 9.983665206136406e-05,
+      "loss": 0.17,
+      "step": 1490
+    },
+    {
+      "grad_norm": 0.3648560643196106,
+      "learning_rate": 9.982990668145075e-05,
+      "loss": 0.2219,
+      "step": 1500
+    },
+    {
+      "grad_norm": 0.45016640424728394,
+      "learning_rate": 9.982302506847534e-05,
+      "loss": 0.1942,
+      "step": 1510
+    },
+    {
+      "grad_norm": 0.49017226696014404,
+      "learning_rate": 9.981600724125189e-05,
+      "loss": 0.1921,
+      "step": 1520
+    },
+    {
+      "grad_norm": 0.4160068929195404,
+      "learning_rate": 9.980885321896685e-05,
+      "loss": 0.1959,
+      "step": 1530
+    },
+    {
+      "grad_norm": 0.4685332477092743,
+      "learning_rate": 9.980156302117905e-05,
+      "loss": 0.2002,
+      "step": 1540
+    },
+    {
+      "grad_norm": 0.4941580295562744,
+      "learning_rate": 9.979413666781963e-05,
+      "loss": 0.1909,
+      "step": 1550
+    },
+    {
+      "grad_norm": 0.4263238310813904,
+      "learning_rate": 9.978657417919193e-05,
+      "loss": 0.2086,
+      "step": 1560
+    },
+    {
+      "grad_norm": 0.35773414373397827,
+      "learning_rate": 9.977887557597153e-05,
+      "loss": 0.1881,
+      "step": 1570
+    },
+    {
+      "grad_norm": 0.5850985050201416,
+      "learning_rate": 9.97710408792061e-05,
+      "loss": 0.2011,
+      "step": 1580
+    },
+    {
+      "grad_norm": 0.43927475810050964,
+      "learning_rate": 9.976307011031542e-05,
+      "loss": 0.1946,
+      "step": 1590
+    },
+    {
+      "grad_norm": 0.38386958837509155,
+      "learning_rate": 9.975496329109126e-05,
+      "loss": 0.1992,
+      "step": 1600
+    },
+    {
+      "grad_norm": 0.3906306326389313,
+      "learning_rate": 9.974672044369732e-05,
+      "loss": 0.195,
+      "step": 1610
+    },
+    {
+      "grad_norm": 0.3716115653514862,
+      "learning_rate": 9.97383415906693e-05,
+      "loss": 0.1886,
+      "step": 1620
+    },
+    {
+      "grad_norm": 0.4166058301925659,
+      "learning_rate": 9.97298267549146e-05,
+      "loss": 0.1942,
+      "step": 1630
+    },
+    {
+      "grad_norm": 0.515879213809967,
+      "learning_rate": 9.972117595971249e-05,
+      "loss": 0.1926,
+      "step": 1640
+    },
+    {
+      "grad_norm": 0.677603542804718,
+      "learning_rate": 9.971238922871391e-05,
+      "loss": 0.2056,
+      "step": 1650
+    },
+    {
+      "grad_norm": 0.37264329195022583,
+      "learning_rate": 9.970346658594142e-05,
+      "loss": 0.1864,
+      "step": 1660
+    },
+    {
+      "grad_norm": 0.6440589427947998,
+      "learning_rate": 9.969440805578923e-05,
+      "loss": 0.2002,
+      "step": 1670
+    },
+    {
+      "grad_norm": 0.39667099714279175,
+      "learning_rate": 9.968521366302298e-05,
+      "loss": 0.1634,
+      "step": 1680
+    },
+    {
+      "grad_norm": 0.49169060587882996,
+      "learning_rate": 9.967588343277981e-05,
+      "loss": 0.1747,
+      "step": 1690
+    },
+    {
+      "grad_norm": 0.7108457088470459,
+      "learning_rate": 9.966641739056818e-05,
+      "loss": 0.22,
+      "step": 1700
+    },
+    {
+      "grad_norm": 0.42115163803100586,
+      "learning_rate": 9.965681556226793e-05,
+      "loss": 0.2028,
+      "step": 1710
+    },
+    {
+      "grad_norm": 0.36843326687812805,
+      "learning_rate": 9.964707797413006e-05,
+      "loss": 0.1866,
+      "step": 1720
+    },
+    {
+      "grad_norm": 0.37428349256515503,
+      "learning_rate": 9.963720465277679e-05,
+      "loss": 0.1804,
+      "step": 1730
+    },
+    {
+      "grad_norm": 0.49117597937583923,
+      "learning_rate": 9.96271956252014e-05,
+      "loss": 0.202,
+      "step": 1740
+    },
+    {
+      "grad_norm": 0.5093969702720642,
+      "learning_rate": 9.961705091876816e-05,
+      "loss": 0.1864,
+      "step": 1750
+    },
+    {
+      "grad_norm": 0.49675390124320984,
+      "learning_rate": 9.960677056121235e-05,
+      "loss": 0.1938,
+      "step": 1760
+    },
+    {
+      "grad_norm": 0.46998053789138794,
+      "learning_rate": 9.959635458064005e-05,
+      "loss": 0.2147,
+      "step": 1770
+    },
+    {
+      "grad_norm": 0.3512859046459198,
+      "learning_rate": 9.958580300552815e-05,
+      "loss": 0.1879,
+      "step": 1780
+    },
+    {
+      "grad_norm": 0.39896416664123535,
+      "learning_rate": 9.957511586472426e-05,
+      "loss": 0.1977,
+      "step": 1790
+    },
+    {
+      "grad_norm": 0.47821298241615295,
+      "learning_rate": 9.956429318744662e-05,
+      "loss": 0.1828,
+      "step": 1800
+    },
+    {
+      "grad_norm": 0.4184552729129791,
+      "learning_rate": 9.955333500328404e-05,
+      "loss": 0.1937,
+      "step": 1810
+    },
+    {
+      "grad_norm": 0.5620543360710144,
+      "learning_rate": 9.95422413421957e-05,
+      "loss": 0.2039,
+      "step": 1820
+    },
+    {
+      "grad_norm": 0.4486919641494751,
+      "learning_rate": 9.953101223451133e-05,
+      "loss": 0.1963,
+      "step": 1830
+    },
+    {
+      "grad_norm": 0.5195168256759644,
+      "learning_rate": 9.951964771093085e-05,
+      "loss": 0.1872,
+      "step": 1840
+    },
+    {
+      "grad_norm": 0.7360044717788696,
+      "learning_rate": 9.950814780252442e-05,
+      "loss": 0.176,
+      "step": 1850
+    },
+    {
+      "grad_norm": 0.3479352295398712,
+      "learning_rate": 9.949651254073236e-05,
+      "loss": 0.1992,
+      "step": 1860
+    },
+    {
+      "grad_norm": 0.6191892623901367,
+      "learning_rate": 9.948474195736504e-05,
+      "loss": 0.1963,
+      "step": 1870
+    },
+    {
+      "grad_norm": 0.4803287386894226,
+      "learning_rate": 9.947283608460277e-05,
+      "loss": 0.1844,
+      "step": 1880
+    },
+    {
+      "grad_norm": 0.6341800689697266,
+      "learning_rate": 9.946079495499577e-05,
+      "loss": 0.1915,
+      "step": 1890
+    },
+    {
+      "grad_norm": 0.5674198865890503,
+      "learning_rate": 9.944861860146401e-05,
+      "loss": 0.1913,
+      "step": 1900
+    },
+    {
+      "grad_norm": 0.3358455002307892,
+      "learning_rate": 9.943630705729719e-05,
+      "loss": 0.193,
+      "step": 1910
+    },
+    {
+      "grad_norm": 0.5320000648498535,
+      "learning_rate": 9.942386035615459e-05,
+      "loss": 0.1918,
+      "step": 1920
+    },
+    {
+      "grad_norm": 0.43139195442199707,
+      "learning_rate": 9.941127853206503e-05,
+      "loss": 0.2048,
+      "step": 1930
+    },
+    {
+      "grad_norm": 0.47216835618019104,
+      "learning_rate": 9.939856161942673e-05,
+      "loss": 0.1933,
+      "step": 1940
+    },
+    {
+      "grad_norm": 0.5149073600769043,
+      "learning_rate": 9.938570965300724e-05,
+      "loss": 0.1995,
+      "step": 1950
+    },
+    {
+      "grad_norm": 0.4058394134044647,
+      "learning_rate": 9.937272266794335e-05,
+      "loss": 0.201,
+      "step": 1960
+    },
+    {
+      "grad_norm": 0.42147955298423767,
+      "learning_rate": 9.935960069974096e-05,
+      "loss": 0.1853,
+      "step": 1970
+    },
+    {
+      "grad_norm": 0.5313990712165833,
+      "learning_rate": 9.934634378427506e-05,
+      "loss": 0.192,
+      "step": 1980
+    },
+    {
+      "grad_norm": 0.37542814016342163,
+      "learning_rate": 9.933295195778954e-05,
+      "loss": 0.2193,
+      "step": 1990
+    },
+    {
+      "grad_norm": 0.41276150941848755,
+      "learning_rate": 9.931942525689715e-05,
+      "loss": 0.188,
+      "step": 2000
+    },
+    {
+      "grad_norm": 0.3326045870780945,
+      "learning_rate": 9.930576371857936e-05,
+      "loss": 0.1798,
+      "step": 2010
+    },
+    {
+      "grad_norm": 0.488316148519516,
+      "learning_rate": 9.929196738018629e-05,
+      "loss": 0.1776,
+      "step": 2020
+    },
+    {
+      "grad_norm": 0.5502291321754456,
+      "learning_rate": 9.927803627943662e-05,
+      "loss": 0.2047,
+      "step": 2030
+    },
+    {
+      "grad_norm": 0.4054112732410431,
+      "learning_rate": 9.926397045441744e-05,
+      "loss": 0.1858,
+      "step": 2040
+    },
+    {
+      "grad_norm": 0.503993034362793,
+      "learning_rate": 9.924976994358417e-05,
+      "loss": 0.2223,
+      "step": 2050
+    },
+    {
+      "grad_norm": 0.3643146753311157,
+      "learning_rate": 9.923543478576048e-05,
+      "loss": 0.1861,
+      "step": 2060
+    },
+    {
+      "grad_norm": 0.4147532284259796,
+      "learning_rate": 9.922096502013813e-05,
+      "loss": 0.1852,
+      "step": 2070
+    },
+    {
+      "grad_norm": 0.5896642804145813,
+      "learning_rate": 9.92063606862769e-05,
+      "loss": 0.1922,
+      "step": 2080
+    },
+    {
+      "grad_norm": 0.6279969811439514,
+      "learning_rate": 9.919162182410453e-05,
+      "loss": 0.1753,
+      "step": 2090
+    },
+    {
+      "grad_norm": 0.5181149840354919,
+      "learning_rate": 9.917674847391645e-05,
+      "loss": 0.1848,
+      "step": 2100
+    },
+    {
+      "grad_norm": 0.40997037291526794,
+      "learning_rate": 9.916174067637584e-05,
+      "loss": 0.1988,
+      "step": 2110
+    },
+    {
+      "grad_norm": 0.551565408706665,
+      "learning_rate": 9.914659847251348e-05,
+      "loss": 0.1978,
+      "step": 2120
+    },
+    {
+      "grad_norm": 0.37171655893325806,
+      "learning_rate": 9.913132190372753e-05,
+      "loss": 0.1706,
+      "step": 2130
+    },
+    {
+      "grad_norm": 0.3925575911998749,
+      "learning_rate": 9.911591101178359e-05,
+      "loss": 0.1841,
+      "step": 2140
+    },
+    {
+      "grad_norm": 0.45162108540534973,
+      "learning_rate": 9.910036583881443e-05,
+      "loss": 0.2014,
+      "step": 2150
+    },
+    {
+      "grad_norm": 0.568889856338501,
+      "learning_rate": 9.908468642731995e-05,
+      "loss": 0.1773,
+      "step": 2160
+    },
+    {
+      "grad_norm": 0.3534856140613556,
+      "learning_rate": 9.906887282016707e-05,
+      "loss": 0.1913,
+      "step": 2170
+    },
+    {
+      "grad_norm": 0.4262492060661316,
+      "learning_rate": 9.90529250605896e-05,
+      "loss": 0.2091,
+      "step": 2180
+    },
+    {
+      "grad_norm": 0.8466174006462097,
+      "learning_rate": 9.903684319218809e-05,
+      "loss": 0.1882,
+      "step": 2190
+    },
+    {
+      "grad_norm": 0.41412338614463806,
+      "learning_rate": 9.902062725892976e-05,
+      "loss": 0.1823,
+      "step": 2200
+    },
+    {
+      "grad_norm": 0.5441012382507324,
+      "learning_rate": 9.900427730514834e-05,
+      "loss": 0.2119,
+      "step": 2210
+    },
+    {
+      "grad_norm": 0.4396047294139862,
+      "learning_rate": 9.8987793375544e-05,
+      "loss": 0.1864,
+      "step": 2220
+    },
+    {
+      "grad_norm": 0.49779215455055237,
+      "learning_rate": 9.897117551518318e-05,
+      "loss": 0.1847,
+      "step": 2230
+    },
+    {
+      "grad_norm": 0.4468015730381012,
+      "learning_rate": 9.895442376949844e-05,
+      "loss": 0.1916,
+      "step": 2240
+    },
+    {
+      "grad_norm": 0.36594513058662415,
+      "learning_rate": 9.893753818428845e-05,
+      "loss": 0.2052,
+      "step": 2250
+    },
+    {
+      "grad_norm": 0.5838186740875244,
+      "learning_rate": 9.892051880571773e-05,
+      "loss": 0.1892,
+      "step": 2260
+    },
+    {
+      "grad_norm": 0.5487616062164307,
+      "learning_rate": 9.890336568031663e-05,
+      "loss": 0.2004,
+      "step": 2270
+    },
+    {
+      "grad_norm": 0.31143733859062195,
+      "learning_rate": 9.888607885498113e-05,
+      "loss": 0.1721,
+      "step": 2280
+    },
+    {
+      "grad_norm": 0.37593698501586914,
+      "learning_rate": 9.886865837697275e-05,
+      "loss": 0.1934,
+      "step": 2290
+    },
+    {
+      "grad_norm": 0.3458669185638428,
+      "learning_rate": 9.88511042939184e-05,
+      "loss": 0.1942,
+      "step": 2300
+    },
+    {
+      "grad_norm": 0.3525335192680359,
+      "learning_rate": 9.883341665381028e-05,
+      "loss": 0.1869,
+      "step": 2310
+    },
+    {
+      "grad_norm": 0.3350130319595337,
+      "learning_rate": 9.881559550500575e-05,
+      "loss": 0.1935,
+      "step": 2320
+    },
+    {
+      "grad_norm": 0.5341783761978149,
+      "learning_rate": 9.879764089622712e-05,
+      "loss": 0.1833,
+      "step": 2330
+    },
+    {
+      "grad_norm": 0.5393840670585632,
+      "learning_rate": 9.87795528765616e-05,
+      "loss": 0.1774,
+      "step": 2340
+    },
+    {
+      "grad_norm": 0.6250892281532288,
+      "learning_rate": 9.876133149546118e-05,
+      "loss": 0.2167,
+      "step": 2350
+    },
+    {
+      "grad_norm": 0.45851725339889526,
+      "learning_rate": 9.874297680274238e-05,
+      "loss": 0.1803,
+      "step": 2360
+    },
+    {
+      "grad_norm": 0.6423644423484802,
+      "learning_rate": 9.872448884858624e-05,
+      "loss": 0.2,
+      "step": 2370
+    },
+    {
+      "grad_norm": 0.4443056881427765,
+      "learning_rate": 9.870586768353815e-05,
+      "loss": 0.1624,
+      "step": 2380
+    },
+    {
+      "grad_norm": 0.6335181593894958,
+      "learning_rate": 9.868711335850764e-05,
+      "loss": 0.2045,
+      "step": 2390
+    },
+    {
+      "grad_norm": 0.38028770685195923,
+      "learning_rate": 9.866822592476833e-05,
+      "loss": 0.186,
+      "step": 2400
+    },
+    {
+      "grad_norm": 0.46251988410949707,
+      "learning_rate": 9.86492054339577e-05,
+      "loss": 0.1824,
+      "step": 2410
+    },
+    {
+      "grad_norm": 0.4443410336971283,
+      "learning_rate": 9.863005193807711e-05,
+      "loss": 0.1678,
+      "step": 2420
+    },
+    {
+      "grad_norm": 0.5266323089599609,
+      "learning_rate": 9.861076548949143e-05,
+      "loss": 0.1713,
+      "step": 2430
+    },
+    {
+      "grad_norm": 0.34131526947021484,
+      "learning_rate": 9.859134614092912e-05,
+      "loss": 0.1838,
+      "step": 2440
+    },
+    {
+      "grad_norm": 0.39628955721855164,
+      "learning_rate": 9.857179394548191e-05,
+      "loss": 0.1934,
+      "step": 2450
+    },
+    {
+      "grad_norm": 0.4386841952800751,
+      "learning_rate": 9.855210895660477e-05,
+      "loss": 0.1756,
+      "step": 2460
+    },
+    {
+      "grad_norm": 0.463236004114151,
+      "learning_rate": 9.853229122811568e-05,
+      "loss": 0.1862,
+      "step": 2470
+    },
+    {
+      "grad_norm": 0.3599270284175873,
+      "learning_rate": 9.851234081419559e-05,
+      "loss": 0.1835,
+      "step": 2480
+    },
+    {
+      "grad_norm": 0.4263663589954376,
+      "learning_rate": 9.849225776938814e-05,
+      "loss": 0.1903,
+      "step": 2490
+    },
+    {
+      "grad_norm": 0.3628334403038025,
+      "learning_rate": 9.847204214859964e-05,
+      "loss": 0.1718,
+      "step": 2500
+    },
+    {
+      "grad_norm": 0.41961997747421265,
+      "learning_rate": 9.845169400709879e-05,
+      "loss": 0.1916,
+      "step": 2510
+    },
+    {
+      "grad_norm": 0.445348858833313,
+      "learning_rate": 9.843121340051664e-05,
+      "loss": 0.1894,
+      "step": 2520
+    },
+    {
+      "grad_norm": 0.5569151043891907,
+      "learning_rate": 9.841060038484641e-05,
+      "loss": 0.1937,
+      "step": 2530
+    },
+    {
+      "grad_norm": 0.3943973481655121,
+      "learning_rate": 9.838985501644328e-05,
+      "loss": 0.1958,
+      "step": 2540
+    },
+    {
+      "grad_norm": 0.5496942400932312,
+      "learning_rate": 9.83689773520243e-05,
+      "loss": 0.2036,
+      "step": 2550
+    },
+    {
+      "grad_norm": 0.3871750235557556,
+      "learning_rate": 9.834796744866819e-05,
+      "loss": 0.1774,
+      "step": 2560
+    },
+    {
+      "grad_norm": 0.4247782528400421,
+      "learning_rate": 9.832682536381525e-05,
+      "loss": 0.1663,
+      "step": 2570
+    },
+    {
+      "grad_norm": 0.4165402054786682,
+      "learning_rate": 9.830555115526711e-05,
+      "loss": 0.1905,
+      "step": 2580
+    },
+    {
+      "grad_norm": 1.2667182683944702,
+      "learning_rate": 9.828414488118667e-05,
+      "loss": 0.2018,
+      "step": 2590
+    },
+    {
+      "grad_norm": 0.45332053303718567,
+      "learning_rate": 9.826260660009785e-05,
+      "loss": 0.1826,
+      "step": 2600
+    },
+    {
+      "grad_norm": 0.3480590283870697,
+      "learning_rate": 9.824093637088547e-05,
+      "loss": 0.1944,
+      "step": 2610
+    },
+    {
+      "grad_norm": 0.3985593020915985,
+      "learning_rate": 9.821913425279514e-05,
+      "loss": 0.1838,
+      "step": 2620
+    },
+    {
+      "grad_norm": 0.5786546468734741,
+      "learning_rate": 9.8197200305433e-05,
+      "loss": 0.1865,
+      "step": 2630
+    },
+    {
+      "grad_norm": 0.4882510304450989,
+      "learning_rate": 9.817513458876564e-05,
+      "loss": 0.19,
+      "step": 2640
+    },
+    {
+      "grad_norm": 0.5400271415710449,
+      "learning_rate": 9.815293716311987e-05,
+      "loss": 0.1842,
+      "step": 2650
+    },
+    {
+      "grad_norm": 0.46679797768592834,
+      "learning_rate": 9.813060808918262e-05,
+      "loss": 0.1699,
+      "step": 2660
+    },
+    {
+      "grad_norm": 0.5496741533279419,
+      "learning_rate": 9.810814742800069e-05,
+      "loss": 0.1694,
+      "step": 2670
+    },
+    {
+      "grad_norm": 0.5625917911529541,
+      "learning_rate": 9.808555524098074e-05,
+      "loss": 0.1885,
+      "step": 2680
+    },
+    {
+      "grad_norm": 0.4316520094871521,
+      "learning_rate": 9.806283158988887e-05,
+      "loss": 0.1777,
+      "step": 2690
+    },
+    {
+      "grad_norm": 0.449280709028244,
+      "learning_rate": 9.803997653685072e-05,
+      "loss": 0.176,
+      "step": 2700
+    },
+    {
+      "grad_norm": 0.40558287501335144,
+      "learning_rate": 9.801699014435112e-05,
+      "loss": 0.1677,
+      "step": 2710
+    },
+    {
+      "grad_norm": 0.5426893830299377,
+      "learning_rate": 9.799387247523398e-05,
+      "loss": 0.1942,
+      "step": 2720
+    },
+    {
+      "grad_norm": 0.35034891963005066,
+      "learning_rate": 9.797062359270215e-05,
+      "loss": 0.1951,
+      "step": 2730
+    },
+    {
+      "grad_norm": 0.43325069546699524,
+      "learning_rate": 9.794724356031715e-05,
+      "loss": 0.1816,
+      "step": 2740
+    },
+    {
+      "grad_norm": 0.4800868034362793,
+      "learning_rate": 9.792373244199913e-05,
+      "loss": 0.1813,
+      "step": 2750
+    },
+    {
+      "grad_norm": 0.38212496042251587,
+      "learning_rate": 9.790009030202658e-05,
+      "loss": 0.161,
+      "step": 2760
+    },
+    {
+      "grad_norm": 0.33050736784935,
+      "learning_rate": 9.78763172050362e-05,
+      "loss": 0.1675,
+      "step": 2770
+    },
+    {
+      "grad_norm": 0.4662337601184845,
+      "learning_rate": 9.785241321602274e-05,
+      "loss": 0.2006,
+      "step": 2780
+    },
+    {
+      "grad_norm": 0.36936289072036743,
+      "learning_rate": 9.782837840033879e-05,
+      "loss": 0.1813,
+      "step": 2790
+    },
+    {
+      "grad_norm": 0.5863960981369019,
+      "learning_rate": 9.780421282369461e-05,
+      "loss": 0.1684,
+      "step": 2800
+    },
+    {
+      "grad_norm": 0.4914627969264984,
+      "learning_rate": 9.777991655215797e-05,
+      "loss": 0.1799,
+      "step": 2810
+    },
+    {
+      "grad_norm": 0.38510528206825256,
+      "learning_rate": 9.775548965215394e-05,
+      "loss": 0.1695,
+      "step": 2820
+    },
+    {
+      "grad_norm": 0.4830227196216583,
+      "learning_rate": 9.773093219046474e-05,
+      "loss": 0.1953,
+      "step": 2830
+    },
+    {
+      "grad_norm": 0.6548153162002563,
+      "learning_rate": 9.770624423422954e-05,
+      "loss": 0.1711,
+      "step": 2840
+    },
+    {
+      "grad_norm": 0.44675204157829285,
+      "learning_rate": 9.768142585094426e-05,
+      "loss": 0.1945,
+      "step": 2850
+    },
+    {
+      "grad_norm": 0.4758172035217285,
+      "learning_rate": 9.765647710846142e-05,
+      "loss": 0.1989,
+      "step": 2860
+    },
+    {
+      "grad_norm": 0.6227725148200989,
+      "learning_rate": 9.763139807498991e-05,
+      "loss": 0.1967,
+      "step": 2870
+    },
+    {
+      "grad_norm": 0.39073339104652405,
+      "learning_rate": 9.760618881909487e-05,
+      "loss": 0.1751,
+      "step": 2880
+    },
+    {
+      "grad_norm": 0.4348808228969574,
+      "learning_rate": 9.758084940969744e-05,
+      "loss": 0.1771,
+      "step": 2890
+    },
+    {
+      "grad_norm": 0.46974802017211914,
+      "learning_rate": 9.755537991607459e-05,
+      "loss": 0.1957,
+      "step": 2900
+    },
+    {
+      "grad_norm": 0.3160678446292877,
+      "learning_rate": 9.752978040785895e-05,
+      "loss": 0.187,
+      "step": 2910
+    },
+    {
+      "grad_norm": 0.6031491160392761,
+      "learning_rate": 9.750405095503859e-05,
+      "loss": 0.181,
+      "step": 2920
+    },
+    {
+      "grad_norm": 0.41250836849212646,
+      "learning_rate": 9.747819162795686e-05,
+      "loss": 0.171,
+      "step": 2930
+    },
+    {
+      "grad_norm": 0.5526098608970642,
+      "learning_rate": 9.745220249731217e-05,
+      "loss": 0.1776,
+      "step": 2940
+    },
+    {
+      "grad_norm": 0.6233943104743958,
+      "learning_rate": 9.742608363415781e-05,
+      "loss": 0.203,
+      "step": 2950
+    },
+    {
+      "grad_norm": 0.4862516224384308,
+      "learning_rate": 9.739983510990176e-05,
+      "loss": 0.1673,
+      "step": 2960
+    },
+    {
+      "grad_norm": 0.5347321033477783,
+      "learning_rate": 9.737345699630647e-05,
+      "loss": 0.1792,
+      "step": 2970
+    },
+    {
+      "grad_norm": 0.6074512004852295,
+      "learning_rate": 9.734694936548869e-05,
+      "loss": 0.1809,
+      "step": 2980
+    },
+    {
+      "grad_norm": 0.45306071639060974,
+      "learning_rate": 9.732031228991932e-05,
+      "loss": 0.1748,
+      "step": 2990
+    },
+    {
+      "grad_norm": 0.3919631838798523,
+      "learning_rate": 9.729354584242302e-05,
+      "loss": 0.1731,
+      "step": 3000
+    },
+    {
+      "grad_norm": 0.6330358386039734,
+      "learning_rate": 9.726665009617832e-05,
+      "loss": 0.1668,
+      "step": 3010
+    },
+    {
+      "grad_norm": 0.4687839448451996,
+      "learning_rate": 9.723962512471714e-05,
+      "loss": 0.166,
+      "step": 3020
+    },
+    {
+      "grad_norm": 1.2356817722320557,
+      "learning_rate": 9.72124710019247e-05,
+      "loss": 0.2017,
+      "step": 3030
+    },
+    {
+      "grad_norm": 0.5399520397186279,
+      "learning_rate": 9.718518780203934e-05,
+      "loss": 0.1764,
+      "step": 3040
+    },
+    {
+      "grad_norm": 0.39416205883026123,
+      "learning_rate": 9.715777559965228e-05,
+      "loss": 0.1654,
+      "step": 3050
+    },
+    {
+      "grad_norm": 0.3908625841140747,
+      "learning_rate": 9.713023446970746e-05,
+      "loss": 0.1879,
+      "step": 3060
+    },
+    {
+      "grad_norm": 0.45571476221084595,
+      "learning_rate": 9.710256448750126e-05,
+      "loss": 0.1756,
+      "step": 3070
+    },
+    {
+      "grad_norm": 0.538749635219574,
+      "learning_rate": 9.707476572868235e-05,
+      "loss": 0.1878,
+      "step": 3080
+    },
+    {
+      "grad_norm": 0.5083677172660828,
+      "learning_rate": 9.704683826925149e-05,
+      "loss": 0.1837,
+      "step": 3090
+    },
+    {
+      "grad_norm": 0.6785595417022705,
+      "learning_rate": 9.701878218556129e-05,
+      "loss": 0.1854,
+      "step": 3100
+    },
+    {
+      "grad_norm": 0.40305230021476746,
+      "learning_rate": 9.699059755431598e-05,
+      "loss": 0.2086,
+      "step": 3110
+    },
+    {
+      "grad_norm": 0.4892082214355469,
+      "learning_rate": 9.696228445257132e-05,
+      "loss": 0.1813,
+      "step": 3120
+    },
+    {
+      "grad_norm": 0.5538605451583862,
+      "learning_rate": 9.693384295773419e-05,
+      "loss": 0.1876,
+      "step": 3130
+    },
+    {
+      "grad_norm": 0.5298478007316589,
+      "learning_rate": 9.690527314756259e-05,
+      "loss": 0.201,
+      "step": 3140
+    },
+    {
+      "grad_norm": 0.37606433033943176,
+      "learning_rate": 9.687657510016527e-05,
+      "loss": 0.1952,
+      "step": 3150
+    },
+    {
+      "grad_norm": 0.3208979666233063,
+      "learning_rate": 9.684774889400161e-05,
+      "loss": 0.2005,
+      "step": 3160
+    },
+    {
+      "grad_norm": 0.4781680107116699,
+      "learning_rate": 9.681879460788135e-05,
+      "loss": 0.218,
+      "step": 3170
+    },
+    {
+      "grad_norm": 0.5140867829322815,
+      "learning_rate": 9.67897123209644e-05,
+      "loss": 0.1873,
+      "step": 3180
+    },
+    {
+      "grad_norm": 0.38380861282348633,
+      "learning_rate": 9.676050211276062e-05,
+      "loss": 0.2082,
+      "step": 3190
+    },
+    {
+      "grad_norm": 0.4517562985420227,
+      "learning_rate": 9.673116406312962e-05,
+      "loss": 0.2026,
+      "step": 3200
+    },
+    {
+      "grad_norm": 0.6685178875923157,
+      "learning_rate": 9.67016982522805e-05,
+      "loss": 0.1809,
+      "step": 3210
+    },
+    {
+      "grad_norm": 0.38690057396888733,
+      "learning_rate": 9.667210476077164e-05,
+      "loss": 0.2038,
+      "step": 3220
+    },
+    {
+      "grad_norm": 0.45636653900146484,
+      "learning_rate": 9.664238366951055e-05,
+      "loss": 0.204,
+      "step": 3230
+    },
+    {
+      "grad_norm": 0.4715133011341095,
+      "learning_rate": 9.661253505975355e-05,
+      "loss": 0.2048,
+      "step": 3240
+    },
+    {
+      "grad_norm": 0.5734756588935852,
+      "learning_rate": 9.658255901310557e-05,
+      "loss": 0.2027,
+      "step": 3250
+    },
+    {
+      "grad_norm": 0.47970259189605713,
+      "learning_rate": 9.655245561152e-05,
+      "loss": 0.1835,
+      "step": 3260
+    },
+    {
+      "grad_norm": 0.7647449374198914,
+      "learning_rate": 9.65222249372984e-05,
+      "loss": 0.1868,
+      "step": 3270
+    },
+    {
+      "grad_norm": 0.728284478187561,
+      "learning_rate": 9.649186707309026e-05,
+      "loss": 0.2021,
+      "step": 3280
+    },
+    {
+      "grad_norm": 0.428120881319046,
+      "learning_rate": 9.646138210189283e-05,
+      "loss": 0.1809,
+      "step": 3290
+    },
+    {
+      "grad_norm": 0.3813256025314331,
+      "learning_rate": 9.643077010705087e-05,
+      "loss": 0.1733,
+      "step": 3300
+    },
+    {
+      "grad_norm": 0.43519726395606995,
+      "learning_rate": 9.640003117225637e-05,
+      "loss": 0.1765,
+      "step": 3310
+    },
+    {
+      "grad_norm": 0.4349793791770935,
+      "learning_rate": 9.636916538154846e-05,
+      "loss": 0.1931,
+      "step": 3320
+    },
+    {
+      "grad_norm": 0.6906169056892395,
+      "learning_rate": 9.633817281931296e-05,
+      "loss": 0.1876,
+      "step": 3330
+    },
+    {
+      "grad_norm": 0.5695355534553528,
+      "learning_rate": 9.630705357028242e-05,
+      "loss": 0.178,
+      "step": 3340
+    },
+    {
+      "grad_norm": 0.5425128936767578,
+      "learning_rate": 9.627580771953563e-05,
+      "loss": 0.1793,
+      "step": 3350
+    },
+    {
+      "grad_norm": 0.7037976980209351,
+      "learning_rate": 9.624443535249759e-05,
+      "loss": 0.2079,
+      "step": 3360
+    },
+    {
+      "grad_norm": 0.5787683129310608,
+      "learning_rate": 9.621293655493913e-05,
+      "loss": 0.1735,
+      "step": 3370
+    },
+    {
+      "grad_norm": 0.43937140703201294,
+      "learning_rate": 9.618131141297675e-05,
+      "loss": 0.197,
+      "step": 3380
+    },
+    {
+      "grad_norm": 1.0462347269058228,
+      "learning_rate": 9.614956001307242e-05,
+      "loss": 0.1935,
+      "step": 3390
+    },
+    {
+      "grad_norm": 0.443409264087677,
+      "learning_rate": 9.611768244203321e-05,
+      "loss": 0.2035,
+      "step": 3400
+    },
+    {
+      "grad_norm": 0.6702489256858826,
+      "learning_rate": 9.60856787870112e-05,
+      "loss": 0.1944,
+      "step": 3410
+    },
+    {
+      "grad_norm": 0.4318901002407074,
+      "learning_rate": 9.605354913550318e-05,
+      "loss": 0.1742,
+      "step": 3420
+    },
+    {
+      "grad_norm": 0.43600714206695557,
+      "learning_rate": 9.602129357535037e-05,
+      "loss": 0.1998,
+      "step": 3430
+    },
+    {
+      "grad_norm": 0.5550727844238281,
+      "learning_rate": 9.598891219473825e-05,
+      "loss": 0.1967,
+      "step": 3440
+    },
+    {
+      "grad_norm": 0.4986667037010193,
+      "learning_rate": 9.595640508219625e-05,
+      "loss": 0.1905,
+      "step": 3450
+    },
+    {
+      "grad_norm": 0.5785936713218689,
+      "learning_rate": 9.592377232659761e-05,
+      "loss": 0.2001,
+      "step": 3460
+    },
+    {
+      "grad_norm": 0.4718359112739563,
+      "learning_rate": 9.589101401715904e-05,
+      "loss": 0.1856,
+      "step": 3470
+    },
+    {
+      "grad_norm": 0.5924352407455444,
+      "learning_rate": 9.585813024344045e-05,
+      "loss": 0.1818,
+      "step": 3480
+    },
+    {
+      "grad_norm": 0.7085385322570801,
+      "learning_rate": 9.58251210953449e-05,
+      "loss": 0.1842,
+      "step": 3490
+    },
+    {
+      "grad_norm": 0.5206746459007263,
+      "learning_rate": 9.579198666311809e-05,
+      "loss": 0.1891,
+      "step": 3500
+    },
+    {
+      "grad_norm": 0.4825865626335144,
+      "learning_rate": 9.575872703734832e-05,
+      "loss": 0.2127,
+      "step": 3510
+    },
+    {
+      "grad_norm": 0.40282726287841797,
+      "learning_rate": 9.572534230896611e-05,
+      "loss": 0.175,
+      "step": 3520
+    },
+    {
+      "grad_norm": 0.4739479124546051,
+      "learning_rate": 9.569183256924403e-05,
+      "loss": 0.1765,
+      "step": 3530
+    },
+    {
+      "grad_norm": 0.4922867715358734,
+      "learning_rate": 9.565819790979646e-05,
+      "loss": 0.203,
+      "step": 3540
+    },
+    {
+      "grad_norm": 0.4966680705547333,
+      "learning_rate": 9.562443842257925e-05,
+      "loss": 0.1892,
+      "step": 3550
+    },
+    {
+      "grad_norm": 0.5746491551399231,
+      "learning_rate": 9.559055419988956e-05,
+      "loss": 0.2099,
+      "step": 3560
+    },
+    {
+      "grad_norm": 1.0019278526306152,
+      "learning_rate": 9.555654533436557e-05,
+      "loss": 0.1942,
+      "step": 3570
+    },
+    {
+      "grad_norm": 0.49017074704170227,
+      "learning_rate": 9.552241191898621e-05,
+      "loss": 0.2015,
+      "step": 3580
+    },
+    {
+      "grad_norm": 0.43805229663848877,
+      "learning_rate": 9.548815404707092e-05,
+      "loss": 0.1769,
+      "step": 3590
+    },
+    {
+      "grad_norm": 0.3613499104976654,
+      "learning_rate": 9.545377181227942e-05,
+      "loss": 0.1654,
+      "step": 3600
+    },
+    {
+      "grad_norm": 0.37925782799720764,
+      "learning_rate": 9.541926530861145e-05,
+      "loss": 0.2002,
+      "step": 3610
+    },
+    {
+      "grad_norm": 0.6143661141395569,
+      "learning_rate": 9.538463463040645e-05,
+      "loss": 0.2052,
+      "step": 3620
+    },
+    {
+      "grad_norm": 0.5052018761634827,
+      "learning_rate": 9.534987987234337e-05,
+      "loss": 0.1859,
+      "step": 3630
+    },
+    {
+      "grad_norm": 0.3767317235469818,
+      "learning_rate": 9.53150011294404e-05,
+      "loss": 0.1887,
+      "step": 3640
+    },
+    {
+      "grad_norm": 0.5335121154785156,
+      "learning_rate": 9.527999849705471e-05,
+      "loss": 0.1829,
+      "step": 3650
+    },
+    {
+      "grad_norm": 0.5821238160133362,
+      "learning_rate": 9.524487207088213e-05,
+      "loss": 0.1907,
+      "step": 3660
+    },
+    {
+      "grad_norm": 0.5295085906982422,
+      "learning_rate": 9.520962194695698e-05,
+      "loss": 0.1936,
+      "step": 3670
+    },
+    {
+      "grad_norm": 0.4983557462692261,
+      "learning_rate": 9.517424822165175e-05,
+      "loss": 0.2013,
+      "step": 3680
+    },
+    {
+      "grad_norm": 0.6450728178024292,
+      "learning_rate": 9.513875099167685e-05,
+      "loss": 0.1865,
+      "step": 3690
+    },
+    {
+      "grad_norm": 0.40041887760162354,
+      "learning_rate": 9.510313035408035e-05,
+      "loss": 0.1858,
+      "step": 3700
+    },
+    {
+      "grad_norm": 0.5237501263618469,
+      "learning_rate": 9.506738640624775e-05,
+      "loss": 0.1728,
+      "step": 3710
+    },
+    {
+      "grad_norm": 0.5804911255836487,
+      "learning_rate": 9.50315192459016e-05,
+      "loss": 0.1822,
+      "step": 3720
+    },
+    {
+      "grad_norm": 0.3805403411388397,
+      "learning_rate": 9.499552897110136e-05,
+      "loss": 0.1709,
+      "step": 3730
+    },
+    {
+      "grad_norm": 0.527192234992981,
+      "learning_rate": 9.495941568024304e-05,
+      "loss": 0.182,
+      "step": 3740
+    },
+    {
+      "grad_norm": 0.44829973578453064,
+      "learning_rate": 9.492317947205904e-05,
+      "loss": 0.1765,
+      "step": 3750
+    },
+    {
+      "grad_norm": 0.5719584822654724,
+      "learning_rate": 9.488682044561775e-05,
+      "loss": 0.1891,
+      "step": 3760
+    },
+    {
+      "grad_norm": 0.5995995402336121,
+      "learning_rate": 9.485033870032335e-05,
+      "loss": 0.2026,
+      "step": 3770
+    },
+    {
+      "grad_norm": 0.6265711784362793,
+      "learning_rate": 9.481373433591556e-05,
+      "loss": 0.1814,
+      "step": 3780
+    },
+    {
+      "grad_norm": 0.4861345589160919,
+      "learning_rate": 9.47770074524693e-05,
+      "loss": 0.18,
+      "step": 3790
+    },
+    {
+      "grad_norm": 0.457229882478714,
+      "learning_rate": 9.474015815039446e-05,
+      "loss": 0.17,
+      "step": 3800
+    },
+    {
+      "grad_norm": 0.5541972517967224,
+      "learning_rate": 9.470318653043565e-05,
+      "loss": 0.1676,
+      "step": 3810
+    },
+    {
+      "grad_norm": 0.4805838465690613,
+      "learning_rate": 9.466609269367185e-05,
+      "loss": 0.1805,
+      "step": 3820
+    },
+    {
+      "grad_norm": 0.6064276695251465,
+      "learning_rate": 9.46288767415162e-05,
+      "loss": 0.1732,
+      "step": 3830
+    },
+    {
+      "grad_norm": 0.3876231610774994,
+      "learning_rate": 9.459153877571567e-05,
+      "loss": 0.1615,
+      "step": 3840
+    },
+    {
+      "grad_norm": 0.6367563009262085,
+      "learning_rate": 9.455407889835087e-05,
+      "loss": 0.1724,
+      "step": 3850
+    },
+    {
+      "grad_norm": 0.5864294171333313,
+      "learning_rate": 9.451649721183564e-05,
+      "loss": 0.1748,
+      "step": 3860
+    },
+    {
+      "grad_norm": 0.49465158581733704,
+      "learning_rate": 9.447879381891692e-05,
+      "loss": 0.1975,
+      "step": 3870
+    },
+    {
+      "grad_norm": 0.46496355533599854,
+      "learning_rate": 9.444096882267428e-05,
+      "loss": 0.1983,
+      "step": 3880
+    },
+    {
+      "grad_norm": 0.6822494864463806,
+      "learning_rate": 9.440302232651988e-05,
+      "loss": 0.1737,
+      "step": 3890
+    },
+    {
+      "grad_norm": 0.4842269718647003,
+      "learning_rate": 9.436495443419795e-05,
+      "loss": 0.1938,
+      "step": 3900
+    },
+    {
+      "grad_norm": 0.4859203100204468,
+      "learning_rate": 9.432676524978466e-05,
+      "loss": 0.1706,
+      "step": 3910
+    },
+    {
+      "grad_norm": 0.5386573672294617,
+      "learning_rate": 9.42884548776878e-05,
+      "loss": 0.1751,
+      "step": 3920
+    },
+    {
+      "grad_norm": 0.4070545732975006,
+      "learning_rate": 9.425002342264646e-05,
+      "loss": 0.2115,
+      "step": 3930
+    },
+    {
+      "grad_norm": 0.5709313154220581,
+      "learning_rate": 9.421147098973077e-05,
+      "loss": 0.1939,
+      "step": 3940
+    },
+    {
+      "grad_norm": 0.6231239438056946,
+      "learning_rate": 9.41727976843416e-05,
+      "loss": 0.1807,
+      "step": 3950
+    },
+    {
+      "grad_norm": 0.5807859301567078,
+      "learning_rate": 9.413400361221029e-05,
+      "loss": 0.2212,
+      "step": 3960
+    },
+    {
+      "grad_norm": 0.4486054480075836,
+      "learning_rate": 9.409508887939835e-05,
+      "loss": 0.1888,
+      "step": 3970
+    },
+    {
+      "grad_norm": 0.5400952696800232,
+      "learning_rate": 9.40560535922972e-05,
+      "loss": 0.1659,
+      "step": 3980
+    },
+    {
+      "grad_norm": 0.4326404929161072,
+      "learning_rate": 9.40168978576278e-05,
+      "loss": 0.1874,
+      "step": 3990
+    },
+    {
+      "grad_norm": 0.53413325548172,
+      "learning_rate": 9.397762178244043e-05,
+      "loss": 0.1838,
+      "step": 4000
+    },
+    {
+      "grad_norm": 0.5266087651252747,
+      "learning_rate": 9.393822547411439e-05,
+      "loss": 0.1985,
+      "step": 4010
+    },
+    {
+      "grad_norm": 0.4578450918197632,
+      "learning_rate": 9.389870904035769e-05,
+      "loss": 0.1697,
+      "step": 4020
+    },
+    {
+      "grad_norm": 0.5176160335540771,
+      "learning_rate": 9.385907258920672e-05,
+      "loss": 0.1761,
+      "step": 4030
+    },
+    {
+      "grad_norm": 0.6311068534851074,
+      "learning_rate": 9.381931622902607e-05,
+      "loss": 0.1615,
+      "step": 4040
+    },
+    {
+      "grad_norm": 0.3930976688861847,
+      "learning_rate": 9.377944006850807e-05,
+      "loss": 0.1874,
+      "step": 4050
+    },
+    {
+      "grad_norm": 0.5806311368942261,
+      "learning_rate": 9.373944421667265e-05,
+      "loss": 0.1859,
+      "step": 4060
+    },
+    {
+      "grad_norm": 0.39180493354797363,
+      "learning_rate": 9.369932878286691e-05,
+      "loss": 0.1819,
+      "step": 4070
+    },
+    {
+      "grad_norm": 0.38805386424064636,
+      "learning_rate": 9.365909387676494e-05,
+      "loss": 0.1739,
+      "step": 4080
+    },
+    {
+      "grad_norm": 0.5136656761169434,
+      "learning_rate": 9.361873960836744e-05,
+      "loss": 0.187,
+      "step": 4090
+    },
+    {
+      "grad_norm": 0.41437581181526184,
+      "learning_rate": 9.357826608800142e-05,
+      "loss": 0.169,
+      "step": 4100
+    },
+    {
+      "grad_norm": 0.5145320296287537,
+      "learning_rate": 9.353767342631994e-05,
+      "loss": 0.201,
+      "step": 4110
+    },
+    {
+      "grad_norm": 0.49589014053344727,
+      "learning_rate": 9.34969617343018e-05,
+      "loss": 0.1778,
+      "step": 4120
+    },
+    {
+      "grad_norm": 0.6092096567153931,
+      "learning_rate": 9.345613112325122e-05,
+      "loss": 0.1799,
+      "step": 4130
+    },
+    {
+      "grad_norm": 0.3946431577205658,
+      "learning_rate": 9.34151817047975e-05,
+      "loss": 0.1754,
+      "step": 4140
+    },
+    {
+      "grad_norm": 0.390053927898407,
+      "learning_rate": 9.33741135908948e-05,
+      "loss": 0.1719,
+      "step": 4150
+    },
+    {
+      "grad_norm": 0.542607843875885,
+      "learning_rate": 9.33329268938218e-05,
+      "loss": 0.1677,
+      "step": 4160
+    },
+    {
+      "grad_norm": 0.5465247631072998,
+      "learning_rate": 9.329162172618132e-05,
+      "loss": 0.1897,
+      "step": 4170
+    },
+    {
+      "grad_norm": 0.5028508901596069,
+      "learning_rate": 9.325019820090013e-05,
+      "loss": 0.1873,
+      "step": 4180
+    },
+    {
+      "grad_norm": 0.5170102119445801,
+      "learning_rate": 9.320865643122855e-05,
+      "loss": 0.1813,
+      "step": 4190
+    },
+    {
+      "grad_norm": 0.3968212306499481,
+      "learning_rate": 9.316699653074023e-05,
+      "loss": 0.1755,
+      "step": 4200
+    },
+    {
+      "grad_norm": 0.6035100817680359,
+      "learning_rate": 9.312521861333172e-05,
+      "loss": 0.1737,
+      "step": 4210
+    },
+    {
+      "grad_norm": 0.5333013534545898,
+      "learning_rate": 9.308332279322224e-05,
+      "loss": 0.1739,
+      "step": 4220
+    },
+    {
+      "grad_norm": 0.508972704410553,
+      "learning_rate": 9.304130918495338e-05,
+      "loss": 0.1789,
+      "step": 4230
+    },
+    {
+      "grad_norm": 0.5775176286697388,
+      "learning_rate": 9.299917790338874e-05,
+      "loss": 0.1782,
+      "step": 4240
+    },
+    {
+      "grad_norm": 0.423092246055603,
+      "learning_rate": 9.295692906371363e-05,
+      "loss": 0.1914,
+      "step": 4250
+    },
+    {
+      "grad_norm": 0.4618014693260193,
+      "learning_rate": 9.291456278143476e-05,
+      "loss": 0.1825,
+      "step": 4260
+    },
+    {
+      "grad_norm": 0.53009432554245,
+      "learning_rate": 9.287207917237994e-05,
+      "loss": 0.1846,
+      "step": 4270
+    },
+    {
+      "grad_norm": 0.4253798723220825,
+      "learning_rate": 9.282947835269773e-05,
+      "loss": 0.1767,
+      "step": 4280
+    },
+    {
+      "grad_norm": 0.4627242386341095,
+      "learning_rate": 9.278676043885715e-05,
+      "loss": 0.186,
+      "step": 4290
+    },
+    {
+      "grad_norm": 0.6383516192436218,
+      "learning_rate": 9.274392554764733e-05,
+      "loss": 0.1727,
+      "step": 4300
+    },
+    {
+      "grad_norm": 0.6066827774047852,
+      "learning_rate": 9.270097379617723e-05,
+      "loss": 0.178,
+      "step": 4310
+    },
+    {
+      "grad_norm": 0.5484395027160645,
+      "learning_rate": 9.26579053018753e-05,
+      "loss": 0.1841,
+      "step": 4320
+    },
+    {
+      "grad_norm": 0.50762939453125,
+      "learning_rate": 9.261472018248918e-05,
+      "loss": 0.1689,
+      "step": 4330
+    },
+    {
+      "grad_norm": 0.35956844687461853,
+      "learning_rate": 9.25714185560853e-05,
+      "loss": 0.1688,
+      "step": 4340
+    },
+    {
+      "grad_norm": 0.5380527377128601,
+      "learning_rate": 9.252800054104868e-05,
+      "loss": 0.1785,
+      "step": 4350
+    },
+    {
+      "grad_norm": 0.3856523931026459,
+      "learning_rate": 9.248446625608252e-05,
+      "loss": 0.1618,
+      "step": 4360
+    },
+    {
+      "grad_norm": 0.48960137367248535,
+      "learning_rate": 9.244081582020789e-05,
+      "loss": 0.1698,
+      "step": 4370
+    },
+    {
+      "grad_norm": 0.575940728187561,
+      "learning_rate": 9.239704935276339e-05,
+      "loss": 0.17,
+      "step": 4380
+    },
+    {
+      "grad_norm": 0.4953281581401825,
+      "learning_rate": 9.235316697340489e-05,
+      "loss": 0.1914,
+      "step": 4390
+    },
+    {
+      "grad_norm": 0.3985150158405304,
+      "learning_rate": 9.230916880210512e-05,
+      "loss": 0.1955,
+      "step": 4400
+    },
+    {
+      "grad_norm": 0.5684506297111511,
+      "learning_rate": 9.226505495915342e-05,
+      "loss": 0.1782,
+      "step": 4410
+    },
+    {
+      "grad_norm": 0.46522441506385803,
+      "learning_rate": 9.222082556515536e-05,
+      "loss": 0.1779,
+      "step": 4420
+    },
+    {
+      "grad_norm": 0.7446576952934265,
+      "learning_rate": 9.217648074103242e-05,
+      "loss": 0.1875,
+      "step": 4430
+    },
+    {
+      "grad_norm": 0.5002114176750183,
+      "learning_rate": 9.213202060802161e-05,
+      "loss": 0.1623,
+      "step": 4440
+    },
+    {
+      "grad_norm": 0.6910802125930786,
+      "learning_rate": 9.208744528767528e-05,
+      "loss": 0.1868,
+      "step": 4450
+    },
+    {
+      "grad_norm": 0.5340420007705688,
+      "learning_rate": 9.204275490186064e-05,
+      "loss": 0.1709,
+      "step": 4460
+    },
+    {
+      "grad_norm": 0.41052910685539246,
+      "learning_rate": 9.199794957275949e-05,
+      "loss": 0.1946,
+      "step": 4470
+    },
+    {
+      "grad_norm": 0.34787020087242126,
+      "learning_rate": 9.19530294228679e-05,
+      "loss": 0.1675,
+      "step": 4480
+    },
+    {
+      "grad_norm": 0.578985869884491,
+      "learning_rate": 9.190799457499583e-05,
+      "loss": 0.1851,
+      "step": 4490
+    },
+    {
+      "grad_norm": 0.37645426392555237,
+      "learning_rate": 9.186284515226686e-05,
+      "loss": 0.1621,
+      "step": 4500
+    },
+    {
+      "grad_norm": 0.5258778929710388,
+      "learning_rate": 9.181758127811777e-05,
+      "loss": 0.1796,
+      "step": 4510
+    },
+    {
+      "grad_norm": 0.45895397663116455,
+      "learning_rate": 9.177220307629825e-05,
+      "loss": 0.176,
+      "step": 4520
+    },
+    {
+      "grad_norm": 0.4690583646297455,
+      "learning_rate": 9.172671067087059e-05,
+      "loss": 0.1694,
+      "step": 4530
+    },
+    {
+      "grad_norm": 0.5963845252990723,
+      "learning_rate": 9.16811041862093e-05,
+      "loss": 0.1766,
+      "step": 4540
+    },
+    {
+      "grad_norm": 0.4553046226501465,
+      "learning_rate": 9.163538374700076e-05,
+      "loss": 0.1905,
+      "step": 4550
+    },
+    {
+      "grad_norm": 0.37714922428131104,
+      "learning_rate": 9.158954947824287e-05,
+      "loss": 0.1786,
+      "step": 4560
+    },
+    {
+      "grad_norm": 0.505889356136322,
+      "learning_rate": 9.154360150524482e-05,
+      "loss": 0.1744,
+      "step": 4570
+    },
+    {
+      "grad_norm": 0.46347787976264954,
+      "learning_rate": 9.14975399536266e-05,
+      "loss": 0.1798,
+      "step": 4580
+    },
+    {
+      "grad_norm": 0.5350095629692078,
+      "learning_rate": 9.14513649493187e-05,
+      "loss": 0.1911,
+      "step": 4590
+    },
+    {
+      "grad_norm": 0.6578544974327087,
+      "learning_rate": 9.140507661856187e-05,
+      "loss": 0.175,
+      "step": 4600
+    },
+    {
+      "grad_norm": 0.6321951746940613,
+      "learning_rate": 9.135867508790661e-05,
+      "loss": 0.1827,
+      "step": 4610
+    },
+    {
+      "grad_norm": 0.60100919008255,
+      "learning_rate": 9.131216048421291e-05,
+      "loss": 0.1818,
+      "step": 4620
+    },
+    {
+      "grad_norm": 0.6358137130737305,
+      "learning_rate": 9.126553293464998e-05,
+      "loss": 0.1831,
+      "step": 4630
+    },
+    {
+      "grad_norm": 0.6207745671272278,
+      "learning_rate": 9.121879256669572e-05,
+      "loss": 0.184,
+      "step": 4640
+    },
+    {
+      "grad_norm": 0.4525275230407715,
+      "learning_rate": 9.117193950813652e-05,
+      "loss": 0.1668,
+      "step": 4650
+    },
+    {
+      "grad_norm": 0.5048314929008484,
+      "learning_rate": 9.112497388706685e-05,
+      "loss": 0.1781,
+      "step": 4660
+    },
+    {
+      "grad_norm": 0.6398943662643433,
+      "learning_rate": 9.10778958318889e-05,
+      "loss": 0.1938,
+      "step": 4670
+    },
+    {
+      "grad_norm": 0.3619368076324463,
+      "learning_rate": 9.103070547131232e-05,
+      "loss": 0.1761,
+      "step": 4680
+    },
+    {
+      "grad_norm": 0.4355509877204895,
+      "learning_rate": 9.098340293435375e-05,
+      "loss": 0.174,
+      "step": 4690
+    },
+    {
+      "grad_norm": 0.37825003266334534,
+      "learning_rate": 9.093598835033649e-05,
+      "loss": 0.1638,
+      "step": 4700
+    },
+    {
+      "grad_norm": 0.5397427082061768,
+      "learning_rate": 9.088846184889021e-05,
+      "loss": 0.1888,
+      "step": 4710
+    },
+    {
+      "grad_norm": 0.4330540895462036,
+      "learning_rate": 9.084082355995057e-05,
+      "loss": 0.1566,
+      "step": 4720
+    },
+    {
+      "grad_norm": 0.40616077184677124,
+      "learning_rate": 9.079307361375882e-05,
+      "loss": 0.1776,
+      "step": 4730
+    },
+    {
+      "grad_norm": 0.4763285517692566,
+      "learning_rate": 9.074521214086149e-05,
+      "loss": 0.1911,
+      "step": 4740
+    },
+    {
+      "grad_norm": 0.6156401634216309,
+      "learning_rate": 9.069723927211001e-05,
+      "loss": 0.1859,
+      "step": 4750
+    },
+    {
+      "grad_norm": 0.3529934287071228,
+      "learning_rate": 9.064915513866037e-05,
+      "loss": 0.166,
+      "step": 4760
+    },
+    {
+      "grad_norm": 0.6315341591835022,
+      "learning_rate": 9.060095987197279e-05,
+      "loss": 0.1812,
+      "step": 4770
+    },
+    {
+      "grad_norm": 0.5327130556106567,
+      "learning_rate": 9.055265360381126e-05,
+      "loss": 0.1664,
+      "step": 4780
+    },
+    {
+      "grad_norm": 0.4192088842391968,
+      "learning_rate": 9.050423646624326e-05,
+      "loss": 0.1733,
+      "step": 4790
+    },
+    {
+      "grad_norm": 0.9486660361289978,
+      "learning_rate": 9.045570859163943e-05,
+      "loss": 0.1671,
+      "step": 4800
+    },
+    {
+      "grad_norm": 0.3756958246231079,
+      "learning_rate": 9.04070701126731e-05,
+      "loss": 0.1924,
+      "step": 4810
+    },
+    {
+      "grad_norm": 0.5123847126960754,
+      "learning_rate": 9.035832116232001e-05,
+      "loss": 0.1707,
+      "step": 4820
+    },
+    {
+      "grad_norm": 0.553472638130188,
+      "learning_rate": 9.030946187385796e-05,
+      "loss": 0.1747,
+      "step": 4830
+    },
+    {
+      "grad_norm": 0.38321569561958313,
+      "learning_rate": 9.026049238086635e-05,
+      "loss": 0.1738,
+      "step": 4840
+    },
+    {
+      "grad_norm": 0.4932466149330139,
+      "learning_rate": 9.021141281722591e-05,
+      "loss": 0.1722,
+      "step": 4850
+    },
+    {
+      "grad_norm": 0.49787822365760803,
+      "learning_rate": 9.01622233171183e-05,
+      "loss": 0.1733,
+      "step": 4860
+    },
+    {
+      "grad_norm": 0.5631378889083862,
+      "learning_rate": 9.011292401502574e-05,
+      "loss": 0.1696,
+      "step": 4870
+    },
+    {
+      "grad_norm": 0.8084385991096497,
+      "learning_rate": 9.006351504573063e-05,
+      "loss": 0.1723,
+      "step": 4880
+    },
+    {
+      "grad_norm": 0.5995148420333862,
+      "learning_rate": 9.001399654431519e-05,
+      "loss": 0.1703,
+      "step": 4890
+    },
+    {
+      "grad_norm": 0.5791090130805969,
+      "learning_rate": 8.996436864616116e-05,
+      "loss": 0.1714,
+      "step": 4900
+    },
+    {
+      "grad_norm": 0.3513377010822296,
+      "learning_rate": 8.991463148694925e-05,
+      "loss": 0.1614,
+      "step": 4910
+    },
+    {
+      "grad_norm": 0.4377618134021759,
+      "learning_rate": 8.986478520265902e-05,
+      "loss": 0.1584,
+      "step": 4920
+    },
+    {
+      "grad_norm": 0.5671657919883728,
+      "learning_rate": 8.981482992956827e-05,
+      "loss": 0.1879,
+      "step": 4930
+    },
+    {
+      "grad_norm": 0.5192087292671204,
+      "learning_rate": 8.976476580425282e-05,
+      "loss": 0.1866,
+      "step": 4940
+    },
+    {
+      "grad_norm": 0.45744383335113525,
+      "learning_rate": 8.971459296358606e-05,
+      "loss": 0.1708,
+      "step": 4950
+    },
+    {
+      "grad_norm": 0.7598880529403687,
+      "learning_rate": 8.966431154473864e-05,
+      "loss": 0.171,
+      "step": 4960
+    },
+    {
+      "grad_norm": 0.4569564759731293,
+      "learning_rate": 8.961392168517803e-05,
+      "loss": 0.2025,
+      "step": 4970
+    },
+    {
+      "grad_norm": 0.429435670375824,
+      "learning_rate": 8.956342352266821e-05,
+      "loss": 0.1774,
+      "step": 4980
+    },
+    {
+      "grad_norm": 0.6855875849723816,
+      "learning_rate": 8.95128171952692e-05,
+      "loss": 0.1768,
+      "step": 4990
+    },
+    {
+      "grad_norm": 0.44373345375061035,
+      "learning_rate": 8.946210284133676e-05,
+      "loss": 0.186,
+      "step": 5000
+    },
+    {
+      "grad_norm": 0.3834450840950012,
+      "learning_rate": 8.941128059952201e-05,
+      "loss": 0.1819,
+      "step": 5010
+    },
+    {
+      "grad_norm": 0.7311763763427734,
+      "learning_rate": 8.936035060877102e-05,
+      "loss": 0.1819,
+      "step": 5020
+    },
+    {
+      "grad_norm": 0.37648001313209534,
+      "learning_rate": 8.930931300832443e-05,
+      "loss": 0.1777,
+      "step": 5030
+    },
+    {
+      "grad_norm": 0.5130906105041504,
+      "learning_rate": 8.925816793771711e-05,
+      "loss": 0.1716,
+      "step": 5040
+    },
+    {
+      "grad_norm": 0.5409080386161804,
+      "learning_rate": 8.92069155367777e-05,
+      "loss": 0.1733,
+      "step": 5050
+    },
+    {
+      "grad_norm": 0.3500835597515106,
+      "learning_rate": 8.915555594562834e-05,
+      "loss": 0.1747,
+      "step": 5060
+    },
+    {
+      "grad_norm": 0.45488786697387695,
+      "learning_rate": 8.910408930468416e-05,
+      "loss": 0.1936,
+      "step": 5070
+    },
+    {
+      "grad_norm": 0.47840750217437744,
+      "learning_rate": 8.905251575465303e-05,
+      "loss": 0.1757,
+      "step": 5080
+    },
+    {
+      "grad_norm": 0.5184291005134583,
+      "learning_rate": 8.900083543653502e-05,
+      "loss": 0.1598,
+      "step": 5090
+    },
+    {
+      "grad_norm": 0.4018440544605255,
+      "learning_rate": 8.894904849162218e-05,
+      "loss": 0.1517,
+      "step": 5100
+    },
+    {
+      "grad_norm": 0.6080432534217834,
+      "learning_rate": 8.889715506149802e-05,
+      "loss": 0.1795,
+      "step": 5110
+    },
+    {
+      "grad_norm": 0.6320006251335144,
+      "learning_rate": 8.884515528803722e-05,
+      "loss": 0.2061,
+      "step": 5120
+    },
+    {
+      "grad_norm": 0.4888564944267273,
+      "learning_rate": 8.879304931340517e-05,
+      "loss": 0.1732,
+      "step": 5130
+    },
+    {
+      "grad_norm": 0.7844697833061218,
+      "learning_rate": 8.874083728005759e-05,
+      "loss": 0.176,
+      "step": 5140
+    },
+    {
+      "grad_norm": 0.4245668649673462,
+      "learning_rate": 8.868851933074021e-05,
+      "loss": 0.1771,
+      "step": 5150
+    },
+    {
+      "grad_norm": 0.4809960424900055,
+      "learning_rate": 8.863609560848829e-05,
+      "loss": 0.1741,
+      "step": 5160
+    },
+    {
+      "grad_norm": 0.5211909413337708,
+      "learning_rate": 8.85835662566263e-05,
+      "loss": 0.1894,
+      "step": 5170
+    },
+    {
+      "grad_norm": 0.6504306793212891,
+      "learning_rate": 8.853093141876747e-05,
+      "loss": 0.1821,
+      "step": 5180
+    },
+    {
+      "grad_norm": 0.7522326111793518,
+      "learning_rate": 8.847819123881343e-05,
+      "loss": 0.1999,
+      "step": 5190
+    },
+    {
+      "grad_norm": 0.46731919050216675,
+      "learning_rate": 8.842534586095383e-05,
+      "loss": 0.1931,
+      "step": 5200
+    },
+    {
+      "grad_norm": 0.3652777373790741,
+      "learning_rate": 8.837239542966593e-05,
+      "loss": 0.1821,
+      "step": 5210
+    },
+    {
+      "grad_norm": 0.541904866695404,
+      "learning_rate": 8.831934008971417e-05,
+      "loss": 0.1789,
+      "step": 5220
+    },
+    {
+      "grad_norm": 0.5020244717597961,
+      "learning_rate": 8.826617998614982e-05,
+      "loss": 0.1774,
+      "step": 5230
+    },
+    {
+      "grad_norm": 0.4835430085659027,
+      "learning_rate": 8.821291526431056e-05,
+      "loss": 0.1699,
+      "step": 5240
+    },
+    {
+      "grad_norm": 0.4528132975101471,
+      "learning_rate": 8.815954606982015e-05,
+      "loss": 0.1873,
+      "step": 5250
+    },
+    {
+      "grad_norm": 0.44556108117103577,
+      "learning_rate": 8.810607254858789e-05,
+      "loss": 0.1692,
+      "step": 5260
+    },
+    {
+      "grad_norm": 0.546000063419342,
+      "learning_rate": 8.805249484680838e-05,
+      "loss": 0.1794,
+      "step": 5270
+    },
+    {
+      "grad_norm": 0.5679325461387634,
+      "learning_rate": 8.799881311096096e-05,
+      "loss": 0.1595,
+      "step": 5280
+    },
+    {
+      "grad_norm": 0.5736382603645325,
+      "learning_rate": 8.794502748780949e-05,
+      "loss": 0.2009,
+      "step": 5290
+    },
+    {
+      "grad_norm": 0.3857686221599579,
+      "learning_rate": 8.78911381244018e-05,
+      "loss": 0.1838,
+      "step": 5300
+    },
+    {
+      "grad_norm": 0.645546555519104,
+      "learning_rate": 8.783714516806933e-05,
+      "loss": 0.1911,
+      "step": 5310
+    },
+    {
+      "grad_norm": 0.5154575705528259,
+      "learning_rate": 8.77830487664268e-05,
+      "loss": 0.1792,
+      "step": 5320
+    },
+    {
+      "grad_norm": 0.5259498357772827,
+      "learning_rate": 8.772884906737167e-05,
+      "loss": 0.1877,
+      "step": 5330
+    },
+    {
+      "grad_norm": 0.5271315574645996,
+      "learning_rate": 8.767454621908387e-05,
+      "loss": 0.1905,
+      "step": 5340
+    },
+    {
+      "grad_norm": 0.5458414554595947,
+      "learning_rate": 8.76201403700253e-05,
+      "loss": 0.2004,
+      "step": 5350
+    },
+    {
+      "grad_norm": 0.555850625038147,
+      "learning_rate": 8.756563166893949e-05,
+      "loss": 0.181,
+      "step": 5360
+    },
+    {
+      "grad_norm": 0.47980746626853943,
+      "learning_rate": 8.751102026485113e-05,
+      "loss": 0.1791,
+      "step": 5370
+    },
+    {
+      "grad_norm": 0.504806637763977,
+      "learning_rate": 8.745630630706571e-05,
+      "loss": 0.1766,
+      "step": 5380
+    },
+    {
+      "grad_norm": 0.5069490075111389,
+      "learning_rate": 8.740148994516912e-05,
+      "loss": 0.1688,
+      "step": 5390
+    },
+    {
+      "grad_norm": 0.6437678337097168,
+      "learning_rate": 8.73465713290272e-05,
+      "loss": 0.1762,
+      "step": 5400
+    },
+    {
+      "grad_norm": 0.4785497486591339,
+      "learning_rate": 8.729155060878533e-05,
+      "loss": 0.1688,
+      "step": 5410
+    },
+    {
+      "grad_norm": 0.3648753762245178,
+      "learning_rate": 8.723642793486809e-05,
+      "loss": 0.1748,
+      "step": 5420
+    },
+    {
+      "grad_norm": 0.4988344609737396,
+      "learning_rate": 8.718120345797873e-05,
+      "loss": 0.1788,
+      "step": 5430
+    },
+    {
+      "grad_norm": 0.4616014361381531,
+      "learning_rate": 8.712587732909889e-05,
+      "loss": 0.1577,
+      "step": 5440
+    },
+    {
+      "grad_norm": 0.4555915594100952,
+      "learning_rate": 8.707044969948806e-05,
+      "loss": 0.1811,
+      "step": 5450
+    },
+    {
+      "grad_norm": 0.43113359808921814,
+      "learning_rate": 8.701492072068329e-05,
+      "loss": 0.1628,
+      "step": 5460
+    },
+    {
+      "grad_norm": 0.8870915770530701,
+      "learning_rate": 8.695929054449869e-05,
+      "loss": 0.1677,
+      "step": 5470
+    },
+    {
+      "grad_norm": 0.6327729821205139,
+      "learning_rate": 8.690355932302501e-05,
+      "loss": 0.1899,
+      "step": 5480
+    },
+    {
+      "grad_norm": 0.5786057114601135,
+      "learning_rate": 8.684772720862931e-05,
+      "loss": 0.1812,
+      "step": 5490
+    },
+    {
+      "grad_norm": 0.39201620221138,
+      "learning_rate": 8.679179435395446e-05,
+      "loss": 0.2254,
+      "step": 5500
+    },
+    {
+      "grad_norm": 0.521139919757843,
+      "learning_rate": 8.673576091191874e-05,
+      "loss": 0.1693,
+      "step": 5510
+    },
+    {
+      "grad_norm": 0.5388842821121216,
+      "learning_rate": 8.667962703571541e-05,
+      "loss": 0.2022,
+      "step": 5520
+    },
+    {
+      "grad_norm": 0.4477870464324951,
+      "learning_rate": 8.662339287881238e-05,
+      "loss": 0.1601,
+      "step": 5530
+    },
+    {
+      "grad_norm": 0.5324650406837463,
+      "learning_rate": 8.656705859495169e-05,
+      "loss": 0.1843,
+      "step": 5540
+    },
+    {
+      "grad_norm": 0.49471986293792725,
+      "learning_rate": 8.651062433814912e-05,
+      "loss": 0.2109,
+      "step": 5550
+    },
+    {
+      "grad_norm": 0.5062776803970337,
+      "learning_rate": 8.645409026269375e-05,
+      "loss": 0.1865,
+      "step": 5560
+    },
+    {
+      "grad_norm": 0.5051210522651672,
+      "learning_rate": 8.639745652314759e-05,
+      "loss": 0.1835,
+      "step": 5570
+    },
+    {
+      "grad_norm": 0.48859837651252747,
+      "learning_rate": 8.634072327434515e-05,
+      "loss": 0.1905,
+      "step": 5580
+    },
+    {
+      "grad_norm": 0.429341584444046,
+      "learning_rate": 8.628389067139294e-05,
+      "loss": 0.1856,
+      "step": 5590
+    },
+    {
+      "grad_norm": 0.6983324885368347,
+      "learning_rate": 8.622695886966911e-05,
+      "loss": 0.1696,
+      "step": 5600
+    },
+    {
+      "grad_norm": 0.4402853548526764,
+      "learning_rate": 8.616992802482308e-05,
+      "loss": 0.1745,
+      "step": 5610
+    },
+    {
+      "grad_norm": 0.3669653534889221,
+      "learning_rate": 8.611279829277496e-05,
+      "loss": 0.1752,
+      "step": 5620
+    },
+    {
+      "grad_norm": 0.5630952715873718,
+      "learning_rate": 8.605556982971528e-05,
+      "loss": 0.1701,
+      "step": 5630
+    },
+    {
+      "grad_norm": 0.598450779914856,
+      "learning_rate": 8.599824279210447e-05,
+      "loss": 0.1867,
+      "step": 5640
+    },
+    {
+      "grad_norm": 0.3196536898612976,
+      "learning_rate": 8.594081733667243e-05,
+      "loss": 0.1681,
+      "step": 5650
+    },
+    {
+      "grad_norm": 0.687271773815155,
+      "learning_rate": 8.58832936204182e-05,
+      "loss": 0.1879,
+      "step": 5660
+    },
+    {
+      "grad_norm": 0.5468834638595581,
+      "learning_rate": 8.582567180060942e-05,
+      "loss": 0.1794,
+      "step": 5670
+    },
+    {
+      "grad_norm": 0.3773702383041382,
+      "learning_rate": 8.576795203478194e-05,
+      "loss": 0.1636,
+      "step": 5680
+    },
+    {
+      "grad_norm": 0.6198897361755371,
+      "learning_rate": 8.571013448073939e-05,
+      "loss": 0.1757,
+      "step": 5690
+    },
+    {
+      "grad_norm": 0.6091999411582947,
+      "learning_rate": 8.565221929655275e-05,
+      "loss": 0.1633,
+      "step": 5700
+    },
+    {
+      "grad_norm": 0.6326218843460083,
+      "learning_rate": 8.559420664055992e-05,
+      "loss": 0.1947,
+      "step": 5710
+    },
+    {
+      "grad_norm": 0.7249149084091187,
+      "learning_rate": 8.553609667136532e-05,
+      "loss": 0.1974,
+      "step": 5720
+    },
+    {
+      "grad_norm": 0.5965058207511902,
+      "learning_rate": 8.547788954783936e-05,
+      "loss": 0.2183,
+      "step": 5730
+    },
+    {
+      "grad_norm": 0.5835154056549072,
+      "learning_rate": 8.541958542911808e-05,
+      "loss": 0.1723,
+      "step": 5740
+    },
+    {
+      "grad_norm": 0.48739296197891235,
+      "learning_rate": 8.536118447460275e-05,
+      "loss": 0.1641,
+      "step": 5750
+    },
+    {
+      "grad_norm": 0.40164250135421753,
+      "learning_rate": 8.530268684395932e-05,
+      "loss": 0.1778,
+      "step": 5760
+    },
+    {
+      "grad_norm": 0.5199743509292603,
+      "learning_rate": 8.524409269711807e-05,
+      "loss": 0.176,
+      "step": 5770
+    },
+    {
+      "grad_norm": 0.50117427110672,
+      "learning_rate": 8.51854021942732e-05,
+      "loss": 0.1855,
+      "step": 5780
+    },
+    {
+      "grad_norm": 0.7282928228378296,
+      "learning_rate": 8.512661549588227e-05,
+      "loss": 0.1899,
+      "step": 5790
+    },
+    {
+      "grad_norm": 0.686956524848938,
+      "learning_rate": 8.506773276266588e-05,
+      "loss": 0.1777,
+      "step": 5800
+    },
+    {
+      "grad_norm": 0.5173795819282532,
+      "learning_rate": 8.500875415560721e-05,
+      "loss": 0.2067,
+      "step": 5810
+    },
+    {
+      "grad_norm": 0.4220134913921356,
+      "learning_rate": 8.494967983595144e-05,
+      "loss": 0.1712,
+      "step": 5820
+    },
+    {
+      "grad_norm": 0.5839557647705078,
+      "learning_rate": 8.489050996520558e-05,
+      "loss": 0.1603,
+      "step": 5830
+    },
+    {
+      "grad_norm": 0.38393619656562805,
+      "learning_rate": 8.483124470513775e-05,
+      "loss": 0.1801,
+      "step": 5840
+    },
+    {
+      "grad_norm": 0.5402277112007141,
+      "learning_rate": 8.477188421777692e-05,
+      "loss": 0.1715,
+      "step": 5850
+    },
+    {
+      "grad_norm": 0.5864099860191345,
+      "learning_rate": 8.47124286654124e-05,
+      "loss": 0.1844,
+      "step": 5860
+    },
+    {
+      "grad_norm": 0.38726454973220825,
+      "learning_rate": 8.465287821059341e-05,
+      "loss": 0.1863,
+      "step": 5870
+    },
+    {
+      "grad_norm": 0.6337236762046814,
+      "learning_rate": 8.45932330161286e-05,
+      "loss": 0.1741,
+      "step": 5880
+    },
+    {
+      "grad_norm": 0.43250012397766113,
+      "learning_rate": 8.453349324508567e-05,
+      "loss": 0.1795,
+      "step": 5890
+    },
+    {
+      "grad_norm": 0.4237213730812073,
+      "learning_rate": 8.447365906079088e-05,
+      "loss": 0.1868,
+      "step": 5900
+    },
+    {
+      "grad_norm": 0.6180346608161926,
+      "learning_rate": 8.441373062682856e-05,
+      "loss": 0.1882,
+      "step": 5910
+    },
+    {
+      "grad_norm": 0.4483833909034729,
+      "learning_rate": 8.43537081070408e-05,
+      "loss": 0.1888,
+      "step": 5920
+    },
+    {
+      "grad_norm": 0.6591970920562744,
+      "learning_rate": 8.429359166552689e-05,
+      "loss": 0.1726,
+      "step": 5930
+    },
+    {
+      "grad_norm": 0.3937123119831085,
+      "learning_rate": 8.423338146664284e-05,
+      "loss": 0.1895,
+      "step": 5940
+    },
+    {
+      "grad_norm": 0.5331619381904602,
+      "learning_rate": 8.417307767500107e-05,
+      "loss": 0.1843,
+      "step": 5950
+    },
+    {
+      "grad_norm": 0.6063985824584961,
+      "learning_rate": 8.411268045546983e-05,
+      "loss": 0.1871,
+      "step": 5960
+    },
+    {
+      "grad_norm": 0.8219554424285889,
+      "learning_rate": 8.405218997317281e-05,
+      "loss": 0.1817,
+      "step": 5970
+    },
+    {
+      "grad_norm": 0.4364663064479828,
+      "learning_rate": 8.399160639348869e-05,
+      "loss": 0.176,
+      "step": 5980
+    },
+    {
+      "grad_norm": 0.751763641834259,
+      "learning_rate": 8.393092988205065e-05,
+      "loss": 0.1687,
+      "step": 5990
+    },
+    {
+      "grad_norm": 0.3435507118701935,
+      "learning_rate": 8.387016060474597e-05,
+      "loss": 0.1576,
+      "step": 6000
+    },
+    {
+      "grad_norm": 0.4536468982696533,
+      "learning_rate": 8.380929872771551e-05,
+      "loss": 0.177,
+      "step": 6010
+    },
+    {
+      "grad_norm": 0.5580184459686279,
+      "learning_rate": 8.374834441735335e-05,
+      "loss": 0.1723,
+      "step": 6020
+    },
+    {
+      "grad_norm": 0.8800753951072693,
+      "learning_rate": 8.368729784030622e-05,
+      "loss": 0.165,
+      "step": 6030
+    },
+    {
+      "grad_norm": 0.6233548521995544,
+      "learning_rate": 8.362615916347315e-05,
+      "loss": 0.1712,
+      "step": 6040
+    },
+    {
+      "grad_norm": 0.6393119096755981,
+      "learning_rate": 8.356492855400493e-05,
+      "loss": 0.1905,
+      "step": 6050
+    },
+    {
+      "grad_norm": 0.4487166404724121,
+      "learning_rate": 8.350360617930371e-05,
+      "loss": 0.1815,
+      "step": 6060
+    },
+    {
+      "grad_norm": 0.5289573669433594,
+      "learning_rate": 8.344219220702255e-05,
+      "loss": 0.1619,
+      "step": 6070
+    },
+    {
+      "grad_norm": 0.47635534405708313,
+      "learning_rate": 8.338068680506485e-05,
+      "loss": 0.1863,
+      "step": 6080
+    },
+    {
+      "grad_norm": 0.7369346022605896,
+      "learning_rate": 8.33190901415841e-05,
+      "loss": 0.1821,
+      "step": 6090
+    },
+    {
+      "grad_norm": 0.6643389463424683,
+      "learning_rate": 8.325740238498317e-05,
+      "loss": 0.1747,
+      "step": 6100
+    },
+    {
+      "grad_norm": 0.459866464138031,
+      "learning_rate": 8.319562370391406e-05,
+      "loss": 0.1725,
+      "step": 6110
+    },
+    {
+      "grad_norm": 0.46379023790359497,
+      "learning_rate": 8.31337542672773e-05,
+      "loss": 0.1777,
+      "step": 6120
+    },
+    {
+      "grad_norm": 0.5364670157432556,
+      "learning_rate": 8.307179424422158e-05,
+      "loss": 0.1862,
+      "step": 6130
+    },
+    {
+      "grad_norm": 0.4546237289905548,
+      "learning_rate": 8.300974380414327e-05,
+      "loss": 0.1794,
+      "step": 6140
+    },
+    {
+      "grad_norm": 0.44631871581077576,
+      "learning_rate": 8.294760311668586e-05,
+      "loss": 0.1814,
+      "step": 6150
+    },
+    {
+      "grad_norm": 0.3880269229412079,
+      "learning_rate": 8.288537235173961e-05,
+      "loss": 0.1686,
+      "step": 6160
+    },
+    {
+      "grad_norm": 0.4075128436088562,
+      "learning_rate": 8.282305167944108e-05,
+      "loss": 0.1758,
+      "step": 6170
+    },
+    {
+      "grad_norm": 0.6632016897201538,
+      "learning_rate": 8.276064127017262e-05,
+      "loss": 0.1995,
+      "step": 6180
+    },
+    {
+      "grad_norm": 0.6579618453979492,
+      "learning_rate": 8.269814129456189e-05,
+      "loss": 0.2047,
+      "step": 6190
+    },
+    {
+      "grad_norm": 0.421772301197052,
+      "learning_rate": 8.263555192348143e-05,
+      "loss": 0.18,
+      "step": 6200
+    },
+    {
+      "grad_norm": 0.6424674987792969,
+      "learning_rate": 8.257287332804819e-05,
+      "loss": 0.1965,
+      "step": 6210
+    },
+    {
+      "grad_norm": 0.3565591275691986,
+      "learning_rate": 8.251010567962307e-05,
+      "loss": 0.1646,
+      "step": 6220
+    },
+    {
+      "grad_norm": 0.4164339601993561,
+      "learning_rate": 8.244724914981041e-05,
+      "loss": 0.1981,
+      "step": 6230
+    },
+    {
+      "grad_norm": 0.4323714077472687,
+      "learning_rate": 8.238430391045757e-05,
+      "loss": 0.1613,
+      "step": 6240
+    },
+    {
+      "grad_norm": 0.39576658606529236,
+      "learning_rate": 8.232127013365445e-05,
+      "loss": 0.1789,
+      "step": 6250
+    },
+    {
+      "grad_norm": 0.4295504689216614,
+      "learning_rate": 8.225814799173295e-05,
+      "loss": 0.1916,
+      "step": 6260
+    },
+    {
+      "grad_norm": 0.4853602945804596,
+      "learning_rate": 8.219493765726663e-05,
+      "loss": 0.2014,
+      "step": 6270
+    },
+    {
+      "grad_norm": 0.3629939556121826,
+      "learning_rate": 8.21316393030701e-05,
+      "loss": 0.1832,
+      "step": 6280
+    },
+    {
+      "grad_norm": 0.5434958934783936,
+      "learning_rate": 8.206825310219865e-05,
+      "loss": 0.1923,
+      "step": 6290
+    },
+    {
+      "grad_norm": 0.36115556955337524,
+      "learning_rate": 8.200477922794776e-05,
+      "loss": 0.1999,
+      "step": 6300
+    },
+    {
+      "grad_norm": 0.6021917462348938,
+      "learning_rate": 8.194121785385256e-05,
+      "loss": 0.1673,
+      "step": 6310
+    },
+    {
+      "grad_norm": 0.4855813682079315,
+      "learning_rate": 8.187756915368741e-05,
+      "loss": 0.1879,
+      "step": 6320
+    },
+    {
+      "grad_norm": 0.475492924451828,
+      "learning_rate": 8.181383330146544e-05,
+      "loss": 0.1908,
+      "step": 6330
+    },
+    {
+      "grad_norm": 0.38350939750671387,
+      "learning_rate": 8.175001047143804e-05,
+      "loss": 0.1964,
+      "step": 6340
+    },
+    {
+      "grad_norm": 0.5156674385070801,
+      "learning_rate": 8.168610083809438e-05,
+      "loss": 0.1821,
+      "step": 6350
+    },
+    {
+      "grad_norm": 0.676490306854248,
+      "learning_rate": 8.162210457616095e-05,
+      "loss": 0.1796,
+      "step": 6360
+    },
+    {
+      "grad_norm": 0.6172734498977661,
+      "learning_rate": 8.155802186060109e-05,
+      "loss": 0.1765,
+      "step": 6370
+    },
+    {
+      "grad_norm": 0.5570704936981201,
+      "learning_rate": 8.149385286661453e-05,
+      "loss": 0.1976,
+      "step": 6380
+    },
+    {
+      "grad_norm": 0.43670567870140076,
+      "learning_rate": 8.14295977696368e-05,
+      "loss": 0.1727,
+      "step": 6390
+    },
+    {
+      "grad_norm": 0.5615594983100891,
+      "learning_rate": 8.13652567453389e-05,
+      "loss": 0.2009,
+      "step": 6400
+    },
+    {
+      "grad_norm": 0.4178938865661621,
+      "learning_rate": 8.130082996962676e-05,
+      "loss": 0.2155,
+      "step": 6410
+    },
+    {
+      "grad_norm": 0.4542387127876282,
+      "learning_rate": 8.123631761864068e-05,
+      "loss": 0.1823,
+      "step": 6420
+    },
+    {
+      "grad_norm": 0.43628746271133423,
+      "learning_rate": 8.1171719868755e-05,
+      "loss": 0.1899,
+      "step": 6430
+    },
+    {
+      "grad_norm": 0.7109262943267822,
+      "learning_rate": 8.110703689657748e-05,
+      "loss": 0.1605,
+      "step": 6440
+    },
+    {
+      "grad_norm": 0.5329083800315857,
+      "learning_rate": 8.104226887894892e-05,
+      "loss": 0.1848,
+      "step": 6450
+    },
+    {
+      "grad_norm": 0.5122144818305969,
+      "learning_rate": 8.097741599294257e-05,
+      "loss": 0.2029,
+      "step": 6460
+    },
+    {
+      "grad_norm": 0.3079993724822998,
+      "learning_rate": 8.091247841586378e-05,
+      "loss": 0.1772,
+      "step": 6470
+    },
+    {
+      "grad_norm": 0.5121235847473145,
+      "learning_rate": 8.084745632524939e-05,
+      "loss": 0.1749,
+      "step": 6480
+    },
+    {
+      "grad_norm": 0.5712058544158936,
+      "learning_rate": 8.07823498988673e-05,
+      "loss": 0.2039,
+      "step": 6490
+    },
+    {
+      "grad_norm": 0.483211874961853,
+      "learning_rate": 8.071715931471602e-05,
+      "loss": 0.1947,
+      "step": 6500
+    },
+    {
+      "grad_norm": 0.4779788553714752,
+      "learning_rate": 8.06518847510241e-05,
+      "loss": 0.1948,
+      "step": 6510
+    },
+    {
+      "grad_norm": 0.396277517080307,
+      "learning_rate": 8.058652638624971e-05,
+      "loss": 0.1778,
+      "step": 6520
+    },
+    {
+      "grad_norm": 0.513851523399353,
+      "learning_rate": 8.052108439908013e-05,
+      "loss": 0.1939,
+      "step": 6530
+    },
+    {
+      "grad_norm": 0.39212316274642944,
+      "learning_rate": 8.045555896843125e-05,
+      "loss": 0.1881,
+      "step": 6540
+    },
+    {
+      "grad_norm": 0.43922436237335205,
+      "learning_rate": 8.03899502734471e-05,
+      "loss": 0.1927,
+      "step": 6550
+    },
+    {
+      "grad_norm": 0.41773778200149536,
+      "learning_rate": 8.032425849349931e-05,
+      "loss": 0.1815,
+      "step": 6560
+    },
+    {
+      "grad_norm": 0.47995758056640625,
+      "learning_rate": 8.025848380818674e-05,
+      "loss": 0.1763,
+      "step": 6570
+    },
+    {
+      "grad_norm": 0.4474988877773285,
+      "learning_rate": 8.019262639733487e-05,
+      "loss": 0.1746,
+      "step": 6580
+    },
+    {
+      "grad_norm": 0.5823827385902405,
+      "learning_rate": 8.012668644099531e-05,
+      "loss": 0.172,
+      "step": 6590
+    },
+    {
+      "grad_norm": 0.4172171950340271,
+      "learning_rate": 8.006066411944542e-05,
+      "loss": 0.1844,
+      "step": 6600
+    },
+    {
+      "grad_norm": 0.5686772465705872,
+      "learning_rate": 7.999455961318769e-05,
+      "loss": 0.1712,
+      "step": 6610
+    },
+    {
+      "grad_norm": 0.49165746569633484,
+      "learning_rate": 7.992837310294932e-05,
+      "loss": 0.1868,
+      "step": 6620
+    },
+    {
+      "grad_norm": 0.40783655643463135,
+      "learning_rate": 7.986210476968167e-05,
+      "loss": 0.1713,
+      "step": 6630
+    },
+    {
+      "grad_norm": 0.347829669713974,
+      "learning_rate": 7.97957547945599e-05,
+      "loss": 0.1573,
+      "step": 6640
+    },
+    {
+      "grad_norm": 0.37543681263923645,
+      "learning_rate": 7.972932335898226e-05,
+      "loss": 0.1685,
+      "step": 6650
+    },
+    {
+      "grad_norm": 0.4008532464504242,
+      "learning_rate": 7.966281064456975e-05,
+      "loss": 0.1706,
+      "step": 6660
+    },
+    {
+      "grad_norm": 0.5448466539382935,
+      "learning_rate": 7.959621683316563e-05,
+      "loss": 0.1563,
+      "step": 6670
+    },
+    {
+      "grad_norm": 0.3708697259426117,
+      "learning_rate": 7.952954210683481e-05,
+      "loss": 0.1664,
+      "step": 6680
+    },
+    {
+      "grad_norm": 0.4621528387069702,
+      "learning_rate": 7.946278664786345e-05,
+      "loss": 0.1897,
+      "step": 6690
+    },
+    {
+      "grad_norm": 0.4081062376499176,
+      "learning_rate": 7.939595063875842e-05,
+      "loss": 0.1639,
+      "step": 6700
+    },
+    {
+      "grad_norm": 0.38687703013420105,
+      "learning_rate": 7.932903426224683e-05,
+      "loss": 0.1777,
+      "step": 6710
+    },
+    {
+      "grad_norm": 0.5351437926292419,
+      "learning_rate": 7.926203770127552e-05,
+      "loss": 0.1742,
+      "step": 6720
+    },
+    {
+      "grad_norm": 0.3460598886013031,
+      "learning_rate": 7.919496113901046e-05,
+      "loss": 0.1792,
+      "step": 6730
+    },
+    {
+      "grad_norm": 0.6417859196662903,
+      "learning_rate": 7.912780475883649e-05,
+      "loss": 0.1651,
+      "step": 6740
+    },
+    {
+      "grad_norm": 0.3893650472164154,
+      "learning_rate": 7.906056874435652e-05,
+      "loss": 0.1706,
+      "step": 6750
+    },
+    {
+      "grad_norm": 0.39047351479530334,
+      "learning_rate": 7.899325327939131e-05,
+      "loss": 0.1952,
+      "step": 6760
+    },
+    {
+      "grad_norm": 0.44595035910606384,
+      "learning_rate": 7.892585854797872e-05,
+      "loss": 0.189,
+      "step": 6770
+    },
+    {
+      "grad_norm": 0.4414966106414795,
+      "learning_rate": 7.88583847343734e-05,
+      "loss": 0.1649,
+      "step": 6780
+    },
+    {
+      "grad_norm": 0.3749309480190277,
+      "learning_rate": 7.879083202304616e-05,
+      "loss": 0.1654,
+      "step": 6790
+    },
+    {
+      "grad_norm": 0.5494127869606018,
+      "learning_rate": 7.872320059868355e-05,
+      "loss": 0.1588,
+      "step": 6800
+    },
+    {
+      "grad_norm": 0.8617106080055237,
+      "learning_rate": 7.865549064618729e-05,
+      "loss": 0.1818,
+      "step": 6810
+    },
+    {
+      "grad_norm": 0.5497077703475952,
+      "learning_rate": 7.858770235067381e-05,
+      "loss": 0.1604,
+      "step": 6820
+    },
+    {
+      "grad_norm": 0.41114872694015503,
+      "learning_rate": 7.851983589747374e-05,
+      "loss": 0.195,
+      "step": 6830
+    },
+    {
+      "grad_norm": 0.38935649394989014,
+      "learning_rate": 7.845189147213133e-05,
+      "loss": 0.1629,
+      "step": 6840
+    },
+    {
+      "grad_norm": 0.5953046679496765,
+      "learning_rate": 7.838386926040407e-05,
+      "loss": 0.192,
+      "step": 6850
+    },
+    {
+      "grad_norm": 0.5348408222198486,
+      "learning_rate": 7.83157694482621e-05,
+      "loss": 0.1746,
+      "step": 6860
+    },
+    {
+      "grad_norm": 0.5624399185180664,
+      "learning_rate": 7.824759222188768e-05,
+      "loss": 0.1753,
+      "step": 6870
+    },
+    {
+      "grad_norm": 0.29335105419158936,
+      "learning_rate": 7.817933776767478e-05,
+      "loss": 0.1697,
+      "step": 6880
+    },
+    {
+      "grad_norm": 0.5341830849647522,
+      "learning_rate": 7.811100627222842e-05,
+      "loss": 0.1824,
+      "step": 6890
+    },
+    {
+      "grad_norm": 0.5268265008926392,
+      "learning_rate": 7.804259792236435e-05,
+      "loss": 0.1606,
+      "step": 6900
+    },
+    {
+      "grad_norm": 0.5236201286315918,
+      "learning_rate": 7.797411290510835e-05,
+      "loss": 0.1817,
+      "step": 6910
+    },
+    {
+      "grad_norm": 0.4060767590999603,
+      "learning_rate": 7.790555140769586e-05,
+      "loss": 0.1691,
+      "step": 6920
+    },
+    {
+      "grad_norm": 0.4897361099720001,
+      "learning_rate": 7.78369136175714e-05,
+      "loss": 0.1776,
+      "step": 6930
+    },
+    {
+      "grad_norm": 0.5901176929473877,
+      "learning_rate": 7.776819972238806e-05,
+      "loss": 0.1488,
+      "step": 6940
+    },
+    {
+      "grad_norm": 0.25817981362342834,
+      "learning_rate": 7.7699409910007e-05,
+      "loss": 0.1809,
+      "step": 6950
+    },
+    {
+      "grad_norm": 0.579413890838623,
+      "learning_rate": 7.763054436849694e-05,
+      "loss": 0.174,
+      "step": 6960
+    },
+    {
+      "grad_norm": 0.3552204668521881,
+      "learning_rate": 7.756160328613364e-05,
+      "loss": 0.1683,
+      "step": 6970
+    },
+    {
+      "grad_norm": 0.4387410879135132,
+      "learning_rate": 7.749258685139942e-05,
+      "loss": 0.1861,
+      "step": 6980
+    },
+    {
+      "grad_norm": 0.4106725752353668,
+      "learning_rate": 7.742349525298253e-05,
+      "loss": 0.1792,
+      "step": 6990
+    },
+    {
+      "grad_norm": 0.5252473950386047,
+      "learning_rate": 7.735432867977679e-05,
+      "loss": 0.1908,
+      "step": 7000
+    },
+    {
+      "grad_norm": 0.43396705389022827,
+      "learning_rate": 7.728508732088096e-05,
+      "loss": 0.1783,
+      "step": 7010
+    },
+    {
+      "grad_norm": 0.4027872681617737,
+      "learning_rate": 7.721577136559825e-05,
+      "loss": 0.1788,
+      "step": 7020
+    },
+    {
+      "grad_norm": 0.34274473786354065,
+      "learning_rate": 7.714638100343588e-05,
+      "loss": 0.1662,
+      "step": 7030
+    },
+    {
+      "grad_norm": 0.5517646670341492,
+      "learning_rate": 7.707691642410444e-05,
+      "loss": 0.1741,
+      "step": 7040
+    },
+    {
+      "grad_norm": 0.3831420838832855,
+      "learning_rate": 7.70073778175174e-05,
+      "loss": 0.1826,
+      "step": 7050
+    },
+    {
+      "grad_norm": 0.43620720505714417,
+      "learning_rate": 7.69377653737907e-05,
+      "loss": 0.1664,
+      "step": 7060
+    },
+    {
+      "grad_norm": 0.4326232075691223,
+      "learning_rate": 7.686807928324209e-05,
+      "loss": 0.1682,
+      "step": 7070
+    },
+    {
+      "grad_norm": 0.49154457449913025,
+      "learning_rate": 7.679831973639065e-05,
+      "loss": 0.16,
+      "step": 7080
+    },
+    {
+      "grad_norm": 0.498117059469223,
+      "learning_rate": 7.672848692395637e-05,
+      "loss": 0.1729,
+      "step": 7090
+    },
+    {
+      "grad_norm": 0.6177932620048523,
+      "learning_rate": 7.665858103685944e-05,
+      "loss": 0.1664,
+      "step": 7100
+    },
+    {
+      "grad_norm": 0.34048134088516235,
+      "learning_rate": 7.658860226621991e-05,
+      "loss": 0.1816,
+      "step": 7110
+    },
+    {
+      "grad_norm": 0.5639463067054749,
+      "learning_rate": 7.651855080335708e-05,
+      "loss": 0.17,
+      "step": 7120
+    },
+    {
+      "grad_norm": 0.5009056329727173,
+      "learning_rate": 7.644842683978896e-05,
+      "loss": 0.2003,
+      "step": 7130
+    },
+    {
+      "grad_norm": 0.4188900291919708,
+      "learning_rate": 7.63782305672318e-05,
+      "loss": 0.1502,
+      "step": 7140
+    },
+    {
+      "grad_norm": 0.5396490693092346,
+      "learning_rate": 7.63079621775995e-05,
+      "loss": 0.1713,
+      "step": 7150
+    },
+    {
+      "grad_norm": 0.5339280962944031,
+      "learning_rate": 7.623762186300319e-05,
+      "loss": 0.1882,
+      "step": 7160
+    },
+    {
+      "grad_norm": 0.5875482559204102,
+      "learning_rate": 7.616720981575057e-05,
+      "loss": 0.187,
+      "step": 7170
+    },
+    {
+      "grad_norm": 0.6965053081512451,
+      "learning_rate": 7.609672622834552e-05,
+      "loss": 0.1889,
+      "step": 7180
+    },
+    {
+      "grad_norm": 0.5891492366790771,
+      "learning_rate": 7.602617129348747e-05,
+      "loss": 0.1601,
+      "step": 7190
+    },
+    {
+      "grad_norm": 0.537082850933075,
+      "learning_rate": 7.595554520407088e-05,
+      "loss": 0.1761,
+      "step": 7200
+    },
+    {
+      "grad_norm": 0.5850155353546143,
+      "learning_rate": 7.588484815318484e-05,
+      "loss": 0.1764,
+      "step": 7210
+    },
+    {
+      "grad_norm": 0.7694187164306641,
+      "learning_rate": 7.581408033411234e-05,
+      "loss": 0.1718,
+      "step": 7220
+    },
+    {
+      "grad_norm": 0.533075749874115,
+      "learning_rate": 7.574324194032995e-05,
+      "loss": 0.1933,
+      "step": 7230
+    },
+    {
+      "grad_norm": 0.684154212474823,
+      "learning_rate": 7.567233316550705e-05,
+      "loss": 0.1606,
+      "step": 7240
+    },
+    {
+      "grad_norm": 0.45348629355430603,
+      "learning_rate": 7.560135420350562e-05,
+      "loss": 0.161,
+      "step": 7250
+    },
+    {
+      "grad_norm": 0.587451696395874,
+      "learning_rate": 7.553030524837935e-05,
+      "loss": 0.1784,
+      "step": 7260
+    },
+    {
+      "grad_norm": 0.4856109619140625,
+      "learning_rate": 7.545918649437341e-05,
+      "loss": 0.2089,
+      "step": 7270
+    },
+    {
+      "grad_norm": 0.4258940517902374,
+      "learning_rate": 7.538799813592377e-05,
+      "loss": 0.1714,
+      "step": 7280
+    },
+    {
+      "grad_norm": 0.4969996511936188,
+      "learning_rate": 7.531674036765662e-05,
+      "loss": 0.1758,
+      "step": 7290
+    },
+    {
+      "grad_norm": 0.35997462272644043,
+      "learning_rate": 7.524541338438807e-05,
+      "loss": 0.1732,
+      "step": 7300
+    },
+    {
+      "grad_norm": 0.5294685959815979,
+      "learning_rate": 7.517401738112328e-05,
+      "loss": 0.1659,
+      "step": 7310
+    },
+    {
+      "grad_norm": 0.42113953828811646,
+      "learning_rate": 7.510255255305628e-05,
+      "loss": 0.1863,
+      "step": 7320
+    },
+    {
+      "grad_norm": 0.46128571033477783,
+      "learning_rate": 7.503101909556911e-05,
+      "loss": 0.1619,
+      "step": 7330
+    },
+    {
+      "grad_norm": 0.548859179019928,
+      "learning_rate": 7.495941720423154e-05,
+      "loss": 0.1545,
+      "step": 7340
+    },
+    {
+      "grad_norm": 0.7683234214782715,
+      "learning_rate": 7.488774707480042e-05,
+      "loss": 0.1792,
+      "step": 7350
+    },
+    {
+      "grad_norm": 0.40643322467803955,
+      "learning_rate": 7.481600890321911e-05,
+      "loss": 0.1779,
+      "step": 7360
+    },
+    {
+      "grad_norm": 0.36510908603668213,
+      "learning_rate": 7.474420288561708e-05,
+      "loss": 0.1621,
+      "step": 7370
+    },
+    {
+      "grad_norm": 0.5000161528587341,
+      "learning_rate": 7.467232921830921e-05,
+      "loss": 0.1749,
+      "step": 7380
+    },
+    {
+      "grad_norm": 0.5171788930892944,
+      "learning_rate": 7.460038809779537e-05,
+      "loss": 0.1814,
+      "step": 7390
+    },
+    {
+      "grad_norm": 0.6573072671890259,
+      "learning_rate": 7.452837972075983e-05,
+      "loss": 0.1836,
+      "step": 7400
+    },
+    {
+      "grad_norm": 0.5616627931594849,
+      "learning_rate": 7.445630428407074e-05,
+      "loss": 0.1659,
+      "step": 7410
+    },
+    {
+      "grad_norm": 0.4225040674209595,
+      "learning_rate": 7.43841619847796e-05,
+      "loss": 0.166,
+      "step": 7420
+    },
+    {
+      "grad_norm": 0.4441474676132202,
+      "learning_rate": 7.431195302012072e-05,
+      "loss": 0.151,
+      "step": 7430
+    },
+    {
+      "grad_norm": 0.48102179169654846,
+      "learning_rate": 7.423967758751061e-05,
+      "loss": 0.1957,
+      "step": 7440
+    },
+    {
+      "grad_norm": 0.5136003494262695,
+      "learning_rate": 7.416733588454758e-05,
+      "loss": 0.1557,
+      "step": 7450
+    },
+    {
+      "grad_norm": 0.6044415235519409,
+      "learning_rate": 7.409492810901106e-05,
+      "loss": 0.1832,
+      "step": 7460
+    },
+    {
+      "grad_norm": 0.4371381402015686,
+      "learning_rate": 7.402245445886116e-05,
+      "loss": 0.1629,
+      "step": 7470
+    },
+    {
+      "grad_norm": 0.4242047965526581,
+      "learning_rate": 7.394991513223806e-05,
+      "loss": 0.1684,
+      "step": 7480
+    },
+    {
+      "grad_norm": 0.9678165912628174,
+      "learning_rate": 7.38773103274615e-05,
+      "loss": 0.1593,
+      "step": 7490
+    },
+    {
+      "grad_norm": 0.39337942004203796,
+      "learning_rate": 7.380464024303028e-05,
+      "loss": 0.1672,
+      "step": 7500
+    },
+    {
+      "grad_norm": 0.553438663482666,
+      "learning_rate": 7.373190507762162e-05,
+      "loss": 0.1635,
+      "step": 7510
+    },
+    {
+      "grad_norm": 0.5425419807434082,
+      "learning_rate": 7.365910503009066e-05,
+      "loss": 0.2008,
+      "step": 7520
+    },
+    {
+      "grad_norm": 0.5195794105529785,
+      "learning_rate": 7.358624029946996e-05,
+      "loss": 0.1716,
+      "step": 7530
+    },
+    {
+      "grad_norm": 0.49776509404182434,
+      "learning_rate": 7.351331108496893e-05,
+      "loss": 0.1549,
+      "step": 7540
+    },
+    {
+      "grad_norm": 0.5533013343811035,
+      "learning_rate": 7.344031758597325e-05,
+      "loss": 0.1748,
+      "step": 7550
+    },
+    {
+      "grad_norm": 0.5506358742713928,
+      "learning_rate": 7.336726000204435e-05,
+      "loss": 0.1759,
+      "step": 7560
+    },
+    {
+      "grad_norm": 0.42112860083580017,
+      "learning_rate": 7.32941385329189e-05,
+      "loss": 0.1626,
+      "step": 7570
+    },
+    {
+      "grad_norm": 1.3983711004257202,
+      "learning_rate": 7.322095337850816e-05,
+      "loss": 0.1704,
+      "step": 7580
+    },
+    {
+      "grad_norm": 0.47714993357658386,
+      "learning_rate": 7.314770473889758e-05,
+      "loss": 0.1684,
+      "step": 7590
+    },
+    {
+      "grad_norm": 0.40627628564834595,
+      "learning_rate": 7.307439281434615e-05,
+      "loss": 0.1886,
+      "step": 7600
+    },
+    {
+      "grad_norm": 0.6334792971611023,
+      "learning_rate": 7.300101780528585e-05,
+      "loss": 0.1664,
+      "step": 7610
+    },
+    {
+      "grad_norm": 0.4064364731311798,
+      "learning_rate": 7.292757991232117e-05,
+      "loss": 0.1968,
+      "step": 7620
+    },
+    {
+      "grad_norm": 0.4562758803367615,
+      "learning_rate": 7.285407933622848e-05,
+      "loss": 0.1813,
+      "step": 7630
+    },
+    {
+      "grad_norm": 0.4465484321117401,
+      "learning_rate": 7.278051627795557e-05,
+      "loss": 0.1923,
+      "step": 7640
+    },
+    {
+      "grad_norm": 0.40902653336524963,
+      "learning_rate": 7.270689093862105e-05,
+      "loss": 0.1606,
+      "step": 7650
+    },
+    {
+      "grad_norm": 0.3922201693058014,
+      "learning_rate": 7.263320351951374e-05,
+      "loss": 0.165,
+      "step": 7660
+    },
+    {
+      "grad_norm": 0.5342893004417419,
+      "learning_rate": 7.255945422209227e-05,
+      "loss": 0.1728,
+      "step": 7670
+    },
+    {
+      "grad_norm": 0.49451762437820435,
+      "learning_rate": 7.248564324798437e-05,
+      "loss": 0.1876,
+      "step": 7680
+    },
+    {
+      "grad_norm": 0.7057358622550964,
+      "learning_rate": 7.241177079898644e-05,
+      "loss": 0.1684,
+      "step": 7690
+    },
+    {
+      "grad_norm": 0.5408640503883362,
+      "learning_rate": 7.233783707706295e-05,
+      "loss": 0.1705,
+      "step": 7700
+    },
+    {
+      "grad_norm": 0.5592444539070129,
+      "learning_rate": 7.226384228434586e-05,
+      "loss": 0.178,
+      "step": 7710
+    },
+    {
+      "grad_norm": 0.670231282711029,
+      "learning_rate": 7.21897866231341e-05,
+      "loss": 0.1646,
+      "step": 7720
+    },
+    {
+      "grad_norm": 0.5700432062149048,
+      "learning_rate": 7.211567029589303e-05,
+      "loss": 0.1743,
+      "step": 7730
+    },
+    {
+      "grad_norm": 0.4519805312156677,
+      "learning_rate": 7.204149350525387e-05,
+      "loss": 0.196,
+      "step": 7740
+    },
+    {
+      "grad_norm": 0.6240270137786865,
+      "learning_rate": 7.196725645401309e-05,
+      "loss": 0.1846,
+      "step": 7750
+    },
+    {
+      "grad_norm": 0.5212076902389526,
+      "learning_rate": 7.1892959345132e-05,
+      "loss": 0.2078,
+      "step": 7760
+    },
+    {
+      "grad_norm": 0.6497604250907898,
+      "learning_rate": 7.181860238173605e-05,
+      "loss": 0.1773,
+      "step": 7770
+    },
+    {
+      "grad_norm": 0.3528762757778168,
+      "learning_rate": 7.174418576711432e-05,
+      "loss": 0.1736,
+      "step": 7780
+    },
+    {
+      "grad_norm": 0.4148101806640625,
+      "learning_rate": 7.1669709704719e-05,
+      "loss": 0.1822,
+      "step": 7790
+    },
+    {
+      "grad_norm": 0.5849226117134094,
+      "learning_rate": 7.159517439816481e-05,
+      "loss": 0.1858,
+      "step": 7800
+    },
+    {
+      "grad_norm": 0.40431728959083557,
+      "learning_rate": 7.152058005122842e-05,
+      "loss": 0.1695,
+      "step": 7810
+    },
+    {
+      "grad_norm": 0.4626309871673584,
+      "learning_rate": 7.144592686784793e-05,
+      "loss": 0.1851,
+      "step": 7820
+    },
+    {
+      "grad_norm": 0.5541772246360779,
+      "learning_rate": 7.137121505212229e-05,
+      "loss": 0.1778,
+      "step": 7830
+    },
+    {
+      "grad_norm": 0.7931815385818481,
+      "learning_rate": 7.129644480831077e-05,
+      "loss": 0.1926,
+      "step": 7840
+    },
+    {
+      "grad_norm": 0.5193843245506287,
+      "learning_rate": 7.122161634083234e-05,
+      "loss": 0.1791,
+      "step": 7850
+    },
+    {
+      "grad_norm": 0.5702765583992004,
+      "learning_rate": 7.114672985426516e-05,
+      "loss": 0.1969,
+      "step": 7860
+    },
+    {
+      "grad_norm": 0.4512436091899872,
+      "learning_rate": 7.107178555334606e-05,
+      "loss": 0.1853,
+      "step": 7870
+    },
+    {
+      "grad_norm": 0.5971438884735107,
+      "learning_rate": 7.099678364296989e-05,
+      "loss": 0.1808,
+      "step": 7880
+    },
+    {
+      "grad_norm": 0.4496137499809265,
+      "learning_rate": 7.0921724328189e-05,
+      "loss": 0.1791,
+      "step": 7890
+    },
+    {
+      "grad_norm": 0.43689867854118347,
+      "learning_rate": 7.084660781421268e-05,
+      "loss": 0.1786,
+      "step": 7900
+    },
+    {
+      "grad_norm": 0.8746722340583801,
+      "learning_rate": 7.077143430640662e-05,
+      "loss": 0.1827,
+      "step": 7910
+    },
+    {
+      "grad_norm": 0.6381582617759705,
+      "learning_rate": 7.069620401029232e-05,
+      "loss": 0.174,
+      "step": 7920
+    },
+    {
+      "grad_norm": 0.50140380859375,
+      "learning_rate": 7.062091713154655e-05,
+      "loss": 0.1944,
+      "step": 7930
+    },
+    {
+      "grad_norm": 0.4641288220882416,
+      "learning_rate": 7.054557387600075e-05,
+      "loss": 0.1883,
+      "step": 7940
+    },
+    {
+      "grad_norm": 0.7470072507858276,
+      "learning_rate": 7.04701744496405e-05,
+      "loss": 0.177,
+      "step": 7950
+    },
+    {
+      "grad_norm": 0.4173056185245514,
+      "learning_rate": 7.039471905860495e-05,
+      "loss": 0.1811,
+      "step": 7960
+    },
+    {
+      "grad_norm": 0.618665337562561,
+      "learning_rate": 7.031920790918628e-05,
+      "loss": 0.1676,
+      "step": 7970
+    },
+    {
+      "grad_norm": 0.39125391840934753,
+      "learning_rate": 7.024364120782906e-05,
+      "loss": 0.1712,
+      "step": 7980
+    },
+    {
+      "grad_norm": 0.7238273620605469,
+      "learning_rate": 7.016801916112978e-05,
+      "loss": 0.1909,
+      "step": 7990
+    },
+    {
+      "grad_norm": 0.46883994340896606,
+      "learning_rate": 7.009234197583623e-05,
+      "loss": 0.1755,
+      "step": 8000
+    },
+    {
+      "grad_norm": 0.4243353307247162,
+      "learning_rate": 7.001660985884692e-05,
+      "loss": 0.1932,
+      "step": 8010
+    },
+    {
+      "grad_norm": 0.5187137722969055,
+      "learning_rate": 6.994082301721063e-05,
+      "loss": 0.1805,
+      "step": 8020
+    },
+    {
+      "grad_norm": 0.4431232511997223,
+      "learning_rate": 6.986498165812563e-05,
+      "loss": 0.1778,
+      "step": 8030
+    },
+    {
+      "grad_norm": 0.566923201084137,
+      "learning_rate": 6.978908598893932e-05,
+      "loss": 0.1733,
+      "step": 8040
+    },
+    {
+      "grad_norm": 0.6274117231369019,
+      "learning_rate": 6.971313621714756e-05,
+      "loss": 0.1689,
+      "step": 8050
+    },
+    {
+      "grad_norm": 0.6123325228691101,
+      "learning_rate": 6.96371325503941e-05,
+      "loss": 0.165,
+      "step": 8060
+    },
+    {
+      "grad_norm": 0.5399617552757263,
+      "learning_rate": 6.956107519647014e-05,
+      "loss": 0.1849,
+      "step": 8070
+    },
+    {
+      "grad_norm": 0.6987713575363159,
+      "learning_rate": 6.94849643633135e-05,
+      "loss": 0.1764,
+      "step": 8080
+    },
+    {
+      "grad_norm": 0.5001187920570374,
+      "learning_rate": 6.940880025900834e-05,
+      "loss": 0.1608,
+      "step": 8090
+    },
+    {
+      "grad_norm": 0.4777175784111023,
+      "learning_rate": 6.933258309178438e-05,
+      "loss": 0.1763,
+      "step": 8100
+    },
+    {
+      "grad_norm": 0.9429814219474792,
+      "learning_rate": 6.925631307001646e-05,
+      "loss": 0.1817,
+      "step": 8110
+    },
+    {
+      "grad_norm": 0.6263329386711121,
+      "learning_rate": 6.91799904022239e-05,
+      "loss": 0.188,
+      "step": 8120
+    },
+    {
+      "grad_norm": 0.648918092250824,
+      "learning_rate": 6.910361529706997e-05,
+      "loss": 0.1686,
+      "step": 8130
+    },
+    {
+      "grad_norm": 0.4071788787841797,
+      "learning_rate": 6.902718796336131e-05,
+      "loss": 0.1574,
+      "step": 8140
+    },
+    {
+      "grad_norm": 0.33340150117874146,
+      "learning_rate": 6.895070861004729e-05,
+      "loss": 0.1599,
+      "step": 8150
+    },
+    {
+      "grad_norm": 0.43489596247673035,
+      "learning_rate": 6.887417744621956e-05,
+      "loss": 0.169,
+      "step": 8160
+    },
+    {
+      "grad_norm": 0.47071486711502075,
+      "learning_rate": 6.87975946811114e-05,
+      "loss": 0.1586,
+      "step": 8170
+    },
+    {
+      "grad_norm": 0.6574937105178833,
+      "learning_rate": 6.872096052409718e-05,
+      "loss": 0.178,
+      "step": 8180
+    },
+    {
+      "grad_norm": 0.5476977825164795,
+      "learning_rate": 6.864427518469174e-05,
+      "loss": 0.1795,
+      "step": 8190
+    },
+    {
+      "grad_norm": 0.5717020034790039,
+      "learning_rate": 6.856753887254986e-05,
+      "loss": 0.1735,
+      "step": 8200
+    },
+    {
+      "grad_norm": 0.5040568113327026,
+      "learning_rate": 6.849075179746572e-05,
+      "loss": 0.1878,
+      "step": 8210
+    },
+    {
+      "grad_norm": 0.42369338870048523,
+      "learning_rate": 6.841391416937221e-05,
+      "loss": 0.1647,
+      "step": 8220
+    },
+    {
+      "grad_norm": 0.5649428963661194,
+      "learning_rate": 6.833702619834053e-05,
+      "loss": 0.1708,
+      "step": 8230
+    },
+    {
+      "grad_norm": 0.5883945822715759,
+      "learning_rate": 6.82600880945794e-05,
+      "loss": 0.2288,
+      "step": 8240
+    },
+    {
+      "grad_norm": 0.6284005641937256,
+      "learning_rate": 6.818310006843468e-05,
+      "loss": 0.1687,
+      "step": 8250
+    },
+    {
+      "grad_norm": 0.4061494767665863,
+      "learning_rate": 6.810606233038868e-05,
+      "loss": 0.1775,
+      "step": 8260
+    },
+    {
+      "grad_norm": 0.46648678183555603,
+      "learning_rate": 6.802897509105966e-05,
+      "loss": 0.1785,
+      "step": 8270
+    },
+    {
+      "grad_norm": 0.48088502883911133,
+      "learning_rate": 6.79518385612012e-05,
+      "loss": 0.1729,
+      "step": 8280
+    },
+    {
+      "grad_norm": 0.5093273520469666,
+      "learning_rate": 6.787465295170157e-05,
+      "loss": 0.1759,
+      "step": 8290
+    },
+    {
+      "grad_norm": 0.471543550491333,
+      "learning_rate": 6.779741847358332e-05,
+      "loss": 0.1749,
+      "step": 8300
+    },
+    {
+      "grad_norm": 0.5014373660087585,
+      "learning_rate": 6.772013533800256e-05,
+      "loss": 0.1593,
+      "step": 8310
+    },
+    {
+      "grad_norm": 0.4633059799671173,
+      "learning_rate": 6.764280375624843e-05,
+      "loss": 0.174,
+      "step": 8320
+    },
+    {
+      "grad_norm": 0.6488783359527588,
+      "learning_rate": 6.756542393974252e-05,
+      "loss": 0.166,
+      "step": 8330
+    },
+    {
+      "grad_norm": 0.49490612745285034,
+      "learning_rate": 6.748799610003828e-05,
+      "loss": 0.1714,
+      "step": 8340
+    },
+    {
+      "grad_norm": 0.47256970405578613,
+      "learning_rate": 6.741052044882048e-05,
+      "loss": 0.1798,
+      "step": 8350
+    },
+    {
+      "grad_norm": 0.6256896257400513,
+      "learning_rate": 6.73329971979046e-05,
+      "loss": 0.1761,
+      "step": 8360
+    },
+    {
+      "grad_norm": 0.7974782586097717,
+      "learning_rate": 6.725542655923625e-05,
+      "loss": 0.1703,
+      "step": 8370
+    },
+    {
+      "grad_norm": 0.4163827896118164,
+      "learning_rate": 6.717780874489057e-05,
+      "loss": 0.1802,
+      "step": 8380
+    },
+    {
+      "grad_norm": 0.5573554039001465,
+      "learning_rate": 6.710014396707172e-05,
+      "loss": 0.1548,
+      "step": 8390
+    },
+    {
+      "grad_norm": 0.42139360308647156,
+      "learning_rate": 6.702243243811221e-05,
+      "loss": 0.1807,
+      "step": 8400
+    },
+    {
+      "grad_norm": 0.580695390701294,
+      "learning_rate": 6.694467437047244e-05,
+      "loss": 0.1713,
+      "step": 8410
+    },
+    {
+      "grad_norm": 0.630988597869873,
+      "learning_rate": 6.686686997673997e-05,
+      "loss": 0.1863,
+      "step": 8420
+    },
+    {
+      "grad_norm": 0.6133594512939453,
+      "learning_rate": 6.678901946962903e-05,
+      "loss": 0.1759,
+      "step": 8430
+    },
+    {
+      "grad_norm": 0.6058676242828369,
+      "learning_rate": 6.671112306197996e-05,
+      "loss": 0.1813,
+      "step": 8440
+    },
+    {
+      "grad_norm": 0.5705500245094299,
+      "learning_rate": 6.663318096675854e-05,
+      "loss": 0.1721,
+      "step": 8450
+    },
+    {
+      "grad_norm": 0.7127603888511658,
+      "learning_rate": 6.655519339705552e-05,
+      "loss": 0.1765,
+      "step": 8460
+    },
+    {
+      "grad_norm": 0.5193121433258057,
+      "learning_rate": 6.647716056608588e-05,
+      "loss": 0.178,
+      "step": 8470
+    },
+    {
+      "grad_norm": 0.3939180374145508,
+      "learning_rate": 6.639908268718843e-05,
+      "loss": 0.1657,
+      "step": 8480
+    },
+    {
+      "grad_norm": 0.45440754294395447,
+      "learning_rate": 6.632095997382514e-05,
+      "loss": 0.1679,
+      "step": 8490
+    },
+    {
+      "grad_norm": 0.4980035722255707,
+      "learning_rate": 6.624279263958047e-05,
+      "loss": 0.1606,
+      "step": 8500
+    },
+    {
+      "grad_norm": 0.48411908745765686,
+      "learning_rate": 6.616458089816097e-05,
+      "loss": 0.1703,
+      "step": 8510
+    },
+    {
+      "grad_norm": 0.5369507670402527,
+      "learning_rate": 6.608632496339454e-05,
+      "loss": 0.1868,
+      "step": 8520
+    },
+    {
+      "grad_norm": 0.46849775314331055,
+      "learning_rate": 6.600802504922988e-05,
+      "loss": 0.1904,
+      "step": 8530
+    },
+    {
+      "grad_norm": 0.4244977831840515,
+      "learning_rate": 6.592968136973604e-05,
+      "loss": 0.169,
+      "step": 8540
+    },
+    {
+      "grad_norm": 0.45252394676208496,
+      "learning_rate": 6.585129413910159e-05,
+      "loss": 0.1766,
+      "step": 8550
+    },
+    {
+      "grad_norm": 0.4946672320365906,
+      "learning_rate": 6.577286357163424e-05,
+      "loss": 0.167,
+      "step": 8560
+    },
+    {
+      "grad_norm": 0.4654735326766968,
+      "learning_rate": 6.569438988176018e-05,
+      "loss": 0.1581,
+      "step": 8570
+    },
+    {
+      "grad_norm": 0.4032638967037201,
+      "learning_rate": 6.561587328402347e-05,
+      "loss": 0.1633,
+      "step": 8580
+    },
+    {
+      "grad_norm": 0.4498487710952759,
+      "learning_rate": 6.553731399308549e-05,
+      "loss": 0.1829,
+      "step": 8590
+    },
+    {
+      "grad_norm": 0.5855516791343689,
+      "learning_rate": 6.545871222372436e-05,
+      "loss": 0.1584,
+      "step": 8600
+    },
+    {
+      "grad_norm": 0.7406018972396851,
+      "learning_rate": 6.538006819083426e-05,
+      "loss": 0.1521,
+      "step": 8610
+    },
+    {
+      "grad_norm": 0.6364853382110596,
+      "learning_rate": 6.530138210942505e-05,
+      "loss": 0.1678,
+      "step": 8620
+    },
+    {
+      "grad_norm": 0.45917588472366333,
+      "learning_rate": 6.522265419462141e-05,
+      "loss": 0.167,
+      "step": 8630
+    },
+    {
+      "grad_norm": 0.5528519153594971,
+      "learning_rate": 6.514388466166248e-05,
+      "loss": 0.166,
+      "step": 8640
+    },
+    {
+      "grad_norm": 0.4700118601322174,
+      "learning_rate": 6.506507372590119e-05,
+      "loss": 0.1764,
+      "step": 8650
+    },
+    {
+      "grad_norm": 0.5571269989013672,
+      "learning_rate": 6.498622160280355e-05,
+      "loss": 0.17,
+      "step": 8660
+    },
+    {
+      "grad_norm": 0.4334975779056549,
+      "learning_rate": 6.490732850794832e-05,
+      "loss": 0.1741,
+      "step": 8670
+    },
+    {
+      "grad_norm": 0.5792575478553772,
+      "learning_rate": 6.482839465702616e-05,
+      "loss": 0.1583,
+      "step": 8680
+    },
+    {
+      "grad_norm": 0.5591619610786438,
+      "learning_rate": 6.474942026583923e-05,
+      "loss": 0.1684,
+      "step": 8690
+    },
+    {
+      "grad_norm": 0.6293572187423706,
+      "learning_rate": 6.467040555030052e-05,
+      "loss": 0.1553,
+      "step": 8700
+    },
+    {
+      "grad_norm": 0.5244634747505188,
+      "learning_rate": 6.459135072643321e-05,
+      "loss": 0.1547,
+      "step": 8710
+    },
+    {
+      "grad_norm": 0.5345861911773682,
+      "learning_rate": 6.451225601037019e-05,
+      "loss": 0.2028,
+      "step": 8720
+    },
+    {
+      "grad_norm": 0.6430583000183105,
+      "learning_rate": 6.443312161835338e-05,
+      "loss": 0.1592,
+      "step": 8730
+    },
+    {
+      "grad_norm": 1.0593215227127075,
+      "learning_rate": 6.43539477667332e-05,
+      "loss": 0.1926,
+      "step": 8740
+    },
+    {
+      "grad_norm": 0.546528160572052,
+      "learning_rate": 6.427473467196793e-05,
+      "loss": 0.1662,
+      "step": 8750
+    },
+    {
+      "grad_norm": 0.5268025994300842,
+      "learning_rate": 6.419548255062315e-05,
+      "loss": 0.1768,
+      "step": 8760
+    },
+    {
+      "grad_norm": 0.5170095562934875,
+      "learning_rate": 6.411619161937112e-05,
+      "loss": 0.1748,
+      "step": 8770
+    },
+    {
+      "grad_norm": 0.5583987236022949,
+      "learning_rate": 6.403686209499022e-05,
+      "loss": 0.1787,
+      "step": 8780
+    },
+    {
+      "grad_norm": 0.5135347247123718,
+      "learning_rate": 6.395749419436437e-05,
+      "loss": 0.1631,
+      "step": 8790
+    },
+    {
+      "grad_norm": 0.44871315360069275,
+      "learning_rate": 6.387808813448234e-05,
+      "loss": 0.1726,
+      "step": 8800
+    },
+    {
+      "grad_norm": 0.6232064366340637,
+      "learning_rate": 6.37986441324373e-05,
+      "loss": 0.1627,
+      "step": 8810
+    },
+    {
+      "grad_norm": 0.5067418813705444,
+      "learning_rate": 6.37191624054261e-05,
+      "loss": 0.1635,
+      "step": 8820
+    },
+    {
+      "grad_norm": 0.5494692325592041,
+      "learning_rate": 6.363964317074872e-05,
+      "loss": 0.1432,
+      "step": 8830
+    },
+    {
+      "grad_norm": 0.5651106834411621,
+      "learning_rate": 6.356008664580776e-05,
+      "loss": 0.1727,
+      "step": 8840
+    },
+    {
+      "grad_norm": 0.52388995885849,
+      "learning_rate": 6.348049304810771e-05,
+      "loss": 0.1634,
+      "step": 8850
+    },
+    {
+      "grad_norm": 0.45826584100723267,
+      "learning_rate": 6.340086259525442e-05,
+      "loss": 0.1606,
+      "step": 8860
+    },
+    {
+      "grad_norm": 0.5106724500656128,
+      "learning_rate": 6.332119550495448e-05,
+      "loss": 0.1525,
+      "step": 8870
+    },
+    {
+      "grad_norm": 0.4510875344276428,
+      "learning_rate": 6.324149199501473e-05,
+      "loss": 0.1669,
+      "step": 8880
+    },
+    {
+      "grad_norm": 0.5665463805198669,
+      "learning_rate": 6.316175228334146e-05,
+      "loss": 0.1532,
+      "step": 8890
+    },
+    {
+      "grad_norm": 0.5110814571380615,
+      "learning_rate": 6.308197658794003e-05,
+      "loss": 0.1814,
+      "step": 8900
+    },
+    {
+      "grad_norm": 0.5583161115646362,
+      "learning_rate": 6.300216512691417e-05,
+      "loss": 0.1732,
+      "step": 8910
+    },
+    {
+      "grad_norm": 0.4035802483558655,
+      "learning_rate": 6.292231811846532e-05,
+      "loss": 0.1573,
+      "step": 8920
+    },
+    {
+      "grad_norm": 0.48649492859840393,
+      "learning_rate": 6.284243578089217e-05,
+      "loss": 0.1684,
+      "step": 8930
+    },
+    {
+      "grad_norm": 0.48168379068374634,
+      "learning_rate": 6.276251833258999e-05,
+      "loss": 0.1696,
+      "step": 8940
+    },
+    {
+      "grad_norm": 0.7830365896224976,
+      "learning_rate": 6.268256599205003e-05,
+      "loss": 0.1818,
+      "step": 8950
+    },
+    {
+      "grad_norm": 0.5242315530776978,
+      "learning_rate": 6.260257897785892e-05,
+      "loss": 0.1613,
+      "step": 8960
+    },
+    {
+      "grad_norm": 0.3354107141494751,
+      "learning_rate": 6.252255750869811e-05,
+      "loss": 0.158,
+      "step": 8970
+    },
+    {
+      "grad_norm": 0.5901716947555542,
+      "learning_rate": 6.244250180334325e-05,
+      "loss": 0.1816,
+      "step": 8980
+    },
+    {
+      "grad_norm": 0.4838642179965973,
+      "learning_rate": 6.236241208066356e-05,
+      "loss": 0.1646,
+      "step": 8990
+    },
+    {
+      "grad_norm": 0.6138313412666321,
+      "learning_rate": 6.228228855962133e-05,
+      "loss": 0.1652,
+      "step": 9000
+    },
+    {
+      "grad_norm": 0.4725053906440735,
+      "learning_rate": 6.220213145927115e-05,
+      "loss": 0.1511,
+      "step": 9010
+    },
+    {
+      "grad_norm": 0.525400698184967,
+      "learning_rate": 6.212194099875951e-05,
+      "loss": 0.1646,
+      "step": 9020
+    },
+    {
+      "grad_norm": 0.5466144680976868,
+      "learning_rate": 6.204171739732405e-05,
+      "loss": 0.1617,
+      "step": 9030
+    },
+    {
+      "grad_norm": 0.6913033723831177,
+      "learning_rate": 6.196146087429303e-05,
+      "loss": 0.1656,
+      "step": 9040
+    },
+    {
+      "grad_norm": 0.6543959975242615,
+      "learning_rate": 6.188117164908474e-05,
+      "loss": 0.179,
+      "step": 9050
+    },
+    {
+      "grad_norm": 0.3706721365451813,
+      "learning_rate": 6.180084994120684e-05,
+      "loss": 0.1601,
+      "step": 9060
+    },
+    {
+      "grad_norm": 0.5568912625312805,
+      "learning_rate": 6.17204959702558e-05,
+      "loss": 0.1659,
+      "step": 9070
+    },
+    {
+      "grad_norm": 0.5273260474205017,
+      "learning_rate": 6.164010995591635e-05,
+      "loss": 0.1797,
+      "step": 9080
+    },
+    {
+      "grad_norm": 0.6631227731704712,
+      "learning_rate": 6.155969211796076e-05,
+      "loss": 0.1769,
+      "step": 9090
+    },
+    {
+      "grad_norm": 0.6078776717185974,
+      "learning_rate": 6.147924267624829e-05,
+      "loss": 0.1742,
+      "step": 9100
+    },
+    {
+      "grad_norm": 0.5191851854324341,
+      "learning_rate": 6.13987618507247e-05,
+      "loss": 0.1555,
+      "step": 9110
+    },
+    {
+      "grad_norm": 0.43232864141464233,
+      "learning_rate": 6.131824986142147e-05,
+      "loss": 0.1657,
+      "step": 9120
+    },
+    {
+      "grad_norm": 0.3777904808521271,
+      "learning_rate": 6.123770692845529e-05,
+      "loss": 0.17,
+      "step": 9130
+    },
+    {
+      "grad_norm": 0.5256659388542175,
+      "learning_rate": 6.11571332720275e-05,
+      "loss": 0.1596,
+      "step": 9140
+    },
+    {
+      "grad_norm": 0.5890393257141113,
+      "learning_rate": 6.107652911242336e-05,
+      "loss": 0.1772,
+      "step": 9150
+    },
+    {
+      "grad_norm": 0.5760863423347473,
+      "learning_rate": 6.0995894670011586e-05,
+      "loss": 0.1554,
+      "step": 9160
+    },
+    {
+      "grad_norm": 0.4065435230731964,
+      "learning_rate": 6.091523016524368e-05,
+      "loss": 0.1517,
+      "step": 9170
+    },
+    {
+      "grad_norm": 0.5711379051208496,
+      "learning_rate": 6.083453581865328e-05,
+      "loss": 0.1845,
+      "step": 9180
+    },
+    {
+      "grad_norm": 0.5990337133407593,
+      "learning_rate": 6.075381185085568e-05,
+      "loss": 0.1746,
+      "step": 9190
+    },
+    {
+      "grad_norm": 0.6722525358200073,
+      "learning_rate": 6.067305848254709e-05,
+      "loss": 0.1806,
+      "step": 9200
+    },
+    {
+      "grad_norm": 0.5812041759490967,
+      "learning_rate": 6.059227593450418e-05,
+      "loss": 0.1641,
+      "step": 9210
+    },
+    {
+      "grad_norm": 0.5105819702148438,
+      "learning_rate": 6.051146442758333e-05,
+      "loss": 0.1726,
+      "step": 9220
+    },
+    {
+      "grad_norm": 0.4999433755874634,
+      "learning_rate": 6.043062418272012e-05,
+      "loss": 0.1636,
+      "step": 9230
+    },
+    {
+      "grad_norm": 0.6747785806655884,
+      "learning_rate": 6.0349755420928666e-05,
+      "loss": 0.1798,
+      "step": 9240
+    },
+    {
+      "grad_norm": 0.5493257641792297,
+      "learning_rate": 6.0268858363301105e-05,
+      "loss": 0.1637,
+      "step": 9250
+    },
+    {
+      "grad_norm": 0.459867388010025,
+      "learning_rate": 6.018793323100689e-05,
+      "loss": 0.1684,
+      "step": 9260
+    },
+    {
+      "grad_norm": 0.4723917543888092,
+      "learning_rate": 6.0106980245292255e-05,
+      "loss": 0.1677,
+      "step": 9270
+    },
+    {
+      "grad_norm": 0.4414682686328888,
+      "learning_rate": 6.002599962747957e-05,
+      "loss": 0.1676,
+      "step": 9280
+    },
+    {
+      "grad_norm": 0.4809039831161499,
+      "learning_rate": 5.994499159896673e-05,
+      "loss": 0.1736,
+      "step": 9290
+    },
+    {
+      "grad_norm": 0.7380583882331848,
+      "learning_rate": 5.9863956381226607e-05,
+      "loss": 0.1839,
+      "step": 9300
+    },
+    {
+      "grad_norm": 0.865779459476471,
+      "learning_rate": 5.9782894195806394e-05,
+      "loss": 0.1788,
+      "step": 9310
+    },
+    {
+      "grad_norm": 0.5601370930671692,
+      "learning_rate": 5.9701805264327004e-05,
+      "loss": 0.1691,
+      "step": 9320
+    },
+    {
+      "grad_norm": 0.427077978849411,
+      "learning_rate": 5.96206898084825e-05,
+      "loss": 0.1579,
+      "step": 9330
+    },
+    {
+      "grad_norm": 0.4620397686958313,
+      "learning_rate": 5.953954805003942e-05,
+      "loss": 0.1692,
+      "step": 9340
+    },
+    {
+      "grad_norm": 0.5680602192878723,
+      "learning_rate": 5.945838021083623e-05,
+      "loss": 0.1565,
+      "step": 9350
+    },
+    {
+      "grad_norm": 0.4300524890422821,
+      "learning_rate": 5.9377186512782714e-05,
+      "loss": 0.1572,
+      "step": 9360
+    },
+    {
+      "grad_norm": 0.5034781694412231,
+      "learning_rate": 5.929596717785935e-05,
+      "loss": 0.1733,
+      "step": 9370
+    },
+    {
+      "grad_norm": 0.7639031410217285,
+      "learning_rate": 5.921472242811668e-05,
+      "loss": 0.1725,
+      "step": 9380
+    },
+    {
+      "grad_norm": 0.43752986192703247,
+      "learning_rate": 5.913345248567475e-05,
+      "loss": 0.1593,
+      "step": 9390
+    },
+    {
+      "grad_norm": 0.3173084557056427,
+      "learning_rate": 5.905215757272248e-05,
+      "loss": 0.1725,
+      "step": 9400
+    },
+    {
+      "grad_norm": 0.4746582806110382,
+      "learning_rate": 5.897083791151706e-05,
+      "loss": 0.1741,
+      "step": 9410
+    },
+    {
+      "grad_norm": 0.6064617037773132,
+      "learning_rate": 5.888949372438336e-05,
+      "loss": 0.1654,
+      "step": 9420
+    },
+    {
+      "grad_norm": 0.561570405960083,
+      "learning_rate": 5.8808125233713255e-05,
+      "loss": 0.1648,
+      "step": 9430
+    },
+    {
+      "grad_norm": 0.4926293194293976,
+      "learning_rate": 5.872673266196509e-05,
+      "loss": 0.1655,
+      "step": 9440
+    },
+    {
+      "grad_norm": 0.46248817443847656,
+      "learning_rate": 5.864531623166305e-05,
+      "loss": 0.1541,
+      "step": 9450
+    },
+    {
+      "grad_norm": 0.9015707969665527,
+      "learning_rate": 5.856387616539656e-05,
+      "loss": 0.175,
+      "step": 9460
+    },
+    {
+      "grad_norm": 0.43535536527633667,
+      "learning_rate": 5.848241268581967e-05,
+      "loss": 0.1758,
+      "step": 9470
+    },
+    {
+      "grad_norm": 0.6100785136222839,
+      "learning_rate": 5.840092601565037e-05,
+      "loss": 0.1567,
+      "step": 9480
+    },
+    {
+      "grad_norm": 0.4814026653766632,
+      "learning_rate": 5.8319416377670144e-05,
+      "loss": 0.1534,
+      "step": 9490
+    },
+    {
+      "grad_norm": 0.7451731562614441,
+      "learning_rate": 5.82378839947232e-05,
+      "loss": 0.1678,
+      "step": 9500
+    },
+    {
+      "grad_norm": 0.48139694333076477,
+      "learning_rate": 5.815632908971599e-05,
+      "loss": 0.1587,
+      "step": 9510
+    },
+    {
+      "grad_norm": 0.5009700655937195,
+      "learning_rate": 5.80747518856165e-05,
+      "loss": 0.1796,
+      "step": 9520
+    },
+    {
+      "grad_norm": 0.44166460633277893,
+      "learning_rate": 5.799315260545367e-05,
+      "loss": 0.1608,
+      "step": 9530
+    },
+    {
+      "grad_norm": 0.5245239734649658,
+      "learning_rate": 5.791153147231686e-05,
+      "loss": 0.1464,
+      "step": 9540
+    },
+    {
+      "grad_norm": 0.4896029233932495,
+      "learning_rate": 5.782988870935509e-05,
+      "loss": 0.1747,
+      "step": 9550
+    },
+    {
+      "grad_norm": 0.4968087375164032,
+      "learning_rate": 5.774822453977657e-05,
+      "loss": 0.1832,
+      "step": 9560
+    },
+    {
+      "grad_norm": 0.35863521695137024,
+      "learning_rate": 5.7666539186848036e-05,
+      "loss": 0.1522,
+      "step": 9570
+    },
+    {
+      "grad_norm": 0.6285898089408875,
+      "learning_rate": 5.758483287389411e-05,
+      "loss": 0.1584,
+      "step": 9580
+    },
+    {
+      "grad_norm": 0.5019512176513672,
+      "learning_rate": 5.7503105824296735e-05,
+      "loss": 0.1672,
+      "step": 9590
+    },
+    {
+      "grad_norm": 0.38427790999412537,
+      "learning_rate": 5.742135826149453e-05,
+      "loss": 0.1591,
+      "step": 9600
+    },
+    {
+      "grad_norm": 0.543759286403656,
+      "learning_rate": 5.7339590408982223e-05,
+      "loss": 0.1652,
+      "step": 9610
+    },
+    {
+      "grad_norm": 0.39145317673683167,
+      "learning_rate": 5.725780249031e-05,
+      "loss": 0.1543,
+      "step": 9620
+    },
+    {
+      "grad_norm": 0.5868179798126221,
+      "learning_rate": 5.717599472908292e-05,
+      "loss": 0.1637,
+      "step": 9630
+    },
+    {
+      "grad_norm": 0.4005380868911743,
+      "learning_rate": 5.7094167348960237e-05,
+      "loss": 0.1601,
+      "step": 9640
+    },
+    {
+      "grad_norm": 0.4945452809333801,
+      "learning_rate": 5.7012320573654945e-05,
+      "loss": 0.1516,
+      "step": 9650
+    },
+    {
+      "grad_norm": 0.7010737061500549,
+      "learning_rate": 5.693045462693295e-05,
+      "loss": 0.1667,
+      "step": 9660
+    },
+    {
+      "grad_norm": 0.5524814128875732,
+      "learning_rate": 5.684856973261266e-05,
+      "loss": 0.1548,
+      "step": 9670
+    },
+    {
+      "grad_norm": 0.5110557079315186,
+      "learning_rate": 5.6766666114564215e-05,
+      "loss": 0.1731,
+      "step": 9680
+    },
+    {
+      "grad_norm": 0.7684085965156555,
+      "learning_rate": 5.668474399670899e-05,
+      "loss": 0.1759,
+      "step": 9690
+    },
+    {
+      "grad_norm": 0.6560841798782349,
+      "learning_rate": 5.660280360301896e-05,
+      "loss": 0.1584,
+      "step": 9700
+    },
+    {
+      "grad_norm": 0.4157427251338959,
+      "learning_rate": 5.652084515751599e-05,
+      "loss": 0.1673,
+      "step": 9710
+    },
+    {
+      "grad_norm": 0.3634550869464874,
+      "learning_rate": 5.643886888427137e-05,
+      "loss": 0.1735,
+      "step": 9720
+    },
+    {
+      "grad_norm": 0.4537419080734253,
+      "learning_rate": 5.6356875007405074e-05,
+      "loss": 0.1736,
+      "step": 9730
+    },
+    {
+      "grad_norm": 0.653682291507721,
+      "learning_rate": 5.627486375108525e-05,
+      "loss": 0.1763,
+      "step": 9740
+    },
+    {
+      "grad_norm": 0.6879625916481018,
+      "learning_rate": 5.619283533952754e-05,
+      "loss": 0.1769,
+      "step": 9750
+    },
+    {
+      "grad_norm": 0.5207549929618835,
+      "learning_rate": 5.6110789996994474e-05,
+      "loss": 0.1793,
+      "step": 9760
+    },
+    {
+      "grad_norm": 0.9255119562149048,
+      "learning_rate": 5.602872794779491e-05,
+      "loss": 0.1659,
+      "step": 9770
+    },
+    {
+      "grad_norm": 0.4505520462989807,
+      "learning_rate": 5.594664941628334e-05,
+      "loss": 0.1801,
+      "step": 9780
+    },
+    {
+      "grad_norm": 1.0005419254302979,
+      "learning_rate": 5.5864554626859324e-05,
+      "loss": 0.1789,
+      "step": 9790
+    },
+    {
+      "grad_norm": 0.5624749064445496,
+      "learning_rate": 5.578244380396691e-05,
+      "loss": 0.184,
+      "step": 9800
+    },
+    {
+      "grad_norm": 0.5277497172355652,
+      "learning_rate": 5.570031717209394e-05,
+      "loss": 0.1674,
+      "step": 9810
+    },
+    {
+      "grad_norm": 0.4932638108730316,
+      "learning_rate": 5.561817495577147e-05,
+      "loss": 0.1562,
+      "step": 9820
+    },
+    {
+      "grad_norm": 0.681172251701355,
+      "learning_rate": 5.5536017379573215e-05,
+      "loss": 0.1838,
+      "step": 9830
+    },
+    {
+      "grad_norm": 0.5972948670387268,
+      "learning_rate": 5.545384466811483e-05,
+      "loss": 0.1864,
+      "step": 9840
+    },
+    {
+      "grad_norm": 0.468509703874588,
+      "learning_rate": 5.5371657046053384e-05,
+      "loss": 0.1698,
+      "step": 9850
+    },
+    {
+      "grad_norm": 0.5842800736427307,
+      "learning_rate": 5.528945473808669e-05,
+      "loss": 0.1886,
+      "step": 9860
+    },
+    {
+      "grad_norm": 0.5157095193862915,
+      "learning_rate": 5.520723796895272e-05,
+      "loss": 0.1904,
+      "step": 9870
+    },
+    {
+      "grad_norm": 0.3780837953090668,
+      "learning_rate": 5.512500696342897e-05,
+      "loss": 0.1775,
+      "step": 9880
+    },
+    {
+      "grad_norm": 0.5193829536437988,
+      "learning_rate": 5.504276194633188e-05,
+      "loss": 0.1642,
+      "step": 9890
+    },
+    {
+      "grad_norm": 0.48715534806251526,
+      "learning_rate": 5.49605031425162e-05,
+      "loss": 0.1715,
+      "step": 9900
+    },
+    {
+      "grad_norm": 0.40378332138061523,
+      "learning_rate": 5.487823077687434e-05,
+      "loss": 0.1505,
+      "step": 9910
+    },
+    {
+      "grad_norm": 0.43677860498428345,
+      "learning_rate": 5.4795945074335806e-05,
+      "loss": 0.1675,
+      "step": 9920
+    },
+    {
+      "grad_norm": 0.6398891806602478,
+      "learning_rate": 5.471364625986657e-05,
+      "loss": 0.1761,
+      "step": 9930
+    },
+    {
+      "grad_norm": 0.3452124297618866,
+      "learning_rate": 5.463133455846845e-05,
+      "loss": 0.1622,
+      "step": 9940
+    },
+    {
+      "grad_norm": 0.4906247854232788,
+      "learning_rate": 5.4549010195178505e-05,
+      "loss": 0.1775,
+      "step": 9950
+    },
+    {
+      "grad_norm": 0.5674220323562622,
+      "learning_rate": 5.446667339506838e-05,
+      "loss": 0.1822,
+      "step": 9960
+    },
+    {
+      "grad_norm": 0.5618964433670044,
+      "learning_rate": 5.4384324383243756e-05,
+      "loss": 0.1704,
+      "step": 9970
+    },
+    {
+      "grad_norm": 0.611150324344635,
+      "learning_rate": 5.430196338484368e-05,
+      "loss": 0.2035,
+      "step": 9980
+    },
+    {
+      "grad_norm": 0.6093125939369202,
+      "learning_rate": 5.4219590625039975e-05,
+      "loss": 0.1643,
+      "step": 9990
+    },
+    {
+      "grad_norm": 0.5724085569381714,
+      "learning_rate": 5.413720632903664e-05,
+      "loss": 0.1778,
+      "step": 10000
+    },
+    {
+      "grad_norm": 0.5971804261207581,
+      "learning_rate": 5.405481072206917e-05,
+      "loss": 0.1713,
+      "step": 10010
+    },
+    {
+      "grad_norm": 0.6137742400169373,
+      "learning_rate": 5.397240402940402e-05,
+      "loss": 0.1732,
+      "step": 10020
+    },
+    {
+      "grad_norm": 0.5185403227806091,
+      "learning_rate": 5.388998647633794e-05,
+      "loss": 0.2019,
+      "step": 10030
+    },
+    {
+      "grad_norm": 0.40620481967926025,
+      "learning_rate": 5.380755828819737e-05,
+      "loss": 0.1731,
+      "step": 10040
+    },
+    {
+      "grad_norm": 0.6932348608970642,
+      "learning_rate": 5.3725119690337846e-05,
+      "loss": 0.1575,
+      "step": 10050
+    },
+    {
+      "grad_norm": 0.4542210102081299,
+      "learning_rate": 5.3642670908143324e-05,
+      "loss": 0.1644,
+      "step": 10060
+    },
+    {
+      "grad_norm": 0.5463055968284607,
+      "learning_rate": 5.356021216702562e-05,
+      "loss": 0.1554,
+      "step": 10070
+    },
+    {
+      "grad_norm": 0.6204454302787781,
+      "learning_rate": 5.347774369242381e-05,
+      "loss": 0.1622,
+      "step": 10080
+    },
+    {
+      "grad_norm": 0.5540537238121033,
+      "learning_rate": 5.3395265709803545e-05,
+      "loss": 0.176,
+      "step": 10090
+    },
+    {
+      "grad_norm": 0.5746768116950989,
+      "learning_rate": 5.331277844465647e-05,
+      "loss": 0.1561,
+      "step": 10100
+    },
+    {
+      "grad_norm": 0.37307244539260864,
+      "learning_rate": 5.323028212249963e-05,
+      "loss": 0.1737,
+      "step": 10110
+    },
+    {
+      "grad_norm": 0.731255292892456,
+      "learning_rate": 5.314777696887481e-05,
+      "loss": 0.1714,
+      "step": 10120
+    },
+    {
+      "grad_norm": 0.6135475635528564,
+      "learning_rate": 5.306526320934796e-05,
+      "loss": 0.18,
+      "step": 10130
+    },
+    {
+      "grad_norm": 0.49417629837989807,
+      "learning_rate": 5.298274106950854e-05,
+      "loss": 0.18,
+      "step": 10140
+    },
+    {
+      "grad_norm": 0.38949549198150635,
+      "learning_rate": 5.290021077496893e-05,
+      "loss": 0.1586,
+      "step": 10150
+    },
+    {
+      "grad_norm": 0.552558422088623,
+      "learning_rate": 5.2817672551363816e-05,
+      "loss": 0.1582,
+      "step": 10160
+    },
+    {
+      "grad_norm": 0.47757411003112793,
+      "learning_rate": 5.273512662434952e-05,
+      "loss": 0.1643,
+      "step": 10170
+    },
+    {
+      "grad_norm": 0.40912917256355286,
+      "learning_rate": 5.265257321960349e-05,
+      "loss": 0.1755,
+      "step": 10180
+    },
+    {
+      "grad_norm": 0.43193644285202026,
+      "learning_rate": 5.257001256282357e-05,
+      "loss": 0.1741,
+      "step": 10190
+    },
+    {
+      "grad_norm": 0.7017669677734375,
+      "learning_rate": 5.248744487972742e-05,
+      "loss": 0.174,
+      "step": 10200
+    },
+    {
+      "grad_norm": 0.6171168088912964,
+      "learning_rate": 5.240487039605196e-05,
+      "loss": 0.1798,
+      "step": 10210
+    },
+    {
+      "grad_norm": 0.5147926807403564,
+      "learning_rate": 5.232228933755267e-05,
+      "loss": 0.1793,
+      "step": 10220
+    },
+    {
+      "grad_norm": 0.6641113758087158,
+      "learning_rate": 5.2239701930003006e-05,
+      "loss": 0.1778,
+      "step": 10230
+    },
+    {
+      "grad_norm": 0.7683154940605164,
+      "learning_rate": 5.215710839919379e-05,
+      "loss": 0.1614,
+      "step": 10240
+    },
+    {
+      "grad_norm": 0.5681193470954895,
+      "learning_rate": 5.207450897093257e-05,
+      "loss": 0.1834,
+      "step": 10250
+    },
+    {
+      "grad_norm": 0.7068840861320496,
+      "learning_rate": 5.1991903871043046e-05,
+      "loss": 0.1783,
+      "step": 10260
+    },
+    {
+      "grad_norm": 0.46191349625587463,
+      "learning_rate": 5.190929332536439e-05,
+      "loss": 0.1523,
+      "step": 10270
+    },
+    {
+      "grad_norm": 0.6000279188156128,
+      "learning_rate": 5.182667755975071e-05,
+      "loss": 0.1635,
+      "step": 10280
+    },
+    {
+      "grad_norm": 0.6448169946670532,
+      "learning_rate": 5.1744056800070315e-05,
+      "loss": 0.156,
+      "step": 10290
+    },
+    {
+      "grad_norm": 0.6426157355308533,
+      "learning_rate": 5.166143127220524e-05,
+      "loss": 0.1781,
+      "step": 10300
+    },
+    {
+      "grad_norm": 0.7184805870056152,
+      "learning_rate": 5.1578801202050485e-05,
+      "loss": 0.1678,
+      "step": 10310
+    },
+    {
+      "grad_norm": 0.527672529220581,
+      "learning_rate": 5.149616681551355e-05,
+      "loss": 0.1586,
+      "step": 10320
+    },
+    {
+      "grad_norm": 0.5092218518257141,
+      "learning_rate": 5.141352833851367e-05,
+      "loss": 0.1835,
+      "step": 10330
+    },
+    {
+      "grad_norm": 0.5019543170928955,
+      "learning_rate": 5.1330885996981285e-05,
+      "loss": 0.1763,
+      "step": 10340
+    },
+    {
+      "grad_norm": 0.4697420299053192,
+      "learning_rate": 5.124824001685741e-05,
+      "loss": 0.1547,
+      "step": 10350
+    },
+    {
+      "grad_norm": 0.40343138575553894,
+      "learning_rate": 5.116559062409298e-05,
+      "loss": 0.1528,
+      "step": 10360
+    },
+    {
+      "grad_norm": 0.6338273882865906,
+      "learning_rate": 5.10829380446483e-05,
+      "loss": 0.1657,
+      "step": 10370
+    },
+    {
+      "grad_norm": 0.6325240731239319,
+      "learning_rate": 5.100028250449235e-05,
+      "loss": 0.1996,
+      "step": 10380
+    },
+    {
+      "grad_norm": 0.6518502235412598,
+      "learning_rate": 5.0917624229602234e-05,
+      "loss": 0.1685,
+      "step": 10390
+    },
+    {
+      "grad_norm": 0.49050095677375793,
+      "learning_rate": 5.0834963445962524e-05,
+      "loss": 0.1684,
+      "step": 10400
+    },
+    {
+      "grad_norm": 0.5025054216384888,
+      "learning_rate": 5.075230037956461e-05,
+      "loss": 0.1784,
+      "step": 10410
+    },
+    {
+      "grad_norm": 0.5804678201675415,
+      "learning_rate": 5.0669635256406213e-05,
+      "loss": 0.1771,
+      "step": 10420
+    },
+    {
+      "grad_norm": 0.7888110876083374,
+      "learning_rate": 5.058696830249058e-05,
+      "loss": 0.1825,
+      "step": 10430
+    },
+    {
+      "grad_norm": 0.5129984021186829,
+      "learning_rate": 5.050429974382602e-05,
+      "loss": 0.1701,
+      "step": 10440
+    },
+    {
+      "grad_norm": 0.5313721895217896,
+      "learning_rate": 5.042162980642523e-05,
+      "loss": 0.1657,
+      "step": 10450
+    },
+    {
+      "grad_norm": 0.4321903884410858,
+      "learning_rate": 5.033895871630462e-05,
+      "loss": 0.185,
+      "step": 10460
+    },
+    {
+      "grad_norm": 0.4177863597869873,
+      "learning_rate": 5.025628669948386e-05,
+      "loss": 0.1586,
+      "step": 10470
+    },
+    {
+      "grad_norm": 0.6729171276092529,
+      "learning_rate": 5.017361398198502e-05,
+      "loss": 0.1456,
+      "step": 10480
+    },
+    {
+      "grad_norm": 0.4558064043521881,
+      "learning_rate": 5.009094078983221e-05,
+      "loss": 0.1567,
+      "step": 10490
+    },
+    {
+      "grad_norm": 0.46107494831085205,
+      "learning_rate": 5.000826734905073e-05,
+      "loss": 0.1786,
+      "step": 10500
+    },
+    {
+      "grad_norm": 0.42808714509010315,
+      "learning_rate": 4.9925593885666645e-05,
+      "loss": 0.1595,
+      "step": 10510
+    },
+    {
+      "grad_norm": 0.4761933386325836,
+      "learning_rate": 4.984292062570602e-05,
+      "loss": 0.1657,
+      "step": 10520
+    },
+    {
+      "grad_norm": 0.424326628446579,
+      "learning_rate": 4.976024779519442e-05,
+      "loss": 0.1634,
+      "step": 10530
+    },
+    {
+      "grad_norm": 0.5124930739402771,
+      "learning_rate": 4.9677575620156194e-05,
+      "loss": 0.1605,
+      "step": 10540
+    },
+    {
+      "grad_norm": 0.5146670341491699,
+      "learning_rate": 4.959490432661391e-05,
+      "loss": 0.1504,
+      "step": 10550
+    },
+    {
+      "grad_norm": 0.5470330119132996,
+      "learning_rate": 4.9512234140587726e-05,
+      "loss": 0.1732,
+      "step": 10560
+    },
+    {
+      "grad_norm": 1.0055515766143799,
+      "learning_rate": 4.942956528809477e-05,
+      "loss": 0.1784,
+      "step": 10570
+    },
+    {
+      "grad_norm": 0.5799973011016846,
+      "learning_rate": 4.934689799514854e-05,
+      "loss": 0.1827,
+      "step": 10580
+    },
+    {
+      "grad_norm": 0.7593075633049011,
+      "learning_rate": 4.926423248775827e-05,
+      "loss": 0.1662,
+      "step": 10590
+    },
+    {
+      "grad_norm": 0.5706773400306702,
+      "learning_rate": 4.918156899192826e-05,
+      "loss": 0.1508,
+      "step": 10600
+    },
+    {
+      "grad_norm": 0.555431604385376,
+      "learning_rate": 4.909890773365738e-05,
+      "loss": 0.1514,
+      "step": 10610
+    },
+    {
+      "grad_norm": 0.5203885436058044,
+      "learning_rate": 4.9016248938938344e-05,
+      "loss": 0.1626,
+      "step": 10620
+    },
+    {
+      "grad_norm": 0.4124124050140381,
+      "learning_rate": 4.8933592833757156e-05,
+      "loss": 0.1706,
+      "step": 10630
+    },
+    {
+      "grad_norm": 0.41043752431869507,
+      "learning_rate": 4.8850939644092435e-05,
+      "loss": 0.1527,
+      "step": 10640
+    },
+    {
+      "grad_norm": 0.48491060733795166,
+      "learning_rate": 4.876828959591485e-05,
+      "loss": 0.1531,
+      "step": 10650
+    },
+    {
+      "grad_norm": 0.4437198042869568,
+      "learning_rate": 4.8685642915186474e-05,
+      "loss": 0.167,
+      "step": 10660
+    },
+    {
+      "grad_norm": 0.5170273780822754,
+      "learning_rate": 4.860299982786018e-05,
+      "loss": 0.1634,
+      "step": 10670
+    },
+    {
+      "grad_norm": 0.6541380286216736,
+      "learning_rate": 4.852036055987901e-05,
+      "loss": 0.1657,
+      "step": 10680
+    },
+    {
+      "grad_norm": 0.7853013277053833,
+      "learning_rate": 4.843772533717558e-05,
+      "loss": 0.1791,
+      "step": 10690
+    },
+    {
+      "grad_norm": 0.5096300840377808,
+      "learning_rate": 4.835509438567142e-05,
+      "loss": 0.1715,
+      "step": 10700
+    },
+    {
+      "grad_norm": 0.5490502715110779,
+      "learning_rate": 4.827246793127639e-05,
+      "loss": 0.1723,
+      "step": 10710
+    },
+    {
+      "grad_norm": 0.4444034993648529,
+      "learning_rate": 4.818984619988807e-05,
+      "loss": 0.1768,
+      "step": 10720
+    },
+    {
+      "grad_norm": 0.8637450933456421,
+      "learning_rate": 4.810722941739115e-05,
+      "loss": 0.172,
+      "step": 10730
+    },
+    {
+      "grad_norm": 0.9409641623497009,
+      "learning_rate": 4.8024617809656684e-05,
+      "loss": 0.1783,
+      "step": 10740
+    },
+    {
+      "grad_norm": 0.5161317586898804,
+      "learning_rate": 4.794201160254171e-05,
+      "loss": 0.172,
+      "step": 10750
+    },
+    {
+      "grad_norm": 0.5636203289031982,
+      "learning_rate": 4.785941102188844e-05,
+      "loss": 0.1924,
+      "step": 10760
+    },
+    {
+      "grad_norm": 0.8947455883026123,
+      "learning_rate": 4.7776816293523686e-05,
+      "loss": 0.1636,
+      "step": 10770
+    },
+    {
+      "grad_norm": 0.48216044902801514,
+      "learning_rate": 4.769422764325832e-05,
+      "loss": 0.1698,
+      "step": 10780
+    },
+    {
+      "grad_norm": 0.613787055015564,
+      "learning_rate": 4.76116452968865e-05,
+      "loss": 0.1605,
+      "step": 10790
+    },
+    {
+      "grad_norm": 0.4093400835990906,
+      "learning_rate": 4.752906948018525e-05,
+      "loss": 0.174,
+      "step": 10800
+    },
+    {
+      "grad_norm": 0.3925022780895233,
+      "learning_rate": 4.7446500418913684e-05,
+      "loss": 0.1727,
+      "step": 10810
+    },
+    {
+      "grad_norm": 0.45032086968421936,
+      "learning_rate": 4.736393833881247e-05,
+      "loss": 0.1708,
+      "step": 10820
+    },
+    {
+      "grad_norm": 0.7255443930625916,
+      "learning_rate": 4.7281383465603194e-05,
+      "loss": 0.1599,
+      "step": 10830
+    },
+    {
+      "grad_norm": 0.5215007066726685,
+      "learning_rate": 4.71988360249877e-05,
+      "loss": 0.1682,
+      "step": 10840
+    },
+    {
+      "grad_norm": 0.4398680627346039,
+      "learning_rate": 4.7116296242647554e-05,
+      "loss": 0.1609,
+      "step": 10850
+    },
+    {
+      "grad_norm": 0.4730145335197449,
+      "learning_rate": 4.703376434424336e-05,
+      "loss": 0.1881,
+      "step": 10860
+    },
+    {
+      "grad_norm": 0.7034695744514465,
+      "learning_rate": 4.695124055541421e-05,
+      "loss": 0.1745,
+      "step": 10870
+    },
+    {
+      "grad_norm": 0.528322696685791,
+      "learning_rate": 4.6868725101776934e-05,
+      "loss": 0.2186,
+      "step": 10880
+    },
+    {
+      "grad_norm": 1.207874059677124,
+      "learning_rate": 4.678621820892567e-05,
+      "loss": 0.1922,
+      "step": 10890
+    },
+    {
+      "grad_norm": 0.3483419120311737,
+      "learning_rate": 4.670372010243111e-05,
+      "loss": 0.1717,
+      "step": 10900
+    },
+    {
+      "grad_norm": 0.37014544010162354,
+      "learning_rate": 4.662123100783992e-05,
+      "loss": 0.1639,
+      "step": 10910
+    },
+    {
+      "grad_norm": 0.48025137186050415,
+      "learning_rate": 4.653875115067415e-05,
+      "loss": 0.1727,
+      "step": 10920
+    },
+    {
+      "grad_norm": 0.4750792384147644,
+      "learning_rate": 4.6456280756430545e-05,
+      "loss": 0.1791,
+      "step": 10930
+    },
+    {
+      "grad_norm": 0.7383691072463989,
+      "learning_rate": 4.637382005058004e-05,
+      "loss": 0.1589,
+      "step": 10940
+    },
+    {
+      "grad_norm": 0.5167818069458008,
+      "learning_rate": 4.629136925856705e-05,
+      "loss": 0.1663,
+      "step": 10950
+    },
+    {
+      "grad_norm": 0.4710721969604492,
+      "learning_rate": 4.6208928605808895e-05,
+      "loss": 0.1731,
+      "step": 10960
+    },
+    {
+      "grad_norm": 0.626711905002594,
+      "learning_rate": 4.612649831769519e-05,
+      "loss": 0.1771,
+      "step": 10970
+    },
+    {
+      "grad_norm": 0.46646878123283386,
+      "learning_rate": 4.604407861958715e-05,
+      "loss": 0.1473,
+      "step": 10980
+    },
+    {
+      "grad_norm": 0.346687376499176,
+      "learning_rate": 4.5961669736817114e-05,
+      "loss": 0.1825,
+      "step": 10990
+    },
+    {
+      "grad_norm": 0.5539637207984924,
+      "learning_rate": 4.5879271894687814e-05,
+      "loss": 0.1642,
+      "step": 11000
+    },
+    {
+      "grad_norm": 0.47491246461868286,
+      "learning_rate": 4.5796885318471826e-05,
+      "loss": 0.157,
+      "step": 11010
+    },
+    {
+      "grad_norm": 0.5279638767242432,
+      "learning_rate": 4.571451023341086e-05,
+      "loss": 0.1738,
+      "step": 11020
+    },
+    {
+      "grad_norm": 0.4065079689025879,
+      "learning_rate": 4.563214686471527e-05,
+      "loss": 0.1734,
+      "step": 11030
+    },
+    {
+      "grad_norm": 0.5166419148445129,
+      "learning_rate": 4.5549795437563365e-05,
+      "loss": 0.1736,
+      "step": 11040
+    },
+    {
+      "grad_norm": 0.40257424116134644,
+      "learning_rate": 4.546745617710081e-05,
+      "loss": 0.1667,
+      "step": 11050
+    },
+    {
+      "grad_norm": 0.5338035821914673,
+      "learning_rate": 4.5385129308440014e-05,
+      "loss": 0.1631,
+      "step": 11060
+    },
+    {
+      "grad_norm": 0.46705541014671326,
+      "learning_rate": 4.530281505665944e-05,
+      "loss": 0.1777,
+      "step": 11070
+    },
+    {
+      "grad_norm": 0.6368256211280823,
+      "learning_rate": 4.5220513646803134e-05,
+      "loss": 0.1768,
+      "step": 11080
+    },
+    {
+      "grad_norm": 0.40383782982826233,
+      "learning_rate": 4.513822530388003e-05,
+      "loss": 0.1753,
+      "step": 11090
+    },
+    {
+      "grad_norm": 0.49315759539604187,
+      "learning_rate": 4.5055950252863296e-05,
+      "loss": 0.1789,
+      "step": 11100
+    },
+    {
+      "grad_norm": 0.4137597680091858,
+      "learning_rate": 4.4973688718689803e-05,
+      "loss": 0.165,
+      "step": 11110
+    },
+    {
+      "grad_norm": 0.7456216216087341,
+      "learning_rate": 4.4891440926259406e-05,
+      "loss": 0.1667,
+      "step": 11120
+    },
+    {
+      "grad_norm": 0.7160358428955078,
+      "learning_rate": 4.480920710043443e-05,
+      "loss": 0.17,
+      "step": 11130
+    },
+    {
+      "grad_norm": 0.4973923861980438,
+      "learning_rate": 4.4726987466039044e-05,
+      "loss": 0.1646,
+      "step": 11140
+    },
+    {
+      "grad_norm": 0.3063754141330719,
+      "learning_rate": 4.46447822478586e-05,
+      "loss": 0.1527,
+      "step": 11150
+    },
+    {
+      "grad_norm": 0.5491044521331787,
+      "learning_rate": 4.4562591670638974e-05,
+      "loss": 0.1882,
+      "step": 11160
+    },
+    {
+      "grad_norm": 0.6108298301696777,
+      "learning_rate": 4.4480415959086105e-05,
+      "loss": 0.1702,
+      "step": 11170
+    },
+    {
+      "grad_norm": 0.6784408688545227,
+      "learning_rate": 4.439825533786522e-05,
+      "loss": 0.1605,
+      "step": 11180
+    },
+    {
+      "grad_norm": 0.590998649597168,
+      "learning_rate": 4.431611003160035e-05,
+      "loss": 0.1842,
+      "step": 11190
+    },
+    {
+      "grad_norm": 0.4587008059024811,
+      "learning_rate": 4.4233980264873636e-05,
+      "loss": 0.1568,
+      "step": 11200
+    },
+    {
+      "grad_norm": 0.4763771891593933,
+      "learning_rate": 4.4151866262224684e-05,
+      "loss": 0.1607,
+      "step": 11210
+    },
+    {
+      "grad_norm": 0.7357561588287354,
+      "learning_rate": 4.406976824815006e-05,
+      "loss": 0.1712,
+      "step": 11220
+    },
+    {
+      "grad_norm": 0.7048335671424866,
+      "learning_rate": 4.3987686447102595e-05,
+      "loss": 0.1705,
+      "step": 11230
+    },
+    {
+      "grad_norm": 0.5278778672218323,
+      "learning_rate": 4.3905621083490804e-05,
+      "loss": 0.1709,
+      "step": 11240
+    },
+    {
+      "grad_norm": 0.44849035143852234,
+      "learning_rate": 4.3823572381678286e-05,
+      "loss": 0.1608,
+      "step": 11250
+    },
+    {
+      "grad_norm": 0.5241603851318359,
+      "learning_rate": 4.374154056598301e-05,
+      "loss": 0.1545,
+      "step": 11260
+    },
+    {
+      "grad_norm": 0.623948872089386,
+      "learning_rate": 4.3659525860676845e-05,
+      "loss": 0.1722,
+      "step": 11270
+    },
+    {
+      "grad_norm": 0.45986148715019226,
+      "learning_rate": 4.3577528489984854e-05,
+      "loss": 0.1478,
+      "step": 11280
+    },
+    {
+      "grad_norm": 0.6198714375495911,
+      "learning_rate": 4.349554867808476e-05,
+      "loss": 0.2065,
+      "step": 11290
+    },
+    {
+      "grad_norm": 0.5176485180854797,
+      "learning_rate": 4.34135866491062e-05,
+      "loss": 0.1707,
+      "step": 11300
+    },
+    {
+      "grad_norm": 0.49329012632369995,
+      "learning_rate": 4.333164262713022e-05,
+      "loss": 0.1663,
+      "step": 11310
+    },
+    {
+      "grad_norm": 0.6420264840126038,
+      "learning_rate": 4.324971683618868e-05,
+      "loss": 0.1619,
+      "step": 11320
+    },
+    {
+      "grad_norm": 0.7733703851699829,
+      "learning_rate": 4.316780950026354e-05,
+      "loss": 0.1783,
+      "step": 11330
+    },
+    {
+      "grad_norm": 0.5912997126579285,
+      "learning_rate": 4.308592084328637e-05,
+      "loss": 0.1479,
+      "step": 11340
+    },
+    {
+      "grad_norm": 0.5698673725128174,
+      "learning_rate": 4.3004051089137576e-05,
+      "loss": 0.1691,
+      "step": 11350
+    },
+    {
+      "grad_norm": 0.7396775484085083,
+      "learning_rate": 4.292220046164597e-05,
+      "loss": 0.1552,
+      "step": 11360
+    },
+    {
+      "grad_norm": 0.4461669921875,
+      "learning_rate": 4.2840369184588035e-05,
+      "loss": 0.1459,
+      "step": 11370
+    },
+    {
+      "grad_norm": 0.7288140654563904,
+      "learning_rate": 4.2758557481687345e-05,
+      "loss": 0.1787,
+      "step": 11380
+    },
+    {
+      "grad_norm": 0.353989839553833,
+      "learning_rate": 4.267676557661403e-05,
+      "loss": 0.1656,
+      "step": 11390
+    },
+    {
+      "grad_norm": 0.4888449013233185,
+      "learning_rate": 4.2594993692983955e-05,
+      "loss": 0.1682,
+      "step": 11400
+    },
+    {
+      "grad_norm": 0.6738421320915222,
+      "learning_rate": 4.251324205435837e-05,
+      "loss": 0.1802,
+      "step": 11410
+    },
+    {
+      "grad_norm": 0.6141536831855774,
+      "learning_rate": 4.243151088424312e-05,
+      "loss": 0.1671,
+      "step": 11420
+    },
+    {
+      "grad_norm": 0.5311849117279053,
+      "learning_rate": 4.234980040608813e-05,
+      "loss": 0.1721,
+      "step": 11430
+    },
+    {
+      "grad_norm": 0.5365890860557556,
+      "learning_rate": 4.22681108432867e-05,
+      "loss": 0.1685,
+      "step": 11440
+    },
+    {
+      "grad_norm": 0.4518149495124817,
+      "learning_rate": 4.2186442419174984e-05,
+      "loss": 0.1722,
+      "step": 11450
+    },
+    {
+      "grad_norm": 0.4242200255393982,
+      "learning_rate": 4.210479535703133e-05,
+      "loss": 0.1552,
+      "step": 11460
+    },
+    {
+      "grad_norm": 0.5872128009796143,
+      "learning_rate": 4.202316988007567e-05,
+      "loss": 0.1674,
+      "step": 11470
+    },
+    {
+      "grad_norm": 0.5117030143737793,
+      "learning_rate": 4.194156621146901e-05,
+      "loss": 0.1537,
+      "step": 11480
+    },
+    {
+      "grad_norm": 0.522005021572113,
+      "learning_rate": 4.1859984574312596e-05,
+      "loss": 0.1609,
+      "step": 11490
+    },
+    {
+      "grad_norm": 0.7396453022956848,
+      "learning_rate": 4.177842519164752e-05,
+      "loss": 0.1777,
+      "step": 11500
+    },
+    {
+      "grad_norm": 0.5400175452232361,
+      "learning_rate": 4.169688828645404e-05,
+      "loss": 0.1679,
+      "step": 11510
+    },
+    {
+      "grad_norm": 0.7011004090309143,
+      "learning_rate": 4.161537408165092e-05,
+      "loss": 0.1581,
+      "step": 11520
+    },
+    {
+      "grad_norm": 0.6122260689735413,
+      "learning_rate": 4.1533882800094924e-05,
+      "loss": 0.1647,
+      "step": 11530
+    },
+    {
+      "grad_norm": 0.38532474637031555,
+      "learning_rate": 4.145241466458005e-05,
+      "loss": 0.1646,
+      "step": 11540
+    },
+    {
+      "grad_norm": 0.41623345017433167,
+      "learning_rate": 4.13709698978371e-05,
+      "loss": 0.155,
+      "step": 11550
+    },
+    {
+      "grad_norm": 0.4000478684902191,
+      "learning_rate": 4.1289548722532944e-05,
+      "loss": 0.1807,
+      "step": 11560
+    },
+    {
+      "grad_norm": 0.6748583912849426,
+      "learning_rate": 4.120815136126999e-05,
+      "loss": 0.1505,
+      "step": 11570
+    },
+    {
+      "grad_norm": 0.40557560324668884,
+      "learning_rate": 4.112677803658548e-05,
+      "loss": 0.1628,
+      "step": 11580
+    },
+    {
+      "grad_norm": 0.4959176182746887,
+      "learning_rate": 4.1045428970951e-05,
+      "loss": 0.1696,
+      "step": 11590
+    },
+    {
+      "grad_norm": 0.707023024559021,
+      "learning_rate": 4.0964104386771785e-05,
+      "loss": 0.1602,
+      "step": 11600
+    },
+    {
+      "grad_norm": 0.4446443021297455,
+      "learning_rate": 4.0882804506386144e-05,
+      "loss": 0.1694,
+      "step": 11610
+    },
+    {
+      "grad_norm": 0.5900254845619202,
+      "learning_rate": 4.080152955206485e-05,
+      "loss": 0.1667,
+      "step": 11620
+    },
+    {
+      "grad_norm": 0.4790717363357544,
+      "learning_rate": 4.0720279746010505e-05,
+      "loss": 0.1884,
+      "step": 11630
+    },
+    {
+      "grad_norm": 0.5196569561958313,
+      "learning_rate": 4.063905531035699e-05,
+      "loss": 0.1545,
+      "step": 11640
+    },
+    {
+      "grad_norm": 0.4423220753669739,
+      "learning_rate": 4.055785646716882e-05,
+      "loss": 0.1424,
+      "step": 11650
+    },
+    {
+      "grad_norm": 0.8388225436210632,
+      "learning_rate": 4.047668343844051e-05,
+      "loss": 0.1539,
+      "step": 11660
+    },
+    {
+      "grad_norm": 0.4646672010421753,
+      "learning_rate": 4.039553644609604e-05,
+      "loss": 0.1683,
+      "step": 11670
+    },
+    {
+      "grad_norm": 0.4878402650356293,
+      "learning_rate": 4.0314415711988176e-05,
+      "loss": 0.1458,
+      "step": 11680
+    },
+    {
+      "grad_norm": 1.0563349723815918,
+      "learning_rate": 4.023332145789792e-05,
+      "loss": 0.2016,
+      "step": 11690
+    },
+    {
+      "grad_norm": 0.5814833641052246,
+      "learning_rate": 4.015225390553385e-05,
+      "loss": 0.1598,
+      "step": 11700
+    },
+    {
+      "grad_norm": 0.5816835761070251,
+      "learning_rate": 4.007121327653158e-05,
+      "loss": 0.1692,
+      "step": 11710
+    },
+    {
+      "grad_norm": 0.5406365394592285,
+      "learning_rate": 3.9990199792453064e-05,
+      "loss": 0.1523,
+      "step": 11720
+    },
+    {
+      "grad_norm": 0.5402758121490479,
+      "learning_rate": 3.9909213674786103e-05,
+      "loss": 0.1677,
+      "step": 11730
+    },
+    {
+      "grad_norm": 0.5855444073677063,
+      "learning_rate": 3.982825514494363e-05,
+      "loss": 0.1691,
+      "step": 11740
+    },
+    {
+      "grad_norm": 0.43407246470451355,
+      "learning_rate": 3.974732442426319e-05,
+      "loss": 0.163,
+      "step": 11750
+    },
+    {
+      "grad_norm": 0.38541123270988464,
+      "learning_rate": 3.966642173400629e-05,
+      "loss": 0.1709,
+      "step": 11760
+    },
+    {
+      "grad_norm": 0.38804277777671814,
+      "learning_rate": 3.9585547295357764e-05,
+      "loss": 0.1567,
+      "step": 11770
+    },
+    {
+      "grad_norm": 0.45247089862823486,
+      "learning_rate": 3.950470132942526e-05,
+      "loss": 0.1642,
+      "step": 11780
+    },
+    {
+      "grad_norm": 0.43455731868743896,
+      "learning_rate": 3.942388405723856e-05,
+      "loss": 0.1878,
+      "step": 11790
+    },
+    {
+      "grad_norm": 0.47529157996177673,
+      "learning_rate": 3.9343095699749e-05,
+      "loss": 0.1792,
+      "step": 11800
+    },
+    {
+      "grad_norm": 0.4261329472064972,
+      "learning_rate": 3.9262336477828874e-05,
+      "loss": 0.181,
+      "step": 11810
+    },
+    {
+      "grad_norm": 0.49119898676872253,
+      "learning_rate": 3.9181606612270794e-05,
+      "loss": 0.1826,
+      "step": 11820
+    },
+    {
+      "grad_norm": 0.5408397316932678,
+      "learning_rate": 3.910090632378713e-05,
+      "loss": 0.1595,
+      "step": 11830
+    },
+    {
+      "grad_norm": 0.4073582589626312,
+      "learning_rate": 3.90202358330094e-05,
+      "loss": 0.1558,
+      "step": 11840
+    },
+    {
+      "grad_norm": 0.47378021478652954,
+      "learning_rate": 3.8939595360487656e-05,
+      "loss": 0.1594,
+      "step": 11850
+    },
+    {
+      "grad_norm": 0.5382238626480103,
+      "learning_rate": 3.885898512668984e-05,
+      "loss": 0.1641,
+      "step": 11860
+    },
+    {
+      "grad_norm": 0.5228734016418457,
+      "learning_rate": 3.877840535200127e-05,
+      "loss": 0.1669,
+      "step": 11870
+    },
+    {
+      "grad_norm": 0.44776326417922974,
+      "learning_rate": 3.869785625672397e-05,
+      "loss": 0.1645,
+      "step": 11880
+    },
+    {
+      "grad_norm": 0.6345827579498291,
+      "learning_rate": 3.8617338061076094e-05,
+      "loss": 0.1694,
+      "step": 11890
+    },
+    {
+      "grad_norm": 0.4533159136772156,
+      "learning_rate": 3.853685098519132e-05,
+      "loss": 0.1763,
+      "step": 11900
+    },
+    {
+      "grad_norm": 0.4740560054779053,
+      "learning_rate": 3.845639524911823e-05,
+      "loss": 0.1733,
+      "step": 11910
+    },
+    {
+      "grad_norm": 0.5324491262435913,
+      "learning_rate": 3.837597107281974e-05,
+      "loss": 0.1399,
+      "step": 11920
+    },
+    {
+      "grad_norm": 0.4555518925189972,
+      "learning_rate": 3.829557867617247e-05,
+      "loss": 0.1693,
+      "step": 11930
+    },
+    {
+      "grad_norm": 0.7510755062103271,
+      "learning_rate": 3.821521827896618e-05,
+      "loss": 0.1816,
+      "step": 11940
+    },
+    {
+      "grad_norm": 0.4699271023273468,
+      "learning_rate": 3.81348901009031e-05,
+      "loss": 0.1752,
+      "step": 11950
+    },
+    {
+      "grad_norm": 0.40360531210899353,
+      "learning_rate": 3.805459436159741e-05,
+      "loss": 0.171,
+      "step": 11960
+    },
+    {
+      "grad_norm": 0.5797461867332458,
+      "learning_rate": 3.797433128057461e-05,
+      "loss": 0.1639,
+      "step": 11970
+    },
+    {
+      "grad_norm": 0.7412588000297546,
+      "learning_rate": 3.789410107727089e-05,
+      "loss": 0.1582,
+      "step": 11980
+    },
+    {
+      "grad_norm": 0.46924036741256714,
+      "learning_rate": 3.781390397103257e-05,
+      "loss": 0.1719,
+      "step": 11990
+    },
+    {
+      "grad_norm": 0.4402959942817688,
+      "learning_rate": 3.7733740181115455e-05,
+      "loss": 0.1636,
+      "step": 12000
+    },
+    {
+      "grad_norm": 0.5117281079292297,
+      "learning_rate": 3.7653609926684306e-05,
+      "loss": 0.1623,
+      "step": 12010
+    },
+    {
+      "grad_norm": 0.4092056453227997,
+      "learning_rate": 3.757351342681217e-05,
+      "loss": 0.1528,
+      "step": 12020
+    },
+    {
+      "grad_norm": 0.4096065163612366,
+      "learning_rate": 3.749345090047982e-05,
+      "loss": 0.1676,
+      "step": 12030
+    },
+    {
+      "grad_norm": 0.5281375646591187,
+      "learning_rate": 3.741342256657515e-05,
+      "loss": 0.1603,
+      "step": 12040
+    },
+    {
+      "grad_norm": 0.43640759587287903,
+      "learning_rate": 3.7333428643892567e-05,
+      "loss": 0.1722,
+      "step": 12050
+    },
+    {
+      "grad_norm": 0.49961090087890625,
+      "learning_rate": 3.725346935113239e-05,
+      "loss": 0.1732,
+      "step": 12060
+    },
+    {
+      "grad_norm": 0.430685818195343,
+      "learning_rate": 3.717354490690029e-05,
+      "loss": 0.1494,
+      "step": 12070
+    },
+    {
+      "grad_norm": 0.6747862100601196,
+      "learning_rate": 3.709365552970664e-05,
+      "loss": 0.181,
+      "step": 12080
+    },
+    {
+      "grad_norm": 0.42452794313430786,
+      "learning_rate": 3.7013801437965945e-05,
+      "loss": 0.1693,
+      "step": 12090
+    },
+    {
+      "grad_norm": 0.39529722929000854,
+      "learning_rate": 3.693398284999623e-05,
+      "loss": 0.1595,
+      "step": 12100
+    },
+    {
+      "grad_norm": 0.4539473056793213,
+      "learning_rate": 3.6854199984018484e-05,
+      "loss": 0.1622,
+      "step": 12110
+    },
+    {
+      "grad_norm": 0.3776356875896454,
+      "learning_rate": 3.677445305815601e-05,
+      "loss": 0.1477,
+      "step": 12120
+    },
+    {
+      "grad_norm": 0.4259239733219147,
+      "learning_rate": 3.669474229043387e-05,
+      "loss": 0.1688,
+      "step": 12130
+    },
+    {
+      "grad_norm": 0.45569169521331787,
+      "learning_rate": 3.6615067898778235e-05,
+      "loss": 0.1519,
+      "step": 12140
+    },
+    {
+      "grad_norm": 0.5427250266075134,
+      "learning_rate": 3.6535430101015866e-05,
+      "loss": 0.173,
+      "step": 12150
+    },
+    {
+      "grad_norm": 0.505456268787384,
+      "learning_rate": 3.645582911487345e-05,
+      "loss": 0.1709,
+      "step": 12160
+    },
+    {
+      "grad_norm": 0.5890184640884399,
+      "learning_rate": 3.637626515797706e-05,
+      "loss": 0.1599,
+      "step": 12170
+    },
+    {
+      "grad_norm": 0.5894227027893066,
+      "learning_rate": 3.629673844785152e-05,
+      "loss": 0.1649,
+      "step": 12180
+    },
+    {
+      "grad_norm": 0.701718270778656,
+      "learning_rate": 3.621724920191979e-05,
+      "loss": 0.1804,
+      "step": 12190
+    },
+    {
+      "grad_norm": 0.6606631875038147,
+      "learning_rate": 3.6137797637502444e-05,
+      "loss": 0.1685,
+      "step": 12200
+    },
+    {
+      "grad_norm": 0.6727001667022705,
+      "learning_rate": 3.6058383971817035e-05,
+      "loss": 0.1778,
+      "step": 12210
+    },
+    {
+      "grad_norm": 0.5566079020500183,
+      "learning_rate": 3.59790084219775e-05,
+      "loss": 0.1709,
+      "step": 12220
+    },
+    {
+      "grad_norm": 0.46569496393203735,
+      "learning_rate": 3.589967120499353e-05,
+      "loss": 0.168,
+      "step": 12230
+    },
+    {
+      "grad_norm": 0.49469733238220215,
+      "learning_rate": 3.5820372537770075e-05,
+      "loss": 0.1671,
+      "step": 12240
+    },
+    {
+      "grad_norm": 0.4463235139846802,
+      "learning_rate": 3.5741112637106655e-05,
+      "loss": 0.1766,
+      "step": 12250
+    },
+    {
+      "grad_norm": 0.5417762398719788,
+      "learning_rate": 3.5661891719696804e-05,
+      "loss": 0.1782,
+      "step": 12260
+    },
+    {
+      "grad_norm": 0.5094807147979736,
+      "learning_rate": 3.5582710002127504e-05,
+      "loss": 0.1764,
+      "step": 12270
+    },
+    {
+      "grad_norm": 0.4968473017215729,
+      "learning_rate": 3.550356770087853e-05,
+      "loss": 0.174,
+      "step": 12280
+    },
+    {
+      "grad_norm": 0.5379506349563599,
+      "learning_rate": 3.5424465032321914e-05,
+      "loss": 0.1664,
+      "step": 12290
+    },
+    {
+      "grad_norm": 0.41273626685142517,
+      "learning_rate": 3.5345402212721335e-05,
+      "loss": 0.1639,
+      "step": 12300
+    },
+    {
+      "grad_norm": 0.4517160952091217,
+      "learning_rate": 3.526637945823152e-05,
+      "loss": 0.1767,
+      "step": 12310
+    },
+    {
+      "grad_norm": 0.5400528907775879,
+      "learning_rate": 3.518739698489767e-05,
+      "loss": 0.1831,
+      "step": 12320
+    },
+    {
+      "grad_norm": 0.32836925983428955,
+      "learning_rate": 3.510845500865485e-05,
+      "loss": 0.1615,
+      "step": 12330
+    },
+    {
+      "grad_norm": 0.35485437512397766,
+      "learning_rate": 3.502955374532739e-05,
+      "loss": 0.1695,
+      "step": 12340
+    },
+    {
+      "grad_norm": 0.5808596611022949,
+      "learning_rate": 3.495069341062836e-05,
+      "loss": 0.1713,
+      "step": 12350
+    },
+    {
+      "grad_norm": 0.5171905159950256,
+      "learning_rate": 3.4871874220158896e-05,
+      "loss": 0.156,
+      "step": 12360
+    },
+    {
+      "grad_norm": 0.5654253959655762,
+      "learning_rate": 3.479309638940762e-05,
+      "loss": 0.1678,
+      "step": 12370
+    },
+    {
+      "grad_norm": 0.5749889612197876,
+      "learning_rate": 3.4714360133750146e-05,
+      "loss": 0.1743,
+      "step": 12380
+    },
+    {
+      "grad_norm": 0.40622127056121826,
+      "learning_rate": 3.463566566844839e-05,
+      "loss": 0.1583,
+      "step": 12390
+    },
+    {
+      "grad_norm": 0.5374818444252014,
+      "learning_rate": 3.4557013208650016e-05,
+      "loss": 0.1596,
+      "step": 12400
+    },
+    {
+      "grad_norm": 0.5075781345367432,
+      "learning_rate": 3.4478402969387857e-05,
+      "loss": 0.1701,
+      "step": 12410
+    },
+    {
+      "grad_norm": 0.4900453984737396,
+      "learning_rate": 3.4399835165579266e-05,
+      "loss": 0.1554,
+      "step": 12420
+    },
+    {
+      "grad_norm": 0.47955799102783203,
+      "learning_rate": 3.4321310012025645e-05,
+      "loss": 0.1727,
+      "step": 12430
+    },
+    {
+      "grad_norm": 0.43693438172340393,
+      "learning_rate": 3.424282772341176e-05,
+      "loss": 0.1636,
+      "step": 12440
+    },
+    {
+      "grad_norm": 0.4201594889163971,
+      "learning_rate": 3.416438851430519e-05,
+      "loss": 0.1801,
+      "step": 12450
+    },
+    {
+      "grad_norm": 0.5778195858001709,
+      "learning_rate": 3.408599259915577e-05,
+      "loss": 0.1749,
+      "step": 12460
+    },
+    {
+      "grad_norm": 0.4619051516056061,
+      "learning_rate": 3.400764019229487e-05,
+      "loss": 0.1578,
+      "step": 12470
+    },
+    {
+      "grad_norm": 0.5605830550193787,
+      "learning_rate": 3.3929331507935035e-05,
+      "loss": 0.1682,
+      "step": 12480
+    },
+    {
+      "grad_norm": 0.453199177980423,
+      "learning_rate": 3.3851066760169196e-05,
+      "loss": 0.1627,
+      "step": 12490
+    },
+    {
+      "grad_norm": 0.5066860914230347,
+      "learning_rate": 3.377284616297021e-05,
+      "loss": 0.1572,
+      "step": 12500
+    },
+    {
+      "grad_norm": 0.4680129885673523,
+      "learning_rate": 3.3694669930190166e-05,
+      "loss": 0.1483,
+      "step": 12510
+    },
+    {
+      "grad_norm": 0.5394432544708252,
+      "learning_rate": 3.36165382755599e-05,
+      "loss": 0.1606,
+      "step": 12520
+    },
+    {
+      "grad_norm": 0.46569517254829407,
+      "learning_rate": 3.35384514126884e-05,
+      "loss": 0.1688,
+      "step": 12530
+    },
+    {
+      "grad_norm": 0.5426195859909058,
+      "learning_rate": 3.3460409555062154e-05,
+      "loss": 0.1671,
+      "step": 12540
+    },
+    {
+      "grad_norm": 0.5004144310951233,
+      "learning_rate": 3.3382412916044645e-05,
+      "loss": 0.1547,
+      "step": 12550
+    },
+    {
+      "grad_norm": 0.7353156805038452,
+      "learning_rate": 3.330446170887566e-05,
+      "loss": 0.1662,
+      "step": 12560
+    },
+    {
+      "grad_norm": 0.4313611686229706,
+      "learning_rate": 3.3226556146670834e-05,
+      "loss": 0.1537,
+      "step": 12570
+    },
+    {
+      "grad_norm": 0.4920187294483185,
+      "learning_rate": 3.314869644242102e-05,
+      "loss": 0.1711,
+      "step": 12580
+    },
+    {
+      "grad_norm": 0.6050612926483154,
+      "learning_rate": 3.3070882808991674e-05,
+      "loss": 0.1664,
+      "step": 12590
+    },
+    {
+      "grad_norm": 0.5446799993515015,
+      "learning_rate": 3.2993115459122305e-05,
+      "loss": 0.18,
+      "step": 12600
+    },
+    {
+      "grad_norm": 0.5016178488731384,
+      "learning_rate": 3.2915394605425835e-05,
+      "loss": 0.1769,
+      "step": 12610
+    },
+    {
+      "grad_norm": 0.603927493095398,
+      "learning_rate": 3.283772046038816e-05,
+      "loss": 0.1601,
+      "step": 12620
+    },
+    {
+      "grad_norm": 0.5412598848342896,
+      "learning_rate": 3.276009323636739e-05,
+      "loss": 0.1719,
+      "step": 12630
+    },
+    {
+      "grad_norm": 0.648412823677063,
+      "learning_rate": 3.268251314559344e-05,
+      "loss": 0.1722,
+      "step": 12640
+    },
+    {
+      "grad_norm": 0.5609948039054871,
+      "learning_rate": 3.2604980400167254e-05,
+      "loss": 0.1958,
+      "step": 12650
+    },
+    {
+      "grad_norm": 0.5060745477676392,
+      "learning_rate": 3.252749521206042e-05,
+      "loss": 0.1708,
+      "step": 12660
+    },
+    {
+      "grad_norm": 0.4371897578239441,
+      "learning_rate": 3.2450057793114494e-05,
+      "loss": 0.1904,
+      "step": 12670
+    },
+    {
+      "grad_norm": 0.40514662861824036,
+      "learning_rate": 3.2372668355040435e-05,
+      "loss": 0.1582,
+      "step": 12680
+    },
+    {
+      "grad_norm": 0.3414977788925171,
+      "learning_rate": 3.2295327109418005e-05,
+      "loss": 0.1698,
+      "step": 12690
+    },
+    {
+      "grad_norm": 0.4250338673591614,
+      "learning_rate": 3.221803426769518e-05,
+      "loss": 0.1658,
+      "step": 12700
+    },
+    {
+      "grad_norm": 0.5071417093276978,
+      "learning_rate": 3.214079004118768e-05,
+      "loss": 0.1702,
+      "step": 12710
+    },
+    {
+      "grad_norm": 0.36038920283317566,
+      "learning_rate": 3.2063594641078234e-05,
+      "loss": 0.1623,
+      "step": 12720
+    },
+    {
+      "grad_norm": 0.380938321352005,
+      "learning_rate": 3.198644827841616e-05,
+      "loss": 0.1842,
+      "step": 12730
+    },
+    {
+      "grad_norm": 0.6350394487380981,
+      "learning_rate": 3.1909351164116654e-05,
+      "loss": 0.1667,
+      "step": 12740
+    },
+    {
+      "grad_norm": 0.4462554156780243,
+      "learning_rate": 3.183230350896026e-05,
+      "loss": 0.1591,
+      "step": 12750
+    },
+    {
+      "grad_norm": 0.5072724223136902,
+      "learning_rate": 3.1755305523592337e-05,
+      "loss": 0.1601,
+      "step": 12760
+    },
+    {
+      "grad_norm": 0.5472372770309448,
+      "learning_rate": 3.167835741852245e-05,
+      "loss": 0.166,
+      "step": 12770
+    },
+    {
+      "grad_norm": 0.49258583784103394,
+      "learning_rate": 3.160145940412378e-05,
+      "loss": 0.1821,
+      "step": 12780
+    },
+    {
+      "grad_norm": 0.5023723244667053,
+      "learning_rate": 3.1524611690632545e-05,
+      "loss": 0.1581,
+      "step": 12790
+    },
+    {
+      "grad_norm": 0.662909746170044,
+      "learning_rate": 3.144781448814746e-05,
+      "loss": 0.1631,
+      "step": 12800
+    },
+    {
+      "grad_norm": 0.3885636627674103,
+      "learning_rate": 3.1371068006629145e-05,
+      "loss": 0.1779,
+      "step": 12810
+    },
+    {
+      "grad_norm": 0.6170032620429993,
+      "learning_rate": 3.129437245589956e-05,
+      "loss": 0.1711,
+      "step": 12820
+    },
+    {
+      "grad_norm": 0.6143956184387207,
+      "learning_rate": 3.121772804564143e-05,
+      "loss": 0.1814,
+      "step": 12830
+    },
+    {
+      "grad_norm": 0.7126495242118835,
+      "learning_rate": 3.11411349853976e-05,
+      "loss": 0.1853,
+      "step": 12840
+    },
+    {
+      "grad_norm": 0.49642619490623474,
+      "learning_rate": 3.10645934845706e-05,
+      "loss": 0.1622,
+      "step": 12850
+    },
+    {
+      "grad_norm": 0.5137234926223755,
+      "learning_rate": 3.098810375242196e-05,
+      "loss": 0.1751,
+      "step": 12860
+    },
+    {
+      "grad_norm": 0.8113284111022949,
+      "learning_rate": 3.0911665998071704e-05,
+      "loss": 0.1607,
+      "step": 12870
+    },
+    {
+      "grad_norm": 0.4306753873825073,
+      "learning_rate": 3.083528043049774e-05,
+      "loss": 0.1765,
+      "step": 12880
+    },
+    {
+      "grad_norm": 0.6716392636299133,
+      "learning_rate": 3.0758947258535255e-05,
+      "loss": 0.1968,
+      "step": 12890
+    },
+    {
+      "grad_norm": 0.45412057638168335,
+      "learning_rate": 3.068266669087625e-05,
+      "loss": 0.1529,
+      "step": 12900
+    },
+    {
+      "grad_norm": 0.40364691615104675,
+      "learning_rate": 3.060643893606887e-05,
+      "loss": 0.1695,
+      "step": 12910
+    },
+    {
+      "grad_norm": 0.428393691778183,
+      "learning_rate": 3.053026420251693e-05,
+      "loss": 0.1689,
+      "step": 12920
+    },
+    {
+      "grad_norm": 0.5287341475486755,
+      "learning_rate": 3.0454142698479183e-05,
+      "loss": 0.1709,
+      "step": 12930
+    },
+    {
+      "grad_norm": 0.45965078473091125,
+      "learning_rate": 3.0378074632068954e-05,
+      "loss": 0.1699,
+      "step": 12940
+    },
+    {
+      "grad_norm": 0.5724154710769653,
+      "learning_rate": 3.0302060211253408e-05,
+      "loss": 0.1825,
+      "step": 12950
+    },
+    {
+      "grad_norm": 0.46713191270828247,
+      "learning_rate": 3.0226099643853073e-05,
+      "loss": 0.1687,
+      "step": 12960
+    },
+    {
+      "grad_norm": 0.351250022649765,
+      "learning_rate": 3.0150193137541283e-05,
+      "loss": 0.1805,
+      "step": 12970
+    },
+    {
+      "grad_norm": 0.5446012616157532,
+      "learning_rate": 3.0074340899843467e-05,
+      "loss": 0.1678,
+      "step": 12980
+    },
+    {
+      "grad_norm": 0.44407615065574646,
+      "learning_rate": 2.999854313813677e-05,
+      "loss": 0.1563,
+      "step": 12990
+    },
+    {
+      "grad_norm": 0.5766637325286865,
+      "learning_rate": 2.9922800059649382e-05,
+      "loss": 0.1504,
+      "step": 13000
+    },
+    {
+      "grad_norm": 0.5381314754486084,
+      "learning_rate": 2.9847111871459976e-05,
+      "loss": 0.1833,
+      "step": 13010
+    },
+    {
+      "grad_norm": 0.5126599669456482,
+      "learning_rate": 2.977147878049721e-05,
+      "loss": 0.1779,
+      "step": 13020
+    },
+    {
+      "grad_norm": 0.4859880208969116,
+      "learning_rate": 2.9695900993539006e-05,
+      "loss": 0.1689,
+      "step": 13030
+    },
+    {
+      "grad_norm": 0.4755982756614685,
+      "learning_rate": 2.9620378717212183e-05,
+      "loss": 0.1598,
+      "step": 13040
+    },
+    {
+      "grad_norm": 0.5181517601013184,
+      "learning_rate": 2.9544912157991745e-05,
+      "loss": 0.1865,
+      "step": 13050
+    },
+    {
+      "grad_norm": 0.5027256608009338,
+      "learning_rate": 2.9469501522200405e-05,
+      "loss": 0.1622,
+      "step": 13060
+    },
+    {
+      "grad_norm": 0.6058418154716492,
+      "learning_rate": 2.9394147016007946e-05,
+      "loss": 0.1509,
+      "step": 13070
+    },
+    {
+      "grad_norm": 0.48936083912849426,
+      "learning_rate": 2.9318848845430702e-05,
+      "loss": 0.2007,
+      "step": 13080
+    },
+    {
+      "grad_norm": 0.4124186933040619,
+      "learning_rate": 2.9243607216331013e-05,
+      "loss": 0.1858,
+      "step": 13090
+    },
+    {
+      "grad_norm": 0.4763948619365692,
+      "learning_rate": 2.916842233441661e-05,
+      "loss": 0.1787,
+      "step": 13100
+    },
+    {
+      "grad_norm": 0.7799022793769836,
+      "learning_rate": 2.90932944052401e-05,
+      "loss": 0.1864,
+      "step": 13110
+    },
+    {
+      "grad_norm": 0.4890362322330475,
+      "learning_rate": 2.9018223634198354e-05,
+      "loss": 0.1566,
+      "step": 13120
+    },
+    {
+      "grad_norm": 0.43637776374816895,
+      "learning_rate": 2.8943210226532025e-05,
+      "loss": 0.1512,
+      "step": 13130
+    },
+    {
+      "grad_norm": 0.6873915195465088,
+      "learning_rate": 2.8868254387324857e-05,
+      "loss": 0.1649,
+      "step": 13140
+    },
+    {
+      "grad_norm": 0.5120935440063477,
+      "learning_rate": 2.8793356321503306e-05,
+      "loss": 0.1709,
+      "step": 13150
+    },
+    {
+      "grad_norm": 0.4122109115123749,
+      "learning_rate": 2.87185162338358e-05,
+      "loss": 0.1743,
+      "step": 13160
+    },
+    {
+      "grad_norm": 0.5945339202880859,
+      "learning_rate": 2.8643734328932253e-05,
+      "loss": 0.1739,
+      "step": 13170
+    },
+    {
+      "grad_norm": 0.592860221862793,
+      "learning_rate": 2.856901081124359e-05,
+      "loss": 0.1686,
+      "step": 13180
+    },
+    {
+      "grad_norm": 0.5247403979301453,
+      "learning_rate": 2.8494345885061002e-05,
+      "loss": 0.1544,
+      "step": 13190
+    },
+    {
+      "grad_norm": 0.40072691440582275,
+      "learning_rate": 2.8419739754515616e-05,
+      "loss": 0.1764,
+      "step": 13200
+    },
+    {
+      "grad_norm": 0.5615106225013733,
+      "learning_rate": 2.8345192623577666e-05,
+      "loss": 0.1572,
+      "step": 13210
+    },
+    {
+      "grad_norm": 0.5447139143943787,
+      "learning_rate": 2.8270704696056193e-05,
+      "loss": 0.1567,
+      "step": 13220
+    },
+    {
+      "grad_norm": 0.44524502754211426,
+      "learning_rate": 2.8196276175598367e-05,
+      "loss": 0.1676,
+      "step": 13230
+    },
+    {
+      "grad_norm": 0.5331522226333618,
+      "learning_rate": 2.8121907265688884e-05,
+      "loss": 0.1751,
+      "step": 13240
+    },
+    {
+      "grad_norm": 0.42695844173431396,
+      "learning_rate": 2.804759816964957e-05,
+      "loss": 0.169,
+      "step": 13250
+    },
+    {
+      "grad_norm": 0.5069171190261841,
+      "learning_rate": 2.797334909063857e-05,
+      "loss": 0.1612,
+      "step": 13260
+    },
+    {
+      "grad_norm": 0.5392596125602722,
+      "learning_rate": 2.7899160231650056e-05,
+      "loss": 0.1784,
+      "step": 13270
+    },
+    {
+      "grad_norm": 0.5258902311325073,
+      "learning_rate": 2.7825031795513585e-05,
+      "loss": 0.1629,
+      "step": 13280
+    },
+    {
+      "grad_norm": 0.5167690515518188,
+      "learning_rate": 2.775096398489341e-05,
+      "loss": 0.1647,
+      "step": 13290
+    },
+    {
+      "grad_norm": 0.6430761218070984,
+      "learning_rate": 2.7676957002288163e-05,
+      "loss": 0.1582,
+      "step": 13300
+    },
+    {
+      "grad_norm": 0.3945392966270447,
+      "learning_rate": 2.760301105003003e-05,
+      "loss": 0.1513,
+      "step": 13310
+    },
+    {
+      "grad_norm": 0.657512903213501,
+      "learning_rate": 2.752912633028446e-05,
+      "loss": 0.1753,
+      "step": 13320
+    },
+    {
+      "grad_norm": 0.6438477635383606,
+      "learning_rate": 2.7455303045049474e-05,
+      "loss": 0.1536,
+      "step": 13330
+    },
+    {
+      "grad_norm": 0.5524279475212097,
+      "learning_rate": 2.7381541396155098e-05,
+      "loss": 0.1628,
+      "step": 13340
+    },
+    {
+      "grad_norm": 0.5443811416625977,
+      "learning_rate": 2.730784158526286e-05,
+      "loss": 0.1697,
+      "step": 13350
+    },
+    {
+      "grad_norm": 0.5870099067687988,
+      "learning_rate": 2.723420381386521e-05,
+      "loss": 0.1722,
+      "step": 13360
+    },
+    {
+      "grad_norm": 0.6483100056648254,
+      "learning_rate": 2.7160628283285018e-05,
+      "loss": 0.1772,
+      "step": 13370
+    },
+    {
+      "grad_norm": 0.5268969535827637,
+      "learning_rate": 2.7087115194675007e-05,
+      "loss": 0.1549,
+      "step": 13380
+    },
+    {
+      "grad_norm": 0.4931395351886749,
+      "learning_rate": 2.701366474901712e-05,
+      "loss": 0.1642,
+      "step": 13390
+    },
+    {
+      "grad_norm": 0.4483206570148468,
+      "learning_rate": 2.6940277147122085e-05,
+      "loss": 0.1664,
+      "step": 13400
+    },
+    {
+      "grad_norm": 0.709114670753479,
+      "learning_rate": 2.686695258962878e-05,
+      "loss": 0.1651,
+      "step": 13410
+    },
+    {
+      "grad_norm": 0.48009762167930603,
+      "learning_rate": 2.679369127700375e-05,
+      "loss": 0.1657,
+      "step": 13420
+    },
+    {
+      "grad_norm": 0.47486114501953125,
+      "learning_rate": 2.672049340954067e-05,
+      "loss": 0.1551,
+      "step": 13430
+    },
+    {
+      "grad_norm": 0.6969658732414246,
+      "learning_rate": 2.6647359187359676e-05,
+      "loss": 0.1712,
+      "step": 13440
+    },
+    {
+      "grad_norm": 0.5913156270980835,
+      "learning_rate": 2.6574288810406946e-05,
+      "loss": 0.1721,
+      "step": 13450
+    },
+    {
+      "grad_norm": 0.6500833630561829,
+      "learning_rate": 2.6501282478454083e-05,
+      "loss": 0.152,
+      "step": 13460
+    },
+    {
+      "grad_norm": 0.664987325668335,
+      "learning_rate": 2.6428340391097618e-05,
+      "loss": 0.1721,
+      "step": 13470
+    },
+    {
+      "grad_norm": 0.6221473813056946,
+      "learning_rate": 2.6355462747758485e-05,
+      "loss": 0.1593,
+      "step": 13480
+    },
+    {
+      "grad_norm": 0.9812434911727905,
+      "learning_rate": 2.6282649747681304e-05,
+      "loss": 0.1624,
+      "step": 13490
+    },
+    {
+      "grad_norm": 0.5992206335067749,
+      "learning_rate": 2.620990158993406e-05,
+      "loss": 0.1512,
+      "step": 13500
+    },
+    {
+      "grad_norm": 0.709544837474823,
+      "learning_rate": 2.6137218473407477e-05,
+      "loss": 0.159,
+      "step": 13510
+    },
+    {
+      "grad_norm": 0.6545948386192322,
+      "learning_rate": 2.606460059681436e-05,
+      "loss": 0.1808,
+      "step": 13520
+    },
+    {
+      "grad_norm": 0.4354262351989746,
+      "learning_rate": 2.599204815868928e-05,
+      "loss": 0.1612,
+      "step": 13530
+    },
+    {
+      "grad_norm": 0.41133973002433777,
+      "learning_rate": 2.5919561357387756e-05,
+      "loss": 0.1628,
+      "step": 13540
+    },
+    {
+      "grad_norm": 0.5813371539115906,
+      "learning_rate": 2.5847140391085972e-05,
+      "loss": 0.1603,
+      "step": 13550
+    },
+    {
+      "grad_norm": 0.47663813829421997,
+      "learning_rate": 2.5774785457780103e-05,
+      "loss": 0.1623,
+      "step": 13560
+    },
+    {
+      "grad_norm": 0.5030827522277832,
+      "learning_rate": 2.5702496755285753e-05,
+      "loss": 0.1562,
+      "step": 13570
+    },
+    {
+      "grad_norm": 0.8624659180641174,
+      "learning_rate": 2.5630274481237483e-05,
+      "loss": 0.1309,
+      "step": 13580
+    },
+    {
+      "grad_norm": 0.4447266161441803,
+      "learning_rate": 2.5558118833088197e-05,
+      "loss": 0.1625,
+      "step": 13590
+    },
+    {
+      "grad_norm": 0.44729140400886536,
+      "learning_rate": 2.548603000810872e-05,
+      "loss": 0.1746,
+      "step": 13600
+    },
+    {
+      "grad_norm": 0.6088227033615112,
+      "learning_rate": 2.5414008203387152e-05,
+      "loss": 0.1737,
+      "step": 13610
+    },
+    {
+      "grad_norm": 0.5989233255386353,
+      "learning_rate": 2.534205361582834e-05,
+      "loss": 0.1663,
+      "step": 13620
+    },
+    {
+      "grad_norm": 0.4933260381221771,
+      "learning_rate": 2.527016644215338e-05,
+      "loss": 0.1589,
+      "step": 13630
+    },
+    {
+      "grad_norm": 0.5559582710266113,
+      "learning_rate": 2.519834687889905e-05,
+      "loss": 0.1503,
+      "step": 13640
+    },
+    {
+      "grad_norm": 0.45066747069358826,
+      "learning_rate": 2.5126595122417295e-05,
+      "loss": 0.1591,
+      "step": 13650
+    },
+    {
+      "grad_norm": 0.6676121950149536,
+      "learning_rate": 2.5054911368874713e-05,
+      "loss": 0.1432,
+      "step": 13660
+    },
+    {
+      "grad_norm": 0.4860077202320099,
+      "learning_rate": 2.4983295814251916e-05,
+      "loss": 0.1558,
+      "step": 13670
+    },
+    {
+      "grad_norm": 0.3921772539615631,
+      "learning_rate": 2.4911748654343105e-05,
+      "loss": 0.1545,
+      "step": 13680
+    },
+    {
+      "grad_norm": 0.6930418610572815,
+      "learning_rate": 2.4840270084755463e-05,
+      "loss": 0.1666,
+      "step": 13690
+    },
+    {
+      "grad_norm": 0.7458028793334961,
+      "learning_rate": 2.4768860300908685e-05,
+      "loss": 0.1631,
+      "step": 13700
+    },
+    {
+      "grad_norm": 0.4085841774940491,
+      "learning_rate": 2.469751949803443e-05,
+      "loss": 0.1531,
+      "step": 13710
+    },
+    {
+      "grad_norm": 0.5243082642555237,
+      "learning_rate": 2.4626247871175666e-05,
+      "loss": 0.1734,
+      "step": 13720
+    },
+    {
+      "grad_norm": 0.6336411237716675,
+      "learning_rate": 2.4555045615186346e-05,
+      "loss": 0.1659,
+      "step": 13730
+    },
+    {
+      "grad_norm": 0.5403310656547546,
+      "learning_rate": 2.4483912924730677e-05,
+      "loss": 0.1635,
+      "step": 13740
+    },
+    {
+      "grad_norm": 0.5413652062416077,
+      "learning_rate": 2.4412849994282742e-05,
+      "loss": 0.1695,
+      "step": 13750
+    },
+    {
+      "grad_norm": 0.9338445663452148,
+      "learning_rate": 2.434185701812592e-05,
+      "loss": 0.1584,
+      "step": 13760
+    },
+    {
+      "grad_norm": 0.6594008207321167,
+      "learning_rate": 2.4270934190352218e-05,
+      "loss": 0.2093,
+      "step": 13770
+    },
+    {
+      "grad_norm": 0.572248637676239,
+      "learning_rate": 2.4200081704861998e-05,
+      "loss": 0.165,
+      "step": 13780
+    },
+    {
+      "grad_norm": 0.895415723323822,
+      "learning_rate": 2.412929975536321e-05,
+      "loss": 0.1621,
+      "step": 13790
+    },
+    {
+      "grad_norm": 0.5335855484008789,
+      "learning_rate": 2.4058588535371017e-05,
+      "loss": 0.1751,
+      "step": 13800
+    },
+    {
+      "grad_norm": 0.41414937376976013,
+      "learning_rate": 2.3987948238207243e-05,
+      "loss": 0.1646,
+      "step": 13810
+    },
+    {
+      "grad_norm": 0.4890303909778595,
+      "learning_rate": 2.3917379056999678e-05,
+      "loss": 0.1757,
+      "step": 13820
+    },
+    {
+      "grad_norm": 0.4044335186481476,
+      "learning_rate": 2.3846881184681824e-05,
+      "loss": 0.1668,
+      "step": 13830
+    },
+    {
+      "grad_norm": 0.5928899049758911,
+      "learning_rate": 2.377645481399214e-05,
+      "loss": 0.1732,
+      "step": 13840
+    },
+    {
+      "grad_norm": 0.5383327007293701,
+      "learning_rate": 2.3706100137473667e-05,
+      "loss": 0.1885,
+      "step": 13850
+    },
+    {
+      "grad_norm": 0.40311476588249207,
+      "learning_rate": 2.3635817347473394e-05,
+      "loss": 0.1576,
+      "step": 13860
+    },
+    {
+      "grad_norm": 0.6410138010978699,
+      "learning_rate": 2.3565606636141757e-05,
+      "loss": 0.1517,
+      "step": 13870
+    },
+    {
+      "grad_norm": 0.42336976528167725,
+      "learning_rate": 2.3495468195432203e-05,
+      "loss": 0.1639,
+      "step": 13880
+    },
+    {
+      "grad_norm": 0.4434325397014618,
+      "learning_rate": 2.3425402217100507e-05,
+      "loss": 0.1813,
+      "step": 13890
+    },
+    {
+      "grad_norm": 0.5273676514625549,
+      "learning_rate": 2.3355408892704424e-05,
+      "loss": 0.177,
+      "step": 13900
+    },
+    {
+      "grad_norm": 1.1010818481445312,
+      "learning_rate": 2.3285488413603003e-05,
+      "loss": 0.1824,
+      "step": 13910
+    },
+    {
+      "grad_norm": 0.4930150806903839,
+      "learning_rate": 2.321564097095615e-05,
+      "loss": 0.1916,
+      "step": 13920
+    },
+    {
+      "grad_norm": 0.7029837965965271,
+      "learning_rate": 2.3145866755724142e-05,
+      "loss": 0.1711,
+      "step": 13930
+    },
+    {
+      "grad_norm": 0.5362256169319153,
+      "learning_rate": 2.307616595866699e-05,
+      "loss": 0.1747,
+      "step": 13940
+    },
+    {
+      "grad_norm": 0.5539709329605103,
+      "learning_rate": 2.3006538770344032e-05,
+      "loss": 0.1834,
+      "step": 13950
+    },
+    {
+      "grad_norm": 0.43245863914489746,
+      "learning_rate": 2.293698538111334e-05,
+      "loss": 0.1701,
+      "step": 13960
+    },
+    {
+      "grad_norm": 0.5766111016273499,
+      "learning_rate": 2.28675059811312e-05,
+      "loss": 0.1703,
+      "step": 13970
+    },
+    {
+      "grad_norm": 0.5906485319137573,
+      "learning_rate": 2.279810076035167e-05,
+      "loss": 0.1729,
+      "step": 13980
+    },
+    {
+      "grad_norm": 0.6132853627204895,
+      "learning_rate": 2.272876990852596e-05,
+      "loss": 0.1595,
+      "step": 13990
+    },
+    {
+      "grad_norm": 0.4491095542907715,
+      "learning_rate": 2.265951361520195e-05,
+      "loss": 0.1703,
+      "step": 14000
+    },
+    {
+      "grad_norm": 0.47304192185401917,
+      "learning_rate": 2.2590332069723748e-05,
+      "loss": 0.1728,
+      "step": 14010
+    },
+    {
+      "grad_norm": 0.6269650459289551,
+      "learning_rate": 2.2521225461231004e-05,
+      "loss": 0.1487,
+      "step": 14020
+    },
+    {
+      "grad_norm": 0.4406624138355255,
+      "learning_rate": 2.2452193978658597e-05,
+      "loss": 0.1742,
+      "step": 14030
+    },
+    {
+      "grad_norm": 0.48336032032966614,
+      "learning_rate": 2.238323781073594e-05,
+      "loss": 0.1599,
+      "step": 14040
+    },
+    {
+      "grad_norm": 0.4881398677825928,
+      "learning_rate": 2.2314357145986552e-05,
+      "loss": 0.1762,
+      "step": 14050
+    },
+    {
+      "grad_norm": 0.40922093391418457,
+      "learning_rate": 2.224555217272757e-05,
+      "loss": 0.1676,
+      "step": 14060
+    },
+    {
+      "grad_norm": 0.47547614574432373,
+      "learning_rate": 2.2176823079069127e-05,
+      "loss": 0.1682,
+      "step": 14070
+    },
+    {
+      "grad_norm": 0.574183464050293,
+      "learning_rate": 2.210817005291398e-05,
+      "loss": 0.1655,
+      "step": 14080
+    },
+    {
+      "grad_norm": 0.5340254306793213,
+      "learning_rate": 2.203959328195686e-05,
+      "loss": 0.1687,
+      "step": 14090
+    },
+    {
+      "grad_norm": 0.5861740112304688,
+      "learning_rate": 2.1971092953684026e-05,
+      "loss": 0.1613,
+      "step": 14100
+    },
+    {
+      "grad_norm": 0.5040154457092285,
+      "learning_rate": 2.1902669255372788e-05,
+      "loss": 0.167,
+      "step": 14110
+    },
+    {
+      "grad_norm": 0.49121519923210144,
+      "learning_rate": 2.1834322374090897e-05,
+      "loss": 0.1661,
+      "step": 14120
+    },
+    {
+      "grad_norm": 0.4335910975933075,
+      "learning_rate": 2.1766052496696153e-05,
+      "loss": 0.1635,
+      "step": 14130
+    },
+    {
+      "grad_norm": 0.4995878040790558,
+      "learning_rate": 2.169785980983577e-05,
+      "loss": 0.1642,
+      "step": 14140
+    },
+    {
+      "grad_norm": 0.5822210907936096,
+      "learning_rate": 2.162974449994593e-05,
+      "loss": 0.1548,
+      "step": 14150
+    },
+    {
+      "grad_norm": 0.5646964311599731,
+      "learning_rate": 2.1561706753251337e-05,
+      "loss": 0.171,
+      "step": 14160
+    },
+    {
+      "grad_norm": 0.5396406054496765,
+      "learning_rate": 2.1493746755764544e-05,
+      "loss": 0.1701,
+      "step": 14170
+    },
+    {
+      "grad_norm": 0.3783027231693268,
+      "learning_rate": 2.1425864693285635e-05,
+      "loss": 0.1629,
+      "step": 14180
+    },
+    {
+      "grad_norm": 0.8163051605224609,
+      "learning_rate": 2.1358060751401547e-05,
+      "loss": 0.1617,
+      "step": 14190
+    },
+    {
+      "grad_norm": 0.5412561297416687,
+      "learning_rate": 2.129033511548566e-05,
+      "loss": 0.1682,
+      "step": 14200
+    },
+    {
+      "grad_norm": 0.48072996735572815,
+      "learning_rate": 2.1222687970697315e-05,
+      "loss": 0.1643,
+      "step": 14210
+    },
+    {
+      "grad_norm": 0.40615180134773254,
+      "learning_rate": 2.1155119501981173e-05,
+      "loss": 0.1665,
+      "step": 14220
+    },
+    {
+      "grad_norm": 0.5327228903770447,
+      "learning_rate": 2.1087629894066895e-05,
+      "loss": 0.1757,
+      "step": 14230
+    },
+    {
+      "grad_norm": 0.5953244566917419,
+      "learning_rate": 2.1020219331468473e-05,
+      "loss": 0.1649,
+      "step": 14240
+    },
+    {
+      "grad_norm": 0.4644988179206848,
+      "learning_rate": 2.095288799848379e-05,
+      "loss": 0.1615,
+      "step": 14250
+    },
+    {
+      "grad_norm": 0.7579044699668884,
+      "learning_rate": 2.088563607919417e-05,
+      "loss": 0.165,
+      "step": 14260
+    },
+    {
+      "grad_norm": 0.5288797616958618,
+      "learning_rate": 2.0818463757463786e-05,
+      "loss": 0.1708,
+      "step": 14270
+    },
+    {
+      "grad_norm": 0.5747759938240051,
+      "learning_rate": 2.0751371216939175e-05,
+      "loss": 0.1556,
+      "step": 14280
+    },
+    {
+      "grad_norm": 0.6156583428382874,
+      "learning_rate": 2.068435864104882e-05,
+      "loss": 0.1548,
+      "step": 14290
+    },
+    {
+      "grad_norm": 0.7416667938232422,
+      "learning_rate": 2.0617426213002506e-05,
+      "loss": 0.157,
+      "step": 14300
+    },
+    {
+      "grad_norm": 0.34732478857040405,
+      "learning_rate": 2.055057411579097e-05,
+      "loss": 0.1651,
+      "step": 14310
+    },
+    {
+      "grad_norm": 0.4653224050998688,
+      "learning_rate": 2.0483802532185286e-05,
+      "loss": 0.1731,
+      "step": 14320
+    },
+    {
+      "grad_norm": 0.6839614510536194,
+      "learning_rate": 2.041711164473638e-05,
+      "loss": 0.1623,
+      "step": 14330
+    },
+    {
+      "grad_norm": 0.5481322407722473,
+      "learning_rate": 2.0350501635774637e-05,
+      "loss": 0.1678,
+      "step": 14340
+    },
+    {
+      "grad_norm": 0.6168639659881592,
+      "learning_rate": 2.0283972687409247e-05,
+      "loss": 0.1749,
+      "step": 14350
+    },
+    {
+      "grad_norm": 0.5664542317390442,
+      "learning_rate": 2.021752498152784e-05,
+      "loss": 0.1532,
+      "step": 14360
+    },
+    {
+      "grad_norm": 0.6281317472457886,
+      "learning_rate": 2.015115869979589e-05,
+      "loss": 0.1769,
+      "step": 14370
+    },
+    {
+      "grad_norm": 0.5325033664703369,
+      "learning_rate": 2.0084874023656265e-05,
+      "loss": 0.1733,
+      "step": 14380
+    },
+    {
+      "grad_norm": 0.569244921207428,
+      "learning_rate": 2.001867113432877e-05,
+      "loss": 0.1736,
+      "step": 14390
+    },
+    {
+      "grad_norm": 0.5267947316169739,
+      "learning_rate": 1.995255021280954e-05,
+      "loss": 0.1703,
+      "step": 14400
+    },
+    {
+      "grad_norm": 0.4406875967979431,
+      "learning_rate": 1.9886511439870688e-05,
+      "loss": 0.1776,
+      "step": 14410
+    },
+    {
+      "grad_norm": 0.5991896986961365,
+      "learning_rate": 1.9820554996059675e-05,
+      "loss": 0.1611,
+      "step": 14420
+    },
+    {
+      "grad_norm": 0.5982379913330078,
+      "learning_rate": 1.9754681061698893e-05,
+      "loss": 0.1607,
+      "step": 14430
+    },
+    {
+      "grad_norm": 0.5041596293449402,
+      "learning_rate": 1.9688889816885185e-05,
+      "loss": 0.172,
+      "step": 14440
+    },
+    {
+      "grad_norm": 0.4544855058193207,
+      "learning_rate": 1.962318144148928e-05,
+      "loss": 0.1674,
+      "step": 14450
+    },
+    {
+      "grad_norm": 0.7839325666427612,
+      "learning_rate": 1.955755611515539e-05,
+      "loss": 0.1748,
+      "step": 14460
+    },
+    {
+      "grad_norm": 0.4136938452720642,
+      "learning_rate": 1.9492014017300642e-05,
+      "loss": 0.1696,
+      "step": 14470
+    },
+    {
+      "grad_norm": 0.33549144864082336,
+      "learning_rate": 1.942655532711461e-05,
+      "loss": 0.1485,
+      "step": 14480
+    },
+    {
+      "grad_norm": 0.569109320640564,
+      "learning_rate": 1.9361180223558882e-05,
+      "loss": 0.1596,
+      "step": 14490
+    },
+    {
+      "grad_norm": 0.6051235795021057,
+      "learning_rate": 1.929588888536647e-05,
+      "loss": 0.1584,
+      "step": 14500
+    },
+    {
+      "grad_norm": 0.5000282526016235,
+      "learning_rate": 1.9230681491041425e-05,
+      "loss": 0.1682,
+      "step": 14510
+    },
+    {
+      "grad_norm": 0.4269227385520935,
+      "learning_rate": 1.9165558218858264e-05,
+      "loss": 0.1658,
+      "step": 14520
+    },
+    {
+      "grad_norm": 0.425680011510849,
+      "learning_rate": 1.9100519246861505e-05,
+      "loss": 0.1463,
+      "step": 14530
+    },
+    {
+      "grad_norm": 0.4468296766281128,
+      "learning_rate": 1.9035564752865248e-05,
+      "loss": 0.1558,
+      "step": 14540
+    },
+    {
+      "grad_norm": 0.5718340873718262,
+      "learning_rate": 1.897069491445258e-05,
+      "loss": 0.1836,
+      "step": 14550
+    },
+    {
+      "grad_norm": 0.5542588829994202,
+      "learning_rate": 1.890590990897515e-05,
+      "loss": 0.1569,
+      "step": 14560
+    },
+    {
+      "grad_norm": 0.6399118304252625,
+      "learning_rate": 1.884120991355272e-05,
+      "loss": 0.1675,
+      "step": 14570
+    },
+    {
+      "grad_norm": 0.67510986328125,
+      "learning_rate": 1.8776595105072576e-05,
+      "loss": 0.1687,
+      "step": 14580
+    },
+    {
+      "grad_norm": 0.9643115401268005,
+      "learning_rate": 1.8712065660189166e-05,
+      "loss": 0.1797,
+      "step": 14590
+    },
+    {
+      "grad_norm": 0.5739313364028931,
+      "learning_rate": 1.8647621755323513e-05,
+      "loss": 0.1719,
+      "step": 14600
+    },
+    {
+      "grad_norm": 0.4867771863937378,
+      "learning_rate": 1.858326356666278e-05,
+      "loss": 0.1656,
+      "step": 14610
+    },
+    {
+      "grad_norm": 0.7240979671478271,
+      "learning_rate": 1.851899127015983e-05,
+      "loss": 0.1658,
+      "step": 14620
+    },
+    {
+      "grad_norm": 0.5568884611129761,
+      "learning_rate": 1.8454805041532626e-05,
+      "loss": 0.1564,
+      "step": 14630
+    },
+    {
+      "grad_norm": 0.49170196056365967,
+      "learning_rate": 1.8390705056263906e-05,
+      "loss": 0.1696,
+      "step": 14640
+    },
+    {
+      "grad_norm": 0.42530348896980286,
+      "learning_rate": 1.832669148960057e-05,
+      "loss": 0.1679,
+      "step": 14650
+    },
+    {
+      "grad_norm": 0.4691551923751831,
+      "learning_rate": 1.8262764516553233e-05,
+      "loss": 0.1724,
+      "step": 14660
+    },
+    {
+      "grad_norm": 0.48202648758888245,
+      "learning_rate": 1.8198924311895843e-05,
+      "loss": 0.1716,
+      "step": 14670
+    },
+    {
+      "grad_norm": 0.4617081880569458,
+      "learning_rate": 1.813517105016505e-05,
+      "loss": 0.1819,
+      "step": 14680
+    },
+    {
+      "grad_norm": 0.7121796011924744,
+      "learning_rate": 1.8071504905659888e-05,
+      "loss": 0.1702,
+      "step": 14690
+    },
+    {
+      "grad_norm": 0.6098193526268005,
+      "learning_rate": 1.800792605244109e-05,
+      "loss": 0.1799,
+      "step": 14700
+    },
+    {
+      "grad_norm": 0.4047832489013672,
+      "learning_rate": 1.7944434664330844e-05,
+      "loss": 0.1673,
+      "step": 14710
+    },
+    {
+      "grad_norm": 0.4286409020423889,
+      "learning_rate": 1.7881030914912212e-05,
+      "loss": 0.1614,
+      "step": 14720
+    },
+    {
+      "grad_norm": 0.5605373978614807,
+      "learning_rate": 1.7817714977528577e-05,
+      "loss": 0.1591,
+      "step": 14730
+    },
+    {
+      "grad_norm": 0.5677973031997681,
+      "learning_rate": 1.7754487025283332e-05,
+      "loss": 0.1605,
+      "step": 14740
+    },
+    {
+      "grad_norm": 0.41900044679641724,
+      "learning_rate": 1.7691347231039275e-05,
+      "loss": 0.169,
+      "step": 14750
+    },
+    {
+      "grad_norm": 0.5106980204582214,
+      "learning_rate": 1.7628295767418164e-05,
+      "loss": 0.1642,
+      "step": 14760
+    },
+    {
+      "grad_norm": 0.4111778140068054,
+      "learning_rate": 1.7565332806800333e-05,
+      "loss": 0.163,
+      "step": 14770
+    },
+    {
+      "grad_norm": 0.638185977935791,
+      "learning_rate": 1.750245852132408e-05,
+      "loss": 0.1787,
+      "step": 14780
+    },
+    {
+      "grad_norm": 0.5195234417915344,
+      "learning_rate": 1.7439673082885323e-05,
+      "loss": 0.1592,
+      "step": 14790
+    },
+    {
+      "grad_norm": 0.4852416217327118,
+      "learning_rate": 1.7376976663137047e-05,
+      "loss": 0.1715,
+      "step": 14800
+    },
+    {
+      "grad_norm": 0.5728936195373535,
+      "learning_rate": 1.7314369433488853e-05,
+      "loss": 0.159,
+      "step": 14810
+    },
+    {
+      "grad_norm": 0.488319993019104,
+      "learning_rate": 1.7251851565106548e-05,
+      "loss": 0.1571,
+      "step": 14820
+    },
+    {
+      "grad_norm": 0.45625239610671997,
+      "learning_rate": 1.7189423228911574e-05,
+      "loss": 0.1594,
+      "step": 14830
+    },
+    {
+      "grad_norm": 0.7218068838119507,
+      "learning_rate": 1.7127084595580606e-05,
+      "loss": 0.1664,
+      "step": 14840
+    },
+    {
+      "grad_norm": 0.4839458465576172,
+      "learning_rate": 1.706483583554513e-05,
+      "loss": 0.1536,
+      "step": 14850
+    },
+    {
+      "grad_norm": 0.7011167407035828,
+      "learning_rate": 1.700267711899083e-05,
+      "loss": 0.1907,
+      "step": 14860
+    },
+    {
+      "grad_norm": 0.4585695266723633,
+      "learning_rate": 1.69406086158573e-05,
+      "loss": 0.1429,
+      "step": 14870
+    },
+    {
+      "grad_norm": 0.4816190004348755,
+      "learning_rate": 1.6878630495837455e-05,
+      "loss": 0.1393,
+      "step": 14880
+    },
+    {
+      "grad_norm": 0.5019491314888,
+      "learning_rate": 1.681674292837707e-05,
+      "loss": 0.1611,
+      "step": 14890
+    },
+    {
+      "grad_norm": 0.6372382640838623,
+      "learning_rate": 1.6754946082674444e-05,
+      "loss": 0.1481,
+      "step": 14900
+    },
+    {
+      "grad_norm": 1.0556327104568481,
+      "learning_rate": 1.6693240127679748e-05,
+      "loss": 0.1616,
+      "step": 14910
+    },
+    {
+      "grad_norm": 0.876680850982666,
+      "learning_rate": 1.663162523209475e-05,
+      "loss": 0.1632,
+      "step": 14920
+    },
+    {
+      "grad_norm": 0.7315483093261719,
+      "learning_rate": 1.6570101564372193e-05,
+      "loss": 0.1747,
+      "step": 14930
+    },
+    {
+      "grad_norm": 0.7636805176734924,
+      "learning_rate": 1.650866929271543e-05,
+      "loss": 0.1638,
+      "step": 14940
+    },
+    {
+      "grad_norm": 0.7136143445968628,
+      "learning_rate": 1.644732858507797e-05,
+      "loss": 0.1603,
+      "step": 14950
+    },
+    {
+      "grad_norm": 0.5374797582626343,
+      "learning_rate": 1.6386079609162943e-05,
+      "loss": 0.1516,
+      "step": 14960
+    },
+    {
+      "grad_norm": 0.5543725490570068,
+      "learning_rate": 1.6324922532422742e-05,
+      "loss": 0.1671,
+      "step": 14970
+    },
+    {
+      "grad_norm": 0.5024927258491516,
+      "learning_rate": 1.6263857522058434e-05,
+      "loss": 0.1536,
+      "step": 14980
+    },
+    {
+      "grad_norm": 0.5799697637557983,
+      "learning_rate": 1.6202884745019443e-05,
+      "loss": 0.162,
+      "step": 14990
+    },
+    {
+      "grad_norm": 0.5031914710998535,
+      "learning_rate": 1.614200436800304e-05,
+      "loss": 0.1523,
+      "step": 15000
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 20000,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 9223372036854775807,
+  "save_steps": 1000,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 0.0,
+  "train_batch_size": 16,
+  "trial_name": null,
+  "trial_params": null
+}