{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 1.0,
  "eval_steps": 500,
  "global_step": 4361,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.00022930520522815867,
      "grad_norm": 237.96550129380253,
      "learning_rate": 7.633587786259541e-06,
      "loss": 8.5752,
      "step": 1
    },
    {
      "epoch": 0.00045861041045631735,
      "grad_norm": 225.32322257541617,
      "learning_rate": 1.5267175572519083e-05,
      "loss": 8.2051,
      "step": 2
    },
    {
      "epoch": 0.000687915615684476,
      "grad_norm": 186.66444826153574,
      "learning_rate": 2.2900763358778628e-05,
      "loss": 7.8221,
      "step": 3
    },
    {
      "epoch": 0.0009172208209126347,
      "grad_norm": 44.05519741908143,
      "learning_rate": 3.0534351145038166e-05,
      "loss": 6.5686,
      "step": 4
    },
    {
      "epoch": 0.0011465260261407934,
      "grad_norm": 21.710094021734562,
      "learning_rate": 3.816793893129771e-05,
      "loss": 5.9476,
      "step": 5
    },
    {
      "epoch": 0.001375831231368952,
      "grad_norm": 15.796254492361477,
      "learning_rate": 4.5801526717557256e-05,
      "loss": 5.4047,
      "step": 6
    },
    {
      "epoch": 0.0016051364365971107,
      "grad_norm": 14.566986869659809,
      "learning_rate": 5.3435114503816794e-05,
      "loss": 5.1066,
      "step": 7
    },
    {
      "epoch": 0.0018344416418252694,
      "grad_norm": 7.799379583583716,
      "learning_rate": 6.106870229007633e-05,
      "loss": 4.7562,
      "step": 8
    },
    {
      "epoch": 0.0020637468470534283,
      "grad_norm": 6.137727523099906,
      "learning_rate": 6.870229007633588e-05,
      "loss": 4.6022,
      "step": 9
    },
    {
      "epoch": 0.002293052052281587,
      "grad_norm": 5.123204300812989,
      "learning_rate": 7.633587786259542e-05,
      "loss": 4.389,
      "step": 10
    },
    {
      "epoch": 0.0025223572575097455,
      "grad_norm": 6.395214016691427,
      "learning_rate": 8.396946564885496e-05,
      "loss": 4.4437,
      "step": 11
    },
    {
      "epoch": 0.002751662462737904,
      "grad_norm": 4.147190517437537,
      "learning_rate": 9.160305343511451e-05,
      "loss": 4.2187,
      "step": 12
    },
    {
      "epoch": 0.002980967667966063,
      "grad_norm": 3.6749602590522006,
      "learning_rate": 9.923664122137405e-05,
      "loss": 4.1846,
      "step": 13
    },
    {
      "epoch": 0.0032102728731942215,
      "grad_norm": 2.7508310989336886,
      "learning_rate": 0.00010687022900763359,
      "loss": 4.0721,
      "step": 14
    },
    {
      "epoch": 0.00343957807842238,
      "grad_norm": 2.0273928899368805,
      "learning_rate": 0.00011450381679389313,
      "loss": 3.9806,
      "step": 15
    },
    {
      "epoch": 0.0036688832836505388,
      "grad_norm": 2.0408170521259517,
      "learning_rate": 0.00012213740458015266,
      "loss": 4.18,
      "step": 16
    },
    {
      "epoch": 0.0038981884888786974,
      "grad_norm": 2.0293471639938496,
      "learning_rate": 0.00012977099236641222,
      "loss": 4.1236,
      "step": 17
    },
    {
      "epoch": 0.0041274936941068565,
      "grad_norm": 1.9412210402521821,
      "learning_rate": 0.00013740458015267177,
      "loss": 4.0652,
      "step": 18
    },
    {
      "epoch": 0.004356798899335015,
      "grad_norm": 1.5808844758579432,
      "learning_rate": 0.0001450381679389313,
      "loss": 3.835,
      "step": 19
    },
    {
      "epoch": 0.004586104104563174,
      "grad_norm": 1.5390240993881632,
      "learning_rate": 0.00015267175572519084,
      "loss": 3.917,
      "step": 20
    },
    {
      "epoch": 0.004815409309791332,
      "grad_norm": 1.5381841786512938,
      "learning_rate": 0.00016030534351145037,
      "loss": 3.9481,
      "step": 21
    },
    {
      "epoch": 0.005044714515019491,
      "grad_norm": 1.0141896062441846,
      "learning_rate": 0.00016793893129770992,
      "loss": 3.9481,
      "step": 22
    },
    {
      "epoch": 0.005274019720247649,
      "grad_norm": 1.0350257469136093,
      "learning_rate": 0.00017557251908396944,
      "loss": 3.9808,
      "step": 23
    },
    {
      "epoch": 0.005503324925475808,
      "grad_norm": 0.9145834602085727,
      "learning_rate": 0.00018320610687022902,
      "loss": 3.8672,
      "step": 24
    },
    {
      "epoch": 0.005732630130703967,
      "grad_norm": 0.9324861434970846,
      "learning_rate": 0.00019083969465648857,
      "loss": 4.01,
      "step": 25
    },
    {
      "epoch": 0.005961935335932126,
      "grad_norm": 0.8576848553880976,
      "learning_rate": 0.0001984732824427481,
      "loss": 3.9915,
      "step": 26
    },
    {
      "epoch": 0.006191240541160285,
      "grad_norm": 0.9100365956223551,
      "learning_rate": 0.00020610687022900765,
      "loss": 3.9587,
      "step": 27
    },
    {
      "epoch": 0.006420545746388443,
      "grad_norm": 0.7587596564246835,
      "learning_rate": 0.00021374045801526718,
      "loss": 3.9474,
      "step": 28
    },
    {
      "epoch": 0.006649850951616602,
      "grad_norm": 0.8653207942341115,
      "learning_rate": 0.00022137404580152673,
      "loss": 3.9361,
      "step": 29
    },
    {
      "epoch": 0.00687915615684476,
      "grad_norm": 0.6960980315846079,
      "learning_rate": 0.00022900763358778625,
      "loss": 3.9232,
      "step": 30
    },
    {
      "epoch": 0.007108461362072919,
      "grad_norm": 0.9368140492199619,
      "learning_rate": 0.0002366412213740458,
      "loss": 3.8846,
      "step": 31
    },
    {
      "epoch": 0.0073377665673010775,
      "grad_norm": 0.9272164069924629,
      "learning_rate": 0.00024427480916030533,
      "loss": 3.689,
      "step": 32
    },
    {
      "epoch": 0.007567071772529237,
      "grad_norm": 0.8847099625129458,
      "learning_rate": 0.0002519083969465649,
      "loss": 3.9409,
      "step": 33
    },
    {
      "epoch": 0.007796376977757395,
      "grad_norm": 0.8062368021680489,
      "learning_rate": 0.00025954198473282443,
      "loss": 3.987,
      "step": 34
    },
    {
      "epoch": 0.008025682182985553,
      "grad_norm": 0.6981610613190473,
      "learning_rate": 0.000267175572519084,
      "loss": 3.8497,
      "step": 35
    },
    {
      "epoch": 0.008254987388213713,
      "grad_norm": 0.5913329922853873,
      "learning_rate": 0.00027480916030534353,
      "loss": 3.9826,
      "step": 36
    },
    {
      "epoch": 0.008484292593441871,
      "grad_norm": 0.6133548817206804,
      "learning_rate": 0.0002824427480916031,
      "loss": 3.879,
      "step": 37
    },
    {
      "epoch": 0.00871359779867003,
      "grad_norm": 0.47577272090117967,
      "learning_rate": 0.0002900763358778626,
      "loss": 3.8246,
      "step": 38
    },
    {
      "epoch": 0.008942903003898188,
      "grad_norm": 0.5654552929638682,
      "learning_rate": 0.00029770992366412214,
      "loss": 3.8637,
      "step": 39
    },
    {
      "epoch": 0.009172208209126348,
      "grad_norm": 0.7010458438880585,
      "learning_rate": 0.0003053435114503817,
      "loss": 3.7782,
      "step": 40
    },
    {
      "epoch": 0.009401513414354506,
      "grad_norm": 0.5199614923179321,
      "learning_rate": 0.00031297709923664124,
      "loss": 3.84,
      "step": 41
    },
    {
      "epoch": 0.009630818619582664,
      "grad_norm": 0.6130106169775211,
      "learning_rate": 0.00032061068702290074,
      "loss": 3.8868,
      "step": 42
    },
    {
      "epoch": 0.009860123824810824,
      "grad_norm": 0.4614477733947819,
      "learning_rate": 0.0003282442748091603,
      "loss": 3.8079,
      "step": 43
    },
    {
      "epoch": 0.010089429030038982,
      "grad_norm": 0.44628762533654287,
      "learning_rate": 0.00033587786259541984,
      "loss": 3.7473,
      "step": 44
    },
    {
      "epoch": 0.01031873423526714,
      "grad_norm": 0.4275247271559422,
      "learning_rate": 0.0003435114503816794,
      "loss": 3.7589,
      "step": 45
    },
    {
      "epoch": 0.010548039440495299,
      "grad_norm": 0.4369611916002911,
      "learning_rate": 0.0003511450381679389,
      "loss": 3.6308,
      "step": 46
    },
    {
      "epoch": 0.010777344645723459,
      "grad_norm": 0.5906347325783963,
      "learning_rate": 0.0003587786259541985,
      "loss": 3.9672,
      "step": 47
    },
    {
      "epoch": 0.011006649850951617,
      "grad_norm": 0.44521753531712455,
      "learning_rate": 0.00036641221374045805,
      "loss": 3.9337,
      "step": 48
    },
    {
      "epoch": 0.011235955056179775,
      "grad_norm": 0.5165545658422241,
      "learning_rate": 0.0003740458015267176,
      "loss": 3.8703,
      "step": 49
    },
    {
      "epoch": 0.011465260261407933,
      "grad_norm": 0.5274254953314194,
      "learning_rate": 0.00038167938931297715,
      "loss": 3.7668,
      "step": 50
    },
    {
      "epoch": 0.011694565466636093,
      "grad_norm": 0.5078940572399943,
      "learning_rate": 0.00038931297709923665,
      "loss": 3.7043,
      "step": 51
    },
    {
      "epoch": 0.011923870671864251,
      "grad_norm": 0.5411643116429496,
      "learning_rate": 0.0003969465648854962,
      "loss": 3.7944,
      "step": 52
    },
    {
      "epoch": 0.01215317587709241,
      "grad_norm": 0.6052261850176506,
      "learning_rate": 0.00040458015267175575,
      "loss": 3.9817,
      "step": 53
    },
    {
      "epoch": 0.01238248108232057,
      "grad_norm": 0.5123064255795436,
      "learning_rate": 0.0004122137404580153,
      "loss": 3.9026,
      "step": 54
    },
    {
      "epoch": 0.012611786287548728,
      "grad_norm": 0.47547616806583204,
      "learning_rate": 0.0004198473282442748,
      "loss": 3.9107,
      "step": 55
    },
    {
      "epoch": 0.012841091492776886,
      "grad_norm": 0.43052317059880874,
      "learning_rate": 0.00042748091603053435,
      "loss": 3.8391,
      "step": 56
    },
    {
      "epoch": 0.013070396698005044,
      "grad_norm": 0.5027459618443271,
      "learning_rate": 0.0004351145038167939,
      "loss": 3.772,
      "step": 57
    },
    {
      "epoch": 0.013299701903233204,
      "grad_norm": 0.395213010894662,
      "learning_rate": 0.00044274809160305345,
      "loss": 3.6595,
      "step": 58
    },
    {
      "epoch": 0.013529007108461362,
      "grad_norm": 0.5778555588594506,
      "learning_rate": 0.00045038167938931295,
      "loss": 3.9573,
      "step": 59
    },
    {
      "epoch": 0.01375831231368952,
      "grad_norm": 0.5379780946157494,
      "learning_rate": 0.0004580152671755725,
      "loss": 3.9204,
      "step": 60
    },
    {
      "epoch": 0.013987617518917679,
      "grad_norm": 0.421211485794834,
      "learning_rate": 0.00046564885496183206,
      "loss": 3.8681,
      "step": 61
    },
    {
      "epoch": 0.014216922724145839,
      "grad_norm": 0.5243542185900986,
      "learning_rate": 0.0004732824427480916,
      "loss": 3.747,
      "step": 62
    },
    {
      "epoch": 0.014446227929373997,
      "grad_norm": 0.5035082340145444,
      "learning_rate": 0.00048091603053435116,
      "loss": 3.8346,
      "step": 63
    },
    {
      "epoch": 0.014675533134602155,
      "grad_norm": 0.5187388616698048,
      "learning_rate": 0.0004885496183206107,
      "loss": 3.8992,
      "step": 64
    },
    {
      "epoch": 0.014904838339830313,
      "grad_norm": 0.42282525829707757,
      "learning_rate": 0.0004961832061068703,
      "loss": 3.8175,
      "step": 65
    },
    {
      "epoch": 0.015134143545058473,
      "grad_norm": 0.48925166161363504,
      "learning_rate": 0.0005038167938931298,
      "loss": 3.8761,
      "step": 66
    },
    {
      "epoch": 0.015363448750286631,
      "grad_norm": 0.45777656641448133,
      "learning_rate": 0.0005114503816793893,
      "loss": 3.8474,
      "step": 67
    },
    {
      "epoch": 0.01559275395551479,
      "grad_norm": 0.44944642908758087,
      "learning_rate": 0.0005190839694656489,
      "loss": 3.9084,
      "step": 68
    },
    {
      "epoch": 0.01582205916074295,
      "grad_norm": 0.43601062373252386,
      "learning_rate": 0.0005267175572519084,
      "loss": 3.8367,
      "step": 69
    },
    {
      "epoch": 0.016051364365971106,
      "grad_norm": 0.43907763813575823,
      "learning_rate": 0.000534351145038168,
      "loss": 3.7934,
      "step": 70
    },
    {
      "epoch": 0.016280669571199266,
      "grad_norm": 0.5106955115504161,
      "learning_rate": 0.0005419847328244275,
      "loss": 3.7287,
      "step": 71
    },
    {
      "epoch": 0.016509974776427426,
      "grad_norm": 0.43103770783062717,
      "learning_rate": 0.0005496183206106871,
      "loss": 3.6848,
      "step": 72
    },
    {
      "epoch": 0.016739279981655582,
      "grad_norm": 0.39113386689989127,
      "learning_rate": 0.0005572519083969466,
      "loss": 3.9804,
      "step": 73
    },
    {
      "epoch": 0.016968585186883742,
      "grad_norm": 0.33547839109469607,
      "learning_rate": 0.0005648854961832062,
      "loss": 3.8287,
      "step": 74
    },
    {
      "epoch": 0.017197890392111902,
      "grad_norm": 0.4777670176999005,
      "learning_rate": 0.0005725190839694656,
      "loss": 3.9267,
      "step": 75
    },
    {
      "epoch": 0.01742719559734006,
      "grad_norm": 0.33702793348426313,
      "learning_rate": 0.0005801526717557252,
      "loss": 3.8088,
      "step": 76
    },
    {
      "epoch": 0.01765650080256822,
      "grad_norm": 0.33891027809927554,
      "learning_rate": 0.0005877862595419848,
      "loss": 3.8885,
      "step": 77
    },
    {
      "epoch": 0.017885806007796375,
      "grad_norm": 0.3244620314426041,
      "learning_rate": 0.0005954198473282443,
      "loss": 3.9139,
      "step": 78
    },
    {
      "epoch": 0.018115111213024535,
      "grad_norm": 0.3369423302177944,
      "learning_rate": 0.0006030534351145039,
      "loss": 3.8184,
      "step": 79
    },
    {
      "epoch": 0.018344416418252695,
      "grad_norm": 0.3431477892267195,
      "learning_rate": 0.0006106870229007634,
      "loss": 3.8797,
      "step": 80
    },
    {
      "epoch": 0.01857372162348085,
      "grad_norm": 0.31884626143498446,
      "learning_rate": 0.000618320610687023,
      "loss": 3.6964,
      "step": 81
    },
    {
      "epoch": 0.01880302682870901,
      "grad_norm": 0.39599890050416886,
      "learning_rate": 0.0006259541984732825,
      "loss": 3.6471,
      "step": 82
    },
    {
      "epoch": 0.01903233203393717,
      "grad_norm": 0.3369013778970282,
      "learning_rate": 0.000633587786259542,
      "loss": 3.8712,
      "step": 83
    },
    {
      "epoch": 0.019261637239165328,
      "grad_norm": 0.3274888425732609,
      "learning_rate": 0.0006412213740458015,
      "loss": 3.8077,
      "step": 84
    },
    {
      "epoch": 0.019490942444393488,
      "grad_norm": 0.3608065020246459,
      "learning_rate": 0.0006488549618320611,
      "loss": 3.6604,
      "step": 85
    },
    {
      "epoch": 0.019720247649621648,
      "grad_norm": 0.34312917722306324,
      "learning_rate": 0.0006564885496183206,
      "loss": 3.8606,
      "step": 86
    },
    {
      "epoch": 0.019949552854849804,
      "grad_norm": 0.44575130579444344,
      "learning_rate": 0.0006641221374045802,
      "loss": 3.7312,
      "step": 87
    },
    {
      "epoch": 0.020178858060077964,
      "grad_norm": 0.3568029770948986,
      "learning_rate": 0.0006717557251908397,
      "loss": 3.6446,
      "step": 88
    },
    {
      "epoch": 0.02040816326530612,
      "grad_norm": 0.2963304049646136,
      "learning_rate": 0.0006793893129770993,
      "loss": 3.702,
      "step": 89
    },
    {
      "epoch": 0.02063746847053428,
      "grad_norm": 0.37527089300310773,
      "learning_rate": 0.0006870229007633588,
      "loss": 3.7316,
      "step": 90
    },
    {
      "epoch": 0.02086677367576244,
      "grad_norm": 0.39746184971216,
      "learning_rate": 0.0006946564885496184,
      "loss": 3.799,
      "step": 91
    },
    {
      "epoch": 0.021096078880990597,
      "grad_norm": 0.45934715544599664,
      "learning_rate": 0.0007022900763358778,
      "loss": 3.8024,
      "step": 92
    },
    {
      "epoch": 0.021325384086218757,
      "grad_norm": 0.4078075928227041,
      "learning_rate": 0.0007099236641221374,
      "loss": 3.8111,
      "step": 93
    },
    {
      "epoch": 0.021554689291446917,
      "grad_norm": 0.41132893002671644,
      "learning_rate": 0.000717557251908397,
      "loss": 3.7962,
      "step": 94
    },
    {
      "epoch": 0.021783994496675074,
      "grad_norm": 0.35292649502471846,
      "learning_rate": 0.0007251908396946565,
      "loss": 3.8192,
      "step": 95
    },
    {
      "epoch": 0.022013299701903234,
      "grad_norm": 0.3564059382725336,
      "learning_rate": 0.0007328244274809161,
      "loss": 3.768,
      "step": 96
    },
    {
      "epoch": 0.022242604907131393,
      "grad_norm": 0.3327033870609853,
      "learning_rate": 0.0007404580152671756,
      "loss": 3.7257,
      "step": 97
    },
    {
      "epoch": 0.02247191011235955,
      "grad_norm": 0.379301303260244,
      "learning_rate": 0.0007480916030534352,
      "loss": 3.8439,
      "step": 98
    },
    {
      "epoch": 0.02270121531758771,
      "grad_norm": 0.42254309230647386,
      "learning_rate": 0.0007557251908396947,
      "loss": 3.8142,
      "step": 99
    },
    {
      "epoch": 0.022930520522815866,
      "grad_norm": 0.3753674684089436,
      "learning_rate": 0.0007633587786259543,
      "loss": 3.8241,
      "step": 100
    },
    {
      "epoch": 0.023159825728044026,
      "grad_norm": 0.30094598361674996,
      "learning_rate": 0.0007709923664122137,
      "loss": 3.8152,
      "step": 101
    },
    {
      "epoch": 0.023389130933272186,
      "grad_norm": 0.42771321752615926,
      "learning_rate": 0.0007786259541984733,
      "loss": 3.7543,
      "step": 102
    },
    {
      "epoch": 0.023618436138500343,
      "grad_norm": 0.3406521506605358,
      "learning_rate": 0.0007862595419847328,
      "loss": 3.7925,
      "step": 103
    },
    {
      "epoch": 0.023847741343728503,
      "grad_norm": 0.28714734473613246,
      "learning_rate": 0.0007938931297709924,
      "loss": 3.7051,
      "step": 104
    },
    {
      "epoch": 0.024077046548956663,
      "grad_norm": 0.37046012440681475,
      "learning_rate": 0.0008015267175572519,
      "loss": 3.7813,
      "step": 105
    },
    {
      "epoch": 0.02430635175418482,
      "grad_norm": 0.3595455940146872,
      "learning_rate": 0.0008091603053435115,
      "loss": 3.7296,
      "step": 106
    },
    {
      "epoch": 0.02453565695941298,
      "grad_norm": 0.29845706023674506,
      "learning_rate": 0.000816793893129771,
      "loss": 3.7849,
      "step": 107
    },
    {
      "epoch": 0.02476496216464114,
      "grad_norm": 0.331885445269052,
      "learning_rate": 0.0008244274809160306,
      "loss": 3.8569,
      "step": 108
    },
    {
      "epoch": 0.024994267369869295,
      "grad_norm": 0.32064749958723343,
      "learning_rate": 0.0008320610687022901,
      "loss": 3.7715,
      "step": 109
    },
    {
      "epoch": 0.025223572575097455,
      "grad_norm": 0.3556151846285433,
      "learning_rate": 0.0008396946564885496,
      "loss": 3.8261,
      "step": 110
    },
    {
      "epoch": 0.025452877780325612,
      "grad_norm": 0.31811818751851817,
      "learning_rate": 0.0008473282442748091,
      "loss": 3.7998,
      "step": 111
    },
    {
      "epoch": 0.025682182985553772,
      "grad_norm": 0.32406624030415493,
      "learning_rate": 0.0008549618320610687,
      "loss": 3.981,
      "step": 112
    },
    {
      "epoch": 0.025911488190781932,
      "grad_norm": 0.29459849898842383,
      "learning_rate": 0.0008625954198473283,
      "loss": 3.7104,
      "step": 113
    },
    {
      "epoch": 0.02614079339601009,
      "grad_norm": 0.2858261768721266,
      "learning_rate": 0.0008702290076335878,
      "loss": 3.7646,
      "step": 114
    },
    {
      "epoch": 0.026370098601238248,
      "grad_norm": 0.2725467237713816,
      "learning_rate": 0.0008778625954198474,
      "loss": 3.8102,
      "step": 115
    },
    {
      "epoch": 0.026599403806466408,
      "grad_norm": 0.29875633247051403,
      "learning_rate": 0.0008854961832061069,
      "loss": 3.7544,
      "step": 116
    },
    {
      "epoch": 0.026828709011694565,
      "grad_norm": 0.3194746877519354,
      "learning_rate": 0.0008931297709923665,
      "loss": 3.6977,
      "step": 117
    },
    {
      "epoch": 0.027058014216922725,
      "grad_norm": 0.32540419806074344,
      "learning_rate": 0.0009007633587786259,
      "loss": 3.7217,
      "step": 118
    },
    {
      "epoch": 0.027287319422150885,
      "grad_norm": 0.35463035038371155,
      "learning_rate": 0.0009083969465648855,
      "loss": 3.8064,
      "step": 119
    },
    {
      "epoch": 0.02751662462737904,
      "grad_norm": 0.5981494749144327,
      "learning_rate": 0.000916030534351145,
      "loss": 3.7471,
      "step": 120
    },
    {
      "epoch": 0.0277459298326072,
      "grad_norm": 0.3535653841710468,
      "learning_rate": 0.0009236641221374046,
      "loss": 3.8604,
      "step": 121
    },
    {
      "epoch": 0.027975235037835357,
      "grad_norm": 0.2586296022683936,
      "learning_rate": 0.0009312977099236641,
      "loss": 3.7158,
      "step": 122
    },
    {
      "epoch": 0.028204540243063517,
      "grad_norm": 0.3181685529282772,
      "learning_rate": 0.0009389312977099237,
      "loss": 3.7411,
      "step": 123
    },
    {
      "epoch": 0.028433845448291677,
      "grad_norm": 0.8204417826037302,
      "learning_rate": 0.0009465648854961832,
      "loss": 3.8172,
      "step": 124
    },
    {
      "epoch": 0.028663150653519834,
      "grad_norm": 0.2981670626009636,
      "learning_rate": 0.0009541984732824428,
      "loss": 3.8523,
      "step": 125
    },
    {
      "epoch": 0.028892455858747994,
      "grad_norm": 0.8628311002916279,
      "learning_rate": 0.0009618320610687023,
      "loss": 3.7724,
      "step": 126
    },
    {
      "epoch": 0.029121761063976154,
      "grad_norm": 1.8012010372349643,
      "learning_rate": 0.0009694656488549618,
      "loss": 3.7944,
      "step": 127
    },
    {
      "epoch": 0.02935106626920431,
      "grad_norm": 4.676364896263532,
      "learning_rate": 0.0009770992366412213,
      "loss": 4.5256,
      "step": 128
    },
    {
      "epoch": 0.02958037147443247,
      "grad_norm": 2.137346037536333,
      "learning_rate": 0.0009847328244274808,
      "loss": 4.2013,
      "step": 129
    },
    {
      "epoch": 0.029809676679660627,
      "grad_norm": 0.44539881677463367,
      "learning_rate": 0.0009923664122137405,
      "loss": 3.7766,
      "step": 130
    },
    {
      "epoch": 0.030038981884888787,
      "grad_norm": 0.46333180999648726,
      "learning_rate": 0.001,
      "loss": 3.6999,
      "step": 131
    },
    {
      "epoch": 0.030268287090116947,
      "grad_norm": 0.3633830124191734,
      "learning_rate": 0.0009999998621016723,
      "loss": 3.8147,
      "step": 132
    },
    {
      "epoch": 0.030497592295345103,
      "grad_norm": 0.3805991735707165,
      "learning_rate": 0.0009999994484067654,
      "loss": 3.8256,
      "step": 133
    },
    {
      "epoch": 0.030726897500573263,
      "grad_norm": 0.3954825082646029,
      "learning_rate": 0.0009999987589155075,
      "loss": 3.7739,
      "step": 134
    },
    {
      "epoch": 0.030956202705801423,
      "grad_norm": 0.9305569350512148,
      "learning_rate": 0.0009999977936282788,
      "loss": 3.8531,
      "step": 135
    },
    {
      "epoch": 0.03118550791102958,
      "grad_norm": 0.28897461030064875,
      "learning_rate": 0.0009999965525456119,
      "loss": 3.8507,
      "step": 136
    },
    {
      "epoch": 0.03141481311625774,
      "grad_norm": 0.25091325636712325,
      "learning_rate": 0.0009999950356681913,
      "loss": 3.5492,
      "step": 137
    },
    {
      "epoch": 0.0316441183214859,
      "grad_norm": 0.3437354505311233,
      "learning_rate": 0.0009999932429968536,
      "loss": 3.8318,
      "step": 138
    },
    {
      "epoch": 0.03187342352671406,
      "grad_norm": 0.36077314379853187,
      "learning_rate": 0.0009999911745325876,
      "loss": 3.9766,
      "step": 139
    },
    {
      "epoch": 0.03210272873194221,
      "grad_norm": 0.2509849384456548,
      "learning_rate": 0.0009999888302765346,
      "loss": 3.6832,
      "step": 140
    },
    {
      "epoch": 0.03233203393717037,
      "grad_norm": 0.3009060036175528,
      "learning_rate": 0.0009999862102299873,
      "loss": 3.7482,
      "step": 141
    },
    {
      "epoch": 0.03256133914239853,
      "grad_norm": 0.32030748051299657,
      "learning_rate": 0.000999983314394391,
      "loss": 3.7934,
      "step": 142
    },
    {
      "epoch": 0.03279064434762669,
      "grad_norm": 0.43794334316990474,
      "learning_rate": 0.0009999801427713433,
      "loss": 3.8652,
      "step": 143
    },
    {
      "epoch": 0.03301994955285485,
      "grad_norm": 0.4000449698861661,
      "learning_rate": 0.000999976695362593,
      "loss": 3.852,
      "step": 144
    },
    {
      "epoch": 0.03324925475808301,
      "grad_norm": 1.2240604559784714,
      "learning_rate": 0.0009999729721700424,
      "loss": 3.7372,
      "step": 145
    },
    {
      "epoch": 0.033478559963311165,
      "grad_norm": 0.2516152673856596,
      "learning_rate": 0.0009999689731957449,
      "loss": 3.6672,
      "step": 146
    },
    {
      "epoch": 0.033707865168539325,
      "grad_norm": 0.282417591358799,
      "learning_rate": 0.000999964698441906,
      "loss": 3.8143,
      "step": 147
    },
    {
      "epoch": 0.033937170373767485,
      "grad_norm": 0.21612240523459403,
      "learning_rate": 0.000999960147910884,
      "loss": 3.7761,
      "step": 148
    },
    {
      "epoch": 0.034166475578995645,
      "grad_norm": 0.246784964971249,
      "learning_rate": 0.0009999553216051892,
      "loss": 3.6696,
      "step": 149
    },
    {
      "epoch": 0.034395780784223805,
      "grad_norm": 0.2518990700215664,
      "learning_rate": 0.000999950219527483,
      "loss": 3.7661,
      "step": 150
    },
    {
      "epoch": 0.03462508598945196,
      "grad_norm": 0.3404786806715568,
      "learning_rate": 0.00099994484168058,
      "loss": 3.891,
      "step": 151
    },
    {
      "epoch": 0.03485439119468012,
      "grad_norm": 0.255497552824681,
      "learning_rate": 0.0009999391880674469,
      "loss": 3.8561,
      "step": 152
    },
    {
      "epoch": 0.03508369639990828,
      "grad_norm": 0.3796665095255569,
      "learning_rate": 0.0009999332586912019,
      "loss": 3.9623,
      "step": 153
    },
    {
      "epoch": 0.03531300160513644,
      "grad_norm": 0.22579151948095777,
      "learning_rate": 0.0009999270535551157,
      "loss": 3.915,
      "step": 154
    },
    {
      "epoch": 0.0355423068103646,
      "grad_norm": 0.26811181557870345,
      "learning_rate": 0.0009999205726626108,
      "loss": 3.5695,
      "step": 155
    },
    {
      "epoch": 0.03577161201559275,
      "grad_norm": 0.19867414863360108,
      "learning_rate": 0.0009999138160172623,
      "loss": 3.6805,
      "step": 156
    },
    {
      "epoch": 0.03600091722082091,
      "grad_norm": 0.2508004760999495,
      "learning_rate": 0.000999906783622797,
      "loss": 3.6051,
      "step": 157
    },
    {
      "epoch": 0.03623022242604907,
      "grad_norm": 1.451413798474198,
      "learning_rate": 0.000999899475483094,
      "loss": 3.7557,
      "step": 158
    },
    {
      "epoch": 0.03645952763127723,
      "grad_norm": 0.2285085918829455,
      "learning_rate": 0.0009998918916021842,
      "loss": 3.8361,
      "step": 159
    },
    {
      "epoch": 0.03668883283650539,
      "grad_norm": 0.19826386995376183,
      "learning_rate": 0.000999884031984251,
      "loss": 3.8634,
      "step": 160
    },
    {
      "epoch": 0.03691813804173355,
      "grad_norm": 0.20623533529573126,
      "learning_rate": 0.0009998758966336297,
      "loss": 3.8994,
      "step": 161
    },
    {
      "epoch": 0.0371474432469617,
      "grad_norm": 0.2550666239603128,
      "learning_rate": 0.0009998674855548075,
      "loss": 3.9142,
      "step": 162
    },
    {
      "epoch": 0.03737674845218986,
      "grad_norm": 0.1979617142398181,
      "learning_rate": 0.0009998587987524242,
      "loss": 3.8922,
      "step": 163
    },
    {
      "epoch": 0.03760605365741802,
      "grad_norm": 0.22989244138382792,
      "learning_rate": 0.0009998498362312714,
      "loss": 3.7219,
      "step": 164
    },
    {
      "epoch": 0.03783535886264618,
      "grad_norm": 0.19794053697533598,
      "learning_rate": 0.0009998405979962926,
      "loss": 3.7956,
      "step": 165
    },
    {
      "epoch": 0.03806466406787434,
      "grad_norm": 1.4257966314796013,
      "learning_rate": 0.0009998310840525833,
      "loss": 3.7031,
      "step": 166
    },
    {
      "epoch": 0.038293969273102496,
      "grad_norm": 0.21183548069089572,
      "learning_rate": 0.000999821294405392,
      "loss": 3.6762,
      "step": 167
    },
    {
      "epoch": 0.038523274478330656,
      "grad_norm": 0.2156297674672091,
      "learning_rate": 0.0009998112290601179,
      "loss": 3.7362,
      "step": 168
    },
    {
      "epoch": 0.038752579683558816,
      "grad_norm": 0.21376734371908832,
      "learning_rate": 0.0009998008880223134,
      "loss": 3.7668,
      "step": 169
    },
    {
      "epoch": 0.038981884888786976,
      "grad_norm": 0.1724831730192065,
      "learning_rate": 0.0009997902712976824,
      "loss": 3.8022,
      "step": 170
    },
    {
      "epoch": 0.039211190094015136,
      "grad_norm": 0.21694891379407266,
      "learning_rate": 0.000999779378892081,
      "loss": 3.6711,
      "step": 171
    },
    {
      "epoch": 0.039440495299243296,
      "grad_norm": 0.19175423705305208,
      "learning_rate": 0.0009997682108115176,
      "loss": 3.5813,
      "step": 172
    },
    {
      "epoch": 0.03966980050447145,
      "grad_norm": 1.4120660918557404,
      "learning_rate": 0.0009997567670621522,
      "loss": 3.7468,
      "step": 173
    },
    {
      "epoch": 0.03989910570969961,
      "grad_norm": 0.18410030043652612,
      "learning_rate": 0.000999745047650297,
      "loss": 3.7921,
      "step": 174
    },
    {
      "epoch": 0.04012841091492777,
      "grad_norm": 0.19038412287070167,
      "learning_rate": 0.0009997330525824165,
      "loss": 3.7402,
      "step": 175
    },
    {
      "epoch": 0.04035771612015593,
      "grad_norm": 0.15475398293474263,
      "learning_rate": 0.0009997207818651273,
      "loss": 3.6467,
      "step": 176
    },
    {
      "epoch": 0.04058702132538409,
      "grad_norm": 0.20034493799446926,
      "learning_rate": 0.0009997082355051976,
      "loss": 3.8951,
      "step": 177
    },
    {
      "epoch": 0.04081632653061224,
      "grad_norm": 0.2395353842779523,
      "learning_rate": 0.0009996954135095479,
      "loss": 3.6665,
      "step": 178
    },
    {
      "epoch": 0.0410456317358404,
      "grad_norm": 0.16923580380883518,
      "learning_rate": 0.000999682315885251,
      "loss": 3.7859,
      "step": 179
    },
    {
      "epoch": 0.04127493694106856,
      "grad_norm": 0.16915929492873066,
      "learning_rate": 0.0009996689426395307,
      "loss": 3.6715,
      "step": 180
    },
    {
      "epoch": 0.04150424214629672,
      "grad_norm": 0.22341905764840872,
      "learning_rate": 0.0009996552937797645,
      "loss": 3.7938,
      "step": 181
    },
    {
      "epoch": 0.04173354735152488,
      "grad_norm": 0.1879567969163752,
      "learning_rate": 0.0009996413693134805,
      "loss": 3.7105,
      "step": 182
    },
    {
      "epoch": 0.04196285255675304,
      "grad_norm": 0.20158131526866313,
      "learning_rate": 0.0009996271692483596,
      "loss": 3.5856,
      "step": 183
    },
    {
      "epoch": 0.042192157761981194,
      "grad_norm": 0.21369425807606826,
      "learning_rate": 0.0009996126935922342,
      "loss": 3.7203,
      "step": 184
    },
    {
      "epoch": 0.042421462967209354,
      "grad_norm": 0.28745076340039805,
      "learning_rate": 0.0009995979423530893,
      "loss": 3.7211,
      "step": 185
    },
    {
      "epoch": 0.042650768172437514,
      "grad_norm": 0.16857776724379497,
      "learning_rate": 0.0009995829155390612,
      "loss": 3.6751,
      "step": 186
    },
    {
      "epoch": 0.042880073377665674,
      "grad_norm": 0.21023510726944866,
      "learning_rate": 0.000999567613158439,
      "loss": 3.6789,
      "step": 187
    },
    {
      "epoch": 0.043109378582893834,
      "grad_norm": 0.7954519908812462,
      "learning_rate": 0.000999552035219663,
      "loss": 3.5844,
      "step": 188
    },
    {
      "epoch": 0.04333868378812199,
      "grad_norm": 0.2146371153942246,
      "learning_rate": 0.0009995361817313263,
      "loss": 3.7881,
      "step": 189
    },
    {
      "epoch": 0.04356798899335015,
      "grad_norm": 0.17404830166164154,
      "learning_rate": 0.000999520052702173,
      "loss": 3.6823,
      "step": 190
    },
    {
      "epoch": 0.04379729419857831,
      "grad_norm": 0.1505387812134299,
      "learning_rate": 0.0009995036481411004,
      "loss": 3.6882,
      "step": 191
    },
    {
      "epoch": 0.04402659940380647,
      "grad_norm": 0.25045477315180226,
      "learning_rate": 0.0009994869680571567,
      "loss": 3.7203,
      "step": 192
    },
    {
      "epoch": 0.04425590460903463,
      "grad_norm": 0.1572973630777354,
      "learning_rate": 0.0009994700124595429,
      "loss": 3.8209,
      "step": 193
    },
    {
      "epoch": 0.04448520981426279,
      "grad_norm": 0.18105233513489447,
      "learning_rate": 0.000999452781357611,
      "loss": 3.7822,
      "step": 194
    },
    {
      "epoch": 0.04471451501949094,
      "grad_norm": 0.1802006078040822,
      "learning_rate": 0.0009994352747608663,
      "loss": 3.6136,
      "step": 195
    },
    {
      "epoch": 0.0449438202247191,
      "grad_norm": 0.17463252842910582,
      "learning_rate": 0.0009994174926789647,
      "loss": 3.8103,
      "step": 196
    },
    {
      "epoch": 0.04517312542994726,
      "grad_norm": 0.15568214099888075,
      "learning_rate": 0.0009993994351217151,
      "loss": 3.781,
      "step": 197
    },
    {
      "epoch": 0.04540243063517542,
      "grad_norm": 0.24628986006003034,
      "learning_rate": 0.0009993811020990778,
      "loss": 3.7835,
      "step": 198
    },
    {
      "epoch": 0.04563173584040358,
      "grad_norm": 0.17216615273135186,
      "learning_rate": 0.000999362493621165,
      "loss": 3.7237,
      "step": 199
    },
    {
      "epoch": 0.04586104104563173,
      "grad_norm": 0.16723297514476004,
      "learning_rate": 0.0009993436096982413,
      "loss": 3.7184,
      "step": 200
    },
    {
      "epoch": 0.04609034625085989,
      "grad_norm": 0.15952747980850007,
      "learning_rate": 0.0009993244503407226,
      "loss": 3.7983,
      "step": 201
    },
    {
      "epoch": 0.04631965145608805,
      "grad_norm": 0.17325373524022825,
      "learning_rate": 0.0009993050155591775,
      "loss": 3.7573,
      "step": 202
    },
    {
      "epoch": 0.04654895666131621,
      "grad_norm": 0.15233634282451267,
      "learning_rate": 0.0009992853053643258,
      "loss": 3.6926,
      "step": 203
    },
    {
      "epoch": 0.04677826186654437,
      "grad_norm": 0.160327272709267,
      "learning_rate": 0.0009992653197670393,
      "loss": 3.7199,
      "step": 204
    },
    {
      "epoch": 0.04700756707177253,
      "grad_norm": 0.1612897164455894,
      "learning_rate": 0.0009992450587783426,
      "loss": 3.6879,
      "step": 205
    },
    {
      "epoch": 0.047236872277000685,
      "grad_norm": 0.16651515625072147,
      "learning_rate": 0.0009992245224094108,
      "loss": 3.7632,
      "step": 206
    },
    {
      "epoch": 0.047466177482228845,
      "grad_norm": 0.22394395981529108,
      "learning_rate": 0.000999203710671572,
      "loss": 3.6552,
      "step": 207
    },
    {
      "epoch": 0.047695482687457005,
      "grad_norm": 0.157723162051367,
      "learning_rate": 0.000999182623576306,
      "loss": 3.6942,
      "step": 208
    },
    {
      "epoch": 0.047924787892685165,
      "grad_norm": 0.1707297105037175,
      "learning_rate": 0.0009991612611352438,
      "loss": 3.631,
      "step": 209
    },
    {
      "epoch": 0.048154093097913325,
      "grad_norm": 0.1689230779167834,
      "learning_rate": 0.0009991396233601693,
      "loss": 3.7375,
      "step": 210
    },
    {
      "epoch": 0.04838339830314148,
      "grad_norm": 0.201120098866728,
      "learning_rate": 0.0009991177102630173,
      "loss": 3.747,
      "step": 211
    },
    {
      "epoch": 0.04861270350836964,
      "grad_norm": 0.1592292762990334,
      "learning_rate": 0.0009990955218558751,
      "loss": 3.6109,
      "step": 212
    },
    {
      "epoch": 0.0488420087135978,
      "grad_norm": 0.17673914861612894,
      "learning_rate": 0.0009990730581509817,
      "loss": 3.5782,
      "step": 213
    },
    {
      "epoch": 0.04907131391882596,
      "grad_norm": 0.19496038134339966,
      "learning_rate": 0.000999050319160728,
      "loss": 3.7062,
      "step": 214
    },
    {
      "epoch": 0.04930061912405412,
      "grad_norm": 0.17673831891944777,
      "learning_rate": 0.0009990273048976566,
      "loss": 3.634,
      "step": 215
    },
    {
      "epoch": 0.04952992432928228,
      "grad_norm": 0.18389874374699203,
      "learning_rate": 0.0009990040153744619,
      "loss": 3.785,
      "step": 216
    },
    {
      "epoch": 0.04975922953451043,
      "grad_norm": 0.28821464807541136,
      "learning_rate": 0.0009989804506039905,
      "loss": 3.6382,
      "step": 217
    },
    {
      "epoch": 0.04998853473973859,
      "grad_norm": 0.16154087843897244,
      "learning_rate": 0.0009989566105992403,
      "loss": 3.7813,
      "step": 218
    },
    {
      "epoch": 0.05021783994496675,
      "grad_norm": 0.1802268363595131,
      "learning_rate": 0.0009989324953733614,
      "loss": 3.8224,
      "step": 219
    },
    {
      "epoch": 0.05044714515019491,
      "grad_norm": 0.17230634816077275,
      "learning_rate": 0.0009989081049396557,
      "loss": 3.5961,
      "step": 220
    },
    {
      "epoch": 0.05067645035542307,
      "grad_norm": 0.23118312051499457,
      "learning_rate": 0.0009988834393115766,
      "loss": 3.6442,
      "step": 221
    },
    {
      "epoch": 0.050905755560651224,
      "grad_norm": 0.18387388373176583,
      "learning_rate": 0.0009988584985027296,
      "loss": 3.7334,
      "step": 222
    },
    {
      "epoch": 0.051135060765879384,
      "grad_norm": 0.17506863521553132,
      "learning_rate": 0.000998833282526872,
      "loss": 3.6768,
      "step": 223
    },
    {
      "epoch": 0.051364365971107544,
      "grad_norm": 0.1723966280673466,
      "learning_rate": 0.0009988077913979125,
      "loss": 3.7913,
      "step": 224
    },
    {
      "epoch": 0.051593671176335704,
      "grad_norm": 0.21249458687913042,
      "learning_rate": 0.0009987820251299122,
      "loss": 3.6484,
      "step": 225
    },
    {
      "epoch": 0.051822976381563864,
      "grad_norm": 0.1670629635368297,
      "learning_rate": 0.0009987559837370832,
      "loss": 3.5802,
      "step": 226
    },
    {
      "epoch": 0.052052281586792024,
      "grad_norm": 0.17377003749920716,
      "learning_rate": 0.00099872966723379,
      "loss": 3.7688,
      "step": 227
    },
    {
      "epoch": 0.05228158679202018,
      "grad_norm": 0.17098855824960021,
      "learning_rate": 0.0009987030756345486,
      "loss": 3.7845,
      "step": 228
    },
    {
      "epoch": 0.052510891997248337,
      "grad_norm": 0.16158569111063492,
      "learning_rate": 0.0009986762089540266,
      "loss": 3.7026,
      "step": 229
    },
    {
      "epoch": 0.052740197202476496,
      "grad_norm": 0.16372372822359899,
      "learning_rate": 0.0009986490672070436,
      "loss": 3.6466,
      "step": 230
    },
    {
      "epoch": 0.052969502407704656,
      "grad_norm": 0.27475117955676476,
      "learning_rate": 0.0009986216504085709,
      "loss": 3.6536,
      "step": 231
    },
    {
      "epoch": 0.053198807612932816,
      "grad_norm": 0.19862160382534294,
      "learning_rate": 0.0009985939585737311,
      "loss": 3.6838,
      "step": 232
    },
    {
      "epoch": 0.05342811281816097,
      "grad_norm": 0.18225328240884486,
      "learning_rate": 0.0009985659917177991,
      "loss": 3.5771,
      "step": 233
    },
    {
      "epoch": 0.05365741802338913,
      "grad_norm": 0.18221190581687494,
      "learning_rate": 0.0009985377498562013,
      "loss": 3.7198,
      "step": 234
    },
    {
      "epoch": 0.05388672322861729,
      "grad_norm": 0.18325182872920723,
      "learning_rate": 0.0009985092330045155,
      "loss": 3.7318,
      "step": 235
    },
    {
      "epoch": 0.05411602843384545,
      "grad_norm": 0.18464656458759332,
      "learning_rate": 0.0009984804411784715,
      "loss": 3.8136,
      "step": 236
    },
    {
      "epoch": 0.05434533363907361,
      "grad_norm": 0.16028715688074813,
      "learning_rate": 0.0009984513743939508,
      "loss": 3.7646,
      "step": 237
    },
    {
      "epoch": 0.05457463884430177,
      "grad_norm": 0.18550466156673576,
      "learning_rate": 0.0009984220326669862,
      "loss": 3.8478,
      "step": 238
    },
    {
      "epoch": 0.05480394404952992,
      "grad_norm": 0.1834105006632199,
      "learning_rate": 0.0009983924160137626,
      "loss": 3.6759,
      "step": 239
    },
    {
      "epoch": 0.05503324925475808,
      "grad_norm": 0.16407503322231465,
      "learning_rate": 0.0009983625244506162,
      "loss": 3.7438,
      "step": 240
    },
    {
      "epoch": 0.05526255445998624,
      "grad_norm": 0.13979115594145872,
      "learning_rate": 0.000998332357994035,
      "loss": 3.5986,
      "step": 241
    },
    {
      "epoch": 0.0554918596652144,
      "grad_norm": 0.1615199027228966,
      "learning_rate": 0.0009983019166606589,
      "loss": 3.7847,
      "step": 242
    },
    {
      "epoch": 0.05572116487044256,
      "grad_norm": 0.1863441006632608,
      "learning_rate": 0.0009982712004672786,
      "loss": 3.6866,
      "step": 243
    },
    {
      "epoch": 0.055950470075670715,
      "grad_norm": 0.21455626450544865,
      "learning_rate": 0.0009982402094308373,
      "loss": 3.8135,
      "step": 244
    },
    {
      "epoch": 0.056179775280898875,
      "grad_norm": 0.18377882271119259,
      "learning_rate": 0.0009982089435684295,
      "loss": 3.7986,
      "step": 245
    },
    {
      "epoch": 0.056409080486127035,
      "grad_norm": 0.2617644536404064,
      "learning_rate": 0.0009981774028973013,
      "loss": 3.761,
      "step": 246
    },
    {
      "epoch": 0.056638385691355195,
      "grad_norm": 0.21303500210659754,
      "learning_rate": 0.0009981455874348499,
      "loss": 3.7557,
      "step": 247
    },
    {
      "epoch": 0.056867690896583355,
      "grad_norm": 0.15574798887890656,
      "learning_rate": 0.0009981134971986248,
      "loss": 3.6618,
      "step": 248
    },
    {
      "epoch": 0.05709699610181151,
      "grad_norm": 0.14349039341444006,
      "learning_rate": 0.0009980811322063269,
      "loss": 3.5523,
      "step": 249
    },
    {
      "epoch": 0.05732630130703967,
      "grad_norm": 0.195636506907488,
      "learning_rate": 0.0009980484924758083,
      "loss": 3.7969,
      "step": 250
    },
    {
      "epoch": 0.05755560651226783,
      "grad_norm": 0.20683873579874432,
      "learning_rate": 0.0009980155780250728,
      "loss": 3.6959,
      "step": 251
    },
    {
      "epoch": 0.05778491171749599,
      "grad_norm": 0.1632172590113636,
      "learning_rate": 0.0009979823888722759,
      "loss": 3.7362,
      "step": 252
    },
    {
      "epoch": 0.05801421692272415,
      "grad_norm": 0.18355485032833035,
      "learning_rate": 0.0009979489250357243,
      "loss": 3.8308,
      "step": 253
    },
    {
      "epoch": 0.05824352212795231,
      "grad_norm": 0.16564882166284686,
      "learning_rate": 0.000997915186533877,
      "loss": 3.6721,
      "step": 254
    },
    {
      "epoch": 0.05847282733318046,
      "grad_norm": 0.1795408264986232,
      "learning_rate": 0.0009978811733853431,
      "loss": 3.7179,
      "step": 255
    },
    {
      "epoch": 0.05870213253840862,
      "grad_norm": 0.1742776863849554,
      "learning_rate": 0.000997846885608885,
      "loss": 3.7521,
      "step": 256
    },
    {
      "epoch": 0.05893143774363678,
      "grad_norm": 0.16885430897648915,
      "learning_rate": 0.0009978123232234147,
      "loss": 3.6326,
      "step": 257
    },
    {
      "epoch": 0.05916074294886494,
      "grad_norm": 0.15947382924262818,
      "learning_rate": 0.0009977774862479972,
      "loss": 3.7866,
      "step": 258
    },
    {
      "epoch": 0.0593900481540931,
      "grad_norm": 0.16648241918572604,
      "learning_rate": 0.000997742374701848,
      "loss": 3.5222,
      "step": 259
    },
    {
      "epoch": 0.05961935335932125,
      "grad_norm": 0.16526200627170357,
      "learning_rate": 0.0009977069886043344,
      "loss": 3.8183,
      "step": 260
    },
    {
      "epoch": 0.05984865856454941,
      "grad_norm": 0.1425675684883045,
      "learning_rate": 0.0009976713279749754,
      "loss": 3.7566,
      "step": 261
    },
    {
      "epoch": 0.06007796376977757,
      "grad_norm": 0.15698310698050427,
      "learning_rate": 0.0009976353928334409,
      "loss": 3.8075,
      "step": 262
    },
    {
      "epoch": 0.06030726897500573,
      "grad_norm": 0.15097214541140647,
      "learning_rate": 0.0009975991831995528,
      "loss": 3.6968,
      "step": 263
    },
    {
      "epoch": 0.06053657418023389,
      "grad_norm": 0.1983079495477396,
      "learning_rate": 0.0009975626990932837,
      "loss": 3.7661,
      "step": 264
    },
    {
      "epoch": 0.06076587938546205,
      "grad_norm": 0.16039140319442782,
      "learning_rate": 0.0009975259405347581,
      "loss": 3.7064,
      "step": 265
    },
    {
      "epoch": 0.060995184590690206,
      "grad_norm": 0.1471879236452574,
      "learning_rate": 0.0009974889075442521,
      "loss": 3.7499,
      "step": 266
    },
    {
      "epoch": 0.061224489795918366,
      "grad_norm": 0.15427185424670362,
      "learning_rate": 0.0009974516001421926,
      "loss": 3.7067,
      "step": 267
    },
    {
      "epoch": 0.061453795001146526,
      "grad_norm": 0.15107061196767463,
      "learning_rate": 0.0009974140183491578,
      "loss": 3.6296,
      "step": 268
    },
    {
      "epoch": 0.061683100206374686,
      "grad_norm": 0.17153339210723492,
      "learning_rate": 0.000997376162185878,
      "loss": 3.7135,
      "step": 269
    },
    {
      "epoch": 0.061912405411602846,
      "grad_norm": 0.15864558409094096,
      "learning_rate": 0.0009973380316732343,
      "loss": 3.6851,
      "step": 270
    },
    {
      "epoch": 0.062141710616831,
      "grad_norm": 0.14693987416900517,
      "learning_rate": 0.0009972996268322594,
      "loss": 3.6937,
      "step": 271
    },
    {
      "epoch": 0.06237101582205916,
      "grad_norm": 0.16393185422508086,
      "learning_rate": 0.0009972609476841367,
      "loss": 3.7933,
      "step": 272
    },
    {
      "epoch": 0.06260032102728733,
      "grad_norm": 0.14820599922914918,
      "learning_rate": 0.0009972219942502017,
      "loss": 3.6485,
      "step": 273
    },
    {
      "epoch": 0.06282962623251548,
      "grad_norm": 0.17553140951539267,
      "learning_rate": 0.000997182766551941,
      "loss": 3.7228,
      "step": 274
    },
    {
      "epoch": 0.06305893143774363,
      "grad_norm": 0.14631525611246174,
      "learning_rate": 0.0009971432646109918,
      "loss": 3.6274,
      "step": 275
    },
    {
      "epoch": 0.0632882366429718,
      "grad_norm": 0.177427161966203,
      "learning_rate": 0.0009971034884491436,
      "loss": 3.6301,
      "step": 276
    },
    {
      "epoch": 0.06351754184819995,
      "grad_norm": 0.14945666919326747,
      "learning_rate": 0.0009970634380883365,
      "loss": 3.5657,
      "step": 277
    },
    {
      "epoch": 0.06374684705342812,
      "grad_norm": 0.19798752733201141,
      "learning_rate": 0.0009970231135506618,
      "loss": 3.7681,
      "step": 278
    },
    {
      "epoch": 0.06397615225865627,
      "grad_norm": 0.19858126180443186,
      "learning_rate": 0.0009969825148583627,
      "loss": 3.7382,
      "step": 279
    },
    {
      "epoch": 0.06420545746388442,
      "grad_norm": 0.14769742478022513,
      "learning_rate": 0.0009969416420338329,
      "loss": 3.6706,
      "step": 280
    },
    {
      "epoch": 0.06443476266911259,
      "grad_norm": 0.15428665334441066,
      "learning_rate": 0.0009969004950996173,
      "loss": 3.6719,
      "step": 281
    },
    {
      "epoch": 0.06466406787434074,
      "grad_norm": 0.192695295497557,
      "learning_rate": 0.000996859074078413,
      "loss": 3.7504,
      "step": 282
    },
    {
      "epoch": 0.06489337307956891,
      "grad_norm": 0.14811660477254268,
      "learning_rate": 0.0009968173789930668,
      "loss": 3.7967,
      "step": 283
    },
    {
      "epoch": 0.06512267828479706,
      "grad_norm": 0.16164273718254707,
      "learning_rate": 0.0009967754098665778,
      "loss": 3.7015,
      "step": 284
    },
    {
      "epoch": 0.06535198349002522,
      "grad_norm": 0.18368515791734258,
      "learning_rate": 0.0009967331667220958,
      "loss": 3.685,
      "step": 285
    },
    {
      "epoch": 0.06558128869525338,
      "grad_norm": 0.18462038484290416,
      "learning_rate": 0.000996690649582922,
      "loss": 3.7156,
      "step": 286
    },
    {
      "epoch": 0.06581059390048154,
      "grad_norm": 0.18090351871322471,
      "learning_rate": 0.0009966478584725086,
      "loss": 3.7926,
      "step": 287
    },
    {
      "epoch": 0.0660398991057097,
      "grad_norm": 0.15871812078384903,
      "learning_rate": 0.0009966047934144584,
      "loss": 3.7801,
      "step": 288
    },
    {
      "epoch": 0.06626920431093786,
      "grad_norm": 0.1600134632299252,
      "learning_rate": 0.0009965614544325263,
      "loss": 3.8192,
      "step": 289
    },
    {
      "epoch": 0.06649850951616602,
      "grad_norm": 0.14423988994360598,
      "learning_rate": 0.0009965178415506175,
      "loss": 3.5849,
      "step": 290
    },
    {
      "epoch": 0.06672781472139418,
      "grad_norm": 0.16123949220166137,
      "learning_rate": 0.000996473954792789,
      "loss": 3.7756,
      "step": 291
    },
    {
      "epoch": 0.06695711992662233,
      "grad_norm": 0.19321433762199075,
      "learning_rate": 0.0009964297941832479,
      "loss": 3.6447,
      "step": 292
    },
    {
      "epoch": 0.0671864251318505,
      "grad_norm": 0.14501168505423792,
      "learning_rate": 0.0009963853597463532,
      "loss": 3.6442,
      "step": 293
    },
    {
      "epoch": 0.06741573033707865,
      "grad_norm": 0.16666337456104682,
      "learning_rate": 0.0009963406515066148,
      "loss": 3.6146,
      "step": 294
    },
    {
      "epoch": 0.06764503554230682,
      "grad_norm": 0.19365163507322172,
      "learning_rate": 0.000996295669488693,
      "loss": 3.7419,
      "step": 295
    },
    {
      "epoch": 0.06787434074753497,
      "grad_norm": 0.17694928698543438,
      "learning_rate": 0.0009962504137173997,
      "loss": 3.7401,
      "step": 296
    },
    {
      "epoch": 0.06810364595276312,
      "grad_norm": 0.16154729546296917,
      "learning_rate": 0.0009962048842176979,
      "loss": 3.6353,
      "step": 297
    },
    {
      "epoch": 0.06833295115799129,
      "grad_norm": 0.16776076036503412,
      "learning_rate": 0.0009961590810147014,
      "loss": 3.712,
      "step": 298
    },
    {
      "epoch": 0.06856225636321944,
      "grad_norm": 0.1784214506048516,
      "learning_rate": 0.0009961130041336748,
      "loss": 3.8056,
      "step": 299
    },
    {
      "epoch": 0.06879156156844761,
      "grad_norm": 0.20398770882284875,
      "learning_rate": 0.0009960666536000338,
      "loss": 3.6357,
      "step": 300
    },
    {
      "epoch": 0.06902086677367576,
      "grad_norm": 0.15737228500785158,
      "learning_rate": 0.0009960200294393449,
      "loss": 3.5447,
      "step": 301
    },
    {
      "epoch": 0.06925017197890392,
      "grad_norm": 0.15398928268331497,
      "learning_rate": 0.000995973131677326,
      "loss": 3.6894,
      "step": 302
    },
    {
      "epoch": 0.06947947718413208,
      "grad_norm": 0.15358508975668356,
      "learning_rate": 0.0009959259603398453,
      "loss": 3.7274,
      "step": 303
    },
    {
      "epoch": 0.06970878238936024,
      "grad_norm": 0.15464920212157102,
      "learning_rate": 0.0009958785154529225,
      "loss": 3.7396,
      "step": 304
    },
    {
      "epoch": 0.0699380875945884,
      "grad_norm": 0.17248385750529166,
      "learning_rate": 0.0009958307970427275,
      "loss": 3.7613,
      "step": 305
    },
    {
      "epoch": 0.07016739279981656,
      "grad_norm": 0.15974555592034087,
      "learning_rate": 0.0009957828051355816,
      "loss": 3.6712,
      "step": 306
    },
    {
      "epoch": 0.07039669800504471,
      "grad_norm": 0.1453020378782658,
      "learning_rate": 0.0009957345397579572,
      "loss": 3.7424,
      "step": 307
    },
    {
      "epoch": 0.07062600321027288,
      "grad_norm": 0.13605573075004063,
      "learning_rate": 0.0009956860009364766,
      "loss": 3.7023,
      "step": 308
    },
    {
      "epoch": 0.07085530841550103,
      "grad_norm": 0.1935853982034738,
      "learning_rate": 0.0009956371886979138,
      "loss": 3.7579,
      "step": 309
    },
    {
      "epoch": 0.0710846136207292,
      "grad_norm": 0.14476001175227035,
      "learning_rate": 0.000995588103069193,
      "loss": 3.587,
      "step": 310
    },
    {
      "epoch": 0.07131391882595735,
      "grad_norm": 0.16644752994980475,
      "learning_rate": 0.00099553874407739,
      "loss": 3.6635,
      "step": 311
    },
    {
      "epoch": 0.0715432240311855,
      "grad_norm": 0.1396906251525954,
      "learning_rate": 0.0009954891117497306,
      "loss": 3.783,
      "step": 312
    },
    {
      "epoch": 0.07177252923641367,
      "grad_norm": 0.18156743658772312,
      "learning_rate": 0.0009954392061135916,
      "loss": 3.8287,
      "step": 313
    },
    {
      "epoch": 0.07200183444164182,
      "grad_norm": 0.16520329655448684,
      "learning_rate": 0.0009953890271965006,
      "loss": 3.6793,
      "step": 314
    },
    {
      "epoch": 0.07223113964686999,
      "grad_norm": 0.14391862140591488,
      "learning_rate": 0.0009953385750261364,
      "loss": 3.6666,
      "step": 315
    },
    {
      "epoch": 0.07246044485209814,
      "grad_norm": 0.17292383853029208,
      "learning_rate": 0.0009952878496303273,
      "loss": 3.572,
      "step": 316
    },
    {
      "epoch": 0.07268975005732631,
      "grad_norm": 0.159812784266416,
      "learning_rate": 0.0009952368510370538,
      "loss": 3.7175,
      "step": 317
    },
    {
      "epoch": 0.07291905526255446,
      "grad_norm": 0.14815060869198163,
      "learning_rate": 0.000995185579274446,
      "loss": 3.6799,
      "step": 318
    },
    {
      "epoch": 0.07314836046778261,
      "grad_norm": 0.1730187649813741,
      "learning_rate": 0.0009951340343707852,
      "loss": 3.7129,
      "step": 319
    },
    {
      "epoch": 0.07337766567301078,
      "grad_norm": 0.18375915636113038,
      "learning_rate": 0.000995082216354503,
      "loss": 3.7531,
      "step": 320
    },
    {
      "epoch": 0.07360697087823893,
      "grad_norm": 0.1899557211808973,
      "learning_rate": 0.0009950301252541823,
      "loss": 3.6124,
      "step": 321
    },
    {
      "epoch": 0.0738362760834671,
      "grad_norm": 0.1687394036363737,
      "learning_rate": 0.000994977761098556,
      "loss": 3.6416,
      "step": 322
    },
    {
      "epoch": 0.07406558128869525,
      "grad_norm": 0.16074591347167283,
      "learning_rate": 0.0009949251239165075,
      "loss": 3.6863,
      "step": 323
    },
    {
      "epoch": 0.0742948864939234,
      "grad_norm": 0.15265684019203504,
      "learning_rate": 0.0009948722137370716,
      "loss": 3.6513,
      "step": 324
    },
    {
      "epoch": 0.07452419169915157,
      "grad_norm": 0.16208426659602074,
      "learning_rate": 0.000994819030589433,
      "loss": 3.582,
      "step": 325
    },
    {
      "epoch": 0.07475349690437973,
      "grad_norm": 0.14528935497346177,
      "learning_rate": 0.000994765574502927,
      "loss": 3.5695,
      "step": 326
    },
    {
      "epoch": 0.0749828021096079,
      "grad_norm": 0.17161169661742023,
      "learning_rate": 0.00099471184550704,
      "loss": 3.5054,
      "step": 327
    },
    {
      "epoch": 0.07521210731483605,
      "grad_norm": 0.1992621081115661,
      "learning_rate": 0.0009946578436314081,
      "loss": 3.7336,
      "step": 328
    },
    {
      "epoch": 0.0754414125200642,
      "grad_norm": 0.15047983178329513,
      "learning_rate": 0.0009946035689058189,
      "loss": 3.57,
      "step": 329
    },
    {
      "epoch": 0.07567071772529237,
      "grad_norm": 0.1491896118477208,
      "learning_rate": 0.0009945490213602093,
      "loss": 3.5983,
      "step": 330
    },
    {
      "epoch": 0.07590002293052052,
      "grad_norm": 0.1850903300664619,
      "learning_rate": 0.0009944942010246681,
      "loss": 3.5417,
      "step": 331
    },
    {
      "epoch": 0.07612932813574869,
      "grad_norm": 0.1506711636272506,
      "learning_rate": 0.0009944391079294334,
      "loss": 3.735,
      "step": 332
    },
    {
      "epoch": 0.07635863334097684,
      "grad_norm": 0.13787887881175712,
      "learning_rate": 0.0009943837421048942,
      "loss": 3.7196,
      "step": 333
    },
    {
      "epoch": 0.07658793854620499,
      "grad_norm": 0.20187781455636636,
      "learning_rate": 0.00099432810358159,
      "loss": 3.6787,
      "step": 334
    },
    {
      "epoch": 0.07681724375143316,
      "grad_norm": 0.15478295879406964,
      "learning_rate": 0.0009942721923902106,
      "loss": 3.6328,
      "step": 335
    },
    {
      "epoch": 0.07704654895666131,
      "grad_norm": 0.1487474562503357,
      "learning_rate": 0.0009942160085615961,
      "loss": 3.6324,
      "step": 336
    },
    {
      "epoch": 0.07727585416188948,
      "grad_norm": 0.1799026389856094,
      "learning_rate": 0.0009941595521267377,
      "loss": 3.7377,
      "step": 337
    },
    {
      "epoch": 0.07750515936711763,
      "grad_norm": 0.1546617893532859,
      "learning_rate": 0.0009941028231167757,
      "loss": 3.6509,
      "step": 338
    },
    {
      "epoch": 0.0777344645723458,
      "grad_norm": 0.15926047876853947,
      "learning_rate": 0.0009940458215630017,
      "loss": 3.6672,
      "step": 339
    },
    {
      "epoch": 0.07796376977757395,
      "grad_norm": 0.14623666742207328,
      "learning_rate": 0.0009939885474968572,
      "loss": 3.6403,
      "step": 340
    },
    {
      "epoch": 0.0781930749828021,
      "grad_norm": 0.18775898607129646,
      "learning_rate": 0.0009939310009499348,
      "loss": 3.6332,
      "step": 341
    },
    {
      "epoch": 0.07842238018803027,
      "grad_norm": 0.1411870416105716,
      "learning_rate": 0.0009938731819539761,
      "loss": 3.6266,
      "step": 342
    },
    {
      "epoch": 0.07865168539325842,
      "grad_norm": 0.15109612202444736,
      "learning_rate": 0.000993815090540874,
      "loss": 3.7417,
      "step": 343
    },
    {
      "epoch": 0.07888099059848659,
      "grad_norm": 0.16200175116035326,
      "learning_rate": 0.0009937567267426714,
      "loss": 3.7302,
      "step": 344
    },
    {
      "epoch": 0.07911029580371474,
      "grad_norm": 0.1602895050176464,
      "learning_rate": 0.000993698090591561,
      "loss": 3.644,
      "step": 345
    },
    {
      "epoch": 0.0793396010089429,
      "grad_norm": 0.15377695505299363,
      "learning_rate": 0.0009936391821198866,
      "loss": 3.585,
      "step": 346
    },
    {
      "epoch": 0.07956890621417106,
      "grad_norm": 0.16891505376370639,
      "learning_rate": 0.0009935800013601416,
      "loss": 3.5765,
      "step": 347
    },
    {
      "epoch": 0.07979821141939922,
      "grad_norm": 0.1759821469679735,
      "learning_rate": 0.0009935205483449694,
      "loss": 3.5599,
      "step": 348
    },
    {
      "epoch": 0.08002751662462738,
      "grad_norm": 0.14779734522802282,
      "learning_rate": 0.000993460823107164,
      "loss": 3.7366,
      "step": 349
    },
    {
      "epoch": 0.08025682182985554,
      "grad_norm": 0.17252093366608215,
      "learning_rate": 0.0009934008256796696,
      "loss": 3.6459,
      "step": 350
    },
    {
      "epoch": 0.08048612703508369,
      "grad_norm": 0.16128261512429712,
      "learning_rate": 0.0009933405560955803,
      "loss": 3.7103,
      "step": 351
    },
    {
      "epoch": 0.08071543224031186,
      "grad_norm": 0.15530162539031375,
      "learning_rate": 0.0009932800143881404,
      "loss": 3.7482,
      "step": 352
    },
    {
      "epoch": 0.08094473744554001,
      "grad_norm": 0.1571431056060872,
      "learning_rate": 0.0009932192005907446,
      "loss": 3.6192,
      "step": 353
    },
    {
      "epoch": 0.08117404265076818,
      "grad_norm": 0.14123023234182006,
      "learning_rate": 0.0009931581147369368,
      "loss": 3.5288,
      "step": 354
    },
    {
      "epoch": 0.08140334785599633,
      "grad_norm": 0.17981950083111473,
      "learning_rate": 0.0009930967568604118,
      "loss": 3.7045,
      "step": 355
    },
    {
      "epoch": 0.08163265306122448,
      "grad_norm": 0.18310178970067942,
      "learning_rate": 0.0009930351269950143,
      "loss": 3.6929,
      "step": 356
    },
    {
      "epoch": 0.08186195826645265,
      "grad_norm": 0.15364256625967238,
      "learning_rate": 0.000992973225174739,
      "loss": 3.7421,
      "step": 357
    },
    {
      "epoch": 0.0820912634716808,
      "grad_norm": 0.15343041942291252,
      "learning_rate": 0.00099291105143373,
      "loss": 3.7571,
      "step": 358
    },
    {
      "epoch": 0.08232056867690897,
      "grad_norm": 0.15103859178862697,
      "learning_rate": 0.0009928486058062827,
      "loss": 3.7532,
      "step": 359
    },
    {
      "epoch": 0.08254987388213712,
      "grad_norm": 0.18107852378134448,
      "learning_rate": 0.000992785888326841,
      "loss": 3.6043,
      "step": 360
    },
    {
      "epoch": 0.08277917908736529,
      "grad_norm": 0.19174508922858516,
      "learning_rate": 0.0009927228990299999,
      "loss": 3.8102,
      "step": 361
    },
    {
      "epoch": 0.08300848429259344,
      "grad_norm": 0.18916846975793303,
      "learning_rate": 0.0009926596379505037,
      "loss": 3.5898,
      "step": 362
    },
    {
      "epoch": 0.0832377894978216,
      "grad_norm": 0.13503563466263388,
      "learning_rate": 0.0009925961051232468,
      "loss": 3.6424,
      "step": 363
    },
    {
      "epoch": 0.08346709470304976,
      "grad_norm": 0.1675225327025962,
      "learning_rate": 0.0009925323005832735,
      "loss": 3.7685,
      "step": 364
    },
    {
      "epoch": 0.08369639990827792,
      "grad_norm": 0.17395260598835807,
      "learning_rate": 0.000992468224365778,
      "loss": 3.8257,
      "step": 365
    },
    {
      "epoch": 0.08392570511350608,
      "grad_norm": 0.1610235051790843,
      "learning_rate": 0.000992403876506104,
      "loss": 3.6938,
      "step": 366
    },
    {
      "epoch": 0.08415501031873424,
      "grad_norm": 0.14135668506610266,
      "learning_rate": 0.000992339257039746,
      "loss": 3.5997,
      "step": 367
    },
    {
      "epoch": 0.08438431552396239,
      "grad_norm": 0.1797352344845781,
      "learning_rate": 0.0009922743660023467,
      "loss": 3.6758,
      "step": 368
    },
    {
      "epoch": 0.08461362072919056,
      "grad_norm": 0.14190128812756203,
      "learning_rate": 0.0009922092034297006,
      "loss": 3.7014,
      "step": 369
    },
    {
      "epoch": 0.08484292593441871,
      "grad_norm": 0.14557136901260873,
      "learning_rate": 0.0009921437693577503,
      "loss": 3.6793,
      "step": 370
    },
    {
      "epoch": 0.08507223113964688,
      "grad_norm": 0.1478384563001219,
      "learning_rate": 0.0009920780638225891,
      "loss": 3.6237,
      "step": 371
    },
    {
      "epoch": 0.08530153634487503,
      "grad_norm": 0.16723563406738057,
      "learning_rate": 0.0009920120868604594,
      "loss": 3.6407,
      "step": 372
    },
    {
      "epoch": 0.08553084155010318,
      "grad_norm": 0.14785884563084387,
      "learning_rate": 0.0009919458385077538,
      "loss": 3.7461,
      "step": 373
    },
    {
      "epoch": 0.08576014675533135,
      "grad_norm": 0.13326055173216772,
      "learning_rate": 0.0009918793188010146,
      "loss": 3.5647,
      "step": 374
    },
    {
      "epoch": 0.0859894519605595,
      "grad_norm": 0.17648783875559199,
      "learning_rate": 0.0009918125277769336,
      "loss": 3.6311,
      "step": 375
    },
    {
      "epoch": 0.08621875716578767,
      "grad_norm": 0.1946815834108251,
      "learning_rate": 0.0009917454654723523,
      "loss": 3.6361,
      "step": 376
    },
    {
      "epoch": 0.08644806237101582,
      "grad_norm": 0.16919092340014708,
      "learning_rate": 0.0009916781319242614,
      "loss": 3.8322,
      "step": 377
    },
    {
      "epoch": 0.08667736757624397,
      "grad_norm": 0.155334023958672,
      "learning_rate": 0.0009916105271698021,
      "loss": 3.7656,
      "step": 378
    },
    {
      "epoch": 0.08690667278147214,
      "grad_norm": 0.15667144683998552,
      "learning_rate": 0.0009915426512462646,
      "loss": 3.6115,
      "step": 379
    },
    {
      "epoch": 0.0871359779867003,
      "grad_norm": 0.14434235676400325,
      "learning_rate": 0.0009914745041910888,
      "loss": 3.6314,
      "step": 380
    },
    {
      "epoch": 0.08736528319192846,
      "grad_norm": 0.19024469910054959,
      "learning_rate": 0.0009914060860418644,
      "loss": 3.5857,
      "step": 381
    },
    {
      "epoch": 0.08759458839715661,
      "grad_norm": 0.14280000270259563,
      "learning_rate": 0.0009913373968363298,
      "loss": 3.5134,
      "step": 382
    },
    {
      "epoch": 0.08782389360238478,
      "grad_norm": 0.15761136581508958,
      "learning_rate": 0.000991268436612374,
      "loss": 3.7306,
      "step": 383
    },
    {
      "epoch": 0.08805319880761293,
      "grad_norm": 0.17688831264081784,
      "learning_rate": 0.0009911992054080347,
      "loss": 3.5415,
      "step": 384
    },
    {
      "epoch": 0.08828250401284109,
      "grad_norm": 0.16658509663328921,
      "learning_rate": 0.0009911297032614997,
      "loss": 3.6606,
      "step": 385
    },
    {
      "epoch": 0.08851180921806925,
      "grad_norm": 0.1791029634131211,
      "learning_rate": 0.0009910599302111057,
      "loss": 3.6384,
      "step": 386
    },
    {
      "epoch": 0.08874111442329741,
      "grad_norm": 0.16360956786292052,
      "learning_rate": 0.000990989886295339,
      "loss": 3.9047,
      "step": 387
    },
    {
      "epoch": 0.08897041962852557,
      "grad_norm": 0.18833225915529367,
      "learning_rate": 0.0009909195715528355,
      "loss": 3.6981,
      "step": 388
    },
    {
      "epoch": 0.08919972483375373,
      "grad_norm": 0.20081356494345756,
      "learning_rate": 0.0009908489860223804,
      "loss": 3.7618,
      "step": 389
    },
    {
      "epoch": 0.08942903003898188,
      "grad_norm": 0.18905310684953205,
      "learning_rate": 0.000990778129742908,
      "loss": 3.5597,
      "step": 390
    },
    {
      "epoch": 0.08965833524421005,
      "grad_norm": 0.16576083557097063,
      "learning_rate": 0.000990707002753502,
      "loss": 3.7051,
      "step": 391
    },
    {
      "epoch": 0.0898876404494382,
      "grad_norm": 0.15812789545802655,
      "learning_rate": 0.000990635605093396,
      "loss": 3.5852,
      "step": 392
    },
    {
      "epoch": 0.09011694565466637,
      "grad_norm": 0.15541166925399003,
      "learning_rate": 0.0009905639368019724,
      "loss": 3.5655,
      "step": 393
    },
    {
      "epoch": 0.09034625085989452,
      "grad_norm": 0.16723991004745364,
      "learning_rate": 0.0009904919979187628,
      "loss": 3.5929,
      "step": 394
    },
    {
      "epoch": 0.09057555606512267,
      "grad_norm": 0.14179931025739612,
      "learning_rate": 0.0009904197884834482,
      "loss": 3.5291,
      "step": 395
    },
    {
      "epoch": 0.09080486127035084,
      "grad_norm": 0.18730052910486453,
      "learning_rate": 0.0009903473085358587,
      "loss": 3.5686,
      "step": 396
    },
    {
      "epoch": 0.09103416647557899,
      "grad_norm": 0.1505186687782768,
      "learning_rate": 0.0009902745581159742,
      "loss": 3.6718,
      "step": 397
    },
    {
      "epoch": 0.09126347168080716,
      "grad_norm": 0.19077555837156335,
      "learning_rate": 0.000990201537263923,
      "loss": 3.5848,
      "step": 398
    },
    {
      "epoch": 0.09149277688603531,
      "grad_norm": 0.1786668611218048,
      "learning_rate": 0.0009901282460199829,
      "loss": 3.6321,
      "step": 399
    },
    {
      "epoch": 0.09172208209126347,
      "grad_norm": 0.1696290599590613,
      "learning_rate": 0.000990054684424581,
      "loss": 3.7327,
      "step": 400
    },
    {
      "epoch": 0.09195138729649163,
      "grad_norm": 0.17882477229690152,
      "learning_rate": 0.0009899808525182935,
      "loss": 3.688,
      "step": 401
    },
    {
      "epoch": 0.09218069250171979,
      "grad_norm": 0.2090026110313227,
      "learning_rate": 0.0009899067503418453,
      "loss": 3.6424,
      "step": 402
    },
    {
      "epoch": 0.09240999770694795,
      "grad_norm": 0.15263174795991313,
      "learning_rate": 0.0009898323779361107,
      "loss": 3.5363,
      "step": 403
    },
    {
      "epoch": 0.0926393029121761,
      "grad_norm": 0.1515216887036575,
      "learning_rate": 0.0009897577353421131,
      "loss": 3.6111,
      "step": 404
    },
    {
      "epoch": 0.09286860811740426,
      "grad_norm": 0.14424110631286563,
      "learning_rate": 0.000989682822601025,
      "loss": 3.7185,
      "step": 405
    },
    {
      "epoch": 0.09309791332263243,
      "grad_norm": 0.1554444649406368,
      "learning_rate": 0.0009896076397541676,
      "loss": 3.5302,
      "step": 406
    },
    {
      "epoch": 0.09332721852786058,
      "grad_norm": 0.18149362711364395,
      "learning_rate": 0.0009895321868430113,
      "loss": 3.5536,
      "step": 407
    },
    {
      "epoch": 0.09355652373308875,
      "grad_norm": 0.17071865767847907,
      "learning_rate": 0.0009894564639091755,
      "loss": 3.6773,
      "step": 408
    },
    {
      "epoch": 0.0937858289383169,
      "grad_norm": 0.16177856262848794,
      "learning_rate": 0.0009893804709944281,
      "loss": 3.6754,
      "step": 409
    },
    {
      "epoch": 0.09401513414354506,
      "grad_norm": 0.1552648204030067,
      "learning_rate": 0.0009893042081406866,
      "loss": 3.7913,
      "step": 410
    },
    {
      "epoch": 0.09424443934877322,
      "grad_norm": 0.1586379372659671,
      "learning_rate": 0.0009892276753900174,
      "loss": 3.6977,
      "step": 411
    },
    {
      "epoch": 0.09447374455400137,
      "grad_norm": 0.15264217627993593,
      "learning_rate": 0.0009891508727846348,
      "loss": 3.6284,
      "step": 412
    },
    {
      "epoch": 0.09470304975922954,
      "grad_norm": 0.1586709185531381,
      "learning_rate": 0.0009890738003669028,
      "loss": 3.6622,
      "step": 413
    },
    {
      "epoch": 0.09493235496445769,
      "grad_norm": 0.1553849482589822,
      "learning_rate": 0.0009889964581793343,
      "loss": 3.6401,
      "step": 414
    },
    {
      "epoch": 0.09516166016968586,
      "grad_norm": 0.1483468950093554,
      "learning_rate": 0.0009889188462645904,
      "loss": 3.6595,
      "step": 415
    },
    {
      "epoch": 0.09539096537491401,
      "grad_norm": 0.1732983583969069,
      "learning_rate": 0.0009888409646654818,
      "loss": 3.8106,
      "step": 416
    },
    {
      "epoch": 0.09562027058014216,
      "grad_norm": 0.13445683790392718,
      "learning_rate": 0.0009887628134249667,
      "loss": 3.6286,
      "step": 417
    },
    {
      "epoch": 0.09584957578537033,
      "grad_norm": 0.1402615873876701,
      "learning_rate": 0.0009886843925861536,
      "loss": 3.6246,
      "step": 418
    },
    {
      "epoch": 0.09607888099059848,
      "grad_norm": 0.15035870701112555,
      "learning_rate": 0.0009886057021922983,
      "loss": 3.7073,
      "step": 419
    },
    {
      "epoch": 0.09630818619582665,
      "grad_norm": 0.15639700850239655,
      "learning_rate": 0.0009885267422868062,
      "loss": 3.7175,
      "step": 420
    },
    {
      "epoch": 0.0965374914010548,
      "grad_norm": 0.1519030097421034,
      "learning_rate": 0.0009884475129132311,
      "loss": 3.687,
      "step": 421
    },
    {
      "epoch": 0.09676679660628296,
      "grad_norm": 0.1386714919528708,
      "learning_rate": 0.0009883680141152752,
      "loss": 3.5762,
      "step": 422
    },
    {
      "epoch": 0.09699610181151112,
      "grad_norm": 0.1285352997724124,
      "learning_rate": 0.0009882882459367897,
      "loss": 3.6053,
      "step": 423
    },
    {
      "epoch": 0.09722540701673928,
      "grad_norm": 0.1398033696555147,
      "learning_rate": 0.0009882082084217742,
      "loss": 3.5839,
      "step": 424
    },
    {
      "epoch": 0.09745471222196744,
      "grad_norm": 0.16554990647011958,
      "learning_rate": 0.0009881279016143766,
      "loss": 3.6086,
      "step": 425
    },
    {
      "epoch": 0.0976840174271956,
      "grad_norm": 0.16838291244188747,
      "learning_rate": 0.0009880473255588936,
      "loss": 3.6663,
      "step": 426
    },
    {
      "epoch": 0.09791332263242375,
      "grad_norm": 0.22274061575365994,
      "learning_rate": 0.0009879664802997707,
      "loss": 3.6029,
      "step": 427
    },
    {
      "epoch": 0.09814262783765192,
      "grad_norm": 0.16690581565360166,
      "learning_rate": 0.0009878853658816015,
      "loss": 3.7775,
      "step": 428
    },
    {
      "epoch": 0.09837193304288007,
      "grad_norm": 0.16612471419999542,
      "learning_rate": 0.000987803982349128,
      "loss": 3.6803,
      "step": 429
    },
    {
      "epoch": 0.09860123824810824,
      "grad_norm": 0.2406528232330766,
      "learning_rate": 0.0009877223297472411,
      "loss": 3.7067,
      "step": 430
    },
    {
      "epoch": 0.09883054345333639,
      "grad_norm": 0.1747494566008279,
      "learning_rate": 0.0009876404081209796,
      "loss": 3.7048,
      "step": 431
    },
    {
      "epoch": 0.09905984865856456,
      "grad_norm": 0.1513186476950181,
      "learning_rate": 0.000987558217515531,
      "loss": 3.6436,
      "step": 432
    },
    {
      "epoch": 0.09928915386379271,
      "grad_norm": 0.171899821962293,
      "learning_rate": 0.000987475757976231,
      "loss": 3.6296,
      "step": 433
    },
    {
      "epoch": 0.09951845906902086,
      "grad_norm": 0.16721349430043553,
      "learning_rate": 0.000987393029548564,
      "loss": 3.5177,
      "step": 434
    },
    {
      "epoch": 0.09974776427424903,
      "grad_norm": 0.16324795382120347,
      "learning_rate": 0.000987310032278162,
      "loss": 3.7044,
      "step": 435
    },
    {
      "epoch": 0.09997706947947718,
      "grad_norm": 0.15547657502513637,
      "learning_rate": 0.0009872267662108063,
      "loss": 3.6716,
      "step": 436
    },
    {
      "epoch": 0.10020637468470535,
      "grad_norm": 0.1666319822653044,
      "learning_rate": 0.0009871432313924254,
      "loss": 3.7497,
      "step": 437
    },
    {
      "epoch": 0.1004356798899335,
      "grad_norm": 0.20357358304741457,
      "learning_rate": 0.000987059427869097,
      "loss": 3.7196,
      "step": 438
    },
    {
      "epoch": 0.10066498509516165,
      "grad_norm": 0.16049303587149427,
      "learning_rate": 0.000986975355687046,
      "loss": 3.6516,
      "step": 439
    },
    {
      "epoch": 0.10089429030038982,
      "grad_norm": 0.14387887239750416,
      "learning_rate": 0.0009868910148926465,
      "loss": 3.6167,
      "step": 440
    },
    {
      "epoch": 0.10112359550561797,
      "grad_norm": 0.1873175236508152,
      "learning_rate": 0.0009868064055324204,
      "loss": 3.6921,
      "step": 441
    },
    {
      "epoch": 0.10135290071084614,
      "grad_norm": 0.14680285964074463,
      "learning_rate": 0.0009867215276530372,
      "loss": 3.6962,
      "step": 442
    },
    {
      "epoch": 0.1015822059160743,
      "grad_norm": 0.13756257364255423,
      "learning_rate": 0.0009866363813013153,
      "loss": 3.6759,
      "step": 443
    },
    {
      "epoch": 0.10181151112130245,
      "grad_norm": 0.17565871511396178,
      "learning_rate": 0.0009865509665242208,
      "loss": 3.6991,
      "step": 444
    },
    {
      "epoch": 0.10204081632653061,
      "grad_norm": 0.20153120916308917,
      "learning_rate": 0.0009864652833688676,
      "loss": 3.7577,
      "step": 445
    },
    {
      "epoch": 0.10227012153175877,
      "grad_norm": 0.15895972827225077,
      "learning_rate": 0.0009863793318825185,
      "loss": 3.6254,
      "step": 446
    },
    {
      "epoch": 0.10249942673698693,
      "grad_norm": 0.17706338916937273,
      "learning_rate": 0.0009862931121125836,
      "loss": 3.7263,
      "step": 447
    },
    {
      "epoch": 0.10272873194221509,
      "grad_norm": 0.15679347838104327,
      "learning_rate": 0.000986206624106621,
      "loss": 3.6475,
      "step": 448
    },
    {
      "epoch": 0.10295803714744324,
      "grad_norm": 0.1543887083774493,
      "learning_rate": 0.000986119867912337,
      "loss": 3.6384,
      "step": 449
    },
    {
      "epoch": 0.10318734235267141,
      "grad_norm": 0.1802573692550327,
      "learning_rate": 0.0009860328435775856,
      "loss": 3.6811,
      "step": 450
    },
    {
      "epoch": 0.10341664755789956,
      "grad_norm": 0.16171401867495414,
      "learning_rate": 0.000985945551150369,
      "loss": 3.5729,
      "step": 451
    },
    {
      "epoch": 0.10364595276312773,
      "grad_norm": 0.16805177887449796,
      "learning_rate": 0.000985857990678837,
      "loss": 3.5369,
      "step": 452
    },
    {
      "epoch": 0.10387525796835588,
      "grad_norm": 0.15648952345245645,
      "learning_rate": 0.0009857701622112876,
      "loss": 3.8807,
      "step": 453
    },
    {
      "epoch": 0.10410456317358405,
      "grad_norm": 0.14737483603461066,
      "learning_rate": 0.0009856820657961661,
      "loss": 3.6625,
      "step": 454
    },
    {
      "epoch": 0.1043338683788122,
      "grad_norm": 0.13857979421001662,
      "learning_rate": 0.000985593701482066,
      "loss": 3.5973,
      "step": 455
    },
    {
      "epoch": 0.10456317358404035,
      "grad_norm": 0.14734907420779395,
      "learning_rate": 0.0009855050693177284,
      "loss": 3.768,
      "step": 456
    },
    {
      "epoch": 0.10479247878926852,
      "grad_norm": 0.14133426237357533,
      "learning_rate": 0.0009854161693520424,
      "loss": 3.5438,
      "step": 457
    },
    {
      "epoch": 0.10502178399449667,
      "grad_norm": 0.1693293754658771,
      "learning_rate": 0.0009853270016340447,
      "loss": 3.7755,
      "step": 458
    },
    {
      "epoch": 0.10525108919972484,
      "grad_norm": 0.13905207966352867,
      "learning_rate": 0.0009852375662129194,
      "loss": 3.523,
      "step": 459
    },
    {
      "epoch": 0.10548039440495299,
      "grad_norm": 0.14306459151308393,
      "learning_rate": 0.0009851478631379982,
      "loss": 3.5307,
      "step": 460
    },
    {
      "epoch": 0.10570969961018115,
      "grad_norm": 0.15899533261611098,
      "learning_rate": 0.0009850578924587613,
      "loss": 3.8136,
      "step": 461
    },
    {
      "epoch": 0.10593900481540931,
      "grad_norm": 0.16257202871654797,
      "learning_rate": 0.0009849676542248356,
      "loss": 3.5549,
      "step": 462
    },
    {
      "epoch": 0.10616831002063747,
      "grad_norm": 0.1686677495037545,
      "learning_rate": 0.000984877148485996,
      "loss": 3.6798,
      "step": 463
    },
    {
      "epoch": 0.10639761522586563,
      "grad_norm": 0.168460864914195,
      "learning_rate": 0.0009847863752921648,
      "loss": 3.7052,
      "step": 464
    },
    {
      "epoch": 0.10662692043109379,
      "grad_norm": 0.14758325575898823,
      "learning_rate": 0.000984695334693412,
      "loss": 3.4721,
      "step": 465
    },
    {
      "epoch": 0.10685622563632194,
      "grad_norm": 0.15565857992472984,
      "learning_rate": 0.0009846040267399547,
      "loss": 3.7322,
      "step": 466
    },
    {
      "epoch": 0.1070855308415501,
      "grad_norm": 0.14263600876902438,
      "learning_rate": 0.000984512451482158,
      "loss": 3.6584,
      "step": 467
    },
    {
      "epoch": 0.10731483604677826,
      "grad_norm": 0.15439393942065924,
      "learning_rate": 0.0009844206089705343,
      "loss": 3.5254,
      "step": 468
    },
    {
      "epoch": 0.10754414125200643,
      "grad_norm": 0.16107639531212778,
      "learning_rate": 0.0009843284992557431,
      "loss": 3.6317,
      "step": 469
    },
    {
      "epoch": 0.10777344645723458,
      "grad_norm": 0.16088139591770956,
      "learning_rate": 0.0009842361223885915,
      "loss": 3.5807,
      "step": 470
    },
    {
      "epoch": 0.10800275166246273,
      "grad_norm": 0.13820151643574324,
      "learning_rate": 0.000984143478420034,
      "loss": 3.5633,
      "step": 471
    },
    {
      "epoch": 0.1082320568676909,
      "grad_norm": 0.14348491989554554,
      "learning_rate": 0.0009840505674011727,
      "loss": 3.5891,
      "step": 472
    },
    {
      "epoch": 0.10846136207291905,
      "grad_norm": 0.1527233924569708,
      "learning_rate": 0.0009839573893832563,
      "loss": 3.6238,
      "step": 473
    },
    {
      "epoch": 0.10869066727814722,
      "grad_norm": 0.16495164677890695,
      "learning_rate": 0.0009838639444176812,
      "loss": 3.625,
      "step": 474
    },
    {
      "epoch": 0.10891997248337537,
      "grad_norm": 0.16037881203967,
      "learning_rate": 0.000983770232555991,
      "loss": 3.4839,
      "step": 475
    },
    {
      "epoch": 0.10914927768860354,
      "grad_norm": 0.15568128481938126,
      "learning_rate": 0.0009836762538498768,
      "loss": 3.7129,
      "step": 476
    },
    {
      "epoch": 0.10937858289383169,
      "grad_norm": 0.14022458531322515,
      "learning_rate": 0.0009835820083511765,
      "loss": 3.6346,
      "step": 477
    },
    {
      "epoch": 0.10960788809905984,
      "grad_norm": 0.15170335072775246,
      "learning_rate": 0.0009834874961118754,
      "loss": 3.6257,
      "step": 478
    },
    {
      "epoch": 0.10983719330428801,
      "grad_norm": 0.1649272335334313,
      "learning_rate": 0.0009833927171841055,
      "loss": 3.6884,
      "step": 479
    },
    {
      "epoch": 0.11006649850951616,
      "grad_norm": 0.13257368512541515,
      "learning_rate": 0.0009832976716201466,
      "loss": 3.6211,
      "step": 480
    },
    {
      "epoch": 0.11029580371474433,
      "grad_norm": 0.17563823374706417,
      "learning_rate": 0.0009832023594724246,
      "loss": 3.694,
      "step": 481
    },
    {
      "epoch": 0.11052510891997248,
      "grad_norm": 0.14949656016005006,
      "learning_rate": 0.0009831067807935137,
      "loss": 3.5494,
      "step": 482
    },
    {
      "epoch": 0.11075441412520064,
      "grad_norm": 0.14796188814264344,
      "learning_rate": 0.0009830109356361344,
      "loss": 3.6953,
      "step": 483
    },
    {
      "epoch": 0.1109837193304288,
      "grad_norm": 0.14445104733249467,
      "learning_rate": 0.0009829148240531537,
      "loss": 3.6585,
      "step": 484
    },
    {
      "epoch": 0.11121302453565696,
      "grad_norm": 0.16580336140645385,
      "learning_rate": 0.0009828184460975867,
      "loss": 3.6775,
      "step": 485
    },
    {
      "epoch": 0.11144232974088512,
      "grad_norm": 0.18408234387422356,
      "learning_rate": 0.0009827218018225944,
      "loss": 3.5344,
      "step": 486
    },
    {
      "epoch": 0.11167163494611328,
      "grad_norm": 0.16796969659134525,
      "learning_rate": 0.0009826248912814855,
      "loss": 3.6897,
      "step": 487
    },
    {
      "epoch": 0.11190094015134143,
      "grad_norm": 0.14118383641075266,
      "learning_rate": 0.0009825277145277147,
      "loss": 3.6896,
      "step": 488
    },
    {
      "epoch": 0.1121302453565696,
      "grad_norm": 0.18898782481018137,
      "learning_rate": 0.0009824302716148847,
      "loss": 3.7356,
      "step": 489
    },
    {
      "epoch": 0.11235955056179775,
      "grad_norm": 0.1392857465141545,
      "learning_rate": 0.0009823325625967442,
      "loss": 3.6553,
      "step": 490
    },
    {
      "epoch": 0.11258885576702592,
      "grad_norm": 0.1604129910736065,
      "learning_rate": 0.0009822345875271884,
      "loss": 3.5545,
      "step": 491
    },
    {
      "epoch": 0.11281816097225407,
      "grad_norm": 0.12954397953367894,
      "learning_rate": 0.00098213634646026,
      "loss": 3.6443,
      "step": 492
    },
    {
      "epoch": 0.11304746617748222,
      "grad_norm": 0.15165065227699925,
      "learning_rate": 0.0009820378394501481,
      "loss": 3.8314,
      "step": 493
    },
    {
      "epoch": 0.11327677138271039,
      "grad_norm": 0.16323952359992575,
      "learning_rate": 0.0009819390665511884,
      "loss": 3.7005,
      "step": 494
    },
    {
      "epoch": 0.11350607658793854,
      "grad_norm": 0.14463740787520465,
      "learning_rate": 0.0009818400278178636,
      "loss": 3.6483,
      "step": 495
    },
    {
      "epoch": 0.11373538179316671,
      "grad_norm": 0.17326238668752716,
      "learning_rate": 0.0009817407233048028,
      "loss": 3.5974,
      "step": 496
    },
    {
      "epoch": 0.11396468699839486,
      "grad_norm": 0.17800273593315957,
      "learning_rate": 0.0009816411530667814,
      "loss": 3.7208,
      "step": 497
    },
    {
      "epoch": 0.11419399220362302,
      "grad_norm": 0.15433696362251292,
      "learning_rate": 0.0009815413171587219,
      "loss": 3.6621,
      "step": 498
    },
    {
      "epoch": 0.11442329740885118,
      "grad_norm": 0.17547421266636393,
      "learning_rate": 0.000981441215635693,
      "loss": 3.3962,
      "step": 499
    },
    {
      "epoch": 0.11465260261407934,
      "grad_norm": 0.14798017203844788,
      "learning_rate": 0.0009813408485529103,
      "loss": 3.5671,
      "step": 500
    },
    {
      "epoch": 0.1148819078193075,
      "grad_norm": 0.15344366535003068,
      "learning_rate": 0.0009812402159657353,
      "loss": 3.6083,
      "step": 501
    },
    {
      "epoch": 0.11511121302453566,
      "grad_norm": 0.14035469710102216,
      "learning_rate": 0.000981139317929676,
      "loss": 3.6005,
      "step": 502
    },
    {
      "epoch": 0.11534051822976382,
      "grad_norm": 0.1269886962911051,
      "learning_rate": 0.000981038154500388,
      "loss": 3.5013,
      "step": 503
    },
    {
      "epoch": 0.11556982343499198,
      "grad_norm": 0.15323355584581275,
      "learning_rate": 0.0009809367257336717,
      "loss": 3.4852,
      "step": 504
    },
    {
      "epoch": 0.11579912864022013,
      "grad_norm": 0.1256690892876832,
      "learning_rate": 0.0009808350316854746,
      "loss": 3.6904,
      "step": 505
    },
    {
      "epoch": 0.1160284338454483,
      "grad_norm": 0.13537918357189757,
      "learning_rate": 0.0009807330724118904,
      "loss": 3.5833,
      "step": 506
    },
    {
      "epoch": 0.11625773905067645,
      "grad_norm": 0.15583747681749938,
      "learning_rate": 0.0009806308479691594,
      "loss": 3.6006,
      "step": 507
    },
    {
      "epoch": 0.11648704425590461,
      "grad_norm": 0.14527652070430463,
      "learning_rate": 0.000980528358413668,
      "loss": 3.543,
      "step": 508
    },
    {
      "epoch": 0.11671634946113277,
      "grad_norm": 0.1514035668702159,
      "learning_rate": 0.0009804256038019482,
      "loss": 3.7017,
      "step": 509
    },
    {
      "epoch": 0.11694565466636092,
      "grad_norm": 0.14768705267007004,
      "learning_rate": 0.0009803225841906792,
      "loss": 3.6902,
      "step": 510
    },
    {
      "epoch": 0.11717495987158909,
      "grad_norm": 0.148374133074262,
      "learning_rate": 0.0009802192996366857,
      "loss": 3.7185,
      "step": 511
    },
    {
      "epoch": 0.11740426507681724,
      "grad_norm": 0.17415459745270576,
      "learning_rate": 0.000980115750196939,
      "loss": 3.6125,
      "step": 512
    },
    {
      "epoch": 0.11763357028204541,
      "grad_norm": 0.15055428549367497,
      "learning_rate": 0.0009800119359285563,
      "loss": 3.6708,
      "step": 513
    },
    {
      "epoch": 0.11786287548727356,
      "grad_norm": 0.15456315526580297,
      "learning_rate": 0.0009799078568888007,
      "loss": 3.6687,
      "step": 514
    },
    {
      "epoch": 0.11809218069250171,
      "grad_norm": 0.15226466795507201,
      "learning_rate": 0.0009798035131350813,
      "loss": 3.5906,
      "step": 515
    },
    {
      "epoch": 0.11832148589772988,
      "grad_norm": 0.13507666041904784,
      "learning_rate": 0.0009796989047249538,
      "loss": 3.5767,
      "step": 516
    },
    {
      "epoch": 0.11855079110295803,
      "grad_norm": 0.1419615783802049,
      "learning_rate": 0.0009795940317161194,
      "loss": 3.673,
      "step": 517
    },
    {
      "epoch": 0.1187800963081862,
      "grad_norm": 0.13603613432963302,
      "learning_rate": 0.0009794888941664253,
      "loss": 3.6312,
      "step": 518
    },
    {
      "epoch": 0.11900940151341435,
      "grad_norm": 0.14250777615016266,
      "learning_rate": 0.0009793834921338646,
      "loss": 3.6614,
      "step": 519
    },
    {
      "epoch": 0.1192387067186425,
      "grad_norm": 0.14618009820307054,
      "learning_rate": 0.0009792778256765763,
      "loss": 3.5958,
      "step": 520
    },
    {
      "epoch": 0.11946801192387067,
      "grad_norm": 0.13368865411946468,
      "learning_rate": 0.0009791718948528457,
      "loss": 3.6292,
      "step": 521
    },
    {
      "epoch": 0.11969731712909883,
      "grad_norm": 0.13971253627039387,
      "learning_rate": 0.0009790656997211032,
      "loss": 3.7153,
      "step": 522
    },
    {
      "epoch": 0.119926622334327,
      "grad_norm": 0.1663648423583044,
      "learning_rate": 0.0009789592403399252,
      "loss": 3.5633,
      "step": 523
    },
    {
      "epoch": 0.12015592753955515,
      "grad_norm": 0.15834587564468747,
      "learning_rate": 0.0009788525167680345,
      "loss": 3.6948,
      "step": 524
    },
    {
      "epoch": 0.12038523274478331,
      "grad_norm": 0.13744383770175675,
      "learning_rate": 0.0009787455290642985,
      "loss": 3.5881,
      "step": 525
    },
    {
      "epoch": 0.12061453795001147,
      "grad_norm": 0.15129229087749727,
      "learning_rate": 0.0009786382772877312,
      "loss": 3.5392,
      "step": 526
    },
    {
      "epoch": 0.12084384315523962,
      "grad_norm": 0.1465585751346376,
      "learning_rate": 0.000978530761497492,
      "loss": 3.6872,
      "step": 527
    },
    {
      "epoch": 0.12107314836046779,
      "grad_norm": 0.14237493397908335,
      "learning_rate": 0.0009784229817528859,
      "loss": 3.6927,
      "step": 528
    },
    {
      "epoch": 0.12130245356569594,
      "grad_norm": 0.18316827569767072,
      "learning_rate": 0.0009783149381133633,
      "loss": 3.8145,
      "step": 529
    },
    {
      "epoch": 0.1215317587709241,
      "grad_norm": 0.1524054907959129,
      "learning_rate": 0.0009782066306385207,
      "loss": 3.6395,
      "step": 530
    },
    {
      "epoch": 0.12176106397615226,
      "grad_norm": 0.13959864984398487,
      "learning_rate": 0.0009780980593880992,
      "loss": 3.7122,
      "step": 531
    },
    {
      "epoch": 0.12199036918138041,
      "grad_norm": 0.1382782999375935,
      "learning_rate": 0.0009779892244219865,
      "loss": 3.6805,
      "step": 532
    },
    {
      "epoch": 0.12221967438660858,
      "grad_norm": 0.15734917485066058,
      "learning_rate": 0.0009778801258002153,
      "loss": 3.5592,
      "step": 533
    },
    {
      "epoch": 0.12244897959183673,
      "grad_norm": 0.14814683739013965,
      "learning_rate": 0.0009777707635829631,
      "loss": 3.6934,
      "step": 534
    },
    {
      "epoch": 0.1226782847970649,
      "grad_norm": 0.15735240360119138,
      "learning_rate": 0.000977661137830554,
      "loss": 3.6362,
      "step": 535
    },
    {
      "epoch": 0.12290759000229305,
      "grad_norm": 0.14183825218890475,
      "learning_rate": 0.0009775512486034563,
      "loss": 3.6802,
      "step": 536
    },
    {
      "epoch": 0.1231368952075212,
      "grad_norm": 0.14408090644532504,
      "learning_rate": 0.0009774410959622845,
      "loss": 3.546,
      "step": 537
    },
    {
      "epoch": 0.12336620041274937,
      "grad_norm": 0.14412506895769767,
      "learning_rate": 0.0009773306799677978,
      "loss": 3.667,
      "step": 538
    },
    {
      "epoch": 0.12359550561797752,
      "grad_norm": 0.1467324919297679,
      "learning_rate": 0.000977220000680901,
      "loss": 3.6713,
      "step": 539
    },
    {
      "epoch": 0.12382481082320569,
      "grad_norm": 0.16372913079744505,
      "learning_rate": 0.0009771090581626445,
      "loss": 3.6296,
      "step": 540
    },
    {
      "epoch": 0.12405411602843384,
      "grad_norm": 0.15099315593040102,
      "learning_rate": 0.000976997852474223,
      "loss": 3.6588,
      "step": 541
    },
    {
      "epoch": 0.124283421233662,
      "grad_norm": 0.15209156312692404,
      "learning_rate": 0.0009768863836769769,
      "loss": 3.6166,
      "step": 542
    },
    {
      "epoch": 0.12451272643889016,
      "grad_norm": 0.13332835495485942,
      "learning_rate": 0.0009767746518323914,
      "loss": 3.6407,
      "step": 543
    },
    {
      "epoch": 0.12474203164411832,
      "grad_norm": 0.15088511510960761,
      "learning_rate": 0.0009766626570020974,
      "loss": 3.6808,
      "step": 544
    },
    {
      "epoch": 0.12497133684934648,
      "grad_norm": 0.1481050791297745,
      "learning_rate": 0.0009765503992478704,
      "loss": 3.7565,
      "step": 545
    },
    {
      "epoch": 0.12520064205457465,
      "grad_norm": 0.13869761052371274,
      "learning_rate": 0.000976437878631631,
      "loss": 3.5443,
      "step": 546
    },
    {
      "epoch": 0.1254299472598028,
      "grad_norm": 0.14373627815898438,
      "learning_rate": 0.0009763250952154449,
      "loss": 3.6949,
      "step": 547
    },
    {
      "epoch": 0.12565925246503096,
      "grad_norm": 0.13536021441215906,
      "learning_rate": 0.0009762120490615225,
      "loss": 3.6,
      "step": 548
    },
    {
      "epoch": 0.1258885576702591,
      "grad_norm": 0.1324155181670053,
      "learning_rate": 0.0009760987402322195,
      "loss": 3.5363,
      "step": 549
    },
    {
      "epoch": 0.12611786287548726,
      "grad_norm": 0.12952017763572696,
      "learning_rate": 0.000975985168790036,
      "loss": 3.5153,
      "step": 550
    },
    {
      "epoch": 0.12634716808071544,
      "grad_norm": 0.1405173993153049,
      "learning_rate": 0.0009758713347976178,
      "loss": 3.7185,
      "step": 551
    },
    {
      "epoch": 0.1265764732859436,
      "grad_norm": 0.13117305997048287,
      "learning_rate": 0.0009757572383177545,
      "loss": 3.5468,
      "step": 552
    },
    {
      "epoch": 0.12680577849117175,
      "grad_norm": 0.1329779359984029,
      "learning_rate": 0.000975642879413381,
      "loss": 3.6883,
      "step": 553
    },
    {
      "epoch": 0.1270350836963999,
      "grad_norm": 0.1460460692545784,
      "learning_rate": 0.0009755282581475768,
      "loss": 3.7274,
      "step": 554
    },
    {
      "epoch": 0.12726438890162806,
      "grad_norm": 0.14661709792906447,
      "learning_rate": 0.0009754133745835665,
      "loss": 3.524,
      "step": 555
    },
    {
      "epoch": 0.12749369410685624,
      "grad_norm": 0.13861596473828358,
      "learning_rate": 0.0009752982287847192,
      "loss": 3.7265,
      "step": 556
    },
    {
      "epoch": 0.1277229993120844,
      "grad_norm": 0.1460535881643949,
      "learning_rate": 0.0009751828208145482,
      "loss": 3.6511,
      "step": 557
    },
    {
      "epoch": 0.12795230451731254,
      "grad_norm": 0.14303426535576522,
      "learning_rate": 0.0009750671507367119,
      "loss": 3.6468,
      "step": 558
    },
    {
      "epoch": 0.1281816097225407,
      "grad_norm": 0.12992475817168983,
      "learning_rate": 0.0009749512186150131,
      "loss": 3.6993,
      "step": 559
    },
    {
      "epoch": 0.12841091492776885,
      "grad_norm": 0.13792461557022143,
      "learning_rate": 0.0009748350245133994,
      "loss": 3.656,
      "step": 560
    },
    {
      "epoch": 0.12864022013299703,
      "grad_norm": 0.1324533866373334,
      "learning_rate": 0.0009747185684959625,
      "loss": 3.6454,
      "step": 561
    },
    {
      "epoch": 0.12886952533822518,
      "grad_norm": 0.17606541000510273,
      "learning_rate": 0.0009746018506269387,
      "loss": 3.5993,
      "step": 562
    },
    {
      "epoch": 0.12909883054345334,
      "grad_norm": 0.15395495768501188,
      "learning_rate": 0.000974484870970709,
      "loss": 3.643,
      "step": 563
    },
    {
      "epoch": 0.1293281357486815,
      "grad_norm": 0.153230625620958,
      "learning_rate": 0.0009743676295917985,
      "loss": 3.6277,
      "step": 564
    },
    {
      "epoch": 0.12955744095390964,
      "grad_norm": 0.14577485453399713,
      "learning_rate": 0.0009742501265548767,
      "loss": 3.6388,
      "step": 565
    },
    {
      "epoch": 0.12978674615913782,
      "grad_norm": 0.1305102419195435,
      "learning_rate": 0.0009741323619247575,
      "loss": 3.5718,
      "step": 566
    },
    {
      "epoch": 0.13001605136436598,
      "grad_norm": 0.1327056791345219,
      "learning_rate": 0.0009740143357663993,
      "loss": 3.5309,
      "step": 567
    },
    {
      "epoch": 0.13024535656959413,
      "grad_norm": 0.13943216803696784,
      "learning_rate": 0.0009738960481449042,
      "loss": 3.7394,
      "step": 568
    },
    {
      "epoch": 0.13047466177482228,
      "grad_norm": 0.13526467810139628,
      "learning_rate": 0.000973777499125519,
      "loss": 3.6537,
      "step": 569
    },
    {
      "epoch": 0.13070396698005043,
      "grad_norm": 0.1572572677433108,
      "learning_rate": 0.0009736586887736346,
      "loss": 3.583,
      "step": 570
    },
    {
      "epoch": 0.13093327218527862,
      "grad_norm": 0.14387959887354335,
      "learning_rate": 0.0009735396171547859,
      "loss": 3.6385,
      "step": 571
    },
    {
      "epoch": 0.13116257739050677,
      "grad_norm": 0.13297884086713804,
      "learning_rate": 0.0009734202843346522,
      "loss": 3.5786,
      "step": 572
    },
    {
      "epoch": 0.13139188259573492,
      "grad_norm": 0.16988698540629307,
      "learning_rate": 0.0009733006903790564,
      "loss": 3.652,
      "step": 573
    },
    {
      "epoch": 0.13162118780096307,
      "grad_norm": 0.17217668745322817,
      "learning_rate": 0.0009731808353539659,
      "loss": 3.583,
      "step": 574
    },
    {
      "epoch": 0.13185049300619123,
      "grad_norm": 0.15695310465569146,
      "learning_rate": 0.0009730607193254922,
      "loss": 3.6956,
      "step": 575
    },
    {
      "epoch": 0.1320797982114194,
      "grad_norm": 0.1437811787586104,
      "learning_rate": 0.0009729403423598899,
      "loss": 3.6935,
      "step": 576
    },
    {
      "epoch": 0.13230910341664756,
      "grad_norm": 0.18465629681157944,
      "learning_rate": 0.0009728197045235585,
      "loss": 3.6598,
      "step": 577
    },
    {
      "epoch": 0.13253840862187571,
      "grad_norm": 0.16619116838695497,
      "learning_rate": 0.0009726988058830409,
      "loss": 3.7813,
      "step": 578
    },
    {
      "epoch": 0.13276771382710387,
      "grad_norm": 0.18712464893518516,
      "learning_rate": 0.0009725776465050242,
      "loss": 3.7055,
      "step": 579
    },
    {
      "epoch": 0.13299701903233205,
      "grad_norm": 0.14667297280562322,
      "learning_rate": 0.0009724562264563388,
      "loss": 3.4937,
      "step": 580
    },
    {
      "epoch": 0.1332263242375602,
      "grad_norm": 0.15283387370062623,
      "learning_rate": 0.0009723345458039594,
      "loss": 3.5446,
      "step": 581
    },
    {
      "epoch": 0.13345562944278835,
      "grad_norm": 0.16884747702029598,
      "learning_rate": 0.0009722126046150042,
      "loss": 3.7389,
      "step": 582
    },
    {
      "epoch": 0.1336849346480165,
      "grad_norm": 0.15401177970509622,
      "learning_rate": 0.000972090402956735,
      "loss": 3.6912,
      "step": 583
    },
    {
      "epoch": 0.13391423985324466,
      "grad_norm": 0.130889340740142,
      "learning_rate": 0.0009719679408965576,
      "loss": 3.6256,
      "step": 584
    },
    {
      "epoch": 0.13414354505847284,
      "grad_norm": 0.14808347718105036,
      "learning_rate": 0.0009718452185020212,
      "loss": 3.6865,
      "step": 585
    },
    {
      "epoch": 0.134372850263701,
      "grad_norm": 0.17826003597051163,
      "learning_rate": 0.0009717222358408186,
      "loss": 3.5852,
      "step": 586
    },
    {
      "epoch": 0.13460215546892915,
      "grad_norm": 0.16672248984785912,
      "learning_rate": 0.0009715989929807862,
      "loss": 3.7678,
      "step": 587
    },
    {
      "epoch": 0.1348314606741573,
      "grad_norm": 0.1630185952670217,
      "learning_rate": 0.0009714754899899041,
      "loss": 3.5379,
      "step": 588
    },
    {
      "epoch": 0.13506076587938545,
      "grad_norm": 0.14752623912159443,
      "learning_rate": 0.0009713517269362955,
      "loss": 3.6207,
      "step": 589
    },
    {
      "epoch": 0.13529007108461363,
      "grad_norm": 0.14729697085470136,
      "learning_rate": 0.0009712277038882273,
      "loss": 3.6896,
      "step": 590
    },
    {
      "epoch": 0.1355193762898418,
      "grad_norm": 0.1571003004841232,
      "learning_rate": 0.0009711034209141101,
      "loss": 3.6273,
      "step": 591
    },
    {
      "epoch": 0.13574868149506994,
      "grad_norm": 0.15399551216145646,
      "learning_rate": 0.0009709788780824971,
      "loss": 3.6186,
      "step": 592
    },
    {
      "epoch": 0.1359779867002981,
      "grad_norm": 0.1298437886458736,
      "learning_rate": 0.0009708540754620856,
      "loss": 3.6306,
      "step": 593
    },
    {
      "epoch": 0.13620729190552625,
      "grad_norm": 0.16046916617576815,
      "learning_rate": 0.0009707290131217159,
      "loss": 3.7428,
      "step": 594
    },
    {
      "epoch": 0.13643659711075443,
      "grad_norm": 0.14060355891305995,
      "learning_rate": 0.0009706036911303713,
      "loss": 3.6183,
      "step": 595
    },
    {
      "epoch": 0.13666590231598258,
      "grad_norm": 0.14880596674514926,
      "learning_rate": 0.0009704781095571787,
      "loss": 3.6716,
      "step": 596
    },
    {
      "epoch": 0.13689520752121073,
      "grad_norm": 0.13673089712682326,
      "learning_rate": 0.0009703522684714083,
      "loss": 3.6485,
      "step": 597
    },
    {
      "epoch": 0.13712451272643889,
      "grad_norm": 0.13628258509351315,
      "learning_rate": 0.0009702261679424728,
      "loss": 3.6245,
      "step": 598
    },
    {
      "epoch": 0.13735381793166704,
      "grad_norm": 0.1489727580728453,
      "learning_rate": 0.0009700998080399286,
      "loss": 3.5562,
      "step": 599
    },
    {
      "epoch": 0.13758312313689522,
      "grad_norm": 0.13911421355045916,
      "learning_rate": 0.000969973188833475,
      "loss": 3.6505,
      "step": 600
    },
    {
      "epoch": 0.13781242834212337,
      "grad_norm": 0.1555393950689739,
      "learning_rate": 0.0009698463103929542,
      "loss": 3.5435,
      "step": 601
    },
    {
      "epoch": 0.13804173354735153,
      "grad_norm": 0.13147609805478155,
      "learning_rate": 0.0009697191727883516,
      "loss": 3.5763,
      "step": 602
    },
    {
      "epoch": 0.13827103875257968,
      "grad_norm": 0.13701232703539107,
      "learning_rate": 0.0009695917760897954,
      "loss": 3.5543,
      "step": 603
    },
    {
      "epoch": 0.13850034395780783,
      "grad_norm": 0.13247047829013814,
      "learning_rate": 0.0009694641203675569,
      "loss": 3.6347,
      "step": 604
    },
    {
      "epoch": 0.138729649163036,
      "grad_norm": 0.1299647262479322,
      "learning_rate": 0.0009693362056920501,
      "loss": 3.5199,
      "step": 605
    },
    {
      "epoch": 0.13895895436826416,
      "grad_norm": 0.13337770218276176,
      "learning_rate": 0.0009692080321338317,
      "loss": 3.6231,
      "step": 606
    },
    {
      "epoch": 0.13918825957349232,
      "grad_norm": 0.1639918747760781,
      "learning_rate": 0.0009690795997636015,
      "loss": 3.6096,
      "step": 607
    },
    {
      "epoch": 0.13941756477872047,
      "grad_norm": 0.16267039351959373,
      "learning_rate": 0.0009689509086522018,
      "loss": 3.6891,
      "step": 608
    },
    {
      "epoch": 0.13964686998394862,
      "grad_norm": 0.1614961758956938,
      "learning_rate": 0.0009688219588706179,
      "loss": 3.6893,
      "step": 609
    },
    {
      "epoch": 0.1398761751891768,
      "grad_norm": 0.1715129684629224,
      "learning_rate": 0.0009686927504899778,
      "loss": 3.7862,
      "step": 610
    },
    {
      "epoch": 0.14010548039440496,
      "grad_norm": 0.1624500935239744,
      "learning_rate": 0.0009685632835815518,
      "loss": 3.5943,
      "step": 611
    },
    {
      "epoch": 0.1403347855996331,
      "grad_norm": 0.16087213171821174,
      "learning_rate": 0.0009684335582167529,
      "loss": 3.5317,
      "step": 612
    },
    {
      "epoch": 0.14056409080486126,
      "grad_norm": 0.14019229566133107,
      "learning_rate": 0.0009683035744671367,
      "loss": 3.5285,
      "step": 613
    },
    {
      "epoch": 0.14079339601008942,
      "grad_norm": 0.14838899276272588,
      "learning_rate": 0.0009681733324044017,
      "loss": 3.5963,
      "step": 614
    },
    {
      "epoch": 0.1410227012153176,
      "grad_norm": 0.1522897055568193,
      "learning_rate": 0.0009680428321003883,
      "loss": 3.5891,
      "step": 615
    },
    {
      "epoch": 0.14125200642054575,
      "grad_norm": 0.13337878390985922,
      "learning_rate": 0.0009679120736270794,
      "loss": 3.4381,
      "step": 616
    },
    {
      "epoch": 0.1414813116257739,
      "grad_norm": 0.1390533659447495,
      "learning_rate": 0.000967781057056601,
      "loss": 3.468,
      "step": 617
    },
    {
      "epoch": 0.14171061683100206,
      "grad_norm": 0.18384986099951334,
      "learning_rate": 0.0009676497824612206,
      "loss": 3.6386,
      "step": 618
    },
    {
      "epoch": 0.1419399220362302,
      "grad_norm": 0.14651238669272565,
      "learning_rate": 0.0009675182499133485,
      "loss": 3.7051,
      "step": 619
    },
    {
      "epoch": 0.1421692272414584,
      "grad_norm": 0.1853348407625658,
      "learning_rate": 0.0009673864594855372,
      "loss": 3.4375,
      "step": 620
    },
    {
      "epoch": 0.14239853244668654,
      "grad_norm": 0.1506028115123924,
      "learning_rate": 0.0009672544112504813,
      "loss": 3.5438,
      "step": 621
    },
    {
      "epoch": 0.1426278376519147,
      "grad_norm": 0.17134709790129632,
      "learning_rate": 0.0009671221052810178,
      "loss": 3.6406,
      "step": 622
    },
    {
      "epoch": 0.14285714285714285,
      "grad_norm": 0.15575252697914296,
      "learning_rate": 0.0009669895416501257,
      "loss": 3.5564,
      "step": 623
    },
    {
      "epoch": 0.143086448062371,
      "grad_norm": 0.13961638096657114,
      "learning_rate": 0.0009668567204309265,
      "loss": 3.6022,
      "step": 624
    },
    {
      "epoch": 0.14331575326759918,
      "grad_norm": 0.14952746226126573,
      "learning_rate": 0.0009667236416966833,
      "loss": 3.5782,
      "step": 625
    },
    {
      "epoch": 0.14354505847282734,
      "grad_norm": 0.1441788335964229,
      "learning_rate": 0.0009665903055208013,
      "loss": 3.5659,
      "step": 626
    },
    {
      "epoch": 0.1437743636780555,
      "grad_norm": 0.17496850171556463,
      "learning_rate": 0.0009664567119768281,
      "loss": 3.5771,
      "step": 627
    },
    {
      "epoch": 0.14400366888328364,
      "grad_norm": 0.13200500754392516,
      "learning_rate": 0.0009663228611384527,
      "loss": 3.6026,
      "step": 628
    },
    {
      "epoch": 0.14423297408851182,
      "grad_norm": 0.13087187480382664,
      "learning_rate": 0.0009661887530795067,
      "loss": 3.6612,
      "step": 629
    },
    {
      "epoch": 0.14446227929373998,
      "grad_norm": 0.13428504031596478,
      "learning_rate": 0.0009660543878739629,
      "loss": 3.6036,
      "step": 630
    },
    {
      "epoch": 0.14469158449896813,
      "grad_norm": 0.1399332040407389,
      "learning_rate": 0.0009659197655959365,
      "loss": 3.554,
      "step": 631
    },
    {
      "epoch": 0.14492088970419628,
      "grad_norm": 0.13340340157896727,
      "learning_rate": 0.000965784886319684,
      "loss": 3.6081,
      "step": 632
    },
    {
      "epoch": 0.14515019490942443,
      "grad_norm": 0.14644296760343642,
      "learning_rate": 0.000965649750119604,
      "loss": 3.5313,
      "step": 633
    },
    {
      "epoch": 0.14537950011465262,
      "grad_norm": 0.1265987796960822,
      "learning_rate": 0.0009655143570702369,
      "loss": 3.7208,
      "step": 634
    },
    {
      "epoch": 0.14560880531988077,
      "grad_norm": 0.1294534904346438,
      "learning_rate": 0.0009653787072462643,
      "loss": 3.6635,
      "step": 635
    },
    {
      "epoch": 0.14583811052510892,
      "grad_norm": 0.14459735636120638,
      "learning_rate": 0.0009652428007225099,
      "loss": 3.6008,
      "step": 636
    },
    {
      "epoch": 0.14606741573033707,
      "grad_norm": 0.14202994310257286,
      "learning_rate": 0.0009651066375739388,
      "loss": 3.6391,
      "step": 637
    },
    {
      "epoch": 0.14629672093556523,
      "grad_norm": 0.14913108134952235,
      "learning_rate": 0.0009649702178756578,
      "loss": 3.6542,
      "step": 638
    },
    {
      "epoch": 0.1465260261407934,
      "grad_norm": 0.1517301974961052,
      "learning_rate": 0.000964833541702915,
      "loss": 3.5274,
      "step": 639
    },
    {
      "epoch": 0.14675533134602156,
      "grad_norm": 0.1497193884747371,
      "learning_rate": 0.0009646966091311,
      "loss": 3.6658,
      "step": 640
    },
    {
      "epoch": 0.14698463655124971,
      "grad_norm": 0.1554947394417242,
      "learning_rate": 0.0009645594202357438,
      "loss": 3.5456,
      "step": 641
    },
    {
      "epoch": 0.14721394175647787,
      "grad_norm": 0.13658592049083682,
      "learning_rate": 0.0009644219750925191,
      "loss": 3.5893,
      "step": 642
    },
    {
      "epoch": 0.14744324696170602,
      "grad_norm": 0.14259821144542686,
      "learning_rate": 0.0009642842737772397,
      "loss": 3.5938,
      "step": 643
    },
    {
      "epoch": 0.1476725521669342,
      "grad_norm": 0.14863692004420914,
      "learning_rate": 0.0009641463163658606,
      "loss": 3.6511,
      "step": 644
    },
    {
      "epoch": 0.14790185737216235,
      "grad_norm": 0.14866428208804616,
      "learning_rate": 0.0009640081029344782,
      "loss": 3.6735,
      "step": 645
    },
    {
      "epoch": 0.1481311625773905,
      "grad_norm": 0.12494474836933409,
      "learning_rate": 0.0009638696335593303,
      "loss": 3.7087,
      "step": 646
    },
    {
      "epoch": 0.14836046778261866,
      "grad_norm": 0.14464642292481553,
      "learning_rate": 0.0009637309083167956,
      "loss": 3.587,
      "step": 647
    },
    {
      "epoch": 0.1485897729878468,
      "grad_norm": 0.14752098984454348,
      "learning_rate": 0.0009635919272833937,
      "loss": 3.4857,
      "step": 648
    },
    {
      "epoch": 0.148819078193075,
      "grad_norm": 0.15158505470590647,
      "learning_rate": 0.0009634526905357859,
      "loss": 3.6526,
      "step": 649
    },
    {
      "epoch": 0.14904838339830315,
      "grad_norm": 0.148367377776824,
      "learning_rate": 0.0009633131981507744,
      "loss": 3.7348,
      "step": 650
    },
    {
      "epoch": 0.1492776886035313,
      "grad_norm": 0.14461226514865447,
      "learning_rate": 0.000963173450205302,
      "loss": 3.5729,
      "step": 651
    },
    {
      "epoch": 0.14950699380875945,
      "grad_norm": 0.13577058245257137,
      "learning_rate": 0.0009630334467764528,
      "loss": 3.6979,
      "step": 652
    },
    {
      "epoch": 0.1497362990139876,
      "grad_norm": 0.1503050915963726,
      "learning_rate": 0.0009628931879414517,
      "loss": 3.647,
      "step": 653
    },
    {
      "epoch": 0.1499656042192158,
      "grad_norm": 0.16206316971529336,
      "learning_rate": 0.0009627526737776646,
      "loss": 3.4787,
      "step": 654
    },
    {
      "epoch": 0.15019490942444394,
      "grad_norm": 0.1426258752311336,
      "learning_rate": 0.0009626119043625983,
      "loss": 3.6438,
      "step": 655
    },
    {
      "epoch": 0.1504242146296721,
      "grad_norm": 0.13395236250667997,
      "learning_rate": 0.0009624708797739001,
      "loss": 3.7633,
      "step": 656
    },
    {
      "epoch": 0.15065351983490025,
      "grad_norm": 0.1384409642324847,
      "learning_rate": 0.0009623296000893582,
      "loss": 3.6402,
      "step": 657
    },
    {
      "epoch": 0.1508828250401284,
      "grad_norm": 0.14034438447934292,
      "learning_rate": 0.0009621880653869016,
      "loss": 3.6407,
      "step": 658
    },
    {
      "epoch": 0.15111213024535658,
      "grad_norm": 0.15334531219516467,
      "learning_rate": 0.0009620462757446,
      "loss": 3.6253,
      "step": 659
    },
    {
      "epoch": 0.15134143545058473,
      "grad_norm": 0.16451490937768276,
      "learning_rate": 0.0009619042312406634,
      "loss": 3.6596,
      "step": 660
    },
    {
      "epoch": 0.15157074065581289,
      "grad_norm": 0.1577020965309097,
      "learning_rate": 0.0009617619319534428,
      "loss": 3.5608,
      "step": 661
    },
    {
      "epoch": 0.15180004586104104,
      "grad_norm": 0.13787567506160134,
      "learning_rate": 0.0009616193779614293,
      "loss": 3.6935,
      "step": 662
    },
    {
      "epoch": 0.1520293510662692,
      "grad_norm": 0.1410140162305034,
      "learning_rate": 0.000961476569343255,
      "loss": 3.5173,
      "step": 663
    },
    {
      "epoch": 0.15225865627149737,
      "grad_norm": 0.14273898589418813,
      "learning_rate": 0.000961333506177692,
      "loss": 3.7372,
      "step": 664
    },
    {
      "epoch": 0.15248796147672553,
      "grad_norm": 0.14965667146338826,
      "learning_rate": 0.0009611901885436529,
      "loss": 3.5875,
      "step": 665
    },
    {
      "epoch": 0.15271726668195368,
      "grad_norm": 0.15665788398567293,
      "learning_rate": 0.0009610466165201911,
      "loss": 3.7064,
      "step": 666
    },
    {
      "epoch": 0.15294657188718183,
      "grad_norm": 0.13784875524887205,
      "learning_rate": 0.0009609027901864996,
      "loss": 3.6827,
      "step": 667
    },
    {
      "epoch": 0.15317587709240998,
      "grad_norm": 0.14548528825206947,
      "learning_rate": 0.0009607587096219124,
      "loss": 3.7409,
      "step": 668
    },
    {
      "epoch": 0.15340518229763817,
      "grad_norm": 0.13387383502197506,
      "learning_rate": 0.0009606143749059029,
      "loss": 3.6075,
      "step": 669
    },
    {
      "epoch": 0.15363448750286632,
      "grad_norm": 0.137326980319962,
      "learning_rate": 0.0009604697861180856,
      "loss": 3.6668,
      "step": 670
    },
    {
      "epoch": 0.15386379270809447,
      "grad_norm": 0.1547948016499996,
      "learning_rate": 0.0009603249433382144,
      "loss": 3.5944,
      "step": 671
    },
    {
      "epoch": 0.15409309791332262,
      "grad_norm": 0.1296745928722884,
      "learning_rate": 0.0009601798466461838,
      "loss": 3.6328,
      "step": 672
    },
    {
      "epoch": 0.1543224031185508,
      "grad_norm": 0.14192985826677185,
      "learning_rate": 0.0009600344961220282,
      "loss": 3.6173,
      "step": 673
    },
    {
      "epoch": 0.15455170832377896,
      "grad_norm": 0.135957517639193,
      "learning_rate": 0.0009598888918459219,
      "loss": 3.5339,
      "step": 674
    },
    {
      "epoch": 0.1547810135290071,
      "grad_norm": 0.13732089494577812,
      "learning_rate": 0.0009597430338981791,
      "loss": 3.6159,
      "step": 675
    },
    {
      "epoch": 0.15501031873423526,
      "grad_norm": 0.16637277893051244,
      "learning_rate": 0.0009595969223592542,
      "loss": 3.6114,
      "step": 676
    },
    {
      "epoch": 0.15523962393946342,
      "grad_norm": 0.15264521071740098,
      "learning_rate": 0.0009594505573097414,
      "loss": 3.6289,
      "step": 677
    },
    {
      "epoch": 0.1554689291446916,
      "grad_norm": 0.14398173777671397,
      "learning_rate": 0.0009593039388303747,
      "loss": 3.4189,
      "step": 678
    },
    {
      "epoch": 0.15569823434991975,
      "grad_norm": 0.146965161097489,
      "learning_rate": 0.0009591570670020277,
      "loss": 3.6246,
      "step": 679
    },
    {
      "epoch": 0.1559275395551479,
      "grad_norm": 0.14514371718870933,
      "learning_rate": 0.0009590099419057142,
      "loss": 3.6307,
      "step": 680
    },
    {
      "epoch": 0.15615684476037606,
      "grad_norm": 0.13606933438504656,
      "learning_rate": 0.0009588625636225871,
      "loss": 3.6609,
      "step": 681
    },
    {
      "epoch": 0.1563861499656042,
      "grad_norm": 0.15522376375538222,
      "learning_rate": 0.0009587149322339394,
      "loss": 3.5771,
      "step": 682
    },
    {
      "epoch": 0.1566154551708324,
      "grad_norm": 0.14251005348772794,
      "learning_rate": 0.0009585670478212036,
      "loss": 3.6345,
      "step": 683
    },
    {
      "epoch": 0.15684476037606054,
      "grad_norm": 0.14539481575935878,
      "learning_rate": 0.0009584189104659517,
      "loss": 3.5223,
      "step": 684
    },
    {
      "epoch": 0.1570740655812887,
      "grad_norm": 0.13157286177291141,
      "learning_rate": 0.0009582705202498956,
      "loss": 3.5961,
      "step": 685
    },
    {
      "epoch": 0.15730337078651685,
      "grad_norm": 0.1471492562778442,
      "learning_rate": 0.0009581218772548859,
      "loss": 3.6528,
      "step": 686
    },
    {
      "epoch": 0.157532675991745,
      "grad_norm": 0.15882779804169567,
      "learning_rate": 0.0009579729815629133,
      "loss": 3.4013,
      "step": 687
    },
    {
      "epoch": 0.15776198119697318,
      "grad_norm": 0.1473567618535874,
      "learning_rate": 0.0009578238332561075,
      "loss": 3.5506,
      "step": 688
    },
    {
      "epoch": 0.15799128640220134,
      "grad_norm": 0.13958600225634335,
      "learning_rate": 0.0009576744324167379,
      "loss": 3.4507,
      "step": 689
    },
    {
      "epoch": 0.1582205916074295,
      "grad_norm": 0.15920290933342907,
      "learning_rate": 0.0009575247791272129,
      "loss": 3.703,
      "step": 690
    },
    {
      "epoch": 0.15844989681265764,
      "grad_norm": 0.13639627377874278,
      "learning_rate": 0.0009573748734700804,
      "loss": 3.5796,
      "step": 691
    },
    {
      "epoch": 0.1586792020178858,
      "grad_norm": 0.13312117005152238,
      "learning_rate": 0.0009572247155280271,
      "loss": 3.6635,
      "step": 692
    },
    {
      "epoch": 0.15890850722311398,
      "grad_norm": 0.14410170443578874,
      "learning_rate": 0.0009570743053838796,
      "loss": 3.7222,
      "step": 693
    },
    {
      "epoch": 0.15913781242834213,
      "grad_norm": 0.1398228964726699,
      "learning_rate": 0.0009569236431206025,
      "loss": 3.657,
      "step": 694
    },
    {
      "epoch": 0.15936711763357028,
      "grad_norm": 0.1330593810073827,
      "learning_rate": 0.0009567727288213005,
      "loss": 3.577,
      "step": 695
    },
    {
      "epoch": 0.15959642283879844,
      "grad_norm": 0.17190175310828537,
      "learning_rate": 0.0009566215625692167,
      "loss": 3.5533,
      "step": 696
    },
    {
      "epoch": 0.1598257280440266,
      "grad_norm": 0.14479484034309573,
      "learning_rate": 0.0009564701444477337,
      "loss": 3.6268,
      "step": 697
    },
    {
      "epoch": 0.16005503324925477,
      "grad_norm": 0.16292745270340617,
      "learning_rate": 0.0009563184745403723,
      "loss": 3.5357,
      "step": 698
    },
    {
      "epoch": 0.16028433845448292,
      "grad_norm": 0.1586788443382945,
      "learning_rate": 0.000956166552930793,
      "loss": 3.7329,
      "step": 699
    },
    {
      "epoch": 0.16051364365971107,
      "grad_norm": 0.1464372478105035,
      "learning_rate": 0.0009560143797027946,
      "loss": 3.5807,
      "step": 700
    },
    {
      "epoch": 0.16074294886493923,
      "grad_norm": 0.17558769439436175,
      "learning_rate": 0.0009558619549403147,
      "loss": 3.5113,
      "step": 701
    },
    {
      "epoch": 0.16097225407016738,
      "grad_norm": 0.154839763375458,
      "learning_rate": 0.0009557092787274299,
      "loss": 3.5552,
      "step": 702
    },
    {
      "epoch": 0.16120155927539556,
      "grad_norm": 0.14165972075355593,
      "learning_rate": 0.0009555563511483555,
      "loss": 3.6246,
      "step": 703
    },
    {
      "epoch": 0.16143086448062371,
      "grad_norm": 0.14900522592078813,
      "learning_rate": 0.0009554031722874451,
      "loss": 3.5722,
      "step": 704
    },
    {
      "epoch": 0.16166016968585187,
      "grad_norm": 0.19421875875089728,
      "learning_rate": 0.0009552497422291912,
      "loss": 3.5784,
      "step": 705
    },
    {
      "epoch": 0.16188947489108002,
      "grad_norm": 0.13781459451433212,
      "learning_rate": 0.000955096061058225,
      "loss": 3.5942,
      "step": 706
    },
    {
      "epoch": 0.16211878009630817,
      "grad_norm": 0.1366816928258114,
      "learning_rate": 0.0009549421288593157,
      "loss": 3.5746,
      "step": 707
    },
    {
      "epoch": 0.16234808530153635,
      "grad_norm": 0.18556665276511572,
      "learning_rate": 0.0009547879457173714,
      "loss": 3.7659,
      "step": 708
    },
    {
      "epoch": 0.1625773905067645,
      "grad_norm": 0.14197409433328345,
      "learning_rate": 0.0009546335117174385,
      "loss": 3.5266,
      "step": 709
    },
    {
      "epoch": 0.16280669571199266,
      "grad_norm": 0.14869473158514948,
      "learning_rate": 0.0009544788269447018,
      "loss": 3.6345,
      "step": 710
    },
    {
      "epoch": 0.1630360009172208,
      "grad_norm": 0.16469830344213177,
      "learning_rate": 0.0009543238914844843,
      "loss": 3.6937,
      "step": 711
    },
    {
      "epoch": 0.16326530612244897,
      "grad_norm": 0.1488509922374584,
      "learning_rate": 0.0009541687054222474,
      "loss": 3.5995,
      "step": 712
    },
    {
      "epoch": 0.16349461132767715,
      "grad_norm": 0.1548302923760706,
      "learning_rate": 0.0009540132688435907,
      "loss": 3.4571,
      "step": 713
    },
    {
      "epoch": 0.1637239165329053,
      "grad_norm": 0.18246580689186986,
      "learning_rate": 0.0009538575818342521,
      "loss": 3.6378,
      "step": 714
    },
    {
      "epoch": 0.16395322173813345,
      "grad_norm": 0.14277051611208777,
      "learning_rate": 0.0009537016444801074,
      "loss": 3.5917,
      "step": 715
    },
    {
      "epoch": 0.1641825269433616,
      "grad_norm": 0.14282843773116427,
      "learning_rate": 0.0009535454568671704,
      "loss": 3.7466,
      "step": 716
    },
    {
      "epoch": 0.16441183214858976,
      "grad_norm": 0.16535589544775675,
      "learning_rate": 0.0009533890190815935,
      "loss": 3.5066,
      "step": 717
    },
    {
      "epoch": 0.16464113735381794,
      "grad_norm": 0.13980454164255762,
      "learning_rate": 0.0009532323312096665,
      "loss": 3.6133,
      "step": 718
    },
    {
      "epoch": 0.1648704425590461,
      "grad_norm": 0.15367812531499384,
      "learning_rate": 0.0009530753933378173,
      "loss": 3.6487,
      "step": 719
    },
    {
      "epoch": 0.16509974776427425,
      "grad_norm": 0.1358130049316476,
      "learning_rate": 0.0009529182055526121,
      "loss": 3.5408,
      "step": 720
    },
    {
      "epoch": 0.1653290529695024,
      "grad_norm": 0.15395906924896202,
      "learning_rate": 0.0009527607679407545,
      "loss": 3.6225,
      "step": 721
    },
    {
      "epoch": 0.16555835817473058,
      "grad_norm": 0.12622515948092464,
      "learning_rate": 0.000952603080589086,
      "loss": 3.5707,
      "step": 722
    },
    {
      "epoch": 0.16578766337995873,
      "grad_norm": 0.1383174124145602,
      "learning_rate": 0.0009524451435845857,
      "loss": 3.5296,
      "step": 723
    },
    {
      "epoch": 0.16601696858518689,
      "grad_norm": 0.1696709433982929,
      "learning_rate": 0.000952286957014371,
      "loss": 3.4846,
      "step": 724
    },
    {
      "epoch": 0.16624627379041504,
      "grad_norm": 0.1489880213817021,
      "learning_rate": 0.0009521285209656963,
      "loss": 3.4864,
      "step": 725
    },
    {
      "epoch": 0.1664755789956432,
      "grad_norm": 0.14462365673623023,
      "learning_rate": 0.0009519698355259538,
      "loss": 3.6949,
      "step": 726
    },
    {
      "epoch": 0.16670488420087137,
      "grad_norm": 0.1245286663143048,
      "learning_rate": 0.0009518109007826734,
      "loss": 3.5057,
      "step": 727
    },
    {
      "epoch": 0.16693418940609953,
      "grad_norm": 0.16288252015776652,
      "learning_rate": 0.0009516517168235226,
      "loss": 3.6743,
      "step": 728
    },
    {
      "epoch": 0.16716349461132768,
      "grad_norm": 0.16423673263027685,
      "learning_rate": 0.0009514922837363059,
      "loss": 3.6197,
      "step": 729
    },
    {
      "epoch": 0.16739279981655583,
      "grad_norm": 0.16964877410834078,
      "learning_rate": 0.0009513326016089658,
      "loss": 3.6186,
      "step": 730
    },
    {
      "epoch": 0.16762210502178398,
      "grad_norm": 0.14159912636173458,
      "learning_rate": 0.0009511726705295817,
      "loss": 3.6814,
      "step": 731
    },
    {
      "epoch": 0.16785141022701217,
      "grad_norm": 0.16409878574240164,
      "learning_rate": 0.0009510124905863709,
      "loss": 3.5228,
      "step": 732
    },
    {
      "epoch": 0.16808071543224032,
      "grad_norm": 0.16474365060374338,
      "learning_rate": 0.000950852061867687,
      "loss": 3.6059,
      "step": 733
    },
    {
      "epoch": 0.16831002063746847,
      "grad_norm": 0.13632388805471063,
      "learning_rate": 0.0009506913844620217,
      "loss": 3.6652,
      "step": 734
    },
    {
      "epoch": 0.16853932584269662,
      "grad_norm": 0.13750165177772164,
      "learning_rate": 0.0009505304584580038,
      "loss": 3.4359,
      "step": 735
    },
    {
      "epoch": 0.16876863104792478,
      "grad_norm": 0.14457272286280537,
      "learning_rate": 0.0009503692839443987,
      "loss": 3.635,
      "step": 736
    },
    {
      "epoch": 0.16899793625315296,
      "grad_norm": 0.1377521029338072,
      "learning_rate": 0.0009502078610101092,
      "loss": 3.6663,
      "step": 737
    },
    {
      "epoch": 0.1692272414583811,
      "grad_norm": 0.13193445771828202,
      "learning_rate": 0.0009500461897441752,
      "loss": 3.614,
      "step": 738
    },
    {
      "epoch": 0.16945654666360926,
      "grad_norm": 0.1591166508000921,
      "learning_rate": 0.0009498842702357736,
      "loss": 3.7309,
      "step": 739
    },
    {
      "epoch": 0.16968585186883742,
      "grad_norm": 0.13833143372997211,
      "learning_rate": 0.0009497221025742177,
      "loss": 3.5262,
      "step": 740
    },
    {
      "epoch": 0.16991515707406557,
      "grad_norm": 0.1265521989852963,
      "learning_rate": 0.0009495596868489587,
      "loss": 3.6669,
      "step": 741
    },
    {
      "epoch": 0.17014446227929375,
      "grad_norm": 0.13095954282273775,
      "learning_rate": 0.0009493970231495835,
      "loss": 3.4893,
      "step": 742
    },
    {
      "epoch": 0.1703737674845219,
      "grad_norm": 0.13257334149127772,
      "learning_rate": 0.0009492341115658167,
      "loss": 3.5788,
      "step": 743
    },
    {
      "epoch": 0.17060307268975006,
      "grad_norm": 0.1532781975516019,
      "learning_rate": 0.0009490709521875189,
      "loss": 3.5288,
      "step": 744
    },
    {
      "epoch": 0.1708323778949782,
      "grad_norm": 0.1460846163437705,
      "learning_rate": 0.0009489075451046879,
      "loss": 3.5653,
      "step": 745
    },
    {
      "epoch": 0.17106168310020636,
      "grad_norm": 0.16853885262608387,
      "learning_rate": 0.000948743890407458,
      "loss": 3.7096,
      "step": 746
    },
    {
      "epoch": 0.17129098830543454,
      "grad_norm": 0.1417648243039852,
      "learning_rate": 0.0009485799881861,
      "loss": 3.664,
      "step": 747
    },
    {
      "epoch": 0.1715202935106627,
      "grad_norm": 0.16394412504548855,
      "learning_rate": 0.0009484158385310212,
      "loss": 3.4651,
      "step": 748
    },
    {
      "epoch": 0.17174959871589085,
      "grad_norm": 0.15068417243191262,
      "learning_rate": 0.0009482514415327654,
      "loss": 3.6379,
      "step": 749
    },
    {
      "epoch": 0.171978903921119,
      "grad_norm": 0.14533502281029645,
      "learning_rate": 0.0009480867972820131,
      "loss": 3.56,
      "step": 750
    },
    {
      "epoch": 0.17220820912634716,
      "grad_norm": 0.16476490150295564,
      "learning_rate": 0.000947921905869581,
      "loss": 3.4676,
      "step": 751
    },
    {
      "epoch": 0.17243751433157534,
      "grad_norm": 0.15347670603288732,
      "learning_rate": 0.0009477567673864216,
      "loss": 3.7437,
      "step": 752
    },
    {
      "epoch": 0.1726668195368035,
      "grad_norm": 0.13388823672308198,
      "learning_rate": 0.0009475913819236248,
      "loss": 3.5106,
      "step": 753
    },
    {
      "epoch": 0.17289612474203164,
      "grad_norm": 0.1403800043693641,
      "learning_rate": 0.0009474257495724156,
      "loss": 3.5813,
      "step": 754
    },
    {
      "epoch": 0.1731254299472598,
      "grad_norm": 0.16373616311649034,
      "learning_rate": 0.0009472598704241561,
      "loss": 3.7027,
      "step": 755
    },
    {
      "epoch": 0.17335473515248795,
      "grad_norm": 0.1674171059219014,
      "learning_rate": 0.000947093744570344,
      "loss": 3.6552,
      "step": 756
    },
    {
      "epoch": 0.17358404035771613,
      "grad_norm": 0.13385459319658366,
      "learning_rate": 0.0009469273721026131,
      "loss": 3.6578,
      "step": 757
    },
    {
      "epoch": 0.17381334556294428,
      "grad_norm": 0.13274658980498208,
      "learning_rate": 0.0009467607531127335,
      "loss": 3.532,
      "step": 758
    },
    {
      "epoch": 0.17404265076817244,
      "grad_norm": 0.16422862986657022,
      "learning_rate": 0.0009465938876926111,
      "loss": 3.5917,
      "step": 759
    },
    {
      "epoch": 0.1742719559734006,
      "grad_norm": 0.14591555103437376,
      "learning_rate": 0.0009464267759342875,
      "loss": 3.5198,
      "step": 760
    },
    {
      "epoch": 0.17450126117862874,
      "grad_norm": 0.1493600158937577,
      "learning_rate": 0.0009462594179299406,
      "loss": 3.6464,
      "step": 761
    },
    {
      "epoch": 0.17473056638385692,
      "grad_norm": 0.1447324547955115,
      "learning_rate": 0.0009460918137718841,
      "loss": 3.4764,
      "step": 762
    },
    {
      "epoch": 0.17495987158908508,
      "grad_norm": 0.15701350373460074,
      "learning_rate": 0.0009459239635525672,
      "loss": 3.6951,
      "step": 763
    },
    {
      "epoch": 0.17518917679431323,
      "grad_norm": 0.15205970213437014,
      "learning_rate": 0.000945755867364575,
      "loss": 3.4948,
      "step": 764
    },
    {
      "epoch": 0.17541848199954138,
      "grad_norm": 0.1577594768414039,
      "learning_rate": 0.0009455875253006281,
      "loss": 3.6915,
      "step": 765
    },
    {
      "epoch": 0.17564778720476956,
      "grad_norm": 0.17532295025718397,
      "learning_rate": 0.0009454189374535829,
      "loss": 3.7164,
      "step": 766
    },
    {
      "epoch": 0.17587709240999772,
      "grad_norm": 0.14725194440000525,
      "learning_rate": 0.0009452501039164315,
      "loss": 3.5877,
      "step": 767
    },
    {
      "epoch": 0.17610639761522587,
      "grad_norm": 0.16854392288298717,
      "learning_rate": 0.0009450810247823012,
      "loss": 3.712,
      "step": 768
    },
    {
      "epoch": 0.17633570282045402,
      "grad_norm": 0.16297892671695452,
      "learning_rate": 0.0009449117001444549,
      "loss": 3.6287,
      "step": 769
    },
    {
      "epoch": 0.17656500802568217,
      "grad_norm": 0.14688810877086383,
      "learning_rate": 0.0009447421300962911,
      "loss": 3.7086,
      "step": 770
    },
    {
      "epoch": 0.17679431323091035,
      "grad_norm": 0.1660537292367693,
      "learning_rate": 0.0009445723147313433,
      "loss": 3.6016,
      "step": 771
    },
    {
      "epoch": 0.1770236184361385,
      "grad_norm": 0.13812498177810173,
      "learning_rate": 0.0009444022541432806,
      "loss": 3.5592,
      "step": 772
    },
    {
      "epoch": 0.17725292364136666,
      "grad_norm": 0.14551630313656833,
      "learning_rate": 0.0009442319484259074,
      "loss": 3.6541,
      "step": 773
    },
    {
      "epoch": 0.17748222884659481,
      "grad_norm": 0.1283911600420412,
      "learning_rate": 0.000944061397673163,
      "loss": 3.4805,
      "step": 774
    },
    {
      "epoch": 0.17771153405182297,
      "grad_norm": 0.13536174574491594,
      "learning_rate": 0.0009438906019791222,
      "loss": 3.5883,
      "step": 775
    },
    {
      "epoch": 0.17794083925705115,
      "grad_norm": 0.12301821106894704,
      "learning_rate": 0.0009437195614379946,
      "loss": 3.6318,
      "step": 776
    },
    {
      "epoch": 0.1781701444622793,
      "grad_norm": 0.13690421929263674,
      "learning_rate": 0.0009435482761441251,
      "loss": 3.7155,
      "step": 777
    },
    {
      "epoch": 0.17839944966750745,
      "grad_norm": 0.12029561027190741,
      "learning_rate": 0.0009433767461919937,
      "loss": 3.5371,
      "step": 778
    },
    {
      "epoch": 0.1786287548727356,
      "grad_norm": 0.1622339213213419,
      "learning_rate": 0.000943204971676215,
      "loss": 3.5156,
      "step": 779
    },
    {
      "epoch": 0.17885806007796376,
      "grad_norm": 0.16176350513543955,
      "learning_rate": 0.0009430329526915386,
      "loss": 3.5858,
      "step": 780
    },
    {
      "epoch": 0.17908736528319194,
      "grad_norm": 0.14249581029999334,
      "learning_rate": 0.0009428606893328493,
      "loss": 3.5365,
      "step": 781
    },
    {
      "epoch": 0.1793166704884201,
      "grad_norm": 0.12733499278668842,
      "learning_rate": 0.000942688181695166,
      "loss": 3.5278,
      "step": 782
    },
    {
      "epoch": 0.17954597569364825,
      "grad_norm": 0.15869051224025948,
      "learning_rate": 0.0009425154298736432,
      "loss": 3.6492,
      "step": 783
    },
    {
      "epoch": 0.1797752808988764,
      "grad_norm": 0.15381859695981565,
      "learning_rate": 0.0009423424339635694,
      "loss": 3.6297,
      "step": 784
    },
    {
      "epoch": 0.18000458610410455,
      "grad_norm": 0.14501302978695882,
      "learning_rate": 0.0009421691940603678,
      "loss": 3.627,
      "step": 785
    },
    {
      "epoch": 0.18023389130933273,
      "grad_norm": 0.1387101240826076,
      "learning_rate": 0.0009419957102595968,
      "loss": 3.6862,
      "step": 786
    },
    {
      "epoch": 0.1804631965145609,
      "grad_norm": 0.1526033554507801,
      "learning_rate": 0.0009418219826569488,
      "loss": 3.4849,
      "step": 787
    },
    {
      "epoch": 0.18069250171978904,
      "grad_norm": 0.14588924960965083,
      "learning_rate": 0.0009416480113482504,
      "loss": 3.5849,
      "step": 788
    },
    {
      "epoch": 0.1809218069250172,
      "grad_norm": 0.1296522642368441,
      "learning_rate": 0.0009414737964294635,
      "loss": 3.5519,
      "step": 789
    },
    {
      "epoch": 0.18115111213024535,
      "grad_norm": 0.13814591190416745,
      "learning_rate": 0.0009412993379966836,
      "loss": 3.801,
      "step": 790
    },
    {
      "epoch": 0.18138041733547353,
      "grad_norm": 0.1521517242982108,
      "learning_rate": 0.000941124636146141,
      "loss": 3.6369,
      "step": 791
    },
    {
      "epoch": 0.18160972254070168,
      "grad_norm": 0.14212068760540478,
      "learning_rate": 0.0009409496909741998,
      "loss": 3.6927,
      "step": 792
    },
    {
      "epoch": 0.18183902774592983,
      "grad_norm": 0.13334612783002225,
      "learning_rate": 0.0009407745025773589,
      "loss": 3.4994,
      "step": 793
    },
    {
      "epoch": 0.18206833295115799,
      "grad_norm": 0.14920659390168292,
      "learning_rate": 0.0009405990710522508,
      "loss": 3.6287,
      "step": 794
    },
    {
      "epoch": 0.18229763815638614,
      "grad_norm": 0.13599495418657576,
      "learning_rate": 0.0009404233964956423,
      "loss": 3.6043,
      "step": 795
    },
    {
      "epoch": 0.18252694336161432,
      "grad_norm": 0.13612172162365055,
      "learning_rate": 0.0009402474790044346,
      "loss": 3.589,
      "step": 796
    },
    {
      "epoch": 0.18275624856684247,
      "grad_norm": 0.1389887637423891,
      "learning_rate": 0.0009400713186756625,
      "loss": 3.6012,
      "step": 797
    },
    {
      "epoch": 0.18298555377207062,
      "grad_norm": 0.14670448054743193,
      "learning_rate": 0.0009398949156064948,
      "loss": 3.6766,
      "step": 798
    },
    {
      "epoch": 0.18321485897729878,
      "grad_norm": 0.17296859339470985,
      "learning_rate": 0.0009397182698942342,
      "loss": 3.5328,
      "step": 799
    },
    {
      "epoch": 0.18344416418252693,
      "grad_norm": 0.15808953669788356,
      "learning_rate": 0.0009395413816363175,
      "loss": 3.7289,
      "step": 800
    },
    {
      "epoch": 0.1836734693877551,
      "grad_norm": 0.13351724504290308,
      "learning_rate": 0.0009393642509303149,
      "loss": 3.7003,
      "step": 801
    },
    {
      "epoch": 0.18390277459298326,
      "grad_norm": 0.18905812697134175,
      "learning_rate": 0.0009391868778739305,
      "loss": 3.7478,
      "step": 802
    },
    {
      "epoch": 0.18413207979821142,
      "grad_norm": 0.17555885435370824,
      "learning_rate": 0.0009390092625650023,
      "loss": 3.5644,
      "step": 803
    },
    {
      "epoch": 0.18436138500343957,
      "grad_norm": 0.1486143158699133,
      "learning_rate": 0.0009388314051015015,
      "loss": 3.555,
      "step": 804
    },
    {
      "epoch": 0.18459069020866772,
      "grad_norm": 0.1561603918974189,
      "learning_rate": 0.0009386533055815332,
      "loss": 3.4917,
      "step": 805
    },
    {
      "epoch": 0.1848199954138959,
      "grad_norm": 0.14612095066789596,
      "learning_rate": 0.0009384749641033358,
      "loss": 3.4329,
      "step": 806
    },
    {
      "epoch": 0.18504930061912406,
      "grad_norm": 0.15797796263666025,
      "learning_rate": 0.0009382963807652813,
      "loss": 3.5086,
      "step": 807
    },
    {
      "epoch": 0.1852786058243522,
      "grad_norm": 0.1455057941394473,
      "learning_rate": 0.0009381175556658752,
      "loss": 3.5763,
      "step": 808
    },
    {
      "epoch": 0.18550791102958036,
      "grad_norm": 0.1739180077695725,
      "learning_rate": 0.000937938488903756,
      "loss": 3.5731,
      "step": 809
    },
    {
      "epoch": 0.18573721623480852,
      "grad_norm": 0.17669439737359816,
      "learning_rate": 0.0009377591805776961,
      "loss": 3.7388,
      "step": 810
    },
    {
      "epoch": 0.1859665214400367,
      "grad_norm": 0.1735328705424633,
      "learning_rate": 0.0009375796307866003,
      "loss": 3.6194,
      "step": 811
    },
    {
      "epoch": 0.18619582664526485,
      "grad_norm": 0.12979279722276274,
      "learning_rate": 0.0009373998396295074,
      "loss": 3.5325,
      "step": 812
    },
    {
      "epoch": 0.186425131850493,
      "grad_norm": 0.17406704705143838,
      "learning_rate": 0.0009372198072055888,
      "loss": 3.6428,
      "step": 813
    },
    {
      "epoch": 0.18665443705572116,
      "grad_norm": 0.1284285513807575,
      "learning_rate": 0.0009370395336141495,
      "loss": 3.446,
      "step": 814
    },
    {
      "epoch": 0.18688374226094934,
      "grad_norm": 0.149907957920771,
      "learning_rate": 0.0009368590189546268,
      "loss": 3.7637,
      "step": 815
    },
    {
      "epoch": 0.1871130474661775,
      "grad_norm": 0.13611744559698918,
      "learning_rate": 0.0009366782633265916,
      "loss": 3.5739,
      "step": 816
    },
    {
      "epoch": 0.18734235267140564,
      "grad_norm": 0.14338985688924372,
      "learning_rate": 0.0009364972668297474,
      "loss": 3.5691,
      "step": 817
    },
    {
      "epoch": 0.1875716578766338,
      "grad_norm": 0.11963420952749053,
      "learning_rate": 0.0009363160295639308,
      "loss": 3.6107,
      "step": 818
    },
    {
      "epoch": 0.18780096308186195,
      "grad_norm": 0.14413159105307466,
      "learning_rate": 0.0009361345516291111,
      "loss": 3.6066,
      "step": 819
    },
    {
      "epoch": 0.18803026828709013,
      "grad_norm": 0.13604555780482613,
      "learning_rate": 0.0009359528331253901,
      "loss": 3.494,
      "step": 820
    },
    {
      "epoch": 0.18825957349231828,
      "grad_norm": 0.14994841621782892,
      "learning_rate": 0.0009357708741530024,
      "loss": 3.7061,
      "step": 821
    },
    {
      "epoch": 0.18848887869754644,
      "grad_norm": 0.1524260858891333,
      "learning_rate": 0.0009355886748123158,
      "loss": 3.5688,
      "step": 822
    },
    {
      "epoch": 0.1887181839027746,
      "grad_norm": 0.14622099090773638,
      "learning_rate": 0.00093540623520383,
      "loss": 3.577,
      "step": 823
    },
    {
      "epoch": 0.18894748910800274,
      "grad_norm": 0.1344305737073058,
      "learning_rate": 0.0009352235554281774,
      "loss": 3.5264,
      "step": 824
    },
    {
      "epoch": 0.18917679431323092,
      "grad_norm": 0.14136385747224695,
      "learning_rate": 0.000935040635586123,
      "loss": 3.4928,
      "step": 825
    },
    {
      "epoch": 0.18940609951845908,
      "grad_norm": 0.14308201674801124,
      "learning_rate": 0.0009348574757785642,
      "loss": 3.6307,
      "step": 826
    },
    {
      "epoch": 0.18963540472368723,
      "grad_norm": 0.14076830800771392,
      "learning_rate": 0.0009346740761065305,
      "loss": 3.5631,
      "step": 827
    },
    {
      "epoch": 0.18986470992891538,
      "grad_norm": 0.13656021258579362,
      "learning_rate": 0.0009344904366711844,
      "loss": 3.6137,
      "step": 828
    },
    {
      "epoch": 0.19009401513414353,
      "grad_norm": 0.14879763172611743,
      "learning_rate": 0.0009343065575738197,
      "loss": 3.517,
      "step": 829
    },
    {
      "epoch": 0.19032332033937172,
      "grad_norm": 0.15409550375714914,
      "learning_rate": 0.0009341224389158631,
      "loss": 3.6226,
      "step": 830
    },
    {
      "epoch": 0.19055262554459987,
      "grad_norm": 0.1407197959202914,
      "learning_rate": 0.0009339380807988733,
      "loss": 3.5084,
      "step": 831
    },
    {
      "epoch": 0.19078193074982802,
      "grad_norm": 0.13055623628065852,
      "learning_rate": 0.0009337534833245409,
      "loss": 3.5887,
      "step": 832
    },
    {
      "epoch": 0.19101123595505617,
      "grad_norm": 0.14818999828668616,
      "learning_rate": 0.0009335686465946887,
      "loss": 3.6675,
      "step": 833
    },
    {
      "epoch": 0.19124054116028433,
      "grad_norm": 0.16974449474977588,
      "learning_rate": 0.0009333835707112713,
      "loss": 3.7368,
      "step": 834
    },
    {
      "epoch": 0.1914698463655125,
      "grad_norm": 0.1651060072318715,
      "learning_rate": 0.0009331982557763754,
      "loss": 3.4995,
      "step": 835
    },
    {
      "epoch": 0.19169915157074066,
      "grad_norm": 0.14916821448915885,
      "learning_rate": 0.0009330127018922195,
      "loss": 3.7141,
      "step": 836
    },
    {
      "epoch": 0.19192845677596881,
      "grad_norm": 0.15445457620483416,
      "learning_rate": 0.0009328269091611537,
      "loss": 3.5788,
      "step": 837
    },
    {
      "epoch": 0.19215776198119697,
      "grad_norm": 0.14804552943402263,
      "learning_rate": 0.00093264087768566,
      "loss": 3.7703,
      "step": 838
    },
    {
      "epoch": 0.19238706718642512,
      "grad_norm": 0.13473205260732274,
      "learning_rate": 0.0009324546075683524,
      "loss": 3.6012,
      "step": 839
    },
    {
      "epoch": 0.1926163723916533,
      "grad_norm": 0.14902816606640282,
      "learning_rate": 0.0009322680989119763,
      "loss": 3.611,
      "step": 840
    },
    {
      "epoch": 0.19284567759688145,
      "grad_norm": 0.13873970266792057,
      "learning_rate": 0.0009320813518194083,
      "loss": 3.5295,
      "step": 841
    },
    {
      "epoch": 0.1930749828021096,
      "grad_norm": 0.14899631503435137,
      "learning_rate": 0.000931894366393657,
      "loss": 3.5417,
      "step": 842
    },
    {
      "epoch": 0.19330428800733776,
      "grad_norm": 0.16883122901416894,
      "learning_rate": 0.0009317071427378624,
      "loss": 3.4126,
      "step": 843
    },
    {
      "epoch": 0.1935335932125659,
      "grad_norm": 0.13822459521804137,
      "learning_rate": 0.0009315196809552956,
      "loss": 3.5749,
      "step": 844
    },
    {
      "epoch": 0.1937628984177941,
      "grad_norm": 0.15130444420805633,
      "learning_rate": 0.0009313319811493594,
      "loss": 3.661,
      "step": 845
    },
    {
      "epoch": 0.19399220362302225,
      "grad_norm": 0.16863390437322567,
      "learning_rate": 0.0009311440434235878,
      "loss": 3.7795,
      "step": 846
    },
    {
      "epoch": 0.1942215088282504,
      "grad_norm": 0.12381820820943545,
      "learning_rate": 0.000930955867881646,
      "loss": 3.455,
      "step": 847
    },
    {
      "epoch": 0.19445081403347855,
      "grad_norm": 0.13940305776375805,
      "learning_rate": 0.0009307674546273301,
      "loss": 3.4985,
      "step": 848
    },
    {
      "epoch": 0.1946801192387067,
      "grad_norm": 0.1396747665733979,
      "learning_rate": 0.0009305788037645681,
      "loss": 3.5891,
      "step": 849
    },
    {
      "epoch": 0.1949094244439349,
      "grad_norm": 0.14519619610319898,
      "learning_rate": 0.000930389915397418,
      "loss": 3.6276,
      "step": 850
    },
    {
      "epoch": 0.19513872964916304,
      "grad_norm": 0.1353512244795869,
      "learning_rate": 0.0009302007896300697,
      "loss": 3.6656,
      "step": 851
    },
    {
      "epoch": 0.1953680348543912,
      "grad_norm": 0.1304853345290525,
      "learning_rate": 0.0009300114265668436,
      "loss": 3.6075,
      "step": 852
    },
    {
      "epoch": 0.19559734005961935,
      "grad_norm": 0.15970163564710566,
      "learning_rate": 0.0009298218263121911,
      "loss": 3.6468,
      "step": 853
    },
    {
      "epoch": 0.1958266452648475,
      "grad_norm": 0.15628951198032945,
      "learning_rate": 0.0009296319889706946,
      "loss": 3.5597,
      "step": 854
    },
    {
      "epoch": 0.19605595047007568,
      "grad_norm": 0.12926260247475196,
      "learning_rate": 0.0009294419146470668,
      "loss": 3.6059,
      "step": 855
    },
    {
      "epoch": 0.19628525567530383,
      "grad_norm": 0.14876776911507014,
      "learning_rate": 0.0009292516034461515,
      "loss": 3.7016,
      "step": 856
    },
    {
      "epoch": 0.19651456088053199,
      "grad_norm": 0.12248378043254364,
      "learning_rate": 0.0009290610554729234,
      "loss": 3.4698,
      "step": 857
    },
    {
      "epoch": 0.19674386608576014,
      "grad_norm": 0.13592303366202962,
      "learning_rate": 0.0009288702708324872,
      "loss": 3.6077,
      "step": 858
    },
    {
      "epoch": 0.19697317129098832,
      "grad_norm": 0.14143640548795863,
      "learning_rate": 0.0009286792496300784,
      "loss": 3.7026,
      "step": 859
    },
    {
      "epoch": 0.19720247649621647,
      "grad_norm": 0.12910887506569385,
      "learning_rate": 0.0009284879919710631,
      "loss": 3.5572,
      "step": 860
    },
    {
      "epoch": 0.19743178170144463,
      "grad_norm": 0.12880822135953143,
      "learning_rate": 0.0009282964979609379,
      "loss": 3.5743,
      "step": 861
    },
    {
      "epoch": 0.19766108690667278,
      "grad_norm": 0.1206600188849409,
      "learning_rate": 0.0009281047677053293,
      "loss": 3.7552,
      "step": 862
    },
    {
      "epoch": 0.19789039211190093,
      "grad_norm": 0.12300107419111646,
      "learning_rate": 0.0009279128013099947,
      "loss": 3.5871,
      "step": 863
    },
    {
      "epoch": 0.1981196973171291,
      "grad_norm": 0.127302860377348,
      "learning_rate": 0.0009277205988808214,
      "loss": 3.5916,
      "step": 864
    },
    {
      "epoch": 0.19834900252235727,
      "grad_norm": 0.13484283895289453,
      "learning_rate": 0.0009275281605238268,
      "loss": 3.479,
      "step": 865
    },
    {
      "epoch": 0.19857830772758542,
      "grad_norm": 0.1270664721822015,
      "learning_rate": 0.0009273354863451589,
      "loss": 3.5986,
      "step": 866
    },
    {
      "epoch": 0.19880761293281357,
      "grad_norm": 0.13282225546233123,
      "learning_rate": 0.0009271425764510953,
      "loss": 3.6191,
      "step": 867
    },
    {
      "epoch": 0.19903691813804172,
      "grad_norm": 0.13820174204021363,
      "learning_rate": 0.000926949430948044,
      "loss": 3.6434,
      "step": 868
    },
    {
      "epoch": 0.1992662233432699,
      "grad_norm": 0.1574113149908797,
      "learning_rate": 0.0009267560499425423,
      "loss": 3.6161,
      "step": 869
    },
    {
      "epoch": 0.19949552854849806,
      "grad_norm": 0.1486629590133486,
      "learning_rate": 0.0009265624335412584,
      "loss": 3.5529,
      "step": 870
    },
    {
      "epoch": 0.1997248337537262,
      "grad_norm": 0.13055533262465013,
      "learning_rate": 0.0009263685818509895,
      "loss": 3.5833,
      "step": 871
    },
    {
      "epoch": 0.19995413895895436,
      "grad_norm": 0.14035779709911536,
      "learning_rate": 0.000926174494978663,
      "loss": 3.5429,
      "step": 872
    },
    {
      "epoch": 0.20018344416418252,
      "grad_norm": 0.17236872687966945,
      "learning_rate": 0.000925980173031336,
      "loss": 3.5629,
      "step": 873
    },
    {
      "epoch": 0.2004127493694107,
      "grad_norm": 0.15767352548688807,
      "learning_rate": 0.0009257856161161952,
      "loss": 3.6296,
      "step": 874
    },
    {
      "epoch": 0.20064205457463885,
      "grad_norm": 0.14356036385502272,
      "learning_rate": 0.0009255908243405567,
      "loss": 3.6064,
      "step": 875
    },
    {
      "epoch": 0.200871359779867,
      "grad_norm": 0.1415751319878037,
      "learning_rate": 0.0009253957978118663,
      "loss": 3.4346,
      "step": 876
    },
    {
      "epoch": 0.20110066498509516,
      "grad_norm": 0.12432296896293302,
      "learning_rate": 0.0009252005366376996,
      "loss": 3.6336,
      "step": 877
    },
    {
      "epoch": 0.2013299701903233,
      "grad_norm": 0.1450439320226253,
      "learning_rate": 0.0009250050409257611,
      "loss": 3.5094,
      "step": 878
    },
    {
      "epoch": 0.2015592753955515,
      "grad_norm": 0.13708736439977282,
      "learning_rate": 0.0009248093107838852,
      "loss": 3.5697,
      "step": 879
    },
    {
      "epoch": 0.20178858060077964,
      "grad_norm": 0.14275596718608097,
      "learning_rate": 0.0009246133463200352,
      "loss": 3.4965,
      "step": 880
    },
    {
      "epoch": 0.2020178858060078,
      "grad_norm": 0.13311489024425277,
      "learning_rate": 0.0009244171476423036,
      "loss": 3.6924,
      "step": 881
    },
    {
      "epoch": 0.20224719101123595,
      "grad_norm": 0.12836433201500838,
      "learning_rate": 0.0009242207148589125,
      "loss": 3.4992,
      "step": 882
    },
    {
      "epoch": 0.2024764962164641,
      "grad_norm": 0.13524069272347214,
      "learning_rate": 0.0009240240480782129,
      "loss": 3.504,
      "step": 883
    },
    {
      "epoch": 0.20270580142169228,
      "grad_norm": 0.13686417781872462,
      "learning_rate": 0.000923827147408685,
      "loss": 3.484,
      "step": 884
    },
    {
      "epoch": 0.20293510662692044,
      "grad_norm": 0.13255325605118579,
      "learning_rate": 0.0009236300129589376,
      "loss": 3.6589,
      "step": 885
    },
    {
      "epoch": 0.2031644118321486,
      "grad_norm": 0.12794645160022747,
      "learning_rate": 0.0009234326448377088,
      "loss": 3.5636,
      "step": 886
    },
    {
      "epoch": 0.20339371703737674,
      "grad_norm": 0.14854377096027907,
      "learning_rate": 0.0009232350431538657,
      "loss": 3.6876,
      "step": 887
    },
    {
      "epoch": 0.2036230222426049,
      "grad_norm": 0.13604290715021025,
      "learning_rate": 0.000923037208016404,
      "loss": 3.5908,
      "step": 888
    },
    {
      "epoch": 0.20385232744783308,
      "grad_norm": 0.1479430072460149,
      "learning_rate": 0.0009228391395344482,
      "loss": 3.6999,
      "step": 889
    },
    {
      "epoch": 0.20408163265306123,
      "grad_norm": 0.14369619733455666,
      "learning_rate": 0.0009226408378172515,
      "loss": 3.581,
      "step": 890
    },
    {
      "epoch": 0.20431093785828938,
      "grad_norm": 0.13459509020901092,
      "learning_rate": 0.000922442302974196,
      "loss": 3.6208,
      "step": 891
    },
    {
      "epoch": 0.20454024306351754,
      "grad_norm": 0.13977394073379307,
      "learning_rate": 0.000922243535114792,
      "loss": 3.6371,
      "step": 892
    },
    {
      "epoch": 0.2047695482687457,
      "grad_norm": 0.15048548370039116,
      "learning_rate": 0.0009220445343486785,
      "loss": 3.613,
      "step": 893
    },
    {
      "epoch": 0.20499885347397387,
      "grad_norm": 0.1466210634813974,
      "learning_rate": 0.0009218453007856231,
      "loss": 3.5875,
      "step": 894
    },
    {
      "epoch": 0.20522815867920202,
      "grad_norm": 0.12820003165684332,
      "learning_rate": 0.0009216458345355217,
      "loss": 3.5896,
      "step": 895
    },
    {
      "epoch": 0.20545746388443017,
      "grad_norm": 0.1411331191428333,
      "learning_rate": 0.0009214461357083985,
      "loss": 3.6408,
      "step": 896
    },
    {
      "epoch": 0.20568676908965833,
      "grad_norm": 0.16723889890910854,
      "learning_rate": 0.0009212462044144061,
      "loss": 3.4692,
      "step": 897
    },
    {
      "epoch": 0.20591607429488648,
      "grad_norm": 0.15301198404540073,
      "learning_rate": 0.0009210460407638253,
      "loss": 3.3825,
      "step": 898
    },
    {
      "epoch": 0.20614537950011466,
      "grad_norm": 0.17326954765855057,
      "learning_rate": 0.0009208456448670648,
      "loss": 3.499,
      "step": 899
    },
    {
      "epoch": 0.20637468470534281,
      "grad_norm": 0.15611186988356715,
      "learning_rate": 0.0009206450168346618,
      "loss": 3.6167,
      "step": 900
    },
    {
      "epoch": 0.20660398991057097,
      "grad_norm": 0.15716670064660662,
      "learning_rate": 0.0009204441567772816,
      "loss": 3.6701,
      "step": 901
    },
    {
      "epoch": 0.20683329511579912,
      "grad_norm": 0.14689260241008165,
      "learning_rate": 0.0009202430648057168,
      "loss": 3.5937,
      "step": 902
    },
    {
      "epoch": 0.20706260032102727,
      "grad_norm": 0.14320577246954097,
      "learning_rate": 0.0009200417410308888,
      "loss": 3.5845,
      "step": 903
    },
    {
      "epoch": 0.20729190552625545,
      "grad_norm": 0.14560658320735256,
      "learning_rate": 0.0009198401855638461,
      "loss": 3.6021,
      "step": 904
    },
    {
      "epoch": 0.2075212107314836,
      "grad_norm": 0.15138084640249788,
      "learning_rate": 0.0009196383985157656,
      "loss": 3.5029,
      "step": 905
    },
    {
      "epoch": 0.20775051593671176,
      "grad_norm": 0.16353544586874177,
      "learning_rate": 0.0009194363799979515,
      "loss": 3.637,
      "step": 906
    },
    {
      "epoch": 0.2079798211419399,
      "grad_norm": 0.14052949881287066,
      "learning_rate": 0.000919234130121836,
      "loss": 3.5699,
      "step": 907
    },
    {
      "epoch": 0.2082091263471681,
      "grad_norm": 0.14299722824343739,
      "learning_rate": 0.000919031648998979,
      "loss": 3.5159,
      "step": 908
    },
    {
      "epoch": 0.20843843155239625,
      "grad_norm": 0.14711037547180347,
      "learning_rate": 0.0009188289367410672,
      "loss": 3.5553,
      "step": 909
    },
    {
      "epoch": 0.2086677367576244,
      "grad_norm": 0.15173272977316915,
      "learning_rate": 0.0009186259934599157,
      "loss": 3.4925,
      "step": 910
    },
    {
      "epoch": 0.20889704196285255,
      "grad_norm": 0.1368241397490562,
      "learning_rate": 0.0009184228192674666,
      "loss": 3.3909,
      "step": 911
    },
    {
      "epoch": 0.2091263471680807,
      "grad_norm": 0.15737028615051063,
      "learning_rate": 0.0009182194142757893,
      "loss": 3.6643,
      "step": 912
    },
    {
      "epoch": 0.2093556523733089,
      "grad_norm": 0.14221695771803494,
      "learning_rate": 0.0009180157785970808,
      "loss": 3.6594,
      "step": 913
    },
    {
      "epoch": 0.20958495757853704,
      "grad_norm": 0.14371702317189522,
      "learning_rate": 0.0009178119123436651,
      "loss": 3.6868,
      "step": 914
    },
    {
      "epoch": 0.2098142627837652,
      "grad_norm": 0.13619854686648106,
      "learning_rate": 0.0009176078156279932,
      "loss": 3.6027,
      "step": 915
    },
    {
      "epoch": 0.21004356798899335,
      "grad_norm": 0.14309907981105635,
      "learning_rate": 0.000917403488562644,
      "loss": 3.6371,
      "step": 916
    },
    {
      "epoch": 0.2102728731942215,
      "grad_norm": 0.13686140742859967,
      "learning_rate": 0.0009171989312603226,
      "loss": 3.5841,
      "step": 917
    },
    {
      "epoch": 0.21050217839944968,
      "grad_norm": 0.14475745660829445,
      "learning_rate": 0.0009169941438338616,
      "loss": 3.676,
      "step": 918
    },
    {
      "epoch": 0.21073148360467783,
      "grad_norm": 0.14529581649727888,
      "learning_rate": 0.0009167891263962202,
      "loss": 3.5385,
      "step": 919
    },
    {
      "epoch": 0.21096078880990599,
      "grad_norm": 0.13979407923990594,
      "learning_rate": 0.0009165838790604848,
      "loss": 3.4924,
      "step": 920
    },
    {
      "epoch": 0.21119009401513414,
      "grad_norm": 0.12268187624869317,
      "learning_rate": 0.0009163784019398685,
      "loss": 3.5998,
      "step": 921
    },
    {
      "epoch": 0.2114193992203623,
      "grad_norm": 0.13947499217630432,
      "learning_rate": 0.0009161726951477107,
      "loss": 3.728,
      "step": 922
    },
    {
      "epoch": 0.21164870442559047,
      "grad_norm": 0.15224647443160624,
      "learning_rate": 0.0009159667587974785,
      "loss": 3.58,
      "step": 923
    },
    {
      "epoch": 0.21187800963081863,
      "grad_norm": 0.12976490989047176,
      "learning_rate": 0.0009157605930027646,
      "loss": 3.5648,
      "step": 924
    },
    {
      "epoch": 0.21210731483604678,
      "grad_norm": 0.12896635673566376,
      "learning_rate": 0.0009155541978772887,
      "loss": 3.496,
      "step": 925
    },
    {
      "epoch": 0.21233662004127493,
      "grad_norm": 0.12962924051865982,
      "learning_rate": 0.0009153475735348973,
      "loss": 3.3471,
      "step": 926
    },
    {
      "epoch": 0.21256592524650308,
      "grad_norm": 0.12565111810075946,
      "learning_rate": 0.0009151407200895625,
      "loss": 3.4912,
      "step": 927
    },
    {
      "epoch": 0.21279523045173127,
      "grad_norm": 0.13190135022492586,
      "learning_rate": 0.0009149336376553836,
      "loss": 3.5397,
      "step": 928
    },
    {
      "epoch": 0.21302453565695942,
      "grad_norm": 0.13503458684602543,
      "learning_rate": 0.000914726326346586,
      "loss": 3.5445,
      "step": 929
    },
    {
      "epoch": 0.21325384086218757,
      "grad_norm": 0.15936131350818214,
      "learning_rate": 0.0009145187862775209,
      "loss": 3.5502,
      "step": 930
    },
    {
      "epoch": 0.21348314606741572,
      "grad_norm": 0.14419729721738034,
      "learning_rate": 0.0009143110175626661,
      "loss": 3.587,
      "step": 931
    },
    {
      "epoch": 0.21371245127264388,
      "grad_norm": 0.1310642132913455,
      "learning_rate": 0.0009141030203166256,
      "loss": 3.621,
      "step": 932
    },
    {
      "epoch": 0.21394175647787206,
      "grad_norm": 0.1652207619030664,
      "learning_rate": 0.0009138947946541291,
      "loss": 3.4831,
      "step": 933
    },
    {
      "epoch": 0.2141710616831002,
      "grad_norm": 0.14100583231352126,
      "learning_rate": 0.0009136863406900328,
      "loss": 3.7369,
      "step": 934
    },
    {
      "epoch": 0.21440036688832836,
      "grad_norm": 0.1376552683581864,
      "learning_rate": 0.0009134776585393181,
      "loss": 3.4716,
      "step": 935
    },
    {
      "epoch": 0.21462967209355652,
      "grad_norm": 0.15535794669790065,
      "learning_rate": 0.0009132687483170929,
      "loss": 3.5179,
      "step": 936
    },
    {
      "epoch": 0.21485897729878467,
      "grad_norm": 0.13535223134614832,
      "learning_rate": 0.0009130596101385906,
      "loss": 3.6605,
      "step": 937
    },
    {
      "epoch": 0.21508828250401285,
      "grad_norm": 0.13694498148077966,
      "learning_rate": 0.0009128502441191706,
      "loss": 3.5121,
      "step": 938
    },
    {
      "epoch": 0.215317587709241,
      "grad_norm": 0.13349987231792154,
      "learning_rate": 0.0009126406503743174,
      "loss": 3.5457,
      "step": 939
    },
    {
      "epoch": 0.21554689291446916,
      "grad_norm": 0.1302531252259541,
      "learning_rate": 0.000912430829019642,
      "loss": 3.5508,
      "step": 940
    },
    {
      "epoch": 0.2157761981196973,
      "grad_norm": 0.14664981078614828,
      "learning_rate": 0.0009122207801708802,
      "loss": 3.5805,
      "step": 941
    },
    {
      "epoch": 0.21600550332492546,
      "grad_norm": 0.127629342567122,
      "learning_rate": 0.0009120105039438936,
      "loss": 3.5767,
      "step": 942
    },
    {
      "epoch": 0.21623480853015364,
      "grad_norm": 0.13868940678373703,
      "learning_rate": 0.0009118000004546689,
      "loss": 3.6165,
      "step": 943
    },
    {
      "epoch": 0.2164641137353818,
      "grad_norm": 0.1253829840932939,
      "learning_rate": 0.0009115892698193189,
      "loss": 3.4228,
      "step": 944
    },
    {
      "epoch": 0.21669341894060995,
      "grad_norm": 0.15469246182793872,
      "learning_rate": 0.0009113783121540807,
      "loss": 3.5025,
      "step": 945
    },
    {
      "epoch": 0.2169227241458381,
      "grad_norm": 0.1326359941333622,
      "learning_rate": 0.0009111671275753176,
      "loss": 3.5333,
      "step": 946
    },
    {
      "epoch": 0.21715202935106626,
      "grad_norm": 0.1632979833761056,
      "learning_rate": 0.0009109557161995172,
      "loss": 3.6465,
      "step": 947
    },
    {
      "epoch": 0.21738133455629444,
      "grad_norm": 0.13736860758772074,
      "learning_rate": 0.000910744078143293,
      "loss": 3.5927,
      "step": 948
    },
    {
      "epoch": 0.2176106397615226,
      "grad_norm": 0.13742281623003655,
      "learning_rate": 0.0009105322135233828,
      "loss": 3.493,
      "step": 949
    },
    {
      "epoch": 0.21783994496675074,
      "grad_norm": 0.14048432298484592,
      "learning_rate": 0.0009103201224566498,
      "loss": 3.5971,
      "step": 950
    },
    {
      "epoch": 0.2180692501719789,
      "grad_norm": 0.12922005724467056,
      "learning_rate": 0.0009101078050600821,
      "loss": 3.6033,
      "step": 951
    },
    {
      "epoch": 0.21829855537720708,
      "grad_norm": 0.13016096051571124,
      "learning_rate": 0.0009098952614507925,
      "loss": 3.6168,
      "step": 952
    },
    {
      "epoch": 0.21852786058243523,
      "grad_norm": 0.15253343901381866,
      "learning_rate": 0.0009096824917460186,
      "loss": 3.4912,
      "step": 953
    },
    {
      "epoch": 0.21875716578766338,
      "grad_norm": 0.1457921328579775,
      "learning_rate": 0.0009094694960631228,
      "loss": 3.4972,
      "step": 954
    },
    {
      "epoch": 0.21898647099289154,
      "grad_norm": 0.1385961287776174,
      "learning_rate": 0.0009092562745195921,
      "loss": 3.7091,
      "step": 955
    },
    {
      "epoch": 0.2192157761981197,
      "grad_norm": 0.1345426455795194,
      "learning_rate": 0.000909042827233038,
      "loss": 3.5767,
      "step": 956
    },
    {
      "epoch": 0.21944508140334787,
      "grad_norm": 0.1409610533633448,
      "learning_rate": 0.0009088291543211967,
      "loss": 3.5138,
      "step": 957
    },
    {
      "epoch": 0.21967438660857602,
      "grad_norm": 0.12554525263200672,
      "learning_rate": 0.0009086152559019286,
      "loss": 3.4966,
      "step": 958
    },
    {
      "epoch": 0.21990369181380418,
      "grad_norm": 0.1333760810756069,
      "learning_rate": 0.0009084011320932188,
      "loss": 3.514,
      "step": 959
    },
    {
      "epoch": 0.22013299701903233,
      "grad_norm": 0.14925415935266403,
      "learning_rate": 0.0009081867830131765,
      "loss": 3.4865,
      "step": 960
    },
    {
      "epoch": 0.22036230222426048,
      "grad_norm": 0.1471498877755403,
      "learning_rate": 0.0009079722087800352,
      "loss": 3.5191,
      "step": 961
    },
    {
      "epoch": 0.22059160742948866,
      "grad_norm": 0.13296319411261423,
      "learning_rate": 0.0009077574095121526,
      "loss": 3.5027,
      "step": 962
    },
    {
      "epoch": 0.22082091263471681,
      "grad_norm": 0.1387457784795869,
      "learning_rate": 0.0009075423853280106,
      "loss": 3.6665,
      "step": 963
    },
    {
      "epoch": 0.22105021783994497,
      "grad_norm": 0.14994754829726983,
      "learning_rate": 0.000907327136346215,
      "loss": 3.4957,
      "step": 964
    },
    {
      "epoch": 0.22127952304517312,
      "grad_norm": 0.13108162952869368,
      "learning_rate": 0.0009071116626854958,
      "loss": 3.596,
      "step": 965
    },
    {
      "epoch": 0.22150882825040127,
      "grad_norm": 0.1385534379099724,
      "learning_rate": 0.0009068959644647068,
      "loss": 3.5652,
      "step": 966
    },
    {
      "epoch": 0.22173813345562945,
      "grad_norm": 0.13711301943459367,
      "learning_rate": 0.0009066800418028256,
      "loss": 3.5466,
      "step": 967
    },
    {
      "epoch": 0.2219674386608576,
      "grad_norm": 0.13078441989983453,
      "learning_rate": 0.0009064638948189539,
      "loss": 3.5428,
      "step": 968
    },
    {
      "epoch": 0.22219674386608576,
      "grad_norm": 0.1527914794013222,
      "learning_rate": 0.0009062475236323168,
      "loss": 3.5105,
      "step": 969
    },
    {
      "epoch": 0.2224260490713139,
      "grad_norm": 0.13671575662699326,
      "learning_rate": 0.0009060309283622632,
      "loss": 3.56,
      "step": 970
    },
    {
      "epoch": 0.22265535427654207,
      "grad_norm": 0.14066530870115684,
      "learning_rate": 0.0009058141091282656,
      "loss": 3.4759,
      "step": 971
    },
    {
      "epoch": 0.22288465948177025,
      "grad_norm": 0.1428041857071868,
      "learning_rate": 0.0009055970660499199,
      "loss": 3.5049,
      "step": 972
    },
    {
      "epoch": 0.2231139646869984,
      "grad_norm": 0.15553124524965356,
      "learning_rate": 0.0009053797992469461,
      "loss": 3.5858,
      "step": 973
    },
    {
      "epoch": 0.22334326989222655,
      "grad_norm": 0.1461260892644773,
      "learning_rate": 0.0009051623088391865,
      "loss": 3.763,
      "step": 974
    },
    {
      "epoch": 0.2235725750974547,
      "grad_norm": 0.13838802349871812,
      "learning_rate": 0.0009049445949466078,
      "loss": 3.5394,
      "step": 975
    },
    {
      "epoch": 0.22380188030268286,
      "grad_norm": 0.1189738190529518,
      "learning_rate": 0.0009047266576892992,
      "loss": 3.5321,
      "step": 976
    },
    {
      "epoch": 0.22403118550791104,
      "grad_norm": 0.18606557388607844,
      "learning_rate": 0.0009045084971874737,
      "loss": 3.5766,
      "step": 977
    },
    {
      "epoch": 0.2242604907131392,
      "grad_norm": 0.15158250261226308,
      "learning_rate": 0.0009042901135614671,
      "loss": 3.4764,
      "step": 978
    },
    {
      "epoch": 0.22448979591836735,
      "grad_norm": 0.12585212028686518,
      "learning_rate": 0.0009040715069317382,
      "loss": 3.4799,
      "step": 979
    },
    {
      "epoch": 0.2247191011235955,
      "grad_norm": 0.1578667010568558,
      "learning_rate": 0.0009038526774188691,
      "loss": 3.5642,
      "step": 980
    },
    {
      "epoch": 0.22494840632882365,
      "grad_norm": 0.13952611322074163,
      "learning_rate": 0.0009036336251435648,
      "loss": 3.5751,
      "step": 981
    },
    {
      "epoch": 0.22517771153405183,
      "grad_norm": 0.14459546133330678,
      "learning_rate": 0.0009034143502266528,
      "loss": 3.4529,
      "step": 982
    },
    {
      "epoch": 0.22540701673928,
      "grad_norm": 0.11972780454992622,
      "learning_rate": 0.0009031948527890839,
      "loss": 3.6154,
      "step": 983
    },
    {
      "epoch": 0.22563632194450814,
      "grad_norm": 0.13159791717160566,
      "learning_rate": 0.0009029751329519311,
      "loss": 3.6733,
      "step": 984
    },
    {
      "epoch": 0.2258656271497363,
      "grad_norm": 0.13186188757389525,
      "learning_rate": 0.000902755190836391,
      "loss": 3.5899,
      "step": 985
    },
    {
      "epoch": 0.22609493235496445,
      "grad_norm": 0.16453804767439997,
      "learning_rate": 0.0009025350265637815,
      "loss": 3.4698,
      "step": 986
    },
    {
      "epoch": 0.22632423756019263,
      "grad_norm": 0.16582620759660818,
      "learning_rate": 0.0009023146402555442,
      "loss": 3.7237,
      "step": 987
    },
    {
      "epoch": 0.22655354276542078,
      "grad_norm": 0.14390379475168652,
      "learning_rate": 0.0009020940320332425,
      "loss": 3.7737,
      "step": 988
    },
    {
      "epoch": 0.22678284797064893,
      "grad_norm": 0.12833405929114092,
      "learning_rate": 0.0009018732020185624,
      "loss": 3.6548,
      "step": 989
    },
    {
      "epoch": 0.22701215317587708,
      "grad_norm": 0.14241721992051937,
      "learning_rate": 0.0009016521503333126,
      "loss": 3.5043,
      "step": 990
    },
    {
      "epoch": 0.22724145838110524,
      "grad_norm": 0.1434962293012857,
      "learning_rate": 0.0009014308770994235,
      "loss": 3.4131,
      "step": 991
    },
    {
      "epoch": 0.22747076358633342,
      "grad_norm": 0.14208559166916157,
      "learning_rate": 0.0009012093824389477,
      "loss": 3.5673,
      "step": 992
    },
    {
      "epoch": 0.22770006879156157,
      "grad_norm": 0.15406299828351594,
      "learning_rate": 0.0009009876664740605,
      "loss": 3.5916,
      "step": 993
    },
    {
      "epoch": 0.22792937399678972,
      "grad_norm": 0.150934034246715,
      "learning_rate": 0.0009007657293270588,
      "loss": 3.5387,
      "step": 994
    },
    {
      "epoch": 0.22815867920201788,
      "grad_norm": 0.14624068244999955,
      "learning_rate": 0.0009005435711203618,
      "loss": 3.4821,
      "step": 995
    },
    {
      "epoch": 0.22838798440724603,
      "grad_norm": 0.14886426781355916,
      "learning_rate": 0.0009003211919765101,
      "loss": 3.6321,
      "step": 996
    },
    {
      "epoch": 0.2286172896124742,
      "grad_norm": 0.1312009223379268,
      "learning_rate": 0.000900098592018167,
      "loss": 3.4544,
      "step": 997
    },
    {
      "epoch": 0.22884659481770236,
      "grad_norm": 0.14246103249761333,
      "learning_rate": 0.0008998757713681169,
      "loss": 3.6063,
      "step": 998
    },
    {
      "epoch": 0.22907590002293052,
      "grad_norm": 0.13054664494924548,
      "learning_rate": 0.0008996527301492663,
      "loss": 3.7081,
      "step": 999
    },
    {
      "epoch": 0.22930520522815867,
      "grad_norm": 0.1419123702258663,
      "learning_rate": 0.0008994294684846431,
      "loss": 3.4853,
      "step": 1000
    },
    {
      "epoch": 0.22953451043338685,
      "grad_norm": 0.13874458801663914,
      "learning_rate": 0.0008992059864973972,
      "loss": 3.4697,
      "step": 1001
    },
    {
      "epoch": 0.229763815638615,
      "grad_norm": 0.13910918720018547,
      "learning_rate": 0.0008989822843107994,
      "loss": 3.5315,
      "step": 1002
    },
    {
      "epoch": 0.22999312084384316,
      "grad_norm": 0.128279039450511,
      "learning_rate": 0.0008987583620482427,
      "loss": 3.6573,
      "step": 1003
    },
    {
      "epoch": 0.2302224260490713,
      "grad_norm": 0.1364074396624975,
      "learning_rate": 0.0008985342198332407,
      "loss": 3.559,
      "step": 1004
    },
    {
      "epoch": 0.23045173125429946,
      "grad_norm": 0.12750273247919058,
      "learning_rate": 0.0008983098577894292,
      "loss": 3.6331,
      "step": 1005
    },
    {
      "epoch": 0.23068103645952764,
      "grad_norm": 0.12827706095262503,
      "learning_rate": 0.0008980852760405646,
      "loss": 3.5877,
      "step": 1006
    },
    {
      "epoch": 0.2309103416647558,
      "grad_norm": 0.13583919416818532,
      "learning_rate": 0.0008978604747105246,
      "loss": 3.5924,
      "step": 1007
    },
    {
      "epoch": 0.23113964686998395,
      "grad_norm": 0.12880201356731297,
      "learning_rate": 0.0008976354539233082,
      "loss": 3.5644,
      "step": 1008
    },
    {
      "epoch": 0.2313689520752121,
      "grad_norm": 0.137404615077182,
      "learning_rate": 0.0008974102138030354,
      "loss": 3.6564,
      "step": 1009
    },
    {
      "epoch": 0.23159825728044026,
      "grad_norm": 0.1348790907397189,
      "learning_rate": 0.0008971847544739468,
      "loss": 3.4629,
      "step": 1010
    },
    {
      "epoch": 0.23182756248566844,
      "grad_norm": 0.17110208804870367,
      "learning_rate": 0.000896959076060405,
      "loss": 3.7396,
      "step": 1011
    },
    {
      "epoch": 0.2320568676908966,
      "grad_norm": 0.14989826974638878,
      "learning_rate": 0.0008967331786868919,
      "loss": 3.6002,
      "step": 1012
    },
    {
      "epoch": 0.23228617289612474,
      "grad_norm": 0.12762441370095892,
      "learning_rate": 0.0008965070624780116,
      "loss": 3.5916,
      "step": 1013
    },
    {
      "epoch": 0.2325154781013529,
      "grad_norm": 0.1424871974959941,
      "learning_rate": 0.0008962807275584879,
      "loss": 3.549,
      "step": 1014
    },
    {
      "epoch": 0.23274478330658105,
      "grad_norm": 0.13888327400951014,
      "learning_rate": 0.0008960541740531658,
      "loss": 3.7545,
      "step": 1015
    },
    {
      "epoch": 0.23297408851180923,
      "grad_norm": 0.14712923088910718,
      "learning_rate": 0.0008958274020870107,
      "loss": 3.7904,
      "step": 1016
    },
    {
      "epoch": 0.23320339371703738,
      "grad_norm": 0.1414200266379628,
      "learning_rate": 0.0008956004117851083,
      "loss": 3.6108,
      "step": 1017
    },
    {
      "epoch": 0.23343269892226554,
      "grad_norm": 0.14423737343329368,
      "learning_rate": 0.0008953732032726652,
      "loss": 3.5009,
      "step": 1018
    },
    {
      "epoch": 0.2336620041274937,
      "grad_norm": 0.13574656107052016,
      "learning_rate": 0.0008951457766750079,
      "loss": 3.5241,
      "step": 1019
    },
    {
      "epoch": 0.23389130933272184,
      "grad_norm": 0.14276836151916852,
      "learning_rate": 0.0008949181321175837,
      "loss": 3.5343,
      "step": 1020
    },
    {
      "epoch": 0.23412061453795002,
      "grad_norm": 0.131466189159688,
      "learning_rate": 0.0008946902697259593,
      "loss": 3.6137,
      "step": 1021
    },
    {
      "epoch": 0.23434991974317818,
      "grad_norm": 0.12331285008271232,
      "learning_rate": 0.0008944621896258225,
      "loss": 3.4622,
      "step": 1022
    },
    {
      "epoch": 0.23457922494840633,
      "grad_norm": 0.12578326286306896,
      "learning_rate": 0.0008942338919429805,
      "loss": 3.6619,
      "step": 1023
    },
    {
      "epoch": 0.23480853015363448,
      "grad_norm": 0.1391147468152276,
      "learning_rate": 0.0008940053768033609,
      "loss": 3.6299,
      "step": 1024
    },
    {
      "epoch": 0.23503783535886263,
      "grad_norm": 0.1210561809445352,
      "learning_rate": 0.0008937766443330113,
      "loss": 3.6315,
      "step": 1025
    },
    {
      "epoch": 0.23526714056409082,
      "grad_norm": 0.14191692072700507,
      "learning_rate": 0.0008935476946580987,
      "loss": 3.5676,
      "step": 1026
    },
    {
      "epoch": 0.23549644576931897,
      "grad_norm": 0.1417210255043145,
      "learning_rate": 0.0008933185279049103,
      "loss": 3.6466,
      "step": 1027
    },
    {
      "epoch": 0.23572575097454712,
      "grad_norm": 0.1413485099322972,
      "learning_rate": 0.000893089144199853,
      "loss": 3.6261,
      "step": 1028
    },
    {
      "epoch": 0.23595505617977527,
      "grad_norm": 0.13751824430300405,
      "learning_rate": 0.0008928595436694532,
      "loss": 3.7396,
      "step": 1029
    },
    {
      "epoch": 0.23618436138500343,
      "grad_norm": 0.14693370926925983,
      "learning_rate": 0.0008926297264403571,
      "loss": 3.5533,
      "step": 1030
    },
    {
      "epoch": 0.2364136665902316,
      "grad_norm": 0.13432235365711923,
      "learning_rate": 0.0008923996926393305,
      "loss": 3.5356,
      "step": 1031
    },
    {
      "epoch": 0.23664297179545976,
      "grad_norm": 0.141160308034562,
      "learning_rate": 0.0008921694423932583,
      "loss": 3.5025,
      "step": 1032
    },
    {
      "epoch": 0.23687227700068791,
      "grad_norm": 0.13750544128781747,
      "learning_rate": 0.0008919389758291449,
      "loss": 3.5865,
      "step": 1033
    },
    {
      "epoch": 0.23710158220591607,
      "grad_norm": 0.14225968523328583,
      "learning_rate": 0.0008917082930741145,
      "loss": 3.5451,
      "step": 1034
    },
    {
      "epoch": 0.23733088741114422,
      "grad_norm": 0.15819401187393933,
      "learning_rate": 0.0008914773942554098,
      "loss": 3.5113,
      "step": 1035
    },
    {
      "epoch": 0.2375601926163724,
      "grad_norm": 0.13951694808959145,
      "learning_rate": 0.0008912462795003932,
      "loss": 3.4751,
      "step": 1036
    },
    {
      "epoch": 0.23778949782160055,
      "grad_norm": 0.15465355885338913,
      "learning_rate": 0.000891014948936546,
      "loss": 3.6182,
      "step": 1037
    },
    {
      "epoch": 0.2380188030268287,
      "grad_norm": 0.13844478085095532,
      "learning_rate": 0.0008907834026914686,
      "loss": 3.5688,
      "step": 1038
    },
    {
      "epoch": 0.23824810823205686,
      "grad_norm": 0.15846292812468354,
      "learning_rate": 0.0008905516408928804,
      "loss": 3.6466,
      "step": 1039
    },
    {
      "epoch": 0.238477413437285,
      "grad_norm": 0.1516441949177498,
      "learning_rate": 0.0008903196636686197,
      "loss": 3.6331,
      "step": 1040
    },
    {
      "epoch": 0.2387067186425132,
      "grad_norm": 0.14402508413146864,
      "learning_rate": 0.0008900874711466434,
      "loss": 3.5851,
      "step": 1041
    },
    {
      "epoch": 0.23893602384774135,
      "grad_norm": 0.13840910108856888,
      "learning_rate": 0.0008898550634550275,
      "loss": 3.5492,
      "step": 1042
    },
    {
      "epoch": 0.2391653290529695,
      "grad_norm": 0.16751257462684488,
      "learning_rate": 0.0008896224407219666,
      "loss": 3.6326,
      "step": 1043
    },
    {
      "epoch": 0.23939463425819765,
      "grad_norm": 0.13326726644741133,
      "learning_rate": 0.0008893896030757737,
      "loss": 3.6198,
      "step": 1044
    },
    {
      "epoch": 0.23962393946342583,
      "grad_norm": 0.12800850240756867,
      "learning_rate": 0.0008891565506448804,
      "loss": 3.6902,
      "step": 1045
    },
    {
      "epoch": 0.239853244668654,
      "grad_norm": 0.13422009528554957,
      "learning_rate": 0.0008889232835578372,
      "loss": 3.5467,
      "step": 1046
    },
    {
      "epoch": 0.24008254987388214,
      "grad_norm": 0.13278517417813582,
      "learning_rate": 0.0008886898019433122,
      "loss": 3.4096,
      "step": 1047
    },
    {
      "epoch": 0.2403118550791103,
      "grad_norm": 0.12429498187013793,
      "learning_rate": 0.0008884561059300926,
      "loss": 3.3996,
      "step": 1048
    },
    {
      "epoch": 0.24054116028433845,
      "grad_norm": 0.14928893042236369,
      "learning_rate": 0.0008882221956470836,
      "loss": 3.6728,
      "step": 1049
    },
    {
      "epoch": 0.24077046548956663,
      "grad_norm": 0.15163331981980627,
      "learning_rate": 0.0008879880712233084,
      "loss": 3.5648,
      "step": 1050
    },
    {
      "epoch": 0.24099977069479478,
      "grad_norm": 0.145042475920795,
      "learning_rate": 0.0008877537327879086,
      "loss": 3.54,
      "step": 1051
    },
    {
      "epoch": 0.24122907590002293,
      "grad_norm": 0.13091118238583527,
      "learning_rate": 0.0008875191804701434,
      "loss": 3.5052,
      "step": 1052
    },
    {
      "epoch": 0.24145838110525109,
      "grad_norm": 0.173157087679912,
      "learning_rate": 0.0008872844143993908,
      "loss": 3.5695,
      "step": 1053
    },
    {
      "epoch": 0.24168768631047924,
      "grad_norm": 0.1394594656556252,
      "learning_rate": 0.0008870494347051458,
      "loss": 3.6148,
      "step": 1054
    },
    {
      "epoch": 0.24191699151570742,
      "grad_norm": 0.12714509356154072,
      "learning_rate": 0.0008868142415170218,
      "loss": 3.6065,
      "step": 1055
    },
    {
      "epoch": 0.24214629672093557,
      "grad_norm": 0.15252576726030917,
      "learning_rate": 0.0008865788349647495,
      "loss": 3.5383,
      "step": 1056
    },
    {
      "epoch": 0.24237560192616373,
      "grad_norm": 0.13742020690110632,
      "learning_rate": 0.0008863432151781781,
      "loss": 3.5631,
      "step": 1057
    },
    {
      "epoch": 0.24260490713139188,
      "grad_norm": 0.12467600904426886,
      "learning_rate": 0.0008861073822872734,
      "loss": 3.5354,
      "step": 1058
    },
    {
      "epoch": 0.24283421233662003,
      "grad_norm": 0.1519934271405295,
      "learning_rate": 0.0008858713364221195,
      "loss": 3.4756,
      "step": 1059
    },
    {
      "epoch": 0.2430635175418482,
      "grad_norm": 0.13976650646743877,
      "learning_rate": 0.0008856350777129178,
      "loss": 3.6753,
      "step": 1060
    },
    {
      "epoch": 0.24329282274707636,
      "grad_norm": 0.12259247612993472,
      "learning_rate": 0.0008853986062899868,
      "loss": 3.4707,
      "step": 1061
    },
    {
      "epoch": 0.24352212795230452,
      "grad_norm": 0.12823305236289978,
      "learning_rate": 0.0008851619222837626,
      "loss": 3.4879,
      "step": 1062
    },
    {
      "epoch": 0.24375143315753267,
      "grad_norm": 0.13211148557637142,
      "learning_rate": 0.0008849250258247986,
      "loss": 3.6081,
      "step": 1063
    },
    {
      "epoch": 0.24398073836276082,
      "grad_norm": 0.14752031938149435,
      "learning_rate": 0.0008846879170437652,
      "loss": 3.6099,
      "step": 1064
    },
    {
      "epoch": 0.244210043567989,
      "grad_norm": 0.13239108561729315,
      "learning_rate": 0.0008844505960714503,
      "loss": 3.5142,
      "step": 1065
    },
    {
      "epoch": 0.24443934877321716,
      "grad_norm": 0.14468104392001285,
      "learning_rate": 0.0008842130630387581,
      "loss": 3.5008,
      "step": 1066
    },
    {
      "epoch": 0.2446686539784453,
      "grad_norm": 0.15006522727700267,
      "learning_rate": 0.0008839753180767108,
      "loss": 3.4577,
      "step": 1067
    },
    {
      "epoch": 0.24489795918367346,
      "grad_norm": 0.13090750043532992,
      "learning_rate": 0.0008837373613164464,
      "loss": 3.4879,
      "step": 1068
    },
    {
      "epoch": 0.24512726438890162,
      "grad_norm": 0.13381227532912426,
      "learning_rate": 0.0008834991928892204,
      "loss": 3.63,
      "step": 1069
    },
    {
      "epoch": 0.2453565695941298,
      "grad_norm": 0.16515856973692855,
      "learning_rate": 0.000883260812926405,
      "loss": 3.6733,
      "step": 1070
    },
    {
      "epoch": 0.24558587479935795,
      "grad_norm": 0.14206188443431095,
      "learning_rate": 0.000883022221559489,
      "loss": 3.6388,
      "step": 1071
    },
    {
      "epoch": 0.2458151800045861,
      "grad_norm": 0.1282172127795998,
      "learning_rate": 0.0008827834189200778,
      "loss": 3.6093,
      "step": 1072
    },
    {
      "epoch": 0.24604448520981426,
      "grad_norm": 0.14870537186660931,
      "learning_rate": 0.0008825444051398934,
      "loss": 3.5914,
      "step": 1073
    },
    {
      "epoch": 0.2462737904150424,
      "grad_norm": 0.12866847690885316,
      "learning_rate": 0.000882305180350774,
      "loss": 3.4713,
      "step": 1074
    },
    {
      "epoch": 0.2465030956202706,
      "grad_norm": 0.13358979776398103,
      "learning_rate": 0.0008820657446846745,
      "loss": 3.6526,
      "step": 1075
    },
    {
      "epoch": 0.24673240082549874,
      "grad_norm": 0.15741103852514707,
      "learning_rate": 0.0008818260982736661,
      "loss": 3.6006,
      "step": 1076
    },
    {
      "epoch": 0.2469617060307269,
      "grad_norm": 0.13045043713425658,
      "learning_rate": 0.000881586241249936,
      "loss": 3.6364,
      "step": 1077
    },
    {
      "epoch": 0.24719101123595505,
      "grad_norm": 0.13711591495334396,
      "learning_rate": 0.0008813461737457877,
      "loss": 3.5515,
      "step": 1078
    },
    {
      "epoch": 0.2474203164411832,
      "grad_norm": 0.1297945988920966,
      "learning_rate": 0.0008811058958936411,
      "loss": 3.499,
      "step": 1079
    },
    {
      "epoch": 0.24764962164641138,
      "grad_norm": 0.15830129174611315,
      "learning_rate": 0.0008808654078260317,
      "loss": 3.7164,
      "step": 1080
    },
    {
      "epoch": 0.24787892685163954,
      "grad_norm": 0.14277551173039418,
      "learning_rate": 0.000880624709675611,
      "loss": 3.7008,
      "step": 1081
    },
    {
      "epoch": 0.2481082320568677,
      "grad_norm": 0.12861861084546197,
      "learning_rate": 0.0008803838015751466,
      "loss": 3.6415,
      "step": 1082
    },
    {
      "epoch": 0.24833753726209584,
      "grad_norm": 0.135127656301749,
      "learning_rate": 0.000880142683657522,
      "loss": 3.6091,
      "step": 1083
    },
    {
      "epoch": 0.248566842467324,
      "grad_norm": 0.12806155450389403,
      "learning_rate": 0.0008799013560557359,
      "loss": 3.4777,
      "step": 1084
    },
    {
      "epoch": 0.24879614767255218,
      "grad_norm": 0.1451855711379548,
      "learning_rate": 0.0008796598189029029,
      "loss": 3.5029,
      "step": 1085
    },
    {
      "epoch": 0.24902545287778033,
      "grad_norm": 0.13850171353676952,
      "learning_rate": 0.0008794180723322536,
      "loss": 3.5052,
      "step": 1086
    },
    {
      "epoch": 0.24925475808300848,
      "grad_norm": 0.1598733703842045,
      "learning_rate": 0.0008791761164771338,
      "loss": 3.6456,
      "step": 1087
    },
    {
      "epoch": 0.24948406328823663,
      "grad_norm": 0.13599904999625806,
      "learning_rate": 0.0008789339514710046,
      "loss": 3.5598,
      "step": 1088
    },
    {
      "epoch": 0.2497133684934648,
      "grad_norm": 0.15561371766298468,
      "learning_rate": 0.0008786915774474424,
      "loss": 3.5257,
      "step": 1089
    },
    {
      "epoch": 0.24994267369869297,
      "grad_norm": 0.14772262829101362,
      "learning_rate": 0.0008784489945401395,
      "loss": 3.595,
      "step": 1090
    },
    {
      "epoch": 0.2501719789039211,
      "grad_norm": 0.14288725134544206,
      "learning_rate": 0.0008782062028829027,
      "loss": 3.4916,
      "step": 1091
    },
    {
      "epoch": 0.2504012841091493,
      "grad_norm": 0.15327066621356808,
      "learning_rate": 0.0008779632026096544,
      "loss": 3.5945,
      "step": 1092
    },
    {
      "epoch": 0.2506305893143774,
      "grad_norm": 0.12628275814146345,
      "learning_rate": 0.0008777199938544318,
      "loss": 3.5936,
      "step": 1093
    },
    {
      "epoch": 0.2508598945196056,
      "grad_norm": 0.1342135857391714,
      "learning_rate": 0.0008774765767513875,
      "loss": 3.4594,
      "step": 1094
    },
    {
      "epoch": 0.25108919972483373,
      "grad_norm": 0.12485072781149054,
      "learning_rate": 0.0008772329514347883,
      "loss": 3.4366,
      "step": 1095
    },
    {
      "epoch": 0.2513185049300619,
      "grad_norm": 0.1363094462167997,
      "learning_rate": 0.0008769891180390167,
      "loss": 3.521,
      "step": 1096
    },
    {
      "epoch": 0.2515478101352901,
      "grad_norm": 0.12104410664890328,
      "learning_rate": 0.0008767450766985694,
      "loss": 3.5565,
      "step": 1097
    },
    {
      "epoch": 0.2517771153405182,
      "grad_norm": 0.11946397327619104,
      "learning_rate": 0.000876500827548058,
      "loss": 3.5808,
      "step": 1098
    },
    {
      "epoch": 0.2520064205457464,
      "grad_norm": 0.12569371737652274,
      "learning_rate": 0.0008762563707222086,
      "loss": 3.5628,
      "step": 1099
    },
    {
      "epoch": 0.2522357257509745,
      "grad_norm": 0.13810154874711833,
      "learning_rate": 0.0008760117063558621,
      "loss": 3.5513,
      "step": 1100
    },
    {
      "epoch": 0.2524650309562027,
      "grad_norm": 0.14926203815805314,
      "learning_rate": 0.0008757668345839738,
      "loss": 3.5795,
      "step": 1101
    },
    {
      "epoch": 0.2526943361614309,
      "grad_norm": 0.13638846751247927,
      "learning_rate": 0.0008755217555416131,
      "loss": 3.5549,
      "step": 1102
    },
    {
      "epoch": 0.252923641366659,
      "grad_norm": 0.15646561471593468,
      "learning_rate": 0.0008752764693639638,
      "loss": 3.4771,
      "step": 1103
    },
    {
      "epoch": 0.2531529465718872,
      "grad_norm": 0.16093730162119096,
      "learning_rate": 0.0008750309761863246,
      "loss": 3.6831,
      "step": 1104
    },
    {
      "epoch": 0.2533822517771153,
      "grad_norm": 0.13438927380853272,
      "learning_rate": 0.0008747852761441078,
      "loss": 3.5089,
      "step": 1105
    },
    {
      "epoch": 0.2536115569823435,
      "grad_norm": 0.15512822677739194,
      "learning_rate": 0.0008745393693728395,
      "loss": 3.4761,
      "step": 1106
    },
    {
      "epoch": 0.2538408621875717,
      "grad_norm": 0.13934079907868874,
      "learning_rate": 0.0008742932560081607,
      "loss": 3.544,
      "step": 1107
    },
    {
      "epoch": 0.2540701673927998,
      "grad_norm": 0.12965220217418685,
      "learning_rate": 0.0008740469361858255,
      "loss": 3.5003,
      "step": 1108
    },
    {
      "epoch": 0.254299472598028,
      "grad_norm": 0.14327196897736505,
      "learning_rate": 0.0008738004100417025,
      "loss": 3.5174,
      "step": 1109
    },
    {
      "epoch": 0.2545287778032561,
      "grad_norm": 0.1451806182349158,
      "learning_rate": 0.0008735536777117739,
      "loss": 3.6596,
      "step": 1110
    },
    {
      "epoch": 0.2547580830084843,
      "grad_norm": 0.13813683985275765,
      "learning_rate": 0.0008733067393321355,
      "loss": 3.6075,
      "step": 1111
    },
    {
      "epoch": 0.2549873882137125,
      "grad_norm": 0.13757213090285897,
      "learning_rate": 0.0008730595950389968,
      "loss": 3.6297,
      "step": 1112
    },
    {
      "epoch": 0.2552166934189406,
      "grad_norm": 0.14446183261877066,
      "learning_rate": 0.000872812244968681,
      "loss": 3.6165,
      "step": 1113
    },
    {
      "epoch": 0.2554459986241688,
      "grad_norm": 0.1505578249982042,
      "learning_rate": 0.0008725646892576249,
      "loss": 3.7008,
      "step": 1114
    },
    {
      "epoch": 0.2556753038293969,
      "grad_norm": 0.1307387827520946,
      "learning_rate": 0.0008723169280423783,
      "loss": 3.5598,
      "step": 1115
    },
    {
      "epoch": 0.2559046090346251,
      "grad_norm": 0.12841841286862068,
      "learning_rate": 0.0008720689614596049,
      "loss": 3.6588,
      "step": 1116
    },
    {
      "epoch": 0.25613391423985327,
      "grad_norm": 0.1358389199552597,
      "learning_rate": 0.0008718207896460811,
      "loss": 3.6254,
      "step": 1117
    },
    {
      "epoch": 0.2563632194450814,
      "grad_norm": 0.1472336910328574,
      "learning_rate": 0.0008715724127386971,
      "loss": 3.6561,
      "step": 1118
    },
    {
      "epoch": 0.2565925246503096,
      "grad_norm": 0.14701450514775924,
      "learning_rate": 0.0008713238308744557,
      "loss": 3.7162,
      "step": 1119
    },
    {
      "epoch": 0.2568218298555377,
      "grad_norm": 0.1396460655742657,
      "learning_rate": 0.0008710750441904733,
      "loss": 3.695,
      "step": 1120
    },
    {
      "epoch": 0.2570511350607659,
      "grad_norm": 0.13791803062359267,
      "learning_rate": 0.0008708260528239789,
      "loss": 3.5468,
      "step": 1121
    },
    {
      "epoch": 0.25728044026599406,
      "grad_norm": 0.1462443931405544,
      "learning_rate": 0.0008705768569123141,
      "loss": 3.4982,
      "step": 1122
    },
    {
      "epoch": 0.2575097454712222,
      "grad_norm": 0.144733013050313,
      "learning_rate": 0.000870327456592934,
      "loss": 3.5718,
      "step": 1123
    },
    {
      "epoch": 0.25773905067645037,
      "grad_norm": 0.1364859622727172,
      "learning_rate": 0.0008700778520034062,
      "loss": 3.5535,
      "step": 1124
    },
    {
      "epoch": 0.2579683558816785,
      "grad_norm": 0.11519092437058902,
      "learning_rate": 0.0008698280432814107,
      "loss": 3.5491,
      "step": 1125
    },
    {
      "epoch": 0.25819766108690667,
      "grad_norm": 0.14536468676124054,
      "learning_rate": 0.0008695780305647405,
      "loss": 3.6288,
      "step": 1126
    },
    {
      "epoch": 0.25842696629213485,
      "grad_norm": 0.1348365662625633,
      "learning_rate": 0.000869327813991301,
      "loss": 3.5713,
      "step": 1127
    },
    {
      "epoch": 0.258656271497363,
      "grad_norm": 0.1274399099811883,
      "learning_rate": 0.0008690773936991098,
      "loss": 3.5439,
      "step": 1128
    },
    {
      "epoch": 0.25888557670259116,
      "grad_norm": 0.1289624233490091,
      "learning_rate": 0.0008688267698262971,
      "loss": 3.5402,
      "step": 1129
    },
    {
      "epoch": 0.2591148819078193,
      "grad_norm": 0.135592776007851,
      "learning_rate": 0.0008685759425111055,
      "loss": 3.4558,
      "step": 1130
    },
    {
      "epoch": 0.25934418711304746,
      "grad_norm": 0.1396686897007555,
      "learning_rate": 0.0008683249118918894,
      "loss": 3.5092,
      "step": 1131
    },
    {
      "epoch": 0.25957349231827564,
      "grad_norm": 0.17599107917933504,
      "learning_rate": 0.0008680736781071159,
      "loss": 3.7661,
      "step": 1132
    },
    {
      "epoch": 0.25980279752350377,
      "grad_norm": 0.14670391172116728,
      "learning_rate": 0.0008678222412953637,
      "loss": 3.414,
      "step": 1133
    },
    {
      "epoch": 0.26003210272873195,
      "grad_norm": 0.1517007696818703,
      "learning_rate": 0.0008675706015953237,
      "loss": 3.6091,
      "step": 1134
    },
    {
      "epoch": 0.2602614079339601,
      "grad_norm": 0.16133657524356834,
      "learning_rate": 0.0008673187591457987,
      "loss": 3.7312,
      "step": 1135
    },
    {
      "epoch": 0.26049071313918826,
      "grad_norm": 0.13477350998957913,
      "learning_rate": 0.0008670667140857032,
      "loss": 3.8274,
      "step": 1136
    },
    {
      "epoch": 0.26072001834441644,
      "grad_norm": 0.13152820331520237,
      "learning_rate": 0.0008668144665540639,
      "loss": 3.6203,
      "step": 1137
    },
    {
      "epoch": 0.26094932354964456,
      "grad_norm": 0.14764698299753495,
      "learning_rate": 0.0008665620166900184,
      "loss": 3.6689,
      "step": 1138
    },
    {
      "epoch": 0.26117862875487274,
      "grad_norm": 0.13568217333333063,
      "learning_rate": 0.0008663093646328167,
      "loss": 3.5896,
      "step": 1139
    },
    {
      "epoch": 0.26140793396010087,
      "grad_norm": 0.13147863231651719,
      "learning_rate": 0.0008660565105218196,
      "loss": 3.5691,
      "step": 1140
    },
    {
      "epoch": 0.26163723916532905,
      "grad_norm": 0.12791666195179968,
      "learning_rate": 0.0008658034544965003,
      "loss": 3.6707,
      "step": 1141
    },
    {
      "epoch": 0.26186654437055723,
      "grad_norm": 0.14824588412746084,
      "learning_rate": 0.0008655501966964423,
      "loss": 3.5557,
      "step": 1142
    },
    {
      "epoch": 0.26209584957578536,
      "grad_norm": 0.1365688473807527,
      "learning_rate": 0.0008652967372613412,
      "loss": 3.5419,
      "step": 1143
    },
    {
      "epoch": 0.26232515478101354,
      "grad_norm": 0.14623741008873833,
      "learning_rate": 0.0008650430763310034,
      "loss": 3.5015,
      "step": 1144
    },
    {
      "epoch": 0.26255445998624166,
      "grad_norm": 0.14401076171544938,
      "learning_rate": 0.0008647892140453466,
      "loss": 3.6434,
      "step": 1145
    },
    {
      "epoch": 0.26278376519146984,
      "grad_norm": 0.12796154666282372,
      "learning_rate": 0.0008645351505443997,
      "loss": 3.5088,
      "step": 1146
    },
    {
      "epoch": 0.263013070396698,
      "grad_norm": 0.14935250523643784,
      "learning_rate": 0.0008642808859683021,
      "loss": 3.7241,
      "step": 1147
    },
    {
      "epoch": 0.26324237560192615,
      "grad_norm": 0.13525335618074147,
      "learning_rate": 0.0008640264204573048,
      "loss": 3.4973,
      "step": 1148
    },
    {
      "epoch": 0.26347168080715433,
      "grad_norm": 0.13088581880852926,
      "learning_rate": 0.0008637717541517689,
      "loss": 3.6123,
      "step": 1149
    },
    {
      "epoch": 0.26370098601238245,
      "grad_norm": 0.139570309580931,
      "learning_rate": 0.0008635168871921669,
      "loss": 3.659,
      "step": 1150
    },
    {
      "epoch": 0.26393029121761064,
      "grad_norm": 0.14851717817714927,
      "learning_rate": 0.0008632618197190816,
      "loss": 3.6578,
      "step": 1151
    },
    {
      "epoch": 0.2641595964228388,
      "grad_norm": 0.13795925194966804,
      "learning_rate": 0.0008630065518732067,
      "loss": 3.5329,
      "step": 1152
    },
    {
      "epoch": 0.26438890162806694,
      "grad_norm": 0.13147109695806572,
      "learning_rate": 0.0008627510837953458,
      "loss": 3.6274,
      "step": 1153
    },
    {
      "epoch": 0.2646182068332951,
      "grad_norm": 0.13646290330302413,
      "learning_rate": 0.0008624954156264137,
      "loss": 3.5426,
      "step": 1154
    },
    {
      "epoch": 0.26484751203852325,
      "grad_norm": 0.1471478940590606,
      "learning_rate": 0.0008622395475074355,
      "loss": 3.4862,
      "step": 1155
    },
    {
      "epoch": 0.26507681724375143,
      "grad_norm": 0.13813641864095788,
      "learning_rate": 0.0008619834795795459,
      "loss": 3.4078,
      "step": 1156
    },
    {
      "epoch": 0.2653061224489796,
      "grad_norm": 0.13293539247657007,
      "learning_rate": 0.0008617272119839903,
      "loss": 3.5047,
      "step": 1157
    },
    {
      "epoch": 0.26553542765420773,
      "grad_norm": 0.1280662305738998,
      "learning_rate": 0.0008614707448621244,
      "loss": 3.4423,
      "step": 1158
    },
    {
      "epoch": 0.2657647328594359,
      "grad_norm": 0.13768456528670867,
      "learning_rate": 0.0008612140783554136,
      "loss": 3.589,
      "step": 1159
    },
    {
      "epoch": 0.2659940380646641,
      "grad_norm": 0.14925015826454263,
      "learning_rate": 0.0008609572126054336,
      "loss": 3.5136,
      "step": 1160
    },
    {
      "epoch": 0.2662233432698922,
      "grad_norm": 0.14687812593341434,
      "learning_rate": 0.0008607001477538696,
      "loss": 3.5604,
      "step": 1161
    },
    {
      "epoch": 0.2664526484751204,
      "grad_norm": 0.14800818274723987,
      "learning_rate": 0.0008604428839425171,
      "loss": 3.6247,
      "step": 1162
    },
    {
      "epoch": 0.2666819536803485,
      "grad_norm": 0.12386274457481873,
      "learning_rate": 0.0008601854213132807,
      "loss": 3.6218,
      "step": 1163
    },
    {
      "epoch": 0.2669112588855767,
      "grad_norm": 0.12408236335031443,
      "learning_rate": 0.0008599277600081757,
      "loss": 3.5635,
      "step": 1164
    },
    {
      "epoch": 0.2671405640908049,
      "grad_norm": 0.1327234928745511,
      "learning_rate": 0.0008596699001693256,
      "loss": 3.5724,
      "step": 1165
    },
    {
      "epoch": 0.267369869296033,
      "grad_norm": 0.12691816168894016,
      "learning_rate": 0.0008594118419389648,
      "loss": 3.5801,
      "step": 1166
    },
    {
      "epoch": 0.2675991745012612,
      "grad_norm": 0.12793510302164263,
      "learning_rate": 0.000859153585459436,
      "loss": 3.6277,
      "step": 1167
    },
    {
      "epoch": 0.2678284797064893,
      "grad_norm": 0.13321730462717465,
      "learning_rate": 0.0008588951308731921,
      "loss": 3.4911,
      "step": 1168
    },
    {
      "epoch": 0.2680577849117175,
      "grad_norm": 0.1455978903421647,
      "learning_rate": 0.0008586364783227949,
      "loss": 3.5642,
      "step": 1169
    },
    {
      "epoch": 0.2682870901169457,
      "grad_norm": 0.13236860050018276,
      "learning_rate": 0.0008583776279509152,
      "loss": 3.4734,
      "step": 1170
    },
    {
      "epoch": 0.2685163953221738,
      "grad_norm": 0.1376824124318853,
      "learning_rate": 0.0008581185799003332,
      "loss": 3.5989,
      "step": 1171
    },
    {
      "epoch": 0.268745700527402,
      "grad_norm": 0.13737430476005985,
      "learning_rate": 0.0008578593343139383,
      "loss": 3.5049,
      "step": 1172
    },
    {
      "epoch": 0.2689750057326301,
      "grad_norm": 0.14872874757132964,
      "learning_rate": 0.0008575998913347283,
      "loss": 3.5601,
      "step": 1173
    },
    {
      "epoch": 0.2692043109378583,
      "grad_norm": 0.13924270209590817,
      "learning_rate": 0.0008573402511058104,
      "loss": 3.5313,
      "step": 1174
    },
    {
      "epoch": 0.2694336161430865,
      "grad_norm": 0.161193842832313,
      "learning_rate": 0.0008570804137704004,
      "loss": 3.5895,
      "step": 1175
    },
    {
      "epoch": 0.2696629213483146,
      "grad_norm": 0.16488875090890687,
      "learning_rate": 0.0008568203794718228,
      "loss": 3.546,
      "step": 1176
    },
    {
      "epoch": 0.2698922265535428,
      "grad_norm": 0.1538141547255964,
      "learning_rate": 0.0008565601483535108,
      "loss": 3.473,
      "step": 1177
    },
    {
      "epoch": 0.2701215317587709,
      "grad_norm": 0.1484959171957521,
      "learning_rate": 0.0008562997205590062,
      "loss": 3.4816,
      "step": 1178
    },
    {
      "epoch": 0.2703508369639991,
      "grad_norm": 0.12140162860833383,
      "learning_rate": 0.0008560390962319591,
      "loss": 3.4323,
      "step": 1179
    },
    {
      "epoch": 0.27058014216922727,
      "grad_norm": 0.1365945697720345,
      "learning_rate": 0.0008557782755161281,
      "loss": 3.5899,
      "step": 1180
    },
    {
      "epoch": 0.2708094473744554,
      "grad_norm": 0.13436368822800257,
      "learning_rate": 0.0008555172585553804,
      "loss": 3.548,
      "step": 1181
    },
    {
      "epoch": 0.2710387525796836,
      "grad_norm": 0.12971116418153486,
      "learning_rate": 0.000855256045493691,
      "loss": 3.4968,
      "step": 1182
    },
    {
      "epoch": 0.2712680577849117,
      "grad_norm": 0.1278181583672108,
      "learning_rate": 0.0008549946364751435,
      "loss": 3.5499,
      "step": 1183
    },
    {
      "epoch": 0.2714973629901399,
      "grad_norm": 0.13693610570859813,
      "learning_rate": 0.000854733031643929,
      "loss": 3.6302,
      "step": 1184
    },
    {
      "epoch": 0.27172666819536806,
      "grad_norm": 0.12996007417982947,
      "learning_rate": 0.0008544712311443475,
      "loss": 3.4858,
      "step": 1185
    },
    {
      "epoch": 0.2719559734005962,
      "grad_norm": 0.13758874606093688,
      "learning_rate": 0.0008542092351208057,
      "loss": 3.5811,
      "step": 1186
    },
    {
      "epoch": 0.27218527860582437,
      "grad_norm": 0.13244677682869782,
      "learning_rate": 0.0008539470437178196,
      "loss": 3.6105,
      "step": 1187
    },
    {
      "epoch": 0.2724145838110525,
      "grad_norm": 0.12397466215593247,
      "learning_rate": 0.0008536846570800117,
      "loss": 3.5,
      "step": 1188
    },
    {
      "epoch": 0.27264388901628067,
      "grad_norm": 0.1385912110835162,
      "learning_rate": 0.000853422075352113,
      "loss": 3.4765,
      "step": 1189
    },
    {
      "epoch": 0.27287319422150885,
      "grad_norm": 0.13013627652138016,
      "learning_rate": 0.0008531592986789616,
      "loss": 3.6297,
      "step": 1190
    },
    {
      "epoch": 0.273102499426737,
      "grad_norm": 0.14763977282186402,
      "learning_rate": 0.0008528963272055035,
      "loss": 3.5387,
      "step": 1191
    },
    {
      "epoch": 0.27333180463196516,
      "grad_norm": 0.14113846470170824,
      "learning_rate": 0.0008526331610767919,
      "loss": 3.5239,
      "step": 1192
    },
    {
      "epoch": 0.2735611098371933,
      "grad_norm": 0.1617349395491954,
      "learning_rate": 0.0008523698004379877,
      "loss": 3.6225,
      "step": 1193
    },
    {
      "epoch": 0.27379041504242146,
      "grad_norm": 0.1432420275983754,
      "learning_rate": 0.0008521062454343586,
      "loss": 3.4242,
      "step": 1194
    },
    {
      "epoch": 0.27401972024764965,
      "grad_norm": 0.14405621265787016,
      "learning_rate": 0.00085184249621128,
      "loss": 3.6321,
      "step": 1195
    },
    {
      "epoch": 0.27424902545287777,
      "grad_norm": 0.14653243659492196,
      "learning_rate": 0.0008515785529142339,
      "loss": 3.7004,
      "step": 1196
    },
    {
      "epoch": 0.27447833065810595,
      "grad_norm": 0.14093207238118866,
      "learning_rate": 0.0008513144156888101,
      "loss": 3.7042,
      "step": 1197
    },
    {
      "epoch": 0.2747076358633341,
      "grad_norm": 0.12272924039615037,
      "learning_rate": 0.0008510500846807045,
      "loss": 3.5859,
      "step": 1198
    },
    {
      "epoch": 0.27493694106856226,
      "grad_norm": 0.14520637760980948,
      "learning_rate": 0.0008507855600357207,
      "loss": 3.5725,
      "step": 1199
    },
    {
      "epoch": 0.27516624627379044,
      "grad_norm": 0.12489966281265422,
      "learning_rate": 0.0008505208418997684,
      "loss": 3.4994,
      "step": 1200
    },
    {
      "epoch": 0.27539555147901856,
      "grad_norm": 0.11108146419252836,
      "learning_rate": 0.0008502559304188644,
      "loss": 3.5908,
      "step": 1201
    },
    {
      "epoch": 0.27562485668424674,
      "grad_norm": 0.13474267072286575,
      "learning_rate": 0.0008499908257391324,
      "loss": 3.4931,
      "step": 1202
    },
    {
      "epoch": 0.27585416188947487,
      "grad_norm": 0.128944177301228,
      "learning_rate": 0.0008497255280068019,
      "loss": 3.5922,
      "step": 1203
    },
    {
      "epoch": 0.27608346709470305,
      "grad_norm": 0.14539289052348553,
      "learning_rate": 0.0008494600373682098,
      "loss": 3.5674,
      "step": 1204
    },
    {
      "epoch": 0.27631277229993123,
      "grad_norm": 0.13188261958218817,
      "learning_rate": 0.0008491943539697986,
      "loss": 3.4414,
      "step": 1205
    },
    {
      "epoch": 0.27654207750515936,
      "grad_norm": 0.1355920828983346,
      "learning_rate": 0.0008489284779581177,
      "loss": 3.4782,
      "step": 1206
    },
    {
      "epoch": 0.27677138271038754,
      "grad_norm": 0.14200450080235066,
      "learning_rate": 0.0008486624094798226,
      "loss": 3.5893,
      "step": 1207
    },
    {
      "epoch": 0.27700068791561566,
      "grad_norm": 0.13420251664155827,
      "learning_rate": 0.0008483961486816747,
      "loss": 3.557,
      "step": 1208
    },
    {
      "epoch": 0.27722999312084384,
      "grad_norm": 0.1425066280519148,
      "learning_rate": 0.0008481296957105417,
      "loss": 3.7042,
      "step": 1209
    },
    {
      "epoch": 0.277459298326072,
      "grad_norm": 0.14411566712749593,
      "learning_rate": 0.0008478630507133973,
      "loss": 3.5828,
      "step": 1210
    },
    {
      "epoch": 0.27768860353130015,
      "grad_norm": 0.15708206029900654,
      "learning_rate": 0.0008475962138373213,
      "loss": 3.5026,
      "step": 1211
    },
    {
      "epoch": 0.27791790873652833,
      "grad_norm": 0.13432930585561983,
      "learning_rate": 0.0008473291852294987,
      "loss": 3.4734,
      "step": 1212
    },
    {
      "epoch": 0.27814721394175645,
      "grad_norm": 0.15026756287581833,
      "learning_rate": 0.0008470619650372211,
      "loss": 3.455,
      "step": 1213
    },
    {
      "epoch": 0.27837651914698464,
      "grad_norm": 0.14159245892538802,
      "learning_rate": 0.0008467945534078852,
      "loss": 3.6047,
      "step": 1214
    },
    {
      "epoch": 0.2786058243522128,
      "grad_norm": 0.14260975877477697,
      "learning_rate": 0.0008465269504889934,
      "loss": 3.5997,
      "step": 1215
    },
    {
      "epoch": 0.27883512955744094,
      "grad_norm": 0.15520340165123664,
      "learning_rate": 0.0008462591564281539,
      "loss": 3.5834,
      "step": 1216
    },
    {
      "epoch": 0.2790644347626691,
      "grad_norm": 0.15039743732178548,
      "learning_rate": 0.0008459911713730799,
      "loss": 3.5889,
      "step": 1217
    },
    {
      "epoch": 0.27929373996789725,
      "grad_norm": 0.1312069176717689,
      "learning_rate": 0.0008457229954715902,
      "loss": 3.5705,
      "step": 1218
    },
    {
      "epoch": 0.27952304517312543,
      "grad_norm": 0.12988778048115093,
      "learning_rate": 0.0008454546288716089,
      "loss": 3.6697,
      "step": 1219
    },
    {
      "epoch": 0.2797523503783536,
      "grad_norm": 0.12456961593039081,
      "learning_rate": 0.0008451860717211652,
      "loss": 3.685,
      "step": 1220
    },
    {
      "epoch": 0.27998165558358173,
      "grad_norm": 0.13538505361226735,
      "learning_rate": 0.0008449173241683935,
      "loss": 3.6388,
      "step": 1221
    },
    {
      "epoch": 0.2802109607888099,
      "grad_norm": 0.12022812039602293,
      "learning_rate": 0.0008446483863615329,
      "loss": 3.4853,
      "step": 1222
    },
    {
      "epoch": 0.28044026599403804,
      "grad_norm": 0.14317388350656618,
      "learning_rate": 0.0008443792584489281,
      "loss": 3.5296,
      "step": 1223
    },
    {
      "epoch": 0.2806695711992662,
      "grad_norm": 0.13409811190296458,
      "learning_rate": 0.0008441099405790278,
      "loss": 3.5504,
      "step": 1224
    },
    {
      "epoch": 0.2808988764044944,
      "grad_norm": 0.13325169055056668,
      "learning_rate": 0.0008438404329003863,
      "loss": 3.59,
      "step": 1225
    },
    {
      "epoch": 0.2811281816097225,
      "grad_norm": 0.13360681098036917,
      "learning_rate": 0.000843570735561662,
      "loss": 3.5262,
      "step": 1226
    },
    {
      "epoch": 0.2813574868149507,
      "grad_norm": 0.14613970165941606,
      "learning_rate": 0.0008433008487116183,
      "loss": 3.7775,
      "step": 1227
    },
    {
      "epoch": 0.28158679202017883,
      "grad_norm": 0.15164625533946693,
      "learning_rate": 0.000843030772499123,
      "loss": 3.5808,
      "step": 1228
    },
    {
      "epoch": 0.281816097225407,
      "grad_norm": 0.12878552384394246,
      "learning_rate": 0.0008427605070731481,
      "loss": 3.4552,
      "step": 1229
    },
    {
      "epoch": 0.2820454024306352,
      "grad_norm": 0.1376737332933127,
      "learning_rate": 0.0008424900525827706,
      "loss": 3.4393,
      "step": 1230
    },
    {
      "epoch": 0.2822747076358633,
      "grad_norm": 0.1489429944935281,
      "learning_rate": 0.0008422194091771708,
      "loss": 3.6534,
      "step": 1231
    },
    {
      "epoch": 0.2825040128410915,
      "grad_norm": 0.13422649912294282,
      "learning_rate": 0.0008419485770056344,
      "loss": 3.6416,
      "step": 1232
    },
    {
      "epoch": 0.2827333180463196,
      "grad_norm": 0.1317339989075956,
      "learning_rate": 0.0008416775562175503,
      "loss": 3.4557,
      "step": 1233
    },
    {
      "epoch": 0.2829626232515478,
      "grad_norm": 0.1326078775873507,
      "learning_rate": 0.0008414063469624118,
      "loss": 3.3836,
      "step": 1234
    },
    {
      "epoch": 0.283191928456776,
      "grad_norm": 0.1388811446852851,
      "learning_rate": 0.000841134949389816,
      "loss": 3.7153,
      "step": 1235
    },
    {
      "epoch": 0.2834212336620041,
      "grad_norm": 0.12779276262266465,
      "learning_rate": 0.0008408633636494642,
      "loss": 3.4844,
      "step": 1236
    },
    {
      "epoch": 0.2836505388672323,
      "grad_norm": 0.13130876372319644,
      "learning_rate": 0.0008405915898911611,
      "loss": 3.657,
      "step": 1237
    },
    {
      "epoch": 0.2838798440724604,
      "grad_norm": 0.13077294354046182,
      "learning_rate": 0.0008403196282648156,
      "loss": 3.4492,
      "step": 1238
    },
    {
      "epoch": 0.2841091492776886,
      "grad_norm": 0.12309338244875954,
      "learning_rate": 0.0008400474789204396,
      "loss": 3.4389,
      "step": 1239
    },
    {
      "epoch": 0.2843384544829168,
      "grad_norm": 0.13610760389478255,
      "learning_rate": 0.0008397751420081488,
      "loss": 3.5068,
      "step": 1240
    },
    {
      "epoch": 0.2845677596881449,
      "grad_norm": 0.12276614936954895,
      "learning_rate": 0.0008395026176781626,
      "loss": 3.4806,
      "step": 1241
    },
    {
      "epoch": 0.2847970648933731,
      "grad_norm": 0.12919760351875498,
      "learning_rate": 0.0008392299060808035,
      "loss": 3.575,
      "step": 1242
    },
    {
      "epoch": 0.2850263700986012,
      "grad_norm": 0.12968867346493362,
      "learning_rate": 0.0008389570073664976,
      "loss": 3.5974,
      "step": 1243
    },
    {
      "epoch": 0.2852556753038294,
      "grad_norm": 0.13740394562251765,
      "learning_rate": 0.0008386839216857738,
      "loss": 3.567,
      "step": 1244
    },
    {
      "epoch": 0.2854849805090576,
      "grad_norm": 0.1495971632343388,
      "learning_rate": 0.0008384106491892642,
      "loss": 3.3729,
      "step": 1245
    },
    {
      "epoch": 0.2857142857142857,
      "grad_norm": 0.16045283628782733,
      "learning_rate": 0.0008381371900277045,
      "loss": 3.5409,
      "step": 1246
    },
    {
      "epoch": 0.2859435909195139,
      "grad_norm": 0.14108882302716066,
      "learning_rate": 0.0008378635443519327,
      "loss": 3.4426,
      "step": 1247
    },
    {
      "epoch": 0.286172896124742,
      "grad_norm": 0.16168864090428026,
      "learning_rate": 0.0008375897123128897,
      "loss": 3.6625,
      "step": 1248
    },
    {
      "epoch": 0.2864022013299702,
      "grad_norm": 0.12831481006276946,
      "learning_rate": 0.0008373156940616199,
      "loss": 3.5165,
      "step": 1249
    },
    {
      "epoch": 0.28663150653519837,
      "grad_norm": 0.13306165319419205,
      "learning_rate": 0.0008370414897492696,
      "loss": 3.5516,
      "step": 1250
    },
    {
      "epoch": 0.2868608117404265,
      "grad_norm": 0.14168523914986836,
      "learning_rate": 0.0008367670995270882,
      "loss": 3.5494,
      "step": 1251
    },
    {
      "epoch": 0.28709011694565467,
      "grad_norm": 0.12253913004413976,
      "learning_rate": 0.0008364925235464274,
      "loss": 3.6372,
      "step": 1252
    },
    {
      "epoch": 0.28731942215088285,
      "grad_norm": 0.11254237450599903,
      "learning_rate": 0.0008362177619587416,
      "loss": 3.4109,
      "step": 1253
    },
    {
      "epoch": 0.287548727356111,
      "grad_norm": 0.12971634329302262,
      "learning_rate": 0.0008359428149155875,
      "loss": 3.6073,
      "step": 1254
    },
    {
      "epoch": 0.28777803256133916,
      "grad_norm": 0.13206277877514946,
      "learning_rate": 0.0008356676825686238,
      "loss": 3.4895,
      "step": 1255
    },
    {
      "epoch": 0.2880073377665673,
      "grad_norm": 0.14551388668462314,
      "learning_rate": 0.0008353923650696118,
      "loss": 3.6412,
      "step": 1256
    },
    {
      "epoch": 0.28823664297179546,
      "grad_norm": 0.1251768315331934,
      "learning_rate": 0.0008351168625704147,
      "loss": 3.6137,
      "step": 1257
    },
    {
      "epoch": 0.28846594817702365,
      "grad_norm": 0.12203960686778809,
      "learning_rate": 0.0008348411752229982,
      "loss": 3.526,
      "step": 1258
    },
    {
      "epoch": 0.28869525338225177,
      "grad_norm": 0.1293969022838271,
      "learning_rate": 0.0008345653031794292,
      "loss": 3.6197,
      "step": 1259
    },
    {
      "epoch": 0.28892455858747995,
      "grad_norm": 0.13645392998967307,
      "learning_rate": 0.000834289246591877,
      "loss": 3.5723,
      "step": 1260
    },
    {
      "epoch": 0.2891538637927081,
      "grad_norm": 0.1435258699179275,
      "learning_rate": 0.0008340130056126125,
      "loss": 3.646,
      "step": 1261
    },
    {
      "epoch": 0.28938316899793626,
      "grad_norm": 0.13544093511727626,
      "learning_rate": 0.0008337365803940084,
      "loss": 3.5208,
      "step": 1262
    },
    {
      "epoch": 0.28961247420316444,
      "grad_norm": 0.12722456234349372,
      "learning_rate": 0.0008334599710885394,
      "loss": 3.5463,
      "step": 1263
    },
    {
      "epoch": 0.28984177940839256,
      "grad_norm": 0.14297260823845293,
      "learning_rate": 0.0008331831778487807,
      "loss": 3.6968,
      "step": 1264
    },
    {
      "epoch": 0.29007108461362074,
      "grad_norm": 0.13653023773776776,
      "learning_rate": 0.0008329062008274098,
      "loss": 3.5393,
      "step": 1265
    },
    {
      "epoch": 0.29030038981884887,
      "grad_norm": 0.13629118668721105,
      "learning_rate": 0.0008326290401772056,
      "loss": 3.5162,
      "step": 1266
    },
    {
      "epoch": 0.29052969502407705,
      "grad_norm": 0.14340079662181449,
      "learning_rate": 0.000832351696051048,
      "loss": 3.5551,
      "step": 1267
    },
    {
      "epoch": 0.29075900022930523,
      "grad_norm": 0.12249235803431972,
      "learning_rate": 0.0008320741686019179,
      "loss": 3.6231,
      "step": 1268
    },
    {
      "epoch": 0.29098830543453336,
      "grad_norm": 0.1563367716874775,
      "learning_rate": 0.000831796457982898,
      "loss": 3.699,
      "step": 1269
    },
    {
      "epoch": 0.29121761063976154,
      "grad_norm": 0.12999024379725546,
      "learning_rate": 0.0008315185643471713,
      "loss": 3.5609,
      "step": 1270
    },
    {
      "epoch": 0.29144691584498966,
      "grad_norm": 0.12698156702901098,
      "learning_rate": 0.0008312404878480222,
      "loss": 3.5568,
      "step": 1271
    },
    {
      "epoch": 0.29167622105021784,
      "grad_norm": 0.13713529479977715,
      "learning_rate": 0.0008309622286388356,
      "loss": 3.6103,
      "step": 1272
    },
    {
      "epoch": 0.291905526255446,
      "grad_norm": 0.13886568542012168,
      "learning_rate": 0.0008306837868730979,
      "loss": 3.491,
      "step": 1273
    },
    {
      "epoch": 0.29213483146067415,
      "grad_norm": 0.1631183509969543,
      "learning_rate": 0.0008304051627043953,
      "loss": 3.6014,
      "step": 1274
    },
    {
      "epoch": 0.29236413666590233,
      "grad_norm": 0.13063502334225044,
      "learning_rate": 0.0008301263562864152,
      "loss": 3.5079,
      "step": 1275
    },
    {
      "epoch": 0.29259344187113046,
      "grad_norm": 0.1239411134817858,
      "learning_rate": 0.0008298473677729452,
      "loss": 3.4133,
      "step": 1276
    },
    {
      "epoch": 0.29282274707635864,
      "grad_norm": 0.14836048441897107,
      "learning_rate": 0.0008295681973178737,
      "loss": 3.5125,
      "step": 1277
    },
    {
      "epoch": 0.2930520522815868,
      "grad_norm": 0.15990842077255105,
      "learning_rate": 0.0008292888450751892,
      "loss": 3.5696,
      "step": 1278
    },
    {
      "epoch": 0.29328135748681494,
      "grad_norm": 0.13438848848554102,
      "learning_rate": 0.0008290093111989804,
      "loss": 3.5937,
      "step": 1279
    },
    {
      "epoch": 0.2935106626920431,
      "grad_norm": 0.15360791160071777,
      "learning_rate": 0.0008287295958434364,
      "loss": 3.6559,
      "step": 1280
    },
    {
      "epoch": 0.29373996789727125,
      "grad_norm": 0.11840879486671224,
      "learning_rate": 0.0008284496991628465,
      "loss": 3.5815,
      "step": 1281
    },
    {
      "epoch": 0.29396927310249943,
      "grad_norm": 0.1233452837569966,
      "learning_rate": 0.0008281696213115995,
      "loss": 3.572,
      "step": 1282
    },
    {
      "epoch": 0.2941985783077276,
      "grad_norm": 0.12216957901875583,
      "learning_rate": 0.0008278893624441847,
      "loss": 3.492,
      "step": 1283
    },
    {
      "epoch": 0.29442788351295573,
      "grad_norm": 0.13316148754044893,
      "learning_rate": 0.000827608922715191,
      "loss": 3.3836,
      "step": 1284
    },
    {
      "epoch": 0.2946571887181839,
      "grad_norm": 0.11690675545369718,
      "learning_rate": 0.000827328302279307,
      "loss": 3.4479,
      "step": 1285
    },
    {
      "epoch": 0.29488649392341204,
      "grad_norm": 0.141552419602808,
      "learning_rate": 0.0008270475012913211,
      "loss": 3.6652,
      "step": 1286
    },
    {
      "epoch": 0.2951157991286402,
      "grad_norm": 0.14246792540286474,
      "learning_rate": 0.0008267665199061211,
      "loss": 3.4162,
      "step": 1287
    },
    {
      "epoch": 0.2953451043338684,
      "grad_norm": 0.13724298866254117,
      "learning_rate": 0.0008264853582786949,
      "loss": 3.333,
      "step": 1288
    },
    {
      "epoch": 0.2955744095390965,
      "grad_norm": 0.14044742805437255,
      "learning_rate": 0.0008262040165641288,
      "loss": 3.6381,
      "step": 1289
    },
    {
      "epoch": 0.2958037147443247,
      "grad_norm": 0.13507960856841916,
      "learning_rate": 0.0008259224949176094,
      "loss": 3.4996,
      "step": 1290
    },
    {
      "epoch": 0.29603301994955283,
      "grad_norm": 0.13515534576122032,
      "learning_rate": 0.0008256407934944219,
      "loss": 3.5092,
      "step": 1291
    },
    {
      "epoch": 0.296262325154781,
      "grad_norm": 0.1301028174309254,
      "learning_rate": 0.0008253589124499512,
      "loss": 3.5692,
      "step": 1292
    },
    {
      "epoch": 0.2964916303600092,
      "grad_norm": 0.13945262371057202,
      "learning_rate": 0.0008250768519396807,
      "loss": 3.6337,
      "step": 1293
    },
    {
      "epoch": 0.2967209355652373,
      "grad_norm": 0.14062886335661148,
      "learning_rate": 0.0008247946121191933,
      "loss": 3.4902,
      "step": 1294
    },
    {
      "epoch": 0.2969502407704655,
      "grad_norm": 0.1301356957356634,
      "learning_rate": 0.0008245121931441706,
      "loss": 3.4271,
      "step": 1295
    },
    {
      "epoch": 0.2971795459756936,
      "grad_norm": 0.14039174294490553,
      "learning_rate": 0.0008242295951703929,
      "loss": 3.6519,
      "step": 1296
    },
    {
      "epoch": 0.2974088511809218,
      "grad_norm": 0.14144107755335933,
      "learning_rate": 0.0008239468183537393,
      "loss": 3.5221,
      "step": 1297
    },
    {
      "epoch": 0.29763815638615,
      "grad_norm": 0.1245184840084144,
      "learning_rate": 0.0008236638628501878,
      "loss": 3.5659,
      "step": 1298
    },
    {
      "epoch": 0.2978674615913781,
      "grad_norm": 0.15226843921463387,
      "learning_rate": 0.0008233807288158146,
      "loss": 3.5627,
      "step": 1299
    },
    {
      "epoch": 0.2980967667966063,
      "grad_norm": 0.12529700945683456,
      "learning_rate": 0.0008230974164067949,
      "loss": 3.4708,
      "step": 1300
    },
    {
      "epoch": 0.2983260720018344,
      "grad_norm": 0.124139185388391,
      "learning_rate": 0.0008228139257794012,
      "loss": 3.4096,
      "step": 1301
    },
    {
      "epoch": 0.2985553772070626,
      "grad_norm": 0.1510339469459239,
      "learning_rate": 0.0008225302570900058,
      "loss": 3.7229,
      "step": 1302
    },
    {
      "epoch": 0.2987846824122908,
      "grad_norm": 0.14570721022193314,
      "learning_rate": 0.0008222464104950778,
      "loss": 3.5532,
      "step": 1303
    },
    {
      "epoch": 0.2990139876175189,
      "grad_norm": 0.14479286422456883,
      "learning_rate": 0.0008219623861511855,
      "loss": 3.5887,
      "step": 1304
    },
    {
      "epoch": 0.2992432928227471,
      "grad_norm": 0.14825466147872876,
      "learning_rate": 0.000821678184214995,
      "loss": 3.6208,
      "step": 1305
    },
    {
      "epoch": 0.2994725980279752,
      "grad_norm": 0.1480829091475768,
      "learning_rate": 0.0008213938048432696,
      "loss": 3.4997,
      "step": 1306
    },
    {
      "epoch": 0.2997019032332034,
      "grad_norm": 0.13587709107961632,
      "learning_rate": 0.0008211092481928716,
      "loss": 3.5903,
      "step": 1307
    },
    {
      "epoch": 0.2999312084384316,
      "grad_norm": 0.14376612907697767,
      "learning_rate": 0.0008208245144207603,
      "loss": 3.4967,
      "step": 1308
    },
    {
      "epoch": 0.3001605136436597,
      "grad_norm": 0.13744049769459377,
      "learning_rate": 0.0008205396036839927,
      "loss": 3.5738,
      "step": 1309
    },
    {
      "epoch": 0.3003898188488879,
      "grad_norm": 0.14119221528130915,
      "learning_rate": 0.0008202545161397241,
      "loss": 3.658,
      "step": 1310
    },
    {
      "epoch": 0.300619124054116,
      "grad_norm": 0.135818378502723,
      "learning_rate": 0.0008199692519452069,
      "loss": 3.5404,
      "step": 1311
    },
    {
      "epoch": 0.3008484292593442,
      "grad_norm": 0.12842515080526173,
      "learning_rate": 0.0008196838112577904,
      "loss": 3.5641,
      "step": 1312
    },
    {
      "epoch": 0.30107773446457237,
      "grad_norm": 0.13007207672408883,
      "learning_rate": 0.0008193981942349224,
      "loss": 3.5311,
      "step": 1313
    },
    {
      "epoch": 0.3013070396698005,
      "grad_norm": 0.13813775116340074,
      "learning_rate": 0.0008191124010341468,
      "loss": 3.5557,
      "step": 1314
    },
    {
      "epoch": 0.3015363448750287,
      "grad_norm": 0.15475000145335055,
      "learning_rate": 0.0008188264318131056,
      "loss": 3.5329,
      "step": 1315
    },
    {
      "epoch": 0.3017656500802568,
      "grad_norm": 0.13410196006449032,
      "learning_rate": 0.0008185402867295373,
      "loss": 3.5865,
      "step": 1316
    },
    {
      "epoch": 0.301994955285485,
      "grad_norm": 0.12992873361543236,
      "learning_rate": 0.0008182539659412776,
      "loss": 3.5346,
      "step": 1317
    },
    {
      "epoch": 0.30222426049071316,
      "grad_norm": 0.1351506643109155,
      "learning_rate": 0.0008179674696062593,
      "loss": 3.4542,
      "step": 1318
    },
    {
      "epoch": 0.3024535656959413,
      "grad_norm": 0.1269261643792217,
      "learning_rate": 0.0008176807978825118,
      "loss": 3.4397,
      "step": 1319
    },
    {
      "epoch": 0.30268287090116947,
      "grad_norm": 0.13554636624206814,
      "learning_rate": 0.0008173939509281613,
      "loss": 3.455,
      "step": 1320
    },
    {
      "epoch": 0.3029121761063976,
      "grad_norm": 0.1548225391288222,
      "learning_rate": 0.0008171069289014306,
      "loss": 3.4737,
      "step": 1321
    },
    {
      "epoch": 0.30314148131162577,
      "grad_norm": 0.15579917867963694,
      "learning_rate": 0.0008168197319606392,
      "loss": 3.68,
      "step": 1322
    },
    {
      "epoch": 0.30337078651685395,
      "grad_norm": 0.12891642754342134,
      "learning_rate": 0.0008165323602642028,
      "loss": 3.6566,
      "step": 1323
    },
    {
      "epoch": 0.3036000917220821,
      "grad_norm": 0.14507023688672555,
      "learning_rate": 0.0008162448139706341,
      "loss": 3.6092,
      "step": 1324
    },
    {
      "epoch": 0.30382939692731026,
      "grad_norm": 0.12799900706628714,
      "learning_rate": 0.0008159570932385414,
      "loss": 3.5455,
      "step": 1325
    },
    {
      "epoch": 0.3040587021325384,
      "grad_norm": 0.14720690444950685,
      "learning_rate": 0.0008156691982266298,
      "loss": 3.4153,
      "step": 1326
    },
    {
      "epoch": 0.30428800733776656,
      "grad_norm": 0.13100182631158244,
      "learning_rate": 0.0008153811290936999,
      "loss": 3.4147,
      "step": 1327
    },
    {
      "epoch": 0.30451731254299474,
      "grad_norm": 0.12742229801149813,
      "learning_rate": 0.0008150928859986488,
      "loss": 3.5557,
      "step": 1328
    },
    {
      "epoch": 0.30474661774822287,
      "grad_norm": 0.13606329153374852,
      "learning_rate": 0.0008148044691004698,
      "loss": 3.5667,
      "step": 1329
    },
    {
      "epoch": 0.30497592295345105,
      "grad_norm": 0.1509675456697617,
      "learning_rate": 0.0008145158785582514,
      "loss": 3.4993,
      "step": 1330
    },
    {
      "epoch": 0.3052052281586792,
      "grad_norm": 0.13233064563796498,
      "learning_rate": 0.0008142271145311783,
      "loss": 3.4922,
      "step": 1331
    },
    {
      "epoch": 0.30543453336390736,
      "grad_norm": 0.1273915385287744,
      "learning_rate": 0.0008139381771785308,
      "loss": 3.5193,
      "step": 1332
    },
    {
      "epoch": 0.30566383856913554,
      "grad_norm": 0.1404283772120039,
      "learning_rate": 0.000813649066659685,
      "loss": 3.5256,
      "step": 1333
    },
    {
      "epoch": 0.30589314377436366,
      "grad_norm": 0.13861130265342478,
      "learning_rate": 0.000813359783134112,
      "loss": 3.6056,
      "step": 1334
    },
    {
      "epoch": 0.30612244897959184,
      "grad_norm": 0.18163867232505154,
      "learning_rate": 0.0008130703267613787,
      "loss": 3.6384,
      "step": 1335
    },
    {
      "epoch": 0.30635175418481997,
      "grad_norm": 0.1634286189174257,
      "learning_rate": 0.0008127806977011474,
      "loss": 3.555,
      "step": 1336
    },
    {
      "epoch": 0.30658105939004815,
      "grad_norm": 0.13801663683816917,
      "learning_rate": 0.0008124908961131759,
      "loss": 3.5794,
      "step": 1337
    },
    {
      "epoch": 0.30681036459527633,
      "grad_norm": 0.13216575288445223,
      "learning_rate": 0.0008122009221573161,
      "loss": 3.5286,
      "step": 1338
    },
    {
      "epoch": 0.30703966980050446,
      "grad_norm": 0.1256565220487945,
      "learning_rate": 0.0008119107759935163,
      "loss": 3.4363,
      "step": 1339
    },
    {
      "epoch": 0.30726897500573264,
      "grad_norm": 0.13723359310476693,
      "learning_rate": 0.000811620457781819,
      "loss": 3.4374,
      "step": 1340
    },
    {
      "epoch": 0.30749828021096076,
      "grad_norm": 0.1454388833751865,
      "learning_rate": 0.0008113299676823615,
      "loss": 3.5174,
      "step": 1341
    },
    {
      "epoch": 0.30772758541618894,
      "grad_norm": 0.13990128945048685,
      "learning_rate": 0.0008110393058553765,
      "loss": 3.579,
      "step": 1342
    },
    {
      "epoch": 0.3079568906214171,
      "grad_norm": 0.16024442027057062,
      "learning_rate": 0.0008107484724611911,
      "loss": 3.6182,
      "step": 1343
    },
    {
      "epoch": 0.30818619582664525,
      "grad_norm": 0.1447268442315348,
      "learning_rate": 0.000810457467660227,
      "loss": 3.6132,
      "step": 1344
    },
    {
      "epoch": 0.30841550103187343,
      "grad_norm": 0.14375893994831543,
      "learning_rate": 0.0008101662916130006,
      "loss": 3.668,
      "step": 1345
    },
    {
      "epoch": 0.3086448062371016,
      "grad_norm": 0.12486632920994109,
      "learning_rate": 0.0008098749444801224,
      "loss": 3.4929,
      "step": 1346
    },
    {
      "epoch": 0.30887411144232974,
      "grad_norm": 0.1304702168752577,
      "learning_rate": 0.0008095834264222979,
      "loss": 3.4248,
      "step": 1347
    },
    {
      "epoch": 0.3091034166475579,
      "grad_norm": 0.12636156971860768,
      "learning_rate": 0.0008092917376003262,
      "loss": 3.5815,
      "step": 1348
    },
    {
      "epoch": 0.30933272185278604,
      "grad_norm": 0.11390642437328163,
      "learning_rate": 0.0008089998781751009,
      "loss": 3.5943,
      "step": 1349
    },
    {
      "epoch": 0.3095620270580142,
      "grad_norm": 0.12468146644170872,
      "learning_rate": 0.00080870784830761,
      "loss": 3.6127,
      "step": 1350
    },
    {
      "epoch": 0.3097913322632424,
      "grad_norm": 0.12273938832524733,
      "learning_rate": 0.0008084156481589349,
      "loss": 3.4631,
      "step": 1351
    },
    {
      "epoch": 0.31002063746847053,
      "grad_norm": 0.12896921980195955,
      "learning_rate": 0.0008081232778902516,
      "loss": 3.5902,
      "step": 1352
    },
    {
      "epoch": 0.3102499426736987,
      "grad_norm": 0.11519108981046823,
      "learning_rate": 0.0008078307376628291,
      "loss": 3.5904,
      "step": 1353
    },
    {
      "epoch": 0.31047924787892683,
      "grad_norm": 0.13044257116206517,
      "learning_rate": 0.0008075380276380311,
      "loss": 3.689,
      "step": 1354
    },
    {
      "epoch": 0.310708553084155,
      "grad_norm": 0.11957297388465317,
      "learning_rate": 0.0008072451479773143,
      "loss": 3.3935,
      "step": 1355
    },
    {
      "epoch": 0.3109378582893832,
      "grad_norm": 0.14478327325872486,
      "learning_rate": 0.0008069520988422291,
      "loss": 3.3977,
      "step": 1356
    },
    {
      "epoch": 0.3111671634946113,
      "grad_norm": 0.13565351520601865,
      "learning_rate": 0.0008066588803944195,
      "loss": 3.6025,
      "step": 1357
    },
    {
      "epoch": 0.3113964686998395,
      "grad_norm": 0.13624720607518656,
      "learning_rate": 0.0008063654927956227,
      "loss": 3.664,
      "step": 1358
    },
    {
      "epoch": 0.3116257739050676,
      "grad_norm": 0.1543827679740697,
      "learning_rate": 0.0008060719362076697,
      "loss": 3.4416,
      "step": 1359
    },
    {
      "epoch": 0.3118550791102958,
      "grad_norm": 0.13261468809879648,
      "learning_rate": 0.000805778210792484,
      "loss": 3.6449,
      "step": 1360
    },
    {
      "epoch": 0.312084384315524,
      "grad_norm": 0.12838147470915393,
      "learning_rate": 0.0008054843167120826,
      "loss": 3.6496,
      "step": 1361
    },
    {
      "epoch": 0.3123136895207521,
      "grad_norm": 0.12179898362545727,
      "learning_rate": 0.0008051902541285758,
      "loss": 3.5225,
      "step": 1362
    },
    {
      "epoch": 0.3125429947259803,
      "grad_norm": 0.14176043825650317,
      "learning_rate": 0.0008048960232041663,
      "loss": 3.4707,
      "step": 1363
    },
    {
      "epoch": 0.3127722999312084,
      "grad_norm": 0.126177280384065,
      "learning_rate": 0.00080460162410115,
      "loss": 3.575,
      "step": 1364
    },
    {
      "epoch": 0.3130016051364366,
      "grad_norm": 0.13602228212554243,
      "learning_rate": 0.0008043070569819153,
      "loss": 3.5306,
      "step": 1365
    },
    {
      "epoch": 0.3132309103416648,
      "grad_norm": 0.1320789076450651,
      "learning_rate": 0.0008040123220089437,
      "loss": 3.5628,
      "step": 1366
    },
    {
      "epoch": 0.3134602155468929,
      "grad_norm": 0.149371717266157,
      "learning_rate": 0.0008037174193448089,
      "loss": 3.6262,
      "step": 1367
    },
    {
      "epoch": 0.3136895207521211,
      "grad_norm": 0.1388075261807259,
      "learning_rate": 0.0008034223491521775,
      "loss": 3.4414,
      "step": 1368
    },
    {
      "epoch": 0.3139188259573492,
      "grad_norm": 0.1348933085648744,
      "learning_rate": 0.0008031271115938077,
      "loss": 3.6324,
      "step": 1369
    },
    {
      "epoch": 0.3141481311625774,
      "grad_norm": 0.13702158373333248,
      "learning_rate": 0.000802831706832551,
      "loss": 3.3918,
      "step": 1370
    },
    {
      "epoch": 0.3143774363678056,
      "grad_norm": 0.14936099048027968,
      "learning_rate": 0.0008025361350313505,
      "loss": 3.4284,
      "step": 1371
    },
    {
      "epoch": 0.3146067415730337,
      "grad_norm": 0.13571589814855384,
      "learning_rate": 0.0008022403963532419,
      "loss": 3.4325,
      "step": 1372
    },
    {
      "epoch": 0.3148360467782619,
      "grad_norm": 0.16054838291375,
      "learning_rate": 0.0008019444909613523,
      "loss": 3.5973,
      "step": 1373
    },
    {
      "epoch": 0.31506535198349,
      "grad_norm": 0.15426017165239791,
      "learning_rate": 0.0008016484190189012,
      "loss": 3.8576,
      "step": 1374
    },
    {
      "epoch": 0.3152946571887182,
      "grad_norm": 0.1353789393557544,
      "learning_rate": 0.0008013521806892003,
      "loss": 3.625,
      "step": 1375
    },
    {
      "epoch": 0.31552396239394637,
      "grad_norm": 0.15223363454694258,
      "learning_rate": 0.0008010557761356522,
      "loss": 3.647,
      "step": 1376
    },
    {
      "epoch": 0.3157532675991745,
      "grad_norm": 0.1596806306109384,
      "learning_rate": 0.000800759205521752,
      "loss": 3.5029,
      "step": 1377
    },
    {
      "epoch": 0.3159825728044027,
      "grad_norm": 0.19453711720247197,
      "learning_rate": 0.0008004624690110859,
      "loss": 3.5258,
      "step": 1378
    },
    {
      "epoch": 0.3162118780096308,
      "grad_norm": 0.12547522211473985,
      "learning_rate": 0.0008001655667673318,
      "loss": 3.634,
      "step": 1379
    },
    {
      "epoch": 0.316441183214859,
      "grad_norm": 0.13830800252096684,
      "learning_rate": 0.0007998684989542588,
      "loss": 3.7134,
      "step": 1380
    },
    {
      "epoch": 0.31667048842008716,
      "grad_norm": 0.13450721849313335,
      "learning_rate": 0.0007995712657357279,
      "loss": 3.5072,
      "step": 1381
    },
    {
      "epoch": 0.3168997936253153,
      "grad_norm": 0.12519752545384494,
      "learning_rate": 0.0007992738672756908,
      "loss": 3.6609,
      "step": 1382
    },
    {
      "epoch": 0.31712909883054347,
      "grad_norm": 0.12714043338772643,
      "learning_rate": 0.0007989763037381904,
      "loss": 3.5543,
      "step": 1383
    },
    {
      "epoch": 0.3173584040357716,
      "grad_norm": 0.13237245910799128,
      "learning_rate": 0.0007986785752873609,
      "loss": 3.393,
      "step": 1384
    },
    {
      "epoch": 0.31758770924099977,
      "grad_norm": 0.11981030013863608,
      "learning_rate": 0.0007983806820874271,
      "loss": 3.5321,
      "step": 1385
    },
    {
      "epoch": 0.31781701444622795,
      "grad_norm": 0.1302292592680686,
      "learning_rate": 0.0007980826243027052,
      "loss": 3.5311,
      "step": 1386
    },
    {
      "epoch": 0.3180463196514561,
      "grad_norm": 0.134848110945448,
      "learning_rate": 0.0007977844020976016,
      "loss": 3.7314,
      "step": 1387
    },
    {
      "epoch": 0.31827562485668426,
      "grad_norm": 0.14011767737458558,
      "learning_rate": 0.0007974860156366139,
      "loss": 3.4896,
      "step": 1388
    },
    {
      "epoch": 0.3185049300619124,
      "grad_norm": 0.1437569877352732,
      "learning_rate": 0.00079718746508433,
      "loss": 3.5139,
      "step": 1389
    },
    {
      "epoch": 0.31873423526714056,
      "grad_norm": 0.13033919839611838,
      "learning_rate": 0.0007968887506054282,
      "loss": 3.5623,
      "step": 1390
    },
    {
      "epoch": 0.31896354047236875,
      "grad_norm": 0.13985745982875292,
      "learning_rate": 0.0007965898723646776,
      "loss": 3.6106,
      "step": 1391
    },
    {
      "epoch": 0.31919284567759687,
      "grad_norm": 0.14511896418279716,
      "learning_rate": 0.0007962908305269375,
      "loss": 3.4994,
      "step": 1392
    },
    {
      "epoch": 0.31942215088282505,
      "grad_norm": 0.16630771470118866,
      "learning_rate": 0.0007959916252571573,
      "loss": 3.585,
      "step": 1393
    },
    {
      "epoch": 0.3196514560880532,
      "grad_norm": 0.1485622988273296,
      "learning_rate": 0.0007956922567203766,
      "loss": 3.4911,
      "step": 1394
    },
    {
      "epoch": 0.31988076129328136,
      "grad_norm": 0.13868634011625902,
      "learning_rate": 0.000795392725081725,
      "loss": 3.5238,
      "step": 1395
    },
    {
      "epoch": 0.32011006649850954,
      "grad_norm": 0.134620303735499,
      "learning_rate": 0.0007950930305064223,
      "loss": 3.4422,
      "step": 1396
    },
    {
      "epoch": 0.32033937170373766,
      "grad_norm": 0.12459118878183424,
      "learning_rate": 0.000794793173159778,
      "loss": 3.5197,
      "step": 1397
    },
    {
      "epoch": 0.32056867690896584,
      "grad_norm": 0.14324452108479405,
      "learning_rate": 0.0007944931532071912,
      "loss": 3.6699,
      "step": 1398
    },
    {
      "epoch": 0.32079798211419397,
      "grad_norm": 0.145187686745591,
      "learning_rate": 0.0007941929708141513,
      "loss": 3.6427,
      "step": 1399
    },
    {
      "epoch": 0.32102728731942215,
      "grad_norm": 0.1372745330496257,
      "learning_rate": 0.0007938926261462366,
      "loss": 3.6035,
      "step": 1400
    },
    {
      "epoch": 0.32125659252465033,
      "grad_norm": 0.12991539821587664,
      "learning_rate": 0.0007935921193691153,
      "loss": 3.4702,
      "step": 1401
    },
    {
      "epoch": 0.32148589772987846,
      "grad_norm": 0.13923245620223715,
      "learning_rate": 0.000793291450648545,
      "loss": 3.6519,
      "step": 1402
    },
    {
      "epoch": 0.32171520293510664,
      "grad_norm": 0.14037524735631082,
      "learning_rate": 0.0007929906201503722,
      "loss": 3.5,
      "step": 1403
    },
    {
      "epoch": 0.32194450814033476,
      "grad_norm": 0.12878242337354667,
      "learning_rate": 0.0007926896280405334,
      "loss": 3.5918,
      "step": 1404
    },
    {
      "epoch": 0.32217381334556294,
      "grad_norm": 0.12861673522155892,
      "learning_rate": 0.0007923884744850536,
      "loss": 3.4952,
      "step": 1405
    },
    {
      "epoch": 0.3224031185507911,
      "grad_norm": 0.13586150940244507,
      "learning_rate": 0.0007920871596500472,
      "loss": 3.6014,
      "step": 1406
    },
    {
      "epoch": 0.32263242375601925,
      "grad_norm": 0.13662658275084072,
      "learning_rate": 0.0007917856837017176,
      "loss": 3.6362,
      "step": 1407
    },
    {
      "epoch": 0.32286172896124743,
      "grad_norm": 0.15141637417325537,
      "learning_rate": 0.0007914840468063566,
      "loss": 3.5169,
      "step": 1408
    },
    {
      "epoch": 0.32309103416647555,
      "grad_norm": 0.14025218888232313,
      "learning_rate": 0.0007911822491303452,
      "loss": 3.3979,
      "step": 1409
    },
    {
      "epoch": 0.32332033937170374,
      "grad_norm": 0.14203187925028155,
      "learning_rate": 0.0007908802908401532,
      "loss": 3.4229,
      "step": 1410
    },
    {
      "epoch": 0.3235496445769319,
      "grad_norm": 0.14492780139168468,
      "learning_rate": 0.0007905781721023382,
      "loss": 3.3877,
      "step": 1411
    },
    {
      "epoch": 0.32377894978216004,
      "grad_norm": 0.1437274533466006,
      "learning_rate": 0.0007902758930835477,
      "loss": 3.5999,
      "step": 1412
    },
    {
      "epoch": 0.3240082549873882,
      "grad_norm": 0.1260932533593268,
      "learning_rate": 0.000789973453950516,
      "loss": 3.4947,
      "step": 1413
    },
    {
      "epoch": 0.32423756019261635,
      "grad_norm": 0.13005088166280737,
      "learning_rate": 0.000789670854870067,
      "loss": 3.6089,
      "step": 1414
    },
    {
      "epoch": 0.32446686539784453,
      "grad_norm": 0.12122740159940056,
      "learning_rate": 0.000789368096009112,
      "loss": 3.4743,
      "step": 1415
    },
    {
      "epoch": 0.3246961706030727,
      "grad_norm": 0.12505216142580072,
      "learning_rate": 0.0007890651775346512,
      "loss": 3.4916,
      "step": 1416
    },
    {
      "epoch": 0.32492547580830083,
      "grad_norm": 0.1416586333449929,
      "learning_rate": 0.0007887620996137721,
      "loss": 3.4806,
      "step": 1417
    },
    {
      "epoch": 0.325154781013529,
      "grad_norm": 0.15835536232245606,
      "learning_rate": 0.0007884588624136504,
      "loss": 3.6078,
      "step": 1418
    },
    {
      "epoch": 0.32538408621875714,
      "grad_norm": 0.1285184842297297,
      "learning_rate": 0.0007881554661015497,
      "loss": 3.4873,
      "step": 1419
    },
    {
      "epoch": 0.3256133914239853,
      "grad_norm": 0.13582343485239745,
      "learning_rate": 0.0007878519108448218,
      "loss": 3.6862,
      "step": 1420
    },
    {
      "epoch": 0.3258426966292135,
      "grad_norm": 0.12780472372225513,
      "learning_rate": 0.0007875481968109051,
      "loss": 3.4349,
      "step": 1421
    },
    {
      "epoch": 0.3260720018344416,
      "grad_norm": 0.14929880137419996,
      "learning_rate": 0.0007872443241673269,
      "loss": 3.6338,
      "step": 1422
    },
    {
      "epoch": 0.3263013070396698,
      "grad_norm": 0.14169352030932636,
      "learning_rate": 0.0007869402930817007,
      "loss": 3.4016,
      "step": 1423
    },
    {
      "epoch": 0.32653061224489793,
      "grad_norm": 0.13106972384931484,
      "learning_rate": 0.0007866361037217284,
      "loss": 3.4552,
      "step": 1424
    },
    {
      "epoch": 0.3267599174501261,
      "grad_norm": 0.14269618564479014,
      "learning_rate": 0.0007863317562551987,
      "loss": 3.5687,
      "step": 1425
    },
    {
      "epoch": 0.3269892226553543,
      "grad_norm": 0.1374513769649523,
      "learning_rate": 0.0007860272508499876,
      "loss": 3.4214,
      "step": 1426
    },
    {
      "epoch": 0.3272185278605824,
      "grad_norm": 0.15426286370275316,
      "learning_rate": 0.0007857225876740584,
      "loss": 3.5827,
      "step": 1427
    },
    {
      "epoch": 0.3274478330658106,
      "grad_norm": 0.14334327601024244,
      "learning_rate": 0.0007854177668954609,
      "loss": 3.5095,
      "step": 1428
    },
    {
      "epoch": 0.3276771382710387,
      "grad_norm": 0.1296318531988706,
      "learning_rate": 0.0007851127886823327,
      "loss": 3.593,
      "step": 1429
    },
    {
      "epoch": 0.3279064434762669,
      "grad_norm": 0.12764324897730478,
      "learning_rate": 0.0007848076532028974,
      "loss": 3.7233,
      "step": 1430
    },
    {
      "epoch": 0.3281357486814951,
      "grad_norm": 0.13447001191296865,
      "learning_rate": 0.0007845023606254658,
      "loss": 3.6423,
      "step": 1431
    },
    {
      "epoch": 0.3283650538867232,
      "grad_norm": 0.13778467170490694,
      "learning_rate": 0.0007841969111184352,
      "loss": 3.5277,
      "step": 1432
    },
    {
      "epoch": 0.3285943590919514,
      "grad_norm": 0.12501854498524914,
      "learning_rate": 0.0007838913048502894,
      "loss": 3.5653,
      "step": 1433
    },
    {
      "epoch": 0.3288236642971795,
      "grad_norm": 0.1490063652247872,
      "learning_rate": 0.0007835855419895992,
      "loss": 3.5231,
      "step": 1434
    },
    {
      "epoch": 0.3290529695024077,
      "grad_norm": 0.1533671082735197,
      "learning_rate": 0.0007832796227050208,
      "loss": 3.6325,
      "step": 1435
    },
    {
      "epoch": 0.3292822747076359,
      "grad_norm": 0.14027971265611172,
      "learning_rate": 0.0007829735471652976,
      "loss": 3.5919,
      "step": 1436
    },
    {
      "epoch": 0.329511579912864,
      "grad_norm": 0.15067975695950006,
      "learning_rate": 0.0007826673155392587,
      "loss": 3.5379,
      "step": 1437
    },
    {
      "epoch": 0.3297408851180922,
      "grad_norm": 0.13361387879486808,
      "learning_rate": 0.0007823609279958194,
      "loss": 3.5133,
      "step": 1438
    },
    {
      "epoch": 0.32997019032332037,
      "grad_norm": 0.1393206490862222,
      "learning_rate": 0.000782054384703981,
      "loss": 3.5487,
      "step": 1439
    },
    {
      "epoch": 0.3301994955285485,
      "grad_norm": 0.14721145687396434,
      "learning_rate": 0.0007817476858328308,
      "loss": 3.4743,
      "step": 1440
    },
    {
      "epoch": 0.3304288007337767,
      "grad_norm": 0.14168135706997956,
      "learning_rate": 0.0007814408315515418,
      "loss": 3.5187,
      "step": 1441
    },
    {
      "epoch": 0.3306581059390048,
      "grad_norm": 0.14238982139531622,
      "learning_rate": 0.0007811338220293729,
      "loss": 3.5356,
      "step": 1442
    },
    {
      "epoch": 0.330887411144233,
      "grad_norm": 0.13068569745602,
      "learning_rate": 0.0007808266574356683,
      "loss": 3.4806,
      "step": 1443
    },
    {
      "epoch": 0.33111671634946116,
      "grad_norm": 0.1293500803135721,
      "learning_rate": 0.000780519337939858,
      "loss": 3.6362,
      "step": 1444
    },
    {
      "epoch": 0.3313460215546893,
      "grad_norm": 0.14805198486737442,
      "learning_rate": 0.0007802118637114573,
      "loss": 3.5343,
      "step": 1445
    },
    {
      "epoch": 0.33157532675991747,
      "grad_norm": 0.1406804628728433,
      "learning_rate": 0.0007799042349200671,
      "loss": 3.6556,
      "step": 1446
    },
    {
      "epoch": 0.3318046319651456,
      "grad_norm": 0.14128383744926776,
      "learning_rate": 0.0007795964517353734,
      "loss": 3.5255,
      "step": 1447
    },
    {
      "epoch": 0.33203393717037377,
      "grad_norm": 0.12952265846664912,
      "learning_rate": 0.0007792885143271472,
      "loss": 3.4782,
      "step": 1448
    },
    {
      "epoch": 0.33226324237560195,
      "grad_norm": 0.14595777571185925,
      "learning_rate": 0.0007789804228652449,
      "loss": 3.3838,
      "step": 1449
    },
    {
      "epoch": 0.3324925475808301,
      "grad_norm": 0.1252539066430055,
      "learning_rate": 0.0007786721775196073,
      "loss": 3.4784,
      "step": 1450
    },
    {
      "epoch": 0.33272185278605826,
      "grad_norm": 0.13913220960475964,
      "learning_rate": 0.0007783637784602609,
      "loss": 3.6732,
      "step": 1451
    },
    {
      "epoch": 0.3329511579912864,
      "grad_norm": 0.13660443824361165,
      "learning_rate": 0.0007780552258573163,
      "loss": 3.5115,
      "step": 1452
    },
    {
      "epoch": 0.33318046319651456,
      "grad_norm": 0.15829787827120748,
      "learning_rate": 0.0007777465198809692,
      "loss": 3.6527,
      "step": 1453
    },
    {
      "epoch": 0.33340976840174275,
      "grad_norm": 0.12473888821079421,
      "learning_rate": 0.0007774376607014994,
      "loss": 3.4114,
      "step": 1454
    },
    {
      "epoch": 0.33363907360697087,
      "grad_norm": 0.12863045644024762,
      "learning_rate": 0.0007771286484892722,
      "loss": 3.4384,
      "step": 1455
    },
    {
      "epoch": 0.33386837881219905,
      "grad_norm": 0.1209219394237598,
      "learning_rate": 0.0007768194834147362,
      "loss": 3.5444,
      "step": 1456
    },
    {
      "epoch": 0.3340976840174272,
      "grad_norm": 0.13469916000535076,
      "learning_rate": 0.000776510165648425,
      "loss": 3.5006,
      "step": 1457
    },
    {
      "epoch": 0.33432698922265536,
      "grad_norm": 0.13583788413077524,
      "learning_rate": 0.000776200695360956,
      "loss": 3.633,
      "step": 1458
    },
    {
      "epoch": 0.33455629442788354,
      "grad_norm": 0.14439402874429813,
      "learning_rate": 0.0007758910727230311,
      "loss": 3.499,
      "step": 1459
    },
    {
      "epoch": 0.33478559963311166,
      "grad_norm": 0.1339484089908917,
      "learning_rate": 0.000775581297905436,
      "loss": 3.5685,
      "step": 1460
    },
    {
      "epoch": 0.33501490483833984,
      "grad_norm": 0.13580660171721926,
      "learning_rate": 0.0007752713710790404,
      "loss": 3.6881,
      "step": 1461
    },
    {
      "epoch": 0.33524421004356797,
      "grad_norm": 0.13255861035983751,
      "learning_rate": 0.000774961292414798,
      "loss": 3.5079,
      "step": 1462
    },
    {
      "epoch": 0.33547351524879615,
      "grad_norm": 0.1433097709501589,
      "learning_rate": 0.0007746510620837459,
      "loss": 3.4609,
      "step": 1463
    },
    {
      "epoch": 0.33570282045402433,
      "grad_norm": 0.12957302798383546,
      "learning_rate": 0.0007743406802570053,
      "loss": 3.4487,
      "step": 1464
    },
    {
      "epoch": 0.33593212565925246,
      "grad_norm": 0.13404031206464775,
      "learning_rate": 0.0007740301471057807,
      "loss": 3.6141,
      "step": 1465
    },
    {
      "epoch": 0.33616143086448064,
      "grad_norm": 0.15022741506092424,
      "learning_rate": 0.0007737194628013599,
      "loss": 3.4982,
      "step": 1466
    },
    {
      "epoch": 0.33639073606970876,
      "grad_norm": 0.13458451865126572,
      "learning_rate": 0.0007734086275151146,
      "loss": 3.5204,
      "step": 1467
    },
    {
      "epoch": 0.33662004127493694,
      "grad_norm": 0.1409817244911584,
      "learning_rate": 0.0007730976414184992,
      "loss": 3.4813,
      "step": 1468
    },
    {
      "epoch": 0.3368493464801651,
      "grad_norm": 0.16341838285790392,
      "learning_rate": 0.0007727865046830517,
      "loss": 3.6891,
      "step": 1469
    },
    {
      "epoch": 0.33707865168539325,
      "grad_norm": 0.1423970035310754,
      "learning_rate": 0.0007724752174803929,
      "loss": 3.6709,
      "step": 1470
    },
    {
      "epoch": 0.33730795689062143,
      "grad_norm": 0.1295606129592353,
      "learning_rate": 0.0007721637799822269,
      "loss": 3.6022,
      "step": 1471
    },
    {
      "epoch": 0.33753726209584956,
      "grad_norm": 0.15495624171474928,
      "learning_rate": 0.0007718521923603404,
      "loss": 3.366,
      "step": 1472
    },
    {
      "epoch": 0.33776656730107774,
      "grad_norm": 0.14215226318128305,
      "learning_rate": 0.0007715404547866032,
      "loss": 3.5464,
      "step": 1473
    },
    {
      "epoch": 0.3379958725063059,
      "grad_norm": 0.1349250106304292,
      "learning_rate": 0.0007712285674329676,
      "loss": 3.5086,
      "step": 1474
    },
    {
      "epoch": 0.33822517771153404,
      "grad_norm": 0.14706684822984553,
      "learning_rate": 0.0007709165304714685,
      "loss": 3.6521,
      "step": 1475
    },
    {
      "epoch": 0.3384544829167622,
      "grad_norm": 0.1377153411265826,
      "learning_rate": 0.0007706043440742235,
      "loss": 3.5054,
      "step": 1476
    },
    {
      "epoch": 0.33868378812199035,
      "grad_norm": 0.14518672875838576,
      "learning_rate": 0.0007702920084134324,
      "loss": 3.5441,
      "step": 1477
    },
    {
      "epoch": 0.33891309332721853,
      "grad_norm": 0.14965968731787957,
      "learning_rate": 0.0007699795236613776,
      "loss": 3.5606,
      "step": 1478
    },
    {
      "epoch": 0.3391423985324467,
      "grad_norm": 0.1550557973180041,
      "learning_rate": 0.0007696668899904236,
      "loss": 3.6113,
      "step": 1479
    },
    {
      "epoch": 0.33937170373767483,
      "grad_norm": 0.15544755289597958,
      "learning_rate": 0.000769354107573017,
      "loss": 3.6024,
      "step": 1480
    },
    {
      "epoch": 0.339601008942903,
      "grad_norm": 0.1609177571262103,
      "learning_rate": 0.0007690411765816864,
      "loss": 3.6955,
      "step": 1481
    },
    {
      "epoch": 0.33983031414813114,
      "grad_norm": 0.1466201343190738,
      "learning_rate": 0.0007687280971890425,
      "loss": 3.532,
      "step": 1482
    },
    {
      "epoch": 0.3400596193533593,
      "grad_norm": 0.1235402365921717,
      "learning_rate": 0.0007684148695677778,
      "loss": 3.5346,
      "step": 1483
    },
    {
      "epoch": 0.3402889245585875,
      "grad_norm": 0.1261001643706058,
      "learning_rate": 0.0007681014938906666,
      "loss": 3.5832,
      "step": 1484
    },
    {
      "epoch": 0.3405182297638156,
      "grad_norm": 0.12314855591069832,
      "learning_rate": 0.000767787970330565,
      "loss": 3.5878,
      "step": 1485
    },
    {
      "epoch": 0.3407475349690438,
      "grad_norm": 0.1193476223810506,
      "learning_rate": 0.0007674742990604101,
      "loss": 3.5446,
      "step": 1486
    },
    {
      "epoch": 0.34097684017427193,
      "grad_norm": 0.12976001952457847,
      "learning_rate": 0.000767160480253221,
      "loss": 3.58,
      "step": 1487
    },
    {
      "epoch": 0.3412061453795001,
      "grad_norm": 0.12253069128055356,
      "learning_rate": 0.0007668465140820984,
      "loss": 3.5619,
      "step": 1488
    },
    {
      "epoch": 0.3414354505847283,
      "grad_norm": 0.13512069842323143,
      "learning_rate": 0.0007665324007202235,
      "loss": 3.5201,
      "step": 1489
    },
    {
      "epoch": 0.3416647557899564,
      "grad_norm": 0.13041432343217998,
      "learning_rate": 0.0007662181403408592,
      "loss": 3.6458,
      "step": 1490
    },
    {
      "epoch": 0.3418940609951846,
      "grad_norm": 0.12005245449098491,
      "learning_rate": 0.0007659037331173498,
      "loss": 3.5281,
      "step": 1491
    },
    {
      "epoch": 0.3421233662004127,
      "grad_norm": 0.14158686380883673,
      "learning_rate": 0.0007655891792231197,
      "loss": 3.4893,
      "step": 1492
    },
    {
      "epoch": 0.3423526714056409,
      "grad_norm": 0.13458452828892853,
      "learning_rate": 0.0007652744788316752,
      "loss": 3.4667,
      "step": 1493
    },
    {
      "epoch": 0.3425819766108691,
      "grad_norm": 0.1346912302039164,
      "learning_rate": 0.0007649596321166025,
      "loss": 3.4667,
      "step": 1494
    },
    {
      "epoch": 0.3428112818160972,
      "grad_norm": 0.1337753782958598,
      "learning_rate": 0.0007646446392515692,
      "loss": 3.5691,
      "step": 1495
    },
    {
      "epoch": 0.3430405870213254,
      "grad_norm": 0.15385554798596077,
      "learning_rate": 0.0007643295004103232,
      "loss": 3.6047,
      "step": 1496
    },
    {
      "epoch": 0.3432698922265535,
      "grad_norm": 0.17136431914071173,
      "learning_rate": 0.000764014215766693,
      "loss": 3.696,
      "step": 1497
    },
    {
      "epoch": 0.3434991974317817,
      "grad_norm": 0.1343340750508588,
      "learning_rate": 0.0007636987854945874,
      "loss": 3.5734,
      "step": 1498
    },
    {
      "epoch": 0.3437285026370099,
      "grad_norm": 0.14076879331963044,
      "learning_rate": 0.0007633832097679958,
      "loss": 3.4757,
      "step": 1499
    },
    {
      "epoch": 0.343957807842238,
      "grad_norm": 0.1422953069330846,
      "learning_rate": 0.0007630674887609876,
      "loss": 3.557,
      "step": 1500
    },
    {
      "epoch": 0.3441871130474662,
      "grad_norm": 0.13264319933549107,
      "learning_rate": 0.0007627516226477122,
      "loss": 3.5816,
      "step": 1501
    },
    {
      "epoch": 0.3444164182526943,
      "grad_norm": 0.14481542290925709,
      "learning_rate": 0.0007624356116023996,
      "loss": 3.4699,
      "step": 1502
    },
    {
      "epoch": 0.3446457234579225,
      "grad_norm": 0.12658184152756022,
      "learning_rate": 0.0007621194557993589,
      "loss": 3.6814,
      "step": 1503
    },
    {
      "epoch": 0.3448750286631507,
      "grad_norm": 0.137051773297617,
      "learning_rate": 0.00076180315541298,
      "loss": 3.6843,
      "step": 1504
    },
    {
      "epoch": 0.3451043338683788,
      "grad_norm": 0.12371199998402063,
      "learning_rate": 0.0007614867106177319,
      "loss": 3.4738,
      "step": 1505
    },
    {
      "epoch": 0.345333639073607,
      "grad_norm": 0.140629434733131,
      "learning_rate": 0.0007611701215881635,
      "loss": 3.5189,
      "step": 1506
    },
    {
      "epoch": 0.3455629442788351,
      "grad_norm": 0.13843271771174803,
      "learning_rate": 0.0007608533884989029,
      "loss": 3.5917,
      "step": 1507
    },
    {
      "epoch": 0.3457922494840633,
      "grad_norm": 0.11995780357376266,
      "learning_rate": 0.000760536511524658,
      "loss": 3.5263,
      "step": 1508
    },
    {
      "epoch": 0.34602155468929147,
      "grad_norm": 0.12138799353805634,
      "learning_rate": 0.0007602194908402166,
      "loss": 3.4864,
      "step": 1509
    },
    {
      "epoch": 0.3462508598945196,
      "grad_norm": 0.12807884463803557,
      "learning_rate": 0.0007599023266204445,
      "loss": 3.5253,
      "step": 1510
    },
    {
      "epoch": 0.34648016509974777,
      "grad_norm": 0.1404074957033638,
      "learning_rate": 0.0007595850190402877,
      "loss": 3.6112,
      "step": 1511
    },
    {
      "epoch": 0.3467094703049759,
      "grad_norm": 0.15180244018730088,
      "learning_rate": 0.0007592675682747708,
      "loss": 3.6239,
      "step": 1512
    },
    {
      "epoch": 0.3469387755102041,
      "grad_norm": 0.12553336283049152,
      "learning_rate": 0.0007589499744989976,
      "loss": 3.4022,
      "step": 1513
    },
    {
      "epoch": 0.34716808071543226,
      "grad_norm": 0.1237191384563885,
      "learning_rate": 0.0007586322378881508,
      "loss": 3.5465,
      "step": 1514
    },
    {
      "epoch": 0.3473973859206604,
      "grad_norm": 0.11770621457209618,
      "learning_rate": 0.0007583143586174916,
      "loss": 3.6072,
      "step": 1515
    },
    {
      "epoch": 0.34762669112588857,
      "grad_norm": 0.12962895247006645,
      "learning_rate": 0.0007579963368623601,
      "loss": 3.5811,
      "step": 1516
    },
    {
      "epoch": 0.3478559963311167,
      "grad_norm": 0.12020858672150277,
      "learning_rate": 0.000757678172798175,
      "loss": 3.5825,
      "step": 1517
    },
    {
      "epoch": 0.34808530153634487,
      "grad_norm": 0.13237430101879946,
      "learning_rate": 0.0007573598666004335,
      "loss": 3.6286,
      "step": 1518
    },
    {
      "epoch": 0.34831460674157305,
      "grad_norm": 0.1304284445063766,
      "learning_rate": 0.0007570414184447112,
      "loss": 3.6164,
      "step": 1519
    },
    {
      "epoch": 0.3485439119468012,
      "grad_norm": 0.13224724083458828,
      "learning_rate": 0.000756722828506662,
      "loss": 3.5447,
      "step": 1520
    },
    {
      "epoch": 0.34877321715202936,
      "grad_norm": 0.14926890115418578,
      "learning_rate": 0.0007564040969620179,
      "loss": 3.5882,
      "step": 1521
    },
    {
      "epoch": 0.3490025223572575,
      "grad_norm": 0.13258338460071914,
      "learning_rate": 0.0007560852239865889,
      "loss": 3.5911,
      "step": 1522
    },
    {
      "epoch": 0.34923182756248566,
      "grad_norm": 0.12987225053017315,
      "learning_rate": 0.0007557662097562636,
      "loss": 3.6874,
      "step": 1523
    },
    {
      "epoch": 0.34946113276771384,
      "grad_norm": 0.14153533475880295,
      "learning_rate": 0.0007554470544470079,
      "loss": 3.5952,
      "step": 1524
    },
    {
      "epoch": 0.34969043797294197,
      "grad_norm": 0.15776491345021007,
      "learning_rate": 0.0007551277582348658,
      "loss": 3.4926,
      "step": 1525
    },
    {
      "epoch": 0.34991974317817015,
      "grad_norm": 0.1482156836833684,
      "learning_rate": 0.0007548083212959587,
      "loss": 3.4606,
      "step": 1526
    },
    {
      "epoch": 0.3501490483833983,
      "grad_norm": 0.13964195390022222,
      "learning_rate": 0.0007544887438064862,
      "loss": 3.5845,
      "step": 1527
    },
    {
      "epoch": 0.35037835358862646,
      "grad_norm": 0.13478613708057574,
      "learning_rate": 0.000754169025942725,
      "loss": 3.4554,
      "step": 1528
    },
    {
      "epoch": 0.35060765879385464,
      "grad_norm": 0.14743118474926667,
      "learning_rate": 0.0007538491678810294,
      "loss": 3.4841,
      "step": 1529
    },
    {
      "epoch": 0.35083696399908276,
      "grad_norm": 0.14030797924097316,
      "learning_rate": 0.0007535291697978309,
      "loss": 3.6297,
      "step": 1530
    },
    {
      "epoch": 0.35106626920431094,
      "grad_norm": 0.12307706156568946,
      "learning_rate": 0.0007532090318696381,
      "loss": 3.5796,
      "step": 1531
    },
    {
      "epoch": 0.3512955744095391,
      "grad_norm": 0.14678925762978023,
      "learning_rate": 0.0007528887542730374,
      "loss": 3.5149,
      "step": 1532
    },
    {
      "epoch": 0.35152487961476725,
      "grad_norm": 0.1310796709238968,
      "learning_rate": 0.0007525683371846913,
      "loss": 3.4217,
      "step": 1533
    },
    {
      "epoch": 0.35175418481999543,
      "grad_norm": 0.14356658651619297,
      "learning_rate": 0.00075224778078134,
      "loss": 3.6911,
      "step": 1534
    },
    {
      "epoch": 0.35198349002522356,
      "grad_norm": 0.13838789112054076,
      "learning_rate": 0.0007519270852398001,
      "loss": 3.5391,
      "step": 1535
    },
    {
      "epoch": 0.35221279523045174,
      "grad_norm": 0.14553503422252762,
      "learning_rate": 0.0007516062507369654,
      "loss": 3.7211,
      "step": 1536
    },
    {
      "epoch": 0.3524421004356799,
      "grad_norm": 0.12967061851054723,
      "learning_rate": 0.000751285277449806,
      "loss": 3.4295,
      "step": 1537
    },
    {
      "epoch": 0.35267140564090804,
      "grad_norm": 0.1373518110926237,
      "learning_rate": 0.0007509641655553683,
      "loss": 3.4268,
      "step": 1538
    },
    {
      "epoch": 0.3529007108461362,
      "grad_norm": 0.12140772582517441,
      "learning_rate": 0.0007506429152307756,
      "loss": 3.4491,
      "step": 1539
    },
    {
      "epoch": 0.35313001605136435,
      "grad_norm": 0.11959890843698859,
      "learning_rate": 0.0007503215266532276,
      "loss": 3.5328,
      "step": 1540
    },
    {
      "epoch": 0.35335932125659253,
      "grad_norm": 0.14007292747355465,
      "learning_rate": 0.00075,
      "loss": 3.6046,
      "step": 1541
    },
    {
      "epoch": 0.3535886264618207,
      "grad_norm": 0.14228040705062067,
      "learning_rate": 0.0007496783354484449,
      "loss": 3.6224,
      "step": 1542
    },
    {
      "epoch": 0.35381793166704884,
      "grad_norm": 0.1226612217736403,
      "learning_rate": 0.00074935653317599,
      "loss": 3.5565,
      "step": 1543
    },
    {
      "epoch": 0.354047236872277,
      "grad_norm": 0.1278624331631339,
      "learning_rate": 0.0007490345933601394,
      "loss": 3.5317,
      "step": 1544
    },
    {
      "epoch": 0.35427654207750514,
      "grad_norm": 0.12168147217362027,
      "learning_rate": 0.000748712516178473,
      "loss": 3.3906,
      "step": 1545
    },
    {
      "epoch": 0.3545058472827333,
      "grad_norm": 0.12997795246352867,
      "learning_rate": 0.0007483903018086466,
      "loss": 3.5368,
      "step": 1546
    },
    {
      "epoch": 0.3547351524879615,
      "grad_norm": 0.13847604793498702,
      "learning_rate": 0.0007480679504283911,
      "loss": 3.5742,
      "step": 1547
    },
    {
      "epoch": 0.35496445769318963,
      "grad_norm": 0.14170130091817515,
      "learning_rate": 0.0007477454622155135,
      "loss": 3.6722,
      "step": 1548
    },
    {
      "epoch": 0.3551937628984178,
      "grad_norm": 0.12696750778413982,
      "learning_rate": 0.0007474228373478964,
      "loss": 3.5695,
      "step": 1549
    },
    {
      "epoch": 0.35542306810364593,
      "grad_norm": 0.1369607968978726,
      "learning_rate": 0.0007471000760034972,
      "loss": 3.6261,
      "step": 1550
    },
    {
      "epoch": 0.3556523733088741,
      "grad_norm": 0.13647777682798848,
      "learning_rate": 0.0007467771783603492,
      "loss": 3.5673,
      "step": 1551
    },
    {
      "epoch": 0.3558816785141023,
      "grad_norm": 0.1280109225195111,
      "learning_rate": 0.0007464541445965602,
      "loss": 3.5548,
      "step": 1552
    },
    {
      "epoch": 0.3561109837193304,
      "grad_norm": 0.14832078852016387,
      "learning_rate": 0.0007461309748903138,
      "loss": 3.5557,
      "step": 1553
    },
    {
      "epoch": 0.3563402889245586,
      "grad_norm": 0.150417821462944,
      "learning_rate": 0.0007458076694198681,
      "loss": 3.5476,
      "step": 1554
    },
    {
      "epoch": 0.3565695941297867,
      "grad_norm": 0.1371690990798973,
      "learning_rate": 0.0007454842283635562,
      "loss": 3.5865,
      "step": 1555
    },
    {
      "epoch": 0.3567988993350149,
      "grad_norm": 0.13181867796145913,
      "learning_rate": 0.0007451606518997862,
      "loss": 3.4638,
      "step": 1556
    },
    {
      "epoch": 0.3570282045402431,
      "grad_norm": 0.1491295365317617,
      "learning_rate": 0.0007448369402070404,
      "loss": 3.4362,
      "step": 1557
    },
    {
      "epoch": 0.3572575097454712,
      "grad_norm": 0.14270701496862767,
      "learning_rate": 0.0007445130934638764,
      "loss": 3.5883,
      "step": 1558
    },
    {
      "epoch": 0.3574868149506994,
      "grad_norm": 0.14609072124536063,
      "learning_rate": 0.0007441891118489254,
      "loss": 3.5646,
      "step": 1559
    },
    {
      "epoch": 0.3577161201559275,
      "grad_norm": 0.1354904331063319,
      "learning_rate": 0.0007438649955408941,
      "loss": 3.4964,
      "step": 1560
    },
    {
      "epoch": 0.3579454253611557,
      "grad_norm": 0.14245997041594477,
      "learning_rate": 0.0007435407447185622,
      "loss": 3.4633,
      "step": 1561
    },
    {
      "epoch": 0.3581747305663839,
      "grad_norm": 0.1407881819933249,
      "learning_rate": 0.000743216359560785,
      "loss": 3.6498,
      "step": 1562
    },
    {
      "epoch": 0.358404035771612,
      "grad_norm": 0.13243982189396136,
      "learning_rate": 0.0007428918402464908,
      "loss": 3.5205,
      "step": 1563
    },
    {
      "epoch": 0.3586333409768402,
      "grad_norm": 0.13192866598040395,
      "learning_rate": 0.0007425671869546822,
      "loss": 3.6241,
      "step": 1564
    },
    {
      "epoch": 0.3588626461820683,
      "grad_norm": 0.12852313294627124,
      "learning_rate": 0.0007422423998644359,
      "loss": 3.417,
      "step": 1565
    },
    {
      "epoch": 0.3590919513872965,
      "grad_norm": 0.13428966876903586,
      "learning_rate": 0.0007419174791549023,
      "loss": 3.6361,
      "step": 1566
    },
    {
      "epoch": 0.3593212565925247,
      "grad_norm": 0.13112091371131876,
      "learning_rate": 0.0007415924250053055,
      "loss": 3.5928,
      "step": 1567
    },
    {
      "epoch": 0.3595505617977528,
      "grad_norm": 0.12927488916980842,
      "learning_rate": 0.0007412672375949431,
      "loss": 3.5717,
      "step": 1568
    },
    {
      "epoch": 0.359779867002981,
      "grad_norm": 0.1260359767085019,
      "learning_rate": 0.0007409419171031865,
      "loss": 3.5095,
      "step": 1569
    },
    {
      "epoch": 0.3600091722082091,
      "grad_norm": 0.13612820946118176,
      "learning_rate": 0.0007406164637094801,
      "loss": 3.5109,
      "step": 1570
    },
    {
      "epoch": 0.3602384774134373,
      "grad_norm": 0.13700205260878306,
      "learning_rate": 0.0007402908775933419,
      "loss": 3.5289,
      "step": 1571
    },
    {
      "epoch": 0.36046778261866547,
      "grad_norm": 0.14950759121769577,
      "learning_rate": 0.0007399651589343632,
      "loss": 3.5375,
      "step": 1572
    },
    {
      "epoch": 0.3606970878238936,
      "grad_norm": 0.1450437622778872,
      "learning_rate": 0.0007396393079122077,
      "loss": 3.4804,
      "step": 1573
    },
    {
      "epoch": 0.3609263930291218,
      "grad_norm": 0.13224480022488314,
      "learning_rate": 0.0007393133247066133,
      "loss": 3.3819,
      "step": 1574
    },
    {
      "epoch": 0.3611556982343499,
      "grad_norm": 0.1304408770057927,
      "learning_rate": 0.0007389872094973896,
      "loss": 3.4715,
      "step": 1575
    },
    {
      "epoch": 0.3613850034395781,
      "grad_norm": 0.12103669348788251,
      "learning_rate": 0.00073866096246442,
      "loss": 3.4571,
      "step": 1576
    },
    {
      "epoch": 0.36161430864480626,
      "grad_norm": 0.1421915506296089,
      "learning_rate": 0.00073833458378766,
      "loss": 3.5309,
      "step": 1577
    },
    {
      "epoch": 0.3618436138500344,
      "grad_norm": 0.1548347183063643,
      "learning_rate": 0.0007380080736471378,
      "loss": 3.5973,
      "step": 1578
    },
    {
      "epoch": 0.36207291905526257,
      "grad_norm": 0.13950869934894994,
      "learning_rate": 0.0007376814322229544,
      "loss": 3.5987,
      "step": 1579
    },
    {
      "epoch": 0.3623022242604907,
      "grad_norm": 0.12819653345534002,
      "learning_rate": 0.0007373546596952828,
      "loss": 3.3934,
      "step": 1580
    },
    {
      "epoch": 0.36253152946571887,
      "grad_norm": 0.14780710375640274,
      "learning_rate": 0.0007370277562443688,
      "loss": 3.6028,
      "step": 1581
    },
    {
      "epoch": 0.36276083467094705,
      "grad_norm": 0.1281929632188211,
      "learning_rate": 0.00073670072205053,
      "loss": 3.6145,
      "step": 1582
    },
    {
      "epoch": 0.3629901398761752,
      "grad_norm": 0.13391236263906253,
      "learning_rate": 0.0007363735572941564,
      "loss": 3.6009,
      "step": 1583
    },
    {
      "epoch": 0.36321944508140336,
      "grad_norm": 0.131054618678178,
      "learning_rate": 0.0007360462621557095,
      "loss": 3.5006,
      "step": 1584
    },
    {
      "epoch": 0.3634487502866315,
      "grad_norm": 0.1333189098362532,
      "learning_rate": 0.0007357188368157236,
      "loss": 3.5374,
      "step": 1585
    },
    {
      "epoch": 0.36367805549185966,
      "grad_norm": 0.13679623580948583,
      "learning_rate": 0.0007353912814548042,
      "loss": 3.5475,
      "step": 1586
    },
    {
      "epoch": 0.36390736069708784,
      "grad_norm": 0.13515456348031032,
      "learning_rate": 0.0007350635962536284,
      "loss": 3.6813,
      "step": 1587
    },
    {
      "epoch": 0.36413666590231597,
      "grad_norm": 0.13190884460225374,
      "learning_rate": 0.0007347357813929454,
      "loss": 3.4941,
      "step": 1588
    },
    {
      "epoch": 0.36436597110754415,
      "grad_norm": 0.12864433581143558,
      "learning_rate": 0.0007344078370535756,
      "loss": 3.5875,
      "step": 1589
    },
    {
      "epoch": 0.3645952763127723,
      "grad_norm": 0.12907288569604944,
      "learning_rate": 0.000734079763416411,
      "loss": 3.4848,
      "step": 1590
    },
    {
      "epoch": 0.36482458151800046,
      "grad_norm": 0.13618599501304304,
      "learning_rate": 0.0007337515606624148,
      "loss": 3.6697,
      "step": 1591
    },
    {
      "epoch": 0.36505388672322864,
      "grad_norm": 0.14904740873560599,
      "learning_rate": 0.0007334232289726212,
      "loss": 3.6029,
      "step": 1592
    },
    {
      "epoch": 0.36528319192845676,
      "grad_norm": 0.14023348697964022,
      "learning_rate": 0.0007330947685281362,
      "loss": 3.5207,
      "step": 1593
    },
    {
      "epoch": 0.36551249713368494,
      "grad_norm": 0.1348735144097619,
      "learning_rate": 0.000732766179510136,
      "loss": 3.5796,
      "step": 1594
    },
    {
      "epoch": 0.36574180233891307,
      "grad_norm": 0.12873882505098272,
      "learning_rate": 0.0007324374620998682,
      "loss": 3.462,
      "step": 1595
    },
    {
      "epoch": 0.36597110754414125,
      "grad_norm": 0.12309734200721215,
      "learning_rate": 0.0007321086164786512,
      "loss": 3.5013,
      "step": 1596
    },
    {
      "epoch": 0.36620041274936943,
      "grad_norm": 0.13641388287359021,
      "learning_rate": 0.000731779642827874,
      "loss": 3.5954,
      "step": 1597
    },
    {
      "epoch": 0.36642971795459756,
      "grad_norm": 0.12173520648539993,
      "learning_rate": 0.0007314505413289963,
      "loss": 3.4082,
      "step": 1598
    },
    {
      "epoch": 0.36665902315982574,
      "grad_norm": 0.13966497743881065,
      "learning_rate": 0.0007311213121635483,
      "loss": 3.5473,
      "step": 1599
    },
    {
      "epoch": 0.36688832836505386,
      "grad_norm": 0.12580840096253984,
      "learning_rate": 0.0007307919555131306,
      "loss": 3.4851,
      "step": 1600
    },
    {
      "epoch": 0.36711763357028204,
      "grad_norm": 0.12818039354274366,
      "learning_rate": 0.0007304624715594139,
      "loss": 3.5918,
      "step": 1601
    },
    {
      "epoch": 0.3673469387755102,
      "grad_norm": 0.14676051824719033,
      "learning_rate": 0.0007301328604841399,
      "loss": 3.4537,
      "step": 1602
    },
    {
      "epoch": 0.36757624398073835,
      "grad_norm": 0.1383162548702085,
      "learning_rate": 0.0007298031224691193,
      "loss": 3.6432,
      "step": 1603
    },
    {
      "epoch": 0.36780554918596653,
      "grad_norm": 0.1685539615268935,
      "learning_rate": 0.0007294732576962335,
      "loss": 3.55,
      "step": 1604
    },
    {
      "epoch": 0.36803485439119465,
      "grad_norm": 0.1398767743193315,
      "learning_rate": 0.0007291432663474339,
      "loss": 3.616,
      "step": 1605
    },
    {
      "epoch": 0.36826415959642284,
      "grad_norm": 0.15730273843864184,
      "learning_rate": 0.0007288131486047414,
      "loss": 3.5711,
      "step": 1606
    },
    {
      "epoch": 0.368493464801651,
      "grad_norm": 0.1311413090098633,
      "learning_rate": 0.0007284829046502467,
      "loss": 3.5539,
      "step": 1607
    },
    {
      "epoch": 0.36872277000687914,
      "grad_norm": 0.11972425886116718,
      "learning_rate": 0.0007281525346661102,
      "loss": 3.4541,
      "step": 1608
    },
    {
      "epoch": 0.3689520752121073,
      "grad_norm": 0.1261695818475225,
      "learning_rate": 0.0007278220388345619,
      "loss": 3.451,
      "step": 1609
    },
    {
      "epoch": 0.36918138041733545,
      "grad_norm": 0.13298963322849322,
      "learning_rate": 0.0007274914173379009,
      "loss": 3.5599,
      "step": 1610
    },
    {
      "epoch": 0.36941068562256363,
      "grad_norm": 0.15172687953016425,
      "learning_rate": 0.0007271606703584958,
      "loss": 3.553,
      "step": 1611
    },
    {
      "epoch": 0.3696399908277918,
      "grad_norm": 0.13059383098959818,
      "learning_rate": 0.0007268297980787846,
      "loss": 3.4956,
      "step": 1612
    },
    {
      "epoch": 0.36986929603301993,
      "grad_norm": 0.15019400535960067,
      "learning_rate": 0.000726498800681274,
      "loss": 3.5022,
      "step": 1613
    },
    {
      "epoch": 0.3700986012382481,
      "grad_norm": 0.15012924570142455,
      "learning_rate": 0.0007261676783485402,
      "loss": 3.5221,
      "step": 1614
    },
    {
      "epoch": 0.37032790644347624,
      "grad_norm": 0.1317387120057884,
      "learning_rate": 0.0007258364312632279,
      "loss": 3.4528,
      "step": 1615
    },
    {
      "epoch": 0.3705572116487044,
      "grad_norm": 0.14005060854511972,
      "learning_rate": 0.0007255050596080509,
      "loss": 3.6177,
      "step": 1616
    },
    {
      "epoch": 0.3707865168539326,
      "grad_norm": 0.13301625902766015,
      "learning_rate": 0.0007251735635657915,
      "loss": 3.5697,
      "step": 1617
    },
    {
      "epoch": 0.3710158220591607,
      "grad_norm": 0.1527904921371237,
      "learning_rate": 0.0007248419433193007,
      "loss": 3.5245,
      "step": 1618
    },
    {
      "epoch": 0.3712451272643889,
      "grad_norm": 0.13624522488116542,
      "learning_rate": 0.000724510199051498,
      "loss": 3.5754,
      "step": 1619
    },
    {
      "epoch": 0.37147443246961703,
      "grad_norm": 0.12798941614554307,
      "learning_rate": 0.0007241783309453714,
      "loss": 3.4956,
      "step": 1620
    },
    {
      "epoch": 0.3717037376748452,
      "grad_norm": 0.13526968357852512,
      "learning_rate": 0.0007238463391839769,
      "loss": 3.5551,
      "step": 1621
    },
    {
      "epoch": 0.3719330428800734,
      "grad_norm": 0.1510546979160967,
      "learning_rate": 0.0007235142239504391,
      "loss": 3.7455,
      "step": 1622
    },
    {
      "epoch": 0.3721623480853015,
      "grad_norm": 0.1331617388271413,
      "learning_rate": 0.0007231819854279508,
      "loss": 3.5627,
      "step": 1623
    },
    {
      "epoch": 0.3723916532905297,
      "grad_norm": 0.14116520201792881,
      "learning_rate": 0.0007228496237997722,
      "loss": 3.5406,
      "step": 1624
    },
    {
      "epoch": 0.3726209584957579,
      "grad_norm": 0.14063268475321458,
      "learning_rate": 0.0007225171392492316,
      "loss": 3.6425,
      "step": 1625
    },
    {
      "epoch": 0.372850263700986,
      "grad_norm": 0.12851233847756388,
      "learning_rate": 0.0007221845319597258,
      "loss": 3.3995,
      "step": 1626
    },
    {
      "epoch": 0.3730795689062142,
      "grad_norm": 0.1427263143819911,
      "learning_rate": 0.0007218518021147182,
      "loss": 3.4744,
      "step": 1627
    },
    {
      "epoch": 0.3733088741114423,
      "grad_norm": 0.12543319145421017,
      "learning_rate": 0.0007215189498977409,
      "loss": 3.4584,
      "step": 1628
    },
    {
      "epoch": 0.3735381793166705,
      "grad_norm": 0.13450772080626597,
      "learning_rate": 0.0007211859754923923,
      "loss": 3.4661,
      "step": 1629
    },
    {
      "epoch": 0.3737674845218987,
      "grad_norm": 0.14382179346816143,
      "learning_rate": 0.0007208528790823394,
      "loss": 3.7316,
      "step": 1630
    },
    {
      "epoch": 0.3739967897271268,
      "grad_norm": 0.13828454417059807,
      "learning_rate": 0.0007205196608513158,
      "loss": 3.5585,
      "step": 1631
    },
    {
      "epoch": 0.374226094932355,
      "grad_norm": 0.14243340653878114,
      "learning_rate": 0.0007201863209831223,
      "loss": 3.4039,
      "step": 1632
    },
    {
      "epoch": 0.3744554001375831,
      "grad_norm": 0.14426807723667087,
      "learning_rate": 0.0007198528596616272,
      "loss": 3.5511,
      "step": 1633
    },
    {
      "epoch": 0.3746847053428113,
      "grad_norm": 0.13773104783094722,
      "learning_rate": 0.0007195192770707654,
      "loss": 3.5819,
      "step": 1634
    },
    {
      "epoch": 0.37491401054803947,
      "grad_norm": 0.13624109748263155,
      "learning_rate": 0.0007191855733945387,
      "loss": 3.4438,
      "step": 1635
    },
    {
      "epoch": 0.3751433157532676,
      "grad_norm": 0.15244335488500355,
      "learning_rate": 0.0007188517488170159,
      "loss": 3.4729,
      "step": 1636
    },
    {
      "epoch": 0.3753726209584958,
      "grad_norm": 0.16137397181282676,
      "learning_rate": 0.0007185178035223327,
      "loss": 3.6156,
      "step": 1637
    },
    {
      "epoch": 0.3756019261637239,
      "grad_norm": 0.15180953191855037,
      "learning_rate": 0.0007181837376946905,
      "loss": 3.4298,
      "step": 1638
    },
    {
      "epoch": 0.3758312313689521,
      "grad_norm": 0.13620473783636333,
      "learning_rate": 0.0007178495515183583,
      "loss": 3.5642,
      "step": 1639
    },
    {
      "epoch": 0.37606053657418026,
      "grad_norm": 0.13279588125768435,
      "learning_rate": 0.0007175152451776708,
      "loss": 3.5199,
      "step": 1640
    },
    {
      "epoch": 0.3762898417794084,
      "grad_norm": 0.12976939257704226,
      "learning_rate": 0.000717180818857029,
      "loss": 3.4962,
      "step": 1641
    },
    {
      "epoch": 0.37651914698463657,
      "grad_norm": 0.15618869293594698,
      "learning_rate": 0.0007168462727409003,
      "loss": 3.5042,
      "step": 1642
    },
    {
      "epoch": 0.3767484521898647,
      "grad_norm": 0.15464698463242207,
      "learning_rate": 0.0007165116070138182,
      "loss": 3.7149,
      "step": 1643
    },
    {
      "epoch": 0.37697775739509287,
      "grad_norm": 0.12428165876386728,
      "learning_rate": 0.0007161768218603821,
      "loss": 3.5734,
      "step": 1644
    },
    {
      "epoch": 0.37720706260032105,
      "grad_norm": 0.13298205518873793,
      "learning_rate": 0.0007158419174652569,
      "loss": 3.4738,
      "step": 1645
    },
    {
      "epoch": 0.3774363678055492,
      "grad_norm": 0.13262843727569068,
      "learning_rate": 0.000715506894013174,
      "loss": 3.4811,
      "step": 1646
    },
    {
      "epoch": 0.37766567301077736,
      "grad_norm": 0.13939770843064833,
      "learning_rate": 0.00071517175168893,
      "loss": 3.5504,
      "step": 1647
    },
    {
      "epoch": 0.3778949782160055,
      "grad_norm": 0.1287306747780721,
      "learning_rate": 0.0007148364906773871,
      "loss": 3.5026,
      "step": 1648
    },
    {
      "epoch": 0.37812428342123366,
      "grad_norm": 0.13579425828360558,
      "learning_rate": 0.0007145011111634732,
      "loss": 3.5849,
      "step": 1649
    },
    {
      "epoch": 0.37835358862646185,
      "grad_norm": 0.12190491441842122,
      "learning_rate": 0.000714165613332181,
      "loss": 3.497,
      "step": 1650
    },
    {
      "epoch": 0.37858289383168997,
      "grad_norm": 0.12386783723713503,
      "learning_rate": 0.0007138299973685694,
      "loss": 3.4852,
      "step": 1651
    },
    {
      "epoch": 0.37881219903691815,
      "grad_norm": 0.12843811640660843,
      "learning_rate": 0.0007134942634577614,
      "loss": 3.5242,
      "step": 1652
    },
    {
      "epoch": 0.3790415042421463,
      "grad_norm": 0.12963544212440006,
      "learning_rate": 0.0007131584117849459,
      "loss": 3.5575,
      "step": 1653
    },
    {
      "epoch": 0.37927080944737446,
      "grad_norm": 0.1258149521514678,
      "learning_rate": 0.0007128224425353764,
      "loss": 3.4048,
      "step": 1654
    },
    {
      "epoch": 0.37950011465260264,
      "grad_norm": 0.12344121144410007,
      "learning_rate": 0.0007124863558943713,
      "loss": 3.5094,
      "step": 1655
    },
    {
      "epoch": 0.37972941985783076,
      "grad_norm": 0.13388242579600085,
      "learning_rate": 0.0007121501520473137,
      "loss": 3.5769,
      "step": 1656
    },
    {
      "epoch": 0.37995872506305894,
      "grad_norm": 0.13673532566631932,
      "learning_rate": 0.0007118138311796514,
      "loss": 3.5284,
      "step": 1657
    },
    {
      "epoch": 0.38018803026828707,
      "grad_norm": 0.12345411727311542,
      "learning_rate": 0.0007114773934768965,
      "loss": 3.49,
      "step": 1658
    },
    {
      "epoch": 0.38041733547351525,
      "grad_norm": 0.1239337500543878,
      "learning_rate": 0.0007111408391246262,
      "loss": 3.3882,
      "step": 1659
    },
    {
      "epoch": 0.38064664067874343,
      "grad_norm": 0.13968635070736873,
      "learning_rate": 0.0007108041683084814,
      "loss": 3.6333,
      "step": 1660
    },
    {
      "epoch": 0.38087594588397156,
      "grad_norm": 0.13537610105014197,
      "learning_rate": 0.0007104673812141675,
      "loss": 3.5624,
      "step": 1661
    },
    {
      "epoch": 0.38110525108919974,
      "grad_norm": 0.1295296137539506,
      "learning_rate": 0.000710130478027454,
      "loss": 3.4704,
      "step": 1662
    },
    {
      "epoch": 0.38133455629442786,
      "grad_norm": 0.13161759349336788,
      "learning_rate": 0.0007097934589341745,
      "loss": 3.6134,
      "step": 1663
    },
    {
      "epoch": 0.38156386149965604,
      "grad_norm": 0.1370738974366995,
      "learning_rate": 0.0007094563241202266,
      "loss": 3.4865,
      "step": 1664
    },
    {
      "epoch": 0.3817931667048842,
      "grad_norm": 0.13644041185393138,
      "learning_rate": 0.0007091190737715711,
      "loss": 3.5192,
      "step": 1665
    },
    {
      "epoch": 0.38202247191011235,
      "grad_norm": 0.14142598616457688,
      "learning_rate": 0.0007087817080742337,
      "loss": 3.642,
      "step": 1666
    },
    {
      "epoch": 0.38225177711534053,
      "grad_norm": 0.13100979330057003,
      "learning_rate": 0.0007084442272143026,
      "loss": 3.3792,
      "step": 1667
    },
    {
      "epoch": 0.38248108232056865,
      "grad_norm": 0.15009400157160332,
      "learning_rate": 0.0007081066313779301,
      "loss": 3.5194,
      "step": 1668
    },
    {
      "epoch": 0.38271038752579684,
      "grad_norm": 0.13550081708673398,
      "learning_rate": 0.000707768920751332,
      "loss": 3.4909,
      "step": 1669
    },
    {
      "epoch": 0.382939692731025,
      "grad_norm": 0.14241451709699426,
      "learning_rate": 0.0007074310955207869,
      "loss": 3.5443,
      "step": 1670
    },
    {
      "epoch": 0.38316899793625314,
      "grad_norm": 0.1295632436621588,
      "learning_rate": 0.0007070931558726373,
      "loss": 3.6084,
      "step": 1671
    },
    {
      "epoch": 0.3833983031414813,
      "grad_norm": 0.13514737290483023,
      "learning_rate": 0.0007067551019932879,
      "loss": 3.6141,
      "step": 1672
    },
    {
      "epoch": 0.38362760834670945,
      "grad_norm": 0.15403342908667436,
      "learning_rate": 0.0007064169340692076,
      "loss": 3.5786,
      "step": 1673
    },
    {
      "epoch": 0.38385691355193763,
      "grad_norm": 0.16369858129456613,
      "learning_rate": 0.0007060786522869271,
      "loss": 3.6303,
      "step": 1674
    },
    {
      "epoch": 0.3840862187571658,
      "grad_norm": 0.13548298113075388,
      "learning_rate": 0.0007057402568330407,
      "loss": 3.5239,
      "step": 1675
    },
    {
      "epoch": 0.38431552396239393,
      "grad_norm": 0.15631438350737914,
      "learning_rate": 0.0007054017478942047,
      "loss": 3.3915,
      "step": 1676
    },
    {
      "epoch": 0.3845448291676221,
      "grad_norm": 0.13772041997638407,
      "learning_rate": 0.0007050631256571389,
      "loss": 3.4425,
      "step": 1677
    },
    {
      "epoch": 0.38477413437285024,
      "grad_norm": 0.1441288664445646,
      "learning_rate": 0.0007047243903086245,
      "loss": 3.5804,
      "step": 1678
    },
    {
      "epoch": 0.3850034395780784,
      "grad_norm": 0.13285791850154266,
      "learning_rate": 0.000704385542035506,
      "loss": 3.5866,
      "step": 1679
    },
    {
      "epoch": 0.3852327447833066,
      "grad_norm": 0.140983018904344,
      "learning_rate": 0.0007040465810246897,
      "loss": 3.5138,
      "step": 1680
    },
    {
      "epoch": 0.3854620499885347,
      "grad_norm": 0.1315965011632515,
      "learning_rate": 0.000703707507463144,
      "loss": 3.5154,
      "step": 1681
    },
    {
      "epoch": 0.3856913551937629,
      "grad_norm": 0.12927273630070069,
      "learning_rate": 0.0007033683215379002,
      "loss": 3.7555,
      "step": 1682
    },
    {
      "epoch": 0.38592066039899103,
      "grad_norm": 0.1346155274037756,
      "learning_rate": 0.0007030290234360505,
      "loss": 3.3823,
      "step": 1683
    },
    {
      "epoch": 0.3861499656042192,
      "grad_norm": 0.12402992540451466,
      "learning_rate": 0.0007026896133447497,
      "loss": 3.627,
      "step": 1684
    },
    {
      "epoch": 0.3863792708094474,
      "grad_norm": 0.13134401672632398,
      "learning_rate": 0.0007023500914512139,
      "loss": 3.4729,
      "step": 1685
    },
    {
      "epoch": 0.3866085760146755,
      "grad_norm": 0.12163827108723309,
      "learning_rate": 0.0007020104579427214,
      "loss": 3.4974,
      "step": 1686
    },
    {
      "epoch": 0.3868378812199037,
      "grad_norm": 0.12275887176116765,
      "learning_rate": 0.0007016707130066116,
      "loss": 3.6473,
      "step": 1687
    },
    {
      "epoch": 0.3870671864251318,
      "grad_norm": 0.1365889254108298,
      "learning_rate": 0.0007013308568302854,
      "loss": 3.457,
      "step": 1688
    },
    {
      "epoch": 0.38729649163036,
      "grad_norm": 0.13924280357003913,
      "learning_rate": 0.0007009908896012055,
      "loss": 3.6146,
      "step": 1689
    },
    {
      "epoch": 0.3875257968355882,
      "grad_norm": 0.13299873311440624,
      "learning_rate": 0.0007006508115068954,
      "loss": 3.6252,
      "step": 1690
    },
    {
      "epoch": 0.3877551020408163,
      "grad_norm": 0.13201516666329038,
      "learning_rate": 0.0007003106227349399,
      "loss": 3.4598,
      "step": 1691
    },
    {
      "epoch": 0.3879844072460445,
      "grad_norm": 0.13146551693523692,
      "learning_rate": 0.0006999703234729847,
      "loss": 3.5319,
      "step": 1692
    },
    {
      "epoch": 0.3882137124512726,
      "grad_norm": 0.12976435159857508,
      "learning_rate": 0.000699629913908737,
      "loss": 3.586,
      "step": 1693
    },
    {
      "epoch": 0.3884430176565008,
      "grad_norm": 0.13288093324413802,
      "learning_rate": 0.0006992893942299641,
      "loss": 3.4253,
      "step": 1694
    },
    {
      "epoch": 0.388672322861729,
      "grad_norm": 0.12075093341438867,
      "learning_rate": 0.0006989487646244943,
      "loss": 3.4155,
      "step": 1695
    },
    {
      "epoch": 0.3889016280669571,
      "grad_norm": 0.13568706657938753,
      "learning_rate": 0.000698608025280217,
      "loss": 3.5071,
      "step": 1696
    },
    {
      "epoch": 0.3891309332721853,
      "grad_norm": 0.12965049348134847,
      "learning_rate": 0.0006982671763850814,
      "loss": 3.4872,
      "step": 1697
    },
    {
      "epoch": 0.3893602384774134,
      "grad_norm": 0.16728101304172324,
      "learning_rate": 0.0006979262181270976,
      "loss": 3.5694,
      "step": 1698
    },
    {
      "epoch": 0.3895895436826416,
      "grad_norm": 0.14477050718377252,
      "learning_rate": 0.0006975851506943359,
      "loss": 3.6896,
      "step": 1699
    },
    {
      "epoch": 0.3898188488878698,
      "grad_norm": 0.12789795483465352,
      "learning_rate": 0.000697243974274927,
      "loss": 3.486,
      "step": 1700
    },
    {
      "epoch": 0.3900481540930979,
      "grad_norm": 0.14704992548954257,
      "learning_rate": 0.0006969026890570611,
      "loss": 3.6727,
      "step": 1701
    },
    {
      "epoch": 0.3902774592983261,
      "grad_norm": 0.13398740058651362,
      "learning_rate": 0.000696561295228989,
      "loss": 3.5407,
      "step": 1702
    },
    {
      "epoch": 0.3905067645035542,
      "grad_norm": 0.13162413677726797,
      "learning_rate": 0.0006962197929790216,
      "loss": 3.5185,
      "step": 1703
    },
    {
      "epoch": 0.3907360697087824,
      "grad_norm": 0.1318790045394086,
      "learning_rate": 0.0006958781824955288,
      "loss": 3.4971,
      "step": 1704
    },
    {
      "epoch": 0.39096537491401057,
      "grad_norm": 0.14719857333185238,
      "learning_rate": 0.0006955364639669409,
      "loss": 3.6415,
      "step": 1705
    },
    {
      "epoch": 0.3911946801192387,
      "grad_norm": 0.15024391182249383,
      "learning_rate": 0.0006951946375817474,
      "loss": 3.5876,
      "step": 1706
    },
    {
      "epoch": 0.39142398532446687,
      "grad_norm": 0.15735277044895682,
      "learning_rate": 0.0006948527035284978,
      "loss": 3.5678,
      "step": 1707
    },
    {
      "epoch": 0.391653290529695,
      "grad_norm": 0.14455576282633614,
      "learning_rate": 0.0006945106619958002,
      "loss": 3.4871,
      "step": 1708
    },
    {
      "epoch": 0.3918825957349232,
      "grad_norm": 0.1365061632436101,
      "learning_rate": 0.0006941685131723225,
      "loss": 3.4427,
      "step": 1709
    },
    {
      "epoch": 0.39211190094015136,
      "grad_norm": 0.1353288990992737,
      "learning_rate": 0.0006938262572467919,
      "loss": 3.5433,
      "step": 1710
    },
    {
      "epoch": 0.3923412061453795,
      "grad_norm": 0.13649307755827306,
      "learning_rate": 0.0006934838944079943,
      "loss": 3.6337,
      "step": 1711
    },
    {
      "epoch": 0.39257051135060766,
      "grad_norm": 0.12901958129797111,
      "learning_rate": 0.0006931414248447749,
      "loss": 3.4846,
      "step": 1712
    },
    {
      "epoch": 0.3927998165558358,
      "grad_norm": 0.15455422436633637,
      "learning_rate": 0.0006927988487460378,
      "loss": 3.5284,
      "step": 1713
    },
    {
      "epoch": 0.39302912176106397,
      "grad_norm": 0.1319802731791124,
      "learning_rate": 0.000692456166300745,
      "loss": 3.6161,
      "step": 1714
    },
    {
      "epoch": 0.39325842696629215,
      "grad_norm": 0.13415705757049468,
      "learning_rate": 0.0006921133776979186,
      "loss": 3.397,
      "step": 1715
    },
    {
      "epoch": 0.3934877321715203,
      "grad_norm": 0.14690860136799505,
      "learning_rate": 0.0006917704831266381,
      "loss": 3.5391,
      "step": 1716
    },
    {
      "epoch": 0.39371703737674846,
      "grad_norm": 0.1271663356332884,
      "learning_rate": 0.0006914274827760418,
      "loss": 3.5118,
      "step": 1717
    },
    {
      "epoch": 0.39394634258197664,
      "grad_norm": 0.1294816845007828,
      "learning_rate": 0.0006910843768353266,
      "loss": 3.5032,
      "step": 1718
    },
    {
      "epoch": 0.39417564778720476,
      "grad_norm": 0.12611082836412413,
      "learning_rate": 0.0006907411654937475,
      "loss": 3.3508,
      "step": 1719
    },
    {
      "epoch": 0.39440495299243294,
      "grad_norm": 0.13355617720815283,
      "learning_rate": 0.0006903978489406174,
      "loss": 3.4576,
      "step": 1720
    },
    {
      "epoch": 0.39463425819766107,
      "grad_norm": 0.1455960440331674,
      "learning_rate": 0.0006900544273653075,
      "loss": 3.5708,
      "step": 1721
    },
    {
      "epoch": 0.39486356340288925,
      "grad_norm": 0.13661593745537712,
      "learning_rate": 0.0006897109009572468,
      "loss": 3.4181,
      "step": 1722
    },
    {
      "epoch": 0.39509286860811743,
      "grad_norm": 0.14657782656363147,
      "learning_rate": 0.000689367269905922,
      "loss": 3.4438,
      "step": 1723
    },
    {
      "epoch": 0.39532217381334556,
      "grad_norm": 0.13191762864338197,
      "learning_rate": 0.000689023534400878,
      "loss": 3.394,
      "step": 1724
    },
    {
      "epoch": 0.39555147901857374,
      "grad_norm": 0.14509494142780238,
      "learning_rate": 0.0006886796946317168,
      "loss": 3.4874,
      "step": 1725
    },
    {
      "epoch": 0.39578078422380186,
      "grad_norm": 0.1384686935478333,
      "learning_rate": 0.0006883357507880985,
      "loss": 3.4439,
      "step": 1726
    },
    {
      "epoch": 0.39601008942903004,
      "grad_norm": 0.1362861058417607,
      "learning_rate": 0.0006879917030597397,
      "loss": 3.4429,
      "step": 1727
    },
    {
      "epoch": 0.3962393946342582,
      "grad_norm": 0.1672496766059863,
      "learning_rate": 0.0006876475516364149,
      "loss": 3.5224,
      "step": 1728
    },
    {
      "epoch": 0.39646869983948635,
      "grad_norm": 0.1420697104004912,
      "learning_rate": 0.0006873032967079561,
      "loss": 3.479,
      "step": 1729
    },
    {
      "epoch": 0.39669800504471453,
      "grad_norm": 0.13872473648694067,
      "learning_rate": 0.0006869589384642516,
      "loss": 3.6334,
      "step": 1730
    },
    {
      "epoch": 0.39692731024994266,
      "grad_norm": 0.13097219082021797,
      "learning_rate": 0.0006866144770952474,
      "loss": 3.3712,
      "step": 1731
    },
    {
      "epoch": 0.39715661545517084,
      "grad_norm": 0.1506871326345639,
      "learning_rate": 0.000686269912790946,
      "loss": 3.5108,
      "step": 1732
    },
    {
      "epoch": 0.397385920660399,
      "grad_norm": 0.13363867543147243,
      "learning_rate": 0.0006859252457414067,
      "loss": 3.5224,
      "step": 1733
    },
    {
      "epoch": 0.39761522586562714,
      "grad_norm": 0.12601192741918313,
      "learning_rate": 0.0006855804761367456,
      "loss": 3.4625,
      "step": 1734
    },
    {
      "epoch": 0.3978445310708553,
      "grad_norm": 0.1558299772058803,
      "learning_rate": 0.0006852356041671351,
      "loss": 3.4778,
      "step": 1735
    },
    {
      "epoch": 0.39807383627608345,
      "grad_norm": 0.14759399307468374,
      "learning_rate": 0.0006848906300228046,
      "loss": 3.6643,
      "step": 1736
    },
    {
      "epoch": 0.39830314148131163,
      "grad_norm": 0.11953142873805458,
      "learning_rate": 0.0006845455538940394,
      "loss": 3.5281,
      "step": 1737
    },
    {
      "epoch": 0.3985324466865398,
      "grad_norm": 0.1310148840766096,
      "learning_rate": 0.0006842003759711811,
      "loss": 3.5529,
      "step": 1738
    },
    {
      "epoch": 0.39876175189176793,
      "grad_norm": 0.1452848386719086,
      "learning_rate": 0.0006838550964446276,
      "loss": 3.5189,
      "step": 1739
    },
    {
      "epoch": 0.3989910570969961,
      "grad_norm": 0.14502144652516047,
      "learning_rate": 0.0006835097155048329,
      "loss": 3.4631,
      "step": 1740
    },
    {
      "epoch": 0.39922036230222424,
      "grad_norm": 0.13103356013774334,
      "learning_rate": 0.0006831642333423067,
      "loss": 3.4005,
      "step": 1741
    },
    {
      "epoch": 0.3994496675074524,
      "grad_norm": 0.14460758712945593,
      "learning_rate": 0.0006828186501476144,
      "loss": 3.5167,
      "step": 1742
    },
    {
      "epoch": 0.3996789727126806,
      "grad_norm": 0.14376302138523162,
      "learning_rate": 0.000682472966111378,
      "loss": 3.4088,
      "step": 1743
    },
    {
      "epoch": 0.3999082779179087,
      "grad_norm": 0.15753694804834026,
      "learning_rate": 0.0006821271814242738,
      "loss": 3.4687,
      "step": 1744
    },
    {
      "epoch": 0.4001375831231369,
      "grad_norm": 0.1534944075291014,
      "learning_rate": 0.0006817812962770348,
      "loss": 3.5864,
      "step": 1745
    },
    {
      "epoch": 0.40036688832836503,
      "grad_norm": 0.1335923116284337,
      "learning_rate": 0.0006814353108604487,
      "loss": 3.484,
      "step": 1746
    },
    {
      "epoch": 0.4005961935335932,
      "grad_norm": 0.17848213710480915,
      "learning_rate": 0.0006810892253653589,
      "loss": 3.6423,
      "step": 1747
    },
    {
      "epoch": 0.4008254987388214,
      "grad_norm": 0.14274883553330034,
      "learning_rate": 0.0006807430399826637,
      "loss": 3.5872,
      "step": 1748
    },
    {
      "epoch": 0.4010548039440495,
      "grad_norm": 0.1522906730069557,
      "learning_rate": 0.0006803967549033167,
      "loss": 3.5681,
      "step": 1749
    },
    {
      "epoch": 0.4012841091492777,
      "grad_norm": 0.14651200220937957,
      "learning_rate": 0.0006800503703183265,
      "loss": 3.707,
      "step": 1750
    },
    {
      "epoch": 0.4015134143545058,
      "grad_norm": 0.13700640532644262,
      "learning_rate": 0.0006797038864187564,
      "loss": 3.5332,
      "step": 1751
    },
    {
      "epoch": 0.401742719559734,
      "grad_norm": 0.15327967895253683,
      "learning_rate": 0.0006793573033957248,
      "loss": 3.5688,
      "step": 1752
    },
    {
      "epoch": 0.4019720247649622,
      "grad_norm": 0.13637263563738533,
      "learning_rate": 0.0006790106214404043,
      "loss": 3.6908,
      "step": 1753
    },
    {
      "epoch": 0.4022013299701903,
      "grad_norm": 0.13825004400887814,
      "learning_rate": 0.0006786638407440224,
      "loss": 3.433,
      "step": 1754
    },
    {
      "epoch": 0.4024306351754185,
      "grad_norm": 0.14494698843879705,
      "learning_rate": 0.0006783169614978614,
      "loss": 3.4939,
      "step": 1755
    },
    {
      "epoch": 0.4026599403806466,
      "grad_norm": 0.15498095939061643,
      "learning_rate": 0.000677969983893257,
      "loss": 3.683,
      "step": 1756
    },
    {
      "epoch": 0.4028892455858748,
      "grad_norm": 0.13164547831751994,
      "learning_rate": 0.0006776229081216001,
      "loss": 3.5428,
      "step": 1757
    },
    {
      "epoch": 0.403118550791103,
      "grad_norm": 0.13111138841725595,
      "learning_rate": 0.0006772757343743352,
      "loss": 3.594,
      "step": 1758
    },
    {
      "epoch": 0.4033478559963311,
      "grad_norm": 0.12046845717732564,
      "learning_rate": 0.0006769284628429611,
      "loss": 3.4225,
      "step": 1759
    },
    {
      "epoch": 0.4035771612015593,
      "grad_norm": 0.14127815893130066,
      "learning_rate": 0.0006765810937190307,
      "loss": 3.601,
      "step": 1760
    },
    {
      "epoch": 0.4038064664067874,
      "grad_norm": 0.13435801316518028,
      "learning_rate": 0.0006762336271941498,
      "loss": 3.4921,
      "step": 1761
    },
    {
      "epoch": 0.4040357716120156,
      "grad_norm": 0.13630719234376143,
      "learning_rate": 0.0006758860634599792,
      "loss": 3.5267,
      "step": 1762
    },
    {
      "epoch": 0.4042650768172438,
      "grad_norm": 0.12877841416791877,
      "learning_rate": 0.0006755384027082326,
      "loss": 3.4443,
      "step": 1763
    },
    {
      "epoch": 0.4044943820224719,
      "grad_norm": 0.1308141278925634,
      "learning_rate": 0.0006751906451306774,
      "loss": 3.442,
      "step": 1764
    },
    {
      "epoch": 0.4047236872277001,
      "grad_norm": 0.13494604206826927,
      "learning_rate": 0.0006748427909191342,
      "loss": 3.5023,
      "step": 1765
    },
    {
      "epoch": 0.4049529924329282,
      "grad_norm": 0.14134069317265865,
      "learning_rate": 0.0006744948402654771,
      "loss": 3.5874,
      "step": 1766
    },
    {
      "epoch": 0.4051822976381564,
      "grad_norm": 0.12187799172691766,
      "learning_rate": 0.0006741467933616335,
      "loss": 3.5599,
      "step": 1767
    },
    {
      "epoch": 0.40541160284338457,
      "grad_norm": 0.1383914862567864,
      "learning_rate": 0.0006737986503995832,
      "loss": 3.573,
      "step": 1768
    },
    {
      "epoch": 0.4056409080486127,
      "grad_norm": 0.12397255284210726,
      "learning_rate": 0.0006734504115713604,
      "loss": 3.4608,
      "step": 1769
    },
    {
      "epoch": 0.4058702132538409,
      "grad_norm": 0.1316707058210004,
      "learning_rate": 0.0006731020770690505,
      "loss": 3.519,
      "step": 1770
    },
    {
      "epoch": 0.406099518459069,
      "grad_norm": 0.11631476021621165,
      "learning_rate": 0.0006727536470847932,
      "loss": 3.6446,
      "step": 1771
    },
    {
      "epoch": 0.4063288236642972,
      "grad_norm": 0.12493223586315153,
      "learning_rate": 0.0006724051218107795,
      "loss": 3.6644,
      "step": 1772
    },
    {
      "epoch": 0.40655812886952536,
      "grad_norm": 0.12719828045200776,
      "learning_rate": 0.000672056501439254,
      "loss": 3.4313,
      "step": 1773
    },
    {
      "epoch": 0.4067874340747535,
      "grad_norm": 0.13075975591802766,
      "learning_rate": 0.0006717077861625134,
      "loss": 3.4993,
      "step": 1774
    },
    {
      "epoch": 0.40701673927998167,
      "grad_norm": 0.13286767211132086,
      "learning_rate": 0.0006713589761729063,
      "loss": 3.5316,
      "step": 1775
    },
    {
      "epoch": 0.4072460444852098,
      "grad_norm": 0.13175824117122287,
      "learning_rate": 0.0006710100716628344,
      "loss": 3.5742,
      "step": 1776
    },
    {
      "epoch": 0.40747534969043797,
      "grad_norm": 0.13102482264498505,
      "learning_rate": 0.0006706610728247508,
      "loss": 3.4937,
      "step": 1777
    },
    {
      "epoch": 0.40770465489566615,
      "grad_norm": 0.12513712682934014,
      "learning_rate": 0.0006703119798511612,
      "loss": 3.4389,
      "step": 1778
    },
    {
      "epoch": 0.4079339601008943,
      "grad_norm": 0.15091215401485286,
      "learning_rate": 0.0006699627929346227,
      "loss": 3.7245,
      "step": 1779
    },
    {
      "epoch": 0.40816326530612246,
      "grad_norm": 0.14195567724696875,
      "learning_rate": 0.0006696135122677447,
      "loss": 3.5425,
      "step": 1780
    },
    {
      "epoch": 0.4083925705113506,
      "grad_norm": 0.12479661317457874,
      "learning_rate": 0.0006692641380431879,
      "loss": 3.4158,
      "step": 1781
    },
    {
      "epoch": 0.40862187571657876,
      "grad_norm": 0.12700437114091162,
      "learning_rate": 0.0006689146704536646,
      "loss": 3.6067,
      "step": 1782
    },
    {
      "epoch": 0.40885118092180694,
      "grad_norm": 0.13093661718455793,
      "learning_rate": 0.0006685651096919393,
      "loss": 3.6175,
      "step": 1783
    },
    {
      "epoch": 0.40908048612703507,
      "grad_norm": 0.1316469765778252,
      "learning_rate": 0.0006682154559508269,
      "loss": 3.4901,
      "step": 1784
    },
    {
      "epoch": 0.40930979133226325,
      "grad_norm": 0.13103047043647242,
      "learning_rate": 0.0006678657094231944,
      "loss": 3.5945,
      "step": 1785
    },
    {
      "epoch": 0.4095390965374914,
      "grad_norm": 0.13283111194622665,
      "learning_rate": 0.0006675158703019593,
      "loss": 3.6335,
      "step": 1786
    },
    {
      "epoch": 0.40976840174271956,
      "grad_norm": 0.12562192152479507,
      "learning_rate": 0.0006671659387800909,
      "loss": 3.4582,
      "step": 1787
    },
    {
      "epoch": 0.40999770694794774,
      "grad_norm": 0.13149877852859077,
      "learning_rate": 0.0006668159150506089,
      "loss": 3.6156,
      "step": 1788
    },
    {
      "epoch": 0.41022701215317586,
      "grad_norm": 0.14536525527966693,
      "learning_rate": 0.000666465799306584,
      "loss": 3.5933,
      "step": 1789
    },
    {
      "epoch": 0.41045631735840404,
      "grad_norm": 0.1380446814594201,
      "learning_rate": 0.0006661155917411378,
      "loss": 3.5363,
      "step": 1790
    },
    {
      "epoch": 0.41068562256363217,
      "grad_norm": 0.12084574949936566,
      "learning_rate": 0.0006657652925474423,
      "loss": 3.533,
      "step": 1791
    },
    {
      "epoch": 0.41091492776886035,
      "grad_norm": 0.14240210593692024,
      "learning_rate": 0.0006654149019187205,
      "loss": 3.5105,
      "step": 1792
    },
    {
      "epoch": 0.41114423297408853,
      "grad_norm": 0.1420252838972675,
      "learning_rate": 0.000665064420048245,
      "loss": 3.4634,
      "step": 1793
    },
    {
      "epoch": 0.41137353817931666,
      "grad_norm": 0.13915280270429095,
      "learning_rate": 0.0006647138471293399,
      "loss": 3.4194,
      "step": 1794
    },
    {
      "epoch": 0.41160284338454484,
      "grad_norm": 0.14672951976311133,
      "learning_rate": 0.0006643631833553785,
      "loss": 3.559,
      "step": 1795
    },
    {
      "epoch": 0.41183214858977296,
      "grad_norm": 0.1425071987227489,
      "learning_rate": 0.0006640124289197845,
      "loss": 3.6348,
      "step": 1796
    },
    {
      "epoch": 0.41206145379500114,
      "grad_norm": 0.14525431706818984,
      "learning_rate": 0.000663661584016032,
      "loss": 3.5011,
      "step": 1797
    },
    {
      "epoch": 0.4122907590002293,
      "grad_norm": 0.14639229133222778,
      "learning_rate": 0.0006633106488376445,
      "loss": 3.5555,
      "step": 1798
    },
    {
      "epoch": 0.41252006420545745,
      "grad_norm": 0.14951025624739397,
      "learning_rate": 0.0006629596235781957,
      "loss": 3.5497,
      "step": 1799
    },
    {
      "epoch": 0.41274936941068563,
      "grad_norm": 0.14241735207428827,
      "learning_rate": 0.0006626085084313083,
      "loss": 3.6944,
      "step": 1800
    },
    {
      "epoch": 0.41297867461591375,
      "grad_norm": 0.14561346190119545,
      "learning_rate": 0.0006622573035906556,
      "loss": 3.3753,
      "step": 1801
    },
    {
      "epoch": 0.41320797982114194,
      "grad_norm": 0.14202334649154638,
      "learning_rate": 0.0006619060092499597,
      "loss": 3.5192,
      "step": 1802
    },
    {
      "epoch": 0.4134372850263701,
      "grad_norm": 0.1427122315713031,
      "learning_rate": 0.0006615546256029921,
      "loss": 3.5039,
      "step": 1803
    },
    {
      "epoch": 0.41366659023159824,
      "grad_norm": 0.14423232412886666,
      "learning_rate": 0.0006612031528435737,
      "loss": 3.6165,
      "step": 1804
    },
    {
      "epoch": 0.4138958954368264,
      "grad_norm": 0.15761269895713007,
      "learning_rate": 0.0006608515911655743,
      "loss": 3.5127,
      "step": 1805
    },
    {
      "epoch": 0.41412520064205455,
      "grad_norm": 0.1328830593694952,
      "learning_rate": 0.0006604999407629137,
      "loss": 3.6051,
      "step": 1806
    },
    {
      "epoch": 0.41435450584728273,
      "grad_norm": 0.1390582827939829,
      "learning_rate": 0.0006601482018295591,
      "loss": 3.5132,
      "step": 1807
    },
    {
      "epoch": 0.4145838110525109,
      "grad_norm": 0.11897118047612894,
      "learning_rate": 0.0006597963745595276,
      "loss": 3.5394,
      "step": 1808
    },
    {
      "epoch": 0.41481311625773903,
      "grad_norm": 0.1562914617089704,
      "learning_rate": 0.0006594444591468851,
      "loss": 3.5118,
      "step": 1809
    },
    {
      "epoch": 0.4150424214629672,
      "grad_norm": 0.15671493797173425,
      "learning_rate": 0.0006590924557857454,
      "loss": 3.2909,
      "step": 1810
    },
    {
      "epoch": 0.4152717266681954,
      "grad_norm": 0.12677838683429657,
      "learning_rate": 0.0006587403646702713,
      "loss": 3.5648,
      "step": 1811
    },
    {
      "epoch": 0.4155010318734235,
      "grad_norm": 0.13002506844952644,
      "learning_rate": 0.0006583881859946739,
      "loss": 3.5176,
      "step": 1812
    },
    {
      "epoch": 0.4157303370786517,
      "grad_norm": 0.12298507105842083,
      "learning_rate": 0.0006580359199532126,
      "loss": 3.6618,
      "step": 1813
    },
    {
      "epoch": 0.4159596422838798,
      "grad_norm": 0.15261490989216037,
      "learning_rate": 0.0006576835667401953,
      "loss": 3.5139,
      "step": 1814
    },
    {
      "epoch": 0.416188947489108,
      "grad_norm": 0.1390249772418868,
      "learning_rate": 0.000657331126549977,
      "loss": 3.5422,
      "step": 1815
    },
    {
      "epoch": 0.4164182526943362,
      "grad_norm": 0.1392141730123498,
      "learning_rate": 0.000656978599576962,
      "loss": 3.6062,
      "step": 1816
    },
    {
      "epoch": 0.4166475578995643,
      "grad_norm": 0.13280061824605632,
      "learning_rate": 0.0006566259860156014,
      "loss": 3.4082,
      "step": 1817
    },
    {
      "epoch": 0.4168768631047925,
      "grad_norm": 0.12385378056938968,
      "learning_rate": 0.0006562732860603949,
      "loss": 3.5958,
      "step": 1818
    },
    {
      "epoch": 0.4171061683100206,
      "grad_norm": 0.13195765326352632,
      "learning_rate": 0.0006559204999058888,
      "loss": 3.4425,
      "step": 1819
    },
    {
      "epoch": 0.4173354735152488,
      "grad_norm": 0.12996328800898407,
      "learning_rate": 0.0006555676277466782,
      "loss": 3.5293,
      "step": 1820
    },
    {
      "epoch": 0.417564778720477,
      "grad_norm": 0.1282816839897133,
      "learning_rate": 0.0006552146697774049,
      "loss": 3.4685,
      "step": 1821
    },
    {
      "epoch": 0.4177940839257051,
      "grad_norm": 0.12581003329004842,
      "learning_rate": 0.0006548616261927578,
      "loss": 3.4561,
      "step": 1822
    },
    {
      "epoch": 0.4180233891309333,
      "grad_norm": 0.1290669144158653,
      "learning_rate": 0.0006545084971874737,
      "loss": 3.4278,
      "step": 1823
    },
    {
      "epoch": 0.4182526943361614,
      "grad_norm": 0.14825086168379314,
      "learning_rate": 0.0006541552829563362,
      "loss": 3.5753,
      "step": 1824
    },
    {
      "epoch": 0.4184819995413896,
      "grad_norm": 0.1288257701678689,
      "learning_rate": 0.0006538019836941758,
      "loss": 3.3853,
      "step": 1825
    },
    {
      "epoch": 0.4187113047466178,
      "grad_norm": 0.1297529900245684,
      "learning_rate": 0.0006534485995958699,
      "loss": 3.4623,
      "step": 1826
    },
    {
      "epoch": 0.4189406099518459,
      "grad_norm": 0.1281120970144617,
      "learning_rate": 0.0006530951308563431,
      "loss": 3.634,
      "step": 1827
    },
    {
      "epoch": 0.4191699151570741,
      "grad_norm": 0.1415921383009029,
      "learning_rate": 0.0006527415776705662,
      "loss": 3.5504,
      "step": 1828
    },
    {
      "epoch": 0.4193992203623022,
      "grad_norm": 0.1273401948037744,
      "learning_rate": 0.0006523879402335567,
      "loss": 3.5321,
      "step": 1829
    },
    {
      "epoch": 0.4196285255675304,
      "grad_norm": 0.13684666940709428,
      "learning_rate": 0.0006520342187403789,
      "loss": 3.5904,
      "step": 1830
    },
    {
      "epoch": 0.41985783077275857,
      "grad_norm": 0.13736358757808476,
      "learning_rate": 0.0006516804133861429,
      "loss": 3.543,
      "step": 1831
    },
    {
      "epoch": 0.4200871359779867,
      "grad_norm": 0.13712557344757623,
      "learning_rate": 0.0006513265243660057,
      "loss": 3.4865,
      "step": 1832
    },
    {
      "epoch": 0.4203164411832149,
      "grad_norm": 0.1326063735412298,
      "learning_rate": 0.0006509725518751698,
      "loss": 3.5693,
      "step": 1833
    },
    {
      "epoch": 0.420545746388443,
      "grad_norm": 0.13187635938972386,
      "learning_rate": 0.0006506184961088844,
      "loss": 3.4874,
      "step": 1834
    },
    {
      "epoch": 0.4207750515936712,
      "grad_norm": 0.13110846903057732,
      "learning_rate": 0.0006502643572624438,
      "loss": 3.4976,
      "step": 1835
    },
    {
      "epoch": 0.42100435679889936,
      "grad_norm": 0.1311759944312106,
      "learning_rate": 0.0006499101355311891,
      "loss": 3.6606,
      "step": 1836
    },
    {
      "epoch": 0.4212336620041275,
      "grad_norm": 0.15159836049065425,
      "learning_rate": 0.0006495558311105064,
      "loss": 3.4801,
      "step": 1837
    },
    {
      "epoch": 0.42146296720935567,
      "grad_norm": 0.12833529015574333,
      "learning_rate": 0.0006492014441958277,
      "loss": 3.5384,
      "step": 1838
    },
    {
      "epoch": 0.4216922724145838,
      "grad_norm": 0.1393815368712179,
      "learning_rate": 0.0006488469749826305,
      "loss": 3.6759,
      "step": 1839
    },
    {
      "epoch": 0.42192157761981197,
      "grad_norm": 0.13840007880089913,
      "learning_rate": 0.0006484924236664376,
      "loss": 3.4629,
      "step": 1840
    },
    {
      "epoch": 0.42215088282504015,
      "grad_norm": 0.1370635313552814,
      "learning_rate": 0.000648137790442817,
      "loss": 3.5708,
      "step": 1841
    },
    {
      "epoch": 0.4223801880302683,
      "grad_norm": 0.14307923943388218,
      "learning_rate": 0.0006477830755073822,
      "loss": 3.5724,
      "step": 1842
    },
    {
      "epoch": 0.42260949323549646,
      "grad_norm": 0.14455434941299034,
      "learning_rate": 0.0006474282790557916,
      "loss": 3.5573,
      "step": 1843
    },
    {
      "epoch": 0.4228387984407246,
      "grad_norm": 0.13649003285676792,
      "learning_rate": 0.0006470734012837485,
      "loss": 3.3932,
      "step": 1844
    },
    {
      "epoch": 0.42306810364595276,
      "grad_norm": 0.12866120291211272,
      "learning_rate": 0.000646718442387001,
      "loss": 3.4975,
      "step": 1845
    },
    {
      "epoch": 0.42329740885118095,
      "grad_norm": 0.1341388215618962,
      "learning_rate": 0.0006463634025613422,
      "loss": 3.466,
      "step": 1846
    },
    {
      "epoch": 0.42352671405640907,
      "grad_norm": 0.12886200139277215,
      "learning_rate": 0.0006460082820026094,
      "loss": 3.5292,
      "step": 1847
    },
    {
      "epoch": 0.42375601926163725,
      "grad_norm": 0.1413595086148348,
      "learning_rate": 0.0006456530809066852,
      "loss": 3.5298,
      "step": 1848
    },
    {
      "epoch": 0.4239853244668654,
      "grad_norm": 0.15563007528348244,
      "learning_rate": 0.0006452977994694959,
      "loss": 3.4364,
      "step": 1849
    },
    {
      "epoch": 0.42421462967209356,
      "grad_norm": 0.14116605708707805,
      "learning_rate": 0.0006449424378870123,
      "loss": 3.495,
      "step": 1850
    },
    {
      "epoch": 0.42444393487732174,
      "grad_norm": 0.14103046941796588,
      "learning_rate": 0.0006445869963552496,
      "loss": 3.629,
      "step": 1851
    },
    {
      "epoch": 0.42467324008254986,
      "grad_norm": 0.146933541877603,
      "learning_rate": 0.0006442314750702666,
      "loss": 3.6241,
      "step": 1852
    },
    {
      "epoch": 0.42490254528777804,
      "grad_norm": 0.12970620076234168,
      "learning_rate": 0.0006438758742281672,
      "loss": 3.5169,
      "step": 1853
    },
    {
      "epoch": 0.42513185049300617,
      "grad_norm": 0.13606631530859872,
      "learning_rate": 0.0006435201940250977,
      "loss": 3.6435,
      "step": 1854
    },
    {
      "epoch": 0.42536115569823435,
      "grad_norm": 0.13246489024662042,
      "learning_rate": 0.0006431644346572495,
      "loss": 3.5616,
      "step": 1855
    },
    {
      "epoch": 0.42559046090346253,
      "grad_norm": 0.14599325989643816,
      "learning_rate": 0.0006428085963208567,
      "loss": 3.4378,
      "step": 1856
    },
    {
      "epoch": 0.42581976610869066,
      "grad_norm": 0.140362464956343,
      "learning_rate": 0.0006424526792121974,
      "loss": 3.6334,
      "step": 1857
    },
    {
      "epoch": 0.42604907131391884,
      "grad_norm": 0.1262022863789821,
      "learning_rate": 0.0006420966835275933,
      "loss": 3.425,
      "step": 1858
    },
    {
      "epoch": 0.42627837651914696,
      "grad_norm": 0.13119233109514394,
      "learning_rate": 0.0006417406094634089,
      "loss": 3.6054,
      "step": 1859
    },
    {
      "epoch": 0.42650768172437514,
      "grad_norm": 0.12799378426796312,
      "learning_rate": 0.0006413844572160527,
      "loss": 3.5381,
      "step": 1860
    },
    {
      "epoch": 0.4267369869296033,
      "grad_norm": 0.1352554228200962,
      "learning_rate": 0.0006410282269819756,
      "loss": 3.401,
      "step": 1861
    },
    {
      "epoch": 0.42696629213483145,
      "grad_norm": 0.14206516346405645,
      "learning_rate": 0.000640671918957672,
      "loss": 3.5176,
      "step": 1862
    },
    {
      "epoch": 0.42719559734005963,
      "grad_norm": 0.13564059674526907,
      "learning_rate": 0.0006403155333396787,
      "loss": 3.5023,
      "step": 1863
    },
    {
      "epoch": 0.42742490254528775,
      "grad_norm": 0.1234378786194567,
      "learning_rate": 0.0006399590703245759,
      "loss": 3.5138,
      "step": 1864
    },
    {
      "epoch": 0.42765420775051594,
      "grad_norm": 0.12574985201545189,
      "learning_rate": 0.0006396025301089863,
      "loss": 3.6788,
      "step": 1865
    },
    {
      "epoch": 0.4278835129557441,
      "grad_norm": 0.12918889504839937,
      "learning_rate": 0.0006392459128895747,
      "loss": 3.4658,
      "step": 1866
    },
    {
      "epoch": 0.42811281816097224,
      "grad_norm": 0.1293616883230675,
      "learning_rate": 0.0006388892188630493,
      "loss": 3.523,
      "step": 1867
    },
    {
      "epoch": 0.4283421233662004,
      "grad_norm": 0.1397694014823953,
      "learning_rate": 0.0006385324482261597,
      "loss": 3.5101,
      "step": 1868
    },
    {
      "epoch": 0.42857142857142855,
      "grad_norm": 0.13488967300319504,
      "learning_rate": 0.0006381756011756982,
      "loss": 3.8001,
      "step": 1869
    },
    {
      "epoch": 0.42880073377665673,
      "grad_norm": 0.12394485399360881,
      "learning_rate": 0.0006378186779084996,
      "loss": 3.4989,
      "step": 1870
    },
    {
      "epoch": 0.4290300389818849,
      "grad_norm": 0.1277574246606033,
      "learning_rate": 0.0006374616786214403,
      "loss": 3.5513,
      "step": 1871
    },
    {
      "epoch": 0.42925934418711303,
      "grad_norm": 0.13302337645966644,
      "learning_rate": 0.0006371046035114384,
      "loss": 3.5371,
      "step": 1872
    },
    {
      "epoch": 0.4294886493923412,
      "grad_norm": 0.1353427871185621,
      "learning_rate": 0.0006367474527754544,
      "loss": 3.4941,
      "step": 1873
    },
    {
      "epoch": 0.42971795459756934,
      "grad_norm": 0.1287249349894737,
      "learning_rate": 0.0006363902266104902,
      "loss": 3.5168,
      "step": 1874
    },
    {
      "epoch": 0.4299472598027975,
      "grad_norm": 0.12380513310825225,
      "learning_rate": 0.0006360329252135894,
      "loss": 3.3829,
      "step": 1875
    },
    {
      "epoch": 0.4301765650080257,
      "grad_norm": 0.11872962224905112,
      "learning_rate": 0.0006356755487818371,
      "loss": 3.5036,
      "step": 1876
    },
    {
      "epoch": 0.4304058702132538,
      "grad_norm": 0.12511988774574895,
      "learning_rate": 0.0006353180975123595,
      "loss": 3.4664,
      "step": 1877
    },
    {
      "epoch": 0.430635175418482,
      "grad_norm": 0.13001958544420197,
      "learning_rate": 0.0006349605716023246,
      "loss": 3.4694,
      "step": 1878
    },
    {
      "epoch": 0.43086448062371013,
      "grad_norm": 0.14386638518778383,
      "learning_rate": 0.0006346029712489413,
      "loss": 3.5891,
      "step": 1879
    },
    {
      "epoch": 0.4310937858289383,
      "grad_norm": 0.12450143548299471,
      "learning_rate": 0.0006342452966494593,
      "loss": 3.3659,
      "step": 1880
    },
    {
      "epoch": 0.4313230910341665,
      "grad_norm": 0.12948456808837921,
      "learning_rate": 0.0006338875480011698,
      "loss": 3.4158,
      "step": 1881
    },
    {
      "epoch": 0.4315523962393946,
      "grad_norm": 0.13930135808247693,
      "learning_rate": 0.0006335297255014043,
      "loss": 3.5717,
      "step": 1882
    },
    {
      "epoch": 0.4317817014446228,
      "grad_norm": 0.1374951242166573,
      "learning_rate": 0.0006331718293475357,
      "loss": 3.4887,
      "step": 1883
    },
    {
      "epoch": 0.4320110066498509,
      "grad_norm": 0.1549608864118214,
      "learning_rate": 0.0006328138597369768,
      "loss": 3.4964,
      "step": 1884
    },
    {
      "epoch": 0.4322403118550791,
      "grad_norm": 0.12163927415898253,
      "learning_rate": 0.0006324558168671811,
      "loss": 3.4564,
      "step": 1885
    },
    {
      "epoch": 0.4324696170603073,
      "grad_norm": 0.13870717453399467,
      "learning_rate": 0.0006320977009356432,
      "loss": 3.4821,
      "step": 1886
    },
    {
      "epoch": 0.4326989222655354,
      "grad_norm": 0.1455017712648699,
      "learning_rate": 0.0006317395121398968,
      "loss": 3.4432,
      "step": 1887
    },
    {
      "epoch": 0.4329282274707636,
      "grad_norm": 0.1416021181391573,
      "learning_rate": 0.0006313812506775169,
      "loss": 3.6491,
      "step": 1888
    },
    {
      "epoch": 0.4331575326759917,
      "grad_norm": 0.1424340455555957,
      "learning_rate": 0.0006310229167461179,
      "loss": 3.5178,
      "step": 1889
    },
    {
      "epoch": 0.4333868378812199,
      "grad_norm": 0.13762110272307448,
      "learning_rate": 0.0006306645105433546,
      "loss": 3.4446,
      "step": 1890
    },
    {
      "epoch": 0.4336161430864481,
      "grad_norm": 0.12921590675501302,
      "learning_rate": 0.0006303060322669214,
      "loss": 3.3777,
      "step": 1891
    },
    {
      "epoch": 0.4338454482916762,
      "grad_norm": 0.13513672581666475,
      "learning_rate": 0.0006299474821145523,
      "loss": 3.4405,
      "step": 1892
    },
    {
      "epoch": 0.4340747534969044,
      "grad_norm": 0.14705151348019266,
      "learning_rate": 0.0006295888602840214,
      "loss": 3.5064,
      "step": 1893
    },
    {
      "epoch": 0.4343040587021325,
      "grad_norm": 0.1380491499497484,
      "learning_rate": 0.0006292301669731421,
      "loss": 3.4896,
      "step": 1894
    },
    {
      "epoch": 0.4345333639073607,
      "grad_norm": 0.13746182871797588,
      "learning_rate": 0.0006288714023797671,
      "loss": 3.5241,
      "step": 1895
    },
    {
      "epoch": 0.4347626691125889,
      "grad_norm": 0.15223615172777016,
      "learning_rate": 0.0006285125667017886,
      "loss": 3.2972,
      "step": 1896
    },
    {
      "epoch": 0.434991974317817,
      "grad_norm": 0.1369686661684975,
      "learning_rate": 0.000628153660137138,
      "loss": 3.5499,
      "step": 1897
    },
    {
      "epoch": 0.4352212795230452,
      "grad_norm": 0.13399974117421762,
      "learning_rate": 0.000627794682883786,
      "loss": 3.5009,
      "step": 1898
    },
    {
      "epoch": 0.4354505847282733,
      "grad_norm": 0.14775683259096017,
      "learning_rate": 0.0006274356351397413,
      "loss": 3.6217,
      "step": 1899
    },
    {
      "epoch": 0.4356798899335015,
      "grad_norm": 0.1481811673380147,
      "learning_rate": 0.000627076517103053,
      "loss": 3.5,
      "step": 1900
    },
    {
      "epoch": 0.43590919513872967,
      "grad_norm": 0.13155137165583042,
      "learning_rate": 0.0006267173289718079,
      "loss": 3.4652,
      "step": 1901
    },
    {
      "epoch": 0.4361385003439578,
      "grad_norm": 0.14525788636073786,
      "learning_rate": 0.0006263580709441319,
      "loss": 3.6361,
      "step": 1902
    },
    {
      "epoch": 0.43636780554918597,
      "grad_norm": 0.1312818218555995,
      "learning_rate": 0.000625998743218189,
      "loss": 3.4767,
      "step": 1903
    },
    {
      "epoch": 0.43659711075441415,
      "grad_norm": 0.14205215753487063,
      "learning_rate": 0.0006256393459921824,
      "loss": 3.5669,
      "step": 1904
    },
    {
      "epoch": 0.4368264159596423,
      "grad_norm": 0.12551104185886794,
      "learning_rate": 0.000625279879464353,
      "loss": 3.3567,
      "step": 1905
    },
    {
      "epoch": 0.43705572116487046,
      "grad_norm": 0.1388868835117466,
      "learning_rate": 0.0006249203438329799,
      "loss": 3.4431,
      "step": 1906
    },
    {
      "epoch": 0.4372850263700986,
      "grad_norm": 0.1315653043579689,
      "learning_rate": 0.000624560739296381,
      "loss": 3.4836,
      "step": 1907
    },
    {
      "epoch": 0.43751433157532676,
      "grad_norm": 0.12700526312457094,
      "learning_rate": 0.0006242010660529115,
      "loss": 3.5795,
      "step": 1908
    },
    {
      "epoch": 0.43774363678055495,
      "grad_norm": 0.13412224131460793,
      "learning_rate": 0.0006238413243009648,
      "loss": 3.4819,
      "step": 1909
    },
    {
      "epoch": 0.43797294198578307,
      "grad_norm": 0.1522259869119444,
      "learning_rate": 0.0006234815142389719,
      "loss": 3.5,
      "step": 1910
    },
    {
      "epoch": 0.43820224719101125,
      "grad_norm": 0.13255669446565782,
      "learning_rate": 0.000623121636065402,
      "loss": 3.5768,
      "step": 1911
    },
    {
      "epoch": 0.4384315523962394,
      "grad_norm": 0.12603162574159882,
      "learning_rate": 0.0006227616899787612,
      "loss": 3.4847,
      "step": 1912
    },
    {
      "epoch": 0.43866085760146756,
      "grad_norm": 0.1293381004168608,
      "learning_rate": 0.0006224016761775933,
      "loss": 3.4313,
      "step": 1913
    },
    {
      "epoch": 0.43889016280669574,
      "grad_norm": 0.13462444877339297,
      "learning_rate": 0.0006220415948604796,
      "loss": 3.531,
      "step": 1914
    },
    {
      "epoch": 0.43911946801192386,
      "grad_norm": 0.14065727261202474,
      "learning_rate": 0.0006216814462260386,
      "loss": 3.4776,
      "step": 1915
    },
    {
      "epoch": 0.43934877321715204,
      "grad_norm": 0.1321257547712339,
      "learning_rate": 0.0006213212304729259,
      "loss": 3.6438,
      "step": 1916
    },
    {
      "epoch": 0.43957807842238017,
      "grad_norm": 0.14294501866899964,
      "learning_rate": 0.0006209609477998338,
      "loss": 3.5377,
      "step": 1917
    },
    {
      "epoch": 0.43980738362760835,
      "grad_norm": 0.16867422666451837,
      "learning_rate": 0.0006206005984054923,
      "loss": 3.4702,
      "step": 1918
    },
    {
      "epoch": 0.44003668883283653,
      "grad_norm": 0.13989417207500127,
      "learning_rate": 0.0006202401824886674,
      "loss": 3.4732,
      "step": 1919
    },
    {
      "epoch": 0.44026599403806466,
      "grad_norm": 0.1581797414380702,
      "learning_rate": 0.0006198797002481622,
      "loss": 3.6604,
      "step": 1920
    },
    {
      "epoch": 0.44049529924329284,
      "grad_norm": 0.1350642156679117,
      "learning_rate": 0.0006195191518828162,
      "loss": 3.6873,
      "step": 1921
    },
    {
      "epoch": 0.44072460444852096,
      "grad_norm": 0.13670919246489277,
      "learning_rate": 0.0006191585375915056,
      "loss": 3.3184,
      "step": 1922
    },
    {
      "epoch": 0.44095390965374914,
      "grad_norm": 0.14633137511550315,
      "learning_rate": 0.0006187978575731427,
      "loss": 3.5621,
      "step": 1923
    },
    {
      "epoch": 0.4411832148589773,
      "grad_norm": 0.1372680963369642,
      "learning_rate": 0.0006184371120266761,
      "loss": 3.6469,
      "step": 1924
    },
    {
      "epoch": 0.44141252006420545,
      "grad_norm": 0.148167162911114,
      "learning_rate": 0.0006180763011510911,
      "loss": 3.6635,
      "step": 1925
    },
    {
      "epoch": 0.44164182526943363,
      "grad_norm": 0.14968878065871624,
      "learning_rate": 0.0006177154251454082,
      "loss": 3.5668,
      "step": 1926
    },
    {
      "epoch": 0.44187113047466176,
      "grad_norm": 0.12784578404501498,
      "learning_rate": 0.000617354484208684,
      "loss": 3.3974,
      "step": 1927
    },
    {
      "epoch": 0.44210043567988994,
      "grad_norm": 0.12763270280160888,
      "learning_rate": 0.0006169934785400116,
      "loss": 3.4463,
      "step": 1928
    },
    {
      "epoch": 0.4423297408851181,
      "grad_norm": 0.1362566348775628,
      "learning_rate": 0.0006166324083385189,
      "loss": 3.5071,
      "step": 1929
    },
    {
      "epoch": 0.44255904609034624,
      "grad_norm": 0.1305499696772877,
      "learning_rate": 0.0006162712738033699,
      "loss": 3.5913,
      "step": 1930
    },
    {
      "epoch": 0.4427883512955744,
      "grad_norm": 0.1365308507013613,
      "learning_rate": 0.0006159100751337642,
      "loss": 3.5846,
      "step": 1931
    },
    {
      "epoch": 0.44301765650080255,
      "grad_norm": 0.13449265012227407,
      "learning_rate": 0.0006155488125289363,
      "loss": 3.5351,
      "step": 1932
    },
    {
      "epoch": 0.44324696170603073,
      "grad_norm": 0.12847786443981427,
      "learning_rate": 0.0006151874861881565,
      "loss": 3.4552,
      "step": 1933
    },
    {
      "epoch": 0.4434762669112589,
      "grad_norm": 0.1473226082269585,
      "learning_rate": 0.0006148260963107297,
      "loss": 3.4822,
      "step": 1934
    },
    {
      "epoch": 0.44370557211648703,
      "grad_norm": 0.12987166447246073,
      "learning_rate": 0.0006144646430959964,
      "loss": 3.6323,
      "step": 1935
    },
    {
      "epoch": 0.4439348773217152,
      "grad_norm": 0.14089710183002874,
      "learning_rate": 0.0006141031267433316,
      "loss": 3.4547,
      "step": 1936
    },
    {
      "epoch": 0.44416418252694334,
      "grad_norm": 0.1475812696141555,
      "learning_rate": 0.0006137415474521454,
      "loss": 3.5135,
      "step": 1937
    },
    {
      "epoch": 0.4443934877321715,
      "grad_norm": 0.17610992367942357,
      "learning_rate": 0.0006133799054218825,
      "loss": 3.5911,
      "step": 1938
    },
    {
      "epoch": 0.4446227929373997,
      "grad_norm": 0.1496561160361629,
      "learning_rate": 0.0006130182008520222,
      "loss": 3.5008,
      "step": 1939
    },
    {
      "epoch": 0.4448520981426278,
      "grad_norm": 0.1420999579365738,
      "learning_rate": 0.0006126564339420784,
      "loss": 3.6295,
      "step": 1940
    },
    {
      "epoch": 0.445081403347856,
      "grad_norm": 0.13300631547960118,
      "learning_rate": 0.000612294604891599,
      "loss": 3.5193,
      "step": 1941
    },
    {
      "epoch": 0.44531070855308413,
      "grad_norm": 0.14644817674561422,
      "learning_rate": 0.000611932713900167,
      "loss": 3.4757,
      "step": 1942
    },
    {
      "epoch": 0.4455400137583123,
      "grad_norm": 0.13794144625059837,
      "learning_rate": 0.0006115707611673986,
      "loss": 3.4295,
      "step": 1943
    },
    {
      "epoch": 0.4457693189635405,
      "grad_norm": 0.1403825638198623,
      "learning_rate": 0.0006112087468929449,
      "loss": 3.5357,
      "step": 1944
    },
    {
      "epoch": 0.4459986241687686,
      "grad_norm": 0.13157122293755344,
      "learning_rate": 0.0006108466712764902,
      "loss": 3.3887,
      "step": 1945
    },
    {
      "epoch": 0.4462279293739968,
      "grad_norm": 0.1484293317657108,
      "learning_rate": 0.000610484534517753,
      "loss": 3.5941,
      "step": 1946
    },
    {
      "epoch": 0.4464572345792249,
      "grad_norm": 0.14651519103993815,
      "learning_rate": 0.0006101223368164858,
      "loss": 3.5341,
      "step": 1947
    },
    {
      "epoch": 0.4466865397844531,
      "grad_norm": 0.12726557422308163,
      "learning_rate": 0.0006097600783724742,
      "loss": 3.6425,
      "step": 1948
    },
    {
      "epoch": 0.4469158449896813,
      "grad_norm": 0.12524592452893754,
      "learning_rate": 0.0006093977593855375,
      "loss": 3.5669,
      "step": 1949
    },
    {
      "epoch": 0.4471451501949094,
      "grad_norm": 0.1409270095102889,
      "learning_rate": 0.0006090353800555286,
      "loss": 3.5329,
      "step": 1950
    },
    {
      "epoch": 0.4473744554001376,
      "grad_norm": 0.13658079040560728,
      "learning_rate": 0.0006086729405823335,
      "loss": 3.4772,
      "step": 1951
    },
    {
      "epoch": 0.4476037606053657,
      "grad_norm": 0.1416998708133019,
      "learning_rate": 0.0006083104411658713,
      "loss": 3.4958,
      "step": 1952
    },
    {
      "epoch": 0.4478330658105939,
      "grad_norm": 0.13094601398422964,
      "learning_rate": 0.0006079478820060943,
      "loss": 3.6096,
      "step": 1953
    },
    {
      "epoch": 0.4480623710158221,
      "grad_norm": 0.12320700462767449,
      "learning_rate": 0.0006075852633029877,
      "loss": 3.5487,
      "step": 1954
    },
    {
      "epoch": 0.4482916762210502,
      "grad_norm": 0.13319141811026203,
      "learning_rate": 0.0006072225852565695,
      "loss": 3.5811,
      "step": 1955
    },
    {
      "epoch": 0.4485209814262784,
      "grad_norm": 0.1344749924343147,
      "learning_rate": 0.0006068598480668905,
      "loss": 3.5556,
      "step": 1956
    },
    {
      "epoch": 0.4487502866315065,
      "grad_norm": 0.14277528809238266,
      "learning_rate": 0.0006064970519340341,
      "loss": 3.5001,
      "step": 1957
    },
    {
      "epoch": 0.4489795918367347,
      "grad_norm": 0.14073606289721868,
      "learning_rate": 0.0006061341970581165,
      "loss": 3.6021,
      "step": 1958
    },
    {
      "epoch": 0.4492088970419629,
      "grad_norm": 0.12370571355542616,
      "learning_rate": 0.0006057712836392856,
      "loss": 3.4901,
      "step": 1959
    },
    {
      "epoch": 0.449438202247191,
      "grad_norm": 0.14168764391758965,
      "learning_rate": 0.0006054083118777221,
      "loss": 3.4968,
      "step": 1960
    },
    {
      "epoch": 0.4496675074524192,
      "grad_norm": 0.14212592867148177,
      "learning_rate": 0.0006050452819736389,
      "loss": 3.6371,
      "step": 1961
    },
    {
      "epoch": 0.4498968126576473,
      "grad_norm": 0.13341059077127282,
      "learning_rate": 0.000604682194127281,
      "loss": 3.5424,
      "step": 1962
    },
    {
      "epoch": 0.4501261178628755,
      "grad_norm": 0.13860016485145776,
      "learning_rate": 0.000604319048538925,
      "loss": 3.5596,
      "step": 1963
    },
    {
      "epoch": 0.45035542306810367,
      "grad_norm": 0.1375967860609339,
      "learning_rate": 0.0006039558454088796,
      "loss": 3.5029,
      "step": 1964
    },
    {
      "epoch": 0.4505847282733318,
      "grad_norm": 0.14925452390066712,
      "learning_rate": 0.0006035925849374855,
      "loss": 3.6456,
      "step": 1965
    },
    {
      "epoch": 0.45081403347856,
      "grad_norm": 0.13842989777688264,
      "learning_rate": 0.0006032292673251142,
      "loss": 3.5303,
      "step": 1966
    },
    {
      "epoch": 0.4510433386837881,
      "grad_norm": 0.14132462036221619,
      "learning_rate": 0.0006028658927721697,
      "loss": 3.555,
      "step": 1967
    },
    {
      "epoch": 0.4512726438890163,
      "grad_norm": 0.1572441401148378,
      "learning_rate": 0.000602502461479087,
      "loss": 3.5994,
      "step": 1968
    },
    {
      "epoch": 0.45150194909424446,
      "grad_norm": 0.1518898935380814,
      "learning_rate": 0.0006021389736463321,
      "loss": 3.4911,
      "step": 1969
    },
    {
      "epoch": 0.4517312542994726,
      "grad_norm": 0.1325134797368519,
      "learning_rate": 0.0006017754294744027,
      "loss": 3.5088,
      "step": 1970
    },
    {
      "epoch": 0.45196055950470077,
      "grad_norm": 0.12595571901909355,
      "learning_rate": 0.0006014118291638271,
      "loss": 3.4307,
      "step": 1971
    },
    {
      "epoch": 0.4521898647099289,
      "grad_norm": 0.13628793099470615,
      "learning_rate": 0.0006010481729151649,
      "loss": 3.5412,
      "step": 1972
    },
    {
      "epoch": 0.45241916991515707,
      "grad_norm": 0.1357405023368633,
      "learning_rate": 0.0006006844609290065,
      "loss": 3.4312,
      "step": 1973
    },
    {
      "epoch": 0.45264847512038525,
      "grad_norm": 0.13168452457173796,
      "learning_rate": 0.000600320693405973,
      "loss": 3.5664,
      "step": 1974
    },
    {
      "epoch": 0.4528777803256134,
      "grad_norm": 0.13746629266392946,
      "learning_rate": 0.0005999568705467161,
      "loss": 3.5948,
      "step": 1975
    },
    {
      "epoch": 0.45310708553084156,
      "grad_norm": 0.1334637768149688,
      "learning_rate": 0.000599592992551918,
      "loss": 3.3883,
      "step": 1976
    },
    {
      "epoch": 0.4533363907360697,
      "grad_norm": 0.144602328781968,
      "learning_rate": 0.0005992290596222915,
      "loss": 3.488,
      "step": 1977
    },
    {
      "epoch": 0.45356569594129786,
      "grad_norm": 0.1606482136469162,
      "learning_rate": 0.0005988650719585795,
      "loss": 3.6256,
      "step": 1978
    },
    {
      "epoch": 0.45379500114652604,
      "grad_norm": 0.14188794977606523,
      "learning_rate": 0.0005985010297615551,
      "loss": 3.4826,
      "step": 1979
    },
    {
      "epoch": 0.45402430635175417,
      "grad_norm": 0.13290975473569616,
      "learning_rate": 0.0005981369332320217,
      "loss": 3.6015,
      "step": 1980
    },
    {
      "epoch": 0.45425361155698235,
      "grad_norm": 0.13083200993316044,
      "learning_rate": 0.0005977727825708123,
      "loss": 3.4798,
      "step": 1981
    },
    {
      "epoch": 0.4544829167622105,
      "grad_norm": 0.1333055432090884,
      "learning_rate": 0.0005974085779787901,
      "loss": 3.5388,
      "step": 1982
    },
    {
      "epoch": 0.45471222196743866,
      "grad_norm": 0.14192159619237266,
      "learning_rate": 0.0005970443196568478,
      "loss": 3.6052,
      "step": 1983
    },
    {
      "epoch": 0.45494152717266684,
      "grad_norm": 0.13550634032538575,
      "learning_rate": 0.000596680007805908,
      "loss": 3.5811,
      "step": 1984
    },
    {
      "epoch": 0.45517083237789496,
      "grad_norm": 0.13926537815276097,
      "learning_rate": 0.0005963156426269227,
      "loss": 3.442,
      "step": 1985
    },
    {
      "epoch": 0.45540013758312314,
      "grad_norm": 0.14398169026706323,
      "learning_rate": 0.0005959512243208731,
      "loss": 3.6084,
      "step": 1986
    },
    {
      "epoch": 0.45562944278835127,
      "grad_norm": 0.1442433712231854,
      "learning_rate": 0.0005955867530887702,
      "loss": 3.4813,
      "step": 1987
    },
    {
      "epoch": 0.45585874799357945,
      "grad_norm": 0.16094935836889498,
      "learning_rate": 0.0005952222291316535,
      "loss": 3.5781,
      "step": 1988
    },
    {
      "epoch": 0.45608805319880763,
      "grad_norm": 0.14058927108812885,
      "learning_rate": 0.0005948576526505923,
      "loss": 3.3457,
      "step": 1989
    },
    {
      "epoch": 0.45631735840403576,
      "grad_norm": 0.13800815401476924,
      "learning_rate": 0.0005944930238466845,
      "loss": 3.4558,
      "step": 1990
    },
    {
      "epoch": 0.45654666360926394,
      "grad_norm": 0.13715517117981635,
      "learning_rate": 0.0005941283429210568,
      "loss": 3.5723,
      "step": 1991
    },
    {
      "epoch": 0.45677596881449206,
      "grad_norm": 0.14496558261297332,
      "learning_rate": 0.0005937636100748648,
      "loss": 3.413,
      "step": 1992
    },
    {
      "epoch": 0.45700527401972024,
      "grad_norm": 0.14502017555546798,
      "learning_rate": 0.0005933988255092926,
      "loss": 3.5113,
      "step": 1993
    },
    {
      "epoch": 0.4572345792249484,
      "grad_norm": 0.13940934205788427,
      "learning_rate": 0.0005930339894255532,
      "loss": 3.4872,
      "step": 1994
    },
    {
      "epoch": 0.45746388443017655,
      "grad_norm": 0.13772950545274912,
      "learning_rate": 0.0005926691020248874,
      "loss": 3.4985,
      "step": 1995
    },
    {
      "epoch": 0.45769318963540473,
      "grad_norm": 0.13345713356003652,
      "learning_rate": 0.000592304163508565,
      "loss": 3.3508,
      "step": 1996
    },
    {
      "epoch": 0.4579224948406329,
      "grad_norm": 0.12130776820278423,
      "learning_rate": 0.0005919391740778833,
      "loss": 3.5687,
      "step": 1997
    },
    {
      "epoch": 0.45815180004586104,
      "grad_norm": 0.133914968439512,
      "learning_rate": 0.0005915741339341682,
      "loss": 3.5971,
      "step": 1998
    },
    {
      "epoch": 0.4583811052510892,
      "grad_norm": 0.13610493202260163,
      "learning_rate": 0.0005912090432787736,
      "loss": 3.5337,
      "step": 1999
    },
    {
      "epoch": 0.45861041045631734,
      "grad_norm": 0.14164654331324525,
      "learning_rate": 0.0005908439023130807,
      "loss": 3.3831,
      "step": 2000
    },
    {
      "epoch": 0.4588397156615455,
      "grad_norm": 0.14524074202099424,
      "learning_rate": 0.000590478711238499,
      "loss": 3.5215,
      "step": 2001
    },
    {
      "epoch": 0.4590690208667737,
      "grad_norm": 0.1864597322834301,
      "learning_rate": 0.0005901134702564656,
      "loss": 3.5233,
      "step": 2002
    },
    {
      "epoch": 0.45929832607200183,
      "grad_norm": 0.1543927294673927,
      "learning_rate": 0.0005897481795684446,
      "loss": 3.4991,
      "step": 2003
    },
    {
      "epoch": 0.45952763127723,
      "grad_norm": 0.14170516824830354,
      "learning_rate": 0.0005893828393759281,
      "loss": 3.501,
      "step": 2004
    },
    {
      "epoch": 0.45975693648245813,
      "grad_norm": 0.14869148285347347,
      "learning_rate": 0.0005890174498804355,
      "loss": 3.6887,
      "step": 2005
    },
    {
      "epoch": 0.4599862416876863,
      "grad_norm": 0.15107759871539225,
      "learning_rate": 0.0005886520112835128,
      "loss": 3.6318,
      "step": 2006
    },
    {
      "epoch": 0.4602155468929145,
      "grad_norm": 0.13875359857752856,
      "learning_rate": 0.0005882865237867339,
      "loss": 3.5957,
      "step": 2007
    },
    {
      "epoch": 0.4604448520981426,
      "grad_norm": 0.1526250460612507,
      "learning_rate": 0.0005879209875916989,
      "loss": 3.4308,
      "step": 2008
    },
    {
      "epoch": 0.4606741573033708,
      "grad_norm": 0.1391828598564802,
      "learning_rate": 0.0005875554029000353,
      "loss": 3.4908,
      "step": 2009
    },
    {
      "epoch": 0.4609034625085989,
      "grad_norm": 0.1354895839022271,
      "learning_rate": 0.0005871897699133973,
      "loss": 3.4905,
      "step": 2010
    },
    {
      "epoch": 0.4611327677138271,
      "grad_norm": 0.13244351091078008,
      "learning_rate": 0.0005868240888334653,
      "loss": 3.4447,
      "step": 2011
    },
    {
      "epoch": 0.4613620729190553,
      "grad_norm": 0.12844318370285615,
      "learning_rate": 0.0005864583598619468,
      "loss": 3.4465,
      "step": 2012
    },
    {
      "epoch": 0.4615913781242834,
      "grad_norm": 0.1307700115413058,
      "learning_rate": 0.0005860925832005753,
      "loss": 3.4487,
      "step": 2013
    },
    {
      "epoch": 0.4618206833295116,
      "grad_norm": 0.13330650508349587,
      "learning_rate": 0.0005857267590511108,
      "loss": 3.4426,
      "step": 2014
    },
    {
      "epoch": 0.4620499885347397,
      "grad_norm": 0.15057558074825086,
      "learning_rate": 0.0005853608876153395,
      "loss": 3.5632,
      "step": 2015
    },
    {
      "epoch": 0.4622792937399679,
      "grad_norm": 0.1382703887578414,
      "learning_rate": 0.0005849949690950735,
      "loss": 3.5398,
      "step": 2016
    },
    {
      "epoch": 0.4625085989451961,
      "grad_norm": 0.13946908609872738,
      "learning_rate": 0.0005846290036921512,
      "loss": 3.5477,
      "step": 2017
    },
    {
      "epoch": 0.4627379041504242,
      "grad_norm": 0.13672613222342098,
      "learning_rate": 0.0005842629916084365,
      "loss": 3.4764,
      "step": 2018
    },
    {
      "epoch": 0.4629672093556524,
      "grad_norm": 0.1433025579666604,
      "learning_rate": 0.0005838969330458195,
      "loss": 3.6546,
      "step": 2019
    },
    {
      "epoch": 0.4631965145608805,
      "grad_norm": 0.14067191076529809,
      "learning_rate": 0.0005835308282062153,
      "loss": 3.6026,
      "step": 2020
    },
    {
      "epoch": 0.4634258197661087,
      "grad_norm": 0.13989326148391226,
      "learning_rate": 0.0005831646772915651,
      "loss": 3.4264,
      "step": 2021
    },
    {
      "epoch": 0.4636551249713369,
      "grad_norm": 0.1407406734738939,
      "learning_rate": 0.0005827984805038352,
      "loss": 3.4918,
      "step": 2022
    },
    {
      "epoch": 0.463884430176565,
      "grad_norm": 0.13456964167060106,
      "learning_rate": 0.0005824322380450173,
      "loss": 3.6519,
      "step": 2023
    },
    {
      "epoch": 0.4641137353817932,
      "grad_norm": 0.15430308210003804,
      "learning_rate": 0.0005820659501171284,
      "loss": 3.5117,
      "step": 2024
    },
    {
      "epoch": 0.4643430405870213,
      "grad_norm": 0.15082194034081753,
      "learning_rate": 0.0005816996169222102,
      "loss": 3.4827,
      "step": 2025
    },
    {
      "epoch": 0.4645723457922495,
      "grad_norm": 0.13275001502240197,
      "learning_rate": 0.00058133323866233,
      "loss": 3.6093,
      "step": 2026
    },
    {
      "epoch": 0.46480165099747767,
      "grad_norm": 0.1488351018288779,
      "learning_rate": 0.0005809668155395793,
      "loss": 3.469,
      "step": 2027
    },
    {
      "epoch": 0.4650309562027058,
      "grad_norm": 0.13838417034783249,
      "learning_rate": 0.0005806003477560748,
      "loss": 3.4849,
      "step": 2028
    },
    {
      "epoch": 0.465260261407934,
      "grad_norm": 0.13493391949681224,
      "learning_rate": 0.0005802338355139578,
      "loss": 3.4256,
      "step": 2029
    },
    {
      "epoch": 0.4654895666131621,
      "grad_norm": 0.13366693063960444,
      "learning_rate": 0.0005798672790153937,
      "loss": 3.6508,
      "step": 2030
    },
    {
      "epoch": 0.4657188718183903,
      "grad_norm": 0.13404350398551573,
      "learning_rate": 0.0005795006784625728,
      "loss": 3.5928,
      "step": 2031
    },
    {
      "epoch": 0.46594817702361846,
      "grad_norm": 0.13563676054388027,
      "learning_rate": 0.0005791340340577094,
      "loss": 3.5619,
      "step": 2032
    },
    {
      "epoch": 0.4661774822288466,
      "grad_norm": 0.1525058837899261,
      "learning_rate": 0.0005787673460030423,
      "loss": 3.5366,
      "step": 2033
    },
    {
      "epoch": 0.46640678743407477,
      "grad_norm": 0.12790684209408035,
      "learning_rate": 0.000578400614500834,
      "loss": 3.3956,
      "step": 2034
    },
    {
      "epoch": 0.4666360926393029,
      "grad_norm": 0.14517765401354374,
      "learning_rate": 0.000578033839753371,
      "loss": 3.3151,
      "step": 2035
    },
    {
      "epoch": 0.46686539784453107,
      "grad_norm": 0.13545470902778395,
      "learning_rate": 0.0005776670219629642,
      "loss": 3.5296,
      "step": 2036
    },
    {
      "epoch": 0.46709470304975925,
      "grad_norm": 0.15492279104628628,
      "learning_rate": 0.0005773001613319476,
      "loss": 3.6292,
      "step": 2037
    },
    {
      "epoch": 0.4673240082549874,
      "grad_norm": 0.1436357424196758,
      "learning_rate": 0.0005769332580626792,
      "loss": 3.4821,
      "step": 2038
    },
    {
      "epoch": 0.46755331346021556,
      "grad_norm": 0.13742295966298262,
      "learning_rate": 0.00057656631235754,
      "loss": 3.4902,
      "step": 2039
    },
    {
      "epoch": 0.4677826186654437,
      "grad_norm": 0.14133806966910323,
      "learning_rate": 0.0005761993244189354,
      "loss": 3.4494,
      "step": 2040
    },
    {
      "epoch": 0.46801192387067186,
      "grad_norm": 0.13289323781227227,
      "learning_rate": 0.0005758322944492929,
      "loss": 3.4092,
      "step": 2041
    },
    {
      "epoch": 0.46824122907590005,
      "grad_norm": 0.1295730775606446,
      "learning_rate": 0.0005754652226510644,
      "loss": 3.6388,
      "step": 2042
    },
    {
      "epoch": 0.46847053428112817,
      "grad_norm": 0.1608650143181302,
      "learning_rate": 0.0005750981092267237,
      "loss": 3.5439,
      "step": 2043
    },
    {
      "epoch": 0.46869983948635635,
      "grad_norm": 0.13428023909130174,
      "learning_rate": 0.0005747309543787683,
      "loss": 3.5289,
      "step": 2044
    },
    {
      "epoch": 0.4689291446915845,
      "grad_norm": 0.14521077120744075,
      "learning_rate": 0.0005743637583097183,
      "loss": 3.5232,
      "step": 2045
    },
    {
      "epoch": 0.46915844989681266,
      "grad_norm": 0.13794051501919474,
      "learning_rate": 0.0005739965212221168,
      "loss": 3.4047,
      "step": 2046
    },
    {
      "epoch": 0.46938775510204084,
      "grad_norm": 0.13807838427931954,
      "learning_rate": 0.0005736292433185291,
      "loss": 3.399,
      "step": 2047
    },
    {
      "epoch": 0.46961706030726896,
      "grad_norm": 0.1340125211101786,
      "learning_rate": 0.0005732619248015433,
      "loss": 3.4106,
      "step": 2048
    },
    {
      "epoch": 0.46984636551249714,
      "grad_norm": 0.13585912314609042,
      "learning_rate": 0.0005728945658737699,
      "loss": 3.5199,
      "step": 2049
    },
    {
      "epoch": 0.47007567071772527,
      "grad_norm": 0.1353777934579441,
      "learning_rate": 0.0005725271667378418,
      "loss": 3.5079,
      "step": 2050
    },
    {
      "epoch": 0.47030497592295345,
      "grad_norm": 0.1411419981679103,
      "learning_rate": 0.0005721597275964133,
      "loss": 3.6115,
      "step": 2051
    },
    {
      "epoch": 0.47053428112818163,
      "grad_norm": 0.12973868472498454,
      "learning_rate": 0.000571792248652162,
      "loss": 3.609,
      "step": 2052
    },
    {
      "epoch": 0.47076358633340976,
      "grad_norm": 0.1408128202580158,
      "learning_rate": 0.0005714247301077865,
      "loss": 3.5045,
      "step": 2053
    },
    {
      "epoch": 0.47099289153863794,
      "grad_norm": 0.12633399031735143,
      "learning_rate": 0.0005710571721660077,
      "loss": 3.4043,
      "step": 2054
    },
    {
      "epoch": 0.47122219674386606,
      "grad_norm": 0.13013298668463194,
      "learning_rate": 0.0005706895750295682,
      "loss": 3.5117,
      "step": 2055
    },
    {
      "epoch": 0.47145150194909424,
      "grad_norm": 0.129500991427283,
      "learning_rate": 0.0005703219389012318,
      "loss": 3.4485,
      "step": 2056
    },
    {
      "epoch": 0.4716808071543224,
      "grad_norm": 0.1338964039140026,
      "learning_rate": 0.0005699542639837844,
      "loss": 3.4362,
      "step": 2057
    },
    {
      "epoch": 0.47191011235955055,
      "grad_norm": 0.1410085620853052,
      "learning_rate": 0.0005695865504800327,
      "loss": 3.349,
      "step": 2058
    },
    {
      "epoch": 0.47213941756477873,
      "grad_norm": 0.14102297433096844,
      "learning_rate": 0.0005692187985928055,
      "loss": 3.4319,
      "step": 2059
    },
    {
      "epoch": 0.47236872277000685,
      "grad_norm": 0.12781060983270176,
      "learning_rate": 0.000568851008524952,
      "loss": 3.5212,
      "step": 2060
    },
    {
      "epoch": 0.47259802797523504,
      "grad_norm": 0.12796771256881292,
      "learning_rate": 0.0005684831804793427,
      "loss": 3.4807,
      "step": 2061
    },
    {
      "epoch": 0.4728273331804632,
      "grad_norm": 0.13425279901515141,
      "learning_rate": 0.0005681153146588693,
      "loss": 3.3684,
      "step": 2062
    },
    {
      "epoch": 0.47305663838569134,
      "grad_norm": 0.13621063004272024,
      "learning_rate": 0.0005677474112664438,
      "loss": 3.4438,
      "step": 2063
    },
    {
      "epoch": 0.4732859435909195,
      "grad_norm": 0.14313748733249518,
      "learning_rate": 0.0005673794705049995,
      "loss": 3.4926,
      "step": 2064
    },
    {
      "epoch": 0.47351524879614765,
      "grad_norm": 0.13641581537193617,
      "learning_rate": 0.0005670114925774899,
      "loss": 3.4351,
      "step": 2065
    },
    {
      "epoch": 0.47374455400137583,
      "grad_norm": 0.13324811027531944,
      "learning_rate": 0.0005666434776868895,
      "loss": 3.4246,
      "step": 2066
    },
    {
      "epoch": 0.473973859206604,
      "grad_norm": 0.1399908527214456,
      "learning_rate": 0.0005662754260361924,
      "loss": 3.5978,
      "step": 2067
    },
    {
      "epoch": 0.47420316441183213,
      "grad_norm": 0.12845276715628431,
      "learning_rate": 0.0005659073378284137,
      "loss": 3.4075,
      "step": 2068
    },
    {
      "epoch": 0.4744324696170603,
      "grad_norm": 0.1369158965934555,
      "learning_rate": 0.0005655392132665884,
      "loss": 3.5886,
      "step": 2069
    },
    {
      "epoch": 0.47466177482228844,
      "grad_norm": 0.1294333384749657,
      "learning_rate": 0.0005651710525537714,
      "loss": 3.432,
      "step": 2070
    },
    {
      "epoch": 0.4748910800275166,
      "grad_norm": 0.12766911869830241,
      "learning_rate": 0.000564802855893038,
      "loss": 3.4438,
      "step": 2071
    },
    {
      "epoch": 0.4751203852327448,
      "grad_norm": 0.1433072332474297,
      "learning_rate": 0.0005644346234874824,
      "loss": 3.6478,
      "step": 2072
    },
    {
      "epoch": 0.4753496904379729,
      "grad_norm": 0.13076111247151934,
      "learning_rate": 0.0005640663555402198,
      "loss": 3.5103,
      "step": 2073
    },
    {
      "epoch": 0.4755789956432011,
      "grad_norm": 0.14487309684570363,
      "learning_rate": 0.0005636980522543838,
      "loss": 3.5438,
      "step": 2074
    },
    {
      "epoch": 0.47580830084842923,
      "grad_norm": 0.14533007587657942,
      "learning_rate": 0.0005633297138331285,
      "loss": 3.6641,
      "step": 2075
    },
    {
      "epoch": 0.4760376060536574,
      "grad_norm": 0.13590454678547748,
      "learning_rate": 0.0005629613404796266,
      "loss": 3.5186,
      "step": 2076
    },
    {
      "epoch": 0.4762669112588856,
      "grad_norm": 0.13171347618697285,
      "learning_rate": 0.0005625929323970705,
      "loss": 3.4934,
      "step": 2077
    },
    {
      "epoch": 0.4764962164641137,
      "grad_norm": 0.14453300923996432,
      "learning_rate": 0.0005622244897886716,
      "loss": 3.5486,
      "step": 2078
    },
    {
      "epoch": 0.4767255216693419,
      "grad_norm": 0.13333339096395827,
      "learning_rate": 0.0005618560128576603,
      "loss": 3.459,
      "step": 2079
    },
    {
      "epoch": 0.47695482687457,
      "grad_norm": 0.16539060253416868,
      "learning_rate": 0.0005614875018072861,
      "loss": 3.4129,
      "step": 2080
    },
    {
      "epoch": 0.4771841320797982,
      "grad_norm": 0.14585103949749204,
      "learning_rate": 0.0005611189568408173,
      "loss": 3.59,
      "step": 2081
    },
    {
      "epoch": 0.4774134372850264,
      "grad_norm": 0.12583183781919488,
      "learning_rate": 0.0005607503781615407,
      "loss": 3.5878,
      "step": 2082
    },
    {
      "epoch": 0.4776427424902545,
      "grad_norm": 0.13031175011103077,
      "learning_rate": 0.0005603817659727619,
      "loss": 3.5842,
      "step": 2083
    },
    {
      "epoch": 0.4778720476954827,
      "grad_norm": 0.16162808981775334,
      "learning_rate": 0.0005600131204778049,
      "loss": 3.5322,
      "step": 2084
    },
    {
      "epoch": 0.4781013529007108,
      "grad_norm": 0.13822143053000438,
      "learning_rate": 0.0005596444418800121,
      "loss": 3.5903,
      "step": 2085
    },
    {
      "epoch": 0.478330658105939,
      "grad_norm": 0.13310988113075617,
      "learning_rate": 0.000559275730382744,
      "loss": 3.4722,
      "step": 2086
    },
    {
      "epoch": 0.4785599633111672,
      "grad_norm": 0.1470071373290441,
      "learning_rate": 0.0005589069861893798,
      "loss": 3.4112,
      "step": 2087
    },
    {
      "epoch": 0.4787892685163953,
      "grad_norm": 0.13909924538809715,
      "learning_rate": 0.0005585382095033159,
      "loss": 3.5047,
      "step": 2088
    },
    {
      "epoch": 0.4790185737216235,
      "grad_norm": 0.14069399433531268,
      "learning_rate": 0.0005581694005279673,
      "loss": 3.4925,
      "step": 2089
    },
    {
      "epoch": 0.47924787892685167,
      "grad_norm": 0.1331618808221091,
      "learning_rate": 0.0005578005594667665,
      "loss": 3.5462,
      "step": 2090
    },
    {
      "epoch": 0.4794771841320798,
      "grad_norm": 0.13684928417061973,
      "learning_rate": 0.0005574316865231637,
      "loss": 3.3405,
      "step": 2091
    },
    {
      "epoch": 0.479706489337308,
      "grad_norm": 0.14243269058116975,
      "learning_rate": 0.0005570627819006268,
      "loss": 3.5982,
      "step": 2092
    },
    {
      "epoch": 0.4799357945425361,
      "grad_norm": 0.14198792287817935,
      "learning_rate": 0.0005566938458026411,
      "loss": 3.4792,
      "step": 2093
    },
    {
      "epoch": 0.4801650997477643,
      "grad_norm": 0.1334849261478509,
      "learning_rate": 0.0005563248784327092,
      "loss": 3.4735,
      "step": 2094
    },
    {
      "epoch": 0.48039440495299246,
      "grad_norm": 0.13222701907111248,
      "learning_rate": 0.0005559558799943514,
      "loss": 3.4183,
      "step": 2095
    },
    {
      "epoch": 0.4806237101582206,
      "grad_norm": 0.1518768248658855,
      "learning_rate": 0.0005555868506911044,
      "loss": 3.5761,
      "step": 2096
    },
    {
      "epoch": 0.48085301536344877,
      "grad_norm": 0.14656547181320442,
      "learning_rate": 0.0005552177907265223,
      "loss": 3.5209,
      "step": 2097
    },
    {
      "epoch": 0.4810823205686769,
      "grad_norm": 0.1482787593471702,
      "learning_rate": 0.0005548487003041762,
      "loss": 3.3947,
      "step": 2098
    },
    {
      "epoch": 0.48131162577390507,
      "grad_norm": 0.14103491753375894,
      "learning_rate": 0.000554479579627654,
      "loss": 3.5712,
      "step": 2099
    },
    {
      "epoch": 0.48154093097913325,
      "grad_norm": 0.13361716667609636,
      "learning_rate": 0.0005541104289005602,
      "loss": 3.6745,
      "step": 2100
    },
    {
      "epoch": 0.4817702361843614,
      "grad_norm": 0.1264920224499176,
      "learning_rate": 0.0005537412483265157,
      "loss": 3.4834,
      "step": 2101
    },
    {
      "epoch": 0.48199954138958956,
      "grad_norm": 0.13255786070288134,
      "learning_rate": 0.0005533720381091582,
      "loss": 3.5193,
      "step": 2102
    },
    {
      "epoch": 0.4822288465948177,
      "grad_norm": 0.15085917210897107,
      "learning_rate": 0.0005530027984521413,
      "loss": 3.5871,
      "step": 2103
    },
    {
      "epoch": 0.48245815180004586,
      "grad_norm": 0.1307801037796017,
      "learning_rate": 0.0005526335295591354,
      "loss": 3.5523,
      "step": 2104
    },
    {
      "epoch": 0.48268745700527405,
      "grad_norm": 0.12808450489698134,
      "learning_rate": 0.0005522642316338268,
      "loss": 3.3717,
      "step": 2105
    },
    {
      "epoch": 0.48291676221050217,
      "grad_norm": 0.13741444375777911,
      "learning_rate": 0.0005518949048799176,
      "loss": 3.5602,
      "step": 2106
    },
    {
      "epoch": 0.48314606741573035,
      "grad_norm": 0.15355821547519344,
      "learning_rate": 0.0005515255495011259,
      "loss": 3.6081,
      "step": 2107
    },
    {
      "epoch": 0.4833753726209585,
      "grad_norm": 0.14038646796668972,
      "learning_rate": 0.0005511561657011859,
      "loss": 3.569,
      "step": 2108
    },
    {
      "epoch": 0.48360467782618666,
      "grad_norm": 0.14088215820144404,
      "learning_rate": 0.0005507867536838472,
      "loss": 3.553,
      "step": 2109
    },
    {
      "epoch": 0.48383398303141484,
      "grad_norm": 0.14630397200531042,
      "learning_rate": 0.0005504173136528746,
      "loss": 3.4931,
      "step": 2110
    },
    {
      "epoch": 0.48406328823664296,
      "grad_norm": 0.1298894511218128,
      "learning_rate": 0.0005500478458120492,
      "loss": 3.4673,
      "step": 2111
    },
    {
      "epoch": 0.48429259344187114,
      "grad_norm": 0.15332340824242338,
      "learning_rate": 0.0005496783503651667,
      "loss": 3.5447,
      "step": 2112
    },
    {
      "epoch": 0.48452189864709927,
      "grad_norm": 0.14109809013047467,
      "learning_rate": 0.0005493088275160387,
      "loss": 3.4223,
      "step": 2113
    },
    {
      "epoch": 0.48475120385232745,
      "grad_norm": 0.13664497187333624,
      "learning_rate": 0.0005489392774684911,
      "loss": 3.469,
      "step": 2114
    },
    {
      "epoch": 0.48498050905755563,
      "grad_norm": 0.1269985189934799,
      "learning_rate": 0.0005485697004263657,
      "loss": 3.4942,
      "step": 2115
    },
    {
      "epoch": 0.48520981426278376,
      "grad_norm": 0.15723948325778297,
      "learning_rate": 0.0005482000965935182,
      "loss": 3.6423,
      "step": 2116
    },
    {
      "epoch": 0.48543911946801194,
      "grad_norm": 0.13103234243859546,
      "learning_rate": 0.0005478304661738199,
      "loss": 3.5049,
      "step": 2117
    },
    {
      "epoch": 0.48566842467324006,
      "grad_norm": 0.13188189422716193,
      "learning_rate": 0.0005474608093711565,
      "loss": 3.5461,
      "step": 2118
    },
    {
      "epoch": 0.48589772987846824,
      "grad_norm": 0.14073334030454435,
      "learning_rate": 0.0005470911263894279,
      "loss": 3.4839,
      "step": 2119
    },
    {
      "epoch": 0.4861270350836964,
      "grad_norm": 0.12395563772124465,
      "learning_rate": 0.0005467214174325494,
      "loss": 3.3682,
      "step": 2120
    },
    {
      "epoch": 0.48635634028892455,
      "grad_norm": 0.1425856436957177,
      "learning_rate": 0.0005463516827044491,
      "loss": 3.4909,
      "step": 2121
    },
    {
      "epoch": 0.48658564549415273,
      "grad_norm": 0.13759160543379165,
      "learning_rate": 0.000545981922409071,
      "loss": 3.4908,
      "step": 2122
    },
    {
      "epoch": 0.48681495069938086,
      "grad_norm": 0.14517009347381546,
      "learning_rate": 0.000545612136750372,
      "loss": 3.4465,
      "step": 2123
    },
    {
      "epoch": 0.48704425590460904,
      "grad_norm": 0.1451056092816682,
      "learning_rate": 0.0005452423259323235,
      "loss": 3.5393,
      "step": 2124
    },
    {
      "epoch": 0.4872735611098372,
      "grad_norm": 0.1281268371196807,
      "learning_rate": 0.0005448724901589107,
      "loss": 3.3569,
      "step": 2125
    },
    {
      "epoch": 0.48750286631506534,
      "grad_norm": 0.14302120728011494,
      "learning_rate": 0.0005445026296341325,
      "loss": 3.5264,
      "step": 2126
    },
    {
      "epoch": 0.4877321715202935,
      "grad_norm": 0.15131268543018483,
      "learning_rate": 0.0005441327445620014,
      "loss": 3.5081,
      "step": 2127
    },
    {
      "epoch": 0.48796147672552165,
      "grad_norm": 0.12932325406889603,
      "learning_rate": 0.0005437628351465435,
      "loss": 3.5715,
      "step": 2128
    },
    {
      "epoch": 0.48819078193074983,
      "grad_norm": 0.14313535836915586,
      "learning_rate": 0.0005433929015917988,
      "loss": 3.5117,
      "step": 2129
    },
    {
      "epoch": 0.488420087135978,
      "grad_norm": 0.141414470977411,
      "learning_rate": 0.0005430229441018197,
      "loss": 3.5937,
      "step": 2130
    },
    {
      "epoch": 0.48864939234120613,
      "grad_norm": 0.13015732629623838,
      "learning_rate": 0.0005426529628806724,
      "loss": 3.5396,
      "step": 2131
    },
    {
      "epoch": 0.4888786975464343,
      "grad_norm": 0.12892739713018775,
      "learning_rate": 0.0005422829581324361,
      "loss": 3.5756,
      "step": 2132
    },
    {
      "epoch": 0.48910800275166244,
      "grad_norm": 0.14494853970216356,
      "learning_rate": 0.0005419129300612029,
      "loss": 3.546,
      "step": 2133
    },
    {
      "epoch": 0.4893373079568906,
      "grad_norm": 0.13512037696497028,
      "learning_rate": 0.000541542878871078,
      "loss": 3.4709,
      "step": 2134
    },
    {
      "epoch": 0.4895666131621188,
      "grad_norm": 0.13525545179300785,
      "learning_rate": 0.000541172804766179,
      "loss": 3.5167,
      "step": 2135
    },
    {
      "epoch": 0.4897959183673469,
      "grad_norm": 0.13087385164802162,
      "learning_rate": 0.0005408027079506361,
      "loss": 3.4675,
      "step": 2136
    },
    {
      "epoch": 0.4900252235725751,
      "grad_norm": 0.1342147643283008,
      "learning_rate": 0.0005404325886285927,
      "loss": 3.5222,
      "step": 2137
    },
    {
      "epoch": 0.49025452877780323,
      "grad_norm": 0.14269329116181548,
      "learning_rate": 0.0005400624470042037,
      "loss": 3.5139,
      "step": 2138
    },
    {
      "epoch": 0.4904838339830314,
      "grad_norm": 0.1572032038527884,
      "learning_rate": 0.000539692283281637,
      "loss": 3.4457,
      "step": 2139
    },
    {
      "epoch": 0.4907131391882596,
      "grad_norm": 0.14202785221681014,
      "learning_rate": 0.0005393220976650724,
      "loss": 3.4278,
      "step": 2140
    },
    {
      "epoch": 0.4909424443934877,
      "grad_norm": 0.13806925623399674,
      "learning_rate": 0.0005389518903587017,
      "loss": 3.568,
      "step": 2141
    },
    {
      "epoch": 0.4911717495987159,
      "grad_norm": 0.13781390923062947,
      "learning_rate": 0.0005385816615667287,
      "loss": 3.4491,
      "step": 2142
    },
    {
      "epoch": 0.491401054803944,
      "grad_norm": 0.1467808424996466,
      "learning_rate": 0.0005382114114933695,
      "loss": 3.4657,
      "step": 2143
    },
    {
      "epoch": 0.4916303600091722,
      "grad_norm": 0.1468233550421187,
      "learning_rate": 0.0005378411403428513,
      "loss": 3.5504,
      "step": 2144
    },
    {
      "epoch": 0.4918596652144004,
      "grad_norm": 0.1471005666055234,
      "learning_rate": 0.0005374708483194132,
      "loss": 3.3423,
      "step": 2145
    },
    {
      "epoch": 0.4920889704196285,
      "grad_norm": 0.1417791088365292,
      "learning_rate": 0.0005371005356273057,
      "loss": 3.4324,
      "step": 2146
    },
    {
      "epoch": 0.4923182756248567,
      "grad_norm": 0.12275768522377589,
      "learning_rate": 0.000536730202470791,
      "loss": 3.3889,
      "step": 2147
    },
    {
      "epoch": 0.4925475808300848,
      "grad_norm": 0.13796486614536457,
      "learning_rate": 0.0005363598490541425,
      "loss": 3.4582,
      "step": 2148
    },
    {
      "epoch": 0.492776886035313,
      "grad_norm": 0.1327539713817846,
      "learning_rate": 0.0005359894755816443,
      "loss": 3.5735,
      "step": 2149
    },
    {
      "epoch": 0.4930061912405412,
      "grad_norm": 0.1325929622804662,
      "learning_rate": 0.0005356190822575921,
      "loss": 3.496,
      "step": 2150
    },
    {
      "epoch": 0.4932354964457693,
      "grad_norm": 0.1313813956331959,
      "learning_rate": 0.0005352486692862926,
      "loss": 3.5351,
      "step": 2151
    },
    {
      "epoch": 0.4934648016509975,
      "grad_norm": 0.1320407494398325,
      "learning_rate": 0.0005348782368720626,
      "loss": 3.4429,
      "step": 2152
    },
    {
      "epoch": 0.4936941068562256,
      "grad_norm": 0.1352519961837643,
      "learning_rate": 0.0005345077852192307,
      "loss": 3.6499,
      "step": 2153
    },
    {
      "epoch": 0.4939234120614538,
      "grad_norm": 0.1329779090928443,
      "learning_rate": 0.0005341373145321351,
      "loss": 3.5331,
      "step": 2154
    },
    {
      "epoch": 0.494152717266682,
      "grad_norm": 0.13688370395753333,
      "learning_rate": 0.0005337668250151254,
      "loss": 3.4045,
      "step": 2155
    },
    {
      "epoch": 0.4943820224719101,
      "grad_norm": 0.15934815261791552,
      "learning_rate": 0.0005333963168725609,
      "loss": 3.366,
      "step": 2156
    },
    {
      "epoch": 0.4946113276771383,
      "grad_norm": 0.12971803180725935,
      "learning_rate": 0.0005330257903088111,
      "loss": 3.4113,
      "step": 2157
    },
    {
      "epoch": 0.4948406328823664,
      "grad_norm": 0.12724385582014514,
      "learning_rate": 0.0005326552455282566,
      "loss": 3.6271,
      "step": 2158
    },
    {
      "epoch": 0.4950699380875946,
      "grad_norm": 0.12892919012564272,
      "learning_rate": 0.000532284682735287,
      "loss": 3.3705,
      "step": 2159
    },
    {
      "epoch": 0.49529924329282277,
      "grad_norm": 0.14553317260808277,
      "learning_rate": 0.0005319141021343024,
      "loss": 3.5535,
      "step": 2160
    },
    {
      "epoch": 0.4955285484980509,
      "grad_norm": 0.1269244748604712,
      "learning_rate": 0.0005315435039297124,
      "loss": 3.4939,
      "step": 2161
    },
    {
      "epoch": 0.49575785370327907,
      "grad_norm": 0.1500287608914004,
      "learning_rate": 0.0005311728883259368,
      "loss": 3.3746,
      "step": 2162
    },
    {
      "epoch": 0.4959871589085072,
      "grad_norm": 0.1473060271718232,
      "learning_rate": 0.0005308022555274046,
      "loss": 3.4901,
      "step": 2163
    },
    {
      "epoch": 0.4962164641137354,
      "grad_norm": 0.14807871299368308,
      "learning_rate": 0.0005304316057385541,
      "loss": 3.3927,
      "step": 2164
    },
    {
      "epoch": 0.49644576931896356,
      "grad_norm": 0.14369531600103083,
      "learning_rate": 0.0005300609391638336,
      "loss": 3.5337,
      "step": 2165
    },
    {
      "epoch": 0.4966750745241917,
      "grad_norm": 0.1305741382516606,
      "learning_rate": 0.0005296902560077,
      "loss": 3.3326,
      "step": 2166
    },
    {
      "epoch": 0.49690437972941986,
      "grad_norm": 0.13582301696148164,
      "learning_rate": 0.0005293195564746201,
      "loss": 3.6343,
      "step": 2167
    },
    {
      "epoch": 0.497133684934648,
      "grad_norm": 0.142372751170207,
      "learning_rate": 0.0005289488407690685,
      "loss": 3.4976,
      "step": 2168
    },
    {
      "epoch": 0.49736299013987617,
      "grad_norm": 0.1352908155100643,
      "learning_rate": 0.0005285781090955304,
      "loss": 3.4213,
      "step": 2169
    },
    {
      "epoch": 0.49759229534510435,
      "grad_norm": 0.1570686223130985,
      "learning_rate": 0.0005282073616584982,
      "loss": 3.5157,
      "step": 2170
    },
    {
      "epoch": 0.4978216005503325,
      "grad_norm": 0.1461758324645867,
      "learning_rate": 0.0005278365986624743,
      "loss": 3.4415,
      "step": 2171
    },
    {
      "epoch": 0.49805090575556066,
      "grad_norm": 0.12963820119311203,
      "learning_rate": 0.0005274658203119684,
      "loss": 3.5861,
      "step": 2172
    },
    {
      "epoch": 0.4982802109607888,
      "grad_norm": 0.1265707599292712,
      "learning_rate": 0.0005270950268115001,
      "loss": 3.5367,
      "step": 2173
    },
    {
      "epoch": 0.49850951616601696,
      "grad_norm": 0.13786046248432737,
      "learning_rate": 0.0005267242183655961,
      "loss": 3.4214,
      "step": 2174
    },
    {
      "epoch": 0.49873882137124514,
      "grad_norm": 0.15005593299163067,
      "learning_rate": 0.0005263533951787919,
      "loss": 3.4641,
      "step": 2175
    },
    {
      "epoch": 0.49896812657647327,
      "grad_norm": 0.14693017407748835,
      "learning_rate": 0.0005259825574556315,
      "loss": 3.5157,
      "step": 2176
    },
    {
      "epoch": 0.49919743178170145,
      "grad_norm": 0.13473560166068685,
      "learning_rate": 0.000525611705400666,
      "loss": 3.4057,
      "step": 2177
    },
    {
      "epoch": 0.4994267369869296,
      "grad_norm": 0.14423124129097292,
      "learning_rate": 0.0005252408392184552,
      "loss": 3.3898,
      "step": 2178
    },
    {
      "epoch": 0.49965604219215776,
      "grad_norm": 0.13981692010809643,
      "learning_rate": 0.0005248699591135664,
      "loss": 3.511,
      "step": 2179
    },
    {
      "epoch": 0.49988534739738594,
      "grad_norm": 0.15894194199567346,
      "learning_rate": 0.0005244990652905744,
      "loss": 3.7506,
      "step": 2180
    },
    {
      "epoch": 0.5001146526026141,
      "grad_norm": 0.14124858210615968,
      "learning_rate": 0.0005241281579540618,
      "loss": 3.5374,
      "step": 2181
    },
    {
      "epoch": 0.5003439578078422,
      "grad_norm": 0.140566441307058,
      "learning_rate": 0.0005237572373086188,
      "loss": 3.497,
      "step": 2182
    },
    {
      "epoch": 0.5005732630130704,
      "grad_norm": 0.1329162752252238,
      "learning_rate": 0.0005233863035588427,
      "loss": 3.3285,
      "step": 2183
    },
    {
      "epoch": 0.5008025682182986,
      "grad_norm": 0.14692289103210202,
      "learning_rate": 0.0005230153569093377,
      "loss": 3.4619,
      "step": 2184
    },
    {
      "epoch": 0.5010318734235267,
      "grad_norm": 0.1762908274196154,
      "learning_rate": 0.0005226443975647161,
      "loss": 3.4334,
      "step": 2185
    },
    {
      "epoch": 0.5012611786287549,
      "grad_norm": 0.15438722740969524,
      "learning_rate": 0.0005222734257295963,
      "loss": 3.534,
      "step": 2186
    },
    {
      "epoch": 0.501490483833983,
      "grad_norm": 0.13460821214164417,
      "learning_rate": 0.0005219024416086036,
      "loss": 3.5333,
      "step": 2187
    },
    {
      "epoch": 0.5017197890392112,
      "grad_norm": 0.1444824901726961,
      "learning_rate": 0.0005215314454063707,
      "loss": 3.417,
      "step": 2188
    },
    {
      "epoch": 0.5019490942444393,
      "grad_norm": 0.15058542975422662,
      "learning_rate": 0.0005211604373275366,
      "loss": 3.3766,
      "step": 2189
    },
    {
      "epoch": 0.5021783994496675,
      "grad_norm": 0.12848856747529616,
      "learning_rate": 0.000520789417576747,
      "loss": 3.3915,
      "step": 2190
    },
    {
      "epoch": 0.5024077046548957,
      "grad_norm": 0.12907874281861556,
      "learning_rate": 0.0005204183863586533,
      "loss": 3.5001,
      "step": 2191
    },
    {
      "epoch": 0.5026370098601238,
      "grad_norm": 0.1481080974719839,
      "learning_rate": 0.0005200473438779147,
      "loss": 3.4587,
      "step": 2192
    },
    {
      "epoch": 0.502866315065352,
      "grad_norm": 0.1383409640682494,
      "learning_rate": 0.0005196762903391951,
      "loss": 3.6544,
      "step": 2193
    },
    {
      "epoch": 0.5030956202705802,
      "grad_norm": 0.12635211469203023,
      "learning_rate": 0.0005193052259471655,
      "loss": 3.5453,
      "step": 2194
    },
    {
      "epoch": 0.5033249254758083,
      "grad_norm": 0.16125130348822958,
      "learning_rate": 0.0005189341509065023,
      "loss": 3.4555,
      "step": 2195
    },
    {
      "epoch": 0.5035542306810364,
      "grad_norm": 0.14671383259378767,
      "learning_rate": 0.0005185630654218881,
      "loss": 3.34,
      "step": 2196
    },
    {
      "epoch": 0.5037835358862646,
      "grad_norm": 0.18336653466735744,
      "learning_rate": 0.0005181919696980112,
      "loss": 3.5368,
      "step": 2197
    },
    {
      "epoch": 0.5040128410914928,
      "grad_norm": 0.14884892584621592,
      "learning_rate": 0.0005178208639395655,
      "loss": 3.6941,
      "step": 2198
    },
    {
      "epoch": 0.5042421462967209,
      "grad_norm": 0.13715642500588668,
      "learning_rate": 0.0005174497483512506,
      "loss": 3.3687,
      "step": 2199
    },
    {
      "epoch": 0.504471451501949,
      "grad_norm": 0.1483542794780037,
      "learning_rate": 0.0005170786231377711,
      "loss": 3.4533,
      "step": 2200
    },
    {
      "epoch": 0.5047007567071773,
      "grad_norm": 0.1386999526545699,
      "learning_rate": 0.0005167074885038374,
      "loss": 3.5863,
      "step": 2201
    },
    {
      "epoch": 0.5049300619124054,
      "grad_norm": 0.1399119898121294,
      "learning_rate": 0.0005163363446541647,
      "loss": 3.4436,
      "step": 2202
    },
    {
      "epoch": 0.5051593671176335,
      "grad_norm": 0.14811629805092866,
      "learning_rate": 0.0005159651917934735,
      "loss": 3.5301,
      "step": 2203
    },
    {
      "epoch": 0.5053886723228618,
      "grad_norm": 0.1369705247143089,
      "learning_rate": 0.0005155940301264896,
      "loss": 3.2546,
      "step": 2204
    },
    {
      "epoch": 0.5056179775280899,
      "grad_norm": 0.147175271918407,
      "learning_rate": 0.0005152228598579428,
      "loss": 3.5008,
      "step": 2205
    },
    {
      "epoch": 0.505847282733318,
      "grad_norm": 0.13067023802970745,
      "learning_rate": 0.0005148516811925684,
      "loss": 3.4596,
      "step": 2206
    },
    {
      "epoch": 0.5060765879385462,
      "grad_norm": 0.1468136139888926,
      "learning_rate": 0.000514480494335106,
      "loss": 3.4415,
      "step": 2207
    },
    {
      "epoch": 0.5063058931437744,
      "grad_norm": 0.1403848796664554,
      "learning_rate": 0.0005141092994902998,
      "loss": 3.5886,
      "step": 2208
    },
    {
      "epoch": 0.5065351983490025,
      "grad_norm": 0.13267106321074248,
      "learning_rate": 0.0005137380968628983,
      "loss": 3.5128,
      "step": 2209
    },
    {
      "epoch": 0.5067645035542306,
      "grad_norm": 0.15416822823530932,
      "learning_rate": 0.0005133668866576544,
      "loss": 3.4971,
      "step": 2210
    },
    {
      "epoch": 0.5069938087594589,
      "grad_norm": 0.13079358530235488,
      "learning_rate": 0.0005129956690793255,
      "loss": 3.5158,
      "step": 2211
    },
    {
      "epoch": 0.507223113964687,
      "grad_norm": 0.14542931285728503,
      "learning_rate": 0.0005126244443326722,
      "loss": 3.5285,
      "step": 2212
    },
    {
      "epoch": 0.5074524191699151,
      "grad_norm": 0.13911011042495552,
      "learning_rate": 0.0005122532126224601,
      "loss": 3.4444,
      "step": 2213
    },
    {
      "epoch": 0.5076817243751434,
      "grad_norm": 0.14273545220495282,
      "learning_rate": 0.0005118819741534577,
      "loss": 3.42,
      "step": 2214
    },
    {
      "epoch": 0.5079110295803715,
      "grad_norm": 0.14665698091846943,
      "learning_rate": 0.0005115107291304378,
      "loss": 3.4354,
      "step": 2215
    },
    {
      "epoch": 0.5081403347855996,
      "grad_norm": 0.1364060922812458,
      "learning_rate": 0.000511139477758177,
      "loss": 3.4028,
      "step": 2216
    },
    {
      "epoch": 0.5083696399908277,
      "grad_norm": 0.15304573143157615,
      "learning_rate": 0.0005107682202414544,
      "loss": 3.464,
      "step": 2217
    },
    {
      "epoch": 0.508598945196056,
      "grad_norm": 0.13450995919653208,
      "learning_rate": 0.0005103969567850538,
      "loss": 3.4434,
      "step": 2218
    },
    {
      "epoch": 0.5088282504012841,
      "grad_norm": 0.13876614629058787,
      "learning_rate": 0.0005100256875937613,
      "loss": 3.3678,
      "step": 2219
    },
    {
      "epoch": 0.5090575556065122,
      "grad_norm": 0.1453407124399724,
      "learning_rate": 0.0005096544128723667,
      "loss": 3.4832,
      "step": 2220
    },
    {
      "epoch": 0.5092868608117405,
      "grad_norm": 0.15437184042520302,
      "learning_rate": 0.0005092831328256625,
      "loss": 3.5823,
      "step": 2221
    },
    {
      "epoch": 0.5095161660169686,
      "grad_norm": 0.1405277699523195,
      "learning_rate": 0.0005089118476584442,
      "loss": 3.4579,
      "step": 2222
    },
    {
      "epoch": 0.5097454712221967,
      "grad_norm": 0.15010771828494765,
      "learning_rate": 0.0005085405575755105,
      "loss": 3.5854,
      "step": 2223
    },
    {
      "epoch": 0.509974776427425,
      "grad_norm": 0.13889066490912286,
      "learning_rate": 0.0005081692627816624,
      "loss": 3.4614,
      "step": 2224
    },
    {
      "epoch": 0.5102040816326531,
      "grad_norm": 0.12554855322588804,
      "learning_rate": 0.0005077979634817034,
      "loss": 3.5663,
      "step": 2225
    },
    {
      "epoch": 0.5104333868378812,
      "grad_norm": 0.14413665549867083,
      "learning_rate": 0.0005074266598804402,
      "loss": 3.4741,
      "step": 2226
    },
    {
      "epoch": 0.5106626920431094,
      "grad_norm": 0.14371954261961956,
      "learning_rate": 0.0005070553521826808,
      "loss": 3.613,
      "step": 2227
    },
    {
      "epoch": 0.5108919972483376,
      "grad_norm": 0.1310033274355903,
      "learning_rate": 0.0005066840405932363,
      "loss": 3.5536,
      "step": 2228
    },
    {
      "epoch": 0.5111213024535657,
      "grad_norm": 0.13437872583250687,
      "learning_rate": 0.00050631272531692,
      "loss": 3.494,
      "step": 2229
    },
    {
      "epoch": 0.5113506076587938,
      "grad_norm": 0.1329512553335537,
      "learning_rate": 0.0005059414065585465,
      "loss": 3.4414,
      "step": 2230
    },
    {
      "epoch": 0.511579912864022,
      "grad_norm": 0.13777157139830237,
      "learning_rate": 0.0005055700845229327,
      "loss": 3.427,
      "step": 2231
    },
    {
      "epoch": 0.5118092180692502,
      "grad_norm": 0.13787058989125026,
      "learning_rate": 0.0005051987594148979,
      "loss": 3.5584,
      "step": 2232
    },
    {
      "epoch": 0.5120385232744783,
      "grad_norm": 0.1305539846827834,
      "learning_rate": 0.000504827431439262,
      "loss": 3.4746,
      "step": 2233
    },
    {
      "epoch": 0.5122678284797065,
      "grad_norm": 0.14722995498237787,
      "learning_rate": 0.0005044561008008472,
      "loss": 3.4371,
      "step": 2234
    },
    {
      "epoch": 0.5124971336849347,
      "grad_norm": 0.1328493393721045,
      "learning_rate": 0.000504084767704477,
      "loss": 3.4276,
      "step": 2235
    },
    {
      "epoch": 0.5127264388901628,
      "grad_norm": 0.15897140061253928,
      "learning_rate": 0.0005037134323549763,
      "loss": 3.4226,
      "step": 2236
    },
    {
      "epoch": 0.512955744095391,
      "grad_norm": 0.1470417571767102,
      "learning_rate": 0.0005033420949571712,
      "loss": 3.6082,
      "step": 2237
    },
    {
      "epoch": 0.5131850493006191,
      "grad_norm": 0.13668650361538495,
      "learning_rate": 0.0005029707557158887,
      "loss": 3.4394,
      "step": 2238
    },
    {
      "epoch": 0.5134143545058473,
      "grad_norm": 0.14171451317396289,
      "learning_rate": 0.0005025994148359574,
      "loss": 3.4816,
      "step": 2239
    },
    {
      "epoch": 0.5136436597110754,
      "grad_norm": 0.1275431083623368,
      "learning_rate": 0.0005022280725222063,
      "loss": 3.4514,
      "step": 2240
    },
    {
      "epoch": 0.5138729649163036,
      "grad_norm": 0.12754541677105602,
      "learning_rate": 0.0005018567289794651,
      "loss": 3.4577,
      "step": 2241
    },
    {
      "epoch": 0.5141022701215318,
      "grad_norm": 0.14806642900446065,
      "learning_rate": 0.0005014853844125648,
      "loss": 3.658,
      "step": 2242
    },
    {
      "epoch": 0.5143315753267599,
      "grad_norm": 0.13855428359280747,
      "learning_rate": 0.0005011140390263362,
      "loss": 3.5483,
      "step": 2243
    },
    {
      "epoch": 0.5145608805319881,
      "grad_norm": 0.1576638542296356,
      "learning_rate": 0.0005007426930256113,
      "loss": 3.4739,
      "step": 2244
    },
    {
      "epoch": 0.5147901857372162,
      "grad_norm": 0.14843887754069687,
      "learning_rate": 0.0005003713466152218,
      "loss": 3.5241,
      "step": 2245
    },
    {
      "epoch": 0.5150194909424444,
      "grad_norm": 0.13055913096248395,
      "learning_rate": 0.0005,
      "loss": 3.4369,
      "step": 2246
    },
    {
      "epoch": 0.5152487961476726,
      "grad_norm": 0.15196220311914804,
      "learning_rate": 0.0004996286533847783,
      "loss": 3.6706,
      "step": 2247
    },
    {
      "epoch": 0.5154781013529007,
      "grad_norm": 0.14315513665314902,
      "learning_rate": 0.0004992573069743887,
      "loss": 3.433,
      "step": 2248
    },
    {
      "epoch": 0.5157074065581289,
      "grad_norm": 0.14594324540839107,
      "learning_rate": 0.000498885960973664,
      "loss": 3.4438,
      "step": 2249
    },
    {
      "epoch": 0.515936711763357,
      "grad_norm": 0.1520536127628956,
      "learning_rate": 0.0004985146155874353,
      "loss": 3.4636,
      "step": 2250
    },
    {
      "epoch": 0.5161660169685852,
      "grad_norm": 0.13081980291535447,
      "learning_rate": 0.000498143271020535,
      "loss": 3.5078,
      "step": 2251
    },
    {
      "epoch": 0.5163953221738133,
      "grad_norm": 0.14584991109277767,
      "learning_rate": 0.0004977719274777938,
      "loss": 3.5144,
      "step": 2252
    },
    {
      "epoch": 0.5166246273790415,
      "grad_norm": 0.14738254331763387,
      "learning_rate": 0.0004974005851640428,
      "loss": 3.5575,
      "step": 2253
    },
    {
      "epoch": 0.5168539325842697,
      "grad_norm": 0.13958269089027683,
      "learning_rate": 0.0004970292442841113,
      "loss": 3.5512,
      "step": 2254
    },
    {
      "epoch": 0.5170832377894978,
      "grad_norm": 0.13976419401052353,
      "learning_rate": 0.000496657905042829,
      "loss": 3.3504,
      "step": 2255
    },
    {
      "epoch": 0.517312542994726,
      "grad_norm": 0.13820230546278117,
      "learning_rate": 0.0004962865676450238,
      "loss": 3.378,
      "step": 2256
    },
    {
      "epoch": 0.5175418481999542,
      "grad_norm": 0.1381322713248955,
      "learning_rate": 0.0004959152322955232,
      "loss": 3.4256,
      "step": 2257
    },
    {
      "epoch": 0.5177711534051823,
      "grad_norm": 0.1350331094959808,
      "learning_rate": 0.0004955438991991528,
      "loss": 3.4539,
      "step": 2258
    },
    {
      "epoch": 0.5180004586104104,
      "grad_norm": 0.1280960178062229,
      "learning_rate": 0.0004951725685607382,
      "loss": 3.4443,
      "step": 2259
    },
    {
      "epoch": 0.5182297638156386,
      "grad_norm": 0.14953225198766149,
      "learning_rate": 0.0004948012405851022,
      "loss": 3.4302,
      "step": 2260
    },
    {
      "epoch": 0.5184590690208668,
      "grad_norm": 0.14651199901318232,
      "learning_rate": 0.0004944299154770673,
      "loss": 3.5333,
      "step": 2261
    },
    {
      "epoch": 0.5186883742260949,
      "grad_norm": 0.1427125738111923,
      "learning_rate": 0.0004940585934414535,
      "loss": 3.5779,
      "step": 2262
    },
    {
      "epoch": 0.518917679431323,
      "grad_norm": 0.13456158117896797,
      "learning_rate": 0.0004936872746830802,
      "loss": 3.4387,
      "step": 2263
    },
    {
      "epoch": 0.5191469846365513,
      "grad_norm": 0.14547288645361536,
      "learning_rate": 0.0004933159594067637,
      "loss": 3.5951,
      "step": 2264
    },
    {
      "epoch": 0.5193762898417794,
      "grad_norm": 0.13856874617939066,
      "learning_rate": 0.0004929446478173195,
      "loss": 3.528,
      "step": 2265
    },
    {
      "epoch": 0.5196055950470075,
      "grad_norm": 0.123502531163859,
      "learning_rate": 0.00049257334011956,
      "loss": 3.3357,
      "step": 2266
    },
    {
      "epoch": 0.5198349002522358,
      "grad_norm": 0.14725576373312682,
      "learning_rate": 0.0004922020365182968,
      "loss": 3.4125,
      "step": 2267
    },
    {
      "epoch": 0.5200642054574639,
      "grad_norm": 0.14486727338209815,
      "learning_rate": 0.0004918307372183378,
      "loss": 3.2631,
      "step": 2268
    },
    {
      "epoch": 0.520293510662692,
      "grad_norm": 0.14470239236924975,
      "learning_rate": 0.0004914594424244897,
      "loss": 3.5216,
      "step": 2269
    },
    {
      "epoch": 0.5205228158679202,
      "grad_norm": 0.1256871108375866,
      "learning_rate": 0.0004910881523415558,
      "loss": 3.6257,
      "step": 2270
    },
    {
      "epoch": 0.5207521210731484,
      "grad_norm": 0.13562697450149017,
      "learning_rate": 0.0004907168671743376,
      "loss": 3.541,
      "step": 2271
    },
    {
      "epoch": 0.5209814262783765,
      "grad_norm": 0.15376849646504284,
      "learning_rate": 0.0004903455871276334,
      "loss": 3.6143,
      "step": 2272
    },
    {
      "epoch": 0.5212107314836046,
      "grad_norm": 0.14106520602917313,
      "learning_rate": 0.0004899743124062387,
      "loss": 3.5701,
      "step": 2273
    },
    {
      "epoch": 0.5214400366888329,
      "grad_norm": 0.1407601893090992,
      "learning_rate": 0.0004896030432149462,
      "loss": 3.6759,
      "step": 2274
    },
    {
      "epoch": 0.521669341894061,
      "grad_norm": 0.14069875925645048,
      "learning_rate": 0.0004892317797585456,
      "loss": 3.5694,
      "step": 2275
    },
    {
      "epoch": 0.5218986470992891,
      "grad_norm": 0.1446632327782973,
      "learning_rate": 0.0004888605222418232,
      "loss": 3.5216,
      "step": 2276
    },
    {
      "epoch": 0.5221279523045174,
      "grad_norm": 0.13987357002101675,
      "learning_rate": 0.0004884892708695623,
      "loss": 3.4965,
      "step": 2277
    },
    {
      "epoch": 0.5223572575097455,
      "grad_norm": 0.14825125157447555,
      "learning_rate": 0.00048811802584654236,
      "loss": 3.5759,
      "step": 2278
    },
    {
      "epoch": 0.5225865627149736,
      "grad_norm": 0.13726053128392252,
      "learning_rate": 0.0004877467873775402,
      "loss": 3.5171,
      "step": 2279
    },
    {
      "epoch": 0.5228158679202017,
      "grad_norm": 0.15417444190003404,
      "learning_rate": 0.00048737555566732787,
      "loss": 3.4524,
      "step": 2280
    },
    {
      "epoch": 0.52304517312543,
      "grad_norm": 0.13857829734839056,
      "learning_rate": 0.00048700433092067473,
      "loss": 3.4985,
      "step": 2281
    },
    {
      "epoch": 0.5232744783306581,
      "grad_norm": 0.15052264473479332,
      "learning_rate": 0.0004866331133423456,
      "loss": 3.323,
      "step": 2282
    },
    {
      "epoch": 0.5235037835358862,
      "grad_norm": 0.13889959766688478,
      "learning_rate": 0.0004862619031371019,
      "loss": 3.5506,
      "step": 2283
    },
    {
      "epoch": 0.5237330887411145,
      "grad_norm": 0.1356667780342131,
      "learning_rate": 0.0004858907005097004,
      "loss": 3.2825,
      "step": 2284
    },
    {
      "epoch": 0.5239623939463426,
      "grad_norm": 0.13822726198947058,
      "learning_rate": 0.0004855195056648942,
      "loss": 3.5309,
      "step": 2285
    },
    {
      "epoch": 0.5241916991515707,
      "grad_norm": 0.14768388173492,
      "learning_rate": 0.0004851483188074317,
      "loss": 3.5806,
      "step": 2286
    },
    {
      "epoch": 0.524421004356799,
      "grad_norm": 0.14618431946178678,
      "learning_rate": 0.00048477714014205734,
      "loss": 3.4968,
      "step": 2287
    },
    {
      "epoch": 0.5246503095620271,
      "grad_norm": 0.1567059552274089,
      "learning_rate": 0.0004844059698735105,
      "loss": 3.4674,
      "step": 2288
    },
    {
      "epoch": 0.5248796147672552,
      "grad_norm": 0.1453880089318148,
      "learning_rate": 0.00048403480820652644,
      "loss": 3.5439,
      "step": 2289
    },
    {
      "epoch": 0.5251089199724833,
      "grad_norm": 0.12991287965994205,
      "learning_rate": 0.00048366365534583533,
      "loss": 3.4917,
      "step": 2290
    },
    {
      "epoch": 0.5253382251777116,
      "grad_norm": 0.1314600118259877,
      "learning_rate": 0.0004832925114961629,
      "loss": 3.5329,
      "step": 2291
    },
    {
      "epoch": 0.5255675303829397,
      "grad_norm": 0.14496302234200567,
      "learning_rate": 0.0004829213768622289,
      "loss": 3.3317,
      "step": 2292
    },
    {
      "epoch": 0.5257968355881678,
      "grad_norm": 0.1592068280277827,
      "learning_rate": 0.0004825502516487497,
      "loss": 3.4911,
      "step": 2293
    },
    {
      "epoch": 0.526026140793396,
      "grad_norm": 0.14781806766652353,
      "learning_rate": 0.00048217913606043457,
      "loss": 3.3314,
      "step": 2294
    },
    {
      "epoch": 0.5262554459986242,
      "grad_norm": 0.17132814350280454,
      "learning_rate": 0.00048180803030198896,
      "loss": 3.6156,
      "step": 2295
    },
    {
      "epoch": 0.5264847512038523,
      "grad_norm": 0.1515823397019961,
      "learning_rate": 0.0004814369345781121,
      "loss": 3.6173,
      "step": 2296
    },
    {
      "epoch": 0.5267140564090805,
      "grad_norm": 0.14221478828972056,
      "learning_rate": 0.0004810658490934979,
      "loss": 3.5587,
      "step": 2297
    },
    {
      "epoch": 0.5269433616143087,
      "grad_norm": 0.15535671311859212,
      "learning_rate": 0.0004806947740528347,
      "loss": 3.5226,
      "step": 2298
    },
    {
      "epoch": 0.5271726668195368,
      "grad_norm": 0.1457262294454733,
      "learning_rate": 0.000480323709660805,
      "loss": 3.5501,
      "step": 2299
    },
    {
      "epoch": 0.5274019720247649,
      "grad_norm": 0.14115831834715445,
      "learning_rate": 0.00047995265612208544,
      "loss": 3.4949,
      "step": 2300
    },
    {
      "epoch": 0.5276312772299931,
      "grad_norm": 0.13223699165263733,
      "learning_rate": 0.0004795816136413467,
      "loss": 3.399,
      "step": 2301
    },
    {
      "epoch": 0.5278605824352213,
      "grad_norm": 0.1502112431280003,
      "learning_rate": 0.00047921058242325317,
      "loss": 3.4662,
      "step": 2302
    },
    {
      "epoch": 0.5280898876404494,
      "grad_norm": 0.13699793666583177,
      "learning_rate": 0.00047883956267246353,
      "loss": 3.5065,
      "step": 2303
    },
    {
      "epoch": 0.5283191928456776,
      "grad_norm": 0.13990267977793694,
      "learning_rate": 0.00047846855459362934,
      "loss": 3.6215,
      "step": 2304
    },
    {
      "epoch": 0.5285484980509058,
      "grad_norm": 0.1454563144011376,
      "learning_rate": 0.00047809755839139657,
      "loss": 3.4996,
      "step": 2305
    },
    {
      "epoch": 0.5287778032561339,
      "grad_norm": 0.13901717100668842,
      "learning_rate": 0.00047772657427040384,
      "loss": 3.6174,
      "step": 2306
    },
    {
      "epoch": 0.5290071084613621,
      "grad_norm": 0.1377473437686566,
      "learning_rate": 0.0004773556024352841,
      "loss": 3.5219,
      "step": 2307
    },
    {
      "epoch": 0.5292364136665902,
      "grad_norm": 0.15017708313937256,
      "learning_rate": 0.00047698464309066233,
      "loss": 3.4916,
      "step": 2308
    },
    {
      "epoch": 0.5294657188718184,
      "grad_norm": 0.13701104868697644,
      "learning_rate": 0.00047661369644115754,
      "loss": 3.5916,
      "step": 2309
    },
    {
      "epoch": 0.5296950240770465,
      "grad_norm": 0.1626179065875042,
      "learning_rate": 0.0004762427626913813,
      "loss": 3.5993,
      "step": 2310
    },
    {
      "epoch": 0.5299243292822747,
      "grad_norm": 0.158845647828866,
      "learning_rate": 0.0004758718420459383,
      "loss": 3.5742,
      "step": 2311
    },
    {
      "epoch": 0.5301536344875029,
      "grad_norm": 0.13562540593407654,
      "learning_rate": 0.00047550093470942576,
      "loss": 3.4063,
      "step": 2312
    },
    {
      "epoch": 0.530382939692731,
      "grad_norm": 0.1324961414112339,
      "learning_rate": 0.0004751300408864339,
      "loss": 3.433,
      "step": 2313
    },
    {
      "epoch": 0.5306122448979592,
      "grad_norm": 0.13592195824623524,
      "learning_rate": 0.0004747591607815449,
      "loss": 3.4171,
      "step": 2314
    },
    {
      "epoch": 0.5308415501031873,
      "grad_norm": 0.14199963982745525,
      "learning_rate": 0.00047438829459933414,
      "loss": 3.4584,
      "step": 2315
    },
    {
      "epoch": 0.5310708553084155,
      "grad_norm": 0.14872076949131852,
      "learning_rate": 0.00047401744254436864,
      "loss": 3.6076,
      "step": 2316
    },
    {
      "epoch": 0.5313001605136437,
      "grad_norm": 0.1445821010534467,
      "learning_rate": 0.0004736466048212082,
      "loss": 3.4402,
      "step": 2317
    },
    {
      "epoch": 0.5315294657188718,
      "grad_norm": 0.14726828670628184,
      "learning_rate": 0.000473275781634404,
      "loss": 3.4977,
      "step": 2318
    },
    {
      "epoch": 0.5317587709241,
      "grad_norm": 0.13840432142492642,
      "learning_rate": 0.0004729049731885002,
      "loss": 3.5798,
      "step": 2319
    },
    {
      "epoch": 0.5319880761293282,
      "grad_norm": 0.1403563458705823,
      "learning_rate": 0.00047253417968803164,
      "loss": 3.5198,
      "step": 2320
    },
    {
      "epoch": 0.5322173813345563,
      "grad_norm": 0.14682694118194445,
      "learning_rate": 0.000472163401337526,
      "loss": 3.4965,
      "step": 2321
    },
    {
      "epoch": 0.5324466865397844,
      "grad_norm": 0.13495516546891628,
      "learning_rate": 0.0004717926383415019,
      "loss": 3.3956,
      "step": 2322
    },
    {
      "epoch": 0.5326759917450126,
      "grad_norm": 0.1345318755201247,
      "learning_rate": 0.00047142189090446985,
      "loss": 3.4085,
      "step": 2323
    },
    {
      "epoch": 0.5329052969502408,
      "grad_norm": 0.14552196392563227,
      "learning_rate": 0.0004710511592309315,
      "loss": 3.417,
      "step": 2324
    },
    {
      "epoch": 0.5331346021554689,
      "grad_norm": 0.15009199445866736,
      "learning_rate": 0.0004706804435253802,
      "loss": 3.5201,
      "step": 2325
    },
    {
      "epoch": 0.533363907360697,
      "grad_norm": 0.14281856631060308,
      "learning_rate": 0.0004703097439923,
      "loss": 3.5521,
      "step": 2326
    },
    {
      "epoch": 0.5335932125659253,
      "grad_norm": 0.13344965323530708,
      "learning_rate": 0.0004699390608361665,
      "loss": 3.3216,
      "step": 2327
    },
    {
      "epoch": 0.5338225177711534,
      "grad_norm": 0.1385263809223454,
      "learning_rate": 0.0004695683942614459,
      "loss": 3.4966,
      "step": 2328
    },
    {
      "epoch": 0.5340518229763815,
      "grad_norm": 0.14785913301060988,
      "learning_rate": 0.0004691977444725955,
      "loss": 3.662,
      "step": 2329
    },
    {
      "epoch": 0.5342811281816098,
      "grad_norm": 0.14379011763915558,
      "learning_rate": 0.0004688271116740632,
      "loss": 3.5018,
      "step": 2330
    },
    {
      "epoch": 0.5345104333868379,
      "grad_norm": 0.14375684274990658,
      "learning_rate": 0.0004684564960702877,
      "loss": 3.543,
      "step": 2331
    },
    {
      "epoch": 0.534739738592066,
      "grad_norm": 0.13393381480986966,
      "learning_rate": 0.0004680858978656976,
      "loss": 3.3888,
      "step": 2332
    },
    {
      "epoch": 0.5349690437972942,
      "grad_norm": 0.13362256419370647,
      "learning_rate": 0.0004677153172647131,
      "loss": 3.4084,
      "step": 2333
    },
    {
      "epoch": 0.5351983490025224,
      "grad_norm": 0.1326759749798979,
      "learning_rate": 0.00046734475447174345,
      "loss": 3.4396,
      "step": 2334
    },
    {
      "epoch": 0.5354276542077505,
      "grad_norm": 0.13192921229194693,
      "learning_rate": 0.00046697420969118894,
      "loss": 3.4886,
      "step": 2335
    },
    {
      "epoch": 0.5356569594129786,
      "grad_norm": 0.14345011308226846,
      "learning_rate": 0.0004666036831274392,
      "loss": 3.5791,
      "step": 2336
    },
    {
      "epoch": 0.5358862646182069,
      "grad_norm": 0.14943682433127095,
      "learning_rate": 0.00046623317498487466,
      "loss": 3.5074,
      "step": 2337
    },
    {
      "epoch": 0.536115569823435,
      "grad_norm": 0.14316517800801687,
      "learning_rate": 0.0004658626854678648,
      "loss": 3.423,
      "step": 2338
    },
    {
      "epoch": 0.5363448750286631,
      "grad_norm": 0.14377407607338477,
      "learning_rate": 0.0004654922147807694,
      "loss": 3.5204,
      "step": 2339
    },
    {
      "epoch": 0.5365741802338914,
      "grad_norm": 0.15349133324225883,
      "learning_rate": 0.00046512176312793734,
      "loss": 3.4803,
      "step": 2340
    },
    {
      "epoch": 0.5368034854391195,
      "grad_norm": 0.14588559338351387,
      "learning_rate": 0.00046475133071370757,
      "loss": 3.5866,
      "step": 2341
    },
    {
      "epoch": 0.5370327906443476,
      "grad_norm": 0.1615552512053596,
      "learning_rate": 0.0004643809177424078,
      "loss": 3.5399,
      "step": 2342
    },
    {
      "epoch": 0.5372620958495757,
      "grad_norm": 0.1372635607752529,
      "learning_rate": 0.00046401052441835574,
      "loss": 3.3514,
      "step": 2343
    },
    {
      "epoch": 0.537491401054804,
      "grad_norm": 0.13889215215993345,
      "learning_rate": 0.0004636401509458575,
      "loss": 3.4713,
      "step": 2344
    },
    {
      "epoch": 0.5377207062600321,
      "grad_norm": 0.22218866538866056,
      "learning_rate": 0.000463269797529209,
      "loss": 3.4105,
      "step": 2345
    },
    {
      "epoch": 0.5379500114652602,
      "grad_norm": 0.1407727024312711,
      "learning_rate": 0.0004628994643726941,
      "loss": 3.4943,
      "step": 2346
    },
    {
      "epoch": 0.5381793166704885,
      "grad_norm": 0.14137485420933896,
      "learning_rate": 0.00046252915168058697,
      "loss": 3.5146,
      "step": 2347
    },
    {
      "epoch": 0.5384086218757166,
      "grad_norm": 0.1422103444687226,
      "learning_rate": 0.00046215885965714875,
      "loss": 3.4886,
      "step": 2348
    },
    {
      "epoch": 0.5386379270809447,
      "grad_norm": 0.1396027708616848,
      "learning_rate": 0.0004617885885066305,
      "loss": 3.438,
      "step": 2349
    },
    {
      "epoch": 0.538867232286173,
      "grad_norm": 0.14257663219773486,
      "learning_rate": 0.0004614183384332712,
      "loss": 3.416,
      "step": 2350
    },
    {
      "epoch": 0.5390965374914011,
      "grad_norm": 0.14361349550608496,
      "learning_rate": 0.0004610481096412984,
      "loss": 3.6805,
      "step": 2351
    },
    {
      "epoch": 0.5393258426966292,
      "grad_norm": 0.14259056164382752,
      "learning_rate": 0.0004606779023349277,
      "loss": 3.3897,
      "step": 2352
    },
    {
      "epoch": 0.5395551479018573,
      "grad_norm": 0.1412333419706847,
      "learning_rate": 0.000460307716718363,
      "loss": 3.5649,
      "step": 2353
    },
    {
      "epoch": 0.5397844531070856,
      "grad_norm": 0.15127062751835416,
      "learning_rate": 0.00045993755299579623,
      "loss": 3.5649,
      "step": 2354
    },
    {
      "epoch": 0.5400137583123137,
      "grad_norm": 0.14850378015615345,
      "learning_rate": 0.0004595674113714074,
      "loss": 3.5884,
      "step": 2355
    },
    {
      "epoch": 0.5402430635175418,
      "grad_norm": 0.13613115352006575,
      "learning_rate": 0.0004591972920493638,
      "loss": 3.4138,
      "step": 2356
    },
    {
      "epoch": 0.54047236872277,
      "grad_norm": 0.1425662369975736,
      "learning_rate": 0.0004588271952338212,
      "loss": 3.4169,
      "step": 2357
    },
    {
      "epoch": 0.5407016739279982,
      "grad_norm": 0.14547648385215084,
      "learning_rate": 0.00045845712112892195,
      "loss": 3.4219,
      "step": 2358
    },
    {
      "epoch": 0.5409309791332263,
      "grad_norm": 0.14940825968476495,
      "learning_rate": 0.00045808706993879714,
      "loss": 3.4378,
      "step": 2359
    },
    {
      "epoch": 0.5411602843384545,
      "grad_norm": 0.18015499092740941,
      "learning_rate": 0.0004577170418675638,
      "loss": 3.5199,
      "step": 2360
    },
    {
      "epoch": 0.5413895895436827,
      "grad_norm": 0.15303602780777584,
      "learning_rate": 0.00045734703711932767,
      "loss": 3.4295,
      "step": 2361
    },
    {
      "epoch": 0.5416188947489108,
      "grad_norm": 0.1326161839952035,
      "learning_rate": 0.00045697705589818036,
      "loss": 3.4539,
      "step": 2362
    },
    {
      "epoch": 0.5418481999541389,
      "grad_norm": 0.14214171150811158,
      "learning_rate": 0.0004566070984082013,
      "loss": 3.5718,
      "step": 2363
    },
    {
      "epoch": 0.5420775051593671,
      "grad_norm": 0.14423617391530452,
      "learning_rate": 0.0004562371648534564,
      "loss": 3.4789,
      "step": 2364
    },
    {
      "epoch": 0.5423068103645953,
      "grad_norm": 0.1526093312934098,
      "learning_rate": 0.00045586725543799865,
      "loss": 3.5084,
      "step": 2365
    },
    {
      "epoch": 0.5425361155698234,
      "grad_norm": 0.14618703912416714,
      "learning_rate": 0.0004554973703658676,
      "loss": 3.4836,
      "step": 2366
    },
    {
      "epoch": 0.5427654207750516,
      "grad_norm": 0.15079073958121245,
      "learning_rate": 0.00045512750984108937,
      "loss": 3.4606,
      "step": 2367
    },
    {
      "epoch": 0.5429947259802798,
      "grad_norm": 0.1471791109063338,
      "learning_rate": 0.0004547576740676765,
      "loss": 3.4872,
      "step": 2368
    },
    {
      "epoch": 0.5432240311855079,
      "grad_norm": 0.1504453559599655,
      "learning_rate": 0.000454387863249628,
      "loss": 3.6679,
      "step": 2369
    },
    {
      "epoch": 0.5434533363907361,
      "grad_norm": 0.1210194838026993,
      "learning_rate": 0.00045401807759092895,
      "loss": 3.4491,
      "step": 2370
    },
    {
      "epoch": 0.5436826415959642,
      "grad_norm": 0.13414956639833633,
      "learning_rate": 0.00045364831729555096,
      "loss": 3.4681,
      "step": 2371
    },
    {
      "epoch": 0.5439119468011924,
      "grad_norm": 0.1261460922256347,
      "learning_rate": 0.0004532785825674507,
      "loss": 3.4917,
      "step": 2372
    },
    {
      "epoch": 0.5441412520064205,
      "grad_norm": 0.15640890057394385,
      "learning_rate": 0.0004529088736105721,
      "loss": 3.4563,
      "step": 2373
    },
    {
      "epoch": 0.5443705572116487,
      "grad_norm": 0.12987486913326907,
      "learning_rate": 0.00045253919062884356,
      "loss": 3.5067,
      "step": 2374
    },
    {
      "epoch": 0.5445998624168769,
      "grad_norm": 0.1373461749765522,
      "learning_rate": 0.0004521695338261802,
      "loss": 3.5499,
      "step": 2375
    },
    {
      "epoch": 0.544829167622105,
      "grad_norm": 0.1533036772357446,
      "learning_rate": 0.0004517999034064818,
      "loss": 3.5807,
      "step": 2376
    },
    {
      "epoch": 0.5450584728273332,
      "grad_norm": 0.1426853347035903,
      "learning_rate": 0.0004514302995736344,
      "loss": 3.4853,
      "step": 2377
    },
    {
      "epoch": 0.5452877780325613,
      "grad_norm": 0.13728677027209005,
      "learning_rate": 0.0004510607225315088,
      "loss": 3.4217,
      "step": 2378
    },
    {
      "epoch": 0.5455170832377895,
      "grad_norm": 0.13568000420197923,
      "learning_rate": 0.0004506911724839613,
      "loss": 3.4499,
      "step": 2379
    },
    {
      "epoch": 0.5457463884430177,
      "grad_norm": 0.15039457353715838,
      "learning_rate": 0.0004503216496348331,
      "loss": 3.5688,
      "step": 2380
    },
    {
      "epoch": 0.5459756936482458,
      "grad_norm": 0.1425239485863866,
      "learning_rate": 0.0004499521541879508,
      "loss": 3.5362,
      "step": 2381
    },
    {
      "epoch": 0.546204998853474,
      "grad_norm": 0.14243651620157075,
      "learning_rate": 0.0004495826863471254,
      "loss": 3.5198,
      "step": 2382
    },
    {
      "epoch": 0.5464343040587021,
      "grad_norm": 0.1644714472610764,
      "learning_rate": 0.00044921324631615303,
      "loss": 3.5676,
      "step": 2383
    },
    {
      "epoch": 0.5466636092639303,
      "grad_norm": 0.13548416367958613,
      "learning_rate": 0.0004488438342988141,
      "loss": 3.4788,
      "step": 2384
    },
    {
      "epoch": 0.5468929144691584,
      "grad_norm": 0.13166306201943284,
      "learning_rate": 0.0004484744504988742,
      "loss": 3.5717,
      "step": 2385
    },
    {
      "epoch": 0.5471222196743866,
      "grad_norm": 0.13688152815735496,
      "learning_rate": 0.0004481050951200824,
      "loss": 3.4728,
      "step": 2386
    },
    {
      "epoch": 0.5473515248796148,
      "grad_norm": 0.1527543602249593,
      "learning_rate": 0.00044773576836617336,
      "loss": 3.4336,
      "step": 2387
    },
    {
      "epoch": 0.5475808300848429,
      "grad_norm": 0.13249553961815483,
      "learning_rate": 0.0004473664704408646,
      "loss": 3.4873,
      "step": 2388
    },
    {
      "epoch": 0.547810135290071,
      "grad_norm": 0.13386872958116952,
      "learning_rate": 0.0004469972015478588,
      "loss": 3.4872,
      "step": 2389
    },
    {
      "epoch": 0.5480394404952993,
      "grad_norm": 0.13944560088566607,
      "learning_rate": 0.00044662796189084194,
      "loss": 3.4001,
      "step": 2390
    },
    {
      "epoch": 0.5482687457005274,
      "grad_norm": 0.141038075488471,
      "learning_rate": 0.0004462587516734844,
      "loss": 3.3715,
      "step": 2391
    },
    {
      "epoch": 0.5484980509057555,
      "grad_norm": 0.13757215082969093,
      "learning_rate": 0.00044588957109943986,
      "loss": 3.5744,
      "step": 2392
    },
    {
      "epoch": 0.5487273561109837,
      "grad_norm": 0.1403302052509228,
      "learning_rate": 0.00044552042037234596,
      "loss": 3.3428,
      "step": 2393
    },
    {
      "epoch": 0.5489566613162119,
      "grad_norm": 0.12963476676143787,
      "learning_rate": 0.0004451512996958237,
      "loss": 3.4206,
      "step": 2394
    },
    {
      "epoch": 0.54918596652144,
      "grad_norm": 0.13546758000781872,
      "learning_rate": 0.00044478220927347774,
      "loss": 3.4055,
      "step": 2395
    },
    {
      "epoch": 0.5494152717266682,
      "grad_norm": 0.14492084124548194,
      "learning_rate": 0.00044441314930889565,
      "loss": 3.5603,
      "step": 2396
    },
    {
      "epoch": 0.5496445769318964,
      "grad_norm": 0.14612808763011026,
      "learning_rate": 0.00044404412000564875,
      "loss": 3.4801,
      "step": 2397
    },
    {
      "epoch": 0.5498738821371245,
      "grad_norm": 0.15930035429894604,
      "learning_rate": 0.0004436751215672906,
      "loss": 3.584,
      "step": 2398
    },
    {
      "epoch": 0.5501031873423526,
      "grad_norm": 0.14605174000608848,
      "learning_rate": 0.000443306154197359,
      "loss": 3.6474,
      "step": 2399
    },
    {
      "epoch": 0.5503324925475809,
      "grad_norm": 0.1392284509307845,
      "learning_rate": 0.00044293721809937316,
      "loss": 3.4641,
      "step": 2400
    },
    {
      "epoch": 0.550561797752809,
      "grad_norm": 0.143054273024242,
      "learning_rate": 0.00044256831347683646,
      "loss": 3.727,
      "step": 2401
    },
    {
      "epoch": 0.5507911029580371,
      "grad_norm": 0.14887491966269153,
      "learning_rate": 0.0004421994405332336,
      "loss": 3.388,
      "step": 2402
    },
    {
      "epoch": 0.5510204081632653,
      "grad_norm": 0.13978723384482425,
      "learning_rate": 0.0004418305994720328,
      "loss": 3.3741,
      "step": 2403
    },
    {
      "epoch": 0.5512497133684935,
      "grad_norm": 0.13971263831471664,
      "learning_rate": 0.0004414617904966841,
      "loss": 3.4874,
      "step": 2404
    },
    {
      "epoch": 0.5514790185737216,
      "grad_norm": 0.14048722203157493,
      "learning_rate": 0.0004410930138106203,
      "loss": 3.6231,
      "step": 2405
    },
    {
      "epoch": 0.5517083237789497,
      "grad_norm": 0.13611543523401717,
      "learning_rate": 0.0004407242696172559,
      "loss": 3.4321,
      "step": 2406
    },
    {
      "epoch": 0.551937628984178,
      "grad_norm": 0.13708275825734748,
      "learning_rate": 0.000440355558119988,
      "loss": 3.6234,
      "step": 2407
    },
    {
      "epoch": 0.5521669341894061,
      "grad_norm": 0.14161255595817235,
      "learning_rate": 0.0004399868795221951,
      "loss": 3.3327,
      "step": 2408
    },
    {
      "epoch": 0.5523962393946342,
      "grad_norm": 0.13891764262237224,
      "learning_rate": 0.00043961823402723814,
      "loss": 3.513,
      "step": 2409
    },
    {
      "epoch": 0.5526255445998625,
      "grad_norm": 0.13443704915793112,
      "learning_rate": 0.00043924962183845953,
      "loss": 3.5577,
      "step": 2410
    },
    {
      "epoch": 0.5528548498050906,
      "grad_norm": 0.13546285161333724,
      "learning_rate": 0.0004388810431591829,
      "loss": 3.6747,
      "step": 2411
    },
    {
      "epoch": 0.5530841550103187,
      "grad_norm": 0.13173006527953965,
      "learning_rate": 0.00043851249819271403,
      "loss": 3.346,
      "step": 2412
    },
    {
      "epoch": 0.553313460215547,
      "grad_norm": 0.13050118235487168,
      "learning_rate": 0.0004381439871423398,
      "loss": 3.4324,
      "step": 2413
    },
    {
      "epoch": 0.5535427654207751,
      "grad_norm": 0.13458981119526164,
      "learning_rate": 0.0004377755102113286,
      "loss": 3.4253,
      "step": 2414
    },
    {
      "epoch": 0.5537720706260032,
      "grad_norm": 0.13151264076993716,
      "learning_rate": 0.00043740706760292966,
      "loss": 3.47,
      "step": 2415
    },
    {
      "epoch": 0.5540013758312313,
      "grad_norm": 0.14817018269647028,
      "learning_rate": 0.00043703865952037353,
      "loss": 3.6448,
      "step": 2416
    },
    {
      "epoch": 0.5542306810364596,
      "grad_norm": 0.13222264376370518,
      "learning_rate": 0.0004366702861668716,
      "loss": 3.5608,
      "step": 2417
    },
    {
      "epoch": 0.5544599862416877,
      "grad_norm": 0.13621502210512842,
      "learning_rate": 0.00043630194774561625,
      "loss": 3.5796,
      "step": 2418
    },
    {
      "epoch": 0.5546892914469158,
      "grad_norm": 0.14225818755638503,
      "learning_rate": 0.00043593364445978036,
      "loss": 3.4639,
      "step": 2419
    },
    {
      "epoch": 0.554918596652144,
      "grad_norm": 0.13274773497949913,
      "learning_rate": 0.0004355653765125178,
      "loss": 3.3525,
      "step": 2420
    },
    {
      "epoch": 0.5551479018573722,
      "grad_norm": 0.15152930901569664,
      "learning_rate": 0.0004351971441069622,
      "loss": 3.5238,
      "step": 2421
    },
    {
      "epoch": 0.5553772070626003,
      "grad_norm": 0.12774747939386305,
      "learning_rate": 0.00043482894744622874,
      "loss": 3.4704,
      "step": 2422
    },
    {
      "epoch": 0.5556065122678285,
      "grad_norm": 0.14044016411703183,
      "learning_rate": 0.0004344607867334116,
      "loss": 3.4495,
      "step": 2423
    },
    {
      "epoch": 0.5558358174730567,
      "grad_norm": 0.12597837172473048,
      "learning_rate": 0.0004340926621715864,
      "loss": 3.4469,
      "step": 2424
    },
    {
      "epoch": 0.5560651226782848,
      "grad_norm": 0.14387258519253,
      "learning_rate": 0.00043372457396380766,
      "loss": 3.433,
      "step": 2425
    },
    {
      "epoch": 0.5562944278835129,
      "grad_norm": 0.14995935455621046,
      "learning_rate": 0.00043335652231311076,
      "loss": 3.5054,
      "step": 2426
    },
    {
      "epoch": 0.5565237330887411,
      "grad_norm": 0.1540347552151845,
      "learning_rate": 0.00043298850742251013,
      "loss": 3.4899,
      "step": 2427
    },
    {
      "epoch": 0.5567530382939693,
      "grad_norm": 0.13575754588246652,
      "learning_rate": 0.0004326205294950006,
      "loss": 3.5985,
      "step": 2428
    },
    {
      "epoch": 0.5569823434991974,
      "grad_norm": 0.14094325130331012,
      "learning_rate": 0.0004322525887335563,
      "loss": 3.4124,
      "step": 2429
    },
    {
      "epoch": 0.5572116487044256,
      "grad_norm": 0.14119734339542975,
      "learning_rate": 0.0004318846853411309,
      "loss": 3.4455,
      "step": 2430
    },
    {
      "epoch": 0.5574409539096538,
      "grad_norm": 0.1394141877787113,
      "learning_rate": 0.00043151681952065734,
      "loss": 3.4327,
      "step": 2431
    },
    {
      "epoch": 0.5576702591148819,
      "grad_norm": 0.14700310970882818,
      "learning_rate": 0.0004311489914750482,
      "loss": 3.5301,
      "step": 2432
    },
    {
      "epoch": 0.5578995643201101,
      "grad_norm": 0.15103811930516237,
      "learning_rate": 0.00043078120140719456,
      "loss": 3.6474,
      "step": 2433
    },
    {
      "epoch": 0.5581288695253382,
      "grad_norm": 0.13137867097387354,
      "learning_rate": 0.0004304134495199674,
      "loss": 3.5439,
      "step": 2434
    },
    {
      "epoch": 0.5583581747305664,
      "grad_norm": 0.1450399712857061,
      "learning_rate": 0.0004300457360162158,
      "loss": 3.539,
      "step": 2435
    },
    {
      "epoch": 0.5585874799357945,
      "grad_norm": 0.14193579924648086,
      "learning_rate": 0.00042967806109876853,
      "loss": 3.4487,
      "step": 2436
    },
    {
      "epoch": 0.5588167851410227,
      "grad_norm": 0.1356598613353377,
      "learning_rate": 0.0004293104249704319,
      "loss": 3.4131,
      "step": 2437
    },
    {
      "epoch": 0.5590460903462509,
      "grad_norm": 0.1360667078379002,
      "learning_rate": 0.00042894282783399246,
      "loss": 3.4918,
      "step": 2438
    },
    {
      "epoch": 0.559275395551479,
      "grad_norm": 0.14740671691371734,
      "learning_rate": 0.00042857526989221355,
      "loss": 3.489,
      "step": 2439
    },
    {
      "epoch": 0.5595047007567072,
      "grad_norm": 0.13718591692873172,
      "learning_rate": 0.00042820775134783817,
      "loss": 3.3328,
      "step": 2440
    },
    {
      "epoch": 0.5597340059619353,
      "grad_norm": 0.1500592106749312,
      "learning_rate": 0.00042784027240358674,
      "loss": 3.3669,
      "step": 2441
    },
    {
      "epoch": 0.5599633111671635,
      "grad_norm": 0.15939551682962402,
      "learning_rate": 0.00042747283326215847,
      "loss": 3.5537,
      "step": 2442
    },
    {
      "epoch": 0.5601926163723917,
      "grad_norm": 0.15535296198264067,
      "learning_rate": 0.0004271054341262301,
      "loss": 3.6172,
      "step": 2443
    },
    {
      "epoch": 0.5604219215776198,
      "grad_norm": 0.148159975823159,
      "learning_rate": 0.00042673807519845675,
      "loss": 3.3355,
      "step": 2444
    },
    {
      "epoch": 0.560651226782848,
      "grad_norm": 0.15414336636791348,
      "learning_rate": 0.000426370756681471,
      "loss": 3.6125,
      "step": 2445
    },
    {
      "epoch": 0.5608805319880761,
      "grad_norm": 0.1663745340158044,
      "learning_rate": 0.0004260034787778833,
      "loss": 3.5111,
      "step": 2446
    },
    {
      "epoch": 0.5611098371933043,
      "grad_norm": 0.1488994564516886,
      "learning_rate": 0.0004256362416902817,
      "loss": 3.5108,
      "step": 2447
    },
    {
      "epoch": 0.5613391423985324,
      "grad_norm": 0.13818548025626515,
      "learning_rate": 0.00042526904562123193,
      "loss": 3.4395,
      "step": 2448
    },
    {
      "epoch": 0.5615684476037606,
      "grad_norm": 0.14703573426167973,
      "learning_rate": 0.00042490189077327637,
      "loss": 3.5261,
      "step": 2449
    },
    {
      "epoch": 0.5617977528089888,
      "grad_norm": 0.13669585208050197,
      "learning_rate": 0.0004245347773489359,
      "loss": 3.637,
      "step": 2450
    },
    {
      "epoch": 0.5620270580142169,
      "grad_norm": 0.14832577358694712,
      "learning_rate": 0.00042416770555070703,
      "loss": 3.4665,
      "step": 2451
    },
    {
      "epoch": 0.562256363219445,
      "grad_norm": 0.14623042551500842,
      "learning_rate": 0.0004238006755810648,
      "loss": 3.4033,
      "step": 2452
    },
    {
      "epoch": 0.5624856684246733,
      "grad_norm": 0.16041777575401636,
      "learning_rate": 0.00042343368764246,
      "loss": 3.4689,
      "step": 2453
    },
    {
      "epoch": 0.5627149736299014,
      "grad_norm": 0.14673495819339408,
      "learning_rate": 0.000423066741937321,
      "loss": 3.5352,
      "step": 2454
    },
    {
      "epoch": 0.5629442788351295,
      "grad_norm": 0.14617795035599954,
      "learning_rate": 0.0004226998386680524,
      "loss": 3.4992,
      "step": 2455
    },
    {
      "epoch": 0.5631735840403577,
      "grad_norm": 0.1563853622822488,
      "learning_rate": 0.00042233297803703583,
      "loss": 3.5325,
      "step": 2456
    },
    {
      "epoch": 0.5634028892455859,
      "grad_norm": 0.13976537369096775,
      "learning_rate": 0.000421966160246629,
      "loss": 3.4908,
      "step": 2457
    },
    {
      "epoch": 0.563632194450814,
      "grad_norm": 0.16814648066960605,
      "learning_rate": 0.0004215993854991662,
      "loss": 3.5497,
      "step": 2458
    },
    {
      "epoch": 0.5638614996560422,
      "grad_norm": 0.1588094272310501,
      "learning_rate": 0.00042123265399695783,
      "loss": 3.4939,
      "step": 2459
    },
    {
      "epoch": 0.5640908048612704,
      "grad_norm": 0.14543181158567464,
      "learning_rate": 0.00042086596594229065,
      "loss": 3.4939,
      "step": 2460
    },
    {
      "epoch": 0.5643201100664985,
      "grad_norm": 0.1404579573198478,
      "learning_rate": 0.0004204993215374273,
      "loss": 3.4222,
      "step": 2461
    },
    {
      "epoch": 0.5645494152717266,
      "grad_norm": 0.1516753068330918,
      "learning_rate": 0.0004201327209846065,
      "loss": 3.486,
      "step": 2462
    },
    {
      "epoch": 0.5647787204769549,
      "grad_norm": 0.13638623348400486,
      "learning_rate": 0.00041976616448604226,
      "loss": 3.4504,
      "step": 2463
    },
    {
      "epoch": 0.565008025682183,
      "grad_norm": 0.142225233308687,
      "learning_rate": 0.0004193996522439253,
      "loss": 3.6904,
      "step": 2464
    },
    {
      "epoch": 0.5652373308874111,
      "grad_norm": 0.14364429515080876,
      "learning_rate": 0.00041903318446042076,
      "loss": 3.4641,
      "step": 2465
    },
    {
      "epoch": 0.5654666360926393,
      "grad_norm": 0.1394199888912572,
      "learning_rate": 0.0004186667613376702,
      "loss": 3.3782,
      "step": 2466
    },
    {
      "epoch": 0.5656959412978675,
      "grad_norm": 0.13597409027365673,
      "learning_rate": 0.00041830038307778984,
      "loss": 3.6337,
      "step": 2467
    },
    {
      "epoch": 0.5659252465030956,
      "grad_norm": 0.13968905298206868,
      "learning_rate": 0.0004179340498828718,
      "loss": 3.5426,
      "step": 2468
    },
    {
      "epoch": 0.5661545517083237,
      "grad_norm": 0.13825166236412512,
      "learning_rate": 0.0004175677619549828,
      "loss": 3.5159,
      "step": 2469
    },
    {
      "epoch": 0.566383856913552,
      "grad_norm": 0.14394531426144827,
      "learning_rate": 0.00041720151949616496,
      "loss": 3.5055,
      "step": 2470
    },
    {
      "epoch": 0.5666131621187801,
      "grad_norm": 0.13522738339309737,
      "learning_rate": 0.000416835322708435,
      "loss": 3.4218,
      "step": 2471
    },
    {
      "epoch": 0.5668424673240082,
      "grad_norm": 0.13999266356148027,
      "learning_rate": 0.0004164691717937848,
      "loss": 3.468,
      "step": 2472
    },
    {
      "epoch": 0.5670717725292365,
      "grad_norm": 0.1383539379436375,
      "learning_rate": 0.00041610306695418056,
      "loss": 3.4817,
      "step": 2473
    },
    {
      "epoch": 0.5673010777344646,
      "grad_norm": 0.13762168192769741,
      "learning_rate": 0.0004157370083915636,
      "loss": 3.4489,
      "step": 2474
    },
    {
      "epoch": 0.5675303829396927,
      "grad_norm": 0.15414343413125717,
      "learning_rate": 0.0004153709963078488,
      "loss": 3.441,
      "step": 2475
    },
    {
      "epoch": 0.5677596881449208,
      "grad_norm": 0.1404454793006845,
      "learning_rate": 0.0004150050309049267,
      "loss": 3.5141,
      "step": 2476
    },
    {
      "epoch": 0.5679889933501491,
      "grad_norm": 0.14211117158072498,
      "learning_rate": 0.0004146391123846606,
      "loss": 3.4802,
      "step": 2477
    },
    {
      "epoch": 0.5682182985553772,
      "grad_norm": 0.15010897177853186,
      "learning_rate": 0.00041427324094888947,
      "loss": 3.4297,
      "step": 2478
    },
    {
      "epoch": 0.5684476037606053,
      "grad_norm": 0.13938429116838666,
      "learning_rate": 0.0004139074167994249,
      "loss": 3.5723,
      "step": 2479
    },
    {
      "epoch": 0.5686769089658336,
      "grad_norm": 0.13257710050309793,
      "learning_rate": 0.0004135416401380535,
      "loss": 3.44,
      "step": 2480
    },
    {
      "epoch": 0.5689062141710617,
      "grad_norm": 0.13110987468203078,
      "learning_rate": 0.00041317591116653486,
      "loss": 3.5316,
      "step": 2481
    },
    {
      "epoch": 0.5691355193762898,
      "grad_norm": 0.13734600243622622,
      "learning_rate": 0.000412810230086603,
      "loss": 3.4264,
      "step": 2482
    },
    {
      "epoch": 0.569364824581518,
      "grad_norm": 0.13989190689693898,
      "learning_rate": 0.0004124445970999648,
      "loss": 3.4557,
      "step": 2483
    },
    {
      "epoch": 0.5695941297867462,
      "grad_norm": 0.14261407450455169,
      "learning_rate": 0.00041207901240830126,
      "loss": 3.5835,
      "step": 2484
    },
    {
      "epoch": 0.5698234349919743,
      "grad_norm": 0.15453406793609675,
      "learning_rate": 0.00041171347621326627,
      "loss": 3.5074,
      "step": 2485
    },
    {
      "epoch": 0.5700527401972024,
      "grad_norm": 0.15135915635307617,
      "learning_rate": 0.0004113479887164873,
      "loss": 3.4422,
      "step": 2486
    },
    {
      "epoch": 0.5702820454024307,
      "grad_norm": 0.14660347229708756,
      "learning_rate": 0.00041098255011956465,
      "loss": 3.6145,
      "step": 2487
    },
    {
      "epoch": 0.5705113506076588,
      "grad_norm": 0.1551186327704907,
      "learning_rate": 0.00041061716062407204,
      "loss": 3.5758,
      "step": 2488
    },
    {
      "epoch": 0.5707406558128869,
      "grad_norm": 0.15583296461849572,
      "learning_rate": 0.00041025182043155547,
      "loss": 3.5117,
      "step": 2489
    },
    {
      "epoch": 0.5709699610181151,
      "grad_norm": 0.1510217296128952,
      "learning_rate": 0.00040988652974353467,
      "loss": 3.4243,
      "step": 2490
    },
    {
      "epoch": 0.5711992662233433,
      "grad_norm": 0.1409510388436008,
      "learning_rate": 0.000409521288761501,
      "loss": 3.4232,
      "step": 2491
    },
    {
      "epoch": 0.5714285714285714,
      "grad_norm": 0.15217160068829103,
      "learning_rate": 0.0004091560976869194,
      "loss": 3.4406,
      "step": 2492
    },
    {
      "epoch": 0.5716578766337996,
      "grad_norm": 0.15052822642518407,
      "learning_rate": 0.00040879095672122646,
      "loss": 3.4395,
      "step": 2493
    },
    {
      "epoch": 0.5718871818390278,
      "grad_norm": 0.14826188672687218,
      "learning_rate": 0.00040842586606583176,
      "loss": 3.6943,
      "step": 2494
    },
    {
      "epoch": 0.5721164870442559,
      "grad_norm": 0.13614917575613808,
      "learning_rate": 0.0004080608259221167,
      "loss": 3.4861,
      "step": 2495
    },
    {
      "epoch": 0.572345792249484,
      "grad_norm": 0.1612566666409039,
      "learning_rate": 0.00040769583649143517,
      "loss": 3.596,
      "step": 2496
    },
    {
      "epoch": 0.5725750974547122,
      "grad_norm": 0.14750376643026727,
      "learning_rate": 0.0004073308979751126,
      "loss": 3.542,
      "step": 2497
    },
    {
      "epoch": 0.5728044026599404,
      "grad_norm": 0.1357186148439824,
      "learning_rate": 0.00040696601057444695,
      "loss": 3.4247,
      "step": 2498
    },
    {
      "epoch": 0.5730337078651685,
      "grad_norm": 0.15481009407345,
      "learning_rate": 0.0004066011744907074,
      "loss": 3.4816,
      "step": 2499
    },
    {
      "epoch": 0.5732630130703967,
      "grad_norm": 0.14787390575248763,
      "learning_rate": 0.0004062363899251353,
      "loss": 3.5018,
      "step": 2500
    },
    {
      "epoch": 0.5734923182756249,
      "grad_norm": 0.14401675082251222,
      "learning_rate": 0.00040587165707894326,
      "loss": 3.5777,
      "step": 2501
    },
    {
      "epoch": 0.573721623480853,
      "grad_norm": 0.151396247605506,
      "learning_rate": 0.00040550697615331565,
      "loss": 3.6728,
      "step": 2502
    },
    {
      "epoch": 0.5739509286860812,
      "grad_norm": 0.13767826970391794,
      "learning_rate": 0.0004051423473494076,
      "loss": 3.4852,
      "step": 2503
    },
    {
      "epoch": 0.5741802338913093,
      "grad_norm": 0.1373279522272641,
      "learning_rate": 0.0004047777708683466,
      "loss": 3.4238,
      "step": 2504
    },
    {
      "epoch": 0.5744095390965375,
      "grad_norm": 0.1495486603855936,
      "learning_rate": 0.0004044132469112299,
      "loss": 3.6487,
      "step": 2505
    },
    {
      "epoch": 0.5746388443017657,
      "grad_norm": 0.15333089537085692,
      "learning_rate": 0.00040404877567912695,
      "loss": 3.4521,
      "step": 2506
    },
    {
      "epoch": 0.5748681495069938,
      "grad_norm": 0.1379362441847097,
      "learning_rate": 0.00040368435737307733,
      "loss": 3.392,
      "step": 2507
    },
    {
      "epoch": 0.575097454712222,
      "grad_norm": 0.14477167012891876,
      "learning_rate": 0.00040331999219409204,
      "loss": 3.3769,
      "step": 2508
    },
    {
      "epoch": 0.5753267599174501,
      "grad_norm": 0.13476982828346767,
      "learning_rate": 0.00040295568034315224,
      "loss": 3.451,
      "step": 2509
    },
    {
      "epoch": 0.5755560651226783,
      "grad_norm": 0.15001975858220049,
      "learning_rate": 0.00040259142202121,
      "loss": 3.6135,
      "step": 2510
    },
    {
      "epoch": 0.5757853703279064,
      "grad_norm": 0.14601471963897045,
      "learning_rate": 0.0004022272174291878,
      "loss": 3.5277,
      "step": 2511
    },
    {
      "epoch": 0.5760146755331346,
      "grad_norm": 0.1353435019953728,
      "learning_rate": 0.00040186306676797843,
      "loss": 3.5901,
      "step": 2512
    },
    {
      "epoch": 0.5762439807383628,
      "grad_norm": 0.13375264621806557,
      "learning_rate": 0.0004014989702384449,
      "loss": 3.4073,
      "step": 2513
    },
    {
      "epoch": 0.5764732859435909,
      "grad_norm": 0.13344819987174864,
      "learning_rate": 0.00040113492804142066,
      "loss": 3.3986,
      "step": 2514
    },
    {
      "epoch": 0.576702591148819,
      "grad_norm": 0.13096756347102564,
      "learning_rate": 0.00040077094037770843,
      "loss": 3.3832,
      "step": 2515
    },
    {
      "epoch": 0.5769318963540473,
      "grad_norm": 0.1417991507932659,
      "learning_rate": 0.00040040700744808206,
      "loss": 3.5186,
      "step": 2516
    },
    {
      "epoch": 0.5771612015592754,
      "grad_norm": 0.16421273555938112,
      "learning_rate": 0.0004000431294532838,
      "loss": 3.5331,
      "step": 2517
    },
    {
      "epoch": 0.5773905067645035,
      "grad_norm": 0.14534150728983541,
      "learning_rate": 0.0003996793065940271,
      "loss": 3.5743,
      "step": 2518
    },
    {
      "epoch": 0.5776198119697317,
      "grad_norm": 0.13767778294211047,
      "learning_rate": 0.0003993155390709935,
      "loss": 3.56,
      "step": 2519
    },
    {
      "epoch": 0.5778491171749599,
      "grad_norm": 0.16693607354743933,
      "learning_rate": 0.0003989518270848351,
      "loss": 3.4929,
      "step": 2520
    },
    {
      "epoch": 0.578078422380188,
      "grad_norm": 0.14645389878135726,
      "learning_rate": 0.0003985881708361729,
      "loss": 3.5015,
      "step": 2521
    },
    {
      "epoch": 0.5783077275854162,
      "grad_norm": 0.1512183878930163,
      "learning_rate": 0.0003982245705255974,
      "loss": 3.4075,
      "step": 2522
    },
    {
      "epoch": 0.5785370327906444,
      "grad_norm": 0.15929215674896835,
      "learning_rate": 0.00039786102635366784,
      "loss": 3.4653,
      "step": 2523
    },
    {
      "epoch": 0.5787663379958725,
      "grad_norm": 0.14586110481583528,
      "learning_rate": 0.00039749753852091305,
      "loss": 3.5064,
      "step": 2524
    },
    {
      "epoch": 0.5789956432011006,
      "grad_norm": 0.15896964389402318,
      "learning_rate": 0.0003971341072278302,
      "loss": 3.4026,
      "step": 2525
    },
    {
      "epoch": 0.5792249484063289,
      "grad_norm": 0.13718998009957795,
      "learning_rate": 0.0003967707326748857,
      "loss": 3.5468,
      "step": 2526
    },
    {
      "epoch": 0.579454253611557,
      "grad_norm": 0.14582230029559384,
      "learning_rate": 0.00039640741506251457,
      "loss": 3.5991,
      "step": 2527
    },
    {
      "epoch": 0.5796835588167851,
      "grad_norm": 0.14551783220174586,
      "learning_rate": 0.0003960441545911204,
      "loss": 3.3846,
      "step": 2528
    },
    {
      "epoch": 0.5799128640220133,
      "grad_norm": 0.1458349203374476,
      "learning_rate": 0.00039568095146107495,
      "loss": 3.4218,
      "step": 2529
    },
    {
      "epoch": 0.5801421692272415,
      "grad_norm": 0.1482387420048935,
      "learning_rate": 0.0003953178058727191,
      "loss": 3.4836,
      "step": 2530
    },
    {
      "epoch": 0.5803714744324696,
      "grad_norm": 0.14133244075246185,
      "learning_rate": 0.00039495471802636096,
      "loss": 3.421,
      "step": 2531
    },
    {
      "epoch": 0.5806007796376977,
      "grad_norm": 0.14173634043190497,
      "learning_rate": 0.000394591688122278,
      "loss": 3.5041,
      "step": 2532
    },
    {
      "epoch": 0.580830084842926,
      "grad_norm": 0.14144248805790877,
      "learning_rate": 0.0003942287163607145,
      "loss": 3.5022,
      "step": 2533
    },
    {
      "epoch": 0.5810593900481541,
      "grad_norm": 0.14625055261221376,
      "learning_rate": 0.0003938658029418837,
      "loss": 3.6015,
      "step": 2534
    },
    {
      "epoch": 0.5812886952533822,
      "grad_norm": 0.13580663297749898,
      "learning_rate": 0.0003935029480659658,
      "loss": 3.3879,
      "step": 2535
    },
    {
      "epoch": 0.5815180004586105,
      "grad_norm": 0.15439658729691538,
      "learning_rate": 0.00039314015193310953,
      "loss": 3.6307,
      "step": 2536
    },
    {
      "epoch": 0.5817473056638386,
      "grad_norm": 0.13889822161843765,
      "learning_rate": 0.00039277741474343054,
      "loss": 3.4773,
      "step": 2537
    },
    {
      "epoch": 0.5819766108690667,
      "grad_norm": 0.12831363390569797,
      "learning_rate": 0.00039241473669701237,
      "loss": 3.3809,
      "step": 2538
    },
    {
      "epoch": 0.5822059160742948,
      "grad_norm": 0.14651000229130165,
      "learning_rate": 0.0003920521179939057,
      "loss": 3.4765,
      "step": 2539
    },
    {
      "epoch": 0.5824352212795231,
      "grad_norm": 0.1427853653267412,
      "learning_rate": 0.00039168955883412874,
      "loss": 3.5248,
      "step": 2540
    },
    {
      "epoch": 0.5826645264847512,
      "grad_norm": 0.1390413910698605,
      "learning_rate": 0.00039132705941766644,
      "loss": 3.4984,
      "step": 2541
    },
    {
      "epoch": 0.5828938316899793,
      "grad_norm": 0.1584103825353161,
      "learning_rate": 0.00039096461994447146,
      "loss": 3.4633,
      "step": 2542
    },
    {
      "epoch": 0.5831231368952076,
      "grad_norm": 0.14869767814886134,
      "learning_rate": 0.0003906022406144624,
      "loss": 3.4564,
      "step": 2543
    },
    {
      "epoch": 0.5833524421004357,
      "grad_norm": 0.14236317752844205,
      "learning_rate": 0.000390239921627526,
      "loss": 3.4986,
      "step": 2544
    },
    {
      "epoch": 0.5835817473056638,
      "grad_norm": 0.14853148030477784,
      "learning_rate": 0.0003898776631835143,
      "loss": 3.5023,
      "step": 2545
    },
    {
      "epoch": 0.583811052510892,
      "grad_norm": 0.14217181733523163,
      "learning_rate": 0.0003895154654822471,
      "loss": 3.5657,
      "step": 2546
    },
    {
      "epoch": 0.5840403577161202,
      "grad_norm": 0.13871941386859973,
      "learning_rate": 0.00038915332872350994,
      "loss": 3.414,
      "step": 2547
    },
    {
      "epoch": 0.5842696629213483,
      "grad_norm": 0.14437233233816596,
      "learning_rate": 0.00038879125310705526,
      "loss": 3.4075,
      "step": 2548
    },
    {
      "epoch": 0.5844989681265764,
      "grad_norm": 0.15879225637190686,
      "learning_rate": 0.00038842923883260135,
      "loss": 3.4275,
      "step": 2549
    },
    {
      "epoch": 0.5847282733318047,
      "grad_norm": 0.17065979919346416,
      "learning_rate": 0.0003880672860998331,
      "loss": 3.6529,
      "step": 2550
    },
    {
      "epoch": 0.5849575785370328,
      "grad_norm": 0.15130249869759424,
      "learning_rate": 0.00038770539510840093,
      "loss": 3.4061,
      "step": 2551
    },
    {
      "epoch": 0.5851868837422609,
      "grad_norm": 0.14654183038525745,
      "learning_rate": 0.0003873435660579217,
      "loss": 3.4687,
      "step": 2552
    },
    {
      "epoch": 0.5854161889474891,
      "grad_norm": 0.1484098990354365,
      "learning_rate": 0.00038698179914797783,
      "loss": 3.3719,
      "step": 2553
    },
    {
      "epoch": 0.5856454941527173,
      "grad_norm": 0.14129207969615806,
      "learning_rate": 0.00038662009457811754,
      "loss": 3.3877,
      "step": 2554
    },
    {
      "epoch": 0.5858747993579454,
      "grad_norm": 0.15225632200179387,
      "learning_rate": 0.0003862584525478545,
      "loss": 3.6188,
      "step": 2555
    },
    {
      "epoch": 0.5861041045631736,
      "grad_norm": 0.14911547013799678,
      "learning_rate": 0.0003858968732566685,
      "loss": 3.6033,
      "step": 2556
    },
    {
      "epoch": 0.5863334097684018,
      "grad_norm": 0.15514279749999638,
      "learning_rate": 0.00038553535690400353,
      "loss": 3.6109,
      "step": 2557
    },
    {
      "epoch": 0.5865627149736299,
      "grad_norm": 0.15848344222022073,
      "learning_rate": 0.00038517390368927044,
      "loss": 3.4139,
      "step": 2558
    },
    {
      "epoch": 0.586792020178858,
      "grad_norm": 0.13467031596654633,
      "learning_rate": 0.00038481251381184355,
      "loss": 3.6125,
      "step": 2559
    },
    {
      "epoch": 0.5870213253840862,
      "grad_norm": 0.14203590528946444,
      "learning_rate": 0.0003844511874710638,
      "loss": 3.4621,
      "step": 2560
    },
    {
      "epoch": 0.5872506305893144,
      "grad_norm": 0.1524907628781939,
      "learning_rate": 0.00038408992486623584,
      "loss": 3.4863,
      "step": 2561
    },
    {
      "epoch": 0.5874799357945425,
      "grad_norm": 0.14158859830048767,
      "learning_rate": 0.00038372872619663013,
      "loss": 3.4399,
      "step": 2562
    },
    {
      "epoch": 0.5877092409997707,
      "grad_norm": 0.1468328223003866,
      "learning_rate": 0.00038336759166148117,
      "loss": 3.3361,
      "step": 2563
    },
    {
      "epoch": 0.5879385462049989,
      "grad_norm": 0.15386949791823917,
      "learning_rate": 0.00038300652145998855,
      "loss": 3.4639,
      "step": 2564
    },
    {
      "epoch": 0.588167851410227,
      "grad_norm": 0.1601559198872609,
      "learning_rate": 0.0003826455157913159,
      "loss": 3.3466,
      "step": 2565
    },
    {
      "epoch": 0.5883971566154552,
      "grad_norm": 0.15580295613989775,
      "learning_rate": 0.00038228457485459187,
      "loss": 3.7014,
      "step": 2566
    },
    {
      "epoch": 0.5886264618206833,
      "grad_norm": 0.16350862988682252,
      "learning_rate": 0.00038192369884890886,
      "loss": 3.6568,
      "step": 2567
    },
    {
      "epoch": 0.5888557670259115,
      "grad_norm": 0.16310369853021076,
      "learning_rate": 0.00038156288797332384,
      "loss": 3.4154,
      "step": 2568
    },
    {
      "epoch": 0.5890850722311396,
      "grad_norm": 0.14198926858660876,
      "learning_rate": 0.00038120214242685723,
      "loss": 3.4711,
      "step": 2569
    },
    {
      "epoch": 0.5893143774363678,
      "grad_norm": 0.16563536885091698,
      "learning_rate": 0.00038084146240849456,
      "loss": 3.3928,
      "step": 2570
    },
    {
      "epoch": 0.589543682641596,
      "grad_norm": 0.17382546522900064,
      "learning_rate": 0.00038048084811718373,
      "loss": 3.5311,
      "step": 2571
    },
    {
      "epoch": 0.5897729878468241,
      "grad_norm": 0.15788810024039554,
      "learning_rate": 0.00038012029975183795,
      "loss": 3.4959,
      "step": 2572
    },
    {
      "epoch": 0.5900022930520523,
      "grad_norm": 0.14210926276947905,
      "learning_rate": 0.0003797598175113327,
      "loss": 3.4501,
      "step": 2573
    },
    {
      "epoch": 0.5902315982572804,
      "grad_norm": 0.16920984186950794,
      "learning_rate": 0.00037939940159450775,
      "loss": 3.5407,
      "step": 2574
    },
    {
      "epoch": 0.5904609034625086,
      "grad_norm": 0.14074166118758477,
      "learning_rate": 0.0003790390522001662,
      "loss": 3.4766,
      "step": 2575
    },
    {
      "epoch": 0.5906902086677368,
      "grad_norm": 0.15381121728468386,
      "learning_rate": 0.00037867876952707425,
      "loss": 3.5698,
      "step": 2576
    },
    {
      "epoch": 0.5909195138729649,
      "grad_norm": 0.15214502522053025,
      "learning_rate": 0.0003783185537739615,
      "loss": 3.4325,
      "step": 2577
    },
    {
      "epoch": 0.591148819078193,
      "grad_norm": 0.14227866125559746,
      "learning_rate": 0.0003779584051395205,
      "loss": 3.3221,
      "step": 2578
    },
    {
      "epoch": 0.5913781242834212,
      "grad_norm": 0.13955888190865673,
      "learning_rate": 0.00037759832382240697,
      "loss": 3.3341,
      "step": 2579
    },
    {
      "epoch": 0.5916074294886494,
      "grad_norm": 0.13852459587070765,
      "learning_rate": 0.000377238310021239,
      "loss": 3.3891,
      "step": 2580
    },
    {
      "epoch": 0.5918367346938775,
      "grad_norm": 0.13223531911509337,
      "learning_rate": 0.00037687836393459826,
      "loss": 3.3835,
      "step": 2581
    },
    {
      "epoch": 0.5920660398991057,
      "grad_norm": 0.14860102142639756,
      "learning_rate": 0.00037651848576102814,
      "loss": 3.6357,
      "step": 2582
    },
    {
      "epoch": 0.5922953451043339,
      "grad_norm": 0.1376023207872264,
      "learning_rate": 0.0003761586756990354,
      "loss": 3.4479,
      "step": 2583
    },
    {
      "epoch": 0.592524650309562,
      "grad_norm": 0.14060574528005895,
      "learning_rate": 0.00037579893394708866,
      "loss": 3.51,
      "step": 2584
    },
    {
      "epoch": 0.5927539555147902,
      "grad_norm": 0.14761485026874546,
      "learning_rate": 0.0003754392607036191,
      "loss": 3.4291,
      "step": 2585
    },
    {
      "epoch": 0.5929832607200184,
      "grad_norm": 0.1444292745528682,
      "learning_rate": 0.00037507965616702014,
      "loss": 3.5712,
      "step": 2586
    },
    {
      "epoch": 0.5932125659252465,
      "grad_norm": 0.14666269248357,
      "learning_rate": 0.0003747201205356472,
      "loss": 3.3508,
      "step": 2587
    },
    {
      "epoch": 0.5934418711304746,
      "grad_norm": 0.13337405298728597,
      "learning_rate": 0.0003743606540078177,
      "loss": 3.4277,
      "step": 2588
    },
    {
      "epoch": 0.5936711763357028,
      "grad_norm": 0.14816933113741984,
      "learning_rate": 0.0003740012567818111,
      "loss": 3.5734,
      "step": 2589
    },
    {
      "epoch": 0.593900481540931,
      "grad_norm": 0.13201202736434078,
      "learning_rate": 0.00037364192905586823,
      "loss": 3.4181,
      "step": 2590
    },
    {
      "epoch": 0.5941297867461591,
      "grad_norm": 0.14328508447797506,
      "learning_rate": 0.0003732826710281922,
      "loss": 3.4194,
      "step": 2591
    },
    {
      "epoch": 0.5943590919513873,
      "grad_norm": 0.14689787311303595,
      "learning_rate": 0.000372923482896947,
      "loss": 3.3586,
      "step": 2592
    },
    {
      "epoch": 0.5945883971566155,
      "grad_norm": 0.14676644191435087,
      "learning_rate": 0.0003725643648602588,
      "loss": 3.4604,
      "step": 2593
    },
    {
      "epoch": 0.5948177023618436,
      "grad_norm": 0.15325576272363708,
      "learning_rate": 0.00037220531711621417,
      "loss": 3.4956,
      "step": 2594
    },
    {
      "epoch": 0.5950470075670717,
      "grad_norm": 0.14812451880072566,
      "learning_rate": 0.0003718463398628621,
      "loss": 3.5972,
      "step": 2595
    },
    {
      "epoch": 0.5952763127723,
      "grad_norm": 0.1487336625696047,
      "learning_rate": 0.00037148743329821143,
      "loss": 3.5219,
      "step": 2596
    },
    {
      "epoch": 0.5955056179775281,
      "grad_norm": 0.13160941856182545,
      "learning_rate": 0.0003711285976202331,
      "loss": 3.5484,
      "step": 2597
    },
    {
      "epoch": 0.5957349231827562,
      "grad_norm": 0.14193816328543238,
      "learning_rate": 0.000370769833026858,
      "loss": 3.3973,
      "step": 2598
    },
    {
      "epoch": 0.5959642283879845,
      "grad_norm": 0.13033386926702828,
      "learning_rate": 0.0003704111397159787,
      "loss": 3.5017,
      "step": 2599
    },
    {
      "epoch": 0.5961935335932126,
      "grad_norm": 0.14151266439566684,
      "learning_rate": 0.0003700525178854478,
      "loss": 3.5686,
      "step": 2600
    },
    {
      "epoch": 0.5964228387984407,
      "grad_norm": 0.14100609156365043,
      "learning_rate": 0.0003696939677330788,
      "loss": 3.4764,
      "step": 2601
    },
    {
      "epoch": 0.5966521440036688,
      "grad_norm": 0.14047283967366453,
      "learning_rate": 0.00036933548945664545,
      "loss": 3.5617,
      "step": 2602
    },
    {
      "epoch": 0.5968814492088971,
      "grad_norm": 0.14332398110613084,
      "learning_rate": 0.00036897708325388213,
      "loss": 3.424,
      "step": 2603
    },
    {
      "epoch": 0.5971107544141252,
      "grad_norm": 0.13853325779550202,
      "learning_rate": 0.00036861874932248316,
      "loss": 3.4092,
      "step": 2604
    },
    {
      "epoch": 0.5973400596193533,
      "grad_norm": 0.13815057822988758,
      "learning_rate": 0.0003682604878601034,
      "loss": 3.5467,
      "step": 2605
    },
    {
      "epoch": 0.5975693648245816,
      "grad_norm": 0.14143235813173272,
      "learning_rate": 0.000367902299064357,
      "loss": 3.4398,
      "step": 2606
    },
    {
      "epoch": 0.5977986700298097,
      "grad_norm": 0.15142456608390964,
      "learning_rate": 0.000367544183132819,
      "loss": 3.6604,
      "step": 2607
    },
    {
      "epoch": 0.5980279752350378,
      "grad_norm": 0.15344054118433265,
      "learning_rate": 0.00036718614026302334,
      "loss": 3.4819,
      "step": 2608
    },
    {
      "epoch": 0.598257280440266,
      "grad_norm": 0.14593524810447056,
      "learning_rate": 0.0003668281706524645,
      "loss": 3.4984,
      "step": 2609
    },
    {
      "epoch": 0.5984865856454942,
      "grad_norm": 0.1637832486251439,
      "learning_rate": 0.0003664702744985957,
      "loss": 3.5224,
      "step": 2610
    },
    {
      "epoch": 0.5987158908507223,
      "grad_norm": 0.13466580152055121,
      "learning_rate": 0.0003661124519988304,
      "loss": 3.4322,
      "step": 2611
    },
    {
      "epoch": 0.5989451960559504,
      "grad_norm": 0.1492440222865526,
      "learning_rate": 0.00036575470335054077,
      "loss": 3.6053,
      "step": 2612
    },
    {
      "epoch": 0.5991745012611787,
      "grad_norm": 0.14115854990055265,
      "learning_rate": 0.00036539702875105893,
      "loss": 3.547,
      "step": 2613
    },
    {
      "epoch": 0.5994038064664068,
      "grad_norm": 0.1522983421805283,
      "learning_rate": 0.00036503942839767546,
      "loss": 3.4983,
      "step": 2614
    },
    {
      "epoch": 0.5996331116716349,
      "grad_norm": 0.16013096860229556,
      "learning_rate": 0.0003646819024876406,
      "loss": 3.4565,
      "step": 2615
    },
    {
      "epoch": 0.5998624168768631,
      "grad_norm": 0.14154549722584878,
      "learning_rate": 0.00036432445121816306,
      "loss": 3.3952,
      "step": 2616
    },
    {
      "epoch": 0.6000917220820913,
      "grad_norm": 0.1461887868967534,
      "learning_rate": 0.0003639670747864107,
      "loss": 3.412,
      "step": 2617
    },
    {
      "epoch": 0.6003210272873194,
      "grad_norm": 0.1548246079285477,
      "learning_rate": 0.00036360977338950986,
      "loss": 3.6023,
      "step": 2618
    },
    {
      "epoch": 0.6005503324925476,
      "grad_norm": 0.14544482896414326,
      "learning_rate": 0.00036325254722454584,
      "loss": 3.5028,
      "step": 2619
    },
    {
      "epoch": 0.6007796376977758,
      "grad_norm": 0.1568670762214345,
      "learning_rate": 0.0003628953964885616,
      "loss": 3.4972,
      "step": 2620
    },
    {
      "epoch": 0.6010089429030039,
      "grad_norm": 0.13452623824620077,
      "learning_rate": 0.00036253832137855997,
      "loss": 3.4744,
      "step": 2621
    },
    {
      "epoch": 0.601238248108232,
      "grad_norm": 0.14654913220144342,
      "learning_rate": 0.00036218132209150044,
      "loss": 3.4445,
      "step": 2622
    },
    {
      "epoch": 0.6014675533134602,
      "grad_norm": 0.16430889927859818,
      "learning_rate": 0.00036182439882430183,
      "loss": 3.6282,
      "step": 2623
    },
    {
      "epoch": 0.6016968585186884,
      "grad_norm": 0.13884650975708543,
      "learning_rate": 0.0003614675517738405,
      "loss": 3.4471,
      "step": 2624
    },
    {
      "epoch": 0.6019261637239165,
      "grad_norm": 0.17389573665693642,
      "learning_rate": 0.00036111078113695096,
      "loss": 3.4213,
      "step": 2625
    },
    {
      "epoch": 0.6021554689291447,
      "grad_norm": 0.13791820744313876,
      "learning_rate": 0.00036075408711042534,
      "loss": 3.5008,
      "step": 2626
    },
    {
      "epoch": 0.6023847741343729,
      "grad_norm": 0.1341611510238957,
      "learning_rate": 0.0003603974698910139,
      "loss": 3.3339,
      "step": 2627
    },
    {
      "epoch": 0.602614079339601,
      "grad_norm": 0.13357341650644705,
      "learning_rate": 0.0003600409296754241,
      "loss": 3.4222,
      "step": 2628
    },
    {
      "epoch": 0.6028433845448292,
      "grad_norm": 0.14583019370222872,
      "learning_rate": 0.0003596844666603214,
      "loss": 3.4266,
      "step": 2629
    },
    {
      "epoch": 0.6030726897500573,
      "grad_norm": 0.1399120749346696,
      "learning_rate": 0.0003593280810423282,
      "loss": 3.5448,
      "step": 2630
    },
    {
      "epoch": 0.6033019949552855,
      "grad_norm": 0.15930927245448523,
      "learning_rate": 0.0003589717730180245,
      "loss": 3.4552,
      "step": 2631
    },
    {
      "epoch": 0.6035313001605136,
      "grad_norm": 0.14485759085382247,
      "learning_rate": 0.0003586155427839473,
      "loss": 3.3797,
      "step": 2632
    },
    {
      "epoch": 0.6037606053657418,
      "grad_norm": 0.13720096915265026,
      "learning_rate": 0.00035825939053659117,
      "loss": 3.5405,
      "step": 2633
    },
    {
      "epoch": 0.60398991057097,
      "grad_norm": 0.13366285195284996,
      "learning_rate": 0.00035790331647240684,
      "loss": 3.4678,
      "step": 2634
    },
    {
      "epoch": 0.6042192157761981,
      "grad_norm": 0.14147696934252743,
      "learning_rate": 0.00035754732078780273,
      "loss": 3.6457,
      "step": 2635
    },
    {
      "epoch": 0.6044485209814263,
      "grad_norm": 0.14043155046953582,
      "learning_rate": 0.0003571914036791435,
      "loss": 3.5871,
      "step": 2636
    },
    {
      "epoch": 0.6046778261866544,
      "grad_norm": 0.14228708358833791,
      "learning_rate": 0.00035683556534275076,
      "loss": 3.3511,
      "step": 2637
    },
    {
      "epoch": 0.6049071313918826,
      "grad_norm": 0.13907340550763295,
      "learning_rate": 0.0003564798059749023,
      "loss": 3.5014,
      "step": 2638
    },
    {
      "epoch": 0.6051364365971108,
      "grad_norm": 0.14295256484728355,
      "learning_rate": 0.00035612412577183303,
      "loss": 3.4511,
      "step": 2639
    },
    {
      "epoch": 0.6053657418023389,
      "grad_norm": 0.136904637012762,
      "learning_rate": 0.0003557685249297333,
      "loss": 3.4975,
      "step": 2640
    },
    {
      "epoch": 0.6055950470075671,
      "grad_norm": 0.14519077386265095,
      "learning_rate": 0.00035541300364475063,
      "loss": 3.4927,
      "step": 2641
    },
    {
      "epoch": 0.6058243522127952,
      "grad_norm": 0.14664364884151584,
      "learning_rate": 0.00035505756211298776,
      "loss": 3.6619,
      "step": 2642
    },
    {
      "epoch": 0.6060536574180234,
      "grad_norm": 0.1426479134705805,
      "learning_rate": 0.0003547022005305043,
      "loss": 3.4395,
      "step": 2643
    },
    {
      "epoch": 0.6062829626232515,
      "grad_norm": 0.13853208113459564,
      "learning_rate": 0.00035434691909331487,
      "loss": 3.3642,
      "step": 2644
    },
    {
      "epoch": 0.6065122678284797,
      "grad_norm": 0.14524563325617298,
      "learning_rate": 0.0003539917179973907,
      "loss": 3.6011,
      "step": 2645
    },
    {
      "epoch": 0.6067415730337079,
      "grad_norm": 0.14753191008956676,
      "learning_rate": 0.00035363659743865795,
      "loss": 3.4888,
      "step": 2646
    },
    {
      "epoch": 0.606970878238936,
      "grad_norm": 0.1443681622832895,
      "learning_rate": 0.00035328155761299917,
      "loss": 3.4259,
      "step": 2647
    },
    {
      "epoch": 0.6072001834441642,
      "grad_norm": 0.13328703924643334,
      "learning_rate": 0.00035292659871625153,
      "loss": 3.3197,
      "step": 2648
    },
    {
      "epoch": 0.6074294886493924,
      "grad_norm": 0.14702820167246486,
      "learning_rate": 0.0003525717209442085,
      "loss": 3.3772,
      "step": 2649
    },
    {
      "epoch": 0.6076587938546205,
      "grad_norm": 0.14481588994518904,
      "learning_rate": 0.00035221692449261785,
      "loss": 3.5274,
      "step": 2650
    },
    {
      "epoch": 0.6078880990598486,
      "grad_norm": 0.13437450057205702,
      "learning_rate": 0.00035186220955718306,
      "loss": 3.3165,
      "step": 2651
    },
    {
      "epoch": 0.6081174042650768,
      "grad_norm": 0.15437403019153229,
      "learning_rate": 0.0003515075763335625,
      "loss": 3.4702,
      "step": 2652
    },
    {
      "epoch": 0.608346709470305,
      "grad_norm": 0.1405924575835941,
      "learning_rate": 0.0003511530250173696,
      "loss": 3.4547,
      "step": 2653
    },
    {
      "epoch": 0.6085760146755331,
      "grad_norm": 0.1537916385350099,
      "learning_rate": 0.0003507985558041723,
      "loss": 3.694,
      "step": 2654
    },
    {
      "epoch": 0.6088053198807613,
      "grad_norm": 0.15018909470553893,
      "learning_rate": 0.00035044416888949364,
      "loss": 3.4421,
      "step": 2655
    },
    {
      "epoch": 0.6090346250859895,
      "grad_norm": 0.15298705404890414,
      "learning_rate": 0.00035008986446881087,
      "loss": 3.476,
      "step": 2656
    },
    {
      "epoch": 0.6092639302912176,
      "grad_norm": 0.15053035934680412,
      "learning_rate": 0.0003497356427375562,
      "loss": 3.3476,
      "step": 2657
    },
    {
      "epoch": 0.6094932354964457,
      "grad_norm": 0.14673372870905313,
      "learning_rate": 0.0003493815038911157,
      "loss": 3.4973,
      "step": 2658
    },
    {
      "epoch": 0.609722540701674,
      "grad_norm": 0.1433680810621852,
      "learning_rate": 0.00034902744812483034,
      "loss": 3.4949,
      "step": 2659
    },
    {
      "epoch": 0.6099518459069021,
      "grad_norm": 0.13390895388030205,
      "learning_rate": 0.0003486734756339943,
      "loss": 3.4235,
      "step": 2660
    },
    {
      "epoch": 0.6101811511121302,
      "grad_norm": 0.1473099466694107,
      "learning_rate": 0.00034831958661385714,
      "loss": 3.4501,
      "step": 2661
    },
    {
      "epoch": 0.6104104563173584,
      "grad_norm": 0.16061970295469488,
      "learning_rate": 0.00034796578125962117,
      "loss": 3.6077,
      "step": 2662
    },
    {
      "epoch": 0.6106397615225866,
      "grad_norm": 0.16508904113594897,
      "learning_rate": 0.0003476120597664434,
      "loss": 3.5376,
      "step": 2663
    },
    {
      "epoch": 0.6108690667278147,
      "grad_norm": 0.1407523673795753,
      "learning_rate": 0.0003472584223294339,
      "loss": 3.3804,
      "step": 2664
    },
    {
      "epoch": 0.6110983719330428,
      "grad_norm": 0.1603882177997997,
      "learning_rate": 0.00034690486914365704,
      "loss": 3.5918,
      "step": 2665
    },
    {
      "epoch": 0.6113276771382711,
      "grad_norm": 0.16398514213102353,
      "learning_rate": 0.00034655140040413005,
      "loss": 3.4121,
      "step": 2666
    },
    {
      "epoch": 0.6115569823434992,
      "grad_norm": 0.1462832913559012,
      "learning_rate": 0.00034619801630582435,
      "loss": 3.63,
      "step": 2667
    },
    {
      "epoch": 0.6117862875487273,
      "grad_norm": 0.15142486470037828,
      "learning_rate": 0.0003458447170436638,
      "loss": 3.4371,
      "step": 2668
    },
    {
      "epoch": 0.6120155927539556,
      "grad_norm": 0.1392048327431358,
      "learning_rate": 0.00034549150281252633,
      "loss": 3.4786,
      "step": 2669
    },
    {
      "epoch": 0.6122448979591837,
      "grad_norm": 0.15305887879918334,
      "learning_rate": 0.0003451383738072422,
      "loss": 3.1979,
      "step": 2670
    },
    {
      "epoch": 0.6124742031644118,
      "grad_norm": 0.16192021909568163,
      "learning_rate": 0.0003447853302225952,
      "loss": 3.6001,
      "step": 2671
    },
    {
      "epoch": 0.6127035083696399,
      "grad_norm": 0.14521650689229026,
      "learning_rate": 0.00034443237225332174,
      "loss": 3.5991,
      "step": 2672
    },
    {
      "epoch": 0.6129328135748682,
      "grad_norm": 0.16851896566522495,
      "learning_rate": 0.00034407950009411126,
      "loss": 3.4251,
      "step": 2673
    },
    {
      "epoch": 0.6131621187800963,
      "grad_norm": 0.1569779302424886,
      "learning_rate": 0.00034372671393960517,
      "loss": 3.5089,
      "step": 2674
    },
    {
      "epoch": 0.6133914239853244,
      "grad_norm": 0.14704539110445705,
      "learning_rate": 0.00034337401398439873,
      "loss": 3.3967,
      "step": 2675
    },
    {
      "epoch": 0.6136207291905527,
      "grad_norm": 0.15741883369873041,
      "learning_rate": 0.0003430214004230381,
      "loss": 3.5743,
      "step": 2676
    },
    {
      "epoch": 0.6138500343957808,
      "grad_norm": 0.14976121545580037,
      "learning_rate": 0.00034266887345002305,
      "loss": 3.3648,
      "step": 2677
    },
    {
      "epoch": 0.6140793396010089,
      "grad_norm": 0.16019830619029649,
      "learning_rate": 0.0003423164332598049,
      "loss": 3.623,
      "step": 2678
    },
    {
      "epoch": 0.6143086448062371,
      "grad_norm": 0.154556499734848,
      "learning_rate": 0.0003419640800467874,
      "loss": 3.5351,
      "step": 2679
    },
    {
      "epoch": 0.6145379500114653,
      "grad_norm": 0.15412494297456739,
      "learning_rate": 0.0003416118140053261,
      "loss": 3.3287,
      "step": 2680
    },
    {
      "epoch": 0.6147672552166934,
      "grad_norm": 0.15388505012846007,
      "learning_rate": 0.0003412596353297288,
      "loss": 3.5019,
      "step": 2681
    },
    {
      "epoch": 0.6149965604219215,
      "grad_norm": 0.14108984721273943,
      "learning_rate": 0.0003409075442142546,
      "loss": 3.403,
      "step": 2682
    },
    {
      "epoch": 0.6152258656271498,
      "grad_norm": 0.17444243112432312,
      "learning_rate": 0.00034055554085311493,
      "loss": 3.3893,
      "step": 2683
    },
    {
      "epoch": 0.6154551708323779,
      "grad_norm": 0.1674863057013658,
      "learning_rate": 0.0003402036254404722,
      "loss": 3.5854,
      "step": 2684
    },
    {
      "epoch": 0.615684476037606,
      "grad_norm": 0.14032506012649454,
      "learning_rate": 0.00033985179817044105,
      "loss": 3.431,
      "step": 2685
    },
    {
      "epoch": 0.6159137812428342,
      "grad_norm": 0.14363185849798552,
      "learning_rate": 0.0003395000592370864,
      "loss": 3.4887,
      "step": 2686
    },
    {
      "epoch": 0.6161430864480624,
      "grad_norm": 0.15411167014689167,
      "learning_rate": 0.0003391484088344257,
      "loss": 3.6049,
      "step": 2687
    },
    {
      "epoch": 0.6163723916532905,
      "grad_norm": 0.14710651193351054,
      "learning_rate": 0.0003387968471564263,
      "loss": 3.5768,
      "step": 2688
    },
    {
      "epoch": 0.6166016968585187,
      "grad_norm": 0.1489055522237036,
      "learning_rate": 0.00033844537439700807,
      "loss": 3.4233,
      "step": 2689
    },
    {
      "epoch": 0.6168310020637469,
      "grad_norm": 0.14706396171709896,
      "learning_rate": 0.00033809399075004036,
      "loss": 3.5352,
      "step": 2690
    },
    {
      "epoch": 0.617060307268975,
      "grad_norm": 0.13782967251521633,
      "learning_rate": 0.00033774269640934445,
      "loss": 3.389,
      "step": 2691
    },
    {
      "epoch": 0.6172896124742032,
      "grad_norm": 0.15002163580095987,
      "learning_rate": 0.00033739149156869166,
      "loss": 3.5247,
      "step": 2692
    },
    {
      "epoch": 0.6175189176794313,
      "grad_norm": 0.1482134557340793,
      "learning_rate": 0.0003370403764218045,
      "loss": 3.417,
      "step": 2693
    },
    {
      "epoch": 0.6177482228846595,
      "grad_norm": 0.14397990207021036,
      "learning_rate": 0.0003366893511623555,
      "loss": 3.4375,
      "step": 2694
    },
    {
      "epoch": 0.6179775280898876,
      "grad_norm": 0.14187227187940357,
      "learning_rate": 0.000336338415983968,
      "loss": 3.4612,
      "step": 2695
    },
    {
      "epoch": 0.6182068332951158,
      "grad_norm": 0.13181164548928045,
      "learning_rate": 0.00033598757108021545,
      "loss": 3.3354,
      "step": 2696
    },
    {
      "epoch": 0.618436138500344,
      "grad_norm": 0.17037250829084014,
      "learning_rate": 0.00033563681664462155,
      "loss": 3.5104,
      "step": 2697
    },
    {
      "epoch": 0.6186654437055721,
      "grad_norm": 0.14931542047736576,
      "learning_rate": 0.00033528615287066005,
      "loss": 3.4932,
      "step": 2698
    },
    {
      "epoch": 0.6188947489108003,
      "grad_norm": 0.1411866112283677,
      "learning_rate": 0.000334935579951755,
      "loss": 3.4344,
      "step": 2699
    },
    {
      "epoch": 0.6191240541160284,
      "grad_norm": 0.14553196651600606,
      "learning_rate": 0.0003345850980812795,
      "loss": 3.3444,
      "step": 2700
    },
    {
      "epoch": 0.6193533593212566,
      "grad_norm": 0.1424286115160167,
      "learning_rate": 0.0003342347074525578,
      "loss": 3.5528,
      "step": 2701
    },
    {
      "epoch": 0.6195826645264848,
      "grad_norm": 0.1538390562141881,
      "learning_rate": 0.00033388440825886216,
      "loss": 3.5619,
      "step": 2702
    },
    {
      "epoch": 0.6198119697317129,
      "grad_norm": 0.1465384935196937,
      "learning_rate": 0.0003335342006934161,
      "loss": 3.4569,
      "step": 2703
    },
    {
      "epoch": 0.6200412749369411,
      "grad_norm": 0.1395103161252614,
      "learning_rate": 0.0003331840849493911,
      "loss": 3.4921,
      "step": 2704
    },
    {
      "epoch": 0.6202705801421692,
      "grad_norm": 0.15136889328707537,
      "learning_rate": 0.00033283406121990914,
      "loss": 3.3684,
      "step": 2705
    },
    {
      "epoch": 0.6204998853473974,
      "grad_norm": 0.13989507454842032,
      "learning_rate": 0.00033248412969804063,
      "loss": 3.5408,
      "step": 2706
    },
    {
      "epoch": 0.6207291905526255,
      "grad_norm": 0.14307736599080212,
      "learning_rate": 0.0003321342905768057,
      "loss": 3.5082,
      "step": 2707
    },
    {
      "epoch": 0.6209584957578537,
      "grad_norm": 0.142418166519866,
      "learning_rate": 0.0003317845440491731,
      "loss": 3.4695,
      "step": 2708
    },
    {
      "epoch": 0.6211878009630819,
      "grad_norm": 0.14147185364948167,
      "learning_rate": 0.00033143489030806086,
      "loss": 3.4282,
      "step": 2709
    },
    {
      "epoch": 0.62141710616831,
      "grad_norm": 0.13566221977395423,
      "learning_rate": 0.0003310853295463353,
      "loss": 3.5217,
      "step": 2710
    },
    {
      "epoch": 0.6216464113735382,
      "grad_norm": 0.13990930272393254,
      "learning_rate": 0.00033073586195681227,
      "loss": 3.5098,
      "step": 2711
    },
    {
      "epoch": 0.6218757165787664,
      "grad_norm": 0.14582328284747592,
      "learning_rate": 0.0003303864877322553,
      "loss": 3.4519,
      "step": 2712
    },
    {
      "epoch": 0.6221050217839945,
      "grad_norm": 0.15047841566797848,
      "learning_rate": 0.00033003720706537736,
      "loss": 3.4683,
      "step": 2713
    },
    {
      "epoch": 0.6223343269892226,
      "grad_norm": 0.15635762305372952,
      "learning_rate": 0.0003296880201488387,
      "loss": 3.5328,
      "step": 2714
    },
    {
      "epoch": 0.6225636321944508,
      "grad_norm": 0.15822474719186805,
      "learning_rate": 0.0003293389271752492,
      "loss": 3.4753,
      "step": 2715
    },
    {
      "epoch": 0.622792937399679,
      "grad_norm": 0.1582494030639349,
      "learning_rate": 0.0003289899283371657,
      "loss": 3.5082,
      "step": 2716
    },
    {
      "epoch": 0.6230222426049071,
      "grad_norm": 0.15662602281623847,
      "learning_rate": 0.00032864102382709374,
      "loss": 3.3946,
      "step": 2717
    },
    {
      "epoch": 0.6232515478101353,
      "grad_norm": 0.148163060142565,
      "learning_rate": 0.0003282922138374868,
      "loss": 3.4678,
      "step": 2718
    },
    {
      "epoch": 0.6234808530153635,
      "grad_norm": 0.14562367660268752,
      "learning_rate": 0.000327943498560746,
      "loss": 3.505,
      "step": 2719
    },
    {
      "epoch": 0.6237101582205916,
      "grad_norm": 0.13901678103994491,
      "learning_rate": 0.0003275948781892205,
      "loss": 3.395,
      "step": 2720
    },
    {
      "epoch": 0.6239394634258197,
      "grad_norm": 0.14078630733604475,
      "learning_rate": 0.00032724635291520694,
      "loss": 3.4428,
      "step": 2721
    },
    {
      "epoch": 0.624168768631048,
      "grad_norm": 0.13618094676910517,
      "learning_rate": 0.00032689792293094937,
      "loss": 3.5296,
      "step": 2722
    },
    {
      "epoch": 0.6243980738362761,
      "grad_norm": 0.13517641955630533,
      "learning_rate": 0.00032654958842863967,
      "loss": 3.3152,
      "step": 2723
    },
    {
      "epoch": 0.6246273790415042,
      "grad_norm": 0.16113928876444833,
      "learning_rate": 0.0003262013496004167,
      "loss": 3.5639,
      "step": 2724
    },
    {
      "epoch": 0.6248566842467324,
      "grad_norm": 0.13909481418698608,
      "learning_rate": 0.0003258532066383667,
      "loss": 3.4172,
      "step": 2725
    },
    {
      "epoch": 0.6250859894519606,
      "grad_norm": 0.15321893502152878,
      "learning_rate": 0.00032550515973452294,
      "loss": 3.6363,
      "step": 2726
    },
    {
      "epoch": 0.6253152946571887,
      "grad_norm": 0.14852891109043553,
      "learning_rate": 0.000325157209080866,
      "loss": 3.4595,
      "step": 2727
    },
    {
      "epoch": 0.6255445998624168,
      "grad_norm": 0.1458746477894711,
      "learning_rate": 0.00032480935486932263,
      "loss": 3.4285,
      "step": 2728
    },
    {
      "epoch": 0.6257739050676451,
      "grad_norm": 0.13776324146870836,
      "learning_rate": 0.00032446159729176743,
      "loss": 3.4156,
      "step": 2729
    },
    {
      "epoch": 0.6260032102728732,
      "grad_norm": 0.1460552066090348,
      "learning_rate": 0.0003241139365400207,
      "loss": 3.4462,
      "step": 2730
    },
    {
      "epoch": 0.6262325154781013,
      "grad_norm": 0.14778217585841213,
      "learning_rate": 0.0003237663728058502,
      "loss": 3.7168,
      "step": 2731
    },
    {
      "epoch": 0.6264618206833296,
      "grad_norm": 0.14880857329565597,
      "learning_rate": 0.0003234189062809695,
      "loss": 3.6249,
      "step": 2732
    },
    {
      "epoch": 0.6266911258885577,
      "grad_norm": 0.1601003728300445,
      "learning_rate": 0.0003230715371570389,
      "loss": 3.4397,
      "step": 2733
    },
    {
      "epoch": 0.6269204310937858,
      "grad_norm": 0.14763277183624451,
      "learning_rate": 0.00032272426562566484,
      "loss": 3.5333,
      "step": 2734
    },
    {
      "epoch": 0.6271497362990139,
      "grad_norm": 0.15831814122742838,
      "learning_rate": 0.00032237709187839996,
      "loss": 3.5664,
      "step": 2735
    },
    {
      "epoch": 0.6273790415042422,
      "grad_norm": 0.1486933761494612,
      "learning_rate": 0.0003220300161067432,
      "loss": 3.4842,
      "step": 2736
    },
    {
      "epoch": 0.6276083467094703,
      "grad_norm": 0.14203828583831324,
      "learning_rate": 0.0003216830385021388,
      "loss": 3.4834,
      "step": 2737
    },
    {
      "epoch": 0.6278376519146984,
      "grad_norm": 0.14331960006973304,
      "learning_rate": 0.0003213361592559777,
      "loss": 3.3845,
      "step": 2738
    },
    {
      "epoch": 0.6280669571199267,
      "grad_norm": 0.14116552029050936,
      "learning_rate": 0.0003209893785595959,
      "loss": 3.5174,
      "step": 2739
    },
    {
      "epoch": 0.6282962623251548,
      "grad_norm": 0.15514860383793302,
      "learning_rate": 0.0003206426966042754,
      "loss": 3.3385,
      "step": 2740
    },
    {
      "epoch": 0.6285255675303829,
      "grad_norm": 0.1496909998141586,
      "learning_rate": 0.00032029611358124366,
      "loss": 3.3943,
      "step": 2741
    },
    {
      "epoch": 0.6287548727356111,
      "grad_norm": 0.1560684180395754,
      "learning_rate": 0.00031994962968167365,
      "loss": 3.3184,
      "step": 2742
    },
    {
      "epoch": 0.6289841779408393,
      "grad_norm": 0.1526124045374692,
      "learning_rate": 0.00031960324509668336,
      "loss": 3.5181,
      "step": 2743
    },
    {
      "epoch": 0.6292134831460674,
      "grad_norm": 0.14724877426162794,
      "learning_rate": 0.00031925696001733637,
      "loss": 3.5734,
      "step": 2744
    },
    {
      "epoch": 0.6294427883512955,
      "grad_norm": 0.13721267774363527,
      "learning_rate": 0.0003189107746346412,
      "loss": 3.5681,
      "step": 2745
    },
    {
      "epoch": 0.6296720935565238,
      "grad_norm": 0.13780420972488036,
      "learning_rate": 0.0003185646891395514,
      "loss": 3.541,
      "step": 2746
    },
    {
      "epoch": 0.6299013987617519,
      "grad_norm": 0.1454628776044396,
      "learning_rate": 0.0003182187037229653,
      "loss": 3.4502,
      "step": 2747
    },
    {
      "epoch": 0.63013070396698,
      "grad_norm": 0.15315740280480922,
      "learning_rate": 0.00031787281857572627,
      "loss": 3.5364,
      "step": 2748
    },
    {
      "epoch": 0.6303600091722082,
      "grad_norm": 0.14037824399093235,
      "learning_rate": 0.0003175270338886221,
      "loss": 3.4238,
      "step": 2749
    },
    {
      "epoch": 0.6305893143774364,
      "grad_norm": 0.14319773111857245,
      "learning_rate": 0.00031718134985238565,
      "loss": 3.3847,
      "step": 2750
    },
    {
      "epoch": 0.6308186195826645,
      "grad_norm": 0.15752421208702003,
      "learning_rate": 0.00031683576665769345,
      "loss": 3.5046,
      "step": 2751
    },
    {
      "epoch": 0.6310479247878927,
      "grad_norm": 0.15585366913939513,
      "learning_rate": 0.0003164902844951673,
      "loss": 3.4174,
      "step": 2752
    },
    {
      "epoch": 0.6312772299931209,
      "grad_norm": 0.1434014890187922,
      "learning_rate": 0.0003161449035553724,
      "loss": 3.4817,
      "step": 2753
    },
    {
      "epoch": 0.631506535198349,
      "grad_norm": 0.1378103783193763,
      "learning_rate": 0.00031579962402881903,
      "loss": 3.4579,
      "step": 2754
    },
    {
      "epoch": 0.6317358404035771,
      "grad_norm": 0.16838125905142717,
      "learning_rate": 0.00031545444610596077,
      "loss": 3.4942,
      "step": 2755
    },
    {
      "epoch": 0.6319651456088053,
      "grad_norm": 0.15796012412150914,
      "learning_rate": 0.0003151093699771955,
      "loss": 3.3157,
      "step": 2756
    },
    {
      "epoch": 0.6321944508140335,
      "grad_norm": 0.1599637207965777,
      "learning_rate": 0.000314764395832865,
      "loss": 3.5029,
      "step": 2757
    },
    {
      "epoch": 0.6324237560192616,
      "grad_norm": 0.15503160631592341,
      "learning_rate": 0.00031441952386325467,
      "loss": 3.3762,
      "step": 2758
    },
    {
      "epoch": 0.6326530612244898,
      "grad_norm": 0.14842478053163435,
      "learning_rate": 0.0003140747542585934,
      "loss": 3.4858,
      "step": 2759
    },
    {
      "epoch": 0.632882366429718,
      "grad_norm": 0.1567260792235641,
      "learning_rate": 0.0003137300872090542,
      "loss": 3.4595,
      "step": 2760
    },
    {
      "epoch": 0.6331116716349461,
      "grad_norm": 0.14181182643523674,
      "learning_rate": 0.00031338552290475266,
      "loss": 3.5213,
      "step": 2761
    },
    {
      "epoch": 0.6333409768401743,
      "grad_norm": 0.1528418715986394,
      "learning_rate": 0.0003130410615357485,
      "loss": 3.4014,
      "step": 2762
    },
    {
      "epoch": 0.6335702820454024,
      "grad_norm": 0.14990901288982295,
      "learning_rate": 0.00031269670329204396,
      "loss": 3.5061,
      "step": 2763
    },
    {
      "epoch": 0.6337995872506306,
      "grad_norm": 0.15701176164122446,
      "learning_rate": 0.00031235244836358525,
      "loss": 3.5657,
      "step": 2764
    },
    {
      "epoch": 0.6340288924558587,
      "grad_norm": 0.14977232936409476,
      "learning_rate": 0.0003120082969402604,
      "loss": 3.4234,
      "step": 2765
    },
    {
      "epoch": 0.6342581976610869,
      "grad_norm": 0.1372048437734487,
      "learning_rate": 0.00031166424921190173,
      "loss": 3.4595,
      "step": 2766
    },
    {
      "epoch": 0.6344875028663151,
      "grad_norm": 0.15404184507914687,
      "learning_rate": 0.00031132030536828314,
      "loss": 3.4976,
      "step": 2767
    },
    {
      "epoch": 0.6347168080715432,
      "grad_norm": 0.14203136879818776,
      "learning_rate": 0.00031097646559912205,
      "loss": 3.4355,
      "step": 2768
    },
    {
      "epoch": 0.6349461132767714,
      "grad_norm": 0.1509036721086356,
      "learning_rate": 0.00031063273009407805,
      "loss": 3.4508,
      "step": 2769
    },
    {
      "epoch": 0.6351754184819995,
      "grad_norm": 0.14720951310975275,
      "learning_rate": 0.0003102890990427535,
      "loss": 3.2708,
      "step": 2770
    },
    {
      "epoch": 0.6354047236872277,
      "grad_norm": 0.15477236277180556,
      "learning_rate": 0.00030994557263469265,
      "loss": 3.4274,
      "step": 2771
    },
    {
      "epoch": 0.6356340288924559,
      "grad_norm": 0.1344014777113198,
      "learning_rate": 0.00030960215105938274,
      "loss": 3.4932,
      "step": 2772
    },
    {
      "epoch": 0.635863334097684,
      "grad_norm": 0.14597300939204041,
      "learning_rate": 0.0003092588345062526,
      "loss": 3.501,
      "step": 2773
    },
    {
      "epoch": 0.6360926393029122,
      "grad_norm": 0.14159381204943247,
      "learning_rate": 0.00030891562316467345,
      "loss": 3.541,
      "step": 2774
    },
    {
      "epoch": 0.6363219445081403,
      "grad_norm": 0.15945041235922067,
      "learning_rate": 0.0003085725172239582,
      "loss": 3.465,
      "step": 2775
    },
    {
      "epoch": 0.6365512497133685,
      "grad_norm": 0.14554036678879773,
      "learning_rate": 0.0003082295168733621,
      "loss": 3.5123,
      "step": 2776
    },
    {
      "epoch": 0.6367805549185966,
      "grad_norm": 0.15571355904245412,
      "learning_rate": 0.0003078866223020815,
      "loss": 3.4656,
      "step": 2777
    },
    {
      "epoch": 0.6370098601238248,
      "grad_norm": 0.14895690081253407,
      "learning_rate": 0.0003075438336992551,
      "loss": 3.4502,
      "step": 2778
    },
    {
      "epoch": 0.637239165329053,
      "grad_norm": 0.15616317547170955,
      "learning_rate": 0.0003072011512539624,
      "loss": 3.5488,
      "step": 2779
    },
    {
      "epoch": 0.6374684705342811,
      "grad_norm": 0.14916139712927778,
      "learning_rate": 0.00030685857515522517,
      "loss": 3.6914,
      "step": 2780
    },
    {
      "epoch": 0.6376977757395093,
      "grad_norm": 0.14813704029869829,
      "learning_rate": 0.00030651610559200574,
      "loss": 3.4595,
      "step": 2781
    },
    {
      "epoch": 0.6379270809447375,
      "grad_norm": 0.13938506298064196,
      "learning_rate": 0.0003061737427532083,
      "loss": 3.4393,
      "step": 2782
    },
    {
      "epoch": 0.6381563861499656,
      "grad_norm": 0.14847192861453137,
      "learning_rate": 0.00030583148682767757,
      "loss": 3.4337,
      "step": 2783
    },
    {
      "epoch": 0.6383856913551937,
      "grad_norm": 0.15990582957744223,
      "learning_rate": 0.0003054893380042001,
      "loss": 3.4128,
      "step": 2784
    },
    {
      "epoch": 0.638614996560422,
      "grad_norm": 0.1396417755668301,
      "learning_rate": 0.00030514729647150243,
      "loss": 3.4518,
      "step": 2785
    },
    {
      "epoch": 0.6388443017656501,
      "grad_norm": 0.14390225663527978,
      "learning_rate": 0.00030480536241825264,
      "loss": 3.4108,
      "step": 2786
    },
    {
      "epoch": 0.6390736069708782,
      "grad_norm": 0.1428192762129616,
      "learning_rate": 0.0003044635360330592,
      "loss": 3.5523,
      "step": 2787
    },
    {
      "epoch": 0.6393029121761064,
      "grad_norm": 0.15895435312498615,
      "learning_rate": 0.0003041218175044713,
      "loss": 3.4004,
      "step": 2788
    },
    {
      "epoch": 0.6395322173813346,
      "grad_norm": 0.15224352190331278,
      "learning_rate": 0.00030378020702097845,
      "loss": 3.437,
      "step": 2789
    },
    {
      "epoch": 0.6397615225865627,
      "grad_norm": 0.14606845351646888,
      "learning_rate": 0.0003034387047710111,
      "loss": 3.558,
      "step": 2790
    },
    {
      "epoch": 0.6399908277917908,
      "grad_norm": 0.16707035311228455,
      "learning_rate": 0.000303097310942939,
      "loss": 3.6732,
      "step": 2791
    },
    {
      "epoch": 0.6402201329970191,
      "grad_norm": 0.15483546395419306,
      "learning_rate": 0.0003027560257250733,
      "loss": 3.3909,
      "step": 2792
    },
    {
      "epoch": 0.6404494382022472,
      "grad_norm": 0.14875662042392826,
      "learning_rate": 0.0003024148493056641,
      "loss": 3.455,
      "step": 2793
    },
    {
      "epoch": 0.6406787434074753,
      "grad_norm": 0.14415425974690046,
      "learning_rate": 0.00030207378187290245,
      "loss": 3.5272,
      "step": 2794
    },
    {
      "epoch": 0.6409080486127036,
      "grad_norm": 0.13945616608432235,
      "learning_rate": 0.00030173282361491865,
      "loss": 3.498,
      "step": 2795
    },
    {
      "epoch": 0.6411373538179317,
      "grad_norm": 0.14193951882246728,
      "learning_rate": 0.0003013919747197832,
      "loss": 3.4311,
      "step": 2796
    },
    {
      "epoch": 0.6413666590231598,
      "grad_norm": 0.15133074287367732,
      "learning_rate": 0.0003010512353755057,
      "loss": 3.3486,
      "step": 2797
    },
    {
      "epoch": 0.6415959642283879,
      "grad_norm": 0.14438566050596446,
      "learning_rate": 0.00030071060577003606,
      "loss": 3.5392,
      "step": 2798
    },
    {
      "epoch": 0.6418252694336162,
      "grad_norm": 0.1514325223546602,
      "learning_rate": 0.00030037008609126313,
      "loss": 3.538,
      "step": 2799
    },
    {
      "epoch": 0.6420545746388443,
      "grad_norm": 0.14724930747265855,
      "learning_rate": 0.00030002967652701534,
      "loss": 3.433,
      "step": 2800
    },
    {
      "epoch": 0.6422838798440724,
      "grad_norm": 0.14454677900105045,
      "learning_rate": 0.0002996893772650602,
      "loss": 3.4105,
      "step": 2801
    },
    {
      "epoch": 0.6425131850493007,
      "grad_norm": 0.15506613434708008,
      "learning_rate": 0.0002993491884931048,
      "loss": 3.4773,
      "step": 2802
    },
    {
      "epoch": 0.6427424902545288,
      "grad_norm": 0.14840913142772083,
      "learning_rate": 0.0002990091103987945,
      "loss": 3.5586,
      "step": 2803
    },
    {
      "epoch": 0.6429717954597569,
      "grad_norm": 0.16755182944130656,
      "learning_rate": 0.0002986691431697148,
      "loss": 3.4936,
      "step": 2804
    },
    {
      "epoch": 0.6432011006649851,
      "grad_norm": 0.1603779794499398,
      "learning_rate": 0.0002983292869933886,
      "loss": 3.3524,
      "step": 2805
    },
    {
      "epoch": 0.6434304058702133,
      "grad_norm": 0.13843960715090656,
      "learning_rate": 0.00029798954205727886,
      "loss": 3.5024,
      "step": 2806
    },
    {
      "epoch": 0.6436597110754414,
      "grad_norm": 0.14112978611979812,
      "learning_rate": 0.0002976499085487862,
      "loss": 3.4975,
      "step": 2807
    },
    {
      "epoch": 0.6438890162806695,
      "grad_norm": 0.14764562594926478,
      "learning_rate": 0.0002973103866552506,
      "loss": 3.3653,
      "step": 2808
    },
    {
      "epoch": 0.6441183214858978,
      "grad_norm": 0.16267429917284473,
      "learning_rate": 0.00029697097656394963,
      "loss": 3.5403,
      "step": 2809
    },
    {
      "epoch": 0.6443476266911259,
      "grad_norm": 0.15335563546340145,
      "learning_rate": 0.0002966316784621,
      "loss": 3.2816,
      "step": 2810
    },
    {
      "epoch": 0.644576931896354,
      "grad_norm": 0.14675414404760406,
      "learning_rate": 0.00029629249253685595,
      "loss": 3.5171,
      "step": 2811
    },
    {
      "epoch": 0.6448062371015822,
      "grad_norm": 0.16097566370720925,
      "learning_rate": 0.0002959534189753105,
      "loss": 3.5911,
      "step": 2812
    },
    {
      "epoch": 0.6450355423068104,
      "grad_norm": 0.1555022516319028,
      "learning_rate": 0.00029561445796449416,
      "loss": 3.4626,
      "step": 2813
    },
    {
      "epoch": 0.6452648475120385,
      "grad_norm": 0.16620835969051567,
      "learning_rate": 0.0002952756096913756,
      "loss": 3.5369,
      "step": 2814
    },
    {
      "epoch": 0.6454941527172667,
      "grad_norm": 0.14134213984258123,
      "learning_rate": 0.0002949368743428612,
      "loss": 3.3984,
      "step": 2815
    },
    {
      "epoch": 0.6457234579224949,
      "grad_norm": 0.14120613802690082,
      "learning_rate": 0.0002945982521057953,
      "loss": 3.4143,
      "step": 2816
    },
    {
      "epoch": 0.645952763127723,
      "grad_norm": 0.1302794815749293,
      "learning_rate": 0.0002942597431669593,
      "loss": 3.3727,
      "step": 2817
    },
    {
      "epoch": 0.6461820683329511,
      "grad_norm": 0.15859630236654276,
      "learning_rate": 0.000293921347713073,
      "loss": 3.5719,
      "step": 2818
    },
    {
      "epoch": 0.6464113735381793,
      "grad_norm": 0.1483606665198632,
      "learning_rate": 0.0002935830659307924,
      "loss": 3.4735,
      "step": 2819
    },
    {
      "epoch": 0.6466406787434075,
      "grad_norm": 0.16155125356551214,
      "learning_rate": 0.0002932448980067122,
      "loss": 3.342,
      "step": 2820
    },
    {
      "epoch": 0.6468699839486356,
      "grad_norm": 0.149004558864426,
      "learning_rate": 0.0002929068441273629,
      "loss": 3.3842,
      "step": 2821
    },
    {
      "epoch": 0.6470992891538638,
      "grad_norm": 0.15361267878762416,
      "learning_rate": 0.0002925689044792132,
      "loss": 3.458,
      "step": 2822
    },
    {
      "epoch": 0.647328594359092,
      "grad_norm": 0.15291685004575226,
      "learning_rate": 0.0002922310792486681,
      "loss": 3.5503,
      "step": 2823
    },
    {
      "epoch": 0.6475578995643201,
      "grad_norm": 0.16481076872402495,
      "learning_rate": 0.00029189336862207003,
      "loss": 3.5381,
      "step": 2824
    },
    {
      "epoch": 0.6477872047695483,
      "grad_norm": 0.1620243942276125,
      "learning_rate": 0.00029155577278569745,
      "loss": 3.4247,
      "step": 2825
    },
    {
      "epoch": 0.6480165099747764,
      "grad_norm": 0.1677856592318214,
      "learning_rate": 0.0002912182919257664,
      "loss": 3.5159,
      "step": 2826
    },
    {
      "epoch": 0.6482458151800046,
      "grad_norm": 0.15235567708823747,
      "learning_rate": 0.00029088092622842895,
      "loss": 3.5404,
      "step": 2827
    },
    {
      "epoch": 0.6484751203852327,
      "grad_norm": 0.1616837160449886,
      "learning_rate": 0.0002905436758797737,
      "loss": 3.5763,
      "step": 2828
    },
    {
      "epoch": 0.6487044255904609,
      "grad_norm": 0.15274425844206194,
      "learning_rate": 0.00029020654106582544,
      "loss": 3.3975,
      "step": 2829
    },
    {
      "epoch": 0.6489337307956891,
      "grad_norm": 0.1468054869801275,
      "learning_rate": 0.000289869521972546,
      "loss": 3.5729,
      "step": 2830
    },
    {
      "epoch": 0.6491630360009172,
      "grad_norm": 0.16249676705823088,
      "learning_rate": 0.0002895326187858326,
      "loss": 3.3435,
      "step": 2831
    },
    {
      "epoch": 0.6493923412061454,
      "grad_norm": 0.1457135591425428,
      "learning_rate": 0.0002891958316915187,
      "loss": 3.4144,
      "step": 2832
    },
    {
      "epoch": 0.6496216464113735,
      "grad_norm": 0.16801231951474405,
      "learning_rate": 0.00028885916087537377,
      "loss": 3.5423,
      "step": 2833
    },
    {
      "epoch": 0.6498509516166017,
      "grad_norm": 0.1576914292330519,
      "learning_rate": 0.0002885226065231035,
      "loss": 3.4495,
      "step": 2834
    },
    {
      "epoch": 0.6500802568218299,
      "grad_norm": 0.14845774806573242,
      "learning_rate": 0.00028818616882034877,
      "loss": 3.3793,
      "step": 2835
    },
    {
      "epoch": 0.650309562027058,
      "grad_norm": 0.16119909530962215,
      "learning_rate": 0.00028784984795268643,
      "loss": 3.4879,
      "step": 2836
    },
    {
      "epoch": 0.6505388672322862,
      "grad_norm": 0.15016767364236921,
      "learning_rate": 0.0002875136441056286,
      "loss": 3.3577,
      "step": 2837
    },
    {
      "epoch": 0.6507681724375143,
      "grad_norm": 0.15364008712127203,
      "learning_rate": 0.00028717755746462373,
      "loss": 3.4245,
      "step": 2838
    },
    {
      "epoch": 0.6509974776427425,
      "grad_norm": 0.169840818631348,
      "learning_rate": 0.000286841588215054,
      "loss": 3.5329,
      "step": 2839
    },
    {
      "epoch": 0.6512267828479706,
      "grad_norm": 0.16673345234312767,
      "learning_rate": 0.0002865057365422386,
      "loss": 3.4643,
      "step": 2840
    },
    {
      "epoch": 0.6514560880531988,
      "grad_norm": 0.1573361253387342,
      "learning_rate": 0.0002861700026314308,
      "loss": 3.5673,
      "step": 2841
    },
    {
      "epoch": 0.651685393258427,
      "grad_norm": 0.16652423751779494,
      "learning_rate": 0.0002858343866678191,
      "loss": 3.6252,
      "step": 2842
    },
    {
      "epoch": 0.6519146984636551,
      "grad_norm": 0.15262347060486264,
      "learning_rate": 0.00028549888883652686,
      "loss": 3.4453,
      "step": 2843
    },
    {
      "epoch": 0.6521440036688833,
      "grad_norm": 0.14417345409393736,
      "learning_rate": 0.0002851635093226129,
      "loss": 3.4988,
      "step": 2844
    },
    {
      "epoch": 0.6523733088741115,
      "grad_norm": 0.17049192176163797,
      "learning_rate": 0.00028482824831107,
      "loss": 3.6553,
      "step": 2845
    },
    {
      "epoch": 0.6526026140793396,
      "grad_norm": 0.1453371892560079,
      "learning_rate": 0.0002844931059868261,
      "loss": 3.5812,
      "step": 2846
    },
    {
      "epoch": 0.6528319192845677,
      "grad_norm": 0.14756904971470175,
      "learning_rate": 0.000284158082534743,
      "loss": 3.4753,
      "step": 2847
    },
    {
      "epoch": 0.6530612244897959,
      "grad_norm": 0.13840810033650047,
      "learning_rate": 0.000283823178139618,
      "loss": 3.4029,
      "step": 2848
    },
    {
      "epoch": 0.6532905296950241,
      "grad_norm": 0.15982166060369757,
      "learning_rate": 0.00028348839298618177,
      "loss": 3.6494,
      "step": 2849
    },
    {
      "epoch": 0.6535198349002522,
      "grad_norm": 0.15119445140333684,
      "learning_rate": 0.0002831537272590997,
      "loss": 3.5489,
      "step": 2850
    },
    {
      "epoch": 0.6537491401054804,
      "grad_norm": 0.13990920965650186,
      "learning_rate": 0.0002828191811429709,
      "loss": 3.5371,
      "step": 2851
    },
    {
      "epoch": 0.6539784453107086,
      "grad_norm": 0.15611846437706636,
      "learning_rate": 0.00028248475482232937,
      "loss": 3.5829,
      "step": 2852
    },
    {
      "epoch": 0.6542077505159367,
      "grad_norm": 0.1481119945065491,
      "learning_rate": 0.00028215044848164164,
      "loss": 3.4278,
      "step": 2853
    },
    {
      "epoch": 0.6544370557211648,
      "grad_norm": 0.16283416460771108,
      "learning_rate": 0.0002818162623053094,
      "loss": 3.403,
      "step": 2854
    },
    {
      "epoch": 0.6546663609263931,
      "grad_norm": 0.15272711617045104,
      "learning_rate": 0.00028148219647766747,
      "loss": 3.374,
      "step": 2855
    },
    {
      "epoch": 0.6548956661316212,
      "grad_norm": 0.14590619193997825,
      "learning_rate": 0.00028114825118298413,
      "loss": 3.5268,
      "step": 2856
    },
    {
      "epoch": 0.6551249713368493,
      "grad_norm": 0.1445197749410317,
      "learning_rate": 0.00028081442660546124,
      "loss": 3.4251,
      "step": 2857
    },
    {
      "epoch": 0.6553542765420775,
      "grad_norm": 0.16125293479419409,
      "learning_rate": 0.00028048072292923465,
      "loss": 3.4626,
      "step": 2858
    },
    {
      "epoch": 0.6555835817473057,
      "grad_norm": 0.15124525419920107,
      "learning_rate": 0.0002801471403383728,
      "loss": 3.4148,
      "step": 2859
    },
    {
      "epoch": 0.6558128869525338,
      "grad_norm": 0.14752327749012284,
      "learning_rate": 0.00027981367901687773,
      "loss": 3.3256,
      "step": 2860
    },
    {
      "epoch": 0.6560421921577619,
      "grad_norm": 0.14998417949361745,
      "learning_rate": 0.00027948033914868415,
      "loss": 3.4521,
      "step": 2861
    },
    {
      "epoch": 0.6562714973629902,
      "grad_norm": 0.14673471546674346,
      "learning_rate": 0.0002791471209176606,
      "loss": 3.3632,
      "step": 2862
    },
    {
      "epoch": 0.6565008025682183,
      "grad_norm": 0.1485358328574425,
      "learning_rate": 0.00027881402450760775,
      "loss": 3.3354,
      "step": 2863
    },
    {
      "epoch": 0.6567301077734464,
      "grad_norm": 0.15709335967502205,
      "learning_rate": 0.00027848105010225933,
      "loss": 3.5265,
      "step": 2864
    },
    {
      "epoch": 0.6569594129786747,
      "grad_norm": 0.14792567559137673,
      "learning_rate": 0.00027814819788528165,
      "loss": 3.4615,
      "step": 2865
    },
    {
      "epoch": 0.6571887181839028,
      "grad_norm": 0.18292322297211525,
      "learning_rate": 0.00027781546804027445,
      "loss": 3.5367,
      "step": 2866
    },
    {
      "epoch": 0.6574180233891309,
      "grad_norm": 0.15316970647205894,
      "learning_rate": 0.00027748286075076836,
      "loss": 3.4169,
      "step": 2867
    },
    {
      "epoch": 0.657647328594359,
      "grad_norm": 0.14731041301588205,
      "learning_rate": 0.00027715037620022797,
      "loss": 3.5127,
      "step": 2868
    },
    {
      "epoch": 0.6578766337995873,
      "grad_norm": 0.14288403976266928,
      "learning_rate": 0.00027681801457204937,
      "loss": 3.4086,
      "step": 2869
    },
    {
      "epoch": 0.6581059390048154,
      "grad_norm": 0.14021167362296771,
      "learning_rate": 0.0002764857760495609,
      "loss": 3.4201,
      "step": 2870
    },
    {
      "epoch": 0.6583352442100435,
      "grad_norm": 0.14803538614238357,
      "learning_rate": 0.00027615366081602306,
      "loss": 3.4557,
      "step": 2871
    },
    {
      "epoch": 0.6585645494152718,
      "grad_norm": 0.1429528537063265,
      "learning_rate": 0.0002758216690546287,
      "loss": 3.4881,
      "step": 2872
    },
    {
      "epoch": 0.6587938546204999,
      "grad_norm": 0.16053698462283944,
      "learning_rate": 0.0002754898009485021,
      "loss": 3.5216,
      "step": 2873
    },
    {
      "epoch": 0.659023159825728,
      "grad_norm": 0.14559274622616758,
      "learning_rate": 0.00027515805668069937,
      "loss": 3.4924,
      "step": 2874
    },
    {
      "epoch": 0.6592524650309562,
      "grad_norm": 0.1539756234607275,
      "learning_rate": 0.0002748264364342085,
      "loss": 3.5007,
      "step": 2875
    },
    {
      "epoch": 0.6594817702361844,
      "grad_norm": 0.14433367478309575,
      "learning_rate": 0.000274494940391949,
      "loss": 3.3404,
      "step": 2876
    },
    {
      "epoch": 0.6597110754414125,
      "grad_norm": 0.14491756539264355,
      "learning_rate": 0.00027416356873677204,
      "loss": 3.5579,
      "step": 2877
    },
    {
      "epoch": 0.6599403806466407,
      "grad_norm": 0.1674629920895873,
      "learning_rate": 0.00027383232165145976,
      "loss": 3.5208,
      "step": 2878
    },
    {
      "epoch": 0.6601696858518689,
      "grad_norm": 0.1592286079112558,
      "learning_rate": 0.0002735011993187258,
      "loss": 3.4296,
      "step": 2879
    },
    {
      "epoch": 0.660398991057097,
      "grad_norm": 0.16504492674315496,
      "learning_rate": 0.0002731702019212156,
      "loss": 3.6206,
      "step": 2880
    },
    {
      "epoch": 0.6606282962623251,
      "grad_norm": 0.1450168690223817,
      "learning_rate": 0.0002728393296415042,
      "loss": 3.493,
      "step": 2881
    },
    {
      "epoch": 0.6608576014675533,
      "grad_norm": 0.15080664217185732,
      "learning_rate": 0.00027250858266209915,
      "loss": 3.3945,
      "step": 2882
    },
    {
      "epoch": 0.6610869066727815,
      "grad_norm": 0.15687292655689816,
      "learning_rate": 0.00027217796116543817,
      "loss": 3.397,
      "step": 2883
    },
    {
      "epoch": 0.6613162118780096,
      "grad_norm": 0.15125974822983185,
      "learning_rate": 0.0002718474653338899,
      "loss": 3.4686,
      "step": 2884
    },
    {
      "epoch": 0.6615455170832378,
      "grad_norm": 0.14191416190920314,
      "learning_rate": 0.0002715170953497532,
      "loss": 3.4685,
      "step": 2885
    },
    {
      "epoch": 0.661774822288466,
      "grad_norm": 0.15116978650502405,
      "learning_rate": 0.0002711868513952587,
      "loss": 3.5293,
      "step": 2886
    },
    {
      "epoch": 0.6620041274936941,
      "grad_norm": 0.15475077887709915,
      "learning_rate": 0.00027085673365256614,
      "loss": 3.442,
      "step": 2887
    },
    {
      "epoch": 0.6622334326989223,
      "grad_norm": 0.1547812370992383,
      "learning_rate": 0.0002705267423037666,
      "loss": 3.5613,
      "step": 2888
    },
    {
      "epoch": 0.6624627379041504,
      "grad_norm": 0.1506734046630414,
      "learning_rate": 0.00027019687753088075,
      "loss": 3.5277,
      "step": 2889
    },
    {
      "epoch": 0.6626920431093786,
      "grad_norm": 0.16794027607547032,
      "learning_rate": 0.0002698671395158603,
      "loss": 3.421,
      "step": 2890
    },
    {
      "epoch": 0.6629213483146067,
      "grad_norm": 0.16178545673061623,
      "learning_rate": 0.00026953752844058597,
      "loss": 3.5361,
      "step": 2891
    },
    {
      "epoch": 0.6631506535198349,
      "grad_norm": 0.1375257391516164,
      "learning_rate": 0.0002692080444868695,
      "loss": 3.378,
      "step": 2892
    },
    {
      "epoch": 0.6633799587250631,
      "grad_norm": 0.15575957476572233,
      "learning_rate": 0.0002688786878364516,
      "loss": 3.4601,
      "step": 2893
    },
    {
      "epoch": 0.6636092639302912,
      "grad_norm": 0.1560631082845894,
      "learning_rate": 0.00026854945867100377,
      "loss": 3.4728,
      "step": 2894
    },
    {
      "epoch": 0.6638385691355194,
      "grad_norm": 0.14991967938484801,
      "learning_rate": 0.00026822035717212597,
      "loss": 3.649,
      "step": 2895
    },
    {
      "epoch": 0.6640678743407475,
      "grad_norm": 0.16480784690725883,
      "learning_rate": 0.0002678913835213488,
      "loss": 3.361,
      "step": 2896
    },
    {
      "epoch": 0.6642971795459757,
      "grad_norm": 0.1444044618285611,
      "learning_rate": 0.00026756253790013193,
      "loss": 3.4764,
      "step": 2897
    },
    {
      "epoch": 0.6645264847512039,
      "grad_norm": 0.15263559164618815,
      "learning_rate": 0.0002672338204898642,
      "loss": 3.5521,
      "step": 2898
    },
    {
      "epoch": 0.664755789956432,
      "grad_norm": 0.157035351051557,
      "learning_rate": 0.0002669052314718641,
      "loss": 3.3967,
      "step": 2899
    },
    {
      "epoch": 0.6649850951616602,
      "grad_norm": 0.14066727693080613,
      "learning_rate": 0.00026657677102737875,
      "loss": 3.4703,
      "step": 2900
    },
    {
      "epoch": 0.6652144003668883,
      "grad_norm": 0.15018392129917071,
      "learning_rate": 0.0002662484393375855,
      "loss": 3.5421,
      "step": 2901
    },
    {
      "epoch": 0.6654437055721165,
      "grad_norm": 0.1409739314404724,
      "learning_rate": 0.00026592023658358904,
      "loss": 3.506,
      "step": 2902
    },
    {
      "epoch": 0.6656730107773446,
      "grad_norm": 0.14744664684850253,
      "learning_rate": 0.00026559216294642446,
      "loss": 3.5419,
      "step": 2903
    },
    {
      "epoch": 0.6659023159825728,
      "grad_norm": 0.15460343780401484,
      "learning_rate": 0.00026526421860705474,
      "loss": 3.5154,
      "step": 2904
    },
    {
      "epoch": 0.666131621187801,
      "grad_norm": 0.1538295526266509,
      "learning_rate": 0.0002649364037463718,
      "loss": 3.5127,
      "step": 2905
    },
    {
      "epoch": 0.6663609263930291,
      "grad_norm": 0.15820147544829222,
      "learning_rate": 0.0002646087185451959,
      "loss": 3.3907,
      "step": 2906
    },
    {
      "epoch": 0.6665902315982573,
      "grad_norm": 0.15588107952256317,
      "learning_rate": 0.0002642811631842764,
      "loss": 3.3979,
      "step": 2907
    },
    {
      "epoch": 0.6668195368034855,
      "grad_norm": 0.1477221805868086,
      "learning_rate": 0.00026395373784429053,
      "loss": 3.5112,
      "step": 2908
    },
    {
      "epoch": 0.6670488420087136,
      "grad_norm": 0.15201478999912435,
      "learning_rate": 0.0002636264427058439,
      "loss": 3.5005,
      "step": 2909
    },
    {
      "epoch": 0.6672781472139417,
      "grad_norm": 0.1563848816070769,
      "learning_rate": 0.00026329927794947,
      "loss": 3.4243,
      "step": 2910
    },
    {
      "epoch": 0.6675074524191699,
      "grad_norm": 0.15287278501796037,
      "learning_rate": 0.00026297224375563123,
      "loss": 3.3896,
      "step": 2911
    },
    {
      "epoch": 0.6677367576243981,
      "grad_norm": 0.1606792125185515,
      "learning_rate": 0.00026264534030471723,
      "loss": 3.5993,
      "step": 2912
    },
    {
      "epoch": 0.6679660628296262,
      "grad_norm": 0.14955742152696422,
      "learning_rate": 0.00026231856777704575,
      "loss": 3.4118,
      "step": 2913
    },
    {
      "epoch": 0.6681953680348544,
      "grad_norm": 0.16047754337273984,
      "learning_rate": 0.0002619919263528622,
      "loss": 3.5524,
      "step": 2914
    },
    {
      "epoch": 0.6684246732400826,
      "grad_norm": 0.16406647108438518,
      "learning_rate": 0.00026166541621234026,
      "loss": 3.4213,
      "step": 2915
    },
    {
      "epoch": 0.6686539784453107,
      "grad_norm": 0.15669916458093142,
      "learning_rate": 0.0002613390375355801,
      "loss": 3.4385,
      "step": 2916
    },
    {
      "epoch": 0.6688832836505388,
      "grad_norm": 0.1515562151476654,
      "learning_rate": 0.00026101279050261045,
      "loss": 3.442,
      "step": 2917
    },
    {
      "epoch": 0.6691125888557671,
      "grad_norm": 0.16659284864374935,
      "learning_rate": 0.0002606866752933869,
      "loss": 3.6398,
      "step": 2918
    },
    {
      "epoch": 0.6693418940609952,
      "grad_norm": 0.18371283125148746,
      "learning_rate": 0.00026036069208779247,
      "loss": 3.5424,
      "step": 2919
    },
    {
      "epoch": 0.6695711992662233,
      "grad_norm": 0.1563832484287315,
      "learning_rate": 0.00026003484106563703,
      "loss": 3.4204,
      "step": 2920
    },
    {
      "epoch": 0.6698005044714515,
      "grad_norm": 0.15828443175357038,
      "learning_rate": 0.0002597091224066581,
      "loss": 3.3325,
      "step": 2921
    },
    {
      "epoch": 0.6700298096766797,
      "grad_norm": 0.15224513823107338,
      "learning_rate": 0.00025938353629052004,
      "loss": 3.4306,
      "step": 2922
    },
    {
      "epoch": 0.6702591148819078,
      "grad_norm": 0.1473855470356984,
      "learning_rate": 0.00025905808289681365,
      "loss": 3.4192,
      "step": 2923
    },
    {
      "epoch": 0.6704884200871359,
      "grad_norm": 0.17838958181875847,
      "learning_rate": 0.00025873276240505685,
      "loss": 3.4577,
      "step": 2924
    },
    {
      "epoch": 0.6707177252923642,
      "grad_norm": 0.14725447678178843,
      "learning_rate": 0.0002584075749946946,
      "loss": 3.3925,
      "step": 2925
    },
    {
      "epoch": 0.6709470304975923,
      "grad_norm": 0.15645952416118802,
      "learning_rate": 0.0002580825208450978,
      "loss": 3.4594,
      "step": 2926
    },
    {
      "epoch": 0.6711763357028204,
      "grad_norm": 0.156612464610885,
      "learning_rate": 0.00025775760013556424,
      "loss": 3.6133,
      "step": 2927
    },
    {
      "epoch": 0.6714056409080487,
      "grad_norm": 0.15271999388155144,
      "learning_rate": 0.00025743281304531785,
      "loss": 3.378,
      "step": 2928
    },
    {
      "epoch": 0.6716349461132768,
      "grad_norm": 0.14910468483636405,
      "learning_rate": 0.0002571081597535095,
      "loss": 3.4037,
      "step": 2929
    },
    {
      "epoch": 0.6718642513185049,
      "grad_norm": 0.13954324924860936,
      "learning_rate": 0.000256783640439215,
      "loss": 3.4418,
      "step": 2930
    },
    {
      "epoch": 0.672093556523733,
      "grad_norm": 0.16769491592649224,
      "learning_rate": 0.00025645925528143776,
      "loss": 3.5232,
      "step": 2931
    },
    {
      "epoch": 0.6723228617289613,
      "grad_norm": 0.1551675474530191,
      "learning_rate": 0.0002561350044591061,
      "loss": 3.3184,
      "step": 2932
    },
    {
      "epoch": 0.6725521669341894,
      "grad_norm": 0.1377198858682951,
      "learning_rate": 0.0002558108881510747,
      "loss": 3.3601,
      "step": 2933
    },
    {
      "epoch": 0.6727814721394175,
      "grad_norm": 0.1568297227353549,
      "learning_rate": 0.0002554869065361237,
      "loss": 3.5025,
      "step": 2934
    },
    {
      "epoch": 0.6730107773446458,
      "grad_norm": 0.14854480155470592,
      "learning_rate": 0.00025516305979295963,
      "loss": 3.4797,
      "step": 2935
    },
    {
      "epoch": 0.6732400825498739,
      "grad_norm": 0.15572161581892688,
      "learning_rate": 0.000254839348100214,
      "loss": 3.3945,
      "step": 2936
    },
    {
      "epoch": 0.673469387755102,
      "grad_norm": 0.1692319311623885,
      "learning_rate": 0.0002545157716364439,
      "loss": 3.5144,
      "step": 2937
    },
    {
      "epoch": 0.6736986929603302,
      "grad_norm": 0.1587590035383548,
      "learning_rate": 0.00025419233058013193,
      "loss": 3.4594,
      "step": 2938
    },
    {
      "epoch": 0.6739279981655584,
      "grad_norm": 0.15197792120774914,
      "learning_rate": 0.00025386902510968624,
      "loss": 3.4583,
      "step": 2939
    },
    {
      "epoch": 0.6741573033707865,
      "grad_norm": 0.14163775229174497,
      "learning_rate": 0.00025354585540343985,
      "loss": 3.3166,
      "step": 2940
    },
    {
      "epoch": 0.6743866085760146,
      "grad_norm": 0.1412541328952194,
      "learning_rate": 0.00025322282163965095,
      "loss": 3.3955,
      "step": 2941
    },
    {
      "epoch": 0.6746159137812429,
      "grad_norm": 0.15388100169140084,
      "learning_rate": 0.0002528999239965027,
      "loss": 3.4198,
      "step": 2942
    },
    {
      "epoch": 0.674845218986471,
      "grad_norm": 0.15652420157798005,
      "learning_rate": 0.00025257716265210384,
      "loss": 3.5495,
      "step": 2943
    },
    {
      "epoch": 0.6750745241916991,
      "grad_norm": 0.15046006807167156,
      "learning_rate": 0.00025225453778448647,
      "loss": 3.5316,
      "step": 2944
    },
    {
      "epoch": 0.6753038293969273,
      "grad_norm": 0.17052464391516667,
      "learning_rate": 0.0002519320495716091,
      "loss": 3.4831,
      "step": 2945
    },
    {
      "epoch": 0.6755331346021555,
      "grad_norm": 0.15778607877736148,
      "learning_rate": 0.00025160969819135366,
      "loss": 3.4537,
      "step": 2946
    },
    {
      "epoch": 0.6757624398073836,
      "grad_norm": 0.1425653565524009,
      "learning_rate": 0.00025128748382152716,
      "loss": 3.4139,
      "step": 2947
    },
    {
      "epoch": 0.6759917450126118,
      "grad_norm": 0.1528809311495654,
      "learning_rate": 0.0002509654066398607,
      "loss": 3.4544,
      "step": 2948
    },
    {
      "epoch": 0.67622105021784,
      "grad_norm": 0.15876918411894203,
      "learning_rate": 0.00025064346682401016,
      "loss": 3.5349,
      "step": 2949
    },
    {
      "epoch": 0.6764503554230681,
      "grad_norm": 0.15072099095134203,
      "learning_rate": 0.0002503216645515553,
      "loss": 3.4895,
      "step": 2950
    },
    {
      "epoch": 0.6766796606282962,
      "grad_norm": 0.15480964368880282,
      "learning_rate": 0.0002500000000000001,
      "loss": 3.3326,
      "step": 2951
    },
    {
      "epoch": 0.6769089658335244,
      "grad_norm": 0.1626273167661917,
      "learning_rate": 0.00024967847334677235,
      "loss": 3.464,
      "step": 2952
    },
    {
      "epoch": 0.6771382710387526,
      "grad_norm": 0.16041586257100593,
      "learning_rate": 0.0002493570847692246,
      "loss": 3.5622,
      "step": 2953
    },
    {
      "epoch": 0.6773675762439807,
      "grad_norm": 0.16118561855677993,
      "learning_rate": 0.00024903583444463187,
      "loss": 3.5198,
      "step": 2954
    },
    {
      "epoch": 0.6775968814492089,
      "grad_norm": 0.15331303889123649,
      "learning_rate": 0.00024871472255019424,
      "loss": 3.332,
      "step": 2955
    },
    {
      "epoch": 0.6778261866544371,
      "grad_norm": 0.15945101933133488,
      "learning_rate": 0.0002483937492630345,
      "loss": 3.4732,
      "step": 2956
    },
    {
      "epoch": 0.6780554918596652,
      "grad_norm": 0.15799064800861032,
      "learning_rate": 0.00024807291476019994,
      "loss": 3.5237,
      "step": 2957
    },
    {
      "epoch": 0.6782847970648934,
      "grad_norm": 0.16622870603958523,
      "learning_rate": 0.00024775221921866005,
      "loss": 3.4788,
      "step": 2958
    },
    {
      "epoch": 0.6785141022701215,
      "grad_norm": 0.1516760114183391,
      "learning_rate": 0.00024743166281530877,
      "loss": 3.4017,
      "step": 2959
    },
    {
      "epoch": 0.6787434074753497,
      "grad_norm": 0.14816595669662408,
      "learning_rate": 0.0002471112457269628,
      "loss": 3.2444,
      "step": 2960
    },
    {
      "epoch": 0.6789727126805778,
      "grad_norm": 0.15388109676454262,
      "learning_rate": 0.000246790968130362,
      "loss": 3.5131,
      "step": 2961
    },
    {
      "epoch": 0.679202017885806,
      "grad_norm": 0.17359747043541535,
      "learning_rate": 0.00024647083020216915,
      "loss": 3.4365,
      "step": 2962
    },
    {
      "epoch": 0.6794313230910342,
      "grad_norm": 0.1562243036392403,
      "learning_rate": 0.0002461508321189706,
      "loss": 3.3336,
      "step": 2963
    },
    {
      "epoch": 0.6796606282962623,
      "grad_norm": 0.16140400436097305,
      "learning_rate": 0.000245830974057275,
      "loss": 3.4867,
      "step": 2964
    },
    {
      "epoch": 0.6798899335014905,
      "grad_norm": 0.14523867226143328,
      "learning_rate": 0.00024551125619351385,
      "loss": 3.4955,
      "step": 2965
    },
    {
      "epoch": 0.6801192387067186,
      "grad_norm": 0.16034029627899185,
      "learning_rate": 0.00024519167870404123,
      "loss": 3.5895,
      "step": 2966
    },
    {
      "epoch": 0.6803485439119468,
      "grad_norm": 0.1428380487679418,
      "learning_rate": 0.00024487224176513453,
      "loss": 3.4886,
      "step": 2967
    },
    {
      "epoch": 0.680577849117175,
      "grad_norm": 0.14513180839786066,
      "learning_rate": 0.00024455294555299217,
      "loss": 3.4133,
      "step": 2968
    },
    {
      "epoch": 0.6808071543224031,
      "grad_norm": 0.17143009983300428,
      "learning_rate": 0.0002442337902437365,
      "loss": 3.4217,
      "step": 2969
    },
    {
      "epoch": 0.6810364595276313,
      "grad_norm": 0.1588322312319082,
      "learning_rate": 0.000243914776013411,
      "loss": 3.334,
      "step": 2970
    },
    {
      "epoch": 0.6812657647328595,
      "grad_norm": 0.15365368616455763,
      "learning_rate": 0.0002435959030379824,
      "loss": 3.523,
      "step": 2971
    },
    {
      "epoch": 0.6814950699380876,
      "grad_norm": 0.139358185994878,
      "learning_rate": 0.0002432771714933381,
      "loss": 3.5256,
      "step": 2972
    },
    {
      "epoch": 0.6817243751433157,
      "grad_norm": 0.15008661336817147,
      "learning_rate": 0.00024295858155528888,
      "loss": 3.4653,
      "step": 2973
    },
    {
      "epoch": 0.6819536803485439,
      "grad_norm": 0.1488126886294541,
      "learning_rate": 0.0002426401333995666,
      "loss": 3.4569,
      "step": 2974
    },
    {
      "epoch": 0.6821829855537721,
      "grad_norm": 0.13779973814718094,
      "learning_rate": 0.00024232182720182523,
      "loss": 3.4216,
      "step": 2975
    },
    {
      "epoch": 0.6824122907590002,
      "grad_norm": 0.1621352797008553,
      "learning_rate": 0.00024200366313764,
      "loss": 3.5005,
      "step": 2976
    },
    {
      "epoch": 0.6826415959642284,
      "grad_norm": 0.15696914706384518,
      "learning_rate": 0.00024168564138250855,
      "loss": 3.5592,
      "step": 2977
    },
    {
      "epoch": 0.6828709011694566,
      "grad_norm": 0.14492415556467536,
      "learning_rate": 0.00024136776211184936,
      "loss": 3.5143,
      "step": 2978
    },
    {
      "epoch": 0.6831002063746847,
      "grad_norm": 0.16687660670394366,
      "learning_rate": 0.00024105002550100246,
      "loss": 3.4995,
      "step": 2979
    },
    {
      "epoch": 0.6833295115799128,
      "grad_norm": 0.1625209160277466,
      "learning_rate": 0.00024073243172522912,
      "loss": 3.4753,
      "step": 2980
    },
    {
      "epoch": 0.6835588167851411,
      "grad_norm": 0.17835336590737486,
      "learning_rate": 0.00024041498095971254,
      "loss": 3.4281,
      "step": 2981
    },
    {
      "epoch": 0.6837881219903692,
      "grad_norm": 0.16045455851830054,
      "learning_rate": 0.00024009767337955553,
      "loss": 3.4402,
      "step": 2982
    },
    {
      "epoch": 0.6840174271955973,
      "grad_norm": 0.15306065695896265,
      "learning_rate": 0.0002397805091597835,
      "loss": 3.4897,
      "step": 2983
    },
    {
      "epoch": 0.6842467324008255,
      "grad_norm": 0.14927160508017653,
      "learning_rate": 0.00023946348847534194,
      "loss": 3.2411,
      "step": 2984
    },
    {
      "epoch": 0.6844760376060537,
      "grad_norm": 0.16569683899670387,
      "learning_rate": 0.0002391466115010973,
      "loss": 3.4871,
      "step": 2985
    },
    {
      "epoch": 0.6847053428112818,
      "grad_norm": 0.1447367311020404,
      "learning_rate": 0.00023882987841183657,
      "loss": 3.4893,
      "step": 2986
    },
    {
      "epoch": 0.6849346480165099,
      "grad_norm": 0.1532088488447268,
      "learning_rate": 0.00023851328938226808,
      "loss": 3.4202,
      "step": 2987
    },
    {
      "epoch": 0.6851639532217382,
      "grad_norm": 0.1708065198464051,
      "learning_rate": 0.00023819684458701996,
      "loss": 3.5338,
      "step": 2988
    },
    {
      "epoch": 0.6853932584269663,
      "grad_norm": 0.1605941040308506,
      "learning_rate": 0.00023788054420064109,
      "loss": 3.5557,
      "step": 2989
    },
    {
      "epoch": 0.6856225636321944,
      "grad_norm": 0.15508669377848197,
      "learning_rate": 0.0002375643883976004,
      "loss": 3.3508,
      "step": 2990
    },
    {
      "epoch": 0.6858518688374227,
      "grad_norm": 0.1575005945901048,
      "learning_rate": 0.00023724837735228773,
      "loss": 3.6209,
      "step": 2991
    },
    {
      "epoch": 0.6860811740426508,
      "grad_norm": 0.15164945622714657,
      "learning_rate": 0.00023693251123901244,
      "loss": 3.4845,
      "step": 2992
    },
    {
      "epoch": 0.6863104792478789,
      "grad_norm": 0.15104127734667425,
      "learning_rate": 0.00023661679023200422,
      "loss": 3.3953,
      "step": 2993
    },
    {
      "epoch": 0.686539784453107,
      "grad_norm": 0.16501149364548034,
      "learning_rate": 0.00023630121450541247,
      "loss": 3.3492,
      "step": 2994
    },
    {
      "epoch": 0.6867690896583353,
      "grad_norm": 0.17739841902977366,
      "learning_rate": 0.00023598578423330714,
      "loss": 3.5598,
      "step": 2995
    },
    {
      "epoch": 0.6869983948635634,
      "grad_norm": 0.17704679764259437,
      "learning_rate": 0.00023567049958967678,
      "loss": 3.3728,
      "step": 2996
    },
    {
      "epoch": 0.6872277000687915,
      "grad_norm": 0.17408304414842926,
      "learning_rate": 0.00023535536074843083,
      "loss": 3.4802,
      "step": 2997
    },
    {
      "epoch": 0.6874570052740198,
      "grad_norm": 0.16158407447499165,
      "learning_rate": 0.0002350403678833976,
      "loss": 3.385,
      "step": 2998
    },
    {
      "epoch": 0.6876863104792479,
      "grad_norm": 0.16332309114261467,
      "learning_rate": 0.00023472552116832502,
      "loss": 3.5148,
      "step": 2999
    },
    {
      "epoch": 0.687915615684476,
      "grad_norm": 0.16144889439146218,
      "learning_rate": 0.00023441082077688025,
      "loss": 3.5984,
      "step": 3000
    },
    {
      "epoch": 0.6881449208897042,
      "grad_norm": 0.16364244587312377,
      "learning_rate": 0.0002340962668826503,
      "loss": 3.3716,
      "step": 3001
    },
    {
      "epoch": 0.6883742260949324,
      "grad_norm": 0.17614186115301317,
      "learning_rate": 0.0002337818596591408,
      "loss": 3.5169,
      "step": 3002
    },
    {
      "epoch": 0.6886035313001605,
      "grad_norm": 0.17097088774862154,
      "learning_rate": 0.00023346759927977663,
      "loss": 3.4748,
      "step": 3003
    },
    {
      "epoch": 0.6888328365053886,
      "grad_norm": 0.1665954828576739,
      "learning_rate": 0.00023315348591790164,
      "loss": 3.5715,
      "step": 3004
    },
    {
      "epoch": 0.6890621417106169,
      "grad_norm": 0.16124789036821568,
      "learning_rate": 0.0002328395197467789,
      "loss": 3.6253,
      "step": 3005
    },
    {
      "epoch": 0.689291446915845,
      "grad_norm": 0.15188078693448936,
      "learning_rate": 0.00023252570093958998,
      "loss": 3.5384,
      "step": 3006
    },
    {
      "epoch": 0.6895207521210731,
      "grad_norm": 0.14919109046882273,
      "learning_rate": 0.00023221202966943515,
      "loss": 3.4821,
      "step": 3007
    },
    {
      "epoch": 0.6897500573263013,
      "grad_norm": 0.1472438375306656,
      "learning_rate": 0.00023189850610933328,
      "loss": 3.4013,
      "step": 3008
    },
    {
      "epoch": 0.6899793625315295,
      "grad_norm": 0.14637685456331193,
      "learning_rate": 0.0002315851304322223,
      "loss": 3.5175,
      "step": 3009
    },
    {
      "epoch": 0.6902086677367576,
      "grad_norm": 0.14695072563075082,
      "learning_rate": 0.0002312719028109575,
      "loss": 3.6139,
      "step": 3010
    },
    {
      "epoch": 0.6904379729419858,
      "grad_norm": 0.15687338745382925,
      "learning_rate": 0.0002309588234183137,
      "loss": 3.5857,
      "step": 3011
    },
    {
      "epoch": 0.690667278147214,
      "grad_norm": 0.16636554034950202,
      "learning_rate": 0.00023064589242698314,
      "loss": 3.3239,
      "step": 3012
    },
    {
      "epoch": 0.6908965833524421,
      "grad_norm": 0.1598860131398424,
      "learning_rate": 0.00023033311000957653,
      "loss": 3.5931,
      "step": 3013
    },
    {
      "epoch": 0.6911258885576702,
      "grad_norm": 0.15050261590991496,
      "learning_rate": 0.00023002047633862232,
      "loss": 3.2561,
      "step": 3014
    },
    {
      "epoch": 0.6913551937628984,
      "grad_norm": 0.14445191911939984,
      "learning_rate": 0.00022970799158656758,
      "loss": 3.4555,
      "step": 3015
    },
    {
      "epoch": 0.6915844989681266,
      "grad_norm": 0.15547301537403588,
      "learning_rate": 0.0002293956559257766,
      "loss": 3.3675,
      "step": 3016
    },
    {
      "epoch": 0.6918138041733547,
      "grad_norm": 0.1479790744279835,
      "learning_rate": 0.0002290834695285316,
      "loss": 3.4886,
      "step": 3017
    },
    {
      "epoch": 0.6920431093785829,
      "grad_norm": 0.15857930760747208,
      "learning_rate": 0.00022877143256703236,
      "loss": 3.5083,
      "step": 3018
    },
    {
      "epoch": 0.6922724145838111,
      "grad_norm": 0.14731568311668408,
      "learning_rate": 0.00022845954521339678,
      "loss": 3.4732,
      "step": 3019
    },
    {
      "epoch": 0.6925017197890392,
      "grad_norm": 0.1492401136188573,
      "learning_rate": 0.0002281478076396596,
      "loss": 3.3237,
      "step": 3020
    },
    {
      "epoch": 0.6927310249942674,
      "grad_norm": 0.15708937218413777,
      "learning_rate": 0.0002278362200177732,
      "loss": 3.4828,
      "step": 3021
    },
    {
      "epoch": 0.6929603301994955,
      "grad_norm": 0.15174659169726817,
      "learning_rate": 0.000227524782519607,
      "loss": 3.3377,
      "step": 3022
    },
    {
      "epoch": 0.6931896354047237,
      "grad_norm": 0.15899319142156398,
      "learning_rate": 0.00022721349531694852,
      "loss": 3.5031,
      "step": 3023
    },
    {
      "epoch": 0.6934189406099518,
      "grad_norm": 0.15188481295389164,
      "learning_rate": 0.00022690235858150083,
      "loss": 3.4723,
      "step": 3024
    },
    {
      "epoch": 0.69364824581518,
      "grad_norm": 0.15753330767568877,
      "learning_rate": 0.0002265913724848855,
      "loss": 3.4678,
      "step": 3025
    },
    {
      "epoch": 0.6938775510204082,
      "grad_norm": 0.16155356758256176,
      "learning_rate": 0.00022628053719864018,
      "loss": 3.5688,
      "step": 3026
    },
    {
      "epoch": 0.6941068562256363,
      "grad_norm": 0.15235938048487127,
      "learning_rate": 0.00022596985289421946,
      "loss": 3.4251,
      "step": 3027
    },
    {
      "epoch": 0.6943361614308645,
      "grad_norm": 0.15225846367135729,
      "learning_rate": 0.00022565931974299463,
      "loss": 3.3738,
      "step": 3028
    },
    {
      "epoch": 0.6945654666360926,
      "grad_norm": 0.15402957381253682,
      "learning_rate": 0.00022534893791625405,
      "loss": 3.4795,
      "step": 3029
    },
    {
      "epoch": 0.6947947718413208,
      "grad_norm": 0.15114939982217984,
      "learning_rate": 0.00022503870758520212,
      "loss": 3.2866,
      "step": 3030
    },
    {
      "epoch": 0.695024077046549,
      "grad_norm": 0.15989945869669966,
      "learning_rate": 0.00022472862892095968,
      "loss": 3.6585,
      "step": 3031
    },
    {
      "epoch": 0.6952533822517771,
      "grad_norm": 0.15031890883649593,
      "learning_rate": 0.000224418702094564,
      "loss": 3.4844,
      "step": 3032
    },
    {
      "epoch": 0.6954826874570053,
      "grad_norm": 0.1514728390292168,
      "learning_rate": 0.00022410892727696896,
      "loss": 3.5739,
      "step": 3033
    },
    {
      "epoch": 0.6957119926622334,
      "grad_norm": 0.16044414519822678,
      "learning_rate": 0.00022379930463904408,
      "loss": 3.4264,
      "step": 3034
    },
    {
      "epoch": 0.6959412978674616,
      "grad_norm": 0.13984517008272843,
      "learning_rate": 0.0002234898343515751,
      "loss": 3.3716,
      "step": 3035
    },
    {
      "epoch": 0.6961706030726897,
      "grad_norm": 0.1453784531596726,
      "learning_rate": 0.00022318051658526368,
      "loss": 3.4874,
      "step": 3036
    },
    {
      "epoch": 0.6963999082779179,
      "grad_norm": 0.1574975411015076,
      "learning_rate": 0.00022287135151072792,
      "loss": 3.5553,
      "step": 3037
    },
    {
      "epoch": 0.6966292134831461,
      "grad_norm": 0.1524460324044553,
      "learning_rate": 0.00022256233929850044,
      "loss": 3.5125,
      "step": 3038
    },
    {
      "epoch": 0.6968585186883742,
      "grad_norm": 0.16363562688818967,
      "learning_rate": 0.00022225348011903096,
      "loss": 3.6763,
      "step": 3039
    },
    {
      "epoch": 0.6970878238936024,
      "grad_norm": 0.169599414261328,
      "learning_rate": 0.00022194477414268384,
      "loss": 3.458,
      "step": 3040
    },
    {
      "epoch": 0.6973171290988306,
      "grad_norm": 0.1625459951757107,
      "learning_rate": 0.0002216362215397393,
      "loss": 3.5114,
      "step": 3041
    },
    {
      "epoch": 0.6975464343040587,
      "grad_norm": 0.15656477745648498,
      "learning_rate": 0.0002213278224803927,
      "loss": 3.4251,
      "step": 3042
    },
    {
      "epoch": 0.6977757395092868,
      "grad_norm": 0.15279156184324144,
      "learning_rate": 0.00022101957713475522,
      "loss": 3.3884,
      "step": 3043
    },
    {
      "epoch": 0.698005044714515,
      "grad_norm": 0.1625906099537145,
      "learning_rate": 0.00022071148567285283,
      "loss": 3.5261,
      "step": 3044
    },
    {
      "epoch": 0.6982343499197432,
      "grad_norm": 0.155239693553817,
      "learning_rate": 0.00022040354826462666,
      "loss": 3.3973,
      "step": 3045
    },
    {
      "epoch": 0.6984636551249713,
      "grad_norm": 0.1420995379193434,
      "learning_rate": 0.00022009576507993274,
      "loss": 3.3883,
      "step": 3046
    },
    {
      "epoch": 0.6986929603301995,
      "grad_norm": 0.1643230936460663,
      "learning_rate": 0.0002197881362885426,
      "loss": 3.4823,
      "step": 3047
    },
    {
      "epoch": 0.6989222655354277,
      "grad_norm": 0.1523448876261538,
      "learning_rate": 0.00021948066206014207,
      "loss": 3.4811,
      "step": 3048
    },
    {
      "epoch": 0.6991515707406558,
      "grad_norm": 0.1540023683610715,
      "learning_rate": 0.0002191733425643318,
      "loss": 3.363,
      "step": 3049
    },
    {
      "epoch": 0.6993808759458839,
      "grad_norm": 0.17740430897210152,
      "learning_rate": 0.00021886617797062702,
      "loss": 3.5092,
      "step": 3050
    },
    {
      "epoch": 0.6996101811511122,
      "grad_norm": 0.1744024368888741,
      "learning_rate": 0.00021855916844845826,
      "loss": 3.3928,
      "step": 3051
    },
    {
      "epoch": 0.6998394863563403,
      "grad_norm": 0.15590788343380638,
      "learning_rate": 0.0002182523141671691,
      "loss": 3.4509,
      "step": 3052
    },
    {
      "epoch": 0.7000687915615684,
      "grad_norm": 0.1696765849988407,
      "learning_rate": 0.00021794561529601898,
      "loss": 3.2853,
      "step": 3053
    },
    {
      "epoch": 0.7002980967667966,
      "grad_norm": 0.1613175444341577,
      "learning_rate": 0.00021763907200418066,
      "loss": 3.3519,
      "step": 3054
    },
    {
      "epoch": 0.7005274019720248,
      "grad_norm": 0.1575862457207413,
      "learning_rate": 0.00021733268446074138,
      "loss": 3.4658,
      "step": 3055
    },
    {
      "epoch": 0.7007567071772529,
      "grad_norm": 0.15636483360994297,
      "learning_rate": 0.00021702645283470234,
      "loss": 3.2944,
      "step": 3056
    },
    {
      "epoch": 0.700986012382481,
      "grad_norm": 0.14216024816383924,
      "learning_rate": 0.00021672037729497917,
      "loss": 3.4538,
      "step": 3057
    },
    {
      "epoch": 0.7012153175877093,
      "grad_norm": 0.15582016372351085,
      "learning_rate": 0.00021641445801040088,
      "loss": 3.3835,
      "step": 3058
    },
    {
      "epoch": 0.7014446227929374,
      "grad_norm": 0.14352381408055395,
      "learning_rate": 0.0002161086951497106,
      "loss": 3.3822,
      "step": 3059
    },
    {
      "epoch": 0.7016739279981655,
      "grad_norm": 0.14671465921877408,
      "learning_rate": 0.00021580308888156496,
      "loss": 3.4618,
      "step": 3060
    },
    {
      "epoch": 0.7019032332033938,
      "grad_norm": 0.152303652884044,
      "learning_rate": 0.00021549763937453442,
      "loss": 3.3787,
      "step": 3061
    },
    {
      "epoch": 0.7021325384086219,
      "grad_norm": 0.1535942597556452,
      "learning_rate": 0.00021519234679710282,
      "loss": 3.3901,
      "step": 3062
    },
    {
      "epoch": 0.70236184361385,
      "grad_norm": 0.16294638886512125,
      "learning_rate": 0.00021488721131766736,
      "loss": 3.5256,
      "step": 3063
    },
    {
      "epoch": 0.7025911488190782,
      "grad_norm": 0.16642513919551447,
      "learning_rate": 0.00021458223310453912,
      "loss": 3.4646,
      "step": 3064
    },
    {
      "epoch": 0.7028204540243064,
      "grad_norm": 0.1398286316247193,
      "learning_rate": 0.00021427741232594183,
      "loss": 3.426,
      "step": 3065
    },
    {
      "epoch": 0.7030497592295345,
      "grad_norm": 0.15763337850375547,
      "learning_rate": 0.00021397274915001252,
      "loss": 3.4388,
      "step": 3066
    },
    {
      "epoch": 0.7032790644347626,
      "grad_norm": 0.1592416742129576,
      "learning_rate": 0.0002136682437448013,
      "loss": 3.5913,
      "step": 3067
    },
    {
      "epoch": 0.7035083696399909,
      "grad_norm": 0.15606189039744503,
      "learning_rate": 0.00021336389627827168,
      "loss": 3.5642,
      "step": 3068
    },
    {
      "epoch": 0.703737674845219,
      "grad_norm": 0.15681537012453556,
      "learning_rate": 0.0002130597069182994,
      "loss": 3.5076,
      "step": 3069
    },
    {
      "epoch": 0.7039669800504471,
      "grad_norm": 0.14286670529686846,
      "learning_rate": 0.0002127556758326733,
      "loss": 3.429,
      "step": 3070
    },
    {
      "epoch": 0.7041962852556753,
      "grad_norm": 0.1617869477457725,
      "learning_rate": 0.0002124518031890948,
      "loss": 3.3827,
      "step": 3071
    },
    {
      "epoch": 0.7044255904609035,
      "grad_norm": 0.18024178703643035,
      "learning_rate": 0.00021214808915517847,
      "loss": 3.4536,
      "step": 3072
    },
    {
      "epoch": 0.7046548956661316,
      "grad_norm": 0.1529939653812452,
      "learning_rate": 0.0002118445338984502,
      "loss": 3.2859,
      "step": 3073
    },
    {
      "epoch": 0.7048842008713598,
      "grad_norm": 0.16246589322595803,
      "learning_rate": 0.0002115411375863497,
      "loss": 3.3276,
      "step": 3074
    },
    {
      "epoch": 0.705113506076588,
      "grad_norm": 0.16059521632909593,
      "learning_rate": 0.00021123790038622808,
      "loss": 3.3414,
      "step": 3075
    },
    {
      "epoch": 0.7053428112818161,
      "grad_norm": 0.1508934268703505,
      "learning_rate": 0.00021093482246534894,
      "loss": 3.2708,
      "step": 3076
    },
    {
      "epoch": 0.7055721164870442,
      "grad_norm": 0.15127371244713053,
      "learning_rate": 0.0002106319039908879,
      "loss": 3.3701,
      "step": 3077
    },
    {
      "epoch": 0.7058014216922724,
      "grad_norm": 0.16101177807102948,
      "learning_rate": 0.00021032914512993312,
      "loss": 3.5555,
      "step": 3078
    },
    {
      "epoch": 0.7060307268975006,
      "grad_norm": 0.17382779025366987,
      "learning_rate": 0.00021002654604948412,
      "loss": 3.4708,
      "step": 3079
    },
    {
      "epoch": 0.7062600321027287,
      "grad_norm": 0.15626968552490314,
      "learning_rate": 0.0002097241069164526,
      "loss": 3.3669,
      "step": 3080
    },
    {
      "epoch": 0.7064893373079569,
      "grad_norm": 0.15035581209011398,
      "learning_rate": 0.00020942182789766172,
      "loss": 3.4494,
      "step": 3081
    },
    {
      "epoch": 0.7067186425131851,
      "grad_norm": 0.17376889088642863,
      "learning_rate": 0.00020911970915984702,
      "loss": 3.4625,
      "step": 3082
    },
    {
      "epoch": 0.7069479477184132,
      "grad_norm": 0.15133028433346318,
      "learning_rate": 0.00020881775086965492,
      "loss": 3.3834,
      "step": 3083
    },
    {
      "epoch": 0.7071772529236414,
      "grad_norm": 0.17081540405951434,
      "learning_rate": 0.00020851595319364358,
      "loss": 3.4952,
      "step": 3084
    },
    {
      "epoch": 0.7074065581288695,
      "grad_norm": 0.18268206468860293,
      "learning_rate": 0.00020821431629828246,
      "loss": 3.5082,
      "step": 3085
    },
    {
      "epoch": 0.7076358633340977,
      "grad_norm": 0.155650771582544,
      "learning_rate": 0.00020791284034995294,
      "loss": 3.5927,
      "step": 3086
    },
    {
      "epoch": 0.7078651685393258,
      "grad_norm": 0.16036696589894672,
      "learning_rate": 0.00020761152551494643,
      "loss": 3.4092,
      "step": 3087
    },
    {
      "epoch": 0.708094473744554,
      "grad_norm": 0.158990364932579,
      "learning_rate": 0.00020731037195946677,
      "loss": 3.5451,
      "step": 3088
    },
    {
      "epoch": 0.7083237789497822,
      "grad_norm": 0.15993402103122173,
      "learning_rate": 0.00020700937984962798,
      "loss": 3.434,
      "step": 3089
    },
    {
      "epoch": 0.7085530841550103,
      "grad_norm": 0.15900427787532134,
      "learning_rate": 0.00020670854935145533,
      "loss": 3.4657,
      "step": 3090
    },
    {
      "epoch": 0.7087823893602385,
      "grad_norm": 0.1509790832240358,
      "learning_rate": 0.0002064078806308848,
      "loss": 3.3971,
      "step": 3091
    },
    {
      "epoch": 0.7090116945654666,
      "grad_norm": 0.13875239266204478,
      "learning_rate": 0.00020610737385376348,
      "loss": 3.3659,
      "step": 3092
    },
    {
      "epoch": 0.7092409997706948,
      "grad_norm": 0.1534162967284318,
      "learning_rate": 0.00020580702918584882,
      "loss": 3.4054,
      "step": 3093
    },
    {
      "epoch": 0.709470304975923,
      "grad_norm": 0.16169527684345722,
      "learning_rate": 0.00020550684679280885,
      "loss": 3.4381,
      "step": 3094
    },
    {
      "epoch": 0.7096996101811511,
      "grad_norm": 0.153371620657946,
      "learning_rate": 0.000205206826840222,
      "loss": 3.426,
      "step": 3095
    },
    {
      "epoch": 0.7099289153863793,
      "grad_norm": 0.14776454658074598,
      "learning_rate": 0.00020490696949357774,
      "loss": 3.6343,
      "step": 3096
    },
    {
      "epoch": 0.7101582205916074,
      "grad_norm": 0.15361184613161782,
      "learning_rate": 0.0002046072749182751,
      "loss": 3.4457,
      "step": 3097
    },
    {
      "epoch": 0.7103875257968356,
      "grad_norm": 0.16466864889999905,
      "learning_rate": 0.0002043077432796236,
      "loss": 3.4824,
      "step": 3098
    },
    {
      "epoch": 0.7106168310020637,
      "grad_norm": 0.1737984239519585,
      "learning_rate": 0.00020400837474284273,
      "loss": 3.5075,
      "step": 3099
    },
    {
      "epoch": 0.7108461362072919,
      "grad_norm": 0.1480652230720716,
      "learning_rate": 0.00020370916947306267,
      "loss": 3.4492,
      "step": 3100
    },
    {
      "epoch": 0.7110754414125201,
      "grad_norm": 0.15312661212765674,
      "learning_rate": 0.0002034101276353224,
      "loss": 3.4572,
      "step": 3101
    },
    {
      "epoch": 0.7113047466177482,
      "grad_norm": 0.15330902849030822,
      "learning_rate": 0.00020311124939457193,
      "loss": 3.4407,
      "step": 3102
    },
    {
      "epoch": 0.7115340518229764,
      "grad_norm": 0.13935959279601703,
      "learning_rate": 0.00020281253491567027,
      "loss": 3.4769,
      "step": 3103
    },
    {
      "epoch": 0.7117633570282046,
      "grad_norm": 0.15426363297099535,
      "learning_rate": 0.0002025139843633863,
      "loss": 3.4958,
      "step": 3104
    },
    {
      "epoch": 0.7119926622334327,
      "grad_norm": 0.13498548050883513,
      "learning_rate": 0.0002022155979023984,
      "loss": 3.4914,
      "step": 3105
    },
    {
      "epoch": 0.7122219674386608,
      "grad_norm": 0.16300413674159211,
      "learning_rate": 0.00020191737569729491,
      "loss": 3.5265,
      "step": 3106
    },
    {
      "epoch": 0.712451272643889,
      "grad_norm": 0.14937531985650118,
      "learning_rate": 0.000201619317912573,
      "loss": 3.5304,
      "step": 3107
    },
    {
      "epoch": 0.7126805778491172,
      "grad_norm": 0.1409862175209832,
      "learning_rate": 0.00020132142471263932,
      "loss": 3.4308,
      "step": 3108
    },
    {
      "epoch": 0.7129098830543453,
      "grad_norm": 0.15750771640392028,
      "learning_rate": 0.00020102369626180962,
      "loss": 3.5508,
      "step": 3109
    },
    {
      "epoch": 0.7131391882595735,
      "grad_norm": 0.176288976289614,
      "learning_rate": 0.00020072613272430922,
      "loss": 3.4916,
      "step": 3110
    },
    {
      "epoch": 0.7133684934648017,
      "grad_norm": 0.1500694169014916,
      "learning_rate": 0.0002004287342642721,
      "loss": 3.506,
      "step": 3111
    },
    {
      "epoch": 0.7135977986700298,
      "grad_norm": 0.1678845018825507,
      "learning_rate": 0.00020013150104574123,
      "loss": 3.2944,
      "step": 3112
    },
    {
      "epoch": 0.7138271038752579,
      "grad_norm": 0.16812684315953172,
      "learning_rate": 0.00019983443323266824,
      "loss": 3.3863,
      "step": 3113
    },
    {
      "epoch": 0.7140564090804862,
      "grad_norm": 0.16692630171891865,
      "learning_rate": 0.0001995375309889143,
      "loss": 3.4803,
      "step": 3114
    },
    {
      "epoch": 0.7142857142857143,
      "grad_norm": 0.16286000126405248,
      "learning_rate": 0.00019924079447824805,
      "loss": 3.4353,
      "step": 3115
    },
    {
      "epoch": 0.7145150194909424,
      "grad_norm": 0.18007491679202955,
      "learning_rate": 0.0001989442238643478,
      "loss": 3.4257,
      "step": 3116
    },
    {
      "epoch": 0.7147443246961706,
      "grad_norm": 0.16313025299190032,
      "learning_rate": 0.00019864781931079977,
      "loss": 3.4823,
      "step": 3117
    },
    {
      "epoch": 0.7149736299013988,
      "grad_norm": 0.18209797406497374,
      "learning_rate": 0.0001983515809810988,
      "loss": 3.4532,
      "step": 3118
    },
    {
      "epoch": 0.7152029351066269,
      "grad_norm": 0.15861262968710335,
      "learning_rate": 0.00019805550903864773,
      "loss": 3.4843,
      "step": 3119
    },
    {
      "epoch": 0.715432240311855,
      "grad_norm": 0.1667089703048363,
      "learning_rate": 0.00019775960364675822,
      "loss": 3.4915,
      "step": 3120
    },
    {
      "epoch": 0.7156615455170833,
      "grad_norm": 0.16322455450497783,
      "learning_rate": 0.00019746386496864948,
      "loss": 3.2929,
      "step": 3121
    },
    {
      "epoch": 0.7158908507223114,
      "grad_norm": 0.15000905545554113,
      "learning_rate": 0.00019716829316744912,
      "loss": 3.455,
      "step": 3122
    },
    {
      "epoch": 0.7161201559275395,
      "grad_norm": 0.16747419902662747,
      "learning_rate": 0.00019687288840619226,
      "loss": 3.4533,
      "step": 3123
    },
    {
      "epoch": 0.7163494611327678,
      "grad_norm": 0.1546292689214374,
      "learning_rate": 0.00019657765084782282,
      "loss": 3.4757,
      "step": 3124
    },
    {
      "epoch": 0.7165787663379959,
      "grad_norm": 0.14999905702356517,
      "learning_rate": 0.0001962825806551911,
      "loss": 3.4025,
      "step": 3125
    },
    {
      "epoch": 0.716808071543224,
      "grad_norm": 0.15574785156064064,
      "learning_rate": 0.00019598767799105637,
      "loss": 3.4491,
      "step": 3126
    },
    {
      "epoch": 0.7170373767484521,
      "grad_norm": 0.16704481691758177,
      "learning_rate": 0.0001956929430180846,
      "loss": 3.4986,
      "step": 3127
    },
    {
      "epoch": 0.7172666819536804,
      "grad_norm": 0.157957807204373,
      "learning_rate": 0.00019539837589885024,
      "loss": 3.5296,
      "step": 3128
    },
    {
      "epoch": 0.7174959871589085,
      "grad_norm": 0.15217635939010815,
      "learning_rate": 0.00019510397679583374,
      "loss": 3.3822,
      "step": 3129
    },
    {
      "epoch": 0.7177252923641366,
      "grad_norm": 0.1491136798531693,
      "learning_rate": 0.00019480974587142425,
      "loss": 3.4941,
      "step": 3130
    },
    {
      "epoch": 0.7179545975693649,
      "grad_norm": 0.16538063209941545,
      "learning_rate": 0.0001945156832879174,
      "loss": 3.4896,
      "step": 3131
    },
    {
      "epoch": 0.718183902774593,
      "grad_norm": 0.16316965043599427,
      "learning_rate": 0.00019422178920751616,
      "loss": 3.6016,
      "step": 3132
    },
    {
      "epoch": 0.7184132079798211,
      "grad_norm": 0.16467420662337626,
      "learning_rate": 0.00019392806379233036,
      "loss": 3.2296,
      "step": 3133
    },
    {
      "epoch": 0.7186425131850493,
      "grad_norm": 0.14603676298974005,
      "learning_rate": 0.0001936345072043773,
      "loss": 3.3309,
      "step": 3134
    },
    {
      "epoch": 0.7188718183902775,
      "grad_norm": 0.16919332912210971,
      "learning_rate": 0.00019334111960558065,
      "loss": 3.5422,
      "step": 3135
    },
    {
      "epoch": 0.7191011235955056,
      "grad_norm": 0.15180484971417926,
      "learning_rate": 0.0001930479011577711,
      "loss": 3.4577,
      "step": 3136
    },
    {
      "epoch": 0.7193304288007337,
      "grad_norm": 0.1515927160616106,
      "learning_rate": 0.00019275485202268573,
      "loss": 3.6226,
      "step": 3137
    },
    {
      "epoch": 0.719559734005962,
      "grad_norm": 0.17231530520709987,
      "learning_rate": 0.0001924619723619691,
      "loss": 3.4721,
      "step": 3138
    },
    {
      "epoch": 0.7197890392111901,
      "grad_norm": 0.16720654572580052,
      "learning_rate": 0.00019216926233717085,
      "loss": 3.3631,
      "step": 3139
    },
    {
      "epoch": 0.7200183444164182,
      "grad_norm": 0.17682100601851694,
      "learning_rate": 0.00019187672210974855,
      "loss": 3.4908,
      "step": 3140
    },
    {
      "epoch": 0.7202476496216464,
      "grad_norm": 0.16959977920286007,
      "learning_rate": 0.00019158435184106498,
      "loss": 3.453,
      "step": 3141
    },
    {
      "epoch": 0.7204769548268746,
      "grad_norm": 0.16177494367860704,
      "learning_rate": 0.00019129215169239017,
      "loss": 3.5374,
      "step": 3142
    },
    {
      "epoch": 0.7207062600321027,
      "grad_norm": 0.15050909092679077,
      "learning_rate": 0.00019100012182489905,
      "loss": 3.6369,
      "step": 3143
    },
    {
      "epoch": 0.7209355652373309,
      "grad_norm": 0.16507395560250782,
      "learning_rate": 0.00019070826239967387,
      "loss": 3.5558,
      "step": 3144
    },
    {
      "epoch": 0.7211648704425591,
      "grad_norm": 0.15898127005842155,
      "learning_rate": 0.00019041657357770226,
      "loss": 3.4815,
      "step": 3145
    },
    {
      "epoch": 0.7213941756477872,
      "grad_norm": 0.15680197081909897,
      "learning_rate": 0.00019012505551987762,
      "loss": 3.4767,
      "step": 3146
    },
    {
      "epoch": 0.7216234808530153,
      "grad_norm": 0.15932399527526186,
      "learning_rate": 0.00018983370838699943,
      "loss": 3.5245,
      "step": 3147
    },
    {
      "epoch": 0.7218527860582435,
      "grad_norm": 0.1598049481408077,
      "learning_rate": 0.00018954253233977297,
      "loss": 3.4828,
      "step": 3148
    },
    {
      "epoch": 0.7220820912634717,
      "grad_norm": 0.16931898053903874,
      "learning_rate": 0.00018925152753880892,
      "loss": 3.5593,
      "step": 3149
    },
    {
      "epoch": 0.7223113964686998,
      "grad_norm": 0.18004357048929576,
      "learning_rate": 0.00018896069414462357,
      "loss": 3.4228,
      "step": 3150
    },
    {
      "epoch": 0.722540701673928,
      "grad_norm": 0.16059006379913027,
      "learning_rate": 0.00018867003231763847,
      "loss": 3.5963,
      "step": 3151
    },
    {
      "epoch": 0.7227700068791562,
      "grad_norm": 0.17748691532014335,
      "learning_rate": 0.00018837954221818132,
      "loss": 3.4421,
      "step": 3152
    },
    {
      "epoch": 0.7229993120843843,
      "grad_norm": 0.16413288880891208,
      "learning_rate": 0.00018808922400648375,
      "loss": 3.6616,
      "step": 3153
    },
    {
      "epoch": 0.7232286172896125,
      "grad_norm": 0.17652265123869804,
      "learning_rate": 0.0001877990778426839,
      "loss": 3.5741,
      "step": 3154
    },
    {
      "epoch": 0.7234579224948406,
      "grad_norm": 0.1722941794680971,
      "learning_rate": 0.00018750910388682428,
      "loss": 3.4818,
      "step": 3155
    },
    {
      "epoch": 0.7236872277000688,
      "grad_norm": 0.15473901590596195,
      "learning_rate": 0.0001872193022988526,
      "loss": 3.4882,
      "step": 3156
    },
    {
      "epoch": 0.723916532905297,
      "grad_norm": 0.15701212970836528,
      "learning_rate": 0.00018692967323862125,
      "loss": 3.4509,
      "step": 3157
    },
    {
      "epoch": 0.7241458381105251,
      "grad_norm": 0.15984072990028547,
      "learning_rate": 0.0001866402168658881,
      "loss": 3.5207,
      "step": 3158
    },
    {
      "epoch": 0.7243751433157533,
      "grad_norm": 0.16635439900077365,
      "learning_rate": 0.00018635093334031517,
      "loss": 3.4171,
      "step": 3159
    },
    {
      "epoch": 0.7246044485209814,
      "grad_norm": 0.1445303969463516,
      "learning_rate": 0.00018606182282146922,
      "loss": 3.3683,
      "step": 3160
    },
    {
      "epoch": 0.7248337537262096,
      "grad_norm": 0.16354196966508627,
      "learning_rate": 0.00018577288546882165,
      "loss": 3.4243,
      "step": 3161
    },
    {
      "epoch": 0.7250630589314377,
      "grad_norm": 0.15338672515940177,
      "learning_rate": 0.00018548412144174859,
      "loss": 3.4735,
      "step": 3162
    },
    {
      "epoch": 0.7252923641366659,
      "grad_norm": 0.18581486307208003,
      "learning_rate": 0.00018519553089953023,
      "loss": 3.383,
      "step": 3163
    },
    {
      "epoch": 0.7255216693418941,
      "grad_norm": 0.1721660854939421,
      "learning_rate": 0.0001849071140013512,
      "loss": 3.4912,
      "step": 3164
    },
    {
      "epoch": 0.7257509745471222,
      "grad_norm": 0.18074408780720397,
      "learning_rate": 0.0001846188709063001,
      "loss": 3.4713,
      "step": 3165
    },
    {
      "epoch": 0.7259802797523504,
      "grad_norm": 0.16256710369163996,
      "learning_rate": 0.00018433080177337043,
      "loss": 3.408,
      "step": 3166
    },
    {
      "epoch": 0.7262095849575786,
      "grad_norm": 0.14708659759612058,
      "learning_rate": 0.00018404290676145857,
      "loss": 3.5011,
      "step": 3167
    },
    {
      "epoch": 0.7264388901628067,
      "grad_norm": 0.1448076756792702,
      "learning_rate": 0.00018375518602936592,
      "loss": 3.3771,
      "step": 3168
    },
    {
      "epoch": 0.7266681953680348,
      "grad_norm": 0.1763720322653574,
      "learning_rate": 0.00018346763973579722,
      "loss": 3.4974,
      "step": 3169
    },
    {
      "epoch": 0.726897500573263,
      "grad_norm": 0.1662235768752532,
      "learning_rate": 0.000183180268039361,
      "loss": 3.3977,
      "step": 3170
    },
    {
      "epoch": 0.7271268057784912,
      "grad_norm": 0.15261527071553177,
      "learning_rate": 0.00018289307109856939,
      "loss": 3.5594,
      "step": 3171
    },
    {
      "epoch": 0.7273561109837193,
      "grad_norm": 0.16451429462141176,
      "learning_rate": 0.00018260604907183875,
      "loss": 3.4991,
      "step": 3172
    },
    {
      "epoch": 0.7275854161889475,
      "grad_norm": 0.17936399453837254,
      "learning_rate": 0.0001823192021174882,
      "loss": 3.448,
      "step": 3173
    },
    {
      "epoch": 0.7278147213941757,
      "grad_norm": 0.1496245803673224,
      "learning_rate": 0.00018203253039374075,
      "loss": 3.4571,
      "step": 3174
    },
    {
      "epoch": 0.7280440265994038,
      "grad_norm": 0.16382961453624864,
      "learning_rate": 0.0001817460340587223,
      "loss": 3.502,
      "step": 3175
    },
    {
      "epoch": 0.7282733318046319,
      "grad_norm": 0.16403217543111664,
      "learning_rate": 0.00018145971327046274,
      "loss": 3.4364,
      "step": 3176
    },
    {
      "epoch": 0.7285026370098602,
      "grad_norm": 0.15982773440091363,
      "learning_rate": 0.00018117356818689445,
      "loss": 3.454,
      "step": 3177
    },
    {
      "epoch": 0.7287319422150883,
      "grad_norm": 0.14988368481122827,
      "learning_rate": 0.00018088759896585323,
      "loss": 3.522,
      "step": 3178
    },
    {
      "epoch": 0.7289612474203164,
      "grad_norm": 0.171686770721971,
      "learning_rate": 0.00018060180576507756,
      "loss": 3.4184,
      "step": 3179
    },
    {
      "epoch": 0.7291905526255446,
      "grad_norm": 0.17090125354913435,
      "learning_rate": 0.00018031618874220963,
      "loss": 3.4418,
      "step": 3180
    },
    {
      "epoch": 0.7294198578307728,
      "grad_norm": 0.17032557483298474,
      "learning_rate": 0.00018003074805479313,
      "loss": 3.4888,
      "step": 3181
    },
    {
      "epoch": 0.7296491630360009,
      "grad_norm": 0.182376985516253,
      "learning_rate": 0.00017974548386027583,
      "loss": 3.3867,
      "step": 3182
    },
    {
      "epoch": 0.729878468241229,
      "grad_norm": 0.16890447565790626,
      "learning_rate": 0.00017946039631600724,
      "loss": 3.5788,
      "step": 3183
    },
    {
      "epoch": 0.7301077734464573,
      "grad_norm": 0.16222293567381482,
      "learning_rate": 0.0001791754855792399,
      "loss": 3.4527,
      "step": 3184
    },
    {
      "epoch": 0.7303370786516854,
      "grad_norm": 0.17344199230954846,
      "learning_rate": 0.00017889075180712837,
      "loss": 3.5503,
      "step": 3185
    },
    {
      "epoch": 0.7305663838569135,
      "grad_norm": 0.14749142091822098,
      "learning_rate": 0.0001786061951567303,
      "loss": 3.3896,
      "step": 3186
    },
    {
      "epoch": 0.7307956890621418,
      "grad_norm": 0.1662300811205438,
      "learning_rate": 0.00017832181578500512,
      "loss": 3.4136,
      "step": 3187
    },
    {
      "epoch": 0.7310249942673699,
      "grad_norm": 0.17336570811749424,
      "learning_rate": 0.00017803761384881445,
      "loss": 3.5087,
      "step": 3188
    },
    {
      "epoch": 0.731254299472598,
      "grad_norm": 0.15179641598467197,
      "learning_rate": 0.0001777535895049221,
      "loss": 3.6845,
      "step": 3189
    },
    {
      "epoch": 0.7314836046778261,
      "grad_norm": 0.1645982059606469,
      "learning_rate": 0.0001774697429099943,
      "loss": 3.2634,
      "step": 3190
    },
    {
      "epoch": 0.7317129098830544,
      "grad_norm": 0.1540375983240674,
      "learning_rate": 0.0001771860742205988,
      "loss": 3.5738,
      "step": 3191
    },
    {
      "epoch": 0.7319422150882825,
      "grad_norm": 0.16033904808853677,
      "learning_rate": 0.00017690258359320533,
      "loss": 3.5374,
      "step": 3192
    },
    {
      "epoch": 0.7321715202935106,
      "grad_norm": 0.15455666625497355,
      "learning_rate": 0.00017661927118418525,
      "loss": 3.4491,
      "step": 3193
    },
    {
      "epoch": 0.7324008254987389,
      "grad_norm": 0.16445555556410024,
      "learning_rate": 0.00017633613714981234,
      "loss": 3.4955,
      "step": 3194
    },
    {
      "epoch": 0.732630130703967,
      "grad_norm": 0.15880472201825,
      "learning_rate": 0.00017605318164626066,
      "loss": 3.62,
      "step": 3195
    },
    {
      "epoch": 0.7328594359091951,
      "grad_norm": 0.16317976566787648,
      "learning_rate": 0.0001757704048296072,
      "loss": 3.5845,
      "step": 3196
    },
    {
      "epoch": 0.7330887411144233,
      "grad_norm": 0.1531568810039128,
      "learning_rate": 0.00017548780685582949,
      "loss": 3.5283,
      "step": 3197
    },
    {
      "epoch": 0.7333180463196515,
      "grad_norm": 0.1472845960008979,
      "learning_rate": 0.00017520538788080677,
      "loss": 3.5705,
      "step": 3198
    },
    {
      "epoch": 0.7335473515248796,
      "grad_norm": 0.16353519149467013,
      "learning_rate": 0.00017492314806031922,
      "loss": 3.6196,
      "step": 3199
    },
    {
      "epoch": 0.7337766567301077,
      "grad_norm": 0.1383005383151244,
      "learning_rate": 0.00017464108755004882,
      "loss": 3.4904,
      "step": 3200
    },
    {
      "epoch": 0.734005961935336,
      "grad_norm": 0.1481201424787422,
      "learning_rate": 0.00017435920650557806,
      "loss": 3.3703,
      "step": 3201
    },
    {
      "epoch": 0.7342352671405641,
      "grad_norm": 0.1491243053790297,
      "learning_rate": 0.00017407750508239067,
      "loss": 3.3874,
      "step": 3202
    },
    {
      "epoch": 0.7344645723457922,
      "grad_norm": 0.1507829352190925,
      "learning_rate": 0.00017379598343587112,
      "loss": 3.5688,
      "step": 3203
    },
    {
      "epoch": 0.7346938775510204,
      "grad_norm": 0.16932541754799188,
      "learning_rate": 0.0001735146417213051,
      "loss": 3.5408,
      "step": 3204
    },
    {
      "epoch": 0.7349231827562486,
      "grad_norm": 0.14646148301914436,
      "learning_rate": 0.00017323348009387878,
      "loss": 3.357,
      "step": 3205
    },
    {
      "epoch": 0.7351524879614767,
      "grad_norm": 0.15524594953872303,
      "learning_rate": 0.00017295249870867896,
      "loss": 3.4451,
      "step": 3206
    },
    {
      "epoch": 0.7353817931667049,
      "grad_norm": 0.15627235632438705,
      "learning_rate": 0.0001726716977206929,
      "loss": 3.5173,
      "step": 3207
    },
    {
      "epoch": 0.7356110983719331,
      "grad_norm": 0.16314577730424715,
      "learning_rate": 0.00017239107728480912,
      "loss": 3.437,
      "step": 3208
    },
    {
      "epoch": 0.7358404035771612,
      "grad_norm": 0.16559249946486024,
      "learning_rate": 0.00017211063755581525,
      "loss": 3.4283,
      "step": 3209
    },
    {
      "epoch": 0.7360697087823893,
      "grad_norm": 0.15786406577187184,
      "learning_rate": 0.0001718303786884005,
      "loss": 3.3942,
      "step": 3210
    },
    {
      "epoch": 0.7362990139876175,
      "grad_norm": 0.1718720259142092,
      "learning_rate": 0.0001715503008371536,
      "loss": 3.5592,
      "step": 3211
    },
    {
      "epoch": 0.7365283191928457,
      "grad_norm": 0.16700751722095203,
      "learning_rate": 0.00017127040415656364,
      "loss": 3.4951,
      "step": 3212
    },
    {
      "epoch": 0.7367576243980738,
      "grad_norm": 0.15221186206210865,
      "learning_rate": 0.0001709906888010196,
      "loss": 3.4707,
      "step": 3213
    },
    {
      "epoch": 0.736986929603302,
      "grad_norm": 0.15561098624931305,
      "learning_rate": 0.00017071115492481087,
      "loss": 3.4753,
      "step": 3214
    },
    {
      "epoch": 0.7372162348085302,
      "grad_norm": 0.15556824951631948,
      "learning_rate": 0.00017043180268212638,
      "loss": 3.4296,
      "step": 3215
    },
    {
      "epoch": 0.7374455400137583,
      "grad_norm": 0.1559127780631232,
      "learning_rate": 0.0001701526322270549,
      "loss": 3.3899,
      "step": 3216
    },
    {
      "epoch": 0.7376748452189865,
      "grad_norm": 0.1588083986404376,
      "learning_rate": 0.00016987364371358481,
      "loss": 3.4738,
      "step": 3217
    },
    {
      "epoch": 0.7379041504242146,
      "grad_norm": 0.1421069819666475,
      "learning_rate": 0.00016959483729560472,
      "loss": 3.3829,
      "step": 3218
    },
    {
      "epoch": 0.7381334556294428,
      "grad_norm": 0.16134438433033785,
      "learning_rate": 0.00016931621312690214,
      "loss": 3.5133,
      "step": 3219
    },
    {
      "epoch": 0.7383627608346709,
      "grad_norm": 0.15399829364691495,
      "learning_rate": 0.00016903777136116437,
      "loss": 3.2956,
      "step": 3220
    },
    {
      "epoch": 0.7385920660398991,
      "grad_norm": 0.1851083483332422,
      "learning_rate": 0.00016875951215197777,
      "loss": 3.444,
      "step": 3221
    },
    {
      "epoch": 0.7388213712451273,
      "grad_norm": 0.17426634623906626,
      "learning_rate": 0.0001684814356528288,
      "loss": 3.5237,
      "step": 3222
    },
    {
      "epoch": 0.7390506764503554,
      "grad_norm": 0.16993796003359538,
      "learning_rate": 0.00016820354201710214,
      "loss": 3.574,
      "step": 3223
    },
    {
      "epoch": 0.7392799816555836,
      "grad_norm": 0.16496828060609656,
      "learning_rate": 0.00016792583139808197,
      "loss": 3.4773,
      "step": 3224
    },
    {
      "epoch": 0.7395092868608117,
      "grad_norm": 0.16363016499163752,
      "learning_rate": 0.00016764830394895203,
      "loss": 3.2589,
      "step": 3225
    },
    {
      "epoch": 0.7397385920660399,
      "grad_norm": 0.14367330405008558,
      "learning_rate": 0.00016737095982279442,
      "loss": 3.5373,
      "step": 3226
    },
    {
      "epoch": 0.7399678972712681,
      "grad_norm": 0.1547862376815086,
      "learning_rate": 0.00016709379917259027,
      "loss": 3.4086,
      "step": 3227
    },
    {
      "epoch": 0.7401972024764962,
      "grad_norm": 0.1676217677501311,
      "learning_rate": 0.00016681682215121936,
      "loss": 3.3922,
      "step": 3228
    },
    {
      "epoch": 0.7404265076817244,
      "grad_norm": 0.16441738785317447,
      "learning_rate": 0.00016654002891146091,
      "loss": 3.5392,
      "step": 3229
    },
    {
      "epoch": 0.7406558128869525,
      "grad_norm": 0.157536202042249,
      "learning_rate": 0.00016626341960599157,
      "loss": 3.722,
      "step": 3230
    },
    {
      "epoch": 0.7408851180921807,
      "grad_norm": 0.14931124427612263,
      "learning_rate": 0.00016598699438738764,
      "loss": 3.5601,
      "step": 3231
    },
    {
      "epoch": 0.7411144232974088,
      "grad_norm": 0.15745735405549144,
      "learning_rate": 0.0001657107534081232,
      "loss": 3.4902,
      "step": 3232
    },
    {
      "epoch": 0.741343728502637,
      "grad_norm": 0.1614519222392803,
      "learning_rate": 0.00016543469682057105,
      "loss": 3.493,
      "step": 3233
    },
    {
      "epoch": 0.7415730337078652,
      "grad_norm": 0.147777236505778,
      "learning_rate": 0.0001651588247770019,
      "loss": 3.3605,
      "step": 3234
    },
    {
      "epoch": 0.7418023389130933,
      "grad_norm": 0.1560416409267665,
      "learning_rate": 0.00016488313742958526,
      "loss": 3.5113,
      "step": 3235
    },
    {
      "epoch": 0.7420316441183215,
      "grad_norm": 0.14440276945097122,
      "learning_rate": 0.0001646076349303884,
      "loss": 3.4742,
      "step": 3236
    },
    {
      "epoch": 0.7422609493235497,
      "grad_norm": 0.1662240764823909,
      "learning_rate": 0.00016433231743137646,
      "loss": 3.5198,
      "step": 3237
    },
    {
      "epoch": 0.7424902545287778,
      "grad_norm": 0.15925513623088447,
      "learning_rate": 0.00016405718508441264,
      "loss": 3.3841,
      "step": 3238
    },
    {
      "epoch": 0.7427195597340059,
      "grad_norm": 0.15945284187265538,
      "learning_rate": 0.0001637822380412584,
      "loss": 3.4916,
      "step": 3239
    },
    {
      "epoch": 0.7429488649392341,
      "grad_norm": 0.14871946100463757,
      "learning_rate": 0.00016350747645357266,
      "loss": 3.5537,
      "step": 3240
    },
    {
      "epoch": 0.7431781701444623,
      "grad_norm": 0.14635700149026218,
      "learning_rate": 0.00016323290047291195,
      "loss": 3.5581,
      "step": 3241
    },
    {
      "epoch": 0.7434074753496904,
      "grad_norm": 0.1600293110583089,
      "learning_rate": 0.0001629585102507304,
      "loss": 3.3286,
      "step": 3242
    },
    {
      "epoch": 0.7436367805549186,
      "grad_norm": 0.15214771700613125,
      "learning_rate": 0.0001626843059383803,
      "loss": 3.5594,
      "step": 3243
    },
    {
      "epoch": 0.7438660857601468,
      "grad_norm": 0.14917179986993595,
      "learning_rate": 0.0001624102876871103,
      "loss": 3.4538,
      "step": 3244
    },
    {
      "epoch": 0.7440953909653749,
      "grad_norm": 0.15006809620073205,
      "learning_rate": 0.00016213645564806752,
      "loss": 3.4627,
      "step": 3245
    },
    {
      "epoch": 0.744324696170603,
      "grad_norm": 0.18020567887910247,
      "learning_rate": 0.0001618628099722957,
      "loss": 3.5385,
      "step": 3246
    },
    {
      "epoch": 0.7445540013758313,
      "grad_norm": 0.15282936637217215,
      "learning_rate": 0.0001615893508107359,
      "loss": 3.4205,
      "step": 3247
    },
    {
      "epoch": 0.7447833065810594,
      "grad_norm": 0.16684224234972997,
      "learning_rate": 0.00016131607831422633,
      "loss": 3.4236,
      "step": 3248
    },
    {
      "epoch": 0.7450126117862875,
      "grad_norm": 0.15692075196840316,
      "learning_rate": 0.00016104299263350252,
      "loss": 3.3734,
      "step": 3249
    },
    {
      "epoch": 0.7452419169915158,
      "grad_norm": 0.16312406538962937,
      "learning_rate": 0.0001607700939191966,
      "loss": 3.3083,
      "step": 3250
    },
    {
      "epoch": 0.7454712221967439,
      "grad_norm": 0.154659624480959,
      "learning_rate": 0.00016049738232183758,
      "loss": 3.5095,
      "step": 3251
    },
    {
      "epoch": 0.745700527401972,
      "grad_norm": 0.16685498366036242,
      "learning_rate": 0.00016022485799185128,
      "loss": 3.5088,
      "step": 3252
    },
    {
      "epoch": 0.7459298326072001,
      "grad_norm": 0.1630076859044962,
      "learning_rate": 0.0001599525210795606,
      "loss": 3.5179,
      "step": 3253
    },
    {
      "epoch": 0.7461591378124284,
      "grad_norm": 0.15983867045537895,
      "learning_rate": 0.0001596803717351845,
      "loss": 3.542,
      "step": 3254
    },
    {
      "epoch": 0.7463884430176565,
      "grad_norm": 0.16199543007222106,
      "learning_rate": 0.00015940841010883889,
      "loss": 3.3032,
      "step": 3255
    },
    {
      "epoch": 0.7466177482228846,
      "grad_norm": 0.17636474232399443,
      "learning_rate": 0.00015913663635053576,
      "loss": 3.5073,
      "step": 3256
    },
    {
      "epoch": 0.7468470534281129,
      "grad_norm": 0.16616606933091493,
      "learning_rate": 0.00015886505061018413,
      "loss": 3.4448,
      "step": 3257
    },
    {
      "epoch": 0.747076358633341,
      "grad_norm": 0.1764752580643292,
      "learning_rate": 0.00015859365303758834,
      "loss": 3.4594,
      "step": 3258
    },
    {
      "epoch": 0.7473056638385691,
      "grad_norm": 0.17745899344169766,
      "learning_rate": 0.0001583224437824498,
      "loss": 3.482,
      "step": 3259
    },
    {
      "epoch": 0.7475349690437973,
      "grad_norm": 0.1636637997338116,
      "learning_rate": 0.0001580514229943657,
      "loss": 3.3879,
      "step": 3260
    },
    {
      "epoch": 0.7477642742490255,
      "grad_norm": 0.1711920753863977,
      "learning_rate": 0.0001577805908228293,
      "loss": 3.6319,
      "step": 3261
    },
    {
      "epoch": 0.7479935794542536,
      "grad_norm": 0.1609691036381285,
      "learning_rate": 0.0001575099474172295,
      "loss": 3.4021,
      "step": 3262
    },
    {
      "epoch": 0.7482228846594817,
      "grad_norm": 0.1733300852473039,
      "learning_rate": 0.00015723949292685191,
      "loss": 3.3781,
      "step": 3263
    },
    {
      "epoch": 0.74845218986471,
      "grad_norm": 0.16390906685219592,
      "learning_rate": 0.00015696922750087715,
      "loss": 3.4106,
      "step": 3264
    },
    {
      "epoch": 0.7486814950699381,
      "grad_norm": 0.17579369004976242,
      "learning_rate": 0.0001566991512883818,
      "loss": 3.3596,
      "step": 3265
    },
    {
      "epoch": 0.7489108002751662,
      "grad_norm": 0.15287843486697592,
      "learning_rate": 0.00015642926443833798,
      "loss": 3.3942,
      "step": 3266
    },
    {
      "epoch": 0.7491401054803944,
      "grad_norm": 0.16255719813446637,
      "learning_rate": 0.00015615956709961378,
      "loss": 3.52,
      "step": 3267
    },
    {
      "epoch": 0.7493694106856226,
      "grad_norm": 0.17034139652464178,
      "learning_rate": 0.0001558900594209723,
      "loss": 3.4119,
      "step": 3268
    },
    {
      "epoch": 0.7495987158908507,
      "grad_norm": 0.18473228843468567,
      "learning_rate": 0.00015562074155107215,
      "loss": 3.5421,
      "step": 3269
    },
    {
      "epoch": 0.7498280210960789,
      "grad_norm": 0.16777959849082877,
      "learning_rate": 0.00015535161363846705,
      "loss": 3.4534,
      "step": 3270
    },
    {
      "epoch": 0.7500573263013071,
      "grad_norm": 0.18467393688455191,
      "learning_rate": 0.0001550826758316068,
      "loss": 3.414,
      "step": 3271
    },
    {
      "epoch": 0.7502866315065352,
      "grad_norm": 0.19082051543799058,
      "learning_rate": 0.00015481392827883489,
      "loss": 3.6147,
      "step": 3272
    },
    {
      "epoch": 0.7505159367117633,
      "grad_norm": 0.1558835993026194,
      "learning_rate": 0.00015454537112839122,
      "loss": 3.2981,
      "step": 3273
    },
    {
      "epoch": 0.7507452419169915,
      "grad_norm": 0.16602865003284623,
      "learning_rate": 0.00015427700452840998,
      "loss": 3.4885,
      "step": 3274
    },
    {
      "epoch": 0.7509745471222197,
      "grad_norm": 0.15997367899915374,
      "learning_rate": 0.00015400882862692033,
      "loss": 3.5132,
      "step": 3275
    },
    {
      "epoch": 0.7512038523274478,
      "grad_norm": 0.16040140177377324,
      "learning_rate": 0.0001537408435718462,
      "loss": 3.3607,
      "step": 3276
    },
    {
      "epoch": 0.751433157532676,
      "grad_norm": 0.1699330952907693,
      "learning_rate": 0.00015347304951100665,
      "loss": 3.3289,
      "step": 3277
    },
    {
      "epoch": 0.7516624627379042,
      "grad_norm": 0.1587329399195259,
      "learning_rate": 0.000153205446592115,
      "loss": 3.4436,
      "step": 3278
    },
    {
      "epoch": 0.7518917679431323,
      "grad_norm": 0.15474619208155269,
      "learning_rate": 0.00015293803496277907,
      "loss": 3.3819,
      "step": 3279
    },
    {
      "epoch": 0.7521210731483605,
      "grad_norm": 0.1696243610033937,
      "learning_rate": 0.00015267081477050133,
      "loss": 3.587,
      "step": 3280
    },
    {
      "epoch": 0.7523503783535886,
      "grad_norm": 0.16234955371911278,
      "learning_rate": 0.00015240378616267886,
      "loss": 3.5044,
      "step": 3281
    },
    {
      "epoch": 0.7525796835588168,
      "grad_norm": 0.15378689169140747,
      "learning_rate": 0.00015213694928660276,
      "loss": 3.4042,
      "step": 3282
    },
    {
      "epoch": 0.7528089887640449,
      "grad_norm": 0.15213889404210548,
      "learning_rate": 0.00015187030428945843,
      "loss": 3.3235,
      "step": 3283
    },
    {
      "epoch": 0.7530382939692731,
      "grad_norm": 0.1586010293423822,
      "learning_rate": 0.00015160385131832532,
      "loss": 3.4021,
      "step": 3284
    },
    {
      "epoch": 0.7532675991745013,
      "grad_norm": 0.1694337641339962,
      "learning_rate": 0.0001513375905201776,
      "loss": 3.5968,
      "step": 3285
    },
    {
      "epoch": 0.7534969043797294,
      "grad_norm": 0.15014880869387393,
      "learning_rate": 0.0001510715220418823,
      "loss": 3.4441,
      "step": 3286
    },
    {
      "epoch": 0.7537262095849576,
      "grad_norm": 0.16231630617288373,
      "learning_rate": 0.00015080564603020142,
      "loss": 3.526,
      "step": 3287
    },
    {
      "epoch": 0.7539555147901857,
      "grad_norm": 0.15740630714594828,
      "learning_rate": 0.00015053996263179032,
      "loss": 3.5103,
      "step": 3288
    },
    {
      "epoch": 0.7541848199954139,
      "grad_norm": 0.1652981715225447,
      "learning_rate": 0.0001502744719931982,
      "loss": 3.4008,
      "step": 3289
    },
    {
      "epoch": 0.7544141252006421,
      "grad_norm": 0.17353491448989172,
      "learning_rate": 0.0001500091742608677,
      "loss": 3.4766,
      "step": 3290
    },
    {
      "epoch": 0.7546434304058702,
      "grad_norm": 0.1894007731934565,
      "learning_rate": 0.00014974406958113558,
      "loss": 3.484,
      "step": 3291
    },
    {
      "epoch": 0.7548727356110984,
      "grad_norm": 0.18470666780671505,
      "learning_rate": 0.00014947915810023171,
      "loss": 3.4743,
      "step": 3292
    },
    {
      "epoch": 0.7551020408163265,
      "grad_norm": 0.1557198885697612,
      "learning_rate": 0.00014921443996427947,
      "loss": 3.3435,
      "step": 3293
    },
    {
      "epoch": 0.7553313460215547,
      "grad_norm": 0.18366031107201553,
      "learning_rate": 0.0001489499153192954,
      "loss": 3.4902,
      "step": 3294
    },
    {
      "epoch": 0.7555606512267828,
      "grad_norm": 0.17326652408891352,
      "learning_rate": 0.0001486855843111901,
      "loss": 3.4658,
      "step": 3295
    },
    {
      "epoch": 0.755789956432011,
      "grad_norm": 0.16151112094740314,
      "learning_rate": 0.00014842144708576604,
      "loss": 3.4132,
      "step": 3296
    },
    {
      "epoch": 0.7560192616372392,
      "grad_norm": 0.16687739444712288,
      "learning_rate": 0.0001481575037887201,
      "loss": 3.4521,
      "step": 3297
    },
    {
      "epoch": 0.7562485668424673,
      "grad_norm": 0.16518004131379374,
      "learning_rate": 0.00014789375456564135,
      "loss": 3.5227,
      "step": 3298
    },
    {
      "epoch": 0.7564778720476955,
      "grad_norm": 0.16096560034958055,
      "learning_rate": 0.00014763019956201253,
      "loss": 3.517,
      "step": 3299
    },
    {
      "epoch": 0.7567071772529237,
      "grad_norm": 0.15260760401100773,
      "learning_rate": 0.00014736683892320812,
      "loss": 3.3804,
      "step": 3300
    },
    {
      "epoch": 0.7569364824581518,
      "grad_norm": 0.15361737917968202,
      "learning_rate": 0.0001471036727944966,
      "loss": 3.5169,
      "step": 3301
    },
    {
      "epoch": 0.7571657876633799,
      "grad_norm": 0.17156096459573722,
      "learning_rate": 0.00014684070132103854,
      "loss": 3.4348,
      "step": 3302
    },
    {
      "epoch": 0.7573950928686081,
      "grad_norm": 0.16693818410950897,
      "learning_rate": 0.0001465779246478872,
      "loss": 3.4225,
      "step": 3303
    },
    {
      "epoch": 0.7576243980738363,
      "grad_norm": 0.1532666787520267,
      "learning_rate": 0.00014631534291998833,
      "loss": 3.4426,
      "step": 3304
    },
    {
      "epoch": 0.7578537032790644,
      "grad_norm": 0.16651553986921905,
      "learning_rate": 0.00014605295628218045,
      "loss": 3.4612,
      "step": 3305
    },
    {
      "epoch": 0.7580830084842926,
      "grad_norm": 0.1585069203418513,
      "learning_rate": 0.0001457907648791943,
      "loss": 3.4139,
      "step": 3306
    },
    {
      "epoch": 0.7583123136895208,
      "grad_norm": 0.1664342284783134,
      "learning_rate": 0.0001455287688556527,
      "loss": 3.3708,
      "step": 3307
    },
    {
      "epoch": 0.7585416188947489,
      "grad_norm": 0.14513907333537776,
      "learning_rate": 0.00014526696835607088,
      "loss": 3.3973,
      "step": 3308
    },
    {
      "epoch": 0.758770924099977,
      "grad_norm": 0.17637876745715442,
      "learning_rate": 0.00014500536352485673,
      "loss": 3.4572,
      "step": 3309
    },
    {
      "epoch": 0.7590002293052053,
      "grad_norm": 0.16008923144642442,
      "learning_rate": 0.00014474395450630895,
      "loss": 3.4459,
      "step": 3310
    },
    {
      "epoch": 0.7592295345104334,
      "grad_norm": 0.14504401178948403,
      "learning_rate": 0.00014448274144461965,
      "loss": 3.3868,
      "step": 3311
    },
    {
      "epoch": 0.7594588397156615,
      "grad_norm": 0.1634517207947283,
      "learning_rate": 0.00014422172448387178,
      "loss": 3.4517,
      "step": 3312
    },
    {
      "epoch": 0.7596881449208897,
      "grad_norm": 0.1479684024476466,
      "learning_rate": 0.00014396090376804112,
      "loss": 3.4261,
      "step": 3313
    },
    {
      "epoch": 0.7599174501261179,
      "grad_norm": 0.17479967237857916,
      "learning_rate": 0.0001437002794409939,
      "loss": 3.6166,
      "step": 3314
    },
    {
      "epoch": 0.760146755331346,
      "grad_norm": 0.16714552939842736,
      "learning_rate": 0.00014343985164648926,
      "loss": 3.2987,
      "step": 3315
    },
    {
      "epoch": 0.7603760605365741,
      "grad_norm": 0.1658265095377136,
      "learning_rate": 0.0001431796205281773,
      "loss": 3.427,
      "step": 3316
    },
    {
      "epoch": 0.7606053657418024,
      "grad_norm": 0.15843503977579393,
      "learning_rate": 0.00014291958622959973,
      "loss": 3.3562,
      "step": 3317
    },
    {
      "epoch": 0.7608346709470305,
      "grad_norm": 0.17081648803186045,
      "learning_rate": 0.00014265974889418959,
      "loss": 3.3203,
      "step": 3318
    },
    {
      "epoch": 0.7610639761522586,
      "grad_norm": 0.14962554136263753,
      "learning_rate": 0.00014240010866527176,
      "loss": 3.399,
      "step": 3319
    },
    {
      "epoch": 0.7612932813574869,
      "grad_norm": 0.16065388544744785,
      "learning_rate": 0.00014214066568606182,
      "loss": 3.4086,
      "step": 3320
    },
    {
      "epoch": 0.761522586562715,
      "grad_norm": 0.16916096659240606,
      "learning_rate": 0.00014188142009966686,
      "loss": 3.386,
      "step": 3321
    },
    {
      "epoch": 0.7617518917679431,
      "grad_norm": 0.15932792549356128,
      "learning_rate": 0.00014162237204908478,
      "loss": 3.4863,
      "step": 3322
    },
    {
      "epoch": 0.7619811969731712,
      "grad_norm": 0.1525818143744201,
      "learning_rate": 0.0001413635216772053,
      "loss": 3.4508,
      "step": 3323
    },
    {
      "epoch": 0.7622105021783995,
      "grad_norm": 0.16362291918822838,
      "learning_rate": 0.00014110486912680793,
      "loss": 3.3763,
      "step": 3324
    },
    {
      "epoch": 0.7624398073836276,
      "grad_norm": 0.152362551772482,
      "learning_rate": 0.000140846414540564,
      "loss": 3.3252,
      "step": 3325
    },
    {
      "epoch": 0.7626691125888557,
      "grad_norm": 0.15246739468835555,
      "learning_rate": 0.00014058815806103543,
      "loss": 3.398,
      "step": 3326
    },
    {
      "epoch": 0.762898417794084,
      "grad_norm": 0.1678697483083912,
      "learning_rate": 0.00014033009983067452,
      "loss": 3.3954,
      "step": 3327
    },
    {
      "epoch": 0.7631277229993121,
      "grad_norm": 0.17197600460695572,
      "learning_rate": 0.00014007223999182444,
      "loss": 3.4945,
      "step": 3328
    },
    {
      "epoch": 0.7633570282045402,
      "grad_norm": 0.17082757384884603,
      "learning_rate": 0.00013981457868671927,
      "loss": 3.5153,
      "step": 3329
    },
    {
      "epoch": 0.7635863334097684,
      "grad_norm": 0.1645359719910603,
      "learning_rate": 0.00013955711605748305,
      "loss": 3.3305,
      "step": 3330
    },
    {
      "epoch": 0.7638156386149966,
      "grad_norm": 0.1793193756080003,
      "learning_rate": 0.0001392998522461305,
      "loss": 3.5251,
      "step": 3331
    },
    {
      "epoch": 0.7640449438202247,
      "grad_norm": 0.17571013171593514,
      "learning_rate": 0.00013904278739456643,
      "loss": 3.7173,
      "step": 3332
    },
    {
      "epoch": 0.7642742490254528,
      "grad_norm": 0.16892913846281338,
      "learning_rate": 0.00013878592164458635,
      "loss": 3.3855,
      "step": 3333
    },
    {
      "epoch": 0.7645035542306811,
      "grad_norm": 0.15006348791717994,
      "learning_rate": 0.00013852925513787567,
      "loss": 3.3316,
      "step": 3334
    },
    {
      "epoch": 0.7647328594359092,
      "grad_norm": 0.1591342347008813,
      "learning_rate": 0.00013827278801600978,
      "loss": 3.3778,
      "step": 3335
    },
    {
      "epoch": 0.7649621646411373,
      "grad_norm": 0.15510621801852995,
      "learning_rate": 0.00013801652042045416,
      "loss": 3.5337,
      "step": 3336
    },
    {
      "epoch": 0.7651914698463655,
      "grad_norm": 0.15550259995469623,
      "learning_rate": 0.0001377604524925647,
      "loss": 3.5105,
      "step": 3337
    },
    {
      "epoch": 0.7654207750515937,
      "grad_norm": 0.1594999960970026,
      "learning_rate": 0.0001375045843735862,
      "loss": 3.4859,
      "step": 3338
    },
    {
      "epoch": 0.7656500802568218,
      "grad_norm": 0.15830170543556393,
      "learning_rate": 0.00013724891620465424,
      "loss": 3.4733,
      "step": 3339
    },
    {
      "epoch": 0.76587938546205,
      "grad_norm": 0.14841320240012926,
      "learning_rate": 0.00013699344812679353,
      "loss": 3.3542,
      "step": 3340
    },
    {
      "epoch": 0.7661086906672782,
      "grad_norm": 0.17627478871547833,
      "learning_rate": 0.0001367381802809185,
      "loss": 3.4091,
      "step": 3341
    },
    {
      "epoch": 0.7663379958725063,
      "grad_norm": 0.1665837829118598,
      "learning_rate": 0.0001364831128078331,
      "loss": 3.259,
      "step": 3342
    },
    {
      "epoch": 0.7665673010777345,
      "grad_norm": 0.1466247574296966,
      "learning_rate": 0.00013622824584823113,
      "loss": 3.4164,
      "step": 3343
    },
    {
      "epoch": 0.7667966062829626,
      "grad_norm": 0.1496609306645941,
      "learning_rate": 0.00013597357954269534,
      "loss": 3.4827,
      "step": 3344
    },
    {
      "epoch": 0.7670259114881908,
      "grad_norm": 0.15779616663508714,
      "learning_rate": 0.00013571911403169795,
      "loss": 3.5201,
      "step": 3345
    },
    {
      "epoch": 0.7672552166934189,
      "grad_norm": 0.17007307885893178,
      "learning_rate": 0.00013546484945560028,
      "loss": 3.468,
      "step": 3346
    },
    {
      "epoch": 0.7674845218986471,
      "grad_norm": 0.15098727819886723,
      "learning_rate": 0.0001352107859546533,
      "loss": 3.392,
      "step": 3347
    },
    {
      "epoch": 0.7677138271038753,
      "grad_norm": 0.16608461005430641,
      "learning_rate": 0.00013495692366899658,
      "loss": 3.4168,
      "step": 3348
    },
    {
      "epoch": 0.7679431323091034,
      "grad_norm": 0.1660900117579587,
      "learning_rate": 0.00013470326273865886,
      "loss": 3.5092,
      "step": 3349
    },
    {
      "epoch": 0.7681724375143316,
      "grad_norm": 0.1600323219576918,
      "learning_rate": 0.0001344498033035576,
      "loss": 3.4295,
      "step": 3350
    },
    {
      "epoch": 0.7684017427195597,
      "grad_norm": 0.1648056221050302,
      "learning_rate": 0.00013419654550349985,
      "loss": 3.5542,
      "step": 3351
    },
    {
      "epoch": 0.7686310479247879,
      "grad_norm": 0.1654895595588831,
      "learning_rate": 0.0001339434894781803,
      "loss": 3.5126,
      "step": 3352
    },
    {
      "epoch": 0.7688603531300161,
      "grad_norm": 0.16460270961105355,
      "learning_rate": 0.00013369063536718346,
      "loss": 3.4823,
      "step": 3353
    },
    {
      "epoch": 0.7690896583352442,
      "grad_norm": 0.16613260798871796,
      "learning_rate": 0.0001334379833099817,
      "loss": 3.3595,
      "step": 3354
    },
    {
      "epoch": 0.7693189635404724,
      "grad_norm": 0.16241328785982817,
      "learning_rate": 0.00013318553344593632,
      "loss": 3.3671,
      "step": 3355
    },
    {
      "epoch": 0.7695482687457005,
      "grad_norm": 0.16437702522367909,
      "learning_rate": 0.0001329332859142967,
      "loss": 3.4962,
      "step": 3356
    },
    {
      "epoch": 0.7697775739509287,
      "grad_norm": 0.158383187277195,
      "learning_rate": 0.00013268124085420136,
      "loss": 3.5075,
      "step": 3357
    },
    {
      "epoch": 0.7700068791561568,
      "grad_norm": 0.16918439385291847,
      "learning_rate": 0.00013242939840467637,
      "loss": 3.4638,
      "step": 3358
    },
    {
      "epoch": 0.770236184361385,
      "grad_norm": 0.16325621495399023,
      "learning_rate": 0.0001321777587046364,
      "loss": 3.4225,
      "step": 3359
    },
    {
      "epoch": 0.7704654895666132,
      "grad_norm": 0.16544654530746458,
      "learning_rate": 0.00013192632189288405,
      "loss": 3.551,
      "step": 3360
    },
    {
      "epoch": 0.7706947947718413,
      "grad_norm": 0.1726383978092923,
      "learning_rate": 0.00013167508810811059,
      "loss": 3.5161,
      "step": 3361
    },
    {
      "epoch": 0.7709240999770695,
      "grad_norm": 0.16186581758992405,
      "learning_rate": 0.00013142405748889458,
      "loss": 3.3517,
      "step": 3362
    },
    {
      "epoch": 0.7711534051822977,
      "grad_norm": 0.17580048508760004,
      "learning_rate": 0.0001311732301737029,
      "loss": 3.5705,
      "step": 3363
    },
    {
      "epoch": 0.7713827103875258,
      "grad_norm": 0.16721892832693563,
      "learning_rate": 0.00013092260630089014,
      "loss": 3.4939,
      "step": 3364
    },
    {
      "epoch": 0.7716120155927539,
      "grad_norm": 0.16446386499355822,
      "learning_rate": 0.0001306721860086991,
      "loss": 3.4858,
      "step": 3365
    },
    {
      "epoch": 0.7718413207979821,
      "grad_norm": 0.16365454872257912,
      "learning_rate": 0.00013042196943525942,
      "loss": 3.3753,
      "step": 3366
    },
    {
      "epoch": 0.7720706260032103,
      "grad_norm": 0.16591186596111257,
      "learning_rate": 0.00013017195671858928,
      "loss": 3.4421,
      "step": 3367
    },
    {
      "epoch": 0.7722999312084384,
      "grad_norm": 0.16501203098404169,
      "learning_rate": 0.0001299221479965939,
      "loss": 3.3498,
      "step": 3368
    },
    {
      "epoch": 0.7725292364136666,
      "grad_norm": 0.17907536833797422,
      "learning_rate": 0.0001296725434070661,
      "loss": 3.495,
      "step": 3369
    },
    {
      "epoch": 0.7727585416188948,
      "grad_norm": 0.16265857941118506,
      "learning_rate": 0.00012942314308768594,
      "loss": 3.5198,
      "step": 3370
    },
    {
      "epoch": 0.7729878468241229,
      "grad_norm": 0.15808854510645323,
      "learning_rate": 0.00012917394717602121,
      "loss": 3.3971,
      "step": 3371
    },
    {
      "epoch": 0.773217152029351,
      "grad_norm": 0.18541055236150167,
      "learning_rate": 0.00012892495580952667,
      "loss": 3.6104,
      "step": 3372
    },
    {
      "epoch": 0.7734464572345793,
      "grad_norm": 0.1634255647912794,
      "learning_rate": 0.00012867616912554426,
      "loss": 3.3938,
      "step": 3373
    },
    {
      "epoch": 0.7736757624398074,
      "grad_norm": 0.16638649780028603,
      "learning_rate": 0.00012842758726130281,
      "loss": 3.3957,
      "step": 3374
    },
    {
      "epoch": 0.7739050676450355,
      "grad_norm": 0.16682332322551138,
      "learning_rate": 0.00012817921035391882,
      "loss": 3.4607,
      "step": 3375
    },
    {
      "epoch": 0.7741343728502637,
      "grad_norm": 0.1645875275893724,
      "learning_rate": 0.00012793103854039518,
      "loss": 3.4911,
      "step": 3376
    },
    {
      "epoch": 0.7743636780554919,
      "grad_norm": 0.16214786741467926,
      "learning_rate": 0.00012768307195762168,
      "loss": 3.4108,
      "step": 3377
    },
    {
      "epoch": 0.77459298326072,
      "grad_norm": 0.1588244681017531,
      "learning_rate": 0.00012743531074237507,
      "loss": 3.5219,
      "step": 3378
    },
    {
      "epoch": 0.7748222884659481,
      "grad_norm": 0.16033892270488775,
      "learning_rate": 0.00012718775503131908,
      "loss": 3.494,
      "step": 3379
    },
    {
      "epoch": 0.7750515936711764,
      "grad_norm": 0.1718148038819317,
      "learning_rate": 0.00012694040496100318,
      "loss": 3.4187,
      "step": 3380
    },
    {
      "epoch": 0.7752808988764045,
      "grad_norm": 0.1812025603142883,
      "learning_rate": 0.0001266932606678646,
      "loss": 3.5744,
      "step": 3381
    },
    {
      "epoch": 0.7755102040816326,
      "grad_norm": 0.1595862381694047,
      "learning_rate": 0.00012644632228822617,
      "loss": 3.3922,
      "step": 3382
    },
    {
      "epoch": 0.7757395092868609,
      "grad_norm": 0.16474646226675607,
      "learning_rate": 0.00012619958995829756,
      "loss": 3.4004,
      "step": 3383
    },
    {
      "epoch": 0.775968814492089,
      "grad_norm": 0.17761488651209914,
      "learning_rate": 0.00012595306381417448,
      "loss": 3.4998,
      "step": 3384
    },
    {
      "epoch": 0.7761981196973171,
      "grad_norm": 0.1754040953957049,
      "learning_rate": 0.0001257067439918394,
      "loss": 3.4272,
      "step": 3385
    },
    {
      "epoch": 0.7764274249025452,
      "grad_norm": 0.1694043479720841,
      "learning_rate": 0.0001254606306271607,
      "loss": 3.4137,
      "step": 3386
    },
    {
      "epoch": 0.7766567301077735,
      "grad_norm": 0.1549591570011813,
      "learning_rate": 0.00012521472385589234,
      "loss": 3.4112,
      "step": 3387
    },
    {
      "epoch": 0.7768860353130016,
      "grad_norm": 0.16142472734013327,
      "learning_rate": 0.00012496902381367543,
      "loss": 3.3438,
      "step": 3388
    },
    {
      "epoch": 0.7771153405182297,
      "grad_norm": 0.1766295253091256,
      "learning_rate": 0.00012472353063603626,
      "loss": 3.4,
      "step": 3389
    },
    {
      "epoch": 0.777344645723458,
      "grad_norm": 0.15332872032471603,
      "learning_rate": 0.00012447824445838718,
      "loss": 3.3967,
      "step": 3390
    },
    {
      "epoch": 0.7775739509286861,
      "grad_norm": 0.16007432935832538,
      "learning_rate": 0.0001242331654160263,
      "loss": 3.4345,
      "step": 3391
    },
    {
      "epoch": 0.7778032561339142,
      "grad_norm": 0.1661758341236528,
      "learning_rate": 0.00012398829364413793,
      "loss": 3.4704,
      "step": 3392
    },
    {
      "epoch": 0.7780325613391424,
      "grad_norm": 0.1539216726633029,
      "learning_rate": 0.0001237436292777914,
      "loss": 3.5253,
      "step": 3393
    },
    {
      "epoch": 0.7782618665443706,
      "grad_norm": 0.16916742170585847,
      "learning_rate": 0.00012349917245194214,
      "loss": 3.4625,
      "step": 3394
    },
    {
      "epoch": 0.7784911717495987,
      "grad_norm": 0.17027645091500576,
      "learning_rate": 0.00012325492330143061,
      "loss": 3.4249,
      "step": 3395
    },
    {
      "epoch": 0.7787204769548268,
      "grad_norm": 0.18072443022101534,
      "learning_rate": 0.0001230108819609833,
      "loss": 3.4426,
      "step": 3396
    },
    {
      "epoch": 0.7789497821600551,
      "grad_norm": 0.15323035966632273,
      "learning_rate": 0.00012276704856521175,
      "loss": 3.3635,
      "step": 3397
    },
    {
      "epoch": 0.7791790873652832,
      "grad_norm": 0.1709534157174997,
      "learning_rate": 0.0001225234232486127,
      "loss": 3.607,
      "step": 3398
    },
    {
      "epoch": 0.7794083925705113,
      "grad_norm": 0.17275306114869624,
      "learning_rate": 0.00012228000614556816,
      "loss": 3.4551,
      "step": 3399
    },
    {
      "epoch": 0.7796376977757395,
      "grad_norm": 0.16665206083592357,
      "learning_rate": 0.00012203679739034579,
      "loss": 3.4046,
      "step": 3400
    },
    {
      "epoch": 0.7798670029809677,
      "grad_norm": 0.1659596902818201,
      "learning_rate": 0.00012179379711709738,
      "loss": 3.4054,
      "step": 3401
    },
    {
      "epoch": 0.7800963081861958,
      "grad_norm": 0.16233944705990674,
      "learning_rate": 0.00012155100545986065,
      "loss": 3.4539,
      "step": 3402
    },
    {
      "epoch": 0.780325613391424,
      "grad_norm": 0.18012869217323937,
      "learning_rate": 0.0001213084225525577,
      "loss": 3.5179,
      "step": 3403
    },
    {
      "epoch": 0.7805549185966522,
      "grad_norm": 0.16492711971978405,
      "learning_rate": 0.00012106604852899566,
      "loss": 3.4435,
      "step": 3404
    },
    {
      "epoch": 0.7807842238018803,
      "grad_norm": 0.16360269767235405,
      "learning_rate": 0.00012082388352286627,
      "loss": 3.4985,
      "step": 3405
    },
    {
      "epoch": 0.7810135290071084,
      "grad_norm": 0.1576216768191429,
      "learning_rate": 0.00012058192766774639,
      "loss": 3.6072,
      "step": 3406
    },
    {
      "epoch": 0.7812428342123366,
      "grad_norm": 0.17317221811968855,
      "learning_rate": 0.00012034018109709716,
      "loss": 3.6188,
      "step": 3407
    },
    {
      "epoch": 0.7814721394175648,
      "grad_norm": 0.1666603718829379,
      "learning_rate": 0.00012009864394426434,
      "loss": 3.4978,
      "step": 3408
    },
    {
      "epoch": 0.7817014446227929,
      "grad_norm": 0.15122232431478408,
      "learning_rate": 0.00011985731634247809,
      "loss": 3.3714,
      "step": 3409
    },
    {
      "epoch": 0.7819307498280211,
      "grad_norm": 0.16699551641681146,
      "learning_rate": 0.00011961619842485333,
      "loss": 3.3348,
      "step": 3410
    },
    {
      "epoch": 0.7821600550332493,
      "grad_norm": 0.1775369108546889,
      "learning_rate": 0.00011937529032438904,
      "loss": 3.4744,
      "step": 3411
    },
    {
      "epoch": 0.7823893602384774,
      "grad_norm": 0.1541581645255735,
      "learning_rate": 0.00011913459217396843,
      "loss": 3.2848,
      "step": 3412
    },
    {
      "epoch": 0.7826186654437056,
      "grad_norm": 0.17074444813178014,
      "learning_rate": 0.00011889410410635887,
      "loss": 3.3801,
      "step": 3413
    },
    {
      "epoch": 0.7828479706489337,
      "grad_norm": 0.1731638272842968,
      "learning_rate": 0.00011865382625421239,
      "loss": 3.3943,
      "step": 3414
    },
    {
      "epoch": 0.7830772758541619,
      "grad_norm": 0.15308277010327836,
      "learning_rate": 0.0001184137587500641,
      "loss": 3.3914,
      "step": 3415
    },
    {
      "epoch": 0.78330658105939,
      "grad_norm": 0.1820684365212057,
      "learning_rate": 0.00011817390172633402,
      "loss": 3.4207,
      "step": 3416
    },
    {
      "epoch": 0.7835358862646182,
      "grad_norm": 0.16472160260172153,
      "learning_rate": 0.00011793425531532564,
      "loss": 3.3351,
      "step": 3417
    },
    {
      "epoch": 0.7837651914698464,
      "grad_norm": 0.1468640502325057,
      "learning_rate": 0.00011769481964922618,
      "loss": 3.3782,
      "step": 3418
    },
    {
      "epoch": 0.7839944966750745,
      "grad_norm": 0.16943788128045348,
      "learning_rate": 0.00011745559486010671,
      "loss": 3.4496,
      "step": 3419
    },
    {
      "epoch": 0.7842238018803027,
      "grad_norm": 0.166873696836669,
      "learning_rate": 0.00011721658107992223,
      "loss": 3.4081,
      "step": 3420
    },
    {
      "epoch": 0.7844531070855308,
      "grad_norm": 0.17052200084739635,
      "learning_rate": 0.00011697777844051105,
      "loss": 3.433,
      "step": 3421
    },
    {
      "epoch": 0.784682412290759,
      "grad_norm": 0.1587820490794032,
      "learning_rate": 0.0001167391870735951,
      "loss": 3.3832,
      "step": 3422
    },
    {
      "epoch": 0.7849117174959872,
      "grad_norm": 0.17183639601199013,
      "learning_rate": 0.00011650080711077964,
      "loss": 3.4824,
      "step": 3423
    },
    {
      "epoch": 0.7851410227012153,
      "grad_norm": 0.1797600625134874,
      "learning_rate": 0.00011626263868355375,
      "loss": 3.3878,
      "step": 3424
    },
    {
      "epoch": 0.7853703279064435,
      "grad_norm": 0.15041702455728062,
      "learning_rate": 0.00011602468192328936,
      "loss": 3.5378,
      "step": 3425
    },
    {
      "epoch": 0.7855996331116716,
      "grad_norm": 0.15997984829034026,
      "learning_rate": 0.00011578693696124193,
      "loss": 3.4643,
      "step": 3426
    },
    {
      "epoch": 0.7858289383168998,
      "grad_norm": 0.15447136093409594,
      "learning_rate": 0.00011554940392854973,
      "loss": 3.4706,
      "step": 3427
    },
    {
      "epoch": 0.7860582435221279,
      "grad_norm": 0.16435230252033362,
      "learning_rate": 0.00011531208295623485,
      "loss": 3.3835,
      "step": 3428
    },
    {
      "epoch": 0.7862875487273561,
      "grad_norm": 0.15825350836728347,
      "learning_rate": 0.00011507497417520146,
      "loss": 3.3566,
      "step": 3429
    },
    {
      "epoch": 0.7865168539325843,
      "grad_norm": 0.1586515057751094,
      "learning_rate": 0.0001148380777162375,
      "loss": 3.5429,
      "step": 3430
    },
    {
      "epoch": 0.7867461591378124,
      "grad_norm": 0.164673626323467,
      "learning_rate": 0.00011460139371001339,
      "loss": 3.4313,
      "step": 3431
    },
    {
      "epoch": 0.7869754643430406,
      "grad_norm": 0.1647407888477585,
      "learning_rate": 0.00011436492228708239,
      "loss": 3.3268,
      "step": 3432
    },
    {
      "epoch": 0.7872047695482688,
      "grad_norm": 0.16220284667566034,
      "learning_rate": 0.00011412866357788049,
      "loss": 3.406,
      "step": 3433
    },
    {
      "epoch": 0.7874340747534969,
      "grad_norm": 0.15926894351547521,
      "learning_rate": 0.00011389261771272663,
      "loss": 3.4851,
      "step": 3434
    },
    {
      "epoch": 0.787663379958725,
      "grad_norm": 0.1743739450220488,
      "learning_rate": 0.00011365678482182207,
      "loss": 3.3189,
      "step": 3435
    },
    {
      "epoch": 0.7878926851639533,
      "grad_norm": 0.16205436427881775,
      "learning_rate": 0.00011342116503525057,
      "loss": 3.3354,
      "step": 3436
    },
    {
      "epoch": 0.7881219903691814,
      "grad_norm": 0.1762258233616548,
      "learning_rate": 0.0001131857584829783,
      "loss": 3.3344,
      "step": 3437
    },
    {
      "epoch": 0.7883512955744095,
      "grad_norm": 0.15917554712363144,
      "learning_rate": 0.00011295056529485426,
      "loss": 3.6379,
      "step": 3438
    },
    {
      "epoch": 0.7885806007796377,
      "grad_norm": 0.17850157758384266,
      "learning_rate": 0.0001127155856006093,
      "loss": 3.5968,
      "step": 3439
    },
    {
      "epoch": 0.7888099059848659,
      "grad_norm": 0.16540722941087432,
      "learning_rate": 0.00011248081952985656,
      "loss": 3.5149,
      "step": 3440
    },
    {
      "epoch": 0.789039211190094,
      "grad_norm": 0.16054789925835045,
      "learning_rate": 0.00011224626721209141,
      "loss": 3.4214,
      "step": 3441
    },
    {
      "epoch": 0.7892685163953221,
      "grad_norm": 0.1709018334796118,
      "learning_rate": 0.0001120119287766917,
      "loss": 3.4176,
      "step": 3442
    },
    {
      "epoch": 0.7894978216005504,
      "grad_norm": 0.1634958097286702,
      "learning_rate": 0.0001117778043529164,
      "loss": 3.5389,
      "step": 3443
    },
    {
      "epoch": 0.7897271268057785,
      "grad_norm": 0.16825417900254533,
      "learning_rate": 0.0001115438940699074,
      "loss": 3.6648,
      "step": 3444
    },
    {
      "epoch": 0.7899564320110066,
      "grad_norm": 0.16376572111458704,
      "learning_rate": 0.0001113101980566879,
      "loss": 3.4182,
      "step": 3445
    },
    {
      "epoch": 0.7901857372162349,
      "grad_norm": 0.16671011044098816,
      "learning_rate": 0.00011107671644216305,
      "loss": 3.4318,
      "step": 3446
    },
    {
      "epoch": 0.790415042421463,
      "grad_norm": 0.16064382436760632,
      "learning_rate": 0.00011084344935511958,
      "loss": 3.2731,
      "step": 3447
    },
    {
      "epoch": 0.7906443476266911,
      "grad_norm": 0.1662751525876473,
      "learning_rate": 0.00011061039692422637,
      "loss": 3.3156,
      "step": 3448
    },
    {
      "epoch": 0.7908736528319192,
      "grad_norm": 0.16477917616527776,
      "learning_rate": 0.00011037755927803345,
      "loss": 3.5702,
      "step": 3449
    },
    {
      "epoch": 0.7911029580371475,
      "grad_norm": 0.1689205525088891,
      "learning_rate": 0.00011014493654497254,
      "loss": 3.451,
      "step": 3450
    },
    {
      "epoch": 0.7913322632423756,
      "grad_norm": 0.16108800968348877,
      "learning_rate": 0.00010991252885335651,
      "loss": 3.431,
      "step": 3451
    },
    {
      "epoch": 0.7915615684476037,
      "grad_norm": 0.16134283514564562,
      "learning_rate": 0.0001096803363313803,
      "loss": 3.453,
      "step": 3452
    },
    {
      "epoch": 0.791790873652832,
      "grad_norm": 0.17480433890026145,
      "learning_rate": 0.00010944835910711958,
      "loss": 3.5181,
      "step": 3453
    },
    {
      "epoch": 0.7920201788580601,
      "grad_norm": 0.15486994944129065,
      "learning_rate": 0.00010921659730853145,
      "loss": 3.3474,
      "step": 3454
    },
    {
      "epoch": 0.7922494840632882,
      "grad_norm": 0.14661393611625698,
      "learning_rate": 0.00010898505106345396,
      "loss": 3.4493,
      "step": 3455
    },
    {
      "epoch": 0.7924787892685164,
      "grad_norm": 0.1805285074566702,
      "learning_rate": 0.00010875372049960696,
      "loss": 3.5512,
      "step": 3456
    },
    {
      "epoch": 0.7927080944737446,
      "grad_norm": 0.15568590128530307,
      "learning_rate": 0.00010852260574459022,
      "loss": 3.4779,
      "step": 3457
    },
    {
      "epoch": 0.7929373996789727,
      "grad_norm": 0.1506969568686167,
      "learning_rate": 0.00010829170692588558,
      "loss": 3.3855,
      "step": 3458
    },
    {
      "epoch": 0.7931667048842008,
      "grad_norm": 0.18245009746493654,
      "learning_rate": 0.00010806102417085512,
      "loss": 3.4875,
      "step": 3459
    },
    {
      "epoch": 0.7933960100894291,
      "grad_norm": 0.15542034679219666,
      "learning_rate": 0.00010783055760674182,
      "loss": 3.475,
      "step": 3460
    },
    {
      "epoch": 0.7936253152946572,
      "grad_norm": 0.1692652795605623,
      "learning_rate": 0.00010760030736066951,
      "loss": 3.4086,
      "step": 3461
    },
    {
      "epoch": 0.7938546204998853,
      "grad_norm": 0.16386927826980519,
      "learning_rate": 0.00010737027355964284,
      "loss": 3.421,
      "step": 3462
    },
    {
      "epoch": 0.7940839257051135,
      "grad_norm": 0.15149496711049526,
      "learning_rate": 0.00010714045633054687,
      "loss": 3.3912,
      "step": 3463
    },
    {
      "epoch": 0.7943132309103417,
      "grad_norm": 0.17500841720050203,
      "learning_rate": 0.00010691085580014714,
      "loss": 3.547,
      "step": 3464
    },
    {
      "epoch": 0.7945425361155698,
      "grad_norm": 0.16830432941838336,
      "learning_rate": 0.00010668147209508971,
      "loss": 3.4693,
      "step": 3465
    },
    {
      "epoch": 0.794771841320798,
      "grad_norm": 0.18276478157080236,
      "learning_rate": 0.00010645230534190148,
      "loss": 3.4104,
      "step": 3466
    },
    {
      "epoch": 0.7950011465260262,
      "grad_norm": 0.16065601287951756,
      "learning_rate": 0.00010622335566698877,
      "loss": 3.3898,
      "step": 3467
    },
    {
      "epoch": 0.7952304517312543,
      "grad_norm": 0.15329693921522228,
      "learning_rate": 0.00010599462319663906,
      "loss": 3.5132,
      "step": 3468
    },
    {
      "epoch": 0.7954597569364824,
      "grad_norm": 0.15774051654042728,
      "learning_rate": 0.00010576610805701942,
      "loss": 3.4403,
      "step": 3469
    },
    {
      "epoch": 0.7956890621417106,
      "grad_norm": 0.18021232371256907,
      "learning_rate": 0.0001055378103741777,
      "loss": 3.4879,
      "step": 3470
    },
    {
      "epoch": 0.7959183673469388,
      "grad_norm": 0.1764237382139878,
      "learning_rate": 0.00010530973027404073,
      "loss": 3.446,
      "step": 3471
    },
    {
      "epoch": 0.7961476725521669,
      "grad_norm": 0.16975231364593488,
      "learning_rate": 0.00010508186788241647,
      "loss": 3.4432,
      "step": 3472
    },
    {
      "epoch": 0.7963769777573951,
      "grad_norm": 0.16011145968988763,
      "learning_rate": 0.00010485422332499212,
      "loss": 3.5371,
      "step": 3473
    },
    {
      "epoch": 0.7966062829626233,
      "grad_norm": 0.16803325946113407,
      "learning_rate": 0.00010462679672733489,
      "loss": 3.299,
      "step": 3474
    },
    {
      "epoch": 0.7968355881678514,
      "grad_norm": 0.16396919409461186,
      "learning_rate": 0.00010439958821489165,
      "loss": 3.4396,
      "step": 3475
    },
    {
      "epoch": 0.7970648933730796,
      "grad_norm": 0.15610637037514233,
      "learning_rate": 0.00010417259791298939,
      "loss": 3.4324,
      "step": 3476
    },
    {
      "epoch": 0.7972941985783077,
      "grad_norm": 0.15582117174431231,
      "learning_rate": 0.00010394582594683428,
      "loss": 3.4111,
      "step": 3477
    },
    {
      "epoch": 0.7975235037835359,
      "grad_norm": 0.1774807808578746,
      "learning_rate": 0.0001037192724415122,
      "loss": 3.3659,
      "step": 3478
    },
    {
      "epoch": 0.797752808988764,
      "grad_norm": 0.1570420073334353,
      "learning_rate": 0.0001034929375219884,
      "loss": 3.2574,
      "step": 3479
    },
    {
      "epoch": 0.7979821141939922,
      "grad_norm": 0.17009496500687227,
      "learning_rate": 0.00010326682131310816,
      "loss": 3.3523,
      "step": 3480
    },
    {
      "epoch": 0.7982114193992204,
      "grad_norm": 0.16110577846815935,
      "learning_rate": 0.00010304092393959514,
      "loss": 3.4406,
      "step": 3481
    },
    {
      "epoch": 0.7984407246044485,
      "grad_norm": 0.15145041662045602,
      "learning_rate": 0.00010281524552605315,
      "loss": 3.3867,
      "step": 3482
    },
    {
      "epoch": 0.7986700298096767,
      "grad_norm": 0.1700598222263911,
      "learning_rate": 0.00010258978619696468,
      "loss": 3.2773,
      "step": 3483
    },
    {
      "epoch": 0.7988993350149048,
      "grad_norm": 0.16480341582448807,
      "learning_rate": 0.00010236454607669199,
      "loss": 3.3627,
      "step": 3484
    },
    {
      "epoch": 0.799128640220133,
      "grad_norm": 0.20374188380387617,
      "learning_rate": 0.00010213952528947551,
      "loss": 3.5467,
      "step": 3485
    },
    {
      "epoch": 0.7993579454253612,
      "grad_norm": 0.16126191772505796,
      "learning_rate": 0.00010191472395943552,
      "loss": 3.4642,
      "step": 3486
    },
    {
      "epoch": 0.7995872506305893,
      "grad_norm": 0.17274204793517745,
      "learning_rate": 0.00010169014221057089,
      "loss": 3.2706,
      "step": 3487
    },
    {
      "epoch": 0.7998165558358175,
      "grad_norm": 0.15240150894683133,
      "learning_rate": 0.00010146578016675934,
      "loss": 3.3721,
      "step": 3488
    },
    {
      "epoch": 0.8000458610410456,
      "grad_norm": 0.16285203303196613,
      "learning_rate": 0.00010124163795175734,
      "loss": 3.4395,
      "step": 3489
    },
    {
      "epoch": 0.8002751662462738,
      "grad_norm": 0.16798146772709813,
      "learning_rate": 0.0001010177156892006,
      "loss": 3.5074,
      "step": 3490
    },
    {
      "epoch": 0.8005044714515019,
      "grad_norm": 0.15733617335745675,
      "learning_rate": 0.00010079401350260287,
      "loss": 3.2495,
      "step": 3491
    },
    {
      "epoch": 0.8007337766567301,
      "grad_norm": 0.1701647974540495,
      "learning_rate": 0.0001005705315153569,
      "loss": 3.3908,
      "step": 3492
    },
    {
      "epoch": 0.8009630818619583,
      "grad_norm": 0.17092253097532326,
      "learning_rate": 0.00010034726985073362,
      "loss": 3.4976,
      "step": 3493
    },
    {
      "epoch": 0.8011923870671864,
      "grad_norm": 0.1631077765781974,
      "learning_rate": 0.00010012422863188314,
      "loss": 3.4744,
      "step": 3494
    },
    {
      "epoch": 0.8014216922724146,
      "grad_norm": 0.16653849588589242,
      "learning_rate": 9.9901407981833e-05,
      "loss": 3.3966,
      "step": 3495
    },
    {
      "epoch": 0.8016509974776428,
      "grad_norm": 0.17154497948513123,
      "learning_rate": 9.967880802348989e-05,
      "loss": 3.5404,
      "step": 3496
    },
    {
      "epoch": 0.8018803026828709,
      "grad_norm": 0.1722700421431167,
      "learning_rate": 9.94564288796384e-05,
      "loss": 3.5575,
      "step": 3497
    },
    {
      "epoch": 0.802109607888099,
      "grad_norm": 0.15717738316711555,
      "learning_rate": 9.923427067294133e-05,
      "loss": 3.4078,
      "step": 3498
    },
    {
      "epoch": 0.8023389130933272,
      "grad_norm": 0.17576620146660707,
      "learning_rate": 9.901233352593953e-05,
      "loss": 3.3821,
      "step": 3499
    },
    {
      "epoch": 0.8025682182985554,
      "grad_norm": 0.1618930008377299,
      "learning_rate": 9.879061756105234e-05,
      "loss": 3.5803,
      "step": 3500
    },
    {
      "epoch": 0.8027975235037835,
      "grad_norm": 0.17253554034560015,
      "learning_rate": 9.856912290057668e-05,
      "loss": 3.3382,
      "step": 3501
    },
    {
      "epoch": 0.8030268287090117,
      "grad_norm": 0.16100904792733675,
      "learning_rate": 9.834784966668747e-05,
      "loss": 3.4409,
      "step": 3502
    },
    {
      "epoch": 0.8032561339142399,
      "grad_norm": 0.1593274638926778,
      "learning_rate": 9.812679798143748e-05,
      "loss": 3.4506,
      "step": 3503
    },
    {
      "epoch": 0.803485439119468,
      "grad_norm": 0.16665541762849656,
      "learning_rate": 9.79059679667575e-05,
      "loss": 3.5697,
      "step": 3504
    },
    {
      "epoch": 0.8037147443246961,
      "grad_norm": 0.17219294211699507,
      "learning_rate": 9.768535974445586e-05,
      "loss": 3.6095,
      "step": 3505
    },
    {
      "epoch": 0.8039440495299244,
      "grad_norm": 0.17168628162024852,
      "learning_rate": 9.746497343621857e-05,
      "loss": 3.3988,
      "step": 3506
    },
    {
      "epoch": 0.8041733547351525,
      "grad_norm": 0.1706161906274764,
      "learning_rate": 9.724480916360906e-05,
      "loss": 3.4763,
      "step": 3507
    },
    {
      "epoch": 0.8044026599403806,
      "grad_norm": 0.15284191456685192,
      "learning_rate": 9.702486704806896e-05,
      "loss": 3.324,
      "step": 3508
    },
    {
      "epoch": 0.8046319651456088,
      "grad_norm": 0.16637740432393006,
      "learning_rate": 9.68051472109162e-05,
      "loss": 3.5672,
      "step": 3509
    },
    {
      "epoch": 0.804861270350837,
      "grad_norm": 0.1693041925947679,
      "learning_rate": 9.658564977334727e-05,
      "loss": 3.4589,
      "step": 3510
    },
    {
      "epoch": 0.8050905755560651,
      "grad_norm": 0.16073414639282124,
      "learning_rate": 9.636637485643529e-05,
      "loss": 3.5589,
      "step": 3511
    },
    {
      "epoch": 0.8053198807612932,
      "grad_norm": 0.1635827789103493,
      "learning_rate": 9.614732258113096e-05,
      "loss": 3.4485,
      "step": 3512
    },
    {
      "epoch": 0.8055491859665215,
      "grad_norm": 0.16795942811976358,
      "learning_rate": 9.592849306826174e-05,
      "loss": 3.4396,
      "step": 3513
    },
    {
      "epoch": 0.8057784911717496,
      "grad_norm": 0.16311606305500856,
      "learning_rate": 9.570988643853296e-05,
      "loss": 3.5438,
      "step": 3514
    },
    {
      "epoch": 0.8060077963769777,
      "grad_norm": 0.1709231658057814,
      "learning_rate": 9.549150281252633e-05,
      "loss": 3.591,
      "step": 3515
    },
    {
      "epoch": 0.806237101582206,
      "grad_norm": 0.1683844216646939,
      "learning_rate": 9.527334231070084e-05,
      "loss": 3.3691,
      "step": 3516
    },
    {
      "epoch": 0.8064664067874341,
      "grad_norm": 0.1780159904647418,
      "learning_rate": 9.505540505339223e-05,
      "loss": 3.4887,
      "step": 3517
    },
    {
      "epoch": 0.8066957119926622,
      "grad_norm": 0.1686590241625279,
      "learning_rate": 9.483769116081348e-05,
      "loss": 3.4816,
      "step": 3518
    },
    {
      "epoch": 0.8069250171978903,
      "grad_norm": 0.17283767209155965,
      "learning_rate": 9.4620200753054e-05,
      "loss": 3.4014,
      "step": 3519
    },
    {
      "epoch": 0.8071543224031186,
      "grad_norm": 0.16628629407523174,
      "learning_rate": 9.440293395008004e-05,
      "loss": 3.3743,
      "step": 3520
    },
    {
      "epoch": 0.8073836276083467,
      "grad_norm": 0.17519340346539228,
      "learning_rate": 9.418589087173441e-05,
      "loss": 3.4373,
      "step": 3521
    },
    {
      "epoch": 0.8076129328135748,
      "grad_norm": 0.16616035729244483,
      "learning_rate": 9.396907163773694e-05,
      "loss": 3.4568,
      "step": 3522
    },
    {
      "epoch": 0.8078422380188031,
      "grad_norm": 0.15984943269024754,
      "learning_rate": 9.375247636768325e-05,
      "loss": 3.3989,
      "step": 3523
    },
    {
      "epoch": 0.8080715432240312,
      "grad_norm": 0.15697220162099607,
      "learning_rate": 9.35361051810461e-05,
      "loss": 3.3565,
      "step": 3524
    },
    {
      "epoch": 0.8083008484292593,
      "grad_norm": 0.14521989359025736,
      "learning_rate": 9.331995819717443e-05,
      "loss": 3.4938,
      "step": 3525
    },
    {
      "epoch": 0.8085301536344875,
      "grad_norm": 0.17044775490058867,
      "learning_rate": 9.310403553529334e-05,
      "loss": 3.5031,
      "step": 3526
    },
    {
      "epoch": 0.8087594588397157,
      "grad_norm": 0.16277916829545042,
      "learning_rate": 9.288833731450419e-05,
      "loss": 3.3744,
      "step": 3527
    },
    {
      "epoch": 0.8089887640449438,
      "grad_norm": 0.16843625799642967,
      "learning_rate": 9.267286365378503e-05,
      "loss": 3.3539,
      "step": 3528
    },
    {
      "epoch": 0.809218069250172,
      "grad_norm": 0.163591055405162,
      "learning_rate": 9.245761467198948e-05,
      "loss": 3.3423,
      "step": 3529
    },
    {
      "epoch": 0.8094473744554002,
      "grad_norm": 0.16949339674534025,
      "learning_rate": 9.22425904878475e-05,
      "loss": 3.6155,
      "step": 3530
    },
    {
      "epoch": 0.8096766796606283,
      "grad_norm": 0.15595498526348375,
      "learning_rate": 9.20277912199648e-05,
      "loss": 3.2982,
      "step": 3531
    },
    {
      "epoch": 0.8099059848658564,
      "grad_norm": 0.18366548049209233,
      "learning_rate": 9.181321698682354e-05,
      "loss": 3.5216,
      "step": 3532
    },
    {
      "epoch": 0.8101352900710846,
      "grad_norm": 0.1848353123476605,
      "learning_rate": 9.159886790678123e-05,
      "loss": 3.5804,
      "step": 3533
    },
    {
      "epoch": 0.8103645952763128,
      "grad_norm": 0.1670538136802275,
      "learning_rate": 9.138474409807145e-05,
      "loss": 3.5125,
      "step": 3534
    },
    {
      "epoch": 0.8105939004815409,
      "grad_norm": 0.1800229500420716,
      "learning_rate": 9.11708456788033e-05,
      "loss": 3.3246,
      "step": 3535
    },
    {
      "epoch": 0.8108232056867691,
      "grad_norm": 0.17587334843465044,
      "learning_rate": 9.095717276696214e-05,
      "loss": 3.5017,
      "step": 3536
    },
    {
      "epoch": 0.8110525108919973,
      "grad_norm": 0.1674400549896529,
      "learning_rate": 9.074372548040793e-05,
      "loss": 3.4961,
      "step": 3537
    },
    {
      "epoch": 0.8112818160972254,
      "grad_norm": 0.17428357647420556,
      "learning_rate": 9.053050393687723e-05,
      "loss": 3.4879,
      "step": 3538
    },
    {
      "epoch": 0.8115111213024536,
      "grad_norm": 0.15874105962600107,
      "learning_rate": 9.031750825398145e-05,
      "loss": 3.3071,
      "step": 3539
    },
    {
      "epoch": 0.8117404265076817,
      "grad_norm": 0.16969063687027036,
      "learning_rate": 9.010473854920759e-05,
      "loss": 3.4737,
      "step": 3540
    },
    {
      "epoch": 0.8119697317129099,
      "grad_norm": 0.1808622540884579,
      "learning_rate": 8.98921949399179e-05,
      "loss": 3.5358,
      "step": 3541
    },
    {
      "epoch": 0.812199036918138,
      "grad_norm": 0.169777397934649,
      "learning_rate": 8.967987754335022e-05,
      "loss": 3.405,
      "step": 3542
    },
    {
      "epoch": 0.8124283421233662,
      "grad_norm": 0.169477938276937,
      "learning_rate": 8.94677864766173e-05,
      "loss": 3.3425,
      "step": 3543
    },
    {
      "epoch": 0.8126576473285944,
      "grad_norm": 0.187865525375078,
      "learning_rate": 8.925592185670711e-05,
      "loss": 3.5406,
      "step": 3544
    },
    {
      "epoch": 0.8128869525338225,
      "grad_norm": 0.15688698703006215,
      "learning_rate": 8.904428380048269e-05,
      "loss": 3.5186,
      "step": 3545
    },
    {
      "epoch": 0.8131162577390507,
      "grad_norm": 0.16929715031721898,
      "learning_rate": 8.883287242468242e-05,
      "loss": 3.3787,
      "step": 3546
    },
    {
      "epoch": 0.8133455629442788,
      "grad_norm": 0.1480443940772312,
      "learning_rate": 8.862168784591929e-05,
      "loss": 3.4184,
      "step": 3547
    },
    {
      "epoch": 0.813574868149507,
      "grad_norm": 0.1592160148850113,
      "learning_rate": 8.841073018068125e-05,
      "loss": 3.4018,
      "step": 3548
    },
    {
      "epoch": 0.8138041733547352,
      "grad_norm": 0.16674071675625682,
      "learning_rate": 8.819999954533115e-05,
      "loss": 3.5292,
      "step": 3549
    },
    {
      "epoch": 0.8140334785599633,
      "grad_norm": 0.16049514182831157,
      "learning_rate": 8.798949605610662e-05,
      "loss": 3.5074,
      "step": 3550
    },
    {
      "epoch": 0.8142627837651915,
      "grad_norm": 0.18135586864735376,
      "learning_rate": 8.777921982911996e-05,
      "loss": 3.3744,
      "step": 3551
    },
    {
      "epoch": 0.8144920889704196,
      "grad_norm": 0.18068534962706295,
      "learning_rate": 8.756917098035799e-05,
      "loss": 3.4082,
      "step": 3552
    },
    {
      "epoch": 0.8147213941756478,
      "grad_norm": 0.17583259847438928,
      "learning_rate": 8.735934962568253e-05,
      "loss": 3.4882,
      "step": 3553
    },
    {
      "epoch": 0.8149506993808759,
      "grad_norm": 0.1707896727857121,
      "learning_rate": 8.714975588082951e-05,
      "loss": 3.2822,
      "step": 3554
    },
    {
      "epoch": 0.8151800045861041,
      "grad_norm": 0.1745812610008835,
      "learning_rate": 8.694038986140945e-05,
      "loss": 3.3712,
      "step": 3555
    },
    {
      "epoch": 0.8154093097913323,
      "grad_norm": 0.1775701260530073,
      "learning_rate": 8.673125168290713e-05,
      "loss": 3.3625,
      "step": 3556
    },
    {
      "epoch": 0.8156386149965604,
      "grad_norm": 0.18879262313917283,
      "learning_rate": 8.652234146068206e-05,
      "loss": 3.4449,
      "step": 3557
    },
    {
      "epoch": 0.8158679202017886,
      "grad_norm": 0.15440002293955096,
      "learning_rate": 8.631365930996732e-05,
      "loss": 3.3317,
      "step": 3558
    },
    {
      "epoch": 0.8160972254070168,
      "grad_norm": 0.17014373350544135,
      "learning_rate": 8.610520534587086e-05,
      "loss": 3.3531,
      "step": 3559
    },
    {
      "epoch": 0.8163265306122449,
      "grad_norm": 0.1769974134290452,
      "learning_rate": 8.589697968337446e-05,
      "loss": 3.4644,
      "step": 3560
    },
    {
      "epoch": 0.816555835817473,
      "grad_norm": 0.1689317824515115,
      "learning_rate": 8.568898243733397e-05,
      "loss": 3.4828,
      "step": 3561
    },
    {
      "epoch": 0.8167851410227012,
      "grad_norm": 0.15961695079481747,
      "learning_rate": 8.548121372247918e-05,
      "loss": 3.3396,
      "step": 3562
    },
    {
      "epoch": 0.8170144462279294,
      "grad_norm": 0.1704498624556381,
      "learning_rate": 8.527367365341409e-05,
      "loss": 3.3629,
      "step": 3563
    },
    {
      "epoch": 0.8172437514331575,
      "grad_norm": 0.20316165950873677,
      "learning_rate": 8.50663623446164e-05,
      "loss": 3.6863,
      "step": 3564
    },
    {
      "epoch": 0.8174730566383857,
      "grad_norm": 0.17355073546806038,
      "learning_rate": 8.485927991043757e-05,
      "loss": 3.2738,
      "step": 3565
    },
    {
      "epoch": 0.8177023618436139,
      "grad_norm": 0.15454952292907861,
      "learning_rate": 8.465242646510279e-05,
      "loss": 3.408,
      "step": 3566
    },
    {
      "epoch": 0.817931667048842,
      "grad_norm": 0.15288105194431303,
      "learning_rate": 8.444580212271125e-05,
      "loss": 3.3178,
      "step": 3567
    },
    {
      "epoch": 0.8181609722540701,
      "grad_norm": 0.17654740326336082,
      "learning_rate": 8.423940699723548e-05,
      "loss": 3.5101,
      "step": 3568
    },
    {
      "epoch": 0.8183902774592984,
      "grad_norm": 0.16193802603846222,
      "learning_rate": 8.403324120252159e-05,
      "loss": 3.3961,
      "step": 3569
    },
    {
      "epoch": 0.8186195826645265,
      "grad_norm": 0.17150742391790313,
      "learning_rate": 8.382730485228923e-05,
      "loss": 3.4211,
      "step": 3570
    },
    {
      "epoch": 0.8188488878697546,
      "grad_norm": 0.15819635566360357,
      "learning_rate": 8.362159806013175e-05,
      "loss": 3.4898,
      "step": 3571
    },
    {
      "epoch": 0.8190781930749828,
      "grad_norm": 0.18592274524652078,
      "learning_rate": 8.341612093951523e-05,
      "loss": 3.5055,
      "step": 3572
    },
    {
      "epoch": 0.819307498280211,
      "grad_norm": 0.15669030965098338,
      "learning_rate": 8.321087360377988e-05,
      "loss": 3.2986,
      "step": 3573
    },
    {
      "epoch": 0.8195368034854391,
      "grad_norm": 0.1589679912211388,
      "learning_rate": 8.300585616613854e-05,
      "loss": 3.4909,
      "step": 3574
    },
    {
      "epoch": 0.8197661086906672,
      "grad_norm": 0.17211473806660446,
      "learning_rate": 8.280106873967752e-05,
      "loss": 3.4489,
      "step": 3575
    },
    {
      "epoch": 0.8199954138958955,
      "grad_norm": 0.17296177865658505,
      "learning_rate": 8.259651143735603e-05,
      "loss": 3.4341,
      "step": 3576
    },
    {
      "epoch": 0.8202247191011236,
      "grad_norm": 0.17924750179503784,
      "learning_rate": 8.239218437200679e-05,
      "loss": 3.3095,
      "step": 3577
    },
    {
      "epoch": 0.8204540243063517,
      "grad_norm": 0.1665900252805461,
      "learning_rate": 8.218808765633512e-05,
      "loss": 3.4892,
      "step": 3578
    },
    {
      "epoch": 0.82068332951158,
      "grad_norm": 0.17452307973764666,
      "learning_rate": 8.198422140291939e-05,
      "loss": 3.5296,
      "step": 3579
    },
    {
      "epoch": 0.8209126347168081,
      "grad_norm": 0.16386192336348096,
      "learning_rate": 8.178058572421077e-05,
      "loss": 3.4851,
      "step": 3580
    },
    {
      "epoch": 0.8211419399220362,
      "grad_norm": 0.17498467326482448,
      "learning_rate": 8.157718073253351e-05,
      "loss": 3.4123,
      "step": 3581
    },
    {
      "epoch": 0.8213712451272643,
      "grad_norm": 0.16259661454702445,
      "learning_rate": 8.137400654008436e-05,
      "loss": 3.488,
      "step": 3582
    },
    {
      "epoch": 0.8216005503324926,
      "grad_norm": 0.1693224304919875,
      "learning_rate": 8.117106325893287e-05,
      "loss": 3.3476,
      "step": 3583
    },
    {
      "epoch": 0.8218298555377207,
      "grad_norm": 0.17123935180375213,
      "learning_rate": 8.096835100102107e-05,
      "loss": 3.4395,
      "step": 3584
    },
    {
      "epoch": 0.8220591607429488,
      "grad_norm": 0.15951013510508677,
      "learning_rate": 8.076586987816404e-05,
      "loss": 3.5332,
      "step": 3585
    },
    {
      "epoch": 0.8222884659481771,
      "grad_norm": 0.1623234503599117,
      "learning_rate": 8.056362000204847e-05,
      "loss": 3.2929,
      "step": 3586
    },
    {
      "epoch": 0.8225177711534052,
      "grad_norm": 0.16303003163071897,
      "learning_rate": 8.036160148423449e-05,
      "loss": 3.6363,
      "step": 3587
    },
    {
      "epoch": 0.8227470763586333,
      "grad_norm": 0.16301270658239836,
      "learning_rate": 8.0159814436154e-05,
      "loss": 3.378,
      "step": 3588
    },
    {
      "epoch": 0.8229763815638615,
      "grad_norm": 0.18155467436192324,
      "learning_rate": 7.995825896911141e-05,
      "loss": 3.3339,
      "step": 3589
    },
    {
      "epoch": 0.8232056867690897,
      "grad_norm": 0.1712236000087474,
      "learning_rate": 7.975693519428318e-05,
      "loss": 3.4085,
      "step": 3590
    },
    {
      "epoch": 0.8234349919743178,
      "grad_norm": 0.1579494659722188,
      "learning_rate": 7.955584322271853e-05,
      "loss": 3.4686,
      "step": 3591
    },
    {
      "epoch": 0.8236642971795459,
      "grad_norm": 0.1737222697601203,
      "learning_rate": 7.935498316533818e-05,
      "loss": 3.2766,
      "step": 3592
    },
    {
      "epoch": 0.8238936023847742,
      "grad_norm": 0.1571289338713734,
      "learning_rate": 7.915435513293523e-05,
      "loss": 3.3639,
      "step": 3593
    },
    {
      "epoch": 0.8241229075900023,
      "grad_norm": 0.1664680834922629,
      "learning_rate": 7.89539592361747e-05,
      "loss": 3.5564,
      "step": 3594
    },
    {
      "epoch": 0.8243522127952304,
      "grad_norm": 0.17479173798394312,
      "learning_rate": 7.875379558559387e-05,
      "loss": 3.4121,
      "step": 3595
    },
    {
      "epoch": 0.8245815180004586,
      "grad_norm": 0.15860777242199717,
      "learning_rate": 7.85538642916015e-05,
      "loss": 3.5469,
      "step": 3596
    },
    {
      "epoch": 0.8248108232056868,
      "grad_norm": 0.16041222251856663,
      "learning_rate": 7.835416546447838e-05,
      "loss": 3.4519,
      "step": 3597
    },
    {
      "epoch": 0.8250401284109149,
      "grad_norm": 0.187463883103448,
      "learning_rate": 7.81546992143769e-05,
      "loss": 3.3953,
      "step": 3598
    },
    {
      "epoch": 0.8252694336161431,
      "grad_norm": 0.18337186387365761,
      "learning_rate": 7.795546565132167e-05,
      "loss": 3.3942,
      "step": 3599
    },
    {
      "epoch": 0.8254987388213713,
      "grad_norm": 0.180703546277697,
      "learning_rate": 7.775646488520816e-05,
      "loss": 3.4102,
      "step": 3600
    },
    {
      "epoch": 0.8257280440265994,
      "grad_norm": 0.1582626935884187,
      "learning_rate": 7.755769702580412e-05,
      "loss": 3.3804,
      "step": 3601
    },
    {
      "epoch": 0.8259573492318275,
      "grad_norm": 0.1935725637436937,
      "learning_rate": 7.735916218274857e-05,
      "loss": 3.3938,
      "step": 3602
    },
    {
      "epoch": 0.8261866544370557,
      "grad_norm": 0.16392998629251573,
      "learning_rate": 7.716086046555193e-05,
      "loss": 3.3992,
      "step": 3603
    },
    {
      "epoch": 0.8264159596422839,
      "grad_norm": 0.17519465259915334,
      "learning_rate": 7.696279198359607e-05,
      "loss": 3.3912,
      "step": 3604
    },
    {
      "epoch": 0.826645264847512,
      "grad_norm": 0.16817454216473637,
      "learning_rate": 7.676495684613432e-05,
      "loss": 3.5264,
      "step": 3605
    },
    {
      "epoch": 0.8268745700527402,
      "grad_norm": 0.1693341149403124,
      "learning_rate": 7.656735516229125e-05,
      "loss": 3.4248,
      "step": 3606
    },
    {
      "epoch": 0.8271038752579684,
      "grad_norm": 0.17658412732561055,
      "learning_rate": 7.636998704106252e-05,
      "loss": 3.6141,
      "step": 3607
    },
    {
      "epoch": 0.8273331804631965,
      "grad_norm": 0.1735188340004837,
      "learning_rate": 7.617285259131501e-05,
      "loss": 3.4518,
      "step": 3608
    },
    {
      "epoch": 0.8275624856684247,
      "grad_norm": 0.16870827157709695,
      "learning_rate": 7.597595192178702e-05,
      "loss": 3.4355,
      "step": 3609
    },
    {
      "epoch": 0.8277917908736528,
      "grad_norm": 0.16524134826755024,
      "learning_rate": 7.577928514108745e-05,
      "loss": 3.4479,
      "step": 3610
    },
    {
      "epoch": 0.828021096078881,
      "grad_norm": 0.16046763204543807,
      "learning_rate": 7.558285235769646e-05,
      "loss": 3.4875,
      "step": 3611
    },
    {
      "epoch": 0.8282504012841091,
      "grad_norm": 0.1632990180450673,
      "learning_rate": 7.538665367996484e-05,
      "loss": 3.463,
      "step": 3612
    },
    {
      "epoch": 0.8284797064893373,
      "grad_norm": 0.1638661374664673,
      "learning_rate": 7.519068921611494e-05,
      "loss": 3.5285,
      "step": 3613
    },
    {
      "epoch": 0.8287090116945655,
      "grad_norm": 0.17651397422038484,
      "learning_rate": 7.499495907423887e-05,
      "loss": 3.4883,
      "step": 3614
    },
    {
      "epoch": 0.8289383168997936,
      "grad_norm": 0.18114897228908938,
      "learning_rate": 7.479946336230047e-05,
      "loss": 3.4295,
      "step": 3615
    },
    {
      "epoch": 0.8291676221050218,
      "grad_norm": 0.16752272269092924,
      "learning_rate": 7.46042021881338e-05,
      "loss": 3.4881,
      "step": 3616
    },
    {
      "epoch": 0.8293969273102499,
      "grad_norm": 0.16708425028691906,
      "learning_rate": 7.440917565944349e-05,
      "loss": 3.4969,
      "step": 3617
    },
    {
      "epoch": 0.8296262325154781,
      "grad_norm": 0.16512262204187056,
      "learning_rate": 7.421438388380486e-05,
      "loss": 3.4934,
      "step": 3618
    },
    {
      "epoch": 0.8298555377207063,
      "grad_norm": 0.16479754999060833,
      "learning_rate": 7.4019826968664e-05,
      "loss": 3.6205,
      "step": 3619
    },
    {
      "epoch": 0.8300848429259344,
      "grad_norm": 0.16963105612188836,
      "learning_rate": 7.382550502133694e-05,
      "loss": 3.512,
      "step": 3620
    },
    {
      "epoch": 0.8303141481311626,
      "grad_norm": 0.17109095450579223,
      "learning_rate": 7.363141814901053e-05,
      "loss": 3.4097,
      "step": 3621
    },
    {
      "epoch": 0.8305434533363908,
      "grad_norm": 0.16983640703976416,
      "learning_rate": 7.343756645874162e-05,
      "loss": 3.3782,
      "step": 3622
    },
    {
      "epoch": 0.8307727585416189,
      "grad_norm": 0.1609711257017234,
      "learning_rate": 7.32439500574577e-05,
      "loss": 3.4449,
      "step": 3623
    },
    {
      "epoch": 0.831002063746847,
      "grad_norm": 0.18336248462730284,
      "learning_rate": 7.305056905195618e-05,
      "loss": 3.5413,
      "step": 3624
    },
    {
      "epoch": 0.8312313689520752,
      "grad_norm": 0.1630580615857299,
      "learning_rate": 7.285742354890473e-05,
      "loss": 3.4086,
      "step": 3625
    },
    {
      "epoch": 0.8314606741573034,
      "grad_norm": 0.17239441431990715,
      "learning_rate": 7.266451365484105e-05,
      "loss": 3.5484,
      "step": 3626
    },
    {
      "epoch": 0.8316899793625315,
      "grad_norm": 0.16061734901761113,
      "learning_rate": 7.247183947617325e-05,
      "loss": 3.3575,
      "step": 3627
    },
    {
      "epoch": 0.8319192845677597,
      "grad_norm": 0.17414824519650604,
      "learning_rate": 7.227940111917863e-05,
      "loss": 3.4211,
      "step": 3628
    },
    {
      "epoch": 0.8321485897729879,
      "grad_norm": 0.16308306890429572,
      "learning_rate": 7.20871986900053e-05,
      "loss": 3.5483,
      "step": 3629
    },
    {
      "epoch": 0.832377894978216,
      "grad_norm": 0.159319276560701,
      "learning_rate": 7.189523229467071e-05,
      "loss": 3.4596,
      "step": 3630
    },
    {
      "epoch": 0.8326072001834441,
      "grad_norm": 0.16404663933965122,
      "learning_rate": 7.170350203906218e-05,
      "loss": 3.3868,
      "step": 3631
    },
    {
      "epoch": 0.8328365053886724,
      "grad_norm": 0.16609132440781713,
      "learning_rate": 7.151200802893681e-05,
      "loss": 3.5261,
      "step": 3632
    },
    {
      "epoch": 0.8330658105939005,
      "grad_norm": 0.16489531609068153,
      "learning_rate": 7.132075036992158e-05,
      "loss": 3.5706,
      "step": 3633
    },
    {
      "epoch": 0.8332951157991286,
      "grad_norm": 0.1661121050600127,
      "learning_rate": 7.112972916751287e-05,
      "loss": 3.408,
      "step": 3634
    },
    {
      "epoch": 0.8335244210043568,
      "grad_norm": 0.16402328721908435,
      "learning_rate": 7.093894452707666e-05,
      "loss": 3.4209,
      "step": 3635
    },
    {
      "epoch": 0.833753726209585,
      "grad_norm": 0.16974648447137994,
      "learning_rate": 7.074839655384835e-05,
      "loss": 3.5061,
      "step": 3636
    },
    {
      "epoch": 0.8339830314148131,
      "grad_norm": 0.16367605746770683,
      "learning_rate": 7.055808535293334e-05,
      "loss": 3.3672,
      "step": 3637
    },
    {
      "epoch": 0.8342123366200412,
      "grad_norm": 0.17915617811522996,
      "learning_rate": 7.036801102930551e-05,
      "loss": 3.4037,
      "step": 3638
    },
    {
      "epoch": 0.8344416418252695,
      "grad_norm": 0.18286706557558693,
      "learning_rate": 7.017817368780888e-05,
      "loss": 3.2725,
      "step": 3639
    },
    {
      "epoch": 0.8346709470304976,
      "grad_norm": 0.1644375063177599,
      "learning_rate": 6.998857343315634e-05,
      "loss": 3.4056,
      "step": 3640
    },
    {
      "epoch": 0.8349002522357257,
      "grad_norm": 0.15952503431330337,
      "learning_rate": 6.979921036993042e-05,
      "loss": 3.4186,
      "step": 3641
    },
    {
      "epoch": 0.835129557440954,
      "grad_norm": 0.17170574346263742,
      "learning_rate": 6.9610084602582e-05,
      "loss": 3.4195,
      "step": 3642
    },
    {
      "epoch": 0.8353588626461821,
      "grad_norm": 0.18713546407455867,
      "learning_rate": 6.942119623543202e-05,
      "loss": 3.474,
      "step": 3643
    },
    {
      "epoch": 0.8355881678514102,
      "grad_norm": 0.18302147031376978,
      "learning_rate": 6.923254537266988e-05,
      "loss": 3.4949,
      "step": 3644
    },
    {
      "epoch": 0.8358174730566383,
      "grad_norm": 0.1743478865900858,
      "learning_rate": 6.904413211835414e-05,
      "loss": 3.5001,
      "step": 3645
    },
    {
      "epoch": 0.8360467782618666,
      "grad_norm": 0.16137822483559608,
      "learning_rate": 6.885595657641213e-05,
      "loss": 3.3958,
      "step": 3646
    },
    {
      "epoch": 0.8362760834670947,
      "grad_norm": 0.1669209401151373,
      "learning_rate": 6.866801885064056e-05,
      "loss": 3.4447,
      "step": 3647
    },
    {
      "epoch": 0.8365053886723228,
      "grad_norm": 0.16224843466411054,
      "learning_rate": 6.848031904470441e-05,
      "loss": 3.4685,
      "step": 3648
    },
    {
      "epoch": 0.8367346938775511,
      "grad_norm": 0.17661829713251212,
      "learning_rate": 6.829285726213769e-05,
      "loss": 3.5112,
      "step": 3649
    },
    {
      "epoch": 0.8369639990827792,
      "grad_norm": 0.17217637929504093,
      "learning_rate": 6.810563360634297e-05,
      "loss": 3.4638,
      "step": 3650
    },
    {
      "epoch": 0.8371933042880073,
      "grad_norm": 0.17046161744316138,
      "learning_rate": 6.79186481805918e-05,
      "loss": 3.4058,
      "step": 3651
    },
    {
      "epoch": 0.8374226094932355,
      "grad_norm": 0.17288882256894764,
      "learning_rate": 6.773190108802374e-05,
      "loss": 3.4262,
      "step": 3652
    },
    {
      "epoch": 0.8376519146984637,
      "grad_norm": 0.2081053400185701,
      "learning_rate": 6.754539243164754e-05,
      "loss": 3.4642,
      "step": 3653
    },
    {
      "epoch": 0.8378812199036918,
      "grad_norm": 0.16908476583481333,
      "learning_rate": 6.735912231433993e-05,
      "loss": 3.4508,
      "step": 3654
    },
    {
      "epoch": 0.8381105251089199,
      "grad_norm": 0.16711225972460259,
      "learning_rate": 6.717309083884654e-05,
      "loss": 3.4376,
      "step": 3655
    },
    {
      "epoch": 0.8383398303141482,
      "grad_norm": 0.16776742730028243,
      "learning_rate": 6.698729810778065e-05,
      "loss": 3.3053,
      "step": 3656
    },
    {
      "epoch": 0.8385691355193763,
      "grad_norm": 0.17279617086590585,
      "learning_rate": 6.680174422362468e-05,
      "loss": 3.4928,
      "step": 3657
    },
    {
      "epoch": 0.8387984407246044,
      "grad_norm": 0.17004769077784668,
      "learning_rate": 6.661642928872874e-05,
      "loss": 3.4593,
      "step": 3658
    },
    {
      "epoch": 0.8390277459298326,
      "grad_norm": 0.16581567644599454,
      "learning_rate": 6.643135340531136e-05,
      "loss": 3.437,
      "step": 3659
    },
    {
      "epoch": 0.8392570511350608,
      "grad_norm": 0.17255608054996013,
      "learning_rate": 6.6246516675459e-05,
      "loss": 3.5364,
      "step": 3660
    },
    {
      "epoch": 0.8394863563402889,
      "grad_norm": 0.17818573020428485,
      "learning_rate": 6.606191920112664e-05,
      "loss": 3.4519,
      "step": 3661
    },
    {
      "epoch": 0.8397156615455171,
      "grad_norm": 0.1739158554235178,
      "learning_rate": 6.587756108413684e-05,
      "loss": 3.3843,
      "step": 3662
    },
    {
      "epoch": 0.8399449667507453,
      "grad_norm": 0.16367550209454376,
      "learning_rate": 6.569344242618036e-05,
      "loss": 3.4598,
      "step": 3663
    },
    {
      "epoch": 0.8401742719559734,
      "grad_norm": 0.16393586260880819,
      "learning_rate": 6.550956332881564e-05,
      "loss": 3.4597,
      "step": 3664
    },
    {
      "epoch": 0.8404035771612015,
      "grad_norm": 0.166805635492427,
      "learning_rate": 6.532592389346958e-05,
      "loss": 3.5212,
      "step": 3665
    },
    {
      "epoch": 0.8406328823664297,
      "grad_norm": 0.17592641340680584,
      "learning_rate": 6.514252422143591e-05,
      "loss": 3.506,
      "step": 3666
    },
    {
      "epoch": 0.8408621875716579,
      "grad_norm": 0.173078651865898,
      "learning_rate": 6.495936441387713e-05,
      "loss": 3.5501,
      "step": 3667
    },
    {
      "epoch": 0.841091492776886,
      "grad_norm": 0.15750457428635753,
      "learning_rate": 6.477644457182275e-05,
      "loss": 3.4603,
      "step": 3668
    },
    {
      "epoch": 0.8413207979821142,
      "grad_norm": 0.15398864354025774,
      "learning_rate": 6.459376479617013e-05,
      "loss": 3.428,
      "step": 3669
    },
    {
      "epoch": 0.8415501031873424,
      "grad_norm": 0.17486403732893285,
      "learning_rate": 6.441132518768417e-05,
      "loss": 3.4853,
      "step": 3670
    },
    {
      "epoch": 0.8417794083925705,
      "grad_norm": 0.19429711503728866,
      "learning_rate": 6.422912584699752e-05,
      "loss": 3.3682,
      "step": 3671
    },
    {
      "epoch": 0.8420087135977987,
      "grad_norm": 0.16399019523278918,
      "learning_rate": 6.404716687461004e-05,
      "loss": 3.3475,
      "step": 3672
    },
    {
      "epoch": 0.8422380188030268,
      "grad_norm": 0.17019908608313056,
      "learning_rate": 6.386544837088904e-05,
      "loss": 3.5069,
      "step": 3673
    },
    {
      "epoch": 0.842467324008255,
      "grad_norm": 0.1572414044224314,
      "learning_rate": 6.36839704360691e-05,
      "loss": 3.4543,
      "step": 3674
    },
    {
      "epoch": 0.8426966292134831,
      "grad_norm": 0.15474497552488067,
      "learning_rate": 6.350273317025251e-05,
      "loss": 3.4802,
      "step": 3675
    },
    {
      "epoch": 0.8429259344187113,
      "grad_norm": 0.16758875004680848,
      "learning_rate": 6.33217366734084e-05,
      "loss": 3.5334,
      "step": 3676
    },
    {
      "epoch": 0.8431552396239395,
      "grad_norm": 0.16251936969515243,
      "learning_rate": 6.314098104537324e-05,
      "loss": 3.4815,
      "step": 3677
    },
    {
      "epoch": 0.8433845448291676,
      "grad_norm": 0.1570436826991895,
      "learning_rate": 6.29604663858505e-05,
      "loss": 3.593,
      "step": 3678
    },
    {
      "epoch": 0.8436138500343958,
      "grad_norm": 0.17125233275975904,
      "learning_rate": 6.278019279441122e-05,
      "loss": 3.2793,
      "step": 3679
    },
    {
      "epoch": 0.8438431552396239,
      "grad_norm": 0.16145794975963357,
      "learning_rate": 6.260016037049259e-05,
      "loss": 3.3852,
      "step": 3680
    },
    {
      "epoch": 0.8440724604448521,
      "grad_norm": 0.17200282669388514,
      "learning_rate": 6.242036921339972e-05,
      "loss": 3.5554,
      "step": 3681
    },
    {
      "epoch": 0.8443017656500803,
      "grad_norm": 0.16487886752497052,
      "learning_rate": 6.224081942230403e-05,
      "loss": 3.3183,
      "step": 3682
    },
    {
      "epoch": 0.8445310708553084,
      "grad_norm": 0.179681622327588,
      "learning_rate": 6.206151109624402e-05,
      "loss": 3.637,
      "step": 3683
    },
    {
      "epoch": 0.8447603760605366,
      "grad_norm": 0.16795918289786527,
      "learning_rate": 6.188244433412482e-05,
      "loss": 3.622,
      "step": 3684
    },
    {
      "epoch": 0.8449896812657647,
      "grad_norm": 0.15305565409538133,
      "learning_rate": 6.170361923471868e-05,
      "loss": 3.5017,
      "step": 3685
    },
    {
      "epoch": 0.8452189864709929,
      "grad_norm": 0.1510576722648952,
      "learning_rate": 6.152503589666425e-05,
      "loss": 3.2689,
      "step": 3686
    },
    {
      "epoch": 0.845448291676221,
      "grad_norm": 0.1663343187440649,
      "learning_rate": 6.134669441846691e-05,
      "loss": 3.4222,
      "step": 3687
    },
    {
      "epoch": 0.8456775968814492,
      "grad_norm": 0.17557390383255722,
      "learning_rate": 6.116859489849846e-05,
      "loss": 3.5246,
      "step": 3688
    },
    {
      "epoch": 0.8459069020866774,
      "grad_norm": 0.15896464911376532,
      "learning_rate": 6.099073743499772e-05,
      "loss": 3.4534,
      "step": 3689
    },
    {
      "epoch": 0.8461362072919055,
      "grad_norm": 0.1626493824021472,
      "learning_rate": 6.081312212606943e-05,
      "loss": 3.5874,
      "step": 3690
    },
    {
      "epoch": 0.8463655124971337,
      "grad_norm": 0.16694619397418436,
      "learning_rate": 6.063574906968511e-05,
      "loss": 3.4055,
      "step": 3691
    },
    {
      "epoch": 0.8465948177023619,
      "grad_norm": 0.16268988229440184,
      "learning_rate": 6.045861836368244e-05,
      "loss": 3.3856,
      "step": 3692
    },
    {
      "epoch": 0.84682412290759,
      "grad_norm": 0.16099661092601633,
      "learning_rate": 6.028173010576582e-05,
      "loss": 3.36,
      "step": 3693
    },
    {
      "epoch": 0.8470534281128181,
      "grad_norm": 0.17136866448830712,
      "learning_rate": 6.0105084393505206e-05,
      "loss": 3.4497,
      "step": 3694
    },
    {
      "epoch": 0.8472827333180463,
      "grad_norm": 0.16310248131085014,
      "learning_rate": 5.9928681324337544e-05,
      "loss": 3.352,
      "step": 3695
    },
    {
      "epoch": 0.8475120385232745,
      "grad_norm": 0.17337173759148433,
      "learning_rate": 5.975252099556544e-05,
      "loss": 3.4192,
      "step": 3696
    },
    {
      "epoch": 0.8477413437285026,
      "grad_norm": 0.15554891378435573,
      "learning_rate": 5.957660350435773e-05,
      "loss": 3.4027,
      "step": 3697
    },
    {
      "epoch": 0.8479706489337308,
      "grad_norm": 0.16327749102900954,
      "learning_rate": 5.940092894774929e-05,
      "loss": 3.4551,
      "step": 3698
    },
    {
      "epoch": 0.848199954138959,
      "grad_norm": 0.16171648166977895,
      "learning_rate": 5.922549742264122e-05,
      "loss": 3.4291,
      "step": 3699
    },
    {
      "epoch": 0.8484292593441871,
      "grad_norm": 0.174034186716906,
      "learning_rate": 5.9050309025800275e-05,
      "loss": 3.506,
      "step": 3700
    },
    {
      "epoch": 0.8486585645494152,
      "grad_norm": 0.16689882659094452,
      "learning_rate": 5.8875363853859166e-05,
      "loss": 3.4498,
      "step": 3701
    },
    {
      "epoch": 0.8488878697546435,
      "grad_norm": 0.17936146672377073,
      "learning_rate": 5.8700662003316366e-05,
      "loss": 3.4046,
      "step": 3702
    },
    {
      "epoch": 0.8491171749598716,
      "grad_norm": 0.1565135413169788,
      "learning_rate": 5.852620357053651e-05,
      "loss": 3.4172,
      "step": 3703
    },
    {
      "epoch": 0.8493464801650997,
      "grad_norm": 0.1830622970322639,
      "learning_rate": 5.835198865174956e-05,
      "loss": 3.5361,
      "step": 3704
    },
    {
      "epoch": 0.8495757853703279,
      "grad_norm": 0.17459018624037706,
      "learning_rate": 5.8178017343051336e-05,
      "loss": 3.4102,
      "step": 3705
    },
    {
      "epoch": 0.8498050905755561,
      "grad_norm": 0.1569825286665608,
      "learning_rate": 5.800428974040311e-05,
      "loss": 3.3478,
      "step": 3706
    },
    {
      "epoch": 0.8500343957807842,
      "grad_norm": 0.16611438508053036,
      "learning_rate": 5.783080593963219e-05,
      "loss": 3.4669,
      "step": 3707
    },
    {
      "epoch": 0.8502637009860123,
      "grad_norm": 0.17731563984073723,
      "learning_rate": 5.7657566036430686e-05,
      "loss": 3.4533,
      "step": 3708
    },
    {
      "epoch": 0.8504930061912406,
      "grad_norm": 0.16765857097589304,
      "learning_rate": 5.748457012635683e-05,
      "loss": 3.4852,
      "step": 3709
    },
    {
      "epoch": 0.8507223113964687,
      "grad_norm": 0.1575473161667482,
      "learning_rate": 5.731181830483395e-05,
      "loss": 3.3362,
      "step": 3710
    },
    {
      "epoch": 0.8509516166016968,
      "grad_norm": 0.1570040814950494,
      "learning_rate": 5.713931066715078e-05,
      "loss": 3.5037,
      "step": 3711
    },
    {
      "epoch": 0.8511809218069251,
      "grad_norm": 0.17174425499771767,
      "learning_rate": 5.696704730846136e-05,
      "loss": 3.5523,
      "step": 3712
    },
    {
      "epoch": 0.8514102270121532,
      "grad_norm": 0.16215484377986752,
      "learning_rate": 5.679502832378497e-05,
      "loss": 3.5433,
      "step": 3713
    },
    {
      "epoch": 0.8516395322173813,
      "grad_norm": 0.17560245020700796,
      "learning_rate": 5.662325380800631e-05,
      "loss": 3.5464,
      "step": 3714
    },
    {
      "epoch": 0.8518688374226095,
      "grad_norm": 0.1643955987084494,
      "learning_rate": 5.645172385587482e-05,
      "loss": 3.5622,
      "step": 3715
    },
    {
      "epoch": 0.8520981426278377,
      "grad_norm": 0.14844951894696484,
      "learning_rate": 5.6280438562005434e-05,
      "loss": 3.3306,
      "step": 3716
    },
    {
      "epoch": 0.8523274478330658,
      "grad_norm": 0.16613443348228743,
      "learning_rate": 5.6109398020877834e-05,
      "loss": 3.4596,
      "step": 3717
    },
    {
      "epoch": 0.8525567530382939,
      "grad_norm": 0.15909585529548703,
      "learning_rate": 5.593860232683712e-05,
      "loss": 3.4321,
      "step": 3718
    },
    {
      "epoch": 0.8527860582435222,
      "grad_norm": 0.1629077521693289,
      "learning_rate": 5.576805157409265e-05,
      "loss": 3.4662,
      "step": 3719
    },
    {
      "epoch": 0.8530153634487503,
      "grad_norm": 0.17163569393036696,
      "learning_rate": 5.5597745856719394e-05,
      "loss": 3.3219,
      "step": 3720
    },
    {
      "epoch": 0.8532446686539784,
      "grad_norm": 0.17153528627730977,
      "learning_rate": 5.542768526865677e-05,
      "loss": 3.4555,
      "step": 3721
    },
    {
      "epoch": 0.8534739738592066,
      "grad_norm": 0.15846070909289156,
      "learning_rate": 5.525786990370901e-05,
      "loss": 3.3401,
      "step": 3722
    },
    {
      "epoch": 0.8537032790644348,
      "grad_norm": 0.19019608028963744,
      "learning_rate": 5.508829985554509e-05,
      "loss": 3.4272,
      "step": 3723
    },
    {
      "epoch": 0.8539325842696629,
      "grad_norm": 0.16771798304614974,
      "learning_rate": 5.4918975217698844e-05,
      "loss": 3.5021,
      "step": 3724
    },
    {
      "epoch": 0.8541618894748911,
      "grad_norm": 0.167597355331385,
      "learning_rate": 5.474989608356856e-05,
      "loss": 3.4531,
      "step": 3725
    },
    {
      "epoch": 0.8543911946801193,
      "grad_norm": 0.16584098370316766,
      "learning_rate": 5.458106254641715e-05,
      "loss": 3.4548,
      "step": 3726
    },
    {
      "epoch": 0.8546204998853474,
      "grad_norm": 0.17230801842659074,
      "learning_rate": 5.441247469937194e-05,
      "loss": 3.4576,
      "step": 3727
    },
    {
      "epoch": 0.8548498050905755,
      "grad_norm": 0.18468119799602378,
      "learning_rate": 5.424413263542522e-05,
      "loss": 3.3967,
      "step": 3728
    },
    {
      "epoch": 0.8550791102958037,
      "grad_norm": 0.16471405484213644,
      "learning_rate": 5.407603644743286e-05,
      "loss": 3.428,
      "step": 3729
    },
    {
      "epoch": 0.8553084155010319,
      "grad_norm": 0.17624542146428682,
      "learning_rate": 5.390818622811594e-05,
      "loss": 3.4045,
      "step": 3730
    },
    {
      "epoch": 0.85553772070626,
      "grad_norm": 0.15010252983151537,
      "learning_rate": 5.374058207005944e-05,
      "loss": 3.3574,
      "step": 3731
    },
    {
      "epoch": 0.8557670259114882,
      "grad_norm": 0.15729251056137403,
      "learning_rate": 5.357322406571269e-05,
      "loss": 3.4444,
      "step": 3732
    },
    {
      "epoch": 0.8559963311167164,
      "grad_norm": 0.16763387338598268,
      "learning_rate": 5.3406112307389066e-05,
      "loss": 3.4396,
      "step": 3733
    },
    {
      "epoch": 0.8562256363219445,
      "grad_norm": 0.17133764841131416,
      "learning_rate": 5.3239246887266554e-05,
      "loss": 3.5388,
      "step": 3734
    },
    {
      "epoch": 0.8564549415271727,
      "grad_norm": 0.18252527129277024,
      "learning_rate": 5.3072627897386926e-05,
      "loss": 3.3766,
      "step": 3735
    },
    {
      "epoch": 0.8566842467324008,
      "grad_norm": 0.16864841232803002,
      "learning_rate": 5.29062554296561e-05,
      "loss": 3.417,
      "step": 3736
    },
    {
      "epoch": 0.856913551937629,
      "grad_norm": 0.17327410172920865,
      "learning_rate": 5.27401295758439e-05,
      "loss": 3.3929,
      "step": 3737
    },
    {
      "epoch": 0.8571428571428571,
      "grad_norm": 0.16447837270285567,
      "learning_rate": 5.25742504275844e-05,
      "loss": 3.4331,
      "step": 3738
    },
    {
      "epoch": 0.8573721623480853,
      "grad_norm": 0.1578141781697319,
      "learning_rate": 5.2408618076375315e-05,
      "loss": 3.4396,
      "step": 3739
    },
    {
      "epoch": 0.8576014675533135,
      "grad_norm": 0.16185331584174262,
      "learning_rate": 5.224323261357844e-05,
      "loss": 3.3427,
      "step": 3740
    },
    {
      "epoch": 0.8578307727585416,
      "grad_norm": 0.18000697973038787,
      "learning_rate": 5.207809413041914e-05,
      "loss": 3.3327,
      "step": 3741
    },
    {
      "epoch": 0.8580600779637698,
      "grad_norm": 0.1620056903236253,
      "learning_rate": 5.1913202717987e-05,
      "loss": 3.2889,
      "step": 3742
    },
    {
      "epoch": 0.8582893831689979,
      "grad_norm": 0.17804391058674174,
      "learning_rate": 5.174855846723459e-05,
      "loss": 3.365,
      "step": 3743
    },
    {
      "epoch": 0.8585186883742261,
      "grad_norm": 0.1778816497780635,
      "learning_rate": 5.158416146897887e-05,
      "loss": 3.6495,
      "step": 3744
    },
    {
      "epoch": 0.8587479935794543,
      "grad_norm": 0.16267276068212688,
      "learning_rate": 5.1420011813900104e-05,
      "loss": 3.3556,
      "step": 3745
    },
    {
      "epoch": 0.8589772987846824,
      "grad_norm": 0.17784800709576953,
      "learning_rate": 5.125610959254212e-05,
      "loss": 3.5562,
      "step": 3746
    },
    {
      "epoch": 0.8592066039899106,
      "grad_norm": 0.16028214800994323,
      "learning_rate": 5.109245489531211e-05,
      "loss": 3.4613,
      "step": 3747
    },
    {
      "epoch": 0.8594359091951387,
      "grad_norm": 0.1734757948259417,
      "learning_rate": 5.092904781248115e-05,
      "loss": 3.3768,
      "step": 3748
    },
    {
      "epoch": 0.8596652144003669,
      "grad_norm": 0.16220539373416207,
      "learning_rate": 5.0765888434183446e-05,
      "loss": 3.4551,
      "step": 3749
    },
    {
      "epoch": 0.859894519605595,
      "grad_norm": 0.17682574950768815,
      "learning_rate": 5.060297685041659e-05,
      "loss": 3.5092,
      "step": 3750
    },
    {
      "epoch": 0.8601238248108232,
      "grad_norm": 0.173166684979041,
      "learning_rate": 5.0440313151041364e-05,
      "loss": 3.3787,
      "step": 3751
    },
    {
      "epoch": 0.8603531300160514,
      "grad_norm": 0.1882395144992159,
      "learning_rate": 5.027789742578226e-05,
      "loss": 3.4171,
      "step": 3752
    },
    {
      "epoch": 0.8605824352212795,
      "grad_norm": 0.16281207912221476,
      "learning_rate": 5.011572976422657e-05,
      "loss": 3.4994,
      "step": 3753
    },
    {
      "epoch": 0.8608117404265077,
      "grad_norm": 0.17636915844769657,
      "learning_rate": 4.995381025582485e-05,
      "loss": 3.3578,
      "step": 3754
    },
    {
      "epoch": 0.8610410456317359,
      "grad_norm": 0.18108543045292913,
      "learning_rate": 4.9792138989890825e-05,
      "loss": 3.3106,
      "step": 3755
    },
    {
      "epoch": 0.861270350836964,
      "grad_norm": 0.16314111159381542,
      "learning_rate": 4.9630716055601434e-05,
      "loss": 3.4481,
      "step": 3756
    },
    {
      "epoch": 0.8614996560421921,
      "grad_norm": 0.15271979092836313,
      "learning_rate": 4.9469541541996234e-05,
      "loss": 3.32,
      "step": 3757
    },
    {
      "epoch": 0.8617289612474203,
      "grad_norm": 0.15435370955894637,
      "learning_rate": 4.9308615537978216e-05,
      "loss": 3.3751,
      "step": 3758
    },
    {
      "epoch": 0.8619582664526485,
      "grad_norm": 0.15434148791972693,
      "learning_rate": 4.914793813231305e-05,
      "loss": 3.323,
      "step": 3759
    },
    {
      "epoch": 0.8621875716578766,
      "grad_norm": 0.17629077750227462,
      "learning_rate": 4.8987509413629284e-05,
      "loss": 3.6906,
      "step": 3760
    },
    {
      "epoch": 0.8624168768631048,
      "grad_norm": 0.1728011721273689,
      "learning_rate": 4.882732947041818e-05,
      "loss": 3.4292,
      "step": 3761
    },
    {
      "epoch": 0.862646182068333,
      "grad_norm": 0.17341802441069115,
      "learning_rate": 4.866739839103418e-05,
      "loss": 3.455,
      "step": 3762
    },
    {
      "epoch": 0.8628754872735611,
      "grad_norm": 0.16300375286663596,
      "learning_rate": 4.850771626369416e-05,
      "loss": 3.3555,
      "step": 3763
    },
    {
      "epoch": 0.8631047924787892,
      "grad_norm": 0.17666686300043727,
      "learning_rate": 4.8348283176477545e-05,
      "loss": 3.3468,
      "step": 3764
    },
    {
      "epoch": 0.8633340976840175,
      "grad_norm": 0.18799064578645314,
      "learning_rate": 4.818909921732662e-05,
      "loss": 3.6027,
      "step": 3765
    },
    {
      "epoch": 0.8635634028892456,
      "grad_norm": 0.16917332028378146,
      "learning_rate": 4.8030164474046286e-05,
      "loss": 3.349,
      "step": 3766
    },
    {
      "epoch": 0.8637927080944737,
      "grad_norm": 0.16371878790365188,
      "learning_rate": 4.787147903430383e-05,
      "loss": 3.5065,
      "step": 3767
    },
    {
      "epoch": 0.8640220132997019,
      "grad_norm": 0.16249523481691208,
      "learning_rate": 4.771304298562912e-05,
      "loss": 3.2939,
      "step": 3768
    },
    {
      "epoch": 0.8642513185049301,
      "grad_norm": 0.15451507592022026,
      "learning_rate": 4.755485641541424e-05,
      "loss": 3.5146,
      "step": 3769
    },
    {
      "epoch": 0.8644806237101582,
      "grad_norm": 0.16708957334263647,
      "learning_rate": 4.7396919410914174e-05,
      "loss": 3.6246,
      "step": 3770
    },
    {
      "epoch": 0.8647099289153863,
      "grad_norm": 0.163314976025884,
      "learning_rate": 4.723923205924557e-05,
      "loss": 3.6226,
      "step": 3771
    },
    {
      "epoch": 0.8649392341206146,
      "grad_norm": 0.17847262884949294,
      "learning_rate": 4.708179444738791e-05,
      "loss": 3.4061,
      "step": 3772
    },
    {
      "epoch": 0.8651685393258427,
      "grad_norm": 0.16171086849962774,
      "learning_rate": 4.6924606662182736e-05,
      "loss": 3.478,
      "step": 3773
    },
    {
      "epoch": 0.8653978445310708,
      "grad_norm": 0.1650599254629266,
      "learning_rate": 4.6767668790333675e-05,
      "loss": 3.4204,
      "step": 3774
    },
    {
      "epoch": 0.8656271497362991,
      "grad_norm": 0.1640087389840356,
      "learning_rate": 4.6610980918406596e-05,
      "loss": 3.5174,
      "step": 3775
    },
    {
      "epoch": 0.8658564549415272,
      "grad_norm": 0.17604873740588375,
      "learning_rate": 4.645454313282965e-05,
      "loss": 3.4272,
      "step": 3776
    },
    {
      "epoch": 0.8660857601467553,
      "grad_norm": 0.16334972487917648,
      "learning_rate": 4.629835551989276e-05,
      "loss": 3.4899,
      "step": 3777
    },
    {
      "epoch": 0.8663150653519834,
      "grad_norm": 0.1777386027361396,
      "learning_rate": 4.614241816574799e-05,
      "loss": 3.5359,
      "step": 3778
    },
    {
      "epoch": 0.8665443705572117,
      "grad_norm": 0.16594374999759287,
      "learning_rate": 4.5986731156409224e-05,
      "loss": 3.3143,
      "step": 3779
    },
    {
      "epoch": 0.8667736757624398,
      "grad_norm": 0.16298148421499994,
      "learning_rate": 4.583129457775253e-05,
      "loss": 3.4678,
      "step": 3780
    },
    {
      "epoch": 0.8670029809676679,
      "grad_norm": 0.17621431965874676,
      "learning_rate": 4.567610851551568e-05,
      "loss": 3.4152,
      "step": 3781
    },
    {
      "epoch": 0.8672322861728962,
      "grad_norm": 0.17334732670068836,
      "learning_rate": 4.5521173055298234e-05,
      "loss": 3.4734,
      "step": 3782
    },
    {
      "epoch": 0.8674615913781243,
      "grad_norm": 0.15783868086809263,
      "learning_rate": 4.536648828256146e-05,
      "loss": 3.5602,
      "step": 3783
    },
    {
      "epoch": 0.8676908965833524,
      "grad_norm": 0.17407706459240804,
      "learning_rate": 4.52120542826287e-05,
      "loss": 3.4222,
      "step": 3784
    },
    {
      "epoch": 0.8679202017885806,
      "grad_norm": 0.1749658569115017,
      "learning_rate": 4.505787114068433e-05,
      "loss": 3.6088,
      "step": 3785
    },
    {
      "epoch": 0.8681495069938088,
      "grad_norm": 0.15689629552398754,
      "learning_rate": 4.490393894177508e-05,
      "loss": 3.4505,
      "step": 3786
    },
    {
      "epoch": 0.8683788121990369,
      "grad_norm": 0.1741852386827198,
      "learning_rate": 4.4750257770808764e-05,
      "loss": 3.3948,
      "step": 3787
    },
    {
      "epoch": 0.868608117404265,
      "grad_norm": 0.18381277154630915,
      "learning_rate": 4.459682771255497e-05,
      "loss": 3.5057,
      "step": 3788
    },
    {
      "epoch": 0.8688374226094933,
      "grad_norm": 0.17356494166135283,
      "learning_rate": 4.444364885164448e-05,
      "loss": 3.5016,
      "step": 3789
    },
    {
      "epoch": 0.8690667278147214,
      "grad_norm": 0.1585731846230267,
      "learning_rate": 4.4290721272570056e-05,
      "loss": 3.3972,
      "step": 3790
    },
    {
      "epoch": 0.8692960330199495,
      "grad_norm": 0.18927794147717528,
      "learning_rate": 4.413804505968533e-05,
      "loss": 3.5212,
      "step": 3791
    },
    {
      "epoch": 0.8695253382251777,
      "grad_norm": 0.16837393298288297,
      "learning_rate": 4.3985620297205487e-05,
      "loss": 3.4788,
      "step": 3792
    },
    {
      "epoch": 0.8697546434304059,
      "grad_norm": 0.16072122929170937,
      "learning_rate": 4.3833447069206944e-05,
      "loss": 3.3463,
      "step": 3793
    },
    {
      "epoch": 0.869983948635634,
      "grad_norm": 0.16228260523731103,
      "learning_rate": 4.3681525459627616e-05,
      "loss": 3.3682,
      "step": 3794
    },
    {
      "epoch": 0.8702132538408622,
      "grad_norm": 0.17288047092152542,
      "learning_rate": 4.352985555226635e-05,
      "loss": 3.401,
      "step": 3795
    },
    {
      "epoch": 0.8704425590460904,
      "grad_norm": 0.17730149094199912,
      "learning_rate": 4.337843743078329e-05,
      "loss": 3.5471,
      "step": 3796
    },
    {
      "epoch": 0.8706718642513185,
      "grad_norm": 0.17197389279979236,
      "learning_rate": 4.322727117869951e-05,
      "loss": 3.444,
      "step": 3797
    },
    {
      "epoch": 0.8709011694565466,
      "grad_norm": 0.17260639634170347,
      "learning_rate": 4.3076356879397594e-05,
      "loss": 3.6108,
      "step": 3798
    },
    {
      "epoch": 0.8711304746617748,
      "grad_norm": 0.16885798128757937,
      "learning_rate": 4.29256946161205e-05,
      "loss": 3.3187,
      "step": 3799
    },
    {
      "epoch": 0.871359779867003,
      "grad_norm": 0.16215524130813558,
      "learning_rate": 4.277528447197282e-05,
      "loss": 3.3973,
      "step": 3800
    },
    {
      "epoch": 0.8715890850722311,
      "grad_norm": 0.18233887091094997,
      "learning_rate": 4.262512652991968e-05,
      "loss": 3.3898,
      "step": 3801
    },
    {
      "epoch": 0.8718183902774593,
      "grad_norm": 0.16192691751714394,
      "learning_rate": 4.247522087278716e-05,
      "loss": 3.4377,
      "step": 3802
    },
    {
      "epoch": 0.8720476954826875,
      "grad_norm": 0.17490532714070844,
      "learning_rate": 4.2325567583262113e-05,
      "loss": 3.5421,
      "step": 3803
    },
    {
      "epoch": 0.8722770006879156,
      "grad_norm": 0.1696159029769085,
      "learning_rate": 4.2176166743892576e-05,
      "loss": 3.4552,
      "step": 3804
    },
    {
      "epoch": 0.8725063058931438,
      "grad_norm": 0.19001114348558443,
      "learning_rate": 4.2027018437086895e-05,
      "loss": 3.4808,
      "step": 3805
    },
    {
      "epoch": 0.8727356110983719,
      "grad_norm": 0.17642362881381085,
      "learning_rate": 4.1878122745114265e-05,
      "loss": 3.5818,
      "step": 3806
    },
    {
      "epoch": 0.8729649163036001,
      "grad_norm": 0.16677470402375263,
      "learning_rate": 4.172947975010449e-05,
      "loss": 3.383,
      "step": 3807
    },
    {
      "epoch": 0.8731942215088283,
      "grad_norm": 0.17226599280229926,
      "learning_rate": 4.158108953404832e-05,
      "loss": 3.3359,
      "step": 3808
    },
    {
      "epoch": 0.8734235267140564,
      "grad_norm": 0.1528964912969297,
      "learning_rate": 4.143295217879645e-05,
      "loss": 3.4325,
      "step": 3809
    },
    {
      "epoch": 0.8736528319192846,
      "grad_norm": 0.17041324401454888,
      "learning_rate": 4.128506776606067e-05,
      "loss": 3.4865,
      "step": 3810
    },
    {
      "epoch": 0.8738821371245127,
      "grad_norm": 0.1640316109298882,
      "learning_rate": 4.113743637741296e-05,
      "loss": 3.5386,
      "step": 3811
    },
    {
      "epoch": 0.8741114423297409,
      "grad_norm": 0.1726089067759072,
      "learning_rate": 4.0990058094285966e-05,
      "loss": 3.5829,
      "step": 3812
    },
    {
      "epoch": 0.874340747534969,
      "grad_norm": 0.1669137054214936,
      "learning_rate": 4.084293299797226e-05,
      "loss": 3.4493,
      "step": 3813
    },
    {
      "epoch": 0.8745700527401972,
      "grad_norm": 0.18815436070034106,
      "learning_rate": 4.069606116962532e-05,
      "loss": 3.4296,
      "step": 3814
    },
    {
      "epoch": 0.8747993579454254,
      "grad_norm": 0.16794833060610412,
      "learning_rate": 4.054944269025862e-05,
      "loss": 3.506,
      "step": 3815
    },
    {
      "epoch": 0.8750286631506535,
      "grad_norm": 0.15740351636005065,
      "learning_rate": 4.0403077640745855e-05,
      "loss": 3.4672,
      "step": 3816
    },
    {
      "epoch": 0.8752579683558817,
      "grad_norm": 0.1599136087330241,
      "learning_rate": 4.025696610182095e-05,
      "loss": 3.4587,
      "step": 3817
    },
    {
      "epoch": 0.8754872735611099,
      "grad_norm": 0.16661288856203402,
      "learning_rate": 4.0111108154078205e-05,
      "loss": 3.3969,
      "step": 3818
    },
    {
      "epoch": 0.875716578766338,
      "grad_norm": 0.1685400433542339,
      "learning_rate": 3.996550387797187e-05,
      "loss": 3.4885,
      "step": 3819
    },
    {
      "epoch": 0.8759458839715661,
      "grad_norm": 0.16411351209105754,
      "learning_rate": 3.9820153353816226e-05,
      "loss": 3.3942,
      "step": 3820
    },
    {
      "epoch": 0.8761751891767943,
      "grad_norm": 0.18921225107496795,
      "learning_rate": 3.9675056661785556e-05,
      "loss": 3.4031,
      "step": 3821
    },
    {
      "epoch": 0.8764044943820225,
      "grad_norm": 0.16830239479355263,
      "learning_rate": 3.9530213881914576e-05,
      "loss": 3.4308,
      "step": 3822
    },
    {
      "epoch": 0.8766337995872506,
      "grad_norm": 0.18038251613747475,
      "learning_rate": 3.9385625094097154e-05,
      "loss": 3.4019,
      "step": 3823
    },
    {
      "epoch": 0.8768631047924788,
      "grad_norm": 0.16082193117955998,
      "learning_rate": 3.924129037808777e-05,
      "loss": 3.4957,
      "step": 3824
    },
    {
      "epoch": 0.877092409997707,
      "grad_norm": 0.17695222565895172,
      "learning_rate": 3.909720981350034e-05,
      "loss": 3.4058,
      "step": 3825
    },
    {
      "epoch": 0.8773217152029351,
      "grad_norm": 0.18264678764025416,
      "learning_rate": 3.895338347980898e-05,
      "loss": 3.6451,
      "step": 3826
    },
    {
      "epoch": 0.8775510204081632,
      "grad_norm": 0.18662995615866573,
      "learning_rate": 3.880981145634704e-05,
      "loss": 3.4681,
      "step": 3827
    },
    {
      "epoch": 0.8777803256133915,
      "grad_norm": 0.17654360365204577,
      "learning_rate": 3.8666493822308077e-05,
      "loss": 3.4325,
      "step": 3828
    },
    {
      "epoch": 0.8780096308186196,
      "grad_norm": 0.1682149922223464,
      "learning_rate": 3.852343065674507e-05,
      "loss": 3.3785,
      "step": 3829
    },
    {
      "epoch": 0.8782389360238477,
      "grad_norm": 0.16865138985122952,
      "learning_rate": 3.838062203857073e-05,
      "loss": 3.4447,
      "step": 3830
    },
    {
      "epoch": 0.8784682412290759,
      "grad_norm": 0.18052750805423406,
      "learning_rate": 3.8238068046557276e-05,
      "loss": 3.422,
      "step": 3831
    },
    {
      "epoch": 0.8786975464343041,
      "grad_norm": 0.172450608279686,
      "learning_rate": 3.8095768759336626e-05,
      "loss": 3.5115,
      "step": 3832
    },
    {
      "epoch": 0.8789268516395322,
      "grad_norm": 0.16862328590205872,
      "learning_rate": 3.795372425540006e-05,
      "loss": 3.5752,
      "step": 3833
    },
    {
      "epoch": 0.8791561568447603,
      "grad_norm": 0.1569800408065627,
      "learning_rate": 3.7811934613098394e-05,
      "loss": 3.5538,
      "step": 3834
    },
    {
      "epoch": 0.8793854620499886,
      "grad_norm": 0.16188565838156466,
      "learning_rate": 3.76703999106418e-05,
      "loss": 3.2997,
      "step": 3835
    },
    {
      "epoch": 0.8796147672552167,
      "grad_norm": 0.17280173993341924,
      "learning_rate": 3.752912022610006e-05,
      "loss": 3.5203,
      "step": 3836
    },
    {
      "epoch": 0.8798440724604448,
      "grad_norm": 0.18793229192856725,
      "learning_rate": 3.7388095637401754e-05,
      "loss": 3.5449,
      "step": 3837
    },
    {
      "epoch": 0.8800733776656731,
      "grad_norm": 0.17407220993346856,
      "learning_rate": 3.72473262223354e-05,
      "loss": 3.4554,
      "step": 3838
    },
    {
      "epoch": 0.8803026828709012,
      "grad_norm": 0.1645987018320664,
      "learning_rate": 3.7106812058548376e-05,
      "loss": 3.3301,
      "step": 3839
    },
    {
      "epoch": 0.8805319880761293,
      "grad_norm": 0.17322236913214173,
      "learning_rate": 3.696655322354736e-05,
      "loss": 3.4799,
      "step": 3840
    },
    {
      "epoch": 0.8807612932813574,
      "grad_norm": 0.16490794481832363,
      "learning_rate": 3.682654979469807e-05,
      "loss": 3.4244,
      "step": 3841
    },
    {
      "epoch": 0.8809905984865857,
      "grad_norm": 0.1860591005571536,
      "learning_rate": 3.668680184922563e-05,
      "loss": 3.4755,
      "step": 3842
    },
    {
      "epoch": 0.8812199036918138,
      "grad_norm": 0.17663636474672187,
      "learning_rate": 3.654730946421403e-05,
      "loss": 3.3725,
      "step": 3843
    },
    {
      "epoch": 0.8814492088970419,
      "grad_norm": 0.16372093324235523,
      "learning_rate": 3.6408072716606344e-05,
      "loss": 3.5871,
      "step": 3844
    },
    {
      "epoch": 0.8816785141022702,
      "grad_norm": 0.1781885377990395,
      "learning_rate": 3.6269091683204466e-05,
      "loss": 3.4925,
      "step": 3845
    },
    {
      "epoch": 0.8819078193074983,
      "grad_norm": 0.16671585070550773,
      "learning_rate": 3.613036644066969e-05,
      "loss": 3.4737,
      "step": 3846
    },
    {
      "epoch": 0.8821371245127264,
      "grad_norm": 0.1704294027009498,
      "learning_rate": 3.5991897065521693e-05,
      "loss": 3.377,
      "step": 3847
    },
    {
      "epoch": 0.8823664297179546,
      "grad_norm": 0.172320922802505,
      "learning_rate": 3.5853683634139434e-05,
      "loss": 3.476,
      "step": 3848
    },
    {
      "epoch": 0.8825957349231828,
      "grad_norm": 0.1658383609221704,
      "learning_rate": 3.571572622276026e-05,
      "loss": 3.5486,
      "step": 3849
    },
    {
      "epoch": 0.8828250401284109,
      "grad_norm": 0.1640463371680034,
      "learning_rate": 3.557802490748097e-05,
      "loss": 3.5749,
      "step": 3850
    },
    {
      "epoch": 0.883054345333639,
      "grad_norm": 0.17675702814734734,
      "learning_rate": 3.544057976425619e-05,
      "loss": 3.4351,
      "step": 3851
    },
    {
      "epoch": 0.8832836505388673,
      "grad_norm": 0.16651663496482824,
      "learning_rate": 3.5303390868900164e-05,
      "loss": 3.4971,
      "step": 3852
    },
    {
      "epoch": 0.8835129557440954,
      "grad_norm": 0.17199979958967418,
      "learning_rate": 3.5166458297085146e-05,
      "loss": 3.2602,
      "step": 3853
    },
    {
      "epoch": 0.8837422609493235,
      "grad_norm": 0.1658890372744537,
      "learning_rate": 3.502978212434227e-05,
      "loss": 3.3325,
      "step": 3854
    },
    {
      "epoch": 0.8839715661545517,
      "grad_norm": 0.1842638596920117,
      "learning_rate": 3.489336242606111e-05,
      "loss": 3.6237,
      "step": 3855
    },
    {
      "epoch": 0.8842008713597799,
      "grad_norm": 0.1641378995731501,
      "learning_rate": 3.4757199277490105e-05,
      "loss": 3.3381,
      "step": 3856
    },
    {
      "epoch": 0.884430176565008,
      "grad_norm": 0.17121872800101484,
      "learning_rate": 3.462129275373577e-05,
      "loss": 3.4615,
      "step": 3857
    },
    {
      "epoch": 0.8846594817702362,
      "grad_norm": 0.16323679208822856,
      "learning_rate": 3.448564292976319e-05,
      "loss": 3.3867,
      "step": 3858
    },
    {
      "epoch": 0.8848887869754644,
      "grad_norm": 0.15921036991403495,
      "learning_rate": 3.4350249880395924e-05,
      "loss": 3.4584,
      "step": 3859
    },
    {
      "epoch": 0.8851180921806925,
      "grad_norm": 0.1641670663154659,
      "learning_rate": 3.4215113680316046e-05,
      "loss": 3.5844,
      "step": 3860
    },
    {
      "epoch": 0.8853473973859206,
      "grad_norm": 0.1704161335754675,
      "learning_rate": 3.408023440406355e-05,
      "loss": 3.4115,
      "step": 3861
    },
    {
      "epoch": 0.8855767025911488,
      "grad_norm": 0.18309647332919585,
      "learning_rate": 3.394561212603708e-05,
      "loss": 3.4067,
      "step": 3862
    },
    {
      "epoch": 0.885806007796377,
      "grad_norm": 0.17679624972945,
      "learning_rate": 3.381124692049331e-05,
      "loss": 3.3773,
      "step": 3863
    },
    {
      "epoch": 0.8860353130016051,
      "grad_norm": 0.15878326701535408,
      "learning_rate": 3.367713886154733e-05,
      "loss": 3.3786,
      "step": 3864
    },
    {
      "epoch": 0.8862646182068333,
      "grad_norm": 0.17458185093342884,
      "learning_rate": 3.354328802317197e-05,
      "loss": 3.4012,
      "step": 3865
    },
    {
      "epoch": 0.8864939234120615,
      "grad_norm": 0.176753142976038,
      "learning_rate": 3.340969447919873e-05,
      "loss": 3.4602,
      "step": 3866
    },
    {
      "epoch": 0.8867232286172896,
      "grad_norm": 0.1753856150671882,
      "learning_rate": 3.327635830331677e-05,
      "loss": 3.437,
      "step": 3867
    },
    {
      "epoch": 0.8869525338225178,
      "grad_norm": 0.16881971208915697,
      "learning_rate": 3.314327956907354e-05,
      "loss": 3.4242,
      "step": 3868
    },
    {
      "epoch": 0.8871818390277459,
      "grad_norm": 0.16619521683998068,
      "learning_rate": 3.3010458349874206e-05,
      "loss": 3.3114,
      "step": 3869
    },
    {
      "epoch": 0.8874111442329741,
      "grad_norm": 0.18298437498062267,
      "learning_rate": 3.287789471898223e-05,
      "loss": 3.405,
      "step": 3870
    },
    {
      "epoch": 0.8876404494382022,
      "grad_norm": 0.16937568335886055,
      "learning_rate": 3.2745588749518775e-05,
      "loss": 3.4725,
      "step": 3871
    },
    {
      "epoch": 0.8878697546434304,
      "grad_norm": 0.17440251680739705,
      "learning_rate": 3.2613540514462914e-05,
      "loss": 3.3389,
      "step": 3872
    },
    {
      "epoch": 0.8880990598486586,
      "grad_norm": 0.1572190782835771,
      "learning_rate": 3.248175008665161e-05,
      "loss": 3.5849,
      "step": 3873
    },
    {
      "epoch": 0.8883283650538867,
      "grad_norm": 0.17643421546733412,
      "learning_rate": 3.235021753877942e-05,
      "loss": 3.3647,
      "step": 3874
    },
    {
      "epoch": 0.8885576702591149,
      "grad_norm": 0.2115011016628796,
      "learning_rate": 3.221894294339911e-05,
      "loss": 3.5018,
      "step": 3875
    },
    {
      "epoch": 0.888786975464343,
      "grad_norm": 0.18422923500831703,
      "learning_rate": 3.208792637292057e-05,
      "loss": 3.5429,
      "step": 3876
    },
    {
      "epoch": 0.8890162806695712,
      "grad_norm": 0.16230221326812513,
      "learning_rate": 3.1957167899611836e-05,
      "loss": 3.491,
      "step": 3877
    },
    {
      "epoch": 0.8892455858747994,
      "grad_norm": 0.1747422895394056,
      "learning_rate": 3.1826667595598435e-05,
      "loss": 3.4733,
      "step": 3878
    },
    {
      "epoch": 0.8894748910800275,
      "grad_norm": 0.17468811835799017,
      "learning_rate": 3.169642553286334e-05,
      "loss": 3.3846,
      "step": 3879
    },
    {
      "epoch": 0.8897041962852557,
      "grad_norm": 0.17581265753480424,
      "learning_rate": 3.156644178324719e-05,
      "loss": 3.475,
      "step": 3880
    },
    {
      "epoch": 0.8899335014904838,
      "grad_norm": 0.17626495125674868,
      "learning_rate": 3.143671641844831e-05,
      "loss": 3.4627,
      "step": 3881
    },
    {
      "epoch": 0.890162806695712,
      "grad_norm": 0.2088723479256713,
      "learning_rate": 3.1307249510022225e-05,
      "loss": 3.3495,
      "step": 3882
    },
    {
      "epoch": 0.8903921119009401,
      "grad_norm": 0.16579323817051164,
      "learning_rate": 3.117804112938205e-05,
      "loss": 3.3713,
      "step": 3883
    },
    {
      "epoch": 0.8906214171061683,
      "grad_norm": 0.17171479721514807,
      "learning_rate": 3.104909134779821e-05,
      "loss": 3.36,
      "step": 3884
    },
    {
      "epoch": 0.8908507223113965,
      "grad_norm": 0.17520697552842132,
      "learning_rate": 3.092040023639869e-05,
      "loss": 3.4392,
      "step": 3885
    },
    {
      "epoch": 0.8910800275166246,
      "grad_norm": 0.18111928625675,
      "learning_rate": 3.079196786616839e-05,
      "loss": 3.4065,
      "step": 3886
    },
    {
      "epoch": 0.8913093327218528,
      "grad_norm": 0.16886021497256462,
      "learning_rate": 3.066379430795002e-05,
      "loss": 3.3534,
      "step": 3887
    },
    {
      "epoch": 0.891538637927081,
      "grad_norm": 0.19344299316260585,
      "learning_rate": 3.053587963244303e-05,
      "loss": 3.4602,
      "step": 3888
    },
    {
      "epoch": 0.8917679431323091,
      "grad_norm": 0.16377810255178446,
      "learning_rate": 3.040822391020459e-05,
      "loss": 3.4069,
      "step": 3889
    },
    {
      "epoch": 0.8919972483375372,
      "grad_norm": 0.1736770779472014,
      "learning_rate": 3.0280827211648377e-05,
      "loss": 3.391,
      "step": 3890
    },
    {
      "epoch": 0.8922265535427654,
      "grad_norm": 0.16371176508582275,
      "learning_rate": 3.0153689607045842e-05,
      "loss": 3.4279,
      "step": 3891
    },
    {
      "epoch": 0.8924558587479936,
      "grad_norm": 0.17303001197814774,
      "learning_rate": 3.0026811166525114e-05,
      "loss": 3.4973,
      "step": 3892
    },
    {
      "epoch": 0.8926851639532217,
      "grad_norm": 0.1564938114903408,
      "learning_rate": 2.9900191960071545e-05,
      "loss": 3.2561,
      "step": 3893
    },
    {
      "epoch": 0.8929144691584499,
      "grad_norm": 0.1788653387050378,
      "learning_rate": 2.9773832057527263e-05,
      "loss": 3.468,
      "step": 3894
    },
    {
      "epoch": 0.8931437743636781,
      "grad_norm": 0.16883317144902935,
      "learning_rate": 2.9647731528591848e-05,
      "loss": 3.323,
      "step": 3895
    },
    {
      "epoch": 0.8933730795689062,
      "grad_norm": 0.1632595902430215,
      "learning_rate": 2.9521890442821275e-05,
      "loss": 3.3099,
      "step": 3896
    },
    {
      "epoch": 0.8936023847741343,
      "grad_norm": 0.16233536293518377,
      "learning_rate": 2.9396308869628795e-05,
      "loss": 3.3556,
      "step": 3897
    },
    {
      "epoch": 0.8938316899793626,
      "grad_norm": 0.15995652971072144,
      "learning_rate": 2.9270986878284167e-05,
      "loss": 3.4693,
      "step": 3898
    },
    {
      "epoch": 0.8940609951845907,
      "grad_norm": 0.16276535001806883,
      "learning_rate": 2.914592453791448e-05,
      "loss": 3.3345,
      "step": 3899
    },
    {
      "epoch": 0.8942903003898188,
      "grad_norm": 0.18319528108262284,
      "learning_rate": 2.902112191750289e-05,
      "loss": 3.441,
      "step": 3900
    },
    {
      "epoch": 0.8945196055950471,
      "grad_norm": 0.1589773110836166,
      "learning_rate": 2.8896579085889994e-05,
      "loss": 3.4891,
      "step": 3901
    },
    {
      "epoch": 0.8947489108002752,
      "grad_norm": 0.159813762116895,
      "learning_rate": 2.8772296111772677e-05,
      "loss": 3.496,
      "step": 3902
    },
    {
      "epoch": 0.8949782160055033,
      "grad_norm": 0.17170366359592762,
      "learning_rate": 2.86482730637046e-05,
      "loss": 3.4618,
      "step": 3903
    },
    {
      "epoch": 0.8952075212107314,
      "grad_norm": 0.17657717556463295,
      "learning_rate": 2.8524510010095984e-05,
      "loss": 3.5794,
      "step": 3904
    },
    {
      "epoch": 0.8954368264159597,
      "grad_norm": 0.19396981763325416,
      "learning_rate": 2.840100701921383e-05,
      "loss": 3.4692,
      "step": 3905
    },
    {
      "epoch": 0.8956661316211878,
      "grad_norm": 0.16832277183634817,
      "learning_rate": 2.8277764159181486e-05,
      "loss": 3.2768,
      "step": 3906
    },
    {
      "epoch": 0.8958954368264159,
      "grad_norm": 0.1678456373202823,
      "learning_rate": 2.8154781497978898e-05,
      "loss": 3.6134,
      "step": 3907
    },
    {
      "epoch": 0.8961247420316442,
      "grad_norm": 0.17169754726595,
      "learning_rate": 2.8032059103442364e-05,
      "loss": 3.3795,
      "step": 3908
    },
    {
      "epoch": 0.8963540472368723,
      "grad_norm": 0.1889602287120023,
      "learning_rate": 2.7909597043265013e-05,
      "loss": 3.5045,
      "step": 3909
    },
    {
      "epoch": 0.8965833524421004,
      "grad_norm": 0.18537943846046845,
      "learning_rate": 2.7787395384995862e-05,
      "loss": 3.4128,
      "step": 3910
    },
    {
      "epoch": 0.8968126576473286,
      "grad_norm": 0.16348708671420867,
      "learning_rate": 2.7665454196040662e-05,
      "loss": 3.5049,
      "step": 3911
    },
    {
      "epoch": 0.8970419628525568,
      "grad_norm": 0.17445862331600415,
      "learning_rate": 2.7543773543661167e-05,
      "loss": 3.3853,
      "step": 3912
    },
    {
      "epoch": 0.8972712680577849,
      "grad_norm": 0.15980426136296105,
      "learning_rate": 2.7422353494975905e-05,
      "loss": 3.3234,
      "step": 3913
    },
    {
      "epoch": 0.897500573263013,
      "grad_norm": 0.17554771528156746,
      "learning_rate": 2.7301194116959084e-05,
      "loss": 3.5021,
      "step": 3914
    },
    {
      "epoch": 0.8977298784682413,
      "grad_norm": 0.16888130100944967,
      "learning_rate": 2.7180295476441573e-05,
      "loss": 3.5372,
      "step": 3915
    },
    {
      "epoch": 0.8979591836734694,
      "grad_norm": 0.1653433892614373,
      "learning_rate": 2.70596576401102e-05,
      "loss": 3.5303,
      "step": 3916
    },
    {
      "epoch": 0.8981884888786975,
      "grad_norm": 0.17965390545028187,
      "learning_rate": 2.6939280674508016e-05,
      "loss": 3.4767,
      "step": 3917
    },
    {
      "epoch": 0.8984177940839257,
      "grad_norm": 0.1589398078452716,
      "learning_rate": 2.681916464603401e-05,
      "loss": 3.4571,
      "step": 3918
    },
    {
      "epoch": 0.8986470992891539,
      "grad_norm": 0.16956778624592383,
      "learning_rate": 2.669930962094358e-05,
      "loss": 3.4325,
      "step": 3919
    },
    {
      "epoch": 0.898876404494382,
      "grad_norm": 0.17567183280493093,
      "learning_rate": 2.6579715665347894e-05,
      "loss": 3.4482,
      "step": 3920
    },
    {
      "epoch": 0.8991057096996102,
      "grad_norm": 0.16552048789079693,
      "learning_rate": 2.6460382845214126e-05,
      "loss": 3.4974,
      "step": 3921
    },
    {
      "epoch": 0.8993350149048384,
      "grad_norm": 0.18201605983385571,
      "learning_rate": 2.6341311226365406e-05,
      "loss": 3.6373,
      "step": 3922
    },
    {
      "epoch": 0.8995643201100665,
      "grad_norm": 0.17024667166421292,
      "learning_rate": 2.6222500874481025e-05,
      "loss": 3.4639,
      "step": 3923
    },
    {
      "epoch": 0.8997936253152946,
      "grad_norm": 0.18354620678386296,
      "learning_rate": 2.6103951855095897e-05,
      "loss": 3.3256,
      "step": 3924
    },
    {
      "epoch": 0.9000229305205228,
      "grad_norm": 0.18405580586574,
      "learning_rate": 2.5985664233600827e-05,
      "loss": 3.4532,
      "step": 3925
    },
    {
      "epoch": 0.900252235725751,
      "grad_norm": 0.20163842082329805,
      "learning_rate": 2.586763807524245e-05,
      "loss": 3.4654,
      "step": 3926
    },
    {
      "epoch": 0.9004815409309791,
      "grad_norm": 0.17087627435770084,
      "learning_rate": 2.574987344512336e-05,
      "loss": 3.4567,
      "step": 3927
    },
    {
      "epoch": 0.9007108461362073,
      "grad_norm": 0.1644768400926769,
      "learning_rate": 2.5632370408201488e-05,
      "loss": 3.4078,
      "step": 3928
    },
    {
      "epoch": 0.9009401513414355,
      "grad_norm": 0.18431593397157228,
      "learning_rate": 2.5515129029290984e-05,
      "loss": 3.5765,
      "step": 3929
    },
    {
      "epoch": 0.9011694565466636,
      "grad_norm": 0.17915087879004854,
      "learning_rate": 2.539814937306123e-05,
      "loss": 3.384,
      "step": 3930
    },
    {
      "epoch": 0.9013987617518918,
      "grad_norm": 0.15930676139576347,
      "learning_rate": 2.5281431504037556e-05,
      "loss": 3.5101,
      "step": 3931
    },
    {
      "epoch": 0.90162806695712,
      "grad_norm": 0.16246201436799954,
      "learning_rate": 2.5164975486600572e-05,
      "loss": 3.3606,
      "step": 3932
    },
    {
      "epoch": 0.9018573721623481,
      "grad_norm": 0.16940595638085662,
      "learning_rate": 2.504878138498684e-05,
      "loss": 3.5288,
      "step": 3933
    },
    {
      "epoch": 0.9020866773675762,
      "grad_norm": 0.17206210094892266,
      "learning_rate": 2.4932849263288147e-05,
      "loss": 3.3498,
      "step": 3934
    },
    {
      "epoch": 0.9023159825728044,
      "grad_norm": 0.17970508922657683,
      "learning_rate": 2.48171791854519e-05,
      "loss": 3.3339,
      "step": 3935
    },
    {
      "epoch": 0.9025452877780326,
      "grad_norm": 0.1763533540251536,
      "learning_rate": 2.470177121528089e-05,
      "loss": 3.4466,
      "step": 3936
    },
    {
      "epoch": 0.9027745929832607,
      "grad_norm": 0.17308074553310368,
      "learning_rate": 2.4586625416433473e-05,
      "loss": 3.4358,
      "step": 3937
    },
    {
      "epoch": 0.9030038981884889,
      "grad_norm": 0.17513225571714425,
      "learning_rate": 2.4471741852423235e-05,
      "loss": 3.3939,
      "step": 3938
    },
    {
      "epoch": 0.903233203393717,
      "grad_norm": 0.15396003121551852,
      "learning_rate": 2.435712058661921e-05,
      "loss": 3.491,
      "step": 3939
    },
    {
      "epoch": 0.9034625085989452,
      "grad_norm": 0.1758133644805841,
      "learning_rate": 2.424276168224565e-05,
      "loss": 3.353,
      "step": 3940
    },
    {
      "epoch": 0.9036918138041734,
      "grad_norm": 0.1793735887081094,
      "learning_rate": 2.4128665202382327e-05,
      "loss": 3.4768,
      "step": 3941
    },
    {
      "epoch": 0.9039211190094015,
      "grad_norm": 0.15949845910585372,
      "learning_rate": 2.4014831209963893e-05,
      "loss": 3.3552,
      "step": 3942
    },
    {
      "epoch": 0.9041504242146297,
      "grad_norm": 0.1592648117707493,
      "learning_rate": 2.3901259767780515e-05,
      "loss": 3.3975,
      "step": 3943
    },
    {
      "epoch": 0.9043797294198578,
      "grad_norm": 0.16881904201256548,
      "learning_rate": 2.378795093847752e-05,
      "loss": 3.2705,
      "step": 3944
    },
    {
      "epoch": 0.904609034625086,
      "grad_norm": 0.1705638574573193,
      "learning_rate": 2.367490478455514e-05,
      "loss": 3.364,
      "step": 3945
    },
    {
      "epoch": 0.9048383398303141,
      "grad_norm": 0.16094936911948068,
      "learning_rate": 2.3562121368368937e-05,
      "loss": 3.362,
      "step": 3946
    },
    {
      "epoch": 0.9050676450355423,
      "grad_norm": 0.178717852093636,
      "learning_rate": 2.3449600752129597e-05,
      "loss": 3.4593,
      "step": 3947
    },
    {
      "epoch": 0.9052969502407705,
      "grad_norm": 0.16002606187054982,
      "learning_rate": 2.3337342997902633e-05,
      "loss": 3.3705,
      "step": 3948
    },
    {
      "epoch": 0.9055262554459986,
      "grad_norm": 0.17874299263903212,
      "learning_rate": 2.3225348167608685e-05,
      "loss": 3.5709,
      "step": 3949
    },
    {
      "epoch": 0.9057555606512268,
      "grad_norm": 0.16600129791312704,
      "learning_rate": 2.3113616323023277e-05,
      "loss": 3.4619,
      "step": 3950
    },
    {
      "epoch": 0.905984865856455,
      "grad_norm": 0.17853104587501267,
      "learning_rate": 2.3002147525777118e-05,
      "loss": 3.6023,
      "step": 3951
    },
    {
      "epoch": 0.9062141710616831,
      "grad_norm": 0.17238525846241648,
      "learning_rate": 2.2890941837355518e-05,
      "loss": 3.4307,
      "step": 3952
    },
    {
      "epoch": 0.9064434762669112,
      "grad_norm": 0.1853462985721937,
      "learning_rate": 2.2779999319098856e-05,
      "loss": 3.4656,
      "step": 3953
    },
    {
      "epoch": 0.9066727814721394,
      "grad_norm": 0.1647736983614884,
      "learning_rate": 2.2669320032202178e-05,
      "loss": 3.4578,
      "step": 3954
    },
    {
      "epoch": 0.9069020866773676,
      "grad_norm": 0.18149664584152783,
      "learning_rate": 2.255890403771571e-05,
      "loss": 3.5456,
      "step": 3955
    },
    {
      "epoch": 0.9071313918825957,
      "grad_norm": 0.1609179994529313,
      "learning_rate": 2.2448751396543787e-05,
      "loss": 3.351,
      "step": 3956
    },
    {
      "epoch": 0.9073606970878239,
      "grad_norm": 0.18658326589641472,
      "learning_rate": 2.233886216944614e-05,
      "loss": 3.3883,
      "step": 3957
    },
    {
      "epoch": 0.9075900022930521,
      "grad_norm": 0.1725270535672448,
      "learning_rate": 2.222923641703689e-05,
      "loss": 3.2864,
      "step": 3958
    },
    {
      "epoch": 0.9078193074982802,
      "grad_norm": 0.16511379334450246,
      "learning_rate": 2.211987419978484e-05,
      "loss": 3.3039,
      "step": 3959
    },
    {
      "epoch": 0.9080486127035083,
      "grad_norm": 0.16709219417038715,
      "learning_rate": 2.2010775578013343e-05,
      "loss": 3.4006,
      "step": 3960
    },
    {
      "epoch": 0.9082779179087366,
      "grad_norm": 0.1630400629958854,
      "learning_rate": 2.1901940611900705e-05,
      "loss": 3.3619,
      "step": 3961
    },
    {
      "epoch": 0.9085072231139647,
      "grad_norm": 0.17369220131716723,
      "learning_rate": 2.179336936147941e-05,
      "loss": 3.3903,
      "step": 3962
    },
    {
      "epoch": 0.9087365283191928,
      "grad_norm": 0.1702905451672036,
      "learning_rate": 2.168506188663666e-05,
      "loss": 3.5561,
      "step": 3963
    },
    {
      "epoch": 0.908965833524421,
      "grad_norm": 0.18328475774033579,
      "learning_rate": 2.157701824711411e-05,
      "loss": 3.4057,
      "step": 3964
    },
    {
      "epoch": 0.9091951387296492,
      "grad_norm": 0.16665990446359175,
      "learning_rate": 2.1469238502507925e-05,
      "loss": 3.4137,
      "step": 3965
    },
    {
      "epoch": 0.9094244439348773,
      "grad_norm": 0.16770894396838892,
      "learning_rate": 2.136172271226877e-05,
      "loss": 3.4684,
      "step": 3966
    },
    {
      "epoch": 0.9096537491401054,
      "grad_norm": 0.17687276458247994,
      "learning_rate": 2.125447093570154e-05,
      "loss": 3.2944,
      "step": 3967
    },
    {
      "epoch": 0.9098830543453337,
      "grad_norm": 0.15255719917163557,
      "learning_rate": 2.114748323196558e-05,
      "loss": 3.3255,
      "step": 3968
    },
    {
      "epoch": 0.9101123595505618,
      "grad_norm": 0.17172581548976518,
      "learning_rate": 2.1040759660074793e-05,
      "loss": 3.4232,
      "step": 3969
    },
    {
      "epoch": 0.9103416647557899,
      "grad_norm": 0.17581458360038169,
      "learning_rate": 2.0934300278896867e-05,
      "loss": 3.3435,
      "step": 3970
    },
    {
      "epoch": 0.9105709699610182,
      "grad_norm": 0.17734781686766848,
      "learning_rate": 2.0828105147154273e-05,
      "loss": 3.3046,
      "step": 3971
    },
    {
      "epoch": 0.9108002751662463,
      "grad_norm": 0.17007862374795546,
      "learning_rate": 2.0722174323423605e-05,
      "loss": 3.5115,
      "step": 3972
    },
    {
      "epoch": 0.9110295803714744,
      "grad_norm": 0.14939912683297218,
      "learning_rate": 2.061650786613545e-05,
      "loss": 3.435,
      "step": 3973
    },
    {
      "epoch": 0.9112588855767025,
      "grad_norm": 0.18967792428204946,
      "learning_rate": 2.0511105833574683e-05,
      "loss": 3.5061,
      "step": 3974
    },
    {
      "epoch": 0.9114881907819308,
      "grad_norm": 0.1752509892626585,
      "learning_rate": 2.040596828388058e-05,
      "loss": 3.4765,
      "step": 3975
    },
    {
      "epoch": 0.9117174959871589,
      "grad_norm": 0.1755198970463125,
      "learning_rate": 2.0301095275046144e-05,
      "loss": 3.3655,
      "step": 3976
    },
    {
      "epoch": 0.911946801192387,
      "grad_norm": 0.17607307720884338,
      "learning_rate": 2.019648686491865e-05,
      "loss": 3.4421,
      "step": 3977
    },
    {
      "epoch": 0.9121761063976153,
      "grad_norm": 0.17362367146825866,
      "learning_rate": 2.009214311119939e-05,
      "loss": 3.4923,
      "step": 3978
    },
    {
      "epoch": 0.9124054116028434,
      "grad_norm": 0.166344545950508,
      "learning_rate": 1.9988064071443767e-05,
      "loss": 3.337,
      "step": 3979
    },
    {
      "epoch": 0.9126347168080715,
      "grad_norm": 0.18520614627440873,
      "learning_rate": 1.9884249803060915e-05,
      "loss": 3.524,
      "step": 3980
    },
    {
      "epoch": 0.9128640220132997,
      "grad_norm": 0.18989275720639023,
      "learning_rate": 1.9780700363314253e-05,
      "loss": 3.4929,
      "step": 3981
    },
    {
      "epoch": 0.9130933272185279,
      "grad_norm": 0.15646203704920694,
      "learning_rate": 1.9677415809320865e-05,
      "loss": 3.3911,
      "step": 3982
    },
    {
      "epoch": 0.913322632423756,
      "grad_norm": 0.17275752561978078,
      "learning_rate": 1.957439619805196e-05,
      "loss": 3.3475,
      "step": 3983
    },
    {
      "epoch": 0.9135519376289841,
      "grad_norm": 0.16731059216787936,
      "learning_rate": 1.947164158633219e-05,
      "loss": 3.4847,
      "step": 3984
    },
    {
      "epoch": 0.9137812428342124,
      "grad_norm": 0.17400058028555856,
      "learning_rate": 1.9369152030840554e-05,
      "loss": 3.4661,
      "step": 3985
    },
    {
      "epoch": 0.9140105480394405,
      "grad_norm": 0.17508828352126707,
      "learning_rate": 1.926692758810955e-05,
      "loss": 3.4808,
      "step": 3986
    },
    {
      "epoch": 0.9142398532446686,
      "grad_norm": 0.19083605165879802,
      "learning_rate": 1.916496831452552e-05,
      "loss": 3.5662,
      "step": 3987
    },
    {
      "epoch": 0.9144691584498968,
      "grad_norm": 0.16487303258803002,
      "learning_rate": 1.9063274266328368e-05,
      "loss": 3.3883,
      "step": 3988
    },
    {
      "epoch": 0.914698463655125,
      "grad_norm": 0.17345303941100135,
      "learning_rate": 1.8961845499611998e-05,
      "loss": 3.6118,
      "step": 3989
    },
    {
      "epoch": 0.9149277688603531,
      "grad_norm": 0.16321763414661397,
      "learning_rate": 1.886068207032382e-05,
      "loss": 3.4293,
      "step": 3990
    },
    {
      "epoch": 0.9151570740655813,
      "grad_norm": 0.17428444697613404,
      "learning_rate": 1.8759784034264925e-05,
      "loss": 3.4292,
      "step": 3991
    },
    {
      "epoch": 0.9153863792708095,
      "grad_norm": 0.16626806869699923,
      "learning_rate": 1.865915144708985e-05,
      "loss": 3.4744,
      "step": 3992
    },
    {
      "epoch": 0.9156156844760376,
      "grad_norm": 0.157648631255949,
      "learning_rate": 1.855878436430708e-05,
      "loss": 3.3714,
      "step": 3993
    },
    {
      "epoch": 0.9158449896812658,
      "grad_norm": 0.16165319828448907,
      "learning_rate": 1.8458682841278162e-05,
      "loss": 3.3609,
      "step": 3994
    },
    {
      "epoch": 0.916074294886494,
      "grad_norm": 0.15968186241820082,
      "learning_rate": 1.835884693321871e-05,
      "loss": 3.3417,
      "step": 3995
    },
    {
      "epoch": 0.9163036000917221,
      "grad_norm": 0.1659516271152944,
      "learning_rate": 1.825927669519728e-05,
      "loss": 3.3686,
      "step": 3996
    },
    {
      "epoch": 0.9165329052969502,
      "grad_norm": 0.16686530359757035,
      "learning_rate": 1.8159972182136386e-05,
      "loss": 3.4869,
      "step": 3997
    },
    {
      "epoch": 0.9167622105021784,
      "grad_norm": 0.1682090565002745,
      "learning_rate": 1.8060933448811546e-05,
      "loss": 3.3268,
      "step": 3998
    },
    {
      "epoch": 0.9169915157074066,
      "grad_norm": 0.1861648478322882,
      "learning_rate": 1.7962160549851945e-05,
      "loss": 3.4803,
      "step": 3999
    },
    {
      "epoch": 0.9172208209126347,
      "grad_norm": 0.17384539993282663,
      "learning_rate": 1.7863653539740122e-05,
      "loss": 3.497,
      "step": 4000
    },
    {
      "epoch": 0.9174501261178629,
      "grad_norm": 0.16627764587325086,
      "learning_rate": 1.776541247281177e-05,
      "loss": 3.3622,
      "step": 4001
    },
    {
      "epoch": 0.917679431323091,
      "grad_norm": 0.1659529073540457,
      "learning_rate": 1.7667437403255938e-05,
      "loss": 3.4696,
      "step": 4002
    },
    {
      "epoch": 0.9179087365283192,
      "grad_norm": 0.1822195449045398,
      "learning_rate": 1.7569728385115224e-05,
      "loss": 3.3058,
      "step": 4003
    },
    {
      "epoch": 0.9181380417335474,
      "grad_norm": 0.17424640829703272,
      "learning_rate": 1.7472285472285186e-05,
      "loss": 3.4101,
      "step": 4004
    },
    {
      "epoch": 0.9183673469387755,
      "grad_norm": 0.1773752870945145,
      "learning_rate": 1.7375108718514665e-05,
      "loss": 3.4584,
      "step": 4005
    },
    {
      "epoch": 0.9185966521440037,
      "grad_norm": 0.17486489819199824,
      "learning_rate": 1.7278198177405614e-05,
      "loss": 3.4475,
      "step": 4006
    },
    {
      "epoch": 0.9188259573492318,
      "grad_norm": 0.18651585881237387,
      "learning_rate": 1.7181553902413438e-05,
      "loss": 3.47,
      "step": 4007
    },
    {
      "epoch": 0.91905526255446,
      "grad_norm": 0.16663832989485935,
      "learning_rate": 1.7085175946846333e-05,
      "loss": 3.5077,
      "step": 4008
    },
    {
      "epoch": 0.9192845677596881,
      "grad_norm": 0.18619654160542048,
      "learning_rate": 1.698906436386577e-05,
      "loss": 3.3924,
      "step": 4009
    },
    {
      "epoch": 0.9195138729649163,
      "grad_norm": 0.16457574606400735,
      "learning_rate": 1.689321920648623e-05,
      "loss": 3.4908,
      "step": 4010
    },
    {
      "epoch": 0.9197431781701445,
      "grad_norm": 0.1687091666176098,
      "learning_rate": 1.679764052757532e-05,
      "loss": 3.381,
      "step": 4011
    },
    {
      "epoch": 0.9199724833753726,
      "grad_norm": 0.18306420020852993,
      "learning_rate": 1.6702328379853525e-05,
      "loss": 3.4057,
      "step": 4012
    },
    {
      "epoch": 0.9202017885806008,
      "grad_norm": 0.16559109160623697,
      "learning_rate": 1.6607282815894464e-05,
      "loss": 3.3689,
      "step": 4013
    },
    {
      "epoch": 0.920431093785829,
      "grad_norm": 0.16670822218783032,
      "learning_rate": 1.6512503888124643e-05,
      "loss": 3.371,
      "step": 4014
    },
    {
      "epoch": 0.9206603989910571,
      "grad_norm": 0.16807146516506113,
      "learning_rate": 1.6417991648823405e-05,
      "loss": 3.4477,
      "step": 4015
    },
    {
      "epoch": 0.9208897041962852,
      "grad_norm": 0.17734245779079202,
      "learning_rate": 1.6323746150122997e-05,
      "loss": 3.4121,
      "step": 4016
    },
    {
      "epoch": 0.9211190094015134,
      "grad_norm": 0.18476551527114746,
      "learning_rate": 1.6229767444008835e-05,
      "loss": 3.3938,
      "step": 4017
    },
    {
      "epoch": 0.9213483146067416,
      "grad_norm": 0.17706112821107792,
      "learning_rate": 1.6136055582318844e-05,
      "loss": 3.4726,
      "step": 4018
    },
    {
      "epoch": 0.9215776198119697,
      "grad_norm": 0.1701503568360483,
      "learning_rate": 1.604261061674378e-05,
      "loss": 3.4023,
      "step": 4019
    },
    {
      "epoch": 0.9218069250171979,
      "grad_norm": 0.17751548842488157,
      "learning_rate": 1.5949432598827307e-05,
      "loss": 3.4051,
      "step": 4020
    },
    {
      "epoch": 0.9220362302224261,
      "grad_norm": 0.1770263301578665,
      "learning_rate": 1.5856521579965865e-05,
      "loss": 3.5109,
      "step": 4021
    },
    {
      "epoch": 0.9222655354276542,
      "grad_norm": 0.16971672164575816,
      "learning_rate": 1.576387761140846e-05,
      "loss": 3.4553,
      "step": 4022
    },
    {
      "epoch": 0.9224948406328823,
      "grad_norm": 0.17453666651804475,
      "learning_rate": 1.5671500744256938e-05,
      "loss": 3.4594,
      "step": 4023
    },
    {
      "epoch": 0.9227241458381106,
      "grad_norm": 0.17752214912528705,
      "learning_rate": 1.5579391029465705e-05,
      "loss": 3.2562,
      "step": 4024
    },
    {
      "epoch": 0.9229534510433387,
      "grad_norm": 0.16945552322149596,
      "learning_rate": 1.5487548517841953e-05,
      "loss": 3.3445,
      "step": 4025
    },
    {
      "epoch": 0.9231827562485668,
      "grad_norm": 0.17831770421763032,
      "learning_rate": 1.539597326004527e-05,
      "loss": 3.5591,
      "step": 4026
    },
    {
      "epoch": 0.923412061453795,
      "grad_norm": 0.1680941686580515,
      "learning_rate": 1.530466530658814e-05,
      "loss": 3.4926,
      "step": 4027
    },
    {
      "epoch": 0.9236413666590232,
      "grad_norm": 0.17755544326111172,
      "learning_rate": 1.5213624707835271e-05,
      "loss": 3.2468,
      "step": 4028
    },
    {
      "epoch": 0.9238706718642513,
      "grad_norm": 0.1786716812690602,
      "learning_rate": 1.5122851514004054e-05,
      "loss": 3.4451,
      "step": 4029
    },
    {
      "epoch": 0.9240999770694794,
      "grad_norm": 0.16709649637939267,
      "learning_rate": 1.503234577516438e-05,
      "loss": 3.3719,
      "step": 4030
    },
    {
      "epoch": 0.9243292822747077,
      "grad_norm": 0.16292326101047558,
      "learning_rate": 1.4942107541238703e-05,
      "loss": 3.481,
      "step": 4031
    },
    {
      "epoch": 0.9245585874799358,
      "grad_norm": 0.17488875936016482,
      "learning_rate": 1.4852136862001764e-05,
      "loss": 3.4415,
      "step": 4032
    },
    {
      "epoch": 0.9247878926851639,
      "grad_norm": 0.17506938609131256,
      "learning_rate": 1.4762433787080809e-05,
      "loss": 3.4768,
      "step": 4033
    },
    {
      "epoch": 0.9250171978903922,
      "grad_norm": 0.17989464770695351,
      "learning_rate": 1.4672998365955314e-05,
      "loss": 3.4267,
      "step": 4034
    },
    {
      "epoch": 0.9252465030956203,
      "grad_norm": 0.17340923090216795,
      "learning_rate": 1.4583830647957541e-05,
      "loss": 3.5233,
      "step": 4035
    },
    {
      "epoch": 0.9254758083008484,
      "grad_norm": 0.1817829855466153,
      "learning_rate": 1.449493068227159e-05,
      "loss": 3.556,
      "step": 4036
    },
    {
      "epoch": 0.9257051135060765,
      "grad_norm": 0.17401617556492138,
      "learning_rate": 1.4406298517934068e-05,
      "loss": 3.5212,
      "step": 4037
    },
    {
      "epoch": 0.9259344187113048,
      "grad_norm": 0.1571092201908352,
      "learning_rate": 1.4317934203834037e-05,
      "loss": 3.4838,
      "step": 4038
    },
    {
      "epoch": 0.9261637239165329,
      "grad_norm": 0.1643403977669219,
      "learning_rate": 1.4229837788712562e-05,
      "loss": 3.4297,
      "step": 4039
    },
    {
      "epoch": 0.926393029121761,
      "grad_norm": 0.17380458315857802,
      "learning_rate": 1.4142009321163051e-05,
      "loss": 3.5065,
      "step": 4040
    },
    {
      "epoch": 0.9266223343269893,
      "grad_norm": 0.17441852416526818,
      "learning_rate": 1.4054448849631085e-05,
      "loss": 3.4108,
      "step": 4041
    },
    {
      "epoch": 0.9268516395322174,
      "grad_norm": 0.16921982177567213,
      "learning_rate": 1.3967156422414474e-05,
      "loss": 3.4655,
      "step": 4042
    },
    {
      "epoch": 0.9270809447374455,
      "grad_norm": 0.18654767555855367,
      "learning_rate": 1.3880132087663145e-05,
      "loss": 3.5112,
      "step": 4043
    },
    {
      "epoch": 0.9273102499426737,
      "grad_norm": 0.1728876518562336,
      "learning_rate": 1.379337589337909e-05,
      "loss": 3.4714,
      "step": 4044
    },
    {
      "epoch": 0.9275395551479019,
      "grad_norm": 0.18387550010508988,
      "learning_rate": 1.3706887887416419e-05,
      "loss": 3.4691,
      "step": 4045
    },
    {
      "epoch": 0.92776886035313,
      "grad_norm": 0.1780731700820299,
      "learning_rate": 1.3620668117481471e-05,
      "loss": 3.4301,
      "step": 4046
    },
    {
      "epoch": 0.9279981655583581,
      "grad_norm": 0.1750654295190873,
      "learning_rate": 1.3534716631132316e-05,
      "loss": 3.435,
      "step": 4047
    },
    {
      "epoch": 0.9282274707635864,
      "grad_norm": 0.16372145177569367,
      "learning_rate": 1.344903347577936e-05,
      "loss": 3.3561,
      "step": 4048
    },
    {
      "epoch": 0.9284567759688145,
      "grad_norm": 0.16155325319817357,
      "learning_rate": 1.3363618698684853e-05,
      "loss": 3.3466,
      "step": 4049
    },
    {
      "epoch": 0.9286860811740426,
      "grad_norm": 0.1808970982654504,
      "learning_rate": 1.3278472346962944e-05,
      "loss": 3.4682,
      "step": 4050
    },
    {
      "epoch": 0.9289153863792708,
      "grad_norm": 0.17551963941980323,
      "learning_rate": 1.3193594467579728e-05,
      "loss": 3.3132,
      "step": 4051
    },
    {
      "epoch": 0.929144691584499,
      "grad_norm": 0.15737496453941227,
      "learning_rate": 1.3108985107353477e-05,
      "loss": 3.2844,
      "step": 4052
    },
    {
      "epoch": 0.9293739967897271,
      "grad_norm": 0.18565245552146087,
      "learning_rate": 1.3024644312954026e-05,
      "loss": 3.2989,
      "step": 4053
    },
    {
      "epoch": 0.9296033019949553,
      "grad_norm": 0.17398511246430767,
      "learning_rate": 1.2940572130903161e-05,
      "loss": 3.4135,
      "step": 4054
    },
    {
      "epoch": 0.9298326072001835,
      "grad_norm": 0.17090641780014457,
      "learning_rate": 1.2856768607574564e-05,
      "loss": 3.4086,
      "step": 4055
    },
    {
      "epoch": 0.9300619124054116,
      "grad_norm": 0.1839185645133012,
      "learning_rate": 1.2773233789193816e-05,
      "loss": 3.3014,
      "step": 4056
    },
    {
      "epoch": 0.9302912176106397,
      "grad_norm": 0.16776962111377194,
      "learning_rate": 1.2689967721837947e-05,
      "loss": 3.3927,
      "step": 4057
    },
    {
      "epoch": 0.930520522815868,
      "grad_norm": 0.19558540445465677,
      "learning_rate": 1.2606970451436105e-05,
      "loss": 3.2843,
      "step": 4058
    },
    {
      "epoch": 0.9307498280210961,
      "grad_norm": 0.18036703499569307,
      "learning_rate": 1.2524242023769006e-05,
      "loss": 3.4305,
      "step": 4059
    },
    {
      "epoch": 0.9309791332263242,
      "grad_norm": 0.16662935401979012,
      "learning_rate": 1.2441782484469145e-05,
      "loss": 3.5169,
      "step": 4060
    },
    {
      "epoch": 0.9312084384315524,
      "grad_norm": 0.17611729434667608,
      "learning_rate": 1.2359591879020526e-05,
      "loss": 3.4422,
      "step": 4061
    },
    {
      "epoch": 0.9314377436367806,
      "grad_norm": 0.17678248391630608,
      "learning_rate": 1.2277670252759e-05,
      "loss": 3.5962,
      "step": 4062
    },
    {
      "epoch": 0.9316670488420087,
      "grad_norm": 0.1728537615840493,
      "learning_rate": 1.2196017650872081e-05,
      "loss": 3.4154,
      "step": 4063
    },
    {
      "epoch": 0.9318963540472369,
      "grad_norm": 0.17804064410056628,
      "learning_rate": 1.2114634118398637e-05,
      "loss": 3.4396,
      "step": 4064
    },
    {
      "epoch": 0.932125659252465,
      "grad_norm": 0.1762505063940152,
      "learning_rate": 1.2033519700229367e-05,
      "loss": 3.5853,
      "step": 4065
    },
    {
      "epoch": 0.9323549644576932,
      "grad_norm": 0.19390056550974547,
      "learning_rate": 1.1952674441106482e-05,
      "loss": 3.5347,
      "step": 4066
    },
    {
      "epoch": 0.9325842696629213,
      "grad_norm": 0.18912936645356548,
      "learning_rate": 1.1872098385623586e-05,
      "loss": 3.4056,
      "step": 4067
    },
    {
      "epoch": 0.9328135748681495,
      "grad_norm": 0.16780112093481545,
      "learning_rate": 1.1791791578226018e-05,
      "loss": 3.5485,
      "step": 4068
    },
    {
      "epoch": 0.9330428800733777,
      "grad_norm": 0.1606182125080837,
      "learning_rate": 1.1711754063210289e-05,
      "loss": 3.4527,
      "step": 4069
    },
    {
      "epoch": 0.9332721852786058,
      "grad_norm": 0.15697699429320813,
      "learning_rate": 1.1631985884724804e-05,
      "loss": 3.3557,
      "step": 4070
    },
    {
      "epoch": 0.933501490483834,
      "grad_norm": 0.19177757913476748,
      "learning_rate": 1.155248708676887e-05,
      "loss": 3.433,
      "step": 4071
    },
    {
      "epoch": 0.9337307956890621,
      "grad_norm": 0.17092602452749509,
      "learning_rate": 1.147325771319374e-05,
      "loss": 3.3806,
      "step": 4072
    },
    {
      "epoch": 0.9339601008942903,
      "grad_norm": 0.1675707205197582,
      "learning_rate": 1.1394297807701737e-05,
      "loss": 3.4093,
      "step": 4073
    },
    {
      "epoch": 0.9341894060995185,
      "grad_norm": 0.1765259436471832,
      "learning_rate": 1.1315607413846518e-05,
      "loss": 3.4177,
      "step": 4074
    },
    {
      "epoch": 0.9344187113047466,
      "grad_norm": 0.16878943940087918,
      "learning_rate": 1.1237186575033254e-05,
      "loss": 3.4089,
      "step": 4075
    },
    {
      "epoch": 0.9346480165099748,
      "grad_norm": 0.1752641739210239,
      "learning_rate": 1.115903533451834e-05,
      "loss": 3.4425,
      "step": 4076
    },
    {
      "epoch": 0.9348773217152029,
      "grad_norm": 0.17246919812111097,
      "learning_rate": 1.1081153735409522e-05,
      "loss": 3.4299,
      "step": 4077
    },
    {
      "epoch": 0.9351066269204311,
      "grad_norm": 0.16617678237010938,
      "learning_rate": 1.1003541820665763e-05,
      "loss": 3.5031,
      "step": 4078
    },
    {
      "epoch": 0.9353359321256592,
      "grad_norm": 0.18325793887558306,
      "learning_rate": 1.0926199633097156e-05,
      "loss": 3.4512,
      "step": 4079
    },
    {
      "epoch": 0.9355652373308874,
      "grad_norm": 0.16559973971317632,
      "learning_rate": 1.0849127215365296e-05,
      "loss": 3.3775,
      "step": 4080
    },
    {
      "epoch": 0.9357945425361156,
      "grad_norm": 0.1853822334247987,
      "learning_rate": 1.0772324609982787e-05,
      "loss": 3.4154,
      "step": 4081
    },
    {
      "epoch": 0.9360238477413437,
      "grad_norm": 0.17027063390450922,
      "learning_rate": 1.0695791859313298e-05,
      "loss": 3.5194,
      "step": 4082
    },
    {
      "epoch": 0.9362531529465719,
      "grad_norm": 0.17473842294057648,
      "learning_rate": 1.0619529005571893e-05,
      "loss": 3.4865,
      "step": 4083
    },
    {
      "epoch": 0.9364824581518001,
      "grad_norm": 0.1780232524127434,
      "learning_rate": 1.0543536090824702e-05,
      "loss": 3.3645,
      "step": 4084
    },
    {
      "epoch": 0.9367117633570282,
      "grad_norm": 0.16470994588993393,
      "learning_rate": 1.0467813156988748e-05,
      "loss": 3.4075,
      "step": 4085
    },
    {
      "epoch": 0.9369410685622563,
      "grad_norm": 0.16573902174813296,
      "learning_rate": 1.03923602458324e-05,
      "loss": 3.2529,
      "step": 4086
    },
    {
      "epoch": 0.9371703737674846,
      "grad_norm": 0.1655892047370745,
      "learning_rate": 1.0317177398975031e-05,
      "loss": 3.4038,
      "step": 4087
    },
    {
      "epoch": 0.9373996789727127,
      "grad_norm": 0.1829516027091889,
      "learning_rate": 1.0242264657886858e-05,
      "loss": 3.5156,
      "step": 4088
    },
    {
      "epoch": 0.9376289841779408,
      "grad_norm": 0.1706524270221836,
      "learning_rate": 1.0167622063889326e-05,
      "loss": 3.5004,
      "step": 4089
    },
    {
      "epoch": 0.937858289383169,
      "grad_norm": 0.171639041529688,
      "learning_rate": 1.0093249658154835e-05,
      "loss": 3.376,
      "step": 4090
    },
    {
      "epoch": 0.9380875945883972,
      "grad_norm": 0.17809166435836293,
      "learning_rate": 1.0019147481706625e-05,
      "loss": 3.4965,
      "step": 4091
    },
    {
      "epoch": 0.9383168997936253,
      "grad_norm": 0.17371848292204123,
      "learning_rate": 9.945315575419e-06,
      "loss": 3.413,
      "step": 4092
    },
    {
      "epoch": 0.9385462049988534,
      "grad_norm": 0.17457646989369371,
      "learning_rate": 9.871753980017051e-06,
      "loss": 3.4716,
      "step": 4093
    },
    {
      "epoch": 0.9387755102040817,
      "grad_norm": 0.16200003647602393,
      "learning_rate": 9.79846273607704e-06,
      "loss": 3.3794,
      "step": 4094
    },
    {
      "epoch": 0.9390048154093098,
      "grad_norm": 0.16649057314300872,
      "learning_rate": 9.725441884025855e-06,
      "loss": 3.4248,
      "step": 4095
    },
    {
      "epoch": 0.9392341206145379,
      "grad_norm": 0.17250531911059644,
      "learning_rate": 9.652691464141272e-06,
      "loss": 3.2608,
      "step": 4096
    },
    {
      "epoch": 0.9394634258197662,
      "grad_norm": 0.16167205056087014,
      "learning_rate": 9.580211516551862e-06,
      "loss": 3.4033,
      "step": 4097
    },
    {
      "epoch": 0.9396927310249943,
      "grad_norm": 0.1733557279244543,
      "learning_rate": 9.50800208123731e-06,
      "loss": 3.6002,
      "step": 4098
    },
    {
      "epoch": 0.9399220362302224,
      "grad_norm": 0.15626751439791062,
      "learning_rate": 9.436063198027589e-06,
      "loss": 3.3266,
      "step": 4099
    },
    {
      "epoch": 0.9401513414354505,
      "grad_norm": 0.1654730181291095,
      "learning_rate": 9.364394906603902e-06,
      "loss": 3.5685,
      "step": 4100
    },
    {
      "epoch": 0.9403806466406788,
      "grad_norm": 0.18358333710429758,
      "learning_rate": 9.292997246497959e-06,
      "loss": 3.398,
      "step": 4101
    },
    {
      "epoch": 0.9406099518459069,
      "grad_norm": 0.18106819016679215,
      "learning_rate": 9.221870257092203e-06,
      "loss": 3.5812,
      "step": 4102
    },
    {
      "epoch": 0.940839257051135,
      "grad_norm": 0.1726400879663816,
      "learning_rate": 9.151013977619693e-06,
      "loss": 3.5392,
      "step": 4103
    },
    {
      "epoch": 0.9410685622563633,
      "grad_norm": 0.17939903292787945,
      "learning_rate": 9.080428447164502e-06,
      "loss": 3.3463,
      "step": 4104
    },
    {
      "epoch": 0.9412978674615914,
      "grad_norm": 0.1728713745797768,
      "learning_rate": 9.010113704661038e-06,
      "loss": 3.4345,
      "step": 4105
    },
    {
      "epoch": 0.9415271726668195,
      "grad_norm": 0.17897396301968332,
      "learning_rate": 8.940069788894389e-06,
      "loss": 3.4753,
      "step": 4106
    },
    {
      "epoch": 0.9417564778720477,
      "grad_norm": 0.1644944983764894,
      "learning_rate": 8.870296738500316e-06,
      "loss": 3.3895,
      "step": 4107
    },
    {
      "epoch": 0.9419857830772759,
      "grad_norm": 0.17635886605425652,
      "learning_rate": 8.800794591965256e-06,
      "loss": 3.4712,
      "step": 4108
    },
    {
      "epoch": 0.942215088282504,
      "grad_norm": 0.16513367968625145,
      "learning_rate": 8.731563387626095e-06,
      "loss": 3.4096,
      "step": 4109
    },
    {
      "epoch": 0.9424443934877321,
      "grad_norm": 0.18065680710617157,
      "learning_rate": 8.662603163670236e-06,
      "loss": 3.358,
      "step": 4110
    },
    {
      "epoch": 0.9426736986929604,
      "grad_norm": 0.18878149446830428,
      "learning_rate": 8.59391395813569e-06,
      "loss": 3.4612,
      "step": 4111
    },
    {
      "epoch": 0.9429030038981885,
      "grad_norm": 0.1653943216162752,
      "learning_rate": 8.525495808911155e-06,
      "loss": 3.4008,
      "step": 4112
    },
    {
      "epoch": 0.9431323091034166,
      "grad_norm": 0.17488607088628633,
      "learning_rate": 8.457348753735328e-06,
      "loss": 3.4782,
      "step": 4113
    },
    {
      "epoch": 0.9433616143086448,
      "grad_norm": 0.1850350692187784,
      "learning_rate": 8.389472830197808e-06,
      "loss": 3.4165,
      "step": 4114
    },
    {
      "epoch": 0.943590919513873,
      "grad_norm": 0.16491556109536382,
      "learning_rate": 8.321868075738593e-06,
      "loss": 3.2609,
      "step": 4115
    },
    {
      "epoch": 0.9438202247191011,
      "grad_norm": 0.1563384305362877,
      "learning_rate": 8.25453452764785e-06,
      "loss": 3.44,
      "step": 4116
    },
    {
      "epoch": 0.9440495299243293,
      "grad_norm": 0.1721308375913384,
      "learning_rate": 8.187472223066371e-06,
      "loss": 3.5494,
      "step": 4117
    },
    {
      "epoch": 0.9442788351295575,
      "grad_norm": 0.16536204614278582,
      "learning_rate": 8.120681198985291e-06,
      "loss": 3.4104,
      "step": 4118
    },
    {
      "epoch": 0.9445081403347856,
      "grad_norm": 0.1721008363042696,
      "learning_rate": 8.054161492246136e-06,
      "loss": 3.4393,
      "step": 4119
    },
    {
      "epoch": 0.9447374455400137,
      "grad_norm": 0.16370345755091317,
      "learning_rate": 7.98791313954067e-06,
      "loss": 3.3873,
      "step": 4120
    },
    {
      "epoch": 0.944966750745242,
      "grad_norm": 0.1732099610776653,
      "learning_rate": 7.921936177411049e-06,
      "loss": 3.4091,
      "step": 4121
    },
    {
      "epoch": 0.9451960559504701,
      "grad_norm": 0.16790821375136394,
      "learning_rate": 7.856230642249718e-06,
      "loss": 3.3396,
      "step": 4122
    },
    {
      "epoch": 0.9454253611556982,
      "grad_norm": 0.18415501092726244,
      "learning_rate": 7.790796570299463e-06,
      "loss": 3.3372,
      "step": 4123
    },
    {
      "epoch": 0.9456546663609264,
      "grad_norm": 0.16544209391169484,
      "learning_rate": 7.725633997653193e-06,
      "loss": 3.4944,
      "step": 4124
    },
    {
      "epoch": 0.9458839715661546,
      "grad_norm": 0.18314870193718688,
      "learning_rate": 7.660742960254207e-06,
      "loss": 3.331,
      "step": 4125
    },
    {
      "epoch": 0.9461132767713827,
      "grad_norm": 0.17348807949427783,
      "learning_rate": 7.59612349389599e-06,
      "loss": 3.5636,
      "step": 4126
    },
    {
      "epoch": 0.9463425819766109,
      "grad_norm": 0.17294930727546146,
      "learning_rate": 7.531775634222138e-06,
      "loss": 3.2988,
      "step": 4127
    },
    {
      "epoch": 0.946571887181839,
      "grad_norm": 0.17328830502193845,
      "learning_rate": 7.467699416726537e-06,
      "loss": 3.4486,
      "step": 4128
    },
    {
      "epoch": 0.9468011923870672,
      "grad_norm": 0.16163401283286277,
      "learning_rate": 7.403894876753192e-06,
      "loss": 3.4348,
      "step": 4129
    },
    {
      "epoch": 0.9470304975922953,
      "grad_norm": 0.17082354424390064,
      "learning_rate": 7.340362049496285e-06,
      "loss": 3.3127,
      "step": 4130
    },
    {
      "epoch": 0.9472598027975235,
      "grad_norm": 0.16912732944207481,
      "learning_rate": 7.277100970000061e-06,
      "loss": 3.4665,
      "step": 4131
    },
    {
      "epoch": 0.9474891080027517,
      "grad_norm": 0.15797686408528705,
      "learning_rate": 7.214111673158941e-06,
      "loss": 3.3928,
      "step": 4132
    },
    {
      "epoch": 0.9477184132079798,
      "grad_norm": 0.16616475354858776,
      "learning_rate": 7.151394193717408e-06,
      "loss": 3.4285,
      "step": 4133
    },
    {
      "epoch": 0.947947718413208,
      "grad_norm": 0.18197516867486835,
      "learning_rate": 7.088948566269959e-06,
      "loss": 3.352,
      "step": 4134
    },
    {
      "epoch": 0.9481770236184361,
      "grad_norm": 0.16713539911230868,
      "learning_rate": 7.026774825261151e-06,
      "loss": 3.3886,
      "step": 4135
    },
    {
      "epoch": 0.9484063288236643,
      "grad_norm": 0.17358424485407187,
      "learning_rate": 6.964873004985717e-06,
      "loss": 3.3978,
      "step": 4136
    },
    {
      "epoch": 0.9486356340288925,
      "grad_norm": 0.1714911043750069,
      "learning_rate": 6.903243139588233e-06,
      "loss": 3.2471,
      "step": 4137
    },
    {
      "epoch": 0.9488649392341206,
      "grad_norm": 0.16757697138633817,
      "learning_rate": 6.841885263063341e-06,
      "loss": 3.4293,
      "step": 4138
    },
    {
      "epoch": 0.9490942444393488,
      "grad_norm": 0.1638294546281909,
      "learning_rate": 6.780799409255522e-06,
      "loss": 3.3424,
      "step": 4139
    },
    {
      "epoch": 0.9493235496445769,
      "grad_norm": 0.17503498076585167,
      "learning_rate": 6.7199856118594916e-06,
      "loss": 3.4129,
      "step": 4140
    },
    {
      "epoch": 0.9495528548498051,
      "grad_norm": 0.17988142044473124,
      "learning_rate": 6.659443904419637e-06,
      "loss": 3.6439,
      "step": 4141
    },
    {
      "epoch": 0.9497821600550332,
      "grad_norm": 0.1877219169415489,
      "learning_rate": 6.599174320330359e-06,
      "loss": 3.6787,
      "step": 4142
    },
    {
      "epoch": 0.9500114652602614,
      "grad_norm": 0.15877785578917583,
      "learning_rate": 6.539176892836008e-06,
      "loss": 3.3322,
      "step": 4143
    },
    {
      "epoch": 0.9502407704654896,
      "grad_norm": 0.17017802363014414,
      "learning_rate": 6.479451655030721e-06,
      "loss": 3.3493,
      "step": 4144
    },
    {
      "epoch": 0.9504700756707177,
      "grad_norm": 0.17237306393247703,
      "learning_rate": 6.4199986398585375e-06,
      "loss": 3.5284,
      "step": 4145
    },
    {
      "epoch": 0.9506993808759459,
      "grad_norm": 0.18179756718081816,
      "learning_rate": 6.3608178801133345e-06,
      "loss": 3.4223,
      "step": 4146
    },
    {
      "epoch": 0.9509286860811741,
      "grad_norm": 0.15872743457807267,
      "learning_rate": 6.3019094084388884e-06,
      "loss": 3.4926,
      "step": 4147
    },
    {
      "epoch": 0.9511579912864022,
      "grad_norm": 0.16924051527764206,
      "learning_rate": 6.243273257328652e-06,
      "loss": 3.5101,
      "step": 4148
    },
    {
      "epoch": 0.9513872964916303,
      "grad_norm": 0.17376881582192572,
      "learning_rate": 6.18490945912592e-06,
      "loss": 3.3751,
      "step": 4149
    },
    {
      "epoch": 0.9516166016968585,
      "grad_norm": 0.1748802412521212,
      "learning_rate": 6.1268180460238854e-06,
      "loss": 3.4052,
      "step": 4150
    },
    {
      "epoch": 0.9518459069020867,
      "grad_norm": 0.1874713177039608,
      "learning_rate": 6.068999050065249e-06,
      "loss": 3.4829,
      "step": 4151
    },
    {
      "epoch": 0.9520752121073148,
      "grad_norm": 0.16670143362301898,
      "learning_rate": 6.011452503142667e-06,
      "loss": 3.3589,
      "step": 4152
    },
    {
      "epoch": 0.952304517312543,
      "grad_norm": 0.1822917358007997,
      "learning_rate": 5.9541784369983586e-06,
      "loss": 3.5831,
      "step": 4153
    },
    {
      "epoch": 0.9525338225177712,
      "grad_norm": 0.17614595903481842,
      "learning_rate": 5.897176883224442e-06,
      "loss": 3.4652,
      "step": 4154
    },
    {
      "epoch": 0.9527631277229993,
      "grad_norm": 0.1786541713970712,
      "learning_rate": 5.840447873262433e-06,
      "loss": 3.551,
      "step": 4155
    },
    {
      "epoch": 0.9529924329282274,
      "grad_norm": 0.1839619648760807,
      "learning_rate": 5.7839914384038014e-06,
      "loss": 3.4585,
      "step": 4156
    },
    {
      "epoch": 0.9532217381334557,
      "grad_norm": 0.1849647179227022,
      "learning_rate": 5.727807609789471e-06,
      "loss": 3.3842,
      "step": 4157
    },
    {
      "epoch": 0.9534510433386838,
      "grad_norm": 0.17207121094843478,
      "learning_rate": 5.671896418410094e-06,
      "loss": 3.4554,
      "step": 4158
    },
    {
      "epoch": 0.9536803485439119,
      "grad_norm": 0.16281389351794798,
      "learning_rate": 5.616257895105892e-06,
      "loss": 3.4675,
      "step": 4159
    },
    {
      "epoch": 0.95390965374914,
      "grad_norm": 0.1689186534715943,
      "learning_rate": 5.560892070566703e-06,
      "loss": 3.4066,
      "step": 4160
    },
    {
      "epoch": 0.9541389589543683,
      "grad_norm": 0.1796642254562253,
      "learning_rate": 5.505798975331933e-06,
      "loss": 3.4002,
      "step": 4161
    },
    {
      "epoch": 0.9543682641595964,
      "grad_norm": 0.1764477993202574,
      "learning_rate": 5.4509786397906046e-06,
      "loss": 3.5512,
      "step": 4162
    },
    {
      "epoch": 0.9545975693648245,
      "grad_norm": 0.17788722102325677,
      "learning_rate": 5.396431094181198e-06,
      "loss": 3.4347,
      "step": 4163
    },
    {
      "epoch": 0.9548268745700528,
      "grad_norm": 0.16572944736048117,
      "learning_rate": 5.3421563685919214e-06,
      "loss": 3.4199,
      "step": 4164
    },
    {
      "epoch": 0.9550561797752809,
      "grad_norm": 0.17344444925707808,
      "learning_rate": 5.288154492960107e-06,
      "loss": 3.5277,
      "step": 4165
    },
    {
      "epoch": 0.955285484980509,
      "grad_norm": 0.18591213180328006,
      "learning_rate": 5.234425497072981e-06,
      "loss": 3.4768,
      "step": 4166
    },
    {
      "epoch": 0.9555147901857373,
      "grad_norm": 0.18326169313546098,
      "learning_rate": 5.1809694105671155e-06,
      "loss": 3.4942,
      "step": 4167
    },
    {
      "epoch": 0.9557440953909654,
      "grad_norm": 0.15842881165892433,
      "learning_rate": 5.127786262928424e-06,
      "loss": 3.4542,
      "step": 4168
    },
    {
      "epoch": 0.9559734005961935,
      "grad_norm": 0.18035072803103633,
      "learning_rate": 5.074876083492441e-06,
      "loss": 3.35,
      "step": 4169
    },
    {
      "epoch": 0.9562027058014216,
      "grad_norm": 0.16778900952214967,
      "learning_rate": 5.022238901444098e-06,
      "loss": 3.5023,
      "step": 4170
    },
    {
      "epoch": 0.9564320110066499,
      "grad_norm": 0.17387740634763885,
      "learning_rate": 4.96987474581767e-06,
      "loss": 3.4697,
      "step": 4171
    },
    {
      "epoch": 0.956661316211878,
      "grad_norm": 0.17562420996025463,
      "learning_rate": 4.917783645496887e-06,
      "loss": 3.5375,
      "step": 4172
    },
    {
      "epoch": 0.9568906214171061,
      "grad_norm": 0.16576468322411733,
      "learning_rate": 4.865965629214819e-06,
      "loss": 3.2844,
      "step": 4173
    },
    {
      "epoch": 0.9571199266223344,
      "grad_norm": 0.17257374528967723,
      "learning_rate": 4.8144207255539914e-06,
      "loss": 3.3639,
      "step": 4174
    },
    {
      "epoch": 0.9573492318275625,
      "grad_norm": 0.15604688728961402,
      "learning_rate": 4.763148962946218e-06,
      "loss": 3.549,
      "step": 4175
    },
    {
      "epoch": 0.9575785370327906,
      "grad_norm": 0.1576777046683438,
      "learning_rate": 4.712150369672652e-06,
      "loss": 3.2934,
      "step": 4176
    },
    {
      "epoch": 0.9578078422380188,
      "grad_norm": 0.15341932063300534,
      "learning_rate": 4.661424973863681e-06,
      "loss": 3.2843,
      "step": 4177
    },
    {
      "epoch": 0.958037147443247,
      "grad_norm": 0.1639320600835105,
      "learning_rate": 4.610972803499313e-06,
      "loss": 3.3115,
      "step": 4178
    },
    {
      "epoch": 0.9582664526484751,
      "grad_norm": 0.16475661941068037,
      "learning_rate": 4.560793886408398e-06,
      "loss": 3.3457,
      "step": 4179
    },
    {
      "epoch": 0.9584957578537033,
      "grad_norm": 0.1646207857706655,
      "learning_rate": 4.510888250269407e-06,
      "loss": 3.471,
      "step": 4180
    },
    {
      "epoch": 0.9587250630589315,
      "grad_norm": 0.18123123111993325,
      "learning_rate": 4.461255922609986e-06,
      "loss": 3.4692,
      "step": 4181
    },
    {
      "epoch": 0.9589543682641596,
      "grad_norm": 0.18287775571178988,
      "learning_rate": 4.411896930806902e-06,
      "loss": 3.5902,
      "step": 4182
    },
    {
      "epoch": 0.9591836734693877,
      "grad_norm": 0.15371411453202177,
      "learning_rate": 4.362811302086267e-06,
      "loss": 3.4444,
      "step": 4183
    },
    {
      "epoch": 0.959412978674616,
      "grad_norm": 0.16923871239810293,
      "learning_rate": 4.313999063523422e-06,
      "loss": 3.5133,
      "step": 4184
    },
    {
      "epoch": 0.9596422838798441,
      "grad_norm": 0.1664734173065585,
      "learning_rate": 4.265460242042885e-06,
      "loss": 3.3305,
      "step": 4185
    },
    {
      "epoch": 0.9598715890850722,
      "grad_norm": 0.17838416775799557,
      "learning_rate": 4.2171948644182946e-06,
      "loss": 3.5024,
      "step": 4186
    },
    {
      "epoch": 0.9601008942903004,
      "grad_norm": 0.17497477900727124,
      "learning_rate": 4.169202957272522e-06,
      "loss": 3.5692,
      "step": 4187
    },
    {
      "epoch": 0.9603301994955286,
      "grad_norm": 0.1705213519905296,
      "learning_rate": 4.121484547077614e-06,
      "loss": 3.3688,
      "step": 4188
    },
    {
      "epoch": 0.9605595047007567,
      "grad_norm": 0.17390800841259302,
      "learning_rate": 4.074039660154738e-06,
      "loss": 3.4272,
      "step": 4189
    },
    {
      "epoch": 0.9607888099059849,
      "grad_norm": 0.17247304968337776,
      "learning_rate": 4.026868322674127e-06,
      "loss": 3.505,
      "step": 4190
    },
    {
      "epoch": 0.961018115111213,
      "grad_norm": 0.19089129240428498,
      "learning_rate": 3.9799705606551325e-06,
      "loss": 3.6176,
      "step": 4191
    },
    {
      "epoch": 0.9612474203164412,
      "grad_norm": 0.16458073066179457,
      "learning_rate": 3.933346399966398e-06,
      "loss": 3.5613,
      "step": 4192
    },
    {
      "epoch": 0.9614767255216693,
      "grad_norm": 0.18018064028313818,
      "learning_rate": 3.886995866325294e-06,
      "loss": 3.4336,
      "step": 4193
    },
    {
      "epoch": 0.9617060307268975,
      "grad_norm": 0.18092053131905478,
      "learning_rate": 3.840918985298647e-06,
      "loss": 3.4074,
      "step": 4194
    },
    {
      "epoch": 0.9619353359321257,
      "grad_norm": 0.1630438474350495,
      "learning_rate": 3.795115782302072e-06,
      "loss": 3.4336,
      "step": 4195
    },
    {
      "epoch": 0.9621646411373538,
      "grad_norm": 0.1757495127297042,
      "learning_rate": 3.7495862826003592e-06,
      "loss": 3.4873,
      "step": 4196
    },
    {
      "epoch": 0.962393946342582,
      "grad_norm": 0.17367670988141518,
      "learning_rate": 3.704330511307197e-06,
      "loss": 3.2697,
      "step": 4197
    },
    {
      "epoch": 0.9626232515478101,
      "grad_norm": 0.18497753499422867,
      "learning_rate": 3.659348493385395e-06,
      "loss": 3.517,
      "step": 4198
    },
    {
      "epoch": 0.9628525567530383,
      "grad_norm": 0.17160916677937432,
      "learning_rate": 3.614640253646828e-06,
      "loss": 3.4692,
      "step": 4199
    },
    {
      "epoch": 0.9630818619582665,
      "grad_norm": 0.1657031981892196,
      "learning_rate": 3.5702058167521035e-06,
      "loss": 3.3798,
      "step": 4200
    },
    {
      "epoch": 0.9633111671634946,
      "grad_norm": 0.15976446410802247,
      "learning_rate": 3.5260452072110594e-06,
      "loss": 3.4233,
      "step": 4201
    },
    {
      "epoch": 0.9635404723687228,
      "grad_norm": 0.18127831847969592,
      "learning_rate": 3.4821584493823776e-06,
      "loss": 3.5242,
      "step": 4202
    },
    {
      "epoch": 0.9637697775739509,
      "grad_norm": 0.1771425465218149,
      "learning_rate": 3.4385455674737498e-06,
      "loss": 3.5039,
      "step": 4203
    },
    {
      "epoch": 0.9639990827791791,
      "grad_norm": 0.16391853508917098,
      "learning_rate": 3.3952065855415994e-06,
      "loss": 3.2425,
      "step": 4204
    },
    {
      "epoch": 0.9642283879844072,
      "grad_norm": 0.16728365030335002,
      "learning_rate": 3.3521415274915256e-06,
      "loss": 3.5253,
      "step": 4205
    },
    {
      "epoch": 0.9644576931896354,
      "grad_norm": 0.17244161024203275,
      "learning_rate": 3.309350417077972e-06,
      "loss": 3.4916,
      "step": 4206
    },
    {
      "epoch": 0.9646869983948636,
      "grad_norm": 0.17960632341714602,
      "learning_rate": 3.2668332779041133e-06,
      "loss": 3.4546,
      "step": 4207
    },
    {
      "epoch": 0.9649163036000917,
      "grad_norm": 0.17674518937707842,
      "learning_rate": 3.2245901334221893e-06,
      "loss": 3.4384,
      "step": 4208
    },
    {
      "epoch": 0.9651456088053199,
      "grad_norm": 0.17788636544368314,
      "learning_rate": 3.1826210069332838e-06,
      "loss": 3.4134,
      "step": 4209
    },
    {
      "epoch": 0.9653749140105481,
      "grad_norm": 0.1586104919142522,
      "learning_rate": 3.1409259215871567e-06,
      "loss": 3.418,
      "step": 4210
    },
    {
      "epoch": 0.9656042192157762,
      "grad_norm": 0.16568137869837846,
      "learning_rate": 3.0995049003826324e-06,
      "loss": 3.4932,
      "step": 4211
    },
    {
      "epoch": 0.9658335244210043,
      "grad_norm": 0.20657148013400997,
      "learning_rate": 3.058357966167213e-06,
      "loss": 3.4713,
      "step": 4212
    },
    {
      "epoch": 0.9660628296262325,
      "grad_norm": 0.17022070440482903,
      "learning_rate": 3.017485141637355e-06,
      "loss": 3.3764,
      "step": 4213
    },
    {
      "epoch": 0.9662921348314607,
      "grad_norm": 0.18080675326150147,
      "learning_rate": 2.9768864493381344e-06,
      "loss": 3.4665,
      "step": 4214
    },
    {
      "epoch": 0.9665214400366888,
      "grad_norm": 0.17520412567717222,
      "learning_rate": 2.9365619116636376e-06,
      "loss": 3.4003,
      "step": 4215
    },
    {
      "epoch": 0.966750745241917,
      "grad_norm": 0.17120588119820743,
      "learning_rate": 2.896511550856462e-06,
      "loss": 3.504,
      "step": 4216
    },
    {
      "epoch": 0.9669800504471452,
      "grad_norm": 0.18072038626850212,
      "learning_rate": 2.856735389008269e-06,
      "loss": 3.392,
      "step": 4217
    },
    {
      "epoch": 0.9672093556523733,
      "grad_norm": 0.17591695783373146,
      "learning_rate": 2.8172334480591755e-06,
      "loss": 3.5421,
      "step": 4218
    },
    {
      "epoch": 0.9674386608576014,
      "grad_norm": 0.17235991635645265,
      "learning_rate": 2.778005749798307e-06,
      "loss": 3.4698,
      "step": 4219
    },
    {
      "epoch": 0.9676679660628297,
      "grad_norm": 0.17087990644324044,
      "learning_rate": 2.739052315863355e-06,
      "loss": 3.4645,
      "step": 4220
    },
    {
      "epoch": 0.9678972712680578,
      "grad_norm": 0.16888045117533004,
      "learning_rate": 2.700373167740744e-06,
      "loss": 3.4871,
      "step": 4221
    },
    {
      "epoch": 0.9681265764732859,
      "grad_norm": 0.18175028504983007,
      "learning_rate": 2.6619683267656845e-06,
      "loss": 3.4569,
      "step": 4222
    },
    {
      "epoch": 0.968355881678514,
      "grad_norm": 0.1742980764904034,
      "learning_rate": 2.62383781412201e-06,
      "loss": 3.497,
      "step": 4223
    },
    {
      "epoch": 0.9685851868837423,
      "grad_norm": 0.16042418993202426,
      "learning_rate": 2.5859816508422286e-06,
      "loss": 3.3707,
      "step": 4224
    },
    {
      "epoch": 0.9688144920889704,
      "grad_norm": 0.17798364956580265,
      "learning_rate": 2.5483998578076373e-06,
      "loss": 3.4034,
      "step": 4225
    },
    {
      "epoch": 0.9690437972941985,
      "grad_norm": 0.1688935555605003,
      "learning_rate": 2.511092455747932e-06,
      "loss": 3.352,
      "step": 4226
    },
    {
      "epoch": 0.9692731024994268,
      "grad_norm": 0.16695498923596483,
      "learning_rate": 2.4740594652418736e-06,
      "loss": 3.2536,
      "step": 4227
    },
    {
      "epoch": 0.9695024077046549,
      "grad_norm": 0.17155855504696987,
      "learning_rate": 2.4373009067163443e-06,
      "loss": 3.3994,
      "step": 4228
    },
    {
      "epoch": 0.969731712909883,
      "grad_norm": 0.1729224498727605,
      "learning_rate": 2.4008168004472917e-06,
      "loss": 3.4277,
      "step": 4229
    },
    {
      "epoch": 0.9699610181151113,
      "grad_norm": 0.17891732502054444,
      "learning_rate": 2.364607166559063e-06,
      "loss": 3.4009,
      "step": 4230
    },
    {
      "epoch": 0.9701903233203394,
      "grad_norm": 0.16929111003469463,
      "learning_rate": 2.3286720250246253e-06,
      "loss": 3.4172,
      "step": 4231
    },
    {
      "epoch": 0.9704196285255675,
      "grad_norm": 0.1807623483402604,
      "learning_rate": 2.2930113956655674e-06,
      "loss": 3.4188,
      "step": 4232
    },
    {
      "epoch": 0.9706489337307956,
      "grad_norm": 0.18586924505979496,
      "learning_rate": 2.2576252981520994e-06,
      "loss": 3.438,
      "step": 4233
    },
    {
      "epoch": 0.9708782389360239,
      "grad_norm": 0.17668598184774487,
      "learning_rate": 2.2225137520029416e-06,
      "loss": 3.4152,
      "step": 4234
    },
    {
      "epoch": 0.971107544141252,
      "grad_norm": 0.17176900751620583,
      "learning_rate": 2.1876767765853233e-06,
      "loss": 3.4169,
      "step": 4235
    },
    {
      "epoch": 0.9713368493464801,
      "grad_norm": 0.16151579804910915,
      "learning_rate": 2.153114391115152e-06,
      "loss": 3.4301,
      "step": 4236
    },
    {
      "epoch": 0.9715661545517084,
      "grad_norm": 0.18373369843544304,
      "learning_rate": 2.118826614656788e-06,
      "loss": 3.5453,
      "step": 4237
    },
    {
      "epoch": 0.9717954597569365,
      "grad_norm": 0.1915763235147539,
      "learning_rate": 2.084813466123159e-06,
      "loss": 3.2869,
      "step": 4238
    },
    {
      "epoch": 0.9720247649621646,
      "grad_norm": 0.16238361414435196,
      "learning_rate": 2.051074964275701e-06,
      "loss": 3.442,
      "step": 4239
    },
    {
      "epoch": 0.9722540701673928,
      "grad_norm": 0.17141381417488008,
      "learning_rate": 2.0176111277242502e-06,
      "loss": 3.3781,
      "step": 4240
    },
    {
      "epoch": 0.972483375372621,
      "grad_norm": 0.18379356949034883,
      "learning_rate": 1.984421974927375e-06,
      "loss": 3.3076,
      "step": 4241
    },
    {
      "epoch": 0.9727126805778491,
      "grad_norm": 0.17730790263733204,
      "learning_rate": 1.95150752419182e-06,
      "loss": 3.3733,
      "step": 4242
    },
    {
      "epoch": 0.9729419857830772,
      "grad_norm": 0.16930800270841626,
      "learning_rate": 1.9188677936731734e-06,
      "loss": 3.3093,
      "step": 4243
    },
    {
      "epoch": 0.9731712909883055,
      "grad_norm": 0.1776110892674154,
      "learning_rate": 1.8865028013751451e-06,
      "loss": 3.5015,
      "step": 4244
    },
    {
      "epoch": 0.9734005961935336,
      "grad_norm": 0.1558259857468744,
      "learning_rate": 1.8544125651501208e-06,
      "loss": 3.4759,
      "step": 4245
    },
    {
      "epoch": 0.9736299013987617,
      "grad_norm": 0.16157662104382167,
      "learning_rate": 1.8225971026987753e-06,
      "loss": 3.4234,
      "step": 4246
    },
    {
      "epoch": 0.97385920660399,
      "grad_norm": 0.17369543221443193,
      "learning_rate": 1.7910564315704035e-06,
      "loss": 3.4803,
      "step": 4247
    },
    {
      "epoch": 0.9740885118092181,
      "grad_norm": 0.17759916272460188,
      "learning_rate": 1.7597905691625893e-06,
      "loss": 3.3395,
      "step": 4248
    },
    {
      "epoch": 0.9743178170144462,
      "grad_norm": 0.18552370937313262,
      "learning_rate": 1.7287995327214257e-06,
      "loss": 3.3317,
      "step": 4249
    },
    {
      "epoch": 0.9745471222196744,
      "grad_norm": 0.16286015374315416,
      "learning_rate": 1.6980833393412388e-06,
      "loss": 3.4323,
      "step": 4250
    },
    {
      "epoch": 0.9747764274249026,
      "grad_norm": 0.16490338658881265,
      "learning_rate": 1.6676420059649754e-06,
      "loss": 3.4244,
      "step": 4251
    },
    {
      "epoch": 0.9750057326301307,
      "grad_norm": 0.16467188890669027,
      "learning_rate": 1.6374755493838155e-06,
      "loss": 3.4656,
      "step": 4252
    },
    {
      "epoch": 0.9752350378353588,
      "grad_norm": 0.17848481620451537,
      "learning_rate": 1.6075839862374485e-06,
      "loss": 3.4351,
      "step": 4253
    },
    {
      "epoch": 0.975464343040587,
      "grad_norm": 0.16508827156400804,
      "learning_rate": 1.5779673330137966e-06,
      "loss": 3.3636,
      "step": 4254
    },
    {
      "epoch": 0.9756936482458152,
      "grad_norm": 0.16459095088502734,
      "learning_rate": 1.5486256060492366e-06,
      "loss": 3.4332,
      "step": 4255
    },
    {
      "epoch": 0.9759229534510433,
      "grad_norm": 0.1813720298567073,
      "learning_rate": 1.5195588215283773e-06,
      "loss": 3.5008,
      "step": 4256
    },
    {
      "epoch": 0.9761522586562715,
      "grad_norm": 0.17026679819180818,
      "learning_rate": 1.4907669954844495e-06,
      "loss": 3.4182,
      "step": 4257
    },
    {
      "epoch": 0.9763815638614997,
      "grad_norm": 0.1848984344531851,
      "learning_rate": 1.462250143798638e-06,
      "loss": 3.4764,
      "step": 4258
    },
    {
      "epoch": 0.9766108690667278,
      "grad_norm": 0.1874749390822543,
      "learning_rate": 1.434008282200805e-06,
      "loss": 3.4771,
      "step": 4259
    },
    {
      "epoch": 0.976840174271956,
      "grad_norm": 0.17835501171400428,
      "learning_rate": 1.406041426268878e-06,
      "loss": 3.363,
      "step": 4260
    },
    {
      "epoch": 0.9770694794771841,
      "grad_norm": 0.163284229953889,
      "learning_rate": 1.3783495914291844e-06,
      "loss": 3.3907,
      "step": 4261
    },
    {
      "epoch": 0.9772987846824123,
      "grad_norm": 0.16833850959528682,
      "learning_rate": 1.3509327929563942e-06,
      "loss": 3.4079,
      "step": 4262
    },
    {
      "epoch": 0.9775280898876404,
      "grad_norm": 0.16976730796082581,
      "learning_rate": 1.3237910459734104e-06,
      "loss": 3.3905,
      "step": 4263
    },
    {
      "epoch": 0.9777573950928686,
      "grad_norm": 0.15972981726231172,
      "learning_rate": 1.2969243654514796e-06,
      "loss": 3.5977,
      "step": 4264
    },
    {
      "epoch": 0.9779867002980968,
      "grad_norm": 0.17022055482716192,
      "learning_rate": 1.270332766210025e-06,
      "loss": 3.3718,
      "step": 4265
    },
    {
      "epoch": 0.9782160055033249,
      "grad_norm": 0.17574356157806428,
      "learning_rate": 1.2440162629168138e-06,
      "loss": 3.5637,
      "step": 4266
    },
    {
      "epoch": 0.9784453107085531,
      "grad_norm": 0.20059759307225428,
      "learning_rate": 1.2179748700879012e-06,
      "loss": 3.4376,
      "step": 4267
    },
    {
      "epoch": 0.9786746159137812,
      "grad_norm": 0.1604061869202046,
      "learning_rate": 1.1922086020874634e-06,
      "loss": 3.4444,
      "step": 4268
    },
    {
      "epoch": 0.9789039211190094,
      "grad_norm": 0.1827622427525679,
      "learning_rate": 1.1667174731280205e-06,
      "loss": 3.6006,
      "step": 4269
    },
    {
      "epoch": 0.9791332263242376,
      "grad_norm": 0.17653265072567642,
      "learning_rate": 1.1415014972703808e-06,
      "loss": 3.5358,
      "step": 4270
    },
    {
      "epoch": 0.9793625315294657,
      "grad_norm": 0.18091059302120632,
      "learning_rate": 1.1165606884234182e-06,
      "loss": 3.4649,
      "step": 4271
    },
    {
      "epoch": 0.9795918367346939,
      "grad_norm": 0.17433298928802216,
      "learning_rate": 1.0918950603444055e-06,
      "loss": 3.5069,
      "step": 4272
    },
    {
      "epoch": 0.9798211419399221,
      "grad_norm": 0.1686261162086502,
      "learning_rate": 1.0675046266386268e-06,
      "loss": 3.4748,
      "step": 4273
    },
    {
      "epoch": 0.9800504471451502,
      "grad_norm": 0.15754789968588004,
      "learning_rate": 1.0433894007597643e-06,
      "loss": 3.3829,
      "step": 4274
    },
    {
      "epoch": 0.9802797523503783,
      "grad_norm": 0.17555748008910793,
      "learning_rate": 1.019549396009567e-06,
      "loss": 3.4457,
      "step": 4275
    },
    {
      "epoch": 0.9805090575556065,
      "grad_norm": 0.17896015441561205,
      "learning_rate": 9.959846255381267e-07,
      "loss": 3.5035,
      "step": 4276
    },
    {
      "epoch": 0.9807383627608347,
      "grad_norm": 0.1750517382741998,
      "learning_rate": 9.726951023434348e-07,
      "loss": 3.4643,
      "step": 4277
    },
    {
      "epoch": 0.9809676679660628,
      "grad_norm": 0.18822278502597037,
      "learning_rate": 9.496808392719935e-07,
      "loss": 3.541,
      "step": 4278
    },
    {
      "epoch": 0.981196973171291,
      "grad_norm": 0.17489824581828023,
      "learning_rate": 9.269418490182591e-07,
      "loss": 3.4025,
      "step": 4279
    },
    {
      "epoch": 0.9814262783765192,
      "grad_norm": 0.1767932672140336,
      "learning_rate": 9.044781441249205e-07,
      "loss": 3.2822,
      "step": 4280
    },
    {
      "epoch": 0.9816555835817473,
      "grad_norm": 0.16532328307712799,
      "learning_rate": 8.822897369827332e-07,
      "loss": 3.5147,
      "step": 4281
    },
    {
      "epoch": 0.9818848887869754,
      "grad_norm": 0.19401518626108874,
      "learning_rate": 8.603766398307956e-07,
      "loss": 3.5431,
      "step": 4282
    },
    {
      "epoch": 0.9821141939922037,
      "grad_norm": 0.1734736630106125,
      "learning_rate": 8.387388647561611e-07,
      "loss": 3.4125,
      "step": 4283
    },
    {
      "epoch": 0.9823434991974318,
      "grad_norm": 0.16591208745467204,
      "learning_rate": 8.173764236940606e-07,
      "loss": 3.462,
      "step": 4284
    },
    {
      "epoch": 0.9825728044026599,
      "grad_norm": 0.1756847363546272,
      "learning_rate": 7.962893284279016e-07,
      "loss": 3.4721,
      "step": 4285
    },
    {
      "epoch": 0.982802109607888,
      "grad_norm": 0.16360990001608108,
      "learning_rate": 7.754775905891575e-07,
      "loss": 3.4369,
      "step": 4286
    },
    {
      "epoch": 0.9830314148131163,
      "grad_norm": 0.16241952848156954,
      "learning_rate": 7.549412216574791e-07,
      "loss": 3.4824,
      "step": 4287
    },
    {
      "epoch": 0.9832607200183444,
      "grad_norm": 0.17107061264264214,
      "learning_rate": 7.346802329605828e-07,
      "loss": 3.4428,
      "step": 4288
    },
    {
      "epoch": 0.9834900252235725,
      "grad_norm": 0.17637611965383063,
      "learning_rate": 7.146946356743067e-07,
      "loss": 3.4348,
      "step": 4289
    },
    {
      "epoch": 0.9837193304288008,
      "grad_norm": 0.18559583810679794,
      "learning_rate": 6.949844408225548e-07,
      "loss": 3.4169,
      "step": 4290
    },
    {
      "epoch": 0.9839486356340289,
      "grad_norm": 0.15505137388669987,
      "learning_rate": 6.755496592773524e-07,
      "loss": 3.4876,
      "step": 4291
    },
    {
      "epoch": 0.984177940839257,
      "grad_norm": 0.1611558528967555,
      "learning_rate": 6.563903017587358e-07,
      "loss": 3.3723,
      "step": 4292
    },
    {
      "epoch": 0.9844072460444853,
      "grad_norm": 0.16857743427973296,
      "learning_rate": 6.375063788349733e-07,
      "loss": 3.4873,
      "step": 4293
    },
    {
      "epoch": 0.9846365512497134,
      "grad_norm": 0.17461869987561704,
      "learning_rate": 6.188979009222329e-07,
      "loss": 3.5246,
      "step": 4294
    },
    {
      "epoch": 0.9848658564549415,
      "grad_norm": 0.17979260196617808,
      "learning_rate": 6.005648782848594e-07,
      "loss": 3.4247,
      "step": 4295
    },
    {
      "epoch": 0.9850951616601696,
      "grad_norm": 0.16476375499931614,
      "learning_rate": 5.825073210352083e-07,
      "loss": 3.5238,
      "step": 4296
    },
    {
      "epoch": 0.9853244668653979,
      "grad_norm": 0.16753410853657516,
      "learning_rate": 5.647252391337565e-07,
      "loss": 3.455,
      "step": 4297
    },
    {
      "epoch": 0.985553772070626,
      "grad_norm": 0.1720591702832646,
      "learning_rate": 5.472186423889358e-07,
      "loss": 3.2953,
      "step": 4298
    },
    {
      "epoch": 0.9857830772758541,
      "grad_norm": 0.176429553575321,
      "learning_rate": 5.299875404572441e-07,
      "loss": 3.4304,
      "step": 4299
    },
    {
      "epoch": 0.9860123824810824,
      "grad_norm": 0.17043608483781164,
      "learning_rate": 5.130319428433005e-07,
      "loss": 3.4857,
      "step": 4300
    },
    {
      "epoch": 0.9862416876863105,
      "grad_norm": 0.16838958163531878,
      "learning_rate": 4.963518588996796e-07,
      "loss": 3.403,
      "step": 4301
    },
    {
      "epoch": 0.9864709928915386,
      "grad_norm": 0.16239879104797922,
      "learning_rate": 4.799472978269659e-07,
      "loss": 3.2837,
      "step": 4302
    },
    {
      "epoch": 0.9867002980967668,
      "grad_norm": 0.16268558647265535,
      "learning_rate": 4.638182686738657e-07,
      "loss": 3.5091,
      "step": 4303
    },
    {
      "epoch": 0.986929603301995,
      "grad_norm": 0.22704521050976129,
      "learning_rate": 4.4796478033704015e-07,
      "loss": 3.3653,
      "step": 4304
    },
    {
      "epoch": 0.9871589085072231,
      "grad_norm": 0.1644616519130453,
      "learning_rate": 4.3238684156110543e-07,
      "loss": 3.3436,
      "step": 4305
    },
    {
      "epoch": 0.9873882137124512,
      "grad_norm": 0.17230879478626837,
      "learning_rate": 4.170844609387991e-07,
      "loss": 3.5609,
      "step": 4306
    },
    {
      "epoch": 0.9876175189176795,
      "grad_norm": 0.17537338556415225,
      "learning_rate": 4.020576469108139e-07,
      "loss": 3.3966,
      "step": 4307
    },
    {
      "epoch": 0.9878468241229076,
      "grad_norm": 0.16797474003938528,
      "learning_rate": 3.873064077657973e-07,
      "loss": 3.2175,
      "step": 4308
    },
    {
      "epoch": 0.9880761293281357,
      "grad_norm": 0.17381755778888644,
      "learning_rate": 3.7283075164046274e-07,
      "loss": 3.4258,
      "step": 4309
    },
    {
      "epoch": 0.988305434533364,
      "grad_norm": 0.16690098058545252,
      "learning_rate": 3.5863068651953436e-07,
      "loss": 3.3893,
      "step": 4310
    },
    {
      "epoch": 0.9885347397385921,
      "grad_norm": 0.18626949951805966,
      "learning_rate": 3.4470622023557995e-07,
      "loss": 3.527,
      "step": 4311
    },
    {
      "epoch": 0.9887640449438202,
      "grad_norm": 0.17575166819853918,
      "learning_rate": 3.3105736046928903e-07,
      "loss": 3.4462,
      "step": 4312
    },
    {
      "epoch": 0.9889933501490484,
      "grad_norm": 0.1906471420090877,
      "learning_rate": 3.176841147492504e-07,
      "loss": 3.4531,
      "step": 4313
    },
    {
      "epoch": 0.9892226553542766,
      "grad_norm": 0.17295939409642452,
      "learning_rate": 3.0458649045211895e-07,
      "loss": 3.2477,
      "step": 4314
    },
    {
      "epoch": 0.9894519605595047,
      "grad_norm": 0.17335884825467515,
      "learning_rate": 2.9176449480244895e-07,
      "loss": 3.4388,
      "step": 4315
    },
    {
      "epoch": 0.9896812657647328,
      "grad_norm": 0.1969696246278778,
      "learning_rate": 2.792181348726941e-07,
      "loss": 3.2816,
      "step": 4316
    },
    {
      "epoch": 0.989910570969961,
      "grad_norm": 0.1628198683590351,
      "learning_rate": 2.6694741758342967e-07,
      "loss": 3.3937,
      "step": 4317
    },
    {
      "epoch": 0.9901398761751892,
      "grad_norm": 0.1696942768988349,
      "learning_rate": 2.549523497030193e-07,
      "loss": 3.5479,
      "step": 4318
    },
    {
      "epoch": 0.9903691813804173,
      "grad_norm": 0.17567087666803233,
      "learning_rate": 2.432329378478926e-07,
      "loss": 3.2472,
      "step": 4319
    },
    {
      "epoch": 0.9905984865856455,
      "grad_norm": 0.17351086815882663,
      "learning_rate": 2.3178918848243413e-07,
      "loss": 3.5444,
      "step": 4320
    },
    {
      "epoch": 0.9908277917908737,
      "grad_norm": 0.18661886876342323,
      "learning_rate": 2.2062110791892798e-07,
      "loss": 3.6157,
      "step": 4321
    },
    {
      "epoch": 0.9910570969961018,
      "grad_norm": 0.1783379396079576,
      "learning_rate": 2.097287023176131e-07,
      "loss": 3.5078,
      "step": 4322
    },
    {
      "epoch": 0.99128640220133,
      "grad_norm": 0.17325508119833108,
      "learning_rate": 1.9911197768662792e-07,
      "loss": 3.5009,
      "step": 4323
    },
    {
      "epoch": 0.9915157074065581,
      "grad_norm": 0.17090596336752678,
      "learning_rate": 1.8877093988212135e-07,
      "loss": 3.3043,
      "step": 4324
    },
    {
      "epoch": 0.9917450126117863,
      "grad_norm": 0.17914234329521223,
      "learning_rate": 1.7870559460814173e-07,
      "loss": 3.3503,
      "step": 4325
    },
    {
      "epoch": 0.9919743178170144,
      "grad_norm": 0.15379785965988932,
      "learning_rate": 1.6891594741663684e-07,
      "loss": 3.4129,
      "step": 4326
    },
    {
      "epoch": 0.9922036230222426,
      "grad_norm": 0.16281806279740366,
      "learning_rate": 1.5940200370750947e-07,
      "loss": 3.5301,
      "step": 4327
    },
    {
      "epoch": 0.9924329282274708,
      "grad_norm": 0.1644978705209607,
      "learning_rate": 1.5016376872867277e-07,
      "loss": 3.5663,
      "step": 4328
    },
    {
      "epoch": 0.9926622334326989,
      "grad_norm": 0.16927321110899848,
      "learning_rate": 1.4120124757577291e-07,
      "loss": 3.5699,
      "step": 4329
    },
    {
      "epoch": 0.9928915386379271,
      "grad_norm": 0.19191896986408047,
      "learning_rate": 1.3251444519246647e-07,
      "loss": 3.5808,
      "step": 4330
    },
    {
      "epoch": 0.9931208438431552,
      "grad_norm": 0.1724538841223402,
      "learning_rate": 1.2410336637047603e-07,
      "loss": 3.403,
      "step": 4331
    },
    {
      "epoch": 0.9933501490483834,
      "grad_norm": 0.16563014045789273,
      "learning_rate": 1.1596801574914606e-07,
      "loss": 3.2012,
      "step": 4332
    },
    {
      "epoch": 0.9935794542536116,
      "grad_norm": 0.1595624260823157,
      "learning_rate": 1.081083978159425e-07,
      "loss": 3.3989,
      "step": 4333
    },
    {
      "epoch": 0.9938087594588397,
      "grad_norm": 0.17601785607583761,
      "learning_rate": 1.0052451690617526e-07,
      "loss": 3.5385,
      "step": 4334
    },
    {
      "epoch": 0.9940380646640679,
      "grad_norm": 0.18081483884526242,
      "learning_rate": 9.321637720310915e-08,
      "loss": 3.4881,
      "step": 4335
    },
    {
      "epoch": 0.994267369869296,
      "grad_norm": 0.1781694835190791,
      "learning_rate": 8.618398273779749e-08,
      "loss": 3.4724,
      "step": 4336
    },
    {
      "epoch": 0.9944966750745242,
      "grad_norm": 0.15386063369428113,
      "learning_rate": 7.942733738924845e-08,
      "loss": 3.315,
      "step": 4337
    },
    {
      "epoch": 0.9947259802797523,
      "grad_norm": 0.18565650047945667,
      "learning_rate": 7.294644488442526e-08,
      "loss": 3.3633,
      "step": 4338
    },
    {
      "epoch": 0.9949552854849805,
      "grad_norm": 0.16568574707824288,
      "learning_rate": 6.6741308798135e-08,
      "loss": 3.3743,
      "step": 4339
    },
    {
      "epoch": 0.9951845906902087,
      "grad_norm": 0.17989512598253077,
      "learning_rate": 6.081193255313977e-08,
      "loss": 3.3187,
      "step": 4340
    },
    {
      "epoch": 0.9954138958954368,
      "grad_norm": 0.17471339375765876,
      "learning_rate": 5.5158319419934546e-08,
      "loss": 3.4519,
      "step": 4341
    },
    {
      "epoch": 0.995643201100665,
      "grad_norm": 0.15744035953032282,
      "learning_rate": 4.978047251708029e-08,
      "loss": 3.3791,
      "step": 4342
    },
    {
      "epoch": 0.9958725063058932,
      "grad_norm": 0.16948149057816905,
      "learning_rate": 4.4678394810981906e-08,
      "loss": 3.2616,
      "step": 4343
    },
    {
      "epoch": 0.9961018115111213,
      "grad_norm": 0.15786162756866334,
      "learning_rate": 3.9852089115888226e-08,
      "loss": 3.2873,
      "step": 4344
    },
    {
      "epoch": 0.9963311167163494,
      "grad_norm": 0.1679875890862027,
      "learning_rate": 3.5301558093947527e-08,
      "loss": 3.247,
      "step": 4345
    },
    {
      "epoch": 0.9965604219215776,
      "grad_norm": 0.17469773596795407,
      "learning_rate": 3.102680425520754e-08,
      "loss": 3.4207,
      "step": 4346
    },
    {
      "epoch": 0.9967897271268058,
      "grad_norm": 0.171382174036622,
      "learning_rate": 2.7027829957559925e-08,
      "loss": 3.3945,
      "step": 4347
    },
    {
      "epoch": 0.9970190323320339,
      "grad_norm": 0.15938427165584856,
      "learning_rate": 2.3304637406906802e-08,
      "loss": 3.3329,
      "step": 4348
    },
    {
      "epoch": 0.997248337537262,
      "grad_norm": 0.1651463998318795,
      "learning_rate": 1.985722865682771e-08,
      "loss": 3.3393,
      "step": 4349
    },
    {
      "epoch": 0.9974776427424903,
      "grad_norm": 0.16496443949430037,
      "learning_rate": 1.6685605608968145e-08,
      "loss": 3.5102,
      "step": 4350
    },
    {
      "epoch": 0.9977069479477184,
      "grad_norm": 0.17482395736835685,
      "learning_rate": 1.3789770012762048e-08,
      "loss": 3.411,
      "step": 4351
    },
    {
      "epoch": 0.9979362531529465,
      "grad_norm": 0.16797867425058485,
      "learning_rate": 1.116972346548728e-08,
      "loss": 3.5849,
      "step": 4352
    },
    {
      "epoch": 0.9981655583581748,
      "grad_norm": 0.16813620319568248,
      "learning_rate": 8.825467412376665e-09,
      "loss": 3.4584,
      "step": 4353
    },
    {
      "epoch": 0.9983948635634029,
      "grad_norm": 0.18134139189656326,
      "learning_rate": 6.757003146506957e-09,
      "loss": 3.5944,
      "step": 4354
    },
    {
      "epoch": 0.998624168768631,
      "grad_norm": 0.1616904899832619,
      "learning_rate": 4.96433180879885e-09,
      "loss": 3.3765,
      "step": 4355
    },
    {
      "epoch": 0.9988534739738592,
      "grad_norm": 0.1762908394302317,
      "learning_rate": 3.4474543881279906e-09,
      "loss": 3.5386,
      "step": 4356
    },
    {
      "epoch": 0.9990827791790874,
      "grad_norm": 0.17410374850323737,
      "learning_rate": 2.206371721158451e-09,
      "loss": 3.3464,
      "step": 4357
    },
    {
      "epoch": 0.9993120843843155,
      "grad_norm": 0.16347766088082094,
      "learning_rate": 1.241084492509259e-09,
      "loss": 3.4046,
      "step": 4358
    },
    {
      "epoch": 0.9995413895895436,
      "grad_norm": 0.18918113488101013,
      "learning_rate": 5.515932345323549e-10,
      "loss": 3.6387,
      "step": 4359
    },
    {
      "epoch": 0.9997706947947719,
      "grad_norm": 0.16685953079149596,
      "learning_rate": 1.37898327645658e-10,
      "loss": 3.4552,
      "step": 4360
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.16955091803169606,
      "learning_rate": 0.0,
      "loss": 3.531,
      "step": 4361
    },
    {
      "epoch": 1.0,
      "step": 4361,
      "total_flos": 22861032554496.0,
      "train_loss": 3.532874656964595,
      "train_runtime": 3070.5661,
      "train_samples_per_second": 181.767,
      "train_steps_per_second": 1.42
    }
  ],
  "logging_steps": 1.0,
  "max_steps": 4361,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 1000.0,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": false,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 22861032554496.0,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}