diff --git "a/checkpoint-456/trainer_state.json" "b/checkpoint-456/trainer_state.json"
--- "a/checkpoint-456/trainer_state.json"
+++ "b/checkpoint-456/trainer_state.json"
@@ -46,3159 +46,3159 @@
     },
     {
       "epoch": 0.03956043956043956,
-      "grad_norm": 1.9752649068832397,
+      "grad_norm": 1.97528076171875,
       "learning_rate": 0.0002,
       "loss": 1.7334,
       "step": 6
     },
     {
       "epoch": 0.046153846153846156,
-      "grad_norm": 2.564279317855835,
+      "grad_norm": 2.583857297897339,
       "learning_rate": 0.00019955654101995565,
-      "loss": 1.8931,
+      "loss": 1.8924,
       "step": 7
     },
     {
       "epoch": 0.05274725274725275,
-      "grad_norm": 1.9491405487060547,
+      "grad_norm": 1.9402282238006592,
       "learning_rate": 0.00019911308203991133,
-      "loss": 1.2738,
+      "loss": 1.273,
       "step": 8
     },
     {
       "epoch": 0.05934065934065934,
-      "grad_norm": 1.026367425918579,
+      "grad_norm": 1.030725359916687,
       "learning_rate": 0.00019866962305986697,
-      "loss": 1.33,
+      "loss": 1.3303,
       "step": 9
     },
     {
       "epoch": 0.06593406593406594,
-      "grad_norm": 1.3561840057373047,
+      "grad_norm": 1.3545081615447998,
       "learning_rate": 0.00019822616407982261,
-      "loss": 1.3474,
+      "loss": 1.3464,
       "step": 10
     },
     {
       "epoch": 0.07252747252747253,
-      "grad_norm": 0.9841963648796082,
+      "grad_norm": 0.9670407176017761,
       "learning_rate": 0.00019778270509977829,
-      "loss": 1.26,
+      "loss": 1.2525,
       "step": 11
     },
     {
       "epoch": 0.07912087912087912,
-      "grad_norm": 0.7599233984947205,
+      "grad_norm": 0.7464919090270996,
       "learning_rate": 0.00019733924611973393,
-      "loss": 1.1589,
+      "loss": 1.1564,
       "step": 12
     },
     {
       "epoch": 0.08571428571428572,
-      "grad_norm": 0.5863561630249023,
+      "grad_norm": 0.5867994427680969,
       "learning_rate": 0.0001968957871396896,
-      "loss": 1.0918,
+      "loss": 1.0917,
       "step": 13
     },
     {
       "epoch": 0.09230769230769231,
-      "grad_norm": 0.614296019077301,
+      "grad_norm": 0.6113730669021606,
       "learning_rate": 0.00019645232815964525,
-      "loss": 1.1588,
+      "loss": 1.1594,
       "step": 14
     },
     {
       "epoch": 0.0989010989010989,
-      "grad_norm": 0.46234244108200073,
+      "grad_norm": 0.4500170052051544,
       "learning_rate": 0.00019600886917960092,
-      "loss": 1.0832,
+      "loss": 1.0796,
       "step": 15
     },
     {
       "epoch": 0.1054945054945055,
-      "grad_norm": 0.5534088015556335,
+      "grad_norm": 0.5538170337677002,
       "learning_rate": 0.00019556541019955653,
-      "loss": 1.1643,
+      "loss": 1.1618,
       "step": 16
     },
     {
       "epoch": 0.11208791208791209,
-      "grad_norm": 0.45435506105422974,
+      "grad_norm": 0.45620211958885193,
       "learning_rate": 0.0001951219512195122,
-      "loss": 1.1515,
+      "loss": 1.1508,
       "step": 17
     },
     {
       "epoch": 0.11868131868131868,
-      "grad_norm": 0.4777897894382477,
+      "grad_norm": 0.4772197902202606,
       "learning_rate": 0.00019467849223946785,
-      "loss": 1.062,
+      "loss": 1.061,
       "step": 18
     },
     {
       "epoch": 0.12527472527472527,
-      "grad_norm": 0.525127649307251,
+      "grad_norm": 0.533805787563324,
       "learning_rate": 0.00019423503325942352,
-      "loss": 1.1474,
+      "loss": 1.1468,
       "step": 19
     },
     {
       "epoch": 0.13186813186813187,
-      "grad_norm": 0.413349986076355,
+      "grad_norm": 0.4162095785140991,
       "learning_rate": 0.00019379157427937917,
-      "loss": 0.9223,
+      "loss": 0.9207,
       "step": 20
     },
     {
       "epoch": 0.13846153846153847,
-      "grad_norm": 0.30619320273399353,
+      "grad_norm": 0.30745798349380493,
       "learning_rate": 0.00019334811529933484,
-      "loss": 0.8226,
+      "loss": 0.8223,
       "step": 21
     },
     {
       "epoch": 0.14505494505494507,
-      "grad_norm": 0.30665796995162964,
+      "grad_norm": 0.29989153146743774,
       "learning_rate": 0.00019290465631929045,
-      "loss": 1.063,
+      "loss": 1.0635,
       "step": 22
     },
     {
       "epoch": 0.15164835164835164,
-      "grad_norm": 0.4375471770763397,
+      "grad_norm": 0.4353378415107727,
       "learning_rate": 0.00019246119733924613,
-      "loss": 1.1137,
+      "loss": 1.1136,
       "step": 23
     },
     {
       "epoch": 0.15824175824175823,
-      "grad_norm": 0.3059830069541931,
+      "grad_norm": 0.30720826983451843,
       "learning_rate": 0.00019201773835920177,
-      "loss": 1.0795,
+      "loss": 1.0796,
       "step": 24
     },
     {
       "epoch": 0.16483516483516483,
-      "grad_norm": 0.303276389837265,
+      "grad_norm": 0.30275797843933105,
       "learning_rate": 0.00019157427937915744,
-      "loss": 0.9357,
+      "loss": 0.9368,
       "step": 25
     },
     {
       "epoch": 0.17142857142857143,
-      "grad_norm": 0.30064496397972107,
+      "grad_norm": 0.3034687042236328,
       "learning_rate": 0.00019113082039911309,
-      "loss": 0.9608,
+      "loss": 0.9616,
       "step": 26
     },
     {
       "epoch": 0.17802197802197803,
-      "grad_norm": 0.3118164539337158,
+      "grad_norm": 0.3129018545150757,
       "learning_rate": 0.00019068736141906876,
-      "loss": 0.976,
+      "loss": 0.9762,
       "step": 27
     },
     {
       "epoch": 0.18461538461538463,
-      "grad_norm": 0.32316967844963074,
+      "grad_norm": 0.32465481758117676,
       "learning_rate": 0.0001902439024390244,
-      "loss": 1.0248,
+      "loss": 1.025,
       "step": 28
     },
     {
       "epoch": 0.1912087912087912,
-      "grad_norm": 0.38039305806159973,
+      "grad_norm": 0.3805822730064392,
       "learning_rate": 0.00018980044345898005,
-      "loss": 1.0935,
+      "loss": 1.0966,
       "step": 29
     },
     {
       "epoch": 0.1978021978021978,
-      "grad_norm": 0.3035508692264557,
+      "grad_norm": 0.30361831188201904,
       "learning_rate": 0.00018935698447893572,
-      "loss": 1.1209,
+      "loss": 1.1199,
       "step": 30
     },
     {
       "epoch": 0.2043956043956044,
-      "grad_norm": 0.3552742600440979,
+      "grad_norm": 0.3538556694984436,
       "learning_rate": 0.00018891352549889136,
-      "loss": 0.8599,
+      "loss": 0.8606,
       "step": 31
     },
     {
       "epoch": 0.210989010989011,
-      "grad_norm": 0.4854717254638672,
+      "grad_norm": 0.4855636954307556,
       "learning_rate": 0.00018847006651884703,
-      "loss": 1.2983,
+      "loss": 1.3002,
       "step": 32
     },
     {
       "epoch": 0.2175824175824176,
-      "grad_norm": 0.3681517541408539,
+      "grad_norm": 0.36801108717918396,
       "learning_rate": 0.00018802660753880268,
       "loss": 1.1441,
       "step": 33
     },
     {
       "epoch": 0.22417582417582418,
-      "grad_norm": 0.303026407957077,
+      "grad_norm": 0.3000083565711975,
       "learning_rate": 0.00018758314855875832,
-      "loss": 1.0872,
+      "loss": 1.0868,
       "step": 34
     },
     {
       "epoch": 0.23076923076923078,
-      "grad_norm": 0.3582163155078888,
+      "grad_norm": 0.35833317041397095,
       "learning_rate": 0.00018713968957871397,
-      "loss": 0.981,
+      "loss": 0.9805,
       "step": 35
     },
     {
       "epoch": 0.23736263736263735,
-      "grad_norm": 0.27061930298805237,
+      "grad_norm": 0.26648440957069397,
       "learning_rate": 0.00018669623059866964,
-      "loss": 0.9192,
+      "loss": 0.9186,
       "step": 36
     },
     {
       "epoch": 0.24395604395604395,
-      "grad_norm": 0.29702866077423096,
+      "grad_norm": 0.29637497663497925,
       "learning_rate": 0.00018625277161862528,
       "loss": 1.0078,
       "step": 37
     },
     {
       "epoch": 0.25054945054945055,
-      "grad_norm": 0.37023022770881653,
+      "grad_norm": 0.3601683974266052,
       "learning_rate": 0.00018580931263858095,
-      "loss": 1.0047,
+      "loss": 1.0045,
       "step": 38
     },
     {
       "epoch": 0.2571428571428571,
-      "grad_norm": 0.3519829511642456,
+      "grad_norm": 0.3521329462528229,
       "learning_rate": 0.0001853658536585366,
-      "loss": 0.999,
+      "loss": 1.0004,
       "step": 39
     },
     {
       "epoch": 0.26373626373626374,
-      "grad_norm": 0.2641372084617615,
+      "grad_norm": 0.2667492628097534,
       "learning_rate": 0.00018492239467849224,
-      "loss": 0.9269,
+      "loss": 0.9271,
       "step": 40
     },
     {
       "epoch": 0.2703296703296703,
-      "grad_norm": 0.3424227833747864,
+      "grad_norm": 0.33176255226135254,
       "learning_rate": 0.0001844789356984479,
-      "loss": 1.0748,
+      "loss": 1.0735,
       "step": 41
     },
     {
       "epoch": 0.27692307692307694,
-      "grad_norm": 0.3919481039047241,
+      "grad_norm": 0.3940621018409729,
       "learning_rate": 0.00018403547671840356,
-      "loss": 1.1445,
+      "loss": 1.1447,
       "step": 42
     },
     {
       "epoch": 0.2835164835164835,
-      "grad_norm": 0.3040602207183838,
+      "grad_norm": 0.3026503324508667,
       "learning_rate": 0.0001835920177383592,
-      "loss": 0.8869,
+      "loss": 0.8857,
       "step": 43
     },
     {
       "epoch": 0.29010989010989013,
-      "grad_norm": 0.2948625683784485,
+      "grad_norm": 0.2967509627342224,
       "learning_rate": 0.00018314855875831487,
-      "loss": 1.0821,
+      "loss": 1.0834,
       "step": 44
     },
     {
       "epoch": 0.2967032967032967,
-      "grad_norm": 0.2821897566318512,
+      "grad_norm": 0.2822628319263458,
       "learning_rate": 0.00018270509977827052,
-      "loss": 0.9714,
+      "loss": 0.9712,
       "step": 45
     },
     {
       "epoch": 0.3032967032967033,
-      "grad_norm": 0.3602289855480194,
+      "grad_norm": 0.36211535334587097,
       "learning_rate": 0.00018226164079822616,
-      "loss": 1.0296,
+      "loss": 1.0312,
       "step": 46
     },
     {
       "epoch": 0.3098901098901099,
-      "grad_norm": 0.3210930824279785,
+      "grad_norm": 0.3197602331638336,
       "learning_rate": 0.00018181818181818183,
-      "loss": 0.9409,
+      "loss": 0.9411,
       "step": 47
     },
     {
       "epoch": 0.31648351648351647,
-      "grad_norm": 0.3485034704208374,
+      "grad_norm": 0.34601065516471863,
       "learning_rate": 0.00018137472283813748,
-      "loss": 0.9463,
+      "loss": 0.9466,
       "step": 48
     },
     {
       "epoch": 0.3230769230769231,
-      "grad_norm": 0.30606183409690857,
+      "grad_norm": 0.3059934973716736,
       "learning_rate": 0.00018093126385809312,
-      "loss": 0.8539,
+      "loss": 0.8534,
       "step": 49
     },
     {
       "epoch": 0.32967032967032966,
-      "grad_norm": 0.30226030945777893,
+      "grad_norm": 0.3049042820930481,
       "learning_rate": 0.0001804878048780488,
-      "loss": 0.9588,
+      "loss": 0.9564,
       "step": 50
     },
     {
       "epoch": 0.3362637362637363,
-      "grad_norm": 0.3496457040309906,
+      "grad_norm": 0.3513273298740387,
       "learning_rate": 0.00018004434589800444,
-      "loss": 1.0444,
+      "loss": 1.0429,
       "step": 51
     },
     {
       "epoch": 0.34285714285714286,
-      "grad_norm": 0.33469706773757935,
+      "grad_norm": 0.33669522404670715,
       "learning_rate": 0.00017960088691796008,
-      "loss": 0.9727,
+      "loss": 0.9706,
       "step": 52
     },
     {
       "epoch": 0.34945054945054943,
-      "grad_norm": 0.3913564383983612,
+      "grad_norm": 0.3956065773963928,
       "learning_rate": 0.00017915742793791575,
-      "loss": 1.0487,
+      "loss": 1.0505,
       "step": 53
     },
     {
       "epoch": 0.35604395604395606,
-      "grad_norm": 0.31554681062698364,
+      "grad_norm": 0.31517383456230164,
       "learning_rate": 0.0001787139689578714,
-      "loss": 1.0484,
+      "loss": 1.0469,
       "step": 54
     },
     {
       "epoch": 0.3626373626373626,
-      "grad_norm": 0.3576411306858063,
+      "grad_norm": 0.36682212352752686,
       "learning_rate": 0.00017827050997782707,
       "loss": 0.9408,
       "step": 55
     },
     {
       "epoch": 0.36923076923076925,
-      "grad_norm": 0.38003620505332947,
+      "grad_norm": 0.38366109132766724,
       "learning_rate": 0.00017782705099778271,
-      "loss": 1.1853,
+      "loss": 1.1865,
       "step": 56
     },
     {
       "epoch": 0.3758241758241758,
-      "grad_norm": 0.2947905659675598,
+      "grad_norm": 0.30710846185684204,
       "learning_rate": 0.00017738359201773839,
-      "loss": 1.0223,
+      "loss": 1.0216,
       "step": 57
     },
     {
       "epoch": 0.3824175824175824,
-      "grad_norm": 0.2967394292354584,
+      "grad_norm": 0.29654064774513245,
       "learning_rate": 0.000176940133037694,
-      "loss": 0.9359,
+      "loss": 0.9368,
       "step": 58
     },
     {
       "epoch": 0.389010989010989,
-      "grad_norm": 0.43115320801734924,
+      "grad_norm": 0.43266645073890686,
       "learning_rate": 0.00017649667405764967,
-      "loss": 0.9404,
+      "loss": 0.9399,
       "step": 59
     },
     {
       "epoch": 0.3956043956043956,
-      "grad_norm": 0.33834582567214966,
+      "grad_norm": 0.3399896025657654,
       "learning_rate": 0.00017605321507760532,
-      "loss": 1.2252,
+      "loss": 1.2272,
       "step": 60
     },
     {
       "epoch": 0.4021978021978022,
-      "grad_norm": 0.35457855463027954,
+      "grad_norm": 0.35805898904800415,
       "learning_rate": 0.000175609756097561,
-      "loss": 1.1279,
+      "loss": 1.1291,
       "step": 61
     },
     {
       "epoch": 0.4087912087912088,
-      "grad_norm": 0.3393542170524597,
+      "grad_norm": 0.3398084044456482,
       "learning_rate": 0.00017516629711751663,
-      "loss": 1.0621,
+      "loss": 1.062,
       "step": 62
     },
     {
       "epoch": 0.4153846153846154,
-      "grad_norm": 0.29777970910072327,
+      "grad_norm": 0.3010111451148987,
       "learning_rate": 0.0001747228381374723,
-      "loss": 0.9896,
+      "loss": 0.9908,
       "step": 63
     },
     {
       "epoch": 0.421978021978022,
-      "grad_norm": 0.3347181975841522,
+      "grad_norm": 0.3359769284725189,
       "learning_rate": 0.00017427937915742792,
-      "loss": 1.0941,
+      "loss": 1.094,
       "step": 64
     },
     {
       "epoch": 0.42857142857142855,
-      "grad_norm": 0.35212668776512146,
+      "grad_norm": 0.35044530034065247,
       "learning_rate": 0.0001738359201773836,
-      "loss": 0.9955,
+      "loss": 0.9933,
       "step": 65
     },
     {
       "epoch": 0.4351648351648352,
-      "grad_norm": 0.4496903717517853,
+      "grad_norm": 0.4499974548816681,
       "learning_rate": 0.00017339246119733924,
-      "loss": 1.099,
+      "loss": 1.0979,
       "step": 66
     },
     {
       "epoch": 0.44175824175824174,
-      "grad_norm": 0.23707008361816406,
+      "grad_norm": 0.2390085607767105,
       "learning_rate": 0.0001729490022172949,
-      "loss": 0.7456,
+      "loss": 0.7452,
       "step": 67
     },
     {
       "epoch": 0.44835164835164837,
-      "grad_norm": 0.2926875650882721,
+      "grad_norm": 0.2939722239971161,
       "learning_rate": 0.00017250554323725056,
-      "loss": 1.1481,
+      "loss": 1.1464,
       "step": 68
     },
     {
       "epoch": 0.45494505494505494,
-      "grad_norm": 0.42252832651138306,
+      "grad_norm": 0.4215264916419983,
       "learning_rate": 0.00017206208425720623,
-      "loss": 1.0156,
+      "loss": 1.0169,
       "step": 69
     },
     {
       "epoch": 0.46153846153846156,
-      "grad_norm": 0.2438354790210724,
+      "grad_norm": 0.24245759844779968,
       "learning_rate": 0.00017161862527716187,
       "loss": 0.7183,
       "step": 70
     },
     {
       "epoch": 0.46813186813186813,
-      "grad_norm": 0.33233973383903503,
+      "grad_norm": 0.33143526315689087,
       "learning_rate": 0.00017117516629711752,
-      "loss": 1.1451,
+      "loss": 1.1435,
       "step": 71
     },
     {
       "epoch": 0.4747252747252747,
-      "grad_norm": 0.30313268303871155,
+      "grad_norm": 0.3057692348957062,
       "learning_rate": 0.0001707317073170732,
-      "loss": 0.7117,
+      "loss": 0.7141,
       "step": 72
     },
     {
       "epoch": 0.48131868131868133,
-      "grad_norm": 0.31673258543014526,
+      "grad_norm": 0.3185351490974426,
       "learning_rate": 0.00017028824833702883,
-      "loss": 0.9695,
+      "loss": 0.9677,
       "step": 73
     },
     {
       "epoch": 0.4879120879120879,
-      "grad_norm": 0.31357434391975403,
+      "grad_norm": 0.3105548322200775,
       "learning_rate": 0.0001698447893569845,
-      "loss": 1.0329,
+      "loss": 1.0319,
       "step": 74
     },
     {
       "epoch": 0.4945054945054945,
-      "grad_norm": 0.26894521713256836,
+      "grad_norm": 0.27544498443603516,
       "learning_rate": 0.00016940133037694015,
-      "loss": 0.8497,
+      "loss": 0.8509,
       "step": 75
     },
     {
       "epoch": 0.5010989010989011,
-      "grad_norm": 0.24665911495685577,
+      "grad_norm": 0.24930939078330994,
       "learning_rate": 0.00016895787139689582,
-      "loss": 0.9803,
+      "loss": 0.9795,
       "step": 76
     },
     {
       "epoch": 0.5076923076923077,
-      "grad_norm": 0.3271327614784241,
+      "grad_norm": 0.326751172542572,
       "learning_rate": 0.00016851441241685144,
-      "loss": 1.0342,
+      "loss": 1.0334,
       "step": 77
     },
     {
       "epoch": 0.5142857142857142,
-      "grad_norm": 0.27658379077911377,
+      "grad_norm": 0.2782577574253082,
       "learning_rate": 0.0001680709534368071,
-      "loss": 0.9028,
+      "loss": 0.9029,
       "step": 78
     },
     {
       "epoch": 0.5208791208791209,
-      "grad_norm": 0.28840765357017517,
+      "grad_norm": 0.29126885533332825,
       "learning_rate": 0.00016762749445676275,
-      "loss": 0.9316,
+      "loss": 0.9306,
       "step": 79
     },
     {
       "epoch": 0.5274725274725275,
-      "grad_norm": 0.25261417031288147,
+      "grad_norm": 0.25332754850387573,
       "learning_rate": 0.00016718403547671842,
-      "loss": 0.9062,
+      "loss": 0.9057,
       "step": 80
     },
     {
       "epoch": 0.5340659340659341,
-      "grad_norm": 0.33007004857063293,
+      "grad_norm": 0.3294142782688141,
       "learning_rate": 0.00016674057649667407,
-      "loss": 0.7618,
+      "loss": 0.7615,
       "step": 81
     },
     {
       "epoch": 0.5406593406593406,
-      "grad_norm": 0.37919148802757263,
+      "grad_norm": 0.39348527789115906,
       "learning_rate": 0.00016629711751662974,
-      "loss": 0.8487,
+      "loss": 0.8525,
       "step": 82
     },
     {
       "epoch": 0.5472527472527473,
-      "grad_norm": 0.3373790979385376,
+      "grad_norm": 0.3311331868171692,
       "learning_rate": 0.00016585365853658536,
-      "loss": 1.0527,
+      "loss": 1.0504,
       "step": 83
     },
     {
       "epoch": 0.5538461538461539,
-      "grad_norm": 0.3170938491821289,
+      "grad_norm": 0.32208502292633057,
       "learning_rate": 0.00016541019955654103,
-      "loss": 1.0429,
+      "loss": 1.043,
       "step": 84
     },
     {
       "epoch": 0.5604395604395604,
-      "grad_norm": 0.3070983290672302,
+      "grad_norm": 0.3057066798210144,
       "learning_rate": 0.00016496674057649667,
-      "loss": 0.917,
+      "loss": 0.9153,
       "step": 85
     },
     {
       "epoch": 0.567032967032967,
-      "grad_norm": 0.3040669560432434,
+      "grad_norm": 0.3061386048793793,
       "learning_rate": 0.00016452328159645234,
-      "loss": 0.8483,
+      "loss": 0.8475,
       "step": 86
     },
     {
       "epoch": 0.5736263736263736,
-      "grad_norm": 0.3397156596183777,
+      "grad_norm": 0.34367796778678894,
       "learning_rate": 0.000164079822616408,
-      "loss": 1.1383,
+      "loss": 1.1397,
       "step": 87
     },
     {
       "epoch": 0.5802197802197803,
-      "grad_norm": 0.3420846164226532,
+      "grad_norm": 0.3496788442134857,
       "learning_rate": 0.00016363636363636366,
-      "loss": 0.9135,
+      "loss": 0.9111,
       "step": 88
     },
     {
       "epoch": 0.5868131868131868,
-      "grad_norm": 0.33865463733673096,
+      "grad_norm": 0.34243008494377136,
       "learning_rate": 0.0001631929046563193,
-      "loss": 0.9746,
+      "loss": 0.9765,
       "step": 89
     },
     {
       "epoch": 0.5934065934065934,
-      "grad_norm": 0.40767139196395874,
+      "grad_norm": 0.4083498418331146,
       "learning_rate": 0.00016274944567627495,
-      "loss": 1.0905,
+      "loss": 1.09,
       "step": 90
     },
     {
       "epoch": 0.6,
-      "grad_norm": 0.32135534286499023,
+      "grad_norm": 0.32501041889190674,
       "learning_rate": 0.0001623059866962306,
-      "loss": 0.9839,
+      "loss": 0.983,
       "step": 91
     },
     {
       "epoch": 0.6065934065934065,
-      "grad_norm": 0.29726099967956543,
+      "grad_norm": 0.2962501645088196,
       "learning_rate": 0.00016186252771618626,
-      "loss": 0.8087,
+      "loss": 0.8079,
       "step": 92
     },
     {
       "epoch": 0.6131868131868132,
-      "grad_norm": 0.3343488574028015,
+      "grad_norm": 0.339017778635025,
       "learning_rate": 0.0001614190687361419,
-      "loss": 1.062,
+      "loss": 1.0645,
       "step": 93
     },
     {
       "epoch": 0.6197802197802198,
-      "grad_norm": 0.3392772972583771,
+      "grad_norm": 0.33595070242881775,
       "learning_rate": 0.00016097560975609758,
-      "loss": 0.9496,
+      "loss": 0.9504,
       "step": 94
     },
     {
       "epoch": 0.6263736263736264,
-      "grad_norm": 0.38109564781188965,
+      "grad_norm": 0.3844953179359436,
       "learning_rate": 0.00016053215077605322,
-      "loss": 0.8332,
+      "loss": 0.8361,
       "step": 95
     },
     {
       "epoch": 0.6329670329670329,
-      "grad_norm": 0.4026820659637451,
+      "grad_norm": 0.4032916724681854,
       "learning_rate": 0.00016008869179600887,
-      "loss": 1.0506,
+      "loss": 1.0501,
       "step": 96
     },
     {
       "epoch": 0.6395604395604395,
-      "grad_norm": 0.32605379819869995,
+      "grad_norm": 0.3297523856163025,
       "learning_rate": 0.00015964523281596454,
-      "loss": 0.9707,
+      "loss": 0.9706,
       "step": 97
     },
     {
       "epoch": 0.6461538461538462,
-      "grad_norm": 0.32885733246803284,
+      "grad_norm": 0.3284400403499603,
       "learning_rate": 0.00015920177383592018,
-      "loss": 0.95,
+      "loss": 0.9507,
       "step": 98
     },
     {
       "epoch": 0.6527472527472528,
-      "grad_norm": 0.2949531078338623,
+      "grad_norm": 0.3026384115219116,
       "learning_rate": 0.00015875831485587586,
-      "loss": 0.9197,
+      "loss": 0.9194,
       "step": 99
     },
     {
       "epoch": 0.6593406593406593,
-      "grad_norm": 0.34610554575920105,
+      "grad_norm": 0.3462526500225067,
       "learning_rate": 0.0001583148558758315,
-      "loss": 0.9544,
+      "loss": 0.9535,
       "step": 100
     },
     {
       "epoch": 0.6659340659340659,
-      "grad_norm": 0.2914392650127411,
+      "grad_norm": 0.2929343283176422,
       "learning_rate": 0.00015787139689578714,
-      "loss": 1.0428,
+      "loss": 1.0426,
       "step": 101
     },
     {
       "epoch": 0.6725274725274726,
-      "grad_norm": 0.33224692940711975,
+      "grad_norm": 0.33232760429382324,
       "learning_rate": 0.0001574279379157428,
-      "loss": 0.9962,
+      "loss": 0.9976,
       "step": 102
     },
     {
       "epoch": 0.6791208791208792,
-      "grad_norm": 0.3527650833129883,
+      "grad_norm": 0.3566662073135376,
       "learning_rate": 0.00015698447893569846,
-      "loss": 0.8515,
+      "loss": 0.8533,
       "step": 103
     },
     {
       "epoch": 0.6857142857142857,
-      "grad_norm": 0.3555796444416046,
+      "grad_norm": 0.3751154839992523,
       "learning_rate": 0.0001565410199556541,
-      "loss": 0.957,
+      "loss": 0.9584,
       "step": 104
     },
     {
       "epoch": 0.6923076923076923,
-      "grad_norm": 0.33348530530929565,
+      "grad_norm": 0.3387792408466339,
       "learning_rate": 0.00015609756097560978,
-      "loss": 0.9111,
+      "loss": 0.9135,
       "step": 105
     },
     {
       "epoch": 0.6989010989010989,
-      "grad_norm": 0.29242005944252014,
+      "grad_norm": 0.2936052978038788,
       "learning_rate": 0.00015565410199556542,
-      "loss": 0.7325,
+      "loss": 0.7321,
       "step": 106
     },
     {
       "epoch": 0.7054945054945055,
-      "grad_norm": 0.3308833837509155,
+      "grad_norm": 0.3313789963722229,
       "learning_rate": 0.00015521064301552106,
-      "loss": 0.9916,
+      "loss": 0.9921,
       "step": 107
     },
     {
       "epoch": 0.7120879120879121,
-      "grad_norm": 0.38151150941848755,
+      "grad_norm": 0.3824083209037781,
       "learning_rate": 0.0001547671840354767,
-      "loss": 0.867,
+      "loss": 0.8662,
       "step": 108
     },
     {
       "epoch": 0.7186813186813187,
-      "grad_norm": 0.2719651758670807,
+      "grad_norm": 0.271816611289978,
       "learning_rate": 0.00015432372505543238,
-      "loss": 0.7214,
+      "loss": 0.7216,
       "step": 109
     },
     {
       "epoch": 0.7252747252747253,
-      "grad_norm": 0.3571125566959381,
+      "grad_norm": 0.3538140654563904,
       "learning_rate": 0.00015388026607538802,
-      "loss": 0.8555,
+      "loss": 0.8546,
       "step": 110
     },
     {
       "epoch": 0.7318681318681318,
-      "grad_norm": 0.3709869980812073,
+      "grad_norm": 0.3695511817932129,
       "learning_rate": 0.0001534368070953437,
-      "loss": 0.9381,
+      "loss": 0.9383,
       "step": 111
     },
     {
       "epoch": 0.7384615384615385,
-      "grad_norm": 0.34119173884391785,
+      "grad_norm": 0.3450765311717987,
       "learning_rate": 0.00015299334811529934,
-      "loss": 1.0286,
+      "loss": 1.0277,
       "step": 112
     },
     {
       "epoch": 0.7450549450549451,
-      "grad_norm": 0.35051465034484863,
+      "grad_norm": 0.3513217270374298,
       "learning_rate": 0.00015254988913525498,
-      "loss": 0.8688,
+      "loss": 0.8685,
       "step": 113
     },
     {
       "epoch": 0.7516483516483516,
-      "grad_norm": 0.3159275949001312,
+      "grad_norm": 0.31267625093460083,
       "learning_rate": 0.00015210643015521066,
-      "loss": 0.9751,
+      "loss": 0.9752,
       "step": 114
     },
     {
       "epoch": 0.7582417582417582,
-      "grad_norm": 0.27043935656547546,
+      "grad_norm": 0.2716124653816223,
       "learning_rate": 0.0001516629711751663,
-      "loss": 0.7634,
+      "loss": 0.7638,
       "step": 115
     },
     {
       "epoch": 0.7648351648351648,
-      "grad_norm": 0.37761369347572327,
+      "grad_norm": 0.3737407922744751,
       "learning_rate": 0.00015121951219512197,
-      "loss": 0.9724,
+      "loss": 0.9719,
       "step": 116
     },
     {
       "epoch": 0.7714285714285715,
-      "grad_norm": 0.3172036111354828,
+      "grad_norm": 0.3199615776538849,
       "learning_rate": 0.00015077605321507762,
-      "loss": 0.8525,
+      "loss": 0.8535,
       "step": 117
     },
     {
       "epoch": 0.778021978021978,
-      "grad_norm": 0.28983673453330994,
+      "grad_norm": 0.29326456785202026,
       "learning_rate": 0.0001503325942350333,
-      "loss": 0.9894,
+      "loss": 0.991,
       "step": 118
     },
     {
       "epoch": 0.7846153846153846,
-      "grad_norm": 0.3155057430267334,
+      "grad_norm": 0.3158397674560547,
       "learning_rate": 0.0001498891352549889,
-      "loss": 0.9778,
+      "loss": 0.9742,
       "step": 119
     },
     {
       "epoch": 0.7912087912087912,
-      "grad_norm": 0.33516910672187805,
+      "grad_norm": 0.340801477432251,
       "learning_rate": 0.00014944567627494458,
-      "loss": 1.0043,
+      "loss": 1.006,
       "step": 120
     },
     {
       "epoch": 0.7978021978021979,
-      "grad_norm": 0.41567182540893555,
+      "grad_norm": 0.4258521795272827,
       "learning_rate": 0.00014900221729490022,
-      "loss": 1.023,
+      "loss": 1.0222,
       "step": 121
     },
     {
       "epoch": 0.8043956043956044,
-      "grad_norm": 0.40814825892448425,
+      "grad_norm": 0.40230363607406616,
       "learning_rate": 0.0001485587583148559,
-      "loss": 1.0267,
+      "loss": 1.0283,
       "step": 122
     },
     {
       "epoch": 0.810989010989011,
-      "grad_norm": 0.297061949968338,
+      "grad_norm": 0.298953652381897,
       "learning_rate": 0.00014811529933481154,
-      "loss": 0.8668,
+      "loss": 0.8674,
       "step": 123
     },
     {
       "epoch": 0.8175824175824176,
-      "grad_norm": 0.3487430512905121,
+      "grad_norm": 0.34640374779701233,
       "learning_rate": 0.0001476718403547672,
-      "loss": 0.9448,
+      "loss": 0.9418,
       "step": 124
     },
     {
       "epoch": 0.8241758241758241,
-      "grad_norm": 0.2947084307670593,
+      "grad_norm": 0.2947790026664734,
       "learning_rate": 0.00014722838137472282,
-      "loss": 0.8624,
+      "loss": 0.8632,
       "step": 125
     },
     {
       "epoch": 0.8307692307692308,
-      "grad_norm": 0.4176642894744873,
+      "grad_norm": 0.42390382289886475,
       "learning_rate": 0.0001467849223946785,
-      "loss": 0.9898,
+      "loss": 0.992,
       "step": 126
     },
     {
       "epoch": 0.8373626373626374,
-      "grad_norm": 0.3396783173084259,
+      "grad_norm": 0.33880630135536194,
       "learning_rate": 0.00014634146341463414,
-      "loss": 0.8457,
+      "loss": 0.8464,
       "step": 127
     },
     {
       "epoch": 0.843956043956044,
-      "grad_norm": 0.30610787868499756,
+      "grad_norm": 0.305600106716156,
       "learning_rate": 0.0001458980044345898,
-      "loss": 0.7489,
+      "loss": 0.7496,
       "step": 128
     },
     {
       "epoch": 0.8505494505494505,
-      "grad_norm": 0.411915123462677,
+      "grad_norm": 0.4102969765663147,
       "learning_rate": 0.00014545454545454546,
-      "loss": 1.0267,
+      "loss": 1.027,
       "step": 129
     },
     {
       "epoch": 0.8571428571428571,
-      "grad_norm": 0.2569451332092285,
+      "grad_norm": 0.25675109028816223,
       "learning_rate": 0.00014501108647450113,
-      "loss": 0.8711,
+      "loss": 0.8703,
       "step": 130
     },
     {
       "epoch": 0.8637362637362638,
-      "grad_norm": 0.3669357895851135,
+      "grad_norm": 0.37069281935691833,
       "learning_rate": 0.00014456762749445675,
-      "loss": 0.978,
+      "loss": 0.9777,
       "step": 131
     },
     {
       "epoch": 0.8703296703296703,
-      "grad_norm": 0.4036771357059479,
+      "grad_norm": 0.4089072644710541,
       "learning_rate": 0.00014412416851441242,
-      "loss": 0.8404,
+      "loss": 0.8419,
       "step": 132
     },
     {
       "epoch": 0.8769230769230769,
-      "grad_norm": 0.30679866671562195,
+      "grad_norm": 0.31191402673721313,
       "learning_rate": 0.00014368070953436806,
-      "loss": 0.9304,
+      "loss": 0.9311,
       "step": 133
     },
     {
       "epoch": 0.8835164835164835,
-      "grad_norm": 0.4402814507484436,
+      "grad_norm": 0.4401726722717285,
       "learning_rate": 0.00014323725055432373,
-      "loss": 0.9296,
+      "loss": 0.9287,
       "step": 134
     },
     {
       "epoch": 0.8901098901098901,
-      "grad_norm": 0.31870439648628235,
+      "grad_norm": 0.3216991424560547,
       "learning_rate": 0.00014279379157427938,
-      "loss": 0.9951,
+      "loss": 0.9944,
       "step": 135
     },
     {
       "epoch": 0.8967032967032967,
-      "grad_norm": 0.36312800645828247,
+      "grad_norm": 0.3641892075538635,
       "learning_rate": 0.00014235033259423505,
-      "loss": 1.0001,
+      "loss": 1.0015,
       "step": 136
     },
     {
       "epoch": 0.9032967032967033,
-      "grad_norm": 0.3176593780517578,
+      "grad_norm": 0.31928136944770813,
       "learning_rate": 0.0001419068736141907,
-      "loss": 0.8874,
+      "loss": 0.888,
       "step": 137
     },
     {
       "epoch": 0.9098901098901099,
-      "grad_norm": 0.34519079327583313,
+      "grad_norm": 0.34381303191185,
       "learning_rate": 0.00014146341463414634,
-      "loss": 0.918,
+      "loss": 0.9181,
       "step": 138
     },
     {
       "epoch": 0.9164835164835164,
-      "grad_norm": 0.34109336137771606,
+      "grad_norm": 0.34465450048446655,
       "learning_rate": 0.000141019955654102,
-      "loss": 0.9881,
+      "loss": 0.9856,
       "step": 139
     },
     {
       "epoch": 0.9230769230769231,
-      "grad_norm": 0.33588215708732605,
+      "grad_norm": 0.34267473220825195,
       "learning_rate": 0.00014057649667405765,
-      "loss": 1.0556,
+      "loss": 1.0567,
       "step": 140
     },
     {
       "epoch": 0.9296703296703297,
-      "grad_norm": 0.3143637180328369,
+      "grad_norm": 0.31306982040405273,
       "learning_rate": 0.00014013303769401332,
-      "loss": 0.8042,
+      "loss": 0.8032,
       "step": 141
     },
     {
       "epoch": 0.9362637362637363,
-      "grad_norm": 0.34192153811454773,
+      "grad_norm": 0.3434275984764099,
       "learning_rate": 0.00013968957871396897,
-      "loss": 0.9389,
+      "loss": 0.9398,
       "step": 142
     },
     {
       "epoch": 0.9428571428571428,
-      "grad_norm": 0.33687394857406616,
+      "grad_norm": 0.33172014355659485,
       "learning_rate": 0.00013924611973392464,
-      "loss": 0.9696,
+      "loss": 0.97,
       "step": 143
     },
     {
       "epoch": 0.9494505494505494,
-      "grad_norm": 0.32633718848228455,
+      "grad_norm": 0.3282938599586487,
       "learning_rate": 0.00013880266075388026,
-      "loss": 0.9389,
+      "loss": 0.9407,
       "step": 144
     },
     {
       "epoch": 0.9560439560439561,
-      "grad_norm": 0.33426982164382935,
+      "grad_norm": 0.332861065864563,
       "learning_rate": 0.00013835920177383593,
-      "loss": 1.0901,
+      "loss": 1.0917,
       "step": 145
     },
     {
       "epoch": 0.9626373626373627,
-      "grad_norm": 0.3398204743862152,
+      "grad_norm": 0.336420476436615,
       "learning_rate": 0.00013791574279379157,
-      "loss": 1.0035,
+      "loss": 1.003,
       "step": 146
     },
     {
       "epoch": 0.9692307692307692,
-      "grad_norm": 0.3482389450073242,
+      "grad_norm": 0.34626010060310364,
       "learning_rate": 0.00013747228381374724,
-      "loss": 1.0233,
+      "loss": 1.0213,
       "step": 147
     },
     {
       "epoch": 0.9758241758241758,
-      "grad_norm": 0.27446088194847107,
+      "grad_norm": 0.2741713225841522,
       "learning_rate": 0.0001370288248337029,
-      "loss": 0.8593,
+      "loss": 0.8592,
       "step": 148
     },
     {
       "epoch": 0.9824175824175824,
-      "grad_norm": 0.3116665184497833,
+      "grad_norm": 0.3122574985027313,
       "learning_rate": 0.00013658536585365856,
-      "loss": 0.9122,
+      "loss": 0.9123,
       "step": 149
     },
     {
       "epoch": 0.989010989010989,
-      "grad_norm": 0.32924315333366394,
+      "grad_norm": 0.32611116766929626,
       "learning_rate": 0.00013614190687361418,
-      "loss": 0.8465,
+      "loss": 0.8468,
       "step": 150
     },
     {
       "epoch": 0.9956043956043956,
-      "grad_norm": 0.3131709396839142,
+      "grad_norm": 0.31028270721435547,
       "learning_rate": 0.00013569844789356985,
-      "loss": 1.1115,
+      "loss": 1.1108,
       "step": 151
     },
     {
       "epoch": 1.0,
-      "grad_norm": 0.5639778971672058,
+      "grad_norm": 0.5587679147720337,
       "learning_rate": 0.0001352549889135255,
-      "loss": 1.0023,
+      "loss": 1.0044,
       "step": 152
     },
     {
       "epoch": 1.0065934065934066,
-      "grad_norm": 0.3395214080810547,
+      "grad_norm": 0.34237536787986755,
       "learning_rate": 0.00013481152993348116,
-      "loss": 0.7579,
+      "loss": 0.7587,
       "step": 153
     },
     {
       "epoch": 1.0131868131868131,
-      "grad_norm": 0.4451121985912323,
+      "grad_norm": 0.4465104639530182,
       "learning_rate": 0.0001343680709534368,
-      "loss": 0.9046,
+      "loss": 0.904,
       "step": 154
     },
     {
       "epoch": 1.0197802197802197,
-      "grad_norm": 0.30658870935440063,
+      "grad_norm": 0.30461183190345764,
       "learning_rate": 0.00013392461197339248,
-      "loss": 0.7993,
+      "loss": 0.7978,
       "step": 155
     },
     {
       "epoch": 1.0263736263736263,
-      "grad_norm": 0.3359564542770386,
+      "grad_norm": 0.33155784010887146,
       "learning_rate": 0.00013348115299334812,
-      "loss": 0.9043,
+      "loss": 0.9028,
       "step": 156
     },
     {
       "epoch": 1.032967032967033,
-      "grad_norm": 0.3504899740219116,
+      "grad_norm": 0.3484174311161041,
       "learning_rate": 0.00013303769401330377,
-      "loss": 0.7881,
+      "loss": 0.7883,
       "step": 157
     },
     {
       "epoch": 1.0395604395604396,
-      "grad_norm": 0.3386143445968628,
+      "grad_norm": 0.3378135561943054,
       "learning_rate": 0.00013259423503325944,
-      "loss": 0.9221,
+      "loss": 0.9246,
       "step": 158
     },
     {
       "epoch": 1.0461538461538462,
-      "grad_norm": 0.28975123167037964,
+      "grad_norm": 0.2909366190433502,
       "learning_rate": 0.00013215077605321509,
-      "loss": 0.7426,
+      "loss": 0.7427,
       "step": 159
     },
     {
       "epoch": 1.0527472527472528,
-      "grad_norm": 0.3702305853366852,
+      "grad_norm": 0.370918869972229,
       "learning_rate": 0.00013170731707317076,
-      "loss": 0.7627,
+      "loss": 0.762,
       "step": 160
     },
     {
       "epoch": 1.0593406593406594,
-      "grad_norm": 0.394538551568985,
+      "grad_norm": 0.3959696888923645,
       "learning_rate": 0.0001312638580931264,
-      "loss": 0.9105,
+      "loss": 0.9099,
       "step": 161
     },
     {
       "epoch": 1.065934065934066,
-      "grad_norm": 0.3724621832370758,
+      "grad_norm": 0.36984387040138245,
       "learning_rate": 0.00013082039911308205,
-      "loss": 0.9514,
+      "loss": 0.9537,
       "step": 162
     },
     {
       "epoch": 1.0725274725274725,
-      "grad_norm": 0.3351310193538666,
+      "grad_norm": 0.33510100841522217,
       "learning_rate": 0.0001303769401330377,
-      "loss": 0.863,
+      "loss": 0.8638,
       "step": 163
     },
     {
       "epoch": 1.079120879120879,
-      "grad_norm": 0.3554590344429016,
+      "grad_norm": 0.35161909461021423,
       "learning_rate": 0.00012993348115299336,
-      "loss": 0.9837,
+      "loss": 0.9841,
       "step": 164
     },
     {
       "epoch": 1.0857142857142856,
-      "grad_norm": 0.3050754964351654,
+      "grad_norm": 0.3048061728477478,
       "learning_rate": 0.000129490022172949,
       "loss": 0.7628,
       "step": 165
     },
     {
       "epoch": 1.0923076923076924,
-      "grad_norm": 0.35037916898727417,
+      "grad_norm": 0.35248255729675293,
       "learning_rate": 0.00012904656319290468,
-      "loss": 0.8715,
+      "loss": 0.8721,
       "step": 166
     },
     {
       "epoch": 1.098901098901099,
-      "grad_norm": 0.4356972873210907,
+      "grad_norm": 0.43364354968070984,
       "learning_rate": 0.00012860310421286032,
       "loss": 0.8514,
       "step": 167
     },
     {
       "epoch": 1.1054945054945056,
-      "grad_norm": 0.44227975606918335,
+      "grad_norm": 0.4386586546897888,
       "learning_rate": 0.00012815964523281597,
-      "loss": 0.8368,
+      "loss": 0.8363,
       "step": 168
     },
     {
       "epoch": 1.1120879120879121,
-      "grad_norm": 0.41575899720191956,
+      "grad_norm": 0.4130457639694214,
       "learning_rate": 0.0001277161862527716,
-      "loss": 0.9083,
+      "loss": 0.9079,
       "step": 169
     },
     {
       "epoch": 1.1186813186813187,
-      "grad_norm": 0.32656052708625793,
+      "grad_norm": 0.32351064682006836,
       "learning_rate": 0.00012727272727272728,
-      "loss": 0.7544,
+      "loss": 0.7551,
       "step": 170
     },
     {
       "epoch": 1.1252747252747253,
-      "grad_norm": 0.4732937812805176,
+      "grad_norm": 0.47145897150039673,
       "learning_rate": 0.00012682926829268293,
-      "loss": 1.0187,
+      "loss": 1.0163,
       "step": 171
     },
     {
       "epoch": 1.1318681318681318,
-      "grad_norm": 0.3214307427406311,
+      "grad_norm": 0.31925448775291443,
       "learning_rate": 0.0001263858093126386,
-      "loss": 0.8781,
+      "loss": 0.877,
       "step": 172
     },
     {
       "epoch": 1.1384615384615384,
-      "grad_norm": 0.4291451573371887,
+      "grad_norm": 0.42638224363327026,
       "learning_rate": 0.00012594235033259424,
-      "loss": 0.869,
+      "loss": 0.867,
       "step": 173
     },
     {
       "epoch": 1.145054945054945,
-      "grad_norm": 0.4516123831272125,
+      "grad_norm": 0.4519364535808563,
       "learning_rate": 0.00012549889135254989,
-      "loss": 0.9012,
+      "loss": 0.8992,
       "step": 174
     },
     {
       "epoch": 1.1516483516483516,
-      "grad_norm": 0.39512842893600464,
+      "grad_norm": 0.39410778880119324,
       "learning_rate": 0.00012505543237250553,
-      "loss": 0.897,
+      "loss": 0.8946,
       "step": 175
     },
     {
       "epoch": 1.1582417582417581,
-      "grad_norm": 0.32635363936424255,
+      "grad_norm": 0.32669949531555176,
       "learning_rate": 0.0001246119733924612,
-      "loss": 0.7872,
+      "loss": 0.7879,
       "step": 176
     },
     {
       "epoch": 1.164835164835165,
-      "grad_norm": 0.34062984585762024,
+      "grad_norm": 0.3455027639865875,
       "learning_rate": 0.00012416851441241685,
-      "loss": 0.7698,
+      "loss": 0.7705,
       "step": 177
     },
     {
       "epoch": 1.1714285714285715,
-      "grad_norm": 0.4236986041069031,
+      "grad_norm": 0.42715129256248474,
       "learning_rate": 0.00012372505543237252,
-      "loss": 0.8203,
+      "loss": 0.8197,
       "step": 178
     },
     {
       "epoch": 1.178021978021978,
-      "grad_norm": 0.3937680721282959,
+      "grad_norm": 0.3946492671966553,
       "learning_rate": 0.00012328159645232816,
       "loss": 0.6399,
       "step": 179
     },
     {
       "epoch": 1.1846153846153846,
-      "grad_norm": 0.45519042015075684,
+      "grad_norm": 0.45343202352523804,
       "learning_rate": 0.0001228381374722838,
-      "loss": 0.9213,
+      "loss": 0.925,
       "step": 180
     },
     {
       "epoch": 1.1912087912087912,
-      "grad_norm": 0.44973817467689514,
+      "grad_norm": 0.44791096448898315,
       "learning_rate": 0.00012239467849223948,
-      "loss": 0.9743,
+      "loss": 0.9734,
       "step": 181
     },
     {
       "epoch": 1.1978021978021978,
-      "grad_norm": 0.42643284797668457,
+      "grad_norm": 0.4247220456600189,
       "learning_rate": 0.00012195121951219512,
-      "loss": 0.9751,
+      "loss": 0.9757,
       "step": 182
     },
     {
       "epoch": 1.2043956043956043,
-      "grad_norm": 0.41849836707115173,
+      "grad_norm": 0.4177721440792084,
       "learning_rate": 0.00012150776053215078,
-      "loss": 0.8524,
+      "loss": 0.8537,
       "step": 183
     },
     {
       "epoch": 1.210989010989011,
-      "grad_norm": 0.48973095417022705,
+      "grad_norm": 0.47911062836647034,
       "learning_rate": 0.00012106430155210644,
-      "loss": 0.722,
+      "loss": 0.7214,
       "step": 184
     },
     {
       "epoch": 1.2175824175824177,
-      "grad_norm": 0.4730561077594757,
+      "grad_norm": 0.470956951379776,
       "learning_rate": 0.0001206208425720621,
-      "loss": 0.8323,
+      "loss": 0.8312,
       "step": 185
     },
     {
       "epoch": 1.2241758241758243,
-      "grad_norm": 0.43756726384162903,
+      "grad_norm": 0.4381925165653229,
       "learning_rate": 0.00012017738359201774,
-      "loss": 0.8261,
+      "loss": 0.8253,
       "step": 186
     },
     {
       "epoch": 1.2307692307692308,
-      "grad_norm": 0.34324491024017334,
+      "grad_norm": 0.3466181457042694,
       "learning_rate": 0.0001197339246119734,
-      "loss": 0.7222,
+      "loss": 0.7228,
       "step": 187
     },
     {
       "epoch": 1.2373626373626374,
-      "grad_norm": 0.4662308990955353,
+      "grad_norm": 0.47227564454078674,
       "learning_rate": 0.00011929046563192906,
-      "loss": 0.9886,
+      "loss": 0.9872,
       "step": 188
     },
     {
       "epoch": 1.243956043956044,
-      "grad_norm": 0.39682698249816895,
+      "grad_norm": 0.3967965543270111,
       "learning_rate": 0.00011884700665188471,
-      "loss": 0.8586,
+      "loss": 0.8568,
       "step": 189
     },
     {
       "epoch": 1.2505494505494505,
-      "grad_norm": 0.4866606891155243,
+      "grad_norm": 0.4912855923175812,
       "learning_rate": 0.00011840354767184036,
-      "loss": 0.9655,
+      "loss": 0.9629,
       "step": 190
     },
     {
       "epoch": 1.2571428571428571,
-      "grad_norm": 0.43162989616394043,
+      "grad_norm": 0.4317874312400818,
       "learning_rate": 0.00011796008869179602,
-      "loss": 0.8334,
+      "loss": 0.8323,
       "step": 191
     },
     {
       "epoch": 1.2637362637362637,
-      "grad_norm": 0.4115634560585022,
+      "grad_norm": 0.4118874967098236,
       "learning_rate": 0.00011751662971175166,
-      "loss": 0.9202,
+      "loss": 0.9194,
       "step": 192
     },
     {
       "epoch": 1.2703296703296703,
-      "grad_norm": 0.45951539278030396,
+      "grad_norm": 0.4587341547012329,
       "learning_rate": 0.00011707317073170732,
-      "loss": 0.903,
+      "loss": 0.9024,
       "step": 193
     },
     {
       "epoch": 1.2769230769230768,
-      "grad_norm": 0.4265710413455963,
+      "grad_norm": 0.4255760908126831,
       "learning_rate": 0.00011662971175166298,
-      "loss": 0.7974,
+      "loss": 0.7961,
       "step": 194
     },
     {
       "epoch": 1.2835164835164834,
-      "grad_norm": 0.4018781781196594,
+      "grad_norm": 0.39803823828697205,
       "learning_rate": 0.00011618625277161863,
-      "loss": 0.7937,
+      "loss": 0.793,
       "step": 195
     },
     {
       "epoch": 1.2901098901098902,
-      "grad_norm": 0.33751317858695984,
+      "grad_norm": 0.33639976382255554,
       "learning_rate": 0.00011574279379157429,
-      "loss": 0.958,
+      "loss": 0.9582,
       "step": 196
     },
     {
       "epoch": 1.2967032967032968,
-      "grad_norm": 0.5202163457870483,
+      "grad_norm": 0.5177255868911743,
       "learning_rate": 0.00011529933481152995,
-      "loss": 0.8871,
+      "loss": 0.8881,
       "step": 197
     },
     {
       "epoch": 1.3032967032967033,
-      "grad_norm": 0.5462297797203064,
+      "grad_norm": 0.5451109409332275,
       "learning_rate": 0.00011485587583148558,
-      "loss": 0.8778,
+      "loss": 0.8771,
       "step": 198
     },
     {
       "epoch": 1.30989010989011,
-      "grad_norm": 0.5294543504714966,
+      "grad_norm": 0.5325527787208557,
       "learning_rate": 0.00011441241685144124,
-      "loss": 0.8053,
+      "loss": 0.8054,
       "step": 199
     },
     {
       "epoch": 1.3164835164835165,
-      "grad_norm": 0.508984386920929,
+      "grad_norm": 0.5048739910125732,
       "learning_rate": 0.0001139689578713969,
-      "loss": 0.9449,
+      "loss": 0.9425,
       "step": 200
     },
     {
       "epoch": 1.323076923076923,
-      "grad_norm": 0.4766629934310913,
+      "grad_norm": 0.477891206741333,
       "learning_rate": 0.00011352549889135255,
-      "loss": 0.9975,
+      "loss": 0.9965,
       "step": 201
     },
     {
       "epoch": 1.3296703296703296,
-      "grad_norm": 0.4836457073688507,
+      "grad_norm": 0.4765445291996002,
       "learning_rate": 0.00011308203991130821,
-      "loss": 1.0674,
+      "loss": 1.0682,
       "step": 202
     },
     {
       "epoch": 1.3362637362637364,
-      "grad_norm": 0.4459066689014435,
+      "grad_norm": 0.44138312339782715,
       "learning_rate": 0.00011263858093126387,
-      "loss": 0.9403,
+      "loss": 0.9393,
       "step": 203
     },
     {
       "epoch": 1.342857142857143,
-      "grad_norm": 0.49527183175086975,
+      "grad_norm": 0.49631035327911377,
       "learning_rate": 0.00011219512195121953,
-      "loss": 1.0174,
+      "loss": 1.018,
       "step": 204
     },
     {
       "epoch": 1.3494505494505495,
-      "grad_norm": 0.4496254026889801,
+      "grad_norm": 0.4567028284072876,
       "learning_rate": 0.00011175166297117516,
-      "loss": 0.8753,
+      "loss": 0.8774,
       "step": 205
     },
     {
       "epoch": 1.3560439560439561,
-      "grad_norm": 0.5478966236114502,
+      "grad_norm": 0.55516117811203,
       "learning_rate": 0.00011130820399113082,
-      "loss": 1.0999,
+      "loss": 1.1016,
       "step": 206
     },
     {
       "epoch": 1.3626373626373627,
-      "grad_norm": 0.4426548182964325,
+      "grad_norm": 0.4408508539199829,
       "learning_rate": 0.00011086474501108647,
-      "loss": 0.7285,
+      "loss": 0.7284,
       "step": 207
     },
     {
       "epoch": 1.3692307692307693,
-      "grad_norm": 0.5293945670127869,
+      "grad_norm": 0.5260791778564453,
       "learning_rate": 0.00011042128603104213,
-      "loss": 0.9999,
+      "loss": 0.9981,
       "step": 208
     },
     {
       "epoch": 1.3758241758241758,
-      "grad_norm": 0.47293853759765625,
+      "grad_norm": 0.47149479389190674,
       "learning_rate": 0.00010997782705099779,
-      "loss": 0.7832,
+      "loss": 0.782,
       "step": 209
     },
     {
       "epoch": 1.3824175824175824,
-      "grad_norm": 0.47598668932914734,
+      "grad_norm": 0.4713144600391388,
       "learning_rate": 0.00010953436807095345,
-      "loss": 0.8358,
+      "loss": 0.8344,
       "step": 210
     },
     {
       "epoch": 1.389010989010989,
-      "grad_norm": 0.426973432302475,
+      "grad_norm": 0.4286600351333618,
       "learning_rate": 0.00010909090909090909,
-      "loss": 0.7659,
+      "loss": 0.7674,
       "step": 211
     },
     {
       "epoch": 1.3956043956043955,
-      "grad_norm": 0.4742601215839386,
+      "grad_norm": 0.47824904322624207,
       "learning_rate": 0.00010864745011086475,
-      "loss": 0.8223,
+      "loss": 0.8219,
       "step": 212
     },
     {
       "epoch": 1.402197802197802,
-      "grad_norm": 0.5977970957756042,
+      "grad_norm": 0.5984587073326111,
       "learning_rate": 0.00010820399113082041,
-      "loss": 0.8722,
+      "loss": 0.8736,
       "step": 213
     },
     {
       "epoch": 1.4087912087912087,
-      "grad_norm": 0.4443981647491455,
+      "grad_norm": 0.4488672912120819,
       "learning_rate": 0.00010776053215077607,
-      "loss": 0.842,
+      "loss": 0.8427,
       "step": 214
     },
     {
       "epoch": 1.4153846153846155,
-      "grad_norm": 0.5637273788452148,
+      "grad_norm": 0.562637448310852,
       "learning_rate": 0.00010731707317073172,
-      "loss": 0.9229,
+      "loss": 0.9239,
       "step": 215
     },
     {
       "epoch": 1.421978021978022,
-      "grad_norm": 0.4368816018104553,
+      "grad_norm": 0.43384426832199097,
       "learning_rate": 0.00010687361419068738,
-      "loss": 0.8706,
+      "loss": 0.8691,
       "step": 216
     },
     {
       "epoch": 1.4285714285714286,
-      "grad_norm": 0.47785332798957825,
+      "grad_norm": 0.47463458776474,
       "learning_rate": 0.00010643015521064301,
-      "loss": 0.8216,
+      "loss": 0.8234,
       "step": 217
     },
     {
       "epoch": 1.4351648351648352,
-      "grad_norm": 0.5267130136489868,
+      "grad_norm": 0.5394907593727112,
       "learning_rate": 0.00010598669623059867,
-      "loss": 0.8562,
+      "loss": 0.8585,
       "step": 218
     },
     {
       "epoch": 1.4417582417582417,
-      "grad_norm": 0.4472438097000122,
+      "grad_norm": 0.4483904242515564,
       "learning_rate": 0.00010554323725055433,
-      "loss": 0.6763,
+      "loss": 0.6775,
       "step": 219
     },
     {
       "epoch": 1.4483516483516483,
-      "grad_norm": 0.5247462391853333,
+      "grad_norm": 0.5243779420852661,
       "learning_rate": 0.00010509977827050999,
-      "loss": 0.9322,
+      "loss": 0.934,
       "step": 220
     },
     {
       "epoch": 1.4549450549450549,
-      "grad_norm": 0.5785446763038635,
+      "grad_norm": 0.5830507278442383,
       "learning_rate": 0.00010465631929046564,
-      "loss": 0.8836,
+      "loss": 0.8841,
       "step": 221
     },
     {
       "epoch": 1.4615384615384617,
-      "grad_norm": 0.5753875970840454,
+      "grad_norm": 0.587237536907196,
       "learning_rate": 0.0001042128603104213,
-      "loss": 0.9538,
+      "loss": 0.9594,
       "step": 222
     },
     {
       "epoch": 1.4681318681318682,
-      "grad_norm": 0.4653920829296112,
+      "grad_norm": 0.46859410405158997,
       "learning_rate": 0.00010376940133037693,
-      "loss": 0.8921,
+      "loss": 0.8952,
       "step": 223
     },
     {
       "epoch": 1.4747252747252748,
-      "grad_norm": 0.6153593063354492,
+      "grad_norm": 0.619542121887207,
       "learning_rate": 0.00010332594235033259,
-      "loss": 0.9249,
+      "loss": 0.9253,
       "step": 224
     },
     {
       "epoch": 1.4813186813186814,
-      "grad_norm": 0.41231054067611694,
+      "grad_norm": 0.41206881403923035,
       "learning_rate": 0.00010288248337028825,
-      "loss": 0.7862,
+      "loss": 0.7864,
       "step": 225
     },
     {
       "epoch": 1.487912087912088,
-      "grad_norm": 0.42926061153411865,
+      "grad_norm": 0.42798635363578796,
       "learning_rate": 0.0001024390243902439,
-      "loss": 0.6944,
+      "loss": 0.6949,
       "step": 226
     },
     {
       "epoch": 1.4945054945054945,
-      "grad_norm": 0.5749198198318481,
+      "grad_norm": 0.5727190971374512,
       "learning_rate": 0.00010199556541019956,
-      "loss": 0.8639,
+      "loss": 0.8635,
       "step": 227
     },
     {
       "epoch": 1.501098901098901,
-      "grad_norm": 0.5089406967163086,
+      "grad_norm": 0.5052634477615356,
       "learning_rate": 0.00010155210643015522,
-      "loss": 0.848,
+      "loss": 0.8481,
       "step": 228
     },
     {
       "epoch": 1.5076923076923077,
-      "grad_norm": 0.47087109088897705,
+      "grad_norm": 0.4678313732147217,
       "learning_rate": 0.00010110864745011087,
-      "loss": 0.7546,
+      "loss": 0.7565,
       "step": 229
     },
     {
       "epoch": 1.5142857142857142,
-      "grad_norm": 0.48008468747138977,
+      "grad_norm": 0.4803018867969513,
       "learning_rate": 0.00010066518847006652,
-      "loss": 0.7108,
+      "loss": 0.7101,
       "step": 230
     },
     {
       "epoch": 1.5208791208791208,
-      "grad_norm": 0.5517432689666748,
+      "grad_norm": 0.5546374917030334,
       "learning_rate": 0.00010022172949002218,
-      "loss": 0.7404,
+      "loss": 0.7391,
       "step": 231
     },
     {
       "epoch": 1.5274725274725274,
-      "grad_norm": 0.5300886631011963,
+      "grad_norm": 0.5278930068016052,
       "learning_rate": 9.977827050997783e-05,
-      "loss": 0.764,
+      "loss": 0.7634,
       "step": 232
     },
     {
       "epoch": 1.534065934065934,
-      "grad_norm": 0.5006000995635986,
+      "grad_norm": 0.5004315972328186,
       "learning_rate": 9.933481152993348e-05,
-      "loss": 0.7367,
+      "loss": 0.7371,
       "step": 233
     },
     {
       "epoch": 1.5406593406593405,
-      "grad_norm": 0.49519893527030945,
+      "grad_norm": 0.4936966598033905,
       "learning_rate": 9.889135254988914e-05,
-      "loss": 0.9038,
+      "loss": 0.901,
       "step": 234
     },
     {
       "epoch": 1.5472527472527473,
-      "grad_norm": 0.5353063344955444,
+      "grad_norm": 0.5347455739974976,
       "learning_rate": 9.84478935698448e-05,
-      "loss": 0.9174,
+      "loss": 0.9187,
       "step": 235
     },
     {
       "epoch": 1.5538461538461539,
-      "grad_norm": 0.4727020859718323,
+      "grad_norm": 0.4705422818660736,
       "learning_rate": 9.800443458980046e-05,
-      "loss": 0.859,
+      "loss": 0.8574,
       "step": 236
     },
     {
       "epoch": 1.5604395604395604,
-      "grad_norm": 0.4980611801147461,
+      "grad_norm": 0.4970325827598572,
       "learning_rate": 9.75609756097561e-05,
-      "loss": 1.0518,
+      "loss": 1.0511,
       "step": 237
     },
     {
       "epoch": 1.567032967032967,
-      "grad_norm": 0.4948464632034302,
+      "grad_norm": 0.49616602063179016,
       "learning_rate": 9.711751662971176e-05,
-      "loss": 0.8243,
+      "loss": 0.8263,
       "step": 238
     },
     {
       "epoch": 1.5736263736263736,
-      "grad_norm": 0.5270423889160156,
+      "grad_norm": 0.5309842228889465,
       "learning_rate": 9.667405764966742e-05,
-      "loss": 0.9291,
+      "loss": 0.9275,
       "step": 239
     },
     {
       "epoch": 1.5802197802197804,
-      "grad_norm": 0.5233663320541382,
+      "grad_norm": 0.5249683856964111,
       "learning_rate": 9.623059866962306e-05,
-      "loss": 1.0248,
+      "loss": 1.0206,
       "step": 240
     },
     {
       "epoch": 1.586813186813187,
-      "grad_norm": 0.590831458568573,
+      "grad_norm": 0.5909983515739441,
       "learning_rate": 9.578713968957872e-05,
-      "loss": 1.119,
+      "loss": 1.1204,
       "step": 241
     },
     {
       "epoch": 1.5934065934065935,
-      "grad_norm": 0.5016458034515381,
+      "grad_norm": 0.4994933605194092,
       "learning_rate": 9.534368070953438e-05,
-      "loss": 0.9389,
+      "loss": 0.9374,
       "step": 242
     },
     {
       "epoch": 1.6,
-      "grad_norm": 0.5507176518440247,
+      "grad_norm": 0.5519099235534668,
       "learning_rate": 9.490022172949002e-05,
-      "loss": 0.7676,
+      "loss": 0.7673,
       "step": 243
     },
     {
       "epoch": 1.6065934065934067,
-      "grad_norm": 0.49372318387031555,
+      "grad_norm": 0.49864885210990906,
       "learning_rate": 9.445676274944568e-05,
-      "loss": 0.8448,
+      "loss": 0.8424,
       "step": 244
     },
     {
       "epoch": 1.6131868131868132,
-      "grad_norm": 0.5039417147636414,
+      "grad_norm": 0.5097473859786987,
       "learning_rate": 9.401330376940134e-05,
-      "loss": 0.8028,
+      "loss": 0.803,
       "step": 245
     },
     {
       "epoch": 1.6197802197802198,
-      "grad_norm": 0.5067917704582214,
+      "grad_norm": 0.512746274471283,
       "learning_rate": 9.356984478935698e-05,
-      "loss": 0.8708,
+      "loss": 0.8722,
       "step": 246
     },
     {
       "epoch": 1.6263736263736264,
-      "grad_norm": 0.517314076423645,
+      "grad_norm": 0.5191940069198608,
       "learning_rate": 9.312638580931264e-05,
-      "loss": 0.748,
+      "loss": 0.7468,
       "step": 247
     },
     {
       "epoch": 1.632967032967033,
-      "grad_norm": 0.5289116501808167,
+      "grad_norm": 0.532005786895752,
       "learning_rate": 9.26829268292683e-05,
-      "loss": 0.9131,
+      "loss": 0.914,
       "step": 248
     },
     {
       "epoch": 1.6395604395604395,
-      "grad_norm": 0.515396773815155,
+      "grad_norm": 0.5155553221702576,
       "learning_rate": 9.223946784922394e-05,
-      "loss": 0.6893,
+      "loss": 0.6895,
       "step": 249
     },
     {
       "epoch": 1.646153846153846,
-      "grad_norm": 0.4606779217720032,
+      "grad_norm": 0.4564570486545563,
       "learning_rate": 9.17960088691796e-05,
-      "loss": 0.6963,
+      "loss": 0.6956,
       "step": 250
     },
     {
       "epoch": 1.6527472527472526,
-      "grad_norm": 0.626171350479126,
+      "grad_norm": 0.6313489079475403,
       "learning_rate": 9.135254988913526e-05,
-      "loss": 0.8437,
+      "loss": 0.8443,
       "step": 251
     },
     {
       "epoch": 1.6593406593406592,
-      "grad_norm": 0.5930360555648804,
+      "grad_norm": 0.5917752981185913,
       "learning_rate": 9.090909090909092e-05,
-      "loss": 0.818,
+      "loss": 0.8156,
       "step": 252
     },
     {
       "epoch": 1.6659340659340658,
-      "grad_norm": 0.5804259181022644,
+      "grad_norm": 0.5950708985328674,
       "learning_rate": 9.046563192904656e-05,
-      "loss": 0.9461,
+      "loss": 0.9472,
       "step": 253
     },
     {
       "epoch": 1.6725274725274726,
-      "grad_norm": 0.5499175190925598,
+      "grad_norm": 0.5504249930381775,
       "learning_rate": 9.002217294900222e-05,
-      "loss": 0.8106,
+      "loss": 0.8124,
       "step": 254
     },
     {
       "epoch": 1.6791208791208792,
-      "grad_norm": 0.3868570625782013,
+      "grad_norm": 0.3866679072380066,
       "learning_rate": 8.957871396895788e-05,
-      "loss": 0.6983,
+      "loss": 0.6982,
       "step": 255
     },
     {
       "epoch": 1.6857142857142857,
-      "grad_norm": 0.572151243686676,
+      "grad_norm": 0.5705950856208801,
       "learning_rate": 8.913525498891354e-05,
-      "loss": 0.7609,
+      "loss": 0.7596,
       "step": 256
     },
     {
       "epoch": 1.6923076923076923,
-      "grad_norm": 0.5721766948699951,
+      "grad_norm": 0.5793991088867188,
       "learning_rate": 8.869179600886919e-05,
-      "loss": 0.8263,
+      "loss": 0.8257,
       "step": 257
     },
     {
       "epoch": 1.6989010989010989,
-      "grad_norm": 0.5553361773490906,
+      "grad_norm": 0.5562020540237427,
       "learning_rate": 8.824833702882484e-05,
-      "loss": 0.9169,
+      "loss": 0.9161,
       "step": 258
     },
     {
       "epoch": 1.7054945054945057,
-      "grad_norm": 0.5333415865898132,
+      "grad_norm": 0.5355163812637329,
       "learning_rate": 8.78048780487805e-05,
-      "loss": 1.0914,
+      "loss": 1.0931,
       "step": 259
     },
     {
       "epoch": 1.7120879120879122,
-      "grad_norm": 0.44406670331954956,
+      "grad_norm": 0.4376404285430908,
       "learning_rate": 8.736141906873615e-05,
-      "loss": 0.7741,
+      "loss": 0.7723,
       "step": 260
     },
     {
       "epoch": 1.7186813186813188,
-      "grad_norm": 0.5063455700874329,
+      "grad_norm": 0.5053808093070984,
       "learning_rate": 8.69179600886918e-05,
-      "loss": 0.9668,
+      "loss": 0.9649,
       "step": 261
     },
     {
       "epoch": 1.7252747252747254,
-      "grad_norm": 0.5529189705848694,
+      "grad_norm": 0.5418540835380554,
       "learning_rate": 8.647450110864746e-05,
-      "loss": 0.9054,
+      "loss": 0.9049,
       "step": 262
     },
     {
       "epoch": 1.731868131868132,
-      "grad_norm": 0.5595840215682983,
+      "grad_norm": 0.557334303855896,
       "learning_rate": 8.603104212860311e-05,
-      "loss": 0.816,
+      "loss": 0.8178,
       "step": 263
     },
     {
       "epoch": 1.7384615384615385,
-      "grad_norm": 0.5390612483024597,
+      "grad_norm": 0.5328534245491028,
       "learning_rate": 8.558758314855876e-05,
-      "loss": 0.7387,
+      "loss": 0.7365,
       "step": 264
     },
     {
       "epoch": 1.745054945054945,
-      "grad_norm": 0.5740283727645874,
+      "grad_norm": 0.5694268941879272,
       "learning_rate": 8.514412416851442e-05,
-      "loss": 0.9647,
+      "loss": 0.962,
       "step": 265
     },
     {
       "epoch": 1.7516483516483516,
-      "grad_norm": 0.6255824565887451,
+      "grad_norm": 0.615943431854248,
       "learning_rate": 8.470066518847007e-05,
-      "loss": 1.01,
+      "loss": 1.0112,
       "step": 266
     },
     {
       "epoch": 1.7582417582417582,
-      "grad_norm": 0.5318125486373901,
+      "grad_norm": 0.5294637680053711,
       "learning_rate": 8.425720620842572e-05,
-      "loss": 0.8678,
+      "loss": 0.8686,
       "step": 267
     },
     {
       "epoch": 1.7648351648351648,
-      "grad_norm": 0.4187081754207611,
+      "grad_norm": 0.41722503304481506,
       "learning_rate": 8.381374722838138e-05,
-      "loss": 0.859,
+      "loss": 0.8581,
       "step": 268
     },
     {
       "epoch": 1.7714285714285714,
-      "grad_norm": 0.5770336985588074,
+      "grad_norm": 0.5750018954277039,
       "learning_rate": 8.337028824833703e-05,
-      "loss": 0.8747,
+      "loss": 0.8743,
       "step": 269
     },
     {
       "epoch": 1.778021978021978,
-      "grad_norm": 0.4633336067199707,
+      "grad_norm": 0.46019288897514343,
       "learning_rate": 8.292682926829268e-05,
-      "loss": 0.9147,
+      "loss": 0.9135,
       "step": 270
     },
     {
       "epoch": 1.7846153846153845,
-      "grad_norm": 0.4776274859905243,
+      "grad_norm": 0.47639933228492737,
       "learning_rate": 8.248337028824834e-05,
-      "loss": 0.9318,
+      "loss": 0.9316,
       "step": 271
     },
     {
       "epoch": 1.791208791208791,
-      "grad_norm": 0.7317682504653931,
+      "grad_norm": 0.7167646884918213,
       "learning_rate": 8.2039911308204e-05,
-      "loss": 0.9723,
+      "loss": 0.9709,
       "step": 272
     },
     {
       "epoch": 1.7978021978021979,
-      "grad_norm": 0.5217792987823486,
+      "grad_norm": 0.5192975401878357,
       "learning_rate": 8.159645232815965e-05,
-      "loss": 0.9407,
+      "loss": 0.9414,
       "step": 273
     },
     {
       "epoch": 1.8043956043956044,
-      "grad_norm": 0.5275446772575378,
+      "grad_norm": 0.5262181162834167,
       "learning_rate": 8.11529933481153e-05,
-      "loss": 0.8643,
+      "loss": 0.8627,
       "step": 274
     },
     {
       "epoch": 1.810989010989011,
-      "grad_norm": 0.669671356678009,
+      "grad_norm": 0.6696521043777466,
       "learning_rate": 8.070953436807095e-05,
-      "loss": 0.8596,
+      "loss": 0.8618,
       "step": 275
     },
     {
       "epoch": 1.8175824175824176,
-      "grad_norm": 0.5502275824546814,
+      "grad_norm": 0.5426575541496277,
       "learning_rate": 8.026607538802661e-05,
-      "loss": 0.8945,
+      "loss": 0.8942,
       "step": 276
     },
     {
       "epoch": 1.8241758241758241,
-      "grad_norm": 0.5572497248649597,
+      "grad_norm": 0.5545498132705688,
       "learning_rate": 7.982261640798227e-05,
-      "loss": 0.9932,
+      "loss": 0.9942,
       "step": 277
     },
     {
       "epoch": 1.830769230769231,
-      "grad_norm": 0.42105263471603394,
+      "grad_norm": 0.42044493556022644,
       "learning_rate": 7.937915742793793e-05,
-      "loss": 0.7142,
+      "loss": 0.7124,
       "step": 278
     },
     {
       "epoch": 1.8373626373626375,
-      "grad_norm": 0.5113288164138794,
+      "grad_norm": 0.5082975625991821,
       "learning_rate": 7.893569844789357e-05,
-      "loss": 0.7668,
+      "loss": 0.7645,
       "step": 279
     },
     {
       "epoch": 1.843956043956044,
-      "grad_norm": 0.4879838824272156,
+      "grad_norm": 0.489961713552475,
       "learning_rate": 7.849223946784923e-05,
-      "loss": 0.8119,
+      "loss": 0.812,
       "step": 280
     },
     {
       "epoch": 1.8505494505494506,
-      "grad_norm": 0.5138363242149353,
+      "grad_norm": 0.5158363580703735,
       "learning_rate": 7.804878048780489e-05,
-      "loss": 0.86,
+      "loss": 0.8612,
       "step": 281
     },
     {
       "epoch": 1.8571428571428572,
-      "grad_norm": 0.42918461561203003,
+      "grad_norm": 0.43079808354377747,
       "learning_rate": 7.760532150776053e-05,
-      "loss": 0.7694,
+      "loss": 0.7685,
       "step": 282
     },
     {
       "epoch": 1.8637362637362638,
-      "grad_norm": 0.5903877019882202,
+      "grad_norm": 0.5886262655258179,
       "learning_rate": 7.716186252771619e-05,
-      "loss": 0.8804,
+      "loss": 0.8799,
       "step": 283
     },
     {
       "epoch": 1.8703296703296703,
-      "grad_norm": 0.5936295986175537,
+      "grad_norm": 0.5958455801010132,
       "learning_rate": 7.671840354767185e-05,
-      "loss": 0.7532,
+      "loss": 0.7535,
       "step": 284
     },
     {
       "epoch": 1.876923076923077,
-      "grad_norm": 0.5549003481864929,
+      "grad_norm": 0.5544354319572449,
       "learning_rate": 7.627494456762749e-05,
-      "loss": 0.8225,
+      "loss": 0.8202,
       "step": 285
     },
     {
       "epoch": 1.8835164835164835,
-      "grad_norm": 0.45071983337402344,
+      "grad_norm": 0.4507337510585785,
       "learning_rate": 7.583148558758315e-05,
-      "loss": 0.9675,
+      "loss": 0.9666,
       "step": 286
     },
     {
       "epoch": 1.89010989010989,
-      "grad_norm": 0.5381602644920349,
+      "grad_norm": 0.537993311882019,
       "learning_rate": 7.538802660753881e-05,
-      "loss": 0.8273,
+      "loss": 0.8264,
       "step": 287
     },
     {
       "epoch": 1.8967032967032966,
-      "grad_norm": 0.5621278882026672,
+      "grad_norm": 0.5618338584899902,
       "learning_rate": 7.494456762749445e-05,
-      "loss": 0.8176,
+      "loss": 0.8165,
       "step": 288
     },
     {
       "epoch": 1.9032967032967032,
-      "grad_norm": 0.5810724496841431,
+      "grad_norm": 0.5855028629302979,
       "learning_rate": 7.450110864745011e-05,
-      "loss": 0.7506,
+      "loss": 0.7508,
       "step": 289
     },
     {
       "epoch": 1.9098901098901098,
-      "grad_norm": 0.5444163084030151,
+      "grad_norm": 0.5481427311897278,
       "learning_rate": 7.405764966740577e-05,
-      "loss": 0.7305,
+      "loss": 0.73,
       "step": 290
     },
     {
       "epoch": 1.9164835164835163,
-      "grad_norm": 0.5373480319976807,
+      "grad_norm": 0.5455793142318726,
       "learning_rate": 7.361419068736141e-05,
-      "loss": 0.8135,
+      "loss": 0.8251,
       "step": 291
     },
     {
       "epoch": 1.9230769230769231,
-      "grad_norm": 0.5446553230285645,
+      "grad_norm": 0.5479074716567993,
       "learning_rate": 7.317073170731707e-05,
-      "loss": 0.8316,
+      "loss": 0.8328,
       "step": 292
     },
     {
       "epoch": 1.9296703296703297,
-      "grad_norm": 0.5608210563659668,
+      "grad_norm": 0.558845579624176,
       "learning_rate": 7.272727272727273e-05,
-      "loss": 0.8076,
+      "loss": 0.8069,
       "step": 293
     },
     {
       "epoch": 1.9362637362637363,
-      "grad_norm": 0.5300861597061157,
+      "grad_norm": 0.5320450663566589,
       "learning_rate": 7.228381374722837e-05,
-      "loss": 0.8772,
+      "loss": 0.8758,
       "step": 294
     },
     {
       "epoch": 1.9428571428571428,
-      "grad_norm": 0.5608508586883545,
+      "grad_norm": 0.56231689453125,
       "learning_rate": 7.184035476718403e-05,
-      "loss": 0.7926,
+      "loss": 0.7928,
       "step": 295
     },
     {
       "epoch": 1.9494505494505494,
-      "grad_norm": 0.6337592601776123,
+      "grad_norm": 0.6288520097732544,
       "learning_rate": 7.139689578713969e-05,
-      "loss": 0.9388,
+      "loss": 0.9385,
       "step": 296
     },
     {
       "epoch": 1.9560439560439562,
-      "grad_norm": 0.6125892996788025,
+      "grad_norm": 0.614974856376648,
       "learning_rate": 7.095343680709535e-05,
-      "loss": 0.9204,
+      "loss": 0.9189,
       "step": 297
     },
     {
       "epoch": 1.9626373626373628,
-      "grad_norm": 0.5068105459213257,
+      "grad_norm": 0.5072554349899292,
       "learning_rate": 7.0509977827051e-05,
-      "loss": 0.8169,
+      "loss": 0.818,
       "step": 298
     },
     {
       "epoch": 1.9692307692307693,
-      "grad_norm": 0.6008275151252747,
+      "grad_norm": 0.6028959155082703,
       "learning_rate": 7.006651884700666e-05,
-      "loss": 0.803,
+      "loss": 0.8039,
       "step": 299
     },
     {
       "epoch": 1.975824175824176,
-      "grad_norm": 0.5600306987762451,
+      "grad_norm": 0.561110258102417,
       "learning_rate": 6.962305986696232e-05,
-      "loss": 0.6676,
+      "loss": 0.667,
       "step": 300
     },
     {
       "epoch": 1.9824175824175825,
-      "grad_norm": 0.5443146824836731,
+      "grad_norm": 0.5425792932510376,
       "learning_rate": 6.917960088691796e-05,
-      "loss": 0.802,
+      "loss": 0.7988,
       "step": 301
     },
     {
       "epoch": 1.989010989010989,
-      "grad_norm": 0.43961313366889954,
+      "grad_norm": 0.4379030168056488,
       "learning_rate": 6.873614190687362e-05,
-      "loss": 0.7341,
+      "loss": 0.7336,
       "step": 302
     },
     {
       "epoch": 1.9956043956043956,
-      "grad_norm": 0.48595428466796875,
+      "grad_norm": 0.4857703447341919,
       "learning_rate": 6.829268292682928e-05,
-      "loss": 0.7199,
+      "loss": 0.7181,
       "step": 303
     },
     {
       "epoch": 2.0,
-      "grad_norm": 0.9165866374969482,
+      "grad_norm": 0.9238166809082031,
       "learning_rate": 6.784922394678492e-05,
-      "loss": 0.9223,
+      "loss": 0.9261,
       "step": 304
     },
     {
       "epoch": 2.0065934065934066,
-      "grad_norm": 0.5227509140968323,
+      "grad_norm": 0.5198847651481628,
       "learning_rate": 6.740576496674058e-05,
-      "loss": 0.7765,
+      "loss": 0.7753,
       "step": 305
     },
     {
       "epoch": 2.013186813186813,
-      "grad_norm": 0.6080646514892578,
+      "grad_norm": 0.6053488254547119,
       "learning_rate": 6.696230598669624e-05,
-      "loss": 0.8298,
+      "loss": 0.8302,
       "step": 306
     },
     {
       "epoch": 2.0197802197802197,
-      "grad_norm": 0.5301607847213745,
+      "grad_norm": 0.5359417200088501,
       "learning_rate": 6.651884700665188e-05,
-      "loss": 0.6396,
+      "loss": 0.6405,
       "step": 307
     },
     {
       "epoch": 2.0263736263736263,
-      "grad_norm": 0.5521807074546814,
+      "grad_norm": 0.554151713848114,
       "learning_rate": 6.607538802660754e-05,
-      "loss": 0.6587,
+      "loss": 0.6575,
       "step": 308
     },
     {
       "epoch": 2.032967032967033,
-      "grad_norm": 0.5167801976203918,
+      "grad_norm": 0.5202828645706177,
       "learning_rate": 6.56319290465632e-05,
-      "loss": 0.8271,
+      "loss": 0.8267,
       "step": 309
     },
     {
       "epoch": 2.0395604395604394,
-      "grad_norm": 0.5947807431221008,
+      "grad_norm": 0.597801923751831,
       "learning_rate": 6.518847006651884e-05,
-      "loss": 0.8893,
+      "loss": 0.8906,
       "step": 310
     },
     {
       "epoch": 2.046153846153846,
-      "grad_norm": 0.5799498558044434,
+      "grad_norm": 0.5805016160011292,
       "learning_rate": 6.47450110864745e-05,
       "loss": 0.6536,
       "step": 311
     },
     {
       "epoch": 2.0527472527472526,
-      "grad_norm": 0.5995151996612549,
+      "grad_norm": 0.5974270701408386,
       "learning_rate": 6.430155210643016e-05,
-      "loss": 0.6882,
+      "loss": 0.6886,
       "step": 312
     },
     {
       "epoch": 2.059340659340659,
-      "grad_norm": 0.528610348701477,
+      "grad_norm": 0.5277975797653198,
       "learning_rate": 6.38580931263858e-05,
-      "loss": 0.5904,
+      "loss": 0.5901,
       "step": 313
     },
     {
       "epoch": 2.065934065934066,
-      "grad_norm": 0.44652125239372253,
+      "grad_norm": 0.4494751989841461,
       "learning_rate": 6.341463414634146e-05,
-      "loss": 0.6652,
+      "loss": 0.6647,
       "step": 314
     },
     {
       "epoch": 2.0725274725274727,
-      "grad_norm": 0.46826040744781494,
+      "grad_norm": 0.46326035261154175,
       "learning_rate": 6.297117516629712e-05,
-      "loss": 0.6307,
+      "loss": 0.6292,
       "step": 315
     },
     {
       "epoch": 2.0791208791208793,
-      "grad_norm": 0.7317270636558533,
+      "grad_norm": 0.7360116243362427,
       "learning_rate": 6.252771618625277e-05,
-      "loss": 0.7194,
+      "loss": 0.7179,
       "step": 316
     },
     {
       "epoch": 2.085714285714286,
-      "grad_norm": 0.5108973979949951,
+      "grad_norm": 0.5109550356864929,
       "learning_rate": 6.208425720620842e-05,
-      "loss": 0.5204,
+      "loss": 0.52,
       "step": 317
     },
     {
       "epoch": 2.0923076923076924,
-      "grad_norm": 0.5635854005813599,
+      "grad_norm": 0.5595861077308655,
       "learning_rate": 6.164079822616408e-05,
-      "loss": 0.8943,
+      "loss": 0.8949,
       "step": 318
     },
     {
       "epoch": 2.098901098901099,
-      "grad_norm": 0.48101961612701416,
+      "grad_norm": 0.47883766889572144,
       "learning_rate": 6.119733924611974e-05,
-      "loss": 0.6351,
+      "loss": 0.6334,
       "step": 319
     },
     {
       "epoch": 2.1054945054945056,
-      "grad_norm": 0.7315899133682251,
+      "grad_norm": 0.7345694303512573,
       "learning_rate": 6.075388026607539e-05,
-      "loss": 0.8634,
+      "loss": 0.862,
       "step": 320
     },
     {
       "epoch": 2.112087912087912,
-      "grad_norm": 0.6516451835632324,
+      "grad_norm": 0.657485842704773,
       "learning_rate": 6.031042128603105e-05,
-      "loss": 0.7284,
+      "loss": 0.7304,
       "step": 321
     },
     {
       "epoch": 2.1186813186813187,
-      "grad_norm": 0.7066317796707153,
+      "grad_norm": 0.706637978553772,
       "learning_rate": 5.98669623059867e-05,
-      "loss": 0.8358,
+      "loss": 0.8356,
       "step": 322
     },
     {
       "epoch": 2.1252747252747253,
-      "grad_norm": 0.6052649021148682,
+      "grad_norm": 0.6115225553512573,
       "learning_rate": 5.942350332594236e-05,
-      "loss": 0.8006,
+      "loss": 0.8008,
       "step": 323
     },
     {
       "epoch": 2.131868131868132,
-      "grad_norm": 0.5642592310905457,
+      "grad_norm": 0.5652472376823425,
       "learning_rate": 5.898004434589801e-05,
-      "loss": 0.7239,
+      "loss": 0.7252,
       "step": 324
     },
     {
       "epoch": 2.1384615384615384,
-      "grad_norm": 0.6396430134773254,
+      "grad_norm": 0.6393257975578308,
       "learning_rate": 5.853658536585366e-05,
-      "loss": 0.9395,
+      "loss": 0.9396,
       "step": 325
     },
     {
       "epoch": 2.145054945054945,
-      "grad_norm": 0.6785988807678223,
+      "grad_norm": 0.6815609335899353,
       "learning_rate": 5.809312638580932e-05,
-      "loss": 0.6952,
+      "loss": 0.6955,
       "step": 326
     },
     {
       "epoch": 2.1516483516483516,
-      "grad_norm": 0.6331364512443542,
+      "grad_norm": 0.6356279253959656,
       "learning_rate": 5.7649667405764975e-05,
-      "loss": 0.6753,
+      "loss": 0.675,
       "step": 327
     },
     {
       "epoch": 2.158241758241758,
-      "grad_norm": 0.514786422252655,
+      "grad_norm": 0.5139729976654053,
       "learning_rate": 5.720620842572062e-05,
-      "loss": 0.6626,
+      "loss": 0.6628,
       "step": 328
     },
     {
       "epoch": 2.1648351648351647,
-      "grad_norm": 0.6315619349479675,
+      "grad_norm": 0.6320257782936096,
       "learning_rate": 5.676274944567628e-05,
-      "loss": 0.6879,
+      "loss": 0.6855,
       "step": 329
     },
     {
       "epoch": 2.1714285714285713,
-      "grad_norm": 0.7786369323730469,
+      "grad_norm": 0.9928913712501526,
       "learning_rate": 5.6319290465631935e-05,
-      "loss": 1.0007,
+      "loss": 1.0035,
       "step": 330
     },
     {
       "epoch": 2.178021978021978,
-      "grad_norm": 0.6195600032806396,
+      "grad_norm": 0.620817244052887,
       "learning_rate": 5.587583148558758e-05,
-      "loss": 0.6651,
+      "loss": 0.6642,
       "step": 331
     },
     {
       "epoch": 2.184615384615385,
-      "grad_norm": 0.7224487662315369,
+      "grad_norm": 0.7170416116714478,
       "learning_rate": 5.543237250554324e-05,
-      "loss": 0.6863,
+      "loss": 0.6867,
       "step": 332
     },
     {
       "epoch": 2.1912087912087914,
-      "grad_norm": 0.5404961109161377,
+      "grad_norm": 0.5419169664382935,
       "learning_rate": 5.4988913525498895e-05,
-      "loss": 0.7605,
+      "loss": 0.7599,
       "step": 333
     },
     {
       "epoch": 2.197802197802198,
-      "grad_norm": 0.6667106747627258,
+      "grad_norm": 0.6690497994422913,
       "learning_rate": 5.4545454545454546e-05,
-      "loss": 0.696,
+      "loss": 0.6972,
       "step": 334
     },
     {
       "epoch": 2.2043956043956046,
-      "grad_norm": 0.6088030338287354,
+      "grad_norm": 0.6096978187561035,
       "learning_rate": 5.4101995565410204e-05,
-      "loss": 0.7073,
+      "loss": 0.7072,
       "step": 335
     },
     {
       "epoch": 2.210989010989011,
-      "grad_norm": 0.6480793952941895,
+      "grad_norm": 0.6517192125320435,
       "learning_rate": 5.365853658536586e-05,
-      "loss": 0.8875,
+      "loss": 0.8895,
       "step": 336
     },
     {
       "epoch": 2.2175824175824177,
-      "grad_norm": 0.724563717842102,
+      "grad_norm": 0.724141001701355,
       "learning_rate": 5.3215077605321506e-05,
-      "loss": 0.7546,
+      "loss": 0.7545,
       "step": 337
     },
     {
       "epoch": 2.2241758241758243,
-      "grad_norm": 0.6789265275001526,
+      "grad_norm": 0.6749389171600342,
       "learning_rate": 5.2771618625277164e-05,
-      "loss": 0.7924,
+      "loss": 0.7943,
       "step": 338
     },
     {
       "epoch": 2.230769230769231,
-      "grad_norm": 0.5973027944564819,
+      "grad_norm": 0.5959587693214417,
       "learning_rate": 5.232815964523282e-05,
-      "loss": 0.775,
+      "loss": 0.7744,
       "step": 339
     },
     {
       "epoch": 2.2373626373626374,
-      "grad_norm": 0.734237790107727,
+      "grad_norm": 0.7323570847511292,
       "learning_rate": 5.1884700665188466e-05,
-      "loss": 0.6643,
+      "loss": 0.6648,
       "step": 340
     },
     {
       "epoch": 2.243956043956044,
-      "grad_norm": 0.6669371724128723,
+      "grad_norm": 0.6649282574653625,
       "learning_rate": 5.1441241685144124e-05,
-      "loss": 0.7901,
+      "loss": 0.7913,
       "step": 341
     },
     {
       "epoch": 2.2505494505494505,
-      "grad_norm": 0.6581075191497803,
+      "grad_norm": 0.6598180532455444,
       "learning_rate": 5.099778270509978e-05,
-      "loss": 0.7027,
+      "loss": 0.7004,
       "step": 342
     },
     {
       "epoch": 2.257142857142857,
-      "grad_norm": 0.8296685814857483,
+      "grad_norm": 0.8298692107200623,
       "learning_rate": 5.0554323725055433e-05,
-      "loss": 0.837,
+      "loss": 0.8346,
       "step": 343
     },
     {
       "epoch": 2.2637362637362637,
-      "grad_norm": 0.6875712275505066,
+      "grad_norm": 0.6844825744628906,
       "learning_rate": 5.011086474501109e-05,
-      "loss": 0.8774,
+      "loss": 0.8751,
       "step": 344
     },
     {
       "epoch": 2.2703296703296703,
-      "grad_norm": 0.6138696670532227,
+      "grad_norm": 0.611853301525116,
       "learning_rate": 4.966740576496674e-05,
-      "loss": 0.7283,
+      "loss": 0.7295,
       "step": 345
     },
     {
       "epoch": 2.276923076923077,
-      "grad_norm": 0.6708604097366333,
+      "grad_norm": 0.671319305896759,
       "learning_rate": 4.92239467849224e-05,
-      "loss": 0.7499,
+      "loss": 0.7503,
       "step": 346
     },
     {
       "epoch": 2.2835164835164834,
-      "grad_norm": 0.7797860503196716,
+      "grad_norm": 0.7785210609436035,
       "learning_rate": 4.878048780487805e-05,
-      "loss": 0.7742,
+      "loss": 0.776,
       "step": 347
     },
     {
       "epoch": 2.29010989010989,
-      "grad_norm": 0.5669825077056885,
+      "grad_norm": 0.5678278207778931,
       "learning_rate": 4.833702882483371e-05,
-      "loss": 0.5829,
+      "loss": 0.5836,
       "step": 348
     },
     {
       "epoch": 2.2967032967032965,
-      "grad_norm": 0.6981120109558105,
+      "grad_norm": 0.7011044025421143,
       "learning_rate": 4.789356984478936e-05,
-      "loss": 0.7853,
+      "loss": 0.7855,
       "step": 349
     },
     {
       "epoch": 2.303296703296703,
-      "grad_norm": 0.7048529982566833,
+      "grad_norm": 0.7039121389389038,
       "learning_rate": 4.745011086474501e-05,
-      "loss": 0.8974,
+      "loss": 0.8968,
       "step": 350
     },
     {
       "epoch": 2.3098901098901097,
-      "grad_norm": 0.6162217259407043,
+      "grad_norm": 0.6213542819023132,
       "learning_rate": 4.700665188470067e-05,
-      "loss": 0.6141,
+      "loss": 0.6145,
       "step": 351
     },
     {
       "epoch": 2.3164835164835162,
-      "grad_norm": 0.58530592918396,
+      "grad_norm": 0.5828162431716919,
       "learning_rate": 4.656319290465632e-05,
-      "loss": 0.657,
+      "loss": 0.6545,
       "step": 352
     },
     {
       "epoch": 2.3230769230769233,
-      "grad_norm": 0.694770336151123,
+      "grad_norm": 0.6973269581794739,
       "learning_rate": 4.611973392461197e-05,
-      "loss": 0.7779,
+      "loss": 0.7797,
       "step": 353
     },
     {
       "epoch": 2.32967032967033,
-      "grad_norm": 0.7460579872131348,
+      "grad_norm": 0.739226222038269,
       "learning_rate": 4.567627494456763e-05,
-      "loss": 0.6942,
+      "loss": 0.6918,
       "step": 354
     },
     {
       "epoch": 2.3362637362637364,
-      "grad_norm": 0.6448771953582764,
+      "grad_norm": 0.6460672616958618,
       "learning_rate": 4.523281596452328e-05,
-      "loss": 0.837,
+      "loss": 0.8387,
       "step": 355
     },
     {
       "epoch": 2.342857142857143,
-      "grad_norm": 0.7290218472480774,
+      "grad_norm": 0.7241936326026917,
       "learning_rate": 4.478935698447894e-05,
-      "loss": 0.6104,
+      "loss": 0.609,
       "step": 356
     },
     {
       "epoch": 2.3494505494505495,
-      "grad_norm": 0.674991250038147,
+      "grad_norm": 0.6753931641578674,
       "learning_rate": 4.4345898004434597e-05,
-      "loss": 0.8655,
+      "loss": 0.8636,
       "step": 357
     },
     {
       "epoch": 2.356043956043956,
-      "grad_norm": 0.6900389194488525,
+      "grad_norm": 0.6909801959991455,
       "learning_rate": 4.390243902439025e-05,
-      "loss": 0.7315,
+      "loss": 0.7321,
       "step": 358
     },
     {
       "epoch": 2.3626373626373627,
-      "grad_norm": 0.666122555732727,
+      "grad_norm": 0.6674776077270508,
       "learning_rate": 4.34589800443459e-05,
-      "loss": 0.5603,
+      "loss": 0.5621,
       "step": 359
     },
     {
       "epoch": 2.3692307692307693,
-      "grad_norm": 0.6514036059379578,
+      "grad_norm": 0.6482294797897339,
       "learning_rate": 4.301552106430156e-05,
-      "loss": 0.6701,
+      "loss": 0.6706,
       "step": 360
     },
     {
       "epoch": 2.375824175824176,
-      "grad_norm": 0.7242164611816406,
+      "grad_norm": 0.7250370979309082,
       "learning_rate": 4.257206208425721e-05,
-      "loss": 0.732,
+      "loss": 0.7313,
       "step": 361
     },
     {
       "epoch": 2.3824175824175824,
-      "grad_norm": 0.7702664136886597,
+      "grad_norm": 0.7682430148124695,
       "learning_rate": 4.212860310421286e-05,
-      "loss": 0.6786,
+      "loss": 0.6804,
       "step": 362
     },
     {
       "epoch": 2.389010989010989,
-      "grad_norm": 0.7631247043609619,
+      "grad_norm": 0.7638933658599854,
       "learning_rate": 4.168514412416852e-05,
-      "loss": 0.7601,
+      "loss": 0.7604,
       "step": 363
     },
     {
       "epoch": 2.3956043956043955,
-      "grad_norm": 0.6547625064849854,
+      "grad_norm": 0.6588174104690552,
       "learning_rate": 4.124168514412417e-05,
-      "loss": 0.7012,
+      "loss": 0.7018,
       "step": 364
     },
     {
       "epoch": 2.402197802197802,
-      "grad_norm": 0.8105188608169556,
+      "grad_norm": 0.8067613840103149,
       "learning_rate": 4.0798226164079826e-05,
-      "loss": 0.7047,
+      "loss": 0.7052,
       "step": 365
     },
     {
       "epoch": 2.4087912087912087,
-      "grad_norm": 0.5581157207489014,
+      "grad_norm": 0.5554311871528625,
       "learning_rate": 4.035476718403548e-05,
-      "loss": 0.627,
+      "loss": 0.6261,
       "step": 366
     },
     {
       "epoch": 2.4153846153846152,
-      "grad_norm": 0.6693578362464905,
+      "grad_norm": 0.6679801940917969,
       "learning_rate": 3.9911308203991135e-05,
-      "loss": 0.5234,
+      "loss": 0.523,
       "step": 367
     },
     {
       "epoch": 2.421978021978022,
-      "grad_norm": 0.7053078413009644,
+      "grad_norm": 0.7038661241531372,
       "learning_rate": 3.9467849223946786e-05,
-      "loss": 0.6758,
+      "loss": 0.6777,
       "step": 368
     },
     {
       "epoch": 2.4285714285714284,
-      "grad_norm": 0.7145379781723022,
+      "grad_norm": 0.7129457592964172,
       "learning_rate": 3.9024390243902444e-05,
-      "loss": 0.6681,
+      "loss": 0.6649,
       "step": 369
     },
     {
       "epoch": 2.4351648351648354,
-      "grad_norm": 0.763256847858429,
+      "grad_norm": 0.7614677548408508,
       "learning_rate": 3.8580931263858095e-05,
-      "loss": 0.8974,
+      "loss": 0.897,
       "step": 370
     },
     {
       "epoch": 2.441758241758242,
-      "grad_norm": 0.7621896862983704,
+      "grad_norm": 0.7623000741004944,
       "learning_rate": 3.8137472283813746e-05,
-      "loss": 0.7955,
+      "loss": 0.7936,
       "step": 371
     },
     {
       "epoch": 2.4483516483516485,
-      "grad_norm": 0.7032443284988403,
+      "grad_norm": 0.7011579871177673,
       "learning_rate": 3.7694013303769404e-05,
-      "loss": 0.778,
+      "loss": 0.7764,
       "step": 372
     },
     {
       "epoch": 2.454945054945055,
-      "grad_norm": 0.5998260974884033,
+      "grad_norm": 0.5975192189216614,
       "learning_rate": 3.7250554323725055e-05,
-      "loss": 0.7306,
+      "loss": 0.728,
       "step": 373
     },
     {
       "epoch": 2.4615384615384617,
-      "grad_norm": 0.5985954403877258,
+      "grad_norm": 0.6005866527557373,
       "learning_rate": 3.6807095343680706e-05,
-      "loss": 0.6777,
+      "loss": 0.6774,
       "step": 374
     },
     {
       "epoch": 2.4681318681318682,
-      "grad_norm": 0.7410600781440735,
+      "grad_norm": 0.7338082790374756,
       "learning_rate": 3.6363636363636364e-05,
-      "loss": 0.8438,
+      "loss": 0.8431,
       "step": 375
     },
     {
       "epoch": 2.474725274725275,
-      "grad_norm": 0.5700070858001709,
+      "grad_norm": 0.5733476877212524,
       "learning_rate": 3.5920177383592015e-05,
-      "loss": 0.9176,
+      "loss": 0.9186,
       "step": 376
     },
     {
       "epoch": 2.4813186813186814,
-      "grad_norm": 0.8185898065567017,
+      "grad_norm": 0.8242601752281189,
       "learning_rate": 3.547671840354767e-05,
-      "loss": 0.7588,
+      "loss": 0.7575,
       "step": 377
     },
     {
       "epoch": 2.487912087912088,
-      "grad_norm": 0.6818128228187561,
+      "grad_norm": 0.6773817539215088,
       "learning_rate": 3.503325942350333e-05,
-      "loss": 0.7676,
+      "loss": 0.7657,
       "step": 378
     },
     {
       "epoch": 2.4945054945054945,
-      "grad_norm": 0.7490910291671753,
+      "grad_norm": 0.7502856254577637,
       "learning_rate": 3.458980044345898e-05,
-      "loss": 0.7052,
+      "loss": 0.7035,
       "step": 379
     },
     {
       "epoch": 2.501098901098901,
-      "grad_norm": 0.7682050466537476,
+      "grad_norm": 0.7681854367256165,
       "learning_rate": 3.414634146341464e-05,
-      "loss": 0.8888,
+      "loss": 0.8887,
       "step": 380
     },
     {
       "epoch": 2.5076923076923077,
-      "grad_norm": 0.8988820910453796,
+      "grad_norm": 0.9010462760925293,
       "learning_rate": 3.370288248337029e-05,
-      "loss": 0.8517,
+      "loss": 0.8513,
       "step": 381
     },
     {
       "epoch": 2.5142857142857142,
-      "grad_norm": 0.7269229292869568,
+      "grad_norm": 0.7274242043495178,
       "learning_rate": 3.325942350332594e-05,
-      "loss": 0.864,
+      "loss": 0.865,
       "step": 382
     },
     {
       "epoch": 2.520879120879121,
-      "grad_norm": 0.688852071762085,
+      "grad_norm": 0.689725935459137,
       "learning_rate": 3.28159645232816e-05,
-      "loss": 0.6955,
+      "loss": 0.6951,
       "step": 383
     },
     {
       "epoch": 2.5274725274725274,
-      "grad_norm": 0.7329714894294739,
+      "grad_norm": 0.7339224219322205,
       "learning_rate": 3.237250554323725e-05,
-      "loss": 0.6581,
+      "loss": 0.6566,
       "step": 384
     },
     {
       "epoch": 2.534065934065934,
-      "grad_norm": 0.7130051255226135,
+      "grad_norm": 0.7123922109603882,
       "learning_rate": 3.19290465631929e-05,
-      "loss": 0.7039,
+      "loss": 0.704,
       "step": 385
     },
     {
       "epoch": 2.5406593406593405,
-      "grad_norm": 0.735274076461792,
+      "grad_norm": 0.7392660975456238,
       "learning_rate": 3.148558758314856e-05,
-      "loss": 0.7355,
+      "loss": 0.7344,
       "step": 386
     },
     {
       "epoch": 2.547252747252747,
-      "grad_norm": 0.8151843547821045,
+      "grad_norm": 0.8106175661087036,
       "learning_rate": 3.104212860310421e-05,
-      "loss": 0.7484,
+      "loss": 0.7474,
       "step": 387
     },
     {
       "epoch": 2.5538461538461537,
-      "grad_norm": 0.6167259216308594,
+      "grad_norm": 0.6163520812988281,
       "learning_rate": 3.059866962305987e-05,
-      "loss": 0.7218,
+      "loss": 0.7197,
       "step": 388
     },
     {
       "epoch": 2.5604395604395602,
-      "grad_norm": 0.7514891028404236,
+      "grad_norm": 0.7567664980888367,
       "learning_rate": 3.0155210643015524e-05,
-      "loss": 0.7335,
+      "loss": 0.7359,
       "step": 389
     },
     {
       "epoch": 2.567032967032967,
-      "grad_norm": 0.8656178712844849,
+      "grad_norm": 0.8751924633979797,
       "learning_rate": 2.971175166297118e-05,
-      "loss": 0.8249,
+      "loss": 0.8216,
       "step": 390
     },
     {
       "epoch": 2.5736263736263734,
-      "grad_norm": 0.7031922936439514,
+      "grad_norm": 0.7038911581039429,
       "learning_rate": 2.926829268292683e-05,
-      "loss": 0.6533,
+      "loss": 0.6534,
       "step": 391
     },
     {
       "epoch": 2.5802197802197804,
-      "grad_norm": 0.7793689370155334,
+      "grad_norm": 0.7836447954177856,
       "learning_rate": 2.8824833702882487e-05,
-      "loss": 0.6836,
+      "loss": 0.6819,
       "step": 392
     },
     {
       "epoch": 2.586813186813187,
-      "grad_norm": 0.7198311686515808,
+      "grad_norm": 0.7203120589256287,
       "learning_rate": 2.838137472283814e-05,
-      "loss": 0.7065,
+      "loss": 0.7075,
       "step": 393
     },
     {
       "epoch": 2.5934065934065935,
-      "grad_norm": 0.7807820439338684,
+      "grad_norm": 0.7772233486175537,
       "learning_rate": 2.793791574279379e-05,
-      "loss": 0.6903,
+      "loss": 0.6887,
       "step": 394
     },
     {
       "epoch": 2.6,
-      "grad_norm": 0.6784717440605164,
+      "grad_norm": 0.6759134531021118,
       "learning_rate": 2.7494456762749448e-05,
-      "loss": 0.6866,
+      "loss": 0.6859,
       "step": 395
     },
     {
       "epoch": 2.6065934065934067,
-      "grad_norm": 0.7901123762130737,
+      "grad_norm": 0.7961100339889526,
       "learning_rate": 2.7050997782705102e-05,
-      "loss": 0.7878,
+      "loss": 0.7887,
       "step": 396
     },
     {
       "epoch": 2.6131868131868132,
-      "grad_norm": 0.6011061072349548,
+      "grad_norm": 0.5994078516960144,
       "learning_rate": 2.6607538802660753e-05,
-      "loss": 0.59,
+      "loss": 0.5902,
       "step": 397
     },
     {
       "epoch": 2.61978021978022,
-      "grad_norm": 0.7180050611495972,
+      "grad_norm": 0.7192034721374512,
       "learning_rate": 2.616407982261641e-05,
-      "loss": 0.7138,
+      "loss": 0.7135,
       "step": 398
     },
     {
       "epoch": 2.6263736263736264,
-      "grad_norm": 0.6538064479827881,
+      "grad_norm": 0.6578332185745239,
       "learning_rate": 2.5720620842572062e-05,
-      "loss": 0.6557,
+      "loss": 0.6544,
       "step": 399
     },
     {
       "epoch": 2.632967032967033,
-      "grad_norm": 0.7195414900779724,
+      "grad_norm": 0.7176105976104736,
       "learning_rate": 2.5277161862527717e-05,
-      "loss": 0.8531,
+      "loss": 0.8515,
       "step": 400
     },
     {
       "epoch": 2.6395604395604395,
-      "grad_norm": 0.7028957605361938,
+      "grad_norm": 0.7048030495643616,
       "learning_rate": 2.483370288248337e-05,
-      "loss": 0.7554,
+      "loss": 0.756,
       "step": 401
     },
     {
       "epoch": 2.646153846153846,
-      "grad_norm": 0.8164121508598328,
+      "grad_norm": 0.8234903216362,
       "learning_rate": 2.4390243902439026e-05,
-      "loss": 0.669,
+      "loss": 0.67,
       "step": 402
     },
     {
       "epoch": 2.6527472527472526,
-      "grad_norm": 0.6855744123458862,
+      "grad_norm": 0.6862563490867615,
       "learning_rate": 2.394678492239468e-05,
-      "loss": 0.668,
+      "loss": 0.6706,
       "step": 403
     },
     {
       "epoch": 2.659340659340659,
-      "grad_norm": 0.6195077300071716,
+      "grad_norm": 0.6193423867225647,
       "learning_rate": 2.3503325942350335e-05,
-      "loss": 0.5571,
+      "loss": 0.5555,
       "step": 404
     },
     {
       "epoch": 2.665934065934066,
-      "grad_norm": 0.7968306541442871,
+      "grad_norm": 0.7917978167533875,
       "learning_rate": 2.3059866962305986e-05,
-      "loss": 0.7287,
+      "loss": 0.727,
       "step": 405
     },
     {
       "epoch": 2.672527472527473,
-      "grad_norm": 0.659389317035675,
+      "grad_norm": 0.6611487865447998,
       "learning_rate": 2.261640798226164e-05,
-      "loss": 0.6338,
+      "loss": 0.6348,
       "step": 406
     },
     {
       "epoch": 2.6791208791208794,
-      "grad_norm": 0.594168484210968,
+      "grad_norm": 0.5944433808326721,
       "learning_rate": 2.2172949002217298e-05,
-      "loss": 0.6009,
+      "loss": 0.6005,
       "step": 407
     },
     {
       "epoch": 2.685714285714286,
-      "grad_norm": 0.6932500004768372,
+      "grad_norm": 0.6937222480773926,
       "learning_rate": 2.172949002217295e-05,
-      "loss": 0.7241,
+      "loss": 0.7213,
       "step": 408
     },
     {
       "epoch": 2.6923076923076925,
-      "grad_norm": 0.6755253672599792,
+      "grad_norm": 0.6782053112983704,
       "learning_rate": 2.1286031042128604e-05,
-      "loss": 0.6216,
+      "loss": 0.6215,
       "step": 409
     },
     {
       "epoch": 2.698901098901099,
-      "grad_norm": 0.7545448541641235,
+      "grad_norm": 0.7478845715522766,
       "learning_rate": 2.084257206208426e-05,
-      "loss": 0.5814,
+      "loss": 0.5813,
       "step": 410
     },
     {
       "epoch": 2.7054945054945057,
-      "grad_norm": 0.636264443397522,
+      "grad_norm": 0.638687789440155,
       "learning_rate": 2.0399113082039913e-05,
-      "loss": 0.8003,
+      "loss": 0.8007,
       "step": 411
     },
     {
       "epoch": 2.7120879120879122,
-      "grad_norm": 0.6745185852050781,
+      "grad_norm": 0.6766836643218994,
       "learning_rate": 1.9955654101995567e-05,
-      "loss": 0.6301,
+      "loss": 0.6302,
       "step": 412
     },
     {
       "epoch": 2.718681318681319,
-      "grad_norm": 0.6747480630874634,
+      "grad_norm": 0.6753154397010803,
       "learning_rate": 1.9512195121951222e-05,
-      "loss": 0.7168,
+      "loss": 0.7166,
       "step": 413
     },
     {
       "epoch": 2.7252747252747254,
-      "grad_norm": 0.7580742835998535,
+      "grad_norm": 0.7590558528900146,
       "learning_rate": 1.9068736141906873e-05,
-      "loss": 0.9029,
+      "loss": 0.9045,
       "step": 414
     },
     {
       "epoch": 2.731868131868132,
-      "grad_norm": 0.7952632308006287,
+      "grad_norm": 0.7930037975311279,
       "learning_rate": 1.8625277161862528e-05,
-      "loss": 0.691,
+      "loss": 0.6906,
       "step": 415
     },
     {
       "epoch": 2.7384615384615385,
-      "grad_norm": 0.7883942723274231,
+      "grad_norm": 0.788776695728302,
       "learning_rate": 1.8181818181818182e-05,
-      "loss": 0.6352,
+      "loss": 0.6344,
       "step": 416
     },
     {
       "epoch": 2.745054945054945,
-      "grad_norm": 0.7801777124404907,
+      "grad_norm": 0.7762596011161804,
       "learning_rate": 1.7738359201773837e-05,
-      "loss": 0.7571,
+      "loss": 0.7564,
       "step": 417
     },
     {
       "epoch": 2.7516483516483516,
-      "grad_norm": 0.7452870011329651,
+      "grad_norm": 0.7417887449264526,
       "learning_rate": 1.729490022172949e-05,
-      "loss": 0.7927,
+      "loss": 0.7932,
       "step": 418
     },
     {
       "epoch": 2.758241758241758,
-      "grad_norm": 0.8206801414489746,
+      "grad_norm": 0.8229939937591553,
       "learning_rate": 1.6851441241685146e-05,
-      "loss": 0.6048,
+      "loss": 0.6047,
       "step": 419
     },
     {
       "epoch": 2.764835164835165,
-      "grad_norm": 0.6999947428703308,
+      "grad_norm": 0.6988521218299866,
       "learning_rate": 1.64079822616408e-05,
-      "loss": 0.6482,
+      "loss": 0.6495,
       "step": 420
     },
     {
       "epoch": 2.7714285714285714,
-      "grad_norm": 0.7890909910202026,
+      "grad_norm": 0.7938661575317383,
       "learning_rate": 1.596452328159645e-05,
-      "loss": 0.6781,
+      "loss": 0.679,
       "step": 421
     },
     {
       "epoch": 2.778021978021978,
-      "grad_norm": 0.8980422616004944,
+      "grad_norm": 0.8902888894081116,
       "learning_rate": 1.5521064301552106e-05,
-      "loss": 0.8028,
+      "loss": 0.7972,
       "step": 422
     },
     {
       "epoch": 2.7846153846153845,
-      "grad_norm": 0.9638545513153076,
+      "grad_norm": 0.9655492305755615,
       "learning_rate": 1.5077605321507762e-05,
-      "loss": 0.8153,
+      "loss": 0.8192,
       "step": 423
     },
     {
       "epoch": 2.791208791208791,
-      "grad_norm": 0.7455162405967712,
+      "grad_norm": 0.7463139295578003,
       "learning_rate": 1.4634146341463415e-05,
-      "loss": 0.6217,
+      "loss": 0.6208,
       "step": 424
     },
     {
       "epoch": 2.7978021978021976,
-      "grad_norm": 0.7139669060707092,
+      "grad_norm": 0.7192083597183228,
       "learning_rate": 1.419068736141907e-05,
-      "loss": 0.5691,
+      "loss": 0.5693,
       "step": 425
     },
     {
       "epoch": 2.804395604395604,
-      "grad_norm": 0.716723620891571,
+      "grad_norm": 0.7135023474693298,
       "learning_rate": 1.3747228381374724e-05,
-      "loss": 0.7262,
+      "loss": 0.7241,
       "step": 426
     },
     {
       "epoch": 2.8109890109890108,
-      "grad_norm": 0.839403510093689,
+      "grad_norm": 0.8401147723197937,
       "learning_rate": 1.3303769401330377e-05,
-      "loss": 0.7527,
+      "loss": 0.7541,
       "step": 427
     },
     {
       "epoch": 2.8175824175824173,
-      "grad_norm": 0.8174664378166199,
+      "grad_norm": 0.8131107091903687,
       "learning_rate": 1.2860310421286031e-05,
-      "loss": 0.8455,
+      "loss": 0.8444,
       "step": 428
     },
     {
       "epoch": 2.824175824175824,
-      "grad_norm": 0.7429272532463074,
+      "grad_norm": 0.7430720925331116,
       "learning_rate": 1.2416851441241686e-05,
-      "loss": 0.7737,
+      "loss": 0.7727,
       "step": 429
     },
     {
       "epoch": 2.830769230769231,
-      "grad_norm": 0.7109377384185791,
+      "grad_norm": 0.705534040927887,
       "learning_rate": 1.197339246119734e-05,
-      "loss": 0.7044,
+      "loss": 0.7042,
       "step": 430
     },
     {
       "epoch": 2.8373626373626375,
-      "grad_norm": 0.7668364644050598,
+      "grad_norm": 0.7639802098274231,
       "learning_rate": 1.1529933481152993e-05,
-      "loss": 0.6097,
+      "loss": 0.6106,
       "step": 431
     },
     {
       "epoch": 2.843956043956044,
-      "grad_norm": 0.7915285229682922,
+      "grad_norm": 0.7918491363525391,
       "learning_rate": 1.1086474501108649e-05,
-      "loss": 0.6081,
+      "loss": 0.6098,
       "step": 432
     },
     {
       "epoch": 2.8505494505494506,
-      "grad_norm": 0.7363781332969666,
+      "grad_norm": 0.737334132194519,
       "learning_rate": 1.0643015521064302e-05,
-      "loss": 0.6031,
+      "loss": 0.6063,
       "step": 433
     },
     {
       "epoch": 2.857142857142857,
-      "grad_norm": 0.7125745415687561,
+      "grad_norm": 0.712175726890564,
       "learning_rate": 1.0199556541019956e-05,
-      "loss": 0.701,
+      "loss": 0.6997,
       "step": 434
     },
     {
       "epoch": 2.8637362637362638,
-      "grad_norm": 0.688061535358429,
+      "grad_norm": 0.690259575843811,
       "learning_rate": 9.756097560975611e-06,
-      "loss": 0.7117,
+      "loss": 0.7119,
       "step": 435
     },
     {
       "epoch": 2.8703296703296703,
-      "grad_norm": 0.8198866844177246,
+      "grad_norm": 0.8218372464179993,
       "learning_rate": 9.312638580931264e-06,
-      "loss": 0.7051,
+      "loss": 0.7084,
       "step": 436
     },
     {
       "epoch": 2.876923076923077,
-      "grad_norm": 0.7329337000846863,
+      "grad_norm": 0.729598879814148,
       "learning_rate": 8.869179600886918e-06,
-      "loss": 0.6565,
+      "loss": 0.6557,
       "step": 437
     },
     {
       "epoch": 2.8835164835164835,
-      "grad_norm": 0.8913673758506775,
+      "grad_norm": 0.8934702277183533,
       "learning_rate": 8.425720620842573e-06,
-      "loss": 0.9004,
+      "loss": 0.9019,
       "step": 438
     },
     {
       "epoch": 2.89010989010989,
-      "grad_norm": 0.732146143913269,
+      "grad_norm": 0.7369548082351685,
       "learning_rate": 7.982261640798226e-06,
-      "loss": 0.6133,
+      "loss": 0.6137,
       "step": 439
     },
     {
       "epoch": 2.8967032967032966,
-      "grad_norm": 0.8003055453300476,
+      "grad_norm": 0.7938423752784729,
       "learning_rate": 7.538802660753881e-06,
-      "loss": 0.7823,
+      "loss": 0.7798,
       "step": 440
     },
     {
       "epoch": 2.903296703296703,
-      "grad_norm": 0.7041755318641663,
+      "grad_norm": 0.7049030065536499,
       "learning_rate": 7.095343680709535e-06,
-      "loss": 0.6966,
+      "loss": 0.698,
       "step": 441
     },
     {
       "epoch": 2.9098901098901098,
-      "grad_norm": 0.7933310270309448,
+      "grad_norm": 0.7887250781059265,
       "learning_rate": 6.651884700665188e-06,
-      "loss": 0.7512,
+      "loss": 0.7487,
       "step": 442
     },
     {
       "epoch": 2.9164835164835163,
-      "grad_norm": 0.5804031491279602,
+      "grad_norm": 0.5802251100540161,
       "learning_rate": 6.208425720620843e-06,
-      "loss": 0.6545,
+      "loss": 0.6557,
       "step": 443
     },
     {
       "epoch": 2.9230769230769234,
-      "grad_norm": 0.7655191421508789,
+      "grad_norm": 0.7482982873916626,
       "learning_rate": 5.7649667405764965e-06,
-      "loss": 0.5979,
+      "loss": 0.5976,
       "step": 444
     },
     {
       "epoch": 2.92967032967033,
-      "grad_norm": 0.830234706401825,
+      "grad_norm": 0.8295899033546448,
       "learning_rate": 5.321507760532151e-06,
-      "loss": 0.8859,
+      "loss": 0.8851,
       "step": 445
     },
     {
       "epoch": 2.9362637362637365,
-      "grad_norm": 0.6376515626907349,
+      "grad_norm": 0.6355248093605042,
       "learning_rate": 4.8780487804878055e-06,
       "loss": 0.5971,
       "step": 446
     },
     {
       "epoch": 2.942857142857143,
-      "grad_norm": 0.7984137535095215,
+      "grad_norm": 0.7999809980392456,
       "learning_rate": 4.434589800443459e-06,
-      "loss": 0.693,
+      "loss": 0.6951,
       "step": 447
     },
     {
       "epoch": 2.9494505494505496,
-      "grad_norm": 0.762963593006134,
+      "grad_norm": 0.7603808045387268,
       "learning_rate": 3.991130820399113e-06,
-      "loss": 0.7014,
+      "loss": 0.7022,
       "step": 448
     },
     {
       "epoch": 2.956043956043956,
-      "grad_norm": 0.809650719165802,
+      "grad_norm": 0.8109670877456665,
       "learning_rate": 3.5476718403547673e-06,
       "loss": 0.6239,
       "step": 449
     },
     {
       "epoch": 2.9626373626373628,
-      "grad_norm": 0.6534344553947449,
+      "grad_norm": 0.6522879600524902,
       "learning_rate": 3.1042128603104214e-06,
-      "loss": 0.8035,
+      "loss": 0.806,
       "step": 450
     },
     {
       "epoch": 2.9692307692307693,
-      "grad_norm": 0.8147944808006287,
+      "grad_norm": 0.8160108923912048,
       "learning_rate": 2.6607538802660755e-06,
-      "loss": 0.741,
+      "loss": 0.7419,
       "step": 451
     },
     {
       "epoch": 2.975824175824176,
-      "grad_norm": 0.6966200470924377,
+      "grad_norm": 0.6966752409934998,
       "learning_rate": 2.2172949002217296e-06,
-      "loss": 0.5363,
+      "loss": 0.5381,
       "step": 452
     },
     {
       "epoch": 2.9824175824175825,
-      "grad_norm": 0.7815210819244385,
+      "grad_norm": 0.7813317179679871,
       "learning_rate": 1.7738359201773837e-06,
-      "loss": 0.7955,
+      "loss": 0.795,
       "step": 453
     },
     {
       "epoch": 2.989010989010989,
-      "grad_norm": 0.7591390609741211,
+      "grad_norm": 0.7595626711845398,
       "learning_rate": 1.3303769401330377e-06,
-      "loss": 0.5946,
+      "loss": 0.596,
       "step": 454
     },
     {
       "epoch": 2.9956043956043956,
-      "grad_norm": 0.8715270757675171,
+      "grad_norm": 0.8695041537284851,
       "learning_rate": 8.869179600886918e-07,
-      "loss": 0.913,
+      "loss": 0.9132,
       "step": 455
     },
     {
       "epoch": 3.0,
-      "grad_norm": 0.9393943548202515,
+      "grad_norm": 0.936522364616394,
       "learning_rate": 4.434589800443459e-07,
-      "loss": 0.7085,
+      "loss": 0.7101,
       "step": 456
     }
   ],