diff --git "a/vlmfreeze_loc_h10_17k/trainer_state.json" "b/vlmfreeze_loc_h10_17k/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/vlmfreeze_loc_h10_17k/trainer_state.json"
@@ -0,0 +1,10234 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.85,
+  "eval_steps": 500,
+  "global_step": 17000,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "grad_norm": 0.49919793009757996,
+      "learning_rate": 9e-07,
+      "loss": 0.3357,
+      "step": 10
+    },
+    {
+      "grad_norm": 0.8183974623680115,
+      "learning_rate": 1.9e-06,
+      "loss": 0.3369,
+      "step": 20
+    },
+    {
+      "grad_norm": 0.3310164213180542,
+      "learning_rate": 2.9e-06,
+      "loss": 0.3183,
+      "step": 30
+    },
+    {
+      "grad_norm": 0.40600425004959106,
+      "learning_rate": 3.9e-06,
+      "loss": 0.3136,
+      "step": 40
+    },
+    {
+      "grad_norm": 0.3595507740974426,
+      "learning_rate": 4.9000000000000005e-06,
+      "loss": 0.3067,
+      "step": 50
+    },
+    {
+      "grad_norm": 0.3398914933204651,
+      "learning_rate": 5.9e-06,
+      "loss": 0.3108,
+      "step": 60
+    },
+    {
+      "grad_norm": 0.3425237834453583,
+      "learning_rate": 6.900000000000001e-06,
+      "loss": 0.3089,
+      "step": 70
+    },
+    {
+      "grad_norm": 0.28334304690361023,
+      "learning_rate": 7.9e-06,
+      "loss": 0.3062,
+      "step": 80
+    },
+    {
+      "grad_norm": 0.3862457573413849,
+      "learning_rate": 8.9e-06,
+      "loss": 0.2839,
+      "step": 90
+    },
+    {
+      "grad_norm": 0.36357825994491577,
+      "learning_rate": 9.900000000000002e-06,
+      "loss": 0.2789,
+      "step": 100
+    },
+    {
+      "grad_norm": 0.2662424147129059,
+      "learning_rate": 1.09e-05,
+      "loss": 0.2645,
+      "step": 110
+    },
+    {
+      "grad_norm": 0.37322670221328735,
+      "learning_rate": 1.19e-05,
+      "loss": 0.2563,
+      "step": 120
+    },
+    {
+      "grad_norm": 0.33626827597618103,
+      "learning_rate": 1.29e-05,
+      "loss": 0.2686,
+      "step": 130
+    },
+    {
+      "grad_norm": 0.31342142820358276,
+      "learning_rate": 1.3900000000000002e-05,
+      "loss": 0.2764,
+      "step": 140
+    },
+    {
+      "grad_norm": 0.3349839746952057,
+      "learning_rate": 1.49e-05,
+      "loss": 0.2545,
+      "step": 150
+    },
+    {
+      "grad_norm": 0.3048800230026245,
+      "learning_rate": 1.59e-05,
+      "loss": 0.2419,
+      "step": 160
+    },
+    {
+      "grad_norm": 0.2844460904598236,
+      "learning_rate": 1.69e-05,
+      "loss": 0.2231,
+      "step": 170
+    },
+    {
+      "grad_norm": 0.4111763834953308,
+      "learning_rate": 1.79e-05,
+      "loss": 0.2434,
+      "step": 180
+    },
+    {
+      "grad_norm": 0.4194934666156769,
+      "learning_rate": 1.8900000000000002e-05,
+      "loss": 0.2346,
+      "step": 190
+    },
+    {
+      "grad_norm": 0.4110984802246094,
+      "learning_rate": 1.9900000000000003e-05,
+      "loss": 0.2309,
+      "step": 200
+    },
+    {
+      "grad_norm": 0.38131290674209595,
+      "learning_rate": 2.09e-05,
+      "loss": 0.2078,
+      "step": 210
+    },
+    {
+      "grad_norm": 0.5574779510498047,
+      "learning_rate": 2.19e-05,
+      "loss": 0.2283,
+      "step": 220
+    },
+    {
+      "grad_norm": 0.46695613861083984,
+      "learning_rate": 2.29e-05,
+      "loss": 0.2209,
+      "step": 230
+    },
+    {
+      "grad_norm": 0.5055718421936035,
+      "learning_rate": 2.39e-05,
+      "loss": 0.2163,
+      "step": 240
+    },
+    {
+      "grad_norm": 0.5080457925796509,
+      "learning_rate": 2.4900000000000002e-05,
+      "loss": 0.2228,
+      "step": 250
+    },
+    {
+      "grad_norm": 0.365084171295166,
+      "learning_rate": 2.5900000000000003e-05,
+      "loss": 0.1649,
+      "step": 260
+    },
+    {
+      "grad_norm": 0.5375341773033142,
+      "learning_rate": 2.6900000000000003e-05,
+      "loss": 0.2042,
+      "step": 270
+    },
+    {
+      "grad_norm": 0.4356468915939331,
+      "learning_rate": 2.7900000000000004e-05,
+      "loss": 0.1965,
+      "step": 280
+    },
+    {
+      "grad_norm": 0.29737231135368347,
+      "learning_rate": 2.8899999999999998e-05,
+      "loss": 0.1847,
+      "step": 290
+    },
+    {
+      "grad_norm": 0.47155651450157166,
+      "learning_rate": 2.9900000000000002e-05,
+      "loss": 0.1779,
+      "step": 300
+    },
+    {
+      "grad_norm": 0.47919100522994995,
+      "learning_rate": 3.09e-05,
+      "loss": 0.1937,
+      "step": 310
+    },
+    {
+      "grad_norm": 0.34574800729751587,
+      "learning_rate": 3.19e-05,
+      "loss": 0.1957,
+      "step": 320
+    },
+    {
+      "grad_norm": 0.5125250220298767,
+      "learning_rate": 3.29e-05,
+      "loss": 0.1938,
+      "step": 330
+    },
+    {
+      "grad_norm": 0.372917115688324,
+      "learning_rate": 3.3900000000000004e-05,
+      "loss": 0.1693,
+      "step": 340
+    },
+    {
+      "grad_norm": 0.3852974474430084,
+      "learning_rate": 3.49e-05,
+      "loss": 0.1893,
+      "step": 350
+    },
+    {
+      "grad_norm": 0.6960095167160034,
+      "learning_rate": 3.59e-05,
+      "loss": 0.1737,
+      "step": 360
+    },
+    {
+      "grad_norm": 0.40932366251945496,
+      "learning_rate": 3.69e-05,
+      "loss": 0.1679,
+      "step": 370
+    },
+    {
+      "grad_norm": 0.38926613330841064,
+      "learning_rate": 3.79e-05,
+      "loss": 0.1795,
+      "step": 380
+    },
+    {
+      "grad_norm": 0.6326749324798584,
+      "learning_rate": 3.8900000000000004e-05,
+      "loss": 0.1988,
+      "step": 390
+    },
+    {
+      "grad_norm": 0.45645421743392944,
+      "learning_rate": 3.99e-05,
+      "loss": 0.179,
+      "step": 400
+    },
+    {
+      "grad_norm": 0.392812579870224,
+      "learning_rate": 4.09e-05,
+      "loss": 0.1662,
+      "step": 410
+    },
+    {
+      "grad_norm": 0.7580351829528809,
+      "learning_rate": 4.19e-05,
+      "loss": 0.185,
+      "step": 420
+    },
+    {
+      "grad_norm": 0.5179138779640198,
+      "learning_rate": 4.29e-05,
+      "loss": 0.1743,
+      "step": 430
+    },
+    {
+      "grad_norm": 0.5835297107696533,
+      "learning_rate": 4.39e-05,
+      "loss": 0.191,
+      "step": 440
+    },
+    {
+      "grad_norm": 0.3952111601829529,
+      "learning_rate": 4.49e-05,
+      "loss": 0.1887,
+      "step": 450
+    },
+    {
+      "grad_norm": 0.4841550588607788,
+      "learning_rate": 4.5900000000000004e-05,
+      "loss": 0.1617,
+      "step": 460
+    },
+    {
+      "grad_norm": 0.5566447973251343,
+      "learning_rate": 4.69e-05,
+      "loss": 0.1516,
+      "step": 470
+    },
+    {
+      "grad_norm": 0.3890221416950226,
+      "learning_rate": 4.79e-05,
+      "loss": 0.1699,
+      "step": 480
+    },
+    {
+      "grad_norm": 0.6405625939369202,
+      "learning_rate": 4.89e-05,
+      "loss": 0.17,
+      "step": 490
+    },
+    {
+      "grad_norm": 0.4669230580329895,
+      "learning_rate": 4.99e-05,
+      "loss": 0.1768,
+      "step": 500
+    },
+    {
+      "grad_norm": 0.5685226321220398,
+      "learning_rate": 5.0900000000000004e-05,
+      "loss": 0.1916,
+      "step": 510
+    },
+    {
+      "grad_norm": 0.4632212817668915,
+      "learning_rate": 5.19e-05,
+      "loss": 0.1606,
+      "step": 520
+    },
+    {
+      "grad_norm": 0.49214452505111694,
+      "learning_rate": 5.2900000000000005e-05,
+      "loss": 0.1602,
+      "step": 530
+    },
+    {
+      "grad_norm": 0.548396646976471,
+      "learning_rate": 5.390000000000001e-05,
+      "loss": 0.1526,
+      "step": 540
+    },
+    {
+      "grad_norm": 0.5027827620506287,
+      "learning_rate": 5.4900000000000006e-05,
+      "loss": 0.1716,
+      "step": 550
+    },
+    {
+      "grad_norm": 0.37354350090026855,
+      "learning_rate": 5.590000000000001e-05,
+      "loss": 0.1803,
+      "step": 560
+    },
+    {
+      "grad_norm": 0.47122082114219666,
+      "learning_rate": 5.69e-05,
+      "loss": 0.1596,
+      "step": 570
+    },
+    {
+      "grad_norm": 0.45578533411026,
+      "learning_rate": 5.79e-05,
+      "loss": 0.1655,
+      "step": 580
+    },
+    {
+      "grad_norm": 0.4837999939918518,
+      "learning_rate": 5.89e-05,
+      "loss": 0.177,
+      "step": 590
+    },
+    {
+      "grad_norm": 0.4454890191555023,
+      "learning_rate": 5.99e-05,
+      "loss": 0.1581,
+      "step": 600
+    },
+    {
+      "grad_norm": 0.41412365436553955,
+      "learning_rate": 6.09e-05,
+      "loss": 0.1687,
+      "step": 610
+    },
+    {
+      "grad_norm": 0.4288477599620819,
+      "learning_rate": 6.19e-05,
+      "loss": 0.1637,
+      "step": 620
+    },
+    {
+      "grad_norm": 0.5093424320220947,
+      "learning_rate": 6.29e-05,
+      "loss": 0.1751,
+      "step": 630
+    },
+    {
+      "grad_norm": 0.3570202589035034,
+      "learning_rate": 6.390000000000001e-05,
+      "loss": 0.1504,
+      "step": 640
+    },
+    {
+      "grad_norm": 0.4416474997997284,
+      "learning_rate": 6.49e-05,
+      "loss": 0.1654,
+      "step": 650
+    },
+    {
+      "grad_norm": 0.553651750087738,
+      "learning_rate": 6.59e-05,
+      "loss": 0.183,
+      "step": 660
+    },
+    {
+      "grad_norm": 0.4252713620662689,
+      "learning_rate": 6.690000000000001e-05,
+      "loss": 0.1825,
+      "step": 670
+    },
+    {
+      "grad_norm": 0.4885893166065216,
+      "learning_rate": 6.790000000000001e-05,
+      "loss": 0.1552,
+      "step": 680
+    },
+    {
+      "grad_norm": 0.5417454242706299,
+      "learning_rate": 6.89e-05,
+      "loss": 0.1539,
+      "step": 690
+    },
+    {
+      "grad_norm": 0.5418331623077393,
+      "learning_rate": 6.99e-05,
+      "loss": 0.1584,
+      "step": 700
+    },
+    {
+      "grad_norm": 0.43238386511802673,
+      "learning_rate": 7.09e-05,
+      "loss": 0.14,
+      "step": 710
+    },
+    {
+      "grad_norm": 0.8995737433433533,
+      "learning_rate": 7.19e-05,
+      "loss": 0.1622,
+      "step": 720
+    },
+    {
+      "grad_norm": 0.45176416635513306,
+      "learning_rate": 7.29e-05,
+      "loss": 0.1853,
+      "step": 730
+    },
+    {
+      "grad_norm": 0.6066661477088928,
+      "learning_rate": 7.390000000000001e-05,
+      "loss": 0.1693,
+      "step": 740
+    },
+    {
+      "grad_norm": 0.44987475872039795,
+      "learning_rate": 7.49e-05,
+      "loss": 0.1401,
+      "step": 750
+    },
+    {
+      "grad_norm": 0.5020248293876648,
+      "learning_rate": 7.59e-05,
+      "loss": 0.1757,
+      "step": 760
+    },
+    {
+      "grad_norm": 0.45413193106651306,
+      "learning_rate": 7.69e-05,
+      "loss": 0.178,
+      "step": 770
+    },
+    {
+      "grad_norm": 0.5021084547042847,
+      "learning_rate": 7.790000000000001e-05,
+      "loss": 0.1598,
+      "step": 780
+    },
+    {
+      "grad_norm": 0.4811430275440216,
+      "learning_rate": 7.890000000000001e-05,
+      "loss": 0.1706,
+      "step": 790
+    },
+    {
+      "grad_norm": 0.49934500455856323,
+      "learning_rate": 7.99e-05,
+      "loss": 0.1525,
+      "step": 800
+    },
+    {
+      "grad_norm": 0.45332571864128113,
+      "learning_rate": 8.090000000000001e-05,
+      "loss": 0.1743,
+      "step": 810
+    },
+    {
+      "grad_norm": 0.3541397452354431,
+      "learning_rate": 8.19e-05,
+      "loss": 0.1694,
+      "step": 820
+    },
+    {
+      "grad_norm": 0.4758358299732208,
+      "learning_rate": 8.29e-05,
+      "loss": 0.1497,
+      "step": 830
+    },
+    {
+      "grad_norm": 0.5417020320892334,
+      "learning_rate": 8.39e-05,
+      "loss": 0.1709,
+      "step": 840
+    },
+    {
+      "grad_norm": 0.5000898241996765,
+      "learning_rate": 8.49e-05,
+      "loss": 0.1691,
+      "step": 850
+    },
+    {
+      "grad_norm": 0.7036383748054504,
+      "learning_rate": 8.59e-05,
+      "loss": 0.1691,
+      "step": 860
+    },
+    {
+      "grad_norm": 0.36348623037338257,
+      "learning_rate": 8.69e-05,
+      "loss": 0.1746,
+      "step": 870
+    },
+    {
+      "grad_norm": 0.4636788070201874,
+      "learning_rate": 8.790000000000001e-05,
+      "loss": 0.1714,
+      "step": 880
+    },
+    {
+      "grad_norm": 0.42352041602134705,
+      "learning_rate": 8.89e-05,
+      "loss": 0.165,
+      "step": 890
+    },
+    {
+      "grad_norm": 0.513974130153656,
+      "learning_rate": 8.99e-05,
+      "loss": 0.1628,
+      "step": 900
+    },
+    {
+      "grad_norm": 0.5464494824409485,
+      "learning_rate": 9.090000000000001e-05,
+      "loss": 0.1845,
+      "step": 910
+    },
+    {
+      "grad_norm": 0.5518175363540649,
+      "learning_rate": 9.190000000000001e-05,
+      "loss": 0.1805,
+      "step": 920
+    },
+    {
+      "grad_norm": 0.7822105884552002,
+      "learning_rate": 9.290000000000001e-05,
+      "loss": 0.1808,
+      "step": 930
+    },
+    {
+      "grad_norm": 0.46754059195518494,
+      "learning_rate": 9.39e-05,
+      "loss": 0.1657,
+      "step": 940
+    },
+    {
+      "grad_norm": 0.29130831360816956,
+      "learning_rate": 9.49e-05,
+      "loss": 0.165,
+      "step": 950
+    },
+    {
+      "grad_norm": 0.49861809611320496,
+      "learning_rate": 9.59e-05,
+      "loss": 0.1577,
+      "step": 960
+    },
+    {
+      "grad_norm": 0.5356115102767944,
+      "learning_rate": 9.69e-05,
+      "loss": 0.1753,
+      "step": 970
+    },
+    {
+      "grad_norm": 0.43957042694091797,
+      "learning_rate": 9.790000000000001e-05,
+      "loss": 0.158,
+      "step": 980
+    },
+    {
+      "grad_norm": 0.6091327667236328,
+      "learning_rate": 9.89e-05,
+      "loss": 0.1844,
+      "step": 990
+    },
+    {
+      "grad_norm": 0.35945528745651245,
+      "learning_rate": 9.99e-05,
+      "loss": 0.1603,
+      "step": 1000
+    },
+    {
+      "grad_norm": 0.4383627772331238,
+      "learning_rate": 9.999994463727085e-05,
+      "loss": 0.1699,
+      "step": 1010
+    },
+    {
+      "grad_norm": 0.590233325958252,
+      "learning_rate": 9.999975326009292e-05,
+      "loss": 0.1629,
+      "step": 1020
+    },
+    {
+      "grad_norm": 0.3979860246181488,
+      "learning_rate": 9.999942518549879e-05,
+      "loss": 0.1826,
+      "step": 1030
+    },
+    {
+      "grad_norm": 0.5467487573623657,
+      "learning_rate": 9.999896041438544e-05,
+      "loss": 0.1592,
+      "step": 1040
+    },
+    {
+      "grad_norm": 0.4409658908843994,
+      "learning_rate": 9.999835894802353e-05,
+      "loss": 0.1612,
+      "step": 1050
+    },
+    {
+      "grad_norm": 0.47142231464385986,
+      "learning_rate": 9.999762078805743e-05,
+      "loss": 0.1785,
+      "step": 1060
+    },
+    {
+      "grad_norm": 0.4026871919631958,
+      "learning_rate": 9.999674593650526e-05,
+      "loss": 0.1478,
+      "step": 1070
+    },
+    {
+      "grad_norm": 0.6867359280586243,
+      "learning_rate": 9.99957343957588e-05,
+      "loss": 0.1952,
+      "step": 1080
+    },
+    {
+      "grad_norm": 0.3656802475452423,
+      "learning_rate": 9.99945861685836e-05,
+      "loss": 0.1449,
+      "step": 1090
+    },
+    {
+      "grad_norm": 0.512914776802063,
+      "learning_rate": 9.999330125811884e-05,
+      "loss": 0.1598,
+      "step": 1100
+    },
+    {
+      "grad_norm": 0.4180309474468231,
+      "learning_rate": 9.999187966787744e-05,
+      "loss": 0.1753,
+      "step": 1110
+    },
+    {
+      "grad_norm": 0.40852007269859314,
+      "learning_rate": 9.999032140174595e-05,
+      "loss": 0.1647,
+      "step": 1120
+    },
+    {
+      "grad_norm": 0.3530569076538086,
+      "learning_rate": 9.998862646398464e-05,
+      "loss": 0.1482,
+      "step": 1130
+    },
+    {
+      "grad_norm": 0.32812801003456116,
+      "learning_rate": 9.998679485922739e-05,
+      "loss": 0.1893,
+      "step": 1140
+    },
+    {
+      "grad_norm": 0.4347979426383972,
+      "learning_rate": 9.998482659248174e-05,
+      "loss": 0.1698,
+      "step": 1150
+    },
+    {
+      "grad_norm": 0.4761258363723755,
+      "learning_rate": 9.998272166912883e-05,
+      "loss": 0.1845,
+      "step": 1160
+    },
+    {
+      "grad_norm": 0.32617080211639404,
+      "learning_rate": 9.998048009492347e-05,
+      "loss": 0.1636,
+      "step": 1170
+    },
+    {
+      "grad_norm": 0.44783279299736023,
+      "learning_rate": 9.997810187599403e-05,
+      "loss": 0.1563,
+      "step": 1180
+    },
+    {
+      "grad_norm": 0.4570409059524536,
+      "learning_rate": 9.997558701884249e-05,
+      "loss": 0.1726,
+      "step": 1190
+    },
+    {
+      "grad_norm": 0.545115053653717,
+      "learning_rate": 9.997293553034433e-05,
+      "loss": 0.1937,
+      "step": 1200
+    },
+    {
+      "grad_norm": 0.5105549693107605,
+      "learning_rate": 9.997014741774866e-05,
+      "loss": 0.1754,
+      "step": 1210
+    },
+    {
+      "grad_norm": 0.7354128360748291,
+      "learning_rate": 9.996722268867803e-05,
+      "loss": 0.1882,
+      "step": 1220
+    },
+    {
+      "grad_norm": 0.4433441758155823,
+      "learning_rate": 9.996416135112858e-05,
+      "loss": 0.1702,
+      "step": 1230
+    },
+    {
+      "grad_norm": 0.4800748825073242,
+      "learning_rate": 9.996096341346988e-05,
+      "loss": 0.1863,
+      "step": 1240
+    },
+    {
+      "grad_norm": 0.5801267623901367,
+      "learning_rate": 9.995762888444495e-05,
+      "loss": 0.1517,
+      "step": 1250
+    },
+    {
+      "grad_norm": 0.4159570634365082,
+      "learning_rate": 9.995415777317027e-05,
+      "loss": 0.1836,
+      "step": 1260
+    },
+    {
+      "grad_norm": 0.34658706188201904,
+      "learning_rate": 9.995055008913574e-05,
+      "loss": 0.1683,
+      "step": 1270
+    },
+    {
+      "grad_norm": 0.4147317707538605,
+      "learning_rate": 9.994680584220463e-05,
+      "loss": 0.1698,
+      "step": 1280
+    },
+    {
+      "grad_norm": 0.4322935938835144,
+      "learning_rate": 9.994292504261355e-05,
+      "loss": 0.1754,
+      "step": 1290
+    },
+    {
+      "grad_norm": 0.45112869143486023,
+      "learning_rate": 9.993890770097247e-05,
+      "loss": 0.1596,
+      "step": 1300
+    },
+    {
+      "grad_norm": 0.3351392149925232,
+      "learning_rate": 9.993475382826467e-05,
+      "loss": 0.1547,
+      "step": 1310
+    },
+    {
+      "grad_norm": 0.6491179466247559,
+      "learning_rate": 9.993046343584664e-05,
+      "loss": 0.1702,
+      "step": 1320
+    },
+    {
+      "grad_norm": 0.5320515632629395,
+      "learning_rate": 9.992603653544816e-05,
+      "loss": 0.1863,
+      "step": 1330
+    },
+    {
+      "grad_norm": 0.43435370922088623,
+      "learning_rate": 9.992147313917222e-05,
+      "loss": 0.1606,
+      "step": 1340
+    },
+    {
+      "grad_norm": 0.5075807571411133,
+      "learning_rate": 9.991677325949497e-05,
+      "loss": 0.1611,
+      "step": 1350
+    },
+    {
+      "grad_norm": 0.4908890426158905,
+      "learning_rate": 9.991193690926568e-05,
+      "loss": 0.1799,
+      "step": 1360
+    },
+    {
+      "grad_norm": 0.7153270244598389,
+      "learning_rate": 9.990696410170678e-05,
+      "loss": 0.151,
+      "step": 1370
+    },
+    {
+      "grad_norm": 0.4066173732280731,
+      "learning_rate": 9.990185485041371e-05,
+      "loss": 0.1652,
+      "step": 1380
+    },
+    {
+      "grad_norm": 0.5917170643806458,
+      "learning_rate": 9.989660916935498e-05,
+      "loss": 0.1818,
+      "step": 1390
+    },
+    {
+      "grad_norm": 0.456473708152771,
+      "learning_rate": 9.989122707287208e-05,
+      "loss": 0.1713,
+      "step": 1400
+    },
+    {
+      "grad_norm": 0.4308083653450012,
+      "learning_rate": 9.988570857567945e-05,
+      "loss": 0.1622,
+      "step": 1410
+    },
+    {
+      "grad_norm": 0.3788297176361084,
+      "learning_rate": 9.988005369286446e-05,
+      "loss": 0.157,
+      "step": 1420
+    },
+    {
+      "grad_norm": 0.5879454016685486,
+      "learning_rate": 9.987426243988734e-05,
+      "loss": 0.1843,
+      "step": 1430
+    },
+    {
+      "grad_norm": 0.4538172483444214,
+      "learning_rate": 9.986833483258114e-05,
+      "loss": 0.1603,
+      "step": 1440
+    },
+    {
+      "grad_norm": 0.4889186918735504,
+      "learning_rate": 9.986227088715173e-05,
+      "loss": 0.1752,
+      "step": 1450
+    },
+    {
+      "grad_norm": 0.4647832214832306,
+      "learning_rate": 9.98560706201777e-05,
+      "loss": 0.1717,
+      "step": 1460
+    },
+    {
+      "grad_norm": 0.39912810921669006,
+      "learning_rate": 9.984973404861036e-05,
+      "loss": 0.1571,
+      "step": 1470
+    },
+    {
+      "grad_norm": 0.4435666501522064,
+      "learning_rate": 9.984326118977361e-05,
+      "loss": 0.1713,
+      "step": 1480
+    },
+    {
+      "grad_norm": 0.48232781887054443,
+      "learning_rate": 9.983665206136406e-05,
+      "loss": 0.1606,
+      "step": 1490
+    },
+    {
+      "grad_norm": 0.3383047580718994,
+      "learning_rate": 9.982990668145075e-05,
+      "loss": 0.1544,
+      "step": 1500
+    },
+    {
+      "grad_norm": 0.4779326319694519,
+      "learning_rate": 9.982302506847534e-05,
+      "loss": 0.1474,
+      "step": 1510
+    },
+    {
+      "grad_norm": 0.5124672055244446,
+      "learning_rate": 9.981600724125189e-05,
+      "loss": 0.1694,
+      "step": 1520
+    },
+    {
+      "grad_norm": 0.44776567816734314,
+      "learning_rate": 9.980885321896685e-05,
+      "loss": 0.1698,
+      "step": 1530
+    },
+    {
+      "grad_norm": 0.4382692575454712,
+      "learning_rate": 9.980156302117905e-05,
+      "loss": 0.1738,
+      "step": 1540
+    },
+    {
+      "grad_norm": 0.4076938331127167,
+      "learning_rate": 9.979413666781963e-05,
+      "loss": 0.1613,
+      "step": 1550
+    },
+    {
+      "grad_norm": 0.469988614320755,
+      "learning_rate": 9.978657417919193e-05,
+      "loss": 0.1657,
+      "step": 1560
+    },
+    {
+      "grad_norm": 0.41605186462402344,
+      "learning_rate": 9.977887557597153e-05,
+      "loss": 0.1623,
+      "step": 1570
+    },
+    {
+      "grad_norm": 0.6079490184783936,
+      "learning_rate": 9.97710408792061e-05,
+      "loss": 0.1952,
+      "step": 1580
+    },
+    {
+      "grad_norm": 0.4101788103580475,
+      "learning_rate": 9.976307011031542e-05,
+      "loss": 0.1711,
+      "step": 1590
+    },
+    {
+      "grad_norm": 0.4563458263874054,
+      "learning_rate": 9.975496329109126e-05,
+      "loss": 0.1515,
+      "step": 1600
+    },
+    {
+      "grad_norm": 0.4573756754398346,
+      "learning_rate": 9.974672044369732e-05,
+      "loss": 0.1597,
+      "step": 1610
+    },
+    {
+      "grad_norm": 0.4121549129486084,
+      "learning_rate": 9.97383415906693e-05,
+      "loss": 0.1501,
+      "step": 1620
+    },
+    {
+      "grad_norm": 0.3829125463962555,
+      "learning_rate": 9.97298267549146e-05,
+      "loss": 0.1685,
+      "step": 1630
+    },
+    {
+      "grad_norm": 0.4750492572784424,
+      "learning_rate": 9.972117595971249e-05,
+      "loss": 0.155,
+      "step": 1640
+    },
+    {
+      "grad_norm": 0.6081451177597046,
+      "learning_rate": 9.971238922871391e-05,
+      "loss": 0.1583,
+      "step": 1650
+    },
+    {
+      "grad_norm": 0.35019928216934204,
+      "learning_rate": 9.970346658594142e-05,
+      "loss": 0.1464,
+      "step": 1660
+    },
+    {
+      "grad_norm": 0.47533974051475525,
+      "learning_rate": 9.969440805578923e-05,
+      "loss": 0.1793,
+      "step": 1670
+    },
+    {
+      "grad_norm": 0.4462675154209137,
+      "learning_rate": 9.968521366302298e-05,
+      "loss": 0.157,
+      "step": 1680
+    },
+    {
+      "grad_norm": 0.4538755714893341,
+      "learning_rate": 9.967588343277981e-05,
+      "loss": 0.1734,
+      "step": 1690
+    },
+    {
+      "grad_norm": 0.6284106969833374,
+      "learning_rate": 9.966641739056818e-05,
+      "loss": 0.1597,
+      "step": 1700
+    },
+    {
+      "grad_norm": 0.41365891695022583,
+      "learning_rate": 9.965681556226793e-05,
+      "loss": 0.1746,
+      "step": 1710
+    },
+    {
+      "grad_norm": 0.44948768615722656,
+      "learning_rate": 9.964707797413006e-05,
+      "loss": 0.1463,
+      "step": 1720
+    },
+    {
+      "grad_norm": 0.36817267537117004,
+      "learning_rate": 9.963720465277679e-05,
+      "loss": 0.1649,
+      "step": 1730
+    },
+    {
+      "grad_norm": 0.3769955635070801,
+      "learning_rate": 9.96271956252014e-05,
+      "loss": 0.1736,
+      "step": 1740
+    },
+    {
+      "grad_norm": 0.5613548755645752,
+      "learning_rate": 9.961705091876816e-05,
+      "loss": 0.174,
+      "step": 1750
+    },
+    {
+      "grad_norm": 0.41043537855148315,
+      "learning_rate": 9.960677056121235e-05,
+      "loss": 0.1729,
+      "step": 1760
+    },
+    {
+      "grad_norm": 0.46817097067832947,
+      "learning_rate": 9.959635458064005e-05,
+      "loss": 0.1766,
+      "step": 1770
+    },
+    {
+      "grad_norm": 0.5097590684890747,
+      "learning_rate": 9.958580300552815e-05,
+      "loss": 0.1654,
+      "step": 1780
+    },
+    {
+      "grad_norm": 0.3588050305843353,
+      "learning_rate": 9.957511586472426e-05,
+      "loss": 0.1621,
+      "step": 1790
+    },
+    {
+      "grad_norm": 0.4156924784183502,
+      "learning_rate": 9.956429318744662e-05,
+      "loss": 0.1537,
+      "step": 1800
+    },
+    {
+      "grad_norm": 0.4562986493110657,
+      "learning_rate": 9.955333500328404e-05,
+      "loss": 0.1528,
+      "step": 1810
+    },
+    {
+      "grad_norm": 0.39460626244544983,
+      "learning_rate": 9.95422413421957e-05,
+      "loss": 0.1693,
+      "step": 1820
+    },
+    {
+      "grad_norm": 0.4456663429737091,
+      "learning_rate": 9.953101223451133e-05,
+      "loss": 0.1788,
+      "step": 1830
+    },
+    {
+      "grad_norm": 0.3802589476108551,
+      "learning_rate": 9.951964771093085e-05,
+      "loss": 0.1731,
+      "step": 1840
+    },
+    {
+      "grad_norm": 0.6012398600578308,
+      "learning_rate": 9.950814780252442e-05,
+      "loss": 0.1504,
+      "step": 1850
+    },
+    {
+      "grad_norm": 0.5096449255943298,
+      "learning_rate": 9.949651254073236e-05,
+      "loss": 0.1629,
+      "step": 1860
+    },
+    {
+      "grad_norm": 0.3866385817527771,
+      "learning_rate": 9.948474195736504e-05,
+      "loss": 0.1617,
+      "step": 1870
+    },
+    {
+      "grad_norm": 0.39717477560043335,
+      "learning_rate": 9.947283608460277e-05,
+      "loss": 0.1719,
+      "step": 1880
+    },
+    {
+      "grad_norm": 0.41292041540145874,
+      "learning_rate": 9.946079495499577e-05,
+      "loss": 0.1554,
+      "step": 1890
+    },
+    {
+      "grad_norm": 0.4445132315158844,
+      "learning_rate": 9.944861860146401e-05,
+      "loss": 0.1511,
+      "step": 1900
+    },
+    {
+      "grad_norm": 0.42302602529525757,
+      "learning_rate": 9.943630705729719e-05,
+      "loss": 0.156,
+      "step": 1910
+    },
+    {
+      "grad_norm": 0.4361373484134674,
+      "learning_rate": 9.942386035615459e-05,
+      "loss": 0.1639,
+      "step": 1920
+    },
+    {
+      "grad_norm": 0.43072396516799927,
+      "learning_rate": 9.941127853206503e-05,
+      "loss": 0.2152,
+      "step": 1930
+    },
+    {
+      "grad_norm": 0.45482319593429565,
+      "learning_rate": 9.939856161942673e-05,
+      "loss": 0.1523,
+      "step": 1940
+    },
+    {
+      "grad_norm": 0.35983747243881226,
+      "learning_rate": 9.938570965300724e-05,
+      "loss": 0.1714,
+      "step": 1950
+    },
+    {
+      "grad_norm": 0.472236305475235,
+      "learning_rate": 9.937272266794335e-05,
+      "loss": 0.166,
+      "step": 1960
+    },
+    {
+      "grad_norm": 0.4279971420764923,
+      "learning_rate": 9.935960069974096e-05,
+      "loss": 0.1731,
+      "step": 1970
+    },
+    {
+      "grad_norm": 0.5904799699783325,
+      "learning_rate": 9.934634378427506e-05,
+      "loss": 0.1756,
+      "step": 1980
+    },
+    {
+      "grad_norm": 0.5814228057861328,
+      "learning_rate": 9.933295195778954e-05,
+      "loss": 0.1873,
+      "step": 1990
+    },
+    {
+      "grad_norm": 0.3843928873538971,
+      "learning_rate": 9.931942525689715e-05,
+      "loss": 0.1987,
+      "step": 2000
+    },
+    {
+      "grad_norm": 0.4723409414291382,
+      "learning_rate": 9.930576371857936e-05,
+      "loss": 0.1544,
+      "step": 2010
+    },
+    {
+      "grad_norm": 0.6833857893943787,
+      "learning_rate": 9.929196738018629e-05,
+      "loss": 0.1586,
+      "step": 2020
+    },
+    {
+      "grad_norm": 0.5064021348953247,
+      "learning_rate": 9.927803627943662e-05,
+      "loss": 0.1722,
+      "step": 2030
+    },
+    {
+      "grad_norm": 0.3276600241661072,
+      "learning_rate": 9.926397045441744e-05,
+      "loss": 0.1464,
+      "step": 2040
+    },
+    {
+      "grad_norm": 0.38793569803237915,
+      "learning_rate": 9.924976994358417e-05,
+      "loss": 0.172,
+      "step": 2050
+    },
+    {
+      "grad_norm": 0.4500002861022949,
+      "learning_rate": 9.923543478576048e-05,
+      "loss": 0.1632,
+      "step": 2060
+    },
+    {
+      "grad_norm": 0.45614978671073914,
+      "learning_rate": 9.922096502013813e-05,
+      "loss": 0.1713,
+      "step": 2070
+    },
+    {
+      "grad_norm": 0.5363633036613464,
+      "learning_rate": 9.92063606862769e-05,
+      "loss": 0.1668,
+      "step": 2080
+    },
+    {
+      "grad_norm": 0.7095813751220703,
+      "learning_rate": 9.919162182410453e-05,
+      "loss": 0.193,
+      "step": 2090
+    },
+    {
+      "grad_norm": 0.3974549174308777,
+      "learning_rate": 9.917674847391645e-05,
+      "loss": 0.1419,
+      "step": 2100
+    },
+    {
+      "grad_norm": 0.6337177753448486,
+      "learning_rate": 9.916174067637584e-05,
+      "loss": 0.1681,
+      "step": 2110
+    },
+    {
+      "grad_norm": 0.46541985869407654,
+      "learning_rate": 9.914659847251348e-05,
+      "loss": 0.1559,
+      "step": 2120
+    },
+    {
+      "grad_norm": 0.35531461238861084,
+      "learning_rate": 9.913132190372753e-05,
+      "loss": 0.1606,
+      "step": 2130
+    },
+    {
+      "grad_norm": 0.43313074111938477,
+      "learning_rate": 9.911591101178359e-05,
+      "loss": 0.1841,
+      "step": 2140
+    },
+    {
+      "grad_norm": 0.5053561925888062,
+      "learning_rate": 9.910036583881443e-05,
+      "loss": 0.1675,
+      "step": 2150
+    },
+    {
+      "grad_norm": 0.5011147856712341,
+      "learning_rate": 9.908468642731995e-05,
+      "loss": 0.1805,
+      "step": 2160
+    },
+    {
+      "grad_norm": 0.4348241686820984,
+      "learning_rate": 9.906887282016707e-05,
+      "loss": 0.1271,
+      "step": 2170
+    },
+    {
+      "grad_norm": 0.41252338886260986,
+      "learning_rate": 9.90529250605896e-05,
+      "loss": 0.1583,
+      "step": 2180
+    },
+    {
+      "grad_norm": 0.5953693985939026,
+      "learning_rate": 9.903684319218809e-05,
+      "loss": 0.1658,
+      "step": 2190
+    },
+    {
+      "grad_norm": 0.5093420147895813,
+      "learning_rate": 9.902062725892976e-05,
+      "loss": 0.1737,
+      "step": 2200
+    },
+    {
+      "grad_norm": 0.3964361846446991,
+      "learning_rate": 9.900427730514834e-05,
+      "loss": 0.1351,
+      "step": 2210
+    },
+    {
+      "grad_norm": 0.5651803016662598,
+      "learning_rate": 9.8987793375544e-05,
+      "loss": 0.1668,
+      "step": 2220
+    },
+    {
+      "grad_norm": 0.47457465529441833,
+      "learning_rate": 9.897117551518318e-05,
+      "loss": 0.1571,
+      "step": 2230
+    },
+    {
+      "grad_norm": 0.4295869469642639,
+      "learning_rate": 9.895442376949844e-05,
+      "loss": 0.1525,
+      "step": 2240
+    },
+    {
+      "grad_norm": 0.45591041445732117,
+      "learning_rate": 9.893753818428845e-05,
+      "loss": 0.1577,
+      "step": 2250
+    },
+    {
+      "grad_norm": 0.3740297555923462,
+      "learning_rate": 9.892051880571773e-05,
+      "loss": 0.1555,
+      "step": 2260
+    },
+    {
+      "grad_norm": 0.34207215905189514,
+      "learning_rate": 9.890336568031663e-05,
+      "loss": 0.1527,
+      "step": 2270
+    },
+    {
+      "grad_norm": 0.48910239338874817,
+      "learning_rate": 9.888607885498113e-05,
+      "loss": 0.1564,
+      "step": 2280
+    },
+    {
+      "grad_norm": 0.4521818161010742,
+      "learning_rate": 9.886865837697275e-05,
+      "loss": 0.1663,
+      "step": 2290
+    },
+    {
+      "grad_norm": 0.41172903776168823,
+      "learning_rate": 9.88511042939184e-05,
+      "loss": 0.1565,
+      "step": 2300
+    },
+    {
+      "grad_norm": 0.392194926738739,
+      "learning_rate": 9.883341665381028e-05,
+      "loss": 0.1521,
+      "step": 2310
+    },
+    {
+      "grad_norm": 0.47184863686561584,
+      "learning_rate": 9.881559550500575e-05,
+      "loss": 0.1508,
+      "step": 2320
+    },
+    {
+      "grad_norm": 0.2775649130344391,
+      "learning_rate": 9.879764089622712e-05,
+      "loss": 0.1595,
+      "step": 2330
+    },
+    {
+      "grad_norm": 0.5155332684516907,
+      "learning_rate": 9.87795528765616e-05,
+      "loss": 0.1761,
+      "step": 2340
+    },
+    {
+      "grad_norm": 0.38401615619659424,
+      "learning_rate": 9.876133149546118e-05,
+      "loss": 0.1609,
+      "step": 2350
+    },
+    {
+      "grad_norm": 0.46312469244003296,
+      "learning_rate": 9.874297680274238e-05,
+      "loss": 0.1644,
+      "step": 2360
+    },
+    {
+      "grad_norm": 0.5208353400230408,
+      "learning_rate": 9.872448884858624e-05,
+      "loss": 0.1597,
+      "step": 2370
+    },
+    {
+      "grad_norm": 0.5717175006866455,
+      "learning_rate": 9.870586768353815e-05,
+      "loss": 0.1757,
+      "step": 2380
+    },
+    {
+      "grad_norm": 0.6189532279968262,
+      "learning_rate": 9.868711335850764e-05,
+      "loss": 0.1379,
+      "step": 2390
+    },
+    {
+      "grad_norm": 0.47163233160972595,
+      "learning_rate": 9.866822592476833e-05,
+      "loss": 0.1581,
+      "step": 2400
+    },
+    {
+      "grad_norm": 0.44464004039764404,
+      "learning_rate": 9.86492054339577e-05,
+      "loss": 0.1441,
+      "step": 2410
+    },
+    {
+      "grad_norm": 0.44643107056617737,
+      "learning_rate": 9.863005193807711e-05,
+      "loss": 0.1671,
+      "step": 2420
+    },
+    {
+      "grad_norm": 0.4269534945487976,
+      "learning_rate": 9.861076548949143e-05,
+      "loss": 0.1611,
+      "step": 2430
+    },
+    {
+      "grad_norm": 0.3271143436431885,
+      "learning_rate": 9.859134614092912e-05,
+      "loss": 0.1695,
+      "step": 2440
+    },
+    {
+      "grad_norm": 0.44474682211875916,
+      "learning_rate": 9.857179394548191e-05,
+      "loss": 0.152,
+      "step": 2450
+    },
+    {
+      "grad_norm": 0.4499410092830658,
+      "learning_rate": 9.855210895660477e-05,
+      "loss": 0.1579,
+      "step": 2460
+    },
+    {
+      "grad_norm": 0.6365428566932678,
+      "learning_rate": 9.853229122811568e-05,
+      "loss": 0.1954,
+      "step": 2470
+    },
+    {
+      "grad_norm": 0.33451956510543823,
+      "learning_rate": 9.851234081419559e-05,
+      "loss": 0.1395,
+      "step": 2480
+    },
+    {
+      "grad_norm": 0.41435596346855164,
+      "learning_rate": 9.849225776938814e-05,
+      "loss": 0.1517,
+      "step": 2490
+    },
+    {
+      "grad_norm": 0.39582446217536926,
+      "learning_rate": 9.847204214859964e-05,
+      "loss": 0.1524,
+      "step": 2500
+    },
+    {
+      "grad_norm": 0.41536810994148254,
+      "learning_rate": 9.845169400709879e-05,
+      "loss": 0.1439,
+      "step": 2510
+    },
+    {
+      "grad_norm": 0.5069689750671387,
+      "learning_rate": 9.843121340051664e-05,
+      "loss": 0.1571,
+      "step": 2520
+    },
+    {
+      "grad_norm": 0.4008333086967468,
+      "learning_rate": 9.841060038484641e-05,
+      "loss": 0.1685,
+      "step": 2530
+    },
+    {
+      "grad_norm": 0.497463196516037,
+      "learning_rate": 9.838985501644328e-05,
+      "loss": 0.1492,
+      "step": 2540
+    },
+    {
+      "grad_norm": 0.49305829405784607,
+      "learning_rate": 9.83689773520243e-05,
+      "loss": 0.1684,
+      "step": 2550
+    },
+    {
+      "grad_norm": 0.49042460322380066,
+      "learning_rate": 9.834796744866819e-05,
+      "loss": 0.1718,
+      "step": 2560
+    },
+    {
+      "grad_norm": 0.38399025797843933,
+      "learning_rate": 9.832682536381525e-05,
+      "loss": 0.1286,
+      "step": 2570
+    },
+    {
+      "grad_norm": 0.6375200152397156,
+      "learning_rate": 9.830555115526711e-05,
+      "loss": 0.1464,
+      "step": 2580
+    },
+    {
+      "grad_norm": 0.4116520881652832,
+      "learning_rate": 9.828414488118667e-05,
+      "loss": 0.1414,
+      "step": 2590
+    },
+    {
+      "grad_norm": 0.6042519807815552,
+      "learning_rate": 9.826260660009785e-05,
+      "loss": 0.1567,
+      "step": 2600
+    },
+    {
+      "grad_norm": 0.8501799702644348,
+      "learning_rate": 9.824093637088547e-05,
+      "loss": 0.1686,
+      "step": 2610
+    },
+    {
+      "grad_norm": 0.33054319024086,
+      "learning_rate": 9.821913425279514e-05,
+      "loss": 0.1785,
+      "step": 2620
+    },
+    {
+      "grad_norm": 0.4994356334209442,
+      "learning_rate": 9.8197200305433e-05,
+      "loss": 0.1663,
+      "step": 2630
+    },
+    {
+      "grad_norm": 0.637952446937561,
+      "learning_rate": 9.817513458876564e-05,
+      "loss": 0.1689,
+      "step": 2640
+    },
+    {
+      "grad_norm": 0.5916326642036438,
+      "learning_rate": 9.815293716311987e-05,
+      "loss": 0.1491,
+      "step": 2650
+    },
+    {
+      "grad_norm": 0.7356970906257629,
+      "learning_rate": 9.813060808918262e-05,
+      "loss": 0.1678,
+      "step": 2660
+    },
+    {
+      "grad_norm": 0.5041266083717346,
+      "learning_rate": 9.810814742800069e-05,
+      "loss": 0.16,
+      "step": 2670
+    },
+    {
+      "grad_norm": 0.427004873752594,
+      "learning_rate": 9.808555524098074e-05,
+      "loss": 0.1548,
+      "step": 2680
+    },
+    {
+      "grad_norm": 0.7032573223114014,
+      "learning_rate": 9.806283158988887e-05,
+      "loss": 0.1627,
+      "step": 2690
+    },
+    {
+      "grad_norm": 0.409481406211853,
+      "learning_rate": 9.803997653685072e-05,
+      "loss": 0.1567,
+      "step": 2700
+    },
+    {
+      "grad_norm": 0.5554487705230713,
+      "learning_rate": 9.801699014435112e-05,
+      "loss": 0.1713,
+      "step": 2710
+    },
+    {
+      "grad_norm": 0.3701111674308777,
+      "learning_rate": 9.799387247523398e-05,
+      "loss": 0.1685,
+      "step": 2720
+    },
+    {
+      "grad_norm": 0.5052011013031006,
+      "learning_rate": 9.797062359270215e-05,
+      "loss": 0.1624,
+      "step": 2730
+    },
+    {
+      "grad_norm": 0.6909388303756714,
+      "learning_rate": 9.794724356031715e-05,
+      "loss": 0.1538,
+      "step": 2740
+    },
+    {
+      "grad_norm": 0.45989280939102173,
+      "learning_rate": 9.792373244199913e-05,
+      "loss": 0.1648,
+      "step": 2750
+    },
+    {
+      "grad_norm": 0.35538172721862793,
+      "learning_rate": 9.790009030202658e-05,
+      "loss": 0.1407,
+      "step": 2760
+    },
+    {
+      "grad_norm": 0.39710453152656555,
+      "learning_rate": 9.78763172050362e-05,
+      "loss": 0.1633,
+      "step": 2770
+    },
+    {
+      "grad_norm": 0.35037481784820557,
+      "learning_rate": 9.785241321602274e-05,
+      "loss": 0.1729,
+      "step": 2780
+    },
+    {
+      "grad_norm": 0.45732420682907104,
+      "learning_rate": 9.782837840033879e-05,
+      "loss": 0.1511,
+      "step": 2790
+    },
+    {
+      "grad_norm": 0.4060347378253937,
+      "learning_rate": 9.780421282369461e-05,
+      "loss": 0.146,
+      "step": 2800
+    },
+    {
+      "grad_norm": 0.46445637941360474,
+      "learning_rate": 9.777991655215797e-05,
+      "loss": 0.1535,
+      "step": 2810
+    },
+    {
+      "grad_norm": 0.5067121386528015,
+      "learning_rate": 9.775548965215394e-05,
+      "loss": 0.155,
+      "step": 2820
+    },
+    {
+      "grad_norm": 0.4202738404273987,
+      "learning_rate": 9.773093219046474e-05,
+      "loss": 0.1638,
+      "step": 2830
+    },
+    {
+      "grad_norm": 0.42346957325935364,
+      "learning_rate": 9.770624423422954e-05,
+      "loss": 0.1553,
+      "step": 2840
+    },
+    {
+      "grad_norm": 0.4668117165565491,
+      "learning_rate": 9.768142585094426e-05,
+      "loss": 0.1694,
+      "step": 2850
+    },
+    {
+      "grad_norm": 0.47686445713043213,
+      "learning_rate": 9.765647710846142e-05,
+      "loss": 0.1552,
+      "step": 2860
+    },
+    {
+      "grad_norm": 0.368721604347229,
+      "learning_rate": 9.763139807498991e-05,
+      "loss": 0.1642,
+      "step": 2870
+    },
+    {
+      "grad_norm": 0.3819839656352997,
+      "learning_rate": 9.760618881909487e-05,
+      "loss": 0.1634,
+      "step": 2880
+    },
+    {
+      "grad_norm": 0.5916709303855896,
+      "learning_rate": 9.758084940969744e-05,
+      "loss": 0.1432,
+      "step": 2890
+    },
+    {
+      "grad_norm": 0.3503391444683075,
+      "learning_rate": 9.755537991607459e-05,
+      "loss": 0.1406,
+      "step": 2900
+    },
+    {
+      "grad_norm": 0.5615946054458618,
+      "learning_rate": 9.752978040785895e-05,
+      "loss": 0.1667,
+      "step": 2910
+    },
+    {
+      "grad_norm": 0.3928743898868561,
+      "learning_rate": 9.750405095503859e-05,
+      "loss": 0.1678,
+      "step": 2920
+    },
+    {
+      "grad_norm": 0.35815370082855225,
+      "learning_rate": 9.747819162795686e-05,
+      "loss": 0.158,
+      "step": 2930
+    },
+    {
+      "grad_norm": 0.489862322807312,
+      "learning_rate": 9.745220249731217e-05,
+      "loss": 0.1524,
+      "step": 2940
+    },
+    {
+      "grad_norm": 0.49589449167251587,
+      "learning_rate": 9.742608363415781e-05,
+      "loss": 0.178,
+      "step": 2950
+    },
+    {
+      "grad_norm": 0.6664644479751587,
+      "learning_rate": 9.739983510990176e-05,
+      "loss": 0.1673,
+      "step": 2960
+    },
+    {
+      "grad_norm": 0.4060116708278656,
+      "learning_rate": 9.737345699630647e-05,
+      "loss": 0.1525,
+      "step": 2970
+    },
+    {
+      "grad_norm": 0.46838751435279846,
+      "learning_rate": 9.734694936548869e-05,
+      "loss": 0.1586,
+      "step": 2980
+    },
+    {
+      "grad_norm": 0.46751508116722107,
+      "learning_rate": 9.732031228991932e-05,
+      "loss": 0.1492,
+      "step": 2990
+    },
+    {
+      "grad_norm": 0.4857649505138397,
+      "learning_rate": 9.729354584242302e-05,
+      "loss": 0.144,
+      "step": 3000
+    },
+    {
+      "grad_norm": 0.36312589049339294,
+      "learning_rate": 9.726665009617832e-05,
+      "loss": 0.1715,
+      "step": 3010
+    },
+    {
+      "grad_norm": 0.42318475246429443,
+      "learning_rate": 9.723962512471714e-05,
+      "loss": 0.1352,
+      "step": 3020
+    },
+    {
+      "grad_norm": 0.5730621218681335,
+      "learning_rate": 9.72124710019247e-05,
+      "loss": 0.1712,
+      "step": 3030
+    },
+    {
+      "grad_norm": 0.34390145540237427,
+      "learning_rate": 9.718518780203934e-05,
+      "loss": 0.1545,
+      "step": 3040
+    },
+    {
+      "grad_norm": 0.5548776388168335,
+      "learning_rate": 9.715777559965228e-05,
+      "loss": 0.1729,
+      "step": 3050
+    },
+    {
+      "grad_norm": 0.4265879988670349,
+      "learning_rate": 9.713023446970746e-05,
+      "loss": 0.1711,
+      "step": 3060
+    },
+    {
+      "grad_norm": 0.4300907850265503,
+      "learning_rate": 9.710256448750126e-05,
+      "loss": 0.1862,
+      "step": 3070
+    },
+    {
+      "grad_norm": 0.3605111539363861,
+      "learning_rate": 9.707476572868235e-05,
+      "loss": 0.1725,
+      "step": 3080
+    },
+    {
+      "grad_norm": 0.4023027718067169,
+      "learning_rate": 9.704683826925149e-05,
+      "loss": 0.1811,
+      "step": 3090
+    },
+    {
+      "grad_norm": 0.5678909420967102,
+      "learning_rate": 9.701878218556129e-05,
+      "loss": 0.1686,
+      "step": 3100
+    },
+    {
+      "grad_norm": 0.3750840425491333,
+      "learning_rate": 9.699059755431598e-05,
+      "loss": 0.1601,
+      "step": 3110
+    },
+    {
+      "grad_norm": 0.8199723362922668,
+      "learning_rate": 9.696228445257132e-05,
+      "loss": 0.1707,
+      "step": 3120
+    },
+    {
+      "grad_norm": 0.5791060924530029,
+      "learning_rate": 9.693384295773419e-05,
+      "loss": 0.1507,
+      "step": 3130
+    },
+    {
+      "grad_norm": 0.5557647943496704,
+      "learning_rate": 9.690527314756259e-05,
+      "loss": 0.15,
+      "step": 3140
+    },
+    {
+      "grad_norm": 0.38979536294937134,
+      "learning_rate": 9.687657510016527e-05,
+      "loss": 0.176,
+      "step": 3150
+    },
+    {
+      "grad_norm": 0.3244946599006653,
+      "learning_rate": 9.684774889400161e-05,
+      "loss": 0.1474,
+      "step": 3160
+    },
+    {
+      "grad_norm": 0.42470288276672363,
+      "learning_rate": 9.681879460788135e-05,
+      "loss": 0.1669,
+      "step": 3170
+    },
+    {
+      "grad_norm": 0.5319637060165405,
+      "learning_rate": 9.67897123209644e-05,
+      "loss": 0.1828,
+      "step": 3180
+    },
+    {
+      "grad_norm": 0.42851802706718445,
+      "learning_rate": 9.676050211276062e-05,
+      "loss": 0.1558,
+      "step": 3190
+    },
+    {
+      "grad_norm": 0.4439624547958374,
+      "learning_rate": 9.673116406312962e-05,
+      "loss": 0.1682,
+      "step": 3200
+    },
+    {
+      "grad_norm": 0.3958091139793396,
+      "learning_rate": 9.67016982522805e-05,
+      "loss": 0.1612,
+      "step": 3210
+    },
+    {
+      "grad_norm": 0.6761658191680908,
+      "learning_rate": 9.667210476077164e-05,
+      "loss": 0.1681,
+      "step": 3220
+    },
+    {
+      "grad_norm": 0.725614607334137,
+      "learning_rate": 9.664238366951055e-05,
+      "loss": 0.1765,
+      "step": 3230
+    },
+    {
+      "grad_norm": 0.5815354585647583,
+      "learning_rate": 9.661253505975355e-05,
+      "loss": 0.1432,
+      "step": 3240
+    },
+    {
+      "grad_norm": 0.5952789783477783,
+      "learning_rate": 9.658255901310557e-05,
+      "loss": 0.1473,
+      "step": 3250
+    },
+    {
+      "grad_norm": 0.7221768498420715,
+      "learning_rate": 9.655245561152e-05,
+      "loss": 0.1621,
+      "step": 3260
+    },
+    {
+      "grad_norm": 0.4501568377017975,
+      "learning_rate": 9.65222249372984e-05,
+      "loss": 0.1806,
+      "step": 3270
+    },
+    {
+      "grad_norm": 0.7780941128730774,
+      "learning_rate": 9.649186707309026e-05,
+      "loss": 0.1974,
+      "step": 3280
+    },
+    {
+      "grad_norm": 0.6010203957557678,
+      "learning_rate": 9.646138210189283e-05,
+      "loss": 0.1692,
+      "step": 3290
+    },
+    {
+      "grad_norm": 0.5620298981666565,
+      "learning_rate": 9.643077010705087e-05,
+      "loss": 0.1572,
+      "step": 3300
+    },
+    {
+      "grad_norm": 0.5052732825279236,
+      "learning_rate": 9.640003117225637e-05,
+      "loss": 0.1469,
+      "step": 3310
+    },
+    {
+      "grad_norm": 0.3884584307670593,
+      "learning_rate": 9.636916538154846e-05,
+      "loss": 0.1621,
+      "step": 3320
+    },
+    {
+      "grad_norm": 0.5137442946434021,
+      "learning_rate": 9.633817281931296e-05,
+      "loss": 0.149,
+      "step": 3330
+    },
+    {
+      "grad_norm": 0.7076863050460815,
+      "learning_rate": 9.630705357028242e-05,
+      "loss": 0.1729,
+      "step": 3340
+    },
+    {
+      "grad_norm": 0.568705141544342,
+      "learning_rate": 9.627580771953563e-05,
+      "loss": 0.1825,
+      "step": 3350
+    },
+    {
+      "grad_norm": 0.4905519485473633,
+      "learning_rate": 9.624443535249759e-05,
+      "loss": 0.1692,
+      "step": 3360
+    },
+    {
+      "grad_norm": 0.6092402935028076,
+      "learning_rate": 9.621293655493913e-05,
+      "loss": 0.1782,
+      "step": 3370
+    },
+    {
+      "grad_norm": 0.4784291982650757,
+      "learning_rate": 9.618131141297675e-05,
+      "loss": 0.1545,
+      "step": 3380
+    },
+    {
+      "grad_norm": 0.5267539024353027,
+      "learning_rate": 9.614956001307242e-05,
+      "loss": 0.1579,
+      "step": 3390
+    },
+    {
+      "grad_norm": 0.49305450916290283,
+      "learning_rate": 9.611768244203321e-05,
+      "loss": 0.1766,
+      "step": 3400
+    },
+    {
+      "grad_norm": 0.590823233127594,
+      "learning_rate": 9.60856787870112e-05,
+      "loss": 0.1642,
+      "step": 3410
+    },
+    {
+      "grad_norm": 0.5640073418617249,
+      "learning_rate": 9.605354913550318e-05,
+      "loss": 0.1514,
+      "step": 3420
+    },
+    {
+      "grad_norm": 0.47256362438201904,
+      "learning_rate": 9.602129357535037e-05,
+      "loss": 0.1719,
+      "step": 3430
+    },
+    {
+      "grad_norm": 0.5772477388381958,
+      "learning_rate": 9.598891219473825e-05,
+      "loss": 0.1585,
+      "step": 3440
+    },
+    {
+      "grad_norm": 0.6106401681900024,
+      "learning_rate": 9.595640508219625e-05,
+      "loss": 0.1622,
+      "step": 3450
+    },
+    {
+      "grad_norm": 0.3641313314437866,
+      "learning_rate": 9.592377232659761e-05,
+      "loss": 0.1646,
+      "step": 3460
+    },
+    {
+      "grad_norm": 0.5276760458946228,
+      "learning_rate": 9.589101401715904e-05,
+      "loss": 0.1557,
+      "step": 3470
+    },
+    {
+      "grad_norm": 0.3947807848453522,
+      "learning_rate": 9.585813024344045e-05,
+      "loss": 0.1581,
+      "step": 3480
+    },
+    {
+      "grad_norm": 0.4768114984035492,
+      "learning_rate": 9.58251210953449e-05,
+      "loss": 0.1632,
+      "step": 3490
+    },
+    {
+      "grad_norm": 0.6272286176681519,
+      "learning_rate": 9.579198666311809e-05,
+      "loss": 0.166,
+      "step": 3500
+    },
+    {
+      "grad_norm": 0.40389296412467957,
+      "learning_rate": 9.575872703734832e-05,
+      "loss": 0.161,
+      "step": 3510
+    },
+    {
+      "grad_norm": 0.5940328240394592,
+      "learning_rate": 9.572534230896611e-05,
+      "loss": 0.1544,
+      "step": 3520
+    },
+    {
+      "grad_norm": 0.615892231464386,
+      "learning_rate": 9.569183256924403e-05,
+      "loss": 0.162,
+      "step": 3530
+    },
+    {
+      "grad_norm": 0.7142258286476135,
+      "learning_rate": 9.565819790979646e-05,
+      "loss": 0.1579,
+      "step": 3540
+    },
+    {
+      "grad_norm": 0.46866926550865173,
+      "learning_rate": 9.562443842257925e-05,
+      "loss": 0.15,
+      "step": 3550
+    },
+    {
+      "grad_norm": 0.31648173928260803,
+      "learning_rate": 9.559055419988956e-05,
+      "loss": 0.1715,
+      "step": 3560
+    },
+    {
+      "grad_norm": 0.4224969446659088,
+      "learning_rate": 9.555654533436557e-05,
+      "loss": 0.1339,
+      "step": 3570
+    },
+    {
+      "grad_norm": 0.5087144374847412,
+      "learning_rate": 9.552241191898621e-05,
+      "loss": 0.1711,
+      "step": 3580
+    },
+    {
+      "grad_norm": 0.4774325489997864,
+      "learning_rate": 9.548815404707092e-05,
+      "loss": 0.1468,
+      "step": 3590
+    },
+    {
+      "grad_norm": 0.5909769535064697,
+      "learning_rate": 9.545377181227942e-05,
+      "loss": 0.16,
+      "step": 3600
+    },
+    {
+      "grad_norm": 0.5786628127098083,
+      "learning_rate": 9.541926530861145e-05,
+      "loss": 0.1355,
+      "step": 3610
+    },
+    {
+      "grad_norm": 0.5397319793701172,
+      "learning_rate": 9.538463463040645e-05,
+      "loss": 0.1693,
+      "step": 3620
+    },
+    {
+      "grad_norm": 0.37220314145088196,
+      "learning_rate": 9.534987987234337e-05,
+      "loss": 0.1408,
+      "step": 3630
+    },
+    {
+      "grad_norm": 0.3543132245540619,
+      "learning_rate": 9.53150011294404e-05,
+      "loss": 0.1564,
+      "step": 3640
+    },
+    {
+      "grad_norm": 0.44777625799179077,
+      "learning_rate": 9.527999849705471e-05,
+      "loss": 0.165,
+      "step": 3650
+    },
+    {
+      "grad_norm": 0.5717726945877075,
+      "learning_rate": 9.524487207088213e-05,
+      "loss": 0.1843,
+      "step": 3660
+    },
+    {
+      "grad_norm": 0.6085203289985657,
+      "learning_rate": 9.520962194695698e-05,
+      "loss": 0.14,
+      "step": 3670
+    },
+    {
+      "grad_norm": 0.4424811005592346,
+      "learning_rate": 9.517424822165175e-05,
+      "loss": 0.1674,
+      "step": 3680
+    },
+    {
+      "grad_norm": 0.6584381461143494,
+      "learning_rate": 9.513875099167685e-05,
+      "loss": 0.1704,
+      "step": 3690
+    },
+    {
+      "grad_norm": 0.4689304232597351,
+      "learning_rate": 9.510313035408035e-05,
+      "loss": 0.1569,
+      "step": 3700
+    },
+    {
+      "grad_norm": 0.380482017993927,
+      "learning_rate": 9.506738640624775e-05,
+      "loss": 0.1465,
+      "step": 3710
+    },
+    {
+      "grad_norm": 0.4750548005104065,
+      "learning_rate": 9.50315192459016e-05,
+      "loss": 0.1408,
+      "step": 3720
+    },
+    {
+      "grad_norm": 0.58077073097229,
+      "learning_rate": 9.499552897110136e-05,
+      "loss": 0.1463,
+      "step": 3730
+    },
+    {
+      "grad_norm": 0.5339252948760986,
+      "learning_rate": 9.495941568024304e-05,
+      "loss": 0.1573,
+      "step": 3740
+    },
+    {
+      "grad_norm": 0.4519467353820801,
+      "learning_rate": 9.492317947205904e-05,
+      "loss": 0.153,
+      "step": 3750
+    },
+    {
+      "grad_norm": 0.4003654420375824,
+      "learning_rate": 9.488682044561775e-05,
+      "loss": 0.1435,
+      "step": 3760
+    },
+    {
+      "grad_norm": 0.6126506924629211,
+      "learning_rate": 9.485033870032335e-05,
+      "loss": 0.1671,
+      "step": 3770
+    },
+    {
+      "grad_norm": 0.5795039534568787,
+      "learning_rate": 9.481373433591556e-05,
+      "loss": 0.1547,
+      "step": 3780
+    },
+    {
+      "grad_norm": 0.5838664174079895,
+      "learning_rate": 9.47770074524693e-05,
+      "loss": 0.1434,
+      "step": 3790
+    },
+    {
+      "grad_norm": 0.4138217866420746,
+      "learning_rate": 9.474015815039446e-05,
+      "loss": 0.1368,
+      "step": 3800
+    },
+    {
+      "grad_norm": 0.588933527469635,
+      "learning_rate": 9.470318653043565e-05,
+      "loss": 0.1387,
+      "step": 3810
+    },
+    {
+      "grad_norm": 0.433880090713501,
+      "learning_rate": 9.466609269367185e-05,
+      "loss": 0.1471,
+      "step": 3820
+    },
+    {
+      "grad_norm": 0.6373485922813416,
+      "learning_rate": 9.46288767415162e-05,
+      "loss": 0.1454,
+      "step": 3830
+    },
+    {
+      "grad_norm": 0.5456726551055908,
+      "learning_rate": 9.459153877571567e-05,
+      "loss": 0.1468,
+      "step": 3840
+    },
+    {
+      "grad_norm": 0.6738699674606323,
+      "learning_rate": 9.455407889835087e-05,
+      "loss": 0.1621,
+      "step": 3850
+    },
+    {
+      "grad_norm": 0.5350501537322998,
+      "learning_rate": 9.451649721183564e-05,
+      "loss": 0.1545,
+      "step": 3860
+    },
+    {
+      "grad_norm": 0.4198494553565979,
+      "learning_rate": 9.447879381891692e-05,
+      "loss": 0.16,
+      "step": 3870
+    },
+    {
+      "grad_norm": 0.5458757281303406,
+      "learning_rate": 9.444096882267428e-05,
+      "loss": 0.1494,
+      "step": 3880
+    },
+    {
+      "grad_norm": 0.6553415656089783,
+      "learning_rate": 9.440302232651988e-05,
+      "loss": 0.1571,
+      "step": 3890
+    },
+    {
+      "grad_norm": 0.5958887338638306,
+      "learning_rate": 9.436495443419795e-05,
+      "loss": 0.1571,
+      "step": 3900
+    },
+    {
+      "grad_norm": 0.5396506786346436,
+      "learning_rate": 9.432676524978466e-05,
+      "loss": 0.1623,
+      "step": 3910
+    },
+    {
+      "grad_norm": 0.5331987142562866,
+      "learning_rate": 9.42884548776878e-05,
+      "loss": 0.1691,
+      "step": 3920
+    },
+    {
+      "grad_norm": 0.5442116260528564,
+      "learning_rate": 9.425002342264646e-05,
+      "loss": 0.154,
+      "step": 3930
+    },
+    {
+      "grad_norm": 0.49666348099708557,
+      "learning_rate": 9.421147098973077e-05,
+      "loss": 0.1495,
+      "step": 3940
+    },
+    {
+      "grad_norm": 0.3620719313621521,
+      "learning_rate": 9.41727976843416e-05,
+      "loss": 0.1556,
+      "step": 3950
+    },
+    {
+      "grad_norm": 0.36253923177719116,
+      "learning_rate": 9.413400361221029e-05,
+      "loss": 0.1352,
+      "step": 3960
+    },
+    {
+      "grad_norm": 0.4730953276157379,
+      "learning_rate": 9.409508887939835e-05,
+      "loss": 0.1697,
+      "step": 3970
+    },
+    {
+      "grad_norm": 0.49778151512145996,
+      "learning_rate": 9.40560535922972e-05,
+      "loss": 0.1506,
+      "step": 3980
+    },
+    {
+      "grad_norm": 0.39833173155784607,
+      "learning_rate": 9.40168978576278e-05,
+      "loss": 0.1418,
+      "step": 3990
+    },
+    {
+      "grad_norm": 0.35803860425949097,
+      "learning_rate": 9.397762178244043e-05,
+      "loss": 0.1462,
+      "step": 4000
+    },
+    {
+      "grad_norm": 0.4511455297470093,
+      "learning_rate": 9.393822547411439e-05,
+      "loss": 0.1493,
+      "step": 4010
+    },
+    {
+      "grad_norm": 0.5506104230880737,
+      "learning_rate": 9.389870904035769e-05,
+      "loss": 0.1674,
+      "step": 4020
+    },
+    {
+      "grad_norm": 0.4511580169200897,
+      "learning_rate": 9.385907258920672e-05,
+      "loss": 0.1393,
+      "step": 4030
+    },
+    {
+      "grad_norm": 0.3634929955005646,
+      "learning_rate": 9.381931622902607e-05,
+      "loss": 0.1473,
+      "step": 4040
+    },
+    {
+      "grad_norm": 0.4699901342391968,
+      "learning_rate": 9.377944006850807e-05,
+      "loss": 0.1484,
+      "step": 4050
+    },
+    {
+      "grad_norm": 0.4500221610069275,
+      "learning_rate": 9.373944421667265e-05,
+      "loss": 0.1715,
+      "step": 4060
+    },
+    {
+      "grad_norm": 0.37292906641960144,
+      "learning_rate": 9.369932878286691e-05,
+      "loss": 0.1341,
+      "step": 4070
+    },
+    {
+      "grad_norm": 0.2896611988544464,
+      "learning_rate": 9.365909387676494e-05,
+      "loss": 0.1367,
+      "step": 4080
+    },
+    {
+      "grad_norm": 0.4624013602733612,
+      "learning_rate": 9.361873960836744e-05,
+      "loss": 0.1522,
+      "step": 4090
+    },
+    {
+      "grad_norm": 0.41886767745018005,
+      "learning_rate": 9.357826608800142e-05,
+      "loss": 0.164,
+      "step": 4100
+    },
+    {
+      "grad_norm": 0.6623914837837219,
+      "learning_rate": 9.353767342631994e-05,
+      "loss": 0.1725,
+      "step": 4110
+    },
+    {
+      "grad_norm": 0.43481212854385376,
+      "learning_rate": 9.34969617343018e-05,
+      "loss": 0.1641,
+      "step": 4120
+    },
+    {
+      "grad_norm": 0.4176675081253052,
+      "learning_rate": 9.345613112325122e-05,
+      "loss": 0.148,
+      "step": 4130
+    },
+    {
+      "grad_norm": 0.4567943215370178,
+      "learning_rate": 9.34151817047975e-05,
+      "loss": 0.1738,
+      "step": 4140
+    },
+    {
+      "grad_norm": 0.5684789419174194,
+      "learning_rate": 9.33741135908948e-05,
+      "loss": 0.152,
+      "step": 4150
+    },
+    {
+      "grad_norm": 0.3954605162143707,
+      "learning_rate": 9.33329268938218e-05,
+      "loss": 0.158,
+      "step": 4160
+    },
+    {
+      "grad_norm": 0.4392043650150299,
+      "learning_rate": 9.329162172618132e-05,
+      "loss": 0.1585,
+      "step": 4170
+    },
+    {
+      "grad_norm": 0.4697527289390564,
+      "learning_rate": 9.325019820090013e-05,
+      "loss": 0.1583,
+      "step": 4180
+    },
+    {
+      "grad_norm": 0.4229051172733307,
+      "learning_rate": 9.320865643122855e-05,
+      "loss": 0.155,
+      "step": 4190
+    },
+    {
+      "grad_norm": 0.48601073026657104,
+      "learning_rate": 9.316699653074023e-05,
+      "loss": 0.1744,
+      "step": 4200
+    },
+    {
+      "grad_norm": 0.5530809760093689,
+      "learning_rate": 9.312521861333172e-05,
+      "loss": 0.1514,
+      "step": 4210
+    },
+    {
+      "grad_norm": 0.6378450393676758,
+      "learning_rate": 9.308332279322224e-05,
+      "loss": 0.1593,
+      "step": 4220
+    },
+    {
+      "grad_norm": 0.3873921036720276,
+      "learning_rate": 9.304130918495338e-05,
+      "loss": 0.1542,
+      "step": 4230
+    },
+    {
+      "grad_norm": 0.4033936858177185,
+      "learning_rate": 9.299917790338874e-05,
+      "loss": 0.1546,
+      "step": 4240
+    },
+    {
+      "grad_norm": 0.46620792150497437,
+      "learning_rate": 9.295692906371363e-05,
+      "loss": 0.1729,
+      "step": 4250
+    },
+    {
+      "grad_norm": 0.4556692838668823,
+      "learning_rate": 9.291456278143476e-05,
+      "loss": 0.1653,
+      "step": 4260
+    },
+    {
+      "grad_norm": 0.4371429979801178,
+      "learning_rate": 9.287207917237994e-05,
+      "loss": 0.1457,
+      "step": 4270
+    },
+    {
+      "grad_norm": 0.472690224647522,
+      "learning_rate": 9.282947835269773e-05,
+      "loss": 0.1671,
+      "step": 4280
+    },
+    {
+      "grad_norm": 0.5881608128547668,
+      "learning_rate": 9.278676043885715e-05,
+      "loss": 0.1494,
+      "step": 4290
+    },
+    {
+      "grad_norm": 0.5272968411445618,
+      "learning_rate": 9.274392554764733e-05,
+      "loss": 0.1314,
+      "step": 4300
+    },
+    {
+      "grad_norm": 0.46066397428512573,
+      "learning_rate": 9.270097379617723e-05,
+      "loss": 0.1489,
+      "step": 4310
+    },
+    {
+      "grad_norm": 0.45993587374687195,
+      "learning_rate": 9.26579053018753e-05,
+      "loss": 0.1681,
+      "step": 4320
+    },
+    {
+      "grad_norm": 0.5181461572647095,
+      "learning_rate": 9.261472018248918e-05,
+      "loss": 0.1555,
+      "step": 4330
+    },
+    {
+      "grad_norm": 0.571498453617096,
+      "learning_rate": 9.25714185560853e-05,
+      "loss": 0.1673,
+      "step": 4340
+    },
+    {
+      "grad_norm": 0.4522469937801361,
+      "learning_rate": 9.252800054104868e-05,
+      "loss": 0.1524,
+      "step": 4350
+    },
+    {
+      "grad_norm": 0.46526020765304565,
+      "learning_rate": 9.248446625608252e-05,
+      "loss": 0.1591,
+      "step": 4360
+    },
+    {
+      "grad_norm": 0.5256683826446533,
+      "learning_rate": 9.244081582020789e-05,
+      "loss": 0.1781,
+      "step": 4370
+    },
+    {
+      "grad_norm": 0.4290837049484253,
+      "learning_rate": 9.239704935276339e-05,
+      "loss": 0.1591,
+      "step": 4380
+    },
+    {
+      "grad_norm": 0.4445662498474121,
+      "learning_rate": 9.235316697340489e-05,
+      "loss": 0.1655,
+      "step": 4390
+    },
+    {
+      "grad_norm": 0.4488997459411621,
+      "learning_rate": 9.230916880210512e-05,
+      "loss": 0.1608,
+      "step": 4400
+    },
+    {
+      "grad_norm": 0.4629778265953064,
+      "learning_rate": 9.226505495915342e-05,
+      "loss": 0.1669,
+      "step": 4410
+    },
+    {
+      "grad_norm": 0.38739582896232605,
+      "learning_rate": 9.222082556515536e-05,
+      "loss": 0.1552,
+      "step": 4420
+    },
+    {
+      "grad_norm": 0.5273932814598083,
+      "learning_rate": 9.217648074103242e-05,
+      "loss": 0.1499,
+      "step": 4430
+    },
+    {
+      "grad_norm": 0.4120842218399048,
+      "learning_rate": 9.213202060802161e-05,
+      "loss": 0.1488,
+      "step": 4440
+    },
+    {
+      "grad_norm": 0.3304223418235779,
+      "learning_rate": 9.208744528767528e-05,
+      "loss": 0.1398,
+      "step": 4450
+    },
+    {
+      "grad_norm": 0.4437789022922516,
+      "learning_rate": 9.204275490186064e-05,
+      "loss": 0.148,
+      "step": 4460
+    },
+    {
+      "grad_norm": 0.4289587438106537,
+      "learning_rate": 9.199794957275949e-05,
+      "loss": 0.1284,
+      "step": 4470
+    },
+    {
+      "grad_norm": 0.4686824381351471,
+      "learning_rate": 9.19530294228679e-05,
+      "loss": 0.1797,
+      "step": 4480
+    },
+    {
+      "grad_norm": 0.5537347793579102,
+      "learning_rate": 9.190799457499583e-05,
+      "loss": 0.156,
+      "step": 4490
+    },
+    {
+      "grad_norm": 0.3553921580314636,
+      "learning_rate": 9.186284515226686e-05,
+      "loss": 0.154,
+      "step": 4500
+    },
+    {
+      "grad_norm": 0.5196880102157593,
+      "learning_rate": 9.181758127811777e-05,
+      "loss": 0.1708,
+      "step": 4510
+    },
+    {
+      "grad_norm": 0.422090619802475,
+      "learning_rate": 9.177220307629825e-05,
+      "loss": 0.1757,
+      "step": 4520
+    },
+    {
+      "grad_norm": 0.5996848940849304,
+      "learning_rate": 9.172671067087059e-05,
+      "loss": 0.1583,
+      "step": 4530
+    },
+    {
+      "grad_norm": 0.6522098779678345,
+      "learning_rate": 9.16811041862093e-05,
+      "loss": 0.1871,
+      "step": 4540
+    },
+    {
+      "grad_norm": 0.604037344455719,
+      "learning_rate": 9.163538374700076e-05,
+      "loss": 0.1495,
+      "step": 4550
+    },
+    {
+      "grad_norm": 0.4153933823108673,
+      "learning_rate": 9.158954947824287e-05,
+      "loss": 0.1683,
+      "step": 4560
+    },
+    {
+      "grad_norm": 0.5726081728935242,
+      "learning_rate": 9.154360150524482e-05,
+      "loss": 0.1523,
+      "step": 4570
+    },
+    {
+      "grad_norm": 0.7292566895484924,
+      "learning_rate": 9.14975399536266e-05,
+      "loss": 0.1734,
+      "step": 4580
+    },
+    {
+      "grad_norm": 0.41257768869400024,
+      "learning_rate": 9.14513649493187e-05,
+      "loss": 0.1785,
+      "step": 4590
+    },
+    {
+      "grad_norm": 0.4115147590637207,
+      "learning_rate": 9.140507661856187e-05,
+      "loss": 0.1551,
+      "step": 4600
+    },
+    {
+      "grad_norm": 0.4570203423500061,
+      "learning_rate": 9.135867508790661e-05,
+      "loss": 0.1777,
+      "step": 4610
+    },
+    {
+      "grad_norm": 0.4845227599143982,
+      "learning_rate": 9.131216048421291e-05,
+      "loss": 0.1679,
+      "step": 4620
+    },
+    {
+      "grad_norm": 0.5000484585762024,
+      "learning_rate": 9.126553293464998e-05,
+      "loss": 0.1731,
+      "step": 4630
+    },
+    {
+      "grad_norm": 0.397193044424057,
+      "learning_rate": 9.121879256669572e-05,
+      "loss": 0.1446,
+      "step": 4640
+    },
+    {
+      "grad_norm": 0.37070292234420776,
+      "learning_rate": 9.117193950813652e-05,
+      "loss": 0.16,
+      "step": 4650
+    },
+    {
+      "grad_norm": 0.32511526346206665,
+      "learning_rate": 9.112497388706685e-05,
+      "loss": 0.1682,
+      "step": 4660
+    },
+    {
+      "grad_norm": 0.2601248323917389,
+      "learning_rate": 9.10778958318889e-05,
+      "loss": 0.1551,
+      "step": 4670
+    },
+    {
+      "grad_norm": 0.3405192494392395,
+      "learning_rate": 9.103070547131232e-05,
+      "loss": 0.1519,
+      "step": 4680
+    },
+    {
+      "grad_norm": 0.43947863578796387,
+      "learning_rate": 9.098340293435375e-05,
+      "loss": 0.1694,
+      "step": 4690
+    },
+    {
+      "grad_norm": 0.3517494797706604,
+      "learning_rate": 9.093598835033649e-05,
+      "loss": 0.1601,
+      "step": 4700
+    },
+    {
+      "grad_norm": 0.4725731313228607,
+      "learning_rate": 9.088846184889021e-05,
+      "loss": 0.1471,
+      "step": 4710
+    },
+    {
+      "grad_norm": 0.5243344902992249,
+      "learning_rate": 9.084082355995057e-05,
+      "loss": 0.1432,
+      "step": 4720
+    },
+    {
+      "grad_norm": 0.4642360508441925,
+      "learning_rate": 9.079307361375882e-05,
+      "loss": 0.1451,
+      "step": 4730
+    },
+    {
+      "grad_norm": 0.528448760509491,
+      "learning_rate": 9.074521214086149e-05,
+      "loss": 0.1568,
+      "step": 4740
+    },
+    {
+      "grad_norm": 0.4241181015968323,
+      "learning_rate": 9.069723927211001e-05,
+      "loss": 0.1509,
+      "step": 4750
+    },
+    {
+      "grad_norm": 0.4279056787490845,
+      "learning_rate": 9.064915513866037e-05,
+      "loss": 0.1556,
+      "step": 4760
+    },
+    {
+      "grad_norm": 0.5071821212768555,
+      "learning_rate": 9.060095987197279e-05,
+      "loss": 0.1796,
+      "step": 4770
+    },
+    {
+      "grad_norm": 0.5389459133148193,
+      "learning_rate": 9.055265360381126e-05,
+      "loss": 0.1616,
+      "step": 4780
+    },
+    {
+      "grad_norm": 0.3819650113582611,
+      "learning_rate": 9.050423646624326e-05,
+      "loss": 0.1443,
+      "step": 4790
+    },
+    {
+      "grad_norm": 0.506536066532135,
+      "learning_rate": 9.045570859163943e-05,
+      "loss": 0.1603,
+      "step": 4800
+    },
+    {
+      "grad_norm": 0.4889078140258789,
+      "learning_rate": 9.04070701126731e-05,
+      "loss": 0.1702,
+      "step": 4810
+    },
+    {
+      "grad_norm": 0.6768327951431274,
+      "learning_rate": 9.035832116232001e-05,
+      "loss": 0.1587,
+      "step": 4820
+    },
+    {
+      "grad_norm": 0.7247352004051208,
+      "learning_rate": 9.030946187385796e-05,
+      "loss": 0.1562,
+      "step": 4830
+    },
+    {
+      "grad_norm": 0.4402683675289154,
+      "learning_rate": 9.026049238086635e-05,
+      "loss": 0.1624,
+      "step": 4840
+    },
+    {
+      "grad_norm": 0.43945467472076416,
+      "learning_rate": 9.021141281722591e-05,
+      "loss": 0.1603,
+      "step": 4850
+    },
+    {
+      "grad_norm": 0.6942856311798096,
+      "learning_rate": 9.01622233171183e-05,
+      "loss": 0.1372,
+      "step": 4860
+    },
+    {
+      "grad_norm": 0.418361634016037,
+      "learning_rate": 9.011292401502574e-05,
+      "loss": 0.1578,
+      "step": 4870
+    },
+    {
+      "grad_norm": 0.36947569251060486,
+      "learning_rate": 9.006351504573063e-05,
+      "loss": 0.1649,
+      "step": 4880
+    },
+    {
+      "grad_norm": 0.41154003143310547,
+      "learning_rate": 9.001399654431519e-05,
+      "loss": 0.146,
+      "step": 4890
+    },
+    {
+      "grad_norm": 0.5292714238166809,
+      "learning_rate": 8.996436864616116e-05,
+      "loss": 0.1389,
+      "step": 4900
+    },
+    {
+      "grad_norm": 0.3531433343887329,
+      "learning_rate": 8.991463148694925e-05,
+      "loss": 0.1667,
+      "step": 4910
+    },
+    {
+      "grad_norm": 0.6159530282020569,
+      "learning_rate": 8.986478520265902e-05,
+      "loss": 0.158,
+      "step": 4920
+    },
+    {
+      "grad_norm": 0.4082534909248352,
+      "learning_rate": 8.981482992956827e-05,
+      "loss": 0.1417,
+      "step": 4930
+    },
+    {
+      "grad_norm": 0.5040385127067566,
+      "learning_rate": 8.976476580425282e-05,
+      "loss": 0.164,
+      "step": 4940
+    },
+    {
+      "grad_norm": 0.3659520447254181,
+      "learning_rate": 8.971459296358606e-05,
+      "loss": 0.1452,
+      "step": 4950
+    },
+    {
+      "grad_norm": 0.6556157469749451,
+      "learning_rate": 8.966431154473864e-05,
+      "loss": 0.165,
+      "step": 4960
+    },
+    {
+      "grad_norm": 0.4559183120727539,
+      "learning_rate": 8.961392168517803e-05,
+      "loss": 0.149,
+      "step": 4970
+    },
+    {
+      "grad_norm": 0.5523380041122437,
+      "learning_rate": 8.956342352266821e-05,
+      "loss": 0.1712,
+      "step": 4980
+    },
+    {
+      "grad_norm": 0.4615600109100342,
+      "learning_rate": 8.95128171952692e-05,
+      "loss": 0.1499,
+      "step": 4990
+    },
+    {
+      "grad_norm": 0.3619142174720764,
+      "learning_rate": 8.946210284133676e-05,
+      "loss": 0.1642,
+      "step": 5000
+    },
+    {
+      "grad_norm": 0.47529518604278564,
+      "learning_rate": 8.941128059952201e-05,
+      "loss": 0.1633,
+      "step": 5010
+    },
+    {
+      "grad_norm": 0.6111720204353333,
+      "learning_rate": 8.936035060877102e-05,
+      "loss": 0.1482,
+      "step": 5020
+    },
+    {
+      "grad_norm": 0.563284695148468,
+      "learning_rate": 8.930931300832443e-05,
+      "loss": 0.1413,
+      "step": 5030
+    },
+    {
+      "grad_norm": 0.699497640132904,
+      "learning_rate": 8.925816793771711e-05,
+      "loss": 0.1502,
+      "step": 5040
+    },
+    {
+      "grad_norm": 0.6201089024543762,
+      "learning_rate": 8.92069155367777e-05,
+      "loss": 0.1509,
+      "step": 5050
+    },
+    {
+      "grad_norm": 0.45711320638656616,
+      "learning_rate": 8.915555594562834e-05,
+      "loss": 0.1629,
+      "step": 5060
+    },
+    {
+      "grad_norm": 0.6208669543266296,
+      "learning_rate": 8.910408930468416e-05,
+      "loss": 0.1415,
+      "step": 5070
+    },
+    {
+      "grad_norm": 0.4821048974990845,
+      "learning_rate": 8.905251575465303e-05,
+      "loss": 0.1648,
+      "step": 5080
+    },
+    {
+      "grad_norm": 0.456248015165329,
+      "learning_rate": 8.900083543653502e-05,
+      "loss": 0.1563,
+      "step": 5090
+    },
+    {
+      "grad_norm": 0.7249851226806641,
+      "learning_rate": 8.894904849162218e-05,
+      "loss": 0.1722,
+      "step": 5100
+    },
+    {
+      "grad_norm": 0.6027569770812988,
+      "learning_rate": 8.889715506149802e-05,
+      "loss": 0.1521,
+      "step": 5110
+    },
+    {
+      "grad_norm": 0.6237264275550842,
+      "learning_rate": 8.884515528803722e-05,
+      "loss": 0.1543,
+      "step": 5120
+    },
+    {
+      "grad_norm": 0.38722893595695496,
+      "learning_rate": 8.879304931340517e-05,
+      "loss": 0.1557,
+      "step": 5130
+    },
+    {
+      "grad_norm": 0.4342791736125946,
+      "learning_rate": 8.874083728005759e-05,
+      "loss": 0.1604,
+      "step": 5140
+    },
+    {
+      "grad_norm": 0.4503703713417053,
+      "learning_rate": 8.868851933074021e-05,
+      "loss": 0.1558,
+      "step": 5150
+    },
+    {
+      "grad_norm": 0.5190554261207581,
+      "learning_rate": 8.863609560848829e-05,
+      "loss": 0.1432,
+      "step": 5160
+    },
+    {
+      "grad_norm": 0.669116735458374,
+      "learning_rate": 8.85835662566263e-05,
+      "loss": 0.1562,
+      "step": 5170
+    },
+    {
+      "grad_norm": 0.4965788722038269,
+      "learning_rate": 8.853093141876747e-05,
+      "loss": 0.1452,
+      "step": 5180
+    },
+    {
+      "grad_norm": 0.6538459062576294,
+      "learning_rate": 8.847819123881343e-05,
+      "loss": 0.1575,
+      "step": 5190
+    },
+    {
+      "grad_norm": 0.5277708172798157,
+      "learning_rate": 8.842534586095383e-05,
+      "loss": 0.154,
+      "step": 5200
+    },
+    {
+      "grad_norm": 0.4874209463596344,
+      "learning_rate": 8.837239542966593e-05,
+      "loss": 0.1515,
+      "step": 5210
+    },
+    {
+      "grad_norm": 0.5868343114852905,
+      "learning_rate": 8.831934008971417e-05,
+      "loss": 0.1566,
+      "step": 5220
+    },
+    {
+      "grad_norm": 0.5318579077720642,
+      "learning_rate": 8.826617998614982e-05,
+      "loss": 0.1845,
+      "step": 5230
+    },
+    {
+      "grad_norm": 0.47259417176246643,
+      "learning_rate": 8.821291526431056e-05,
+      "loss": 0.1494,
+      "step": 5240
+    },
+    {
+      "grad_norm": 0.4665931761264801,
+      "learning_rate": 8.815954606982015e-05,
+      "loss": 0.1674,
+      "step": 5250
+    },
+    {
+      "grad_norm": 0.6050563454627991,
+      "learning_rate": 8.810607254858789e-05,
+      "loss": 0.1712,
+      "step": 5260
+    },
+    {
+      "grad_norm": 0.5646398663520813,
+      "learning_rate": 8.805249484680838e-05,
+      "loss": 0.1647,
+      "step": 5270
+    },
+    {
+      "grad_norm": 0.5276556611061096,
+      "learning_rate": 8.799881311096096e-05,
+      "loss": 0.1624,
+      "step": 5280
+    },
+    {
+      "grad_norm": 0.3966107964515686,
+      "learning_rate": 8.794502748780949e-05,
+      "loss": 0.1627,
+      "step": 5290
+    },
+    {
+      "grad_norm": 0.5557110905647278,
+      "learning_rate": 8.78911381244018e-05,
+      "loss": 0.1617,
+      "step": 5300
+    },
+    {
+      "grad_norm": 0.4603351652622223,
+      "learning_rate": 8.783714516806933e-05,
+      "loss": 0.1534,
+      "step": 5310
+    },
+    {
+      "grad_norm": 0.5961066484451294,
+      "learning_rate": 8.77830487664268e-05,
+      "loss": 0.1486,
+      "step": 5320
+    },
+    {
+      "grad_norm": 0.444274365901947,
+      "learning_rate": 8.772884906737167e-05,
+      "loss": 0.1733,
+      "step": 5330
+    },
+    {
+      "grad_norm": 0.5912826061248779,
+      "learning_rate": 8.767454621908387e-05,
+      "loss": 0.1624,
+      "step": 5340
+    },
+    {
+      "grad_norm": 0.4391452968120575,
+      "learning_rate": 8.76201403700253e-05,
+      "loss": 0.1732,
+      "step": 5350
+    },
+    {
+      "grad_norm": 0.6658621430397034,
+      "learning_rate": 8.756563166893949e-05,
+      "loss": 0.1544,
+      "step": 5360
+    },
+    {
+      "grad_norm": 0.3312908411026001,
+      "learning_rate": 8.751102026485113e-05,
+      "loss": 0.1513,
+      "step": 5370
+    },
+    {
+      "grad_norm": 0.5003913044929504,
+      "learning_rate": 8.745630630706571e-05,
+      "loss": 0.1529,
+      "step": 5380
+    },
+    {
+      "grad_norm": 0.765087902545929,
+      "learning_rate": 8.740148994516912e-05,
+      "loss": 0.1517,
+      "step": 5390
+    },
+    {
+      "grad_norm": 0.829515814781189,
+      "learning_rate": 8.73465713290272e-05,
+      "loss": 0.1521,
+      "step": 5400
+    },
+    {
+      "grad_norm": 0.5427308082580566,
+      "learning_rate": 8.729155060878533e-05,
+      "loss": 0.1512,
+      "step": 5410
+    },
+    {
+      "grad_norm": 0.5107832551002502,
+      "learning_rate": 8.723642793486809e-05,
+      "loss": 0.1396,
+      "step": 5420
+    },
+    {
+      "grad_norm": 0.549967885017395,
+      "learning_rate": 8.718120345797873e-05,
+      "loss": 0.1477,
+      "step": 5430
+    },
+    {
+      "grad_norm": 0.4028843939304352,
+      "learning_rate": 8.712587732909889e-05,
+      "loss": 0.1437,
+      "step": 5440
+    },
+    {
+      "grad_norm": 0.5434556603431702,
+      "learning_rate": 8.707044969948806e-05,
+      "loss": 0.1441,
+      "step": 5450
+    },
+    {
+      "grad_norm": 0.6278228163719177,
+      "learning_rate": 8.701492072068329e-05,
+      "loss": 0.1271,
+      "step": 5460
+    },
+    {
+      "grad_norm": 0.5705100297927856,
+      "learning_rate": 8.695929054449869e-05,
+      "loss": 0.1445,
+      "step": 5470
+    },
+    {
+      "grad_norm": 0.5904441475868225,
+      "learning_rate": 8.690355932302501e-05,
+      "loss": 0.1567,
+      "step": 5480
+    },
+    {
+      "grad_norm": 0.6486173868179321,
+      "learning_rate": 8.684772720862931e-05,
+      "loss": 0.1422,
+      "step": 5490
+    },
+    {
+      "grad_norm": 0.5612136721611023,
+      "learning_rate": 8.679179435395446e-05,
+      "loss": 0.1416,
+      "step": 5500
+    },
+    {
+      "grad_norm": 0.5251556038856506,
+      "learning_rate": 8.673576091191874e-05,
+      "loss": 0.1408,
+      "step": 5510
+    },
+    {
+      "grad_norm": 0.5542381405830383,
+      "learning_rate": 8.667962703571541e-05,
+      "loss": 0.1537,
+      "step": 5520
+    },
+    {
+      "grad_norm": 0.6489263772964478,
+      "learning_rate": 8.662339287881238e-05,
+      "loss": 0.1624,
+      "step": 5530
+    },
+    {
+      "grad_norm": 0.6997092366218567,
+      "learning_rate": 8.656705859495169e-05,
+      "loss": 0.1826,
+      "step": 5540
+    },
+    {
+      "grad_norm": 0.5688534379005432,
+      "learning_rate": 8.651062433814912e-05,
+      "loss": 0.1373,
+      "step": 5550
+    },
+    {
+      "grad_norm": 0.7175577282905579,
+      "learning_rate": 8.645409026269375e-05,
+      "loss": 0.1575,
+      "step": 5560
+    },
+    {
+      "grad_norm": 0.570445716381073,
+      "learning_rate": 8.639745652314759e-05,
+      "loss": 0.1844,
+      "step": 5570
+    },
+    {
+      "grad_norm": 0.42648932337760925,
+      "learning_rate": 8.634072327434515e-05,
+      "loss": 0.1757,
+      "step": 5580
+    },
+    {
+      "grad_norm": 0.3774489164352417,
+      "learning_rate": 8.628389067139294e-05,
+      "loss": 0.1663,
+      "step": 5590
+    },
+    {
+      "grad_norm": 0.5803174376487732,
+      "learning_rate": 8.622695886966911e-05,
+      "loss": 0.1573,
+      "step": 5600
+    },
+    {
+      "grad_norm": 0.4728555679321289,
+      "learning_rate": 8.616992802482308e-05,
+      "loss": 0.1717,
+      "step": 5610
+    },
+    {
+      "grad_norm": 0.4463382065296173,
+      "learning_rate": 8.611279829277496e-05,
+      "loss": 0.1553,
+      "step": 5620
+    },
+    {
+      "grad_norm": 0.473662406206131,
+      "learning_rate": 8.605556982971528e-05,
+      "loss": 0.1776,
+      "step": 5630
+    },
+    {
+      "grad_norm": 0.46911701560020447,
+      "learning_rate": 8.599824279210447e-05,
+      "loss": 0.1414,
+      "step": 5640
+    },
+    {
+      "grad_norm": 0.5945715308189392,
+      "learning_rate": 8.594081733667243e-05,
+      "loss": 0.161,
+      "step": 5650
+    },
+    {
+      "grad_norm": 0.8718485832214355,
+      "learning_rate": 8.58832936204182e-05,
+      "loss": 0.1581,
+      "step": 5660
+    },
+    {
+      "grad_norm": 0.5202658772468567,
+      "learning_rate": 8.582567180060942e-05,
+      "loss": 0.1537,
+      "step": 5670
+    },
+    {
+      "grad_norm": 0.5831074714660645,
+      "learning_rate": 8.576795203478194e-05,
+      "loss": 0.1741,
+      "step": 5680
+    },
+    {
+      "grad_norm": 0.4111710786819458,
+      "learning_rate": 8.571013448073939e-05,
+      "loss": 0.1529,
+      "step": 5690
+    },
+    {
+      "grad_norm": 0.38849571347236633,
+      "learning_rate": 8.565221929655275e-05,
+      "loss": 0.141,
+      "step": 5700
+    },
+    {
+      "grad_norm": 0.5735211372375488,
+      "learning_rate": 8.559420664055992e-05,
+      "loss": 0.1486,
+      "step": 5710
+    },
+    {
+      "grad_norm": 0.5859473943710327,
+      "learning_rate": 8.553609667136532e-05,
+      "loss": 0.1545,
+      "step": 5720
+    },
+    {
+      "grad_norm": 0.6030755043029785,
+      "learning_rate": 8.547788954783936e-05,
+      "loss": 0.1488,
+      "step": 5730
+    },
+    {
+      "grad_norm": 0.5109389424324036,
+      "learning_rate": 8.541958542911808e-05,
+      "loss": 0.147,
+      "step": 5740
+    },
+    {
+      "grad_norm": 0.4166112542152405,
+      "learning_rate": 8.536118447460275e-05,
+      "loss": 0.1568,
+      "step": 5750
+    },
+    {
+      "grad_norm": 0.43787088990211487,
+      "learning_rate": 8.530268684395932e-05,
+      "loss": 0.1572,
+      "step": 5760
+    },
+    {
+      "grad_norm": 0.39980342984199524,
+      "learning_rate": 8.524409269711807e-05,
+      "loss": 0.1452,
+      "step": 5770
+    },
+    {
+      "grad_norm": 0.4353829324245453,
+      "learning_rate": 8.51854021942732e-05,
+      "loss": 0.1584,
+      "step": 5780
+    },
+    {
+      "grad_norm": 0.5610677003860474,
+      "learning_rate": 8.512661549588227e-05,
+      "loss": 0.1643,
+      "step": 5790
+    },
+    {
+      "grad_norm": 0.4289039373397827,
+      "learning_rate": 8.506773276266588e-05,
+      "loss": 0.1462,
+      "step": 5800
+    },
+    {
+      "grad_norm": 0.5160802006721497,
+      "learning_rate": 8.500875415560721e-05,
+      "loss": 0.1689,
+      "step": 5810
+    },
+    {
+      "grad_norm": 0.7192560434341431,
+      "learning_rate": 8.494967983595144e-05,
+      "loss": 0.1682,
+      "step": 5820
+    },
+    {
+      "grad_norm": 0.7457773089408875,
+      "learning_rate": 8.489050996520558e-05,
+      "loss": 0.182,
+      "step": 5830
+    },
+    {
+      "grad_norm": 0.5927083492279053,
+      "learning_rate": 8.483124470513775e-05,
+      "loss": 0.1529,
+      "step": 5840
+    },
+    {
+      "grad_norm": 0.5590975284576416,
+      "learning_rate": 8.477188421777692e-05,
+      "loss": 0.1466,
+      "step": 5850
+    },
+    {
+      "grad_norm": 0.4434778094291687,
+      "learning_rate": 8.47124286654124e-05,
+      "loss": 0.1411,
+      "step": 5860
+    },
+    {
+      "grad_norm": 0.44979992508888245,
+      "learning_rate": 8.465287821059341e-05,
+      "loss": 0.1508,
+      "step": 5870
+    },
+    {
+      "grad_norm": 0.4769991338253021,
+      "learning_rate": 8.45932330161286e-05,
+      "loss": 0.1579,
+      "step": 5880
+    },
+    {
+      "grad_norm": 0.668977677822113,
+      "learning_rate": 8.453349324508567e-05,
+      "loss": 0.1546,
+      "step": 5890
+    },
+    {
+      "grad_norm": 0.43769145011901855,
+      "learning_rate": 8.447365906079088e-05,
+      "loss": 0.1449,
+      "step": 5900
+    },
+    {
+      "grad_norm": 0.7127582430839539,
+      "learning_rate": 8.441373062682856e-05,
+      "loss": 0.1581,
+      "step": 5910
+    },
+    {
+      "grad_norm": 0.5985525250434875,
+      "learning_rate": 8.43537081070408e-05,
+      "loss": 0.1686,
+      "step": 5920
+    },
+    {
+      "grad_norm": 0.3619884252548218,
+      "learning_rate": 8.429359166552689e-05,
+      "loss": 0.1653,
+      "step": 5930
+    },
+    {
+      "grad_norm": 0.46753373742103577,
+      "learning_rate": 8.423338146664284e-05,
+      "loss": 0.1437,
+      "step": 5940
+    },
+    {
+      "grad_norm": 0.5531954765319824,
+      "learning_rate": 8.417307767500107e-05,
+      "loss": 0.1548,
+      "step": 5950
+    },
+    {
+      "grad_norm": 0.41177940368652344,
+      "learning_rate": 8.411268045546983e-05,
+      "loss": 0.1371,
+      "step": 5960
+    },
+    {
+      "grad_norm": 0.518706202507019,
+      "learning_rate": 8.405218997317281e-05,
+      "loss": 0.1592,
+      "step": 5970
+    },
+    {
+      "grad_norm": 0.40269067883491516,
+      "learning_rate": 8.399160639348869e-05,
+      "loss": 0.1505,
+      "step": 5980
+    },
+    {
+      "grad_norm": 0.4423786997795105,
+      "learning_rate": 8.393092988205065e-05,
+      "loss": 0.1587,
+      "step": 5990
+    },
+    {
+      "grad_norm": 0.48152950406074524,
+      "learning_rate": 8.387016060474597e-05,
+      "loss": 0.1508,
+      "step": 6000
+    },
+    {
+      "grad_norm": 0.8399495482444763,
+      "learning_rate": 8.380929872771551e-05,
+      "loss": 0.1713,
+      "step": 6010
+    },
+    {
+      "grad_norm": 0.3561352491378784,
+      "learning_rate": 8.374834441735335e-05,
+      "loss": 0.148,
+      "step": 6020
+    },
+    {
+      "grad_norm": 0.46161335706710815,
+      "learning_rate": 8.368729784030622e-05,
+      "loss": 0.1405,
+      "step": 6030
+    },
+    {
+      "grad_norm": 0.3662930130958557,
+      "learning_rate": 8.362615916347315e-05,
+      "loss": 0.1565,
+      "step": 6040
+    },
+    {
+      "grad_norm": 0.5010324716567993,
+      "learning_rate": 8.356492855400493e-05,
+      "loss": 0.1485,
+      "step": 6050
+    },
+    {
+      "grad_norm": 0.7411253452301025,
+      "learning_rate": 8.350360617930371e-05,
+      "loss": 0.1489,
+      "step": 6060
+    },
+    {
+      "grad_norm": 0.46456313133239746,
+      "learning_rate": 8.344219220702255e-05,
+      "loss": 0.1662,
+      "step": 6070
+    },
+    {
+      "grad_norm": 0.46667420864105225,
+      "learning_rate": 8.338068680506485e-05,
+      "loss": 0.1396,
+      "step": 6080
+    },
+    {
+      "grad_norm": 0.5470563173294067,
+      "learning_rate": 8.33190901415841e-05,
+      "loss": 0.1525,
+      "step": 6090
+    },
+    {
+      "grad_norm": 0.3678164482116699,
+      "learning_rate": 8.325740238498317e-05,
+      "loss": 0.1556,
+      "step": 6100
+    },
+    {
+      "grad_norm": 0.41196396946907043,
+      "learning_rate": 8.319562370391406e-05,
+      "loss": 0.1602,
+      "step": 6110
+    },
+    {
+      "grad_norm": 0.5237336754798889,
+      "learning_rate": 8.31337542672773e-05,
+      "loss": 0.1638,
+      "step": 6120
+    },
+    {
+      "grad_norm": 0.44463083148002625,
+      "learning_rate": 8.307179424422158e-05,
+      "loss": 0.1737,
+      "step": 6130
+    },
+    {
+      "grad_norm": 0.4130384922027588,
+      "learning_rate": 8.300974380414327e-05,
+      "loss": 0.1536,
+      "step": 6140
+    },
+    {
+      "grad_norm": 0.3876592814922333,
+      "learning_rate": 8.294760311668586e-05,
+      "loss": 0.1566,
+      "step": 6150
+    },
+    {
+      "grad_norm": 0.4410976469516754,
+      "learning_rate": 8.288537235173961e-05,
+      "loss": 0.143,
+      "step": 6160
+    },
+    {
+      "grad_norm": 0.49194395542144775,
+      "learning_rate": 8.282305167944108e-05,
+      "loss": 0.1793,
+      "step": 6170
+    },
+    {
+      "grad_norm": 0.47688379883766174,
+      "learning_rate": 8.276064127017262e-05,
+      "loss": 0.1507,
+      "step": 6180
+    },
+    {
+      "grad_norm": 0.37545105814933777,
+      "learning_rate": 8.269814129456189e-05,
+      "loss": 0.1406,
+      "step": 6190
+    },
+    {
+      "grad_norm": 0.4378091096878052,
+      "learning_rate": 8.263555192348143e-05,
+      "loss": 0.1555,
+      "step": 6200
+    },
+    {
+      "grad_norm": 0.367202490568161,
+      "learning_rate": 8.257287332804819e-05,
+      "loss": 0.164,
+      "step": 6210
+    },
+    {
+      "grad_norm": 0.4738454818725586,
+      "learning_rate": 8.251010567962307e-05,
+      "loss": 0.1595,
+      "step": 6220
+    },
+    {
+      "grad_norm": 0.7073342800140381,
+      "learning_rate": 8.244724914981041e-05,
+      "loss": 0.1556,
+      "step": 6230
+    },
+    {
+      "grad_norm": 0.48685553669929504,
+      "learning_rate": 8.238430391045757e-05,
+      "loss": 0.1778,
+      "step": 6240
+    },
+    {
+      "grad_norm": 0.5090844035148621,
+      "learning_rate": 8.232127013365445e-05,
+      "loss": 0.1943,
+      "step": 6250
+    },
+    {
+      "grad_norm": 0.38908103108406067,
+      "learning_rate": 8.225814799173295e-05,
+      "loss": 0.1666,
+      "step": 6260
+    },
+    {
+      "grad_norm": 0.4548185467720032,
+      "learning_rate": 8.219493765726663e-05,
+      "loss": 0.141,
+      "step": 6270
+    },
+    {
+      "grad_norm": 0.3863275945186615,
+      "learning_rate": 8.21316393030701e-05,
+      "loss": 0.1606,
+      "step": 6280
+    },
+    {
+      "grad_norm": 0.6323210597038269,
+      "learning_rate": 8.206825310219865e-05,
+      "loss": 0.1618,
+      "step": 6290
+    },
+    {
+      "grad_norm": 0.6593117713928223,
+      "learning_rate": 8.200477922794776e-05,
+      "loss": 0.152,
+      "step": 6300
+    },
+    {
+      "grad_norm": 0.4435170590877533,
+      "learning_rate": 8.194121785385256e-05,
+      "loss": 0.1481,
+      "step": 6310
+    },
+    {
+      "grad_norm": 0.31953907012939453,
+      "learning_rate": 8.187756915368741e-05,
+      "loss": 0.1515,
+      "step": 6320
+    },
+    {
+      "grad_norm": 0.4692417085170746,
+      "learning_rate": 8.181383330146544e-05,
+      "loss": 0.1613,
+      "step": 6330
+    },
+    {
+      "grad_norm": 0.83287113904953,
+      "learning_rate": 8.175001047143804e-05,
+      "loss": 0.1661,
+      "step": 6340
+    },
+    {
+      "grad_norm": 0.40431779623031616,
+      "learning_rate": 8.168610083809438e-05,
+      "loss": 0.1584,
+      "step": 6350
+    },
+    {
+      "grad_norm": 0.5592981576919556,
+      "learning_rate": 8.162210457616095e-05,
+      "loss": 0.1589,
+      "step": 6360
+    },
+    {
+      "grad_norm": 0.43560945987701416,
+      "learning_rate": 8.155802186060109e-05,
+      "loss": 0.1574,
+      "step": 6370
+    },
+    {
+      "grad_norm": 0.4271409213542938,
+      "learning_rate": 8.149385286661453e-05,
+      "loss": 0.1378,
+      "step": 6380
+    },
+    {
+      "grad_norm": 0.37510257959365845,
+      "learning_rate": 8.14295977696368e-05,
+      "loss": 0.1514,
+      "step": 6390
+    },
+    {
+      "grad_norm": 0.4203036427497864,
+      "learning_rate": 8.13652567453389e-05,
+      "loss": 0.1569,
+      "step": 6400
+    },
+    {
+      "grad_norm": 0.49708136916160583,
+      "learning_rate": 8.130082996962676e-05,
+      "loss": 0.1648,
+      "step": 6410
+    },
+    {
+      "grad_norm": 0.4301004707813263,
+      "learning_rate": 8.123631761864068e-05,
+      "loss": 0.1413,
+      "step": 6420
+    },
+    {
+      "grad_norm": 0.5896795988082886,
+      "learning_rate": 8.1171719868755e-05,
+      "loss": 0.1778,
+      "step": 6430
+    },
+    {
+      "grad_norm": 0.3781304359436035,
+      "learning_rate": 8.110703689657748e-05,
+      "loss": 0.156,
+      "step": 6440
+    },
+    {
+      "grad_norm": 0.46969467401504517,
+      "learning_rate": 8.104226887894892e-05,
+      "loss": 0.1569,
+      "step": 6450
+    },
+    {
+      "grad_norm": 0.25774848461151123,
+      "learning_rate": 8.097741599294257e-05,
+      "loss": 0.1378,
+      "step": 6460
+    },
+    {
+      "grad_norm": 0.45624351501464844,
+      "learning_rate": 8.091247841586378e-05,
+      "loss": 0.162,
+      "step": 6470
+    },
+    {
+      "grad_norm": 0.3723549246788025,
+      "learning_rate": 8.084745632524939e-05,
+      "loss": 0.1563,
+      "step": 6480
+    },
+    {
+      "grad_norm": 0.6368786692619324,
+      "learning_rate": 8.07823498988673e-05,
+      "loss": 0.1628,
+      "step": 6490
+    },
+    {
+      "grad_norm": 0.5796650052070618,
+      "learning_rate": 8.071715931471602e-05,
+      "loss": 0.1743,
+      "step": 6500
+    },
+    {
+      "grad_norm": 0.7003113627433777,
+      "learning_rate": 8.06518847510241e-05,
+      "loss": 0.1258,
+      "step": 6510
+    },
+    {
+      "grad_norm": 0.45986080169677734,
+      "learning_rate": 8.058652638624971e-05,
+      "loss": 0.1625,
+      "step": 6520
+    },
+    {
+      "grad_norm": 0.4257431924343109,
+      "learning_rate": 8.052108439908013e-05,
+      "loss": 0.1468,
+      "step": 6530
+    },
+    {
+      "grad_norm": 0.42671462893486023,
+      "learning_rate": 8.045555896843125e-05,
+      "loss": 0.1607,
+      "step": 6540
+    },
+    {
+      "grad_norm": 0.4996664226055145,
+      "learning_rate": 8.03899502734471e-05,
+      "loss": 0.1539,
+      "step": 6550
+    },
+    {
+      "grad_norm": 0.36821869015693665,
+      "learning_rate": 8.032425849349931e-05,
+      "loss": 0.1447,
+      "step": 6560
+    },
+    {
+      "grad_norm": 0.47465208172798157,
+      "learning_rate": 8.025848380818674e-05,
+      "loss": 0.1557,
+      "step": 6570
+    },
+    {
+      "grad_norm": 0.5191715359687805,
+      "learning_rate": 8.019262639733487e-05,
+      "loss": 0.161,
+      "step": 6580
+    },
+    {
+      "grad_norm": 0.6478346586227417,
+      "learning_rate": 8.012668644099531e-05,
+      "loss": 0.1731,
+      "step": 6590
+    },
+    {
+      "grad_norm": 0.3149877190589905,
+      "learning_rate": 8.006066411944542e-05,
+      "loss": 0.1455,
+      "step": 6600
+    },
+    {
+      "grad_norm": 0.5958502292633057,
+      "learning_rate": 7.999455961318769e-05,
+      "loss": 0.1595,
+      "step": 6610
+    },
+    {
+      "grad_norm": 0.43935564160346985,
+      "learning_rate": 7.992837310294932e-05,
+      "loss": 0.1425,
+      "step": 6620
+    },
+    {
+      "grad_norm": 0.581080973148346,
+      "learning_rate": 7.986210476968167e-05,
+      "loss": 0.1601,
+      "step": 6630
+    },
+    {
+      "grad_norm": 0.4066057503223419,
+      "learning_rate": 7.97957547945599e-05,
+      "loss": 0.1432,
+      "step": 6640
+    },
+    {
+      "grad_norm": 0.5545112490653992,
+      "learning_rate": 7.972932335898226e-05,
+      "loss": 0.1318,
+      "step": 6650
+    },
+    {
+      "grad_norm": 0.5526306629180908,
+      "learning_rate": 7.966281064456975e-05,
+      "loss": 0.1514,
+      "step": 6660
+    },
+    {
+      "grad_norm": 0.37425753474235535,
+      "learning_rate": 7.959621683316563e-05,
+      "loss": 0.1766,
+      "step": 6670
+    },
+    {
+      "grad_norm": 0.5164843797683716,
+      "learning_rate": 7.952954210683481e-05,
+      "loss": 0.1393,
+      "step": 6680
+    },
+    {
+      "grad_norm": 0.49167394638061523,
+      "learning_rate": 7.946278664786345e-05,
+      "loss": 0.1527,
+      "step": 6690
+    },
+    {
+      "grad_norm": 0.3772335350513458,
+      "learning_rate": 7.939595063875842e-05,
+      "loss": 0.1521,
+      "step": 6700
+    },
+    {
+      "grad_norm": 0.5985110998153687,
+      "learning_rate": 7.932903426224683e-05,
+      "loss": 0.1393,
+      "step": 6710
+    },
+    {
+      "grad_norm": 0.4745233952999115,
+      "learning_rate": 7.926203770127552e-05,
+      "loss": 0.1521,
+      "step": 6720
+    },
+    {
+      "grad_norm": 0.37146180868148804,
+      "learning_rate": 7.919496113901046e-05,
+      "loss": 0.1529,
+      "step": 6730
+    },
+    {
+      "grad_norm": 0.3370974361896515,
+      "learning_rate": 7.912780475883649e-05,
+      "loss": 0.1478,
+      "step": 6740
+    },
+    {
+      "grad_norm": 0.5201083421707153,
+      "learning_rate": 7.906056874435652e-05,
+      "loss": 0.1583,
+      "step": 6750
+    },
+    {
+      "grad_norm": 0.5765872001647949,
+      "learning_rate": 7.899325327939131e-05,
+      "loss": 0.1496,
+      "step": 6760
+    },
+    {
+      "grad_norm": 0.6143363118171692,
+      "learning_rate": 7.892585854797872e-05,
+      "loss": 0.1533,
+      "step": 6770
+    },
+    {
+      "grad_norm": 0.35201141238212585,
+      "learning_rate": 7.88583847343734e-05,
+      "loss": 0.1434,
+      "step": 6780
+    },
+    {
+      "grad_norm": 0.47996991872787476,
+      "learning_rate": 7.879083202304616e-05,
+      "loss": 0.164,
+      "step": 6790
+    },
+    {
+      "grad_norm": 0.3810211718082428,
+      "learning_rate": 7.872320059868355e-05,
+      "loss": 0.1419,
+      "step": 6800
+    },
+    {
+      "grad_norm": 0.5872716903686523,
+      "learning_rate": 7.865549064618729e-05,
+      "loss": 0.1539,
+      "step": 6810
+    },
+    {
+      "grad_norm": 0.5600290894508362,
+      "learning_rate": 7.858770235067381e-05,
+      "loss": 0.166,
+      "step": 6820
+    },
+    {
+      "grad_norm": 0.3578312397003174,
+      "learning_rate": 7.851983589747374e-05,
+      "loss": 0.1484,
+      "step": 6830
+    },
+    {
+      "grad_norm": 0.42482301592826843,
+      "learning_rate": 7.845189147213133e-05,
+      "loss": 0.1381,
+      "step": 6840
+    },
+    {
+      "grad_norm": 0.38345247507095337,
+      "learning_rate": 7.838386926040407e-05,
+      "loss": 0.1603,
+      "step": 6850
+    },
+    {
+      "grad_norm": 0.39441102743148804,
+      "learning_rate": 7.83157694482621e-05,
+      "loss": 0.1318,
+      "step": 6860
+    },
+    {
+      "grad_norm": 0.35376253724098206,
+      "learning_rate": 7.824759222188768e-05,
+      "loss": 0.1503,
+      "step": 6870
+    },
+    {
+      "grad_norm": 0.4473683834075928,
+      "learning_rate": 7.817933776767478e-05,
+      "loss": 0.1463,
+      "step": 6880
+    },
+    {
+      "grad_norm": 0.616813600063324,
+      "learning_rate": 7.811100627222842e-05,
+      "loss": 0.1569,
+      "step": 6890
+    },
+    {
+      "grad_norm": 0.46018558740615845,
+      "learning_rate": 7.804259792236435e-05,
+      "loss": 0.1562,
+      "step": 6900
+    },
+    {
+      "grad_norm": 0.36712005734443665,
+      "learning_rate": 7.797411290510835e-05,
+      "loss": 0.1473,
+      "step": 6910
+    },
+    {
+      "grad_norm": 0.6315376162528992,
+      "learning_rate": 7.790555140769586e-05,
+      "loss": 0.1469,
+      "step": 6920
+    },
+    {
+      "grad_norm": 0.362397700548172,
+      "learning_rate": 7.78369136175714e-05,
+      "loss": 0.1602,
+      "step": 6930
+    },
+    {
+      "grad_norm": 0.5630201101303101,
+      "learning_rate": 7.776819972238806e-05,
+      "loss": 0.163,
+      "step": 6940
+    },
+    {
+      "grad_norm": 0.3359009325504303,
+      "learning_rate": 7.7699409910007e-05,
+      "loss": 0.1518,
+      "step": 6950
+    },
+    {
+      "grad_norm": 0.4823044538497925,
+      "learning_rate": 7.763054436849694e-05,
+      "loss": 0.1357,
+      "step": 6960
+    },
+    {
+      "grad_norm": 0.5044270753860474,
+      "learning_rate": 7.756160328613364e-05,
+      "loss": 0.1427,
+      "step": 6970
+    },
+    {
+      "grad_norm": 0.5942718982696533,
+      "learning_rate": 7.749258685139942e-05,
+      "loss": 0.1517,
+      "step": 6980
+    },
+    {
+      "grad_norm": 0.5309360027313232,
+      "learning_rate": 7.742349525298253e-05,
+      "loss": 0.1879,
+      "step": 6990
+    },
+    {
+      "grad_norm": 0.43510329723358154,
+      "learning_rate": 7.735432867977679e-05,
+      "loss": 0.146,
+      "step": 7000
+    },
+    {
+      "grad_norm": 0.38771185278892517,
+      "learning_rate": 7.728508732088096e-05,
+      "loss": 0.1466,
+      "step": 7010
+    },
+    {
+      "grad_norm": 0.4187186360359192,
+      "learning_rate": 7.721577136559825e-05,
+      "loss": 0.1401,
+      "step": 7020
+    },
+    {
+      "grad_norm": 0.6778857707977295,
+      "learning_rate": 7.714638100343588e-05,
+      "loss": 0.1538,
+      "step": 7030
+    },
+    {
+      "grad_norm": 0.3753938674926758,
+      "learning_rate": 7.707691642410444e-05,
+      "loss": 0.1354,
+      "step": 7040
+    },
+    {
+      "grad_norm": 0.4636106491088867,
+      "learning_rate": 7.70073778175174e-05,
+      "loss": 0.1532,
+      "step": 7050
+    },
+    {
+      "grad_norm": 0.3907829821109772,
+      "learning_rate": 7.69377653737907e-05,
+      "loss": 0.1453,
+      "step": 7060
+    },
+    {
+      "grad_norm": 0.4850373864173889,
+      "learning_rate": 7.686807928324209e-05,
+      "loss": 0.1493,
+      "step": 7070
+    },
+    {
+      "grad_norm": 0.5411980152130127,
+      "learning_rate": 7.679831973639065e-05,
+      "loss": 0.2102,
+      "step": 7080
+    },
+    {
+      "grad_norm": 0.48953086137771606,
+      "learning_rate": 7.672848692395637e-05,
+      "loss": 0.1591,
+      "step": 7090
+    },
+    {
+      "grad_norm": 0.4727027118206024,
+      "learning_rate": 7.665858103685944e-05,
+      "loss": 0.1485,
+      "step": 7100
+    },
+    {
+      "grad_norm": 0.34940749406814575,
+      "learning_rate": 7.658860226621991e-05,
+      "loss": 0.1524,
+      "step": 7110
+    },
+    {
+      "grad_norm": 0.33032819628715515,
+      "learning_rate": 7.651855080335708e-05,
+      "loss": 0.1458,
+      "step": 7120
+    },
+    {
+      "grad_norm": 0.5604700446128845,
+      "learning_rate": 7.644842683978896e-05,
+      "loss": 0.1554,
+      "step": 7130
+    },
+    {
+      "grad_norm": 0.6467476487159729,
+      "learning_rate": 7.63782305672318e-05,
+      "loss": 0.1558,
+      "step": 7140
+    },
+    {
+      "grad_norm": 0.4750521779060364,
+      "learning_rate": 7.63079621775995e-05,
+      "loss": 0.1574,
+      "step": 7150
+    },
+    {
+      "grad_norm": 0.7652590870857239,
+      "learning_rate": 7.623762186300319e-05,
+      "loss": 0.1635,
+      "step": 7160
+    },
+    {
+      "grad_norm": 0.6367971897125244,
+      "learning_rate": 7.616720981575057e-05,
+      "loss": 0.1465,
+      "step": 7170
+    },
+    {
+      "grad_norm": 0.5346390008926392,
+      "learning_rate": 7.609672622834552e-05,
+      "loss": 0.1703,
+      "step": 7180
+    },
+    {
+      "grad_norm": 0.386788010597229,
+      "learning_rate": 7.602617129348747e-05,
+      "loss": 0.1425,
+      "step": 7190
+    },
+    {
+      "grad_norm": 0.4682758152484894,
+      "learning_rate": 7.595554520407088e-05,
+      "loss": 0.1498,
+      "step": 7200
+    },
+    {
+      "grad_norm": 0.46089550852775574,
+      "learning_rate": 7.588484815318484e-05,
+      "loss": 0.1275,
+      "step": 7210
+    },
+    {
+      "grad_norm": 0.4261797070503235,
+      "learning_rate": 7.581408033411234e-05,
+      "loss": 0.1557,
+      "step": 7220
+    },
+    {
+      "grad_norm": 0.2952896058559418,
+      "learning_rate": 7.574324194032995e-05,
+      "loss": 0.1236,
+      "step": 7230
+    },
+    {
+      "grad_norm": 0.4678933918476105,
+      "learning_rate": 7.567233316550705e-05,
+      "loss": 0.1745,
+      "step": 7240
+    },
+    {
+      "grad_norm": 0.5451762676239014,
+      "learning_rate": 7.560135420350562e-05,
+      "loss": 0.16,
+      "step": 7250
+    },
+    {
+      "grad_norm": 0.6013763546943665,
+      "learning_rate": 7.553030524837935e-05,
+      "loss": 0.1553,
+      "step": 7260
+    },
+    {
+      "grad_norm": 0.43584924936294556,
+      "learning_rate": 7.545918649437341e-05,
+      "loss": 0.152,
+      "step": 7270
+    },
+    {
+      "grad_norm": 0.6572586894035339,
+      "learning_rate": 7.538799813592377e-05,
+      "loss": 0.1527,
+      "step": 7280
+    },
+    {
+      "grad_norm": 0.49582788348197937,
+      "learning_rate": 7.531674036765662e-05,
+      "loss": 0.1405,
+      "step": 7290
+    },
+    {
+      "grad_norm": 0.5729706287384033,
+      "learning_rate": 7.524541338438807e-05,
+      "loss": 0.1579,
+      "step": 7300
+    },
+    {
+      "grad_norm": 0.48084717988967896,
+      "learning_rate": 7.517401738112328e-05,
+      "loss": 0.1478,
+      "step": 7310
+    },
+    {
+      "grad_norm": 0.3920782506465912,
+      "learning_rate": 7.510255255305628e-05,
+      "loss": 0.1465,
+      "step": 7320
+    },
+    {
+      "grad_norm": 0.42413967847824097,
+      "learning_rate": 7.503101909556911e-05,
+      "loss": 0.1599,
+      "step": 7330
+    },
+    {
+      "grad_norm": 0.6260297894477844,
+      "learning_rate": 7.495941720423154e-05,
+      "loss": 0.16,
+      "step": 7340
+    },
+    {
+      "grad_norm": 0.4718172550201416,
+      "learning_rate": 7.488774707480042e-05,
+      "loss": 0.153,
+      "step": 7350
+    },
+    {
+      "grad_norm": 0.5255489349365234,
+      "learning_rate": 7.481600890321911e-05,
+      "loss": 0.1603,
+      "step": 7360
+    },
+    {
+      "grad_norm": 0.5330467820167542,
+      "learning_rate": 7.474420288561708e-05,
+      "loss": 0.1461,
+      "step": 7370
+    },
+    {
+      "grad_norm": 0.430868536233902,
+      "learning_rate": 7.467232921830921e-05,
+      "loss": 0.1515,
+      "step": 7380
+    },
+    {
+      "grad_norm": 0.5155881643295288,
+      "learning_rate": 7.460038809779537e-05,
+      "loss": 0.1343,
+      "step": 7390
+    },
+    {
+      "grad_norm": 0.6267361640930176,
+      "learning_rate": 7.452837972075983e-05,
+      "loss": 0.1832,
+      "step": 7400
+    },
+    {
+      "grad_norm": 0.5033665299415588,
+      "learning_rate": 7.445630428407074e-05,
+      "loss": 0.1376,
+      "step": 7410
+    },
+    {
+      "grad_norm": 0.4798707365989685,
+      "learning_rate": 7.43841619847796e-05,
+      "loss": 0.1658,
+      "step": 7420
+    },
+    {
+      "grad_norm": 0.5167217254638672,
+      "learning_rate": 7.431195302012072e-05,
+      "loss": 0.1419,
+      "step": 7430
+    },
+    {
+      "grad_norm": 0.5399263501167297,
+      "learning_rate": 7.423967758751061e-05,
+      "loss": 0.1697,
+      "step": 7440
+    },
+    {
+      "grad_norm": 0.4054155647754669,
+      "learning_rate": 7.416733588454758e-05,
+      "loss": 0.1474,
+      "step": 7450
+    },
+    {
+      "grad_norm": 0.48530200123786926,
+      "learning_rate": 7.409492810901106e-05,
+      "loss": 0.1431,
+      "step": 7460
+    },
+    {
+      "grad_norm": 0.5296711325645447,
+      "learning_rate": 7.402245445886116e-05,
+      "loss": 0.1634,
+      "step": 7470
+    },
+    {
+      "grad_norm": 0.44249144196510315,
+      "learning_rate": 7.394991513223806e-05,
+      "loss": 0.1525,
+      "step": 7480
+    },
+    {
+      "grad_norm": 0.5170490741729736,
+      "learning_rate": 7.38773103274615e-05,
+      "loss": 0.1444,
+      "step": 7490
+    },
+    {
+      "grad_norm": 0.3813028037548065,
+      "learning_rate": 7.380464024303028e-05,
+      "loss": 0.1656,
+      "step": 7500
+    },
+    {
+      "grad_norm": 0.47864624857902527,
+      "learning_rate": 7.373190507762162e-05,
+      "loss": 0.1534,
+      "step": 7510
+    },
+    {
+      "grad_norm": 0.44420570135116577,
+      "learning_rate": 7.365910503009066e-05,
+      "loss": 0.1211,
+      "step": 7520
+    },
+    {
+      "grad_norm": 0.41292932629585266,
+      "learning_rate": 7.358624029946996e-05,
+      "loss": 0.1405,
+      "step": 7530
+    },
+    {
+      "grad_norm": 0.5182114243507385,
+      "learning_rate": 7.351331108496893e-05,
+      "loss": 0.1288,
+      "step": 7540
+    },
+    {
+      "grad_norm": 0.41996482014656067,
+      "learning_rate": 7.344031758597325e-05,
+      "loss": 0.1357,
+      "step": 7550
+    },
+    {
+      "grad_norm": 0.38613900542259216,
+      "learning_rate": 7.336726000204435e-05,
+      "loss": 0.1542,
+      "step": 7560
+    },
+    {
+      "grad_norm": 0.33762499690055847,
+      "learning_rate": 7.32941385329189e-05,
+      "loss": 0.1264,
+      "step": 7570
+    },
+    {
+      "grad_norm": 0.43300339579582214,
+      "learning_rate": 7.322095337850816e-05,
+      "loss": 0.1423,
+      "step": 7580
+    },
+    {
+      "grad_norm": 0.4532369375228882,
+      "learning_rate": 7.314770473889758e-05,
+      "loss": 0.1649,
+      "step": 7590
+    },
+    {
+      "grad_norm": 0.5500749349594116,
+      "learning_rate": 7.307439281434615e-05,
+      "loss": 0.1653,
+      "step": 7600
+    },
+    {
+      "grad_norm": 0.42430853843688965,
+      "learning_rate": 7.300101780528585e-05,
+      "loss": 0.1428,
+      "step": 7610
+    },
+    {
+      "grad_norm": 0.4829023480415344,
+      "learning_rate": 7.292757991232117e-05,
+      "loss": 0.1497,
+      "step": 7620
+    },
+    {
+      "grad_norm": 0.47607797384262085,
+      "learning_rate": 7.285407933622848e-05,
+      "loss": 0.1604,
+      "step": 7630
+    },
+    {
+      "grad_norm": 0.2768373489379883,
+      "learning_rate": 7.278051627795557e-05,
+      "loss": 0.1316,
+      "step": 7640
+    },
+    {
+      "grad_norm": 0.5142151117324829,
+      "learning_rate": 7.270689093862105e-05,
+      "loss": 0.1606,
+      "step": 7650
+    },
+    {
+      "grad_norm": 0.48016801476478577,
+      "learning_rate": 7.263320351951374e-05,
+      "loss": 0.1778,
+      "step": 7660
+    },
+    {
+      "grad_norm": 0.5532330274581909,
+      "learning_rate": 7.255945422209227e-05,
+      "loss": 0.1619,
+      "step": 7670
+    },
+    {
+      "grad_norm": 0.6146724224090576,
+      "learning_rate": 7.248564324798437e-05,
+      "loss": 0.1597,
+      "step": 7680
+    },
+    {
+      "grad_norm": 0.3743759095668793,
+      "learning_rate": 7.241177079898644e-05,
+      "loss": 0.1282,
+      "step": 7690
+    },
+    {
+      "grad_norm": 0.4911789894104004,
+      "learning_rate": 7.233783707706295e-05,
+      "loss": 0.1542,
+      "step": 7700
+    },
+    {
+      "grad_norm": 0.4771135747432709,
+      "learning_rate": 7.226384228434586e-05,
+      "loss": 0.1689,
+      "step": 7710
+    },
+    {
+      "grad_norm": 0.43502840399742126,
+      "learning_rate": 7.21897866231341e-05,
+      "loss": 0.1346,
+      "step": 7720
+    },
+    {
+      "grad_norm": 0.40363478660583496,
+      "learning_rate": 7.211567029589303e-05,
+      "loss": 0.1473,
+      "step": 7730
+    },
+    {
+      "grad_norm": 0.45180439949035645,
+      "learning_rate": 7.204149350525387e-05,
+      "loss": 0.1406,
+      "step": 7740
+    },
+    {
+      "grad_norm": 1.1957563161849976,
+      "learning_rate": 7.196725645401309e-05,
+      "loss": 0.1453,
+      "step": 7750
+    },
+    {
+      "grad_norm": 0.48904576897621155,
+      "learning_rate": 7.1892959345132e-05,
+      "loss": 0.1559,
+      "step": 7760
+    },
+    {
+      "grad_norm": 0.7125906348228455,
+      "learning_rate": 7.181860238173605e-05,
+      "loss": 0.1524,
+      "step": 7770
+    },
+    {
+      "grad_norm": 0.6729773879051208,
+      "learning_rate": 7.174418576711432e-05,
+      "loss": 0.1502,
+      "step": 7780
+    },
+    {
+      "grad_norm": 0.4506104290485382,
+      "learning_rate": 7.1669709704719e-05,
+      "loss": 0.1585,
+      "step": 7790
+    },
+    {
+      "grad_norm": 0.4825182855129242,
+      "learning_rate": 7.159517439816481e-05,
+      "loss": 0.1661,
+      "step": 7800
+    },
+    {
+      "grad_norm": 0.35956642031669617,
+      "learning_rate": 7.152058005122842e-05,
+      "loss": 0.138,
+      "step": 7810
+    },
+    {
+      "grad_norm": 0.436401903629303,
+      "learning_rate": 7.144592686784793e-05,
+      "loss": 0.164,
+      "step": 7820
+    },
+    {
+      "grad_norm": 0.421114981174469,
+      "learning_rate": 7.137121505212229e-05,
+      "loss": 0.1379,
+      "step": 7830
+    },
+    {
+      "grad_norm": 0.8368301391601562,
+      "learning_rate": 7.129644480831077e-05,
+      "loss": 0.1634,
+      "step": 7840
+    },
+    {
+      "grad_norm": 0.4869307279586792,
+      "learning_rate": 7.122161634083234e-05,
+      "loss": 0.1479,
+      "step": 7850
+    },
+    {
+      "grad_norm": 0.48510006070137024,
+      "learning_rate": 7.114672985426516e-05,
+      "loss": 0.1399,
+      "step": 7860
+    },
+    {
+      "grad_norm": 0.5700347423553467,
+      "learning_rate": 7.107178555334606e-05,
+      "loss": 0.1697,
+      "step": 7870
+    },
+    {
+      "grad_norm": 0.5160481333732605,
+      "learning_rate": 7.099678364296989e-05,
+      "loss": 0.1551,
+      "step": 7880
+    },
+    {
+      "grad_norm": 0.453804612159729,
+      "learning_rate": 7.0921724328189e-05,
+      "loss": 0.154,
+      "step": 7890
+    },
+    {
+      "grad_norm": 0.40567076206207275,
+      "learning_rate": 7.084660781421268e-05,
+      "loss": 0.1392,
+      "step": 7900
+    },
+    {
+      "grad_norm": 0.6939504146575928,
+      "learning_rate": 7.077143430640662e-05,
+      "loss": 0.158,
+      "step": 7910
+    },
+    {
+      "grad_norm": 0.4799591302871704,
+      "learning_rate": 7.069620401029232e-05,
+      "loss": 0.165,
+      "step": 7920
+    },
+    {
+      "grad_norm": 0.44246557354927063,
+      "learning_rate": 7.062091713154655e-05,
+      "loss": 0.1571,
+      "step": 7930
+    },
+    {
+      "grad_norm": 0.7225956916809082,
+      "learning_rate": 7.054557387600075e-05,
+      "loss": 0.1605,
+      "step": 7940
+    },
+    {
+      "grad_norm": 0.4085860252380371,
+      "learning_rate": 7.04701744496405e-05,
+      "loss": 0.1473,
+      "step": 7950
+    },
+    {
+      "grad_norm": 0.36924096941947937,
+      "learning_rate": 7.039471905860495e-05,
+      "loss": 0.1316,
+      "step": 7960
+    },
+    {
+      "grad_norm": 0.5347712635993958,
+      "learning_rate": 7.031920790918628e-05,
+      "loss": 0.1653,
+      "step": 7970
+    },
+    {
+      "grad_norm": 0.5151801109313965,
+      "learning_rate": 7.024364120782906e-05,
+      "loss": 0.1621,
+      "step": 7980
+    },
+    {
+      "grad_norm": 0.6688697934150696,
+      "learning_rate": 7.016801916112978e-05,
+      "loss": 0.1587,
+      "step": 7990
+    },
+    {
+      "grad_norm": 0.4618225693702698,
+      "learning_rate": 7.009234197583623e-05,
+      "loss": 0.157,
+      "step": 8000
+    },
+    {
+      "grad_norm": 0.4520286023616791,
+      "learning_rate": 7.001660985884692e-05,
+      "loss": 0.1474,
+      "step": 8010
+    },
+    {
+      "grad_norm": 0.46954602003097534,
+      "learning_rate": 6.994082301721063e-05,
+      "loss": 0.1673,
+      "step": 8020
+    },
+    {
+      "grad_norm": 0.5362978577613831,
+      "learning_rate": 6.986498165812563e-05,
+      "loss": 0.1525,
+      "step": 8030
+    },
+    {
+      "grad_norm": 0.6156564950942993,
+      "learning_rate": 6.978908598893932e-05,
+      "loss": 0.145,
+      "step": 8040
+    },
+    {
+      "grad_norm": 0.3331153988838196,
+      "learning_rate": 6.971313621714756e-05,
+      "loss": 0.1575,
+      "step": 8050
+    },
+    {
+      "grad_norm": 0.40469953417778015,
+      "learning_rate": 6.96371325503941e-05,
+      "loss": 0.1617,
+      "step": 8060
+    },
+    {
+      "grad_norm": 0.6529107093811035,
+      "learning_rate": 6.956107519647014e-05,
+      "loss": 0.168,
+      "step": 8070
+    },
+    {
+      "grad_norm": 0.2709386944770813,
+      "learning_rate": 6.94849643633135e-05,
+      "loss": 0.1502,
+      "step": 8080
+    },
+    {
+      "grad_norm": 0.3172459304332733,
+      "learning_rate": 6.940880025900834e-05,
+      "loss": 0.1694,
+      "step": 8090
+    },
+    {
+      "grad_norm": 0.5133095979690552,
+      "learning_rate": 6.933258309178438e-05,
+      "loss": 0.1592,
+      "step": 8100
+    },
+    {
+      "grad_norm": 0.4597356915473938,
+      "learning_rate": 6.925631307001646e-05,
+      "loss": 0.137,
+      "step": 8110
+    },
+    {
+      "grad_norm": 0.4502180218696594,
+      "learning_rate": 6.91799904022239e-05,
+      "loss": 0.1446,
+      "step": 8120
+    },
+    {
+      "grad_norm": 0.6499724984169006,
+      "learning_rate": 6.910361529706997e-05,
+      "loss": 0.1238,
+      "step": 8130
+    },
+    {
+      "grad_norm": 0.36773473024368286,
+      "learning_rate": 6.902718796336131e-05,
+      "loss": 0.1516,
+      "step": 8140
+    },
+    {
+      "grad_norm": 0.8023806810379028,
+      "learning_rate": 6.895070861004729e-05,
+      "loss": 0.1584,
+      "step": 8150
+    },
+    {
+      "grad_norm": 0.6120730042457581,
+      "learning_rate": 6.887417744621956e-05,
+      "loss": 0.164,
+      "step": 8160
+    },
+    {
+      "grad_norm": 0.7398198246955872,
+      "learning_rate": 6.87975946811114e-05,
+      "loss": 0.1472,
+      "step": 8170
+    },
+    {
+      "grad_norm": 0.4539482593536377,
+      "learning_rate": 6.872096052409718e-05,
+      "loss": 0.1453,
+      "step": 8180
+    },
+    {
+      "grad_norm": 0.42565491795539856,
+      "learning_rate": 6.864427518469174e-05,
+      "loss": 0.1472,
+      "step": 8190
+    },
+    {
+      "grad_norm": 0.6299214959144592,
+      "learning_rate": 6.856753887254986e-05,
+      "loss": 0.1658,
+      "step": 8200
+    },
+    {
+      "grad_norm": 0.5207350254058838,
+      "learning_rate": 6.849075179746572e-05,
+      "loss": 0.1716,
+      "step": 8210
+    },
+    {
+      "grad_norm": 0.6267195343971252,
+      "learning_rate": 6.841391416937221e-05,
+      "loss": 0.1692,
+      "step": 8220
+    },
+    {
+      "grad_norm": 0.6877052187919617,
+      "learning_rate": 6.833702619834053e-05,
+      "loss": 0.1354,
+      "step": 8230
+    },
+    {
+      "grad_norm": 0.5795918703079224,
+      "learning_rate": 6.82600880945794e-05,
+      "loss": 0.1707,
+      "step": 8240
+    },
+    {
+      "grad_norm": 0.47131434082984924,
+      "learning_rate": 6.818310006843468e-05,
+      "loss": 0.153,
+      "step": 8250
+    },
+    {
+      "grad_norm": 0.5225297808647156,
+      "learning_rate": 6.810606233038868e-05,
+      "loss": 0.1526,
+      "step": 8260
+    },
+    {
+      "grad_norm": 0.3766261339187622,
+      "learning_rate": 6.802897509105966e-05,
+      "loss": 0.1461,
+      "step": 8270
+    },
+    {
+      "grad_norm": 0.47061875462532043,
+      "learning_rate": 6.79518385612012e-05,
+      "loss": 0.1582,
+      "step": 8280
+    },
+    {
+      "grad_norm": 0.5938490033149719,
+      "learning_rate": 6.787465295170157e-05,
+      "loss": 0.1535,
+      "step": 8290
+    },
+    {
+      "grad_norm": 0.3773987293243408,
+      "learning_rate": 6.779741847358332e-05,
+      "loss": 0.1726,
+      "step": 8300
+    },
+    {
+      "grad_norm": 0.5251126885414124,
+      "learning_rate": 6.772013533800256e-05,
+      "loss": 0.1464,
+      "step": 8310
+    },
+    {
+      "grad_norm": 0.47159793972969055,
+      "learning_rate": 6.764280375624843e-05,
+      "loss": 0.1435,
+      "step": 8320
+    },
+    {
+      "grad_norm": 0.5718153119087219,
+      "learning_rate": 6.756542393974252e-05,
+      "loss": 0.1483,
+      "step": 8330
+    },
+    {
+      "grad_norm": 0.5215834379196167,
+      "learning_rate": 6.748799610003828e-05,
+      "loss": 0.1714,
+      "step": 8340
+    },
+    {
+      "grad_norm": 0.41711899638175964,
+      "learning_rate": 6.741052044882048e-05,
+      "loss": 0.1532,
+      "step": 8350
+    },
+    {
+      "grad_norm": 0.4833110272884369,
+      "learning_rate": 6.73329971979046e-05,
+      "loss": 0.177,
+      "step": 8360
+    },
+    {
+      "grad_norm": 0.5375620126724243,
+      "learning_rate": 6.725542655923625e-05,
+      "loss": 0.1419,
+      "step": 8370
+    },
+    {
+      "grad_norm": 0.4572528302669525,
+      "learning_rate": 6.717780874489057e-05,
+      "loss": 0.139,
+      "step": 8380
+    },
+    {
+      "grad_norm": 0.5163964033126831,
+      "learning_rate": 6.710014396707172e-05,
+      "loss": 0.1533,
+      "step": 8390
+    },
+    {
+      "grad_norm": 0.4021039307117462,
+      "learning_rate": 6.702243243811221e-05,
+      "loss": 0.1526,
+      "step": 8400
+    },
+    {
+      "grad_norm": 0.6673436760902405,
+      "learning_rate": 6.694467437047244e-05,
+      "loss": 0.1531,
+      "step": 8410
+    },
+    {
+      "grad_norm": 0.376609206199646,
+      "learning_rate": 6.686686997673997e-05,
+      "loss": 0.1613,
+      "step": 8420
+    },
+    {
+      "grad_norm": 0.43886974453926086,
+      "learning_rate": 6.678901946962903e-05,
+      "loss": 0.1188,
+      "step": 8430
+    },
+    {
+      "grad_norm": 0.7995914816856384,
+      "learning_rate": 6.671112306197996e-05,
+      "loss": 0.1694,
+      "step": 8440
+    },
+    {
+      "grad_norm": 0.5564230680465698,
+      "learning_rate": 6.663318096675854e-05,
+      "loss": 0.1459,
+      "step": 8450
+    },
+    {
+      "grad_norm": 0.5078952312469482,
+      "learning_rate": 6.655519339705552e-05,
+      "loss": 0.1586,
+      "step": 8460
+    },
+    {
+      "grad_norm": 0.4987128674983978,
+      "learning_rate": 6.647716056608588e-05,
+      "loss": 0.1547,
+      "step": 8470
+    },
+    {
+      "grad_norm": 0.695870578289032,
+      "learning_rate": 6.639908268718843e-05,
+      "loss": 0.1755,
+      "step": 8480
+    },
+    {
+      "grad_norm": 0.44118109345436096,
+      "learning_rate": 6.632095997382514e-05,
+      "loss": 0.151,
+      "step": 8490
+    },
+    {
+      "grad_norm": 0.4051033556461334,
+      "learning_rate": 6.624279263958047e-05,
+      "loss": 0.1619,
+      "step": 8500
+    },
+    {
+      "grad_norm": 0.4849059581756592,
+      "learning_rate": 6.616458089816097e-05,
+      "loss": 0.1486,
+      "step": 8510
+    },
+    {
+      "grad_norm": 0.4353429973125458,
+      "learning_rate": 6.608632496339454e-05,
+      "loss": 0.1476,
+      "step": 8520
+    },
+    {
+      "grad_norm": 0.6226187944412231,
+      "learning_rate": 6.600802504922988e-05,
+      "loss": 0.1373,
+      "step": 8530
+    },
+    {
+      "grad_norm": 0.37061500549316406,
+      "learning_rate": 6.592968136973604e-05,
+      "loss": 0.134,
+      "step": 8540
+    },
+    {
+      "grad_norm": 0.5091410875320435,
+      "learning_rate": 6.585129413910159e-05,
+      "loss": 0.151,
+      "step": 8550
+    },
+    {
+      "grad_norm": 0.4716707170009613,
+      "learning_rate": 6.577286357163424e-05,
+      "loss": 0.1384,
+      "step": 8560
+    },
+    {
+      "grad_norm": 0.5448651313781738,
+      "learning_rate": 6.569438988176018e-05,
+      "loss": 0.153,
+      "step": 8570
+    },
+    {
+      "grad_norm": 0.6176994442939758,
+      "learning_rate": 6.561587328402347e-05,
+      "loss": 0.1435,
+      "step": 8580
+    },
+    {
+      "grad_norm": 0.6173880100250244,
+      "learning_rate": 6.553731399308549e-05,
+      "loss": 0.1386,
+      "step": 8590
+    },
+    {
+      "grad_norm": 0.5722159743309021,
+      "learning_rate": 6.545871222372436e-05,
+      "loss": 0.162,
+      "step": 8600
+    },
+    {
+      "grad_norm": 0.45739296078681946,
+      "learning_rate": 6.538006819083426e-05,
+      "loss": 0.1515,
+      "step": 8610
+    },
+    {
+      "grad_norm": 0.5461430549621582,
+      "learning_rate": 6.530138210942505e-05,
+      "loss": 0.151,
+      "step": 8620
+    },
+    {
+      "grad_norm": 0.5969565510749817,
+      "learning_rate": 6.522265419462141e-05,
+      "loss": 0.1395,
+      "step": 8630
+    },
+    {
+      "grad_norm": 0.69148188829422,
+      "learning_rate": 6.514388466166248e-05,
+      "loss": 0.1521,
+      "step": 8640
+    },
+    {
+      "grad_norm": 0.49734556674957275,
+      "learning_rate": 6.506507372590119e-05,
+      "loss": 0.1198,
+      "step": 8650
+    },
+    {
+      "grad_norm": 0.6039820313453674,
+      "learning_rate": 6.498622160280355e-05,
+      "loss": 0.1425,
+      "step": 8660
+    },
+    {
+      "grad_norm": 0.5308656096458435,
+      "learning_rate": 6.490732850794832e-05,
+      "loss": 0.1737,
+      "step": 8670
+    },
+    {
+      "grad_norm": 0.47831395268440247,
+      "learning_rate": 6.482839465702616e-05,
+      "loss": 0.1407,
+      "step": 8680
+    },
+    {
+      "grad_norm": 0.5315905213356018,
+      "learning_rate": 6.474942026583923e-05,
+      "loss": 0.12,
+      "step": 8690
+    },
+    {
+      "grad_norm": 0.35923418402671814,
+      "learning_rate": 6.467040555030052e-05,
+      "loss": 0.1435,
+      "step": 8700
+    },
+    {
+      "grad_norm": 0.5326019525527954,
+      "learning_rate": 6.459135072643321e-05,
+      "loss": 0.137,
+      "step": 8710
+    },
+    {
+      "grad_norm": 0.5948243737220764,
+      "learning_rate": 6.451225601037019e-05,
+      "loss": 0.1428,
+      "step": 8720
+    },
+    {
+      "grad_norm": 0.5500339269638062,
+      "learning_rate": 6.443312161835338e-05,
+      "loss": 0.16,
+      "step": 8730
+    },
+    {
+      "grad_norm": 0.469795286655426,
+      "learning_rate": 6.43539477667332e-05,
+      "loss": 0.1429,
+      "step": 8740
+    },
+    {
+      "grad_norm": 0.35395967960357666,
+      "learning_rate": 6.427473467196793e-05,
+      "loss": 0.1429,
+      "step": 8750
+    },
+    {
+      "grad_norm": 0.4958198070526123,
+      "learning_rate": 6.419548255062315e-05,
+      "loss": 0.133,
+      "step": 8760
+    },
+    {
+      "grad_norm": 0.6807100176811218,
+      "learning_rate": 6.411619161937112e-05,
+      "loss": 0.1716,
+      "step": 8770
+    },
+    {
+      "grad_norm": 0.4612419307231903,
+      "learning_rate": 6.403686209499022e-05,
+      "loss": 0.1627,
+      "step": 8780
+    },
+    {
+      "grad_norm": 0.5603896379470825,
+      "learning_rate": 6.395749419436437e-05,
+      "loss": 0.1451,
+      "step": 8790
+    },
+    {
+      "grad_norm": 0.5466400980949402,
+      "learning_rate": 6.387808813448234e-05,
+      "loss": 0.1547,
+      "step": 8800
+    },
+    {
+      "grad_norm": 0.6266804337501526,
+      "learning_rate": 6.37986441324373e-05,
+      "loss": 0.1409,
+      "step": 8810
+    },
+    {
+      "grad_norm": 0.4320460557937622,
+      "learning_rate": 6.37191624054261e-05,
+      "loss": 0.1509,
+      "step": 8820
+    },
+    {
+      "grad_norm": 0.46681129932403564,
+      "learning_rate": 6.363964317074872e-05,
+      "loss": 0.1394,
+      "step": 8830
+    },
+    {
+      "grad_norm": 0.505926787853241,
+      "learning_rate": 6.356008664580776e-05,
+      "loss": 0.1557,
+      "step": 8840
+    },
+    {
+      "grad_norm": 0.33630436658859253,
+      "learning_rate": 6.348049304810771e-05,
+      "loss": 0.1219,
+      "step": 8850
+    },
+    {
+      "grad_norm": 0.510668933391571,
+      "learning_rate": 6.340086259525442e-05,
+      "loss": 0.1461,
+      "step": 8860
+    },
+    {
+      "grad_norm": 0.5635788440704346,
+      "learning_rate": 6.332119550495448e-05,
+      "loss": 0.1274,
+      "step": 8870
+    },
+    {
+      "grad_norm": 0.4122416377067566,
+      "learning_rate": 6.324149199501473e-05,
+      "loss": 0.153,
+      "step": 8880
+    },
+    {
+      "grad_norm": 0.5904040336608887,
+      "learning_rate": 6.316175228334146e-05,
+      "loss": 0.1487,
+      "step": 8890
+    },
+    {
+      "grad_norm": 0.6465409398078918,
+      "learning_rate": 6.308197658794003e-05,
+      "loss": 0.1421,
+      "step": 8900
+    },
+    {
+      "grad_norm": 0.6611448526382446,
+      "learning_rate": 6.300216512691417e-05,
+      "loss": 0.1486,
+      "step": 8910
+    },
+    {
+      "grad_norm": 0.5517676472663879,
+      "learning_rate": 6.292231811846532e-05,
+      "loss": 0.1233,
+      "step": 8920
+    },
+    {
+      "grad_norm": 0.7486810088157654,
+      "learning_rate": 6.284243578089217e-05,
+      "loss": 0.1417,
+      "step": 8930
+    },
+    {
+      "grad_norm": 0.5255800485610962,
+      "learning_rate": 6.276251833258999e-05,
+      "loss": 0.1593,
+      "step": 8940
+    },
+    {
+      "grad_norm": 0.46037963032722473,
+      "learning_rate": 6.268256599205003e-05,
+      "loss": 0.1422,
+      "step": 8950
+    },
+    {
+      "grad_norm": 0.5746683478355408,
+      "learning_rate": 6.260257897785892e-05,
+      "loss": 0.1363,
+      "step": 8960
+    },
+    {
+      "grad_norm": 0.444522887468338,
+      "learning_rate": 6.252255750869811e-05,
+      "loss": 0.1387,
+      "step": 8970
+    },
+    {
+      "grad_norm": 0.6860119104385376,
+      "learning_rate": 6.244250180334325e-05,
+      "loss": 0.1535,
+      "step": 8980
+    },
+    {
+      "grad_norm": 0.5499054789543152,
+      "learning_rate": 6.236241208066356e-05,
+      "loss": 0.1503,
+      "step": 8990
+    },
+    {
+      "grad_norm": 0.5161921381950378,
+      "learning_rate": 6.228228855962133e-05,
+      "loss": 0.1526,
+      "step": 9000
+    },
+    {
+      "grad_norm": 0.3715263903141022,
+      "learning_rate": 6.220213145927115e-05,
+      "loss": 0.1389,
+      "step": 9010
+    },
+    {
+      "grad_norm": 0.44927752017974854,
+      "learning_rate": 6.212194099875951e-05,
+      "loss": 0.12,
+      "step": 9020
+    },
+    {
+      "grad_norm": 0.5410082936286926,
+      "learning_rate": 6.204171739732405e-05,
+      "loss": 0.1593,
+      "step": 9030
+    },
+    {
+      "grad_norm": 0.4667140543460846,
+      "learning_rate": 6.196146087429303e-05,
+      "loss": 0.1159,
+      "step": 9040
+    },
+    {
+      "grad_norm": 0.6709445714950562,
+      "learning_rate": 6.188117164908474e-05,
+      "loss": 0.1412,
+      "step": 9050
+    },
+    {
+      "grad_norm": 0.6438217163085938,
+      "learning_rate": 6.180084994120684e-05,
+      "loss": 0.1325,
+      "step": 9060
+    },
+    {
+      "grad_norm": 0.4224262833595276,
+      "learning_rate": 6.17204959702558e-05,
+      "loss": 0.1529,
+      "step": 9070
+    },
+    {
+      "grad_norm": 0.45982444286346436,
+      "learning_rate": 6.164010995591635e-05,
+      "loss": 0.1507,
+      "step": 9080
+    },
+    {
+      "grad_norm": 0.5061719417572021,
+      "learning_rate": 6.155969211796076e-05,
+      "loss": 0.1453,
+      "step": 9090
+    },
+    {
+      "grad_norm": 0.475137323141098,
+      "learning_rate": 6.147924267624829e-05,
+      "loss": 0.1468,
+      "step": 9100
+    },
+    {
+      "grad_norm": 0.9040607810020447,
+      "learning_rate": 6.13987618507247e-05,
+      "loss": 0.1401,
+      "step": 9110
+    },
+    {
+      "grad_norm": 0.5963755249977112,
+      "learning_rate": 6.131824986142147e-05,
+      "loss": 0.1409,
+      "step": 9120
+    },
+    {
+      "grad_norm": 0.5166276097297668,
+      "learning_rate": 6.123770692845529e-05,
+      "loss": 0.1501,
+      "step": 9130
+    },
+    {
+      "grad_norm": 0.6420073509216309,
+      "learning_rate": 6.11571332720275e-05,
+      "loss": 0.1289,
+      "step": 9140
+    },
+    {
+      "grad_norm": 0.7091841101646423,
+      "learning_rate": 6.107652911242336e-05,
+      "loss": 0.1717,
+      "step": 9150
+    },
+    {
+      "grad_norm": 0.588521420955658,
+      "learning_rate": 6.0995894670011586e-05,
+      "loss": 0.1452,
+      "step": 9160
+    },
+    {
+      "grad_norm": 0.4155677855014801,
+      "learning_rate": 6.091523016524368e-05,
+      "loss": 0.1548,
+      "step": 9170
+    },
+    {
+      "grad_norm": 0.5204802751541138,
+      "learning_rate": 6.083453581865328e-05,
+      "loss": 0.1464,
+      "step": 9180
+    },
+    {
+      "grad_norm": 0.4198887050151825,
+      "learning_rate": 6.075381185085568e-05,
+      "loss": 0.1487,
+      "step": 9190
+    },
+    {
+      "grad_norm": 0.6765252947807312,
+      "learning_rate": 6.067305848254709e-05,
+      "loss": 0.1518,
+      "step": 9200
+    },
+    {
+      "grad_norm": 0.49510884284973145,
+      "learning_rate": 6.059227593450418e-05,
+      "loss": 0.137,
+      "step": 9210
+    },
+    {
+      "grad_norm": 0.5212628841400146,
+      "learning_rate": 6.051146442758333e-05,
+      "loss": 0.1433,
+      "step": 9220
+    },
+    {
+      "grad_norm": 0.42623719573020935,
+      "learning_rate": 6.043062418272012e-05,
+      "loss": 0.1449,
+      "step": 9230
+    },
+    {
+      "grad_norm": 0.6151301860809326,
+      "learning_rate": 6.0349755420928666e-05,
+      "loss": 0.1343,
+      "step": 9240
+    },
+    {
+      "grad_norm": 0.4641484022140503,
+      "learning_rate": 6.0268858363301105e-05,
+      "loss": 0.1696,
+      "step": 9250
+    },
+    {
+      "grad_norm": 0.45424482226371765,
+      "learning_rate": 6.018793323100689e-05,
+      "loss": 0.1561,
+      "step": 9260
+    },
+    {
+      "grad_norm": 0.5596998333930969,
+      "learning_rate": 6.0106980245292255e-05,
+      "loss": 0.145,
+      "step": 9270
+    },
+    {
+      "grad_norm": 0.640203595161438,
+      "learning_rate": 6.002599962747957e-05,
+      "loss": 0.1579,
+      "step": 9280
+    },
+    {
+      "grad_norm": 0.3602248430252075,
+      "learning_rate": 5.994499159896673e-05,
+      "loss": 0.12,
+      "step": 9290
+    },
+    {
+      "grad_norm": 0.44463831186294556,
+      "learning_rate": 5.9863956381226607e-05,
+      "loss": 0.1485,
+      "step": 9300
+    },
+    {
+      "grad_norm": 0.47846466302871704,
+      "learning_rate": 5.9782894195806394e-05,
+      "loss": 0.1539,
+      "step": 9310
+    },
+    {
+      "grad_norm": 0.34202128648757935,
+      "learning_rate": 5.9701805264327004e-05,
+      "loss": 0.1356,
+      "step": 9320
+    },
+    {
+      "grad_norm": 0.4801376461982727,
+      "learning_rate": 5.96206898084825e-05,
+      "loss": 0.1484,
+      "step": 9330
+    },
+    {
+      "grad_norm": 0.3704809248447418,
+      "learning_rate": 5.953954805003942e-05,
+      "loss": 0.1399,
+      "step": 9340
+    },
+    {
+      "grad_norm": 0.5061681866645813,
+      "learning_rate": 5.945838021083623e-05,
+      "loss": 0.1466,
+      "step": 9350
+    },
+    {
+      "grad_norm": 0.40298745036125183,
+      "learning_rate": 5.9377186512782714e-05,
+      "loss": 0.1293,
+      "step": 9360
+    },
+    {
+      "grad_norm": 0.33222824335098267,
+      "learning_rate": 5.929596717785935e-05,
+      "loss": 0.1508,
+      "step": 9370
+    },
+    {
+      "grad_norm": 0.34811481833457947,
+      "learning_rate": 5.921472242811668e-05,
+      "loss": 0.1562,
+      "step": 9380
+    },
+    {
+      "grad_norm": 0.5632371306419373,
+      "learning_rate": 5.913345248567475e-05,
+      "loss": 0.1514,
+      "step": 9390
+    },
+    {
+      "grad_norm": 0.48682403564453125,
+      "learning_rate": 5.905215757272248e-05,
+      "loss": 0.1396,
+      "step": 9400
+    },
+    {
+      "grad_norm": 0.4803490936756134,
+      "learning_rate": 5.897083791151706e-05,
+      "loss": 0.1465,
+      "step": 9410
+    },
+    {
+      "grad_norm": 0.6670511364936829,
+      "learning_rate": 5.888949372438336e-05,
+      "loss": 0.1529,
+      "step": 9420
+    },
+    {
+      "grad_norm": 0.5470581650733948,
+      "learning_rate": 5.8808125233713255e-05,
+      "loss": 0.1424,
+      "step": 9430
+    },
+    {
+      "grad_norm": 0.4502539038658142,
+      "learning_rate": 5.872673266196509e-05,
+      "loss": 0.138,
+      "step": 9440
+    },
+    {
+      "grad_norm": 0.41475042700767517,
+      "learning_rate": 5.864531623166305e-05,
+      "loss": 0.1419,
+      "step": 9450
+    },
+    {
+      "grad_norm": 0.5697763562202454,
+      "learning_rate": 5.856387616539656e-05,
+      "loss": 0.1426,
+      "step": 9460
+    },
+    {
+      "grad_norm": 0.5014544725418091,
+      "learning_rate": 5.848241268581967e-05,
+      "loss": 0.1634,
+      "step": 9470
+    },
+    {
+      "grad_norm": 0.44875621795654297,
+      "learning_rate": 5.840092601565037e-05,
+      "loss": 0.1605,
+      "step": 9480
+    },
+    {
+      "grad_norm": 0.4355970323085785,
+      "learning_rate": 5.8319416377670144e-05,
+      "loss": 0.1425,
+      "step": 9490
+    },
+    {
+      "grad_norm": 0.5939949750900269,
+      "learning_rate": 5.82378839947232e-05,
+      "loss": 0.1497,
+      "step": 9500
+    },
+    {
+      "grad_norm": 0.4647330343723297,
+      "learning_rate": 5.815632908971599e-05,
+      "loss": 0.1428,
+      "step": 9510
+    },
+    {
+      "grad_norm": 0.5400416851043701,
+      "learning_rate": 5.80747518856165e-05,
+      "loss": 0.1488,
+      "step": 9520
+    },
+    {
+      "grad_norm": 0.4881412386894226,
+      "learning_rate": 5.799315260545367e-05,
+      "loss": 0.1318,
+      "step": 9530
+    },
+    {
+      "grad_norm": 0.6033586859703064,
+      "learning_rate": 5.791153147231686e-05,
+      "loss": 0.1426,
+      "step": 9540
+    },
+    {
+      "grad_norm": 0.5710471272468567,
+      "learning_rate": 5.782988870935509e-05,
+      "loss": 0.1483,
+      "step": 9550
+    },
+    {
+      "grad_norm": 0.8451299071311951,
+      "learning_rate": 5.774822453977657e-05,
+      "loss": 0.1457,
+      "step": 9560
+    },
+    {
+      "grad_norm": 0.5806753635406494,
+      "learning_rate": 5.7666539186848036e-05,
+      "loss": 0.1362,
+      "step": 9570
+    },
+    {
+      "grad_norm": 0.5162105560302734,
+      "learning_rate": 5.758483287389411e-05,
+      "loss": 0.1433,
+      "step": 9580
+    },
+    {
+      "grad_norm": 0.4983697533607483,
+      "learning_rate": 5.7503105824296735e-05,
+      "loss": 0.1341,
+      "step": 9590
+    },
+    {
+      "grad_norm": 0.3076488971710205,
+      "learning_rate": 5.742135826149453e-05,
+      "loss": 0.1536,
+      "step": 9600
+    },
+    {
+      "grad_norm": 0.523889422416687,
+      "learning_rate": 5.7339590408982223e-05,
+      "loss": 0.1665,
+      "step": 9610
+    },
+    {
+      "grad_norm": 0.6269903182983398,
+      "learning_rate": 5.725780249031e-05,
+      "loss": 0.1434,
+      "step": 9620
+    },
+    {
+      "grad_norm": 0.5015751719474792,
+      "learning_rate": 5.717599472908292e-05,
+      "loss": 0.1316,
+      "step": 9630
+    },
+    {
+      "grad_norm": 0.4077630639076233,
+      "learning_rate": 5.7094167348960237e-05,
+      "loss": 0.1431,
+      "step": 9640
+    },
+    {
+      "grad_norm": 0.5937637090682983,
+      "learning_rate": 5.7012320573654945e-05,
+      "loss": 0.1452,
+      "step": 9650
+    },
+    {
+      "grad_norm": 0.5231539607048035,
+      "learning_rate": 5.693045462693295e-05,
+      "loss": 0.1627,
+      "step": 9660
+    },
+    {
+      "grad_norm": 0.4319014847278595,
+      "learning_rate": 5.684856973261266e-05,
+      "loss": 0.1557,
+      "step": 9670
+    },
+    {
+      "grad_norm": 0.5718788504600525,
+      "learning_rate": 5.6766666114564215e-05,
+      "loss": 0.1437,
+      "step": 9680
+    },
+    {
+      "grad_norm": 0.43910127878189087,
+      "learning_rate": 5.668474399670899e-05,
+      "loss": 0.1403,
+      "step": 9690
+    },
+    {
+      "grad_norm": 0.48854154348373413,
+      "learning_rate": 5.660280360301896e-05,
+      "loss": 0.1342,
+      "step": 9700
+    },
+    {
+      "grad_norm": 0.37922403216362,
+      "learning_rate": 5.652084515751599e-05,
+      "loss": 0.1493,
+      "step": 9710
+    },
+    {
+      "grad_norm": 0.5543588995933533,
+      "learning_rate": 5.643886888427137e-05,
+      "loss": 0.1716,
+      "step": 9720
+    },
+    {
+      "grad_norm": 0.4352636933326721,
+      "learning_rate": 5.6356875007405074e-05,
+      "loss": 0.1524,
+      "step": 9730
+    },
+    {
+      "grad_norm": 0.43619707226753235,
+      "learning_rate": 5.627486375108525e-05,
+      "loss": 0.1532,
+      "step": 9740
+    },
+    {
+      "grad_norm": 0.5202144980430603,
+      "learning_rate": 5.619283533952754e-05,
+      "loss": 0.14,
+      "step": 9750
+    },
+    {
+      "grad_norm": 0.391480028629303,
+      "learning_rate": 5.6110789996994474e-05,
+      "loss": 0.1382,
+      "step": 9760
+    },
+    {
+      "grad_norm": 0.6704453825950623,
+      "learning_rate": 5.602872794779491e-05,
+      "loss": 0.1586,
+      "step": 9770
+    },
+    {
+      "grad_norm": 0.4123976230621338,
+      "learning_rate": 5.594664941628334e-05,
+      "loss": 0.1613,
+      "step": 9780
+    },
+    {
+      "grad_norm": 0.520669162273407,
+      "learning_rate": 5.5864554626859324e-05,
+      "loss": 0.1519,
+      "step": 9790
+    },
+    {
+      "grad_norm": 0.3191356062889099,
+      "learning_rate": 5.578244380396691e-05,
+      "loss": 0.1443,
+      "step": 9800
+    },
+    {
+      "grad_norm": 0.43067264556884766,
+      "learning_rate": 5.570031717209394e-05,
+      "loss": 0.1538,
+      "step": 9810
+    },
+    {
+      "grad_norm": 0.8429540395736694,
+      "learning_rate": 5.561817495577147e-05,
+      "loss": 0.1609,
+      "step": 9820
+    },
+    {
+      "grad_norm": 0.3827391564846039,
+      "learning_rate": 5.5536017379573215e-05,
+      "loss": 0.1622,
+      "step": 9830
+    },
+    {
+      "grad_norm": 0.3954344689846039,
+      "learning_rate": 5.545384466811483e-05,
+      "loss": 0.156,
+      "step": 9840
+    },
+    {
+      "grad_norm": 0.41019681096076965,
+      "learning_rate": 5.5371657046053384e-05,
+      "loss": 0.1475,
+      "step": 9850
+    },
+    {
+      "grad_norm": 0.42015039920806885,
+      "learning_rate": 5.528945473808669e-05,
+      "loss": 0.1536,
+      "step": 9860
+    },
+    {
+      "grad_norm": 0.36494842171669006,
+      "learning_rate": 5.520723796895272e-05,
+      "loss": 0.1342,
+      "step": 9870
+    },
+    {
+      "grad_norm": 0.36835306882858276,
+      "learning_rate": 5.512500696342897e-05,
+      "loss": 0.1358,
+      "step": 9880
+    },
+    {
+      "grad_norm": 0.5547810792922974,
+      "learning_rate": 5.504276194633188e-05,
+      "loss": 0.1417,
+      "step": 9890
+    },
+    {
+      "grad_norm": 0.6774442791938782,
+      "learning_rate": 5.49605031425162e-05,
+      "loss": 0.158,
+      "step": 9900
+    },
+    {
+      "grad_norm": 0.5542992353439331,
+      "learning_rate": 5.487823077687434e-05,
+      "loss": 0.1726,
+      "step": 9910
+    },
+    {
+      "grad_norm": 0.4831373691558838,
+      "learning_rate": 5.4795945074335806e-05,
+      "loss": 0.1643,
+      "step": 9920
+    },
+    {
+      "grad_norm": 0.5753509402275085,
+      "learning_rate": 5.471364625986657e-05,
+      "loss": 0.1524,
+      "step": 9930
+    },
+    {
+      "grad_norm": 0.5133906602859497,
+      "learning_rate": 5.463133455846845e-05,
+      "loss": 0.1627,
+      "step": 9940
+    },
+    {
+      "grad_norm": 0.337358295917511,
+      "learning_rate": 5.4549010195178505e-05,
+      "loss": 0.1184,
+      "step": 9950
+    },
+    {
+      "grad_norm": 0.3677103817462921,
+      "learning_rate": 5.446667339506838e-05,
+      "loss": 0.1523,
+      "step": 9960
+    },
+    {
+      "grad_norm": 0.4786983132362366,
+      "learning_rate": 5.4384324383243756e-05,
+      "loss": 0.1483,
+      "step": 9970
+    },
+    {
+      "grad_norm": 0.5571385622024536,
+      "learning_rate": 5.430196338484368e-05,
+      "loss": 0.1297,
+      "step": 9980
+    },
+    {
+      "grad_norm": 0.42291325330734253,
+      "learning_rate": 5.4219590625039975e-05,
+      "loss": 0.1406,
+      "step": 9990
+    },
+    {
+      "grad_norm": 0.6355092525482178,
+      "learning_rate": 5.413720632903664e-05,
+      "loss": 0.1483,
+      "step": 10000
+    },
+    {
+      "grad_norm": 0.4612889289855957,
+      "learning_rate": 5.405481072206917e-05,
+      "loss": 0.165,
+      "step": 10010
+    },
+    {
+      "grad_norm": 0.5882695913314819,
+      "learning_rate": 5.397240402940402e-05,
+      "loss": 0.1438,
+      "step": 10020
+    },
+    {
+      "grad_norm": 0.5175050497055054,
+      "learning_rate": 5.388998647633794e-05,
+      "loss": 0.1473,
+      "step": 10030
+    },
+    {
+      "grad_norm": 0.3760899305343628,
+      "learning_rate": 5.380755828819737e-05,
+      "loss": 0.1499,
+      "step": 10040
+    },
+    {
+      "grad_norm": 0.5052558779716492,
+      "learning_rate": 5.3725119690337846e-05,
+      "loss": 0.1528,
+      "step": 10050
+    },
+    {
+      "grad_norm": 0.5411064624786377,
+      "learning_rate": 5.3642670908143324e-05,
+      "loss": 0.143,
+      "step": 10060
+    },
+    {
+      "grad_norm": 0.4038064479827881,
+      "learning_rate": 5.356021216702562e-05,
+      "loss": 0.1514,
+      "step": 10070
+    },
+    {
+      "grad_norm": 0.4561954438686371,
+      "learning_rate": 5.347774369242381e-05,
+      "loss": 0.1542,
+      "step": 10080
+    },
+    {
+      "grad_norm": 0.5946084856987,
+      "learning_rate": 5.3395265709803545e-05,
+      "loss": 0.143,
+      "step": 10090
+    },
+    {
+      "grad_norm": 0.7121728658676147,
+      "learning_rate": 5.331277844465647e-05,
+      "loss": 0.1427,
+      "step": 10100
+    },
+    {
+      "grad_norm": 0.2856979966163635,
+      "learning_rate": 5.323028212249963e-05,
+      "loss": 0.1438,
+      "step": 10110
+    },
+    {
+      "grad_norm": 0.4259609282016754,
+      "learning_rate": 5.314777696887481e-05,
+      "loss": 0.15,
+      "step": 10120
+    },
+    {
+      "grad_norm": 0.6267874240875244,
+      "learning_rate": 5.306526320934796e-05,
+      "loss": 0.1434,
+      "step": 10130
+    },
+    {
+      "grad_norm": 0.49387502670288086,
+      "learning_rate": 5.298274106950854e-05,
+      "loss": 0.1662,
+      "step": 10140
+    },
+    {
+      "grad_norm": 0.45723411440849304,
+      "learning_rate": 5.290021077496893e-05,
+      "loss": 0.1577,
+      "step": 10150
+    },
+    {
+      "grad_norm": 0.4757936894893646,
+      "learning_rate": 5.2817672551363816e-05,
+      "loss": 0.1562,
+      "step": 10160
+    },
+    {
+      "grad_norm": 0.5539591908454895,
+      "learning_rate": 5.273512662434952e-05,
+      "loss": 0.1482,
+      "step": 10170
+    },
+    {
+      "grad_norm": 0.3503054976463318,
+      "learning_rate": 5.265257321960349e-05,
+      "loss": 0.1535,
+      "step": 10180
+    },
+    {
+      "grad_norm": 0.49046820402145386,
+      "learning_rate": 5.257001256282357e-05,
+      "loss": 0.1548,
+      "step": 10190
+    },
+    {
+      "grad_norm": 0.4313761591911316,
+      "learning_rate": 5.248744487972742e-05,
+      "loss": 0.1374,
+      "step": 10200
+    },
+    {
+      "grad_norm": 0.42987844347953796,
+      "learning_rate": 5.240487039605196e-05,
+      "loss": 0.1742,
+      "step": 10210
+    },
+    {
+      "grad_norm": 0.45236629247665405,
+      "learning_rate": 5.232228933755267e-05,
+      "loss": 0.138,
+      "step": 10220
+    },
+    {
+      "grad_norm": 0.4238169193267822,
+      "learning_rate": 5.2239701930003006e-05,
+      "loss": 0.1611,
+      "step": 10230
+    },
+    {
+      "grad_norm": 0.4235691428184509,
+      "learning_rate": 5.215710839919379e-05,
+      "loss": 0.1305,
+      "step": 10240
+    },
+    {
+      "grad_norm": 0.5592643618583679,
+      "learning_rate": 5.207450897093257e-05,
+      "loss": 0.1522,
+      "step": 10250
+    },
+    {
+      "grad_norm": 0.7223462462425232,
+      "learning_rate": 5.1991903871043046e-05,
+      "loss": 0.1488,
+      "step": 10260
+    },
+    {
+      "grad_norm": 0.46924272179603577,
+      "learning_rate": 5.190929332536439e-05,
+      "loss": 0.1552,
+      "step": 10270
+    },
+    {
+      "grad_norm": 0.4227829873561859,
+      "learning_rate": 5.182667755975071e-05,
+      "loss": 0.1509,
+      "step": 10280
+    },
+    {
+      "grad_norm": 0.6450467109680176,
+      "learning_rate": 5.1744056800070315e-05,
+      "loss": 0.1446,
+      "step": 10290
+    },
+    {
+      "grad_norm": 0.38730642199516296,
+      "learning_rate": 5.166143127220524e-05,
+      "loss": 0.1585,
+      "step": 10300
+    },
+    {
+      "grad_norm": 0.4760332405567169,
+      "learning_rate": 5.1578801202050485e-05,
+      "loss": 0.1621,
+      "step": 10310
+    },
+    {
+      "grad_norm": 0.377533882856369,
+      "learning_rate": 5.149616681551355e-05,
+      "loss": 0.1432,
+      "step": 10320
+    },
+    {
+      "grad_norm": 0.5672144889831543,
+      "learning_rate": 5.141352833851367e-05,
+      "loss": 0.1527,
+      "step": 10330
+    },
+    {
+      "grad_norm": 0.37412023544311523,
+      "learning_rate": 5.1330885996981285e-05,
+      "loss": 0.16,
+      "step": 10340
+    },
+    {
+      "grad_norm": 0.3990615904331207,
+      "learning_rate": 5.124824001685741e-05,
+      "loss": 0.1521,
+      "step": 10350
+    },
+    {
+      "grad_norm": 0.3501453101634979,
+      "learning_rate": 5.116559062409298e-05,
+      "loss": 0.1526,
+      "step": 10360
+    },
+    {
+      "grad_norm": 0.6271255612373352,
+      "learning_rate": 5.10829380446483e-05,
+      "loss": 0.1565,
+      "step": 10370
+    },
+    {
+      "grad_norm": 0.5372224450111389,
+      "learning_rate": 5.100028250449235e-05,
+      "loss": 0.1552,
+      "step": 10380
+    },
+    {
+      "grad_norm": 0.4000946879386902,
+      "learning_rate": 5.0917624229602234e-05,
+      "loss": 0.1726,
+      "step": 10390
+    },
+    {
+      "grad_norm": 0.49024948477745056,
+      "learning_rate": 5.0834963445962524e-05,
+      "loss": 0.1391,
+      "step": 10400
+    },
+    {
+      "grad_norm": 1.1074209213256836,
+      "learning_rate": 5.075230037956461e-05,
+      "loss": 0.1482,
+      "step": 10410
+    },
+    {
+      "grad_norm": 0.523674488067627,
+      "learning_rate": 5.0669635256406213e-05,
+      "loss": 0.1279,
+      "step": 10420
+    },
+    {
+      "grad_norm": 0.551565945148468,
+      "learning_rate": 5.058696830249058e-05,
+      "loss": 0.134,
+      "step": 10430
+    },
+    {
+      "grad_norm": 0.3917821943759918,
+      "learning_rate": 5.050429974382602e-05,
+      "loss": 0.1696,
+      "step": 10440
+    },
+    {
+      "grad_norm": 0.46342769265174866,
+      "learning_rate": 5.042162980642523e-05,
+      "loss": 0.1398,
+      "step": 10450
+    },
+    {
+      "grad_norm": 0.4497271180152893,
+      "learning_rate": 5.033895871630462e-05,
+      "loss": 0.1432,
+      "step": 10460
+    },
+    {
+      "grad_norm": 0.44396674633026123,
+      "learning_rate": 5.025628669948386e-05,
+      "loss": 0.1475,
+      "step": 10470
+    },
+    {
+      "grad_norm": 0.4294814169406891,
+      "learning_rate": 5.017361398198502e-05,
+      "loss": 0.1419,
+      "step": 10480
+    },
+    {
+      "grad_norm": 0.5722894072532654,
+      "learning_rate": 5.009094078983221e-05,
+      "loss": 0.142,
+      "step": 10490
+    },
+    {
+      "grad_norm": 0.5636281371116638,
+      "learning_rate": 5.000826734905073e-05,
+      "loss": 0.1632,
+      "step": 10500
+    },
+    {
+      "grad_norm": 0.5491324067115784,
+      "learning_rate": 4.9925593885666645e-05,
+      "loss": 0.1583,
+      "step": 10510
+    },
+    {
+      "grad_norm": 0.6083096861839294,
+      "learning_rate": 4.984292062570602e-05,
+      "loss": 0.171,
+      "step": 10520
+    },
+    {
+      "grad_norm": 0.930322527885437,
+      "learning_rate": 4.976024779519442e-05,
+      "loss": 0.1822,
+      "step": 10530
+    },
+    {
+      "grad_norm": 0.5122578740119934,
+      "learning_rate": 4.9677575620156194e-05,
+      "loss": 0.1483,
+      "step": 10540
+    },
+    {
+      "grad_norm": 0.4673050045967102,
+      "learning_rate": 4.959490432661391e-05,
+      "loss": 0.1328,
+      "step": 10550
+    },
+    {
+      "grad_norm": 0.4568622410297394,
+      "learning_rate": 4.9512234140587726e-05,
+      "loss": 0.1541,
+      "step": 10560
+    },
+    {
+      "grad_norm": 0.49044907093048096,
+      "learning_rate": 4.942956528809477e-05,
+      "loss": 0.1594,
+      "step": 10570
+    },
+    {
+      "grad_norm": 0.6256377696990967,
+      "learning_rate": 4.934689799514854e-05,
+      "loss": 0.1739,
+      "step": 10580
+    },
+    {
+      "grad_norm": 0.48133817315101624,
+      "learning_rate": 4.926423248775827e-05,
+      "loss": 0.1502,
+      "step": 10590
+    },
+    {
+      "grad_norm": 0.6031012535095215,
+      "learning_rate": 4.918156899192826e-05,
+      "loss": 0.1522,
+      "step": 10600
+    },
+    {
+      "grad_norm": 0.383358895778656,
+      "learning_rate": 4.909890773365738e-05,
+      "loss": 0.1424,
+      "step": 10610
+    },
+    {
+      "grad_norm": 0.4280951917171478,
+      "learning_rate": 4.9016248938938344e-05,
+      "loss": 0.1483,
+      "step": 10620
+    },
+    {
+      "grad_norm": 0.40035688877105713,
+      "learning_rate": 4.8933592833757156e-05,
+      "loss": 0.1433,
+      "step": 10630
+    },
+    {
+      "grad_norm": 0.43933045864105225,
+      "learning_rate": 4.8850939644092435e-05,
+      "loss": 0.136,
+      "step": 10640
+    },
+    {
+      "grad_norm": 0.36851996183395386,
+      "learning_rate": 4.876828959591485e-05,
+      "loss": 0.1256,
+      "step": 10650
+    },
+    {
+      "grad_norm": 0.5466973781585693,
+      "learning_rate": 4.8685642915186474e-05,
+      "loss": 0.1431,
+      "step": 10660
+    },
+    {
+      "grad_norm": 0.6948999762535095,
+      "learning_rate": 4.860299982786018e-05,
+      "loss": 0.1577,
+      "step": 10670
+    },
+    {
+      "grad_norm": 0.33470630645751953,
+      "learning_rate": 4.852036055987901e-05,
+      "loss": 0.1239,
+      "step": 10680
+    },
+    {
+      "grad_norm": 0.5142731666564941,
+      "learning_rate": 4.843772533717558e-05,
+      "loss": 0.1614,
+      "step": 10690
+    },
+    {
+      "grad_norm": 0.48010584712028503,
+      "learning_rate": 4.835509438567142e-05,
+      "loss": 0.1573,
+      "step": 10700
+    },
+    {
+      "grad_norm": 0.4586421549320221,
+      "learning_rate": 4.827246793127639e-05,
+      "loss": 0.1403,
+      "step": 10710
+    },
+    {
+      "grad_norm": 0.7138292789459229,
+      "learning_rate": 4.818984619988807e-05,
+      "loss": 0.1762,
+      "step": 10720
+    },
+    {
+      "grad_norm": 0.42487064003944397,
+      "learning_rate": 4.810722941739115e-05,
+      "loss": 0.1357,
+      "step": 10730
+    },
+    {
+      "grad_norm": 0.5380846858024597,
+      "learning_rate": 4.8024617809656684e-05,
+      "loss": 0.1421,
+      "step": 10740
+    },
+    {
+      "grad_norm": 0.5628655552864075,
+      "learning_rate": 4.794201160254171e-05,
+      "loss": 0.1522,
+      "step": 10750
+    },
+    {
+      "grad_norm": 0.4800051152706146,
+      "learning_rate": 4.785941102188844e-05,
+      "loss": 0.141,
+      "step": 10760
+    },
+    {
+      "grad_norm": 0.42914098501205444,
+      "learning_rate": 4.7776816293523686e-05,
+      "loss": 0.1323,
+      "step": 10770
+    },
+    {
+      "grad_norm": 0.48330816626548767,
+      "learning_rate": 4.769422764325832e-05,
+      "loss": 0.1632,
+      "step": 10780
+    },
+    {
+      "grad_norm": 0.44986045360565186,
+      "learning_rate": 4.76116452968865e-05,
+      "loss": 0.1554,
+      "step": 10790
+    },
+    {
+      "grad_norm": 0.5064200162887573,
+      "learning_rate": 4.752906948018525e-05,
+      "loss": 0.1556,
+      "step": 10800
+    },
+    {
+      "grad_norm": 0.4190466105937958,
+      "learning_rate": 4.7446500418913684e-05,
+      "loss": 0.1396,
+      "step": 10810
+    },
+    {
+      "grad_norm": 0.715853214263916,
+      "learning_rate": 4.736393833881247e-05,
+      "loss": 0.1416,
+      "step": 10820
+    },
+    {
+      "grad_norm": 0.36505401134490967,
+      "learning_rate": 4.7281383465603194e-05,
+      "loss": 0.1401,
+      "step": 10830
+    },
+    {
+      "grad_norm": 0.4902125895023346,
+      "learning_rate": 4.71988360249877e-05,
+      "loss": 0.1403,
+      "step": 10840
+    },
+    {
+      "grad_norm": 0.46639740467071533,
+      "learning_rate": 4.7116296242647554e-05,
+      "loss": 0.1529,
+      "step": 10850
+    },
+    {
+      "grad_norm": 0.6440649628639221,
+      "learning_rate": 4.703376434424336e-05,
+      "loss": 0.1459,
+      "step": 10860
+    },
+    {
+      "grad_norm": 0.38722774386405945,
+      "learning_rate": 4.695124055541421e-05,
+      "loss": 0.1539,
+      "step": 10870
+    },
+    {
+      "grad_norm": 0.5275564193725586,
+      "learning_rate": 4.6868725101776934e-05,
+      "loss": 0.141,
+      "step": 10880
+    },
+    {
+      "grad_norm": 0.4437614977359772,
+      "learning_rate": 4.678621820892567e-05,
+      "loss": 0.1392,
+      "step": 10890
+    },
+    {
+      "grad_norm": 0.5153006911277771,
+      "learning_rate": 4.670372010243111e-05,
+      "loss": 0.1476,
+      "step": 10900
+    },
+    {
+      "grad_norm": 0.5635380148887634,
+      "learning_rate": 4.662123100783992e-05,
+      "loss": 0.1335,
+      "step": 10910
+    },
+    {
+      "grad_norm": 0.3748430609703064,
+      "learning_rate": 4.653875115067415e-05,
+      "loss": 0.1311,
+      "step": 10920
+    },
+    {
+      "grad_norm": 0.3062388002872467,
+      "learning_rate": 4.6456280756430545e-05,
+      "loss": 0.1275,
+      "step": 10930
+    },
+    {
+      "grad_norm": 0.33864331245422363,
+      "learning_rate": 4.637382005058004e-05,
+      "loss": 0.1232,
+      "step": 10940
+    },
+    {
+      "grad_norm": 0.7302923798561096,
+      "learning_rate": 4.629136925856705e-05,
+      "loss": 0.1289,
+      "step": 10950
+    },
+    {
+      "grad_norm": 0.5158184766769409,
+      "learning_rate": 4.6208928605808895e-05,
+      "loss": 0.17,
+      "step": 10960
+    },
+    {
+      "grad_norm": 0.45864617824554443,
+      "learning_rate": 4.612649831769519e-05,
+      "loss": 0.1689,
+      "step": 10970
+    },
+    {
+      "grad_norm": 0.4657990038394928,
+      "learning_rate": 4.604407861958715e-05,
+      "loss": 0.136,
+      "step": 10980
+    },
+    {
+      "grad_norm": 0.37360793352127075,
+      "learning_rate": 4.5961669736817114e-05,
+      "loss": 0.1326,
+      "step": 10990
+    },
+    {
+      "grad_norm": 0.8401086926460266,
+      "learning_rate": 4.5879271894687814e-05,
+      "loss": 0.1397,
+      "step": 11000
+    },
+    {
+      "grad_norm": 0.4285585880279541,
+      "learning_rate": 4.5796885318471826e-05,
+      "loss": 0.1454,
+      "step": 11010
+    },
+    {
+      "grad_norm": 0.35717225074768066,
+      "learning_rate": 4.571451023341086e-05,
+      "loss": 0.1349,
+      "step": 11020
+    },
+    {
+      "grad_norm": 0.6365489363670349,
+      "learning_rate": 4.563214686471527e-05,
+      "loss": 0.1393,
+      "step": 11030
+    },
+    {
+      "grad_norm": 0.6525166630744934,
+      "learning_rate": 4.5549795437563365e-05,
+      "loss": 0.1519,
+      "step": 11040
+    },
+    {
+      "grad_norm": 0.8059558868408203,
+      "learning_rate": 4.546745617710081e-05,
+      "loss": 0.1497,
+      "step": 11050
+    },
+    {
+      "grad_norm": 0.3655293881893158,
+      "learning_rate": 4.5385129308440014e-05,
+      "loss": 0.1458,
+      "step": 11060
+    },
+    {
+      "grad_norm": 0.6059927344322205,
+      "learning_rate": 4.530281505665944e-05,
+      "loss": 0.1276,
+      "step": 11070
+    },
+    {
+      "grad_norm": 0.519593358039856,
+      "learning_rate": 4.5220513646803134e-05,
+      "loss": 0.1476,
+      "step": 11080
+    },
+    {
+      "grad_norm": 0.5255056023597717,
+      "learning_rate": 4.513822530388003e-05,
+      "loss": 0.1343,
+      "step": 11090
+    },
+    {
+      "grad_norm": 0.5028067827224731,
+      "learning_rate": 4.5055950252863296e-05,
+      "loss": 0.131,
+      "step": 11100
+    },
+    {
+      "grad_norm": 0.41324442625045776,
+      "learning_rate": 4.4973688718689803e-05,
+      "loss": 0.1412,
+      "step": 11110
+    },
+    {
+      "grad_norm": 0.5088866949081421,
+      "learning_rate": 4.4891440926259406e-05,
+      "loss": 0.1342,
+      "step": 11120
+    },
+    {
+      "grad_norm": 0.448533833026886,
+      "learning_rate": 4.480920710043443e-05,
+      "loss": 0.1357,
+      "step": 11130
+    },
+    {
+      "grad_norm": 0.5223669409751892,
+      "learning_rate": 4.4726987466039044e-05,
+      "loss": 0.1353,
+      "step": 11140
+    },
+    {
+      "grad_norm": 0.3482387661933899,
+      "learning_rate": 4.46447822478586e-05,
+      "loss": 0.1251,
+      "step": 11150
+    },
+    {
+      "grad_norm": 0.7929878234863281,
+      "learning_rate": 4.4562591670638974e-05,
+      "loss": 0.1477,
+      "step": 11160
+    },
+    {
+      "grad_norm": 0.4207678437232971,
+      "learning_rate": 4.4480415959086105e-05,
+      "loss": 0.1301,
+      "step": 11170
+    },
+    {
+      "grad_norm": 0.4079393744468689,
+      "learning_rate": 4.439825533786522e-05,
+      "loss": 0.1374,
+      "step": 11180
+    },
+    {
+      "grad_norm": 0.5330106616020203,
+      "learning_rate": 4.431611003160035e-05,
+      "loss": 0.1419,
+      "step": 11190
+    },
+    {
+      "grad_norm": 0.5333981513977051,
+      "learning_rate": 4.4233980264873636e-05,
+      "loss": 0.1497,
+      "step": 11200
+    },
+    {
+      "grad_norm": 0.5646610856056213,
+      "learning_rate": 4.4151866262224684e-05,
+      "loss": 0.153,
+      "step": 11210
+    },
+    {
+      "grad_norm": 0.7247358560562134,
+      "learning_rate": 4.406976824815006e-05,
+      "loss": 0.1406,
+      "step": 11220
+    },
+    {
+      "grad_norm": 0.48595622181892395,
+      "learning_rate": 4.3987686447102595e-05,
+      "loss": 0.161,
+      "step": 11230
+    },
+    {
+      "grad_norm": 0.3686184287071228,
+      "learning_rate": 4.3905621083490804e-05,
+      "loss": 0.143,
+      "step": 11240
+    },
+    {
+      "grad_norm": 0.28096747398376465,
+      "learning_rate": 4.3823572381678286e-05,
+      "loss": 0.1239,
+      "step": 11250
+    },
+    {
+      "grad_norm": 0.49489763379096985,
+      "learning_rate": 4.374154056598301e-05,
+      "loss": 0.1356,
+      "step": 11260
+    },
+    {
+      "grad_norm": 0.37762606143951416,
+      "learning_rate": 4.3659525860676845e-05,
+      "loss": 0.1411,
+      "step": 11270
+    },
+    {
+      "grad_norm": 0.4733501374721527,
+      "learning_rate": 4.3577528489984854e-05,
+      "loss": 0.1469,
+      "step": 11280
+    },
+    {
+      "grad_norm": 0.716112494468689,
+      "learning_rate": 4.349554867808476e-05,
+      "loss": 0.1464,
+      "step": 11290
+    },
+    {
+      "grad_norm": 0.6870642900466919,
+      "learning_rate": 4.34135866491062e-05,
+      "loss": 0.1757,
+      "step": 11300
+    },
+    {
+      "grad_norm": 0.47758838534355164,
+      "learning_rate": 4.333164262713022e-05,
+      "loss": 0.1588,
+      "step": 11310
+    },
+    {
+      "grad_norm": 0.4254021942615509,
+      "learning_rate": 4.324971683618868e-05,
+      "loss": 0.1582,
+      "step": 11320
+    },
+    {
+      "grad_norm": 0.37129178643226624,
+      "learning_rate": 4.316780950026354e-05,
+      "loss": 0.1823,
+      "step": 11330
+    },
+    {
+      "grad_norm": 0.42764630913734436,
+      "learning_rate": 4.308592084328637e-05,
+      "loss": 0.1403,
+      "step": 11340
+    },
+    {
+      "grad_norm": 0.3811211585998535,
+      "learning_rate": 4.3004051089137576e-05,
+      "loss": 0.138,
+      "step": 11350
+    },
+    {
+      "grad_norm": 0.4244554042816162,
+      "learning_rate": 4.292220046164597e-05,
+      "loss": 0.1383,
+      "step": 11360
+    },
+    {
+      "grad_norm": 0.5916116833686829,
+      "learning_rate": 4.2840369184588035e-05,
+      "loss": 0.1462,
+      "step": 11370
+    },
+    {
+      "grad_norm": 0.5111758708953857,
+      "learning_rate": 4.2758557481687345e-05,
+      "loss": 0.1358,
+      "step": 11380
+    },
+    {
+      "grad_norm": 0.454848974943161,
+      "learning_rate": 4.267676557661403e-05,
+      "loss": 0.1426,
+      "step": 11390
+    },
+    {
+      "grad_norm": 0.41505560278892517,
+      "learning_rate": 4.2594993692983955e-05,
+      "loss": 0.1243,
+      "step": 11400
+    },
+    {
+      "grad_norm": 0.3787071704864502,
+      "learning_rate": 4.251324205435837e-05,
+      "loss": 0.1371,
+      "step": 11410
+    },
+    {
+      "grad_norm": 0.45758819580078125,
+      "learning_rate": 4.243151088424312e-05,
+      "loss": 0.1621,
+      "step": 11420
+    },
+    {
+      "grad_norm": 0.5442251563072205,
+      "learning_rate": 4.234980040608813e-05,
+      "loss": 0.1525,
+      "step": 11430
+    },
+    {
+      "grad_norm": 0.6122934818267822,
+      "learning_rate": 4.22681108432867e-05,
+      "loss": 0.1565,
+      "step": 11440
+    },
+    {
+      "grad_norm": 0.3289790451526642,
+      "learning_rate": 4.2186442419174984e-05,
+      "loss": 0.1627,
+      "step": 11450
+    },
+    {
+      "grad_norm": 0.676074743270874,
+      "learning_rate": 4.210479535703133e-05,
+      "loss": 0.1534,
+      "step": 11460
+    },
+    {
+      "grad_norm": 0.35827261209487915,
+      "learning_rate": 4.202316988007567e-05,
+      "loss": 0.151,
+      "step": 11470
+    },
+    {
+      "grad_norm": 0.5259619951248169,
+      "learning_rate": 4.194156621146901e-05,
+      "loss": 0.129,
+      "step": 11480
+    },
+    {
+      "grad_norm": 0.43125808238983154,
+      "learning_rate": 4.1859984574312596e-05,
+      "loss": 0.1281,
+      "step": 11490
+    },
+    {
+      "grad_norm": 0.4761846661567688,
+      "learning_rate": 4.177842519164752e-05,
+      "loss": 0.1416,
+      "step": 11500
+    },
+    {
+      "grad_norm": 0.4888117015361786,
+      "learning_rate": 4.169688828645404e-05,
+      "loss": 0.1213,
+      "step": 11510
+    },
+    {
+      "grad_norm": 0.5855244398117065,
+      "learning_rate": 4.161537408165092e-05,
+      "loss": 0.1357,
+      "step": 11520
+    },
+    {
+      "grad_norm": 0.5944448709487915,
+      "learning_rate": 4.1533882800094924e-05,
+      "loss": 0.1402,
+      "step": 11530
+    },
+    {
+      "grad_norm": 0.5820303559303284,
+      "learning_rate": 4.145241466458005e-05,
+      "loss": 0.1359,
+      "step": 11540
+    },
+    {
+      "grad_norm": 0.521994948387146,
+      "learning_rate": 4.13709698978371e-05,
+      "loss": 0.1427,
+      "step": 11550
+    },
+    {
+      "grad_norm": 0.6419322490692139,
+      "learning_rate": 4.1289548722532944e-05,
+      "loss": 0.1164,
+      "step": 11560
+    },
+    {
+      "grad_norm": 0.529522716999054,
+      "learning_rate": 4.120815136126999e-05,
+      "loss": 0.1515,
+      "step": 11570
+    },
+    {
+      "grad_norm": 0.3663100600242615,
+      "learning_rate": 4.112677803658548e-05,
+      "loss": 0.1491,
+      "step": 11580
+    },
+    {
+      "grad_norm": 0.5234674215316772,
+      "learning_rate": 4.1045428970951e-05,
+      "loss": 0.1492,
+      "step": 11590
+    },
+    {
+      "grad_norm": 0.47266897559165955,
+      "learning_rate": 4.0964104386771785e-05,
+      "loss": 0.133,
+      "step": 11600
+    },
+    {
+      "grad_norm": 0.4186302423477173,
+      "learning_rate": 4.0882804506386144e-05,
+      "loss": 0.1352,
+      "step": 11610
+    },
+    {
+      "grad_norm": 0.6860542297363281,
+      "learning_rate": 4.080152955206485e-05,
+      "loss": 0.1596,
+      "step": 11620
+    },
+    {
+      "grad_norm": 0.3779941201210022,
+      "learning_rate": 4.0720279746010505e-05,
+      "loss": 0.139,
+      "step": 11630
+    },
+    {
+      "grad_norm": 0.4613971710205078,
+      "learning_rate": 4.063905531035699e-05,
+      "loss": 0.1161,
+      "step": 11640
+    },
+    {
+      "grad_norm": 0.34459319710731506,
+      "learning_rate": 4.055785646716882e-05,
+      "loss": 0.1524,
+      "step": 11650
+    },
+    {
+      "grad_norm": 0.3938691020011902,
+      "learning_rate": 4.047668343844051e-05,
+      "loss": 0.1672,
+      "step": 11660
+    },
+    {
+      "grad_norm": 0.5669636130332947,
+      "learning_rate": 4.039553644609604e-05,
+      "loss": 0.1404,
+      "step": 11670
+    },
+    {
+      "grad_norm": 0.4859471023082733,
+      "learning_rate": 4.0314415711988176e-05,
+      "loss": 0.1434,
+      "step": 11680
+    },
+    {
+      "grad_norm": 0.47179779410362244,
+      "learning_rate": 4.023332145789792e-05,
+      "loss": 0.1428,
+      "step": 11690
+    },
+    {
+      "grad_norm": 0.4263125956058502,
+      "learning_rate": 4.015225390553385e-05,
+      "loss": 0.1453,
+      "step": 11700
+    },
+    {
+      "grad_norm": 0.5297496914863586,
+      "learning_rate": 4.007121327653158e-05,
+      "loss": 0.138,
+      "step": 11710
+    },
+    {
+      "grad_norm": 0.49534937739372253,
+      "learning_rate": 3.9990199792453064e-05,
+      "loss": 0.1374,
+      "step": 11720
+    },
+    {
+      "grad_norm": 0.4692939817905426,
+      "learning_rate": 3.9909213674786103e-05,
+      "loss": 0.1345,
+      "step": 11730
+    },
+    {
+      "grad_norm": 0.6668260097503662,
+      "learning_rate": 3.982825514494363e-05,
+      "loss": 0.1567,
+      "step": 11740
+    },
+    {
+      "grad_norm": 0.6448221206665039,
+      "learning_rate": 3.974732442426319e-05,
+      "loss": 0.167,
+      "step": 11750
+    },
+    {
+      "grad_norm": 0.5559336543083191,
+      "learning_rate": 3.966642173400629e-05,
+      "loss": 0.1494,
+      "step": 11760
+    },
+    {
+      "grad_norm": 0.5132073760032654,
+      "learning_rate": 3.9585547295357764e-05,
+      "loss": 0.1503,
+      "step": 11770
+    },
+    {
+      "grad_norm": 0.4282352328300476,
+      "learning_rate": 3.950470132942526e-05,
+      "loss": 0.1259,
+      "step": 11780
+    },
+    {
+      "grad_norm": 0.4071321487426758,
+      "learning_rate": 3.942388405723856e-05,
+      "loss": 0.1511,
+      "step": 11790
+    },
+    {
+      "grad_norm": 0.4684301018714905,
+      "learning_rate": 3.9343095699749e-05,
+      "loss": 0.1456,
+      "step": 11800
+    },
+    {
+      "grad_norm": 0.5778101682662964,
+      "learning_rate": 3.9262336477828874e-05,
+      "loss": 0.1354,
+      "step": 11810
+    },
+    {
+      "grad_norm": 0.3759154975414276,
+      "learning_rate": 3.9181606612270794e-05,
+      "loss": 0.1434,
+      "step": 11820
+    },
+    {
+      "grad_norm": 0.5054191946983337,
+      "learning_rate": 3.910090632378713e-05,
+      "loss": 0.1726,
+      "step": 11830
+    },
+    {
+      "grad_norm": 0.4369896650314331,
+      "learning_rate": 3.90202358330094e-05,
+      "loss": 0.1351,
+      "step": 11840
+    },
+    {
+      "grad_norm": 0.6141563057899475,
+      "learning_rate": 3.8939595360487656e-05,
+      "loss": 0.1635,
+      "step": 11850
+    },
+    {
+      "grad_norm": 0.4740920066833496,
+      "learning_rate": 3.885898512668984e-05,
+      "loss": 0.1461,
+      "step": 11860
+    },
+    {
+      "grad_norm": 0.5038125514984131,
+      "learning_rate": 3.877840535200127e-05,
+      "loss": 0.1464,
+      "step": 11870
+    },
+    {
+      "grad_norm": 0.6280749440193176,
+      "learning_rate": 3.869785625672397e-05,
+      "loss": 0.1393,
+      "step": 11880
+    },
+    {
+      "grad_norm": 0.5136828422546387,
+      "learning_rate": 3.8617338061076094e-05,
+      "loss": 0.1602,
+      "step": 11890
+    },
+    {
+      "grad_norm": 0.44834092259407043,
+      "learning_rate": 3.853685098519132e-05,
+      "loss": 0.1525,
+      "step": 11900
+    },
+    {
+      "grad_norm": 0.492308109998703,
+      "learning_rate": 3.845639524911823e-05,
+      "loss": 0.1414,
+      "step": 11910
+    },
+    {
+      "grad_norm": 0.5485745668411255,
+      "learning_rate": 3.837597107281974e-05,
+      "loss": 0.1453,
+      "step": 11920
+    },
+    {
+      "grad_norm": 0.4516947567462921,
+      "learning_rate": 3.829557867617247e-05,
+      "loss": 0.1348,
+      "step": 11930
+    },
+    {
+      "grad_norm": 0.5421794652938843,
+      "learning_rate": 3.821521827896618e-05,
+      "loss": 0.1352,
+      "step": 11940
+    },
+    {
+      "grad_norm": 0.6226781606674194,
+      "learning_rate": 3.81348901009031e-05,
+      "loss": 0.1635,
+      "step": 11950
+    },
+    {
+      "grad_norm": 0.5080947279930115,
+      "learning_rate": 3.805459436159741e-05,
+      "loss": 0.1508,
+      "step": 11960
+    },
+    {
+      "grad_norm": 0.5345335602760315,
+      "learning_rate": 3.797433128057461e-05,
+      "loss": 0.1388,
+      "step": 11970
+    },
+    {
+      "grad_norm": 0.5552277565002441,
+      "learning_rate": 3.789410107727089e-05,
+      "loss": 0.1549,
+      "step": 11980
+    },
+    {
+      "grad_norm": 0.6635316014289856,
+      "learning_rate": 3.781390397103257e-05,
+      "loss": 0.155,
+      "step": 11990
+    },
+    {
+      "grad_norm": 0.4638996422290802,
+      "learning_rate": 3.7733740181115455e-05,
+      "loss": 0.1275,
+      "step": 12000
+    },
+    {
+      "grad_norm": 0.44815176725387573,
+      "learning_rate": 3.7653609926684306e-05,
+      "loss": 0.158,
+      "step": 12010
+    },
+    {
+      "grad_norm": 0.4426247775554657,
+      "learning_rate": 3.757351342681217e-05,
+      "loss": 0.1471,
+      "step": 12020
+    },
+    {
+      "grad_norm": 0.5822455883026123,
+      "learning_rate": 3.749345090047982e-05,
+      "loss": 0.1496,
+      "step": 12030
+    },
+    {
+      "grad_norm": 0.35467076301574707,
+      "learning_rate": 3.741342256657515e-05,
+      "loss": 0.1573,
+      "step": 12040
+    },
+    {
+      "grad_norm": 0.5544428825378418,
+      "learning_rate": 3.7333428643892567e-05,
+      "loss": 0.1307,
+      "step": 12050
+    },
+    {
+      "grad_norm": 0.4993782639503479,
+      "learning_rate": 3.725346935113239e-05,
+      "loss": 0.1319,
+      "step": 12060
+    },
+    {
+      "grad_norm": 0.5056540966033936,
+      "learning_rate": 3.717354490690029e-05,
+      "loss": 0.1511,
+      "step": 12070
+    },
+    {
+      "grad_norm": 0.6148380041122437,
+      "learning_rate": 3.709365552970664e-05,
+      "loss": 0.1407,
+      "step": 12080
+    },
+    {
+      "grad_norm": 0.5592396855354309,
+      "learning_rate": 3.7013801437965945e-05,
+      "loss": 0.1566,
+      "step": 12090
+    },
+    {
+      "grad_norm": 0.8606341481208801,
+      "learning_rate": 3.693398284999623e-05,
+      "loss": 0.1635,
+      "step": 12100
+    },
+    {
+      "grad_norm": 0.7090617418289185,
+      "learning_rate": 3.6854199984018484e-05,
+      "loss": 0.1492,
+      "step": 12110
+    },
+    {
+      "grad_norm": 0.6167499423027039,
+      "learning_rate": 3.677445305815601e-05,
+      "loss": 0.1748,
+      "step": 12120
+    },
+    {
+      "grad_norm": 0.6157158613204956,
+      "learning_rate": 3.669474229043387e-05,
+      "loss": 0.1577,
+      "step": 12130
+    },
+    {
+      "grad_norm": 0.551916778087616,
+      "learning_rate": 3.6615067898778235e-05,
+      "loss": 0.1474,
+      "step": 12140
+    },
+    {
+      "grad_norm": 0.42716094851493835,
+      "learning_rate": 3.6535430101015866e-05,
+      "loss": 0.1707,
+      "step": 12150
+    },
+    {
+      "grad_norm": 0.4142579138278961,
+      "learning_rate": 3.645582911487345e-05,
+      "loss": 0.1281,
+      "step": 12160
+    },
+    {
+      "grad_norm": 0.5094059705734253,
+      "learning_rate": 3.637626515797706e-05,
+      "loss": 0.1536,
+      "step": 12170
+    },
+    {
+      "grad_norm": 0.4376841187477112,
+      "learning_rate": 3.629673844785152e-05,
+      "loss": 0.137,
+      "step": 12180
+    },
+    {
+      "grad_norm": 0.38608500361442566,
+      "learning_rate": 3.621724920191979e-05,
+      "loss": 0.1475,
+      "step": 12190
+    },
+    {
+      "grad_norm": 0.47520437836647034,
+      "learning_rate": 3.6137797637502444e-05,
+      "loss": 0.1358,
+      "step": 12200
+    },
+    {
+      "grad_norm": 0.36206260323524475,
+      "learning_rate": 3.6058383971817035e-05,
+      "loss": 0.144,
+      "step": 12210
+    },
+    {
+      "grad_norm": 0.5303428769111633,
+      "learning_rate": 3.59790084219775e-05,
+      "loss": 0.1433,
+      "step": 12220
+    },
+    {
+      "grad_norm": 0.4407895803451538,
+      "learning_rate": 3.589967120499353e-05,
+      "loss": 0.1427,
+      "step": 12230
+    },
+    {
+      "grad_norm": 0.6267043948173523,
+      "learning_rate": 3.5820372537770075e-05,
+      "loss": 0.1529,
+      "step": 12240
+    },
+    {
+      "grad_norm": 0.8943570852279663,
+      "learning_rate": 3.5741112637106655e-05,
+      "loss": 0.1583,
+      "step": 12250
+    },
+    {
+      "grad_norm": 0.603672981262207,
+      "learning_rate": 3.5661891719696804e-05,
+      "loss": 0.16,
+      "step": 12260
+    },
+    {
+      "grad_norm": 0.4142937660217285,
+      "learning_rate": 3.5582710002127504e-05,
+      "loss": 0.1385,
+      "step": 12270
+    },
+    {
+      "grad_norm": 0.44892171025276184,
+      "learning_rate": 3.550356770087853e-05,
+      "loss": 0.1455,
+      "step": 12280
+    },
+    {
+      "grad_norm": 0.4810429811477661,
+      "learning_rate": 3.5424465032321914e-05,
+      "loss": 0.1288,
+      "step": 12290
+    },
+    {
+      "grad_norm": 0.5047153830528259,
+      "learning_rate": 3.5345402212721335e-05,
+      "loss": 0.1636,
+      "step": 12300
+    },
+    {
+      "grad_norm": 0.4368864595890045,
+      "learning_rate": 3.526637945823152e-05,
+      "loss": 0.1396,
+      "step": 12310
+    },
+    {
+      "grad_norm": 0.4284287989139557,
+      "learning_rate": 3.518739698489767e-05,
+      "loss": 0.1297,
+      "step": 12320
+    },
+    {
+      "grad_norm": 0.40356409549713135,
+      "learning_rate": 3.510845500865485e-05,
+      "loss": 0.14,
+      "step": 12330
+    },
+    {
+      "grad_norm": 0.3771931231021881,
+      "learning_rate": 3.502955374532739e-05,
+      "loss": 0.1409,
+      "step": 12340
+    },
+    {
+      "grad_norm": 0.49628838896751404,
+      "learning_rate": 3.495069341062836e-05,
+      "loss": 0.1374,
+      "step": 12350
+    },
+    {
+      "grad_norm": 0.5035080313682556,
+      "learning_rate": 3.4871874220158896e-05,
+      "loss": 0.1405,
+      "step": 12360
+    },
+    {
+      "grad_norm": 0.9876883625984192,
+      "learning_rate": 3.479309638940762e-05,
+      "loss": 0.1736,
+      "step": 12370
+    },
+    {
+      "grad_norm": 0.634549081325531,
+      "learning_rate": 3.4714360133750146e-05,
+      "loss": 0.1648,
+      "step": 12380
+    },
+    {
+      "grad_norm": 0.518318772315979,
+      "learning_rate": 3.463566566844839e-05,
+      "loss": 0.1537,
+      "step": 12390
+    },
+    {
+      "grad_norm": 0.5775787234306335,
+      "learning_rate": 3.4557013208650016e-05,
+      "loss": 0.1426,
+      "step": 12400
+    },
+    {
+      "grad_norm": 0.4486111104488373,
+      "learning_rate": 3.4478402969387857e-05,
+      "loss": 0.1518,
+      "step": 12410
+    },
+    {
+      "grad_norm": 0.5648380517959595,
+      "learning_rate": 3.4399835165579266e-05,
+      "loss": 0.1588,
+      "step": 12420
+    },
+    {
+      "grad_norm": 0.46043774485588074,
+      "learning_rate": 3.4321310012025645e-05,
+      "loss": 0.1351,
+      "step": 12430
+    },
+    {
+      "grad_norm": 0.5836649537086487,
+      "learning_rate": 3.424282772341176e-05,
+      "loss": 0.1393,
+      "step": 12440
+    },
+    {
+      "grad_norm": 0.5107713937759399,
+      "learning_rate": 3.416438851430519e-05,
+      "loss": 0.1663,
+      "step": 12450
+    },
+    {
+      "grad_norm": 0.5084676742553711,
+      "learning_rate": 3.408599259915577e-05,
+      "loss": 0.1412,
+      "step": 12460
+    },
+    {
+      "grad_norm": 0.49307137727737427,
+      "learning_rate": 3.400764019229487e-05,
+      "loss": 0.1568,
+      "step": 12470
+    },
+    {
+      "grad_norm": 0.4890282154083252,
+      "learning_rate": 3.3929331507935035e-05,
+      "loss": 0.1485,
+      "step": 12480
+    },
+    {
+      "grad_norm": 0.5722495913505554,
+      "learning_rate": 3.3851066760169196e-05,
+      "loss": 0.1413,
+      "step": 12490
+    },
+    {
+      "grad_norm": 0.5228545069694519,
+      "learning_rate": 3.377284616297021e-05,
+      "loss": 0.1621,
+      "step": 12500
+    },
+    {
+      "grad_norm": 0.5254539847373962,
+      "learning_rate": 3.3694669930190166e-05,
+      "loss": 0.1419,
+      "step": 12510
+    },
+    {
+      "grad_norm": 0.4499201476573944,
+      "learning_rate": 3.36165382755599e-05,
+      "loss": 0.1451,
+      "step": 12520
+    },
+    {
+      "grad_norm": 0.47527462244033813,
+      "learning_rate": 3.35384514126884e-05,
+      "loss": 0.158,
+      "step": 12530
+    },
+    {
+      "grad_norm": 0.4404827654361725,
+      "learning_rate": 3.3460409555062154e-05,
+      "loss": 0.1349,
+      "step": 12540
+    },
+    {
+      "grad_norm": 0.5054007768630981,
+      "learning_rate": 3.3382412916044645e-05,
+      "loss": 0.1528,
+      "step": 12550
+    },
+    {
+      "grad_norm": 0.5412774085998535,
+      "learning_rate": 3.330446170887566e-05,
+      "loss": 0.1357,
+      "step": 12560
+    },
+    {
+      "grad_norm": 0.44708800315856934,
+      "learning_rate": 3.3226556146670834e-05,
+      "loss": 0.1555,
+      "step": 12570
+    },
+    {
+      "grad_norm": 0.31748852133750916,
+      "learning_rate": 3.314869644242102e-05,
+      "loss": 0.1378,
+      "step": 12580
+    },
+    {
+      "grad_norm": 0.42069295048713684,
+      "learning_rate": 3.3070882808991674e-05,
+      "loss": 0.1646,
+      "step": 12590
+    },
+    {
+      "grad_norm": 0.3729683458805084,
+      "learning_rate": 3.2993115459122305e-05,
+      "loss": 0.1449,
+      "step": 12600
+    },
+    {
+      "grad_norm": 0.6514593362808228,
+      "learning_rate": 3.2915394605425835e-05,
+      "loss": 0.1471,
+      "step": 12610
+    },
+    {
+      "grad_norm": 0.620712161064148,
+      "learning_rate": 3.283772046038816e-05,
+      "loss": 0.1592,
+      "step": 12620
+    },
+    {
+      "grad_norm": 0.3733891248703003,
+      "learning_rate": 3.276009323636739e-05,
+      "loss": 0.1473,
+      "step": 12630
+    },
+    {
+      "grad_norm": 0.5199699401855469,
+      "learning_rate": 3.268251314559344e-05,
+      "loss": 0.1459,
+      "step": 12640
+    },
+    {
+      "grad_norm": 0.5111045837402344,
+      "learning_rate": 3.2604980400167254e-05,
+      "loss": 0.1399,
+      "step": 12650
+    },
+    {
+      "grad_norm": 0.4543409049510956,
+      "learning_rate": 3.252749521206042e-05,
+      "loss": 0.1193,
+      "step": 12660
+    },
+    {
+      "grad_norm": 0.34513652324676514,
+      "learning_rate": 3.2450057793114494e-05,
+      "loss": 0.1698,
+      "step": 12670
+    },
+    {
+      "grad_norm": 0.44166332483291626,
+      "learning_rate": 3.2372668355040435e-05,
+      "loss": 0.1449,
+      "step": 12680
+    },
+    {
+      "grad_norm": 0.4303935170173645,
+      "learning_rate": 3.2295327109418005e-05,
+      "loss": 0.1321,
+      "step": 12690
+    },
+    {
+      "grad_norm": 0.8056159615516663,
+      "learning_rate": 3.221803426769518e-05,
+      "loss": 0.1621,
+      "step": 12700
+    },
+    {
+      "grad_norm": 0.39342600107192993,
+      "learning_rate": 3.214079004118768e-05,
+      "loss": 0.1364,
+      "step": 12710
+    },
+    {
+      "grad_norm": 0.45766669511795044,
+      "learning_rate": 3.2063594641078234e-05,
+      "loss": 0.135,
+      "step": 12720
+    },
+    {
+      "grad_norm": 0.5896368622779846,
+      "learning_rate": 3.198644827841616e-05,
+      "loss": 0.1428,
+      "step": 12730
+    },
+    {
+      "grad_norm": 0.6812804341316223,
+      "learning_rate": 3.1909351164116654e-05,
+      "loss": 0.1549,
+      "step": 12740
+    },
+    {
+      "grad_norm": 0.49864867329597473,
+      "learning_rate": 3.183230350896026e-05,
+      "loss": 0.1535,
+      "step": 12750
+    },
+    {
+      "grad_norm": 0.5077038407325745,
+      "learning_rate": 3.1755305523592337e-05,
+      "loss": 0.163,
+      "step": 12760
+    },
+    {
+      "grad_norm": 0.5962294936180115,
+      "learning_rate": 3.167835741852245e-05,
+      "loss": 0.1547,
+      "step": 12770
+    },
+    {
+      "grad_norm": 0.7617970108985901,
+      "learning_rate": 3.160145940412378e-05,
+      "loss": 0.1566,
+      "step": 12780
+    },
+    {
+      "grad_norm": 0.32674744725227356,
+      "learning_rate": 3.1524611690632545e-05,
+      "loss": 0.1292,
+      "step": 12790
+    },
+    {
+      "grad_norm": 0.48363614082336426,
+      "learning_rate": 3.144781448814746e-05,
+      "loss": 0.1586,
+      "step": 12800
+    },
+    {
+      "grad_norm": 0.44200846552848816,
+      "learning_rate": 3.1371068006629145e-05,
+      "loss": 0.1537,
+      "step": 12810
+    },
+    {
+      "grad_norm": 0.45461201667785645,
+      "learning_rate": 3.129437245589956e-05,
+      "loss": 0.1508,
+      "step": 12820
+    },
+    {
+      "grad_norm": 0.40469852089881897,
+      "learning_rate": 3.121772804564143e-05,
+      "loss": 0.1471,
+      "step": 12830
+    },
+    {
+      "grad_norm": 0.44410720467567444,
+      "learning_rate": 3.11411349853976e-05,
+      "loss": 0.1646,
+      "step": 12840
+    },
+    {
+      "grad_norm": 0.4673011898994446,
+      "learning_rate": 3.10645934845706e-05,
+      "loss": 0.1476,
+      "step": 12850
+    },
+    {
+      "grad_norm": 0.6184592843055725,
+      "learning_rate": 3.098810375242196e-05,
+      "loss": 0.1347,
+      "step": 12860
+    },
+    {
+      "grad_norm": 0.5264120697975159,
+      "learning_rate": 3.0911665998071704e-05,
+      "loss": 0.1466,
+      "step": 12870
+    },
+    {
+      "grad_norm": 0.5913290977478027,
+      "learning_rate": 3.083528043049774e-05,
+      "loss": 0.1454,
+      "step": 12880
+    },
+    {
+      "grad_norm": 0.4904285967350006,
+      "learning_rate": 3.0758947258535255e-05,
+      "loss": 0.1449,
+      "step": 12890
+    },
+    {
+      "grad_norm": 0.5076789855957031,
+      "learning_rate": 3.068266669087625e-05,
+      "loss": 0.1488,
+      "step": 12900
+    },
+    {
+      "grad_norm": 0.4661986231803894,
+      "learning_rate": 3.060643893606887e-05,
+      "loss": 0.1373,
+      "step": 12910
+    },
+    {
+      "grad_norm": 0.6898033022880554,
+      "learning_rate": 3.053026420251693e-05,
+      "loss": 0.1628,
+      "step": 12920
+    },
+    {
+      "grad_norm": 0.5707530379295349,
+      "learning_rate": 3.0454142698479183e-05,
+      "loss": 0.1309,
+      "step": 12930
+    },
+    {
+      "grad_norm": 0.6992397904396057,
+      "learning_rate": 3.0378074632068954e-05,
+      "loss": 0.1405,
+      "step": 12940
+    },
+    {
+      "grad_norm": 0.33566418290138245,
+      "learning_rate": 3.0302060211253408e-05,
+      "loss": 0.1417,
+      "step": 12950
+    },
+    {
+      "grad_norm": 0.5328617095947266,
+      "learning_rate": 3.0226099643853073e-05,
+      "loss": 0.1374,
+      "step": 12960
+    },
+    {
+      "grad_norm": 0.4831804633140564,
+      "learning_rate": 3.0150193137541283e-05,
+      "loss": 0.1452,
+      "step": 12970
+    },
+    {
+      "grad_norm": 0.38728657364845276,
+      "learning_rate": 3.0074340899843467e-05,
+      "loss": 0.1392,
+      "step": 12980
+    },
+    {
+      "grad_norm": 0.5172281861305237,
+      "learning_rate": 2.999854313813677e-05,
+      "loss": 0.1574,
+      "step": 12990
+    },
+    {
+      "grad_norm": 0.5316351056098938,
+      "learning_rate": 2.9922800059649382e-05,
+      "loss": 0.1504,
+      "step": 13000
+    },
+    {
+      "grad_norm": 0.5502157211303711,
+      "learning_rate": 2.9847111871459976e-05,
+      "loss": 0.1335,
+      "step": 13010
+    },
+    {
+      "grad_norm": 0.5915395617485046,
+      "learning_rate": 2.977147878049721e-05,
+      "loss": 0.1203,
+      "step": 13020
+    },
+    {
+      "grad_norm": 0.44677820801734924,
+      "learning_rate": 2.9695900993539006e-05,
+      "loss": 0.1248,
+      "step": 13030
+    },
+    {
+      "grad_norm": 0.43962833285331726,
+      "learning_rate": 2.9620378717212183e-05,
+      "loss": 0.1474,
+      "step": 13040
+    },
+    {
+      "grad_norm": 0.49852675199508667,
+      "learning_rate": 2.9544912157991745e-05,
+      "loss": 0.1393,
+      "step": 13050
+    },
+    {
+      "grad_norm": 0.37782490253448486,
+      "learning_rate": 2.9469501522200405e-05,
+      "loss": 0.1344,
+      "step": 13060
+    },
+    {
+      "grad_norm": 0.668069064617157,
+      "learning_rate": 2.9394147016007946e-05,
+      "loss": 0.1541,
+      "step": 13070
+    },
+    {
+      "grad_norm": 0.7251052260398865,
+      "learning_rate": 2.9318848845430702e-05,
+      "loss": 0.1373,
+      "step": 13080
+    },
+    {
+      "grad_norm": 0.6519511938095093,
+      "learning_rate": 2.9243607216331013e-05,
+      "loss": 0.1516,
+      "step": 13090
+    },
+    {
+      "grad_norm": 0.4199202358722687,
+      "learning_rate": 2.916842233441661e-05,
+      "loss": 0.1595,
+      "step": 13100
+    },
+    {
+      "grad_norm": 0.8728315830230713,
+      "learning_rate": 2.90932944052401e-05,
+      "loss": 0.1607,
+      "step": 13110
+    },
+    {
+      "grad_norm": 0.5271692872047424,
+      "learning_rate": 2.9018223634198354e-05,
+      "loss": 0.1647,
+      "step": 13120
+    },
+    {
+      "grad_norm": 0.43006017804145813,
+      "learning_rate": 2.8943210226532025e-05,
+      "loss": 0.1426,
+      "step": 13130
+    },
+    {
+      "grad_norm": 0.46635010838508606,
+      "learning_rate": 2.8868254387324857e-05,
+      "loss": 0.1501,
+      "step": 13140
+    },
+    {
+      "grad_norm": 0.37150195240974426,
+      "learning_rate": 2.8793356321503306e-05,
+      "loss": 0.1301,
+      "step": 13150
+    },
+    {
+      "grad_norm": 0.34532731771469116,
+      "learning_rate": 2.87185162338358e-05,
+      "loss": 0.1375,
+      "step": 13160
+    },
+    {
+      "grad_norm": 0.5564374327659607,
+      "learning_rate": 2.8643734328932253e-05,
+      "loss": 0.1619,
+      "step": 13170
+    },
+    {
+      "grad_norm": 0.5679888129234314,
+      "learning_rate": 2.856901081124359e-05,
+      "loss": 0.1462,
+      "step": 13180
+    },
+    {
+      "grad_norm": 0.5806342363357544,
+      "learning_rate": 2.8494345885061002e-05,
+      "loss": 0.1499,
+      "step": 13190
+    },
+    {
+      "grad_norm": 0.4439584016799927,
+      "learning_rate": 2.8419739754515616e-05,
+      "loss": 0.145,
+      "step": 13200
+    },
+    {
+      "grad_norm": 0.5426590442657471,
+      "learning_rate": 2.8345192623577666e-05,
+      "loss": 0.1412,
+      "step": 13210
+    },
+    {
+      "grad_norm": 0.4567379355430603,
+      "learning_rate": 2.8270704696056193e-05,
+      "loss": 0.1447,
+      "step": 13220
+    },
+    {
+      "grad_norm": 0.7855887413024902,
+      "learning_rate": 2.8196276175598367e-05,
+      "loss": 0.1598,
+      "step": 13230
+    },
+    {
+      "grad_norm": 0.6222705841064453,
+      "learning_rate": 2.8121907265688884e-05,
+      "loss": 0.1507,
+      "step": 13240
+    },
+    {
+      "grad_norm": 0.4166463315486908,
+      "learning_rate": 2.804759816964957e-05,
+      "loss": 0.1385,
+      "step": 13250
+    },
+    {
+      "grad_norm": 0.4524863660335541,
+      "learning_rate": 2.797334909063857e-05,
+      "loss": 0.1445,
+      "step": 13260
+    },
+    {
+      "grad_norm": 0.4537992775440216,
+      "learning_rate": 2.7899160231650056e-05,
+      "loss": 0.1366,
+      "step": 13270
+    },
+    {
+      "grad_norm": 0.8089361786842346,
+      "learning_rate": 2.7825031795513585e-05,
+      "loss": 0.1462,
+      "step": 13280
+    },
+    {
+      "grad_norm": 0.41903209686279297,
+      "learning_rate": 2.775096398489341e-05,
+      "loss": 0.1464,
+      "step": 13290
+    },
+    {
+      "grad_norm": 0.5523425340652466,
+      "learning_rate": 2.7676957002288163e-05,
+      "loss": 0.1435,
+      "step": 13300
+    },
+    {
+      "grad_norm": 0.42928025126457214,
+      "learning_rate": 2.760301105003003e-05,
+      "loss": 0.1396,
+      "step": 13310
+    },
+    {
+      "grad_norm": 0.4648435711860657,
+      "learning_rate": 2.752912633028446e-05,
+      "loss": 0.1259,
+      "step": 13320
+    },
+    {
+      "grad_norm": 0.5977752208709717,
+      "learning_rate": 2.7455303045049474e-05,
+      "loss": 0.1435,
+      "step": 13330
+    },
+    {
+      "grad_norm": 0.6296465396881104,
+      "learning_rate": 2.7381541396155098e-05,
+      "loss": 0.1605,
+      "step": 13340
+    },
+    {
+      "grad_norm": 0.2933214604854584,
+      "learning_rate": 2.730784158526286e-05,
+      "loss": 0.1371,
+      "step": 13350
+    },
+    {
+      "grad_norm": 0.5645100474357605,
+      "learning_rate": 2.723420381386521e-05,
+      "loss": 0.1191,
+      "step": 13360
+    },
+    {
+      "grad_norm": 0.3983377516269684,
+      "learning_rate": 2.7160628283285018e-05,
+      "loss": 0.1345,
+      "step": 13370
+    },
+    {
+      "grad_norm": 1.033931851387024,
+      "learning_rate": 2.7087115194675007e-05,
+      "loss": 0.1737,
+      "step": 13380
+    },
+    {
+      "grad_norm": 0.32144275307655334,
+      "learning_rate": 2.701366474901712e-05,
+      "loss": 0.1331,
+      "step": 13390
+    },
+    {
+      "grad_norm": 0.6477856636047363,
+      "learning_rate": 2.6940277147122085e-05,
+      "loss": 0.1376,
+      "step": 13400
+    },
+    {
+      "grad_norm": 0.6014312505722046,
+      "learning_rate": 2.686695258962878e-05,
+      "loss": 0.1445,
+      "step": 13410
+    },
+    {
+      "grad_norm": 0.4717272222042084,
+      "learning_rate": 2.679369127700375e-05,
+      "loss": 0.1279,
+      "step": 13420
+    },
+    {
+      "grad_norm": 0.47569066286087036,
+      "learning_rate": 2.672049340954067e-05,
+      "loss": 0.1512,
+      "step": 13430
+    },
+    {
+      "grad_norm": 0.6994011998176575,
+      "learning_rate": 2.6647359187359676e-05,
+      "loss": 0.1496,
+      "step": 13440
+    },
+    {
+      "grad_norm": 0.8393667340278625,
+      "learning_rate": 2.6574288810406946e-05,
+      "loss": 0.1561,
+      "step": 13450
+    },
+    {
+      "grad_norm": 0.4607236683368683,
+      "learning_rate": 2.6501282478454083e-05,
+      "loss": 0.1316,
+      "step": 13460
+    },
+    {
+      "grad_norm": 0.5732179880142212,
+      "learning_rate": 2.6428340391097618e-05,
+      "loss": 0.1444,
+      "step": 13470
+    },
+    {
+      "grad_norm": 0.6297001242637634,
+      "learning_rate": 2.6355462747758485e-05,
+      "loss": 0.1304,
+      "step": 13480
+    },
+    {
+      "grad_norm": 0.45855289697647095,
+      "learning_rate": 2.6282649747681304e-05,
+      "loss": 0.1606,
+      "step": 13490
+    },
+    {
+      "grad_norm": 0.42114052176475525,
+      "learning_rate": 2.620990158993406e-05,
+      "loss": 0.1376,
+      "step": 13500
+    },
+    {
+      "grad_norm": 0.3901847004890442,
+      "learning_rate": 2.6137218473407477e-05,
+      "loss": 0.1292,
+      "step": 13510
+    },
+    {
+      "grad_norm": 0.40595537424087524,
+      "learning_rate": 2.606460059681436e-05,
+      "loss": 0.1412,
+      "step": 13520
+    },
+    {
+      "grad_norm": 0.5803327560424805,
+      "learning_rate": 2.599204815868928e-05,
+      "loss": 0.142,
+      "step": 13530
+    },
+    {
+      "grad_norm": 0.4926631450653076,
+      "learning_rate": 2.5919561357387756e-05,
+      "loss": 0.147,
+      "step": 13540
+    },
+    {
+      "grad_norm": 0.4088830053806305,
+      "learning_rate": 2.5847140391085972e-05,
+      "loss": 0.1285,
+      "step": 13550
+    },
+    {
+      "grad_norm": 0.4611234664916992,
+      "learning_rate": 2.5774785457780103e-05,
+      "loss": 0.1284,
+      "step": 13560
+    },
+    {
+      "grad_norm": 0.5129578709602356,
+      "learning_rate": 2.5702496755285753e-05,
+      "loss": 0.1352,
+      "step": 13570
+    },
+    {
+      "grad_norm": 0.43149107694625854,
+      "learning_rate": 2.5630274481237483e-05,
+      "loss": 0.1458,
+      "step": 13580
+    },
+    {
+      "grad_norm": 0.4427432417869568,
+      "learning_rate": 2.5558118833088197e-05,
+      "loss": 0.1444,
+      "step": 13590
+    },
+    {
+      "grad_norm": 0.3868066370487213,
+      "learning_rate": 2.548603000810872e-05,
+      "loss": 0.1302,
+      "step": 13600
+    },
+    {
+      "grad_norm": 0.645939826965332,
+      "learning_rate": 2.5414008203387152e-05,
+      "loss": 0.1429,
+      "step": 13610
+    },
+    {
+      "grad_norm": 0.3478547930717468,
+      "learning_rate": 2.534205361582834e-05,
+      "loss": 0.1389,
+      "step": 13620
+    },
+    {
+      "grad_norm": 0.32033225893974304,
+      "learning_rate": 2.527016644215338e-05,
+      "loss": 0.123,
+      "step": 13630
+    },
+    {
+      "grad_norm": 0.47429198026657104,
+      "learning_rate": 2.519834687889905e-05,
+      "loss": 0.159,
+      "step": 13640
+    },
+    {
+      "grad_norm": 0.683470606803894,
+      "learning_rate": 2.5126595122417295e-05,
+      "loss": 0.1464,
+      "step": 13650
+    },
+    {
+      "grad_norm": 0.6066814064979553,
+      "learning_rate": 2.5054911368874713e-05,
+      "loss": 0.1336,
+      "step": 13660
+    },
+    {
+      "grad_norm": 0.5541272759437561,
+      "learning_rate": 2.4983295814251916e-05,
+      "loss": 0.1468,
+      "step": 13670
+    },
+    {
+      "grad_norm": 0.6147645711898804,
+      "learning_rate": 2.4911748654343105e-05,
+      "loss": 0.134,
+      "step": 13680
+    },
+    {
+      "grad_norm": 0.5629763603210449,
+      "learning_rate": 2.4840270084755463e-05,
+      "loss": 0.1453,
+      "step": 13690
+    },
+    {
+      "grad_norm": 0.8916916847229004,
+      "learning_rate": 2.4768860300908685e-05,
+      "loss": 0.144,
+      "step": 13700
+    },
+    {
+      "grad_norm": 0.4556906819343567,
+      "learning_rate": 2.469751949803443e-05,
+      "loss": 0.148,
+      "step": 13710
+    },
+    {
+      "grad_norm": 0.5079695582389832,
+      "learning_rate": 2.4626247871175666e-05,
+      "loss": 0.1605,
+      "step": 13720
+    },
+    {
+      "grad_norm": 0.6419221758842468,
+      "learning_rate": 2.4555045615186346e-05,
+      "loss": 0.1557,
+      "step": 13730
+    },
+    {
+      "grad_norm": 0.3996198773384094,
+      "learning_rate": 2.4483912924730677e-05,
+      "loss": 0.1334,
+      "step": 13740
+    },
+    {
+      "grad_norm": 0.5776199698448181,
+      "learning_rate": 2.4412849994282742e-05,
+      "loss": 0.1378,
+      "step": 13750
+    },
+    {
+      "grad_norm": 0.40003812313079834,
+      "learning_rate": 2.434185701812592e-05,
+      "loss": 0.1355,
+      "step": 13760
+    },
+    {
+      "grad_norm": 0.4128742814064026,
+      "learning_rate": 2.4270934190352218e-05,
+      "loss": 0.1418,
+      "step": 13770
+    },
+    {
+      "grad_norm": 0.5939120054244995,
+      "learning_rate": 2.4200081704861998e-05,
+      "loss": 0.1388,
+      "step": 13780
+    },
+    {
+      "grad_norm": 0.7074214816093445,
+      "learning_rate": 2.412929975536321e-05,
+      "loss": 0.1375,
+      "step": 13790
+    },
+    {
+      "grad_norm": 0.458082914352417,
+      "learning_rate": 2.4058588535371017e-05,
+      "loss": 0.1383,
+      "step": 13800
+    },
+    {
+      "grad_norm": 0.617110550403595,
+      "learning_rate": 2.3987948238207243e-05,
+      "loss": 0.1559,
+      "step": 13810
+    },
+    {
+      "grad_norm": 0.45726099610328674,
+      "learning_rate": 2.3917379056999678e-05,
+      "loss": 0.1381,
+      "step": 13820
+    },
+    {
+      "grad_norm": 0.4458962082862854,
+      "learning_rate": 2.3846881184681824e-05,
+      "loss": 0.1368,
+      "step": 13830
+    },
+    {
+      "grad_norm": 0.468458354473114,
+      "learning_rate": 2.377645481399214e-05,
+      "loss": 0.147,
+      "step": 13840
+    },
+    {
+      "grad_norm": 0.5195881724357605,
+      "learning_rate": 2.3706100137473667e-05,
+      "loss": 0.1657,
+      "step": 13850
+    },
+    {
+      "grad_norm": 0.4110904335975647,
+      "learning_rate": 2.3635817347473394e-05,
+      "loss": 0.1425,
+      "step": 13860
+    },
+    {
+      "grad_norm": 0.4793466031551361,
+      "learning_rate": 2.3565606636141757e-05,
+      "loss": 0.1399,
+      "step": 13870
+    },
+    {
+      "grad_norm": 0.5409266948699951,
+      "learning_rate": 2.3495468195432203e-05,
+      "loss": 0.1452,
+      "step": 13880
+    },
+    {
+      "grad_norm": 0.453700989484787,
+      "learning_rate": 2.3425402217100507e-05,
+      "loss": 0.1476,
+      "step": 13890
+    },
+    {
+      "grad_norm": 0.45534783601760864,
+      "learning_rate": 2.3355408892704424e-05,
+      "loss": 0.1459,
+      "step": 13900
+    },
+    {
+      "grad_norm": 0.4195924699306488,
+      "learning_rate": 2.3285488413603003e-05,
+      "loss": 0.1388,
+      "step": 13910
+    },
+    {
+      "grad_norm": 0.4560341536998749,
+      "learning_rate": 2.321564097095615e-05,
+      "loss": 0.1771,
+      "step": 13920
+    },
+    {
+      "grad_norm": 0.49362289905548096,
+      "learning_rate": 2.3145866755724142e-05,
+      "loss": 0.1382,
+      "step": 13930
+    },
+    {
+      "grad_norm": 0.4538297653198242,
+      "learning_rate": 2.307616595866699e-05,
+      "loss": 0.1612,
+      "step": 13940
+    },
+    {
+      "grad_norm": 0.6492700576782227,
+      "learning_rate": 2.3006538770344032e-05,
+      "loss": 0.137,
+      "step": 13950
+    },
+    {
+      "grad_norm": 0.46616366505622864,
+      "learning_rate": 2.293698538111334e-05,
+      "loss": 0.1358,
+      "step": 13960
+    },
+    {
+      "grad_norm": 0.5867716073989868,
+      "learning_rate": 2.28675059811312e-05,
+      "loss": 0.1347,
+      "step": 13970
+    },
+    {
+      "grad_norm": 0.620273768901825,
+      "learning_rate": 2.279810076035167e-05,
+      "loss": 0.1601,
+      "step": 13980
+    },
+    {
+      "grad_norm": 0.47465258836746216,
+      "learning_rate": 2.272876990852596e-05,
+      "loss": 0.1375,
+      "step": 13990
+    },
+    {
+      "grad_norm": 1.0628660917282104,
+      "learning_rate": 2.265951361520195e-05,
+      "loss": 0.1684,
+      "step": 14000
+    },
+    {
+      "grad_norm": 0.7413018941879272,
+      "learning_rate": 2.2590332069723748e-05,
+      "loss": 0.1477,
+      "step": 14010
+    },
+    {
+      "grad_norm": 0.43072959780693054,
+      "learning_rate": 2.2521225461231004e-05,
+      "loss": 0.1442,
+      "step": 14020
+    },
+    {
+      "grad_norm": 0.47115612030029297,
+      "learning_rate": 2.2452193978658597e-05,
+      "loss": 0.1493,
+      "step": 14030
+    },
+    {
+      "grad_norm": 0.5778260827064514,
+      "learning_rate": 2.238323781073594e-05,
+      "loss": 0.1398,
+      "step": 14040
+    },
+    {
+      "grad_norm": 0.7411498427391052,
+      "learning_rate": 2.2314357145986552e-05,
+      "loss": 0.1452,
+      "step": 14050
+    },
+    {
+      "grad_norm": 0.4282691180706024,
+      "learning_rate": 2.224555217272757e-05,
+      "loss": 0.1568,
+      "step": 14060
+    },
+    {
+      "grad_norm": 0.6388832926750183,
+      "learning_rate": 2.2176823079069127e-05,
+      "loss": 0.1532,
+      "step": 14070
+    },
+    {
+      "grad_norm": 0.40495604276657104,
+      "learning_rate": 2.210817005291398e-05,
+      "loss": 0.1556,
+      "step": 14080
+    },
+    {
+      "grad_norm": 0.41548284888267517,
+      "learning_rate": 2.203959328195686e-05,
+      "loss": 0.1416,
+      "step": 14090
+    },
+    {
+      "grad_norm": 0.37998533248901367,
+      "learning_rate": 2.1971092953684026e-05,
+      "loss": 0.1343,
+      "step": 14100
+    },
+    {
+      "grad_norm": 0.7624937891960144,
+      "learning_rate": 2.1902669255372788e-05,
+      "loss": 0.1409,
+      "step": 14110
+    },
+    {
+      "grad_norm": 0.6046563982963562,
+      "learning_rate": 2.1834322374090897e-05,
+      "loss": 0.1355,
+      "step": 14120
+    },
+    {
+      "grad_norm": 0.4905014634132385,
+      "learning_rate": 2.1766052496696153e-05,
+      "loss": 0.1578,
+      "step": 14130
+    },
+    {
+      "grad_norm": 0.5940402150154114,
+      "learning_rate": 2.169785980983577e-05,
+      "loss": 0.1456,
+      "step": 14140
+    },
+    {
+      "grad_norm": 0.6443553566932678,
+      "learning_rate": 2.162974449994593e-05,
+      "loss": 0.1421,
+      "step": 14150
+    },
+    {
+      "grad_norm": 0.4725497364997864,
+      "learning_rate": 2.1561706753251337e-05,
+      "loss": 0.1386,
+      "step": 14160
+    },
+    {
+      "grad_norm": 0.4976433217525482,
+      "learning_rate": 2.1493746755764544e-05,
+      "loss": 0.1359,
+      "step": 14170
+    },
+    {
+      "grad_norm": 0.4799959361553192,
+      "learning_rate": 2.1425864693285635e-05,
+      "loss": 0.1532,
+      "step": 14180
+    },
+    {
+      "grad_norm": 0.5584168434143066,
+      "learning_rate": 2.1358060751401547e-05,
+      "loss": 0.1522,
+      "step": 14190
+    },
+    {
+      "grad_norm": 0.430121511220932,
+      "learning_rate": 2.129033511548566e-05,
+      "loss": 0.148,
+      "step": 14200
+    },
+    {
+      "grad_norm": 0.6414793133735657,
+      "learning_rate": 2.1222687970697315e-05,
+      "loss": 0.1588,
+      "step": 14210
+    },
+    {
+      "grad_norm": 0.5209755897521973,
+      "learning_rate": 2.1155119501981173e-05,
+      "loss": 0.142,
+      "step": 14220
+    },
+    {
+      "grad_norm": 0.608585774898529,
+      "learning_rate": 2.1087629894066895e-05,
+      "loss": 0.1634,
+      "step": 14230
+    },
+    {
+      "grad_norm": 0.45329952239990234,
+      "learning_rate": 2.1020219331468473e-05,
+      "loss": 0.133,
+      "step": 14240
+    },
+    {
+      "grad_norm": 0.6077437400817871,
+      "learning_rate": 2.095288799848379e-05,
+      "loss": 0.151,
+      "step": 14250
+    },
+    {
+      "grad_norm": 0.46086400747299194,
+      "learning_rate": 2.088563607919417e-05,
+      "loss": 0.1549,
+      "step": 14260
+    },
+    {
+      "grad_norm": 0.5258955955505371,
+      "learning_rate": 2.0818463757463786e-05,
+      "loss": 0.1368,
+      "step": 14270
+    },
+    {
+      "grad_norm": 0.5024899840354919,
+      "learning_rate": 2.0751371216939175e-05,
+      "loss": 0.1534,
+      "step": 14280
+    },
+    {
+      "grad_norm": 0.49977365136146545,
+      "learning_rate": 2.068435864104882e-05,
+      "loss": 0.145,
+      "step": 14290
+    },
+    {
+      "grad_norm": 0.8212709426879883,
+      "learning_rate": 2.0617426213002506e-05,
+      "loss": 0.1554,
+      "step": 14300
+    },
+    {
+      "grad_norm": 0.7351389527320862,
+      "learning_rate": 2.055057411579097e-05,
+      "loss": 0.1493,
+      "step": 14310
+    },
+    {
+      "grad_norm": 0.4591904878616333,
+      "learning_rate": 2.0483802532185286e-05,
+      "loss": 0.1508,
+      "step": 14320
+    },
+    {
+      "grad_norm": 0.7639979720115662,
+      "learning_rate": 2.041711164473638e-05,
+      "loss": 0.1501,
+      "step": 14330
+    },
+    {
+      "grad_norm": 0.578839123249054,
+      "learning_rate": 2.0350501635774637e-05,
+      "loss": 0.1621,
+      "step": 14340
+    },
+    {
+      "grad_norm": 0.7311603426933289,
+      "learning_rate": 2.0283972687409247e-05,
+      "loss": 0.1431,
+      "step": 14350
+    },
+    {
+      "grad_norm": 0.404704213142395,
+      "learning_rate": 2.021752498152784e-05,
+      "loss": 0.1541,
+      "step": 14360
+    },
+    {
+      "grad_norm": 0.5350682139396667,
+      "learning_rate": 2.015115869979589e-05,
+      "loss": 0.148,
+      "step": 14370
+    },
+    {
+      "grad_norm": 0.6514018177986145,
+      "learning_rate": 2.0084874023656265e-05,
+      "loss": 0.1379,
+      "step": 14380
+    },
+    {
+      "grad_norm": 0.5937101244926453,
+      "learning_rate": 2.001867113432877e-05,
+      "loss": 0.1423,
+      "step": 14390
+    },
+    {
+      "grad_norm": 0.5568202137947083,
+      "learning_rate": 1.995255021280954e-05,
+      "loss": 0.1226,
+      "step": 14400
+    },
+    {
+      "grad_norm": 0.366691529750824,
+      "learning_rate": 1.9886511439870688e-05,
+      "loss": 0.1347,
+      "step": 14410
+    },
+    {
+      "grad_norm": 0.5125223994255066,
+      "learning_rate": 1.9820554996059675e-05,
+      "loss": 0.1608,
+      "step": 14420
+    },
+    {
+      "grad_norm": 0.5805913209915161,
+      "learning_rate": 1.9754681061698893e-05,
+      "loss": 0.1566,
+      "step": 14430
+    },
+    {
+      "grad_norm": 0.5162914991378784,
+      "learning_rate": 1.9688889816885185e-05,
+      "loss": 0.1642,
+      "step": 14440
+    },
+    {
+      "grad_norm": 0.4152345359325409,
+      "learning_rate": 1.962318144148928e-05,
+      "loss": 0.1391,
+      "step": 14450
+    },
+    {
+      "grad_norm": 0.48711293935775757,
+      "learning_rate": 1.955755611515539e-05,
+      "loss": 0.1502,
+      "step": 14460
+    },
+    {
+      "grad_norm": 0.3295830488204956,
+      "learning_rate": 1.9492014017300642e-05,
+      "loss": 0.1381,
+      "step": 14470
+    },
+    {
+      "grad_norm": 0.3542550504207611,
+      "learning_rate": 1.942655532711461e-05,
+      "loss": 0.1511,
+      "step": 14480
+    },
+    {
+      "grad_norm": 0.42618438601493835,
+      "learning_rate": 1.9361180223558882e-05,
+      "loss": 0.133,
+      "step": 14490
+    },
+    {
+      "grad_norm": 0.5635879635810852,
+      "learning_rate": 1.929588888536647e-05,
+      "loss": 0.1512,
+      "step": 14500
+    },
+    {
+      "grad_norm": 0.525734007358551,
+      "learning_rate": 1.9230681491041425e-05,
+      "loss": 0.1395,
+      "step": 14510
+    },
+    {
+      "grad_norm": 0.6089289784431458,
+      "learning_rate": 1.9165558218858264e-05,
+      "loss": 0.1576,
+      "step": 14520
+    },
+    {
+      "grad_norm": 0.5419020056724548,
+      "learning_rate": 1.9100519246861505e-05,
+      "loss": 0.1241,
+      "step": 14530
+    },
+    {
+      "grad_norm": 0.6943102478981018,
+      "learning_rate": 1.9035564752865248e-05,
+      "loss": 0.154,
+      "step": 14540
+    },
+    {
+      "grad_norm": 0.5433569550514221,
+      "learning_rate": 1.897069491445258e-05,
+      "loss": 0.1629,
+      "step": 14550
+    },
+    {
+      "grad_norm": 0.5426835417747498,
+      "learning_rate": 1.890590990897515e-05,
+      "loss": 0.1458,
+      "step": 14560
+    },
+    {
+      "grad_norm": 0.37228840589523315,
+      "learning_rate": 1.884120991355272e-05,
+      "loss": 0.1458,
+      "step": 14570
+    },
+    {
+      "grad_norm": 0.5086845755577087,
+      "learning_rate": 1.8776595105072576e-05,
+      "loss": 0.1369,
+      "step": 14580
+    },
+    {
+      "grad_norm": 0.6947071552276611,
+      "learning_rate": 1.8712065660189166e-05,
+      "loss": 0.1544,
+      "step": 14590
+    },
+    {
+      "grad_norm": 0.4524993300437927,
+      "learning_rate": 1.8647621755323513e-05,
+      "loss": 0.1386,
+      "step": 14600
+    },
+    {
+      "grad_norm": 0.39710119366645813,
+      "learning_rate": 1.858326356666278e-05,
+      "loss": 0.1567,
+      "step": 14610
+    },
+    {
+      "grad_norm": 0.49423453211784363,
+      "learning_rate": 1.851899127015983e-05,
+      "loss": 0.1454,
+      "step": 14620
+    },
+    {
+      "grad_norm": 0.48335862159729004,
+      "learning_rate": 1.8454805041532626e-05,
+      "loss": 0.1488,
+      "step": 14630
+    },
+    {
+      "grad_norm": 0.45463165640830994,
+      "learning_rate": 1.8390705056263906e-05,
+      "loss": 0.1621,
+      "step": 14640
+    },
+    {
+      "grad_norm": 0.44413816928863525,
+      "learning_rate": 1.832669148960057e-05,
+      "loss": 0.1411,
+      "step": 14650
+    },
+    {
+      "grad_norm": 0.4332204759120941,
+      "learning_rate": 1.8262764516553233e-05,
+      "loss": 0.1316,
+      "step": 14660
+    },
+    {
+      "grad_norm": 0.5654910802841187,
+      "learning_rate": 1.8198924311895843e-05,
+      "loss": 0.1523,
+      "step": 14670
+    },
+    {
+      "grad_norm": 0.44807732105255127,
+      "learning_rate": 1.813517105016505e-05,
+      "loss": 0.1538,
+      "step": 14680
+    },
+    {
+      "grad_norm": 0.44858354330062866,
+      "learning_rate": 1.8071504905659888e-05,
+      "loss": 0.154,
+      "step": 14690
+    },
+    {
+      "grad_norm": 0.6337711215019226,
+      "learning_rate": 1.800792605244109e-05,
+      "loss": 0.1379,
+      "step": 14700
+    },
+    {
+      "grad_norm": 0.4808521866798401,
+      "learning_rate": 1.7944434664330844e-05,
+      "loss": 0.1468,
+      "step": 14710
+    },
+    {
+      "grad_norm": 0.3633001446723938,
+      "learning_rate": 1.7881030914912212e-05,
+      "loss": 0.147,
+      "step": 14720
+    },
+    {
+      "grad_norm": 0.6506484150886536,
+      "learning_rate": 1.7817714977528577e-05,
+      "loss": 0.1398,
+      "step": 14730
+    },
+    {
+      "grad_norm": 0.5819616317749023,
+      "learning_rate": 1.7754487025283332e-05,
+      "loss": 0.1535,
+      "step": 14740
+    },
+    {
+      "grad_norm": 0.581115186214447,
+      "learning_rate": 1.7691347231039275e-05,
+      "loss": 0.1654,
+      "step": 14750
+    },
+    {
+      "grad_norm": 0.597812294960022,
+      "learning_rate": 1.7628295767418164e-05,
+      "loss": 0.1298,
+      "step": 14760
+    },
+    {
+      "grad_norm": 0.4996192753314972,
+      "learning_rate": 1.7565332806800333e-05,
+      "loss": 0.152,
+      "step": 14770
+    },
+    {
+      "grad_norm": 0.6354101300239563,
+      "learning_rate": 1.750245852132408e-05,
+      "loss": 0.1414,
+      "step": 14780
+    },
+    {
+      "grad_norm": 0.42595967650413513,
+      "learning_rate": 1.7439673082885323e-05,
+      "loss": 0.1322,
+      "step": 14790
+    },
+    {
+      "grad_norm": 0.5699095726013184,
+      "learning_rate": 1.7376976663137047e-05,
+      "loss": 0.1441,
+      "step": 14800
+    },
+    {
+      "grad_norm": 0.7154938578605652,
+      "learning_rate": 1.7314369433488853e-05,
+      "loss": 0.1423,
+      "step": 14810
+    },
+    {
+      "grad_norm": 0.34929409623146057,
+      "learning_rate": 1.7251851565106548e-05,
+      "loss": 0.1295,
+      "step": 14820
+    },
+    {
+      "grad_norm": 0.3786975145339966,
+      "learning_rate": 1.7189423228911574e-05,
+      "loss": 0.1382,
+      "step": 14830
+    },
+    {
+      "grad_norm": 0.6986491084098816,
+      "learning_rate": 1.7127084595580606e-05,
+      "loss": 0.1474,
+      "step": 14840
+    },
+    {
+      "grad_norm": 0.20330744981765747,
+      "learning_rate": 1.706483583554513e-05,
+      "loss": 0.1198,
+      "step": 14850
+    },
+    {
+      "grad_norm": 0.49908575415611267,
+      "learning_rate": 1.700267711899083e-05,
+      "loss": 0.1491,
+      "step": 14860
+    },
+    {
+      "grad_norm": 0.3722893297672272,
+      "learning_rate": 1.69406086158573e-05,
+      "loss": 0.1277,
+      "step": 14870
+    },
+    {
+      "grad_norm": 0.3950440287590027,
+      "learning_rate": 1.6878630495837455e-05,
+      "loss": 0.13,
+      "step": 14880
+    },
+    {
+      "grad_norm": 0.7562043070793152,
+      "learning_rate": 1.681674292837707e-05,
+      "loss": 0.1508,
+      "step": 14890
+    },
+    {
+      "grad_norm": 0.36514854431152344,
+      "learning_rate": 1.6754946082674444e-05,
+      "loss": 0.1366,
+      "step": 14900
+    },
+    {
+      "grad_norm": 0.5401731133460999,
+      "learning_rate": 1.6693240127679748e-05,
+      "loss": 0.133,
+      "step": 14910
+    },
+    {
+      "grad_norm": 0.4642062187194824,
+      "learning_rate": 1.663162523209475e-05,
+      "loss": 0.1491,
+      "step": 14920
+    },
+    {
+      "grad_norm": 0.541653037071228,
+      "learning_rate": 1.6570101564372193e-05,
+      "loss": 0.138,
+      "step": 14930
+    },
+    {
+      "grad_norm": 0.32297763228416443,
+      "learning_rate": 1.650866929271543e-05,
+      "loss": 0.1248,
+      "step": 14940
+    },
+    {
+      "grad_norm": 0.46302124857902527,
+      "learning_rate": 1.644732858507797e-05,
+      "loss": 0.143,
+      "step": 14950
+    },
+    {
+      "grad_norm": 0.5050262808799744,
+      "learning_rate": 1.6386079609162943e-05,
+      "loss": 0.1665,
+      "step": 14960
+    },
+    {
+      "grad_norm": 0.6757389903068542,
+      "learning_rate": 1.6324922532422742e-05,
+      "loss": 0.1581,
+      "step": 14970
+    },
+    {
+      "grad_norm": 0.4199475347995758,
+      "learning_rate": 1.6263857522058434e-05,
+      "loss": 0.1431,
+      "step": 14980
+    },
+    {
+      "grad_norm": 0.4102480709552765,
+      "learning_rate": 1.6202884745019443e-05,
+      "loss": 0.1308,
+      "step": 14990
+    },
+    {
+      "grad_norm": 0.548195481300354,
+      "learning_rate": 1.614200436800304e-05,
+      "loss": 0.153,
+      "step": 15000
+    },
+    {
+      "grad_norm": 0.6598746180534363,
+      "learning_rate": 1.6081216557453814e-05,
+      "loss": 0.1406,
+      "step": 15010
+    },
+    {
+      "grad_norm": 0.38730570673942566,
+      "learning_rate": 1.6020521479563367e-05,
+      "loss": 0.1311,
+      "step": 15020
+    },
+    {
+      "grad_norm": 0.47192880511283875,
+      "learning_rate": 1.5959919300269654e-05,
+      "loss": 0.1407,
+      "step": 15030
+    },
+    {
+      "grad_norm": 0.46144768595695496,
+      "learning_rate": 1.5899410185256764e-05,
+      "loss": 0.133,
+      "step": 15040
+    },
+    {
+      "grad_norm": 0.6259526014328003,
+      "learning_rate": 1.583899429995431e-05,
+      "loss": 0.1462,
+      "step": 15050
+    },
+    {
+      "grad_norm": 0.4829665720462799,
+      "learning_rate": 1.5778671809536993e-05,
+      "loss": 0.1427,
+      "step": 15060
+    },
+    {
+      "grad_norm": 0.3559143543243408,
+      "learning_rate": 1.5718442878924246e-05,
+      "loss": 0.1386,
+      "step": 15070
+    },
+    {
+      "grad_norm": 0.4935402274131775,
+      "learning_rate": 1.5658307672779593e-05,
+      "loss": 0.1587,
+      "step": 15080
+    },
+    {
+      "grad_norm": 0.4345477521419525,
+      "learning_rate": 1.5598266355510427e-05,
+      "loss": 0.1314,
+      "step": 15090
+    },
+    {
+      "grad_norm": 0.4949833154678345,
+      "learning_rate": 1.553831909126744e-05,
+      "loss": 0.1415,
+      "step": 15100
+    },
+    {
+      "grad_norm": 0.48413971066474915,
+      "learning_rate": 1.5478466043944135e-05,
+      "loss": 0.1481,
+      "step": 15110
+    },
+    {
+      "grad_norm": 0.6730433106422424,
+      "learning_rate": 1.5418707377176468e-05,
+      "loss": 0.1362,
+      "step": 15120
+    },
+    {
+      "grad_norm": 0.39549338817596436,
+      "learning_rate": 1.535904325434233e-05,
+      "loss": 0.1502,
+      "step": 15130
+    },
+    {
+      "grad_norm": 0.4284266233444214,
+      "learning_rate": 1.529947383856118e-05,
+      "loss": 0.1458,
+      "step": 15140
+    },
+    {
+      "grad_norm": 0.5123095512390137,
+      "learning_rate": 1.5239999292693524e-05,
+      "loss": 0.1404,
+      "step": 15150
+    },
+    {
+      "grad_norm": 0.7664332389831543,
+      "learning_rate": 1.5180619779340505e-05,
+      "loss": 0.148,
+      "step": 15160
+    },
+    {
+      "grad_norm": 0.6258096694946289,
+      "learning_rate": 1.5121335460843428e-05,
+      "loss": 0.136,
+      "step": 15170
+    },
+    {
+      "grad_norm": 0.4324125647544861,
+      "learning_rate": 1.5062146499283347e-05,
+      "loss": 0.1358,
+      "step": 15180
+    },
+    {
+      "grad_norm": 0.47857117652893066,
+      "learning_rate": 1.5003053056480643e-05,
+      "loss": 0.1302,
+      "step": 15190
+    },
+    {
+      "grad_norm": 0.5040610432624817,
+      "learning_rate": 1.4944055293994551e-05,
+      "loss": 0.1545,
+      "step": 15200
+    },
+    {
+      "grad_norm": 0.5491944551467896,
+      "learning_rate": 1.4885153373122656e-05,
+      "loss": 0.1225,
+      "step": 15210
+    },
+    {
+      "grad_norm": 0.566630482673645,
+      "learning_rate": 1.482634745490059e-05,
+      "loss": 0.1529,
+      "step": 15220
+    },
+    {
+      "grad_norm": 0.6260606646537781,
+      "learning_rate": 1.4767637700101466e-05,
+      "loss": 0.1453,
+      "step": 15230
+    },
+    {
+      "grad_norm": 0.45122262835502625,
+      "learning_rate": 1.4709024269235528e-05,
+      "loss": 0.1374,
+      "step": 15240
+    },
+    {
+      "grad_norm": 0.5693982839584351,
+      "learning_rate": 1.4650507322549684e-05,
+      "loss": 0.1398,
+      "step": 15250
+    },
+    {
+      "grad_norm": 0.512087345123291,
+      "learning_rate": 1.4592087020026972e-05,
+      "loss": 0.1471,
+      "step": 15260
+    },
+    {
+      "grad_norm": 0.525966227054596,
+      "learning_rate": 1.4533763521386318e-05,
+      "loss": 0.1247,
+      "step": 15270
+    },
+    {
+      "grad_norm": 0.48545825481414795,
+      "learning_rate": 1.44755369860819e-05,
+      "loss": 0.1341,
+      "step": 15280
+    },
+    {
+      "grad_norm": 0.4079943001270294,
+      "learning_rate": 1.441740757330287e-05,
+      "loss": 0.1527,
+      "step": 15290
+    },
+    {
+      "grad_norm": 0.49379998445510864,
+      "learning_rate": 1.4359375441972844e-05,
+      "loss": 0.1311,
+      "step": 15300
+    },
+    {
+      "grad_norm": 0.617524266242981,
+      "learning_rate": 1.4301440750749395e-05,
+      "loss": 0.1444,
+      "step": 15310
+    },
+    {
+      "grad_norm": 0.4012261629104614,
+      "learning_rate": 1.4243603658023808e-05,
+      "loss": 0.1327,
+      "step": 15320
+    },
+    {
+      "grad_norm": 0.5284041166305542,
+      "learning_rate": 1.4185864321920444e-05,
+      "loss": 0.1456,
+      "step": 15330
+    },
+    {
+      "grad_norm": 0.5602944493293762,
+      "learning_rate": 1.4128222900296485e-05,
+      "loss": 0.134,
+      "step": 15340
+    },
+    {
+      "grad_norm": 0.6466161012649536,
+      "learning_rate": 1.407067955074135e-05,
+      "loss": 0.1359,
+      "step": 15350
+    },
+    {
+      "grad_norm": 0.40096116065979004,
+      "learning_rate": 1.4013234430576356e-05,
+      "loss": 0.1182,
+      "step": 15360
+    },
+    {
+      "grad_norm": 0.7968366146087646,
+      "learning_rate": 1.3955887696854286e-05,
+      "loss": 0.1543,
+      "step": 15370
+    },
+    {
+      "grad_norm": 0.5351948738098145,
+      "learning_rate": 1.38986395063589e-05,
+      "loss": 0.1431,
+      "step": 15380
+    },
+    {
+      "grad_norm": 0.4936433434486389,
+      "learning_rate": 1.3841490015604597e-05,
+      "loss": 0.1429,
+      "step": 15390
+    },
+    {
+      "grad_norm": 0.3997669219970703,
+      "learning_rate": 1.3784439380835879e-05,
+      "loss": 0.1329,
+      "step": 15400
+    },
+    {
+      "grad_norm": 0.4618523120880127,
+      "learning_rate": 1.3727487758026986e-05,
+      "loss": 0.1566,
+      "step": 15410
+    },
+    {
+      "grad_norm": 0.4930196702480316,
+      "learning_rate": 1.3670635302881525e-05,
+      "loss": 0.1462,
+      "step": 15420
+    },
+    {
+      "grad_norm": 0.42736363410949707,
+      "learning_rate": 1.3613882170831888e-05,
+      "loss": 0.1467,
+      "step": 15430
+    },
+    {
+      "grad_norm": 0.5205985307693481,
+      "learning_rate": 1.355722851703901e-05,
+      "loss": 0.1164,
+      "step": 15440
+    },
+    {
+      "grad_norm": 0.6934687495231628,
+      "learning_rate": 1.3500674496391814e-05,
+      "loss": 0.1401,
+      "step": 15450
+    },
+    {
+      "grad_norm": 0.6401660442352295,
+      "learning_rate": 1.3444220263506795e-05,
+      "loss": 0.1428,
+      "step": 15460
+    },
+    {
+      "grad_norm": 0.6236344575881958,
+      "learning_rate": 1.3387865972727714e-05,
+      "loss": 0.1325,
+      "step": 15470
+    },
+    {
+      "grad_norm": 0.44572579860687256,
+      "learning_rate": 1.3331611778125036e-05,
+      "loss": 0.1384,
+      "step": 15480
+    },
+    {
+      "grad_norm": 0.5028837323188782,
+      "learning_rate": 1.3275457833495564e-05,
+      "loss": 0.1352,
+      "step": 15490
+    },
+    {
+      "grad_norm": 0.43543121218681335,
+      "learning_rate": 1.3219404292362065e-05,
+      "loss": 0.1387,
+      "step": 15500
+    },
+    {
+      "grad_norm": 0.4481230080127716,
+      "learning_rate": 1.3163451307972751e-05,
+      "loss": 0.1413,
+      "step": 15510
+    },
+    {
+      "grad_norm": 0.4134395122528076,
+      "learning_rate": 1.3107599033300977e-05,
+      "loss": 0.1381,
+      "step": 15520
+    },
+    {
+      "grad_norm": 0.7629886865615845,
+      "learning_rate": 1.305184762104471e-05,
+      "loss": 0.1617,
+      "step": 15530
+    },
+    {
+      "grad_norm": 0.37617412209510803,
+      "learning_rate": 1.2996197223626178e-05,
+      "loss": 0.1432,
+      "step": 15540
+    },
+    {
+      "grad_norm": 0.6572557687759399,
+      "learning_rate": 1.2940647993191457e-05,
+      "loss": 0.1421,
+      "step": 15550
+    },
+    {
+      "grad_norm": 0.5591655969619751,
+      "learning_rate": 1.2885200081610005e-05,
+      "loss": 0.1307,
+      "step": 15560
+    },
+    {
+      "grad_norm": 0.3976919651031494,
+      "learning_rate": 1.2829853640474316e-05,
+      "loss": 0.1389,
+      "step": 15570
+    },
+    {
+      "grad_norm": 0.6312011480331421,
+      "learning_rate": 1.2774608821099438e-05,
+      "loss": 0.1285,
+      "step": 15580
+    },
+    {
+      "grad_norm": 0.5990042090415955,
+      "learning_rate": 1.2719465774522577e-05,
+      "loss": 0.1339,
+      "step": 15590
+    },
+    {
+      "grad_norm": 0.5309950113296509,
+      "learning_rate": 1.2664424651502755e-05,
+      "loss": 0.1558,
+      "step": 15600
+    },
+    {
+      "grad_norm": 0.47596848011016846,
+      "learning_rate": 1.260948560252026e-05,
+      "loss": 0.1508,
+      "step": 15610
+    },
+    {
+      "grad_norm": 0.4416780173778534,
+      "learning_rate": 1.2554648777776396e-05,
+      "loss": 0.1314,
+      "step": 15620
+    },
+    {
+      "grad_norm": 0.4963531792163849,
+      "learning_rate": 1.2499914327192919e-05,
+      "loss": 0.1355,
+      "step": 15630
+    },
+    {
+      "grad_norm": 0.40797120332717896,
+      "learning_rate": 1.2445282400411722e-05,
+      "loss": 0.1235,
+      "step": 15640
+    },
+    {
+      "grad_norm": 0.47188374400138855,
+      "learning_rate": 1.2390753146794437e-05,
+      "loss": 0.1323,
+      "step": 15650
+    },
+    {
+      "grad_norm": 0.6476052403450012,
+      "learning_rate": 1.2336326715421925e-05,
+      "loss": 0.1295,
+      "step": 15660
+    },
+    {
+      "grad_norm": 0.3983314335346222,
+      "learning_rate": 1.2282003255094005e-05,
+      "loss": 0.1325,
+      "step": 15670
+    },
+    {
+      "grad_norm": 0.40132319927215576,
+      "learning_rate": 1.2227782914328928e-05,
+      "loss": 0.123,
+      "step": 15680
+    },
+    {
+      "grad_norm": 0.3788958787918091,
+      "learning_rate": 1.2173665841363018e-05,
+      "loss": 0.124,
+      "step": 15690
+    },
+    {
+      "grad_norm": 0.4068279564380646,
+      "learning_rate": 1.211965218415032e-05,
+      "loss": 0.1355,
+      "step": 15700
+    },
+    {
+      "grad_norm": 0.6024231910705566,
+      "learning_rate": 1.2065742090362082e-05,
+      "loss": 0.1353,
+      "step": 15710
+    },
+    {
+      "grad_norm": 1.1619590520858765,
+      "learning_rate": 1.2011935707386457e-05,
+      "loss": 0.1541,
+      "step": 15720
+    },
+    {
+      "grad_norm": 0.4276230037212372,
+      "learning_rate": 1.1958233182328044e-05,
+      "loss": 0.1351,
+      "step": 15730
+    },
+    {
+      "grad_norm": 0.6093833446502686,
+      "learning_rate": 1.1904634662007474e-05,
+      "loss": 0.1377,
+      "step": 15740
+    },
+    {
+      "grad_norm": 0.4416544437408447,
+      "learning_rate": 1.1851140292961088e-05,
+      "loss": 0.1304,
+      "step": 15750
+    },
+    {
+      "grad_norm": 0.4658486247062683,
+      "learning_rate": 1.1797750221440424e-05,
+      "loss": 0.1394,
+      "step": 15760
+    },
+    {
+      "grad_norm": 0.4282757341861725,
+      "learning_rate": 1.1744464593411897e-05,
+      "loss": 0.1257,
+      "step": 15770
+    },
+    {
+      "grad_norm": 1.1826274394989014,
+      "learning_rate": 1.1691283554556399e-05,
+      "loss": 0.1387,
+      "step": 15780
+    },
+    {
+      "grad_norm": 0.45218122005462646,
+      "learning_rate": 1.1638207250268834e-05,
+      "loss": 0.1251,
+      "step": 15790
+    },
+    {
+      "grad_norm": 0.41805794835090637,
+      "learning_rate": 1.158523582565782e-05,
+      "loss": 0.1475,
+      "step": 15800
+    },
+    {
+      "grad_norm": 0.5664446353912354,
+      "learning_rate": 1.1532369425545192e-05,
+      "loss": 0.1344,
+      "step": 15810
+    },
+    {
+      "grad_norm": 0.532878577709198,
+      "learning_rate": 1.1479608194465662e-05,
+      "loss": 0.1324,
+      "step": 15820
+    },
+    {
+      "grad_norm": 0.4573632478713989,
+      "learning_rate": 1.1426952276666442e-05,
+      "loss": 0.1313,
+      "step": 15830
+    },
+    {
+      "grad_norm": 0.5342668294906616,
+      "learning_rate": 1.1374401816106778e-05,
+      "loss": 0.1414,
+      "step": 15840
+    },
+    {
+      "grad_norm": 0.6519197225570679,
+      "learning_rate": 1.1321956956457646e-05,
+      "loss": 0.122,
+      "step": 15850
+    },
+    {
+      "grad_norm": 0.3816360533237457,
+      "learning_rate": 1.1269617841101277e-05,
+      "loss": 0.1432,
+      "step": 15860
+    },
+    {
+      "grad_norm": 0.5894538164138794,
+      "learning_rate": 1.1217384613130804e-05,
+      "loss": 0.139,
+      "step": 15870
+    },
+    {
+      "grad_norm": 0.6336177587509155,
+      "learning_rate": 1.11652574153499e-05,
+      "loss": 0.1174,
+      "step": 15880
+    },
+    {
+      "grad_norm": 0.5145868062973022,
+      "learning_rate": 1.1113236390272303e-05,
+      "loss": 0.1381,
+      "step": 15890
+    },
+    {
+      "grad_norm": 0.41678327322006226,
+      "learning_rate": 1.106132168012155e-05,
+      "loss": 0.1555,
+      "step": 15900
+    },
+    {
+      "grad_norm": 0.4801405966281891,
+      "learning_rate": 1.1009513426830448e-05,
+      "loss": 0.1411,
+      "step": 15910
+    },
+    {
+      "grad_norm": 0.48061782121658325,
+      "learning_rate": 1.0957811772040777e-05,
+      "loss": 0.1311,
+      "step": 15920
+    },
+    {
+      "grad_norm": 0.3757217824459076,
+      "learning_rate": 1.0906216857102913e-05,
+      "loss": 0.1461,
+      "step": 15930
+    },
+    {
+      "grad_norm": 0.5418319702148438,
+      "learning_rate": 1.0854728823075355e-05,
+      "loss": 0.1221,
+      "step": 15940
+    },
+    {
+      "grad_norm": 0.4384613335132599,
+      "learning_rate": 1.0803347810724452e-05,
+      "loss": 0.1185,
+      "step": 15950
+    },
+    {
+      "grad_norm": 0.4609908163547516,
+      "learning_rate": 1.0752073960523911e-05,
+      "loss": 0.1241,
+      "step": 15960
+    },
+    {
+      "grad_norm": 0.529299795627594,
+      "learning_rate": 1.070090741265447e-05,
+      "loss": 0.1457,
+      "step": 15970
+    },
+    {
+      "grad_norm": 0.5654266476631165,
+      "learning_rate": 1.0649848307003547e-05,
+      "loss": 0.1634,
+      "step": 15980
+    },
+    {
+      "grad_norm": 0.45760494470596313,
+      "learning_rate": 1.0598896783164757e-05,
+      "loss": 0.1304,
+      "step": 15990
+    },
+    {
+      "grad_norm": 0.4073363244533539,
+      "learning_rate": 1.0548052980437645e-05,
+      "loss": 0.1373,
+      "step": 16000
+    },
+    {
+      "grad_norm": 0.4601219892501831,
+      "learning_rate": 1.049731703782722e-05,
+      "loss": 0.1299,
+      "step": 16010
+    },
+    {
+      "grad_norm": 0.5139146447181702,
+      "learning_rate": 1.0446689094043587e-05,
+      "loss": 0.1475,
+      "step": 16020
+    },
+    {
+      "grad_norm": 0.4405770003795624,
+      "learning_rate": 1.039616928750165e-05,
+      "loss": 0.1196,
+      "step": 16030
+    },
+    {
+      "grad_norm": 0.5012580156326294,
+      "learning_rate": 1.0345757756320612e-05,
+      "loss": 0.121,
+      "step": 16040
+    },
+    {
+      "grad_norm": 0.5846120119094849,
+      "learning_rate": 1.0295454638323666e-05,
+      "loss": 0.1315,
+      "step": 16050
+    },
+    {
+      "grad_norm": 0.46412724256515503,
+      "learning_rate": 1.0245260071037632e-05,
+      "loss": 0.1214,
+      "step": 16060
+    },
+    {
+      "grad_norm": 0.567833662033081,
+      "learning_rate": 1.0195174191692518e-05,
+      "loss": 0.1339,
+      "step": 16070
+    },
+    {
+      "grad_norm": 0.5218148827552795,
+      "learning_rate": 1.014519713722124e-05,
+      "loss": 0.1308,
+      "step": 16080
+    },
+    {
+      "grad_norm": 0.45488137006759644,
+      "learning_rate": 1.0095329044259132e-05,
+      "loss": 0.1546,
+      "step": 16090
+    },
+    {
+      "grad_norm": 0.326812744140625,
+      "learning_rate": 1.004557004914365e-05,
+      "loss": 0.1324,
+      "step": 16100
+    },
+    {
+      "grad_norm": 0.6174618601799011,
+      "learning_rate": 9.995920287914007e-06,
+      "loss": 0.1287,
+      "step": 16110
+    },
+    {
+      "grad_norm": 0.46022677421569824,
+      "learning_rate": 9.946379896310737e-06,
+      "loss": 0.1288,
+      "step": 16120
+    },
+    {
+      "grad_norm": 0.6942744255065918,
+      "learning_rate": 9.896949009775396e-06,
+      "loss": 0.1419,
+      "step": 16130
+    },
+    {
+      "grad_norm": 0.5942294001579285,
+      "learning_rate": 9.847627763450134e-06,
+      "loss": 0.1485,
+      "step": 16140
+    },
+    {
+      "grad_norm": 0.5537288188934326,
+      "learning_rate": 9.798416292177337e-06,
+      "loss": 0.1145,
+      "step": 16150
+    },
+    {
+      "grad_norm": 0.4877749979496002,
+      "learning_rate": 9.74931473049932e-06,
+      "loss": 0.1371,
+      "step": 16160
+    },
+    {
+      "grad_norm": 0.5071951746940613,
+      "learning_rate": 9.700323212657847e-06,
+      "loss": 0.1433,
+      "step": 16170
+    },
+    {
+      "grad_norm": 0.5510461330413818,
+      "learning_rate": 9.65144187259388e-06,
+      "loss": 0.1286,
+      "step": 16180
+    },
+    {
+      "grad_norm": 0.5004866123199463,
+      "learning_rate": 9.602670843947132e-06,
+      "loss": 0.1282,
+      "step": 16190
+    },
+    {
+      "grad_norm": 0.45095255970954895,
+      "learning_rate": 9.554010260055713e-06,
+      "loss": 0.1434,
+      "step": 16200
+    },
+    {
+      "grad_norm": 0.5912445187568665,
+      "learning_rate": 9.505460253955834e-06,
+      "loss": 0.1476,
+      "step": 16210
+    },
+    {
+      "grad_norm": 0.39689475297927856,
+      "learning_rate": 9.457020958381324e-06,
+      "loss": 0.1238,
+      "step": 16220
+    },
+    {
+      "grad_norm": 0.3529171347618103,
+      "learning_rate": 9.408692505763395e-06,
+      "loss": 0.1202,
+      "step": 16230
+    },
+    {
+      "grad_norm": 0.3619577884674072,
+      "learning_rate": 9.360475028230181e-06,
+      "loss": 0.1336,
+      "step": 16240
+    },
+    {
+      "grad_norm": 0.5226307511329651,
+      "learning_rate": 9.312368657606412e-06,
+      "loss": 0.1574,
+      "step": 16250
+    },
+    {
+      "grad_norm": 0.3679366111755371,
+      "learning_rate": 9.264373525413096e-06,
+      "loss": 0.1262,
+      "step": 16260
+    },
+    {
+      "grad_norm": 0.48918986320495605,
+      "learning_rate": 9.216489762867058e-06,
+      "loss": 0.1371,
+      "step": 16270
+    },
+    {
+      "grad_norm": 0.37229111790657043,
+      "learning_rate": 9.168717500880708e-06,
+      "loss": 0.1365,
+      "step": 16280
+    },
+    {
+      "grad_norm": 0.6316254138946533,
+      "learning_rate": 9.121056870061574e-06,
+      "loss": 0.1323,
+      "step": 16290
+    },
+    {
+      "grad_norm": 0.7448822259902954,
+      "learning_rate": 9.073508000711983e-06,
+      "loss": 0.152,
+      "step": 16300
+    },
+    {
+      "grad_norm": 0.455748587846756,
+      "learning_rate": 9.026071022828758e-06,
+      "loss": 0.1464,
+      "step": 16310
+    },
+    {
+      "grad_norm": 0.5494582056999207,
+      "learning_rate": 8.978746066102771e-06,
+      "loss": 0.1495,
+      "step": 16320
+    },
+    {
+      "grad_norm": 0.6252233386039734,
+      "learning_rate": 8.931533259918634e-06,
+      "loss": 0.1496,
+      "step": 16330
+    },
+    {
+      "grad_norm": 0.5149800181388855,
+      "learning_rate": 8.884432733354382e-06,
+      "loss": 0.1278,
+      "step": 16340
+    },
+    {
+      "grad_norm": 0.407632976770401,
+      "learning_rate": 8.837444615181029e-06,
+      "loss": 0.1485,
+      "step": 16350
+    },
+    {
+      "grad_norm": 0.5810245871543884,
+      "learning_rate": 8.790569033862323e-06,
+      "loss": 0.1261,
+      "step": 16360
+    },
+    {
+      "grad_norm": 0.6462126970291138,
+      "learning_rate": 8.7438061175543e-06,
+      "loss": 0.1348,
+      "step": 16370
+    },
+    {
+      "grad_norm": 0.5198664665222168,
+      "learning_rate": 8.697155994104978e-06,
+      "loss": 0.1425,
+      "step": 16380
+    },
+    {
+      "grad_norm": 0.46222320199012756,
+      "learning_rate": 8.650618791054033e-06,
+      "loss": 0.1281,
+      "step": 16390
+    },
+    {
+      "grad_norm": 0.5257467031478882,
+      "learning_rate": 8.604194635632373e-06,
+      "loss": 0.1505,
+      "step": 16400
+    },
+    {
+      "grad_norm": 0.5276866555213928,
+      "learning_rate": 8.557883654761906e-06,
+      "loss": 0.1478,
+      "step": 16410
+    },
+    {
+      "grad_norm": 0.6495941281318665,
+      "learning_rate": 8.511685975055061e-06,
+      "loss": 0.1324,
+      "step": 16420
+    },
+    {
+      "grad_norm": 0.4367246925830841,
+      "learning_rate": 8.46560172281452e-06,
+      "loss": 0.1511,
+      "step": 16430
+    },
+    {
+      "grad_norm": 0.4822598993778229,
+      "learning_rate": 8.419631024032893e-06,
+      "loss": 0.133,
+      "step": 16440
+    },
+    {
+      "grad_norm": 0.7041584253311157,
+      "learning_rate": 8.373774004392293e-06,
+      "loss": 0.1239,
+      "step": 16450
+    },
+    {
+      "grad_norm": 0.39874666929244995,
+      "learning_rate": 8.32803078926409e-06,
+      "loss": 0.1315,
+      "step": 16460
+    },
+    {
+      "grad_norm": 0.6261482238769531,
+      "learning_rate": 8.282401503708454e-06,
+      "loss": 0.1676,
+      "step": 16470
+    },
+    {
+      "grad_norm": 0.5142518281936646,
+      "learning_rate": 8.23688627247412e-06,
+      "loss": 0.1417,
+      "step": 16480
+    },
+    {
+      "grad_norm": 0.68407142162323,
+      "learning_rate": 8.191485219998007e-06,
+      "loss": 0.1411,
+      "step": 16490
+    },
+    {
+      "grad_norm": 0.5007943511009216,
+      "learning_rate": 8.146198470404843e-06,
+      "loss": 0.1318,
+      "step": 16500
+    },
+    {
+      "grad_norm": 0.4244157075881958,
+      "learning_rate": 8.101026147506897e-06,
+      "loss": 0.1499,
+      "step": 16510
+    },
+    {
+      "grad_norm": 0.4448772668838501,
+      "learning_rate": 8.05596837480353e-06,
+      "loss": 0.1379,
+      "step": 16520
+    },
+    {
+      "grad_norm": 0.4883294999599457,
+      "learning_rate": 8.011025275480998e-06,
+      "loss": 0.1532,
+      "step": 16530
+    },
+    {
+      "grad_norm": 0.41193774342536926,
+      "learning_rate": 7.966196972412027e-06,
+      "loss": 0.1367,
+      "step": 16540
+    },
+    {
+      "grad_norm": 0.41936343908309937,
+      "learning_rate": 7.92148358815547e-06,
+      "loss": 0.1551,
+      "step": 16550
+    },
+    {
+      "grad_norm": 0.5695376992225647,
+      "learning_rate": 7.87688524495604e-06,
+      "loss": 0.1502,
+      "step": 16560
+    },
+    {
+      "grad_norm": 0.4159487783908844,
+      "learning_rate": 7.83240206474386e-06,
+      "loss": 0.1393,
+      "step": 16570
+    },
+    {
+      "grad_norm": 0.5019527673721313,
+      "learning_rate": 7.788034169134272e-06,
+      "loss": 0.1474,
+      "step": 16580
+    },
+    {
+      "grad_norm": 0.3667476773262024,
+      "learning_rate": 7.743781679427414e-06,
+      "loss": 0.1263,
+      "step": 16590
+    },
+    {
+      "grad_norm": 0.5808566808700562,
+      "learning_rate": 7.699644716607895e-06,
+      "loss": 0.1456,
+      "step": 16600
+    },
+    {
+      "grad_norm": 0.4646558165550232,
+      "learning_rate": 7.655623401344486e-06,
+      "loss": 0.1242,
+      "step": 16610
+    },
+    {
+      "grad_norm": 0.7283293008804321,
+      "learning_rate": 7.611717853989775e-06,
+      "loss": 0.1292,
+      "step": 16620
+    },
+    {
+      "grad_norm": 0.3644586205482483,
+      "learning_rate": 7.567928194579854e-06,
+      "loss": 0.1646,
+      "step": 16630
+    },
+    {
+      "grad_norm": 0.5412976741790771,
+      "learning_rate": 7.524254542833997e-06,
+      "loss": 0.1375,
+      "step": 16640
+    },
+    {
+      "grad_norm": 0.5743188858032227,
+      "learning_rate": 7.480697018154286e-06,
+      "loss": 0.1388,
+      "step": 16650
+    },
+    {
+      "grad_norm": 0.48704782128334045,
+      "learning_rate": 7.437255739625332e-06,
+      "loss": 0.1429,
+      "step": 16660
+    },
+    {
+      "grad_norm": 0.5118150115013123,
+      "learning_rate": 7.393930826013923e-06,
+      "loss": 0.1371,
+      "step": 16670
+    },
+    {
+      "grad_norm": 0.5060493350028992,
+      "learning_rate": 7.350722395768722e-06,
+      "loss": 0.1404,
+      "step": 16680
+    },
+    {
+      "grad_norm": 0.6362712383270264,
+      "learning_rate": 7.307630567019963e-06,
+      "loss": 0.1448,
+      "step": 16690
+    },
+    {
+      "grad_norm": 0.3849766254425049,
+      "learning_rate": 7.264655457579e-06,
+      "loss": 0.1427,
+      "step": 16700
+    },
+    {
+      "grad_norm": 0.6498367786407471,
+      "learning_rate": 7.221797184938184e-06,
+      "loss": 0.1618,
+      "step": 16710
+    },
+    {
+      "grad_norm": 0.4325396716594696,
+      "learning_rate": 7.179055866270373e-06,
+      "loss": 0.1406,
+      "step": 16720
+    },
+    {
+      "grad_norm": 0.5065338015556335,
+      "learning_rate": 7.136431618428707e-06,
+      "loss": 0.1326,
+      "step": 16730
+    },
+    {
+      "grad_norm": 0.5200700163841248,
+      "learning_rate": 7.09392455794628e-06,
+      "loss": 0.1418,
+      "step": 16740
+    },
+    {
+      "grad_norm": 0.6769509315490723,
+      "learning_rate": 7.051534801035725e-06,
+      "loss": 0.145,
+      "step": 16750
+    },
+    {
+      "grad_norm": 0.6287325620651245,
+      "learning_rate": 7.00926246358905e-06,
+      "loss": 0.1572,
+      "step": 16760
+    },
+    {
+      "grad_norm": 0.42523446679115295,
+      "learning_rate": 6.967107661177191e-06,
+      "loss": 0.1494,
+      "step": 16770
+    },
+    {
+      "grad_norm": 0.4803038239479065,
+      "learning_rate": 6.925070509049786e-06,
+      "loss": 0.1473,
+      "step": 16780
+    },
+    {
+      "grad_norm": 0.42848366498947144,
+      "learning_rate": 6.883151122134812e-06,
+      "loss": 0.1543,
+      "step": 16790
+    },
+    {
+      "grad_norm": 0.46495023369789124,
+      "learning_rate": 6.8413496150382394e-06,
+      "loss": 0.138,
+      "step": 16800
+    },
+    {
+      "grad_norm": 0.6437684893608093,
+      "learning_rate": 6.7996661020438165e-06,
+      "loss": 0.1451,
+      "step": 16810
+    },
+    {
+      "grad_norm": 0.47337549924850464,
+      "learning_rate": 6.758100697112662e-06,
+      "loss": 0.1457,
+      "step": 16820
+    },
+    {
+      "grad_norm": 0.6507829427719116,
+      "learning_rate": 6.716653513883026e-06,
+      "loss": 0.1535,
+      "step": 16830
+    },
+    {
+      "grad_norm": 0.6391059756278992,
+      "learning_rate": 6.675324665669913e-06,
+      "loss": 0.1501,
+      "step": 16840
+    },
+    {
+      "grad_norm": 0.7159315347671509,
+      "learning_rate": 6.634114265464803e-06,
+      "loss": 0.1743,
+      "step": 16850
+    },
+    {
+      "grad_norm": 0.5213925838470459,
+      "learning_rate": 6.59302242593538e-06,
+      "loss": 0.1437,
+      "step": 16860
+    },
+    {
+      "grad_norm": 0.5864213109016418,
+      "learning_rate": 6.552049259425141e-06,
+      "loss": 0.156,
+      "step": 16870
+    },
+    {
+      "grad_norm": 0.7387944459915161,
+      "learning_rate": 6.511194877953181e-06,
+      "loss": 0.1529,
+      "step": 16880
+    },
+    {
+      "grad_norm": 0.5863958597183228,
+      "learning_rate": 6.470459393213813e-06,
+      "loss": 0.1534,
+      "step": 16890
+    },
+    {
+      "grad_norm": 0.5395596027374268,
+      "learning_rate": 6.429842916576279e-06,
+      "loss": 0.1312,
+      "step": 16900
+    },
+    {
+      "grad_norm": 0.49101313948631287,
+      "learning_rate": 6.389345559084503e-06,
+      "loss": 0.1505,
+      "step": 16910
+    },
+    {
+      "grad_norm": 0.5647670030593872,
+      "learning_rate": 6.348967431456682e-06,
+      "loss": 0.1789,
+      "step": 16920
+    },
+    {
+      "grad_norm": 0.6265051364898682,
+      "learning_rate": 6.30870864408511e-06,
+      "loss": 0.154,
+      "step": 16930
+    },
+    {
+      "grad_norm": 0.4168553352355957,
+      "learning_rate": 6.268569307035754e-06,
+      "loss": 0.138,
+      "step": 16940
+    },
+    {
+      "grad_norm": 0.5133265852928162,
+      "learning_rate": 6.228549530048022e-06,
+      "loss": 0.1394,
+      "step": 16950
+    },
+    {
+      "grad_norm": 0.6816703081130981,
+      "learning_rate": 6.1886494225344814e-06,
+      "loss": 0.1533,
+      "step": 16960
+    },
+    {
+      "grad_norm": 0.619709312915802,
+      "learning_rate": 6.148869093580479e-06,
+      "loss": 0.144,
+      "step": 16970
+    },
+    {
+      "grad_norm": 0.38436588644981384,
+      "learning_rate": 6.109208651943921e-06,
+      "loss": 0.1303,
+      "step": 16980
+    },
+    {
+      "grad_norm": 0.698114275932312,
+      "learning_rate": 6.069668206054946e-06,
+      "loss": 0.1433,
+      "step": 16990
+    },
+    {
+      "grad_norm": 0.4472900927066803,
+      "learning_rate": 6.0302478640156145e-06,
+      "loss": 0.148,
+      "step": 17000
+    }
+  ],
+  "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
+}