diff --git "a/trainer_state.json" "b/trainer_state.json"
deleted file mode 100644--- "a/trainer_state.json"
+++ /dev/null
@@ -1,15309 +0,0 @@
-{
-  "best_metric": null,
-  "best_model_checkpoint": null,
-  "epoch": 1.0,
-  "eval_steps": 500,
-  "global_step": 2181,
-  "is_hyper_param_search": false,
-  "is_local_process_zero": true,
-  "is_world_process_zero": true,
-  "log_history": [
-    {
-      "epoch": 0.0004585052728106373,
-      "grad_norm": 80.66014194372936,
-      "learning_rate": 1.5151515151515153e-05,
-      "loss": 6.9841,
-      "step": 1
-    },
-    {
-      "epoch": 0.0009170105456212746,
-      "grad_norm": 96.28892490957462,
-      "learning_rate": 3.0303030303030306e-05,
-      "loss": 7.0714,
-      "step": 2
-    },
-    {
-      "epoch": 0.001375515818431912,
-      "grad_norm": 63.7930159904684,
-      "learning_rate": 4.545454545454546e-05,
-      "loss": 6.7445,
-      "step": 3
-    },
-    {
-      "epoch": 0.0018340210912425492,
-      "grad_norm": 76.5374934997506,
-      "learning_rate": 6.060606060606061e-05,
-      "loss": 5.7079,
-      "step": 4
-    },
-    {
-      "epoch": 0.0022925263640531865,
-      "grad_norm": 21.740763040169888,
-      "learning_rate": 7.575757575757576e-05,
-      "loss": 5.1082,
-      "step": 5
-    },
-    {
-      "epoch": 0.002751031636863824,
-      "grad_norm": 21.301622976786533,
-      "learning_rate": 9.090909090909092e-05,
-      "loss": 5.0354,
-      "step": 6
-    },
-    {
-      "epoch": 0.0032095369096744614,
-      "grad_norm": 15.494304879713727,
-      "learning_rate": 0.00010606060606060606,
-      "loss": 4.7451,
-      "step": 7
-    },
-    {
-      "epoch": 0.0036680421824850985,
-      "grad_norm": 14.641158302877827,
-      "learning_rate": 0.00012121212121212122,
-      "loss": 4.6418,
-      "step": 8
-    },
-    {
-      "epoch": 0.0041265474552957355,
-      "grad_norm": 10.053374233601124,
-      "learning_rate": 0.00013636363636363637,
-      "loss": 4.3787,
-      "step": 9
-    },
-    {
-      "epoch": 0.004585052728106373,
-      "grad_norm": 8.86182753442264,
-      "learning_rate": 0.00015151515151515152,
-      "loss": 4.2858,
-      "step": 10
-    },
-    {
-      "epoch": 0.0050435580009170105,
-      "grad_norm": 11.643452453292435,
-      "learning_rate": 0.00016666666666666666,
-      "loss": 4.2391,
-      "step": 11
-    },
-    {
-      "epoch": 0.005502063273727648,
-      "grad_norm": 10.45687586227522,
-      "learning_rate": 0.00018181818181818183,
-      "loss": 4.2892,
-      "step": 12
-    },
-    {
-      "epoch": 0.005960568546538285,
-      "grad_norm": 11.459512436050144,
-      "learning_rate": 0.00019696969696969695,
-      "loss": 4.1703,
-      "step": 13
-    },
-    {
-      "epoch": 0.006419073819348923,
-      "grad_norm": 8.159919737168092,
-      "learning_rate": 0.00021212121212121213,
-      "loss": 4.2319,
-      "step": 14
-    },
-    {
-      "epoch": 0.0068775790921595595,
-      "grad_norm": 6.736499322987043,
-      "learning_rate": 0.00022727272727272727,
-      "loss": 4.0712,
-      "step": 15
-    },
-    {
-      "epoch": 0.007336084364970197,
-      "grad_norm": 9.494499013006847,
-      "learning_rate": 0.00024242424242424245,
-      "loss": 4.0221,
-      "step": 16
-    },
-    {
-      "epoch": 0.007794589637780834,
-      "grad_norm": 10.81104645785115,
-      "learning_rate": 0.00025757575757575756,
-      "loss": 3.9841,
-      "step": 17
-    },
-    {
-      "epoch": 0.008253094910591471,
-      "grad_norm": 12.395236986171414,
-      "learning_rate": 0.00027272727272727274,
-      "loss": 4.0025,
-      "step": 18
-    },
-    {
-      "epoch": 0.00871160018340211,
-      "grad_norm": 5.384158225076754,
-      "learning_rate": 0.0002878787878787879,
-      "loss": 3.9311,
-      "step": 19
-    },
-    {
-      "epoch": 0.009170105456212746,
-      "grad_norm": 6.582592389555483,
-      "learning_rate": 0.00030303030303030303,
-      "loss": 3.7949,
-      "step": 20
-    },
-    {
-      "epoch": 0.009628610729023384,
-      "grad_norm": 3.0100241079119474,
-      "learning_rate": 0.0003181818181818182,
-      "loss": 3.9122,
-      "step": 21
-    },
-    {
-      "epoch": 0.010087116001834021,
-      "grad_norm": 2.186816198045507,
-      "learning_rate": 0.0003333333333333333,
-      "loss": 3.9184,
-      "step": 22
-    },
-    {
-      "epoch": 0.01054562127464466,
-      "grad_norm": 1.9720611788914193,
-      "learning_rate": 0.0003484848484848485,
-      "loss": 3.7836,
-      "step": 23
-    },
-    {
-      "epoch": 0.011004126547455296,
-      "grad_norm": 1.5574131225079753,
-      "learning_rate": 0.00036363636363636367,
-      "loss": 3.7492,
-      "step": 24
-    },
-    {
-      "epoch": 0.011462631820265932,
-      "grad_norm": 1.4095896476813865,
-      "learning_rate": 0.0003787878787878788,
-      "loss": 3.8042,
-      "step": 25
-    },
-    {
-      "epoch": 0.01192113709307657,
-      "grad_norm": 1.3599812228766146,
-      "learning_rate": 0.0003939393939393939,
-      "loss": 3.679,
-      "step": 26
-    },
-    {
-      "epoch": 0.012379642365887207,
-      "grad_norm": 1.5303158312964409,
-      "learning_rate": 0.00040909090909090913,
-      "loss": 3.6645,
-      "step": 27
-    },
-    {
-      "epoch": 0.012838147638697846,
-      "grad_norm": 1.3983364628142794,
-      "learning_rate": 0.00042424242424242425,
-      "loss": 3.7072,
-      "step": 28
-    },
-    {
-      "epoch": 0.013296652911508482,
-      "grad_norm": 1.1950038043034172,
-      "learning_rate": 0.0004393939393939394,
-      "loss": 3.7279,
-      "step": 29
-    },
-    {
-      "epoch": 0.013755158184319119,
-      "grad_norm": 1.1490905966123481,
-      "learning_rate": 0.00045454545454545455,
-      "loss": 3.7133,
-      "step": 30
-    },
-    {
-      "epoch": 0.014213663457129757,
-      "grad_norm": 1.008215636475388,
-      "learning_rate": 0.0004696969696969697,
-      "loss": 3.6522,
-      "step": 31
-    },
-    {
-      "epoch": 0.014672168729940394,
-      "grad_norm": 0.894838890330922,
-      "learning_rate": 0.0004848484848484849,
-      "loss": 3.7101,
-      "step": 32
-    },
-    {
-      "epoch": 0.015130674002751032,
-      "grad_norm": 1.355893266474629,
-      "learning_rate": 0.0005,
-      "loss": 3.7161,
-      "step": 33
-    },
-    {
-      "epoch": 0.015589179275561669,
-      "grad_norm": 1.0159817710877006,
-      "learning_rate": 0.0005151515151515151,
-      "loss": 3.5887,
-      "step": 34
-    },
-    {
-      "epoch": 0.016047684548372305,
-      "grad_norm": 1.0284710621443132,
-      "learning_rate": 0.0005303030303030302,
-      "loss": 3.6596,
-      "step": 35
-    },
-    {
-      "epoch": 0.016506189821182942,
-      "grad_norm": 0.8875995894923159,
-      "learning_rate": 0.0005454545454545455,
-      "loss": 3.6496,
-      "step": 36
-    },
-    {
-      "epoch": 0.016964695093993582,
-      "grad_norm": 0.8461350484395553,
-      "learning_rate": 0.0005606060606060606,
-      "loss": 3.5863,
-      "step": 37
-    },
-    {
-      "epoch": 0.01742320036680422,
-      "grad_norm": 0.8401185988756101,
-      "learning_rate": 0.0005757575757575758,
-      "loss": 3.6505,
-      "step": 38
-    },
-    {
-      "epoch": 0.017881705639614855,
-      "grad_norm": 0.7376779356776575,
-      "learning_rate": 0.0005909090909090909,
-      "loss": 3.5117,
-      "step": 39
-    },
-    {
-      "epoch": 0.018340210912425492,
-      "grad_norm": 0.9018522173704681,
-      "learning_rate": 0.0006060606060606061,
-      "loss": 3.5356,
-      "step": 40
-    },
-    {
-      "epoch": 0.01879871618523613,
-      "grad_norm": 1.1776047235108242,
-      "learning_rate": 0.0006212121212121212,
-      "loss": 3.5186,
-      "step": 41
-    },
-    {
-      "epoch": 0.01925722145804677,
-      "grad_norm": 0.7592915766691384,
-      "learning_rate": 0.0006363636363636364,
-      "loss": 3.5541,
-      "step": 42
-    },
-    {
-      "epoch": 0.019715726730857405,
-      "grad_norm": 1.1406502055459768,
-      "learning_rate": 0.0006515151515151515,
-      "loss": 3.4653,
-      "step": 43
-    },
-    {
-      "epoch": 0.020174232003668042,
-      "grad_norm": 0.783997505338904,
-      "learning_rate": 0.0006666666666666666,
-      "loss": 3.5407,
-      "step": 44
-    },
-    {
-      "epoch": 0.02063273727647868,
-      "grad_norm": 0.7145370717341182,
-      "learning_rate": 0.0006818181818181818,
-      "loss": 3.5236,
-      "step": 45
-    },
-    {
-      "epoch": 0.02109124254928932,
-      "grad_norm": 0.6941066439124494,
-      "learning_rate": 0.000696969696969697,
-      "loss": 3.4469,
-      "step": 46
-    },
-    {
-      "epoch": 0.021549747822099955,
-      "grad_norm": 0.6761709165456139,
-      "learning_rate": 0.0007121212121212122,
-      "loss": 3.4892,
-      "step": 47
-    },
-    {
-      "epoch": 0.02200825309491059,
-      "grad_norm": 0.9243304084809468,
-      "learning_rate": 0.0007272727272727273,
-      "loss": 3.5224,
-      "step": 48
-    },
-    {
-      "epoch": 0.02246675836772123,
-      "grad_norm": 0.9711128508038874,
-      "learning_rate": 0.0007424242424242425,
-      "loss": 3.4684,
-      "step": 49
-    },
-    {
-      "epoch": 0.022925263640531865,
-      "grad_norm": 0.6834501439207199,
-      "learning_rate": 0.0007575757575757576,
-      "loss": 3.4137,
-      "step": 50
-    },
-    {
-      "epoch": 0.023383768913342505,
-      "grad_norm": 0.9000509075289157,
-      "learning_rate": 0.0007727272727272727,
-      "loss": 3.3845,
-      "step": 51
-    },
-    {
-      "epoch": 0.02384227418615314,
-      "grad_norm": 0.8914282631391856,
-      "learning_rate": 0.0007878787878787878,
-      "loss": 3.58,
-      "step": 52
-    },
-    {
-      "epoch": 0.024300779458963778,
-      "grad_norm": 0.8446758172333043,
-      "learning_rate": 0.000803030303030303,
-      "loss": 3.4406,
-      "step": 53
-    },
-    {
-      "epoch": 0.024759284731774415,
-      "grad_norm": 0.8503434649135407,
-      "learning_rate": 0.0008181818181818183,
-      "loss": 3.4819,
-      "step": 54
-    },
-    {
-      "epoch": 0.02521779000458505,
-      "grad_norm": 0.9329321494195977,
-      "learning_rate": 0.0008333333333333334,
-      "loss": 3.471,
-      "step": 55
-    },
-    {
-      "epoch": 0.02567629527739569,
-      "grad_norm": 1.0909666485759235,
-      "learning_rate": 0.0008484848484848485,
-      "loss": 3.4436,
-      "step": 56
-    },
-    {
-      "epoch": 0.026134800550206328,
-      "grad_norm": 0.9495428360161083,
-      "learning_rate": 0.0008636363636363636,
-      "loss": 3.4487,
-      "step": 57
-    },
-    {
-      "epoch": 0.026593305823016965,
-      "grad_norm": 0.9163970829122245,
-      "learning_rate": 0.0008787878787878789,
-      "loss": 3.5042,
-      "step": 58
-    },
-    {
-      "epoch": 0.0270518110958276,
-      "grad_norm": 0.7866566606764968,
-      "learning_rate": 0.000893939393939394,
-      "loss": 3.2726,
-      "step": 59
-    },
-    {
-      "epoch": 0.027510316368638238,
-      "grad_norm": 0.8659441062808783,
-      "learning_rate": 0.0009090909090909091,
-      "loss": 3.3909,
-      "step": 60
-    },
-    {
-      "epoch": 0.027968821641448878,
-      "grad_norm": 0.8583475516624618,
-      "learning_rate": 0.0009242424242424242,
-      "loss": 3.519,
-      "step": 61
-    },
-    {
-      "epoch": 0.028427326914259515,
-      "grad_norm": 1.3033776360837526,
-      "learning_rate": 0.0009393939393939394,
-      "loss": 3.4303,
-      "step": 62
-    },
-    {
-      "epoch": 0.02888583218707015,
-      "grad_norm": 1.1908936515662136,
-      "learning_rate": 0.0009545454545454546,
-      "loss": 3.4387,
-      "step": 63
-    },
-    {
-      "epoch": 0.029344337459880788,
-      "grad_norm": 0.9993815633153912,
-      "learning_rate": 0.0009696969696969698,
-      "loss": 3.3656,
-      "step": 64
-    },
-    {
-      "epoch": 0.029802842732691424,
-      "grad_norm": 1.1473680523552312,
-      "learning_rate": 0.000984848484848485,
-      "loss": 3.3111,
-      "step": 65
-    },
-    {
-      "epoch": 0.030261348005502064,
-      "grad_norm": 1.2346546280392177,
-      "learning_rate": 0.001,
-      "loss": 3.3226,
-      "step": 66
-    },
-    {
-      "epoch": 0.0307198532783127,
-      "grad_norm": 1.0743760115212055,
-      "learning_rate": 0.0009999994484067654,
-      "loss": 3.3572,
-      "step": 67
-    },
-    {
-      "epoch": 0.031178358551123338,
-      "grad_norm": 1.049776742252738,
-      "learning_rate": 0.0009999977936282788,
-      "loss": 3.4081,
-      "step": 68
-    },
-    {
-      "epoch": 0.03163686382393398,
-      "grad_norm": 1.0817500633038226,
-      "learning_rate": 0.0009999950356681913,
-      "loss": 3.4216,
-      "step": 69
-    },
-    {
-      "epoch": 0.03209536909674461,
-      "grad_norm": 1.3501114015069227,
-      "learning_rate": 0.0009999911745325876,
-      "loss": 3.3673,
-      "step": 70
-    },
-    {
-      "epoch": 0.03255387436955525,
-      "grad_norm": 1.4123840777124592,
-      "learning_rate": 0.0009999862102299873,
-      "loss": 3.3579,
-      "step": 71
-    },
-    {
-      "epoch": 0.033012379642365884,
-      "grad_norm": 1.2830076910731105,
-      "learning_rate": 0.0009999801427713433,
-      "loss": 3.3226,
-      "step": 72
-    },
-    {
-      "epoch": 0.033470884915176524,
-      "grad_norm": 1.1785852780872696,
-      "learning_rate": 0.0009999729721700424,
-      "loss": 3.2519,
-      "step": 73
-    },
-    {
-      "epoch": 0.033929390187987164,
-      "grad_norm": 1.0904183624775445,
-      "learning_rate": 0.000999964698441906,
-      "loss": 3.2925,
-      "step": 74
-    },
-    {
-      "epoch": 0.0343878954607978,
-      "grad_norm": 1.0221144363250596,
-      "learning_rate": 0.0009999553216051892,
-      "loss": 3.2013,
-      "step": 75
-    },
-    {
-      "epoch": 0.03484640073360844,
-      "grad_norm": 1.041963912952068,
-      "learning_rate": 0.00099994484168058,
-      "loss": 3.3587,
-      "step": 76
-    },
-    {
-      "epoch": 0.03530490600641907,
-      "grad_norm": 1.1583924966935402,
-      "learning_rate": 0.0009999332586912019,
-      "loss": 3.4388,
-      "step": 77
-    },
-    {
-      "epoch": 0.03576341127922971,
-      "grad_norm": 1.025006848223709,
-      "learning_rate": 0.0009999205726626108,
-      "loss": 3.2671,
-      "step": 78
-    },
-    {
-      "epoch": 0.03622191655204035,
-      "grad_norm": 0.9300867383494789,
-      "learning_rate": 0.000999906783622797,
-      "loss": 3.2857,
-      "step": 79
-    },
-    {
-      "epoch": 0.036680421824850984,
-      "grad_norm": 0.9426177102687758,
-      "learning_rate": 0.0009998918916021842,
-      "loss": 3.2622,
-      "step": 80
-    },
-    {
-      "epoch": 0.037138927097661624,
-      "grad_norm": 0.8509126084088136,
-      "learning_rate": 0.0009998758966336297,
-      "loss": 3.2837,
-      "step": 81
-    },
-    {
-      "epoch": 0.03759743237047226,
-      "grad_norm": 1.0440840330104177,
-      "learning_rate": 0.0009998587987524242,
-      "loss": 3.2867,
-      "step": 82
-    },
-    {
-      "epoch": 0.0380559376432829,
-      "grad_norm": 1.3771408981097388,
-      "learning_rate": 0.0009998405979962926,
-      "loss": 3.2079,
-      "step": 83
-    },
-    {
-      "epoch": 0.03851444291609354,
-      "grad_norm": 1.0879501484008387,
-      "learning_rate": 0.000999821294405392,
-      "loss": 3.3465,
-      "step": 84
-    },
-    {
-      "epoch": 0.03897294818890417,
-      "grad_norm": 1.0510906903653494,
-      "learning_rate": 0.0009998008880223134,
-      "loss": 3.2552,
-      "step": 85
-    },
-    {
-      "epoch": 0.03943145346171481,
-      "grad_norm": 0.9786288877330902,
-      "learning_rate": 0.000999779378892081,
-      "loss": 3.3383,
-      "step": 86
-    },
-    {
-      "epoch": 0.039889958734525444,
-      "grad_norm": 1.102060138796188,
-      "learning_rate": 0.0009997567670621522,
-      "loss": 3.2856,
-      "step": 87
-    },
-    {
-      "epoch": 0.040348464007336084,
-      "grad_norm": 1.0084230593598118,
-      "learning_rate": 0.0009997330525824165,
-      "loss": 3.2426,
-      "step": 88
-    },
-    {
-      "epoch": 0.040806969280146724,
-      "grad_norm": 1.151927248798999,
-      "learning_rate": 0.0009997082355051976,
-      "loss": 3.2684,
-      "step": 89
-    },
-    {
-      "epoch": 0.04126547455295736,
-      "grad_norm": 1.0333154739850925,
-      "learning_rate": 0.000999682315885251,
-      "loss": 3.2063,
-      "step": 90
-    },
-    {
-      "epoch": 0.041723979825768,
-      "grad_norm": 0.995071795685202,
-      "learning_rate": 0.0009996552937797645,
-      "loss": 3.3013,
-      "step": 91
-    },
-    {
-      "epoch": 0.04218248509857864,
-      "grad_norm": 1.1003014443063381,
-      "learning_rate": 0.0009996271692483596,
-      "loss": 3.2194,
-      "step": 92
-    },
-    {
-      "epoch": 0.04264099037138927,
-      "grad_norm": 0.9327888011318094,
-      "learning_rate": 0.0009995979423530893,
-      "loss": 3.2041,
-      "step": 93
-    },
-    {
-      "epoch": 0.04309949564419991,
-      "grad_norm": 1.313753393844391,
-      "learning_rate": 0.000999567613158439,
-      "loss": 3.0976,
-      "step": 94
-    },
-    {
-      "epoch": 0.04355800091701054,
-      "grad_norm": 0.958501225504884,
-      "learning_rate": 0.0009995361817313263,
-      "loss": 3.1696,
-      "step": 95
-    },
-    {
-      "epoch": 0.04401650618982118,
-      "grad_norm": 1.1256844728144912,
-      "learning_rate": 0.0009995036481411004,
-      "loss": 3.2345,
-      "step": 96
-    },
-    {
-      "epoch": 0.044475011462631824,
-      "grad_norm": 1.080389549617024,
-      "learning_rate": 0.0009994700124595429,
-      "loss": 3.1914,
-      "step": 97
-    },
-    {
-      "epoch": 0.04493351673544246,
-      "grad_norm": 0.9566044400662667,
-      "learning_rate": 0.0009994352747608663,
-      "loss": 3.1861,
-      "step": 98
-    },
-    {
-      "epoch": 0.0453920220082531,
-      "grad_norm": 0.9885136129478937,
-      "learning_rate": 0.0009993994351217151,
-      "loss": 3.1835,
-      "step": 99
-    },
-    {
-      "epoch": 0.04585052728106373,
-      "grad_norm": 1.0944402653822762,
-      "learning_rate": 0.000999362493621165,
-      "loss": 3.1025,
-      "step": 100
-    },
-    {
-      "epoch": 0.04630903255387437,
-      "grad_norm": 1.152977319890535,
-      "learning_rate": 0.0009993244503407226,
-      "loss": 3.155,
-      "step": 101
-    },
-    {
-      "epoch": 0.04676753782668501,
-      "grad_norm": 1.0760442412303444,
-      "learning_rate": 0.0009992853053643258,
-      "loss": 3.1312,
-      "step": 102
-    },
-    {
-      "epoch": 0.04722604309949564,
-      "grad_norm": 0.9670021012702773,
-      "learning_rate": 0.0009992450587783426,
-      "loss": 3.1499,
-      "step": 103
-    },
-    {
-      "epoch": 0.04768454837230628,
-      "grad_norm": 1.0333989017379546,
-      "learning_rate": 0.000999203710671572,
-      "loss": 3.0816,
-      "step": 104
-    },
-    {
-      "epoch": 0.048143053645116916,
-      "grad_norm": 0.9724961947692106,
-      "learning_rate": 0.0009991612611352438,
-      "loss": 3.0655,
-      "step": 105
-    },
-    {
-      "epoch": 0.048601558917927556,
-      "grad_norm": 0.9803891035611217,
-      "learning_rate": 0.0009991177102630173,
-      "loss": 3.246,
-      "step": 106
-    },
-    {
-      "epoch": 0.049060064190738197,
-      "grad_norm": 0.9422055962312962,
-      "learning_rate": 0.0009990730581509817,
-      "loss": 3.0568,
-      "step": 107
-    },
-    {
-      "epoch": 0.04951856946354883,
-      "grad_norm": 0.9548355557051555,
-      "learning_rate": 0.0009990273048976566,
-      "loss": 3.1765,
-      "step": 108
-    },
-    {
-      "epoch": 0.04997707473635947,
-      "grad_norm": 0.9758560849089173,
-      "learning_rate": 0.0009989804506039905,
-      "loss": 3.071,
-      "step": 109
-    },
-    {
-      "epoch": 0.0504355800091701,
-      "grad_norm": 1.0092997712262735,
-      "learning_rate": 0.0009989324953733614,
-      "loss": 3.1735,
-      "step": 110
-    },
-    {
-      "epoch": 0.05089408528198074,
-      "grad_norm": 1.0645283348741235,
-      "learning_rate": 0.0009988834393115766,
-      "loss": 3.072,
-      "step": 111
-    },
-    {
-      "epoch": 0.05135259055479138,
-      "grad_norm": 1.037200543424908,
-      "learning_rate": 0.000998833282526872,
-      "loss": 3.0813,
-      "step": 112
-    },
-    {
-      "epoch": 0.051811095827602016,
-      "grad_norm": 1.1079038861073334,
-      "learning_rate": 0.0009987820251299122,
-      "loss": 3.1153,
-      "step": 113
-    },
-    {
-      "epoch": 0.052269601100412656,
-      "grad_norm": 1.0753480852957635,
-      "learning_rate": 0.00099872966723379,
-      "loss": 3.0776,
-      "step": 114
-    },
-    {
-      "epoch": 0.05272810637322329,
-      "grad_norm": 0.9948182850375495,
-      "learning_rate": 0.0009986762089540266,
-      "loss": 3.1204,
-      "step": 115
-    },
-    {
-      "epoch": 0.05318661164603393,
-      "grad_norm": 1.0123243070751304,
-      "learning_rate": 0.0009986216504085709,
-      "loss": 3.1169,
-      "step": 116
-    },
-    {
-      "epoch": 0.05364511691884457,
-      "grad_norm": 1.0274005999264875,
-      "learning_rate": 0.0009985659917177991,
-      "loss": 3.1109,
-      "step": 117
-    },
-    {
-      "epoch": 0.0541036221916552,
-      "grad_norm": 0.9902328791221441,
-      "learning_rate": 0.0009985092330045155,
-      "loss": 3.052,
-      "step": 118
-    },
-    {
-      "epoch": 0.05456212746446584,
-      "grad_norm": 1.112642146590195,
-      "learning_rate": 0.0009984513743939508,
-      "loss": 2.9245,
-      "step": 119
-    },
-    {
-      "epoch": 0.055020632737276476,
-      "grad_norm": 0.96567386465752,
-      "learning_rate": 0.0009983924160137626,
-      "loss": 3.0404,
-      "step": 120
-    },
-    {
-      "epoch": 0.055479138010087116,
-      "grad_norm": 0.94464787850937,
-      "learning_rate": 0.000998332357994035,
-      "loss": 3.066,
-      "step": 121
-    },
-    {
-      "epoch": 0.055937643282897756,
-      "grad_norm": 0.9451276498937964,
-      "learning_rate": 0.0009982712004672786,
-      "loss": 3.0875,
-      "step": 122
-    },
-    {
-      "epoch": 0.05639614855570839,
-      "grad_norm": 1.087906790258818,
-      "learning_rate": 0.0009982089435684295,
-      "loss": 3.0403,
-      "step": 123
-    },
-    {
-      "epoch": 0.05685465382851903,
-      "grad_norm": 0.9458271479833253,
-      "learning_rate": 0.0009981455874348499,
-      "loss": 2.9707,
-      "step": 124
-    },
-    {
-      "epoch": 0.05731315910132966,
-      "grad_norm": 0.9916336778876604,
-      "learning_rate": 0.0009980811322063269,
-      "loss": 3.0225,
-      "step": 125
-    },
-    {
-      "epoch": 0.0577716643741403,
-      "grad_norm": 0.9923762255560588,
-      "learning_rate": 0.0009980155780250728,
-      "loss": 3.0229,
-      "step": 126
-    },
-    {
-      "epoch": 0.05823016964695094,
-      "grad_norm": 1.1893755839744062,
-      "learning_rate": 0.0009979489250357243,
-      "loss": 3.0079,
-      "step": 127
-    },
-    {
-      "epoch": 0.058688674919761576,
-      "grad_norm": 0.9488359580142646,
-      "learning_rate": 0.0009978811733853431,
-      "loss": 2.999,
-      "step": 128
-    },
-    {
-      "epoch": 0.059147180192572216,
-      "grad_norm": 1.010065658568233,
-      "learning_rate": 0.0009978123232234147,
-      "loss": 2.9367,
-      "step": 129
-    },
-    {
-      "epoch": 0.05960568546538285,
-      "grad_norm": 0.9800901874234773,
-      "learning_rate": 0.000997742374701848,
-      "loss": 3.1495,
-      "step": 130
-    },
-    {
-      "epoch": 0.06006419073819349,
-      "grad_norm": 1.159783508969774,
-      "learning_rate": 0.0009976713279749754,
-      "loss": 2.9605,
-      "step": 131
-    },
-    {
-      "epoch": 0.06052269601100413,
-      "grad_norm": 1.100633688722624,
-      "learning_rate": 0.0009975991831995528,
-      "loss": 2.9843,
-      "step": 132
-    },
-    {
-      "epoch": 0.06098120128381476,
-      "grad_norm": 1.102031007858914,
-      "learning_rate": 0.0009975259405347581,
-      "loss": 2.9772,
-      "step": 133
-    },
-    {
-      "epoch": 0.0614397065566254,
-      "grad_norm": 1.11557073080476,
-      "learning_rate": 0.0009974516001421926,
-      "loss": 2.9492,
-      "step": 134
-    },
-    {
-      "epoch": 0.061898211829436035,
-      "grad_norm": 1.008810138666867,
-      "learning_rate": 0.000997376162185878,
-      "loss": 2.9546,
-      "step": 135
-    },
-    {
-      "epoch": 0.062356717102246675,
-      "grad_norm": 1.0767288999171818,
-      "learning_rate": 0.0009972996268322594,
-      "loss": 2.9794,
-      "step": 136
-    },
-    {
-      "epoch": 0.06281522237505731,
-      "grad_norm": 1.0518783869472075,
-      "learning_rate": 0.0009972219942502017,
-      "loss": 2.9189,
-      "step": 137
-    },
-    {
-      "epoch": 0.06327372764786796,
-      "grad_norm": 1.0043428901040614,
-      "learning_rate": 0.0009971432646109918,
-      "loss": 2.9522,
-      "step": 138
-    },
-    {
-      "epoch": 0.06373223292067859,
-      "grad_norm": 1.0736490533622907,
-      "learning_rate": 0.0009970634380883365,
-      "loss": 2.8784,
-      "step": 139
-    },
-    {
-      "epoch": 0.06419073819348922,
-      "grad_norm": 1.1466930713661942,
-      "learning_rate": 0.0009969825148583627,
-      "loss": 2.9204,
-      "step": 140
-    },
-    {
-      "epoch": 0.06464924346629987,
-      "grad_norm": 1.098902587828927,
-      "learning_rate": 0.0009969004950996173,
-      "loss": 3.0238,
-      "step": 141
-    },
-    {
-      "epoch": 0.0651077487391105,
-      "grad_norm": 1.0346840681234006,
-      "learning_rate": 0.0009968173789930668,
-      "loss": 2.9666,
-      "step": 142
-    },
-    {
-      "epoch": 0.06556625401192114,
-      "grad_norm": 1.0588377177893822,
-      "learning_rate": 0.0009967331667220958,
-      "loss": 2.889,
-      "step": 143
-    },
-    {
-      "epoch": 0.06602475928473177,
-      "grad_norm": 1.0760485068674208,
-      "learning_rate": 0.0009966478584725086,
-      "loss": 2.9288,
-      "step": 144
-    },
-    {
-      "epoch": 0.06648326455754242,
-      "grad_norm": 1.0916325824279938,
-      "learning_rate": 0.0009965614544325263,
-      "loss": 2.9384,
-      "step": 145
-    },
-    {
-      "epoch": 0.06694176983035305,
-      "grad_norm": 1.0615867373904504,
-      "learning_rate": 0.000996473954792789,
-      "loss": 2.8083,
-      "step": 146
-    },
-    {
-      "epoch": 0.06740027510316368,
-      "grad_norm": 1.827888101528001,
-      "learning_rate": 0.0009963853597463532,
-      "loss": 2.8215,
-      "step": 147
-    },
-    {
-      "epoch": 0.06785878037597433,
-      "grad_norm": 1.1244101095403432,
-      "learning_rate": 0.000996295669488693,
-      "loss": 2.8482,
-      "step": 148
-    },
-    {
-      "epoch": 0.06831728564878496,
-      "grad_norm": 1.222543343687254,
-      "learning_rate": 0.0009962048842176979,
-      "loss": 2.9654,
-      "step": 149
-    },
-    {
-      "epoch": 0.0687757909215956,
-      "grad_norm": 1.1102036979073797,
-      "learning_rate": 0.0009961130041336748,
-      "loss": 2.8001,
-      "step": 150
-    },
-    {
-      "epoch": 0.06923429619440624,
-      "grad_norm": 1.2483869247552615,
-      "learning_rate": 0.0009960200294393449,
-      "loss": 2.8694,
-      "step": 151
-    },
-    {
-      "epoch": 0.06969280146721687,
-      "grad_norm": 1.0859626833136278,
-      "learning_rate": 0.0009959259603398453,
-      "loss": 2.8159,
-      "step": 152
-    },
-    {
-      "epoch": 0.07015130674002751,
-      "grad_norm": 1.0284250890709725,
-      "learning_rate": 0.0009958307970427275,
-      "loss": 2.8778,
-      "step": 153
-    },
-    {
-      "epoch": 0.07060981201283814,
-      "grad_norm": 1.0681128618486486,
-      "learning_rate": 0.0009957345397579572,
-      "loss": 2.891,
-      "step": 154
-    },
-    {
-      "epoch": 0.07106831728564879,
-      "grad_norm": 0.9656128520726124,
-      "learning_rate": 0.0009956371886979138,
-      "loss": 2.8084,
-      "step": 155
-    },
-    {
-      "epoch": 0.07152682255845942,
-      "grad_norm": 0.9084007740298274,
-      "learning_rate": 0.00099553874407739,
-      "loss": 2.7262,
-      "step": 156
-    },
-    {
-      "epoch": 0.07198532783127005,
-      "grad_norm": 1.0476147569849856,
-      "learning_rate": 0.0009954392061135916,
-      "loss": 2.8734,
-      "step": 157
-    },
-    {
-      "epoch": 0.0724438331040807,
-      "grad_norm": 1.1863174192902035,
-      "learning_rate": 0.0009953385750261364,
-      "loss": 2.8887,
-      "step": 158
-    },
-    {
-      "epoch": 0.07290233837689133,
-      "grad_norm": 0.9970534483877987,
-      "learning_rate": 0.0009952368510370538,
-      "loss": 2.7782,
-      "step": 159
-    },
-    {
-      "epoch": 0.07336084364970197,
-      "grad_norm": 0.9288939472190667,
-      "learning_rate": 0.0009951340343707852,
-      "loss": 2.8051,
-      "step": 160
-    },
-    {
-      "epoch": 0.07381934892251261,
-      "grad_norm": 1.039030624022607,
-      "learning_rate": 0.0009950301252541823,
-      "loss": 2.7478,
-      "step": 161
-    },
-    {
-      "epoch": 0.07427785419532325,
-      "grad_norm": 1.0263650056614357,
-      "learning_rate": 0.0009949251239165075,
-      "loss": 2.7989,
-      "step": 162
-    },
-    {
-      "epoch": 0.07473635946813388,
-      "grad_norm": 1.075555420164327,
-      "learning_rate": 0.000994819030589433,
-      "loss": 2.8432,
-      "step": 163
-    },
-    {
-      "epoch": 0.07519486474094451,
-      "grad_norm": 0.9594967496919021,
-      "learning_rate": 0.00099471184550704,
-      "loss": 2.7438,
-      "step": 164
-    },
-    {
-      "epoch": 0.07565337001375516,
-      "grad_norm": 1.2645734639570843,
-      "learning_rate": 0.0009946035689058189,
-      "loss": 2.7968,
-      "step": 165
-    },
-    {
-      "epoch": 0.0761118752865658,
-      "grad_norm": 0.9786298707980176,
-      "learning_rate": 0.0009944942010246681,
-      "loss": 2.7402,
-      "step": 166
-    },
-    {
-      "epoch": 0.07657038055937643,
-      "grad_norm": 1.024436368599405,
-      "learning_rate": 0.0009943837421048942,
-      "loss": 2.6742,
-      "step": 167
-    },
-    {
-      "epoch": 0.07702888583218707,
-      "grad_norm": 1.0394772126848457,
-      "learning_rate": 0.0009942721923902106,
-      "loss": 2.7194,
-      "step": 168
-    },
-    {
-      "epoch": 0.07748739110499771,
-      "grad_norm": 1.1480102019862903,
-      "learning_rate": 0.0009941595521267377,
-      "loss": 2.7017,
-      "step": 169
-    },
-    {
-      "epoch": 0.07794589637780834,
-      "grad_norm": 1.1624271663543033,
-      "learning_rate": 0.0009940458215630017,
-      "loss": 2.714,
-      "step": 170
-    },
-    {
-      "epoch": 0.07840440165061899,
-      "grad_norm": 1.199150081153361,
-      "learning_rate": 0.0009939310009499348,
-      "loss": 2.7036,
-      "step": 171
-    },
-    {
-      "epoch": 0.07886290692342962,
-      "grad_norm": 1.0429334634533718,
-      "learning_rate": 0.000993815090540874,
-      "loss": 2.7626,
-      "step": 172
-    },
-    {
-      "epoch": 0.07932141219624025,
-      "grad_norm": 1.1383347560243362,
-      "learning_rate": 0.000993698090591561,
-      "loss": 2.7554,
-      "step": 173
-    },
-    {
-      "epoch": 0.07977991746905089,
-      "grad_norm": 1.0089542294324998,
-      "learning_rate": 0.0009935800013601416,
-      "loss": 2.7116,
-      "step": 174
-    },
-    {
-      "epoch": 0.08023842274186153,
-      "grad_norm": 1.2174273678291487,
-      "learning_rate": 0.000993460823107164,
-      "loss": 2.6312,
-      "step": 175
-    },
-    {
-      "epoch": 0.08069692801467217,
-      "grad_norm": 1.1200492321942064,
-      "learning_rate": 0.0009933405560955803,
-      "loss": 2.7355,
-      "step": 176
-    },
-    {
-      "epoch": 0.0811554332874828,
-      "grad_norm": 1.0198201554847668,
-      "learning_rate": 0.0009932192005907446,
-      "loss": 2.7033,
-      "step": 177
-    },
-    {
-      "epoch": 0.08161393856029345,
-      "grad_norm": 1.3041155669400353,
-      "learning_rate": 0.0009930967568604118,
-      "loss": 2.7327,
-      "step": 178
-    },
-    {
-      "epoch": 0.08207244383310408,
-      "grad_norm": 1.065015101604865,
-      "learning_rate": 0.000992973225174739,
-      "loss": 2.7108,
-      "step": 179
-    },
-    {
-      "epoch": 0.08253094910591471,
-      "grad_norm": 0.9061530057203948,
-      "learning_rate": 0.0009928486058062827,
-      "loss": 2.6731,
-      "step": 180
-    },
-    {
-      "epoch": 0.08298945437872536,
-      "grad_norm": 1.054353120567777,
-      "learning_rate": 0.0009927228990299999,
-      "loss": 2.7079,
-      "step": 181
-    },
-    {
-      "epoch": 0.083447959651536,
-      "grad_norm": 1.0953215586611484,
-      "learning_rate": 0.0009925961051232468,
-      "loss": 2.6381,
-      "step": 182
-    },
-    {
-      "epoch": 0.08390646492434663,
-      "grad_norm": 0.9513250062214248,
-      "learning_rate": 0.000992468224365778,
-      "loss": 2.7373,
-      "step": 183
-    },
-    {
-      "epoch": 0.08436497019715727,
-      "grad_norm": 0.950665971918881,
-      "learning_rate": 0.000992339257039746,
-      "loss": 2.6616,
-      "step": 184
-    },
-    {
-      "epoch": 0.08482347546996791,
-      "grad_norm": 0.9549753678904646,
-      "learning_rate": 0.0009922092034297006,
-      "loss": 2.7514,
-      "step": 185
-    },
-    {
-      "epoch": 0.08528198074277854,
-      "grad_norm": 1.1828491840191873,
-      "learning_rate": 0.0009920780638225891,
-      "loss": 2.6252,
-      "step": 186
-    },
-    {
-      "epoch": 0.08574048601558917,
-      "grad_norm": 1.0522401933438232,
-      "learning_rate": 0.0009919458385077538,
-      "loss": 2.6176,
-      "step": 187
-    },
-    {
-      "epoch": 0.08619899128839982,
-      "grad_norm": 0.9983079166712243,
-      "learning_rate": 0.0009918125277769336,
-      "loss": 2.5861,
-      "step": 188
-    },
-    {
-      "epoch": 0.08665749656121045,
-      "grad_norm": 1.0580091296154863,
-      "learning_rate": 0.0009916781319242614,
-      "loss": 2.6871,
-      "step": 189
-    },
-    {
-      "epoch": 0.08711600183402109,
-      "grad_norm": 1.1027085319305565,
-      "learning_rate": 0.0009915426512462646,
-      "loss": 2.6434,
-      "step": 190
-    },
-    {
-      "epoch": 0.08757450710683173,
-      "grad_norm": 0.9820481095577646,
-      "learning_rate": 0.0009914060860418644,
-      "loss": 2.4997,
-      "step": 191
-    },
-    {
-      "epoch": 0.08803301237964237,
-      "grad_norm": 1.2231128262124527,
-      "learning_rate": 0.000991268436612374,
-      "loss": 2.6188,
-      "step": 192
-    },
-    {
-      "epoch": 0.088491517652453,
-      "grad_norm": 1.0663558569029885,
-      "learning_rate": 0.0009911297032614997,
-      "loss": 2.6811,
-      "step": 193
-    },
-    {
-      "epoch": 0.08895002292526365,
-      "grad_norm": 1.0305404249350487,
-      "learning_rate": 0.000990989886295339,
-      "loss": 2.6945,
-      "step": 194
-    },
-    {
-      "epoch": 0.08940852819807428,
-      "grad_norm": 1.0004863282869267,
-      "learning_rate": 0.0009908489860223804,
-      "loss": 2.7399,
-      "step": 195
-    },
-    {
-      "epoch": 0.08986703347088491,
-      "grad_norm": 0.9360166964752862,
-      "learning_rate": 0.000990707002753502,
-      "loss": 2.7262,
-      "step": 196
-    },
-    {
-      "epoch": 0.09032553874369555,
-      "grad_norm": 0.8641375226291981,
-      "learning_rate": 0.0009905639368019724,
-      "loss": 2.6671,
-      "step": 197
-    },
-    {
-      "epoch": 0.0907840440165062,
-      "grad_norm": 1.1134660705442792,
-      "learning_rate": 0.0009904197884834482,
-      "loss": 2.5338,
-      "step": 198
-    },
-    {
-      "epoch": 0.09124254928931683,
-      "grad_norm": 0.972405726464288,
-      "learning_rate": 0.0009902745581159742,
-      "loss": 2.6238,
-      "step": 199
-    },
-    {
-      "epoch": 0.09170105456212746,
-      "grad_norm": 1.0510180203913753,
-      "learning_rate": 0.0009901282460199829,
-      "loss": 2.6931,
-      "step": 200
-    },
-    {
-      "epoch": 0.0921595598349381,
-      "grad_norm": 1.1636523862028245,
-      "learning_rate": 0.0009899808525182935,
-      "loss": 2.678,
-      "step": 201
-    },
-    {
-      "epoch": 0.09261806510774874,
-      "grad_norm": 0.9636777717779481,
-      "learning_rate": 0.0009898323779361107,
-      "loss": 2.6085,
-      "step": 202
-    },
-    {
-      "epoch": 0.09307657038055937,
-      "grad_norm": 0.8841922398883925,
-      "learning_rate": 0.000989682822601025,
-      "loss": 2.622,
-      "step": 203
-    },
-    {
-      "epoch": 0.09353507565337002,
-      "grad_norm": 0.868904441717743,
-      "learning_rate": 0.0009895321868430113,
-      "loss": 2.6326,
-      "step": 204
-    },
-    {
-      "epoch": 0.09399358092618065,
-      "grad_norm": 0.9791301745752887,
-      "learning_rate": 0.0009893804709944281,
-      "loss": 2.7037,
-      "step": 205
-    },
-    {
-      "epoch": 0.09445208619899129,
-      "grad_norm": 0.8529032547084557,
-      "learning_rate": 0.0009892276753900174,
-      "loss": 2.6404,
-      "step": 206
-    },
-    {
-      "epoch": 0.09491059147180192,
-      "grad_norm": 0.898358663956271,
-      "learning_rate": 0.0009890738003669028,
-      "loss": 2.6708,
-      "step": 207
-    },
-    {
-      "epoch": 0.09536909674461257,
-      "grad_norm": 0.9664888965504623,
-      "learning_rate": 0.0009889188462645904,
-      "loss": 2.6224,
-      "step": 208
-    },
-    {
-      "epoch": 0.0958276020174232,
-      "grad_norm": 0.8427690647316801,
-      "learning_rate": 0.0009887628134249667,
-      "loss": 2.498,
-      "step": 209
-    },
-    {
-      "epoch": 0.09628610729023383,
-      "grad_norm": 0.9058292487128633,
-      "learning_rate": 0.0009886057021922983,
-      "loss": 2.6171,
-      "step": 210
-    },
-    {
-      "epoch": 0.09674461256304448,
-      "grad_norm": 0.9264681517183185,
-      "learning_rate": 0.0009884475129132311,
-      "loss": 2.6377,
-      "step": 211
-    },
-    {
-      "epoch": 0.09720311783585511,
-      "grad_norm": 0.8603539147490995,
-      "learning_rate": 0.0009882882459367897,
-      "loss": 2.6084,
-      "step": 212
-    },
-    {
-      "epoch": 0.09766162310866575,
-      "grad_norm": 0.8803034363411403,
-      "learning_rate": 0.0009881279016143766,
-      "loss": 2.5984,
-      "step": 213
-    },
-    {
-      "epoch": 0.09812012838147639,
-      "grad_norm": 0.8898036059366264,
-      "learning_rate": 0.0009879664802997707,
-      "loss": 2.5254,
-      "step": 214
-    },
-    {
-      "epoch": 0.09857863365428703,
-      "grad_norm": 0.8732477651437989,
-      "learning_rate": 0.000987803982349128,
-      "loss": 2.5809,
-      "step": 215
-    },
-    {
-      "epoch": 0.09903713892709766,
-      "grad_norm": 0.805687794769314,
-      "learning_rate": 0.0009876404081209796,
-      "loss": 2.5912,
-      "step": 216
-    },
-    {
-      "epoch": 0.09949564419990829,
-      "grad_norm": 0.8925140407937739,
-      "learning_rate": 0.000987475757976231,
-      "loss": 2.5253,
-      "step": 217
-    },
-    {
-      "epoch": 0.09995414947271894,
-      "grad_norm": 0.8313766345052097,
-      "learning_rate": 0.000987310032278162,
-      "loss": 2.5746,
-      "step": 218
-    },
-    {
-      "epoch": 0.10041265474552957,
-      "grad_norm": 0.8432478497706013,
-      "learning_rate": 0.0009871432313924254,
-      "loss": 2.6461,
-      "step": 219
-    },
-    {
-      "epoch": 0.1008711600183402,
-      "grad_norm": 0.9069182875126073,
-      "learning_rate": 0.000986975355687046,
-      "loss": 2.5991,
-      "step": 220
-    },
-    {
-      "epoch": 0.10132966529115085,
-      "grad_norm": 0.8728692308183966,
-      "learning_rate": 0.0009868064055324204,
-      "loss": 2.655,
-      "step": 221
-    },
-    {
-      "epoch": 0.10178817056396149,
-      "grad_norm": 0.8197205363218673,
-      "learning_rate": 0.0009866363813013153,
-      "loss": 2.6155,
-      "step": 222
-    },
-    {
-      "epoch": 0.10224667583677212,
-      "grad_norm": 0.8584399712358266,
-      "learning_rate": 0.0009864652833688676,
-      "loss": 2.591,
-      "step": 223
-    },
-    {
-      "epoch": 0.10270518110958277,
-      "grad_norm": 0.7993725265860971,
-      "learning_rate": 0.0009862931121125836,
-      "loss": 2.526,
-      "step": 224
-    },
-    {
-      "epoch": 0.1031636863823934,
-      "grad_norm": 0.7863806689227006,
-      "learning_rate": 0.000986119867912337,
-      "loss": 2.543,
-      "step": 225
-    },
-    {
-      "epoch": 0.10362219165520403,
-      "grad_norm": 0.8275959561867755,
-      "learning_rate": 0.000985945551150369,
-      "loss": 2.572,
-      "step": 226
-    },
-    {
-      "epoch": 0.10408069692801467,
-      "grad_norm": 0.8537180985582624,
-      "learning_rate": 0.0009857701622112876,
-      "loss": 2.5479,
-      "step": 227
-    },
-    {
-      "epoch": 0.10453920220082531,
-      "grad_norm": 0.9019377308965434,
-      "learning_rate": 0.000985593701482066,
-      "loss": 2.5344,
-      "step": 228
-    },
-    {
-      "epoch": 0.10499770747363595,
-      "grad_norm": 0.8026199314349757,
-      "learning_rate": 0.0009854161693520424,
-      "loss": 2.6123,
-      "step": 229
-    },
-    {
-      "epoch": 0.10545621274644658,
-      "grad_norm": 0.8774422113269473,
-      "learning_rate": 0.0009852375662129194,
-      "loss": 2.6137,
-      "step": 230
-    },
-    {
-      "epoch": 0.10591471801925723,
-      "grad_norm": 0.8259393647221029,
-      "learning_rate": 0.0009850578924587613,
-      "loss": 2.5544,
-      "step": 231
-    },
-    {
-      "epoch": 0.10637322329206786,
-      "grad_norm": 0.8385034492535076,
-      "learning_rate": 0.000984877148485996,
-      "loss": 2.5784,
-      "step": 232
-    },
-    {
-      "epoch": 0.10683172856487849,
-      "grad_norm": 0.7886607730829454,
-      "learning_rate": 0.000984695334693412,
-      "loss": 2.5691,
-      "step": 233
-    },
-    {
-      "epoch": 0.10729023383768914,
-      "grad_norm": 1.1157401911987421,
-      "learning_rate": 0.000984512451482158,
-      "loss": 2.5742,
-      "step": 234
-    },
-    {
-      "epoch": 0.10774873911049977,
-      "grad_norm": 0.83627813464661,
-      "learning_rate": 0.0009843284992557431,
-      "loss": 2.4674,
-      "step": 235
-    },
-    {
-      "epoch": 0.1082072443833104,
-      "grad_norm": 0.857477032911942,
-      "learning_rate": 0.000984143478420034,
-      "loss": 2.5851,
-      "step": 236
-    },
-    {
-      "epoch": 0.10866574965612105,
-      "grad_norm": 0.797881046483444,
-      "learning_rate": 0.0009839573893832563,
-      "loss": 2.4562,
-      "step": 237
-    },
-    {
-      "epoch": 0.10912425492893169,
-      "grad_norm": 0.7962270473717661,
-      "learning_rate": 0.000983770232555991,
-      "loss": 2.5276,
-      "step": 238
-    },
-    {
-      "epoch": 0.10958276020174232,
-      "grad_norm": 0.7981365989226473,
-      "learning_rate": 0.0009835820083511765,
-      "loss": 2.5035,
-      "step": 239
-    },
-    {
-      "epoch": 0.11004126547455295,
-      "grad_norm": 0.7892872863742335,
-      "learning_rate": 0.0009833927171841055,
-      "loss": 2.5863,
-      "step": 240
-    },
-    {
-      "epoch": 0.1104997707473636,
-      "grad_norm": 0.8663696602739149,
-      "learning_rate": 0.0009832023594724246,
-      "loss": 2.5975,
-      "step": 241
-    },
-    {
-      "epoch": 0.11095827602017423,
-      "grad_norm": 0.8840096330173806,
-      "learning_rate": 0.0009830109356361344,
-      "loss": 2.5481,
-      "step": 242
-    },
-    {
-      "epoch": 0.11141678129298486,
-      "grad_norm": 0.9614491907876213,
-      "learning_rate": 0.0009828184460975867,
-      "loss": 2.6423,
-      "step": 243
-    },
-    {
-      "epoch": 0.11187528656579551,
-      "grad_norm": 0.945179459967468,
-      "learning_rate": 0.0009826248912814855,
-      "loss": 2.5357,
-      "step": 244
-    },
-    {
-      "epoch": 0.11233379183860615,
-      "grad_norm": 1.7894058980749061,
-      "learning_rate": 0.0009824302716148847,
-      "loss": 2.5543,
-      "step": 245
-    },
-    {
-      "epoch": 0.11279229711141678,
-      "grad_norm": 1.0068679165317196,
-      "learning_rate": 0.0009822345875271884,
-      "loss": 2.5451,
-      "step": 246
-    },
-    {
-      "epoch": 0.11325080238422743,
-      "grad_norm": 0.8577746521481551,
-      "learning_rate": 0.0009820378394501481,
-      "loss": 2.5565,
-      "step": 247
-    },
-    {
-      "epoch": 0.11370930765703806,
-      "grad_norm": 0.9424505802356529,
-      "learning_rate": 0.0009818400278178636,
-      "loss": 2.5089,
-      "step": 248
-    },
-    {
-      "epoch": 0.11416781292984869,
-      "grad_norm": 0.8466953025354731,
-      "learning_rate": 0.0009816411530667814,
-      "loss": 2.6273,
-      "step": 249
-    },
-    {
-      "epoch": 0.11462631820265932,
-      "grad_norm": 0.9674513017981115,
-      "learning_rate": 0.000981441215635693,
-      "loss": 2.5213,
-      "step": 250
-    },
-    {
-      "epoch": 0.11508482347546997,
-      "grad_norm": 1.4404937665231627,
-      "learning_rate": 0.0009812402159657353,
-      "loss": 2.6017,
-      "step": 251
-    },
-    {
-      "epoch": 0.1155433287482806,
-      "grad_norm": 0.8070264899578764,
-      "learning_rate": 0.000981038154500388,
-      "loss": 2.5638,
-      "step": 252
-    },
-    {
-      "epoch": 0.11600183402109124,
-      "grad_norm": 0.8681724320513512,
-      "learning_rate": 0.0009808350316854746,
-      "loss": 2.5813,
-      "step": 253
-    },
-    {
-      "epoch": 0.11646033929390188,
-      "grad_norm": 0.9013105526548899,
-      "learning_rate": 0.0009806308479691594,
-      "loss": 2.4437,
-      "step": 254
-    },
-    {
-      "epoch": 0.11691884456671252,
-      "grad_norm": 0.8285670565410778,
-      "learning_rate": 0.0009804256038019482,
-      "loss": 2.5775,
-      "step": 255
-    },
-    {
-      "epoch": 0.11737734983952315,
-      "grad_norm": 0.9257051158938029,
-      "learning_rate": 0.0009802192996366857,
-      "loss": 2.5405,
-      "step": 256
-    },
-    {
-      "epoch": 0.1178358551123338,
-      "grad_norm": 0.8991399139070363,
-      "learning_rate": 0.0009800119359285563,
-      "loss": 2.572,
-      "step": 257
-    },
-    {
-      "epoch": 0.11829436038514443,
-      "grad_norm": 0.8832897519824833,
-      "learning_rate": 0.0009798035131350813,
-      "loss": 2.5731,
-      "step": 258
-    },
-    {
-      "epoch": 0.11875286565795506,
-      "grad_norm": 0.8884847449208705,
-      "learning_rate": 0.0009795940317161194,
-      "loss": 2.4984,
-      "step": 259
-    },
-    {
-      "epoch": 0.1192113709307657,
-      "grad_norm": 1.1814688514354819,
-      "learning_rate": 0.0009793834921338646,
-      "loss": 2.574,
-      "step": 260
-    },
-    {
-      "epoch": 0.11966987620357634,
-      "grad_norm": 1.5070765013399725,
-      "learning_rate": 0.0009791718948528457,
-      "loss": 2.4977,
-      "step": 261
-    },
-    {
-      "epoch": 0.12012838147638698,
-      "grad_norm": 0.9031004112230903,
-      "learning_rate": 0.0009789592403399252,
-      "loss": 2.4779,
-      "step": 262
-    },
-    {
-      "epoch": 0.12058688674919761,
-      "grad_norm": 0.8114756051919401,
-      "learning_rate": 0.0009787455290642985,
-      "loss": 2.6736,
-      "step": 263
-    },
-    {
-      "epoch": 0.12104539202200826,
-      "grad_norm": 0.8263858248719389,
-      "learning_rate": 0.000978530761497492,
-      "loss": 2.5542,
-      "step": 264
-    },
-    {
-      "epoch": 0.12150389729481889,
-      "grad_norm": 0.8707645814290397,
-      "learning_rate": 0.0009783149381133633,
-      "loss": 2.5256,
-      "step": 265
-    },
-    {
-      "epoch": 0.12196240256762952,
-      "grad_norm": 0.8945045925316212,
-      "learning_rate": 0.0009780980593880992,
-      "loss": 2.562,
-      "step": 266
-    },
-    {
-      "epoch": 0.12242090784044017,
-      "grad_norm": 0.836084718864861,
-      "learning_rate": 0.0009778801258002153,
-      "loss": 2.5398,
-      "step": 267
-    },
-    {
-      "epoch": 0.1228794131132508,
-      "grad_norm": 0.855405686505703,
-      "learning_rate": 0.000977661137830554,
-      "loss": 2.4971,
-      "step": 268
-    },
-    {
-      "epoch": 0.12333791838606144,
-      "grad_norm": 0.9017193190117249,
-      "learning_rate": 0.0009774410959622845,
-      "loss": 2.5965,
-      "step": 269
-    },
-    {
-      "epoch": 0.12379642365887207,
-      "grad_norm": 0.9258034191414712,
-      "learning_rate": 0.000977220000680901,
-      "loss": 2.5083,
-      "step": 270
-    },
-    {
-      "epoch": 0.12425492893168272,
-      "grad_norm": 0.8544623407873332,
-      "learning_rate": 0.000976997852474223,
-      "loss": 2.5583,
-      "step": 271
-    },
-    {
-      "epoch": 0.12471343420449335,
-      "grad_norm": 0.8304185472190929,
-      "learning_rate": 0.0009767746518323914,
-      "loss": 2.5758,
-      "step": 272
-    },
-    {
-      "epoch": 0.12517193947730398,
-      "grad_norm": 0.8461439308676527,
-      "learning_rate": 0.0009765503992478704,
-      "loss": 2.4568,
-      "step": 273
-    },
-    {
-      "epoch": 0.12563044475011462,
-      "grad_norm": 0.7851630962011668,
-      "learning_rate": 0.0009763250952154449,
-      "loss": 2.4484,
-      "step": 274
-    },
-    {
-      "epoch": 0.12608895002292525,
-      "grad_norm": 0.7437596553638574,
-      "learning_rate": 0.0009760987402322195,
-      "loss": 2.4244,
-      "step": 275
-    },
-    {
-      "epoch": 0.1265474552957359,
-      "grad_norm": 0.7645446696581057,
-      "learning_rate": 0.0009758713347976178,
-      "loss": 2.5202,
-      "step": 276
-    },
-    {
-      "epoch": 0.12700596056854654,
-      "grad_norm": 0.7480538457540715,
-      "learning_rate": 0.000975642879413381,
-      "loss": 2.4774,
-      "step": 277
-    },
-    {
-      "epoch": 0.12746446584135718,
-      "grad_norm": 0.747329904296222,
-      "learning_rate": 0.0009754133745835665,
-      "loss": 2.5923,
-      "step": 278
-    },
-    {
-      "epoch": 0.1279229711141678,
-      "grad_norm": 0.8269686858272265,
-      "learning_rate": 0.0009751828208145482,
-      "loss": 2.5485,
-      "step": 279
-    },
-    {
-      "epoch": 0.12838147638697844,
-      "grad_norm": 0.7991184946190629,
-      "learning_rate": 0.0009749512186150131,
-      "loss": 2.4705,
-      "step": 280
-    },
-    {
-      "epoch": 0.12883998165978908,
-      "grad_norm": 0.7524121280026368,
-      "learning_rate": 0.0009747185684959625,
-      "loss": 2.4838,
-      "step": 281
-    },
-    {
-      "epoch": 0.12929848693259974,
-      "grad_norm": 0.8170171703756699,
-      "learning_rate": 0.000974484870970709,
-      "loss": 2.4948,
-      "step": 282
-    },
-    {
-      "epoch": 0.12975699220541037,
-      "grad_norm": 0.814606744128704,
-      "learning_rate": 0.0009742501265548767,
-      "loss": 2.5075,
-      "step": 283
-    },
-    {
-      "epoch": 0.130215497478221,
-      "grad_norm": 0.7552517804098113,
-      "learning_rate": 0.0009740143357663993,
-      "loss": 2.4401,
-      "step": 284
-    },
-    {
-      "epoch": 0.13067400275103164,
-      "grad_norm": 0.7920298129385891,
-      "learning_rate": 0.000973777499125519,
-      "loss": 2.5958,
-      "step": 285
-    },
-    {
-      "epoch": 0.13113250802384227,
-      "grad_norm": 0.7627744471833945,
-      "learning_rate": 0.0009735396171547859,
-      "loss": 2.4618,
-      "step": 286
-    },
-    {
-      "epoch": 0.1315910132966529,
-      "grad_norm": 0.7017523788578347,
-      "learning_rate": 0.0009733006903790564,
-      "loss": 2.5229,
-      "step": 287
-    },
-    {
-      "epoch": 0.13204951856946354,
-      "grad_norm": 0.7303301506333497,
-      "learning_rate": 0.0009730607193254922,
-      "loss": 2.5368,
-      "step": 288
-    },
-    {
-      "epoch": 0.1325080238422742,
-      "grad_norm": 0.7873750314064523,
-      "learning_rate": 0.0009728197045235585,
-      "loss": 2.4305,
-      "step": 289
-    },
-    {
-      "epoch": 0.13296652911508483,
-      "grad_norm": 0.7437438150908053,
-      "learning_rate": 0.0009725776465050242,
-      "loss": 2.5205,
-      "step": 290
-    },
-    {
-      "epoch": 0.13342503438789546,
-      "grad_norm": 0.7090699568501801,
-      "learning_rate": 0.0009723345458039594,
-      "loss": 2.5189,
-      "step": 291
-    },
-    {
-      "epoch": 0.1338835396607061,
-      "grad_norm": 0.7408548041082708,
-      "learning_rate": 0.000972090402956735,
-      "loss": 2.5486,
-      "step": 292
-    },
-    {
-      "epoch": 0.13434204493351673,
-      "grad_norm": 0.7562390407076534,
-      "learning_rate": 0.0009718452185020212,
-      "loss": 2.4372,
-      "step": 293
-    },
-    {
-      "epoch": 0.13480055020632736,
-      "grad_norm": 1.1213582479504831,
-      "learning_rate": 0.0009715989929807862,
-      "loss": 2.506,
-      "step": 294
-    },
-    {
-      "epoch": 0.13525905547913802,
-      "grad_norm": 0.6944503066275778,
-      "learning_rate": 0.0009713517269362955,
-      "loss": 2.4199,
-      "step": 295
-    },
-    {
-      "epoch": 0.13571756075194866,
-      "grad_norm": 0.8365287018747677,
-      "learning_rate": 0.0009711034209141101,
-      "loss": 2.4547,
-      "step": 296
-    },
-    {
-      "epoch": 0.1361760660247593,
-      "grad_norm": 0.7781502847292924,
-      "learning_rate": 0.0009708540754620856,
-      "loss": 2.4739,
-      "step": 297
-    },
-    {
-      "epoch": 0.13663457129756992,
-      "grad_norm": 0.817813640113753,
-      "learning_rate": 0.0009706036911303713,
-      "loss": 2.4713,
-      "step": 298
-    },
-    {
-      "epoch": 0.13709307657038056,
-      "grad_norm": 0.757629111801681,
-      "learning_rate": 0.0009703522684714083,
-      "loss": 2.4535,
-      "step": 299
-    },
-    {
-      "epoch": 0.1375515818431912,
-      "grad_norm": 0.7873595756474979,
-      "learning_rate": 0.0009700998080399286,
-      "loss": 2.4807,
-      "step": 300
-    },
-    {
-      "epoch": 0.13801008711600182,
-      "grad_norm": 0.7372082091360098,
-      "learning_rate": 0.0009698463103929542,
-      "loss": 2.4845,
-      "step": 301
-    },
-    {
-      "epoch": 0.13846859238881248,
-      "grad_norm": 0.8292677379184695,
-      "learning_rate": 0.0009695917760897954,
-      "loss": 2.4355,
-      "step": 302
-    },
-    {
-      "epoch": 0.13892709766162312,
-      "grad_norm": 1.3597460108928376,
-      "learning_rate": 0.0009693362056920501,
-      "loss": 2.5387,
-      "step": 303
-    },
-    {
-      "epoch": 0.13938560293443375,
-      "grad_norm": 0.8769006724059404,
-      "learning_rate": 0.0009690795997636015,
-      "loss": 2.5536,
-      "step": 304
-    },
-    {
-      "epoch": 0.13984410820724438,
-      "grad_norm": 0.8868002796292577,
-      "learning_rate": 0.0009688219588706179,
-      "loss": 2.5367,
-      "step": 305
-    },
-    {
-      "epoch": 0.14030261348005502,
-      "grad_norm": 0.774918955352645,
-      "learning_rate": 0.0009685632835815518,
-      "loss": 2.4599,
-      "step": 306
-    },
-    {
-      "epoch": 0.14076111875286565,
-      "grad_norm": 0.8440718710198023,
-      "learning_rate": 0.0009683035744671367,
-      "loss": 2.4269,
-      "step": 307
-    },
-    {
-      "epoch": 0.14121962402567628,
-      "grad_norm": 0.8341920823494009,
-      "learning_rate": 0.0009680428321003883,
-      "loss": 2.4892,
-      "step": 308
-    },
-    {
-      "epoch": 0.14167812929848694,
-      "grad_norm": 0.7949236235962552,
-      "learning_rate": 0.000967781057056601,
-      "loss": 2.3997,
-      "step": 309
-    },
-    {
-      "epoch": 0.14213663457129758,
-      "grad_norm": 0.8152857779102205,
-      "learning_rate": 0.0009675182499133485,
-      "loss": 2.4652,
-      "step": 310
-    },
-    {
-      "epoch": 0.1425951398441082,
-      "grad_norm": 0.8307917943380024,
-      "learning_rate": 0.0009672544112504813,
-      "loss": 2.4885,
-      "step": 311
-    },
-    {
-      "epoch": 0.14305364511691884,
-      "grad_norm": 1.24091123373615,
-      "learning_rate": 0.0009669895416501257,
-      "loss": 2.589,
-      "step": 312
-    },
-    {
-      "epoch": 0.14351215038972948,
-      "grad_norm": 0.7949633868670231,
-      "learning_rate": 0.0009667236416966833,
-      "loss": 2.5003,
-      "step": 313
-    },
-    {
-      "epoch": 0.1439706556625401,
-      "grad_norm": 0.7509646492274797,
-      "learning_rate": 0.0009664567119768281,
-      "loss": 2.4143,
-      "step": 314
-    },
-    {
-      "epoch": 0.14442916093535077,
-      "grad_norm": 0.7664396075451531,
-      "learning_rate": 0.0009661887530795067,
-      "loss": 2.4732,
-      "step": 315
-    },
-    {
-      "epoch": 0.1448876662081614,
-      "grad_norm": 0.8489182939136704,
-      "learning_rate": 0.0009659197655959365,
-      "loss": 2.5938,
-      "step": 316
-    },
-    {
-      "epoch": 0.14534617148097204,
-      "grad_norm": 0.816093382022774,
-      "learning_rate": 0.000965649750119604,
-      "loss": 2.4537,
-      "step": 317
-    },
-    {
-      "epoch": 0.14580467675378267,
-      "grad_norm": 0.7603026394669137,
-      "learning_rate": 0.0009653787072462643,
-      "loss": 2.5234,
-      "step": 318
-    },
-    {
-      "epoch": 0.1462631820265933,
-      "grad_norm": 0.7558215203281234,
-      "learning_rate": 0.0009651066375739388,
-      "loss": 2.504,
-      "step": 319
-    },
-    {
-      "epoch": 0.14672168729940394,
-      "grad_norm": 0.8252374516963078,
-      "learning_rate": 0.000964833541702915,
-      "loss": 2.5244,
-      "step": 320
-    },
-    {
-      "epoch": 0.14718019257221457,
-      "grad_norm": 0.8118569604524664,
-      "learning_rate": 0.0009645594202357438,
-      "loss": 2.454,
-      "step": 321
-    },
-    {
-      "epoch": 0.14763869784502523,
-      "grad_norm": 0.8394209102264499,
-      "learning_rate": 0.0009642842737772397,
-      "loss": 2.3292,
-      "step": 322
-    },
-    {
-      "epoch": 0.14809720311783586,
-      "grad_norm": 1.005690261364726,
-      "learning_rate": 0.0009640081029344782,
-      "loss": 2.4252,
-      "step": 323
-    },
-    {
-      "epoch": 0.1485557083906465,
-      "grad_norm": 0.7860130298769012,
-      "learning_rate": 0.0009637309083167956,
-      "loss": 2.4447,
-      "step": 324
-    },
-    {
-      "epoch": 0.14901421366345713,
-      "grad_norm": 0.8332590619221837,
-      "learning_rate": 0.0009634526905357859,
-      "loss": 2.436,
-      "step": 325
-    },
-    {
-      "epoch": 0.14947271893626776,
-      "grad_norm": 0.713902986244657,
-      "learning_rate": 0.000963173450205302,
-      "loss": 2.4863,
-      "step": 326
-    },
-    {
-      "epoch": 0.1499312242090784,
-      "grad_norm": 0.7439034461336351,
-      "learning_rate": 0.0009628931879414517,
-      "loss": 2.5582,
-      "step": 327
-    },
-    {
-      "epoch": 0.15038972948188903,
-      "grad_norm": 0.741114275865419,
-      "learning_rate": 0.0009626119043625983,
-      "loss": 2.6073,
-      "step": 328
-    },
-    {
-      "epoch": 0.1508482347546997,
-      "grad_norm": 0.7145763754447997,
-      "learning_rate": 0.0009623296000893582,
-      "loss": 2.4297,
-      "step": 329
-    },
-    {
-      "epoch": 0.15130674002751032,
-      "grad_norm": 0.697589193127181,
-      "learning_rate": 0.0009620462757446,
-      "loss": 2.5055,
-      "step": 330
-    },
-    {
-      "epoch": 0.15176524530032096,
-      "grad_norm": 0.7610210481763144,
-      "learning_rate": 0.0009617619319534428,
-      "loss": 2.5152,
-      "step": 331
-    },
-    {
-      "epoch": 0.1522237505731316,
-      "grad_norm": 0.74177442116675,
-      "learning_rate": 0.000961476569343255,
-      "loss": 2.56,
-      "step": 332
-    },
-    {
-      "epoch": 0.15268225584594222,
-      "grad_norm": 0.6990964552151436,
-      "learning_rate": 0.0009611901885436529,
-      "loss": 2.5011,
-      "step": 333
-    },
-    {
-      "epoch": 0.15314076111875286,
-      "grad_norm": 0.6958906393419879,
-      "learning_rate": 0.0009609027901864996,
-      "loss": 2.4934,
-      "step": 334
-    },
-    {
-      "epoch": 0.15359926639156352,
-      "grad_norm": 0.8134790740483817,
-      "learning_rate": 0.0009606143749059029,
-      "loss": 2.5048,
-      "step": 335
-    },
-    {
-      "epoch": 0.15405777166437415,
-      "grad_norm": 0.7136731774532791,
-      "learning_rate": 0.0009603249433382144,
-      "loss": 2.4244,
-      "step": 336
-    },
-    {
-      "epoch": 0.15451627693718478,
-      "grad_norm": 0.761671531265195,
-      "learning_rate": 0.0009600344961220282,
-      "loss": 2.4173,
-      "step": 337
-    },
-    {
-      "epoch": 0.15497478220999542,
-      "grad_norm": 0.7197276238986177,
-      "learning_rate": 0.0009597430338981791,
-      "loss": 2.4663,
-      "step": 338
-    },
-    {
-      "epoch": 0.15543328748280605,
-      "grad_norm": 0.7453609729569829,
-      "learning_rate": 0.0009594505573097414,
-      "loss": 2.4829,
-      "step": 339
-    },
-    {
-      "epoch": 0.15589179275561668,
-      "grad_norm": 0.7261905384589481,
-      "learning_rate": 0.0009591570670020277,
-      "loss": 2.4735,
-      "step": 340
-    },
-    {
-      "epoch": 0.15635029802842731,
-      "grad_norm": 0.8689035285134682,
-      "learning_rate": 0.0009588625636225871,
-      "loss": 2.5105,
-      "step": 341
-    },
-    {
-      "epoch": 0.15680880330123798,
-      "grad_norm": 0.7578896152694875,
-      "learning_rate": 0.0009585670478212036,
-      "loss": 2.5498,
-      "step": 342
-    },
-    {
-      "epoch": 0.1572673085740486,
-      "grad_norm": 0.7920529034174378,
-      "learning_rate": 0.0009582705202498956,
-      "loss": 2.4992,
-      "step": 343
-    },
-    {
-      "epoch": 0.15772581384685924,
-      "grad_norm": 0.808159644854977,
-      "learning_rate": 0.0009579729815629133,
-      "loss": 2.4294,
-      "step": 344
-    },
-    {
-      "epoch": 0.15818431911966988,
-      "grad_norm": 1.2631614610865987,
-      "learning_rate": 0.0009576744324167379,
-      "loss": 2.3992,
-      "step": 345
-    },
-    {
-      "epoch": 0.1586428243924805,
-      "grad_norm": 0.6931349978933036,
-      "learning_rate": 0.0009573748734700804,
-      "loss": 2.5483,
-      "step": 346
-    },
-    {
-      "epoch": 0.15910132966529114,
-      "grad_norm": 0.7782081821779203,
-      "learning_rate": 0.0009570743053838796,
-      "loss": 2.4443,
-      "step": 347
-    },
-    {
-      "epoch": 0.15955983493810177,
-      "grad_norm": 0.7492100189190506,
-      "learning_rate": 0.0009567727288213005,
-      "loss": 2.5603,
-      "step": 348
-    },
-    {
-      "epoch": 0.16001834021091244,
-      "grad_norm": 0.793393771070654,
-      "learning_rate": 0.0009564701444477337,
-      "loss": 2.4843,
-      "step": 349
-    },
-    {
-      "epoch": 0.16047684548372307,
-      "grad_norm": 0.7240139128884583,
-      "learning_rate": 0.000956166552930793,
-      "loss": 2.4368,
-      "step": 350
-    },
-    {
-      "epoch": 0.1609353507565337,
-      "grad_norm": 0.9097846581379125,
-      "learning_rate": 0.0009558619549403147,
-      "loss": 2.5594,
-      "step": 351
-    },
-    {
-      "epoch": 0.16139385602934433,
-      "grad_norm": 0.7076395550340109,
-      "learning_rate": 0.0009555563511483555,
-      "loss": 2.5106,
-      "step": 352
-    },
-    {
-      "epoch": 0.16185236130215497,
-      "grad_norm": 0.7438938036282859,
-      "learning_rate": 0.0009552497422291912,
-      "loss": 2.5447,
-      "step": 353
-    },
-    {
-      "epoch": 0.1623108665749656,
-      "grad_norm": 3.9776359844129816,
-      "learning_rate": 0.0009549421288593157,
-      "loss": 2.5393,
-      "step": 354
-    },
-    {
-      "epoch": 0.16276937184777626,
-      "grad_norm": 0.74218772336053,
-      "learning_rate": 0.0009546335117174385,
-      "loss": 2.468,
-      "step": 355
-    },
-    {
-      "epoch": 0.1632278771205869,
-      "grad_norm": 0.8137854453929196,
-      "learning_rate": 0.0009543238914844843,
-      "loss": 2.5098,
-      "step": 356
-    },
-    {
-      "epoch": 0.16368638239339753,
-      "grad_norm": 0.7743151786203668,
-      "learning_rate": 0.0009540132688435907,
-      "loss": 2.5511,
-      "step": 357
-    },
-    {
-      "epoch": 0.16414488766620816,
-      "grad_norm": 0.6742836763337464,
-      "learning_rate": 0.0009537016444801074,
-      "loss": 2.506,
-      "step": 358
-    },
-    {
-      "epoch": 0.1646033929390188,
-      "grad_norm": 0.7598181718022291,
-      "learning_rate": 0.0009533890190815935,
-      "loss": 2.5043,
-      "step": 359
-    },
-    {
-      "epoch": 0.16506189821182943,
-      "grad_norm": 0.6996976418540687,
-      "learning_rate": 0.0009530753933378173,
-      "loss": 2.5136,
-      "step": 360
-    },
-    {
-      "epoch": 0.16552040348464006,
-      "grad_norm": 0.715119360533776,
-      "learning_rate": 0.0009527607679407545,
-      "loss": 2.4374,
-      "step": 361
-    },
-    {
-      "epoch": 0.16597890875745072,
-      "grad_norm": 0.7048041390553927,
-      "learning_rate": 0.0009524451435845857,
-      "loss": 2.4928,
-      "step": 362
-    },
-    {
-      "epoch": 0.16643741403026135,
-      "grad_norm": 0.7748620483098821,
-      "learning_rate": 0.0009521285209656963,
-      "loss": 2.5043,
-      "step": 363
-    },
-    {
-      "epoch": 0.166895919303072,
-      "grad_norm": 0.7009460970985532,
-      "learning_rate": 0.0009518109007826734,
-      "loss": 2.4696,
-      "step": 364
-    },
-    {
-      "epoch": 0.16735442457588262,
-      "grad_norm": 0.6897838744075913,
-      "learning_rate": 0.0009514922837363059,
-      "loss": 2.4914,
-      "step": 365
-    },
-    {
-      "epoch": 0.16781292984869325,
-      "grad_norm": 0.7620296598745006,
-      "learning_rate": 0.0009511726705295817,
-      "loss": 2.4067,
-      "step": 366
-    },
-    {
-      "epoch": 0.1682714351215039,
-      "grad_norm": 0.6445615139994112,
-      "learning_rate": 0.000950852061867687,
-      "loss": 2.5036,
-      "step": 367
-    },
-    {
-      "epoch": 0.16872994039431455,
-      "grad_norm": 0.7003147663122766,
-      "learning_rate": 0.0009505304584580038,
-      "loss": 2.5692,
-      "step": 368
-    },
-    {
-      "epoch": 0.16918844566712518,
-      "grad_norm": 0.6708398176317568,
-      "learning_rate": 0.0009502078610101092,
-      "loss": 2.527,
-      "step": 369
-    },
-    {
-      "epoch": 0.16964695093993581,
-      "grad_norm": 0.6679995990483795,
-      "learning_rate": 0.0009498842702357736,
-      "loss": 2.4838,
-      "step": 370
-    },
-    {
-      "epoch": 0.17010545621274645,
-      "grad_norm": 0.6805627992468459,
-      "learning_rate": 0.0009495596868489587,
-      "loss": 2.5933,
-      "step": 371
-    },
-    {
-      "epoch": 0.17056396148555708,
-      "grad_norm": 0.7045910373289027,
-      "learning_rate": 0.0009492341115658167,
-      "loss": 2.4895,
-      "step": 372
-    },
-    {
-      "epoch": 0.1710224667583677,
-      "grad_norm": 0.7603140796767655,
-      "learning_rate": 0.0009489075451046879,
-      "loss": 2.5619,
-      "step": 373
-    },
-    {
-      "epoch": 0.17148097203117835,
-      "grad_norm": 0.6594595106795479,
-      "learning_rate": 0.0009485799881861,
-      "loss": 2.5626,
-      "step": 374
-    },
-    {
-      "epoch": 0.171939477303989,
-      "grad_norm": 0.6504438571080672,
-      "learning_rate": 0.0009482514415327654,
-      "loss": 2.5188,
-      "step": 375
-    },
-    {
-      "epoch": 0.17239798257679964,
-      "grad_norm": 0.7300360272548702,
-      "learning_rate": 0.000947921905869581,
-      "loss": 2.4195,
-      "step": 376
-    },
-    {
-      "epoch": 0.17285648784961027,
-      "grad_norm": 0.7167498237904364,
-      "learning_rate": 0.0009475913819236248,
-      "loss": 2.5,
-      "step": 377
-    },
-    {
-      "epoch": 0.1733149931224209,
-      "grad_norm": 0.6325571154143905,
-      "learning_rate": 0.0009472598704241561,
-      "loss": 2.4793,
-      "step": 378
-    },
-    {
-      "epoch": 0.17377349839523154,
-      "grad_norm": 0.7067019949939857,
-      "learning_rate": 0.0009469273721026131,
-      "loss": 2.3391,
-      "step": 379
-    },
-    {
-      "epoch": 0.17423200366804217,
-      "grad_norm": 0.7375313217903855,
-      "learning_rate": 0.0009465938876926111,
-      "loss": 2.5098,
-      "step": 380
-    },
-    {
-      "epoch": 0.1746905089408528,
-      "grad_norm": 0.7090199960083187,
-      "learning_rate": 0.0009462594179299406,
-      "loss": 2.4606,
-      "step": 381
-    },
-    {
-      "epoch": 0.17514901421366347,
-      "grad_norm": 0.6837019024171153,
-      "learning_rate": 0.0009459239635525672,
-      "loss": 2.4614,
-      "step": 382
-    },
-    {
-      "epoch": 0.1756075194864741,
-      "grad_norm": 0.7557117277292827,
-      "learning_rate": 0.0009455875253006281,
-      "loss": 2.4704,
-      "step": 383
-    },
-    {
-      "epoch": 0.17606602475928473,
-      "grad_norm": 0.6636145107929177,
-      "learning_rate": 0.0009452501039164315,
-      "loss": 2.486,
-      "step": 384
-    },
-    {
-      "epoch": 0.17652453003209537,
-      "grad_norm": 0.6735018112813426,
-      "learning_rate": 0.0009449117001444549,
-      "loss": 2.4724,
-      "step": 385
-    },
-    {
-      "epoch": 0.176983035304906,
-      "grad_norm": 0.6622018494562905,
-      "learning_rate": 0.0009445723147313433,
-      "loss": 2.4624,
-      "step": 386
-    },
-    {
-      "epoch": 0.17744154057771663,
-      "grad_norm": 0.647873163251299,
-      "learning_rate": 0.0009442319484259074,
-      "loss": 2.4773,
-      "step": 387
-    },
-    {
-      "epoch": 0.1779000458505273,
-      "grad_norm": 0.6456408064291912,
-      "learning_rate": 0.0009438906019791222,
-      "loss": 2.4891,
-      "step": 388
-    },
-    {
-      "epoch": 0.17835855112333793,
-      "grad_norm": 0.6464738124664497,
-      "learning_rate": 0.0009435482761441251,
-      "loss": 2.4984,
-      "step": 389
-    },
-    {
-      "epoch": 0.17881705639614856,
-      "grad_norm": 0.6706534869283223,
-      "learning_rate": 0.000943204971676215,
-      "loss": 2.4794,
-      "step": 390
-    },
-    {
-      "epoch": 0.1792755616689592,
-      "grad_norm": 0.6319128738446849,
-      "learning_rate": 0.0009428606893328493,
-      "loss": 2.4219,
-      "step": 391
-    },
-    {
-      "epoch": 0.17973406694176983,
-      "grad_norm": 0.6384020506614136,
-      "learning_rate": 0.0009425154298736432,
-      "loss": 2.4505,
-      "step": 392
-    },
-    {
-      "epoch": 0.18019257221458046,
-      "grad_norm": 0.6259673632319245,
-      "learning_rate": 0.0009421691940603678,
-      "loss": 2.4508,
-      "step": 393
-    },
-    {
-      "epoch": 0.1806510774873911,
-      "grad_norm": 0.6634237832583242,
-      "learning_rate": 0.0009418219826569488,
-      "loss": 2.5295,
-      "step": 394
-    },
-    {
-      "epoch": 0.18110958276020175,
-      "grad_norm": 0.8229975317039904,
-      "learning_rate": 0.0009414737964294635,
-      "loss": 2.434,
-      "step": 395
-    },
-    {
-      "epoch": 0.1815680880330124,
-      "grad_norm": 0.7540097391050903,
-      "learning_rate": 0.000941124636146141,
-      "loss": 2.5144,
-      "step": 396
-    },
-    {
-      "epoch": 0.18202659330582302,
-      "grad_norm": 0.6655663557926031,
-      "learning_rate": 0.0009407745025773589,
-      "loss": 2.4118,
-      "step": 397
-    },
-    {
-      "epoch": 0.18248509857863365,
-      "grad_norm": 0.6940193775837854,
-      "learning_rate": 0.0009404233964956423,
-      "loss": 2.3767,
-      "step": 398
-    },
-    {
-      "epoch": 0.1829436038514443,
-      "grad_norm": 0.6529303949530829,
-      "learning_rate": 0.0009400713186756625,
-      "loss": 2.4281,
-      "step": 399
-    },
-    {
-      "epoch": 0.18340210912425492,
-      "grad_norm": 0.747328546560019,
-      "learning_rate": 0.0009397182698942342,
-      "loss": 2.5048,
-      "step": 400
-    },
-    {
-      "epoch": 0.18386061439706555,
-      "grad_norm": 0.640203977626051,
-      "learning_rate": 0.0009393642509303149,
-      "loss": 2.385,
-      "step": 401
-    },
-    {
-      "epoch": 0.1843191196698762,
-      "grad_norm": 0.6762399384672041,
-      "learning_rate": 0.0009390092625650023,
-      "loss": 2.4985,
-      "step": 402
-    },
-    {
-      "epoch": 0.18477762494268685,
-      "grad_norm": 0.6095050610309389,
-      "learning_rate": 0.0009386533055815332,
-      "loss": 2.4301,
-      "step": 403
-    },
-    {
-      "epoch": 0.18523613021549748,
-      "grad_norm": 0.6529479909936484,
-      "learning_rate": 0.0009382963807652813,
-      "loss": 2.5307,
-      "step": 404
-    },
-    {
-      "epoch": 0.1856946354883081,
-      "grad_norm": 0.6456318712988236,
-      "learning_rate": 0.000937938488903756,
-      "loss": 2.4612,
-      "step": 405
-    },
-    {
-      "epoch": 0.18615314076111875,
-      "grad_norm": 0.6123131513010611,
-      "learning_rate": 0.0009375796307866003,
-      "loss": 2.463,
-      "step": 406
-    },
-    {
-      "epoch": 0.18661164603392938,
-      "grad_norm": 0.6519140863672865,
-      "learning_rate": 0.0009372198072055888,
-      "loss": 2.4406,
-      "step": 407
-    },
-    {
-      "epoch": 0.18707015130674004,
-      "grad_norm": 0.7563658931274678,
-      "learning_rate": 0.0009368590189546268,
-      "loss": 2.3888,
-      "step": 408
-    },
-    {
-      "epoch": 0.18752865657955067,
-      "grad_norm": 0.6765602000139052,
-      "learning_rate": 0.0009364972668297474,
-      "loss": 2.498,
-      "step": 409
-    },
-    {
-      "epoch": 0.1879871618523613,
-      "grad_norm": 0.6534111418828514,
-      "learning_rate": 0.0009361345516291111,
-      "loss": 2.4813,
-      "step": 410
-    },
-    {
-      "epoch": 0.18844566712517194,
-      "grad_norm": 0.6990942837577676,
-      "learning_rate": 0.0009357708741530024,
-      "loss": 2.4579,
-      "step": 411
-    },
-    {
-      "epoch": 0.18890417239798257,
-      "grad_norm": 0.6309549291407727,
-      "learning_rate": 0.00093540623520383,
-      "loss": 2.3471,
-      "step": 412
-    },
-    {
-      "epoch": 0.1893626776707932,
-      "grad_norm": 0.6482908168452525,
-      "learning_rate": 0.000935040635586123,
-      "loss": 2.415,
-      "step": 413
-    },
-    {
-      "epoch": 0.18982118294360384,
-      "grad_norm": 0.6385744743124974,
-      "learning_rate": 0.0009346740761065305,
-      "loss": 2.5046,
-      "step": 414
-    },
-    {
-      "epoch": 0.1902796882164145,
-      "grad_norm": 0.5891567794460227,
-      "learning_rate": 0.0009343065575738197,
-      "loss": 2.4787,
-      "step": 415
-    },
-    {
-      "epoch": 0.19073819348922513,
-      "grad_norm": 0.6219816874607428,
-      "learning_rate": 0.0009339380807988733,
-      "loss": 2.4682,
-      "step": 416
-    },
-    {
-      "epoch": 0.19119669876203577,
-      "grad_norm": 0.613031668735731,
-      "learning_rate": 0.0009335686465946887,
-      "loss": 2.3315,
-      "step": 417
-    },
-    {
-      "epoch": 0.1916552040348464,
-      "grad_norm": 0.6554529154618803,
-      "learning_rate": 0.0009331982557763754,
-      "loss": 2.3863,
-      "step": 418
-    },
-    {
-      "epoch": 0.19211370930765703,
-      "grad_norm": 0.6296659595334743,
-      "learning_rate": 0.0009328269091611537,
-      "loss": 2.3759,
-      "step": 419
-    },
-    {
-      "epoch": 0.19257221458046767,
-      "grad_norm": 0.6586472632234228,
-      "learning_rate": 0.0009324546075683524,
-      "loss": 2.4305,
-      "step": 420
-    },
-    {
-      "epoch": 0.19303071985327833,
-      "grad_norm": 0.6764279689488539,
-      "learning_rate": 0.0009320813518194083,
-      "loss": 2.4758,
-      "step": 421
-    },
-    {
-      "epoch": 0.19348922512608896,
-      "grad_norm": 0.671230171332417,
-      "learning_rate": 0.0009317071427378624,
-      "loss": 2.3971,
-      "step": 422
-    },
-    {
-      "epoch": 0.1939477303988996,
-      "grad_norm": 0.6880891281327903,
-      "learning_rate": 0.0009313319811493594,
-      "loss": 2.3712,
-      "step": 423
-    },
-    {
-      "epoch": 0.19440623567171023,
-      "grad_norm": 0.7584257072076925,
-      "learning_rate": 0.000930955867881646,
-      "loss": 2.4859,
-      "step": 424
-    },
-    {
-      "epoch": 0.19486474094452086,
-      "grad_norm": 0.8111016672824936,
-      "learning_rate": 0.0009305788037645681,
-      "loss": 2.4122,
-      "step": 425
-    },
-    {
-      "epoch": 0.1953232462173315,
-      "grad_norm": 0.6531359030244724,
-      "learning_rate": 0.0009302007896300697,
-      "loss": 2.4057,
-      "step": 426
-    },
-    {
-      "epoch": 0.19578175149014213,
-      "grad_norm": 0.630254967850073,
-      "learning_rate": 0.0009298218263121911,
-      "loss": 2.4026,
-      "step": 427
-    },
-    {
-      "epoch": 0.19624025676295279,
-      "grad_norm": 0.7037653917450202,
-      "learning_rate": 0.0009294419146470668,
-      "loss": 2.4657,
-      "step": 428
-    },
-    {
-      "epoch": 0.19669876203576342,
-      "grad_norm": 0.6891796184855002,
-      "learning_rate": 0.0009290610554729234,
-      "loss": 2.4917,
-      "step": 429
-    },
-    {
-      "epoch": 0.19715726730857405,
-      "grad_norm": 0.6284961783300207,
-      "learning_rate": 0.0009286792496300784,
-      "loss": 2.4242,
-      "step": 430
-    },
-    {
-      "epoch": 0.19761577258138469,
-      "grad_norm": 0.6913430840388772,
-      "learning_rate": 0.0009282964979609379,
-      "loss": 2.4664,
-      "step": 431
-    },
-    {
-      "epoch": 0.19807427785419532,
-      "grad_norm": 0.6551259033929393,
-      "learning_rate": 0.0009279128013099947,
-      "loss": 2.5306,
-      "step": 432
-    },
-    {
-      "epoch": 0.19853278312700595,
-      "grad_norm": 0.6657712199899886,
-      "learning_rate": 0.0009275281605238268,
-      "loss": 2.5235,
-      "step": 433
-    },
-    {
-      "epoch": 0.19899128839981658,
-      "grad_norm": 0.660846508754099,
-      "learning_rate": 0.0009271425764510953,
-      "loss": 2.455,
-      "step": 434
-    },
-    {
-      "epoch": 0.19944979367262725,
-      "grad_norm": 0.6480067237656526,
-      "learning_rate": 0.0009267560499425423,
-      "loss": 2.4534,
-      "step": 435
-    },
-    {
-      "epoch": 0.19990829894543788,
-      "grad_norm": 0.659627171616556,
-      "learning_rate": 0.0009263685818509895,
-      "loss": 2.5122,
-      "step": 436
-    },
-    {
-      "epoch": 0.2003668042182485,
-      "grad_norm": 0.6845722899088686,
-      "learning_rate": 0.000925980173031336,
-      "loss": 2.4593,
-      "step": 437
-    },
-    {
-      "epoch": 0.20082530949105915,
-      "grad_norm": 0.663738159253785,
-      "learning_rate": 0.0009255908243405567,
-      "loss": 2.4365,
-      "step": 438
-    },
-    {
-      "epoch": 0.20128381476386978,
-      "grad_norm": 0.6451297419675094,
-      "learning_rate": 0.0009252005366376996,
-      "loss": 2.4524,
-      "step": 439
-    },
-    {
-      "epoch": 0.2017423200366804,
-      "grad_norm": 0.6558884085004961,
-      "learning_rate": 0.0009248093107838852,
-      "loss": 2.4576,
-      "step": 440
-    },
-    {
-      "epoch": 0.20220082530949107,
-      "grad_norm": 0.6690837334032369,
-      "learning_rate": 0.0009244171476423036,
-      "loss": 2.4225,
-      "step": 441
-    },
-    {
-      "epoch": 0.2026593305823017,
-      "grad_norm": 0.6455798509281659,
-      "learning_rate": 0.0009240240480782129,
-      "loss": 2.4194,
-      "step": 442
-    },
-    {
-      "epoch": 0.20311783585511234,
-      "grad_norm": 0.6727381041004469,
-      "learning_rate": 0.0009236300129589376,
-      "loss": 2.4427,
-      "step": 443
-    },
-    {
-      "epoch": 0.20357634112792297,
-      "grad_norm": 0.8197174783780022,
-      "learning_rate": 0.0009232350431538657,
-      "loss": 2.4315,
-      "step": 444
-    },
-    {
-      "epoch": 0.2040348464007336,
-      "grad_norm": 0.7004923822103954,
-      "learning_rate": 0.0009228391395344482,
-      "loss": 2.3716,
-      "step": 445
-    },
-    {
-      "epoch": 0.20449335167354424,
-      "grad_norm": 0.647513342515878,
-      "learning_rate": 0.000922442302974196,
-      "loss": 2.4102,
-      "step": 446
-    },
-    {
-      "epoch": 0.20495185694635487,
-      "grad_norm": 0.6697139886630802,
-      "learning_rate": 0.0009220445343486785,
-      "loss": 2.4885,
-      "step": 447
-    },
-    {
-      "epoch": 0.20541036221916553,
-      "grad_norm": 0.7203379942784551,
-      "learning_rate": 0.0009216458345355217,
-      "loss": 2.42,
-      "step": 448
-    },
-    {
-      "epoch": 0.20586886749197617,
-      "grad_norm": 0.6556099934850383,
-      "learning_rate": 0.0009212462044144061,
-      "loss": 2.4964,
-      "step": 449
-    },
-    {
-      "epoch": 0.2063273727647868,
-      "grad_norm": 0.7029353766914647,
-      "learning_rate": 0.0009208456448670648,
-      "loss": 2.4039,
-      "step": 450
-    },
-    {
-      "epoch": 0.20678587803759743,
-      "grad_norm": 0.6514711173771484,
-      "learning_rate": 0.0009204441567772816,
-      "loss": 2.4214,
-      "step": 451
-    },
-    {
-      "epoch": 0.20724438331040806,
-      "grad_norm": 0.6305882593618566,
-      "learning_rate": 0.0009200417410308888,
-      "loss": 2.4797,
-      "step": 452
-    },
-    {
-      "epoch": 0.2077028885832187,
-      "grad_norm": 0.6596088517056036,
-      "learning_rate": 0.0009196383985157656,
-      "loss": 2.3545,
-      "step": 453
-    },
-    {
-      "epoch": 0.20816139385602933,
-      "grad_norm": 0.6899286034078523,
-      "learning_rate": 0.000919234130121836,
-      "loss": 2.4377,
-      "step": 454
-    },
-    {
-      "epoch": 0.20861989912884,
-      "grad_norm": 0.6818442225055324,
-      "learning_rate": 0.0009188289367410672,
-      "loss": 2.4392,
-      "step": 455
-    },
-    {
-      "epoch": 0.20907840440165062,
-      "grad_norm": 0.674073990940869,
-      "learning_rate": 0.0009184228192674666,
-      "loss": 2.48,
-      "step": 456
-    },
-    {
-      "epoch": 0.20953690967446126,
-      "grad_norm": 0.6505991970739209,
-      "learning_rate": 0.0009180157785970808,
-      "loss": 2.3656,
-      "step": 457
-    },
-    {
-      "epoch": 0.2099954149472719,
-      "grad_norm": 0.6394606270871431,
-      "learning_rate": 0.0009176078156279932,
-      "loss": 2.4789,
-      "step": 458
-    },
-    {
-      "epoch": 0.21045392022008252,
-      "grad_norm": 0.6445404355521662,
-      "learning_rate": 0.0009171989312603226,
-      "loss": 2.4287,
-      "step": 459
-    },
-    {
-      "epoch": 0.21091242549289316,
-      "grad_norm": 0.6745457815017084,
-      "learning_rate": 0.0009167891263962202,
-      "loss": 2.3418,
-      "step": 460
-    },
-    {
-      "epoch": 0.21137093076570382,
-      "grad_norm": 0.5762108238035328,
-      "learning_rate": 0.0009163784019398685,
-      "loss": 2.3483,
-      "step": 461
-    },
-    {
-      "epoch": 0.21182943603851445,
-      "grad_norm": 0.7394876853395154,
-      "learning_rate": 0.0009159667587974785,
-      "loss": 2.4782,
-      "step": 462
-    },
-    {
-      "epoch": 0.21228794131132508,
-      "grad_norm": 0.6264216451996296,
-      "learning_rate": 0.0009155541978772887,
-      "loss": 2.3946,
-      "step": 463
-    },
-    {
-      "epoch": 0.21274644658413572,
-      "grad_norm": 0.627025988465465,
-      "learning_rate": 0.0009151407200895625,
-      "loss": 2.4751,
-      "step": 464
-    },
-    {
-      "epoch": 0.21320495185694635,
-      "grad_norm": 0.6175081121602854,
-      "learning_rate": 0.000914726326346586,
-      "loss": 2.3738,
-      "step": 465
-    },
-    {
-      "epoch": 0.21366345712975698,
-      "grad_norm": 0.6336004300873654,
-      "learning_rate": 0.0009143110175626661,
-      "loss": 2.4049,
-      "step": 466
-    },
-    {
-      "epoch": 0.21412196240256762,
-      "grad_norm": 0.6029487554792996,
-      "learning_rate": 0.0009138947946541291,
-      "loss": 2.4125,
-      "step": 467
-    },
-    {
-      "epoch": 0.21458046767537828,
-      "grad_norm": 0.6499478243986,
-      "learning_rate": 0.0009134776585393181,
-      "loss": 2.4473,
-      "step": 468
-    },
-    {
-      "epoch": 0.2150389729481889,
-      "grad_norm": 0.6092853212697944,
-      "learning_rate": 0.0009130596101385906,
-      "loss": 2.3732,
-      "step": 469
-    },
-    {
-      "epoch": 0.21549747822099954,
-      "grad_norm": 0.6081814796225781,
-      "learning_rate": 0.0009126406503743174,
-      "loss": 2.3616,
-      "step": 470
-    },
-    {
-      "epoch": 0.21595598349381018,
-      "grad_norm": 0.6322757993298066,
-      "learning_rate": 0.0009122207801708802,
-      "loss": 2.3785,
-      "step": 471
-    },
-    {
-      "epoch": 0.2164144887666208,
-      "grad_norm": 0.6777772824586898,
-      "learning_rate": 0.0009118000004546689,
-      "loss": 2.441,
-      "step": 472
-    },
-    {
-      "epoch": 0.21687299403943144,
-      "grad_norm": 0.6840724052661372,
-      "learning_rate": 0.0009113783121540807,
-      "loss": 2.3955,
-      "step": 473
-    },
-    {
-      "epoch": 0.2173314993122421,
-      "grad_norm": 0.6955931862455399,
-      "learning_rate": 0.0009109557161995172,
-      "loss": 2.3662,
-      "step": 474
-    },
-    {
-      "epoch": 0.21779000458505274,
-      "grad_norm": 0.6625097187612561,
-      "learning_rate": 0.0009105322135233828,
-      "loss": 2.3336,
-      "step": 475
-    },
-    {
-      "epoch": 0.21824850985786337,
-      "grad_norm": 0.7015368022014182,
-      "learning_rate": 0.0009101078050600821,
-      "loss": 2.4815,
-      "step": 476
-    },
-    {
-      "epoch": 0.218707015130674,
-      "grad_norm": 0.6239769835113129,
-      "learning_rate": 0.0009096824917460186,
-      "loss": 2.3479,
-      "step": 477
-    },
-    {
-      "epoch": 0.21916552040348464,
-      "grad_norm": 0.5745790682657247,
-      "learning_rate": 0.0009092562745195921,
-      "loss": 2.379,
-      "step": 478
-    },
-    {
-      "epoch": 0.21962402567629527,
-      "grad_norm": 0.6543245686890503,
-      "learning_rate": 0.0009088291543211967,
-      "loss": 2.4277,
-      "step": 479
-    },
-    {
-      "epoch": 0.2200825309491059,
-      "grad_norm": 0.6453109241118828,
-      "learning_rate": 0.0009084011320932188,
-      "loss": 2.511,
-      "step": 480
-    },
-    {
-      "epoch": 0.22054103622191656,
-      "grad_norm": 2.4877928085252226,
-      "learning_rate": 0.0009079722087800352,
-      "loss": 2.4683,
-      "step": 481
-    },
-    {
-      "epoch": 0.2209995414947272,
-      "grad_norm": 0.647300962064528,
-      "learning_rate": 0.0009075423853280106,
-      "loss": 2.452,
-      "step": 482
-    },
-    {
-      "epoch": 0.22145804676753783,
-      "grad_norm": 0.6425303001854006,
-      "learning_rate": 0.0009071116626854958,
-      "loss": 2.4651,
-      "step": 483
-    },
-    {
-      "epoch": 0.22191655204034846,
-      "grad_norm": 0.622091834737993,
-      "learning_rate": 0.0009066800418028256,
-      "loss": 2.4299,
-      "step": 484
-    },
-    {
-      "epoch": 0.2223750573131591,
-      "grad_norm": 0.769876294127082,
-      "learning_rate": 0.0009062475236323168,
-      "loss": 2.4632,
-      "step": 485
-    },
-    {
-      "epoch": 0.22283356258596973,
-      "grad_norm": 0.6828136079635185,
-      "learning_rate": 0.0009058141091282656,
-      "loss": 2.4191,
-      "step": 486
-    },
-    {
-      "epoch": 0.22329206785878036,
-      "grad_norm": 0.6067146223131924,
-      "learning_rate": 0.0009053797992469461,
-      "loss": 2.4226,
-      "step": 487
-    },
-    {
-      "epoch": 0.22375057313159102,
-      "grad_norm": 0.61042331464537,
-      "learning_rate": 0.0009049445949466078,
-      "loss": 2.4379,
-      "step": 488
-    },
-    {
-      "epoch": 0.22420907840440166,
-      "grad_norm": 0.6434775495825543,
-      "learning_rate": 0.0009045084971874737,
-      "loss": 2.4961,
-      "step": 489
-    },
-    {
-      "epoch": 0.2246675836772123,
-      "grad_norm": 0.648164087916946,
-      "learning_rate": 0.0009040715069317382,
-      "loss": 2.4488,
-      "step": 490
-    },
-    {
-      "epoch": 0.22512608895002292,
-      "grad_norm": 0.598703710797765,
-      "learning_rate": 0.0009036336251435648,
-      "loss": 2.3667,
-      "step": 491
-    },
-    {
-      "epoch": 0.22558459422283356,
-      "grad_norm": 0.6301388702462853,
-      "learning_rate": 0.0009031948527890839,
-      "loss": 2.4664,
-      "step": 492
-    },
-    {
-      "epoch": 0.2260430994956442,
-      "grad_norm": 0.6383756032779206,
-      "learning_rate": 0.000902755190836391,
-      "loss": 2.4934,
-      "step": 493
-    },
-    {
-      "epoch": 0.22650160476845485,
-      "grad_norm": 0.6311516560800575,
-      "learning_rate": 0.0009023146402555442,
-      "loss": 2.3832,
-      "step": 494
-    },
-    {
-      "epoch": 0.22696011004126548,
-      "grad_norm": 0.6608858450190411,
-      "learning_rate": 0.0009018732020185624,
-      "loss": 2.4473,
-      "step": 495
-    },
-    {
-      "epoch": 0.22741861531407612,
-      "grad_norm": 0.6745631835394463,
-      "learning_rate": 0.0009014308770994235,
-      "loss": 2.4859,
-      "step": 496
-    },
-    {
-      "epoch": 0.22787712058688675,
-      "grad_norm": 0.674277771665461,
-      "learning_rate": 0.0009009876664740605,
-      "loss": 2.4061,
-      "step": 497
-    },
-    {
-      "epoch": 0.22833562585969738,
-      "grad_norm": 0.7415945934865369,
-      "learning_rate": 0.0009005435711203618,
-      "loss": 2.4769,
-      "step": 498
-    },
-    {
-      "epoch": 0.22879413113250802,
-      "grad_norm": 0.6341516981125057,
-      "learning_rate": 0.000900098592018167,
-      "loss": 2.3807,
-      "step": 499
-    },
-    {
-      "epoch": 0.22925263640531865,
-      "grad_norm": 0.6633835266617267,
-      "learning_rate": 0.0008996527301492663,
-      "loss": 2.4111,
-      "step": 500
-    },
-    {
-      "epoch": 0.2297111416781293,
-      "grad_norm": 0.6425304101632402,
-      "learning_rate": 0.0008992059864973972,
-      "loss": 2.4305,
-      "step": 501
-    },
-    {
-      "epoch": 0.23016964695093994,
-      "grad_norm": 0.6655889452512519,
-      "learning_rate": 0.0008987583620482427,
-      "loss": 2.481,
-      "step": 502
-    },
-    {
-      "epoch": 0.23062815222375058,
-      "grad_norm": 0.6794806369776747,
-      "learning_rate": 0.0008983098577894292,
-      "loss": 2.3152,
-      "step": 503
-    },
-    {
-      "epoch": 0.2310866574965612,
-      "grad_norm": 0.6500066946417357,
-      "learning_rate": 0.0008978604747105246,
-      "loss": 2.4187,
-      "step": 504
-    },
-    {
-      "epoch": 0.23154516276937184,
-      "grad_norm": 0.6171710290929213,
-      "learning_rate": 0.0008974102138030354,
-      "loss": 2.4143,
-      "step": 505
-    },
-    {
-      "epoch": 0.23200366804218248,
-      "grad_norm": 0.7147337830232946,
-      "learning_rate": 0.000896959076060405,
-      "loss": 2.5136,
-      "step": 506
-    },
-    {
-      "epoch": 0.2324621733149931,
-      "grad_norm": 0.6466559711060365,
-      "learning_rate": 0.0008965070624780116,
-      "loss": 2.407,
-      "step": 507
-    },
-    {
-      "epoch": 0.23292067858780377,
-      "grad_norm": 0.5775224059847199,
-      "learning_rate": 0.0008960541740531658,
-      "loss": 2.4122,
-      "step": 508
-    },
-    {
-      "epoch": 0.2333791838606144,
-      "grad_norm": 0.633283976101755,
-      "learning_rate": 0.0008956004117851083,
-      "loss": 2.3945,
-      "step": 509
-    },
-    {
-      "epoch": 0.23383768913342504,
-      "grad_norm": 0.6714688131252854,
-      "learning_rate": 0.0008951457766750079,
-      "loss": 2.3571,
-      "step": 510
-    },
-    {
-      "epoch": 0.23429619440623567,
-      "grad_norm": 0.5974129630824366,
-      "learning_rate": 0.0008946902697259593,
-      "loss": 2.4555,
-      "step": 511
-    },
-    {
-      "epoch": 0.2347546996790463,
-      "grad_norm": 0.641635075583706,
-      "learning_rate": 0.0008942338919429805,
-      "loss": 2.3894,
-      "step": 512
-    },
-    {
-      "epoch": 0.23521320495185694,
-      "grad_norm": 0.6545075507966298,
-      "learning_rate": 0.0008937766443330113,
-      "loss": 2.4421,
-      "step": 513
-    },
-    {
-      "epoch": 0.2356717102246676,
-      "grad_norm": 0.5993079362040434,
-      "learning_rate": 0.0008933185279049103,
-      "loss": 2.3124,
-      "step": 514
-    },
-    {
-      "epoch": 0.23613021549747823,
-      "grad_norm": 0.61153517208945,
-      "learning_rate": 0.0008928595436694532,
-      "loss": 2.3709,
-      "step": 515
-    },
-    {
-      "epoch": 0.23658872077028886,
-      "grad_norm": 0.7068683304180072,
-      "learning_rate": 0.0008923996926393305,
-      "loss": 2.4605,
-      "step": 516
-    },
-    {
-      "epoch": 0.2370472260430995,
-      "grad_norm": 0.6210985607134849,
-      "learning_rate": 0.0008919389758291449,
-      "loss": 2.5096,
-      "step": 517
-    },
-    {
-      "epoch": 0.23750573131591013,
-      "grad_norm": 0.576043655587415,
-      "learning_rate": 0.0008914773942554098,
-      "loss": 2.4615,
-      "step": 518
-    },
-    {
-      "epoch": 0.23796423658872076,
-      "grad_norm": 0.6343526200870906,
-      "learning_rate": 0.000891014948936546,
-      "loss": 2.4101,
-      "step": 519
-    },
-    {
-      "epoch": 0.2384227418615314,
-      "grad_norm": 0.651532110813708,
-      "learning_rate": 0.0008905516408928804,
-      "loss": 2.4504,
-      "step": 520
-    },
-    {
-      "epoch": 0.23888124713434206,
-      "grad_norm": 0.6153683253577205,
-      "learning_rate": 0.0008900874711466434,
-      "loss": 2.4358,
-      "step": 521
-    },
-    {
-      "epoch": 0.2393397524071527,
-      "grad_norm": 0.5509137022264555,
-      "learning_rate": 0.0008896224407219666,
-      "loss": 2.3555,
-      "step": 522
-    },
-    {
-      "epoch": 0.23979825767996332,
-      "grad_norm": 0.6463543881515971,
-      "learning_rate": 0.0008891565506448804,
-      "loss": 2.4603,
-      "step": 523
-    },
-    {
-      "epoch": 0.24025676295277396,
-      "grad_norm": 0.6308673215592945,
-      "learning_rate": 0.0008886898019433122,
-      "loss": 2.4113,
-      "step": 524
-    },
-    {
-      "epoch": 0.2407152682255846,
-      "grad_norm": 0.6273750223673461,
-      "learning_rate": 0.0008882221956470836,
-      "loss": 2.3956,
-      "step": 525
-    },
-    {
-      "epoch": 0.24117377349839522,
-      "grad_norm": 0.5813598869103455,
-      "learning_rate": 0.0008877537327879086,
-      "loss": 2.3435,
-      "step": 526
-    },
-    {
-      "epoch": 0.24163227877120586,
-      "grad_norm": 0.6146916262317736,
-      "learning_rate": 0.0008872844143993908,
-      "loss": 2.342,
-      "step": 527
-    },
-    {
-      "epoch": 0.24209078404401652,
-      "grad_norm": 0.6448913958797008,
-      "learning_rate": 0.0008868142415170218,
-      "loss": 2.4076,
-      "step": 528
-    },
-    {
-      "epoch": 0.24254928931682715,
-      "grad_norm": 0.5816067581359745,
-      "learning_rate": 0.0008863432151781781,
-      "loss": 2.4148,
-      "step": 529
-    },
-    {
-      "epoch": 0.24300779458963778,
-      "grad_norm": 0.6070224774753479,
-      "learning_rate": 0.0008858713364221195,
-      "loss": 2.261,
-      "step": 530
-    },
-    {
-      "epoch": 0.24346629986244842,
-      "grad_norm": 0.6169830561593425,
-      "learning_rate": 0.0008853986062899868,
-      "loss": 2.4152,
-      "step": 531
-    },
-    {
-      "epoch": 0.24392480513525905,
-      "grad_norm": 0.5745566095489744,
-      "learning_rate": 0.0008849250258247986,
-      "loss": 2.4338,
-      "step": 532
-    },
-    {
-      "epoch": 0.24438331040806968,
-      "grad_norm": 0.5822226966350785,
-      "learning_rate": 0.0008844505960714503,
-      "loss": 2.421,
-      "step": 533
-    },
-    {
-      "epoch": 0.24484181568088034,
-      "grad_norm": 0.5875510473850603,
-      "learning_rate": 0.0008839753180767108,
-      "loss": 2.3953,
-      "step": 534
-    },
-    {
-      "epoch": 0.24530032095369098,
-      "grad_norm": 0.6251127884467537,
-      "learning_rate": 0.0008834991928892204,
-      "loss": 2.3348,
-      "step": 535
-    },
-    {
-      "epoch": 0.2457588262265016,
-      "grad_norm": 0.6078198904023948,
-      "learning_rate": 0.000883022221559489,
-      "loss": 2.4133,
-      "step": 536
-    },
-    {
-      "epoch": 0.24621733149931224,
-      "grad_norm": 0.6174403575011894,
-      "learning_rate": 0.0008825444051398934,
-      "loss": 2.4854,
-      "step": 537
-    },
-    {
-      "epoch": 0.24667583677212288,
-      "grad_norm": 0.6346737780512752,
-      "learning_rate": 0.0008820657446846745,
-      "loss": 2.4542,
-      "step": 538
-    },
-    {
-      "epoch": 0.2471343420449335,
-      "grad_norm": 0.6228160316531164,
-      "learning_rate": 0.000881586241249936,
-      "loss": 2.4127,
-      "step": 539
-    },
-    {
-      "epoch": 0.24759284731774414,
-      "grad_norm": 0.5512334361562646,
-      "learning_rate": 0.0008811058958936411,
-      "loss": 2.3826,
-      "step": 540
-    },
-    {
-      "epoch": 0.2480513525905548,
-      "grad_norm": 0.6403160556896487,
-      "learning_rate": 0.000880624709675611,
-      "loss": 2.3928,
-      "step": 541
-    },
-    {
-      "epoch": 0.24850985786336544,
-      "grad_norm": 0.6179684584495759,
-      "learning_rate": 0.000880142683657522,
-      "loss": 2.4553,
-      "step": 542
-    },
-    {
-      "epoch": 0.24896836313617607,
-      "grad_norm": 0.5815597326775519,
-      "learning_rate": 0.0008796598189029029,
-      "loss": 2.4398,
-      "step": 543
-    },
-    {
-      "epoch": 0.2494268684089867,
-      "grad_norm": 0.5795760167134859,
-      "learning_rate": 0.0008791761164771338,
-      "loss": 2.3633,
-      "step": 544
-    },
-    {
-      "epoch": 0.24988537368179733,
-      "grad_norm": 0.6510872768479001,
-      "learning_rate": 0.0008786915774474424,
-      "loss": 2.4055,
-      "step": 545
-    },
-    {
-      "epoch": 0.25034387895460797,
-      "grad_norm": 0.6130487732592328,
-      "learning_rate": 0.0008782062028829027,
-      "loss": 2.4971,
-      "step": 546
-    },
-    {
-      "epoch": 0.25080238422741863,
-      "grad_norm": 0.7207033695566853,
-      "learning_rate": 0.0008777199938544318,
-      "loss": 2.388,
-      "step": 547
-    },
-    {
-      "epoch": 0.25126088950022923,
-      "grad_norm": 0.5829958622952096,
-      "learning_rate": 0.0008772329514347883,
-      "loss": 2.3951,
-      "step": 548
-    },
-    {
-      "epoch": 0.2517193947730399,
-      "grad_norm": 0.68651786465506,
-      "learning_rate": 0.0008767450766985694,
-      "loss": 2.4389,
-      "step": 549
-    },
-    {
-      "epoch": 0.2521779000458505,
-      "grad_norm": 0.6996551828547495,
-      "learning_rate": 0.0008762563707222086,
-      "loss": 2.4115,
-      "step": 550
-    },
-    {
-      "epoch": 0.25263640531866116,
-      "grad_norm": 0.6432911462771954,
-      "learning_rate": 0.0008757668345839738,
-      "loss": 2.3995,
-      "step": 551
-    },
-    {
-      "epoch": 0.2530949105914718,
-      "grad_norm": 0.7168805513017877,
-      "learning_rate": 0.0008752764693639638,
-      "loss": 2.4984,
-      "step": 552
-    },
-    {
-      "epoch": 0.2535534158642824,
-      "grad_norm": 0.6120223895832259,
-      "learning_rate": 0.0008747852761441078,
-      "loss": 2.3412,
-      "step": 553
-    },
-    {
-      "epoch": 0.2540119211370931,
-      "grad_norm": 0.6120326384263118,
-      "learning_rate": 0.0008742932560081607,
-      "loss": 2.4222,
-      "step": 554
-    },
-    {
-      "epoch": 0.2544704264099037,
-      "grad_norm": 0.8321804326483784,
-      "learning_rate": 0.0008738004100417025,
-      "loss": 2.3449,
-      "step": 555
-    },
-    {
-      "epoch": 0.25492893168271435,
-      "grad_norm": 0.6360396877170151,
-      "learning_rate": 0.0008733067393321355,
-      "loss": 2.3949,
-      "step": 556
-    },
-    {
-      "epoch": 0.255387436955525,
-      "grad_norm": 0.6158321307494375,
-      "learning_rate": 0.000872812244968681,
-      "loss": 2.4707,
-      "step": 557
-    },
-    {
-      "epoch": 0.2558459422283356,
-      "grad_norm": 0.6488923655503851,
-      "learning_rate": 0.0008723169280423783,
-      "loss": 2.3999,
-      "step": 558
-    },
-    {
-      "epoch": 0.2563044475011463,
-      "grad_norm": 0.620561963431552,
-      "learning_rate": 0.0008718207896460811,
-      "loss": 2.436,
-      "step": 559
-    },
-    {
-      "epoch": 0.2567629527739569,
-      "grad_norm": 0.6551485566231218,
-      "learning_rate": 0.0008713238308744557,
-      "loss": 2.3913,
-      "step": 560
-    },
-    {
-      "epoch": 0.25722145804676755,
-      "grad_norm": 0.7537133721828555,
-      "learning_rate": 0.0008708260528239789,
-      "loss": 2.3701,
-      "step": 561
-    },
-    {
-      "epoch": 0.25767996331957815,
-      "grad_norm": 0.6096840979678798,
-      "learning_rate": 0.000870327456592934,
-      "loss": 2.4281,
-      "step": 562
-    },
-    {
-      "epoch": 0.2581384685923888,
-      "grad_norm": 0.7331767064498679,
-      "learning_rate": 0.0008698280432814107,
-      "loss": 2.3973,
-      "step": 563
-    },
-    {
-      "epoch": 0.2585969738651995,
-      "grad_norm": 0.6135881915892857,
-      "learning_rate": 0.000869327813991301,
-      "loss": 2.3592,
-      "step": 564
-    },
-    {
-      "epoch": 0.2590554791380101,
-      "grad_norm": 0.6467882496359131,
-      "learning_rate": 0.0008688267698262971,
-      "loss": 2.2886,
-      "step": 565
-    },
-    {
-      "epoch": 0.25951398441082074,
-      "grad_norm": 0.999638650472179,
-      "learning_rate": 0.0008683249118918894,
-      "loss": 2.3788,
-      "step": 566
-    },
-    {
-      "epoch": 0.25997248968363135,
-      "grad_norm": 0.6550974952564148,
-      "learning_rate": 0.0008678222412953637,
-      "loss": 2.4487,
-      "step": 567
-    },
-    {
-      "epoch": 0.260430994956442,
-      "grad_norm": 0.615593335662876,
-      "learning_rate": 0.0008673187591457987,
-      "loss": 2.401,
-      "step": 568
-    },
-    {
-      "epoch": 0.2608895002292526,
-      "grad_norm": 0.7973844241570424,
-      "learning_rate": 0.0008668144665540639,
-      "loss": 2.4176,
-      "step": 569
-    },
-    {
-      "epoch": 0.2613480055020633,
-      "grad_norm": 0.5811852144244235,
-      "learning_rate": 0.0008663093646328167,
-      "loss": 2.3819,
-      "step": 570
-    },
-    {
-      "epoch": 0.26180651077487394,
-      "grad_norm": 0.6562302549547977,
-      "learning_rate": 0.0008658034544965003,
-      "loss": 2.3624,
-      "step": 571
-    },
-    {
-      "epoch": 0.26226501604768454,
-      "grad_norm": 0.6912495410586338,
-      "learning_rate": 0.0008652967372613412,
-      "loss": 2.5354,
-      "step": 572
-    },
-    {
-      "epoch": 0.2627235213204952,
-      "grad_norm": 0.6557707988312041,
-      "learning_rate": 0.0008647892140453466,
-      "loss": 2.4458,
-      "step": 573
-    },
-    {
-      "epoch": 0.2631820265933058,
-      "grad_norm": 0.6259790744448414,
-      "learning_rate": 0.0008642808859683021,
-      "loss": 2.3894,
-      "step": 574
-    },
-    {
-      "epoch": 0.26364053186611647,
-      "grad_norm": 0.6415646992795874,
-      "learning_rate": 0.0008637717541517689,
-      "loss": 2.357,
-      "step": 575
-    },
-    {
-      "epoch": 0.2640990371389271,
-      "grad_norm": 0.8318402536767932,
-      "learning_rate": 0.0008632618197190816,
-      "loss": 2.3491,
-      "step": 576
-    },
-    {
-      "epoch": 0.26455754241173773,
-      "grad_norm": 0.6300881819859533,
-      "learning_rate": 0.0008627510837953458,
-      "loss": 2.4768,
-      "step": 577
-    },
-    {
-      "epoch": 0.2650160476845484,
-      "grad_norm": 0.8724661825115051,
-      "learning_rate": 0.0008622395475074355,
-      "loss": 2.3431,
-      "step": 578
-    },
-    {
-      "epoch": 0.265474552957359,
-      "grad_norm": 0.6532616605005043,
-      "learning_rate": 0.0008617272119839903,
-      "loss": 2.3677,
-      "step": 579
-    },
-    {
-      "epoch": 0.26593305823016966,
-      "grad_norm": 1.7441227859661272,
-      "learning_rate": 0.0008612140783554136,
-      "loss": 2.3862,
-      "step": 580
-    },
-    {
-      "epoch": 0.26639156350298027,
-      "grad_norm": 0.566709188314118,
-      "learning_rate": 0.0008607001477538696,
-      "loss": 2.3304,
-      "step": 581
-    },
-    {
-      "epoch": 0.2668500687757909,
-      "grad_norm": 0.6219301260423922,
-      "learning_rate": 0.0008601854213132807,
-      "loss": 2.3596,
-      "step": 582
-    },
-    {
-      "epoch": 0.26730857404860153,
-      "grad_norm": 0.7374677897674008,
-      "learning_rate": 0.0008596699001693256,
-      "loss": 2.392,
-      "step": 583
-    },
-    {
-      "epoch": 0.2677670793214122,
-      "grad_norm": 0.7888901228332025,
-      "learning_rate": 0.000859153585459436,
-      "loss": 2.4629,
-      "step": 584
-    },
-    {
-      "epoch": 0.26822558459422285,
-      "grad_norm": 0.9860957733624505,
-      "learning_rate": 0.0008586364783227949,
-      "loss": 2.4051,
-      "step": 585
-    },
-    {
-      "epoch": 0.26868408986703346,
-      "grad_norm": 0.6461056443757641,
-      "learning_rate": 0.0008581185799003332,
-      "loss": 2.4074,
-      "step": 586
-    },
-    {
-      "epoch": 0.2691425951398441,
-      "grad_norm": 1.4713724286428653,
-      "learning_rate": 0.0008575998913347283,
-      "loss": 2.3163,
-      "step": 587
-    },
-    {
-      "epoch": 0.2696011004126547,
-      "grad_norm": 0.6421984078546545,
-      "learning_rate": 0.0008570804137704004,
-      "loss": 2.3457,
-      "step": 588
-    },
-    {
-      "epoch": 0.2700596056854654,
-      "grad_norm": 0.6674464104759813,
-      "learning_rate": 0.0008565601483535108,
-      "loss": 2.3828,
-      "step": 589
-    },
-    {
-      "epoch": 0.27051811095827605,
-      "grad_norm": 0.6719039804008141,
-      "learning_rate": 0.0008560390962319591,
-      "loss": 2.3585,
-      "step": 590
-    },
-    {
-      "epoch": 0.27097661623108665,
-      "grad_norm": 0.6795110425501392,
-      "learning_rate": 0.0008555172585553804,
-      "loss": 2.4293,
-      "step": 591
-    },
-    {
-      "epoch": 0.2714351215038973,
-      "grad_norm": 0.6351580105780029,
-      "learning_rate": 0.0008549946364751435,
-      "loss": 2.415,
-      "step": 592
-    },
-    {
-      "epoch": 0.2718936267767079,
-      "grad_norm": 0.7148797007937459,
-      "learning_rate": 0.0008544712311443475,
-      "loss": 2.4315,
-      "step": 593
-    },
-    {
-      "epoch": 0.2723521320495186,
-      "grad_norm": 1.851694099863968,
-      "learning_rate": 0.0008539470437178196,
-      "loss": 2.2914,
-      "step": 594
-    },
-    {
-      "epoch": 0.2728106373223292,
-      "grad_norm": 0.6342451813878438,
-      "learning_rate": 0.000853422075352113,
-      "loss": 2.4077,
-      "step": 595
-    },
-    {
-      "epoch": 0.27326914259513985,
-      "grad_norm": 0.7029853764955869,
-      "learning_rate": 0.0008528963272055035,
-      "loss": 2.4256,
-      "step": 596
-    },
-    {
-      "epoch": 0.2737276478679505,
-      "grad_norm": 0.6664641158833858,
-      "learning_rate": 0.0008523698004379877,
-      "loss": 2.4455,
-      "step": 597
-    },
-    {
-      "epoch": 0.2741861531407611,
-      "grad_norm": 0.5921595879417815,
-      "learning_rate": 0.00085184249621128,
-      "loss": 2.4003,
-      "step": 598
-    },
-    {
-      "epoch": 0.2746446584135718,
-      "grad_norm": 0.6417749361541993,
-      "learning_rate": 0.0008513144156888101,
-      "loss": 2.4017,
-      "step": 599
-    },
-    {
-      "epoch": 0.2751031636863824,
-      "grad_norm": 0.6499953835680485,
-      "learning_rate": 0.0008507855600357207,
-      "loss": 2.3316,
-      "step": 600
-    },
-    {
-      "epoch": 0.27556166895919304,
-      "grad_norm": 0.6602010655286489,
-      "learning_rate": 0.0008502559304188644,
-      "loss": 2.3683,
-      "step": 601
-    },
-    {
-      "epoch": 0.27602017423200365,
-      "grad_norm": 0.593701073746697,
-      "learning_rate": 0.0008497255280068019,
-      "loss": 2.4438,
-      "step": 602
-    },
-    {
-      "epoch": 0.2764786795048143,
-      "grad_norm": 0.6533312260282143,
-      "learning_rate": 0.0008491943539697986,
-      "loss": 2.3941,
-      "step": 603
-    },
-    {
-      "epoch": 0.27693718477762497,
-      "grad_norm": 0.6415643957047807,
-      "learning_rate": 0.0008486624094798226,
-      "loss": 2.4155,
-      "step": 604
-    },
-    {
-      "epoch": 0.2773956900504356,
-      "grad_norm": 0.5820743555629746,
-      "learning_rate": 0.0008481296957105417,
-      "loss": 2.4701,
-      "step": 605
-    },
-    {
-      "epoch": 0.27785419532324623,
-      "grad_norm": 0.6178660146936497,
-      "learning_rate": 0.0008475962138373213,
-      "loss": 2.4498,
-      "step": 606
-    },
-    {
-      "epoch": 0.27831270059605684,
-      "grad_norm": 0.5506147974775482,
-      "learning_rate": 0.0008470619650372211,
-      "loss": 2.4039,
-      "step": 607
-    },
-    {
-      "epoch": 0.2787712058688675,
-      "grad_norm": 0.5851854833108201,
-      "learning_rate": 0.0008465269504889934,
-      "loss": 2.3649,
-      "step": 608
-    },
-    {
-      "epoch": 0.2792297111416781,
-      "grad_norm": 0.6378613364666753,
-      "learning_rate": 0.0008459911713730799,
-      "loss": 2.5504,
-      "step": 609
-    },
-    {
-      "epoch": 0.27968821641448877,
-      "grad_norm": 0.6052537996070505,
-      "learning_rate": 0.0008454546288716089,
-      "loss": 2.353,
-      "step": 610
-    },
-    {
-      "epoch": 0.2801467216872994,
-      "grad_norm": 0.5863471085615826,
-      "learning_rate": 0.0008449173241683935,
-      "loss": 2.3448,
-      "step": 611
-    },
-    {
-      "epoch": 0.28060522696011003,
-      "grad_norm": 0.659980321965251,
-      "learning_rate": 0.0008443792584489281,
-      "loss": 2.3695,
-      "step": 612
-    },
-    {
-      "epoch": 0.2810637322329207,
-      "grad_norm": 0.7056978766503686,
-      "learning_rate": 0.0008438404329003863,
-      "loss": 2.358,
-      "step": 613
-    },
-    {
-      "epoch": 0.2815222375057313,
-      "grad_norm": 0.680202248449271,
-      "learning_rate": 0.0008433008487116183,
-      "loss": 2.3823,
-      "step": 614
-    },
-    {
-      "epoch": 0.28198074277854196,
-      "grad_norm": 0.5894231012770508,
-      "learning_rate": 0.0008427605070731481,
-      "loss": 2.536,
-      "step": 615
-    },
-    {
-      "epoch": 0.28243924805135256,
-      "grad_norm": 0.7987039485264056,
-      "learning_rate": 0.0008422194091771708,
-      "loss": 2.3664,
-      "step": 616
-    },
-    {
-      "epoch": 0.2828977533241632,
-      "grad_norm": 0.5657062899874121,
-      "learning_rate": 0.0008416775562175503,
-      "loss": 2.3937,
-      "step": 617
-    },
-    {
-      "epoch": 0.2833562585969739,
-      "grad_norm": 0.6045964357971475,
-      "learning_rate": 0.000841134949389816,
-      "loss": 2.3485,
-      "step": 618
-    },
-    {
-      "epoch": 0.2838147638697845,
-      "grad_norm": 0.6193435237582906,
-      "learning_rate": 0.0008405915898911611,
-      "loss": 2.4265,
-      "step": 619
-    },
-    {
-      "epoch": 0.28427326914259515,
-      "grad_norm": 0.9988469387900049,
-      "learning_rate": 0.0008400474789204396,
-      "loss": 2.4171,
-      "step": 620
-    },
-    {
-      "epoch": 0.28473177441540576,
-      "grad_norm": 0.6158690463841565,
-      "learning_rate": 0.0008395026176781626,
-      "loss": 2.4208,
-      "step": 621
-    },
-    {
-      "epoch": 0.2851902796882164,
-      "grad_norm": 0.5709538692982119,
-      "learning_rate": 0.0008389570073664976,
-      "loss": 2.3062,
-      "step": 622
-    },
-    {
-      "epoch": 0.285648784961027,
-      "grad_norm": 0.655304508993323,
-      "learning_rate": 0.0008384106491892642,
-      "loss": 2.2976,
-      "step": 623
-    },
-    {
-      "epoch": 0.2861072902338377,
-      "grad_norm": 0.6599159686927974,
-      "learning_rate": 0.0008378635443519327,
-      "loss": 2.4104,
-      "step": 624
-    },
-    {
-      "epoch": 0.28656579550664835,
-      "grad_norm": 0.7226768587369921,
-      "learning_rate": 0.0008373156940616199,
-      "loss": 2.3466,
-      "step": 625
-    },
-    {
-      "epoch": 0.28702430077945895,
-      "grad_norm": 0.5829514982671197,
-      "learning_rate": 0.0008367670995270882,
-      "loss": 2.3184,
-      "step": 626
-    },
-    {
-      "epoch": 0.2874828060522696,
-      "grad_norm": 0.5769595014026729,
-      "learning_rate": 0.0008362177619587416,
-      "loss": 2.434,
-      "step": 627
-    },
-    {
-      "epoch": 0.2879413113250802,
-      "grad_norm": 0.669351115317637,
-      "learning_rate": 0.0008356676825686238,
-      "loss": 2.378,
-      "step": 628
-    },
-    {
-      "epoch": 0.2883998165978909,
-      "grad_norm": 0.6965680997779802,
-      "learning_rate": 0.0008351168625704147,
-      "loss": 2.4493,
-      "step": 629
-    },
-    {
-      "epoch": 0.28885832187070154,
-      "grad_norm": 0.6347242121758895,
-      "learning_rate": 0.0008345653031794292,
-      "loss": 2.3806,
-      "step": 630
-    },
-    {
-      "epoch": 0.28931682714351215,
-      "grad_norm": 0.6340682831946298,
-      "learning_rate": 0.0008340130056126125,
-      "loss": 2.4033,
-      "step": 631
-    },
-    {
-      "epoch": 0.2897753324163228,
-      "grad_norm": 0.6289716969371879,
-      "learning_rate": 0.0008334599710885394,
-      "loss": 2.4836,
-      "step": 632
-    },
-    {
-      "epoch": 0.2902338376891334,
-      "grad_norm": 0.6446829971217706,
-      "learning_rate": 0.0008329062008274098,
-      "loss": 2.4304,
-      "step": 633
-    },
-    {
-      "epoch": 0.2906923429619441,
-      "grad_norm": 0.6380094631391859,
-      "learning_rate": 0.000832351696051048,
-      "loss": 2.3509,
-      "step": 634
-    },
-    {
-      "epoch": 0.2911508482347547,
-      "grad_norm": 0.6658190093722898,
-      "learning_rate": 0.000831796457982898,
-      "loss": 2.432,
-      "step": 635
-    },
-    {
-      "epoch": 0.29160935350756534,
-      "grad_norm": 0.6158649700228938,
-      "learning_rate": 0.0008312404878480222,
-      "loss": 2.3907,
-      "step": 636
-    },
-    {
-      "epoch": 0.292067858780376,
-      "grad_norm": 0.6272716925381977,
-      "learning_rate": 0.0008306837868730979,
-      "loss": 2.3566,
-      "step": 637
-    },
-    {
-      "epoch": 0.2925263640531866,
-      "grad_norm": 0.5686513433082397,
-      "learning_rate": 0.0008301263562864152,
-      "loss": 2.4703,
-      "step": 638
-    },
-    {
-      "epoch": 0.29298486932599727,
-      "grad_norm": 0.5639732473746389,
-      "learning_rate": 0.0008295681973178737,
-      "loss": 2.4147,
-      "step": 639
-    },
-    {
-      "epoch": 0.29344337459880787,
-      "grad_norm": 0.5615379918326923,
-      "learning_rate": 0.0008290093111989804,
-      "loss": 2.3952,
-      "step": 640
-    },
-    {
-      "epoch": 0.29390187987161853,
-      "grad_norm": 0.6383979790429217,
-      "learning_rate": 0.0008284496991628465,
-      "loss": 2.2389,
-      "step": 641
-    },
-    {
-      "epoch": 0.29436038514442914,
-      "grad_norm": 0.5681405041332358,
-      "learning_rate": 0.0008278893624441847,
-      "loss": 2.3721,
-      "step": 642
-    },
-    {
-      "epoch": 0.2948188904172398,
-      "grad_norm": 0.7396960288753919,
-      "learning_rate": 0.000827328302279307,
-      "loss": 2.3112,
-      "step": 643
-    },
-    {
-      "epoch": 0.29527739569005046,
-      "grad_norm": 0.6424452751599146,
-      "learning_rate": 0.0008267665199061211,
-      "loss": 2.3921,
-      "step": 644
-    },
-    {
-      "epoch": 0.29573590096286106,
-      "grad_norm": 0.742896686081527,
-      "learning_rate": 0.0008262040165641288,
-      "loss": 2.448,
-      "step": 645
-    },
-    {
-      "epoch": 0.2961944062356717,
-      "grad_norm": 0.8512349488530159,
-      "learning_rate": 0.0008256407934944219,
-      "loss": 2.4115,
-      "step": 646
-    },
-    {
-      "epoch": 0.29665291150848233,
-      "grad_norm": 0.5954084148103094,
-      "learning_rate": 0.0008250768519396807,
-      "loss": 2.4253,
-      "step": 647
-    },
-    {
-      "epoch": 0.297111416781293,
-      "grad_norm": 0.7018705396412835,
-      "learning_rate": 0.0008245121931441706,
-      "loss": 2.3119,
-      "step": 648
-    },
-    {
-      "epoch": 0.2975699220541036,
-      "grad_norm": 0.5712491773473823,
-      "learning_rate": 0.0008239468183537393,
-      "loss": 2.3577,
-      "step": 649
-    },
-    {
-      "epoch": 0.29802842732691426,
-      "grad_norm": 0.8243692498520805,
-      "learning_rate": 0.0008233807288158146,
-      "loss": 2.397,
-      "step": 650
-    },
-    {
-      "epoch": 0.2984869325997249,
-      "grad_norm": 0.5593523610900946,
-      "learning_rate": 0.0008228139257794012,
-      "loss": 2.3391,
-      "step": 651
-    },
-    {
-      "epoch": 0.2989454378725355,
-      "grad_norm": 0.6451744966231737,
-      "learning_rate": 0.0008222464104950778,
-      "loss": 2.4352,
-      "step": 652
-    },
-    {
-      "epoch": 0.2994039431453462,
-      "grad_norm": 0.6608851175173895,
-      "learning_rate": 0.000821678184214995,
-      "loss": 2.35,
-      "step": 653
-    },
-    {
-      "epoch": 0.2998624484181568,
-      "grad_norm": 0.5428989857855885,
-      "learning_rate": 0.0008211092481928716,
-      "loss": 2.417,
-      "step": 654
-    },
-    {
-      "epoch": 0.30032095369096745,
-      "grad_norm": 0.5803469041625958,
-      "learning_rate": 0.0008205396036839927,
-      "loss": 2.4706,
-      "step": 655
-    },
-    {
-      "epoch": 0.30077945896377806,
-      "grad_norm": 0.5935688142328672,
-      "learning_rate": 0.0008199692519452069,
-      "loss": 2.4299,
-      "step": 656
-    },
-    {
-      "epoch": 0.3012379642365887,
-      "grad_norm": 0.573867056800319,
-      "learning_rate": 0.0008193981942349224,
-      "loss": 2.4653,
-      "step": 657
-    },
-    {
-      "epoch": 0.3016964695093994,
-      "grad_norm": 0.6191072549010077,
-      "learning_rate": 0.0008188264318131056,
-      "loss": 2.3428,
-      "step": 658
-    },
-    {
-      "epoch": 0.30215497478221,
-      "grad_norm": 0.6251636523700811,
-      "learning_rate": 0.0008182539659412776,
-      "loss": 2.3895,
-      "step": 659
-    },
-    {
-      "epoch": 0.30261348005502064,
-      "grad_norm": 0.5636288287984793,
-      "learning_rate": 0.0008176807978825118,
-      "loss": 2.3727,
-      "step": 660
-    },
-    {
-      "epoch": 0.30307198532783125,
-      "grad_norm": 0.5390883306520359,
-      "learning_rate": 0.0008171069289014306,
-      "loss": 2.4969,
-      "step": 661
-    },
-    {
-      "epoch": 0.3035304906006419,
-      "grad_norm": 0.5732480725907481,
-      "learning_rate": 0.0008165323602642028,
-      "loss": 2.3715,
-      "step": 662
-    },
-    {
-      "epoch": 0.30398899587345257,
-      "grad_norm": 3.6177347456196465,
-      "learning_rate": 0.0008159570932385414,
-      "loss": 2.3576,
-      "step": 663
-    },
-    {
-      "epoch": 0.3044475011462632,
-      "grad_norm": 0.5641180412489363,
-      "learning_rate": 0.0008153811290936999,
-      "loss": 2.4042,
-      "step": 664
-    },
-    {
-      "epoch": 0.30490600641907384,
-      "grad_norm": 0.5509988071044499,
-      "learning_rate": 0.0008148044691004698,
-      "loss": 2.364,
-      "step": 665
-    },
-    {
-      "epoch": 0.30536451169188444,
-      "grad_norm": 0.5901829724221357,
-      "learning_rate": 0.0008142271145311783,
-      "loss": 2.2927,
-      "step": 666
-    },
-    {
-      "epoch": 0.3058230169646951,
-      "grad_norm": 0.7700350199866965,
-      "learning_rate": 0.000813649066659685,
-      "loss": 2.4374,
-      "step": 667
-    },
-    {
-      "epoch": 0.3062815222375057,
-      "grad_norm": 0.5948709686864219,
-      "learning_rate": 0.0008130703267613787,
-      "loss": 2.4811,
-      "step": 668
-    },
-    {
-      "epoch": 0.30674002751031637,
-      "grad_norm": 0.6059193853046889,
-      "learning_rate": 0.0008124908961131759,
-      "loss": 2.3791,
-      "step": 669
-    },
-    {
-      "epoch": 0.30719853278312703,
-      "grad_norm": 0.7954897428006166,
-      "learning_rate": 0.0008119107759935163,
-      "loss": 2.4448,
-      "step": 670
-    },
-    {
-      "epoch": 0.30765703805593764,
-      "grad_norm": 0.6279410795777252,
-      "learning_rate": 0.0008113299676823615,
-      "loss": 2.4549,
-      "step": 671
-    },
-    {
-      "epoch": 0.3081155433287483,
-      "grad_norm": 0.5762268472568115,
-      "learning_rate": 0.0008107484724611911,
-      "loss": 2.3724,
-      "step": 672
-    },
-    {
-      "epoch": 0.3085740486015589,
-      "grad_norm": 0.5821682239575681,
-      "learning_rate": 0.0008101662916130006,
-      "loss": 2.3662,
-      "step": 673
-    },
-    {
-      "epoch": 0.30903255387436956,
-      "grad_norm": 0.6025952285172688,
-      "learning_rate": 0.0008095834264222979,
-      "loss": 2.3817,
-      "step": 674
-    },
-    {
-      "epoch": 0.30949105914718017,
-      "grad_norm": 0.5946770259431051,
-      "learning_rate": 0.0008089998781751009,
-      "loss": 2.4299,
-      "step": 675
-    },
-    {
-      "epoch": 0.30994956441999083,
-      "grad_norm": 0.6328583201476777,
-      "learning_rate": 0.0008084156481589349,
-      "loss": 2.3723,
-      "step": 676
-    },
-    {
-      "epoch": 0.3104080696928015,
-      "grad_norm": 0.5722329772011179,
-      "learning_rate": 0.0008078307376628291,
-      "loss": 2.4428,
-      "step": 677
-    },
-    {
-      "epoch": 0.3108665749656121,
-      "grad_norm": 0.6942430756368884,
-      "learning_rate": 0.0008072451479773143,
-      "loss": 2.3023,
-      "step": 678
-    },
-    {
-      "epoch": 0.31132508023842276,
-      "grad_norm": 0.6110375966271333,
-      "learning_rate": 0.0008066588803944195,
-      "loss": 2.46,
-      "step": 679
-    },
-    {
-      "epoch": 0.31178358551123336,
-      "grad_norm": 0.5540512078468336,
-      "learning_rate": 0.0008060719362076697,
-      "loss": 2.3399,
-      "step": 680
-    },
-    {
-      "epoch": 0.312242090784044,
-      "grad_norm": 0.5459858915143045,
-      "learning_rate": 0.0008054843167120826,
-      "loss": 2.3863,
-      "step": 681
-    },
-    {
-      "epoch": 0.31270059605685463,
-      "grad_norm": 0.5842362388215133,
-      "learning_rate": 0.0008048960232041663,
-      "loss": 2.4243,
-      "step": 682
-    },
-    {
-      "epoch": 0.3131591013296653,
-      "grad_norm": 0.5863435969147561,
-      "learning_rate": 0.0008043070569819153,
-      "loss": 2.3668,
-      "step": 683
-    },
-    {
-      "epoch": 0.31361760660247595,
-      "grad_norm": 0.5739990514745751,
-      "learning_rate": 0.0008037174193448089,
-      "loss": 2.3152,
-      "step": 684
-    },
-    {
-      "epoch": 0.31407611187528656,
-      "grad_norm": 0.8733051273313819,
-      "learning_rate": 0.0008031271115938077,
-      "loss": 2.4157,
-      "step": 685
-    },
-    {
-      "epoch": 0.3145346171480972,
-      "grad_norm": 0.5895464734998546,
-      "learning_rate": 0.0008025361350313505,
-      "loss": 2.3626,
-      "step": 686
-    },
-    {
-      "epoch": 0.3149931224209078,
-      "grad_norm": 0.6051246209239318,
-      "learning_rate": 0.0008019444909613523,
-      "loss": 2.3291,
-      "step": 687
-    },
-    {
-      "epoch": 0.3154516276937185,
-      "grad_norm": 0.5752628852209671,
-      "learning_rate": 0.0008013521806892003,
-      "loss": 2.4508,
-      "step": 688
-    },
-    {
-      "epoch": 0.3159101329665291,
-      "grad_norm": 0.6224267769555628,
-      "learning_rate": 0.000800759205521752,
-      "loss": 2.3669,
-      "step": 689
-    },
-    {
-      "epoch": 0.31636863823933975,
-      "grad_norm": 0.61907264452091,
-      "learning_rate": 0.0008001655667673318,
-      "loss": 2.3237,
-      "step": 690
-    },
-    {
-      "epoch": 0.3168271435121504,
-      "grad_norm": 0.6536870000073488,
-      "learning_rate": 0.0007995712657357279,
-      "loss": 2.411,
-      "step": 691
-    },
-    {
-      "epoch": 0.317285648784961,
-      "grad_norm": 0.6567422038886473,
-      "learning_rate": 0.0007989763037381904,
-      "loss": 2.4442,
-      "step": 692
-    },
-    {
-      "epoch": 0.3177441540577717,
-      "grad_norm": 0.651776556889196,
-      "learning_rate": 0.0007983806820874271,
-      "loss": 2.3418,
-      "step": 693
-    },
-    {
-      "epoch": 0.3182026593305823,
-      "grad_norm": 0.5941715481195426,
-      "learning_rate": 0.0007977844020976016,
-      "loss": 2.4767,
-      "step": 694
-    },
-    {
-      "epoch": 0.31866116460339294,
-      "grad_norm": 0.5936247176221312,
-      "learning_rate": 0.00079718746508433,
-      "loss": 2.3354,
-      "step": 695
-    },
-    {
-      "epoch": 0.31911966987620355,
-      "grad_norm": 0.6456627383917539,
-      "learning_rate": 0.0007965898723646776,
-      "loss": 2.3406,
-      "step": 696
-    },
-    {
-      "epoch": 0.3195781751490142,
-      "grad_norm": 0.5867664334766225,
-      "learning_rate": 0.0007959916252571573,
-      "loss": 2.263,
-      "step": 697
-    },
-    {
-      "epoch": 0.32003668042182487,
-      "grad_norm": 0.6246452669688579,
-      "learning_rate": 0.000795392725081725,
-      "loss": 2.4126,
-      "step": 698
-    },
-    {
-      "epoch": 0.3204951856946355,
-      "grad_norm": 0.5550095052085721,
-      "learning_rate": 0.000794793173159778,
-      "loss": 2.3372,
-      "step": 699
-    },
-    {
-      "epoch": 0.32095369096744614,
-      "grad_norm": 0.7622127208949454,
-      "learning_rate": 0.0007941929708141513,
-      "loss": 2.3773,
-      "step": 700
-    },
-    {
-      "epoch": 0.32141219624025674,
-      "grad_norm": 0.5915735166711423,
-      "learning_rate": 0.0007935921193691153,
-      "loss": 2.3933,
-      "step": 701
-    },
-    {
-      "epoch": 0.3218707015130674,
-      "grad_norm": 1.0058150013144858,
-      "learning_rate": 0.0007929906201503722,
-      "loss": 2.3864,
-      "step": 702
-    },
-    {
-      "epoch": 0.32232920678587806,
-      "grad_norm": 0.5600675213841385,
-      "learning_rate": 0.0007923884744850536,
-      "loss": 2.422,
-      "step": 703
-    },
-    {
-      "epoch": 0.32278771205868867,
-      "grad_norm": 1.0247870030957493,
-      "learning_rate": 0.0007917856837017176,
-      "loss": 2.3831,
-      "step": 704
-    },
-    {
-      "epoch": 0.32324621733149933,
-      "grad_norm": 0.5786134130340941,
-      "learning_rate": 0.0007911822491303452,
-      "loss": 2.3803,
-      "step": 705
-    },
-    {
-      "epoch": 0.32370472260430994,
-      "grad_norm": 0.6291139428902199,
-      "learning_rate": 0.0007905781721023382,
-      "loss": 2.3475,
-      "step": 706
-    },
-    {
-      "epoch": 0.3241632278771206,
-      "grad_norm": 0.6024819204788013,
-      "learning_rate": 0.000789973453950516,
-      "loss": 2.4517,
-      "step": 707
-    },
-    {
-      "epoch": 0.3246217331499312,
-      "grad_norm": 0.6402301220071819,
-      "learning_rate": 0.000789368096009112,
-      "loss": 2.3107,
-      "step": 708
-    },
-    {
-      "epoch": 0.32508023842274186,
-      "grad_norm": 0.585464957545079,
-      "learning_rate": 0.0007887620996137721,
-      "loss": 2.3516,
-      "step": 709
-    },
-    {
-      "epoch": 0.3255387436955525,
-      "grad_norm": 0.5798591016047451,
-      "learning_rate": 0.0007881554661015497,
-      "loss": 2.4029,
-      "step": 710
-    },
-    {
-      "epoch": 0.32599724896836313,
-      "grad_norm": 0.5718999335450506,
-      "learning_rate": 0.0007875481968109051,
-      "loss": 2.3517,
-      "step": 711
-    },
-    {
-      "epoch": 0.3264557542411738,
-      "grad_norm": 0.5779544443421062,
-      "learning_rate": 0.0007869402930817007,
-      "loss": 2.2685,
-      "step": 712
-    },
-    {
-      "epoch": 0.3269142595139844,
-      "grad_norm": 0.580961355359017,
-      "learning_rate": 0.0007863317562551987,
-      "loss": 2.356,
-      "step": 713
-    },
-    {
-      "epoch": 0.32737276478679506,
-      "grad_norm": 0.5929542278376141,
-      "learning_rate": 0.0007857225876740584,
-      "loss": 2.4563,
-      "step": 714
-    },
-    {
-      "epoch": 0.32783127005960566,
-      "grad_norm": 0.5668242118899094,
-      "learning_rate": 0.0007851127886823327,
-      "loss": 2.2594,
-      "step": 715
-    },
-    {
-      "epoch": 0.3282897753324163,
-      "grad_norm": 0.5747291527058214,
-      "learning_rate": 0.0007845023606254658,
-      "loss": 2.2851,
-      "step": 716
-    },
-    {
-      "epoch": 0.328748280605227,
-      "grad_norm": 0.5524730551168953,
-      "learning_rate": 0.0007838913048502894,
-      "loss": 2.3466,
-      "step": 717
-    },
-    {
-      "epoch": 0.3292067858780376,
-      "grad_norm": 0.6427325895356404,
-      "learning_rate": 0.0007832796227050208,
-      "loss": 2.4446,
-      "step": 718
-    },
-    {
-      "epoch": 0.32966529115084825,
-      "grad_norm": 0.5392042279087655,
-      "learning_rate": 0.0007826673155392587,
-      "loss": 2.3675,
-      "step": 719
-    },
-    {
-      "epoch": 0.33012379642365886,
-      "grad_norm": 0.572547613247159,
-      "learning_rate": 0.000782054384703981,
-      "loss": 2.3457,
-      "step": 720
-    },
-    {
-      "epoch": 0.3305823016964695,
-      "grad_norm": 0.5617201611858595,
-      "learning_rate": 0.0007814408315515418,
-      "loss": 2.395,
-      "step": 721
-    },
-    {
-      "epoch": 0.3310408069692801,
-      "grad_norm": 0.5500275962920742,
-      "learning_rate": 0.0007808266574356683,
-      "loss": 2.331,
-      "step": 722
-    },
-    {
-      "epoch": 0.3314993122420908,
-      "grad_norm": 0.5275951075645025,
-      "learning_rate": 0.0007802118637114573,
-      "loss": 2.381,
-      "step": 723
-    },
-    {
-      "epoch": 0.33195781751490144,
-      "grad_norm": 0.6048151466541565,
-      "learning_rate": 0.0007795964517353734,
-      "loss": 2.2991,
-      "step": 724
-    },
-    {
-      "epoch": 0.33241632278771205,
-      "grad_norm": 0.5890686521237348,
-      "learning_rate": 0.0007789804228652449,
-      "loss": 2.3384,
-      "step": 725
-    },
-    {
-      "epoch": 0.3328748280605227,
-      "grad_norm": 0.5654798231808897,
-      "learning_rate": 0.0007783637784602609,
-      "loss": 2.3094,
-      "step": 726
-    },
-    {
-      "epoch": 0.3333333333333333,
-      "grad_norm": 0.5253620492243938,
-      "learning_rate": 0.0007777465198809692,
-      "loss": 2.3846,
-      "step": 727
-    },
-    {
-      "epoch": 0.333791838606144,
-      "grad_norm": 0.5843023669157159,
-      "learning_rate": 0.0007771286484892722,
-      "loss": 2.3035,
-      "step": 728
-    },
-    {
-      "epoch": 0.3342503438789546,
-      "grad_norm": 0.6677173230542522,
-      "learning_rate": 0.000776510165648425,
-      "loss": 2.3222,
-      "step": 729
-    },
-    {
-      "epoch": 0.33470884915176524,
-      "grad_norm": 1.1137603720088098,
-      "learning_rate": 0.0007758910727230311,
-      "loss": 2.3577,
-      "step": 730
-    },
-    {
-      "epoch": 0.3351673544245759,
-      "grad_norm": 0.5165392934141462,
-      "learning_rate": 0.0007752713710790404,
-      "loss": 2.2443,
-      "step": 731
-    },
-    {
-      "epoch": 0.3356258596973865,
-      "grad_norm": 0.5804590586370538,
-      "learning_rate": 0.0007746510620837459,
-      "loss": 2.3726,
-      "step": 732
-    },
-    {
-      "epoch": 0.33608436497019717,
-      "grad_norm": 0.531415235850521,
-      "learning_rate": 0.0007740301471057807,
-      "loss": 2.3073,
-      "step": 733
-    },
-    {
-      "epoch": 0.3365428702430078,
-      "grad_norm": 0.5165552233877524,
-      "learning_rate": 0.0007734086275151146,
-      "loss": 2.3414,
-      "step": 734
-    },
-    {
-      "epoch": 0.33700137551581844,
-      "grad_norm": 0.5810379536807678,
-      "learning_rate": 0.0007727865046830517,
-      "loss": 2.407,
-      "step": 735
-    },
-    {
-      "epoch": 0.3374598807886291,
-      "grad_norm": 0.5562780617449304,
-      "learning_rate": 0.0007721637799822269,
-      "loss": 2.3151,
-      "step": 736
-    },
-    {
-      "epoch": 0.3379183860614397,
-      "grad_norm": 0.5680659247729638,
-      "learning_rate": 0.0007715404547866032,
-      "loss": 2.4252,
-      "step": 737
-    },
-    {
-      "epoch": 0.33837689133425036,
-      "grad_norm": 0.5612904699772373,
-      "learning_rate": 0.0007709165304714685,
-      "loss": 2.3404,
-      "step": 738
-    },
-    {
-      "epoch": 0.33883539660706097,
-      "grad_norm": 0.5962227929075719,
-      "learning_rate": 0.0007702920084134324,
-      "loss": 2.4104,
-      "step": 739
-    },
-    {
-      "epoch": 0.33929390187987163,
-      "grad_norm": 0.5731556208037039,
-      "learning_rate": 0.0007696668899904236,
-      "loss": 2.3974,
-      "step": 740
-    },
-    {
-      "epoch": 0.33975240715268223,
-      "grad_norm": 0.5548057992469195,
-      "learning_rate": 0.0007690411765816864,
-      "loss": 2.3755,
-      "step": 741
-    },
-    {
-      "epoch": 0.3402109124254929,
-      "grad_norm": 0.5273382221921197,
-      "learning_rate": 0.0007684148695677778,
-      "loss": 2.345,
-      "step": 742
-    },
-    {
-      "epoch": 0.34066941769830356,
-      "grad_norm": 0.4937646492368576,
-      "learning_rate": 0.000767787970330565,
-      "loss": 2.3215,
-      "step": 743
-    },
-    {
-      "epoch": 0.34112792297111416,
-      "grad_norm": 0.5486595416271625,
-      "learning_rate": 0.000767160480253221,
-      "loss": 2.4023,
-      "step": 744
-    },
-    {
-      "epoch": 0.3415864282439248,
-      "grad_norm": 0.5107707356174915,
-      "learning_rate": 0.0007665324007202235,
-      "loss": 2.4235,
-      "step": 745
-    },
-    {
-      "epoch": 0.3420449335167354,
-      "grad_norm": 0.5123872748926068,
-      "learning_rate": 0.0007659037331173498,
-      "loss": 2.3502,
-      "step": 746
-    },
-    {
-      "epoch": 0.3425034387895461,
-      "grad_norm": 0.5528714219446537,
-      "learning_rate": 0.0007652744788316752,
-      "loss": 2.4187,
-      "step": 747
-    },
-    {
-      "epoch": 0.3429619440623567,
-      "grad_norm": 0.5318493214834359,
-      "learning_rate": 0.0007646446392515692,
-      "loss": 2.3406,
-      "step": 748
-    },
-    {
-      "epoch": 0.34342044933516735,
-      "grad_norm": 0.6034727112163654,
-      "learning_rate": 0.000764014215766693,
-      "loss": 2.4329,
-      "step": 749
-    },
-    {
-      "epoch": 0.343878954607978,
-      "grad_norm": 0.5540419006565971,
-      "learning_rate": 0.0007633832097679958,
-      "loss": 2.3368,
-      "step": 750
-    },
-    {
-      "epoch": 0.3443374598807886,
-      "grad_norm": 0.5491173241455549,
-      "learning_rate": 0.0007627516226477122,
-      "loss": 2.2917,
-      "step": 751
-    },
-    {
-      "epoch": 0.3447959651535993,
-      "grad_norm": 0.5931523514631889,
-      "learning_rate": 0.0007621194557993589,
-      "loss": 2.3601,
-      "step": 752
-    },
-    {
-      "epoch": 0.3452544704264099,
-      "grad_norm": 0.5574352918482477,
-      "learning_rate": 0.0007614867106177319,
-      "loss": 2.395,
-      "step": 753
-    },
-    {
-      "epoch": 0.34571297569922055,
-      "grad_norm": 0.5418231857421993,
-      "learning_rate": 0.0007608533884989029,
-      "loss": 2.3588,
-      "step": 754
-    },
-    {
-      "epoch": 0.34617148097203115,
-      "grad_norm": 0.5725066310983835,
-      "learning_rate": 0.0007602194908402166,
-      "loss": 2.4666,
-      "step": 755
-    },
-    {
-      "epoch": 0.3466299862448418,
-      "grad_norm": 0.571971142768323,
-      "learning_rate": 0.0007595850190402877,
-      "loss": 2.3205,
-      "step": 756
-    },
-    {
-      "epoch": 0.3470884915176525,
-      "grad_norm": 0.5539502456356745,
-      "learning_rate": 0.0007589499744989976,
-      "loss": 2.3801,
-      "step": 757
-    },
-    {
-      "epoch": 0.3475469967904631,
-      "grad_norm": 0.5484520104484344,
-      "learning_rate": 0.0007583143586174916,
-      "loss": 2.3837,
-      "step": 758
-    },
-    {
-      "epoch": 0.34800550206327374,
-      "grad_norm": 0.5489177817287902,
-      "learning_rate": 0.000757678172798175,
-      "loss": 2.3439,
-      "step": 759
-    },
-    {
-      "epoch": 0.34846400733608435,
-      "grad_norm": 0.5517184889123504,
-      "learning_rate": 0.0007570414184447112,
-      "loss": 2.3345,
-      "step": 760
-    },
-    {
-      "epoch": 0.348922512608895,
-      "grad_norm": 0.6067129295611734,
-      "learning_rate": 0.0007564040969620179,
-      "loss": 2.3932,
-      "step": 761
-    },
-    {
-      "epoch": 0.3493810178817056,
-      "grad_norm": 0.5659785795708996,
-      "learning_rate": 0.0007557662097562636,
-      "loss": 2.3684,
-      "step": 762
-    },
-    {
-      "epoch": 0.3498395231545163,
-      "grad_norm": 0.602214140784335,
-      "learning_rate": 0.0007551277582348658,
-      "loss": 2.3844,
-      "step": 763
-    },
-    {
-      "epoch": 0.35029802842732694,
-      "grad_norm": 0.5382053979095284,
-      "learning_rate": 0.0007544887438064862,
-      "loss": 2.3859,
-      "step": 764
-    },
-    {
-      "epoch": 0.35075653370013754,
-      "grad_norm": 0.5204632634463788,
-      "learning_rate": 0.0007538491678810294,
-      "loss": 2.3986,
-      "step": 765
-    },
-    {
-      "epoch": 0.3512150389729482,
-      "grad_norm": 0.594692514647606,
-      "learning_rate": 0.0007532090318696381,
-      "loss": 2.3809,
-      "step": 766
-    },
-    {
-      "epoch": 0.3516735442457588,
-      "grad_norm": 0.5989275936949552,
-      "learning_rate": 0.0007525683371846913,
-      "loss": 2.4009,
-      "step": 767
-    },
-    {
-      "epoch": 0.35213204951856947,
-      "grad_norm": 0.5410151046963885,
-      "learning_rate": 0.0007519270852398001,
-      "loss": 2.4295,
-      "step": 768
-    },
-    {
-      "epoch": 0.35259055479138013,
-      "grad_norm": 0.6446029198781277,
-      "learning_rate": 0.000751285277449806,
-      "loss": 2.4578,
-      "step": 769
-    },
-    {
-      "epoch": 0.35304906006419073,
-      "grad_norm": 0.5787025750950012,
-      "learning_rate": 0.0007506429152307756,
-      "loss": 2.5038,
-      "step": 770
-    },
-    {
-      "epoch": 0.3535075653370014,
-      "grad_norm": 0.532794142510667,
-      "learning_rate": 0.00075,
-      "loss": 2.305,
-      "step": 771
-    },
-    {
-      "epoch": 0.353966070609812,
-      "grad_norm": 0.5885308441482923,
-      "learning_rate": 0.00074935653317599,
-      "loss": 2.345,
-      "step": 772
-    },
-    {
-      "epoch": 0.35442457588262266,
-      "grad_norm": 0.5527066248660742,
-      "learning_rate": 0.000748712516178473,
-      "loss": 2.406,
-      "step": 773
-    },
-    {
-      "epoch": 0.35488308115543327,
-      "grad_norm": 0.5422296637426552,
-      "learning_rate": 0.0007480679504283911,
-      "loss": 2.363,
-      "step": 774
-    },
-    {
-      "epoch": 0.3553415864282439,
-      "grad_norm": 0.5807454624726437,
-      "learning_rate": 0.0007474228373478964,
-      "loss": 2.3696,
-      "step": 775
-    },
-    {
-      "epoch": 0.3558000917010546,
-      "grad_norm": 0.6126224255810685,
-      "learning_rate": 0.0007467771783603492,
-      "loss": 2.4065,
-      "step": 776
-    },
-    {
-      "epoch": 0.3562585969738652,
-      "grad_norm": 0.5939529058127212,
-      "learning_rate": 0.0007461309748903138,
-      "loss": 2.3653,
-      "step": 777
-    },
-    {
-      "epoch": 0.35671710224667585,
-      "grad_norm": 0.5602703488201014,
-      "learning_rate": 0.0007454842283635562,
-      "loss": 2.5086,
-      "step": 778
-    },
-    {
-      "epoch": 0.35717560751948646,
-      "grad_norm": 0.5567078717084607,
-      "learning_rate": 0.0007448369402070404,
-      "loss": 2.3516,
-      "step": 779
-    },
-    {
-      "epoch": 0.3576341127922971,
-      "grad_norm": 0.6024826180251579,
-      "learning_rate": 0.0007441891118489254,
-      "loss": 2.4081,
-      "step": 780
-    },
-    {
-      "epoch": 0.3580926180651077,
-      "grad_norm": 0.513786704451278,
-      "learning_rate": 0.0007435407447185622,
-      "loss": 2.4464,
-      "step": 781
-    },
-    {
-      "epoch": 0.3585511233379184,
-      "grad_norm": 0.5265211564007609,
-      "learning_rate": 0.0007428918402464908,
-      "loss": 2.3413,
-      "step": 782
-    },
-    {
-      "epoch": 0.35900962861072905,
-      "grad_norm": 0.5593588406144061,
-      "learning_rate": 0.0007422423998644359,
-      "loss": 2.3995,
-      "step": 783
-    },
-    {
-      "epoch": 0.35946813388353965,
-      "grad_norm": 0.5512323293052026,
-      "learning_rate": 0.0007415924250053055,
-      "loss": 2.3595,
-      "step": 784
-    },
-    {
-      "epoch": 0.3599266391563503,
-      "grad_norm": 0.5605591187402648,
-      "learning_rate": 0.0007409419171031865,
-      "loss": 2.3324,
-      "step": 785
-    },
-    {
-      "epoch": 0.3603851444291609,
-      "grad_norm": 0.539507295616766,
-      "learning_rate": 0.0007402908775933419,
-      "loss": 2.3387,
-      "step": 786
-    },
-    {
-      "epoch": 0.3608436497019716,
-      "grad_norm": 0.5536252873058062,
-      "learning_rate": 0.0007396393079122077,
-      "loss": 2.3917,
-      "step": 787
-    },
-    {
-      "epoch": 0.3613021549747822,
-      "grad_norm": 0.5483361056821187,
-      "learning_rate": 0.0007389872094973896,
-      "loss": 2.2607,
-      "step": 788
-    },
-    {
-      "epoch": 0.36176066024759285,
-      "grad_norm": 0.5916647132042852,
-      "learning_rate": 0.00073833458378766,
-      "loss": 2.5241,
-      "step": 789
-    },
-    {
-      "epoch": 0.3622191655204035,
-      "grad_norm": 0.56979508108908,
-      "learning_rate": 0.0007376814322229544,
-      "loss": 2.466,
-      "step": 790
-    },
-    {
-      "epoch": 0.3626776707932141,
-      "grad_norm": 0.5650822880881783,
-      "learning_rate": 0.0007370277562443688,
-      "loss": 2.4701,
-      "step": 791
-    },
-    {
-      "epoch": 0.3631361760660248,
-      "grad_norm": 0.545385351829367,
-      "learning_rate": 0.0007363735572941564,
-      "loss": 2.2991,
-      "step": 792
-    },
-    {
-      "epoch": 0.3635946813388354,
-      "grad_norm": 0.5833609374582766,
-      "learning_rate": 0.0007357188368157236,
-      "loss": 2.4106,
-      "step": 793
-    },
-    {
-      "epoch": 0.36405318661164604,
-      "grad_norm": 0.5578095718849222,
-      "learning_rate": 0.0007350635962536284,
-      "loss": 2.2636,
-      "step": 794
-    },
-    {
-      "epoch": 0.36451169188445665,
-      "grad_norm": 0.5311939537763459,
-      "learning_rate": 0.0007344078370535756,
-      "loss": 2.2939,
-      "step": 795
-    },
-    {
-      "epoch": 0.3649701971572673,
-      "grad_norm": 0.5398077999909714,
-      "learning_rate": 0.0007337515606624148,
-      "loss": 2.428,
-      "step": 796
-    },
-    {
-      "epoch": 0.36542870243007797,
-      "grad_norm": 0.5126886853289304,
-      "learning_rate": 0.0007330947685281362,
-      "loss": 2.4285,
-      "step": 797
-    },
-    {
-      "epoch": 0.3658872077028886,
-      "grad_norm": 0.5835088716488485,
-      "learning_rate": 0.0007324374620998682,
-      "loss": 2.3992,
-      "step": 798
-    },
-    {
-      "epoch": 0.36634571297569923,
-      "grad_norm": 0.5133176795378716,
-      "learning_rate": 0.000731779642827874,
-      "loss": 2.2159,
-      "step": 799
-    },
-    {
-      "epoch": 0.36680421824850984,
-      "grad_norm": 0.5763532601560792,
-      "learning_rate": 0.0007311213121635483,
-      "loss": 2.3294,
-      "step": 800
-    },
-    {
-      "epoch": 0.3672627235213205,
-      "grad_norm": 0.5233610000772326,
-      "learning_rate": 0.0007304624715594139,
-      "loss": 2.3301,
-      "step": 801
-    },
-    {
-      "epoch": 0.3677212287941311,
-      "grad_norm": 1.0920130956156373,
-      "learning_rate": 0.0007298031224691193,
-      "loss": 2.4273,
-      "step": 802
-    },
-    {
-      "epoch": 0.36817973406694177,
-      "grad_norm": 0.5146766038259072,
-      "learning_rate": 0.0007291432663474339,
-      "loss": 2.3554,
-      "step": 803
-    },
-    {
-      "epoch": 0.3686382393397524,
-      "grad_norm": 0.5241391346582333,
-      "learning_rate": 0.0007284829046502467,
-      "loss": 2.2939,
-      "step": 804
-    },
-    {
-      "epoch": 0.36909674461256303,
-      "grad_norm": 0.5019224109696822,
-      "learning_rate": 0.0007278220388345619,
-      "loss": 2.3456,
-      "step": 805
-    },
-    {
-      "epoch": 0.3695552498853737,
-      "grad_norm": 0.5179750589379198,
-      "learning_rate": 0.0007271606703584958,
-      "loss": 2.3676,
-      "step": 806
-    },
-    {
-      "epoch": 0.3700137551581843,
-      "grad_norm": 0.5421509608982283,
-      "learning_rate": 0.000726498800681274,
-      "loss": 2.3122,
-      "step": 807
-    },
-    {
-      "epoch": 0.37047226043099496,
-      "grad_norm": 0.4936573196220727,
-      "learning_rate": 0.0007258364312632279,
-      "loss": 2.3559,
-      "step": 808
-    },
-    {
-      "epoch": 0.3709307657038056,
-      "grad_norm": 0.5676742876477151,
-      "learning_rate": 0.0007251735635657915,
-      "loss": 2.306,
-      "step": 809
-    },
-    {
-      "epoch": 0.3713892709766162,
-      "grad_norm": 0.5427334332574643,
-      "learning_rate": 0.000724510199051498,
-      "loss": 2.2686,
-      "step": 810
-    },
-    {
-      "epoch": 0.3718477762494269,
-      "grad_norm": 0.5528211169621853,
-      "learning_rate": 0.0007238463391839769,
-      "loss": 2.396,
-      "step": 811
-    },
-    {
-      "epoch": 0.3723062815222375,
-      "grad_norm": 0.6131206810218987,
-      "learning_rate": 0.0007231819854279508,
-      "loss": 2.4299,
-      "step": 812
-    },
-    {
-      "epoch": 0.37276478679504815,
-      "grad_norm": 0.5782833343676114,
-      "learning_rate": 0.0007225171392492316,
-      "loss": 2.3093,
-      "step": 813
-    },
-    {
-      "epoch": 0.37322329206785876,
-      "grad_norm": 0.5769746065983292,
-      "learning_rate": 0.0007218518021147182,
-      "loss": 2.4235,
-      "step": 814
-    },
-    {
-      "epoch": 0.3736817973406694,
-      "grad_norm": 0.5322000888487295,
-      "learning_rate": 0.0007211859754923923,
-      "loss": 2.3831,
-      "step": 815
-    },
-    {
-      "epoch": 0.3741403026134801,
-      "grad_norm": 0.5609297269442653,
-      "learning_rate": 0.0007205196608513158,
-      "loss": 2.4017,
-      "step": 816
-    },
-    {
-      "epoch": 0.3745988078862907,
-      "grad_norm": 0.5537981808940872,
-      "learning_rate": 0.0007198528596616272,
-      "loss": 2.2833,
-      "step": 817
-    },
-    {
-      "epoch": 0.37505731315910135,
-      "grad_norm": 0.5192308346090793,
-      "learning_rate": 0.0007191855733945387,
-      "loss": 2.3031,
-      "step": 818
-    },
-    {
-      "epoch": 0.37551581843191195,
-      "grad_norm": 0.5685590517390261,
-      "learning_rate": 0.0007185178035223327,
-      "loss": 2.3369,
-      "step": 819
-    },
-    {
-      "epoch": 0.3759743237047226,
-      "grad_norm": 0.5636232069993984,
-      "learning_rate": 0.0007178495515183583,
-      "loss": 2.457,
-      "step": 820
-    },
-    {
-      "epoch": 0.3764328289775332,
-      "grad_norm": 0.5649715399474239,
-      "learning_rate": 0.000717180818857029,
-      "loss": 2.2897,
-      "step": 821
-    },
-    {
-      "epoch": 0.3768913342503439,
-      "grad_norm": 0.5552433762462149,
-      "learning_rate": 0.0007165116070138182,
-      "loss": 2.3563,
-      "step": 822
-    },
-    {
-      "epoch": 0.37734983952315454,
-      "grad_norm": 0.5197023671459275,
-      "learning_rate": 0.0007158419174652569,
-      "loss": 2.3649,
-      "step": 823
-    },
-    {
-      "epoch": 0.37780834479596515,
-      "grad_norm": 0.5726460815118224,
-      "learning_rate": 0.00071517175168893,
-      "loss": 2.3699,
-      "step": 824
-    },
-    {
-      "epoch": 0.3782668500687758,
-      "grad_norm": 0.5478881872144105,
-      "learning_rate": 0.0007145011111634732,
-      "loss": 2.3023,
-      "step": 825
-    },
-    {
-      "epoch": 0.3787253553415864,
-      "grad_norm": 0.5665792474376192,
-      "learning_rate": 0.0007138299973685694,
-      "loss": 2.3934,
-      "step": 826
-    },
-    {
-      "epoch": 0.3791838606143971,
-      "grad_norm": 0.5219168421306043,
-      "learning_rate": 0.0007131584117849459,
-      "loss": 2.3914,
-      "step": 827
-    },
-    {
-      "epoch": 0.3796423658872077,
-      "grad_norm": 0.5922823502551955,
-      "learning_rate": 0.0007124863558943713,
-      "loss": 2.3213,
-      "step": 828
-    },
-    {
-      "epoch": 0.38010087116001834,
-      "grad_norm": 0.5307645429169339,
-      "learning_rate": 0.0007118138311796514,
-      "loss": 2.2625,
-      "step": 829
-    },
-    {
-      "epoch": 0.380559376432829,
-      "grad_norm": 0.6129415784027249,
-      "learning_rate": 0.0007111408391246262,
-      "loss": 2.4053,
-      "step": 830
-    },
-    {
-      "epoch": 0.3810178817056396,
-      "grad_norm": 0.566391199832776,
-      "learning_rate": 0.0007104673812141675,
-      "loss": 2.3545,
-      "step": 831
-    },
-    {
-      "epoch": 0.38147638697845027,
-      "grad_norm": 0.5008191138744033,
-      "learning_rate": 0.0007097934589341745,
-      "loss": 2.3098,
-      "step": 832
-    },
-    {
-      "epoch": 0.38193489225126087,
-      "grad_norm": 0.5243376435554258,
-      "learning_rate": 0.0007091190737715711,
-      "loss": 2.3269,
-      "step": 833
-    },
-    {
-      "epoch": 0.38239339752407153,
-      "grad_norm": 0.5179782225208717,
-      "learning_rate": 0.0007084442272143026,
-      "loss": 2.3827,
-      "step": 834
-    },
-    {
-      "epoch": 0.38285190279688214,
-      "grad_norm": 0.5018330892736436,
-      "learning_rate": 0.000707768920751332,
-      "loss": 2.4632,
-      "step": 835
-    },
-    {
-      "epoch": 0.3833104080696928,
-      "grad_norm": 0.5588237478183992,
-      "learning_rate": 0.0007070931558726373,
-      "loss": 2.3747,
-      "step": 836
-    },
-    {
-      "epoch": 0.38376891334250346,
-      "grad_norm": 0.7143093154192723,
-      "learning_rate": 0.0007064169340692076,
-      "loss": 2.3591,
-      "step": 837
-    },
-    {
-      "epoch": 0.38422741861531406,
-      "grad_norm": 0.5249792654762456,
-      "learning_rate": 0.0007057402568330407,
-      "loss": 2.4361,
-      "step": 838
-    },
-    {
-      "epoch": 0.3846859238881247,
-      "grad_norm": 0.6373106503185764,
-      "learning_rate": 0.0007050631256571389,
-      "loss": 2.4136,
-      "step": 839
-    },
-    {
-      "epoch": 0.38514442916093533,
-      "grad_norm": 0.5161242391402878,
-      "learning_rate": 0.000704385542035506,
-      "loss": 2.3553,
-      "step": 840
-    },
-    {
-      "epoch": 0.385602934433746,
-      "grad_norm": 0.5442057541227984,
-      "learning_rate": 0.000703707507463144,
-      "loss": 2.3559,
-      "step": 841
-    },
-    {
-      "epoch": 0.38606143970655665,
-      "grad_norm": 0.5530404083612418,
-      "learning_rate": 0.0007030290234360505,
-      "loss": 2.2666,
-      "step": 842
-    },
-    {
-      "epoch": 0.38651994497936726,
-      "grad_norm": 0.5633976881550155,
-      "learning_rate": 0.0007023500914512139,
-      "loss": 2.2983,
-      "step": 843
-    },
-    {
-      "epoch": 0.3869784502521779,
-      "grad_norm": 0.5713693484924441,
-      "learning_rate": 0.0007016707130066116,
-      "loss": 2.2872,
-      "step": 844
-    },
-    {
-      "epoch": 0.3874369555249885,
-      "grad_norm": 0.5568579868432181,
-      "learning_rate": 0.0007009908896012055,
-      "loss": 2.3927,
-      "step": 845
-    },
-    {
-      "epoch": 0.3878954607977992,
-      "grad_norm": 0.5693922847179034,
-      "learning_rate": 0.0007003106227349399,
-      "loss": 2.3972,
-      "step": 846
-    },
-    {
-      "epoch": 0.3883539660706098,
-      "grad_norm": 0.948635611452431,
-      "learning_rate": 0.000699629913908737,
-      "loss": 2.3474,
-      "step": 847
-    },
-    {
-      "epoch": 0.38881247134342045,
-      "grad_norm": 0.5807621064977793,
-      "learning_rate": 0.0006989487646244943,
-      "loss": 2.3086,
-      "step": 848
-    },
-    {
-      "epoch": 0.3892709766162311,
-      "grad_norm": 0.5171774309577701,
-      "learning_rate": 0.0006982671763850814,
-      "loss": 2.2383,
-      "step": 849
-    },
-    {
-      "epoch": 0.3897294818890417,
-      "grad_norm": 0.5611400506189765,
-      "learning_rate": 0.0006975851506943359,
-      "loss": 2.281,
-      "step": 850
-    },
-    {
-      "epoch": 0.3901879871618524,
-      "grad_norm": 0.5247530582964096,
-      "learning_rate": 0.0006969026890570611,
-      "loss": 2.3833,
-      "step": 851
-    },
-    {
-      "epoch": 0.390646492434663,
-      "grad_norm": 0.5420365188700074,
-      "learning_rate": 0.0006962197929790216,
-      "loss": 2.3442,
-      "step": 852
-    },
-    {
-      "epoch": 0.39110499770747365,
-      "grad_norm": 0.5539234571817386,
-      "learning_rate": 0.0006955364639669409,
-      "loss": 2.3384,
-      "step": 853
-    },
-    {
-      "epoch": 0.39156350298028425,
-      "grad_norm": 0.6073951149915158,
-      "learning_rate": 0.0006948527035284978,
-      "loss": 2.3662,
-      "step": 854
-    },
-    {
-      "epoch": 0.3920220082530949,
-      "grad_norm": 0.5583389914636869,
-      "learning_rate": 0.0006941685131723225,
-      "loss": 2.3873,
-      "step": 855
-    },
-    {
-      "epoch": 0.39248051352590557,
-      "grad_norm": 0.534441863766848,
-      "learning_rate": 0.0006934838944079943,
-      "loss": 2.3947,
-      "step": 856
-    },
-    {
-      "epoch": 0.3929390187987162,
-      "grad_norm": 0.5209752979615028,
-      "learning_rate": 0.0006927988487460378,
-      "loss": 2.2844,
-      "step": 857
-    },
-    {
-      "epoch": 0.39339752407152684,
-      "grad_norm": 0.5398201710089527,
-      "learning_rate": 0.0006921133776979186,
-      "loss": 2.3205,
-      "step": 858
-    },
-    {
-      "epoch": 0.39385602934433744,
-      "grad_norm": 0.514941295283389,
-      "learning_rate": 0.0006914274827760418,
-      "loss": 2.3849,
-      "step": 859
-    },
-    {
-      "epoch": 0.3943145346171481,
-      "grad_norm": 0.5003262271433511,
-      "learning_rate": 0.0006907411654937475,
-      "loss": 2.4111,
-      "step": 860
-    },
-    {
-      "epoch": 0.3947730398899587,
-      "grad_norm": 0.502881975132895,
-      "learning_rate": 0.0006900544273653075,
-      "loss": 2.3418,
-      "step": 861
-    },
-    {
-      "epoch": 0.39523154516276937,
-      "grad_norm": 0.5226331997251873,
-      "learning_rate": 0.000689367269905922,
-      "loss": 2.3324,
-      "step": 862
-    },
-    {
-      "epoch": 0.39569005043558003,
-      "grad_norm": 0.5204036133749118,
-      "learning_rate": 0.0006886796946317168,
-      "loss": 2.4028,
-      "step": 863
-    },
-    {
-      "epoch": 0.39614855570839064,
-      "grad_norm": 0.5578286052847883,
-      "learning_rate": 0.0006879917030597397,
-      "loss": 2.326,
-      "step": 864
-    },
-    {
-      "epoch": 0.3966070609812013,
-      "grad_norm": 0.5359771796630176,
-      "learning_rate": 0.0006873032967079561,
-      "loss": 2.3558,
-      "step": 865
-    },
-    {
-      "epoch": 0.3970655662540119,
-      "grad_norm": 0.5035672096088962,
-      "learning_rate": 0.0006866144770952474,
-      "loss": 2.3397,
-      "step": 866
-    },
-    {
-      "epoch": 0.39752407152682256,
-      "grad_norm": 0.5110921533725696,
-      "learning_rate": 0.0006859252457414067,
-      "loss": 2.3766,
-      "step": 867
-    },
-    {
-      "epoch": 0.39798257679963317,
-      "grad_norm": 0.5922498277740145,
-      "learning_rate": 0.0006852356041671351,
-      "loss": 2.3648,
-      "step": 868
-    },
-    {
-      "epoch": 0.39844108207244383,
-      "grad_norm": 0.5421259373517148,
-      "learning_rate": 0.0006845455538940394,
-      "loss": 2.3671,
-      "step": 869
-    },
-    {
-      "epoch": 0.3988995873452545,
-      "grad_norm": 0.5316622406470337,
-      "learning_rate": 0.0006838550964446276,
-      "loss": 2.3283,
-      "step": 870
-    },
-    {
-      "epoch": 0.3993580926180651,
-      "grad_norm": 0.5379781441972117,
-      "learning_rate": 0.0006831642333423067,
-      "loss": 2.4368,
-      "step": 871
-    },
-    {
-      "epoch": 0.39981659789087576,
-      "grad_norm": 0.5099006066252737,
-      "learning_rate": 0.000682472966111378,
-      "loss": 2.3364,
-      "step": 872
-    },
-    {
-      "epoch": 0.40027510316368636,
-      "grad_norm": 0.5380008590441111,
-      "learning_rate": 0.0006817812962770348,
-      "loss": 2.365,
-      "step": 873
-    },
-    {
-      "epoch": 0.400733608436497,
-      "grad_norm": 0.5408655078428792,
-      "learning_rate": 0.0006810892253653589,
-      "loss": 2.4194,
-      "step": 874
-    },
-    {
-      "epoch": 0.40119211370930763,
-      "grad_norm": 0.5202575523915198,
-      "learning_rate": 0.0006803967549033167,
-      "loss": 2.4549,
-      "step": 875
-    },
-    {
-      "epoch": 0.4016506189821183,
-      "grad_norm": 0.5415480546694929,
-      "learning_rate": 0.0006797038864187564,
-      "loss": 2.3058,
-      "step": 876
-    },
-    {
-      "epoch": 0.40210912425492895,
-      "grad_norm": 0.565713200291752,
-      "learning_rate": 0.0006790106214404043,
-      "loss": 2.3528,
-      "step": 877
-    },
-    {
-      "epoch": 0.40256762952773956,
-      "grad_norm": 0.4997842435412165,
-      "learning_rate": 0.0006783169614978614,
-      "loss": 2.3143,
-      "step": 878
-    },
-    {
-      "epoch": 0.4030261348005502,
-      "grad_norm": 0.5664633876245978,
-      "learning_rate": 0.0006776229081216001,
-      "loss": 2.334,
-      "step": 879
-    },
-    {
-      "epoch": 0.4034846400733608,
-      "grad_norm": 0.5795858432451744,
-      "learning_rate": 0.0006769284628429611,
-      "loss": 2.3828,
-      "step": 880
-    },
-    {
-      "epoch": 0.4039431453461715,
-      "grad_norm": 0.5786740540728272,
-      "learning_rate": 0.0006762336271941498,
-      "loss": 2.4168,
-      "step": 881
-    },
-    {
-      "epoch": 0.40440165061898214,
-      "grad_norm": 0.5341054428484557,
-      "learning_rate": 0.0006755384027082326,
-      "loss": 2.3147,
-      "step": 882
-    },
-    {
-      "epoch": 0.40486015589179275,
-      "grad_norm": 0.6213691939616937,
-      "learning_rate": 0.0006748427909191342,
-      "loss": 2.4052,
-      "step": 883
-    },
-    {
-      "epoch": 0.4053186611646034,
-      "grad_norm": 0.6547204673027162,
-      "learning_rate": 0.0006741467933616335,
-      "loss": 2.3846,
-      "step": 884
-    },
-    {
-      "epoch": 0.405777166437414,
-      "grad_norm": 0.5679095493852216,
-      "learning_rate": 0.0006734504115713604,
-      "loss": 2.242,
-      "step": 885
-    },
-    {
-      "epoch": 0.4062356717102247,
-      "grad_norm": 0.5392949296017379,
-      "learning_rate": 0.0006727536470847932,
-      "loss": 2.3578,
-      "step": 886
-    },
-    {
-      "epoch": 0.4066941769830353,
-      "grad_norm": 0.530351160212528,
-      "learning_rate": 0.000672056501439254,
-      "loss": 2.357,
-      "step": 887
-    },
-    {
-      "epoch": 0.40715268225584594,
-      "grad_norm": 0.6368528933090891,
-      "learning_rate": 0.0006713589761729063,
-      "loss": 2.2924,
-      "step": 888
-    },
-    {
-      "epoch": 0.4076111875286566,
-      "grad_norm": 0.5239599879777793,
-      "learning_rate": 0.0006706610728247508,
-      "loss": 2.2652,
-      "step": 889
-    },
-    {
-      "epoch": 0.4080696928014672,
-      "grad_norm": 0.5433608897185997,
-      "learning_rate": 0.0006699627929346227,
-      "loss": 2.3686,
-      "step": 890
-    },
-    {
-      "epoch": 0.40852819807427787,
-      "grad_norm": 0.5614779686141015,
-      "learning_rate": 0.0006692641380431879,
-      "loss": 2.4064,
-      "step": 891
-    },
-    {
-      "epoch": 0.4089867033470885,
-      "grad_norm": 0.49578266194279563,
-      "learning_rate": 0.0006685651096919393,
-      "loss": 2.3038,
-      "step": 892
-    },
-    {
-      "epoch": 0.40944520861989914,
-      "grad_norm": 0.5349206806229141,
-      "learning_rate": 0.0006678657094231944,
-      "loss": 2.3557,
-      "step": 893
-    },
-    {
-      "epoch": 0.40990371389270974,
-      "grad_norm": 0.6741660851398205,
-      "learning_rate": 0.0006671659387800909,
-      "loss": 2.3327,
-      "step": 894
-    },
-    {
-      "epoch": 0.4103622191655204,
-      "grad_norm": 0.529054706413139,
-      "learning_rate": 0.000666465799306584,
-      "loss": 2.3547,
-      "step": 895
-    },
-    {
-      "epoch": 0.41082072443833106,
-      "grad_norm": 2.6590922747900865,
-      "learning_rate": 0.0006657652925474423,
-      "loss": 2.3911,
-      "step": 896
-    },
-    {
-      "epoch": 0.41127922971114167,
-      "grad_norm": 0.541416801218056,
-      "learning_rate": 0.000665064420048245,
-      "loss": 2.3009,
-      "step": 897
-    },
-    {
-      "epoch": 0.41173773498395233,
-      "grad_norm": 0.5995602812845273,
-      "learning_rate": 0.0006643631833553785,
-      "loss": 2.3611,
-      "step": 898
-    },
-    {
-      "epoch": 0.41219624025676294,
-      "grad_norm": 0.586047333317652,
-      "learning_rate": 0.000663661584016032,
-      "loss": 2.3838,
-      "step": 899
-    },
-    {
-      "epoch": 0.4126547455295736,
-      "grad_norm": 0.5501463332107863,
-      "learning_rate": 0.0006629596235781957,
-      "loss": 2.3142,
-      "step": 900
-    },
-    {
-      "epoch": 0.4131132508023842,
-      "grad_norm": 0.5566671091187362,
-      "learning_rate": 0.0006622573035906556,
-      "loss": 2.3501,
-      "step": 901
-    },
-    {
-      "epoch": 0.41357175607519486,
-      "grad_norm": 0.578271200165006,
-      "learning_rate": 0.0006615546256029921,
-      "loss": 2.2409,
-      "step": 902
-    },
-    {
-      "epoch": 0.4140302613480055,
-      "grad_norm": 0.532133655138365,
-      "learning_rate": 0.0006608515911655743,
-      "loss": 2.4128,
-      "step": 903
-    },
-    {
-      "epoch": 0.41448876662081613,
-      "grad_norm": 0.4951196643195566,
-      "learning_rate": 0.0006601482018295591,
-      "loss": 2.328,
-      "step": 904
-    },
-    {
-      "epoch": 0.4149472718936268,
-      "grad_norm": 0.520469692983277,
-      "learning_rate": 0.0006594444591468851,
-      "loss": 2.4142,
-      "step": 905
-    },
-    {
-      "epoch": 0.4154057771664374,
-      "grad_norm": 0.5073998689009289,
-      "learning_rate": 0.0006587403646702713,
-      "loss": 2.3617,
-      "step": 906
-    },
-    {
-      "epoch": 0.41586428243924806,
-      "grad_norm": 0.5175875183818545,
-      "learning_rate": 0.0006580359199532126,
-      "loss": 2.2793,
-      "step": 907
-    },
-    {
-      "epoch": 0.41632278771205866,
-      "grad_norm": 0.57740726855504,
-      "learning_rate": 0.000657331126549977,
-      "loss": 2.2904,
-      "step": 908
-    },
-    {
-      "epoch": 0.4167812929848693,
-      "grad_norm": 0.701423879659659,
-      "learning_rate": 0.0006566259860156014,
-      "loss": 2.298,
-      "step": 909
-    },
-    {
-      "epoch": 0.41723979825768,
-      "grad_norm": 0.535510527375861,
-      "learning_rate": 0.0006559204999058888,
-      "loss": 2.2964,
-      "step": 910
-    },
-    {
-      "epoch": 0.4176983035304906,
-      "grad_norm": 0.5256628619444164,
-      "learning_rate": 0.0006552146697774049,
-      "loss": 2.3387,
-      "step": 911
-    },
-    {
-      "epoch": 0.41815680880330125,
-      "grad_norm": 0.5225861635493781,
-      "learning_rate": 0.0006545084971874737,
-      "loss": 2.2647,
-      "step": 912
-    },
-    {
-      "epoch": 0.41861531407611186,
-      "grad_norm": 0.5218815688366558,
-      "learning_rate": 0.0006538019836941758,
-      "loss": 2.3751,
-      "step": 913
-    },
-    {
-      "epoch": 0.4190738193489225,
-      "grad_norm": 0.5578669042746827,
-      "learning_rate": 0.0006530951308563431,
-      "loss": 2.3562,
-      "step": 914
-    },
-    {
-      "epoch": 0.4195323246217332,
-      "grad_norm": 0.9392581175210497,
-      "learning_rate": 0.0006523879402335567,
-      "loss": 2.3624,
-      "step": 915
-    },
-    {
-      "epoch": 0.4199908298945438,
-      "grad_norm": 0.5415878495266302,
-      "learning_rate": 0.0006516804133861429,
-      "loss": 2.4367,
-      "step": 916
-    },
-    {
-      "epoch": 0.42044933516735444,
-      "grad_norm": 0.5506350170120378,
-      "learning_rate": 0.0006509725518751698,
-      "loss": 2.3002,
-      "step": 917
-    },
-    {
-      "epoch": 0.42090784044016505,
-      "grad_norm": 0.5050028595066641,
-      "learning_rate": 0.0006502643572624438,
-      "loss": 2.3566,
-      "step": 918
-    },
-    {
-      "epoch": 0.4213663457129757,
-      "grad_norm": 0.5036094391132381,
-      "learning_rate": 0.0006495558311105064,
-      "loss": 2.3071,
-      "step": 919
-    },
-    {
-      "epoch": 0.4218248509857863,
-      "grad_norm": 0.5209996208582774,
-      "learning_rate": 0.0006488469749826305,
-      "loss": 2.3898,
-      "step": 920
-    },
-    {
-      "epoch": 0.422283356258597,
-      "grad_norm": 0.5246965980947982,
-      "learning_rate": 0.000648137790442817,
-      "loss": 2.3371,
-      "step": 921
-    },
-    {
-      "epoch": 0.42274186153140764,
-      "grad_norm": 0.5379272491322112,
-      "learning_rate": 0.0006474282790557916,
-      "loss": 2.2943,
-      "step": 922
-    },
-    {
-      "epoch": 0.42320036680421824,
-      "grad_norm": 0.5211777129187266,
-      "learning_rate": 0.000646718442387001,
-      "loss": 2.3986,
-      "step": 923
-    },
-    {
-      "epoch": 0.4236588720770289,
-      "grad_norm": 0.522307915872655,
-      "learning_rate": 0.0006460082820026094,
-      "loss": 2.3195,
-      "step": 924
-    },
-    {
-      "epoch": 0.4241173773498395,
-      "grad_norm": 0.513350423115602,
-      "learning_rate": 0.0006452977994694959,
-      "loss": 2.412,
-      "step": 925
-    },
-    {
-      "epoch": 0.42457588262265017,
-      "grad_norm": 0.5610438604318756,
-      "learning_rate": 0.0006445869963552496,
-      "loss": 2.3671,
-      "step": 926
-    },
-    {
-      "epoch": 0.4250343878954608,
-      "grad_norm": 0.5658152104757717,
-      "learning_rate": 0.0006438758742281672,
-      "loss": 2.3608,
-      "step": 927
-    },
-    {
-      "epoch": 0.42549289316827144,
-      "grad_norm": 0.5324757382042445,
-      "learning_rate": 0.0006431644346572495,
-      "loss": 2.3646,
-      "step": 928
-    },
-    {
-      "epoch": 0.4259513984410821,
-      "grad_norm": 0.5264991091118995,
-      "learning_rate": 0.0006424526792121974,
-      "loss": 2.3294,
-      "step": 929
-    },
-    {
-      "epoch": 0.4264099037138927,
-      "grad_norm": 0.5340779245057918,
-      "learning_rate": 0.0006417406094634089,
-      "loss": 2.3161,
-      "step": 930
-    },
-    {
-      "epoch": 0.42686840898670336,
-      "grad_norm": 0.5370586312772534,
-      "learning_rate": 0.0006410282269819756,
-      "loss": 2.3421,
-      "step": 931
-    },
-    {
-      "epoch": 0.42732691425951397,
-      "grad_norm": 0.5595684257445558,
-      "learning_rate": 0.0006403155333396787,
-      "loss": 2.2591,
-      "step": 932
-    },
-    {
-      "epoch": 0.42778541953232463,
-      "grad_norm": 0.5540813913159776,
-      "learning_rate": 0.0006396025301089863,
-      "loss": 2.4437,
-      "step": 933
-    },
-    {
-      "epoch": 0.42824392480513523,
-      "grad_norm": 0.5377539419228148,
-      "learning_rate": 0.0006388892188630493,
-      "loss": 2.4049,
-      "step": 934
-    },
-    {
-      "epoch": 0.4287024300779459,
-      "grad_norm": 0.5296942415325342,
-      "learning_rate": 0.0006381756011756982,
-      "loss": 2.4196,
-      "step": 935
-    },
-    {
-      "epoch": 0.42916093535075656,
-      "grad_norm": 0.5314545020854935,
-      "learning_rate": 0.0006374616786214403,
-      "loss": 2.2886,
-      "step": 936
-    },
-    {
-      "epoch": 0.42961944062356716,
-      "grad_norm": 0.5117343007747824,
-      "learning_rate": 0.0006367474527754544,
-      "loss": 2.3709,
-      "step": 937
-    },
-    {
-      "epoch": 0.4300779458963778,
-      "grad_norm": 0.4985783184190204,
-      "learning_rate": 0.0006360329252135894,
-      "loss": 2.3249,
-      "step": 938
-    },
-    {
-      "epoch": 0.4305364511691884,
-      "grad_norm": 0.544426460176613,
-      "learning_rate": 0.0006353180975123595,
-      "loss": 2.3236,
-      "step": 939
-    },
-    {
-      "epoch": 0.4309949564419991,
-      "grad_norm": 0.5245154649490273,
-      "learning_rate": 0.0006346029712489413,
-      "loss": 2.3386,
-      "step": 940
-    },
-    {
-      "epoch": 0.4314534617148097,
-      "grad_norm": 0.5945867388179042,
-      "learning_rate": 0.0006338875480011698,
-      "loss": 2.3717,
-      "step": 941
-    },
-    {
-      "epoch": 0.43191196698762035,
-      "grad_norm": 0.5065036793561452,
-      "learning_rate": 0.0006331718293475357,
-      "loss": 2.3392,
-      "step": 942
-    },
-    {
-      "epoch": 0.432370472260431,
-      "grad_norm": 0.5231872011748798,
-      "learning_rate": 0.0006324558168671811,
-      "loss": 2.4427,
-      "step": 943
-    },
-    {
-      "epoch": 0.4328289775332416,
-      "grad_norm": 0.5258907183522172,
-      "learning_rate": 0.0006317395121398968,
-      "loss": 2.3498,
-      "step": 944
-    },
-    {
-      "epoch": 0.4332874828060523,
-      "grad_norm": 0.5100044233773697,
-      "learning_rate": 0.0006310229167461179,
-      "loss": 2.331,
-      "step": 945
-    },
-    {
-      "epoch": 0.4337459880788629,
-      "grad_norm": 0.5615499645684479,
-      "learning_rate": 0.0006303060322669214,
-      "loss": 2.3501,
-      "step": 946
-    },
-    {
-      "epoch": 0.43420449335167355,
-      "grad_norm": 0.554779704919431,
-      "learning_rate": 0.0006295888602840214,
-      "loss": 2.281,
-      "step": 947
-    },
-    {
-      "epoch": 0.4346629986244842,
-      "grad_norm": 0.5235513640685839,
-      "learning_rate": 0.0006288714023797671,
-      "loss": 2.3445,
-      "step": 948
-    },
-    {
-      "epoch": 0.4351215038972948,
-      "grad_norm": 0.5468245634168923,
-      "learning_rate": 0.000628153660137138,
-      "loss": 2.366,
-      "step": 949
-    },
-    {
-      "epoch": 0.4355800091701055,
-      "grad_norm": 0.5315832903070925,
-      "learning_rate": 0.0006274356351397413,
-      "loss": 2.2923,
-      "step": 950
-    },
-    {
-      "epoch": 0.4360385144429161,
-      "grad_norm": 0.5377372847081662,
-      "learning_rate": 0.0006267173289718079,
-      "loss": 2.2839,
-      "step": 951
-    },
-    {
-      "epoch": 0.43649701971572674,
-      "grad_norm": 0.5493829767367672,
-      "learning_rate": 0.000625998743218189,
-      "loss": 2.3411,
-      "step": 952
-    },
-    {
-      "epoch": 0.43695552498853735,
-      "grad_norm": 0.5970662729097328,
-      "learning_rate": 0.000625279879464353,
-      "loss": 2.3304,
-      "step": 953
-    },
-    {
-      "epoch": 0.437414030261348,
-      "grad_norm": 0.5511523230708906,
-      "learning_rate": 0.000624560739296381,
-      "loss": 2.3901,
-      "step": 954
-    },
-    {
-      "epoch": 0.43787253553415867,
-      "grad_norm": 0.5098893526528427,
-      "learning_rate": 0.0006238413243009648,
-      "loss": 2.3054,
-      "step": 955
-    },
-    {
-      "epoch": 0.4383310408069693,
-      "grad_norm": 0.5057477436650281,
-      "learning_rate": 0.000623121636065402,
-      "loss": 2.2589,
-      "step": 956
-    },
-    {
-      "epoch": 0.43878954607977994,
-      "grad_norm": 0.5136456478334754,
-      "learning_rate": 0.0006224016761775933,
-      "loss": 2.3229,
-      "step": 957
-    },
-    {
-      "epoch": 0.43924805135259054,
-      "grad_norm": 0.5320447558734738,
-      "learning_rate": 0.0006216814462260386,
-      "loss": 2.458,
-      "step": 958
-    },
-    {
-      "epoch": 0.4397065566254012,
-      "grad_norm": 0.5113781986095742,
-      "learning_rate": 0.0006209609477998338,
-      "loss": 2.348,
-      "step": 959
-    },
-    {
-      "epoch": 0.4401650618982118,
-      "grad_norm": 0.5014432163737782,
-      "learning_rate": 0.0006202401824886674,
-      "loss": 2.3272,
-      "step": 960
-    },
-    {
-      "epoch": 0.44062356717102247,
-      "grad_norm": 0.5046580727995201,
-      "learning_rate": 0.0006195191518828162,
-      "loss": 2.2908,
-      "step": 961
-    },
-    {
-      "epoch": 0.44108207244383313,
-      "grad_norm": 0.49555092332905754,
-      "learning_rate": 0.0006187978575731427,
-      "loss": 2.2938,
-      "step": 962
-    },
-    {
-      "epoch": 0.44154057771664373,
-      "grad_norm": 0.4889095291203438,
-      "learning_rate": 0.0006180763011510911,
-      "loss": 2.3558,
-      "step": 963
-    },
-    {
-      "epoch": 0.4419990829894544,
-      "grad_norm": 0.5264514157258534,
-      "learning_rate": 0.000617354484208684,
-      "loss": 2.2988,
-      "step": 964
-    },
-    {
-      "epoch": 0.442457588262265,
-      "grad_norm": 0.5107484336410916,
-      "learning_rate": 0.0006166324083385189,
-      "loss": 2.2967,
-      "step": 965
-    },
-    {
-      "epoch": 0.44291609353507566,
-      "grad_norm": 0.48827685354160866,
-      "learning_rate": 0.0006159100751337642,
-      "loss": 2.298,
-      "step": 966
-    },
-    {
-      "epoch": 0.44337459880788627,
-      "grad_norm": 0.5194734054017448,
-      "learning_rate": 0.0006151874861881565,
-      "loss": 2.2503,
-      "step": 967
-    },
-    {
-      "epoch": 0.4438331040806969,
-      "grad_norm": 0.5216001788779987,
-      "learning_rate": 0.0006144646430959964,
-      "loss": 2.3657,
-      "step": 968
-    },
-    {
-      "epoch": 0.4442916093535076,
-      "grad_norm": 0.4927478131300791,
-      "learning_rate": 0.0006137415474521454,
-      "loss": 2.3662,
-      "step": 969
-    },
-    {
-      "epoch": 0.4447501146263182,
-      "grad_norm": 0.5162232880841452,
-      "learning_rate": 0.0006130182008520222,
-      "loss": 2.3468,
-      "step": 970
-    },
-    {
-      "epoch": 0.44520861989912885,
-      "grad_norm": 0.5240323624009054,
-      "learning_rate": 0.000612294604891599,
-      "loss": 2.3458,
-      "step": 971
-    },
-    {
-      "epoch": 0.44566712517193946,
-      "grad_norm": 0.5212647034755511,
-      "learning_rate": 0.0006115707611673986,
-      "loss": 2.3827,
-      "step": 972
-    },
-    {
-      "epoch": 0.4461256304447501,
-      "grad_norm": 0.5219294407032694,
-      "learning_rate": 0.0006108466712764902,
-      "loss": 2.2846,
-      "step": 973
-    },
-    {
-      "epoch": 0.4465841357175607,
-      "grad_norm": 0.49592623031773037,
-      "learning_rate": 0.0006101223368164858,
-      "loss": 2.3407,
-      "step": 974
-    },
-    {
-      "epoch": 0.4470426409903714,
-      "grad_norm": 0.5158698106338042,
-      "learning_rate": 0.0006093977593855375,
-      "loss": 2.2964,
-      "step": 975
-    },
-    {
-      "epoch": 0.44750114626318205,
-      "grad_norm": 0.5406196094651438,
-      "learning_rate": 0.0006086729405823335,
-      "loss": 2.3647,
-      "step": 976
-    },
-    {
-      "epoch": 0.44795965153599265,
-      "grad_norm": 0.5308116347170453,
-      "learning_rate": 0.0006079478820060943,
-      "loss": 2.3579,
-      "step": 977
-    },
-    {
-      "epoch": 0.4484181568088033,
-      "grad_norm": 0.48837923854846327,
-      "learning_rate": 0.0006072225852565695,
-      "loss": 2.3208,
-      "step": 978
-    },
-    {
-      "epoch": 0.4488766620816139,
-      "grad_norm": 0.5081242881092637,
-      "learning_rate": 0.0006064970519340341,
-      "loss": 2.3953,
-      "step": 979
-    },
-    {
-      "epoch": 0.4493351673544246,
-      "grad_norm": 0.5140611375700801,
-      "learning_rate": 0.0006057712836392856,
-      "loss": 2.2934,
-      "step": 980
-    },
-    {
-      "epoch": 0.4497936726272352,
-      "grad_norm": 0.5131146718609415,
-      "learning_rate": 0.0006050452819736389,
-      "loss": 2.3395,
-      "step": 981
-    },
-    {
-      "epoch": 0.45025217790004585,
-      "grad_norm": 0.5078264830571466,
-      "learning_rate": 0.000604319048538925,
-      "loss": 2.3273,
-      "step": 982
-    },
-    {
-      "epoch": 0.4507106831728565,
-      "grad_norm": 0.5433768676404067,
-      "learning_rate": 0.0006035925849374855,
-      "loss": 2.3612,
-      "step": 983
-    },
-    {
-      "epoch": 0.4511691884456671,
-      "grad_norm": 0.5491195041830904,
-      "learning_rate": 0.0006028658927721697,
-      "loss": 2.3395,
-      "step": 984
-    },
-    {
-      "epoch": 0.4516276937184778,
-      "grad_norm": 0.49770145896786605,
-      "learning_rate": 0.0006021389736463321,
-      "loss": 2.3232,
-      "step": 985
-    },
-    {
-      "epoch": 0.4520861989912884,
-      "grad_norm": 0.525438009220575,
-      "learning_rate": 0.0006014118291638271,
-      "loss": 2.2397,
-      "step": 986
-    },
-    {
-      "epoch": 0.45254470426409904,
-      "grad_norm": 0.5351837142019337,
-      "learning_rate": 0.0006006844609290065,
-      "loss": 2.3978,
-      "step": 987
-    },
-    {
-      "epoch": 0.4530032095369097,
-      "grad_norm": 0.5229811416004296,
-      "learning_rate": 0.0005999568705467161,
-      "loss": 2.3904,
-      "step": 988
-    },
-    {
-      "epoch": 0.4534617148097203,
-      "grad_norm": 0.5144381919324981,
-      "learning_rate": 0.0005992290596222915,
-      "loss": 2.3837,
-      "step": 989
-    },
-    {
-      "epoch": 0.45392022008253097,
-      "grad_norm": 0.541565904430512,
-      "learning_rate": 0.0005985010297615551,
-      "loss": 2.3109,
-      "step": 990
-    },
-    {
-      "epoch": 0.4543787253553416,
-      "grad_norm": 0.5552510652520087,
-      "learning_rate": 0.0005977727825708123,
-      "loss": 2.3422,
-      "step": 991
-    },
-    {
-      "epoch": 0.45483723062815223,
-      "grad_norm": 0.5368693231459033,
-      "learning_rate": 0.0005970443196568478,
-      "loss": 2.2898,
-      "step": 992
-    },
-    {
-      "epoch": 0.45529573590096284,
-      "grad_norm": 0.5520657391160676,
-      "learning_rate": 0.0005963156426269227,
-      "loss": 2.2233,
-      "step": 993
-    },
-    {
-      "epoch": 0.4557542411737735,
-      "grad_norm": 0.5481284044701286,
-      "learning_rate": 0.0005955867530887702,
-      "loss": 2.2933,
-      "step": 994
-    },
-    {
-      "epoch": 0.45621274644658416,
-      "grad_norm": 0.5758738876646136,
-      "learning_rate": 0.0005948576526505923,
-      "loss": 2.3395,
-      "step": 995
-    },
-    {
-      "epoch": 0.45667125171939477,
-      "grad_norm": 0.5293990156289529,
-      "learning_rate": 0.0005941283429210568,
-      "loss": 2.3169,
-      "step": 996
-    },
-    {
-      "epoch": 0.4571297569922054,
-      "grad_norm": 0.5419753241707896,
-      "learning_rate": 0.0005933988255092926,
-      "loss": 2.3704,
-      "step": 997
-    },
-    {
-      "epoch": 0.45758826226501603,
-      "grad_norm": 0.6044870618639858,
-      "learning_rate": 0.0005926691020248874,
-      "loss": 2.2749,
-      "step": 998
-    },
-    {
-      "epoch": 0.4580467675378267,
-      "grad_norm": 0.5690474163540851,
-      "learning_rate": 0.0005919391740778833,
-      "loss": 2.3379,
-      "step": 999
-    },
-    {
-      "epoch": 0.4585052728106373,
-      "grad_norm": 0.5254870927978359,
-      "learning_rate": 0.0005912090432787736,
-      "loss": 2.4109,
-      "step": 1000
-    },
-    {
-      "epoch": 0.45896377808344796,
-      "grad_norm": 0.5084821119267197,
-      "learning_rate": 0.000590478711238499,
-      "loss": 2.1924,
-      "step": 1001
-    },
-    {
-      "epoch": 0.4594222833562586,
-      "grad_norm": 0.5320567816102622,
-      "learning_rate": 0.0005897481795684446,
-      "loss": 2.2982,
-      "step": 1002
-    },
-    {
-      "epoch": 0.4598807886290692,
-      "grad_norm": 0.5171938087890373,
-      "learning_rate": 0.0005890174498804355,
-      "loss": 2.2907,
-      "step": 1003
-    },
-    {
-      "epoch": 0.4603392939018799,
-      "grad_norm": 0.5676519413031678,
-      "learning_rate": 0.0005882865237867339,
-      "loss": 2.3785,
-      "step": 1004
-    },
-    {
-      "epoch": 0.4607977991746905,
-      "grad_norm": 0.5183551812235855,
-      "learning_rate": 0.0005875554029000353,
-      "loss": 2.2933,
-      "step": 1005
-    },
-    {
-      "epoch": 0.46125630444750115,
-      "grad_norm": 0.46389927188439806,
-      "learning_rate": 0.0005868240888334653,
-      "loss": 2.2253,
-      "step": 1006
-    },
-    {
-      "epoch": 0.46171480972031176,
-      "grad_norm": 0.5133475160716632,
-      "learning_rate": 0.0005860925832005753,
-      "loss": 2.2434,
-      "step": 1007
-    },
-    {
-      "epoch": 0.4621733149931224,
-      "grad_norm": 0.5181339792411381,
-      "learning_rate": 0.0005853608876153395,
-      "loss": 2.3654,
-      "step": 1008
-    },
-    {
-      "epoch": 0.4626318202659331,
-      "grad_norm": 0.5155949069041423,
-      "learning_rate": 0.0005846290036921512,
-      "loss": 2.3155,
-      "step": 1009
-    },
-    {
-      "epoch": 0.4630903255387437,
-      "grad_norm": 0.5613921067862823,
-      "learning_rate": 0.0005838969330458195,
-      "loss": 2.2294,
-      "step": 1010
-    },
-    {
-      "epoch": 0.46354883081155435,
-      "grad_norm": 0.5335292059251702,
-      "learning_rate": 0.0005831646772915651,
-      "loss": 2.4538,
-      "step": 1011
-    },
-    {
-      "epoch": 0.46400733608436495,
-      "grad_norm": 0.5150026090344495,
-      "learning_rate": 0.0005824322380450173,
-      "loss": 2.303,
-      "step": 1012
-    },
-    {
-      "epoch": 0.4644658413571756,
-      "grad_norm": 0.5211176133929212,
-      "learning_rate": 0.0005816996169222102,
-      "loss": 2.4213,
-      "step": 1013
-    },
-    {
-      "epoch": 0.4649243466299862,
-      "grad_norm": 0.510608896856762,
-      "learning_rate": 0.0005809668155395793,
-      "loss": 2.3518,
-      "step": 1014
-    },
-    {
-      "epoch": 0.4653828519027969,
-      "grad_norm": 0.5021826865287341,
-      "learning_rate": 0.0005802338355139578,
-      "loss": 2.3217,
-      "step": 1015
-    },
-    {
-      "epoch": 0.46584135717560754,
-      "grad_norm": 0.48427732445631627,
-      "learning_rate": 0.0005795006784625728,
-      "loss": 2.3523,
-      "step": 1016
-    },
-    {
-      "epoch": 0.46629986244841815,
-      "grad_norm": 0.5194348256092414,
-      "learning_rate": 0.0005787673460030423,
-      "loss": 2.3887,
-      "step": 1017
-    },
-    {
-      "epoch": 0.4667583677212288,
-      "grad_norm": 0.5102005421873763,
-      "learning_rate": 0.000578033839753371,
-      "loss": 2.2478,
-      "step": 1018
-    },
-    {
-      "epoch": 0.4672168729940394,
-      "grad_norm": 0.5086101152768956,
-      "learning_rate": 0.0005773001613319476,
-      "loss": 2.2574,
-      "step": 1019
-    },
-    {
-      "epoch": 0.4676753782668501,
-      "grad_norm": 0.4979275658697624,
-      "learning_rate": 0.00057656631235754,
-      "loss": 2.2779,
-      "step": 1020
-    },
-    {
-      "epoch": 0.46813388353966073,
-      "grad_norm": 0.5158103576744597,
-      "learning_rate": 0.0005758322944492929,
-      "loss": 2.2914,
-      "step": 1021
-    },
-    {
-      "epoch": 0.46859238881247134,
-      "grad_norm": 0.5410258239686794,
-      "learning_rate": 0.0005750981092267237,
-      "loss": 2.3542,
-      "step": 1022
-    },
-    {
-      "epoch": 0.469050894085282,
-      "grad_norm": 0.5109814198454611,
-      "learning_rate": 0.0005743637583097183,
-      "loss": 2.2486,
-      "step": 1023
-    },
-    {
-      "epoch": 0.4695093993580926,
-      "grad_norm": 0.5419522913004196,
-      "learning_rate": 0.0005736292433185291,
-      "loss": 2.3244,
-      "step": 1024
-    },
-    {
-      "epoch": 0.46996790463090327,
-      "grad_norm": 0.5749021707699442,
-      "learning_rate": 0.0005728945658737699,
-      "loss": 2.4558,
-      "step": 1025
-    },
-    {
-      "epoch": 0.47042640990371387,
-      "grad_norm": 0.5501950986341766,
-      "learning_rate": 0.0005721597275964133,
-      "loss": 2.3745,
-      "step": 1026
-    },
-    {
-      "epoch": 0.47088491517652453,
-      "grad_norm": 0.48726171818188874,
-      "learning_rate": 0.0005714247301077865,
-      "loss": 2.3239,
-      "step": 1027
-    },
-    {
-      "epoch": 0.4713434204493352,
-      "grad_norm": 0.5095694301860038,
-      "learning_rate": 0.0005706895750295682,
-      "loss": 2.3107,
-      "step": 1028
-    },
-    {
-      "epoch": 0.4718019257221458,
-      "grad_norm": 0.5651333702423922,
-      "learning_rate": 0.0005699542639837844,
-      "loss": 2.3277,
-      "step": 1029
-    },
-    {
-      "epoch": 0.47226043099495646,
-      "grad_norm": 0.5594690538747507,
-      "learning_rate": 0.0005692187985928055,
-      "loss": 2.3229,
-      "step": 1030
-    },
-    {
-      "epoch": 0.47271893626776706,
-      "grad_norm": 0.5439923373549331,
-      "learning_rate": 0.0005684831804793427,
-      "loss": 2.2896,
-      "step": 1031
-    },
-    {
-      "epoch": 0.4731774415405777,
-      "grad_norm": 0.5191865293645964,
-      "learning_rate": 0.0005677474112664438,
-      "loss": 2.3506,
-      "step": 1032
-    },
-    {
-      "epoch": 0.47363594681338833,
-      "grad_norm": 0.49266831460404376,
-      "learning_rate": 0.0005670114925774899,
-      "loss": 2.3187,
-      "step": 1033
-    },
-    {
-      "epoch": 0.474094452086199,
-      "grad_norm": 0.5469604736187815,
-      "learning_rate": 0.0005662754260361924,
-      "loss": 2.391,
-      "step": 1034
-    },
-    {
-      "epoch": 0.47455295735900965,
-      "grad_norm": 0.5097605035422281,
-      "learning_rate": 0.0005655392132665884,
-      "loss": 2.366,
-      "step": 1035
-    },
-    {
-      "epoch": 0.47501146263182026,
-      "grad_norm": 0.5546219461281521,
-      "learning_rate": 0.000564802855893038,
-      "loss": 2.3223,
-      "step": 1036
-    },
-    {
-      "epoch": 0.4754699679046309,
-      "grad_norm": 0.5208139152433666,
-      "learning_rate": 0.0005640663555402198,
-      "loss": 2.3449,
-      "step": 1037
-    },
-    {
-      "epoch": 0.4759284731774415,
-      "grad_norm": 0.5291182207260684,
-      "learning_rate": 0.0005633297138331285,
-      "loss": 2.2458,
-      "step": 1038
-    },
-    {
-      "epoch": 0.4763869784502522,
-      "grad_norm": 0.5240093237475879,
-      "learning_rate": 0.0005625929323970705,
-      "loss": 2.3207,
-      "step": 1039
-    },
-    {
-      "epoch": 0.4768454837230628,
-      "grad_norm": 0.508027181041645,
-      "learning_rate": 0.0005618560128576603,
-      "loss": 2.2611,
-      "step": 1040
-    },
-    {
-      "epoch": 0.47730398899587345,
-      "grad_norm": 0.5498829033456516,
-      "learning_rate": 0.0005611189568408173,
-      "loss": 2.3493,
-      "step": 1041
-    },
-    {
-      "epoch": 0.4777624942686841,
-      "grad_norm": 0.5197343167264477,
-      "learning_rate": 0.0005603817659727619,
-      "loss": 2.3636,
-      "step": 1042
-    },
-    {
-      "epoch": 0.4782209995414947,
-      "grad_norm": 0.5176535299542298,
-      "learning_rate": 0.0005596444418800121,
-      "loss": 2.2596,
-      "step": 1043
-    },
-    {
-      "epoch": 0.4786795048143054,
-      "grad_norm": 0.5080009129256254,
-      "learning_rate": 0.0005589069861893798,
-      "loss": 2.2631,
-      "step": 1044
-    },
-    {
-      "epoch": 0.479138010087116,
-      "grad_norm": 0.4985842682917299,
-      "learning_rate": 0.0005581694005279673,
-      "loss": 2.3579,
-      "step": 1045
-    },
-    {
-      "epoch": 0.47959651535992665,
-      "grad_norm": 0.5199024319327121,
-      "learning_rate": 0.0005574316865231637,
-      "loss": 2.3903,
-      "step": 1046
-    },
-    {
-      "epoch": 0.48005502063273725,
-      "grad_norm": 0.5387063664077895,
-      "learning_rate": 0.0005566938458026411,
-      "loss": 2.3205,
-      "step": 1047
-    },
-    {
-      "epoch": 0.4805135259055479,
-      "grad_norm": 0.5292817901911053,
-      "learning_rate": 0.0005559558799943514,
-      "loss": 2.3011,
-      "step": 1048
-    },
-    {
-      "epoch": 0.48097203117835857,
-      "grad_norm": 0.5876307291233482,
-      "learning_rate": 0.0005552177907265223,
-      "loss": 2.3756,
-      "step": 1049
-    },
-    {
-      "epoch": 0.4814305364511692,
-      "grad_norm": 0.5466765490711343,
-      "learning_rate": 0.000554479579627654,
-      "loss": 2.2629,
-      "step": 1050
-    },
-    {
-      "epoch": 0.48188904172397984,
-      "grad_norm": 0.51616949422207,
-      "learning_rate": 0.0005537412483265157,
-      "loss": 2.3961,
-      "step": 1051
-    },
-    {
-      "epoch": 0.48234754699679044,
-      "grad_norm": 0.5588471654369795,
-      "learning_rate": 0.0005530027984521413,
-      "loss": 2.3298,
-      "step": 1052
-    },
-    {
-      "epoch": 0.4828060522696011,
-      "grad_norm": 0.5273767019502302,
-      "learning_rate": 0.0005522642316338268,
-      "loss": 2.2478,
-      "step": 1053
-    },
-    {
-      "epoch": 0.4832645575424117,
-      "grad_norm": 0.5129114397089616,
-      "learning_rate": 0.0005515255495011259,
-      "loss": 2.2741,
-      "step": 1054
-    },
-    {
-      "epoch": 0.48372306281522237,
-      "grad_norm": 0.509225691162859,
-      "learning_rate": 0.0005507867536838472,
-      "loss": 2.2902,
-      "step": 1055
-    },
-    {
-      "epoch": 0.48418156808803303,
-      "grad_norm": 0.4976674934985514,
-      "learning_rate": 0.0005500478458120492,
-      "loss": 2.2948,
-      "step": 1056
-    },
-    {
-      "epoch": 0.48464007336084364,
-      "grad_norm": 0.5216645166721955,
-      "learning_rate": 0.0005493088275160387,
-      "loss": 2.3619,
-      "step": 1057
-    },
-    {
-      "epoch": 0.4850985786336543,
-      "grad_norm": 0.5017758610279464,
-      "learning_rate": 0.0005485697004263657,
-      "loss": 2.3594,
-      "step": 1058
-    },
-    {
-      "epoch": 0.4855570839064649,
-      "grad_norm": 0.4905370736187145,
-      "learning_rate": 0.0005478304661738199,
-      "loss": 2.3717,
-      "step": 1059
-    },
-    {
-      "epoch": 0.48601558917927556,
-      "grad_norm": 0.541600348833721,
-      "learning_rate": 0.0005470911263894279,
-      "loss": 2.2612,
-      "step": 1060
-    },
-    {
-      "epoch": 0.4864740944520862,
-      "grad_norm": 0.5100002847176492,
-      "learning_rate": 0.0005463516827044491,
-      "loss": 2.3423,
-      "step": 1061
-    },
-    {
-      "epoch": 0.48693259972489683,
-      "grad_norm": 0.5266384238044451,
-      "learning_rate": 0.000545612136750372,
-      "loss": 2.372,
-      "step": 1062
-    },
-    {
-      "epoch": 0.4873911049977075,
-      "grad_norm": 0.4871968420442387,
-      "learning_rate": 0.0005448724901589107,
-      "loss": 2.3202,
-      "step": 1063
-    },
-    {
-      "epoch": 0.4878496102705181,
-      "grad_norm": 0.518646299042885,
-      "learning_rate": 0.0005441327445620014,
-      "loss": 2.3784,
-      "step": 1064
-    },
-    {
-      "epoch": 0.48830811554332876,
-      "grad_norm": 0.5181826333928549,
-      "learning_rate": 0.0005433929015917988,
-      "loss": 2.293,
-      "step": 1065
-    },
-    {
-      "epoch": 0.48876662081613936,
-      "grad_norm": 0.4838765999287423,
-      "learning_rate": 0.0005426529628806724,
-      "loss": 2.262,
-      "step": 1066
-    },
-    {
-      "epoch": 0.48922512608895,
-      "grad_norm": 0.5481109113643567,
-      "learning_rate": 0.0005419129300612029,
-      "loss": 2.2609,
-      "step": 1067
-    },
-    {
-      "epoch": 0.4896836313617607,
-      "grad_norm": 0.5187892365410456,
-      "learning_rate": 0.000541172804766179,
-      "loss": 2.3238,
-      "step": 1068
-    },
-    {
-      "epoch": 0.4901421366345713,
-      "grad_norm": 0.5206082275824148,
-      "learning_rate": 0.0005404325886285927,
-      "loss": 2.1714,
-      "step": 1069
-    },
-    {
-      "epoch": 0.49060064190738195,
-      "grad_norm": 0.485595231082355,
-      "learning_rate": 0.000539692283281637,
-      "loss": 2.3656,
-      "step": 1070
-    },
-    {
-      "epoch": 0.49105914718019256,
-      "grad_norm": 0.5194125127181125,
-      "learning_rate": 0.0005389518903587017,
-      "loss": 2.34,
-      "step": 1071
-    },
-    {
-      "epoch": 0.4915176524530032,
-      "grad_norm": 0.5008911376942706,
-      "learning_rate": 0.0005382114114933695,
-      "loss": 2.2248,
-      "step": 1072
-    },
-    {
-      "epoch": 0.4919761577258138,
-      "grad_norm": 0.4916901928547721,
-      "learning_rate": 0.0005374708483194132,
-      "loss": 2.2194,
-      "step": 1073
-    },
-    {
-      "epoch": 0.4924346629986245,
-      "grad_norm": 0.5229336115612288,
-      "learning_rate": 0.000536730202470791,
-      "loss": 2.3339,
-      "step": 1074
-    },
-    {
-      "epoch": 0.49289316827143514,
-      "grad_norm": 0.5128726972528522,
-      "learning_rate": 0.0005359894755816443,
-      "loss": 2.3577,
-      "step": 1075
-    },
-    {
-      "epoch": 0.49335167354424575,
-      "grad_norm": 0.5263219817943153,
-      "learning_rate": 0.0005352486692862926,
-      "loss": 2.2835,
-      "step": 1076
-    },
-    {
-      "epoch": 0.4938101788170564,
-      "grad_norm": 0.5261585495779395,
-      "learning_rate": 0.0005345077852192307,
-      "loss": 2.2519,
-      "step": 1077
-    },
-    {
-      "epoch": 0.494268684089867,
-      "grad_norm": 0.4834979542492456,
-      "learning_rate": 0.0005337668250151254,
-      "loss": 2.3607,
-      "step": 1078
-    },
-    {
-      "epoch": 0.4947271893626777,
-      "grad_norm": 0.505555061589212,
-      "learning_rate": 0.0005330257903088111,
-      "loss": 2.2738,
-      "step": 1079
-    },
-    {
-      "epoch": 0.4951856946354883,
-      "grad_norm": 0.5137439756956772,
-      "learning_rate": 0.000532284682735287,
-      "loss": 2.3956,
-      "step": 1080
-    },
-    {
-      "epoch": 0.49564419990829894,
-      "grad_norm": 0.5310359183386704,
-      "learning_rate": 0.0005315435039297124,
-      "loss": 2.2679,
-      "step": 1081
-    },
-    {
-      "epoch": 0.4961027051811096,
-      "grad_norm": 0.5037058766404054,
-      "learning_rate": 0.0005308022555274046,
-      "loss": 2.3311,
-      "step": 1082
-    },
-    {
-      "epoch": 0.4965612104539202,
-      "grad_norm": 0.5305208080121452,
-      "learning_rate": 0.0005300609391638336,
-      "loss": 2.3785,
-      "step": 1083
-    },
-    {
-      "epoch": 0.49701971572673087,
-      "grad_norm": 0.5346876635037718,
-      "learning_rate": 0.0005293195564746201,
-      "loss": 2.2848,
-      "step": 1084
-    },
-    {
-      "epoch": 0.4974782209995415,
-      "grad_norm": 0.5126900647666846,
-      "learning_rate": 0.0005285781090955304,
-      "loss": 2.4006,
-      "step": 1085
-    },
-    {
-      "epoch": 0.49793672627235214,
-      "grad_norm": 0.49762058784339125,
-      "learning_rate": 0.0005278365986624743,
-      "loss": 2.282,
-      "step": 1086
-    },
-    {
-      "epoch": 0.49839523154516274,
-      "grad_norm": 0.5286134721345837,
-      "learning_rate": 0.0005270950268115001,
-      "loss": 2.3861,
-      "step": 1087
-    },
-    {
-      "epoch": 0.4988537368179734,
-      "grad_norm": 0.5085532471135135,
-      "learning_rate": 0.0005263533951787919,
-      "loss": 2.3382,
-      "step": 1088
-    },
-    {
-      "epoch": 0.49931224209078406,
-      "grad_norm": 0.49029926235376703,
-      "learning_rate": 0.000525611705400666,
-      "loss": 2.3439,
-      "step": 1089
-    },
-    {
-      "epoch": 0.49977074736359467,
-      "grad_norm": 0.5226941608676834,
-      "learning_rate": 0.0005248699591135664,
-      "loss": 2.3816,
-      "step": 1090
-    },
-    {
-      "epoch": 0.5002292526364053,
-      "grad_norm": 0.5471993433950821,
-      "learning_rate": 0.0005241281579540618,
-      "loss": 2.3312,
-      "step": 1091
-    },
-    {
-      "epoch": 0.5006877579092159,
-      "grad_norm": 0.5577185020166487,
-      "learning_rate": 0.0005233863035588427,
-      "loss": 2.3542,
-      "step": 1092
-    },
-    {
-      "epoch": 0.5011462631820266,
-      "grad_norm": 0.518074927902605,
-      "learning_rate": 0.0005226443975647161,
-      "loss": 2.3229,
-      "step": 1093
-    },
-    {
-      "epoch": 0.5016047684548373,
-      "grad_norm": 0.4957147064464304,
-      "learning_rate": 0.0005219024416086036,
-      "loss": 2.2977,
-      "step": 1094
-    },
-    {
-      "epoch": 0.5020632737276479,
-      "grad_norm": 0.5080781972858949,
-      "learning_rate": 0.0005211604373275366,
-      "loss": 2.3653,
-      "step": 1095
-    },
-    {
-      "epoch": 0.5025217790004585,
-      "grad_norm": 0.4608277804254298,
-      "learning_rate": 0.0005204183863586533,
-      "loss": 2.3012,
-      "step": 1096
-    },
-    {
-      "epoch": 0.5029802842732691,
-      "grad_norm": 0.5225269164555655,
-      "learning_rate": 0.0005196762903391951,
-      "loss": 2.3529,
-      "step": 1097
-    },
-    {
-      "epoch": 0.5034387895460798,
-      "grad_norm": 0.5152343248489477,
-      "learning_rate": 0.0005189341509065023,
-      "loss": 2.3482,
-      "step": 1098
-    },
-    {
-      "epoch": 0.5038972948188905,
-      "grad_norm": 0.49267390012618256,
-      "learning_rate": 0.0005181919696980112,
-      "loss": 2.3037,
-      "step": 1099
-    },
-    {
-      "epoch": 0.504355800091701,
-      "grad_norm": 0.5194725766698038,
-      "learning_rate": 0.0005174497483512506,
-      "loss": 2.3256,
-      "step": 1100
-    },
-    {
-      "epoch": 0.5048143053645117,
-      "grad_norm": 0.49905007878199314,
-      "learning_rate": 0.0005167074885038374,
-      "loss": 2.3432,
-      "step": 1101
-    },
-    {
-      "epoch": 0.5052728106373223,
-      "grad_norm": 0.49960677812280174,
-      "learning_rate": 0.0005159651917934735,
-      "loss": 2.1905,
-      "step": 1102
-    },
-    {
-      "epoch": 0.505731315910133,
-      "grad_norm": 0.4956888924934793,
-      "learning_rate": 0.0005152228598579428,
-      "loss": 2.3172,
-      "step": 1103
-    },
-    {
-      "epoch": 0.5061898211829436,
-      "grad_norm": 0.5005746838956409,
-      "learning_rate": 0.000514480494335106,
-      "loss": 2.2013,
-      "step": 1104
-    },
-    {
-      "epoch": 0.5066483264557542,
-      "grad_norm": 0.5448923325135996,
-      "learning_rate": 0.0005137380968628983,
-      "loss": 2.365,
-      "step": 1105
-    },
-    {
-      "epoch": 0.5071068317285649,
-      "grad_norm": 0.5019719914868719,
-      "learning_rate": 0.0005129956690793255,
-      "loss": 2.3063,
-      "step": 1106
-    },
-    {
-      "epoch": 0.5075653370013755,
-      "grad_norm": 0.5185821780947759,
-      "learning_rate": 0.0005122532126224601,
-      "loss": 2.3128,
-      "step": 1107
-    },
-    {
-      "epoch": 0.5080238422741862,
-      "grad_norm": 0.5350893435194712,
-      "learning_rate": 0.0005115107291304378,
-      "loss": 2.2742,
-      "step": 1108
-    },
-    {
-      "epoch": 0.5084823475469968,
-      "grad_norm": 0.520073454035653,
-      "learning_rate": 0.0005107682202414544,
-      "loss": 2.3225,
-      "step": 1109
-    },
-    {
-      "epoch": 0.5089408528198074,
-      "grad_norm": 0.5450740720447317,
-      "learning_rate": 0.0005100256875937613,
-      "loss": 2.3666,
-      "step": 1110
-    },
-    {
-      "epoch": 0.509399358092618,
-      "grad_norm": 0.5144549253467725,
-      "learning_rate": 0.0005092831328256625,
-      "loss": 2.3402,
-      "step": 1111
-    },
-    {
-      "epoch": 0.5098578633654287,
-      "grad_norm": 0.51448134409717,
-      "learning_rate": 0.0005085405575755105,
-      "loss": 2.3787,
-      "step": 1112
-    },
-    {
-      "epoch": 0.5103163686382394,
-      "grad_norm": 0.5660825040501342,
-      "learning_rate": 0.0005077979634817034,
-      "loss": 2.3001,
-      "step": 1113
-    },
-    {
-      "epoch": 0.51077487391105,
-      "grad_norm": 0.657334439116442,
-      "learning_rate": 0.0005070553521826808,
-      "loss": 2.3185,
-      "step": 1114
-    },
-    {
-      "epoch": 0.5112333791838606,
-      "grad_norm": 0.5321529204052988,
-      "learning_rate": 0.00050631272531692,
-      "loss": 2.2621,
-      "step": 1115
-    },
-    {
-      "epoch": 0.5116918844566712,
-      "grad_norm": 0.5421455359675029,
-      "learning_rate": 0.0005055700845229327,
-      "loss": 2.3687,
-      "step": 1116
-    },
-    {
-      "epoch": 0.5121503897294819,
-      "grad_norm": 0.576518341734363,
-      "learning_rate": 0.000504827431439262,
-      "loss": 2.327,
-      "step": 1117
-    },
-    {
-      "epoch": 0.5126088950022926,
-      "grad_norm": 0.5574498698008828,
-      "learning_rate": 0.000504084767704477,
-      "loss": 2.3514,
-      "step": 1118
-    },
-    {
-      "epoch": 0.5130674002751031,
-      "grad_norm": 0.5426996036986057,
-      "learning_rate": 0.0005033420949571712,
-      "loss": 2.4069,
-      "step": 1119
-    },
-    {
-      "epoch": 0.5135259055479138,
-      "grad_norm": 0.542047456552381,
-      "learning_rate": 0.0005025994148359574,
-      "loss": 2.2547,
-      "step": 1120
-    },
-    {
-      "epoch": 0.5139844108207244,
-      "grad_norm": 0.5015127577915873,
-      "learning_rate": 0.0005018567289794651,
-      "loss": 2.2639,
-      "step": 1121
-    },
-    {
-      "epoch": 0.5144429160935351,
-      "grad_norm": 0.5475852247935928,
-      "learning_rate": 0.0005011140390263362,
-      "loss": 2.3594,
-      "step": 1122
-    },
-    {
-      "epoch": 0.5149014213663458,
-      "grad_norm": 0.5227006444783086,
-      "learning_rate": 0.0005003713466152218,
-      "loss": 2.2924,
-      "step": 1123
-    },
-    {
-      "epoch": 0.5153599266391563,
-      "grad_norm": 0.5340588688427017,
-      "learning_rate": 0.0004996286533847783,
-      "loss": 2.2168,
-      "step": 1124
-    },
-    {
-      "epoch": 0.515818431911967,
-      "grad_norm": 0.48351149212339406,
-      "learning_rate": 0.000498885960973664,
-      "loss": 2.2754,
-      "step": 1125
-    },
-    {
-      "epoch": 0.5162769371847776,
-      "grad_norm": 0.5393205908995412,
-      "learning_rate": 0.000498143271020535,
-      "loss": 2.2921,
-      "step": 1126
-    },
-    {
-      "epoch": 0.5167354424575883,
-      "grad_norm": 0.47694607581752985,
-      "learning_rate": 0.0004974005851640428,
-      "loss": 2.3928,
-      "step": 1127
-    },
-    {
-      "epoch": 0.517193947730399,
-      "grad_norm": 0.5604429306459202,
-      "learning_rate": 0.000496657905042829,
-      "loss": 2.464,
-      "step": 1128
-    },
-    {
-      "epoch": 0.5176524530032095,
-      "grad_norm": 0.47708737258397643,
-      "learning_rate": 0.0004959152322955232,
-      "loss": 2.3841,
-      "step": 1129
-    },
-    {
-      "epoch": 0.5181109582760202,
-      "grad_norm": 0.48778398322229166,
-      "learning_rate": 0.0004951725685607382,
-      "loss": 2.2378,
-      "step": 1130
-    },
-    {
-      "epoch": 0.5185694635488308,
-      "grad_norm": 0.5078605030828082,
-      "learning_rate": 0.0004944299154770673,
-      "loss": 2.2013,
-      "step": 1131
-    },
-    {
-      "epoch": 0.5190279688216415,
-      "grad_norm": 0.4996807014386485,
-      "learning_rate": 0.0004936872746830802,
-      "loss": 2.3087,
-      "step": 1132
-    },
-    {
-      "epoch": 0.519486474094452,
-      "grad_norm": 0.5579583048263448,
-      "learning_rate": 0.0004929446478173195,
-      "loss": 2.2836,
-      "step": 1133
-    },
-    {
-      "epoch": 0.5199449793672627,
-      "grad_norm": 0.5227761823816844,
-      "learning_rate": 0.0004922020365182968,
-      "loss": 2.3975,
-      "step": 1134
-    },
-    {
-      "epoch": 0.5204034846400734,
-      "grad_norm": 0.4896507003885574,
-      "learning_rate": 0.0004914594424244897,
-      "loss": 2.381,
-      "step": 1135
-    },
-    {
-      "epoch": 0.520861989912884,
-      "grad_norm": 0.5076402871936739,
-      "learning_rate": 0.0004907168671743376,
-      "loss": 2.2973,
-      "step": 1136
-    },
-    {
-      "epoch": 0.5213204951856947,
-      "grad_norm": 0.5646963768888968,
-      "learning_rate": 0.0004899743124062387,
-      "loss": 2.3536,
-      "step": 1137
-    },
-    {
-      "epoch": 0.5217790004585052,
-      "grad_norm": 0.5295301709311149,
-      "learning_rate": 0.0004892317797585456,
-      "loss": 2.2474,
-      "step": 1138
-    },
-    {
-      "epoch": 0.5222375057313159,
-      "grad_norm": 0.5226887052716515,
-      "learning_rate": 0.0004884892708695623,
-      "loss": 2.3552,
-      "step": 1139
-    },
-    {
-      "epoch": 0.5226960110041265,
-      "grad_norm": 0.4764971447662442,
-      "learning_rate": 0.0004877467873775402,
-      "loss": 2.2024,
-      "step": 1140
-    },
-    {
-      "epoch": 0.5231545162769372,
-      "grad_norm": 0.5463461505038101,
-      "learning_rate": 0.00048700433092067473,
-      "loss": 2.3352,
-      "step": 1141
-    },
-    {
-      "epoch": 0.5236130215497479,
-      "grad_norm": 0.5516987893512786,
-      "learning_rate": 0.0004862619031371019,
-      "loss": 2.3858,
-      "step": 1142
-    },
-    {
-      "epoch": 0.5240715268225584,
-      "grad_norm": 0.6122657386660546,
-      "learning_rate": 0.0004855195056648942,
-      "loss": 2.3616,
-      "step": 1143
-    },
-    {
-      "epoch": 0.5245300320953691,
-      "grad_norm": 0.5230066479832287,
-      "learning_rate": 0.00048477714014205734,
-      "loss": 2.2984,
-      "step": 1144
-    },
-    {
-      "epoch": 0.5249885373681797,
-      "grad_norm": 0.6258026499265087,
-      "learning_rate": 0.00048403480820652644,
-      "loss": 2.2881,
-      "step": 1145
-    },
-    {
-      "epoch": 0.5254470426409904,
-      "grad_norm": 0.5234980590416085,
-      "learning_rate": 0.0004832925114961629,
-      "loss": 2.3068,
-      "step": 1146
-    },
-    {
-      "epoch": 0.5259055479138011,
-      "grad_norm": 0.48452922001223997,
-      "learning_rate": 0.0004825502516487497,
-      "loss": 2.2457,
-      "step": 1147
-    },
-    {
-      "epoch": 0.5263640531866116,
-      "grad_norm": 0.5710419112081698,
-      "learning_rate": 0.00048180803030198896,
-      "loss": 2.4448,
-      "step": 1148
-    },
-    {
-      "epoch": 0.5268225584594223,
-      "grad_norm": 0.5868368345933771,
-      "learning_rate": 0.0004810658490934979,
-      "loss": 2.3374,
-      "step": 1149
-    },
-    {
-      "epoch": 0.5272810637322329,
-      "grad_norm": 0.5769756832342341,
-      "learning_rate": 0.000480323709660805,
-      "loss": 2.3063,
-      "step": 1150
-    },
-    {
-      "epoch": 0.5277395690050436,
-      "grad_norm": 0.5762922615829852,
-      "learning_rate": 0.0004795816136413467,
-      "loss": 2.2978,
-      "step": 1151
-    },
-    {
-      "epoch": 0.5281980742778541,
-      "grad_norm": 0.5196589760162287,
-      "learning_rate": 0.00047883956267246353,
-      "loss": 2.3233,
-      "step": 1152
-    },
-    {
-      "epoch": 0.5286565795506648,
-      "grad_norm": 0.5125201684520355,
-      "learning_rate": 0.00047809755839139657,
-      "loss": 2.3083,
-      "step": 1153
-    },
-    {
-      "epoch": 0.5291150848234755,
-      "grad_norm": 0.5301772152376623,
-      "learning_rate": 0.0004773556024352841,
-      "loss": 2.3393,
-      "step": 1154
-    },
-    {
-      "epoch": 0.5295735900962861,
-      "grad_norm": 0.493850104303461,
-      "learning_rate": 0.00047661369644115754,
-      "loss": 2.3892,
-      "step": 1155
-    },
-    {
-      "epoch": 0.5300320953690968,
-      "grad_norm": 0.536219189283294,
-      "learning_rate": 0.0004758718420459383,
-      "loss": 2.2448,
-      "step": 1156
-    },
-    {
-      "epoch": 0.5304906006419073,
-      "grad_norm": 0.5072994903085426,
-      "learning_rate": 0.0004751300408864339,
-      "loss": 2.3543,
-      "step": 1157
-    },
-    {
-      "epoch": 0.530949105914718,
-      "grad_norm": 0.4860964162301103,
-      "learning_rate": 0.00047438829459933414,
-      "loss": 2.2702,
-      "step": 1158
-    },
-    {
-      "epoch": 0.5314076111875287,
-      "grad_norm": 0.5192980363859238,
-      "learning_rate": 0.0004736466048212082,
-      "loss": 2.2925,
-      "step": 1159
-    },
-    {
-      "epoch": 0.5318661164603393,
-      "grad_norm": 0.5827049392378201,
-      "learning_rate": 0.0004729049731885002,
-      "loss": 2.3585,
-      "step": 1160
-    },
-    {
-      "epoch": 0.53232462173315,
-      "grad_norm": 0.5495720130896946,
-      "learning_rate": 0.000472163401337526,
-      "loss": 2.2271,
-      "step": 1161
-    },
-    {
-      "epoch": 0.5327831270059605,
-      "grad_norm": 0.5630144266623233,
-      "learning_rate": 0.00047142189090446985,
-      "loss": 2.2634,
-      "step": 1162
-    },
-    {
-      "epoch": 0.5332416322787712,
-      "grad_norm": 0.49030553113268244,
-      "learning_rate": 0.0004706804435253802,
-      "loss": 2.2366,
-      "step": 1163
-    },
-    {
-      "epoch": 0.5337001375515819,
-      "grad_norm": 0.5147289972210803,
-      "learning_rate": 0.0004699390608361665,
-      "loss": 2.3256,
-      "step": 1164
-    },
-    {
-      "epoch": 0.5341586428243925,
-      "grad_norm": 0.6139593669167133,
-      "learning_rate": 0.0004691977444725955,
-      "loss": 2.2354,
-      "step": 1165
-    },
-    {
-      "epoch": 0.5346171480972031,
-      "grad_norm": 0.48052439280211917,
-      "learning_rate": 0.0004684564960702877,
-      "loss": 2.3489,
-      "step": 1166
-    },
-    {
-      "epoch": 0.5350756533700137,
-      "grad_norm": 0.49985393145179574,
-      "learning_rate": 0.0004677153172647131,
-      "loss": 2.3482,
-      "step": 1167
-    },
-    {
-      "epoch": 0.5355341586428244,
-      "grad_norm": 0.5162277818721523,
-      "learning_rate": 0.00046697420969118894,
-      "loss": 2.2725,
-      "step": 1168
-    },
-    {
-      "epoch": 0.535992663915635,
-      "grad_norm": 0.49028203550180144,
-      "learning_rate": 0.00046623317498487466,
-      "loss": 2.3763,
-      "step": 1169
-    },
-    {
-      "epoch": 0.5364511691884457,
-      "grad_norm": 0.5916990680886414,
-      "learning_rate": 0.0004654922147807694,
-      "loss": 2.303,
-      "step": 1170
-    },
-    {
-      "epoch": 0.5369096744612563,
-      "grad_norm": 0.4829264430500059,
-      "learning_rate": 0.00046475133071370757,
-      "loss": 2.2011,
-      "step": 1171
-    },
-    {
-      "epoch": 0.5373681797340669,
-      "grad_norm": 0.5503109568945079,
-      "learning_rate": 0.00046401052441835574,
-      "loss": 2.4203,
-      "step": 1172
-    },
-    {
-      "epoch": 0.5378266850068776,
-      "grad_norm": 0.4911297435754877,
-      "learning_rate": 0.000463269797529209,
-      "loss": 2.261,
-      "step": 1173
-    },
-    {
-      "epoch": 0.5382851902796882,
-      "grad_norm": 0.4690828085648031,
-      "learning_rate": 0.00046252915168058697,
-      "loss": 2.2228,
-      "step": 1174
-    },
-    {
-      "epoch": 0.5387436955524989,
-      "grad_norm": 0.5056195470735605,
-      "learning_rate": 0.0004617885885066305,
-      "loss": 2.2882,
-      "step": 1175
-    },
-    {
-      "epoch": 0.5392022008253095,
-      "grad_norm": 0.496921912723091,
-      "learning_rate": 0.0004610481096412984,
-      "loss": 2.336,
-      "step": 1176
-    },
-    {
-      "epoch": 0.5396607060981201,
-      "grad_norm": 0.5065036730355171,
-      "learning_rate": 0.000460307716718363,
-      "loss": 2.3053,
-      "step": 1177
-    },
-    {
-      "epoch": 0.5401192113709308,
-      "grad_norm": 0.531978943075051,
-      "learning_rate": 0.0004595674113714074,
-      "loss": 2.3028,
-      "step": 1178
-    },
-    {
-      "epoch": 0.5405777166437414,
-      "grad_norm": 0.5190974199795869,
-      "learning_rate": 0.0004588271952338212,
-      "loss": 2.2893,
-      "step": 1179
-    },
-    {
-      "epoch": 0.5410362219165521,
-      "grad_norm": 0.5208401743866742,
-      "learning_rate": 0.00045808706993879714,
-      "loss": 2.2591,
-      "step": 1180
-    },
-    {
-      "epoch": 0.5414947271893626,
-      "grad_norm": 0.5430948665002586,
-      "learning_rate": 0.00045734703711932767,
-      "loss": 2.4066,
-      "step": 1181
-    },
-    {
-      "epoch": 0.5419532324621733,
-      "grad_norm": 0.5274243944370243,
-      "learning_rate": 0.0004566070984082013,
-      "loss": 2.3808,
-      "step": 1182
-    },
-    {
-      "epoch": 0.542411737734984,
-      "grad_norm": 0.5396137785490254,
-      "learning_rate": 0.00045586725543799865,
-      "loss": 2.3515,
-      "step": 1183
-    },
-    {
-      "epoch": 0.5428702430077946,
-      "grad_norm": 0.5076216825188604,
-      "learning_rate": 0.00045512750984108937,
-      "loss": 2.3545,
-      "step": 1184
-    },
-    {
-      "epoch": 0.5433287482806052,
-      "grad_norm": 0.5012844878237079,
-      "learning_rate": 0.000454387863249628,
-      "loss": 2.2981,
-      "step": 1185
-    },
-    {
-      "epoch": 0.5437872535534158,
-      "grad_norm": 0.5256149976621481,
-      "learning_rate": 0.00045364831729555096,
-      "loss": 2.3735,
-      "step": 1186
-    },
-    {
-      "epoch": 0.5442457588262265,
-      "grad_norm": 0.5474733768118928,
-      "learning_rate": 0.0004529088736105721,
-      "loss": 2.4076,
-      "step": 1187
-    },
-    {
-      "epoch": 0.5447042640990372,
-      "grad_norm": 0.4862787198150333,
-      "learning_rate": 0.0004521695338261802,
-      "loss": 2.3455,
-      "step": 1188
-    },
-    {
-      "epoch": 0.5451627693718478,
-      "grad_norm": 0.5780409918219676,
-      "learning_rate": 0.0004514302995736344,
-      "loss": 2.2315,
-      "step": 1189
-    },
-    {
-      "epoch": 0.5456212746446584,
-      "grad_norm": 0.49411544127804646,
-      "learning_rate": 0.0004506911724839613,
-      "loss": 2.3669,
-      "step": 1190
-    },
-    {
-      "epoch": 0.546079779917469,
-      "grad_norm": 0.4768066938510597,
-      "learning_rate": 0.0004499521541879508,
-      "loss": 2.3591,
-      "step": 1191
-    },
-    {
-      "epoch": 0.5465382851902797,
-      "grad_norm": 0.5078026766010292,
-      "learning_rate": 0.00044921324631615303,
-      "loss": 2.335,
-      "step": 1192
-    },
-    {
-      "epoch": 0.5469967904630904,
-      "grad_norm": 0.558369816521602,
-      "learning_rate": 0.0004484744504988742,
-      "loss": 2.3068,
-      "step": 1193
-    },
-    {
-      "epoch": 0.547455295735901,
-      "grad_norm": 0.5410546454842105,
-      "learning_rate": 0.00044773576836617336,
-      "loss": 2.2941,
-      "step": 1194
-    },
-    {
-      "epoch": 0.5479138010087116,
-      "grad_norm": 0.5100561897382591,
-      "learning_rate": 0.0004469972015478588,
-      "loss": 2.2436,
-      "step": 1195
-    },
-    {
-      "epoch": 0.5483723062815222,
-      "grad_norm": 0.5316899032846605,
-      "learning_rate": 0.0004462587516734844,
-      "loss": 2.1805,
-      "step": 1196
-    },
-    {
-      "epoch": 0.5488308115543329,
-      "grad_norm": 0.5389273398895282,
-      "learning_rate": 0.00044552042037234596,
-      "loss": 2.3957,
-      "step": 1197
-    },
-    {
-      "epoch": 0.5492893168271435,
-      "grad_norm": 0.6453075289390467,
-      "learning_rate": 0.00044478220927347774,
-      "loss": 2.3385,
-      "step": 1198
-    },
-    {
-      "epoch": 0.5497478220999541,
-      "grad_norm": 0.485989371728771,
-      "learning_rate": 0.00044404412000564875,
-      "loss": 2.279,
-      "step": 1199
-    },
-    {
-      "epoch": 0.5502063273727648,
-      "grad_norm": 0.8764219605626095,
-      "learning_rate": 0.000443306154197359,
-      "loss": 2.2883,
-      "step": 1200
-    },
-    {
-      "epoch": 0.5506648326455754,
-      "grad_norm": 0.5225809157841347,
-      "learning_rate": 0.00044256831347683646,
-      "loss": 2.3673,
-      "step": 1201
-    },
-    {
-      "epoch": 0.5511233379183861,
-      "grad_norm": 0.5562592214067121,
-      "learning_rate": 0.0004418305994720328,
-      "loss": 2.2901,
-      "step": 1202
-    },
-    {
-      "epoch": 0.5515818431911967,
-      "grad_norm": 0.5489453115536549,
-      "learning_rate": 0.0004410930138106203,
-      "loss": 2.2618,
-      "step": 1203
-    },
-    {
-      "epoch": 0.5520403484640073,
-      "grad_norm": 0.516657615768289,
-      "learning_rate": 0.000440355558119988,
-      "loss": 2.3328,
-      "step": 1204
-    },
-    {
-      "epoch": 0.552498853736818,
-      "grad_norm": 0.535956361966292,
-      "learning_rate": 0.00043961823402723814,
-      "loss": 2.2679,
-      "step": 1205
-    },
-    {
-      "epoch": 0.5529573590096286,
-      "grad_norm": 0.5903729991722448,
-      "learning_rate": 0.0004388810431591829,
-      "loss": 2.2707,
-      "step": 1206
-    },
-    {
-      "epoch": 0.5534158642824393,
-      "grad_norm": 0.5469311770468273,
-      "learning_rate": 0.0004381439871423398,
-      "loss": 2.3573,
-      "step": 1207
-    },
-    {
-      "epoch": 0.5538743695552499,
-      "grad_norm": 0.538109440904654,
-      "learning_rate": 0.00043740706760292966,
-      "loss": 2.3242,
-      "step": 1208
-    },
-    {
-      "epoch": 0.5543328748280605,
-      "grad_norm": 0.5164135390645902,
-      "learning_rate": 0.0004366702861668716,
-      "loss": 2.3187,
-      "step": 1209
-    },
-    {
-      "epoch": 0.5547913801008711,
-      "grad_norm": 0.8088187993540111,
-      "learning_rate": 0.00043593364445978036,
-      "loss": 2.2986,
-      "step": 1210
-    },
-    {
-      "epoch": 0.5552498853736818,
-      "grad_norm": 0.49339963732347714,
-      "learning_rate": 0.0004351971441069622,
-      "loss": 2.3146,
-      "step": 1211
-    },
-    {
-      "epoch": 0.5557083906464925,
-      "grad_norm": 0.5183922310887656,
-      "learning_rate": 0.0004344607867334116,
-      "loss": 2.3419,
-      "step": 1212
-    },
-    {
-      "epoch": 0.556166895919303,
-      "grad_norm": 0.48980690715463915,
-      "learning_rate": 0.00043372457396380766,
-      "loss": 2.2932,
-      "step": 1213
-    },
-    {
-      "epoch": 0.5566254011921137,
-      "grad_norm": 0.5373894258309051,
-      "learning_rate": 0.00043298850742251013,
-      "loss": 2.2906,
-      "step": 1214
-    },
-    {
-      "epoch": 0.5570839064649243,
-      "grad_norm": 0.5139362147749471,
-      "learning_rate": 0.0004322525887335563,
-      "loss": 2.207,
-      "step": 1215
-    },
-    {
-      "epoch": 0.557542411737735,
-      "grad_norm": 0.509576047663656,
-      "learning_rate": 0.00043151681952065734,
-      "loss": 2.2694,
-      "step": 1216
-    },
-    {
-      "epoch": 0.5580009170105457,
-      "grad_norm": 0.5120161883308073,
-      "learning_rate": 0.00043078120140719456,
-      "loss": 2.3139,
-      "step": 1217
-    },
-    {
-      "epoch": 0.5584594222833562,
-      "grad_norm": 0.5892187900680396,
-      "learning_rate": 0.0004300457360162158,
-      "loss": 2.2295,
-      "step": 1218
-    },
-    {
-      "epoch": 0.5589179275561669,
-      "grad_norm": 0.5295940698398411,
-      "learning_rate": 0.0004293104249704319,
-      "loss": 2.2651,
-      "step": 1219
-    },
-    {
-      "epoch": 0.5593764328289775,
-      "grad_norm": 0.5349948378156093,
-      "learning_rate": 0.00042857526989221355,
-      "loss": 2.2568,
-      "step": 1220
-    },
-    {
-      "epoch": 0.5598349381017882,
-      "grad_norm": 0.5576797331691116,
-      "learning_rate": 0.00042784027240358674,
-      "loss": 2.2847,
-      "step": 1221
-    },
-    {
-      "epoch": 0.5602934433745989,
-      "grad_norm": 0.4984046100061123,
-      "learning_rate": 0.0004271054341262301,
-      "loss": 2.3079,
-      "step": 1222
-    },
-    {
-      "epoch": 0.5607519486474094,
-      "grad_norm": 0.4910297607137429,
-      "learning_rate": 0.000426370756681471,
-      "loss": 2.3065,
-      "step": 1223
-    },
-    {
-      "epoch": 0.5612104539202201,
-      "grad_norm": 0.5217612989904924,
-      "learning_rate": 0.0004256362416902817,
-      "loss": 2.2837,
-      "step": 1224
-    },
-    {
-      "epoch": 0.5616689591930307,
-      "grad_norm": 0.5162330946711731,
-      "learning_rate": 0.00042490189077327637,
-      "loss": 2.3074,
-      "step": 1225
-    },
-    {
-      "epoch": 0.5621274644658414,
-      "grad_norm": 0.6371822675738126,
-      "learning_rate": 0.00042416770555070703,
-      "loss": 2.3108,
-      "step": 1226
-    },
-    {
-      "epoch": 0.562585969738652,
-      "grad_norm": 0.5138406358468603,
-      "learning_rate": 0.00042343368764246,
-      "loss": 2.2799,
-      "step": 1227
-    },
-    {
-      "epoch": 0.5630444750114626,
-      "grad_norm": 0.5487218192597215,
-      "learning_rate": 0.0004226998386680524,
-      "loss": 2.3308,
-      "step": 1228
-    },
-    {
-      "epoch": 0.5635029802842733,
-      "grad_norm": 0.5432481477846867,
-      "learning_rate": 0.000421966160246629,
-      "loss": 2.3732,
-      "step": 1229
-    },
-    {
-      "epoch": 0.5639614855570839,
-      "grad_norm": 0.48140085045981723,
-      "learning_rate": 0.00042123265399695783,
-      "loss": 2.194,
-      "step": 1230
-    },
-    {
-      "epoch": 0.5644199908298946,
-      "grad_norm": 0.531135701933841,
-      "learning_rate": 0.0004204993215374273,
-      "loss": 2.2998,
-      "step": 1231
-    },
-    {
-      "epoch": 0.5648784961027051,
-      "grad_norm": 0.5720048683022793,
-      "learning_rate": 0.00041976616448604226,
-      "loss": 2.4751,
-      "step": 1232
-    },
-    {
-      "epoch": 0.5653370013755158,
-      "grad_norm": 0.5810436065402614,
-      "learning_rate": 0.00041903318446042076,
-      "loss": 2.2789,
-      "step": 1233
-    },
-    {
-      "epoch": 0.5657955066483265,
-      "grad_norm": 0.47586922144492344,
-      "learning_rate": 0.00041830038307778984,
-      "loss": 2.2795,
-      "step": 1234
-    },
-    {
-      "epoch": 0.5662540119211371,
-      "grad_norm": 0.4823347022084893,
-      "learning_rate": 0.0004175677619549828,
-      "loss": 2.2427,
-      "step": 1235
-    },
-    {
-      "epoch": 0.5667125171939478,
-      "grad_norm": 0.5011860888476942,
-      "learning_rate": 0.000416835322708435,
-      "loss": 2.2282,
-      "step": 1236
-    },
-    {
-      "epoch": 0.5671710224667583,
-      "grad_norm": 0.4813189110886366,
-      "learning_rate": 0.00041610306695418056,
-      "loss": 2.2746,
-      "step": 1237
-    },
-    {
-      "epoch": 0.567629527739569,
-      "grad_norm": 0.5250953456997385,
-      "learning_rate": 0.0004153709963078488,
-      "loss": 2.2829,
-      "step": 1238
-    },
-    {
-      "epoch": 0.5680880330123796,
-      "grad_norm": 0.5312998900320953,
-      "learning_rate": 0.0004146391123846606,
-      "loss": 2.4515,
-      "step": 1239
-    },
-    {
-      "epoch": 0.5685465382851903,
-      "grad_norm": 0.5145861542290342,
-      "learning_rate": 0.0004139074167994249,
-      "loss": 2.2809,
-      "step": 1240
-    },
-    {
-      "epoch": 0.569005043558001,
-      "grad_norm": 0.4730417614352528,
-      "learning_rate": 0.00041317591116653486,
-      "loss": 2.2825,
-      "step": 1241
-    },
-    {
-      "epoch": 0.5694635488308115,
-      "grad_norm": 0.4938007955965544,
-      "learning_rate": 0.0004124445970999648,
-      "loss": 2.2974,
-      "step": 1242
-    },
-    {
-      "epoch": 0.5699220541036222,
-      "grad_norm": 0.5133015466903679,
-      "learning_rate": 0.00041171347621326627,
-      "loss": 2.3237,
-      "step": 1243
-    },
-    {
-      "epoch": 0.5703805593764328,
-      "grad_norm": 0.49555181438363516,
-      "learning_rate": 0.00041098255011956465,
-      "loss": 2.2801,
-      "step": 1244
-    },
-    {
-      "epoch": 0.5708390646492435,
-      "grad_norm": 0.5059982135971404,
-      "learning_rate": 0.00041025182043155547,
-      "loss": 2.3738,
-      "step": 1245
-    },
-    {
-      "epoch": 0.571297569922054,
-      "grad_norm": 0.49192312307835284,
-      "learning_rate": 0.000409521288761501,
-      "loss": 2.2514,
-      "step": 1246
-    },
-    {
-      "epoch": 0.5717560751948647,
-      "grad_norm": 0.5187566122942779,
-      "learning_rate": 0.00040879095672122646,
-      "loss": 2.2183,
-      "step": 1247
-    },
-    {
-      "epoch": 0.5722145804676754,
-      "grad_norm": 0.5512322040689093,
-      "learning_rate": 0.0004080608259221167,
-      "loss": 2.3191,
-      "step": 1248
-    },
-    {
-      "epoch": 0.572673085740486,
-      "grad_norm": 0.5515292299718093,
-      "learning_rate": 0.0004073308979751126,
-      "loss": 2.3293,
-      "step": 1249
-    },
-    {
-      "epoch": 0.5731315910132967,
-      "grad_norm": 0.5468821974263677,
-      "learning_rate": 0.0004066011744907074,
-      "loss": 2.2967,
-      "step": 1250
-    },
-    {
-      "epoch": 0.5735900962861072,
-      "grad_norm": 0.5278176249250481,
-      "learning_rate": 0.00040587165707894326,
-      "loss": 2.4115,
-      "step": 1251
-    },
-    {
-      "epoch": 0.5740486015589179,
-      "grad_norm": 0.7662448429712779,
-      "learning_rate": 0.0004051423473494076,
-      "loss": 2.314,
-      "step": 1252
-    },
-    {
-      "epoch": 0.5745071068317286,
-      "grad_norm": 0.5327157965493358,
-      "learning_rate": 0.0004044132469112299,
-      "loss": 2.2734,
-      "step": 1253
-    },
-    {
-      "epoch": 0.5749656121045392,
-      "grad_norm": 0.5523654417107869,
-      "learning_rate": 0.00040368435737307733,
-      "loss": 2.3534,
-      "step": 1254
-    },
-    {
-      "epoch": 0.5754241173773499,
-      "grad_norm": 0.49869870044897324,
-      "learning_rate": 0.00040295568034315224,
-      "loss": 2.3399,
-      "step": 1255
-    },
-    {
-      "epoch": 0.5758826226501604,
-      "grad_norm": 0.48404554441572006,
-      "learning_rate": 0.0004022272174291878,
-      "loss": 2.3287,
-      "step": 1256
-    },
-    {
-      "epoch": 0.5763411279229711,
-      "grad_norm": 0.5146095645291623,
-      "learning_rate": 0.0004014989702384449,
-      "loss": 2.1945,
-      "step": 1257
-    },
-    {
-      "epoch": 0.5767996331957818,
-      "grad_norm": 0.49018961153037066,
-      "learning_rate": 0.00040077094037770843,
-      "loss": 2.282,
-      "step": 1258
-    },
-    {
-      "epoch": 0.5772581384685924,
-      "grad_norm": 0.5292928148303959,
-      "learning_rate": 0.0004000431294532838,
-      "loss": 2.3635,
-      "step": 1259
-    },
-    {
-      "epoch": 0.5777166437414031,
-      "grad_norm": 0.5194768715845305,
-      "learning_rate": 0.0003993155390709935,
-      "loss": 2.2896,
-      "step": 1260
-    },
-    {
-      "epoch": 0.5781751490142136,
-      "grad_norm": 0.49660566296231473,
-      "learning_rate": 0.0003985881708361729,
-      "loss": 2.189,
-      "step": 1261
-    },
-    {
-      "epoch": 0.5786336542870243,
-      "grad_norm": 0.5881681466437416,
-      "learning_rate": 0.00039786102635366784,
-      "loss": 2.3581,
-      "step": 1262
-    },
-    {
-      "epoch": 0.579092159559835,
-      "grad_norm": 0.5151898889988641,
-      "learning_rate": 0.0003971341072278302,
-      "loss": 2.2375,
-      "step": 1263
-    },
-    {
-      "epoch": 0.5795506648326456,
-      "grad_norm": 0.5944310510752396,
-      "learning_rate": 0.00039640741506251457,
-      "loss": 2.1782,
-      "step": 1264
-    },
-    {
-      "epoch": 0.5800091701054562,
-      "grad_norm": 0.5316210141058031,
-      "learning_rate": 0.00039568095146107495,
-      "loss": 2.4185,
-      "step": 1265
-    },
-    {
-      "epoch": 0.5804676753782668,
-      "grad_norm": 0.5346957931071892,
-      "learning_rate": 0.00039495471802636096,
-      "loss": 2.275,
-      "step": 1266
-    },
-    {
-      "epoch": 0.5809261806510775,
-      "grad_norm": 0.5287316231966706,
-      "learning_rate": 0.0003942287163607145,
-      "loss": 2.2532,
-      "step": 1267
-    },
-    {
-      "epoch": 0.5813846859238881,
-      "grad_norm": 0.5148937904520897,
-      "learning_rate": 0.0003935029480659658,
-      "loss": 2.2817,
-      "step": 1268
-    },
-    {
-      "epoch": 0.5818431911966988,
-      "grad_norm": 0.5260948403712259,
-      "learning_rate": 0.00039277741474343054,
-      "loss": 2.3145,
-      "step": 1269
-    },
-    {
-      "epoch": 0.5823016964695094,
-      "grad_norm": 0.5332861951772798,
-      "learning_rate": 0.0003920521179939057,
-      "loss": 2.3143,
-      "step": 1270
-    },
-    {
-      "epoch": 0.58276020174232,
-      "grad_norm": 0.5139526952126439,
-      "learning_rate": 0.00039132705941766644,
-      "loss": 2.4208,
-      "step": 1271
-    },
-    {
-      "epoch": 0.5832187070151307,
-      "grad_norm": 0.5270669780382023,
-      "learning_rate": 0.0003906022406144624,
-      "loss": 2.2527,
-      "step": 1272
-    },
-    {
-      "epoch": 0.5836772122879413,
-      "grad_norm": 0.485800046944185,
-      "learning_rate": 0.0003898776631835143,
-      "loss": 2.309,
-      "step": 1273
-    },
-    {
-      "epoch": 0.584135717560752,
-      "grad_norm": 0.5164569346262999,
-      "learning_rate": 0.00038915332872350994,
-      "loss": 2.2395,
-      "step": 1274
-    },
-    {
-      "epoch": 0.5845942228335625,
-      "grad_norm": 0.4866183966556747,
-      "learning_rate": 0.00038842923883260135,
-      "loss": 2.2769,
-      "step": 1275
-    },
-    {
-      "epoch": 0.5850527281063732,
-      "grad_norm": 0.5915647248482845,
-      "learning_rate": 0.00038770539510840093,
-      "loss": 2.2828,
-      "step": 1276
-    },
-    {
-      "epoch": 0.5855112333791839,
-      "grad_norm": 0.4863048221374928,
-      "learning_rate": 0.00038698179914797783,
-      "loss": 2.3219,
-      "step": 1277
-    },
-    {
-      "epoch": 0.5859697386519945,
-      "grad_norm": 0.5506459170913212,
-      "learning_rate": 0.0003862584525478545,
-      "loss": 2.2827,
-      "step": 1278
-    },
-    {
-      "epoch": 0.5864282439248051,
-      "grad_norm": 0.5278023701308207,
-      "learning_rate": 0.00038553535690400353,
-      "loss": 2.2613,
-      "step": 1279
-    },
-    {
-      "epoch": 0.5868867491976157,
-      "grad_norm": 0.5430932436130381,
-      "learning_rate": 0.00038481251381184355,
-      "loss": 2.2573,
-      "step": 1280
-    },
-    {
-      "epoch": 0.5873452544704264,
-      "grad_norm": 0.5069319614780216,
-      "learning_rate": 0.00038408992486623584,
-      "loss": 2.3071,
-      "step": 1281
-    },
-    {
-      "epoch": 0.5878037597432371,
-      "grad_norm": 0.5247421013400542,
-      "learning_rate": 0.00038336759166148117,
-      "loss": 2.3383,
-      "step": 1282
-    },
-    {
-      "epoch": 0.5882622650160477,
-      "grad_norm": 0.5225701957616619,
-      "learning_rate": 0.0003826455157913159,
-      "loss": 2.2752,
-      "step": 1283
-    },
-    {
-      "epoch": 0.5887207702888583,
-      "grad_norm": 0.4974888653511051,
-      "learning_rate": 0.00038192369884890886,
-      "loss": 2.265,
-      "step": 1284
-    },
-    {
-      "epoch": 0.5891792755616689,
-      "grad_norm": 0.5841060620489029,
-      "learning_rate": 0.00038120214242685723,
-      "loss": 2.2607,
-      "step": 1285
-    },
-    {
-      "epoch": 0.5896377808344796,
-      "grad_norm": 0.47453858135529997,
-      "learning_rate": 0.00038048084811718373,
-      "loss": 2.3048,
-      "step": 1286
-    },
-    {
-      "epoch": 0.5900962861072903,
-      "grad_norm": 0.49179841478582426,
-      "learning_rate": 0.0003797598175113327,
-      "loss": 2.3454,
-      "step": 1287
-    },
-    {
-      "epoch": 0.5905547913801009,
-      "grad_norm": 0.5010584309814975,
-      "learning_rate": 0.0003790390522001662,
-      "loss": 2.3166,
-      "step": 1288
-    },
-    {
-      "epoch": 0.5910132966529115,
-      "grad_norm": 0.5040574210901609,
-      "learning_rate": 0.0003783185537739615,
-      "loss": 2.332,
-      "step": 1289
-    },
-    {
-      "epoch": 0.5914718019257221,
-      "grad_norm": 0.5487449948302137,
-      "learning_rate": 0.00037759832382240697,
-      "loss": 2.2806,
-      "step": 1290
-    },
-    {
-      "epoch": 0.5919303071985328,
-      "grad_norm": 0.5025437255955114,
-      "learning_rate": 0.00037687836393459826,
-      "loss": 2.2625,
-      "step": 1291
-    },
-    {
-      "epoch": 0.5923888124713435,
-      "grad_norm": 0.5439310928074973,
-      "learning_rate": 0.0003761586756990354,
-      "loss": 2.3813,
-      "step": 1292
-    },
-    {
-      "epoch": 0.5928473177441541,
-      "grad_norm": 0.48207059659842094,
-      "learning_rate": 0.0003754392607036191,
-      "loss": 2.2744,
-      "step": 1293
-    },
-    {
-      "epoch": 0.5933058230169647,
-      "grad_norm": 0.5228307155190857,
-      "learning_rate": 0.0003747201205356472,
-      "loss": 2.3705,
-      "step": 1294
-    },
-    {
-      "epoch": 0.5937643282897753,
-      "grad_norm": 0.5156803792236302,
-      "learning_rate": 0.0003740012567818111,
-      "loss": 2.2727,
-      "step": 1295
-    },
-    {
-      "epoch": 0.594222833562586,
-      "grad_norm": 0.49295780689083235,
-      "learning_rate": 0.0003732826710281922,
-      "loss": 2.323,
-      "step": 1296
-    },
-    {
-      "epoch": 0.5946813388353966,
-      "grad_norm": 0.5209183849488812,
-      "learning_rate": 0.0003725643648602588,
-      "loss": 2.2329,
-      "step": 1297
-    },
-    {
-      "epoch": 0.5951398441082072,
-      "grad_norm": 0.5345100128139496,
-      "learning_rate": 0.0003718463398628621,
-      "loss": 2.2795,
-      "step": 1298
-    },
-    {
-      "epoch": 0.5955983493810179,
-      "grad_norm": 0.4817790795385162,
-      "learning_rate": 0.0003711285976202331,
-      "loss": 2.3003,
-      "step": 1299
-    },
-    {
-      "epoch": 0.5960568546538285,
-      "grad_norm": 0.5091282838075971,
-      "learning_rate": 0.0003704111397159787,
-      "loss": 2.2519,
-      "step": 1300
-    },
-    {
-      "epoch": 0.5965153599266392,
-      "grad_norm": 0.475524495706934,
-      "learning_rate": 0.0003696939677330788,
-      "loss": 2.2573,
-      "step": 1301
-    },
-    {
-      "epoch": 0.5969738651994498,
-      "grad_norm": 0.524200962677336,
-      "learning_rate": 0.00036897708325388213,
-      "loss": 2.3149,
-      "step": 1302
-    },
-    {
-      "epoch": 0.5974323704722604,
-      "grad_norm": 0.5311818508656727,
-      "learning_rate": 0.0003682604878601034,
-      "loss": 2.2397,
-      "step": 1303
-    },
-    {
-      "epoch": 0.597890875745071,
-      "grad_norm": 0.5027183234398438,
-      "learning_rate": 0.000367544183132819,
-      "loss": 2.2316,
-      "step": 1304
-    },
-    {
-      "epoch": 0.5983493810178817,
-      "grad_norm": 0.5388755605934765,
-      "learning_rate": 0.0003668281706524645,
-      "loss": 2.2731,
-      "step": 1305
-    },
-    {
-      "epoch": 0.5988078862906924,
-      "grad_norm": 0.516660060977736,
-      "learning_rate": 0.0003661124519988304,
-      "loss": 2.2715,
-      "step": 1306
-    },
-    {
-      "epoch": 0.599266391563503,
-      "grad_norm": 0.5148722240789517,
-      "learning_rate": 0.00036539702875105893,
-      "loss": 2.3057,
-      "step": 1307
-    },
-    {
-      "epoch": 0.5997248968363136,
-      "grad_norm": 0.4883904443477065,
-      "learning_rate": 0.0003646819024876406,
-      "loss": 2.2801,
-      "step": 1308
-    },
-    {
-      "epoch": 0.6001834021091242,
-      "grad_norm": 0.5074150047358399,
-      "learning_rate": 0.0003639670747864107,
-      "loss": 2.2652,
-      "step": 1309
-    },
-    {
-      "epoch": 0.6006419073819349,
-      "grad_norm": 0.5144485985028897,
-      "learning_rate": 0.00036325254722454584,
-      "loss": 2.2704,
-      "step": 1310
-    },
-    {
-      "epoch": 0.6011004126547456,
-      "grad_norm": 0.5049179876994025,
-      "learning_rate": 0.00036253832137855997,
-      "loss": 2.2252,
-      "step": 1311
-    },
-    {
-      "epoch": 0.6015589179275561,
-      "grad_norm": 0.512991183489867,
-      "learning_rate": 0.00036182439882430183,
-      "loss": 2.3121,
-      "step": 1312
-    },
-    {
-      "epoch": 0.6020174232003668,
-      "grad_norm": 0.5267320966649242,
-      "learning_rate": 0.00036111078113695096,
-      "loss": 2.325,
-      "step": 1313
-    },
-    {
-      "epoch": 0.6024759284731774,
-      "grad_norm": 0.49756608679878483,
-      "learning_rate": 0.0003603974698910139,
-      "loss": 2.2957,
-      "step": 1314
-    },
-    {
-      "epoch": 0.6029344337459881,
-      "grad_norm": 0.48039177449043563,
-      "learning_rate": 0.0003596844666603214,
-      "loss": 2.2723,
-      "step": 1315
-    },
-    {
-      "epoch": 0.6033929390187988,
-      "grad_norm": 0.5009053650953867,
-      "learning_rate": 0.0003589717730180245,
-      "loss": 2.2922,
-      "step": 1316
-    },
-    {
-      "epoch": 0.6038514442916093,
-      "grad_norm": 0.5134833895692668,
-      "learning_rate": 0.00035825939053659117,
-      "loss": 2.284,
-      "step": 1317
-    },
-    {
-      "epoch": 0.60430994956442,
-      "grad_norm": 0.5292067730408965,
-      "learning_rate": 0.00035754732078780273,
-      "loss": 2.3117,
-      "step": 1318
-    },
-    {
-      "epoch": 0.6047684548372306,
-      "grad_norm": 0.5126018450324754,
-      "learning_rate": 0.00035683556534275076,
-      "loss": 2.2622,
-      "step": 1319
-    },
-    {
-      "epoch": 0.6052269601100413,
-      "grad_norm": 0.5012979360843415,
-      "learning_rate": 0.00035612412577183303,
-      "loss": 2.2201,
-      "step": 1320
-    },
-    {
-      "epoch": 0.605685465382852,
-      "grad_norm": 0.4962456887746889,
-      "learning_rate": 0.00035541300364475063,
-      "loss": 2.3048,
-      "step": 1321
-    },
-    {
-      "epoch": 0.6061439706556625,
-      "grad_norm": 0.8863674111872646,
-      "learning_rate": 0.0003547022005305043,
-      "loss": 2.2243,
-      "step": 1322
-    },
-    {
-      "epoch": 0.6066024759284732,
-      "grad_norm": 0.5289618361668265,
-      "learning_rate": 0.0003539917179973907,
-      "loss": 2.249,
-      "step": 1323
-    },
-    {
-      "epoch": 0.6070609812012838,
-      "grad_norm": 0.517862038296709,
-      "learning_rate": 0.00035328155761299917,
-      "loss": 2.3238,
-      "step": 1324
-    },
-    {
-      "epoch": 0.6075194864740945,
-      "grad_norm": 0.517042271201594,
-      "learning_rate": 0.0003525717209442085,
-      "loss": 2.3241,
-      "step": 1325
-    },
-    {
-      "epoch": 0.6079779917469051,
-      "grad_norm": 0.4923804142548179,
-      "learning_rate": 0.00035186220955718306,
-      "loss": 2.2955,
-      "step": 1326
-    },
-    {
-      "epoch": 0.6084364970197157,
-      "grad_norm": 0.6245138663776206,
-      "learning_rate": 0.0003511530250173696,
-      "loss": 2.2964,
-      "step": 1327
-    },
-    {
-      "epoch": 0.6088950022925264,
-      "grad_norm": 0.5118300271664651,
-      "learning_rate": 0.00035044416888949364,
-      "loss": 2.3173,
-      "step": 1328
-    },
-    {
-      "epoch": 0.609353507565337,
-      "grad_norm": 0.5653076075086632,
-      "learning_rate": 0.0003497356427375562,
-      "loss": 2.3516,
-      "step": 1329
-    },
-    {
-      "epoch": 0.6098120128381477,
-      "grad_norm": 0.509245174855858,
-      "learning_rate": 0.00034902744812483034,
-      "loss": 2.4165,
-      "step": 1330
-    },
-    {
-      "epoch": 0.6102705181109582,
-      "grad_norm": 0.529059755298975,
-      "learning_rate": 0.00034831958661385714,
-      "loss": 2.379,
-      "step": 1331
-    },
-    {
-      "epoch": 0.6107290233837689,
-      "grad_norm": 0.48496748934328665,
-      "learning_rate": 0.0003476120597664434,
-      "loss": 2.2644,
-      "step": 1332
-    },
-    {
-      "epoch": 0.6111875286565795,
-      "grad_norm": 0.5217588466740565,
-      "learning_rate": 0.00034690486914365704,
-      "loss": 2.3122,
-      "step": 1333
-    },
-    {
-      "epoch": 0.6116460339293902,
-      "grad_norm": 0.559715292367092,
-      "learning_rate": 0.00034619801630582435,
-      "loss": 2.2741,
-      "step": 1334
-    },
-    {
-      "epoch": 0.6121045392022009,
-      "grad_norm": 0.5535583998719366,
-      "learning_rate": 0.00034549150281252633,
-      "loss": 2.3828,
-      "step": 1335
-    },
-    {
-      "epoch": 0.6125630444750114,
-      "grad_norm": 0.5299653135905453,
-      "learning_rate": 0.0003447853302225952,
-      "loss": 2.3387,
-      "step": 1336
-    },
-    {
-      "epoch": 0.6130215497478221,
-      "grad_norm": 0.51657716232863,
-      "learning_rate": 0.00034407950009411126,
-      "loss": 2.3809,
-      "step": 1337
-    },
-    {
-      "epoch": 0.6134800550206327,
-      "grad_norm": 0.4816142037696413,
-      "learning_rate": 0.00034337401398439873,
-      "loss": 2.2749,
-      "step": 1338
-    },
-    {
-      "epoch": 0.6139385602934434,
-      "grad_norm": 0.5065257360289724,
-      "learning_rate": 0.00034266887345002305,
-      "loss": 2.3761,
-      "step": 1339
-    },
-    {
-      "epoch": 0.6143970655662541,
-      "grad_norm": 0.5532320205707356,
-      "learning_rate": 0.0003419640800467874,
-      "loss": 2.3053,
-      "step": 1340
-    },
-    {
-      "epoch": 0.6148555708390646,
-      "grad_norm": 0.4935485075517298,
-      "learning_rate": 0.0003412596353297288,
-      "loss": 2.3551,
-      "step": 1341
-    },
-    {
-      "epoch": 0.6153140761118753,
-      "grad_norm": 0.5285287413305256,
-      "learning_rate": 0.00034055554085311493,
-      "loss": 2.3847,
-      "step": 1342
-    },
-    {
-      "epoch": 0.6157725813846859,
-      "grad_norm": 0.5207109213811997,
-      "learning_rate": 0.00033985179817044105,
-      "loss": 2.2345,
-      "step": 1343
-    },
-    {
-      "epoch": 0.6162310866574966,
-      "grad_norm": 0.5669655724549134,
-      "learning_rate": 0.0003391484088344257,
-      "loss": 2.3355,
-      "step": 1344
-    },
-    {
-      "epoch": 0.6166895919303071,
-      "grad_norm": 0.5273054089118442,
-      "learning_rate": 0.00033844537439700807,
-      "loss": 2.2001,
-      "step": 1345
-    },
-    {
-      "epoch": 0.6171480972031178,
-      "grad_norm": 0.49257130337369703,
-      "learning_rate": 0.00033774269640934445,
-      "loss": 2.2967,
-      "step": 1346
-    },
-    {
-      "epoch": 0.6176066024759285,
-      "grad_norm": 0.5074114988310836,
-      "learning_rate": 0.0003370403764218045,
-      "loss": 2.2798,
-      "step": 1347
-    },
-    {
-      "epoch": 0.6180651077487391,
-      "grad_norm": 0.4971473617503481,
-      "learning_rate": 0.000336338415983968,
-      "loss": 2.2981,
-      "step": 1348
-    },
-    {
-      "epoch": 0.6185236130215498,
-      "grad_norm": 0.49820339744614855,
-      "learning_rate": 0.00033563681664462155,
-      "loss": 2.312,
-      "step": 1349
-    },
-    {
-      "epoch": 0.6189821182943603,
-      "grad_norm": 0.49968374740234944,
-      "learning_rate": 0.000334935579951755,
-      "loss": 2.3603,
-      "step": 1350
-    },
-    {
-      "epoch": 0.619440623567171,
-      "grad_norm": 0.4825445963341056,
-      "learning_rate": 0.0003342347074525578,
-      "loss": 2.2315,
-      "step": 1351
-    },
-    {
-      "epoch": 0.6198991288399817,
-      "grad_norm": 0.4740107477394949,
-      "learning_rate": 0.0003335342006934161,
-      "loss": 2.2649,
-      "step": 1352
-    },
-    {
-      "epoch": 0.6203576341127923,
-      "grad_norm": 0.5200218688661572,
-      "learning_rate": 0.00033283406121990914,
-      "loss": 2.3321,
-      "step": 1353
-    },
-    {
-      "epoch": 0.620816139385603,
-      "grad_norm": 0.49192777992360437,
-      "learning_rate": 0.0003321342905768057,
-      "loss": 2.2444,
-      "step": 1354
-    },
-    {
-      "epoch": 0.6212746446584135,
-      "grad_norm": 0.4949678459369133,
-      "learning_rate": 0.00033143489030806086,
-      "loss": 2.255,
-      "step": 1355
-    },
-    {
-      "epoch": 0.6217331499312242,
-      "grad_norm": 0.4825017762522047,
-      "learning_rate": 0.00033073586195681227,
-      "loss": 2.2615,
-      "step": 1356
-    },
-    {
-      "epoch": 0.6221916552040349,
-      "grad_norm": 0.5092521715663748,
-      "learning_rate": 0.00033003720706537736,
-      "loss": 2.2764,
-      "step": 1357
-    },
-    {
-      "epoch": 0.6226501604768455,
-      "grad_norm": 0.5075684905682752,
-      "learning_rate": 0.0003293389271752492,
-      "loss": 2.1358,
-      "step": 1358
-    },
-    {
-      "epoch": 0.6231086657496562,
-      "grad_norm": 0.5232618412140702,
-      "learning_rate": 0.00032864102382709374,
-      "loss": 2.2901,
-      "step": 1359
-    },
-    {
-      "epoch": 0.6235671710224667,
-      "grad_norm": 0.5494674621153885,
-      "learning_rate": 0.000327943498560746,
-      "loss": 2.2745,
-      "step": 1360
-    },
-    {
-      "epoch": 0.6240256762952774,
-      "grad_norm": 0.5096643333996131,
-      "learning_rate": 0.00032724635291520694,
-      "loss": 2.219,
-      "step": 1361
-    },
-    {
-      "epoch": 0.624484181568088,
-      "grad_norm": 0.5289575531210651,
-      "learning_rate": 0.00032654958842863967,
-      "loss": 2.2886,
-      "step": 1362
-    },
-    {
-      "epoch": 0.6249426868408987,
-      "grad_norm": 0.509013004725559,
-      "learning_rate": 0.0003258532066383667,
-      "loss": 2.2099,
-      "step": 1363
-    },
-    {
-      "epoch": 0.6254011921137093,
-      "grad_norm": 0.5180562042612443,
-      "learning_rate": 0.000325157209080866,
-      "loss": 2.316,
-      "step": 1364
-    },
-    {
-      "epoch": 0.6258596973865199,
-      "grad_norm": 0.4958814834237173,
-      "learning_rate": 0.00032446159729176743,
-      "loss": 2.3195,
-      "step": 1365
-    },
-    {
-      "epoch": 0.6263182026593306,
-      "grad_norm": 0.5523266634446398,
-      "learning_rate": 0.0003237663728058502,
-      "loss": 2.2742,
-      "step": 1366
-    },
-    {
-      "epoch": 0.6267767079321412,
-      "grad_norm": 0.48416649316488397,
-      "learning_rate": 0.0003230715371570389,
-      "loss": 2.227,
-      "step": 1367
-    },
-    {
-      "epoch": 0.6272352132049519,
-      "grad_norm": 0.5103367658350448,
-      "learning_rate": 0.00032237709187839996,
-      "loss": 2.2286,
-      "step": 1368
-    },
-    {
-      "epoch": 0.6276937184777625,
-      "grad_norm": 0.48405900624940207,
-      "learning_rate": 0.0003216830385021388,
-      "loss": 2.378,
-      "step": 1369
-    },
-    {
-      "epoch": 0.6281522237505731,
-      "grad_norm": 0.5722334156150723,
-      "learning_rate": 0.0003209893785595959,
-      "loss": 2.277,
-      "step": 1370
-    },
-    {
-      "epoch": 0.6286107290233838,
-      "grad_norm": 0.5367255162427883,
-      "learning_rate": 0.00032029611358124366,
-      "loss": 2.3556,
-      "step": 1371
-    },
-    {
-      "epoch": 0.6290692342961944,
-      "grad_norm": 0.4802678953656144,
-      "learning_rate": 0.00031960324509668336,
-      "loss": 2.2731,
-      "step": 1372
-    },
-    {
-      "epoch": 0.6295277395690051,
-      "grad_norm": 0.5183454866532284,
-      "learning_rate": 0.0003189107746346412,
-      "loss": 2.2827,
-      "step": 1373
-    },
-    {
-      "epoch": 0.6299862448418156,
-      "grad_norm": 0.5260334675461894,
-      "learning_rate": 0.0003182187037229653,
-      "loss": 2.2858,
-      "step": 1374
-    },
-    {
-      "epoch": 0.6304447501146263,
-      "grad_norm": 0.5473090569472365,
-      "learning_rate": 0.0003175270338886221,
-      "loss": 2.3333,
-      "step": 1375
-    },
-    {
-      "epoch": 0.630903255387437,
-      "grad_norm": 0.5206769316009642,
-      "learning_rate": 0.00031683576665769345,
-      "loss": 2.2918,
-      "step": 1376
-    },
-    {
-      "epoch": 0.6313617606602476,
-      "grad_norm": 0.5354725808173415,
-      "learning_rate": 0.0003161449035553724,
-      "loss": 2.3652,
-      "step": 1377
-    },
-    {
-      "epoch": 0.6318202659330582,
-      "grad_norm": 0.49704998467554784,
-      "learning_rate": 0.00031545444610596077,
-      "loss": 2.3192,
-      "step": 1378
-    },
-    {
-      "epoch": 0.6322787712058688,
-      "grad_norm": 0.5179105795316825,
-      "learning_rate": 0.000314764395832865,
-      "loss": 2.3029,
-      "step": 1379
-    },
-    {
-      "epoch": 0.6327372764786795,
-      "grad_norm": 0.6213060653794928,
-      "learning_rate": 0.0003140747542585934,
-      "loss": 2.2083,
-      "step": 1380
-    },
-    {
-      "epoch": 0.6331957817514902,
-      "grad_norm": 0.5215848846407494,
-      "learning_rate": 0.00031338552290475266,
-      "loss": 2.2675,
-      "step": 1381
-    },
-    {
-      "epoch": 0.6336542870243008,
-      "grad_norm": 0.5112743834750455,
-      "learning_rate": 0.00031269670329204396,
-      "loss": 2.2609,
-      "step": 1382
-    },
-    {
-      "epoch": 0.6341127922971114,
-      "grad_norm": 0.5351977464935733,
-      "learning_rate": 0.0003120082969402604,
-      "loss": 2.2449,
-      "step": 1383
-    },
-    {
-      "epoch": 0.634571297569922,
-      "grad_norm": 0.5228880511235061,
-      "learning_rate": 0.00031132030536828314,
-      "loss": 2.3792,
-      "step": 1384
-    },
-    {
-      "epoch": 0.6350298028427327,
-      "grad_norm": 0.4992996615894805,
-      "learning_rate": 0.00031063273009407805,
-      "loss": 2.2107,
-      "step": 1385
-    },
-    {
-      "epoch": 0.6354883081155434,
-      "grad_norm": 0.5077301747274104,
-      "learning_rate": 0.00030994557263469265,
-      "loss": 2.3865,
-      "step": 1386
-    },
-    {
-      "epoch": 0.635946813388354,
-      "grad_norm": 0.5141957203700439,
-      "learning_rate": 0.0003092588345062526,
-      "loss": 2.3017,
-      "step": 1387
-    },
-    {
-      "epoch": 0.6364053186611646,
-      "grad_norm": 0.5007500259470499,
-      "learning_rate": 0.0003085725172239582,
-      "loss": 2.263,
-      "step": 1388
-    },
-    {
-      "epoch": 0.6368638239339752,
-      "grad_norm": 0.5218813935048109,
-      "learning_rate": 0.0003078866223020815,
-      "loss": 2.3438,
-      "step": 1389
-    },
-    {
-      "epoch": 0.6373223292067859,
-      "grad_norm": 0.5016180670749802,
-      "learning_rate": 0.0003072011512539624,
-      "loss": 2.2819,
-      "step": 1390
-    },
-    {
-      "epoch": 0.6377808344795965,
-      "grad_norm": 0.526039927703199,
-      "learning_rate": 0.00030651610559200574,
-      "loss": 2.2112,
-      "step": 1391
-    },
-    {
-      "epoch": 0.6382393397524071,
-      "grad_norm": 0.5495002349980955,
-      "learning_rate": 0.00030583148682767757,
-      "loss": 2.2907,
-      "step": 1392
-    },
-    {
-      "epoch": 0.6386978450252178,
-      "grad_norm": 0.6034309538453339,
-      "learning_rate": 0.00030514729647150243,
-      "loss": 2.3004,
-      "step": 1393
-    },
-    {
-      "epoch": 0.6391563502980284,
-      "grad_norm": 0.47659678804838623,
-      "learning_rate": 0.0003044635360330592,
-      "loss": 2.2911,
-      "step": 1394
-    },
-    {
-      "epoch": 0.6396148555708391,
-      "grad_norm": 0.5201583784397313,
-      "learning_rate": 0.00030378020702097845,
-      "loss": 2.3236,
-      "step": 1395
-    },
-    {
-      "epoch": 0.6400733608436497,
-      "grad_norm": 0.5075178093667224,
-      "learning_rate": 0.000303097310942939,
-      "loss": 2.1476,
-      "step": 1396
-    },
-    {
-      "epoch": 0.6405318661164603,
-      "grad_norm": 0.5213480127743028,
-      "learning_rate": 0.0003024148493056641,
-      "loss": 2.3057,
-      "step": 1397
-    },
-    {
-      "epoch": 0.640990371389271,
-      "grad_norm": 0.5049739579904373,
-      "learning_rate": 0.00030173282361491865,
-      "loss": 2.1986,
-      "step": 1398
-    },
-    {
-      "epoch": 0.6414488766620816,
-      "grad_norm": 0.49726681183802973,
-      "learning_rate": 0.0003010512353755057,
-      "loss": 2.2876,
-      "step": 1399
-    },
-    {
-      "epoch": 0.6419073819348923,
-      "grad_norm": 0.524522784753267,
-      "learning_rate": 0.00030037008609126313,
-      "loss": 2.2916,
-      "step": 1400
-    },
-    {
-      "epoch": 0.6423658872077029,
-      "grad_norm": 0.49792744663189464,
-      "learning_rate": 0.0002996893772650602,
-      "loss": 2.3609,
-      "step": 1401
-    },
-    {
-      "epoch": 0.6428243924805135,
-      "grad_norm": 0.49913883507966356,
-      "learning_rate": 0.0002990091103987945,
-      "loss": 2.2465,
-      "step": 1402
-    },
-    {
-      "epoch": 0.6432828977533241,
-      "grad_norm": 0.5107442683018881,
-      "learning_rate": 0.0002983292869933886,
-      "loss": 2.3294,
-      "step": 1403
-    },
-    {
-      "epoch": 0.6437414030261348,
-      "grad_norm": 0.49925674109601015,
-      "learning_rate": 0.0002976499085487862,
-      "loss": 2.3014,
-      "step": 1404
-    },
-    {
-      "epoch": 0.6441999082989455,
-      "grad_norm": 0.5372439302615921,
-      "learning_rate": 0.00029697097656394963,
-      "loss": 2.3133,
-      "step": 1405
-    },
-    {
-      "epoch": 0.6446584135717561,
-      "grad_norm": 0.5364605951705181,
-      "learning_rate": 0.00029629249253685595,
-      "loss": 2.3385,
-      "step": 1406
-    },
-    {
-      "epoch": 0.6451169188445667,
-      "grad_norm": 0.5035565670653194,
-      "learning_rate": 0.00029561445796449416,
-      "loss": 2.2862,
-      "step": 1407
-    },
-    {
-      "epoch": 0.6455754241173773,
-      "grad_norm": 0.5158063804788108,
-      "learning_rate": 0.0002949368743428612,
-      "loss": 2.272,
-      "step": 1408
-    },
-    {
-      "epoch": 0.646033929390188,
-      "grad_norm": 0.5005564308288862,
-      "learning_rate": 0.0002942597431669593,
-      "loss": 2.3271,
-      "step": 1409
-    },
-    {
-      "epoch": 0.6464924346629987,
-      "grad_norm": 0.49724471273185133,
-      "learning_rate": 0.0002935830659307924,
-      "loss": 2.3287,
-      "step": 1410
-    },
-    {
-      "epoch": 0.6469509399358092,
-      "grad_norm": 0.49277502049098476,
-      "learning_rate": 0.0002929068441273629,
-      "loss": 2.2648,
-      "step": 1411
-    },
-    {
-      "epoch": 0.6474094452086199,
-      "grad_norm": 0.4834026170318713,
-      "learning_rate": 0.0002922310792486681,
-      "loss": 2.2764,
-      "step": 1412
-    },
-    {
-      "epoch": 0.6478679504814305,
-      "grad_norm": 0.49203048637343144,
-      "learning_rate": 0.00029155577278569745,
-      "loss": 2.2619,
-      "step": 1413
-    },
-    {
-      "epoch": 0.6483264557542412,
-      "grad_norm": 0.5119134174595378,
-      "learning_rate": 0.00029088092622842895,
-      "loss": 2.2887,
-      "step": 1414
-    },
-    {
-      "epoch": 0.6487849610270519,
-      "grad_norm": 0.5268702014422324,
-      "learning_rate": 0.00029020654106582544,
-      "loss": 2.2328,
-      "step": 1415
-    },
-    {
-      "epoch": 0.6492434662998624,
-      "grad_norm": 0.5154608766483364,
-      "learning_rate": 0.0002895326187858326,
-      "loss": 2.3357,
-      "step": 1416
-    },
-    {
-      "epoch": 0.6497019715726731,
-      "grad_norm": 0.47895909093909694,
-      "learning_rate": 0.00028885916087537377,
-      "loss": 2.2594,
-      "step": 1417
-    },
-    {
-      "epoch": 0.6501604768454837,
-      "grad_norm": 0.48106728805211885,
-      "learning_rate": 0.00028818616882034877,
-      "loss": 2.3817,
-      "step": 1418
-    },
-    {
-      "epoch": 0.6506189821182944,
-      "grad_norm": 0.5382890400198277,
-      "learning_rate": 0.0002875136441056286,
-      "loss": 2.1685,
-      "step": 1419
-    },
-    {
-      "epoch": 0.651077487391105,
-      "grad_norm": 0.49579661540200487,
-      "learning_rate": 0.000286841588215054,
-      "loss": 2.2542,
-      "step": 1420
-    },
-    {
-      "epoch": 0.6515359926639156,
-      "grad_norm": 0.5014482389094856,
-      "learning_rate": 0.0002861700026314308,
-      "loss": 2.3294,
-      "step": 1421
-    },
-    {
-      "epoch": 0.6519944979367263,
-      "grad_norm": 0.48547989172446276,
-      "learning_rate": 0.00028549888883652686,
-      "loss": 2.2486,
-      "step": 1422
-    },
-    {
-      "epoch": 0.6524530032095369,
-      "grad_norm": 0.4985687048885817,
-      "learning_rate": 0.00028482824831107,
-      "loss": 2.2004,
-      "step": 1423
-    },
-    {
-      "epoch": 0.6529115084823476,
-      "grad_norm": 0.4898445960002192,
-      "learning_rate": 0.000284158082534743,
-      "loss": 2.2715,
-      "step": 1424
-    },
-    {
-      "epoch": 0.6533700137551581,
-      "grad_norm": 0.5079687192310951,
-      "learning_rate": 0.00028348839298618177,
-      "loss": 2.2884,
-      "step": 1425
-    },
-    {
-      "epoch": 0.6538285190279688,
-      "grad_norm": 0.4995091585986773,
-      "learning_rate": 0.0002828191811429709,
-      "loss": 2.2797,
-      "step": 1426
-    },
-    {
-      "epoch": 0.6542870243007795,
-      "grad_norm": 0.5283098942545613,
-      "learning_rate": 0.00028215044848164164,
-      "loss": 2.3156,
-      "step": 1427
-    },
-    {
-      "epoch": 0.6547455295735901,
-      "grad_norm": 0.5084085123574805,
-      "learning_rate": 0.00028148219647766747,
-      "loss": 2.2179,
-      "step": 1428
-    },
-    {
-      "epoch": 0.6552040348464008,
-      "grad_norm": 0.5201432768126203,
-      "learning_rate": 0.00028081442660546124,
-      "loss": 2.2712,
-      "step": 1429
-    },
-    {
-      "epoch": 0.6556625401192113,
-      "grad_norm": 0.5017990322701781,
-      "learning_rate": 0.0002801471403383728,
-      "loss": 2.2513,
-      "step": 1430
-    },
-    {
-      "epoch": 0.656121045392022,
-      "grad_norm": 0.4933798039534146,
-      "learning_rate": 0.00027948033914868415,
-      "loss": 2.2542,
-      "step": 1431
-    },
-    {
-      "epoch": 0.6565795506648326,
-      "grad_norm": 0.506621200213779,
-      "learning_rate": 0.00027881402450760775,
-      "loss": 2.3205,
-      "step": 1432
-    },
-    {
-      "epoch": 0.6570380559376433,
-      "grad_norm": 0.5072407832849437,
-      "learning_rate": 0.00027814819788528165,
-      "loss": 2.2183,
-      "step": 1433
-    },
-    {
-      "epoch": 0.657496561210454,
-      "grad_norm": 0.5589481122572417,
-      "learning_rate": 0.00027748286075076836,
-      "loss": 2.2647,
-      "step": 1434
-    },
-    {
-      "epoch": 0.6579550664832645,
-      "grad_norm": 0.4961071473709942,
-      "learning_rate": 0.00027681801457204937,
-      "loss": 2.2979,
-      "step": 1435
-    },
-    {
-      "epoch": 0.6584135717560752,
-      "grad_norm": 0.4889169142333946,
-      "learning_rate": 0.00027615366081602306,
-      "loss": 2.2839,
-      "step": 1436
-    },
-    {
-      "epoch": 0.6588720770288858,
-      "grad_norm": 0.5282861908789778,
-      "learning_rate": 0.0002754898009485021,
-      "loss": 2.3349,
-      "step": 1437
-    },
-    {
-      "epoch": 0.6593305823016965,
-      "grad_norm": 0.5330031149955284,
-      "learning_rate": 0.0002748264364342085,
-      "loss": 2.2385,
-      "step": 1438
-    },
-    {
-      "epoch": 0.6597890875745072,
-      "grad_norm": 0.5045755168740601,
-      "learning_rate": 0.00027416356873677204,
-      "loss": 2.2654,
-      "step": 1439
-    },
-    {
-      "epoch": 0.6602475928473177,
-      "grad_norm": 0.4761678716046884,
-      "learning_rate": 0.0002735011993187258,
-      "loss": 2.2722,
-      "step": 1440
-    },
-    {
-      "epoch": 0.6607060981201284,
-      "grad_norm": 0.5158878788792216,
-      "learning_rate": 0.0002728393296415042,
-      "loss": 2.1876,
-      "step": 1441
-    },
-    {
-      "epoch": 0.661164603392939,
-      "grad_norm": 0.4992236474816284,
-      "learning_rate": 0.00027217796116543817,
-      "loss": 2.3502,
-      "step": 1442
-    },
-    {
-      "epoch": 0.6616231086657497,
-      "grad_norm": 0.46931171810152256,
-      "learning_rate": 0.0002715170953497532,
-      "loss": 2.2716,
-      "step": 1443
-    },
-    {
-      "epoch": 0.6620816139385602,
-      "grad_norm": 0.5180135283328481,
-      "learning_rate": 0.00027085673365256614,
-      "loss": 2.2565,
-      "step": 1444
-    },
-    {
-      "epoch": 0.6625401192113709,
-      "grad_norm": 0.5041099285024941,
-      "learning_rate": 0.00027019687753088075,
-      "loss": 2.2547,
-      "step": 1445
-    },
-    {
-      "epoch": 0.6629986244841816,
-      "grad_norm": 0.49757014580325987,
-      "learning_rate": 0.00026953752844058597,
-      "loss": 2.335,
-      "step": 1446
-    },
-    {
-      "epoch": 0.6634571297569922,
-      "grad_norm": 0.4910663799425603,
-      "learning_rate": 0.0002688786878364516,
-      "loss": 2.2158,
-      "step": 1447
-    },
-    {
-      "epoch": 0.6639156350298029,
-      "grad_norm": 0.502141895681862,
-      "learning_rate": 0.00026822035717212597,
-      "loss": 2.2105,
-      "step": 1448
-    },
-    {
-      "epoch": 0.6643741403026134,
-      "grad_norm": 0.5173969410514904,
-      "learning_rate": 0.00026756253790013193,
-      "loss": 2.3025,
-      "step": 1449
-    },
-    {
-      "epoch": 0.6648326455754241,
-      "grad_norm": 0.5007007596771331,
-      "learning_rate": 0.0002669052314718641,
-      "loss": 2.2038,
-      "step": 1450
-    },
-    {
-      "epoch": 0.6652911508482348,
-      "grad_norm": 0.49659005419308766,
-      "learning_rate": 0.0002662484393375855,
-      "loss": 2.2347,
-      "step": 1451
-    },
-    {
-      "epoch": 0.6657496561210454,
-      "grad_norm": 0.5155980421915982,
-      "learning_rate": 0.00026559216294642446,
-      "loss": 2.2391,
-      "step": 1452
-    },
-    {
-      "epoch": 0.6662081613938561,
-      "grad_norm": 0.5207985541036593,
-      "learning_rate": 0.0002649364037463718,
-      "loss": 2.3582,
-      "step": 1453
-    },
-    {
-      "epoch": 0.6666666666666666,
-      "grad_norm": 0.5344497226525378,
-      "learning_rate": 0.0002642811631842764,
-      "loss": 2.273,
-      "step": 1454
-    },
-    {
-      "epoch": 0.6671251719394773,
-      "grad_norm": 0.5012929871691693,
-      "learning_rate": 0.0002636264427058439,
-      "loss": 2.254,
-      "step": 1455
-    },
-    {
-      "epoch": 0.667583677212288,
-      "grad_norm": 0.5191177837285388,
-      "learning_rate": 0.00026297224375563123,
-      "loss": 2.3976,
-      "step": 1456
-    },
-    {
-      "epoch": 0.6680421824850986,
-      "grad_norm": 0.49785031245176836,
-      "learning_rate": 0.00026231856777704575,
-      "loss": 2.2438,
-      "step": 1457
-    },
-    {
-      "epoch": 0.6685006877579092,
-      "grad_norm": 0.4965407209927301,
-      "learning_rate": 0.00026166541621234026,
-      "loss": 2.2835,
-      "step": 1458
-    },
-    {
-      "epoch": 0.6689591930307198,
-      "grad_norm": 0.5133329783371949,
-      "learning_rate": 0.00026101279050261045,
-      "loss": 2.3516,
-      "step": 1459
-    },
-    {
-      "epoch": 0.6694176983035305,
-      "grad_norm": 0.523139413249944,
-      "learning_rate": 0.00026036069208779247,
-      "loss": 2.3511,
-      "step": 1460
-    },
-    {
-      "epoch": 0.6698762035763411,
-      "grad_norm": 0.4933644760929774,
-      "learning_rate": 0.0002597091224066581,
-      "loss": 2.2724,
-      "step": 1461
-    },
-    {
-      "epoch": 0.6703347088491518,
-      "grad_norm": 0.5005166701662377,
-      "learning_rate": 0.00025905808289681365,
-      "loss": 2.2639,
-      "step": 1462
-    },
-    {
-      "epoch": 0.6707932141219624,
-      "grad_norm": 0.5149230441562146,
-      "learning_rate": 0.0002584075749946946,
-      "loss": 2.2076,
-      "step": 1463
-    },
-    {
-      "epoch": 0.671251719394773,
-      "grad_norm": 0.5113360066950257,
-      "learning_rate": 0.00025775760013556424,
-      "loss": 2.3295,
-      "step": 1464
-    },
-    {
-      "epoch": 0.6717102246675837,
-      "grad_norm": 0.5397005499026559,
-      "learning_rate": 0.0002571081597535095,
-      "loss": 2.2325,
-      "step": 1465
-    },
-    {
-      "epoch": 0.6721687299403943,
-      "grad_norm": 0.5482257304339322,
-      "learning_rate": 0.00025645925528143776,
-      "loss": 2.2701,
-      "step": 1466
-    },
-    {
-      "epoch": 0.672627235213205,
-      "grad_norm": 0.4978618517735032,
-      "learning_rate": 0.0002558108881510747,
-      "loss": 2.3384,
-      "step": 1467
-    },
-    {
-      "epoch": 0.6730857404860155,
-      "grad_norm": 0.5111897713330397,
-      "learning_rate": 0.00025516305979295963,
-      "loss": 2.2228,
-      "step": 1468
-    },
-    {
-      "epoch": 0.6735442457588262,
-      "grad_norm": 0.534474744022487,
-      "learning_rate": 0.0002545157716364439,
-      "loss": 2.2681,
-      "step": 1469
-    },
-    {
-      "epoch": 0.6740027510316369,
-      "grad_norm": 0.5392606472989532,
-      "learning_rate": 0.00025386902510968624,
-      "loss": 2.2569,
-      "step": 1470
-    },
-    {
-      "epoch": 0.6744612563044475,
-      "grad_norm": 0.5373988067469657,
-      "learning_rate": 0.00025322282163965095,
-      "loss": 2.302,
-      "step": 1471
-    },
-    {
-      "epoch": 0.6749197615772582,
-      "grad_norm": 0.5382271071324495,
-      "learning_rate": 0.00025257716265210384,
-      "loss": 2.2995,
-      "step": 1472
-    },
-    {
-      "epoch": 0.6753782668500687,
-      "grad_norm": 0.5432000750197926,
-      "learning_rate": 0.0002519320495716091,
-      "loss": 2.2432,
-      "step": 1473
-    },
-    {
-      "epoch": 0.6758367721228794,
-      "grad_norm": 0.4985209711935529,
-      "learning_rate": 0.00025128748382152716,
-      "loss": 2.3616,
-      "step": 1474
-    },
-    {
-      "epoch": 0.6762952773956901,
-      "grad_norm": 0.4947732070514017,
-      "learning_rate": 0.00025064346682401016,
-      "loss": 2.2455,
-      "step": 1475
-    },
-    {
-      "epoch": 0.6767537826685007,
-      "grad_norm": 0.5386824034460025,
-      "learning_rate": 0.0002500000000000001,
-      "loss": 2.2743,
-      "step": 1476
-    },
-    {
-      "epoch": 0.6772122879413113,
-      "grad_norm": 0.5367061302035454,
-      "learning_rate": 0.0002493570847692246,
-      "loss": 2.1632,
-      "step": 1477
-    },
-    {
-      "epoch": 0.6776707932141219,
-      "grad_norm": 0.49361652636741465,
-      "learning_rate": 0.00024871472255019424,
-      "loss": 2.2637,
-      "step": 1478
-    },
-    {
-      "epoch": 0.6781292984869326,
-      "grad_norm": 0.5331018473084127,
-      "learning_rate": 0.00024807291476019994,
-      "loss": 2.2612,
-      "step": 1479
-    },
-    {
-      "epoch": 0.6785878037597433,
-      "grad_norm": 0.5656486739425826,
-      "learning_rate": 0.00024743166281530877,
-      "loss": 2.3078,
-      "step": 1480
-    },
-    {
-      "epoch": 0.6790463090325539,
-      "grad_norm": 0.5340557155030317,
-      "learning_rate": 0.000246790968130362,
-      "loss": 2.304,
-      "step": 1481
-    },
-    {
-      "epoch": 0.6795048143053645,
-      "grad_norm": 0.52661741069275,
-      "learning_rate": 0.0002461508321189706,
-      "loss": 2.2829,
-      "step": 1482
-    },
-    {
-      "epoch": 0.6799633195781751,
-      "grad_norm": 0.4877757506391307,
-      "learning_rate": 0.00024551125619351385,
-      "loss": 2.2873,
-      "step": 1483
-    },
-    {
-      "epoch": 0.6804218248509858,
-      "grad_norm": 0.5080155922467434,
-      "learning_rate": 0.00024487224176513453,
-      "loss": 2.2892,
-      "step": 1484
-    },
-    {
-      "epoch": 0.6808803301237965,
-      "grad_norm": 0.5064489039004293,
-      "learning_rate": 0.0002442337902437365,
-      "loss": 2.2467,
-      "step": 1485
-    },
-    {
-      "epoch": 0.6813388353966071,
-      "grad_norm": 0.5334397422382213,
-      "learning_rate": 0.0002435959030379824,
-      "loss": 2.2689,
-      "step": 1486
-    },
-    {
-      "epoch": 0.6817973406694177,
-      "grad_norm": 0.5059613682867113,
-      "learning_rate": 0.00024295858155528888,
-      "loss": 2.2056,
-      "step": 1487
-    },
-    {
-      "epoch": 0.6822558459422283,
-      "grad_norm": 0.4905363006598041,
-      "learning_rate": 0.00024232182720182523,
-      "loss": 2.2481,
-      "step": 1488
-    },
-    {
-      "epoch": 0.682714351215039,
-      "grad_norm": 0.6107749511895805,
-      "learning_rate": 0.00024168564138250855,
-      "loss": 2.2865,
-      "step": 1489
-    },
-    {
-      "epoch": 0.6831728564878496,
-      "grad_norm": 0.5469308945923277,
-      "learning_rate": 0.00024105002550100246,
-      "loss": 2.3312,
-      "step": 1490
-    },
-    {
-      "epoch": 0.6836313617606602,
-      "grad_norm": 0.4869412804382035,
-      "learning_rate": 0.00024041498095971254,
-      "loss": 2.2619,
-      "step": 1491
-    },
-    {
-      "epoch": 0.6840898670334709,
-      "grad_norm": 0.8035552336370282,
-      "learning_rate": 0.0002397805091597835,
-      "loss": 2.2803,
-      "step": 1492
-    },
-    {
-      "epoch": 0.6845483723062815,
-      "grad_norm": 0.498719158785719,
-      "learning_rate": 0.0002391466115010973,
-      "loss": 2.3297,
-      "step": 1493
-    },
-    {
-      "epoch": 0.6850068775790922,
-      "grad_norm": 0.5288687726347915,
-      "learning_rate": 0.00023851328938226808,
-      "loss": 2.2384,
-      "step": 1494
-    },
-    {
-      "epoch": 0.6854653828519028,
-      "grad_norm": 0.7520340920800699,
-      "learning_rate": 0.00023788054420064109,
-      "loss": 2.3007,
-      "step": 1495
-    },
-    {
-      "epoch": 0.6859238881247134,
-      "grad_norm": 0.4991923713934602,
-      "learning_rate": 0.00023724837735228773,
-      "loss": 2.2931,
-      "step": 1496
-    },
-    {
-      "epoch": 0.686382393397524,
-      "grad_norm": 0.5467404059414905,
-      "learning_rate": 0.00023661679023200422,
-      "loss": 2.3256,
-      "step": 1497
-    },
-    {
-      "epoch": 0.6868408986703347,
-      "grad_norm": 0.46737176504580874,
-      "learning_rate": 0.00023598578423330714,
-      "loss": 2.1968,
-      "step": 1498
-    },
-    {
-      "epoch": 0.6872994039431454,
-      "grad_norm": 0.5112992518679131,
-      "learning_rate": 0.00023535536074843083,
-      "loss": 2.2924,
-      "step": 1499
-    },
-    {
-      "epoch": 0.687757909215956,
-      "grad_norm": 0.5201526827326766,
-      "learning_rate": 0.00023472552116832502,
-      "loss": 2.2233,
-      "step": 1500
-    },
-    {
-      "epoch": 0.6882164144887666,
-      "grad_norm": 0.4963854336801691,
-      "learning_rate": 0.0002340962668826503,
-      "loss": 2.2483,
-      "step": 1501
-    },
-    {
-      "epoch": 0.6886749197615772,
-      "grad_norm": 0.48969453735667995,
-      "learning_rate": 0.00023346759927977663,
-      "loss": 2.2818,
-      "step": 1502
-    },
-    {
-      "epoch": 0.6891334250343879,
-      "grad_norm": 0.5103920852155451,
-      "learning_rate": 0.0002328395197467789,
-      "loss": 2.2395,
-      "step": 1503
-    },
-    {
-      "epoch": 0.6895919303071986,
-      "grad_norm": 0.5166665042700211,
-      "learning_rate": 0.00023221202966943515,
-      "loss": 2.3293,
-      "step": 1504
-    },
-    {
-      "epoch": 0.6900504355800092,
-      "grad_norm": 0.5689172472973941,
-      "learning_rate": 0.0002315851304322223,
-      "loss": 2.252,
-      "step": 1505
-    },
-    {
-      "epoch": 0.6905089408528198,
-      "grad_norm": 0.5100926249291342,
-      "learning_rate": 0.0002309588234183137,
-      "loss": 2.3699,
-      "step": 1506
-    },
-    {
-      "epoch": 0.6909674461256304,
-      "grad_norm": 0.5033524086217923,
-      "learning_rate": 0.00023033311000957653,
-      "loss": 2.2878,
-      "step": 1507
-    },
-    {
-      "epoch": 0.6914259513984411,
-      "grad_norm": 0.5055376982998195,
-      "learning_rate": 0.00022970799158656758,
-      "loss": 2.271,
-      "step": 1508
-    },
-    {
-      "epoch": 0.6918844566712518,
-      "grad_norm": 0.509573136907041,
-      "learning_rate": 0.0002290834695285316,
-      "loss": 2.3409,
-      "step": 1509
-    },
-    {
-      "epoch": 0.6923429619440623,
-      "grad_norm": 0.5003164498833251,
-      "learning_rate": 0.00022845954521339678,
-      "loss": 2.2954,
-      "step": 1510
-    },
-    {
-      "epoch": 0.692801467216873,
-      "grad_norm": 0.5022003137109337,
-      "learning_rate": 0.0002278362200177732,
-      "loss": 2.2003,
-      "step": 1511
-    },
-    {
-      "epoch": 0.6932599724896836,
-      "grad_norm": 0.5589986332904947,
-      "learning_rate": 0.00022721349531694852,
-      "loss": 2.275,
-      "step": 1512
-    },
-    {
-      "epoch": 0.6937184777624943,
-      "grad_norm": 0.5370550409924908,
-      "learning_rate": 0.0002265913724848855,
-      "loss": 2.246,
-      "step": 1513
-    },
-    {
-      "epoch": 0.694176983035305,
-      "grad_norm": 0.49967162121608827,
-      "learning_rate": 0.00022596985289421946,
-      "loss": 2.3223,
-      "step": 1514
-    },
-    {
-      "epoch": 0.6946354883081155,
-      "grad_norm": 0.4962682098097238,
-      "learning_rate": 0.00022534893791625405,
-      "loss": 2.2599,
-      "step": 1515
-    },
-    {
-      "epoch": 0.6950939935809262,
-      "grad_norm": 0.4540478633591032,
-      "learning_rate": 0.00022472862892095968,
-      "loss": 2.1541,
-      "step": 1516
-    },
-    {
-      "epoch": 0.6955524988537368,
-      "grad_norm": 0.5275195283567394,
-      "learning_rate": 0.00022410892727696896,
-      "loss": 2.3622,
-      "step": 1517
-    },
-    {
-      "epoch": 0.6960110041265475,
-      "grad_norm": 0.5214051881238637,
-      "learning_rate": 0.0002234898343515751,
-      "loss": 2.215,
-      "step": 1518
-    },
-    {
-      "epoch": 0.6964695093993581,
-      "grad_norm": 0.4820383606850711,
-      "learning_rate": 0.00022287135151072792,
-      "loss": 2.2504,
-      "step": 1519
-    },
-    {
-      "epoch": 0.6969280146721687,
-      "grad_norm": 0.5377054996318741,
-      "learning_rate": 0.00022225348011903096,
-      "loss": 2.3231,
-      "step": 1520
-    },
-    {
-      "epoch": 0.6973865199449794,
-      "grad_norm": 0.4968503090902347,
-      "learning_rate": 0.0002216362215397393,
-      "loss": 2.1977,
-      "step": 1521
-    },
-    {
-      "epoch": 0.69784502521779,
-      "grad_norm": 0.5121136042389722,
-      "learning_rate": 0.00022101957713475522,
-      "loss": 2.3437,
-      "step": 1522
-    },
-    {
-      "epoch": 0.6983035304906007,
-      "grad_norm": 0.5136811435519598,
-      "learning_rate": 0.00022040354826462666,
-      "loss": 2.3594,
-      "step": 1523
-    },
-    {
-      "epoch": 0.6987620357634112,
-      "grad_norm": 0.49683165441763794,
-      "learning_rate": 0.0002197881362885426,
-      "loss": 2.1914,
-      "step": 1524
-    },
-    {
-      "epoch": 0.6992205410362219,
-      "grad_norm": 0.5834174538135315,
-      "learning_rate": 0.0002191733425643318,
-      "loss": 2.3034,
-      "step": 1525
-    },
-    {
-      "epoch": 0.6996790463090325,
-      "grad_norm": 0.5150628807609408,
-      "learning_rate": 0.00021855916844845826,
-      "loss": 2.2364,
-      "step": 1526
-    },
-    {
-      "epoch": 0.7001375515818432,
-      "grad_norm": 0.5275923772000638,
-      "learning_rate": 0.00021794561529601898,
-      "loss": 2.2357,
-      "step": 1527
-    },
-    {
-      "epoch": 0.7005960568546539,
-      "grad_norm": 0.512760515835184,
-      "learning_rate": 0.00021733268446074138,
-      "loss": 2.3588,
-      "step": 1528
-    },
-    {
-      "epoch": 0.7010545621274644,
-      "grad_norm": 0.548710063075275,
-      "learning_rate": 0.00021672037729497917,
-      "loss": 2.2882,
-      "step": 1529
-    },
-    {
-      "epoch": 0.7015130674002751,
-      "grad_norm": 0.5253476358839626,
-      "learning_rate": 0.0002161086951497106,
-      "loss": 2.2513,
-      "step": 1530
-    },
-    {
-      "epoch": 0.7019715726730857,
-      "grad_norm": 0.6117222927436697,
-      "learning_rate": 0.00021549763937453442,
-      "loss": 2.1866,
-      "step": 1531
-    },
-    {
-      "epoch": 0.7024300779458964,
-      "grad_norm": 0.49037788506631147,
-      "learning_rate": 0.00021488721131766736,
-      "loss": 2.3144,
-      "step": 1532
-    },
-    {
-      "epoch": 0.7028885832187071,
-      "grad_norm": 0.5199797427039786,
-      "learning_rate": 0.00021427741232594183,
-      "loss": 2.2855,
-      "step": 1533
-    },
-    {
-      "epoch": 0.7033470884915176,
-      "grad_norm": 0.5011795754521229,
-      "learning_rate": 0.0002136682437448013,
-      "loss": 2.3325,
-      "step": 1534
-    },
-    {
-      "epoch": 0.7038055937643283,
-      "grad_norm": 0.5077968844892193,
-      "learning_rate": 0.0002130597069182994,
-      "loss": 2.2127,
-      "step": 1535
-    },
-    {
-      "epoch": 0.7042640990371389,
-      "grad_norm": 0.4979062087431842,
-      "learning_rate": 0.0002124518031890948,
-      "loss": 2.2628,
-      "step": 1536
-    },
-    {
-      "epoch": 0.7047226043099496,
-      "grad_norm": 0.5039373340186529,
-      "learning_rate": 0.0002118445338984502,
-      "loss": 2.2986,
-      "step": 1537
-    },
-    {
-      "epoch": 0.7051811095827603,
-      "grad_norm": 0.49804187356083157,
-      "learning_rate": 0.00021123790038622808,
-      "loss": 2.2686,
-      "step": 1538
-    },
-    {
-      "epoch": 0.7056396148555708,
-      "grad_norm": 0.5142286312572508,
-      "learning_rate": 0.0002106319039908879,
-      "loss": 2.2807,
-      "step": 1539
-    },
-    {
-      "epoch": 0.7060981201283815,
-      "grad_norm": 0.4924012592269383,
-      "learning_rate": 0.00021002654604948412,
-      "loss": 2.288,
-      "step": 1540
-    },
-    {
-      "epoch": 0.7065566254011921,
-      "grad_norm": 0.49024068652132796,
-      "learning_rate": 0.00020942182789766172,
-      "loss": 2.3064,
-      "step": 1541
-    },
-    {
-      "epoch": 0.7070151306740028,
-      "grad_norm": 0.5073368297657994,
-      "learning_rate": 0.00020881775086965492,
-      "loss": 2.2311,
-      "step": 1542
-    },
-    {
-      "epoch": 0.7074736359468133,
-      "grad_norm": 0.497515780447841,
-      "learning_rate": 0.00020821431629828246,
-      "loss": 2.2338,
-      "step": 1543
-    },
-    {
-      "epoch": 0.707932141219624,
-      "grad_norm": 0.5220372848717107,
-      "learning_rate": 0.00020761152551494643,
-      "loss": 2.3008,
-      "step": 1544
-    },
-    {
-      "epoch": 0.7083906464924347,
-      "grad_norm": 0.49733627151621623,
-      "learning_rate": 0.00020700937984962798,
-      "loss": 2.2449,
-      "step": 1545
-    },
-    {
-      "epoch": 0.7088491517652453,
-      "grad_norm": 0.5008268962345094,
-      "learning_rate": 0.0002064078806308848,
-      "loss": 2.268,
-      "step": 1546
-    },
-    {
-      "epoch": 0.709307657038056,
-      "grad_norm": 0.4906716985719888,
-      "learning_rate": 0.00020580702918584882,
-      "loss": 2.2723,
-      "step": 1547
-    },
-    {
-      "epoch": 0.7097661623108665,
-      "grad_norm": 0.48894794784714773,
-      "learning_rate": 0.000205206826840222,
-      "loss": 2.3245,
-      "step": 1548
-    },
-    {
-      "epoch": 0.7102246675836772,
-      "grad_norm": 0.5005411857462807,
-      "learning_rate": 0.0002046072749182751,
-      "loss": 2.2818,
-      "step": 1549
-    },
-    {
-      "epoch": 0.7106831728564879,
-      "grad_norm": 0.6166778126185692,
-      "learning_rate": 0.00020400837474284273,
-      "loss": 2.2346,
-      "step": 1550
-    },
-    {
-      "epoch": 0.7111416781292985,
-      "grad_norm": 0.4895922356393894,
-      "learning_rate": 0.0002034101276353224,
-      "loss": 2.1969,
-      "step": 1551
-    },
-    {
-      "epoch": 0.7116001834021092,
-      "grad_norm": 0.5446004722988269,
-      "learning_rate": 0.00020281253491567027,
-      "loss": 2.2177,
-      "step": 1552
-    },
-    {
-      "epoch": 0.7120586886749197,
-      "grad_norm": 0.5129057316178701,
-      "learning_rate": 0.0002022155979023984,
-      "loss": 2.309,
-      "step": 1553
-    },
-    {
-      "epoch": 0.7125171939477304,
-      "grad_norm": 0.4978667613729111,
-      "learning_rate": 0.000201619317912573,
-      "loss": 2.2382,
-      "step": 1554
-    },
-    {
-      "epoch": 0.712975699220541,
-      "grad_norm": 0.5109748973470088,
-      "learning_rate": 0.00020102369626180962,
-      "loss": 2.3605,
-      "step": 1555
-    },
-    {
-      "epoch": 0.7134342044933517,
-      "grad_norm": 0.48961065635989787,
-      "learning_rate": 0.0002004287342642721,
-      "loss": 2.3781,
-      "step": 1556
-    },
-    {
-      "epoch": 0.7138927097661623,
-      "grad_norm": 0.490238789760803,
-      "learning_rate": 0.00019983443323266824,
-      "loss": 2.1813,
-      "step": 1557
-    },
-    {
-      "epoch": 0.7143512150389729,
-      "grad_norm": 0.5025841714713433,
-      "learning_rate": 0.00019924079447824805,
-      "loss": 2.3474,
-      "step": 1558
-    },
-    {
-      "epoch": 0.7148097203117836,
-      "grad_norm": 0.5109545699656275,
-      "learning_rate": 0.00019864781931079977,
-      "loss": 2.2949,
-      "step": 1559
-    },
-    {
-      "epoch": 0.7152682255845942,
-      "grad_norm": 0.5089490555660663,
-      "learning_rate": 0.00019805550903864773,
-      "loss": 2.2785,
-      "step": 1560
-    },
-    {
-      "epoch": 0.7157267308574049,
-      "grad_norm": 0.514978295187205,
-      "learning_rate": 0.00019746386496864948,
-      "loss": 2.3042,
-      "step": 1561
-    },
-    {
-      "epoch": 0.7161852361302155,
-      "grad_norm": 0.5361849411217643,
-      "learning_rate": 0.00019687288840619226,
-      "loss": 2.3258,
-      "step": 1562
-    },
-    {
-      "epoch": 0.7166437414030261,
-      "grad_norm": 0.5182962485257994,
-      "learning_rate": 0.0001962825806551911,
-      "loss": 2.2553,
-      "step": 1563
-    },
-    {
-      "epoch": 0.7171022466758368,
-      "grad_norm": 0.48102513777490796,
-      "learning_rate": 0.0001956929430180846,
-      "loss": 2.274,
-      "step": 1564
-    },
-    {
-      "epoch": 0.7175607519486474,
-      "grad_norm": 0.4798105129123405,
-      "learning_rate": 0.00019510397679583374,
-      "loss": 2.2418,
-      "step": 1565
-    },
-    {
-      "epoch": 0.7180192572214581,
-      "grad_norm": 0.4759130455727512,
-      "learning_rate": 0.0001945156832879174,
-      "loss": 2.2409,
-      "step": 1566
-    },
-    {
-      "epoch": 0.7184777624942686,
-      "grad_norm": 0.5220771032466857,
-      "learning_rate": 0.00019392806379233036,
-      "loss": 2.3695,
-      "step": 1567
-    },
-    {
-      "epoch": 0.7189362677670793,
-      "grad_norm": 0.49393028149769624,
-      "learning_rate": 0.00019334111960558065,
-      "loss": 2.213,
-      "step": 1568
-    },
-    {
-      "epoch": 0.71939477303989,
-      "grad_norm": 0.5192019556869057,
-      "learning_rate": 0.00019275485202268573,
-      "loss": 2.3188,
-      "step": 1569
-    },
-    {
-      "epoch": 0.7198532783127006,
-      "grad_norm": 0.4915482851472972,
-      "learning_rate": 0.00019216926233717085,
-      "loss": 2.1846,
-      "step": 1570
-    },
-    {
-      "epoch": 0.7203117835855112,
-      "grad_norm": 0.5022954300096992,
-      "learning_rate": 0.00019158435184106498,
-      "loss": 2.2909,
-      "step": 1571
-    },
-    {
-      "epoch": 0.7207702888583218,
-      "grad_norm": 0.5213811299813487,
-      "learning_rate": 0.00019100012182489905,
-      "loss": 2.1713,
-      "step": 1572
-    },
-    {
-      "epoch": 0.7212287941311325,
-      "grad_norm": 0.5110758828488375,
-      "learning_rate": 0.00019041657357770226,
-      "loss": 2.2426,
-      "step": 1573
-    },
-    {
-      "epoch": 0.7216872994039432,
-      "grad_norm": 0.5346243270063613,
-      "learning_rate": 0.00018983370838699943,
-      "loss": 2.2832,
-      "step": 1574
-    },
-    {
-      "epoch": 0.7221458046767538,
-      "grad_norm": 0.501673501098792,
-      "learning_rate": 0.00018925152753880892,
-      "loss": 2.2778,
-      "step": 1575
-    },
-    {
-      "epoch": 0.7226043099495644,
-      "grad_norm": 0.5069176774731274,
-      "learning_rate": 0.00018867003231763847,
-      "loss": 2.2521,
-      "step": 1576
-    },
-    {
-      "epoch": 0.723062815222375,
-      "grad_norm": 0.5362866218825197,
-      "learning_rate": 0.00018808922400648375,
-      "loss": 2.301,
-      "step": 1577
-    },
-    {
-      "epoch": 0.7235213204951857,
-      "grad_norm": 0.5618891530570317,
-      "learning_rate": 0.00018750910388682428,
-      "loss": 2.3861,
-      "step": 1578
-    },
-    {
-      "epoch": 0.7239798257679964,
-      "grad_norm": 0.49087980966101996,
-      "learning_rate": 0.00018692967323862125,
-      "loss": 2.2933,
-      "step": 1579
-    },
-    {
-      "epoch": 0.724438331040807,
-      "grad_norm": 0.49602413627723213,
-      "learning_rate": 0.00018635093334031517,
-      "loss": 2.2803,
-      "step": 1580
-    },
-    {
-      "epoch": 0.7248968363136176,
-      "grad_norm": 0.5312611661116198,
-      "learning_rate": 0.00018577288546882165,
-      "loss": 2.2739,
-      "step": 1581
-    },
-    {
-      "epoch": 0.7253553415864282,
-      "grad_norm": 0.4870346961176397,
-      "learning_rate": 0.00018519553089953023,
-      "loss": 2.2808,
-      "step": 1582
-    },
-    {
-      "epoch": 0.7258138468592389,
-      "grad_norm": 0.5146205711782817,
-      "learning_rate": 0.0001846188709063001,
-      "loss": 2.3087,
-      "step": 1583
-    },
-    {
-      "epoch": 0.7262723521320495,
-      "grad_norm": 0.5024317157967517,
-      "learning_rate": 0.00018404290676145857,
-      "loss": 2.274,
-      "step": 1584
-    },
-    {
-      "epoch": 0.7267308574048602,
-      "grad_norm": 0.4849880709549088,
-      "learning_rate": 0.00018346763973579722,
-      "loss": 2.2853,
-      "step": 1585
-    },
-    {
-      "epoch": 0.7271893626776708,
-      "grad_norm": 0.5040264032594898,
-      "learning_rate": 0.00018289307109856939,
-      "loss": 2.3104,
-      "step": 1586
-    },
-    {
-      "epoch": 0.7276478679504814,
-      "grad_norm": 0.49117312157794163,
-      "learning_rate": 0.0001823192021174882,
-      "loss": 2.2826,
-      "step": 1587
-    },
-    {
-      "epoch": 0.7281063732232921,
-      "grad_norm": 0.5135994887840122,
-      "learning_rate": 0.0001817460340587223,
-      "loss": 2.2811,
-      "step": 1588
-    },
-    {
-      "epoch": 0.7285648784961027,
-      "grad_norm": 0.506084139057811,
-      "learning_rate": 0.00018117356818689445,
-      "loss": 2.1961,
-      "step": 1589
-    },
-    {
-      "epoch": 0.7290233837689133,
-      "grad_norm": 0.5028432885278785,
-      "learning_rate": 0.00018060180576507756,
-      "loss": 2.2831,
-      "step": 1590
-    },
-    {
-      "epoch": 0.729481889041724,
-      "grad_norm": 0.4956208517947566,
-      "learning_rate": 0.00018003074805479313,
-      "loss": 2.2256,
-      "step": 1591
-    },
-    {
-      "epoch": 0.7299403943145346,
-      "grad_norm": 0.49462834150596535,
-      "learning_rate": 0.00017946039631600724,
-      "loss": 2.2683,
-      "step": 1592
-    },
-    {
-      "epoch": 0.7303988995873453,
-      "grad_norm": 0.5825578891982529,
-      "learning_rate": 0.00017889075180712837,
-      "loss": 2.3349,
-      "step": 1593
-    },
-    {
-      "epoch": 0.7308574048601559,
-      "grad_norm": 0.5145605854226915,
-      "learning_rate": 0.00017832181578500512,
-      "loss": 2.1115,
-      "step": 1594
-    },
-    {
-      "epoch": 0.7313159101329665,
-      "grad_norm": 0.5261957888312558,
-      "learning_rate": 0.0001777535895049221,
-      "loss": 2.4003,
-      "step": 1595
-    },
-    {
-      "epoch": 0.7317744154057771,
-      "grad_norm": 0.5014541864191274,
-      "learning_rate": 0.0001771860742205988,
-      "loss": 2.3584,
-      "step": 1596
-    },
-    {
-      "epoch": 0.7322329206785878,
-      "grad_norm": 0.5298541977717675,
-      "learning_rate": 0.00017661927118418525,
-      "loss": 2.245,
-      "step": 1597
-    },
-    {
-      "epoch": 0.7326914259513985,
-      "grad_norm": 0.5734058049957399,
-      "learning_rate": 0.00017605318164626066,
-      "loss": 2.2788,
-      "step": 1598
-    },
-    {
-      "epoch": 0.7331499312242091,
-      "grad_norm": 0.5094601249670941,
-      "learning_rate": 0.00017548780685582949,
-      "loss": 2.2802,
-      "step": 1599
-    },
-    {
-      "epoch": 0.7336084364970197,
-      "grad_norm": 0.5071482083435719,
-      "learning_rate": 0.00017492314806031922,
-      "loss": 2.2958,
-      "step": 1600
-    },
-    {
-      "epoch": 0.7340669417698303,
-      "grad_norm": 0.4936986527950718,
-      "learning_rate": 0.00017435920650557806,
-      "loss": 2.2194,
-      "step": 1601
-    },
-    {
-      "epoch": 0.734525447042641,
-      "grad_norm": 0.5483532971441242,
-      "learning_rate": 0.00017379598343587112,
-      "loss": 2.2442,
-      "step": 1602
-    },
-    {
-      "epoch": 0.7349839523154517,
-      "grad_norm": 0.5214242302218202,
-      "learning_rate": 0.00017323348009387878,
-      "loss": 2.23,
-      "step": 1603
-    },
-    {
-      "epoch": 0.7354424575882622,
-      "grad_norm": 0.5116066519835499,
-      "learning_rate": 0.0001726716977206929,
-      "loss": 2.274,
-      "step": 1604
-    },
-    {
-      "epoch": 0.7359009628610729,
-      "grad_norm": 0.5587096552543058,
-      "learning_rate": 0.00017211063755581525,
-      "loss": 2.3722,
-      "step": 1605
-    },
-    {
-      "epoch": 0.7363594681338835,
-      "grad_norm": 0.5291895744436085,
-      "learning_rate": 0.0001715503008371536,
-      "loss": 2.244,
-      "step": 1606
-    },
-    {
-      "epoch": 0.7368179734066942,
-      "grad_norm": 0.5665200129489439,
-      "learning_rate": 0.0001709906888010196,
-      "loss": 2.3025,
-      "step": 1607
-    },
-    {
-      "epoch": 0.7372764786795049,
-      "grad_norm": 0.5147818626759852,
-      "learning_rate": 0.00017043180268212638,
-      "loss": 2.2424,
-      "step": 1608
-    },
-    {
-      "epoch": 0.7377349839523154,
-      "grad_norm": 0.513651904109542,
-      "learning_rate": 0.00016987364371358481,
-      "loss": 2.26,
-      "step": 1609
-    },
-    {
-      "epoch": 0.7381934892251261,
-      "grad_norm": 0.5284704750332726,
-      "learning_rate": 0.00016931621312690214,
-      "loss": 2.3969,
-      "step": 1610
-    },
-    {
-      "epoch": 0.7386519944979367,
-      "grad_norm": 0.48008163779101043,
-      "learning_rate": 0.00016875951215197777,
-      "loss": 2.1397,
-      "step": 1611
-    },
-    {
-      "epoch": 0.7391104997707474,
-      "grad_norm": 0.5450826671724655,
-      "learning_rate": 0.00016820354201710214,
-      "loss": 2.413,
-      "step": 1612
-    },
-    {
-      "epoch": 0.739569005043558,
-      "grad_norm": 0.5458841243071932,
-      "learning_rate": 0.00016764830394895203,
-      "loss": 2.2924,
-      "step": 1613
-    },
-    {
-      "epoch": 0.7400275103163686,
-      "grad_norm": 0.5430483755381278,
-      "learning_rate": 0.00016709379917259027,
-      "loss": 2.2291,
-      "step": 1614
-    },
-    {
-      "epoch": 0.7404860155891793,
-      "grad_norm": 0.485466928749901,
-      "learning_rate": 0.00016654002891146091,
-      "loss": 2.301,
-      "step": 1615
-    },
-    {
-      "epoch": 0.7409445208619899,
-      "grad_norm": 0.47131363933672066,
-      "learning_rate": 0.00016598699438738764,
-      "loss": 2.2861,
-      "step": 1616
-    },
-    {
-      "epoch": 0.7414030261348006,
-      "grad_norm": 0.5438475677238771,
-      "learning_rate": 0.00016543469682057105,
-      "loss": 2.3937,
-      "step": 1617
-    },
-    {
-      "epoch": 0.7418615314076112,
-      "grad_norm": 0.5197168216882007,
-      "learning_rate": 0.00016488313742958526,
-      "loss": 2.2158,
-      "step": 1618
-    },
-    {
-      "epoch": 0.7423200366804218,
-      "grad_norm": 0.5281288517068743,
-      "learning_rate": 0.00016433231743137646,
-      "loss": 2.2649,
-      "step": 1619
-    },
-    {
-      "epoch": 0.7427785419532325,
-      "grad_norm": 0.5101812144340996,
-      "learning_rate": 0.0001637822380412584,
-      "loss": 2.4063,
-      "step": 1620
-    },
-    {
-      "epoch": 0.7432370472260431,
-      "grad_norm": 0.522654977532835,
-      "learning_rate": 0.00016323290047291195,
-      "loss": 2.3015,
-      "step": 1621
-    },
-    {
-      "epoch": 0.7436955524988538,
-      "grad_norm": 0.5397711925044155,
-      "learning_rate": 0.0001626843059383803,
-      "loss": 2.3052,
-      "step": 1622
-    },
-    {
-      "epoch": 0.7441540577716643,
-      "grad_norm": 0.4986582547087325,
-      "learning_rate": 0.00016213645564806752,
-      "loss": 2.2791,
-      "step": 1623
-    },
-    {
-      "epoch": 0.744612563044475,
-      "grad_norm": 0.5356683877902318,
-      "learning_rate": 0.0001615893508107359,
-      "loss": 2.2917,
-      "step": 1624
-    },
-    {
-      "epoch": 0.7450710683172856,
-      "grad_norm": 0.5380156059132745,
-      "learning_rate": 0.00016104299263350252,
-      "loss": 2.218,
-      "step": 1625
-    },
-    {
-      "epoch": 0.7455295735900963,
-      "grad_norm": 0.5161052403685286,
-      "learning_rate": 0.00016049738232183758,
-      "loss": 2.2433,
-      "step": 1626
-    },
-    {
-      "epoch": 0.745988078862907,
-      "grad_norm": 0.5367766084243679,
-      "learning_rate": 0.0001599525210795606,
-      "loss": 2.2369,
-      "step": 1627
-    },
-    {
-      "epoch": 0.7464465841357175,
-      "grad_norm": 0.5517603977531025,
-      "learning_rate": 0.00015940841010883889,
-      "loss": 2.2386,
-      "step": 1628
-    },
-    {
-      "epoch": 0.7469050894085282,
-      "grad_norm": 0.5537663308866031,
-      "learning_rate": 0.00015886505061018413,
-      "loss": 2.316,
-      "step": 1629
-    },
-    {
-      "epoch": 0.7473635946813388,
-      "grad_norm": 0.5226936125266187,
-      "learning_rate": 0.0001583224437824498,
-      "loss": 2.2613,
-      "step": 1630
-    },
-    {
-      "epoch": 0.7478220999541495,
-      "grad_norm": 0.478184876917597,
-      "learning_rate": 0.0001577805908228293,
-      "loss": 2.3289,
-      "step": 1631
-    },
-    {
-      "epoch": 0.7482806052269602,
-      "grad_norm": 0.4805998853410383,
-      "learning_rate": 0.00015723949292685191,
-      "loss": 2.2373,
-      "step": 1632
-    },
-    {
-      "epoch": 0.7487391104997707,
-      "grad_norm": 0.5023159332901762,
-      "learning_rate": 0.0001566991512883818,
-      "loss": 2.2527,
-      "step": 1633
-    },
-    {
-      "epoch": 0.7491976157725814,
-      "grad_norm": 0.5139454280473759,
-      "learning_rate": 0.00015615956709961378,
-      "loss": 2.276,
-      "step": 1634
-    },
-    {
-      "epoch": 0.749656121045392,
-      "grad_norm": 0.5306010612882976,
-      "learning_rate": 0.00015562074155107215,
-      "loss": 2.3191,
-      "step": 1635
-    },
-    {
-      "epoch": 0.7501146263182027,
-      "grad_norm": 0.523751204211952,
-      "learning_rate": 0.0001550826758316068,
-      "loss": 2.3241,
-      "step": 1636
-    },
-    {
-      "epoch": 0.7505731315910132,
-      "grad_norm": 0.5382647022678302,
-      "learning_rate": 0.00015454537112839122,
-      "loss": 2.2815,
-      "step": 1637
-    },
-    {
-      "epoch": 0.7510316368638239,
-      "grad_norm": 0.5304537963883867,
-      "learning_rate": 0.00015400882862692033,
-      "loss": 2.1706,
-      "step": 1638
-    },
-    {
-      "epoch": 0.7514901421366346,
-      "grad_norm": 0.5239939551958582,
-      "learning_rate": 0.00015347304951100665,
-      "loss": 2.2446,
-      "step": 1639
-    },
-    {
-      "epoch": 0.7519486474094452,
-      "grad_norm": 0.4859966409641667,
-      "learning_rate": 0.00015293803496277907,
-      "loss": 2.2362,
-      "step": 1640
-    },
-    {
-      "epoch": 0.7524071526822559,
-      "grad_norm": 0.47509377035431727,
-      "learning_rate": 0.00015240378616267886,
-      "loss": 2.2475,
-      "step": 1641
-    },
-    {
-      "epoch": 0.7528656579550664,
-      "grad_norm": 0.5146263020983162,
-      "learning_rate": 0.00015187030428945843,
-      "loss": 2.3112,
-      "step": 1642
-    },
-    {
-      "epoch": 0.7533241632278771,
-      "grad_norm": 0.4983099537137803,
-      "learning_rate": 0.0001513375905201776,
-      "loss": 2.2238,
-      "step": 1643
-    },
-    {
-      "epoch": 0.7537826685006878,
-      "grad_norm": 0.47542893620526394,
-      "learning_rate": 0.00015080564603020142,
-      "loss": 2.3425,
-      "step": 1644
-    },
-    {
-      "epoch": 0.7542411737734984,
-      "grad_norm": 0.5020174257120055,
-      "learning_rate": 0.0001502744719931982,
-      "loss": 2.3045,
-      "step": 1645
-    },
-    {
-      "epoch": 0.7546996790463091,
-      "grad_norm": 0.5303639017308275,
-      "learning_rate": 0.00014974406958113558,
-      "loss": 2.3083,
-      "step": 1646
-    },
-    {
-      "epoch": 0.7551581843191196,
-      "grad_norm": 0.5149487554762554,
-      "learning_rate": 0.00014921443996427947,
-      "loss": 2.2209,
-      "step": 1647
-    },
-    {
-      "epoch": 0.7556166895919303,
-      "grad_norm": 0.5233154739564948,
-      "learning_rate": 0.0001486855843111901,
-      "loss": 2.2827,
-      "step": 1648
-    },
-    {
-      "epoch": 0.756075194864741,
-      "grad_norm": 0.5288261846978599,
-      "learning_rate": 0.0001481575037887201,
-      "loss": 2.3166,
-      "step": 1649
-    },
-    {
-      "epoch": 0.7565337001375516,
-      "grad_norm": 0.5092875958742993,
-      "learning_rate": 0.00014763019956201253,
-      "loss": 2.1823,
-      "step": 1650
-    },
-    {
-      "epoch": 0.7569922054103623,
-      "grad_norm": 0.5034820412266606,
-      "learning_rate": 0.0001471036727944966,
-      "loss": 2.2316,
-      "step": 1651
-    },
-    {
-      "epoch": 0.7574507106831728,
-      "grad_norm": 0.49919249522653886,
-      "learning_rate": 0.0001465779246478872,
-      "loss": 2.2678,
-      "step": 1652
-    },
-    {
-      "epoch": 0.7579092159559835,
-      "grad_norm": 0.5066584910876444,
-      "learning_rate": 0.00014605295628218045,
-      "loss": 2.2668,
-      "step": 1653
-    },
-    {
-      "epoch": 0.7583677212287941,
-      "grad_norm": 0.4911056741355541,
-      "learning_rate": 0.0001455287688556527,
-      "loss": 2.2377,
-      "step": 1654
-    },
-    {
-      "epoch": 0.7588262265016048,
-      "grad_norm": 0.5649929674270537,
-      "learning_rate": 0.00014500536352485673,
-      "loss": 2.2396,
-      "step": 1655
-    },
-    {
-      "epoch": 0.7592847317744154,
-      "grad_norm": 0.5782915846747136,
-      "learning_rate": 0.00014448274144461965,
-      "loss": 2.376,
-      "step": 1656
-    },
-    {
-      "epoch": 0.759743237047226,
-      "grad_norm": 0.5221704424115859,
-      "learning_rate": 0.00014396090376804112,
-      "loss": 2.3263,
-      "step": 1657
-    },
-    {
-      "epoch": 0.7602017423200367,
-      "grad_norm": 0.507419964720307,
-      "learning_rate": 0.00014343985164648926,
-      "loss": 2.2579,
-      "step": 1658
-    },
-    {
-      "epoch": 0.7606602475928473,
-      "grad_norm": 0.4970410081660402,
-      "learning_rate": 0.00014291958622959973,
-      "loss": 2.3156,
-      "step": 1659
-    },
-    {
-      "epoch": 0.761118752865658,
-      "grad_norm": 0.51444612852573,
-      "learning_rate": 0.00014240010866527176,
-      "loss": 2.2897,
-      "step": 1660
-    },
-    {
-      "epoch": 0.7615772581384685,
-      "grad_norm": 0.5064294575634252,
-      "learning_rate": 0.00014188142009966686,
-      "loss": 2.2585,
-      "step": 1661
-    },
-    {
-      "epoch": 0.7620357634112792,
-      "grad_norm": 0.4996135082178294,
-      "learning_rate": 0.0001413635216772053,
-      "loss": 2.2456,
-      "step": 1662
-    },
-    {
-      "epoch": 0.7624942686840899,
-      "grad_norm": 0.5525468476166044,
-      "learning_rate": 0.000140846414540564,
-      "loss": 2.2191,
-      "step": 1663
-    },
-    {
-      "epoch": 0.7629527739569005,
-      "grad_norm": 0.4890314678376398,
-      "learning_rate": 0.00014033009983067452,
-      "loss": 2.3098,
-      "step": 1664
-    },
-    {
-      "epoch": 0.7634112792297112,
-      "grad_norm": 0.4875408538716374,
-      "learning_rate": 0.00013981457868671927,
-      "loss": 2.3288,
-      "step": 1665
-    },
-    {
-      "epoch": 0.7638697845025217,
-      "grad_norm": 0.4996471305387497,
-      "learning_rate": 0.0001392998522461305,
-      "loss": 2.2694,
-      "step": 1666
-    },
-    {
-      "epoch": 0.7643282897753324,
-      "grad_norm": 0.49337121300112863,
-      "learning_rate": 0.00013878592164458635,
-      "loss": 2.2448,
-      "step": 1667
-    },
-    {
-      "epoch": 0.7647867950481431,
-      "grad_norm": 0.49664067594438827,
-      "learning_rate": 0.00013827278801600978,
-      "loss": 2.2241,
-      "step": 1668
-    },
-    {
-      "epoch": 0.7652453003209537,
-      "grad_norm": 0.49748376118857446,
-      "learning_rate": 0.0001377604524925647,
-      "loss": 2.2839,
-      "step": 1669
-    },
-    {
-      "epoch": 0.7657038055937643,
-      "grad_norm": 0.5268143389453419,
-      "learning_rate": 0.00013724891620465424,
-      "loss": 2.2595,
-      "step": 1670
-    },
-    {
-      "epoch": 0.7661623108665749,
-      "grad_norm": 0.532025145337782,
-      "learning_rate": 0.0001367381802809185,
-      "loss": 2.1769,
-      "step": 1671
-    },
-    {
-      "epoch": 0.7666208161393856,
-      "grad_norm": 0.507015228486743,
-      "learning_rate": 0.00013622824584823113,
-      "loss": 2.2768,
-      "step": 1672
-    },
-    {
-      "epoch": 0.7670793214121963,
-      "grad_norm": 0.5037955209358083,
-      "learning_rate": 0.00013571911403169795,
-      "loss": 2.3189,
-      "step": 1673
-    },
-    {
-      "epoch": 0.7675378266850069,
-      "grad_norm": 0.535079604249999,
-      "learning_rate": 0.0001352107859546533,
-      "loss": 2.2531,
-      "step": 1674
-    },
-    {
-      "epoch": 0.7679963319578175,
-      "grad_norm": 0.5310713129152423,
-      "learning_rate": 0.00013470326273865886,
-      "loss": 2.2501,
-      "step": 1675
-    },
-    {
-      "epoch": 0.7684548372306281,
-      "grad_norm": 0.5603498540857571,
-      "learning_rate": 0.00013419654550349985,
-      "loss": 2.2382,
-      "step": 1676
-    },
-    {
-      "epoch": 0.7689133425034388,
-      "grad_norm": 0.49972484652662125,
-      "learning_rate": 0.00013369063536718346,
-      "loss": 2.3163,
-      "step": 1677
-    },
-    {
-      "epoch": 0.7693718477762495,
-      "grad_norm": 0.5404738071897214,
-      "learning_rate": 0.00013318553344593632,
-      "loss": 2.2849,
-      "step": 1678
-    },
-    {
-      "epoch": 0.7698303530490601,
-      "grad_norm": 0.5369411668947259,
-      "learning_rate": 0.00013268124085420136,
-      "loss": 2.2328,
-      "step": 1679
-    },
-    {
-      "epoch": 0.7702888583218707,
-      "grad_norm": 0.5259471138912987,
-      "learning_rate": 0.0001321777587046364,
-      "loss": 2.1805,
-      "step": 1680
-    },
-    {
-      "epoch": 0.7707473635946813,
-      "grad_norm": 0.5003727656314584,
-      "learning_rate": 0.00013167508810811059,
-      "loss": 2.2815,
-      "step": 1681
-    },
-    {
-      "epoch": 0.771205868867492,
-      "grad_norm": 0.5000465661722135,
-      "learning_rate": 0.0001311732301737029,
-      "loss": 2.2351,
-      "step": 1682
-    },
-    {
-      "epoch": 0.7716643741403026,
-      "grad_norm": 0.5159891224158503,
-      "learning_rate": 0.0001306721860086991,
-      "loss": 2.2663,
-      "step": 1683
-    },
-    {
-      "epoch": 0.7721228794131133,
-      "grad_norm": 0.5165969115139841,
-      "learning_rate": 0.00013017195671858928,
-      "loss": 2.2266,
-      "step": 1684
-    },
-    {
-      "epoch": 0.7725813846859239,
-      "grad_norm": 0.49689462316723704,
-      "learning_rate": 0.0001296725434070661,
-      "loss": 2.221,
-      "step": 1685
-    },
-    {
-      "epoch": 0.7730398899587345,
-      "grad_norm": 0.5104840879661785,
-      "learning_rate": 0.00012917394717602121,
-      "loss": 2.2795,
-      "step": 1686
-    },
-    {
-      "epoch": 0.7734983952315452,
-      "grad_norm": 0.5001178957294948,
-      "learning_rate": 0.00012867616912554426,
-      "loss": 2.3171,
-      "step": 1687
-    },
-    {
-      "epoch": 0.7739569005043558,
-      "grad_norm": 0.5165342621061031,
-      "learning_rate": 0.00012817921035391882,
-      "loss": 2.237,
-      "step": 1688
-    },
-    {
-      "epoch": 0.7744154057771664,
-      "grad_norm": 0.5011108341195203,
-      "learning_rate": 0.00012768307195762168,
-      "loss": 2.3098,
-      "step": 1689
-    },
-    {
-      "epoch": 0.774873911049977,
-      "grad_norm": 0.4955822457665162,
-      "learning_rate": 0.00012718775503131908,
-      "loss": 2.25,
-      "step": 1690
-    },
-    {
-      "epoch": 0.7753324163227877,
-      "grad_norm": 0.48813522551483324,
-      "learning_rate": 0.0001266932606678646,
-      "loss": 2.3756,
-      "step": 1691
-    },
-    {
-      "epoch": 0.7757909215955984,
-      "grad_norm": 0.4942500580830044,
-      "learning_rate": 0.00012619958995829756,
-      "loss": 2.2978,
-      "step": 1692
-    },
-    {
-      "epoch": 0.776249426868409,
-      "grad_norm": 0.502716320572478,
-      "learning_rate": 0.0001257067439918394,
-      "loss": 2.2286,
-      "step": 1693
-    },
-    {
-      "epoch": 0.7767079321412196,
-      "grad_norm": 0.5035498905952134,
-      "learning_rate": 0.00012521472385589234,
-      "loss": 2.2558,
-      "step": 1694
-    },
-    {
-      "epoch": 0.7771664374140302,
-      "grad_norm": 0.5022827072508828,
-      "learning_rate": 0.00012472353063603626,
-      "loss": 2.2179,
-      "step": 1695
-    },
-    {
-      "epoch": 0.7776249426868409,
-      "grad_norm": 0.5324243077560508,
-      "learning_rate": 0.0001242331654160263,
-      "loss": 2.3146,
-      "step": 1696
-    },
-    {
-      "epoch": 0.7780834479596516,
-      "grad_norm": 0.49641365762048895,
-      "learning_rate": 0.0001237436292777914,
-      "loss": 2.3053,
-      "step": 1697
-    },
-    {
-      "epoch": 0.7785419532324622,
-      "grad_norm": 0.4982007221480574,
-      "learning_rate": 0.00012325492330143061,
-      "loss": 2.3106,
-      "step": 1698
-    },
-    {
-      "epoch": 0.7790004585052728,
-      "grad_norm": 0.47491957847268534,
-      "learning_rate": 0.00012276704856521175,
-      "loss": 2.3097,
-      "step": 1699
-    },
-    {
-      "epoch": 0.7794589637780834,
-      "grad_norm": 0.500228658420741,
-      "learning_rate": 0.00012228000614556816,
-      "loss": 2.3055,
-      "step": 1700
-    },
-    {
-      "epoch": 0.7799174690508941,
-      "grad_norm": 0.5038985279049725,
-      "learning_rate": 0.00012179379711709738,
-      "loss": 2.3258,
-      "step": 1701
-    },
-    {
-      "epoch": 0.7803759743237048,
-      "grad_norm": 0.5111462647610527,
-      "learning_rate": 0.0001213084225525577,
-      "loss": 2.2645,
-      "step": 1702
-    },
-    {
-      "epoch": 0.7808344795965153,
-      "grad_norm": 0.494703341849124,
-      "learning_rate": 0.00012082388352286627,
-      "loss": 2.2748,
-      "step": 1703
-    },
-    {
-      "epoch": 0.781292984869326,
-      "grad_norm": 0.48509049038072016,
-      "learning_rate": 0.00012034018109709716,
-      "loss": 2.2249,
-      "step": 1704
-    },
-    {
-      "epoch": 0.7817514901421366,
-      "grad_norm": 0.5029325047664066,
-      "learning_rate": 0.00011985731634247809,
-      "loss": 2.2184,
-      "step": 1705
-    },
-    {
-      "epoch": 0.7822099954149473,
-      "grad_norm": 0.5057381893042493,
-      "learning_rate": 0.00011937529032438904,
-      "loss": 2.332,
-      "step": 1706
-    },
-    {
-      "epoch": 0.782668500687758,
-      "grad_norm": 0.5597363932038633,
-      "learning_rate": 0.00011889410410635887,
-      "loss": 2.2623,
-      "step": 1707
-    },
-    {
-      "epoch": 0.7831270059605685,
-      "grad_norm": 0.48761239506571197,
-      "learning_rate": 0.0001184137587500641,
-      "loss": 2.2851,
-      "step": 1708
-    },
-    {
-      "epoch": 0.7835855112333792,
-      "grad_norm": 0.50022488315428,
-      "learning_rate": 0.00011793425531532564,
-      "loss": 2.1899,
-      "step": 1709
-    },
-    {
-      "epoch": 0.7840440165061898,
-      "grad_norm": 0.4814767174411258,
-      "learning_rate": 0.00011745559486010671,
-      "loss": 2.314,
-      "step": 1710
-    },
-    {
-      "epoch": 0.7845025217790005,
-      "grad_norm": 0.48498750134559787,
-      "learning_rate": 0.00011697777844051105,
-      "loss": 2.2434,
-      "step": 1711
-    },
-    {
-      "epoch": 0.7849610270518111,
-      "grad_norm": 0.5022849498766453,
-      "learning_rate": 0.00011650080711077964,
-      "loss": 2.3654,
-      "step": 1712
-    },
-    {
-      "epoch": 0.7854195323246217,
-      "grad_norm": 0.4880379220727716,
-      "learning_rate": 0.00011602468192328936,
-      "loss": 2.2504,
-      "step": 1713
-    },
-    {
-      "epoch": 0.7858780375974324,
-      "grad_norm": 0.4873568857116793,
-      "learning_rate": 0.00011554940392854973,
-      "loss": 2.3086,
-      "step": 1714
-    },
-    {
-      "epoch": 0.786336542870243,
-      "grad_norm": 0.508966800077224,
-      "learning_rate": 0.00011507497417520146,
-      "loss": 2.3696,
-      "step": 1715
-    },
-    {
-      "epoch": 0.7867950481430537,
-      "grad_norm": 0.4817779516996919,
-      "learning_rate": 0.00011460139371001339,
-      "loss": 2.1886,
-      "step": 1716
-    },
-    {
-      "epoch": 0.7872535534158643,
-      "grad_norm": 0.5132080976496017,
-      "learning_rate": 0.00011412866357788049,
-      "loss": 2.2165,
-      "step": 1717
-    },
-    {
-      "epoch": 0.7877120586886749,
-      "grad_norm": 0.503278539669577,
-      "learning_rate": 0.00011365678482182207,
-      "loss": 2.2164,
-      "step": 1718
-    },
-    {
-      "epoch": 0.7881705639614855,
-      "grad_norm": 0.5341916279094643,
-      "learning_rate": 0.0001131857584829783,
-      "loss": 2.2162,
-      "step": 1719
-    },
-    {
-      "epoch": 0.7886290692342962,
-      "grad_norm": 0.4831429705947554,
-      "learning_rate": 0.0001127155856006093,
-      "loss": 2.1875,
-      "step": 1720
-    },
-    {
-      "epoch": 0.7890875745071069,
-      "grad_norm": 0.5204056002921584,
-      "learning_rate": 0.00011224626721209141,
-      "loss": 2.2368,
-      "step": 1721
-    },
-    {
-      "epoch": 0.7895460797799174,
-      "grad_norm": 0.9259965216010957,
-      "learning_rate": 0.0001117778043529164,
-      "loss": 2.3176,
-      "step": 1722
-    },
-    {
-      "epoch": 0.7900045850527281,
-      "grad_norm": 0.5009032606201166,
-      "learning_rate": 0.0001113101980566879,
-      "loss": 2.1806,
-      "step": 1723
-    },
-    {
-      "epoch": 0.7904630903255387,
-      "grad_norm": 0.5105752403133017,
-      "learning_rate": 0.00011084344935511958,
-      "loss": 2.3447,
-      "step": 1724
-    },
-    {
-      "epoch": 0.7909215955983494,
-      "grad_norm": 0.501870165293479,
-      "learning_rate": 0.00011037755927803345,
-      "loss": 2.1246,
-      "step": 1725
-    },
-    {
-      "epoch": 0.7913801008711601,
-      "grad_norm": 0.4988166512213042,
-      "learning_rate": 0.00010991252885335651,
-      "loss": 2.2708,
-      "step": 1726
-    },
-    {
-      "epoch": 0.7918386061439706,
-      "grad_norm": 0.5233353777180758,
-      "learning_rate": 0.00010944835910711958,
-      "loss": 2.2546,
-      "step": 1727
-    },
-    {
-      "epoch": 0.7922971114167813,
-      "grad_norm": 0.4933388423913291,
-      "learning_rate": 0.00010898505106345396,
-      "loss": 2.1309,
-      "step": 1728
-    },
-    {
-      "epoch": 0.7927556166895919,
-      "grad_norm": 0.5278693396821831,
-      "learning_rate": 0.00010852260574459022,
-      "loss": 2.2288,
-      "step": 1729
-    },
-    {
-      "epoch": 0.7932141219624026,
-      "grad_norm": 0.5174287863977814,
-      "learning_rate": 0.00010806102417085512,
-      "loss": 2.2106,
-      "step": 1730
-    },
-    {
-      "epoch": 0.7936726272352133,
-      "grad_norm": 0.533301498958857,
-      "learning_rate": 0.00010760030736066951,
-      "loss": 2.2455,
-      "step": 1731
-    },
-    {
-      "epoch": 0.7941311325080238,
-      "grad_norm": 0.5326582953354907,
-      "learning_rate": 0.00010714045633054687,
-      "loss": 2.2556,
-      "step": 1732
-    },
-    {
-      "epoch": 0.7945896377808345,
-      "grad_norm": 0.5272674182996155,
-      "learning_rate": 0.00010668147209508971,
-      "loss": 2.2696,
-      "step": 1733
-    },
-    {
-      "epoch": 0.7950481430536451,
-      "grad_norm": 0.5058865689457673,
-      "learning_rate": 0.00010622335566698877,
-      "loss": 2.2569,
-      "step": 1734
-    },
-    {
-      "epoch": 0.7955066483264558,
-      "grad_norm": 0.49287164842836273,
-      "learning_rate": 0.00010576610805701942,
-      "loss": 2.282,
-      "step": 1735
-    },
-    {
-      "epoch": 0.7959651535992663,
-      "grad_norm": 0.5013551951992586,
-      "learning_rate": 0.00010530973027404073,
-      "loss": 2.2396,
-      "step": 1736
-    },
-    {
-      "epoch": 0.796423658872077,
-      "grad_norm": 0.48830216398749426,
-      "learning_rate": 0.00010485422332499212,
-      "loss": 2.3542,
-      "step": 1737
-    },
-    {
-      "epoch": 0.7968821641448877,
-      "grad_norm": 0.48675442985515416,
-      "learning_rate": 0.00010439958821489165,
-      "loss": 2.2077,
-      "step": 1738
-    },
-    {
-      "epoch": 0.7973406694176983,
-      "grad_norm": 0.49663794183073784,
-      "learning_rate": 0.00010394582594683428,
-      "loss": 2.2303,
-      "step": 1739
-    },
-    {
-      "epoch": 0.797799174690509,
-      "grad_norm": 0.5016365193848195,
-      "learning_rate": 0.0001034929375219884,
-      "loss": 2.2259,
-      "step": 1740
-    },
-    {
-      "epoch": 0.7982576799633195,
-      "grad_norm": 0.5081935814416603,
-      "learning_rate": 0.00010304092393959514,
-      "loss": 2.2621,
-      "step": 1741
-    },
-    {
-      "epoch": 0.7987161852361302,
-      "grad_norm": 0.5189679370595702,
-      "learning_rate": 0.00010258978619696468,
-      "loss": 2.249,
-      "step": 1742
-    },
-    {
-      "epoch": 0.7991746905089409,
-      "grad_norm": 0.5090710490952277,
-      "learning_rate": 0.00010213952528947551,
-      "loss": 2.2578,
-      "step": 1743
-    },
-    {
-      "epoch": 0.7996331957817515,
-      "grad_norm": 0.5052986249702603,
-      "learning_rate": 0.00010169014221057089,
-      "loss": 2.2945,
-      "step": 1744
-    },
-    {
-      "epoch": 0.8000917010545622,
-      "grad_norm": 0.5183950854773649,
-      "learning_rate": 0.00010124163795175734,
-      "loss": 2.1364,
-      "step": 1745
-    },
-    {
-      "epoch": 0.8005502063273727,
-      "grad_norm": 0.47797739062116956,
-      "learning_rate": 0.00010079401350260287,
-      "loss": 2.2758,
-      "step": 1746
-    },
-    {
-      "epoch": 0.8010087116001834,
-      "grad_norm": 0.5155985560824953,
-      "learning_rate": 0.00010034726985073362,
-      "loss": 2.3324,
-      "step": 1747
-    },
-    {
-      "epoch": 0.801467216872994,
-      "grad_norm": 0.5150870979509828,
-      "learning_rate": 9.9901407981833e-05,
-      "loss": 2.2875,
-      "step": 1748
-    },
-    {
-      "epoch": 0.8019257221458047,
-      "grad_norm": 0.5086891207494453,
-      "learning_rate": 9.94564288796384e-05,
-      "loss": 2.2253,
-      "step": 1749
-    },
-    {
-      "epoch": 0.8023842274186153,
-      "grad_norm": 0.5254576161282207,
-      "learning_rate": 9.901233352593953e-05,
-      "loss": 2.2721,
-      "step": 1750
-    },
-    {
-      "epoch": 0.8028427326914259,
-      "grad_norm": 0.5325042203772471,
-      "learning_rate": 9.856912290057668e-05,
-      "loss": 2.2293,
-      "step": 1751
-    },
-    {
-      "epoch": 0.8033012379642366,
-      "grad_norm": 0.49089618931066786,
-      "learning_rate": 9.812679798143748e-05,
-      "loss": 2.1479,
-      "step": 1752
-    },
-    {
-      "epoch": 0.8037597432370472,
-      "grad_norm": 0.5037337761601366,
-      "learning_rate": 9.768535974445586e-05,
-      "loss": 2.2092,
-      "step": 1753
-    },
-    {
-      "epoch": 0.8042182485098579,
-      "grad_norm": 0.5431443996327089,
-      "learning_rate": 9.724480916360906e-05,
-      "loss": 2.2767,
-      "step": 1754
-    },
-    {
-      "epoch": 0.8046767537826685,
-      "grad_norm": 0.48104305466228914,
-      "learning_rate": 9.68051472109162e-05,
-      "loss": 2.2768,
-      "step": 1755
-    },
-    {
-      "epoch": 0.8051352590554791,
-      "grad_norm": 0.5074238507226475,
-      "learning_rate": 9.636637485643529e-05,
-      "loss": 2.3062,
-      "step": 1756
-    },
-    {
-      "epoch": 0.8055937643282898,
-      "grad_norm": 0.5272550490626172,
-      "learning_rate": 9.592849306826174e-05,
-      "loss": 2.2471,
-      "step": 1757
-    },
-    {
-      "epoch": 0.8060522696011004,
-      "grad_norm": 0.5018569898526621,
-      "learning_rate": 9.549150281252633e-05,
-      "loss": 2.3351,
-      "step": 1758
-    },
-    {
-      "epoch": 0.8065107748739111,
-      "grad_norm": 0.5193894267108414,
-      "learning_rate": 9.505540505339223e-05,
-      "loss": 2.1829,
-      "step": 1759
-    },
-    {
-      "epoch": 0.8069692801467216,
-      "grad_norm": 0.517992659213132,
-      "learning_rate": 9.4620200753054e-05,
-      "loss": 2.2815,
-      "step": 1760
-    },
-    {
-      "epoch": 0.8074277854195323,
-      "grad_norm": 0.5054432050423933,
-      "learning_rate": 9.418589087173441e-05,
-      "loss": 2.2536,
-      "step": 1761
-    },
-    {
-      "epoch": 0.807886290692343,
-      "grad_norm": 0.5248523189304971,
-      "learning_rate": 9.375247636768325e-05,
-      "loss": 2.2703,
-      "step": 1762
-    },
-    {
-      "epoch": 0.8083447959651536,
-      "grad_norm": 0.514167869814317,
-      "learning_rate": 9.331995819717443e-05,
-      "loss": 2.3436,
-      "step": 1763
-    },
-    {
-      "epoch": 0.8088033012379643,
-      "grad_norm": 0.7925431824006471,
-      "learning_rate": 9.288833731450419e-05,
-      "loss": 2.2981,
-      "step": 1764
-    },
-    {
-      "epoch": 0.8092618065107748,
-      "grad_norm": 0.48283092993487253,
-      "learning_rate": 9.245761467198948e-05,
-      "loss": 2.3569,
-      "step": 1765
-    },
-    {
-      "epoch": 0.8097203117835855,
-      "grad_norm": 0.5286747874540788,
-      "learning_rate": 9.20277912199648e-05,
-      "loss": 2.1788,
-      "step": 1766
-    },
-    {
-      "epoch": 0.8101788170563962,
-      "grad_norm": 0.49568140453521203,
-      "learning_rate": 9.159886790678123e-05,
-      "loss": 2.2141,
-      "step": 1767
-    },
-    {
-      "epoch": 0.8106373223292068,
-      "grad_norm": 0.4969436487872497,
-      "learning_rate": 9.11708456788033e-05,
-      "loss": 2.2289,
-      "step": 1768
-    },
-    {
-      "epoch": 0.8110958276020174,
-      "grad_norm": 0.5273645335093186,
-      "learning_rate": 9.074372548040793e-05,
-      "loss": 2.3003,
-      "step": 1769
-    },
-    {
-      "epoch": 0.811554332874828,
-      "grad_norm": 0.494357272667024,
-      "learning_rate": 9.031750825398145e-05,
-      "loss": 2.2757,
-      "step": 1770
-    },
-    {
-      "epoch": 0.8120128381476387,
-      "grad_norm": 0.4948340757237826,
-      "learning_rate": 8.98921949399179e-05,
-      "loss": 2.2736,
-      "step": 1771
-    },
-    {
-      "epoch": 0.8124713434204494,
-      "grad_norm": 0.5330473256081285,
-      "learning_rate": 8.94677864766173e-05,
-      "loss": 2.2799,
-      "step": 1772
-    },
-    {
-      "epoch": 0.81292984869326,
-      "grad_norm": 0.517664984846126,
-      "learning_rate": 8.904428380048269e-05,
-      "loss": 2.2577,
-      "step": 1773
-    },
-    {
-      "epoch": 0.8133883539660706,
-      "grad_norm": 0.4951842502926313,
-      "learning_rate": 8.862168784591929e-05,
-      "loss": 2.2065,
-      "step": 1774
-    },
-    {
-      "epoch": 0.8138468592388812,
-      "grad_norm": 0.5075064855810636,
-      "learning_rate": 8.819999954533115e-05,
-      "loss": 2.2267,
-      "step": 1775
-    },
-    {
-      "epoch": 0.8143053645116919,
-      "grad_norm": 0.49798636302600247,
-      "learning_rate": 8.777921982911996e-05,
-      "loss": 2.3182,
-      "step": 1776
-    },
-    {
-      "epoch": 0.8147638697845025,
-      "grad_norm": 0.4892698059734464,
-      "learning_rate": 8.735934962568253e-05,
-      "loss": 2.258,
-      "step": 1777
-    },
-    {
-      "epoch": 0.8152223750573132,
-      "grad_norm": 0.48052943871784937,
-      "learning_rate": 8.694038986140945e-05,
-      "loss": 2.3376,
-      "step": 1778
-    },
-    {
-      "epoch": 0.8156808803301238,
-      "grad_norm": 0.5256521372265549,
-      "learning_rate": 8.652234146068206e-05,
-      "loss": 2.1649,
-      "step": 1779
-    },
-    {
-      "epoch": 0.8161393856029344,
-      "grad_norm": 0.49089451883340596,
-      "learning_rate": 8.610520534587086e-05,
-      "loss": 2.1913,
-      "step": 1780
-    },
-    {
-      "epoch": 0.8165978908757451,
-      "grad_norm": 0.5280943101139074,
-      "learning_rate": 8.568898243733397e-05,
-      "loss": 2.2115,
-      "step": 1781
-    },
-    {
-      "epoch": 0.8170563961485557,
-      "grad_norm": 0.523025782685486,
-      "learning_rate": 8.527367365341409e-05,
-      "loss": 2.2837,
-      "step": 1782
-    },
-    {
-      "epoch": 0.8175149014213663,
-      "grad_norm": 0.5175160240309032,
-      "learning_rate": 8.485927991043757e-05,
-      "loss": 2.251,
-      "step": 1783
-    },
-    {
-      "epoch": 0.817973406694177,
-      "grad_norm": 0.5297378703518707,
-      "learning_rate": 8.444580212271125e-05,
-      "loss": 2.3466,
-      "step": 1784
-    },
-    {
-      "epoch": 0.8184319119669876,
-      "grad_norm": 0.5033985274322866,
-      "learning_rate": 8.403324120252159e-05,
-      "loss": 2.3098,
-      "step": 1785
-    },
-    {
-      "epoch": 0.8188904172397983,
-      "grad_norm": 0.47693405363177177,
-      "learning_rate": 8.362159806013175e-05,
-      "loss": 2.2173,
-      "step": 1786
-    },
-    {
-      "epoch": 0.8193489225126089,
-      "grad_norm": 0.52985403650314,
-      "learning_rate": 8.321087360377988e-05,
-      "loss": 2.3411,
-      "step": 1787
-    },
-    {
-      "epoch": 0.8198074277854195,
-      "grad_norm": 0.5261901297142636,
-      "learning_rate": 8.280106873967752e-05,
-      "loss": 2.2767,
-      "step": 1788
-    },
-    {
-      "epoch": 0.8202659330582301,
-      "grad_norm": 0.507448878223163,
-      "learning_rate": 8.239218437200679e-05,
-      "loss": 2.1825,
-      "step": 1789
-    },
-    {
-      "epoch": 0.8207244383310408,
-      "grad_norm": 0.5050717130484134,
-      "learning_rate": 8.198422140291939e-05,
-      "loss": 2.2667,
-      "step": 1790
-    },
-    {
-      "epoch": 0.8211829436038515,
-      "grad_norm": 0.5051591838917546,
-      "learning_rate": 8.157718073253351e-05,
-      "loss": 2.2743,
-      "step": 1791
-    },
-    {
-      "epoch": 0.8216414488766621,
-      "grad_norm": 0.4909781948342899,
-      "learning_rate": 8.117106325893287e-05,
-      "loss": 2.275,
-      "step": 1792
-    },
-    {
-      "epoch": 0.8220999541494727,
-      "grad_norm": 0.4930040472870836,
-      "learning_rate": 8.076586987816404e-05,
-      "loss": 2.3104,
-      "step": 1793
-    },
-    {
-      "epoch": 0.8225584594222833,
-      "grad_norm": 0.4766073345079148,
-      "learning_rate": 8.036160148423449e-05,
-      "loss": 2.1955,
-      "step": 1794
-    },
-    {
-      "epoch": 0.823016964695094,
-      "grad_norm": 0.524861349013132,
-      "learning_rate": 7.995825896911141e-05,
-      "loss": 2.2422,
-      "step": 1795
-    },
-    {
-      "epoch": 0.8234754699679047,
-      "grad_norm": 0.491950374155697,
-      "learning_rate": 7.955584322271853e-05,
-      "loss": 2.2333,
-      "step": 1796
-    },
-    {
-      "epoch": 0.8239339752407153,
-      "grad_norm": 0.5010458071576674,
-      "learning_rate": 7.915435513293523e-05,
-      "loss": 2.2098,
-      "step": 1797
-    },
-    {
-      "epoch": 0.8243924805135259,
-      "grad_norm": 0.503126680348386,
-      "learning_rate": 7.875379558559387e-05,
-      "loss": 2.327,
-      "step": 1798
-    },
-    {
-      "epoch": 0.8248509857863365,
-      "grad_norm": 0.5035020765386163,
-      "learning_rate": 7.835416546447838e-05,
-      "loss": 2.295,
-      "step": 1799
-    },
-    {
-      "epoch": 0.8253094910591472,
-      "grad_norm": 0.5056951011691955,
-      "learning_rate": 7.795546565132167e-05,
-      "loss": 2.245,
-      "step": 1800
-    },
-    {
-      "epoch": 0.8257679963319579,
-      "grad_norm": 0.5402398885006325,
-      "learning_rate": 7.755769702580412e-05,
-      "loss": 2.3013,
-      "step": 1801
-    },
-    {
-      "epoch": 0.8262265016047684,
-      "grad_norm": 0.7695048619359408,
-      "learning_rate": 7.716086046555193e-05,
-      "loss": 2.2902,
-      "step": 1802
-    },
-    {
-      "epoch": 0.8266850068775791,
-      "grad_norm": 0.5237617558105836,
-      "learning_rate": 7.676495684613432e-05,
-      "loss": 2.2918,
-      "step": 1803
-    },
-    {
-      "epoch": 0.8271435121503897,
-      "grad_norm": 0.5312495770698155,
-      "learning_rate": 7.636998704106252e-05,
-      "loss": 2.2934,
-      "step": 1804
-    },
-    {
-      "epoch": 0.8276020174232004,
-      "grad_norm": 0.49460307422865696,
-      "learning_rate": 7.597595192178702e-05,
-      "loss": 2.2161,
-      "step": 1805
-    },
-    {
-      "epoch": 0.828060522696011,
-      "grad_norm": 0.4967959755665021,
-      "learning_rate": 7.558285235769646e-05,
-      "loss": 2.273,
-      "step": 1806
-    },
-    {
-      "epoch": 0.8285190279688216,
-      "grad_norm": 0.4844449020714296,
-      "learning_rate": 7.519068921611494e-05,
-      "loss": 2.3208,
-      "step": 1807
-    },
-    {
-      "epoch": 0.8289775332416323,
-      "grad_norm": 0.526931032961215,
-      "learning_rate": 7.479946336230047e-05,
-      "loss": 2.2556,
-      "step": 1808
-    },
-    {
-      "epoch": 0.8294360385144429,
-      "grad_norm": 0.4976579206208824,
-      "learning_rate": 7.440917565944349e-05,
-      "loss": 2.1602,
-      "step": 1809
-    },
-    {
-      "epoch": 0.8298945437872536,
-      "grad_norm": 0.4904767907958703,
-      "learning_rate": 7.4019826968664e-05,
-      "loss": 2.127,
-      "step": 1810
-    },
-    {
-      "epoch": 0.8303530490600642,
-      "grad_norm": 0.4997255833748969,
-      "learning_rate": 7.363141814901053e-05,
-      "loss": 2.2327,
-      "step": 1811
-    },
-    {
-      "epoch": 0.8308115543328748,
-      "grad_norm": 0.495423465462944,
-      "learning_rate": 7.32439500574577e-05,
-      "loss": 2.1823,
-      "step": 1812
-    },
-    {
-      "epoch": 0.8312700596056855,
-      "grad_norm": 0.48497043820007507,
-      "learning_rate": 7.285742354890473e-05,
-      "loss": 2.2794,
-      "step": 1813
-    },
-    {
-      "epoch": 0.8317285648784961,
-      "grad_norm": 0.531769779811923,
-      "learning_rate": 7.247183947617325e-05,
-      "loss": 2.3277,
-      "step": 1814
-    },
-    {
-      "epoch": 0.8321870701513068,
-      "grad_norm": 0.49045516345864154,
-      "learning_rate": 7.20871986900053e-05,
-      "loss": 2.3177,
-      "step": 1815
-    },
-    {
-      "epoch": 0.8326455754241173,
-      "grad_norm": 0.5065363419295845,
-      "learning_rate": 7.170350203906218e-05,
-      "loss": 2.2531,
-      "step": 1816
-    },
-    {
-      "epoch": 0.833104080696928,
-      "grad_norm": 0.49752407871435816,
-      "learning_rate": 7.132075036992158e-05,
-      "loss": 2.25,
-      "step": 1817
-    },
-    {
-      "epoch": 0.8335625859697386,
-      "grad_norm": 0.5183199231651351,
-      "learning_rate": 7.093894452707666e-05,
-      "loss": 2.1435,
-      "step": 1818
-    },
-    {
-      "epoch": 0.8340210912425493,
-      "grad_norm": 0.5117556164197722,
-      "learning_rate": 7.055808535293334e-05,
-      "loss": 2.3336,
-      "step": 1819
-    },
-    {
-      "epoch": 0.83447959651536,
-      "grad_norm": 0.47411425273253854,
-      "learning_rate": 7.017817368780888e-05,
-      "loss": 2.2597,
-      "step": 1820
-    },
-    {
-      "epoch": 0.8349381017881705,
-      "grad_norm": 0.5337890479401532,
-      "learning_rate": 6.979921036993042e-05,
-      "loss": 2.2176,
-      "step": 1821
-    },
-    {
-      "epoch": 0.8353966070609812,
-      "grad_norm": 0.46748380178204074,
-      "learning_rate": 6.942119623543202e-05,
-      "loss": 2.3123,
-      "step": 1822
-    },
-    {
-      "epoch": 0.8358551123337918,
-      "grad_norm": 0.49731868658759276,
-      "learning_rate": 6.904413211835414e-05,
-      "loss": 2.2839,
-      "step": 1823
-    },
-    {
-      "epoch": 0.8363136176066025,
-      "grad_norm": 0.5088243739949531,
-      "learning_rate": 6.866801885064056e-05,
-      "loss": 2.2071,
-      "step": 1824
-    },
-    {
-      "epoch": 0.8367721228794132,
-      "grad_norm": 0.478708326454388,
-      "learning_rate": 6.829285726213769e-05,
-      "loss": 2.2518,
-      "step": 1825
-    },
-    {
-      "epoch": 0.8372306281522237,
-      "grad_norm": 0.5196706024317812,
-      "learning_rate": 6.79186481805918e-05,
-      "loss": 2.261,
-      "step": 1826
-    },
-    {
-      "epoch": 0.8376891334250344,
-      "grad_norm": 0.5552700746054026,
-      "learning_rate": 6.754539243164754e-05,
-      "loss": 2.1879,
-      "step": 1827
-    },
-    {
-      "epoch": 0.838147638697845,
-      "grad_norm": 0.49468242585428823,
-      "learning_rate": 6.717309083884654e-05,
-      "loss": 2.3612,
-      "step": 1828
-    },
-    {
-      "epoch": 0.8386061439706557,
-      "grad_norm": 0.5071540793674566,
-      "learning_rate": 6.680174422362468e-05,
-      "loss": 2.2132,
-      "step": 1829
-    },
-    {
-      "epoch": 0.8390646492434664,
-      "grad_norm": 0.5455426791084509,
-      "learning_rate": 6.643135340531136e-05,
-      "loss": 2.3289,
-      "step": 1830
-    },
-    {
-      "epoch": 0.8395231545162769,
-      "grad_norm": 0.4735439662090218,
-      "learning_rate": 6.606191920112664e-05,
-      "loss": 2.2824,
-      "step": 1831
-    },
-    {
-      "epoch": 0.8399816597890876,
-      "grad_norm": 0.49664494503679024,
-      "learning_rate": 6.569344242618036e-05,
-      "loss": 2.2723,
-      "step": 1832
-    },
-    {
-      "epoch": 0.8404401650618982,
-      "grad_norm": 0.6022450884941623,
-      "learning_rate": 6.532592389346958e-05,
-      "loss": 2.2847,
-      "step": 1833
-    },
-    {
-      "epoch": 0.8408986703347089,
-      "grad_norm": 0.5387162832201386,
-      "learning_rate": 6.495936441387713e-05,
-      "loss": 2.2608,
-      "step": 1834
-    },
-    {
-      "epoch": 0.8413571756075194,
-      "grad_norm": 0.5241259912720828,
-      "learning_rate": 6.459376479617013e-05,
-      "loss": 2.249,
-      "step": 1835
-    },
-    {
-      "epoch": 0.8418156808803301,
-      "grad_norm": 0.5170384715625865,
-      "learning_rate": 6.422912584699752e-05,
-      "loss": 2.2831,
-      "step": 1836
-    },
-    {
-      "epoch": 0.8422741861531408,
-      "grad_norm": 0.5222311207419498,
-      "learning_rate": 6.386544837088904e-05,
-      "loss": 2.3738,
-      "step": 1837
-    },
-    {
-      "epoch": 0.8427326914259514,
-      "grad_norm": 0.5119799252751762,
-      "learning_rate": 6.350273317025251e-05,
-      "loss": 2.1221,
-      "step": 1838
-    },
-    {
-      "epoch": 0.8431911966987621,
-      "grad_norm": 0.48825979281009435,
-      "learning_rate": 6.314098104537324e-05,
-      "loss": 2.2176,
-      "step": 1839
-    },
-    {
-      "epoch": 0.8436497019715726,
-      "grad_norm": 0.5549189026541644,
-      "learning_rate": 6.278019279441122e-05,
-      "loss": 2.2166,
-      "step": 1840
-    },
-    {
-      "epoch": 0.8441082072443833,
-      "grad_norm": 0.5025642738771449,
-      "learning_rate": 6.242036921339972e-05,
-      "loss": 2.1464,
-      "step": 1841
-    },
-    {
-      "epoch": 0.844566712517194,
-      "grad_norm": 0.4986752269862454,
-      "learning_rate": 6.206151109624402e-05,
-      "loss": 2.2447,
-      "step": 1842
-    },
-    {
-      "epoch": 0.8450252177900046,
-      "grad_norm": 0.5217264424837558,
-      "learning_rate": 6.170361923471868e-05,
-      "loss": 2.2307,
-      "step": 1843
-    },
-    {
-      "epoch": 0.8454837230628153,
-      "grad_norm": 0.4980412818323487,
-      "learning_rate": 6.134669441846691e-05,
-      "loss": 2.252,
-      "step": 1844
-    },
-    {
-      "epoch": 0.8459422283356258,
-      "grad_norm": 0.5606099408506974,
-      "learning_rate": 6.099073743499772e-05,
-      "loss": 2.2746,
-      "step": 1845
-    },
-    {
-      "epoch": 0.8464007336084365,
-      "grad_norm": 0.5595824586089312,
-      "learning_rate": 6.063574906968511e-05,
-      "loss": 2.341,
-      "step": 1846
-    },
-    {
-      "epoch": 0.8468592388812471,
-      "grad_norm": 2.777085773810831,
-      "learning_rate": 6.028173010576582e-05,
-      "loss": 2.1878,
-      "step": 1847
-    },
-    {
-      "epoch": 0.8473177441540578,
-      "grad_norm": 0.5081281313808489,
-      "learning_rate": 5.9928681324337544e-05,
-      "loss": 2.1731,
-      "step": 1848
-    },
-    {
-      "epoch": 0.8477762494268684,
-      "grad_norm": 0.5046436238518828,
-      "learning_rate": 5.957660350435773e-05,
-      "loss": 2.2631,
-      "step": 1849
-    },
-    {
-      "epoch": 0.848234754699679,
-      "grad_norm": 0.5027002880830774,
-      "learning_rate": 5.922549742264122e-05,
-      "loss": 2.1366,
-      "step": 1850
-    },
-    {
-      "epoch": 0.8486932599724897,
-      "grad_norm": 0.47483032201790276,
-      "learning_rate": 5.8875363853859166e-05,
-      "loss": 2.2671,
-      "step": 1851
-    },
-    {
-      "epoch": 0.8491517652453003,
-      "grad_norm": 0.5079562123681465,
-      "learning_rate": 5.852620357053651e-05,
-      "loss": 2.2182,
-      "step": 1852
-    },
-    {
-      "epoch": 0.849610270518111,
-      "grad_norm": 0.5187091150692763,
-      "learning_rate": 5.8178017343051336e-05,
-      "loss": 2.2243,
-      "step": 1853
-    },
-    {
-      "epoch": 0.8500687757909215,
-      "grad_norm": 0.5233552733821646,
-      "learning_rate": 5.783080593963219e-05,
-      "loss": 2.2735,
-      "step": 1854
-    },
-    {
-      "epoch": 0.8505272810637322,
-      "grad_norm": 0.5133546860245916,
-      "learning_rate": 5.748457012635683e-05,
-      "loss": 2.2381,
-      "step": 1855
-    },
-    {
-      "epoch": 0.8509857863365429,
-      "grad_norm": 0.5802687697333587,
-      "learning_rate": 5.713931066715078e-05,
-      "loss": 2.3383,
-      "step": 1856
-    },
-    {
-      "epoch": 0.8514442916093535,
-      "grad_norm": 0.5387981762755145,
-      "learning_rate": 5.679502832378497e-05,
-      "loss": 2.2543,
-      "step": 1857
-    },
-    {
-      "epoch": 0.8519027968821642,
-      "grad_norm": 0.512767896963743,
-      "learning_rate": 5.645172385587482e-05,
-      "loss": 2.2178,
-      "step": 1858
-    },
-    {
-      "epoch": 0.8523613021549747,
-      "grad_norm": 0.5438007887412479,
-      "learning_rate": 5.6109398020877834e-05,
-      "loss": 2.2474,
-      "step": 1859
-    },
-    {
-      "epoch": 0.8528198074277854,
-      "grad_norm": 0.554511422599296,
-      "learning_rate": 5.576805157409265e-05,
-      "loss": 2.3331,
-      "step": 1860
-    },
-    {
-      "epoch": 0.8532783127005961,
-      "grad_norm": 0.5146565425368497,
-      "learning_rate": 5.542768526865677e-05,
-      "loss": 2.3048,
-      "step": 1861
-    },
-    {
-      "epoch": 0.8537368179734067,
-      "grad_norm": 0.5056175981785931,
-      "learning_rate": 5.508829985554509e-05,
-      "loss": 2.2649,
-      "step": 1862
-    },
-    {
-      "epoch": 0.8541953232462174,
-      "grad_norm": 0.5378730187377647,
-      "learning_rate": 5.474989608356856e-05,
-      "loss": 2.2431,
-      "step": 1863
-    },
-    {
-      "epoch": 0.8546538285190279,
-      "grad_norm": 0.5042360360041317,
-      "learning_rate": 5.441247469937194e-05,
-      "loss": 2.2171,
-      "step": 1864
-    },
-    {
-      "epoch": 0.8551123337918386,
-      "grad_norm": 0.5419037201137222,
-      "learning_rate": 5.407603644743286e-05,
-      "loss": 2.3189,
-      "step": 1865
-    },
-    {
-      "epoch": 0.8555708390646493,
-      "grad_norm": 0.5263951153503227,
-      "learning_rate": 5.374058207005944e-05,
-      "loss": 2.1785,
-      "step": 1866
-    },
-    {
-      "epoch": 0.8560293443374599,
-      "grad_norm": 0.5536770299338741,
-      "learning_rate": 5.3406112307389066e-05,
-      "loss": 2.2788,
-      "step": 1867
-    },
-    {
-      "epoch": 0.8564878496102705,
-      "grad_norm": 0.5840216273128637,
-      "learning_rate": 5.3072627897386926e-05,
-      "loss": 2.3698,
-      "step": 1868
-    },
-    {
-      "epoch": 0.8569463548830811,
-      "grad_norm": 0.5678096143041548,
-      "learning_rate": 5.27401295758439e-05,
-      "loss": 2.3298,
-      "step": 1869
-    },
-    {
-      "epoch": 0.8574048601558918,
-      "grad_norm": 0.5106873999152276,
-      "learning_rate": 5.2408618076375315e-05,
-      "loss": 2.2352,
-      "step": 1870
-    },
-    {
-      "epoch": 0.8578633654287025,
-      "grad_norm": 0.5844942995556011,
-      "learning_rate": 5.207809413041914e-05,
-      "loss": 2.2874,
-      "step": 1871
-    },
-    {
-      "epoch": 0.8583218707015131,
-      "grad_norm": 0.5473537327259399,
-      "learning_rate": 5.174855846723459e-05,
-      "loss": 2.1546,
-      "step": 1872
-    },
-    {
-      "epoch": 0.8587803759743237,
-      "grad_norm": 0.5006007001970149,
-      "learning_rate": 5.1420011813900104e-05,
-      "loss": 2.3257,
-      "step": 1873
-    },
-    {
-      "epoch": 0.8592388812471343,
-      "grad_norm": 0.530974648753256,
-      "learning_rate": 5.109245489531211e-05,
-      "loss": 2.2047,
-      "step": 1874
-    },
-    {
-      "epoch": 0.859697386519945,
-      "grad_norm": 0.5081803617100108,
-      "learning_rate": 5.0765888434183446e-05,
-      "loss": 2.1567,
-      "step": 1875
-    },
-    {
-      "epoch": 0.8601558917927556,
-      "grad_norm": 0.5078007034160346,
-      "learning_rate": 5.0440313151041364e-05,
-      "loss": 2.2828,
-      "step": 1876
-    },
-    {
-      "epoch": 0.8606143970655663,
-      "grad_norm": 0.5041400478863098,
-      "learning_rate": 5.011572976422657e-05,
-      "loss": 2.205,
-      "step": 1877
-    },
-    {
-      "epoch": 0.8610729023383769,
-      "grad_norm": 0.5420110396379942,
-      "learning_rate": 4.9792138989890825e-05,
-      "loss": 2.316,
-      "step": 1878
-    },
-    {
-      "epoch": 0.8615314076111875,
-      "grad_norm": 0.5381884459939758,
-      "learning_rate": 4.9469541541996234e-05,
-      "loss": 2.2109,
-      "step": 1879
-    },
-    {
-      "epoch": 0.8619899128839982,
-      "grad_norm": 0.5097377123626542,
-      "learning_rate": 4.914793813231305e-05,
-      "loss": 2.2694,
-      "step": 1880
-    },
-    {
-      "epoch": 0.8624484181568088,
-      "grad_norm": 0.9673346202812244,
-      "learning_rate": 4.882732947041818e-05,
-      "loss": 2.2993,
-      "step": 1881
-    },
-    {
-      "epoch": 0.8629069234296194,
-      "grad_norm": 0.5149817391799207,
-      "learning_rate": 4.850771626369416e-05,
-      "loss": 2.2442,
-      "step": 1882
-    },
-    {
-      "epoch": 0.86336542870243,
-      "grad_norm": 0.5200678375355416,
-      "learning_rate": 4.818909921732662e-05,
-      "loss": 2.2385,
-      "step": 1883
-    },
-    {
-      "epoch": 0.8638239339752407,
-      "grad_norm": 0.5269470517509824,
-      "learning_rate": 4.787147903430383e-05,
-      "loss": 2.299,
-      "step": 1884
-    },
-    {
-      "epoch": 0.8642824392480514,
-      "grad_norm": 0.5015662398618534,
-      "learning_rate": 4.755485641541424e-05,
-      "loss": 2.2727,
-      "step": 1885
-    },
-    {
-      "epoch": 0.864740944520862,
-      "grad_norm": 0.5713692957217279,
-      "learning_rate": 4.723923205924557e-05,
-      "loss": 2.2684,
-      "step": 1886
-    },
-    {
-      "epoch": 0.8651994497936726,
-      "grad_norm": 0.5703396447789385,
-      "learning_rate": 4.6924606662182736e-05,
-      "loss": 2.2838,
-      "step": 1887
-    },
-    {
-      "epoch": 0.8656579550664832,
-      "grad_norm": 0.5796166739564688,
-      "learning_rate": 4.6610980918406596e-05,
-      "loss": 2.2573,
-      "step": 1888
-    },
-    {
-      "epoch": 0.8661164603392939,
-      "grad_norm": 0.5195539887864064,
-      "learning_rate": 4.629835551989276e-05,
-      "loss": 2.306,
-      "step": 1889
-    },
-    {
-      "epoch": 0.8665749656121046,
-      "grad_norm": 0.5277774590257224,
-      "learning_rate": 4.5986731156409224e-05,
-      "loss": 2.1872,
-      "step": 1890
-    },
-    {
-      "epoch": 0.8670334708849152,
-      "grad_norm": 0.5591210220516896,
-      "learning_rate": 4.567610851551568e-05,
-      "loss": 2.2785,
-      "step": 1891
-    },
-    {
-      "epoch": 0.8674919761577258,
-      "grad_norm": 0.4964997423590212,
-      "learning_rate": 4.536648828256146e-05,
-      "loss": 2.2206,
-      "step": 1892
-    },
-    {
-      "epoch": 0.8679504814305364,
-      "grad_norm": 0.5098769513514946,
-      "learning_rate": 4.505787114068433e-05,
-      "loss": 2.2716,
-      "step": 1893
-    },
-    {
-      "epoch": 0.8684089867033471,
-      "grad_norm": 0.515784743360786,
-      "learning_rate": 4.4750257770808764e-05,
-      "loss": 2.1961,
-      "step": 1894
-    },
-    {
-      "epoch": 0.8688674919761578,
-      "grad_norm": 0.5149139498891653,
-      "learning_rate": 4.444364885164448e-05,
-      "loss": 2.1848,
-      "step": 1895
-    },
-    {
-      "epoch": 0.8693259972489684,
-      "grad_norm": 0.5831567405934395,
-      "learning_rate": 4.413804505968533e-05,
-      "loss": 2.246,
-      "step": 1896
-    },
-    {
-      "epoch": 0.869784502521779,
-      "grad_norm": 0.5138706640316592,
-      "learning_rate": 4.3833447069206944e-05,
-      "loss": 2.2388,
-      "step": 1897
-    },
-    {
-      "epoch": 0.8702430077945896,
-      "grad_norm": 0.4969601218617517,
-      "learning_rate": 4.352985555226635e-05,
-      "loss": 2.2044,
-      "step": 1898
-    },
-    {
-      "epoch": 0.8707015130674003,
-      "grad_norm": 0.600292293746169,
-      "learning_rate": 4.322727117869951e-05,
-      "loss": 2.278,
-      "step": 1899
-    },
-    {
-      "epoch": 0.871160018340211,
-      "grad_norm": 0.5304954353465424,
-      "learning_rate": 4.29256946161205e-05,
-      "loss": 2.2705,
-      "step": 1900
-    },
-    {
-      "epoch": 0.8716185236130215,
-      "grad_norm": 0.5120384649871955,
-      "learning_rate": 4.262512652991968e-05,
-      "loss": 2.2936,
-      "step": 1901
-    },
-    {
-      "epoch": 0.8720770288858322,
-      "grad_norm": 0.5537094937411666,
-      "learning_rate": 4.2325567583262113e-05,
-      "loss": 2.2762,
-      "step": 1902
-    },
-    {
-      "epoch": 0.8725355341586428,
-      "grad_norm": 0.5162128318191855,
-      "learning_rate": 4.2027018437086895e-05,
-      "loss": 2.251,
-      "step": 1903
-    },
-    {
-      "epoch": 0.8729940394314535,
-      "grad_norm": 0.4870221854282856,
-      "learning_rate": 4.172947975010449e-05,
-      "loss": 2.2301,
-      "step": 1904
-    },
-    {
-      "epoch": 0.8734525447042641,
-      "grad_norm": 0.527014737896177,
-      "learning_rate": 4.143295217879645e-05,
-      "loss": 2.2429,
-      "step": 1905
-    },
-    {
-      "epoch": 0.8739110499770747,
-      "grad_norm": 0.514726289415392,
-      "learning_rate": 4.113743637741296e-05,
-      "loss": 2.1404,
-      "step": 1906
-    },
-    {
-      "epoch": 0.8743695552498854,
-      "grad_norm": 0.497077047076629,
-      "learning_rate": 4.084293299797226e-05,
-      "loss": 2.1908,
-      "step": 1907
-    },
-    {
-      "epoch": 0.874828060522696,
-      "grad_norm": 0.7189944429186459,
-      "learning_rate": 4.054944269025862e-05,
-      "loss": 2.2364,
-      "step": 1908
-    },
-    {
-      "epoch": 0.8752865657955067,
-      "grad_norm": 0.537613410268457,
-      "learning_rate": 4.025696610182095e-05,
-      "loss": 2.2556,
-      "step": 1909
-    },
-    {
-      "epoch": 0.8757450710683173,
-      "grad_norm": 0.5380865023916365,
-      "learning_rate": 3.996550387797187e-05,
-      "loss": 2.1999,
-      "step": 1910
-    },
-    {
-      "epoch": 0.8762035763411279,
-      "grad_norm": 0.5383686129732749,
-      "learning_rate": 3.9675056661785556e-05,
-      "loss": 2.3295,
-      "step": 1911
-    },
-    {
-      "epoch": 0.8766620816139385,
-      "grad_norm": 0.49841375909143676,
-      "learning_rate": 3.9385625094097154e-05,
-      "loss": 2.2067,
-      "step": 1912
-    },
-    {
-      "epoch": 0.8771205868867492,
-      "grad_norm": 0.5207496169977238,
-      "learning_rate": 3.909720981350034e-05,
-      "loss": 2.2976,
-      "step": 1913
-    },
-    {
-      "epoch": 0.8775790921595599,
-      "grad_norm": 0.5068821502392448,
-      "learning_rate": 3.880981145634704e-05,
-      "loss": 2.2593,
-      "step": 1914
-    },
-    {
-      "epoch": 0.8780375974323704,
-      "grad_norm": 0.5660325963639762,
-      "learning_rate": 3.852343065674507e-05,
-      "loss": 2.291,
-      "step": 1915
-    },
-    {
-      "epoch": 0.8784961027051811,
-      "grad_norm": 0.5216042549592605,
-      "learning_rate": 3.8238068046557276e-05,
-      "loss": 2.3148,
-      "step": 1916
-    },
-    {
-      "epoch": 0.8789546079779917,
-      "grad_norm": 0.5140261400710355,
-      "learning_rate": 3.795372425540006e-05,
-      "loss": 2.2588,
-      "step": 1917
-    },
-    {
-      "epoch": 0.8794131132508024,
-      "grad_norm": 0.518571310082231,
-      "learning_rate": 3.76703999106418e-05,
-      "loss": 2.2929,
-      "step": 1918
-    },
-    {
-      "epoch": 0.8798716185236131,
-      "grad_norm": 0.4693581594706169,
-      "learning_rate": 3.7388095637401754e-05,
-      "loss": 2.3125,
-      "step": 1919
-    },
-    {
-      "epoch": 0.8803301237964236,
-      "grad_norm": 0.5289697320130735,
-      "learning_rate": 3.7106812058548376e-05,
-      "loss": 2.3446,
-      "step": 1920
-    },
-    {
-      "epoch": 0.8807886290692343,
-      "grad_norm": 0.48740805643299756,
-      "learning_rate": 3.682654979469807e-05,
-      "loss": 2.3056,
-      "step": 1921
-    },
-    {
-      "epoch": 0.8812471343420449,
-      "grad_norm": 0.5220822048937759,
-      "learning_rate": 3.654730946421403e-05,
-      "loss": 2.286,
-      "step": 1922
-    },
-    {
-      "epoch": 0.8817056396148556,
-      "grad_norm": 0.49633736523601835,
-      "learning_rate": 3.6269091683204466e-05,
-      "loss": 2.3351,
-      "step": 1923
-    },
-    {
-      "epoch": 0.8821641448876663,
-      "grad_norm": 0.4992133705725392,
-      "learning_rate": 3.5991897065521693e-05,
-      "loss": 2.3691,
-      "step": 1924
-    },
-    {
-      "epoch": 0.8826226501604768,
-      "grad_norm": 0.5153097201193427,
-      "learning_rate": 3.571572622276026e-05,
-      "loss": 2.2197,
-      "step": 1925
-    },
-    {
-      "epoch": 0.8830811554332875,
-      "grad_norm": 0.5004038741880003,
-      "learning_rate": 3.544057976425619e-05,
-      "loss": 2.2952,
-      "step": 1926
-    },
-    {
-      "epoch": 0.8835396607060981,
-      "grad_norm": 0.5242842945692862,
-      "learning_rate": 3.5166458297085146e-05,
-      "loss": 2.2847,
-      "step": 1927
-    },
-    {
-      "epoch": 0.8839981659789088,
-      "grad_norm": 0.5266468725159641,
-      "learning_rate": 3.489336242606111e-05,
-      "loss": 2.2241,
-      "step": 1928
-    },
-    {
-      "epoch": 0.8844566712517193,
-      "grad_norm": 0.5143837969949387,
-      "learning_rate": 3.462129275373577e-05,
-      "loss": 2.2058,
-      "step": 1929
-    },
-    {
-      "epoch": 0.88491517652453,
-      "grad_norm": 0.5325768711234949,
-      "learning_rate": 3.4350249880395924e-05,
-      "loss": 2.2497,
-      "step": 1930
-    },
-    {
-      "epoch": 0.8853736817973407,
-      "grad_norm": 0.4943160839410224,
-      "learning_rate": 3.408023440406355e-05,
-      "loss": 2.2835,
-      "step": 1931
-    },
-    {
-      "epoch": 0.8858321870701513,
-      "grad_norm": 0.496859895184936,
-      "learning_rate": 3.381124692049331e-05,
-      "loss": 2.2772,
-      "step": 1932
-    },
-    {
-      "epoch": 0.886290692342962,
-      "grad_norm": 0.526740686536386,
-      "learning_rate": 3.354328802317197e-05,
-      "loss": 2.2406,
-      "step": 1933
-    },
-    {
-      "epoch": 0.8867491976157725,
-      "grad_norm": 0.5129598647989948,
-      "learning_rate": 3.327635830331677e-05,
-      "loss": 2.1653,
-      "step": 1934
-    },
-    {
-      "epoch": 0.8872077028885832,
-      "grad_norm": 0.5378867406330848,
-      "learning_rate": 3.3010458349874206e-05,
-      "loss": 2.2747,
-      "step": 1935
-    },
-    {
-      "epoch": 0.8876662081613939,
-      "grad_norm": 0.5359917254843849,
-      "learning_rate": 3.2745588749518775e-05,
-      "loss": 2.2805,
-      "step": 1936
-    },
-    {
-      "epoch": 0.8881247134342045,
-      "grad_norm": 0.5357973794360179,
-      "learning_rate": 3.248175008665161e-05,
-      "loss": 2.3076,
-      "step": 1937
-    },
-    {
-      "epoch": 0.8885832187070152,
-      "grad_norm": 0.5176048181956041,
-      "learning_rate": 3.221894294339911e-05,
-      "loss": 2.3247,
-      "step": 1938
-    },
-    {
-      "epoch": 0.8890417239798257,
-      "grad_norm": 0.6230732008251234,
-      "learning_rate": 3.1957167899611836e-05,
-      "loss": 2.3102,
-      "step": 1939
-    },
-    {
-      "epoch": 0.8895002292526364,
-      "grad_norm": 0.5199602496245991,
-      "learning_rate": 3.169642553286334e-05,
-      "loss": 2.2755,
-      "step": 1940
-    },
-    {
-      "epoch": 0.889958734525447,
-      "grad_norm": 0.48290516896078417,
-      "learning_rate": 3.143671641844831e-05,
-      "loss": 2.2086,
-      "step": 1941
-    },
-    {
-      "epoch": 0.8904172397982577,
-      "grad_norm": 0.4773706504131942,
-      "learning_rate": 3.117804112938205e-05,
-      "loss": 2.2415,
-      "step": 1942
-    },
-    {
-      "epoch": 0.8908757450710684,
-      "grad_norm": 0.5111186983352134,
-      "learning_rate": 3.092040023639869e-05,
-      "loss": 2.2797,
-      "step": 1943
-    },
-    {
-      "epoch": 0.8913342503438789,
-      "grad_norm": 0.5045498253926202,
-      "learning_rate": 3.066379430795002e-05,
-      "loss": 2.2637,
-      "step": 1944
-    },
-    {
-      "epoch": 0.8917927556166896,
-      "grad_norm": 0.46864341142384747,
-      "learning_rate": 3.040822391020459e-05,
-      "loss": 2.1774,
-      "step": 1945
-    },
-    {
-      "epoch": 0.8922512608895002,
-      "grad_norm": 0.49650610002846596,
-      "learning_rate": 3.0153689607045842e-05,
-      "loss": 2.1465,
-      "step": 1946
-    },
-    {
-      "epoch": 0.8927097661623109,
-      "grad_norm": 0.5024622176293131,
-      "learning_rate": 2.9900191960071545e-05,
-      "loss": 2.321,
-      "step": 1947
-    },
-    {
-      "epoch": 0.8931682714351215,
-      "grad_norm": 0.519856751624489,
-      "learning_rate": 2.9647731528591848e-05,
-      "loss": 2.244,
-      "step": 1948
-    },
-    {
-      "epoch": 0.8936267767079321,
-      "grad_norm": 0.49942340529265744,
-      "learning_rate": 2.9396308869628795e-05,
-      "loss": 2.2521,
-      "step": 1949
-    },
-    {
-      "epoch": 0.8940852819807428,
-      "grad_norm": 0.5198442493901192,
-      "learning_rate": 2.914592453791448e-05,
-      "loss": 2.3054,
-      "step": 1950
-    },
-    {
-      "epoch": 0.8945437872535534,
-      "grad_norm": 0.5115738223955972,
-      "learning_rate": 2.8896579085889994e-05,
-      "loss": 2.3506,
-      "step": 1951
-    },
-    {
-      "epoch": 0.8950022925263641,
-      "grad_norm": 0.5062799052045392,
-      "learning_rate": 2.86482730637046e-05,
-      "loss": 2.2783,
-      "step": 1952
-    },
-    {
-      "epoch": 0.8954607977991746,
-      "grad_norm": 0.494740066953936,
-      "learning_rate": 2.840100701921383e-05,
-      "loss": 2.2184,
-      "step": 1953
-    },
-    {
-      "epoch": 0.8959193030719853,
-      "grad_norm": 0.49770272451428266,
-      "learning_rate": 2.8154781497978898e-05,
-      "loss": 2.3572,
-      "step": 1954
-    },
-    {
-      "epoch": 0.896377808344796,
-      "grad_norm": 0.48988604554545284,
-      "learning_rate": 2.7909597043265013e-05,
-      "loss": 2.2189,
-      "step": 1955
-    },
-    {
-      "epoch": 0.8968363136176066,
-      "grad_norm": 0.5079297035935748,
-      "learning_rate": 2.7665454196040662e-05,
-      "loss": 2.1731,
-      "step": 1956
-    },
-    {
-      "epoch": 0.8972948188904173,
-      "grad_norm": 0.49566624250006996,
-      "learning_rate": 2.7422353494975905e-05,
-      "loss": 2.2731,
-      "step": 1957
-    },
-    {
-      "epoch": 0.8977533241632278,
-      "grad_norm": 0.5094451712345172,
-      "learning_rate": 2.7180295476441573e-05,
-      "loss": 2.1862,
-      "step": 1958
-    },
-    {
-      "epoch": 0.8982118294360385,
-      "grad_norm": 0.5210175844960181,
-      "learning_rate": 2.6939280674508016e-05,
-      "loss": 2.3383,
-      "step": 1959
-    },
-    {
-      "epoch": 0.8986703347088492,
-      "grad_norm": 0.5078671127993354,
-      "learning_rate": 2.669930962094358e-05,
-      "loss": 2.2065,
-      "step": 1960
-    },
-    {
-      "epoch": 0.8991288399816598,
-      "grad_norm": 0.49418824234063996,
-      "learning_rate": 2.6460382845214126e-05,
-      "loss": 2.3209,
-      "step": 1961
-    },
-    {
-      "epoch": 0.8995873452544704,
-      "grad_norm": 0.5175732844054391,
-      "learning_rate": 2.6222500874481025e-05,
-      "loss": 2.1442,
-      "step": 1962
-    },
-    {
-      "epoch": 0.900045850527281,
-      "grad_norm": 0.5307175827187469,
-      "learning_rate": 2.5985664233600827e-05,
-      "loss": 2.291,
-      "step": 1963
-    },
-    {
-      "epoch": 0.9005043558000917,
-      "grad_norm": 0.518729119273772,
-      "learning_rate": 2.574987344512336e-05,
-      "loss": 2.1978,
-      "step": 1964
-    },
-    {
-      "epoch": 0.9009628610729024,
-      "grad_norm": 0.5099649653883829,
-      "learning_rate": 2.5515129029290984e-05,
-      "loss": 2.1932,
-      "step": 1965
-    },
-    {
-      "epoch": 0.901421366345713,
-      "grad_norm": 0.5000445374708067,
-      "learning_rate": 2.5281431504037556e-05,
-      "loss": 2.2645,
-      "step": 1966
-    },
-    {
-      "epoch": 0.9018798716185236,
-      "grad_norm": 0.5159944040557487,
-      "learning_rate": 2.504878138498684e-05,
-      "loss": 2.1427,
-      "step": 1967
-    },
-    {
-      "epoch": 0.9023383768913342,
-      "grad_norm": 0.5216327428939966,
-      "learning_rate": 2.48171791854519e-05,
-      "loss": 2.3033,
-      "step": 1968
-    },
-    {
-      "epoch": 0.9027968821641449,
-      "grad_norm": 0.5193788521829803,
-      "learning_rate": 2.4586625416433473e-05,
-      "loss": 2.2885,
-      "step": 1969
-    },
-    {
-      "epoch": 0.9032553874369555,
-      "grad_norm": 0.49932254334805337,
-      "learning_rate": 2.435712058661921e-05,
-      "loss": 2.2504,
-      "step": 1970
-    },
-    {
-      "epoch": 0.9037138927097662,
-      "grad_norm": 0.5228032263182277,
-      "learning_rate": 2.4128665202382327e-05,
-      "loss": 2.2651,
-      "step": 1971
-    },
-    {
-      "epoch": 0.9041723979825768,
-      "grad_norm": 0.5342880076869575,
-      "learning_rate": 2.3901259767780515e-05,
-      "loss": 2.1805,
-      "step": 1972
-    },
-    {
-      "epoch": 0.9046309032553874,
-      "grad_norm": 0.517037648609628,
-      "learning_rate": 2.367490478455514e-05,
-      "loss": 2.2488,
-      "step": 1973
-    },
-    {
-      "epoch": 0.9050894085281981,
-      "grad_norm": 0.5299764828262314,
-      "learning_rate": 2.3449600752129597e-05,
-      "loss": 2.2321,
-      "step": 1974
-    },
-    {
-      "epoch": 0.9055479138010087,
-      "grad_norm": 0.5112711702137571,
-      "learning_rate": 2.3225348167608685e-05,
-      "loss": 2.2814,
-      "step": 1975
-    },
-    {
-      "epoch": 0.9060064190738194,
-      "grad_norm": 0.4927268485009986,
-      "learning_rate": 2.3002147525777118e-05,
-      "loss": 2.1909,
-      "step": 1976
-    },
-    {
-      "epoch": 0.90646492434663,
-      "grad_norm": 0.534093579198666,
-      "learning_rate": 2.2779999319098856e-05,
-      "loss": 2.2316,
-      "step": 1977
-    },
-    {
-      "epoch": 0.9069234296194406,
-      "grad_norm": 0.5209549594467968,
-      "learning_rate": 2.255890403771571e-05,
-      "loss": 2.261,
-      "step": 1978
-    },
-    {
-      "epoch": 0.9073819348922513,
-      "grad_norm": 0.5350295179121762,
-      "learning_rate": 2.233886216944614e-05,
-      "loss": 2.1474,
-      "step": 1979
-    },
-    {
-      "epoch": 0.9078404401650619,
-      "grad_norm": 0.524195607377346,
-      "learning_rate": 2.211987419978484e-05,
-      "loss": 2.2695,
-      "step": 1980
-    },
-    {
-      "epoch": 0.9082989454378725,
-      "grad_norm": 0.5305110910811646,
-      "learning_rate": 2.1901940611900705e-05,
-      "loss": 2.3617,
-      "step": 1981
-    },
-    {
-      "epoch": 0.9087574507106831,
-      "grad_norm": 0.5100149510030066,
-      "learning_rate": 2.168506188663666e-05,
-      "loss": 2.2328,
-      "step": 1982
-    },
-    {
-      "epoch": 0.9092159559834938,
-      "grad_norm": 0.5302396514572292,
-      "learning_rate": 2.1469238502507925e-05,
-      "loss": 2.2918,
-      "step": 1983
-    },
-    {
-      "epoch": 0.9096744612563045,
-      "grad_norm": 0.5409159465227895,
-      "learning_rate": 2.125447093570154e-05,
-      "loss": 2.2732,
-      "step": 1984
-    },
-    {
-      "epoch": 0.9101329665291151,
-      "grad_norm": 0.5166085178307243,
-      "learning_rate": 2.1040759660074793e-05,
-      "loss": 2.1494,
-      "step": 1985
-    },
-    {
-      "epoch": 0.9105914718019257,
-      "grad_norm": 0.5374267355790167,
-      "learning_rate": 2.0828105147154273e-05,
-      "loss": 2.1994,
-      "step": 1986
-    },
-    {
-      "epoch": 0.9110499770747363,
-      "grad_norm": 0.4977550773763996,
-      "learning_rate": 2.061650786613545e-05,
-      "loss": 2.2956,
-      "step": 1987
-    },
-    {
-      "epoch": 0.911508482347547,
-      "grad_norm": 0.5815616099415107,
-      "learning_rate": 2.040596828388058e-05,
-      "loss": 2.2406,
-      "step": 1988
-    },
-    {
-      "epoch": 0.9119669876203577,
-      "grad_norm": 0.5531483387333666,
-      "learning_rate": 2.019648686491865e-05,
-      "loss": 2.1301,
-      "step": 1989
-    },
-    {
-      "epoch": 0.9124254928931683,
-      "grad_norm": 0.5313751852451184,
-      "learning_rate": 1.9988064071443767e-05,
-      "loss": 2.2108,
-      "step": 1990
-    },
-    {
-      "epoch": 0.9128839981659789,
-      "grad_norm": 0.5520204277776866,
-      "learning_rate": 1.9780700363314253e-05,
-      "loss": 2.3305,
-      "step": 1991
-    },
-    {
-      "epoch": 0.9133425034387895,
-      "grad_norm": 0.5267695511780115,
-      "learning_rate": 1.957439619805196e-05,
-      "loss": 2.2511,
-      "step": 1992
-    },
-    {
-      "epoch": 0.9138010087116002,
-      "grad_norm": 0.47998511594946097,
-      "learning_rate": 1.9369152030840554e-05,
-      "loss": 2.2742,
-      "step": 1993
-    },
-    {
-      "epoch": 0.9142595139844109,
-      "grad_norm": 0.5443646797904254,
-      "learning_rate": 1.916496831452552e-05,
-      "loss": 2.3601,
-      "step": 1994
-    },
-    {
-      "epoch": 0.9147180192572214,
-      "grad_norm": 0.5141453940109932,
-      "learning_rate": 1.8961845499611998e-05,
-      "loss": 2.3163,
-      "step": 1995
-    },
-    {
-      "epoch": 0.9151765245300321,
-      "grad_norm": 0.5347284533198813,
-      "learning_rate": 1.8759784034264925e-05,
-      "loss": 2.1786,
-      "step": 1996
-    },
-    {
-      "epoch": 0.9156350298028427,
-      "grad_norm": 0.5231865891558373,
-      "learning_rate": 1.855878436430708e-05,
-      "loss": 2.1871,
-      "step": 1997
-    },
-    {
-      "epoch": 0.9160935350756534,
-      "grad_norm": 0.5103004635099752,
-      "learning_rate": 1.835884693321871e-05,
-      "loss": 2.2899,
-      "step": 1998
-    },
-    {
-      "epoch": 0.916552040348464,
-      "grad_norm": 0.4735595054034461,
-      "learning_rate": 1.8159972182136386e-05,
-      "loss": 2.2732,
-      "step": 1999
-    },
-    {
-      "epoch": 0.9170105456212746,
-      "grad_norm": 0.51805964936498,
-      "learning_rate": 1.7962160549851945e-05,
-      "loss": 2.2438,
-      "step": 2000
-    },
-    {
-      "epoch": 0.9174690508940853,
-      "grad_norm": 0.5261237270589799,
-      "learning_rate": 1.776541247281177e-05,
-      "loss": 2.211,
-      "step": 2001
-    },
-    {
-      "epoch": 0.9179275561668959,
-      "grad_norm": 0.5113187347552365,
-      "learning_rate": 1.7569728385115224e-05,
-      "loss": 2.2599,
-      "step": 2002
-    },
-    {
-      "epoch": 0.9183860614397066,
-      "grad_norm": 0.5456327709329971,
-      "learning_rate": 1.7375108718514665e-05,
-      "loss": 2.2239,
-      "step": 2003
-    },
-    {
-      "epoch": 0.9188445667125172,
-      "grad_norm": 0.5131713553129895,
-      "learning_rate": 1.7181553902413438e-05,
-      "loss": 2.1882,
-      "step": 2004
-    },
-    {
-      "epoch": 0.9193030719853278,
-      "grad_norm": 0.49380465463237183,
-      "learning_rate": 1.698906436386577e-05,
-      "loss": 2.2315,
-      "step": 2005
-    },
-    {
-      "epoch": 0.9197615772581385,
-      "grad_norm": 0.5117814270429939,
-      "learning_rate": 1.679764052757532e-05,
-      "loss": 2.2841,
-      "step": 2006
-    },
-    {
-      "epoch": 0.9202200825309491,
-      "grad_norm": 0.5286359502965893,
-      "learning_rate": 1.6607282815894464e-05,
-      "loss": 2.355,
-      "step": 2007
-    },
-    {
-      "epoch": 0.9206785878037598,
-      "grad_norm": 0.589518735602969,
-      "learning_rate": 1.6417991648823405e-05,
-      "loss": 2.293,
-      "step": 2008
-    },
-    {
-      "epoch": 0.9211370930765704,
-      "grad_norm": 0.525564869894698,
-      "learning_rate": 1.6229767444008835e-05,
-      "loss": 2.23,
-      "step": 2009
-    },
-    {
-      "epoch": 0.921595598349381,
-      "grad_norm": 0.4992631596087589,
-      "learning_rate": 1.604261061674378e-05,
-      "loss": 2.3156,
-      "step": 2010
-    },
-    {
-      "epoch": 0.9220541036221916,
-      "grad_norm": 0.5106951622849519,
-      "learning_rate": 1.5856521579965865e-05,
-      "loss": 2.2193,
-      "step": 2011
-    },
-    {
-      "epoch": 0.9225126088950023,
-      "grad_norm": 0.5141902943732626,
-      "learning_rate": 1.5671500744256938e-05,
-      "loss": 2.242,
-      "step": 2012
-    },
-    {
-      "epoch": 0.922971114167813,
-      "grad_norm": 0.5012787469653646,
-      "learning_rate": 1.5487548517841953e-05,
-      "loss": 2.1969,
-      "step": 2013
-    },
-    {
-      "epoch": 0.9234296194406235,
-      "grad_norm": 0.5431941873566634,
-      "learning_rate": 1.530466530658814e-05,
-      "loss": 2.2655,
-      "step": 2014
-    },
-    {
-      "epoch": 0.9238881247134342,
-      "grad_norm": 0.5230493868111545,
-      "learning_rate": 1.5122851514004054e-05,
-      "loss": 2.1859,
-      "step": 2015
-    },
-    {
-      "epoch": 0.9243466299862448,
-      "grad_norm": 0.511863490784643,
-      "learning_rate": 1.4942107541238703e-05,
-      "loss": 2.2171,
-      "step": 2016
-    },
-    {
-      "epoch": 0.9248051352590555,
-      "grad_norm": 0.5123227171083122,
-      "learning_rate": 1.4762433787080809e-05,
-      "loss": 2.2478,
-      "step": 2017
-    },
-    {
-      "epoch": 0.9252636405318662,
-      "grad_norm": 0.51438710329265,
-      "learning_rate": 1.4583830647957541e-05,
-      "loss": 2.2179,
-      "step": 2018
-    },
-    {
-      "epoch": 0.9257221458046767,
-      "grad_norm": 0.5130901723482908,
-      "learning_rate": 1.4406298517934068e-05,
-      "loss": 2.2542,
-      "step": 2019
-    },
-    {
-      "epoch": 0.9261806510774874,
-      "grad_norm": 0.501274680600378,
-      "learning_rate": 1.4229837788712562e-05,
-      "loss": 2.3112,
-      "step": 2020
-    },
-    {
-      "epoch": 0.926639156350298,
-      "grad_norm": 0.5060227438220692,
-      "learning_rate": 1.4054448849631085e-05,
-      "loss": 2.1789,
-      "step": 2021
-    },
-    {
-      "epoch": 0.9270976616231087,
-      "grad_norm": 0.5466209630509076,
-      "learning_rate": 1.3880132087663145e-05,
-      "loss": 2.249,
-      "step": 2022
-    },
-    {
-      "epoch": 0.9275561668959194,
-      "grad_norm": 0.49870839854426485,
-      "learning_rate": 1.3706887887416419e-05,
-      "loss": 2.249,
-      "step": 2023
-    },
-    {
-      "epoch": 0.9280146721687299,
-      "grad_norm": 0.5422610275554967,
-      "learning_rate": 1.3534716631132316e-05,
-      "loss": 2.2189,
-      "step": 2024
-    },
-    {
-      "epoch": 0.9284731774415406,
-      "grad_norm": 0.5056809957510171,
-      "learning_rate": 1.3363618698684853e-05,
-      "loss": 2.2315,
-      "step": 2025
-    },
-    {
-      "epoch": 0.9289316827143512,
-      "grad_norm": 0.5010440341323672,
-      "learning_rate": 1.3193594467579728e-05,
-      "loss": 2.1848,
-      "step": 2026
-    },
-    {
-      "epoch": 0.9293901879871619,
-      "grad_norm": 0.503084717624002,
-      "learning_rate": 1.3024644312954026e-05,
-      "loss": 2.2964,
-      "step": 2027
-    },
-    {
-      "epoch": 0.9298486932599724,
-      "grad_norm": 0.5155359835035311,
-      "learning_rate": 1.2856768607574564e-05,
-      "loss": 2.2448,
-      "step": 2028
-    },
-    {
-      "epoch": 0.9303071985327831,
-      "grad_norm": 0.494733116200242,
-      "learning_rate": 1.2689967721837947e-05,
-      "loss": 2.1965,
-      "step": 2029
-    },
-    {
-      "epoch": 0.9307657038055938,
-      "grad_norm": 0.5162254212119592,
-      "learning_rate": 1.2524242023769006e-05,
-      "loss": 2.1983,
-      "step": 2030
-    },
-    {
-      "epoch": 0.9312242090784044,
-      "grad_norm": 0.5031994010779061,
-      "learning_rate": 1.2359591879020526e-05,
-      "loss": 2.2651,
-      "step": 2031
-    },
-    {
-      "epoch": 0.9316827143512151,
-      "grad_norm": 0.5032464101841924,
-      "learning_rate": 1.2196017650872081e-05,
-      "loss": 2.3449,
-      "step": 2032
-    },
-    {
-      "epoch": 0.9321412196240256,
-      "grad_norm": 0.5524458660881685,
-      "learning_rate": 1.2033519700229367e-05,
-      "loss": 2.2448,
-      "step": 2033
-    },
-    {
-      "epoch": 0.9325997248968363,
-      "grad_norm": 0.5379975946499309,
-      "learning_rate": 1.1872098385623586e-05,
-      "loss": 2.2183,
-      "step": 2034
-    },
-    {
-      "epoch": 0.933058230169647,
-      "grad_norm": 0.5136154081670692,
-      "learning_rate": 1.1711754063210289e-05,
-      "loss": 2.328,
-      "step": 2035
-    },
-    {
-      "epoch": 0.9335167354424576,
-      "grad_norm": 0.5003473077503876,
-      "learning_rate": 1.155248708676887e-05,
-      "loss": 2.2597,
-      "step": 2036
-    },
-    {
-      "epoch": 0.9339752407152683,
-      "grad_norm": 0.7062697472174687,
-      "learning_rate": 1.1394297807701737e-05,
-      "loss": 2.1385,
-      "step": 2037
-    },
-    {
-      "epoch": 0.9344337459880788,
-      "grad_norm": 0.5190996711455947,
-      "learning_rate": 1.1237186575033254e-05,
-      "loss": 2.199,
-      "step": 2038
-    },
-    {
-      "epoch": 0.9348922512608895,
-      "grad_norm": 0.6225750451625697,
-      "learning_rate": 1.1081153735409522e-05,
-      "loss": 2.2159,
-      "step": 2039
-    },
-    {
-      "epoch": 0.9353507565337001,
-      "grad_norm": 0.513048940590949,
-      "learning_rate": 1.0926199633097156e-05,
-      "loss": 2.2106,
-      "step": 2040
-    },
-    {
-      "epoch": 0.9358092618065108,
-      "grad_norm": 0.7493061356864148,
-      "learning_rate": 1.0772324609982787e-05,
-      "loss": 2.1357,
-      "step": 2041
-    },
-    {
-      "epoch": 0.9362677670793215,
-      "grad_norm": 0.5037856226753217,
-      "learning_rate": 1.0619529005571893e-05,
-      "loss": 2.215,
-      "step": 2042
-    },
-    {
-      "epoch": 0.936726272352132,
-      "grad_norm": 0.4881458792194505,
-      "learning_rate": 1.0467813156988748e-05,
-      "loss": 2.2018,
-      "step": 2043
-    },
-    {
-      "epoch": 0.9371847776249427,
-      "grad_norm": 0.5236379938676736,
-      "learning_rate": 1.0317177398975031e-05,
-      "loss": 2.3144,
-      "step": 2044
-    },
-    {
-      "epoch": 0.9376432828977533,
-      "grad_norm": 0.4791574999893036,
-      "learning_rate": 1.0167622063889326e-05,
-      "loss": 2.1322,
-      "step": 2045
-    },
-    {
-      "epoch": 0.938101788170564,
-      "grad_norm": 0.505451875717103,
-      "learning_rate": 1.0019147481706625e-05,
-      "loss": 2.2308,
-      "step": 2046
-    },
-    {
-      "epoch": 0.9385602934433745,
-      "grad_norm": 0.5325579298513811,
-      "learning_rate": 9.871753980017051e-06,
-      "loss": 2.2203,
-      "step": 2047
-    },
-    {
-      "epoch": 0.9390187987161852,
-      "grad_norm": 0.4991847090745052,
-      "learning_rate": 9.725441884025855e-06,
-      "loss": 2.1964,
-      "step": 2048
-    },
-    {
-      "epoch": 0.9394773039889959,
-      "grad_norm": 0.5012645078665976,
-      "learning_rate": 9.580211516551862e-06,
-      "loss": 2.2281,
-      "step": 2049
-    },
-    {
-      "epoch": 0.9399358092618065,
-      "grad_norm": 0.536787770829371,
-      "learning_rate": 9.436063198027589e-06,
-      "loss": 2.2988,
-      "step": 2050
-    },
-    {
-      "epoch": 0.9403943145346172,
-      "grad_norm": 0.5197382430334139,
-      "learning_rate": 9.292997246497959e-06,
-      "loss": 2.3612,
-      "step": 2051
-    },
-    {
-      "epoch": 0.9408528198074277,
-      "grad_norm": 0.5260517138384843,
-      "learning_rate": 9.151013977619693e-06,
-      "loss": 2.3018,
-      "step": 2052
-    },
-    {
-      "epoch": 0.9413113250802384,
-      "grad_norm": 0.5014265156280914,
-      "learning_rate": 9.010113704661038e-06,
-      "loss": 2.1708,
-      "step": 2053
-    },
-    {
-      "epoch": 0.9417698303530491,
-      "grad_norm": 0.5285851763039264,
-      "learning_rate": 8.870296738500316e-06,
-      "loss": 2.3125,
-      "step": 2054
-    },
-    {
-      "epoch": 0.9422283356258597,
-      "grad_norm": 0.5098176592545269,
-      "learning_rate": 8.731563387626095e-06,
-      "loss": 2.2444,
-      "step": 2055
-    },
-    {
-      "epoch": 0.9426868408986704,
-      "grad_norm": 0.5055157737677624,
-      "learning_rate": 8.59391395813569e-06,
-      "loss": 2.3052,
-      "step": 2056
-    },
-    {
-      "epoch": 0.9431453461714809,
-      "grad_norm": 0.4917603079994781,
-      "learning_rate": 8.457348753735328e-06,
-      "loss": 2.303,
-      "step": 2057
-    },
-    {
-      "epoch": 0.9436038514442916,
-      "grad_norm": 0.491799301750492,
-      "learning_rate": 8.321868075738593e-06,
-      "loss": 2.2721,
-      "step": 2058
-    },
-    {
-      "epoch": 0.9440623567171023,
-      "grad_norm": 0.482131776430265,
-      "learning_rate": 8.187472223066371e-06,
-      "loss": 2.2374,
-      "step": 2059
-    },
-    {
-      "epoch": 0.9445208619899129,
-      "grad_norm": 0.5213869178801601,
-      "learning_rate": 8.054161492246136e-06,
-      "loss": 2.1607,
-      "step": 2060
-    },
-    {
-      "epoch": 0.9449793672627235,
-      "grad_norm": 0.533237980351357,
-      "learning_rate": 7.921936177411049e-06,
-      "loss": 2.2602,
-      "step": 2061
-    },
-    {
-      "epoch": 0.9454378725355341,
-      "grad_norm": 0.5911959618431935,
-      "learning_rate": 7.790796570299463e-06,
-      "loss": 2.2363,
-      "step": 2062
-    },
-    {
-      "epoch": 0.9458963778083448,
-      "grad_norm": 0.49063687915205934,
-      "learning_rate": 7.660742960254207e-06,
-      "loss": 2.2077,
-      "step": 2063
-    },
-    {
-      "epoch": 0.9463548830811555,
-      "grad_norm": 0.5036183304466069,
-      "learning_rate": 7.531775634222138e-06,
-      "loss": 2.178,
-      "step": 2064
-    },
-    {
-      "epoch": 0.9468133883539661,
-      "grad_norm": 0.49947464643153566,
-      "learning_rate": 7.403894876753192e-06,
-      "loss": 2.2137,
-      "step": 2065
-    },
-    {
-      "epoch": 0.9472718936267767,
-      "grad_norm": 0.5080009101756622,
-      "learning_rate": 7.277100970000061e-06,
-      "loss": 2.2315,
-      "step": 2066
-    },
-    {
-      "epoch": 0.9477303988995873,
-      "grad_norm": 0.48645791067995336,
-      "learning_rate": 7.151394193717408e-06,
-      "loss": 2.2676,
-      "step": 2067
-    },
-    {
-      "epoch": 0.948188904172398,
-      "grad_norm": 0.5343421642301939,
-      "learning_rate": 7.026774825261151e-06,
-      "loss": 2.2825,
-      "step": 2068
-    },
-    {
-      "epoch": 0.9486474094452086,
-      "grad_norm": 0.5014323805123012,
-      "learning_rate": 6.903243139588233e-06,
-      "loss": 2.2763,
-      "step": 2069
-    },
-    {
-      "epoch": 0.9491059147180193,
-      "grad_norm": 0.5154310926589517,
-      "learning_rate": 6.780799409255522e-06,
-      "loss": 2.2693,
-      "step": 2070
-    },
-    {
-      "epoch": 0.9495644199908299,
-      "grad_norm": 0.5339186091847842,
-      "learning_rate": 6.659443904419637e-06,
-      "loss": 2.3303,
-      "step": 2071
-    },
-    {
-      "epoch": 0.9500229252636405,
-      "grad_norm": 0.48513418677186954,
-      "learning_rate": 6.539176892836008e-06,
-      "loss": 2.2893,
-      "step": 2072
-    },
-    {
-      "epoch": 0.9504814305364512,
-      "grad_norm": 0.5446527880175672,
-      "learning_rate": 6.4199986398585375e-06,
-      "loss": 2.2056,
-      "step": 2073
-    },
-    {
-      "epoch": 0.9509399358092618,
-      "grad_norm": 0.4882540915804193,
-      "learning_rate": 6.3019094084388884e-06,
-      "loss": 2.2731,
-      "step": 2074
-    },
-    {
-      "epoch": 0.9513984410820725,
-      "grad_norm": 0.5240640906539814,
-      "learning_rate": 6.18490945912592e-06,
-      "loss": 2.2572,
-      "step": 2075
-    },
-    {
-      "epoch": 0.951856946354883,
-      "grad_norm": 0.5087336718973248,
-      "learning_rate": 6.068999050065249e-06,
-      "loss": 2.1672,
-      "step": 2076
-    },
-    {
-      "epoch": 0.9523154516276937,
-      "grad_norm": 0.5109627509111413,
-      "learning_rate": 5.9541784369983586e-06,
-      "loss": 2.1712,
-      "step": 2077
-    },
-    {
-      "epoch": 0.9527739569005044,
-      "grad_norm": 0.5008958592114598,
-      "learning_rate": 5.840447873262433e-06,
-      "loss": 2.1958,
-      "step": 2078
-    },
-    {
-      "epoch": 0.953232462173315,
-      "grad_norm": 0.5200080406907405,
-      "learning_rate": 5.727807609789471e-06,
-      "loss": 2.2051,
-      "step": 2079
-    },
-    {
-      "epoch": 0.9536909674461256,
-      "grad_norm": 0.5508094853914992,
-      "learning_rate": 5.616257895105892e-06,
-      "loss": 2.2829,
-      "step": 2080
-    },
-    {
-      "epoch": 0.9541494727189362,
-      "grad_norm": 0.5134632732718317,
-      "learning_rate": 5.505798975331933e-06,
-      "loss": 2.2014,
-      "step": 2081
-    },
-    {
-      "epoch": 0.9546079779917469,
-      "grad_norm": 0.5128699774900491,
-      "learning_rate": 5.396431094181198e-06,
-      "loss": 2.2456,
-      "step": 2082
-    },
-    {
-      "epoch": 0.9550664832645576,
-      "grad_norm": 0.5421996108500615,
-      "learning_rate": 5.288154492960107e-06,
-      "loss": 2.1953,
-      "step": 2083
-    },
-    {
-      "epoch": 0.9555249885373682,
-      "grad_norm": 0.5267189697994638,
-      "learning_rate": 5.1809694105671155e-06,
-      "loss": 2.3309,
-      "step": 2084
-    },
-    {
-      "epoch": 0.9559834938101788,
-      "grad_norm": 0.514134263295045,
-      "learning_rate": 5.074876083492441e-06,
-      "loss": 2.1512,
-      "step": 2085
-    },
-    {
-      "epoch": 0.9564419990829894,
-      "grad_norm": 0.5172078414047186,
-      "learning_rate": 4.96987474581767e-06,
-      "loss": 2.2118,
-      "step": 2086
-    },
-    {
-      "epoch": 0.9569005043558001,
-      "grad_norm": 0.5203904176985656,
-      "learning_rate": 4.865965629214819e-06,
-      "loss": 2.3104,
-      "step": 2087
-    },
-    {
-      "epoch": 0.9573590096286108,
-      "grad_norm": 0.5189236899071782,
-      "learning_rate": 4.763148962946218e-06,
-      "loss": 2.1321,
-      "step": 2088
-    },
-    {
-      "epoch": 0.9578175149014214,
-      "grad_norm": 0.531179658575282,
-      "learning_rate": 4.661424973863681e-06,
-      "loss": 2.2656,
-      "step": 2089
-    },
-    {
-      "epoch": 0.958276020174232,
-      "grad_norm": 0.533911292166662,
-      "learning_rate": 4.560793886408398e-06,
-      "loss": 2.2905,
-      "step": 2090
-    },
-    {
-      "epoch": 0.9587345254470426,
-      "grad_norm": 0.5028530874990772,
-      "learning_rate": 4.461255922609986e-06,
-      "loss": 2.2754,
-      "step": 2091
-    },
-    {
-      "epoch": 0.9591930307198533,
-      "grad_norm": 0.48670078330246036,
-      "learning_rate": 4.362811302086267e-06,
-      "loss": 2.2496,
-      "step": 2092
-    },
-    {
-      "epoch": 0.959651535992664,
-      "grad_norm": 0.4958732703577284,
-      "learning_rate": 4.265460242042885e-06,
-      "loss": 2.1309,
-      "step": 2093
-    },
-    {
-      "epoch": 0.9601100412654745,
-      "grad_norm": 0.5253916054844128,
-      "learning_rate": 4.169202957272522e-06,
-      "loss": 2.2034,
-      "step": 2094
-    },
-    {
-      "epoch": 0.9605685465382852,
-      "grad_norm": 0.5068801544081865,
-      "learning_rate": 4.074039660154738e-06,
-      "loss": 2.2603,
-      "step": 2095
-    },
-    {
-      "epoch": 0.9610270518110958,
-      "grad_norm": 0.5154803647353869,
-      "learning_rate": 3.9799705606551325e-06,
-      "loss": 2.2693,
-      "step": 2096
-    },
-    {
-      "epoch": 0.9614855570839065,
-      "grad_norm": 0.5099619627993038,
-      "learning_rate": 3.886995866325294e-06,
-      "loss": 2.299,
-      "step": 2097
-    },
-    {
-      "epoch": 0.9619440623567171,
-      "grad_norm": 0.522574162246845,
-      "learning_rate": 3.795115782302072e-06,
-      "loss": 2.2754,
-      "step": 2098
-    },
-    {
-      "epoch": 0.9624025676295277,
-      "grad_norm": 0.532803703750089,
-      "learning_rate": 3.704330511307197e-06,
-      "loss": 2.2728,
-      "step": 2099
-    },
-    {
-      "epoch": 0.9628610729023384,
-      "grad_norm": 0.503374375343764,
-      "learning_rate": 3.614640253646828e-06,
-      "loss": 2.1984,
-      "step": 2100
-    },
-    {
-      "epoch": 0.963319578175149,
-      "grad_norm": 0.4963114267472325,
-      "learning_rate": 3.5260452072110594e-06,
-      "loss": 2.257,
-      "step": 2101
-    },
-    {
-      "epoch": 0.9637780834479597,
-      "grad_norm": 0.5320721940496795,
-      "learning_rate": 3.4385455674737498e-06,
-      "loss": 2.2542,
-      "step": 2102
-    },
-    {
-      "epoch": 0.9642365887207703,
-      "grad_norm": 0.535013735026119,
-      "learning_rate": 3.3521415274915256e-06,
-      "loss": 2.2721,
-      "step": 2103
-    },
-    {
-      "epoch": 0.9646950939935809,
-      "grad_norm": 0.5277708838189081,
-      "learning_rate": 3.2668332779041133e-06,
-      "loss": 2.2405,
-      "step": 2104
-    },
-    {
-      "epoch": 0.9651535992663915,
-      "grad_norm": 0.4990997133683891,
-      "learning_rate": 3.1826210069332838e-06,
-      "loss": 2.2025,
-      "step": 2105
-    },
-    {
-      "epoch": 0.9656121045392022,
-      "grad_norm": 0.6684755259366332,
-      "learning_rate": 3.0995049003826324e-06,
-      "loss": 2.2754,
-      "step": 2106
-    },
-    {
-      "epoch": 0.9660706098120129,
-      "grad_norm": 0.5180861910188997,
-      "learning_rate": 3.017485141637355e-06,
-      "loss": 2.2023,
-      "step": 2107
-    },
-    {
-      "epoch": 0.9665291150848234,
-      "grad_norm": 0.5682713642832486,
-      "learning_rate": 2.9365619116636376e-06,
-      "loss": 2.3051,
-      "step": 2108
-    },
-    {
-      "epoch": 0.9669876203576341,
-      "grad_norm": 0.5110699331889148,
-      "learning_rate": 2.856735389008269e-06,
-      "loss": 2.2314,
-      "step": 2109
-    },
-    {
-      "epoch": 0.9674461256304447,
-      "grad_norm": 0.5849160109877934,
-      "learning_rate": 2.778005749798307e-06,
-      "loss": 2.2072,
-      "step": 2110
-    },
-    {
-      "epoch": 0.9679046309032554,
-      "grad_norm": 0.5294564121508625,
-      "learning_rate": 2.700373167740744e-06,
-      "loss": 2.2527,
-      "step": 2111
-    },
-    {
-      "epoch": 0.9683631361760661,
-      "grad_norm": 0.5150770315511337,
-      "learning_rate": 2.62383781412201e-06,
-      "loss": 2.1806,
-      "step": 2112
-    },
-    {
-      "epoch": 0.9688216414488766,
-      "grad_norm": 0.5798446431435137,
-      "learning_rate": 2.5483998578076373e-06,
-      "loss": 2.2707,
-      "step": 2113
-    },
-    {
-      "epoch": 0.9692801467216873,
-      "grad_norm": 0.4934827286074145,
-      "learning_rate": 2.4740594652418736e-06,
-      "loss": 2.2677,
-      "step": 2114
-    },
-    {
-      "epoch": 0.9697386519944979,
-      "grad_norm": 0.49023961756855244,
-      "learning_rate": 2.4008168004472917e-06,
-      "loss": 2.1883,
-      "step": 2115
-    },
-    {
-      "epoch": 0.9701971572673086,
-      "grad_norm": 0.512578058685763,
-      "learning_rate": 2.3286720250246253e-06,
-      "loss": 2.2577,
-      "step": 2116
-    },
-    {
-      "epoch": 0.9706556625401193,
-      "grad_norm": 0.49536020077633036,
-      "learning_rate": 2.2576252981520994e-06,
-      "loss": 2.2225,
-      "step": 2117
-    },
-    {
-      "epoch": 0.9711141678129298,
-      "grad_norm": 0.512731799916307,
-      "learning_rate": 2.1876767765853233e-06,
-      "loss": 2.2351,
-      "step": 2118
-    },
-    {
-      "epoch": 0.9715726730857405,
-      "grad_norm": 0.5038311702774062,
-      "learning_rate": 2.118826614656788e-06,
-      "loss": 2.1852,
-      "step": 2119
-    },
-    {
-      "epoch": 0.9720311783585511,
-      "grad_norm": 0.5251241973623888,
-      "learning_rate": 2.051074964275701e-06,
-      "loss": 2.1705,
-      "step": 2120
-    },
-    {
-      "epoch": 0.9724896836313618,
-      "grad_norm": 0.49538755699946896,
-      "learning_rate": 1.984421974927375e-06,
-      "loss": 2.1981,
-      "step": 2121
-    },
-    {
-      "epoch": 0.9729481889041725,
-      "grad_norm": 0.5208183448343481,
-      "learning_rate": 1.9188677936731734e-06,
-      "loss": 2.2312,
-      "step": 2122
-    },
-    {
-      "epoch": 0.973406694176983,
-      "grad_norm": 0.5168265638515198,
-      "learning_rate": 1.8544125651501208e-06,
-      "loss": 2.2725,
-      "step": 2123
-    },
-    {
-      "epoch": 0.9738651994497937,
-      "grad_norm": 0.5093185530655802,
-      "learning_rate": 1.7910564315704035e-06,
-      "loss": 2.1933,
-      "step": 2124
-    },
-    {
-      "epoch": 0.9743237047226043,
-      "grad_norm": 0.5432235285971039,
-      "learning_rate": 1.7287995327214257e-06,
-      "loss": 2.3548,
-      "step": 2125
-    },
-    {
-      "epoch": 0.974782209995415,
-      "grad_norm": 0.5502255056961028,
-      "learning_rate": 1.6676420059649754e-06,
-      "loss": 2.3629,
-      "step": 2126
-    },
-    {
-      "epoch": 0.9752407152682255,
-      "grad_norm": 0.49782959251482944,
-      "learning_rate": 1.6075839862374485e-06,
-      "loss": 2.1715,
-      "step": 2127
-    },
-    {
-      "epoch": 0.9756992205410362,
-      "grad_norm": 0.5172148272963718,
-      "learning_rate": 1.5486256060492366e-06,
-      "loss": 2.3354,
-      "step": 2128
-    },
-    {
-      "epoch": 0.9761577258138469,
-      "grad_norm": 0.5148912802173865,
-      "learning_rate": 1.4907669954844495e-06,
-      "loss": 2.1885,
-      "step": 2129
-    },
-    {
-      "epoch": 0.9766162310866575,
-      "grad_norm": 0.5128009535172008,
-      "learning_rate": 1.434008282200805e-06,
-      "loss": 2.3103,
-      "step": 2130
-    },
-    {
-      "epoch": 0.9770747363594682,
-      "grad_norm": 0.5010728206793448,
-      "learning_rate": 1.3783495914291844e-06,
-      "loss": 2.2376,
-      "step": 2131
-    },
-    {
-      "epoch": 0.9775332416322787,
-      "grad_norm": 0.5345866659699305,
-      "learning_rate": 1.3237910459734104e-06,
-      "loss": 2.3078,
-      "step": 2132
-    },
-    {
-      "epoch": 0.9779917469050894,
-      "grad_norm": 0.5115918387036305,
-      "learning_rate": 1.270332766210025e-06,
-      "loss": 2.2611,
-      "step": 2133
-    },
-    {
-      "epoch": 0.9784502521779,
-      "grad_norm": 0.5257374142349257,
-      "learning_rate": 1.2179748700879012e-06,
-      "loss": 2.2635,
-      "step": 2134
-    },
-    {
-      "epoch": 0.9789087574507107,
-      "grad_norm": 0.5201385939322453,
-      "learning_rate": 1.1667174731280205e-06,
-      "loss": 2.2458,
-      "step": 2135
-    },
-    {
-      "epoch": 0.9793672627235214,
-      "grad_norm": 0.5530558864621438,
-      "learning_rate": 1.1165606884234182e-06,
-      "loss": 2.1993,
-      "step": 2136
-    },
-    {
-      "epoch": 0.9798257679963319,
-      "grad_norm": 0.5283407753367914,
-      "learning_rate": 1.0675046266386268e-06,
-      "loss": 2.2407,
-      "step": 2137
-    },
-    {
-      "epoch": 0.9802842732691426,
-      "grad_norm": 0.5177492279510663,
-      "learning_rate": 1.019549396009567e-06,
-      "loss": 2.2912,
-      "step": 2138
-    },
-    {
-      "epoch": 0.9807427785419532,
-      "grad_norm": 0.5004194286716147,
-      "learning_rate": 9.726951023434348e-07,
-      "loss": 2.314,
-      "step": 2139
-    },
-    {
-      "epoch": 0.9812012838147639,
-      "grad_norm": 0.5263430024979978,
-      "learning_rate": 9.269418490182591e-07,
-      "loss": 2.2857,
-      "step": 2140
-    },
-    {
-      "epoch": 0.9816597890875745,
-      "grad_norm": 0.5250569160333022,
-      "learning_rate": 8.822897369827332e-07,
-      "loss": 2.2898,
-      "step": 2141
-    },
-    {
-      "epoch": 0.9821182943603851,
-      "grad_norm": 0.5115450922033579,
-      "learning_rate": 8.387388647561611e-07,
-      "loss": 2.1989,
-      "step": 2142
-    },
-    {
-      "epoch": 0.9825767996331958,
-      "grad_norm": 0.5156315253126624,
-      "learning_rate": 7.962893284279016e-07,
-      "loss": 2.2679,
-      "step": 2143
-    },
-    {
-      "epoch": 0.9830353049060064,
-      "grad_norm": 0.5091837303661778,
-      "learning_rate": 7.549412216574791e-07,
-      "loss": 2.2655,
-      "step": 2144
-    },
-    {
-      "epoch": 0.9834938101788171,
-      "grad_norm": 0.49484654959269747,
-      "learning_rate": 7.146946356743067e-07,
-      "loss": 2.2091,
-      "step": 2145
-    },
-    {
-      "epoch": 0.9839523154516276,
-      "grad_norm": 0.5012984543004315,
-      "learning_rate": 6.755496592773524e-07,
-      "loss": 2.2089,
-      "step": 2146
-    },
-    {
-      "epoch": 0.9844108207244383,
-      "grad_norm": 0.5109535940168121,
-      "learning_rate": 6.375063788349733e-07,
-      "loss": 2.2111,
-      "step": 2147
-    },
-    {
-      "epoch": 0.984869325997249,
-      "grad_norm": 0.4847054464710344,
-      "learning_rate": 6.005648782848594e-07,
-      "loss": 2.1209,
-      "step": 2148
-    },
-    {
-      "epoch": 0.9853278312700596,
-      "grad_norm": 0.523851618935093,
-      "learning_rate": 5.647252391337565e-07,
-      "loss": 2.2696,
-      "step": 2149
-    },
-    {
-      "epoch": 0.9857863365428703,
-      "grad_norm": 0.4984931323285539,
-      "learning_rate": 5.299875404572441e-07,
-      "loss": 2.2008,
-      "step": 2150
-    },
-    {
-      "epoch": 0.9862448418156808,
-      "grad_norm": 0.5004780955557484,
-      "learning_rate": 4.963518588996796e-07,
-      "loss": 2.2982,
-      "step": 2151
-    },
-    {
-      "epoch": 0.9867033470884915,
-      "grad_norm": 0.4984825596446015,
-      "learning_rate": 4.638182686738657e-07,
-      "loss": 2.2703,
-      "step": 2152
-    },
-    {
-      "epoch": 0.9871618523613022,
-      "grad_norm": 0.5044095652445582,
-      "learning_rate": 4.3238684156110543e-07,
-      "loss": 2.2916,
-      "step": 2153
-    },
-    {
-      "epoch": 0.9876203576341128,
-      "grad_norm": 0.48332130847548654,
-      "learning_rate": 4.020576469108139e-07,
-      "loss": 2.2613,
-      "step": 2154
-    },
-    {
-      "epoch": 0.9880788629069235,
-      "grad_norm": 0.5121533106132787,
-      "learning_rate": 3.7283075164046274e-07,
-      "loss": 2.1839,
-      "step": 2155
-    },
-    {
-      "epoch": 0.988537368179734,
-      "grad_norm": 0.5082694679787694,
-      "learning_rate": 3.4470622023557995e-07,
-      "loss": 2.1638,
-      "step": 2156
-    },
-    {
-      "epoch": 0.9889958734525447,
-      "grad_norm": 0.49850277860940745,
-      "learning_rate": 3.176841147492504e-07,
-      "loss": 2.1726,
-      "step": 2157
-    },
-    {
-      "epoch": 0.9894543787253554,
-      "grad_norm": 0.49289136419378393,
-      "learning_rate": 2.9176449480244895e-07,
-      "loss": 2.2554,
-      "step": 2158
-    },
-    {
-      "epoch": 0.989912883998166,
-      "grad_norm": 0.5003012807912481,
-      "learning_rate": 2.6694741758342967e-07,
-      "loss": 2.1866,
-      "step": 2159
-    },
-    {
-      "epoch": 0.9903713892709766,
-      "grad_norm": 0.49231310170877096,
-      "learning_rate": 2.432329378478926e-07,
-      "loss": 2.2289,
-      "step": 2160
-    },
-    {
-      "epoch": 0.9908298945437872,
-      "grad_norm": 0.5385852444696893,
-      "learning_rate": 2.2062110791892798e-07,
-      "loss": 2.2578,
-      "step": 2161
-    },
-    {
-      "epoch": 0.9912883998165979,
-      "grad_norm": 0.5125898611522903,
-      "learning_rate": 1.9911197768662792e-07,
-      "loss": 2.2947,
-      "step": 2162
-    },
-    {
-      "epoch": 0.9917469050894085,
-      "grad_norm": 0.522793720226793,
-      "learning_rate": 1.7870559460814173e-07,
-      "loss": 2.2979,
-      "step": 2163
-    },
-    {
-      "epoch": 0.9922054103622192,
-      "grad_norm": 0.494346811226961,
-      "learning_rate": 1.5940200370750947e-07,
-      "loss": 2.1912,
-      "step": 2164
-    },
-    {
-      "epoch": 0.9926639156350298,
-      "grad_norm": 0.5190983524721302,
-      "learning_rate": 1.4120124757577291e-07,
-      "loss": 2.1855,
-      "step": 2165
-    },
-    {
-      "epoch": 0.9931224209078404,
-      "grad_norm": 0.4970171027285987,
-      "learning_rate": 1.2410336637047603e-07,
-      "loss": 2.3172,
-      "step": 2166
-    },
-    {
-      "epoch": 0.9935809261806511,
-      "grad_norm": 0.49969641821637195,
-      "learning_rate": 1.081083978159425e-07,
-      "loss": 2.2389,
-      "step": 2167
-    },
-    {
-      "epoch": 0.9940394314534617,
-      "grad_norm": 0.5297864944687216,
-      "learning_rate": 9.321637720310915e-08,
-      "loss": 2.2423,
-      "step": 2168
-    },
-    {
-      "epoch": 0.9944979367262724,
-      "grad_norm": 0.5059231106730568,
-      "learning_rate": 7.942733738924845e-08,
-      "loss": 2.2521,
-      "step": 2169
-    },
-    {
-      "epoch": 0.994956441999083,
-      "grad_norm": 0.4935679041612263,
-      "learning_rate": 6.6741308798135e-08,
-      "loss": 2.3631,
-      "step": 2170
-    },
-    {
-      "epoch": 0.9954149472718936,
-      "grad_norm": 0.5167095336504519,
-      "learning_rate": 5.5158319419934546e-08,
-      "loss": 2.1866,
-      "step": 2171
-    },
-    {
-      "epoch": 0.9958734525447043,
-      "grad_norm": 0.5366491467915796,
-      "learning_rate": 4.4678394810981906e-08,
-      "loss": 2.1805,
-      "step": 2172
-    },
-    {
-      "epoch": 0.9963319578175149,
-      "grad_norm": 0.5033517749143199,
-      "learning_rate": 3.5301558093947527e-08,
-      "loss": 2.3386,
-      "step": 2173
-    },
-    {
-      "epoch": 0.9967904630903255,
-      "grad_norm": 0.5280228293342136,
-      "learning_rate": 2.7027829957559925e-08,
-      "loss": 2.261,
-      "step": 2174
-    },
-    {
-      "epoch": 0.9972489683631361,
-      "grad_norm": 0.5149366347274347,
-      "learning_rate": 1.985722865682771e-08,
-      "loss": 2.2349,
-      "step": 2175
-    },
-    {
-      "epoch": 0.9977074736359468,
-      "grad_norm": 0.5043324095755742,
-      "learning_rate": 1.3789770012762048e-08,
-      "loss": 2.1956,
-      "step": 2176
-    },
-    {
-      "epoch": 0.9981659789087575,
-      "grad_norm": 0.5098821410483256,
-      "learning_rate": 8.825467412376665e-09,
-      "loss": 2.2555,
-      "step": 2177
-    },
-    {
-      "epoch": 0.9986244841815681,
-      "grad_norm": 0.5328397337848737,
-      "learning_rate": 4.96433180879885e-09,
-      "loss": 2.2323,
-      "step": 2178
-    },
-    {
-      "epoch": 0.9990829894543787,
-      "grad_norm": 0.5414030161244278,
-      "learning_rate": 2.206371721158451e-09,
-      "loss": 2.296,
-      "step": 2179
-    },
-    {
-      "epoch": 0.9995414947271893,
-      "grad_norm": 0.5091179162027744,
-      "learning_rate": 5.515932345323549e-10,
-      "loss": 2.2578,
-      "step": 2180
-    },
-    {
-      "epoch": 1.0,
-      "grad_norm": 0.4948074344843498,
-      "learning_rate": 0.0,
-      "loss": 2.3178,
-      "step": 2181
-    },
-    {
-      "epoch": 1.0,
-      "step": 2181,
-      "total_flos": 4.315341437722952e+17,
-      "train_loss": 2.419363842815065,
-      "train_runtime": 7017.9167,
-      "train_samples_per_second": 79.529,
-      "train_steps_per_second": 0.311
-    }
-  ],
-  "logging_steps": 1.0,
-  "max_steps": 2181,
-  "num_input_tokens_seen": 0,
-  "num_train_epochs": 1,
-  "save_steps": 24000,
-  "stateful_callbacks": {
-    "TrainerControl": {
-      "args": {
-        "should_epoch_stop": false,
-        "should_evaluate": false,
-        "should_log": false,
-        "should_save": false,
-        "should_training_stop": false
-      },
-      "attributes": {}
-    }
-  },
-  "total_flos": 4.315341437722952e+17,
-  "train_batch_size": 32,
-  "trial_name": null,
-  "trial_params": null
-}