{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.273037542662116,
  "eval_steps": 500,
  "global_step": 9990,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.00022753128555176336,
      "grad_norm": 12.279616595766896,
      "learning_rate": 1.25e-06,
      "loss": 0.1112,
      "step": 1
    },
    {
      "epoch": 0.0004550625711035267,
      "grad_norm": 7.2629005990045785,
      "learning_rate": 1.2499999936130725e-06,
      "loss": 0.1394,
      "step": 2
    },
    {
      "epoch": 0.0006825938566552901,
      "grad_norm": 9.061755966937882,
      "learning_rate": 1.2499999744522896e-06,
      "loss": 0.266,
      "step": 3
    },
    {
      "epoch": 0.0009101251422070534,
      "grad_norm": 4.366742138714415,
      "learning_rate": 1.2499999425176518e-06,
      "loss": 0.1258,
      "step": 4
    },
    {
      "epoch": 0.0011376564277588168,
      "grad_norm": 8.875041328214682,
      "learning_rate": 1.2499998978091598e-06,
      "loss": 0.1749,
      "step": 5
    },
    {
      "epoch": 0.0013651877133105802,
      "grad_norm": 8.800851302952648,
      "learning_rate": 1.2499998403268147e-06,
      "loss": 0.2,
      "step": 6
    },
    {
      "epoch": 0.0015927189988623437,
      "grad_norm": 3.6087293622719416,
      "learning_rate": 1.2499997700706173e-06,
      "loss": 0.1398,
      "step": 7
    },
    {
      "epoch": 0.0018202502844141069,
      "grad_norm": 3.42318784466345,
      "learning_rate": 1.2499996870405692e-06,
      "loss": 0.1226,
      "step": 8
    },
    {
      "epoch": 0.0020477815699658703,
      "grad_norm": 10.818031536677639,
      "learning_rate": 1.2499995912366722e-06,
      "loss": 0.171,
      "step": 9
    },
    {
      "epoch": 0.0022753128555176336,
      "grad_norm": 4.021373220825307,
      "learning_rate": 1.2499994826589282e-06,
      "loss": 0.1326,
      "step": 10
    },
    {
      "epoch": 0.002502844141069397,
      "grad_norm": 23.4957115965293,
      "learning_rate": 1.2499993613073393e-06,
      "loss": 0.1235,
      "step": 11
    },
    {
      "epoch": 0.0027303754266211604,
      "grad_norm": 21.545758763913824,
      "learning_rate": 1.2499992271819083e-06,
      "loss": 0.0925,
      "step": 12
    },
    {
      "epoch": 0.0029579067121729237,
      "grad_norm": 3.510359233156002,
      "learning_rate": 1.2499990802826377e-06,
      "loss": 0.0985,
      "step": 13
    },
    {
      "epoch": 0.0031854379977246873,
      "grad_norm": 3.9794392758691632,
      "learning_rate": 1.2499989206095304e-06,
      "loss": 0.0906,
      "step": 14
    },
    {
      "epoch": 0.0034129692832764505,
      "grad_norm": 2.4759900062021503,
      "learning_rate": 1.2499987481625899e-06,
      "loss": 0.0746,
      "step": 15
    },
    {
      "epoch": 0.0036405005688282138,
      "grad_norm": 7.91228269690243,
      "learning_rate": 1.2499985629418195e-06,
      "loss": 0.1381,
      "step": 16
    },
    {
      "epoch": 0.0038680318543799774,
      "grad_norm": 3.050431958605509,
      "learning_rate": 1.2499983649472233e-06,
      "loss": 0.0509,
      "step": 17
    },
    {
      "epoch": 0.004095563139931741,
      "grad_norm": 9.537504830414553,
      "learning_rate": 1.249998154178805e-06,
      "loss": 0.0658,
      "step": 18
    },
    {
      "epoch": 0.004323094425483504,
      "grad_norm": 5.663160798723299,
      "learning_rate": 1.2499979306365692e-06,
      "loss": 0.097,
      "step": 19
    },
    {
      "epoch": 0.004550625711035267,
      "grad_norm": 13.411158998212743,
      "learning_rate": 1.2499976943205202e-06,
      "loss": 0.1,
      "step": 20
    },
    {
      "epoch": 0.00477815699658703,
      "grad_norm": 3.6621965264316194,
      "learning_rate": 1.249997445230663e-06,
      "loss": 0.0698,
      "step": 21
    },
    {
      "epoch": 0.005005688282138794,
      "grad_norm": 31.354306922977063,
      "learning_rate": 1.2499971833670026e-06,
      "loss": 0.1129,
      "step": 22
    },
    {
      "epoch": 0.005233219567690558,
      "grad_norm": 4.025175447301565,
      "learning_rate": 1.2499969087295443e-06,
      "loss": 0.1076,
      "step": 23
    },
    {
      "epoch": 0.005460750853242321,
      "grad_norm": 3.7989823695443863,
      "learning_rate": 1.249996621318294e-06,
      "loss": 0.0902,
      "step": 24
    },
    {
      "epoch": 0.005688282138794084,
      "grad_norm": 2.2185511399815545,
      "learning_rate": 1.2499963211332573e-06,
      "loss": 0.0398,
      "step": 25
    },
    {
      "epoch": 0.005915813424345847,
      "grad_norm": 3.775038573479949,
      "learning_rate": 1.2499960081744405e-06,
      "loss": 0.1145,
      "step": 26
    },
    {
      "epoch": 0.0061433447098976105,
      "grad_norm": 7.295114622894012,
      "learning_rate": 1.24999568244185e-06,
      "loss": 0.1379,
      "step": 27
    },
    {
      "epoch": 0.006370875995449375,
      "grad_norm": 21.290899699279613,
      "learning_rate": 1.249995343935492e-06,
      "loss": 0.0995,
      "step": 28
    },
    {
      "epoch": 0.006598407281001138,
      "grad_norm": 3.593142181899087,
      "learning_rate": 1.2499949926553743e-06,
      "loss": 0.0973,
      "step": 29
    },
    {
      "epoch": 0.006825938566552901,
      "grad_norm": 7.737004888367107,
      "learning_rate": 1.2499946286015032e-06,
      "loss": 0.09,
      "step": 30
    },
    {
      "epoch": 0.007053469852104664,
      "grad_norm": 2.955413576195516,
      "learning_rate": 1.2499942517738867e-06,
      "loss": 0.0625,
      "step": 31
    },
    {
      "epoch": 0.0072810011376564275,
      "grad_norm": 4.138230802608221,
      "learning_rate": 1.2499938621725322e-06,
      "loss": 0.0749,
      "step": 32
    },
    {
      "epoch": 0.007508532423208191,
      "grad_norm": 2.2958240484880115,
      "learning_rate": 1.2499934597974478e-06,
      "loss": 0.0986,
      "step": 33
    },
    {
      "epoch": 0.007736063708759955,
      "grad_norm": 3.9060226417706256,
      "learning_rate": 1.2499930446486416e-06,
      "loss": 0.1296,
      "step": 34
    },
    {
      "epoch": 0.007963594994311717,
      "grad_norm": 1.5544518249405173,
      "learning_rate": 1.2499926167261224e-06,
      "loss": 0.0598,
      "step": 35
    },
    {
      "epoch": 0.008191126279863481,
      "grad_norm": 5.2971839668544884,
      "learning_rate": 1.2499921760298987e-06,
      "loss": 0.1216,
      "step": 36
    },
    {
      "epoch": 0.008418657565415245,
      "grad_norm": 6.168021178604936,
      "learning_rate": 1.2499917225599796e-06,
      "loss": 0.1257,
      "step": 37
    },
    {
      "epoch": 0.008646188850967008,
      "grad_norm": 5.087771081836934,
      "learning_rate": 1.2499912563163742e-06,
      "loss": 0.0695,
      "step": 38
    },
    {
      "epoch": 0.008873720136518772,
      "grad_norm": 7.633032375201886,
      "learning_rate": 1.249990777299092e-06,
      "loss": 0.0522,
      "step": 39
    },
    {
      "epoch": 0.009101251422070534,
      "grad_norm": 2.9414497498297734,
      "learning_rate": 1.249990285508143e-06,
      "loss": 0.081,
      "step": 40
    },
    {
      "epoch": 0.009328782707622298,
      "grad_norm": 4.449234015706431,
      "learning_rate": 1.2499897809435374e-06,
      "loss": 0.1104,
      "step": 41
    },
    {
      "epoch": 0.00955631399317406,
      "grad_norm": 17.29783348718493,
      "learning_rate": 1.249989263605285e-06,
      "loss": 0.0952,
      "step": 42
    },
    {
      "epoch": 0.009783845278725825,
      "grad_norm": 4.357226856628825,
      "learning_rate": 1.249988733493397e-06,
      "loss": 0.0639,
      "step": 43
    },
    {
      "epoch": 0.010011376564277589,
      "grad_norm": 4.248351802911978,
      "learning_rate": 1.2499881906078836e-06,
      "loss": 0.0845,
      "step": 44
    },
    {
      "epoch": 0.010238907849829351,
      "grad_norm": 2.230593274135571,
      "learning_rate": 1.2499876349487564e-06,
      "loss": 0.1031,
      "step": 45
    },
    {
      "epoch": 0.010466439135381115,
      "grad_norm": 5.02333552530202,
      "learning_rate": 1.2499870665160262e-06,
      "loss": 0.0763,
      "step": 46
    },
    {
      "epoch": 0.010693970420932878,
      "grad_norm": 1.7336741239976143,
      "learning_rate": 1.2499864853097054e-06,
      "loss": 0.0495,
      "step": 47
    },
    {
      "epoch": 0.010921501706484642,
      "grad_norm": 4.9863935956267325,
      "learning_rate": 1.2499858913298053e-06,
      "loss": 0.1674,
      "step": 48
    },
    {
      "epoch": 0.011149032992036406,
      "grad_norm": 2.8583968641628967,
      "learning_rate": 1.249985284576338e-06,
      "loss": 0.0728,
      "step": 49
    },
    {
      "epoch": 0.011376564277588168,
      "grad_norm": 5.82977520269559,
      "learning_rate": 1.2499846650493164e-06,
      "loss": 0.1076,
      "step": 50
    },
    {
      "epoch": 0.011604095563139932,
      "grad_norm": 2.407346961463521,
      "learning_rate": 1.2499840327487528e-06,
      "loss": 0.0542,
      "step": 51
    },
    {
      "epoch": 0.011831626848691695,
      "grad_norm": 1.7581308929371289,
      "learning_rate": 1.24998338767466e-06,
      "loss": 0.0851,
      "step": 52
    },
    {
      "epoch": 0.012059158134243459,
      "grad_norm": 3.6069730541490315,
      "learning_rate": 1.2499827298270515e-06,
      "loss": 0.0771,
      "step": 53
    },
    {
      "epoch": 0.012286689419795221,
      "grad_norm": 2.6384830282527965,
      "learning_rate": 1.2499820592059405e-06,
      "loss": 0.1336,
      "step": 54
    },
    {
      "epoch": 0.012514220705346985,
      "grad_norm": 3.1162890015900793,
      "learning_rate": 1.2499813758113409e-06,
      "loss": 0.1058,
      "step": 55
    },
    {
      "epoch": 0.01274175199089875,
      "grad_norm": 3.5247536707308966,
      "learning_rate": 1.2499806796432665e-06,
      "loss": 0.0717,
      "step": 56
    },
    {
      "epoch": 0.012969283276450512,
      "grad_norm": 2.2371436604921584,
      "learning_rate": 1.2499799707017315e-06,
      "loss": 0.07,
      "step": 57
    },
    {
      "epoch": 0.013196814562002276,
      "grad_norm": 5.912440334168246,
      "learning_rate": 1.2499792489867508e-06,
      "loss": 0.1588,
      "step": 58
    },
    {
      "epoch": 0.013424345847554038,
      "grad_norm": 1.6415965725415542,
      "learning_rate": 1.2499785144983386e-06,
      "loss": 0.0587,
      "step": 59
    },
    {
      "epoch": 0.013651877133105802,
      "grad_norm": 3.921979088226587,
      "learning_rate": 1.24997776723651e-06,
      "loss": 0.1403,
      "step": 60
    },
    {
      "epoch": 0.013879408418657566,
      "grad_norm": 4.854246136713843,
      "learning_rate": 1.2499770072012809e-06,
      "loss": 0.0706,
      "step": 61
    },
    {
      "epoch": 0.014106939704209329,
      "grad_norm": 2.249851975045142,
      "learning_rate": 1.2499762343926661e-06,
      "loss": 0.0988,
      "step": 62
    },
    {
      "epoch": 0.014334470989761093,
      "grad_norm": 2.5833858090734956,
      "learning_rate": 1.2499754488106817e-06,
      "loss": 0.0618,
      "step": 63
    },
    {
      "epoch": 0.014562002275312855,
      "grad_norm": 3.523489637935598,
      "learning_rate": 1.2499746504553436e-06,
      "loss": 0.0788,
      "step": 64
    },
    {
      "epoch": 0.01478953356086462,
      "grad_norm": 1.4862582066571077,
      "learning_rate": 1.2499738393266684e-06,
      "loss": 0.0574,
      "step": 65
    },
    {
      "epoch": 0.015017064846416382,
      "grad_norm": 3.5354676602669506,
      "learning_rate": 1.2499730154246726e-06,
      "loss": 0.0742,
      "step": 66
    },
    {
      "epoch": 0.015244596131968146,
      "grad_norm": 4.702124240777191,
      "learning_rate": 1.2499721787493726e-06,
      "loss": 0.0994,
      "step": 67
    },
    {
      "epoch": 0.01547212741751991,
      "grad_norm": 2.2735257693436854,
      "learning_rate": 1.2499713293007862e-06,
      "loss": 0.0728,
      "step": 68
    },
    {
      "epoch": 0.015699658703071672,
      "grad_norm": 1.7727324277447247,
      "learning_rate": 1.2499704670789301e-06,
      "loss": 0.0492,
      "step": 69
    },
    {
      "epoch": 0.015927189988623434,
      "grad_norm": 2.0842450205223977,
      "learning_rate": 1.2499695920838225e-06,
      "loss": 0.0552,
      "step": 70
    },
    {
      "epoch": 0.0161547212741752,
      "grad_norm": 3.5228642040697116,
      "learning_rate": 1.2499687043154809e-06,
      "loss": 0.0556,
      "step": 71
    },
    {
      "epoch": 0.016382252559726963,
      "grad_norm": 2.797328413443058,
      "learning_rate": 1.2499678037739235e-06,
      "loss": 0.0519,
      "step": 72
    },
    {
      "epoch": 0.016609783845278725,
      "grad_norm": 3.349987636757497,
      "learning_rate": 1.2499668904591688e-06,
      "loss": 0.1053,
      "step": 73
    },
    {
      "epoch": 0.01683731513083049,
      "grad_norm": 4.71488578944105,
      "learning_rate": 1.2499659643712356e-06,
      "loss": 0.1072,
      "step": 74
    },
    {
      "epoch": 0.017064846416382253,
      "grad_norm": 1.9864905040446388,
      "learning_rate": 1.2499650255101425e-06,
      "loss": 0.1129,
      "step": 75
    },
    {
      "epoch": 0.017292377701934016,
      "grad_norm": 3.9786000792659073,
      "learning_rate": 1.2499640738759088e-06,
      "loss": 0.0798,
      "step": 76
    },
    {
      "epoch": 0.017519908987485778,
      "grad_norm": 3.293845214395531,
      "learning_rate": 1.249963109468554e-06,
      "loss": 0.1235,
      "step": 77
    },
    {
      "epoch": 0.017747440273037544,
      "grad_norm": 5.489256171335804,
      "learning_rate": 1.2499621322880979e-06,
      "loss": 0.0439,
      "step": 78
    },
    {
      "epoch": 0.017974971558589306,
      "grad_norm": 5.65488778928505,
      "learning_rate": 1.2499611423345604e-06,
      "loss": 0.0715,
      "step": 79
    },
    {
      "epoch": 0.01820250284414107,
      "grad_norm": 1.7907281324160922,
      "learning_rate": 1.2499601396079617e-06,
      "loss": 0.0668,
      "step": 80
    },
    {
      "epoch": 0.018430034129692834,
      "grad_norm": 2.5432248323972377,
      "learning_rate": 1.2499591241083222e-06,
      "loss": 0.0836,
      "step": 81
    },
    {
      "epoch": 0.018657565415244597,
      "grad_norm": 2.6549098221779146,
      "learning_rate": 1.2499580958356628e-06,
      "loss": 0.0612,
      "step": 82
    },
    {
      "epoch": 0.01888509670079636,
      "grad_norm": 3.6428276701171685,
      "learning_rate": 1.2499570547900045e-06,
      "loss": 0.0713,
      "step": 83
    },
    {
      "epoch": 0.01911262798634812,
      "grad_norm": 3.8990952294988994,
      "learning_rate": 1.2499560009713684e-06,
      "loss": 0.1046,
      "step": 84
    },
    {
      "epoch": 0.019340159271899887,
      "grad_norm": 1.7241928848576353,
      "learning_rate": 1.2499549343797764e-06,
      "loss": 0.0759,
      "step": 85
    },
    {
      "epoch": 0.01956769055745165,
      "grad_norm": 2.2613238963696545,
      "learning_rate": 1.24995385501525e-06,
      "loss": 0.0931,
      "step": 86
    },
    {
      "epoch": 0.019795221843003412,
      "grad_norm": 2.3167104475270492,
      "learning_rate": 1.2499527628778116e-06,
      "loss": 0.0775,
      "step": 87
    },
    {
      "epoch": 0.020022753128555178,
      "grad_norm": 2.631127335335558,
      "learning_rate": 1.2499516579674831e-06,
      "loss": 0.0911,
      "step": 88
    },
    {
      "epoch": 0.02025028441410694,
      "grad_norm": 3.902893778838773,
      "learning_rate": 1.2499505402842872e-06,
      "loss": 0.1129,
      "step": 89
    },
    {
      "epoch": 0.020477815699658702,
      "grad_norm": 2.6988246720898905,
      "learning_rate": 1.2499494098282469e-06,
      "loss": 0.088,
      "step": 90
    },
    {
      "epoch": 0.020705346985210465,
      "grad_norm": 1.4451215893923708,
      "learning_rate": 1.2499482665993851e-06,
      "loss": 0.0521,
      "step": 91
    },
    {
      "epoch": 0.02093287827076223,
      "grad_norm": 3.920423356576455,
      "learning_rate": 1.2499471105977252e-06,
      "loss": 0.079,
      "step": 92
    },
    {
      "epoch": 0.021160409556313993,
      "grad_norm": 3.1520828274033486,
      "learning_rate": 1.249945941823291e-06,
      "loss": 0.1206,
      "step": 93
    },
    {
      "epoch": 0.021387940841865755,
      "grad_norm": 6.767303206340345,
      "learning_rate": 1.2499447602761063e-06,
      "loss": 0.2231,
      "step": 94
    },
    {
      "epoch": 0.02161547212741752,
      "grad_norm": 8.857984803581953,
      "learning_rate": 1.2499435659561954e-06,
      "loss": 0.2288,
      "step": 95
    },
    {
      "epoch": 0.021843003412969283,
      "grad_norm": 3.2203370791462302,
      "learning_rate": 1.2499423588635823e-06,
      "loss": 0.1181,
      "step": 96
    },
    {
      "epoch": 0.022070534698521046,
      "grad_norm": 2.3890790390173864,
      "learning_rate": 1.2499411389982919e-06,
      "loss": 0.0546,
      "step": 97
    },
    {
      "epoch": 0.02229806598407281,
      "grad_norm": 3.9314083232645953,
      "learning_rate": 1.2499399063603492e-06,
      "loss": 0.1202,
      "step": 98
    },
    {
      "epoch": 0.022525597269624574,
      "grad_norm": 1.534409734668097,
      "learning_rate": 1.2499386609497793e-06,
      "loss": 0.0575,
      "step": 99
    },
    {
      "epoch": 0.022753128555176336,
      "grad_norm": 2.6915769569708163,
      "learning_rate": 1.2499374027666078e-06,
      "loss": 0.0865,
      "step": 100
    },
    {
      "epoch": 0.0229806598407281,
      "grad_norm": 1.9363626293552272,
      "learning_rate": 1.2499361318108602e-06,
      "loss": 0.0691,
      "step": 101
    },
    {
      "epoch": 0.023208191126279865,
      "grad_norm": 1.706755813278197,
      "learning_rate": 1.2499348480825627e-06,
      "loss": 0.0694,
      "step": 102
    },
    {
      "epoch": 0.023435722411831627,
      "grad_norm": 2.4058707201091902,
      "learning_rate": 1.2499335515817413e-06,
      "loss": 0.0873,
      "step": 103
    },
    {
      "epoch": 0.02366325369738339,
      "grad_norm": 1.4799661409619025,
      "learning_rate": 1.2499322423084226e-06,
      "loss": 0.0489,
      "step": 104
    },
    {
      "epoch": 0.023890784982935155,
      "grad_norm": 1.5529996293769865,
      "learning_rate": 1.2499309202626336e-06,
      "loss": 0.0489,
      "step": 105
    },
    {
      "epoch": 0.024118316268486917,
      "grad_norm": 1.2094533521149893,
      "learning_rate": 1.249929585444401e-06,
      "loss": 0.0555,
      "step": 106
    },
    {
      "epoch": 0.02434584755403868,
      "grad_norm": 3.3515777105454805,
      "learning_rate": 1.2499282378537522e-06,
      "loss": 0.0869,
      "step": 107
    },
    {
      "epoch": 0.024573378839590442,
      "grad_norm": 3.049734475824735,
      "learning_rate": 1.2499268774907144e-06,
      "loss": 0.0436,
      "step": 108
    },
    {
      "epoch": 0.024800910125142208,
      "grad_norm": 1.466216427195955,
      "learning_rate": 1.249925504355316e-06,
      "loss": 0.0602,
      "step": 109
    },
    {
      "epoch": 0.02502844141069397,
      "grad_norm": 1.7889692240619717,
      "learning_rate": 1.2499241184475848e-06,
      "loss": 0.0485,
      "step": 110
    },
    {
      "epoch": 0.025255972696245733,
      "grad_norm": 1.601570429523166,
      "learning_rate": 1.249922719767549e-06,
      "loss": 0.0657,
      "step": 111
    },
    {
      "epoch": 0.0254835039817975,
      "grad_norm": 1.9895197648670602,
      "learning_rate": 1.2499213083152374e-06,
      "loss": 0.0613,
      "step": 112
    },
    {
      "epoch": 0.02571103526734926,
      "grad_norm": 1.559725191147961,
      "learning_rate": 1.2499198840906787e-06,
      "loss": 0.0376,
      "step": 113
    },
    {
      "epoch": 0.025938566552901023,
      "grad_norm": 1.8823702931097668,
      "learning_rate": 1.249918447093902e-06,
      "loss": 0.0441,
      "step": 114
    },
    {
      "epoch": 0.026166097838452786,
      "grad_norm": 2.8320457267761707,
      "learning_rate": 1.249916997324937e-06,
      "loss": 0.1287,
      "step": 115
    },
    {
      "epoch": 0.02639362912400455,
      "grad_norm": 4.382916227720041,
      "learning_rate": 1.2499155347838129e-06,
      "loss": 0.0828,
      "step": 116
    },
    {
      "epoch": 0.026621160409556314,
      "grad_norm": 3.200832910369845,
      "learning_rate": 1.2499140594705596e-06,
      "loss": 0.0621,
      "step": 117
    },
    {
      "epoch": 0.026848691695108076,
      "grad_norm": 1.6914803041014228,
      "learning_rate": 1.2499125713852076e-06,
      "loss": 0.0778,
      "step": 118
    },
    {
      "epoch": 0.027076222980659842,
      "grad_norm": 1.192120885936281,
      "learning_rate": 1.2499110705277869e-06,
      "loss": 0.0505,
      "step": 119
    },
    {
      "epoch": 0.027303754266211604,
      "grad_norm": 2.0573836463025716,
      "learning_rate": 1.2499095568983284e-06,
      "loss": 0.0802,
      "step": 120
    },
    {
      "epoch": 0.027531285551763367,
      "grad_norm": 1.4902545114620074,
      "learning_rate": 1.2499080304968634e-06,
      "loss": 0.0565,
      "step": 121
    },
    {
      "epoch": 0.027758816837315133,
      "grad_norm": 5.341407524309934,
      "learning_rate": 1.2499064913234222e-06,
      "loss": 0.0556,
      "step": 122
    },
    {
      "epoch": 0.027986348122866895,
      "grad_norm": 1.9959451584520156,
      "learning_rate": 1.249904939378037e-06,
      "loss": 0.0425,
      "step": 123
    },
    {
      "epoch": 0.028213879408418657,
      "grad_norm": 4.8411742944908225,
      "learning_rate": 1.2499033746607395e-06,
      "loss": 0.1189,
      "step": 124
    },
    {
      "epoch": 0.02844141069397042,
      "grad_norm": 2.6968922160166624,
      "learning_rate": 1.2499017971715614e-06,
      "loss": 0.0952,
      "step": 125
    },
    {
      "epoch": 0.028668941979522185,
      "grad_norm": 2.498689273522412,
      "learning_rate": 1.2499002069105348e-06,
      "loss": 0.0609,
      "step": 126
    },
    {
      "epoch": 0.028896473265073948,
      "grad_norm": 1.7926923107505222,
      "learning_rate": 1.2498986038776926e-06,
      "loss": 0.0674,
      "step": 127
    },
    {
      "epoch": 0.02912400455062571,
      "grad_norm": 2.951592426546495,
      "learning_rate": 1.2498969880730671e-06,
      "loss": 0.0581,
      "step": 128
    },
    {
      "epoch": 0.029351535836177476,
      "grad_norm": 2.546012031191214,
      "learning_rate": 1.249895359496692e-06,
      "loss": 0.0604,
      "step": 129
    },
    {
      "epoch": 0.02957906712172924,
      "grad_norm": 2.982654740949648,
      "learning_rate": 1.2498937181486e-06,
      "loss": 0.1317,
      "step": 130
    },
    {
      "epoch": 0.029806598407281,
      "grad_norm": 3.35692876580473,
      "learning_rate": 1.2498920640288248e-06,
      "loss": 0.1357,
      "step": 131
    },
    {
      "epoch": 0.030034129692832763,
      "grad_norm": 3.0581666005631694,
      "learning_rate": 1.2498903971374005e-06,
      "loss": 0.1404,
      "step": 132
    },
    {
      "epoch": 0.03026166097838453,
      "grad_norm": 2.8194208797307936,
      "learning_rate": 1.2498887174743606e-06,
      "loss": 0.1139,
      "step": 133
    },
    {
      "epoch": 0.03048919226393629,
      "grad_norm": 1.307812448511231,
      "learning_rate": 1.24988702503974e-06,
      "loss": 0.0439,
      "step": 134
    },
    {
      "epoch": 0.030716723549488054,
      "grad_norm": 2.1828107869380178,
      "learning_rate": 1.2498853198335728e-06,
      "loss": 0.0714,
      "step": 135
    },
    {
      "epoch": 0.03094425483503982,
      "grad_norm": 2.7687164066652588,
      "learning_rate": 1.2498836018558942e-06,
      "loss": 0.0588,
      "step": 136
    },
    {
      "epoch": 0.031171786120591582,
      "grad_norm": 2.0398223571287124,
      "learning_rate": 1.2498818711067392e-06,
      "loss": 0.0788,
      "step": 137
    },
    {
      "epoch": 0.031399317406143344,
      "grad_norm": 1.544887443922682,
      "learning_rate": 1.2498801275861433e-06,
      "loss": 0.0424,
      "step": 138
    },
    {
      "epoch": 0.03162684869169511,
      "grad_norm": 1.9943988491119002,
      "learning_rate": 1.2498783712941418e-06,
      "loss": 0.0509,
      "step": 139
    },
    {
      "epoch": 0.03185437997724687,
      "grad_norm": 2.2756991800533726,
      "learning_rate": 1.2498766022307709e-06,
      "loss": 0.077,
      "step": 140
    },
    {
      "epoch": 0.032081911262798635,
      "grad_norm": 2.111047377421262,
      "learning_rate": 1.2498748203960665e-06,
      "loss": 0.0698,
      "step": 141
    },
    {
      "epoch": 0.0323094425483504,
      "grad_norm": 5.455732324109059,
      "learning_rate": 1.2498730257900655e-06,
      "loss": 0.0464,
      "step": 142
    },
    {
      "epoch": 0.03253697383390216,
      "grad_norm": 5.8400695805570475,
      "learning_rate": 1.249871218412804e-06,
      "loss": 0.0808,
      "step": 143
    },
    {
      "epoch": 0.032764505119453925,
      "grad_norm": 1.6154990753356673,
      "learning_rate": 1.2498693982643192e-06,
      "loss": 0.0579,
      "step": 144
    },
    {
      "epoch": 0.03299203640500569,
      "grad_norm": 2.6179374503343644,
      "learning_rate": 1.2498675653446485e-06,
      "loss": 0.0539,
      "step": 145
    },
    {
      "epoch": 0.03321956769055745,
      "grad_norm": 1.8802237147910983,
      "learning_rate": 1.249865719653829e-06,
      "loss": 0.0562,
      "step": 146
    },
    {
      "epoch": 0.033447098976109216,
      "grad_norm": 1.9601893311780516,
      "learning_rate": 1.2498638611918985e-06,
      "loss": 0.0842,
      "step": 147
    },
    {
      "epoch": 0.03367463026166098,
      "grad_norm": 3.2621758627603814,
      "learning_rate": 1.249861989958895e-06,
      "loss": 0.0707,
      "step": 148
    },
    {
      "epoch": 0.03390216154721274,
      "grad_norm": 1.7359031480428844,
      "learning_rate": 1.2498601059548572e-06,
      "loss": 0.0552,
      "step": 149
    },
    {
      "epoch": 0.034129692832764506,
      "grad_norm": 1.9031742914383765,
      "learning_rate": 1.2498582091798228e-06,
      "loss": 0.0551,
      "step": 150
    },
    {
      "epoch": 0.034357224118316265,
      "grad_norm": 1.6839469596143435,
      "learning_rate": 1.2498562996338312e-06,
      "loss": 0.0549,
      "step": 151
    },
    {
      "epoch": 0.03458475540386803,
      "grad_norm": 4.770073312573169,
      "learning_rate": 1.249854377316921e-06,
      "loss": 0.0922,
      "step": 152
    },
    {
      "epoch": 0.0348122866894198,
      "grad_norm": 9.77865071194089,
      "learning_rate": 1.2498524422291319e-06,
      "loss": 0.0656,
      "step": 153
    },
    {
      "epoch": 0.035039817974971556,
      "grad_norm": 3.279720923464348,
      "learning_rate": 1.2498504943705033e-06,
      "loss": 0.0628,
      "step": 154
    },
    {
      "epoch": 0.03526734926052332,
      "grad_norm": 2.5593319197975086,
      "learning_rate": 1.249848533741075e-06,
      "loss": 0.1116,
      "step": 155
    },
    {
      "epoch": 0.03549488054607509,
      "grad_norm": 2.7020759561258796,
      "learning_rate": 1.2498465603408865e-06,
      "loss": 0.0818,
      "step": 156
    },
    {
      "epoch": 0.035722411831626846,
      "grad_norm": 1.1827259380181383,
      "learning_rate": 1.2498445741699792e-06,
      "loss": 0.0412,
      "step": 157
    },
    {
      "epoch": 0.03594994311717861,
      "grad_norm": 3.090835953132798,
      "learning_rate": 1.249842575228393e-06,
      "loss": 0.0922,
      "step": 158
    },
    {
      "epoch": 0.03617747440273038,
      "grad_norm": 4.286688880757478,
      "learning_rate": 1.249840563516169e-06,
      "loss": 0.0921,
      "step": 159
    },
    {
      "epoch": 0.03640500568828214,
      "grad_norm": 3.826111531217773,
      "learning_rate": 1.249838539033348e-06,
      "loss": 0.0711,
      "step": 160
    },
    {
      "epoch": 0.0366325369738339,
      "grad_norm": 1.286547340104366,
      "learning_rate": 1.2498365017799715e-06,
      "loss": 0.0363,
      "step": 161
    },
    {
      "epoch": 0.03686006825938567,
      "grad_norm": 0.9115912875215112,
      "learning_rate": 1.2498344517560815e-06,
      "loss": 0.0499,
      "step": 162
    },
    {
      "epoch": 0.03708759954493743,
      "grad_norm": 1.967900659157127,
      "learning_rate": 1.2498323889617198e-06,
      "loss": 0.0642,
      "step": 163
    },
    {
      "epoch": 0.03731513083048919,
      "grad_norm": 20.150143075771535,
      "learning_rate": 1.2498303133969281e-06,
      "loss": 0.0764,
      "step": 164
    },
    {
      "epoch": 0.03754266211604096,
      "grad_norm": 1.8715348615633682,
      "learning_rate": 1.2498282250617492e-06,
      "loss": 0.0756,
      "step": 165
    },
    {
      "epoch": 0.03777019340159272,
      "grad_norm": 1.4158902127506805,
      "learning_rate": 1.2498261239562257e-06,
      "loss": 0.0669,
      "step": 166
    },
    {
      "epoch": 0.037997724687144484,
      "grad_norm": 2.5083436605654583,
      "learning_rate": 1.2498240100804005e-06,
      "loss": 0.0642,
      "step": 167
    },
    {
      "epoch": 0.03822525597269624,
      "grad_norm": 4.201645953801033,
      "learning_rate": 1.249821883434317e-06,
      "loss": 0.0804,
      "step": 168
    },
    {
      "epoch": 0.03845278725824801,
      "grad_norm": 1.8056911269518585,
      "learning_rate": 1.2498197440180182e-06,
      "loss": 0.0826,
      "step": 169
    },
    {
      "epoch": 0.038680318543799774,
      "grad_norm": 2.5577509171659174,
      "learning_rate": 1.2498175918315484e-06,
      "loss": 0.0782,
      "step": 170
    },
    {
      "epoch": 0.03890784982935153,
      "grad_norm": 1.6347218920613042,
      "learning_rate": 1.2498154268749513e-06,
      "loss": 0.0526,
      "step": 171
    },
    {
      "epoch": 0.0391353811149033,
      "grad_norm": 2.089011530912283,
      "learning_rate": 1.249813249148271e-06,
      "loss": 0.0845,
      "step": 172
    },
    {
      "epoch": 0.039362912400455065,
      "grad_norm": 2.0840328778691557,
      "learning_rate": 1.2498110586515525e-06,
      "loss": 0.0642,
      "step": 173
    },
    {
      "epoch": 0.039590443686006824,
      "grad_norm": 2.63166064853352,
      "learning_rate": 1.2498088553848398e-06,
      "loss": 0.0896,
      "step": 174
    },
    {
      "epoch": 0.03981797497155859,
      "grad_norm": 1.7790239100772818,
      "learning_rate": 1.2498066393481787e-06,
      "loss": 0.0887,
      "step": 175
    },
    {
      "epoch": 0.040045506257110355,
      "grad_norm": 2.0484275395799374,
      "learning_rate": 1.249804410541614e-06,
      "loss": 0.0757,
      "step": 176
    },
    {
      "epoch": 0.040273037542662114,
      "grad_norm": 2.7339764522222674,
      "learning_rate": 1.2498021689651916e-06,
      "loss": 0.0752,
      "step": 177
    },
    {
      "epoch": 0.04050056882821388,
      "grad_norm": 1.7202036539272256,
      "learning_rate": 1.249799914618957e-06,
      "loss": 0.0571,
      "step": 178
    },
    {
      "epoch": 0.040728100113765646,
      "grad_norm": 0.9595120325924671,
      "learning_rate": 1.2497976475029566e-06,
      "loss": 0.0431,
      "step": 179
    },
    {
      "epoch": 0.040955631399317405,
      "grad_norm": 2.2881459955255283,
      "learning_rate": 1.2497953676172364e-06,
      "loss": 0.0714,
      "step": 180
    },
    {
      "epoch": 0.04118316268486917,
      "grad_norm": 1.1818626309445535,
      "learning_rate": 1.2497930749618431e-06,
      "loss": 0.0352,
      "step": 181
    },
    {
      "epoch": 0.04141069397042093,
      "grad_norm": 1.7233391740704784,
      "learning_rate": 1.2497907695368238e-06,
      "loss": 0.0469,
      "step": 182
    },
    {
      "epoch": 0.041638225255972695,
      "grad_norm": 1.260961347917145,
      "learning_rate": 1.2497884513422253e-06,
      "loss": 0.0545,
      "step": 183
    },
    {
      "epoch": 0.04186575654152446,
      "grad_norm": 1.350181559062471,
      "learning_rate": 1.249786120378095e-06,
      "loss": 0.0579,
      "step": 184
    },
    {
      "epoch": 0.04209328782707622,
      "grad_norm": 1.4312208218600422,
      "learning_rate": 1.2497837766444806e-06,
      "loss": 0.0671,
      "step": 185
    },
    {
      "epoch": 0.042320819112627986,
      "grad_norm": 3.0862470606278527,
      "learning_rate": 1.2497814201414304e-06,
      "loss": 0.1552,
      "step": 186
    },
    {
      "epoch": 0.04254835039817975,
      "grad_norm": 2.010075620155332,
      "learning_rate": 1.249779050868992e-06,
      "loss": 0.0819,
      "step": 187
    },
    {
      "epoch": 0.04277588168373151,
      "grad_norm": 1.5041508185690229,
      "learning_rate": 1.249776668827214e-06,
      "loss": 0.0555,
      "step": 188
    },
    {
      "epoch": 0.043003412969283276,
      "grad_norm": 15.160154885670842,
      "learning_rate": 1.249774274016145e-06,
      "loss": 0.1377,
      "step": 189
    },
    {
      "epoch": 0.04323094425483504,
      "grad_norm": 1.3900253472134594,
      "learning_rate": 1.2497718664358341e-06,
      "loss": 0.0573,
      "step": 190
    },
    {
      "epoch": 0.0434584755403868,
      "grad_norm": 1.5383922685465743,
      "learning_rate": 1.2497694460863307e-06,
      "loss": 0.0986,
      "step": 191
    },
    {
      "epoch": 0.04368600682593857,
      "grad_norm": 2.2430359603043266,
      "learning_rate": 1.2497670129676838e-06,
      "loss": 0.0771,
      "step": 192
    },
    {
      "epoch": 0.04391353811149033,
      "grad_norm": 2.1439327799346333,
      "learning_rate": 1.2497645670799436e-06,
      "loss": 0.0436,
      "step": 193
    },
    {
      "epoch": 0.04414106939704209,
      "grad_norm": 1.771166715500092,
      "learning_rate": 1.2497621084231595e-06,
      "loss": 0.0523,
      "step": 194
    },
    {
      "epoch": 0.04436860068259386,
      "grad_norm": 1.4299897341433283,
      "learning_rate": 1.2497596369973823e-06,
      "loss": 0.0401,
      "step": 195
    },
    {
      "epoch": 0.04459613196814562,
      "grad_norm": 2.3614356932574654,
      "learning_rate": 1.2497571528026623e-06,
      "loss": 0.0888,
      "step": 196
    },
    {
      "epoch": 0.04482366325369738,
      "grad_norm": 1.8304520866772858,
      "learning_rate": 1.2497546558390503e-06,
      "loss": 0.065,
      "step": 197
    },
    {
      "epoch": 0.04505119453924915,
      "grad_norm": 1.8912148609925916,
      "learning_rate": 1.2497521461065973e-06,
      "loss": 0.0601,
      "step": 198
    },
    {
      "epoch": 0.04527872582480091,
      "grad_norm": 3.522073583380884,
      "learning_rate": 1.2497496236053547e-06,
      "loss": 0.0841,
      "step": 199
    },
    {
      "epoch": 0.04550625711035267,
      "grad_norm": 1.3143196144497622,
      "learning_rate": 1.2497470883353738e-06,
      "loss": 0.0485,
      "step": 200
    },
    {
      "epoch": 0.04573378839590444,
      "grad_norm": 2.251264141887164,
      "learning_rate": 1.2497445402967068e-06,
      "loss": 0.1068,
      "step": 201
    },
    {
      "epoch": 0.0459613196814562,
      "grad_norm": 1.5072412944239943,
      "learning_rate": 1.2497419794894053e-06,
      "loss": 0.0685,
      "step": 202
    },
    {
      "epoch": 0.04618885096700796,
      "grad_norm": 3.468383203818734,
      "learning_rate": 1.249739405913522e-06,
      "loss": 0.0815,
      "step": 203
    },
    {
      "epoch": 0.04641638225255973,
      "grad_norm": 2.799795848526847,
      "learning_rate": 1.2497368195691095e-06,
      "loss": 0.0611,
      "step": 204
    },
    {
      "epoch": 0.04664391353811149,
      "grad_norm": 1.8317350709529971,
      "learning_rate": 1.2497342204562205e-06,
      "loss": 0.0645,
      "step": 205
    },
    {
      "epoch": 0.046871444823663254,
      "grad_norm": 1.8552065397519177,
      "learning_rate": 1.2497316085749081e-06,
      "loss": 0.0475,
      "step": 206
    },
    {
      "epoch": 0.04709897610921502,
      "grad_norm": 33.648509264650215,
      "learning_rate": 1.249728983925226e-06,
      "loss": 0.4463,
      "step": 207
    },
    {
      "epoch": 0.04732650739476678,
      "grad_norm": 3.6371540108131555,
      "learning_rate": 1.2497263465072274e-06,
      "loss": 0.1261,
      "step": 208
    },
    {
      "epoch": 0.047554038680318544,
      "grad_norm": 4.579077765212189,
      "learning_rate": 1.2497236963209663e-06,
      "loss": 0.1537,
      "step": 209
    },
    {
      "epoch": 0.04778156996587031,
      "grad_norm": 1.8349256873988717,
      "learning_rate": 1.2497210333664972e-06,
      "loss": 0.0905,
      "step": 210
    },
    {
      "epoch": 0.04800910125142207,
      "grad_norm": 1.0349470677126553,
      "learning_rate": 1.2497183576438743e-06,
      "loss": 0.0383,
      "step": 211
    },
    {
      "epoch": 0.048236632536973835,
      "grad_norm": 1.7552990958997892,
      "learning_rate": 1.2497156691531523e-06,
      "loss": 0.0667,
      "step": 212
    },
    {
      "epoch": 0.048464163822525594,
      "grad_norm": 1.1378619448565346,
      "learning_rate": 1.249712967894386e-06,
      "loss": 0.0494,
      "step": 213
    },
    {
      "epoch": 0.04869169510807736,
      "grad_norm": 2.098843760453871,
      "learning_rate": 1.2497102538676308e-06,
      "loss": 0.0683,
      "step": 214
    },
    {
      "epoch": 0.048919226393629126,
      "grad_norm": 2.4023627244072996,
      "learning_rate": 1.249707527072942e-06,
      "loss": 0.0752,
      "step": 215
    },
    {
      "epoch": 0.049146757679180884,
      "grad_norm": 3.0741883094346405,
      "learning_rate": 1.2497047875103757e-06,
      "loss": 0.1576,
      "step": 216
    },
    {
      "epoch": 0.04937428896473265,
      "grad_norm": 2.109066858277886,
      "learning_rate": 1.2497020351799875e-06,
      "loss": 0.0834,
      "step": 217
    },
    {
      "epoch": 0.049601820250284416,
      "grad_norm": 1.1860658004774445,
      "learning_rate": 1.2496992700818335e-06,
      "loss": 0.0487,
      "step": 218
    },
    {
      "epoch": 0.049829351535836175,
      "grad_norm": 2.78284096499592,
      "learning_rate": 1.249696492215971e-06,
      "loss": 0.0743,
      "step": 219
    },
    {
      "epoch": 0.05005688282138794,
      "grad_norm": 2.111446571336187,
      "learning_rate": 1.249693701582456e-06,
      "loss": 0.0381,
      "step": 220
    },
    {
      "epoch": 0.05028441410693971,
      "grad_norm": 1.8439980435313363,
      "learning_rate": 1.2496908981813458e-06,
      "loss": 0.0821,
      "step": 221
    },
    {
      "epoch": 0.050511945392491465,
      "grad_norm": 2.1292431200987165,
      "learning_rate": 1.2496880820126977e-06,
      "loss": 0.102,
      "step": 222
    },
    {
      "epoch": 0.05073947667804323,
      "grad_norm": 1.58572513687618,
      "learning_rate": 1.2496852530765695e-06,
      "loss": 0.0451,
      "step": 223
    },
    {
      "epoch": 0.050967007963595,
      "grad_norm": 3.028551047012946,
      "learning_rate": 1.2496824113730186e-06,
      "loss": 0.1259,
      "step": 224
    },
    {
      "epoch": 0.051194539249146756,
      "grad_norm": 1.2656190346220688,
      "learning_rate": 1.2496795569021033e-06,
      "loss": 0.0593,
      "step": 225
    },
    {
      "epoch": 0.05142207053469852,
      "grad_norm": 2.5365255203279444,
      "learning_rate": 1.2496766896638819e-06,
      "loss": 0.0754,
      "step": 226
    },
    {
      "epoch": 0.05164960182025029,
      "grad_norm": 2.246271311279298,
      "learning_rate": 1.249673809658413e-06,
      "loss": 0.0689,
      "step": 227
    },
    {
      "epoch": 0.05187713310580205,
      "grad_norm": 1.212864625758916,
      "learning_rate": 1.2496709168857555e-06,
      "loss": 0.0451,
      "step": 228
    },
    {
      "epoch": 0.05210466439135381,
      "grad_norm": 2.806330657303488,
      "learning_rate": 1.2496680113459683e-06,
      "loss": 0.1473,
      "step": 229
    },
    {
      "epoch": 0.05233219567690557,
      "grad_norm": 2.652722899111948,
      "learning_rate": 1.2496650930391113e-06,
      "loss": 0.1155,
      "step": 230
    },
    {
      "epoch": 0.05255972696245734,
      "grad_norm": 2.7863683658107696,
      "learning_rate": 1.2496621619652435e-06,
      "loss": 0.0939,
      "step": 231
    },
    {
      "epoch": 0.0527872582480091,
      "grad_norm": 2.0852803666211925,
      "learning_rate": 1.2496592181244253e-06,
      "loss": 0.0385,
      "step": 232
    },
    {
      "epoch": 0.05301478953356086,
      "grad_norm": 2.3109089889052274,
      "learning_rate": 1.249656261516717e-06,
      "loss": 0.0591,
      "step": 233
    },
    {
      "epoch": 0.05324232081911263,
      "grad_norm": 1.657658325406227,
      "learning_rate": 1.2496532921421781e-06,
      "loss": 0.0666,
      "step": 234
    },
    {
      "epoch": 0.053469852104664393,
      "grad_norm": 1.8341397873796956,
      "learning_rate": 1.2496503100008704e-06,
      "loss": 0.0893,
      "step": 235
    },
    {
      "epoch": 0.05369738339021615,
      "grad_norm": 1.3722825156667726,
      "learning_rate": 1.249647315092854e-06,
      "loss": 0.0445,
      "step": 236
    },
    {
      "epoch": 0.05392491467576792,
      "grad_norm": 2.4225397650308187,
      "learning_rate": 1.2496443074181905e-06,
      "loss": 0.0783,
      "step": 237
    },
    {
      "epoch": 0.054152445961319684,
      "grad_norm": 1.27450689566742,
      "learning_rate": 1.2496412869769415e-06,
      "loss": 0.0482,
      "step": 238
    },
    {
      "epoch": 0.05437997724687144,
      "grad_norm": 1.5788523588867425,
      "learning_rate": 1.2496382537691686e-06,
      "loss": 0.0559,
      "step": 239
    },
    {
      "epoch": 0.05460750853242321,
      "grad_norm": 2.455674592198026,
      "learning_rate": 1.2496352077949336e-06,
      "loss": 0.0686,
      "step": 240
    },
    {
      "epoch": 0.054835039817974975,
      "grad_norm": 0.9822051281919313,
      "learning_rate": 1.249632149054299e-06,
      "loss": 0.0333,
      "step": 241
    },
    {
      "epoch": 0.05506257110352673,
      "grad_norm": 1.4918328919143296,
      "learning_rate": 1.249629077547327e-06,
      "loss": 0.0564,
      "step": 242
    },
    {
      "epoch": 0.0552901023890785,
      "grad_norm": 2.2431233119775973,
      "learning_rate": 1.2496259932740813e-06,
      "loss": 0.0998,
      "step": 243
    },
    {
      "epoch": 0.055517633674630265,
      "grad_norm": 1.0547810468664673,
      "learning_rate": 1.2496228962346236e-06,
      "loss": 0.054,
      "step": 244
    },
    {
      "epoch": 0.055745164960182024,
      "grad_norm": 4.2608959679243785,
      "learning_rate": 1.249619786429018e-06,
      "loss": 0.1324,
      "step": 245
    },
    {
      "epoch": 0.05597269624573379,
      "grad_norm": 0.9597443879271953,
      "learning_rate": 1.2496166638573278e-06,
      "loss": 0.0545,
      "step": 246
    },
    {
      "epoch": 0.05620022753128555,
      "grad_norm": 1.0564009660406621,
      "learning_rate": 1.2496135285196172e-06,
      "loss": 0.0409,
      "step": 247
    },
    {
      "epoch": 0.056427758816837315,
      "grad_norm": 5.181656955922701,
      "learning_rate": 1.2496103804159497e-06,
      "loss": 0.0514,
      "step": 248
    },
    {
      "epoch": 0.05665529010238908,
      "grad_norm": 31.956127145863782,
      "learning_rate": 1.2496072195463904e-06,
      "loss": 0.5089,
      "step": 249
    },
    {
      "epoch": 0.05688282138794084,
      "grad_norm": 1.6118693718642072,
      "learning_rate": 1.249604045911003e-06,
      "loss": 0.0601,
      "step": 250
    },
    {
      "epoch": 0.057110352673492605,
      "grad_norm": 1.8753798893546312,
      "learning_rate": 1.249600859509853e-06,
      "loss": 0.0781,
      "step": 251
    },
    {
      "epoch": 0.05733788395904437,
      "grad_norm": 1.9173456172296954,
      "learning_rate": 1.2495976603430054e-06,
      "loss": 0.0998,
      "step": 252
    },
    {
      "epoch": 0.05756541524459613,
      "grad_norm": 1.7446295444585516,
      "learning_rate": 1.2495944484105254e-06,
      "loss": 0.0348,
      "step": 253
    },
    {
      "epoch": 0.057792946530147896,
      "grad_norm": 3.1221291176703816,
      "learning_rate": 1.2495912237124787e-06,
      "loss": 0.0467,
      "step": 254
    },
    {
      "epoch": 0.05802047781569966,
      "grad_norm": 3.8018086885799454,
      "learning_rate": 1.2495879862489312e-06,
      "loss": 0.0893,
      "step": 255
    },
    {
      "epoch": 0.05824800910125142,
      "grad_norm": 7.114525761671455,
      "learning_rate": 1.2495847360199495e-06,
      "loss": 0.0647,
      "step": 256
    },
    {
      "epoch": 0.058475540386803186,
      "grad_norm": 1.861881748798064,
      "learning_rate": 1.2495814730255993e-06,
      "loss": 0.0832,
      "step": 257
    },
    {
      "epoch": 0.05870307167235495,
      "grad_norm": 1.920791237896296,
      "learning_rate": 1.2495781972659479e-06,
      "loss": 0.0346,
      "step": 258
    },
    {
      "epoch": 0.05893060295790671,
      "grad_norm": 1.9123959618493658,
      "learning_rate": 1.2495749087410618e-06,
      "loss": 0.0924,
      "step": 259
    },
    {
      "epoch": 0.05915813424345848,
      "grad_norm": 4.29696428710133,
      "learning_rate": 1.2495716074510087e-06,
      "loss": 0.0503,
      "step": 260
    },
    {
      "epoch": 0.059385665529010236,
      "grad_norm": 1.518204196410266,
      "learning_rate": 1.2495682933958555e-06,
      "loss": 0.0516,
      "step": 261
    },
    {
      "epoch": 0.059613196814562,
      "grad_norm": 4.632688881661792,
      "learning_rate": 1.2495649665756705e-06,
      "loss": 0.1211,
      "step": 262
    },
    {
      "epoch": 0.05984072810011377,
      "grad_norm": 2.486723056340893,
      "learning_rate": 1.2495616269905212e-06,
      "loss": 0.0811,
      "step": 263
    },
    {
      "epoch": 0.060068259385665526,
      "grad_norm": 1.581969386319132,
      "learning_rate": 1.2495582746404762e-06,
      "loss": 0.0589,
      "step": 264
    },
    {
      "epoch": 0.06029579067121729,
      "grad_norm": 1.5286635653264549,
      "learning_rate": 1.249554909525604e-06,
      "loss": 0.0638,
      "step": 265
    },
    {
      "epoch": 0.06052332195676906,
      "grad_norm": 1.2992103388623448,
      "learning_rate": 1.249551531645973e-06,
      "loss": 0.0309,
      "step": 266
    },
    {
      "epoch": 0.06075085324232082,
      "grad_norm": 1.4096677819832404,
      "learning_rate": 1.2495481410016527e-06,
      "loss": 0.0779,
      "step": 267
    },
    {
      "epoch": 0.06097838452787258,
      "grad_norm": 2.5850002425545586,
      "learning_rate": 1.2495447375927122e-06,
      "loss": 0.0718,
      "step": 268
    },
    {
      "epoch": 0.06120591581342435,
      "grad_norm": 2.2171422082649155,
      "learning_rate": 1.2495413214192209e-06,
      "loss": 0.0761,
      "step": 269
    },
    {
      "epoch": 0.06143344709897611,
      "grad_norm": 1.8061938933671926,
      "learning_rate": 1.2495378924812486e-06,
      "loss": 0.068,
      "step": 270
    },
    {
      "epoch": 0.06166097838452787,
      "grad_norm": 1.805390591341637,
      "learning_rate": 1.2495344507788662e-06,
      "loss": 0.0589,
      "step": 271
    },
    {
      "epoch": 0.06188850967007964,
      "grad_norm": 28.567737212004268,
      "learning_rate": 1.249530996312143e-06,
      "loss": 0.2672,
      "step": 272
    },
    {
      "epoch": 0.0621160409556314,
      "grad_norm": 2.023457290937467,
      "learning_rate": 1.2495275290811499e-06,
      "loss": 0.0762,
      "step": 273
    },
    {
      "epoch": 0.062343572241183164,
      "grad_norm": 1.726293880974768,
      "learning_rate": 1.2495240490859581e-06,
      "loss": 0.1124,
      "step": 274
    },
    {
      "epoch": 0.06257110352673492,
      "grad_norm": 3.5654068745258094,
      "learning_rate": 1.2495205563266384e-06,
      "loss": 0.08,
      "step": 275
    },
    {
      "epoch": 0.06279863481228669,
      "grad_norm": 3.330534204865081,
      "learning_rate": 1.2495170508032624e-06,
      "loss": 0.1063,
      "step": 276
    },
    {
      "epoch": 0.06302616609783845,
      "grad_norm": 2.7347891521994265,
      "learning_rate": 1.2495135325159015e-06,
      "loss": 0.0544,
      "step": 277
    },
    {
      "epoch": 0.06325369738339022,
      "grad_norm": 1.6191657189271829,
      "learning_rate": 1.2495100014646277e-06,
      "loss": 0.0399,
      "step": 278
    },
    {
      "epoch": 0.06348122866894199,
      "grad_norm": 2.2760450280718163,
      "learning_rate": 1.2495064576495134e-06,
      "loss": 0.0842,
      "step": 279
    },
    {
      "epoch": 0.06370875995449374,
      "grad_norm": 1.26417795343513,
      "learning_rate": 1.2495029010706306e-06,
      "loss": 0.0396,
      "step": 280
    },
    {
      "epoch": 0.0639362912400455,
      "grad_norm": 2.452124440380249,
      "learning_rate": 1.2494993317280524e-06,
      "loss": 0.057,
      "step": 281
    },
    {
      "epoch": 0.06416382252559727,
      "grad_norm": 1.6768179224480646,
      "learning_rate": 1.2494957496218516e-06,
      "loss": 0.0778,
      "step": 282
    },
    {
      "epoch": 0.06439135381114904,
      "grad_norm": 1.148511080426964,
      "learning_rate": 1.2494921547521013e-06,
      "loss": 0.0593,
      "step": 283
    },
    {
      "epoch": 0.0646188850967008,
      "grad_norm": 2.6954847712989687,
      "learning_rate": 1.249488547118875e-06,
      "loss": 0.0581,
      "step": 284
    },
    {
      "epoch": 0.06484641638225255,
      "grad_norm": 1.7038701082720327,
      "learning_rate": 1.2494849267222466e-06,
      "loss": 0.0467,
      "step": 285
    },
    {
      "epoch": 0.06507394766780432,
      "grad_norm": 2.0205984871282654,
      "learning_rate": 1.24948129356229e-06,
      "loss": 0.08,
      "step": 286
    },
    {
      "epoch": 0.06530147895335608,
      "grad_norm": 2.4430734451001093,
      "learning_rate": 1.2494776476390793e-06,
      "loss": 0.0689,
      "step": 287
    },
    {
      "epoch": 0.06552901023890785,
      "grad_norm": 2.2074548221975223,
      "learning_rate": 1.2494739889526894e-06,
      "loss": 0.036,
      "step": 288
    },
    {
      "epoch": 0.06575654152445962,
      "grad_norm": 2.3839227269087115,
      "learning_rate": 1.2494703175031946e-06,
      "loss": 0.1017,
      "step": 289
    },
    {
      "epoch": 0.06598407281001138,
      "grad_norm": 1.297764082661735,
      "learning_rate": 1.2494666332906702e-06,
      "loss": 0.0428,
      "step": 290
    },
    {
      "epoch": 0.06621160409556313,
      "grad_norm": 2.29700118159146,
      "learning_rate": 1.2494629363151916e-06,
      "loss": 0.0564,
      "step": 291
    },
    {
      "epoch": 0.0664391353811149,
      "grad_norm": 10.343219429302799,
      "learning_rate": 1.2494592265768343e-06,
      "loss": 0.2299,
      "step": 292
    },
    {
      "epoch": 0.06666666666666667,
      "grad_norm": 1.7119116189327344,
      "learning_rate": 1.2494555040756737e-06,
      "loss": 0.0491,
      "step": 293
    },
    {
      "epoch": 0.06689419795221843,
      "grad_norm": 1.5996315416913807,
      "learning_rate": 1.2494517688117867e-06,
      "loss": 0.0569,
      "step": 294
    },
    {
      "epoch": 0.0671217292377702,
      "grad_norm": 2.2996053958249068,
      "learning_rate": 1.2494480207852489e-06,
      "loss": 0.0561,
      "step": 295
    },
    {
      "epoch": 0.06734926052332196,
      "grad_norm": 2.4074335741025346,
      "learning_rate": 1.249444259996137e-06,
      "loss": 0.0578,
      "step": 296
    },
    {
      "epoch": 0.06757679180887372,
      "grad_norm": 3.415732508941554,
      "learning_rate": 1.2494404864445284e-06,
      "loss": 0.0617,
      "step": 297
    },
    {
      "epoch": 0.06780432309442548,
      "grad_norm": 2.777678894153543,
      "learning_rate": 1.2494367001304996e-06,
      "loss": 0.081,
      "step": 298
    },
    {
      "epoch": 0.06803185437997725,
      "grad_norm": 1.6387141538383663,
      "learning_rate": 1.2494329010541284e-06,
      "loss": 0.0387,
      "step": 299
    },
    {
      "epoch": 0.06825938566552901,
      "grad_norm": 2.069756645203544,
      "learning_rate": 1.2494290892154922e-06,
      "loss": 0.0742,
      "step": 300
    },
    {
      "epoch": 0.06848691695108078,
      "grad_norm": 3.4307396136508332,
      "learning_rate": 1.2494252646146692e-06,
      "loss": 0.0957,
      "step": 301
    },
    {
      "epoch": 0.06871444823663253,
      "grad_norm": 2.0232162640823974,
      "learning_rate": 1.249421427251737e-06,
      "loss": 0.0522,
      "step": 302
    },
    {
      "epoch": 0.0689419795221843,
      "grad_norm": 2.146160892723777,
      "learning_rate": 1.2494175771267748e-06,
      "loss": 0.0896,
      "step": 303
    },
    {
      "epoch": 0.06916951080773606,
      "grad_norm": 2.940508414612204,
      "learning_rate": 1.2494137142398607e-06,
      "loss": 0.0776,
      "step": 304
    },
    {
      "epoch": 0.06939704209328783,
      "grad_norm": 4.306654470652337,
      "learning_rate": 1.249409838591074e-06,
      "loss": 0.112,
      "step": 305
    },
    {
      "epoch": 0.0696245733788396,
      "grad_norm": 2.698854439380438,
      "learning_rate": 1.2494059501804937e-06,
      "loss": 0.0922,
      "step": 306
    },
    {
      "epoch": 0.06985210466439136,
      "grad_norm": 2.684055328227073,
      "learning_rate": 1.249402049008199e-06,
      "loss": 0.0603,
      "step": 307
    },
    {
      "epoch": 0.07007963594994311,
      "grad_norm": 1.6527179950360487,
      "learning_rate": 1.2493981350742704e-06,
      "loss": 0.035,
      "step": 308
    },
    {
      "epoch": 0.07030716723549488,
      "grad_norm": 2.632569906843673,
      "learning_rate": 1.2493942083787872e-06,
      "loss": 0.0483,
      "step": 309
    },
    {
      "epoch": 0.07053469852104664,
      "grad_norm": 1.8419184740112247,
      "learning_rate": 1.2493902689218299e-06,
      "loss": 0.0592,
      "step": 310
    },
    {
      "epoch": 0.07076222980659841,
      "grad_norm": 4.098447223521742,
      "learning_rate": 1.249386316703479e-06,
      "loss": 0.0647,
      "step": 311
    },
    {
      "epoch": 0.07098976109215017,
      "grad_norm": 3.425692782797119,
      "learning_rate": 1.2493823517238154e-06,
      "loss": 0.0304,
      "step": 312
    },
    {
      "epoch": 0.07121729237770194,
      "grad_norm": 2.09845005259606,
      "learning_rate": 1.2493783739829202e-06,
      "loss": 0.0841,
      "step": 313
    },
    {
      "epoch": 0.07144482366325369,
      "grad_norm": 1.3953874008324176,
      "learning_rate": 1.2493743834808741e-06,
      "loss": 0.0572,
      "step": 314
    },
    {
      "epoch": 0.07167235494880546,
      "grad_norm": 2.623892824594455,
      "learning_rate": 1.2493703802177594e-06,
      "loss": 0.0625,
      "step": 315
    },
    {
      "epoch": 0.07189988623435722,
      "grad_norm": 1.2851580205849038,
      "learning_rate": 1.2493663641936576e-06,
      "loss": 0.0321,
      "step": 316
    },
    {
      "epoch": 0.07212741751990899,
      "grad_norm": 1.9343201552647253,
      "learning_rate": 1.2493623354086507e-06,
      "loss": 0.067,
      "step": 317
    },
    {
      "epoch": 0.07235494880546076,
      "grad_norm": 2.2726283028135286,
      "learning_rate": 1.2493582938628213e-06,
      "loss": 0.0797,
      "step": 318
    },
    {
      "epoch": 0.07258248009101251,
      "grad_norm": 1.669835013133446,
      "learning_rate": 1.2493542395562516e-06,
      "loss": 0.0534,
      "step": 319
    },
    {
      "epoch": 0.07281001137656427,
      "grad_norm": 1.684538229380955,
      "learning_rate": 1.2493501724890247e-06,
      "loss": 0.0728,
      "step": 320
    },
    {
      "epoch": 0.07303754266211604,
      "grad_norm": 1.9071354849813447,
      "learning_rate": 1.249346092661224e-06,
      "loss": 0.0678,
      "step": 321
    },
    {
      "epoch": 0.0732650739476678,
      "grad_norm": 1.7042139773464355,
      "learning_rate": 1.2493420000729322e-06,
      "loss": 0.0784,
      "step": 322
    },
    {
      "epoch": 0.07349260523321957,
      "grad_norm": 1.6735213517836012,
      "learning_rate": 1.2493378947242336e-06,
      "loss": 0.0776,
      "step": 323
    },
    {
      "epoch": 0.07372013651877134,
      "grad_norm": 1.5646112601537545,
      "learning_rate": 1.2493337766152119e-06,
      "loss": 0.0582,
      "step": 324
    },
    {
      "epoch": 0.07394766780432309,
      "grad_norm": 1.5543937889365869,
      "learning_rate": 1.249329645745951e-06,
      "loss": 0.0512,
      "step": 325
    },
    {
      "epoch": 0.07417519908987485,
      "grad_norm": 1.982236801977647,
      "learning_rate": 1.2493255021165357e-06,
      "loss": 0.0804,
      "step": 326
    },
    {
      "epoch": 0.07440273037542662,
      "grad_norm": 1.2394712998486772,
      "learning_rate": 1.2493213457270504e-06,
      "loss": 0.0534,
      "step": 327
    },
    {
      "epoch": 0.07463026166097839,
      "grad_norm": 1.7803933921864556,
      "learning_rate": 1.2493171765775804e-06,
      "loss": 0.0687,
      "step": 328
    },
    {
      "epoch": 0.07485779294653015,
      "grad_norm": 2.379962120888543,
      "learning_rate": 1.2493129946682107e-06,
      "loss": 0.071,
      "step": 329
    },
    {
      "epoch": 0.07508532423208192,
      "grad_norm": 1.589657975033299,
      "learning_rate": 1.2493087999990263e-06,
      "loss": 0.0739,
      "step": 330
    },
    {
      "epoch": 0.07531285551763367,
      "grad_norm": 3.043606210421758,
      "learning_rate": 1.249304592570114e-06,
      "loss": 0.1518,
      "step": 331
    },
    {
      "epoch": 0.07554038680318544,
      "grad_norm": 1.4392385212589442,
      "learning_rate": 1.2493003723815588e-06,
      "loss": 0.0556,
      "step": 332
    },
    {
      "epoch": 0.0757679180887372,
      "grad_norm": 2.0188482312629397,
      "learning_rate": 1.2492961394334474e-06,
      "loss": 0.0662,
      "step": 333
    },
    {
      "epoch": 0.07599544937428897,
      "grad_norm": 1.3600493699697815,
      "learning_rate": 1.2492918937258663e-06,
      "loss": 0.042,
      "step": 334
    },
    {
      "epoch": 0.07622298065984073,
      "grad_norm": 2.2314121339792714,
      "learning_rate": 1.2492876352589024e-06,
      "loss": 0.0751,
      "step": 335
    },
    {
      "epoch": 0.07645051194539249,
      "grad_norm": 1.4434673698644476,
      "learning_rate": 1.2492833640326424e-06,
      "loss": 0.0359,
      "step": 336
    },
    {
      "epoch": 0.07667804323094425,
      "grad_norm": 1.9053597077248743,
      "learning_rate": 1.2492790800471738e-06,
      "loss": 0.0881,
      "step": 337
    },
    {
      "epoch": 0.07690557451649602,
      "grad_norm": 0.9804503872776514,
      "learning_rate": 1.249274783302584e-06,
      "loss": 0.0332,
      "step": 338
    },
    {
      "epoch": 0.07713310580204778,
      "grad_norm": 1.439933416412126,
      "learning_rate": 1.249270473798961e-06,
      "loss": 0.0363,
      "step": 339
    },
    {
      "epoch": 0.07736063708759955,
      "grad_norm": 1.8018763642808848,
      "learning_rate": 1.249266151536393e-06,
      "loss": 0.0652,
      "step": 340
    },
    {
      "epoch": 0.07758816837315131,
      "grad_norm": 1.605885586174771,
      "learning_rate": 1.249261816514968e-06,
      "loss": 0.0483,
      "step": 341
    },
    {
      "epoch": 0.07781569965870307,
      "grad_norm": 1.5475812110829932,
      "learning_rate": 1.2492574687347747e-06,
      "loss": 0.0518,
      "step": 342
    },
    {
      "epoch": 0.07804323094425483,
      "grad_norm": 1.3861334147739204,
      "learning_rate": 1.249253108195902e-06,
      "loss": 0.0438,
      "step": 343
    },
    {
      "epoch": 0.0782707622298066,
      "grad_norm": 1.6280816638823692,
      "learning_rate": 1.249248734898439e-06,
      "loss": 0.0804,
      "step": 344
    },
    {
      "epoch": 0.07849829351535836,
      "grad_norm": 1.945038730906618,
      "learning_rate": 1.2492443488424753e-06,
      "loss": 0.08,
      "step": 345
    },
    {
      "epoch": 0.07872582480091013,
      "grad_norm": 1.163435954016696,
      "learning_rate": 1.2492399500281002e-06,
      "loss": 0.0404,
      "step": 346
    },
    {
      "epoch": 0.07895335608646188,
      "grad_norm": 2.0650618862775225,
      "learning_rate": 1.2492355384554039e-06,
      "loss": 0.0965,
      "step": 347
    },
    {
      "epoch": 0.07918088737201365,
      "grad_norm": 1.2688387295944672,
      "learning_rate": 1.2492311141244764e-06,
      "loss": 0.0387,
      "step": 348
    },
    {
      "epoch": 0.07940841865756541,
      "grad_norm": 2.2968428653497472,
      "learning_rate": 1.249226677035408e-06,
      "loss": 0.0569,
      "step": 349
    },
    {
      "epoch": 0.07963594994311718,
      "grad_norm": 1.3014564625890543,
      "learning_rate": 1.2492222271882896e-06,
      "loss": 0.0418,
      "step": 350
    },
    {
      "epoch": 0.07986348122866894,
      "grad_norm": 2.3510870610479433,
      "learning_rate": 1.2492177645832121e-06,
      "loss": 0.0932,
      "step": 351
    },
    {
      "epoch": 0.08009101251422071,
      "grad_norm": 1.4510684428455591,
      "learning_rate": 1.2492132892202668e-06,
      "loss": 0.0513,
      "step": 352
    },
    {
      "epoch": 0.08031854379977246,
      "grad_norm": 4.594413722551582,
      "learning_rate": 1.2492088010995449e-06,
      "loss": 0.1235,
      "step": 353
    },
    {
      "epoch": 0.08054607508532423,
      "grad_norm": 2.137036810353962,
      "learning_rate": 1.2492043002211385e-06,
      "loss": 0.1026,
      "step": 354
    },
    {
      "epoch": 0.080773606370876,
      "grad_norm": 0.7577275839475693,
      "learning_rate": 1.2491997865851392e-06,
      "loss": 0.0368,
      "step": 355
    },
    {
      "epoch": 0.08100113765642776,
      "grad_norm": 1.0494266975406659,
      "learning_rate": 1.2491952601916395e-06,
      "loss": 0.0385,
      "step": 356
    },
    {
      "epoch": 0.08122866894197953,
      "grad_norm": 1.0423156372262818,
      "learning_rate": 1.2491907210407319e-06,
      "loss": 0.0424,
      "step": 357
    },
    {
      "epoch": 0.08145620022753129,
      "grad_norm": 1.150214646729361,
      "learning_rate": 1.249186169132509e-06,
      "loss": 0.0694,
      "step": 358
    },
    {
      "epoch": 0.08168373151308304,
      "grad_norm": 2.382439951131263,
      "learning_rate": 1.2491816044670641e-06,
      "loss": 0.0367,
      "step": 359
    },
    {
      "epoch": 0.08191126279863481,
      "grad_norm": 1.7484749214098252,
      "learning_rate": 1.24917702704449e-06,
      "loss": 0.0677,
      "step": 360
    },
    {
      "epoch": 0.08213879408418658,
      "grad_norm": 2.1328792490346227,
      "learning_rate": 1.2491724368648808e-06,
      "loss": 0.049,
      "step": 361
    },
    {
      "epoch": 0.08236632536973834,
      "grad_norm": 0.6514880974265386,
      "learning_rate": 1.2491678339283303e-06,
      "loss": 0.0217,
      "step": 362
    },
    {
      "epoch": 0.08259385665529011,
      "grad_norm": 1.3802700887569803,
      "learning_rate": 1.249163218234932e-06,
      "loss": 0.0406,
      "step": 363
    },
    {
      "epoch": 0.08282138794084186,
      "grad_norm": 1.7918385557024892,
      "learning_rate": 1.249158589784781e-06,
      "loss": 0.064,
      "step": 364
    },
    {
      "epoch": 0.08304891922639362,
      "grad_norm": 1.2060165323705336,
      "learning_rate": 1.2491539485779713e-06,
      "loss": 0.0367,
      "step": 365
    },
    {
      "epoch": 0.08327645051194539,
      "grad_norm": 1.2166249313910695,
      "learning_rate": 1.2491492946145981e-06,
      "loss": 0.0512,
      "step": 366
    },
    {
      "epoch": 0.08350398179749716,
      "grad_norm": 1.3048815510766842,
      "learning_rate": 1.2491446278947563e-06,
      "loss": 0.0554,
      "step": 367
    },
    {
      "epoch": 0.08373151308304892,
      "grad_norm": 1.9459326713846232,
      "learning_rate": 1.2491399484185413e-06,
      "loss": 0.0488,
      "step": 368
    },
    {
      "epoch": 0.08395904436860069,
      "grad_norm": 1.7915232664153333,
      "learning_rate": 1.249135256186049e-06,
      "loss": 0.1229,
      "step": 369
    },
    {
      "epoch": 0.08418657565415244,
      "grad_norm": 0.9354087864668412,
      "learning_rate": 1.249130551197375e-06,
      "loss": 0.0383,
      "step": 370
    },
    {
      "epoch": 0.0844141069397042,
      "grad_norm": 2.07264306868735,
      "learning_rate": 1.2491258334526155e-06,
      "loss": 0.0794,
      "step": 371
    },
    {
      "epoch": 0.08464163822525597,
      "grad_norm": 1.4534854537948092,
      "learning_rate": 1.2491211029518672e-06,
      "loss": 0.0282,
      "step": 372
    },
    {
      "epoch": 0.08486916951080774,
      "grad_norm": 2.0044431124055984,
      "learning_rate": 1.2491163596952264e-06,
      "loss": 0.0709,
      "step": 373
    },
    {
      "epoch": 0.0850967007963595,
      "grad_norm": 1.6305074169029203,
      "learning_rate": 1.2491116036827902e-06,
      "loss": 0.0407,
      "step": 374
    },
    {
      "epoch": 0.08532423208191127,
      "grad_norm": 1.223898789967783,
      "learning_rate": 1.2491068349146559e-06,
      "loss": 0.0595,
      "step": 375
    },
    {
      "epoch": 0.08555176336746302,
      "grad_norm": 1.7895390232916124,
      "learning_rate": 1.249102053390921e-06,
      "loss": 0.1034,
      "step": 376
    },
    {
      "epoch": 0.08577929465301479,
      "grad_norm": 1.6725786460952021,
      "learning_rate": 1.249097259111683e-06,
      "loss": 0.0955,
      "step": 377
    },
    {
      "epoch": 0.08600682593856655,
      "grad_norm": 2.4309538605657735,
      "learning_rate": 1.24909245207704e-06,
      "loss": 0.0794,
      "step": 378
    },
    {
      "epoch": 0.08623435722411832,
      "grad_norm": 0.8982937912678952,
      "learning_rate": 1.2490876322870904e-06,
      "loss": 0.0302,
      "step": 379
    },
    {
      "epoch": 0.08646188850967008,
      "grad_norm": 3.0756164983832,
      "learning_rate": 1.2490827997419325e-06,
      "loss": 0.0549,
      "step": 380
    },
    {
      "epoch": 0.08668941979522184,
      "grad_norm": 3.676872613008804,
      "learning_rate": 1.249077954441665e-06,
      "loss": 0.141,
      "step": 381
    },
    {
      "epoch": 0.0869169510807736,
      "grad_norm": 1.4373261550715828,
      "learning_rate": 1.249073096386387e-06,
      "loss": 0.0392,
      "step": 382
    },
    {
      "epoch": 0.08714448236632537,
      "grad_norm": 3.1988616398871006,
      "learning_rate": 1.249068225576198e-06,
      "loss": 0.0915,
      "step": 383
    },
    {
      "epoch": 0.08737201365187713,
      "grad_norm": 1.5619645893707617,
      "learning_rate": 1.2490633420111974e-06,
      "loss": 0.0541,
      "step": 384
    },
    {
      "epoch": 0.0875995449374289,
      "grad_norm": 0.9211933924050371,
      "learning_rate": 1.249058445691485e-06,
      "loss": 0.0337,
      "step": 385
    },
    {
      "epoch": 0.08782707622298067,
      "grad_norm": 1.1767496728905957,
      "learning_rate": 1.2490535366171607e-06,
      "loss": 0.039,
      "step": 386
    },
    {
      "epoch": 0.08805460750853242,
      "grad_norm": 2.9534787325055274,
      "learning_rate": 1.249048614788325e-06,
      "loss": 0.1048,
      "step": 387
    },
    {
      "epoch": 0.08828213879408418,
      "grad_norm": 6.651807262283111,
      "learning_rate": 1.249043680205079e-06,
      "loss": 0.0494,
      "step": 388
    },
    {
      "epoch": 0.08850967007963595,
      "grad_norm": 2.5123377668968745,
      "learning_rate": 1.2490387328675226e-06,
      "loss": 0.1199,
      "step": 389
    },
    {
      "epoch": 0.08873720136518772,
      "grad_norm": 1.5502594476140923,
      "learning_rate": 1.2490337727757576e-06,
      "loss": 0.0503,
      "step": 390
    },
    {
      "epoch": 0.08896473265073948,
      "grad_norm": 5.640796726611394,
      "learning_rate": 1.249028799929885e-06,
      "loss": 0.06,
      "step": 391
    },
    {
      "epoch": 0.08919226393629125,
      "grad_norm": 1.2047358188375967,
      "learning_rate": 1.2490238143300066e-06,
      "loss": 0.0427,
      "step": 392
    },
    {
      "epoch": 0.089419795221843,
      "grad_norm": 1.5816796650719427,
      "learning_rate": 1.2490188159762243e-06,
      "loss": 0.0665,
      "step": 393
    },
    {
      "epoch": 0.08964732650739476,
      "grad_norm": 2.8164102954322505,
      "learning_rate": 1.2490138048686405e-06,
      "loss": 0.076,
      "step": 394
    },
    {
      "epoch": 0.08987485779294653,
      "grad_norm": 2.0775035442513596,
      "learning_rate": 1.249008781007357e-06,
      "loss": 0.0857,
      "step": 395
    },
    {
      "epoch": 0.0901023890784983,
      "grad_norm": 1.3816134442939907,
      "learning_rate": 1.2490037443924768e-06,
      "loss": 0.0647,
      "step": 396
    },
    {
      "epoch": 0.09032992036405006,
      "grad_norm": 5.3602328587776205,
      "learning_rate": 1.2489986950241032e-06,
      "loss": 0.0857,
      "step": 397
    },
    {
      "epoch": 0.09055745164960181,
      "grad_norm": 1.3668732893642688,
      "learning_rate": 1.2489936329023387e-06,
      "loss": 0.0513,
      "step": 398
    },
    {
      "epoch": 0.09078498293515358,
      "grad_norm": 1.3818985376090422,
      "learning_rate": 1.2489885580272874e-06,
      "loss": 0.0643,
      "step": 399
    },
    {
      "epoch": 0.09101251422070535,
      "grad_norm": 1.1645733063202421,
      "learning_rate": 1.2489834703990527e-06,
      "loss": 0.0564,
      "step": 400
    },
    {
      "epoch": 0.09124004550625711,
      "grad_norm": 2.579861013447331,
      "learning_rate": 1.2489783700177385e-06,
      "loss": 0.0844,
      "step": 401
    },
    {
      "epoch": 0.09146757679180888,
      "grad_norm": 1.7416779142182774,
      "learning_rate": 1.2489732568834492e-06,
      "loss": 0.0874,
      "step": 402
    },
    {
      "epoch": 0.09169510807736064,
      "grad_norm": 1.2036980088831546,
      "learning_rate": 1.2489681309962895e-06,
      "loss": 0.0605,
      "step": 403
    },
    {
      "epoch": 0.0919226393629124,
      "grad_norm": 2.823337671795372,
      "learning_rate": 1.2489629923563637e-06,
      "loss": 0.0563,
      "step": 404
    },
    {
      "epoch": 0.09215017064846416,
      "grad_norm": 3.5147572243595406,
      "learning_rate": 1.2489578409637774e-06,
      "loss": 0.0962,
      "step": 405
    },
    {
      "epoch": 0.09237770193401593,
      "grad_norm": 2.441595167267241,
      "learning_rate": 1.2489526768186352e-06,
      "loss": 0.0409,
      "step": 406
    },
    {
      "epoch": 0.09260523321956769,
      "grad_norm": 1.606289452778594,
      "learning_rate": 1.2489474999210434e-06,
      "loss": 0.083,
      "step": 407
    },
    {
      "epoch": 0.09283276450511946,
      "grad_norm": 2.0108948875054358,
      "learning_rate": 1.2489423102711068e-06,
      "loss": 0.0424,
      "step": 408
    },
    {
      "epoch": 0.09306029579067122,
      "grad_norm": 1.9517497199537932,
      "learning_rate": 1.2489371078689326e-06,
      "loss": 0.063,
      "step": 409
    },
    {
      "epoch": 0.09328782707622298,
      "grad_norm": 2.1984153851855903,
      "learning_rate": 1.2489318927146263e-06,
      "loss": 0.0607,
      "step": 410
    },
    {
      "epoch": 0.09351535836177474,
      "grad_norm": 0.9821028313036854,
      "learning_rate": 1.2489266648082951e-06,
      "loss": 0.0269,
      "step": 411
    },
    {
      "epoch": 0.09374288964732651,
      "grad_norm": 1.4642895316806703,
      "learning_rate": 1.2489214241500453e-06,
      "loss": 0.0773,
      "step": 412
    },
    {
      "epoch": 0.09397042093287827,
      "grad_norm": 2.6315717094149598,
      "learning_rate": 1.2489161707399843e-06,
      "loss": 0.1042,
      "step": 413
    },
    {
      "epoch": 0.09419795221843004,
      "grad_norm": 1.2730067445859696,
      "learning_rate": 1.2489109045782194e-06,
      "loss": 0.0397,
      "step": 414
    },
    {
      "epoch": 0.09442548350398179,
      "grad_norm": 1.0214233244231261,
      "learning_rate": 1.2489056256648582e-06,
      "loss": 0.0297,
      "step": 415
    },
    {
      "epoch": 0.09465301478953356,
      "grad_norm": 1.8895414872949885,
      "learning_rate": 1.2489003340000089e-06,
      "loss": 0.1027,
      "step": 416
    },
    {
      "epoch": 0.09488054607508532,
      "grad_norm": 1.2431995135273388,
      "learning_rate": 1.2488950295837792e-06,
      "loss": 0.0694,
      "step": 417
    },
    {
      "epoch": 0.09510807736063709,
      "grad_norm": 2.741996145452892,
      "learning_rate": 1.2488897124162777e-06,
      "loss": 0.0815,
      "step": 418
    },
    {
      "epoch": 0.09533560864618885,
      "grad_norm": 1.711178715312897,
      "learning_rate": 1.248884382497613e-06,
      "loss": 0.0955,
      "step": 419
    },
    {
      "epoch": 0.09556313993174062,
      "grad_norm": 2.609523312381175,
      "learning_rate": 1.2488790398278941e-06,
      "loss": 0.1047,
      "step": 420
    },
    {
      "epoch": 0.09579067121729237,
      "grad_norm": 1.53212810794613,
      "learning_rate": 1.2488736844072304e-06,
      "loss": 0.0585,
      "step": 421
    },
    {
      "epoch": 0.09601820250284414,
      "grad_norm": 1.4174025848206742,
      "learning_rate": 1.248868316235731e-06,
      "loss": 0.0387,
      "step": 422
    },
    {
      "epoch": 0.0962457337883959,
      "grad_norm": 2.1766082548840644,
      "learning_rate": 1.2488629353135059e-06,
      "loss": 0.0883,
      "step": 423
    },
    {
      "epoch": 0.09647326507394767,
      "grad_norm": 1.2878325712311849,
      "learning_rate": 1.2488575416406649e-06,
      "loss": 0.0469,
      "step": 424
    },
    {
      "epoch": 0.09670079635949944,
      "grad_norm": 2.1138235879975222,
      "learning_rate": 1.2488521352173183e-06,
      "loss": 0.0531,
      "step": 425
    },
    {
      "epoch": 0.09692832764505119,
      "grad_norm": 1.6090227620325084,
      "learning_rate": 1.2488467160435765e-06,
      "loss": 0.0532,
      "step": 426
    },
    {
      "epoch": 0.09715585893060295,
      "grad_norm": 1.5021035863134842,
      "learning_rate": 1.2488412841195505e-06,
      "loss": 0.0805,
      "step": 427
    },
    {
      "epoch": 0.09738339021615472,
      "grad_norm": 2.0944361790568515,
      "learning_rate": 1.2488358394453512e-06,
      "loss": 0.0864,
      "step": 428
    },
    {
      "epoch": 0.09761092150170649,
      "grad_norm": 1.3336921136142907,
      "learning_rate": 1.2488303820210897e-06,
      "loss": 0.0405,
      "step": 429
    },
    {
      "epoch": 0.09783845278725825,
      "grad_norm": 1.4602632552266945,
      "learning_rate": 1.2488249118468776e-06,
      "loss": 0.0319,
      "step": 430
    },
    {
      "epoch": 0.09806598407281002,
      "grad_norm": 1.1300406263272467,
      "learning_rate": 1.248819428922827e-06,
      "loss": 0.0328,
      "step": 431
    },
    {
      "epoch": 0.09829351535836177,
      "grad_norm": 3.2436410504809006,
      "learning_rate": 1.2488139332490495e-06,
      "loss": 0.1052,
      "step": 432
    },
    {
      "epoch": 0.09852104664391353,
      "grad_norm": 1.1456421976312734,
      "learning_rate": 1.248808424825658e-06,
      "loss": 0.0455,
      "step": 433
    },
    {
      "epoch": 0.0987485779294653,
      "grad_norm": 1.8967128040132528,
      "learning_rate": 1.2488029036527645e-06,
      "loss": 0.0521,
      "step": 434
    },
    {
      "epoch": 0.09897610921501707,
      "grad_norm": 2.22293609570488,
      "learning_rate": 1.2487973697304822e-06,
      "loss": 0.0401,
      "step": 435
    },
    {
      "epoch": 0.09920364050056883,
      "grad_norm": 1.3201279140220556,
      "learning_rate": 1.248791823058924e-06,
      "loss": 0.0656,
      "step": 436
    },
    {
      "epoch": 0.0994311717861206,
      "grad_norm": 1.8801560325756854,
      "learning_rate": 1.2487862636382034e-06,
      "loss": 0.036,
      "step": 437
    },
    {
      "epoch": 0.09965870307167235,
      "grad_norm": 0.9413516301280102,
      "learning_rate": 1.248780691468434e-06,
      "loss": 0.0369,
      "step": 438
    },
    {
      "epoch": 0.09988623435722412,
      "grad_norm": 1.6280926081257134,
      "learning_rate": 1.2487751065497296e-06,
      "loss": 0.0679,
      "step": 439
    },
    {
      "epoch": 0.10011376564277588,
      "grad_norm": 1.1483666492978277,
      "learning_rate": 1.2487695088822044e-06,
      "loss": 0.0476,
      "step": 440
    },
    {
      "epoch": 0.10034129692832765,
      "grad_norm": 1.091509019893692,
      "learning_rate": 1.2487638984659729e-06,
      "loss": 0.0267,
      "step": 441
    },
    {
      "epoch": 0.10056882821387941,
      "grad_norm": 1.2894939135732226,
      "learning_rate": 1.2487582753011496e-06,
      "loss": 0.0379,
      "step": 442
    },
    {
      "epoch": 0.10079635949943117,
      "grad_norm": 1.5539892877637305,
      "learning_rate": 1.2487526393878497e-06,
      "loss": 0.0717,
      "step": 443
    },
    {
      "epoch": 0.10102389078498293,
      "grad_norm": 1.0799436334247106,
      "learning_rate": 1.248746990726188e-06,
      "loss": 0.0494,
      "step": 444
    },
    {
      "epoch": 0.1012514220705347,
      "grad_norm": 1.0482657304812906,
      "learning_rate": 1.2487413293162803e-06,
      "loss": 0.0328,
      "step": 445
    },
    {
      "epoch": 0.10147895335608646,
      "grad_norm": 2.974661390553103,
      "learning_rate": 1.2487356551582421e-06,
      "loss": 0.1436,
      "step": 446
    },
    {
      "epoch": 0.10170648464163823,
      "grad_norm": 2.6800805691171843,
      "learning_rate": 1.2487299682521893e-06,
      "loss": 0.1383,
      "step": 447
    },
    {
      "epoch": 0.10193401592719,
      "grad_norm": 2.2359167995553233,
      "learning_rate": 1.2487242685982384e-06,
      "loss": 0.0485,
      "step": 448
    },
    {
      "epoch": 0.10216154721274175,
      "grad_norm": 2.818745463327436,
      "learning_rate": 1.2487185561965057e-06,
      "loss": 0.0446,
      "step": 449
    },
    {
      "epoch": 0.10238907849829351,
      "grad_norm": 0.6315900981706886,
      "learning_rate": 1.248712831047108e-06,
      "loss": 0.0186,
      "step": 450
    },
    {
      "epoch": 0.10261660978384528,
      "grad_norm": 1.9674239723403535,
      "learning_rate": 1.2487070931501624e-06,
      "loss": 0.0673,
      "step": 451
    },
    {
      "epoch": 0.10284414106939704,
      "grad_norm": 1.6885325551942292,
      "learning_rate": 1.2487013425057858e-06,
      "loss": 0.0767,
      "step": 452
    },
    {
      "epoch": 0.10307167235494881,
      "grad_norm": 1.891051810210748,
      "learning_rate": 1.2486955791140964e-06,
      "loss": 0.063,
      "step": 453
    },
    {
      "epoch": 0.10329920364050058,
      "grad_norm": 1.1912814980714257,
      "learning_rate": 1.2486898029752113e-06,
      "loss": 0.0493,
      "step": 454
    },
    {
      "epoch": 0.10352673492605233,
      "grad_norm": 0.8883039975316582,
      "learning_rate": 1.248684014089249e-06,
      "loss": 0.0277,
      "step": 455
    },
    {
      "epoch": 0.1037542662116041,
      "grad_norm": 1.306066876530539,
      "learning_rate": 1.2486782124563277e-06,
      "loss": 0.0544,
      "step": 456
    },
    {
      "epoch": 0.10398179749715586,
      "grad_norm": 1.5420420577450527,
      "learning_rate": 1.2486723980765659e-06,
      "loss": 0.077,
      "step": 457
    },
    {
      "epoch": 0.10420932878270762,
      "grad_norm": 2.0367683349755823,
      "learning_rate": 1.2486665709500826e-06,
      "loss": 0.0649,
      "step": 458
    },
    {
      "epoch": 0.10443686006825939,
      "grad_norm": 1.1542715096556455,
      "learning_rate": 1.2486607310769965e-06,
      "loss": 0.0586,
      "step": 459
    },
    {
      "epoch": 0.10466439135381114,
      "grad_norm": 2.061046992152943,
      "learning_rate": 1.2486548784574275e-06,
      "loss": 0.0491,
      "step": 460
    },
    {
      "epoch": 0.10489192263936291,
      "grad_norm": 1.0629825041281826,
      "learning_rate": 1.2486490130914948e-06,
      "loss": 0.0445,
      "step": 461
    },
    {
      "epoch": 0.10511945392491467,
      "grad_norm": 1.415699131985968,
      "learning_rate": 1.2486431349793185e-06,
      "loss": 0.0679,
      "step": 462
    },
    {
      "epoch": 0.10534698521046644,
      "grad_norm": 1.8548752695365796,
      "learning_rate": 1.2486372441210188e-06,
      "loss": 0.0514,
      "step": 463
    },
    {
      "epoch": 0.1055745164960182,
      "grad_norm": 1.1301759724239804,
      "learning_rate": 1.248631340516716e-06,
      "loss": 0.0381,
      "step": 464
    },
    {
      "epoch": 0.10580204778156997,
      "grad_norm": 1.408994037607189,
      "learning_rate": 1.2486254241665302e-06,
      "loss": 0.0692,
      "step": 465
    },
    {
      "epoch": 0.10602957906712172,
      "grad_norm": 1.6631340659469183,
      "learning_rate": 1.2486194950705831e-06,
      "loss": 0.0454,
      "step": 466
    },
    {
      "epoch": 0.10625711035267349,
      "grad_norm": 4.446207788916147,
      "learning_rate": 1.248613553228996e-06,
      "loss": 0.051,
      "step": 467
    },
    {
      "epoch": 0.10648464163822526,
      "grad_norm": 2.558672760104107,
      "learning_rate": 1.2486075986418896e-06,
      "loss": 0.0581,
      "step": 468
    },
    {
      "epoch": 0.10671217292377702,
      "grad_norm": 1.1293994671100194,
      "learning_rate": 1.248601631309386e-06,
      "loss": 0.0601,
      "step": 469
    },
    {
      "epoch": 0.10693970420932879,
      "grad_norm": 0.9858665143030374,
      "learning_rate": 1.2485956512316072e-06,
      "loss": 0.042,
      "step": 470
    },
    {
      "epoch": 0.10716723549488055,
      "grad_norm": 1.0238223731711866,
      "learning_rate": 1.2485896584086754e-06,
      "loss": 0.0419,
      "step": 471
    },
    {
      "epoch": 0.1073947667804323,
      "grad_norm": 1.4636163379221103,
      "learning_rate": 1.248583652840713e-06,
      "loss": 0.0763,
      "step": 472
    },
    {
      "epoch": 0.10762229806598407,
      "grad_norm": 2.3052142467408534,
      "learning_rate": 1.2485776345278427e-06,
      "loss": 0.0577,
      "step": 473
    },
    {
      "epoch": 0.10784982935153584,
      "grad_norm": 1.797428495175526,
      "learning_rate": 1.2485716034701876e-06,
      "loss": 0.0664,
      "step": 474
    },
    {
      "epoch": 0.1080773606370876,
      "grad_norm": 0.9502391190499007,
      "learning_rate": 1.2485655596678712e-06,
      "loss": 0.032,
      "step": 475
    },
    {
      "epoch": 0.10830489192263937,
      "grad_norm": 2.0597597691340033,
      "learning_rate": 1.2485595031210164e-06,
      "loss": 0.0693,
      "step": 476
    },
    {
      "epoch": 0.10853242320819112,
      "grad_norm": 1.6422248848311762,
      "learning_rate": 1.2485534338297475e-06,
      "loss": 0.036,
      "step": 477
    },
    {
      "epoch": 0.10875995449374289,
      "grad_norm": 1.0205877451233056,
      "learning_rate": 1.2485473517941884e-06,
      "loss": 0.036,
      "step": 478
    },
    {
      "epoch": 0.10898748577929465,
      "grad_norm": 1.352496943979393,
      "learning_rate": 1.2485412570144633e-06,
      "loss": 0.0619,
      "step": 479
    },
    {
      "epoch": 0.10921501706484642,
      "grad_norm": 1.1460532851585836,
      "learning_rate": 1.2485351494906969e-06,
      "loss": 0.0457,
      "step": 480
    },
    {
      "epoch": 0.10944254835039818,
      "grad_norm": 2.136834742910293,
      "learning_rate": 1.2485290292230142e-06,
      "loss": 0.1229,
      "step": 481
    },
    {
      "epoch": 0.10967007963594995,
      "grad_norm": 1.2780826832089733,
      "learning_rate": 1.24852289621154e-06,
      "loss": 0.0295,
      "step": 482
    },
    {
      "epoch": 0.1098976109215017,
      "grad_norm": 1.5165261909270789,
      "learning_rate": 1.2485167504563995e-06,
      "loss": 0.0605,
      "step": 483
    },
    {
      "epoch": 0.11012514220705347,
      "grad_norm": 2.5598525552258153,
      "learning_rate": 1.2485105919577187e-06,
      "loss": 0.1343,
      "step": 484
    },
    {
      "epoch": 0.11035267349260523,
      "grad_norm": 1.2948078456387253,
      "learning_rate": 1.2485044207156233e-06,
      "loss": 0.0462,
      "step": 485
    },
    {
      "epoch": 0.110580204778157,
      "grad_norm": 2.2282842690273674,
      "learning_rate": 1.2484982367302395e-06,
      "loss": 0.0906,
      "step": 486
    },
    {
      "epoch": 0.11080773606370876,
      "grad_norm": 1.58668388631447,
      "learning_rate": 1.2484920400016936e-06,
      "loss": 0.0754,
      "step": 487
    },
    {
      "epoch": 0.11103526734926053,
      "grad_norm": 1.8486367636195544,
      "learning_rate": 1.2484858305301122e-06,
      "loss": 0.0858,
      "step": 488
    },
    {
      "epoch": 0.11126279863481228,
      "grad_norm": 1.2490242519124817,
      "learning_rate": 1.2484796083156222e-06,
      "loss": 0.0412,
      "step": 489
    },
    {
      "epoch": 0.11149032992036405,
      "grad_norm": 2.4293013886013024,
      "learning_rate": 1.2484733733583511e-06,
      "loss": 0.0347,
      "step": 490
    },
    {
      "epoch": 0.11171786120591581,
      "grad_norm": 1.3895430378196065,
      "learning_rate": 1.248467125658426e-06,
      "loss": 0.0508,
      "step": 491
    },
    {
      "epoch": 0.11194539249146758,
      "grad_norm": 1.0989498931762587,
      "learning_rate": 1.2484608652159746e-06,
      "loss": 0.0376,
      "step": 492
    },
    {
      "epoch": 0.11217292377701935,
      "grad_norm": 1.2816896914965858,
      "learning_rate": 1.248454592031125e-06,
      "loss": 0.0325,
      "step": 493
    },
    {
      "epoch": 0.1124004550625711,
      "grad_norm": 1.3834104899544186,
      "learning_rate": 1.2484483061040054e-06,
      "loss": 0.0502,
      "step": 494
    },
    {
      "epoch": 0.11262798634812286,
      "grad_norm": 1.8438437167351667,
      "learning_rate": 1.2484420074347441e-06,
      "loss": 0.0675,
      "step": 495
    },
    {
      "epoch": 0.11285551763367463,
      "grad_norm": 1.3847331523036337,
      "learning_rate": 1.24843569602347e-06,
      "loss": 0.0708,
      "step": 496
    },
    {
      "epoch": 0.1130830489192264,
      "grad_norm": 1.4695794221134664,
      "learning_rate": 1.2484293718703119e-06,
      "loss": 0.0677,
      "step": 497
    },
    {
      "epoch": 0.11331058020477816,
      "grad_norm": 1.4197704649286171,
      "learning_rate": 1.2484230349753994e-06,
      "loss": 0.0275,
      "step": 498
    },
    {
      "epoch": 0.11353811149032993,
      "grad_norm": 1.9758484599559376,
      "learning_rate": 1.2484166853388617e-06,
      "loss": 0.0846,
      "step": 499
    },
    {
      "epoch": 0.11376564277588168,
      "grad_norm": 1.4480733215282273,
      "learning_rate": 1.2484103229608288e-06,
      "loss": 0.0302,
      "step": 500
    },
    {
      "epoch": 0.11399317406143344,
      "grad_norm": 1.6803783059061204,
      "learning_rate": 1.2484039478414305e-06,
      "loss": 0.033,
      "step": 501
    },
    {
      "epoch": 0.11422070534698521,
      "grad_norm": 1.5093539895040036,
      "learning_rate": 1.2483975599807972e-06,
      "loss": 0.0592,
      "step": 502
    },
    {
      "epoch": 0.11444823663253698,
      "grad_norm": 1.2912092446277637,
      "learning_rate": 1.2483911593790595e-06,
      "loss": 0.0341,
      "step": 503
    },
    {
      "epoch": 0.11467576791808874,
      "grad_norm": 4.053757173255741,
      "learning_rate": 1.2483847460363482e-06,
      "loss": 0.08,
      "step": 504
    },
    {
      "epoch": 0.1149032992036405,
      "grad_norm": 1.8561008975587792,
      "learning_rate": 1.2483783199527943e-06,
      "loss": 0.0417,
      "step": 505
    },
    {
      "epoch": 0.11513083048919226,
      "grad_norm": 1.7954101019883197,
      "learning_rate": 1.2483718811285296e-06,
      "loss": 0.098,
      "step": 506
    },
    {
      "epoch": 0.11535836177474403,
      "grad_norm": 0.90376517227131,
      "learning_rate": 1.2483654295636848e-06,
      "loss": 0.0204,
      "step": 507
    },
    {
      "epoch": 0.11558589306029579,
      "grad_norm": 1.0402238740609373,
      "learning_rate": 1.2483589652583924e-06,
      "loss": 0.0393,
      "step": 508
    },
    {
      "epoch": 0.11581342434584756,
      "grad_norm": 1.2739463431329932,
      "learning_rate": 1.2483524882127846e-06,
      "loss": 0.0361,
      "step": 509
    },
    {
      "epoch": 0.11604095563139932,
      "grad_norm": 1.8542368928683455,
      "learning_rate": 1.2483459984269933e-06,
      "loss": 0.062,
      "step": 510
    },
    {
      "epoch": 0.11626848691695107,
      "grad_norm": 1.4901585211244064,
      "learning_rate": 1.2483394959011514e-06,
      "loss": 0.0362,
      "step": 511
    },
    {
      "epoch": 0.11649601820250284,
      "grad_norm": 1.655677974364157,
      "learning_rate": 1.248332980635392e-06,
      "loss": 0.068,
      "step": 512
    },
    {
      "epoch": 0.1167235494880546,
      "grad_norm": 1.40135662829084,
      "learning_rate": 1.2483264526298478e-06,
      "loss": 0.0467,
      "step": 513
    },
    {
      "epoch": 0.11695108077360637,
      "grad_norm": 0.9801745008839565,
      "learning_rate": 1.2483199118846525e-06,
      "loss": 0.0547,
      "step": 514
    },
    {
      "epoch": 0.11717861205915814,
      "grad_norm": 0.7956319000441687,
      "learning_rate": 1.2483133583999399e-06,
      "loss": 0.0292,
      "step": 515
    },
    {
      "epoch": 0.1174061433447099,
      "grad_norm": 1.5822262084465146,
      "learning_rate": 1.2483067921758439e-06,
      "loss": 0.0442,
      "step": 516
    },
    {
      "epoch": 0.11763367463026166,
      "grad_norm": 1.788875375154368,
      "learning_rate": 1.2483002132124983e-06,
      "loss": 0.1027,
      "step": 517
    },
    {
      "epoch": 0.11786120591581342,
      "grad_norm": 1.12691319203208,
      "learning_rate": 1.2482936215100382e-06,
      "loss": 0.0392,
      "step": 518
    },
    {
      "epoch": 0.11808873720136519,
      "grad_norm": 1.4278197951300633,
      "learning_rate": 1.2482870170685978e-06,
      "loss": 0.0309,
      "step": 519
    },
    {
      "epoch": 0.11831626848691695,
      "grad_norm": 2.3408918893379176,
      "learning_rate": 1.2482803998883122e-06,
      "loss": 0.0554,
      "step": 520
    },
    {
      "epoch": 0.11854379977246872,
      "grad_norm": 1.308272654811059,
      "learning_rate": 1.2482737699693168e-06,
      "loss": 0.0386,
      "step": 521
    },
    {
      "epoch": 0.11877133105802047,
      "grad_norm": 1.4643677550086747,
      "learning_rate": 1.248267127311747e-06,
      "loss": 0.0517,
      "step": 522
    },
    {
      "epoch": 0.11899886234357224,
      "grad_norm": 1.9375531511223079,
      "learning_rate": 1.2482604719157386e-06,
      "loss": 0.0547,
      "step": 523
    },
    {
      "epoch": 0.119226393629124,
      "grad_norm": 1.4368931736724269,
      "learning_rate": 1.2482538037814277e-06,
      "loss": 0.0594,
      "step": 524
    },
    {
      "epoch": 0.11945392491467577,
      "grad_norm": 2.3983593700848864,
      "learning_rate": 1.2482471229089502e-06,
      "loss": 0.0479,
      "step": 525
    },
    {
      "epoch": 0.11968145620022753,
      "grad_norm": 1.5406589025382933,
      "learning_rate": 1.2482404292984431e-06,
      "loss": 0.0468,
      "step": 526
    },
    {
      "epoch": 0.1199089874857793,
      "grad_norm": 2.7569107151011254,
      "learning_rate": 1.248233722950043e-06,
      "loss": 0.1653,
      "step": 527
    },
    {
      "epoch": 0.12013651877133105,
      "grad_norm": 1.2498018017145271,
      "learning_rate": 1.2482270038638872e-06,
      "loss": 0.0376,
      "step": 528
    },
    {
      "epoch": 0.12036405005688282,
      "grad_norm": 1.663550002347151,
      "learning_rate": 1.2482202720401128e-06,
      "loss": 0.0336,
      "step": 529
    },
    {
      "epoch": 0.12059158134243458,
      "grad_norm": 1.302350240596974,
      "learning_rate": 1.248213527478857e-06,
      "loss": 0.058,
      "step": 530
    },
    {
      "epoch": 0.12081911262798635,
      "grad_norm": 1.6402862563370133,
      "learning_rate": 1.2482067701802583e-06,
      "loss": 0.0826,
      "step": 531
    },
    {
      "epoch": 0.12104664391353812,
      "grad_norm": 1.1268914835455415,
      "learning_rate": 1.2482000001444547e-06,
      "loss": 0.0524,
      "step": 532
    },
    {
      "epoch": 0.12127417519908988,
      "grad_norm": 1.877525140915591,
      "learning_rate": 1.2481932173715845e-06,
      "loss": 0.0539,
      "step": 533
    },
    {
      "epoch": 0.12150170648464163,
      "grad_norm": 1.928222388008525,
      "learning_rate": 1.2481864218617859e-06,
      "loss": 0.1134,
      "step": 534
    },
    {
      "epoch": 0.1217292377701934,
      "grad_norm": 1.2293579919391775,
      "learning_rate": 1.2481796136151984e-06,
      "loss": 0.0516,
      "step": 535
    },
    {
      "epoch": 0.12195676905574517,
      "grad_norm": 1.5890038701436435,
      "learning_rate": 1.2481727926319609e-06,
      "loss": 0.0922,
      "step": 536
    },
    {
      "epoch": 0.12218430034129693,
      "grad_norm": 1.850040007143475,
      "learning_rate": 1.2481659589122127e-06,
      "loss": 0.075,
      "step": 537
    },
    {
      "epoch": 0.1224118316268487,
      "grad_norm": 2.5725083852053747,
      "learning_rate": 1.2481591124560934e-06,
      "loss": 0.0891,
      "step": 538
    },
    {
      "epoch": 0.12263936291240045,
      "grad_norm": 1.768076465759339,
      "learning_rate": 1.2481522532637435e-06,
      "loss": 0.0533,
      "step": 539
    },
    {
      "epoch": 0.12286689419795221,
      "grad_norm": 0.9070123837857468,
      "learning_rate": 1.2481453813353026e-06,
      "loss": 0.0294,
      "step": 540
    },
    {
      "epoch": 0.12309442548350398,
      "grad_norm": 1.625916805043753,
      "learning_rate": 1.2481384966709116e-06,
      "loss": 0.0323,
      "step": 541
    },
    {
      "epoch": 0.12332195676905575,
      "grad_norm": 2.291411200127928,
      "learning_rate": 1.2481315992707104e-06,
      "loss": 0.0964,
      "step": 542
    },
    {
      "epoch": 0.12354948805460751,
      "grad_norm": 1.7452732363406978,
      "learning_rate": 1.248124689134841e-06,
      "loss": 0.0555,
      "step": 543
    },
    {
      "epoch": 0.12377701934015928,
      "grad_norm": 1.4048824029212816,
      "learning_rate": 1.2481177662634438e-06,
      "loss": 0.0551,
      "step": 544
    },
    {
      "epoch": 0.12400455062571103,
      "grad_norm": 6.9359786939510695,
      "learning_rate": 1.2481108306566609e-06,
      "loss": 0.0746,
      "step": 545
    },
    {
      "epoch": 0.1242320819112628,
      "grad_norm": 1.2916063155011999,
      "learning_rate": 1.2481038823146338e-06,
      "loss": 0.0245,
      "step": 546
    },
    {
      "epoch": 0.12445961319681456,
      "grad_norm": 1.9432907622995745,
      "learning_rate": 1.2480969212375043e-06,
      "loss": 0.1119,
      "step": 547
    },
    {
      "epoch": 0.12468714448236633,
      "grad_norm": 2.9617568197379347,
      "learning_rate": 1.2480899474254151e-06,
      "loss": 0.0878,
      "step": 548
    },
    {
      "epoch": 0.12491467576791809,
      "grad_norm": 0.9954987625211155,
      "learning_rate": 1.2480829608785085e-06,
      "loss": 0.0397,
      "step": 549
    },
    {
      "epoch": 0.12514220705346984,
      "grad_norm": 1.2186721476872615,
      "learning_rate": 1.2480759615969273e-06,
      "loss": 0.0462,
      "step": 550
    },
    {
      "epoch": 0.12536973833902162,
      "grad_norm": 1.1723077588981092,
      "learning_rate": 1.2480689495808144e-06,
      "loss": 0.0646,
      "step": 551
    },
    {
      "epoch": 0.12559726962457338,
      "grad_norm": 1.4338215761686757,
      "learning_rate": 1.2480619248303133e-06,
      "loss": 0.0806,
      "step": 552
    },
    {
      "epoch": 0.12582480091012513,
      "grad_norm": 1.841000108666981,
      "learning_rate": 1.2480548873455675e-06,
      "loss": 0.0827,
      "step": 553
    },
    {
      "epoch": 0.1260523321956769,
      "grad_norm": 2.5765107503856144,
      "learning_rate": 1.248047837126721e-06,
      "loss": 0.1326,
      "step": 554
    },
    {
      "epoch": 0.12627986348122866,
      "grad_norm": 1.3867433794433537,
      "learning_rate": 1.248040774173918e-06,
      "loss": 0.0588,
      "step": 555
    },
    {
      "epoch": 0.12650739476678044,
      "grad_norm": 1.3761837680071665,
      "learning_rate": 1.248033698487302e-06,
      "loss": 0.0495,
      "step": 556
    },
    {
      "epoch": 0.1267349260523322,
      "grad_norm": 1.1704965217659915,
      "learning_rate": 1.2480266100670189e-06,
      "loss": 0.0467,
      "step": 557
    },
    {
      "epoch": 0.12696245733788397,
      "grad_norm": 2.106768640375489,
      "learning_rate": 1.2480195089132125e-06,
      "loss": 0.0426,
      "step": 558
    },
    {
      "epoch": 0.12718998862343572,
      "grad_norm": 1.1575398708268563,
      "learning_rate": 1.2480123950260284e-06,
      "loss": 0.0413,
      "step": 559
    },
    {
      "epoch": 0.12741751990898748,
      "grad_norm": 9.831393771519632,
      "learning_rate": 1.248005268405612e-06,
      "loss": 0.0434,
      "step": 560
    },
    {
      "epoch": 0.12764505119453926,
      "grad_norm": 1.0820393760670333,
      "learning_rate": 1.2479981290521087e-06,
      "loss": 0.034,
      "step": 561
    },
    {
      "epoch": 0.127872582480091,
      "grad_norm": 2.206902342168789,
      "learning_rate": 1.2479909769656648e-06,
      "loss": 0.073,
      "step": 562
    },
    {
      "epoch": 0.1281001137656428,
      "grad_norm": 1.5027865833496339,
      "learning_rate": 1.2479838121464263e-06,
      "loss": 0.0744,
      "step": 563
    },
    {
      "epoch": 0.12832764505119454,
      "grad_norm": 2.001324966678472,
      "learning_rate": 1.2479766345945395e-06,
      "loss": 0.0457,
      "step": 564
    },
    {
      "epoch": 0.1285551763367463,
      "grad_norm": 0.9498156004808115,
      "learning_rate": 1.2479694443101513e-06,
      "loss": 0.0442,
      "step": 565
    },
    {
      "epoch": 0.12878270762229807,
      "grad_norm": 1.427402300347758,
      "learning_rate": 1.2479622412934087e-06,
      "loss": 0.0774,
      "step": 566
    },
    {
      "epoch": 0.12901023890784982,
      "grad_norm": 1.338474573791641,
      "learning_rate": 1.2479550255444586e-06,
      "loss": 0.0498,
      "step": 567
    },
    {
      "epoch": 0.1292377701934016,
      "grad_norm": 1.0332062154173083,
      "learning_rate": 1.2479477970634487e-06,
      "loss": 0.0575,
      "step": 568
    },
    {
      "epoch": 0.12946530147895335,
      "grad_norm": 2.342085432201913,
      "learning_rate": 1.2479405558505267e-06,
      "loss": 0.0776,
      "step": 569
    },
    {
      "epoch": 0.1296928327645051,
      "grad_norm": 1.7614392055133243,
      "learning_rate": 1.247933301905841e-06,
      "loss": 0.0626,
      "step": 570
    },
    {
      "epoch": 0.12992036405005689,
      "grad_norm": 1.401551595719724,
      "learning_rate": 1.2479260352295388e-06,
      "loss": 0.0354,
      "step": 571
    },
    {
      "epoch": 0.13014789533560864,
      "grad_norm": 1.4755845320165697,
      "learning_rate": 1.2479187558217697e-06,
      "loss": 0.0407,
      "step": 572
    },
    {
      "epoch": 0.13037542662116042,
      "grad_norm": 0.7585780841566192,
      "learning_rate": 1.247911463682682e-06,
      "loss": 0.0376,
      "step": 573
    },
    {
      "epoch": 0.13060295790671217,
      "grad_norm": 2.2124015942087496,
      "learning_rate": 1.2479041588124247e-06,
      "loss": 0.0537,
      "step": 574
    },
    {
      "epoch": 0.13083048919226395,
      "grad_norm": 1.0766166316852028,
      "learning_rate": 1.2478968412111471e-06,
      "loss": 0.0292,
      "step": 575
    },
    {
      "epoch": 0.1310580204778157,
      "grad_norm": 0.995096761634457,
      "learning_rate": 1.247889510878999e-06,
      "loss": 0.0523,
      "step": 576
    },
    {
      "epoch": 0.13128555176336745,
      "grad_norm": 1.6016154355497043,
      "learning_rate": 1.24788216781613e-06,
      "loss": 0.0634,
      "step": 577
    },
    {
      "epoch": 0.13151308304891923,
      "grad_norm": 1.1448425600800458,
      "learning_rate": 1.2478748120226902e-06,
      "loss": 0.0449,
      "step": 578
    },
    {
      "epoch": 0.13174061433447098,
      "grad_norm": 1.1092744505453933,
      "learning_rate": 1.2478674434988299e-06,
      "loss": 0.0377,
      "step": 579
    },
    {
      "epoch": 0.13196814562002276,
      "grad_norm": 0.8431904554396206,
      "learning_rate": 1.2478600622447001e-06,
      "loss": 0.0404,
      "step": 580
    },
    {
      "epoch": 0.13219567690557452,
      "grad_norm": 1.1168682097019669,
      "learning_rate": 1.2478526682604512e-06,
      "loss": 0.0324,
      "step": 581
    },
    {
      "epoch": 0.13242320819112627,
      "grad_norm": 1.3377796205096442,
      "learning_rate": 1.2478452615462345e-06,
      "loss": 0.0663,
      "step": 582
    },
    {
      "epoch": 0.13265073947667805,
      "grad_norm": 3.2228218888338667,
      "learning_rate": 1.247837842102201e-06,
      "loss": 0.1264,
      "step": 583
    },
    {
      "epoch": 0.1328782707622298,
      "grad_norm": 1.5298131963208237,
      "learning_rate": 1.2478304099285031e-06,
      "loss": 0.0525,
      "step": 584
    },
    {
      "epoch": 0.13310580204778158,
      "grad_norm": 1.1458963041110781,
      "learning_rate": 1.2478229650252921e-06,
      "loss": 0.0464,
      "step": 585
    },
    {
      "epoch": 0.13333333333333333,
      "grad_norm": 1.7278138919558592,
      "learning_rate": 1.2478155073927204e-06,
      "loss": 0.0556,
      "step": 586
    },
    {
      "epoch": 0.13356086461888508,
      "grad_norm": 1.2475569782092948,
      "learning_rate": 1.2478080370309404e-06,
      "loss": 0.0428,
      "step": 587
    },
    {
      "epoch": 0.13378839590443686,
      "grad_norm": 1.088342211454506,
      "learning_rate": 1.2478005539401046e-06,
      "loss": 0.0295,
      "step": 588
    },
    {
      "epoch": 0.13401592718998862,
      "grad_norm": 1.7587707539606798,
      "learning_rate": 1.2477930581203663e-06,
      "loss": 0.0442,
      "step": 589
    },
    {
      "epoch": 0.1342434584755404,
      "grad_norm": 1.1629708095404674,
      "learning_rate": 1.2477855495718782e-06,
      "loss": 0.0548,
      "step": 590
    },
    {
      "epoch": 0.13447098976109215,
      "grad_norm": 1.551063066789381,
      "learning_rate": 1.2477780282947942e-06,
      "loss": 0.037,
      "step": 591
    },
    {
      "epoch": 0.13469852104664393,
      "grad_norm": 1.0301289542876222,
      "learning_rate": 1.2477704942892677e-06,
      "loss": 0.0443,
      "step": 592
    },
    {
      "epoch": 0.13492605233219568,
      "grad_norm": 3.67842928304707,
      "learning_rate": 1.2477629475554532e-06,
      "loss": 0.0909,
      "step": 593
    },
    {
      "epoch": 0.13515358361774743,
      "grad_norm": 1.6590539267977245,
      "learning_rate": 1.2477553880935043e-06,
      "loss": 0.0523,
      "step": 594
    },
    {
      "epoch": 0.1353811149032992,
      "grad_norm": 1.9972940081293025,
      "learning_rate": 1.2477478159035758e-06,
      "loss": 0.0745,
      "step": 595
    },
    {
      "epoch": 0.13560864618885096,
      "grad_norm": 2.5034292233621693,
      "learning_rate": 1.2477402309858226e-06,
      "loss": 0.0731,
      "step": 596
    },
    {
      "epoch": 0.13583617747440274,
      "grad_norm": 1.7292171438267574,
      "learning_rate": 1.2477326333403995e-06,
      "loss": 0.0947,
      "step": 597
    },
    {
      "epoch": 0.1360637087599545,
      "grad_norm": 1.0893865478354072,
      "learning_rate": 1.2477250229674618e-06,
      "loss": 0.036,
      "step": 598
    },
    {
      "epoch": 0.13629124004550625,
      "grad_norm": 2.0220916101870854,
      "learning_rate": 1.2477173998671653e-06,
      "loss": 0.0695,
      "step": 599
    },
    {
      "epoch": 0.13651877133105803,
      "grad_norm": 1.6663966081636725,
      "learning_rate": 1.2477097640396655e-06,
      "loss": 0.0239,
      "step": 600
    },
    {
      "epoch": 0.13674630261660978,
      "grad_norm": 2.160820824239015,
      "learning_rate": 1.2477021154851185e-06,
      "loss": 0.041,
      "step": 601
    },
    {
      "epoch": 0.13697383390216156,
      "grad_norm": 1.7560963022061893,
      "learning_rate": 1.2476944542036806e-06,
      "loss": 0.0646,
      "step": 602
    },
    {
      "epoch": 0.1372013651877133,
      "grad_norm": 1.0045761668266382,
      "learning_rate": 1.2476867801955086e-06,
      "loss": 0.0319,
      "step": 603
    },
    {
      "epoch": 0.13742889647326506,
      "grad_norm": 1.1722883805586992,
      "learning_rate": 1.247679093460759e-06,
      "loss": 0.0421,
      "step": 604
    },
    {
      "epoch": 0.13765642775881684,
      "grad_norm": 1.5520795851774112,
      "learning_rate": 1.2476713939995895e-06,
      "loss": 0.0545,
      "step": 605
    },
    {
      "epoch": 0.1378839590443686,
      "grad_norm": 1.5341083321869782,
      "learning_rate": 1.2476636818121568e-06,
      "loss": 0.0587,
      "step": 606
    },
    {
      "epoch": 0.13811149032992037,
      "grad_norm": 1.427789001931161,
      "learning_rate": 1.247655956898619e-06,
      "loss": 0.0783,
      "step": 607
    },
    {
      "epoch": 0.13833902161547212,
      "grad_norm": 1.6612253896109521,
      "learning_rate": 1.2476482192591335e-06,
      "loss": 0.0454,
      "step": 608
    },
    {
      "epoch": 0.1385665529010239,
      "grad_norm": 4.06982296073183,
      "learning_rate": 1.247640468893859e-06,
      "loss": 0.0518,
      "step": 609
    },
    {
      "epoch": 0.13879408418657566,
      "grad_norm": 0.9984415328300282,
      "learning_rate": 1.2476327058029534e-06,
      "loss": 0.0286,
      "step": 610
    },
    {
      "epoch": 0.1390216154721274,
      "grad_norm": 1.5505955919902168,
      "learning_rate": 1.2476249299865757e-06,
      "loss": 0.0679,
      "step": 611
    },
    {
      "epoch": 0.1392491467576792,
      "grad_norm": 1.1777984076454868,
      "learning_rate": 1.2476171414448847e-06,
      "loss": 0.0333,
      "step": 612
    },
    {
      "epoch": 0.13947667804323094,
      "grad_norm": 1.5134654111327426,
      "learning_rate": 1.2476093401780397e-06,
      "loss": 0.0464,
      "step": 613
    },
    {
      "epoch": 0.13970420932878272,
      "grad_norm": 8.868555700718426,
      "learning_rate": 1.2476015261861998e-06,
      "loss": 0.0588,
      "step": 614
    },
    {
      "epoch": 0.13993174061433447,
      "grad_norm": 1.07185240181413,
      "learning_rate": 1.247593699469525e-06,
      "loss": 0.0508,
      "step": 615
    },
    {
      "epoch": 0.14015927189988622,
      "grad_norm": 1.0467231615661654,
      "learning_rate": 1.2475858600281754e-06,
      "loss": 0.0375,
      "step": 616
    },
    {
      "epoch": 0.140386803185438,
      "grad_norm": 1.6914112039280367,
      "learning_rate": 1.247578007862311e-06,
      "loss": 0.0385,
      "step": 617
    },
    {
      "epoch": 0.14061433447098975,
      "grad_norm": 1.6162335462442214,
      "learning_rate": 1.2475701429720923e-06,
      "loss": 0.0617,
      "step": 618
    },
    {
      "epoch": 0.14084186575654153,
      "grad_norm": 0.8727030068232594,
      "learning_rate": 1.24756226535768e-06,
      "loss": 0.0388,
      "step": 619
    },
    {
      "epoch": 0.1410693970420933,
      "grad_norm": 1.5566079211643449,
      "learning_rate": 1.2475543750192352e-06,
      "loss": 0.094,
      "step": 620
    },
    {
      "epoch": 0.14129692832764504,
      "grad_norm": 1.1473885930159387,
      "learning_rate": 1.2475464719569192e-06,
      "loss": 0.0563,
      "step": 621
    },
    {
      "epoch": 0.14152445961319682,
      "grad_norm": 1.111358042304426,
      "learning_rate": 1.2475385561708934e-06,
      "loss": 0.058,
      "step": 622
    },
    {
      "epoch": 0.14175199089874857,
      "grad_norm": 0.8961189135726134,
      "learning_rate": 1.2475306276613194e-06,
      "loss": 0.0393,
      "step": 623
    },
    {
      "epoch": 0.14197952218430035,
      "grad_norm": 3.442046327889576,
      "learning_rate": 1.2475226864283596e-06,
      "loss": 0.0713,
      "step": 624
    },
    {
      "epoch": 0.1422070534698521,
      "grad_norm": 4.150666880510046,
      "learning_rate": 1.2475147324721764e-06,
      "loss": 0.0442,
      "step": 625
    },
    {
      "epoch": 0.14243458475540388,
      "grad_norm": 0.9296406182877743,
      "learning_rate": 1.2475067657929319e-06,
      "loss": 0.0414,
      "step": 626
    },
    {
      "epoch": 0.14266211604095563,
      "grad_norm": 1.0829545070929647,
      "learning_rate": 1.2474987863907894e-06,
      "loss": 0.0451,
      "step": 627
    },
    {
      "epoch": 0.14288964732650739,
      "grad_norm": 1.8334147294224943,
      "learning_rate": 1.2474907942659116e-06,
      "loss": 0.0771,
      "step": 628
    },
    {
      "epoch": 0.14311717861205916,
      "grad_norm": 1.7815672754482244,
      "learning_rate": 1.247482789418462e-06,
      "loss": 0.0897,
      "step": 629
    },
    {
      "epoch": 0.14334470989761092,
      "grad_norm": 1.3733903816478978,
      "learning_rate": 1.2474747718486044e-06,
      "loss": 0.0358,
      "step": 630
    },
    {
      "epoch": 0.1435722411831627,
      "grad_norm": 2.28709865860884,
      "learning_rate": 1.2474667415565022e-06,
      "loss": 0.0652,
      "step": 631
    },
    {
      "epoch": 0.14379977246871445,
      "grad_norm": 1.231780324099585,
      "learning_rate": 1.24745869854232e-06,
      "loss": 0.0425,
      "step": 632
    },
    {
      "epoch": 0.1440273037542662,
      "grad_norm": 1.335370738840473,
      "learning_rate": 1.2474506428062219e-06,
      "loss": 0.0422,
      "step": 633
    },
    {
      "epoch": 0.14425483503981798,
      "grad_norm": 2.966338619611093,
      "learning_rate": 1.2474425743483726e-06,
      "loss": 0.1214,
      "step": 634
    },
    {
      "epoch": 0.14448236632536973,
      "grad_norm": 0.8989763932220884,
      "learning_rate": 1.2474344931689371e-06,
      "loss": 0.0341,
      "step": 635
    },
    {
      "epoch": 0.1447098976109215,
      "grad_norm": 0.937533756927628,
      "learning_rate": 1.2474263992680805e-06,
      "loss": 0.0394,
      "step": 636
    },
    {
      "epoch": 0.14493742889647326,
      "grad_norm": 1.8912948524392619,
      "learning_rate": 1.247418292645968e-06,
      "loss": 0.058,
      "step": 637
    },
    {
      "epoch": 0.14516496018202502,
      "grad_norm": 2.6567132856925877,
      "learning_rate": 1.2474101733027659e-06,
      "loss": 0.1061,
      "step": 638
    },
    {
      "epoch": 0.1453924914675768,
      "grad_norm": 1.0544164694645533,
      "learning_rate": 1.2474020412386395e-06,
      "loss": 0.0327,
      "step": 639
    },
    {
      "epoch": 0.14562002275312855,
      "grad_norm": 0.8862493084977525,
      "learning_rate": 1.2473938964537551e-06,
      "loss": 0.0337,
      "step": 640
    },
    {
      "epoch": 0.14584755403868033,
      "grad_norm": 1.1660786205758522,
      "learning_rate": 1.2473857389482797e-06,
      "loss": 0.0379,
      "step": 641
    },
    {
      "epoch": 0.14607508532423208,
      "grad_norm": 1.275063352619887,
      "learning_rate": 1.2473775687223794e-06,
      "loss": 0.0677,
      "step": 642
    },
    {
      "epoch": 0.14630261660978386,
      "grad_norm": 1.033368353907448,
      "learning_rate": 1.2473693857762215e-06,
      "loss": 0.0499,
      "step": 643
    },
    {
      "epoch": 0.1465301478953356,
      "grad_norm": 0.8176955754401027,
      "learning_rate": 1.247361190109973e-06,
      "loss": 0.028,
      "step": 644
    },
    {
      "epoch": 0.14675767918088736,
      "grad_norm": 1.5868301680057004,
      "learning_rate": 1.2473529817238016e-06,
      "loss": 0.0758,
      "step": 645
    },
    {
      "epoch": 0.14698521046643914,
      "grad_norm": 1.1116607935229232,
      "learning_rate": 1.2473447606178754e-06,
      "loss": 0.0464,
      "step": 646
    },
    {
      "epoch": 0.1472127417519909,
      "grad_norm": 1.3737040045295303,
      "learning_rate": 1.2473365267923617e-06,
      "loss": 0.038,
      "step": 647
    },
    {
      "epoch": 0.14744027303754267,
      "grad_norm": 1.6851756106039961,
      "learning_rate": 1.2473282802474293e-06,
      "loss": 0.0521,
      "step": 648
    },
    {
      "epoch": 0.14766780432309443,
      "grad_norm": 1.067298154630741,
      "learning_rate": 1.2473200209832465e-06,
      "loss": 0.0339,
      "step": 649
    },
    {
      "epoch": 0.14789533560864618,
      "grad_norm": 3.389712283735002,
      "learning_rate": 1.2473117489999823e-06,
      "loss": 0.1032,
      "step": 650
    },
    {
      "epoch": 0.14812286689419796,
      "grad_norm": 1.6568457682006943,
      "learning_rate": 1.2473034642978057e-06,
      "loss": 0.0619,
      "step": 651
    },
    {
      "epoch": 0.1483503981797497,
      "grad_norm": 1.490856549646787,
      "learning_rate": 1.247295166876886e-06,
      "loss": 0.0516,
      "step": 652
    },
    {
      "epoch": 0.1485779294653015,
      "grad_norm": 1.1851614081448008,
      "learning_rate": 1.2472868567373924e-06,
      "loss": 0.0431,
      "step": 653
    },
    {
      "epoch": 0.14880546075085324,
      "grad_norm": 1.644166757760147,
      "learning_rate": 1.2472785338794953e-06,
      "loss": 0.082,
      "step": 654
    },
    {
      "epoch": 0.149032992036405,
      "grad_norm": 1.556466054889254,
      "learning_rate": 1.247270198303365e-06,
      "loss": 0.0573,
      "step": 655
    },
    {
      "epoch": 0.14926052332195677,
      "grad_norm": 2.233896039541245,
      "learning_rate": 1.247261850009171e-06,
      "loss": 0.1073,
      "step": 656
    },
    {
      "epoch": 0.14948805460750852,
      "grad_norm": 1.5970780316095814,
      "learning_rate": 1.2472534889970848e-06,
      "loss": 0.0741,
      "step": 657
    },
    {
      "epoch": 0.1497155858930603,
      "grad_norm": 1.2400037573314944,
      "learning_rate": 1.2472451152672766e-06,
      "loss": 0.0483,
      "step": 658
    },
    {
      "epoch": 0.14994311717861206,
      "grad_norm": 0.7241628580276863,
      "learning_rate": 1.2472367288199177e-06,
      "loss": 0.0294,
      "step": 659
    },
    {
      "epoch": 0.15017064846416384,
      "grad_norm": 1.7234352219783362,
      "learning_rate": 1.2472283296551798e-06,
      "loss": 0.0462,
      "step": 660
    },
    {
      "epoch": 0.1503981797497156,
      "grad_norm": 1.4734355231360154,
      "learning_rate": 1.2472199177732346e-06,
      "loss": 0.0582,
      "step": 661
    },
    {
      "epoch": 0.15062571103526734,
      "grad_norm": 2.1378423903625317,
      "learning_rate": 1.2472114931742537e-06,
      "loss": 0.0584,
      "step": 662
    },
    {
      "epoch": 0.15085324232081912,
      "grad_norm": 10.24253063417928,
      "learning_rate": 1.2472030558584093e-06,
      "loss": 0.0477,
      "step": 663
    },
    {
      "epoch": 0.15108077360637087,
      "grad_norm": 1.8691143785999473,
      "learning_rate": 1.2471946058258742e-06,
      "loss": 0.0739,
      "step": 664
    },
    {
      "epoch": 0.15130830489192265,
      "grad_norm": 2.241077277544209,
      "learning_rate": 1.2471861430768205e-06,
      "loss": 0.0672,
      "step": 665
    },
    {
      "epoch": 0.1515358361774744,
      "grad_norm": 2.2720454211658385,
      "learning_rate": 1.2471776676114217e-06,
      "loss": 0.0744,
      "step": 666
    },
    {
      "epoch": 0.15176336746302616,
      "grad_norm": 1.5599655593025843,
      "learning_rate": 1.2471691794298508e-06,
      "loss": 0.0708,
      "step": 667
    },
    {
      "epoch": 0.15199089874857794,
      "grad_norm": 1.2105729421418516,
      "learning_rate": 1.2471606785322814e-06,
      "loss": 0.0408,
      "step": 668
    },
    {
      "epoch": 0.1522184300341297,
      "grad_norm": 1.2672422777930856,
      "learning_rate": 1.247152164918887e-06,
      "loss": 0.0406,
      "step": 669
    },
    {
      "epoch": 0.15244596131968147,
      "grad_norm": 1.4429003913192175,
      "learning_rate": 1.247143638589842e-06,
      "loss": 0.0537,
      "step": 670
    },
    {
      "epoch": 0.15267349260523322,
      "grad_norm": 2.4501715529962667,
      "learning_rate": 1.2471350995453203e-06,
      "loss": 0.0658,
      "step": 671
    },
    {
      "epoch": 0.15290102389078497,
      "grad_norm": 1.48398143402129,
      "learning_rate": 1.2471265477854966e-06,
      "loss": 0.0556,
      "step": 672
    },
    {
      "epoch": 0.15312855517633675,
      "grad_norm": 2.1744278885096278,
      "learning_rate": 1.2471179833105454e-06,
      "loss": 0.0404,
      "step": 673
    },
    {
      "epoch": 0.1533560864618885,
      "grad_norm": 6.901371774679421,
      "learning_rate": 1.2471094061206422e-06,
      "loss": 0.0869,
      "step": 674
    },
    {
      "epoch": 0.15358361774744028,
      "grad_norm": 1.1601432497447364,
      "learning_rate": 1.247100816215962e-06,
      "loss": 0.0379,
      "step": 675
    },
    {
      "epoch": 0.15381114903299203,
      "grad_norm": 1.2820824697846318,
      "learning_rate": 1.2470922135966806e-06,
      "loss": 0.032,
      "step": 676
    },
    {
      "epoch": 0.1540386803185438,
      "grad_norm": 1.2705594517037477,
      "learning_rate": 1.2470835982629736e-06,
      "loss": 0.0494,
      "step": 677
    },
    {
      "epoch": 0.15426621160409557,
      "grad_norm": 1.5982674187017603,
      "learning_rate": 1.247074970215017e-06,
      "loss": 0.0495,
      "step": 678
    },
    {
      "epoch": 0.15449374288964732,
      "grad_norm": 1.7109186334283122,
      "learning_rate": 1.2470663294529873e-06,
      "loss": 0.0414,
      "step": 679
    },
    {
      "epoch": 0.1547212741751991,
      "grad_norm": 1.0715248913545208,
      "learning_rate": 1.2470576759770612e-06,
      "loss": 0.0325,
      "step": 680
    },
    {
      "epoch": 0.15494880546075085,
      "grad_norm": 1.824156765255701,
      "learning_rate": 1.2470490097874155e-06,
      "loss": 0.0466,
      "step": 681
    },
    {
      "epoch": 0.15517633674630263,
      "grad_norm": 2.4373458213618897,
      "learning_rate": 1.247040330884227e-06,
      "loss": 0.0539,
      "step": 682
    },
    {
      "epoch": 0.15540386803185438,
      "grad_norm": 1.0065301428430162,
      "learning_rate": 1.2470316392676738e-06,
      "loss": 0.0525,
      "step": 683
    },
    {
      "epoch": 0.15563139931740613,
      "grad_norm": 1.5815427583099921,
      "learning_rate": 1.2470229349379326e-06,
      "loss": 0.0519,
      "step": 684
    },
    {
      "epoch": 0.1558589306029579,
      "grad_norm": 1.9652039642446397,
      "learning_rate": 1.2470142178951822e-06,
      "loss": 0.0768,
      "step": 685
    },
    {
      "epoch": 0.15608646188850966,
      "grad_norm": 1.6149386041596687,
      "learning_rate": 1.2470054881396002e-06,
      "loss": 0.0832,
      "step": 686
    },
    {
      "epoch": 0.15631399317406144,
      "grad_norm": 1.1987854832655134,
      "learning_rate": 1.246996745671365e-06,
      "loss": 0.0659,
      "step": 687
    },
    {
      "epoch": 0.1565415244596132,
      "grad_norm": 1.3921581311290316,
      "learning_rate": 1.2469879904906556e-06,
      "loss": 0.0477,
      "step": 688
    },
    {
      "epoch": 0.15676905574516495,
      "grad_norm": 1.5291818763479217,
      "learning_rate": 1.2469792225976507e-06,
      "loss": 0.0597,
      "step": 689
    },
    {
      "epoch": 0.15699658703071673,
      "grad_norm": 1.1978579028524097,
      "learning_rate": 1.2469704419925296e-06,
      "loss": 0.0551,
      "step": 690
    },
    {
      "epoch": 0.15722411831626848,
      "grad_norm": 2.2562422781402827,
      "learning_rate": 1.246961648675472e-06,
      "loss": 0.0709,
      "step": 691
    },
    {
      "epoch": 0.15745164960182026,
      "grad_norm": 1.947406282057691,
      "learning_rate": 1.246952842646657e-06,
      "loss": 0.115,
      "step": 692
    },
    {
      "epoch": 0.157679180887372,
      "grad_norm": 1.2450058586298152,
      "learning_rate": 1.2469440239062653e-06,
      "loss": 0.0499,
      "step": 693
    },
    {
      "epoch": 0.15790671217292376,
      "grad_norm": 1.8197388877884058,
      "learning_rate": 1.2469351924544766e-06,
      "loss": 0.0927,
      "step": 694
    },
    {
      "epoch": 0.15813424345847554,
      "grad_norm": 1.904635209687238,
      "learning_rate": 1.2469263482914716e-06,
      "loss": 0.0814,
      "step": 695
    },
    {
      "epoch": 0.1583617747440273,
      "grad_norm": 1.4011214554570595,
      "learning_rate": 1.246917491417431e-06,
      "loss": 0.0585,
      "step": 696
    },
    {
      "epoch": 0.15858930602957907,
      "grad_norm": 1.9445852493644915,
      "learning_rate": 1.246908621832536e-06,
      "loss": 0.1005,
      "step": 697
    },
    {
      "epoch": 0.15881683731513083,
      "grad_norm": 1.363848979307397,
      "learning_rate": 1.2468997395369677e-06,
      "loss": 0.0717,
      "step": 698
    },
    {
      "epoch": 0.1590443686006826,
      "grad_norm": 1.3903385726286603,
      "learning_rate": 1.2468908445309077e-06,
      "loss": 0.0557,
      "step": 699
    },
    {
      "epoch": 0.15927189988623436,
      "grad_norm": 0.8725919788890233,
      "learning_rate": 1.2468819368145376e-06,
      "loss": 0.0365,
      "step": 700
    },
    {
      "epoch": 0.1594994311717861,
      "grad_norm": 1.33354116221911,
      "learning_rate": 1.2468730163880398e-06,
      "loss": 0.0486,
      "step": 701
    },
    {
      "epoch": 0.1597269624573379,
      "grad_norm": 4.544549924474667,
      "learning_rate": 1.2468640832515962e-06,
      "loss": 0.0302,
      "step": 702
    },
    {
      "epoch": 0.15995449374288964,
      "grad_norm": 2.9489517825532308,
      "learning_rate": 1.24685513740539e-06,
      "loss": 0.0904,
      "step": 703
    },
    {
      "epoch": 0.16018202502844142,
      "grad_norm": 1.3714756073146785,
      "learning_rate": 1.2468461788496036e-06,
      "loss": 0.056,
      "step": 704
    },
    {
      "epoch": 0.16040955631399317,
      "grad_norm": 1.9329458786445137,
      "learning_rate": 1.24683720758442e-06,
      "loss": 0.0848,
      "step": 705
    },
    {
      "epoch": 0.16063708759954493,
      "grad_norm": 1.0521231330897036,
      "learning_rate": 1.2468282236100226e-06,
      "loss": 0.0438,
      "step": 706
    },
    {
      "epoch": 0.1608646188850967,
      "grad_norm": 1.7554793724577804,
      "learning_rate": 1.2468192269265955e-06,
      "loss": 0.1102,
      "step": 707
    },
    {
      "epoch": 0.16109215017064846,
      "grad_norm": 1.0071708604696206,
      "learning_rate": 1.246810217534322e-06,
      "loss": 0.0266,
      "step": 708
    },
    {
      "epoch": 0.16131968145620024,
      "grad_norm": 1.3334260212411804,
      "learning_rate": 1.2468011954333864e-06,
      "loss": 0.0609,
      "step": 709
    },
    {
      "epoch": 0.161547212741752,
      "grad_norm": 1.1880594144178185,
      "learning_rate": 1.2467921606239734e-06,
      "loss": 0.0465,
      "step": 710
    },
    {
      "epoch": 0.16177474402730374,
      "grad_norm": 2.7036981936596542,
      "learning_rate": 1.2467831131062672e-06,
      "loss": 0.0516,
      "step": 711
    },
    {
      "epoch": 0.16200227531285552,
      "grad_norm": 1.416736687413605,
      "learning_rate": 1.2467740528804528e-06,
      "loss": 0.0552,
      "step": 712
    },
    {
      "epoch": 0.16222980659840727,
      "grad_norm": 1.3251728729578673,
      "learning_rate": 1.2467649799467156e-06,
      "loss": 0.0615,
      "step": 713
    },
    {
      "epoch": 0.16245733788395905,
      "grad_norm": 0.9167577669577527,
      "learning_rate": 1.246755894305241e-06,
      "loss": 0.0312,
      "step": 714
    },
    {
      "epoch": 0.1626848691695108,
      "grad_norm": 1.4735705097484915,
      "learning_rate": 1.2467467959562143e-06,
      "loss": 0.0676,
      "step": 715
    },
    {
      "epoch": 0.16291240045506258,
      "grad_norm": 1.6890201461846701,
      "learning_rate": 1.2467376848998221e-06,
      "loss": 0.0597,
      "step": 716
    },
    {
      "epoch": 0.16313993174061434,
      "grad_norm": 1.4659558773380117,
      "learning_rate": 1.2467285611362501e-06,
      "loss": 0.052,
      "step": 717
    },
    {
      "epoch": 0.1633674630261661,
      "grad_norm": 1.7717923581360278,
      "learning_rate": 1.2467194246656851e-06,
      "loss": 0.0563,
      "step": 718
    },
    {
      "epoch": 0.16359499431171787,
      "grad_norm": 1.0756915461750918,
      "learning_rate": 1.2467102754883136e-06,
      "loss": 0.0411,
      "step": 719
    },
    {
      "epoch": 0.16382252559726962,
      "grad_norm": 1.2683417009383617,
      "learning_rate": 1.2467011136043228e-06,
      "loss": 0.0454,
      "step": 720
    },
    {
      "epoch": 0.1640500568828214,
      "grad_norm": 1.5509079166930662,
      "learning_rate": 1.2466919390138995e-06,
      "loss": 0.0745,
      "step": 721
    },
    {
      "epoch": 0.16427758816837315,
      "grad_norm": 1.799801190377342,
      "learning_rate": 1.246682751717232e-06,
      "loss": 0.0531,
      "step": 722
    },
    {
      "epoch": 0.1645051194539249,
      "grad_norm": 1.734480174185366,
      "learning_rate": 1.2466735517145074e-06,
      "loss": 0.0792,
      "step": 723
    },
    {
      "epoch": 0.16473265073947668,
      "grad_norm": 1.961760175196997,
      "learning_rate": 1.2466643390059138e-06,
      "loss": 0.0685,
      "step": 724
    },
    {
      "epoch": 0.16496018202502843,
      "grad_norm": 0.8466307246278059,
      "learning_rate": 1.2466551135916398e-06,
      "loss": 0.0229,
      "step": 725
    },
    {
      "epoch": 0.16518771331058021,
      "grad_norm": 1.2173112499342529,
      "learning_rate": 1.2466458754718737e-06,
      "loss": 0.0632,
      "step": 726
    },
    {
      "epoch": 0.16541524459613197,
      "grad_norm": 2.194872247572006,
      "learning_rate": 1.2466366246468045e-06,
      "loss": 0.0432,
      "step": 727
    },
    {
      "epoch": 0.16564277588168372,
      "grad_norm": 1.4416201742124337,
      "learning_rate": 1.246627361116621e-06,
      "loss": 0.0619,
      "step": 728
    },
    {
      "epoch": 0.1658703071672355,
      "grad_norm": 1.8390552268184857,
      "learning_rate": 1.246618084881513e-06,
      "loss": 0.1148,
      "step": 729
    },
    {
      "epoch": 0.16609783845278725,
      "grad_norm": 1.7117932742809556,
      "learning_rate": 1.2466087959416695e-06,
      "loss": 0.0354,
      "step": 730
    },
    {
      "epoch": 0.16632536973833903,
      "grad_norm": 2.1473525159969644,
      "learning_rate": 1.2465994942972805e-06,
      "loss": 0.0479,
      "step": 731
    },
    {
      "epoch": 0.16655290102389078,
      "grad_norm": 2.0228728542376846,
      "learning_rate": 1.2465901799485366e-06,
      "loss": 0.0819,
      "step": 732
    },
    {
      "epoch": 0.16678043230944256,
      "grad_norm": 0.7672679660904389,
      "learning_rate": 1.2465808528956277e-06,
      "loss": 0.0251,
      "step": 733
    },
    {
      "epoch": 0.1670079635949943,
      "grad_norm": 6.58955540161242,
      "learning_rate": 1.2465715131387446e-06,
      "loss": 0.0608,
      "step": 734
    },
    {
      "epoch": 0.16723549488054607,
      "grad_norm": 2.0572059988722873,
      "learning_rate": 1.2465621606780778e-06,
      "loss": 0.0841,
      "step": 735
    },
    {
      "epoch": 0.16746302616609784,
      "grad_norm": 1.628470993180389,
      "learning_rate": 1.2465527955138191e-06,
      "loss": 0.059,
      "step": 736
    },
    {
      "epoch": 0.1676905574516496,
      "grad_norm": 1.0886679327758462,
      "learning_rate": 1.2465434176461596e-06,
      "loss": 0.0419,
      "step": 737
    },
    {
      "epoch": 0.16791808873720138,
      "grad_norm": 1.4286021962604534,
      "learning_rate": 1.2465340270752908e-06,
      "loss": 0.0481,
      "step": 738
    },
    {
      "epoch": 0.16814562002275313,
      "grad_norm": 0.8663035560041373,
      "learning_rate": 1.2465246238014047e-06,
      "loss": 0.0295,
      "step": 739
    },
    {
      "epoch": 0.16837315130830488,
      "grad_norm": 1.8501418707953075,
      "learning_rate": 1.2465152078246936e-06,
      "loss": 0.0829,
      "step": 740
    },
    {
      "epoch": 0.16860068259385666,
      "grad_norm": 1.311824892930283,
      "learning_rate": 1.24650577914535e-06,
      "loss": 0.0418,
      "step": 741
    },
    {
      "epoch": 0.1688282138794084,
      "grad_norm": 1.249901682454073,
      "learning_rate": 1.2464963377635667e-06,
      "loss": 0.0479,
      "step": 742
    },
    {
      "epoch": 0.1690557451649602,
      "grad_norm": 1.9982982994055205,
      "learning_rate": 1.246486883679536e-06,
      "loss": 0.112,
      "step": 743
    },
    {
      "epoch": 0.16928327645051194,
      "grad_norm": 1.8891194873050854,
      "learning_rate": 1.246477416893452e-06,
      "loss": 0.0489,
      "step": 744
    },
    {
      "epoch": 0.1695108077360637,
      "grad_norm": 1.3265078136556134,
      "learning_rate": 1.2464679374055074e-06,
      "loss": 0.0492,
      "step": 745
    },
    {
      "epoch": 0.16973833902161548,
      "grad_norm": 1.1620433482725554,
      "learning_rate": 1.2464584452158968e-06,
      "loss": 0.0217,
      "step": 746
    },
    {
      "epoch": 0.16996587030716723,
      "grad_norm": 1.8320124858819533,
      "learning_rate": 1.2464489403248133e-06,
      "loss": 0.057,
      "step": 747
    },
    {
      "epoch": 0.170193401592719,
      "grad_norm": 0.8996339413245623,
      "learning_rate": 1.246439422732452e-06,
      "loss": 0.0308,
      "step": 748
    },
    {
      "epoch": 0.17042093287827076,
      "grad_norm": 1.1199200581464024,
      "learning_rate": 1.2464298924390066e-06,
      "loss": 0.0449,
      "step": 749
    },
    {
      "epoch": 0.17064846416382254,
      "grad_norm": 1.6610024470498899,
      "learning_rate": 1.2464203494446725e-06,
      "loss": 0.0667,
      "step": 750
    },
    {
      "epoch": 0.1708759954493743,
      "grad_norm": 1.1007524784666796,
      "learning_rate": 1.2464107937496444e-06,
      "loss": 0.0341,
      "step": 751
    },
    {
      "epoch": 0.17110352673492604,
      "grad_norm": 1.0836506765705751,
      "learning_rate": 1.246401225354118e-06,
      "loss": 0.0397,
      "step": 752
    },
    {
      "epoch": 0.17133105802047782,
      "grad_norm": 1.320442517854178,
      "learning_rate": 1.2463916442582883e-06,
      "loss": 0.0694,
      "step": 753
    },
    {
      "epoch": 0.17155858930602957,
      "grad_norm": 1.0431861088552385,
      "learning_rate": 1.2463820504623516e-06,
      "loss": 0.0285,
      "step": 754
    },
    {
      "epoch": 0.17178612059158135,
      "grad_norm": 0.9128127625945931,
      "learning_rate": 1.246372443966504e-06,
      "loss": 0.0394,
      "step": 755
    },
    {
      "epoch": 0.1720136518771331,
      "grad_norm": 1.3898389984950463,
      "learning_rate": 1.246362824770941e-06,
      "loss": 0.0451,
      "step": 756
    },
    {
      "epoch": 0.17224118316268486,
      "grad_norm": 0.9752697300280603,
      "learning_rate": 1.2463531928758605e-06,
      "loss": 0.0398,
      "step": 757
    },
    {
      "epoch": 0.17246871444823664,
      "grad_norm": 1.8211753063899252,
      "learning_rate": 1.2463435482814585e-06,
      "loss": 0.0947,
      "step": 758
    },
    {
      "epoch": 0.1726962457337884,
      "grad_norm": 1.5787941280691695,
      "learning_rate": 1.246333890987932e-06,
      "loss": 0.1047,
      "step": 759
    },
    {
      "epoch": 0.17292377701934017,
      "grad_norm": 1.8356072487697,
      "learning_rate": 1.246324220995479e-06,
      "loss": 0.0729,
      "step": 760
    },
    {
      "epoch": 0.17315130830489192,
      "grad_norm": 4.476837441740864,
      "learning_rate": 1.2463145383042966e-06,
      "loss": 0.0686,
      "step": 761
    },
    {
      "epoch": 0.17337883959044367,
      "grad_norm": 252.52198381660855,
      "learning_rate": 1.2463048429145832e-06,
      "loss": 1.0972,
      "step": 762
    },
    {
      "epoch": 0.17360637087599545,
      "grad_norm": 1.6008327751394016,
      "learning_rate": 1.2462951348265364e-06,
      "loss": 0.067,
      "step": 763
    },
    {
      "epoch": 0.1738339021615472,
      "grad_norm": 1.087447761528399,
      "learning_rate": 1.2462854140403553e-06,
      "loss": 0.0433,
      "step": 764
    },
    {
      "epoch": 0.17406143344709898,
      "grad_norm": 2.5989190208674384,
      "learning_rate": 1.2462756805562378e-06,
      "loss": 0.1183,
      "step": 765
    },
    {
      "epoch": 0.17428896473265074,
      "grad_norm": 1.0290974538586322,
      "learning_rate": 1.2462659343743832e-06,
      "loss": 0.048,
      "step": 766
    },
    {
      "epoch": 0.17451649601820252,
      "grad_norm": 1.2617664816926566,
      "learning_rate": 1.2462561754949908e-06,
      "loss": 0.0498,
      "step": 767
    },
    {
      "epoch": 0.17474402730375427,
      "grad_norm": 1.6415043644592435,
      "learning_rate": 1.2462464039182598e-06,
      "loss": 0.062,
      "step": 768
    },
    {
      "epoch": 0.17497155858930602,
      "grad_norm": 1.6673873794852758,
      "learning_rate": 1.2462366196443903e-06,
      "loss": 0.0656,
      "step": 769
    },
    {
      "epoch": 0.1751990898748578,
      "grad_norm": 3.3205898186422718,
      "learning_rate": 1.246226822673582e-06,
      "loss": 0.1071,
      "step": 770
    },
    {
      "epoch": 0.17542662116040955,
      "grad_norm": 1.280610915532866,
      "learning_rate": 1.2462170130060351e-06,
      "loss": 0.0318,
      "step": 771
    },
    {
      "epoch": 0.17565415244596133,
      "grad_norm": 1.9843858659809077,
      "learning_rate": 1.24620719064195e-06,
      "loss": 0.0829,
      "step": 772
    },
    {
      "epoch": 0.17588168373151308,
      "grad_norm": 2.184973202731946,
      "learning_rate": 1.246197355581528e-06,
      "loss": 0.0998,
      "step": 773
    },
    {
      "epoch": 0.17610921501706484,
      "grad_norm": 1.361748313569626,
      "learning_rate": 1.2461875078249694e-06,
      "loss": 0.0459,
      "step": 774
    },
    {
      "epoch": 0.17633674630261661,
      "grad_norm": 1.5219919911848705,
      "learning_rate": 1.246177647372476e-06,
      "loss": 0.0539,
      "step": 775
    },
    {
      "epoch": 0.17656427758816837,
      "grad_norm": 49.0672993271897,
      "learning_rate": 1.246167774224249e-06,
      "loss": 0.5261,
      "step": 776
    },
    {
      "epoch": 0.17679180887372015,
      "grad_norm": 1.5576377429332795,
      "learning_rate": 1.2461578883804903e-06,
      "loss": 0.036,
      "step": 777
    },
    {
      "epoch": 0.1770193401592719,
      "grad_norm": 0.869488380021361,
      "learning_rate": 1.246147989841402e-06,
      "loss": 0.0252,
      "step": 778
    },
    {
      "epoch": 0.17724687144482365,
      "grad_norm": 1.2894143452262044,
      "learning_rate": 1.2461380786071863e-06,
      "loss": 0.0401,
      "step": 779
    },
    {
      "epoch": 0.17747440273037543,
      "grad_norm": 1.4297168158807256,
      "learning_rate": 1.246128154678046e-06,
      "loss": 0.0586,
      "step": 780
    },
    {
      "epoch": 0.17770193401592718,
      "grad_norm": 4.251777831616221,
      "learning_rate": 1.2461182180541835e-06,
      "loss": 0.0484,
      "step": 781
    },
    {
      "epoch": 0.17792946530147896,
      "grad_norm": 1.1899715722609847,
      "learning_rate": 1.2461082687358022e-06,
      "loss": 0.0391,
      "step": 782
    },
    {
      "epoch": 0.1781569965870307,
      "grad_norm": 1.0511863654570817,
      "learning_rate": 1.2460983067231055e-06,
      "loss": 0.0424,
      "step": 783
    },
    {
      "epoch": 0.1783845278725825,
      "grad_norm": 1.6108410955721733,
      "learning_rate": 1.246088332016297e-06,
      "loss": 0.0453,
      "step": 784
    },
    {
      "epoch": 0.17861205915813425,
      "grad_norm": 1.014732446092101,
      "learning_rate": 1.2460783446155802e-06,
      "loss": 0.0398,
      "step": 785
    },
    {
      "epoch": 0.178839590443686,
      "grad_norm": 3.365599666409623,
      "learning_rate": 1.2460683445211596e-06,
      "loss": 0.1579,
      "step": 786
    },
    {
      "epoch": 0.17906712172923778,
      "grad_norm": 2.0790980638653984,
      "learning_rate": 1.2460583317332395e-06,
      "loss": 0.1013,
      "step": 787
    },
    {
      "epoch": 0.17929465301478953,
      "grad_norm": 1.0292372040908986,
      "learning_rate": 1.2460483062520246e-06,
      "loss": 0.0399,
      "step": 788
    },
    {
      "epoch": 0.1795221843003413,
      "grad_norm": 1.056684060389556,
      "learning_rate": 1.2460382680777196e-06,
      "loss": 0.0342,
      "step": 789
    },
    {
      "epoch": 0.17974971558589306,
      "grad_norm": 1.3734567420277208,
      "learning_rate": 1.2460282172105298e-06,
      "loss": 0.0673,
      "step": 790
    },
    {
      "epoch": 0.1799772468714448,
      "grad_norm": 1.519969265957327,
      "learning_rate": 1.2460181536506608e-06,
      "loss": 0.0775,
      "step": 791
    },
    {
      "epoch": 0.1802047781569966,
      "grad_norm": 1.514321767455213,
      "learning_rate": 1.2460080773983177e-06,
      "loss": 0.0659,
      "step": 792
    },
    {
      "epoch": 0.18043230944254834,
      "grad_norm": 1.0361490916169322,
      "learning_rate": 1.2459979884537072e-06,
      "loss": 0.0424,
      "step": 793
    },
    {
      "epoch": 0.18065984072810012,
      "grad_norm": 1.3102760225972407,
      "learning_rate": 1.2459878868170348e-06,
      "loss": 0.0476,
      "step": 794
    },
    {
      "epoch": 0.18088737201365188,
      "grad_norm": 1.4645693998145335,
      "learning_rate": 1.2459777724885075e-06,
      "loss": 0.0522,
      "step": 795
    },
    {
      "epoch": 0.18111490329920363,
      "grad_norm": 1.6043429433204082,
      "learning_rate": 1.2459676454683318e-06,
      "loss": 0.0415,
      "step": 796
    },
    {
      "epoch": 0.1813424345847554,
      "grad_norm": 2.3566952879735963,
      "learning_rate": 1.2459575057567144e-06,
      "loss": 0.0916,
      "step": 797
    },
    {
      "epoch": 0.18156996587030716,
      "grad_norm": 1.5769029444288707,
      "learning_rate": 1.245947353353863e-06,
      "loss": 0.0596,
      "step": 798
    },
    {
      "epoch": 0.18179749715585894,
      "grad_norm": 2.293268133767521,
      "learning_rate": 1.245937188259985e-06,
      "loss": 0.1087,
      "step": 799
    },
    {
      "epoch": 0.1820250284414107,
      "grad_norm": 2.0227046414956096,
      "learning_rate": 1.245927010475288e-06,
      "loss": 0.0986,
      "step": 800
    },
    {
      "epoch": 0.18225255972696247,
      "grad_norm": 1.4895840436039498,
      "learning_rate": 1.24591681999998e-06,
      "loss": 0.0633,
      "step": 801
    },
    {
      "epoch": 0.18248009101251422,
      "grad_norm": 1.9671715135309942,
      "learning_rate": 1.2459066168342693e-06,
      "loss": 0.0551,
      "step": 802
    },
    {
      "epoch": 0.18270762229806597,
      "grad_norm": 2.1525267147108322,
      "learning_rate": 1.2458964009783646e-06,
      "loss": 0.0809,
      "step": 803
    },
    {
      "epoch": 0.18293515358361775,
      "grad_norm": 1.2864874343026942,
      "learning_rate": 1.2458861724324745e-06,
      "loss": 0.064,
      "step": 804
    },
    {
      "epoch": 0.1831626848691695,
      "grad_norm": 3.1883103987924444,
      "learning_rate": 1.2458759311968084e-06,
      "loss": 0.1303,
      "step": 805
    },
    {
      "epoch": 0.1833902161547213,
      "grad_norm": 1.0904025457159896,
      "learning_rate": 1.245865677271575e-06,
      "loss": 0.0419,
      "step": 806
    },
    {
      "epoch": 0.18361774744027304,
      "grad_norm": 1.4716173202867617,
      "learning_rate": 1.2458554106569844e-06,
      "loss": 0.072,
      "step": 807
    },
    {
      "epoch": 0.1838452787258248,
      "grad_norm": 0.9197168745315846,
      "learning_rate": 1.2458451313532463e-06,
      "loss": 0.04,
      "step": 808
    },
    {
      "epoch": 0.18407281001137657,
      "grad_norm": 1.2386933488526581,
      "learning_rate": 1.2458348393605708e-06,
      "loss": 0.0383,
      "step": 809
    },
    {
      "epoch": 0.18430034129692832,
      "grad_norm": 1.2129381734644349,
      "learning_rate": 1.2458245346791678e-06,
      "loss": 0.0486,
      "step": 810
    },
    {
      "epoch": 0.1845278725824801,
      "grad_norm": 1.0772770126401692,
      "learning_rate": 1.2458142173092486e-06,
      "loss": 0.0654,
      "step": 811
    },
    {
      "epoch": 0.18475540386803185,
      "grad_norm": 5.609205856069292,
      "learning_rate": 1.2458038872510237e-06,
      "loss": 0.0565,
      "step": 812
    },
    {
      "epoch": 0.1849829351535836,
      "grad_norm": 1.1271779547323408,
      "learning_rate": 1.2457935445047042e-06,
      "loss": 0.0414,
      "step": 813
    },
    {
      "epoch": 0.18521046643913538,
      "grad_norm": 1.1585514417402933,
      "learning_rate": 1.2457831890705018e-06,
      "loss": 0.0646,
      "step": 814
    },
    {
      "epoch": 0.18543799772468714,
      "grad_norm": 1.6262463401225817,
      "learning_rate": 1.2457728209486279e-06,
      "loss": 0.0555,
      "step": 815
    },
    {
      "epoch": 0.18566552901023892,
      "grad_norm": 1.6971282783367077,
      "learning_rate": 1.2457624401392943e-06,
      "loss": 0.0803,
      "step": 816
    },
    {
      "epoch": 0.18589306029579067,
      "grad_norm": 1.5890865721783058,
      "learning_rate": 1.2457520466427135e-06,
      "loss": 0.0637,
      "step": 817
    },
    {
      "epoch": 0.18612059158134245,
      "grad_norm": 1.663587359707044,
      "learning_rate": 1.2457416404590974e-06,
      "loss": 0.063,
      "step": 818
    },
    {
      "epoch": 0.1863481228668942,
      "grad_norm": 1.4589989522369327,
      "learning_rate": 1.2457312215886592e-06,
      "loss": 0.0665,
      "step": 819
    },
    {
      "epoch": 0.18657565415244595,
      "grad_norm": 1.2689131726245948,
      "learning_rate": 1.2457207900316115e-06,
      "loss": 0.0653,
      "step": 820
    },
    {
      "epoch": 0.18680318543799773,
      "grad_norm": 1.2763445646366045,
      "learning_rate": 1.245710345788168e-06,
      "loss": 0.0691,
      "step": 821
    },
    {
      "epoch": 0.18703071672354948,
      "grad_norm": 2.1044109818141523,
      "learning_rate": 1.2456998888585414e-06,
      "loss": 0.0334,
      "step": 822
    },
    {
      "epoch": 0.18725824800910126,
      "grad_norm": 1.1582960528680695,
      "learning_rate": 1.245689419242946e-06,
      "loss": 0.0513,
      "step": 823
    },
    {
      "epoch": 0.18748577929465302,
      "grad_norm": 1.4301422945996285,
      "learning_rate": 1.2456789369415955e-06,
      "loss": 0.0632,
      "step": 824
    },
    {
      "epoch": 0.18771331058020477,
      "grad_norm": 1.5961327622673143,
      "learning_rate": 1.2456684419547044e-06,
      "loss": 0.0915,
      "step": 825
    },
    {
      "epoch": 0.18794084186575655,
      "grad_norm": 1.156039499623056,
      "learning_rate": 1.245657934282487e-06,
      "loss": 0.0406,
      "step": 826
    },
    {
      "epoch": 0.1881683731513083,
      "grad_norm": 0.7964273734238357,
      "learning_rate": 1.245647413925158e-06,
      "loss": 0.0339,
      "step": 827
    },
    {
      "epoch": 0.18839590443686008,
      "grad_norm": 1.3110827687978466,
      "learning_rate": 1.2456368808829327e-06,
      "loss": 0.0315,
      "step": 828
    },
    {
      "epoch": 0.18862343572241183,
      "grad_norm": 2.167581797519473,
      "learning_rate": 1.2456263351560261e-06,
      "loss": 0.0944,
      "step": 829
    },
    {
      "epoch": 0.18885096700796358,
      "grad_norm": 1.7008701552637757,
      "learning_rate": 1.2456157767446538e-06,
      "loss": 0.0675,
      "step": 830
    },
    {
      "epoch": 0.18907849829351536,
      "grad_norm": 0.9862167570499358,
      "learning_rate": 1.245605205649032e-06,
      "loss": 0.0283,
      "step": 831
    },
    {
      "epoch": 0.18930602957906711,
      "grad_norm": 1.4504322451994889,
      "learning_rate": 1.245594621869376e-06,
      "loss": 0.0855,
      "step": 832
    },
    {
      "epoch": 0.1895335608646189,
      "grad_norm": 0.8637440563578112,
      "learning_rate": 1.2455840254059026e-06,
      "loss": 0.0457,
      "step": 833
    },
    {
      "epoch": 0.18976109215017065,
      "grad_norm": 1.365450351316701,
      "learning_rate": 1.2455734162588282e-06,
      "loss": 0.0523,
      "step": 834
    },
    {
      "epoch": 0.1899886234357224,
      "grad_norm": 1.9203688742265796,
      "learning_rate": 1.2455627944283697e-06,
      "loss": 0.0421,
      "step": 835
    },
    {
      "epoch": 0.19021615472127418,
      "grad_norm": 1.947700834519252,
      "learning_rate": 1.245552159914744e-06,
      "loss": 0.0631,
      "step": 836
    },
    {
      "epoch": 0.19044368600682593,
      "grad_norm": 1.274745482531776,
      "learning_rate": 1.245541512718169e-06,
      "loss": 0.0408,
      "step": 837
    },
    {
      "epoch": 0.1906712172923777,
      "grad_norm": 1.2840926844609297,
      "learning_rate": 1.245530852838862e-06,
      "loss": 0.0435,
      "step": 838
    },
    {
      "epoch": 0.19089874857792946,
      "grad_norm": 1.5640003481699476,
      "learning_rate": 1.2455201802770405e-06,
      "loss": 0.0616,
      "step": 839
    },
    {
      "epoch": 0.19112627986348124,
      "grad_norm": 1.2705630922274525,
      "learning_rate": 1.245509495032923e-06,
      "loss": 0.0604,
      "step": 840
    },
    {
      "epoch": 0.191353811149033,
      "grad_norm": 1.2721765422307298,
      "learning_rate": 1.2454987971067278e-06,
      "loss": 0.0431,
      "step": 841
    },
    {
      "epoch": 0.19158134243458474,
      "grad_norm": 1.7662664353990207,
      "learning_rate": 1.2454880864986737e-06,
      "loss": 0.0881,
      "step": 842
    },
    {
      "epoch": 0.19180887372013652,
      "grad_norm": 0.8548688062918027,
      "learning_rate": 1.2454773632089795e-06,
      "loss": 0.0219,
      "step": 843
    },
    {
      "epoch": 0.19203640500568828,
      "grad_norm": 1.1833145714689994,
      "learning_rate": 1.2454666272378644e-06,
      "loss": 0.0502,
      "step": 844
    },
    {
      "epoch": 0.19226393629124006,
      "grad_norm": 2.15061289434072,
      "learning_rate": 1.2454558785855475e-06,
      "loss": 0.0687,
      "step": 845
    },
    {
      "epoch": 0.1924914675767918,
      "grad_norm": 1.0201954402445583,
      "learning_rate": 1.245445117252249e-06,
      "loss": 0.0377,
      "step": 846
    },
    {
      "epoch": 0.19271899886234356,
      "grad_norm": 2.256502679796923,
      "learning_rate": 1.2454343432381886e-06,
      "loss": 0.1344,
      "step": 847
    },
    {
      "epoch": 0.19294653014789534,
      "grad_norm": 0.9607197396814332,
      "learning_rate": 1.2454235565435862e-06,
      "loss": 0.0293,
      "step": 848
    },
    {
      "epoch": 0.1931740614334471,
      "grad_norm": 1.1041321579517591,
      "learning_rate": 1.2454127571686629e-06,
      "loss": 0.0588,
      "step": 849
    },
    {
      "epoch": 0.19340159271899887,
      "grad_norm": 1.551888854493328,
      "learning_rate": 1.245401945113639e-06,
      "loss": 0.0884,
      "step": 850
    },
    {
      "epoch": 0.19362912400455062,
      "grad_norm": 1.2207595212151812,
      "learning_rate": 1.2453911203787355e-06,
      "loss": 0.0495,
      "step": 851
    },
    {
      "epoch": 0.19385665529010238,
      "grad_norm": 3.151128795398426,
      "learning_rate": 1.2453802829641736e-06,
      "loss": 0.0959,
      "step": 852
    },
    {
      "epoch": 0.19408418657565416,
      "grad_norm": 1.0306148148350838,
      "learning_rate": 1.2453694328701752e-06,
      "loss": 0.0362,
      "step": 853
    },
    {
      "epoch": 0.1943117178612059,
      "grad_norm": 1.4686426621438413,
      "learning_rate": 1.2453585700969614e-06,
      "loss": 0.0519,
      "step": 854
    },
    {
      "epoch": 0.1945392491467577,
      "grad_norm": 4.418162916580233,
      "learning_rate": 1.2453476946447547e-06,
      "loss": 0.0718,
      "step": 855
    },
    {
      "epoch": 0.19476678043230944,
      "grad_norm": 1.2510611267805927,
      "learning_rate": 1.2453368065137772e-06,
      "loss": 0.0471,
      "step": 856
    },
    {
      "epoch": 0.19499431171786122,
      "grad_norm": 3.8279701209646566,
      "learning_rate": 1.2453259057042514e-06,
      "loss": 0.0962,
      "step": 857
    },
    {
      "epoch": 0.19522184300341297,
      "grad_norm": 0.9857769216686043,
      "learning_rate": 1.2453149922164003e-06,
      "loss": 0.0361,
      "step": 858
    },
    {
      "epoch": 0.19544937428896472,
      "grad_norm": 1.2075815978254054,
      "learning_rate": 1.2453040660504468e-06,
      "loss": 0.0361,
      "step": 859
    },
    {
      "epoch": 0.1956769055745165,
      "grad_norm": 1.5382334773010704,
      "learning_rate": 1.2452931272066141e-06,
      "loss": 0.0489,
      "step": 860
    },
    {
      "epoch": 0.19590443686006825,
      "grad_norm": 1.2911029078532041,
      "learning_rate": 1.245282175685126e-06,
      "loss": 0.0743,
      "step": 861
    },
    {
      "epoch": 0.19613196814562003,
      "grad_norm": 2.432722962095318,
      "learning_rate": 1.2452712114862063e-06,
      "loss": 0.106,
      "step": 862
    },
    {
      "epoch": 0.19635949943117179,
      "grad_norm": 1.3235947439225328,
      "learning_rate": 1.245260234610079e-06,
      "loss": 0.045,
      "step": 863
    },
    {
      "epoch": 0.19658703071672354,
      "grad_norm": 1.1324676509010458,
      "learning_rate": 1.2452492450569682e-06,
      "loss": 0.0554,
      "step": 864
    },
    {
      "epoch": 0.19681456200227532,
      "grad_norm": 2.2307637995876863,
      "learning_rate": 1.245238242827099e-06,
      "loss": 0.0832,
      "step": 865
    },
    {
      "epoch": 0.19704209328782707,
      "grad_norm": 1.4079128192407386,
      "learning_rate": 1.245227227920696e-06,
      "loss": 0.079,
      "step": 866
    },
    {
      "epoch": 0.19726962457337885,
      "grad_norm": 1.9175023199810017,
      "learning_rate": 1.2452162003379842e-06,
      "loss": 0.0917,
      "step": 867
    },
    {
      "epoch": 0.1974971558589306,
      "grad_norm": 1.731489599286668,
      "learning_rate": 1.2452051600791891e-06,
      "loss": 0.0571,
      "step": 868
    },
    {
      "epoch": 0.19772468714448235,
      "grad_norm": 1.3722463786220256,
      "learning_rate": 1.2451941071445367e-06,
      "loss": 0.039,
      "step": 869
    },
    {
      "epoch": 0.19795221843003413,
      "grad_norm": 1.287569176842529,
      "learning_rate": 1.2451830415342524e-06,
      "loss": 0.0427,
      "step": 870
    },
    {
      "epoch": 0.19817974971558588,
      "grad_norm": 2.2505234649573795,
      "learning_rate": 1.2451719632485627e-06,
      "loss": 0.0606,
      "step": 871
    },
    {
      "epoch": 0.19840728100113766,
      "grad_norm": 1.3085408197305148,
      "learning_rate": 1.2451608722876938e-06,
      "loss": 0.0659,
      "step": 872
    },
    {
      "epoch": 0.19863481228668942,
      "grad_norm": 2.3075742049076085,
      "learning_rate": 1.2451497686518722e-06,
      "loss": 0.0762,
      "step": 873
    },
    {
      "epoch": 0.1988623435722412,
      "grad_norm": 1.1650796320462304,
      "learning_rate": 1.2451386523413252e-06,
      "loss": 0.0559,
      "step": 874
    },
    {
      "epoch": 0.19908987485779295,
      "grad_norm": 1.199672804247241,
      "learning_rate": 1.24512752335628e-06,
      "loss": 0.0417,
      "step": 875
    },
    {
      "epoch": 0.1993174061433447,
      "grad_norm": 1.4954309542074338,
      "learning_rate": 1.2451163816969639e-06,
      "loss": 0.0841,
      "step": 876
    },
    {
      "epoch": 0.19954493742889648,
      "grad_norm": 1.4893257668939828,
      "learning_rate": 1.2451052273636045e-06,
      "loss": 0.0639,
      "step": 877
    },
    {
      "epoch": 0.19977246871444823,
      "grad_norm": 1.1681659159861986,
      "learning_rate": 1.24509406035643e-06,
      "loss": 0.0487,
      "step": 878
    },
    {
      "epoch": 0.2,
      "grad_norm": 1.337002748039853,
      "learning_rate": 1.2450828806756685e-06,
      "loss": 0.0408,
      "step": 879
    },
    {
      "epoch": 0.20022753128555176,
      "grad_norm": 1.238199679938109,
      "learning_rate": 1.245071688321549e-06,
      "loss": 0.0452,
      "step": 880
    },
    {
      "epoch": 0.20045506257110352,
      "grad_norm": 1.4016981334625263,
      "learning_rate": 1.2450604832942991e-06,
      "loss": 0.0462,
      "step": 881
    },
    {
      "epoch": 0.2006825938566553,
      "grad_norm": 1.4932886959066234,
      "learning_rate": 1.245049265594149e-06,
      "loss": 0.0735,
      "step": 882
    },
    {
      "epoch": 0.20091012514220705,
      "grad_norm": 1.2357598702477623,
      "learning_rate": 1.2450380352213271e-06,
      "loss": 0.0504,
      "step": 883
    },
    {
      "epoch": 0.20113765642775883,
      "grad_norm": 2.015377601632808,
      "learning_rate": 1.2450267921760636e-06,
      "loss": 0.0523,
      "step": 884
    },
    {
      "epoch": 0.20136518771331058,
      "grad_norm": 1.8296148009850803,
      "learning_rate": 1.2450155364585878e-06,
      "loss": 0.0554,
      "step": 885
    },
    {
      "epoch": 0.20159271899886233,
      "grad_norm": 1.9872439975590221,
      "learning_rate": 1.2450042680691301e-06,
      "loss": 0.0737,
      "step": 886
    },
    {
      "epoch": 0.2018202502844141,
      "grad_norm": 0.832951332510299,
      "learning_rate": 1.2449929870079206e-06,
      "loss": 0.0457,
      "step": 887
    },
    {
      "epoch": 0.20204778156996586,
      "grad_norm": 0.7896517265085776,
      "learning_rate": 1.24498169327519e-06,
      "loss": 0.0261,
      "step": 888
    },
    {
      "epoch": 0.20227531285551764,
      "grad_norm": 1.3980548207185421,
      "learning_rate": 1.2449703868711688e-06,
      "loss": 0.061,
      "step": 889
    },
    {
      "epoch": 0.2025028441410694,
      "grad_norm": 1.3109982570334282,
      "learning_rate": 1.2449590677960886e-06,
      "loss": 0.0525,
      "step": 890
    },
    {
      "epoch": 0.20273037542662117,
      "grad_norm": 0.8611577720266457,
      "learning_rate": 1.2449477360501802e-06,
      "loss": 0.0297,
      "step": 891
    },
    {
      "epoch": 0.20295790671217293,
      "grad_norm": 2.2537154917621267,
      "learning_rate": 1.2449363916336756e-06,
      "loss": 0.0658,
      "step": 892
    },
    {
      "epoch": 0.20318543799772468,
      "grad_norm": 2.7860866459618823,
      "learning_rate": 1.2449250345468065e-06,
      "loss": 0.0853,
      "step": 893
    },
    {
      "epoch": 0.20341296928327646,
      "grad_norm": 2.063772825536578,
      "learning_rate": 1.244913664789805e-06,
      "loss": 0.0413,
      "step": 894
    },
    {
      "epoch": 0.2036405005688282,
      "grad_norm": 1.3514226592309118,
      "learning_rate": 1.2449022823629036e-06,
      "loss": 0.0445,
      "step": 895
    },
    {
      "epoch": 0.20386803185438,
      "grad_norm": 1.2525160508368907,
      "learning_rate": 1.2448908872663347e-06,
      "loss": 0.0337,
      "step": 896
    },
    {
      "epoch": 0.20409556313993174,
      "grad_norm": 0.9460587754259561,
      "learning_rate": 1.2448794795003313e-06,
      "loss": 0.0391,
      "step": 897
    },
    {
      "epoch": 0.2043230944254835,
      "grad_norm": 1.146769061895453,
      "learning_rate": 1.2448680590651269e-06,
      "loss": 0.0618,
      "step": 898
    },
    {
      "epoch": 0.20455062571103527,
      "grad_norm": 1.277169361575558,
      "learning_rate": 1.2448566259609543e-06,
      "loss": 0.0479,
      "step": 899
    },
    {
      "epoch": 0.20477815699658702,
      "grad_norm": 1.2913475907417995,
      "learning_rate": 1.2448451801880476e-06,
      "loss": 0.0352,
      "step": 900
    },
    {
      "epoch": 0.2050056882821388,
      "grad_norm": 1.614343484279519,
      "learning_rate": 1.2448337217466404e-06,
      "loss": 0.0672,
      "step": 901
    },
    {
      "epoch": 0.20523321956769056,
      "grad_norm": 1.1122189034817742,
      "learning_rate": 1.2448222506369675e-06,
      "loss": 0.0385,
      "step": 902
    },
    {
      "epoch": 0.2054607508532423,
      "grad_norm": 1.5480813920575183,
      "learning_rate": 1.2448107668592626e-06,
      "loss": 0.0696,
      "step": 903
    },
    {
      "epoch": 0.2056882821387941,
      "grad_norm": 2.133323122403055,
      "learning_rate": 1.244799270413761e-06,
      "loss": 0.0726,
      "step": 904
    },
    {
      "epoch": 0.20591581342434584,
      "grad_norm": 1.9375085657983182,
      "learning_rate": 1.2447877613006972e-06,
      "loss": 0.0762,
      "step": 905
    },
    {
      "epoch": 0.20614334470989762,
      "grad_norm": 2.138121754084785,
      "learning_rate": 1.244776239520307e-06,
      "loss": 0.1251,
      "step": 906
    },
    {
      "epoch": 0.20637087599544937,
      "grad_norm": 69.30343641276347,
      "learning_rate": 1.244764705072825e-06,
      "loss": 1.5633,
      "step": 907
    },
    {
      "epoch": 0.20659840728100115,
      "grad_norm": 0.9136079997375776,
      "learning_rate": 1.2447531579584878e-06,
      "loss": 0.0359,
      "step": 908
    },
    {
      "epoch": 0.2068259385665529,
      "grad_norm": 2.258737842455169,
      "learning_rate": 1.2447415981775312e-06,
      "loss": 0.0815,
      "step": 909
    },
    {
      "epoch": 0.20705346985210465,
      "grad_norm": 1.1302894088592583,
      "learning_rate": 1.2447300257301912e-06,
      "loss": 0.0377,
      "step": 910
    },
    {
      "epoch": 0.20728100113765643,
      "grad_norm": 1.4921976481695052,
      "learning_rate": 1.2447184406167045e-06,
      "loss": 0.0867,
      "step": 911
    },
    {
      "epoch": 0.2075085324232082,
      "grad_norm": 2.2983270765657546,
      "learning_rate": 1.2447068428373077e-06,
      "loss": 0.0726,
      "step": 912
    },
    {
      "epoch": 0.20773606370875997,
      "grad_norm": 1.9446016021679782,
      "learning_rate": 1.244695232392238e-06,
      "loss": 0.064,
      "step": 913
    },
    {
      "epoch": 0.20796359499431172,
      "grad_norm": 0.9872094991366439,
      "learning_rate": 1.2446836092817328e-06,
      "loss": 0.0432,
      "step": 914
    },
    {
      "epoch": 0.20819112627986347,
      "grad_norm": 6.934093481791269,
      "learning_rate": 1.2446719735060293e-06,
      "loss": 0.0945,
      "step": 915
    },
    {
      "epoch": 0.20841865756541525,
      "grad_norm": 1.360488111746127,
      "learning_rate": 1.2446603250653658e-06,
      "loss": 0.0603,
      "step": 916
    },
    {
      "epoch": 0.208646188850967,
      "grad_norm": 1.3310461950551042,
      "learning_rate": 1.24464866395998e-06,
      "loss": 0.0365,
      "step": 917
    },
    {
      "epoch": 0.20887372013651878,
      "grad_norm": 0.9182393241316875,
      "learning_rate": 1.2446369901901102e-06,
      "loss": 0.0559,
      "step": 918
    },
    {
      "epoch": 0.20910125142207053,
      "grad_norm": 1.4023619755477905,
      "learning_rate": 1.2446253037559952e-06,
      "loss": 0.0457,
      "step": 919
    },
    {
      "epoch": 0.20932878270762229,
      "grad_norm": 1.1256168684655514,
      "learning_rate": 1.2446136046578739e-06,
      "loss": 0.0455,
      "step": 920
    },
    {
      "epoch": 0.20955631399317406,
      "grad_norm": 1.9433578174394295,
      "learning_rate": 1.2446018928959853e-06,
      "loss": 0.0962,
      "step": 921
    },
    {
      "epoch": 0.20978384527872582,
      "grad_norm": 1.5957841959716605,
      "learning_rate": 1.2445901684705685e-06,
      "loss": 0.0408,
      "step": 922
    },
    {
      "epoch": 0.2100113765642776,
      "grad_norm": 1.075018126077203,
      "learning_rate": 1.2445784313818638e-06,
      "loss": 0.0378,
      "step": 923
    },
    {
      "epoch": 0.21023890784982935,
      "grad_norm": 1.5370330862679797,
      "learning_rate": 1.2445666816301102e-06,
      "loss": 0.0578,
      "step": 924
    },
    {
      "epoch": 0.21046643913538113,
      "grad_norm": 1.4943129389397118,
      "learning_rate": 1.2445549192155487e-06,
      "loss": 0.0875,
      "step": 925
    },
    {
      "epoch": 0.21069397042093288,
      "grad_norm": 1.1898394401405206,
      "learning_rate": 1.244543144138419e-06,
      "loss": 0.0486,
      "step": 926
    },
    {
      "epoch": 0.21092150170648463,
      "grad_norm": 1.6954763665720234,
      "learning_rate": 1.2445313563989624e-06,
      "loss": 0.0641,
      "step": 927
    },
    {
      "epoch": 0.2111490329920364,
      "grad_norm": 1.0426224473753456,
      "learning_rate": 1.2445195559974194e-06,
      "loss": 0.0404,
      "step": 928
    },
    {
      "epoch": 0.21137656427758816,
      "grad_norm": 1.6481528761515403,
      "learning_rate": 1.244507742934031e-06,
      "loss": 0.0839,
      "step": 929
    },
    {
      "epoch": 0.21160409556313994,
      "grad_norm": 0.9619424275618367,
      "learning_rate": 1.2444959172090393e-06,
      "loss": 0.0273,
      "step": 930
    },
    {
      "epoch": 0.2118316268486917,
      "grad_norm": 3.69544520271876,
      "learning_rate": 1.2444840788226854e-06,
      "loss": 0.1652,
      "step": 931
    },
    {
      "epoch": 0.21205915813424345,
      "grad_norm": 1.804306156177365,
      "learning_rate": 1.2444722277752114e-06,
      "loss": 0.1027,
      "step": 932
    },
    {
      "epoch": 0.21228668941979523,
      "grad_norm": 1.768050037917839,
      "learning_rate": 1.2444603640668596e-06,
      "loss": 0.095,
      "step": 933
    },
    {
      "epoch": 0.21251422070534698,
      "grad_norm": 1.1285321294388682,
      "learning_rate": 1.2444484876978725e-06,
      "loss": 0.0465,
      "step": 934
    },
    {
      "epoch": 0.21274175199089876,
      "grad_norm": 1.7249462106069577,
      "learning_rate": 1.2444365986684929e-06,
      "loss": 0.0842,
      "step": 935
    },
    {
      "epoch": 0.2129692832764505,
      "grad_norm": 1.2306924671997328,
      "learning_rate": 1.2444246969789633e-06,
      "loss": 0.0447,
      "step": 936
    },
    {
      "epoch": 0.21319681456200226,
      "grad_norm": 1.1194256262692541,
      "learning_rate": 1.2444127826295277e-06,
      "loss": 0.0387,
      "step": 937
    },
    {
      "epoch": 0.21342434584755404,
      "grad_norm": 0.9357846968568098,
      "learning_rate": 1.244400855620429e-06,
      "loss": 0.0311,
      "step": 938
    },
    {
      "epoch": 0.2136518771331058,
      "grad_norm": 0.9399277842933605,
      "learning_rate": 1.2443889159519113e-06,
      "loss": 0.0408,
      "step": 939
    },
    {
      "epoch": 0.21387940841865757,
      "grad_norm": 1.038736805523899,
      "learning_rate": 1.2443769636242185e-06,
      "loss": 0.0573,
      "step": 940
    },
    {
      "epoch": 0.21410693970420933,
      "grad_norm": 1.179286632772412,
      "learning_rate": 1.244364998637595e-06,
      "loss": 0.0578,
      "step": 941
    },
    {
      "epoch": 0.2143344709897611,
      "grad_norm": 3.1404649465639594,
      "learning_rate": 1.2443530209922848e-06,
      "loss": 0.0549,
      "step": 942
    },
    {
      "epoch": 0.21456200227531286,
      "grad_norm": 1.4304654332903906,
      "learning_rate": 1.2443410306885337e-06,
      "loss": 0.0408,
      "step": 943
    },
    {
      "epoch": 0.2147895335608646,
      "grad_norm": 1.3926671923537388,
      "learning_rate": 1.244329027726586e-06,
      "loss": 0.0675,
      "step": 944
    },
    {
      "epoch": 0.2150170648464164,
      "grad_norm": 0.8473568816417418,
      "learning_rate": 1.2443170121066872e-06,
      "loss": 0.0388,
      "step": 945
    },
    {
      "epoch": 0.21524459613196814,
      "grad_norm": 1.7456532933762987,
      "learning_rate": 1.2443049838290827e-06,
      "loss": 0.0655,
      "step": 946
    },
    {
      "epoch": 0.21547212741751992,
      "grad_norm": 5.065824955296096,
      "learning_rate": 1.2442929428940186e-06,
      "loss": 0.196,
      "step": 947
    },
    {
      "epoch": 0.21569965870307167,
      "grad_norm": 0.9216519176270381,
      "learning_rate": 1.2442808893017414e-06,
      "loss": 0.0376,
      "step": 948
    },
    {
      "epoch": 0.21592718998862342,
      "grad_norm": 1.111383779669666,
      "learning_rate": 1.2442688230524965e-06,
      "loss": 0.0403,
      "step": 949
    },
    {
      "epoch": 0.2161547212741752,
      "grad_norm": 1.557530476173142,
      "learning_rate": 1.244256744146531e-06,
      "loss": 0.0365,
      "step": 950
    },
    {
      "epoch": 0.21638225255972696,
      "grad_norm": 1.1787378400008193,
      "learning_rate": 1.244244652584092e-06,
      "loss": 0.0507,
      "step": 951
    },
    {
      "epoch": 0.21660978384527874,
      "grad_norm": 0.9890913425273592,
      "learning_rate": 1.2442325483654263e-06,
      "loss": 0.0533,
      "step": 952
    },
    {
      "epoch": 0.2168373151308305,
      "grad_norm": 2.1206498633906294,
      "learning_rate": 1.2442204314907812e-06,
      "loss": 0.0794,
      "step": 953
    },
    {
      "epoch": 0.21706484641638224,
      "grad_norm": 1.3153818338446894,
      "learning_rate": 1.2442083019604047e-06,
      "loss": 0.0706,
      "step": 954
    },
    {
      "epoch": 0.21729237770193402,
      "grad_norm": 1.2497584639698338,
      "learning_rate": 1.2441961597745447e-06,
      "loss": 0.0474,
      "step": 955
    },
    {
      "epoch": 0.21751990898748577,
      "grad_norm": 1.0089839713441826,
      "learning_rate": 1.244184004933449e-06,
      "loss": 0.0354,
      "step": 956
    },
    {
      "epoch": 0.21774744027303755,
      "grad_norm": 1.967078988299315,
      "learning_rate": 1.2441718374373662e-06,
      "loss": 0.0371,
      "step": 957
    },
    {
      "epoch": 0.2179749715585893,
      "grad_norm": 1.1714442804547354,
      "learning_rate": 1.244159657286545e-06,
      "loss": 0.0452,
      "step": 958
    },
    {
      "epoch": 0.21820250284414108,
      "grad_norm": 0.940255872741034,
      "learning_rate": 1.2441474644812345e-06,
      "loss": 0.0363,
      "step": 959
    },
    {
      "epoch": 0.21843003412969283,
      "grad_norm": 1.100910872233546,
      "learning_rate": 1.2441352590216836e-06,
      "loss": 0.0357,
      "step": 960
    },
    {
      "epoch": 0.2186575654152446,
      "grad_norm": 1.9969980893987056,
      "learning_rate": 1.244123040908142e-06,
      "loss": 0.1572,
      "step": 961
    },
    {
      "epoch": 0.21888509670079637,
      "grad_norm": 1.5961804013592509,
      "learning_rate": 1.2441108101408592e-06,
      "loss": 0.0425,
      "step": 962
    },
    {
      "epoch": 0.21911262798634812,
      "grad_norm": 2.1162445389078024,
      "learning_rate": 1.2440985667200853e-06,
      "loss": 0.0517,
      "step": 963
    },
    {
      "epoch": 0.2193401592718999,
      "grad_norm": 1.064967765846801,
      "learning_rate": 1.2440863106460705e-06,
      "loss": 0.023,
      "step": 964
    },
    {
      "epoch": 0.21956769055745165,
      "grad_norm": 2.08098035050502,
      "learning_rate": 1.2440740419190655e-06,
      "loss": 0.0796,
      "step": 965
    },
    {
      "epoch": 0.2197952218430034,
      "grad_norm": 2.4340967444170136,
      "learning_rate": 1.2440617605393208e-06,
      "loss": 0.0673,
      "step": 966
    },
    {
      "epoch": 0.22002275312855518,
      "grad_norm": 2.139025801414634,
      "learning_rate": 1.2440494665070874e-06,
      "loss": 0.102,
      "step": 967
    },
    {
      "epoch": 0.22025028441410693,
      "grad_norm": 1.3877680351054698,
      "learning_rate": 1.2440371598226165e-06,
      "loss": 0.0588,
      "step": 968
    },
    {
      "epoch": 0.2204778156996587,
      "grad_norm": 1.3210352246509975,
      "learning_rate": 1.2440248404861598e-06,
      "loss": 0.0528,
      "step": 969
    },
    {
      "epoch": 0.22070534698521047,
      "grad_norm": 1.4713104262343806,
      "learning_rate": 1.2440125084979693e-06,
      "loss": 0.0468,
      "step": 970
    },
    {
      "epoch": 0.22093287827076222,
      "grad_norm": 1.4854620779239043,
      "learning_rate": 1.2440001638582965e-06,
      "loss": 0.0494,
      "step": 971
    },
    {
      "epoch": 0.221160409556314,
      "grad_norm": 1.1733972765082468,
      "learning_rate": 1.2439878065673944e-06,
      "loss": 0.0517,
      "step": 972
    },
    {
      "epoch": 0.22138794084186575,
      "grad_norm": 2.818817007281193,
      "learning_rate": 1.2439754366255149e-06,
      "loss": 0.0332,
      "step": 973
    },
    {
      "epoch": 0.22161547212741753,
      "grad_norm": 1.3519809642179033,
      "learning_rate": 1.2439630540329111e-06,
      "loss": 0.0714,
      "step": 974
    },
    {
      "epoch": 0.22184300341296928,
      "grad_norm": 1.2366129488699829,
      "learning_rate": 1.2439506587898358e-06,
      "loss": 0.0307,
      "step": 975
    },
    {
      "epoch": 0.22207053469852106,
      "grad_norm": 1.2264022197392555,
      "learning_rate": 1.243938250896543e-06,
      "loss": 0.0505,
      "step": 976
    },
    {
      "epoch": 0.2222980659840728,
      "grad_norm": 1.7725437520836973,
      "learning_rate": 1.2439258303532858e-06,
      "loss": 0.0551,
      "step": 977
    },
    {
      "epoch": 0.22252559726962456,
      "grad_norm": 0.9663835721436027,
      "learning_rate": 1.243913397160318e-06,
      "loss": 0.0375,
      "step": 978
    },
    {
      "epoch": 0.22275312855517634,
      "grad_norm": 0.8743576163827176,
      "learning_rate": 1.2439009513178938e-06,
      "loss": 0.0245,
      "step": 979
    },
    {
      "epoch": 0.2229806598407281,
      "grad_norm": 1.054127962193629,
      "learning_rate": 1.2438884928262678e-06,
      "loss": 0.0323,
      "step": 980
    },
    {
      "epoch": 0.22320819112627988,
      "grad_norm": 0.9874001278398739,
      "learning_rate": 1.2438760216856944e-06,
      "loss": 0.0435,
      "step": 981
    },
    {
      "epoch": 0.22343572241183163,
      "grad_norm": 1.3544331214991197,
      "learning_rate": 1.2438635378964284e-06,
      "loss": 0.0707,
      "step": 982
    },
    {
      "epoch": 0.22366325369738338,
      "grad_norm": 1.3272223315121154,
      "learning_rate": 1.2438510414587251e-06,
      "loss": 0.0553,
      "step": 983
    },
    {
      "epoch": 0.22389078498293516,
      "grad_norm": 1.4246609306991453,
      "learning_rate": 1.24383853237284e-06,
      "loss": 0.0616,
      "step": 984
    },
    {
      "epoch": 0.2241183162684869,
      "grad_norm": 1.3509367739810239,
      "learning_rate": 1.2438260106390285e-06,
      "loss": 0.054,
      "step": 985
    },
    {
      "epoch": 0.2243458475540387,
      "grad_norm": 2.743016610014617,
      "learning_rate": 1.2438134762575467e-06,
      "loss": 0.1654,
      "step": 986
    },
    {
      "epoch": 0.22457337883959044,
      "grad_norm": 1.1104234557718136,
      "learning_rate": 1.243800929228651e-06,
      "loss": 0.0438,
      "step": 987
    },
    {
      "epoch": 0.2248009101251422,
      "grad_norm": 1.5813628573176925,
      "learning_rate": 1.2437883695525974e-06,
      "loss": 0.0577,
      "step": 988
    },
    {
      "epoch": 0.22502844141069397,
      "grad_norm": 0.9637284966057095,
      "learning_rate": 1.2437757972296427e-06,
      "loss": 0.0441,
      "step": 989
    },
    {
      "epoch": 0.22525597269624573,
      "grad_norm": 1.2736418998037702,
      "learning_rate": 1.2437632122600442e-06,
      "loss": 0.0627,
      "step": 990
    },
    {
      "epoch": 0.2254835039817975,
      "grad_norm": 1.704338839028225,
      "learning_rate": 1.2437506146440587e-06,
      "loss": 0.0794,
      "step": 991
    },
    {
      "epoch": 0.22571103526734926,
      "grad_norm": 1.3083723593285776,
      "learning_rate": 1.243738004381944e-06,
      "loss": 0.0367,
      "step": 992
    },
    {
      "epoch": 0.225938566552901,
      "grad_norm": 1.2372256320814667,
      "learning_rate": 1.2437253814739572e-06,
      "loss": 0.0478,
      "step": 993
    },
    {
      "epoch": 0.2261660978384528,
      "grad_norm": 2.064272199741316,
      "learning_rate": 1.2437127459203572e-06,
      "loss": 0.0673,
      "step": 994
    },
    {
      "epoch": 0.22639362912400454,
      "grad_norm": 0.9919600393162777,
      "learning_rate": 1.2437000977214015e-06,
      "loss": 0.031,
      "step": 995
    },
    {
      "epoch": 0.22662116040955632,
      "grad_norm": 1.6080832002027023,
      "learning_rate": 1.243687436877349e-06,
      "loss": 0.0841,
      "step": 996
    },
    {
      "epoch": 0.22684869169510807,
      "grad_norm": 1.3930180180919087,
      "learning_rate": 1.2436747633884583e-06,
      "loss": 0.0616,
      "step": 997
    },
    {
      "epoch": 0.22707622298065985,
      "grad_norm": 4.70664655158806,
      "learning_rate": 1.2436620772549885e-06,
      "loss": 0.1541,
      "step": 998
    },
    {
      "epoch": 0.2273037542662116,
      "grad_norm": 2.3548939922944556,
      "learning_rate": 1.243649378477199e-06,
      "loss": 0.1161,
      "step": 999
    },
    {
      "epoch": 0.22753128555176336,
      "grad_norm": 1.9186203633212395,
      "learning_rate": 1.2436366670553491e-06,
      "loss": 0.0725,
      "step": 1000
    },
    {
      "epoch": 0.22775881683731514,
      "grad_norm": 1.726012700048525,
      "learning_rate": 1.2436239429896988e-06,
      "loss": 0.1038,
      "step": 1001
    },
    {
      "epoch": 0.2279863481228669,
      "grad_norm": 1.7981785549763478,
      "learning_rate": 1.2436112062805081e-06,
      "loss": 0.0485,
      "step": 1002
    },
    {
      "epoch": 0.22821387940841867,
      "grad_norm": 1.935791028636196,
      "learning_rate": 1.2435984569280372e-06,
      "loss": 0.0773,
      "step": 1003
    },
    {
      "epoch": 0.22844141069397042,
      "grad_norm": 1.4700273599627696,
      "learning_rate": 1.2435856949325467e-06,
      "loss": 0.0584,
      "step": 1004
    },
    {
      "epoch": 0.22866894197952217,
      "grad_norm": 4.712742088498197,
      "learning_rate": 1.2435729202942972e-06,
      "loss": 0.073,
      "step": 1005
    },
    {
      "epoch": 0.22889647326507395,
      "grad_norm": 1.3146906101281273,
      "learning_rate": 1.2435601330135506e-06,
      "loss": 0.0357,
      "step": 1006
    },
    {
      "epoch": 0.2291240045506257,
      "grad_norm": 2.3674281228532066,
      "learning_rate": 1.2435473330905674e-06,
      "loss": 0.0701,
      "step": 1007
    },
    {
      "epoch": 0.22935153583617748,
      "grad_norm": 1.2459812264205032,
      "learning_rate": 1.2435345205256097e-06,
      "loss": 0.0375,
      "step": 1008
    },
    {
      "epoch": 0.22957906712172924,
      "grad_norm": 1.4505448834484096,
      "learning_rate": 1.243521695318939e-06,
      "loss": 0.057,
      "step": 1009
    },
    {
      "epoch": 0.229806598407281,
      "grad_norm": 1.014039827241364,
      "learning_rate": 1.2435088574708178e-06,
      "loss": 0.027,
      "step": 1010
    },
    {
      "epoch": 0.23003412969283277,
      "grad_norm": 2.0117138797089766,
      "learning_rate": 1.2434960069815083e-06,
      "loss": 0.0583,
      "step": 1011
    },
    {
      "epoch": 0.23026166097838452,
      "grad_norm": 2.319983736807929,
      "learning_rate": 1.243483143851273e-06,
      "loss": 0.0687,
      "step": 1012
    },
    {
      "epoch": 0.2304891922639363,
      "grad_norm": 1.5313544208925962,
      "learning_rate": 1.2434702680803751e-06,
      "loss": 0.0531,
      "step": 1013
    },
    {
      "epoch": 0.23071672354948805,
      "grad_norm": 2.4005097874561616,
      "learning_rate": 1.2434573796690774e-06,
      "loss": 0.0929,
      "step": 1014
    },
    {
      "epoch": 0.23094425483503983,
      "grad_norm": 1.4786754339201256,
      "learning_rate": 1.2434444786176435e-06,
      "loss": 0.072,
      "step": 1015
    },
    {
      "epoch": 0.23117178612059158,
      "grad_norm": 1.0190695534603456,
      "learning_rate": 1.2434315649263372e-06,
      "loss": 0.0336,
      "step": 1016
    },
    {
      "epoch": 0.23139931740614333,
      "grad_norm": 0.85632474720689,
      "learning_rate": 1.2434186385954225e-06,
      "loss": 0.035,
      "step": 1017
    },
    {
      "epoch": 0.23162684869169511,
      "grad_norm": 1.2607686804487748,
      "learning_rate": 1.243405699625163e-06,
      "loss": 0.0442,
      "step": 1018
    },
    {
      "epoch": 0.23185437997724687,
      "grad_norm": 1.2717127046145194,
      "learning_rate": 1.243392748015824e-06,
      "loss": 0.0377,
      "step": 1019
    },
    {
      "epoch": 0.23208191126279865,
      "grad_norm": 0.9597540032341592,
      "learning_rate": 1.2433797837676694e-06,
      "loss": 0.0434,
      "step": 1020
    },
    {
      "epoch": 0.2323094425483504,
      "grad_norm": 0.9627749018367474,
      "learning_rate": 1.2433668068809648e-06,
      "loss": 0.0303,
      "step": 1021
    },
    {
      "epoch": 0.23253697383390215,
      "grad_norm": 1.2513646734831103,
      "learning_rate": 1.243353817355975e-06,
      "loss": 0.039,
      "step": 1022
    },
    {
      "epoch": 0.23276450511945393,
      "grad_norm": 1.2104231875454026,
      "learning_rate": 1.2433408151929655e-06,
      "loss": 0.0361,
      "step": 1023
    },
    {
      "epoch": 0.23299203640500568,
      "grad_norm": 1.0202244699658645,
      "learning_rate": 1.2433278003922026e-06,
      "loss": 0.0351,
      "step": 1024
    },
    {
      "epoch": 0.23321956769055746,
      "grad_norm": 1.4263990244044498,
      "learning_rate": 1.2433147729539514e-06,
      "loss": 0.0815,
      "step": 1025
    },
    {
      "epoch": 0.2334470989761092,
      "grad_norm": 1.5336205920808743,
      "learning_rate": 1.2433017328784788e-06,
      "loss": 0.0637,
      "step": 1026
    },
    {
      "epoch": 0.23367463026166096,
      "grad_norm": 1.4842765531420945,
      "learning_rate": 1.2432886801660513e-06,
      "loss": 0.0619,
      "step": 1027
    },
    {
      "epoch": 0.23390216154721274,
      "grad_norm": 0.9615217436557268,
      "learning_rate": 1.2432756148169354e-06,
      "loss": 0.0424,
      "step": 1028
    },
    {
      "epoch": 0.2341296928327645,
      "grad_norm": 2.5880116161383273,
      "learning_rate": 1.2432625368313983e-06,
      "loss": 0.0925,
      "step": 1029
    },
    {
      "epoch": 0.23435722411831628,
      "grad_norm": 1.1806940004387316,
      "learning_rate": 1.2432494462097072e-06,
      "loss": 0.0547,
      "step": 1030
    },
    {
      "epoch": 0.23458475540386803,
      "grad_norm": 0.9929303526421415,
      "learning_rate": 1.2432363429521295e-06,
      "loss": 0.0529,
      "step": 1031
    },
    {
      "epoch": 0.2348122866894198,
      "grad_norm": 0.9571908888077277,
      "learning_rate": 1.2432232270589335e-06,
      "loss": 0.0334,
      "step": 1032
    },
    {
      "epoch": 0.23503981797497156,
      "grad_norm": 2.067393897730002,
      "learning_rate": 1.2432100985303868e-06,
      "loss": 0.106,
      "step": 1033
    },
    {
      "epoch": 0.2352673492605233,
      "grad_norm": 0.8691321322200917,
      "learning_rate": 1.243196957366758e-06,
      "loss": 0.0346,
      "step": 1034
    },
    {
      "epoch": 0.2354948805460751,
      "grad_norm": 1.5305165337075897,
      "learning_rate": 1.2431838035683155e-06,
      "loss": 0.0848,
      "step": 1035
    },
    {
      "epoch": 0.23572241183162684,
      "grad_norm": 1.3223616364406545,
      "learning_rate": 1.2431706371353282e-06,
      "loss": 0.0687,
      "step": 1036
    },
    {
      "epoch": 0.23594994311717862,
      "grad_norm": 1.928604930493819,
      "learning_rate": 1.2431574580680653e-06,
      "loss": 0.0649,
      "step": 1037
    },
    {
      "epoch": 0.23617747440273038,
      "grad_norm": 1.2898613078943457,
      "learning_rate": 1.2431442663667958e-06,
      "loss": 0.0542,
      "step": 1038
    },
    {
      "epoch": 0.23640500568828213,
      "grad_norm": 1.0765527225532854,
      "learning_rate": 1.2431310620317898e-06,
      "loss": 0.0601,
      "step": 1039
    },
    {
      "epoch": 0.2366325369738339,
      "grad_norm": 0.7732520151584088,
      "learning_rate": 1.2431178450633168e-06,
      "loss": 0.0285,
      "step": 1040
    },
    {
      "epoch": 0.23686006825938566,
      "grad_norm": 1.1685154356326117,
      "learning_rate": 1.2431046154616473e-06,
      "loss": 0.0356,
      "step": 1041
    },
    {
      "epoch": 0.23708759954493744,
      "grad_norm": 1.3543617648083628,
      "learning_rate": 1.2430913732270512e-06,
      "loss": 0.0446,
      "step": 1042
    },
    {
      "epoch": 0.2373151308304892,
      "grad_norm": 1.707962300050113,
      "learning_rate": 1.2430781183597995e-06,
      "loss": 0.0555,
      "step": 1043
    },
    {
      "epoch": 0.23754266211604094,
      "grad_norm": 1.2841428640766417,
      "learning_rate": 1.243064850860163e-06,
      "loss": 0.0447,
      "step": 1044
    },
    {
      "epoch": 0.23777019340159272,
      "grad_norm": 6.261159556049287,
      "learning_rate": 1.243051570728413e-06,
      "loss": 0.0749,
      "step": 1045
    },
    {
      "epoch": 0.23799772468714447,
      "grad_norm": 1.1364071139867686,
      "learning_rate": 1.2430382779648208e-06,
      "loss": 0.0326,
      "step": 1046
    },
    {
      "epoch": 0.23822525597269625,
      "grad_norm": 1.484581864381779,
      "learning_rate": 1.243024972569658e-06,
      "loss": 0.0645,
      "step": 1047
    },
    {
      "epoch": 0.238452787258248,
      "grad_norm": 1.1545473703015605,
      "learning_rate": 1.2430116545431966e-06,
      "loss": 0.0481,
      "step": 1048
    },
    {
      "epoch": 0.23868031854379979,
      "grad_norm": 0.8983805133665866,
      "learning_rate": 1.2429983238857088e-06,
      "loss": 0.0369,
      "step": 1049
    },
    {
      "epoch": 0.23890784982935154,
      "grad_norm": 1.068742935236811,
      "learning_rate": 1.2429849805974673e-06,
      "loss": 0.039,
      "step": 1050
    },
    {
      "epoch": 0.2391353811149033,
      "grad_norm": 1.4873654976644912,
      "learning_rate": 1.2429716246787444e-06,
      "loss": 0.0312,
      "step": 1051
    },
    {
      "epoch": 0.23936291240045507,
      "grad_norm": 0.9449501930869969,
      "learning_rate": 1.242958256129813e-06,
      "loss": 0.0567,
      "step": 1052
    },
    {
      "epoch": 0.23959044368600682,
      "grad_norm": 1.6243209526441913,
      "learning_rate": 1.242944874950947e-06,
      "loss": 0.0647,
      "step": 1053
    },
    {
      "epoch": 0.2398179749715586,
      "grad_norm": 1.4569612537340462,
      "learning_rate": 1.2429314811424192e-06,
      "loss": 0.0758,
      "step": 1054
    },
    {
      "epoch": 0.24004550625711035,
      "grad_norm": 1.5825027859681509,
      "learning_rate": 1.242918074704504e-06,
      "loss": 0.0621,
      "step": 1055
    },
    {
      "epoch": 0.2402730375426621,
      "grad_norm": 1.0511003317201362,
      "learning_rate": 1.2429046556374747e-06,
      "loss": 0.037,
      "step": 1056
    },
    {
      "epoch": 0.24050056882821388,
      "grad_norm": 1.5928024296302492,
      "learning_rate": 1.2428912239416057e-06,
      "loss": 0.0453,
      "step": 1057
    },
    {
      "epoch": 0.24072810011376564,
      "grad_norm": 1.1044477930026537,
      "learning_rate": 1.242877779617172e-06,
      "loss": 0.0419,
      "step": 1058
    },
    {
      "epoch": 0.24095563139931742,
      "grad_norm": 1.3035104956572268,
      "learning_rate": 1.242864322664448e-06,
      "loss": 0.0531,
      "step": 1059
    },
    {
      "epoch": 0.24118316268486917,
      "grad_norm": 1.4034057924806689,
      "learning_rate": 1.2428508530837088e-06,
      "loss": 0.0753,
      "step": 1060
    },
    {
      "epoch": 0.24141069397042092,
      "grad_norm": 45.147488919338755,
      "learning_rate": 1.2428373708752298e-06,
      "loss": 0.4166,
      "step": 1061
    },
    {
      "epoch": 0.2416382252559727,
      "grad_norm": 1.804049071201822,
      "learning_rate": 1.2428238760392862e-06,
      "loss": 0.0881,
      "step": 1062
    },
    {
      "epoch": 0.24186575654152445,
      "grad_norm": 1.2662139418382417,
      "learning_rate": 1.2428103685761543e-06,
      "loss": 0.0592,
      "step": 1063
    },
    {
      "epoch": 0.24209328782707623,
      "grad_norm": 2.633674196903193,
      "learning_rate": 1.2427968484861097e-06,
      "loss": 0.1104,
      "step": 1064
    },
    {
      "epoch": 0.24232081911262798,
      "grad_norm": 1.437345835314508,
      "learning_rate": 1.2427833157694292e-06,
      "loss": 0.059,
      "step": 1065
    },
    {
      "epoch": 0.24254835039817976,
      "grad_norm": 1.4759939750585893,
      "learning_rate": 1.2427697704263892e-06,
      "loss": 0.0564,
      "step": 1066
    },
    {
      "epoch": 0.24277588168373151,
      "grad_norm": 1.4003983063384298,
      "learning_rate": 1.2427562124572663e-06,
      "loss": 0.0756,
      "step": 1067
    },
    {
      "epoch": 0.24300341296928327,
      "grad_norm": 1.2893015359567934,
      "learning_rate": 1.2427426418623377e-06,
      "loss": 0.0581,
      "step": 1068
    },
    {
      "epoch": 0.24323094425483505,
      "grad_norm": 0.9083897855186556,
      "learning_rate": 1.242729058641881e-06,
      "loss": 0.0353,
      "step": 1069
    },
    {
      "epoch": 0.2434584755403868,
      "grad_norm": 2.339042375493057,
      "learning_rate": 1.2427154627961737e-06,
      "loss": 0.1366,
      "step": 1070
    },
    {
      "epoch": 0.24368600682593858,
      "grad_norm": 2.1438385712601593,
      "learning_rate": 1.2427018543254935e-06,
      "loss": 0.082,
      "step": 1071
    },
    {
      "epoch": 0.24391353811149033,
      "grad_norm": 1.627313372349946,
      "learning_rate": 1.2426882332301187e-06,
      "loss": 0.076,
      "step": 1072
    },
    {
      "epoch": 0.24414106939704208,
      "grad_norm": 0.7791355914048339,
      "learning_rate": 1.2426745995103277e-06,
      "loss": 0.0309,
      "step": 1073
    },
    {
      "epoch": 0.24436860068259386,
      "grad_norm": 1.1486991259203154,
      "learning_rate": 1.242660953166399e-06,
      "loss": 0.0509,
      "step": 1074
    },
    {
      "epoch": 0.2445961319681456,
      "grad_norm": 1.7757375833942548,
      "learning_rate": 1.2426472941986117e-06,
      "loss": 0.0731,
      "step": 1075
    },
    {
      "epoch": 0.2448236632536974,
      "grad_norm": 1.4454374080664514,
      "learning_rate": 1.2426336226072449e-06,
      "loss": 0.0868,
      "step": 1076
    },
    {
      "epoch": 0.24505119453924915,
      "grad_norm": 1.0698078663784565,
      "learning_rate": 1.242619938392578e-06,
      "loss": 0.037,
      "step": 1077
    },
    {
      "epoch": 0.2452787258248009,
      "grad_norm": 1.4863388002827098,
      "learning_rate": 1.2426062415548907e-06,
      "loss": 0.0677,
      "step": 1078
    },
    {
      "epoch": 0.24550625711035268,
      "grad_norm": 0.8732937705027402,
      "learning_rate": 1.2425925320944628e-06,
      "loss": 0.0293,
      "step": 1079
    },
    {
      "epoch": 0.24573378839590443,
      "grad_norm": 1.2595288446902135,
      "learning_rate": 1.2425788100115747e-06,
      "loss": 0.056,
      "step": 1080
    },
    {
      "epoch": 0.2459613196814562,
      "grad_norm": 1.5998408970282165,
      "learning_rate": 1.2425650753065065e-06,
      "loss": 0.0764,
      "step": 1081
    },
    {
      "epoch": 0.24618885096700796,
      "grad_norm": 1.8027463727170587,
      "learning_rate": 1.2425513279795395e-06,
      "loss": 0.0615,
      "step": 1082
    },
    {
      "epoch": 0.24641638225255974,
      "grad_norm": 1.5654018901455387,
      "learning_rate": 1.2425375680309543e-06,
      "loss": 0.0661,
      "step": 1083
    },
    {
      "epoch": 0.2466439135381115,
      "grad_norm": 1.0676862677438599,
      "learning_rate": 1.2425237954610322e-06,
      "loss": 0.0469,
      "step": 1084
    },
    {
      "epoch": 0.24687144482366324,
      "grad_norm": 1.097337057406586,
      "learning_rate": 1.2425100102700547e-06,
      "loss": 0.0457,
      "step": 1085
    },
    {
      "epoch": 0.24709897610921502,
      "grad_norm": 1.7432427011380645,
      "learning_rate": 1.2424962124583033e-06,
      "loss": 0.0566,
      "step": 1086
    },
    {
      "epoch": 0.24732650739476678,
      "grad_norm": 1.2401111182610864,
      "learning_rate": 1.2424824020260603e-06,
      "loss": 0.0471,
      "step": 1087
    },
    {
      "epoch": 0.24755403868031856,
      "grad_norm": 1.6086762249686064,
      "learning_rate": 1.2424685789736077e-06,
      "loss": 0.0712,
      "step": 1088
    },
    {
      "epoch": 0.2477815699658703,
      "grad_norm": 2.223604715087674,
      "learning_rate": 1.2424547433012284e-06,
      "loss": 0.059,
      "step": 1089
    },
    {
      "epoch": 0.24800910125142206,
      "grad_norm": 1.763749275351197,
      "learning_rate": 1.2424408950092049e-06,
      "loss": 0.0768,
      "step": 1090
    },
    {
      "epoch": 0.24823663253697384,
      "grad_norm": 1.7828992657348108,
      "learning_rate": 1.2424270340978204e-06,
      "loss": 0.0794,
      "step": 1091
    },
    {
      "epoch": 0.2484641638225256,
      "grad_norm": 1.5001450767436246,
      "learning_rate": 1.2424131605673582e-06,
      "loss": 0.0693,
      "step": 1092
    },
    {
      "epoch": 0.24869169510807737,
      "grad_norm": 1.2301421279887537,
      "learning_rate": 1.2423992744181015e-06,
      "loss": 0.0512,
      "step": 1093
    },
    {
      "epoch": 0.24891922639362912,
      "grad_norm": 1.2844864578504145,
      "learning_rate": 1.2423853756503343e-06,
      "loss": 0.0447,
      "step": 1094
    },
    {
      "epoch": 0.24914675767918087,
      "grad_norm": 1.7196177827780137,
      "learning_rate": 1.2423714642643408e-06,
      "loss": 0.1182,
      "step": 1095
    },
    {
      "epoch": 0.24937428896473265,
      "grad_norm": 1.46657616590301,
      "learning_rate": 1.2423575402604051e-06,
      "loss": 0.0656,
      "step": 1096
    },
    {
      "epoch": 0.2496018202502844,
      "grad_norm": 1.326342830248352,
      "learning_rate": 1.2423436036388122e-06,
      "loss": 0.0544,
      "step": 1097
    },
    {
      "epoch": 0.24982935153583619,
      "grad_norm": 1.952008099521945,
      "learning_rate": 1.2423296543998465e-06,
      "loss": 0.078,
      "step": 1098
    },
    {
      "epoch": 0.25005688282138794,
      "grad_norm": 2.219926087555926,
      "learning_rate": 1.2423156925437932e-06,
      "loss": 0.0976,
      "step": 1099
    },
    {
      "epoch": 0.2502844141069397,
      "grad_norm": 1.4678969382750013,
      "learning_rate": 1.2423017180709376e-06,
      "loss": 0.0727,
      "step": 1100
    },
    {
      "epoch": 0.25051194539249144,
      "grad_norm": 1.9083840830954104,
      "learning_rate": 1.2422877309815656e-06,
      "loss": 0.0647,
      "step": 1101
    },
    {
      "epoch": 0.25073947667804325,
      "grad_norm": 1.3418972776139215,
      "learning_rate": 1.242273731275963e-06,
      "loss": 0.0616,
      "step": 1102
    },
    {
      "epoch": 0.250967007963595,
      "grad_norm": 1.0799686257073313,
      "learning_rate": 1.2422597189544155e-06,
      "loss": 0.0306,
      "step": 1103
    },
    {
      "epoch": 0.25119453924914675,
      "grad_norm": 1.4786340245230432,
      "learning_rate": 1.2422456940172101e-06,
      "loss": 0.0752,
      "step": 1104
    },
    {
      "epoch": 0.2514220705346985,
      "grad_norm": 1.3574468048142418,
      "learning_rate": 1.2422316564646331e-06,
      "loss": 0.0469,
      "step": 1105
    },
    {
      "epoch": 0.25164960182025026,
      "grad_norm": 1.0996737224918256,
      "learning_rate": 1.2422176062969713e-06,
      "loss": 0.06,
      "step": 1106
    },
    {
      "epoch": 0.25187713310580206,
      "grad_norm": 1.2079586241887055,
      "learning_rate": 1.2422035435145121e-06,
      "loss": 0.0417,
      "step": 1107
    },
    {
      "epoch": 0.2521046643913538,
      "grad_norm": 1.8484139155779635,
      "learning_rate": 1.2421894681175428e-06,
      "loss": 0.1201,
      "step": 1108
    },
    {
      "epoch": 0.25233219567690557,
      "grad_norm": 2.2964405416217977,
      "learning_rate": 1.2421753801063511e-06,
      "loss": 0.0661,
      "step": 1109
    },
    {
      "epoch": 0.2525597269624573,
      "grad_norm": 1.233798632970271,
      "learning_rate": 1.2421612794812248e-06,
      "loss": 0.0533,
      "step": 1110
    },
    {
      "epoch": 0.25278725824800913,
      "grad_norm": 0.8794754874897922,
      "learning_rate": 1.2421471662424525e-06,
      "loss": 0.0374,
      "step": 1111
    },
    {
      "epoch": 0.2530147895335609,
      "grad_norm": 1.1655828487104456,
      "learning_rate": 1.2421330403903222e-06,
      "loss": 0.0417,
      "step": 1112
    },
    {
      "epoch": 0.25324232081911263,
      "grad_norm": 1.978604856786128,
      "learning_rate": 1.2421189019251228e-06,
      "loss": 0.0613,
      "step": 1113
    },
    {
      "epoch": 0.2534698521046644,
      "grad_norm": 2.776603540550928,
      "learning_rate": 1.2421047508471433e-06,
      "loss": 0.0945,
      "step": 1114
    },
    {
      "epoch": 0.25369738339021614,
      "grad_norm": 1.4281147024591225,
      "learning_rate": 1.242090587156673e-06,
      "loss": 0.0634,
      "step": 1115
    },
    {
      "epoch": 0.25392491467576794,
      "grad_norm": 1.9687323288611094,
      "learning_rate": 1.242076410854001e-06,
      "loss": 0.0637,
      "step": 1116
    },
    {
      "epoch": 0.2541524459613197,
      "grad_norm": 0.9824968425418638,
      "learning_rate": 1.2420622219394174e-06,
      "loss": 0.0361,
      "step": 1117
    },
    {
      "epoch": 0.25437997724687145,
      "grad_norm": 1.1748756467082628,
      "learning_rate": 1.2420480204132117e-06,
      "loss": 0.0438,
      "step": 1118
    },
    {
      "epoch": 0.2546075085324232,
      "grad_norm": 1.1687970818584577,
      "learning_rate": 1.242033806275675e-06,
      "loss": 0.0517,
      "step": 1119
    },
    {
      "epoch": 0.25483503981797495,
      "grad_norm": 0.7243821856689469,
      "learning_rate": 1.2420195795270973e-06,
      "loss": 0.0341,
      "step": 1120
    },
    {
      "epoch": 0.25506257110352676,
      "grad_norm": 1.5334379290667297,
      "learning_rate": 1.2420053401677693e-06,
      "loss": 0.0834,
      "step": 1121
    },
    {
      "epoch": 0.2552901023890785,
      "grad_norm": 1.1475477659525273,
      "learning_rate": 1.241991088197982e-06,
      "loss": 0.0409,
      "step": 1122
    },
    {
      "epoch": 0.25551763367463026,
      "grad_norm": 1.0609745683681193,
      "learning_rate": 1.241976823618027e-06,
      "loss": 0.0521,
      "step": 1123
    },
    {
      "epoch": 0.255745164960182,
      "grad_norm": 2.0039923086463602,
      "learning_rate": 1.241962546428196e-06,
      "loss": 0.0912,
      "step": 1124
    },
    {
      "epoch": 0.25597269624573377,
      "grad_norm": 2.1894810646011846,
      "learning_rate": 1.24194825662878e-06,
      "loss": 0.0785,
      "step": 1125
    },
    {
      "epoch": 0.2562002275312856,
      "grad_norm": 1.4851813238564915,
      "learning_rate": 1.2419339542200715e-06,
      "loss": 0.0688,
      "step": 1126
    },
    {
      "epoch": 0.2564277588168373,
      "grad_norm": 1.2728572943407013,
      "learning_rate": 1.241919639202363e-06,
      "loss": 0.0794,
      "step": 1127
    },
    {
      "epoch": 0.2566552901023891,
      "grad_norm": 1.8082441837013492,
      "learning_rate": 1.2419053115759468e-06,
      "loss": 0.0677,
      "step": 1128
    },
    {
      "epoch": 0.25688282138794083,
      "grad_norm": 2.0609613364729267,
      "learning_rate": 1.2418909713411161e-06,
      "loss": 0.08,
      "step": 1129
    },
    {
      "epoch": 0.2571103526734926,
      "grad_norm": 1.533485951338072,
      "learning_rate": 1.2418766184981634e-06,
      "loss": 0.0606,
      "step": 1130
    },
    {
      "epoch": 0.2573378839590444,
      "grad_norm": 1.4347136829017464,
      "learning_rate": 1.2418622530473825e-06,
      "loss": 0.0825,
      "step": 1131
    },
    {
      "epoch": 0.25756541524459614,
      "grad_norm": 0.8595888516934124,
      "learning_rate": 1.2418478749890672e-06,
      "loss": 0.0351,
      "step": 1132
    },
    {
      "epoch": 0.2577929465301479,
      "grad_norm": 0.8988536535150867,
      "learning_rate": 1.2418334843235105e-06,
      "loss": 0.0465,
      "step": 1133
    },
    {
      "epoch": 0.25802047781569964,
      "grad_norm": 0.8536639563076862,
      "learning_rate": 1.2418190810510075e-06,
      "loss": 0.0258,
      "step": 1134
    },
    {
      "epoch": 0.2582480091012514,
      "grad_norm": 1.8139538056092865,
      "learning_rate": 1.2418046651718518e-06,
      "loss": 0.0545,
      "step": 1135
    },
    {
      "epoch": 0.2584755403868032,
      "grad_norm": 1.0709084305362149,
      "learning_rate": 1.2417902366863386e-06,
      "loss": 0.0488,
      "step": 1136
    },
    {
      "epoch": 0.25870307167235496,
      "grad_norm": 1.5965977494143468,
      "learning_rate": 1.2417757955947623e-06,
      "loss": 0.0549,
      "step": 1137
    },
    {
      "epoch": 0.2589306029579067,
      "grad_norm": 1.258938457199683,
      "learning_rate": 1.2417613418974187e-06,
      "loss": 0.0422,
      "step": 1138
    },
    {
      "epoch": 0.25915813424345846,
      "grad_norm": 0.9954800667569594,
      "learning_rate": 1.2417468755946025e-06,
      "loss": 0.0297,
      "step": 1139
    },
    {
      "epoch": 0.2593856655290102,
      "grad_norm": 1.410994342027921,
      "learning_rate": 1.2417323966866097e-06,
      "loss": 0.054,
      "step": 1140
    },
    {
      "epoch": 0.259613196814562,
      "grad_norm": 1.5107757792275114,
      "learning_rate": 1.2417179051737364e-06,
      "loss": 0.0589,
      "step": 1141
    },
    {
      "epoch": 0.25984072810011377,
      "grad_norm": 1.6398671824805653,
      "learning_rate": 1.2417034010562784e-06,
      "loss": 0.0666,
      "step": 1142
    },
    {
      "epoch": 0.2600682593856655,
      "grad_norm": 1.3520000022781709,
      "learning_rate": 1.2416888843345323e-06,
      "loss": 0.0416,
      "step": 1143
    },
    {
      "epoch": 0.2602957906712173,
      "grad_norm": 2.915495796493339,
      "learning_rate": 1.2416743550087951e-06,
      "loss": 0.1129,
      "step": 1144
    },
    {
      "epoch": 0.2605233219567691,
      "grad_norm": 1.2846258547817568,
      "learning_rate": 1.241659813079363e-06,
      "loss": 0.0579,
      "step": 1145
    },
    {
      "epoch": 0.26075085324232083,
      "grad_norm": 2.7427597460101936,
      "learning_rate": 1.2416452585465342e-06,
      "loss": 0.0702,
      "step": 1146
    },
    {
      "epoch": 0.2609783845278726,
      "grad_norm": 0.5884709313504546,
      "learning_rate": 1.2416306914106053e-06,
      "loss": 0.0237,
      "step": 1147
    },
    {
      "epoch": 0.26120591581342434,
      "grad_norm": 1.2939407540414039,
      "learning_rate": 1.2416161116718744e-06,
      "loss": 0.0583,
      "step": 1148
    },
    {
      "epoch": 0.2614334470989761,
      "grad_norm": 1.96584405463017,
      "learning_rate": 1.2416015193306397e-06,
      "loss": 0.0597,
      "step": 1149
    },
    {
      "epoch": 0.2616609783845279,
      "grad_norm": 1.367578620212531,
      "learning_rate": 1.241586914387199e-06,
      "loss": 0.0525,
      "step": 1150
    },
    {
      "epoch": 0.26188850967007965,
      "grad_norm": 1.5579301663665894,
      "learning_rate": 1.2415722968418508e-06,
      "loss": 0.0701,
      "step": 1151
    },
    {
      "epoch": 0.2621160409556314,
      "grad_norm": 2.113746394897606,
      "learning_rate": 1.2415576666948945e-06,
      "loss": 0.1158,
      "step": 1152
    },
    {
      "epoch": 0.26234357224118315,
      "grad_norm": 1.450032906329583,
      "learning_rate": 1.2415430239466283e-06,
      "loss": 0.0602,
      "step": 1153
    },
    {
      "epoch": 0.2625711035267349,
      "grad_norm": 1.3272413443266664,
      "learning_rate": 1.241528368597352e-06,
      "loss": 0.0523,
      "step": 1154
    },
    {
      "epoch": 0.2627986348122867,
      "grad_norm": 1.179039651442814,
      "learning_rate": 1.2415137006473649e-06,
      "loss": 0.0647,
      "step": 1155
    },
    {
      "epoch": 0.26302616609783847,
      "grad_norm": 1.145520426253407,
      "learning_rate": 1.241499020096967e-06,
      "loss": 0.0477,
      "step": 1156
    },
    {
      "epoch": 0.2632536973833902,
      "grad_norm": 1.328947872893636,
      "learning_rate": 1.2414843269464579e-06,
      "loss": 0.0452,
      "step": 1157
    },
    {
      "epoch": 0.26348122866894197,
      "grad_norm": 1.8843518385578133,
      "learning_rate": 1.2414696211961386e-06,
      "loss": 0.0603,
      "step": 1158
    },
    {
      "epoch": 0.2637087599544937,
      "grad_norm": 2.1467164256006943,
      "learning_rate": 1.2414549028463087e-06,
      "loss": 0.0908,
      "step": 1159
    },
    {
      "epoch": 0.26393629124004553,
      "grad_norm": 1.1918005997211691,
      "learning_rate": 1.2414401718972703e-06,
      "loss": 0.0576,
      "step": 1160
    },
    {
      "epoch": 0.2641638225255973,
      "grad_norm": 1.2786644068555166,
      "learning_rate": 1.2414254283493232e-06,
      "loss": 0.0416,
      "step": 1161
    },
    {
      "epoch": 0.26439135381114903,
      "grad_norm": 0.9452023930772737,
      "learning_rate": 1.2414106722027694e-06,
      "loss": 0.0537,
      "step": 1162
    },
    {
      "epoch": 0.2646188850967008,
      "grad_norm": 1.1756293508640332,
      "learning_rate": 1.2413959034579104e-06,
      "loss": 0.0511,
      "step": 1163
    },
    {
      "epoch": 0.26484641638225254,
      "grad_norm": 1.0472201568449635,
      "learning_rate": 1.2413811221150478e-06,
      "loss": 0.0442,
      "step": 1164
    },
    {
      "epoch": 0.26507394766780434,
      "grad_norm": 1.916314085543847,
      "learning_rate": 1.2413663281744843e-06,
      "loss": 0.0768,
      "step": 1165
    },
    {
      "epoch": 0.2653014789533561,
      "grad_norm": 1.0894771604167501,
      "learning_rate": 1.2413515216365216e-06,
      "loss": 0.0602,
      "step": 1166
    },
    {
      "epoch": 0.26552901023890785,
      "grad_norm": 1.00975683219884,
      "learning_rate": 1.2413367025014628e-06,
      "loss": 0.0432,
      "step": 1167
    },
    {
      "epoch": 0.2657565415244596,
      "grad_norm": 2.1765273948785913,
      "learning_rate": 1.2413218707696103e-06,
      "loss": 0.1186,
      "step": 1168
    },
    {
      "epoch": 0.26598407281001135,
      "grad_norm": 1.5268905455044306,
      "learning_rate": 1.2413070264412677e-06,
      "loss": 0.0671,
      "step": 1169
    },
    {
      "epoch": 0.26621160409556316,
      "grad_norm": 0.8198195830517153,
      "learning_rate": 1.2412921695167381e-06,
      "loss": 0.0303,
      "step": 1170
    },
    {
      "epoch": 0.2664391353811149,
      "grad_norm": 0.916743439302086,
      "learning_rate": 1.2412772999963253e-06,
      "loss": 0.0519,
      "step": 1171
    },
    {
      "epoch": 0.26666666666666666,
      "grad_norm": 0.9592599841051566,
      "learning_rate": 1.2412624178803332e-06,
      "loss": 0.0465,
      "step": 1172
    },
    {
      "epoch": 0.2668941979522184,
      "grad_norm": 1.2610976325050403,
      "learning_rate": 1.2412475231690656e-06,
      "loss": 0.0345,
      "step": 1173
    },
    {
      "epoch": 0.26712172923777017,
      "grad_norm": 1.5016024974886808,
      "learning_rate": 1.2412326158628275e-06,
      "loss": 0.0714,
      "step": 1174
    },
    {
      "epoch": 0.267349260523322,
      "grad_norm": 1.4314762978749895,
      "learning_rate": 1.2412176959619232e-06,
      "loss": 0.0591,
      "step": 1175
    },
    {
      "epoch": 0.2675767918088737,
      "grad_norm": 2.2495091900368562,
      "learning_rate": 1.2412027634666578e-06,
      "loss": 0.1059,
      "step": 1176
    },
    {
      "epoch": 0.2678043230944255,
      "grad_norm": 1.706282565066305,
      "learning_rate": 1.2411878183773366e-06,
      "loss": 0.0665,
      "step": 1177
    },
    {
      "epoch": 0.26803185437997723,
      "grad_norm": 0.7041103017092823,
      "learning_rate": 1.2411728606942647e-06,
      "loss": 0.0218,
      "step": 1178
    },
    {
      "epoch": 0.26825938566552904,
      "grad_norm": 0.9725397335622218,
      "learning_rate": 1.241157890417748e-06,
      "loss": 0.0452,
      "step": 1179
    },
    {
      "epoch": 0.2684869169510808,
      "grad_norm": 1.9030708757385315,
      "learning_rate": 1.2411429075480923e-06,
      "loss": 0.0827,
      "step": 1180
    },
    {
      "epoch": 0.26871444823663254,
      "grad_norm": 1.1973351649414545,
      "learning_rate": 1.2411279120856042e-06,
      "loss": 0.0476,
      "step": 1181
    },
    {
      "epoch": 0.2689419795221843,
      "grad_norm": 1.1061135820635588,
      "learning_rate": 1.24111290403059e-06,
      "loss": 0.0387,
      "step": 1182
    },
    {
      "epoch": 0.26916951080773605,
      "grad_norm": 1.1523226678516665,
      "learning_rate": 1.2410978833833564e-06,
      "loss": 0.0614,
      "step": 1183
    },
    {
      "epoch": 0.26939704209328785,
      "grad_norm": 1.0514874828378626,
      "learning_rate": 1.2410828501442104e-06,
      "loss": 0.0439,
      "step": 1184
    },
    {
      "epoch": 0.2696245733788396,
      "grad_norm": 1.5362061949507542,
      "learning_rate": 1.2410678043134591e-06,
      "loss": 0.0804,
      "step": 1185
    },
    {
      "epoch": 0.26985210466439136,
      "grad_norm": 1.8173832515444213,
      "learning_rate": 1.2410527458914103e-06,
      "loss": 0.063,
      "step": 1186
    },
    {
      "epoch": 0.2700796359499431,
      "grad_norm": 1.556320593990832,
      "learning_rate": 1.2410376748783714e-06,
      "loss": 0.0795,
      "step": 1187
    },
    {
      "epoch": 0.27030716723549486,
      "grad_norm": 1.321348596107485,
      "learning_rate": 1.241022591274651e-06,
      "loss": 0.0802,
      "step": 1188
    },
    {
      "epoch": 0.27053469852104667,
      "grad_norm": 1.0652846616165823,
      "learning_rate": 1.2410074950805567e-06,
      "loss": 0.036,
      "step": 1189
    },
    {
      "epoch": 0.2707622298065984,
      "grad_norm": 1.2080698971206665,
      "learning_rate": 1.2409923862963973e-06,
      "loss": 0.0488,
      "step": 1190
    },
    {
      "epoch": 0.27098976109215017,
      "grad_norm": 1.0230330657283313,
      "learning_rate": 1.240977264922482e-06,
      "loss": 0.0408,
      "step": 1191
    },
    {
      "epoch": 0.2712172923777019,
      "grad_norm": 1.243454259162499,
      "learning_rate": 1.2409621309591195e-06,
      "loss": 0.0359,
      "step": 1192
    },
    {
      "epoch": 0.2714448236632537,
      "grad_norm": 1.3001395180964366,
      "learning_rate": 1.2409469844066188e-06,
      "loss": 0.0454,
      "step": 1193
    },
    {
      "epoch": 0.2716723549488055,
      "grad_norm": 1.4147947246201094,
      "learning_rate": 1.2409318252652899e-06,
      "loss": 0.0673,
      "step": 1194
    },
    {
      "epoch": 0.27189988623435724,
      "grad_norm": 2.0607810868090977,
      "learning_rate": 1.2409166535354428e-06,
      "loss": 0.1099,
      "step": 1195
    },
    {
      "epoch": 0.272127417519909,
      "grad_norm": 0.8856660147431201,
      "learning_rate": 1.2409014692173872e-06,
      "loss": 0.037,
      "step": 1196
    },
    {
      "epoch": 0.27235494880546074,
      "grad_norm": 0.8588236308520142,
      "learning_rate": 1.240886272311433e-06,
      "loss": 0.0317,
      "step": 1197
    },
    {
      "epoch": 0.2725824800910125,
      "grad_norm": 1.2966972282013072,
      "learning_rate": 1.240871062817892e-06,
      "loss": 0.0519,
      "step": 1198
    },
    {
      "epoch": 0.2728100113765643,
      "grad_norm": 1.0141950682821237,
      "learning_rate": 1.240855840737074e-06,
      "loss": 0.0362,
      "step": 1199
    },
    {
      "epoch": 0.27303754266211605,
      "grad_norm": 1.1327532340524933,
      "learning_rate": 1.2408406060692909e-06,
      "loss": 0.0411,
      "step": 1200
    },
    {
      "epoch": 0.2732650739476678,
      "grad_norm": 38.94813584323084,
      "learning_rate": 1.2408253588148532e-06,
      "loss": 0.4954,
      "step": 1201
    },
    {
      "epoch": 0.27349260523321955,
      "grad_norm": 1.3419172753176944,
      "learning_rate": 1.2408100989740735e-06,
      "loss": 0.0629,
      "step": 1202
    },
    {
      "epoch": 0.2737201365187713,
      "grad_norm": 1.164418605632623,
      "learning_rate": 1.2407948265472628e-06,
      "loss": 0.0327,
      "step": 1203
    },
    {
      "epoch": 0.2739476678043231,
      "grad_norm": 1.2988536193125668,
      "learning_rate": 1.2407795415347336e-06,
      "loss": 0.0567,
      "step": 1204
    },
    {
      "epoch": 0.27417519908987487,
      "grad_norm": 1.361893473954927,
      "learning_rate": 1.2407642439367986e-06,
      "loss": 0.0438,
      "step": 1205
    },
    {
      "epoch": 0.2744027303754266,
      "grad_norm": 1.0391897172547626,
      "learning_rate": 1.24074893375377e-06,
      "loss": 0.0557,
      "step": 1206
    },
    {
      "epoch": 0.27463026166097837,
      "grad_norm": 0.9962942412430917,
      "learning_rate": 1.2407336109859607e-06,
      "loss": 0.0371,
      "step": 1207
    },
    {
      "epoch": 0.2748577929465301,
      "grad_norm": 1.215727870574286,
      "learning_rate": 1.2407182756336844e-06,
      "loss": 0.0437,
      "step": 1208
    },
    {
      "epoch": 0.27508532423208193,
      "grad_norm": 1.8508042963184421,
      "learning_rate": 1.240702927697254e-06,
      "loss": 0.0626,
      "step": 1209
    },
    {
      "epoch": 0.2753128555176337,
      "grad_norm": 1.8022236633570414,
      "learning_rate": 1.2406875671769837e-06,
      "loss": 0.0584,
      "step": 1210
    },
    {
      "epoch": 0.27554038680318543,
      "grad_norm": 1.606815955580643,
      "learning_rate": 1.2406721940731866e-06,
      "loss": 0.0599,
      "step": 1211
    },
    {
      "epoch": 0.2757679180887372,
      "grad_norm": 1.0652074757978982,
      "learning_rate": 1.2406568083861776e-06,
      "loss": 0.0531,
      "step": 1212
    },
    {
      "epoch": 0.27599544937428894,
      "grad_norm": 1.4187270182072722,
      "learning_rate": 1.2406414101162708e-06,
      "loss": 0.0528,
      "step": 1213
    },
    {
      "epoch": 0.27622298065984074,
      "grad_norm": 1.0888371326459914,
      "learning_rate": 1.2406259992637815e-06,
      "loss": 0.0552,
      "step": 1214
    },
    {
      "epoch": 0.2764505119453925,
      "grad_norm": 2.1125996368288553,
      "learning_rate": 1.240610575829024e-06,
      "loss": 0.0735,
      "step": 1215
    },
    {
      "epoch": 0.27667804323094425,
      "grad_norm": 1.288876395538611,
      "learning_rate": 1.2405951398123136e-06,
      "loss": 0.0553,
      "step": 1216
    },
    {
      "epoch": 0.276905574516496,
      "grad_norm": 1.711303610363005,
      "learning_rate": 1.2405796912139662e-06,
      "loss": 0.0673,
      "step": 1217
    },
    {
      "epoch": 0.2771331058020478,
      "grad_norm": 1.3330966398511392,
      "learning_rate": 1.240564230034297e-06,
      "loss": 0.0347,
      "step": 1218
    },
    {
      "epoch": 0.27736063708759956,
      "grad_norm": 0.9030680288819154,
      "learning_rate": 1.2405487562736226e-06,
      "loss": 0.0365,
      "step": 1219
    },
    {
      "epoch": 0.2775881683731513,
      "grad_norm": 1.20973479021967,
      "learning_rate": 1.240533269932259e-06,
      "loss": 0.0647,
      "step": 1220
    },
    {
      "epoch": 0.27781569965870306,
      "grad_norm": 1.2330537480211692,
      "learning_rate": 1.2405177710105223e-06,
      "loss": 0.0415,
      "step": 1221
    },
    {
      "epoch": 0.2780432309442548,
      "grad_norm": 1.4447131783576561,
      "learning_rate": 1.24050225950873e-06,
      "loss": 0.04,
      "step": 1222
    },
    {
      "epoch": 0.2782707622298066,
      "grad_norm": 1.5348321099487485,
      "learning_rate": 1.2404867354271984e-06,
      "loss": 0.0699,
      "step": 1223
    },
    {
      "epoch": 0.2784982935153584,
      "grad_norm": 1.103797642514096,
      "learning_rate": 1.2404711987662452e-06,
      "loss": 0.0501,
      "step": 1224
    },
    {
      "epoch": 0.2787258248009101,
      "grad_norm": 1.8543277255058668,
      "learning_rate": 1.240455649526188e-06,
      "loss": 0.0779,
      "step": 1225
    },
    {
      "epoch": 0.2789533560864619,
      "grad_norm": 1.461515751555683,
      "learning_rate": 1.2404400877073446e-06,
      "loss": 0.0589,
      "step": 1226
    },
    {
      "epoch": 0.27918088737201363,
      "grad_norm": 1.7282210636065953,
      "learning_rate": 1.2404245133100328e-06,
      "loss": 0.0616,
      "step": 1227
    },
    {
      "epoch": 0.27940841865756544,
      "grad_norm": 0.9274643279367238,
      "learning_rate": 1.240408926334571e-06,
      "loss": 0.0259,
      "step": 1228
    },
    {
      "epoch": 0.2796359499431172,
      "grad_norm": 1.033501452595113,
      "learning_rate": 1.240393326781278e-06,
      "loss": 0.0388,
      "step": 1229
    },
    {
      "epoch": 0.27986348122866894,
      "grad_norm": 1.2277681625095047,
      "learning_rate": 1.2403777146504722e-06,
      "loss": 0.0417,
      "step": 1230
    },
    {
      "epoch": 0.2800910125142207,
      "grad_norm": 1.6056103902463323,
      "learning_rate": 1.240362089942473e-06,
      "loss": 0.0869,
      "step": 1231
    },
    {
      "epoch": 0.28031854379977245,
      "grad_norm": 0.9379340266438139,
      "learning_rate": 1.2403464526575997e-06,
      "loss": 0.0288,
      "step": 1232
    },
    {
      "epoch": 0.28054607508532425,
      "grad_norm": 1.791400262370738,
      "learning_rate": 1.240330802796172e-06,
      "loss": 0.0494,
      "step": 1233
    },
    {
      "epoch": 0.280773606370876,
      "grad_norm": 1.5180897393064074,
      "learning_rate": 1.2403151403585093e-06,
      "loss": 0.0555,
      "step": 1234
    },
    {
      "epoch": 0.28100113765642776,
      "grad_norm": 1.3796072286824306,
      "learning_rate": 1.240299465344932e-06,
      "loss": 0.0405,
      "step": 1235
    },
    {
      "epoch": 0.2812286689419795,
      "grad_norm": 0.968124324669557,
      "learning_rate": 1.2402837777557608e-06,
      "loss": 0.0462,
      "step": 1236
    },
    {
      "epoch": 0.28145620022753126,
      "grad_norm": 2.8732703815868157,
      "learning_rate": 1.240268077591316e-06,
      "loss": 0.1432,
      "step": 1237
    },
    {
      "epoch": 0.28168373151308307,
      "grad_norm": 1.7096825525845536,
      "learning_rate": 1.2402523648519184e-06,
      "loss": 0.0709,
      "step": 1238
    },
    {
      "epoch": 0.2819112627986348,
      "grad_norm": 1.4537193421874715,
      "learning_rate": 1.2402366395378892e-06,
      "loss": 0.0477,
      "step": 1239
    },
    {
      "epoch": 0.2821387940841866,
      "grad_norm": 1.0525576505096776,
      "learning_rate": 1.24022090164955e-06,
      "loss": 0.0381,
      "step": 1240
    },
    {
      "epoch": 0.2823663253697383,
      "grad_norm": 2.0691434586350135,
      "learning_rate": 1.240205151187222e-06,
      "loss": 0.0749,
      "step": 1241
    },
    {
      "epoch": 0.2825938566552901,
      "grad_norm": 1.4855492490109166,
      "learning_rate": 1.2401893881512278e-06,
      "loss": 0.0486,
      "step": 1242
    },
    {
      "epoch": 0.2828213879408419,
      "grad_norm": 2.3507672202858427,
      "learning_rate": 1.240173612541889e-06,
      "loss": 0.1187,
      "step": 1243
    },
    {
      "epoch": 0.28304891922639364,
      "grad_norm": 1.4271939145537942,
      "learning_rate": 1.2401578243595281e-06,
      "loss": 0.0459,
      "step": 1244
    },
    {
      "epoch": 0.2832764505119454,
      "grad_norm": 0.7153929012800043,
      "learning_rate": 1.2401420236044678e-06,
      "loss": 0.0203,
      "step": 1245
    },
    {
      "epoch": 0.28350398179749714,
      "grad_norm": 1.8926578555618896,
      "learning_rate": 1.2401262102770314e-06,
      "loss": 0.0939,
      "step": 1246
    },
    {
      "epoch": 0.2837315130830489,
      "grad_norm": 1.5820745105272767,
      "learning_rate": 1.2401103843775416e-06,
      "loss": 0.0885,
      "step": 1247
    },
    {
      "epoch": 0.2839590443686007,
      "grad_norm": 1.3120392754023362,
      "learning_rate": 1.240094545906322e-06,
      "loss": 0.0629,
      "step": 1248
    },
    {
      "epoch": 0.28418657565415245,
      "grad_norm": 1.2780609773747653,
      "learning_rate": 1.2400786948636966e-06,
      "loss": 0.0669,
      "step": 1249
    },
    {
      "epoch": 0.2844141069397042,
      "grad_norm": 1.4394753485796945,
      "learning_rate": 1.2400628312499892e-06,
      "loss": 0.0457,
      "step": 1250
    },
    {
      "epoch": 0.28464163822525596,
      "grad_norm": 1.4902299382519306,
      "learning_rate": 1.2400469550655239e-06,
      "loss": 0.0788,
      "step": 1251
    },
    {
      "epoch": 0.28486916951080776,
      "grad_norm": 2.1373698726270614,
      "learning_rate": 1.240031066310625e-06,
      "loss": 0.0709,
      "step": 1252
    },
    {
      "epoch": 0.2850967007963595,
      "grad_norm": 2.6364110769644467,
      "learning_rate": 1.2400151649856176e-06,
      "loss": 0.1474,
      "step": 1253
    },
    {
      "epoch": 0.28532423208191127,
      "grad_norm": 1.54489334981108,
      "learning_rate": 1.2399992510908266e-06,
      "loss": 0.0681,
      "step": 1254
    },
    {
      "epoch": 0.285551763367463,
      "grad_norm": 1.2035560270921895,
      "learning_rate": 1.2399833246265772e-06,
      "loss": 0.0624,
      "step": 1255
    },
    {
      "epoch": 0.28577929465301477,
      "grad_norm": 2.3200664566107427,
      "learning_rate": 1.2399673855931951e-06,
      "loss": 0.0948,
      "step": 1256
    },
    {
      "epoch": 0.2860068259385666,
      "grad_norm": 0.911546303051303,
      "learning_rate": 1.2399514339910058e-06,
      "loss": 0.0371,
      "step": 1257
    },
    {
      "epoch": 0.28623435722411833,
      "grad_norm": 1.1459525470306917,
      "learning_rate": 1.2399354698203353e-06,
      "loss": 0.063,
      "step": 1258
    },
    {
      "epoch": 0.2864618885096701,
      "grad_norm": 5.90970905115882,
      "learning_rate": 1.2399194930815103e-06,
      "loss": 0.2573,
      "step": 1259
    },
    {
      "epoch": 0.28668941979522183,
      "grad_norm": 1.6075560430361318,
      "learning_rate": 1.2399035037748567e-06,
      "loss": 0.0712,
      "step": 1260
    },
    {
      "epoch": 0.2869169510807736,
      "grad_norm": 1.059504335911741,
      "learning_rate": 1.2398875019007017e-06,
      "loss": 0.0336,
      "step": 1261
    },
    {
      "epoch": 0.2871444823663254,
      "grad_norm": 1.0386465638630533,
      "learning_rate": 1.2398714874593722e-06,
      "loss": 0.028,
      "step": 1262
    },
    {
      "epoch": 0.28737201365187715,
      "grad_norm": 1.2034325585223464,
      "learning_rate": 1.2398554604511958e-06,
      "loss": 0.0426,
      "step": 1263
    },
    {
      "epoch": 0.2875995449374289,
      "grad_norm": 1.130614264631673,
      "learning_rate": 1.2398394208764998e-06,
      "loss": 0.0449,
      "step": 1264
    },
    {
      "epoch": 0.28782707622298065,
      "grad_norm": 1.8759349581624183,
      "learning_rate": 1.239823368735612e-06,
      "loss": 0.0738,
      "step": 1265
    },
    {
      "epoch": 0.2880546075085324,
      "grad_norm": 1.534877012833876,
      "learning_rate": 1.2398073040288605e-06,
      "loss": 0.0565,
      "step": 1266
    },
    {
      "epoch": 0.2882821387940842,
      "grad_norm": 1.1199756497261975,
      "learning_rate": 1.2397912267565738e-06,
      "loss": 0.0434,
      "step": 1267
    },
    {
      "epoch": 0.28850967007963596,
      "grad_norm": 1.2767125637243415,
      "learning_rate": 1.2397751369190803e-06,
      "loss": 0.053,
      "step": 1268
    },
    {
      "epoch": 0.2887372013651877,
      "grad_norm": 1.6866930161803586,
      "learning_rate": 1.2397590345167088e-06,
      "loss": 0.0977,
      "step": 1269
    },
    {
      "epoch": 0.28896473265073946,
      "grad_norm": 1.2795958593835728,
      "learning_rate": 1.2397429195497887e-06,
      "loss": 0.0435,
      "step": 1270
    },
    {
      "epoch": 0.2891922639362912,
      "grad_norm": 1.2653384358613777,
      "learning_rate": 1.239726792018649e-06,
      "loss": 0.0366,
      "step": 1271
    },
    {
      "epoch": 0.289419795221843,
      "grad_norm": 1.175013233181025,
      "learning_rate": 1.2397106519236197e-06,
      "loss": 0.0589,
      "step": 1272
    },
    {
      "epoch": 0.2896473265073948,
      "grad_norm": 0.8118499461560013,
      "learning_rate": 1.2396944992650303e-06,
      "loss": 0.0341,
      "step": 1273
    },
    {
      "epoch": 0.2898748577929465,
      "grad_norm": 1.7245088180198505,
      "learning_rate": 1.2396783340432111e-06,
      "loss": 0.0712,
      "step": 1274
    },
    {
      "epoch": 0.2901023890784983,
      "grad_norm": 1.453392643848313,
      "learning_rate": 1.2396621562584925e-06,
      "loss": 0.0552,
      "step": 1275
    },
    {
      "epoch": 0.29032992036405003,
      "grad_norm": 0.9686308053813132,
      "learning_rate": 1.2396459659112052e-06,
      "loss": 0.0298,
      "step": 1276
    },
    {
      "epoch": 0.29055745164960184,
      "grad_norm": 1.2190161105816313,
      "learning_rate": 1.23962976300168e-06,
      "loss": 0.0657,
      "step": 1277
    },
    {
      "epoch": 0.2907849829351536,
      "grad_norm": 1.4400910874681545,
      "learning_rate": 1.2396135475302478e-06,
      "loss": 0.0689,
      "step": 1278
    },
    {
      "epoch": 0.29101251422070534,
      "grad_norm": 1.6072456813020977,
      "learning_rate": 1.2395973194972406e-06,
      "loss": 0.0519,
      "step": 1279
    },
    {
      "epoch": 0.2912400455062571,
      "grad_norm": 0.9134133498246579,
      "learning_rate": 1.2395810789029898e-06,
      "loss": 0.0305,
      "step": 1280
    },
    {
      "epoch": 0.29146757679180885,
      "grad_norm": 1.4368452567354464,
      "learning_rate": 1.2395648257478271e-06,
      "loss": 0.0842,
      "step": 1281
    },
    {
      "epoch": 0.29169510807736065,
      "grad_norm": 1.2627048123993603,
      "learning_rate": 1.239548560032085e-06,
      "loss": 0.0556,
      "step": 1282
    },
    {
      "epoch": 0.2919226393629124,
      "grad_norm": 1.8157780106000843,
      "learning_rate": 1.239532281756096e-06,
      "loss": 0.0812,
      "step": 1283
    },
    {
      "epoch": 0.29215017064846416,
      "grad_norm": 1.7031472200076767,
      "learning_rate": 1.2395159909201924e-06,
      "loss": 0.1025,
      "step": 1284
    },
    {
      "epoch": 0.2923777019340159,
      "grad_norm": 1.4512078193177647,
      "learning_rate": 1.2394996875247075e-06,
      "loss": 0.0878,
      "step": 1285
    },
    {
      "epoch": 0.2926052332195677,
      "grad_norm": 1.1558527774450407,
      "learning_rate": 1.2394833715699743e-06,
      "loss": 0.0517,
      "step": 1286
    },
    {
      "epoch": 0.29283276450511947,
      "grad_norm": 1.1961425724018928,
      "learning_rate": 1.2394670430563264e-06,
      "loss": 0.0822,
      "step": 1287
    },
    {
      "epoch": 0.2930602957906712,
      "grad_norm": 1.1257477613693543,
      "learning_rate": 1.2394507019840974e-06,
      "loss": 0.0433,
      "step": 1288
    },
    {
      "epoch": 0.293287827076223,
      "grad_norm": 1.7490981885592791,
      "learning_rate": 1.2394343483536215e-06,
      "loss": 0.0731,
      "step": 1289
    },
    {
      "epoch": 0.2935153583617747,
      "grad_norm": 0.7736925600695324,
      "learning_rate": 1.2394179821652326e-06,
      "loss": 0.048,
      "step": 1290
    },
    {
      "epoch": 0.29374288964732653,
      "grad_norm": 1.2742121789528693,
      "learning_rate": 1.2394016034192654e-06,
      "loss": 0.0523,
      "step": 1291
    },
    {
      "epoch": 0.2939704209328783,
      "grad_norm": 1.4246615440711465,
      "learning_rate": 1.2393852121160548e-06,
      "loss": 0.0813,
      "step": 1292
    },
    {
      "epoch": 0.29419795221843004,
      "grad_norm": 1.656527882332843,
      "learning_rate": 1.2393688082559357e-06,
      "loss": 0.0699,
      "step": 1293
    },
    {
      "epoch": 0.2944254835039818,
      "grad_norm": 1.2504233198660668,
      "learning_rate": 1.2393523918392433e-06,
      "loss": 0.0457,
      "step": 1294
    },
    {
      "epoch": 0.29465301478953354,
      "grad_norm": 2.1835557352018458,
      "learning_rate": 1.2393359628663133e-06,
      "loss": 0.1184,
      "step": 1295
    },
    {
      "epoch": 0.29488054607508535,
      "grad_norm": 1.1347886652336971,
      "learning_rate": 1.239319521337481e-06,
      "loss": 0.0655,
      "step": 1296
    },
    {
      "epoch": 0.2951080773606371,
      "grad_norm": 1.1763338033692377,
      "learning_rate": 1.2393030672530828e-06,
      "loss": 0.0568,
      "step": 1297
    },
    {
      "epoch": 0.29533560864618885,
      "grad_norm": 1.3990303438853744,
      "learning_rate": 1.239286600613455e-06,
      "loss": 0.041,
      "step": 1298
    },
    {
      "epoch": 0.2955631399317406,
      "grad_norm": 0.9401022615309382,
      "learning_rate": 1.2392701214189343e-06,
      "loss": 0.0349,
      "step": 1299
    },
    {
      "epoch": 0.29579067121729236,
      "grad_norm": 2.3948134916737516,
      "learning_rate": 1.2392536296698571e-06,
      "loss": 0.1472,
      "step": 1300
    },
    {
      "epoch": 0.29601820250284416,
      "grad_norm": 1.320538326644519,
      "learning_rate": 1.2392371253665605e-06,
      "loss": 0.0767,
      "step": 1301
    },
    {
      "epoch": 0.2962457337883959,
      "grad_norm": 1.354776649937033,
      "learning_rate": 1.2392206085093823e-06,
      "loss": 0.0584,
      "step": 1302
    },
    {
      "epoch": 0.29647326507394767,
      "grad_norm": 1.4302761520764427,
      "learning_rate": 1.2392040790986594e-06,
      "loss": 0.0823,
      "step": 1303
    },
    {
      "epoch": 0.2967007963594994,
      "grad_norm": 1.4047156580871927,
      "learning_rate": 1.2391875371347303e-06,
      "loss": 0.0549,
      "step": 1304
    },
    {
      "epoch": 0.29692832764505117,
      "grad_norm": 0.7970562047989252,
      "learning_rate": 1.2391709826179327e-06,
      "loss": 0.0336,
      "step": 1305
    },
    {
      "epoch": 0.297155858930603,
      "grad_norm": 1.3727411496357969,
      "learning_rate": 1.239154415548605e-06,
      "loss": 0.0392,
      "step": 1306
    },
    {
      "epoch": 0.29738339021615473,
      "grad_norm": 0.9054264955476324,
      "learning_rate": 1.2391378359270859e-06,
      "loss": 0.0389,
      "step": 1307
    },
    {
      "epoch": 0.2976109215017065,
      "grad_norm": 1.814490926829392,
      "learning_rate": 1.2391212437537138e-06,
      "loss": 0.072,
      "step": 1308
    },
    {
      "epoch": 0.29783845278725823,
      "grad_norm": 1.7618287946971354,
      "learning_rate": 1.2391046390288287e-06,
      "loss": 0.058,
      "step": 1309
    },
    {
      "epoch": 0.29806598407281,
      "grad_norm": 1.5269715641275399,
      "learning_rate": 1.2390880217527692e-06,
      "loss": 0.0656,
      "step": 1310
    },
    {
      "epoch": 0.2982935153583618,
      "grad_norm": 1.262642874805734,
      "learning_rate": 1.2390713919258752e-06,
      "loss": 0.0532,
      "step": 1311
    },
    {
      "epoch": 0.29852104664391355,
      "grad_norm": 1.9617860637916495,
      "learning_rate": 1.2390547495484866e-06,
      "loss": 0.1163,
      "step": 1312
    },
    {
      "epoch": 0.2987485779294653,
      "grad_norm": 1.288057411790914,
      "learning_rate": 1.2390380946209436e-06,
      "loss": 0.0508,
      "step": 1313
    },
    {
      "epoch": 0.29897610921501705,
      "grad_norm": 1.1943768298905832,
      "learning_rate": 1.2390214271435863e-06,
      "loss": 0.0388,
      "step": 1314
    },
    {
      "epoch": 0.2992036405005688,
      "grad_norm": 1.291453023419144,
      "learning_rate": 1.2390047471167557e-06,
      "loss": 0.0459,
      "step": 1315
    },
    {
      "epoch": 0.2994311717861206,
      "grad_norm": 1.014250520116699,
      "learning_rate": 1.2389880545407926e-06,
      "loss": 0.0415,
      "step": 1316
    },
    {
      "epoch": 0.29965870307167236,
      "grad_norm": 0.8604020785970597,
      "learning_rate": 1.2389713494160379e-06,
      "loss": 0.0256,
      "step": 1317
    },
    {
      "epoch": 0.2998862343572241,
      "grad_norm": 1.5717854535146285,
      "learning_rate": 1.2389546317428335e-06,
      "loss": 0.0659,
      "step": 1318
    },
    {
      "epoch": 0.30011376564277586,
      "grad_norm": 1.1712963748867264,
      "learning_rate": 1.2389379015215208e-06,
      "loss": 0.0576,
      "step": 1319
    },
    {
      "epoch": 0.3003412969283277,
      "grad_norm": 1.9673409422404795,
      "learning_rate": 1.2389211587524416e-06,
      "loss": 0.0724,
      "step": 1320
    },
    {
      "epoch": 0.3005688282138794,
      "grad_norm": 1.7254424987946275,
      "learning_rate": 1.2389044034359383e-06,
      "loss": 0.0667,
      "step": 1321
    },
    {
      "epoch": 0.3007963594994312,
      "grad_norm": 1.7304834437682375,
      "learning_rate": 1.2388876355723533e-06,
      "loss": 0.0669,
      "step": 1322
    },
    {
      "epoch": 0.30102389078498293,
      "grad_norm": 1.5970096808225174,
      "learning_rate": 1.2388708551620295e-06,
      "loss": 0.0606,
      "step": 1323
    },
    {
      "epoch": 0.3012514220705347,
      "grad_norm": 1.4122576326239549,
      "learning_rate": 1.2388540622053095e-06,
      "loss": 0.0521,
      "step": 1324
    },
    {
      "epoch": 0.3014789533560865,
      "grad_norm": 0.955203381580325,
      "learning_rate": 1.2388372567025367e-06,
      "loss": 0.0338,
      "step": 1325
    },
    {
      "epoch": 0.30170648464163824,
      "grad_norm": 1.8026618764076134,
      "learning_rate": 1.2388204386540546e-06,
      "loss": 0.0667,
      "step": 1326
    },
    {
      "epoch": 0.30193401592719,
      "grad_norm": 1.4712519094439878,
      "learning_rate": 1.238803608060207e-06,
      "loss": 0.0573,
      "step": 1327
    },
    {
      "epoch": 0.30216154721274174,
      "grad_norm": 1.7736677880474818,
      "learning_rate": 1.2387867649213376e-06,
      "loss": 0.066,
      "step": 1328
    },
    {
      "epoch": 0.3023890784982935,
      "grad_norm": 2.3477972521623465,
      "learning_rate": 1.2387699092377908e-06,
      "loss": 0.1076,
      "step": 1329
    },
    {
      "epoch": 0.3026166097838453,
      "grad_norm": 1.1616522138628205,
      "learning_rate": 1.2387530410099113e-06,
      "loss": 0.0439,
      "step": 1330
    },
    {
      "epoch": 0.30284414106939705,
      "grad_norm": 1.4363882653748854,
      "learning_rate": 1.2387361602380436e-06,
      "loss": 0.0571,
      "step": 1331
    },
    {
      "epoch": 0.3030716723549488,
      "grad_norm": 0.7445259748127024,
      "learning_rate": 1.238719266922533e-06,
      "loss": 0.0254,
      "step": 1332
    },
    {
      "epoch": 0.30329920364050056,
      "grad_norm": 1.5848968542889414,
      "learning_rate": 1.238702361063724e-06,
      "loss": 0.0656,
      "step": 1333
    },
    {
      "epoch": 0.3035267349260523,
      "grad_norm": 6.520543814327786,
      "learning_rate": 1.2386854426619633e-06,
      "loss": 0.168,
      "step": 1334
    },
    {
      "epoch": 0.3037542662116041,
      "grad_norm": 1.1879669128675703,
      "learning_rate": 1.2386685117175956e-06,
      "loss": 0.0414,
      "step": 1335
    },
    {
      "epoch": 0.30398179749715587,
      "grad_norm": 1.627644346798232,
      "learning_rate": 1.2386515682309676e-06,
      "loss": 0.0524,
      "step": 1336
    },
    {
      "epoch": 0.3042093287827076,
      "grad_norm": 1.5535932914971018,
      "learning_rate": 1.2386346122024253e-06,
      "loss": 0.0783,
      "step": 1337
    },
    {
      "epoch": 0.3044368600682594,
      "grad_norm": 1.9629237703618092,
      "learning_rate": 1.2386176436323154e-06,
      "loss": 0.0682,
      "step": 1338
    },
    {
      "epoch": 0.3046643913538111,
      "grad_norm": 1.1645585598673585,
      "learning_rate": 1.2386006625209847e-06,
      "loss": 0.0459,
      "step": 1339
    },
    {
      "epoch": 0.30489192263936293,
      "grad_norm": 1.7630197143088098,
      "learning_rate": 1.2385836688687802e-06,
      "loss": 0.0934,
      "step": 1340
    },
    {
      "epoch": 0.3051194539249147,
      "grad_norm": 1.6191880044317468,
      "learning_rate": 1.2385666626760493e-06,
      "loss": 0.0633,
      "step": 1341
    },
    {
      "epoch": 0.30534698521046644,
      "grad_norm": 1.8913527150034817,
      "learning_rate": 1.2385496439431395e-06,
      "loss": 0.0914,
      "step": 1342
    },
    {
      "epoch": 0.3055745164960182,
      "grad_norm": 1.445418394402971,
      "learning_rate": 1.2385326126703986e-06,
      "loss": 0.0687,
      "step": 1343
    },
    {
      "epoch": 0.30580204778156994,
      "grad_norm": 1.8288680303956055,
      "learning_rate": 1.2385155688581746e-06,
      "loss": 0.0739,
      "step": 1344
    },
    {
      "epoch": 0.30602957906712175,
      "grad_norm": 1.0257171242110983,
      "learning_rate": 1.238498512506816e-06,
      "loss": 0.0384,
      "step": 1345
    },
    {
      "epoch": 0.3062571103526735,
      "grad_norm": 1.8747685549411612,
      "learning_rate": 1.2384814436166715e-06,
      "loss": 0.0818,
      "step": 1346
    },
    {
      "epoch": 0.30648464163822525,
      "grad_norm": 1.2409585891810266,
      "learning_rate": 1.2384643621880898e-06,
      "loss": 0.0559,
      "step": 1347
    },
    {
      "epoch": 0.306712172923777,
      "grad_norm": 1.5549364252372342,
      "learning_rate": 1.2384472682214201e-06,
      "loss": 0.0633,
      "step": 1348
    },
    {
      "epoch": 0.30693970420932876,
      "grad_norm": 2.521195686268785,
      "learning_rate": 1.2384301617170116e-06,
      "loss": 0.076,
      "step": 1349
    },
    {
      "epoch": 0.30716723549488056,
      "grad_norm": 1.5977696260072916,
      "learning_rate": 1.2384130426752142e-06,
      "loss": 0.0849,
      "step": 1350
    },
    {
      "epoch": 0.3073947667804323,
      "grad_norm": 1.8196313465168323,
      "learning_rate": 1.2383959110963775e-06,
      "loss": 0.0585,
      "step": 1351
    },
    {
      "epoch": 0.30762229806598407,
      "grad_norm": 1.349543502368748,
      "learning_rate": 1.2383787669808518e-06,
      "loss": 0.0627,
      "step": 1352
    },
    {
      "epoch": 0.3078498293515358,
      "grad_norm": 1.6887455424782774,
      "learning_rate": 1.2383616103289871e-06,
      "loss": 0.0497,
      "step": 1353
    },
    {
      "epoch": 0.3080773606370876,
      "grad_norm": 0.6215713527290292,
      "learning_rate": 1.238344441141135e-06,
      "loss": 0.0178,
      "step": 1354
    },
    {
      "epoch": 0.3083048919226394,
      "grad_norm": 1.640454410435492,
      "learning_rate": 1.2383272594176454e-06,
      "loss": 0.0638,
      "step": 1355
    },
    {
      "epoch": 0.30853242320819113,
      "grad_norm": 2.278012967969371,
      "learning_rate": 1.23831006515887e-06,
      "loss": 0.0815,
      "step": 1356
    },
    {
      "epoch": 0.3087599544937429,
      "grad_norm": 1.272401712072303,
      "learning_rate": 1.2382928583651601e-06,
      "loss": 0.0654,
      "step": 1357
    },
    {
      "epoch": 0.30898748577929463,
      "grad_norm": 2.586692416689192,
      "learning_rate": 1.2382756390368674e-06,
      "loss": 0.0821,
      "step": 1358
    },
    {
      "epoch": 0.30921501706484644,
      "grad_norm": 1.0904063045445367,
      "learning_rate": 1.2382584071743438e-06,
      "loss": 0.0461,
      "step": 1359
    },
    {
      "epoch": 0.3094425483503982,
      "grad_norm": 1.8779937542492704,
      "learning_rate": 1.2382411627779414e-06,
      "loss": 0.0883,
      "step": 1360
    },
    {
      "epoch": 0.30967007963594995,
      "grad_norm": 1.9814293020102243,
      "learning_rate": 1.2382239058480128e-06,
      "loss": 0.0778,
      "step": 1361
    },
    {
      "epoch": 0.3098976109215017,
      "grad_norm": 1.4818784709766222,
      "learning_rate": 1.2382066363849106e-06,
      "loss": 0.0476,
      "step": 1362
    },
    {
      "epoch": 0.31012514220705345,
      "grad_norm": 1.5363329257464977,
      "learning_rate": 1.2381893543889878e-06,
      "loss": 0.0539,
      "step": 1363
    },
    {
      "epoch": 0.31035267349260526,
      "grad_norm": 1.58062550907191,
      "learning_rate": 1.2381720598605976e-06,
      "loss": 0.0761,
      "step": 1364
    },
    {
      "epoch": 0.310580204778157,
      "grad_norm": 1.358541490432856,
      "learning_rate": 1.2381547528000934e-06,
      "loss": 0.0923,
      "step": 1365
    },
    {
      "epoch": 0.31080773606370876,
      "grad_norm": 1.0458300282796855,
      "learning_rate": 1.238137433207829e-06,
      "loss": 0.0377,
      "step": 1366
    },
    {
      "epoch": 0.3110352673492605,
      "grad_norm": 1.183742190968498,
      "learning_rate": 1.2381201010841585e-06,
      "loss": 0.0402,
      "step": 1367
    },
    {
      "epoch": 0.31126279863481227,
      "grad_norm": 1.2571454436583878,
      "learning_rate": 1.2381027564294359e-06,
      "loss": 0.062,
      "step": 1368
    },
    {
      "epoch": 0.3114903299203641,
      "grad_norm": 1.503878059552754,
      "learning_rate": 1.238085399244016e-06,
      "loss": 0.0751,
      "step": 1369
    },
    {
      "epoch": 0.3117178612059158,
      "grad_norm": 2.1756129961580317,
      "learning_rate": 1.2380680295282532e-06,
      "loss": 0.099,
      "step": 1370
    },
    {
      "epoch": 0.3119453924914676,
      "grad_norm": 1.478282276295607,
      "learning_rate": 1.2380506472825025e-06,
      "loss": 0.0451,
      "step": 1371
    },
    {
      "epoch": 0.31217292377701933,
      "grad_norm": 1.210809744948102,
      "learning_rate": 1.2380332525071194e-06,
      "loss": 0.0371,
      "step": 1372
    },
    {
      "epoch": 0.3124004550625711,
      "grad_norm": 1.4368943652552293,
      "learning_rate": 1.2380158452024595e-06,
      "loss": 0.0407,
      "step": 1373
    },
    {
      "epoch": 0.3126279863481229,
      "grad_norm": 0.7640327894504174,
      "learning_rate": 1.2379984253688783e-06,
      "loss": 0.0337,
      "step": 1374
    },
    {
      "epoch": 0.31285551763367464,
      "grad_norm": 1.8295475899815037,
      "learning_rate": 1.237980993006732e-06,
      "loss": 0.1147,
      "step": 1375
    },
    {
      "epoch": 0.3130830489192264,
      "grad_norm": 1.4807139063710786,
      "learning_rate": 1.2379635481163768e-06,
      "loss": 0.0743,
      "step": 1376
    },
    {
      "epoch": 0.31331058020477814,
      "grad_norm": 2.448286916518216,
      "learning_rate": 1.2379460906981692e-06,
      "loss": 0.0534,
      "step": 1377
    },
    {
      "epoch": 0.3135381114903299,
      "grad_norm": 0.9023973811346913,
      "learning_rate": 1.237928620752466e-06,
      "loss": 0.052,
      "step": 1378
    },
    {
      "epoch": 0.3137656427758817,
      "grad_norm": 2.013947590986497,
      "learning_rate": 1.2379111382796246e-06,
      "loss": 0.0575,
      "step": 1379
    },
    {
      "epoch": 0.31399317406143346,
      "grad_norm": 1.6032640725703056,
      "learning_rate": 1.2378936432800017e-06,
      "loss": 0.0706,
      "step": 1380
    },
    {
      "epoch": 0.3142207053469852,
      "grad_norm": 1.239992146480516,
      "learning_rate": 1.2378761357539554e-06,
      "loss": 0.0569,
      "step": 1381
    },
    {
      "epoch": 0.31444823663253696,
      "grad_norm": 1.0642429061701875,
      "learning_rate": 1.2378586157018434e-06,
      "loss": 0.0381,
      "step": 1382
    },
    {
      "epoch": 0.3146757679180887,
      "grad_norm": 1.1087768689334148,
      "learning_rate": 1.2378410831240235e-06,
      "loss": 0.0558,
      "step": 1383
    },
    {
      "epoch": 0.3149032992036405,
      "grad_norm": 1.9572648591918616,
      "learning_rate": 1.2378235380208542e-06,
      "loss": 0.0738,
      "step": 1384
    },
    {
      "epoch": 0.31513083048919227,
      "grad_norm": 0.9929386411259282,
      "learning_rate": 1.2378059803926941e-06,
      "loss": 0.0394,
      "step": 1385
    },
    {
      "epoch": 0.315358361774744,
      "grad_norm": 1.202982993512815,
      "learning_rate": 1.2377884102399023e-06,
      "loss": 0.0461,
      "step": 1386
    },
    {
      "epoch": 0.3155858930602958,
      "grad_norm": 1.1752667673190271,
      "learning_rate": 1.2377708275628375e-06,
      "loss": 0.0447,
      "step": 1387
    },
    {
      "epoch": 0.3158134243458475,
      "grad_norm": 1.5891864361864994,
      "learning_rate": 1.2377532323618593e-06,
      "loss": 0.0456,
      "step": 1388
    },
    {
      "epoch": 0.31604095563139933,
      "grad_norm": 1.6341111744531334,
      "learning_rate": 1.2377356246373271e-06,
      "loss": 0.069,
      "step": 1389
    },
    {
      "epoch": 0.3162684869169511,
      "grad_norm": 1.440731938439493,
      "learning_rate": 1.2377180043896012e-06,
      "loss": 0.0595,
      "step": 1390
    },
    {
      "epoch": 0.31649601820250284,
      "grad_norm": 0.9756352806439903,
      "learning_rate": 1.2377003716190411e-06,
      "loss": 0.0469,
      "step": 1391
    },
    {
      "epoch": 0.3167235494880546,
      "grad_norm": 1.4676165718616923,
      "learning_rate": 1.2376827263260078e-06,
      "loss": 0.0516,
      "step": 1392
    },
    {
      "epoch": 0.3169510807736064,
      "grad_norm": 1.5939577885228815,
      "learning_rate": 1.2376650685108612e-06,
      "loss": 0.0503,
      "step": 1393
    },
    {
      "epoch": 0.31717861205915815,
      "grad_norm": 2.1816410059989098,
      "learning_rate": 1.2376473981739632e-06,
      "loss": 0.0929,
      "step": 1394
    },
    {
      "epoch": 0.3174061433447099,
      "grad_norm": 2.718683093963833,
      "learning_rate": 1.237629715315674e-06,
      "loss": 0.1129,
      "step": 1395
    },
    {
      "epoch": 0.31763367463026165,
      "grad_norm": 0.9979838242263193,
      "learning_rate": 1.2376120199363554e-06,
      "loss": 0.0536,
      "step": 1396
    },
    {
      "epoch": 0.3178612059158134,
      "grad_norm": 1.9720298867063506,
      "learning_rate": 1.2375943120363692e-06,
      "loss": 0.0969,
      "step": 1397
    },
    {
      "epoch": 0.3180887372013652,
      "grad_norm": 0.6502262811463818,
      "learning_rate": 1.2375765916160773e-06,
      "loss": 0.0287,
      "step": 1398
    },
    {
      "epoch": 0.31831626848691696,
      "grad_norm": 1.3837265125988647,
      "learning_rate": 1.2375588586758415e-06,
      "loss": 0.0441,
      "step": 1399
    },
    {
      "epoch": 0.3185437997724687,
      "grad_norm": 1.593794924686227,
      "learning_rate": 1.2375411132160245e-06,
      "loss": 0.0595,
      "step": 1400
    },
    {
      "epoch": 0.31877133105802047,
      "grad_norm": 1.776611213440541,
      "learning_rate": 1.2375233552369892e-06,
      "loss": 0.0694,
      "step": 1401
    },
    {
      "epoch": 0.3189988623435722,
      "grad_norm": 1.2580919074923334,
      "learning_rate": 1.237505584739098e-06,
      "loss": 0.0491,
      "step": 1402
    },
    {
      "epoch": 0.31922639362912403,
      "grad_norm": 0.994481303076797,
      "learning_rate": 1.2374878017227147e-06,
      "loss": 0.0291,
      "step": 1403
    },
    {
      "epoch": 0.3194539249146758,
      "grad_norm": 0.7265755658718445,
      "learning_rate": 1.237470006188202e-06,
      "loss": 0.027,
      "step": 1404
    },
    {
      "epoch": 0.31968145620022753,
      "grad_norm": 0.9576296817818531,
      "learning_rate": 1.2374521981359245e-06,
      "loss": 0.0423,
      "step": 1405
    },
    {
      "epoch": 0.3199089874857793,
      "grad_norm": 1.7776464530772098,
      "learning_rate": 1.2374343775662456e-06,
      "loss": 0.0764,
      "step": 1406
    },
    {
      "epoch": 0.32013651877133104,
      "grad_norm": 1.2848189857025718,
      "learning_rate": 1.2374165444795296e-06,
      "loss": 0.0398,
      "step": 1407
    },
    {
      "epoch": 0.32036405005688284,
      "grad_norm": 2.0114145603579625,
      "learning_rate": 1.237398698876141e-06,
      "loss": 0.0934,
      "step": 1408
    },
    {
      "epoch": 0.3205915813424346,
      "grad_norm": 1.3984905897111508,
      "learning_rate": 1.2373808407564446e-06,
      "loss": 0.0513,
      "step": 1409
    },
    {
      "epoch": 0.32081911262798635,
      "grad_norm": 0.9464490409610342,
      "learning_rate": 1.2373629701208053e-06,
      "loss": 0.0369,
      "step": 1410
    },
    {
      "epoch": 0.3210466439135381,
      "grad_norm": 1.4072284744296988,
      "learning_rate": 1.2373450869695883e-06,
      "loss": 0.0479,
      "step": 1411
    },
    {
      "epoch": 0.32127417519908985,
      "grad_norm": 1.5797071346301892,
      "learning_rate": 1.2373271913031593e-06,
      "loss": 0.0581,
      "step": 1412
    },
    {
      "epoch": 0.32150170648464166,
      "grad_norm": 1.5741999856445812,
      "learning_rate": 1.237309283121884e-06,
      "loss": 0.0695,
      "step": 1413
    },
    {
      "epoch": 0.3217292377701934,
      "grad_norm": 1.004256336444922,
      "learning_rate": 1.237291362426128e-06,
      "loss": 0.0584,
      "step": 1414
    },
    {
      "epoch": 0.32195676905574516,
      "grad_norm": 1.2610248823720653,
      "learning_rate": 1.2372734292162584e-06,
      "loss": 0.0508,
      "step": 1415
    },
    {
      "epoch": 0.3221843003412969,
      "grad_norm": 1.138966176373832,
      "learning_rate": 1.237255483492641e-06,
      "loss": 0.0346,
      "step": 1416
    },
    {
      "epoch": 0.32241183162684867,
      "grad_norm": 1.279726266655994,
      "learning_rate": 1.2372375252556429e-06,
      "loss": 0.0526,
      "step": 1417
    },
    {
      "epoch": 0.3226393629124005,
      "grad_norm": 2.259244357087649,
      "learning_rate": 1.2372195545056308e-06,
      "loss": 0.0737,
      "step": 1418
    },
    {
      "epoch": 0.3228668941979522,
      "grad_norm": 1.260233063486729,
      "learning_rate": 1.2372015712429725e-06,
      "loss": 0.0439,
      "step": 1419
    },
    {
      "epoch": 0.323094425483504,
      "grad_norm": 1.180315594146752,
      "learning_rate": 1.237183575468035e-06,
      "loss": 0.0506,
      "step": 1420
    },
    {
      "epoch": 0.32332195676905573,
      "grad_norm": 0.9777167019905004,
      "learning_rate": 1.2371655671811866e-06,
      "loss": 0.0308,
      "step": 1421
    },
    {
      "epoch": 0.3235494880546075,
      "grad_norm": 0.8734921365662196,
      "learning_rate": 1.237147546382795e-06,
      "loss": 0.0281,
      "step": 1422
    },
    {
      "epoch": 0.3237770193401593,
      "grad_norm": 1.1112810388031804,
      "learning_rate": 1.237129513073229e-06,
      "loss": 0.04,
      "step": 1423
    },
    {
      "epoch": 0.32400455062571104,
      "grad_norm": 1.156890481720808,
      "learning_rate": 1.2371114672528565e-06,
      "loss": 0.0593,
      "step": 1424
    },
    {
      "epoch": 0.3242320819112628,
      "grad_norm": 0.8688160126362955,
      "learning_rate": 1.2370934089220466e-06,
      "loss": 0.0239,
      "step": 1425
    },
    {
      "epoch": 0.32445961319681454,
      "grad_norm": 1.1430764409107144,
      "learning_rate": 1.2370753380811685e-06,
      "loss": 0.0343,
      "step": 1426
    },
    {
      "epoch": 0.32468714448236635,
      "grad_norm": 0.94277640364479,
      "learning_rate": 1.2370572547305915e-06,
      "loss": 0.0254,
      "step": 1427
    },
    {
      "epoch": 0.3249146757679181,
      "grad_norm": 0.8543512868236353,
      "learning_rate": 1.237039158870685e-06,
      "loss": 0.0359,
      "step": 1428
    },
    {
      "epoch": 0.32514220705346986,
      "grad_norm": 0.9268593896369735,
      "learning_rate": 1.237021050501819e-06,
      "loss": 0.0462,
      "step": 1429
    },
    {
      "epoch": 0.3253697383390216,
      "grad_norm": 1.0492131527271786,
      "learning_rate": 1.2370029296243638e-06,
      "loss": 0.0483,
      "step": 1430
    },
    {
      "epoch": 0.32559726962457336,
      "grad_norm": 1.2265145141334126,
      "learning_rate": 1.2369847962386893e-06,
      "loss": 0.0605,
      "step": 1431
    },
    {
      "epoch": 0.32582480091012517,
      "grad_norm": 1.4614421557097579,
      "learning_rate": 1.2369666503451665e-06,
      "loss": 0.0792,
      "step": 1432
    },
    {
      "epoch": 0.3260523321956769,
      "grad_norm": 1.3551154178225278,
      "learning_rate": 1.2369484919441662e-06,
      "loss": 0.0504,
      "step": 1433
    },
    {
      "epoch": 0.32627986348122867,
      "grad_norm": 2.070524215459429,
      "learning_rate": 1.2369303210360592e-06,
      "loss": 0.0813,
      "step": 1434
    },
    {
      "epoch": 0.3265073947667804,
      "grad_norm": 2.674887998532007,
      "learning_rate": 1.2369121376212174e-06,
      "loss": 0.1356,
      "step": 1435
    },
    {
      "epoch": 0.3267349260523322,
      "grad_norm": 2.0059123123271543,
      "learning_rate": 1.236893941700012e-06,
      "loss": 0.0815,
      "step": 1436
    },
    {
      "epoch": 0.326962457337884,
      "grad_norm": 1.614422308046532,
      "learning_rate": 1.236875733272815e-06,
      "loss": 0.0632,
      "step": 1437
    },
    {
      "epoch": 0.32718998862343573,
      "grad_norm": 1.0273415965735717,
      "learning_rate": 1.236857512339999e-06,
      "loss": 0.0636,
      "step": 1438
    },
    {
      "epoch": 0.3274175199089875,
      "grad_norm": 2.063298633572998,
      "learning_rate": 1.2368392789019356e-06,
      "loss": 0.1028,
      "step": 1439
    },
    {
      "epoch": 0.32764505119453924,
      "grad_norm": 1.0762629811135433,
      "learning_rate": 1.2368210329589982e-06,
      "loss": 0.0287,
      "step": 1440
    },
    {
      "epoch": 0.327872582480091,
      "grad_norm": 1.5998871926979066,
      "learning_rate": 1.236802774511559e-06,
      "loss": 0.0862,
      "step": 1441
    },
    {
      "epoch": 0.3281001137656428,
      "grad_norm": 1.4906993606341739,
      "learning_rate": 1.2367845035599919e-06,
      "loss": 0.094,
      "step": 1442
    },
    {
      "epoch": 0.32832764505119455,
      "grad_norm": 0.8173626977187779,
      "learning_rate": 1.2367662201046698e-06,
      "loss": 0.0348,
      "step": 1443
    },
    {
      "epoch": 0.3285551763367463,
      "grad_norm": 1.260244704516005,
      "learning_rate": 1.2367479241459666e-06,
      "loss": 0.0388,
      "step": 1444
    },
    {
      "epoch": 0.32878270762229805,
      "grad_norm": 1.2263377322773583,
      "learning_rate": 1.2367296156842562e-06,
      "loss": 0.0514,
      "step": 1445
    },
    {
      "epoch": 0.3290102389078498,
      "grad_norm": 1.1202578543878008,
      "learning_rate": 1.2367112947199128e-06,
      "loss": 0.0332,
      "step": 1446
    },
    {
      "epoch": 0.3292377701934016,
      "grad_norm": 1.3736971576588957,
      "learning_rate": 1.2366929612533109e-06,
      "loss": 0.0539,
      "step": 1447
    },
    {
      "epoch": 0.32946530147895337,
      "grad_norm": 1.2893217491287035,
      "learning_rate": 1.2366746152848249e-06,
      "loss": 0.058,
      "step": 1448
    },
    {
      "epoch": 0.3296928327645051,
      "grad_norm": 1.4381578160793065,
      "learning_rate": 1.2366562568148301e-06,
      "loss": 0.0521,
      "step": 1449
    },
    {
      "epoch": 0.32992036405005687,
      "grad_norm": 1.6844461549267769,
      "learning_rate": 1.2366378858437016e-06,
      "loss": 0.1036,
      "step": 1450
    },
    {
      "epoch": 0.3301478953356086,
      "grad_norm": 1.397177590337732,
      "learning_rate": 1.2366195023718152e-06,
      "loss": 0.0534,
      "step": 1451
    },
    {
      "epoch": 0.33037542662116043,
      "grad_norm": 1.0456464972065773,
      "learning_rate": 1.2366011063995458e-06,
      "loss": 0.0558,
      "step": 1452
    },
    {
      "epoch": 0.3306029579067122,
      "grad_norm": 1.9835006920707845,
      "learning_rate": 1.2365826979272702e-06,
      "loss": 0.1249,
      "step": 1453
    },
    {
      "epoch": 0.33083048919226393,
      "grad_norm": 2.6314184120557376,
      "learning_rate": 1.2365642769553644e-06,
      "loss": 0.1484,
      "step": 1454
    },
    {
      "epoch": 0.3310580204778157,
      "grad_norm": 1.4200073439085166,
      "learning_rate": 1.2365458434842046e-06,
      "loss": 0.0483,
      "step": 1455
    },
    {
      "epoch": 0.33128555176336744,
      "grad_norm": 1.1143001688120713,
      "learning_rate": 1.2365273975141675e-06,
      "loss": 0.0369,
      "step": 1456
    },
    {
      "epoch": 0.33151308304891924,
      "grad_norm": 1.0377938140431064,
      "learning_rate": 1.2365089390456308e-06,
      "loss": 0.0518,
      "step": 1457
    },
    {
      "epoch": 0.331740614334471,
      "grad_norm": 1.1599848352897109,
      "learning_rate": 1.236490468078971e-06,
      "loss": 0.0432,
      "step": 1458
    },
    {
      "epoch": 0.33196814562002275,
      "grad_norm": 1.5939156162998587,
      "learning_rate": 1.2364719846145662e-06,
      "loss": 0.0479,
      "step": 1459
    },
    {
      "epoch": 0.3321956769055745,
      "grad_norm": 1.0154765270633663,
      "learning_rate": 1.2364534886527937e-06,
      "loss": 0.0431,
      "step": 1460
    },
    {
      "epoch": 0.3324232081911263,
      "grad_norm": 1.8584242211532112,
      "learning_rate": 1.2364349801940317e-06,
      "loss": 0.1002,
      "step": 1461
    },
    {
      "epoch": 0.33265073947667806,
      "grad_norm": 1.3694201211070989,
      "learning_rate": 1.2364164592386588e-06,
      "loss": 0.0733,
      "step": 1462
    },
    {
      "epoch": 0.3328782707622298,
      "grad_norm": 1.9119781832863982,
      "learning_rate": 1.2363979257870528e-06,
      "loss": 0.0696,
      "step": 1463
    },
    {
      "epoch": 0.33310580204778156,
      "grad_norm": 1.9001104021651412,
      "learning_rate": 1.2363793798395932e-06,
      "loss": 0.0572,
      "step": 1464
    },
    {
      "epoch": 0.3333333333333333,
      "grad_norm": 2.498897834454089,
      "learning_rate": 1.2363608213966588e-06,
      "loss": 0.0848,
      "step": 1465
    },
    {
      "epoch": 0.3335608646188851,
      "grad_norm": 1.1620781742745308,
      "learning_rate": 1.2363422504586286e-06,
      "loss": 0.0504,
      "step": 1466
    },
    {
      "epoch": 0.3337883959044369,
      "grad_norm": 1.2505263772670705,
      "learning_rate": 1.2363236670258827e-06,
      "loss": 0.05,
      "step": 1467
    },
    {
      "epoch": 0.3340159271899886,
      "grad_norm": 1.4633039559699816,
      "learning_rate": 1.2363050710988003e-06,
      "loss": 0.079,
      "step": 1468
    },
    {
      "epoch": 0.3342434584755404,
      "grad_norm": 1.2463657051053827,
      "learning_rate": 1.236286462677762e-06,
      "loss": 0.0418,
      "step": 1469
    },
    {
      "epoch": 0.33447098976109213,
      "grad_norm": 1.683583709704665,
      "learning_rate": 1.236267841763148e-06,
      "loss": 0.0812,
      "step": 1470
    },
    {
      "epoch": 0.33469852104664394,
      "grad_norm": 0.7484223072827635,
      "learning_rate": 1.2362492083553387e-06,
      "loss": 0.0356,
      "step": 1471
    },
    {
      "epoch": 0.3349260523321957,
      "grad_norm": 1.639323852335928,
      "learning_rate": 1.236230562454715e-06,
      "loss": 0.0401,
      "step": 1472
    },
    {
      "epoch": 0.33515358361774744,
      "grad_norm": 1.2064100614992055,
      "learning_rate": 1.2362119040616582e-06,
      "loss": 0.0677,
      "step": 1473
    },
    {
      "epoch": 0.3353811149032992,
      "grad_norm": 1.107381501336236,
      "learning_rate": 1.2361932331765492e-06,
      "loss": 0.0404,
      "step": 1474
    },
    {
      "epoch": 0.33560864618885095,
      "grad_norm": 1.4911394528443533,
      "learning_rate": 1.2361745497997702e-06,
      "loss": 0.0623,
      "step": 1475
    },
    {
      "epoch": 0.33583617747440275,
      "grad_norm": 1.2262949506307204,
      "learning_rate": 1.2361558539317023e-06,
      "loss": 0.0651,
      "step": 1476
    },
    {
      "epoch": 0.3360637087599545,
      "grad_norm": 0.9519435590816607,
      "learning_rate": 1.2361371455727284e-06,
      "loss": 0.0404,
      "step": 1477
    },
    {
      "epoch": 0.33629124004550626,
      "grad_norm": 1.5689690961881004,
      "learning_rate": 1.2361184247232302e-06,
      "loss": 0.087,
      "step": 1478
    },
    {
      "epoch": 0.336518771331058,
      "grad_norm": 5.906098295875717,
      "learning_rate": 1.2360996913835907e-06,
      "loss": 0.236,
      "step": 1479
    },
    {
      "epoch": 0.33674630261660976,
      "grad_norm": 0.8654279575626805,
      "learning_rate": 1.2360809455541928e-06,
      "loss": 0.0549,
      "step": 1480
    },
    {
      "epoch": 0.33697383390216157,
      "grad_norm": 1.776832317476583,
      "learning_rate": 1.2360621872354195e-06,
      "loss": 0.0453,
      "step": 1481
    },
    {
      "epoch": 0.3372013651877133,
      "grad_norm": 1.347252653711477,
      "learning_rate": 1.236043416427654e-06,
      "loss": 0.0697,
      "step": 1482
    },
    {
      "epoch": 0.33742889647326507,
      "grad_norm": 1.2864149376371685,
      "learning_rate": 1.2360246331312804e-06,
      "loss": 0.0381,
      "step": 1483
    },
    {
      "epoch": 0.3376564277588168,
      "grad_norm": 1.6142130648155326,
      "learning_rate": 1.2360058373466821e-06,
      "loss": 0.0477,
      "step": 1484
    },
    {
      "epoch": 0.3378839590443686,
      "grad_norm": 2.106670043543402,
      "learning_rate": 1.2359870290742437e-06,
      "loss": 0.097,
      "step": 1485
    },
    {
      "epoch": 0.3381114903299204,
      "grad_norm": 1.2681226361832132,
      "learning_rate": 1.2359682083143494e-06,
      "loss": 0.0682,
      "step": 1486
    },
    {
      "epoch": 0.33833902161547214,
      "grad_norm": 0.8649596148543436,
      "learning_rate": 1.2359493750673835e-06,
      "loss": 0.0302,
      "step": 1487
    },
    {
      "epoch": 0.3385665529010239,
      "grad_norm": 1.3533089587285108,
      "learning_rate": 1.2359305293337316e-06,
      "loss": 0.0526,
      "step": 1488
    },
    {
      "epoch": 0.33879408418657564,
      "grad_norm": 0.7446203717116848,
      "learning_rate": 1.2359116711137785e-06,
      "loss": 0.0247,
      "step": 1489
    },
    {
      "epoch": 0.3390216154721274,
      "grad_norm": 1.1703255026021544,
      "learning_rate": 1.2358928004079095e-06,
      "loss": 0.0469,
      "step": 1490
    },
    {
      "epoch": 0.3392491467576792,
      "grad_norm": 2.2956044408157554,
      "learning_rate": 1.2358739172165108e-06,
      "loss": 0.1262,
      "step": 1491
    },
    {
      "epoch": 0.33947667804323095,
      "grad_norm": 1.425210539470094,
      "learning_rate": 1.235855021539968e-06,
      "loss": 0.0539,
      "step": 1492
    },
    {
      "epoch": 0.3397042093287827,
      "grad_norm": 1.2958224989383627,
      "learning_rate": 1.2358361133786668e-06,
      "loss": 0.0417,
      "step": 1493
    },
    {
      "epoch": 0.33993174061433445,
      "grad_norm": 0.8383879482270241,
      "learning_rate": 1.2358171927329946e-06,
      "loss": 0.0411,
      "step": 1494
    },
    {
      "epoch": 0.34015927189988626,
      "grad_norm": 1.0816879292496617,
      "learning_rate": 1.2357982596033374e-06,
      "loss": 0.0321,
      "step": 1495
    },
    {
      "epoch": 0.340386803185438,
      "grad_norm": 1.2228392090454494,
      "learning_rate": 1.2357793139900823e-06,
      "loss": 0.076,
      "step": 1496
    },
    {
      "epoch": 0.34061433447098977,
      "grad_norm": 0.9376305546071186,
      "learning_rate": 1.2357603558936168e-06,
      "loss": 0.0294,
      "step": 1497
    },
    {
      "epoch": 0.3408418657565415,
      "grad_norm": 2.032138065689827,
      "learning_rate": 1.2357413853143282e-06,
      "loss": 0.1073,
      "step": 1498
    },
    {
      "epoch": 0.34106939704209327,
      "grad_norm": 1.203066065214765,
      "learning_rate": 1.2357224022526041e-06,
      "loss": 0.0585,
      "step": 1499
    },
    {
      "epoch": 0.3412969283276451,
      "grad_norm": 1.5240232506976108,
      "learning_rate": 1.2357034067088327e-06,
      "loss": 0.0566,
      "step": 1500
    },
    {
      "epoch": 0.34152445961319683,
      "grad_norm": 1.4069261769751387,
      "learning_rate": 1.235684398683402e-06,
      "loss": 0.0911,
      "step": 1501
    },
    {
      "epoch": 0.3417519908987486,
      "grad_norm": 1.2794137407821935,
      "learning_rate": 1.2356653781767009e-06,
      "loss": 0.0387,
      "step": 1502
    },
    {
      "epoch": 0.34197952218430033,
      "grad_norm": 1.7747949717550477,
      "learning_rate": 1.2356463451891174e-06,
      "loss": 0.0928,
      "step": 1503
    },
    {
      "epoch": 0.3422070534698521,
      "grad_norm": 1.0081856733209773,
      "learning_rate": 1.2356272997210414e-06,
      "loss": 0.0289,
      "step": 1504
    },
    {
      "epoch": 0.3424345847554039,
      "grad_norm": 1.8655840562261654,
      "learning_rate": 1.2356082417728612e-06,
      "loss": 0.0837,
      "step": 1505
    },
    {
      "epoch": 0.34266211604095564,
      "grad_norm": 0.9436733972164556,
      "learning_rate": 1.2355891713449672e-06,
      "loss": 0.0406,
      "step": 1506
    },
    {
      "epoch": 0.3428896473265074,
      "grad_norm": 1.2922255822019764,
      "learning_rate": 1.2355700884377485e-06,
      "loss": 0.0363,
      "step": 1507
    },
    {
      "epoch": 0.34311717861205915,
      "grad_norm": 1.3543650706739145,
      "learning_rate": 1.2355509930515958e-06,
      "loss": 0.0697,
      "step": 1508
    },
    {
      "epoch": 0.3433447098976109,
      "grad_norm": 1.3352155504337506,
      "learning_rate": 1.2355318851868987e-06,
      "loss": 0.0723,
      "step": 1509
    },
    {
      "epoch": 0.3435722411831627,
      "grad_norm": 0.8330395399945464,
      "learning_rate": 1.235512764844048e-06,
      "loss": 0.0427,
      "step": 1510
    },
    {
      "epoch": 0.34379977246871446,
      "grad_norm": 0.7599055129001592,
      "learning_rate": 1.2354936320234345e-06,
      "loss": 0.0275,
      "step": 1511
    },
    {
      "epoch": 0.3440273037542662,
      "grad_norm": 1.107814773629098,
      "learning_rate": 1.2354744867254493e-06,
      "loss": 0.0519,
      "step": 1512
    },
    {
      "epoch": 0.34425483503981796,
      "grad_norm": 0.9874953818051542,
      "learning_rate": 1.2354553289504836e-06,
      "loss": 0.0276,
      "step": 1513
    },
    {
      "epoch": 0.3444823663253697,
      "grad_norm": 1.4195908498124084,
      "learning_rate": 1.2354361586989287e-06,
      "loss": 0.0658,
      "step": 1514
    },
    {
      "epoch": 0.3447098976109215,
      "grad_norm": 1.1536105740853742,
      "learning_rate": 1.235416975971177e-06,
      "loss": 0.0693,
      "step": 1515
    },
    {
      "epoch": 0.3449374288964733,
      "grad_norm": 1.1208196458039765,
      "learning_rate": 1.2353977807676205e-06,
      "loss": 0.0451,
      "step": 1516
    },
    {
      "epoch": 0.345164960182025,
      "grad_norm": 1.6664132616304859,
      "learning_rate": 1.2353785730886506e-06,
      "loss": 0.0659,
      "step": 1517
    },
    {
      "epoch": 0.3453924914675768,
      "grad_norm": 1.6331880132019159,
      "learning_rate": 1.235359352934661e-06,
      "loss": 0.0518,
      "step": 1518
    },
    {
      "epoch": 0.34562002275312853,
      "grad_norm": 1.9450367879491788,
      "learning_rate": 1.235340120306044e-06,
      "loss": 0.108,
      "step": 1519
    },
    {
      "epoch": 0.34584755403868034,
      "grad_norm": 1.4426224264383893,
      "learning_rate": 1.2353208752031925e-06,
      "loss": 0.0598,
      "step": 1520
    },
    {
      "epoch": 0.3460750853242321,
      "grad_norm": 1.2111901419847948,
      "learning_rate": 1.2353016176265002e-06,
      "loss": 0.0304,
      "step": 1521
    },
    {
      "epoch": 0.34630261660978384,
      "grad_norm": 0.841669446586875,
      "learning_rate": 1.2352823475763603e-06,
      "loss": 0.029,
      "step": 1522
    },
    {
      "epoch": 0.3465301478953356,
      "grad_norm": 1.6324873257088213,
      "learning_rate": 1.2352630650531672e-06,
      "loss": 0.0737,
      "step": 1523
    },
    {
      "epoch": 0.34675767918088735,
      "grad_norm": 1.5631645650045773,
      "learning_rate": 1.2352437700573147e-06,
      "loss": 0.0555,
      "step": 1524
    },
    {
      "epoch": 0.34698521046643915,
      "grad_norm": 1.3689087939433322,
      "learning_rate": 1.235224462589197e-06,
      "loss": 0.0699,
      "step": 1525
    },
    {
      "epoch": 0.3472127417519909,
      "grad_norm": 1.729486346174357,
      "learning_rate": 1.2352051426492089e-06,
      "loss": 0.0795,
      "step": 1526
    },
    {
      "epoch": 0.34744027303754266,
      "grad_norm": 1.2401649575621747,
      "learning_rate": 1.2351858102377455e-06,
      "loss": 0.0459,
      "step": 1527
    },
    {
      "epoch": 0.3476678043230944,
      "grad_norm": 1.1736397484657297,
      "learning_rate": 1.2351664653552012e-06,
      "loss": 0.0374,
      "step": 1528
    },
    {
      "epoch": 0.34789533560864616,
      "grad_norm": 0.8775289996354035,
      "learning_rate": 1.235147108001972e-06,
      "loss": 0.0415,
      "step": 1529
    },
    {
      "epoch": 0.34812286689419797,
      "grad_norm": 1.5150051901632806,
      "learning_rate": 1.2351277381784532e-06,
      "loss": 0.0499,
      "step": 1530
    },
    {
      "epoch": 0.3483503981797497,
      "grad_norm": 0.7367153174028898,
      "learning_rate": 1.235108355885041e-06,
      "loss": 0.0405,
      "step": 1531
    },
    {
      "epoch": 0.3485779294653015,
      "grad_norm": 1.0973156282921959,
      "learning_rate": 1.2350889611221315e-06,
      "loss": 0.0593,
      "step": 1532
    },
    {
      "epoch": 0.3488054607508532,
      "grad_norm": 1.2029539420964899,
      "learning_rate": 1.2350695538901207e-06,
      "loss": 0.052,
      "step": 1533
    },
    {
      "epoch": 0.34903299203640503,
      "grad_norm": 1.2254789545948275,
      "learning_rate": 1.2350501341894055e-06,
      "loss": 0.0442,
      "step": 1534
    },
    {
      "epoch": 0.3492605233219568,
      "grad_norm": 1.173455391781939,
      "learning_rate": 1.235030702020383e-06,
      "loss": 0.0627,
      "step": 1535
    },
    {
      "epoch": 0.34948805460750854,
      "grad_norm": 1.1576335152673878,
      "learning_rate": 1.23501125738345e-06,
      "loss": 0.0467,
      "step": 1536
    },
    {
      "epoch": 0.3497155858930603,
      "grad_norm": 1.0478326366866813,
      "learning_rate": 1.2349918002790043e-06,
      "loss": 0.0409,
      "step": 1537
    },
    {
      "epoch": 0.34994311717861204,
      "grad_norm": 0.7625837183082101,
      "learning_rate": 1.2349723307074432e-06,
      "loss": 0.037,
      "step": 1538
    },
    {
      "epoch": 0.35017064846416385,
      "grad_norm": 1.2469855224758288,
      "learning_rate": 1.2349528486691648e-06,
      "loss": 0.0668,
      "step": 1539
    },
    {
      "epoch": 0.3503981797497156,
      "grad_norm": 1.0826914171234396,
      "learning_rate": 1.2349333541645672e-06,
      "loss": 0.0443,
      "step": 1540
    },
    {
      "epoch": 0.35062571103526735,
      "grad_norm": 1.195949344430507,
      "learning_rate": 1.2349138471940489e-06,
      "loss": 0.0362,
      "step": 1541
    },
    {
      "epoch": 0.3508532423208191,
      "grad_norm": 1.2921680529996942,
      "learning_rate": 1.2348943277580086e-06,
      "loss": 0.0534,
      "step": 1542
    },
    {
      "epoch": 0.35108077360637086,
      "grad_norm": 0.713008372789458,
      "learning_rate": 1.2348747958568452e-06,
      "loss": 0.0211,
      "step": 1543
    },
    {
      "epoch": 0.35130830489192266,
      "grad_norm": 1.5116593555386681,
      "learning_rate": 1.2348552514909579e-06,
      "loss": 0.0712,
      "step": 1544
    },
    {
      "epoch": 0.3515358361774744,
      "grad_norm": 1.0248863840699733,
      "learning_rate": 1.2348356946607462e-06,
      "loss": 0.0305,
      "step": 1545
    },
    {
      "epoch": 0.35176336746302617,
      "grad_norm": 1.1409421918168505,
      "learning_rate": 1.2348161253666096e-06,
      "loss": 0.0529,
      "step": 1546
    },
    {
      "epoch": 0.3519908987485779,
      "grad_norm": 1.768834998548112,
      "learning_rate": 1.2347965436089484e-06,
      "loss": 0.0888,
      "step": 1547
    },
    {
      "epoch": 0.35221843003412967,
      "grad_norm": 1.4622739241779608,
      "learning_rate": 1.2347769493881625e-06,
      "loss": 0.0466,
      "step": 1548
    },
    {
      "epoch": 0.3524459613196815,
      "grad_norm": 0.6584121088377994,
      "learning_rate": 1.2347573427046527e-06,
      "loss": 0.0348,
      "step": 1549
    },
    {
      "epoch": 0.35267349260523323,
      "grad_norm": 2.2141006455976604,
      "learning_rate": 1.2347377235588193e-06,
      "loss": 0.1474,
      "step": 1550
    },
    {
      "epoch": 0.352901023890785,
      "grad_norm": 1.19254071976006,
      "learning_rate": 1.2347180919510637e-06,
      "loss": 0.0511,
      "step": 1551
    },
    {
      "epoch": 0.35312855517633673,
      "grad_norm": 1.4469931390196393,
      "learning_rate": 1.234698447881787e-06,
      "loss": 0.0457,
      "step": 1552
    },
    {
      "epoch": 0.3533560864618885,
      "grad_norm": 1.0953678059195961,
      "learning_rate": 1.2346787913513904e-06,
      "loss": 0.0351,
      "step": 1553
    },
    {
      "epoch": 0.3535836177474403,
      "grad_norm": 1.2220799955288675,
      "learning_rate": 1.234659122360276e-06,
      "loss": 0.0605,
      "step": 1554
    },
    {
      "epoch": 0.35381114903299204,
      "grad_norm": 0.7125746349183502,
      "learning_rate": 1.2346394409088457e-06,
      "loss": 0.029,
      "step": 1555
    },
    {
      "epoch": 0.3540386803185438,
      "grad_norm": 0.7700035617297257,
      "learning_rate": 1.2346197469975016e-06,
      "loss": 0.0399,
      "step": 1556
    },
    {
      "epoch": 0.35426621160409555,
      "grad_norm": 1.4287678945741027,
      "learning_rate": 1.2346000406266466e-06,
      "loss": 0.0849,
      "step": 1557
    },
    {
      "epoch": 0.3544937428896473,
      "grad_norm": 1.4371209706481878,
      "learning_rate": 1.2345803217966829e-06,
      "loss": 0.0497,
      "step": 1558
    },
    {
      "epoch": 0.3547212741751991,
      "grad_norm": 1.5348631747891068,
      "learning_rate": 1.2345605905080141e-06,
      "loss": 0.0625,
      "step": 1559
    },
    {
      "epoch": 0.35494880546075086,
      "grad_norm": 1.8970470757470723,
      "learning_rate": 1.234540846761043e-06,
      "loss": 0.0613,
      "step": 1560
    },
    {
      "epoch": 0.3551763367463026,
      "grad_norm": 2.2382508192831416,
      "learning_rate": 1.2345210905561733e-06,
      "loss": 0.0889,
      "step": 1561
    },
    {
      "epoch": 0.35540386803185436,
      "grad_norm": 0.9465351044370347,
      "learning_rate": 1.2345013218938089e-06,
      "loss": 0.0478,
      "step": 1562
    },
    {
      "epoch": 0.3556313993174061,
      "grad_norm": 0.7802531004508686,
      "learning_rate": 1.2344815407743537e-06,
      "loss": 0.0265,
      "step": 1563
    },
    {
      "epoch": 0.3558589306029579,
      "grad_norm": 1.3777470813147898,
      "learning_rate": 1.2344617471982119e-06,
      "loss": 0.0595,
      "step": 1564
    },
    {
      "epoch": 0.3560864618885097,
      "grad_norm": 1.9091835939621151,
      "learning_rate": 1.2344419411657883e-06,
      "loss": 0.0652,
      "step": 1565
    },
    {
      "epoch": 0.3563139931740614,
      "grad_norm": 1.8891000138650136,
      "learning_rate": 1.2344221226774874e-06,
      "loss": 0.0894,
      "step": 1566
    },
    {
      "epoch": 0.3565415244596132,
      "grad_norm": 2.3333751351582968,
      "learning_rate": 1.2344022917337147e-06,
      "loss": 0.1068,
      "step": 1567
    },
    {
      "epoch": 0.356769055745165,
      "grad_norm": 1.5543425365273886,
      "learning_rate": 1.234382448334875e-06,
      "loss": 0.0549,
      "step": 1568
    },
    {
      "epoch": 0.35699658703071674,
      "grad_norm": 0.7049933258516752,
      "learning_rate": 1.2343625924813741e-06,
      "loss": 0.024,
      "step": 1569
    },
    {
      "epoch": 0.3572241183162685,
      "grad_norm": 1.7675084632445353,
      "learning_rate": 1.234342724173618e-06,
      "loss": 0.0485,
      "step": 1570
    },
    {
      "epoch": 0.35745164960182024,
      "grad_norm": 1.3808807084264496,
      "learning_rate": 1.2343228434120124e-06,
      "loss": 0.067,
      "step": 1571
    },
    {
      "epoch": 0.357679180887372,
      "grad_norm": 1.2453962587688918,
      "learning_rate": 1.2343029501969638e-06,
      "loss": 0.0524,
      "step": 1572
    },
    {
      "epoch": 0.3579067121729238,
      "grad_norm": 1.343656065138551,
      "learning_rate": 1.2342830445288788e-06,
      "loss": 0.0699,
      "step": 1573
    },
    {
      "epoch": 0.35813424345847555,
      "grad_norm": 1.0013673586639926,
      "learning_rate": 1.2342631264081643e-06,
      "loss": 0.051,
      "step": 1574
    },
    {
      "epoch": 0.3583617747440273,
      "grad_norm": 1.1273372730897988,
      "learning_rate": 1.234243195835227e-06,
      "loss": 0.0564,
      "step": 1575
    },
    {
      "epoch": 0.35858930602957906,
      "grad_norm": 1.3457068038564708,
      "learning_rate": 1.234223252810475e-06,
      "loss": 0.0445,
      "step": 1576
    },
    {
      "epoch": 0.3588168373151308,
      "grad_norm": 1.1036016217002649,
      "learning_rate": 1.2342032973343152e-06,
      "loss": 0.0494,
      "step": 1577
    },
    {
      "epoch": 0.3590443686006826,
      "grad_norm": 1.1088066035106612,
      "learning_rate": 1.2341833294071558e-06,
      "loss": 0.0483,
      "step": 1578
    },
    {
      "epoch": 0.35927189988623437,
      "grad_norm": 1.81529439176539,
      "learning_rate": 1.2341633490294046e-06,
      "loss": 0.1067,
      "step": 1579
    },
    {
      "epoch": 0.3594994311717861,
      "grad_norm": 1.7525830328724732,
      "learning_rate": 1.2341433562014705e-06,
      "loss": 0.0977,
      "step": 1580
    },
    {
      "epoch": 0.3597269624573379,
      "grad_norm": 1.1410352301000704,
      "learning_rate": 1.2341233509237616e-06,
      "loss": 0.0383,
      "step": 1581
    },
    {
      "epoch": 0.3599544937428896,
      "grad_norm": 1.3291484690769528,
      "learning_rate": 1.234103333196687e-06,
      "loss": 0.0414,
      "step": 1582
    },
    {
      "epoch": 0.36018202502844143,
      "grad_norm": 1.8297830185646766,
      "learning_rate": 1.2340833030206558e-06,
      "loss": 0.0578,
      "step": 1583
    },
    {
      "epoch": 0.3604095563139932,
      "grad_norm": 1.4265200616837415,
      "learning_rate": 1.2340632603960774e-06,
      "loss": 0.0533,
      "step": 1584
    },
    {
      "epoch": 0.36063708759954494,
      "grad_norm": 1.5565200581725265,
      "learning_rate": 1.2340432053233615e-06,
      "loss": 0.0558,
      "step": 1585
    },
    {
      "epoch": 0.3608646188850967,
      "grad_norm": 0.8252310606061587,
      "learning_rate": 1.2340231378029177e-06,
      "loss": 0.0415,
      "step": 1586
    },
    {
      "epoch": 0.36109215017064844,
      "grad_norm": 1.1625630250416708,
      "learning_rate": 1.2340030578351564e-06,
      "loss": 0.0536,
      "step": 1587
    },
    {
      "epoch": 0.36131968145620025,
      "grad_norm": 1.4833717042031473,
      "learning_rate": 1.2339829654204878e-06,
      "loss": 0.054,
      "step": 1588
    },
    {
      "epoch": 0.361547212741752,
      "grad_norm": 1.2867709045464473,
      "learning_rate": 1.2339628605593229e-06,
      "loss": 0.0486,
      "step": 1589
    },
    {
      "epoch": 0.36177474402730375,
      "grad_norm": 1.7822365018575843,
      "learning_rate": 1.2339427432520722e-06,
      "loss": 0.0538,
      "step": 1590
    },
    {
      "epoch": 0.3620022753128555,
      "grad_norm": 1.7621725916894804,
      "learning_rate": 1.2339226134991471e-06,
      "loss": 0.0742,
      "step": 1591
    },
    {
      "epoch": 0.36222980659840726,
      "grad_norm": 2.2648964065061223,
      "learning_rate": 1.2339024713009592e-06,
      "loss": 0.0702,
      "step": 1592
    },
    {
      "epoch": 0.36245733788395906,
      "grad_norm": 1.3626520454398885,
      "learning_rate": 1.2338823166579197e-06,
      "loss": 0.0896,
      "step": 1593
    },
    {
      "epoch": 0.3626848691695108,
      "grad_norm": 1.3432892967430223,
      "learning_rate": 1.2338621495704409e-06,
      "loss": 0.0442,
      "step": 1594
    },
    {
      "epoch": 0.36291240045506257,
      "grad_norm": 1.920266034051075,
      "learning_rate": 1.2338419700389349e-06,
      "loss": 0.0949,
      "step": 1595
    },
    {
      "epoch": 0.3631399317406143,
      "grad_norm": 1.556412496157174,
      "learning_rate": 1.2338217780638137e-06,
      "loss": 0.0496,
      "step": 1596
    },
    {
      "epoch": 0.36336746302616607,
      "grad_norm": 2.160817778943729,
      "learning_rate": 1.2338015736454908e-06,
      "loss": 0.1306,
      "step": 1597
    },
    {
      "epoch": 0.3635949943117179,
      "grad_norm": 1.3086877468472204,
      "learning_rate": 1.2337813567843784e-06,
      "loss": 0.0441,
      "step": 1598
    },
    {
      "epoch": 0.36382252559726963,
      "grad_norm": 1.1991943484955294,
      "learning_rate": 1.2337611274808901e-06,
      "loss": 0.0684,
      "step": 1599
    },
    {
      "epoch": 0.3640500568828214,
      "grad_norm": 1.223483082152576,
      "learning_rate": 1.2337408857354394e-06,
      "loss": 0.0464,
      "step": 1600
    },
    {
      "epoch": 0.36427758816837313,
      "grad_norm": 0.8679079039579355,
      "learning_rate": 1.2337206315484396e-06,
      "loss": 0.036,
      "step": 1601
    },
    {
      "epoch": 0.36450511945392494,
      "grad_norm": 1.1879547293476054,
      "learning_rate": 1.2337003649203049e-06,
      "loss": 0.0481,
      "step": 1602
    },
    {
      "epoch": 0.3647326507394767,
      "grad_norm": 1.3048080561875424,
      "learning_rate": 1.2336800858514498e-06,
      "loss": 0.057,
      "step": 1603
    },
    {
      "epoch": 0.36496018202502845,
      "grad_norm": 1.3761316306984896,
      "learning_rate": 1.2336597943422883e-06,
      "loss": 0.0806,
      "step": 1604
    },
    {
      "epoch": 0.3651877133105802,
      "grad_norm": 1.3617767409383836,
      "learning_rate": 1.2336394903932353e-06,
      "loss": 0.0636,
      "step": 1605
    },
    {
      "epoch": 0.36541524459613195,
      "grad_norm": 0.8022215628059131,
      "learning_rate": 1.233619174004706e-06,
      "loss": 0.0419,
      "step": 1606
    },
    {
      "epoch": 0.36564277588168376,
      "grad_norm": 1.0273812494717272,
      "learning_rate": 1.233598845177115e-06,
      "loss": 0.052,
      "step": 1607
    },
    {
      "epoch": 0.3658703071672355,
      "grad_norm": 1.262245877196861,
      "learning_rate": 1.2335785039108787e-06,
      "loss": 0.0494,
      "step": 1608
    },
    {
      "epoch": 0.36609783845278726,
      "grad_norm": 1.32897975384147,
      "learning_rate": 1.233558150206412e-06,
      "loss": 0.069,
      "step": 1609
    },
    {
      "epoch": 0.366325369738339,
      "grad_norm": 1.2103326320958596,
      "learning_rate": 1.2335377840641314e-06,
      "loss": 0.0388,
      "step": 1610
    },
    {
      "epoch": 0.36655290102389076,
      "grad_norm": 1.6475678381055654,
      "learning_rate": 1.233517405484453e-06,
      "loss": 0.0584,
      "step": 1611
    },
    {
      "epoch": 0.3667804323094426,
      "grad_norm": 0.8570759088496013,
      "learning_rate": 1.2334970144677929e-06,
      "loss": 0.0268,
      "step": 1612
    },
    {
      "epoch": 0.3670079635949943,
      "grad_norm": 1.2115212249577887,
      "learning_rate": 1.2334766110145684e-06,
      "loss": 0.0492,
      "step": 1613
    },
    {
      "epoch": 0.3672354948805461,
      "grad_norm": 1.6505586244119963,
      "learning_rate": 1.2334561951251967e-06,
      "loss": 0.0696,
      "step": 1614
    },
    {
      "epoch": 0.36746302616609783,
      "grad_norm": 1.0033818903908105,
      "learning_rate": 1.2334357668000943e-06,
      "loss": 0.0485,
      "step": 1615
    },
    {
      "epoch": 0.3676905574516496,
      "grad_norm": 0.8359701120507522,
      "learning_rate": 1.2334153260396795e-06,
      "loss": 0.0428,
      "step": 1616
    },
    {
      "epoch": 0.3679180887372014,
      "grad_norm": 1.3307257141956392,
      "learning_rate": 1.2333948728443692e-06,
      "loss": 0.0609,
      "step": 1617
    },
    {
      "epoch": 0.36814562002275314,
      "grad_norm": 0.9161807224363929,
      "learning_rate": 1.2333744072145824e-06,
      "loss": 0.0357,
      "step": 1618
    },
    {
      "epoch": 0.3683731513083049,
      "grad_norm": 1.5055305786265103,
      "learning_rate": 1.2333539291507365e-06,
      "loss": 0.1025,
      "step": 1619
    },
    {
      "epoch": 0.36860068259385664,
      "grad_norm": 1.936271932194028,
      "learning_rate": 1.2333334386532507e-06,
      "loss": 0.073,
      "step": 1620
    },
    {
      "epoch": 0.3688282138794084,
      "grad_norm": 0.9636982724990819,
      "learning_rate": 1.2333129357225434e-06,
      "loss": 0.0366,
      "step": 1621
    },
    {
      "epoch": 0.3690557451649602,
      "grad_norm": 1.4864238412613773,
      "learning_rate": 1.2332924203590341e-06,
      "loss": 0.0579,
      "step": 1622
    },
    {
      "epoch": 0.36928327645051195,
      "grad_norm": 1.1141779480557945,
      "learning_rate": 1.2332718925631414e-06,
      "loss": 0.0626,
      "step": 1623
    },
    {
      "epoch": 0.3695108077360637,
      "grad_norm": 1.4180072004359519,
      "learning_rate": 1.2332513523352853e-06,
      "loss": 0.0583,
      "step": 1624
    },
    {
      "epoch": 0.36973833902161546,
      "grad_norm": 1.3854251776786142,
      "learning_rate": 1.2332307996758854e-06,
      "loss": 0.0499,
      "step": 1625
    },
    {
      "epoch": 0.3699658703071672,
      "grad_norm": 2.0139670039119766,
      "learning_rate": 1.233210234585362e-06,
      "loss": 0.083,
      "step": 1626
    },
    {
      "epoch": 0.370193401592719,
      "grad_norm": 1.951694104084049,
      "learning_rate": 1.2331896570641354e-06,
      "loss": 0.08,
      "step": 1627
    },
    {
      "epoch": 0.37042093287827077,
      "grad_norm": 1.2223571341809563,
      "learning_rate": 1.233169067112626e-06,
      "loss": 0.0515,
      "step": 1628
    },
    {
      "epoch": 0.3706484641638225,
      "grad_norm": 1.845536817940592,
      "learning_rate": 1.2331484647312545e-06,
      "loss": 0.1018,
      "step": 1629
    },
    {
      "epoch": 0.3708759954493743,
      "grad_norm": 1.2829067703722639,
      "learning_rate": 1.2331278499204423e-06,
      "loss": 0.0577,
      "step": 1630
    },
    {
      "epoch": 0.371103526734926,
      "grad_norm": 1.5585929030011767,
      "learning_rate": 1.2331072226806107e-06,
      "loss": 0.0466,
      "step": 1631
    },
    {
      "epoch": 0.37133105802047783,
      "grad_norm": 1.6427062066371947,
      "learning_rate": 1.233086583012181e-06,
      "loss": 0.0855,
      "step": 1632
    },
    {
      "epoch": 0.3715585893060296,
      "grad_norm": 1.1255582962725759,
      "learning_rate": 1.233065930915575e-06,
      "loss": 0.0514,
      "step": 1633
    },
    {
      "epoch": 0.37178612059158134,
      "grad_norm": 1.1524468415244595,
      "learning_rate": 1.2330452663912155e-06,
      "loss": 0.0489,
      "step": 1634
    },
    {
      "epoch": 0.3720136518771331,
      "grad_norm": 3.45030750590545,
      "learning_rate": 1.233024589439524e-06,
      "loss": 0.0984,
      "step": 1635
    },
    {
      "epoch": 0.3722411831626849,
      "grad_norm": 1.3099017670753825,
      "learning_rate": 1.2330039000609233e-06,
      "loss": 0.0402,
      "step": 1636
    },
    {
      "epoch": 0.37246871444823665,
      "grad_norm": 1.9185688290740377,
      "learning_rate": 1.2329831982558365e-06,
      "loss": 0.0612,
      "step": 1637
    },
    {
      "epoch": 0.3726962457337884,
      "grad_norm": 1.0515979623712708,
      "learning_rate": 1.2329624840246867e-06,
      "loss": 0.0341,
      "step": 1638
    },
    {
      "epoch": 0.37292377701934015,
      "grad_norm": 1.0104966308830603,
      "learning_rate": 1.2329417573678974e-06,
      "loss": 0.0363,
      "step": 1639
    },
    {
      "epoch": 0.3731513083048919,
      "grad_norm": 0.8767867018961214,
      "learning_rate": 1.2329210182858915e-06,
      "loss": 0.0358,
      "step": 1640
    },
    {
      "epoch": 0.3733788395904437,
      "grad_norm": 1.2786801769613858,
      "learning_rate": 1.2329002667790937e-06,
      "loss": 0.0727,
      "step": 1641
    },
    {
      "epoch": 0.37360637087599546,
      "grad_norm": 1.0430455326740677,
      "learning_rate": 1.2328795028479275e-06,
      "loss": 0.0293,
      "step": 1642
    },
    {
      "epoch": 0.3738339021615472,
      "grad_norm": 1.3214304805194543,
      "learning_rate": 1.2328587264928176e-06,
      "loss": 0.0487,
      "step": 1643
    },
    {
      "epoch": 0.37406143344709897,
      "grad_norm": 2.17352926519868,
      "learning_rate": 1.2328379377141885e-06,
      "loss": 0.1179,
      "step": 1644
    },
    {
      "epoch": 0.3742889647326507,
      "grad_norm": 1.6384210150692329,
      "learning_rate": 1.2328171365124655e-06,
      "loss": 0.0356,
      "step": 1645
    },
    {
      "epoch": 0.3745164960182025,
      "grad_norm": 1.0746654458597755,
      "learning_rate": 1.2327963228880733e-06,
      "loss": 0.0348,
      "step": 1646
    },
    {
      "epoch": 0.3747440273037543,
      "grad_norm": 1.1554129673310896,
      "learning_rate": 1.2327754968414372e-06,
      "loss": 0.0559,
      "step": 1647
    },
    {
      "epoch": 0.37497155858930603,
      "grad_norm": 1.381713868304537,
      "learning_rate": 1.232754658372983e-06,
      "loss": 0.0429,
      "step": 1648
    },
    {
      "epoch": 0.3751990898748578,
      "grad_norm": 1.0461255300086107,
      "learning_rate": 1.2327338074831366e-06,
      "loss": 0.0498,
      "step": 1649
    },
    {
      "epoch": 0.37542662116040953,
      "grad_norm": 0.9696935937818977,
      "learning_rate": 1.2327129441723242e-06,
      "loss": 0.0282,
      "step": 1650
    },
    {
      "epoch": 0.37565415244596134,
      "grad_norm": 2.070008088112287,
      "learning_rate": 1.2326920684409724e-06,
      "loss": 0.0936,
      "step": 1651
    },
    {
      "epoch": 0.3758816837315131,
      "grad_norm": 1.9843357113833955,
      "learning_rate": 1.2326711802895077e-06,
      "loss": 0.068,
      "step": 1652
    },
    {
      "epoch": 0.37610921501706485,
      "grad_norm": 1.3824487035448465,
      "learning_rate": 1.2326502797183568e-06,
      "loss": 0.0609,
      "step": 1653
    },
    {
      "epoch": 0.3763367463026166,
      "grad_norm": 1.4651223595589276,
      "learning_rate": 1.2326293667279472e-06,
      "loss": 0.0448,
      "step": 1654
    },
    {
      "epoch": 0.37656427758816835,
      "grad_norm": 0.9774603013378036,
      "learning_rate": 1.232608441318706e-06,
      "loss": 0.0282,
      "step": 1655
    },
    {
      "epoch": 0.37679180887372016,
      "grad_norm": 1.1003503115295237,
      "learning_rate": 1.232587503491061e-06,
      "loss": 0.0447,
      "step": 1656
    },
    {
      "epoch": 0.3770193401592719,
      "grad_norm": 1.8648740699841149,
      "learning_rate": 1.2325665532454403e-06,
      "loss": 0.0879,
      "step": 1657
    },
    {
      "epoch": 0.37724687144482366,
      "grad_norm": 1.4508296738594828,
      "learning_rate": 1.2325455905822719e-06,
      "loss": 0.048,
      "step": 1658
    },
    {
      "epoch": 0.3774744027303754,
      "grad_norm": 1.5446190575894356,
      "learning_rate": 1.2325246155019844e-06,
      "loss": 0.0713,
      "step": 1659
    },
    {
      "epoch": 0.37770193401592717,
      "grad_norm": 0.657557765969774,
      "learning_rate": 1.2325036280050063e-06,
      "loss": 0.0252,
      "step": 1660
    },
    {
      "epoch": 0.377929465301479,
      "grad_norm": 0.7023883517908608,
      "learning_rate": 1.2324826280917664e-06,
      "loss": 0.0222,
      "step": 1661
    },
    {
      "epoch": 0.3781569965870307,
      "grad_norm": 1.2534079220919763,
      "learning_rate": 1.2324616157626943e-06,
      "loss": 0.0467,
      "step": 1662
    },
    {
      "epoch": 0.3783845278725825,
      "grad_norm": 1.4009224674260483,
      "learning_rate": 1.2324405910182195e-06,
      "loss": 0.0796,
      "step": 1663
    },
    {
      "epoch": 0.37861205915813423,
      "grad_norm": 2.444160860575137,
      "learning_rate": 1.2324195538587713e-06,
      "loss": 0.0772,
      "step": 1664
    },
    {
      "epoch": 0.378839590443686,
      "grad_norm": 1.1670387195512752,
      "learning_rate": 1.23239850428478e-06,
      "loss": 0.0648,
      "step": 1665
    },
    {
      "epoch": 0.3790671217292378,
      "grad_norm": 1.1973819971714186,
      "learning_rate": 1.2323774422966756e-06,
      "loss": 0.0466,
      "step": 1666
    },
    {
      "epoch": 0.37929465301478954,
      "grad_norm": 2.3889623346599484,
      "learning_rate": 1.2323563678948885e-06,
      "loss": 0.0995,
      "step": 1667
    },
    {
      "epoch": 0.3795221843003413,
      "grad_norm": 1.0719186425987624,
      "learning_rate": 1.2323352810798498e-06,
      "loss": 0.0434,
      "step": 1668
    },
    {
      "epoch": 0.37974971558589304,
      "grad_norm": 1.5135458293019917,
      "learning_rate": 1.23231418185199e-06,
      "loss": 0.1,
      "step": 1669
    },
    {
      "epoch": 0.3799772468714448,
      "grad_norm": 1.6715828908867492,
      "learning_rate": 1.2322930702117406e-06,
      "loss": 0.0654,
      "step": 1670
    },
    {
      "epoch": 0.3802047781569966,
      "grad_norm": 1.1117468704392939,
      "learning_rate": 1.232271946159533e-06,
      "loss": 0.0391,
      "step": 1671
    },
    {
      "epoch": 0.38043230944254836,
      "grad_norm": 1.7072937667796886,
      "learning_rate": 1.2322508096957992e-06,
      "loss": 0.0597,
      "step": 1672
    },
    {
      "epoch": 0.3806598407281001,
      "grad_norm": 2.695721182769726,
      "learning_rate": 1.2322296608209709e-06,
      "loss": 0.099,
      "step": 1673
    },
    {
      "epoch": 0.38088737201365186,
      "grad_norm": 2.29172385709693,
      "learning_rate": 1.2322084995354805e-06,
      "loss": 0.0939,
      "step": 1674
    },
    {
      "epoch": 0.38111490329920367,
      "grad_norm": 1.1798892776179337,
      "learning_rate": 1.2321873258397602e-06,
      "loss": 0.0544,
      "step": 1675
    },
    {
      "epoch": 0.3813424345847554,
      "grad_norm": 1.4030630144768808,
      "learning_rate": 1.232166139734243e-06,
      "loss": 0.0533,
      "step": 1676
    },
    {
      "epoch": 0.38156996587030717,
      "grad_norm": 1.2796792180280614,
      "learning_rate": 1.2321449412193622e-06,
      "loss": 0.0607,
      "step": 1677
    },
    {
      "epoch": 0.3817974971558589,
      "grad_norm": 1.3330548972713963,
      "learning_rate": 1.2321237302955505e-06,
      "loss": 0.0474,
      "step": 1678
    },
    {
      "epoch": 0.3820250284414107,
      "grad_norm": 1.0575562905738431,
      "learning_rate": 1.2321025069632416e-06,
      "loss": 0.0495,
      "step": 1679
    },
    {
      "epoch": 0.3822525597269625,
      "grad_norm": 1.1768156879508973,
      "learning_rate": 1.2320812712228694e-06,
      "loss": 0.0451,
      "step": 1680
    },
    {
      "epoch": 0.38248009101251423,
      "grad_norm": 0.7767797500459915,
      "learning_rate": 1.2320600230748677e-06,
      "loss": 0.0347,
      "step": 1681
    },
    {
      "epoch": 0.382707622298066,
      "grad_norm": 1.0249897852947059,
      "learning_rate": 1.232038762519671e-06,
      "loss": 0.0337,
      "step": 1682
    },
    {
      "epoch": 0.38293515358361774,
      "grad_norm": 1.6718567212147661,
      "learning_rate": 1.2320174895577138e-06,
      "loss": 0.1132,
      "step": 1683
    },
    {
      "epoch": 0.3831626848691695,
      "grad_norm": 0.994384807721514,
      "learning_rate": 1.2319962041894307e-06,
      "loss": 0.0362,
      "step": 1684
    },
    {
      "epoch": 0.3833902161547213,
      "grad_norm": 2.303334030981133,
      "learning_rate": 1.2319749064152569e-06,
      "loss": 0.1277,
      "step": 1685
    },
    {
      "epoch": 0.38361774744027305,
      "grad_norm": 1.609337504264491,
      "learning_rate": 1.2319535962356277e-06,
      "loss": 0.0575,
      "step": 1686
    },
    {
      "epoch": 0.3838452787258248,
      "grad_norm": 1.2011777956026055,
      "learning_rate": 1.2319322736509784e-06,
      "loss": 0.0552,
      "step": 1687
    },
    {
      "epoch": 0.38407281001137655,
      "grad_norm": 1.7322047315095004,
      "learning_rate": 1.2319109386617452e-06,
      "loss": 0.0753,
      "step": 1688
    },
    {
      "epoch": 0.3843003412969283,
      "grad_norm": 1.4045767611712419,
      "learning_rate": 1.2318895912683638e-06,
      "loss": 0.0726,
      "step": 1689
    },
    {
      "epoch": 0.3845278725824801,
      "grad_norm": 0.8734396508754521,
      "learning_rate": 1.2318682314712706e-06,
      "loss": 0.029,
      "step": 1690
    },
    {
      "epoch": 0.38475540386803186,
      "grad_norm": 3.125984298027021,
      "learning_rate": 1.2318468592709022e-06,
      "loss": 0.0794,
      "step": 1691
    },
    {
      "epoch": 0.3849829351535836,
      "grad_norm": 0.9959306519128199,
      "learning_rate": 1.2318254746676954e-06,
      "loss": 0.0504,
      "step": 1692
    },
    {
      "epoch": 0.38521046643913537,
      "grad_norm": 1.0411287511380087,
      "learning_rate": 1.2318040776620872e-06,
      "loss": 0.0423,
      "step": 1693
    },
    {
      "epoch": 0.3854379977246871,
      "grad_norm": 1.299794772066647,
      "learning_rate": 1.231782668254515e-06,
      "loss": 0.0633,
      "step": 1694
    },
    {
      "epoch": 0.3856655290102389,
      "grad_norm": 1.5302626083298043,
      "learning_rate": 1.2317612464454161e-06,
      "loss": 0.0581,
      "step": 1695
    },
    {
      "epoch": 0.3858930602957907,
      "grad_norm": 1.3051944839085843,
      "learning_rate": 1.2317398122352289e-06,
      "loss": 0.0524,
      "step": 1696
    },
    {
      "epoch": 0.38612059158134243,
      "grad_norm": 2.5811565293035166,
      "learning_rate": 1.2317183656243912e-06,
      "loss": 0.1304,
      "step": 1697
    },
    {
      "epoch": 0.3863481228668942,
      "grad_norm": 0.7679176052329142,
      "learning_rate": 1.231696906613341e-06,
      "loss": 0.0311,
      "step": 1698
    },
    {
      "epoch": 0.38657565415244594,
      "grad_norm": 1.2458183734125623,
      "learning_rate": 1.2316754352025173e-06,
      "loss": 0.0601,
      "step": 1699
    },
    {
      "epoch": 0.38680318543799774,
      "grad_norm": 1.4002525637777081,
      "learning_rate": 1.2316539513923585e-06,
      "loss": 0.052,
      "step": 1700
    },
    {
      "epoch": 0.3870307167235495,
      "grad_norm": 1.40244177955916,
      "learning_rate": 1.2316324551833042e-06,
      "loss": 0.0798,
      "step": 1701
    },
    {
      "epoch": 0.38725824800910125,
      "grad_norm": 0.636865674778868,
      "learning_rate": 1.2316109465757934e-06,
      "loss": 0.0236,
      "step": 1702
    },
    {
      "epoch": 0.387485779294653,
      "grad_norm": 0.911548468016125,
      "learning_rate": 1.231589425570266e-06,
      "loss": 0.048,
      "step": 1703
    },
    {
      "epoch": 0.38771331058020475,
      "grad_norm": 0.916105640198358,
      "learning_rate": 1.2315678921671615e-06,
      "loss": 0.0559,
      "step": 1704
    },
    {
      "epoch": 0.38794084186575656,
      "grad_norm": 1.1465421388050945,
      "learning_rate": 1.2315463463669202e-06,
      "loss": 0.0561,
      "step": 1705
    },
    {
      "epoch": 0.3881683731513083,
      "grad_norm": 1.441271671684643,
      "learning_rate": 1.2315247881699825e-06,
      "loss": 0.0507,
      "step": 1706
    },
    {
      "epoch": 0.38839590443686006,
      "grad_norm": 1.5270606085562182,
      "learning_rate": 1.2315032175767887e-06,
      "loss": 0.049,
      "step": 1707
    },
    {
      "epoch": 0.3886234357224118,
      "grad_norm": 1.5250540029402313,
      "learning_rate": 1.23148163458778e-06,
      "loss": 0.0597,
      "step": 1708
    },
    {
      "epoch": 0.3888509670079636,
      "grad_norm": 1.4207097878273431,
      "learning_rate": 1.2314600392033974e-06,
      "loss": 0.0676,
      "step": 1709
    },
    {
      "epoch": 0.3890784982935154,
      "grad_norm": 1.5896950089365356,
      "learning_rate": 1.2314384314240824e-06,
      "loss": 0.0617,
      "step": 1710
    },
    {
      "epoch": 0.3893060295790671,
      "grad_norm": 2.036043003063475,
      "learning_rate": 1.2314168112502765e-06,
      "loss": 0.1324,
      "step": 1711
    },
    {
      "epoch": 0.3895335608646189,
      "grad_norm": 1.1640099693184898,
      "learning_rate": 1.2313951786824213e-06,
      "loss": 0.0634,
      "step": 1712
    },
    {
      "epoch": 0.38976109215017063,
      "grad_norm": 1.4865385157328637,
      "learning_rate": 1.2313735337209593e-06,
      "loss": 0.0584,
      "step": 1713
    },
    {
      "epoch": 0.38998862343572244,
      "grad_norm": 1.252450965977985,
      "learning_rate": 1.231351876366333e-06,
      "loss": 0.0368,
      "step": 1714
    },
    {
      "epoch": 0.3902161547212742,
      "grad_norm": 1.4371658281059463,
      "learning_rate": 1.2313302066189846e-06,
      "loss": 0.0557,
      "step": 1715
    },
    {
      "epoch": 0.39044368600682594,
      "grad_norm": 1.579902726372128,
      "learning_rate": 1.2313085244793573e-06,
      "loss": 0.063,
      "step": 1716
    },
    {
      "epoch": 0.3906712172923777,
      "grad_norm": 2.3512912102608277,
      "learning_rate": 1.2312868299478944e-06,
      "loss": 0.0774,
      "step": 1717
    },
    {
      "epoch": 0.39089874857792944,
      "grad_norm": 1.2308435632473214,
      "learning_rate": 1.2312651230250387e-06,
      "loss": 0.0585,
      "step": 1718
    },
    {
      "epoch": 0.39112627986348125,
      "grad_norm": 1.0173462925036059,
      "learning_rate": 1.2312434037112345e-06,
      "loss": 0.0306,
      "step": 1719
    },
    {
      "epoch": 0.391353811149033,
      "grad_norm": 1.3611629215931131,
      "learning_rate": 1.2312216720069251e-06,
      "loss": 0.0822,
      "step": 1720
    },
    {
      "epoch": 0.39158134243458476,
      "grad_norm": 2.0854012174946135,
      "learning_rate": 1.2311999279125552e-06,
      "loss": 0.0877,
      "step": 1721
    },
    {
      "epoch": 0.3918088737201365,
      "grad_norm": 0.9865613588959612,
      "learning_rate": 1.2311781714285689e-06,
      "loss": 0.0557,
      "step": 1722
    },
    {
      "epoch": 0.39203640500568826,
      "grad_norm": 1.4304461856241661,
      "learning_rate": 1.231156402555411e-06,
      "loss": 0.0586,
      "step": 1723
    },
    {
      "epoch": 0.39226393629124007,
      "grad_norm": 1.0337361927182938,
      "learning_rate": 1.2311346212935262e-06,
      "loss": 0.0433,
      "step": 1724
    },
    {
      "epoch": 0.3924914675767918,
      "grad_norm": 0.7476757498111144,
      "learning_rate": 1.23111282764336e-06,
      "loss": 0.0303,
      "step": 1725
    },
    {
      "epoch": 0.39271899886234357,
      "grad_norm": 1.1404795389862592,
      "learning_rate": 1.2310910216053576e-06,
      "loss": 0.0518,
      "step": 1726
    },
    {
      "epoch": 0.3929465301478953,
      "grad_norm": 1.6483693490580467,
      "learning_rate": 1.2310692031799646e-06,
      "loss": 0.0754,
      "step": 1727
    },
    {
      "epoch": 0.3931740614334471,
      "grad_norm": 1.0959398843175312,
      "learning_rate": 1.2310473723676272e-06,
      "loss": 0.0444,
      "step": 1728
    },
    {
      "epoch": 0.3934015927189989,
      "grad_norm": 4.125333379715717,
      "learning_rate": 1.2310255291687913e-06,
      "loss": 0.0891,
      "step": 1729
    },
    {
      "epoch": 0.39362912400455063,
      "grad_norm": 1.4792691340855277,
      "learning_rate": 1.2310036735839037e-06,
      "loss": 0.0414,
      "step": 1730
    },
    {
      "epoch": 0.3938566552901024,
      "grad_norm": 1.4532148859913319,
      "learning_rate": 1.2309818056134108e-06,
      "loss": 0.0631,
      "step": 1731
    },
    {
      "epoch": 0.39408418657565414,
      "grad_norm": 0.7598305700170358,
      "learning_rate": 1.2309599252577593e-06,
      "loss": 0.026,
      "step": 1732
    },
    {
      "epoch": 0.3943117178612059,
      "grad_norm": 1.3606922738087333,
      "learning_rate": 1.230938032517397e-06,
      "loss": 0.0845,
      "step": 1733
    },
    {
      "epoch": 0.3945392491467577,
      "grad_norm": 0.9458332790402086,
      "learning_rate": 1.2309161273927708e-06,
      "loss": 0.0405,
      "step": 1734
    },
    {
      "epoch": 0.39476678043230945,
      "grad_norm": 1.338928802144235,
      "learning_rate": 1.2308942098843289e-06,
      "loss": 0.0506,
      "step": 1735
    },
    {
      "epoch": 0.3949943117178612,
      "grad_norm": 0.9875822099203543,
      "learning_rate": 1.2308722799925188e-06,
      "loss": 0.0474,
      "step": 1736
    },
    {
      "epoch": 0.39522184300341295,
      "grad_norm": 1.144643764479203,
      "learning_rate": 1.2308503377177887e-06,
      "loss": 0.0482,
      "step": 1737
    },
    {
      "epoch": 0.3954493742889647,
      "grad_norm": 1.2433153307693834,
      "learning_rate": 1.2308283830605877e-06,
      "loss": 0.0382,
      "step": 1738
    },
    {
      "epoch": 0.3956769055745165,
      "grad_norm": 0.8868244265843614,
      "learning_rate": 1.230806416021364e-06,
      "loss": 0.0312,
      "step": 1739
    },
    {
      "epoch": 0.39590443686006827,
      "grad_norm": 1.2733746862834583,
      "learning_rate": 1.2307844366005665e-06,
      "loss": 0.0558,
      "step": 1740
    },
    {
      "epoch": 0.39613196814562,
      "grad_norm": 1.5929660791826068,
      "learning_rate": 1.2307624447986446e-06,
      "loss": 0.0939,
      "step": 1741
    },
    {
      "epoch": 0.39635949943117177,
      "grad_norm": 1.6706588043136787,
      "learning_rate": 1.2307404406160476e-06,
      "loss": 0.0856,
      "step": 1742
    },
    {
      "epoch": 0.3965870307167236,
      "grad_norm": 1.4925539377680157,
      "learning_rate": 1.2307184240532255e-06,
      "loss": 0.051,
      "step": 1743
    },
    {
      "epoch": 0.39681456200227533,
      "grad_norm": 0.8314484619002069,
      "learning_rate": 1.2306963951106283e-06,
      "loss": 0.0398,
      "step": 1744
    },
    {
      "epoch": 0.3970420932878271,
      "grad_norm": 1.8438134884760746,
      "learning_rate": 1.2306743537887058e-06,
      "loss": 0.1539,
      "step": 1745
    },
    {
      "epoch": 0.39726962457337883,
      "grad_norm": 1.1012953571620363,
      "learning_rate": 1.2306523000879086e-06,
      "loss": 0.0395,
      "step": 1746
    },
    {
      "epoch": 0.3974971558589306,
      "grad_norm": 1.2928925052093003,
      "learning_rate": 1.230630234008688e-06,
      "loss": 0.0807,
      "step": 1747
    },
    {
      "epoch": 0.3977246871444824,
      "grad_norm": 0.9015308252174471,
      "learning_rate": 1.2306081555514942e-06,
      "loss": 0.0286,
      "step": 1748
    },
    {
      "epoch": 0.39795221843003414,
      "grad_norm": 1.4467952739416505,
      "learning_rate": 1.2305860647167792e-06,
      "loss": 0.0557,
      "step": 1749
    },
    {
      "epoch": 0.3981797497155859,
      "grad_norm": 0.8808019982734657,
      "learning_rate": 1.2305639615049938e-06,
      "loss": 0.033,
      "step": 1750
    },
    {
      "epoch": 0.39840728100113765,
      "grad_norm": 1.3896950910533592,
      "learning_rate": 1.2305418459165902e-06,
      "loss": 0.0425,
      "step": 1751
    },
    {
      "epoch": 0.3986348122866894,
      "grad_norm": 1.0876862572580541,
      "learning_rate": 1.2305197179520203e-06,
      "loss": 0.0529,
      "step": 1752
    },
    {
      "epoch": 0.3988623435722412,
      "grad_norm": 1.3270997207033957,
      "learning_rate": 1.2304975776117362e-06,
      "loss": 0.0539,
      "step": 1753
    },
    {
      "epoch": 0.39908987485779296,
      "grad_norm": 0.7035727027322921,
      "learning_rate": 1.2304754248961906e-06,
      "loss": 0.0319,
      "step": 1754
    },
    {
      "epoch": 0.3993174061433447,
      "grad_norm": 1.2785559710612535,
      "learning_rate": 1.2304532598058363e-06,
      "loss": 0.0481,
      "step": 1755
    },
    {
      "epoch": 0.39954493742889646,
      "grad_norm": 1.0072143636243023,
      "learning_rate": 1.230431082341126e-06,
      "loss": 0.0481,
      "step": 1756
    },
    {
      "epoch": 0.3997724687144482,
      "grad_norm": 1.063335290490128,
      "learning_rate": 1.2304088925025133e-06,
      "loss": 0.0405,
      "step": 1757
    },
    {
      "epoch": 0.4,
      "grad_norm": 1.3383471149297288,
      "learning_rate": 1.2303866902904515e-06,
      "loss": 0.0551,
      "step": 1758
    },
    {
      "epoch": 0.4002275312855518,
      "grad_norm": 1.3624741606263786,
      "learning_rate": 1.2303644757053945e-06,
      "loss": 0.0608,
      "step": 1759
    },
    {
      "epoch": 0.4004550625711035,
      "grad_norm": 1.5467274319242612,
      "learning_rate": 1.2303422487477965e-06,
      "loss": 0.0824,
      "step": 1760
    },
    {
      "epoch": 0.4006825938566553,
      "grad_norm": 0.9214849308730754,
      "learning_rate": 1.230320009418111e-06,
      "loss": 0.0367,
      "step": 1761
    },
    {
      "epoch": 0.40091012514220703,
      "grad_norm": 1.1758563975131424,
      "learning_rate": 1.2302977577167937e-06,
      "loss": 0.0549,
      "step": 1762
    },
    {
      "epoch": 0.40113765642775884,
      "grad_norm": 1.2826994675732137,
      "learning_rate": 1.2302754936442986e-06,
      "loss": 0.0591,
      "step": 1763
    },
    {
      "epoch": 0.4013651877133106,
      "grad_norm": 1.1321972166341843,
      "learning_rate": 1.2302532172010809e-06,
      "loss": 0.0309,
      "step": 1764
    },
    {
      "epoch": 0.40159271899886234,
      "grad_norm": 0.8928863966424516,
      "learning_rate": 1.2302309283875958e-06,
      "loss": 0.0443,
      "step": 1765
    },
    {
      "epoch": 0.4018202502844141,
      "grad_norm": 0.9140748312352776,
      "learning_rate": 1.230208627204299e-06,
      "loss": 0.0462,
      "step": 1766
    },
    {
      "epoch": 0.40204778156996585,
      "grad_norm": 1.365652466902771,
      "learning_rate": 1.2301863136516463e-06,
      "loss": 0.0575,
      "step": 1767
    },
    {
      "epoch": 0.40227531285551765,
      "grad_norm": 1.8435555715075644,
      "learning_rate": 1.2301639877300937e-06,
      "loss": 0.0835,
      "step": 1768
    },
    {
      "epoch": 0.4025028441410694,
      "grad_norm": 1.0698988090252757,
      "learning_rate": 1.2301416494400974e-06,
      "loss": 0.039,
      "step": 1769
    },
    {
      "epoch": 0.40273037542662116,
      "grad_norm": 1.5392783242527026,
      "learning_rate": 1.2301192987821142e-06,
      "loss": 0.0635,
      "step": 1770
    },
    {
      "epoch": 0.4029579067121729,
      "grad_norm": 1.992163233200094,
      "learning_rate": 1.2300969357566008e-06,
      "loss": 0.0687,
      "step": 1771
    },
    {
      "epoch": 0.40318543799772466,
      "grad_norm": 1.1143248581214018,
      "learning_rate": 1.230074560364014e-06,
      "loss": 0.0431,
      "step": 1772
    },
    {
      "epoch": 0.40341296928327647,
      "grad_norm": 0.9736384244561508,
      "learning_rate": 1.2300521726048114e-06,
      "loss": 0.0502,
      "step": 1773
    },
    {
      "epoch": 0.4036405005688282,
      "grad_norm": 1.6202139485972007,
      "learning_rate": 1.2300297724794506e-06,
      "loss": 0.0501,
      "step": 1774
    },
    {
      "epoch": 0.40386803185437997,
      "grad_norm": 2.6531740859909854,
      "learning_rate": 1.2300073599883892e-06,
      "loss": 0.0674,
      "step": 1775
    },
    {
      "epoch": 0.4040955631399317,
      "grad_norm": 1.4336293766578745,
      "learning_rate": 1.2299849351320854e-06,
      "loss": 0.0513,
      "step": 1776
    },
    {
      "epoch": 0.40432309442548353,
      "grad_norm": 0.9141317140330615,
      "learning_rate": 1.2299624979109976e-06,
      "loss": 0.0407,
      "step": 1777
    },
    {
      "epoch": 0.4045506257110353,
      "grad_norm": 1.1533277774875146,
      "learning_rate": 1.229940048325584e-06,
      "loss": 0.0561,
      "step": 1778
    },
    {
      "epoch": 0.40477815699658704,
      "grad_norm": 1.502695522459793,
      "learning_rate": 1.229917586376304e-06,
      "loss": 0.045,
      "step": 1779
    },
    {
      "epoch": 0.4050056882821388,
      "grad_norm": 1.8452181062352009,
      "learning_rate": 1.2298951120636163e-06,
      "loss": 0.0805,
      "step": 1780
    },
    {
      "epoch": 0.40523321956769054,
      "grad_norm": 1.5002423894959154,
      "learning_rate": 1.2298726253879802e-06,
      "loss": 0.0599,
      "step": 1781
    },
    {
      "epoch": 0.40546075085324235,
      "grad_norm": 1.988959282415028,
      "learning_rate": 1.2298501263498557e-06,
      "loss": 0.0951,
      "step": 1782
    },
    {
      "epoch": 0.4056882821387941,
      "grad_norm": 1.0188710069870275,
      "learning_rate": 1.229827614949702e-06,
      "loss": 0.0451,
      "step": 1783
    },
    {
      "epoch": 0.40591581342434585,
      "grad_norm": 1.44475701906824,
      "learning_rate": 1.22980509118798e-06,
      "loss": 0.0707,
      "step": 1784
    },
    {
      "epoch": 0.4061433447098976,
      "grad_norm": 1.5656238194585963,
      "learning_rate": 1.2297825550651491e-06,
      "loss": 0.0658,
      "step": 1785
    },
    {
      "epoch": 0.40637087599544935,
      "grad_norm": 1.1011276149008262,
      "learning_rate": 1.2297600065816707e-06,
      "loss": 0.0786,
      "step": 1786
    },
    {
      "epoch": 0.40659840728100116,
      "grad_norm": 1.5148687422019658,
      "learning_rate": 1.229737445738005e-06,
      "loss": 0.0519,
      "step": 1787
    },
    {
      "epoch": 0.4068259385665529,
      "grad_norm": 2.402810934984467,
      "learning_rate": 1.2297148725346137e-06,
      "loss": 0.0844,
      "step": 1788
    },
    {
      "epoch": 0.40705346985210467,
      "grad_norm": 0.9099961278233849,
      "learning_rate": 1.229692286971958e-06,
      "loss": 0.0249,
      "step": 1789
    },
    {
      "epoch": 0.4072810011376564,
      "grad_norm": 1.4154667950181927,
      "learning_rate": 1.2296696890504992e-06,
      "loss": 0.0473,
      "step": 1790
    },
    {
      "epoch": 0.40750853242320817,
      "grad_norm": 0.6978007467353329,
      "learning_rate": 1.2296470787706993e-06,
      "loss": 0.0294,
      "step": 1791
    },
    {
      "epoch": 0.40773606370876,
      "grad_norm": 0.7284846144551249,
      "learning_rate": 1.2296244561330206e-06,
      "loss": 0.0294,
      "step": 1792
    },
    {
      "epoch": 0.40796359499431173,
      "grad_norm": 1.3976750286698911,
      "learning_rate": 1.2296018211379253e-06,
      "loss": 0.0719,
      "step": 1793
    },
    {
      "epoch": 0.4081911262798635,
      "grad_norm": 1.7561963591958132,
      "learning_rate": 1.229579173785876e-06,
      "loss": 0.0929,
      "step": 1794
    },
    {
      "epoch": 0.40841865756541523,
      "grad_norm": 1.2130144152724294,
      "learning_rate": 1.2295565140773357e-06,
      "loss": 0.055,
      "step": 1795
    },
    {
      "epoch": 0.408646188850967,
      "grad_norm": 1.5931552656845787,
      "learning_rate": 1.2295338420127673e-06,
      "loss": 0.0996,
      "step": 1796
    },
    {
      "epoch": 0.4088737201365188,
      "grad_norm": 2.213301169642664,
      "learning_rate": 1.2295111575926344e-06,
      "loss": 0.0757,
      "step": 1797
    },
    {
      "epoch": 0.40910125142207054,
      "grad_norm": 0.7230584272794064,
      "learning_rate": 1.2294884608174007e-06,
      "loss": 0.0268,
      "step": 1798
    },
    {
      "epoch": 0.4093287827076223,
      "grad_norm": 1.8735567431020124,
      "learning_rate": 1.2294657516875297e-06,
      "loss": 0.0777,
      "step": 1799
    },
    {
      "epoch": 0.40955631399317405,
      "grad_norm": 1.4508553485648286,
      "learning_rate": 1.229443030203486e-06,
      "loss": 0.0592,
      "step": 1800
    },
    {
      "epoch": 0.4097838452787258,
      "grad_norm": 1.649254441958079,
      "learning_rate": 1.2294202963657335e-06,
      "loss": 0.0663,
      "step": 1801
    },
    {
      "epoch": 0.4100113765642776,
      "grad_norm": 1.2184400652023568,
      "learning_rate": 1.2293975501747372e-06,
      "loss": 0.058,
      "step": 1802
    },
    {
      "epoch": 0.41023890784982936,
      "grad_norm": 1.3977455445674583,
      "learning_rate": 1.229374791630962e-06,
      "loss": 0.0513,
      "step": 1803
    },
    {
      "epoch": 0.4104664391353811,
      "grad_norm": 1.7032814483710292,
      "learning_rate": 1.2293520207348727e-06,
      "loss": 0.0526,
      "step": 1804
    },
    {
      "epoch": 0.41069397042093286,
      "grad_norm": 1.506685007842939,
      "learning_rate": 1.229329237486935e-06,
      "loss": 0.0584,
      "step": 1805
    },
    {
      "epoch": 0.4109215017064846,
      "grad_norm": 1.1882085263706217,
      "learning_rate": 1.2293064418876145e-06,
      "loss": 0.0379,
      "step": 1806
    },
    {
      "epoch": 0.4111490329920364,
      "grad_norm": 0.8676322681005716,
      "learning_rate": 1.2292836339373771e-06,
      "loss": 0.0423,
      "step": 1807
    },
    {
      "epoch": 0.4113765642775882,
      "grad_norm": 1.0357464791970643,
      "learning_rate": 1.2292608136366887e-06,
      "loss": 0.0384,
      "step": 1808
    },
    {
      "epoch": 0.4116040955631399,
      "grad_norm": 1.2686531396882506,
      "learning_rate": 1.2292379809860162e-06,
      "loss": 0.048,
      "step": 1809
    },
    {
      "epoch": 0.4118316268486917,
      "grad_norm": 0.6203768192444805,
      "learning_rate": 1.2292151359858258e-06,
      "loss": 0.028,
      "step": 1810
    },
    {
      "epoch": 0.4120591581342435,
      "grad_norm": 0.9010901499957009,
      "learning_rate": 1.2291922786365846e-06,
      "loss": 0.0341,
      "step": 1811
    },
    {
      "epoch": 0.41228668941979524,
      "grad_norm": 1.1445391336093391,
      "learning_rate": 1.2291694089387599e-06,
      "loss": 0.0479,
      "step": 1812
    },
    {
      "epoch": 0.412514220705347,
      "grad_norm": 1.6521327765404437,
      "learning_rate": 1.2291465268928187e-06,
      "loss": 0.1017,
      "step": 1813
    },
    {
      "epoch": 0.41274175199089874,
      "grad_norm": 1.79397743679785,
      "learning_rate": 1.2291236324992291e-06,
      "loss": 0.1033,
      "step": 1814
    },
    {
      "epoch": 0.4129692832764505,
      "grad_norm": 1.5152525260856162,
      "learning_rate": 1.229100725758459e-06,
      "loss": 0.0685,
      "step": 1815
    },
    {
      "epoch": 0.4131968145620023,
      "grad_norm": 1.483120669394339,
      "learning_rate": 1.2290778066709763e-06,
      "loss": 0.0487,
      "step": 1816
    },
    {
      "epoch": 0.41342434584755405,
      "grad_norm": 0.9540882649733521,
      "learning_rate": 1.2290548752372494e-06,
      "loss": 0.0311,
      "step": 1817
    },
    {
      "epoch": 0.4136518771331058,
      "grad_norm": 0.7907197059168389,
      "learning_rate": 1.2290319314577473e-06,
      "loss": 0.0319,
      "step": 1818
    },
    {
      "epoch": 0.41387940841865756,
      "grad_norm": 0.9184118117485006,
      "learning_rate": 1.2290089753329386e-06,
      "loss": 0.0364,
      "step": 1819
    },
    {
      "epoch": 0.4141069397042093,
      "grad_norm": 0.7846265232624993,
      "learning_rate": 1.2289860068632929e-06,
      "loss": 0.0294,
      "step": 1820
    },
    {
      "epoch": 0.4143344709897611,
      "grad_norm": 1.507108248570942,
      "learning_rate": 1.228963026049279e-06,
      "loss": 0.0664,
      "step": 1821
    },
    {
      "epoch": 0.41456200227531287,
      "grad_norm": 1.9973321813531546,
      "learning_rate": 1.228940032891367e-06,
      "loss": 0.0638,
      "step": 1822
    },
    {
      "epoch": 0.4147895335608646,
      "grad_norm": 1.5615035868464353,
      "learning_rate": 1.2289170273900272e-06,
      "loss": 0.0643,
      "step": 1823
    },
    {
      "epoch": 0.4150170648464164,
      "grad_norm": 0.9932945336330554,
      "learning_rate": 1.228894009545729e-06,
      "loss": 0.0645,
      "step": 1824
    },
    {
      "epoch": 0.4152445961319681,
      "grad_norm": 1.5587019691492865,
      "learning_rate": 1.2288709793589434e-06,
      "loss": 0.1044,
      "step": 1825
    },
    {
      "epoch": 0.41547212741751993,
      "grad_norm": 1.786679686535424,
      "learning_rate": 1.2288479368301408e-06,
      "loss": 0.0724,
      "step": 1826
    },
    {
      "epoch": 0.4156996587030717,
      "grad_norm": 1.690300262458873,
      "learning_rate": 1.2288248819597922e-06,
      "loss": 0.091,
      "step": 1827
    },
    {
      "epoch": 0.41592718998862344,
      "grad_norm": 0.7331004990082142,
      "learning_rate": 1.228801814748369e-06,
      "loss": 0.0219,
      "step": 1828
    },
    {
      "epoch": 0.4161547212741752,
      "grad_norm": 1.3161127926479836,
      "learning_rate": 1.2287787351963427e-06,
      "loss": 0.0549,
      "step": 1829
    },
    {
      "epoch": 0.41638225255972694,
      "grad_norm": 2.165472731866584,
      "learning_rate": 1.2287556433041845e-06,
      "loss": 0.0976,
      "step": 1830
    },
    {
      "epoch": 0.41660978384527875,
      "grad_norm": 1.5021088638901345,
      "learning_rate": 1.2287325390723669e-06,
      "loss": 0.0676,
      "step": 1831
    },
    {
      "epoch": 0.4168373151308305,
      "grad_norm": 1.776482868261773,
      "learning_rate": 1.2287094225013618e-06,
      "loss": 0.0955,
      "step": 1832
    },
    {
      "epoch": 0.41706484641638225,
      "grad_norm": 1.4781242043850178,
      "learning_rate": 1.2286862935916416e-06,
      "loss": 0.087,
      "step": 1833
    },
    {
      "epoch": 0.417292377701934,
      "grad_norm": 0.5015421036696045,
      "learning_rate": 1.2286631523436793e-06,
      "loss": 0.0127,
      "step": 1834
    },
    {
      "epoch": 0.41751990898748575,
      "grad_norm": 1.0956161872763561,
      "learning_rate": 1.2286399987579478e-06,
      "loss": 0.0569,
      "step": 1835
    },
    {
      "epoch": 0.41774744027303756,
      "grad_norm": 0.9950125745841572,
      "learning_rate": 1.2286168328349202e-06,
      "loss": 0.0464,
      "step": 1836
    },
    {
      "epoch": 0.4179749715585893,
      "grad_norm": 0.9975401241840096,
      "learning_rate": 1.2285936545750698e-06,
      "loss": 0.0484,
      "step": 1837
    },
    {
      "epoch": 0.41820250284414107,
      "grad_norm": 1.1811400302907533,
      "learning_rate": 1.2285704639788707e-06,
      "loss": 0.0543,
      "step": 1838
    },
    {
      "epoch": 0.4184300341296928,
      "grad_norm": 1.3042909512695973,
      "learning_rate": 1.2285472610467969e-06,
      "loss": 0.0426,
      "step": 1839
    },
    {
      "epoch": 0.41865756541524457,
      "grad_norm": 0.9151776818093462,
      "learning_rate": 1.228524045779322e-06,
      "loss": 0.0403,
      "step": 1840
    },
    {
      "epoch": 0.4188850967007964,
      "grad_norm": 1.6251189485123572,
      "learning_rate": 1.2285008181769212e-06,
      "loss": 0.0503,
      "step": 1841
    },
    {
      "epoch": 0.41911262798634813,
      "grad_norm": 1.3392895278247294,
      "learning_rate": 1.228477578240069e-06,
      "loss": 0.0489,
      "step": 1842
    },
    {
      "epoch": 0.4193401592718999,
      "grad_norm": 1.2464077367956892,
      "learning_rate": 1.22845432596924e-06,
      "loss": 0.0405,
      "step": 1843
    },
    {
      "epoch": 0.41956769055745163,
      "grad_norm": 1.1623606536272444,
      "learning_rate": 1.22843106136491e-06,
      "loss": 0.0358,
      "step": 1844
    },
    {
      "epoch": 0.4197952218430034,
      "grad_norm": 1.0651018669316255,
      "learning_rate": 1.2284077844275543e-06,
      "loss": 0.0416,
      "step": 1845
    },
    {
      "epoch": 0.4200227531285552,
      "grad_norm": 1.84496334967716,
      "learning_rate": 1.2283844951576484e-06,
      "loss": 0.0468,
      "step": 1846
    },
    {
      "epoch": 0.42025028441410694,
      "grad_norm": 1.0014097401871376,
      "learning_rate": 1.2283611935556686e-06,
      "loss": 0.0334,
      "step": 1847
    },
    {
      "epoch": 0.4204778156996587,
      "grad_norm": 1.2875305555648238,
      "learning_rate": 1.2283378796220909e-06,
      "loss": 0.0568,
      "step": 1848
    },
    {
      "epoch": 0.42070534698521045,
      "grad_norm": 2.1705410042003095,
      "learning_rate": 1.228314553357392e-06,
      "loss": 0.0814,
      "step": 1849
    },
    {
      "epoch": 0.42093287827076226,
      "grad_norm": 0.9864984577581978,
      "learning_rate": 1.2282912147620483e-06,
      "loss": 0.035,
      "step": 1850
    },
    {
      "epoch": 0.421160409556314,
      "grad_norm": 1.0361264682055424,
      "learning_rate": 1.2282678638365373e-06,
      "loss": 0.0555,
      "step": 1851
    },
    {
      "epoch": 0.42138794084186576,
      "grad_norm": 1.2255325101947971,
      "learning_rate": 1.2282445005813359e-06,
      "loss": 0.0487,
      "step": 1852
    },
    {
      "epoch": 0.4216154721274175,
      "grad_norm": 1.21533819095114,
      "learning_rate": 1.2282211249969217e-06,
      "loss": 0.0401,
      "step": 1853
    },
    {
      "epoch": 0.42184300341296926,
      "grad_norm": 1.3664419425404826,
      "learning_rate": 1.2281977370837725e-06,
      "loss": 0.0615,
      "step": 1854
    },
    {
      "epoch": 0.42207053469852107,
      "grad_norm": 1.0812048471076985,
      "learning_rate": 1.2281743368423662e-06,
      "loss": 0.0346,
      "step": 1855
    },
    {
      "epoch": 0.4222980659840728,
      "grad_norm": 1.60512477075616,
      "learning_rate": 1.2281509242731813e-06,
      "loss": 0.0945,
      "step": 1856
    },
    {
      "epoch": 0.4225255972696246,
      "grad_norm": 1.6388845900863238,
      "learning_rate": 1.228127499376696e-06,
      "loss": 0.0818,
      "step": 1857
    },
    {
      "epoch": 0.4227531285551763,
      "grad_norm": 0.9782543566946358,
      "learning_rate": 1.228104062153389e-06,
      "loss": 0.0417,
      "step": 1858
    },
    {
      "epoch": 0.4229806598407281,
      "grad_norm": 0.9911432963854596,
      "learning_rate": 1.2280806126037396e-06,
      "loss": 0.0652,
      "step": 1859
    },
    {
      "epoch": 0.4232081911262799,
      "grad_norm": 1.6481089463408192,
      "learning_rate": 1.2280571507282272e-06,
      "loss": 0.104,
      "step": 1860
    },
    {
      "epoch": 0.42343572241183164,
      "grad_norm": 1.1739506923046776,
      "learning_rate": 1.2280336765273309e-06,
      "loss": 0.0499,
      "step": 1861
    },
    {
      "epoch": 0.4236632536973834,
      "grad_norm": 1.0261276957174543,
      "learning_rate": 1.2280101900015306e-06,
      "loss": 0.0467,
      "step": 1862
    },
    {
      "epoch": 0.42389078498293514,
      "grad_norm": 1.4414669057942235,
      "learning_rate": 1.2279866911513064e-06,
      "loss": 0.0559,
      "step": 1863
    },
    {
      "epoch": 0.4241183162684869,
      "grad_norm": 1.4975361597068904,
      "learning_rate": 1.2279631799771386e-06,
      "loss": 0.0816,
      "step": 1864
    },
    {
      "epoch": 0.4243458475540387,
      "grad_norm": 1.9040003169842432,
      "learning_rate": 1.2279396564795077e-06,
      "loss": 0.0914,
      "step": 1865
    },
    {
      "epoch": 0.42457337883959045,
      "grad_norm": 1.3040862552521917,
      "learning_rate": 1.2279161206588944e-06,
      "loss": 0.0402,
      "step": 1866
    },
    {
      "epoch": 0.4248009101251422,
      "grad_norm": 1.4037836584117365,
      "learning_rate": 1.2278925725157798e-06,
      "loss": 0.0583,
      "step": 1867
    },
    {
      "epoch": 0.42502844141069396,
      "grad_norm": 0.9243184821654356,
      "learning_rate": 1.2278690120506451e-06,
      "loss": 0.0461,
      "step": 1868
    },
    {
      "epoch": 0.4252559726962457,
      "grad_norm": 0.9991936622504879,
      "learning_rate": 1.2278454392639722e-06,
      "loss": 0.0342,
      "step": 1869
    },
    {
      "epoch": 0.4254835039817975,
      "grad_norm": 0.8487853121482433,
      "learning_rate": 1.2278218541562422e-06,
      "loss": 0.0301,
      "step": 1870
    },
    {
      "epoch": 0.42571103526734927,
      "grad_norm": 1.6907022090274095,
      "learning_rate": 1.2277982567279377e-06,
      "loss": 0.0786,
      "step": 1871
    },
    {
      "epoch": 0.425938566552901,
      "grad_norm": 1.7052726832363685,
      "learning_rate": 1.2277746469795407e-06,
      "loss": 0.0737,
      "step": 1872
    },
    {
      "epoch": 0.4261660978384528,
      "grad_norm": 1.3329829339197778,
      "learning_rate": 1.227751024911534e-06,
      "loss": 0.055,
      "step": 1873
    },
    {
      "epoch": 0.4263936291240045,
      "grad_norm": 1.3273033221545125,
      "learning_rate": 1.2277273905244002e-06,
      "loss": 0.0484,
      "step": 1874
    },
    {
      "epoch": 0.42662116040955633,
      "grad_norm": 1.1982373731004552,
      "learning_rate": 1.2277037438186224e-06,
      "loss": 0.0466,
      "step": 1875
    },
    {
      "epoch": 0.4268486916951081,
      "grad_norm": 1.0093718042807764,
      "learning_rate": 1.2276800847946839e-06,
      "loss": 0.0363,
      "step": 1876
    },
    {
      "epoch": 0.42707622298065984,
      "grad_norm": 0.9811547399478636,
      "learning_rate": 1.227656413453068e-06,
      "loss": 0.0425,
      "step": 1877
    },
    {
      "epoch": 0.4273037542662116,
      "grad_norm": 1.146528135934977,
      "learning_rate": 1.227632729794259e-06,
      "loss": 0.0485,
      "step": 1878
    },
    {
      "epoch": 0.42753128555176334,
      "grad_norm": 1.5824604547576673,
      "learning_rate": 1.2276090338187403e-06,
      "loss": 0.0487,
      "step": 1879
    },
    {
      "epoch": 0.42775881683731515,
      "grad_norm": 1.3160117730674203,
      "learning_rate": 1.2275853255269967e-06,
      "loss": 0.0446,
      "step": 1880
    },
    {
      "epoch": 0.4279863481228669,
      "grad_norm": 1.3027261584989582,
      "learning_rate": 1.2275616049195129e-06,
      "loss": 0.0395,
      "step": 1881
    },
    {
      "epoch": 0.42821387940841865,
      "grad_norm": 1.0814544003095643,
      "learning_rate": 1.2275378719967733e-06,
      "loss": 0.0292,
      "step": 1882
    },
    {
      "epoch": 0.4284414106939704,
      "grad_norm": 0.7587179405824312,
      "learning_rate": 1.227514126759263e-06,
      "loss": 0.0256,
      "step": 1883
    },
    {
      "epoch": 0.4286689419795222,
      "grad_norm": 1.4079924288332035,
      "learning_rate": 1.2274903692074674e-06,
      "loss": 0.0636,
      "step": 1884
    },
    {
      "epoch": 0.42889647326507396,
      "grad_norm": 1.024771832790498,
      "learning_rate": 1.2274665993418722e-06,
      "loss": 0.0399,
      "step": 1885
    },
    {
      "epoch": 0.4291240045506257,
      "grad_norm": 1.1395572191018248,
      "learning_rate": 1.227442817162963e-06,
      "loss": 0.0427,
      "step": 1886
    },
    {
      "epoch": 0.42935153583617747,
      "grad_norm": 1.9966203221580272,
      "learning_rate": 1.227419022671226e-06,
      "loss": 0.1059,
      "step": 1887
    },
    {
      "epoch": 0.4295790671217292,
      "grad_norm": 1.6299484076057391,
      "learning_rate": 1.2273952158671472e-06,
      "loss": 0.0633,
      "step": 1888
    },
    {
      "epoch": 0.429806598407281,
      "grad_norm": 1.2903207410298125,
      "learning_rate": 1.2273713967512137e-06,
      "loss": 0.0618,
      "step": 1889
    },
    {
      "epoch": 0.4300341296928328,
      "grad_norm": 1.3026692339522694,
      "learning_rate": 1.227347565323912e-06,
      "loss": 0.0795,
      "step": 1890
    },
    {
      "epoch": 0.43026166097838453,
      "grad_norm": 1.175449820116291,
      "learning_rate": 1.2273237215857293e-06,
      "loss": 0.0401,
      "step": 1891
    },
    {
      "epoch": 0.4304891922639363,
      "grad_norm": 1.6959835304167243,
      "learning_rate": 1.2272998655371526e-06,
      "loss": 0.0825,
      "step": 1892
    },
    {
      "epoch": 0.43071672354948803,
      "grad_norm": 1.4197022325382747,
      "learning_rate": 1.22727599717867e-06,
      "loss": 0.0576,
      "step": 1893
    },
    {
      "epoch": 0.43094425483503984,
      "grad_norm": 1.2471339436500026,
      "learning_rate": 1.2272521165107687e-06,
      "loss": 0.0722,
      "step": 1894
    },
    {
      "epoch": 0.4311717861205916,
      "grad_norm": 1.3643717577874817,
      "learning_rate": 1.2272282235339372e-06,
      "loss": 0.0449,
      "step": 1895
    },
    {
      "epoch": 0.43139931740614335,
      "grad_norm": 1.1744612621898987,
      "learning_rate": 1.2272043182486638e-06,
      "loss": 0.0416,
      "step": 1896
    },
    {
      "epoch": 0.4316268486916951,
      "grad_norm": 1.0384312678976468,
      "learning_rate": 1.227180400655437e-06,
      "loss": 0.0399,
      "step": 1897
    },
    {
      "epoch": 0.43185437997724685,
      "grad_norm": 1.7243528985527803,
      "learning_rate": 1.2271564707547457e-06,
      "loss": 0.1495,
      "step": 1898
    },
    {
      "epoch": 0.43208191126279866,
      "grad_norm": 0.9548507541370914,
      "learning_rate": 1.227132528547079e-06,
      "loss": 0.0415,
      "step": 1899
    },
    {
      "epoch": 0.4323094425483504,
      "grad_norm": 1.3617841660517656,
      "learning_rate": 1.2271085740329261e-06,
      "loss": 0.0607,
      "step": 1900
    },
    {
      "epoch": 0.43253697383390216,
      "grad_norm": 1.6402508487098346,
      "learning_rate": 1.2270846072127764e-06,
      "loss": 0.064,
      "step": 1901
    },
    {
      "epoch": 0.4327645051194539,
      "grad_norm": 1.131250294762535,
      "learning_rate": 1.2270606280871205e-06,
      "loss": 0.034,
      "step": 1902
    },
    {
      "epoch": 0.43299203640500566,
      "grad_norm": 1.2541444770869459,
      "learning_rate": 1.2270366366564476e-06,
      "loss": 0.0354,
      "step": 1903
    },
    {
      "epoch": 0.43321956769055747,
      "grad_norm": 1.1469941807844757,
      "learning_rate": 1.2270126329212486e-06,
      "loss": 0.0568,
      "step": 1904
    },
    {
      "epoch": 0.4334470989761092,
      "grad_norm": 0.9467488862348428,
      "learning_rate": 1.2269886168820138e-06,
      "loss": 0.0433,
      "step": 1905
    },
    {
      "epoch": 0.433674630261661,
      "grad_norm": 2.421156992660593,
      "learning_rate": 1.2269645885392342e-06,
      "loss": 0.1575,
      "step": 1906
    },
    {
      "epoch": 0.43390216154721273,
      "grad_norm": 1.2853239102420944,
      "learning_rate": 1.226940547893401e-06,
      "loss": 0.0523,
      "step": 1907
    },
    {
      "epoch": 0.4341296928327645,
      "grad_norm": 0.7677705648665133,
      "learning_rate": 1.2269164949450052e-06,
      "loss": 0.0367,
      "step": 1908
    },
    {
      "epoch": 0.4343572241183163,
      "grad_norm": 1.7484475434950735,
      "learning_rate": 1.2268924296945387e-06,
      "loss": 0.0551,
      "step": 1909
    },
    {
      "epoch": 0.43458475540386804,
      "grad_norm": 1.8872851215095336,
      "learning_rate": 1.2268683521424932e-06,
      "loss": 0.0713,
      "step": 1910
    },
    {
      "epoch": 0.4348122866894198,
      "grad_norm": 1.2082493686491087,
      "learning_rate": 1.226844262289361e-06,
      "loss": 0.0411,
      "step": 1911
    },
    {
      "epoch": 0.43503981797497154,
      "grad_norm": 1.7963587387869338,
      "learning_rate": 1.2268201601356342e-06,
      "loss": 0.0829,
      "step": 1912
    },
    {
      "epoch": 0.4352673492605233,
      "grad_norm": 1.2079928907070248,
      "learning_rate": 1.2267960456818054e-06,
      "loss": 0.0399,
      "step": 1913
    },
    {
      "epoch": 0.4354948805460751,
      "grad_norm": 1.3116442013593013,
      "learning_rate": 1.2267719189283676e-06,
      "loss": 0.0483,
      "step": 1914
    },
    {
      "epoch": 0.43572241183162685,
      "grad_norm": 1.3635629491104682,
      "learning_rate": 1.2267477798758141e-06,
      "loss": 0.0415,
      "step": 1915
    },
    {
      "epoch": 0.4359499431171786,
      "grad_norm": 1.1777595750610328,
      "learning_rate": 1.2267236285246376e-06,
      "loss": 0.0635,
      "step": 1916
    },
    {
      "epoch": 0.43617747440273036,
      "grad_norm": 1.54425482806484,
      "learning_rate": 1.2266994648753325e-06,
      "loss": 0.0713,
      "step": 1917
    },
    {
      "epoch": 0.43640500568828217,
      "grad_norm": 1.3128341631768956,
      "learning_rate": 1.2266752889283923e-06,
      "loss": 0.064,
      "step": 1918
    },
    {
      "epoch": 0.4366325369738339,
      "grad_norm": 1.3460506007652075,
      "learning_rate": 1.226651100684311e-06,
      "loss": 0.0489,
      "step": 1919
    },
    {
      "epoch": 0.43686006825938567,
      "grad_norm": 1.3320121603668689,
      "learning_rate": 1.2266269001435829e-06,
      "loss": 0.0756,
      "step": 1920
    },
    {
      "epoch": 0.4370875995449374,
      "grad_norm": 1.1661019863589135,
      "learning_rate": 1.226602687306703e-06,
      "loss": 0.0313,
      "step": 1921
    },
    {
      "epoch": 0.4373151308304892,
      "grad_norm": 1.320896853626411,
      "learning_rate": 1.226578462174166e-06,
      "loss": 0.0421,
      "step": 1922
    },
    {
      "epoch": 0.437542662116041,
      "grad_norm": 2.159753678644095,
      "learning_rate": 1.2265542247464668e-06,
      "loss": 0.1307,
      "step": 1923
    },
    {
      "epoch": 0.43777019340159273,
      "grad_norm": 1.5563003879211055,
      "learning_rate": 1.226529975024101e-06,
      "loss": 0.0593,
      "step": 1924
    },
    {
      "epoch": 0.4379977246871445,
      "grad_norm": 0.9689169184561489,
      "learning_rate": 1.2265057130075641e-06,
      "loss": 0.0552,
      "step": 1925
    },
    {
      "epoch": 0.43822525597269624,
      "grad_norm": 1.4053723286715865,
      "learning_rate": 1.2264814386973523e-06,
      "loss": 0.0761,
      "step": 1926
    },
    {
      "epoch": 0.438452787258248,
      "grad_norm": 1.373860475632686,
      "learning_rate": 1.2264571520939612e-06,
      "loss": 0.061,
      "step": 1927
    },
    {
      "epoch": 0.4386803185437998,
      "grad_norm": 1.5229851486555135,
      "learning_rate": 1.2264328531978875e-06,
      "loss": 0.0871,
      "step": 1928
    },
    {
      "epoch": 0.43890784982935155,
      "grad_norm": 0.9510671111937553,
      "learning_rate": 1.2264085420096277e-06,
      "loss": 0.0372,
      "step": 1929
    },
    {
      "epoch": 0.4391353811149033,
      "grad_norm": 1.411692319953813,
      "learning_rate": 1.226384218529679e-06,
      "loss": 0.0744,
      "step": 1930
    },
    {
      "epoch": 0.43936291240045505,
      "grad_norm": 0.9987666361393835,
      "learning_rate": 1.2263598827585379e-06,
      "loss": 0.0392,
      "step": 1931
    },
    {
      "epoch": 0.4395904436860068,
      "grad_norm": 0.8661437238334854,
      "learning_rate": 1.2263355346967023e-06,
      "loss": 0.0418,
      "step": 1932
    },
    {
      "epoch": 0.4398179749715586,
      "grad_norm": 1.2249390316316808,
      "learning_rate": 1.2263111743446697e-06,
      "loss": 0.0375,
      "step": 1933
    },
    {
      "epoch": 0.44004550625711036,
      "grad_norm": 1.5944859887412242,
      "learning_rate": 1.2262868017029377e-06,
      "loss": 0.062,
      "step": 1934
    },
    {
      "epoch": 0.4402730375426621,
      "grad_norm": 1.3941560967832323,
      "learning_rate": 1.226262416772005e-06,
      "loss": 0.065,
      "step": 1935
    },
    {
      "epoch": 0.44050056882821387,
      "grad_norm": 0.9120544787723815,
      "learning_rate": 1.2262380195523696e-06,
      "loss": 0.0392,
      "step": 1936
    },
    {
      "epoch": 0.4407281001137656,
      "grad_norm": 1.764018460971706,
      "learning_rate": 1.2262136100445303e-06,
      "loss": 0.0805,
      "step": 1937
    },
    {
      "epoch": 0.4409556313993174,
      "grad_norm": 1.1144365245430385,
      "learning_rate": 1.2261891882489855e-06,
      "loss": 0.0394,
      "step": 1938
    },
    {
      "epoch": 0.4411831626848692,
      "grad_norm": 1.1821432018479405,
      "learning_rate": 1.226164754166235e-06,
      "loss": 0.0411,
      "step": 1939
    },
    {
      "epoch": 0.44141069397042093,
      "grad_norm": 1.2454202923866486,
      "learning_rate": 1.2261403077967778e-06,
      "loss": 0.0441,
      "step": 1940
    },
    {
      "epoch": 0.4416382252559727,
      "grad_norm": 1.271626432566145,
      "learning_rate": 1.2261158491411136e-06,
      "loss": 0.0452,
      "step": 1941
    },
    {
      "epoch": 0.44186575654152443,
      "grad_norm": 1.0596255221554163,
      "learning_rate": 1.2260913781997425e-06,
      "loss": 0.0394,
      "step": 1942
    },
    {
      "epoch": 0.44209328782707624,
      "grad_norm": 1.4836582697063838,
      "learning_rate": 1.2260668949731644e-06,
      "loss": 0.0579,
      "step": 1943
    },
    {
      "epoch": 0.442320819112628,
      "grad_norm": 1.687109016999551,
      "learning_rate": 1.2260423994618798e-06,
      "loss": 0.0544,
      "step": 1944
    },
    {
      "epoch": 0.44254835039817975,
      "grad_norm": 1.2086563543761475,
      "learning_rate": 1.2260178916663892e-06,
      "loss": 0.0351,
      "step": 1945
    },
    {
      "epoch": 0.4427758816837315,
      "grad_norm": 0.9870058509060623,
      "learning_rate": 1.2259933715871935e-06,
      "loss": 0.0458,
      "step": 1946
    },
    {
      "epoch": 0.44300341296928325,
      "grad_norm": 1.2117861018085792,
      "learning_rate": 1.2259688392247942e-06,
      "loss": 0.0466,
      "step": 1947
    },
    {
      "epoch": 0.44323094425483506,
      "grad_norm": 1.4026535262541224,
      "learning_rate": 1.2259442945796926e-06,
      "loss": 0.0695,
      "step": 1948
    },
    {
      "epoch": 0.4434584755403868,
      "grad_norm": 0.7731288867011977,
      "learning_rate": 1.2259197376523898e-06,
      "loss": 0.033,
      "step": 1949
    },
    {
      "epoch": 0.44368600682593856,
      "grad_norm": 0.7657804826696037,
      "learning_rate": 1.2258951684433883e-06,
      "loss": 0.041,
      "step": 1950
    },
    {
      "epoch": 0.4439135381114903,
      "grad_norm": 1.8552107522089532,
      "learning_rate": 1.2258705869531901e-06,
      "loss": 0.0944,
      "step": 1951
    },
    {
      "epoch": 0.4441410693970421,
      "grad_norm": 1.111099312238907,
      "learning_rate": 1.2258459931822974e-06,
      "loss": 0.0651,
      "step": 1952
    },
    {
      "epoch": 0.4443686006825939,
      "grad_norm": 1.6141386501644925,
      "learning_rate": 1.225821387131213e-06,
      "loss": 0.0574,
      "step": 1953
    },
    {
      "epoch": 0.4445961319681456,
      "grad_norm": 0.9741551342534741,
      "learning_rate": 1.22579676880044e-06,
      "loss": 0.0398,
      "step": 1954
    },
    {
      "epoch": 0.4448236632536974,
      "grad_norm": 1.0138692072583615,
      "learning_rate": 1.2257721381904811e-06,
      "loss": 0.0547,
      "step": 1955
    },
    {
      "epoch": 0.44505119453924913,
      "grad_norm": 1.4460053506809138,
      "learning_rate": 1.22574749530184e-06,
      "loss": 0.0624,
      "step": 1956
    },
    {
      "epoch": 0.44527872582480094,
      "grad_norm": 1.9350781623487736,
      "learning_rate": 1.2257228401350205e-06,
      "loss": 0.0694,
      "step": 1957
    },
    {
      "epoch": 0.4455062571103527,
      "grad_norm": 0.8261029534789511,
      "learning_rate": 1.2256981726905262e-06,
      "loss": 0.0295,
      "step": 1958
    },
    {
      "epoch": 0.44573378839590444,
      "grad_norm": 1.6085761442666993,
      "learning_rate": 1.2256734929688612e-06,
      "loss": 0.0699,
      "step": 1959
    },
    {
      "epoch": 0.4459613196814562,
      "grad_norm": 1.2323407982068377,
      "learning_rate": 1.2256488009705303e-06,
      "loss": 0.0409,
      "step": 1960
    },
    {
      "epoch": 0.44618885096700794,
      "grad_norm": 1.6360239341263807,
      "learning_rate": 1.225624096696038e-06,
      "loss": 0.0645,
      "step": 1961
    },
    {
      "epoch": 0.44641638225255975,
      "grad_norm": 0.9576742303137104,
      "learning_rate": 1.225599380145889e-06,
      "loss": 0.0531,
      "step": 1962
    },
    {
      "epoch": 0.4466439135381115,
      "grad_norm": 1.545322326325835,
      "learning_rate": 1.2255746513205889e-06,
      "loss": 0.058,
      "step": 1963
    },
    {
      "epoch": 0.44687144482366326,
      "grad_norm": 0.9718630260311564,
      "learning_rate": 1.2255499102206423e-06,
      "loss": 0.0303,
      "step": 1964
    },
    {
      "epoch": 0.447098976109215,
      "grad_norm": 1.992196356794709,
      "learning_rate": 1.2255251568465558e-06,
      "loss": 0.0711,
      "step": 1965
    },
    {
      "epoch": 0.44732650739476676,
      "grad_norm": 1.758374736199823,
      "learning_rate": 1.2255003911988348e-06,
      "loss": 0.097,
      "step": 1966
    },
    {
      "epoch": 0.44755403868031857,
      "grad_norm": 1.1999149710277988,
      "learning_rate": 1.2254756132779855e-06,
      "loss": 0.0762,
      "step": 1967
    },
    {
      "epoch": 0.4477815699658703,
      "grad_norm": 0.750734148431945,
      "learning_rate": 1.2254508230845144e-06,
      "loss": 0.0366,
      "step": 1968
    },
    {
      "epoch": 0.44800910125142207,
      "grad_norm": 0.9217944145406265,
      "learning_rate": 1.2254260206189283e-06,
      "loss": 0.0388,
      "step": 1969
    },
    {
      "epoch": 0.4482366325369738,
      "grad_norm": 1.0142165579258442,
      "learning_rate": 1.2254012058817337e-06,
      "loss": 0.0582,
      "step": 1970
    },
    {
      "epoch": 0.4484641638225256,
      "grad_norm": 1.7763236230644281,
      "learning_rate": 1.2253763788734384e-06,
      "loss": 0.049,
      "step": 1971
    },
    {
      "epoch": 0.4486916951080774,
      "grad_norm": 1.1246346527787483,
      "learning_rate": 1.2253515395945492e-06,
      "loss": 0.0605,
      "step": 1972
    },
    {
      "epoch": 0.44891922639362913,
      "grad_norm": 1.0749214639270723,
      "learning_rate": 1.2253266880455742e-06,
      "loss": 0.0617,
      "step": 1973
    },
    {
      "epoch": 0.4491467576791809,
      "grad_norm": 0.7801246159886271,
      "learning_rate": 1.225301824227021e-06,
      "loss": 0.028,
      "step": 1974
    },
    {
      "epoch": 0.44937428896473264,
      "grad_norm": 1.3195521199092928,
      "learning_rate": 1.2252769481393979e-06,
      "loss": 0.0471,
      "step": 1975
    },
    {
      "epoch": 0.4496018202502844,
      "grad_norm": 1.5596892760854768,
      "learning_rate": 1.2252520597832132e-06,
      "loss": 0.0337,
      "step": 1976
    },
    {
      "epoch": 0.4498293515358362,
      "grad_norm": 0.8505107972505791,
      "learning_rate": 1.2252271591589759e-06,
      "loss": 0.0398,
      "step": 1977
    },
    {
      "epoch": 0.45005688282138795,
      "grad_norm": 1.1325746921296718,
      "learning_rate": 1.2252022462671947e-06,
      "loss": 0.0486,
      "step": 1978
    },
    {
      "epoch": 0.4502844141069397,
      "grad_norm": 1.813164319739497,
      "learning_rate": 1.2251773211083789e-06,
      "loss": 0.0991,
      "step": 1979
    },
    {
      "epoch": 0.45051194539249145,
      "grad_norm": 2.0540653923112164,
      "learning_rate": 1.225152383683038e-06,
      "loss": 0.0985,
      "step": 1980
    },
    {
      "epoch": 0.4507394766780432,
      "grad_norm": 1.3064957694509771,
      "learning_rate": 1.225127433991681e-06,
      "loss": 0.0408,
      "step": 1981
    },
    {
      "epoch": 0.450967007963595,
      "grad_norm": 1.055797400460298,
      "learning_rate": 1.2251024720348186e-06,
      "loss": 0.0427,
      "step": 1982
    },
    {
      "epoch": 0.45119453924914676,
      "grad_norm": 1.1174321415520143,
      "learning_rate": 1.2250774978129606e-06,
      "loss": 0.0436,
      "step": 1983
    },
    {
      "epoch": 0.4514220705346985,
      "grad_norm": 1.7220669335096757,
      "learning_rate": 1.2250525113266175e-06,
      "loss": 0.0855,
      "step": 1984
    },
    {
      "epoch": 0.45164960182025027,
      "grad_norm": 1.061543218680689,
      "learning_rate": 1.2250275125763002e-06,
      "loss": 0.0386,
      "step": 1985
    },
    {
      "epoch": 0.451877133105802,
      "grad_norm": 1.4190135452138255,
      "learning_rate": 1.2250025015625194e-06,
      "loss": 0.0616,
      "step": 1986
    },
    {
      "epoch": 0.4521046643913538,
      "grad_norm": 1.8336951622768547,
      "learning_rate": 1.2249774782857863e-06,
      "loss": 0.0527,
      "step": 1987
    },
    {
      "epoch": 0.4523321956769056,
      "grad_norm": 1.3808921921483752,
      "learning_rate": 1.2249524427466123e-06,
      "loss": 0.0449,
      "step": 1988
    },
    {
      "epoch": 0.45255972696245733,
      "grad_norm": 1.0413790190683898,
      "learning_rate": 1.2249273949455092e-06,
      "loss": 0.0571,
      "step": 1989
    },
    {
      "epoch": 0.4527872582480091,
      "grad_norm": 1.1506775133869898,
      "learning_rate": 1.224902334882989e-06,
      "loss": 0.0362,
      "step": 1990
    },
    {
      "epoch": 0.4530147895335609,
      "grad_norm": 0.5659500114505296,
      "learning_rate": 1.2248772625595633e-06,
      "loss": 0.0243,
      "step": 1991
    },
    {
      "epoch": 0.45324232081911264,
      "grad_norm": 1.8858126631285193,
      "learning_rate": 1.2248521779757455e-06,
      "loss": 0.0921,
      "step": 1992
    },
    {
      "epoch": 0.4534698521046644,
      "grad_norm": 1.4534701341984988,
      "learning_rate": 1.2248270811320473e-06,
      "loss": 0.0626,
      "step": 1993
    },
    {
      "epoch": 0.45369738339021615,
      "grad_norm": 0.957481343542944,
      "learning_rate": 1.2248019720289826e-06,
      "loss": 0.0273,
      "step": 1994
    },
    {
      "epoch": 0.4539249146757679,
      "grad_norm": 0.8791382523836305,
      "learning_rate": 1.2247768506670639e-06,
      "loss": 0.0589,
      "step": 1995
    },
    {
      "epoch": 0.4541524459613197,
      "grad_norm": 2.2598713011024794,
      "learning_rate": 1.2247517170468045e-06,
      "loss": 0.0993,
      "step": 1996
    },
    {
      "epoch": 0.45437997724687146,
      "grad_norm": 0.9870730424185967,
      "learning_rate": 1.2247265711687186e-06,
      "loss": 0.0355,
      "step": 1997
    },
    {
      "epoch": 0.4546075085324232,
      "grad_norm": 1.1900169451289013,
      "learning_rate": 1.2247014130333198e-06,
      "loss": 0.0525,
      "step": 1998
    },
    {
      "epoch": 0.45483503981797496,
      "grad_norm": 1.384236012560106,
      "learning_rate": 1.2246762426411227e-06,
      "loss": 0.0653,
      "step": 1999
    },
    {
      "epoch": 0.4550625711035267,
      "grad_norm": 1.3459272018496387,
      "learning_rate": 1.2246510599926412e-06,
      "loss": 0.0587,
      "step": 2000
    },
    {
      "epoch": 0.4552901023890785,
      "grad_norm": 1.4831181130397366,
      "learning_rate": 1.2246258650883904e-06,
      "loss": 0.0632,
      "step": 2001
    },
    {
      "epoch": 0.4555176336746303,
      "grad_norm": 1.4710214394162384,
      "learning_rate": 1.224600657928885e-06,
      "loss": 0.0707,
      "step": 2002
    },
    {
      "epoch": 0.455745164960182,
      "grad_norm": 1.6318800586995355,
      "learning_rate": 1.2245754385146402e-06,
      "loss": 0.0737,
      "step": 2003
    },
    {
      "epoch": 0.4559726962457338,
      "grad_norm": 1.3377789919521947,
      "learning_rate": 1.2245502068461718e-06,
      "loss": 0.0549,
      "step": 2004
    },
    {
      "epoch": 0.45620022753128553,
      "grad_norm": 1.5464491108432568,
      "learning_rate": 1.224524962923995e-06,
      "loss": 0.0414,
      "step": 2005
    },
    {
      "epoch": 0.45642775881683734,
      "grad_norm": 1.7217302295238814,
      "learning_rate": 1.2244997067486258e-06,
      "loss": 0.1082,
      "step": 2006
    },
    {
      "epoch": 0.4566552901023891,
      "grad_norm": 1.3057544355327637,
      "learning_rate": 1.2244744383205807e-06,
      "loss": 0.0553,
      "step": 2007
    },
    {
      "epoch": 0.45688282138794084,
      "grad_norm": 1.4993635003547536,
      "learning_rate": 1.224449157640376e-06,
      "loss": 0.0694,
      "step": 2008
    },
    {
      "epoch": 0.4571103526734926,
      "grad_norm": 1.4893330797563609,
      "learning_rate": 1.2244238647085283e-06,
      "loss": 0.0715,
      "step": 2009
    },
    {
      "epoch": 0.45733788395904434,
      "grad_norm": 1.8914472286199435,
      "learning_rate": 1.2243985595255546e-06,
      "loss": 0.0577,
      "step": 2010
    },
    {
      "epoch": 0.45756541524459615,
      "grad_norm": 1.8242839299021114,
      "learning_rate": 1.224373242091972e-06,
      "loss": 0.0565,
      "step": 2011
    },
    {
      "epoch": 0.4577929465301479,
      "grad_norm": 1.0856942810712096,
      "learning_rate": 1.224347912408298e-06,
      "loss": 0.0357,
      "step": 2012
    },
    {
      "epoch": 0.45802047781569966,
      "grad_norm": 1.3194141628678389,
      "learning_rate": 1.2243225704750506e-06,
      "loss": 0.0449,
      "step": 2013
    },
    {
      "epoch": 0.4582480091012514,
      "grad_norm": 1.860571028270787,
      "learning_rate": 1.2242972162927472e-06,
      "loss": 0.0855,
      "step": 2014
    },
    {
      "epoch": 0.45847554038680316,
      "grad_norm": 1.0986966810344647,
      "learning_rate": 1.2242718498619063e-06,
      "loss": 0.048,
      "step": 2015
    },
    {
      "epoch": 0.45870307167235497,
      "grad_norm": 1.286949991423601,
      "learning_rate": 1.2242464711830463e-06,
      "loss": 0.0648,
      "step": 2016
    },
    {
      "epoch": 0.4589306029579067,
      "grad_norm": 1.7867407908291044,
      "learning_rate": 1.224221080256686e-06,
      "loss": 0.0656,
      "step": 2017
    },
    {
      "epoch": 0.45915813424345847,
      "grad_norm": 0.9964611889527885,
      "learning_rate": 1.2241956770833444e-06,
      "loss": 0.0287,
      "step": 2018
    },
    {
      "epoch": 0.4593856655290102,
      "grad_norm": 0.758262966491813,
      "learning_rate": 1.2241702616635403e-06,
      "loss": 0.0266,
      "step": 2019
    },
    {
      "epoch": 0.459613196814562,
      "grad_norm": 2.5974139992788707,
      "learning_rate": 1.2241448339977931e-06,
      "loss": 0.111,
      "step": 2020
    },
    {
      "epoch": 0.4598407281001138,
      "grad_norm": 0.771625496801363,
      "learning_rate": 1.2241193940866234e-06,
      "loss": 0.0291,
      "step": 2021
    },
    {
      "epoch": 0.46006825938566553,
      "grad_norm": 1.04981191475765,
      "learning_rate": 1.22409394193055e-06,
      "loss": 0.0445,
      "step": 2022
    },
    {
      "epoch": 0.4602957906712173,
      "grad_norm": 1.1765006507267777,
      "learning_rate": 1.224068477530094e-06,
      "loss": 0.0534,
      "step": 2023
    },
    {
      "epoch": 0.46052332195676904,
      "grad_norm": 1.7004930482631033,
      "learning_rate": 1.2240430008857751e-06,
      "loss": 0.1093,
      "step": 2024
    },
    {
      "epoch": 0.46075085324232085,
      "grad_norm": 0.9880326133501596,
      "learning_rate": 1.2240175119981146e-06,
      "loss": 0.0596,
      "step": 2025
    },
    {
      "epoch": 0.4609783845278726,
      "grad_norm": 0.9331212134306713,
      "learning_rate": 1.223992010867633e-06,
      "loss": 0.0364,
      "step": 2026
    },
    {
      "epoch": 0.46120591581342435,
      "grad_norm": 0.592705357456653,
      "learning_rate": 1.223966497494852e-06,
      "loss": 0.0267,
      "step": 2027
    },
    {
      "epoch": 0.4614334470989761,
      "grad_norm": 2.103103329890297,
      "learning_rate": 1.2239409718802927e-06,
      "loss": 0.0789,
      "step": 2028
    },
    {
      "epoch": 0.46166097838452785,
      "grad_norm": 1.2774828791183843,
      "learning_rate": 1.2239154340244766e-06,
      "loss": 0.0617,
      "step": 2029
    },
    {
      "epoch": 0.46188850967007966,
      "grad_norm": 1.9492863350229845,
      "learning_rate": 1.2238898839279262e-06,
      "loss": 0.0715,
      "step": 2030
    },
    {
      "epoch": 0.4621160409556314,
      "grad_norm": 1.7906002686883862,
      "learning_rate": 1.223864321591163e-06,
      "loss": 0.0795,
      "step": 2031
    },
    {
      "epoch": 0.46234357224118316,
      "grad_norm": 1.0120261476046106,
      "learning_rate": 1.2238387470147103e-06,
      "loss": 0.0397,
      "step": 2032
    },
    {
      "epoch": 0.4625711035267349,
      "grad_norm": 1.4202832673434427,
      "learning_rate": 1.22381316019909e-06,
      "loss": 0.0551,
      "step": 2033
    },
    {
      "epoch": 0.46279863481228667,
      "grad_norm": 0.9745272894465992,
      "learning_rate": 1.2237875611448254e-06,
      "loss": 0.0634,
      "step": 2034
    },
    {
      "epoch": 0.4630261660978385,
      "grad_norm": 0.7432221539823165,
      "learning_rate": 1.2237619498524397e-06,
      "loss": 0.0218,
      "step": 2035
    },
    {
      "epoch": 0.46325369738339023,
      "grad_norm": 1.9819427368647708,
      "learning_rate": 1.2237363263224564e-06,
      "loss": 0.1136,
      "step": 2036
    },
    {
      "epoch": 0.463481228668942,
      "grad_norm": 1.538163547331391,
      "learning_rate": 1.2237106905553991e-06,
      "loss": 0.0581,
      "step": 2037
    },
    {
      "epoch": 0.46370875995449373,
      "grad_norm": 1.1781797084882926,
      "learning_rate": 1.2236850425517918e-06,
      "loss": 0.0593,
      "step": 2038
    },
    {
      "epoch": 0.4639362912400455,
      "grad_norm": 1.7417190267667673,
      "learning_rate": 1.2236593823121586e-06,
      "loss": 0.092,
      "step": 2039
    },
    {
      "epoch": 0.4641638225255973,
      "grad_norm": 1.2090125054365826,
      "learning_rate": 1.223633709837024e-06,
      "loss": 0.0238,
      "step": 2040
    },
    {
      "epoch": 0.46439135381114904,
      "grad_norm": 0.8736586233913058,
      "learning_rate": 1.2236080251269129e-06,
      "loss": 0.0436,
      "step": 2041
    },
    {
      "epoch": 0.4646188850967008,
      "grad_norm": 1.4877382030386759,
      "learning_rate": 1.2235823281823498e-06,
      "loss": 0.0673,
      "step": 2042
    },
    {
      "epoch": 0.46484641638225255,
      "grad_norm": 1.8784786331322982,
      "learning_rate": 1.2235566190038602e-06,
      "loss": 0.0872,
      "step": 2043
    },
    {
      "epoch": 0.4650739476678043,
      "grad_norm": 0.9749064913400318,
      "learning_rate": 1.2235308975919697e-06,
      "loss": 0.0408,
      "step": 2044
    },
    {
      "epoch": 0.4653014789533561,
      "grad_norm": 1.6562176997205134,
      "learning_rate": 1.2235051639472037e-06,
      "loss": 0.0593,
      "step": 2045
    },
    {
      "epoch": 0.46552901023890786,
      "grad_norm": 7.178097985744176,
      "learning_rate": 1.2234794180700883e-06,
      "loss": 0.2677,
      "step": 2046
    },
    {
      "epoch": 0.4657565415244596,
      "grad_norm": 1.2664824384389577,
      "learning_rate": 1.2234536599611496e-06,
      "loss": 0.0578,
      "step": 2047
    },
    {
      "epoch": 0.46598407281001136,
      "grad_norm": 1.0979349025374328,
      "learning_rate": 1.2234278896209142e-06,
      "loss": 0.053,
      "step": 2048
    },
    {
      "epoch": 0.4662116040955631,
      "grad_norm": 0.6490727646887132,
      "learning_rate": 1.2234021070499086e-06,
      "loss": 0.0262,
      "step": 2049
    },
    {
      "epoch": 0.4664391353811149,
      "grad_norm": 0.9140641916499196,
      "learning_rate": 1.22337631224866e-06,
      "loss": 0.0325,
      "step": 2050
    },
    {
      "epoch": 0.4666666666666667,
      "grad_norm": 1.0654195689636294,
      "learning_rate": 1.2233505052176952e-06,
      "loss": 0.0539,
      "step": 2051
    },
    {
      "epoch": 0.4668941979522184,
      "grad_norm": 1.2246413980257993,
      "learning_rate": 1.223324685957542e-06,
      "loss": 0.0539,
      "step": 2052
    },
    {
      "epoch": 0.4671217292377702,
      "grad_norm": 0.9496001567387945,
      "learning_rate": 1.2232988544687282e-06,
      "loss": 0.0523,
      "step": 2053
    },
    {
      "epoch": 0.46734926052332193,
      "grad_norm": 1.1280442595857205,
      "learning_rate": 1.2232730107517813e-06,
      "loss": 0.0483,
      "step": 2054
    },
    {
      "epoch": 0.46757679180887374,
      "grad_norm": 0.939380252104929,
      "learning_rate": 1.22324715480723e-06,
      "loss": 0.032,
      "step": 2055
    },
    {
      "epoch": 0.4678043230944255,
      "grad_norm": 1.56220155639698,
      "learning_rate": 1.2232212866356022e-06,
      "loss": 0.0744,
      "step": 2056
    },
    {
      "epoch": 0.46803185437997724,
      "grad_norm": 0.8923779970732735,
      "learning_rate": 1.2231954062374272e-06,
      "loss": 0.0278,
      "step": 2057
    },
    {
      "epoch": 0.468259385665529,
      "grad_norm": 1.366515353592018,
      "learning_rate": 1.2231695136132333e-06,
      "loss": 0.0699,
      "step": 2058
    },
    {
      "epoch": 0.4684869169510808,
      "grad_norm": 2.0148188928409616,
      "learning_rate": 1.2231436087635504e-06,
      "loss": 0.0599,
      "step": 2059
    },
    {
      "epoch": 0.46871444823663255,
      "grad_norm": 0.8270174628565943,
      "learning_rate": 1.2231176916889072e-06,
      "loss": 0.027,
      "step": 2060
    },
    {
      "epoch": 0.4689419795221843,
      "grad_norm": 1.2532661876313687,
      "learning_rate": 1.223091762389834e-06,
      "loss": 0.0684,
      "step": 2061
    },
    {
      "epoch": 0.46916951080773606,
      "grad_norm": 0.961835609964009,
      "learning_rate": 1.2230658208668606e-06,
      "loss": 0.0419,
      "step": 2062
    },
    {
      "epoch": 0.4693970420932878,
      "grad_norm": 1.2485009940950207,
      "learning_rate": 1.223039867120517e-06,
      "loss": 0.06,
      "step": 2063
    },
    {
      "epoch": 0.4696245733788396,
      "grad_norm": 1.0550492918928736,
      "learning_rate": 1.2230139011513338e-06,
      "loss": 0.0325,
      "step": 2064
    },
    {
      "epoch": 0.46985210466439137,
      "grad_norm": 1.0002830612932734,
      "learning_rate": 1.2229879229598416e-06,
      "loss": 0.0309,
      "step": 2065
    },
    {
      "epoch": 0.4700796359499431,
      "grad_norm": 1.0559040750849908,
      "learning_rate": 1.2229619325465717e-06,
      "loss": 0.044,
      "step": 2066
    },
    {
      "epoch": 0.47030716723549487,
      "grad_norm": 1.8211747050798512,
      "learning_rate": 1.2229359299120548e-06,
      "loss": 0.0544,
      "step": 2067
    },
    {
      "epoch": 0.4705346985210466,
      "grad_norm": 1.1244511125230865,
      "learning_rate": 1.2229099150568224e-06,
      "loss": 0.0458,
      "step": 2068
    },
    {
      "epoch": 0.47076222980659843,
      "grad_norm": 1.2446047468564831,
      "learning_rate": 1.2228838879814065e-06,
      "loss": 0.0454,
      "step": 2069
    },
    {
      "epoch": 0.4709897610921502,
      "grad_norm": 2.7103883547474132,
      "learning_rate": 1.222857848686339e-06,
      "loss": 0.1164,
      "step": 2070
    },
    {
      "epoch": 0.47121729237770194,
      "grad_norm": 1.0755709452712496,
      "learning_rate": 1.2228317971721518e-06,
      "loss": 0.0497,
      "step": 2071
    },
    {
      "epoch": 0.4714448236632537,
      "grad_norm": 1.3822921197656612,
      "learning_rate": 1.2228057334393777e-06,
      "loss": 0.0606,
      "step": 2072
    },
    {
      "epoch": 0.47167235494880544,
      "grad_norm": 1.436834207352602,
      "learning_rate": 1.2227796574885493e-06,
      "loss": 0.0759,
      "step": 2073
    },
    {
      "epoch": 0.47189988623435725,
      "grad_norm": 0.8895801225433585,
      "learning_rate": 1.2227535693201994e-06,
      "loss": 0.0278,
      "step": 2074
    },
    {
      "epoch": 0.472127417519909,
      "grad_norm": 0.9871504029067106,
      "learning_rate": 1.2227274689348612e-06,
      "loss": 0.0339,
      "step": 2075
    },
    {
      "epoch": 0.47235494880546075,
      "grad_norm": 2.4304810090092217,
      "learning_rate": 1.2227013563330684e-06,
      "loss": 0.1278,
      "step": 2076
    },
    {
      "epoch": 0.4725824800910125,
      "grad_norm": 1.3611006420880847,
      "learning_rate": 1.2226752315153543e-06,
      "loss": 0.056,
      "step": 2077
    },
    {
      "epoch": 0.47281001137656425,
      "grad_norm": 1.2835019677874184,
      "learning_rate": 1.2226490944822533e-06,
      "loss": 0.0495,
      "step": 2078
    },
    {
      "epoch": 0.47303754266211606,
      "grad_norm": 0.9616764487213456,
      "learning_rate": 1.2226229452342991e-06,
      "loss": 0.0346,
      "step": 2079
    },
    {
      "epoch": 0.4732650739476678,
      "grad_norm": 1.0696378165797904,
      "learning_rate": 1.2225967837720265e-06,
      "loss": 0.0408,
      "step": 2080
    },
    {
      "epoch": 0.47349260523321957,
      "grad_norm": 1.40355131716461,
      "learning_rate": 1.2225706100959702e-06,
      "loss": 0.0567,
      "step": 2081
    },
    {
      "epoch": 0.4737201365187713,
      "grad_norm": 1.228435046083943,
      "learning_rate": 1.2225444242066648e-06,
      "loss": 0.0483,
      "step": 2082
    },
    {
      "epoch": 0.47394766780432307,
      "grad_norm": 2.414708407757396,
      "learning_rate": 1.2225182261046458e-06,
      "loss": 0.0817,
      "step": 2083
    },
    {
      "epoch": 0.4741751990898749,
      "grad_norm": 1.3036262799520595,
      "learning_rate": 1.2224920157904487e-06,
      "loss": 0.0552,
      "step": 2084
    },
    {
      "epoch": 0.47440273037542663,
      "grad_norm": 1.4961223747917856,
      "learning_rate": 1.222465793264609e-06,
      "loss": 0.0653,
      "step": 2085
    },
    {
      "epoch": 0.4746302616609784,
      "grad_norm": 0.9962870632246056,
      "learning_rate": 1.2224395585276626e-06,
      "loss": 0.059,
      "step": 2086
    },
    {
      "epoch": 0.47485779294653013,
      "grad_norm": 1.1085210761396838,
      "learning_rate": 1.2224133115801458e-06,
      "loss": 0.0415,
      "step": 2087
    },
    {
      "epoch": 0.4750853242320819,
      "grad_norm": 2.147238444020511,
      "learning_rate": 1.222387052422595e-06,
      "loss": 0.0832,
      "step": 2088
    },
    {
      "epoch": 0.4753128555176337,
      "grad_norm": 1.1789213043757707,
      "learning_rate": 1.222360781055547e-06,
      "loss": 0.0439,
      "step": 2089
    },
    {
      "epoch": 0.47554038680318544,
      "grad_norm": 1.0136823057916216,
      "learning_rate": 1.2223344974795386e-06,
      "loss": 0.0491,
      "step": 2090
    },
    {
      "epoch": 0.4757679180887372,
      "grad_norm": 1.0762034069369268,
      "learning_rate": 1.2223082016951071e-06,
      "loss": 0.0414,
      "step": 2091
    },
    {
      "epoch": 0.47599544937428895,
      "grad_norm": 0.7861883267498344,
      "learning_rate": 1.2222818937027898e-06,
      "loss": 0.0225,
      "step": 2092
    },
    {
      "epoch": 0.47622298065984076,
      "grad_norm": 1.4307219009209873,
      "learning_rate": 1.2222555735031245e-06,
      "loss": 0.0518,
      "step": 2093
    },
    {
      "epoch": 0.4764505119453925,
      "grad_norm": 1.7521480356699026,
      "learning_rate": 1.222229241096649e-06,
      "loss": 0.0686,
      "step": 2094
    },
    {
      "epoch": 0.47667804323094426,
      "grad_norm": 1.084257413613486,
      "learning_rate": 1.2222028964839017e-06,
      "loss": 0.0623,
      "step": 2095
    },
    {
      "epoch": 0.476905574516496,
      "grad_norm": 0.9853812502387866,
      "learning_rate": 1.2221765396654209e-06,
      "loss": 0.0322,
      "step": 2096
    },
    {
      "epoch": 0.47713310580204776,
      "grad_norm": 1.593782379839956,
      "learning_rate": 1.2221501706417454e-06,
      "loss": 0.0776,
      "step": 2097
    },
    {
      "epoch": 0.47736063708759957,
      "grad_norm": 1.2316203058040796,
      "learning_rate": 1.2221237894134138e-06,
      "loss": 0.0415,
      "step": 2098
    },
    {
      "epoch": 0.4775881683731513,
      "grad_norm": 1.7492243550591817,
      "learning_rate": 1.222097395980966e-06,
      "loss": 0.0863,
      "step": 2099
    },
    {
      "epoch": 0.4778156996587031,
      "grad_norm": 1.5123822677786471,
      "learning_rate": 1.2220709903449403e-06,
      "loss": 0.0839,
      "step": 2100
    },
    {
      "epoch": 0.4780432309442548,
      "grad_norm": 0.6925370489166974,
      "learning_rate": 1.2220445725058775e-06,
      "loss": 0.036,
      "step": 2101
    },
    {
      "epoch": 0.4782707622298066,
      "grad_norm": 1.416495751104088,
      "learning_rate": 1.222018142464317e-06,
      "loss": 0.0745,
      "step": 2102
    },
    {
      "epoch": 0.4784982935153584,
      "grad_norm": 1.3844263789079834,
      "learning_rate": 1.221991700220799e-06,
      "loss": 0.0635,
      "step": 2103
    },
    {
      "epoch": 0.47872582480091014,
      "grad_norm": 1.293007537973823,
      "learning_rate": 1.221965245775864e-06,
      "loss": 0.0489,
      "step": 2104
    },
    {
      "epoch": 0.4789533560864619,
      "grad_norm": 1.4567681046992622,
      "learning_rate": 1.2219387791300527e-06,
      "loss": 0.0645,
      "step": 2105
    },
    {
      "epoch": 0.47918088737201364,
      "grad_norm": 0.9331875777291676,
      "learning_rate": 1.2219123002839058e-06,
      "loss": 0.0374,
      "step": 2106
    },
    {
      "epoch": 0.4794084186575654,
      "grad_norm": 1.0880231881224058,
      "learning_rate": 1.2218858092379648e-06,
      "loss": 0.0495,
      "step": 2107
    },
    {
      "epoch": 0.4796359499431172,
      "grad_norm": 2.087839894098663,
      "learning_rate": 1.221859305992771e-06,
      "loss": 0.0775,
      "step": 2108
    },
    {
      "epoch": 0.47986348122866895,
      "grad_norm": 2.0258749667718408,
      "learning_rate": 1.2218327905488662e-06,
      "loss": 0.0928,
      "step": 2109
    },
    {
      "epoch": 0.4800910125142207,
      "grad_norm": 1.077278940685259,
      "learning_rate": 1.221806262906792e-06,
      "loss": 0.0395,
      "step": 2110
    },
    {
      "epoch": 0.48031854379977246,
      "grad_norm": 1.3800617021949446,
      "learning_rate": 1.221779723067091e-06,
      "loss": 0.0547,
      "step": 2111
    },
    {
      "epoch": 0.4805460750853242,
      "grad_norm": 1.1882918051107954,
      "learning_rate": 1.2217531710303053e-06,
      "loss": 0.0646,
      "step": 2112
    },
    {
      "epoch": 0.480773606370876,
      "grad_norm": 1.648037642948223,
      "learning_rate": 1.2217266067969778e-06,
      "loss": 0.0883,
      "step": 2113
    },
    {
      "epoch": 0.48100113765642777,
      "grad_norm": 1.1114503968376748,
      "learning_rate": 1.221700030367651e-06,
      "loss": 0.0532,
      "step": 2114
    },
    {
      "epoch": 0.4812286689419795,
      "grad_norm": 2.318260685332167,
      "learning_rate": 1.2216734417428686e-06,
      "loss": 0.1156,
      "step": 2115
    },
    {
      "epoch": 0.4814562002275313,
      "grad_norm": 0.943158220881948,
      "learning_rate": 1.2216468409231738e-06,
      "loss": 0.0478,
      "step": 2116
    },
    {
      "epoch": 0.481683731513083,
      "grad_norm": 1.3059340047526014,
      "learning_rate": 1.2216202279091104e-06,
      "loss": 0.0474,
      "step": 2117
    },
    {
      "epoch": 0.48191126279863483,
      "grad_norm": 1.5108825640696393,
      "learning_rate": 1.2215936027012221e-06,
      "loss": 0.0528,
      "step": 2118
    },
    {
      "epoch": 0.4821387940841866,
      "grad_norm": 1.6389721165810114,
      "learning_rate": 1.2215669653000532e-06,
      "loss": 0.0636,
      "step": 2119
    },
    {
      "epoch": 0.48236632536973834,
      "grad_norm": 0.8263598171620422,
      "learning_rate": 1.2215403157061478e-06,
      "loss": 0.0288,
      "step": 2120
    },
    {
      "epoch": 0.4825938566552901,
      "grad_norm": 1.9615999942786497,
      "learning_rate": 1.2215136539200512e-06,
      "loss": 0.0735,
      "step": 2121
    },
    {
      "epoch": 0.48282138794084184,
      "grad_norm": 1.052892038422689,
      "learning_rate": 1.2214869799423078e-06,
      "loss": 0.0375,
      "step": 2122
    },
    {
      "epoch": 0.48304891922639365,
      "grad_norm": 1.0184495668702673,
      "learning_rate": 1.2214602937734632e-06,
      "loss": 0.0457,
      "step": 2123
    },
    {
      "epoch": 0.4832764505119454,
      "grad_norm": 2.1042829626272868,
      "learning_rate": 1.2214335954140624e-06,
      "loss": 0.1008,
      "step": 2124
    },
    {
      "epoch": 0.48350398179749715,
      "grad_norm": 1.2825506743830564,
      "learning_rate": 1.221406884864651e-06,
      "loss": 0.0542,
      "step": 2125
    },
    {
      "epoch": 0.4837315130830489,
      "grad_norm": 1.2404980369925334,
      "learning_rate": 1.2213801621257754e-06,
      "loss": 0.076,
      "step": 2126
    },
    {
      "epoch": 0.48395904436860065,
      "grad_norm": 1.6535875011294203,
      "learning_rate": 1.2213534271979815e-06,
      "loss": 0.0826,
      "step": 2127
    },
    {
      "epoch": 0.48418657565415246,
      "grad_norm": 1.5742759100449686,
      "learning_rate": 1.2213266800818158e-06,
      "loss": 0.0722,
      "step": 2128
    },
    {
      "epoch": 0.4844141069397042,
      "grad_norm": 1.4590518459696438,
      "learning_rate": 1.2212999207778246e-06,
      "loss": 0.0626,
      "step": 2129
    },
    {
      "epoch": 0.48464163822525597,
      "grad_norm": 1.032592661119547,
      "learning_rate": 1.2212731492865552e-06,
      "loss": 0.0352,
      "step": 2130
    },
    {
      "epoch": 0.4848691695108077,
      "grad_norm": 1.943068406348077,
      "learning_rate": 1.2212463656085548e-06,
      "loss": 0.0878,
      "step": 2131
    },
    {
      "epoch": 0.4850967007963595,
      "grad_norm": 1.4222918495364802,
      "learning_rate": 1.2212195697443704e-06,
      "loss": 0.0662,
      "step": 2132
    },
    {
      "epoch": 0.4853242320819113,
      "grad_norm": 1.0835698936451348,
      "learning_rate": 1.2211927616945502e-06,
      "loss": 0.0315,
      "step": 2133
    },
    {
      "epoch": 0.48555176336746303,
      "grad_norm": 1.4804001036145082,
      "learning_rate": 1.2211659414596417e-06,
      "loss": 0.057,
      "step": 2134
    },
    {
      "epoch": 0.4857792946530148,
      "grad_norm": 1.0372510496516572,
      "learning_rate": 1.2211391090401931e-06,
      "loss": 0.0347,
      "step": 2135
    },
    {
      "epoch": 0.48600682593856653,
      "grad_norm": 1.4539414447995862,
      "learning_rate": 1.2211122644367531e-06,
      "loss": 0.0546,
      "step": 2136
    },
    {
      "epoch": 0.48623435722411834,
      "grad_norm": 0.8272464584595134,
      "learning_rate": 1.2210854076498699e-06,
      "loss": 0.0291,
      "step": 2137
    },
    {
      "epoch": 0.4864618885096701,
      "grad_norm": 1.082723119747633,
      "learning_rate": 1.2210585386800927e-06,
      "loss": 0.0631,
      "step": 2138
    },
    {
      "epoch": 0.48668941979522184,
      "grad_norm": 1.7613862158027351,
      "learning_rate": 1.2210316575279707e-06,
      "loss": 0.0516,
      "step": 2139
    },
    {
      "epoch": 0.4869169510807736,
      "grad_norm": 1.05018126679848,
      "learning_rate": 1.221004764194053e-06,
      "loss": 0.0503,
      "step": 2140
    },
    {
      "epoch": 0.48714448236632535,
      "grad_norm": 1.4765291215619578,
      "learning_rate": 1.22097785867889e-06,
      "loss": 0.0462,
      "step": 2141
    },
    {
      "epoch": 0.48737201365187716,
      "grad_norm": 1.6033668171289224,
      "learning_rate": 1.2209509409830304e-06,
      "loss": 0.0656,
      "step": 2142
    },
    {
      "epoch": 0.4875995449374289,
      "grad_norm": 1.2469591088986771,
      "learning_rate": 1.2209240111070254e-06,
      "loss": 0.0644,
      "step": 2143
    },
    {
      "epoch": 0.48782707622298066,
      "grad_norm": 1.676030146319255,
      "learning_rate": 1.2208970690514247e-06,
      "loss": 0.0502,
      "step": 2144
    },
    {
      "epoch": 0.4880546075085324,
      "grad_norm": 0.89826172703067,
      "learning_rate": 1.2208701148167795e-06,
      "loss": 0.0378,
      "step": 2145
    },
    {
      "epoch": 0.48828213879408416,
      "grad_norm": 1.2953659435102682,
      "learning_rate": 1.2208431484036405e-06,
      "loss": 0.0532,
      "step": 2146
    },
    {
      "epoch": 0.48850967007963597,
      "grad_norm": 1.6937135620824826,
      "learning_rate": 1.2208161698125584e-06,
      "loss": 0.0777,
      "step": 2147
    },
    {
      "epoch": 0.4887372013651877,
      "grad_norm": 1.4516409315976513,
      "learning_rate": 1.2207891790440852e-06,
      "loss": 0.0532,
      "step": 2148
    },
    {
      "epoch": 0.4889647326507395,
      "grad_norm": 1.466966804713568,
      "learning_rate": 1.2207621760987723e-06,
      "loss": 0.0542,
      "step": 2149
    },
    {
      "epoch": 0.4891922639362912,
      "grad_norm": 1.5403030963325346,
      "learning_rate": 1.2207351609771718e-06,
      "loss": 0.0699,
      "step": 2150
    },
    {
      "epoch": 0.489419795221843,
      "grad_norm": 1.1658099032169384,
      "learning_rate": 1.2207081336798352e-06,
      "loss": 0.0747,
      "step": 2151
    },
    {
      "epoch": 0.4896473265073948,
      "grad_norm": 1.272828829650441,
      "learning_rate": 1.2206810942073158e-06,
      "loss": 0.0339,
      "step": 2152
    },
    {
      "epoch": 0.48987485779294654,
      "grad_norm": 0.8638651055871553,
      "learning_rate": 1.2206540425601653e-06,
      "loss": 0.0258,
      "step": 2153
    },
    {
      "epoch": 0.4901023890784983,
      "grad_norm": 1.1406798846203596,
      "learning_rate": 1.2206269787389374e-06,
      "loss": 0.0434,
      "step": 2154
    },
    {
      "epoch": 0.49032992036405004,
      "grad_norm": 1.3546994084602113,
      "learning_rate": 1.220599902744185e-06,
      "loss": 0.0505,
      "step": 2155
    },
    {
      "epoch": 0.4905574516496018,
      "grad_norm": 0.8028716165840231,
      "learning_rate": 1.220572814576461e-06,
      "loss": 0.0252,
      "step": 2156
    },
    {
      "epoch": 0.4907849829351536,
      "grad_norm": 1.5525391069467074,
      "learning_rate": 1.2205457142363197e-06,
      "loss": 0.0671,
      "step": 2157
    },
    {
      "epoch": 0.49101251422070535,
      "grad_norm": 1.5565330239281656,
      "learning_rate": 1.2205186017243146e-06,
      "loss": 0.0485,
      "step": 2158
    },
    {
      "epoch": 0.4912400455062571,
      "grad_norm": 0.9990569085398061,
      "learning_rate": 1.2204914770409999e-06,
      "loss": 0.0379,
      "step": 2159
    },
    {
      "epoch": 0.49146757679180886,
      "grad_norm": 1.3885909890564816,
      "learning_rate": 1.22046434018693e-06,
      "loss": 0.0815,
      "step": 2160
    },
    {
      "epoch": 0.4916951080773606,
      "grad_norm": 1.0037322866680125,
      "learning_rate": 1.2204371911626597e-06,
      "loss": 0.0409,
      "step": 2161
    },
    {
      "epoch": 0.4919226393629124,
      "grad_norm": 2.005545489895357,
      "learning_rate": 1.2204100299687436e-06,
      "loss": 0.1019,
      "step": 2162
    },
    {
      "epoch": 0.49215017064846417,
      "grad_norm": 1.5901419467739697,
      "learning_rate": 1.2203828566057368e-06,
      "loss": 0.0497,
      "step": 2163
    },
    {
      "epoch": 0.4923777019340159,
      "grad_norm": 1.0024861467422042,
      "learning_rate": 1.2203556710741948e-06,
      "loss": 0.0537,
      "step": 2164
    },
    {
      "epoch": 0.4926052332195677,
      "grad_norm": 1.1764359755022478,
      "learning_rate": 1.2203284733746734e-06,
      "loss": 0.0609,
      "step": 2165
    },
    {
      "epoch": 0.4928327645051195,
      "grad_norm": 1.4780687059085984,
      "learning_rate": 1.2203012635077283e-06,
      "loss": 0.0467,
      "step": 2166
    },
    {
      "epoch": 0.49306029579067123,
      "grad_norm": 1.0636152584968832,
      "learning_rate": 1.2202740414739153e-06,
      "loss": 0.0334,
      "step": 2167
    },
    {
      "epoch": 0.493287827076223,
      "grad_norm": 0.9777542514618388,
      "learning_rate": 1.2202468072737914e-06,
      "loss": 0.0299,
      "step": 2168
    },
    {
      "epoch": 0.49351535836177474,
      "grad_norm": 1.099998345093073,
      "learning_rate": 1.2202195609079128e-06,
      "loss": 0.0349,
      "step": 2169
    },
    {
      "epoch": 0.4937428896473265,
      "grad_norm": 0.9946626000618751,
      "learning_rate": 1.2201923023768365e-06,
      "loss": 0.0439,
      "step": 2170
    },
    {
      "epoch": 0.4939704209328783,
      "grad_norm": 2.1311963123579707,
      "learning_rate": 1.2201650316811194e-06,
      "loss": 0.0871,
      "step": 2171
    },
    {
      "epoch": 0.49419795221843005,
      "grad_norm": 1.492842268786691,
      "learning_rate": 1.2201377488213195e-06,
      "loss": 0.0851,
      "step": 2172
    },
    {
      "epoch": 0.4944254835039818,
      "grad_norm": 2.5237867181473876,
      "learning_rate": 1.2201104537979934e-06,
      "loss": 0.0637,
      "step": 2173
    },
    {
      "epoch": 0.49465301478953355,
      "grad_norm": 2.556432817981331,
      "learning_rate": 1.2200831466117e-06,
      "loss": 0.1105,
      "step": 2174
    },
    {
      "epoch": 0.4948805460750853,
      "grad_norm": 0.8981926808894833,
      "learning_rate": 1.2200558272629967e-06,
      "loss": 0.03,
      "step": 2175
    },
    {
      "epoch": 0.4951080773606371,
      "grad_norm": 1.6324971068278198,
      "learning_rate": 1.2200284957524421e-06,
      "loss": 0.063,
      "step": 2176
    },
    {
      "epoch": 0.49533560864618886,
      "grad_norm": 1.0341597824022535,
      "learning_rate": 1.2200011520805947e-06,
      "loss": 0.0533,
      "step": 2177
    },
    {
      "epoch": 0.4955631399317406,
      "grad_norm": 0.8107094209291485,
      "learning_rate": 1.2199737962480136e-06,
      "loss": 0.0427,
      "step": 2178
    },
    {
      "epoch": 0.49579067121729237,
      "grad_norm": 1.106894286453914,
      "learning_rate": 1.2199464282552576e-06,
      "loss": 0.0594,
      "step": 2179
    },
    {
      "epoch": 0.4960182025028441,
      "grad_norm": 1.7291263581636043,
      "learning_rate": 1.2199190481028864e-06,
      "loss": 0.0675,
      "step": 2180
    },
    {
      "epoch": 0.4962457337883959,
      "grad_norm": 0.9412668127265857,
      "learning_rate": 1.2198916557914592e-06,
      "loss": 0.0364,
      "step": 2181
    },
    {
      "epoch": 0.4964732650739477,
      "grad_norm": 1.4114602905566138,
      "learning_rate": 1.2198642513215362e-06,
      "loss": 0.0652,
      "step": 2182
    },
    {
      "epoch": 0.49670079635949943,
      "grad_norm": 0.609976074205654,
      "learning_rate": 1.2198368346936774e-06,
      "loss": 0.03,
      "step": 2183
    },
    {
      "epoch": 0.4969283276450512,
      "grad_norm": 1.2570918433725233,
      "learning_rate": 1.219809405908443e-06,
      "loss": 0.0479,
      "step": 2184
    },
    {
      "epoch": 0.49715585893060293,
      "grad_norm": 1.0852743744371969,
      "learning_rate": 1.2197819649663934e-06,
      "loss": 0.0383,
      "step": 2185
    },
    {
      "epoch": 0.49738339021615474,
      "grad_norm": 0.8202048600920889,
      "learning_rate": 1.21975451186809e-06,
      "loss": 0.0242,
      "step": 2186
    },
    {
      "epoch": 0.4976109215017065,
      "grad_norm": 1.180630635637642,
      "learning_rate": 1.2197270466140936e-06,
      "loss": 0.0597,
      "step": 2187
    },
    {
      "epoch": 0.49783845278725825,
      "grad_norm": 0.5499273259791628,
      "learning_rate": 1.2196995692049656e-06,
      "loss": 0.0194,
      "step": 2188
    },
    {
      "epoch": 0.49806598407281,
      "grad_norm": 1.3699181548074078,
      "learning_rate": 1.2196720796412675e-06,
      "loss": 0.0729,
      "step": 2189
    },
    {
      "epoch": 0.49829351535836175,
      "grad_norm": 1.6813076102860163,
      "learning_rate": 1.219644577923561e-06,
      "loss": 0.0576,
      "step": 2190
    },
    {
      "epoch": 0.49852104664391356,
      "grad_norm": 1.3365991781516084,
      "learning_rate": 1.2196170640524087e-06,
      "loss": 0.0682,
      "step": 2191
    },
    {
      "epoch": 0.4987485779294653,
      "grad_norm": 1.5433585155887166,
      "learning_rate": 1.2195895380283725e-06,
      "loss": 0.0712,
      "step": 2192
    },
    {
      "epoch": 0.49897610921501706,
      "grad_norm": 0.7540413090381725,
      "learning_rate": 1.219561999852015e-06,
      "loss": 0.0358,
      "step": 2193
    },
    {
      "epoch": 0.4992036405005688,
      "grad_norm": 1.0443802296277052,
      "learning_rate": 1.2195344495238993e-06,
      "loss": 0.0443,
      "step": 2194
    },
    {
      "epoch": 0.49943117178612056,
      "grad_norm": 1.0620744241395403,
      "learning_rate": 1.2195068870445878e-06,
      "loss": 0.0317,
      "step": 2195
    },
    {
      "epoch": 0.49965870307167237,
      "grad_norm": 0.8699498301198716,
      "learning_rate": 1.2194793124146447e-06,
      "loss": 0.0291,
      "step": 2196
    },
    {
      "epoch": 0.4998862343572241,
      "grad_norm": 1.4610167793964797,
      "learning_rate": 1.2194517256346333e-06,
      "loss": 0.0701,
      "step": 2197
    },
    {
      "epoch": 0.5001137656427759,
      "grad_norm": 1.2434868855241086,
      "learning_rate": 1.2194241267051172e-06,
      "loss": 0.0454,
      "step": 2198
    },
    {
      "epoch": 0.5003412969283276,
      "grad_norm": 1.4551693858285264,
      "learning_rate": 1.2193965156266605e-06,
      "loss": 0.0547,
      "step": 2199
    },
    {
      "epoch": 0.5005688282138794,
      "grad_norm": 1.9356847413883649,
      "learning_rate": 1.2193688923998277e-06,
      "loss": 0.0644,
      "step": 2200
    },
    {
      "epoch": 0.5007963594994311,
      "grad_norm": 1.1503513652252215,
      "learning_rate": 1.219341257025183e-06,
      "loss": 0.0474,
      "step": 2201
    },
    {
      "epoch": 0.5010238907849829,
      "grad_norm": 1.5368491876593886,
      "learning_rate": 1.2193136095032918e-06,
      "loss": 0.0573,
      "step": 2202
    },
    {
      "epoch": 0.5012514220705347,
      "grad_norm": 1.7879629590621755,
      "learning_rate": 1.2192859498347189e-06,
      "loss": 0.0917,
      "step": 2203
    },
    {
      "epoch": 0.5014789533560865,
      "grad_norm": 1.5178773819159281,
      "learning_rate": 1.2192582780200293e-06,
      "loss": 0.0622,
      "step": 2204
    },
    {
      "epoch": 0.5017064846416383,
      "grad_norm": 1.3331971965539129,
      "learning_rate": 1.219230594059789e-06,
      "loss": 0.0649,
      "step": 2205
    },
    {
      "epoch": 0.50193401592719,
      "grad_norm": 2.165221254848427,
      "learning_rate": 1.2192028979545637e-06,
      "loss": 0.1342,
      "step": 2206
    },
    {
      "epoch": 0.5021615472127418,
      "grad_norm": 1.743824072315693,
      "learning_rate": 1.2191751897049192e-06,
      "loss": 0.0792,
      "step": 2207
    },
    {
      "epoch": 0.5023890784982935,
      "grad_norm": 1.3495321864370744,
      "learning_rate": 1.2191474693114223e-06,
      "loss": 0.0553,
      "step": 2208
    },
    {
      "epoch": 0.5026166097838453,
      "grad_norm": 1.2047719291625048,
      "learning_rate": 1.2191197367746389e-06,
      "loss": 0.0493,
      "step": 2209
    },
    {
      "epoch": 0.502844141069397,
      "grad_norm": 0.7568498406042676,
      "learning_rate": 1.2190919920951363e-06,
      "loss": 0.0425,
      "step": 2210
    },
    {
      "epoch": 0.5030716723549488,
      "grad_norm": 1.0905266622637186,
      "learning_rate": 1.2190642352734814e-06,
      "loss": 0.0392,
      "step": 2211
    },
    {
      "epoch": 0.5032992036405005,
      "grad_norm": 1.0772245840535826,
      "learning_rate": 1.2190364663102417e-06,
      "loss": 0.0602,
      "step": 2212
    },
    {
      "epoch": 0.5035267349260524,
      "grad_norm": 1.2112393942159567,
      "learning_rate": 1.2190086852059844e-06,
      "loss": 0.0561,
      "step": 2213
    },
    {
      "epoch": 0.5037542662116041,
      "grad_norm": 1.164381821277193,
      "learning_rate": 1.2189808919612773e-06,
      "loss": 0.0567,
      "step": 2214
    },
    {
      "epoch": 0.5039817974971559,
      "grad_norm": 1.980841011359919,
      "learning_rate": 1.2189530865766888e-06,
      "loss": 0.1002,
      "step": 2215
    },
    {
      "epoch": 0.5042093287827076,
      "grad_norm": 1.0195390074518678,
      "learning_rate": 1.218925269052787e-06,
      "loss": 0.0378,
      "step": 2216
    },
    {
      "epoch": 0.5044368600682594,
      "grad_norm": 1.58039989452312,
      "learning_rate": 1.2188974393901404e-06,
      "loss": 0.079,
      "step": 2217
    },
    {
      "epoch": 0.5046643913538111,
      "grad_norm": 1.7632794902075288,
      "learning_rate": 1.2188695975893177e-06,
      "loss": 0.0704,
      "step": 2218
    },
    {
      "epoch": 0.5048919226393629,
      "grad_norm": 1.8641645104616156,
      "learning_rate": 1.218841743650888e-06,
      "loss": 0.0767,
      "step": 2219
    },
    {
      "epoch": 0.5051194539249146,
      "grad_norm": 0.7736340615610193,
      "learning_rate": 1.2188138775754207e-06,
      "loss": 0.0255,
      "step": 2220
    },
    {
      "epoch": 0.5053469852104664,
      "grad_norm": 1.706014884062171,
      "learning_rate": 1.2187859993634854e-06,
      "loss": 0.0842,
      "step": 2221
    },
    {
      "epoch": 0.5055745164960183,
      "grad_norm": 1.0275306984300476,
      "learning_rate": 1.2187581090156518e-06,
      "loss": 0.0372,
      "step": 2222
    },
    {
      "epoch": 0.50580204778157,
      "grad_norm": 1.560530367319729,
      "learning_rate": 1.2187302065324896e-06,
      "loss": 0.0957,
      "step": 2223
    },
    {
      "epoch": 0.5060295790671218,
      "grad_norm": 1.1142447765551218,
      "learning_rate": 1.2187022919145695e-06,
      "loss": 0.0372,
      "step": 2224
    },
    {
      "epoch": 0.5062571103526735,
      "grad_norm": 2.0014178600777157,
      "learning_rate": 1.2186743651624617e-06,
      "loss": 0.1222,
      "step": 2225
    },
    {
      "epoch": 0.5064846416382253,
      "grad_norm": 0.6371915222683989,
      "learning_rate": 1.2186464262767372e-06,
      "loss": 0.0303,
      "step": 2226
    },
    {
      "epoch": 0.506712172923777,
      "grad_norm": 1.1677216592762003,
      "learning_rate": 1.2186184752579671e-06,
      "loss": 0.0375,
      "step": 2227
    },
    {
      "epoch": 0.5069397042093288,
      "grad_norm": 1.0363110917575087,
      "learning_rate": 1.2185905121067223e-06,
      "loss": 0.0465,
      "step": 2228
    },
    {
      "epoch": 0.5071672354948805,
      "grad_norm": 0.7880205913361013,
      "learning_rate": 1.218562536823575e-06,
      "loss": 0.0461,
      "step": 2229
    },
    {
      "epoch": 0.5073947667804323,
      "grad_norm": 1.5494942731094494,
      "learning_rate": 1.2185345494090959e-06,
      "loss": 0.0633,
      "step": 2230
    },
    {
      "epoch": 0.507622298065984,
      "grad_norm": 1.24395335626353,
      "learning_rate": 1.218506549863858e-06,
      "loss": 0.037,
      "step": 2231
    },
    {
      "epoch": 0.5078498293515359,
      "grad_norm": 0.8497919211723142,
      "learning_rate": 1.2184785381884332e-06,
      "loss": 0.0281,
      "step": 2232
    },
    {
      "epoch": 0.5080773606370876,
      "grad_norm": 1.6659795749856976,
      "learning_rate": 1.218450514383394e-06,
      "loss": 0.0684,
      "step": 2233
    },
    {
      "epoch": 0.5083048919226394,
      "grad_norm": 1.2993624404216275,
      "learning_rate": 1.2184224784493127e-06,
      "loss": 0.0737,
      "step": 2234
    },
    {
      "epoch": 0.5085324232081911,
      "grad_norm": 2.2363218045516287,
      "learning_rate": 1.218394430386763e-06,
      "loss": 0.1023,
      "step": 2235
    },
    {
      "epoch": 0.5087599544937429,
      "grad_norm": 1.9367152712705378,
      "learning_rate": 1.2183663701963181e-06,
      "loss": 0.0801,
      "step": 2236
    },
    {
      "epoch": 0.5089874857792946,
      "grad_norm": 1.683178965938102,
      "learning_rate": 1.218338297878551e-06,
      "loss": 0.0511,
      "step": 2237
    },
    {
      "epoch": 0.5092150170648464,
      "grad_norm": 1.2570305483870696,
      "learning_rate": 1.2183102134340361e-06,
      "loss": 0.0531,
      "step": 2238
    },
    {
      "epoch": 0.5094425483503981,
      "grad_norm": 1.3387251635869548,
      "learning_rate": 1.2182821168633466e-06,
      "loss": 0.0488,
      "step": 2239
    },
    {
      "epoch": 0.5096700796359499,
      "grad_norm": 0.9109698538663267,
      "learning_rate": 1.2182540081670574e-06,
      "loss": 0.0297,
      "step": 2240
    },
    {
      "epoch": 0.5098976109215017,
      "grad_norm": 2.0458637944350917,
      "learning_rate": 1.2182258873457428e-06,
      "loss": 0.0932,
      "step": 2241
    },
    {
      "epoch": 0.5101251422070535,
      "grad_norm": 1.546517047083001,
      "learning_rate": 1.2181977543999776e-06,
      "loss": 0.0831,
      "step": 2242
    },
    {
      "epoch": 0.5103526734926053,
      "grad_norm": 0.5799180817857912,
      "learning_rate": 1.2181696093303363e-06,
      "loss": 0.0246,
      "step": 2243
    },
    {
      "epoch": 0.510580204778157,
      "grad_norm": 1.5842605922722366,
      "learning_rate": 1.218141452137395e-06,
      "loss": 0.1109,
      "step": 2244
    },
    {
      "epoch": 0.5108077360637088,
      "grad_norm": 1.1518936456507234,
      "learning_rate": 1.2181132828217284e-06,
      "loss": 0.0392,
      "step": 2245
    },
    {
      "epoch": 0.5110352673492605,
      "grad_norm": 1.5601510600984896,
      "learning_rate": 1.2180851013839127e-06,
      "loss": 0.0516,
      "step": 2246
    },
    {
      "epoch": 0.5112627986348123,
      "grad_norm": 1.1933267652037376,
      "learning_rate": 1.2180569078245236e-06,
      "loss": 0.0768,
      "step": 2247
    },
    {
      "epoch": 0.511490329920364,
      "grad_norm": 1.0276357760825077,
      "learning_rate": 1.2180287021441372e-06,
      "loss": 0.0392,
      "step": 2248
    },
    {
      "epoch": 0.5117178612059158,
      "grad_norm": 1.4568173829581255,
      "learning_rate": 1.2180004843433305e-06,
      "loss": 0.0711,
      "step": 2249
    },
    {
      "epoch": 0.5119453924914675,
      "grad_norm": 1.6296361312434235,
      "learning_rate": 1.21797225442268e-06,
      "loss": 0.061,
      "step": 2250
    },
    {
      "epoch": 0.5121729237770194,
      "grad_norm": 1.4099321985545326,
      "learning_rate": 1.2179440123827625e-06,
      "loss": 0.0562,
      "step": 2251
    },
    {
      "epoch": 0.5124004550625711,
      "grad_norm": 1.0111117892815282,
      "learning_rate": 1.2179157582241554e-06,
      "loss": 0.0425,
      "step": 2252
    },
    {
      "epoch": 0.5126279863481229,
      "grad_norm": 1.100003009354464,
      "learning_rate": 1.2178874919474359e-06,
      "loss": 0.059,
      "step": 2253
    },
    {
      "epoch": 0.5128555176336747,
      "grad_norm": 1.325363901130602,
      "learning_rate": 1.217859213553182e-06,
      "loss": 0.0711,
      "step": 2254
    },
    {
      "epoch": 0.5130830489192264,
      "grad_norm": 0.846300456406513,
      "learning_rate": 1.2178309230419714e-06,
      "loss": 0.0271,
      "step": 2255
    },
    {
      "epoch": 0.5133105802047782,
      "grad_norm": 1.2410062157733406,
      "learning_rate": 1.2178026204143827e-06,
      "loss": 0.0556,
      "step": 2256
    },
    {
      "epoch": 0.5135381114903299,
      "grad_norm": 1.6420974990981174,
      "learning_rate": 1.217774305670994e-06,
      "loss": 0.0712,
      "step": 2257
    },
    {
      "epoch": 0.5137656427758817,
      "grad_norm": 1.2130253789178054,
      "learning_rate": 1.2177459788123841e-06,
      "loss": 0.045,
      "step": 2258
    },
    {
      "epoch": 0.5139931740614334,
      "grad_norm": 1.5302885687128625,
      "learning_rate": 1.217717639839132e-06,
      "loss": 0.0788,
      "step": 2259
    },
    {
      "epoch": 0.5142207053469852,
      "grad_norm": 0.9160947105126066,
      "learning_rate": 1.2176892887518166e-06,
      "loss": 0.0381,
      "step": 2260
    },
    {
      "epoch": 0.514448236632537,
      "grad_norm": 0.9460716271280594,
      "learning_rate": 1.2176609255510176e-06,
      "loss": 0.0453,
      "step": 2261
    },
    {
      "epoch": 0.5146757679180888,
      "grad_norm": 0.6452347430540285,
      "learning_rate": 1.217632550237315e-06,
      "loss": 0.0225,
      "step": 2262
    },
    {
      "epoch": 0.5149032992036405,
      "grad_norm": 1.1495647599088414,
      "learning_rate": 1.2176041628112884e-06,
      "loss": 0.0506,
      "step": 2263
    },
    {
      "epoch": 0.5151308304891923,
      "grad_norm": 1.4009996525001283,
      "learning_rate": 1.217575763273518e-06,
      "loss": 0.0667,
      "step": 2264
    },
    {
      "epoch": 0.515358361774744,
      "grad_norm": 0.9674758245027556,
      "learning_rate": 1.217547351624584e-06,
      "loss": 0.0383,
      "step": 2265
    },
    {
      "epoch": 0.5155858930602958,
      "grad_norm": 1.6049309598180266,
      "learning_rate": 1.2175189278650677e-06,
      "loss": 0.0685,
      "step": 2266
    },
    {
      "epoch": 0.5158134243458475,
      "grad_norm": 1.047951830927466,
      "learning_rate": 1.2174904919955493e-06,
      "loss": 0.031,
      "step": 2267
    },
    {
      "epoch": 0.5160409556313993,
      "grad_norm": 1.0060223411128129,
      "learning_rate": 1.2174620440166105e-06,
      "loss": 0.0442,
      "step": 2268
    },
    {
      "epoch": 0.516268486916951,
      "grad_norm": 1.2945214325099708,
      "learning_rate": 1.2174335839288326e-06,
      "loss": 0.0558,
      "step": 2269
    },
    {
      "epoch": 0.5164960182025028,
      "grad_norm": 1.2199251228288925,
      "learning_rate": 1.2174051117327972e-06,
      "loss": 0.0398,
      "step": 2270
    },
    {
      "epoch": 0.5167235494880547,
      "grad_norm": 1.8177803789021154,
      "learning_rate": 1.2173766274290864e-06,
      "loss": 0.0618,
      "step": 2271
    },
    {
      "epoch": 0.5169510807736064,
      "grad_norm": 2.1534503937526557,
      "learning_rate": 1.217348131018282e-06,
      "loss": 0.0946,
      "step": 2272
    },
    {
      "epoch": 0.5171786120591582,
      "grad_norm": 1.335439582615335,
      "learning_rate": 1.2173196225009666e-06,
      "loss": 0.0467,
      "step": 2273
    },
    {
      "epoch": 0.5174061433447099,
      "grad_norm": 3.39634994855923,
      "learning_rate": 1.217291101877723e-06,
      "loss": 0.1597,
      "step": 2274
    },
    {
      "epoch": 0.5176336746302617,
      "grad_norm": 13.151087600523141,
      "learning_rate": 1.2172625691491342e-06,
      "loss": 0.061,
      "step": 2275
    },
    {
      "epoch": 0.5178612059158134,
      "grad_norm": 0.9388770540310672,
      "learning_rate": 1.217234024315783e-06,
      "loss": 0.0469,
      "step": 2276
    },
    {
      "epoch": 0.5180887372013652,
      "grad_norm": 1.237623086215354,
      "learning_rate": 1.2172054673782527e-06,
      "loss": 0.0494,
      "step": 2277
    },
    {
      "epoch": 0.5183162684869169,
      "grad_norm": 1.3918576506801177,
      "learning_rate": 1.2171768983371276e-06,
      "loss": 0.0688,
      "step": 2278
    },
    {
      "epoch": 0.5185437997724687,
      "grad_norm": 1.2775225468090683,
      "learning_rate": 1.217148317192991e-06,
      "loss": 0.034,
      "step": 2279
    },
    {
      "epoch": 0.5187713310580204,
      "grad_norm": 1.2629250238471204,
      "learning_rate": 1.2171197239464274e-06,
      "loss": 0.0477,
      "step": 2280
    },
    {
      "epoch": 0.5189988623435723,
      "grad_norm": 0.9222994348691986,
      "learning_rate": 1.217091118598021e-06,
      "loss": 0.0518,
      "step": 2281
    },
    {
      "epoch": 0.519226393629124,
      "grad_norm": 1.1733465242259884,
      "learning_rate": 1.2170625011483565e-06,
      "loss": 0.0362,
      "step": 2282
    },
    {
      "epoch": 0.5194539249146758,
      "grad_norm": 1.0540409111201516,
      "learning_rate": 1.2170338715980187e-06,
      "loss": 0.0333,
      "step": 2283
    },
    {
      "epoch": 0.5196814562002275,
      "grad_norm": 1.0390405907126188,
      "learning_rate": 1.217005229947593e-06,
      "loss": 0.0307,
      "step": 2284
    },
    {
      "epoch": 0.5199089874857793,
      "grad_norm": 0.7575429344893744,
      "learning_rate": 1.2169765761976646e-06,
      "loss": 0.0416,
      "step": 2285
    },
    {
      "epoch": 0.520136518771331,
      "grad_norm": 1.3326371343495533,
      "learning_rate": 1.216947910348819e-06,
      "loss": 0.054,
      "step": 2286
    },
    {
      "epoch": 0.5203640500568828,
      "grad_norm": 1.386966435546428,
      "learning_rate": 1.2169192324016423e-06,
      "loss": 0.0552,
      "step": 2287
    },
    {
      "epoch": 0.5205915813424346,
      "grad_norm": 1.957649195968788,
      "learning_rate": 1.2168905423567205e-06,
      "loss": 0.0941,
      "step": 2288
    },
    {
      "epoch": 0.5208191126279863,
      "grad_norm": 2.448802297286155,
      "learning_rate": 1.21686184021464e-06,
      "loss": 0.1116,
      "step": 2289
    },
    {
      "epoch": 0.5210466439135382,
      "grad_norm": 1.1844985987836856,
      "learning_rate": 1.2168331259759875e-06,
      "loss": 0.0449,
      "step": 2290
    },
    {
      "epoch": 0.5212741751990899,
      "grad_norm": 1.8104263725287626,
      "learning_rate": 1.2168043996413497e-06,
      "loss": 0.0575,
      "step": 2291
    },
    {
      "epoch": 0.5215017064846417,
      "grad_norm": 1.1147839216238626,
      "learning_rate": 1.216775661211314e-06,
      "loss": 0.05,
      "step": 2292
    },
    {
      "epoch": 0.5217292377701934,
      "grad_norm": 1.543703515430326,
      "learning_rate": 1.2167469106864673e-06,
      "loss": 0.0813,
      "step": 2293
    },
    {
      "epoch": 0.5219567690557452,
      "grad_norm": 1.5787563749864177,
      "learning_rate": 1.2167181480673977e-06,
      "loss": 0.0843,
      "step": 2294
    },
    {
      "epoch": 0.5221843003412969,
      "grad_norm": 1.3389543319065593,
      "learning_rate": 1.2166893733546927e-06,
      "loss": 0.0854,
      "step": 2295
    },
    {
      "epoch": 0.5224118316268487,
      "grad_norm": 1.79843376105108,
      "learning_rate": 1.2166605865489406e-06,
      "loss": 0.111,
      "step": 2296
    },
    {
      "epoch": 0.5226393629124004,
      "grad_norm": 1.2843560350462406,
      "learning_rate": 1.2166317876507296e-06,
      "loss": 0.0617,
      "step": 2297
    },
    {
      "epoch": 0.5228668941979522,
      "grad_norm": 1.2028371451089743,
      "learning_rate": 1.2166029766606486e-06,
      "loss": 0.0721,
      "step": 2298
    },
    {
      "epoch": 0.5230944254835039,
      "grad_norm": 2.882673882589606,
      "learning_rate": 1.2165741535792861e-06,
      "loss": 0.1223,
      "step": 2299
    },
    {
      "epoch": 0.5233219567690558,
      "grad_norm": 1.3155493992738794,
      "learning_rate": 1.2165453184072312e-06,
      "loss": 0.0591,
      "step": 2300
    },
    {
      "epoch": 0.5235494880546075,
      "grad_norm": 1.6629360145623788,
      "learning_rate": 1.2165164711450735e-06,
      "loss": 0.0775,
      "step": 2301
    },
    {
      "epoch": 0.5237770193401593,
      "grad_norm": 1.3013507571586798,
      "learning_rate": 1.2164876117934024e-06,
      "loss": 0.0638,
      "step": 2302
    },
    {
      "epoch": 0.524004550625711,
      "grad_norm": 2.648828931200833,
      "learning_rate": 1.2164587403528078e-06,
      "loss": 0.0864,
      "step": 2303
    },
    {
      "epoch": 0.5242320819112628,
      "grad_norm": 1.4929308932558032,
      "learning_rate": 1.2164298568238797e-06,
      "loss": 0.0679,
      "step": 2304
    },
    {
      "epoch": 0.5244596131968146,
      "grad_norm": 0.8718556925311854,
      "learning_rate": 1.2164009612072085e-06,
      "loss": 0.0359,
      "step": 2305
    },
    {
      "epoch": 0.5246871444823663,
      "grad_norm": 0.9783095053030553,
      "learning_rate": 1.2163720535033845e-06,
      "loss": 0.0442,
      "step": 2306
    },
    {
      "epoch": 0.5249146757679181,
      "grad_norm": 1.5130471240864227,
      "learning_rate": 1.2163431337129993e-06,
      "loss": 0.0512,
      "step": 2307
    },
    {
      "epoch": 0.5251422070534698,
      "grad_norm": 1.5432509321722079,
      "learning_rate": 1.216314201836643e-06,
      "loss": 0.0704,
      "step": 2308
    },
    {
      "epoch": 0.5253697383390216,
      "grad_norm": 1.069477672250475,
      "learning_rate": 1.2162852578749076e-06,
      "loss": 0.0445,
      "step": 2309
    },
    {
      "epoch": 0.5255972696245734,
      "grad_norm": 1.831811758429303,
      "learning_rate": 1.2162563018283843e-06,
      "loss": 0.086,
      "step": 2310
    },
    {
      "epoch": 0.5258248009101252,
      "grad_norm": 0.9600707211513121,
      "learning_rate": 1.216227333697665e-06,
      "loss": 0.0387,
      "step": 2311
    },
    {
      "epoch": 0.5260523321956769,
      "grad_norm": 1.0398262718871747,
      "learning_rate": 1.216198353483342e-06,
      "loss": 0.0336,
      "step": 2312
    },
    {
      "epoch": 0.5262798634812287,
      "grad_norm": 0.7659268750272631,
      "learning_rate": 1.2161693611860072e-06,
      "loss": 0.0292,
      "step": 2313
    },
    {
      "epoch": 0.5265073947667804,
      "grad_norm": 1.003194143526719,
      "learning_rate": 1.2161403568062533e-06,
      "loss": 0.0446,
      "step": 2314
    },
    {
      "epoch": 0.5267349260523322,
      "grad_norm": 0.8822650643996454,
      "learning_rate": 1.2161113403446733e-06,
      "loss": 0.036,
      "step": 2315
    },
    {
      "epoch": 0.5269624573378839,
      "grad_norm": 1.2445369505027863,
      "learning_rate": 1.21608231180186e-06,
      "loss": 0.056,
      "step": 2316
    },
    {
      "epoch": 0.5271899886234357,
      "grad_norm": 1.9110079675084388,
      "learning_rate": 1.2160532711784066e-06,
      "loss": 0.085,
      "step": 2317
    },
    {
      "epoch": 0.5274175199089874,
      "grad_norm": 1.3201545262151622,
      "learning_rate": 1.2160242184749069e-06,
      "loss": 0.0596,
      "step": 2318
    },
    {
      "epoch": 0.5276450511945392,
      "grad_norm": 1.3929491912851009,
      "learning_rate": 1.2159951536919547e-06,
      "loss": 0.0575,
      "step": 2319
    },
    {
      "epoch": 0.5278725824800911,
      "grad_norm": 1.4551403932656901,
      "learning_rate": 1.2159660768301438e-06,
      "loss": 0.0492,
      "step": 2320
    },
    {
      "epoch": 0.5281001137656428,
      "grad_norm": 1.4130548917935428,
      "learning_rate": 1.2159369878900687e-06,
      "loss": 0.047,
      "step": 2321
    },
    {
      "epoch": 0.5283276450511946,
      "grad_norm": 1.4452980530222412,
      "learning_rate": 1.2159078868723238e-06,
      "loss": 0.0525,
      "step": 2322
    },
    {
      "epoch": 0.5285551763367463,
      "grad_norm": 1.6104531531662376,
      "learning_rate": 1.2158787737775037e-06,
      "loss": 0.0755,
      "step": 2323
    },
    {
      "epoch": 0.5287827076222981,
      "grad_norm": 0.8559219056147712,
      "learning_rate": 1.2158496486062039e-06,
      "loss": 0.0415,
      "step": 2324
    },
    {
      "epoch": 0.5290102389078498,
      "grad_norm": 1.789699350536378,
      "learning_rate": 1.215820511359019e-06,
      "loss": 0.081,
      "step": 2325
    },
    {
      "epoch": 0.5292377701934016,
      "grad_norm": 1.3025849756724603,
      "learning_rate": 1.215791362036545e-06,
      "loss": 0.0485,
      "step": 2326
    },
    {
      "epoch": 0.5294653014789533,
      "grad_norm": 1.8682155789508528,
      "learning_rate": 1.2157622006393777e-06,
      "loss": 0.0634,
      "step": 2327
    },
    {
      "epoch": 0.5296928327645051,
      "grad_norm": 1.176130717608952,
      "learning_rate": 1.2157330271681129e-06,
      "loss": 0.039,
      "step": 2328
    },
    {
      "epoch": 0.5299203640500569,
      "grad_norm": 1.5070209136147386,
      "learning_rate": 1.215703841623347e-06,
      "loss": 0.0515,
      "step": 2329
    },
    {
      "epoch": 0.5301478953356087,
      "grad_norm": 1.2847130041310237,
      "learning_rate": 1.2156746440056762e-06,
      "loss": 0.0515,
      "step": 2330
    },
    {
      "epoch": 0.5303754266211604,
      "grad_norm": 2.141191883255034,
      "learning_rate": 1.2156454343156976e-06,
      "loss": 0.0828,
      "step": 2331
    },
    {
      "epoch": 0.5306029579067122,
      "grad_norm": 0.9805091402589389,
      "learning_rate": 1.2156162125540078e-06,
      "loss": 0.0315,
      "step": 2332
    },
    {
      "epoch": 0.5308304891922639,
      "grad_norm": 0.7809668560849007,
      "learning_rate": 1.2155869787212046e-06,
      "loss": 0.0365,
      "step": 2333
    },
    {
      "epoch": 0.5310580204778157,
      "grad_norm": 1.0616410014514985,
      "learning_rate": 1.215557732817885e-06,
      "loss": 0.0754,
      "step": 2334
    },
    {
      "epoch": 0.5312855517633674,
      "grad_norm": 1.74228651169713,
      "learning_rate": 1.2155284748446469e-06,
      "loss": 0.0628,
      "step": 2335
    },
    {
      "epoch": 0.5315130830489192,
      "grad_norm": 1.8258581718716098,
      "learning_rate": 1.2154992048020882e-06,
      "loss": 0.0957,
      "step": 2336
    },
    {
      "epoch": 0.531740614334471,
      "grad_norm": 1.5209500494569268,
      "learning_rate": 1.2154699226908072e-06,
      "loss": 0.0719,
      "step": 2337
    },
    {
      "epoch": 0.5319681456200227,
      "grad_norm": 0.9184689674787639,
      "learning_rate": 1.2154406285114025e-06,
      "loss": 0.0512,
      "step": 2338
    },
    {
      "epoch": 0.5321956769055746,
      "grad_norm": 0.92211428137911,
      "learning_rate": 1.2154113222644727e-06,
      "loss": 0.048,
      "step": 2339
    },
    {
      "epoch": 0.5324232081911263,
      "grad_norm": 1.2486653702619341,
      "learning_rate": 1.2153820039506167e-06,
      "loss": 0.036,
      "step": 2340
    },
    {
      "epoch": 0.5326507394766781,
      "grad_norm": 1.4084664381645906,
      "learning_rate": 1.2153526735704337e-06,
      "loss": 0.0641,
      "step": 2341
    },
    {
      "epoch": 0.5328782707622298,
      "grad_norm": 1.006089995573004,
      "learning_rate": 1.2153233311245234e-06,
      "loss": 0.0471,
      "step": 2342
    },
    {
      "epoch": 0.5331058020477816,
      "grad_norm": 1.2533267107040431,
      "learning_rate": 1.2152939766134852e-06,
      "loss": 0.044,
      "step": 2343
    },
    {
      "epoch": 0.5333333333333333,
      "grad_norm": 1.7169848907323204,
      "learning_rate": 1.2152646100379193e-06,
      "loss": 0.0979,
      "step": 2344
    },
    {
      "epoch": 0.5335608646188851,
      "grad_norm": 1.1457088139957912,
      "learning_rate": 1.2152352313984257e-06,
      "loss": 0.0539,
      "step": 2345
    },
    {
      "epoch": 0.5337883959044368,
      "grad_norm": 0.9527962003003786,
      "learning_rate": 1.2152058406956049e-06,
      "loss": 0.0389,
      "step": 2346
    },
    {
      "epoch": 0.5340159271899886,
      "grad_norm": 1.5909759190824926,
      "learning_rate": 1.2151764379300578e-06,
      "loss": 0.058,
      "step": 2347
    },
    {
      "epoch": 0.5342434584755403,
      "grad_norm": 1.065418317407388,
      "learning_rate": 1.2151470231023851e-06,
      "loss": 0.0304,
      "step": 2348
    },
    {
      "epoch": 0.5344709897610922,
      "grad_norm": 1.007958090684921,
      "learning_rate": 1.2151175962131881e-06,
      "loss": 0.049,
      "step": 2349
    },
    {
      "epoch": 0.534698521046644,
      "grad_norm": 1.7395647605444997,
      "learning_rate": 1.215088157263068e-06,
      "loss": 0.06,
      "step": 2350
    },
    {
      "epoch": 0.5349260523321957,
      "grad_norm": 1.0555560500880805,
      "learning_rate": 1.2150587062526267e-06,
      "loss": 0.0466,
      "step": 2351
    },
    {
      "epoch": 0.5351535836177475,
      "grad_norm": 1.4330290353052704,
      "learning_rate": 1.2150292431824663e-06,
      "loss": 0.1029,
      "step": 2352
    },
    {
      "epoch": 0.5353811149032992,
      "grad_norm": 1.0120120325632629,
      "learning_rate": 1.2149997680531886e-06,
      "loss": 0.0372,
      "step": 2353
    },
    {
      "epoch": 0.535608646188851,
      "grad_norm": 1.6445264766335106,
      "learning_rate": 1.214970280865396e-06,
      "loss": 0.0601,
      "step": 2354
    },
    {
      "epoch": 0.5358361774744027,
      "grad_norm": 1.479298949629959,
      "learning_rate": 1.2149407816196917e-06,
      "loss": 0.0713,
      "step": 2355
    },
    {
      "epoch": 0.5360637087599545,
      "grad_norm": 1.0674415697842976,
      "learning_rate": 1.214911270316678e-06,
      "loss": 0.0529,
      "step": 2356
    },
    {
      "epoch": 0.5362912400455062,
      "grad_norm": 1.1601322180484739,
      "learning_rate": 1.2148817469569584e-06,
      "loss": 0.0762,
      "step": 2357
    },
    {
      "epoch": 0.5365187713310581,
      "grad_norm": 0.6289268827845146,
      "learning_rate": 1.214852211541136e-06,
      "loss": 0.0271,
      "step": 2358
    },
    {
      "epoch": 0.5367463026166098,
      "grad_norm": 1.127391002234741,
      "learning_rate": 1.2148226640698148e-06,
      "loss": 0.0374,
      "step": 2359
    },
    {
      "epoch": 0.5369738339021616,
      "grad_norm": 1.4660398732133397,
      "learning_rate": 1.2147931045435988e-06,
      "loss": 0.0851,
      "step": 2360
    },
    {
      "epoch": 0.5372013651877133,
      "grad_norm": 1.4378197818055753,
      "learning_rate": 1.2147635329630916e-06,
      "loss": 0.0887,
      "step": 2361
    },
    {
      "epoch": 0.5374288964732651,
      "grad_norm": 1.0407552632451906,
      "learning_rate": 1.214733949328898e-06,
      "loss": 0.0274,
      "step": 2362
    },
    {
      "epoch": 0.5376564277588168,
      "grad_norm": 1.2421238364546752,
      "learning_rate": 1.2147043536416226e-06,
      "loss": 0.0547,
      "step": 2363
    },
    {
      "epoch": 0.5378839590443686,
      "grad_norm": 1.9182602099931712,
      "learning_rate": 1.21467474590187e-06,
      "loss": 0.1061,
      "step": 2364
    },
    {
      "epoch": 0.5381114903299203,
      "grad_norm": 1.7756462089099831,
      "learning_rate": 1.2146451261102458e-06,
      "loss": 0.0633,
      "step": 2365
    },
    {
      "epoch": 0.5383390216154721,
      "grad_norm": 0.9945218699288574,
      "learning_rate": 1.2146154942673548e-06,
      "loss": 0.0458,
      "step": 2366
    },
    {
      "epoch": 0.5385665529010238,
      "grad_norm": 0.8213845393107763,
      "learning_rate": 1.2145858503738032e-06,
      "loss": 0.0425,
      "step": 2367
    },
    {
      "epoch": 0.5387940841865757,
      "grad_norm": 1.8457471393460891,
      "learning_rate": 1.2145561944301963e-06,
      "loss": 0.0922,
      "step": 2368
    },
    {
      "epoch": 0.5390216154721275,
      "grad_norm": 1.264742540597221,
      "learning_rate": 1.2145265264371406e-06,
      "loss": 0.0668,
      "step": 2369
    },
    {
      "epoch": 0.5392491467576792,
      "grad_norm": 0.8301569695908675,
      "learning_rate": 1.2144968463952425e-06,
      "loss": 0.0312,
      "step": 2370
    },
    {
      "epoch": 0.539476678043231,
      "grad_norm": 1.4005462540595164,
      "learning_rate": 1.2144671543051085e-06,
      "loss": 0.0385,
      "step": 2371
    },
    {
      "epoch": 0.5397042093287827,
      "grad_norm": 1.0427937857498775,
      "learning_rate": 1.2144374501673454e-06,
      "loss": 0.0405,
      "step": 2372
    },
    {
      "epoch": 0.5399317406143345,
      "grad_norm": 2.509030531929466,
      "learning_rate": 1.2144077339825603e-06,
      "loss": 0.129,
      "step": 2373
    },
    {
      "epoch": 0.5401592718998862,
      "grad_norm": 1.4490653559115954,
      "learning_rate": 1.2143780057513605e-06,
      "loss": 0.076,
      "step": 2374
    },
    {
      "epoch": 0.540386803185438,
      "grad_norm": 1.1167399295770026,
      "learning_rate": 1.2143482654743535e-06,
      "loss": 0.0433,
      "step": 2375
    },
    {
      "epoch": 0.5406143344709897,
      "grad_norm": 1.1732119937209518,
      "learning_rate": 1.2143185131521475e-06,
      "loss": 0.0639,
      "step": 2376
    },
    {
      "epoch": 0.5408418657565415,
      "grad_norm": 1.083852720432599,
      "learning_rate": 1.2142887487853503e-06,
      "loss": 0.0534,
      "step": 2377
    },
    {
      "epoch": 0.5410693970420933,
      "grad_norm": 1.2045837567267914,
      "learning_rate": 1.2142589723745705e-06,
      "loss": 0.047,
      "step": 2378
    },
    {
      "epoch": 0.5412969283276451,
      "grad_norm": 1.149610953840798,
      "learning_rate": 1.2142291839204163e-06,
      "loss": 0.0451,
      "step": 2379
    },
    {
      "epoch": 0.5415244596131968,
      "grad_norm": 1.17145770612206,
      "learning_rate": 1.2141993834234967e-06,
      "loss": 0.0594,
      "step": 2380
    },
    {
      "epoch": 0.5417519908987486,
      "grad_norm": 0.8279362230767302,
      "learning_rate": 1.2141695708844209e-06,
      "loss": 0.0331,
      "step": 2381
    },
    {
      "epoch": 0.5419795221843003,
      "grad_norm": 1.1783984776248138,
      "learning_rate": 1.214139746303798e-06,
      "loss": 0.0479,
      "step": 2382
    },
    {
      "epoch": 0.5422070534698521,
      "grad_norm": 1.6263426873577607,
      "learning_rate": 1.2141099096822376e-06,
      "loss": 0.0788,
      "step": 2383
    },
    {
      "epoch": 0.5424345847554038,
      "grad_norm": 1.339603186745851,
      "learning_rate": 1.2140800610203497e-06,
      "loss": 0.0819,
      "step": 2384
    },
    {
      "epoch": 0.5426621160409556,
      "grad_norm": 1.1850152950049393,
      "learning_rate": 1.214050200318744e-06,
      "loss": 0.0515,
      "step": 2385
    },
    {
      "epoch": 0.5428896473265074,
      "grad_norm": 1.4608498983467832,
      "learning_rate": 1.214020327578031e-06,
      "loss": 0.0528,
      "step": 2386
    },
    {
      "epoch": 0.5431171786120591,
      "grad_norm": 1.1158219646844105,
      "learning_rate": 1.2139904427988213e-06,
      "loss": 0.0328,
      "step": 2387
    },
    {
      "epoch": 0.543344709897611,
      "grad_norm": 0.8262393644685219,
      "learning_rate": 1.2139605459817259e-06,
      "loss": 0.0415,
      "step": 2388
    },
    {
      "epoch": 0.5435722411831627,
      "grad_norm": 1.096976415569224,
      "learning_rate": 1.2139306371273552e-06,
      "loss": 0.0619,
      "step": 2389
    },
    {
      "epoch": 0.5437997724687145,
      "grad_norm": 0.8704738503788053,
      "learning_rate": 1.213900716236321e-06,
      "loss": 0.0522,
      "step": 2390
    },
    {
      "epoch": 0.5440273037542662,
      "grad_norm": 0.9955866102135448,
      "learning_rate": 1.2138707833092345e-06,
      "loss": 0.0606,
      "step": 2391
    },
    {
      "epoch": 0.544254835039818,
      "grad_norm": 1.488547151459133,
      "learning_rate": 1.213840838346708e-06,
      "loss": 0.0871,
      "step": 2392
    },
    {
      "epoch": 0.5444823663253697,
      "grad_norm": 1.3949089434078865,
      "learning_rate": 1.213810881349353e-06,
      "loss": 0.0833,
      "step": 2393
    },
    {
      "epoch": 0.5447098976109215,
      "grad_norm": 1.5448754824040467,
      "learning_rate": 1.213780912317782e-06,
      "loss": 0.0567,
      "step": 2394
    },
    {
      "epoch": 0.5449374288964732,
      "grad_norm": 1.2930562589697765,
      "learning_rate": 1.2137509312526074e-06,
      "loss": 0.0486,
      "step": 2395
    },
    {
      "epoch": 0.545164960182025,
      "grad_norm": 0.8149733951933631,
      "learning_rate": 1.213720938154442e-06,
      "loss": 0.0447,
      "step": 2396
    },
    {
      "epoch": 0.5453924914675768,
      "grad_norm": 1.5207149596968195,
      "learning_rate": 1.2136909330238988e-06,
      "loss": 0.0712,
      "step": 2397
    },
    {
      "epoch": 0.5456200227531286,
      "grad_norm": 1.6094750936001516,
      "learning_rate": 1.213660915861591e-06,
      "loss": 0.0717,
      "step": 2398
    },
    {
      "epoch": 0.5458475540386803,
      "grad_norm": 1.0354263989549572,
      "learning_rate": 1.2136308866681323e-06,
      "loss": 0.0543,
      "step": 2399
    },
    {
      "epoch": 0.5460750853242321,
      "grad_norm": 1.312237280962152,
      "learning_rate": 1.2136008454441363e-06,
      "loss": 0.0651,
      "step": 2400
    },
    {
      "epoch": 0.5463026166097839,
      "grad_norm": 0.9265942352701293,
      "learning_rate": 1.213570792190217e-06,
      "loss": 0.0399,
      "step": 2401
    },
    {
      "epoch": 0.5465301478953356,
      "grad_norm": 1.3165932395587605,
      "learning_rate": 1.2135407269069885e-06,
      "loss": 0.0711,
      "step": 2402
    },
    {
      "epoch": 0.5467576791808874,
      "grad_norm": 1.3250744955676776,
      "learning_rate": 1.2135106495950655e-06,
      "loss": 0.055,
      "step": 2403
    },
    {
      "epoch": 0.5469852104664391,
      "grad_norm": 1.1269361373493476,
      "learning_rate": 1.2134805602550625e-06,
      "loss": 0.0391,
      "step": 2404
    },
    {
      "epoch": 0.5472127417519909,
      "grad_norm": 1.6294590878339321,
      "learning_rate": 1.2134504588875948e-06,
      "loss": 0.0752,
      "step": 2405
    },
    {
      "epoch": 0.5474402730375426,
      "grad_norm": 1.942161162101733,
      "learning_rate": 1.2134203454932772e-06,
      "loss": 0.111,
      "step": 2406
    },
    {
      "epoch": 0.5476678043230945,
      "grad_norm": 0.9091105494763078,
      "learning_rate": 1.2133902200727256e-06,
      "loss": 0.0366,
      "step": 2407
    },
    {
      "epoch": 0.5478953356086462,
      "grad_norm": 0.8734404453486602,
      "learning_rate": 1.2133600826265555e-06,
      "loss": 0.0339,
      "step": 2408
    },
    {
      "epoch": 0.548122866894198,
      "grad_norm": 1.4745463801324197,
      "learning_rate": 1.2133299331553826e-06,
      "loss": 0.0446,
      "step": 2409
    },
    {
      "epoch": 0.5483503981797497,
      "grad_norm": 0.9275974290923358,
      "learning_rate": 1.2132997716598236e-06,
      "loss": 0.0395,
      "step": 2410
    },
    {
      "epoch": 0.5485779294653015,
      "grad_norm": 0.8344512585289461,
      "learning_rate": 1.2132695981404943e-06,
      "loss": 0.0315,
      "step": 2411
    },
    {
      "epoch": 0.5488054607508532,
      "grad_norm": 1.6903517654405738,
      "learning_rate": 1.2132394125980122e-06,
      "loss": 0.0618,
      "step": 2412
    },
    {
      "epoch": 0.549032992036405,
      "grad_norm": 1.5596571212256216,
      "learning_rate": 1.2132092150329936e-06,
      "loss": 0.0623,
      "step": 2413
    },
    {
      "epoch": 0.5492605233219567,
      "grad_norm": 1.5970478431414654,
      "learning_rate": 1.213179005446056e-06,
      "loss": 0.0676,
      "step": 2414
    },
    {
      "epoch": 0.5494880546075085,
      "grad_norm": 1.1358043470989139,
      "learning_rate": 1.2131487838378167e-06,
      "loss": 0.0444,
      "step": 2415
    },
    {
      "epoch": 0.5497155858930602,
      "grad_norm": 0.7736657460795575,
      "learning_rate": 1.2131185502088932e-06,
      "loss": 0.0234,
      "step": 2416
    },
    {
      "epoch": 0.5499431171786121,
      "grad_norm": 1.0171839684581578,
      "learning_rate": 1.213088304559904e-06,
      "loss": 0.0354,
      "step": 2417
    },
    {
      "epoch": 0.5501706484641639,
      "grad_norm": 1.5638848346839926,
      "learning_rate": 1.2130580468914665e-06,
      "loss": 0.0907,
      "step": 2418
    },
    {
      "epoch": 0.5503981797497156,
      "grad_norm": 1.195018951013028,
      "learning_rate": 1.2130277772041999e-06,
      "loss": 0.0492,
      "step": 2419
    },
    {
      "epoch": 0.5506257110352674,
      "grad_norm": 1.0599309139929776,
      "learning_rate": 1.2129974954987222e-06,
      "loss": 0.0468,
      "step": 2420
    },
    {
      "epoch": 0.5508532423208191,
      "grad_norm": 0.9174363109363153,
      "learning_rate": 1.2129672017756524e-06,
      "loss": 0.0396,
      "step": 2421
    },
    {
      "epoch": 0.5510807736063709,
      "grad_norm": 0.996062027313234,
      "learning_rate": 1.2129368960356102e-06,
      "loss": 0.0461,
      "step": 2422
    },
    {
      "epoch": 0.5513083048919226,
      "grad_norm": 1.1748800644809907,
      "learning_rate": 1.2129065782792142e-06,
      "loss": 0.0468,
      "step": 2423
    },
    {
      "epoch": 0.5515358361774744,
      "grad_norm": 1.0572109261778357,
      "learning_rate": 1.2128762485070848e-06,
      "loss": 0.0337,
      "step": 2424
    },
    {
      "epoch": 0.5517633674630261,
      "grad_norm": 0.7078117825584842,
      "learning_rate": 1.2128459067198414e-06,
      "loss": 0.0279,
      "step": 2425
    },
    {
      "epoch": 0.5519908987485779,
      "grad_norm": 1.2504565872577411,
      "learning_rate": 1.2128155529181042e-06,
      "loss": 0.0432,
      "step": 2426
    },
    {
      "epoch": 0.5522184300341297,
      "grad_norm": 2.0545311286439656,
      "learning_rate": 1.2127851871024937e-06,
      "loss": 0.13,
      "step": 2427
    },
    {
      "epoch": 0.5524459613196815,
      "grad_norm": 1.6325039766646547,
      "learning_rate": 1.2127548092736305e-06,
      "loss": 0.0712,
      "step": 2428
    },
    {
      "epoch": 0.5526734926052332,
      "grad_norm": 1.4302972588358862,
      "learning_rate": 1.2127244194321353e-06,
      "loss": 0.0928,
      "step": 2429
    },
    {
      "epoch": 0.552901023890785,
      "grad_norm": 0.9939753923200273,
      "learning_rate": 1.2126940175786294e-06,
      "loss": 0.0537,
      "step": 2430
    },
    {
      "epoch": 0.5531285551763367,
      "grad_norm": 1.1837217608085497,
      "learning_rate": 1.212663603713734e-06,
      "loss": 0.0411,
      "step": 2431
    },
    {
      "epoch": 0.5533560864618885,
      "grad_norm": 1.0395121186554928,
      "learning_rate": 1.212633177838071e-06,
      "loss": 0.0499,
      "step": 2432
    },
    {
      "epoch": 0.5535836177474402,
      "grad_norm": 0.9514659467888016,
      "learning_rate": 1.2126027399522617e-06,
      "loss": 0.0453,
      "step": 2433
    },
    {
      "epoch": 0.553811149032992,
      "grad_norm": 1.6880603772633955,
      "learning_rate": 1.2125722900569288e-06,
      "loss": 0.0745,
      "step": 2434
    },
    {
      "epoch": 0.5540386803185438,
      "grad_norm": 1.8855045585505437,
      "learning_rate": 1.2125418281526944e-06,
      "loss": 0.1082,
      "step": 2435
    },
    {
      "epoch": 0.5542662116040956,
      "grad_norm": 1.2963290000539638,
      "learning_rate": 1.212511354240181e-06,
      "loss": 0.0571,
      "step": 2436
    },
    {
      "epoch": 0.5544937428896474,
      "grad_norm": 1.075287013825043,
      "learning_rate": 1.2124808683200113e-06,
      "loss": 0.0522,
      "step": 2437
    },
    {
      "epoch": 0.5547212741751991,
      "grad_norm": 1.5923929041091236,
      "learning_rate": 1.2124503703928088e-06,
      "loss": 0.0634,
      "step": 2438
    },
    {
      "epoch": 0.5549488054607509,
      "grad_norm": 1.832683570099282,
      "learning_rate": 1.2124198604591965e-06,
      "loss": 0.0659,
      "step": 2439
    },
    {
      "epoch": 0.5551763367463026,
      "grad_norm": 1.4530576893525877,
      "learning_rate": 1.2123893385197982e-06,
      "loss": 0.0508,
      "step": 2440
    },
    {
      "epoch": 0.5554038680318544,
      "grad_norm": 2.2165292342455967,
      "learning_rate": 1.2123588045752373e-06,
      "loss": 0.0788,
      "step": 2441
    },
    {
      "epoch": 0.5556313993174061,
      "grad_norm": 0.7214271689112302,
      "learning_rate": 1.2123282586261384e-06,
      "loss": 0.0373,
      "step": 2442
    },
    {
      "epoch": 0.5558589306029579,
      "grad_norm": 1.733249440041101,
      "learning_rate": 1.2122977006731256e-06,
      "loss": 0.0849,
      "step": 2443
    },
    {
      "epoch": 0.5560864618885096,
      "grad_norm": 0.9877695265064453,
      "learning_rate": 1.2122671307168232e-06,
      "loss": 0.0293,
      "step": 2444
    },
    {
      "epoch": 0.5563139931740614,
      "grad_norm": 2.369119537407654,
      "learning_rate": 1.212236548757856e-06,
      "loss": 0.1193,
      "step": 2445
    },
    {
      "epoch": 0.5565415244596132,
      "grad_norm": 0.9124531328873139,
      "learning_rate": 1.2122059547968496e-06,
      "loss": 0.0304,
      "step": 2446
    },
    {
      "epoch": 0.556769055745165,
      "grad_norm": 2.08793896339006,
      "learning_rate": 1.2121753488344286e-06,
      "loss": 0.102,
      "step": 2447
    },
    {
      "epoch": 0.5569965870307167,
      "grad_norm": 1.128455134895707,
      "learning_rate": 1.212144730871219e-06,
      "loss": 0.0446,
      "step": 2448
    },
    {
      "epoch": 0.5572241183162685,
      "grad_norm": 1.6401919927448718,
      "learning_rate": 1.2121141009078462e-06,
      "loss": 0.0557,
      "step": 2449
    },
    {
      "epoch": 0.5574516496018203,
      "grad_norm": 1.058585660351393,
      "learning_rate": 1.2120834589449368e-06,
      "loss": 0.0397,
      "step": 2450
    },
    {
      "epoch": 0.557679180887372,
      "grad_norm": 1.850497858074919,
      "learning_rate": 1.2120528049831165e-06,
      "loss": 0.0898,
      "step": 2451
    },
    {
      "epoch": 0.5579067121729238,
      "grad_norm": 2.1238500476076614,
      "learning_rate": 1.212022139023012e-06,
      "loss": 0.0775,
      "step": 2452
    },
    {
      "epoch": 0.5581342434584755,
      "grad_norm": 1.1451793439610103,
      "learning_rate": 1.21199146106525e-06,
      "loss": 0.0395,
      "step": 2453
    },
    {
      "epoch": 0.5583617747440273,
      "grad_norm": 0.8655064333461899,
      "learning_rate": 1.2119607711104574e-06,
      "loss": 0.0414,
      "step": 2454
    },
    {
      "epoch": 0.558589306029579,
      "grad_norm": 0.7739545180233858,
      "learning_rate": 1.211930069159262e-06,
      "loss": 0.0376,
      "step": 2455
    },
    {
      "epoch": 0.5588168373151309,
      "grad_norm": 1.1255668925149045,
      "learning_rate": 1.2118993552122907e-06,
      "loss": 0.0615,
      "step": 2456
    },
    {
      "epoch": 0.5590443686006826,
      "grad_norm": 0.8837865527117891,
      "learning_rate": 1.2118686292701715e-06,
      "loss": 0.033,
      "step": 2457
    },
    {
      "epoch": 0.5592718998862344,
      "grad_norm": 1.4857021615037345,
      "learning_rate": 1.211837891333532e-06,
      "loss": 0.0515,
      "step": 2458
    },
    {
      "epoch": 0.5594994311717861,
      "grad_norm": 1.687364075587055,
      "learning_rate": 1.211807141403001e-06,
      "loss": 0.0788,
      "step": 2459
    },
    {
      "epoch": 0.5597269624573379,
      "grad_norm": 1.1101722611046627,
      "learning_rate": 1.211776379479207e-06,
      "loss": 0.0361,
      "step": 2460
    },
    {
      "epoch": 0.5599544937428896,
      "grad_norm": 1.4952697824972245,
      "learning_rate": 1.211745605562778e-06,
      "loss": 0.0557,
      "step": 2461
    },
    {
      "epoch": 0.5601820250284414,
      "grad_norm": 0.8923044749255172,
      "learning_rate": 1.2117148196543436e-06,
      "loss": 0.0577,
      "step": 2462
    },
    {
      "epoch": 0.5604095563139931,
      "grad_norm": 2.562794461200439,
      "learning_rate": 1.2116840217545329e-06,
      "loss": 0.1001,
      "step": 2463
    },
    {
      "epoch": 0.5606370875995449,
      "grad_norm": 0.9586201769696352,
      "learning_rate": 1.211653211863975e-06,
      "loss": 0.0502,
      "step": 2464
    },
    {
      "epoch": 0.5608646188850968,
      "grad_norm": 1.4257435635791937,
      "learning_rate": 1.2116223899833e-06,
      "loss": 0.0544,
      "step": 2465
    },
    {
      "epoch": 0.5610921501706485,
      "grad_norm": 1.1082780136920223,
      "learning_rate": 1.2115915561131376e-06,
      "loss": 0.0368,
      "step": 2466
    },
    {
      "epoch": 0.5613196814562003,
      "grad_norm": 1.1191301112997847,
      "learning_rate": 1.211560710254118e-06,
      "loss": 0.0346,
      "step": 2467
    },
    {
      "epoch": 0.561547212741752,
      "grad_norm": 1.5912996699076145,
      "learning_rate": 1.211529852406872e-06,
      "loss": 0.0596,
      "step": 2468
    },
    {
      "epoch": 0.5617747440273038,
      "grad_norm": 1.3053443148766963,
      "learning_rate": 1.2114989825720298e-06,
      "loss": 0.0617,
      "step": 2469
    },
    {
      "epoch": 0.5620022753128555,
      "grad_norm": 1.935737911886316,
      "learning_rate": 1.2114681007502227e-06,
      "loss": 0.0713,
      "step": 2470
    },
    {
      "epoch": 0.5622298065984073,
      "grad_norm": 1.6784690811076477,
      "learning_rate": 1.2114372069420815e-06,
      "loss": 0.0898,
      "step": 2471
    },
    {
      "epoch": 0.562457337883959,
      "grad_norm": 1.6677773674255931,
      "learning_rate": 1.2114063011482378e-06,
      "loss": 0.0515,
      "step": 2472
    },
    {
      "epoch": 0.5626848691695108,
      "grad_norm": 0.7751790351928534,
      "learning_rate": 1.2113753833693234e-06,
      "loss": 0.0276,
      "step": 2473
    },
    {
      "epoch": 0.5629124004550625,
      "grad_norm": 1.5172528416500193,
      "learning_rate": 1.2113444536059699e-06,
      "loss": 0.078,
      "step": 2474
    },
    {
      "epoch": 0.5631399317406144,
      "grad_norm": 1.0306067244802375,
      "learning_rate": 1.2113135118588096e-06,
      "loss": 0.0601,
      "step": 2475
    },
    {
      "epoch": 0.5633674630261661,
      "grad_norm": 0.9276551177060361,
      "learning_rate": 1.2112825581284752e-06,
      "loss": 0.055,
      "step": 2476
    },
    {
      "epoch": 0.5635949943117179,
      "grad_norm": 1.2647808751954976,
      "learning_rate": 1.2112515924155987e-06,
      "loss": 0.048,
      "step": 2477
    },
    {
      "epoch": 0.5638225255972696,
      "grad_norm": 1.1803809621410315,
      "learning_rate": 1.2112206147208134e-06,
      "loss": 0.0513,
      "step": 2478
    },
    {
      "epoch": 0.5640500568828214,
      "grad_norm": 1.8816100409430934,
      "learning_rate": 1.2111896250447525e-06,
      "loss": 0.0972,
      "step": 2479
    },
    {
      "epoch": 0.5642775881683731,
      "grad_norm": 1.2512746487710684,
      "learning_rate": 1.211158623388049e-06,
      "loss": 0.0416,
      "step": 2480
    },
    {
      "epoch": 0.5645051194539249,
      "grad_norm": 1.1080723770280754,
      "learning_rate": 1.2111276097513369e-06,
      "loss": 0.044,
      "step": 2481
    },
    {
      "epoch": 0.5647326507394766,
      "grad_norm": 1.060740374175569,
      "learning_rate": 1.2110965841352498e-06,
      "loss": 0.0633,
      "step": 2482
    },
    {
      "epoch": 0.5649601820250284,
      "grad_norm": 1.0075017492283267,
      "learning_rate": 1.211065546540422e-06,
      "loss": 0.0341,
      "step": 2483
    },
    {
      "epoch": 0.5651877133105802,
      "grad_norm": 0.7064865281725817,
      "learning_rate": 1.2110344969674877e-06,
      "loss": 0.0286,
      "step": 2484
    },
    {
      "epoch": 0.565415244596132,
      "grad_norm": 1.1946068302141277,
      "learning_rate": 1.2110034354170816e-06,
      "loss": 0.0659,
      "step": 2485
    },
    {
      "epoch": 0.5656427758816838,
      "grad_norm": 0.9370117142292799,
      "learning_rate": 1.2109723618898383e-06,
      "loss": 0.0288,
      "step": 2486
    },
    {
      "epoch": 0.5658703071672355,
      "grad_norm": 2.2018132198270046,
      "learning_rate": 1.2109412763863933e-06,
      "loss": 0.1172,
      "step": 2487
    },
    {
      "epoch": 0.5660978384527873,
      "grad_norm": 1.1995855232529598,
      "learning_rate": 1.2109101789073815e-06,
      "loss": 0.0511,
      "step": 2488
    },
    {
      "epoch": 0.566325369738339,
      "grad_norm": 1.1031248857035114,
      "learning_rate": 1.2108790694534389e-06,
      "loss": 0.064,
      "step": 2489
    },
    {
      "epoch": 0.5665529010238908,
      "grad_norm": 1.738059897032761,
      "learning_rate": 1.2108479480252011e-06,
      "loss": 0.0756,
      "step": 2490
    },
    {
      "epoch": 0.5667804323094425,
      "grad_norm": 1.8262819909910217,
      "learning_rate": 1.210816814623304e-06,
      "loss": 0.1204,
      "step": 2491
    },
    {
      "epoch": 0.5670079635949943,
      "grad_norm": 0.9990130696869628,
      "learning_rate": 1.2107856692483843e-06,
      "loss": 0.0357,
      "step": 2492
    },
    {
      "epoch": 0.567235494880546,
      "grad_norm": 1.3613754270975984,
      "learning_rate": 1.2107545119010783e-06,
      "loss": 0.0638,
      "step": 2493
    },
    {
      "epoch": 0.5674630261660978,
      "grad_norm": 1.2474975066272391,
      "learning_rate": 1.2107233425820229e-06,
      "loss": 0.0599,
      "step": 2494
    },
    {
      "epoch": 0.5676905574516496,
      "grad_norm": 1.0265382923896895,
      "learning_rate": 1.2106921612918549e-06,
      "loss": 0.0385,
      "step": 2495
    },
    {
      "epoch": 0.5679180887372014,
      "grad_norm": 1.9111381375185976,
      "learning_rate": 1.2106609680312117e-06,
      "loss": 0.0686,
      "step": 2496
    },
    {
      "epoch": 0.5681456200227532,
      "grad_norm": 1.2560597740440176,
      "learning_rate": 1.210629762800731e-06,
      "loss": 0.0642,
      "step": 2497
    },
    {
      "epoch": 0.5683731513083049,
      "grad_norm": 1.0400297969186523,
      "learning_rate": 1.2105985456010506e-06,
      "loss": 0.0419,
      "step": 2498
    },
    {
      "epoch": 0.5686006825938567,
      "grad_norm": 1.5436476934145886,
      "learning_rate": 1.2105673164328081e-06,
      "loss": 0.0771,
      "step": 2499
    },
    {
      "epoch": 0.5688282138794084,
      "grad_norm": 1.2147723738279594,
      "learning_rate": 1.2105360752966424e-06,
      "loss": 0.044,
      "step": 2500
    },
    {
      "epoch": 0.5690557451649602,
      "grad_norm": 1.2940548926007058,
      "learning_rate": 1.2105048221931915e-06,
      "loss": 0.0556,
      "step": 2501
    },
    {
      "epoch": 0.5692832764505119,
      "grad_norm": 1.1343959115342286,
      "learning_rate": 1.2104735571230944e-06,
      "loss": 0.0325,
      "step": 2502
    },
    {
      "epoch": 0.5695108077360637,
      "grad_norm": 0.7696868285921366,
      "learning_rate": 1.21044228008699e-06,
      "loss": 0.0321,
      "step": 2503
    },
    {
      "epoch": 0.5697383390216155,
      "grad_norm": 1.277277873025092,
      "learning_rate": 1.2104109910855176e-06,
      "loss": 0.0597,
      "step": 2504
    },
    {
      "epoch": 0.5699658703071673,
      "grad_norm": 1.433825874950651,
      "learning_rate": 1.2103796901193166e-06,
      "loss": 0.0763,
      "step": 2505
    },
    {
      "epoch": 0.570193401592719,
      "grad_norm": 0.8577837659660162,
      "learning_rate": 1.210348377189027e-06,
      "loss": 0.0284,
      "step": 2506
    },
    {
      "epoch": 0.5704209328782708,
      "grad_norm": 1.1944391480100813,
      "learning_rate": 1.2103170522952885e-06,
      "loss": 0.0562,
      "step": 2507
    },
    {
      "epoch": 0.5706484641638225,
      "grad_norm": 1.283294042740657,
      "learning_rate": 1.2102857154387413e-06,
      "loss": 0.0468,
      "step": 2508
    },
    {
      "epoch": 0.5708759954493743,
      "grad_norm": 1.7437964279202711,
      "learning_rate": 1.210254366620026e-06,
      "loss": 0.097,
      "step": 2509
    },
    {
      "epoch": 0.571103526734926,
      "grad_norm": 1.224329122924591,
      "learning_rate": 1.2102230058397832e-06,
      "loss": 0.0595,
      "step": 2510
    },
    {
      "epoch": 0.5713310580204778,
      "grad_norm": 1.3007053728417897,
      "learning_rate": 1.210191633098654e-06,
      "loss": 0.0685,
      "step": 2511
    },
    {
      "epoch": 0.5715585893060295,
      "grad_norm": 1.1429406193383025,
      "learning_rate": 1.2101602483972797e-06,
      "loss": 0.048,
      "step": 2512
    },
    {
      "epoch": 0.5717861205915813,
      "grad_norm": 0.7595899866293734,
      "learning_rate": 1.2101288517363016e-06,
      "loss": 0.0194,
      "step": 2513
    },
    {
      "epoch": 0.5720136518771332,
      "grad_norm": 1.025256317864455,
      "learning_rate": 1.2100974431163614e-06,
      "loss": 0.0423,
      "step": 2514
    },
    {
      "epoch": 0.5722411831626849,
      "grad_norm": 1.053909591806734,
      "learning_rate": 1.2100660225381008e-06,
      "loss": 0.0284,
      "step": 2515
    },
    {
      "epoch": 0.5724687144482367,
      "grad_norm": 1.384456570911114,
      "learning_rate": 1.2100345900021624e-06,
      "loss": 0.0659,
      "step": 2516
    },
    {
      "epoch": 0.5726962457337884,
      "grad_norm": 1.3325135654239626,
      "learning_rate": 1.2100031455091883e-06,
      "loss": 0.0675,
      "step": 2517
    },
    {
      "epoch": 0.5729237770193402,
      "grad_norm": 1.179139361018548,
      "learning_rate": 1.2099716890598212e-06,
      "loss": 0.0475,
      "step": 2518
    },
    {
      "epoch": 0.5731513083048919,
      "grad_norm": 1.1347488592526416,
      "learning_rate": 1.2099402206547042e-06,
      "loss": 0.0563,
      "step": 2519
    },
    {
      "epoch": 0.5733788395904437,
      "grad_norm": 1.2845284052496178,
      "learning_rate": 1.2099087402944805e-06,
      "loss": 0.053,
      "step": 2520
    },
    {
      "epoch": 0.5736063708759954,
      "grad_norm": 0.9939680061445038,
      "learning_rate": 1.2098772479797933e-06,
      "loss": 0.0388,
      "step": 2521
    },
    {
      "epoch": 0.5738339021615472,
      "grad_norm": 1.2948668621318795,
      "learning_rate": 1.2098457437112862e-06,
      "loss": 0.0451,
      "step": 2522
    },
    {
      "epoch": 0.5740614334470989,
      "grad_norm": 1.6486523140737381,
      "learning_rate": 1.2098142274896033e-06,
      "loss": 0.0722,
      "step": 2523
    },
    {
      "epoch": 0.5742889647326508,
      "grad_norm": 1.1690776697359815,
      "learning_rate": 1.2097826993153886e-06,
      "loss": 0.0504,
      "step": 2524
    },
    {
      "epoch": 0.5745164960182025,
      "grad_norm": 0.8812949923396778,
      "learning_rate": 1.2097511591892863e-06,
      "loss": 0.0232,
      "step": 2525
    },
    {
      "epoch": 0.5747440273037543,
      "grad_norm": 1.6847623228179178,
      "learning_rate": 1.2097196071119415e-06,
      "loss": 0.0643,
      "step": 2526
    },
    {
      "epoch": 0.574971558589306,
      "grad_norm": 1.5124757371829332,
      "learning_rate": 1.2096880430839985e-06,
      "loss": 0.0859,
      "step": 2527
    },
    {
      "epoch": 0.5751990898748578,
      "grad_norm": 1.6240299660391282,
      "learning_rate": 1.2096564671061031e-06,
      "loss": 0.0637,
      "step": 2528
    },
    {
      "epoch": 0.5754266211604095,
      "grad_norm": 1.5970094647815496,
      "learning_rate": 1.2096248791789e-06,
      "loss": 0.0654,
      "step": 2529
    },
    {
      "epoch": 0.5756541524459613,
      "grad_norm": 1.381679821195058,
      "learning_rate": 1.2095932793030353e-06,
      "loss": 0.0513,
      "step": 2530
    },
    {
      "epoch": 0.575881683731513,
      "grad_norm": 1.468449586854502,
      "learning_rate": 1.2095616674791543e-06,
      "loss": 0.064,
      "step": 2531
    },
    {
      "epoch": 0.5761092150170648,
      "grad_norm": 1.3812350043911665,
      "learning_rate": 1.2095300437079034e-06,
      "loss": 0.0532,
      "step": 2532
    },
    {
      "epoch": 0.5763367463026167,
      "grad_norm": 0.9730226451517775,
      "learning_rate": 1.2094984079899292e-06,
      "loss": 0.0355,
      "step": 2533
    },
    {
      "epoch": 0.5765642775881684,
      "grad_norm": 1.0694961505250529,
      "learning_rate": 1.2094667603258779e-06,
      "loss": 0.0404,
      "step": 2534
    },
    {
      "epoch": 0.5767918088737202,
      "grad_norm": 1.2344551784501863,
      "learning_rate": 1.2094351007163962e-06,
      "loss": 0.0472,
      "step": 2535
    },
    {
      "epoch": 0.5770193401592719,
      "grad_norm": 1.256788629460907,
      "learning_rate": 1.2094034291621315e-06,
      "loss": 0.0744,
      "step": 2536
    },
    {
      "epoch": 0.5772468714448237,
      "grad_norm": 1.1106837372555467,
      "learning_rate": 1.2093717456637311e-06,
      "loss": 0.0338,
      "step": 2537
    },
    {
      "epoch": 0.5774744027303754,
      "grad_norm": 1.0354356057027125,
      "learning_rate": 1.2093400502218422e-06,
      "loss": 0.0525,
      "step": 2538
    },
    {
      "epoch": 0.5777019340159272,
      "grad_norm": 1.3796812821808617,
      "learning_rate": 1.209308342837113e-06,
      "loss": 0.0582,
      "step": 2539
    },
    {
      "epoch": 0.5779294653014789,
      "grad_norm": 1.0890955420481698,
      "learning_rate": 1.2092766235101917e-06,
      "loss": 0.0597,
      "step": 2540
    },
    {
      "epoch": 0.5781569965870307,
      "grad_norm": 1.0327303160169794,
      "learning_rate": 1.2092448922417258e-06,
      "loss": 0.0549,
      "step": 2541
    },
    {
      "epoch": 0.5783845278725824,
      "grad_norm": 1.0446209152101336,
      "learning_rate": 1.2092131490323644e-06,
      "loss": 0.0512,
      "step": 2542
    },
    {
      "epoch": 0.5786120591581343,
      "grad_norm": 1.1872918690030627,
      "learning_rate": 1.2091813938827563e-06,
      "loss": 0.0467,
      "step": 2543
    },
    {
      "epoch": 0.578839590443686,
      "grad_norm": 1.6213845013127128,
      "learning_rate": 1.2091496267935502e-06,
      "loss": 0.0913,
      "step": 2544
    },
    {
      "epoch": 0.5790671217292378,
      "grad_norm": 0.25107695804250185,
      "learning_rate": 1.2091178477653957e-06,
      "loss": 0.017,
      "step": 2545
    },
    {
      "epoch": 0.5792946530147896,
      "grad_norm": 0.8996867474301544,
      "learning_rate": 1.209086056798942e-06,
      "loss": 0.0386,
      "step": 2546
    },
    {
      "epoch": 0.5795221843003413,
      "grad_norm": 2.2594305507095944,
      "learning_rate": 1.2090542538948392e-06,
      "loss": 0.1133,
      "step": 2547
    },
    {
      "epoch": 0.579749715585893,
      "grad_norm": 1.1393658831306852,
      "learning_rate": 1.209022439053737e-06,
      "loss": 0.0445,
      "step": 2548
    },
    {
      "epoch": 0.5799772468714448,
      "grad_norm": 1.2977706300782228,
      "learning_rate": 1.2089906122762859e-06,
      "loss": 0.0754,
      "step": 2549
    },
    {
      "epoch": 0.5802047781569966,
      "grad_norm": 1.8878513858209192,
      "learning_rate": 1.208958773563136e-06,
      "loss": 0.0896,
      "step": 2550
    },
    {
      "epoch": 0.5804323094425483,
      "grad_norm": 1.8406180655967013,
      "learning_rate": 1.2089269229149383e-06,
      "loss": 0.0884,
      "step": 2551
    },
    {
      "epoch": 0.5806598407281001,
      "grad_norm": 1.0239444959470219,
      "learning_rate": 1.2088950603323438e-06,
      "loss": 0.0461,
      "step": 2552
    },
    {
      "epoch": 0.5808873720136519,
      "grad_norm": 1.0773801965795455,
      "learning_rate": 1.2088631858160033e-06,
      "loss": 0.0415,
      "step": 2553
    },
    {
      "epoch": 0.5811149032992037,
      "grad_norm": 0.9310673305069316,
      "learning_rate": 1.2088312993665689e-06,
      "loss": 0.042,
      "step": 2554
    },
    {
      "epoch": 0.5813424345847554,
      "grad_norm": 1.3375692899110723,
      "learning_rate": 1.208799400984692e-06,
      "loss": 0.0727,
      "step": 2555
    },
    {
      "epoch": 0.5815699658703072,
      "grad_norm": 1.4881435052347582,
      "learning_rate": 1.2087674906710242e-06,
      "loss": 0.0482,
      "step": 2556
    },
    {
      "epoch": 0.5817974971558589,
      "grad_norm": 0.9553759696828109,
      "learning_rate": 1.2087355684262183e-06,
      "loss": 0.0631,
      "step": 2557
    },
    {
      "epoch": 0.5820250284414107,
      "grad_norm": 0.6274163364554146,
      "learning_rate": 1.2087036342509265e-06,
      "loss": 0.0219,
      "step": 2558
    },
    {
      "epoch": 0.5822525597269624,
      "grad_norm": 1.3757750182752106,
      "learning_rate": 1.208671688145801e-06,
      "loss": 0.0626,
      "step": 2559
    },
    {
      "epoch": 0.5824800910125142,
      "grad_norm": 2.481636121235935,
      "learning_rate": 1.2086397301114955e-06,
      "loss": 0.1418,
      "step": 2560
    },
    {
      "epoch": 0.5827076222980659,
      "grad_norm": 1.0378265608573296,
      "learning_rate": 1.2086077601486627e-06,
      "loss": 0.0576,
      "step": 2561
    },
    {
      "epoch": 0.5829351535836177,
      "grad_norm": 1.4924581569226714,
      "learning_rate": 1.2085757782579562e-06,
      "loss": 0.0816,
      "step": 2562
    },
    {
      "epoch": 0.5831626848691696,
      "grad_norm": 0.8886308787621965,
      "learning_rate": 1.2085437844400293e-06,
      "loss": 0.0278,
      "step": 2563
    },
    {
      "epoch": 0.5833902161547213,
      "grad_norm": 1.9559855395157613,
      "learning_rate": 1.2085117786955366e-06,
      "loss": 0.107,
      "step": 2564
    },
    {
      "epoch": 0.5836177474402731,
      "grad_norm": 1.3285689681822581,
      "learning_rate": 1.2084797610251314e-06,
      "loss": 0.0593,
      "step": 2565
    },
    {
      "epoch": 0.5838452787258248,
      "grad_norm": 1.042908390638745,
      "learning_rate": 1.2084477314294688e-06,
      "loss": 0.0299,
      "step": 2566
    },
    {
      "epoch": 0.5840728100113766,
      "grad_norm": 0.9604010784829523,
      "learning_rate": 1.2084156899092028e-06,
      "loss": 0.0336,
      "step": 2567
    },
    {
      "epoch": 0.5843003412969283,
      "grad_norm": 0.9621929837602977,
      "learning_rate": 1.2083836364649888e-06,
      "loss": 0.0431,
      "step": 2568
    },
    {
      "epoch": 0.5845278725824801,
      "grad_norm": 1.2097619610555346,
      "learning_rate": 1.2083515710974813e-06,
      "loss": 0.0446,
      "step": 2569
    },
    {
      "epoch": 0.5847554038680318,
      "grad_norm": 1.3118996444529896,
      "learning_rate": 1.2083194938073363e-06,
      "loss": 0.0491,
      "step": 2570
    },
    {
      "epoch": 0.5849829351535836,
      "grad_norm": 2.2788662451246915,
      "learning_rate": 1.2082874045952094e-06,
      "loss": 0.0905,
      "step": 2571
    },
    {
      "epoch": 0.5852104664391354,
      "grad_norm": 1.7999056722845332,
      "learning_rate": 1.2082553034617559e-06,
      "loss": 0.0798,
      "step": 2572
    },
    {
      "epoch": 0.5854379977246872,
      "grad_norm": 1.1859950837264095,
      "learning_rate": 1.2082231904076323e-06,
      "loss": 0.0361,
      "step": 2573
    },
    {
      "epoch": 0.5856655290102389,
      "grad_norm": 2.3662421804783618,
      "learning_rate": 1.2081910654334948e-06,
      "loss": 0.0962,
      "step": 2574
    },
    {
      "epoch": 0.5858930602957907,
      "grad_norm": 1.248609581967699,
      "learning_rate": 1.2081589285399998e-06,
      "loss": 0.0509,
      "step": 2575
    },
    {
      "epoch": 0.5861205915813424,
      "grad_norm": 1.1728856094231628,
      "learning_rate": 1.2081267797278043e-06,
      "loss": 0.0474,
      "step": 2576
    },
    {
      "epoch": 0.5863481228668942,
      "grad_norm": 1.3290932236966655,
      "learning_rate": 1.2080946189975656e-06,
      "loss": 0.0608,
      "step": 2577
    },
    {
      "epoch": 0.586575654152446,
      "grad_norm": 1.3099426521521451,
      "learning_rate": 1.2080624463499407e-06,
      "loss": 0.0697,
      "step": 2578
    },
    {
      "epoch": 0.5868031854379977,
      "grad_norm": 1.4265140149288442,
      "learning_rate": 1.2080302617855874e-06,
      "loss": 0.0802,
      "step": 2579
    },
    {
      "epoch": 0.5870307167235495,
      "grad_norm": 0.6748962342724948,
      "learning_rate": 1.2079980653051629e-06,
      "loss": 0.0194,
      "step": 2580
    },
    {
      "epoch": 0.5872582480091012,
      "grad_norm": 1.1857133228322694,
      "learning_rate": 1.207965856909326e-06,
      "loss": 0.0603,
      "step": 2581
    },
    {
      "epoch": 0.5874857792946531,
      "grad_norm": 1.4269876330750941,
      "learning_rate": 1.2079336365987345e-06,
      "loss": 0.0674,
      "step": 2582
    },
    {
      "epoch": 0.5877133105802048,
      "grad_norm": 1.1171535242191324,
      "learning_rate": 1.2079014043740471e-06,
      "loss": 0.0474,
      "step": 2583
    },
    {
      "epoch": 0.5879408418657566,
      "grad_norm": 1.049798577134009,
      "learning_rate": 1.2078691602359224e-06,
      "loss": 0.0515,
      "step": 2584
    },
    {
      "epoch": 0.5881683731513083,
      "grad_norm": 1.4079703732165962,
      "learning_rate": 1.2078369041850197e-06,
      "loss": 0.0614,
      "step": 2585
    },
    {
      "epoch": 0.5883959044368601,
      "grad_norm": 0.6740789590290865,
      "learning_rate": 1.207804636221998e-06,
      "loss": 0.0259,
      "step": 2586
    },
    {
      "epoch": 0.5886234357224118,
      "grad_norm": 1.189214147562108,
      "learning_rate": 1.207772356347517e-06,
      "loss": 0.0527,
      "step": 2587
    },
    {
      "epoch": 0.5888509670079636,
      "grad_norm": 1.1360915553171849,
      "learning_rate": 1.2077400645622363e-06,
      "loss": 0.0458,
      "step": 2588
    },
    {
      "epoch": 0.5890784982935153,
      "grad_norm": 0.9198535003511792,
      "learning_rate": 1.207707760866816e-06,
      "loss": 0.0402,
      "step": 2589
    },
    {
      "epoch": 0.5893060295790671,
      "grad_norm": 1.174521842214451,
      "learning_rate": 1.207675445261916e-06,
      "loss": 0.0421,
      "step": 2590
    },
    {
      "epoch": 0.5895335608646188,
      "grad_norm": 2.8929933375186287,
      "learning_rate": 1.207643117748197e-06,
      "loss": 0.1536,
      "step": 2591
    },
    {
      "epoch": 0.5897610921501707,
      "grad_norm": 1.5474448385855322,
      "learning_rate": 1.2076107783263199e-06,
      "loss": 0.0567,
      "step": 2592
    },
    {
      "epoch": 0.5899886234357224,
      "grad_norm": 0.8451598056278512,
      "learning_rate": 1.2075784269969457e-06,
      "loss": 0.0362,
      "step": 2593
    },
    {
      "epoch": 0.5902161547212742,
      "grad_norm": 0.7031869485161736,
      "learning_rate": 1.2075460637607351e-06,
      "loss": 0.0312,
      "step": 2594
    },
    {
      "epoch": 0.590443686006826,
      "grad_norm": 1.449466507606748,
      "learning_rate": 1.20751368861835e-06,
      "loss": 0.0573,
      "step": 2595
    },
    {
      "epoch": 0.5906712172923777,
      "grad_norm": 1.1927092104354498,
      "learning_rate": 1.2074813015704518e-06,
      "loss": 0.0402,
      "step": 2596
    },
    {
      "epoch": 0.5908987485779295,
      "grad_norm": 1.636503401466634,
      "learning_rate": 1.2074489026177024e-06,
      "loss": 0.0657,
      "step": 2597
    },
    {
      "epoch": 0.5911262798634812,
      "grad_norm": 1.4909987612945546,
      "learning_rate": 1.2074164917607644e-06,
      "loss": 0.0658,
      "step": 2598
    },
    {
      "epoch": 0.591353811149033,
      "grad_norm": 1.1952099163133536,
      "learning_rate": 1.2073840690003e-06,
      "loss": 0.0603,
      "step": 2599
    },
    {
      "epoch": 0.5915813424345847,
      "grad_norm": 1.0622134341599558,
      "learning_rate": 1.2073516343369717e-06,
      "loss": 0.0459,
      "step": 2600
    },
    {
      "epoch": 0.5918088737201365,
      "grad_norm": 0.8032008417926766,
      "learning_rate": 1.2073191877714424e-06,
      "loss": 0.0368,
      "step": 2601
    },
    {
      "epoch": 0.5920364050056883,
      "grad_norm": 1.1983356217235557,
      "learning_rate": 1.2072867293043755e-06,
      "loss": 0.045,
      "step": 2602
    },
    {
      "epoch": 0.5922639362912401,
      "grad_norm": 1.1811729971845715,
      "learning_rate": 1.2072542589364343e-06,
      "loss": 0.0413,
      "step": 2603
    },
    {
      "epoch": 0.5924914675767918,
      "grad_norm": 0.9199950791965811,
      "learning_rate": 1.2072217766682822e-06,
      "loss": 0.0471,
      "step": 2604
    },
    {
      "epoch": 0.5927189988623436,
      "grad_norm": 1.5522989255979738,
      "learning_rate": 1.2071892825005835e-06,
      "loss": 0.0531,
      "step": 2605
    },
    {
      "epoch": 0.5929465301478953,
      "grad_norm": 2.34185081988946,
      "learning_rate": 1.207156776434002e-06,
      "loss": 0.1052,
      "step": 2606
    },
    {
      "epoch": 0.5931740614334471,
      "grad_norm": 1.6203202208735543,
      "learning_rate": 1.2071242584692022e-06,
      "loss": 0.0824,
      "step": 2607
    },
    {
      "epoch": 0.5934015927189988,
      "grad_norm": 1.8225272194053617,
      "learning_rate": 1.2070917286068486e-06,
      "loss": 0.0816,
      "step": 2608
    },
    {
      "epoch": 0.5936291240045506,
      "grad_norm": 1.0482127292144585,
      "learning_rate": 1.2070591868476062e-06,
      "loss": 0.0336,
      "step": 2609
    },
    {
      "epoch": 0.5938566552901023,
      "grad_norm": 1.2383829160698303,
      "learning_rate": 1.20702663319214e-06,
      "loss": 0.0558,
      "step": 2610
    },
    {
      "epoch": 0.5940841865756542,
      "grad_norm": 1.0803065071888958,
      "learning_rate": 1.2069940676411154e-06,
      "loss": 0.0663,
      "step": 2611
    },
    {
      "epoch": 0.594311717861206,
      "grad_norm": 1.5171982637367767,
      "learning_rate": 1.2069614901951978e-06,
      "loss": 0.0601,
      "step": 2612
    },
    {
      "epoch": 0.5945392491467577,
      "grad_norm": 1.3847567182951876,
      "learning_rate": 1.2069289008550533e-06,
      "loss": 0.0611,
      "step": 2613
    },
    {
      "epoch": 0.5947667804323095,
      "grad_norm": 0.9955435148692463,
      "learning_rate": 1.2068962996213476e-06,
      "loss": 0.0403,
      "step": 2614
    },
    {
      "epoch": 0.5949943117178612,
      "grad_norm": 1.13405892693637,
      "learning_rate": 1.2068636864947475e-06,
      "loss": 0.0519,
      "step": 2615
    },
    {
      "epoch": 0.595221843003413,
      "grad_norm": 0.7900528990598964,
      "learning_rate": 1.206831061475919e-06,
      "loss": 0.027,
      "step": 2616
    },
    {
      "epoch": 0.5954493742889647,
      "grad_norm": 0.6257774504076663,
      "learning_rate": 1.2067984245655292e-06,
      "loss": 0.0174,
      "step": 2617
    },
    {
      "epoch": 0.5956769055745165,
      "grad_norm": 0.7037772555599539,
      "learning_rate": 1.2067657757642453e-06,
      "loss": 0.0168,
      "step": 2618
    },
    {
      "epoch": 0.5959044368600682,
      "grad_norm": 0.8538766117483473,
      "learning_rate": 1.2067331150727343e-06,
      "loss": 0.0382,
      "step": 2619
    },
    {
      "epoch": 0.59613196814562,
      "grad_norm": 0.9862142615994054,
      "learning_rate": 1.206700442491664e-06,
      "loss": 0.0359,
      "step": 2620
    },
    {
      "epoch": 0.5963594994311718,
      "grad_norm": 1.5267273993929875,
      "learning_rate": 1.2066677580217018e-06,
      "loss": 0.0625,
      "step": 2621
    },
    {
      "epoch": 0.5965870307167236,
      "grad_norm": 1.5028260871816748,
      "learning_rate": 1.2066350616635159e-06,
      "loss": 0.0545,
      "step": 2622
    },
    {
      "epoch": 0.5968145620022753,
      "grad_norm": 2.64736254484509,
      "learning_rate": 1.2066023534177746e-06,
      "loss": 0.1266,
      "step": 2623
    },
    {
      "epoch": 0.5970420932878271,
      "grad_norm": 1.4120526861644525,
      "learning_rate": 1.2065696332851463e-06,
      "loss": 0.0591,
      "step": 2624
    },
    {
      "epoch": 0.5972696245733788,
      "grad_norm": 0.9067523643094818,
      "learning_rate": 1.2065369012662999e-06,
      "loss": 0.0331,
      "step": 2625
    },
    {
      "epoch": 0.5974971558589306,
      "grad_norm": 1.0988059318675343,
      "learning_rate": 1.206504157361904e-06,
      "loss": 0.0605,
      "step": 2626
    },
    {
      "epoch": 0.5977246871444823,
      "grad_norm": 1.2270739952449312,
      "learning_rate": 1.2064714015726283e-06,
      "loss": 0.056,
      "step": 2627
    },
    {
      "epoch": 0.5979522184300341,
      "grad_norm": 0.9257304826598889,
      "learning_rate": 1.2064386338991423e-06,
      "loss": 0.0433,
      "step": 2628
    },
    {
      "epoch": 0.5981797497155859,
      "grad_norm": 1.8761997788068991,
      "learning_rate": 1.2064058543421152e-06,
      "loss": 0.084,
      "step": 2629
    },
    {
      "epoch": 0.5984072810011376,
      "grad_norm": 1.5801176338760017,
      "learning_rate": 1.2063730629022173e-06,
      "loss": 0.0419,
      "step": 2630
    },
    {
      "epoch": 0.5986348122866895,
      "grad_norm": 1.1971080645956096,
      "learning_rate": 1.2063402595801187e-06,
      "loss": 0.0438,
      "step": 2631
    },
    {
      "epoch": 0.5988623435722412,
      "grad_norm": 1.405844210266892,
      "learning_rate": 1.2063074443764897e-06,
      "loss": 0.0624,
      "step": 2632
    },
    {
      "epoch": 0.599089874857793,
      "grad_norm": 1.5341316627448498,
      "learning_rate": 1.2062746172920014e-06,
      "loss": 0.088,
      "step": 2633
    },
    {
      "epoch": 0.5993174061433447,
      "grad_norm": 1.0531520309166145,
      "learning_rate": 1.2062417783273246e-06,
      "loss": 0.0474,
      "step": 2634
    },
    {
      "epoch": 0.5995449374288965,
      "grad_norm": 1.36821624035935,
      "learning_rate": 1.2062089274831299e-06,
      "loss": 0.0677,
      "step": 2635
    },
    {
      "epoch": 0.5997724687144482,
      "grad_norm": 0.7040976905373539,
      "learning_rate": 1.2061760647600894e-06,
      "loss": 0.0308,
      "step": 2636
    },
    {
      "epoch": 0.6,
      "grad_norm": 0.9178208746030599,
      "learning_rate": 1.2061431901588747e-06,
      "loss": 0.0205,
      "step": 2637
    },
    {
      "epoch": 0.6002275312855517,
      "grad_norm": 0.9105858796703735,
      "learning_rate": 1.2061103036801573e-06,
      "loss": 0.0442,
      "step": 2638
    },
    {
      "epoch": 0.6004550625711035,
      "grad_norm": 1.40951436107761,
      "learning_rate": 1.2060774053246096e-06,
      "loss": 0.0644,
      "step": 2639
    },
    {
      "epoch": 0.6006825938566553,
      "grad_norm": 1.1409764583060835,
      "learning_rate": 1.2060444950929038e-06,
      "loss": 0.0705,
      "step": 2640
    },
    {
      "epoch": 0.6009101251422071,
      "grad_norm": 1.1909546375772935,
      "learning_rate": 1.2060115729857128e-06,
      "loss": 0.0515,
      "step": 2641
    },
    {
      "epoch": 0.6011376564277588,
      "grad_norm": 1.2179103001644596,
      "learning_rate": 1.2059786390037093e-06,
      "loss": 0.0387,
      "step": 2642
    },
    {
      "epoch": 0.6013651877133106,
      "grad_norm": 1.135969807370913,
      "learning_rate": 1.2059456931475663e-06,
      "loss": 0.0224,
      "step": 2643
    },
    {
      "epoch": 0.6015927189988624,
      "grad_norm": 1.6455009465660149,
      "learning_rate": 1.2059127354179573e-06,
      "loss": 0.1139,
      "step": 2644
    },
    {
      "epoch": 0.6018202502844141,
      "grad_norm": 1.0904940664660454,
      "learning_rate": 1.205879765815556e-06,
      "loss": 0.0409,
      "step": 2645
    },
    {
      "epoch": 0.6020477815699659,
      "grad_norm": 1.8187560107849106,
      "learning_rate": 1.205846784341036e-06,
      "loss": 0.0746,
      "step": 2646
    },
    {
      "epoch": 0.6022753128555176,
      "grad_norm": 1.2846083666249735,
      "learning_rate": 1.2058137909950719e-06,
      "loss": 0.0397,
      "step": 2647
    },
    {
      "epoch": 0.6025028441410694,
      "grad_norm": 1.3139384696968406,
      "learning_rate": 1.2057807857783371e-06,
      "loss": 0.0513,
      "step": 2648
    },
    {
      "epoch": 0.6027303754266211,
      "grad_norm": 1.5334002395835504,
      "learning_rate": 1.205747768691507e-06,
      "loss": 0.0451,
      "step": 2649
    },
    {
      "epoch": 0.602957906712173,
      "grad_norm": 0.9668095767338745,
      "learning_rate": 1.2057147397352559e-06,
      "loss": 0.0519,
      "step": 2650
    },
    {
      "epoch": 0.6031854379977247,
      "grad_norm": 1.1043891282747957,
      "learning_rate": 1.2056816989102591e-06,
      "loss": 0.0421,
      "step": 2651
    },
    {
      "epoch": 0.6034129692832765,
      "grad_norm": 1.63627667644995,
      "learning_rate": 1.2056486462171918e-06,
      "loss": 0.0715,
      "step": 2652
    },
    {
      "epoch": 0.6036405005688282,
      "grad_norm": 0.986630375032043,
      "learning_rate": 1.2056155816567297e-06,
      "loss": 0.0497,
      "step": 2653
    },
    {
      "epoch": 0.60386803185438,
      "grad_norm": 1.198264842052037,
      "learning_rate": 1.2055825052295486e-06,
      "loss": 0.0565,
      "step": 2654
    },
    {
      "epoch": 0.6040955631399317,
      "grad_norm": 0.7380211191011931,
      "learning_rate": 1.205549416936324e-06,
      "loss": 0.0423,
      "step": 2655
    },
    {
      "epoch": 0.6043230944254835,
      "grad_norm": 0.8003290473544856,
      "learning_rate": 1.2055163167777328e-06,
      "loss": 0.057,
      "step": 2656
    },
    {
      "epoch": 0.6045506257110352,
      "grad_norm": 1.2699158922395328,
      "learning_rate": 1.2054832047544512e-06,
      "loss": 0.0588,
      "step": 2657
    },
    {
      "epoch": 0.604778156996587,
      "grad_norm": 0.7610615046539442,
      "learning_rate": 1.205450080867156e-06,
      "loss": 0.0343,
      "step": 2658
    },
    {
      "epoch": 0.6050056882821387,
      "grad_norm": 1.3103630700346172,
      "learning_rate": 1.205416945116524e-06,
      "loss": 0.0406,
      "step": 2659
    },
    {
      "epoch": 0.6052332195676906,
      "grad_norm": 1.0165185597393296,
      "learning_rate": 1.2053837975032328e-06,
      "loss": 0.0337,
      "step": 2660
    },
    {
      "epoch": 0.6054607508532424,
      "grad_norm": 1.387250593088855,
      "learning_rate": 1.2053506380279597e-06,
      "loss": 0.0603,
      "step": 2661
    },
    {
      "epoch": 0.6056882821387941,
      "grad_norm": 1.335719349119118,
      "learning_rate": 1.2053174666913826e-06,
      "loss": 0.0729,
      "step": 2662
    },
    {
      "epoch": 0.6059158134243459,
      "grad_norm": 0.7777911730162685,
      "learning_rate": 1.2052842834941791e-06,
      "loss": 0.0351,
      "step": 2663
    },
    {
      "epoch": 0.6061433447098976,
      "grad_norm": 0.9791113605180757,
      "learning_rate": 1.2052510884370274e-06,
      "loss": 0.0447,
      "step": 2664
    },
    {
      "epoch": 0.6063708759954494,
      "grad_norm": 0.967866940301246,
      "learning_rate": 1.2052178815206064e-06,
      "loss": 0.0371,
      "step": 2665
    },
    {
      "epoch": 0.6065984072810011,
      "grad_norm": 1.534725370801602,
      "learning_rate": 1.2051846627455946e-06,
      "loss": 0.0826,
      "step": 2666
    },
    {
      "epoch": 0.6068259385665529,
      "grad_norm": 0.9691101488135158,
      "learning_rate": 1.2051514321126705e-06,
      "loss": 0.0486,
      "step": 2667
    },
    {
      "epoch": 0.6070534698521046,
      "grad_norm": 1.352223067560777,
      "learning_rate": 1.2051181896225139e-06,
      "loss": 0.0496,
      "step": 2668
    },
    {
      "epoch": 0.6072810011376564,
      "grad_norm": 0.9959104730289088,
      "learning_rate": 1.205084935275804e-06,
      "loss": 0.0458,
      "step": 2669
    },
    {
      "epoch": 0.6075085324232082,
      "grad_norm": 1.8958304200279301,
      "learning_rate": 1.20505166907322e-06,
      "loss": 0.0702,
      "step": 2670
    },
    {
      "epoch": 0.60773606370876,
      "grad_norm": 1.1373672856837178,
      "learning_rate": 1.2050183910154425e-06,
      "loss": 0.0642,
      "step": 2671
    },
    {
      "epoch": 0.6079635949943117,
      "grad_norm": 1.5146568276899794,
      "learning_rate": 1.2049851011031514e-06,
      "loss": 0.0615,
      "step": 2672
    },
    {
      "epoch": 0.6081911262798635,
      "grad_norm": 1.3486666030528254,
      "learning_rate": 1.2049517993370269e-06,
      "loss": 0.0407,
      "step": 2673
    },
    {
      "epoch": 0.6084186575654152,
      "grad_norm": 1.7553759733095198,
      "learning_rate": 1.2049184857177498e-06,
      "loss": 0.089,
      "step": 2674
    },
    {
      "epoch": 0.608646188850967,
      "grad_norm": 1.6341921000073825,
      "learning_rate": 1.204885160246001e-06,
      "loss": 0.1034,
      "step": 2675
    },
    {
      "epoch": 0.6088737201365187,
      "grad_norm": 1.2886908765685168,
      "learning_rate": 1.2048518229224613e-06,
      "loss": 0.0371,
      "step": 2676
    },
    {
      "epoch": 0.6091012514220705,
      "grad_norm": 1.147794266112336,
      "learning_rate": 1.2048184737478124e-06,
      "loss": 0.0513,
      "step": 2677
    },
    {
      "epoch": 0.6093287827076223,
      "grad_norm": 1.108848937036823,
      "learning_rate": 1.2047851127227358e-06,
      "loss": 0.0372,
      "step": 2678
    },
    {
      "epoch": 0.6095563139931741,
      "grad_norm": 0.9494871582849728,
      "learning_rate": 1.2047517398479135e-06,
      "loss": 0.0313,
      "step": 2679
    },
    {
      "epoch": 0.6097838452787259,
      "grad_norm": 1.4403996402965795,
      "learning_rate": 1.204718355124027e-06,
      "loss": 0.0648,
      "step": 2680
    },
    {
      "epoch": 0.6100113765642776,
      "grad_norm": 1.1467814441502786,
      "learning_rate": 1.2046849585517595e-06,
      "loss": 0.0363,
      "step": 2681
    },
    {
      "epoch": 0.6102389078498294,
      "grad_norm": 0.8769285585937754,
      "learning_rate": 1.2046515501317927e-06,
      "loss": 0.0439,
      "step": 2682
    },
    {
      "epoch": 0.6104664391353811,
      "grad_norm": 1.0907507673380405,
      "learning_rate": 1.2046181298648101e-06,
      "loss": 0.0362,
      "step": 2683
    },
    {
      "epoch": 0.6106939704209329,
      "grad_norm": 1.4400540200529313,
      "learning_rate": 1.2045846977514943e-06,
      "loss": 0.0695,
      "step": 2684
    },
    {
      "epoch": 0.6109215017064846,
      "grad_norm": 0.926017693420237,
      "learning_rate": 1.2045512537925287e-06,
      "loss": 0.0319,
      "step": 2685
    },
    {
      "epoch": 0.6111490329920364,
      "grad_norm": 1.9422364837523665,
      "learning_rate": 1.2045177979885969e-06,
      "loss": 0.0678,
      "step": 2686
    },
    {
      "epoch": 0.6113765642775881,
      "grad_norm": 1.0780377835715493,
      "learning_rate": 1.2044843303403827e-06,
      "loss": 0.051,
      "step": 2687
    },
    {
      "epoch": 0.6116040955631399,
      "grad_norm": 1.522022597658668,
      "learning_rate": 1.20445085084857e-06,
      "loss": 0.0917,
      "step": 2688
    },
    {
      "epoch": 0.6118316268486917,
      "grad_norm": 1.0239752154710169,
      "learning_rate": 1.204417359513843e-06,
      "loss": 0.0286,
      "step": 2689
    },
    {
      "epoch": 0.6120591581342435,
      "grad_norm": 1.150678969827508,
      "learning_rate": 1.2043838563368865e-06,
      "loss": 0.0554,
      "step": 2690
    },
    {
      "epoch": 0.6122866894197952,
      "grad_norm": 1.2280896133913035,
      "learning_rate": 1.204350341318385e-06,
      "loss": 0.0562,
      "step": 2691
    },
    {
      "epoch": 0.612514220705347,
      "grad_norm": 1.0374592547169612,
      "learning_rate": 1.2043168144590237e-06,
      "loss": 0.0483,
      "step": 2692
    },
    {
      "epoch": 0.6127417519908988,
      "grad_norm": 1.8353693399206994,
      "learning_rate": 1.2042832757594875e-06,
      "loss": 0.0901,
      "step": 2693
    },
    {
      "epoch": 0.6129692832764505,
      "grad_norm": 1.6915158293911816,
      "learning_rate": 1.204249725220462e-06,
      "loss": 0.0741,
      "step": 2694
    },
    {
      "epoch": 0.6131968145620023,
      "grad_norm": 1.18563782209136,
      "learning_rate": 1.204216162842633e-06,
      "loss": 0.0406,
      "step": 2695
    },
    {
      "epoch": 0.613424345847554,
      "grad_norm": 1.2961769715929403,
      "learning_rate": 1.2041825886266866e-06,
      "loss": 0.0449,
      "step": 2696
    },
    {
      "epoch": 0.6136518771331058,
      "grad_norm": 1.5168600612694154,
      "learning_rate": 1.2041490025733089e-06,
      "loss": 0.0853,
      "step": 2697
    },
    {
      "epoch": 0.6138794084186575,
      "grad_norm": 1.0552035992692197,
      "learning_rate": 1.2041154046831859e-06,
      "loss": 0.0459,
      "step": 2698
    },
    {
      "epoch": 0.6141069397042094,
      "grad_norm": 1.0052589171106947,
      "learning_rate": 1.2040817949570046e-06,
      "loss": 0.0519,
      "step": 2699
    },
    {
      "epoch": 0.6143344709897611,
      "grad_norm": 1.5847978638614886,
      "learning_rate": 1.2040481733954523e-06,
      "loss": 0.0489,
      "step": 2700
    },
    {
      "epoch": 0.6145620022753129,
      "grad_norm": 1.5433588402630525,
      "learning_rate": 1.2040145399992157e-06,
      "loss": 0.0554,
      "step": 2701
    },
    {
      "epoch": 0.6147895335608646,
      "grad_norm": 0.9199676231545262,
      "learning_rate": 1.203980894768982e-06,
      "loss": 0.051,
      "step": 2702
    },
    {
      "epoch": 0.6150170648464164,
      "grad_norm": 1.6984060525727276,
      "learning_rate": 1.2039472377054395e-06,
      "loss": 0.0951,
      "step": 2703
    },
    {
      "epoch": 0.6152445961319681,
      "grad_norm": 1.9254054628589912,
      "learning_rate": 1.2039135688092757e-06,
      "loss": 0.0614,
      "step": 2704
    },
    {
      "epoch": 0.6154721274175199,
      "grad_norm": 1.3832191004617072,
      "learning_rate": 1.203879888081179e-06,
      "loss": 0.0666,
      "step": 2705
    },
    {
      "epoch": 0.6156996587030716,
      "grad_norm": 1.915733750626152,
      "learning_rate": 1.203846195521837e-06,
      "loss": 0.0622,
      "step": 2706
    },
    {
      "epoch": 0.6159271899886234,
      "grad_norm": 1.1800428967306633,
      "learning_rate": 1.2038124911319393e-06,
      "loss": 0.0428,
      "step": 2707
    },
    {
      "epoch": 0.6161547212741753,
      "grad_norm": 1.464495992586222,
      "learning_rate": 1.2037787749121741e-06,
      "loss": 0.069,
      "step": 2708
    },
    {
      "epoch": 0.616382252559727,
      "grad_norm": 1.1760191210505098,
      "learning_rate": 1.203745046863231e-06,
      "loss": 0.0503,
      "step": 2709
    },
    {
      "epoch": 0.6166097838452788,
      "grad_norm": 1.2442850515401647,
      "learning_rate": 1.203711306985799e-06,
      "loss": 0.04,
      "step": 2710
    },
    {
      "epoch": 0.6168373151308305,
      "grad_norm": 0.865019588027552,
      "learning_rate": 1.2036775552805674e-06,
      "loss": 0.0294,
      "step": 2711
    },
    {
      "epoch": 0.6170648464163823,
      "grad_norm": 2.6756015679269978,
      "learning_rate": 1.2036437917482267e-06,
      "loss": 0.0814,
      "step": 2712
    },
    {
      "epoch": 0.617292377701934,
      "grad_norm": 0.8893160007809519,
      "learning_rate": 1.2036100163894665e-06,
      "loss": 0.0434,
      "step": 2713
    },
    {
      "epoch": 0.6175199089874858,
      "grad_norm": 1.5121328039736281,
      "learning_rate": 1.2035762292049772e-06,
      "loss": 0.0848,
      "step": 2714
    },
    {
      "epoch": 0.6177474402730375,
      "grad_norm": 1.1556104934545364,
      "learning_rate": 1.2035424301954496e-06,
      "loss": 0.0564,
      "step": 2715
    },
    {
      "epoch": 0.6179749715585893,
      "grad_norm": 1.344567899518094,
      "learning_rate": 1.2035086193615743e-06,
      "loss": 0.0449,
      "step": 2716
    },
    {
      "epoch": 0.618202502844141,
      "grad_norm": 1.6415013868308683,
      "learning_rate": 1.203474796704042e-06,
      "loss": 0.0598,
      "step": 2717
    },
    {
      "epoch": 0.6184300341296929,
      "grad_norm": 0.9852915602376644,
      "learning_rate": 1.2034409622235444e-06,
      "loss": 0.0285,
      "step": 2718
    },
    {
      "epoch": 0.6186575654152446,
      "grad_norm": 1.1301434136989037,
      "learning_rate": 1.203407115920773e-06,
      "loss": 0.0473,
      "step": 2719
    },
    {
      "epoch": 0.6188850967007964,
      "grad_norm": 1.3150440764406408,
      "learning_rate": 1.2033732577964194e-06,
      "loss": 0.0717,
      "step": 2720
    },
    {
      "epoch": 0.6191126279863481,
      "grad_norm": 1.4850057002676174,
      "learning_rate": 1.2033393878511756e-06,
      "loss": 0.0589,
      "step": 2721
    },
    {
      "epoch": 0.6193401592718999,
      "grad_norm": 1.5995015548370441,
      "learning_rate": 1.203305506085734e-06,
      "loss": 0.0705,
      "step": 2722
    },
    {
      "epoch": 0.6195676905574516,
      "grad_norm": 0.8404790504075466,
      "learning_rate": 1.2032716125007868e-06,
      "loss": 0.0437,
      "step": 2723
    },
    {
      "epoch": 0.6197952218430034,
      "grad_norm": 1.3772954832300879,
      "learning_rate": 1.2032377070970268e-06,
      "loss": 0.0318,
      "step": 2724
    },
    {
      "epoch": 0.6200227531285551,
      "grad_norm": 1.4515256482801804,
      "learning_rate": 1.2032037898751475e-06,
      "loss": 0.0473,
      "step": 2725
    },
    {
      "epoch": 0.6202502844141069,
      "grad_norm": 0.7545612524043191,
      "learning_rate": 1.2031698608358414e-06,
      "loss": 0.0269,
      "step": 2726
    },
    {
      "epoch": 0.6204778156996587,
      "grad_norm": 0.743007367054589,
      "learning_rate": 1.2031359199798021e-06,
      "loss": 0.0283,
      "step": 2727
    },
    {
      "epoch": 0.6207053469852105,
      "grad_norm": 1.2718184804954529,
      "learning_rate": 1.2031019673077237e-06,
      "loss": 0.0634,
      "step": 2728
    },
    {
      "epoch": 0.6209328782707623,
      "grad_norm": 1.775193564914863,
      "learning_rate": 1.2030680028202995e-06,
      "loss": 0.1335,
      "step": 2729
    },
    {
      "epoch": 0.621160409556314,
      "grad_norm": 1.1547031507192722,
      "learning_rate": 1.2030340265182242e-06,
      "loss": 0.026,
      "step": 2730
    },
    {
      "epoch": 0.6213879408418658,
      "grad_norm": 1.81101747698367,
      "learning_rate": 1.2030000384021919e-06,
      "loss": 0.1018,
      "step": 2731
    },
    {
      "epoch": 0.6216154721274175,
      "grad_norm": 1.5054245916089843,
      "learning_rate": 1.2029660384728974e-06,
      "loss": 0.0495,
      "step": 2732
    },
    {
      "epoch": 0.6218430034129693,
      "grad_norm": 1.6121127283946548,
      "learning_rate": 1.2029320267310359e-06,
      "loss": 0.0782,
      "step": 2733
    },
    {
      "epoch": 0.622070534698521,
      "grad_norm": 1.6042883442076186,
      "learning_rate": 1.2028980031773018e-06,
      "loss": 0.0589,
      "step": 2734
    },
    {
      "epoch": 0.6222980659840728,
      "grad_norm": 1.2540922996339912,
      "learning_rate": 1.202863967812391e-06,
      "loss": 0.0774,
      "step": 2735
    },
    {
      "epoch": 0.6225255972696245,
      "grad_norm": 1.518635659663246,
      "learning_rate": 1.2028299206369991e-06,
      "loss": 0.1072,
      "step": 2736
    },
    {
      "epoch": 0.6227531285551763,
      "grad_norm": 1.3308844577317762,
      "learning_rate": 1.2027958616518218e-06,
      "loss": 0.0445,
      "step": 2737
    },
    {
      "epoch": 0.6229806598407281,
      "grad_norm": 2.209735756502863,
      "learning_rate": 1.2027617908575553e-06,
      "loss": 0.1124,
      "step": 2738
    },
    {
      "epoch": 0.6232081911262799,
      "grad_norm": 1.3929457690467317,
      "learning_rate": 1.2027277082548958e-06,
      "loss": 0.0545,
      "step": 2739
    },
    {
      "epoch": 0.6234357224118316,
      "grad_norm": 1.051968724737917,
      "learning_rate": 1.20269361384454e-06,
      "loss": 0.033,
      "step": 2740
    },
    {
      "epoch": 0.6236632536973834,
      "grad_norm": 1.6764017429869924,
      "learning_rate": 1.2026595076271848e-06,
      "loss": 0.0834,
      "step": 2741
    },
    {
      "epoch": 0.6238907849829352,
      "grad_norm": 1.6397713261851863,
      "learning_rate": 1.2026253896035273e-06,
      "loss": 0.0693,
      "step": 2742
    },
    {
      "epoch": 0.6241183162684869,
      "grad_norm": 1.591510561765584,
      "learning_rate": 1.2025912597742646e-06,
      "loss": 0.0768,
      "step": 2743
    },
    {
      "epoch": 0.6243458475540387,
      "grad_norm": 1.5008432457963459,
      "learning_rate": 1.2025571181400944e-06,
      "loss": 0.0771,
      "step": 2744
    },
    {
      "epoch": 0.6245733788395904,
      "grad_norm": 1.362876261935557,
      "learning_rate": 1.2025229647017145e-06,
      "loss": 0.0523,
      "step": 2745
    },
    {
      "epoch": 0.6248009101251422,
      "grad_norm": 2.2117120292707977,
      "learning_rate": 1.2024887994598227e-06,
      "loss": 0.0941,
      "step": 2746
    },
    {
      "epoch": 0.625028441410694,
      "grad_norm": 0.962824420377618,
      "learning_rate": 1.2024546224151176e-06,
      "loss": 0.0339,
      "step": 2747
    },
    {
      "epoch": 0.6252559726962458,
      "grad_norm": 0.8088610019406368,
      "learning_rate": 1.2024204335682977e-06,
      "loss": 0.0269,
      "step": 2748
    },
    {
      "epoch": 0.6254835039817975,
      "grad_norm": 2.415791185781217,
      "learning_rate": 1.2023862329200613e-06,
      "loss": 0.0925,
      "step": 2749
    },
    {
      "epoch": 0.6257110352673493,
      "grad_norm": 1.0450350621113902,
      "learning_rate": 1.2023520204711078e-06,
      "loss": 0.0539,
      "step": 2750
    },
    {
      "epoch": 0.625938566552901,
      "grad_norm": 1.3226699575436216,
      "learning_rate": 1.2023177962221366e-06,
      "loss": 0.098,
      "step": 2751
    },
    {
      "epoch": 0.6261660978384528,
      "grad_norm": 1.6169539390839551,
      "learning_rate": 1.2022835601738467e-06,
      "loss": 0.0674,
      "step": 2752
    },
    {
      "epoch": 0.6263936291240045,
      "grad_norm": 0.926101104299797,
      "learning_rate": 1.2022493123269383e-06,
      "loss": 0.0326,
      "step": 2753
    },
    {
      "epoch": 0.6266211604095563,
      "grad_norm": 1.7687331075293982,
      "learning_rate": 1.202215052682111e-06,
      "loss": 0.1007,
      "step": 2754
    },
    {
      "epoch": 0.626848691695108,
      "grad_norm": 1.2611712167362608,
      "learning_rate": 1.2021807812400652e-06,
      "loss": 0.0739,
      "step": 2755
    },
    {
      "epoch": 0.6270762229806598,
      "grad_norm": 1.5800647590282382,
      "learning_rate": 1.2021464980015014e-06,
      "loss": 0.078,
      "step": 2756
    },
    {
      "epoch": 0.6273037542662117,
      "grad_norm": 1.2594753097389422,
      "learning_rate": 1.20211220296712e-06,
      "loss": 0.0642,
      "step": 2757
    },
    {
      "epoch": 0.6275312855517634,
      "grad_norm": 1.121495104083563,
      "learning_rate": 1.2020778961376223e-06,
      "loss": 0.0328,
      "step": 2758
    },
    {
      "epoch": 0.6277588168373152,
      "grad_norm": 1.4683080872027747,
      "learning_rate": 1.202043577513709e-06,
      "loss": 0.0902,
      "step": 2759
    },
    {
      "epoch": 0.6279863481228669,
      "grad_norm": 1.8245567164154206,
      "learning_rate": 1.202009247096082e-06,
      "loss": 0.0761,
      "step": 2760
    },
    {
      "epoch": 0.6282138794084187,
      "grad_norm": 1.1125392914595493,
      "learning_rate": 1.2019749048854426e-06,
      "loss": 0.0646,
      "step": 2761
    },
    {
      "epoch": 0.6284414106939704,
      "grad_norm": 0.8642533280519576,
      "learning_rate": 1.2019405508824927e-06,
      "loss": 0.0324,
      "step": 2762
    },
    {
      "epoch": 0.6286689419795222,
      "grad_norm": 1.7625633245035697,
      "learning_rate": 1.201906185087935e-06,
      "loss": 0.0776,
      "step": 2763
    },
    {
      "epoch": 0.6288964732650739,
      "grad_norm": 1.9920964212571328,
      "learning_rate": 1.201871807502471e-06,
      "loss": 0.0628,
      "step": 2764
    },
    {
      "epoch": 0.6291240045506257,
      "grad_norm": 1.371538527134396,
      "learning_rate": 1.2018374181268039e-06,
      "loss": 0.054,
      "step": 2765
    },
    {
      "epoch": 0.6293515358361774,
      "grad_norm": 1.603991934514247,
      "learning_rate": 1.2018030169616363e-06,
      "loss": 0.0675,
      "step": 2766
    },
    {
      "epoch": 0.6295790671217293,
      "grad_norm": 1.3082298269977088,
      "learning_rate": 1.2017686040076715e-06,
      "loss": 0.059,
      "step": 2767
    },
    {
      "epoch": 0.629806598407281,
      "grad_norm": 1.0629082940323715,
      "learning_rate": 1.2017341792656129e-06,
      "loss": 0.0526,
      "step": 2768
    },
    {
      "epoch": 0.6300341296928328,
      "grad_norm": 1.5246818248480305,
      "learning_rate": 1.2016997427361635e-06,
      "loss": 0.0489,
      "step": 2769
    },
    {
      "epoch": 0.6302616609783845,
      "grad_norm": 1.4969549226937944,
      "learning_rate": 1.201665294420028e-06,
      "loss": 0.079,
      "step": 2770
    },
    {
      "epoch": 0.6304891922639363,
      "grad_norm": 1.195617244868444,
      "learning_rate": 1.2016308343179098e-06,
      "loss": 0.038,
      "step": 2771
    },
    {
      "epoch": 0.630716723549488,
      "grad_norm": 1.0734726876278198,
      "learning_rate": 1.2015963624305132e-06,
      "loss": 0.0528,
      "step": 2772
    },
    {
      "epoch": 0.6309442548350398,
      "grad_norm": 0.9060818686627535,
      "learning_rate": 1.201561878758543e-06,
      "loss": 0.0321,
      "step": 2773
    },
    {
      "epoch": 0.6311717861205915,
      "grad_norm": 1.9200212018830873,
      "learning_rate": 1.2015273833027041e-06,
      "loss": 0.0827,
      "step": 2774
    },
    {
      "epoch": 0.6313993174061433,
      "grad_norm": 1.1822527617116831,
      "learning_rate": 1.2014928760637014e-06,
      "loss": 0.0537,
      "step": 2775
    },
    {
      "epoch": 0.631626848691695,
      "grad_norm": 1.618374665019338,
      "learning_rate": 1.20145835704224e-06,
      "loss": 0.0563,
      "step": 2776
    },
    {
      "epoch": 0.6318543799772469,
      "grad_norm": 2.134594638963389,
      "learning_rate": 1.2014238262390254e-06,
      "loss": 0.0709,
      "step": 2777
    },
    {
      "epoch": 0.6320819112627987,
      "grad_norm": 2.1175666116051968,
      "learning_rate": 1.2013892836547635e-06,
      "loss": 0.0924,
      "step": 2778
    },
    {
      "epoch": 0.6323094425483504,
      "grad_norm": 1.5175276686793735,
      "learning_rate": 1.2013547292901605e-06,
      "loss": 0.0818,
      "step": 2779
    },
    {
      "epoch": 0.6325369738339022,
      "grad_norm": 1.1839298832479785,
      "learning_rate": 1.2013201631459222e-06,
      "loss": 0.0709,
      "step": 2780
    },
    {
      "epoch": 0.6327645051194539,
      "grad_norm": 2.288412160525939,
      "learning_rate": 1.201285585222755e-06,
      "loss": 0.105,
      "step": 2781
    },
    {
      "epoch": 0.6329920364050057,
      "grad_norm": 1.569225058207049,
      "learning_rate": 1.2012509955213664e-06,
      "loss": 0.0477,
      "step": 2782
    },
    {
      "epoch": 0.6332195676905574,
      "grad_norm": 0.7422297833198885,
      "learning_rate": 1.2012163940424624e-06,
      "loss": 0.0359,
      "step": 2783
    },
    {
      "epoch": 0.6334470989761092,
      "grad_norm": 1.009892072927878,
      "learning_rate": 1.2011817807867507e-06,
      "loss": 0.0472,
      "step": 2784
    },
    {
      "epoch": 0.6336746302616609,
      "grad_norm": 1.8711194663634534,
      "learning_rate": 1.2011471557549387e-06,
      "loss": 0.0621,
      "step": 2785
    },
    {
      "epoch": 0.6339021615472128,
      "grad_norm": 1.0981859377257484,
      "learning_rate": 1.2011125189477339e-06,
      "loss": 0.0407,
      "step": 2786
    },
    {
      "epoch": 0.6341296928327645,
      "grad_norm": 2.3667510812808903,
      "learning_rate": 1.2010778703658441e-06,
      "loss": 0.103,
      "step": 2787
    },
    {
      "epoch": 0.6343572241183163,
      "grad_norm": 1.4344646107903631,
      "learning_rate": 1.2010432100099781e-06,
      "loss": 0.0796,
      "step": 2788
    },
    {
      "epoch": 0.634584755403868,
      "grad_norm": 1.6112614680461406,
      "learning_rate": 1.2010085378808437e-06,
      "loss": 0.0635,
      "step": 2789
    },
    {
      "epoch": 0.6348122866894198,
      "grad_norm": 1.8958308802132453,
      "learning_rate": 1.2009738539791497e-06,
      "loss": 0.1027,
      "step": 2790
    },
    {
      "epoch": 0.6350398179749716,
      "grad_norm": 1.4317662705377017,
      "learning_rate": 1.2009391583056048e-06,
      "loss": 0.0689,
      "step": 2791
    },
    {
      "epoch": 0.6352673492605233,
      "grad_norm": 1.2007483338261875,
      "learning_rate": 1.2009044508609182e-06,
      "loss": 0.0584,
      "step": 2792
    },
    {
      "epoch": 0.6354948805460751,
      "grad_norm": 1.9010319599598573,
      "learning_rate": 1.2008697316457997e-06,
      "loss": 0.0966,
      "step": 2793
    },
    {
      "epoch": 0.6357224118316268,
      "grad_norm": 0.9068372322632697,
      "learning_rate": 1.2008350006609584e-06,
      "loss": 0.0319,
      "step": 2794
    },
    {
      "epoch": 0.6359499431171786,
      "grad_norm": 1.9186704917037192,
      "learning_rate": 1.2008002579071043e-06,
      "loss": 0.0834,
      "step": 2795
    },
    {
      "epoch": 0.6361774744027304,
      "grad_norm": 1.5719062287923162,
      "learning_rate": 1.2007655033849474e-06,
      "loss": 0.0719,
      "step": 2796
    },
    {
      "epoch": 0.6364050056882822,
      "grad_norm": 0.8675484564571867,
      "learning_rate": 1.2007307370951983e-06,
      "loss": 0.0382,
      "step": 2797
    },
    {
      "epoch": 0.6366325369738339,
      "grad_norm": 1.2487471267893597,
      "learning_rate": 1.200695959038567e-06,
      "loss": 0.0541,
      "step": 2798
    },
    {
      "epoch": 0.6368600682593857,
      "grad_norm": 1.340853119074798,
      "learning_rate": 1.2006611692157648e-06,
      "loss": 0.0625,
      "step": 2799
    },
    {
      "epoch": 0.6370875995449374,
      "grad_norm": 1.0090275039380872,
      "learning_rate": 1.2006263676275026e-06,
      "loss": 0.0386,
      "step": 2800
    },
    {
      "epoch": 0.6373151308304892,
      "grad_norm": 1.3604638479008277,
      "learning_rate": 1.2005915542744915e-06,
      "loss": 0.0579,
      "step": 2801
    },
    {
      "epoch": 0.6375426621160409,
      "grad_norm": 1.3626136251906178,
      "learning_rate": 1.2005567291574434e-06,
      "loss": 0.0506,
      "step": 2802
    },
    {
      "epoch": 0.6377701934015927,
      "grad_norm": 2.24330408064307,
      "learning_rate": 1.2005218922770695e-06,
      "loss": 0.0934,
      "step": 2803
    },
    {
      "epoch": 0.6379977246871444,
      "grad_norm": 1.652449071090047,
      "learning_rate": 1.2004870436340824e-06,
      "loss": 0.0916,
      "step": 2804
    },
    {
      "epoch": 0.6382252559726962,
      "grad_norm": 1.335757404355839,
      "learning_rate": 1.2004521832291943e-06,
      "loss": 0.0319,
      "step": 2805
    },
    {
      "epoch": 0.6384527872582481,
      "grad_norm": 1.2669579033763387,
      "learning_rate": 1.200417311063117e-06,
      "loss": 0.0452,
      "step": 2806
    },
    {
      "epoch": 0.6386803185437998,
      "grad_norm": 1.3480996457776724,
      "learning_rate": 1.200382427136564e-06,
      "loss": 0.0664,
      "step": 2807
    },
    {
      "epoch": 0.6389078498293516,
      "grad_norm": 1.6036199545645866,
      "learning_rate": 1.2003475314502477e-06,
      "loss": 0.0715,
      "step": 2808
    },
    {
      "epoch": 0.6391353811149033,
      "grad_norm": 1.0938499350027229,
      "learning_rate": 1.200312624004882e-06,
      "loss": 0.0467,
      "step": 2809
    },
    {
      "epoch": 0.6393629124004551,
      "grad_norm": 1.301558947668022,
      "learning_rate": 1.2002777048011794e-06,
      "loss": 0.0611,
      "step": 2810
    },
    {
      "epoch": 0.6395904436860068,
      "grad_norm": 1.2950309349789517,
      "learning_rate": 1.2002427738398543e-06,
      "loss": 0.0501,
      "step": 2811
    },
    {
      "epoch": 0.6398179749715586,
      "grad_norm": 1.074515099435565,
      "learning_rate": 1.2002078311216205e-06,
      "loss": 0.0273,
      "step": 2812
    },
    {
      "epoch": 0.6400455062571103,
      "grad_norm": 0.9534944533034787,
      "learning_rate": 1.2001728766471919e-06,
      "loss": 0.0517,
      "step": 2813
    },
    {
      "epoch": 0.6402730375426621,
      "grad_norm": 1.5158088143964041,
      "learning_rate": 1.2001379104172832e-06,
      "loss": 0.0638,
      "step": 2814
    },
    {
      "epoch": 0.6405005688282139,
      "grad_norm": 1.247966598668176,
      "learning_rate": 1.2001029324326087e-06,
      "loss": 0.054,
      "step": 2815
    },
    {
      "epoch": 0.6407281001137657,
      "grad_norm": 0.9514285262896064,
      "learning_rate": 1.2000679426938838e-06,
      "loss": 0.0324,
      "step": 2816
    },
    {
      "epoch": 0.6409556313993174,
      "grad_norm": 1.5357576363213208,
      "learning_rate": 1.2000329412018233e-06,
      "loss": 0.0504,
      "step": 2817
    },
    {
      "epoch": 0.6411831626848692,
      "grad_norm": 1.3766532946507657,
      "learning_rate": 1.1999979279571425e-06,
      "loss": 0.0785,
      "step": 2818
    },
    {
      "epoch": 0.6414106939704209,
      "grad_norm": 1.1854802730452816,
      "learning_rate": 1.1999629029605572e-06,
      "loss": 0.0471,
      "step": 2819
    },
    {
      "epoch": 0.6416382252559727,
      "grad_norm": 0.9120829198047485,
      "learning_rate": 1.1999278662127832e-06,
      "loss": 0.0465,
      "step": 2820
    },
    {
      "epoch": 0.6418657565415244,
      "grad_norm": 1.215725225022476,
      "learning_rate": 1.1998928177145363e-06,
      "loss": 0.0539,
      "step": 2821
    },
    {
      "epoch": 0.6420932878270762,
      "grad_norm": 1.6055605292211583,
      "learning_rate": 1.1998577574665334e-06,
      "loss": 0.0794,
      "step": 2822
    },
    {
      "epoch": 0.642320819112628,
      "grad_norm": 1.5194681984361975,
      "learning_rate": 1.1998226854694906e-06,
      "loss": 0.0775,
      "step": 2823
    },
    {
      "epoch": 0.6425483503981797,
      "grad_norm": 1.217305433486093,
      "learning_rate": 1.1997876017241248e-06,
      "loss": 0.0363,
      "step": 2824
    },
    {
      "epoch": 0.6427758816837316,
      "grad_norm": 1.2554857295036752,
      "learning_rate": 1.199752506231153e-06,
      "loss": 0.0451,
      "step": 2825
    },
    {
      "epoch": 0.6430034129692833,
      "grad_norm": 1.4903034342688748,
      "learning_rate": 1.1997173989912928e-06,
      "loss": 0.0651,
      "step": 2826
    },
    {
      "epoch": 0.6432309442548351,
      "grad_norm": 0.9479490152328633,
      "learning_rate": 1.1996822800052614e-06,
      "loss": 0.0416,
      "step": 2827
    },
    {
      "epoch": 0.6434584755403868,
      "grad_norm": 1.3630232299101448,
      "learning_rate": 1.1996471492737767e-06,
      "loss": 0.053,
      "step": 2828
    },
    {
      "epoch": 0.6436860068259386,
      "grad_norm": 0.8310504363214145,
      "learning_rate": 1.1996120067975568e-06,
      "loss": 0.0401,
      "step": 2829
    },
    {
      "epoch": 0.6439135381114903,
      "grad_norm": 0.7817607162647777,
      "learning_rate": 1.1995768525773195e-06,
      "loss": 0.0381,
      "step": 2830
    },
    {
      "epoch": 0.6441410693970421,
      "grad_norm": 1.19151900025748,
      "learning_rate": 1.1995416866137837e-06,
      "loss": 0.0401,
      "step": 2831
    },
    {
      "epoch": 0.6443686006825938,
      "grad_norm": 0.8480287553862164,
      "learning_rate": 1.1995065089076682e-06,
      "loss": 0.0427,
      "step": 2832
    },
    {
      "epoch": 0.6445961319681456,
      "grad_norm": 1.3823956321024167,
      "learning_rate": 1.199471319459692e-06,
      "loss": 0.0678,
      "step": 2833
    },
    {
      "epoch": 0.6448236632536973,
      "grad_norm": 1.0881726943732215,
      "learning_rate": 1.199436118270574e-06,
      "loss": 0.0463,
      "step": 2834
    },
    {
      "epoch": 0.6450511945392492,
      "grad_norm": 0.9603443906924007,
      "learning_rate": 1.1994009053410336e-06,
      "loss": 0.0334,
      "step": 2835
    },
    {
      "epoch": 0.645278725824801,
      "grad_norm": 0.7257173068914297,
      "learning_rate": 1.1993656806717906e-06,
      "loss": 0.022,
      "step": 2836
    },
    {
      "epoch": 0.6455062571103527,
      "grad_norm": 1.3460255228704814,
      "learning_rate": 1.199330444263565e-06,
      "loss": 0.0589,
      "step": 2837
    },
    {
      "epoch": 0.6457337883959045,
      "grad_norm": 1.8598440039360982,
      "learning_rate": 1.1992951961170774e-06,
      "loss": 0.0884,
      "step": 2838
    },
    {
      "epoch": 0.6459613196814562,
      "grad_norm": 1.7631231436296406,
      "learning_rate": 1.1992599362330474e-06,
      "loss": 0.0691,
      "step": 2839
    },
    {
      "epoch": 0.646188850967008,
      "grad_norm": 1.0081908118372565,
      "learning_rate": 1.199224664612196e-06,
      "loss": 0.046,
      "step": 2840
    },
    {
      "epoch": 0.6464163822525597,
      "grad_norm": 1.5846986206428932,
      "learning_rate": 1.199189381255244e-06,
      "loss": 0.0819,
      "step": 2841
    },
    {
      "epoch": 0.6466439135381115,
      "grad_norm": 1.3965515241420483,
      "learning_rate": 1.199154086162913e-06,
      "loss": 0.0675,
      "step": 2842
    },
    {
      "epoch": 0.6468714448236632,
      "grad_norm": 1.7830102963673449,
      "learning_rate": 1.1991187793359239e-06,
      "loss": 0.1173,
      "step": 2843
    },
    {
      "epoch": 0.647098976109215,
      "grad_norm": 0.7979390598594543,
      "learning_rate": 1.1990834607749981e-06,
      "loss": 0.0288,
      "step": 2844
    },
    {
      "epoch": 0.6473265073947668,
      "grad_norm": 1.2345551184177335,
      "learning_rate": 1.199048130480858e-06,
      "loss": 0.0823,
      "step": 2845
    },
    {
      "epoch": 0.6475540386803186,
      "grad_norm": 1.3892698568903075,
      "learning_rate": 1.1990127884542251e-06,
      "loss": 0.0739,
      "step": 2846
    },
    {
      "epoch": 0.6477815699658703,
      "grad_norm": 1.5536465235111332,
      "learning_rate": 1.1989774346958225e-06,
      "loss": 0.0725,
      "step": 2847
    },
    {
      "epoch": 0.6480091012514221,
      "grad_norm": 1.482520572829848,
      "learning_rate": 1.1989420692063723e-06,
      "loss": 0.0902,
      "step": 2848
    },
    {
      "epoch": 0.6482366325369738,
      "grad_norm": 1.0973672503057945,
      "learning_rate": 1.198906691986597e-06,
      "loss": 0.0567,
      "step": 2849
    },
    {
      "epoch": 0.6484641638225256,
      "grad_norm": 1.974255381506691,
      "learning_rate": 1.1988713030372202e-06,
      "loss": 0.1019,
      "step": 2850
    },
    {
      "epoch": 0.6486916951080773,
      "grad_norm": 0.9822063316024513,
      "learning_rate": 1.198835902358965e-06,
      "loss": 0.0556,
      "step": 2851
    },
    {
      "epoch": 0.6489192263936291,
      "grad_norm": 0.9883294090430303,
      "learning_rate": 1.1988004899525547e-06,
      "loss": 0.0511,
      "step": 2852
    },
    {
      "epoch": 0.6491467576791808,
      "grad_norm": 1.7242005203993112,
      "learning_rate": 1.1987650658187133e-06,
      "loss": 0.0684,
      "step": 2853
    },
    {
      "epoch": 0.6493742889647327,
      "grad_norm": 1.163292058133356,
      "learning_rate": 1.1987296299581648e-06,
      "loss": 0.055,
      "step": 2854
    },
    {
      "epoch": 0.6496018202502845,
      "grad_norm": 0.842300844455492,
      "learning_rate": 1.1986941823716333e-06,
      "loss": 0.0361,
      "step": 2855
    },
    {
      "epoch": 0.6498293515358362,
      "grad_norm": 0.7877836086747282,
      "learning_rate": 1.1986587230598437e-06,
      "loss": 0.0269,
      "step": 2856
    },
    {
      "epoch": 0.650056882821388,
      "grad_norm": 1.1854138002017856,
      "learning_rate": 1.19862325202352e-06,
      "loss": 0.0789,
      "step": 2857
    },
    {
      "epoch": 0.6502844141069397,
      "grad_norm": 1.5193193171125443,
      "learning_rate": 1.198587769263388e-06,
      "loss": 0.0647,
      "step": 2858
    },
    {
      "epoch": 0.6505119453924915,
      "grad_norm": 0.9471414553218402,
      "learning_rate": 1.198552274780172e-06,
      "loss": 0.0394,
      "step": 2859
    },
    {
      "epoch": 0.6507394766780432,
      "grad_norm": 1.142561403387133,
      "learning_rate": 1.1985167685745982e-06,
      "loss": 0.0427,
      "step": 2860
    },
    {
      "epoch": 0.650967007963595,
      "grad_norm": 1.163361004688848,
      "learning_rate": 1.198481250647392e-06,
      "loss": 0.0583,
      "step": 2861
    },
    {
      "epoch": 0.6511945392491467,
      "grad_norm": 1.2956201317841325,
      "learning_rate": 1.1984457209992792e-06,
      "loss": 0.0497,
      "step": 2862
    },
    {
      "epoch": 0.6514220705346985,
      "grad_norm": 1.0871624809389135,
      "learning_rate": 1.1984101796309862e-06,
      "loss": 0.0719,
      "step": 2863
    },
    {
      "epoch": 0.6516496018202503,
      "grad_norm": 1.3146485801211771,
      "learning_rate": 1.1983746265432392e-06,
      "loss": 0.0589,
      "step": 2864
    },
    {
      "epoch": 0.6518771331058021,
      "grad_norm": 1.5681925610927387,
      "learning_rate": 1.1983390617367649e-06,
      "loss": 0.0701,
      "step": 2865
    },
    {
      "epoch": 0.6521046643913538,
      "grad_norm": 2.2916879452331926,
      "learning_rate": 1.1983034852122902e-06,
      "loss": 0.1033,
      "step": 2866
    },
    {
      "epoch": 0.6523321956769056,
      "grad_norm": 1.0785758663467682,
      "learning_rate": 1.1982678969705425e-06,
      "loss": 0.0264,
      "step": 2867
    },
    {
      "epoch": 0.6525597269624573,
      "grad_norm": 1.8613288950042697,
      "learning_rate": 1.1982322970122485e-06,
      "loss": 0.066,
      "step": 2868
    },
    {
      "epoch": 0.6527872582480091,
      "grad_norm": 1.1933202650904684,
      "learning_rate": 1.1981966853381364e-06,
      "loss": 0.0553,
      "step": 2869
    },
    {
      "epoch": 0.6530147895335608,
      "grad_norm": 0.9553057216020039,
      "learning_rate": 1.1981610619489337e-06,
      "loss": 0.0391,
      "step": 2870
    },
    {
      "epoch": 0.6532423208191126,
      "grad_norm": 1.1836315703365476,
      "learning_rate": 1.1981254268453684e-06,
      "loss": 0.054,
      "step": 2871
    },
    {
      "epoch": 0.6534698521046644,
      "grad_norm": 1.276765336289362,
      "learning_rate": 1.1980897800281694e-06,
      "loss": 0.0585,
      "step": 2872
    },
    {
      "epoch": 0.6536973833902161,
      "grad_norm": 1.5829433588482489,
      "learning_rate": 1.1980541214980646e-06,
      "loss": 0.0811,
      "step": 2873
    },
    {
      "epoch": 0.653924914675768,
      "grad_norm": 2.2299790037798854,
      "learning_rate": 1.1980184512557833e-06,
      "loss": 0.0838,
      "step": 2874
    },
    {
      "epoch": 0.6541524459613197,
      "grad_norm": 1.8246918071788611,
      "learning_rate": 1.1979827693020541e-06,
      "loss": 0.0778,
      "step": 2875
    },
    {
      "epoch": 0.6543799772468715,
      "grad_norm": 0.8023113638584155,
      "learning_rate": 1.1979470756376064e-06,
      "loss": 0.0374,
      "step": 2876
    },
    {
      "epoch": 0.6546075085324232,
      "grad_norm": 1.0665996787354195,
      "learning_rate": 1.1979113702631697e-06,
      "loss": 0.0642,
      "step": 2877
    },
    {
      "epoch": 0.654835039817975,
      "grad_norm": 1.0607142907030562,
      "learning_rate": 1.197875653179474e-06,
      "loss": 0.0379,
      "step": 2878
    },
    {
      "epoch": 0.6550625711035267,
      "grad_norm": 1.509330774335647,
      "learning_rate": 1.1978399243872492e-06,
      "loss": 0.0429,
      "step": 2879
    },
    {
      "epoch": 0.6552901023890785,
      "grad_norm": 1.1080169650246614,
      "learning_rate": 1.1978041838872253e-06,
      "loss": 0.041,
      "step": 2880
    },
    {
      "epoch": 0.6555176336746302,
      "grad_norm": 0.8616722922724181,
      "learning_rate": 1.197768431680133e-06,
      "loss": 0.0308,
      "step": 2881
    },
    {
      "epoch": 0.655745164960182,
      "grad_norm": 1.625281473798501,
      "learning_rate": 1.197732667766703e-06,
      "loss": 0.0732,
      "step": 2882
    },
    {
      "epoch": 0.6559726962457337,
      "grad_norm": 2.0203110844795056,
      "learning_rate": 1.1976968921476662e-06,
      "loss": 0.0885,
      "step": 2883
    },
    {
      "epoch": 0.6562002275312856,
      "grad_norm": 1.325667144252201,
      "learning_rate": 1.1976611048237534e-06,
      "loss": 0.0471,
      "step": 2884
    },
    {
      "epoch": 0.6564277588168373,
      "grad_norm": 1.108094823287875,
      "learning_rate": 1.1976253057956968e-06,
      "loss": 0.0486,
      "step": 2885
    },
    {
      "epoch": 0.6566552901023891,
      "grad_norm": 1.5070679596895156,
      "learning_rate": 1.1975894950642276e-06,
      "loss": 0.0609,
      "step": 2886
    },
    {
      "epoch": 0.6568828213879409,
      "grad_norm": 1.8747406123784733,
      "learning_rate": 1.1975536726300776e-06,
      "loss": 0.084,
      "step": 2887
    },
    {
      "epoch": 0.6571103526734926,
      "grad_norm": 1.0968771480268504,
      "learning_rate": 1.1975178384939793e-06,
      "loss": 0.0518,
      "step": 2888
    },
    {
      "epoch": 0.6573378839590444,
      "grad_norm": 1.2149776100702219,
      "learning_rate": 1.197481992656665e-06,
      "loss": 0.0663,
      "step": 2889
    },
    {
      "epoch": 0.6575654152445961,
      "grad_norm": 1.0219972948941853,
      "learning_rate": 1.1974461351188668e-06,
      "loss": 0.0421,
      "step": 2890
    },
    {
      "epoch": 0.6577929465301479,
      "grad_norm": 1.0668851298436286,
      "learning_rate": 1.1974102658813183e-06,
      "loss": 0.0372,
      "step": 2891
    },
    {
      "epoch": 0.6580204778156996,
      "grad_norm": 0.9280161130235592,
      "learning_rate": 1.1973743849447522e-06,
      "loss": 0.041,
      "step": 2892
    },
    {
      "epoch": 0.6582480091012515,
      "grad_norm": 1.5811942617888033,
      "learning_rate": 1.197338492309902e-06,
      "loss": 0.0644,
      "step": 2893
    },
    {
      "epoch": 0.6584755403868032,
      "grad_norm": 0.9301441187543878,
      "learning_rate": 1.1973025879775011e-06,
      "loss": 0.0452,
      "step": 2894
    },
    {
      "epoch": 0.658703071672355,
      "grad_norm": 1.4309181679010228,
      "learning_rate": 1.1972666719482833e-06,
      "loss": 0.0594,
      "step": 2895
    },
    {
      "epoch": 0.6589306029579067,
      "grad_norm": 1.2896695199681876,
      "learning_rate": 1.197230744222983e-06,
      "loss": 0.0462,
      "step": 2896
    },
    {
      "epoch": 0.6591581342434585,
      "grad_norm": 1.3825152304918809,
      "learning_rate": 1.1971948048023343e-06,
      "loss": 0.0645,
      "step": 2897
    },
    {
      "epoch": 0.6593856655290102,
      "grad_norm": 1.5004332728601224,
      "learning_rate": 1.1971588536870717e-06,
      "loss": 0.0709,
      "step": 2898
    },
    {
      "epoch": 0.659613196814562,
      "grad_norm": 1.4884045373736343,
      "learning_rate": 1.19712289087793e-06,
      "loss": 0.0779,
      "step": 2899
    },
    {
      "epoch": 0.6598407281001137,
      "grad_norm": 1.8129968316744782,
      "learning_rate": 1.197086916375644e-06,
      "loss": 0.0848,
      "step": 2900
    },
    {
      "epoch": 0.6600682593856655,
      "grad_norm": 0.6798855950472126,
      "learning_rate": 1.1970509301809493e-06,
      "loss": 0.0182,
      "step": 2901
    },
    {
      "epoch": 0.6602957906712172,
      "grad_norm": 1.3632444561301862,
      "learning_rate": 1.1970149322945812e-06,
      "loss": 0.0885,
      "step": 2902
    },
    {
      "epoch": 0.6605233219567691,
      "grad_norm": 1.1160426689933205,
      "learning_rate": 1.1969789227172755e-06,
      "loss": 0.0467,
      "step": 2903
    },
    {
      "epoch": 0.6607508532423209,
      "grad_norm": 0.9677303090234676,
      "learning_rate": 1.1969429014497684e-06,
      "loss": 0.0257,
      "step": 2904
    },
    {
      "epoch": 0.6609783845278726,
      "grad_norm": 17.353739180619908,
      "learning_rate": 1.1969068684927954e-06,
      "loss": 0.0818,
      "step": 2905
    },
    {
      "epoch": 0.6612059158134244,
      "grad_norm": 1.2792498241135908,
      "learning_rate": 1.1968708238470936e-06,
      "loss": 0.058,
      "step": 2906
    },
    {
      "epoch": 0.6614334470989761,
      "grad_norm": 1.4555868612084297,
      "learning_rate": 1.1968347675133995e-06,
      "loss": 0.0562,
      "step": 2907
    },
    {
      "epoch": 0.6616609783845279,
      "grad_norm": 1.5729405842398636,
      "learning_rate": 1.19679869949245e-06,
      "loss": 0.0901,
      "step": 2908
    },
    {
      "epoch": 0.6618885096700796,
      "grad_norm": 2.205720994640489,
      "learning_rate": 1.1967626197849824e-06,
      "loss": 0.0712,
      "step": 2909
    },
    {
      "epoch": 0.6621160409556314,
      "grad_norm": 2.2706627078782478,
      "learning_rate": 1.1967265283917339e-06,
      "loss": 0.1286,
      "step": 2910
    },
    {
      "epoch": 0.6623435722411831,
      "grad_norm": 1.3583089823132908,
      "learning_rate": 1.1966904253134422e-06,
      "loss": 0.051,
      "step": 2911
    },
    {
      "epoch": 0.6625711035267349,
      "grad_norm": 1.2040227987956196,
      "learning_rate": 1.1966543105508454e-06,
      "loss": 0.0303,
      "step": 2912
    },
    {
      "epoch": 0.6627986348122867,
      "grad_norm": 1.8898241749294495,
      "learning_rate": 1.1966181841046812e-06,
      "loss": 0.1013,
      "step": 2913
    },
    {
      "epoch": 0.6630261660978385,
      "grad_norm": 1.1212373105622575,
      "learning_rate": 1.1965820459756882e-06,
      "loss": 0.0423,
      "step": 2914
    },
    {
      "epoch": 0.6632536973833902,
      "grad_norm": 1.4942787777207431,
      "learning_rate": 1.1965458961646051e-06,
      "loss": 0.074,
      "step": 2915
    },
    {
      "epoch": 0.663481228668942,
      "grad_norm": 1.2554067286772215,
      "learning_rate": 1.1965097346721707e-06,
      "loss": 0.0569,
      "step": 2916
    },
    {
      "epoch": 0.6637087599544937,
      "grad_norm": 2.064257552726771,
      "learning_rate": 1.1964735614991237e-06,
      "loss": 0.1222,
      "step": 2917
    },
    {
      "epoch": 0.6639362912400455,
      "grad_norm": 1.7127511151118642,
      "learning_rate": 1.196437376646204e-06,
      "loss": 0.059,
      "step": 2918
    },
    {
      "epoch": 0.6641638225255972,
      "grad_norm": 1.747734548367446,
      "learning_rate": 1.1964011801141505e-06,
      "loss": 0.0699,
      "step": 2919
    },
    {
      "epoch": 0.664391353811149,
      "grad_norm": 1.7260170721527037,
      "learning_rate": 1.1963649719037037e-06,
      "loss": 0.0682,
      "step": 2920
    },
    {
      "epoch": 0.6646188850967008,
      "grad_norm": 1.0412792846070889,
      "learning_rate": 1.196328752015603e-06,
      "loss": 0.0469,
      "step": 2921
    },
    {
      "epoch": 0.6648464163822526,
      "grad_norm": 1.0652258289806211,
      "learning_rate": 1.1962925204505894e-06,
      "loss": 0.0292,
      "step": 2922
    },
    {
      "epoch": 0.6650739476678044,
      "grad_norm": 1.05165136662707,
      "learning_rate": 1.1962562772094024e-06,
      "loss": 0.0707,
      "step": 2923
    },
    {
      "epoch": 0.6653014789533561,
      "grad_norm": 0.877541367497127,
      "learning_rate": 1.1962200222927836e-06,
      "loss": 0.0382,
      "step": 2924
    },
    {
      "epoch": 0.6655290102389079,
      "grad_norm": 1.0400570459306038,
      "learning_rate": 1.1961837557014736e-06,
      "loss": 0.0334,
      "step": 2925
    },
    {
      "epoch": 0.6657565415244596,
      "grad_norm": 1.6730636365537126,
      "learning_rate": 1.196147477436214e-06,
      "loss": 0.1133,
      "step": 2926
    },
    {
      "epoch": 0.6659840728100114,
      "grad_norm": 0.9388002444280849,
      "learning_rate": 1.1961111874977455e-06,
      "loss": 0.0425,
      "step": 2927
    },
    {
      "epoch": 0.6662116040955631,
      "grad_norm": 1.2971959782913498,
      "learning_rate": 1.1960748858868104e-06,
      "loss": 0.0598,
      "step": 2928
    },
    {
      "epoch": 0.6664391353811149,
      "grad_norm": 1.6257500523277735,
      "learning_rate": 1.1960385726041507e-06,
      "loss": 0.0773,
      "step": 2929
    },
    {
      "epoch": 0.6666666666666666,
      "grad_norm": 1.113728651042982,
      "learning_rate": 1.1960022476505082e-06,
      "loss": 0.0406,
      "step": 2930
    },
    {
      "epoch": 0.6668941979522184,
      "grad_norm": 0.8908601808254871,
      "learning_rate": 1.1959659110266256e-06,
      "loss": 0.0392,
      "step": 2931
    },
    {
      "epoch": 0.6671217292377702,
      "grad_norm": 1.6390262789672483,
      "learning_rate": 1.1959295627332454e-06,
      "loss": 0.0888,
      "step": 2932
    },
    {
      "epoch": 0.667349260523322,
      "grad_norm": 1.4346808601596435,
      "learning_rate": 1.1958932027711106e-06,
      "loss": 0.0534,
      "step": 2933
    },
    {
      "epoch": 0.6675767918088737,
      "grad_norm": 0.9348135708089949,
      "learning_rate": 1.1958568311409643e-06,
      "loss": 0.0472,
      "step": 2934
    },
    {
      "epoch": 0.6678043230944255,
      "grad_norm": 1.159822973816305,
      "learning_rate": 1.1958204478435497e-06,
      "loss": 0.0602,
      "step": 2935
    },
    {
      "epoch": 0.6680318543799773,
      "grad_norm": 1.0013241030005304,
      "learning_rate": 1.195784052879611e-06,
      "loss": 0.0347,
      "step": 2936
    },
    {
      "epoch": 0.668259385665529,
      "grad_norm": 0.7712539814391252,
      "learning_rate": 1.195747646249891e-06,
      "loss": 0.0238,
      "step": 2937
    },
    {
      "epoch": 0.6684869169510808,
      "grad_norm": 1.8913673239815798,
      "learning_rate": 1.1957112279551347e-06,
      "loss": 0.0967,
      "step": 2938
    },
    {
      "epoch": 0.6687144482366325,
      "grad_norm": 1.3385349237767592,
      "learning_rate": 1.195674797996086e-06,
      "loss": 0.0452,
      "step": 2939
    },
    {
      "epoch": 0.6689419795221843,
      "grad_norm": 1.8314127610766537,
      "learning_rate": 1.1956383563734897e-06,
      "loss": 0.0888,
      "step": 2940
    },
    {
      "epoch": 0.669169510807736,
      "grad_norm": 1.4820883842924408,
      "learning_rate": 1.1956019030880902e-06,
      "loss": 0.0629,
      "step": 2941
    },
    {
      "epoch": 0.6693970420932879,
      "grad_norm": 0.9377075991001586,
      "learning_rate": 1.1955654381406331e-06,
      "loss": 0.0541,
      "step": 2942
    },
    {
      "epoch": 0.6696245733788396,
      "grad_norm": 1.1625402940924978,
      "learning_rate": 1.1955289615318632e-06,
      "loss": 0.0583,
      "step": 2943
    },
    {
      "epoch": 0.6698521046643914,
      "grad_norm": 0.9043356889096931,
      "learning_rate": 1.1954924732625264e-06,
      "loss": 0.0386,
      "step": 2944
    },
    {
      "epoch": 0.6700796359499431,
      "grad_norm": 1.0364482997261704,
      "learning_rate": 1.1954559733333681e-06,
      "loss": 0.0353,
      "step": 2945
    },
    {
      "epoch": 0.6703071672354949,
      "grad_norm": 1.6526326184491673,
      "learning_rate": 1.1954194617451345e-06,
      "loss": 0.0531,
      "step": 2946
    },
    {
      "epoch": 0.6705346985210466,
      "grad_norm": 0.9766619653449208,
      "learning_rate": 1.1953829384985716e-06,
      "loss": 0.0391,
      "step": 2947
    },
    {
      "epoch": 0.6707622298065984,
      "grad_norm": 2.163216907401221,
      "learning_rate": 1.1953464035944262e-06,
      "loss": 0.0835,
      "step": 2948
    },
    {
      "epoch": 0.6709897610921501,
      "grad_norm": 1.4844803459994427,
      "learning_rate": 1.195309857033445e-06,
      "loss": 0.0842,
      "step": 2949
    },
    {
      "epoch": 0.6712172923777019,
      "grad_norm": 0.6644024917990391,
      "learning_rate": 1.1952732988163745e-06,
      "loss": 0.0347,
      "step": 2950
    },
    {
      "epoch": 0.6714448236632536,
      "grad_norm": 1.3392424005542798,
      "learning_rate": 1.1952367289439624e-06,
      "loss": 0.0593,
      "step": 2951
    },
    {
      "epoch": 0.6716723549488055,
      "grad_norm": 1.5693502580306737,
      "learning_rate": 1.1952001474169558e-06,
      "loss": 0.0723,
      "step": 2952
    },
    {
      "epoch": 0.6718998862343573,
      "grad_norm": 1.3882537221447417,
      "learning_rate": 1.1951635542361025e-06,
      "loss": 0.0482,
      "step": 2953
    },
    {
      "epoch": 0.672127417519909,
      "grad_norm": 1.6367400788566016,
      "learning_rate": 1.1951269494021503e-06,
      "loss": 0.0541,
      "step": 2954
    },
    {
      "epoch": 0.6723549488054608,
      "grad_norm": 1.2264958404782558,
      "learning_rate": 1.1950903329158475e-06,
      "loss": 0.0543,
      "step": 2955
    },
    {
      "epoch": 0.6725824800910125,
      "grad_norm": 0.9638518358259208,
      "learning_rate": 1.1950537047779424e-06,
      "loss": 0.0425,
      "step": 2956
    },
    {
      "epoch": 0.6728100113765643,
      "grad_norm": 0.7797808668666366,
      "learning_rate": 1.1950170649891836e-06,
      "loss": 0.0266,
      "step": 2957
    },
    {
      "epoch": 0.673037542662116,
      "grad_norm": 1.256662868306741,
      "learning_rate": 1.1949804135503196e-06,
      "loss": 0.0762,
      "step": 2958
    },
    {
      "epoch": 0.6732650739476678,
      "grad_norm": 1.0816193366991107,
      "learning_rate": 1.1949437504621e-06,
      "loss": 0.047,
      "step": 2959
    },
    {
      "epoch": 0.6734926052332195,
      "grad_norm": 1.5164857137673318,
      "learning_rate": 1.194907075725274e-06,
      "loss": 0.0995,
      "step": 2960
    },
    {
      "epoch": 0.6737201365187714,
      "grad_norm": 1.1360766280459287,
      "learning_rate": 1.1948703893405911e-06,
      "loss": 0.0427,
      "step": 2961
    },
    {
      "epoch": 0.6739476678043231,
      "grad_norm": 1.342451075159084,
      "learning_rate": 1.194833691308801e-06,
      "loss": 0.0422,
      "step": 2962
    },
    {
      "epoch": 0.6741751990898749,
      "grad_norm": 1.7047601826121286,
      "learning_rate": 1.194796981630654e-06,
      "loss": 0.0532,
      "step": 2963
    },
    {
      "epoch": 0.6744027303754266,
      "grad_norm": 1.3289514422481006,
      "learning_rate": 1.1947602603069002e-06,
      "loss": 0.0648,
      "step": 2964
    },
    {
      "epoch": 0.6746302616609784,
      "grad_norm": 1.8203278250275807,
      "learning_rate": 1.19472352733829e-06,
      "loss": 0.0591,
      "step": 2965
    },
    {
      "epoch": 0.6748577929465301,
      "grad_norm": 0.9130861662390535,
      "learning_rate": 1.1946867827255744e-06,
      "loss": 0.0392,
      "step": 2966
    },
    {
      "epoch": 0.6750853242320819,
      "grad_norm": 2.1792488178961866,
      "learning_rate": 1.1946500264695044e-06,
      "loss": 0.0852,
      "step": 2967
    },
    {
      "epoch": 0.6753128555176336,
      "grad_norm": 1.3430961956554572,
      "learning_rate": 1.194613258570831e-06,
      "loss": 0.054,
      "step": 2968
    },
    {
      "epoch": 0.6755403868031854,
      "grad_norm": 1.0194337724675724,
      "learning_rate": 1.1945764790303059e-06,
      "loss": 0.0577,
      "step": 2969
    },
    {
      "epoch": 0.6757679180887372,
      "grad_norm": 1.414677659719635,
      "learning_rate": 1.1945396878486805e-06,
      "loss": 0.0575,
      "step": 2970
    },
    {
      "epoch": 0.675995449374289,
      "grad_norm": 1.3881918329356409,
      "learning_rate": 1.194502885026707e-06,
      "loss": 0.07,
      "step": 2971
    },
    {
      "epoch": 0.6762229806598408,
      "grad_norm": 0.8742447812198808,
      "learning_rate": 1.1944660705651375e-06,
      "loss": 0.0545,
      "step": 2972
    },
    {
      "epoch": 0.6764505119453925,
      "grad_norm": 0.9000949345293963,
      "learning_rate": 1.1944292444647248e-06,
      "loss": 0.0465,
      "step": 2973
    },
    {
      "epoch": 0.6766780432309443,
      "grad_norm": 1.0764234435889581,
      "learning_rate": 1.1943924067262208e-06,
      "loss": 0.0432,
      "step": 2974
    },
    {
      "epoch": 0.676905574516496,
      "grad_norm": 1.249115475491158,
      "learning_rate": 1.194355557350379e-06,
      "loss": 0.0548,
      "step": 2975
    },
    {
      "epoch": 0.6771331058020478,
      "grad_norm": 1.067373727393787,
      "learning_rate": 1.1943186963379522e-06,
      "loss": 0.0528,
      "step": 2976
    },
    {
      "epoch": 0.6773606370875995,
      "grad_norm": 1.5676460836455215,
      "learning_rate": 1.194281823689694e-06,
      "loss": 0.0672,
      "step": 2977
    },
    {
      "epoch": 0.6775881683731513,
      "grad_norm": 1.334005489011124,
      "learning_rate": 1.194244939406358e-06,
      "loss": 0.0565,
      "step": 2978
    },
    {
      "epoch": 0.677815699658703,
      "grad_norm": 1.324440667565256,
      "learning_rate": 1.1942080434886978e-06,
      "loss": 0.0488,
      "step": 2979
    },
    {
      "epoch": 0.6780432309442548,
      "grad_norm": 1.155754728183158,
      "learning_rate": 1.1941711359374678e-06,
      "loss": 0.0622,
      "step": 2980
    },
    {
      "epoch": 0.6782707622298066,
      "grad_norm": 0.7220306306891887,
      "learning_rate": 1.194134216753422e-06,
      "loss": 0.0363,
      "step": 2981
    },
    {
      "epoch": 0.6784982935153584,
      "grad_norm": 2.259664003454956,
      "learning_rate": 1.1940972859373151e-06,
      "loss": 0.084,
      "step": 2982
    },
    {
      "epoch": 0.6787258248009101,
      "grad_norm": 0.9209026025756932,
      "learning_rate": 1.194060343489902e-06,
      "loss": 0.0247,
      "step": 2983
    },
    {
      "epoch": 0.6789533560864619,
      "grad_norm": 2.237931268641456,
      "learning_rate": 1.1940233894119377e-06,
      "loss": 0.1003,
      "step": 2984
    },
    {
      "epoch": 0.6791808873720137,
      "grad_norm": 1.45745994111441,
      "learning_rate": 1.1939864237041774e-06,
      "loss": 0.0656,
      "step": 2985
    },
    {
      "epoch": 0.6794084186575654,
      "grad_norm": 0.9550126427103799,
      "learning_rate": 1.1939494463673767e-06,
      "loss": 0.0442,
      "step": 2986
    },
    {
      "epoch": 0.6796359499431172,
      "grad_norm": 1.9833777656739968,
      "learning_rate": 1.1939124574022914e-06,
      "loss": 0.0904,
      "step": 2987
    },
    {
      "epoch": 0.6798634812286689,
      "grad_norm": 1.3661977813438515,
      "learning_rate": 1.1938754568096771e-06,
      "loss": 0.0644,
      "step": 2988
    },
    {
      "epoch": 0.6800910125142207,
      "grad_norm": 0.8616871036933312,
      "learning_rate": 1.1938384445902905e-06,
      "loss": 0.0393,
      "step": 2989
    },
    {
      "epoch": 0.6803185437997725,
      "grad_norm": 1.1588349083327987,
      "learning_rate": 1.1938014207448877e-06,
      "loss": 0.047,
      "step": 2990
    },
    {
      "epoch": 0.6805460750853243,
      "grad_norm": 1.0841199268841493,
      "learning_rate": 1.1937643852742258e-06,
      "loss": 0.0315,
      "step": 2991
    },
    {
      "epoch": 0.680773606370876,
      "grad_norm": 4.593658878524584,
      "learning_rate": 1.1937273381790615e-06,
      "loss": 0.1461,
      "step": 2992
    },
    {
      "epoch": 0.6810011376564278,
      "grad_norm": 1.1490732435329922,
      "learning_rate": 1.1936902794601518e-06,
      "loss": 0.0479,
      "step": 2993
    },
    {
      "epoch": 0.6812286689419795,
      "grad_norm": 1.450896272579947,
      "learning_rate": 1.1936532091182544e-06,
      "loss": 0.0613,
      "step": 2994
    },
    {
      "epoch": 0.6814562002275313,
      "grad_norm": 1.2205030830111818,
      "learning_rate": 1.1936161271541268e-06,
      "loss": 0.0626,
      "step": 2995
    },
    {
      "epoch": 0.681683731513083,
      "grad_norm": 1.1653942327121265,
      "learning_rate": 1.1935790335685272e-06,
      "loss": 0.0464,
      "step": 2996
    },
    {
      "epoch": 0.6819112627986348,
      "grad_norm": 1.9153169444858846,
      "learning_rate": 1.193541928362213e-06,
      "loss": 0.0732,
      "step": 2997
    },
    {
      "epoch": 0.6821387940841865,
      "grad_norm": 1.4815686264003924,
      "learning_rate": 1.1935048115359432e-06,
      "loss": 0.0638,
      "step": 2998
    },
    {
      "epoch": 0.6823663253697383,
      "grad_norm": 1.6915818417968453,
      "learning_rate": 1.1934676830904763e-06,
      "loss": 0.0773,
      "step": 2999
    },
    {
      "epoch": 0.6825938566552902,
      "grad_norm": 1.3182256509758368,
      "learning_rate": 1.193430543026571e-06,
      "loss": 0.0627,
      "step": 3000
    },
    {
      "epoch": 0.6828213879408419,
      "grad_norm": 1.6374773596627972,
      "learning_rate": 1.1933933913449867e-06,
      "loss": 0.1202,
      "step": 3001
    },
    {
      "epoch": 0.6830489192263937,
      "grad_norm": 1.6225368015279966,
      "learning_rate": 1.193356228046482e-06,
      "loss": 0.1016,
      "step": 3002
    },
    {
      "epoch": 0.6832764505119454,
      "grad_norm": 1.158897733977453,
      "learning_rate": 1.1933190531318172e-06,
      "loss": 0.0455,
      "step": 3003
    },
    {
      "epoch": 0.6835039817974972,
      "grad_norm": 1.2180725170363247,
      "learning_rate": 1.1932818666017516e-06,
      "loss": 0.0335,
      "step": 3004
    },
    {
      "epoch": 0.6837315130830489,
      "grad_norm": 1.5088158398690277,
      "learning_rate": 1.1932446684570455e-06,
      "loss": 0.0478,
      "step": 3005
    },
    {
      "epoch": 0.6839590443686007,
      "grad_norm": 1.0577989164533361,
      "learning_rate": 1.1932074586984592e-06,
      "loss": 0.0411,
      "step": 3006
    },
    {
      "epoch": 0.6841865756541524,
      "grad_norm": 1.265620028127131,
      "learning_rate": 1.1931702373267527e-06,
      "loss": 0.0373,
      "step": 3007
    },
    {
      "epoch": 0.6844141069397042,
      "grad_norm": 1.015589756317499,
      "learning_rate": 1.1931330043426872e-06,
      "loss": 0.0701,
      "step": 3008
    },
    {
      "epoch": 0.6846416382252559,
      "grad_norm": 0.9468195016052887,
      "learning_rate": 1.1930957597470238e-06,
      "loss": 0.0502,
      "step": 3009
    },
    {
      "epoch": 0.6848691695108078,
      "grad_norm": 1.4788516581774798,
      "learning_rate": 1.1930585035405235e-06,
      "loss": 0.0579,
      "step": 3010
    },
    {
      "epoch": 0.6850967007963595,
      "grad_norm": 1.5755558972332642,
      "learning_rate": 1.1930212357239475e-06,
      "loss": 0.0525,
      "step": 3011
    },
    {
      "epoch": 0.6853242320819113,
      "grad_norm": 1.2998868209788172,
      "learning_rate": 1.1929839562980579e-06,
      "loss": 0.0585,
      "step": 3012
    },
    {
      "epoch": 0.685551763367463,
      "grad_norm": 1.5356635666634295,
      "learning_rate": 1.1929466652636164e-06,
      "loss": 0.0604,
      "step": 3013
    },
    {
      "epoch": 0.6857792946530148,
      "grad_norm": 2.0893811231577524,
      "learning_rate": 1.1929093626213852e-06,
      "loss": 0.1015,
      "step": 3014
    },
    {
      "epoch": 0.6860068259385665,
      "grad_norm": 0.5321529547637093,
      "learning_rate": 1.1928720483721269e-06,
      "loss": 0.0139,
      "step": 3015
    },
    {
      "epoch": 0.6862343572241183,
      "grad_norm": 1.0580052416688368,
      "learning_rate": 1.1928347225166035e-06,
      "loss": 0.0499,
      "step": 3016
    },
    {
      "epoch": 0.68646188850967,
      "grad_norm": 1.3320645495655943,
      "learning_rate": 1.1927973850555785e-06,
      "loss": 0.0465,
      "step": 3017
    },
    {
      "epoch": 0.6866894197952218,
      "grad_norm": 1.3194332531036985,
      "learning_rate": 1.192760035989815e-06,
      "loss": 0.0489,
      "step": 3018
    },
    {
      "epoch": 0.6869169510807736,
      "grad_norm": 1.7873894152489311,
      "learning_rate": 1.192722675320076e-06,
      "loss": 0.0683,
      "step": 3019
    },
    {
      "epoch": 0.6871444823663254,
      "grad_norm": 1.533578323181438,
      "learning_rate": 1.1926853030471253e-06,
      "loss": 0.0536,
      "step": 3020
    },
    {
      "epoch": 0.6873720136518772,
      "grad_norm": 0.9261434578357977,
      "learning_rate": 1.1926479191717267e-06,
      "loss": 0.0437,
      "step": 3021
    },
    {
      "epoch": 0.6875995449374289,
      "grad_norm": 1.1444794832316916,
      "learning_rate": 1.1926105236946443e-06,
      "loss": 0.0577,
      "step": 3022
    },
    {
      "epoch": 0.6878270762229807,
      "grad_norm": 1.335750162762178,
      "learning_rate": 1.192573116616642e-06,
      "loss": 0.0529,
      "step": 3023
    },
    {
      "epoch": 0.6880546075085324,
      "grad_norm": 1.434078699306857,
      "learning_rate": 1.192535697938485e-06,
      "loss": 0.0529,
      "step": 3024
    },
    {
      "epoch": 0.6882821387940842,
      "grad_norm": 1.3416490998400092,
      "learning_rate": 1.1924982676609377e-06,
      "loss": 0.074,
      "step": 3025
    },
    {
      "epoch": 0.6885096700796359,
      "grad_norm": 1.3232433001494412,
      "learning_rate": 1.1924608257847651e-06,
      "loss": 0.0451,
      "step": 3026
    },
    {
      "epoch": 0.6887372013651877,
      "grad_norm": 0.8764669981180951,
      "learning_rate": 1.1924233723107322e-06,
      "loss": 0.0484,
      "step": 3027
    },
    {
      "epoch": 0.6889647326507394,
      "grad_norm": 1.1655688024633921,
      "learning_rate": 1.1923859072396051e-06,
      "loss": 0.0606,
      "step": 3028
    },
    {
      "epoch": 0.6891922639362913,
      "grad_norm": 1.0179281217422282,
      "learning_rate": 1.1923484305721489e-06,
      "loss": 0.0466,
      "step": 3029
    },
    {
      "epoch": 0.689419795221843,
      "grad_norm": 0.8659456244009691,
      "learning_rate": 1.19231094230913e-06,
      "loss": 0.0404,
      "step": 3030
    },
    {
      "epoch": 0.6896473265073948,
      "grad_norm": 2.0636576063861565,
      "learning_rate": 1.1922734424513144e-06,
      "loss": 0.0784,
      "step": 3031
    },
    {
      "epoch": 0.6898748577929465,
      "grad_norm": 0.8695263646618282,
      "learning_rate": 1.1922359309994685e-06,
      "loss": 0.0277,
      "step": 3032
    },
    {
      "epoch": 0.6901023890784983,
      "grad_norm": 1.4190056652396834,
      "learning_rate": 1.192198407954359e-06,
      "loss": 0.0616,
      "step": 3033
    },
    {
      "epoch": 0.69032992036405,
      "grad_norm": 1.4753814874094142,
      "learning_rate": 1.192160873316753e-06,
      "loss": 0.0517,
      "step": 3034
    },
    {
      "epoch": 0.6905574516496018,
      "grad_norm": 1.5554050029090873,
      "learning_rate": 1.1921233270874174e-06,
      "loss": 0.0563,
      "step": 3035
    },
    {
      "epoch": 0.6907849829351536,
      "grad_norm": 0.7022912796986149,
      "learning_rate": 1.1920857692671196e-06,
      "loss": 0.0312,
      "step": 3036
    },
    {
      "epoch": 0.6910125142207053,
      "grad_norm": 2.0009143095316646,
      "learning_rate": 1.192048199856627e-06,
      "loss": 0.0994,
      "step": 3037
    },
    {
      "epoch": 0.6912400455062571,
      "grad_norm": 2.134476815905357,
      "learning_rate": 1.192010618856708e-06,
      "loss": 0.1024,
      "step": 3038
    },
    {
      "epoch": 0.6914675767918089,
      "grad_norm": 1.1849138073132004,
      "learning_rate": 1.1919730262681304e-06,
      "loss": 0.055,
      "step": 3039
    },
    {
      "epoch": 0.6916951080773607,
      "grad_norm": 1.845998643453397,
      "learning_rate": 1.1919354220916624e-06,
      "loss": 0.0983,
      "step": 3040
    },
    {
      "epoch": 0.6919226393629124,
      "grad_norm": 2.449072929938712,
      "learning_rate": 1.1918978063280726e-06,
      "loss": 0.0849,
      "step": 3041
    },
    {
      "epoch": 0.6921501706484642,
      "grad_norm": 0.8729079689297836,
      "learning_rate": 1.1918601789781299e-06,
      "loss": 0.0306,
      "step": 3042
    },
    {
      "epoch": 0.6923777019340159,
      "grad_norm": 1.780329225544261,
      "learning_rate": 1.1918225400426032e-06,
      "loss": 0.0689,
      "step": 3043
    },
    {
      "epoch": 0.6926052332195677,
      "grad_norm": 1.063152792810893,
      "learning_rate": 1.191784889522262e-06,
      "loss": 0.0408,
      "step": 3044
    },
    {
      "epoch": 0.6928327645051194,
      "grad_norm": 1.6901254212789465,
      "learning_rate": 1.1917472274178757e-06,
      "loss": 0.0846,
      "step": 3045
    },
    {
      "epoch": 0.6930602957906712,
      "grad_norm": 1.4002132139198793,
      "learning_rate": 1.191709553730214e-06,
      "loss": 0.0772,
      "step": 3046
    },
    {
      "epoch": 0.6932878270762229,
      "grad_norm": 1.5076734596253167,
      "learning_rate": 1.1916718684600469e-06,
      "loss": 0.0819,
      "step": 3047
    },
    {
      "epoch": 0.6935153583617747,
      "grad_norm": 1.0960152159965912,
      "learning_rate": 1.1916341716081446e-06,
      "loss": 0.049,
      "step": 3048
    },
    {
      "epoch": 0.6937428896473266,
      "grad_norm": 2.557391206512559,
      "learning_rate": 1.1915964631752775e-06,
      "loss": 0.0876,
      "step": 3049
    },
    {
      "epoch": 0.6939704209328783,
      "grad_norm": 0.920645510461067,
      "learning_rate": 1.1915587431622164e-06,
      "loss": 0.0307,
      "step": 3050
    },
    {
      "epoch": 0.6941979522184301,
      "grad_norm": 1.04173789253993,
      "learning_rate": 1.1915210115697324e-06,
      "loss": 0.0539,
      "step": 3051
    },
    {
      "epoch": 0.6944254835039818,
      "grad_norm": 1.6810913949726196,
      "learning_rate": 1.1914832683985962e-06,
      "loss": 0.0544,
      "step": 3052
    },
    {
      "epoch": 0.6946530147895336,
      "grad_norm": 1.850644903401837,
      "learning_rate": 1.1914455136495796e-06,
      "loss": 0.0899,
      "step": 3053
    },
    {
      "epoch": 0.6948805460750853,
      "grad_norm": 0.7506486967154027,
      "learning_rate": 1.191407747323454e-06,
      "loss": 0.0265,
      "step": 3054
    },
    {
      "epoch": 0.6951080773606371,
      "grad_norm": 2.1012986938124696,
      "learning_rate": 1.1913699694209914e-06,
      "loss": 0.1144,
      "step": 3055
    },
    {
      "epoch": 0.6953356086461888,
      "grad_norm": 1.2089722077198737,
      "learning_rate": 1.191332179942964e-06,
      "loss": 0.0609,
      "step": 3056
    },
    {
      "epoch": 0.6955631399317406,
      "grad_norm": 1.5324615273992814,
      "learning_rate": 1.1912943788901438e-06,
      "loss": 0.067,
      "step": 3057
    },
    {
      "epoch": 0.6957906712172923,
      "grad_norm": 1.4377403085013738,
      "learning_rate": 1.191256566263304e-06,
      "loss": 0.0841,
      "step": 3058
    },
    {
      "epoch": 0.6960182025028442,
      "grad_norm": 1.7028519956627337,
      "learning_rate": 1.1912187420632165e-06,
      "loss": 0.0796,
      "step": 3059
    },
    {
      "epoch": 0.6962457337883959,
      "grad_norm": 1.535145639508393,
      "learning_rate": 1.1911809062906552e-06,
      "loss": 0.0581,
      "step": 3060
    },
    {
      "epoch": 0.6964732650739477,
      "grad_norm": 1.3876603600761275,
      "learning_rate": 1.1911430589463931e-06,
      "loss": 0.0548,
      "step": 3061
    },
    {
      "epoch": 0.6967007963594994,
      "grad_norm": 2.3585378108380057,
      "learning_rate": 1.1911052000312038e-06,
      "loss": 0.1131,
      "step": 3062
    },
    {
      "epoch": 0.6969283276450512,
      "grad_norm": 1.224179370923149,
      "learning_rate": 1.1910673295458607e-06,
      "loss": 0.0733,
      "step": 3063
    },
    {
      "epoch": 0.697155858930603,
      "grad_norm": 1.080344689666754,
      "learning_rate": 1.1910294474911382e-06,
      "loss": 0.0339,
      "step": 3064
    },
    {
      "epoch": 0.6973833902161547,
      "grad_norm": 1.132793254903834,
      "learning_rate": 1.1909915538678105e-06,
      "loss": 0.0668,
      "step": 3065
    },
    {
      "epoch": 0.6976109215017064,
      "grad_norm": 0.9670480449446429,
      "learning_rate": 1.1909536486766522e-06,
      "loss": 0.0613,
      "step": 3066
    },
    {
      "epoch": 0.6978384527872582,
      "grad_norm": 0.8333929861107748,
      "learning_rate": 1.1909157319184373e-06,
      "loss": 0.0313,
      "step": 3067
    },
    {
      "epoch": 0.6980659840728101,
      "grad_norm": 1.5088053682741849,
      "learning_rate": 1.1908778035939416e-06,
      "loss": 0.0723,
      "step": 3068
    },
    {
      "epoch": 0.6982935153583618,
      "grad_norm": 2.496852043007627,
      "learning_rate": 1.1908398637039398e-06,
      "loss": 0.1006,
      "step": 3069
    },
    {
      "epoch": 0.6985210466439136,
      "grad_norm": 1.340716909671611,
      "learning_rate": 1.1908019122492077e-06,
      "loss": 0.0447,
      "step": 3070
    },
    {
      "epoch": 0.6987485779294653,
      "grad_norm": 2.0378089803817887,
      "learning_rate": 1.1907639492305205e-06,
      "loss": 0.0808,
      "step": 3071
    },
    {
      "epoch": 0.6989761092150171,
      "grad_norm": 0.7800931599177211,
      "learning_rate": 1.1907259746486547e-06,
      "loss": 0.0358,
      "step": 3072
    },
    {
      "epoch": 0.6992036405005688,
      "grad_norm": 0.9202110065520366,
      "learning_rate": 1.1906879885043856e-06,
      "loss": 0.0571,
      "step": 3073
    },
    {
      "epoch": 0.6994311717861206,
      "grad_norm": 0.7009452078498103,
      "learning_rate": 1.1906499907984903e-06,
      "loss": 0.0318,
      "step": 3074
    },
    {
      "epoch": 0.6996587030716723,
      "grad_norm": 1.1314308444671177,
      "learning_rate": 1.190611981531745e-06,
      "loss": 0.0446,
      "step": 3075
    },
    {
      "epoch": 0.6998862343572241,
      "grad_norm": 1.332217799137746,
      "learning_rate": 1.1905739607049267e-06,
      "loss": 0.0744,
      "step": 3076
    },
    {
      "epoch": 0.7001137656427758,
      "grad_norm": 0.9698337291959647,
      "learning_rate": 1.1905359283188126e-06,
      "loss": 0.0368,
      "step": 3077
    },
    {
      "epoch": 0.7003412969283277,
      "grad_norm": 0.5221877787168306,
      "learning_rate": 1.1904978843741796e-06,
      "loss": 0.0167,
      "step": 3078
    },
    {
      "epoch": 0.7005688282138794,
      "grad_norm": 1.4498948354718102,
      "learning_rate": 1.1904598288718055e-06,
      "loss": 0.0727,
      "step": 3079
    },
    {
      "epoch": 0.7007963594994312,
      "grad_norm": 1.5160712367792761,
      "learning_rate": 1.1904217618124684e-06,
      "loss": 0.0787,
      "step": 3080
    },
    {
      "epoch": 0.701023890784983,
      "grad_norm": 1.0604730462463068,
      "learning_rate": 1.1903836831969458e-06,
      "loss": 0.0639,
      "step": 3081
    },
    {
      "epoch": 0.7012514220705347,
      "grad_norm": 0.7275851025633676,
      "learning_rate": 1.190345593026016e-06,
      "loss": 0.0286,
      "step": 3082
    },
    {
      "epoch": 0.7014789533560865,
      "grad_norm": 1.4202168912069701,
      "learning_rate": 1.1903074913004579e-06,
      "loss": 0.0717,
      "step": 3083
    },
    {
      "epoch": 0.7017064846416382,
      "grad_norm": 1.6514017164531183,
      "learning_rate": 1.19026937802105e-06,
      "loss": 0.0787,
      "step": 3084
    },
    {
      "epoch": 0.70193401592719,
      "grad_norm": 0.6956539774170061,
      "learning_rate": 1.1902312531885712e-06,
      "loss": 0.0252,
      "step": 3085
    },
    {
      "epoch": 0.7021615472127417,
      "grad_norm": 1.5569684152870062,
      "learning_rate": 1.1901931168038007e-06,
      "loss": 0.0685,
      "step": 3086
    },
    {
      "epoch": 0.7023890784982935,
      "grad_norm": 1.2406976952968178,
      "learning_rate": 1.190154968867518e-06,
      "loss": 0.0454,
      "step": 3087
    },
    {
      "epoch": 0.7026166097838453,
      "grad_norm": 0.813921596018141,
      "learning_rate": 1.190116809380503e-06,
      "loss": 0.0253,
      "step": 3088
    },
    {
      "epoch": 0.7028441410693971,
      "grad_norm": 1.0275424113403453,
      "learning_rate": 1.190078638343535e-06,
      "loss": 0.0514,
      "step": 3089
    },
    {
      "epoch": 0.7030716723549488,
      "grad_norm": 1.0727110658094747,
      "learning_rate": 1.1900404557573948e-06,
      "loss": 0.0453,
      "step": 3090
    },
    {
      "epoch": 0.7032992036405006,
      "grad_norm": 1.6426665504516007,
      "learning_rate": 1.1900022616228624e-06,
      "loss": 0.0832,
      "step": 3091
    },
    {
      "epoch": 0.7035267349260523,
      "grad_norm": 1.9180762135067262,
      "learning_rate": 1.1899640559407186e-06,
      "loss": 0.0934,
      "step": 3092
    },
    {
      "epoch": 0.7037542662116041,
      "grad_norm": 1.7251637250688259,
      "learning_rate": 1.189925838711744e-06,
      "loss": 0.0744,
      "step": 3093
    },
    {
      "epoch": 0.7039817974971558,
      "grad_norm": 2.301009331337415,
      "learning_rate": 1.18988760993672e-06,
      "loss": 0.1061,
      "step": 3094
    },
    {
      "epoch": 0.7042093287827076,
      "grad_norm": 0.8730951635346749,
      "learning_rate": 1.1898493696164279e-06,
      "loss": 0.0251,
      "step": 3095
    },
    {
      "epoch": 0.7044368600682593,
      "grad_norm": 0.6226893588047131,
      "learning_rate": 1.1898111177516488e-06,
      "loss": 0.0396,
      "step": 3096
    },
    {
      "epoch": 0.7046643913538112,
      "grad_norm": 2.39998461661108,
      "learning_rate": 1.1897728543431653e-06,
      "loss": 0.1219,
      "step": 3097
    },
    {
      "epoch": 0.704891922639363,
      "grad_norm": 1.834814384170055,
      "learning_rate": 1.1897345793917589e-06,
      "loss": 0.0615,
      "step": 3098
    },
    {
      "epoch": 0.7051194539249147,
      "grad_norm": 1.0235189868055947,
      "learning_rate": 1.1896962928982116e-06,
      "loss": 0.0256,
      "step": 3099
    },
    {
      "epoch": 0.7053469852104665,
      "grad_norm": 1.084986587012269,
      "learning_rate": 1.1896579948633067e-06,
      "loss": 0.0839,
      "step": 3100
    },
    {
      "epoch": 0.7055745164960182,
      "grad_norm": 1.0758159672428005,
      "learning_rate": 1.1896196852878262e-06,
      "loss": 0.0554,
      "step": 3101
    },
    {
      "epoch": 0.70580204778157,
      "grad_norm": 1.653037063661443,
      "learning_rate": 1.1895813641725535e-06,
      "loss": 0.0592,
      "step": 3102
    },
    {
      "epoch": 0.7060295790671217,
      "grad_norm": 0.888432730316198,
      "learning_rate": 1.1895430315182719e-06,
      "loss": 0.0346,
      "step": 3103
    },
    {
      "epoch": 0.7062571103526735,
      "grad_norm": 0.7144453665457149,
      "learning_rate": 1.1895046873257644e-06,
      "loss": 0.0339,
      "step": 3104
    },
    {
      "epoch": 0.7064846416382252,
      "grad_norm": 1.1974179578050035,
      "learning_rate": 1.189466331595815e-06,
      "loss": 0.0584,
      "step": 3105
    },
    {
      "epoch": 0.706712172923777,
      "grad_norm": 2.0074894159842804,
      "learning_rate": 1.1894279643292074e-06,
      "loss": 0.0702,
      "step": 3106
    },
    {
      "epoch": 0.7069397042093288,
      "grad_norm": 1.1258269572183623,
      "learning_rate": 1.1893895855267262e-06,
      "loss": 0.0483,
      "step": 3107
    },
    {
      "epoch": 0.7071672354948806,
      "grad_norm": 1.8909664822572771,
      "learning_rate": 1.1893511951891553e-06,
      "loss": 0.0426,
      "step": 3108
    },
    {
      "epoch": 0.7073947667804323,
      "grad_norm": 0.9858584509738625,
      "learning_rate": 1.1893127933172794e-06,
      "loss": 0.0358,
      "step": 3109
    },
    {
      "epoch": 0.7076222980659841,
      "grad_norm": 1.6798238240138743,
      "learning_rate": 1.1892743799118838e-06,
      "loss": 0.0926,
      "step": 3110
    },
    {
      "epoch": 0.7078498293515358,
      "grad_norm": 1.8662422210150158,
      "learning_rate": 1.189235954973753e-06,
      "loss": 0.0913,
      "step": 3111
    },
    {
      "epoch": 0.7080773606370876,
      "grad_norm": 1.2592951116026574,
      "learning_rate": 1.189197518503673e-06,
      "loss": 0.0834,
      "step": 3112
    },
    {
      "epoch": 0.7083048919226393,
      "grad_norm": 2.2263635127158707,
      "learning_rate": 1.1891590705024288e-06,
      "loss": 0.1094,
      "step": 3113
    },
    {
      "epoch": 0.7085324232081911,
      "grad_norm": 1.2398530882425465,
      "learning_rate": 1.1891206109708065e-06,
      "loss": 0.0634,
      "step": 3114
    },
    {
      "epoch": 0.7087599544937428,
      "grad_norm": 0.8888283163380168,
      "learning_rate": 1.1890821399095917e-06,
      "loss": 0.0518,
      "step": 3115
    },
    {
      "epoch": 0.7089874857792946,
      "grad_norm": 1.7877536696152463,
      "learning_rate": 1.1890436573195714e-06,
      "loss": 0.0799,
      "step": 3116
    },
    {
      "epoch": 0.7092150170648465,
      "grad_norm": 0.6507653429669747,
      "learning_rate": 1.1890051632015315e-06,
      "loss": 0.0177,
      "step": 3117
    },
    {
      "epoch": 0.7094425483503982,
      "grad_norm": 0.810817143611541,
      "learning_rate": 1.1889666575562593e-06,
      "loss": 0.0294,
      "step": 3118
    },
    {
      "epoch": 0.70967007963595,
      "grad_norm": 1.351260714026528,
      "learning_rate": 1.1889281403845413e-06,
      "loss": 0.0764,
      "step": 3119
    },
    {
      "epoch": 0.7098976109215017,
      "grad_norm": 0.9662989115635772,
      "learning_rate": 1.1888896116871649e-06,
      "loss": 0.0592,
      "step": 3120
    },
    {
      "epoch": 0.7101251422070535,
      "grad_norm": 1.3131475158616286,
      "learning_rate": 1.1888510714649176e-06,
      "loss": 0.0485,
      "step": 3121
    },
    {
      "epoch": 0.7103526734926052,
      "grad_norm": 1.4877436902230308,
      "learning_rate": 1.1888125197185867e-06,
      "loss": 0.0786,
      "step": 3122
    },
    {
      "epoch": 0.710580204778157,
      "grad_norm": 0.9404627141658739,
      "learning_rate": 1.1887739564489608e-06,
      "loss": 0.0363,
      "step": 3123
    },
    {
      "epoch": 0.7108077360637087,
      "grad_norm": 1.664364740849866,
      "learning_rate": 1.1887353816568277e-06,
      "loss": 0.0836,
      "step": 3124
    },
    {
      "epoch": 0.7110352673492605,
      "grad_norm": 1.1581179507287511,
      "learning_rate": 1.188696795342976e-06,
      "loss": 0.0447,
      "step": 3125
    },
    {
      "epoch": 0.7112627986348122,
      "grad_norm": 1.4278480990598597,
      "learning_rate": 1.188658197508194e-06,
      "loss": 0.0702,
      "step": 3126
    },
    {
      "epoch": 0.7114903299203641,
      "grad_norm": 1.8166789381335957,
      "learning_rate": 1.1886195881532705e-06,
      "loss": 0.0756,
      "step": 3127
    },
    {
      "epoch": 0.7117178612059158,
      "grad_norm": 1.0532098621527515,
      "learning_rate": 1.1885809672789953e-06,
      "loss": 0.0427,
      "step": 3128
    },
    {
      "epoch": 0.7119453924914676,
      "grad_norm": 1.1496591085120342,
      "learning_rate": 1.188542334886157e-06,
      "loss": 0.0345,
      "step": 3129
    },
    {
      "epoch": 0.7121729237770194,
      "grad_norm": 1.3871333419583287,
      "learning_rate": 1.1885036909755454e-06,
      "loss": 0.0591,
      "step": 3130
    },
    {
      "epoch": 0.7124004550625711,
      "grad_norm": 1.4115338701891487,
      "learning_rate": 1.1884650355479505e-06,
      "loss": 0.0824,
      "step": 3131
    },
    {
      "epoch": 0.7126279863481229,
      "grad_norm": 1.4867197919013324,
      "learning_rate": 1.1884263686041622e-06,
      "loss": 0.0793,
      "step": 3132
    },
    {
      "epoch": 0.7128555176336746,
      "grad_norm": 1.2824607789793006,
      "learning_rate": 1.1883876901449707e-06,
      "loss": 0.0451,
      "step": 3133
    },
    {
      "epoch": 0.7130830489192264,
      "grad_norm": 0.6708570443551615,
      "learning_rate": 1.1883490001711667e-06,
      "loss": 0.0219,
      "step": 3134
    },
    {
      "epoch": 0.7133105802047781,
      "grad_norm": 2.015395007692978,
      "learning_rate": 1.1883102986835408e-06,
      "loss": 0.0757,
      "step": 3135
    },
    {
      "epoch": 0.71353811149033,
      "grad_norm": 1.6702641230416622,
      "learning_rate": 1.1882715856828842e-06,
      "loss": 0.075,
      "step": 3136
    },
    {
      "epoch": 0.7137656427758817,
      "grad_norm": 0.912883335407981,
      "learning_rate": 1.1882328611699879e-06,
      "loss": 0.0388,
      "step": 3137
    },
    {
      "epoch": 0.7139931740614335,
      "grad_norm": 1.9915888368964108,
      "learning_rate": 1.1881941251456434e-06,
      "loss": 0.0736,
      "step": 3138
    },
    {
      "epoch": 0.7142207053469852,
      "grad_norm": 1.432753078445199,
      "learning_rate": 1.1881553776106423e-06,
      "loss": 0.0686,
      "step": 3139
    },
    {
      "epoch": 0.714448236632537,
      "grad_norm": 1.8572484999557712,
      "learning_rate": 1.1881166185657765e-06,
      "loss": 0.0639,
      "step": 3140
    },
    {
      "epoch": 0.7146757679180887,
      "grad_norm": 1.1698682146531856,
      "learning_rate": 1.1880778480118388e-06,
      "loss": 0.0506,
      "step": 3141
    },
    {
      "epoch": 0.7149032992036405,
      "grad_norm": 1.4247011889264531,
      "learning_rate": 1.1880390659496207e-06,
      "loss": 0.0688,
      "step": 3142
    },
    {
      "epoch": 0.7151308304891922,
      "grad_norm": 1.1897422778029059,
      "learning_rate": 1.1880002723799155e-06,
      "loss": 0.0488,
      "step": 3143
    },
    {
      "epoch": 0.715358361774744,
      "grad_norm": 0.7067410656877349,
      "learning_rate": 1.1879614673035158e-06,
      "loss": 0.0316,
      "step": 3144
    },
    {
      "epoch": 0.7155858930602957,
      "grad_norm": 1.1073184850973958,
      "learning_rate": 1.1879226507212146e-06,
      "loss": 0.04,
      "step": 3145
    },
    {
      "epoch": 0.7158134243458476,
      "grad_norm": 1.551329079095373,
      "learning_rate": 1.1878838226338054e-06,
      "loss": 0.0449,
      "step": 3146
    },
    {
      "epoch": 0.7160409556313994,
      "grad_norm": 1.2160812373705194,
      "learning_rate": 1.187844983042082e-06,
      "loss": 0.0547,
      "step": 3147
    },
    {
      "epoch": 0.7162684869169511,
      "grad_norm": 1.463195276621994,
      "learning_rate": 1.1878061319468376e-06,
      "loss": 0.0604,
      "step": 3148
    },
    {
      "epoch": 0.7164960182025029,
      "grad_norm": 0.7715205659161936,
      "learning_rate": 1.1877672693488669e-06,
      "loss": 0.0395,
      "step": 3149
    },
    {
      "epoch": 0.7167235494880546,
      "grad_norm": 0.9856101752123785,
      "learning_rate": 1.1877283952489636e-06,
      "loss": 0.0419,
      "step": 3150
    },
    {
      "epoch": 0.7169510807736064,
      "grad_norm": 1.1172173173435798,
      "learning_rate": 1.1876895096479226e-06,
      "loss": 0.044,
      "step": 3151
    },
    {
      "epoch": 0.7171786120591581,
      "grad_norm": 1.7824313127710856,
      "learning_rate": 1.1876506125465386e-06,
      "loss": 0.0882,
      "step": 3152
    },
    {
      "epoch": 0.7174061433447099,
      "grad_norm": 0.6797130682621123,
      "learning_rate": 1.1876117039456065e-06,
      "loss": 0.0241,
      "step": 3153
    },
    {
      "epoch": 0.7176336746302616,
      "grad_norm": 1.8479763283658723,
      "learning_rate": 1.1875727838459213e-06,
      "loss": 0.0718,
      "step": 3154
    },
    {
      "epoch": 0.7178612059158134,
      "grad_norm": 0.8348967152826982,
      "learning_rate": 1.187533852248279e-06,
      "loss": 0.0293,
      "step": 3155
    },
    {
      "epoch": 0.7180887372013652,
      "grad_norm": 1.0039148482185,
      "learning_rate": 1.1874949091534749e-06,
      "loss": 0.0465,
      "step": 3156
    },
    {
      "epoch": 0.718316268486917,
      "grad_norm": 1.6124637787966882,
      "learning_rate": 1.1874559545623049e-06,
      "loss": 0.093,
      "step": 3157
    },
    {
      "epoch": 0.7185437997724687,
      "grad_norm": 1.3428421165694695,
      "learning_rate": 1.1874169884755654e-06,
      "loss": 0.068,
      "step": 3158
    },
    {
      "epoch": 0.7187713310580205,
      "grad_norm": 0.9564008146931468,
      "learning_rate": 1.1873780108940527e-06,
      "loss": 0.0424,
      "step": 3159
    },
    {
      "epoch": 0.7189988623435722,
      "grad_norm": 0.8282933582416673,
      "learning_rate": 1.1873390218185636e-06,
      "loss": 0.0237,
      "step": 3160
    },
    {
      "epoch": 0.719226393629124,
      "grad_norm": 0.8573276208421515,
      "learning_rate": 1.1873000212498942e-06,
      "loss": 0.0431,
      "step": 3161
    },
    {
      "epoch": 0.7194539249146757,
      "grad_norm": 0.9698959738736916,
      "learning_rate": 1.1872610091888426e-06,
      "loss": 0.0444,
      "step": 3162
    },
    {
      "epoch": 0.7196814562002275,
      "grad_norm": 1.8230237032369256,
      "learning_rate": 1.1872219856362057e-06,
      "loss": 0.0878,
      "step": 3163
    },
    {
      "epoch": 0.7199089874857793,
      "grad_norm": 1.1950764632445416,
      "learning_rate": 1.187182950592781e-06,
      "loss": 0.0559,
      "step": 3164
    },
    {
      "epoch": 0.7201365187713311,
      "grad_norm": 1.1584194920218946,
      "learning_rate": 1.1871439040593663e-06,
      "loss": 0.0659,
      "step": 3165
    },
    {
      "epoch": 0.7203640500568829,
      "grad_norm": 1.149871056931848,
      "learning_rate": 1.1871048460367598e-06,
      "loss": 0.0385,
      "step": 3166
    },
    {
      "epoch": 0.7205915813424346,
      "grad_norm": 1.1250408047699638,
      "learning_rate": 1.1870657765257595e-06,
      "loss": 0.041,
      "step": 3167
    },
    {
      "epoch": 0.7208191126279864,
      "grad_norm": 0.9750498880779663,
      "learning_rate": 1.1870266955271645e-06,
      "loss": 0.0286,
      "step": 3168
    },
    {
      "epoch": 0.7210466439135381,
      "grad_norm": 1.93920042584572,
      "learning_rate": 1.186987603041773e-06,
      "loss": 0.1038,
      "step": 3169
    },
    {
      "epoch": 0.7212741751990899,
      "grad_norm": 1.1716818159995082,
      "learning_rate": 1.1869484990703839e-06,
      "loss": 0.0567,
      "step": 3170
    },
    {
      "epoch": 0.7215017064846416,
      "grad_norm": 1.39986755402065,
      "learning_rate": 1.1869093836137968e-06,
      "loss": 0.0763,
      "step": 3171
    },
    {
      "epoch": 0.7217292377701934,
      "grad_norm": 1.3078245254309557,
      "learning_rate": 1.186870256672811e-06,
      "loss": 0.061,
      "step": 3172
    },
    {
      "epoch": 0.7219567690557451,
      "grad_norm": 1.3174384504374115,
      "learning_rate": 1.1868311182482262e-06,
      "loss": 0.0416,
      "step": 3173
    },
    {
      "epoch": 0.7221843003412969,
      "grad_norm": 1.404891211071726,
      "learning_rate": 1.1867919683408421e-06,
      "loss": 0.0272,
      "step": 3174
    },
    {
      "epoch": 0.7224118316268487,
      "grad_norm": 1.4160327712086356,
      "learning_rate": 1.1867528069514591e-06,
      "loss": 0.0863,
      "step": 3175
    },
    {
      "epoch": 0.7226393629124005,
      "grad_norm": 1.400708180969138,
      "learning_rate": 1.1867136340808778e-06,
      "loss": 0.0502,
      "step": 3176
    },
    {
      "epoch": 0.7228668941979522,
      "grad_norm": 1.6148464588484233,
      "learning_rate": 1.1866744497298982e-06,
      "loss": 0.0693,
      "step": 3177
    },
    {
      "epoch": 0.723094425483504,
      "grad_norm": 1.0345610557360225,
      "learning_rate": 1.1866352538993216e-06,
      "loss": 0.0441,
      "step": 3178
    },
    {
      "epoch": 0.7233219567690558,
      "grad_norm": 1.2686233759159034,
      "learning_rate": 1.1865960465899492e-06,
      "loss": 0.0658,
      "step": 3179
    },
    {
      "epoch": 0.7235494880546075,
      "grad_norm": 1.0561518957357936,
      "learning_rate": 1.186556827802582e-06,
      "loss": 0.0316,
      "step": 3180
    },
    {
      "epoch": 0.7237770193401593,
      "grad_norm": 1.2859639053927798,
      "learning_rate": 1.1865175975380218e-06,
      "loss": 0.0748,
      "step": 3181
    },
    {
      "epoch": 0.724004550625711,
      "grad_norm": 1.0612859349624024,
      "learning_rate": 1.18647835579707e-06,
      "loss": 0.0533,
      "step": 3182
    },
    {
      "epoch": 0.7242320819112628,
      "grad_norm": 1.252216188797781,
      "learning_rate": 1.186439102580529e-06,
      "loss": 0.0582,
      "step": 3183
    },
    {
      "epoch": 0.7244596131968145,
      "grad_norm": 1.4352458095090412,
      "learning_rate": 1.1863998378892011e-06,
      "loss": 0.0498,
      "step": 3184
    },
    {
      "epoch": 0.7246871444823664,
      "grad_norm": 2.834373155221246,
      "learning_rate": 1.1863605617238885e-06,
      "loss": 0.1159,
      "step": 3185
    },
    {
      "epoch": 0.7249146757679181,
      "grad_norm": 1.1067917536466692,
      "learning_rate": 1.1863212740853941e-06,
      "loss": 0.0721,
      "step": 3186
    },
    {
      "epoch": 0.7251422070534699,
      "grad_norm": 1.2158906431759549,
      "learning_rate": 1.1862819749745212e-06,
      "loss": 0.0272,
      "step": 3187
    },
    {
      "epoch": 0.7253697383390216,
      "grad_norm": 1.6050627766428684,
      "learning_rate": 1.1862426643920722e-06,
      "loss": 0.0567,
      "step": 3188
    },
    {
      "epoch": 0.7255972696245734,
      "grad_norm": 1.0752979084787344,
      "learning_rate": 1.1862033423388513e-06,
      "loss": 0.0442,
      "step": 3189
    },
    {
      "epoch": 0.7258248009101251,
      "grad_norm": 0.9789837014009594,
      "learning_rate": 1.1861640088156617e-06,
      "loss": 0.0524,
      "step": 3190
    },
    {
      "epoch": 0.7260523321956769,
      "grad_norm": 1.085883546609774,
      "learning_rate": 1.1861246638233077e-06,
      "loss": 0.043,
      "step": 3191
    },
    {
      "epoch": 0.7262798634812286,
      "grad_norm": 2.3885050844926305,
      "learning_rate": 1.1860853073625931e-06,
      "loss": 0.052,
      "step": 3192
    },
    {
      "epoch": 0.7265073947667804,
      "grad_norm": 1.8498560328174254,
      "learning_rate": 1.1860459394343223e-06,
      "loss": 0.1156,
      "step": 3193
    },
    {
      "epoch": 0.7267349260523321,
      "grad_norm": 1.1970597608121667,
      "learning_rate": 1.1860065600393002e-06,
      "loss": 0.0409,
      "step": 3194
    },
    {
      "epoch": 0.726962457337884,
      "grad_norm": 1.5076572874006593,
      "learning_rate": 1.1859671691783315e-06,
      "loss": 0.0561,
      "step": 3195
    },
    {
      "epoch": 0.7271899886234358,
      "grad_norm": 0.7392429484644525,
      "learning_rate": 1.1859277668522209e-06,
      "loss": 0.0228,
      "step": 3196
    },
    {
      "epoch": 0.7274175199089875,
      "grad_norm": 2.2057392375122973,
      "learning_rate": 1.1858883530617743e-06,
      "loss": 0.1463,
      "step": 3197
    },
    {
      "epoch": 0.7276450511945393,
      "grad_norm": 2.23022548380175,
      "learning_rate": 1.185848927807797e-06,
      "loss": 0.0798,
      "step": 3198
    },
    {
      "epoch": 0.727872582480091,
      "grad_norm": 1.7101734591640143,
      "learning_rate": 1.1858094910910945e-06,
      "loss": 0.0721,
      "step": 3199
    },
    {
      "epoch": 0.7281001137656428,
      "grad_norm": 1.013052127850482,
      "learning_rate": 1.1857700429124733e-06,
      "loss": 0.044,
      "step": 3200
    },
    {
      "epoch": 0.7283276450511945,
      "grad_norm": 0.9338207707410625,
      "learning_rate": 1.1857305832727395e-06,
      "loss": 0.0356,
      "step": 3201
    },
    {
      "epoch": 0.7285551763367463,
      "grad_norm": 1.151011668858322,
      "learning_rate": 1.1856911121726993e-06,
      "loss": 0.055,
      "step": 3202
    },
    {
      "epoch": 0.728782707622298,
      "grad_norm": 0.882191391961338,
      "learning_rate": 1.1856516296131596e-06,
      "loss": 0.0293,
      "step": 3203
    },
    {
      "epoch": 0.7290102389078499,
      "grad_norm": 1.701498982454748,
      "learning_rate": 1.1856121355949276e-06,
      "loss": 0.0632,
      "step": 3204
    },
    {
      "epoch": 0.7292377701934016,
      "grad_norm": 0.9653458303805711,
      "learning_rate": 1.18557263011881e-06,
      "loss": 0.0357,
      "step": 3205
    },
    {
      "epoch": 0.7294653014789534,
      "grad_norm": 1.2647182576075804,
      "learning_rate": 1.1855331131856146e-06,
      "loss": 0.0988,
      "step": 3206
    },
    {
      "epoch": 0.7296928327645051,
      "grad_norm": 1.2708027008753917,
      "learning_rate": 1.185493584796149e-06,
      "loss": 0.0659,
      "step": 3207
    },
    {
      "epoch": 0.7299203640500569,
      "grad_norm": 1.7886995517155944,
      "learning_rate": 1.185454044951221e-06,
      "loss": 0.0779,
      "step": 3208
    },
    {
      "epoch": 0.7301478953356086,
      "grad_norm": 0.6845065442131042,
      "learning_rate": 1.1854144936516388e-06,
      "loss": 0.0325,
      "step": 3209
    },
    {
      "epoch": 0.7303754266211604,
      "grad_norm": 0.9425373691725761,
      "learning_rate": 1.1853749308982107e-06,
      "loss": 0.0353,
      "step": 3210
    },
    {
      "epoch": 0.7306029579067121,
      "grad_norm": 1.6009640208057585,
      "learning_rate": 1.1853353566917452e-06,
      "loss": 0.0689,
      "step": 3211
    },
    {
      "epoch": 0.7308304891922639,
      "grad_norm": 0.9565911200831438,
      "learning_rate": 1.1852957710330511e-06,
      "loss": 0.0438,
      "step": 3212
    },
    {
      "epoch": 0.7310580204778157,
      "grad_norm": 1.6166034190859908,
      "learning_rate": 1.1852561739229377e-06,
      "loss": 0.0521,
      "step": 3213
    },
    {
      "epoch": 0.7312855517633675,
      "grad_norm": 2.3048115272176983,
      "learning_rate": 1.1852165653622141e-06,
      "loss": 0.1036,
      "step": 3214
    },
    {
      "epoch": 0.7315130830489193,
      "grad_norm": 0.6867861750300609,
      "learning_rate": 1.18517694535169e-06,
      "loss": 0.0285,
      "step": 3215
    },
    {
      "epoch": 0.731740614334471,
      "grad_norm": 0.7222977767557096,
      "learning_rate": 1.1851373138921752e-06,
      "loss": 0.0332,
      "step": 3216
    },
    {
      "epoch": 0.7319681456200228,
      "grad_norm": 1.0427932498841996,
      "learning_rate": 1.1850976709844792e-06,
      "loss": 0.0654,
      "step": 3217
    },
    {
      "epoch": 0.7321956769055745,
      "grad_norm": 0.7626880126713813,
      "learning_rate": 1.1850580166294127e-06,
      "loss": 0.0257,
      "step": 3218
    },
    {
      "epoch": 0.7324232081911263,
      "grad_norm": 1.3204712038737856,
      "learning_rate": 1.1850183508277862e-06,
      "loss": 0.0821,
      "step": 3219
    },
    {
      "epoch": 0.732650739476678,
      "grad_norm": 1.920439901931252,
      "learning_rate": 1.18497867358041e-06,
      "loss": 0.0843,
      "step": 3220
    },
    {
      "epoch": 0.7328782707622298,
      "grad_norm": 1.4946908757395634,
      "learning_rate": 1.1849389848880955e-06,
      "loss": 0.075,
      "step": 3221
    },
    {
      "epoch": 0.7331058020477815,
      "grad_norm": 0.5605078244225655,
      "learning_rate": 1.1848992847516535e-06,
      "loss": 0.0188,
      "step": 3222
    },
    {
      "epoch": 0.7333333333333333,
      "grad_norm": 0.7658587959864014,
      "learning_rate": 1.1848595731718955e-06,
      "loss": 0.0246,
      "step": 3223
    },
    {
      "epoch": 0.7335608646188851,
      "grad_norm": 1.6693233826267546,
      "learning_rate": 1.1848198501496331e-06,
      "loss": 0.0709,
      "step": 3224
    },
    {
      "epoch": 0.7337883959044369,
      "grad_norm": 1.6483972044168969,
      "learning_rate": 1.1847801156856783e-06,
      "loss": 0.0735,
      "step": 3225
    },
    {
      "epoch": 0.7340159271899886,
      "grad_norm": 1.4807700617572261,
      "learning_rate": 1.1847403697808433e-06,
      "loss": 0.0838,
      "step": 3226
    },
    {
      "epoch": 0.7342434584755404,
      "grad_norm": 1.485456481779637,
      "learning_rate": 1.18470061243594e-06,
      "loss": 0.0687,
      "step": 3227
    },
    {
      "epoch": 0.7344709897610922,
      "grad_norm": 1.4565020245267035,
      "learning_rate": 1.1846608436517813e-06,
      "loss": 0.0593,
      "step": 3228
    },
    {
      "epoch": 0.7346985210466439,
      "grad_norm": 1.3974828149244531,
      "learning_rate": 1.1846210634291799e-06,
      "loss": 0.0683,
      "step": 3229
    },
    {
      "epoch": 0.7349260523321957,
      "grad_norm": 0.9824198436565561,
      "learning_rate": 1.184581271768949e-06,
      "loss": 0.0474,
      "step": 3230
    },
    {
      "epoch": 0.7351535836177474,
      "grad_norm": 1.4337603662453289,
      "learning_rate": 1.1845414686719014e-06,
      "loss": 0.0621,
      "step": 3231
    },
    {
      "epoch": 0.7353811149032992,
      "grad_norm": 0.8374788974669934,
      "learning_rate": 1.1845016541388513e-06,
      "loss": 0.0388,
      "step": 3232
    },
    {
      "epoch": 0.7356086461888509,
      "grad_norm": 1.6437938029709063,
      "learning_rate": 1.184461828170612e-06,
      "loss": 0.0822,
      "step": 3233
    },
    {
      "epoch": 0.7358361774744028,
      "grad_norm": 0.9054278416004449,
      "learning_rate": 1.1844219907679973e-06,
      "loss": 0.0364,
      "step": 3234
    },
    {
      "epoch": 0.7360637087599545,
      "grad_norm": 1.2269654818634814,
      "learning_rate": 1.184382141931822e-06,
      "loss": 0.0474,
      "step": 3235
    },
    {
      "epoch": 0.7362912400455063,
      "grad_norm": 1.2170865361290026,
      "learning_rate": 1.1843422816628998e-06,
      "loss": 0.0463,
      "step": 3236
    },
    {
      "epoch": 0.736518771331058,
      "grad_norm": 1.224432806747687,
      "learning_rate": 1.184302409962046e-06,
      "loss": 0.047,
      "step": 3237
    },
    {
      "epoch": 0.7367463026166098,
      "grad_norm": 1.0965112824005279,
      "learning_rate": 1.1842625268300754e-06,
      "loss": 0.0359,
      "step": 3238
    },
    {
      "epoch": 0.7369738339021615,
      "grad_norm": 1.0155537003395356,
      "learning_rate": 1.1842226322678028e-06,
      "loss": 0.0325,
      "step": 3239
    },
    {
      "epoch": 0.7372013651877133,
      "grad_norm": 1.6425941417778662,
      "learning_rate": 1.1841827262760436e-06,
      "loss": 0.0666,
      "step": 3240
    },
    {
      "epoch": 0.737428896473265,
      "grad_norm": 1.0729010350000274,
      "learning_rate": 1.1841428088556137e-06,
      "loss": 0.0446,
      "step": 3241
    },
    {
      "epoch": 0.7376564277588168,
      "grad_norm": 1.5054940909406098,
      "learning_rate": 1.184102880007329e-06,
      "loss": 0.0734,
      "step": 3242
    },
    {
      "epoch": 0.7378839590443687,
      "grad_norm": 1.4685805990987935,
      "learning_rate": 1.1840629397320052e-06,
      "loss": 0.0657,
      "step": 3243
    },
    {
      "epoch": 0.7381114903299204,
      "grad_norm": 1.0491905607582221,
      "learning_rate": 1.1840229880304589e-06,
      "loss": 0.0553,
      "step": 3244
    },
    {
      "epoch": 0.7383390216154722,
      "grad_norm": 1.6183523557552497,
      "learning_rate": 1.1839830249035062e-06,
      "loss": 0.0401,
      "step": 3245
    },
    {
      "epoch": 0.7385665529010239,
      "grad_norm": 2.0313633648707885,
      "learning_rate": 1.1839430503519645e-06,
      "loss": 0.0871,
      "step": 3246
    },
    {
      "epoch": 0.7387940841865757,
      "grad_norm": 2.2817905288401485,
      "learning_rate": 1.1839030643766505e-06,
      "loss": 0.0876,
      "step": 3247
    },
    {
      "epoch": 0.7390216154721274,
      "grad_norm": 1.2662039437671349,
      "learning_rate": 1.1838630669783814e-06,
      "loss": 0.0597,
      "step": 3248
    },
    {
      "epoch": 0.7392491467576792,
      "grad_norm": 2.4757641907590116,
      "learning_rate": 1.1838230581579746e-06,
      "loss": 0.0364,
      "step": 3249
    },
    {
      "epoch": 0.7394766780432309,
      "grad_norm": 1.509162194685489,
      "learning_rate": 1.183783037916248e-06,
      "loss": 0.0728,
      "step": 3250
    },
    {
      "epoch": 0.7397042093287827,
      "grad_norm": 1.8117671336209638,
      "learning_rate": 1.1837430062540196e-06,
      "loss": 0.074,
      "step": 3251
    },
    {
      "epoch": 0.7399317406143344,
      "grad_norm": 1.426185256628181,
      "learning_rate": 1.1837029631721072e-06,
      "loss": 0.0663,
      "step": 3252
    },
    {
      "epoch": 0.7401592718998863,
      "grad_norm": 1.930245275398741,
      "learning_rate": 1.1836629086713296e-06,
      "loss": 0.0579,
      "step": 3253
    },
    {
      "epoch": 0.740386803185438,
      "grad_norm": 2.323978623936963,
      "learning_rate": 1.1836228427525054e-06,
      "loss": 0.0314,
      "step": 3254
    },
    {
      "epoch": 0.7406143344709898,
      "grad_norm": 1.2652157180134698,
      "learning_rate": 1.183582765416453e-06,
      "loss": 0.0451,
      "step": 3255
    },
    {
      "epoch": 0.7408418657565415,
      "grad_norm": 1.1001275189243191,
      "learning_rate": 1.1835426766639923e-06,
      "loss": 0.034,
      "step": 3256
    },
    {
      "epoch": 0.7410693970420933,
      "grad_norm": 1.2716136324777447,
      "learning_rate": 1.183502576495942e-06,
      "loss": 0.0434,
      "step": 3257
    },
    {
      "epoch": 0.741296928327645,
      "grad_norm": 1.8985708463935187,
      "learning_rate": 1.1834624649131218e-06,
      "loss": 0.0825,
      "step": 3258
    },
    {
      "epoch": 0.7415244596131968,
      "grad_norm": 0.9934071400912606,
      "learning_rate": 1.1834223419163518e-06,
      "loss": 0.0332,
      "step": 3259
    },
    {
      "epoch": 0.7417519908987485,
      "grad_norm": 1.9148887694046421,
      "learning_rate": 1.1833822075064517e-06,
      "loss": 0.0288,
      "step": 3260
    },
    {
      "epoch": 0.7419795221843003,
      "grad_norm": 1.1004472895017543,
      "learning_rate": 1.183342061684242e-06,
      "loss": 0.0487,
      "step": 3261
    },
    {
      "epoch": 0.742207053469852,
      "grad_norm": 1.203746668995667,
      "learning_rate": 1.183301904450543e-06,
      "loss": 0.0423,
      "step": 3262
    },
    {
      "epoch": 0.7424345847554039,
      "grad_norm": 2.131599554211613,
      "learning_rate": 1.1832617358061756e-06,
      "loss": 0.078,
      "step": 3263
    },
    {
      "epoch": 0.7426621160409557,
      "grad_norm": 1.3232000144442189,
      "learning_rate": 1.1832215557519608e-06,
      "loss": 0.0542,
      "step": 3264
    },
    {
      "epoch": 0.7428896473265074,
      "grad_norm": 1.3347834617776861,
      "learning_rate": 1.1831813642887196e-06,
      "loss": 0.0571,
      "step": 3265
    },
    {
      "epoch": 0.7431171786120592,
      "grad_norm": 1.491233272879141,
      "learning_rate": 1.1831411614172735e-06,
      "loss": 0.0706,
      "step": 3266
    },
    {
      "epoch": 0.7433447098976109,
      "grad_norm": 2.338157694562792,
      "learning_rate": 1.1831009471384445e-06,
      "loss": 0.0591,
      "step": 3267
    },
    {
      "epoch": 0.7435722411831627,
      "grad_norm": 1.1906122468487437,
      "learning_rate": 1.1830607214530543e-06,
      "loss": 0.0693,
      "step": 3268
    },
    {
      "epoch": 0.7437997724687144,
      "grad_norm": 1.2940810772686644,
      "learning_rate": 1.1830204843619248e-06,
      "loss": 0.0517,
      "step": 3269
    },
    {
      "epoch": 0.7440273037542662,
      "grad_norm": 2.214601640279187,
      "learning_rate": 1.1829802358658785e-06,
      "loss": 0.1046,
      "step": 3270
    },
    {
      "epoch": 0.7442548350398179,
      "grad_norm": 1.039009352489729,
      "learning_rate": 1.1829399759657383e-06,
      "loss": 0.0592,
      "step": 3271
    },
    {
      "epoch": 0.7444823663253698,
      "grad_norm": 1.8684769750059191,
      "learning_rate": 1.1828997046623267e-06,
      "loss": 0.093,
      "step": 3272
    },
    {
      "epoch": 0.7447098976109215,
      "grad_norm": 1.823521554100394,
      "learning_rate": 1.1828594219564669e-06,
      "loss": 0.049,
      "step": 3273
    },
    {
      "epoch": 0.7449374288964733,
      "grad_norm": 1.8271992779263462,
      "learning_rate": 1.182819127848982e-06,
      "loss": 0.0303,
      "step": 3274
    },
    {
      "epoch": 0.745164960182025,
      "grad_norm": 3.0795350516203928,
      "learning_rate": 1.1827788223406959e-06,
      "loss": 0.1191,
      "step": 3275
    },
    {
      "epoch": 0.7453924914675768,
      "grad_norm": 1.7589178709629525,
      "learning_rate": 1.1827385054324323e-06,
      "loss": 0.0616,
      "step": 3276
    },
    {
      "epoch": 0.7456200227531286,
      "grad_norm": 2.4742706309940976,
      "learning_rate": 1.1826981771250148e-06,
      "loss": 0.077,
      "step": 3277
    },
    {
      "epoch": 0.7458475540386803,
      "grad_norm": 1.9750018572038446,
      "learning_rate": 1.1826578374192681e-06,
      "loss": 0.0795,
      "step": 3278
    },
    {
      "epoch": 0.7460750853242321,
      "grad_norm": 1.5314832082240002,
      "learning_rate": 1.1826174863160168e-06,
      "loss": 0.0694,
      "step": 3279
    },
    {
      "epoch": 0.7463026166097838,
      "grad_norm": 0.9199631256267508,
      "learning_rate": 1.182577123816085e-06,
      "loss": 0.0407,
      "step": 3280
    },
    {
      "epoch": 0.7465301478953356,
      "grad_norm": 1.6996895039706184,
      "learning_rate": 1.1825367499202978e-06,
      "loss": 0.0648,
      "step": 3281
    },
    {
      "epoch": 0.7467576791808874,
      "grad_norm": 1.538865347545186,
      "learning_rate": 1.1824963646294806e-06,
      "loss": 0.0483,
      "step": 3282
    },
    {
      "epoch": 0.7469852104664392,
      "grad_norm": 1.5246278126883535,
      "learning_rate": 1.1824559679444588e-06,
      "loss": 0.0652,
      "step": 3283
    },
    {
      "epoch": 0.7472127417519909,
      "grad_norm": 0.7990923695187152,
      "learning_rate": 1.182415559866058e-06,
      "loss": 0.0416,
      "step": 3284
    },
    {
      "epoch": 0.7474402730375427,
      "grad_norm": 1.0564862923399663,
      "learning_rate": 1.182375140395104e-06,
      "loss": 0.0414,
      "step": 3285
    },
    {
      "epoch": 0.7476678043230944,
      "grad_norm": 1.1846269731523205,
      "learning_rate": 1.1823347095324228e-06,
      "loss": 0.0621,
      "step": 3286
    },
    {
      "epoch": 0.7478953356086462,
      "grad_norm": 0.8569640753381744,
      "learning_rate": 1.1822942672788409e-06,
      "loss": 0.0278,
      "step": 3287
    },
    {
      "epoch": 0.7481228668941979,
      "grad_norm": 1.571726141718777,
      "learning_rate": 1.1822538136351849e-06,
      "loss": 0.0613,
      "step": 3288
    },
    {
      "epoch": 0.7483503981797497,
      "grad_norm": 1.1282851528775475,
      "learning_rate": 1.1822133486022815e-06,
      "loss": 0.0392,
      "step": 3289
    },
    {
      "epoch": 0.7485779294653014,
      "grad_norm": 1.0669556173185228,
      "learning_rate": 1.1821728721809577e-06,
      "loss": 0.0294,
      "step": 3290
    },
    {
      "epoch": 0.7488054607508532,
      "grad_norm": 2.466980162201729,
      "learning_rate": 1.1821323843720408e-06,
      "loss": 0.0745,
      "step": 3291
    },
    {
      "epoch": 0.749032992036405,
      "grad_norm": 1.8125453554610331,
      "learning_rate": 1.1820918851763582e-06,
      "loss": 0.0798,
      "step": 3292
    },
    {
      "epoch": 0.7492605233219568,
      "grad_norm": 1.6160104087285256,
      "learning_rate": 1.182051374594738e-06,
      "loss": 0.0769,
      "step": 3293
    },
    {
      "epoch": 0.7494880546075086,
      "grad_norm": 1.5687919942801445,
      "learning_rate": 1.1820108526280076e-06,
      "loss": 0.0626,
      "step": 3294
    },
    {
      "epoch": 0.7497155858930603,
      "grad_norm": 0.9398572843854899,
      "learning_rate": 1.1819703192769955e-06,
      "loss": 0.0482,
      "step": 3295
    },
    {
      "epoch": 0.7499431171786121,
      "grad_norm": 12.232356704164316,
      "learning_rate": 1.1819297745425304e-06,
      "loss": 0.0627,
      "step": 3296
    },
    {
      "epoch": 0.7501706484641638,
      "grad_norm": 1.4183355936794655,
      "learning_rate": 1.1818892184254404e-06,
      "loss": 0.0731,
      "step": 3297
    },
    {
      "epoch": 0.7503981797497156,
      "grad_norm": 0.7929826048625227,
      "learning_rate": 1.1818486509265547e-06,
      "loss": 0.028,
      "step": 3298
    },
    {
      "epoch": 0.7506257110352673,
      "grad_norm": 0.9321132746109633,
      "learning_rate": 1.1818080720467026e-06,
      "loss": 0.0406,
      "step": 3299
    },
    {
      "epoch": 0.7508532423208191,
      "grad_norm": 0.7184037331999952,
      "learning_rate": 1.1817674817867131e-06,
      "loss": 0.0292,
      "step": 3300
    },
    {
      "epoch": 0.7510807736063708,
      "grad_norm": 1.7585855147942302,
      "learning_rate": 1.181726880147416e-06,
      "loss": 0.074,
      "step": 3301
    },
    {
      "epoch": 0.7513083048919227,
      "grad_norm": 0.9632647926307707,
      "learning_rate": 1.181686267129641e-06,
      "loss": 0.0318,
      "step": 3302
    },
    {
      "epoch": 0.7515358361774744,
      "grad_norm": 1.2947286461920264,
      "learning_rate": 1.1816456427342181e-06,
      "loss": 0.0649,
      "step": 3303
    },
    {
      "epoch": 0.7517633674630262,
      "grad_norm": 0.9051783521749419,
      "learning_rate": 1.181605006961978e-06,
      "loss": 0.0318,
      "step": 3304
    },
    {
      "epoch": 0.7519908987485779,
      "grad_norm": 1.9818984386721539,
      "learning_rate": 1.1815643598137507e-06,
      "loss": 0.0909,
      "step": 3305
    },
    {
      "epoch": 0.7522184300341297,
      "grad_norm": 1.1059310163256162,
      "learning_rate": 1.1815237012903675e-06,
      "loss": 0.0422,
      "step": 3306
    },
    {
      "epoch": 0.7524459613196814,
      "grad_norm": 2.4093572946300865,
      "learning_rate": 1.1814830313926589e-06,
      "loss": 0.0944,
      "step": 3307
    },
    {
      "epoch": 0.7526734926052332,
      "grad_norm": 1.2458152163270595,
      "learning_rate": 1.1814423501214562e-06,
      "loss": 0.0422,
      "step": 3308
    },
    {
      "epoch": 0.752901023890785,
      "grad_norm": 1.1846230368435793,
      "learning_rate": 1.1814016574775909e-06,
      "loss": 0.0459,
      "step": 3309
    },
    {
      "epoch": 0.7531285551763367,
      "grad_norm": 0.7274927229665066,
      "learning_rate": 1.1813609534618948e-06,
      "loss": 0.031,
      "step": 3310
    },
    {
      "epoch": 0.7533560864618886,
      "grad_norm": 1.0084301910129103,
      "learning_rate": 1.1813202380751998e-06,
      "loss": 0.022,
      "step": 3311
    },
    {
      "epoch": 0.7535836177474403,
      "grad_norm": 1.2248641094215589,
      "learning_rate": 1.1812795113183378e-06,
      "loss": 0.0519,
      "step": 3312
    },
    {
      "epoch": 0.7538111490329921,
      "grad_norm": 1.2989392125838315,
      "learning_rate": 1.1812387731921415e-06,
      "loss": 0.0418,
      "step": 3313
    },
    {
      "epoch": 0.7540386803185438,
      "grad_norm": 1.8035598035830265,
      "learning_rate": 1.1811980236974435e-06,
      "loss": 0.047,
      "step": 3314
    },
    {
      "epoch": 0.7542662116040956,
      "grad_norm": 0.9445160919482642,
      "learning_rate": 1.1811572628350764e-06,
      "loss": 0.053,
      "step": 3315
    },
    {
      "epoch": 0.7544937428896473,
      "grad_norm": 1.820531092229777,
      "learning_rate": 1.1811164906058735e-06,
      "loss": 0.0898,
      "step": 3316
    },
    {
      "epoch": 0.7547212741751991,
      "grad_norm": 1.7017364368903272,
      "learning_rate": 1.1810757070106678e-06,
      "loss": 0.064,
      "step": 3317
    },
    {
      "epoch": 0.7549488054607508,
      "grad_norm": 1.589156926554848,
      "learning_rate": 1.1810349120502932e-06,
      "loss": 0.069,
      "step": 3318
    },
    {
      "epoch": 0.7551763367463026,
      "grad_norm": 0.9315330165580284,
      "learning_rate": 1.1809941057255834e-06,
      "loss": 0.043,
      "step": 3319
    },
    {
      "epoch": 0.7554038680318543,
      "grad_norm": 1.7002202165396425,
      "learning_rate": 1.1809532880373721e-06,
      "loss": 0.0662,
      "step": 3320
    },
    {
      "epoch": 0.7556313993174062,
      "grad_norm": 1.4641161178529223,
      "learning_rate": 1.180912458986494e-06,
      "loss": 0.0696,
      "step": 3321
    },
    {
      "epoch": 0.755858930602958,
      "grad_norm": 1.5316937073382941,
      "learning_rate": 1.180871618573783e-06,
      "loss": 0.0532,
      "step": 3322
    },
    {
      "epoch": 0.7560864618885097,
      "grad_norm": 1.5604240478591438,
      "learning_rate": 1.1808307668000745e-06,
      "loss": 0.0511,
      "step": 3323
    },
    {
      "epoch": 0.7563139931740614,
      "grad_norm": 1.7255746868374173,
      "learning_rate": 1.180789903666203e-06,
      "loss": 0.0646,
      "step": 3324
    },
    {
      "epoch": 0.7565415244596132,
      "grad_norm": 1.3143153209229907,
      "learning_rate": 1.1807490291730036e-06,
      "loss": 0.0792,
      "step": 3325
    },
    {
      "epoch": 0.756769055745165,
      "grad_norm": 0.8135022202076733,
      "learning_rate": 1.1807081433213122e-06,
      "loss": 0.0432,
      "step": 3326
    },
    {
      "epoch": 0.7569965870307167,
      "grad_norm": 1.4086605703937913,
      "learning_rate": 1.1806672461119637e-06,
      "loss": 0.0774,
      "step": 3327
    },
    {
      "epoch": 0.7572241183162685,
      "grad_norm": 1.7398967165213481,
      "learning_rate": 1.1806263375457947e-06,
      "loss": 0.0598,
      "step": 3328
    },
    {
      "epoch": 0.7574516496018202,
      "grad_norm": 1.4489241741578003,
      "learning_rate": 1.1805854176236406e-06,
      "loss": 0.0655,
      "step": 3329
    },
    {
      "epoch": 0.757679180887372,
      "grad_norm": 2.000585789486283,
      "learning_rate": 1.1805444863463384e-06,
      "loss": 0.1195,
      "step": 3330
    },
    {
      "epoch": 0.7579067121729238,
      "grad_norm": 1.1274809608332625,
      "learning_rate": 1.180503543714724e-06,
      "loss": 0.0355,
      "step": 3331
    },
    {
      "epoch": 0.7581342434584756,
      "grad_norm": 0.8018211568884192,
      "learning_rate": 1.1804625897296345e-06,
      "loss": 0.0354,
      "step": 3332
    },
    {
      "epoch": 0.7583617747440273,
      "grad_norm": 1.8306901932048323,
      "learning_rate": 1.1804216243919074e-06,
      "loss": 0.0604,
      "step": 3333
    },
    {
      "epoch": 0.7585893060295791,
      "grad_norm": 1.0198114562975762,
      "learning_rate": 1.1803806477023792e-06,
      "loss": 0.034,
      "step": 3334
    },
    {
      "epoch": 0.7588168373151308,
      "grad_norm": 1.0883117507836249,
      "learning_rate": 1.1803396596618878e-06,
      "loss": 0.0476,
      "step": 3335
    },
    {
      "epoch": 0.7590443686006826,
      "grad_norm": 1.2568287515264616,
      "learning_rate": 1.1802986602712705e-06,
      "loss": 0.0415,
      "step": 3336
    },
    {
      "epoch": 0.7592718998862343,
      "grad_norm": 1.5230920956011404,
      "learning_rate": 1.1802576495313657e-06,
      "loss": 0.0516,
      "step": 3337
    },
    {
      "epoch": 0.7594994311717861,
      "grad_norm": 3.0018010476610044,
      "learning_rate": 1.1802166274430116e-06,
      "loss": 0.0773,
      "step": 3338
    },
    {
      "epoch": 0.7597269624573378,
      "grad_norm": 0.8881506140039246,
      "learning_rate": 1.1801755940070464e-06,
      "loss": 0.0332,
      "step": 3339
    },
    {
      "epoch": 0.7599544937428896,
      "grad_norm": 1.5510159044642893,
      "learning_rate": 1.1801345492243087e-06,
      "loss": 0.0653,
      "step": 3340
    },
    {
      "epoch": 0.7601820250284415,
      "grad_norm": 1.732180484199383,
      "learning_rate": 1.1800934930956378e-06,
      "loss": 0.0681,
      "step": 3341
    },
    {
      "epoch": 0.7604095563139932,
      "grad_norm": 1.0655757603385652,
      "learning_rate": 1.1800524256218724e-06,
      "loss": 0.0635,
      "step": 3342
    },
    {
      "epoch": 0.760637087599545,
      "grad_norm": 1.3481251487359558,
      "learning_rate": 1.1800113468038518e-06,
      "loss": 0.0685,
      "step": 3343
    },
    {
      "epoch": 0.7608646188850967,
      "grad_norm": 2.2560803054292418,
      "learning_rate": 1.1799702566424159e-06,
      "loss": 0.1302,
      "step": 3344
    },
    {
      "epoch": 0.7610921501706485,
      "grad_norm": 1.6837334743270123,
      "learning_rate": 1.1799291551384042e-06,
      "loss": 0.0842,
      "step": 3345
    },
    {
      "epoch": 0.7613196814562002,
      "grad_norm": 1.3300039883273982,
      "learning_rate": 1.1798880422926568e-06,
      "loss": 0.0518,
      "step": 3346
    },
    {
      "epoch": 0.761547212741752,
      "grad_norm": 1.8839734220786841,
      "learning_rate": 1.1798469181060143e-06,
      "loss": 0.0994,
      "step": 3347
    },
    {
      "epoch": 0.7617747440273037,
      "grad_norm": 1.600150172121613,
      "learning_rate": 1.1798057825793167e-06,
      "loss": 0.0651,
      "step": 3348
    },
    {
      "epoch": 0.7620022753128555,
      "grad_norm": 1.3818697167200884,
      "learning_rate": 1.179764635713405e-06,
      "loss": 0.0473,
      "step": 3349
    },
    {
      "epoch": 0.7622298065984073,
      "grad_norm": 0.8317911257581198,
      "learning_rate": 1.1797234775091204e-06,
      "loss": 0.0337,
      "step": 3350
    },
    {
      "epoch": 0.7624573378839591,
      "grad_norm": 1.1808932366446725,
      "learning_rate": 1.1796823079673036e-06,
      "loss": 0.0376,
      "step": 3351
    },
    {
      "epoch": 0.7626848691695108,
      "grad_norm": 1.6710801191008726,
      "learning_rate": 1.1796411270887965e-06,
      "loss": 0.0592,
      "step": 3352
    },
    {
      "epoch": 0.7629124004550626,
      "grad_norm": 1.0835763404496392,
      "learning_rate": 1.1795999348744403e-06,
      "loss": 0.0525,
      "step": 3353
    },
    {
      "epoch": 0.7631399317406143,
      "grad_norm": 1.2436113476656212,
      "learning_rate": 1.1795587313250773e-06,
      "loss": 0.0451,
      "step": 3354
    },
    {
      "epoch": 0.7633674630261661,
      "grad_norm": 1.1914849618578496,
      "learning_rate": 1.1795175164415493e-06,
      "loss": 0.0578,
      "step": 3355
    },
    {
      "epoch": 0.7635949943117178,
      "grad_norm": 2.3441953010272867,
      "learning_rate": 1.1794762902246988e-06,
      "loss": 0.1105,
      "step": 3356
    },
    {
      "epoch": 0.7638225255972696,
      "grad_norm": 2.1756802775613364,
      "learning_rate": 1.1794350526753688e-06,
      "loss": 0.1026,
      "step": 3357
    },
    {
      "epoch": 0.7640500568828213,
      "grad_norm": 2.3594222993387652,
      "learning_rate": 1.1793938037944014e-06,
      "loss": 0.1163,
      "step": 3358
    },
    {
      "epoch": 0.7642775881683731,
      "grad_norm": 0.9175490163523685,
      "learning_rate": 1.1793525435826399e-06,
      "loss": 0.0287,
      "step": 3359
    },
    {
      "epoch": 0.764505119453925,
      "grad_norm": 1.9177635612847674,
      "learning_rate": 1.1793112720409277e-06,
      "loss": 0.0945,
      "step": 3360
    },
    {
      "epoch": 0.7647326507394767,
      "grad_norm": 1.0136874739515975,
      "learning_rate": 1.1792699891701085e-06,
      "loss": 0.0389,
      "step": 3361
    },
    {
      "epoch": 0.7649601820250285,
      "grad_norm": 0.9880415043802684,
      "learning_rate": 1.1792286949710254e-06,
      "loss": 0.052,
      "step": 3362
    },
    {
      "epoch": 0.7651877133105802,
      "grad_norm": 1.5402287968172736,
      "learning_rate": 1.1791873894445233e-06,
      "loss": 0.0698,
      "step": 3363
    },
    {
      "epoch": 0.765415244596132,
      "grad_norm": 1.3968644273802429,
      "learning_rate": 1.1791460725914455e-06,
      "loss": 0.065,
      "step": 3364
    },
    {
      "epoch": 0.7656427758816837,
      "grad_norm": 1.8023364682651095,
      "learning_rate": 1.179104744412637e-06,
      "loss": 0.0653,
      "step": 3365
    },
    {
      "epoch": 0.7658703071672355,
      "grad_norm": 1.3938733296222532,
      "learning_rate": 1.1790634049089425e-06,
      "loss": 0.0604,
      "step": 3366
    },
    {
      "epoch": 0.7660978384527872,
      "grad_norm": 1.6406363966121917,
      "learning_rate": 1.1790220540812063e-06,
      "loss": 0.0781,
      "step": 3367
    },
    {
      "epoch": 0.766325369738339,
      "grad_norm": 1.2609639774979144,
      "learning_rate": 1.1789806919302743e-06,
      "loss": 0.0429,
      "step": 3368
    },
    {
      "epoch": 0.7665529010238907,
      "grad_norm": 1.4055922666036842,
      "learning_rate": 1.1789393184569914e-06,
      "loss": 0.0447,
      "step": 3369
    },
    {
      "epoch": 0.7667804323094426,
      "grad_norm": 1.2844015411070886,
      "learning_rate": 1.1788979336622034e-06,
      "loss": 0.0686,
      "step": 3370
    },
    {
      "epoch": 0.7670079635949943,
      "grad_norm": 1.2894781437661027,
      "learning_rate": 1.178856537546756e-06,
      "loss": 0.0676,
      "step": 3371
    },
    {
      "epoch": 0.7672354948805461,
      "grad_norm": 1.0960231571013044,
      "learning_rate": 1.1788151301114952e-06,
      "loss": 0.0472,
      "step": 3372
    },
    {
      "epoch": 0.7674630261660979,
      "grad_norm": 1.504396802843696,
      "learning_rate": 1.1787737113572678e-06,
      "loss": 0.0588,
      "step": 3373
    },
    {
      "epoch": 0.7676905574516496,
      "grad_norm": 1.6683614299229446,
      "learning_rate": 1.1787322812849196e-06,
      "loss": 0.0711,
      "step": 3374
    },
    {
      "epoch": 0.7679180887372014,
      "grad_norm": 2.581634805020921,
      "learning_rate": 1.1786908398952977e-06,
      "loss": 0.1279,
      "step": 3375
    },
    {
      "epoch": 0.7681456200227531,
      "grad_norm": 1.1979082190366048,
      "learning_rate": 1.1786493871892491e-06,
      "loss": 0.0387,
      "step": 3376
    },
    {
      "epoch": 0.7683731513083049,
      "grad_norm": 0.7863088959802915,
      "learning_rate": 1.178607923167621e-06,
      "loss": 0.032,
      "step": 3377
    },
    {
      "epoch": 0.7686006825938566,
      "grad_norm": 1.395956623470284,
      "learning_rate": 1.1785664478312607e-06,
      "loss": 0.0557,
      "step": 3378
    },
    {
      "epoch": 0.7688282138794085,
      "grad_norm": 1.3990313414663118,
      "learning_rate": 1.1785249611810163e-06,
      "loss": 0.0675,
      "step": 3379
    },
    {
      "epoch": 0.7690557451649602,
      "grad_norm": 1.5727682457288055,
      "learning_rate": 1.1784834632177352e-06,
      "loss": 0.0488,
      "step": 3380
    },
    {
      "epoch": 0.769283276450512,
      "grad_norm": 1.2997891762689524,
      "learning_rate": 1.178441953942266e-06,
      "loss": 0.0497,
      "step": 3381
    },
    {
      "epoch": 0.7695108077360637,
      "grad_norm": 1.1951369765878084,
      "learning_rate": 1.1784004333554565e-06,
      "loss": 0.043,
      "step": 3382
    },
    {
      "epoch": 0.7697383390216155,
      "grad_norm": 1.3760280462231655,
      "learning_rate": 1.178358901458156e-06,
      "loss": 0.0637,
      "step": 3383
    },
    {
      "epoch": 0.7699658703071672,
      "grad_norm": 1.167678907961144,
      "learning_rate": 1.1783173582512127e-06,
      "loss": 0.0504,
      "step": 3384
    },
    {
      "epoch": 0.770193401592719,
      "grad_norm": 1.1188494063998051,
      "learning_rate": 1.178275803735476e-06,
      "loss": 0.0396,
      "step": 3385
    },
    {
      "epoch": 0.7704209328782707,
      "grad_norm": 1.6858173282426545,
      "learning_rate": 1.1782342379117954e-06,
      "loss": 0.0749,
      "step": 3386
    },
    {
      "epoch": 0.7706484641638225,
      "grad_norm": 1.6532847814641132,
      "learning_rate": 1.17819266078102e-06,
      "loss": 0.0531,
      "step": 3387
    },
    {
      "epoch": 0.7708759954493742,
      "grad_norm": 1.309511727163334,
      "learning_rate": 1.1781510723439995e-06,
      "loss": 0.0619,
      "step": 3388
    },
    {
      "epoch": 0.7711035267349261,
      "grad_norm": 1.8164710685094503,
      "learning_rate": 1.1781094726015842e-06,
      "loss": 0.0849,
      "step": 3389
    },
    {
      "epoch": 0.7713310580204779,
      "grad_norm": 1.5604092892535402,
      "learning_rate": 1.1780678615546245e-06,
      "loss": 0.0718,
      "step": 3390
    },
    {
      "epoch": 0.7715585893060296,
      "grad_norm": 1.3284971139146458,
      "learning_rate": 1.1780262392039706e-06,
      "loss": 0.0532,
      "step": 3391
    },
    {
      "epoch": 0.7717861205915814,
      "grad_norm": 0.8981286940096231,
      "learning_rate": 1.177984605550473e-06,
      "loss": 0.038,
      "step": 3392
    },
    {
      "epoch": 0.7720136518771331,
      "grad_norm": 1.2803703590807574,
      "learning_rate": 1.177942960594983e-06,
      "loss": 0.0381,
      "step": 3393
    },
    {
      "epoch": 0.7722411831626849,
      "grad_norm": 1.0532494793515876,
      "learning_rate": 1.1779013043383516e-06,
      "loss": 0.0274,
      "step": 3394
    },
    {
      "epoch": 0.7724687144482366,
      "grad_norm": 1.0370979550305721,
      "learning_rate": 1.17785963678143e-06,
      "loss": 0.0527,
      "step": 3395
    },
    {
      "epoch": 0.7726962457337884,
      "grad_norm": 0.8794501213484224,
      "learning_rate": 1.1778179579250699e-06,
      "loss": 0.0252,
      "step": 3396
    },
    {
      "epoch": 0.7729237770193401,
      "grad_norm": 1.2333346024054503,
      "learning_rate": 1.1777762677701232e-06,
      "loss": 0.0443,
      "step": 3397
    },
    {
      "epoch": 0.7731513083048919,
      "grad_norm": 1.0027591833023217,
      "learning_rate": 1.1777345663174419e-06,
      "loss": 0.0689,
      "step": 3398
    },
    {
      "epoch": 0.7733788395904437,
      "grad_norm": 1.1808711247974837,
      "learning_rate": 1.1776928535678784e-06,
      "loss": 0.0575,
      "step": 3399
    },
    {
      "epoch": 0.7736063708759955,
      "grad_norm": 1.3093615145015158,
      "learning_rate": 1.1776511295222852e-06,
      "loss": 0.0602,
      "step": 3400
    },
    {
      "epoch": 0.7738339021615472,
      "grad_norm": 1.488841199985786,
      "learning_rate": 1.177609394181515e-06,
      "loss": 0.0713,
      "step": 3401
    },
    {
      "epoch": 0.774061433447099,
      "grad_norm": 2.280972914382851,
      "learning_rate": 1.177567647546421e-06,
      "loss": 0.1191,
      "step": 3402
    },
    {
      "epoch": 0.7742889647326507,
      "grad_norm": 1.6714940535505132,
      "learning_rate": 1.177525889617856e-06,
      "loss": 0.0872,
      "step": 3403
    },
    {
      "epoch": 0.7745164960182025,
      "grad_norm": 0.7472197048359,
      "learning_rate": 1.177484120396674e-06,
      "loss": 0.0307,
      "step": 3404
    },
    {
      "epoch": 0.7747440273037542,
      "grad_norm": 0.9070022095243302,
      "learning_rate": 1.1774423398837282e-06,
      "loss": 0.0269,
      "step": 3405
    },
    {
      "epoch": 0.774971558589306,
      "grad_norm": 0.8847029817903657,
      "learning_rate": 1.177400548079873e-06,
      "loss": 0.038,
      "step": 3406
    },
    {
      "epoch": 0.7751990898748577,
      "grad_norm": 1.0038791040094233,
      "learning_rate": 1.177358744985962e-06,
      "loss": 0.0514,
      "step": 3407
    },
    {
      "epoch": 0.7754266211604095,
      "grad_norm": 0.9653957279075387,
      "learning_rate": 1.1773169306028498e-06,
      "loss": 0.0376,
      "step": 3408
    },
    {
      "epoch": 0.7756541524459614,
      "grad_norm": 1.025381489223499,
      "learning_rate": 1.1772751049313911e-06,
      "loss": 0.0404,
      "step": 3409
    },
    {
      "epoch": 0.7758816837315131,
      "grad_norm": 4.749905802391231,
      "learning_rate": 1.1772332679724408e-06,
      "loss": 0.0609,
      "step": 3410
    },
    {
      "epoch": 0.7761092150170649,
      "grad_norm": 0.9550524873306326,
      "learning_rate": 1.1771914197268538e-06,
      "loss": 0.0303,
      "step": 3411
    },
    {
      "epoch": 0.7763367463026166,
      "grad_norm": 0.7920355825602112,
      "learning_rate": 1.1771495601954856e-06,
      "loss": 0.0357,
      "step": 3412
    },
    {
      "epoch": 0.7765642775881684,
      "grad_norm": 1.0204752879838752,
      "learning_rate": 1.1771076893791914e-06,
      "loss": 0.0296,
      "step": 3413
    },
    {
      "epoch": 0.7767918088737201,
      "grad_norm": 1.1240956814281524,
      "learning_rate": 1.1770658072788272e-06,
      "loss": 0.0326,
      "step": 3414
    },
    {
      "epoch": 0.7770193401592719,
      "grad_norm": 2.188614006253551,
      "learning_rate": 1.1770239138952492e-06,
      "loss": 0.0814,
      "step": 3415
    },
    {
      "epoch": 0.7772468714448236,
      "grad_norm": 2.1028001694138916,
      "learning_rate": 1.1769820092293132e-06,
      "loss": 0.1016,
      "step": 3416
    },
    {
      "epoch": 0.7774744027303754,
      "grad_norm": 1.3041493538358495,
      "learning_rate": 1.1769400932818758e-06,
      "loss": 0.0689,
      "step": 3417
    },
    {
      "epoch": 0.7777019340159272,
      "grad_norm": 0.8082774287076608,
      "learning_rate": 1.1768981660537938e-06,
      "loss": 0.0465,
      "step": 3418
    },
    {
      "epoch": 0.777929465301479,
      "grad_norm": 0.9743628621052991,
      "learning_rate": 1.1768562275459242e-06,
      "loss": 0.0515,
      "step": 3419
    },
    {
      "epoch": 0.7781569965870307,
      "grad_norm": 1.3767418297539598,
      "learning_rate": 1.1768142777591235e-06,
      "loss": 0.0765,
      "step": 3420
    },
    {
      "epoch": 0.7783845278725825,
      "grad_norm": 3.29832363084199,
      "learning_rate": 1.17677231669425e-06,
      "loss": 0.1072,
      "step": 3421
    },
    {
      "epoch": 0.7786120591581343,
      "grad_norm": 1.3112653588991108,
      "learning_rate": 1.1767303443521608e-06,
      "loss": 0.0412,
      "step": 3422
    },
    {
      "epoch": 0.778839590443686,
      "grad_norm": 1.3040229403784145,
      "learning_rate": 1.1766883607337137e-06,
      "loss": 0.0748,
      "step": 3423
    },
    {
      "epoch": 0.7790671217292378,
      "grad_norm": 1.5632822983270742,
      "learning_rate": 1.176646365839767e-06,
      "loss": 0.0565,
      "step": 3424
    },
    {
      "epoch": 0.7792946530147895,
      "grad_norm": 0.8997709014511377,
      "learning_rate": 1.1766043596711787e-06,
      "loss": 0.0612,
      "step": 3425
    },
    {
      "epoch": 0.7795221843003413,
      "grad_norm": 1.0705681786627015,
      "learning_rate": 1.1765623422288078e-06,
      "loss": 0.0353,
      "step": 3426
    },
    {
      "epoch": 0.779749715585893,
      "grad_norm": 0.9644671179029716,
      "learning_rate": 1.1765203135135126e-06,
      "loss": 0.0371,
      "step": 3427
    },
    {
      "epoch": 0.7799772468714449,
      "grad_norm": 1.1638233882315414,
      "learning_rate": 1.176478273526152e-06,
      "loss": 0.0519,
      "step": 3428
    },
    {
      "epoch": 0.7802047781569966,
      "grad_norm": 1.6611610210478276,
      "learning_rate": 1.1764362222675857e-06,
      "loss": 0.0733,
      "step": 3429
    },
    {
      "epoch": 0.7804323094425484,
      "grad_norm": 1.3309978458504417,
      "learning_rate": 1.176394159738673e-06,
      "loss": 0.0658,
      "step": 3430
    },
    {
      "epoch": 0.7806598407281001,
      "grad_norm": 1.379255981946512,
      "learning_rate": 1.1763520859402735e-06,
      "loss": 0.0803,
      "step": 3431
    },
    {
      "epoch": 0.7808873720136519,
      "grad_norm": 1.2786437648131908,
      "learning_rate": 1.176310000873247e-06,
      "loss": 0.0523,
      "step": 3432
    },
    {
      "epoch": 0.7811149032992036,
      "grad_norm": 1.382541657289892,
      "learning_rate": 1.1762679045384537e-06,
      "loss": 0.0593,
      "step": 3433
    },
    {
      "epoch": 0.7813424345847554,
      "grad_norm": 1.141255046714792,
      "learning_rate": 1.1762257969367543e-06,
      "loss": 0.0404,
      "step": 3434
    },
    {
      "epoch": 0.7815699658703071,
      "grad_norm": 1.6145161686719605,
      "learning_rate": 1.176183678069009e-06,
      "loss": 0.0636,
      "step": 3435
    },
    {
      "epoch": 0.7817974971558589,
      "grad_norm": 2.0980684433512,
      "learning_rate": 1.1761415479360784e-06,
      "loss": 0.0895,
      "step": 3436
    },
    {
      "epoch": 0.7820250284414106,
      "grad_norm": 1.1114214017016597,
      "learning_rate": 1.1760994065388246e-06,
      "loss": 0.0443,
      "step": 3437
    },
    {
      "epoch": 0.7822525597269625,
      "grad_norm": 1.8826452294482943,
      "learning_rate": 1.1760572538781077e-06,
      "loss": 0.0543,
      "step": 3438
    },
    {
      "epoch": 0.7824800910125143,
      "grad_norm": 0.8873377874884448,
      "learning_rate": 1.17601508995479e-06,
      "loss": 0.0376,
      "step": 3439
    },
    {
      "epoch": 0.782707622298066,
      "grad_norm": 1.113759690507731,
      "learning_rate": 1.175972914769733e-06,
      "loss": 0.0387,
      "step": 3440
    },
    {
      "epoch": 0.7829351535836178,
      "grad_norm": 0.880447245871124,
      "learning_rate": 1.1759307283237986e-06,
      "loss": 0.0364,
      "step": 3441
    },
    {
      "epoch": 0.7831626848691695,
      "grad_norm": 0.687251787996786,
      "learning_rate": 1.175888530617849e-06,
      "loss": 0.0272,
      "step": 3442
    },
    {
      "epoch": 0.7833902161547213,
      "grad_norm": 1.9024845236880248,
      "learning_rate": 1.175846321652747e-06,
      "loss": 0.0815,
      "step": 3443
    },
    {
      "epoch": 0.783617747440273,
      "grad_norm": 0.9590041079160219,
      "learning_rate": 1.1758041014293548e-06,
      "loss": 0.0503,
      "step": 3444
    },
    {
      "epoch": 0.7838452787258248,
      "grad_norm": 0.8857462180203992,
      "learning_rate": 1.1757618699485353e-06,
      "loss": 0.0472,
      "step": 3445
    },
    {
      "epoch": 0.7840728100113765,
      "grad_norm": 1.299303333784786,
      "learning_rate": 1.1757196272111524e-06,
      "loss": 0.0555,
      "step": 3446
    },
    {
      "epoch": 0.7843003412969284,
      "grad_norm": 0.845660192083495,
      "learning_rate": 1.1756773732180684e-06,
      "loss": 0.025,
      "step": 3447
    },
    {
      "epoch": 0.7845278725824801,
      "grad_norm": 2.1486155155967737,
      "learning_rate": 1.1756351079701477e-06,
      "loss": 0.086,
      "step": 3448
    },
    {
      "epoch": 0.7847554038680319,
      "grad_norm": 1.373331795777772,
      "learning_rate": 1.1755928314682537e-06,
      "loss": 0.0432,
      "step": 3449
    },
    {
      "epoch": 0.7849829351535836,
      "grad_norm": 0.8852881088520247,
      "learning_rate": 1.1755505437132507e-06,
      "loss": 0.0291,
      "step": 3450
    },
    {
      "epoch": 0.7852104664391354,
      "grad_norm": 2.454713357414872,
      "learning_rate": 1.1755082447060029e-06,
      "loss": 0.1266,
      "step": 3451
    },
    {
      "epoch": 0.7854379977246871,
      "grad_norm": 1.4196724985388591,
      "learning_rate": 1.1754659344473747e-06,
      "loss": 0.0548,
      "step": 3452
    },
    {
      "epoch": 0.7856655290102389,
      "grad_norm": 1.0229434892505915,
      "learning_rate": 1.175423612938231e-06,
      "loss": 0.0558,
      "step": 3453
    },
    {
      "epoch": 0.7858930602957906,
      "grad_norm": 1.3800295427652078,
      "learning_rate": 1.1753812801794368e-06,
      "loss": 0.0617,
      "step": 3454
    },
    {
      "epoch": 0.7861205915813424,
      "grad_norm": 1.5307677991589903,
      "learning_rate": 1.175338936171857e-06,
      "loss": 0.0551,
      "step": 3455
    },
    {
      "epoch": 0.7863481228668942,
      "grad_norm": 1.1267953232901513,
      "learning_rate": 1.1752965809163574e-06,
      "loss": 0.0427,
      "step": 3456
    },
    {
      "epoch": 0.786575654152446,
      "grad_norm": 1.227541921588263,
      "learning_rate": 1.1752542144138033e-06,
      "loss": 0.0486,
      "step": 3457
    },
    {
      "epoch": 0.7868031854379978,
      "grad_norm": 1.679516124458468,
      "learning_rate": 1.1752118366650608e-06,
      "loss": 0.0589,
      "step": 3458
    },
    {
      "epoch": 0.7870307167235495,
      "grad_norm": 1.0236231221195786,
      "learning_rate": 1.1751694476709962e-06,
      "loss": 0.0306,
      "step": 3459
    },
    {
      "epoch": 0.7872582480091013,
      "grad_norm": 1.3331183973827108,
      "learning_rate": 1.1751270474324757e-06,
      "loss": 0.0651,
      "step": 3460
    },
    {
      "epoch": 0.787485779294653,
      "grad_norm": 1.3868556370726,
      "learning_rate": 1.1750846359503657e-06,
      "loss": 0.0753,
      "step": 3461
    },
    {
      "epoch": 0.7877133105802048,
      "grad_norm": 1.2017443079409735,
      "learning_rate": 1.1750422132255334e-06,
      "loss": 0.0626,
      "step": 3462
    },
    {
      "epoch": 0.7879408418657565,
      "grad_norm": 1.4474690077464614,
      "learning_rate": 1.1749997792588453e-06,
      "loss": 0.0449,
      "step": 3463
    },
    {
      "epoch": 0.7881683731513083,
      "grad_norm": 1.2518797374298931,
      "learning_rate": 1.1749573340511693e-06,
      "loss": 0.0621,
      "step": 3464
    },
    {
      "epoch": 0.78839590443686,
      "grad_norm": 1.639438514025765,
      "learning_rate": 1.1749148776033723e-06,
      "loss": 0.063,
      "step": 3465
    },
    {
      "epoch": 0.7886234357224118,
      "grad_norm": 0.8717910913192211,
      "learning_rate": 1.1748724099163225e-06,
      "loss": 0.042,
      "step": 3466
    },
    {
      "epoch": 0.7888509670079636,
      "grad_norm": 1.144252834465786,
      "learning_rate": 1.1748299309908878e-06,
      "loss": 0.0396,
      "step": 3467
    },
    {
      "epoch": 0.7890784982935154,
      "grad_norm": 1.2552868730516495,
      "learning_rate": 1.174787440827936e-06,
      "loss": 0.0561,
      "step": 3468
    },
    {
      "epoch": 0.7893060295790671,
      "grad_norm": 0.9915360367623126,
      "learning_rate": 1.1747449394283361e-06,
      "loss": 0.0417,
      "step": 3469
    },
    {
      "epoch": 0.7895335608646189,
      "grad_norm": 2.2809743574627612,
      "learning_rate": 1.174702426792956e-06,
      "loss": 0.122,
      "step": 3470
    },
    {
      "epoch": 0.7897610921501707,
      "grad_norm": 0.9319050863528758,
      "learning_rate": 1.1746599029226656e-06,
      "loss": 0.024,
      "step": 3471
    },
    {
      "epoch": 0.7899886234357224,
      "grad_norm": 1.0243621293291936,
      "learning_rate": 1.174617367818333e-06,
      "loss": 0.0334,
      "step": 3472
    },
    {
      "epoch": 0.7902161547212742,
      "grad_norm": 0.5162499932362775,
      "learning_rate": 1.174574821480828e-06,
      "loss": 0.028,
      "step": 3473
    },
    {
      "epoch": 0.7904436860068259,
      "grad_norm": 1.1251631282315664,
      "learning_rate": 1.1745322639110203e-06,
      "loss": 0.0314,
      "step": 3474
    },
    {
      "epoch": 0.7906712172923777,
      "grad_norm": 0.881794120596432,
      "learning_rate": 1.1744896951097798e-06,
      "loss": 0.0453,
      "step": 3475
    },
    {
      "epoch": 0.7908987485779294,
      "grad_norm": 0.7316213076139558,
      "learning_rate": 1.1744471150779758e-06,
      "loss": 0.0183,
      "step": 3476
    },
    {
      "epoch": 0.7911262798634813,
      "grad_norm": 1.6604024020521415,
      "learning_rate": 1.1744045238164793e-06,
      "loss": 0.0587,
      "step": 3477
    },
    {
      "epoch": 0.791353811149033,
      "grad_norm": 1.3826539577284072,
      "learning_rate": 1.1743619213261604e-06,
      "loss": 0.0626,
      "step": 3478
    },
    {
      "epoch": 0.7915813424345848,
      "grad_norm": 0.9873190669247738,
      "learning_rate": 1.1743193076078901e-06,
      "loss": 0.0407,
      "step": 3479
    },
    {
      "epoch": 0.7918088737201365,
      "grad_norm": 2.2069878657118243,
      "learning_rate": 1.174276682662539e-06,
      "loss": 0.0907,
      "step": 3480
    },
    {
      "epoch": 0.7920364050056883,
      "grad_norm": 0.9695329009047923,
      "learning_rate": 1.1742340464909786e-06,
      "loss": 0.0444,
      "step": 3481
    },
    {
      "epoch": 0.79226393629124,
      "grad_norm": 1.18743378027036,
      "learning_rate": 1.1741913990940801e-06,
      "loss": 0.0417,
      "step": 3482
    },
    {
      "epoch": 0.7924914675767918,
      "grad_norm": 1.2335459566427895,
      "learning_rate": 1.174148740472715e-06,
      "loss": 0.0357,
      "step": 3483
    },
    {
      "epoch": 0.7927189988623435,
      "grad_norm": 1.7123789864089465,
      "learning_rate": 1.1741060706277557e-06,
      "loss": 0.0549,
      "step": 3484
    },
    {
      "epoch": 0.7929465301478953,
      "grad_norm": 1.0088683967525243,
      "learning_rate": 1.1740633895600738e-06,
      "loss": 0.0487,
      "step": 3485
    },
    {
      "epoch": 0.7931740614334472,
      "grad_norm": 0.9729774800349912,
      "learning_rate": 1.1740206972705418e-06,
      "loss": 0.0487,
      "step": 3486
    },
    {
      "epoch": 0.7934015927189989,
      "grad_norm": 1.357974224485339,
      "learning_rate": 1.1739779937600323e-06,
      "loss": 0.0501,
      "step": 3487
    },
    {
      "epoch": 0.7936291240045507,
      "grad_norm": 1.2508424825549256,
      "learning_rate": 1.1739352790294177e-06,
      "loss": 0.0373,
      "step": 3488
    },
    {
      "epoch": 0.7938566552901024,
      "grad_norm": 1.3686653687913515,
      "learning_rate": 1.1738925530795716e-06,
      "loss": 0.0517,
      "step": 3489
    },
    {
      "epoch": 0.7940841865756542,
      "grad_norm": 1.3919271731699832,
      "learning_rate": 1.173849815911367e-06,
      "loss": 0.0513,
      "step": 3490
    },
    {
      "epoch": 0.7943117178612059,
      "grad_norm": 1.0732983866451746,
      "learning_rate": 1.1738070675256773e-06,
      "loss": 0.0485,
      "step": 3491
    },
    {
      "epoch": 0.7945392491467577,
      "grad_norm": 0.8920921807243749,
      "learning_rate": 1.1737643079233763e-06,
      "loss": 0.0423,
      "step": 3492
    },
    {
      "epoch": 0.7947667804323094,
      "grad_norm": 0.8985878758463217,
      "learning_rate": 1.1737215371053376e-06,
      "loss": 0.0315,
      "step": 3493
    },
    {
      "epoch": 0.7949943117178612,
      "grad_norm": 1.4217871233611445,
      "learning_rate": 1.1736787550724357e-06,
      "loss": 0.0634,
      "step": 3494
    },
    {
      "epoch": 0.7952218430034129,
      "grad_norm": 1.6601308191313218,
      "learning_rate": 1.1736359618255452e-06,
      "loss": 0.072,
      "step": 3495
    },
    {
      "epoch": 0.7954493742889648,
      "grad_norm": 1.9516506015444255,
      "learning_rate": 1.1735931573655402e-06,
      "loss": 0.0845,
      "step": 3496
    },
    {
      "epoch": 0.7956769055745165,
      "grad_norm": 13.731764530376946,
      "learning_rate": 1.1735503416932957e-06,
      "loss": 0.3361,
      "step": 3497
    },
    {
      "epoch": 0.7959044368600683,
      "grad_norm": 0.8458278086989922,
      "learning_rate": 1.1735075148096869e-06,
      "loss": 0.0459,
      "step": 3498
    },
    {
      "epoch": 0.79613196814562,
      "grad_norm": 1.0325577020668653,
      "learning_rate": 1.173464676715589e-06,
      "loss": 0.0491,
      "step": 3499
    },
    {
      "epoch": 0.7963594994311718,
      "grad_norm": 1.0941809224515442,
      "learning_rate": 1.1734218274118775e-06,
      "loss": 0.0509,
      "step": 3500
    },
    {
      "epoch": 0.7965870307167235,
      "grad_norm": 1.4998412599053896,
      "learning_rate": 1.1733789668994285e-06,
      "loss": 0.0674,
      "step": 3501
    },
    {
      "epoch": 0.7968145620022753,
      "grad_norm": 1.4510084408943071,
      "learning_rate": 1.1733360951791176e-06,
      "loss": 0.078,
      "step": 3502
    },
    {
      "epoch": 0.797042093287827,
      "grad_norm": 0.8199454039615341,
      "learning_rate": 1.173293212251821e-06,
      "loss": 0.0277,
      "step": 3503
    },
    {
      "epoch": 0.7972696245733788,
      "grad_norm": 1.4797806837589549,
      "learning_rate": 1.1732503181184155e-06,
      "loss": 0.0457,
      "step": 3504
    },
    {
      "epoch": 0.7974971558589306,
      "grad_norm": 1.6954088656786952,
      "learning_rate": 1.1732074127797773e-06,
      "loss": 0.0431,
      "step": 3505
    },
    {
      "epoch": 0.7977246871444824,
      "grad_norm": 0.78592343457484,
      "learning_rate": 1.1731644962367838e-06,
      "loss": 0.0289,
      "step": 3506
    },
    {
      "epoch": 0.7979522184300342,
      "grad_norm": 1.0800031446768374,
      "learning_rate": 1.173121568490312e-06,
      "loss": 0.068,
      "step": 3507
    },
    {
      "epoch": 0.7981797497155859,
      "grad_norm": 1.4924469013770905,
      "learning_rate": 1.173078629541239e-06,
      "loss": 0.0986,
      "step": 3508
    },
    {
      "epoch": 0.7984072810011377,
      "grad_norm": 1.0339163270277598,
      "learning_rate": 1.1730356793904426e-06,
      "loss": 0.0393,
      "step": 3509
    },
    {
      "epoch": 0.7986348122866894,
      "grad_norm": 2.1650674647440002,
      "learning_rate": 1.1729927180388008e-06,
      "loss": 0.0748,
      "step": 3510
    },
    {
      "epoch": 0.7988623435722412,
      "grad_norm": 1.3477613309939473,
      "learning_rate": 1.172949745487191e-06,
      "loss": 0.0511,
      "step": 3511
    },
    {
      "epoch": 0.7990898748577929,
      "grad_norm": 1.941320037871878,
      "learning_rate": 1.1729067617364923e-06,
      "loss": 0.0653,
      "step": 3512
    },
    {
      "epoch": 0.7993174061433447,
      "grad_norm": 1.3711705573827695,
      "learning_rate": 1.1728637667875825e-06,
      "loss": 0.0583,
      "step": 3513
    },
    {
      "epoch": 0.7995449374288964,
      "grad_norm": 1.4216556543524117,
      "learning_rate": 1.172820760641341e-06,
      "loss": 0.0767,
      "step": 3514
    },
    {
      "epoch": 0.7997724687144482,
      "grad_norm": 1.0065561289408549,
      "learning_rate": 1.1727777432986465e-06,
      "loss": 0.0469,
      "step": 3515
    },
    {
      "epoch": 0.8,
      "grad_norm": 1.3114054375469062,
      "learning_rate": 1.1727347147603778e-06,
      "loss": 0.051,
      "step": 3516
    },
    {
      "epoch": 0.8002275312855518,
      "grad_norm": 1.1635377251522268,
      "learning_rate": 1.1726916750274148e-06,
      "loss": 0.0425,
      "step": 3517
    },
    {
      "epoch": 0.8004550625711035,
      "grad_norm": 0.7109505941253476,
      "learning_rate": 1.172648624100637e-06,
      "loss": 0.0335,
      "step": 3518
    },
    {
      "epoch": 0.8006825938566553,
      "grad_norm": 2.140244010424768,
      "learning_rate": 1.1726055619809245e-06,
      "loss": 0.0764,
      "step": 3519
    },
    {
      "epoch": 0.800910125142207,
      "grad_norm": 1.444555156138925,
      "learning_rate": 1.172562488669157e-06,
      "loss": 0.0618,
      "step": 3520
    },
    {
      "epoch": 0.8011376564277588,
      "grad_norm": 1.7685121918701119,
      "learning_rate": 1.172519404166215e-06,
      "loss": 0.0857,
      "step": 3521
    },
    {
      "epoch": 0.8013651877133106,
      "grad_norm": 1.713827324352964,
      "learning_rate": 1.1724763084729792e-06,
      "loss": 0.0633,
      "step": 3522
    },
    {
      "epoch": 0.8015927189988623,
      "grad_norm": 1.1050834097261653,
      "learning_rate": 1.1724332015903303e-06,
      "loss": 0.0519,
      "step": 3523
    },
    {
      "epoch": 0.8018202502844141,
      "grad_norm": 1.6510556663604237,
      "learning_rate": 1.1723900835191494e-06,
      "loss": 0.0789,
      "step": 3524
    },
    {
      "epoch": 0.8020477815699659,
      "grad_norm": 0.9667126871814873,
      "learning_rate": 1.1723469542603174e-06,
      "loss": 0.042,
      "step": 3525
    },
    {
      "epoch": 0.8022753128555177,
      "grad_norm": 0.8657263063272098,
      "learning_rate": 1.1723038138147165e-06,
      "loss": 0.0313,
      "step": 3526
    },
    {
      "epoch": 0.8025028441410694,
      "grad_norm": 1.856452808149009,
      "learning_rate": 1.1722606621832278e-06,
      "loss": 0.0896,
      "step": 3527
    },
    {
      "epoch": 0.8027303754266212,
      "grad_norm": 0.6848148476167641,
      "learning_rate": 1.1722174993667335e-06,
      "loss": 0.0183,
      "step": 3528
    },
    {
      "epoch": 0.8029579067121729,
      "grad_norm": 1.019323438363341,
      "learning_rate": 1.1721743253661155e-06,
      "loss": 0.051,
      "step": 3529
    },
    {
      "epoch": 0.8031854379977247,
      "grad_norm": 0.6693914231036064,
      "learning_rate": 1.1721311401822566e-06,
      "loss": 0.0232,
      "step": 3530
    },
    {
      "epoch": 0.8034129692832764,
      "grad_norm": 0.920308592294961,
      "learning_rate": 1.1720879438160391e-06,
      "loss": 0.0305,
      "step": 3531
    },
    {
      "epoch": 0.8036405005688282,
      "grad_norm": 1.3695873071245632,
      "learning_rate": 1.1720447362683458e-06,
      "loss": 0.045,
      "step": 3532
    },
    {
      "epoch": 0.8038680318543799,
      "grad_norm": 1.7871009286158004,
      "learning_rate": 1.1720015175400602e-06,
      "loss": 0.0834,
      "step": 3533
    },
    {
      "epoch": 0.8040955631399317,
      "grad_norm": 0.6832287827660394,
      "learning_rate": 1.1719582876320655e-06,
      "loss": 0.0263,
      "step": 3534
    },
    {
      "epoch": 0.8043230944254836,
      "grad_norm": 0.7627184412086807,
      "learning_rate": 1.1719150465452447e-06,
      "loss": 0.0262,
      "step": 3535
    },
    {
      "epoch": 0.8045506257110353,
      "grad_norm": 1.716803621327735,
      "learning_rate": 1.1718717942804822e-06,
      "loss": 0.0734,
      "step": 3536
    },
    {
      "epoch": 0.8047781569965871,
      "grad_norm": 2.0486680528761854,
      "learning_rate": 1.1718285308386618e-06,
      "loss": 0.1516,
      "step": 3537
    },
    {
      "epoch": 0.8050056882821388,
      "grad_norm": 1.3104153747023377,
      "learning_rate": 1.1717852562206678e-06,
      "loss": 0.0414,
      "step": 3538
    },
    {
      "epoch": 0.8052332195676906,
      "grad_norm": 1.5524472443736617,
      "learning_rate": 1.1717419704273844e-06,
      "loss": 0.0532,
      "step": 3539
    },
    {
      "epoch": 0.8054607508532423,
      "grad_norm": 1.8181142023265993,
      "learning_rate": 1.1716986734596963e-06,
      "loss": 0.074,
      "step": 3540
    },
    {
      "epoch": 0.8056882821387941,
      "grad_norm": 1.5770991987541603,
      "learning_rate": 1.1716553653184887e-06,
      "loss": 0.0572,
      "step": 3541
    },
    {
      "epoch": 0.8059158134243458,
      "grad_norm": 1.603046856744777,
      "learning_rate": 1.1716120460046464e-06,
      "loss": 0.068,
      "step": 3542
    },
    {
      "epoch": 0.8061433447098976,
      "grad_norm": 0.9490597103722859,
      "learning_rate": 1.1715687155190553e-06,
      "loss": 0.044,
      "step": 3543
    },
    {
      "epoch": 0.8063708759954493,
      "grad_norm": 1.4475288739954406,
      "learning_rate": 1.1715253738626005e-06,
      "loss": 0.0859,
      "step": 3544
    },
    {
      "epoch": 0.8065984072810012,
      "grad_norm": 2.4651858821386132,
      "learning_rate": 1.171482021036168e-06,
      "loss": 0.1099,
      "step": 3545
    },
    {
      "epoch": 0.8068259385665529,
      "grad_norm": 0.6592130419111449,
      "learning_rate": 1.1714386570406439e-06,
      "loss": 0.0271,
      "step": 3546
    },
    {
      "epoch": 0.8070534698521047,
      "grad_norm": 1.7063955755708686,
      "learning_rate": 1.1713952818769142e-06,
      "loss": 0.111,
      "step": 3547
    },
    {
      "epoch": 0.8072810011376564,
      "grad_norm": 2.169506097036596,
      "learning_rate": 1.1713518955458657e-06,
      "loss": 0.0733,
      "step": 3548
    },
    {
      "epoch": 0.8075085324232082,
      "grad_norm": 1.2288435774277289,
      "learning_rate": 1.1713084980483849e-06,
      "loss": 0.042,
      "step": 3549
    },
    {
      "epoch": 0.8077360637087599,
      "grad_norm": 1.577397877205651,
      "learning_rate": 1.1712650893853591e-06,
      "loss": 0.0831,
      "step": 3550
    },
    {
      "epoch": 0.8079635949943117,
      "grad_norm": 1.1381216086876993,
      "learning_rate": 1.1712216695576753e-06,
      "loss": 0.0531,
      "step": 3551
    },
    {
      "epoch": 0.8081911262798634,
      "grad_norm": 1.1404596418271893,
      "learning_rate": 1.171178238566221e-06,
      "loss": 0.043,
      "step": 3552
    },
    {
      "epoch": 0.8084186575654152,
      "grad_norm": 1.317622472587367,
      "learning_rate": 1.1711347964118837e-06,
      "loss": 0.0529,
      "step": 3553
    },
    {
      "epoch": 0.8086461888509671,
      "grad_norm": 1.065643129987413,
      "learning_rate": 1.1710913430955514e-06,
      "loss": 0.0375,
      "step": 3554
    },
    {
      "epoch": 0.8088737201365188,
      "grad_norm": 1.3409199390597606,
      "learning_rate": 1.171047878618112e-06,
      "loss": 0.0424,
      "step": 3555
    },
    {
      "epoch": 0.8091012514220706,
      "grad_norm": 1.795521953652107,
      "learning_rate": 1.1710044029804542e-06,
      "loss": 0.0649,
      "step": 3556
    },
    {
      "epoch": 0.8093287827076223,
      "grad_norm": 1.4702215735242095,
      "learning_rate": 1.1709609161834663e-06,
      "loss": 0.0714,
      "step": 3557
    },
    {
      "epoch": 0.8095563139931741,
      "grad_norm": 1.4542304303142697,
      "learning_rate": 1.1709174182280371e-06,
      "loss": 0.074,
      "step": 3558
    },
    {
      "epoch": 0.8097838452787258,
      "grad_norm": 0.9222037242585802,
      "learning_rate": 1.1708739091150557e-06,
      "loss": 0.0479,
      "step": 3559
    },
    {
      "epoch": 0.8100113765642776,
      "grad_norm": 1.1706470717553774,
      "learning_rate": 1.1708303888454113e-06,
      "loss": 0.0499,
      "step": 3560
    },
    {
      "epoch": 0.8102389078498293,
      "grad_norm": 1.4734537784856048,
      "learning_rate": 1.1707868574199934e-06,
      "loss": 0.0684,
      "step": 3561
    },
    {
      "epoch": 0.8104664391353811,
      "grad_norm": 2.40284003722661,
      "learning_rate": 1.1707433148396918e-06,
      "loss": 0.1156,
      "step": 3562
    },
    {
      "epoch": 0.8106939704209328,
      "grad_norm": 1.283544001327138,
      "learning_rate": 1.1706997611053963e-06,
      "loss": 0.041,
      "step": 3563
    },
    {
      "epoch": 0.8109215017064847,
      "grad_norm": 1.2655368943151744,
      "learning_rate": 1.1706561962179968e-06,
      "loss": 0.0517,
      "step": 3564
    },
    {
      "epoch": 0.8111490329920364,
      "grad_norm": 1.2742964978863571,
      "learning_rate": 1.1706126201783844e-06,
      "loss": 0.0601,
      "step": 3565
    },
    {
      "epoch": 0.8113765642775882,
      "grad_norm": 1.1638624014319168,
      "learning_rate": 1.170569032987449e-06,
      "loss": 0.0408,
      "step": 3566
    },
    {
      "epoch": 0.81160409556314,
      "grad_norm": 1.3521980505141613,
      "learning_rate": 1.1705254346460818e-06,
      "loss": 0.0494,
      "step": 3567
    },
    {
      "epoch": 0.8118316268486917,
      "grad_norm": 1.002566765933997,
      "learning_rate": 1.170481825155174e-06,
      "loss": 0.0608,
      "step": 3568
    },
    {
      "epoch": 0.8120591581342435,
      "grad_norm": 1.0785551144421794,
      "learning_rate": 1.1704382045156162e-06,
      "loss": 0.0443,
      "step": 3569
    },
    {
      "epoch": 0.8122866894197952,
      "grad_norm": 0.9426048538607938,
      "learning_rate": 1.1703945727283008e-06,
      "loss": 0.0293,
      "step": 3570
    },
    {
      "epoch": 0.812514220705347,
      "grad_norm": 0.9121003153189483,
      "learning_rate": 1.1703509297941193e-06,
      "loss": 0.0469,
      "step": 3571
    },
    {
      "epoch": 0.8127417519908987,
      "grad_norm": 1.1024950352965313,
      "learning_rate": 1.1703072757139633e-06,
      "loss": 0.0432,
      "step": 3572
    },
    {
      "epoch": 0.8129692832764505,
      "grad_norm": 1.5884989646335343,
      "learning_rate": 1.1702636104887253e-06,
      "loss": 0.0567,
      "step": 3573
    },
    {
      "epoch": 0.8131968145620023,
      "grad_norm": 1.3193688692811854,
      "learning_rate": 1.1702199341192977e-06,
      "loss": 0.0492,
      "step": 3574
    },
    {
      "epoch": 0.8134243458475541,
      "grad_norm": 1.278989450905917,
      "learning_rate": 1.1701762466065733e-06,
      "loss": 0.0673,
      "step": 3575
    },
    {
      "epoch": 0.8136518771331058,
      "grad_norm": 1.2742872087629682,
      "learning_rate": 1.1701325479514446e-06,
      "loss": 0.0515,
      "step": 3576
    },
    {
      "epoch": 0.8138794084186576,
      "grad_norm": 1.0146683924087292,
      "learning_rate": 1.1700888381548052e-06,
      "loss": 0.0313,
      "step": 3577
    },
    {
      "epoch": 0.8141069397042093,
      "grad_norm": 1.2456789380747535,
      "learning_rate": 1.1700451172175482e-06,
      "loss": 0.0795,
      "step": 3578
    },
    {
      "epoch": 0.8143344709897611,
      "grad_norm": 1.0481095629438555,
      "learning_rate": 1.1700013851405673e-06,
      "loss": 0.0388,
      "step": 3579
    },
    {
      "epoch": 0.8145620022753128,
      "grad_norm": 2.2650366394721293,
      "learning_rate": 1.169957641924756e-06,
      "loss": 0.0791,
      "step": 3580
    },
    {
      "epoch": 0.8147895335608646,
      "grad_norm": 1.0382772002032885,
      "learning_rate": 1.1699138875710087e-06,
      "loss": 0.0378,
      "step": 3581
    },
    {
      "epoch": 0.8150170648464163,
      "grad_norm": 1.1375187926566632,
      "learning_rate": 1.1698701220802194e-06,
      "loss": 0.046,
      "step": 3582
    },
    {
      "epoch": 0.8152445961319681,
      "grad_norm": 1.1605133417672624,
      "learning_rate": 1.1698263454532827e-06,
      "loss": 0.0563,
      "step": 3583
    },
    {
      "epoch": 0.81547212741752,
      "grad_norm": 1.382705077452947,
      "learning_rate": 1.1697825576910933e-06,
      "loss": 0.0388,
      "step": 3584
    },
    {
      "epoch": 0.8156996587030717,
      "grad_norm": 24.85184337955777,
      "learning_rate": 1.1697387587945461e-06,
      "loss": 0.4189,
      "step": 3585
    },
    {
      "epoch": 0.8159271899886235,
      "grad_norm": 1.375750049796865,
      "learning_rate": 1.1696949487645365e-06,
      "loss": 0.0746,
      "step": 3586
    },
    {
      "epoch": 0.8161547212741752,
      "grad_norm": 1.9135630568236137,
      "learning_rate": 1.1696511276019595e-06,
      "loss": 0.0963,
      "step": 3587
    },
    {
      "epoch": 0.816382252559727,
      "grad_norm": 1.4697481500847787,
      "learning_rate": 1.169607295307711e-06,
      "loss": 0.0468,
      "step": 3588
    },
    {
      "epoch": 0.8166097838452787,
      "grad_norm": 1.8024021280856608,
      "learning_rate": 1.169563451882687e-06,
      "loss": 0.0734,
      "step": 3589
    },
    {
      "epoch": 0.8168373151308305,
      "grad_norm": 1.0001763795597935,
      "learning_rate": 1.1695195973277831e-06,
      "loss": 0.0377,
      "step": 3590
    },
    {
      "epoch": 0.8170648464163822,
      "grad_norm": 1.6468051827674142,
      "learning_rate": 1.169475731643896e-06,
      "loss": 0.0757,
      "step": 3591
    },
    {
      "epoch": 0.817292377701934,
      "grad_norm": 1.1181893907591056,
      "learning_rate": 1.169431854831922e-06,
      "loss": 0.0423,
      "step": 3592
    },
    {
      "epoch": 0.8175199089874858,
      "grad_norm": 1.153273064949388,
      "learning_rate": 1.169387966892758e-06,
      "loss": 0.0697,
      "step": 3593
    },
    {
      "epoch": 0.8177474402730376,
      "grad_norm": 1.694259041690198,
      "learning_rate": 1.169344067827301e-06,
      "loss": 0.0942,
      "step": 3594
    },
    {
      "epoch": 0.8179749715585893,
      "grad_norm": 1.1843299797048426,
      "learning_rate": 1.1693001576364483e-06,
      "loss": 0.0649,
      "step": 3595
    },
    {
      "epoch": 0.8182025028441411,
      "grad_norm": 1.5290468150322878,
      "learning_rate": 1.1692562363210969e-06,
      "loss": 0.0586,
      "step": 3596
    },
    {
      "epoch": 0.8184300341296928,
      "grad_norm": 1.020107068762973,
      "learning_rate": 1.169212303882145e-06,
      "loss": 0.045,
      "step": 3597
    },
    {
      "epoch": 0.8186575654152446,
      "grad_norm": 0.8683619857454433,
      "learning_rate": 1.1691683603204901e-06,
      "loss": 0.0628,
      "step": 3598
    },
    {
      "epoch": 0.8188850967007963,
      "grad_norm": 1.2042022871006006,
      "learning_rate": 1.1691244056370307e-06,
      "loss": 0.0461,
      "step": 3599
    },
    {
      "epoch": 0.8191126279863481,
      "grad_norm": 0.9629993703014715,
      "learning_rate": 1.169080439832665e-06,
      "loss": 0.0429,
      "step": 3600
    },
    {
      "epoch": 0.8193401592718998,
      "grad_norm": 1.3152436305816528,
      "learning_rate": 1.1690364629082915e-06,
      "loss": 0.0414,
      "step": 3601
    },
    {
      "epoch": 0.8195676905574516,
      "grad_norm": 1.2407550380695793,
      "learning_rate": 1.168992474864809e-06,
      "loss": 0.0426,
      "step": 3602
    },
    {
      "epoch": 0.8197952218430035,
      "grad_norm": 1.7099317456821084,
      "learning_rate": 1.1689484757031167e-06,
      "loss": 0.0661,
      "step": 3603
    },
    {
      "epoch": 0.8200227531285552,
      "grad_norm": 2.4921680206760173,
      "learning_rate": 1.1689044654241136e-06,
      "loss": 0.0975,
      "step": 3604
    },
    {
      "epoch": 0.820250284414107,
      "grad_norm": 0.9191594230152784,
      "learning_rate": 1.1688604440286994e-06,
      "loss": 0.0427,
      "step": 3605
    },
    {
      "epoch": 0.8204778156996587,
      "grad_norm": 1.5850443718709968,
      "learning_rate": 1.1688164115177737e-06,
      "loss": 0.064,
      "step": 3606
    },
    {
      "epoch": 0.8207053469852105,
      "grad_norm": 1.4447088659158112,
      "learning_rate": 1.1687723678922367e-06,
      "loss": 0.0644,
      "step": 3607
    },
    {
      "epoch": 0.8209328782707622,
      "grad_norm": 1.1876210795397029,
      "learning_rate": 1.1687283131529882e-06,
      "loss": 0.05,
      "step": 3608
    },
    {
      "epoch": 0.821160409556314,
      "grad_norm": 1.1124560475691172,
      "learning_rate": 1.1686842473009286e-06,
      "loss": 0.052,
      "step": 3609
    },
    {
      "epoch": 0.8213879408418657,
      "grad_norm": 1.1061762100158978,
      "learning_rate": 1.168640170336959e-06,
      "loss": 0.0386,
      "step": 3610
    },
    {
      "epoch": 0.8216154721274175,
      "grad_norm": 1.44352318845533,
      "learning_rate": 1.16859608226198e-06,
      "loss": 0.063,
      "step": 3611
    },
    {
      "epoch": 0.8218430034129692,
      "grad_norm": 1.6539718566592532,
      "learning_rate": 1.1685519830768923e-06,
      "loss": 0.0874,
      "step": 3612
    },
    {
      "epoch": 0.8220705346985211,
      "grad_norm": 1.6565633171924414,
      "learning_rate": 1.1685078727825978e-06,
      "loss": 0.0589,
      "step": 3613
    },
    {
      "epoch": 0.8222980659840728,
      "grad_norm": 1.461206103355595,
      "learning_rate": 1.1684637513799976e-06,
      "loss": 0.0627,
      "step": 3614
    },
    {
      "epoch": 0.8225255972696246,
      "grad_norm": 0.9887213175076792,
      "learning_rate": 1.1684196188699936e-06,
      "loss": 0.0459,
      "step": 3615
    },
    {
      "epoch": 0.8227531285551763,
      "grad_norm": 1.3363304037819075,
      "learning_rate": 1.168375475253488e-06,
      "loss": 0.0467,
      "step": 3616
    },
    {
      "epoch": 0.8229806598407281,
      "grad_norm": 1.178055634767022,
      "learning_rate": 1.168331320531383e-06,
      "loss": 0.0471,
      "step": 3617
    },
    {
      "epoch": 0.8232081911262799,
      "grad_norm": 1.586381066611804,
      "learning_rate": 1.1682871547045805e-06,
      "loss": 0.0708,
      "step": 3618
    },
    {
      "epoch": 0.8234357224118316,
      "grad_norm": 1.209859608299182,
      "learning_rate": 1.1682429777739836e-06,
      "loss": 0.0513,
      "step": 3619
    },
    {
      "epoch": 0.8236632536973834,
      "grad_norm": 1.7764116713616094,
      "learning_rate": 1.168198789740495e-06,
      "loss": 0.0638,
      "step": 3620
    },
    {
      "epoch": 0.8238907849829351,
      "grad_norm": 0.7763254752373752,
      "learning_rate": 1.1681545906050184e-06,
      "loss": 0.0343,
      "step": 3621
    },
    {
      "epoch": 0.824118316268487,
      "grad_norm": 1.1001948497692553,
      "learning_rate": 1.1681103803684564e-06,
      "loss": 0.056,
      "step": 3622
    },
    {
      "epoch": 0.8243458475540387,
      "grad_norm": 1.273383386134855,
      "learning_rate": 1.168066159031713e-06,
      "loss": 0.0468,
      "step": 3623
    },
    {
      "epoch": 0.8245733788395905,
      "grad_norm": 1.5388101293640168,
      "learning_rate": 1.1680219265956918e-06,
      "loss": 0.0558,
      "step": 3624
    },
    {
      "epoch": 0.8248009101251422,
      "grad_norm": 1.1316172883077475,
      "learning_rate": 1.167977683061297e-06,
      "loss": 0.0434,
      "step": 3625
    },
    {
      "epoch": 0.825028441410694,
      "grad_norm": 1.3402414111306105,
      "learning_rate": 1.1679334284294328e-06,
      "loss": 0.0601,
      "step": 3626
    },
    {
      "epoch": 0.8252559726962457,
      "grad_norm": 0.8814362527241864,
      "learning_rate": 1.1678891627010036e-06,
      "loss": 0.0498,
      "step": 3627
    },
    {
      "epoch": 0.8254835039817975,
      "grad_norm": 1.1662546055909706,
      "learning_rate": 1.1678448858769139e-06,
      "loss": 0.0493,
      "step": 3628
    },
    {
      "epoch": 0.8257110352673492,
      "grad_norm": 1.8562897841170296,
      "learning_rate": 1.1678005979580694e-06,
      "loss": 0.0834,
      "step": 3629
    },
    {
      "epoch": 0.825938566552901,
      "grad_norm": 0.8447551801193321,
      "learning_rate": 1.1677562989453745e-06,
      "loss": 0.0283,
      "step": 3630
    },
    {
      "epoch": 0.8261660978384527,
      "grad_norm": 1.0072638573362171,
      "learning_rate": 1.1677119888397348e-06,
      "loss": 0.049,
      "step": 3631
    },
    {
      "epoch": 0.8263936291240046,
      "grad_norm": 1.433958970376907,
      "learning_rate": 1.1676676676420562e-06,
      "loss": 0.0601,
      "step": 3632
    },
    {
      "epoch": 0.8266211604095564,
      "grad_norm": 1.123241025021127,
      "learning_rate": 1.1676233353532442e-06,
      "loss": 0.046,
      "step": 3633
    },
    {
      "epoch": 0.8268486916951081,
      "grad_norm": 0.7445525822110918,
      "learning_rate": 1.167578991974205e-06,
      "loss": 0.0181,
      "step": 3634
    },
    {
      "epoch": 0.8270762229806599,
      "grad_norm": 0.7798344340235215,
      "learning_rate": 1.167534637505845e-06,
      "loss": 0.0272,
      "step": 3635
    },
    {
      "epoch": 0.8273037542662116,
      "grad_norm": 1.0329542174985942,
      "learning_rate": 1.1674902719490704e-06,
      "loss": 0.0283,
      "step": 3636
    },
    {
      "epoch": 0.8275312855517634,
      "grad_norm": 1.3105774584535312,
      "learning_rate": 1.1674458953047883e-06,
      "loss": 0.0768,
      "step": 3637
    },
    {
      "epoch": 0.8277588168373151,
      "grad_norm": 1.403061308791836,
      "learning_rate": 1.1674015075739054e-06,
      "loss": 0.0524,
      "step": 3638
    },
    {
      "epoch": 0.8279863481228669,
      "grad_norm": 1.525826516533729,
      "learning_rate": 1.1673571087573293e-06,
      "loss": 0.035,
      "step": 3639
    },
    {
      "epoch": 0.8282138794084186,
      "grad_norm": 1.578985773093579,
      "learning_rate": 1.167312698855967e-06,
      "loss": 0.0686,
      "step": 3640
    },
    {
      "epoch": 0.8284414106939704,
      "grad_norm": 0.8598678068916811,
      "learning_rate": 1.1672682778707262e-06,
      "loss": 0.0434,
      "step": 3641
    },
    {
      "epoch": 0.8286689419795222,
      "grad_norm": 1.1746515581399284,
      "learning_rate": 1.167223845802515e-06,
      "loss": 0.0683,
      "step": 3642
    },
    {
      "epoch": 0.828896473265074,
      "grad_norm": 1.101963672107574,
      "learning_rate": 1.1671794026522417e-06,
      "loss": 0.0446,
      "step": 3643
    },
    {
      "epoch": 0.8291240045506257,
      "grad_norm": 1.29174643335613,
      "learning_rate": 1.1671349484208142e-06,
      "loss": 0.0631,
      "step": 3644
    },
    {
      "epoch": 0.8293515358361775,
      "grad_norm": 14.688190742848866,
      "learning_rate": 1.1670904831091412e-06,
      "loss": 0.3068,
      "step": 3645
    },
    {
      "epoch": 0.8295790671217292,
      "grad_norm": 1.997677369000564,
      "learning_rate": 1.1670460067181313e-06,
      "loss": 0.097,
      "step": 3646
    },
    {
      "epoch": 0.829806598407281,
      "grad_norm": 2.0170155054153844,
      "learning_rate": 1.167001519248694e-06,
      "loss": 0.091,
      "step": 3647
    },
    {
      "epoch": 0.8300341296928327,
      "grad_norm": 1.445412257420854,
      "learning_rate": 1.1669570207017384e-06,
      "loss": 0.0451,
      "step": 3648
    },
    {
      "epoch": 0.8302616609783845,
      "grad_norm": 1.5233401207808903,
      "learning_rate": 1.1669125110781737e-06,
      "loss": 0.0606,
      "step": 3649
    },
    {
      "epoch": 0.8304891922639362,
      "grad_norm": 0.7835188426142936,
      "learning_rate": 1.1668679903789095e-06,
      "loss": 0.0332,
      "step": 3650
    },
    {
      "epoch": 0.830716723549488,
      "grad_norm": 1.4515524685426144,
      "learning_rate": 1.1668234586048562e-06,
      "loss": 0.0596,
      "step": 3651
    },
    {
      "epoch": 0.8309442548350399,
      "grad_norm": 0.9655835502300143,
      "learning_rate": 1.1667789157569237e-06,
      "loss": 0.0409,
      "step": 3652
    },
    {
      "epoch": 0.8311717861205916,
      "grad_norm": 1.393653817047434,
      "learning_rate": 1.1667343618360224e-06,
      "loss": 0.0469,
      "step": 3653
    },
    {
      "epoch": 0.8313993174061434,
      "grad_norm": 1.483284458995605,
      "learning_rate": 1.166689796843063e-06,
      "loss": 0.071,
      "step": 3654
    },
    {
      "epoch": 0.8316268486916951,
      "grad_norm": 2.0171503740210004,
      "learning_rate": 1.166645220778956e-06,
      "loss": 0.113,
      "step": 3655
    },
    {
      "epoch": 0.8318543799772469,
      "grad_norm": 1.1504742515281876,
      "learning_rate": 1.1666006336446127e-06,
      "loss": 0.0524,
      "step": 3656
    },
    {
      "epoch": 0.8320819112627986,
      "grad_norm": 45.01747083144473,
      "learning_rate": 1.1665560354409444e-06,
      "loss": 0.4523,
      "step": 3657
    },
    {
      "epoch": 0.8323094425483504,
      "grad_norm": 1.112433200629069,
      "learning_rate": 1.1665114261688625e-06,
      "loss": 0.0668,
      "step": 3658
    },
    {
      "epoch": 0.8325369738339021,
      "grad_norm": 0.8757946163066259,
      "learning_rate": 1.1664668058292789e-06,
      "loss": 0.031,
      "step": 3659
    },
    {
      "epoch": 0.8327645051194539,
      "grad_norm": 1.4231045164348157,
      "learning_rate": 1.1664221744231052e-06,
      "loss": 0.0557,
      "step": 3660
    },
    {
      "epoch": 0.8329920364050057,
      "grad_norm": 0.817775869454815,
      "learning_rate": 1.1663775319512539e-06,
      "loss": 0.033,
      "step": 3661
    },
    {
      "epoch": 0.8332195676905575,
      "grad_norm": 1.1781931600979176,
      "learning_rate": 1.1663328784146375e-06,
      "loss": 0.0642,
      "step": 3662
    },
    {
      "epoch": 0.8334470989761092,
      "grad_norm": 0.9259664097181735,
      "learning_rate": 1.1662882138141684e-06,
      "loss": 0.04,
      "step": 3663
    },
    {
      "epoch": 0.833674630261661,
      "grad_norm": 0.9223727967515527,
      "learning_rate": 1.1662435381507595e-06,
      "loss": 0.0348,
      "step": 3664
    },
    {
      "epoch": 0.8339021615472128,
      "grad_norm": 2.505531351215165,
      "learning_rate": 1.1661988514253237e-06,
      "loss": 0.0541,
      "step": 3665
    },
    {
      "epoch": 0.8341296928327645,
      "grad_norm": 0.8185437890691741,
      "learning_rate": 1.1661541536387748e-06,
      "loss": 0.0304,
      "step": 3666
    },
    {
      "epoch": 0.8343572241183163,
      "grad_norm": 1.4951738593038986,
      "learning_rate": 1.166109444792026e-06,
      "loss": 0.0621,
      "step": 3667
    },
    {
      "epoch": 0.834584755403868,
      "grad_norm": 1.3025368287248427,
      "learning_rate": 1.1660647248859913e-06,
      "loss": 0.0516,
      "step": 3668
    },
    {
      "epoch": 0.8348122866894198,
      "grad_norm": 1.2218241621957284,
      "learning_rate": 1.1660199939215845e-06,
      "loss": 0.0425,
      "step": 3669
    },
    {
      "epoch": 0.8350398179749715,
      "grad_norm": 1.3776947069628735,
      "learning_rate": 1.1659752518997197e-06,
      "loss": 0.0398,
      "step": 3670
    },
    {
      "epoch": 0.8352673492605234,
      "grad_norm": 1.1397273421298342,
      "learning_rate": 1.1659304988213115e-06,
      "loss": 0.065,
      "step": 3671
    },
    {
      "epoch": 0.8354948805460751,
      "grad_norm": 1.0434688435298627,
      "learning_rate": 1.1658857346872745e-06,
      "loss": 0.0566,
      "step": 3672
    },
    {
      "epoch": 0.8357224118316269,
      "grad_norm": 1.5440568255758365,
      "learning_rate": 1.165840959498524e-06,
      "loss": 0.0688,
      "step": 3673
    },
    {
      "epoch": 0.8359499431171786,
      "grad_norm": 1.115123935337787,
      "learning_rate": 1.1657961732559745e-06,
      "loss": 0.032,
      "step": 3674
    },
    {
      "epoch": 0.8361774744027304,
      "grad_norm": 2.009941649682246,
      "learning_rate": 1.1657513759605417e-06,
      "loss": 0.1067,
      "step": 3675
    },
    {
      "epoch": 0.8364050056882821,
      "grad_norm": 1.0878721650862986,
      "learning_rate": 1.1657065676131412e-06,
      "loss": 0.039,
      "step": 3676
    },
    {
      "epoch": 0.8366325369738339,
      "grad_norm": 1.2823447557648144,
      "learning_rate": 1.1656617482146886e-06,
      "loss": 0.0565,
      "step": 3677
    },
    {
      "epoch": 0.8368600682593856,
      "grad_norm": 0.8640896631578039,
      "learning_rate": 1.1656169177661e-06,
      "loss": 0.0286,
      "step": 3678
    },
    {
      "epoch": 0.8370875995449374,
      "grad_norm": 1.5431241138574887,
      "learning_rate": 1.165572076268292e-06,
      "loss": 0.0777,
      "step": 3679
    },
    {
      "epoch": 0.8373151308304891,
      "grad_norm": 2.182613338071724,
      "learning_rate": 1.1655272237221804e-06,
      "loss": 0.0811,
      "step": 3680
    },
    {
      "epoch": 0.837542662116041,
      "grad_norm": 1.6601496114451646,
      "learning_rate": 1.1654823601286826e-06,
      "loss": 0.0705,
      "step": 3681
    },
    {
      "epoch": 0.8377701934015928,
      "grad_norm": 0.9845739508563496,
      "learning_rate": 1.165437485488715e-06,
      "loss": 0.0352,
      "step": 3682
    },
    {
      "epoch": 0.8379977246871445,
      "grad_norm": 1.7268236108563282,
      "learning_rate": 1.165392599803195e-06,
      "loss": 0.0813,
      "step": 3683
    },
    {
      "epoch": 0.8382252559726963,
      "grad_norm": 1.7016772440974146,
      "learning_rate": 1.16534770307304e-06,
      "loss": 0.0686,
      "step": 3684
    },
    {
      "epoch": 0.838452787258248,
      "grad_norm": 1.1115991957924602,
      "learning_rate": 1.1653027952991676e-06,
      "loss": 0.0453,
      "step": 3685
    },
    {
      "epoch": 0.8386803185437998,
      "grad_norm": 1.1625036802126243,
      "learning_rate": 1.1652578764824953e-06,
      "loss": 0.0547,
      "step": 3686
    },
    {
      "epoch": 0.8389078498293515,
      "grad_norm": 1.2752562383666506,
      "learning_rate": 1.1652129466239417e-06,
      "loss": 0.0557,
      "step": 3687
    },
    {
      "epoch": 0.8391353811149033,
      "grad_norm": 1.1708119141481568,
      "learning_rate": 1.1651680057244247e-06,
      "loss": 0.0543,
      "step": 3688
    },
    {
      "epoch": 0.839362912400455,
      "grad_norm": 1.5686634859706143,
      "learning_rate": 1.165123053784863e-06,
      "loss": 0.0631,
      "step": 3689
    },
    {
      "epoch": 0.8395904436860068,
      "grad_norm": 0.7862204435509706,
      "learning_rate": 1.1650780908061753e-06,
      "loss": 0.0268,
      "step": 3690
    },
    {
      "epoch": 0.8398179749715586,
      "grad_norm": 0.8894683444434565,
      "learning_rate": 1.1650331167892805e-06,
      "loss": 0.0447,
      "step": 3691
    },
    {
      "epoch": 0.8400455062571104,
      "grad_norm": 1.1634157087855133,
      "learning_rate": 1.1649881317350979e-06,
      "loss": 0.0563,
      "step": 3692
    },
    {
      "epoch": 0.8402730375426621,
      "grad_norm": 0.8154942879262764,
      "learning_rate": 1.1649431356445467e-06,
      "loss": 0.0286,
      "step": 3693
    },
    {
      "epoch": 0.8405005688282139,
      "grad_norm": 1.377478587198256,
      "learning_rate": 1.1648981285185464e-06,
      "loss": 0.0616,
      "step": 3694
    },
    {
      "epoch": 0.8407281001137656,
      "grad_norm": 1.4564713307212853,
      "learning_rate": 1.1648531103580178e-06,
      "loss": 0.0537,
      "step": 3695
    },
    {
      "epoch": 0.8409556313993174,
      "grad_norm": 1.1442584647082312,
      "learning_rate": 1.1648080811638797e-06,
      "loss": 0.0514,
      "step": 3696
    },
    {
      "epoch": 0.8411831626848691,
      "grad_norm": 0.6059546954643502,
      "learning_rate": 1.1647630409370533e-06,
      "loss": 0.024,
      "step": 3697
    },
    {
      "epoch": 0.8414106939704209,
      "grad_norm": 0.845020155561466,
      "learning_rate": 1.1647179896784588e-06,
      "loss": 0.0297,
      "step": 3698
    },
    {
      "epoch": 0.8416382252559726,
      "grad_norm": 1.1265316983407672,
      "learning_rate": 1.1646729273890172e-06,
      "loss": 0.0385,
      "step": 3699
    },
    {
      "epoch": 0.8418657565415245,
      "grad_norm": 1.5081431023506024,
      "learning_rate": 1.1646278540696493e-06,
      "loss": 0.0791,
      "step": 3700
    },
    {
      "epoch": 0.8420932878270763,
      "grad_norm": 1.0623449631960677,
      "learning_rate": 1.1645827697212762e-06,
      "loss": 0.0366,
      "step": 3701
    },
    {
      "epoch": 0.842320819112628,
      "grad_norm": 1.0697058302573508,
      "learning_rate": 1.1645376743448194e-06,
      "loss": 0.0365,
      "step": 3702
    },
    {
      "epoch": 0.8425483503981798,
      "grad_norm": 1.6166494157045328,
      "learning_rate": 1.1644925679412008e-06,
      "loss": 0.0917,
      "step": 3703
    },
    {
      "epoch": 0.8427758816837315,
      "grad_norm": 0.8909409395783954,
      "learning_rate": 1.1644474505113423e-06,
      "loss": 0.0483,
      "step": 3704
    },
    {
      "epoch": 0.8430034129692833,
      "grad_norm": 1.2947960727076373,
      "learning_rate": 1.1644023220561657e-06,
      "loss": 0.0841,
      "step": 3705
    },
    {
      "epoch": 0.843230944254835,
      "grad_norm": 2.338681558244469,
      "learning_rate": 1.1643571825765935e-06,
      "loss": 0.0884,
      "step": 3706
    },
    {
      "epoch": 0.8434584755403868,
      "grad_norm": 1.0808444836814672,
      "learning_rate": 1.1643120320735481e-06,
      "loss": 0.0468,
      "step": 3707
    },
    {
      "epoch": 0.8436860068259385,
      "grad_norm": 3.1483674383789007,
      "learning_rate": 1.164266870547953e-06,
      "loss": 0.1461,
      "step": 3708
    },
    {
      "epoch": 0.8439135381114903,
      "grad_norm": 1.1429682418848544,
      "learning_rate": 1.16422169800073e-06,
      "loss": 0.0575,
      "step": 3709
    },
    {
      "epoch": 0.8441410693970421,
      "grad_norm": 1.4092804732703959,
      "learning_rate": 1.1641765144328035e-06,
      "loss": 0.0824,
      "step": 3710
    },
    {
      "epoch": 0.8443686006825939,
      "grad_norm": 0.9907223592123003,
      "learning_rate": 1.1641313198450966e-06,
      "loss": 0.0408,
      "step": 3711
    },
    {
      "epoch": 0.8445961319681456,
      "grad_norm": 1.1953791862365195,
      "learning_rate": 1.1640861142385326e-06,
      "loss": 0.0417,
      "step": 3712
    },
    {
      "epoch": 0.8448236632536974,
      "grad_norm": 1.0300444272783411,
      "learning_rate": 1.1640408976140358e-06,
      "loss": 0.0292,
      "step": 3713
    },
    {
      "epoch": 0.8450511945392492,
      "grad_norm": 0.9396034384804057,
      "learning_rate": 1.1639956699725303e-06,
      "loss": 0.0251,
      "step": 3714
    },
    {
      "epoch": 0.8452787258248009,
      "grad_norm": 1.0450241907142919,
      "learning_rate": 1.1639504313149403e-06,
      "loss": 0.0427,
      "step": 3715
    },
    {
      "epoch": 0.8455062571103527,
      "grad_norm": 1.2352022092638624,
      "learning_rate": 1.1639051816421906e-06,
      "loss": 0.0622,
      "step": 3716
    },
    {
      "epoch": 0.8457337883959044,
      "grad_norm": 1.7465197987251446,
      "learning_rate": 1.163859920955206e-06,
      "loss": 0.0646,
      "step": 3717
    },
    {
      "epoch": 0.8459613196814562,
      "grad_norm": 1.1198862623805916,
      "learning_rate": 1.1638146492549112e-06,
      "loss": 0.057,
      "step": 3718
    },
    {
      "epoch": 0.8461888509670079,
      "grad_norm": 0.9891506478501289,
      "learning_rate": 1.163769366542232e-06,
      "loss": 0.043,
      "step": 3719
    },
    {
      "epoch": 0.8464163822525598,
      "grad_norm": 0.8307264131023004,
      "learning_rate": 1.1637240728180937e-06,
      "loss": 0.0374,
      "step": 3720
    },
    {
      "epoch": 0.8466439135381115,
      "grad_norm": 1.2015516317579826,
      "learning_rate": 1.163678768083422e-06,
      "loss": 0.0542,
      "step": 3721
    },
    {
      "epoch": 0.8468714448236633,
      "grad_norm": 0.8651705828559965,
      "learning_rate": 1.1636334523391426e-06,
      "loss": 0.0333,
      "step": 3722
    },
    {
      "epoch": 0.847098976109215,
      "grad_norm": 1.8721051492604341,
      "learning_rate": 1.1635881255861821e-06,
      "loss": 0.1021,
      "step": 3723
    },
    {
      "epoch": 0.8473265073947668,
      "grad_norm": 1.7947647079315026,
      "learning_rate": 1.1635427878254667e-06,
      "loss": 0.0741,
      "step": 3724
    },
    {
      "epoch": 0.8475540386803185,
      "grad_norm": 2.1673257995211506,
      "learning_rate": 1.163497439057923e-06,
      "loss": 0.084,
      "step": 3725
    },
    {
      "epoch": 0.8477815699658703,
      "grad_norm": 1.1269894105236045,
      "learning_rate": 1.1634520792844778e-06,
      "loss": 0.0675,
      "step": 3726
    },
    {
      "epoch": 0.848009101251422,
      "grad_norm": 1.205510809010935,
      "learning_rate": 1.1634067085060583e-06,
      "loss": 0.0495,
      "step": 3727
    },
    {
      "epoch": 0.8482366325369738,
      "grad_norm": 0.691849376485242,
      "learning_rate": 1.1633613267235915e-06,
      "loss": 0.0466,
      "step": 3728
    },
    {
      "epoch": 0.8484641638225257,
      "grad_norm": 1.7367118780028674,
      "learning_rate": 1.1633159339380054e-06,
      "loss": 0.0796,
      "step": 3729
    },
    {
      "epoch": 0.8486916951080774,
      "grad_norm": 1.0717660762894743,
      "learning_rate": 1.1632705301502272e-06,
      "loss": 0.04,
      "step": 3730
    },
    {
      "epoch": 0.8489192263936292,
      "grad_norm": 1.5860944536440218,
      "learning_rate": 1.1632251153611855e-06,
      "loss": 0.0494,
      "step": 3731
    },
    {
      "epoch": 0.8491467576791809,
      "grad_norm": 1.8503643346395675,
      "learning_rate": 1.163179689571808e-06,
      "loss": 0.0747,
      "step": 3732
    },
    {
      "epoch": 0.8493742889647327,
      "grad_norm": 1.51685414443071,
      "learning_rate": 1.1631342527830234e-06,
      "loss": 0.0605,
      "step": 3733
    },
    {
      "epoch": 0.8496018202502844,
      "grad_norm": 5.054888475581482,
      "learning_rate": 1.16308880499576e-06,
      "loss": 0.1453,
      "step": 3734
    },
    {
      "epoch": 0.8498293515358362,
      "grad_norm": 0.9938442986727953,
      "learning_rate": 1.163043346210947e-06,
      "loss": 0.0402,
      "step": 3735
    },
    {
      "epoch": 0.8500568828213879,
      "grad_norm": 1.5560463292525197,
      "learning_rate": 1.1629978764295133e-06,
      "loss": 0.0735,
      "step": 3736
    },
    {
      "epoch": 0.8502844141069397,
      "grad_norm": 1.6867577558544848,
      "learning_rate": 1.1629523956523883e-06,
      "loss": 0.0488,
      "step": 3737
    },
    {
      "epoch": 0.8505119453924914,
      "grad_norm": 1.0404339887342127,
      "learning_rate": 1.1629069038805018e-06,
      "loss": 0.0328,
      "step": 3738
    },
    {
      "epoch": 0.8507394766780433,
      "grad_norm": 1.2922834659902496,
      "learning_rate": 1.162861401114783e-06,
      "loss": 0.0643,
      "step": 3739
    },
    {
      "epoch": 0.850967007963595,
      "grad_norm": 1.0645197047514645,
      "learning_rate": 1.1628158873561624e-06,
      "loss": 0.0527,
      "step": 3740
    },
    {
      "epoch": 0.8511945392491468,
      "grad_norm": 1.2986459720013372,
      "learning_rate": 1.16277036260557e-06,
      "loss": 0.0562,
      "step": 3741
    },
    {
      "epoch": 0.8514220705346985,
      "grad_norm": 0.9856861581794601,
      "learning_rate": 1.1627248268639363e-06,
      "loss": 0.0395,
      "step": 3742
    },
    {
      "epoch": 0.8516496018202503,
      "grad_norm": 1.0675074473205326,
      "learning_rate": 1.1626792801321917e-06,
      "loss": 0.0675,
      "step": 3743
    },
    {
      "epoch": 0.851877133105802,
      "grad_norm": 2.38754548600943,
      "learning_rate": 1.1626337224112676e-06,
      "loss": 0.0971,
      "step": 3744
    },
    {
      "epoch": 0.8521046643913538,
      "grad_norm": 1.7266469693034243,
      "learning_rate": 1.1625881537020948e-06,
      "loss": 0.0671,
      "step": 3745
    },
    {
      "epoch": 0.8523321956769055,
      "grad_norm": 1.280551624533034,
      "learning_rate": 1.1625425740056046e-06,
      "loss": 0.0377,
      "step": 3746
    },
    {
      "epoch": 0.8525597269624573,
      "grad_norm": 1.237081721406354,
      "learning_rate": 1.1624969833227287e-06,
      "loss": 0.059,
      "step": 3747
    },
    {
      "epoch": 0.852787258248009,
      "grad_norm": 0.9836714855093116,
      "learning_rate": 1.1624513816543988e-06,
      "loss": 0.0362,
      "step": 3748
    },
    {
      "epoch": 0.8530147895335609,
      "grad_norm": 1.9867742968928817,
      "learning_rate": 1.162405769001547e-06,
      "loss": 0.0645,
      "step": 3749
    },
    {
      "epoch": 0.8532423208191127,
      "grad_norm": 0.8182858794812171,
      "learning_rate": 1.1623601453651053e-06,
      "loss": 0.0355,
      "step": 3750
    },
    {
      "epoch": 0.8534698521046644,
      "grad_norm": 1.4917273253143806,
      "learning_rate": 1.1623145107460065e-06,
      "loss": 0.0503,
      "step": 3751
    },
    {
      "epoch": 0.8536973833902162,
      "grad_norm": 0.7287193210043196,
      "learning_rate": 1.1622688651451833e-06,
      "loss": 0.0307,
      "step": 3752
    },
    {
      "epoch": 0.8539249146757679,
      "grad_norm": 1.4100424416120585,
      "learning_rate": 1.1622232085635683e-06,
      "loss": 0.0826,
      "step": 3753
    },
    {
      "epoch": 0.8541524459613197,
      "grad_norm": 1.2920863855722,
      "learning_rate": 1.162177541002095e-06,
      "loss": 0.057,
      "step": 3754
    },
    {
      "epoch": 0.8543799772468714,
      "grad_norm": 0.9760542142070106,
      "learning_rate": 1.162131862461696e-06,
      "loss": 0.0424,
      "step": 3755
    },
    {
      "epoch": 0.8546075085324232,
      "grad_norm": 1.1155922665776505,
      "learning_rate": 1.1620861729433062e-06,
      "loss": 0.0548,
      "step": 3756
    },
    {
      "epoch": 0.8548350398179749,
      "grad_norm": 2.0265637739770646,
      "learning_rate": 1.1620404724478582e-06,
      "loss": 0.1044,
      "step": 3757
    },
    {
      "epoch": 0.8550625711035267,
      "grad_norm": 1.2011014044384718,
      "learning_rate": 1.1619947609762867e-06,
      "loss": 0.0564,
      "step": 3758
    },
    {
      "epoch": 0.8552901023890785,
      "grad_norm": 6.130934132803181,
      "learning_rate": 1.1619490385295255e-06,
      "loss": 0.2002,
      "step": 3759
    },
    {
      "epoch": 0.8555176336746303,
      "grad_norm": 1.113651539392019,
      "learning_rate": 1.1619033051085094e-06,
      "loss": 0.0652,
      "step": 3760
    },
    {
      "epoch": 0.855745164960182,
      "grad_norm": 1.286325265245382,
      "learning_rate": 1.161857560714173e-06,
      "loss": 0.0588,
      "step": 3761
    },
    {
      "epoch": 0.8559726962457338,
      "grad_norm": 1.7600474072446048,
      "learning_rate": 1.1618118053474514e-06,
      "loss": 0.0893,
      "step": 3762
    },
    {
      "epoch": 0.8562002275312856,
      "grad_norm": 1.6589882412706265,
      "learning_rate": 1.1617660390092794e-06,
      "loss": 0.0745,
      "step": 3763
    },
    {
      "epoch": 0.8564277588168373,
      "grad_norm": 1.2790558914988392,
      "learning_rate": 1.161720261700593e-06,
      "loss": 0.057,
      "step": 3764
    },
    {
      "epoch": 0.856655290102389,
      "grad_norm": 1.0102177907448584,
      "learning_rate": 1.161674473422327e-06,
      "loss": 0.0413,
      "step": 3765
    },
    {
      "epoch": 0.8568828213879408,
      "grad_norm": 1.5326235862910704,
      "learning_rate": 1.1616286741754178e-06,
      "loss": 0.0605,
      "step": 3766
    },
    {
      "epoch": 0.8571103526734926,
      "grad_norm": 0.8465015567779428,
      "learning_rate": 1.1615828639608013e-06,
      "loss": 0.0312,
      "step": 3767
    },
    {
      "epoch": 0.8573378839590444,
      "grad_norm": 1.5793350610162067,
      "learning_rate": 1.1615370427794138e-06,
      "loss": 0.0945,
      "step": 3768
    },
    {
      "epoch": 0.8575654152445962,
      "grad_norm": 1.2162812251052018,
      "learning_rate": 1.1614912106321916e-06,
      "loss": 0.0752,
      "step": 3769
    },
    {
      "epoch": 0.8577929465301479,
      "grad_norm": 1.8496686418444572,
      "learning_rate": 1.1614453675200718e-06,
      "loss": 0.0722,
      "step": 3770
    },
    {
      "epoch": 0.8580204778156997,
      "grad_norm": 1.6913533101626717,
      "learning_rate": 1.1613995134439912e-06,
      "loss": 0.0725,
      "step": 3771
    },
    {
      "epoch": 0.8582480091012514,
      "grad_norm": 3.096645898390307,
      "learning_rate": 1.1613536484048866e-06,
      "loss": 0.1055,
      "step": 3772
    },
    {
      "epoch": 0.8584755403868032,
      "grad_norm": 1.6446149120178801,
      "learning_rate": 1.161307772403696e-06,
      "loss": 0.0597,
      "step": 3773
    },
    {
      "epoch": 0.8587030716723549,
      "grad_norm": 1.5393305823965842,
      "learning_rate": 1.1612618854413566e-06,
      "loss": 0.0732,
      "step": 3774
    },
    {
      "epoch": 0.8589306029579067,
      "grad_norm": 1.9421552759578808,
      "learning_rate": 1.1612159875188065e-06,
      "loss": 0.0693,
      "step": 3775
    },
    {
      "epoch": 0.8591581342434584,
      "grad_norm": 1.3707910624756399,
      "learning_rate": 1.1611700786369835e-06,
      "loss": 0.0585,
      "step": 3776
    },
    {
      "epoch": 0.8593856655290102,
      "grad_norm": 1.5332539464674453,
      "learning_rate": 1.161124158796826e-06,
      "loss": 0.0679,
      "step": 3777
    },
    {
      "epoch": 0.859613196814562,
      "grad_norm": 0.8257875510353138,
      "learning_rate": 1.1610782279992728e-06,
      "loss": 0.0405,
      "step": 3778
    },
    {
      "epoch": 0.8598407281001138,
      "grad_norm": 0.8774289661633246,
      "learning_rate": 1.161032286245262e-06,
      "loss": 0.0303,
      "step": 3779
    },
    {
      "epoch": 0.8600682593856656,
      "grad_norm": 2.167218867354055,
      "learning_rate": 1.1609863335357332e-06,
      "loss": 0.077,
      "step": 3780
    },
    {
      "epoch": 0.8602957906712173,
      "grad_norm": 1.8941215544947247,
      "learning_rate": 1.1609403698716255e-06,
      "loss": 0.0786,
      "step": 3781
    },
    {
      "epoch": 0.8605233219567691,
      "grad_norm": 1.1276115513513825,
      "learning_rate": 1.160894395253878e-06,
      "loss": 0.0537,
      "step": 3782
    },
    {
      "epoch": 0.8607508532423208,
      "grad_norm": 1.7706734723923294,
      "learning_rate": 1.1608484096834306e-06,
      "loss": 0.0922,
      "step": 3783
    },
    {
      "epoch": 0.8609783845278726,
      "grad_norm": 1.2558738456157548,
      "learning_rate": 1.1608024131612231e-06,
      "loss": 0.0652,
      "step": 3784
    },
    {
      "epoch": 0.8612059158134243,
      "grad_norm": 0.9032710072829923,
      "learning_rate": 1.1607564056881953e-06,
      "loss": 0.0379,
      "step": 3785
    },
    {
      "epoch": 0.8614334470989761,
      "grad_norm": 1.190052750152249,
      "learning_rate": 1.160710387265288e-06,
      "loss": 0.0444,
      "step": 3786
    },
    {
      "epoch": 0.8616609783845278,
      "grad_norm": 0.9786079939895075,
      "learning_rate": 1.1606643578934414e-06,
      "loss": 0.0418,
      "step": 3787
    },
    {
      "epoch": 0.8618885096700797,
      "grad_norm": 2.273287744213828,
      "learning_rate": 1.1606183175735963e-06,
      "loss": 0.0738,
      "step": 3788
    },
    {
      "epoch": 0.8621160409556314,
      "grad_norm": 1.0946877128994326,
      "learning_rate": 1.1605722663066938e-06,
      "loss": 0.0473,
      "step": 3789
    },
    {
      "epoch": 0.8623435722411832,
      "grad_norm": 1.3819454647043339,
      "learning_rate": 1.1605262040936752e-06,
      "loss": 0.0735,
      "step": 3790
    },
    {
      "epoch": 0.8625711035267349,
      "grad_norm": 1.263046575159459,
      "learning_rate": 1.1604801309354815e-06,
      "loss": 0.0595,
      "step": 3791
    },
    {
      "epoch": 0.8627986348122867,
      "grad_norm": 1.0133652978063783,
      "learning_rate": 1.1604340468330546e-06,
      "loss": 0.0307,
      "step": 3792
    },
    {
      "epoch": 0.8630261660978384,
      "grad_norm": 1.2379613127015137,
      "learning_rate": 1.1603879517873366e-06,
      "loss": 0.0517,
      "step": 3793
    },
    {
      "epoch": 0.8632536973833902,
      "grad_norm": 1.1750697280707245,
      "learning_rate": 1.160341845799269e-06,
      "loss": 0.0482,
      "step": 3794
    },
    {
      "epoch": 0.863481228668942,
      "grad_norm": 1.6791780297526506,
      "learning_rate": 1.160295728869795e-06,
      "loss": 0.1042,
      "step": 3795
    },
    {
      "epoch": 0.8637087599544937,
      "grad_norm": 1.1451895610966802,
      "learning_rate": 1.1602496009998562e-06,
      "loss": 0.0483,
      "step": 3796
    },
    {
      "epoch": 0.8639362912400455,
      "grad_norm": 0.737910160878505,
      "learning_rate": 1.160203462190396e-06,
      "loss": 0.0264,
      "step": 3797
    },
    {
      "epoch": 0.8641638225255973,
      "grad_norm": 1.0917499952318759,
      "learning_rate": 1.1601573124423573e-06,
      "loss": 0.0553,
      "step": 3798
    },
    {
      "epoch": 0.8643913538111491,
      "grad_norm": 1.4396023675409693,
      "learning_rate": 1.1601111517566831e-06,
      "loss": 0.0593,
      "step": 3799
    },
    {
      "epoch": 0.8646188850967008,
      "grad_norm": 1.302598540309529,
      "learning_rate": 1.1600649801343173e-06,
      "loss": 0.0761,
      "step": 3800
    },
    {
      "epoch": 0.8648464163822526,
      "grad_norm": 0.9487038984159104,
      "learning_rate": 1.1600187975762029e-06,
      "loss": 0.0533,
      "step": 3801
    },
    {
      "epoch": 0.8650739476678043,
      "grad_norm": 0.8318957052334727,
      "learning_rate": 1.159972604083284e-06,
      "loss": 0.0281,
      "step": 3802
    },
    {
      "epoch": 0.8653014789533561,
      "grad_norm": 1.2107437712653222,
      "learning_rate": 1.159926399656505e-06,
      "loss": 0.0465,
      "step": 3803
    },
    {
      "epoch": 0.8655290102389078,
      "grad_norm": 1.4023241592897753,
      "learning_rate": 1.1598801842968103e-06,
      "loss": 0.0395,
      "step": 3804
    },
    {
      "epoch": 0.8657565415244596,
      "grad_norm": 1.021600867386226,
      "learning_rate": 1.1598339580051439e-06,
      "loss": 0.042,
      "step": 3805
    },
    {
      "epoch": 0.8659840728100113,
      "grad_norm": 1.0092710955033857,
      "learning_rate": 1.159787720782451e-06,
      "loss": 0.0513,
      "step": 3806
    },
    {
      "epoch": 0.8662116040955632,
      "grad_norm": 1.517102557965297,
      "learning_rate": 1.1597414726296764e-06,
      "loss": 0.0823,
      "step": 3807
    },
    {
      "epoch": 0.8664391353811149,
      "grad_norm": 1.354754585750691,
      "learning_rate": 1.1596952135477656e-06,
      "loss": 0.0619,
      "step": 3808
    },
    {
      "epoch": 0.8666666666666667,
      "grad_norm": 2.107511992431661,
      "learning_rate": 1.1596489435376638e-06,
      "loss": 0.1053,
      "step": 3809
    },
    {
      "epoch": 0.8668941979522184,
      "grad_norm": 1.660224356687729,
      "learning_rate": 1.1596026626003168e-06,
      "loss": 0.1028,
      "step": 3810
    },
    {
      "epoch": 0.8671217292377702,
      "grad_norm": 1.1996277276445466,
      "learning_rate": 1.1595563707366705e-06,
      "loss": 0.0802,
      "step": 3811
    },
    {
      "epoch": 0.867349260523322,
      "grad_norm": 1.3676154758246724,
      "learning_rate": 1.1595100679476707e-06,
      "loss": 0.0662,
      "step": 3812
    },
    {
      "epoch": 0.8675767918088737,
      "grad_norm": 1.2961263261205425,
      "learning_rate": 1.1594637542342644e-06,
      "loss": 0.0569,
      "step": 3813
    },
    {
      "epoch": 0.8678043230944255,
      "grad_norm": 1.0638382967882807,
      "learning_rate": 1.1594174295973976e-06,
      "loss": 0.0527,
      "step": 3814
    },
    {
      "epoch": 0.8680318543799772,
      "grad_norm": 1.2853107355416882,
      "learning_rate": 1.1593710940380172e-06,
      "loss": 0.053,
      "step": 3815
    },
    {
      "epoch": 0.868259385665529,
      "grad_norm": 1.4112153541469374,
      "learning_rate": 1.1593247475570704e-06,
      "loss": 0.0652,
      "step": 3816
    },
    {
      "epoch": 0.8684869169510808,
      "grad_norm": 0.6600043862976881,
      "learning_rate": 1.1592783901555043e-06,
      "loss": 0.0235,
      "step": 3817
    },
    {
      "epoch": 0.8687144482366326,
      "grad_norm": 1.483562739056723,
      "learning_rate": 1.1592320218342665e-06,
      "loss": 0.0635,
      "step": 3818
    },
    {
      "epoch": 0.8689419795221843,
      "grad_norm": 1.0821712914528594,
      "learning_rate": 1.1591856425943044e-06,
      "loss": 0.0576,
      "step": 3819
    },
    {
      "epoch": 0.8691695108077361,
      "grad_norm": 1.2023944040312258,
      "learning_rate": 1.159139252436566e-06,
      "loss": 0.0448,
      "step": 3820
    },
    {
      "epoch": 0.8693970420932878,
      "grad_norm": 1.2261008981178048,
      "learning_rate": 1.1590928513619997e-06,
      "loss": 0.079,
      "step": 3821
    },
    {
      "epoch": 0.8696245733788396,
      "grad_norm": 0.9103768778718383,
      "learning_rate": 1.1590464393715536e-06,
      "loss": 0.0319,
      "step": 3822
    },
    {
      "epoch": 0.8698521046643913,
      "grad_norm": 1.0450339587824766,
      "learning_rate": 1.1590000164661763e-06,
      "loss": 0.0632,
      "step": 3823
    },
    {
      "epoch": 0.8700796359499431,
      "grad_norm": 1.0245424665524807,
      "learning_rate": 1.1589535826468168e-06,
      "loss": 0.0615,
      "step": 3824
    },
    {
      "epoch": 0.8703071672354948,
      "grad_norm": 1.624369945490237,
      "learning_rate": 1.158907137914424e-06,
      "loss": 0.0988,
      "step": 3825
    },
    {
      "epoch": 0.8705346985210466,
      "grad_norm": 1.1284177897194427,
      "learning_rate": 1.158860682269947e-06,
      "loss": 0.0488,
      "step": 3826
    },
    {
      "epoch": 0.8707622298065985,
      "grad_norm": 1.2276202548958175,
      "learning_rate": 1.1588142157143353e-06,
      "loss": 0.0608,
      "step": 3827
    },
    {
      "epoch": 0.8709897610921502,
      "grad_norm": 1.3822371198286005,
      "learning_rate": 1.1587677382485386e-06,
      "loss": 0.0468,
      "step": 3828
    },
    {
      "epoch": 0.871217292377702,
      "grad_norm": 1.106888167025077,
      "learning_rate": 1.158721249873507e-06,
      "loss": 0.0636,
      "step": 3829
    },
    {
      "epoch": 0.8714448236632537,
      "grad_norm": 0.9393892132586369,
      "learning_rate": 1.1586747505901904e-06,
      "loss": 0.0416,
      "step": 3830
    },
    {
      "epoch": 0.8716723549488055,
      "grad_norm": 1.1085193545572236,
      "learning_rate": 1.1586282403995395e-06,
      "loss": 0.0541,
      "step": 3831
    },
    {
      "epoch": 0.8718998862343572,
      "grad_norm": 1.345809504190513,
      "learning_rate": 1.1585817193025046e-06,
      "loss": 0.0513,
      "step": 3832
    },
    {
      "epoch": 0.872127417519909,
      "grad_norm": 0.8494118924798915,
      "learning_rate": 1.1585351873000365e-06,
      "loss": 0.03,
      "step": 3833
    },
    {
      "epoch": 0.8723549488054607,
      "grad_norm": 1.2837019051092946,
      "learning_rate": 1.1584886443930863e-06,
      "loss": 0.0529,
      "step": 3834
    },
    {
      "epoch": 0.8725824800910125,
      "grad_norm": 1.6558407244456526,
      "learning_rate": 1.1584420905826051e-06,
      "loss": 0.0833,
      "step": 3835
    },
    {
      "epoch": 0.8728100113765643,
      "grad_norm": 2.9844675595475887,
      "learning_rate": 1.1583955258695447e-06,
      "loss": 0.1321,
      "step": 3836
    },
    {
      "epoch": 0.8730375426621161,
      "grad_norm": 1.2652070426430007,
      "learning_rate": 1.1583489502548566e-06,
      "loss": 0.0795,
      "step": 3837
    },
    {
      "epoch": 0.8732650739476678,
      "grad_norm": 2.0137801774238344,
      "learning_rate": 1.1583023637394928e-06,
      "loss": 0.066,
      "step": 3838
    },
    {
      "epoch": 0.8734926052332196,
      "grad_norm": 1.2580173572147197,
      "learning_rate": 1.1582557663244052e-06,
      "loss": 0.0768,
      "step": 3839
    },
    {
      "epoch": 0.8737201365187713,
      "grad_norm": 1.300183703835354,
      "learning_rate": 1.1582091580105464e-06,
      "loss": 0.0609,
      "step": 3840
    },
    {
      "epoch": 0.8739476678043231,
      "grad_norm": 1.6278123129636566,
      "learning_rate": 1.158162538798869e-06,
      "loss": 0.0459,
      "step": 3841
    },
    {
      "epoch": 0.8741751990898748,
      "grad_norm": 1.4397538286299336,
      "learning_rate": 1.158115908690326e-06,
      "loss": 0.0895,
      "step": 3842
    },
    {
      "epoch": 0.8744027303754266,
      "grad_norm": 0.6209497855296535,
      "learning_rate": 1.1580692676858699e-06,
      "loss": 0.0383,
      "step": 3843
    },
    {
      "epoch": 0.8746302616609783,
      "grad_norm": 1.7840447917226478,
      "learning_rate": 1.1580226157864542e-06,
      "loss": 0.0761,
      "step": 3844
    },
    {
      "epoch": 0.8748577929465301,
      "grad_norm": 1.3094120438469554,
      "learning_rate": 1.1579759529930324e-06,
      "loss": 0.0403,
      "step": 3845
    },
    {
      "epoch": 0.875085324232082,
      "grad_norm": 2.141642949759066,
      "learning_rate": 1.1579292793065583e-06,
      "loss": 0.0849,
      "step": 3846
    },
    {
      "epoch": 0.8753128555176337,
      "grad_norm": 1.3320582954398776,
      "learning_rate": 1.157882594727986e-06,
      "loss": 0.0581,
      "step": 3847
    },
    {
      "epoch": 0.8755403868031855,
      "grad_norm": 2.7977842973683478,
      "learning_rate": 1.1578358992582689e-06,
      "loss": 0.0733,
      "step": 3848
    },
    {
      "epoch": 0.8757679180887372,
      "grad_norm": 1.273538584951798,
      "learning_rate": 1.1577891928983622e-06,
      "loss": 0.075,
      "step": 3849
    },
    {
      "epoch": 0.875995449374289,
      "grad_norm": 1.189871225174409,
      "learning_rate": 1.15774247564922e-06,
      "loss": 0.0533,
      "step": 3850
    },
    {
      "epoch": 0.8762229806598407,
      "grad_norm": 1.3416154016304795,
      "learning_rate": 1.1576957475117973e-06,
      "loss": 0.0687,
      "step": 3851
    },
    {
      "epoch": 0.8764505119453925,
      "grad_norm": 1.6440039914240385,
      "learning_rate": 1.1576490084870493e-06,
      "loss": 0.0573,
      "step": 3852
    },
    {
      "epoch": 0.8766780432309442,
      "grad_norm": 1.2895981463406705,
      "learning_rate": 1.1576022585759308e-06,
      "loss": 0.0626,
      "step": 3853
    },
    {
      "epoch": 0.876905574516496,
      "grad_norm": 1.0617961429852751,
      "learning_rate": 1.1575554977793975e-06,
      "loss": 0.0425,
      "step": 3854
    },
    {
      "epoch": 0.8771331058020477,
      "grad_norm": 0.7670265616309823,
      "learning_rate": 1.1575087260984056e-06,
      "loss": 0.0297,
      "step": 3855
    },
    {
      "epoch": 0.8773606370875996,
      "grad_norm": 0.9091034119562531,
      "learning_rate": 1.1574619435339101e-06,
      "loss": 0.0505,
      "step": 3856
    },
    {
      "epoch": 0.8775881683731513,
      "grad_norm": 0.721924678198625,
      "learning_rate": 1.157415150086868e-06,
      "loss": 0.0334,
      "step": 3857
    },
    {
      "epoch": 0.8778156996587031,
      "grad_norm": 1.028224217020184,
      "learning_rate": 1.1573683457582349e-06,
      "loss": 0.0504,
      "step": 3858
    },
    {
      "epoch": 0.8780432309442548,
      "grad_norm": 0.9265903973410824,
      "learning_rate": 1.157321530548968e-06,
      "loss": 0.0384,
      "step": 3859
    },
    {
      "epoch": 0.8782707622298066,
      "grad_norm": 1.0369597512628028,
      "learning_rate": 1.157274704460024e-06,
      "loss": 0.0421,
      "step": 3860
    },
    {
      "epoch": 0.8784982935153584,
      "grad_norm": 0.8116397369448615,
      "learning_rate": 1.1572278674923598e-06,
      "loss": 0.0382,
      "step": 3861
    },
    {
      "epoch": 0.8787258248009101,
      "grad_norm": 1.366642693958556,
      "learning_rate": 1.1571810196469326e-06,
      "loss": 0.0886,
      "step": 3862
    },
    {
      "epoch": 0.8789533560864619,
      "grad_norm": 0.9890955350681031,
      "learning_rate": 1.1571341609247003e-06,
      "loss": 0.0449,
      "step": 3863
    },
    {
      "epoch": 0.8791808873720136,
      "grad_norm": 1.4331436845492822,
      "learning_rate": 1.1570872913266202e-06,
      "loss": 0.0823,
      "step": 3864
    },
    {
      "epoch": 0.8794084186575654,
      "grad_norm": 0.9885779670626057,
      "learning_rate": 1.1570404108536501e-06,
      "loss": 0.0291,
      "step": 3865
    },
    {
      "epoch": 0.8796359499431172,
      "grad_norm": 1.0999651301917803,
      "learning_rate": 1.1569935195067487e-06,
      "loss": 0.041,
      "step": 3866
    },
    {
      "epoch": 0.879863481228669,
      "grad_norm": 1.1538168389141554,
      "learning_rate": 1.1569466172868737e-06,
      "loss": 0.0308,
      "step": 3867
    },
    {
      "epoch": 0.8800910125142207,
      "grad_norm": 0.737445292579344,
      "learning_rate": 1.1568997041949843e-06,
      "loss": 0.0304,
      "step": 3868
    },
    {
      "epoch": 0.8803185437997725,
      "grad_norm": 1.0078853967937846,
      "learning_rate": 1.156852780232039e-06,
      "loss": 0.0373,
      "step": 3869
    },
    {
      "epoch": 0.8805460750853242,
      "grad_norm": 1.1880834513401282,
      "learning_rate": 1.156805845398997e-06,
      "loss": 0.0483,
      "step": 3870
    },
    {
      "epoch": 0.880773606370876,
      "grad_norm": 1.0071545839978109,
      "learning_rate": 1.1567588996968173e-06,
      "loss": 0.0379,
      "step": 3871
    },
    {
      "epoch": 0.8810011376564277,
      "grad_norm": 1.0549498497637813,
      "learning_rate": 1.1567119431264598e-06,
      "loss": 0.058,
      "step": 3872
    },
    {
      "epoch": 0.8812286689419795,
      "grad_norm": 1.5102339027747615,
      "learning_rate": 1.156664975688884e-06,
      "loss": 0.0607,
      "step": 3873
    },
    {
      "epoch": 0.8814562002275312,
      "grad_norm": 1.8905446477817547,
      "learning_rate": 1.1566179973850496e-06,
      "loss": 0.11,
      "step": 3874
    },
    {
      "epoch": 0.8816837315130831,
      "grad_norm": 0.6411112129672515,
      "learning_rate": 1.156571008215917e-06,
      "loss": 0.0307,
      "step": 3875
    },
    {
      "epoch": 0.8819112627986349,
      "grad_norm": 0.748342490109174,
      "learning_rate": 1.1565240081824466e-06,
      "loss": 0.0284,
      "step": 3876
    },
    {
      "epoch": 0.8821387940841866,
      "grad_norm": 1.354485160913724,
      "learning_rate": 1.1564769972855987e-06,
      "loss": 0.0491,
      "step": 3877
    },
    {
      "epoch": 0.8823663253697384,
      "grad_norm": 1.0100612900654617,
      "learning_rate": 1.1564299755263345e-06,
      "loss": 0.061,
      "step": 3878
    },
    {
      "epoch": 0.8825938566552901,
      "grad_norm": 1.5275951944341757,
      "learning_rate": 1.1563829429056148e-06,
      "loss": 0.0997,
      "step": 3879
    },
    {
      "epoch": 0.8828213879408419,
      "grad_norm": 1.2004586776908042,
      "learning_rate": 1.156335899424401e-06,
      "loss": 0.0407,
      "step": 3880
    },
    {
      "epoch": 0.8830489192263936,
      "grad_norm": 1.191033638879632,
      "learning_rate": 1.1562888450836544e-06,
      "loss": 0.0705,
      "step": 3881
    },
    {
      "epoch": 0.8832764505119454,
      "grad_norm": 2.5519305254154117,
      "learning_rate": 1.156241779884337e-06,
      "loss": 0.0995,
      "step": 3882
    },
    {
      "epoch": 0.8835039817974971,
      "grad_norm": 1.1448529875713787,
      "learning_rate": 1.1561947038274104e-06,
      "loss": 0.0409,
      "step": 3883
    },
    {
      "epoch": 0.8837315130830489,
      "grad_norm": 1.3287084471615846,
      "learning_rate": 1.156147616913837e-06,
      "loss": 0.0658,
      "step": 3884
    },
    {
      "epoch": 0.8839590443686007,
      "grad_norm": 1.607480285345695,
      "learning_rate": 1.156100519144579e-06,
      "loss": 0.065,
      "step": 3885
    },
    {
      "epoch": 0.8841865756541525,
      "grad_norm": 0.8350311325241287,
      "learning_rate": 1.156053410520599e-06,
      "loss": 0.028,
      "step": 3886
    },
    {
      "epoch": 0.8844141069397042,
      "grad_norm": 1.2493913843113744,
      "learning_rate": 1.15600629104286e-06,
      "loss": 0.054,
      "step": 3887
    },
    {
      "epoch": 0.884641638225256,
      "grad_norm": 0.8917197871451082,
      "learning_rate": 1.1559591607123248e-06,
      "loss": 0.0529,
      "step": 3888
    },
    {
      "epoch": 0.8848691695108077,
      "grad_norm": 0.9543671098376685,
      "learning_rate": 1.1559120195299566e-06,
      "loss": 0.0482,
      "step": 3889
    },
    {
      "epoch": 0.8850967007963595,
      "grad_norm": 2.016256579364082,
      "learning_rate": 1.1558648674967191e-06,
      "loss": 0.0989,
      "step": 3890
    },
    {
      "epoch": 0.8853242320819112,
      "grad_norm": 1.1851720940142052,
      "learning_rate": 1.1558177046135761e-06,
      "loss": 0.0478,
      "step": 3891
    },
    {
      "epoch": 0.885551763367463,
      "grad_norm": 1.4662655950086683,
      "learning_rate": 1.1557705308814914e-06,
      "loss": 0.058,
      "step": 3892
    },
    {
      "epoch": 0.8857792946530147,
      "grad_norm": 0.7173696835484785,
      "learning_rate": 1.155723346301429e-06,
      "loss": 0.0316,
      "step": 3893
    },
    {
      "epoch": 0.8860068259385665,
      "grad_norm": 1.2706849402987643,
      "learning_rate": 1.1556761508743532e-06,
      "loss": 0.0822,
      "step": 3894
    },
    {
      "epoch": 0.8862343572241184,
      "grad_norm": 1.609473277966293,
      "learning_rate": 1.1556289446012292e-06,
      "loss": 0.0687,
      "step": 3895
    },
    {
      "epoch": 0.8864618885096701,
      "grad_norm": 1.388883872823661,
      "learning_rate": 1.155581727483021e-06,
      "loss": 0.0583,
      "step": 3896
    },
    {
      "epoch": 0.8866894197952219,
      "grad_norm": 1.2281956278668817,
      "learning_rate": 1.1555344995206941e-06,
      "loss": 0.0627,
      "step": 3897
    },
    {
      "epoch": 0.8869169510807736,
      "grad_norm": 0.814456870112058,
      "learning_rate": 1.1554872607152138e-06,
      "loss": 0.0294,
      "step": 3898
    },
    {
      "epoch": 0.8871444823663254,
      "grad_norm": 2.0519731112865855,
      "learning_rate": 1.1554400110675453e-06,
      "loss": 0.054,
      "step": 3899
    },
    {
      "epoch": 0.8873720136518771,
      "grad_norm": 0.61176670012776,
      "learning_rate": 1.1553927505786543e-06,
      "loss": 0.0195,
      "step": 3900
    },
    {
      "epoch": 0.8875995449374289,
      "grad_norm": 1.0712439727415621,
      "learning_rate": 1.1553454792495072e-06,
      "loss": 0.0397,
      "step": 3901
    },
    {
      "epoch": 0.8878270762229806,
      "grad_norm": 1.1593590248854935,
      "learning_rate": 1.1552981970810694e-06,
      "loss": 0.0551,
      "step": 3902
    },
    {
      "epoch": 0.8880546075085324,
      "grad_norm": 1.082392405172485,
      "learning_rate": 1.1552509040743078e-06,
      "loss": 0.0423,
      "step": 3903
    },
    {
      "epoch": 0.8882821387940842,
      "grad_norm": 1.2264806717691155,
      "learning_rate": 1.1552036002301891e-06,
      "loss": 0.0614,
      "step": 3904
    },
    {
      "epoch": 0.888509670079636,
      "grad_norm": 1.8769370812382,
      "learning_rate": 1.1551562855496796e-06,
      "loss": 0.1051,
      "step": 3905
    },
    {
      "epoch": 0.8887372013651877,
      "grad_norm": 0.8452780282160288,
      "learning_rate": 1.1551089600337465e-06,
      "loss": 0.0456,
      "step": 3906
    },
    {
      "epoch": 0.8889647326507395,
      "grad_norm": 1.0073082653206435,
      "learning_rate": 1.1550616236833574e-06,
      "loss": 0.0527,
      "step": 3907
    },
    {
      "epoch": 0.8891922639362912,
      "grad_norm": 1.4600634269759383,
      "learning_rate": 1.155014276499479e-06,
      "loss": 0.0589,
      "step": 3908
    },
    {
      "epoch": 0.889419795221843,
      "grad_norm": 0.9260386118996091,
      "learning_rate": 1.1549669184830796e-06,
      "loss": 0.0296,
      "step": 3909
    },
    {
      "epoch": 0.8896473265073948,
      "grad_norm": 1.4635888779797275,
      "learning_rate": 1.1549195496351271e-06,
      "loss": 0.092,
      "step": 3910
    },
    {
      "epoch": 0.8898748577929465,
      "grad_norm": 1.0776139556085673,
      "learning_rate": 1.1548721699565896e-06,
      "loss": 0.0517,
      "step": 3911
    },
    {
      "epoch": 0.8901023890784983,
      "grad_norm": 1.9827438270104123,
      "learning_rate": 1.1548247794484353e-06,
      "loss": 0.0953,
      "step": 3912
    },
    {
      "epoch": 0.89032992036405,
      "grad_norm": 1.0572692038363223,
      "learning_rate": 1.1547773781116326e-06,
      "loss": 0.0493,
      "step": 3913
    },
    {
      "epoch": 0.8905574516496019,
      "grad_norm": 0.9208219945202761,
      "learning_rate": 1.1547299659471509e-06,
      "loss": 0.0347,
      "step": 3914
    },
    {
      "epoch": 0.8907849829351536,
      "grad_norm": 0.7636650309650683,
      "learning_rate": 1.1546825429559585e-06,
      "loss": 0.0394,
      "step": 3915
    },
    {
      "epoch": 0.8910125142207054,
      "grad_norm": 1.8570270747724098,
      "learning_rate": 1.1546351091390253e-06,
      "loss": 0.0756,
      "step": 3916
    },
    {
      "epoch": 0.8912400455062571,
      "grad_norm": 2.2137237056291625,
      "learning_rate": 1.1545876644973202e-06,
      "loss": 0.0552,
      "step": 3917
    },
    {
      "epoch": 0.8914675767918089,
      "grad_norm": 1.314419860717642,
      "learning_rate": 1.1545402090318133e-06,
      "loss": 0.0746,
      "step": 3918
    },
    {
      "epoch": 0.8916951080773606,
      "grad_norm": 1.0660978255955464,
      "learning_rate": 1.1544927427434743e-06,
      "loss": 0.0544,
      "step": 3919
    },
    {
      "epoch": 0.8919226393629124,
      "grad_norm": 1.2951418524657141,
      "learning_rate": 1.1544452656332733e-06,
      "loss": 0.0585,
      "step": 3920
    },
    {
      "epoch": 0.8921501706484641,
      "grad_norm": 1.266569155658392,
      "learning_rate": 1.1543977777021808e-06,
      "loss": 0.0593,
      "step": 3921
    },
    {
      "epoch": 0.8923777019340159,
      "grad_norm": 1.0985430730465215,
      "learning_rate": 1.1543502789511671e-06,
      "loss": 0.057,
      "step": 3922
    },
    {
      "epoch": 0.8926052332195676,
      "grad_norm": 1.0051340610905692,
      "learning_rate": 1.1543027693812033e-06,
      "loss": 0.044,
      "step": 3923
    },
    {
      "epoch": 0.8928327645051195,
      "grad_norm": 0.8979646754734382,
      "learning_rate": 1.15425524899326e-06,
      "loss": 0.0381,
      "step": 3924
    },
    {
      "epoch": 0.8930602957906713,
      "grad_norm": 1.8955096750982607,
      "learning_rate": 1.154207717788309e-06,
      "loss": 0.0698,
      "step": 3925
    },
    {
      "epoch": 0.893287827076223,
      "grad_norm": 0.7837134343494108,
      "learning_rate": 1.1541601757673216e-06,
      "loss": 0.021,
      "step": 3926
    },
    {
      "epoch": 0.8935153583617748,
      "grad_norm": 1.406918667470208,
      "learning_rate": 1.154112622931269e-06,
      "loss": 0.0633,
      "step": 3927
    },
    {
      "epoch": 0.8937428896473265,
      "grad_norm": 1.6626757430888262,
      "learning_rate": 1.1540650592811233e-06,
      "loss": 0.0802,
      "step": 3928
    },
    {
      "epoch": 0.8939704209328783,
      "grad_norm": 2.7246018561325025,
      "learning_rate": 1.154017484817857e-06,
      "loss": 0.1196,
      "step": 3929
    },
    {
      "epoch": 0.89419795221843,
      "grad_norm": 1.3964197751853908,
      "learning_rate": 1.1539698995424423e-06,
      "loss": 0.0619,
      "step": 3930
    },
    {
      "epoch": 0.8944254835039818,
      "grad_norm": 1.2631409164792344,
      "learning_rate": 1.1539223034558513e-06,
      "loss": 0.0513,
      "step": 3931
    },
    {
      "epoch": 0.8946530147895335,
      "grad_norm": 1.0255766384543599,
      "learning_rate": 1.1538746965590572e-06,
      "loss": 0.0428,
      "step": 3932
    },
    {
      "epoch": 0.8948805460750853,
      "grad_norm": 0.4848782114580742,
      "learning_rate": 1.153827078853033e-06,
      "loss": 0.0267,
      "step": 3933
    },
    {
      "epoch": 0.8951080773606371,
      "grad_norm": 1.0020187672794154,
      "learning_rate": 1.1537794503387516e-06,
      "loss": 0.0494,
      "step": 3934
    },
    {
      "epoch": 0.8953356086461889,
      "grad_norm": 1.83637730371515,
      "learning_rate": 1.1537318110171867e-06,
      "loss": 0.0759,
      "step": 3935
    },
    {
      "epoch": 0.8955631399317406,
      "grad_norm": 1.9684228773442944,
      "learning_rate": 1.153684160889312e-06,
      "loss": 0.1322,
      "step": 3936
    },
    {
      "epoch": 0.8957906712172924,
      "grad_norm": 2.0616350638849172,
      "learning_rate": 1.1536364999561011e-06,
      "loss": 0.0855,
      "step": 3937
    },
    {
      "epoch": 0.8960182025028441,
      "grad_norm": 4.2442677487614,
      "learning_rate": 1.1535888282185283e-06,
      "loss": 0.1942,
      "step": 3938
    },
    {
      "epoch": 0.8962457337883959,
      "grad_norm": 1.5621794154117725,
      "learning_rate": 1.1535411456775682e-06,
      "loss": 0.0611,
      "step": 3939
    },
    {
      "epoch": 0.8964732650739476,
      "grad_norm": 1.0048473835575604,
      "learning_rate": 1.1534934523341952e-06,
      "loss": 0.0484,
      "step": 3940
    },
    {
      "epoch": 0.8967007963594994,
      "grad_norm": 0.9747410927457769,
      "learning_rate": 1.1534457481893834e-06,
      "loss": 0.0321,
      "step": 3941
    },
    {
      "epoch": 0.8969283276450511,
      "grad_norm": 0.8739461699351664,
      "learning_rate": 1.1533980332441085e-06,
      "loss": 0.0363,
      "step": 3942
    },
    {
      "epoch": 0.897155858930603,
      "grad_norm": 1.1216851883585208,
      "learning_rate": 1.1533503074993455e-06,
      "loss": 0.0423,
      "step": 3943
    },
    {
      "epoch": 0.8973833902161548,
      "grad_norm": 0.9832239028189882,
      "learning_rate": 1.15330257095607e-06,
      "loss": 0.0458,
      "step": 3944
    },
    {
      "epoch": 0.8976109215017065,
      "grad_norm": 0.8256408828846264,
      "learning_rate": 1.1532548236152574e-06,
      "loss": 0.0376,
      "step": 3945
    },
    {
      "epoch": 0.8978384527872583,
      "grad_norm": 2.409677737424434,
      "learning_rate": 1.1532070654778838e-06,
      "loss": 0.1075,
      "step": 3946
    },
    {
      "epoch": 0.89806598407281,
      "grad_norm": 1.6618831194214136,
      "learning_rate": 1.1531592965449249e-06,
      "loss": 0.0724,
      "step": 3947
    },
    {
      "epoch": 0.8982935153583618,
      "grad_norm": 1.1965129320838301,
      "learning_rate": 1.1531115168173574e-06,
      "loss": 0.0563,
      "step": 3948
    },
    {
      "epoch": 0.8985210466439135,
      "grad_norm": 1.6580223852928602,
      "learning_rate": 1.1530637262961574e-06,
      "loss": 0.0901,
      "step": 3949
    },
    {
      "epoch": 0.8987485779294653,
      "grad_norm": 2.0705952551342124,
      "learning_rate": 1.1530159249823022e-06,
      "loss": 0.1004,
      "step": 3950
    },
    {
      "epoch": 0.898976109215017,
      "grad_norm": 1.0142357819912748,
      "learning_rate": 1.1529681128767686e-06,
      "loss": 0.0412,
      "step": 3951
    },
    {
      "epoch": 0.8992036405005688,
      "grad_norm": 1.1350339928761526,
      "learning_rate": 1.1529202899805336e-06,
      "loss": 0.0538,
      "step": 3952
    },
    {
      "epoch": 0.8994311717861206,
      "grad_norm": 1.5799378264286232,
      "learning_rate": 1.1528724562945748e-06,
      "loss": 0.0833,
      "step": 3953
    },
    {
      "epoch": 0.8996587030716724,
      "grad_norm": 1.1208895769557774,
      "learning_rate": 1.1528246118198697e-06,
      "loss": 0.039,
      "step": 3954
    },
    {
      "epoch": 0.8998862343572241,
      "grad_norm": 1.160212672909766,
      "learning_rate": 1.1527767565573961e-06,
      "loss": 0.0271,
      "step": 3955
    },
    {
      "epoch": 0.9001137656427759,
      "grad_norm": 1.689340759598001,
      "learning_rate": 1.152728890508132e-06,
      "loss": 0.085,
      "step": 3956
    },
    {
      "epoch": 0.9003412969283277,
      "grad_norm": 6.049824984821008,
      "learning_rate": 1.1526810136730562e-06,
      "loss": 0.0767,
      "step": 3957
    },
    {
      "epoch": 0.9005688282138794,
      "grad_norm": 1.5293674377800182,
      "learning_rate": 1.1526331260531467e-06,
      "loss": 0.0652,
      "step": 3958
    },
    {
      "epoch": 0.9007963594994312,
      "grad_norm": 0.9268964770876272,
      "learning_rate": 1.1525852276493825e-06,
      "loss": 0.0373,
      "step": 3959
    },
    {
      "epoch": 0.9010238907849829,
      "grad_norm": 1.8729631573608785,
      "learning_rate": 1.1525373184627426e-06,
      "loss": 0.0883,
      "step": 3960
    },
    {
      "epoch": 0.9012514220705347,
      "grad_norm": 1.3972976693669097,
      "learning_rate": 1.1524893984942059e-06,
      "loss": 0.056,
      "step": 3961
    },
    {
      "epoch": 0.9014789533560864,
      "grad_norm": 1.0645332178479103,
      "learning_rate": 1.152441467744752e-06,
      "loss": 0.0482,
      "step": 3962
    },
    {
      "epoch": 0.9017064846416383,
      "grad_norm": 0.8491387015963779,
      "learning_rate": 1.1523935262153604e-06,
      "loss": 0.037,
      "step": 3963
    },
    {
      "epoch": 0.90193401592719,
      "grad_norm": 1.9173276752149597,
      "learning_rate": 1.152345573907011e-06,
      "loss": 0.0756,
      "step": 3964
    },
    {
      "epoch": 0.9021615472127418,
      "grad_norm": 1.0482343731205466,
      "learning_rate": 1.1522976108206838e-06,
      "loss": 0.0426,
      "step": 3965
    },
    {
      "epoch": 0.9023890784982935,
      "grad_norm": 1.5912063871488282,
      "learning_rate": 1.1522496369573592e-06,
      "loss": 0.0635,
      "step": 3966
    },
    {
      "epoch": 0.9026166097838453,
      "grad_norm": 1.0885136963654516,
      "learning_rate": 1.1522016523180177e-06,
      "loss": 0.0467,
      "step": 3967
    },
    {
      "epoch": 0.902844141069397,
      "grad_norm": 1.0253021604975354,
      "learning_rate": 1.15215365690364e-06,
      "loss": 0.0402,
      "step": 3968
    },
    {
      "epoch": 0.9030716723549488,
      "grad_norm": 1.6954250726492197,
      "learning_rate": 1.1521056507152068e-06,
      "loss": 0.0775,
      "step": 3969
    },
    {
      "epoch": 0.9032992036405005,
      "grad_norm": 2.193731037581251,
      "learning_rate": 1.1520576337536995e-06,
      "loss": 0.1161,
      "step": 3970
    },
    {
      "epoch": 0.9035267349260523,
      "grad_norm": 1.7077975253874977,
      "learning_rate": 1.1520096060200995e-06,
      "loss": 0.0947,
      "step": 3971
    },
    {
      "epoch": 0.903754266211604,
      "grad_norm": 1.5399563597984878,
      "learning_rate": 1.1519615675153884e-06,
      "loss": 0.1044,
      "step": 3972
    },
    {
      "epoch": 0.9039817974971559,
      "grad_norm": 1.7617983578353158,
      "learning_rate": 1.1519135182405477e-06,
      "loss": 0.0684,
      "step": 3973
    },
    {
      "epoch": 0.9042093287827077,
      "grad_norm": 1.3968116012863456,
      "learning_rate": 1.1518654581965597e-06,
      "loss": 0.0688,
      "step": 3974
    },
    {
      "epoch": 0.9044368600682594,
      "grad_norm": 1.1385754513385686,
      "learning_rate": 1.1518173873844068e-06,
      "loss": 0.0349,
      "step": 3975
    },
    {
      "epoch": 0.9046643913538112,
      "grad_norm": 0.8893866615862518,
      "learning_rate": 1.1517693058050714e-06,
      "loss": 0.0309,
      "step": 3976
    },
    {
      "epoch": 0.9048919226393629,
      "grad_norm": 1.539477830489965,
      "learning_rate": 1.151721213459536e-06,
      "loss": 0.0689,
      "step": 3977
    },
    {
      "epoch": 0.9051194539249147,
      "grad_norm": 1.1195140916495427,
      "learning_rate": 1.1516731103487836e-06,
      "loss": 0.051,
      "step": 3978
    },
    {
      "epoch": 0.9053469852104664,
      "grad_norm": 0.7220926512528832,
      "learning_rate": 1.1516249964737974e-06,
      "loss": 0.019,
      "step": 3979
    },
    {
      "epoch": 0.9055745164960182,
      "grad_norm": 1.0047441878180272,
      "learning_rate": 1.1515768718355607e-06,
      "loss": 0.0536,
      "step": 3980
    },
    {
      "epoch": 0.9058020477815699,
      "grad_norm": 1.1154458357876431,
      "learning_rate": 1.1515287364350573e-06,
      "loss": 0.0436,
      "step": 3981
    },
    {
      "epoch": 0.9060295790671218,
      "grad_norm": 1.083449314088406,
      "learning_rate": 1.1514805902732706e-06,
      "loss": 0.0403,
      "step": 3982
    },
    {
      "epoch": 0.9062571103526735,
      "grad_norm": 1.0839313718633037,
      "learning_rate": 1.151432433351185e-06,
      "loss": 0.0494,
      "step": 3983
    },
    {
      "epoch": 0.9064846416382253,
      "grad_norm": 1.6843996604661877,
      "learning_rate": 1.1513842656697844e-06,
      "loss": 0.0731,
      "step": 3984
    },
    {
      "epoch": 0.906712172923777,
      "grad_norm": 0.8893497549287934,
      "learning_rate": 1.1513360872300535e-06,
      "loss": 0.024,
      "step": 3985
    },
    {
      "epoch": 0.9069397042093288,
      "grad_norm": 1.6031300617012416,
      "learning_rate": 1.1512878980329771e-06,
      "loss": 0.0705,
      "step": 3986
    },
    {
      "epoch": 0.9071672354948805,
      "grad_norm": 1.129106422024724,
      "learning_rate": 1.1512396980795399e-06,
      "loss": 0.0568,
      "step": 3987
    },
    {
      "epoch": 0.9073947667804323,
      "grad_norm": 1.228932149349224,
      "learning_rate": 1.1511914873707269e-06,
      "loss": 0.0462,
      "step": 3988
    },
    {
      "epoch": 0.907622298065984,
      "grad_norm": 1.636371433136822,
      "learning_rate": 1.1511432659075234e-06,
      "loss": 0.0733,
      "step": 3989
    },
    {
      "epoch": 0.9078498293515358,
      "grad_norm": 1.1425256617133182,
      "learning_rate": 1.1510950336909154e-06,
      "loss": 0.0628,
      "step": 3990
    },
    {
      "epoch": 0.9080773606370875,
      "grad_norm": 1.01819882213196,
      "learning_rate": 1.1510467907218883e-06,
      "loss": 0.0442,
      "step": 3991
    },
    {
      "epoch": 0.9083048919226394,
      "grad_norm": 1.245888941943274,
      "learning_rate": 1.1509985370014283e-06,
      "loss": 0.0551,
      "step": 3992
    },
    {
      "epoch": 0.9085324232081912,
      "grad_norm": 1.3036968814546137,
      "learning_rate": 1.1509502725305214e-06,
      "loss": 0.0531,
      "step": 3993
    },
    {
      "epoch": 0.9087599544937429,
      "grad_norm": 0.9765098064954344,
      "learning_rate": 1.1509019973101542e-06,
      "loss": 0.0415,
      "step": 3994
    },
    {
      "epoch": 0.9089874857792947,
      "grad_norm": 0.8156281011799231,
      "learning_rate": 1.1508537113413134e-06,
      "loss": 0.041,
      "step": 3995
    },
    {
      "epoch": 0.9092150170648464,
      "grad_norm": 1.3936521974976825,
      "learning_rate": 1.1508054146249858e-06,
      "loss": 0.0397,
      "step": 3996
    },
    {
      "epoch": 0.9094425483503982,
      "grad_norm": 1.0646687699415507,
      "learning_rate": 1.1507571071621585e-06,
      "loss": 0.0357,
      "step": 3997
    },
    {
      "epoch": 0.9096700796359499,
      "grad_norm": 1.1522419098336925,
      "learning_rate": 1.1507087889538186e-06,
      "loss": 0.048,
      "step": 3998
    },
    {
      "epoch": 0.9098976109215017,
      "grad_norm": 1.2076402547928258,
      "learning_rate": 1.1506604600009542e-06,
      "loss": 0.0586,
      "step": 3999
    },
    {
      "epoch": 0.9101251422070534,
      "grad_norm": 1.0133219849273498,
      "learning_rate": 1.1506121203045524e-06,
      "loss": 0.0619,
      "step": 4000
    },
    {
      "epoch": 0.9103526734926052,
      "grad_norm": 0.8291805370083495,
      "learning_rate": 1.1505637698656014e-06,
      "loss": 0.0474,
      "step": 4001
    },
    {
      "epoch": 0.910580204778157,
      "grad_norm": 1.5524783948323588,
      "learning_rate": 1.1505154086850898e-06,
      "loss": 0.0696,
      "step": 4002
    },
    {
      "epoch": 0.9108077360637088,
      "grad_norm": 1.0896886188901207,
      "learning_rate": 1.1504670367640054e-06,
      "loss": 0.0349,
      "step": 4003
    },
    {
      "epoch": 0.9110352673492605,
      "grad_norm": 1.4368712196848394,
      "learning_rate": 1.1504186541033371e-06,
      "loss": 0.0972,
      "step": 4004
    },
    {
      "epoch": 0.9112627986348123,
      "grad_norm": 1.1609052845323367,
      "learning_rate": 1.150370260704074e-06,
      "loss": 0.0344,
      "step": 4005
    },
    {
      "epoch": 0.911490329920364,
      "grad_norm": 0.7261107332342037,
      "learning_rate": 1.1503218565672047e-06,
      "loss": 0.0378,
      "step": 4006
    },
    {
      "epoch": 0.9117178612059158,
      "grad_norm": 1.3224085375415395,
      "learning_rate": 1.1502734416937188e-06,
      "loss": 0.0806,
      "step": 4007
    },
    {
      "epoch": 0.9119453924914676,
      "grad_norm": 1.3655919422766778,
      "learning_rate": 1.150225016084606e-06,
      "loss": 0.0822,
      "step": 4008
    },
    {
      "epoch": 0.9121729237770193,
      "grad_norm": 0.8123755839338497,
      "learning_rate": 1.1501765797408558e-06,
      "loss": 0.0337,
      "step": 4009
    },
    {
      "epoch": 0.9124004550625711,
      "grad_norm": 1.8652137193155605,
      "learning_rate": 1.1501281326634578e-06,
      "loss": 0.0612,
      "step": 4010
    },
    {
      "epoch": 0.9126279863481229,
      "grad_norm": 0.9706539819037835,
      "learning_rate": 1.1500796748534026e-06,
      "loss": 0.0559,
      "step": 4011
    },
    {
      "epoch": 0.9128555176336747,
      "grad_norm": 0.6605470896041641,
      "learning_rate": 1.1500312063116803e-06,
      "loss": 0.0277,
      "step": 4012
    },
    {
      "epoch": 0.9130830489192264,
      "grad_norm": 2.0635463828713445,
      "learning_rate": 1.149982727039282e-06,
      "loss": 0.1237,
      "step": 4013
    },
    {
      "epoch": 0.9133105802047782,
      "grad_norm": 1.2700127801482208,
      "learning_rate": 1.149934237037198e-06,
      "loss": 0.0519,
      "step": 4014
    },
    {
      "epoch": 0.9135381114903299,
      "grad_norm": 1.0543579404973509,
      "learning_rate": 1.1498857363064198e-06,
      "loss": 0.0318,
      "step": 4015
    },
    {
      "epoch": 0.9137656427758817,
      "grad_norm": 1.3887452700800633,
      "learning_rate": 1.1498372248479383e-06,
      "loss": 0.0776,
      "step": 4016
    },
    {
      "epoch": 0.9139931740614334,
      "grad_norm": 1.1548765049955128,
      "learning_rate": 1.1497887026627451e-06,
      "loss": 0.0631,
      "step": 4017
    },
    {
      "epoch": 0.9142207053469852,
      "grad_norm": 1.4155291898606746,
      "learning_rate": 1.1497401697518318e-06,
      "loss": 0.0757,
      "step": 4018
    },
    {
      "epoch": 0.9144482366325369,
      "grad_norm": 1.5411778856885947,
      "learning_rate": 1.1496916261161908e-06,
      "loss": 0.0845,
      "step": 4019
    },
    {
      "epoch": 0.9146757679180887,
      "grad_norm": 0.8243491395066448,
      "learning_rate": 1.1496430717568136e-06,
      "loss": 0.0304,
      "step": 4020
    },
    {
      "epoch": 0.9149032992036406,
      "grad_norm": 1.0014800679630642,
      "learning_rate": 1.149594506674693e-06,
      "loss": 0.0443,
      "step": 4021
    },
    {
      "epoch": 0.9151308304891923,
      "grad_norm": 0.9825886288199939,
      "learning_rate": 1.1495459308708212e-06,
      "loss": 0.0285,
      "step": 4022
    },
    {
      "epoch": 0.9153583617747441,
      "grad_norm": 0.6708245548677098,
      "learning_rate": 1.1494973443461915e-06,
      "loss": 0.0258,
      "step": 4023
    },
    {
      "epoch": 0.9155858930602958,
      "grad_norm": 1.0354775698510426,
      "learning_rate": 1.1494487471017965e-06,
      "loss": 0.0516,
      "step": 4024
    },
    {
      "epoch": 0.9158134243458476,
      "grad_norm": 1.4705781324392022,
      "learning_rate": 1.1494001391386298e-06,
      "loss": 0.0588,
      "step": 4025
    },
    {
      "epoch": 0.9160409556313993,
      "grad_norm": 1.4496056117656928,
      "learning_rate": 1.1493515204576844e-06,
      "loss": 0.0453,
      "step": 4026
    },
    {
      "epoch": 0.9162684869169511,
      "grad_norm": 1.2256180174274225,
      "learning_rate": 1.1493028910599544e-06,
      "loss": 0.0474,
      "step": 4027
    },
    {
      "epoch": 0.9164960182025028,
      "grad_norm": 1.241479459058262,
      "learning_rate": 1.1492542509464333e-06,
      "loss": 0.0499,
      "step": 4028
    },
    {
      "epoch": 0.9167235494880546,
      "grad_norm": 1.0655237448875803,
      "learning_rate": 1.1492056001181157e-06,
      "loss": 0.04,
      "step": 4029
    },
    {
      "epoch": 0.9169510807736063,
      "grad_norm": 0.9311299042715484,
      "learning_rate": 1.1491569385759953e-06,
      "loss": 0.0353,
      "step": 4030
    },
    {
      "epoch": 0.9171786120591582,
      "grad_norm": 1.9476966702422809,
      "learning_rate": 1.1491082663210675e-06,
      "loss": 0.0822,
      "step": 4031
    },
    {
      "epoch": 0.9174061433447099,
      "grad_norm": 1.7080369705282124,
      "learning_rate": 1.1490595833543263e-06,
      "loss": 0.0921,
      "step": 4032
    },
    {
      "epoch": 0.9176336746302617,
      "grad_norm": 0.9700416110415793,
      "learning_rate": 1.1490108896767672e-06,
      "loss": 0.0376,
      "step": 4033
    },
    {
      "epoch": 0.9178612059158134,
      "grad_norm": 1.79218464701242,
      "learning_rate": 1.1489621852893849e-06,
      "loss": 0.0793,
      "step": 4034
    },
    {
      "epoch": 0.9180887372013652,
      "grad_norm": 1.5337857838361597,
      "learning_rate": 1.1489134701931753e-06,
      "loss": 0.0734,
      "step": 4035
    },
    {
      "epoch": 0.9183162684869169,
      "grad_norm": 2.2908857950335393,
      "learning_rate": 1.1488647443891339e-06,
      "loss": 0.1363,
      "step": 4036
    },
    {
      "epoch": 0.9185437997724687,
      "grad_norm": 0.9807386709388579,
      "learning_rate": 1.1488160078782565e-06,
      "loss": 0.0553,
      "step": 4037
    },
    {
      "epoch": 0.9187713310580204,
      "grad_norm": 1.610140836174014,
      "learning_rate": 1.148767260661539e-06,
      "loss": 0.0535,
      "step": 4038
    },
    {
      "epoch": 0.9189988623435722,
      "grad_norm": 1.8010870128097782,
      "learning_rate": 1.1487185027399783e-06,
      "loss": 0.0809,
      "step": 4039
    },
    {
      "epoch": 0.919226393629124,
      "grad_norm": 0.9946177354686938,
      "learning_rate": 1.1486697341145703e-06,
      "loss": 0.0308,
      "step": 4040
    },
    {
      "epoch": 0.9194539249146758,
      "grad_norm": 1.1243952505434611,
      "learning_rate": 1.148620954786312e-06,
      "loss": 0.0403,
      "step": 4041
    },
    {
      "epoch": 0.9196814562002276,
      "grad_norm": 1.1676663366717939,
      "learning_rate": 1.1485721647562005e-06,
      "loss": 0.0642,
      "step": 4042
    },
    {
      "epoch": 0.9199089874857793,
      "grad_norm": 1.2805472497836035,
      "learning_rate": 1.1485233640252328e-06,
      "loss": 0.0495,
      "step": 4043
    },
    {
      "epoch": 0.9201365187713311,
      "grad_norm": 1.176529756201314,
      "learning_rate": 1.1484745525944063e-06,
      "loss": 0.0425,
      "step": 4044
    },
    {
      "epoch": 0.9203640500568828,
      "grad_norm": 1.4292232195143082,
      "learning_rate": 1.1484257304647187e-06,
      "loss": 0.0792,
      "step": 4045
    },
    {
      "epoch": 0.9205915813424346,
      "grad_norm": 1.7932020137107854,
      "learning_rate": 1.1483768976371677e-06,
      "loss": 0.0727,
      "step": 4046
    },
    {
      "epoch": 0.9208191126279863,
      "grad_norm": 1.6644682540806308,
      "learning_rate": 1.1483280541127513e-06,
      "loss": 0.062,
      "step": 4047
    },
    {
      "epoch": 0.9210466439135381,
      "grad_norm": 1.1510951655579869,
      "learning_rate": 1.1482791998924681e-06,
      "loss": 0.0434,
      "step": 4048
    },
    {
      "epoch": 0.9212741751990898,
      "grad_norm": 1.19496202376025,
      "learning_rate": 1.1482303349773164e-06,
      "loss": 0.0679,
      "step": 4049
    },
    {
      "epoch": 0.9215017064846417,
      "grad_norm": 1.154368424237523,
      "learning_rate": 1.1481814593682946e-06,
      "loss": 0.0396,
      "step": 4050
    },
    {
      "epoch": 0.9217292377701934,
      "grad_norm": 1.3809033535643431,
      "learning_rate": 1.1481325730664023e-06,
      "loss": 0.0772,
      "step": 4051
    },
    {
      "epoch": 0.9219567690557452,
      "grad_norm": 0.9785189844102623,
      "learning_rate": 1.148083676072638e-06,
      "loss": 0.039,
      "step": 4052
    },
    {
      "epoch": 0.922184300341297,
      "grad_norm": 1.4126999693949231,
      "learning_rate": 1.1480347683880016e-06,
      "loss": 0.0704,
      "step": 4053
    },
    {
      "epoch": 0.9224118316268487,
      "grad_norm": 0.8860494719620977,
      "learning_rate": 1.1479858500134924e-06,
      "loss": 0.0269,
      "step": 4054
    },
    {
      "epoch": 0.9226393629124005,
      "grad_norm": 1.4070564902702778,
      "learning_rate": 1.14793692095011e-06,
      "loss": 0.0775,
      "step": 4055
    },
    {
      "epoch": 0.9228668941979522,
      "grad_norm": 1.1117624571631268,
      "learning_rate": 1.147887981198855e-06,
      "loss": 0.0474,
      "step": 4056
    },
    {
      "epoch": 0.923094425483504,
      "grad_norm": 1.4621020662224937,
      "learning_rate": 1.147839030760727e-06,
      "loss": 0.069,
      "step": 4057
    },
    {
      "epoch": 0.9233219567690557,
      "grad_norm": 1.4729100301695202,
      "learning_rate": 1.1477900696367269e-06,
      "loss": 0.0696,
      "step": 4058
    },
    {
      "epoch": 0.9235494880546075,
      "grad_norm": 0.9208872591470431,
      "learning_rate": 1.147741097827855e-06,
      "loss": 0.0365,
      "step": 4059
    },
    {
      "epoch": 0.9237770193401593,
      "grad_norm": 0.9240892354847577,
      "learning_rate": 1.1476921153351126e-06,
      "loss": 0.0405,
      "step": 4060
    },
    {
      "epoch": 0.9240045506257111,
      "grad_norm": 1.4959558432174869,
      "learning_rate": 1.1476431221595005e-06,
      "loss": 0.0463,
      "step": 4061
    },
    {
      "epoch": 0.9242320819112628,
      "grad_norm": 1.332519816167777,
      "learning_rate": 1.1475941183020203e-06,
      "loss": 0.0658,
      "step": 4062
    },
    {
      "epoch": 0.9244596131968146,
      "grad_norm": 2.187146405056203,
      "learning_rate": 1.1475451037636733e-06,
      "loss": 0.0943,
      "step": 4063
    },
    {
      "epoch": 0.9246871444823663,
      "grad_norm": 1.3435930034506374,
      "learning_rate": 1.1474960785454615e-06,
      "loss": 0.0726,
      "step": 4064
    },
    {
      "epoch": 0.9249146757679181,
      "grad_norm": 1.3012028567242737,
      "learning_rate": 1.1474470426483868e-06,
      "loss": 0.0528,
      "step": 4065
    },
    {
      "epoch": 0.9251422070534698,
      "grad_norm": 1.0819099650921233,
      "learning_rate": 1.1473979960734513e-06,
      "loss": 0.0426,
      "step": 4066
    },
    {
      "epoch": 0.9253697383390216,
      "grad_norm": 0.781744966406713,
      "learning_rate": 1.1473489388216574e-06,
      "loss": 0.0367,
      "step": 4067
    },
    {
      "epoch": 0.9255972696245733,
      "grad_norm": 1.151095346965489,
      "learning_rate": 1.1472998708940079e-06,
      "loss": 0.0619,
      "step": 4068
    },
    {
      "epoch": 0.9258248009101251,
      "grad_norm": 1.3142684365826751,
      "learning_rate": 1.1472507922915056e-06,
      "loss": 0.0446,
      "step": 4069
    },
    {
      "epoch": 0.926052332195677,
      "grad_norm": 1.3392798642883632,
      "learning_rate": 1.1472017030151536e-06,
      "loss": 0.0675,
      "step": 4070
    },
    {
      "epoch": 0.9262798634812287,
      "grad_norm": 0.7396122034506598,
      "learning_rate": 1.147152603065955e-06,
      "loss": 0.0436,
      "step": 4071
    },
    {
      "epoch": 0.9265073947667805,
      "grad_norm": 1.3292817664212946,
      "learning_rate": 1.1471034924449133e-06,
      "loss": 0.0562,
      "step": 4072
    },
    {
      "epoch": 0.9267349260523322,
      "grad_norm": 1.9912088396901564,
      "learning_rate": 1.1470543711530328e-06,
      "loss": 0.0629,
      "step": 4073
    },
    {
      "epoch": 0.926962457337884,
      "grad_norm": 1.3570840904239112,
      "learning_rate": 1.147005239191317e-06,
      "loss": 0.0667,
      "step": 4074
    },
    {
      "epoch": 0.9271899886234357,
      "grad_norm": 1.2559298173099012,
      "learning_rate": 1.1469560965607699e-06,
      "loss": 0.0518,
      "step": 4075
    },
    {
      "epoch": 0.9274175199089875,
      "grad_norm": 1.0348081180884323,
      "learning_rate": 1.1469069432623965e-06,
      "loss": 0.0517,
      "step": 4076
    },
    {
      "epoch": 0.9276450511945392,
      "grad_norm": 1.0484875570709722,
      "learning_rate": 1.1468577792972004e-06,
      "loss": 0.0465,
      "step": 4077
    },
    {
      "epoch": 0.927872582480091,
      "grad_norm": 1.09203642071682,
      "learning_rate": 1.1468086046661874e-06,
      "loss": 0.0548,
      "step": 4078
    },
    {
      "epoch": 0.9281001137656428,
      "grad_norm": 1.7712907658090584,
      "learning_rate": 1.146759419370362e-06,
      "loss": 0.1043,
      "step": 4079
    },
    {
      "epoch": 0.9283276450511946,
      "grad_norm": 0.8782441484375744,
      "learning_rate": 1.14671022341073e-06,
      "loss": 0.0241,
      "step": 4080
    },
    {
      "epoch": 0.9285551763367463,
      "grad_norm": 1.3964199439198308,
      "learning_rate": 1.1466610167882963e-06,
      "loss": 0.0789,
      "step": 4081
    },
    {
      "epoch": 0.9287827076222981,
      "grad_norm": 1.3164169043622294,
      "learning_rate": 1.1466117995040666e-06,
      "loss": 0.0635,
      "step": 4082
    },
    {
      "epoch": 0.9290102389078498,
      "grad_norm": 1.2381160175958454,
      "learning_rate": 1.1465625715590473e-06,
      "loss": 0.0427,
      "step": 4083
    },
    {
      "epoch": 0.9292377701934016,
      "grad_norm": 1.0698402996548797,
      "learning_rate": 1.146513332954244e-06,
      "loss": 0.0626,
      "step": 4084
    },
    {
      "epoch": 0.9294653014789533,
      "grad_norm": 1.534739951538693,
      "learning_rate": 1.1464640836906635e-06,
      "loss": 0.0587,
      "step": 4085
    },
    {
      "epoch": 0.9296928327645051,
      "grad_norm": 1.0552598639917363,
      "learning_rate": 1.146414823769312e-06,
      "loss": 0.046,
      "step": 4086
    },
    {
      "epoch": 0.9299203640500568,
      "grad_norm": 0.7826783824026263,
      "learning_rate": 1.1463655531911963e-06,
      "loss": 0.0308,
      "step": 4087
    },
    {
      "epoch": 0.9301478953356086,
      "grad_norm": 0.8273022304812389,
      "learning_rate": 1.1463162719573236e-06,
      "loss": 0.031,
      "step": 4088
    },
    {
      "epoch": 0.9303754266211605,
      "grad_norm": 1.3288942117067748,
      "learning_rate": 1.1462669800687012e-06,
      "loss": 0.0564,
      "step": 4089
    },
    {
      "epoch": 0.9306029579067122,
      "grad_norm": 0.8195424504406558,
      "learning_rate": 1.1462176775263365e-06,
      "loss": 0.0349,
      "step": 4090
    },
    {
      "epoch": 0.930830489192264,
      "grad_norm": 1.69823108141483,
      "learning_rate": 1.1461683643312366e-06,
      "loss": 0.0698,
      "step": 4091
    },
    {
      "epoch": 0.9310580204778157,
      "grad_norm": 1.5933893580265006,
      "learning_rate": 1.1461190404844103e-06,
      "loss": 0.0748,
      "step": 4092
    },
    {
      "epoch": 0.9312855517633675,
      "grad_norm": 1.4412482142890468,
      "learning_rate": 1.1460697059868648e-06,
      "loss": 0.086,
      "step": 4093
    },
    {
      "epoch": 0.9315130830489192,
      "grad_norm": 1.1113383236310055,
      "learning_rate": 1.146020360839609e-06,
      "loss": 0.0379,
      "step": 4094
    },
    {
      "epoch": 0.931740614334471,
      "grad_norm": 0.7200940731856813,
      "learning_rate": 1.1459710050436513e-06,
      "loss": 0.0356,
      "step": 4095
    },
    {
      "epoch": 0.9319681456200227,
      "grad_norm": 1.4853749484298682,
      "learning_rate": 1.1459216385999999e-06,
      "loss": 0.0559,
      "step": 4096
    },
    {
      "epoch": 0.9321956769055745,
      "grad_norm": 0.9574272073778142,
      "learning_rate": 1.1458722615096648e-06,
      "loss": 0.0276,
      "step": 4097
    },
    {
      "epoch": 0.9324232081911262,
      "grad_norm": 0.8174617084003479,
      "learning_rate": 1.1458228737736542e-06,
      "loss": 0.0327,
      "step": 4098
    },
    {
      "epoch": 0.9326507394766781,
      "grad_norm": 1.3250207983841498,
      "learning_rate": 1.145773475392978e-06,
      "loss": 0.0516,
      "step": 4099
    },
    {
      "epoch": 0.9328782707622298,
      "grad_norm": 1.22249857755224,
      "learning_rate": 1.145724066368646e-06,
      "loss": 0.0637,
      "step": 4100
    },
    {
      "epoch": 0.9331058020477816,
      "grad_norm": 1.4663762999847116,
      "learning_rate": 1.1456746467016675e-06,
      "loss": 0.0567,
      "step": 4101
    },
    {
      "epoch": 0.9333333333333333,
      "grad_norm": 1.4030554644294082,
      "learning_rate": 1.1456252163930528e-06,
      "loss": 0.0584,
      "step": 4102
    },
    {
      "epoch": 0.9335608646188851,
      "grad_norm": 1.5109418446571456,
      "learning_rate": 1.1455757754438122e-06,
      "loss": 0.0497,
      "step": 4103
    },
    {
      "epoch": 0.9337883959044369,
      "grad_norm": 1.0191531582003377,
      "learning_rate": 1.1455263238549563e-06,
      "loss": 0.0562,
      "step": 4104
    },
    {
      "epoch": 0.9340159271899886,
      "grad_norm": 0.9931115670125917,
      "learning_rate": 1.1454768616274955e-06,
      "loss": 0.0494,
      "step": 4105
    },
    {
      "epoch": 0.9342434584755404,
      "grad_norm": 1.0861122916963688,
      "learning_rate": 1.1454273887624407e-06,
      "loss": 0.0548,
      "step": 4106
    },
    {
      "epoch": 0.9344709897610921,
      "grad_norm": 0.9828952905986,
      "learning_rate": 1.1453779052608032e-06,
      "loss": 0.0374,
      "step": 4107
    },
    {
      "epoch": 0.9346985210466439,
      "grad_norm": 1.5222459429366253,
      "learning_rate": 1.1453284111235947e-06,
      "loss": 0.0818,
      "step": 4108
    },
    {
      "epoch": 0.9349260523321957,
      "grad_norm": 1.1153500509733376,
      "learning_rate": 1.145278906351826e-06,
      "loss": 0.0409,
      "step": 4109
    },
    {
      "epoch": 0.9351535836177475,
      "grad_norm": 1.0587413994265493,
      "learning_rate": 1.1452293909465095e-06,
      "loss": 0.0669,
      "step": 4110
    },
    {
      "epoch": 0.9353811149032992,
      "grad_norm": 0.809823129379949,
      "learning_rate": 1.145179864908657e-06,
      "loss": 0.0304,
      "step": 4111
    },
    {
      "epoch": 0.935608646188851,
      "grad_norm": 1.173627001973459,
      "learning_rate": 1.1451303282392808e-06,
      "loss": 0.046,
      "step": 4112
    },
    {
      "epoch": 0.9358361774744027,
      "grad_norm": 1.301208608992828,
      "learning_rate": 1.145080780939393e-06,
      "loss": 0.0457,
      "step": 4113
    },
    {
      "epoch": 0.9360637087599545,
      "grad_norm": 1.1311235005804838,
      "learning_rate": 1.145031223010007e-06,
      "loss": 0.0293,
      "step": 4114
    },
    {
      "epoch": 0.9362912400455062,
      "grad_norm": 1.754429989040699,
      "learning_rate": 1.1449816544521347e-06,
      "loss": 0.0962,
      "step": 4115
    },
    {
      "epoch": 0.936518771331058,
      "grad_norm": 1.1523409260488855,
      "learning_rate": 1.1449320752667898e-06,
      "loss": 0.0622,
      "step": 4116
    },
    {
      "epoch": 0.9367463026166097,
      "grad_norm": 0.9453806040405214,
      "learning_rate": 1.1448824854549856e-06,
      "loss": 0.0303,
      "step": 4117
    },
    {
      "epoch": 0.9369738339021616,
      "grad_norm": 1.270090887527193,
      "learning_rate": 1.1448328850177356e-06,
      "loss": 0.057,
      "step": 4118
    },
    {
      "epoch": 0.9372013651877134,
      "grad_norm": 1.1742323486339012,
      "learning_rate": 1.1447832739560533e-06,
      "loss": 0.0689,
      "step": 4119
    },
    {
      "epoch": 0.9374288964732651,
      "grad_norm": 1.0965078798779573,
      "learning_rate": 1.1447336522709528e-06,
      "loss": 0.0616,
      "step": 4120
    },
    {
      "epoch": 0.9376564277588169,
      "grad_norm": 1.1403738547142621,
      "learning_rate": 1.1446840199634483e-06,
      "loss": 0.0681,
      "step": 4121
    },
    {
      "epoch": 0.9378839590443686,
      "grad_norm": 1.176539532491646,
      "learning_rate": 1.1446343770345544e-06,
      "loss": 0.0369,
      "step": 4122
    },
    {
      "epoch": 0.9381114903299204,
      "grad_norm": 0.8950308841444714,
      "learning_rate": 1.1445847234852853e-06,
      "loss": 0.0508,
      "step": 4123
    },
    {
      "epoch": 0.9383390216154721,
      "grad_norm": 2.6033063822464553,
      "learning_rate": 1.1445350593166559e-06,
      "loss": 0.0865,
      "step": 4124
    },
    {
      "epoch": 0.9385665529010239,
      "grad_norm": 1.4419905128708395,
      "learning_rate": 1.1444853845296816e-06,
      "loss": 0.0922,
      "step": 4125
    },
    {
      "epoch": 0.9387940841865756,
      "grad_norm": 0.810209427625322,
      "learning_rate": 1.1444356991253774e-06,
      "loss": 0.0302,
      "step": 4126
    },
    {
      "epoch": 0.9390216154721274,
      "grad_norm": 1.4059215847764233,
      "learning_rate": 1.1443860031047589e-06,
      "loss": 0.0584,
      "step": 4127
    },
    {
      "epoch": 0.9392491467576792,
      "grad_norm": 1.3946503335439686,
      "learning_rate": 1.1443362964688416e-06,
      "loss": 0.0665,
      "step": 4128
    },
    {
      "epoch": 0.939476678043231,
      "grad_norm": 2.123996937731276,
      "learning_rate": 1.1442865792186413e-06,
      "loss": 0.0869,
      "step": 4129
    },
    {
      "epoch": 0.9397042093287827,
      "grad_norm": 1.0634024569590812,
      "learning_rate": 1.1442368513551746e-06,
      "loss": 0.0418,
      "step": 4130
    },
    {
      "epoch": 0.9399317406143345,
      "grad_norm": 1.063642439356403,
      "learning_rate": 1.1441871128794576e-06,
      "loss": 0.0341,
      "step": 4131
    },
    {
      "epoch": 0.9401592718998862,
      "grad_norm": 0.9460769468351368,
      "learning_rate": 1.1441373637925068e-06,
      "loss": 0.0533,
      "step": 4132
    },
    {
      "epoch": 0.940386803185438,
      "grad_norm": 1.764351905359598,
      "learning_rate": 1.1440876040953392e-06,
      "loss": 0.1242,
      "step": 4133
    },
    {
      "epoch": 0.9406143344709897,
      "grad_norm": 1.4558757328006684,
      "learning_rate": 1.1440378337889713e-06,
      "loss": 0.0763,
      "step": 4134
    },
    {
      "epoch": 0.9408418657565415,
      "grad_norm": 0.9021601245337508,
      "learning_rate": 1.143988052874421e-06,
      "loss": 0.0409,
      "step": 4135
    },
    {
      "epoch": 0.9410693970420932,
      "grad_norm": 1.7188429838557109,
      "learning_rate": 1.143938261352705e-06,
      "loss": 0.069,
      "step": 4136
    },
    {
      "epoch": 0.941296928327645,
      "grad_norm": 1.2572237568519273,
      "learning_rate": 1.1438884592248416e-06,
      "loss": 0.0368,
      "step": 4137
    },
    {
      "epoch": 0.9415244596131969,
      "grad_norm": 1.2286297911170685,
      "learning_rate": 1.1438386464918483e-06,
      "loss": 0.0371,
      "step": 4138
    },
    {
      "epoch": 0.9417519908987486,
      "grad_norm": 1.664747740752945,
      "learning_rate": 1.1437888231547434e-06,
      "loss": 0.0793,
      "step": 4139
    },
    {
      "epoch": 0.9419795221843004,
      "grad_norm": 0.9027458023794878,
      "learning_rate": 1.143738989214545e-06,
      "loss": 0.0359,
      "step": 4140
    },
    {
      "epoch": 0.9422070534698521,
      "grad_norm": 0.9844227932686808,
      "learning_rate": 1.1436891446722718e-06,
      "loss": 0.0398,
      "step": 4141
    },
    {
      "epoch": 0.9424345847554039,
      "grad_norm": 2.473079458381896,
      "learning_rate": 1.1436392895289423e-06,
      "loss": 0.1426,
      "step": 4142
    },
    {
      "epoch": 0.9426621160409556,
      "grad_norm": 1.6695549407826555,
      "learning_rate": 1.1435894237855754e-06,
      "loss": 0.0569,
      "step": 4143
    },
    {
      "epoch": 0.9428896473265074,
      "grad_norm": 0.9951741584954418,
      "learning_rate": 1.1435395474431906e-06,
      "loss": 0.0354,
      "step": 4144
    },
    {
      "epoch": 0.9431171786120591,
      "grad_norm": 1.3165779040583256,
      "learning_rate": 1.143489660502807e-06,
      "loss": 0.062,
      "step": 4145
    },
    {
      "epoch": 0.9433447098976109,
      "grad_norm": 1.1419121851198617,
      "learning_rate": 1.1434397629654445e-06,
      "loss": 0.0628,
      "step": 4146
    },
    {
      "epoch": 0.9435722411831626,
      "grad_norm": 1.1143992408718018,
      "learning_rate": 1.1433898548321226e-06,
      "loss": 0.0461,
      "step": 4147
    },
    {
      "epoch": 0.9437997724687145,
      "grad_norm": 1.4419058001216918,
      "learning_rate": 1.1433399361038614e-06,
      "loss": 0.0622,
      "step": 4148
    },
    {
      "epoch": 0.9440273037542662,
      "grad_norm": 1.0878632797009244,
      "learning_rate": 1.1432900067816813e-06,
      "loss": 0.051,
      "step": 4149
    },
    {
      "epoch": 0.944254835039818,
      "grad_norm": 0.8678017410178684,
      "learning_rate": 1.1432400668666028e-06,
      "loss": 0.036,
      "step": 4150
    },
    {
      "epoch": 0.9444823663253697,
      "grad_norm": 2.5565159682952325,
      "learning_rate": 1.1431901163596462e-06,
      "loss": 0.1481,
      "step": 4151
    },
    {
      "epoch": 0.9447098976109215,
      "grad_norm": 1.1719746596557403,
      "learning_rate": 1.1431401552618327e-06,
      "loss": 0.0654,
      "step": 4152
    },
    {
      "epoch": 0.9449374288964733,
      "grad_norm": 0.7399300932811994,
      "learning_rate": 1.1430901835741833e-06,
      "loss": 0.0426,
      "step": 4153
    },
    {
      "epoch": 0.945164960182025,
      "grad_norm": 1.5392915168318702,
      "learning_rate": 1.1430402012977195e-06,
      "loss": 0.064,
      "step": 4154
    },
    {
      "epoch": 0.9453924914675768,
      "grad_norm": 0.8642603027613659,
      "learning_rate": 1.1429902084334627e-06,
      "loss": 0.0391,
      "step": 4155
    },
    {
      "epoch": 0.9456200227531285,
      "grad_norm": 1.5792578097649426,
      "learning_rate": 1.1429402049824348e-06,
      "loss": 0.0864,
      "step": 4156
    },
    {
      "epoch": 0.9458475540386804,
      "grad_norm": 1.0656243196471682,
      "learning_rate": 1.1428901909456575e-06,
      "loss": 0.0643,
      "step": 4157
    },
    {
      "epoch": 0.9460750853242321,
      "grad_norm": 0.7834798421355951,
      "learning_rate": 1.1428401663241533e-06,
      "loss": 0.0255,
      "step": 4158
    },
    {
      "epoch": 0.9463026166097839,
      "grad_norm": 0.7290481198035724,
      "learning_rate": 1.1427901311189444e-06,
      "loss": 0.0347,
      "step": 4159
    },
    {
      "epoch": 0.9465301478953356,
      "grad_norm": 1.0906839196972904,
      "learning_rate": 1.1427400853310536e-06,
      "loss": 0.0416,
      "step": 4160
    },
    {
      "epoch": 0.9467576791808874,
      "grad_norm": 0.7797257513708735,
      "learning_rate": 1.1426900289615034e-06,
      "loss": 0.0332,
      "step": 4161
    },
    {
      "epoch": 0.9469852104664391,
      "grad_norm": 0.9828933034646482,
      "learning_rate": 1.1426399620113174e-06,
      "loss": 0.0426,
      "step": 4162
    },
    {
      "epoch": 0.9472127417519909,
      "grad_norm": 1.0815564353692642,
      "learning_rate": 1.1425898844815183e-06,
      "loss": 0.03,
      "step": 4163
    },
    {
      "epoch": 0.9474402730375426,
      "grad_norm": 1.3070693062923497,
      "learning_rate": 1.1425397963731303e-06,
      "loss": 0.0761,
      "step": 4164
    },
    {
      "epoch": 0.9476678043230944,
      "grad_norm": 1.9681632277415542,
      "learning_rate": 1.1424896976871763e-06,
      "loss": 0.0751,
      "step": 4165
    },
    {
      "epoch": 0.9478953356086461,
      "grad_norm": 1.3403750162598331,
      "learning_rate": 1.1424395884246808e-06,
      "loss": 0.0645,
      "step": 4166
    },
    {
      "epoch": 0.948122866894198,
      "grad_norm": 1.4856178047017083,
      "learning_rate": 1.1423894685866677e-06,
      "loss": 0.0524,
      "step": 4167
    },
    {
      "epoch": 0.9483503981797498,
      "grad_norm": 2.390346331051792,
      "learning_rate": 1.1423393381741614e-06,
      "loss": 0.1007,
      "step": 4168
    },
    {
      "epoch": 0.9485779294653015,
      "grad_norm": 1.9335775336716192,
      "learning_rate": 1.1422891971881867e-06,
      "loss": 0.0909,
      "step": 4169
    },
    {
      "epoch": 0.9488054607508533,
      "grad_norm": 1.4846712557080712,
      "learning_rate": 1.142239045629768e-06,
      "loss": 0.0706,
      "step": 4170
    },
    {
      "epoch": 0.949032992036405,
      "grad_norm": 1.2534969921049077,
      "learning_rate": 1.1421888834999306e-06,
      "loss": 0.0637,
      "step": 4171
    },
    {
      "epoch": 0.9492605233219568,
      "grad_norm": 1.5488362147727934,
      "learning_rate": 1.1421387107996993e-06,
      "loss": 0.065,
      "step": 4172
    },
    {
      "epoch": 0.9494880546075085,
      "grad_norm": 1.8544510157017915,
      "learning_rate": 1.1420885275301001e-06,
      "loss": 0.0783,
      "step": 4173
    },
    {
      "epoch": 0.9497155858930603,
      "grad_norm": 1.2775832861301255,
      "learning_rate": 1.1420383336921583e-06,
      "loss": 0.0628,
      "step": 4174
    },
    {
      "epoch": 0.949943117178612,
      "grad_norm": 1.1882749927515044,
      "learning_rate": 1.1419881292869e-06,
      "loss": 0.053,
      "step": 4175
    },
    {
      "epoch": 0.9501706484641638,
      "grad_norm": 1.1112594425917803,
      "learning_rate": 1.1419379143153511e-06,
      "loss": 0.0391,
      "step": 4176
    },
    {
      "epoch": 0.9503981797497156,
      "grad_norm": 1.4092744149498044,
      "learning_rate": 1.1418876887785379e-06,
      "loss": 0.0588,
      "step": 4177
    },
    {
      "epoch": 0.9506257110352674,
      "grad_norm": 1.2859140834039406,
      "learning_rate": 1.1418374526774872e-06,
      "loss": 0.0449,
      "step": 4178
    },
    {
      "epoch": 0.9508532423208191,
      "grad_norm": 1.7602664958939818,
      "learning_rate": 1.1417872060132251e-06,
      "loss": 0.066,
      "step": 4179
    },
    {
      "epoch": 0.9510807736063709,
      "grad_norm": 1.7687663881511184,
      "learning_rate": 1.1417369487867793e-06,
      "loss": 0.0737,
      "step": 4180
    },
    {
      "epoch": 0.9513083048919226,
      "grad_norm": 1.0504917072097133,
      "learning_rate": 1.1416866809991763e-06,
      "loss": 0.0571,
      "step": 4181
    },
    {
      "epoch": 0.9515358361774744,
      "grad_norm": 0.9175209845403075,
      "learning_rate": 1.1416364026514443e-06,
      "loss": 0.0473,
      "step": 4182
    },
    {
      "epoch": 0.9517633674630261,
      "grad_norm": 1.1584022734622255,
      "learning_rate": 1.1415861137446099e-06,
      "loss": 0.0615,
      "step": 4183
    },
    {
      "epoch": 0.9519908987485779,
      "grad_norm": 0.9605566486118688,
      "learning_rate": 1.1415358142797018e-06,
      "loss": 0.0468,
      "step": 4184
    },
    {
      "epoch": 0.9522184300341296,
      "grad_norm": 1.0547428180217657,
      "learning_rate": 1.1414855042577474e-06,
      "loss": 0.0578,
      "step": 4185
    },
    {
      "epoch": 0.9524459613196815,
      "grad_norm": 1.664335132859367,
      "learning_rate": 1.1414351836797755e-06,
      "loss": 0.0929,
      "step": 4186
    },
    {
      "epoch": 0.9526734926052333,
      "grad_norm": 1.326192413657784,
      "learning_rate": 1.1413848525468139e-06,
      "loss": 0.0583,
      "step": 4187
    },
    {
      "epoch": 0.952901023890785,
      "grad_norm": 1.475570664457709,
      "learning_rate": 1.1413345108598916e-06,
      "loss": 0.0626,
      "step": 4188
    },
    {
      "epoch": 0.9531285551763368,
      "grad_norm": 1.7815203562597235,
      "learning_rate": 1.1412841586200378e-06,
      "loss": 0.0717,
      "step": 4189
    },
    {
      "epoch": 0.9533560864618885,
      "grad_norm": 1.7331207453848758,
      "learning_rate": 1.1412337958282812e-06,
      "loss": 0.0621,
      "step": 4190
    },
    {
      "epoch": 0.9535836177474403,
      "grad_norm": 0.5286320379408398,
      "learning_rate": 1.1411834224856514e-06,
      "loss": 0.0279,
      "step": 4191
    },
    {
      "epoch": 0.953811149032992,
      "grad_norm": 0.8095894659074508,
      "learning_rate": 1.1411330385931776e-06,
      "loss": 0.0342,
      "step": 4192
    },
    {
      "epoch": 0.9540386803185438,
      "grad_norm": 1.1720468896203664,
      "learning_rate": 1.1410826441518898e-06,
      "loss": 0.0769,
      "step": 4193
    },
    {
      "epoch": 0.9542662116040955,
      "grad_norm": 1.6215729451396426,
      "learning_rate": 1.1410322391628179e-06,
      "loss": 0.064,
      "step": 4194
    },
    {
      "epoch": 0.9544937428896473,
      "grad_norm": 0.9975722592735233,
      "learning_rate": 1.140981823626992e-06,
      "loss": 0.033,
      "step": 4195
    },
    {
      "epoch": 0.9547212741751991,
      "grad_norm": 0.961294779276341,
      "learning_rate": 1.1409313975454429e-06,
      "loss": 0.0556,
      "step": 4196
    },
    {
      "epoch": 0.9549488054607509,
      "grad_norm": 1.195259518530016,
      "learning_rate": 1.1408809609192007e-06,
      "loss": 0.0685,
      "step": 4197
    },
    {
      "epoch": 0.9551763367463026,
      "grad_norm": 1.2021438816670518,
      "learning_rate": 1.1408305137492963e-06,
      "loss": 0.0567,
      "step": 4198
    },
    {
      "epoch": 0.9554038680318544,
      "grad_norm": 1.0136210761213766,
      "learning_rate": 1.1407800560367612e-06,
      "loss": 0.0304,
      "step": 4199
    },
    {
      "epoch": 0.9556313993174061,
      "grad_norm": 0.8478875067932324,
      "learning_rate": 1.140729587782626e-06,
      "loss": 0.0439,
      "step": 4200
    },
    {
      "epoch": 0.9558589306029579,
      "grad_norm": 1.660804148976066,
      "learning_rate": 1.1406791089879229e-06,
      "loss": 0.0731,
      "step": 4201
    },
    {
      "epoch": 0.9560864618885097,
      "grad_norm": 0.8895932107889849,
      "learning_rate": 1.1406286196536832e-06,
      "loss": 0.0363,
      "step": 4202
    },
    {
      "epoch": 0.9563139931740614,
      "grad_norm": 2.248321391825998,
      "learning_rate": 1.1405781197809388e-06,
      "loss": 0.0736,
      "step": 4203
    },
    {
      "epoch": 0.9565415244596132,
      "grad_norm": 0.5970206546121392,
      "learning_rate": 1.1405276093707218e-06,
      "loss": 0.02,
      "step": 4204
    },
    {
      "epoch": 0.9567690557451649,
      "grad_norm": 1.263024166664947,
      "learning_rate": 1.1404770884240645e-06,
      "loss": 0.0523,
      "step": 4205
    },
    {
      "epoch": 0.9569965870307168,
      "grad_norm": 1.3072039796205928,
      "learning_rate": 1.1404265569419998e-06,
      "loss": 0.0606,
      "step": 4206
    },
    {
      "epoch": 0.9572241183162685,
      "grad_norm": 1.2388621360387058,
      "learning_rate": 1.14037601492556e-06,
      "loss": 0.065,
      "step": 4207
    },
    {
      "epoch": 0.9574516496018203,
      "grad_norm": 0.6867586774624046,
      "learning_rate": 1.1403254623757785e-06,
      "loss": 0.0258,
      "step": 4208
    },
    {
      "epoch": 0.957679180887372,
      "grad_norm": 1.296038153739218,
      "learning_rate": 1.1402748992936881e-06,
      "loss": 0.0563,
      "step": 4209
    },
    {
      "epoch": 0.9579067121729238,
      "grad_norm": 1.3170165094768458,
      "learning_rate": 1.1402243256803228e-06,
      "loss": 0.0537,
      "step": 4210
    },
    {
      "epoch": 0.9581342434584755,
      "grad_norm": 2.115381990971279,
      "learning_rate": 1.1401737415367157e-06,
      "loss": 0.1093,
      "step": 4211
    },
    {
      "epoch": 0.9583617747440273,
      "grad_norm": 1.1322388794198333,
      "learning_rate": 1.1401231468639008e-06,
      "loss": 0.0477,
      "step": 4212
    },
    {
      "epoch": 0.958589306029579,
      "grad_norm": 1.7074659597722202,
      "learning_rate": 1.140072541662912e-06,
      "loss": 0.0883,
      "step": 4213
    },
    {
      "epoch": 0.9588168373151308,
      "grad_norm": 1.6179437373612184,
      "learning_rate": 1.1400219259347842e-06,
      "loss": 0.0583,
      "step": 4214
    },
    {
      "epoch": 0.9590443686006825,
      "grad_norm": 1.0176578115657482,
      "learning_rate": 1.139971299680551e-06,
      "loss": 0.0551,
      "step": 4215
    },
    {
      "epoch": 0.9592718998862344,
      "grad_norm": 1.07590619993706,
      "learning_rate": 1.1399206629012478e-06,
      "loss": 0.0367,
      "step": 4216
    },
    {
      "epoch": 0.9594994311717862,
      "grad_norm": 0.9678873702253064,
      "learning_rate": 1.1398700155979092e-06,
      "loss": 0.0394,
      "step": 4217
    },
    {
      "epoch": 0.9597269624573379,
      "grad_norm": 1.1238559057706377,
      "learning_rate": 1.1398193577715705e-06,
      "loss": 0.0623,
      "step": 4218
    },
    {
      "epoch": 0.9599544937428897,
      "grad_norm": 1.2051383992211526,
      "learning_rate": 1.1397686894232671e-06,
      "loss": 0.048,
      "step": 4219
    },
    {
      "epoch": 0.9601820250284414,
      "grad_norm": 1.5391029437375636,
      "learning_rate": 1.1397180105540343e-06,
      "loss": 0.0575,
      "step": 4220
    },
    {
      "epoch": 0.9604095563139932,
      "grad_norm": 1.620882109175618,
      "learning_rate": 1.1396673211649078e-06,
      "loss": 0.0401,
      "step": 4221
    },
    {
      "epoch": 0.9606370875995449,
      "grad_norm": 1.4900621743746256,
      "learning_rate": 1.139616621256924e-06,
      "loss": 0.0488,
      "step": 4222
    },
    {
      "epoch": 0.9608646188850967,
      "grad_norm": 1.2199319551548797,
      "learning_rate": 1.1395659108311192e-06,
      "loss": 0.0568,
      "step": 4223
    },
    {
      "epoch": 0.9610921501706484,
      "grad_norm": 0.7265505005040674,
      "learning_rate": 1.1395151898885293e-06,
      "loss": 0.03,
      "step": 4224
    },
    {
      "epoch": 0.9613196814562003,
      "grad_norm": 1.1651056301933327,
      "learning_rate": 1.1394644584301912e-06,
      "loss": 0.0496,
      "step": 4225
    },
    {
      "epoch": 0.961547212741752,
      "grad_norm": 1.3085058571423405,
      "learning_rate": 1.1394137164571418e-06,
      "loss": 0.0488,
      "step": 4226
    },
    {
      "epoch": 0.9617747440273038,
      "grad_norm": 1.3459918801148398,
      "learning_rate": 1.1393629639704182e-06,
      "loss": 0.0525,
      "step": 4227
    },
    {
      "epoch": 0.9620022753128555,
      "grad_norm": 1.8692658121925165,
      "learning_rate": 1.1393122009710575e-06,
      "loss": 0.0859,
      "step": 4228
    },
    {
      "epoch": 0.9622298065984073,
      "grad_norm": 1.1703507912941524,
      "learning_rate": 1.1392614274600975e-06,
      "loss": 0.048,
      "step": 4229
    },
    {
      "epoch": 0.962457337883959,
      "grad_norm": 1.0308219099107716,
      "learning_rate": 1.1392106434385754e-06,
      "loss": 0.0502,
      "step": 4230
    },
    {
      "epoch": 0.9626848691695108,
      "grad_norm": 1.67972447648695,
      "learning_rate": 1.1391598489075298e-06,
      "loss": 0.0898,
      "step": 4231
    },
    {
      "epoch": 0.9629124004550625,
      "grad_norm": 0.9225444046119701,
      "learning_rate": 1.1391090438679986e-06,
      "loss": 0.0589,
      "step": 4232
    },
    {
      "epoch": 0.9631399317406143,
      "grad_norm": 0.8217198859736241,
      "learning_rate": 1.1390582283210199e-06,
      "loss": 0.0515,
      "step": 4233
    },
    {
      "epoch": 0.963367463026166,
      "grad_norm": 1.4166504015117989,
      "learning_rate": 1.1390074022676325e-06,
      "loss": 0.0434,
      "step": 4234
    },
    {
      "epoch": 0.9635949943117179,
      "grad_norm": 1.8246388670882077,
      "learning_rate": 1.1389565657088752e-06,
      "loss": 0.0696,
      "step": 4235
    },
    {
      "epoch": 0.9638225255972697,
      "grad_norm": 1.984425508646575,
      "learning_rate": 1.1389057186457868e-06,
      "loss": 0.139,
      "step": 4236
    },
    {
      "epoch": 0.9640500568828214,
      "grad_norm": 2.480312140442047,
      "learning_rate": 1.1388548610794069e-06,
      "loss": 0.1062,
      "step": 4237
    },
    {
      "epoch": 0.9642775881683732,
      "grad_norm": 1.0531496575917552,
      "learning_rate": 1.1388039930107747e-06,
      "loss": 0.0416,
      "step": 4238
    },
    {
      "epoch": 0.9645051194539249,
      "grad_norm": 1.4873297621523849,
      "learning_rate": 1.1387531144409297e-06,
      "loss": 0.0418,
      "step": 4239
    },
    {
      "epoch": 0.9647326507394767,
      "grad_norm": 0.9204083159334068,
      "learning_rate": 1.138702225370912e-06,
      "loss": 0.0365,
      "step": 4240
    },
    {
      "epoch": 0.9649601820250284,
      "grad_norm": 1.193250713664879,
      "learning_rate": 1.1386513258017617e-06,
      "loss": 0.057,
      "step": 4241
    },
    {
      "epoch": 0.9651877133105802,
      "grad_norm": 1.1349796974488853,
      "learning_rate": 1.138600415734519e-06,
      "loss": 0.0321,
      "step": 4242
    },
    {
      "epoch": 0.9654152445961319,
      "grad_norm": 1.0770052024267605,
      "learning_rate": 1.1385494951702245e-06,
      "loss": 0.0683,
      "step": 4243
    },
    {
      "epoch": 0.9656427758816837,
      "grad_norm": 1.4502511589216216,
      "learning_rate": 1.1384985641099187e-06,
      "loss": 0.0729,
      "step": 4244
    },
    {
      "epoch": 0.9658703071672355,
      "grad_norm": 1.1356940815876257,
      "learning_rate": 1.1384476225546426e-06,
      "loss": 0.0366,
      "step": 4245
    },
    {
      "epoch": 0.9660978384527873,
      "grad_norm": 0.9898762934279589,
      "learning_rate": 1.1383966705054377e-06,
      "loss": 0.0408,
      "step": 4246
    },
    {
      "epoch": 0.966325369738339,
      "grad_norm": 1.7320554731048066,
      "learning_rate": 1.1383457079633448e-06,
      "loss": 0.0668,
      "step": 4247
    },
    {
      "epoch": 0.9665529010238908,
      "grad_norm": 1.1627771449836428,
      "learning_rate": 1.138294734929406e-06,
      "loss": 0.0565,
      "step": 4248
    },
    {
      "epoch": 0.9667804323094426,
      "grad_norm": 1.4667218418425674,
      "learning_rate": 1.1382437514046627e-06,
      "loss": 0.0572,
      "step": 4249
    },
    {
      "epoch": 0.9670079635949943,
      "grad_norm": 1.3102476422588094,
      "learning_rate": 1.1381927573901572e-06,
      "loss": 0.045,
      "step": 4250
    },
    {
      "epoch": 0.967235494880546,
      "grad_norm": 0.9659550904946845,
      "learning_rate": 1.1381417528869316e-06,
      "loss": 0.0562,
      "step": 4251
    },
    {
      "epoch": 0.9674630261660978,
      "grad_norm": 1.208698720888652,
      "learning_rate": 1.1380907378960282e-06,
      "loss": 0.0601,
      "step": 4252
    },
    {
      "epoch": 0.9676905574516496,
      "grad_norm": 1.3703900042435244,
      "learning_rate": 1.13803971241849e-06,
      "loss": 0.0498,
      "step": 4253
    },
    {
      "epoch": 0.9679180887372013,
      "grad_norm": 1.3965879380443744,
      "learning_rate": 1.1379886764553596e-06,
      "loss": 0.0457,
      "step": 4254
    },
    {
      "epoch": 0.9681456200227532,
      "grad_norm": 1.4360003219255868,
      "learning_rate": 1.1379376300076803e-06,
      "loss": 0.0689,
      "step": 4255
    },
    {
      "epoch": 0.9683731513083049,
      "grad_norm": 1.1977161122633277,
      "learning_rate": 1.137886573076495e-06,
      "loss": 0.0455,
      "step": 4256
    },
    {
      "epoch": 0.9686006825938567,
      "grad_norm": 1.5336831401293278,
      "learning_rate": 1.1378355056628474e-06,
      "loss": 0.0749,
      "step": 4257
    },
    {
      "epoch": 0.9688282138794084,
      "grad_norm": 1.842256713795995,
      "learning_rate": 1.1377844277677815e-06,
      "loss": 0.0702,
      "step": 4258
    },
    {
      "epoch": 0.9690557451649602,
      "grad_norm": 1.3257914742909727,
      "learning_rate": 1.1377333393923408e-06,
      "loss": 0.0427,
      "step": 4259
    },
    {
      "epoch": 0.9692832764505119,
      "grad_norm": 0.9651098248872604,
      "learning_rate": 1.1376822405375698e-06,
      "loss": 0.0309,
      "step": 4260
    },
    {
      "epoch": 0.9695108077360637,
      "grad_norm": 1.406730207103498,
      "learning_rate": 1.1376311312045128e-06,
      "loss": 0.0467,
      "step": 4261
    },
    {
      "epoch": 0.9697383390216154,
      "grad_norm": 1.1340714191157526,
      "learning_rate": 1.1375800113942144e-06,
      "loss": 0.0414,
      "step": 4262
    },
    {
      "epoch": 0.9699658703071672,
      "grad_norm": 1.0587038466584116,
      "learning_rate": 1.137528881107719e-06,
      "loss": 0.0411,
      "step": 4263
    },
    {
      "epoch": 0.970193401592719,
      "grad_norm": 1.375758096041912,
      "learning_rate": 1.137477740346072e-06,
      "loss": 0.0532,
      "step": 4264
    },
    {
      "epoch": 0.9704209328782708,
      "grad_norm": 1.3742948141863867,
      "learning_rate": 1.1374265891103187e-06,
      "loss": 0.0581,
      "step": 4265
    },
    {
      "epoch": 0.9706484641638226,
      "grad_norm": 1.8817386199561381,
      "learning_rate": 1.1373754274015044e-06,
      "loss": 0.0567,
      "step": 4266
    },
    {
      "epoch": 0.9708759954493743,
      "grad_norm": 1.8821493407596268,
      "learning_rate": 1.1373242552206744e-06,
      "loss": 0.0881,
      "step": 4267
    },
    {
      "epoch": 0.9711035267349261,
      "grad_norm": 1.2074529205184104,
      "learning_rate": 1.1372730725688754e-06,
      "loss": 0.0634,
      "step": 4268
    },
    {
      "epoch": 0.9713310580204778,
      "grad_norm": 1.193289676669864,
      "learning_rate": 1.1372218794471527e-06,
      "loss": 0.0435,
      "step": 4269
    },
    {
      "epoch": 0.9715585893060296,
      "grad_norm": 1.119752010069939,
      "learning_rate": 1.1371706758565529e-06,
      "loss": 0.0391,
      "step": 4270
    },
    {
      "epoch": 0.9717861205915813,
      "grad_norm": 1.4265721705104009,
      "learning_rate": 1.1371194617981224e-06,
      "loss": 0.0483,
      "step": 4271
    },
    {
      "epoch": 0.9720136518771331,
      "grad_norm": 2.0794425956744775,
      "learning_rate": 1.137068237272908e-06,
      "loss": 0.0806,
      "step": 4272
    },
    {
      "epoch": 0.9722411831626848,
      "grad_norm": 0.8292125442794063,
      "learning_rate": 1.1370170022819569e-06,
      "loss": 0.0266,
      "step": 4273
    },
    {
      "epoch": 0.9724687144482367,
      "grad_norm": 1.5988253258989058,
      "learning_rate": 1.1369657568263157e-06,
      "loss": 0.064,
      "step": 4274
    },
    {
      "epoch": 0.9726962457337884,
      "grad_norm": 1.4884666998450276,
      "learning_rate": 1.1369145009070323e-06,
      "loss": 0.0595,
      "step": 4275
    },
    {
      "epoch": 0.9729237770193402,
      "grad_norm": 1.0787898499294646,
      "learning_rate": 1.1368632345251538e-06,
      "loss": 0.0449,
      "step": 4276
    },
    {
      "epoch": 0.9731513083048919,
      "grad_norm": 1.2761275024398588,
      "learning_rate": 1.1368119576817283e-06,
      "loss": 0.0473,
      "step": 4277
    },
    {
      "epoch": 0.9733788395904437,
      "grad_norm": 0.9499141582616558,
      "learning_rate": 1.1367606703778037e-06,
      "loss": 0.0376,
      "step": 4278
    },
    {
      "epoch": 0.9736063708759954,
      "grad_norm": 1.0046565168504633,
      "learning_rate": 1.1367093726144283e-06,
      "loss": 0.0432,
      "step": 4279
    },
    {
      "epoch": 0.9738339021615472,
      "grad_norm": 1.0659038125870088,
      "learning_rate": 1.1366580643926506e-06,
      "loss": 0.0404,
      "step": 4280
    },
    {
      "epoch": 0.974061433447099,
      "grad_norm": 1.595011229834194,
      "learning_rate": 1.1366067457135188e-06,
      "loss": 0.0794,
      "step": 4281
    },
    {
      "epoch": 0.9742889647326507,
      "grad_norm": 1.968290906084048,
      "learning_rate": 1.1365554165780823e-06,
      "loss": 0.0673,
      "step": 4282
    },
    {
      "epoch": 0.9745164960182024,
      "grad_norm": 1.511125938151934,
      "learning_rate": 1.13650407698739e-06,
      "loss": 0.0347,
      "step": 4283
    },
    {
      "epoch": 0.9747440273037543,
      "grad_norm": 1.6319247438179008,
      "learning_rate": 1.136452726942491e-06,
      "loss": 0.0582,
      "step": 4284
    },
    {
      "epoch": 0.9749715585893061,
      "grad_norm": 1.3918200009388155,
      "learning_rate": 1.1364013664444351e-06,
      "loss": 0.053,
      "step": 4285
    },
    {
      "epoch": 0.9751990898748578,
      "grad_norm": 1.5863047987952235,
      "learning_rate": 1.1363499954942717e-06,
      "loss": 0.0704,
      "step": 4286
    },
    {
      "epoch": 0.9754266211604096,
      "grad_norm": 0.9811314326155632,
      "learning_rate": 1.1362986140930509e-06,
      "loss": 0.0341,
      "step": 4287
    },
    {
      "epoch": 0.9756541524459613,
      "grad_norm": 1.7299011978403993,
      "learning_rate": 1.1362472222418228e-06,
      "loss": 0.0831,
      "step": 4288
    },
    {
      "epoch": 0.9758816837315131,
      "grad_norm": 1.4033619817926257,
      "learning_rate": 1.1361958199416378e-06,
      "loss": 0.0559,
      "step": 4289
    },
    {
      "epoch": 0.9761092150170648,
      "grad_norm": 1.6430727260435547,
      "learning_rate": 1.1361444071935467e-06,
      "loss": 0.0666,
      "step": 4290
    },
    {
      "epoch": 0.9763367463026166,
      "grad_norm": 1.5397750371074748,
      "learning_rate": 1.1360929839985998e-06,
      "loss": 0.0954,
      "step": 4291
    },
    {
      "epoch": 0.9765642775881683,
      "grad_norm": 1.1112858103130956,
      "learning_rate": 1.1360415503578485e-06,
      "loss": 0.0524,
      "step": 4292
    },
    {
      "epoch": 0.9767918088737202,
      "grad_norm": 2.049966956578,
      "learning_rate": 1.1359901062723437e-06,
      "loss": 0.1009,
      "step": 4293
    },
    {
      "epoch": 0.9770193401592719,
      "grad_norm": 2.123284701337966,
      "learning_rate": 1.1359386517431366e-06,
      "loss": 0.0741,
      "step": 4294
    },
    {
      "epoch": 0.9772468714448237,
      "grad_norm": 1.8057163155909235,
      "learning_rate": 1.1358871867712797e-06,
      "loss": 0.1112,
      "step": 4295
    },
    {
      "epoch": 0.9774744027303754,
      "grad_norm": 0.6815528807119272,
      "learning_rate": 1.1358357113578242e-06,
      "loss": 0.0279,
      "step": 4296
    },
    {
      "epoch": 0.9777019340159272,
      "grad_norm": 1.9569360573215808,
      "learning_rate": 1.1357842255038222e-06,
      "loss": 0.0712,
      "step": 4297
    },
    {
      "epoch": 0.977929465301479,
      "grad_norm": 1.2238824938808284,
      "learning_rate": 1.1357327292103266e-06,
      "loss": 0.068,
      "step": 4298
    },
    {
      "epoch": 0.9781569965870307,
      "grad_norm": 1.3361935047633542,
      "learning_rate": 1.135681222478389e-06,
      "loss": 0.0593,
      "step": 4299
    },
    {
      "epoch": 0.9783845278725825,
      "grad_norm": 0.9204735606624523,
      "learning_rate": 1.1356297053090623e-06,
      "loss": 0.0433,
      "step": 4300
    },
    {
      "epoch": 0.9786120591581342,
      "grad_norm": 1.6545159240714982,
      "learning_rate": 1.1355781777033998e-06,
      "loss": 0.0716,
      "step": 4301
    },
    {
      "epoch": 0.978839590443686,
      "grad_norm": 1.3389750327964336,
      "learning_rate": 1.1355266396624545e-06,
      "loss": 0.0788,
      "step": 4302
    },
    {
      "epoch": 0.9790671217292378,
      "grad_norm": 1.2850002640158098,
      "learning_rate": 1.1354750911872795e-06,
      "loss": 0.0563,
      "step": 4303
    },
    {
      "epoch": 0.9792946530147896,
      "grad_norm": 2.1844096287203585,
      "learning_rate": 1.1354235322789286e-06,
      "loss": 0.1006,
      "step": 4304
    },
    {
      "epoch": 0.9795221843003413,
      "grad_norm": 1.2563539759355846,
      "learning_rate": 1.1353719629384554e-06,
      "loss": 0.0732,
      "step": 4305
    },
    {
      "epoch": 0.9797497155858931,
      "grad_norm": 1.4434315214333129,
      "learning_rate": 1.135320383166914e-06,
      "loss": 0.0704,
      "step": 4306
    },
    {
      "epoch": 0.9799772468714448,
      "grad_norm": 1.5960905043054407,
      "learning_rate": 1.1352687929653586e-06,
      "loss": 0.0837,
      "step": 4307
    },
    {
      "epoch": 0.9802047781569966,
      "grad_norm": 1.4152051878506084,
      "learning_rate": 1.1352171923348438e-06,
      "loss": 0.06,
      "step": 4308
    },
    {
      "epoch": 0.9804323094425483,
      "grad_norm": 1.3288142940732306,
      "learning_rate": 1.1351655812764236e-06,
      "loss": 0.0604,
      "step": 4309
    },
    {
      "epoch": 0.9806598407281001,
      "grad_norm": 0.9762848070268094,
      "learning_rate": 1.1351139597911536e-06,
      "loss": 0.0392,
      "step": 4310
    },
    {
      "epoch": 0.9808873720136518,
      "grad_norm": 2.015722359424784,
      "learning_rate": 1.1350623278800884e-06,
      "loss": 0.0572,
      "step": 4311
    },
    {
      "epoch": 0.9811149032992036,
      "grad_norm": 0.7081344517803854,
      "learning_rate": 1.1350106855442833e-06,
      "loss": 0.0276,
      "step": 4312
    },
    {
      "epoch": 0.9813424345847555,
      "grad_norm": 0.846185742491215,
      "learning_rate": 1.134959032784794e-06,
      "loss": 0.026,
      "step": 4313
    },
    {
      "epoch": 0.9815699658703072,
      "grad_norm": 1.3352329035862636,
      "learning_rate": 1.1349073696026759e-06,
      "loss": 0.057,
      "step": 4314
    },
    {
      "epoch": 0.981797497155859,
      "grad_norm": 0.8695753542918584,
      "learning_rate": 1.1348556959989848e-06,
      "loss": 0.0229,
      "step": 4315
    },
    {
      "epoch": 0.9820250284414107,
      "grad_norm": 1.3890644514305241,
      "learning_rate": 1.1348040119747771e-06,
      "loss": 0.052,
      "step": 4316
    },
    {
      "epoch": 0.9822525597269625,
      "grad_norm": 1.870756610162731,
      "learning_rate": 1.1347523175311092e-06,
      "loss": 0.0653,
      "step": 4317
    },
    {
      "epoch": 0.9824800910125142,
      "grad_norm": 1.4997312982720814,
      "learning_rate": 1.1347006126690377e-06,
      "loss": 0.1015,
      "step": 4318
    },
    {
      "epoch": 0.982707622298066,
      "grad_norm": 1.0165185093361495,
      "learning_rate": 1.1346488973896188e-06,
      "loss": 0.0291,
      "step": 4319
    },
    {
      "epoch": 0.9829351535836177,
      "grad_norm": 1.380478111887299,
      "learning_rate": 1.13459717169391e-06,
      "loss": 0.0632,
      "step": 4320
    },
    {
      "epoch": 0.9831626848691695,
      "grad_norm": 1.641625597159548,
      "learning_rate": 1.1345454355829682e-06,
      "loss": 0.0921,
      "step": 4321
    },
    {
      "epoch": 0.9833902161547212,
      "grad_norm": 1.0317618560602355,
      "learning_rate": 1.1344936890578508e-06,
      "loss": 0.0407,
      "step": 4322
    },
    {
      "epoch": 0.9836177474402731,
      "grad_norm": 1.310711884960193,
      "learning_rate": 1.1344419321196156e-06,
      "loss": 0.0685,
      "step": 4323
    },
    {
      "epoch": 0.9838452787258248,
      "grad_norm": 1.1047706035095923,
      "learning_rate": 1.1343901647693204e-06,
      "loss": 0.0565,
      "step": 4324
    },
    {
      "epoch": 0.9840728100113766,
      "grad_norm": 0.7843402054936461,
      "learning_rate": 1.134338387008023e-06,
      "loss": 0.0304,
      "step": 4325
    },
    {
      "epoch": 0.9843003412969283,
      "grad_norm": 1.1819853377536353,
      "learning_rate": 1.134286598836782e-06,
      "loss": 0.0661,
      "step": 4326
    },
    {
      "epoch": 0.9845278725824801,
      "grad_norm": 1.0132047480735697,
      "learning_rate": 1.1342348002566553e-06,
      "loss": 0.0327,
      "step": 4327
    },
    {
      "epoch": 0.9847554038680318,
      "grad_norm": 0.8226016309372456,
      "learning_rate": 1.1341829912687023e-06,
      "loss": 0.0437,
      "step": 4328
    },
    {
      "epoch": 0.9849829351535836,
      "grad_norm": 1.254492363059247,
      "learning_rate": 1.134131171873981e-06,
      "loss": 0.048,
      "step": 4329
    },
    {
      "epoch": 0.9852104664391353,
      "grad_norm": 1.9410455746016961,
      "learning_rate": 1.1340793420735514e-06,
      "loss": 0.0688,
      "step": 4330
    },
    {
      "epoch": 0.9854379977246871,
      "grad_norm": 1.2149172788419882,
      "learning_rate": 1.1340275018684722e-06,
      "loss": 0.055,
      "step": 4331
    },
    {
      "epoch": 0.985665529010239,
      "grad_norm": 1.135352471755647,
      "learning_rate": 1.133975651259803e-06,
      "loss": 0.0367,
      "step": 4332
    },
    {
      "epoch": 0.9858930602957907,
      "grad_norm": 1.3243570338612756,
      "learning_rate": 1.1339237902486037e-06,
      "loss": 0.0546,
      "step": 4333
    },
    {
      "epoch": 0.9861205915813425,
      "grad_norm": 1.3460820001864684,
      "learning_rate": 1.1338719188359343e-06,
      "loss": 0.0588,
      "step": 4334
    },
    {
      "epoch": 0.9863481228668942,
      "grad_norm": 0.9640088217580686,
      "learning_rate": 1.1338200370228546e-06,
      "loss": 0.0266,
      "step": 4335
    },
    {
      "epoch": 0.986575654152446,
      "grad_norm": 1.0087770932683686,
      "learning_rate": 1.1337681448104254e-06,
      "loss": 0.0549,
      "step": 4336
    },
    {
      "epoch": 0.9868031854379977,
      "grad_norm": 1.0628308289235069,
      "learning_rate": 1.1337162421997072e-06,
      "loss": 0.06,
      "step": 4337
    },
    {
      "epoch": 0.9870307167235495,
      "grad_norm": 0.9968869621613908,
      "learning_rate": 1.1336643291917604e-06,
      "loss": 0.0406,
      "step": 4338
    },
    {
      "epoch": 0.9872582480091012,
      "grad_norm": 1.3414457698222066,
      "learning_rate": 1.1336124057876464e-06,
      "loss": 0.059,
      "step": 4339
    },
    {
      "epoch": 0.987485779294653,
      "grad_norm": 1.1669110242397096,
      "learning_rate": 1.1335604719884264e-06,
      "loss": 0.0534,
      "step": 4340
    },
    {
      "epoch": 0.9877133105802047,
      "grad_norm": 2.0594889831279968,
      "learning_rate": 1.1335085277951616e-06,
      "loss": 0.108,
      "step": 4341
    },
    {
      "epoch": 0.9879408418657566,
      "grad_norm": 1.4063745875081306,
      "learning_rate": 1.1334565732089138e-06,
      "loss": 0.085,
      "step": 4342
    },
    {
      "epoch": 0.9881683731513083,
      "grad_norm": 1.501936982721738,
      "learning_rate": 1.133404608230745e-06,
      "loss": 0.0857,
      "step": 4343
    },
    {
      "epoch": 0.9883959044368601,
      "grad_norm": 1.2396246613907491,
      "learning_rate": 1.1333526328617168e-06,
      "loss": 0.0783,
      "step": 4344
    },
    {
      "epoch": 0.9886234357224118,
      "grad_norm": 1.4534107446508395,
      "learning_rate": 1.133300647102892e-06,
      "loss": 0.0359,
      "step": 4345
    },
    {
      "epoch": 0.9888509670079636,
      "grad_norm": 1.5794373443169556,
      "learning_rate": 1.1332486509553328e-06,
      "loss": 0.0433,
      "step": 4346
    },
    {
      "epoch": 0.9890784982935154,
      "grad_norm": 1.2904275476428004,
      "learning_rate": 1.133196644420102e-06,
      "loss": 0.0554,
      "step": 4347
    },
    {
      "epoch": 0.9893060295790671,
      "grad_norm": 1.1570569489010505,
      "learning_rate": 1.1331446274982625e-06,
      "loss": 0.0472,
      "step": 4348
    },
    {
      "epoch": 0.9895335608646189,
      "grad_norm": 1.145673861183969,
      "learning_rate": 1.1330926001908777e-06,
      "loss": 0.0458,
      "step": 4349
    },
    {
      "epoch": 0.9897610921501706,
      "grad_norm": 0.896667580653844,
      "learning_rate": 1.1330405624990104e-06,
      "loss": 0.0234,
      "step": 4350
    },
    {
      "epoch": 0.9899886234357224,
      "grad_norm": 0.9213529689842777,
      "learning_rate": 1.1329885144237243e-06,
      "loss": 0.0381,
      "step": 4351
    },
    {
      "epoch": 0.9902161547212742,
      "grad_norm": 1.0487384771504538,
      "learning_rate": 1.1329364559660836e-06,
      "loss": 0.0364,
      "step": 4352
    },
    {
      "epoch": 0.990443686006826,
      "grad_norm": 1.4233320390454558,
      "learning_rate": 1.132884387127152e-06,
      "loss": 0.0527,
      "step": 4353
    },
    {
      "epoch": 0.9906712172923777,
      "grad_norm": 1.4228059163710602,
      "learning_rate": 1.1328323079079934e-06,
      "loss": 0.0554,
      "step": 4354
    },
    {
      "epoch": 0.9908987485779295,
      "grad_norm": 1.1059331642608046,
      "learning_rate": 1.1327802183096725e-06,
      "loss": 0.0341,
      "step": 4355
    },
    {
      "epoch": 0.9911262798634812,
      "grad_norm": 0.9827406192573771,
      "learning_rate": 1.1327281183332542e-06,
      "loss": 0.0367,
      "step": 4356
    },
    {
      "epoch": 0.991353811149033,
      "grad_norm": 1.1884213703636224,
      "learning_rate": 1.1326760079798027e-06,
      "loss": 0.0491,
      "step": 4357
    },
    {
      "epoch": 0.9915813424345847,
      "grad_norm": 1.656723752079525,
      "learning_rate": 1.1326238872503837e-06,
      "loss": 0.045,
      "step": 4358
    },
    {
      "epoch": 0.9918088737201365,
      "grad_norm": 1.0844028312793037,
      "learning_rate": 1.1325717561460617e-06,
      "loss": 0.0339,
      "step": 4359
    },
    {
      "epoch": 0.9920364050056882,
      "grad_norm": 1.0891260266689944,
      "learning_rate": 1.132519614667903e-06,
      "loss": 0.0466,
      "step": 4360
    },
    {
      "epoch": 0.9922639362912401,
      "grad_norm": 1.7100161526383912,
      "learning_rate": 1.1324674628169725e-06,
      "loss": 0.0889,
      "step": 4361
    },
    {
      "epoch": 0.9924914675767919,
      "grad_norm": 1.0880829131016376,
      "learning_rate": 1.1324153005943367e-06,
      "loss": 0.0431,
      "step": 4362
    },
    {
      "epoch": 0.9927189988623436,
      "grad_norm": 0.8884823165051863,
      "learning_rate": 1.1323631280010611e-06,
      "loss": 0.0447,
      "step": 4363
    },
    {
      "epoch": 0.9929465301478954,
      "grad_norm": 1.8533973089871183,
      "learning_rate": 1.1323109450382128e-06,
      "loss": 0.082,
      "step": 4364
    },
    {
      "epoch": 0.9931740614334471,
      "grad_norm": 0.9974325526425775,
      "learning_rate": 1.1322587517068576e-06,
      "loss": 0.0319,
      "step": 4365
    },
    {
      "epoch": 0.9934015927189989,
      "grad_norm": 0.9460101960780305,
      "learning_rate": 1.1322065480080625e-06,
      "loss": 0.0433,
      "step": 4366
    },
    {
      "epoch": 0.9936291240045506,
      "grad_norm": 2.537972684614883,
      "learning_rate": 1.1321543339428946e-06,
      "loss": 0.1302,
      "step": 4367
    },
    {
      "epoch": 0.9938566552901024,
      "grad_norm": 0.9361607430120513,
      "learning_rate": 1.132102109512421e-06,
      "loss": 0.0414,
      "step": 4368
    },
    {
      "epoch": 0.9940841865756541,
      "grad_norm": 1.3835872703988714,
      "learning_rate": 1.1320498747177088e-06,
      "loss": 0.0539,
      "step": 4369
    },
    {
      "epoch": 0.9943117178612059,
      "grad_norm": 1.4613596711487815,
      "learning_rate": 1.1319976295598258e-06,
      "loss": 0.0695,
      "step": 4370
    },
    {
      "epoch": 0.9945392491467577,
      "grad_norm": 1.3564131183677717,
      "learning_rate": 1.1319453740398397e-06,
      "loss": 0.0438,
      "step": 4371
    },
    {
      "epoch": 0.9947667804323095,
      "grad_norm": 1.312368604701528,
      "learning_rate": 1.1318931081588188e-06,
      "loss": 0.0539,
      "step": 4372
    },
    {
      "epoch": 0.9949943117178612,
      "grad_norm": 2.852729748963175,
      "learning_rate": 1.1318408319178308e-06,
      "loss": 0.1051,
      "step": 4373
    },
    {
      "epoch": 0.995221843003413,
      "grad_norm": 1.174771913378081,
      "learning_rate": 1.1317885453179448e-06,
      "loss": 0.0345,
      "step": 4374
    },
    {
      "epoch": 0.9954493742889647,
      "grad_norm": 1.156608166877398,
      "learning_rate": 1.131736248360229e-06,
      "loss": 0.0507,
      "step": 4375
    },
    {
      "epoch": 0.9956769055745165,
      "grad_norm": 1.7246435858174831,
      "learning_rate": 1.1316839410457523e-06,
      "loss": 0.0753,
      "step": 4376
    },
    {
      "epoch": 0.9959044368600682,
      "grad_norm": 1.190785748773095,
      "learning_rate": 1.1316316233755837e-06,
      "loss": 0.048,
      "step": 4377
    },
    {
      "epoch": 0.99613196814562,
      "grad_norm": 1.3884121672018674,
      "learning_rate": 1.1315792953507924e-06,
      "loss": 0.054,
      "step": 4378
    },
    {
      "epoch": 0.9963594994311717,
      "grad_norm": 1.1099500717065285,
      "learning_rate": 1.1315269569724483e-06,
      "loss": 0.0578,
      "step": 4379
    },
    {
      "epoch": 0.9965870307167235,
      "grad_norm": 1.1511233150222433,
      "learning_rate": 1.131474608241621e-06,
      "loss": 0.0709,
      "step": 4380
    },
    {
      "epoch": 0.9968145620022754,
      "grad_norm": 0.9955642314650807,
      "learning_rate": 1.1314222491593798e-06,
      "loss": 0.0358,
      "step": 4381
    },
    {
      "epoch": 0.9970420932878271,
      "grad_norm": 1.434440905613479,
      "learning_rate": 1.1313698797267958e-06,
      "loss": 0.0486,
      "step": 4382
    },
    {
      "epoch": 0.9972696245733789,
      "grad_norm": 2.8294165726638822,
      "learning_rate": 1.1313174999449384e-06,
      "loss": 0.0813,
      "step": 4383
    },
    {
      "epoch": 0.9974971558589306,
      "grad_norm": 1.2319487216248912,
      "learning_rate": 1.1312651098148788e-06,
      "loss": 0.0432,
      "step": 4384
    },
    {
      "epoch": 0.9977246871444824,
      "grad_norm": 1.2286306009661918,
      "learning_rate": 1.1312127093376876e-06,
      "loss": 0.0498,
      "step": 4385
    },
    {
      "epoch": 0.9979522184300341,
      "grad_norm": 1.4020060740030407,
      "learning_rate": 1.1311602985144358e-06,
      "loss": 0.0618,
      "step": 4386
    },
    {
      "epoch": 0.9981797497155859,
      "grad_norm": 1.2218921395546485,
      "learning_rate": 1.1311078773461942e-06,
      "loss": 0.0704,
      "step": 4387
    },
    {
      "epoch": 0.9984072810011376,
      "grad_norm": 1.2701236465901293,
      "learning_rate": 1.1310554458340345e-06,
      "loss": 0.0494,
      "step": 4388
    },
    {
      "epoch": 0.9986348122866894,
      "grad_norm": 1.2231395933687852,
      "learning_rate": 1.1310030039790285e-06,
      "loss": 0.0548,
      "step": 4389
    },
    {
      "epoch": 0.9988623435722411,
      "grad_norm": 1.228744395379064,
      "learning_rate": 1.1309505517822476e-06,
      "loss": 0.0456,
      "step": 4390
    },
    {
      "epoch": 0.999089874857793,
      "grad_norm": 1.0175121426182296,
      "learning_rate": 1.1308980892447641e-06,
      "loss": 0.049,
      "step": 4391
    },
    {
      "epoch": 0.9993174061433447,
      "grad_norm": 1.144658935981794,
      "learning_rate": 1.1308456163676501e-06,
      "loss": 0.0487,
      "step": 4392
    },
    {
      "epoch": 0.9995449374288965,
      "grad_norm": 1.1385156907253697,
      "learning_rate": 1.1307931331519783e-06,
      "loss": 0.051,
      "step": 4393
    },
    {
      "epoch": 0.9997724687144482,
      "grad_norm": 1.2941484330489015,
      "learning_rate": 1.1307406395988211e-06,
      "loss": 0.0587,
      "step": 4394
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.7645843200332254,
      "learning_rate": 1.1306881357092513e-06,
      "loss": 0.0273,
      "step": 4395
    },
    {
      "epoch": 1.0002275312855518,
      "grad_norm": 0.5440562375961928,
      "learning_rate": 1.1306356214843423e-06,
      "loss": 0.0133,
      "step": 4396
    },
    {
      "epoch": 1.0004550625711035,
      "grad_norm": 1.048557888593317,
      "learning_rate": 1.1305830969251672e-06,
      "loss": 0.0234,
      "step": 4397
    },
    {
      "epoch": 1.0006825938566553,
      "grad_norm": 0.7566376417333784,
      "learning_rate": 1.1305305620327994e-06,
      "loss": 0.0208,
      "step": 4398
    },
    {
      "epoch": 1.000910125142207,
      "grad_norm": 0.6722933508685324,
      "learning_rate": 1.1304780168083128e-06,
      "loss": 0.0252,
      "step": 4399
    },
    {
      "epoch": 1.0011376564277588,
      "grad_norm": 0.9157174579220916,
      "learning_rate": 1.1304254612527815e-06,
      "loss": 0.0292,
      "step": 4400
    },
    {
      "epoch": 1.0013651877133105,
      "grad_norm": 0.7682748184714413,
      "learning_rate": 1.130372895367279e-06,
      "loss": 0.0305,
      "step": 4401
    },
    {
      "epoch": 1.0015927189988623,
      "grad_norm": 0.4692149259132462,
      "learning_rate": 1.1303203191528803e-06,
      "loss": 0.011,
      "step": 4402
    },
    {
      "epoch": 1.001820250284414,
      "grad_norm": 0.6034838800549486,
      "learning_rate": 1.1302677326106598e-06,
      "loss": 0.0158,
      "step": 4403
    },
    {
      "epoch": 1.0020477815699658,
      "grad_norm": 0.5139829099740163,
      "learning_rate": 1.130215135741692e-06,
      "loss": 0.0097,
      "step": 4404
    },
    {
      "epoch": 1.0022753128555177,
      "grad_norm": 0.6233625169929209,
      "learning_rate": 1.1301625285470522e-06,
      "loss": 0.0209,
      "step": 4405
    },
    {
      "epoch": 1.0025028441410695,
      "grad_norm": 0.8715203529693301,
      "learning_rate": 1.1301099110278156e-06,
      "loss": 0.0349,
      "step": 4406
    },
    {
      "epoch": 1.0027303754266212,
      "grad_norm": 0.5809712252863496,
      "learning_rate": 1.1300572831850574e-06,
      "loss": 0.0173,
      "step": 4407
    },
    {
      "epoch": 1.002957906712173,
      "grad_norm": 0.869776635220801,
      "learning_rate": 1.1300046450198532e-06,
      "loss": 0.0349,
      "step": 4408
    },
    {
      "epoch": 1.0031854379977247,
      "grad_norm": 0.6891595017115912,
      "learning_rate": 1.1299519965332791e-06,
      "loss": 0.0218,
      "step": 4409
    },
    {
      "epoch": 1.0034129692832765,
      "grad_norm": 2.771272397471203,
      "learning_rate": 1.1298993377264108e-06,
      "loss": 0.1173,
      "step": 4410
    },
    {
      "epoch": 1.0036405005688283,
      "grad_norm": 0.6496153724080771,
      "learning_rate": 1.129846668600325e-06,
      "loss": 0.0165,
      "step": 4411
    },
    {
      "epoch": 1.00386803185438,
      "grad_norm": 0.9105625202645007,
      "learning_rate": 1.1297939891560975e-06,
      "loss": 0.0356,
      "step": 4412
    },
    {
      "epoch": 1.0040955631399318,
      "grad_norm": 0.954002149635865,
      "learning_rate": 1.1297412993948054e-06,
      "loss": 0.0419,
      "step": 4413
    },
    {
      "epoch": 1.0043230944254835,
      "grad_norm": 0.8420422197036498,
      "learning_rate": 1.1296885993175255e-06,
      "loss": 0.0249,
      "step": 4414
    },
    {
      "epoch": 1.0045506257110353,
      "grad_norm": 1.1194729404929513,
      "learning_rate": 1.1296358889253351e-06,
      "loss": 0.0282,
      "step": 4415
    },
    {
      "epoch": 1.004778156996587,
      "grad_norm": 0.664863679519903,
      "learning_rate": 1.1295831682193115e-06,
      "loss": 0.0232,
      "step": 4416
    },
    {
      "epoch": 1.0050056882821388,
      "grad_norm": 0.7352288120040293,
      "learning_rate": 1.1295304372005316e-06,
      "loss": 0.0308,
      "step": 4417
    },
    {
      "epoch": 1.0052332195676905,
      "grad_norm": 0.9912305346960562,
      "learning_rate": 1.129477695870074e-06,
      "loss": 0.0454,
      "step": 4418
    },
    {
      "epoch": 1.0054607508532423,
      "grad_norm": 0.5967781019650417,
      "learning_rate": 1.129424944229016e-06,
      "loss": 0.0145,
      "step": 4419
    },
    {
      "epoch": 1.005688282138794,
      "grad_norm": 0.8293958411943283,
      "learning_rate": 1.1293721822784359e-06,
      "loss": 0.0189,
      "step": 4420
    },
    {
      "epoch": 1.0059158134243458,
      "grad_norm": 0.9305770629796571,
      "learning_rate": 1.1293194100194121e-06,
      "loss": 0.0258,
      "step": 4421
    },
    {
      "epoch": 1.0061433447098975,
      "grad_norm": 0.7187164975510966,
      "learning_rate": 1.1292666274530232e-06,
      "loss": 0.0165,
      "step": 4422
    },
    {
      "epoch": 1.0063708759954493,
      "grad_norm": 0.6821469033121376,
      "learning_rate": 1.129213834580348e-06,
      "loss": 0.0147,
      "step": 4423
    },
    {
      "epoch": 1.006598407281001,
      "grad_norm": 0.5603983030125896,
      "learning_rate": 1.1291610314024653e-06,
      "loss": 0.0135,
      "step": 4424
    },
    {
      "epoch": 1.006825938566553,
      "grad_norm": 0.7667715549060277,
      "learning_rate": 1.1291082179204548e-06,
      "loss": 0.0202,
      "step": 4425
    },
    {
      "epoch": 1.0070534698521048,
      "grad_norm": 0.5411620244180794,
      "learning_rate": 1.1290553941353954e-06,
      "loss": 0.0103,
      "step": 4426
    },
    {
      "epoch": 1.0072810011376565,
      "grad_norm": 0.6055971311258457,
      "learning_rate": 1.1290025600483667e-06,
      "loss": 0.0228,
      "step": 4427
    },
    {
      "epoch": 1.0075085324232083,
      "grad_norm": 1.2109757139019808,
      "learning_rate": 1.1289497156604487e-06,
      "loss": 0.0437,
      "step": 4428
    },
    {
      "epoch": 1.00773606370876,
      "grad_norm": 0.5370646714370468,
      "learning_rate": 1.1288968609727216e-06,
      "loss": 0.0085,
      "step": 4429
    },
    {
      "epoch": 1.0079635949943118,
      "grad_norm": 0.9137582688267473,
      "learning_rate": 1.1288439959862654e-06,
      "loss": 0.0349,
      "step": 4430
    },
    {
      "epoch": 1.0081911262798635,
      "grad_norm": 0.6202104523174599,
      "learning_rate": 1.128791120702161e-06,
      "loss": 0.0244,
      "step": 4431
    },
    {
      "epoch": 1.0084186575654153,
      "grad_norm": 1.264397001473755,
      "learning_rate": 1.1287382351214884e-06,
      "loss": 0.0385,
      "step": 4432
    },
    {
      "epoch": 1.008646188850967,
      "grad_norm": 0.4733349727968622,
      "learning_rate": 1.128685339245329e-06,
      "loss": 0.0157,
      "step": 4433
    },
    {
      "epoch": 1.0088737201365188,
      "grad_norm": 0.7488096549319501,
      "learning_rate": 1.1286324330747637e-06,
      "loss": 0.0237,
      "step": 4434
    },
    {
      "epoch": 1.0091012514220705,
      "grad_norm": 0.6215090350542927,
      "learning_rate": 1.1285795166108735e-06,
      "loss": 0.0165,
      "step": 4435
    },
    {
      "epoch": 1.0093287827076223,
      "grad_norm": 0.9977188909479956,
      "learning_rate": 1.1285265898547406e-06,
      "loss": 0.0269,
      "step": 4436
    },
    {
      "epoch": 1.009556313993174,
      "grad_norm": 1.3828220847485446,
      "learning_rate": 1.1284736528074464e-06,
      "loss": 0.0327,
      "step": 4437
    },
    {
      "epoch": 1.0097838452787258,
      "grad_norm": 0.8745376030427436,
      "learning_rate": 1.1284207054700727e-06,
      "loss": 0.0223,
      "step": 4438
    },
    {
      "epoch": 1.0100113765642775,
      "grad_norm": 0.710421250041984,
      "learning_rate": 1.1283677478437016e-06,
      "loss": 0.0206,
      "step": 4439
    },
    {
      "epoch": 1.0102389078498293,
      "grad_norm": 0.9294364308935749,
      "learning_rate": 1.1283147799294158e-06,
      "loss": 0.023,
      "step": 4440
    },
    {
      "epoch": 1.010466439135381,
      "grad_norm": 1.1336723842425673,
      "learning_rate": 1.1282618017282977e-06,
      "loss": 0.0543,
      "step": 4441
    },
    {
      "epoch": 1.0106939704209328,
      "grad_norm": 0.6236066585507253,
      "learning_rate": 1.1282088132414297e-06,
      "loss": 0.0177,
      "step": 4442
    },
    {
      "epoch": 1.0109215017064845,
      "grad_norm": 0.4538882994044952,
      "learning_rate": 1.1281558144698956e-06,
      "loss": 0.005,
      "step": 4443
    },
    {
      "epoch": 1.0111490329920365,
      "grad_norm": 0.9922582222331733,
      "learning_rate": 1.128102805414778e-06,
      "loss": 0.024,
      "step": 4444
    },
    {
      "epoch": 1.0113765642775883,
      "grad_norm": 0.939282752346539,
      "learning_rate": 1.1280497860771603e-06,
      "loss": 0.0373,
      "step": 4445
    },
    {
      "epoch": 1.01160409556314,
      "grad_norm": 0.6323505737761085,
      "learning_rate": 1.1279967564581264e-06,
      "loss": 0.0121,
      "step": 4446
    },
    {
      "epoch": 1.0118316268486918,
      "grad_norm": 1.0956673474878567,
      "learning_rate": 1.12794371655876e-06,
      "loss": 0.02,
      "step": 4447
    },
    {
      "epoch": 1.0120591581342435,
      "grad_norm": 0.9129627353008005,
      "learning_rate": 1.127890666380145e-06,
      "loss": 0.025,
      "step": 4448
    },
    {
      "epoch": 1.0122866894197953,
      "grad_norm": 0.46435154090423725,
      "learning_rate": 1.1278376059233658e-06,
      "loss": 0.0083,
      "step": 4449
    },
    {
      "epoch": 1.012514220705347,
      "grad_norm": 0.823583660478901,
      "learning_rate": 1.127784535189507e-06,
      "loss": 0.0242,
      "step": 4450
    },
    {
      "epoch": 1.0127417519908988,
      "grad_norm": 1.244611532738964,
      "learning_rate": 1.127731454179653e-06,
      "loss": 0.0477,
      "step": 4451
    },
    {
      "epoch": 1.0129692832764505,
      "grad_norm": 0.6536691853613382,
      "learning_rate": 1.1276783628948887e-06,
      "loss": 0.0203,
      "step": 4452
    },
    {
      "epoch": 1.0131968145620023,
      "grad_norm": 0.7200287149249455,
      "learning_rate": 1.1276252613362995e-06,
      "loss": 0.0238,
      "step": 4453
    },
    {
      "epoch": 1.013424345847554,
      "grad_norm": 0.8521168883159115,
      "learning_rate": 1.1275721495049702e-06,
      "loss": 0.0308,
      "step": 4454
    },
    {
      "epoch": 1.0136518771331058,
      "grad_norm": 0.6812814467705595,
      "learning_rate": 1.1275190274019867e-06,
      "loss": 0.0223,
      "step": 4455
    },
    {
      "epoch": 1.0138794084186575,
      "grad_norm": 0.8073573367282825,
      "learning_rate": 1.1274658950284347e-06,
      "loss": 0.0159,
      "step": 4456
    },
    {
      "epoch": 1.0141069397042093,
      "grad_norm": 0.9929398468232915,
      "learning_rate": 1.1274127523854e-06,
      "loss": 0.0174,
      "step": 4457
    },
    {
      "epoch": 1.014334470989761,
      "grad_norm": 0.8110259409039485,
      "learning_rate": 1.1273595994739688e-06,
      "loss": 0.0231,
      "step": 4458
    },
    {
      "epoch": 1.0145620022753128,
      "grad_norm": 0.8503939357738989,
      "learning_rate": 1.1273064362952272e-06,
      "loss": 0.0181,
      "step": 4459
    },
    {
      "epoch": 1.0147895335608645,
      "grad_norm": 0.952005853323446,
      "learning_rate": 1.1272532628502621e-06,
      "loss": 0.011,
      "step": 4460
    },
    {
      "epoch": 1.0150170648464163,
      "grad_norm": 0.979632492475044,
      "learning_rate": 1.1272000791401602e-06,
      "loss": 0.0289,
      "step": 4461
    },
    {
      "epoch": 1.015244596131968,
      "grad_norm": 0.9141108737856498,
      "learning_rate": 1.1271468851660084e-06,
      "loss": 0.0325,
      "step": 4462
    },
    {
      "epoch": 1.01547212741752,
      "grad_norm": 0.7032331801339051,
      "learning_rate": 1.127093680928894e-06,
      "loss": 0.0188,
      "step": 4463
    },
    {
      "epoch": 1.0156996587030718,
      "grad_norm": 0.9755139681497196,
      "learning_rate": 1.1270404664299042e-06,
      "loss": 0.0212,
      "step": 4464
    },
    {
      "epoch": 1.0159271899886235,
      "grad_norm": 0.623696536501648,
      "learning_rate": 1.1269872416701267e-06,
      "loss": 0.0143,
      "step": 4465
    },
    {
      "epoch": 1.0161547212741753,
      "grad_norm": 0.5413765005494656,
      "learning_rate": 1.1269340066506493e-06,
      "loss": 0.0101,
      "step": 4466
    },
    {
      "epoch": 1.016382252559727,
      "grad_norm": 0.6689689779774286,
      "learning_rate": 1.12688076137256e-06,
      "loss": 0.0114,
      "step": 4467
    },
    {
      "epoch": 1.0166097838452788,
      "grad_norm": 0.6453292072094783,
      "learning_rate": 1.1268275058369472e-06,
      "loss": 0.0107,
      "step": 4468
    },
    {
      "epoch": 1.0168373151308305,
      "grad_norm": 1.1332110733517493,
      "learning_rate": 1.1267742400448992e-06,
      "loss": 0.0217,
      "step": 4469
    },
    {
      "epoch": 1.0170648464163823,
      "grad_norm": 0.630093494299339,
      "learning_rate": 1.1267209639975046e-06,
      "loss": 0.0089,
      "step": 4470
    },
    {
      "epoch": 1.017292377701934,
      "grad_norm": 1.0118536500106963,
      "learning_rate": 1.1266676776958523e-06,
      "loss": 0.0208,
      "step": 4471
    },
    {
      "epoch": 1.0175199089874858,
      "grad_norm": 1.0518953123729393,
      "learning_rate": 1.1266143811410317e-06,
      "loss": 0.0201,
      "step": 4472
    },
    {
      "epoch": 1.0177474402730375,
      "grad_norm": 0.757424135828587,
      "learning_rate": 1.1265610743341316e-06,
      "loss": 0.0217,
      "step": 4473
    },
    {
      "epoch": 1.0179749715585893,
      "grad_norm": 0.77124606602971,
      "learning_rate": 1.1265077572762418e-06,
      "loss": 0.0262,
      "step": 4474
    },
    {
      "epoch": 1.018202502844141,
      "grad_norm": 1.7312334858326726,
      "learning_rate": 1.1264544299684518e-06,
      "loss": 0.0511,
      "step": 4475
    },
    {
      "epoch": 1.0184300341296928,
      "grad_norm": 0.8077840361458416,
      "learning_rate": 1.1264010924118518e-06,
      "loss": 0.0156,
      "step": 4476
    },
    {
      "epoch": 1.0186575654152445,
      "grad_norm": 0.9380910246867491,
      "learning_rate": 1.1263477446075315e-06,
      "loss": 0.0387,
      "step": 4477
    },
    {
      "epoch": 1.0188850967007963,
      "grad_norm": 0.7652818963146196,
      "learning_rate": 1.1262943865565818e-06,
      "loss": 0.0173,
      "step": 4478
    },
    {
      "epoch": 1.019112627986348,
      "grad_norm": 1.3337882989968501,
      "learning_rate": 1.1262410182600927e-06,
      "loss": 0.0396,
      "step": 4479
    },
    {
      "epoch": 1.0193401592718998,
      "grad_norm": 0.45812987319685416,
      "learning_rate": 1.1261876397191554e-06,
      "loss": 0.015,
      "step": 4480
    },
    {
      "epoch": 1.0195676905574516,
      "grad_norm": 0.7602276965168047,
      "learning_rate": 1.1261342509348604e-06,
      "loss": 0.0202,
      "step": 4481
    },
    {
      "epoch": 1.0197952218430033,
      "grad_norm": 0.9601272476880518,
      "learning_rate": 1.126080851908299e-06,
      "loss": 0.0245,
      "step": 4482
    },
    {
      "epoch": 1.0200227531285553,
      "grad_norm": 0.9302450496938718,
      "learning_rate": 1.1260274426405629e-06,
      "loss": 0.0251,
      "step": 4483
    },
    {
      "epoch": 1.020250284414107,
      "grad_norm": 1.064523523991496,
      "learning_rate": 1.1259740231327434e-06,
      "loss": 0.0295,
      "step": 4484
    },
    {
      "epoch": 1.0204778156996588,
      "grad_norm": 1.4809063729774259,
      "learning_rate": 1.1259205933859325e-06,
      "loss": 0.0268,
      "step": 4485
    },
    {
      "epoch": 1.0207053469852105,
      "grad_norm": 1.2332212440546455,
      "learning_rate": 1.1258671534012216e-06,
      "loss": 0.0432,
      "step": 4486
    },
    {
      "epoch": 1.0209328782707623,
      "grad_norm": 0.7799860888625535,
      "learning_rate": 1.1258137031797037e-06,
      "loss": 0.0124,
      "step": 4487
    },
    {
      "epoch": 1.021160409556314,
      "grad_norm": 0.759267946121651,
      "learning_rate": 1.125760242722471e-06,
      "loss": 0.0091,
      "step": 4488
    },
    {
      "epoch": 1.0213879408418658,
      "grad_norm": 0.891169288145688,
      "learning_rate": 1.1257067720306159e-06,
      "loss": 0.0286,
      "step": 4489
    },
    {
      "epoch": 1.0216154721274175,
      "grad_norm": 1.0432400031887874,
      "learning_rate": 1.1256532911052313e-06,
      "loss": 0.0334,
      "step": 4490
    },
    {
      "epoch": 1.0218430034129693,
      "grad_norm": 1.0537890506909147,
      "learning_rate": 1.1255997999474105e-06,
      "loss": 0.02,
      "step": 4491
    },
    {
      "epoch": 1.022070534698521,
      "grad_norm": 0.7817598770936856,
      "learning_rate": 1.1255462985582465e-06,
      "loss": 0.017,
      "step": 4492
    },
    {
      "epoch": 1.0222980659840728,
      "grad_norm": 0.9383161200612665,
      "learning_rate": 1.125492786938833e-06,
      "loss": 0.011,
      "step": 4493
    },
    {
      "epoch": 1.0225255972696246,
      "grad_norm": 1.7998282210253833,
      "learning_rate": 1.1254392650902633e-06,
      "loss": 0.0469,
      "step": 4494
    },
    {
      "epoch": 1.0227531285551763,
      "grad_norm": 0.9796354836057241,
      "learning_rate": 1.1253857330136316e-06,
      "loss": 0.0252,
      "step": 4495
    },
    {
      "epoch": 1.022980659840728,
      "grad_norm": 0.8615898866534331,
      "learning_rate": 1.125332190710032e-06,
      "loss": 0.0103,
      "step": 4496
    },
    {
      "epoch": 1.0232081911262798,
      "grad_norm": 1.066758301808706,
      "learning_rate": 1.125278638180559e-06,
      "loss": 0.0176,
      "step": 4497
    },
    {
      "epoch": 1.0234357224118316,
      "grad_norm": 0.7537124723757486,
      "learning_rate": 1.1252250754263064e-06,
      "loss": 0.0243,
      "step": 4498
    },
    {
      "epoch": 1.0236632536973833,
      "grad_norm": 0.7310817973699005,
      "learning_rate": 1.1251715024483695e-06,
      "loss": 0.0213,
      "step": 4499
    },
    {
      "epoch": 1.023890784982935,
      "grad_norm": 0.8578238852047406,
      "learning_rate": 1.125117919247843e-06,
      "loss": 0.0319,
      "step": 4500
    },
    {
      "epoch": 1.0241183162684868,
      "grad_norm": 0.9913368014981602,
      "learning_rate": 1.1250643258258225e-06,
      "loss": 0.0208,
      "step": 4501
    },
    {
      "epoch": 1.0243458475540388,
      "grad_norm": 1.2683092562362872,
      "learning_rate": 1.1250107221834027e-06,
      "loss": 0.0504,
      "step": 4502
    },
    {
      "epoch": 1.0245733788395905,
      "grad_norm": 0.7751769469902451,
      "learning_rate": 1.12495710832168e-06,
      "loss": 0.0278,
      "step": 4503
    },
    {
      "epoch": 1.0248009101251423,
      "grad_norm": 0.7428863858706001,
      "learning_rate": 1.1249034842417489e-06,
      "loss": 0.0141,
      "step": 4504
    },
    {
      "epoch": 1.025028441410694,
      "grad_norm": 0.7095783022904831,
      "learning_rate": 1.1248498499447065e-06,
      "loss": 0.019,
      "step": 4505
    },
    {
      "epoch": 1.0252559726962458,
      "grad_norm": 1.0630325639796825,
      "learning_rate": 1.1247962054316485e-06,
      "loss": 0.0439,
      "step": 4506
    },
    {
      "epoch": 1.0254835039817976,
      "grad_norm": 0.8040239643121114,
      "learning_rate": 1.1247425507036715e-06,
      "loss": 0.0189,
      "step": 4507
    },
    {
      "epoch": 1.0257110352673493,
      "grad_norm": 1.8105078367140066,
      "learning_rate": 1.1246888857618719e-06,
      "loss": 0.0633,
      "step": 4508
    },
    {
      "epoch": 1.025938566552901,
      "grad_norm": 1.3028583651888686,
      "learning_rate": 1.1246352106073466e-06,
      "loss": 0.0233,
      "step": 4509
    },
    {
      "epoch": 1.0261660978384528,
      "grad_norm": 1.3773748170809028,
      "learning_rate": 1.1245815252411928e-06,
      "loss": 0.0563,
      "step": 4510
    },
    {
      "epoch": 1.0263936291240046,
      "grad_norm": 1.362685911234451,
      "learning_rate": 1.1245278296645073e-06,
      "loss": 0.0257,
      "step": 4511
    },
    {
      "epoch": 1.0266211604095563,
      "grad_norm": 0.7561719123407112,
      "learning_rate": 1.124474123878388e-06,
      "loss": 0.0137,
      "step": 4512
    },
    {
      "epoch": 1.026848691695108,
      "grad_norm": 0.6802103464690515,
      "learning_rate": 1.1244204078839325e-06,
      "loss": 0.0148,
      "step": 4513
    },
    {
      "epoch": 1.0270762229806598,
      "grad_norm": 1.023770837391378,
      "learning_rate": 1.1243666816822382e-06,
      "loss": 0.0383,
      "step": 4514
    },
    {
      "epoch": 1.0273037542662116,
      "grad_norm": 1.105198417368013,
      "learning_rate": 1.1243129452744036e-06,
      "loss": 0.0169,
      "step": 4515
    },
    {
      "epoch": 1.0275312855517633,
      "grad_norm": 0.8424066208751216,
      "learning_rate": 1.1242591986615268e-06,
      "loss": 0.0308,
      "step": 4516
    },
    {
      "epoch": 1.027758816837315,
      "grad_norm": 0.666774886288571,
      "learning_rate": 1.1242054418447063e-06,
      "loss": 0.0093,
      "step": 4517
    },
    {
      "epoch": 1.0279863481228668,
      "grad_norm": 0.6569322003852172,
      "learning_rate": 1.1241516748250408e-06,
      "loss": 0.0118,
      "step": 4518
    },
    {
      "epoch": 1.0282138794084186,
      "grad_norm": 1.2076404281398103,
      "learning_rate": 1.1240978976036294e-06,
      "loss": 0.0489,
      "step": 4519
    },
    {
      "epoch": 1.0284414106939703,
      "grad_norm": 1.1551438155853788,
      "learning_rate": 1.124044110181571e-06,
      "loss": 0.0311,
      "step": 4520
    },
    {
      "epoch": 1.028668941979522,
      "grad_norm": 1.389595859599812,
      "learning_rate": 1.1239903125599648e-06,
      "loss": 0.0491,
      "step": 4521
    },
    {
      "epoch": 1.028896473265074,
      "grad_norm": 1.0333918284834915,
      "learning_rate": 1.1239365047399106e-06,
      "loss": 0.0404,
      "step": 4522
    },
    {
      "epoch": 1.0291240045506258,
      "grad_norm": 0.8599318194273396,
      "learning_rate": 1.1238826867225077e-06,
      "loss": 0.0197,
      "step": 4523
    },
    {
      "epoch": 1.0293515358361776,
      "grad_norm": 0.8304017967466861,
      "learning_rate": 1.1238288585088567e-06,
      "loss": 0.0171,
      "step": 4524
    },
    {
      "epoch": 1.0295790671217293,
      "grad_norm": 1.136424973741861,
      "learning_rate": 1.1237750201000574e-06,
      "loss": 0.0316,
      "step": 4525
    },
    {
      "epoch": 1.029806598407281,
      "grad_norm": 1.4555472105148153,
      "learning_rate": 1.1237211714972098e-06,
      "loss": 0.062,
      "step": 4526
    },
    {
      "epoch": 1.0300341296928328,
      "grad_norm": 1.3336128205554842,
      "learning_rate": 1.1236673127014152e-06,
      "loss": 0.0243,
      "step": 4527
    },
    {
      "epoch": 1.0302616609783846,
      "grad_norm": 0.9479976161551643,
      "learning_rate": 1.1236134437137738e-06,
      "loss": 0.022,
      "step": 4528
    },
    {
      "epoch": 1.0304891922639363,
      "grad_norm": 0.851610494023467,
      "learning_rate": 1.1235595645353869e-06,
      "loss": 0.0304,
      "step": 4529
    },
    {
      "epoch": 1.030716723549488,
      "grad_norm": 0.8155261530618487,
      "learning_rate": 1.1235056751673554e-06,
      "loss": 0.017,
      "step": 4530
    },
    {
      "epoch": 1.0309442548350398,
      "grad_norm": 1.1243416422484538,
      "learning_rate": 1.123451775610781e-06,
      "loss": 0.0248,
      "step": 4531
    },
    {
      "epoch": 1.0311717861205916,
      "grad_norm": 0.6551716981893974,
      "learning_rate": 1.1233978658667651e-06,
      "loss": 0.0251,
      "step": 4532
    },
    {
      "epoch": 1.0313993174061433,
      "grad_norm": 0.9437616699526395,
      "learning_rate": 1.1233439459364097e-06,
      "loss": 0.0228,
      "step": 4533
    },
    {
      "epoch": 1.031626848691695,
      "grad_norm": 1.245047538582983,
      "learning_rate": 1.1232900158208166e-06,
      "loss": 0.0148,
      "step": 4534
    },
    {
      "epoch": 1.0318543799772468,
      "grad_norm": 0.9658390993690719,
      "learning_rate": 1.1232360755210883e-06,
      "loss": 0.0166,
      "step": 4535
    },
    {
      "epoch": 1.0320819112627986,
      "grad_norm": 1.1047045638558348,
      "learning_rate": 1.123182125038327e-06,
      "loss": 0.0316,
      "step": 4536
    },
    {
      "epoch": 1.0323094425483503,
      "grad_norm": 1.3208237077699243,
      "learning_rate": 1.1231281643736353e-06,
      "loss": 0.0303,
      "step": 4537
    },
    {
      "epoch": 1.032536973833902,
      "grad_norm": 0.7809775193316096,
      "learning_rate": 1.1230741935281163e-06,
      "loss": 0.0196,
      "step": 4538
    },
    {
      "epoch": 1.0327645051194538,
      "grad_norm": 0.991878604252392,
      "learning_rate": 1.123020212502873e-06,
      "loss": 0.02,
      "step": 4539
    },
    {
      "epoch": 1.0329920364050056,
      "grad_norm": 1.3869865701838429,
      "learning_rate": 1.1229662212990088e-06,
      "loss": 0.0612,
      "step": 4540
    },
    {
      "epoch": 1.0332195676905576,
      "grad_norm": 0.619858198046472,
      "learning_rate": 1.1229122199176268e-06,
      "loss": 0.0072,
      "step": 4541
    },
    {
      "epoch": 1.0334470989761093,
      "grad_norm": 0.7040478929448408,
      "learning_rate": 1.1228582083598311e-06,
      "loss": 0.0257,
      "step": 4542
    },
    {
      "epoch": 1.033674630261661,
      "grad_norm": 1.182074276350949,
      "learning_rate": 1.122804186626725e-06,
      "loss": 0.0206,
      "step": 4543
    },
    {
      "epoch": 1.0339021615472128,
      "grad_norm": 0.42845477644473695,
      "learning_rate": 1.1227501547194133e-06,
      "loss": 0.0111,
      "step": 4544
    },
    {
      "epoch": 1.0341296928327646,
      "grad_norm": 1.5120577926077867,
      "learning_rate": 1.1226961126390001e-06,
      "loss": 0.0402,
      "step": 4545
    },
    {
      "epoch": 1.0343572241183163,
      "grad_norm": 1.0258672333609278,
      "learning_rate": 1.1226420603865898e-06,
      "loss": 0.0266,
      "step": 4546
    },
    {
      "epoch": 1.034584755403868,
      "grad_norm": 0.6935436273674827,
      "learning_rate": 1.122587997963287e-06,
      "loss": 0.021,
      "step": 4547
    },
    {
      "epoch": 1.0348122866894198,
      "grad_norm": 2.2382185926621383,
      "learning_rate": 1.122533925370197e-06,
      "loss": 0.0311,
      "step": 4548
    },
    {
      "epoch": 1.0350398179749716,
      "grad_norm": 0.8597189295835992,
      "learning_rate": 1.1224798426084246e-06,
      "loss": 0.0251,
      "step": 4549
    },
    {
      "epoch": 1.0352673492605233,
      "grad_norm": 1.2187642427431231,
      "learning_rate": 1.1224257496790756e-06,
      "loss": 0.0307,
      "step": 4550
    },
    {
      "epoch": 1.035494880546075,
      "grad_norm": 0.8235532448084206,
      "learning_rate": 1.122371646583255e-06,
      "loss": 0.016,
      "step": 4551
    },
    {
      "epoch": 1.0357224118316268,
      "grad_norm": 1.1637372096776504,
      "learning_rate": 1.1223175333220688e-06,
      "loss": 0.0249,
      "step": 4552
    },
    {
      "epoch": 1.0359499431171786,
      "grad_norm": 1.2013323108837044,
      "learning_rate": 1.122263409896623e-06,
      "loss": 0.0211,
      "step": 4553
    },
    {
      "epoch": 1.0361774744027303,
      "grad_norm": 1.5241558913138389,
      "learning_rate": 1.1222092763080242e-06,
      "loss": 0.0614,
      "step": 4554
    },
    {
      "epoch": 1.036405005688282,
      "grad_norm": 0.9609458503216893,
      "learning_rate": 1.1221551325573779e-06,
      "loss": 0.0345,
      "step": 4555
    },
    {
      "epoch": 1.0366325369738338,
      "grad_norm": 1.067326425845944,
      "learning_rate": 1.1221009786457914e-06,
      "loss": 0.0251,
      "step": 4556
    },
    {
      "epoch": 1.0368600682593856,
      "grad_norm": 0.6952960225267327,
      "learning_rate": 1.1220468145743713e-06,
      "loss": 0.0158,
      "step": 4557
    },
    {
      "epoch": 1.0370875995449373,
      "grad_norm": 1.5415742411379196,
      "learning_rate": 1.1219926403442247e-06,
      "loss": 0.0398,
      "step": 4558
    },
    {
      "epoch": 1.037315130830489,
      "grad_norm": 1.0675139745280051,
      "learning_rate": 1.1219384559564587e-06,
      "loss": 0.0277,
      "step": 4559
    },
    {
      "epoch": 1.0375426621160408,
      "grad_norm": 0.7777707864923777,
      "learning_rate": 1.1218842614121806e-06,
      "loss": 0.0247,
      "step": 4560
    },
    {
      "epoch": 1.0377701934015928,
      "grad_norm": 1.6672178552337855,
      "learning_rate": 1.1218300567124983e-06,
      "loss": 0.0342,
      "step": 4561
    },
    {
      "epoch": 1.0379977246871446,
      "grad_norm": 1.0152141504441292,
      "learning_rate": 1.1217758418585195e-06,
      "loss": 0.0426,
      "step": 4562
    },
    {
      "epoch": 1.0382252559726963,
      "grad_norm": 1.0296746623158612,
      "learning_rate": 1.1217216168513522e-06,
      "loss": 0.0225,
      "step": 4563
    },
    {
      "epoch": 1.038452787258248,
      "grad_norm": 1.970275220352162,
      "learning_rate": 1.1216673816921048e-06,
      "loss": 0.0876,
      "step": 4564
    },
    {
      "epoch": 1.0386803185437998,
      "grad_norm": 1.1493211366430227,
      "learning_rate": 1.1216131363818859e-06,
      "loss": 0.0191,
      "step": 4565
    },
    {
      "epoch": 1.0389078498293516,
      "grad_norm": 0.9951641567260908,
      "learning_rate": 1.1215588809218038e-06,
      "loss": 0.013,
      "step": 4566
    },
    {
      "epoch": 1.0391353811149033,
      "grad_norm": 0.8580754292791186,
      "learning_rate": 1.1215046153129678e-06,
      "loss": 0.0183,
      "step": 4567
    },
    {
      "epoch": 1.039362912400455,
      "grad_norm": 1.4552256935681152,
      "learning_rate": 1.1214503395564866e-06,
      "loss": 0.0245,
      "step": 4568
    },
    {
      "epoch": 1.0395904436860068,
      "grad_norm": 0.6094461237074116,
      "learning_rate": 1.1213960536534698e-06,
      "loss": 0.0088,
      "step": 4569
    },
    {
      "epoch": 1.0398179749715586,
      "grad_norm": 1.6872520162524116,
      "learning_rate": 1.1213417576050267e-06,
      "loss": 0.0465,
      "step": 4570
    },
    {
      "epoch": 1.0400455062571103,
      "grad_norm": 0.8147199481078186,
      "learning_rate": 1.1212874514122669e-06,
      "loss": 0.0188,
      "step": 4571
    },
    {
      "epoch": 1.040273037542662,
      "grad_norm": 1.6710841073159661,
      "learning_rate": 1.1212331350763007e-06,
      "loss": 0.0183,
      "step": 4572
    },
    {
      "epoch": 1.0405005688282138,
      "grad_norm": 0.7521582291067465,
      "learning_rate": 1.1211788085982381e-06,
      "loss": 0.0141,
      "step": 4573
    },
    {
      "epoch": 1.0407281001137656,
      "grad_norm": 1.3851917572634034,
      "learning_rate": 1.1211244719791892e-06,
      "loss": 0.0306,
      "step": 4574
    },
    {
      "epoch": 1.0409556313993173,
      "grad_norm": 0.5793785303293703,
      "learning_rate": 1.1210701252202647e-06,
      "loss": 0.0106,
      "step": 4575
    },
    {
      "epoch": 1.041183162684869,
      "grad_norm": 1.0136644099615377,
      "learning_rate": 1.1210157683225753e-06,
      "loss": 0.0165,
      "step": 4576
    },
    {
      "epoch": 1.0414106939704209,
      "grad_norm": 0.45591615471037517,
      "learning_rate": 1.1209614012872323e-06,
      "loss": 0.0084,
      "step": 4577
    },
    {
      "epoch": 1.0416382252559726,
      "grad_norm": 1.3441880570081561,
      "learning_rate": 1.1209070241153462e-06,
      "loss": 0.031,
      "step": 4578
    },
    {
      "epoch": 1.0418657565415244,
      "grad_norm": 1.1464751988060768,
      "learning_rate": 1.1208526368080288e-06,
      "loss": 0.0332,
      "step": 4579
    },
    {
      "epoch": 1.0420932878270763,
      "grad_norm": 0.7884324251000338,
      "learning_rate": 1.120798239366392e-06,
      "loss": 0.0168,
      "step": 4580
    },
    {
      "epoch": 1.042320819112628,
      "grad_norm": 2.046969798862839,
      "learning_rate": 1.1207438317915468e-06,
      "loss": 0.0229,
      "step": 4581
    },
    {
      "epoch": 1.0425483503981798,
      "grad_norm": 0.6622639696340208,
      "learning_rate": 1.1206894140846055e-06,
      "loss": 0.0087,
      "step": 4582
    },
    {
      "epoch": 1.0427758816837316,
      "grad_norm": 0.7403190930069853,
      "learning_rate": 1.1206349862466807e-06,
      "loss": 0.015,
      "step": 4583
    },
    {
      "epoch": 1.0430034129692833,
      "grad_norm": 2.3626414035054175,
      "learning_rate": 1.1205805482788846e-06,
      "loss": 0.0417,
      "step": 4584
    },
    {
      "epoch": 1.043230944254835,
      "grad_norm": 2.471483368904896,
      "learning_rate": 1.1205261001823293e-06,
      "loss": 0.044,
      "step": 4585
    },
    {
      "epoch": 1.0434584755403868,
      "grad_norm": 2.416437888927223,
      "learning_rate": 1.1204716419581281e-06,
      "loss": 0.0098,
      "step": 4586
    },
    {
      "epoch": 1.0436860068259386,
      "grad_norm": 1.467076361857702,
      "learning_rate": 1.1204171736073942e-06,
      "loss": 0.0134,
      "step": 4587
    },
    {
      "epoch": 1.0439135381114903,
      "grad_norm": 1.1268621078307177,
      "learning_rate": 1.1203626951312405e-06,
      "loss": 0.0349,
      "step": 4588
    },
    {
      "epoch": 1.044141069397042,
      "grad_norm": 0.7695063155900477,
      "learning_rate": 1.1203082065307805e-06,
      "loss": 0.012,
      "step": 4589
    },
    {
      "epoch": 1.0443686006825939,
      "grad_norm": 1.446038392903303,
      "learning_rate": 1.1202537078071277e-06,
      "loss": 0.0547,
      "step": 4590
    },
    {
      "epoch": 1.0445961319681456,
      "grad_norm": 0.8969845928910839,
      "learning_rate": 1.1201991989613963e-06,
      "loss": 0.0111,
      "step": 4591
    },
    {
      "epoch": 1.0448236632536974,
      "grad_norm": 0.8115056392147382,
      "learning_rate": 1.1201446799947003e-06,
      "loss": 0.0092,
      "step": 4592
    },
    {
      "epoch": 1.045051194539249,
      "grad_norm": 1.2473917829189507,
      "learning_rate": 1.1200901509081537e-06,
      "loss": 0.0137,
      "step": 4593
    },
    {
      "epoch": 1.0452787258248009,
      "grad_norm": 1.0896608769495246,
      "learning_rate": 1.120035611702871e-06,
      "loss": 0.0209,
      "step": 4594
    },
    {
      "epoch": 1.0455062571103526,
      "grad_norm": 0.7977339624737255,
      "learning_rate": 1.1199810623799673e-06,
      "loss": 0.0176,
      "step": 4595
    },
    {
      "epoch": 1.0457337883959044,
      "grad_norm": 1.1634422015699433,
      "learning_rate": 1.119926502940557e-06,
      "loss": 0.0188,
      "step": 4596
    },
    {
      "epoch": 1.0459613196814561,
      "grad_norm": 1.092858194939433,
      "learning_rate": 1.1198719333857555e-06,
      "loss": 0.0312,
      "step": 4597
    },
    {
      "epoch": 1.0461888509670079,
      "grad_norm": 0.836130183985909,
      "learning_rate": 1.119817353716678e-06,
      "loss": 0.012,
      "step": 4598
    },
    {
      "epoch": 1.0464163822525596,
      "grad_norm": 0.767678738883129,
      "learning_rate": 1.11976276393444e-06,
      "loss": 0.0114,
      "step": 4599
    },
    {
      "epoch": 1.0466439135381116,
      "grad_norm": 1.1278576378072667,
      "learning_rate": 1.1197081640401572e-06,
      "loss": 0.0245,
      "step": 4600
    },
    {
      "epoch": 1.0468714448236633,
      "grad_norm": 1.3358445644446304,
      "learning_rate": 1.1196535540349453e-06,
      "loss": 0.0414,
      "step": 4601
    },
    {
      "epoch": 1.047098976109215,
      "grad_norm": 2.748783161026122,
      "learning_rate": 1.119598933919921e-06,
      "loss": 0.0374,
      "step": 4602
    },
    {
      "epoch": 1.0473265073947668,
      "grad_norm": 0.7346665523199191,
      "learning_rate": 1.1195443036962002e-06,
      "loss": 0.0137,
      "step": 4603
    },
    {
      "epoch": 1.0475540386803186,
      "grad_norm": 0.7926661565300546,
      "learning_rate": 1.1194896633648996e-06,
      "loss": 0.0234,
      "step": 4604
    },
    {
      "epoch": 1.0477815699658704,
      "grad_norm": 1.2266234047001268,
      "learning_rate": 1.1194350129271358e-06,
      "loss": 0.0218,
      "step": 4605
    },
    {
      "epoch": 1.048009101251422,
      "grad_norm": 1.1781905579948284,
      "learning_rate": 1.119380352384026e-06,
      "loss": 0.0298,
      "step": 4606
    },
    {
      "epoch": 1.0482366325369739,
      "grad_norm": 0.8103390645469172,
      "learning_rate": 1.1193256817366871e-06,
      "loss": 0.014,
      "step": 4607
    },
    {
      "epoch": 1.0484641638225256,
      "grad_norm": 1.2189569921919905,
      "learning_rate": 1.1192710009862365e-06,
      "loss": 0.0252,
      "step": 4608
    },
    {
      "epoch": 1.0486916951080774,
      "grad_norm": 0.6581776967473145,
      "learning_rate": 1.1192163101337921e-06,
      "loss": 0.0129,
      "step": 4609
    },
    {
      "epoch": 1.0489192263936291,
      "grad_norm": 0.7853755589384823,
      "learning_rate": 1.1191616091804712e-06,
      "loss": 0.0346,
      "step": 4610
    },
    {
      "epoch": 1.0491467576791809,
      "grad_norm": 0.48263244857420745,
      "learning_rate": 1.1191068981273919e-06,
      "loss": 0.0113,
      "step": 4611
    },
    {
      "epoch": 1.0493742889647326,
      "grad_norm": 0.9545125321950874,
      "learning_rate": 1.1190521769756729e-06,
      "loss": 0.0254,
      "step": 4612
    },
    {
      "epoch": 1.0496018202502844,
      "grad_norm": 1.247134305539268,
      "learning_rate": 1.118997445726432e-06,
      "loss": 0.0413,
      "step": 4613
    },
    {
      "epoch": 1.0498293515358361,
      "grad_norm": 1.1502298928735868,
      "learning_rate": 1.118942704380788e-06,
      "loss": 0.0137,
      "step": 4614
    },
    {
      "epoch": 1.0500568828213879,
      "grad_norm": 1.6594682241469216,
      "learning_rate": 1.11888795293986e-06,
      "loss": 0.0326,
      "step": 4615
    },
    {
      "epoch": 1.0502844141069396,
      "grad_norm": 0.7549275144222661,
      "learning_rate": 1.1188331914047666e-06,
      "loss": 0.0138,
      "step": 4616
    },
    {
      "epoch": 1.0505119453924914,
      "grad_norm": 0.8321159003301328,
      "learning_rate": 1.1187784197766269e-06,
      "loss": 0.0184,
      "step": 4617
    },
    {
      "epoch": 1.0507394766780431,
      "grad_norm": 1.209949785555879,
      "learning_rate": 1.1187236380565608e-06,
      "loss": 0.0251,
      "step": 4618
    },
    {
      "epoch": 1.050967007963595,
      "grad_norm": 0.7955918295120653,
      "learning_rate": 1.1186688462456879e-06,
      "loss": 0.0135,
      "step": 4619
    },
    {
      "epoch": 1.0511945392491469,
      "grad_norm": 1.1615893854076145,
      "learning_rate": 1.118614044345128e-06,
      "loss": 0.0282,
      "step": 4620
    },
    {
      "epoch": 1.0514220705346986,
      "grad_norm": 1.273989641669821,
      "learning_rate": 1.1185592323560006e-06,
      "loss": 0.038,
      "step": 4621
    },
    {
      "epoch": 1.0516496018202504,
      "grad_norm": 1.0165071618076773,
      "learning_rate": 1.1185044102794267e-06,
      "loss": 0.0328,
      "step": 4622
    },
    {
      "epoch": 1.051877133105802,
      "grad_norm": 1.1328617777044239,
      "learning_rate": 1.1184495781165263e-06,
      "loss": 0.0274,
      "step": 4623
    },
    {
      "epoch": 1.0521046643913539,
      "grad_norm": 1.1825756024120404,
      "learning_rate": 1.1183947358684203e-06,
      "loss": 0.0276,
      "step": 4624
    },
    {
      "epoch": 1.0523321956769056,
      "grad_norm": 0.517945392477519,
      "learning_rate": 1.1183398835362298e-06,
      "loss": 0.0174,
      "step": 4625
    },
    {
      "epoch": 1.0525597269624574,
      "grad_norm": 0.5843868560216139,
      "learning_rate": 1.1182850211210752e-06,
      "loss": 0.0161,
      "step": 4626
    },
    {
      "epoch": 1.0527872582480091,
      "grad_norm": 1.2595121663614455,
      "learning_rate": 1.1182301486240782e-06,
      "loss": 0.0359,
      "step": 4627
    },
    {
      "epoch": 1.0530147895335609,
      "grad_norm": 1.0715185413956108,
      "learning_rate": 1.1181752660463604e-06,
      "loss": 0.0214,
      "step": 4628
    },
    {
      "epoch": 1.0532423208191126,
      "grad_norm": 0.7795449522978944,
      "learning_rate": 1.1181203733890433e-06,
      "loss": 0.0177,
      "step": 4629
    },
    {
      "epoch": 1.0534698521046644,
      "grad_norm": 0.9800972489749913,
      "learning_rate": 1.118065470653249e-06,
      "loss": 0.0347,
      "step": 4630
    },
    {
      "epoch": 1.0536973833902161,
      "grad_norm": 1.0970470234975969,
      "learning_rate": 1.1180105578400993e-06,
      "loss": 0.0433,
      "step": 4631
    },
    {
      "epoch": 1.0539249146757679,
      "grad_norm": 1.0800910071281984,
      "learning_rate": 1.117955634950717e-06,
      "loss": 0.0227,
      "step": 4632
    },
    {
      "epoch": 1.0541524459613196,
      "grad_norm": 1.3530664066926406,
      "learning_rate": 1.117900701986224e-06,
      "loss": 0.0343,
      "step": 4633
    },
    {
      "epoch": 1.0543799772468714,
      "grad_norm": 1.4133875078213476,
      "learning_rate": 1.1178457589477434e-06,
      "loss": 0.027,
      "step": 4634
    },
    {
      "epoch": 1.0546075085324231,
      "grad_norm": 0.8440744418991889,
      "learning_rate": 1.1177908058363984e-06,
      "loss": 0.0098,
      "step": 4635
    },
    {
      "epoch": 1.0548350398179749,
      "grad_norm": 1.092273870951362,
      "learning_rate": 1.1177358426533115e-06,
      "loss": 0.02,
      "step": 4636
    },
    {
      "epoch": 1.0550625711035266,
      "grad_norm": 0.9187729071178374,
      "learning_rate": 1.1176808693996067e-06,
      "loss": 0.0183,
      "step": 4637
    },
    {
      "epoch": 1.0552901023890784,
      "grad_norm": 1.1090047987424088,
      "learning_rate": 1.117625886076407e-06,
      "loss": 0.0193,
      "step": 4638
    },
    {
      "epoch": 1.0555176336746304,
      "grad_norm": 1.0008875695337944,
      "learning_rate": 1.1175708926848363e-06,
      "loss": 0.0189,
      "step": 4639
    },
    {
      "epoch": 1.0557451649601821,
      "grad_norm": 2.724829300569884,
      "learning_rate": 1.1175158892260187e-06,
      "loss": 0.0362,
      "step": 4640
    },
    {
      "epoch": 1.0559726962457339,
      "grad_norm": 1.0478197516778056,
      "learning_rate": 1.1174608757010785e-06,
      "loss": 0.0152,
      "step": 4641
    },
    {
      "epoch": 1.0562002275312856,
      "grad_norm": 1.590272661414187,
      "learning_rate": 1.1174058521111398e-06,
      "loss": 0.0436,
      "step": 4642
    },
    {
      "epoch": 1.0564277588168374,
      "grad_norm": 0.8472430737385992,
      "learning_rate": 1.1173508184573273e-06,
      "loss": 0.0138,
      "step": 4643
    },
    {
      "epoch": 1.0566552901023891,
      "grad_norm": 0.8371060858404447,
      "learning_rate": 1.1172957747407657e-06,
      "loss": 0.0157,
      "step": 4644
    },
    {
      "epoch": 1.0568828213879409,
      "grad_norm": 0.9592085999503343,
      "learning_rate": 1.11724072096258e-06,
      "loss": 0.0323,
      "step": 4645
    },
    {
      "epoch": 1.0571103526734926,
      "grad_norm": 1.0524749179147639,
      "learning_rate": 1.1171856571238958e-06,
      "loss": 0.0296,
      "step": 4646
    },
    {
      "epoch": 1.0573378839590444,
      "grad_norm": 0.8834933966797389,
      "learning_rate": 1.1171305832258378e-06,
      "loss": 0.0218,
      "step": 4647
    },
    {
      "epoch": 1.0575654152445961,
      "grad_norm": 0.7743496062659715,
      "learning_rate": 1.117075499269532e-06,
      "loss": 0.0183,
      "step": 4648
    },
    {
      "epoch": 1.0577929465301479,
      "grad_norm": 1.0385592346274115,
      "learning_rate": 1.1170204052561045e-06,
      "loss": 0.0205,
      "step": 4649
    },
    {
      "epoch": 1.0580204778156996,
      "grad_norm": 0.905517805098956,
      "learning_rate": 1.1169653011866806e-06,
      "loss": 0.0274,
      "step": 4650
    },
    {
      "epoch": 1.0582480091012514,
      "grad_norm": 0.7908819622644908,
      "learning_rate": 1.1169101870623872e-06,
      "loss": 0.0137,
      "step": 4651
    },
    {
      "epoch": 1.0584755403868031,
      "grad_norm": 1.061810097297376,
      "learning_rate": 1.1168550628843506e-06,
      "loss": 0.0206,
      "step": 4652
    },
    {
      "epoch": 1.058703071672355,
      "grad_norm": 1.6341823289625474,
      "learning_rate": 1.116799928653697e-06,
      "loss": 0.0258,
      "step": 4653
    },
    {
      "epoch": 1.0589306029579066,
      "grad_norm": 1.4294137058728704,
      "learning_rate": 1.1167447843715536e-06,
      "loss": 0.0415,
      "step": 4654
    },
    {
      "epoch": 1.0591581342434584,
      "grad_norm": 1.9352681748212788,
      "learning_rate": 1.1166896300390475e-06,
      "loss": 0.0398,
      "step": 4655
    },
    {
      "epoch": 1.0593856655290101,
      "grad_norm": 0.623531295073113,
      "learning_rate": 1.1166344656573058e-06,
      "loss": 0.0096,
      "step": 4656
    },
    {
      "epoch": 1.059613196814562,
      "grad_norm": 0.7942878422407874,
      "learning_rate": 1.116579291227456e-06,
      "loss": 0.0246,
      "step": 4657
    },
    {
      "epoch": 1.0598407281001139,
      "grad_norm": 1.2220975624229167,
      "learning_rate": 1.1165241067506258e-06,
      "loss": 0.0361,
      "step": 4658
    },
    {
      "epoch": 1.0600682593856656,
      "grad_norm": 1.1286970072887417,
      "learning_rate": 1.116468912227943e-06,
      "loss": 0.0263,
      "step": 4659
    },
    {
      "epoch": 1.0602957906712174,
      "grad_norm": 1.030142424103505,
      "learning_rate": 1.1164137076605359e-06,
      "loss": 0.0179,
      "step": 4660
    },
    {
      "epoch": 1.0605233219567691,
      "grad_norm": 1.6790275883871597,
      "learning_rate": 1.1163584930495323e-06,
      "loss": 0.0422,
      "step": 4661
    },
    {
      "epoch": 1.0607508532423209,
      "grad_norm": 1.3814052631606326,
      "learning_rate": 1.1163032683960612e-06,
      "loss": 0.0209,
      "step": 4662
    },
    {
      "epoch": 1.0609783845278726,
      "grad_norm": 1.225549316988825,
      "learning_rate": 1.116248033701251e-06,
      "loss": 0.0271,
      "step": 4663
    },
    {
      "epoch": 1.0612059158134244,
      "grad_norm": 0.7598712432018329,
      "learning_rate": 1.1161927889662307e-06,
      "loss": 0.009,
      "step": 4664
    },
    {
      "epoch": 1.0614334470989761,
      "grad_norm": 0.940084790434426,
      "learning_rate": 1.1161375341921293e-06,
      "loss": 0.0248,
      "step": 4665
    },
    {
      "epoch": 1.0616609783845279,
      "grad_norm": 0.5873146342311836,
      "learning_rate": 1.1160822693800761e-06,
      "loss": 0.0078,
      "step": 4666
    },
    {
      "epoch": 1.0618885096700796,
      "grad_norm": 1.4244470289207833,
      "learning_rate": 1.116026994531201e-06,
      "loss": 0.0248,
      "step": 4667
    },
    {
      "epoch": 1.0621160409556314,
      "grad_norm": 1.0520117462705862,
      "learning_rate": 1.1159717096466332e-06,
      "loss": 0.0302,
      "step": 4668
    },
    {
      "epoch": 1.0623435722411831,
      "grad_norm": 1.062118932659138,
      "learning_rate": 1.1159164147275026e-06,
      "loss": 0.022,
      "step": 4669
    },
    {
      "epoch": 1.062571103526735,
      "grad_norm": 1.1122338472373463,
      "learning_rate": 1.11586110977494e-06,
      "loss": 0.0164,
      "step": 4670
    },
    {
      "epoch": 1.0627986348122866,
      "grad_norm": 0.8000529898204216,
      "learning_rate": 1.1158057947900749e-06,
      "loss": 0.0217,
      "step": 4671
    },
    {
      "epoch": 1.0630261660978384,
      "grad_norm": 0.6219194713042089,
      "learning_rate": 1.1157504697740384e-06,
      "loss": 0.0138,
      "step": 4672
    },
    {
      "epoch": 1.0632536973833902,
      "grad_norm": 1.3392201881028478,
      "learning_rate": 1.1156951347279612e-06,
      "loss": 0.028,
      "step": 4673
    },
    {
      "epoch": 1.063481228668942,
      "grad_norm": 1.7333136164138454,
      "learning_rate": 1.1156397896529739e-06,
      "loss": 0.05,
      "step": 4674
    },
    {
      "epoch": 1.0637087599544937,
      "grad_norm": 1.5262332959624183,
      "learning_rate": 1.1155844345502079e-06,
      "loss": 0.0312,
      "step": 4675
    },
    {
      "epoch": 1.0639362912400454,
      "grad_norm": 0.9932229349703479,
      "learning_rate": 1.1155290694207946e-06,
      "loss": 0.0226,
      "step": 4676
    },
    {
      "epoch": 1.0641638225255972,
      "grad_norm": 1.2628573637308051,
      "learning_rate": 1.1154736942658655e-06,
      "loss": 0.0214,
      "step": 4677
    },
    {
      "epoch": 1.0643913538111491,
      "grad_norm": 1.4330647372008305,
      "learning_rate": 1.1154183090865523e-06,
      "loss": 0.0256,
      "step": 4678
    },
    {
      "epoch": 1.0646188850967009,
      "grad_norm": 1.042200851154514,
      "learning_rate": 1.1153629138839869e-06,
      "loss": 0.0293,
      "step": 4679
    },
    {
      "epoch": 1.0648464163822526,
      "grad_norm": 0.9462778481987262,
      "learning_rate": 1.115307508659302e-06,
      "loss": 0.0245,
      "step": 4680
    },
    {
      "epoch": 1.0650739476678044,
      "grad_norm": 1.6564513386390822,
      "learning_rate": 1.115252093413629e-06,
      "loss": 0.0496,
      "step": 4681
    },
    {
      "epoch": 1.0653014789533561,
      "grad_norm": 1.0868647502168836,
      "learning_rate": 1.1151966681481013e-06,
      "loss": 0.0113,
      "step": 4682
    },
    {
      "epoch": 1.065529010238908,
      "grad_norm": 1.115171642211064,
      "learning_rate": 1.1151412328638516e-06,
      "loss": 0.0224,
      "step": 4683
    },
    {
      "epoch": 1.0657565415244596,
      "grad_norm": 0.5398893154965461,
      "learning_rate": 1.1150857875620129e-06,
      "loss": 0.0139,
      "step": 4684
    },
    {
      "epoch": 1.0659840728100114,
      "grad_norm": 1.0871499987134439,
      "learning_rate": 1.1150303322437179e-06,
      "loss": 0.0134,
      "step": 4685
    },
    {
      "epoch": 1.0662116040955631,
      "grad_norm": 1.1070037508570774,
      "learning_rate": 1.1149748669101005e-06,
      "loss": 0.0221,
      "step": 4686
    },
    {
      "epoch": 1.066439135381115,
      "grad_norm": 0.9425411653474299,
      "learning_rate": 1.1149193915622942e-06,
      "loss": 0.0398,
      "step": 4687
    },
    {
      "epoch": 1.0666666666666667,
      "grad_norm": 1.1772890828333267,
      "learning_rate": 1.1148639062014325e-06,
      "loss": 0.0211,
      "step": 4688
    },
    {
      "epoch": 1.0668941979522184,
      "grad_norm": 1.2557238706849523,
      "learning_rate": 1.11480841082865e-06,
      "loss": 0.0301,
      "step": 4689
    },
    {
      "epoch": 1.0671217292377702,
      "grad_norm": 0.7994780640965474,
      "learning_rate": 1.1147529054450805e-06,
      "loss": 0.0201,
      "step": 4690
    },
    {
      "epoch": 1.067349260523322,
      "grad_norm": 1.1399734758992979,
      "learning_rate": 1.1146973900518587e-06,
      "loss": 0.0293,
      "step": 4691
    },
    {
      "epoch": 1.0675767918088737,
      "grad_norm": 1.3046065395944408,
      "learning_rate": 1.1146418646501189e-06,
      "loss": 0.0286,
      "step": 4692
    },
    {
      "epoch": 1.0678043230944254,
      "grad_norm": 0.8748609055579856,
      "learning_rate": 1.114586329240996e-06,
      "loss": 0.0211,
      "step": 4693
    },
    {
      "epoch": 1.0680318543799772,
      "grad_norm": 0.8739676230672317,
      "learning_rate": 1.1145307838256255e-06,
      "loss": 0.011,
      "step": 4694
    },
    {
      "epoch": 1.068259385665529,
      "grad_norm": 1.249091385352741,
      "learning_rate": 1.1144752284051422e-06,
      "loss": 0.0301,
      "step": 4695
    },
    {
      "epoch": 1.068486916951081,
      "grad_norm": 1.562897636481294,
      "learning_rate": 1.1144196629806817e-06,
      "loss": 0.0501,
      "step": 4696
    },
    {
      "epoch": 1.0687144482366326,
      "grad_norm": 0.9143311380884951,
      "learning_rate": 1.1143640875533795e-06,
      "loss": 0.0123,
      "step": 4697
    },
    {
      "epoch": 1.0689419795221844,
      "grad_norm": 1.6200825529791434,
      "learning_rate": 1.1143085021243717e-06,
      "loss": 0.0225,
      "step": 4698
    },
    {
      "epoch": 1.0691695108077361,
      "grad_norm": 1.177872950810007,
      "learning_rate": 1.1142529066947941e-06,
      "loss": 0.0346,
      "step": 4699
    },
    {
      "epoch": 1.069397042093288,
      "grad_norm": 1.108013166594474,
      "learning_rate": 1.1141973012657834e-06,
      "loss": 0.0267,
      "step": 4700
    },
    {
      "epoch": 1.0696245733788396,
      "grad_norm": 1.0112425933651041,
      "learning_rate": 1.1141416858384753e-06,
      "loss": 0.0311,
      "step": 4701
    },
    {
      "epoch": 1.0698521046643914,
      "grad_norm": 1.095144486124691,
      "learning_rate": 1.1140860604140076e-06,
      "loss": 0.0286,
      "step": 4702
    },
    {
      "epoch": 1.0700796359499432,
      "grad_norm": 0.806336687142019,
      "learning_rate": 1.114030424993516e-06,
      "loss": 0.0142,
      "step": 4703
    },
    {
      "epoch": 1.070307167235495,
      "grad_norm": 0.7834952051331864,
      "learning_rate": 1.1139747795781382e-06,
      "loss": 0.0101,
      "step": 4704
    },
    {
      "epoch": 1.0705346985210467,
      "grad_norm": 0.7472560755343798,
      "learning_rate": 1.1139191241690116e-06,
      "loss": 0.0139,
      "step": 4705
    },
    {
      "epoch": 1.0707622298065984,
      "grad_norm": 0.5618125746941464,
      "learning_rate": 1.1138634587672734e-06,
      "loss": 0.0075,
      "step": 4706
    },
    {
      "epoch": 1.0709897610921502,
      "grad_norm": 0.7202976835755431,
      "learning_rate": 1.1138077833740616e-06,
      "loss": 0.0207,
      "step": 4707
    },
    {
      "epoch": 1.071217292377702,
      "grad_norm": 0.7307887063589182,
      "learning_rate": 1.1137520979905138e-06,
      "loss": 0.0223,
      "step": 4708
    },
    {
      "epoch": 1.0714448236632537,
      "grad_norm": 1.1867538184942938,
      "learning_rate": 1.1136964026177683e-06,
      "loss": 0.0288,
      "step": 4709
    },
    {
      "epoch": 1.0716723549488054,
      "grad_norm": 1.2840239837956307,
      "learning_rate": 1.113640697256963e-06,
      "loss": 0.0184,
      "step": 4710
    },
    {
      "epoch": 1.0718998862343572,
      "grad_norm": 1.1345754398737768,
      "learning_rate": 1.113584981909237e-06,
      "loss": 0.0238,
      "step": 4711
    },
    {
      "epoch": 1.072127417519909,
      "grad_norm": 0.9685288187674116,
      "learning_rate": 1.1135292565757288e-06,
      "loss": 0.0227,
      "step": 4712
    },
    {
      "epoch": 1.0723549488054607,
      "grad_norm": 0.7081273094248537,
      "learning_rate": 1.1134735212575772e-06,
      "loss": 0.0102,
      "step": 4713
    },
    {
      "epoch": 1.0725824800910124,
      "grad_norm": 0.9192931039025396,
      "learning_rate": 1.1134177759559216e-06,
      "loss": 0.0244,
      "step": 4714
    },
    {
      "epoch": 1.0728100113765642,
      "grad_norm": 0.9284668402003036,
      "learning_rate": 1.1133620206719011e-06,
      "loss": 0.0243,
      "step": 4715
    },
    {
      "epoch": 1.073037542662116,
      "grad_norm": 0.9115758932784177,
      "learning_rate": 1.1133062554066551e-06,
      "loss": 0.0205,
      "step": 4716
    },
    {
      "epoch": 1.073265073947668,
      "grad_norm": 0.8734771500689966,
      "learning_rate": 1.1132504801613237e-06,
      "loss": 0.0213,
      "step": 4717
    },
    {
      "epoch": 1.0734926052332197,
      "grad_norm": 1.126626337283734,
      "learning_rate": 1.1131946949370467e-06,
      "loss": 0.0327,
      "step": 4718
    },
    {
      "epoch": 1.0737201365187714,
      "grad_norm": 0.65003673022336,
      "learning_rate": 1.113138899734964e-06,
      "loss": 0.0162,
      "step": 4719
    },
    {
      "epoch": 1.0739476678043232,
      "grad_norm": 1.6524784863925304,
      "learning_rate": 1.1130830945562165e-06,
      "loss": 0.046,
      "step": 4720
    },
    {
      "epoch": 1.074175199089875,
      "grad_norm": 0.9704754115890377,
      "learning_rate": 1.1130272794019442e-06,
      "loss": 0.0165,
      "step": 4721
    },
    {
      "epoch": 1.0744027303754267,
      "grad_norm": 0.841189725514391,
      "learning_rate": 1.1129714542732882e-06,
      "loss": 0.0209,
      "step": 4722
    },
    {
      "epoch": 1.0746302616609784,
      "grad_norm": 0.6874778292132357,
      "learning_rate": 1.1129156191713893e-06,
      "loss": 0.0121,
      "step": 4723
    },
    {
      "epoch": 1.0748577929465302,
      "grad_norm": 0.9632218181611477,
      "learning_rate": 1.1128597740973886e-06,
      "loss": 0.0388,
      "step": 4724
    },
    {
      "epoch": 1.075085324232082,
      "grad_norm": 0.7703489214733116,
      "learning_rate": 1.1128039190524278e-06,
      "loss": 0.0059,
      "step": 4725
    },
    {
      "epoch": 1.0753128555176337,
      "grad_norm": 1.9572066317541847,
      "learning_rate": 1.112748054037648e-06,
      "loss": 0.0262,
      "step": 4726
    },
    {
      "epoch": 1.0755403868031854,
      "grad_norm": 1.0824319203641397,
      "learning_rate": 1.1126921790541915e-06,
      "loss": 0.0149,
      "step": 4727
    },
    {
      "epoch": 1.0757679180887372,
      "grad_norm": 0.670440042251469,
      "learning_rate": 1.1126362941032e-06,
      "loss": 0.0063,
      "step": 4728
    },
    {
      "epoch": 1.075995449374289,
      "grad_norm": 0.7800421077003349,
      "learning_rate": 1.1125803991858156e-06,
      "loss": 0.0277,
      "step": 4729
    },
    {
      "epoch": 1.0762229806598407,
      "grad_norm": 1.106352016812619,
      "learning_rate": 1.1125244943031809e-06,
      "loss": 0.0123,
      "step": 4730
    },
    {
      "epoch": 1.0764505119453924,
      "grad_norm": 1.1161991172358912,
      "learning_rate": 1.1124685794564383e-06,
      "loss": 0.026,
      "step": 4731
    },
    {
      "epoch": 1.0766780432309442,
      "grad_norm": 1.7881963402264416,
      "learning_rate": 1.1124126546467307e-06,
      "loss": 0.0385,
      "step": 4732
    },
    {
      "epoch": 1.076905574516496,
      "grad_norm": 1.2868577345652183,
      "learning_rate": 1.1123567198752012e-06,
      "loss": 0.0273,
      "step": 4733
    },
    {
      "epoch": 1.0771331058020477,
      "grad_norm": 2.0351098504278085,
      "learning_rate": 1.1123007751429928e-06,
      "loss": 0.0568,
      "step": 4734
    },
    {
      "epoch": 1.0773606370875997,
      "grad_norm": 0.9002868646294088,
      "learning_rate": 1.1122448204512493e-06,
      "loss": 0.0253,
      "step": 4735
    },
    {
      "epoch": 1.0775881683731514,
      "grad_norm": 0.9063013425685674,
      "learning_rate": 1.1121888558011136e-06,
      "loss": 0.0117,
      "step": 4736
    },
    {
      "epoch": 1.0778156996587032,
      "grad_norm": 0.5859089097561659,
      "learning_rate": 1.11213288119373e-06,
      "loss": 0.0074,
      "step": 4737
    },
    {
      "epoch": 1.078043230944255,
      "grad_norm": 1.3479752927519844,
      "learning_rate": 1.1120768966302424e-06,
      "loss": 0.0249,
      "step": 4738
    },
    {
      "epoch": 1.0782707622298067,
      "grad_norm": 0.7352263154143484,
      "learning_rate": 1.1120209021117953e-06,
      "loss": 0.0173,
      "step": 4739
    },
    {
      "epoch": 1.0784982935153584,
      "grad_norm": 0.7374112860235839,
      "learning_rate": 1.111964897639533e-06,
      "loss": 0.0174,
      "step": 4740
    },
    {
      "epoch": 1.0787258248009102,
      "grad_norm": 1.287591128879902,
      "learning_rate": 1.1119088832145999e-06,
      "loss": 0.0326,
      "step": 4741
    },
    {
      "epoch": 1.078953356086462,
      "grad_norm": 1.0787165555211229,
      "learning_rate": 1.1118528588381408e-06,
      "loss": 0.0192,
      "step": 4742
    },
    {
      "epoch": 1.0791808873720137,
      "grad_norm": 1.0815099906932875,
      "learning_rate": 1.111796824511301e-06,
      "loss": 0.0216,
      "step": 4743
    },
    {
      "epoch": 1.0794084186575654,
      "grad_norm": 0.8664538220627443,
      "learning_rate": 1.1117407802352257e-06,
      "loss": 0.0127,
      "step": 4744
    },
    {
      "epoch": 1.0796359499431172,
      "grad_norm": 1.33972724465675,
      "learning_rate": 1.11168472601106e-06,
      "loss": 0.0335,
      "step": 4745
    },
    {
      "epoch": 1.079863481228669,
      "grad_norm": 1.4666132789919955,
      "learning_rate": 1.1116286618399502e-06,
      "loss": 0.0457,
      "step": 4746
    },
    {
      "epoch": 1.0800910125142207,
      "grad_norm": 0.6790281378613789,
      "learning_rate": 1.1115725877230416e-06,
      "loss": 0.0102,
      "step": 4747
    },
    {
      "epoch": 1.0803185437997724,
      "grad_norm": 0.9620611501361773,
      "learning_rate": 1.1115165036614803e-06,
      "loss": 0.0261,
      "step": 4748
    },
    {
      "epoch": 1.0805460750853242,
      "grad_norm": 1.1237272287467874,
      "learning_rate": 1.1114604096564128e-06,
      "loss": 0.0316,
      "step": 4749
    },
    {
      "epoch": 1.080773606370876,
      "grad_norm": 1.0088779610949172,
      "learning_rate": 1.1114043057089855e-06,
      "loss": 0.0225,
      "step": 4750
    },
    {
      "epoch": 1.0810011376564277,
      "grad_norm": 1.497609697625705,
      "learning_rate": 1.1113481918203447e-06,
      "loss": 0.0442,
      "step": 4751
    },
    {
      "epoch": 1.0812286689419794,
      "grad_norm": 0.8914252374944144,
      "learning_rate": 1.111292067991638e-06,
      "loss": 0.0165,
      "step": 4752
    },
    {
      "epoch": 1.0814562002275312,
      "grad_norm": 1.276532043770101,
      "learning_rate": 1.1112359342240118e-06,
      "loss": 0.0387,
      "step": 4753
    },
    {
      "epoch": 1.081683731513083,
      "grad_norm": 0.9077763034087386,
      "learning_rate": 1.1111797905186137e-06,
      "loss": 0.0141,
      "step": 4754
    },
    {
      "epoch": 1.0819112627986347,
      "grad_norm": 1.6466072809663417,
      "learning_rate": 1.111123636876591e-06,
      "loss": 0.0349,
      "step": 4755
    },
    {
      "epoch": 1.0821387940841867,
      "grad_norm": 0.9430919880459814,
      "learning_rate": 1.1110674732990915e-06,
      "loss": 0.016,
      "step": 4756
    },
    {
      "epoch": 1.0823663253697384,
      "grad_norm": 1.6404185778699842,
      "learning_rate": 1.1110112997872627e-06,
      "loss": 0.0309,
      "step": 4757
    },
    {
      "epoch": 1.0825938566552902,
      "grad_norm": 0.9479110834953683,
      "learning_rate": 1.1109551163422535e-06,
      "loss": 0.0189,
      "step": 4758
    },
    {
      "epoch": 1.082821387940842,
      "grad_norm": 0.9870118719493604,
      "learning_rate": 1.1108989229652115e-06,
      "loss": 0.0221,
      "step": 4759
    },
    {
      "epoch": 1.0830489192263937,
      "grad_norm": 1.7588111531610175,
      "learning_rate": 1.1108427196572854e-06,
      "loss": 0.0327,
      "step": 4760
    },
    {
      "epoch": 1.0832764505119454,
      "grad_norm": 0.9021346736655393,
      "learning_rate": 1.110786506419624e-06,
      "loss": 0.036,
      "step": 4761
    },
    {
      "epoch": 1.0835039817974972,
      "grad_norm": 0.8266880922651803,
      "learning_rate": 1.110730283253376e-06,
      "loss": 0.0279,
      "step": 4762
    },
    {
      "epoch": 1.083731513083049,
      "grad_norm": 0.8011724413374426,
      "learning_rate": 1.1106740501596904e-06,
      "loss": 0.0195,
      "step": 4763
    },
    {
      "epoch": 1.0839590443686007,
      "grad_norm": 0.9073688021992705,
      "learning_rate": 1.110617807139717e-06,
      "loss": 0.023,
      "step": 4764
    },
    {
      "epoch": 1.0841865756541524,
      "grad_norm": 1.022169451105952,
      "learning_rate": 1.1105615541946049e-06,
      "loss": 0.0209,
      "step": 4765
    },
    {
      "epoch": 1.0844141069397042,
      "grad_norm": 0.43405837742396564,
      "learning_rate": 1.1105052913255038e-06,
      "loss": 0.0044,
      "step": 4766
    },
    {
      "epoch": 1.084641638225256,
      "grad_norm": 1.383667776402452,
      "learning_rate": 1.1104490185335638e-06,
      "loss": 0.0287,
      "step": 4767
    },
    {
      "epoch": 1.0848691695108077,
      "grad_norm": 0.9054919782471123,
      "learning_rate": 1.1103927358199349e-06,
      "loss": 0.0191,
      "step": 4768
    },
    {
      "epoch": 1.0850967007963594,
      "grad_norm": 1.2358115777885788,
      "learning_rate": 1.1103364431857672e-06,
      "loss": 0.0311,
      "step": 4769
    },
    {
      "epoch": 1.0853242320819112,
      "grad_norm": 0.7048379529354732,
      "learning_rate": 1.1102801406322118e-06,
      "loss": 0.0104,
      "step": 4770
    },
    {
      "epoch": 1.085551763367463,
      "grad_norm": 1.1457330667513694,
      "learning_rate": 1.110223828160419e-06,
      "loss": 0.0212,
      "step": 4771
    },
    {
      "epoch": 1.0857792946530147,
      "grad_norm": 0.7058751162839924,
      "learning_rate": 1.1101675057715396e-06,
      "loss": 0.0092,
      "step": 4772
    },
    {
      "epoch": 1.0860068259385665,
      "grad_norm": 1.1868874800053821,
      "learning_rate": 1.110111173466725e-06,
      "loss": 0.0416,
      "step": 4773
    },
    {
      "epoch": 1.0862343572241184,
      "grad_norm": 1.1919649645556245,
      "learning_rate": 1.1100548312471266e-06,
      "loss": 0.0287,
      "step": 4774
    },
    {
      "epoch": 1.0864618885096702,
      "grad_norm": 1.0824756647963032,
      "learning_rate": 1.1099984791138957e-06,
      "loss": 0.0152,
      "step": 4775
    },
    {
      "epoch": 1.086689419795222,
      "grad_norm": 1.4691023279738935,
      "learning_rate": 1.109942117068184e-06,
      "loss": 0.0201,
      "step": 4776
    },
    {
      "epoch": 1.0869169510807737,
      "grad_norm": 0.8472040472970495,
      "learning_rate": 1.1098857451111437e-06,
      "loss": 0.0142,
      "step": 4777
    },
    {
      "epoch": 1.0871444823663254,
      "grad_norm": 0.8480336544082094,
      "learning_rate": 1.1098293632439267e-06,
      "loss": 0.009,
      "step": 4778
    },
    {
      "epoch": 1.0873720136518772,
      "grad_norm": 1.1145190027001157,
      "learning_rate": 1.1097729714676855e-06,
      "loss": 0.0143,
      "step": 4779
    },
    {
      "epoch": 1.087599544937429,
      "grad_norm": 1.04224113160413,
      "learning_rate": 1.1097165697835726e-06,
      "loss": 0.024,
      "step": 4780
    },
    {
      "epoch": 1.0878270762229807,
      "grad_norm": 1.0620838647799002,
      "learning_rate": 1.1096601581927407e-06,
      "loss": 0.0178,
      "step": 4781
    },
    {
      "epoch": 1.0880546075085324,
      "grad_norm": 4.011373250629747,
      "learning_rate": 1.109603736696343e-06,
      "loss": 0.1504,
      "step": 4782
    },
    {
      "epoch": 1.0882821387940842,
      "grad_norm": 1.5174512485770184,
      "learning_rate": 1.1095473052955322e-06,
      "loss": 0.0405,
      "step": 4783
    },
    {
      "epoch": 1.088509670079636,
      "grad_norm": 0.884113509373173,
      "learning_rate": 1.1094908639914617e-06,
      "loss": 0.0224,
      "step": 4784
    },
    {
      "epoch": 1.0887372013651877,
      "grad_norm": 1.3619635032370507,
      "learning_rate": 1.1094344127852855e-06,
      "loss": 0.037,
      "step": 4785
    },
    {
      "epoch": 1.0889647326507395,
      "grad_norm": 1.9047416105199788,
      "learning_rate": 1.1093779516781571e-06,
      "loss": 0.0361,
      "step": 4786
    },
    {
      "epoch": 1.0891922639362912,
      "grad_norm": 1.1728707520462738,
      "learning_rate": 1.1093214806712305e-06,
      "loss": 0.0389,
      "step": 4787
    },
    {
      "epoch": 1.089419795221843,
      "grad_norm": 1.3254160466529121,
      "learning_rate": 1.1092649997656597e-06,
      "loss": 0.0232,
      "step": 4788
    },
    {
      "epoch": 1.0896473265073947,
      "grad_norm": 1.0725037617108157,
      "learning_rate": 1.1092085089625992e-06,
      "loss": 0.0164,
      "step": 4789
    },
    {
      "epoch": 1.0898748577929465,
      "grad_norm": 0.9758736007788956,
      "learning_rate": 1.1091520082632037e-06,
      "loss": 0.0218,
      "step": 4790
    },
    {
      "epoch": 1.0901023890784982,
      "grad_norm": 0.8477305521502946,
      "learning_rate": 1.1090954976686277e-06,
      "loss": 0.0245,
      "step": 4791
    },
    {
      "epoch": 1.09032992036405,
      "grad_norm": 0.9962558192273867,
      "learning_rate": 1.1090389771800264e-06,
      "loss": 0.0287,
      "step": 4792
    },
    {
      "epoch": 1.0905574516496017,
      "grad_norm": 1.2473669308596411,
      "learning_rate": 1.1089824467985549e-06,
      "loss": 0.0229,
      "step": 4793
    },
    {
      "epoch": 1.0907849829351535,
      "grad_norm": 2.510749402051243,
      "learning_rate": 1.1089259065253684e-06,
      "loss": 0.0259,
      "step": 4794
    },
    {
      "epoch": 1.0910125142207054,
      "grad_norm": 1.149992705234752,
      "learning_rate": 1.1088693563616226e-06,
      "loss": 0.0247,
      "step": 4795
    },
    {
      "epoch": 1.0912400455062572,
      "grad_norm": 0.5940460865164179,
      "learning_rate": 1.1088127963084736e-06,
      "loss": 0.0171,
      "step": 4796
    },
    {
      "epoch": 1.091467576791809,
      "grad_norm": 0.9164786766368149,
      "learning_rate": 1.108756226367077e-06,
      "loss": 0.0317,
      "step": 4797
    },
    {
      "epoch": 1.0916951080773607,
      "grad_norm": 1.1244091740172437,
      "learning_rate": 1.108699646538589e-06,
      "loss": 0.0255,
      "step": 4798
    },
    {
      "epoch": 1.0919226393629125,
      "grad_norm": 1.437903004135215,
      "learning_rate": 1.108643056824166e-06,
      "loss": 0.0464,
      "step": 4799
    },
    {
      "epoch": 1.0921501706484642,
      "grad_norm": 0.9254577109068766,
      "learning_rate": 1.108586457224965e-06,
      "loss": 0.0194,
      "step": 4800
    },
    {
      "epoch": 1.092377701934016,
      "grad_norm": 1.1931575439362083,
      "learning_rate": 1.1085298477421421e-06,
      "loss": 0.015,
      "step": 4801
    },
    {
      "epoch": 1.0926052332195677,
      "grad_norm": 0.6083574023703965,
      "learning_rate": 1.1084732283768548e-06,
      "loss": 0.0057,
      "step": 4802
    },
    {
      "epoch": 1.0928327645051195,
      "grad_norm": 0.8597843041320873,
      "learning_rate": 1.1084165991302601e-06,
      "loss": 0.0117,
      "step": 4803
    },
    {
      "epoch": 1.0930602957906712,
      "grad_norm": 1.4390855321709966,
      "learning_rate": 1.1083599600035155e-06,
      "loss": 0.0712,
      "step": 4804
    },
    {
      "epoch": 1.093287827076223,
      "grad_norm": 0.9189557088411734,
      "learning_rate": 1.1083033109977787e-06,
      "loss": 0.0248,
      "step": 4805
    },
    {
      "epoch": 1.0935153583617747,
      "grad_norm": 0.7847663302953185,
      "learning_rate": 1.1082466521142072e-06,
      "loss": 0.0138,
      "step": 4806
    },
    {
      "epoch": 1.0937428896473265,
      "grad_norm": 1.0427794259157794,
      "learning_rate": 1.1081899833539592e-06,
      "loss": 0.022,
      "step": 4807
    },
    {
      "epoch": 1.0939704209328782,
      "grad_norm": 0.7963208995055476,
      "learning_rate": 1.1081333047181928e-06,
      "loss": 0.0148,
      "step": 4808
    },
    {
      "epoch": 1.09419795221843,
      "grad_norm": 1.2861710540407572,
      "learning_rate": 1.1080766162080664e-06,
      "loss": 0.044,
      "step": 4809
    },
    {
      "epoch": 1.0944254835039817,
      "grad_norm": 1.0606108464391806,
      "learning_rate": 1.1080199178247388e-06,
      "loss": 0.0227,
      "step": 4810
    },
    {
      "epoch": 1.0946530147895335,
      "grad_norm": 1.0178256141989168,
      "learning_rate": 1.1079632095693688e-06,
      "loss": 0.0217,
      "step": 4811
    },
    {
      "epoch": 1.0948805460750852,
      "grad_norm": 0.3733490330289167,
      "learning_rate": 1.107906491443115e-06,
      "loss": 0.0041,
      "step": 4812
    },
    {
      "epoch": 1.0951080773606372,
      "grad_norm": 0.970660447630168,
      "learning_rate": 1.1078497634471373e-06,
      "loss": 0.0191,
      "step": 4813
    },
    {
      "epoch": 1.095335608646189,
      "grad_norm": 0.9571317568580685,
      "learning_rate": 1.1077930255825944e-06,
      "loss": 0.0146,
      "step": 4814
    },
    {
      "epoch": 1.0955631399317407,
      "grad_norm": 1.4227698048199817,
      "learning_rate": 1.1077362778506464e-06,
      "loss": 0.0259,
      "step": 4815
    },
    {
      "epoch": 1.0957906712172925,
      "grad_norm": 0.7937432954740787,
      "learning_rate": 1.107679520252453e-06,
      "loss": 0.0215,
      "step": 4816
    },
    {
      "epoch": 1.0960182025028442,
      "grad_norm": 1.2433291885534135,
      "learning_rate": 1.107622752789174e-06,
      "loss": 0.0271,
      "step": 4817
    },
    {
      "epoch": 1.096245733788396,
      "grad_norm": 0.5775769553380794,
      "learning_rate": 1.10756597546197e-06,
      "loss": 0.0067,
      "step": 4818
    },
    {
      "epoch": 1.0964732650739477,
      "grad_norm": 0.43052018257915775,
      "learning_rate": 1.1075091882720012e-06,
      "loss": 0.0047,
      "step": 4819
    },
    {
      "epoch": 1.0967007963594995,
      "grad_norm": 0.6965681550864847,
      "learning_rate": 1.1074523912204282e-06,
      "loss": 0.0069,
      "step": 4820
    },
    {
      "epoch": 1.0969283276450512,
      "grad_norm": 1.950972202735377,
      "learning_rate": 1.107395584308412e-06,
      "loss": 0.0384,
      "step": 4821
    },
    {
      "epoch": 1.097155858930603,
      "grad_norm": 1.2755184486485458,
      "learning_rate": 1.1073387675371134e-06,
      "loss": 0.021,
      "step": 4822
    },
    {
      "epoch": 1.0973833902161547,
      "grad_norm": 0.9195102681671121,
      "learning_rate": 1.1072819409076937e-06,
      "loss": 0.0233,
      "step": 4823
    },
    {
      "epoch": 1.0976109215017065,
      "grad_norm": 1.4467096094976202,
      "learning_rate": 1.1072251044213146e-06,
      "loss": 0.063,
      "step": 4824
    },
    {
      "epoch": 1.0978384527872582,
      "grad_norm": 0.9276103854462384,
      "learning_rate": 1.1071682580791375e-06,
      "loss": 0.0186,
      "step": 4825
    },
    {
      "epoch": 1.09806598407281,
      "grad_norm": 0.6788301923178457,
      "learning_rate": 1.107111401882324e-06,
      "loss": 0.0178,
      "step": 4826
    },
    {
      "epoch": 1.0982935153583617,
      "grad_norm": 0.8922254579510575,
      "learning_rate": 1.1070545358320367e-06,
      "loss": 0.0205,
      "step": 4827
    },
    {
      "epoch": 1.0985210466439135,
      "grad_norm": 1.5932312378336655,
      "learning_rate": 1.1069976599294374e-06,
      "loss": 0.0296,
      "step": 4828
    },
    {
      "epoch": 1.0987485779294652,
      "grad_norm": 1.3474193896215136,
      "learning_rate": 1.1069407741756884e-06,
      "loss": 0.0108,
      "step": 4829
    },
    {
      "epoch": 1.098976109215017,
      "grad_norm": 1.286192731643991,
      "learning_rate": 1.106883878571953e-06,
      "loss": 0.0306,
      "step": 4830
    },
    {
      "epoch": 1.0992036405005687,
      "grad_norm": 1.7859019215489673,
      "learning_rate": 1.1068269731193936e-06,
      "loss": 0.0467,
      "step": 4831
    },
    {
      "epoch": 1.0994311717861205,
      "grad_norm": 1.0673033994327072,
      "learning_rate": 1.106770057819173e-06,
      "loss": 0.011,
      "step": 4832
    },
    {
      "epoch": 1.0996587030716722,
      "grad_norm": 1.1919269451119046,
      "learning_rate": 1.1067131326724551e-06,
      "loss": 0.0275,
      "step": 4833
    },
    {
      "epoch": 1.0998862343572242,
      "grad_norm": 1.0610839759729118,
      "learning_rate": 1.106656197680403e-06,
      "loss": 0.0175,
      "step": 4834
    },
    {
      "epoch": 1.100113765642776,
      "grad_norm": 0.9398630402901391,
      "learning_rate": 1.10659925284418e-06,
      "loss": 0.028,
      "step": 4835
    },
    {
      "epoch": 1.1003412969283277,
      "grad_norm": 1.0905561404271749,
      "learning_rate": 1.1065422981649506e-06,
      "loss": 0.0408,
      "step": 4836
    },
    {
      "epoch": 1.1005688282138795,
      "grad_norm": 1.1760138038912833,
      "learning_rate": 1.1064853336438782e-06,
      "loss": 0.0491,
      "step": 4837
    },
    {
      "epoch": 1.1007963594994312,
      "grad_norm": 0.8244510688827178,
      "learning_rate": 1.1064283592821276e-06,
      "loss": 0.0197,
      "step": 4838
    },
    {
      "epoch": 1.101023890784983,
      "grad_norm": 0.747177088030229,
      "learning_rate": 1.106371375080863e-06,
      "loss": 0.017,
      "step": 4839
    },
    {
      "epoch": 1.1012514220705347,
      "grad_norm": 1.1336438441896446,
      "learning_rate": 1.106314381041249e-06,
      "loss": 0.0253,
      "step": 4840
    },
    {
      "epoch": 1.1014789533560865,
      "grad_norm": 0.5114058401483557,
      "learning_rate": 1.1062573771644506e-06,
      "loss": 0.004,
      "step": 4841
    },
    {
      "epoch": 1.1017064846416382,
      "grad_norm": 0.5746769321004888,
      "learning_rate": 1.106200363451633e-06,
      "loss": 0.0094,
      "step": 4842
    },
    {
      "epoch": 1.10193401592719,
      "grad_norm": 0.7680543679306212,
      "learning_rate": 1.1061433399039608e-06,
      "loss": 0.029,
      "step": 4843
    },
    {
      "epoch": 1.1021615472127417,
      "grad_norm": 0.7493587776816238,
      "learning_rate": 1.1060863065226002e-06,
      "loss": 0.0091,
      "step": 4844
    },
    {
      "epoch": 1.1023890784982935,
      "grad_norm": 1.3476928607388319,
      "learning_rate": 1.1060292633087167e-06,
      "loss": 0.0436,
      "step": 4845
    },
    {
      "epoch": 1.1026166097838452,
      "grad_norm": 1.087918078217182,
      "learning_rate": 1.1059722102634756e-06,
      "loss": 0.0201,
      "step": 4846
    },
    {
      "epoch": 1.102844141069397,
      "grad_norm": 1.0929486099530539,
      "learning_rate": 1.1059151473880439e-06,
      "loss": 0.0328,
      "step": 4847
    },
    {
      "epoch": 1.1030716723549487,
      "grad_norm": 1.3189660985869427,
      "learning_rate": 1.105858074683587e-06,
      "loss": 0.0142,
      "step": 4848
    },
    {
      "epoch": 1.1032992036405005,
      "grad_norm": 1.052151916181327,
      "learning_rate": 1.1058009921512717e-06,
      "loss": 0.0195,
      "step": 4849
    },
    {
      "epoch": 1.1035267349260522,
      "grad_norm": 1.0009913456254373,
      "learning_rate": 1.1057438997922648e-06,
      "loss": 0.022,
      "step": 4850
    },
    {
      "epoch": 1.103754266211604,
      "grad_norm": 0.9615349865705245,
      "learning_rate": 1.105686797607733e-06,
      "loss": 0.0111,
      "step": 4851
    },
    {
      "epoch": 1.103981797497156,
      "grad_norm": 1.3130785181889084,
      "learning_rate": 1.1056296855988432e-06,
      "loss": 0.0235,
      "step": 4852
    },
    {
      "epoch": 1.1042093287827077,
      "grad_norm": 0.8672033139838674,
      "learning_rate": 1.105572563766763e-06,
      "loss": 0.0243,
      "step": 4853
    },
    {
      "epoch": 1.1044368600682595,
      "grad_norm": 1.0127688429987634,
      "learning_rate": 1.1055154321126597e-06,
      "loss": 0.0153,
      "step": 4854
    },
    {
      "epoch": 1.1046643913538112,
      "grad_norm": 1.2643260607838762,
      "learning_rate": 1.105458290637701e-06,
      "loss": 0.04,
      "step": 4855
    },
    {
      "epoch": 1.104891922639363,
      "grad_norm": 0.6193207760495852,
      "learning_rate": 1.105401139343055e-06,
      "loss": 0.0095,
      "step": 4856
    },
    {
      "epoch": 1.1051194539249147,
      "grad_norm": 0.9678581892125716,
      "learning_rate": 1.105343978229889e-06,
      "loss": 0.0137,
      "step": 4857
    },
    {
      "epoch": 1.1053469852104665,
      "grad_norm": 1.2738862816924261,
      "learning_rate": 1.1052868072993723e-06,
      "loss": 0.0324,
      "step": 4858
    },
    {
      "epoch": 1.1055745164960182,
      "grad_norm": 2.111284557158375,
      "learning_rate": 1.1052296265526726e-06,
      "loss": 0.0436,
      "step": 4859
    },
    {
      "epoch": 1.10580204778157,
      "grad_norm": 1.7291659207116172,
      "learning_rate": 1.105172435990959e-06,
      "loss": 0.0443,
      "step": 4860
    },
    {
      "epoch": 1.1060295790671217,
      "grad_norm": 1.4583333294669347,
      "learning_rate": 1.1051152356154e-06,
      "loss": 0.0316,
      "step": 4861
    },
    {
      "epoch": 1.1062571103526735,
      "grad_norm": 0.9218920816273283,
      "learning_rate": 1.105058025427165e-06,
      "loss": 0.017,
      "step": 4862
    },
    {
      "epoch": 1.1064846416382252,
      "grad_norm": 1.2813414812852448,
      "learning_rate": 1.105000805427423e-06,
      "loss": 0.0262,
      "step": 4863
    },
    {
      "epoch": 1.106712172923777,
      "grad_norm": 1.6993734853125684,
      "learning_rate": 1.1049435756173439e-06,
      "loss": 0.0219,
      "step": 4864
    },
    {
      "epoch": 1.1069397042093287,
      "grad_norm": 0.8219715997662334,
      "learning_rate": 1.104886335998097e-06,
      "loss": 0.0166,
      "step": 4865
    },
    {
      "epoch": 1.1071672354948805,
      "grad_norm": 1.4898463042222714,
      "learning_rate": 1.104829086570852e-06,
      "loss": 0.0244,
      "step": 4866
    },
    {
      "epoch": 1.1073947667804322,
      "grad_norm": 0.8450834919239356,
      "learning_rate": 1.1047718273367794e-06,
      "loss": 0.0281,
      "step": 4867
    },
    {
      "epoch": 1.107622298065984,
      "grad_norm": 0.6448103593188148,
      "learning_rate": 1.1047145582970494e-06,
      "loss": 0.0086,
      "step": 4868
    },
    {
      "epoch": 1.1078498293515358,
      "grad_norm": 1.1979257342162584,
      "learning_rate": 1.1046572794528324e-06,
      "loss": 0.0295,
      "step": 4869
    },
    {
      "epoch": 1.1080773606370875,
      "grad_norm": 0.704626187571558,
      "learning_rate": 1.104599990805299e-06,
      "loss": 0.0141,
      "step": 4870
    },
    {
      "epoch": 1.1083048919226393,
      "grad_norm": 1.2920126084899004,
      "learning_rate": 1.1045426923556198e-06,
      "loss": 0.0226,
      "step": 4871
    },
    {
      "epoch": 1.108532423208191,
      "grad_norm": 0.9357951370023873,
      "learning_rate": 1.1044853841049668e-06,
      "loss": 0.0146,
      "step": 4872
    },
    {
      "epoch": 1.108759954493743,
      "grad_norm": 0.8290368347528208,
      "learning_rate": 1.1044280660545103e-06,
      "loss": 0.0098,
      "step": 4873
    },
    {
      "epoch": 1.1089874857792947,
      "grad_norm": 1.067183885716191,
      "learning_rate": 1.1043707382054223e-06,
      "loss": 0.0206,
      "step": 4874
    },
    {
      "epoch": 1.1092150170648465,
      "grad_norm": 0.9196445315136,
      "learning_rate": 1.1043134005588743e-06,
      "loss": 0.0196,
      "step": 4875
    },
    {
      "epoch": 1.1094425483503982,
      "grad_norm": 1.5830957213473278,
      "learning_rate": 1.1042560531160381e-06,
      "loss": 0.0426,
      "step": 4876
    },
    {
      "epoch": 1.10967007963595,
      "grad_norm": 1.0495445338165712,
      "learning_rate": 1.104198695878086e-06,
      "loss": 0.0223,
      "step": 4877
    },
    {
      "epoch": 1.1098976109215017,
      "grad_norm": 0.5938411642446392,
      "learning_rate": 1.1041413288461903e-06,
      "loss": 0.0118,
      "step": 4878
    },
    {
      "epoch": 1.1101251422070535,
      "grad_norm": 1.286777509620217,
      "learning_rate": 1.1040839520215233e-06,
      "loss": 0.0333,
      "step": 4879
    },
    {
      "epoch": 1.1103526734926052,
      "grad_norm": 0.9877268530404503,
      "learning_rate": 1.1040265654052575e-06,
      "loss": 0.0223,
      "step": 4880
    },
    {
      "epoch": 1.110580204778157,
      "grad_norm": 0.6647674925503221,
      "learning_rate": 1.103969168998566e-06,
      "loss": 0.0332,
      "step": 4881
    },
    {
      "epoch": 1.1108077360637088,
      "grad_norm": 1.2643667840897899,
      "learning_rate": 1.1039117628026222e-06,
      "loss": 0.0458,
      "step": 4882
    },
    {
      "epoch": 1.1110352673492605,
      "grad_norm": 0.9162239690822223,
      "learning_rate": 1.1038543468185988e-06,
      "loss": 0.0169,
      "step": 4883
    },
    {
      "epoch": 1.1112627986348123,
      "grad_norm": 0.8043833710259668,
      "learning_rate": 1.1037969210476696e-06,
      "loss": 0.013,
      "step": 4884
    },
    {
      "epoch": 1.111490329920364,
      "grad_norm": 0.9360911016853952,
      "learning_rate": 1.1037394854910082e-06,
      "loss": 0.0299,
      "step": 4885
    },
    {
      "epoch": 1.1117178612059158,
      "grad_norm": 0.8478138590871992,
      "learning_rate": 1.1036820401497884e-06,
      "loss": 0.0104,
      "step": 4886
    },
    {
      "epoch": 1.1119453924914675,
      "grad_norm": 0.7885738829466076,
      "learning_rate": 1.1036245850251844e-06,
      "loss": 0.0119,
      "step": 4887
    },
    {
      "epoch": 1.1121729237770193,
      "grad_norm": 0.711711942639679,
      "learning_rate": 1.1035671201183706e-06,
      "loss": 0.0167,
      "step": 4888
    },
    {
      "epoch": 1.112400455062571,
      "grad_norm": 0.6664367950082724,
      "learning_rate": 1.1035096454305213e-06,
      "loss": 0.0146,
      "step": 4889
    },
    {
      "epoch": 1.1126279863481228,
      "grad_norm": 1.5180095449935846,
      "learning_rate": 1.103452160962811e-06,
      "loss": 0.0206,
      "step": 4890
    },
    {
      "epoch": 1.1128555176336747,
      "grad_norm": 0.9314557397364934,
      "learning_rate": 1.103394666716415e-06,
      "loss": 0.0399,
      "step": 4891
    },
    {
      "epoch": 1.1130830489192265,
      "grad_norm": 0.7240617420566345,
      "learning_rate": 1.1033371626925079e-06,
      "loss": 0.0104,
      "step": 4892
    },
    {
      "epoch": 1.1133105802047782,
      "grad_norm": 1.000892668316659,
      "learning_rate": 1.1032796488922653e-06,
      "loss": 0.0186,
      "step": 4893
    },
    {
      "epoch": 1.11353811149033,
      "grad_norm": 1.0694112940902762,
      "learning_rate": 1.103222125316863e-06,
      "loss": 0.0151,
      "step": 4894
    },
    {
      "epoch": 1.1137656427758817,
      "grad_norm": 0.9676066050033223,
      "learning_rate": 1.1031645919674758e-06,
      "loss": 0.0139,
      "step": 4895
    },
    {
      "epoch": 1.1139931740614335,
      "grad_norm": 1.0428454696146314,
      "learning_rate": 1.10310704884528e-06,
      "loss": 0.0306,
      "step": 4896
    },
    {
      "epoch": 1.1142207053469853,
      "grad_norm": 1.2326650915996091,
      "learning_rate": 1.1030494959514521e-06,
      "loss": 0.0296,
      "step": 4897
    },
    {
      "epoch": 1.114448236632537,
      "grad_norm": 1.23258544073363,
      "learning_rate": 1.1029919332871678e-06,
      "loss": 0.0365,
      "step": 4898
    },
    {
      "epoch": 1.1146757679180888,
      "grad_norm": 0.9035657907747793,
      "learning_rate": 1.102934360853604e-06,
      "loss": 0.0128,
      "step": 4899
    },
    {
      "epoch": 1.1149032992036405,
      "grad_norm": 1.1444479344404774,
      "learning_rate": 1.1028767786519368e-06,
      "loss": 0.0141,
      "step": 4900
    },
    {
      "epoch": 1.1151308304891923,
      "grad_norm": 1.2429676441421305,
      "learning_rate": 1.1028191866833438e-06,
      "loss": 0.0285,
      "step": 4901
    },
    {
      "epoch": 1.115358361774744,
      "grad_norm": 0.9590079950756828,
      "learning_rate": 1.1027615849490014e-06,
      "loss": 0.0112,
      "step": 4902
    },
    {
      "epoch": 1.1155858930602958,
      "grad_norm": 1.572926999023847,
      "learning_rate": 1.1027039734500872e-06,
      "loss": 0.0369,
      "step": 4903
    },
    {
      "epoch": 1.1158134243458475,
      "grad_norm": 1.1352227453212334,
      "learning_rate": 1.102646352187779e-06,
      "loss": 0.0359,
      "step": 4904
    },
    {
      "epoch": 1.1160409556313993,
      "grad_norm": 1.1716125851334267,
      "learning_rate": 1.1025887211632538e-06,
      "loss": 0.02,
      "step": 4905
    },
    {
      "epoch": 1.116268486916951,
      "grad_norm": 1.1832800935639822,
      "learning_rate": 1.1025310803776898e-06,
      "loss": 0.0415,
      "step": 4906
    },
    {
      "epoch": 1.1164960182025028,
      "grad_norm": 1.1136610636887923,
      "learning_rate": 1.1024734298322655e-06,
      "loss": 0.0184,
      "step": 4907
    },
    {
      "epoch": 1.1167235494880545,
      "grad_norm": 1.1211312468158372,
      "learning_rate": 1.1024157695281582e-06,
      "loss": 0.029,
      "step": 4908
    },
    {
      "epoch": 1.1169510807736063,
      "grad_norm": 0.9972384903553551,
      "learning_rate": 1.1023580994665472e-06,
      "loss": 0.0201,
      "step": 4909
    },
    {
      "epoch": 1.117178612059158,
      "grad_norm": 1.7707402126553615,
      "learning_rate": 1.1023004196486108e-06,
      "loss": 0.043,
      "step": 4910
    },
    {
      "epoch": 1.11740614334471,
      "grad_norm": 1.134530398414991,
      "learning_rate": 1.102242730075528e-06,
      "loss": 0.0225,
      "step": 4911
    },
    {
      "epoch": 1.1176336746302618,
      "grad_norm": 1.627488244438378,
      "learning_rate": 1.1021850307484776e-06,
      "loss": 0.02,
      "step": 4912
    },
    {
      "epoch": 1.1178612059158135,
      "grad_norm": 1.1040646049948346,
      "learning_rate": 1.1021273216686397e-06,
      "loss": 0.0173,
      "step": 4913
    },
    {
      "epoch": 1.1180887372013653,
      "grad_norm": 1.5838051197154315,
      "learning_rate": 1.1020696028371926e-06,
      "loss": 0.024,
      "step": 4914
    },
    {
      "epoch": 1.118316268486917,
      "grad_norm": 0.8992624282859318,
      "learning_rate": 1.1020118742553166e-06,
      "loss": 0.0226,
      "step": 4915
    },
    {
      "epoch": 1.1185437997724688,
      "grad_norm": 1.3853418643898612,
      "learning_rate": 1.1019541359241917e-06,
      "loss": 0.0277,
      "step": 4916
    },
    {
      "epoch": 1.1187713310580205,
      "grad_norm": 1.0102231219601017,
      "learning_rate": 1.1018963878449976e-06,
      "loss": 0.013,
      "step": 4917
    },
    {
      "epoch": 1.1189988623435723,
      "grad_norm": 1.1790884765143759,
      "learning_rate": 1.1018386300189148e-06,
      "loss": 0.0176,
      "step": 4918
    },
    {
      "epoch": 1.119226393629124,
      "grad_norm": 0.7433554447296712,
      "learning_rate": 1.1017808624471237e-06,
      "loss": 0.0196,
      "step": 4919
    },
    {
      "epoch": 1.1194539249146758,
      "grad_norm": 0.9334325084247519,
      "learning_rate": 1.101723085130805e-06,
      "loss": 0.0241,
      "step": 4920
    },
    {
      "epoch": 1.1196814562002275,
      "grad_norm": 0.7643706032076052,
      "learning_rate": 1.1016652980711392e-06,
      "loss": 0.0138,
      "step": 4921
    },
    {
      "epoch": 1.1199089874857793,
      "grad_norm": 0.7474429513186569,
      "learning_rate": 1.1016075012693082e-06,
      "loss": 0.0266,
      "step": 4922
    },
    {
      "epoch": 1.120136518771331,
      "grad_norm": 1.1711233382681683,
      "learning_rate": 1.1015496947264923e-06,
      "loss": 0.0324,
      "step": 4923
    },
    {
      "epoch": 1.1203640500568828,
      "grad_norm": 2.534130515487931,
      "learning_rate": 1.1014918784438736e-06,
      "loss": 0.0624,
      "step": 4924
    },
    {
      "epoch": 1.1205915813424345,
      "grad_norm": 1.4712142976504,
      "learning_rate": 1.1014340524226337e-06,
      "loss": 0.0425,
      "step": 4925
    },
    {
      "epoch": 1.1208191126279863,
      "grad_norm": 0.8897734161115947,
      "learning_rate": 1.101376216663954e-06,
      "loss": 0.0146,
      "step": 4926
    },
    {
      "epoch": 1.121046643913538,
      "grad_norm": 1.6323492107077655,
      "learning_rate": 1.101318371169017e-06,
      "loss": 0.0446,
      "step": 4927
    },
    {
      "epoch": 1.1212741751990898,
      "grad_norm": 0.9646427394649012,
      "learning_rate": 1.1012605159390048e-06,
      "loss": 0.0312,
      "step": 4928
    },
    {
      "epoch": 1.1215017064846415,
      "grad_norm": 0.7920927385811001,
      "learning_rate": 1.1012026509751e-06,
      "loss": 0.0194,
      "step": 4929
    },
    {
      "epoch": 1.1217292377701935,
      "grad_norm": 1.1574113409003755,
      "learning_rate": 1.1011447762784849e-06,
      "loss": 0.0237,
      "step": 4930
    },
    {
      "epoch": 1.1219567690557453,
      "grad_norm": 1.2517329969278503,
      "learning_rate": 1.1010868918503429e-06,
      "loss": 0.0207,
      "step": 4931
    },
    {
      "epoch": 1.122184300341297,
      "grad_norm": 0.8071040660470671,
      "learning_rate": 1.1010289976918565e-06,
      "loss": 0.0207,
      "step": 4932
    },
    {
      "epoch": 1.1224118316268488,
      "grad_norm": 0.9373790639087559,
      "learning_rate": 1.1009710938042093e-06,
      "loss": 0.0153,
      "step": 4933
    },
    {
      "epoch": 1.1226393629124005,
      "grad_norm": 1.2887447933260439,
      "learning_rate": 1.1009131801885848e-06,
      "loss": 0.0239,
      "step": 4934
    },
    {
      "epoch": 1.1228668941979523,
      "grad_norm": 1.40441656204075,
      "learning_rate": 1.100855256846166e-06,
      "loss": 0.0343,
      "step": 4935
    },
    {
      "epoch": 1.123094425483504,
      "grad_norm": 0.689175568925837,
      "learning_rate": 1.1007973237781377e-06,
      "loss": 0.0148,
      "step": 4936
    },
    {
      "epoch": 1.1233219567690558,
      "grad_norm": 1.1787781392551855,
      "learning_rate": 1.1007393809856834e-06,
      "loss": 0.0286,
      "step": 4937
    },
    {
      "epoch": 1.1235494880546075,
      "grad_norm": 1.0271974875036538,
      "learning_rate": 1.1006814284699873e-06,
      "loss": 0.0373,
      "step": 4938
    },
    {
      "epoch": 1.1237770193401593,
      "grad_norm": 0.8297711019866251,
      "learning_rate": 1.100623466232234e-06,
      "loss": 0.0261,
      "step": 4939
    },
    {
      "epoch": 1.124004550625711,
      "grad_norm": 1.0858267777295185,
      "learning_rate": 1.1005654942736082e-06,
      "loss": 0.0261,
      "step": 4940
    },
    {
      "epoch": 1.1242320819112628,
      "grad_norm": 0.6790255085030746,
      "learning_rate": 1.1005075125952946e-06,
      "loss": 0.0156,
      "step": 4941
    },
    {
      "epoch": 1.1244596131968145,
      "grad_norm": 1.0737443834894482,
      "learning_rate": 1.1004495211984783e-06,
      "loss": 0.0342,
      "step": 4942
    },
    {
      "epoch": 1.1246871444823663,
      "grad_norm": 1.1759005758316916,
      "learning_rate": 1.1003915200843446e-06,
      "loss": 0.0296,
      "step": 4943
    },
    {
      "epoch": 1.124914675767918,
      "grad_norm": 1.4502960478357807,
      "learning_rate": 1.1003335092540787e-06,
      "loss": 0.0338,
      "step": 4944
    },
    {
      "epoch": 1.1251422070534698,
      "grad_norm": 1.0232031902807623,
      "learning_rate": 1.1002754887088665e-06,
      "loss": 0.0275,
      "step": 4945
    },
    {
      "epoch": 1.1253697383390215,
      "grad_norm": 1.1276836760259192,
      "learning_rate": 1.1002174584498938e-06,
      "loss": 0.018,
      "step": 4946
    },
    {
      "epoch": 1.1255972696245733,
      "grad_norm": 1.121912066969083,
      "learning_rate": 1.1001594184783464e-06,
      "loss": 0.0224,
      "step": 4947
    },
    {
      "epoch": 1.125824800910125,
      "grad_norm": 1.1026078266089805,
      "learning_rate": 1.1001013687954109e-06,
      "loss": 0.0265,
      "step": 4948
    },
    {
      "epoch": 1.1260523321956768,
      "grad_norm": 0.6996142941788891,
      "learning_rate": 1.1000433094022735e-06,
      "loss": 0.0204,
      "step": 4949
    },
    {
      "epoch": 1.1262798634812285,
      "grad_norm": 1.022975678784142,
      "learning_rate": 1.0999852403001208e-06,
      "loss": 0.0189,
      "step": 4950
    },
    {
      "epoch": 1.1265073947667805,
      "grad_norm": 1.2687033993187262,
      "learning_rate": 1.0999271614901396e-06,
      "loss": 0.0141,
      "step": 4951
    },
    {
      "epoch": 1.1267349260523323,
      "grad_norm": 1.799956791253596,
      "learning_rate": 1.099869072973517e-06,
      "loss": 0.0146,
      "step": 4952
    },
    {
      "epoch": 1.126962457337884,
      "grad_norm": 1.5492585986994671,
      "learning_rate": 1.0998109747514404e-06,
      "loss": 0.0447,
      "step": 4953
    },
    {
      "epoch": 1.1271899886234358,
      "grad_norm": 0.7598908512501374,
      "learning_rate": 1.099752866825097e-06,
      "loss": 0.0166,
      "step": 4954
    },
    {
      "epoch": 1.1274175199089875,
      "grad_norm": 0.6362839552889372,
      "learning_rate": 1.0996947491956745e-06,
      "loss": 0.0113,
      "step": 4955
    },
    {
      "epoch": 1.1276450511945393,
      "grad_norm": 1.0336058491900288,
      "learning_rate": 1.0996366218643607e-06,
      "loss": 0.0156,
      "step": 4956
    },
    {
      "epoch": 1.127872582480091,
      "grad_norm": 0.7457309100980003,
      "learning_rate": 1.0995784848323434e-06,
      "loss": 0.0151,
      "step": 4957
    },
    {
      "epoch": 1.1281001137656428,
      "grad_norm": 1.649804543332714,
      "learning_rate": 1.0995203381008112e-06,
      "loss": 0.0347,
      "step": 4958
    },
    {
      "epoch": 1.1283276450511945,
      "grad_norm": 1.2533333259409183,
      "learning_rate": 1.099462181670952e-06,
      "loss": 0.0288,
      "step": 4959
    },
    {
      "epoch": 1.1285551763367463,
      "grad_norm": 1.1428064969183531,
      "learning_rate": 1.0994040155439553e-06,
      "loss": 0.0263,
      "step": 4960
    },
    {
      "epoch": 1.128782707622298,
      "grad_norm": 0.7012991923657113,
      "learning_rate": 1.0993458397210092e-06,
      "loss": 0.0108,
      "step": 4961
    },
    {
      "epoch": 1.1290102389078498,
      "grad_norm": 1.319347853084701,
      "learning_rate": 1.0992876542033026e-06,
      "loss": 0.0232,
      "step": 4962
    },
    {
      "epoch": 1.1292377701934015,
      "grad_norm": 0.8983600851477538,
      "learning_rate": 1.0992294589920252e-06,
      "loss": 0.015,
      "step": 4963
    },
    {
      "epoch": 1.1294653014789533,
      "grad_norm": 0.6582052508958808,
      "learning_rate": 1.099171254088366e-06,
      "loss": 0.0129,
      "step": 4964
    },
    {
      "epoch": 1.129692832764505,
      "grad_norm": 2.0776942729185306,
      "learning_rate": 1.0991130394935148e-06,
      "loss": 0.0128,
      "step": 4965
    },
    {
      "epoch": 1.1299203640500568,
      "grad_norm": 1.3980552891281348,
      "learning_rate": 1.0990548152086616e-06,
      "loss": 0.0158,
      "step": 4966
    },
    {
      "epoch": 1.1301478953356086,
      "grad_norm": 1.399217668963092,
      "learning_rate": 1.098996581234996e-06,
      "loss": 0.0214,
      "step": 4967
    },
    {
      "epoch": 1.1303754266211605,
      "grad_norm": 1.0723453956148017,
      "learning_rate": 1.0989383375737081e-06,
      "loss": 0.0124,
      "step": 4968
    },
    {
      "epoch": 1.1306029579067123,
      "grad_norm": 1.1545644366726804,
      "learning_rate": 1.098880084225989e-06,
      "loss": 0.0412,
      "step": 4969
    },
    {
      "epoch": 1.130830489192264,
      "grad_norm": 0.942557146311076,
      "learning_rate": 1.0988218211930285e-06,
      "loss": 0.0201,
      "step": 4970
    },
    {
      "epoch": 1.1310580204778158,
      "grad_norm": 1.0043398319888264,
      "learning_rate": 1.0987635484760178e-06,
      "loss": 0.0198,
      "step": 4971
    },
    {
      "epoch": 1.1312855517633675,
      "grad_norm": 1.5698009312279446,
      "learning_rate": 1.098705266076148e-06,
      "loss": 0.0258,
      "step": 4972
    },
    {
      "epoch": 1.1315130830489193,
      "grad_norm": 1.4423264895380257,
      "learning_rate": 1.0986469739946102e-06,
      "loss": 0.0325,
      "step": 4973
    },
    {
      "epoch": 1.131740614334471,
      "grad_norm": 1.2136974558872335,
      "learning_rate": 1.0985886722325954e-06,
      "loss": 0.0217,
      "step": 4974
    },
    {
      "epoch": 1.1319681456200228,
      "grad_norm": 1.0091129092645157,
      "learning_rate": 1.0985303607912956e-06,
      "loss": 0.0134,
      "step": 4975
    },
    {
      "epoch": 1.1321956769055745,
      "grad_norm": 0.9544088452985304,
      "learning_rate": 1.0984720396719024e-06,
      "loss": 0.0243,
      "step": 4976
    },
    {
      "epoch": 1.1324232081911263,
      "grad_norm": 1.03400426394427,
      "learning_rate": 1.098413708875608e-06,
      "loss": 0.0165,
      "step": 4977
    },
    {
      "epoch": 1.132650739476678,
      "grad_norm": 0.9584699961425377,
      "learning_rate": 1.098355368403604e-06,
      "loss": 0.0159,
      "step": 4978
    },
    {
      "epoch": 1.1328782707622298,
      "grad_norm": 1.1633397537719241,
      "learning_rate": 1.0982970182570837e-06,
      "loss": 0.027,
      "step": 4979
    },
    {
      "epoch": 1.1331058020477816,
      "grad_norm": 0.6150701132014431,
      "learning_rate": 1.098238658437239e-06,
      "loss": 0.0134,
      "step": 4980
    },
    {
      "epoch": 1.1333333333333333,
      "grad_norm": 0.9628332044868487,
      "learning_rate": 1.0981802889452627e-06,
      "loss": 0.0244,
      "step": 4981
    },
    {
      "epoch": 1.133560864618885,
      "grad_norm": 0.8522049310117307,
      "learning_rate": 1.0981219097823479e-06,
      "loss": 0.0106,
      "step": 4982
    },
    {
      "epoch": 1.1337883959044368,
      "grad_norm": 0.9473163604101656,
      "learning_rate": 1.0980635209496878e-06,
      "loss": 0.024,
      "step": 4983
    },
    {
      "epoch": 1.1340159271899886,
      "grad_norm": 0.8002876838010067,
      "learning_rate": 1.0980051224484756e-06,
      "loss": 0.0083,
      "step": 4984
    },
    {
      "epoch": 1.1342434584755403,
      "grad_norm": 1.275340108026526,
      "learning_rate": 1.0979467142799052e-06,
      "loss": 0.0425,
      "step": 4985
    },
    {
      "epoch": 1.134470989761092,
      "grad_norm": 1.1242771585253066,
      "learning_rate": 1.0978882964451698e-06,
      "loss": 0.0235,
      "step": 4986
    },
    {
      "epoch": 1.1346985210466438,
      "grad_norm": 0.8316998784778787,
      "learning_rate": 1.097829868945464e-06,
      "loss": 0.0138,
      "step": 4987
    },
    {
      "epoch": 1.1349260523321956,
      "grad_norm": 1.2826664766461733,
      "learning_rate": 1.0977714317819812e-06,
      "loss": 0.0148,
      "step": 4988
    },
    {
      "epoch": 1.1351535836177473,
      "grad_norm": 1.7211760416844804,
      "learning_rate": 1.0977129849559165e-06,
      "loss": 0.0336,
      "step": 4989
    },
    {
      "epoch": 1.1353811149032993,
      "grad_norm": 0.8143435710252414,
      "learning_rate": 1.0976545284684642e-06,
      "loss": 0.0066,
      "step": 4990
    },
    {
      "epoch": 1.135608646188851,
      "grad_norm": 1.0342548252924428,
      "learning_rate": 1.0975960623208188e-06,
      "loss": 0.0162,
      "step": 4991
    },
    {
      "epoch": 1.1358361774744028,
      "grad_norm": 1.7590039011675342,
      "learning_rate": 1.0975375865141753e-06,
      "loss": 0.0362,
      "step": 4992
    },
    {
      "epoch": 1.1360637087599545,
      "grad_norm": 1.6822978128433002,
      "learning_rate": 1.097479101049729e-06,
      "loss": 0.0185,
      "step": 4993
    },
    {
      "epoch": 1.1362912400455063,
      "grad_norm": 1.2344613037091736,
      "learning_rate": 1.0974206059286752e-06,
      "loss": 0.0374,
      "step": 4994
    },
    {
      "epoch": 1.136518771331058,
      "grad_norm": 0.5280046364865614,
      "learning_rate": 1.0973621011522096e-06,
      "loss": 0.0084,
      "step": 4995
    },
    {
      "epoch": 1.1367463026166098,
      "grad_norm": 1.9687955658616516,
      "learning_rate": 1.0973035867215276e-06,
      "loss": 0.0336,
      "step": 4996
    },
    {
      "epoch": 1.1369738339021616,
      "grad_norm": 1.022624248487459,
      "learning_rate": 1.0972450626378254e-06,
      "loss": 0.0225,
      "step": 4997
    },
    {
      "epoch": 1.1372013651877133,
      "grad_norm": 0.48108692573634865,
      "learning_rate": 1.0971865289022988e-06,
      "loss": 0.008,
      "step": 4998
    },
    {
      "epoch": 1.137428896473265,
      "grad_norm": 0.9799413189160909,
      "learning_rate": 1.0971279855161442e-06,
      "loss": 0.0394,
      "step": 4999
    },
    {
      "epoch": 1.1376564277588168,
      "grad_norm": 1.466751462689922,
      "learning_rate": 1.0970694324805586e-06,
      "loss": 0.0167,
      "step": 5000
    },
    {
      "epoch": 1.1378839590443686,
      "grad_norm": 1.5693112940714422,
      "learning_rate": 1.0970108697967382e-06,
      "loss": 0.0325,
      "step": 5001
    },
    {
      "epoch": 1.1381114903299203,
      "grad_norm": 0.9255284204543452,
      "learning_rate": 1.09695229746588e-06,
      "loss": 0.0362,
      "step": 5002
    },
    {
      "epoch": 1.138339021615472,
      "grad_norm": 1.7087407221932875,
      "learning_rate": 1.0968937154891812e-06,
      "loss": 0.052,
      "step": 5003
    },
    {
      "epoch": 1.1385665529010238,
      "grad_norm": 2.428840915012362,
      "learning_rate": 1.096835123867839e-06,
      "loss": 0.0366,
      "step": 5004
    },
    {
      "epoch": 1.1387940841865756,
      "grad_norm": 0.6595273191874941,
      "learning_rate": 1.0967765226030512e-06,
      "loss": 0.0215,
      "step": 5005
    },
    {
      "epoch": 1.1390216154721273,
      "grad_norm": 1.2073236201929451,
      "learning_rate": 1.0967179116960153e-06,
      "loss": 0.0431,
      "step": 5006
    },
    {
      "epoch": 1.1392491467576793,
      "grad_norm": 1.517298131220002,
      "learning_rate": 1.096659291147929e-06,
      "loss": 0.0493,
      "step": 5007
    },
    {
      "epoch": 1.139476678043231,
      "grad_norm": 0.8459583474402816,
      "learning_rate": 1.0966006609599908e-06,
      "loss": 0.0161,
      "step": 5008
    },
    {
      "epoch": 1.1397042093287828,
      "grad_norm": 0.9402719162645292,
      "learning_rate": 1.0965420211333984e-06,
      "loss": 0.0199,
      "step": 5009
    },
    {
      "epoch": 1.1399317406143346,
      "grad_norm": 0.8824336019611823,
      "learning_rate": 1.0964833716693512e-06,
      "loss": 0.0236,
      "step": 5010
    },
    {
      "epoch": 1.1401592718998863,
      "grad_norm": 0.9061765730464424,
      "learning_rate": 1.096424712569047e-06,
      "loss": 0.0144,
      "step": 5011
    },
    {
      "epoch": 1.140386803185438,
      "grad_norm": 1.1463154922222984,
      "learning_rate": 1.0963660438336851e-06,
      "loss": 0.0405,
      "step": 5012
    },
    {
      "epoch": 1.1406143344709898,
      "grad_norm": 0.6924366133128478,
      "learning_rate": 1.0963073654644645e-06,
      "loss": 0.0139,
      "step": 5013
    },
    {
      "epoch": 1.1408418657565416,
      "grad_norm": 0.9228757161234781,
      "learning_rate": 1.0962486774625847e-06,
      "loss": 0.0227,
      "step": 5014
    },
    {
      "epoch": 1.1410693970420933,
      "grad_norm": 0.9665478631542946,
      "learning_rate": 1.096189979829245e-06,
      "loss": 0.0273,
      "step": 5015
    },
    {
      "epoch": 1.141296928327645,
      "grad_norm": 0.9370551975836519,
      "learning_rate": 1.096131272565645e-06,
      "loss": 0.0201,
      "step": 5016
    },
    {
      "epoch": 1.1415244596131968,
      "grad_norm": 1.0359990365775988,
      "learning_rate": 1.0960725556729845e-06,
      "loss": 0.0188,
      "step": 5017
    },
    {
      "epoch": 1.1417519908987486,
      "grad_norm": 0.6081258142385076,
      "learning_rate": 1.0960138291524637e-06,
      "loss": 0.0123,
      "step": 5018
    },
    {
      "epoch": 1.1419795221843003,
      "grad_norm": 0.8151694196582907,
      "learning_rate": 1.095955093005283e-06,
      "loss": 0.0178,
      "step": 5019
    },
    {
      "epoch": 1.142207053469852,
      "grad_norm": 0.972811628969212,
      "learning_rate": 1.0958963472326426e-06,
      "loss": 0.0114,
      "step": 5020
    },
    {
      "epoch": 1.1424345847554038,
      "grad_norm": 0.7206858563005965,
      "learning_rate": 1.0958375918357433e-06,
      "loss": 0.011,
      "step": 5021
    },
    {
      "epoch": 1.1426621160409556,
      "grad_norm": 0.8388894087062686,
      "learning_rate": 1.095778826815786e-06,
      "loss": 0.0214,
      "step": 5022
    },
    {
      "epoch": 1.1428896473265073,
      "grad_norm": 1.4288807810887136,
      "learning_rate": 1.0957200521739715e-06,
      "loss": 0.0315,
      "step": 5023
    },
    {
      "epoch": 1.143117178612059,
      "grad_norm": 0.9634756984934681,
      "learning_rate": 1.0956612679115012e-06,
      "loss": 0.0338,
      "step": 5024
    },
    {
      "epoch": 1.1433447098976108,
      "grad_norm": 1.2246259764288705,
      "learning_rate": 1.0956024740295767e-06,
      "loss": 0.0339,
      "step": 5025
    },
    {
      "epoch": 1.1435722411831626,
      "grad_norm": 0.9727990066152786,
      "learning_rate": 1.0955436705293996e-06,
      "loss": 0.0186,
      "step": 5026
    },
    {
      "epoch": 1.1437997724687143,
      "grad_norm": 1.4271212460831064,
      "learning_rate": 1.0954848574121715e-06,
      "loss": 0.0407,
      "step": 5027
    },
    {
      "epoch": 1.144027303754266,
      "grad_norm": 1.1642512101338942,
      "learning_rate": 1.0954260346790944e-06,
      "loss": 0.0305,
      "step": 5028
    },
    {
      "epoch": 1.144254835039818,
      "grad_norm": 3.382254274133764,
      "learning_rate": 1.0953672023313709e-06,
      "loss": 0.0462,
      "step": 5029
    },
    {
      "epoch": 1.1444823663253698,
      "grad_norm": 1.3043659225114033,
      "learning_rate": 1.0953083603702031e-06,
      "loss": 0.0269,
      "step": 5030
    },
    {
      "epoch": 1.1447098976109216,
      "grad_norm": 0.8941373741234482,
      "learning_rate": 1.0952495087967939e-06,
      "loss": 0.01,
      "step": 5031
    },
    {
      "epoch": 1.1449374288964733,
      "grad_norm": 1.0345429364832142,
      "learning_rate": 1.095190647612346e-06,
      "loss": 0.0313,
      "step": 5032
    },
    {
      "epoch": 1.145164960182025,
      "grad_norm": 0.9469945046545952,
      "learning_rate": 1.0951317768180623e-06,
      "loss": 0.0185,
      "step": 5033
    },
    {
      "epoch": 1.1453924914675768,
      "grad_norm": 1.5417022710277841,
      "learning_rate": 1.0950728964151457e-06,
      "loss": 0.0477,
      "step": 5034
    },
    {
      "epoch": 1.1456200227531286,
      "grad_norm": 1.2090565974916883,
      "learning_rate": 1.0950140064048005e-06,
      "loss": 0.0304,
      "step": 5035
    },
    {
      "epoch": 1.1458475540386803,
      "grad_norm": 1.6222929147315774,
      "learning_rate": 1.0949551067882297e-06,
      "loss": 0.0449,
      "step": 5036
    },
    {
      "epoch": 1.146075085324232,
      "grad_norm": 0.8458047187793855,
      "learning_rate": 1.094896197566637e-06,
      "loss": 0.0177,
      "step": 5037
    },
    {
      "epoch": 1.1463026166097838,
      "grad_norm": 0.8642635965418713,
      "learning_rate": 1.0948372787412267e-06,
      "loss": 0.019,
      "step": 5038
    },
    {
      "epoch": 1.1465301478953356,
      "grad_norm": 1.6962052412561806,
      "learning_rate": 1.094778350313203e-06,
      "loss": 0.0388,
      "step": 5039
    },
    {
      "epoch": 1.1467576791808873,
      "grad_norm": 1.2082274113676141,
      "learning_rate": 1.09471941228377e-06,
      "loss": 0.0443,
      "step": 5040
    },
    {
      "epoch": 1.146985210466439,
      "grad_norm": 1.551649129101969,
      "learning_rate": 1.0946604646541327e-06,
      "loss": 0.0458,
      "step": 5041
    },
    {
      "epoch": 1.1472127417519908,
      "grad_norm": 0.6322582451330618,
      "learning_rate": 1.0946015074254957e-06,
      "loss": 0.0063,
      "step": 5042
    },
    {
      "epoch": 1.1474402730375426,
      "grad_norm": 1.2390695869307191,
      "learning_rate": 1.0945425405990636e-06,
      "loss": 0.0438,
      "step": 5043
    },
    {
      "epoch": 1.1476678043230943,
      "grad_norm": 1.171113561669916,
      "learning_rate": 1.094483564176042e-06,
      "loss": 0.0304,
      "step": 5044
    },
    {
      "epoch": 1.147895335608646,
      "grad_norm": 1.111705225220342,
      "learning_rate": 1.0944245781576363e-06,
      "loss": 0.0199,
      "step": 5045
    },
    {
      "epoch": 1.148122866894198,
      "grad_norm": 0.7788906673524727,
      "learning_rate": 1.0943655825450517e-06,
      "loss": 0.017,
      "step": 5046
    },
    {
      "epoch": 1.1483503981797498,
      "grad_norm": 1.37808766490078,
      "learning_rate": 1.0943065773394943e-06,
      "loss": 0.027,
      "step": 5047
    },
    {
      "epoch": 1.1485779294653016,
      "grad_norm": 1.0016121975434131,
      "learning_rate": 1.0942475625421701e-06,
      "loss": 0.0217,
      "step": 5048
    },
    {
      "epoch": 1.1488054607508533,
      "grad_norm": 1.5673100085633027,
      "learning_rate": 1.094188538154285e-06,
      "loss": 0.039,
      "step": 5049
    },
    {
      "epoch": 1.149032992036405,
      "grad_norm": 1.1060010064378178,
      "learning_rate": 1.0941295041770453e-06,
      "loss": 0.0158,
      "step": 5050
    },
    {
      "epoch": 1.1492605233219568,
      "grad_norm": 1.1961653202374867,
      "learning_rate": 1.0940704606116578e-06,
      "loss": 0.0197,
      "step": 5051
    },
    {
      "epoch": 1.1494880546075086,
      "grad_norm": 0.9641941727834898,
      "learning_rate": 1.0940114074593292e-06,
      "loss": 0.0141,
      "step": 5052
    },
    {
      "epoch": 1.1497155858930603,
      "grad_norm": 0.6461548194155666,
      "learning_rate": 1.0939523447212665e-06,
      "loss": 0.0074,
      "step": 5053
    },
    {
      "epoch": 1.149943117178612,
      "grad_norm": 1.125395408330258,
      "learning_rate": 1.0938932723986766e-06,
      "loss": 0.0316,
      "step": 5054
    },
    {
      "epoch": 1.1501706484641638,
      "grad_norm": 0.6999270740525377,
      "learning_rate": 1.0938341904927669e-06,
      "loss": 0.0115,
      "step": 5055
    },
    {
      "epoch": 1.1503981797497156,
      "grad_norm": 0.824514611548941,
      "learning_rate": 1.093775099004745e-06,
      "loss": 0.0146,
      "step": 5056
    },
    {
      "epoch": 1.1506257110352673,
      "grad_norm": 1.2958988293911842,
      "learning_rate": 1.0937159979358186e-06,
      "loss": 0.0285,
      "step": 5057
    },
    {
      "epoch": 1.150853242320819,
      "grad_norm": 0.5661057032643344,
      "learning_rate": 1.0936568872871958e-06,
      "loss": 0.0085,
      "step": 5058
    },
    {
      "epoch": 1.1510807736063708,
      "grad_norm": 0.8911202539112528,
      "learning_rate": 1.0935977670600843e-06,
      "loss": 0.0249,
      "step": 5059
    },
    {
      "epoch": 1.1513083048919226,
      "grad_norm": 0.6119358774484015,
      "learning_rate": 1.0935386372556928e-06,
      "loss": 0.0081,
      "step": 5060
    },
    {
      "epoch": 1.1515358361774743,
      "grad_norm": 1.1732745638400495,
      "learning_rate": 1.0934794978752295e-06,
      "loss": 0.0304,
      "step": 5061
    },
    {
      "epoch": 1.151763367463026,
      "grad_norm": 1.4610976896610823,
      "learning_rate": 1.0934203489199033e-06,
      "loss": 0.026,
      "step": 5062
    },
    {
      "epoch": 1.1519908987485779,
      "grad_norm": 0.8751176409958491,
      "learning_rate": 1.093361190390923e-06,
      "loss": 0.0129,
      "step": 5063
    },
    {
      "epoch": 1.1522184300341296,
      "grad_norm": 0.9095379255951723,
      "learning_rate": 1.0933020222894978e-06,
      "loss": 0.0189,
      "step": 5064
    },
    {
      "epoch": 1.1524459613196814,
      "grad_norm": 0.886509655802644,
      "learning_rate": 1.0932428446168369e-06,
      "loss": 0.0094,
      "step": 5065
    },
    {
      "epoch": 1.152673492605233,
      "grad_norm": 1.0165308626390206,
      "learning_rate": 1.0931836573741498e-06,
      "loss": 0.0194,
      "step": 5066
    },
    {
      "epoch": 1.1529010238907849,
      "grad_norm": 2.2862767542657307,
      "learning_rate": 1.093124460562646e-06,
      "loss": 0.03,
      "step": 5067
    },
    {
      "epoch": 1.1531285551763368,
      "grad_norm": 3.0841316980515,
      "learning_rate": 1.0930652541835357e-06,
      "loss": 0.1031,
      "step": 5068
    },
    {
      "epoch": 1.1533560864618886,
      "grad_norm": 2.057276950609037,
      "learning_rate": 1.093006038238029e-06,
      "loss": 0.0498,
      "step": 5069
    },
    {
      "epoch": 1.1535836177474403,
      "grad_norm": 0.8535534714067085,
      "learning_rate": 1.0929468127273357e-06,
      "loss": 0.0262,
      "step": 5070
    },
    {
      "epoch": 1.153811149032992,
      "grad_norm": 0.7487264216032792,
      "learning_rate": 1.0928875776526667e-06,
      "loss": 0.0187,
      "step": 5071
    },
    {
      "epoch": 1.1540386803185438,
      "grad_norm": 1.011442354116883,
      "learning_rate": 1.0928283330152325e-06,
      "loss": 0.0237,
      "step": 5072
    },
    {
      "epoch": 1.1542662116040956,
      "grad_norm": 1.2385558290850083,
      "learning_rate": 1.092769078816244e-06,
      "loss": 0.0576,
      "step": 5073
    },
    {
      "epoch": 1.1544937428896473,
      "grad_norm": 1.2917165551359526,
      "learning_rate": 1.092709815056912e-06,
      "loss": 0.0211,
      "step": 5074
    },
    {
      "epoch": 1.154721274175199,
      "grad_norm": 1.0590341552017861,
      "learning_rate": 1.0926505417384482e-06,
      "loss": 0.0303,
      "step": 5075
    },
    {
      "epoch": 1.1549488054607508,
      "grad_norm": 0.9189460734693743,
      "learning_rate": 1.0925912588620637e-06,
      "loss": 0.0147,
      "step": 5076
    },
    {
      "epoch": 1.1551763367463026,
      "grad_norm": 0.998558466002832,
      "learning_rate": 1.0925319664289703e-06,
      "loss": 0.0219,
      "step": 5077
    },
    {
      "epoch": 1.1554038680318544,
      "grad_norm": 0.7945299298414331,
      "learning_rate": 1.0924726644403797e-06,
      "loss": 0.0176,
      "step": 5078
    },
    {
      "epoch": 1.155631399317406,
      "grad_norm": 1.3566259925414048,
      "learning_rate": 1.0924133528975039e-06,
      "loss": 0.0295,
      "step": 5079
    },
    {
      "epoch": 1.1558589306029579,
      "grad_norm": 0.8074114789352509,
      "learning_rate": 1.0923540318015552e-06,
      "loss": 0.022,
      "step": 5080
    },
    {
      "epoch": 1.1560864618885096,
      "grad_norm": 1.3411051591390433,
      "learning_rate": 1.092294701153746e-06,
      "loss": 0.029,
      "step": 5081
    },
    {
      "epoch": 1.1563139931740614,
      "grad_norm": 1.423080211058415,
      "learning_rate": 1.092235360955289e-06,
      "loss": 0.0503,
      "step": 5082
    },
    {
      "epoch": 1.1565415244596131,
      "grad_norm": 0.8213852817117168,
      "learning_rate": 1.092176011207397e-06,
      "loss": 0.021,
      "step": 5083
    },
    {
      "epoch": 1.1567690557451649,
      "grad_norm": 1.2933428126076256,
      "learning_rate": 1.0921166519112828e-06,
      "loss": 0.0442,
      "step": 5084
    },
    {
      "epoch": 1.1569965870307168,
      "grad_norm": 0.5857327857396418,
      "learning_rate": 1.0920572830681597e-06,
      "loss": 0.0107,
      "step": 5085
    },
    {
      "epoch": 1.1572241183162686,
      "grad_norm": 1.0152274364913478,
      "learning_rate": 1.0919979046792411e-06,
      "loss": 0.021,
      "step": 5086
    },
    {
      "epoch": 1.1574516496018203,
      "grad_norm": 1.6197058508435338,
      "learning_rate": 1.0919385167457408e-06,
      "loss": 0.0317,
      "step": 5087
    },
    {
      "epoch": 1.157679180887372,
      "grad_norm": 0.8321181279990257,
      "learning_rate": 1.0918791192688722e-06,
      "loss": 0.0083,
      "step": 5088
    },
    {
      "epoch": 1.1579067121729238,
      "grad_norm": 1.1078885245620338,
      "learning_rate": 1.0918197122498495e-06,
      "loss": 0.0417,
      "step": 5089
    },
    {
      "epoch": 1.1581342434584756,
      "grad_norm": 0.8666450018016355,
      "learning_rate": 1.0917602956898867e-06,
      "loss": 0.0235,
      "step": 5090
    },
    {
      "epoch": 1.1583617747440274,
      "grad_norm": 0.9389403634525508,
      "learning_rate": 1.0917008695901985e-06,
      "loss": 0.0161,
      "step": 5091
    },
    {
      "epoch": 1.158589306029579,
      "grad_norm": 1.6343340981795507,
      "learning_rate": 1.091641433951999e-06,
      "loss": 0.0589,
      "step": 5092
    },
    {
      "epoch": 1.1588168373151309,
      "grad_norm": 1.0221675248406095,
      "learning_rate": 1.0915819887765034e-06,
      "loss": 0.0209,
      "step": 5093
    },
    {
      "epoch": 1.1590443686006826,
      "grad_norm": 1.2783954239375135,
      "learning_rate": 1.0915225340649264e-06,
      "loss": 0.0314,
      "step": 5094
    },
    {
      "epoch": 1.1592718998862344,
      "grad_norm": 1.0876136046711122,
      "learning_rate": 1.091463069818483e-06,
      "loss": 0.0183,
      "step": 5095
    },
    {
      "epoch": 1.159499431171786,
      "grad_norm": 0.92935249083099,
      "learning_rate": 1.091403596038389e-06,
      "loss": 0.0284,
      "step": 5096
    },
    {
      "epoch": 1.1597269624573379,
      "grad_norm": 1.532892418760994,
      "learning_rate": 1.0913441127258596e-06,
      "loss": 0.0229,
      "step": 5097
    },
    {
      "epoch": 1.1599544937428896,
      "grad_norm": 1.0034643002001107,
      "learning_rate": 1.0912846198821105e-06,
      "loss": 0.0176,
      "step": 5098
    },
    {
      "epoch": 1.1601820250284414,
      "grad_norm": 1.1190616568728453,
      "learning_rate": 1.091225117508358e-06,
      "loss": 0.0201,
      "step": 5099
    },
    {
      "epoch": 1.1604095563139931,
      "grad_norm": 0.7869200780957125,
      "learning_rate": 1.0911656056058175e-06,
      "loss": 0.0085,
      "step": 5100
    },
    {
      "epoch": 1.1606370875995449,
      "grad_norm": 0.7075180696463311,
      "learning_rate": 1.0911060841757063e-06,
      "loss": 0.0143,
      "step": 5101
    },
    {
      "epoch": 1.1608646188850966,
      "grad_norm": 0.6587063658622069,
      "learning_rate": 1.09104655321924e-06,
      "loss": 0.0174,
      "step": 5102
    },
    {
      "epoch": 1.1610921501706484,
      "grad_norm": 1.2851325080822802,
      "learning_rate": 1.0909870127376358e-06,
      "loss": 0.0347,
      "step": 5103
    },
    {
      "epoch": 1.1613196814562001,
      "grad_norm": 1.4645102454670038,
      "learning_rate": 1.0909274627321106e-06,
      "loss": 0.0412,
      "step": 5104
    },
    {
      "epoch": 1.1615472127417519,
      "grad_norm": 1.983767386413822,
      "learning_rate": 1.090867903203881e-06,
      "loss": 0.0224,
      "step": 5105
    },
    {
      "epoch": 1.1617747440273036,
      "grad_norm": 1.2382132521135707,
      "learning_rate": 1.090808334154165e-06,
      "loss": 0.0172,
      "step": 5106
    },
    {
      "epoch": 1.1620022753128556,
      "grad_norm": 0.8691675756314506,
      "learning_rate": 1.0907487555841797e-06,
      "loss": 0.0158,
      "step": 5107
    },
    {
      "epoch": 1.1622298065984074,
      "grad_norm": 2.1283896982776853,
      "learning_rate": 1.0906891674951426e-06,
      "loss": 0.0345,
      "step": 5108
    },
    {
      "epoch": 1.162457337883959,
      "grad_norm": 1.6543688130648284,
      "learning_rate": 1.090629569888272e-06,
      "loss": 0.0491,
      "step": 5109
    },
    {
      "epoch": 1.1626848691695109,
      "grad_norm": 0.6293933164268644,
      "learning_rate": 1.0905699627647857e-06,
      "loss": 0.0094,
      "step": 5110
    },
    {
      "epoch": 1.1629124004550626,
      "grad_norm": 1.6953242717121608,
      "learning_rate": 1.090510346125902e-06,
      "loss": 0.0238,
      "step": 5111
    },
    {
      "epoch": 1.1631399317406144,
      "grad_norm": 1.159765223167702,
      "learning_rate": 1.0904507199728392e-06,
      "loss": 0.0406,
      "step": 5112
    },
    {
      "epoch": 1.1633674630261661,
      "grad_norm": 1.3015729382676753,
      "learning_rate": 1.0903910843068163e-06,
      "loss": 0.0318,
      "step": 5113
    },
    {
      "epoch": 1.1635949943117179,
      "grad_norm": 0.8775703929826588,
      "learning_rate": 1.090331439129052e-06,
      "loss": 0.0087,
      "step": 5114
    },
    {
      "epoch": 1.1638225255972696,
      "grad_norm": 2.0908466705122963,
      "learning_rate": 1.0902717844407651e-06,
      "loss": 0.0187,
      "step": 5115
    },
    {
      "epoch": 1.1640500568828214,
      "grad_norm": 0.8211627405295623,
      "learning_rate": 1.0902121202431754e-06,
      "loss": 0.0214,
      "step": 5116
    },
    {
      "epoch": 1.1642775881683731,
      "grad_norm": 1.0607426301957161,
      "learning_rate": 1.0901524465375015e-06,
      "loss": 0.0305,
      "step": 5117
    },
    {
      "epoch": 1.1645051194539249,
      "grad_norm": 1.0553962833230413,
      "learning_rate": 1.0900927633249638e-06,
      "loss": 0.0217,
      "step": 5118
    },
    {
      "epoch": 1.1647326507394766,
      "grad_norm": 1.0244204786724485,
      "learning_rate": 1.0900330706067818e-06,
      "loss": 0.0264,
      "step": 5119
    },
    {
      "epoch": 1.1649601820250284,
      "grad_norm": 0.9707054088367129,
      "learning_rate": 1.0899733683841753e-06,
      "loss": 0.0162,
      "step": 5120
    },
    {
      "epoch": 1.1651877133105801,
      "grad_norm": 1.1311434603614883,
      "learning_rate": 1.0899136566583647e-06,
      "loss": 0.028,
      "step": 5121
    },
    {
      "epoch": 1.1654152445961319,
      "grad_norm": 0.6912191457949579,
      "learning_rate": 1.0898539354305706e-06,
      "loss": 0.0099,
      "step": 5122
    },
    {
      "epoch": 1.1656427758816836,
      "grad_norm": 0.7902212704972374,
      "learning_rate": 1.0897942047020131e-06,
      "loss": 0.0128,
      "step": 5123
    },
    {
      "epoch": 1.1658703071672356,
      "grad_norm": 0.5967448578657041,
      "learning_rate": 1.0897344644739139e-06,
      "loss": 0.0081,
      "step": 5124
    },
    {
      "epoch": 1.1660978384527874,
      "grad_norm": 0.804672182956438,
      "learning_rate": 1.089674714747493e-06,
      "loss": 0.0121,
      "step": 5125
    },
    {
      "epoch": 1.1663253697383391,
      "grad_norm": 1.289356948925363,
      "learning_rate": 1.0896149555239717e-06,
      "loss": 0.0322,
      "step": 5126
    },
    {
      "epoch": 1.1665529010238909,
      "grad_norm": 1.8000751184542518,
      "learning_rate": 1.0895551868045718e-06,
      "loss": 0.055,
      "step": 5127
    },
    {
      "epoch": 1.1667804323094426,
      "grad_norm": 0.7403604907569553,
      "learning_rate": 1.0894954085905147e-06,
      "loss": 0.0213,
      "step": 5128
    },
    {
      "epoch": 1.1670079635949944,
      "grad_norm": 0.7673485205599404,
      "learning_rate": 1.0894356208830223e-06,
      "loss": 0.0146,
      "step": 5129
    },
    {
      "epoch": 1.1672354948805461,
      "grad_norm": 0.8057347260016878,
      "learning_rate": 1.089375823683316e-06,
      "loss": 0.0114,
      "step": 5130
    },
    {
      "epoch": 1.1674630261660979,
      "grad_norm": 1.031201971745674,
      "learning_rate": 1.0893160169926186e-06,
      "loss": 0.0224,
      "step": 5131
    },
    {
      "epoch": 1.1676905574516496,
      "grad_norm": 1.1471763380869269,
      "learning_rate": 1.0892562008121522e-06,
      "loss": 0.0199,
      "step": 5132
    },
    {
      "epoch": 1.1679180887372014,
      "grad_norm": 1.1194487112088174,
      "learning_rate": 1.0891963751431392e-06,
      "loss": 0.0202,
      "step": 5133
    },
    {
      "epoch": 1.1681456200227531,
      "grad_norm": 0.9431094841741555,
      "learning_rate": 1.0891365399868022e-06,
      "loss": 0.0242,
      "step": 5134
    },
    {
      "epoch": 1.1683731513083049,
      "grad_norm": 0.8152241328521926,
      "learning_rate": 1.0890766953443646e-06,
      "loss": 0.0135,
      "step": 5135
    },
    {
      "epoch": 1.1686006825938566,
      "grad_norm": 1.111189670935272,
      "learning_rate": 1.0890168412170493e-06,
      "loss": 0.0313,
      "step": 5136
    },
    {
      "epoch": 1.1688282138794084,
      "grad_norm": 1.0125526520638173,
      "learning_rate": 1.0889569776060796e-06,
      "loss": 0.0219,
      "step": 5137
    },
    {
      "epoch": 1.1690557451649601,
      "grad_norm": 1.1702719514469648,
      "learning_rate": 1.088897104512679e-06,
      "loss": 0.0349,
      "step": 5138
    },
    {
      "epoch": 1.1692832764505119,
      "grad_norm": 0.8014688872464257,
      "learning_rate": 1.0888372219380709e-06,
      "loss": 0.0134,
      "step": 5139
    },
    {
      "epoch": 1.1695108077360636,
      "grad_norm": 1.2120196426423329,
      "learning_rate": 1.0887773298834798e-06,
      "loss": 0.0218,
      "step": 5140
    },
    {
      "epoch": 1.1697383390216154,
      "grad_norm": 0.5093518570185542,
      "learning_rate": 1.0887174283501293e-06,
      "loss": 0.0069,
      "step": 5141
    },
    {
      "epoch": 1.1699658703071671,
      "grad_norm": 1.0586184070871325,
      "learning_rate": 1.0886575173392435e-06,
      "loss": 0.0212,
      "step": 5142
    },
    {
      "epoch": 1.170193401592719,
      "grad_norm": 0.9121062929597813,
      "learning_rate": 1.0885975968520476e-06,
      "loss": 0.01,
      "step": 5143
    },
    {
      "epoch": 1.1704209328782706,
      "grad_norm": 1.1192151515949293,
      "learning_rate": 1.0885376668897656e-06,
      "loss": 0.0297,
      "step": 5144
    },
    {
      "epoch": 1.1706484641638226,
      "grad_norm": 1.1083415447915677,
      "learning_rate": 1.0884777274536228e-06,
      "loss": 0.0288,
      "step": 5145
    },
    {
      "epoch": 1.1708759954493744,
      "grad_norm": 1.1157775165463915,
      "learning_rate": 1.0884177785448441e-06,
      "loss": 0.0179,
      "step": 5146
    },
    {
      "epoch": 1.1711035267349261,
      "grad_norm": 0.6942923206091653,
      "learning_rate": 1.0883578201646546e-06,
      "loss": 0.0186,
      "step": 5147
    },
    {
      "epoch": 1.1713310580204779,
      "grad_norm": 1.0660205113669436,
      "learning_rate": 1.08829785231428e-06,
      "loss": 0.0154,
      "step": 5148
    },
    {
      "epoch": 1.1715585893060296,
      "grad_norm": 1.2226468081458537,
      "learning_rate": 1.0882378749949456e-06,
      "loss": 0.027,
      "step": 5149
    },
    {
      "epoch": 1.1717861205915814,
      "grad_norm": 0.6736525742954345,
      "learning_rate": 1.0881778882078774e-06,
      "loss": 0.0104,
      "step": 5150
    },
    {
      "epoch": 1.1720136518771331,
      "grad_norm": 1.0175690285786059,
      "learning_rate": 1.0881178919543016e-06,
      "loss": 0.0209,
      "step": 5151
    },
    {
      "epoch": 1.1722411831626849,
      "grad_norm": 1.0274399291271918,
      "learning_rate": 1.0880578862354444e-06,
      "loss": 0.0169,
      "step": 5152
    },
    {
      "epoch": 1.1724687144482366,
      "grad_norm": 1.174579618024413,
      "learning_rate": 1.087997871052532e-06,
      "loss": 0.023,
      "step": 5153
    },
    {
      "epoch": 1.1726962457337884,
      "grad_norm": 1.0784512457128186,
      "learning_rate": 1.0879378464067906e-06,
      "loss": 0.0505,
      "step": 5154
    },
    {
      "epoch": 1.1729237770193401,
      "grad_norm": 1.2684504792555313,
      "learning_rate": 1.0878778122994477e-06,
      "loss": 0.0199,
      "step": 5155
    },
    {
      "epoch": 1.173151308304892,
      "grad_norm": 1.451789483548959,
      "learning_rate": 1.0878177687317302e-06,
      "loss": 0.0433,
      "step": 5156
    },
    {
      "epoch": 1.1733788395904436,
      "grad_norm": 0.9973415681599368,
      "learning_rate": 1.0877577157048648e-06,
      "loss": 0.0205,
      "step": 5157
    },
    {
      "epoch": 1.1736063708759954,
      "grad_norm": 2.814855357951941,
      "learning_rate": 1.0876976532200797e-06,
      "loss": 0.0309,
      "step": 5158
    },
    {
      "epoch": 1.1738339021615471,
      "grad_norm": 0.8916291744362826,
      "learning_rate": 1.0876375812786017e-06,
      "loss": 0.0207,
      "step": 5159
    },
    {
      "epoch": 1.174061433447099,
      "grad_norm": 0.7454883697223361,
      "learning_rate": 1.0875774998816586e-06,
      "loss": 0.0181,
      "step": 5160
    },
    {
      "epoch": 1.1742889647326507,
      "grad_norm": 1.5976521141289088,
      "learning_rate": 1.087517409030479e-06,
      "loss": 0.0352,
      "step": 5161
    },
    {
      "epoch": 1.1745164960182026,
      "grad_norm": 0.9023925994855438,
      "learning_rate": 1.0874573087262902e-06,
      "loss": 0.0189,
      "step": 5162
    },
    {
      "epoch": 1.1747440273037544,
      "grad_norm": 1.021608393538654,
      "learning_rate": 1.087397198970321e-06,
      "loss": 0.0248,
      "step": 5163
    },
    {
      "epoch": 1.1749715585893061,
      "grad_norm": 1.5179996755409662,
      "learning_rate": 1.0873370797638002e-06,
      "loss": 0.0346,
      "step": 5164
    },
    {
      "epoch": 1.1751990898748579,
      "grad_norm": 1.388608360949784,
      "learning_rate": 1.0872769511079561e-06,
      "loss": 0.0371,
      "step": 5165
    },
    {
      "epoch": 1.1754266211604096,
      "grad_norm": 0.9944177229048421,
      "learning_rate": 1.0872168130040175e-06,
      "loss": 0.0255,
      "step": 5166
    },
    {
      "epoch": 1.1756541524459614,
      "grad_norm": 1.460644807474802,
      "learning_rate": 1.087156665453214e-06,
      "loss": 0.0267,
      "step": 5167
    },
    {
      "epoch": 1.1758816837315131,
      "grad_norm": 1.201538811332985,
      "learning_rate": 1.0870965084567748e-06,
      "loss": 0.0322,
      "step": 5168
    },
    {
      "epoch": 1.176109215017065,
      "grad_norm": 0.9737685044159669,
      "learning_rate": 1.087036342015929e-06,
      "loss": 0.0238,
      "step": 5169
    },
    {
      "epoch": 1.1763367463026166,
      "grad_norm": 1.1123560277419813,
      "learning_rate": 1.0869761661319064e-06,
      "loss": 0.0323,
      "step": 5170
    },
    {
      "epoch": 1.1765642775881684,
      "grad_norm": 1.0917701556058614,
      "learning_rate": 1.0869159808059373e-06,
      "loss": 0.0239,
      "step": 5171
    },
    {
      "epoch": 1.1767918088737201,
      "grad_norm": 0.9254801632517604,
      "learning_rate": 1.0868557860392516e-06,
      "loss": 0.0225,
      "step": 5172
    },
    {
      "epoch": 1.177019340159272,
      "grad_norm": 1.1647756504926867,
      "learning_rate": 1.0867955818330792e-06,
      "loss": 0.0202,
      "step": 5173
    },
    {
      "epoch": 1.1772468714448237,
      "grad_norm": 0.9089361069774722,
      "learning_rate": 1.086735368188651e-06,
      "loss": 0.0167,
      "step": 5174
    },
    {
      "epoch": 1.1774744027303754,
      "grad_norm": 1.5111973761987758,
      "learning_rate": 1.0866751451071974e-06,
      "loss": 0.0295,
      "step": 5175
    },
    {
      "epoch": 1.1777019340159272,
      "grad_norm": 0.6596228779983846,
      "learning_rate": 1.0866149125899495e-06,
      "loss": 0.019,
      "step": 5176
    },
    {
      "epoch": 1.177929465301479,
      "grad_norm": 1.6565664804110074,
      "learning_rate": 1.086554670638138e-06,
      "loss": 0.031,
      "step": 5177
    },
    {
      "epoch": 1.1781569965870307,
      "grad_norm": 0.7431382614926502,
      "learning_rate": 1.0864944192529946e-06,
      "loss": 0.0125,
      "step": 5178
    },
    {
      "epoch": 1.1783845278725824,
      "grad_norm": 1.2886691602870353,
      "learning_rate": 1.08643415843575e-06,
      "loss": 0.0132,
      "step": 5179
    },
    {
      "epoch": 1.1786120591581342,
      "grad_norm": 1.2945110536777915,
      "learning_rate": 1.0863738881876369e-06,
      "loss": 0.0453,
      "step": 5180
    },
    {
      "epoch": 1.178839590443686,
      "grad_norm": 1.6944088405196147,
      "learning_rate": 1.0863136085098862e-06,
      "loss": 0.028,
      "step": 5181
    },
    {
      "epoch": 1.1790671217292377,
      "grad_norm": 0.9678714964974754,
      "learning_rate": 1.08625331940373e-06,
      "loss": 0.0186,
      "step": 5182
    },
    {
      "epoch": 1.1792946530147894,
      "grad_norm": 0.8164594441461164,
      "learning_rate": 1.086193020870401e-06,
      "loss": 0.0197,
      "step": 5183
    },
    {
      "epoch": 1.1795221843003414,
      "grad_norm": 0.847422876421066,
      "learning_rate": 1.0861327129111313e-06,
      "loss": 0.014,
      "step": 5184
    },
    {
      "epoch": 1.1797497155858931,
      "grad_norm": 0.8408136915565151,
      "learning_rate": 1.0860723955271533e-06,
      "loss": 0.0141,
      "step": 5185
    },
    {
      "epoch": 1.179977246871445,
      "grad_norm": 1.4185159384093218,
      "learning_rate": 1.0860120687196998e-06,
      "loss": 0.0174,
      "step": 5186
    },
    {
      "epoch": 1.1802047781569966,
      "grad_norm": 1.3653351030007033,
      "learning_rate": 1.0859517324900042e-06,
      "loss": 0.0318,
      "step": 5187
    },
    {
      "epoch": 1.1804323094425484,
      "grad_norm": 0.9616143252233853,
      "learning_rate": 1.0858913868392993e-06,
      "loss": 0.0324,
      "step": 5188
    },
    {
      "epoch": 1.1806598407281002,
      "grad_norm": 0.8826104173449131,
      "learning_rate": 1.0858310317688184e-06,
      "loss": 0.0194,
      "step": 5189
    },
    {
      "epoch": 1.180887372013652,
      "grad_norm": 0.573005849494334,
      "learning_rate": 1.0857706672797954e-06,
      "loss": 0.007,
      "step": 5190
    },
    {
      "epoch": 1.1811149032992037,
      "grad_norm": 0.792841731677897,
      "learning_rate": 1.0857102933734636e-06,
      "loss": 0.0228,
      "step": 5191
    },
    {
      "epoch": 1.1813424345847554,
      "grad_norm": 1.1198194956030572,
      "learning_rate": 1.0856499100510575e-06,
      "loss": 0.0204,
      "step": 5192
    },
    {
      "epoch": 1.1815699658703072,
      "grad_norm": 1.4060671628835655,
      "learning_rate": 1.0855895173138107e-06,
      "loss": 0.0335,
      "step": 5193
    },
    {
      "epoch": 1.181797497155859,
      "grad_norm": 1.0332252844504048,
      "learning_rate": 1.0855291151629576e-06,
      "loss": 0.0248,
      "step": 5194
    },
    {
      "epoch": 1.1820250284414107,
      "grad_norm": 1.3939283564653595,
      "learning_rate": 1.0854687035997329e-06,
      "loss": 0.0503,
      "step": 5195
    },
    {
      "epoch": 1.1822525597269624,
      "grad_norm": 1.389414773323681,
      "learning_rate": 1.0854082826253712e-06,
      "loss": 0.0179,
      "step": 5196
    },
    {
      "epoch": 1.1824800910125142,
      "grad_norm": 1.3843557347025581,
      "learning_rate": 1.0853478522411072e-06,
      "loss": 0.0235,
      "step": 5197
    },
    {
      "epoch": 1.182707622298066,
      "grad_norm": 3.1968855275158368,
      "learning_rate": 1.0852874124481764e-06,
      "loss": 0.0411,
      "step": 5198
    },
    {
      "epoch": 1.1829351535836177,
      "grad_norm": 1.4480239507963797,
      "learning_rate": 1.0852269632478138e-06,
      "loss": 0.0363,
      "step": 5199
    },
    {
      "epoch": 1.1831626848691694,
      "grad_norm": 1.9126854541087823,
      "learning_rate": 1.085166504641255e-06,
      "loss": 0.0181,
      "step": 5200
    },
    {
      "epoch": 1.1833902161547214,
      "grad_norm": 1.1213805046183494,
      "learning_rate": 1.0851060366297356e-06,
      "loss": 0.0353,
      "step": 5201
    },
    {
      "epoch": 1.1836177474402731,
      "grad_norm": 2.0688040255631126,
      "learning_rate": 1.0850455592144916e-06,
      "loss": 0.0451,
      "step": 5202
    },
    {
      "epoch": 1.183845278725825,
      "grad_norm": 1.7510399141137207,
      "learning_rate": 1.0849850723967585e-06,
      "loss": 0.0109,
      "step": 5203
    },
    {
      "epoch": 1.1840728100113767,
      "grad_norm": 6.664020652459381,
      "learning_rate": 1.0849245761777733e-06,
      "loss": 0.0381,
      "step": 5204
    },
    {
      "epoch": 1.1843003412969284,
      "grad_norm": 2.621359696329597,
      "learning_rate": 1.0848640705587718e-06,
      "loss": 0.0216,
      "step": 5205
    },
    {
      "epoch": 1.1845278725824802,
      "grad_norm": 1.8030419214813977,
      "learning_rate": 1.0848035555409911e-06,
      "loss": 0.0303,
      "step": 5206
    },
    {
      "epoch": 1.184755403868032,
      "grad_norm": 1.186220362313326,
      "learning_rate": 1.0847430311256676e-06,
      "loss": 0.0197,
      "step": 5207
    },
    {
      "epoch": 1.1849829351535837,
      "grad_norm": 1.5048336995036211,
      "learning_rate": 1.0846824973140388e-06,
      "loss": 0.0185,
      "step": 5208
    },
    {
      "epoch": 1.1852104664391354,
      "grad_norm": 1.8320739469192913,
      "learning_rate": 1.0846219541073417e-06,
      "loss": 0.0217,
      "step": 5209
    },
    {
      "epoch": 1.1854379977246872,
      "grad_norm": 1.6975307868867775,
      "learning_rate": 1.084561401506813e-06,
      "loss": 0.0491,
      "step": 5210
    },
    {
      "epoch": 1.185665529010239,
      "grad_norm": 2.1632818389721153,
      "learning_rate": 1.0845008395136915e-06,
      "loss": 0.057,
      "step": 5211
    },
    {
      "epoch": 1.1858930602957907,
      "grad_norm": 1.271479589500172,
      "learning_rate": 1.0844402681292144e-06,
      "loss": 0.0369,
      "step": 5212
    },
    {
      "epoch": 1.1861205915813424,
      "grad_norm": 1.0411295086764045,
      "learning_rate": 1.084379687354619e-06,
      "loss": 0.0097,
      "step": 5213
    },
    {
      "epoch": 1.1863481228668942,
      "grad_norm": 2.6559141117447225,
      "learning_rate": 1.0843190971911447e-06,
      "loss": 0.0228,
      "step": 5214
    },
    {
      "epoch": 1.186575654152446,
      "grad_norm": 1.5202925528678082,
      "learning_rate": 1.0842584976400292e-06,
      "loss": 0.0338,
      "step": 5215
    },
    {
      "epoch": 1.1868031854379977,
      "grad_norm": 1.1093380023536548,
      "learning_rate": 1.084197888702511e-06,
      "loss": 0.0168,
      "step": 5216
    },
    {
      "epoch": 1.1870307167235494,
      "grad_norm": 1.5307787654530016,
      "learning_rate": 1.0841372703798289e-06,
      "loss": 0.0278,
      "step": 5217
    },
    {
      "epoch": 1.1872582480091012,
      "grad_norm": 1.3322857454820025,
      "learning_rate": 1.0840766426732219e-06,
      "loss": 0.0347,
      "step": 5218
    },
    {
      "epoch": 1.187485779294653,
      "grad_norm": 0.8608991873903781,
      "learning_rate": 1.0840160055839291e-06,
      "loss": 0.0238,
      "step": 5219
    },
    {
      "epoch": 1.1877133105802047,
      "grad_norm": 1.119129138655594,
      "learning_rate": 1.08395535911319e-06,
      "loss": 0.0131,
      "step": 5220
    },
    {
      "epoch": 1.1879408418657564,
      "grad_norm": 0.9520127980932701,
      "learning_rate": 1.0838947032622436e-06,
      "loss": 0.0187,
      "step": 5221
    },
    {
      "epoch": 1.1881683731513082,
      "grad_norm": 1.0013994149687084,
      "learning_rate": 1.0838340380323297e-06,
      "loss": 0.0217,
      "step": 5222
    },
    {
      "epoch": 1.1883959044368602,
      "grad_norm": 2.3034636462642166,
      "learning_rate": 1.083773363424689e-06,
      "loss": 0.0253,
      "step": 5223
    },
    {
      "epoch": 1.188623435722412,
      "grad_norm": 0.9604142125712335,
      "learning_rate": 1.0837126794405603e-06,
      "loss": 0.0304,
      "step": 5224
    },
    {
      "epoch": 1.1888509670079637,
      "grad_norm": 0.9680611296826206,
      "learning_rate": 1.083651986081185e-06,
      "loss": 0.0233,
      "step": 5225
    },
    {
      "epoch": 1.1890784982935154,
      "grad_norm": 1.1564232624147677,
      "learning_rate": 1.0835912833478029e-06,
      "loss": 0.0514,
      "step": 5226
    },
    {
      "epoch": 1.1893060295790672,
      "grad_norm": 1.0708217476839645,
      "learning_rate": 1.0835305712416546e-06,
      "loss": 0.0244,
      "step": 5227
    },
    {
      "epoch": 1.189533560864619,
      "grad_norm": 1.1356456632073215,
      "learning_rate": 1.0834698497639817e-06,
      "loss": 0.0453,
      "step": 5228
    },
    {
      "epoch": 1.1897610921501707,
      "grad_norm": 0.9643109850754426,
      "learning_rate": 1.0834091189160243e-06,
      "loss": 0.0148,
      "step": 5229
    },
    {
      "epoch": 1.1899886234357224,
      "grad_norm": 1.4647639680120725,
      "learning_rate": 1.0833483786990243e-06,
      "loss": 0.0398,
      "step": 5230
    },
    {
      "epoch": 1.1902161547212742,
      "grad_norm": 0.6967268185621851,
      "learning_rate": 1.0832876291142228e-06,
      "loss": 0.0153,
      "step": 5231
    },
    {
      "epoch": 1.190443686006826,
      "grad_norm": 1.5226277581599168,
      "learning_rate": 1.0832268701628613e-06,
      "loss": 0.0387,
      "step": 5232
    },
    {
      "epoch": 1.1906712172923777,
      "grad_norm": 1.5725234513516795,
      "learning_rate": 1.083166101846182e-06,
      "loss": 0.059,
      "step": 5233
    },
    {
      "epoch": 1.1908987485779294,
      "grad_norm": 1.0092630282850967,
      "learning_rate": 1.0831053241654265e-06,
      "loss": 0.019,
      "step": 5234
    },
    {
      "epoch": 1.1911262798634812,
      "grad_norm": 1.1102622210079196,
      "learning_rate": 1.0830445371218374e-06,
      "loss": 0.0234,
      "step": 5235
    },
    {
      "epoch": 1.191353811149033,
      "grad_norm": 1.2854110941773407,
      "learning_rate": 1.0829837407166565e-06,
      "loss": 0.0243,
      "step": 5236
    },
    {
      "epoch": 1.1915813424345847,
      "grad_norm": 0.38016640183110134,
      "learning_rate": 1.082922934951127e-06,
      "loss": 0.0134,
      "step": 5237
    },
    {
      "epoch": 1.1918088737201364,
      "grad_norm": 1.168241176946775,
      "learning_rate": 1.082862119826491e-06,
      "loss": 0.031,
      "step": 5238
    },
    {
      "epoch": 1.1920364050056882,
      "grad_norm": 1.1960841025032027,
      "learning_rate": 1.0828012953439921e-06,
      "loss": 0.052,
      "step": 5239
    },
    {
      "epoch": 1.1922639362912402,
      "grad_norm": 0.7158528800285552,
      "learning_rate": 1.082740461504873e-06,
      "loss": 0.011,
      "step": 5240
    },
    {
      "epoch": 1.192491467576792,
      "grad_norm": 1.1123900753579041,
      "learning_rate": 1.0826796183103774e-06,
      "loss": 0.0442,
      "step": 5241
    },
    {
      "epoch": 1.1927189988623437,
      "grad_norm": 1.3167322420939016,
      "learning_rate": 1.0826187657617484e-06,
      "loss": 0.0351,
      "step": 5242
    },
    {
      "epoch": 1.1929465301478954,
      "grad_norm": 1.8256977932069058,
      "learning_rate": 1.0825579038602298e-06,
      "loss": 0.0228,
      "step": 5243
    },
    {
      "epoch": 1.1931740614334472,
      "grad_norm": 0.8153271667842001,
      "learning_rate": 1.082497032607066e-06,
      "loss": 0.0218,
      "step": 5244
    },
    {
      "epoch": 1.193401592718999,
      "grad_norm": 1.13052228865416,
      "learning_rate": 1.0824361520035004e-06,
      "loss": 0.0269,
      "step": 5245
    },
    {
      "epoch": 1.1936291240045507,
      "grad_norm": 0.8769614800574006,
      "learning_rate": 1.0823752620507778e-06,
      "loss": 0.0196,
      "step": 5246
    },
    {
      "epoch": 1.1938566552901024,
      "grad_norm": 2.442105507702053,
      "learning_rate": 1.0823143627501423e-06,
      "loss": 0.0223,
      "step": 5247
    },
    {
      "epoch": 1.1940841865756542,
      "grad_norm": 0.9690120767439246,
      "learning_rate": 1.082253454102839e-06,
      "loss": 0.0293,
      "step": 5248
    },
    {
      "epoch": 1.194311717861206,
      "grad_norm": 0.9656302532397552,
      "learning_rate": 1.0821925361101124e-06,
      "loss": 0.0194,
      "step": 5249
    },
    {
      "epoch": 1.1945392491467577,
      "grad_norm": 1.2220527685129399,
      "learning_rate": 1.0821316087732075e-06,
      "loss": 0.0361,
      "step": 5250
    },
    {
      "epoch": 1.1947667804323094,
      "grad_norm": 1.1131169147903066,
      "learning_rate": 1.0820706720933698e-06,
      "loss": 0.0282,
      "step": 5251
    },
    {
      "epoch": 1.1949943117178612,
      "grad_norm": 1.1331333681292557,
      "learning_rate": 1.0820097260718448e-06,
      "loss": 0.0117,
      "step": 5252
    },
    {
      "epoch": 1.195221843003413,
      "grad_norm": 1.126019194986974,
      "learning_rate": 1.081948770709878e-06,
      "loss": 0.0142,
      "step": 5253
    },
    {
      "epoch": 1.1954493742889647,
      "grad_norm": 1.9300596010130329,
      "learning_rate": 1.0818878060087151e-06,
      "loss": 0.0305,
      "step": 5254
    },
    {
      "epoch": 1.1956769055745164,
      "grad_norm": 3.8484100820913096,
      "learning_rate": 1.081826831969602e-06,
      "loss": 0.0175,
      "step": 5255
    },
    {
      "epoch": 1.1959044368600682,
      "grad_norm": 2.6177072747321493,
      "learning_rate": 1.0817658485937855e-06,
      "loss": 0.0356,
      "step": 5256
    },
    {
      "epoch": 1.19613196814562,
      "grad_norm": 1.8647005779081272,
      "learning_rate": 1.0817048558825114e-06,
      "loss": 0.0416,
      "step": 5257
    },
    {
      "epoch": 1.1963594994311717,
      "grad_norm": 1.8201241368258048,
      "learning_rate": 1.0816438538370262e-06,
      "loss": 0.03,
      "step": 5258
    },
    {
      "epoch": 1.1965870307167235,
      "grad_norm": 1.309838446791344,
      "learning_rate": 1.0815828424585772e-06,
      "loss": 0.0221,
      "step": 5259
    },
    {
      "epoch": 1.1968145620022752,
      "grad_norm": 2.2126363422898736,
      "learning_rate": 1.081521821748411e-06,
      "loss": 0.01,
      "step": 5260
    },
    {
      "epoch": 1.197042093287827,
      "grad_norm": 1.5225649498436484,
      "learning_rate": 1.081460791707775e-06,
      "loss": 0.0106,
      "step": 5261
    },
    {
      "epoch": 1.197269624573379,
      "grad_norm": 2.926124794782369,
      "learning_rate": 1.0813997523379163e-06,
      "loss": 0.0545,
      "step": 5262
    },
    {
      "epoch": 1.1974971558589307,
      "grad_norm": 3.121781275137615,
      "learning_rate": 1.0813387036400825e-06,
      "loss": 0.039,
      "step": 5263
    },
    {
      "epoch": 1.1977246871444824,
      "grad_norm": 2.991132032943111,
      "learning_rate": 1.081277645615521e-06,
      "loss": 0.0217,
      "step": 5264
    },
    {
      "epoch": 1.1979522184300342,
      "grad_norm": 0.7510540265715727,
      "learning_rate": 1.0812165782654806e-06,
      "loss": 0.0113,
      "step": 5265
    },
    {
      "epoch": 1.198179749715586,
      "grad_norm": 0.9442454590864651,
      "learning_rate": 1.0811555015912086e-06,
      "loss": 0.0128,
      "step": 5266
    },
    {
      "epoch": 1.1984072810011377,
      "grad_norm": 1.0369867334237874,
      "learning_rate": 1.0810944155939536e-06,
      "loss": 0.0289,
      "step": 5267
    },
    {
      "epoch": 1.1986348122866894,
      "grad_norm": 1.2893237225891867,
      "learning_rate": 1.081033320274964e-06,
      "loss": 0.0225,
      "step": 5268
    },
    {
      "epoch": 1.1988623435722412,
      "grad_norm": 1.4348798625399937,
      "learning_rate": 1.0809722156354884e-06,
      "loss": 0.0196,
      "step": 5269
    },
    {
      "epoch": 1.199089874857793,
      "grad_norm": 0.7783433954066992,
      "learning_rate": 1.0809111016767762e-06,
      "loss": 0.0132,
      "step": 5270
    },
    {
      "epoch": 1.1993174061433447,
      "grad_norm": 2.29125273839501,
      "learning_rate": 1.0808499784000756e-06,
      "loss": 0.0155,
      "step": 5271
    },
    {
      "epoch": 1.1995449374288965,
      "grad_norm": 1.8766245713656982,
      "learning_rate": 1.0807888458066364e-06,
      "loss": 0.0405,
      "step": 5272
    },
    {
      "epoch": 1.1997724687144482,
      "grad_norm": 2.0497369482648997,
      "learning_rate": 1.0807277038977083e-06,
      "loss": 0.0483,
      "step": 5273
    },
    {
      "epoch": 1.2,
      "grad_norm": 1.3799009247077467,
      "learning_rate": 1.0806665526745403e-06,
      "loss": 0.024,
      "step": 5274
    },
    {
      "epoch": 1.2002275312855517,
      "grad_norm": 0.9219092351160844,
      "learning_rate": 1.0806053921383823e-06,
      "loss": 0.0206,
      "step": 5275
    },
    {
      "epoch": 1.2004550625711035,
      "grad_norm": 0.7062087076877579,
      "learning_rate": 1.0805442222904846e-06,
      "loss": 0.0197,
      "step": 5276
    },
    {
      "epoch": 1.2006825938566552,
      "grad_norm": 1.1815501920868743,
      "learning_rate": 1.0804830431320972e-06,
      "loss": 0.0311,
      "step": 5277
    },
    {
      "epoch": 1.200910125142207,
      "grad_norm": 1.4194267958205675,
      "learning_rate": 1.0804218546644708e-06,
      "loss": 0.0269,
      "step": 5278
    },
    {
      "epoch": 1.201137656427759,
      "grad_norm": 1.5374413742413802,
      "learning_rate": 1.0803606568888557e-06,
      "loss": 0.0371,
      "step": 5279
    },
    {
      "epoch": 1.2013651877133107,
      "grad_norm": 1.4162269983170825,
      "learning_rate": 1.0802994498065027e-06,
      "loss": 0.0433,
      "step": 5280
    },
    {
      "epoch": 1.2015927189988624,
      "grad_norm": 1.3982466009914463,
      "learning_rate": 1.0802382334186627e-06,
      "loss": 0.02,
      "step": 5281
    },
    {
      "epoch": 1.2018202502844142,
      "grad_norm": 1.0353374922029346,
      "learning_rate": 1.080177007726587e-06,
      "loss": 0.0314,
      "step": 5282
    },
    {
      "epoch": 1.202047781569966,
      "grad_norm": 1.4144577926145716,
      "learning_rate": 1.080115772731527e-06,
      "loss": 0.0237,
      "step": 5283
    },
    {
      "epoch": 1.2022753128555177,
      "grad_norm": 1.270246359088111,
      "learning_rate": 1.0800545284347338e-06,
      "loss": 0.0278,
      "step": 5284
    },
    {
      "epoch": 1.2025028441410694,
      "grad_norm": 1.7811867988957832,
      "learning_rate": 1.0799932748374598e-06,
      "loss": 0.0351,
      "step": 5285
    },
    {
      "epoch": 1.2027303754266212,
      "grad_norm": 0.811331257305328,
      "learning_rate": 1.0799320119409562e-06,
      "loss": 0.0154,
      "step": 5286
    },
    {
      "epoch": 1.202957906712173,
      "grad_norm": 1.326967549928065,
      "learning_rate": 1.0798707397464756e-06,
      "loss": 0.0251,
      "step": 5287
    },
    {
      "epoch": 1.2031854379977247,
      "grad_norm": 1.2668064811271582,
      "learning_rate": 1.0798094582552703e-06,
      "loss": 0.0214,
      "step": 5288
    },
    {
      "epoch": 1.2034129692832765,
      "grad_norm": 1.7316590095255162,
      "learning_rate": 1.0797481674685925e-06,
      "loss": 0.0595,
      "step": 5289
    },
    {
      "epoch": 1.2036405005688282,
      "grad_norm": 0.8626668121425645,
      "learning_rate": 1.0796868673876947e-06,
      "loss": 0.0138,
      "step": 5290
    },
    {
      "epoch": 1.20386803185438,
      "grad_norm": 1.688909270364021,
      "learning_rate": 1.0796255580138303e-06,
      "loss": 0.0376,
      "step": 5291
    },
    {
      "epoch": 1.2040955631399317,
      "grad_norm": 1.3043440282984107,
      "learning_rate": 1.0795642393482523e-06,
      "loss": 0.0511,
      "step": 5292
    },
    {
      "epoch": 1.2043230944254835,
      "grad_norm": 1.2730397806370202,
      "learning_rate": 1.0795029113922136e-06,
      "loss": 0.0232,
      "step": 5293
    },
    {
      "epoch": 1.2045506257110352,
      "grad_norm": 0.5816881949668542,
      "learning_rate": 1.0794415741469677e-06,
      "loss": 0.0086,
      "step": 5294
    },
    {
      "epoch": 1.204778156996587,
      "grad_norm": 1.0835756813832158,
      "learning_rate": 1.0793802276137683e-06,
      "loss": 0.0204,
      "step": 5295
    },
    {
      "epoch": 1.2050056882821387,
      "grad_norm": 1.8276533196679292,
      "learning_rate": 1.0793188717938693e-06,
      "loss": 0.0304,
      "step": 5296
    },
    {
      "epoch": 1.2052332195676905,
      "grad_norm": 2.167032474707277,
      "learning_rate": 1.0792575066885245e-06,
      "loss": 0.0331,
      "step": 5297
    },
    {
      "epoch": 1.2054607508532422,
      "grad_norm": 0.7493889811458411,
      "learning_rate": 1.0791961322989882e-06,
      "loss": 0.0147,
      "step": 5298
    },
    {
      "epoch": 1.205688282138794,
      "grad_norm": 1.329356172948289,
      "learning_rate": 1.0791347486265147e-06,
      "loss": 0.0353,
      "step": 5299
    },
    {
      "epoch": 1.2059158134243457,
      "grad_norm": 0.8797921570409813,
      "learning_rate": 1.0790733556723589e-06,
      "loss": 0.0242,
      "step": 5300
    },
    {
      "epoch": 1.2061433447098977,
      "grad_norm": 0.8299000303047913,
      "learning_rate": 1.079011953437775e-06,
      "loss": 0.0131,
      "step": 5301
    },
    {
      "epoch": 1.2063708759954495,
      "grad_norm": 0.5695113688949743,
      "learning_rate": 1.0789505419240185e-06,
      "loss": 0.0091,
      "step": 5302
    },
    {
      "epoch": 1.2065984072810012,
      "grad_norm": 1.007416588849143,
      "learning_rate": 1.0788891211323442e-06,
      "loss": 0.0211,
      "step": 5303
    },
    {
      "epoch": 1.206825938566553,
      "grad_norm": 1.0207685534429214,
      "learning_rate": 1.0788276910640074e-06,
      "loss": 0.0167,
      "step": 5304
    },
    {
      "epoch": 1.2070534698521047,
      "grad_norm": 0.9531066995071166,
      "learning_rate": 1.0787662517202641e-06,
      "loss": 0.0264,
      "step": 5305
    },
    {
      "epoch": 1.2072810011376565,
      "grad_norm": 0.8593609085890438,
      "learning_rate": 1.0787048031023693e-06,
      "loss": 0.0192,
      "step": 5306
    },
    {
      "epoch": 1.2075085324232082,
      "grad_norm": 0.7570873924698162,
      "learning_rate": 1.0786433452115794e-06,
      "loss": 0.0107,
      "step": 5307
    },
    {
      "epoch": 1.20773606370876,
      "grad_norm": 0.6192632652453488,
      "learning_rate": 1.0785818780491502e-06,
      "loss": 0.01,
      "step": 5308
    },
    {
      "epoch": 1.2079635949943117,
      "grad_norm": 1.1762597971964037,
      "learning_rate": 1.0785204016163384e-06,
      "loss": 0.017,
      "step": 5309
    },
    {
      "epoch": 1.2081911262798635,
      "grad_norm": 0.8264299611729337,
      "learning_rate": 1.0784589159143999e-06,
      "loss": 0.0137,
      "step": 5310
    },
    {
      "epoch": 1.2084186575654152,
      "grad_norm": 0.5022797081392271,
      "learning_rate": 1.0783974209445915e-06,
      "loss": 0.0069,
      "step": 5311
    },
    {
      "epoch": 1.208646188850967,
      "grad_norm": 0.6561114668173221,
      "learning_rate": 1.0783359167081705e-06,
      "loss": 0.0178,
      "step": 5312
    },
    {
      "epoch": 1.2088737201365187,
      "grad_norm": 1.860198594863872,
      "learning_rate": 1.0782744032063935e-06,
      "loss": 0.0395,
      "step": 5313
    },
    {
      "epoch": 1.2091012514220705,
      "grad_norm": 1.2557407515136827,
      "learning_rate": 1.078212880440518e-06,
      "loss": 0.0198,
      "step": 5314
    },
    {
      "epoch": 1.2093287827076222,
      "grad_norm": 0.9774237973356178,
      "learning_rate": 1.0781513484118008e-06,
      "loss": 0.033,
      "step": 5315
    },
    {
      "epoch": 1.209556313993174,
      "grad_norm": 1.1404558362365051,
      "learning_rate": 1.0780898071215004e-06,
      "loss": 0.0153,
      "step": 5316
    },
    {
      "epoch": 1.2097838452787257,
      "grad_norm": 1.2711927135952792,
      "learning_rate": 1.0780282565708738e-06,
      "loss": 0.0452,
      "step": 5317
    },
    {
      "epoch": 1.2100113765642777,
      "grad_norm": 1.572981203889357,
      "learning_rate": 1.0779666967611796e-06,
      "loss": 0.0366,
      "step": 5318
    },
    {
      "epoch": 1.2102389078498295,
      "grad_norm": 1.0432053268711592,
      "learning_rate": 1.0779051276936755e-06,
      "loss": 0.0193,
      "step": 5319
    },
    {
      "epoch": 1.2104664391353812,
      "grad_norm": 1.108281271730141,
      "learning_rate": 1.0778435493696202e-06,
      "loss": 0.0514,
      "step": 5320
    },
    {
      "epoch": 1.210693970420933,
      "grad_norm": 1.0307341066968732,
      "learning_rate": 1.0777819617902718e-06,
      "loss": 0.0232,
      "step": 5321
    },
    {
      "epoch": 1.2109215017064847,
      "grad_norm": 1.1269375808316393,
      "learning_rate": 1.0777203649568896e-06,
      "loss": 0.0111,
      "step": 5322
    },
    {
      "epoch": 1.2111490329920365,
      "grad_norm": 0.8681106412081948,
      "learning_rate": 1.077658758870732e-06,
      "loss": 0.0134,
      "step": 5323
    },
    {
      "epoch": 1.2113765642775882,
      "grad_norm": 1.6310011066128904,
      "learning_rate": 1.0775971435330588e-06,
      "loss": 0.0443,
      "step": 5324
    },
    {
      "epoch": 1.21160409556314,
      "grad_norm": 1.9800926111893868,
      "learning_rate": 1.0775355189451286e-06,
      "loss": 0.0491,
      "step": 5325
    },
    {
      "epoch": 1.2118316268486917,
      "grad_norm": 1.0735676656079027,
      "learning_rate": 1.0774738851082011e-06,
      "loss": 0.0137,
      "step": 5326
    },
    {
      "epoch": 1.2120591581342435,
      "grad_norm": 1.4112445459881595,
      "learning_rate": 1.0774122420235363e-06,
      "loss": 0.0267,
      "step": 5327
    },
    {
      "epoch": 1.2122866894197952,
      "grad_norm": 1.6264263710804632,
      "learning_rate": 1.0773505896923936e-06,
      "loss": 0.0601,
      "step": 5328
    },
    {
      "epoch": 1.212514220705347,
      "grad_norm": 1.4435922348969275,
      "learning_rate": 1.0772889281160335e-06,
      "loss": 0.0487,
      "step": 5329
    },
    {
      "epoch": 1.2127417519908987,
      "grad_norm": 1.6845252736050549,
      "learning_rate": 1.0772272572957158e-06,
      "loss": 0.0384,
      "step": 5330
    },
    {
      "epoch": 1.2129692832764505,
      "grad_norm": 0.620039095627728,
      "learning_rate": 1.0771655772327013e-06,
      "loss": 0.0087,
      "step": 5331
    },
    {
      "epoch": 1.2131968145620022,
      "grad_norm": 1.0155486269794136,
      "learning_rate": 1.0771038879282505e-06,
      "loss": 0.0316,
      "step": 5332
    },
    {
      "epoch": 1.213424345847554,
      "grad_norm": 1.0720369582435914,
      "learning_rate": 1.0770421893836243e-06,
      "loss": 0.0297,
      "step": 5333
    },
    {
      "epoch": 1.2136518771331057,
      "grad_norm": 1.0641000248930415,
      "learning_rate": 1.0769804816000835e-06,
      "loss": 0.0228,
      "step": 5334
    },
    {
      "epoch": 1.2138794084186575,
      "grad_norm": 1.3185310795364764,
      "learning_rate": 1.0769187645788895e-06,
      "loss": 0.0325,
      "step": 5335
    },
    {
      "epoch": 1.2141069397042092,
      "grad_norm": 0.8019919288175986,
      "learning_rate": 1.0768570383213035e-06,
      "loss": 0.0108,
      "step": 5336
    },
    {
      "epoch": 1.214334470989761,
      "grad_norm": 1.0171180200964656,
      "learning_rate": 1.0767953028285872e-06,
      "loss": 0.0162,
      "step": 5337
    },
    {
      "epoch": 1.2145620022753127,
      "grad_norm": 0.6898326225080136,
      "learning_rate": 1.0767335581020024e-06,
      "loss": 0.0105,
      "step": 5338
    },
    {
      "epoch": 1.2147895335608645,
      "grad_norm": 0.5713039580718985,
      "learning_rate": 1.076671804142811e-06,
      "loss": 0.0077,
      "step": 5339
    },
    {
      "epoch": 1.2150170648464165,
      "grad_norm": 1.2459220754175229,
      "learning_rate": 1.076610040952275e-06,
      "loss": 0.0332,
      "step": 5340
    },
    {
      "epoch": 1.2152445961319682,
      "grad_norm": 1.1187776697921044,
      "learning_rate": 1.076548268531657e-06,
      "loss": 0.0191,
      "step": 5341
    },
    {
      "epoch": 1.21547212741752,
      "grad_norm": 2.0317760107761704,
      "learning_rate": 1.0764864868822194e-06,
      "loss": 0.0516,
      "step": 5342
    },
    {
      "epoch": 1.2156996587030717,
      "grad_norm": 1.9233690667339316,
      "learning_rate": 1.0764246960052247e-06,
      "loss": 0.0134,
      "step": 5343
    },
    {
      "epoch": 1.2159271899886235,
      "grad_norm": 1.6064839097939867,
      "learning_rate": 1.0763628959019359e-06,
      "loss": 0.0209,
      "step": 5344
    },
    {
      "epoch": 1.2161547212741752,
      "grad_norm": 1.2879343782626302,
      "learning_rate": 1.076301086573616e-06,
      "loss": 0.016,
      "step": 5345
    },
    {
      "epoch": 1.216382252559727,
      "grad_norm": 1.1521103656454588,
      "learning_rate": 1.076239268021529e-06,
      "loss": 0.0294,
      "step": 5346
    },
    {
      "epoch": 1.2166097838452787,
      "grad_norm": 0.7645894367138184,
      "learning_rate": 1.0761774402469375e-06,
      "loss": 0.0146,
      "step": 5347
    },
    {
      "epoch": 1.2168373151308305,
      "grad_norm": 1.0900116364078585,
      "learning_rate": 1.0761156032511052e-06,
      "loss": 0.0148,
      "step": 5348
    },
    {
      "epoch": 1.2170648464163822,
      "grad_norm": 1.0426468341641788,
      "learning_rate": 1.0760537570352963e-06,
      "loss": 0.0205,
      "step": 5349
    },
    {
      "epoch": 1.217292377701934,
      "grad_norm": 0.968560865785141,
      "learning_rate": 1.0759919016007747e-06,
      "loss": 0.028,
      "step": 5350
    },
    {
      "epoch": 1.2175199089874857,
      "grad_norm": 1.3908329783187319,
      "learning_rate": 1.0759300369488046e-06,
      "loss": 0.0311,
      "step": 5351
    },
    {
      "epoch": 1.2177474402730375,
      "grad_norm": 0.9336674383856225,
      "learning_rate": 1.0758681630806502e-06,
      "loss": 0.0126,
      "step": 5352
    },
    {
      "epoch": 1.2179749715585892,
      "grad_norm": 0.9822161525178283,
      "learning_rate": 1.0758062799975765e-06,
      "loss": 0.0328,
      "step": 5353
    },
    {
      "epoch": 1.218202502844141,
      "grad_norm": 1.56822129701628,
      "learning_rate": 1.075744387700848e-06,
      "loss": 0.0543,
      "step": 5354
    },
    {
      "epoch": 1.2184300341296928,
      "grad_norm": 0.8508512043785544,
      "learning_rate": 1.0756824861917294e-06,
      "loss": 0.0186,
      "step": 5355
    },
    {
      "epoch": 1.2186575654152445,
      "grad_norm": 1.6471064921915508,
      "learning_rate": 1.0756205754714866e-06,
      "loss": 0.0419,
      "step": 5356
    },
    {
      "epoch": 1.2188850967007965,
      "grad_norm": 0.9710323188808565,
      "learning_rate": 1.0755586555413845e-06,
      "loss": 0.0165,
      "step": 5357
    },
    {
      "epoch": 1.2191126279863482,
      "grad_norm": 1.1740838398504168,
      "learning_rate": 1.0754967264026882e-06,
      "loss": 0.0171,
      "step": 5358
    },
    {
      "epoch": 1.2193401592719,
      "grad_norm": 0.821398345986984,
      "learning_rate": 1.0754347880566643e-06,
      "loss": 0.0144,
      "step": 5359
    },
    {
      "epoch": 1.2195676905574517,
      "grad_norm": 0.737155613985261,
      "learning_rate": 1.075372840504578e-06,
      "loss": 0.0215,
      "step": 5360
    },
    {
      "epoch": 1.2197952218430035,
      "grad_norm": 1.2248403884170755,
      "learning_rate": 1.0753108837476958e-06,
      "loss": 0.023,
      "step": 5361
    },
    {
      "epoch": 1.2200227531285552,
      "grad_norm": 1.5213992441300637,
      "learning_rate": 1.0752489177872839e-06,
      "loss": 0.0211,
      "step": 5362
    },
    {
      "epoch": 1.220250284414107,
      "grad_norm": 0.9276211966706847,
      "learning_rate": 1.0751869426246086e-06,
      "loss": 0.0248,
      "step": 5363
    },
    {
      "epoch": 1.2204778156996587,
      "grad_norm": 0.8339386449394393,
      "learning_rate": 1.0751249582609368e-06,
      "loss": 0.0121,
      "step": 5364
    },
    {
      "epoch": 1.2207053469852105,
      "grad_norm": 0.814490781430545,
      "learning_rate": 1.075062964697535e-06,
      "loss": 0.0169,
      "step": 5365
    },
    {
      "epoch": 1.2209328782707622,
      "grad_norm": 1.368386097389221,
      "learning_rate": 1.0750009619356706e-06,
      "loss": 0.0406,
      "step": 5366
    },
    {
      "epoch": 1.221160409556314,
      "grad_norm": 1.0996742222363038,
      "learning_rate": 1.0749389499766106e-06,
      "loss": 0.0259,
      "step": 5367
    },
    {
      "epoch": 1.2213879408418657,
      "grad_norm": 1.468334437179015,
      "learning_rate": 1.0748769288216226e-06,
      "loss": 0.0272,
      "step": 5368
    },
    {
      "epoch": 1.2216154721274175,
      "grad_norm": 1.3908865445482075,
      "learning_rate": 1.074814898471974e-06,
      "loss": 0.0461,
      "step": 5369
    },
    {
      "epoch": 1.2218430034129693,
      "grad_norm": 1.3737903263713396,
      "learning_rate": 1.0747528589289327e-06,
      "loss": 0.0109,
      "step": 5370
    },
    {
      "epoch": 1.222070534698521,
      "grad_norm": 0.9334616229837778,
      "learning_rate": 1.0746908101937666e-06,
      "loss": 0.0225,
      "step": 5371
    },
    {
      "epoch": 1.2222980659840728,
      "grad_norm": 3.4821828449919656,
      "learning_rate": 1.0746287522677439e-06,
      "loss": 0.1442,
      "step": 5372
    },
    {
      "epoch": 1.2225255972696245,
      "grad_norm": 1.1612571872192434,
      "learning_rate": 1.074566685152133e-06,
      "loss": 0.024,
      "step": 5373
    },
    {
      "epoch": 1.2227531285551763,
      "grad_norm": 0.6082887587191537,
      "learning_rate": 1.0745046088482025e-06,
      "loss": 0.0091,
      "step": 5374
    },
    {
      "epoch": 1.222980659840728,
      "grad_norm": 0.8947279579511637,
      "learning_rate": 1.074442523357221e-06,
      "loss": 0.018,
      "step": 5375
    },
    {
      "epoch": 1.2232081911262798,
      "grad_norm": 0.8578464895598875,
      "learning_rate": 1.0743804286804573e-06,
      "loss": 0.0083,
      "step": 5376
    },
    {
      "epoch": 1.2234357224118315,
      "grad_norm": 0.9832599316516126,
      "learning_rate": 1.0743183248191806e-06,
      "loss": 0.0112,
      "step": 5377
    },
    {
      "epoch": 1.2236632536973833,
      "grad_norm": 0.9942384197604868,
      "learning_rate": 1.0742562117746604e-06,
      "loss": 0.0266,
      "step": 5378
    },
    {
      "epoch": 1.2238907849829352,
      "grad_norm": 0.907525459976151,
      "learning_rate": 1.074194089548166e-06,
      "loss": 0.0182,
      "step": 5379
    },
    {
      "epoch": 1.224118316268487,
      "grad_norm": 0.6044100806377944,
      "learning_rate": 1.0741319581409667e-06,
      "loss": 0.0132,
      "step": 5380
    },
    {
      "epoch": 1.2243458475540387,
      "grad_norm": 0.7851876560740085,
      "learning_rate": 1.0740698175543332e-06,
      "loss": 0.008,
      "step": 5381
    },
    {
      "epoch": 1.2245733788395905,
      "grad_norm": 1.1184528122514146,
      "learning_rate": 1.0740076677895348e-06,
      "loss": 0.0295,
      "step": 5382
    },
    {
      "epoch": 1.2248009101251423,
      "grad_norm": 1.0110121442428808,
      "learning_rate": 1.0739455088478422e-06,
      "loss": 0.0246,
      "step": 5383
    },
    {
      "epoch": 1.225028441410694,
      "grad_norm": 1.3430003631618936,
      "learning_rate": 1.0738833407305254e-06,
      "loss": 0.0311,
      "step": 5384
    },
    {
      "epoch": 1.2252559726962458,
      "grad_norm": 0.714109028797083,
      "learning_rate": 1.0738211634388554e-06,
      "loss": 0.0137,
      "step": 5385
    },
    {
      "epoch": 1.2254835039817975,
      "grad_norm": 1.7977112803389215,
      "learning_rate": 1.0737589769741025e-06,
      "loss": 0.0311,
      "step": 5386
    },
    {
      "epoch": 1.2257110352673493,
      "grad_norm": 1.595010039888779,
      "learning_rate": 1.0736967813375382e-06,
      "loss": 0.0264,
      "step": 5387
    },
    {
      "epoch": 1.225938566552901,
      "grad_norm": 0.9599396412918755,
      "learning_rate": 1.0736345765304335e-06,
      "loss": 0.0183,
      "step": 5388
    },
    {
      "epoch": 1.2261660978384528,
      "grad_norm": 1.4002183993615684,
      "learning_rate": 1.0735723625540596e-06,
      "loss": 0.0584,
      "step": 5389
    },
    {
      "epoch": 1.2263936291240045,
      "grad_norm": 0.5289624966838866,
      "learning_rate": 1.073510139409688e-06,
      "loss": 0.0101,
      "step": 5390
    },
    {
      "epoch": 1.2266211604095563,
      "grad_norm": 1.057508406840753,
      "learning_rate": 1.0734479070985908e-06,
      "loss": 0.0142,
      "step": 5391
    },
    {
      "epoch": 1.226848691695108,
      "grad_norm": 0.7921993020441022,
      "learning_rate": 1.0733856656220396e-06,
      "loss": 0.0111,
      "step": 5392
    },
    {
      "epoch": 1.2270762229806598,
      "grad_norm": 0.7881195760466057,
      "learning_rate": 1.0733234149813065e-06,
      "loss": 0.0251,
      "step": 5393
    },
    {
      "epoch": 1.2273037542662115,
      "grad_norm": 0.7174133775125424,
      "learning_rate": 1.0732611551776639e-06,
      "loss": 0.0123,
      "step": 5394
    },
    {
      "epoch": 1.2275312855517633,
      "grad_norm": 1.2047358857096602,
      "learning_rate": 1.0731988862123841e-06,
      "loss": 0.012,
      "step": 5395
    },
    {
      "epoch": 1.2277588168373152,
      "grad_norm": 0.8067610993565308,
      "learning_rate": 1.07313660808674e-06,
      "loss": 0.0093,
      "step": 5396
    },
    {
      "epoch": 1.227986348122867,
      "grad_norm": 0.9414182769464757,
      "learning_rate": 1.0730743208020044e-06,
      "loss": 0.0138,
      "step": 5397
    },
    {
      "epoch": 1.2282138794084188,
      "grad_norm": 1.0848537565553817,
      "learning_rate": 1.0730120243594504e-06,
      "loss": 0.0235,
      "step": 5398
    },
    {
      "epoch": 1.2284414106939705,
      "grad_norm": 1.0189138352280012,
      "learning_rate": 1.0729497187603508e-06,
      "loss": 0.0264,
      "step": 5399
    },
    {
      "epoch": 1.2286689419795223,
      "grad_norm": 1.0924034497075088,
      "learning_rate": 1.0728874040059798e-06,
      "loss": 0.0115,
      "step": 5400
    },
    {
      "epoch": 1.228896473265074,
      "grad_norm": 0.6433664647544001,
      "learning_rate": 1.07282508009761e-06,
      "loss": 0.01,
      "step": 5401
    },
    {
      "epoch": 1.2291240045506258,
      "grad_norm": 0.7843372574416333,
      "learning_rate": 1.072762747036516e-06,
      "loss": 0.0215,
      "step": 5402
    },
    {
      "epoch": 1.2293515358361775,
      "grad_norm": 1.0890903190836254,
      "learning_rate": 1.0727004048239715e-06,
      "loss": 0.0272,
      "step": 5403
    },
    {
      "epoch": 1.2295790671217293,
      "grad_norm": 0.8996000171907133,
      "learning_rate": 1.0726380534612507e-06,
      "loss": 0.0171,
      "step": 5404
    },
    {
      "epoch": 1.229806598407281,
      "grad_norm": 0.8250699810825667,
      "learning_rate": 1.0725756929496277e-06,
      "loss": 0.007,
      "step": 5405
    },
    {
      "epoch": 1.2300341296928328,
      "grad_norm": 1.0264353557572934,
      "learning_rate": 1.0725133232903773e-06,
      "loss": 0.0345,
      "step": 5406
    },
    {
      "epoch": 1.2302616609783845,
      "grad_norm": 1.137616173296977,
      "learning_rate": 1.0724509444847741e-06,
      "loss": 0.036,
      "step": 5407
    },
    {
      "epoch": 1.2304891922639363,
      "grad_norm": 1.3664447826228183,
      "learning_rate": 1.0723885565340933e-06,
      "loss": 0.0285,
      "step": 5408
    },
    {
      "epoch": 1.230716723549488,
      "grad_norm": 1.1602041449374105,
      "learning_rate": 1.0723261594396095e-06,
      "loss": 0.0332,
      "step": 5409
    },
    {
      "epoch": 1.2309442548350398,
      "grad_norm": 1.090010681276093,
      "learning_rate": 1.0722637532025984e-06,
      "loss": 0.0218,
      "step": 5410
    },
    {
      "epoch": 1.2311717861205915,
      "grad_norm": 1.4413107732179307,
      "learning_rate": 1.0722013378243354e-06,
      "loss": 0.0321,
      "step": 5411
    },
    {
      "epoch": 1.2313993174061433,
      "grad_norm": 1.8979249234317928,
      "learning_rate": 1.0721389133060958e-06,
      "loss": 0.0266,
      "step": 5412
    },
    {
      "epoch": 1.231626848691695,
      "grad_norm": 1.7124518471345664,
      "learning_rate": 1.0720764796491559e-06,
      "loss": 0.0281,
      "step": 5413
    },
    {
      "epoch": 1.2318543799772468,
      "grad_norm": 1.466076682039334,
      "learning_rate": 1.0720140368547915e-06,
      "loss": 0.0445,
      "step": 5414
    },
    {
      "epoch": 1.2320819112627985,
      "grad_norm": 1.307580416194323,
      "learning_rate": 1.0719515849242787e-06,
      "loss": 0.0233,
      "step": 5415
    },
    {
      "epoch": 1.2323094425483503,
      "grad_norm": 0.8318083081384708,
      "learning_rate": 1.0718891238588943e-06,
      "loss": 0.022,
      "step": 5416
    },
    {
      "epoch": 1.232536973833902,
      "grad_norm": 1.3996049895553466,
      "learning_rate": 1.0718266536599145e-06,
      "loss": 0.046,
      "step": 5417
    },
    {
      "epoch": 1.232764505119454,
      "grad_norm": 1.4470303584385065,
      "learning_rate": 1.0717641743286163e-06,
      "loss": 0.0315,
      "step": 5418
    },
    {
      "epoch": 1.2329920364050058,
      "grad_norm": 1.251825573353438,
      "learning_rate": 1.0717016858662766e-06,
      "loss": 0.0187,
      "step": 5419
    },
    {
      "epoch": 1.2332195676905575,
      "grad_norm": 1.2025353214638472,
      "learning_rate": 1.0716391882741722e-06,
      "loss": 0.0512,
      "step": 5420
    },
    {
      "epoch": 1.2334470989761093,
      "grad_norm": 1.1125647041962026,
      "learning_rate": 1.0715766815535812e-06,
      "loss": 0.0191,
      "step": 5421
    },
    {
      "epoch": 1.233674630261661,
      "grad_norm": 1.0140086609338004,
      "learning_rate": 1.0715141657057805e-06,
      "loss": 0.0286,
      "step": 5422
    },
    {
      "epoch": 1.2339021615472128,
      "grad_norm": 1.6413125694512114,
      "learning_rate": 1.0714516407320482e-06,
      "loss": 0.0277,
      "step": 5423
    },
    {
      "epoch": 1.2341296928327645,
      "grad_norm": 1.113912252435204,
      "learning_rate": 1.0713891066336619e-06,
      "loss": 0.0459,
      "step": 5424
    },
    {
      "epoch": 1.2343572241183163,
      "grad_norm": 1.4211151899069139,
      "learning_rate": 1.0713265634118998e-06,
      "loss": 0.03,
      "step": 5425
    },
    {
      "epoch": 1.234584755403868,
      "grad_norm": 0.5439788203180792,
      "learning_rate": 1.0712640110680398e-06,
      "loss": 0.0051,
      "step": 5426
    },
    {
      "epoch": 1.2348122866894198,
      "grad_norm": 1.1066572512164194,
      "learning_rate": 1.071201449603361e-06,
      "loss": 0.031,
      "step": 5427
    },
    {
      "epoch": 1.2350398179749715,
      "grad_norm": 1.0734688953953,
      "learning_rate": 1.0711388790191418e-06,
      "loss": 0.0159,
      "step": 5428
    },
    {
      "epoch": 1.2352673492605233,
      "grad_norm": 0.9674963008176792,
      "learning_rate": 1.071076299316661e-06,
      "loss": 0.0292,
      "step": 5429
    },
    {
      "epoch": 1.235494880546075,
      "grad_norm": 0.8648315211359087,
      "learning_rate": 1.0710137104971973e-06,
      "loss": 0.015,
      "step": 5430
    },
    {
      "epoch": 1.2357224118316268,
      "grad_norm": 1.0177696413333797,
      "learning_rate": 1.0709511125620306e-06,
      "loss": 0.0187,
      "step": 5431
    },
    {
      "epoch": 1.2359499431171785,
      "grad_norm": 1.080479875370983,
      "learning_rate": 1.0708885055124396e-06,
      "loss": 0.0202,
      "step": 5432
    },
    {
      "epoch": 1.2361774744027303,
      "grad_norm": 1.2213741714117063,
      "learning_rate": 1.0708258893497043e-06,
      "loss": 0.0347,
      "step": 5433
    },
    {
      "epoch": 1.236405005688282,
      "grad_norm": 0.6102850699242401,
      "learning_rate": 1.0707632640751042e-06,
      "loss": 0.011,
      "step": 5434
    },
    {
      "epoch": 1.236632536973834,
      "grad_norm": 0.7162928033199788,
      "learning_rate": 1.0707006296899194e-06,
      "loss": 0.0104,
      "step": 5435
    },
    {
      "epoch": 1.2368600682593858,
      "grad_norm": 1.6643057799440983,
      "learning_rate": 1.0706379861954299e-06,
      "loss": 0.0474,
      "step": 5436
    },
    {
      "epoch": 1.2370875995449375,
      "grad_norm": 1.0636970768259353,
      "learning_rate": 1.0705753335929162e-06,
      "loss": 0.0441,
      "step": 5437
    },
    {
      "epoch": 1.2373151308304893,
      "grad_norm": 0.5899981023336623,
      "learning_rate": 1.070512671883659e-06,
      "loss": 0.0081,
      "step": 5438
    },
    {
      "epoch": 1.237542662116041,
      "grad_norm": 1.1358807681666672,
      "learning_rate": 1.0704500010689383e-06,
      "loss": 0.0231,
      "step": 5439
    },
    {
      "epoch": 1.2377701934015928,
      "grad_norm": 1.483924786030886,
      "learning_rate": 1.0703873211500356e-06,
      "loss": 0.0434,
      "step": 5440
    },
    {
      "epoch": 1.2379977246871445,
      "grad_norm": 0.5564175099477741,
      "learning_rate": 1.0703246321282316e-06,
      "loss": 0.0111,
      "step": 5441
    },
    {
      "epoch": 1.2382252559726963,
      "grad_norm": 0.6919558166199928,
      "learning_rate": 1.0702619340048077e-06,
      "loss": 0.0107,
      "step": 5442
    },
    {
      "epoch": 1.238452787258248,
      "grad_norm": 0.916701734430617,
      "learning_rate": 1.0701992267810454e-06,
      "loss": 0.0095,
      "step": 5443
    },
    {
      "epoch": 1.2386803185437998,
      "grad_norm": 0.6806190525241188,
      "learning_rate": 1.0701365104582262e-06,
      "loss": 0.0166,
      "step": 5444
    },
    {
      "epoch": 1.2389078498293515,
      "grad_norm": 0.9812678457663916,
      "learning_rate": 1.070073785037632e-06,
      "loss": 0.0179,
      "step": 5445
    },
    {
      "epoch": 1.2391353811149033,
      "grad_norm": 1.2184326624890787,
      "learning_rate": 1.0700110505205447e-06,
      "loss": 0.0285,
      "step": 5446
    },
    {
      "epoch": 1.239362912400455,
      "grad_norm": 0.9122394182634046,
      "learning_rate": 1.0699483069082468e-06,
      "loss": 0.0206,
      "step": 5447
    },
    {
      "epoch": 1.2395904436860068,
      "grad_norm": 1.776304008174226,
      "learning_rate": 1.0698855542020201e-06,
      "loss": 0.0406,
      "step": 5448
    },
    {
      "epoch": 1.2398179749715585,
      "grad_norm": 2.144224863401225,
      "learning_rate": 1.0698227924031474e-06,
      "loss": 0.0269,
      "step": 5449
    },
    {
      "epoch": 1.2400455062571103,
      "grad_norm": 0.8197781138940033,
      "learning_rate": 1.0697600215129113e-06,
      "loss": 0.0132,
      "step": 5450
    },
    {
      "epoch": 1.240273037542662,
      "grad_norm": 1.112356089568793,
      "learning_rate": 1.0696972415325954e-06,
      "loss": 0.0188,
      "step": 5451
    },
    {
      "epoch": 1.2405005688282138,
      "grad_norm": 1.5519612957572413,
      "learning_rate": 1.069634452463482e-06,
      "loss": 0.0649,
      "step": 5452
    },
    {
      "epoch": 1.2407281001137656,
      "grad_norm": 1.7090640826753272,
      "learning_rate": 1.0695716543068548e-06,
      "loss": 0.021,
      "step": 5453
    },
    {
      "epoch": 1.2409556313993173,
      "grad_norm": 0.9894882585365135,
      "learning_rate": 1.0695088470639973e-06,
      "loss": 0.031,
      "step": 5454
    },
    {
      "epoch": 1.241183162684869,
      "grad_norm": 1.2111142205989058,
      "learning_rate": 1.069446030736193e-06,
      "loss": 0.0328,
      "step": 5455
    },
    {
      "epoch": 1.2414106939704208,
      "grad_norm": 0.8409351330327227,
      "learning_rate": 1.0693832053247256e-06,
      "loss": 0.0147,
      "step": 5456
    },
    {
      "epoch": 1.2416382252559728,
      "grad_norm": 1.2297382097518612,
      "learning_rate": 1.0693203708308792e-06,
      "loss": 0.018,
      "step": 5457
    },
    {
      "epoch": 1.2418657565415245,
      "grad_norm": 0.7856520845586189,
      "learning_rate": 1.0692575272559385e-06,
      "loss": 0.0132,
      "step": 5458
    },
    {
      "epoch": 1.2420932878270763,
      "grad_norm": 2.9495226963416092,
      "learning_rate": 1.0691946746011874e-06,
      "loss": 0.0753,
      "step": 5459
    },
    {
      "epoch": 1.242320819112628,
      "grad_norm": 1.1311124384842985,
      "learning_rate": 1.0691318128679107e-06,
      "loss": 0.0282,
      "step": 5460
    },
    {
      "epoch": 1.2425483503981798,
      "grad_norm": 1.2641663291780139,
      "learning_rate": 1.0690689420573933e-06,
      "loss": 0.0359,
      "step": 5461
    },
    {
      "epoch": 1.2427758816837315,
      "grad_norm": 1.5841963404042003,
      "learning_rate": 1.0690060621709198e-06,
      "loss": 0.0334,
      "step": 5462
    },
    {
      "epoch": 1.2430034129692833,
      "grad_norm": 0.7353352974419007,
      "learning_rate": 1.0689431732097754e-06,
      "loss": 0.0272,
      "step": 5463
    },
    {
      "epoch": 1.243230944254835,
      "grad_norm": 1.0500642122401156,
      "learning_rate": 1.0688802751752458e-06,
      "loss": 0.0275,
      "step": 5464
    },
    {
      "epoch": 1.2434584755403868,
      "grad_norm": 0.7175451483028665,
      "learning_rate": 1.0688173680686164e-06,
      "loss": 0.0114,
      "step": 5465
    },
    {
      "epoch": 1.2436860068259386,
      "grad_norm": 0.5867289910981424,
      "learning_rate": 1.0687544518911726e-06,
      "loss": 0.0111,
      "step": 5466
    },
    {
      "epoch": 1.2439135381114903,
      "grad_norm": 0.6712196944383615,
      "learning_rate": 1.0686915266442005e-06,
      "loss": 0.0162,
      "step": 5467
    },
    {
      "epoch": 1.244141069397042,
      "grad_norm": 1.2463644416881146,
      "learning_rate": 1.0686285923289863e-06,
      "loss": 0.0442,
      "step": 5468
    },
    {
      "epoch": 1.2443686006825938,
      "grad_norm": 0.8847672705631197,
      "learning_rate": 1.0685656489468161e-06,
      "loss": 0.0137,
      "step": 5469
    },
    {
      "epoch": 1.2445961319681456,
      "grad_norm": 0.8604702504051126,
      "learning_rate": 1.0685026964989764e-06,
      "loss": 0.0168,
      "step": 5470
    },
    {
      "epoch": 1.2448236632536973,
      "grad_norm": 1.418689898486085,
      "learning_rate": 1.0684397349867537e-06,
      "loss": 0.0292,
      "step": 5471
    },
    {
      "epoch": 1.245051194539249,
      "grad_norm": 1.846189023146859,
      "learning_rate": 1.068376764411435e-06,
      "loss": 0.0448,
      "step": 5472
    },
    {
      "epoch": 1.2452787258248008,
      "grad_norm": 1.252366571384877,
      "learning_rate": 1.0683137847743076e-06,
      "loss": 0.0301,
      "step": 5473
    },
    {
      "epoch": 1.2455062571103528,
      "grad_norm": 1.2815205403700491,
      "learning_rate": 1.0682507960766578e-06,
      "loss": 0.0317,
      "step": 5474
    },
    {
      "epoch": 1.2457337883959045,
      "grad_norm": 1.5712499828336253,
      "learning_rate": 1.0681877983197738e-06,
      "loss": 0.0511,
      "step": 5475
    },
    {
      "epoch": 1.2459613196814563,
      "grad_norm": 1.3120502665157674,
      "learning_rate": 1.0681247915049428e-06,
      "loss": 0.0239,
      "step": 5476
    },
    {
      "epoch": 1.246188850967008,
      "grad_norm": 1.2372734878283551,
      "learning_rate": 1.0680617756334527e-06,
      "loss": 0.025,
      "step": 5477
    },
    {
      "epoch": 1.2464163822525598,
      "grad_norm": 1.1291827306006323,
      "learning_rate": 1.0679987507065912e-06,
      "loss": 0.0155,
      "step": 5478
    },
    {
      "epoch": 1.2466439135381115,
      "grad_norm": 1.1166618335200997,
      "learning_rate": 1.0679357167256465e-06,
      "loss": 0.015,
      "step": 5479
    },
    {
      "epoch": 1.2468714448236633,
      "grad_norm": 0.9525747638788947,
      "learning_rate": 1.067872673691907e-06,
      "loss": 0.0379,
      "step": 5480
    },
    {
      "epoch": 1.247098976109215,
      "grad_norm": 0.9015026474953439,
      "learning_rate": 1.0678096216066611e-06,
      "loss": 0.0171,
      "step": 5481
    },
    {
      "epoch": 1.2473265073947668,
      "grad_norm": 0.7251387961734896,
      "learning_rate": 1.0677465604711975e-06,
      "loss": 0.0183,
      "step": 5482
    },
    {
      "epoch": 1.2475540386803186,
      "grad_norm": 1.1041778984913362,
      "learning_rate": 1.0676834902868051e-06,
      "loss": 0.0449,
      "step": 5483
    },
    {
      "epoch": 1.2477815699658703,
      "grad_norm": 1.5486690407695638,
      "learning_rate": 1.067620411054773e-06,
      "loss": 0.0559,
      "step": 5484
    },
    {
      "epoch": 1.248009101251422,
      "grad_norm": 1.0936433230281484,
      "learning_rate": 1.0675573227763903e-06,
      "loss": 0.041,
      "step": 5485
    },
    {
      "epoch": 1.2482366325369738,
      "grad_norm": 0.7723592633898496,
      "learning_rate": 1.0674942254529463e-06,
      "loss": 0.0196,
      "step": 5486
    },
    {
      "epoch": 1.2484641638225256,
      "grad_norm": 1.3114855555869747,
      "learning_rate": 1.0674311190857308e-06,
      "loss": 0.0231,
      "step": 5487
    },
    {
      "epoch": 1.2486916951080773,
      "grad_norm": 1.781546684980227,
      "learning_rate": 1.0673680036760333e-06,
      "loss": 0.0336,
      "step": 5488
    },
    {
      "epoch": 1.248919226393629,
      "grad_norm": 1.4863914973995318,
      "learning_rate": 1.0673048792251443e-06,
      "loss": 0.0284,
      "step": 5489
    },
    {
      "epoch": 1.2491467576791808,
      "grad_norm": 0.7255040029884252,
      "learning_rate": 1.0672417457343535e-06,
      "loss": 0.0119,
      "step": 5490
    },
    {
      "epoch": 1.2493742889647326,
      "grad_norm": 1.1732515422757923,
      "learning_rate": 1.0671786032049512e-06,
      "loss": 0.0247,
      "step": 5491
    },
    {
      "epoch": 1.2496018202502843,
      "grad_norm": 0.8454940508394108,
      "learning_rate": 1.0671154516382283e-06,
      "loss": 0.0126,
      "step": 5492
    },
    {
      "epoch": 1.249829351535836,
      "grad_norm": 1.7717426170006738,
      "learning_rate": 1.067052291035475e-06,
      "loss": 0.0288,
      "step": 5493
    },
    {
      "epoch": 1.2500568828213878,
      "grad_norm": 0.9733708679345942,
      "learning_rate": 1.0669891213979826e-06,
      "loss": 0.0424,
      "step": 5494
    },
    {
      "epoch": 1.2502844141069396,
      "grad_norm": 0.8099462019509774,
      "learning_rate": 1.066925942727042e-06,
      "loss": 0.0135,
      "step": 5495
    },
    {
      "epoch": 1.2505119453924913,
      "grad_norm": 1.0927120951454266,
      "learning_rate": 1.0668627550239444e-06,
      "loss": 0.02,
      "step": 5496
    },
    {
      "epoch": 1.2507394766780433,
      "grad_norm": 0.9823491330955677,
      "learning_rate": 1.0667995582899815e-06,
      "loss": 0.0117,
      "step": 5497
    },
    {
      "epoch": 1.250967007963595,
      "grad_norm": 1.3444249454765562,
      "learning_rate": 1.0667363525264446e-06,
      "loss": 0.0272,
      "step": 5498
    },
    {
      "epoch": 1.2511945392491468,
      "grad_norm": 1.5794185821366846,
      "learning_rate": 1.0666731377346257e-06,
      "loss": 0.0429,
      "step": 5499
    },
    {
      "epoch": 1.2514220705346986,
      "grad_norm": 0.827330081267108,
      "learning_rate": 1.0666099139158168e-06,
      "loss": 0.0133,
      "step": 5500
    },
    {
      "epoch": 1.2516496018202503,
      "grad_norm": 1.5562286915608332,
      "learning_rate": 1.06654668107131e-06,
      "loss": 0.037,
      "step": 5501
    },
    {
      "epoch": 1.251877133105802,
      "grad_norm": 1.8342841703579758,
      "learning_rate": 1.0664834392023975e-06,
      "loss": 0.0369,
      "step": 5502
    },
    {
      "epoch": 1.2521046643913538,
      "grad_norm": 1.9754022342502022,
      "learning_rate": 1.0664201883103722e-06,
      "loss": 0.0576,
      "step": 5503
    },
    {
      "epoch": 1.2523321956769056,
      "grad_norm": 1.704003080347443,
      "learning_rate": 1.066356928396527e-06,
      "loss": 0.041,
      "step": 5504
    },
    {
      "epoch": 1.2525597269624573,
      "grad_norm": 0.8152446360977106,
      "learning_rate": 1.066293659462154e-06,
      "loss": 0.0092,
      "step": 5505
    },
    {
      "epoch": 1.252787258248009,
      "grad_norm": 1.187665186303167,
      "learning_rate": 1.066230381508547e-06,
      "loss": 0.03,
      "step": 5506
    },
    {
      "epoch": 1.2530147895335608,
      "grad_norm": 1.515571401940605,
      "learning_rate": 1.0661670945369991e-06,
      "loss": 0.0658,
      "step": 5507
    },
    {
      "epoch": 1.2532423208191126,
      "grad_norm": 0.7829970004154015,
      "learning_rate": 1.0661037985488037e-06,
      "loss": 0.0221,
      "step": 5508
    },
    {
      "epoch": 1.2534698521046643,
      "grad_norm": 0.9344601178136276,
      "learning_rate": 1.0660404935452545e-06,
      "loss": 0.0176,
      "step": 5509
    },
    {
      "epoch": 1.253697383390216,
      "grad_norm": 1.1741631037543796,
      "learning_rate": 1.0659771795276451e-06,
      "loss": 0.0383,
      "step": 5510
    },
    {
      "epoch": 1.253924914675768,
      "grad_norm": 0.6952087144812199,
      "learning_rate": 1.06591385649727e-06,
      "loss": 0.0166,
      "step": 5511
    },
    {
      "epoch": 1.2541524459613198,
      "grad_norm": 1.148195580887031,
      "learning_rate": 1.0658505244554233e-06,
      "loss": 0.0336,
      "step": 5512
    },
    {
      "epoch": 1.2543799772468716,
      "grad_norm": 0.9686864698151374,
      "learning_rate": 1.0657871834033992e-06,
      "loss": 0.017,
      "step": 5513
    },
    {
      "epoch": 1.2546075085324233,
      "grad_norm": 0.6541574267823076,
      "learning_rate": 1.0657238333424922e-06,
      "loss": 0.0149,
      "step": 5514
    },
    {
      "epoch": 1.254835039817975,
      "grad_norm": 0.682956550720555,
      "learning_rate": 1.0656604742739974e-06,
      "loss": 0.0106,
      "step": 5515
    },
    {
      "epoch": 1.2550625711035268,
      "grad_norm": 1.6100348965042812,
      "learning_rate": 1.0655971061992093e-06,
      "loss": 0.0445,
      "step": 5516
    },
    {
      "epoch": 1.2552901023890786,
      "grad_norm": 1.1979735909844647,
      "learning_rate": 1.0655337291194235e-06,
      "loss": 0.0369,
      "step": 5517
    },
    {
      "epoch": 1.2555176336746303,
      "grad_norm": 0.7815830769361026,
      "learning_rate": 1.0654703430359348e-06,
      "loss": 0.0231,
      "step": 5518
    },
    {
      "epoch": 1.255745164960182,
      "grad_norm": 0.6929286637595476,
      "learning_rate": 1.065406947950039e-06,
      "loss": 0.0106,
      "step": 5519
    },
    {
      "epoch": 1.2559726962457338,
      "grad_norm": 1.1236746085503946,
      "learning_rate": 1.065343543863032e-06,
      "loss": 0.0344,
      "step": 5520
    },
    {
      "epoch": 1.2562002275312856,
      "grad_norm": 0.570375538462011,
      "learning_rate": 1.0652801307762093e-06,
      "loss": 0.0154,
      "step": 5521
    },
    {
      "epoch": 1.2564277588168373,
      "grad_norm": 1.182265579469843,
      "learning_rate": 1.065216708690867e-06,
      "loss": 0.0251,
      "step": 5522
    },
    {
      "epoch": 1.256655290102389,
      "grad_norm": 1.0748877350333434,
      "learning_rate": 1.0651532776083014e-06,
      "loss": 0.0252,
      "step": 5523
    },
    {
      "epoch": 1.2568828213879408,
      "grad_norm": 0.6801381732217795,
      "learning_rate": 1.0650898375298088e-06,
      "loss": 0.0175,
      "step": 5524
    },
    {
      "epoch": 1.2571103526734926,
      "grad_norm": 0.8236115723758932,
      "learning_rate": 1.0650263884566863e-06,
      "loss": 0.0122,
      "step": 5525
    },
    {
      "epoch": 1.2573378839590443,
      "grad_norm": 1.2072865539422744,
      "learning_rate": 1.06496293039023e-06,
      "loss": 0.0307,
      "step": 5526
    },
    {
      "epoch": 1.257565415244596,
      "grad_norm": 0.8638078889384445,
      "learning_rate": 1.0648994633317373e-06,
      "loss": 0.0338,
      "step": 5527
    },
    {
      "epoch": 1.2577929465301478,
      "grad_norm": 1.3903600138449066,
      "learning_rate": 1.064835987282505e-06,
      "loss": 0.0427,
      "step": 5528
    },
    {
      "epoch": 1.2580204778156996,
      "grad_norm": 1.2546291636301692,
      "learning_rate": 1.0647725022438307e-06,
      "loss": 0.0159,
      "step": 5529
    },
    {
      "epoch": 1.2582480091012513,
      "grad_norm": 0.756663737422429,
      "learning_rate": 1.0647090082170118e-06,
      "loss": 0.013,
      "step": 5530
    },
    {
      "epoch": 1.258475540386803,
      "grad_norm": 0.8566876022289612,
      "learning_rate": 1.0646455052033463e-06,
      "loss": 0.0201,
      "step": 5531
    },
    {
      "epoch": 1.2587030716723548,
      "grad_norm": 1.1638253388866646,
      "learning_rate": 1.0645819932041317e-06,
      "loss": 0.0166,
      "step": 5532
    },
    {
      "epoch": 1.2589306029579066,
      "grad_norm": 1.1215067967587442,
      "learning_rate": 1.064518472220666e-06,
      "loss": 0.0138,
      "step": 5533
    },
    {
      "epoch": 1.2591581342434583,
      "grad_norm": 1.2121221098665633,
      "learning_rate": 1.064454942254248e-06,
      "loss": 0.0214,
      "step": 5534
    },
    {
      "epoch": 1.25938566552901,
      "grad_norm": 0.7454316468560616,
      "learning_rate": 1.0643914033061757e-06,
      "loss": 0.0129,
      "step": 5535
    },
    {
      "epoch": 1.259613196814562,
      "grad_norm": 1.7146758293381863,
      "learning_rate": 1.0643278553777477e-06,
      "loss": 0.0243,
      "step": 5536
    },
    {
      "epoch": 1.2598407281001138,
      "grad_norm": 32.10042100778171,
      "learning_rate": 1.0642642984702632e-06,
      "loss": 0.4248,
      "step": 5537
    },
    {
      "epoch": 1.2600682593856656,
      "grad_norm": 1.0484198163247316,
      "learning_rate": 1.0642007325850207e-06,
      "loss": 0.0287,
      "step": 5538
    },
    {
      "epoch": 1.2602957906712173,
      "grad_norm": 2.3948890958568487,
      "learning_rate": 1.0641371577233197e-06,
      "loss": 0.0669,
      "step": 5539
    },
    {
      "epoch": 1.260523321956769,
      "grad_norm": 0.9801146567578284,
      "learning_rate": 1.0640735738864594e-06,
      "loss": 0.0162,
      "step": 5540
    },
    {
      "epoch": 1.2607508532423208,
      "grad_norm": 0.9922579714313251,
      "learning_rate": 1.0640099810757394e-06,
      "loss": 0.0181,
      "step": 5541
    },
    {
      "epoch": 1.2609783845278726,
      "grad_norm": 0.8815528286158882,
      "learning_rate": 1.0639463792924592e-06,
      "loss": 0.0217,
      "step": 5542
    },
    {
      "epoch": 1.2612059158134243,
      "grad_norm": 0.8238976020134005,
      "learning_rate": 1.0638827685379191e-06,
      "loss": 0.0145,
      "step": 5543
    },
    {
      "epoch": 1.261433447098976,
      "grad_norm": 1.1437101469135258,
      "learning_rate": 1.063819148813419e-06,
      "loss": 0.0248,
      "step": 5544
    },
    {
      "epoch": 1.2616609783845278,
      "grad_norm": 2.5729371322737475,
      "learning_rate": 1.063755520120259e-06,
      "loss": 0.0719,
      "step": 5545
    },
    {
      "epoch": 1.2618885096700796,
      "grad_norm": 0.742178098402897,
      "learning_rate": 1.0636918824597397e-06,
      "loss": 0.011,
      "step": 5546
    },
    {
      "epoch": 1.2621160409556313,
      "grad_norm": 1.2993072879022185,
      "learning_rate": 1.063628235833162e-06,
      "loss": 0.0127,
      "step": 5547
    },
    {
      "epoch": 1.262343572241183,
      "grad_norm": 1.050947969366834,
      "learning_rate": 1.0635645802418263e-06,
      "loss": 0.0203,
      "step": 5548
    },
    {
      "epoch": 1.2625711035267349,
      "grad_norm": 1.0789000058560003,
      "learning_rate": 1.0635009156870338e-06,
      "loss": 0.0292,
      "step": 5549
    },
    {
      "epoch": 1.2627986348122868,
      "grad_norm": 0.7771290887161867,
      "learning_rate": 1.0634372421700858e-06,
      "loss": 0.0081,
      "step": 5550
    },
    {
      "epoch": 1.2630261660978386,
      "grad_norm": 1.1448879442832878,
      "learning_rate": 1.0633735596922834e-06,
      "loss": 0.0215,
      "step": 5551
    },
    {
      "epoch": 1.2632536973833903,
      "grad_norm": 0.5229559835200444,
      "learning_rate": 1.0633098682549282e-06,
      "loss": 0.0104,
      "step": 5552
    },
    {
      "epoch": 1.263481228668942,
      "grad_norm": 0.9799373836364904,
      "learning_rate": 1.063246167859322e-06,
      "loss": 0.0282,
      "step": 5553
    },
    {
      "epoch": 1.2637087599544938,
      "grad_norm": 1.5057299070563284,
      "learning_rate": 1.0631824585067668e-06,
      "loss": 0.0223,
      "step": 5554
    },
    {
      "epoch": 1.2639362912400456,
      "grad_norm": 0.8319291390882159,
      "learning_rate": 1.0631187401985647e-06,
      "loss": 0.028,
      "step": 5555
    },
    {
      "epoch": 1.2641638225255973,
      "grad_norm": 1.0598953809242282,
      "learning_rate": 1.0630550129360179e-06,
      "loss": 0.0215,
      "step": 5556
    },
    {
      "epoch": 1.264391353811149,
      "grad_norm": 0.9828596775853349,
      "learning_rate": 1.062991276720429e-06,
      "loss": 0.0207,
      "step": 5557
    },
    {
      "epoch": 1.2646188850967008,
      "grad_norm": 0.7615549939570654,
      "learning_rate": 1.0629275315531005e-06,
      "loss": 0.0103,
      "step": 5558
    },
    {
      "epoch": 1.2648464163822526,
      "grad_norm": 1.4943347768705173,
      "learning_rate": 1.0628637774353351e-06,
      "loss": 0.0393,
      "step": 5559
    },
    {
      "epoch": 1.2650739476678043,
      "grad_norm": 0.8670381526948084,
      "learning_rate": 1.062800014368436e-06,
      "loss": 0.0151,
      "step": 5560
    },
    {
      "epoch": 1.265301478953356,
      "grad_norm": 0.9076348935605973,
      "learning_rate": 1.0627362423537065e-06,
      "loss": 0.023,
      "step": 5561
    },
    {
      "epoch": 1.2655290102389078,
      "grad_norm": 2.452173267393995,
      "learning_rate": 1.06267246139245e-06,
      "loss": 0.0468,
      "step": 5562
    },
    {
      "epoch": 1.2657565415244596,
      "grad_norm": 2.285019099369935,
      "learning_rate": 1.06260867148597e-06,
      "loss": 0.0157,
      "step": 5563
    },
    {
      "epoch": 1.2659840728100114,
      "grad_norm": 1.5281975862148434,
      "learning_rate": 1.06254487263557e-06,
      "loss": 0.032,
      "step": 5564
    },
    {
      "epoch": 1.266211604095563,
      "grad_norm": 1.3368814016585056,
      "learning_rate": 1.062481064842554e-06,
      "loss": 0.0187,
      "step": 5565
    },
    {
      "epoch": 1.2664391353811149,
      "grad_norm": 1.4457536111569913,
      "learning_rate": 1.0624172481082265e-06,
      "loss": 0.0314,
      "step": 5566
    },
    {
      "epoch": 1.2666666666666666,
      "grad_norm": 0.8433976153186795,
      "learning_rate": 1.0623534224338916e-06,
      "loss": 0.0221,
      "step": 5567
    },
    {
      "epoch": 1.2668941979522184,
      "grad_norm": 0.7106818851766153,
      "learning_rate": 1.0622895878208535e-06,
      "loss": 0.0105,
      "step": 5568
    },
    {
      "epoch": 1.2671217292377701,
      "grad_norm": 1.7306335155218464,
      "learning_rate": 1.0622257442704174e-06,
      "loss": 0.0545,
      "step": 5569
    },
    {
      "epoch": 1.2673492605233219,
      "grad_norm": 1.2056459642628792,
      "learning_rate": 1.0621618917838874e-06,
      "loss": 0.0328,
      "step": 5570
    },
    {
      "epoch": 1.2675767918088736,
      "grad_norm": 1.0152701077745823,
      "learning_rate": 1.062098030362569e-06,
      "loss": 0.0206,
      "step": 5571
    },
    {
      "epoch": 1.2678043230944254,
      "grad_norm": 1.3569352741344352,
      "learning_rate": 1.0620341600077675e-06,
      "loss": 0.0202,
      "step": 5572
    },
    {
      "epoch": 1.2680318543799771,
      "grad_norm": 1.2714138176681336,
      "learning_rate": 1.0619702807207881e-06,
      "loss": 0.0223,
      "step": 5573
    },
    {
      "epoch": 1.268259385665529,
      "grad_norm": 1.1902953540344818,
      "learning_rate": 1.0619063925029367e-06,
      "loss": 0.0184,
      "step": 5574
    },
    {
      "epoch": 1.2684869169510808,
      "grad_norm": 0.8099951263697308,
      "learning_rate": 1.0618424953555186e-06,
      "loss": 0.0262,
      "step": 5575
    },
    {
      "epoch": 1.2687144482366326,
      "grad_norm": 0.7670050924473597,
      "learning_rate": 1.0617785892798399e-06,
      "loss": 0.0179,
      "step": 5576
    },
    {
      "epoch": 1.2689419795221843,
      "grad_norm": 1.2166230545046535,
      "learning_rate": 1.061714674277207e-06,
      "loss": 0.0272,
      "step": 5577
    },
    {
      "epoch": 1.269169510807736,
      "grad_norm": 1.4328861976608087,
      "learning_rate": 1.0616507503489255e-06,
      "loss": 0.0381,
      "step": 5578
    },
    {
      "epoch": 1.2693970420932879,
      "grad_norm": 1.346759860874677,
      "learning_rate": 1.0615868174963025e-06,
      "loss": 0.0239,
      "step": 5579
    },
    {
      "epoch": 1.2696245733788396,
      "grad_norm": 1.0046272104433696,
      "learning_rate": 1.0615228757206448e-06,
      "loss": 0.0192,
      "step": 5580
    },
    {
      "epoch": 1.2698521046643914,
      "grad_norm": 1.3021385069582194,
      "learning_rate": 1.0614589250232588e-06,
      "loss": 0.036,
      "step": 5581
    },
    {
      "epoch": 1.270079635949943,
      "grad_norm": 1.032731442532869,
      "learning_rate": 1.0613949654054518e-06,
      "loss": 0.0166,
      "step": 5582
    },
    {
      "epoch": 1.2703071672354949,
      "grad_norm": 1.2618108062315003,
      "learning_rate": 1.0613309968685308e-06,
      "loss": 0.0239,
      "step": 5583
    },
    {
      "epoch": 1.2705346985210466,
      "grad_norm": 0.6758570490387608,
      "learning_rate": 1.0612670194138033e-06,
      "loss": 0.0098,
      "step": 5584
    },
    {
      "epoch": 1.2707622298065984,
      "grad_norm": 1.114733408671628,
      "learning_rate": 1.061203033042577e-06,
      "loss": 0.0234,
      "step": 5585
    },
    {
      "epoch": 1.2709897610921501,
      "grad_norm": 1.2312044813034186,
      "learning_rate": 1.0611390377561596e-06,
      "loss": 0.0259,
      "step": 5586
    },
    {
      "epoch": 1.2712172923777019,
      "grad_norm": 0.9957995441313681,
      "learning_rate": 1.0610750335558589e-06,
      "loss": 0.0211,
      "step": 5587
    },
    {
      "epoch": 1.2714448236632536,
      "grad_norm": 0.8395539672302367,
      "learning_rate": 1.0610110204429832e-06,
      "loss": 0.0297,
      "step": 5588
    },
    {
      "epoch": 1.2716723549488056,
      "grad_norm": 1.3385709006255209,
      "learning_rate": 1.060946998418841e-06,
      "loss": 0.0416,
      "step": 5589
    },
    {
      "epoch": 1.2718998862343573,
      "grad_norm": 0.7572806427176023,
      "learning_rate": 1.06088296748474e-06,
      "loss": 0.0125,
      "step": 5590
    },
    {
      "epoch": 1.272127417519909,
      "grad_norm": 0.8812091759388678,
      "learning_rate": 1.0608189276419898e-06,
      "loss": 0.0181,
      "step": 5591
    },
    {
      "epoch": 1.2723549488054609,
      "grad_norm": 1.0677368834859697,
      "learning_rate": 1.0607548788918989e-06,
      "loss": 0.0203,
      "step": 5592
    },
    {
      "epoch": 1.2725824800910126,
      "grad_norm": 1.2209088913629613,
      "learning_rate": 1.0606908212357764e-06,
      "loss": 0.0225,
      "step": 5593
    },
    {
      "epoch": 1.2728100113765644,
      "grad_norm": 0.7838470533440405,
      "learning_rate": 1.0606267546749312e-06,
      "loss": 0.0148,
      "step": 5594
    },
    {
      "epoch": 1.273037542662116,
      "grad_norm": 0.7993976759347979,
      "learning_rate": 1.0605626792106729e-06,
      "loss": 0.0127,
      "step": 5595
    },
    {
      "epoch": 1.2732650739476679,
      "grad_norm": 1.0529954663721204,
      "learning_rate": 1.0604985948443115e-06,
      "loss": 0.0214,
      "step": 5596
    },
    {
      "epoch": 1.2734926052332196,
      "grad_norm": 1.3398092166838957,
      "learning_rate": 1.0604345015771561e-06,
      "loss": 0.0268,
      "step": 5597
    },
    {
      "epoch": 1.2737201365187714,
      "grad_norm": 1.0634782276914523,
      "learning_rate": 1.060370399410517e-06,
      "loss": 0.0231,
      "step": 5598
    },
    {
      "epoch": 1.2739476678043231,
      "grad_norm": 0.989167972365425,
      "learning_rate": 1.0603062883457044e-06,
      "loss": 0.0224,
      "step": 5599
    },
    {
      "epoch": 1.2741751990898749,
      "grad_norm": 0.9634331289782628,
      "learning_rate": 1.0602421683840283e-06,
      "loss": 0.0132,
      "step": 5600
    },
    {
      "epoch": 1.2744027303754266,
      "grad_norm": 9.022054267189066,
      "learning_rate": 1.0601780395267997e-06,
      "loss": 0.2103,
      "step": 5601
    },
    {
      "epoch": 1.2746302616609784,
      "grad_norm": 1.1619135610982998,
      "learning_rate": 1.0601139017753286e-06,
      "loss": 0.029,
      "step": 5602
    },
    {
      "epoch": 1.2748577929465301,
      "grad_norm": 1.4434474276395861,
      "learning_rate": 1.0600497551309263e-06,
      "loss": 0.033,
      "step": 5603
    },
    {
      "epoch": 1.2750853242320819,
      "grad_norm": 1.1497890055189814,
      "learning_rate": 1.0599855995949038e-06,
      "loss": 0.0391,
      "step": 5604
    },
    {
      "epoch": 1.2753128555176336,
      "grad_norm": 1.6021652391632355,
      "learning_rate": 1.0599214351685724e-06,
      "loss": 0.0319,
      "step": 5605
    },
    {
      "epoch": 1.2755403868031854,
      "grad_norm": 1.1732970121103377,
      "learning_rate": 1.0598572618532433e-06,
      "loss": 0.0181,
      "step": 5606
    },
    {
      "epoch": 1.2757679180887371,
      "grad_norm": 1.3340500937166788,
      "learning_rate": 1.0597930796502282e-06,
      "loss": 0.0256,
      "step": 5607
    },
    {
      "epoch": 1.2759954493742889,
      "grad_norm": 1.2579949420508676,
      "learning_rate": 1.059728888560839e-06,
      "loss": 0.0319,
      "step": 5608
    },
    {
      "epoch": 1.2762229806598406,
      "grad_norm": 1.0747003018755463,
      "learning_rate": 1.059664688586387e-06,
      "loss": 0.0215,
      "step": 5609
    },
    {
      "epoch": 1.2764505119453924,
      "grad_norm": 1.2386685548422143,
      "learning_rate": 1.0596004797281853e-06,
      "loss": 0.0392,
      "step": 5610
    },
    {
      "epoch": 1.2766780432309441,
      "grad_norm": 1.5312108773003696,
      "learning_rate": 1.0595362619875455e-06,
      "loss": 0.0187,
      "step": 5611
    },
    {
      "epoch": 1.276905574516496,
      "grad_norm": 1.027132364578428,
      "learning_rate": 1.0594720353657802e-06,
      "loss": 0.0254,
      "step": 5612
    },
    {
      "epoch": 1.2771331058020479,
      "grad_norm": 1.942710900166514,
      "learning_rate": 1.0594077998642025e-06,
      "loss": 0.0309,
      "step": 5613
    },
    {
      "epoch": 1.2773606370875996,
      "grad_norm": 2.5894507431035714,
      "learning_rate": 1.0593435554841247e-06,
      "loss": 0.0629,
      "step": 5614
    },
    {
      "epoch": 1.2775881683731514,
      "grad_norm": 0.8039189924018694,
      "learning_rate": 1.0592793022268603e-06,
      "loss": 0.0142,
      "step": 5615
    },
    {
      "epoch": 1.2778156996587031,
      "grad_norm": 1.109060979102289,
      "learning_rate": 1.0592150400937222e-06,
      "loss": 0.0216,
      "step": 5616
    },
    {
      "epoch": 1.2780432309442549,
      "grad_norm": 1.909595005287003,
      "learning_rate": 1.059150769086024e-06,
      "loss": 0.0948,
      "step": 5617
    },
    {
      "epoch": 1.2782707622298066,
      "grad_norm": 1.7646662291191562,
      "learning_rate": 1.059086489205079e-06,
      "loss": 0.0768,
      "step": 5618
    },
    {
      "epoch": 1.2784982935153584,
      "grad_norm": 0.6261291448525245,
      "learning_rate": 1.0590222004522012e-06,
      "loss": 0.0147,
      "step": 5619
    },
    {
      "epoch": 1.2787258248009101,
      "grad_norm": 1.257814887901369,
      "learning_rate": 1.0589579028287045e-06,
      "loss": 0.0327,
      "step": 5620
    },
    {
      "epoch": 1.2789533560864619,
      "grad_norm": 1.118224185059013,
      "learning_rate": 1.0588935963359032e-06,
      "loss": 0.0334,
      "step": 5621
    },
    {
      "epoch": 1.2791808873720136,
      "grad_norm": 1.524708364397306,
      "learning_rate": 1.0588292809751112e-06,
      "loss": 0.0404,
      "step": 5622
    },
    {
      "epoch": 1.2794084186575654,
      "grad_norm": 0.8685923465972103,
      "learning_rate": 1.0587649567476434e-06,
      "loss": 0.0194,
      "step": 5623
    },
    {
      "epoch": 1.2796359499431171,
      "grad_norm": 0.9025577846970224,
      "learning_rate": 1.0587006236548142e-06,
      "loss": 0.0159,
      "step": 5624
    },
    {
      "epoch": 1.2798634812286689,
      "grad_norm": 0.9060583601498521,
      "learning_rate": 1.0586362816979383e-06,
      "loss": 0.0174,
      "step": 5625
    },
    {
      "epoch": 1.2800910125142206,
      "grad_norm": 0.6398548695145103,
      "learning_rate": 1.0585719308783316e-06,
      "loss": 0.0116,
      "step": 5626
    },
    {
      "epoch": 1.2803185437997724,
      "grad_norm": 0.9028145222778184,
      "learning_rate": 1.058507571197308e-06,
      "loss": 0.0066,
      "step": 5627
    },
    {
      "epoch": 1.2805460750853244,
      "grad_norm": 0.7127546236721658,
      "learning_rate": 1.058443202656184e-06,
      "loss": 0.0112,
      "step": 5628
    },
    {
      "epoch": 1.2807736063708761,
      "grad_norm": 1.1810732350923896,
      "learning_rate": 1.0583788252562745e-06,
      "loss": 0.0388,
      "step": 5629
    },
    {
      "epoch": 1.2810011376564279,
      "grad_norm": 1.3548807980738902,
      "learning_rate": 1.058314438998896e-06,
      "loss": 0.068,
      "step": 5630
    },
    {
      "epoch": 1.2812286689419796,
      "grad_norm": 0.8405440766301588,
      "learning_rate": 1.0582500438853631e-06,
      "loss": 0.0217,
      "step": 5631
    },
    {
      "epoch": 1.2814562002275314,
      "grad_norm": 1.1902880934548667,
      "learning_rate": 1.0581856399169934e-06,
      "loss": 0.0264,
      "step": 5632
    },
    {
      "epoch": 1.2816837315130831,
      "grad_norm": 0.9685340307904585,
      "learning_rate": 1.0581212270951024e-06,
      "loss": 0.0135,
      "step": 5633
    },
    {
      "epoch": 1.2819112627986349,
      "grad_norm": 0.7297977851768284,
      "learning_rate": 1.058056805421007e-06,
      "loss": 0.0209,
      "step": 5634
    },
    {
      "epoch": 1.2821387940841866,
      "grad_norm": 1.0753232968315138,
      "learning_rate": 1.057992374896023e-06,
      "loss": 0.0237,
      "step": 5635
    },
    {
      "epoch": 1.2823663253697384,
      "grad_norm": 1.4454879483916625,
      "learning_rate": 1.0579279355214683e-06,
      "loss": 0.026,
      "step": 5636
    },
    {
      "epoch": 1.2825938566552901,
      "grad_norm": 1.260224633378464,
      "learning_rate": 1.0578634872986592e-06,
      "loss": 0.0184,
      "step": 5637
    },
    {
      "epoch": 1.2828213879408419,
      "grad_norm": 1.9252940374710767,
      "learning_rate": 1.0577990302289136e-06,
      "loss": 0.0293,
      "step": 5638
    },
    {
      "epoch": 1.2830489192263936,
      "grad_norm": 1.4343768366824854,
      "learning_rate": 1.0577345643135482e-06,
      "loss": 0.0408,
      "step": 5639
    },
    {
      "epoch": 1.2832764505119454,
      "grad_norm": 0.8768828863677058,
      "learning_rate": 1.0576700895538809e-06,
      "loss": 0.0213,
      "step": 5640
    },
    {
      "epoch": 1.2835039817974971,
      "grad_norm": 0.8309215455146005,
      "learning_rate": 1.0576056059512292e-06,
      "loss": 0.0205,
      "step": 5641
    },
    {
      "epoch": 1.283731513083049,
      "grad_norm": 0.95278530564801,
      "learning_rate": 1.057541113506911e-06,
      "loss": 0.0229,
      "step": 5642
    },
    {
      "epoch": 1.2839590443686006,
      "grad_norm": 2.512244215865245,
      "learning_rate": 1.057476612222245e-06,
      "loss": 0.0373,
      "step": 5643
    },
    {
      "epoch": 1.2841865756541524,
      "grad_norm": 1.2294843329469738,
      "learning_rate": 1.057412102098549e-06,
      "loss": 0.0384,
      "step": 5644
    },
    {
      "epoch": 1.2844141069397041,
      "grad_norm": 2.1898724117931843,
      "learning_rate": 1.0573475831371416e-06,
      "loss": 0.0251,
      "step": 5645
    },
    {
      "epoch": 1.284641638225256,
      "grad_norm": 1.5651747311949955,
      "learning_rate": 1.0572830553393412e-06,
      "loss": 0.0298,
      "step": 5646
    },
    {
      "epoch": 1.2848691695108077,
      "grad_norm": 1.0523784768363587,
      "learning_rate": 1.057218518706467e-06,
      "loss": 0.0201,
      "step": 5647
    },
    {
      "epoch": 1.2850967007963594,
      "grad_norm": 1.8703075631421116,
      "learning_rate": 1.0571539732398378e-06,
      "loss": 0.0195,
      "step": 5648
    },
    {
      "epoch": 1.2853242320819112,
      "grad_norm": 2.013096542848063,
      "learning_rate": 1.057089418940773e-06,
      "loss": 0.0299,
      "step": 5649
    },
    {
      "epoch": 1.285551763367463,
      "grad_norm": 1.3017641104414657,
      "learning_rate": 1.0570248558105915e-06,
      "loss": 0.0278,
      "step": 5650
    },
    {
      "epoch": 1.2857792946530147,
      "grad_norm": 0.688625948886133,
      "learning_rate": 1.0569602838506136e-06,
      "loss": 0.0088,
      "step": 5651
    },
    {
      "epoch": 1.2860068259385666,
      "grad_norm": 1.0503675437019973,
      "learning_rate": 1.0568957030621582e-06,
      "loss": 0.0135,
      "step": 5652
    },
    {
      "epoch": 1.2862343572241184,
      "grad_norm": 1.1035321913801368,
      "learning_rate": 1.0568311134465457e-06,
      "loss": 0.0322,
      "step": 5653
    },
    {
      "epoch": 1.2864618885096701,
      "grad_norm": 0.8543649037890663,
      "learning_rate": 1.056766515005096e-06,
      "loss": 0.0124,
      "step": 5654
    },
    {
      "epoch": 1.286689419795222,
      "grad_norm": 1.1122886475152938,
      "learning_rate": 1.0567019077391296e-06,
      "loss": 0.0235,
      "step": 5655
    },
    {
      "epoch": 1.2869169510807736,
      "grad_norm": 1.1079256097467238,
      "learning_rate": 1.056637291649967e-06,
      "loss": 0.0191,
      "step": 5656
    },
    {
      "epoch": 1.2871444823663254,
      "grad_norm": 2.5082757586881863,
      "learning_rate": 1.0565726667389284e-06,
      "loss": 0.0147,
      "step": 5657
    },
    {
      "epoch": 1.2873720136518771,
      "grad_norm": 0.6994602690994648,
      "learning_rate": 1.056508033007335e-06,
      "loss": 0.0146,
      "step": 5658
    },
    {
      "epoch": 1.287599544937429,
      "grad_norm": 1.156297278545635,
      "learning_rate": 1.0564433904565078e-06,
      "loss": 0.0206,
      "step": 5659
    },
    {
      "epoch": 1.2878270762229806,
      "grad_norm": 1.0348771757586008,
      "learning_rate": 1.0563787390877677e-06,
      "loss": 0.0304,
      "step": 5660
    },
    {
      "epoch": 1.2880546075085324,
      "grad_norm": 1.6172923570680333,
      "learning_rate": 1.0563140789024363e-06,
      "loss": 0.0393,
      "step": 5661
    },
    {
      "epoch": 1.2882821387940842,
      "grad_norm": 0.7384196511271945,
      "learning_rate": 1.0562494099018346e-06,
      "loss": 0.0238,
      "step": 5662
    },
    {
      "epoch": 1.288509670079636,
      "grad_norm": 0.8883280220843028,
      "learning_rate": 1.0561847320872853e-06,
      "loss": 0.0184,
      "step": 5663
    },
    {
      "epoch": 1.2887372013651877,
      "grad_norm": 1.7381905580936334,
      "learning_rate": 1.0561200454601097e-06,
      "loss": 0.0362,
      "step": 5664
    },
    {
      "epoch": 1.2889647326507394,
      "grad_norm": 1.187480383757449,
      "learning_rate": 1.0560553500216298e-06,
      "loss": 0.0352,
      "step": 5665
    },
    {
      "epoch": 1.2891922639362912,
      "grad_norm": 1.2836678409259843,
      "learning_rate": 1.0559906457731678e-06,
      "loss": 0.0234,
      "step": 5666
    },
    {
      "epoch": 1.2894197952218431,
      "grad_norm": 0.9869735102302846,
      "learning_rate": 1.0559259327160465e-06,
      "loss": 0.0176,
      "step": 5667
    },
    {
      "epoch": 1.2896473265073949,
      "grad_norm": 0.5781054590113305,
      "learning_rate": 1.0558612108515883e-06,
      "loss": 0.0103,
      "step": 5668
    },
    {
      "epoch": 1.2898748577929466,
      "grad_norm": 0.7247426227910199,
      "learning_rate": 1.0557964801811162e-06,
      "loss": 0.0148,
      "step": 5669
    },
    {
      "epoch": 1.2901023890784984,
      "grad_norm": 1.5311127998553455,
      "learning_rate": 1.0557317407059529e-06,
      "loss": 0.009,
      "step": 5670
    },
    {
      "epoch": 1.2903299203640501,
      "grad_norm": 1.0196457464528732,
      "learning_rate": 1.0556669924274217e-06,
      "loss": 0.0154,
      "step": 5671
    },
    {
      "epoch": 1.290557451649602,
      "grad_norm": 1.3612360034540798,
      "learning_rate": 1.0556022353468459e-06,
      "loss": 0.0158,
      "step": 5672
    },
    {
      "epoch": 1.2907849829351536,
      "grad_norm": 1.0790279937441036,
      "learning_rate": 1.055537469465549e-06,
      "loss": 0.0166,
      "step": 5673
    },
    {
      "epoch": 1.2910125142207054,
      "grad_norm": 1.1840170080924126,
      "learning_rate": 1.0554726947848545e-06,
      "loss": 0.0198,
      "step": 5674
    },
    {
      "epoch": 1.2912400455062572,
      "grad_norm": 0.9790455049633395,
      "learning_rate": 1.0554079113060869e-06,
      "loss": 0.014,
      "step": 5675
    },
    {
      "epoch": 1.291467576791809,
      "grad_norm": 0.9375698560069781,
      "learning_rate": 1.0553431190305695e-06,
      "loss": 0.0193,
      "step": 5676
    },
    {
      "epoch": 1.2916951080773607,
      "grad_norm": 0.9460244225991181,
      "learning_rate": 1.055278317959627e-06,
      "loss": 0.022,
      "step": 5677
    },
    {
      "epoch": 1.2919226393629124,
      "grad_norm": 37.51370385688921,
      "learning_rate": 1.0552135080945839e-06,
      "loss": 0.0196,
      "step": 5678
    },
    {
      "epoch": 1.2921501706484642,
      "grad_norm": 0.7870557071298617,
      "learning_rate": 1.0551486894367643e-06,
      "loss": 0.0292,
      "step": 5679
    },
    {
      "epoch": 1.292377701934016,
      "grad_norm": 1.005395594404911,
      "learning_rate": 1.0550838619874933e-06,
      "loss": 0.0186,
      "step": 5680
    },
    {
      "epoch": 1.2926052332195677,
      "grad_norm": 1.3329240605685746,
      "learning_rate": 1.055019025748096e-06,
      "loss": 0.0303,
      "step": 5681
    },
    {
      "epoch": 1.2928327645051194,
      "grad_norm": 0.8592212919369031,
      "learning_rate": 1.0549541807198974e-06,
      "loss": 0.0169,
      "step": 5682
    },
    {
      "epoch": 1.2930602957906712,
      "grad_norm": 1.0894816397080542,
      "learning_rate": 1.0548893269042226e-06,
      "loss": 0.0147,
      "step": 5683
    },
    {
      "epoch": 1.293287827076223,
      "grad_norm": 1.2756290606689868,
      "learning_rate": 1.0548244643023972e-06,
      "loss": 0.0376,
      "step": 5684
    },
    {
      "epoch": 1.2935153583617747,
      "grad_norm": 0.7933354902103573,
      "learning_rate": 1.054759592915747e-06,
      "loss": 0.0204,
      "step": 5685
    },
    {
      "epoch": 1.2937428896473264,
      "grad_norm": 0.9717459762266182,
      "learning_rate": 1.054694712745598e-06,
      "loss": 0.0217,
      "step": 5686
    },
    {
      "epoch": 1.2939704209328782,
      "grad_norm": 1.4817148974153636,
      "learning_rate": 1.0546298237932757e-06,
      "loss": 0.0395,
      "step": 5687
    },
    {
      "epoch": 1.29419795221843,
      "grad_norm": 0.7570513970170758,
      "learning_rate": 1.0545649260601068e-06,
      "loss": 0.0177,
      "step": 5688
    },
    {
      "epoch": 1.2944254835039817,
      "grad_norm": 0.8213620969513302,
      "learning_rate": 1.0545000195474175e-06,
      "loss": 0.0123,
      "step": 5689
    },
    {
      "epoch": 1.2946530147895334,
      "grad_norm": 2.570113325707381,
      "learning_rate": 1.0544351042565344e-06,
      "loss": 0.1389,
      "step": 5690
    },
    {
      "epoch": 1.2948805460750854,
      "grad_norm": 0.94694153472363,
      "learning_rate": 1.0543701801887842e-06,
      "loss": 0.0129,
      "step": 5691
    },
    {
      "epoch": 1.2951080773606372,
      "grad_norm": 1.1658997120682517,
      "learning_rate": 1.054305247345494e-06,
      "loss": 0.0247,
      "step": 5692
    },
    {
      "epoch": 1.295335608646189,
      "grad_norm": 0.6388376726663733,
      "learning_rate": 1.0542403057279907e-06,
      "loss": 0.0079,
      "step": 5693
    },
    {
      "epoch": 1.2955631399317407,
      "grad_norm": 1.5533598170833833,
      "learning_rate": 1.0541753553376016e-06,
      "loss": 0.0552,
      "step": 5694
    },
    {
      "epoch": 1.2957906712172924,
      "grad_norm": 1.3900915621379943,
      "learning_rate": 1.0541103961756543e-06,
      "loss": 0.0276,
      "step": 5695
    },
    {
      "epoch": 1.2960182025028442,
      "grad_norm": 1.3170728163032424,
      "learning_rate": 1.0540454282434765e-06,
      "loss": 0.0292,
      "step": 5696
    },
    {
      "epoch": 1.296245733788396,
      "grad_norm": 0.9274424986111619,
      "learning_rate": 1.0539804515423955e-06,
      "loss": 0.0248,
      "step": 5697
    },
    {
      "epoch": 1.2964732650739477,
      "grad_norm": 1.002255995914125,
      "learning_rate": 1.0539154660737401e-06,
      "loss": 0.0283,
      "step": 5698
    },
    {
      "epoch": 1.2967007963594994,
      "grad_norm": 1.0473043573590481,
      "learning_rate": 1.053850471838838e-06,
      "loss": 0.014,
      "step": 5699
    },
    {
      "epoch": 1.2969283276450512,
      "grad_norm": 0.9807190528337413,
      "learning_rate": 1.0537854688390175e-06,
      "loss": 0.0247,
      "step": 5700
    },
    {
      "epoch": 1.297155858930603,
      "grad_norm": 0.6854049395195209,
      "learning_rate": 1.0537204570756076e-06,
      "loss": 0.0153,
      "step": 5701
    },
    {
      "epoch": 1.2973833902161547,
      "grad_norm": 1.0640768424401603,
      "learning_rate": 1.0536554365499367e-06,
      "loss": 0.0268,
      "step": 5702
    },
    {
      "epoch": 1.2976109215017064,
      "grad_norm": 1.4228991728545564,
      "learning_rate": 1.0535904072633334e-06,
      "loss": 0.0211,
      "step": 5703
    },
    {
      "epoch": 1.2978384527872582,
      "grad_norm": 1.0465889798908117,
      "learning_rate": 1.0535253692171273e-06,
      "loss": 0.0208,
      "step": 5704
    },
    {
      "epoch": 1.29806598407281,
      "grad_norm": 1.1180867272354698,
      "learning_rate": 1.0534603224126474e-06,
      "loss": 0.0426,
      "step": 5705
    },
    {
      "epoch": 1.298293515358362,
      "grad_norm": 1.505044776171652,
      "learning_rate": 1.0533952668512231e-06,
      "loss": 0.04,
      "step": 5706
    },
    {
      "epoch": 1.2985210466439137,
      "grad_norm": 0.6143108397733672,
      "learning_rate": 1.0533302025341843e-06,
      "loss": 0.0121,
      "step": 5707
    },
    {
      "epoch": 1.2987485779294654,
      "grad_norm": 1.127765980451607,
      "learning_rate": 1.0532651294628607e-06,
      "loss": 0.0316,
      "step": 5708
    },
    {
      "epoch": 1.2989761092150172,
      "grad_norm": 1.1519987874665927,
      "learning_rate": 1.053200047638582e-06,
      "loss": 0.0306,
      "step": 5709
    },
    {
      "epoch": 1.299203640500569,
      "grad_norm": 0.9637531006099895,
      "learning_rate": 1.0531349570626787e-06,
      "loss": 0.0209,
      "step": 5710
    },
    {
      "epoch": 1.2994311717861207,
      "grad_norm": 0.653830233371083,
      "learning_rate": 1.0530698577364807e-06,
      "loss": 0.011,
      "step": 5711
    },
    {
      "epoch": 1.2996587030716724,
      "grad_norm": 1.6778292566216655,
      "learning_rate": 1.053004749661319e-06,
      "loss": 0.0327,
      "step": 5712
    },
    {
      "epoch": 1.2998862343572242,
      "grad_norm": 0.8508161175719957,
      "learning_rate": 1.0529396328385238e-06,
      "loss": 0.0298,
      "step": 5713
    },
    {
      "epoch": 1.300113765642776,
      "grad_norm": 1.9250422937435294,
      "learning_rate": 1.0528745072694266e-06,
      "loss": 0.0616,
      "step": 5714
    },
    {
      "epoch": 1.3003412969283277,
      "grad_norm": 1.8275774370143938,
      "learning_rate": 1.052809372955358e-06,
      "loss": 0.0453,
      "step": 5715
    },
    {
      "epoch": 1.3005688282138794,
      "grad_norm": 0.7928558918855286,
      "learning_rate": 1.0527442298976492e-06,
      "loss": 0.0112,
      "step": 5716
    },
    {
      "epoch": 1.3007963594994312,
      "grad_norm": 1.6775736020855938,
      "learning_rate": 1.0526790780976318e-06,
      "loss": 0.0386,
      "step": 5717
    },
    {
      "epoch": 1.301023890784983,
      "grad_norm": 0.7524597201626481,
      "learning_rate": 1.052613917556637e-06,
      "loss": 0.0126,
      "step": 5718
    },
    {
      "epoch": 1.3012514220705347,
      "grad_norm": 1.7153199401873807,
      "learning_rate": 1.0525487482759975e-06,
      "loss": 0.0136,
      "step": 5719
    },
    {
      "epoch": 1.3014789533560864,
      "grad_norm": 1.0942059838392535,
      "learning_rate": 1.052483570257044e-06,
      "loss": 0.0365,
      "step": 5720
    },
    {
      "epoch": 1.3017064846416382,
      "grad_norm": 1.0739474507348372,
      "learning_rate": 1.0524183835011095e-06,
      "loss": 0.0147,
      "step": 5721
    },
    {
      "epoch": 1.30193401592719,
      "grad_norm": 1.1602347838561342,
      "learning_rate": 1.052353188009526e-06,
      "loss": 0.0438,
      "step": 5722
    },
    {
      "epoch": 1.3021615472127417,
      "grad_norm": 0.774345620272605,
      "learning_rate": 1.052287983783626e-06,
      "loss": 0.0129,
      "step": 5723
    },
    {
      "epoch": 1.3023890784982934,
      "grad_norm": 1.551981652235957,
      "learning_rate": 1.052222770824742e-06,
      "loss": 0.034,
      "step": 5724
    },
    {
      "epoch": 1.3026166097838452,
      "grad_norm": 1.073745426616395,
      "learning_rate": 1.0521575491342074e-06,
      "loss": 0.0151,
      "step": 5725
    },
    {
      "epoch": 1.302844141069397,
      "grad_norm": 1.2659073131244354,
      "learning_rate": 1.0520923187133544e-06,
      "loss": 0.0273,
      "step": 5726
    },
    {
      "epoch": 1.3030716723549487,
      "grad_norm": 1.559253636601821,
      "learning_rate": 1.0520270795635167e-06,
      "loss": 0.0433,
      "step": 5727
    },
    {
      "epoch": 1.3032992036405004,
      "grad_norm": 1.246986769425595,
      "learning_rate": 1.0519618316860274e-06,
      "loss": 0.0251,
      "step": 5728
    },
    {
      "epoch": 1.3035267349260522,
      "grad_norm": 1.013075136339921,
      "learning_rate": 1.0518965750822204e-06,
      "loss": 0.0248,
      "step": 5729
    },
    {
      "epoch": 1.3037542662116042,
      "grad_norm": 0.776622690041485,
      "learning_rate": 1.0518313097534292e-06,
      "loss": 0.0197,
      "step": 5730
    },
    {
      "epoch": 1.303981797497156,
      "grad_norm": 1.8936915924731421,
      "learning_rate": 1.0517660357009877e-06,
      "loss": 0.0323,
      "step": 5731
    },
    {
      "epoch": 1.3042093287827077,
      "grad_norm": 0.9731351344937886,
      "learning_rate": 1.0517007529262301e-06,
      "loss": 0.0201,
      "step": 5732
    },
    {
      "epoch": 1.3044368600682594,
      "grad_norm": 1.4132176366415885,
      "learning_rate": 1.0516354614304905e-06,
      "loss": 0.0448,
      "step": 5733
    },
    {
      "epoch": 1.3046643913538112,
      "grad_norm": 1.2312639755556094,
      "learning_rate": 1.0515701612151035e-06,
      "loss": 0.0363,
      "step": 5734
    },
    {
      "epoch": 1.304891922639363,
      "grad_norm": 0.9680235672394335,
      "learning_rate": 1.0515048522814034e-06,
      "loss": 0.0244,
      "step": 5735
    },
    {
      "epoch": 1.3051194539249147,
      "grad_norm": 0.8942191857528213,
      "learning_rate": 1.0514395346307254e-06,
      "loss": 0.0241,
      "step": 5736
    },
    {
      "epoch": 1.3053469852104664,
      "grad_norm": 1.4700138740863975,
      "learning_rate": 1.0513742082644043e-06,
      "loss": 0.0346,
      "step": 5737
    },
    {
      "epoch": 1.3055745164960182,
      "grad_norm": 1.1078337385555872,
      "learning_rate": 1.0513088731837753e-06,
      "loss": 0.0305,
      "step": 5738
    },
    {
      "epoch": 1.30580204778157,
      "grad_norm": 1.2748420988398181,
      "learning_rate": 1.0512435293901737e-06,
      "loss": 0.0285,
      "step": 5739
    },
    {
      "epoch": 1.3060295790671217,
      "grad_norm": 1.2790988355743922,
      "learning_rate": 1.051178176884935e-06,
      "loss": 0.0492,
      "step": 5740
    },
    {
      "epoch": 1.3062571103526734,
      "grad_norm": 1.1980282713090238,
      "learning_rate": 1.0511128156693947e-06,
      "loss": 0.0216,
      "step": 5741
    },
    {
      "epoch": 1.3064846416382252,
      "grad_norm": 0.8208046025134933,
      "learning_rate": 1.0510474457448888e-06,
      "loss": 0.0178,
      "step": 5742
    },
    {
      "epoch": 1.306712172923777,
      "grad_norm": 1.8069229595272986,
      "learning_rate": 1.0509820671127535e-06,
      "loss": 0.0199,
      "step": 5743
    },
    {
      "epoch": 1.3069397042093287,
      "grad_norm": 1.6865296714089357,
      "learning_rate": 1.050916679774325e-06,
      "loss": 0.0202,
      "step": 5744
    },
    {
      "epoch": 1.3071672354948807,
      "grad_norm": 1.0275125368796856,
      "learning_rate": 1.0508512837309394e-06,
      "loss": 0.0154,
      "step": 5745
    },
    {
      "epoch": 1.3073947667804324,
      "grad_norm": 0.6133934114428197,
      "learning_rate": 1.0507858789839336e-06,
      "loss": 0.0107,
      "step": 5746
    },
    {
      "epoch": 1.3076222980659842,
      "grad_norm": 1.080087958680723,
      "learning_rate": 1.0507204655346442e-06,
      "loss": 0.0314,
      "step": 5747
    },
    {
      "epoch": 1.307849829351536,
      "grad_norm": 1.3749276553621517,
      "learning_rate": 1.050655043384408e-06,
      "loss": 0.0318,
      "step": 5748
    },
    {
      "epoch": 1.3080773606370877,
      "grad_norm": 1.2280486437236124,
      "learning_rate": 1.0505896125345624e-06,
      "loss": 0.0199,
      "step": 5749
    },
    {
      "epoch": 1.3083048919226394,
      "grad_norm": 1.465755912941539,
      "learning_rate": 1.0505241729864446e-06,
      "loss": 0.0517,
      "step": 5750
    },
    {
      "epoch": 1.3085324232081912,
      "grad_norm": 0.839754511989302,
      "learning_rate": 1.050458724741392e-06,
      "loss": 0.0138,
      "step": 5751
    },
    {
      "epoch": 1.308759954493743,
      "grad_norm": 1.3071289617497805,
      "learning_rate": 1.0503932678007423e-06,
      "loss": 0.0268,
      "step": 5752
    },
    {
      "epoch": 1.3089874857792947,
      "grad_norm": 1.5584387532798485,
      "learning_rate": 1.0503278021658331e-06,
      "loss": 0.0279,
      "step": 5753
    },
    {
      "epoch": 1.3092150170648464,
      "grad_norm": 1.3103126621131902,
      "learning_rate": 1.0502623278380025e-06,
      "loss": 0.0229,
      "step": 5754
    },
    {
      "epoch": 1.3094425483503982,
      "grad_norm": 1.112498635945538,
      "learning_rate": 1.050196844818589e-06,
      "loss": 0.0234,
      "step": 5755
    },
    {
      "epoch": 1.30967007963595,
      "grad_norm": 0.9046518747362639,
      "learning_rate": 1.0501313531089306e-06,
      "loss": 0.0099,
      "step": 5756
    },
    {
      "epoch": 1.3098976109215017,
      "grad_norm": 0.7326140936427108,
      "learning_rate": 1.050065852710366e-06,
      "loss": 0.0117,
      "step": 5757
    },
    {
      "epoch": 1.3101251422070535,
      "grad_norm": 1.1004333953253604,
      "learning_rate": 1.0500003436242338e-06,
      "loss": 0.0323,
      "step": 5758
    },
    {
      "epoch": 1.3103526734926052,
      "grad_norm": 1.4635856283030002,
      "learning_rate": 1.049934825851873e-06,
      "loss": 0.0357,
      "step": 5759
    },
    {
      "epoch": 1.310580204778157,
      "grad_norm": 1.448988151398696,
      "learning_rate": 1.0498692993946225e-06,
      "loss": 0.0298,
      "step": 5760
    },
    {
      "epoch": 1.3108077360637087,
      "grad_norm": 1.0352196860191685,
      "learning_rate": 1.0498037642538215e-06,
      "loss": 0.0146,
      "step": 5761
    },
    {
      "epoch": 1.3110352673492605,
      "grad_norm": 1.2609870327772956,
      "learning_rate": 1.0497382204308099e-06,
      "loss": 0.0207,
      "step": 5762
    },
    {
      "epoch": 1.3112627986348122,
      "grad_norm": 1.2395935801958684,
      "learning_rate": 1.0496726679269265e-06,
      "loss": 0.042,
      "step": 5763
    },
    {
      "epoch": 1.311490329920364,
      "grad_norm": 1.0739173139510374,
      "learning_rate": 1.0496071067435118e-06,
      "loss": 0.0394,
      "step": 5764
    },
    {
      "epoch": 1.3117178612059157,
      "grad_norm": 2.604100179137865,
      "learning_rate": 1.0495415368819057e-06,
      "loss": 0.1258,
      "step": 5765
    },
    {
      "epoch": 1.3119453924914675,
      "grad_norm": 1.393030141056355,
      "learning_rate": 1.0494759583434478e-06,
      "loss": 0.0233,
      "step": 5766
    },
    {
      "epoch": 1.3121729237770192,
      "grad_norm": 1.0166107127362798,
      "learning_rate": 1.0494103711294786e-06,
      "loss": 0.0172,
      "step": 5767
    },
    {
      "epoch": 1.312400455062571,
      "grad_norm": 1.4792549020627224,
      "learning_rate": 1.049344775241339e-06,
      "loss": 0.0415,
      "step": 5768
    },
    {
      "epoch": 1.312627986348123,
      "grad_norm": 1.743954914946822,
      "learning_rate": 1.049279170680369e-06,
      "loss": 0.0559,
      "step": 5769
    },
    {
      "epoch": 1.3128555176336747,
      "grad_norm": 1.8129298367616118,
      "learning_rate": 1.0492135574479097e-06,
      "loss": 0.0416,
      "step": 5770
    },
    {
      "epoch": 1.3130830489192264,
      "grad_norm": 1.253009038271569,
      "learning_rate": 1.0491479355453026e-06,
      "loss": 0.0265,
      "step": 5771
    },
    {
      "epoch": 1.3133105802047782,
      "grad_norm": 0.9975790029687652,
      "learning_rate": 1.0490823049738884e-06,
      "loss": 0.0205,
      "step": 5772
    },
    {
      "epoch": 1.31353811149033,
      "grad_norm": 0.698949514657323,
      "learning_rate": 1.0490166657350084e-06,
      "loss": 0.0211,
      "step": 5773
    },
    {
      "epoch": 1.3137656427758817,
      "grad_norm": 1.2257870676619147,
      "learning_rate": 1.0489510178300043e-06,
      "loss": 0.0315,
      "step": 5774
    },
    {
      "epoch": 1.3139931740614335,
      "grad_norm": 1.0706430296176082,
      "learning_rate": 1.0488853612602178e-06,
      "loss": 0.0165,
      "step": 5775
    },
    {
      "epoch": 1.3142207053469852,
      "grad_norm": 1.1156453691438755,
      "learning_rate": 1.0488196960269912e-06,
      "loss": 0.0282,
      "step": 5776
    },
    {
      "epoch": 1.314448236632537,
      "grad_norm": 1.2538263082028707,
      "learning_rate": 1.0487540221316659e-06,
      "loss": 0.0265,
      "step": 5777
    },
    {
      "epoch": 1.3146757679180887,
      "grad_norm": 1.0773176692864428,
      "learning_rate": 1.0486883395755845e-06,
      "loss": 0.0395,
      "step": 5778
    },
    {
      "epoch": 1.3149032992036405,
      "grad_norm": 1.5883851817408943,
      "learning_rate": 1.0486226483600894e-06,
      "loss": 0.0766,
      "step": 5779
    },
    {
      "epoch": 1.3151308304891922,
      "grad_norm": 1.9633238080883928,
      "learning_rate": 1.0485569484865231e-06,
      "loss": 0.0495,
      "step": 5780
    },
    {
      "epoch": 1.315358361774744,
      "grad_norm": 0.9192459981660208,
      "learning_rate": 1.0484912399562285e-06,
      "loss": 0.0152,
      "step": 5781
    },
    {
      "epoch": 1.3155858930602957,
      "grad_norm": 1.0996145724289152,
      "learning_rate": 1.0484255227705487e-06,
      "loss": 0.0197,
      "step": 5782
    },
    {
      "epoch": 1.3158134243458475,
      "grad_norm": 0.9854990061171917,
      "learning_rate": 1.0483597969308266e-06,
      "loss": 0.0138,
      "step": 5783
    },
    {
      "epoch": 1.3160409556313994,
      "grad_norm": 1.8517240238943,
      "learning_rate": 1.0482940624384054e-06,
      "loss": 0.0352,
      "step": 5784
    },
    {
      "epoch": 1.3162684869169512,
      "grad_norm": 1.225130304804711,
      "learning_rate": 1.048228319294629e-06,
      "loss": 0.0431,
      "step": 5785
    },
    {
      "epoch": 1.316496018202503,
      "grad_norm": 1.0183330800820867,
      "learning_rate": 1.0481625675008409e-06,
      "loss": 0.0286,
      "step": 5786
    },
    {
      "epoch": 1.3167235494880547,
      "grad_norm": 0.9507764029567124,
      "learning_rate": 1.048096807058385e-06,
      "loss": 0.0185,
      "step": 5787
    },
    {
      "epoch": 1.3169510807736065,
      "grad_norm": 0.8930023522137188,
      "learning_rate": 1.0480310379686048e-06,
      "loss": 0.0155,
      "step": 5788
    },
    {
      "epoch": 1.3171786120591582,
      "grad_norm": 1.2745535497453038,
      "learning_rate": 1.0479652602328453e-06,
      "loss": 0.0189,
      "step": 5789
    },
    {
      "epoch": 1.31740614334471,
      "grad_norm": 0.9488946384445988,
      "learning_rate": 1.0478994738524504e-06,
      "loss": 0.0233,
      "step": 5790
    },
    {
      "epoch": 1.3176336746302617,
      "grad_norm": 1.2465521747877613,
      "learning_rate": 1.047833678828765e-06,
      "loss": 0.0246,
      "step": 5791
    },
    {
      "epoch": 1.3178612059158135,
      "grad_norm": 1.0606853469341604,
      "learning_rate": 1.0477678751631332e-06,
      "loss": 0.0249,
      "step": 5792
    },
    {
      "epoch": 1.3180887372013652,
      "grad_norm": 0.7931700678264818,
      "learning_rate": 1.0477020628569005e-06,
      "loss": 0.0155,
      "step": 5793
    },
    {
      "epoch": 1.318316268486917,
      "grad_norm": 1.1055038224148857,
      "learning_rate": 1.0476362419114117e-06,
      "loss": 0.0554,
      "step": 5794
    },
    {
      "epoch": 1.3185437997724687,
      "grad_norm": 1.14334987480646,
      "learning_rate": 1.047570412328012e-06,
      "loss": 0.0233,
      "step": 5795
    },
    {
      "epoch": 1.3187713310580205,
      "grad_norm": 1.0946218965974854,
      "learning_rate": 1.0475045741080473e-06,
      "loss": 0.0493,
      "step": 5796
    },
    {
      "epoch": 1.3189988623435722,
      "grad_norm": 0.5654984905711095,
      "learning_rate": 1.0474387272528627e-06,
      "loss": 0.0139,
      "step": 5797
    },
    {
      "epoch": 1.319226393629124,
      "grad_norm": 1.1814496866598772,
      "learning_rate": 1.047372871763804e-06,
      "loss": 0.0267,
      "step": 5798
    },
    {
      "epoch": 1.3194539249146757,
      "grad_norm": 0.7187378148607076,
      "learning_rate": 1.0473070076422176e-06,
      "loss": 0.0177,
      "step": 5799
    },
    {
      "epoch": 1.3196814562002275,
      "grad_norm": 1.1593289596572192,
      "learning_rate": 1.0472411348894492e-06,
      "loss": 0.0283,
      "step": 5800
    },
    {
      "epoch": 1.3199089874857792,
      "grad_norm": 1.0028279682721826,
      "learning_rate": 1.0471752535068455e-06,
      "loss": 0.0195,
      "step": 5801
    },
    {
      "epoch": 1.320136518771331,
      "grad_norm": 0.8421345274979064,
      "learning_rate": 1.0471093634957528e-06,
      "loss": 0.0161,
      "step": 5802
    },
    {
      "epoch": 1.3203640500568827,
      "grad_norm": 1.42589130831637,
      "learning_rate": 1.0470434648575175e-06,
      "loss": 0.0239,
      "step": 5803
    },
    {
      "epoch": 1.3205915813424345,
      "grad_norm": 1.0307165726126566,
      "learning_rate": 1.046977557593487e-06,
      "loss": 0.0296,
      "step": 5804
    },
    {
      "epoch": 1.3208191126279862,
      "grad_norm": 1.434881555216667,
      "learning_rate": 1.0469116417050078e-06,
      "loss": 0.0287,
      "step": 5805
    },
    {
      "epoch": 1.321046643913538,
      "grad_norm": 1.3073002962401254,
      "learning_rate": 1.0468457171934276e-06,
      "loss": 0.0385,
      "step": 5806
    },
    {
      "epoch": 1.3212741751990897,
      "grad_norm": 1.359726380040625,
      "learning_rate": 1.0467797840600934e-06,
      "loss": 0.0558,
      "step": 5807
    },
    {
      "epoch": 1.3215017064846417,
      "grad_norm": 1.1817435732611496,
      "learning_rate": 1.0467138423063529e-06,
      "loss": 0.0308,
      "step": 5808
    },
    {
      "epoch": 1.3217292377701935,
      "grad_norm": 1.1727355771812176,
      "learning_rate": 1.0466478919335538e-06,
      "loss": 0.0126,
      "step": 5809
    },
    {
      "epoch": 1.3219567690557452,
      "grad_norm": 0.9282827475194554,
      "learning_rate": 1.0465819329430439e-06,
      "loss": 0.043,
      "step": 5810
    },
    {
      "epoch": 1.322184300341297,
      "grad_norm": 1.0289989606262906,
      "learning_rate": 1.0465159653361716e-06,
      "loss": 0.0323,
      "step": 5811
    },
    {
      "epoch": 1.3224118316268487,
      "grad_norm": 0.8011760159128629,
      "learning_rate": 1.0464499891142847e-06,
      "loss": 0.019,
      "step": 5812
    },
    {
      "epoch": 1.3226393629124005,
      "grad_norm": 0.926556955845452,
      "learning_rate": 1.046384004278732e-06,
      "loss": 0.0167,
      "step": 5813
    },
    {
      "epoch": 1.3228668941979522,
      "grad_norm": 1.0337574749197451,
      "learning_rate": 1.0463180108308615e-06,
      "loss": 0.0149,
      "step": 5814
    },
    {
      "epoch": 1.323094425483504,
      "grad_norm": 0.7687290002927235,
      "learning_rate": 1.0462520087720231e-06,
      "loss": 0.0195,
      "step": 5815
    },
    {
      "epoch": 1.3233219567690557,
      "grad_norm": 0.793358906518525,
      "learning_rate": 1.0461859981035649e-06,
      "loss": 0.0234,
      "step": 5816
    },
    {
      "epoch": 1.3235494880546075,
      "grad_norm": 1.29769101133269,
      "learning_rate": 1.0461199788268364e-06,
      "loss": 0.0315,
      "step": 5817
    },
    {
      "epoch": 1.3237770193401592,
      "grad_norm": 1.3102647580631441,
      "learning_rate": 1.0460539509431865e-06,
      "loss": 0.0243,
      "step": 5818
    },
    {
      "epoch": 1.324004550625711,
      "grad_norm": 1.072047311381633,
      "learning_rate": 1.045987914453965e-06,
      "loss": 0.0273,
      "step": 5819
    },
    {
      "epoch": 1.3242320819112627,
      "grad_norm": 0.6326140291009906,
      "learning_rate": 1.0459218693605216e-06,
      "loss": 0.0135,
      "step": 5820
    },
    {
      "epoch": 1.3244596131968145,
      "grad_norm": 0.7804292447629503,
      "learning_rate": 1.0458558156642063e-06,
      "loss": 0.0192,
      "step": 5821
    },
    {
      "epoch": 1.3246871444823665,
      "grad_norm": 1.2322347552231876,
      "learning_rate": 1.0457897533663686e-06,
      "loss": 0.029,
      "step": 5822
    },
    {
      "epoch": 1.3249146757679182,
      "grad_norm": 1.0345091680985516,
      "learning_rate": 1.0457236824683592e-06,
      "loss": 0.0297,
      "step": 5823
    },
    {
      "epoch": 1.32514220705347,
      "grad_norm": 1.2808097927518385,
      "learning_rate": 1.045657602971528e-06,
      "loss": 0.0247,
      "step": 5824
    },
    {
      "epoch": 1.3253697383390217,
      "grad_norm": 0.7749625600644442,
      "learning_rate": 1.0455915148772262e-06,
      "loss": 0.0182,
      "step": 5825
    },
    {
      "epoch": 1.3255972696245735,
      "grad_norm": 0.8652793933619614,
      "learning_rate": 1.0455254181868037e-06,
      "loss": 0.0151,
      "step": 5826
    },
    {
      "epoch": 1.3258248009101252,
      "grad_norm": 1.237413731510675,
      "learning_rate": 1.0454593129016121e-06,
      "loss": 0.0172,
      "step": 5827
    },
    {
      "epoch": 1.326052332195677,
      "grad_norm": 1.00775831243054,
      "learning_rate": 1.045393199023002e-06,
      "loss": 0.0355,
      "step": 5828
    },
    {
      "epoch": 1.3262798634812287,
      "grad_norm": 0.8408054705160839,
      "learning_rate": 1.0453270765523247e-06,
      "loss": 0.0206,
      "step": 5829
    },
    {
      "epoch": 1.3265073947667805,
      "grad_norm": 0.7148322270057228,
      "learning_rate": 1.045260945490932e-06,
      "loss": 0.0183,
      "step": 5830
    },
    {
      "epoch": 1.3267349260523322,
      "grad_norm": 0.8189570497153293,
      "learning_rate": 1.045194805840175e-06,
      "loss": 0.015,
      "step": 5831
    },
    {
      "epoch": 1.326962457337884,
      "grad_norm": 1.4734645621134759,
      "learning_rate": 1.045128657601406e-06,
      "loss": 0.0406,
      "step": 5832
    },
    {
      "epoch": 1.3271899886234357,
      "grad_norm": 0.6441526578319029,
      "learning_rate": 1.0450625007759765e-06,
      "loss": 0.0135,
      "step": 5833
    },
    {
      "epoch": 1.3274175199089875,
      "grad_norm": 0.5585116452445247,
      "learning_rate": 1.044996335365239e-06,
      "loss": 0.0089,
      "step": 5834
    },
    {
      "epoch": 1.3276450511945392,
      "grad_norm": 1.2564939600468337,
      "learning_rate": 1.0449301613705453e-06,
      "loss": 0.0312,
      "step": 5835
    },
    {
      "epoch": 1.327872582480091,
      "grad_norm": 0.7995562758285696,
      "learning_rate": 1.0448639787932482e-06,
      "loss": 0.0185,
      "step": 5836
    },
    {
      "epoch": 1.3281001137656427,
      "grad_norm": 0.6950191248274259,
      "learning_rate": 1.0447977876347005e-06,
      "loss": 0.0166,
      "step": 5837
    },
    {
      "epoch": 1.3283276450511945,
      "grad_norm": 1.1554827500771103,
      "learning_rate": 1.0447315878962547e-06,
      "loss": 0.0158,
      "step": 5838
    },
    {
      "epoch": 1.3285551763367462,
      "grad_norm": 0.7747805069607538,
      "learning_rate": 1.044665379579264e-06,
      "loss": 0.0129,
      "step": 5839
    },
    {
      "epoch": 1.328782707622298,
      "grad_norm": 0.80632772075669,
      "learning_rate": 1.0445991626850816e-06,
      "loss": 0.0206,
      "step": 5840
    },
    {
      "epoch": 1.3290102389078498,
      "grad_norm": 1.5896873106614287,
      "learning_rate": 1.0445329372150607e-06,
      "loss": 0.0389,
      "step": 5841
    },
    {
      "epoch": 1.3292377701934015,
      "grad_norm": 1.072021072181124,
      "learning_rate": 1.0444667031705549e-06,
      "loss": 0.0303,
      "step": 5842
    },
    {
      "epoch": 1.3294653014789533,
      "grad_norm": 2.222905444464611,
      "learning_rate": 1.0444004605529178e-06,
      "loss": 0.0631,
      "step": 5843
    },
    {
      "epoch": 1.329692832764505,
      "grad_norm": 0.8281594466779323,
      "learning_rate": 1.0443342093635036e-06,
      "loss": 0.022,
      "step": 5844
    },
    {
      "epoch": 1.3299203640500568,
      "grad_norm": 1.6272382185157073,
      "learning_rate": 1.044267949603666e-06,
      "loss": 0.0546,
      "step": 5845
    },
    {
      "epoch": 1.3301478953356085,
      "grad_norm": 0.9091509980029293,
      "learning_rate": 1.0442016812747594e-06,
      "loss": 0.0236,
      "step": 5846
    },
    {
      "epoch": 1.3303754266211605,
      "grad_norm": 0.85308617988147,
      "learning_rate": 1.0441354043781381e-06,
      "loss": 0.0136,
      "step": 5847
    },
    {
      "epoch": 1.3306029579067122,
      "grad_norm": 0.44808170737486047,
      "learning_rate": 1.0440691189151567e-06,
      "loss": 0.007,
      "step": 5848
    },
    {
      "epoch": 1.330830489192264,
      "grad_norm": 0.8973824993734406,
      "learning_rate": 1.0440028248871702e-06,
      "loss": 0.0177,
      "step": 5849
    },
    {
      "epoch": 1.3310580204778157,
      "grad_norm": 1.7619238754120394,
      "learning_rate": 1.0439365222955332e-06,
      "loss": 0.0331,
      "step": 5850
    },
    {
      "epoch": 1.3312855517633675,
      "grad_norm": 0.8705762797622655,
      "learning_rate": 1.043870211141601e-06,
      "loss": 0.0172,
      "step": 5851
    },
    {
      "epoch": 1.3315130830489192,
      "grad_norm": 1.096308879196295,
      "learning_rate": 1.0438038914267287e-06,
      "loss": 0.0274,
      "step": 5852
    },
    {
      "epoch": 1.331740614334471,
      "grad_norm": 0.8197895561671952,
      "learning_rate": 1.043737563152272e-06,
      "loss": 0.0172,
      "step": 5853
    },
    {
      "epoch": 1.3319681456200227,
      "grad_norm": 1.4719340019230323,
      "learning_rate": 1.0436712263195862e-06,
      "loss": 0.0141,
      "step": 5854
    },
    {
      "epoch": 1.3321956769055745,
      "grad_norm": 0.9648112032732078,
      "learning_rate": 1.0436048809300273e-06,
      "loss": 0.0276,
      "step": 5855
    },
    {
      "epoch": 1.3324232081911263,
      "grad_norm": 0.852726738751836,
      "learning_rate": 1.0435385269849515e-06,
      "loss": 0.0288,
      "step": 5856
    },
    {
      "epoch": 1.332650739476678,
      "grad_norm": 1.721578698981377,
      "learning_rate": 1.0434721644857146e-06,
      "loss": 0.043,
      "step": 5857
    },
    {
      "epoch": 1.3328782707622298,
      "grad_norm": 0.623574972886459,
      "learning_rate": 1.043405793433673e-06,
      "loss": 0.0162,
      "step": 5858
    },
    {
      "epoch": 1.3331058020477815,
      "grad_norm": 0.7840847487395295,
      "learning_rate": 1.0433394138301835e-06,
      "loss": 0.0168,
      "step": 5859
    },
    {
      "epoch": 1.3333333333333333,
      "grad_norm": 1.1889353006124617,
      "learning_rate": 1.0432730256766022e-06,
      "loss": 0.0291,
      "step": 5860
    },
    {
      "epoch": 1.3335608646188852,
      "grad_norm": 1.5874275317235012,
      "learning_rate": 1.0432066289742864e-06,
      "loss": 0.0482,
      "step": 5861
    },
    {
      "epoch": 1.333788395904437,
      "grad_norm": 0.8058160492544123,
      "learning_rate": 1.0431402237245932e-06,
      "loss": 0.0111,
      "step": 5862
    },
    {
      "epoch": 1.3340159271899887,
      "grad_norm": 0.6561988259057422,
      "learning_rate": 1.0430738099288794e-06,
      "loss": 0.0112,
      "step": 5863
    },
    {
      "epoch": 1.3342434584755405,
      "grad_norm": 1.0674110824492682,
      "learning_rate": 1.0430073875885026e-06,
      "loss": 0.0305,
      "step": 5864
    },
    {
      "epoch": 1.3344709897610922,
      "grad_norm": 0.6325379384484306,
      "learning_rate": 1.0429409567048205e-06,
      "loss": 0.0088,
      "step": 5865
    },
    {
      "epoch": 1.334698521046644,
      "grad_norm": 1.0168781752483684,
      "learning_rate": 1.0428745172791905e-06,
      "loss": 0.0387,
      "step": 5866
    },
    {
      "epoch": 1.3349260523321957,
      "grad_norm": 1.7297475905815625,
      "learning_rate": 1.0428080693129708e-06,
      "loss": 0.0589,
      "step": 5867
    },
    {
      "epoch": 1.3351535836177475,
      "grad_norm": 1.1628295813004157,
      "learning_rate": 1.0427416128075192e-06,
      "loss": 0.0193,
      "step": 5868
    },
    {
      "epoch": 1.3353811149032992,
      "grad_norm": 1.2631903831165703,
      "learning_rate": 1.0426751477641941e-06,
      "loss": 0.0324,
      "step": 5869
    },
    {
      "epoch": 1.335608646188851,
      "grad_norm": 1.4329045968752339,
      "learning_rate": 1.042608674184354e-06,
      "loss": 0.0255,
      "step": 5870
    },
    {
      "epoch": 1.3358361774744028,
      "grad_norm": 2.479731151758897,
      "learning_rate": 1.0425421920693575e-06,
      "loss": 0.0585,
      "step": 5871
    },
    {
      "epoch": 1.3360637087599545,
      "grad_norm": 0.9244796148915133,
      "learning_rate": 1.042475701420563e-06,
      "loss": 0.0195,
      "step": 5872
    },
    {
      "epoch": 1.3362912400455063,
      "grad_norm": 0.8951775033309297,
      "learning_rate": 1.04240920223933e-06,
      "loss": 0.0214,
      "step": 5873
    },
    {
      "epoch": 1.336518771331058,
      "grad_norm": 0.7274004184798262,
      "learning_rate": 1.0423426945270174e-06,
      "loss": 0.0219,
      "step": 5874
    },
    {
      "epoch": 1.3367463026166098,
      "grad_norm": 0.9074676685298566,
      "learning_rate": 1.0422761782849842e-06,
      "loss": 0.0315,
      "step": 5875
    },
    {
      "epoch": 1.3369738339021615,
      "grad_norm": 1.9629567837065391,
      "learning_rate": 1.0422096535145902e-06,
      "loss": 0.0489,
      "step": 5876
    },
    {
      "epoch": 1.3372013651877133,
      "grad_norm": 1.1684048882515632,
      "learning_rate": 1.042143120217195e-06,
      "loss": 0.0293,
      "step": 5877
    },
    {
      "epoch": 1.337428896473265,
      "grad_norm": 1.364293734448395,
      "learning_rate": 1.0420765783941586e-06,
      "loss": 0.0375,
      "step": 5878
    },
    {
      "epoch": 1.3376564277588168,
      "grad_norm": 1.039811324655209,
      "learning_rate": 1.0420100280468404e-06,
      "loss": 0.0265,
      "step": 5879
    },
    {
      "epoch": 1.3378839590443685,
      "grad_norm": 0.8625486316278609,
      "learning_rate": 1.0419434691766012e-06,
      "loss": 0.0268,
      "step": 5880
    },
    {
      "epoch": 1.3381114903299203,
      "grad_norm": 0.7261697725405176,
      "learning_rate": 1.041876901784801e-06,
      "loss": 0.0147,
      "step": 5881
    },
    {
      "epoch": 1.338339021615472,
      "grad_norm": 0.8914798651369713,
      "learning_rate": 1.0418103258728001e-06,
      "loss": 0.0233,
      "step": 5882
    },
    {
      "epoch": 1.3385665529010238,
      "grad_norm": 2.1255378013198962,
      "learning_rate": 1.04174374144196e-06,
      "loss": 0.0277,
      "step": 5883
    },
    {
      "epoch": 1.3387940841865755,
      "grad_norm": 1.1620134309649157,
      "learning_rate": 1.0416771484936409e-06,
      "loss": 0.0206,
      "step": 5884
    },
    {
      "epoch": 1.3390216154721273,
      "grad_norm": 1.787592523953072,
      "learning_rate": 1.041610547029204e-06,
      "loss": 0.0594,
      "step": 5885
    },
    {
      "epoch": 1.3392491467576793,
      "grad_norm": 1.3432064804257042,
      "learning_rate": 1.0415439370500104e-06,
      "loss": 0.0271,
      "step": 5886
    },
    {
      "epoch": 1.339476678043231,
      "grad_norm": 1.4695495810180548,
      "learning_rate": 1.0414773185574214e-06,
      "loss": 0.0632,
      "step": 5887
    },
    {
      "epoch": 1.3397042093287828,
      "grad_norm": 1.002580161934096,
      "learning_rate": 1.0414106915527991e-06,
      "loss": 0.034,
      "step": 5888
    },
    {
      "epoch": 1.3399317406143345,
      "grad_norm": 1.2230261450463973,
      "learning_rate": 1.041344056037505e-06,
      "loss": 0.0527,
      "step": 5889
    },
    {
      "epoch": 1.3401592718998863,
      "grad_norm": 1.5508468505953605,
      "learning_rate": 1.0412774120129004e-06,
      "loss": 0.0364,
      "step": 5890
    },
    {
      "epoch": 1.340386803185438,
      "grad_norm": 1.1182067053381388,
      "learning_rate": 1.0412107594803484e-06,
      "loss": 0.0224,
      "step": 5891
    },
    {
      "epoch": 1.3406143344709898,
      "grad_norm": 1.334755248496988,
      "learning_rate": 1.0411440984412103e-06,
      "loss": 0.0306,
      "step": 5892
    },
    {
      "epoch": 1.3408418657565415,
      "grad_norm": 1.0312638416250242,
      "learning_rate": 1.0410774288968492e-06,
      "loss": 0.0273,
      "step": 5893
    },
    {
      "epoch": 1.3410693970420933,
      "grad_norm": 0.8814569099212579,
      "learning_rate": 1.0410107508486272e-06,
      "loss": 0.0175,
      "step": 5894
    },
    {
      "epoch": 1.341296928327645,
      "grad_norm": 0.9202861833902302,
      "learning_rate": 1.0409440642979077e-06,
      "loss": 0.0258,
      "step": 5895
    },
    {
      "epoch": 1.3415244596131968,
      "grad_norm": 1.0927206734816597,
      "learning_rate": 1.040877369246053e-06,
      "loss": 0.0429,
      "step": 5896
    },
    {
      "epoch": 1.3417519908987485,
      "grad_norm": 1.3413817549764846,
      "learning_rate": 1.0408106656944267e-06,
      "loss": 0.0248,
      "step": 5897
    },
    {
      "epoch": 1.3419795221843003,
      "grad_norm": 0.8153253460902535,
      "learning_rate": 1.0407439536443919e-06,
      "loss": 0.0208,
      "step": 5898
    },
    {
      "epoch": 1.342207053469852,
      "grad_norm": 0.7094173678000857,
      "learning_rate": 1.040677233097312e-06,
      "loss": 0.0238,
      "step": 5899
    },
    {
      "epoch": 1.342434584755404,
      "grad_norm": 0.6444151744345981,
      "learning_rate": 1.0406105040545509e-06,
      "loss": 0.0161,
      "step": 5900
    },
    {
      "epoch": 1.3426621160409558,
      "grad_norm": 1.379128100384456,
      "learning_rate": 1.0405437665174719e-06,
      "loss": 0.0388,
      "step": 5901
    },
    {
      "epoch": 1.3428896473265075,
      "grad_norm": 0.8946926394570293,
      "learning_rate": 1.0404770204874396e-06,
      "loss": 0.0267,
      "step": 5902
    },
    {
      "epoch": 1.3431171786120593,
      "grad_norm": 2.042177852704746,
      "learning_rate": 1.040410265965818e-06,
      "loss": 0.0404,
      "step": 5903
    },
    {
      "epoch": 1.343344709897611,
      "grad_norm": 1.3077906262962142,
      "learning_rate": 1.0403435029539711e-06,
      "loss": 0.0308,
      "step": 5904
    },
    {
      "epoch": 1.3435722411831628,
      "grad_norm": 1.0324100339589763,
      "learning_rate": 1.0402767314532638e-06,
      "loss": 0.0206,
      "step": 5905
    },
    {
      "epoch": 1.3437997724687145,
      "grad_norm": 1.0265663580127615,
      "learning_rate": 1.0402099514650607e-06,
      "loss": 0.0324,
      "step": 5906
    },
    {
      "epoch": 1.3440273037542663,
      "grad_norm": 0.7549200836426128,
      "learning_rate": 1.0401431629907267e-06,
      "loss": 0.0121,
      "step": 5907
    },
    {
      "epoch": 1.344254835039818,
      "grad_norm": 0.7290621969799262,
      "learning_rate": 1.0400763660316265e-06,
      "loss": 0.0131,
      "step": 5908
    },
    {
      "epoch": 1.3444823663253698,
      "grad_norm": 1.119125346963153,
      "learning_rate": 1.0400095605891258e-06,
      "loss": 0.0423,
      "step": 5909
    },
    {
      "epoch": 1.3447098976109215,
      "grad_norm": 1.4041545259060262,
      "learning_rate": 1.0399427466645895e-06,
      "loss": 0.0256,
      "step": 5910
    },
    {
      "epoch": 1.3449374288964733,
      "grad_norm": 1.0699960135340252,
      "learning_rate": 1.0398759242593834e-06,
      "loss": 0.0193,
      "step": 5911
    },
    {
      "epoch": 1.345164960182025,
      "grad_norm": 1.4394734119461763,
      "learning_rate": 1.0398090933748733e-06,
      "loss": 0.0451,
      "step": 5912
    },
    {
      "epoch": 1.3453924914675768,
      "grad_norm": 0.7328296507836,
      "learning_rate": 1.039742254012425e-06,
      "loss": 0.0117,
      "step": 5913
    },
    {
      "epoch": 1.3456200227531285,
      "grad_norm": 0.9315877067634672,
      "learning_rate": 1.0396754061734047e-06,
      "loss": 0.0291,
      "step": 5914
    },
    {
      "epoch": 1.3458475540386803,
      "grad_norm": 1.1230101632788236,
      "learning_rate": 1.0396085498591785e-06,
      "loss": 0.0134,
      "step": 5915
    },
    {
      "epoch": 1.346075085324232,
      "grad_norm": 1.0983618909120694,
      "learning_rate": 1.0395416850711127e-06,
      "loss": 0.02,
      "step": 5916
    },
    {
      "epoch": 1.3463026166097838,
      "grad_norm": 0.6035953597253712,
      "learning_rate": 1.0394748118105743e-06,
      "loss": 0.0092,
      "step": 5917
    },
    {
      "epoch": 1.3465301478953355,
      "grad_norm": 0.5057823769699702,
      "learning_rate": 1.0394079300789296e-06,
      "loss": 0.006,
      "step": 5918
    },
    {
      "epoch": 1.3467576791808873,
      "grad_norm": 1.2058457720453324,
      "learning_rate": 1.0393410398775459e-06,
      "loss": 0.0381,
      "step": 5919
    },
    {
      "epoch": 1.346985210466439,
      "grad_norm": 1.0689739159271736,
      "learning_rate": 1.03927414120779e-06,
      "loss": 0.0128,
      "step": 5920
    },
    {
      "epoch": 1.3472127417519908,
      "grad_norm": 1.3055479840732933,
      "learning_rate": 1.0392072340710296e-06,
      "loss": 0.0441,
      "step": 5921
    },
    {
      "epoch": 1.3474402730375425,
      "grad_norm": 1.387509390007852,
      "learning_rate": 1.0391403184686318e-06,
      "loss": 0.0302,
      "step": 5922
    },
    {
      "epoch": 1.3476678043230943,
      "grad_norm": 1.1536233789260337,
      "learning_rate": 1.0390733944019645e-06,
      "loss": 0.0202,
      "step": 5923
    },
    {
      "epoch": 1.347895335608646,
      "grad_norm": 0.9911221130460622,
      "learning_rate": 1.0390064618723952e-06,
      "loss": 0.0153,
      "step": 5924
    },
    {
      "epoch": 1.348122866894198,
      "grad_norm": 0.9561197103111988,
      "learning_rate": 1.038939520881292e-06,
      "loss": 0.0114,
      "step": 5925
    },
    {
      "epoch": 1.3483503981797498,
      "grad_norm": 1.102900309683922,
      "learning_rate": 1.038872571430023e-06,
      "loss": 0.0337,
      "step": 5926
    },
    {
      "epoch": 1.3485779294653015,
      "grad_norm": 1.2902418265691835,
      "learning_rate": 1.038805613519957e-06,
      "loss": 0.04,
      "step": 5927
    },
    {
      "epoch": 1.3488054607508533,
      "grad_norm": 1.420709768769003,
      "learning_rate": 1.038738647152462e-06,
      "loss": 0.0199,
      "step": 5928
    },
    {
      "epoch": 1.349032992036405,
      "grad_norm": 0.8364236936432923,
      "learning_rate": 1.0386716723289063e-06,
      "loss": 0.0222,
      "step": 5929
    },
    {
      "epoch": 1.3492605233219568,
      "grad_norm": 1.0505205577839964,
      "learning_rate": 1.0386046890506596e-06,
      "loss": 0.0248,
      "step": 5930
    },
    {
      "epoch": 1.3494880546075085,
      "grad_norm": 1.401571027676419,
      "learning_rate": 1.0385376973190906e-06,
      "loss": 0.0399,
      "step": 5931
    },
    {
      "epoch": 1.3497155858930603,
      "grad_norm": 1.1032877927835119,
      "learning_rate": 1.0384706971355683e-06,
      "loss": 0.0273,
      "step": 5932
    },
    {
      "epoch": 1.349943117178612,
      "grad_norm": 0.7007490772962891,
      "learning_rate": 1.038403688501462e-06,
      "loss": 0.0102,
      "step": 5933
    },
    {
      "epoch": 1.3501706484641638,
      "grad_norm": 0.8472863824911138,
      "learning_rate": 1.0383366714181419e-06,
      "loss": 0.011,
      "step": 5934
    },
    {
      "epoch": 1.3503981797497155,
      "grad_norm": 0.8977164627217846,
      "learning_rate": 1.038269645886977e-06,
      "loss": 0.0167,
      "step": 5935
    },
    {
      "epoch": 1.3506257110352673,
      "grad_norm": 1.2746342659674816,
      "learning_rate": 1.0382026119093372e-06,
      "loss": 0.0272,
      "step": 5936
    },
    {
      "epoch": 1.350853242320819,
      "grad_norm": 1.5516990871708691,
      "learning_rate": 1.038135569486593e-06,
      "loss": 0.0292,
      "step": 5937
    },
    {
      "epoch": 1.3510807736063708,
      "grad_norm": 1.9165286699452406,
      "learning_rate": 1.0380685186201143e-06,
      "loss": 0.0594,
      "step": 5938
    },
    {
      "epoch": 1.3513083048919228,
      "grad_norm": 1.3297283101826065,
      "learning_rate": 1.0380014593112714e-06,
      "loss": 0.0568,
      "step": 5939
    },
    {
      "epoch": 1.3515358361774745,
      "grad_norm": 1.2661268564639319,
      "learning_rate": 1.0379343915614354e-06,
      "loss": 0.0433,
      "step": 5940
    },
    {
      "epoch": 1.3517633674630263,
      "grad_norm": 1.0448046867479912,
      "learning_rate": 1.0378673153719764e-06,
      "loss": 0.0288,
      "step": 5941
    },
    {
      "epoch": 1.351990898748578,
      "grad_norm": 0.6263965338196285,
      "learning_rate": 1.0378002307442659e-06,
      "loss": 0.0067,
      "step": 5942
    },
    {
      "epoch": 1.3522184300341298,
      "grad_norm": 0.9186544964954529,
      "learning_rate": 1.0377331376796745e-06,
      "loss": 0.034,
      "step": 5943
    },
    {
      "epoch": 1.3524459613196815,
      "grad_norm": 1.3175528661105624,
      "learning_rate": 1.0376660361795738e-06,
      "loss": 0.0577,
      "step": 5944
    },
    {
      "epoch": 1.3526734926052333,
      "grad_norm": 1.4599781406050027,
      "learning_rate": 1.0375989262453348e-06,
      "loss": 0.0447,
      "step": 5945
    },
    {
      "epoch": 1.352901023890785,
      "grad_norm": 1.1156489974661725,
      "learning_rate": 1.0375318078783294e-06,
      "loss": 0.0511,
      "step": 5946
    },
    {
      "epoch": 1.3531285551763368,
      "grad_norm": 0.9483213458533444,
      "learning_rate": 1.0374646810799297e-06,
      "loss": 0.0238,
      "step": 5947
    },
    {
      "epoch": 1.3533560864618885,
      "grad_norm": 1.2661883390274016,
      "learning_rate": 1.037397545851507e-06,
      "loss": 0.0429,
      "step": 5948
    },
    {
      "epoch": 1.3535836177474403,
      "grad_norm": 1.0018616531819395,
      "learning_rate": 1.0373304021944338e-06,
      "loss": 0.0221,
      "step": 5949
    },
    {
      "epoch": 1.353811149032992,
      "grad_norm": 0.7808472720700552,
      "learning_rate": 1.0372632501100826e-06,
      "loss": 0.0091,
      "step": 5950
    },
    {
      "epoch": 1.3540386803185438,
      "grad_norm": 0.8140478794271282,
      "learning_rate": 1.0371960895998252e-06,
      "loss": 0.0145,
      "step": 5951
    },
    {
      "epoch": 1.3542662116040955,
      "grad_norm": 0.9117276219302244,
      "learning_rate": 1.0371289206650349e-06,
      "loss": 0.0201,
      "step": 5952
    },
    {
      "epoch": 1.3544937428896473,
      "grad_norm": 1.2061163633959278,
      "learning_rate": 1.0370617433070842e-06,
      "loss": 0.0322,
      "step": 5953
    },
    {
      "epoch": 1.354721274175199,
      "grad_norm": 0.7876722921495687,
      "learning_rate": 1.036994557527346e-06,
      "loss": 0.0144,
      "step": 5954
    },
    {
      "epoch": 1.3549488054607508,
      "grad_norm": 0.6613445307925959,
      "learning_rate": 1.0369273633271936e-06,
      "loss": 0.0069,
      "step": 5955
    },
    {
      "epoch": 1.3551763367463026,
      "grad_norm": 0.6345409248116708,
      "learning_rate": 1.0368601607080004e-06,
      "loss": 0.0155,
      "step": 5956
    },
    {
      "epoch": 1.3554038680318543,
      "grad_norm": 1.8147704904537438,
      "learning_rate": 1.0367929496711397e-06,
      "loss": 0.039,
      "step": 5957
    },
    {
      "epoch": 1.355631399317406,
      "grad_norm": 0.833931276693109,
      "learning_rate": 1.0367257302179853e-06,
      "loss": 0.0146,
      "step": 5958
    },
    {
      "epoch": 1.3558589306029578,
      "grad_norm": 0.7692458341928677,
      "learning_rate": 1.036658502349911e-06,
      "loss": 0.0125,
      "step": 5959
    },
    {
      "epoch": 1.3560864618885096,
      "grad_norm": 0.8091203823295923,
      "learning_rate": 1.0365912660682908e-06,
      "loss": 0.0216,
      "step": 5960
    },
    {
      "epoch": 1.3563139931740613,
      "grad_norm": 2.503585891485645,
      "learning_rate": 1.036524021374499e-06,
      "loss": 0.0419,
      "step": 5961
    },
    {
      "epoch": 1.356541524459613,
      "grad_norm": 1.0464032831404197,
      "learning_rate": 1.0364567682699098e-06,
      "loss": 0.0279,
      "step": 5962
    },
    {
      "epoch": 1.356769055745165,
      "grad_norm": 1.0400014436949596,
      "learning_rate": 1.036389506755898e-06,
      "loss": 0.0202,
      "step": 5963
    },
    {
      "epoch": 1.3569965870307168,
      "grad_norm": 0.9813071406643438,
      "learning_rate": 1.036322236833838e-06,
      "loss": 0.0235,
      "step": 5964
    },
    {
      "epoch": 1.3572241183162685,
      "grad_norm": 1.118519599019973,
      "learning_rate": 1.0362549585051046e-06,
      "loss": 0.0308,
      "step": 5965
    },
    {
      "epoch": 1.3574516496018203,
      "grad_norm": 0.5898336323005567,
      "learning_rate": 1.0361876717710731e-06,
      "loss": 0.013,
      "step": 5966
    },
    {
      "epoch": 1.357679180887372,
      "grad_norm": 1.361363746809614,
      "learning_rate": 1.0361203766331187e-06,
      "loss": 0.034,
      "step": 5967
    },
    {
      "epoch": 1.3579067121729238,
      "grad_norm": 0.6464652035747823,
      "learning_rate": 1.036053073092617e-06,
      "loss": 0.0129,
      "step": 5968
    },
    {
      "epoch": 1.3581342434584756,
      "grad_norm": 1.8759837911851749,
      "learning_rate": 1.0359857611509428e-06,
      "loss": 0.0321,
      "step": 5969
    },
    {
      "epoch": 1.3583617747440273,
      "grad_norm": 1.619283823736038,
      "learning_rate": 1.0359184408094726e-06,
      "loss": 0.0301,
      "step": 5970
    },
    {
      "epoch": 1.358589306029579,
      "grad_norm": 2.0895320562539723,
      "learning_rate": 1.0358511120695819e-06,
      "loss": 0.0571,
      "step": 5971
    },
    {
      "epoch": 1.3588168373151308,
      "grad_norm": 1.1559191193621738,
      "learning_rate": 1.0357837749326471e-06,
      "loss": 0.0305,
      "step": 5972
    },
    {
      "epoch": 1.3590443686006826,
      "grad_norm": 1.1597343434931056,
      "learning_rate": 1.0357164294000442e-06,
      "loss": 0.0301,
      "step": 5973
    },
    {
      "epoch": 1.3592718998862343,
      "grad_norm": 1.1311805121187837,
      "learning_rate": 1.0356490754731496e-06,
      "loss": 0.022,
      "step": 5974
    },
    {
      "epoch": 1.359499431171786,
      "grad_norm": 0.8235807140315661,
      "learning_rate": 1.03558171315334e-06,
      "loss": 0.0125,
      "step": 5975
    },
    {
      "epoch": 1.3597269624573378,
      "grad_norm": 0.8840629724459017,
      "learning_rate": 1.0355143424419922e-06,
      "loss": 0.0124,
      "step": 5976
    },
    {
      "epoch": 1.3599544937428896,
      "grad_norm": 1.674273107123145,
      "learning_rate": 1.035446963340483e-06,
      "loss": 0.0611,
      "step": 5977
    },
    {
      "epoch": 1.3601820250284415,
      "grad_norm": 1.0994944800187416,
      "learning_rate": 1.0353795758501894e-06,
      "loss": 0.0184,
      "step": 5978
    },
    {
      "epoch": 1.3604095563139933,
      "grad_norm": 1.0588312713917787,
      "learning_rate": 1.0353121799724892e-06,
      "loss": 0.0194,
      "step": 5979
    },
    {
      "epoch": 1.360637087599545,
      "grad_norm": 1.3104261203754106,
      "learning_rate": 1.0352447757087592e-06,
      "loss": 0.0296,
      "step": 5980
    },
    {
      "epoch": 1.3608646188850968,
      "grad_norm": 1.3004787031070375,
      "learning_rate": 1.0351773630603774e-06,
      "loss": 0.0204,
      "step": 5981
    },
    {
      "epoch": 1.3610921501706486,
      "grad_norm": 1.0659064927726833,
      "learning_rate": 1.0351099420287213e-06,
      "loss": 0.036,
      "step": 5982
    },
    {
      "epoch": 1.3613196814562003,
      "grad_norm": 1.1812994279020221,
      "learning_rate": 1.0350425126151694e-06,
      "loss": 0.0248,
      "step": 5983
    },
    {
      "epoch": 1.361547212741752,
      "grad_norm": 0.9622198048005316,
      "learning_rate": 1.0349750748210994e-06,
      "loss": 0.0132,
      "step": 5984
    },
    {
      "epoch": 1.3617747440273038,
      "grad_norm": 0.7133679642974851,
      "learning_rate": 1.0349076286478897e-06,
      "loss": 0.0086,
      "step": 5985
    },
    {
      "epoch": 1.3620022753128556,
      "grad_norm": 1.3396910291108761,
      "learning_rate": 1.0348401740969188e-06,
      "loss": 0.0457,
      "step": 5986
    },
    {
      "epoch": 1.3622298065984073,
      "grad_norm": 0.976016551467127,
      "learning_rate": 1.0347727111695652e-06,
      "loss": 0.0182,
      "step": 5987
    },
    {
      "epoch": 1.362457337883959,
      "grad_norm": 1.2706452532311896,
      "learning_rate": 1.0347052398672079e-06,
      "loss": 0.0337,
      "step": 5988
    },
    {
      "epoch": 1.3626848691695108,
      "grad_norm": 1.1241759656924852,
      "learning_rate": 1.034637760191226e-06,
      "loss": 0.0451,
      "step": 5989
    },
    {
      "epoch": 1.3629124004550626,
      "grad_norm": 0.9870356931300208,
      "learning_rate": 1.0345702721429982e-06,
      "loss": 0.0292,
      "step": 5990
    },
    {
      "epoch": 1.3631399317406143,
      "grad_norm": 0.7055169330755962,
      "learning_rate": 1.0345027757239044e-06,
      "loss": 0.0146,
      "step": 5991
    },
    {
      "epoch": 1.363367463026166,
      "grad_norm": 1.4129407289578744,
      "learning_rate": 1.0344352709353237e-06,
      "loss": 0.049,
      "step": 5992
    },
    {
      "epoch": 1.3635949943117178,
      "grad_norm": 0.6615348229173598,
      "learning_rate": 1.034367757778636e-06,
      "loss": 0.0161,
      "step": 5993
    },
    {
      "epoch": 1.3638225255972696,
      "grad_norm": 1.3319609442015357,
      "learning_rate": 1.034300236255221e-06,
      "loss": 0.0142,
      "step": 5994
    },
    {
      "epoch": 1.3640500568828213,
      "grad_norm": 1.101573823862616,
      "learning_rate": 1.0342327063664587e-06,
      "loss": 0.0252,
      "step": 5995
    },
    {
      "epoch": 1.364277588168373,
      "grad_norm": 0.49851149150245927,
      "learning_rate": 1.0341651681137293e-06,
      "loss": 0.0058,
      "step": 5996
    },
    {
      "epoch": 1.3645051194539248,
      "grad_norm": 0.7916106342215998,
      "learning_rate": 1.0340976214984136e-06,
      "loss": 0.012,
      "step": 5997
    },
    {
      "epoch": 1.3647326507394766,
      "grad_norm": 1.3033775616866645,
      "learning_rate": 1.0340300665218913e-06,
      "loss": 0.0224,
      "step": 5998
    },
    {
      "epoch": 1.3649601820250283,
      "grad_norm": 1.1530071908333883,
      "learning_rate": 1.0339625031855438e-06,
      "loss": 0.0264,
      "step": 5999
    },
    {
      "epoch": 1.36518771331058,
      "grad_norm": 1.7119840122096044,
      "learning_rate": 1.0338949314907515e-06,
      "loss": 0.0542,
      "step": 6000
    },
    {
      "epoch": 1.3654152445961318,
      "grad_norm": 1.0229925474062507,
      "learning_rate": 1.0338273514388958e-06,
      "loss": 0.0439,
      "step": 6001
    },
    {
      "epoch": 1.3656427758816838,
      "grad_norm": 1.2452426043381837,
      "learning_rate": 1.0337597630313578e-06,
      "loss": 0.015,
      "step": 6002
    },
    {
      "epoch": 1.3658703071672356,
      "grad_norm": 1.8653310977298578,
      "learning_rate": 1.0336921662695188e-06,
      "loss": 0.0488,
      "step": 6003
    },
    {
      "epoch": 1.3660978384527873,
      "grad_norm": 0.6741864416257192,
      "learning_rate": 1.0336245611547605e-06,
      "loss": 0.0103,
      "step": 6004
    },
    {
      "epoch": 1.366325369738339,
      "grad_norm": 0.7051952693815983,
      "learning_rate": 1.0335569476884643e-06,
      "loss": 0.014,
      "step": 6005
    },
    {
      "epoch": 1.3665529010238908,
      "grad_norm": 0.9708525738875694,
      "learning_rate": 1.0334893258720124e-06,
      "loss": 0.0151,
      "step": 6006
    },
    {
      "epoch": 1.3667804323094426,
      "grad_norm": 1.109837397667383,
      "learning_rate": 1.033421695706787e-06,
      "loss": 0.0233,
      "step": 6007
    },
    {
      "epoch": 1.3670079635949943,
      "grad_norm": 0.9083533358364363,
      "learning_rate": 1.03335405719417e-06,
      "loss": 0.0135,
      "step": 6008
    },
    {
      "epoch": 1.367235494880546,
      "grad_norm": 1.1720567751480195,
      "learning_rate": 1.0332864103355438e-06,
      "loss": 0.015,
      "step": 6009
    },
    {
      "epoch": 1.3674630261660978,
      "grad_norm": 0.7630354357142868,
      "learning_rate": 1.0332187551322914e-06,
      "loss": 0.0128,
      "step": 6010
    },
    {
      "epoch": 1.3676905574516496,
      "grad_norm": 0.6765984321238594,
      "learning_rate": 1.0331510915857951e-06,
      "loss": 0.0138,
      "step": 6011
    },
    {
      "epoch": 1.3679180887372013,
      "grad_norm": 0.528880382934291,
      "learning_rate": 1.0330834196974378e-06,
      "loss": 0.006,
      "step": 6012
    },
    {
      "epoch": 1.368145620022753,
      "grad_norm": 0.9749392678640607,
      "learning_rate": 1.0330157394686031e-06,
      "loss": 0.0291,
      "step": 6013
    },
    {
      "epoch": 1.3683731513083048,
      "grad_norm": 1.6667345822917836,
      "learning_rate": 1.0329480509006738e-06,
      "loss": 0.0284,
      "step": 6014
    },
    {
      "epoch": 1.3686006825938566,
      "grad_norm": 0.9719828123924177,
      "learning_rate": 1.0328803539950332e-06,
      "loss": 0.0161,
      "step": 6015
    },
    {
      "epoch": 1.3688282138794083,
      "grad_norm": 0.6687579179263169,
      "learning_rate": 1.0328126487530657e-06,
      "loss": 0.0171,
      "step": 6016
    },
    {
      "epoch": 1.3690557451649603,
      "grad_norm": 0.8484600929868883,
      "learning_rate": 1.0327449351761542e-06,
      "loss": 0.0234,
      "step": 6017
    },
    {
      "epoch": 1.369283276450512,
      "grad_norm": 0.9980151540533014,
      "learning_rate": 1.0326772132656828e-06,
      "loss": 0.0228,
      "step": 6018
    },
    {
      "epoch": 1.3695108077360638,
      "grad_norm": 1.432111946362771,
      "learning_rate": 1.0326094830230362e-06,
      "loss": 0.0265,
      "step": 6019
    },
    {
      "epoch": 1.3697383390216156,
      "grad_norm": 3.205956674239623,
      "learning_rate": 1.032541744449598e-06,
      "loss": 0.0793,
      "step": 6020
    },
    {
      "epoch": 1.3699658703071673,
      "grad_norm": 1.033249260757496,
      "learning_rate": 1.0324739975467529e-06,
      "loss": 0.042,
      "step": 6021
    },
    {
      "epoch": 1.370193401592719,
      "grad_norm": 0.9157158705373819,
      "learning_rate": 1.0324062423158857e-06,
      "loss": 0.0294,
      "step": 6022
    },
    {
      "epoch": 1.3704209328782708,
      "grad_norm": 0.9368902048923686,
      "learning_rate": 1.0323384787583809e-06,
      "loss": 0.0244,
      "step": 6023
    },
    {
      "epoch": 1.3706484641638226,
      "grad_norm": 1.4416896631494844,
      "learning_rate": 1.0322707068756238e-06,
      "loss": 0.0353,
      "step": 6024
    },
    {
      "epoch": 1.3708759954493743,
      "grad_norm": 1.2550874861994528,
      "learning_rate": 1.0322029266689992e-06,
      "loss": 0.058,
      "step": 6025
    },
    {
      "epoch": 1.371103526734926,
      "grad_norm": 1.3124105011895348,
      "learning_rate": 1.0321351381398926e-06,
      "loss": 0.0274,
      "step": 6026
    },
    {
      "epoch": 1.3713310580204778,
      "grad_norm": 0.8889050886123729,
      "learning_rate": 1.0320673412896891e-06,
      "loss": 0.0268,
      "step": 6027
    },
    {
      "epoch": 1.3715585893060296,
      "grad_norm": 1.1494781940988754,
      "learning_rate": 1.0319995361197752e-06,
      "loss": 0.025,
      "step": 6028
    },
    {
      "epoch": 1.3717861205915813,
      "grad_norm": 1.3212407031910525,
      "learning_rate": 1.0319317226315358e-06,
      "loss": 0.0298,
      "step": 6029
    },
    {
      "epoch": 1.372013651877133,
      "grad_norm": 1.286243454157273,
      "learning_rate": 1.0318639008263572e-06,
      "loss": 0.0342,
      "step": 6030
    },
    {
      "epoch": 1.3722411831626848,
      "grad_norm": 0.8914612174450923,
      "learning_rate": 1.0317960707056256e-06,
      "loss": 0.0126,
      "step": 6031
    },
    {
      "epoch": 1.3724687144482366,
      "grad_norm": 0.7573600841393602,
      "learning_rate": 1.0317282322707275e-06,
      "loss": 0.0197,
      "step": 6032
    },
    {
      "epoch": 1.3726962457337883,
      "grad_norm": 0.5345772069608944,
      "learning_rate": 1.0316603855230492e-06,
      "loss": 0.0111,
      "step": 6033
    },
    {
      "epoch": 1.37292377701934,
      "grad_norm": 0.8338590952425786,
      "learning_rate": 1.0315925304639773e-06,
      "loss": 0.0179,
      "step": 6034
    },
    {
      "epoch": 1.3731513083048918,
      "grad_norm": 0.9724921448714786,
      "learning_rate": 1.0315246670948988e-06,
      "loss": 0.0169,
      "step": 6035
    },
    {
      "epoch": 1.3733788395904436,
      "grad_norm": 0.873659813292738,
      "learning_rate": 1.0314567954172006e-06,
      "loss": 0.0277,
      "step": 6036
    },
    {
      "epoch": 1.3736063708759954,
      "grad_norm": 0.9265881375432587,
      "learning_rate": 1.03138891543227e-06,
      "loss": 0.0364,
      "step": 6037
    },
    {
      "epoch": 1.373833902161547,
      "grad_norm": 1.1953217652768424,
      "learning_rate": 1.031321027141494e-06,
      "loss": 0.0279,
      "step": 6038
    },
    {
      "epoch": 1.3740614334470989,
      "grad_norm": 1.6682562329608188,
      "learning_rate": 1.0312531305462607e-06,
      "loss": 0.0443,
      "step": 6039
    },
    {
      "epoch": 1.3742889647326506,
      "grad_norm": 1.1077668460967074,
      "learning_rate": 1.031185225647957e-06,
      "loss": 0.0163,
      "step": 6040
    },
    {
      "epoch": 1.3745164960182026,
      "grad_norm": 0.648925246369876,
      "learning_rate": 1.0311173124479715e-06,
      "loss": 0.0091,
      "step": 6041
    },
    {
      "epoch": 1.3747440273037543,
      "grad_norm": 0.8886275125074955,
      "learning_rate": 1.0310493909476916e-06,
      "loss": 0.0224,
      "step": 6042
    },
    {
      "epoch": 1.374971558589306,
      "grad_norm": 0.8524740196914463,
      "learning_rate": 1.0309814611485062e-06,
      "loss": 0.0259,
      "step": 6043
    },
    {
      "epoch": 1.3751990898748578,
      "grad_norm": 0.9742799558119335,
      "learning_rate": 1.0309135230518028e-06,
      "loss": 0.0223,
      "step": 6044
    },
    {
      "epoch": 1.3754266211604096,
      "grad_norm": 1.383114536961926,
      "learning_rate": 1.0308455766589706e-06,
      "loss": 0.0319,
      "step": 6045
    },
    {
      "epoch": 1.3756541524459613,
      "grad_norm": 1.3723012718452987,
      "learning_rate": 1.030777621971398e-06,
      "loss": 0.0293,
      "step": 6046
    },
    {
      "epoch": 1.375881683731513,
      "grad_norm": 2.4276688201868044,
      "learning_rate": 1.0307096589904742e-06,
      "loss": 0.0765,
      "step": 6047
    },
    {
      "epoch": 1.3761092150170648,
      "grad_norm": 1.3054528485981858,
      "learning_rate": 1.030641687717588e-06,
      "loss": 0.0282,
      "step": 6048
    },
    {
      "epoch": 1.3763367463026166,
      "grad_norm": 1.3658620692254781,
      "learning_rate": 1.0305737081541283e-06,
      "loss": 0.019,
      "step": 6049
    },
    {
      "epoch": 1.3765642775881684,
      "grad_norm": 1.2988678220894119,
      "learning_rate": 1.0305057203014848e-06,
      "loss": 0.0457,
      "step": 6050
    },
    {
      "epoch": 1.37679180887372,
      "grad_norm": 1.8263728967860204,
      "learning_rate": 1.0304377241610472e-06,
      "loss": 0.0681,
      "step": 6051
    },
    {
      "epoch": 1.3770193401592719,
      "grad_norm": 0.48009564319798714,
      "learning_rate": 1.030369719734205e-06,
      "loss": 0.0078,
      "step": 6052
    },
    {
      "epoch": 1.3772468714448236,
      "grad_norm": 1.5065012818800154,
      "learning_rate": 1.0303017070223482e-06,
      "loss": 0.0227,
      "step": 6053
    },
    {
      "epoch": 1.3774744027303754,
      "grad_norm": 0.5740072329076822,
      "learning_rate": 1.0302336860268667e-06,
      "loss": 0.0198,
      "step": 6054
    },
    {
      "epoch": 1.377701934015927,
      "grad_norm": 0.9008705562256887,
      "learning_rate": 1.0301656567491507e-06,
      "loss": 0.0209,
      "step": 6055
    },
    {
      "epoch": 1.377929465301479,
      "grad_norm": 0.9052594350778648,
      "learning_rate": 1.0300976191905907e-06,
      "loss": 0.0134,
      "step": 6056
    },
    {
      "epoch": 1.3781569965870308,
      "grad_norm": 1.008952514422943,
      "learning_rate": 1.0300295733525774e-06,
      "loss": 0.026,
      "step": 6057
    },
    {
      "epoch": 1.3783845278725826,
      "grad_norm": 0.8655700871947668,
      "learning_rate": 1.0299615192365015e-06,
      "loss": 0.0169,
      "step": 6058
    },
    {
      "epoch": 1.3786120591581343,
      "grad_norm": 1.0614823863183105,
      "learning_rate": 1.0298934568437535e-06,
      "loss": 0.025,
      "step": 6059
    },
    {
      "epoch": 1.378839590443686,
      "grad_norm": 1.2671893505632115,
      "learning_rate": 1.029825386175725e-06,
      "loss": 0.032,
      "step": 6060
    },
    {
      "epoch": 1.3790671217292378,
      "grad_norm": 0.8414566963740582,
      "learning_rate": 1.029757307233807e-06,
      "loss": 0.029,
      "step": 6061
    },
    {
      "epoch": 1.3792946530147896,
      "grad_norm": 1.4410574086008923,
      "learning_rate": 1.0296892200193908e-06,
      "loss": 0.0407,
      "step": 6062
    },
    {
      "epoch": 1.3795221843003413,
      "grad_norm": 1.160321097652087,
      "learning_rate": 1.029621124533868e-06,
      "loss": 0.0177,
      "step": 6063
    },
    {
      "epoch": 1.379749715585893,
      "grad_norm": 1.0069056688464664,
      "learning_rate": 1.0295530207786307e-06,
      "loss": 0.0153,
      "step": 6064
    },
    {
      "epoch": 1.3799772468714449,
      "grad_norm": 0.9343986607113232,
      "learning_rate": 1.0294849087550703e-06,
      "loss": 0.0236,
      "step": 6065
    },
    {
      "epoch": 1.3802047781569966,
      "grad_norm": 1.154725705666541,
      "learning_rate": 1.0294167884645795e-06,
      "loss": 0.0223,
      "step": 6066
    },
    {
      "epoch": 1.3804323094425484,
      "grad_norm": 1.106739391483581,
      "learning_rate": 1.02934865990855e-06,
      "loss": 0.0261,
      "step": 6067
    },
    {
      "epoch": 1.3806598407281,
      "grad_norm": 0.48989173855238655,
      "learning_rate": 1.0292805230883743e-06,
      "loss": 0.0084,
      "step": 6068
    },
    {
      "epoch": 1.3808873720136519,
      "grad_norm": 1.146572789777464,
      "learning_rate": 1.0292123780054452e-06,
      "loss": 0.0149,
      "step": 6069
    },
    {
      "epoch": 1.3811149032992036,
      "grad_norm": 1.508059128952438,
      "learning_rate": 1.0291442246611555e-06,
      "loss": 0.0348,
      "step": 6070
    },
    {
      "epoch": 1.3813424345847554,
      "grad_norm": 0.7179341697610161,
      "learning_rate": 1.029076063056898e-06,
      "loss": 0.0163,
      "step": 6071
    },
    {
      "epoch": 1.3815699658703071,
      "grad_norm": 1.6491912488062737,
      "learning_rate": 1.0290078931940656e-06,
      "loss": 0.0648,
      "step": 6072
    },
    {
      "epoch": 1.3817974971558589,
      "grad_norm": 0.7870360955520767,
      "learning_rate": 1.028939715074052e-06,
      "loss": 0.0152,
      "step": 6073
    },
    {
      "epoch": 1.3820250284414106,
      "grad_norm": 0.8138866022930463,
      "learning_rate": 1.0288715286982504e-06,
      "loss": 0.0189,
      "step": 6074
    },
    {
      "epoch": 1.3822525597269624,
      "grad_norm": 0.5554949851167332,
      "learning_rate": 1.0288033340680543e-06,
      "loss": 0.0088,
      "step": 6075
    },
    {
      "epoch": 1.3824800910125141,
      "grad_norm": 1.3233523342981628,
      "learning_rate": 1.0287351311848574e-06,
      "loss": 0.0433,
      "step": 6076
    },
    {
      "epoch": 1.3827076222980659,
      "grad_norm": 0.6777854674940008,
      "learning_rate": 1.028666920050054e-06,
      "loss": 0.0084,
      "step": 6077
    },
    {
      "epoch": 1.3829351535836176,
      "grad_norm": 1.2431302079486768,
      "learning_rate": 1.0285987006650381e-06,
      "loss": 0.0264,
      "step": 6078
    },
    {
      "epoch": 1.3831626848691694,
      "grad_norm": 1.0244884125500628,
      "learning_rate": 1.028530473031204e-06,
      "loss": 0.0354,
      "step": 6079
    },
    {
      "epoch": 1.3833902161547214,
      "grad_norm": 0.7987683729662631,
      "learning_rate": 1.0284622371499457e-06,
      "loss": 0.0082,
      "step": 6080
    },
    {
      "epoch": 1.383617747440273,
      "grad_norm": 1.1972261120154417,
      "learning_rate": 1.0283939930226584e-06,
      "loss": 0.0416,
      "step": 6081
    },
    {
      "epoch": 1.3838452787258249,
      "grad_norm": 0.7965597712940404,
      "learning_rate": 1.0283257406507366e-06,
      "loss": 0.0168,
      "step": 6082
    },
    {
      "epoch": 1.3840728100113766,
      "grad_norm": 1.678085519895521,
      "learning_rate": 1.0282574800355755e-06,
      "loss": 0.0396,
      "step": 6083
    },
    {
      "epoch": 1.3843003412969284,
      "grad_norm": 1.1602726953842701,
      "learning_rate": 1.0281892111785699e-06,
      "loss": 0.03,
      "step": 6084
    },
    {
      "epoch": 1.3845278725824801,
      "grad_norm": 0.5738281114784806,
      "learning_rate": 1.0281209340811151e-06,
      "loss": 0.0128,
      "step": 6085
    },
    {
      "epoch": 1.3847554038680319,
      "grad_norm": 1.2307920377037582,
      "learning_rate": 1.0280526487446069e-06,
      "loss": 0.0298,
      "step": 6086
    },
    {
      "epoch": 1.3849829351535836,
      "grad_norm": 1.0906060725259301,
      "learning_rate": 1.0279843551704409e-06,
      "loss": 0.0102,
      "step": 6087
    },
    {
      "epoch": 1.3852104664391354,
      "grad_norm": 1.1283393095113754,
      "learning_rate": 1.0279160533600121e-06,
      "loss": 0.0367,
      "step": 6088
    },
    {
      "epoch": 1.3854379977246871,
      "grad_norm": 1.1681818545410683,
      "learning_rate": 1.0278477433147176e-06,
      "loss": 0.0416,
      "step": 6089
    },
    {
      "epoch": 1.3856655290102389,
      "grad_norm": 1.1415077838638938,
      "learning_rate": 1.0277794250359529e-06,
      "loss": 0.0271,
      "step": 6090
    },
    {
      "epoch": 1.3858930602957906,
      "grad_norm": 1.5760408905808865,
      "learning_rate": 1.0277110985251142e-06,
      "loss": 0.0389,
      "step": 6091
    },
    {
      "epoch": 1.3861205915813424,
      "grad_norm": 2.721888747642395,
      "learning_rate": 1.0276427637835984e-06,
      "loss": 0.0733,
      "step": 6092
    },
    {
      "epoch": 1.3863481228668941,
      "grad_norm": 0.6394797109060689,
      "learning_rate": 1.0275744208128019e-06,
      "loss": 0.0085,
      "step": 6093
    },
    {
      "epoch": 1.3865756541524459,
      "grad_norm": 0.8175458312812267,
      "learning_rate": 1.0275060696141215e-06,
      "loss": 0.0158,
      "step": 6094
    },
    {
      "epoch": 1.3868031854379979,
      "grad_norm": 0.8768873105674826,
      "learning_rate": 1.027437710188954e-06,
      "loss": 0.0252,
      "step": 6095
    },
    {
      "epoch": 1.3870307167235496,
      "grad_norm": 1.1764335482126738,
      "learning_rate": 1.027369342538697e-06,
      "loss": 0.0383,
      "step": 6096
    },
    {
      "epoch": 1.3872582480091014,
      "grad_norm": 0.9523241021653495,
      "learning_rate": 1.0273009666647472e-06,
      "loss": 0.0148,
      "step": 6097
    },
    {
      "epoch": 1.387485779294653,
      "grad_norm": 1.1658646040112732,
      "learning_rate": 1.0272325825685028e-06,
      "loss": 0.0245,
      "step": 6098
    },
    {
      "epoch": 1.3877133105802049,
      "grad_norm": 0.7975610013435775,
      "learning_rate": 1.027164190251361e-06,
      "loss": 0.0165,
      "step": 6099
    },
    {
      "epoch": 1.3879408418657566,
      "grad_norm": 1.2045558237977083,
      "learning_rate": 1.0270957897147196e-06,
      "loss": 0.034,
      "step": 6100
    },
    {
      "epoch": 1.3881683731513084,
      "grad_norm": 1.1965276205251034,
      "learning_rate": 1.0270273809599764e-06,
      "loss": 0.0311,
      "step": 6101
    },
    {
      "epoch": 1.3883959044368601,
      "grad_norm": 0.851268325678731,
      "learning_rate": 1.0269589639885302e-06,
      "loss": 0.0167,
      "step": 6102
    },
    {
      "epoch": 1.3886234357224119,
      "grad_norm": 1.3240030725330403,
      "learning_rate": 1.0268905388017788e-06,
      "loss": 0.0269,
      "step": 6103
    },
    {
      "epoch": 1.3888509670079636,
      "grad_norm": 1.0202558011027276,
      "learning_rate": 1.0268221054011208e-06,
      "loss": 0.027,
      "step": 6104
    },
    {
      "epoch": 1.3890784982935154,
      "grad_norm": 1.2434564945488167,
      "learning_rate": 1.026753663787955e-06,
      "loss": 0.0185,
      "step": 6105
    },
    {
      "epoch": 1.3893060295790671,
      "grad_norm": 1.186497952145426,
      "learning_rate": 1.0266852139636799e-06,
      "loss": 0.0225,
      "step": 6106
    },
    {
      "epoch": 1.3895335608646189,
      "grad_norm": 1.6183733781772422,
      "learning_rate": 1.0266167559296946e-06,
      "loss": 0.0483,
      "step": 6107
    },
    {
      "epoch": 1.3897610921501706,
      "grad_norm": 2.1320506255788056,
      "learning_rate": 1.0265482896873986e-06,
      "loss": 0.0398,
      "step": 6108
    },
    {
      "epoch": 1.3899886234357224,
      "grad_norm": 0.7471768184650087,
      "learning_rate": 1.0264798152381907e-06,
      "loss": 0.0144,
      "step": 6109
    },
    {
      "epoch": 1.3902161547212741,
      "grad_norm": 1.1509914195740398,
      "learning_rate": 1.026411332583471e-06,
      "loss": 0.0267,
      "step": 6110
    },
    {
      "epoch": 1.3904436860068259,
      "grad_norm": 1.6574823707279855,
      "learning_rate": 1.0263428417246385e-06,
      "loss": 0.0395,
      "step": 6111
    },
    {
      "epoch": 1.3906712172923776,
      "grad_norm": 1.097456018217718,
      "learning_rate": 1.0262743426630935e-06,
      "loss": 0.0306,
      "step": 6112
    },
    {
      "epoch": 1.3908987485779294,
      "grad_norm": 0.8275409605053028,
      "learning_rate": 1.0262058354002357e-06,
      "loss": 0.0253,
      "step": 6113
    },
    {
      "epoch": 1.3911262798634811,
      "grad_norm": 0.741933960938598,
      "learning_rate": 1.0261373199374655e-06,
      "loss": 0.0117,
      "step": 6114
    },
    {
      "epoch": 1.391353811149033,
      "grad_norm": 0.806245283558632,
      "learning_rate": 1.026068796276183e-06,
      "loss": 0.0163,
      "step": 6115
    },
    {
      "epoch": 1.3915813424345846,
      "grad_norm": 1.2256667129702425,
      "learning_rate": 1.0260002644177892e-06,
      "loss": 0.0335,
      "step": 6116
    },
    {
      "epoch": 1.3918088737201364,
      "grad_norm": 1.433487886403296,
      "learning_rate": 1.025931724363684e-06,
      "loss": 0.0302,
      "step": 6117
    },
    {
      "epoch": 1.3920364050056881,
      "grad_norm": 1.2323013972098553,
      "learning_rate": 1.0258631761152687e-06,
      "loss": 0.0423,
      "step": 6118
    },
    {
      "epoch": 1.3922639362912401,
      "grad_norm": 1.844647486739168,
      "learning_rate": 1.0257946196739444e-06,
      "loss": 0.063,
      "step": 6119
    },
    {
      "epoch": 1.3924914675767919,
      "grad_norm": 0.8259588697399817,
      "learning_rate": 1.025726055041112e-06,
      "loss": 0.0201,
      "step": 6120
    },
    {
      "epoch": 1.3927189988623436,
      "grad_norm": 0.4898924275811768,
      "learning_rate": 1.0256574822181727e-06,
      "loss": 0.0174,
      "step": 6121
    },
    {
      "epoch": 1.3929465301478954,
      "grad_norm": 0.8058598212354585,
      "learning_rate": 1.0255889012065285e-06,
      "loss": 0.0172,
      "step": 6122
    },
    {
      "epoch": 1.3931740614334471,
      "grad_norm": 1.4437392885866793,
      "learning_rate": 1.025520312007581e-06,
      "loss": 0.0302,
      "step": 6123
    },
    {
      "epoch": 1.3934015927189989,
      "grad_norm": 1.1533771806604551,
      "learning_rate": 1.0254517146227314e-06,
      "loss": 0.0303,
      "step": 6124
    },
    {
      "epoch": 1.3936291240045506,
      "grad_norm": 1.7256511601711935,
      "learning_rate": 1.0253831090533823e-06,
      "loss": 0.0466,
      "step": 6125
    },
    {
      "epoch": 1.3938566552901024,
      "grad_norm": 1.6375877890419952,
      "learning_rate": 1.0253144953009357e-06,
      "loss": 0.0416,
      "step": 6126
    },
    {
      "epoch": 1.3940841865756541,
      "grad_norm": 1.210907244464415,
      "learning_rate": 1.025245873366794e-06,
      "loss": 0.0202,
      "step": 6127
    },
    {
      "epoch": 1.394311717861206,
      "grad_norm": 1.7819981672345377,
      "learning_rate": 1.0251772432523596e-06,
      "loss": 0.0485,
      "step": 6128
    },
    {
      "epoch": 1.3945392491467576,
      "grad_norm": 2.224440960495445,
      "learning_rate": 1.0251086049590355e-06,
      "loss": 0.0656,
      "step": 6129
    },
    {
      "epoch": 1.3947667804323094,
      "grad_norm": 1.0114089903725545,
      "learning_rate": 1.0250399584882239e-06,
      "loss": 0.0204,
      "step": 6130
    },
    {
      "epoch": 1.3949943117178611,
      "grad_norm": 1.0071772232385818,
      "learning_rate": 1.0249713038413285e-06,
      "loss": 0.0168,
      "step": 6131
    },
    {
      "epoch": 1.395221843003413,
      "grad_norm": 1.0545053650829948,
      "learning_rate": 1.024902641019752e-06,
      "loss": 0.0256,
      "step": 6132
    },
    {
      "epoch": 1.3954493742889647,
      "grad_norm": 0.901290123486691,
      "learning_rate": 1.024833970024898e-06,
      "loss": 0.0197,
      "step": 6133
    },
    {
      "epoch": 1.3956769055745166,
      "grad_norm": 1.3496311004496413,
      "learning_rate": 1.0247652908581697e-06,
      "loss": 0.0306,
      "step": 6134
    },
    {
      "epoch": 1.3959044368600684,
      "grad_norm": 1.3181866205349257,
      "learning_rate": 1.0246966035209712e-06,
      "loss": 0.0274,
      "step": 6135
    },
    {
      "epoch": 1.3961319681456201,
      "grad_norm": 1.1655739118413535,
      "learning_rate": 1.024627908014706e-06,
      "loss": 0.0272,
      "step": 6136
    },
    {
      "epoch": 1.3963594994311719,
      "grad_norm": 0.6907363078845099,
      "learning_rate": 1.0245592043407784e-06,
      "loss": 0.0159,
      "step": 6137
    },
    {
      "epoch": 1.3965870307167236,
      "grad_norm": 0.7193043243021984,
      "learning_rate": 1.0244904925005924e-06,
      "loss": 0.0096,
      "step": 6138
    },
    {
      "epoch": 1.3968145620022754,
      "grad_norm": 1.10393335812822,
      "learning_rate": 1.0244217724955523e-06,
      "loss": 0.0364,
      "step": 6139
    },
    {
      "epoch": 1.3970420932878271,
      "grad_norm": 1.0603654771057396,
      "learning_rate": 1.0243530443270627e-06,
      "loss": 0.0183,
      "step": 6140
    },
    {
      "epoch": 1.3972696245733789,
      "grad_norm": 0.7222265706967784,
      "learning_rate": 1.0242843079965281e-06,
      "loss": 0.0101,
      "step": 6141
    },
    {
      "epoch": 1.3974971558589306,
      "grad_norm": 1.6811181869493026,
      "learning_rate": 1.024215563505354e-06,
      "loss": 0.0257,
      "step": 6142
    },
    {
      "epoch": 1.3977246871444824,
      "grad_norm": 0.6300121361838538,
      "learning_rate": 1.0241468108549443e-06,
      "loss": 0.0109,
      "step": 6143
    },
    {
      "epoch": 1.3979522184300341,
      "grad_norm": 0.942491659732893,
      "learning_rate": 1.024078050046705e-06,
      "loss": 0.0307,
      "step": 6144
    },
    {
      "epoch": 1.398179749715586,
      "grad_norm": 1.455940520193875,
      "learning_rate": 1.0240092810820412e-06,
      "loss": 0.0234,
      "step": 6145
    },
    {
      "epoch": 1.3984072810011376,
      "grad_norm": 1.8583374809051696,
      "learning_rate": 1.0239405039623585e-06,
      "loss": 0.0429,
      "step": 6146
    },
    {
      "epoch": 1.3986348122866894,
      "grad_norm": 0.9130814882867986,
      "learning_rate": 1.0238717186890625e-06,
      "loss": 0.018,
      "step": 6147
    },
    {
      "epoch": 1.3988623435722412,
      "grad_norm": 1.1845147398658868,
      "learning_rate": 1.0238029252635591e-06,
      "loss": 0.0247,
      "step": 6148
    },
    {
      "epoch": 1.399089874857793,
      "grad_norm": 0.839565054462471,
      "learning_rate": 1.0237341236872544e-06,
      "loss": 0.0117,
      "step": 6149
    },
    {
      "epoch": 1.3993174061433447,
      "grad_norm": 0.7511833754662133,
      "learning_rate": 1.0236653139615542e-06,
      "loss": 0.0142,
      "step": 6150
    },
    {
      "epoch": 1.3995449374288964,
      "grad_norm": 0.7794622692038276,
      "learning_rate": 1.0235964960878655e-06,
      "loss": 0.0179,
      "step": 6151
    },
    {
      "epoch": 1.3997724687144482,
      "grad_norm": 0.9503219335351416,
      "learning_rate": 1.023527670067594e-06,
      "loss": 0.0213,
      "step": 6152
    },
    {
      "epoch": 1.4,
      "grad_norm": 1.10300561276624,
      "learning_rate": 1.023458835902147e-06,
      "loss": 0.0364,
      "step": 6153
    },
    {
      "epoch": 1.4002275312855517,
      "grad_norm": 1.28977779817073,
      "learning_rate": 1.0233899935929311e-06,
      "loss": 0.0373,
      "step": 6154
    },
    {
      "epoch": 1.4004550625711034,
      "grad_norm": 11.609520950996195,
      "learning_rate": 1.0233211431413534e-06,
      "loss": 0.0171,
      "step": 6155
    },
    {
      "epoch": 1.4006825938566552,
      "grad_norm": 0.8504313269319569,
      "learning_rate": 1.023252284548821e-06,
      "loss": 0.0224,
      "step": 6156
    },
    {
      "epoch": 1.400910125142207,
      "grad_norm": 1.4928827347926052,
      "learning_rate": 1.0231834178167412e-06,
      "loss": 0.04,
      "step": 6157
    },
    {
      "epoch": 1.401137656427759,
      "grad_norm": 0.8946676834128838,
      "learning_rate": 1.0231145429465216e-06,
      "loss": 0.0179,
      "step": 6158
    },
    {
      "epoch": 1.4013651877133106,
      "grad_norm": 1.1255364229538867,
      "learning_rate": 1.02304565993957e-06,
      "loss": 0.0204,
      "step": 6159
    },
    {
      "epoch": 1.4015927189988624,
      "grad_norm": 0.9409196567271754,
      "learning_rate": 1.022976768797294e-06,
      "loss": 0.0135,
      "step": 6160
    },
    {
      "epoch": 1.4018202502844141,
      "grad_norm": 1.6268367112917375,
      "learning_rate": 1.0229078695211015e-06,
      "loss": 0.0306,
      "step": 6161
    },
    {
      "epoch": 1.402047781569966,
      "grad_norm": 1.3902005456097397,
      "learning_rate": 1.0228389621124011e-06,
      "loss": 0.0304,
      "step": 6162
    },
    {
      "epoch": 1.4022753128555177,
      "grad_norm": 1.5439359170326747,
      "learning_rate": 1.022770046572601e-06,
      "loss": 0.0418,
      "step": 6163
    },
    {
      "epoch": 1.4025028441410694,
      "grad_norm": 1.0405833401494775,
      "learning_rate": 1.0227011229031095e-06,
      "loss": 0.027,
      "step": 6164
    },
    {
      "epoch": 1.4027303754266212,
      "grad_norm": 1.0531666297415323,
      "learning_rate": 1.0226321911053353e-06,
      "loss": 0.0154,
      "step": 6165
    },
    {
      "epoch": 1.402957906712173,
      "grad_norm": 1.1548106181399105,
      "learning_rate": 1.0225632511806873e-06,
      "loss": 0.0195,
      "step": 6166
    },
    {
      "epoch": 1.4031854379977247,
      "grad_norm": 0.8440928382130881,
      "learning_rate": 1.0224943031305747e-06,
      "loss": 0.0149,
      "step": 6167
    },
    {
      "epoch": 1.4034129692832764,
      "grad_norm": 1.262449747166055,
      "learning_rate": 1.0224253469564067e-06,
      "loss": 0.029,
      "step": 6168
    },
    {
      "epoch": 1.4036405005688282,
      "grad_norm": 1.4654938363999002,
      "learning_rate": 1.0223563826595923e-06,
      "loss": 0.0461,
      "step": 6169
    },
    {
      "epoch": 1.40386803185438,
      "grad_norm": 1.980710283153395,
      "learning_rate": 1.0222874102415412e-06,
      "loss": 0.0344,
      "step": 6170
    },
    {
      "epoch": 1.4040955631399317,
      "grad_norm": 1.5348477981590474,
      "learning_rate": 1.0222184297036628e-06,
      "loss": 0.0411,
      "step": 6171
    },
    {
      "epoch": 1.4043230944254836,
      "grad_norm": 1.4468014368091555,
      "learning_rate": 1.0221494410473674e-06,
      "loss": 0.0326,
      "step": 6172
    },
    {
      "epoch": 1.4045506257110354,
      "grad_norm": 1.0219615062149323,
      "learning_rate": 1.0220804442740648e-06,
      "loss": 0.0211,
      "step": 6173
    },
    {
      "epoch": 1.4047781569965871,
      "grad_norm": 1.0935211989568192,
      "learning_rate": 1.022011439385165e-06,
      "loss": 0.0193,
      "step": 6174
    },
    {
      "epoch": 1.405005688282139,
      "grad_norm": 1.2466607977301363,
      "learning_rate": 1.0219424263820784e-06,
      "loss": 0.0313,
      "step": 6175
    },
    {
      "epoch": 1.4052332195676907,
      "grad_norm": 1.3325638727923352,
      "learning_rate": 1.0218734052662158e-06,
      "loss": 0.0285,
      "step": 6176
    },
    {
      "epoch": 1.4054607508532424,
      "grad_norm": 1.2957750834848885,
      "learning_rate": 1.0218043760389875e-06,
      "loss": 0.04,
      "step": 6177
    },
    {
      "epoch": 1.4056882821387942,
      "grad_norm": 0.836332776641381,
      "learning_rate": 1.0217353387018045e-06,
      "loss": 0.0126,
      "step": 6178
    },
    {
      "epoch": 1.405915813424346,
      "grad_norm": 1.0248086467082256,
      "learning_rate": 1.0216662932560779e-06,
      "loss": 0.0201,
      "step": 6179
    },
    {
      "epoch": 1.4061433447098977,
      "grad_norm": 1.1101913879706153,
      "learning_rate": 1.0215972397032185e-06,
      "loss": 0.0195,
      "step": 6180
    },
    {
      "epoch": 1.4063708759954494,
      "grad_norm": 1.314224111106247,
      "learning_rate": 1.0215281780446378e-06,
      "loss": 0.0369,
      "step": 6181
    },
    {
      "epoch": 1.4065984072810012,
      "grad_norm": 1.4531464271881769,
      "learning_rate": 1.0214591082817477e-06,
      "loss": 0.0473,
      "step": 6182
    },
    {
      "epoch": 1.406825938566553,
      "grad_norm": 1.420628464660094,
      "learning_rate": 1.0213900304159592e-06,
      "loss": 0.0244,
      "step": 6183
    },
    {
      "epoch": 1.4070534698521047,
      "grad_norm": 1.039601484802879,
      "learning_rate": 1.0213209444486844e-06,
      "loss": 0.0141,
      "step": 6184
    },
    {
      "epoch": 1.4072810011376564,
      "grad_norm": 1.4560729122411875,
      "learning_rate": 1.0212518503813356e-06,
      "loss": 0.0354,
      "step": 6185
    },
    {
      "epoch": 1.4075085324232082,
      "grad_norm": 1.1373268608112097,
      "learning_rate": 1.0211827482153244e-06,
      "loss": 0.0256,
      "step": 6186
    },
    {
      "epoch": 1.40773606370876,
      "grad_norm": 1.9894618121379821,
      "learning_rate": 1.0211136379520636e-06,
      "loss": 0.0619,
      "step": 6187
    },
    {
      "epoch": 1.4079635949943117,
      "grad_norm": 1.0161488340364868,
      "learning_rate": 1.0210445195929653e-06,
      "loss": 0.0115,
      "step": 6188
    },
    {
      "epoch": 1.4081911262798634,
      "grad_norm": 1.209292084474128,
      "learning_rate": 1.0209753931394426e-06,
      "loss": 0.0385,
      "step": 6189
    },
    {
      "epoch": 1.4084186575654152,
      "grad_norm": 0.5428346750526308,
      "learning_rate": 1.0209062585929077e-06,
      "loss": 0.0073,
      "step": 6190
    },
    {
      "epoch": 1.408646188850967,
      "grad_norm": 1.264679875063447,
      "learning_rate": 1.0208371159547742e-06,
      "loss": 0.0157,
      "step": 6191
    },
    {
      "epoch": 1.4088737201365187,
      "grad_norm": 0.8975530088077646,
      "learning_rate": 1.020767965226455e-06,
      "loss": 0.0226,
      "step": 6192
    },
    {
      "epoch": 1.4091012514220704,
      "grad_norm": 1.2170421461893541,
      "learning_rate": 1.0206988064093633e-06,
      "loss": 0.0229,
      "step": 6193
    },
    {
      "epoch": 1.4093287827076222,
      "grad_norm": 1.4713267946729547,
      "learning_rate": 1.0206296395049128e-06,
      "loss": 0.026,
      "step": 6194
    },
    {
      "epoch": 1.409556313993174,
      "grad_norm": 0.799539608207469,
      "learning_rate": 1.020560464514517e-06,
      "loss": 0.0083,
      "step": 6195
    },
    {
      "epoch": 1.4097838452787257,
      "grad_norm": 1.2012299824313988,
      "learning_rate": 1.0204912814395898e-06,
      "loss": 0.0313,
      "step": 6196
    },
    {
      "epoch": 1.4100113765642777,
      "grad_norm": 0.6653702932749492,
      "learning_rate": 1.020422090281545e-06,
      "loss": 0.0091,
      "step": 6197
    },
    {
      "epoch": 1.4102389078498294,
      "grad_norm": 0.9721621164097385,
      "learning_rate": 1.0203528910417967e-06,
      "loss": 0.028,
      "step": 6198
    },
    {
      "epoch": 1.4104664391353812,
      "grad_norm": 1.0457933906877754,
      "learning_rate": 1.0202836837217597e-06,
      "loss": 0.0191,
      "step": 6199
    },
    {
      "epoch": 1.410693970420933,
      "grad_norm": 0.8908414799276027,
      "learning_rate": 1.0202144683228478e-06,
      "loss": 0.0208,
      "step": 6200
    },
    {
      "epoch": 1.4109215017064847,
      "grad_norm": 1.0085943408848066,
      "learning_rate": 1.0201452448464762e-06,
      "loss": 0.0195,
      "step": 6201
    },
    {
      "epoch": 1.4111490329920364,
      "grad_norm": 4.061894671138372,
      "learning_rate": 1.0200760132940597e-06,
      "loss": 0.0959,
      "step": 6202
    },
    {
      "epoch": 1.4113765642775882,
      "grad_norm": 0.681487143918558,
      "learning_rate": 1.0200067736670125e-06,
      "loss": 0.0199,
      "step": 6203
    },
    {
      "epoch": 1.41160409556314,
      "grad_norm": 0.9015094053704671,
      "learning_rate": 1.0199375259667505e-06,
      "loss": 0.0361,
      "step": 6204
    },
    {
      "epoch": 1.4118316268486917,
      "grad_norm": 1.3272825952215426,
      "learning_rate": 1.0198682701946889e-06,
      "loss": 0.0534,
      "step": 6205
    },
    {
      "epoch": 1.4120591581342434,
      "grad_norm": 1.132521458300331,
      "learning_rate": 1.0197990063522428e-06,
      "loss": 0.0212,
      "step": 6206
    },
    {
      "epoch": 1.4122866894197952,
      "grad_norm": 0.8792620678025737,
      "learning_rate": 1.0197297344408284e-06,
      "loss": 0.0347,
      "step": 6207
    },
    {
      "epoch": 1.412514220705347,
      "grad_norm": 1.4367192638283304,
      "learning_rate": 1.0196604544618607e-06,
      "loss": 0.0303,
      "step": 6208
    },
    {
      "epoch": 1.4127417519908987,
      "grad_norm": 1.0722086308564236,
      "learning_rate": 1.0195911664167562e-06,
      "loss": 0.0212,
      "step": 6209
    },
    {
      "epoch": 1.4129692832764504,
      "grad_norm": 2.002458559768931,
      "learning_rate": 1.0195218703069311e-06,
      "loss": 0.0213,
      "step": 6210
    },
    {
      "epoch": 1.4131968145620024,
      "grad_norm": 1.304428975916366,
      "learning_rate": 1.0194525661338014e-06,
      "loss": 0.0279,
      "step": 6211
    },
    {
      "epoch": 1.4134243458475542,
      "grad_norm": 1.0699552185731178,
      "learning_rate": 1.0193832538987838e-06,
      "loss": 0.0169,
      "step": 6212
    },
    {
      "epoch": 1.413651877133106,
      "grad_norm": 0.7360253534935643,
      "learning_rate": 1.0193139336032945e-06,
      "loss": 0.0152,
      "step": 6213
    },
    {
      "epoch": 1.4138794084186577,
      "grad_norm": 0.7178080008393654,
      "learning_rate": 1.0192446052487505e-06,
      "loss": 0.028,
      "step": 6214
    },
    {
      "epoch": 1.4141069397042094,
      "grad_norm": 1.0835136894120514,
      "learning_rate": 1.0191752688365691e-06,
      "loss": 0.045,
      "step": 6215
    },
    {
      "epoch": 1.4143344709897612,
      "grad_norm": 1.1029027025275766,
      "learning_rate": 1.019105924368167e-06,
      "loss": 0.0259,
      "step": 6216
    },
    {
      "epoch": 1.414562002275313,
      "grad_norm": 0.8923013888405551,
      "learning_rate": 1.0190365718449616e-06,
      "loss": 0.0116,
      "step": 6217
    },
    {
      "epoch": 1.4147895335608647,
      "grad_norm": 1.2539584305545126,
      "learning_rate": 1.0189672112683704e-06,
      "loss": 0.0195,
      "step": 6218
    },
    {
      "epoch": 1.4150170648464164,
      "grad_norm": 1.3274709438566543,
      "learning_rate": 1.0188978426398107e-06,
      "loss": 0.036,
      "step": 6219
    },
    {
      "epoch": 1.4152445961319682,
      "grad_norm": 1.0618852132476584,
      "learning_rate": 1.0188284659607007e-06,
      "loss": 0.0281,
      "step": 6220
    },
    {
      "epoch": 1.41547212741752,
      "grad_norm": 0.9193492374472514,
      "learning_rate": 1.018759081232458e-06,
      "loss": 0.0374,
      "step": 6221
    },
    {
      "epoch": 1.4156996587030717,
      "grad_norm": 0.5586627911962394,
      "learning_rate": 1.0186896884565005e-06,
      "loss": 0.0096,
      "step": 6222
    },
    {
      "epoch": 1.4159271899886234,
      "grad_norm": 0.6214003540618175,
      "learning_rate": 1.0186202876342473e-06,
      "loss": 0.0107,
      "step": 6223
    },
    {
      "epoch": 1.4161547212741752,
      "grad_norm": 0.9172538558057997,
      "learning_rate": 1.0185508787671162e-06,
      "loss": 0.0298,
      "step": 6224
    },
    {
      "epoch": 1.416382252559727,
      "grad_norm": 1.2046274625206315,
      "learning_rate": 1.0184814618565257e-06,
      "loss": 0.0355,
      "step": 6225
    },
    {
      "epoch": 1.4166097838452787,
      "grad_norm": 1.0097184663278695,
      "learning_rate": 1.0184120369038948e-06,
      "loss": 0.0296,
      "step": 6226
    },
    {
      "epoch": 1.4168373151308304,
      "grad_norm": 1.6454059603970002,
      "learning_rate": 1.0183426039106425e-06,
      "loss": 0.0514,
      "step": 6227
    },
    {
      "epoch": 1.4170648464163822,
      "grad_norm": 1.8322270655293305,
      "learning_rate": 1.0182731628781876e-06,
      "loss": 0.0295,
      "step": 6228
    },
    {
      "epoch": 1.417292377701934,
      "grad_norm": 0.9245095966214673,
      "learning_rate": 1.0182037138079494e-06,
      "loss": 0.0261,
      "step": 6229
    },
    {
      "epoch": 1.4175199089874857,
      "grad_norm": 1.9220130390307328,
      "learning_rate": 1.0181342567013477e-06,
      "loss": 0.0398,
      "step": 6230
    },
    {
      "epoch": 1.4177474402730375,
      "grad_norm": 0.8271648914590224,
      "learning_rate": 1.0180647915598017e-06,
      "loss": 0.0261,
      "step": 6231
    },
    {
      "epoch": 1.4179749715585892,
      "grad_norm": 1.0929333293095134,
      "learning_rate": 1.017995318384731e-06,
      "loss": 0.0413,
      "step": 6232
    },
    {
      "epoch": 1.418202502844141,
      "grad_norm": 1.0356716624792288,
      "learning_rate": 1.017925837177556e-06,
      "loss": 0.0248,
      "step": 6233
    },
    {
      "epoch": 1.4184300341296927,
      "grad_norm": 1.0477341375465965,
      "learning_rate": 1.0178563479396964e-06,
      "loss": 0.0291,
      "step": 6234
    },
    {
      "epoch": 1.4186575654152445,
      "grad_norm": 5.47070249270738,
      "learning_rate": 1.0177868506725725e-06,
      "loss": 0.0942,
      "step": 6235
    },
    {
      "epoch": 1.4188850967007964,
      "grad_norm": 1.4224582455243118,
      "learning_rate": 1.0177173453776047e-06,
      "loss": 0.0342,
      "step": 6236
    },
    {
      "epoch": 1.4191126279863482,
      "grad_norm": 1.9233599739151919,
      "learning_rate": 1.0176478320562136e-06,
      "loss": 0.056,
      "step": 6237
    },
    {
      "epoch": 1.4193401592719,
      "grad_norm": 0.9663548468064327,
      "learning_rate": 1.01757831070982e-06,
      "loss": 0.0276,
      "step": 6238
    },
    {
      "epoch": 1.4195676905574517,
      "grad_norm": 1.357744989256782,
      "learning_rate": 1.0175087813398446e-06,
      "loss": 0.0514,
      "step": 6239
    },
    {
      "epoch": 1.4197952218430034,
      "grad_norm": 0.7221163854356452,
      "learning_rate": 1.0174392439477087e-06,
      "loss": 0.0149,
      "step": 6240
    },
    {
      "epoch": 1.4200227531285552,
      "grad_norm": 1.587061634386172,
      "learning_rate": 1.0173696985348333e-06,
      "loss": 0.0596,
      "step": 6241
    },
    {
      "epoch": 1.420250284414107,
      "grad_norm": 0.7502705383802173,
      "learning_rate": 1.0173001451026396e-06,
      "loss": 0.0093,
      "step": 6242
    },
    {
      "epoch": 1.4204778156996587,
      "grad_norm": 0.7101344663498327,
      "learning_rate": 1.0172305836525498e-06,
      "loss": 0.0205,
      "step": 6243
    },
    {
      "epoch": 1.4207053469852104,
      "grad_norm": 0.754211158471996,
      "learning_rate": 1.017161014185985e-06,
      "loss": 0.0143,
      "step": 6244
    },
    {
      "epoch": 1.4209328782707622,
      "grad_norm": 0.9800518919994446,
      "learning_rate": 1.0170914367043672e-06,
      "loss": 0.0254,
      "step": 6245
    },
    {
      "epoch": 1.421160409556314,
      "grad_norm": 0.8703204775778092,
      "learning_rate": 1.0170218512091188e-06,
      "loss": 0.0154,
      "step": 6246
    },
    {
      "epoch": 1.4213879408418657,
      "grad_norm": 0.844078965636516,
      "learning_rate": 1.0169522577016614e-06,
      "loss": 0.022,
      "step": 6247
    },
    {
      "epoch": 1.4216154721274175,
      "grad_norm": 0.675422468791866,
      "learning_rate": 1.016882656183418e-06,
      "loss": 0.0082,
      "step": 6248
    },
    {
      "epoch": 1.4218430034129692,
      "grad_norm": 1.1004804368030876,
      "learning_rate": 1.0168130466558106e-06,
      "loss": 0.0125,
      "step": 6249
    },
    {
      "epoch": 1.4220705346985212,
      "grad_norm": 0.6727724492875137,
      "learning_rate": 1.0167434291202622e-06,
      "loss": 0.0126,
      "step": 6250
    },
    {
      "epoch": 1.422298065984073,
      "grad_norm": 0.9283584425159678,
      "learning_rate": 1.0166738035781954e-06,
      "loss": 0.0304,
      "step": 6251
    },
    {
      "epoch": 1.4225255972696247,
      "grad_norm": 0.9319135535277618,
      "learning_rate": 1.0166041700310334e-06,
      "loss": 0.0154,
      "step": 6252
    },
    {
      "epoch": 1.4227531285551764,
      "grad_norm": 0.9010921406225174,
      "learning_rate": 1.0165345284801995e-06,
      "loss": 0.0253,
      "step": 6253
    },
    {
      "epoch": 1.4229806598407282,
      "grad_norm": 1.2295648463971003,
      "learning_rate": 1.0164648789271167e-06,
      "loss": 0.0501,
      "step": 6254
    },
    {
      "epoch": 1.42320819112628,
      "grad_norm": 1.2697785438424416,
      "learning_rate": 1.016395221373209e-06,
      "loss": 0.046,
      "step": 6255
    },
    {
      "epoch": 1.4234357224118317,
      "grad_norm": 1.1004200253409915,
      "learning_rate": 1.0163255558198995e-06,
      "loss": 0.0248,
      "step": 6256
    },
    {
      "epoch": 1.4236632536973834,
      "grad_norm": 1.0079675426588957,
      "learning_rate": 1.0162558822686123e-06,
      "loss": 0.0323,
      "step": 6257
    },
    {
      "epoch": 1.4238907849829352,
      "grad_norm": 1.5356918423524708,
      "learning_rate": 1.0161862007207715e-06,
      "loss": 0.0522,
      "step": 6258
    },
    {
      "epoch": 1.424118316268487,
      "grad_norm": 0.753426366532939,
      "learning_rate": 1.0161165111778013e-06,
      "loss": 0.0105,
      "step": 6259
    },
    {
      "epoch": 1.4243458475540387,
      "grad_norm": 0.7423159215816296,
      "learning_rate": 1.0160468136411258e-06,
      "loss": 0.017,
      "step": 6260
    },
    {
      "epoch": 1.4245733788395905,
      "grad_norm": 1.5638797678798362,
      "learning_rate": 1.0159771081121697e-06,
      "loss": 0.041,
      "step": 6261
    },
    {
      "epoch": 1.4248009101251422,
      "grad_norm": 1.3149626939167967,
      "learning_rate": 1.0159073945923575e-06,
      "loss": 0.0332,
      "step": 6262
    },
    {
      "epoch": 1.425028441410694,
      "grad_norm": 1.573177812301038,
      "learning_rate": 1.015837673083114e-06,
      "loss": 0.0519,
      "step": 6263
    },
    {
      "epoch": 1.4252559726962457,
      "grad_norm": 1.0751260698885305,
      "learning_rate": 1.0157679435858643e-06,
      "loss": 0.0305,
      "step": 6264
    },
    {
      "epoch": 1.4254835039817975,
      "grad_norm": 0.9230599364770781,
      "learning_rate": 1.0156982061020335e-06,
      "loss": 0.0179,
      "step": 6265
    },
    {
      "epoch": 1.4257110352673492,
      "grad_norm": 1.4181072269182498,
      "learning_rate": 1.0156284606330468e-06,
      "loss": 0.0491,
      "step": 6266
    },
    {
      "epoch": 1.425938566552901,
      "grad_norm": 0.9090067280998548,
      "learning_rate": 1.0155587071803298e-06,
      "loss": 0.0232,
      "step": 6267
    },
    {
      "epoch": 1.4261660978384527,
      "grad_norm": 1.383666121761416,
      "learning_rate": 1.0154889457453082e-06,
      "loss": 0.0241,
      "step": 6268
    },
    {
      "epoch": 1.4263936291240045,
      "grad_norm": 0.7621859675263093,
      "learning_rate": 1.0154191763294077e-06,
      "loss": 0.0235,
      "step": 6269
    },
    {
      "epoch": 1.4266211604095562,
      "grad_norm": 1.2108840735457889,
      "learning_rate": 1.015349398934054e-06,
      "loss": 0.0401,
      "step": 6270
    },
    {
      "epoch": 1.426848691695108,
      "grad_norm": 0.7260177397917599,
      "learning_rate": 1.0152796135606739e-06,
      "loss": 0.011,
      "step": 6271
    },
    {
      "epoch": 1.4270762229806597,
      "grad_norm": 0.7749106615077918,
      "learning_rate": 1.015209820210693e-06,
      "loss": 0.0223,
      "step": 6272
    },
    {
      "epoch": 1.4273037542662115,
      "grad_norm": 1.0764688037172783,
      "learning_rate": 1.015140018885538e-06,
      "loss": 0.0347,
      "step": 6273
    },
    {
      "epoch": 1.4275312855517632,
      "grad_norm": 0.7315552904513073,
      "learning_rate": 1.0150702095866354e-06,
      "loss": 0.0146,
      "step": 6274
    },
    {
      "epoch": 1.4277588168373152,
      "grad_norm": 1.4725847050231393,
      "learning_rate": 1.0150003923154124e-06,
      "loss": 0.0286,
      "step": 6275
    },
    {
      "epoch": 1.427986348122867,
      "grad_norm": 0.8172080214482219,
      "learning_rate": 1.0149305670732953e-06,
      "loss": 0.0172,
      "step": 6276
    },
    {
      "epoch": 1.4282138794084187,
      "grad_norm": 0.9065092467190615,
      "learning_rate": 1.0148607338617118e-06,
      "loss": 0.026,
      "step": 6277
    },
    {
      "epoch": 1.4284414106939705,
      "grad_norm": 1.055574219901434,
      "learning_rate": 1.0147908926820887e-06,
      "loss": 0.0325,
      "step": 6278
    },
    {
      "epoch": 1.4286689419795222,
      "grad_norm": 1.3895263245690272,
      "learning_rate": 1.014721043535854e-06,
      "loss": 0.0309,
      "step": 6279
    },
    {
      "epoch": 1.428896473265074,
      "grad_norm": 1.3548353082702096,
      "learning_rate": 1.0146511864244344e-06,
      "loss": 0.02,
      "step": 6280
    },
    {
      "epoch": 1.4291240045506257,
      "grad_norm": 0.996001028629038,
      "learning_rate": 1.0145813213492587e-06,
      "loss": 0.0206,
      "step": 6281
    },
    {
      "epoch": 1.4293515358361775,
      "grad_norm": 1.5083691859247366,
      "learning_rate": 1.0145114483117539e-06,
      "loss": 0.0256,
      "step": 6282
    },
    {
      "epoch": 1.4295790671217292,
      "grad_norm": 0.8549655620512744,
      "learning_rate": 1.0144415673133485e-06,
      "loss": 0.0239,
      "step": 6283
    },
    {
      "epoch": 1.429806598407281,
      "grad_norm": 1.3514935766543403,
      "learning_rate": 1.0143716783554709e-06,
      "loss": 0.021,
      "step": 6284
    },
    {
      "epoch": 1.4300341296928327,
      "grad_norm": 1.1741487580376555,
      "learning_rate": 1.0143017814395489e-06,
      "loss": 0.0162,
      "step": 6285
    },
    {
      "epoch": 1.4302616609783845,
      "grad_norm": 0.9905348741277206,
      "learning_rate": 1.0142318765670117e-06,
      "loss": 0.0204,
      "step": 6286
    },
    {
      "epoch": 1.4304891922639362,
      "grad_norm": 1.5929745991128101,
      "learning_rate": 1.0141619637392878e-06,
      "loss": 0.0469,
      "step": 6287
    },
    {
      "epoch": 1.430716723549488,
      "grad_norm": 3.2298820198178886,
      "learning_rate": 1.0140920429578061e-06,
      "loss": 0.1133,
      "step": 6288
    },
    {
      "epoch": 1.43094425483504,
      "grad_norm": 1.4247664034317506,
      "learning_rate": 1.0140221142239957e-06,
      "loss": 0.028,
      "step": 6289
    },
    {
      "epoch": 1.4311717861205917,
      "grad_norm": 0.7515197976704004,
      "learning_rate": 1.0139521775392856e-06,
      "loss": 0.0304,
      "step": 6290
    },
    {
      "epoch": 1.4313993174061435,
      "grad_norm": 0.9354685764612675,
      "learning_rate": 1.0138822329051052e-06,
      "loss": 0.0182,
      "step": 6291
    },
    {
      "epoch": 1.4316268486916952,
      "grad_norm": 1.0152548776003998,
      "learning_rate": 1.0138122803228843e-06,
      "loss": 0.0161,
      "step": 6292
    },
    {
      "epoch": 1.431854379977247,
      "grad_norm": 1.2022208108630814,
      "learning_rate": 1.0137423197940527e-06,
      "loss": 0.0154,
      "step": 6293
    },
    {
      "epoch": 1.4320819112627987,
      "grad_norm": 1.095549117230315,
      "learning_rate": 1.0136723513200396e-06,
      "loss": 0.0291,
      "step": 6294
    },
    {
      "epoch": 1.4323094425483505,
      "grad_norm": 1.5045560124052015,
      "learning_rate": 1.0136023749022759e-06,
      "loss": 0.0491,
      "step": 6295
    },
    {
      "epoch": 1.4325369738339022,
      "grad_norm": 1.6455974388894132,
      "learning_rate": 1.013532390542191e-06,
      "loss": 0.0169,
      "step": 6296
    },
    {
      "epoch": 1.432764505119454,
      "grad_norm": 1.3646221962527803,
      "learning_rate": 1.0134623982412156e-06,
      "loss": 0.0343,
      "step": 6297
    },
    {
      "epoch": 1.4329920364050057,
      "grad_norm": 1.7956652104090614,
      "learning_rate": 1.0133923980007804e-06,
      "loss": 0.0231,
      "step": 6298
    },
    {
      "epoch": 1.4332195676905575,
      "grad_norm": 0.9926711065918172,
      "learning_rate": 1.013322389822316e-06,
      "loss": 0.0149,
      "step": 6299
    },
    {
      "epoch": 1.4334470989761092,
      "grad_norm": 1.3656153995620692,
      "learning_rate": 1.0132523737072528e-06,
      "loss": 0.0342,
      "step": 6300
    },
    {
      "epoch": 1.433674630261661,
      "grad_norm": 0.7242163481709295,
      "learning_rate": 1.0131823496570222e-06,
      "loss": 0.0118,
      "step": 6301
    },
    {
      "epoch": 1.4339021615472127,
      "grad_norm": 0.8670547748875898,
      "learning_rate": 1.0131123176730555e-06,
      "loss": 0.0221,
      "step": 6302
    },
    {
      "epoch": 1.4341296928327645,
      "grad_norm": 1.5995659761831407,
      "learning_rate": 1.0130422777567837e-06,
      "loss": 0.0308,
      "step": 6303
    },
    {
      "epoch": 1.4343572241183162,
      "grad_norm": 1.0654736743914723,
      "learning_rate": 1.0129722299096385e-06,
      "loss": 0.0221,
      "step": 6304
    },
    {
      "epoch": 1.434584755403868,
      "grad_norm": 0.7322779348375932,
      "learning_rate": 1.0129021741330514e-06,
      "loss": 0.017,
      "step": 6305
    },
    {
      "epoch": 1.4348122866894197,
      "grad_norm": 1.371952571606394,
      "learning_rate": 1.0128321104284541e-06,
      "loss": 0.0339,
      "step": 6306
    },
    {
      "epoch": 1.4350398179749715,
      "grad_norm": 1.6200877785866907,
      "learning_rate": 1.0127620387972789e-06,
      "loss": 0.0389,
      "step": 6307
    },
    {
      "epoch": 1.4352673492605232,
      "grad_norm": 1.3440385321706123,
      "learning_rate": 1.0126919592409578e-06,
      "loss": 0.0194,
      "step": 6308
    },
    {
      "epoch": 1.435494880546075,
      "grad_norm": 1.5916470330075667,
      "learning_rate": 1.012621871760923e-06,
      "loss": 0.0454,
      "step": 6309
    },
    {
      "epoch": 1.4357224118316267,
      "grad_norm": 0.9723442385422495,
      "learning_rate": 1.012551776358607e-06,
      "loss": 0.0207,
      "step": 6310
    },
    {
      "epoch": 1.4359499431171785,
      "grad_norm": 1.327551352285537,
      "learning_rate": 1.0124816730354428e-06,
      "loss": 0.0317,
      "step": 6311
    },
    {
      "epoch": 1.4361774744027302,
      "grad_norm": 1.1035989928859236,
      "learning_rate": 1.0124115617928626e-06,
      "loss": 0.0205,
      "step": 6312
    },
    {
      "epoch": 1.4364050056882822,
      "grad_norm": 0.752935991153831,
      "learning_rate": 1.0123414426322995e-06,
      "loss": 0.0168,
      "step": 6313
    },
    {
      "epoch": 1.436632536973834,
      "grad_norm": 1.5986259540246315,
      "learning_rate": 1.0122713155551867e-06,
      "loss": 0.0655,
      "step": 6314
    },
    {
      "epoch": 1.4368600682593857,
      "grad_norm": 1.251656443778033,
      "learning_rate": 1.0122011805629574e-06,
      "loss": 0.0381,
      "step": 6315
    },
    {
      "epoch": 1.4370875995449375,
      "grad_norm": 1.0440643194056838,
      "learning_rate": 1.0121310376570454e-06,
      "loss": 0.024,
      "step": 6316
    },
    {
      "epoch": 1.4373151308304892,
      "grad_norm": 2.2417861503091276,
      "learning_rate": 1.0120608868388837e-06,
      "loss": 0.045,
      "step": 6317
    },
    {
      "epoch": 1.437542662116041,
      "grad_norm": 1.1700571736014582,
      "learning_rate": 1.0119907281099066e-06,
      "loss": 0.0401,
      "step": 6318
    },
    {
      "epoch": 1.4377701934015927,
      "grad_norm": 0.9069348186314079,
      "learning_rate": 1.0119205614715476e-06,
      "loss": 0.0176,
      "step": 6319
    },
    {
      "epoch": 1.4379977246871445,
      "grad_norm": 1.9355113829065758,
      "learning_rate": 1.011850386925241e-06,
      "loss": 0.0294,
      "step": 6320
    },
    {
      "epoch": 1.4382252559726962,
      "grad_norm": 1.4052995825330266,
      "learning_rate": 1.011780204472421e-06,
      "loss": 0.0155,
      "step": 6321
    },
    {
      "epoch": 1.438452787258248,
      "grad_norm": 1.1704734179841323,
      "learning_rate": 1.0117100141145217e-06,
      "loss": 0.0319,
      "step": 6322
    },
    {
      "epoch": 1.4386803185437997,
      "grad_norm": 1.3977711378669946,
      "learning_rate": 1.0116398158529782e-06,
      "loss": 0.0358,
      "step": 6323
    },
    {
      "epoch": 1.4389078498293515,
      "grad_norm": 1.0414067986088198,
      "learning_rate": 1.0115696096892247e-06,
      "loss": 0.0293,
      "step": 6324
    },
    {
      "epoch": 1.4391353811149032,
      "grad_norm": 1.081914953392961,
      "learning_rate": 1.0114993956246968e-06,
      "loss": 0.0227,
      "step": 6325
    },
    {
      "epoch": 1.439362912400455,
      "grad_norm": 0.9696704441650997,
      "learning_rate": 1.0114291736608289e-06,
      "loss": 0.0269,
      "step": 6326
    },
    {
      "epoch": 1.4395904436860067,
      "grad_norm": 1.0191772217507478,
      "learning_rate": 1.0113589437990562e-06,
      "loss": 0.0266,
      "step": 6327
    },
    {
      "epoch": 1.4398179749715587,
      "grad_norm": 0.9746405255609321,
      "learning_rate": 1.0112887060408144e-06,
      "loss": 0.0301,
      "step": 6328
    },
    {
      "epoch": 1.4400455062571105,
      "grad_norm": 0.871919751178116,
      "learning_rate": 1.0112184603875387e-06,
      "loss": 0.0153,
      "step": 6329
    },
    {
      "epoch": 1.4402730375426622,
      "grad_norm": 1.052877314511932,
      "learning_rate": 1.0111482068406655e-06,
      "loss": 0.0118,
      "step": 6330
    },
    {
      "epoch": 1.440500568828214,
      "grad_norm": 1.418051417100814,
      "learning_rate": 1.0110779454016299e-06,
      "loss": 0.0269,
      "step": 6331
    },
    {
      "epoch": 1.4407281001137657,
      "grad_norm": 0.7054078297819081,
      "learning_rate": 1.011007676071868e-06,
      "loss": 0.0128,
      "step": 6332
    },
    {
      "epoch": 1.4409556313993175,
      "grad_norm": 0.9151449048408781,
      "learning_rate": 1.0109373988528161e-06,
      "loss": 0.0188,
      "step": 6333
    },
    {
      "epoch": 1.4411831626848692,
      "grad_norm": 0.6914366508726025,
      "learning_rate": 1.010867113745911e-06,
      "loss": 0.0152,
      "step": 6334
    },
    {
      "epoch": 1.441410693970421,
      "grad_norm": 0.8898176154301342,
      "learning_rate": 1.0107968207525884e-06,
      "loss": 0.0178,
      "step": 6335
    },
    {
      "epoch": 1.4416382252559727,
      "grad_norm": 1.24440607526392,
      "learning_rate": 1.0107265198742855e-06,
      "loss": 0.0426,
      "step": 6336
    },
    {
      "epoch": 1.4418657565415245,
      "grad_norm": 1.195011653706321,
      "learning_rate": 1.010656211112439e-06,
      "loss": 0.0279,
      "step": 6337
    },
    {
      "epoch": 1.4420932878270762,
      "grad_norm": 1.4854065146041637,
      "learning_rate": 1.0105858944684856e-06,
      "loss": 0.023,
      "step": 6338
    },
    {
      "epoch": 1.442320819112628,
      "grad_norm": 0.5511163443762787,
      "learning_rate": 1.010515569943863e-06,
      "loss": 0.0116,
      "step": 6339
    },
    {
      "epoch": 1.4425483503981797,
      "grad_norm": 1.2043446207357231,
      "learning_rate": 1.0104452375400078e-06,
      "loss": 0.0202,
      "step": 6340
    },
    {
      "epoch": 1.4427758816837315,
      "grad_norm": 1.3000326710853358,
      "learning_rate": 1.0103748972583582e-06,
      "loss": 0.0465,
      "step": 6341
    },
    {
      "epoch": 1.4430034129692833,
      "grad_norm": 1.3675162595675465,
      "learning_rate": 1.0103045491003514e-06,
      "loss": 0.0323,
      "step": 6342
    },
    {
      "epoch": 1.443230944254835,
      "grad_norm": 0.7420030229566207,
      "learning_rate": 1.0102341930674252e-06,
      "loss": 0.0237,
      "step": 6343
    },
    {
      "epoch": 1.4434584755403868,
      "grad_norm": 1.122028425409587,
      "learning_rate": 1.0101638291610176e-06,
      "loss": 0.0437,
      "step": 6344
    },
    {
      "epoch": 1.4436860068259385,
      "grad_norm": 2.271929983664725,
      "learning_rate": 1.0100934573825668e-06,
      "loss": 0.0384,
      "step": 6345
    },
    {
      "epoch": 1.4439135381114903,
      "grad_norm": 2.0024195181198214,
      "learning_rate": 1.010023077733511e-06,
      "loss": 0.0459,
      "step": 6346
    },
    {
      "epoch": 1.444141069397042,
      "grad_norm": 1.2513837267760826,
      "learning_rate": 1.0099526902152886e-06,
      "loss": 0.0245,
      "step": 6347
    },
    {
      "epoch": 1.4443686006825938,
      "grad_norm": 0.8662028583353463,
      "learning_rate": 1.0098822948293382e-06,
      "loss": 0.0222,
      "step": 6348
    },
    {
      "epoch": 1.4445961319681455,
      "grad_norm": 1.3336701254562313,
      "learning_rate": 1.0098118915770985e-06,
      "loss": 0.0467,
      "step": 6349
    },
    {
      "epoch": 1.4448236632536973,
      "grad_norm": 1.301840631614471,
      "learning_rate": 1.0097414804600087e-06,
      "loss": 0.036,
      "step": 6350
    },
    {
      "epoch": 1.445051194539249,
      "grad_norm": 0.7933102665199757,
      "learning_rate": 1.0096710614795077e-06,
      "loss": 0.0134,
      "step": 6351
    },
    {
      "epoch": 1.445278725824801,
      "grad_norm": 1.6967569868348495,
      "learning_rate": 1.0096006346370344e-06,
      "loss": 0.0453,
      "step": 6352
    },
    {
      "epoch": 1.4455062571103527,
      "grad_norm": 1.9726304694822052,
      "learning_rate": 1.0095301999340285e-06,
      "loss": 0.0373,
      "step": 6353
    },
    {
      "epoch": 1.4457337883959045,
      "grad_norm": 1.118511972338071,
      "learning_rate": 1.0094597573719297e-06,
      "loss": 0.0306,
      "step": 6354
    },
    {
      "epoch": 1.4459613196814562,
      "grad_norm": 1.3932750265447595,
      "learning_rate": 1.0093893069521777e-06,
      "loss": 0.0254,
      "step": 6355
    },
    {
      "epoch": 1.446188850967008,
      "grad_norm": 0.8939501658395617,
      "learning_rate": 1.009318848676212e-06,
      "loss": 0.0154,
      "step": 6356
    },
    {
      "epoch": 1.4464163822525598,
      "grad_norm": 0.8973575337225248,
      "learning_rate": 1.0092483825454729e-06,
      "loss": 0.0333,
      "step": 6357
    },
    {
      "epoch": 1.4466439135381115,
      "grad_norm": 0.8404430334263568,
      "learning_rate": 1.0091779085614006e-06,
      "loss": 0.0114,
      "step": 6358
    },
    {
      "epoch": 1.4468714448236633,
      "grad_norm": 0.554800530533077,
      "learning_rate": 1.0091074267254355e-06,
      "loss": 0.0072,
      "step": 6359
    },
    {
      "epoch": 1.447098976109215,
      "grad_norm": 1.3794649246980744,
      "learning_rate": 1.009036937039018e-06,
      "loss": 0.0345,
      "step": 6360
    },
    {
      "epoch": 1.4473265073947668,
      "grad_norm": 0.9682097442548065,
      "learning_rate": 1.008966439503589e-06,
      "loss": 0.0387,
      "step": 6361
    },
    {
      "epoch": 1.4475540386803185,
      "grad_norm": 0.9402002903723223,
      "learning_rate": 1.008895934120589e-06,
      "loss": 0.0147,
      "step": 6362
    },
    {
      "epoch": 1.4477815699658703,
      "grad_norm": 1.1075685325307638,
      "learning_rate": 1.0088254208914593e-06,
      "loss": 0.0223,
      "step": 6363
    },
    {
      "epoch": 1.448009101251422,
      "grad_norm": 1.2528562350335635,
      "learning_rate": 1.0087548998176409e-06,
      "loss": 0.0224,
      "step": 6364
    },
    {
      "epoch": 1.4482366325369738,
      "grad_norm": 1.0342471484913283,
      "learning_rate": 1.008684370900575e-06,
      "loss": 0.0391,
      "step": 6365
    },
    {
      "epoch": 1.4484641638225255,
      "grad_norm": 0.541143297382043,
      "learning_rate": 1.0086138341417035e-06,
      "loss": 0.01,
      "step": 6366
    },
    {
      "epoch": 1.4486916951080775,
      "grad_norm": 1.6811565885259934,
      "learning_rate": 1.0085432895424674e-06,
      "loss": 0.0642,
      "step": 6367
    },
    {
      "epoch": 1.4489192263936292,
      "grad_norm": 0.7476243414143443,
      "learning_rate": 1.0084727371043094e-06,
      "loss": 0.0168,
      "step": 6368
    },
    {
      "epoch": 1.449146757679181,
      "grad_norm": 1.1464144466295128,
      "learning_rate": 1.0084021768286706e-06,
      "loss": 0.0245,
      "step": 6369
    },
    {
      "epoch": 1.4493742889647327,
      "grad_norm": 1.359249500318607,
      "learning_rate": 1.0083316087169935e-06,
      "loss": 0.0627,
      "step": 6370
    },
    {
      "epoch": 1.4496018202502845,
      "grad_norm": 2.075043421424735,
      "learning_rate": 1.0082610327707204e-06,
      "loss": 0.0234,
      "step": 6371
    },
    {
      "epoch": 1.4498293515358363,
      "grad_norm": 1.7107015538291868,
      "learning_rate": 1.0081904489912937e-06,
      "loss": 0.0327,
      "step": 6372
    },
    {
      "epoch": 1.450056882821388,
      "grad_norm": 1.1424090301708094,
      "learning_rate": 1.008119857380156e-06,
      "loss": 0.0227,
      "step": 6373
    },
    {
      "epoch": 1.4502844141069398,
      "grad_norm": 1.8234753567513349,
      "learning_rate": 1.0080492579387503e-06,
      "loss": 0.0455,
      "step": 6374
    },
    {
      "epoch": 1.4505119453924915,
      "grad_norm": 0.7117928424230715,
      "learning_rate": 1.007978650668519e-06,
      "loss": 0.0081,
      "step": 6375
    },
    {
      "epoch": 1.4507394766780433,
      "grad_norm": 0.9207913101740711,
      "learning_rate": 1.0079080355709057e-06,
      "loss": 0.0194,
      "step": 6376
    },
    {
      "epoch": 1.450967007963595,
      "grad_norm": 1.8055497591635554,
      "learning_rate": 1.007837412647353e-06,
      "loss": 0.0275,
      "step": 6377
    },
    {
      "epoch": 1.4511945392491468,
      "grad_norm": 1.0900059173773669,
      "learning_rate": 1.007766781899305e-06,
      "loss": 0.0196,
      "step": 6378
    },
    {
      "epoch": 1.4514220705346985,
      "grad_norm": 0.7549560575688554,
      "learning_rate": 1.0076961433282051e-06,
      "loss": 0.0102,
      "step": 6379
    },
    {
      "epoch": 1.4516496018202503,
      "grad_norm": 1.0184772578883805,
      "learning_rate": 1.0076254969354967e-06,
      "loss": 0.0188,
      "step": 6380
    },
    {
      "epoch": 1.451877133105802,
      "grad_norm": 0.941415605985606,
      "learning_rate": 1.0075548427226241e-06,
      "loss": 0.0349,
      "step": 6381
    },
    {
      "epoch": 1.4521046643913538,
      "grad_norm": 0.8934463803567299,
      "learning_rate": 1.007484180691031e-06,
      "loss": 0.0147,
      "step": 6382
    },
    {
      "epoch": 1.4523321956769055,
      "grad_norm": 1.6447864640806058,
      "learning_rate": 1.0074135108421616e-06,
      "loss": 0.0414,
      "step": 6383
    },
    {
      "epoch": 1.4525597269624573,
      "grad_norm": 0.7619214840074895,
      "learning_rate": 1.0073428331774605e-06,
      "loss": 0.0194,
      "step": 6384
    },
    {
      "epoch": 1.452787258248009,
      "grad_norm": 1.042967290966513,
      "learning_rate": 1.007272147698372e-06,
      "loss": 0.0216,
      "step": 6385
    },
    {
      "epoch": 1.4530147895335608,
      "grad_norm": 0.7609050337711548,
      "learning_rate": 1.0072014544063412e-06,
      "loss": 0.0108,
      "step": 6386
    },
    {
      "epoch": 1.4532423208191125,
      "grad_norm": 0.8698525287916742,
      "learning_rate": 1.0071307533028125e-06,
      "loss": 0.029,
      "step": 6387
    },
    {
      "epoch": 1.4534698521046643,
      "grad_norm": 0.8501257290692393,
      "learning_rate": 1.007060044389231e-06,
      "loss": 0.0345,
      "step": 6388
    },
    {
      "epoch": 1.453697383390216,
      "grad_norm": 0.9391000541726464,
      "learning_rate": 1.006989327667042e-06,
      "loss": 0.0256,
      "step": 6389
    },
    {
      "epoch": 1.4539249146757678,
      "grad_norm": 1.8647278292453908,
      "learning_rate": 1.0069186031376906e-06,
      "loss": 0.0392,
      "step": 6390
    },
    {
      "epoch": 1.4541524459613198,
      "grad_norm": 0.7819994697112215,
      "learning_rate": 1.0068478708026224e-06,
      "loss": 0.0122,
      "step": 6391
    },
    {
      "epoch": 1.4543799772468715,
      "grad_norm": 0.585110905309953,
      "learning_rate": 1.0067771306632832e-06,
      "loss": 0.0126,
      "step": 6392
    },
    {
      "epoch": 1.4546075085324233,
      "grad_norm": 0.9364890195145283,
      "learning_rate": 1.0067063827211184e-06,
      "loss": 0.0136,
      "step": 6393
    },
    {
      "epoch": 1.454835039817975,
      "grad_norm": 0.6498349687781143,
      "learning_rate": 1.0066356269775744e-06,
      "loss": 0.0103,
      "step": 6394
    },
    {
      "epoch": 1.4550625711035268,
      "grad_norm": 0.869864688376475,
      "learning_rate": 1.0065648634340971e-06,
      "loss": 0.0219,
      "step": 6395
    },
    {
      "epoch": 1.4552901023890785,
      "grad_norm": 1.1931980839721092,
      "learning_rate": 1.0064940920921328e-06,
      "loss": 0.0287,
      "step": 6396
    },
    {
      "epoch": 1.4555176336746303,
      "grad_norm": 1.3209455372197918,
      "learning_rate": 1.006423312953128e-06,
      "loss": 0.0482,
      "step": 6397
    },
    {
      "epoch": 1.455745164960182,
      "grad_norm": 1.5390872099106532,
      "learning_rate": 1.0063525260185288e-06,
      "loss": 0.0114,
      "step": 6398
    },
    {
      "epoch": 1.4559726962457338,
      "grad_norm": 0.7920775851368451,
      "learning_rate": 1.0062817312897826e-06,
      "loss": 0.0124,
      "step": 6399
    },
    {
      "epoch": 1.4562002275312855,
      "grad_norm": 0.9021573629870503,
      "learning_rate": 1.0062109287683364e-06,
      "loss": 0.0191,
      "step": 6400
    },
    {
      "epoch": 1.4564277588168373,
      "grad_norm": 0.9639937227508991,
      "learning_rate": 1.0061401184556366e-06,
      "loss": 0.0165,
      "step": 6401
    },
    {
      "epoch": 1.456655290102389,
      "grad_norm": 1.3012355962020263,
      "learning_rate": 1.006069300353131e-06,
      "loss": 0.0381,
      "step": 6402
    },
    {
      "epoch": 1.4568828213879408,
      "grad_norm": 1.421780962954957,
      "learning_rate": 1.005998474462267e-06,
      "loss": 0.0278,
      "step": 6403
    },
    {
      "epoch": 1.4571103526734925,
      "grad_norm": 0.7020431371817161,
      "learning_rate": 1.0059276407844915e-06,
      "loss": 0.0136,
      "step": 6404
    },
    {
      "epoch": 1.4573378839590443,
      "grad_norm": 0.610596243564309,
      "learning_rate": 1.0058567993212528e-06,
      "loss": 0.0105,
      "step": 6405
    },
    {
      "epoch": 1.4575654152445963,
      "grad_norm": 1.6868440616932545,
      "learning_rate": 1.005785950073999e-06,
      "loss": 0.0662,
      "step": 6406
    },
    {
      "epoch": 1.457792946530148,
      "grad_norm": 1.3076997172219658,
      "learning_rate": 1.0057150930441772e-06,
      "loss": 0.0494,
      "step": 6407
    },
    {
      "epoch": 1.4580204778156998,
      "grad_norm": 1.8180694190438493,
      "learning_rate": 1.0056442282332365e-06,
      "loss": 0.0256,
      "step": 6408
    },
    {
      "epoch": 1.4582480091012515,
      "grad_norm": 0.7591625918315809,
      "learning_rate": 1.0055733556426248e-06,
      "loss": 0.0251,
      "step": 6409
    },
    {
      "epoch": 1.4584755403868033,
      "grad_norm": 0.8738294046187783,
      "learning_rate": 1.0055024752737906e-06,
      "loss": 0.0264,
      "step": 6410
    },
    {
      "epoch": 1.458703071672355,
      "grad_norm": 0.9064382129589225,
      "learning_rate": 1.0054315871281828e-06,
      "loss": 0.0214,
      "step": 6411
    },
    {
      "epoch": 1.4589306029579068,
      "grad_norm": 1.6826022489740848,
      "learning_rate": 1.00536069120725e-06,
      "loss": 0.0397,
      "step": 6412
    },
    {
      "epoch": 1.4591581342434585,
      "grad_norm": 0.8081842544846835,
      "learning_rate": 1.0052897875124412e-06,
      "loss": 0.0179,
      "step": 6413
    },
    {
      "epoch": 1.4593856655290103,
      "grad_norm": 1.0489435189018754,
      "learning_rate": 1.005218876045206e-06,
      "loss": 0.0258,
      "step": 6414
    },
    {
      "epoch": 1.459613196814562,
      "grad_norm": 0.9107275649992217,
      "learning_rate": 1.0051479568069927e-06,
      "loss": 0.0159,
      "step": 6415
    },
    {
      "epoch": 1.4598407281001138,
      "grad_norm": 0.7882760746921257,
      "learning_rate": 1.0050770297992517e-06,
      "loss": 0.0188,
      "step": 6416
    },
    {
      "epoch": 1.4600682593856655,
      "grad_norm": 0.494600424451502,
      "learning_rate": 1.0050060950234324e-06,
      "loss": 0.0091,
      "step": 6417
    },
    {
      "epoch": 1.4602957906712173,
      "grad_norm": 0.8887210333396641,
      "learning_rate": 1.0049351524809842e-06,
      "loss": 0.0234,
      "step": 6418
    },
    {
      "epoch": 1.460523321956769,
      "grad_norm": 0.8818727125075025,
      "learning_rate": 1.0048642021733576e-06,
      "loss": 0.0203,
      "step": 6419
    },
    {
      "epoch": 1.4607508532423208,
      "grad_norm": 0.8286346226624387,
      "learning_rate": 1.0047932441020022e-06,
      "loss": 0.0141,
      "step": 6420
    },
    {
      "epoch": 1.4609783845278725,
      "grad_norm": 1.0017852943824,
      "learning_rate": 1.0047222782683686e-06,
      "loss": 0.0184,
      "step": 6421
    },
    {
      "epoch": 1.4612059158134243,
      "grad_norm": 1.0992796278491528,
      "learning_rate": 1.0046513046739069e-06,
      "loss": 0.0177,
      "step": 6422
    },
    {
      "epoch": 1.461433447098976,
      "grad_norm": 1.0966388790320833,
      "learning_rate": 1.0045803233200679e-06,
      "loss": 0.0229,
      "step": 6423
    },
    {
      "epoch": 1.4616609783845278,
      "grad_norm": 1.1017149602413754,
      "learning_rate": 1.0045093342083022e-06,
      "loss": 0.0191,
      "step": 6424
    },
    {
      "epoch": 1.4618885096700796,
      "grad_norm": 1.6564794597904529,
      "learning_rate": 1.0044383373400608e-06,
      "loss": 0.0371,
      "step": 6425
    },
    {
      "epoch": 1.4621160409556313,
      "grad_norm": 2.1788289964022813,
      "learning_rate": 1.0043673327167946e-06,
      "loss": 0.0461,
      "step": 6426
    },
    {
      "epoch": 1.462343572241183,
      "grad_norm": 0.9615176745408612,
      "learning_rate": 1.004296320339955e-06,
      "loss": 0.0163,
      "step": 6427
    },
    {
      "epoch": 1.4625711035267348,
      "grad_norm": 0.9958220543488695,
      "learning_rate": 1.0042253002109933e-06,
      "loss": 0.0214,
      "step": 6428
    },
    {
      "epoch": 1.4627986348122866,
      "grad_norm": 0.78557582039727,
      "learning_rate": 1.004154272331361e-06,
      "loss": 0.0136,
      "step": 6429
    },
    {
      "epoch": 1.4630261660978385,
      "grad_norm": 1.0936828441296087,
      "learning_rate": 1.0040832367025097e-06,
      "loss": 0.0219,
      "step": 6430
    },
    {
      "epoch": 1.4632536973833903,
      "grad_norm": 0.7390343812119207,
      "learning_rate": 1.0040121933258912e-06,
      "loss": 0.01,
      "step": 6431
    },
    {
      "epoch": 1.463481228668942,
      "grad_norm": 1.4582085989040046,
      "learning_rate": 1.0039411422029578e-06,
      "loss": 0.0313,
      "step": 6432
    },
    {
      "epoch": 1.4637087599544938,
      "grad_norm": 0.9523062629655219,
      "learning_rate": 1.0038700833351612e-06,
      "loss": 0.0245,
      "step": 6433
    },
    {
      "epoch": 1.4639362912400455,
      "grad_norm": 1.2205425053268553,
      "learning_rate": 1.0037990167239542e-06,
      "loss": 0.0243,
      "step": 6434
    },
    {
      "epoch": 1.4641638225255973,
      "grad_norm": 1.3769758384207615,
      "learning_rate": 1.0037279423707888e-06,
      "loss": 0.0284,
      "step": 6435
    },
    {
      "epoch": 1.464391353811149,
      "grad_norm": 1.6348768046013533,
      "learning_rate": 1.0036568602771179e-06,
      "loss": 0.0674,
      "step": 6436
    },
    {
      "epoch": 1.4646188850967008,
      "grad_norm": 0.7786100751808739,
      "learning_rate": 1.0035857704443944e-06,
      "loss": 0.0112,
      "step": 6437
    },
    {
      "epoch": 1.4648464163822525,
      "grad_norm": 1.2999313140929,
      "learning_rate": 1.0035146728740712e-06,
      "loss": 0.0364,
      "step": 6438
    },
    {
      "epoch": 1.4650739476678043,
      "grad_norm": 1.6927396967754325,
      "learning_rate": 1.0034435675676012e-06,
      "loss": 0.027,
      "step": 6439
    },
    {
      "epoch": 1.465301478953356,
      "grad_norm": 0.7196939694803933,
      "learning_rate": 1.0033724545264378e-06,
      "loss": 0.0061,
      "step": 6440
    },
    {
      "epoch": 1.4655290102389078,
      "grad_norm": 0.8716526278131789,
      "learning_rate": 1.0033013337520342e-06,
      "loss": 0.0146,
      "step": 6441
    },
    {
      "epoch": 1.4657565415244596,
      "grad_norm": 1.048978533991283,
      "learning_rate": 1.0032302052458443e-06,
      "loss": 0.0315,
      "step": 6442
    },
    {
      "epoch": 1.4659840728100113,
      "grad_norm": 1.0777036173381194,
      "learning_rate": 1.0031590690093214e-06,
      "loss": 0.0334,
      "step": 6443
    },
    {
      "epoch": 1.466211604095563,
      "grad_norm": 0.4736534501721457,
      "learning_rate": 1.0030879250439203e-06,
      "loss": 0.0114,
      "step": 6444
    },
    {
      "epoch": 1.466439135381115,
      "grad_norm": 0.6021310354060467,
      "learning_rate": 1.003016773351094e-06,
      "loss": 0.0097,
      "step": 6445
    },
    {
      "epoch": 1.4666666666666668,
      "grad_norm": 0.8649430411936431,
      "learning_rate": 1.0029456139322973e-06,
      "loss": 0.0162,
      "step": 6446
    },
    {
      "epoch": 1.4668941979522185,
      "grad_norm": 1.543095421141427,
      "learning_rate": 1.0028744467889845e-06,
      "loss": 0.0291,
      "step": 6447
    },
    {
      "epoch": 1.4671217292377703,
      "grad_norm": 1.5988742387248307,
      "learning_rate": 1.0028032719226097e-06,
      "loss": 0.0373,
      "step": 6448
    },
    {
      "epoch": 1.467349260523322,
      "grad_norm": 2.0643928637015563,
      "learning_rate": 1.0027320893346284e-06,
      "loss": 0.0459,
      "step": 6449
    },
    {
      "epoch": 1.4675767918088738,
      "grad_norm": 2.220711557926011,
      "learning_rate": 1.0026608990264946e-06,
      "loss": 0.0321,
      "step": 6450
    },
    {
      "epoch": 1.4678043230944255,
      "grad_norm": 0.9445356318647576,
      "learning_rate": 1.0025897009996639e-06,
      "loss": 0.0226,
      "step": 6451
    },
    {
      "epoch": 1.4680318543799773,
      "grad_norm": 0.6330305963301595,
      "learning_rate": 1.0025184952555911e-06,
      "loss": 0.0205,
      "step": 6452
    },
    {
      "epoch": 1.468259385665529,
      "grad_norm": 0.9293483419699585,
      "learning_rate": 1.0024472817957318e-06,
      "loss": 0.0085,
      "step": 6453
    },
    {
      "epoch": 1.4684869169510808,
      "grad_norm": 1.4315734788723715,
      "learning_rate": 1.0023760606215412e-06,
      "loss": 0.0367,
      "step": 6454
    },
    {
      "epoch": 1.4687144482366326,
      "grad_norm": 1.1555294375278542,
      "learning_rate": 1.0023048317344752e-06,
      "loss": 0.0213,
      "step": 6455
    },
    {
      "epoch": 1.4689419795221843,
      "grad_norm": 1.0518687824248476,
      "learning_rate": 1.0022335951359894e-06,
      "loss": 0.0175,
      "step": 6456
    },
    {
      "epoch": 1.469169510807736,
      "grad_norm": 0.951733146012008,
      "learning_rate": 1.0021623508275397e-06,
      "loss": 0.027,
      "step": 6457
    },
    {
      "epoch": 1.4693970420932878,
      "grad_norm": 1.9723448140457374,
      "learning_rate": 1.0020910988105822e-06,
      "loss": 0.0413,
      "step": 6458
    },
    {
      "epoch": 1.4696245733788396,
      "grad_norm": 1.1686953011150991,
      "learning_rate": 1.0020198390865735e-06,
      "loss": 0.032,
      "step": 6459
    },
    {
      "epoch": 1.4698521046643913,
      "grad_norm": 0.8845231672296715,
      "learning_rate": 1.0019485716569696e-06,
      "loss": 0.0192,
      "step": 6460
    },
    {
      "epoch": 1.470079635949943,
      "grad_norm": 1.34301819211037,
      "learning_rate": 1.0018772965232272e-06,
      "loss": 0.0272,
      "step": 6461
    },
    {
      "epoch": 1.4703071672354948,
      "grad_norm": 1.0900651393805174,
      "learning_rate": 1.0018060136868033e-06,
      "loss": 0.0398,
      "step": 6462
    },
    {
      "epoch": 1.4705346985210466,
      "grad_norm": 0.8738815245685796,
      "learning_rate": 1.0017347231491544e-06,
      "loss": 0.0088,
      "step": 6463
    },
    {
      "epoch": 1.4707622298065983,
      "grad_norm": 1.0394239050490557,
      "learning_rate": 1.0016634249117378e-06,
      "loss": 0.0181,
      "step": 6464
    },
    {
      "epoch": 1.47098976109215,
      "grad_norm": 1.684560812659319,
      "learning_rate": 1.0015921189760105e-06,
      "loss": 0.0782,
      "step": 6465
    },
    {
      "epoch": 1.4712172923777018,
      "grad_norm": 0.6453980486743178,
      "learning_rate": 1.00152080534343e-06,
      "loss": 0.0194,
      "step": 6466
    },
    {
      "epoch": 1.4714448236632536,
      "grad_norm": 0.7102456235440718,
      "learning_rate": 1.001449484015454e-06,
      "loss": 0.0173,
      "step": 6467
    },
    {
      "epoch": 1.4716723549488053,
      "grad_norm": 0.9401483471225579,
      "learning_rate": 1.0013781549935396e-06,
      "loss": 0.0304,
      "step": 6468
    },
    {
      "epoch": 1.4718998862343573,
      "grad_norm": 1.3395349325920776,
      "learning_rate": 1.0013068182791454e-06,
      "loss": 0.025,
      "step": 6469
    },
    {
      "epoch": 1.472127417519909,
      "grad_norm": 0.8791378015144556,
      "learning_rate": 1.0012354738737288e-06,
      "loss": 0.0331,
      "step": 6470
    },
    {
      "epoch": 1.4723549488054608,
      "grad_norm": 0.9918757720929136,
      "learning_rate": 1.0011641217787481e-06,
      "loss": 0.017,
      "step": 6471
    },
    {
      "epoch": 1.4725824800910126,
      "grad_norm": 1.0478775077595175,
      "learning_rate": 1.001092761995662e-06,
      "loss": 0.0391,
      "step": 6472
    },
    {
      "epoch": 1.4728100113765643,
      "grad_norm": 0.8503563945359769,
      "learning_rate": 1.0010213945259282e-06,
      "loss": 0.0202,
      "step": 6473
    },
    {
      "epoch": 1.473037542662116,
      "grad_norm": 1.222389885020765,
      "learning_rate": 1.000950019371006e-06,
      "loss": 0.0237,
      "step": 6474
    },
    {
      "epoch": 1.4732650739476678,
      "grad_norm": 0.8048051018799113,
      "learning_rate": 1.0008786365323539e-06,
      "loss": 0.0178,
      "step": 6475
    },
    {
      "epoch": 1.4734926052332196,
      "grad_norm": 1.5299229006902042,
      "learning_rate": 1.0008072460114308e-06,
      "loss": 0.0328,
      "step": 6476
    },
    {
      "epoch": 1.4737201365187713,
      "grad_norm": 1.6732137429792273,
      "learning_rate": 1.0007358478096959e-06,
      "loss": 0.0471,
      "step": 6477
    },
    {
      "epoch": 1.473947667804323,
      "grad_norm": 0.43763199665482255,
      "learning_rate": 1.0006644419286084e-06,
      "loss": 0.0098,
      "step": 6478
    },
    {
      "epoch": 1.4741751990898748,
      "grad_norm": 0.7028792539926481,
      "learning_rate": 1.0005930283696277e-06,
      "loss": 0.0085,
      "step": 6479
    },
    {
      "epoch": 1.4744027303754266,
      "grad_norm": 0.9331242445594948,
      "learning_rate": 1.0005216071342133e-06,
      "loss": 0.0197,
      "step": 6480
    },
    {
      "epoch": 1.4746302616609783,
      "grad_norm": 1.8900811534734059,
      "learning_rate": 1.000450178223825e-06,
      "loss": 0.0396,
      "step": 6481
    },
    {
      "epoch": 1.47485779294653,
      "grad_norm": 1.15876808023037,
      "learning_rate": 1.0003787416399226e-06,
      "loss": 0.0302,
      "step": 6482
    },
    {
      "epoch": 1.4750853242320818,
      "grad_norm": 1.3311084179933157,
      "learning_rate": 1.0003072973839665e-06,
      "loss": 0.0214,
      "step": 6483
    },
    {
      "epoch": 1.4753128555176338,
      "grad_norm": 0.7978453155730808,
      "learning_rate": 1.0002358454574163e-06,
      "loss": 0.0126,
      "step": 6484
    },
    {
      "epoch": 1.4755403868031856,
      "grad_norm": 1.3200907472733454,
      "learning_rate": 1.0001643858617326e-06,
      "loss": 0.0372,
      "step": 6485
    },
    {
      "epoch": 1.4757679180887373,
      "grad_norm": 0.7587699333301391,
      "learning_rate": 1.0000929185983762e-06,
      "loss": 0.0145,
      "step": 6486
    },
    {
      "epoch": 1.475995449374289,
      "grad_norm": 1.115889703946624,
      "learning_rate": 1.0000214436688074e-06,
      "loss": 0.0221,
      "step": 6487
    },
    {
      "epoch": 1.4762229806598408,
      "grad_norm": 1.7530265279400412,
      "learning_rate": 9.99949961074487e-07,
      "loss": 0.0517,
      "step": 6488
    },
    {
      "epoch": 1.4764505119453926,
      "grad_norm": 1.4986267673780238,
      "learning_rate": 9.998784708168762e-07,
      "loss": 0.0281,
      "step": 6489
    },
    {
      "epoch": 1.4766780432309443,
      "grad_norm": 1.1981264017896749,
      "learning_rate": 9.998069728974357e-07,
      "loss": 0.0351,
      "step": 6490
    },
    {
      "epoch": 1.476905574516496,
      "grad_norm": 0.8147906912659221,
      "learning_rate": 9.997354673176273e-07,
      "loss": 0.0211,
      "step": 6491
    },
    {
      "epoch": 1.4771331058020478,
      "grad_norm": 0.7325162068189855,
      "learning_rate": 9.996639540789124e-07,
      "loss": 0.015,
      "step": 6492
    },
    {
      "epoch": 1.4773606370875996,
      "grad_norm": 1.0318994529873236,
      "learning_rate": 9.995924331827521e-07,
      "loss": 0.0223,
      "step": 6493
    },
    {
      "epoch": 1.4775881683731513,
      "grad_norm": 1.0766661165747942,
      "learning_rate": 9.99520904630609e-07,
      "loss": 0.0217,
      "step": 6494
    },
    {
      "epoch": 1.477815699658703,
      "grad_norm": 1.052496175953845,
      "learning_rate": 9.994493684239443e-07,
      "loss": 0.0261,
      "step": 6495
    },
    {
      "epoch": 1.4780432309442548,
      "grad_norm": 0.740359492998119,
      "learning_rate": 9.993778245642202e-07,
      "loss": 0.0218,
      "step": 6496
    },
    {
      "epoch": 1.4782707622298066,
      "grad_norm": 1.1088535016409602,
      "learning_rate": 9.99306273052899e-07,
      "loss": 0.0158,
      "step": 6497
    },
    {
      "epoch": 1.4784982935153583,
      "grad_norm": 1.4976458680888785,
      "learning_rate": 9.99234713891443e-07,
      "loss": 0.021,
      "step": 6498
    },
    {
      "epoch": 1.47872582480091,
      "grad_norm": 0.6855826863773865,
      "learning_rate": 9.99163147081315e-07,
      "loss": 0.0099,
      "step": 6499
    },
    {
      "epoch": 1.4789533560864618,
      "grad_norm": 1.0945214632210196,
      "learning_rate": 9.990915726239774e-07,
      "loss": 0.0297,
      "step": 6500
    },
    {
      "epoch": 1.4791808873720136,
      "grad_norm": 1.0597968699238194,
      "learning_rate": 9.990199905208933e-07,
      "loss": 0.0391,
      "step": 6501
    },
    {
      "epoch": 1.4794084186575653,
      "grad_norm": 1.3770835022888792,
      "learning_rate": 9.989484007735256e-07,
      "loss": 0.0389,
      "step": 6502
    },
    {
      "epoch": 1.479635949943117,
      "grad_norm": 1.8160093879173487,
      "learning_rate": 9.988768033833374e-07,
      "loss": 0.0681,
      "step": 6503
    },
    {
      "epoch": 1.4798634812286688,
      "grad_norm": 1.7490337161767424,
      "learning_rate": 9.98805198351792e-07,
      "loss": 0.0662,
      "step": 6504
    },
    {
      "epoch": 1.4800910125142206,
      "grad_norm": 0.6476476571772248,
      "learning_rate": 9.98733585680353e-07,
      "loss": 0.0105,
      "step": 6505
    },
    {
      "epoch": 1.4803185437997723,
      "grad_norm": 0.7755957452558864,
      "learning_rate": 9.986619653704838e-07,
      "loss": 0.0139,
      "step": 6506
    },
    {
      "epoch": 1.480546075085324,
      "grad_norm": 0.633426255815866,
      "learning_rate": 9.985903374236487e-07,
      "loss": 0.017,
      "step": 6507
    },
    {
      "epoch": 1.480773606370876,
      "grad_norm": 0.8678772008391588,
      "learning_rate": 9.985187018413108e-07,
      "loss": 0.0257,
      "step": 6508
    },
    {
      "epoch": 1.4810011376564278,
      "grad_norm": 0.8694342307699764,
      "learning_rate": 9.98447058624935e-07,
      "loss": 0.0172,
      "step": 6509
    },
    {
      "epoch": 1.4812286689419796,
      "grad_norm": 1.2761367957965584,
      "learning_rate": 9.983754077759852e-07,
      "loss": 0.0369,
      "step": 6510
    },
    {
      "epoch": 1.4814562002275313,
      "grad_norm": 1.0645165105201808,
      "learning_rate": 9.98303749295926e-07,
      "loss": 0.03,
      "step": 6511
    },
    {
      "epoch": 1.481683731513083,
      "grad_norm": 1.1149897643794233,
      "learning_rate": 9.982320831862217e-07,
      "loss": 0.0244,
      "step": 6512
    },
    {
      "epoch": 1.4819112627986348,
      "grad_norm": 1.5255781412882516,
      "learning_rate": 9.981604094483374e-07,
      "loss": 0.0314,
      "step": 6513
    },
    {
      "epoch": 1.4821387940841866,
      "grad_norm": 1.43555128988365,
      "learning_rate": 9.980887280837377e-07,
      "loss": 0.0441,
      "step": 6514
    },
    {
      "epoch": 1.4823663253697383,
      "grad_norm": 1.16961605961412,
      "learning_rate": 9.980170390938873e-07,
      "loss": 0.0308,
      "step": 6515
    },
    {
      "epoch": 1.48259385665529,
      "grad_norm": 1.0707706605010894,
      "learning_rate": 9.979453424802522e-07,
      "loss": 0.0244,
      "step": 6516
    },
    {
      "epoch": 1.4828213879408418,
      "grad_norm": 1.2895035901019312,
      "learning_rate": 9.978736382442969e-07,
      "loss": 0.0494,
      "step": 6517
    },
    {
      "epoch": 1.4830489192263936,
      "grad_norm": 1.310843233111791,
      "learning_rate": 9.978019263874875e-07,
      "loss": 0.0301,
      "step": 6518
    },
    {
      "epoch": 1.4832764505119453,
      "grad_norm": 1.6244622239104036,
      "learning_rate": 9.977302069112896e-07,
      "loss": 0.0271,
      "step": 6519
    },
    {
      "epoch": 1.483503981797497,
      "grad_norm": 1.3621603406802865,
      "learning_rate": 9.97658479817169e-07,
      "loss": 0.0281,
      "step": 6520
    },
    {
      "epoch": 1.4837315130830488,
      "grad_norm": 0.7915875199989744,
      "learning_rate": 9.975867451065913e-07,
      "loss": 0.0114,
      "step": 6521
    },
    {
      "epoch": 1.4839590443686006,
      "grad_norm": 2.0905935656543404,
      "learning_rate": 9.97515002781023e-07,
      "loss": 0.0765,
      "step": 6522
    },
    {
      "epoch": 1.4841865756541526,
      "grad_norm": 1.2649643198155474,
      "learning_rate": 9.9744325284193e-07,
      "loss": 0.0327,
      "step": 6523
    },
    {
      "epoch": 1.4844141069397043,
      "grad_norm": 1.0390661565491186,
      "learning_rate": 9.973714952907792e-07,
      "loss": 0.027,
      "step": 6524
    },
    {
      "epoch": 1.484641638225256,
      "grad_norm": 1.5278981168386385,
      "learning_rate": 9.97299730129037e-07,
      "loss": 0.0397,
      "step": 6525
    },
    {
      "epoch": 1.4848691695108078,
      "grad_norm": 1.021782093064846,
      "learning_rate": 9.972279573581705e-07,
      "loss": 0.0194,
      "step": 6526
    },
    {
      "epoch": 1.4850967007963596,
      "grad_norm": 0.8266467944017118,
      "learning_rate": 9.97156176979646e-07,
      "loss": 0.022,
      "step": 6527
    },
    {
      "epoch": 1.4853242320819113,
      "grad_norm": 0.7868124068525434,
      "learning_rate": 9.970843889949305e-07,
      "loss": 0.011,
      "step": 6528
    },
    {
      "epoch": 1.485551763367463,
      "grad_norm": 1.189191061369537,
      "learning_rate": 9.97012593405492e-07,
      "loss": 0.0361,
      "step": 6529
    },
    {
      "epoch": 1.4857792946530148,
      "grad_norm": 1.2164226385849939,
      "learning_rate": 9.969407902127972e-07,
      "loss": 0.0171,
      "step": 6530
    },
    {
      "epoch": 1.4860068259385666,
      "grad_norm": 1.2574211803000732,
      "learning_rate": 9.968689794183137e-07,
      "loss": 0.0261,
      "step": 6531
    },
    {
      "epoch": 1.4862343572241183,
      "grad_norm": 1.3800274592660757,
      "learning_rate": 9.967971610235095e-07,
      "loss": 0.0298,
      "step": 6532
    },
    {
      "epoch": 1.48646188850967,
      "grad_norm": 2.596631249407389,
      "learning_rate": 9.967253350298522e-07,
      "loss": 0.0824,
      "step": 6533
    },
    {
      "epoch": 1.4866894197952218,
      "grad_norm": 1.2827191671719944,
      "learning_rate": 9.966535014388098e-07,
      "loss": 0.0133,
      "step": 6534
    },
    {
      "epoch": 1.4869169510807736,
      "grad_norm": 0.9864326301137617,
      "learning_rate": 9.965816602518505e-07,
      "loss": 0.0169,
      "step": 6535
    },
    {
      "epoch": 1.4871444823663253,
      "grad_norm": 1.3480233639513728,
      "learning_rate": 9.965098114704425e-07,
      "loss": 0.0469,
      "step": 6536
    },
    {
      "epoch": 1.487372013651877,
      "grad_norm": 0.685810034151907,
      "learning_rate": 9.964379550960544e-07,
      "loss": 0.0167,
      "step": 6537
    },
    {
      "epoch": 1.4875995449374289,
      "grad_norm": 0.9568981168441036,
      "learning_rate": 9.96366091130155e-07,
      "loss": 0.0314,
      "step": 6538
    },
    {
      "epoch": 1.4878270762229806,
      "grad_norm": 1.6341154779837714,
      "learning_rate": 9.962942195742125e-07,
      "loss": 0.0577,
      "step": 6539
    },
    {
      "epoch": 1.4880546075085324,
      "grad_norm": 8.811099545201115,
      "learning_rate": 9.96222340429696e-07,
      "loss": 0.1751,
      "step": 6540
    },
    {
      "epoch": 1.488282138794084,
      "grad_norm": 1.0114318005924805,
      "learning_rate": 9.96150453698075e-07,
      "loss": 0.029,
      "step": 6541
    },
    {
      "epoch": 1.4885096700796359,
      "grad_norm": 1.4441091534455175,
      "learning_rate": 9.960785593808187e-07,
      "loss": 0.0153,
      "step": 6542
    },
    {
      "epoch": 1.4887372013651876,
      "grad_norm": 48.46661837605496,
      "learning_rate": 9.960066574793959e-07,
      "loss": 0.488,
      "step": 6543
    },
    {
      "epoch": 1.4889647326507394,
      "grad_norm": 1.1015052782252979,
      "learning_rate": 9.959347479952764e-07,
      "loss": 0.0251,
      "step": 6544
    },
    {
      "epoch": 1.4891922639362911,
      "grad_norm": 1.708134539419919,
      "learning_rate": 9.958628309299303e-07,
      "loss": 0.0352,
      "step": 6545
    },
    {
      "epoch": 1.4894197952218429,
      "grad_norm": 1.1503598719279833,
      "learning_rate": 9.95790906284827e-07,
      "loss": 0.0181,
      "step": 6546
    },
    {
      "epoch": 1.4896473265073948,
      "grad_norm": 0.6162346862876698,
      "learning_rate": 9.957189740614364e-07,
      "loss": 0.0136,
      "step": 6547
    },
    {
      "epoch": 1.4898748577929466,
      "grad_norm": 0.9769826135275878,
      "learning_rate": 9.956470342612292e-07,
      "loss": 0.0321,
      "step": 6548
    },
    {
      "epoch": 1.4901023890784983,
      "grad_norm": 1.2898048218947735,
      "learning_rate": 9.955750868856753e-07,
      "loss": 0.0292,
      "step": 6549
    },
    {
      "epoch": 1.49032992036405,
      "grad_norm": 1.0495260184817348,
      "learning_rate": 9.955031319362455e-07,
      "loss": 0.0225,
      "step": 6550
    },
    {
      "epoch": 1.4905574516496019,
      "grad_norm": 0.9451273648772042,
      "learning_rate": 9.954311694144101e-07,
      "loss": 0.0365,
      "step": 6551
    },
    {
      "epoch": 1.4907849829351536,
      "grad_norm": 0.7141248917734317,
      "learning_rate": 9.9535919932164e-07,
      "loss": 0.0136,
      "step": 6552
    },
    {
      "epoch": 1.4910125142207054,
      "grad_norm": 1.5055029123037509,
      "learning_rate": 9.952872216594062e-07,
      "loss": 0.0327,
      "step": 6553
    },
    {
      "epoch": 1.491240045506257,
      "grad_norm": 1.2197166533456572,
      "learning_rate": 9.952152364291795e-07,
      "loss": 0.0182,
      "step": 6554
    },
    {
      "epoch": 1.4914675767918089,
      "grad_norm": 1.5013632322134673,
      "learning_rate": 9.951432436324317e-07,
      "loss": 0.058,
      "step": 6555
    },
    {
      "epoch": 1.4916951080773606,
      "grad_norm": 0.8229751970283513,
      "learning_rate": 9.950712432706338e-07,
      "loss": 0.0126,
      "step": 6556
    },
    {
      "epoch": 1.4919226393629124,
      "grad_norm": 0.9036383949717536,
      "learning_rate": 9.949992353452575e-07,
      "loss": 0.0185,
      "step": 6557
    },
    {
      "epoch": 1.4921501706484641,
      "grad_norm": 1.5314393854266743,
      "learning_rate": 9.949272198577741e-07,
      "loss": 0.0386,
      "step": 6558
    },
    {
      "epoch": 1.4923777019340159,
      "grad_norm": 1.2094828607200816,
      "learning_rate": 9.948551968096562e-07,
      "loss": 0.0248,
      "step": 6559
    },
    {
      "epoch": 1.4926052332195676,
      "grad_norm": 1.1532482925479424,
      "learning_rate": 9.947831662023751e-07,
      "loss": 0.0208,
      "step": 6560
    },
    {
      "epoch": 1.4928327645051196,
      "grad_norm": 0.9806602475157391,
      "learning_rate": 9.947111280374036e-07,
      "loss": 0.027,
      "step": 6561
    },
    {
      "epoch": 1.4930602957906713,
      "grad_norm": 1.60568299137254,
      "learning_rate": 9.946390823162136e-07,
      "loss": 0.0771,
      "step": 6562
    },
    {
      "epoch": 1.493287827076223,
      "grad_norm": 1.8385530312374776,
      "learning_rate": 9.945670290402778e-07,
      "loss": 0.027,
      "step": 6563
    },
    {
      "epoch": 1.4935153583617748,
      "grad_norm": 1.6642618859811353,
      "learning_rate": 9.944949682110689e-07,
      "loss": 0.0303,
      "step": 6564
    },
    {
      "epoch": 1.4937428896473266,
      "grad_norm": 1.6989309916692763,
      "learning_rate": 9.944228998300592e-07,
      "loss": 0.0306,
      "step": 6565
    },
    {
      "epoch": 1.4939704209328784,
      "grad_norm": 1.1418893235884928,
      "learning_rate": 9.943508238987223e-07,
      "loss": 0.0248,
      "step": 6566
    },
    {
      "epoch": 1.49419795221843,
      "grad_norm": 1.1063431497011869,
      "learning_rate": 9.942787404185307e-07,
      "loss": 0.021,
      "step": 6567
    },
    {
      "epoch": 1.4944254835039819,
      "grad_norm": 1.0900849834712658,
      "learning_rate": 9.942066493909582e-07,
      "loss": 0.039,
      "step": 6568
    },
    {
      "epoch": 1.4946530147895336,
      "grad_norm": 0.8812585966390157,
      "learning_rate": 9.941345508174778e-07,
      "loss": 0.0254,
      "step": 6569
    },
    {
      "epoch": 1.4948805460750854,
      "grad_norm": 0.9753213224487992,
      "learning_rate": 9.940624446995633e-07,
      "loss": 0.0231,
      "step": 6570
    },
    {
      "epoch": 1.4951080773606371,
      "grad_norm": 1.2089422433395995,
      "learning_rate": 9.939903310386883e-07,
      "loss": 0.0259,
      "step": 6571
    },
    {
      "epoch": 1.4953356086461889,
      "grad_norm": 0.5864095115048934,
      "learning_rate": 9.939182098363267e-07,
      "loss": 0.0128,
      "step": 6572
    },
    {
      "epoch": 1.4955631399317406,
      "grad_norm": 0.9001389442956854,
      "learning_rate": 9.938460810939526e-07,
      "loss": 0.0106,
      "step": 6573
    },
    {
      "epoch": 1.4957906712172924,
      "grad_norm": 1.0616564747090018,
      "learning_rate": 9.9377394481304e-07,
      "loss": 0.0293,
      "step": 6574
    },
    {
      "epoch": 1.4960182025028441,
      "grad_norm": 1.0975433183868526,
      "learning_rate": 9.937018009950637e-07,
      "loss": 0.0438,
      "step": 6575
    },
    {
      "epoch": 1.4962457337883959,
      "grad_norm": 0.8757355770920893,
      "learning_rate": 9.936296496414974e-07,
      "loss": 0.0155,
      "step": 6576
    },
    {
      "epoch": 1.4964732650739476,
      "grad_norm": 1.296977963478537,
      "learning_rate": 9.935574907538162e-07,
      "loss": 0.0299,
      "step": 6577
    },
    {
      "epoch": 1.4967007963594994,
      "grad_norm": 1.5770938202420768,
      "learning_rate": 9.93485324333495e-07,
      "loss": 0.0447,
      "step": 6578
    },
    {
      "epoch": 1.4969283276450511,
      "grad_norm": 1.8187781113713664,
      "learning_rate": 9.934131503820086e-07,
      "loss": 0.0376,
      "step": 6579
    },
    {
      "epoch": 1.4971558589306029,
      "grad_norm": 1.1921366812845895,
      "learning_rate": 9.933409689008323e-07,
      "loss": 0.021,
      "step": 6580
    },
    {
      "epoch": 1.4973833902161546,
      "grad_norm": 0.8690783796494854,
      "learning_rate": 9.932687798914408e-07,
      "loss": 0.0112,
      "step": 6581
    },
    {
      "epoch": 1.4976109215017064,
      "grad_norm": 0.7177586988388654,
      "learning_rate": 9.9319658335531e-07,
      "loss": 0.0203,
      "step": 6582
    },
    {
      "epoch": 1.4978384527872581,
      "grad_norm": 0.9905715701722341,
      "learning_rate": 9.931243792939157e-07,
      "loss": 0.0282,
      "step": 6583
    },
    {
      "epoch": 1.4980659840728099,
      "grad_norm": 0.5676357177462131,
      "learning_rate": 9.93052167708733e-07,
      "loss": 0.0154,
      "step": 6584
    },
    {
      "epoch": 1.4982935153583616,
      "grad_norm": 1.4780735039451949,
      "learning_rate": 9.929799486012381e-07,
      "loss": 0.0318,
      "step": 6585
    },
    {
      "epoch": 1.4985210466439136,
      "grad_norm": 0.8150823907413614,
      "learning_rate": 9.92907721972907e-07,
      "loss": 0.0158,
      "step": 6586
    },
    {
      "epoch": 1.4987485779294654,
      "grad_norm": 1.3145997284823179,
      "learning_rate": 9.928354878252156e-07,
      "loss": 0.0217,
      "step": 6587
    },
    {
      "epoch": 1.4989761092150171,
      "grad_norm": 1.3597840014937403,
      "learning_rate": 9.927632461596409e-07,
      "loss": 0.0229,
      "step": 6588
    },
    {
      "epoch": 1.4992036405005689,
      "grad_norm": 1.8450367592169252,
      "learning_rate": 9.926909969776588e-07,
      "loss": 0.0265,
      "step": 6589
    },
    {
      "epoch": 1.4994311717861206,
      "grad_norm": 1.0598284283095365,
      "learning_rate": 9.926187402807461e-07,
      "loss": 0.0196,
      "step": 6590
    },
    {
      "epoch": 1.4996587030716724,
      "grad_norm": 1.5414138432676403,
      "learning_rate": 9.925464760703796e-07,
      "loss": 0.0456,
      "step": 6591
    },
    {
      "epoch": 1.4998862343572241,
      "grad_norm": 1.3712705609079485,
      "learning_rate": 9.924742043480361e-07,
      "loss": 0.035,
      "step": 6592
    },
    {
      "epoch": 1.5001137656427759,
      "grad_norm": 0.9971920077226126,
      "learning_rate": 9.924019251151932e-07,
      "loss": 0.0123,
      "step": 6593
    },
    {
      "epoch": 1.5003412969283276,
      "grad_norm": 1.367423618774891,
      "learning_rate": 9.923296383733274e-07,
      "loss": 0.0332,
      "step": 6594
    },
    {
      "epoch": 1.5005688282138794,
      "grad_norm": 0.8103598457125527,
      "learning_rate": 9.92257344123917e-07,
      "loss": 0.015,
      "step": 6595
    },
    {
      "epoch": 1.5007963594994311,
      "grad_norm": 1.5852508733151163,
      "learning_rate": 9.921850423684387e-07,
      "loss": 0.0393,
      "step": 6596
    },
    {
      "epoch": 1.5010238907849829,
      "grad_norm": 0.7983856504921896,
      "learning_rate": 9.921127331083708e-07,
      "loss": 0.0155,
      "step": 6597
    },
    {
      "epoch": 1.5012514220705349,
      "grad_norm": 0.8110345399367997,
      "learning_rate": 9.92040416345191e-07,
      "loss": 0.0106,
      "step": 6598
    },
    {
      "epoch": 1.5014789533560866,
      "grad_norm": 1.057033281890757,
      "learning_rate": 9.91968092080377e-07,
      "loss": 0.0364,
      "step": 6599
    },
    {
      "epoch": 1.5017064846416384,
      "grad_norm": 0.763089743036924,
      "learning_rate": 9.918957603154076e-07,
      "loss": 0.0098,
      "step": 6600
    },
    {
      "epoch": 1.5019340159271901,
      "grad_norm": 1.349260836853775,
      "learning_rate": 9.918234210517606e-07,
      "loss": 0.0291,
      "step": 6601
    },
    {
      "epoch": 1.5021615472127419,
      "grad_norm": 0.9124150356814112,
      "learning_rate": 9.917510742909147e-07,
      "loss": 0.0359,
      "step": 6602
    },
    {
      "epoch": 1.5023890784982936,
      "grad_norm": 1.0789877329504083,
      "learning_rate": 9.916787200343487e-07,
      "loss": 0.0271,
      "step": 6603
    },
    {
      "epoch": 1.5026166097838454,
      "grad_norm": 0.8830240075896032,
      "learning_rate": 9.91606358283541e-07,
      "loss": 0.023,
      "step": 6604
    },
    {
      "epoch": 1.5028441410693971,
      "grad_norm": 0.9189612669050022,
      "learning_rate": 9.915339890399707e-07,
      "loss": 0.0187,
      "step": 6605
    },
    {
      "epoch": 1.5030716723549489,
      "grad_norm": 1.1223886637061031,
      "learning_rate": 9.914616123051172e-07,
      "loss": 0.0185,
      "step": 6606
    },
    {
      "epoch": 1.5032992036405006,
      "grad_norm": 1.1895451968922761,
      "learning_rate": 9.913892280804593e-07,
      "loss": 0.0219,
      "step": 6607
    },
    {
      "epoch": 1.5035267349260524,
      "grad_norm": 1.33425573815634,
      "learning_rate": 9.913168363674768e-07,
      "loss": 0.0273,
      "step": 6608
    },
    {
      "epoch": 1.5037542662116041,
      "grad_norm": 0.9253483092504505,
      "learning_rate": 9.912444371676489e-07,
      "loss": 0.0219,
      "step": 6609
    },
    {
      "epoch": 1.5039817974971559,
      "grad_norm": 1.1040945433765377,
      "learning_rate": 9.911720304824555e-07,
      "loss": 0.0206,
      "step": 6610
    },
    {
      "epoch": 1.5042093287827076,
      "grad_norm": 1.3224220455519788,
      "learning_rate": 9.910996163133762e-07,
      "loss": 0.0228,
      "step": 6611
    },
    {
      "epoch": 1.5044368600682594,
      "grad_norm": 1.4206586266356582,
      "learning_rate": 9.910271946618913e-07,
      "loss": 0.04,
      "step": 6612
    },
    {
      "epoch": 1.5046643913538111,
      "grad_norm": 2.5555274852028464,
      "learning_rate": 9.90954765529481e-07,
      "loss": 0.0588,
      "step": 6613
    },
    {
      "epoch": 1.5048919226393629,
      "grad_norm": 1.4075662347643996,
      "learning_rate": 9.908823289176255e-07,
      "loss": 0.0497,
      "step": 6614
    },
    {
      "epoch": 1.5051194539249146,
      "grad_norm": 1.8116951308916345,
      "learning_rate": 9.90809884827805e-07,
      "loss": 0.0628,
      "step": 6615
    },
    {
      "epoch": 1.5053469852104664,
      "grad_norm": 1.1027641178320662,
      "learning_rate": 9.907374332615007e-07,
      "loss": 0.0507,
      "step": 6616
    },
    {
      "epoch": 1.5055745164960181,
      "grad_norm": 0.6991004095047127,
      "learning_rate": 9.90664974220193e-07,
      "loss": 0.013,
      "step": 6617
    },
    {
      "epoch": 1.50580204778157,
      "grad_norm": 1.15443604824671,
      "learning_rate": 9.90592507705363e-07,
      "loss": 0.0198,
      "step": 6618
    },
    {
      "epoch": 1.5060295790671216,
      "grad_norm": 1.2034338936063906,
      "learning_rate": 9.905200337184915e-07,
      "loss": 0.0229,
      "step": 6619
    },
    {
      "epoch": 1.5062571103526734,
      "grad_norm": 1.4942841146862913,
      "learning_rate": 9.904475522610602e-07,
      "loss": 0.0322,
      "step": 6620
    },
    {
      "epoch": 1.5064846416382252,
      "grad_norm": 1.6969990633258083,
      "learning_rate": 9.9037506333455e-07,
      "loss": 0.0807,
      "step": 6621
    },
    {
      "epoch": 1.506712172923777,
      "grad_norm": 1.6210050209374949,
      "learning_rate": 9.90302566940443e-07,
      "loss": 0.0719,
      "step": 6622
    },
    {
      "epoch": 1.5069397042093287,
      "grad_norm": 1.2474940776817767,
      "learning_rate": 9.902300630802201e-07,
      "loss": 0.0399,
      "step": 6623
    },
    {
      "epoch": 1.5071672354948804,
      "grad_norm": 0.837642435435654,
      "learning_rate": 9.901575517553636e-07,
      "loss": 0.0157,
      "step": 6624
    },
    {
      "epoch": 1.5073947667804322,
      "grad_norm": 1.1560577549497977,
      "learning_rate": 9.900850329673559e-07,
      "loss": 0.0121,
      "step": 6625
    },
    {
      "epoch": 1.507622298065984,
      "grad_norm": 1.6091563866332808,
      "learning_rate": 9.900125067176782e-07,
      "loss": 0.0332,
      "step": 6626
    },
    {
      "epoch": 1.5078498293515359,
      "grad_norm": 0.5882706831462519,
      "learning_rate": 9.899399730078138e-07,
      "loss": 0.0221,
      "step": 6627
    },
    {
      "epoch": 1.5080773606370876,
      "grad_norm": 1.4498045295631503,
      "learning_rate": 9.898674318392446e-07,
      "loss": 0.0185,
      "step": 6628
    },
    {
      "epoch": 1.5083048919226394,
      "grad_norm": 1.4230136293074578,
      "learning_rate": 9.897948832134532e-07,
      "loss": 0.0249,
      "step": 6629
    },
    {
      "epoch": 1.5085324232081911,
      "grad_norm": 0.8420732883563372,
      "learning_rate": 9.897223271319227e-07,
      "loss": 0.0244,
      "step": 6630
    },
    {
      "epoch": 1.508759954493743,
      "grad_norm": 1.4719539102826928,
      "learning_rate": 9.896497635961357e-07,
      "loss": 0.0309,
      "step": 6631
    },
    {
      "epoch": 1.5089874857792946,
      "grad_norm": 0.8467326556394097,
      "learning_rate": 9.895771926075753e-07,
      "loss": 0.0252,
      "step": 6632
    },
    {
      "epoch": 1.5092150170648464,
      "grad_norm": 0.7904313040506917,
      "learning_rate": 9.895046141677248e-07,
      "loss": 0.0173,
      "step": 6633
    },
    {
      "epoch": 1.5094425483503981,
      "grad_norm": 1.141524501415206,
      "learning_rate": 9.894320282780675e-07,
      "loss": 0.0336,
      "step": 6634
    },
    {
      "epoch": 1.50967007963595,
      "grad_norm": 0.7509190456969517,
      "learning_rate": 9.89359434940087e-07,
      "loss": 0.0159,
      "step": 6635
    },
    {
      "epoch": 1.5098976109215017,
      "grad_norm": 1.8948621758043176,
      "learning_rate": 9.89286834155267e-07,
      "loss": 0.0376,
      "step": 6636
    },
    {
      "epoch": 1.5101251422070536,
      "grad_norm": 0.7530416159112719,
      "learning_rate": 9.89214225925091e-07,
      "loss": 0.0157,
      "step": 6637
    },
    {
      "epoch": 1.5103526734926054,
      "grad_norm": 0.7653184176286111,
      "learning_rate": 9.891416102510436e-07,
      "loss": 0.0186,
      "step": 6638
    },
    {
      "epoch": 1.5105802047781571,
      "grad_norm": 1.0986906061359811,
      "learning_rate": 9.890689871346084e-07,
      "loss": 0.0369,
      "step": 6639
    },
    {
      "epoch": 1.5108077360637089,
      "grad_norm": 1.0874136319612577,
      "learning_rate": 9.8899635657727e-07,
      "loss": 0.0265,
      "step": 6640
    },
    {
      "epoch": 1.5110352673492606,
      "grad_norm": 0.8527398550564912,
      "learning_rate": 9.889237185805126e-07,
      "loss": 0.0122,
      "step": 6641
    },
    {
      "epoch": 1.5112627986348124,
      "grad_norm": 0.6283273575316396,
      "learning_rate": 9.88851073145821e-07,
      "loss": 0.0199,
      "step": 6642
    },
    {
      "epoch": 1.5114903299203641,
      "grad_norm": 1.4497090224286908,
      "learning_rate": 9.887784202746797e-07,
      "loss": 0.0333,
      "step": 6643
    },
    {
      "epoch": 1.511717861205916,
      "grad_norm": 1.4938621283787956,
      "learning_rate": 9.887057599685735e-07,
      "loss": 0.0569,
      "step": 6644
    },
    {
      "epoch": 1.5119453924914676,
      "grad_norm": 1.6336417251582953,
      "learning_rate": 9.88633092228988e-07,
      "loss": 0.0462,
      "step": 6645
    },
    {
      "epoch": 1.5121729237770194,
      "grad_norm": 1.2241304248465237,
      "learning_rate": 9.885604170574081e-07,
      "loss": 0.0174,
      "step": 6646
    },
    {
      "epoch": 1.5124004550625711,
      "grad_norm": 1.2606080177082761,
      "learning_rate": 9.884877344553189e-07,
      "loss": 0.0174,
      "step": 6647
    },
    {
      "epoch": 1.512627986348123,
      "grad_norm": 1.8808732730138569,
      "learning_rate": 9.884150444242063e-07,
      "loss": 0.0638,
      "step": 6648
    },
    {
      "epoch": 1.5128555176336747,
      "grad_norm": 0.9107630601145861,
      "learning_rate": 9.883423469655553e-07,
      "loss": 0.0235,
      "step": 6649
    },
    {
      "epoch": 1.5130830489192264,
      "grad_norm": 0.7714393934298326,
      "learning_rate": 9.882696420808526e-07,
      "loss": 0.0195,
      "step": 6650
    },
    {
      "epoch": 1.5133105802047782,
      "grad_norm": 0.8951764186258134,
      "learning_rate": 9.881969297715836e-07,
      "loss": 0.0184,
      "step": 6651
    },
    {
      "epoch": 1.51353811149033,
      "grad_norm": 0.7339540735377944,
      "learning_rate": 9.881242100392346e-07,
      "loss": 0.0131,
      "step": 6652
    },
    {
      "epoch": 1.5137656427758817,
      "grad_norm": 0.7233697664885694,
      "learning_rate": 9.880514828852916e-07,
      "loss": 0.0131,
      "step": 6653
    },
    {
      "epoch": 1.5139931740614334,
      "grad_norm": 0.6553569496775881,
      "learning_rate": 9.879787483112413e-07,
      "loss": 0.019,
      "step": 6654
    },
    {
      "epoch": 1.5142207053469852,
      "grad_norm": 1.246700822413907,
      "learning_rate": 9.879060063185702e-07,
      "loss": 0.0255,
      "step": 6655
    },
    {
      "epoch": 1.514448236632537,
      "grad_norm": 0.8464872917857196,
      "learning_rate": 9.878332569087647e-07,
      "loss": 0.0211,
      "step": 6656
    },
    {
      "epoch": 1.5146757679180887,
      "grad_norm": 0.8410971019174226,
      "learning_rate": 9.877605000833122e-07,
      "loss": 0.0163,
      "step": 6657
    },
    {
      "epoch": 1.5149032992036404,
      "grad_norm": 1.256002488770331,
      "learning_rate": 9.876877358436996e-07,
      "loss": 0.0159,
      "step": 6658
    },
    {
      "epoch": 1.5151308304891922,
      "grad_norm": 0.9727260780233405,
      "learning_rate": 9.876149641914135e-07,
      "loss": 0.0224,
      "step": 6659
    },
    {
      "epoch": 1.515358361774744,
      "grad_norm": 1.3058738134780012,
      "learning_rate": 9.875421851279419e-07,
      "loss": 0.0336,
      "step": 6660
    },
    {
      "epoch": 1.5155858930602957,
      "grad_norm": 1.0425717492602375,
      "learning_rate": 9.874693986547717e-07,
      "loss": 0.0177,
      "step": 6661
    },
    {
      "epoch": 1.5158134243458474,
      "grad_norm": 1.6217091401895596,
      "learning_rate": 9.87396604773391e-07,
      "loss": 0.0591,
      "step": 6662
    },
    {
      "epoch": 1.5160409556313992,
      "grad_norm": 0.9444794944797769,
      "learning_rate": 9.873238034852875e-07,
      "loss": 0.0129,
      "step": 6663
    },
    {
      "epoch": 1.516268486916951,
      "grad_norm": 1.5545596261258932,
      "learning_rate": 9.872509947919489e-07,
      "loss": 0.0464,
      "step": 6664
    },
    {
      "epoch": 1.5164960182025027,
      "grad_norm": 1.2645221488362741,
      "learning_rate": 9.871781786948636e-07,
      "loss": 0.019,
      "step": 6665
    },
    {
      "epoch": 1.5167235494880547,
      "grad_norm": 1.1669774401635575,
      "learning_rate": 9.871053551955194e-07,
      "loss": 0.0259,
      "step": 6666
    },
    {
      "epoch": 1.5169510807736064,
      "grad_norm": 1.3872929854862293,
      "learning_rate": 9.87032524295405e-07,
      "loss": 0.0349,
      "step": 6667
    },
    {
      "epoch": 1.5171786120591582,
      "grad_norm": 0.7264415536162637,
      "learning_rate": 9.869596859960087e-07,
      "loss": 0.0104,
      "step": 6668
    },
    {
      "epoch": 1.51740614334471,
      "grad_norm": 1.4722269411341886,
      "learning_rate": 9.868868402988194e-07,
      "loss": 0.0355,
      "step": 6669
    },
    {
      "epoch": 1.5176336746302617,
      "grad_norm": 0.9845541443981577,
      "learning_rate": 9.86813987205326e-07,
      "loss": 0.0319,
      "step": 6670
    },
    {
      "epoch": 1.5178612059158134,
      "grad_norm": 1.5625166554766878,
      "learning_rate": 9.867411267170171e-07,
      "loss": 0.0492,
      "step": 6671
    },
    {
      "epoch": 1.5180887372013652,
      "grad_norm": 1.4742159712737606,
      "learning_rate": 9.866682588353823e-07,
      "loss": 0.0382,
      "step": 6672
    },
    {
      "epoch": 1.518316268486917,
      "grad_norm": 0.9155861801364314,
      "learning_rate": 9.865953835619105e-07,
      "loss": 0.0197,
      "step": 6673
    },
    {
      "epoch": 1.5185437997724687,
      "grad_norm": 0.8446641198873437,
      "learning_rate": 9.865225008980913e-07,
      "loss": 0.0195,
      "step": 6674
    },
    {
      "epoch": 1.5187713310580204,
      "grad_norm": 0.9541679354683429,
      "learning_rate": 9.864496108454142e-07,
      "loss": 0.0304,
      "step": 6675
    },
    {
      "epoch": 1.5189988623435724,
      "grad_norm": 0.7771524650823696,
      "learning_rate": 9.863767134053691e-07,
      "loss": 0.0228,
      "step": 6676
    },
    {
      "epoch": 1.5192263936291241,
      "grad_norm": 0.9791099549816529,
      "learning_rate": 9.86303808579446e-07,
      "loss": 0.0296,
      "step": 6677
    },
    {
      "epoch": 1.519453924914676,
      "grad_norm": 1.2308294541790132,
      "learning_rate": 9.862308963691344e-07,
      "loss": 0.0416,
      "step": 6678
    },
    {
      "epoch": 1.5196814562002277,
      "grad_norm": 1.2569665779197494,
      "learning_rate": 9.86157976775925e-07,
      "loss": 0.0529,
      "step": 6679
    },
    {
      "epoch": 1.5199089874857794,
      "grad_norm": 1.0520527959182502,
      "learning_rate": 9.86085049801308e-07,
      "loss": 0.023,
      "step": 6680
    },
    {
      "epoch": 1.5201365187713312,
      "grad_norm": 0.954887600544443,
      "learning_rate": 9.860121154467738e-07,
      "loss": 0.0151,
      "step": 6681
    },
    {
      "epoch": 1.520364050056883,
      "grad_norm": 1.533955827216088,
      "learning_rate": 9.859391737138132e-07,
      "loss": 0.0302,
      "step": 6682
    },
    {
      "epoch": 1.5205915813424347,
      "grad_norm": 1.0229766183656177,
      "learning_rate": 9.85866224603917e-07,
      "loss": 0.0228,
      "step": 6683
    },
    {
      "epoch": 1.5208191126279864,
      "grad_norm": 1.3237978242768538,
      "learning_rate": 9.85793268118576e-07,
      "loss": 0.0457,
      "step": 6684
    },
    {
      "epoch": 1.5210466439135382,
      "grad_norm": 1.321958954848268,
      "learning_rate": 9.857203042592813e-07,
      "loss": 0.0218,
      "step": 6685
    },
    {
      "epoch": 1.52127417519909,
      "grad_norm": 0.9025338290350718,
      "learning_rate": 9.856473330275243e-07,
      "loss": 0.0153,
      "step": 6686
    },
    {
      "epoch": 1.5215017064846417,
      "grad_norm": 1.5634553678197673,
      "learning_rate": 9.855743544247962e-07,
      "loss": 0.0504,
      "step": 6687
    },
    {
      "epoch": 1.5217292377701934,
      "grad_norm": 1.2829828788158382,
      "learning_rate": 9.855013684525888e-07,
      "loss": 0.026,
      "step": 6688
    },
    {
      "epoch": 1.5219567690557452,
      "grad_norm": 1.03731641686974,
      "learning_rate": 9.854283751123935e-07,
      "loss": 0.0182,
      "step": 6689
    },
    {
      "epoch": 1.522184300341297,
      "grad_norm": 0.983918255174861,
      "learning_rate": 9.853553744057023e-07,
      "loss": 0.0122,
      "step": 6690
    },
    {
      "epoch": 1.5224118316268487,
      "grad_norm": 1.016391471307343,
      "learning_rate": 9.852823663340074e-07,
      "loss": 0.02,
      "step": 6691
    },
    {
      "epoch": 1.5226393629124004,
      "grad_norm": 2.2858262315266753,
      "learning_rate": 9.852093508988006e-07,
      "loss": 0.0629,
      "step": 6692
    },
    {
      "epoch": 1.5228668941979522,
      "grad_norm": 1.0878900203208213,
      "learning_rate": 9.851363281015745e-07,
      "loss": 0.021,
      "step": 6693
    },
    {
      "epoch": 1.523094425483504,
      "grad_norm": 0.9233126923128159,
      "learning_rate": 9.850632979438211e-07,
      "loss": 0.0202,
      "step": 6694
    },
    {
      "epoch": 1.5233219567690557,
      "grad_norm": 1.4496178728052294,
      "learning_rate": 9.849902604270335e-07,
      "loss": 0.0378,
      "step": 6695
    },
    {
      "epoch": 1.5235494880546074,
      "grad_norm": 0.627804195846366,
      "learning_rate": 9.849172155527044e-07,
      "loss": 0.0096,
      "step": 6696
    },
    {
      "epoch": 1.5237770193401592,
      "grad_norm": 1.152818962293583,
      "learning_rate": 9.848441633223266e-07,
      "loss": 0.0178,
      "step": 6697
    },
    {
      "epoch": 1.524004550625711,
      "grad_norm": 1.6797152892325085,
      "learning_rate": 9.847711037373928e-07,
      "loss": 0.0463,
      "step": 6698
    },
    {
      "epoch": 1.5242320819112627,
      "grad_norm": 1.367812533566416,
      "learning_rate": 9.846980367993968e-07,
      "loss": 0.0389,
      "step": 6699
    },
    {
      "epoch": 1.5244596131968144,
      "grad_norm": 1.6336085740093287,
      "learning_rate": 9.846249625098317e-07,
      "loss": 0.0467,
      "step": 6700
    },
    {
      "epoch": 1.5246871444823662,
      "grad_norm": 1.581198385688688,
      "learning_rate": 9.845518808701906e-07,
      "loss": 0.0297,
      "step": 6701
    },
    {
      "epoch": 1.524914675767918,
      "grad_norm": 1.6561397145815502,
      "learning_rate": 9.84478791881968e-07,
      "loss": 0.0458,
      "step": 6702
    },
    {
      "epoch": 1.5251422070534697,
      "grad_norm": 1.1249706240685309,
      "learning_rate": 9.844056955466571e-07,
      "loss": 0.0189,
      "step": 6703
    },
    {
      "epoch": 1.5253697383390215,
      "grad_norm": 0.5097362940197786,
      "learning_rate": 9.84332591865752e-07,
      "loss": 0.0076,
      "step": 6704
    },
    {
      "epoch": 1.5255972696245734,
      "grad_norm": 1.4051486139237497,
      "learning_rate": 9.842594808407467e-07,
      "loss": 0.0363,
      "step": 6705
    },
    {
      "epoch": 1.5258248009101252,
      "grad_norm": 1.7908955623268843,
      "learning_rate": 9.841863624731358e-07,
      "loss": 0.0349,
      "step": 6706
    },
    {
      "epoch": 1.526052332195677,
      "grad_norm": 1.4826315493924698,
      "learning_rate": 9.841132367644133e-07,
      "loss": 0.0277,
      "step": 6707
    },
    {
      "epoch": 1.5262798634812287,
      "grad_norm": 0.971764725095612,
      "learning_rate": 9.840401037160737e-07,
      "loss": 0.0124,
      "step": 6708
    },
    {
      "epoch": 1.5265073947667804,
      "grad_norm": 1.19269631627323,
      "learning_rate": 9.839669633296122e-07,
      "loss": 0.0162,
      "step": 6709
    },
    {
      "epoch": 1.5267349260523322,
      "grad_norm": 1.282535697041421,
      "learning_rate": 9.838938156065236e-07,
      "loss": 0.0314,
      "step": 6710
    },
    {
      "epoch": 1.526962457337884,
      "grad_norm": 0.9152937675240786,
      "learning_rate": 9.838206605483024e-07,
      "loss": 0.0174,
      "step": 6711
    },
    {
      "epoch": 1.5271899886234357,
      "grad_norm": 0.911742361618213,
      "learning_rate": 9.83747498156444e-07,
      "loss": 0.0139,
      "step": 6712
    },
    {
      "epoch": 1.5274175199089874,
      "grad_norm": 1.4421627011208242,
      "learning_rate": 9.836743284324438e-07,
      "loss": 0.0547,
      "step": 6713
    },
    {
      "epoch": 1.5276450511945392,
      "grad_norm": 1.0048756243159007,
      "learning_rate": 9.836011513777975e-07,
      "loss": 0.0269,
      "step": 6714
    },
    {
      "epoch": 1.5278725824800912,
      "grad_norm": 0.9415494134871707,
      "learning_rate": 9.835279669940002e-07,
      "loss": 0.0247,
      "step": 6715
    },
    {
      "epoch": 1.528100113765643,
      "grad_norm": 1.2117322988897647,
      "learning_rate": 9.834547752825477e-07,
      "loss": 0.0136,
      "step": 6716
    },
    {
      "epoch": 1.5283276450511947,
      "grad_norm": 0.8858649899732157,
      "learning_rate": 9.833815762449364e-07,
      "loss": 0.0261,
      "step": 6717
    },
    {
      "epoch": 1.5285551763367464,
      "grad_norm": 1.7990398165637471,
      "learning_rate": 9.833083698826618e-07,
      "loss": 0.0425,
      "step": 6718
    },
    {
      "epoch": 1.5287827076222982,
      "grad_norm": 0.97024081947437,
      "learning_rate": 9.832351561972204e-07,
      "loss": 0.0208,
      "step": 6719
    },
    {
      "epoch": 1.52901023890785,
      "grad_norm": 1.6886694635546007,
      "learning_rate": 9.831619351901083e-07,
      "loss": 0.0345,
      "step": 6720
    },
    {
      "epoch": 1.5292377701934017,
      "grad_norm": 0.9587538553610925,
      "learning_rate": 9.830887068628223e-07,
      "loss": 0.0135,
      "step": 6721
    },
    {
      "epoch": 1.5294653014789534,
      "grad_norm": 1.1164415198948383,
      "learning_rate": 9.830154712168591e-07,
      "loss": 0.0344,
      "step": 6722
    },
    {
      "epoch": 1.5296928327645052,
      "grad_norm": 1.8672159588691146,
      "learning_rate": 9.829422282537152e-07,
      "loss": 0.0527,
      "step": 6723
    },
    {
      "epoch": 1.529920364050057,
      "grad_norm": 0.9012193013849213,
      "learning_rate": 9.828689779748877e-07,
      "loss": 0.0167,
      "step": 6724
    },
    {
      "epoch": 1.5301478953356087,
      "grad_norm": 1.3942555518548752,
      "learning_rate": 9.827957203818737e-07,
      "loss": 0.0302,
      "step": 6725
    },
    {
      "epoch": 1.5303754266211604,
      "grad_norm": 1.4460186732987468,
      "learning_rate": 9.827224554761705e-07,
      "loss": 0.0339,
      "step": 6726
    },
    {
      "epoch": 1.5306029579067122,
      "grad_norm": 1.1521131254229502,
      "learning_rate": 9.826491832592754e-07,
      "loss": 0.0398,
      "step": 6727
    },
    {
      "epoch": 1.530830489192264,
      "grad_norm": 1.2216298964138947,
      "learning_rate": 9.825759037326861e-07,
      "loss": 0.0341,
      "step": 6728
    },
    {
      "epoch": 1.5310580204778157,
      "grad_norm": 0.6507030554372302,
      "learning_rate": 9.825026168979001e-07,
      "loss": 0.01,
      "step": 6729
    },
    {
      "epoch": 1.5312855517633674,
      "grad_norm": 0.7749797892824488,
      "learning_rate": 9.824293227564154e-07,
      "loss": 0.0217,
      "step": 6730
    },
    {
      "epoch": 1.5315130830489192,
      "grad_norm": 1.597607520935028,
      "learning_rate": 9.8235602130973e-07,
      "loss": 0.0294,
      "step": 6731
    },
    {
      "epoch": 1.531740614334471,
      "grad_norm": 0.8423108293730993,
      "learning_rate": 9.822827125593417e-07,
      "loss": 0.013,
      "step": 6732
    },
    {
      "epoch": 1.5319681456200227,
      "grad_norm": 0.7967776857206855,
      "learning_rate": 9.822093965067492e-07,
      "loss": 0.0286,
      "step": 6733
    },
    {
      "epoch": 1.5321956769055745,
      "grad_norm": 1.1464494724303058,
      "learning_rate": 9.821360731534512e-07,
      "loss": 0.0194,
      "step": 6734
    },
    {
      "epoch": 1.5324232081911262,
      "grad_norm": 0.811112555766526,
      "learning_rate": 9.820627425009455e-07,
      "loss": 0.0136,
      "step": 6735
    },
    {
      "epoch": 1.532650739476678,
      "grad_norm": 1.2307368178100238,
      "learning_rate": 9.819894045507315e-07,
      "loss": 0.0355,
      "step": 6736
    },
    {
      "epoch": 1.5328782707622297,
      "grad_norm": 2.5108588990818093,
      "learning_rate": 9.819160593043076e-07,
      "loss": 0.0587,
      "step": 6737
    },
    {
      "epoch": 1.5331058020477815,
      "grad_norm": 1.136986286417985,
      "learning_rate": 9.818427067631733e-07,
      "loss": 0.029,
      "step": 6738
    },
    {
      "epoch": 1.5333333333333332,
      "grad_norm": 1.4896083735623669,
      "learning_rate": 9.817693469288276e-07,
      "loss": 0.0408,
      "step": 6739
    },
    {
      "epoch": 1.533560864618885,
      "grad_norm": 1.1158307587709726,
      "learning_rate": 9.8169597980277e-07,
      "loss": 0.0288,
      "step": 6740
    },
    {
      "epoch": 1.5337883959044367,
      "grad_norm": 1.5678967902284398,
      "learning_rate": 9.816226053864996e-07,
      "loss": 0.0423,
      "step": 6741
    },
    {
      "epoch": 1.5340159271899885,
      "grad_norm": 1.0092054577952496,
      "learning_rate": 9.815492236815163e-07,
      "loss": 0.0257,
      "step": 6742
    },
    {
      "epoch": 1.5342434584755402,
      "grad_norm": 1.6864118266108366,
      "learning_rate": 9.8147583468932e-07,
      "loss": 0.0533,
      "step": 6743
    },
    {
      "epoch": 1.5344709897610922,
      "grad_norm": 0.8285650933249512,
      "learning_rate": 9.814024384114102e-07,
      "loss": 0.0133,
      "step": 6744
    },
    {
      "epoch": 1.534698521046644,
      "grad_norm": 0.7990007997593029,
      "learning_rate": 9.813290348492874e-07,
      "loss": 0.0145,
      "step": 6745
    },
    {
      "epoch": 1.5349260523321957,
      "grad_norm": 1.1452431912291574,
      "learning_rate": 9.812556240044518e-07,
      "loss": 0.0278,
      "step": 6746
    },
    {
      "epoch": 1.5351535836177475,
      "grad_norm": 0.9830841833772611,
      "learning_rate": 9.811822058784038e-07,
      "loss": 0.033,
      "step": 6747
    },
    {
      "epoch": 1.5353811149032992,
      "grad_norm": 1.5315817385842814,
      "learning_rate": 9.811087804726436e-07,
      "loss": 0.0445,
      "step": 6748
    },
    {
      "epoch": 1.535608646188851,
      "grad_norm": 0.8680360060196064,
      "learning_rate": 9.810353477886722e-07,
      "loss": 0.0151,
      "step": 6749
    },
    {
      "epoch": 1.5358361774744027,
      "grad_norm": 0.7341865104680492,
      "learning_rate": 9.809619078279904e-07,
      "loss": 0.0126,
      "step": 6750
    },
    {
      "epoch": 1.5360637087599545,
      "grad_norm": 0.862170127361045,
      "learning_rate": 9.80888460592099e-07,
      "loss": 0.0184,
      "step": 6751
    },
    {
      "epoch": 1.5362912400455062,
      "grad_norm": 1.7419206219624046,
      "learning_rate": 9.808150060824995e-07,
      "loss": 0.057,
      "step": 6752
    },
    {
      "epoch": 1.5365187713310582,
      "grad_norm": 1.571383267693236,
      "learning_rate": 9.807415443006926e-07,
      "loss": 0.0463,
      "step": 6753
    },
    {
      "epoch": 1.53674630261661,
      "grad_norm": 0.6689110433682907,
      "learning_rate": 9.806680752481803e-07,
      "loss": 0.0128,
      "step": 6754
    },
    {
      "epoch": 1.5369738339021617,
      "grad_norm": 1.3775118149606538,
      "learning_rate": 9.805945989264638e-07,
      "loss": 0.0273,
      "step": 6755
    },
    {
      "epoch": 1.5372013651877134,
      "grad_norm": 0.9715191112955526,
      "learning_rate": 9.805211153370448e-07,
      "loss": 0.0214,
      "step": 6756
    },
    {
      "epoch": 1.5374288964732652,
      "grad_norm": 1.0205394119019446,
      "learning_rate": 9.804476244814253e-07,
      "loss": 0.0164,
      "step": 6757
    },
    {
      "epoch": 1.537656427758817,
      "grad_norm": 1.434249017393466,
      "learning_rate": 9.803741263611076e-07,
      "loss": 0.0577,
      "step": 6758
    },
    {
      "epoch": 1.5378839590443687,
      "grad_norm": 1.1853467808691651,
      "learning_rate": 9.803006209775936e-07,
      "loss": 0.031,
      "step": 6759
    },
    {
      "epoch": 1.5381114903299204,
      "grad_norm": 0.9049396065493746,
      "learning_rate": 9.802271083323855e-07,
      "loss": 0.0204,
      "step": 6760
    },
    {
      "epoch": 1.5383390216154722,
      "grad_norm": 1.3834697594871388,
      "learning_rate": 9.801535884269859e-07,
      "loss": 0.0368,
      "step": 6761
    },
    {
      "epoch": 1.538566552901024,
      "grad_norm": 1.2885752750926514,
      "learning_rate": 9.800800612628971e-07,
      "loss": 0.0415,
      "step": 6762
    },
    {
      "epoch": 1.5387940841865757,
      "grad_norm": 1.5134152583249854,
      "learning_rate": 9.800065268416226e-07,
      "loss": 0.0283,
      "step": 6763
    },
    {
      "epoch": 1.5390216154721275,
      "grad_norm": 1.0536222290569848,
      "learning_rate": 9.799329851646643e-07,
      "loss": 0.0197,
      "step": 6764
    },
    {
      "epoch": 1.5392491467576792,
      "grad_norm": 1.4686741972372306,
      "learning_rate": 9.798594362335265e-07,
      "loss": 0.0245,
      "step": 6765
    },
    {
      "epoch": 1.539476678043231,
      "grad_norm": 1.1690678201506683,
      "learning_rate": 9.797858800497112e-07,
      "loss": 0.0157,
      "step": 6766
    },
    {
      "epoch": 1.5397042093287827,
      "grad_norm": 0.8238513238523729,
      "learning_rate": 9.797123166147224e-07,
      "loss": 0.0147,
      "step": 6767
    },
    {
      "epoch": 1.5399317406143345,
      "grad_norm": 1.0558486625656787,
      "learning_rate": 9.796387459300635e-07,
      "loss": 0.024,
      "step": 6768
    },
    {
      "epoch": 1.5401592718998862,
      "grad_norm": 1.1353917596760223,
      "learning_rate": 9.795651679972382e-07,
      "loss": 0.0309,
      "step": 6769
    },
    {
      "epoch": 1.540386803185438,
      "grad_norm": 0.8517135065926781,
      "learning_rate": 9.7949158281775e-07,
      "loss": 0.0184,
      "step": 6770
    },
    {
      "epoch": 1.5406143344709897,
      "grad_norm": 1.4294422070332768,
      "learning_rate": 9.794179903931035e-07,
      "loss": 0.0277,
      "step": 6771
    },
    {
      "epoch": 1.5408418657565415,
      "grad_norm": 1.2869738269014956,
      "learning_rate": 9.79344390724802e-07,
      "loss": 0.0257,
      "step": 6772
    },
    {
      "epoch": 1.5410693970420932,
      "grad_norm": 0.8960870538958028,
      "learning_rate": 9.7927078381435e-07,
      "loss": 0.0256,
      "step": 6773
    },
    {
      "epoch": 1.541296928327645,
      "grad_norm": 0.8475858169518586,
      "learning_rate": 9.791971696632523e-07,
      "loss": 0.0181,
      "step": 6774
    },
    {
      "epoch": 1.5415244596131967,
      "grad_norm": 1.44665286178453,
      "learning_rate": 9.79123548273013e-07,
      "loss": 0.0419,
      "step": 6775
    },
    {
      "epoch": 1.5417519908987485,
      "grad_norm": 0.7037598146385168,
      "learning_rate": 9.79049919645137e-07,
      "loss": 0.0119,
      "step": 6776
    },
    {
      "epoch": 1.5419795221843002,
      "grad_norm": 0.9126315131039067,
      "learning_rate": 9.78976283781129e-07,
      "loss": 0.0189,
      "step": 6777
    },
    {
      "epoch": 1.542207053469852,
      "grad_norm": 1.204330528834964,
      "learning_rate": 9.78902640682494e-07,
      "loss": 0.0226,
      "step": 6778
    },
    {
      "epoch": 1.5424345847554037,
      "grad_norm": 1.4964522539113427,
      "learning_rate": 9.788289903507373e-07,
      "loss": 0.0396,
      "step": 6779
    },
    {
      "epoch": 1.5426621160409555,
      "grad_norm": 1.5964692402160778,
      "learning_rate": 9.787553327873637e-07,
      "loss": 0.0481,
      "step": 6780
    },
    {
      "epoch": 1.5428896473265072,
      "grad_norm": 1.2787948266252873,
      "learning_rate": 9.786816679938794e-07,
      "loss": 0.0313,
      "step": 6781
    },
    {
      "epoch": 1.543117178612059,
      "grad_norm": 0.9053360200297276,
      "learning_rate": 9.786079959717892e-07,
      "loss": 0.0181,
      "step": 6782
    },
    {
      "epoch": 1.543344709897611,
      "grad_norm": 1.2025054194441325,
      "learning_rate": 9.785343167225994e-07,
      "loss": 0.025,
      "step": 6783
    },
    {
      "epoch": 1.5435722411831627,
      "grad_norm": 1.768093481908682,
      "learning_rate": 9.784606302478155e-07,
      "loss": 0.086,
      "step": 6784
    },
    {
      "epoch": 1.5437997724687145,
      "grad_norm": 0.9078465345048391,
      "learning_rate": 9.783869365489437e-07,
      "loss": 0.0115,
      "step": 6785
    },
    {
      "epoch": 1.5440273037542662,
      "grad_norm": 1.1497977464117497,
      "learning_rate": 9.783132356274901e-07,
      "loss": 0.0216,
      "step": 6786
    },
    {
      "epoch": 1.544254835039818,
      "grad_norm": 0.5313589583578873,
      "learning_rate": 9.78239527484961e-07,
      "loss": 0.0087,
      "step": 6787
    },
    {
      "epoch": 1.5444823663253697,
      "grad_norm": 0.9748789217096225,
      "learning_rate": 9.781658121228628e-07,
      "loss": 0.0318,
      "step": 6788
    },
    {
      "epoch": 1.5447098976109215,
      "grad_norm": 2.4694914461680844,
      "learning_rate": 9.780920895427025e-07,
      "loss": 0.05,
      "step": 6789
    },
    {
      "epoch": 1.5449374288964732,
      "grad_norm": 1.30619233097482,
      "learning_rate": 9.780183597459864e-07,
      "loss": 0.0352,
      "step": 6790
    },
    {
      "epoch": 1.545164960182025,
      "grad_norm": 1.8138249647505036,
      "learning_rate": 9.779446227342216e-07,
      "loss": 0.0507,
      "step": 6791
    },
    {
      "epoch": 1.545392491467577,
      "grad_norm": 1.2364189313694403,
      "learning_rate": 9.77870878508915e-07,
      "loss": 0.0194,
      "step": 6792
    },
    {
      "epoch": 1.5456200227531287,
      "grad_norm": 0.9954733274363685,
      "learning_rate": 9.77797127071574e-07,
      "loss": 0.0228,
      "step": 6793
    },
    {
      "epoch": 1.5458475540386805,
      "grad_norm": 1.1973963103082081,
      "learning_rate": 9.777233684237056e-07,
      "loss": 0.0259,
      "step": 6794
    },
    {
      "epoch": 1.5460750853242322,
      "grad_norm": 0.9207738182775058,
      "learning_rate": 9.776496025668174e-07,
      "loss": 0.0132,
      "step": 6795
    },
    {
      "epoch": 1.546302616609784,
      "grad_norm": 0.7174193444984276,
      "learning_rate": 9.775758295024177e-07,
      "loss": 0.0289,
      "step": 6796
    },
    {
      "epoch": 1.5465301478953357,
      "grad_norm": 1.2241595374734087,
      "learning_rate": 9.775020492320136e-07,
      "loss": 0.0378,
      "step": 6797
    },
    {
      "epoch": 1.5467576791808875,
      "grad_norm": 1.8511371140948074,
      "learning_rate": 9.774282617571129e-07,
      "loss": 0.032,
      "step": 6798
    },
    {
      "epoch": 1.5469852104664392,
      "grad_norm": 1.0367978069734165,
      "learning_rate": 9.773544670792243e-07,
      "loss": 0.0408,
      "step": 6799
    },
    {
      "epoch": 1.547212741751991,
      "grad_norm": 1.227189360276986,
      "learning_rate": 9.772806651998555e-07,
      "loss": 0.0256,
      "step": 6800
    },
    {
      "epoch": 1.5474402730375427,
      "grad_norm": 0.8034203266831912,
      "learning_rate": 9.772068561205152e-07,
      "loss": 0.0243,
      "step": 6801
    },
    {
      "epoch": 1.5476678043230945,
      "grad_norm": 1.053114722831617,
      "learning_rate": 9.771330398427118e-07,
      "loss": 0.0314,
      "step": 6802
    },
    {
      "epoch": 1.5478953356086462,
      "grad_norm": 0.657404541342885,
      "learning_rate": 9.770592163679539e-07,
      "loss": 0.0105,
      "step": 6803
    },
    {
      "epoch": 1.548122866894198,
      "grad_norm": 1.050779540517071,
      "learning_rate": 9.769853856977503e-07,
      "loss": 0.0245,
      "step": 6804
    },
    {
      "epoch": 1.5483503981797497,
      "grad_norm": 0.9962720723555235,
      "learning_rate": 9.769115478336102e-07,
      "loss": 0.0253,
      "step": 6805
    },
    {
      "epoch": 1.5485779294653015,
      "grad_norm": 1.8462344743111663,
      "learning_rate": 9.768377027770427e-07,
      "loss": 0.0327,
      "step": 6806
    },
    {
      "epoch": 1.5488054607508532,
      "grad_norm": 1.0035831512384688,
      "learning_rate": 9.767638505295566e-07,
      "loss": 0.0331,
      "step": 6807
    },
    {
      "epoch": 1.549032992036405,
      "grad_norm": 0.8002363836105062,
      "learning_rate": 9.766899910926617e-07,
      "loss": 0.0219,
      "step": 6808
    },
    {
      "epoch": 1.5492605233219567,
      "grad_norm": 1.069327907986938,
      "learning_rate": 9.766161244678675e-07,
      "loss": 0.0482,
      "step": 6809
    },
    {
      "epoch": 1.5494880546075085,
      "grad_norm": 0.7883539436623613,
      "learning_rate": 9.765422506566837e-07,
      "loss": 0.0127,
      "step": 6810
    },
    {
      "epoch": 1.5497155858930602,
      "grad_norm": 0.7269884120486411,
      "learning_rate": 9.7646836966062e-07,
      "loss": 0.0127,
      "step": 6811
    },
    {
      "epoch": 1.549943117178612,
      "grad_norm": 0.9990408828346841,
      "learning_rate": 9.763944814811866e-07,
      "loss": 0.0277,
      "step": 6812
    },
    {
      "epoch": 1.5501706484641637,
      "grad_norm": 0.8768522206549924,
      "learning_rate": 9.763205861198935e-07,
      "loss": 0.0218,
      "step": 6813
    },
    {
      "epoch": 1.5503981797497155,
      "grad_norm": 1.330953541460739,
      "learning_rate": 9.76246683578251e-07,
      "loss": 0.0279,
      "step": 6814
    },
    {
      "epoch": 1.5506257110352673,
      "grad_norm": 0.7770272018475644,
      "learning_rate": 9.761727738577698e-07,
      "loss": 0.0228,
      "step": 6815
    },
    {
      "epoch": 1.550853242320819,
      "grad_norm": 1.8387771356381233,
      "learning_rate": 9.760988569599602e-07,
      "loss": 0.0673,
      "step": 6816
    },
    {
      "epoch": 1.5510807736063708,
      "grad_norm": 1.2266606216673168,
      "learning_rate": 9.760249328863328e-07,
      "loss": 0.039,
      "step": 6817
    },
    {
      "epoch": 1.5513083048919225,
      "grad_norm": 1.0188689397210966,
      "learning_rate": 9.759510016383987e-07,
      "loss": 0.0207,
      "step": 6818
    },
    {
      "epoch": 1.5515358361774743,
      "grad_norm": 1.0579016282881235,
      "learning_rate": 9.758770632176688e-07,
      "loss": 0.0158,
      "step": 6819
    },
    {
      "epoch": 1.551763367463026,
      "grad_norm": 0.9612658877602658,
      "learning_rate": 9.758031176256543e-07,
      "loss": 0.0168,
      "step": 6820
    },
    {
      "epoch": 1.5519908987485778,
      "grad_norm": 2.3193682962515774,
      "learning_rate": 9.757291648638666e-07,
      "loss": 0.0746,
      "step": 6821
    },
    {
      "epoch": 1.5522184300341297,
      "grad_norm": 1.096008882913452,
      "learning_rate": 9.756552049338174e-07,
      "loss": 0.0177,
      "step": 6822
    },
    {
      "epoch": 1.5524459613196815,
      "grad_norm": 1.0739238458685665,
      "learning_rate": 9.755812378370177e-07,
      "loss": 0.0242,
      "step": 6823
    },
    {
      "epoch": 1.5526734926052332,
      "grad_norm": 0.9537760996671094,
      "learning_rate": 9.755072635749795e-07,
      "loss": 0.0256,
      "step": 6824
    },
    {
      "epoch": 1.552901023890785,
      "grad_norm": 0.8524455774216739,
      "learning_rate": 9.754332821492148e-07,
      "loss": 0.012,
      "step": 6825
    },
    {
      "epoch": 1.5531285551763367,
      "grad_norm": 1.459998149204111,
      "learning_rate": 9.753592935612358e-07,
      "loss": 0.0431,
      "step": 6826
    },
    {
      "epoch": 1.5533560864618885,
      "grad_norm": 0.9652354910786201,
      "learning_rate": 9.752852978125544e-07,
      "loss": 0.022,
      "step": 6827
    },
    {
      "epoch": 1.5535836177474402,
      "grad_norm": 0.6879353796285916,
      "learning_rate": 9.75211294904683e-07,
      "loss": 0.0135,
      "step": 6828
    },
    {
      "epoch": 1.553811149032992,
      "grad_norm": 1.4518274964176225,
      "learning_rate": 9.75137284839134e-07,
      "loss": 0.0377,
      "step": 6829
    },
    {
      "epoch": 1.5540386803185438,
      "grad_norm": 1.6705449868902968,
      "learning_rate": 9.750632676174201e-07,
      "loss": 0.0476,
      "step": 6830
    },
    {
      "epoch": 1.5542662116040957,
      "grad_norm": 1.2689213544556226,
      "learning_rate": 9.749892432410544e-07,
      "loss": 0.0282,
      "step": 6831
    },
    {
      "epoch": 1.5544937428896475,
      "grad_norm": 1.8099847360827157,
      "learning_rate": 9.749152117115494e-07,
      "loss": 0.0356,
      "step": 6832
    },
    {
      "epoch": 1.5547212741751992,
      "grad_norm": 1.5405555526689485,
      "learning_rate": 9.748411730304184e-07,
      "loss": 0.0444,
      "step": 6833
    },
    {
      "epoch": 1.554948805460751,
      "grad_norm": 0.8694412690578796,
      "learning_rate": 9.747671271991746e-07,
      "loss": 0.013,
      "step": 6834
    },
    {
      "epoch": 1.5551763367463027,
      "grad_norm": 1.02126526467446,
      "learning_rate": 9.746930742193307e-07,
      "loss": 0.0239,
      "step": 6835
    },
    {
      "epoch": 1.5554038680318545,
      "grad_norm": 0.7213813598412102,
      "learning_rate": 9.746190140924014e-07,
      "loss": 0.0222,
      "step": 6836
    },
    {
      "epoch": 1.5556313993174062,
      "grad_norm": 1.1742233662911234,
      "learning_rate": 9.745449468198997e-07,
      "loss": 0.0302,
      "step": 6837
    },
    {
      "epoch": 1.555858930602958,
      "grad_norm": 1.0483098687522778,
      "learning_rate": 9.744708724033393e-07,
      "loss": 0.0197,
      "step": 6838
    },
    {
      "epoch": 1.5560864618885097,
      "grad_norm": 1.2683705572721342,
      "learning_rate": 9.743967908442343e-07,
      "loss": 0.0414,
      "step": 6839
    },
    {
      "epoch": 1.5563139931740615,
      "grad_norm": 0.8649452535857219,
      "learning_rate": 9.743227021440988e-07,
      "loss": 0.0221,
      "step": 6840
    },
    {
      "epoch": 1.5565415244596132,
      "grad_norm": 1.1327479895451653,
      "learning_rate": 9.74248606304447e-07,
      "loss": 0.0233,
      "step": 6841
    },
    {
      "epoch": 1.556769055745165,
      "grad_norm": 0.853522919213304,
      "learning_rate": 9.741745033267932e-07,
      "loss": 0.0171,
      "step": 6842
    },
    {
      "epoch": 1.5569965870307167,
      "grad_norm": 1.4600706299872448,
      "learning_rate": 9.741003932126522e-07,
      "loss": 0.0273,
      "step": 6843
    },
    {
      "epoch": 1.5572241183162685,
      "grad_norm": 1.9026227073360096,
      "learning_rate": 9.740262759635386e-07,
      "loss": 0.053,
      "step": 6844
    },
    {
      "epoch": 1.5574516496018203,
      "grad_norm": 0.9978504859142656,
      "learning_rate": 9.739521515809669e-07,
      "loss": 0.0261,
      "step": 6845
    },
    {
      "epoch": 1.557679180887372,
      "grad_norm": 1.200148401809416,
      "learning_rate": 9.738780200664525e-07,
      "loss": 0.0254,
      "step": 6846
    },
    {
      "epoch": 1.5579067121729238,
      "grad_norm": 1.394694694968866,
      "learning_rate": 9.738038814215102e-07,
      "loss": 0.0246,
      "step": 6847
    },
    {
      "epoch": 1.5581342434584755,
      "grad_norm": 1.2354636254991684,
      "learning_rate": 9.737297356476554e-07,
      "loss": 0.0234,
      "step": 6848
    },
    {
      "epoch": 1.5583617747440273,
      "grad_norm": 1.27065517757966,
      "learning_rate": 9.736555827464034e-07,
      "loss": 0.0497,
      "step": 6849
    },
    {
      "epoch": 1.558589306029579,
      "grad_norm": 0.5183623239377243,
      "learning_rate": 9.7358142271927e-07,
      "loss": 0.0114,
      "step": 6850
    },
    {
      "epoch": 1.5588168373151308,
      "grad_norm": 0.8794614624216165,
      "learning_rate": 9.735072555677705e-07,
      "loss": 0.0202,
      "step": 6851
    },
    {
      "epoch": 1.5590443686006825,
      "grad_norm": 1.621272386650821,
      "learning_rate": 9.73433081293421e-07,
      "loss": 0.0386,
      "step": 6852
    },
    {
      "epoch": 1.5592718998862343,
      "grad_norm": 1.1922606611608835,
      "learning_rate": 9.733588998977376e-07,
      "loss": 0.0222,
      "step": 6853
    },
    {
      "epoch": 1.559499431171786,
      "grad_norm": 1.1849209727802208,
      "learning_rate": 9.73284711382236e-07,
      "loss": 0.0201,
      "step": 6854
    },
    {
      "epoch": 1.5597269624573378,
      "grad_norm": 1.6673549617864156,
      "learning_rate": 9.732105157484332e-07,
      "loss": 0.056,
      "step": 6855
    },
    {
      "epoch": 1.5599544937428895,
      "grad_norm": 1.3125965381213744,
      "learning_rate": 9.731363129978447e-07,
      "loss": 0.0228,
      "step": 6856
    },
    {
      "epoch": 1.5601820250284413,
      "grad_norm": 1.6685078337354073,
      "learning_rate": 9.730621031319878e-07,
      "loss": 0.0288,
      "step": 6857
    },
    {
      "epoch": 1.560409556313993,
      "grad_norm": 1.2959301226520625,
      "learning_rate": 9.729878861523788e-07,
      "loss": 0.0274,
      "step": 6858
    },
    {
      "epoch": 1.5606370875995448,
      "grad_norm": 0.8526003882872059,
      "learning_rate": 9.729136620605347e-07,
      "loss": 0.0162,
      "step": 6859
    },
    {
      "epoch": 1.5608646188850968,
      "grad_norm": 4.875759566271044,
      "learning_rate": 9.728394308579727e-07,
      "loss": 0.0988,
      "step": 6860
    },
    {
      "epoch": 1.5610921501706485,
      "grad_norm": 0.9454552747509546,
      "learning_rate": 9.727651925462098e-07,
      "loss": 0.0238,
      "step": 6861
    },
    {
      "epoch": 1.5613196814562003,
      "grad_norm": 0.8040446441336326,
      "learning_rate": 9.726909471267632e-07,
      "loss": 0.0169,
      "step": 6862
    },
    {
      "epoch": 1.561547212741752,
      "grad_norm": 0.975335759529535,
      "learning_rate": 9.726166946011503e-07,
      "loss": 0.0208,
      "step": 6863
    },
    {
      "epoch": 1.5617747440273038,
      "grad_norm": 0.995135796770694,
      "learning_rate": 9.72542434970889e-07,
      "loss": 0.0316,
      "step": 6864
    },
    {
      "epoch": 1.5620022753128555,
      "grad_norm": 1.531607130104655,
      "learning_rate": 9.724681682374965e-07,
      "loss": 0.0526,
      "step": 6865
    },
    {
      "epoch": 1.5622298065984073,
      "grad_norm": 0.980914346176978,
      "learning_rate": 9.723938944024913e-07,
      "loss": 0.016,
      "step": 6866
    },
    {
      "epoch": 1.562457337883959,
      "grad_norm": 2.20124488164597,
      "learning_rate": 9.72319613467391e-07,
      "loss": 0.0452,
      "step": 6867
    },
    {
      "epoch": 1.5626848691695108,
      "grad_norm": 1.4818225810862038,
      "learning_rate": 9.722453254337139e-07,
      "loss": 0.0406,
      "step": 6868
    },
    {
      "epoch": 1.5629124004550625,
      "grad_norm": 1.2655098223089418,
      "learning_rate": 9.721710303029783e-07,
      "loss": 0.0235,
      "step": 6869
    },
    {
      "epoch": 1.5631399317406145,
      "grad_norm": 1.1710684519730319,
      "learning_rate": 9.720967280767026e-07,
      "loss": 0.0208,
      "step": 6870
    },
    {
      "epoch": 1.5633674630261662,
      "grad_norm": 1.1745759284941633,
      "learning_rate": 9.720224187564057e-07,
      "loss": 0.0244,
      "step": 6871
    },
    {
      "epoch": 1.563594994311718,
      "grad_norm": 0.8708806170643246,
      "learning_rate": 9.719481023436059e-07,
      "loss": 0.0218,
      "step": 6872
    },
    {
      "epoch": 1.5638225255972698,
      "grad_norm": 1.2108273614761154,
      "learning_rate": 9.718737788398223e-07,
      "loss": 0.0329,
      "step": 6873
    },
    {
      "epoch": 1.5640500568828215,
      "grad_norm": 2.0062378591587646,
      "learning_rate": 9.71799448246574e-07,
      "loss": 0.0602,
      "step": 6874
    },
    {
      "epoch": 1.5642775881683733,
      "grad_norm": 0.6282582638053863,
      "learning_rate": 9.717251105653799e-07,
      "loss": 0.0078,
      "step": 6875
    },
    {
      "epoch": 1.564505119453925,
      "grad_norm": 1.8411837776806999,
      "learning_rate": 9.716507657977597e-07,
      "loss": 0.0311,
      "step": 6876
    },
    {
      "epoch": 1.5647326507394768,
      "grad_norm": 1.5565448824175396,
      "learning_rate": 9.715764139452327e-07,
      "loss": 0.0324,
      "step": 6877
    },
    {
      "epoch": 1.5649601820250285,
      "grad_norm": 1.1011446306121413,
      "learning_rate": 9.715020550093185e-07,
      "loss": 0.0206,
      "step": 6878
    },
    {
      "epoch": 1.5651877133105803,
      "grad_norm": 1.4189312012820676,
      "learning_rate": 9.71427688991537e-07,
      "loss": 0.0366,
      "step": 6879
    },
    {
      "epoch": 1.565415244596132,
      "grad_norm": 1.1440113593957346,
      "learning_rate": 9.713533158934079e-07,
      "loss": 0.0163,
      "step": 6880
    },
    {
      "epoch": 1.5656427758816838,
      "grad_norm": 0.8339533986631594,
      "learning_rate": 9.712789357164512e-07,
      "loss": 0.0177,
      "step": 6881
    },
    {
      "epoch": 1.5658703071672355,
      "grad_norm": 1.3924735109528363,
      "learning_rate": 9.712045484621874e-07,
      "loss": 0.0387,
      "step": 6882
    },
    {
      "epoch": 1.5660978384527873,
      "grad_norm": 0.9259666291165694,
      "learning_rate": 9.711301541321365e-07,
      "loss": 0.0177,
      "step": 6883
    },
    {
      "epoch": 1.566325369738339,
      "grad_norm": 1.3135763130095726,
      "learning_rate": 9.710557527278195e-07,
      "loss": 0.0305,
      "step": 6884
    },
    {
      "epoch": 1.5665529010238908,
      "grad_norm": 1.1297719627959584,
      "learning_rate": 9.709813442507565e-07,
      "loss": 0.0247,
      "step": 6885
    },
    {
      "epoch": 1.5667804323094425,
      "grad_norm": 1.4600258713674805,
      "learning_rate": 9.709069287024684e-07,
      "loss": 0.0411,
      "step": 6886
    },
    {
      "epoch": 1.5670079635949943,
      "grad_norm": 1.0383415502711744,
      "learning_rate": 9.708325060844761e-07,
      "loss": 0.0202,
      "step": 6887
    },
    {
      "epoch": 1.567235494880546,
      "grad_norm": 1.6819369088665528,
      "learning_rate": 9.707580763983008e-07,
      "loss": 0.0348,
      "step": 6888
    },
    {
      "epoch": 1.5674630261660978,
      "grad_norm": 1.1909010495383987,
      "learning_rate": 9.706836396454638e-07,
      "loss": 0.0226,
      "step": 6889
    },
    {
      "epoch": 1.5676905574516495,
      "grad_norm": 1.1133637552881577,
      "learning_rate": 9.70609195827486e-07,
      "loss": 0.0187,
      "step": 6890
    },
    {
      "epoch": 1.5679180887372013,
      "grad_norm": 1.8834546240381704,
      "learning_rate": 9.705347449458896e-07,
      "loss": 0.0615,
      "step": 6891
    },
    {
      "epoch": 1.568145620022753,
      "grad_norm": 1.1406718003995044,
      "learning_rate": 9.704602870021954e-07,
      "loss": 0.0306,
      "step": 6892
    },
    {
      "epoch": 1.5683731513083048,
      "grad_norm": 1.0476596873190438,
      "learning_rate": 9.70385821997926e-07,
      "loss": 0.0257,
      "step": 6893
    },
    {
      "epoch": 1.5686006825938565,
      "grad_norm": 1.505185232649498,
      "learning_rate": 9.703113499346026e-07,
      "loss": 0.0329,
      "step": 6894
    },
    {
      "epoch": 1.5688282138794083,
      "grad_norm": 1.3766493287200758,
      "learning_rate": 9.70236870813748e-07,
      "loss": 0.0303,
      "step": 6895
    },
    {
      "epoch": 1.56905574516496,
      "grad_norm": 0.8517840379820023,
      "learning_rate": 9.701623846368836e-07,
      "loss": 0.0135,
      "step": 6896
    },
    {
      "epoch": 1.5692832764505118,
      "grad_norm": 1.4103036090074965,
      "learning_rate": 9.700878914055325e-07,
      "loss": 0.0531,
      "step": 6897
    },
    {
      "epoch": 1.5695108077360636,
      "grad_norm": 3.6980656854925904,
      "learning_rate": 9.700133911212168e-07,
      "loss": 0.0921,
      "step": 6898
    },
    {
      "epoch": 1.5697383390216155,
      "grad_norm": 0.6914093108994455,
      "learning_rate": 9.699388837854593e-07,
      "loss": 0.0099,
      "step": 6899
    },
    {
      "epoch": 1.5699658703071673,
      "grad_norm": 0.9100695130578376,
      "learning_rate": 9.698643693997827e-07,
      "loss": 0.0245,
      "step": 6900
    },
    {
      "epoch": 1.570193401592719,
      "grad_norm": 0.8627006881738912,
      "learning_rate": 9.697898479657098e-07,
      "loss": 0.0146,
      "step": 6901
    },
    {
      "epoch": 1.5704209328782708,
      "grad_norm": 1.4110615783342308,
      "learning_rate": 9.697153194847641e-07,
      "loss": 0.0236,
      "step": 6902
    },
    {
      "epoch": 1.5706484641638225,
      "grad_norm": 1.1347870516723162,
      "learning_rate": 9.696407839584684e-07,
      "loss": 0.0336,
      "step": 6903
    },
    {
      "epoch": 1.5708759954493743,
      "grad_norm": 1.2538699432855616,
      "learning_rate": 9.695662413883466e-07,
      "loss": 0.0172,
      "step": 6904
    },
    {
      "epoch": 1.571103526734926,
      "grad_norm": 0.8716398094298144,
      "learning_rate": 9.694916917759218e-07,
      "loss": 0.0186,
      "step": 6905
    },
    {
      "epoch": 1.5713310580204778,
      "grad_norm": 0.845944219449997,
      "learning_rate": 9.694171351227175e-07,
      "loss": 0.0172,
      "step": 6906
    },
    {
      "epoch": 1.5715585893060295,
      "grad_norm": 1.5123986101385158,
      "learning_rate": 9.693425714302577e-07,
      "loss": 0.025,
      "step": 6907
    },
    {
      "epoch": 1.5717861205915813,
      "grad_norm": 1.0592873353102206,
      "learning_rate": 9.692680007000663e-07,
      "loss": 0.0182,
      "step": 6908
    },
    {
      "epoch": 1.5720136518771333,
      "grad_norm": 1.5343957838140057,
      "learning_rate": 9.691934229336677e-07,
      "loss": 0.0301,
      "step": 6909
    },
    {
      "epoch": 1.572241183162685,
      "grad_norm": 1.5525566912067996,
      "learning_rate": 9.69118838132586e-07,
      "loss": 0.0298,
      "step": 6910
    },
    {
      "epoch": 1.5724687144482368,
      "grad_norm": 0.9091899940543618,
      "learning_rate": 9.69044246298345e-07,
      "loss": 0.0364,
      "step": 6911
    },
    {
      "epoch": 1.5726962457337885,
      "grad_norm": 0.7851322451427193,
      "learning_rate": 9.689696474324703e-07,
      "loss": 0.0187,
      "step": 6912
    },
    {
      "epoch": 1.5729237770193403,
      "grad_norm": 0.9342893603380591,
      "learning_rate": 9.688950415364855e-07,
      "loss": 0.0376,
      "step": 6913
    },
    {
      "epoch": 1.573151308304892,
      "grad_norm": 2.3291078950763624,
      "learning_rate": 9.68820428611916e-07,
      "loss": 0.0834,
      "step": 6914
    },
    {
      "epoch": 1.5733788395904438,
      "grad_norm": 1.3019341126571247,
      "learning_rate": 9.687458086602866e-07,
      "loss": 0.0308,
      "step": 6915
    },
    {
      "epoch": 1.5736063708759955,
      "grad_norm": 0.772079275849239,
      "learning_rate": 9.686711816831226e-07,
      "loss": 0.015,
      "step": 6916
    },
    {
      "epoch": 1.5738339021615473,
      "grad_norm": 0.9630973601051392,
      "learning_rate": 9.68596547681949e-07,
      "loss": 0.0279,
      "step": 6917
    },
    {
      "epoch": 1.574061433447099,
      "grad_norm": 1.734908432077837,
      "learning_rate": 9.68521906658291e-07,
      "loss": 0.0595,
      "step": 6918
    },
    {
      "epoch": 1.5742889647326508,
      "grad_norm": 1.4035571745166036,
      "learning_rate": 9.684472586136745e-07,
      "loss": 0.0214,
      "step": 6919
    },
    {
      "epoch": 1.5745164960182025,
      "grad_norm": 1.3254699748716203,
      "learning_rate": 9.68372603549625e-07,
      "loss": 0.0362,
      "step": 6920
    },
    {
      "epoch": 1.5747440273037543,
      "grad_norm": 1.4628284794894173,
      "learning_rate": 9.682979414676682e-07,
      "loss": 0.0256,
      "step": 6921
    },
    {
      "epoch": 1.574971558589306,
      "grad_norm": 1.3653689349920086,
      "learning_rate": 9.682232723693305e-07,
      "loss": 0.0258,
      "step": 6922
    },
    {
      "epoch": 1.5751990898748578,
      "grad_norm": 0.7209251346481367,
      "learning_rate": 9.681485962561377e-07,
      "loss": 0.0174,
      "step": 6923
    },
    {
      "epoch": 1.5754266211604095,
      "grad_norm": 1.0955934147353268,
      "learning_rate": 9.680739131296158e-07,
      "loss": 0.0208,
      "step": 6924
    },
    {
      "epoch": 1.5756541524459613,
      "grad_norm": 1.2439741485652382,
      "learning_rate": 9.679992229912914e-07,
      "loss": 0.0305,
      "step": 6925
    },
    {
      "epoch": 1.575881683731513,
      "grad_norm": 1.4366682730583082,
      "learning_rate": 9.67924525842691e-07,
      "loss": 0.0306,
      "step": 6926
    },
    {
      "epoch": 1.5761092150170648,
      "grad_norm": 1.3379090050907754,
      "learning_rate": 9.678498216853416e-07,
      "loss": 0.0411,
      "step": 6927
    },
    {
      "epoch": 1.5763367463026166,
      "grad_norm": 1.1244782762370413,
      "learning_rate": 9.677751105207696e-07,
      "loss": 0.0267,
      "step": 6928
    },
    {
      "epoch": 1.5765642775881683,
      "grad_norm": 1.0334857986896164,
      "learning_rate": 9.677003923505025e-07,
      "loss": 0.0208,
      "step": 6929
    },
    {
      "epoch": 1.57679180887372,
      "grad_norm": 0.6704299998949605,
      "learning_rate": 9.676256671760665e-07,
      "loss": 0.0068,
      "step": 6930
    },
    {
      "epoch": 1.5770193401592718,
      "grad_norm": 0.9533956478964293,
      "learning_rate": 9.675509349989896e-07,
      "loss": 0.0213,
      "step": 6931
    },
    {
      "epoch": 1.5772468714448236,
      "grad_norm": 0.9244996451919156,
      "learning_rate": 9.674761958207986e-07,
      "loss": 0.018,
      "step": 6932
    },
    {
      "epoch": 1.5774744027303753,
      "grad_norm": 0.6496056903137244,
      "learning_rate": 9.674014496430218e-07,
      "loss": 0.0098,
      "step": 6933
    },
    {
      "epoch": 1.577701934015927,
      "grad_norm": 0.9270304301068275,
      "learning_rate": 9.673266964671862e-07,
      "loss": 0.017,
      "step": 6934
    },
    {
      "epoch": 1.5779294653014788,
      "grad_norm": 2.313850498927015,
      "learning_rate": 9.6725193629482e-07,
      "loss": 0.0177,
      "step": 6935
    },
    {
      "epoch": 1.5781569965870306,
      "grad_norm": 0.8972440977156996,
      "learning_rate": 9.671771691274508e-07,
      "loss": 0.0155,
      "step": 6936
    },
    {
      "epoch": 1.5783845278725823,
      "grad_norm": 0.9084538782830909,
      "learning_rate": 9.671023949666073e-07,
      "loss": 0.0144,
      "step": 6937
    },
    {
      "epoch": 1.5786120591581343,
      "grad_norm": 0.8179863989404683,
      "learning_rate": 9.67027613813817e-07,
      "loss": 0.0205,
      "step": 6938
    },
    {
      "epoch": 1.578839590443686,
      "grad_norm": 1.4846267938383064,
      "learning_rate": 9.66952825670609e-07,
      "loss": 0.044,
      "step": 6939
    },
    {
      "epoch": 1.5790671217292378,
      "grad_norm": 1.18034980748661,
      "learning_rate": 9.66878030538511e-07,
      "loss": 0.0326,
      "step": 6940
    },
    {
      "epoch": 1.5792946530147896,
      "grad_norm": 1.256043529456657,
      "learning_rate": 9.668032284190529e-07,
      "loss": 0.0165,
      "step": 6941
    },
    {
      "epoch": 1.5795221843003413,
      "grad_norm": 0.8899435581756949,
      "learning_rate": 9.667284193137622e-07,
      "loss": 0.0208,
      "step": 6942
    },
    {
      "epoch": 1.579749715585893,
      "grad_norm": 1.0826065036291013,
      "learning_rate": 9.666536032241687e-07,
      "loss": 0.0264,
      "step": 6943
    },
    {
      "epoch": 1.5799772468714448,
      "grad_norm": 1.0986489683562306,
      "learning_rate": 9.66578780151801e-07,
      "loss": 0.0357,
      "step": 6944
    },
    {
      "epoch": 1.5802047781569966,
      "grad_norm": 0.7910773749797155,
      "learning_rate": 9.66503950098189e-07,
      "loss": 0.0124,
      "step": 6945
    },
    {
      "epoch": 1.5804323094425483,
      "grad_norm": 1.2608845947842933,
      "learning_rate": 9.664291130648616e-07,
      "loss": 0.0219,
      "step": 6946
    },
    {
      "epoch": 1.5806598407281,
      "grad_norm": 0.896154253642754,
      "learning_rate": 9.663542690533485e-07,
      "loss": 0.0123,
      "step": 6947
    },
    {
      "epoch": 1.580887372013652,
      "grad_norm": 0.9731058492147535,
      "learning_rate": 9.66279418065179e-07,
      "loss": 0.0188,
      "step": 6948
    },
    {
      "epoch": 1.5811149032992038,
      "grad_norm": 2.180460150678355,
      "learning_rate": 9.662045601018834e-07,
      "loss": 0.0376,
      "step": 6949
    },
    {
      "epoch": 1.5813424345847555,
      "grad_norm": 1.4286588206093294,
      "learning_rate": 9.661296951649914e-07,
      "loss": 0.0409,
      "step": 6950
    },
    {
      "epoch": 1.5815699658703073,
      "grad_norm": 1.3765902238160939,
      "learning_rate": 9.660548232560331e-07,
      "loss": 0.0463,
      "step": 6951
    },
    {
      "epoch": 1.581797497155859,
      "grad_norm": 1.185065939414716,
      "learning_rate": 9.659799443765392e-07,
      "loss": 0.0448,
      "step": 6952
    },
    {
      "epoch": 1.5820250284414108,
      "grad_norm": 1.337116480171379,
      "learning_rate": 9.659050585280394e-07,
      "loss": 0.026,
      "step": 6953
    },
    {
      "epoch": 1.5822525597269625,
      "grad_norm": 1.020718028178023,
      "learning_rate": 9.658301657120646e-07,
      "loss": 0.0281,
      "step": 6954
    },
    {
      "epoch": 1.5824800910125143,
      "grad_norm": 0.9055039875506272,
      "learning_rate": 9.657552659301455e-07,
      "loss": 0.0219,
      "step": 6955
    },
    {
      "epoch": 1.582707622298066,
      "grad_norm": 1.6953021742731489,
      "learning_rate": 9.656803591838126e-07,
      "loss": 0.0678,
      "step": 6956
    },
    {
      "epoch": 1.5829351535836178,
      "grad_norm": 1.0772903707549362,
      "learning_rate": 9.656054454745973e-07,
      "loss": 0.0345,
      "step": 6957
    },
    {
      "epoch": 1.5831626848691696,
      "grad_norm": 1.1587537360572808,
      "learning_rate": 9.655305248040302e-07,
      "loss": 0.0359,
      "step": 6958
    },
    {
      "epoch": 1.5833902161547213,
      "grad_norm": 0.9105264852735304,
      "learning_rate": 9.654555971736431e-07,
      "loss": 0.0282,
      "step": 6959
    },
    {
      "epoch": 1.583617747440273,
      "grad_norm": 1.220396846866995,
      "learning_rate": 9.653806625849671e-07,
      "loss": 0.0155,
      "step": 6960
    },
    {
      "epoch": 1.5838452787258248,
      "grad_norm": 0.7086608428171763,
      "learning_rate": 9.653057210395338e-07,
      "loss": 0.0112,
      "step": 6961
    },
    {
      "epoch": 1.5840728100113766,
      "grad_norm": 1.075979091036995,
      "learning_rate": 9.652307725388746e-07,
      "loss": 0.0189,
      "step": 6962
    },
    {
      "epoch": 1.5843003412969283,
      "grad_norm": 1.5170568607738595,
      "learning_rate": 9.651558170845216e-07,
      "loss": 0.0338,
      "step": 6963
    },
    {
      "epoch": 1.58452787258248,
      "grad_norm": 0.7652438799921432,
      "learning_rate": 9.650808546780068e-07,
      "loss": 0.0144,
      "step": 6964
    },
    {
      "epoch": 1.5847554038680318,
      "grad_norm": 1.9783702731659012,
      "learning_rate": 9.65005885320862e-07,
      "loss": 0.0512,
      "step": 6965
    },
    {
      "epoch": 1.5849829351535836,
      "grad_norm": 1.2326785631564439,
      "learning_rate": 9.649309090146194e-07,
      "loss": 0.0308,
      "step": 6966
    },
    {
      "epoch": 1.5852104664391353,
      "grad_norm": 0.9273837401389944,
      "learning_rate": 9.64855925760812e-07,
      "loss": 0.0099,
      "step": 6967
    },
    {
      "epoch": 1.585437997724687,
      "grad_norm": 2.082982278260053,
      "learning_rate": 9.647809355609715e-07,
      "loss": 0.0815,
      "step": 6968
    },
    {
      "epoch": 1.5856655290102388,
      "grad_norm": 1.154389332279708,
      "learning_rate": 9.647059384166314e-07,
      "loss": 0.0199,
      "step": 6969
    },
    {
      "epoch": 1.5858930602957906,
      "grad_norm": 1.3155932055166657,
      "learning_rate": 9.646309343293237e-07,
      "loss": 0.042,
      "step": 6970
    },
    {
      "epoch": 1.5861205915813423,
      "grad_norm": 0.7089433372912292,
      "learning_rate": 9.645559233005819e-07,
      "loss": 0.016,
      "step": 6971
    },
    {
      "epoch": 1.586348122866894,
      "grad_norm": 0.9808906887148743,
      "learning_rate": 9.644809053319388e-07,
      "loss": 0.0152,
      "step": 6972
    },
    {
      "epoch": 1.5865756541524458,
      "grad_norm": 1.154371513813329,
      "learning_rate": 9.644058804249276e-07,
      "loss": 0.0157,
      "step": 6973
    },
    {
      "epoch": 1.5868031854379976,
      "grad_norm": 1.1758967067439108,
      "learning_rate": 9.64330848581082e-07,
      "loss": 0.051,
      "step": 6974
    },
    {
      "epoch": 1.5870307167235493,
      "grad_norm": 1.0737810620882602,
      "learning_rate": 9.642558098019353e-07,
      "loss": 0.0295,
      "step": 6975
    },
    {
      "epoch": 1.587258248009101,
      "grad_norm": 1.288007211828143,
      "learning_rate": 9.641807640890212e-07,
      "loss": 0.0406,
      "step": 6976
    },
    {
      "epoch": 1.587485779294653,
      "grad_norm": 0.8343056698359398,
      "learning_rate": 9.64105711443873e-07,
      "loss": 0.0229,
      "step": 6977
    },
    {
      "epoch": 1.5877133105802048,
      "grad_norm": 2.336951974280419,
      "learning_rate": 9.640306518680257e-07,
      "loss": 0.0516,
      "step": 6978
    },
    {
      "epoch": 1.5879408418657566,
      "grad_norm": 1.471956955036482,
      "learning_rate": 9.639555853630126e-07,
      "loss": 0.0328,
      "step": 6979
    },
    {
      "epoch": 1.5881683731513083,
      "grad_norm": 1.2981308020075435,
      "learning_rate": 9.63880511930368e-07,
      "loss": 0.0315,
      "step": 6980
    },
    {
      "epoch": 1.58839590443686,
      "grad_norm": 1.142394389444487,
      "learning_rate": 9.638054315716264e-07,
      "loss": 0.0304,
      "step": 6981
    },
    {
      "epoch": 1.5886234357224118,
      "grad_norm": 1.578253454753235,
      "learning_rate": 9.637303442883223e-07,
      "loss": 0.0333,
      "step": 6982
    },
    {
      "epoch": 1.5888509670079636,
      "grad_norm": 0.8057669195788537,
      "learning_rate": 9.636552500819903e-07,
      "loss": 0.0113,
      "step": 6983
    },
    {
      "epoch": 1.5890784982935153,
      "grad_norm": 1.0137918642215906,
      "learning_rate": 9.635801489541652e-07,
      "loss": 0.0174,
      "step": 6984
    },
    {
      "epoch": 1.589306029579067,
      "grad_norm": 0.764528773311874,
      "learning_rate": 9.635050409063818e-07,
      "loss": 0.0138,
      "step": 6985
    },
    {
      "epoch": 1.5895335608646188,
      "grad_norm": 1.0778573956251072,
      "learning_rate": 9.634299259401756e-07,
      "loss": 0.0266,
      "step": 6986
    },
    {
      "epoch": 1.5897610921501708,
      "grad_norm": 0.9923931257391829,
      "learning_rate": 9.633548040570815e-07,
      "loss": 0.0304,
      "step": 6987
    },
    {
      "epoch": 1.5899886234357226,
      "grad_norm": 1.6429925247966484,
      "learning_rate": 9.632796752586345e-07,
      "loss": 0.062,
      "step": 6988
    },
    {
      "epoch": 1.5902161547212743,
      "grad_norm": 0.9225605285785448,
      "learning_rate": 9.632045395463708e-07,
      "loss": 0.0226,
      "step": 6989
    },
    {
      "epoch": 1.590443686006826,
      "grad_norm": 1.3067475369066461,
      "learning_rate": 9.631293969218256e-07,
      "loss": 0.0369,
      "step": 6990
    },
    {
      "epoch": 1.5906712172923778,
      "grad_norm": 1.4050369350662906,
      "learning_rate": 9.63054247386535e-07,
      "loss": 0.0373,
      "step": 6991
    },
    {
      "epoch": 1.5908987485779296,
      "grad_norm": 1.1648943302861068,
      "learning_rate": 9.629790909420344e-07,
      "loss": 0.0246,
      "step": 6992
    },
    {
      "epoch": 1.5911262798634813,
      "grad_norm": 1.9942288503602235,
      "learning_rate": 9.629039275898603e-07,
      "loss": 0.0334,
      "step": 6993
    },
    {
      "epoch": 1.591353811149033,
      "grad_norm": 0.732963655692899,
      "learning_rate": 9.628287573315488e-07,
      "loss": 0.0166,
      "step": 6994
    },
    {
      "epoch": 1.5915813424345848,
      "grad_norm": 1.2794851527570241,
      "learning_rate": 9.627535801686359e-07,
      "loss": 0.0213,
      "step": 6995
    },
    {
      "epoch": 1.5918088737201366,
      "grad_norm": 0.8394812636488541,
      "learning_rate": 9.626783961026586e-07,
      "loss": 0.0207,
      "step": 6996
    },
    {
      "epoch": 1.5920364050056883,
      "grad_norm": 1.6392112440185473,
      "learning_rate": 9.626032051351534e-07,
      "loss": 0.0401,
      "step": 6997
    },
    {
      "epoch": 1.59226393629124,
      "grad_norm": 0.744033455630395,
      "learning_rate": 9.62528007267657e-07,
      "loss": 0.0228,
      "step": 6998
    },
    {
      "epoch": 1.5924914675767918,
      "grad_norm": 1.1583696590619958,
      "learning_rate": 9.62452802501706e-07,
      "loss": 0.0285,
      "step": 6999
    },
    {
      "epoch": 1.5927189988623436,
      "grad_norm": 1.0576690714175723,
      "learning_rate": 9.62377590838838e-07,
      "loss": 0.0264,
      "step": 7000
    },
    {
      "epoch": 1.5929465301478953,
      "grad_norm": 1.275729463204334,
      "learning_rate": 9.623023722805898e-07,
      "loss": 0.0288,
      "step": 7001
    },
    {
      "epoch": 1.593174061433447,
      "grad_norm": 1.1375669002340378,
      "learning_rate": 9.62227146828499e-07,
      "loss": 0.0186,
      "step": 7002
    },
    {
      "epoch": 1.5934015927189988,
      "grad_norm": 1.612053860660408,
      "learning_rate": 9.621519144841028e-07,
      "loss": 0.0388,
      "step": 7003
    },
    {
      "epoch": 1.5936291240045506,
      "grad_norm": 1.0591497663172853,
      "learning_rate": 9.62076675248939e-07,
      "loss": 0.0183,
      "step": 7004
    },
    {
      "epoch": 1.5938566552901023,
      "grad_norm": 1.325781849288007,
      "learning_rate": 9.620014291245452e-07,
      "loss": 0.0326,
      "step": 7005
    },
    {
      "epoch": 1.594084186575654,
      "grad_norm": 1.2729296118881142,
      "learning_rate": 9.619261761124592e-07,
      "loss": 0.0193,
      "step": 7006
    },
    {
      "epoch": 1.5943117178612058,
      "grad_norm": 1.681909742131678,
      "learning_rate": 9.618509162142196e-07,
      "loss": 0.0475,
      "step": 7007
    },
    {
      "epoch": 1.5945392491467576,
      "grad_norm": 0.7128073458713802,
      "learning_rate": 9.61775649431364e-07,
      "loss": 0.0121,
      "step": 7008
    },
    {
      "epoch": 1.5947667804323093,
      "grad_norm": 1.3690331746720998,
      "learning_rate": 9.617003757654309e-07,
      "loss": 0.0678,
      "step": 7009
    },
    {
      "epoch": 1.594994311717861,
      "grad_norm": 2.0090900821884397,
      "learning_rate": 9.616250952179586e-07,
      "loss": 0.0745,
      "step": 7010
    },
    {
      "epoch": 1.5952218430034129,
      "grad_norm": 0.8477768247832146,
      "learning_rate": 9.615498077904865e-07,
      "loss": 0.0171,
      "step": 7011
    },
    {
      "epoch": 1.5954493742889646,
      "grad_norm": 1.319320154301809,
      "learning_rate": 9.614745134845518e-07,
      "loss": 0.0162,
      "step": 7012
    },
    {
      "epoch": 1.5956769055745164,
      "grad_norm": 1.1420941610839186,
      "learning_rate": 9.613992123016948e-07,
      "loss": 0.019,
      "step": 7013
    },
    {
      "epoch": 1.595904436860068,
      "grad_norm": 0.9134071733158048,
      "learning_rate": 9.61323904243454e-07,
      "loss": 0.0152,
      "step": 7014
    },
    {
      "epoch": 1.5961319681456199,
      "grad_norm": 0.7303097608530996,
      "learning_rate": 9.612485893113682e-07,
      "loss": 0.0278,
      "step": 7015
    },
    {
      "epoch": 1.5963594994311718,
      "grad_norm": 1.0612669762930091,
      "learning_rate": 9.611732675069773e-07,
      "loss": 0.0261,
      "step": 7016
    },
    {
      "epoch": 1.5965870307167236,
      "grad_norm": 1.3408790033343598,
      "learning_rate": 9.610979388318206e-07,
      "loss": 0.0209,
      "step": 7017
    },
    {
      "epoch": 1.5968145620022753,
      "grad_norm": 0.7958312148744503,
      "learning_rate": 9.610226032874374e-07,
      "loss": 0.0143,
      "step": 7018
    },
    {
      "epoch": 1.597042093287827,
      "grad_norm": 0.937747305014374,
      "learning_rate": 9.609472608753676e-07,
      "loss": 0.028,
      "step": 7019
    },
    {
      "epoch": 1.5972696245733788,
      "grad_norm": 1.8013387061820019,
      "learning_rate": 9.60871911597151e-07,
      "loss": 0.0424,
      "step": 7020
    },
    {
      "epoch": 1.5974971558589306,
      "grad_norm": 0.7645277826475468,
      "learning_rate": 9.607965554543276e-07,
      "loss": 0.0296,
      "step": 7021
    },
    {
      "epoch": 1.5977246871444823,
      "grad_norm": 3.9610847778692526,
      "learning_rate": 9.607211924484378e-07,
      "loss": 0.0255,
      "step": 7022
    },
    {
      "epoch": 1.597952218430034,
      "grad_norm": 0.8342016000959991,
      "learning_rate": 9.606458225810214e-07,
      "loss": 0.0212,
      "step": 7023
    },
    {
      "epoch": 1.5981797497155859,
      "grad_norm": 1.2153427389880582,
      "learning_rate": 9.605704458536193e-07,
      "loss": 0.0266,
      "step": 7024
    },
    {
      "epoch": 1.5984072810011376,
      "grad_norm": 1.2102515172387232,
      "learning_rate": 9.604950622677717e-07,
      "loss": 0.025,
      "step": 7025
    },
    {
      "epoch": 1.5986348122866896,
      "grad_norm": 0.8437385307802597,
      "learning_rate": 9.604196718250197e-07,
      "loss": 0.0233,
      "step": 7026
    },
    {
      "epoch": 1.5988623435722413,
      "grad_norm": 1.0488726295589468,
      "learning_rate": 9.603442745269036e-07,
      "loss": 0.0252,
      "step": 7027
    },
    {
      "epoch": 1.599089874857793,
      "grad_norm": 1.6157731581386559,
      "learning_rate": 9.602688703749648e-07,
      "loss": 0.043,
      "step": 7028
    },
    {
      "epoch": 1.5993174061433448,
      "grad_norm": 1.303474155434243,
      "learning_rate": 9.601934593707444e-07,
      "loss": 0.0532,
      "step": 7029
    },
    {
      "epoch": 1.5995449374288966,
      "grad_norm": 1.2151510476102094,
      "learning_rate": 9.601180415157834e-07,
      "loss": 0.0337,
      "step": 7030
    },
    {
      "epoch": 1.5997724687144483,
      "grad_norm": 2.003854866210954,
      "learning_rate": 9.600426168116237e-07,
      "loss": 0.0616,
      "step": 7031
    },
    {
      "epoch": 1.6,
      "grad_norm": 1.1432974018636979,
      "learning_rate": 9.599671852598062e-07,
      "loss": 0.0266,
      "step": 7032
    },
    {
      "epoch": 1.6002275312855518,
      "grad_norm": 1.0901706676214284,
      "learning_rate": 9.59891746861873e-07,
      "loss": 0.0457,
      "step": 7033
    },
    {
      "epoch": 1.6004550625711036,
      "grad_norm": 1.1518684169841051,
      "learning_rate": 9.598163016193656e-07,
      "loss": 0.0343,
      "step": 7034
    },
    {
      "epoch": 1.6006825938566553,
      "grad_norm": 0.9541937097744977,
      "learning_rate": 9.597408495338266e-07,
      "loss": 0.0209,
      "step": 7035
    },
    {
      "epoch": 1.600910125142207,
      "grad_norm": 1.065605058803838,
      "learning_rate": 9.596653906067974e-07,
      "loss": 0.0174,
      "step": 7036
    },
    {
      "epoch": 1.6011376564277588,
      "grad_norm": 1.3338536559778928,
      "learning_rate": 9.595899248398209e-07,
      "loss": 0.0121,
      "step": 7037
    },
    {
      "epoch": 1.6013651877133106,
      "grad_norm": 1.0338937176005598,
      "learning_rate": 9.595144522344386e-07,
      "loss": 0.0312,
      "step": 7038
    },
    {
      "epoch": 1.6015927189988624,
      "grad_norm": 1.0835957785189174,
      "learning_rate": 9.594389727921937e-07,
      "loss": 0.0321,
      "step": 7039
    },
    {
      "epoch": 1.601820250284414,
      "grad_norm": 1.4114015950065997,
      "learning_rate": 9.593634865146286e-07,
      "loss": 0.0361,
      "step": 7040
    },
    {
      "epoch": 1.6020477815699659,
      "grad_norm": 0.8252106392926577,
      "learning_rate": 9.592879934032864e-07,
      "loss": 0.0237,
      "step": 7041
    },
    {
      "epoch": 1.6022753128555176,
      "grad_norm": 0.844534530767116,
      "learning_rate": 9.5921249345971e-07,
      "loss": 0.0202,
      "step": 7042
    },
    {
      "epoch": 1.6025028441410694,
      "grad_norm": 0.9928822413510855,
      "learning_rate": 9.59136986685442e-07,
      "loss": 0.0299,
      "step": 7043
    },
    {
      "epoch": 1.6027303754266211,
      "grad_norm": 0.9196169522806006,
      "learning_rate": 9.59061473082026e-07,
      "loss": 0.0214,
      "step": 7044
    },
    {
      "epoch": 1.6029579067121729,
      "grad_norm": 1.1755612158971747,
      "learning_rate": 9.589859526510053e-07,
      "loss": 0.0353,
      "step": 7045
    },
    {
      "epoch": 1.6031854379977246,
      "grad_norm": 1.148468492505787,
      "learning_rate": 9.589104253939233e-07,
      "loss": 0.028,
      "step": 7046
    },
    {
      "epoch": 1.6034129692832764,
      "grad_norm": 1.6602307360868302,
      "learning_rate": 9.588348913123238e-07,
      "loss": 0.0441,
      "step": 7047
    },
    {
      "epoch": 1.6036405005688281,
      "grad_norm": 0.6872308598203596,
      "learning_rate": 9.587593504077506e-07,
      "loss": 0.0098,
      "step": 7048
    },
    {
      "epoch": 1.6038680318543799,
      "grad_norm": 1.9397616045566994,
      "learning_rate": 9.586838026817475e-07,
      "loss": 0.044,
      "step": 7049
    },
    {
      "epoch": 1.6040955631399316,
      "grad_norm": 0.5914971934167476,
      "learning_rate": 9.586082481358587e-07,
      "loss": 0.0166,
      "step": 7050
    },
    {
      "epoch": 1.6043230944254834,
      "grad_norm": 1.1549598175673381,
      "learning_rate": 9.58532686771628e-07,
      "loss": 0.0387,
      "step": 7051
    },
    {
      "epoch": 1.6045506257110351,
      "grad_norm": 1.300805798462683,
      "learning_rate": 9.584571185906002e-07,
      "loss": 0.0419,
      "step": 7052
    },
    {
      "epoch": 1.6047781569965869,
      "grad_norm": 1.663258284565078,
      "learning_rate": 9.583815435943195e-07,
      "loss": 0.0545,
      "step": 7053
    },
    {
      "epoch": 1.6050056882821386,
      "grad_norm": 1.6346928610912306,
      "learning_rate": 9.583059617843306e-07,
      "loss": 0.0531,
      "step": 7054
    },
    {
      "epoch": 1.6052332195676906,
      "grad_norm": 1.3362888182930384,
      "learning_rate": 9.582303731621784e-07,
      "loss": 0.0339,
      "step": 7055
    },
    {
      "epoch": 1.6054607508532424,
      "grad_norm": 1.2524212164281763,
      "learning_rate": 9.581547777294076e-07,
      "loss": 0.037,
      "step": 7056
    },
    {
      "epoch": 1.605688282138794,
      "grad_norm": 1.1246757203173876,
      "learning_rate": 9.580791754875631e-07,
      "loss": 0.028,
      "step": 7057
    },
    {
      "epoch": 1.6059158134243459,
      "grad_norm": 1.333681435778776,
      "learning_rate": 9.580035664381905e-07,
      "loss": 0.0228,
      "step": 7058
    },
    {
      "epoch": 1.6061433447098976,
      "grad_norm": 0.9594094533322884,
      "learning_rate": 9.579279505828348e-07,
      "loss": 0.0143,
      "step": 7059
    },
    {
      "epoch": 1.6063708759954494,
      "grad_norm": 0.9563960168156239,
      "learning_rate": 9.578523279230414e-07,
      "loss": 0.0163,
      "step": 7060
    },
    {
      "epoch": 1.6065984072810011,
      "grad_norm": 5.9699340851931835,
      "learning_rate": 9.577766984603562e-07,
      "loss": 0.0241,
      "step": 7061
    },
    {
      "epoch": 1.6068259385665529,
      "grad_norm": 0.7339313234548795,
      "learning_rate": 9.577010621963249e-07,
      "loss": 0.0231,
      "step": 7062
    },
    {
      "epoch": 1.6070534698521046,
      "grad_norm": 0.6163762811246133,
      "learning_rate": 9.576254191324926e-07,
      "loss": 0.0111,
      "step": 7063
    },
    {
      "epoch": 1.6072810011376564,
      "grad_norm": 1.226228473815418,
      "learning_rate": 9.575497692704063e-07,
      "loss": 0.0258,
      "step": 7064
    },
    {
      "epoch": 1.6075085324232083,
      "grad_norm": 0.8877466161700153,
      "learning_rate": 9.574741126116118e-07,
      "loss": 0.0208,
      "step": 7065
    },
    {
      "epoch": 1.60773606370876,
      "grad_norm": 0.9889765681955124,
      "learning_rate": 9.573984491576554e-07,
      "loss": 0.0187,
      "step": 7066
    },
    {
      "epoch": 1.6079635949943119,
      "grad_norm": 1.0927095972515526,
      "learning_rate": 9.573227789100833e-07,
      "loss": 0.0284,
      "step": 7067
    },
    {
      "epoch": 1.6081911262798636,
      "grad_norm": 1.243753628953149,
      "learning_rate": 9.572471018704422e-07,
      "loss": 0.0313,
      "step": 7068
    },
    {
      "epoch": 1.6084186575654154,
      "grad_norm": 0.8906322800855065,
      "learning_rate": 9.57171418040279e-07,
      "loss": 0.0161,
      "step": 7069
    },
    {
      "epoch": 1.608646188850967,
      "grad_norm": 1.049183753999333,
      "learning_rate": 9.570957274211399e-07,
      "loss": 0.0159,
      "step": 7070
    },
    {
      "epoch": 1.6088737201365189,
      "grad_norm": 0.7870709759753043,
      "learning_rate": 9.570200300145727e-07,
      "loss": 0.0226,
      "step": 7071
    },
    {
      "epoch": 1.6091012514220706,
      "grad_norm": 0.9038235928408551,
      "learning_rate": 9.569443258221243e-07,
      "loss": 0.0271,
      "step": 7072
    },
    {
      "epoch": 1.6093287827076224,
      "grad_norm": 0.5708462387231178,
      "learning_rate": 9.568686148453412e-07,
      "loss": 0.01,
      "step": 7073
    },
    {
      "epoch": 1.6095563139931741,
      "grad_norm": 3.0375077188874386,
      "learning_rate": 9.56792897085772e-07,
      "loss": 0.0262,
      "step": 7074
    },
    {
      "epoch": 1.6097838452787259,
      "grad_norm": 1.0056015986896478,
      "learning_rate": 9.567171725449635e-07,
      "loss": 0.0212,
      "step": 7075
    },
    {
      "epoch": 1.6100113765642776,
      "grad_norm": 0.5035348315325101,
      "learning_rate": 9.566414412244635e-07,
      "loss": 0.0092,
      "step": 7076
    },
    {
      "epoch": 1.6102389078498294,
      "grad_norm": 0.7901614524633361,
      "learning_rate": 9.565657031258196e-07,
      "loss": 0.0181,
      "step": 7077
    },
    {
      "epoch": 1.6104664391353811,
      "grad_norm": 1.0025231926932794,
      "learning_rate": 9.564899582505802e-07,
      "loss": 0.0192,
      "step": 7078
    },
    {
      "epoch": 1.6106939704209329,
      "grad_norm": 1.5425227306548526,
      "learning_rate": 9.56414206600293e-07,
      "loss": 0.0264,
      "step": 7079
    },
    {
      "epoch": 1.6109215017064846,
      "grad_norm": 0.8753251618688311,
      "learning_rate": 9.563384481765064e-07,
      "loss": 0.0146,
      "step": 7080
    },
    {
      "epoch": 1.6111490329920364,
      "grad_norm": 1.6211547505794992,
      "learning_rate": 9.562626829807689e-07,
      "loss": 0.0337,
      "step": 7081
    },
    {
      "epoch": 1.6113765642775881,
      "grad_norm": 1.0293396736787777,
      "learning_rate": 9.561869110146288e-07,
      "loss": 0.0242,
      "step": 7082
    },
    {
      "epoch": 1.6116040955631399,
      "grad_norm": 1.170655178365064,
      "learning_rate": 9.561111322796346e-07,
      "loss": 0.0202,
      "step": 7083
    },
    {
      "epoch": 1.6118316268486916,
      "grad_norm": 0.9824361143419721,
      "learning_rate": 9.560353467773354e-07,
      "loss": 0.0191,
      "step": 7084
    },
    {
      "epoch": 1.6120591581342434,
      "grad_norm": 1.6512507172023276,
      "learning_rate": 9.5595955450928e-07,
      "loss": 0.0912,
      "step": 7085
    },
    {
      "epoch": 1.6122866894197951,
      "grad_norm": 1.0057053210926752,
      "learning_rate": 9.558837554770173e-07,
      "loss": 0.0117,
      "step": 7086
    },
    {
      "epoch": 1.612514220705347,
      "grad_norm": 1.152283256946704,
      "learning_rate": 9.558079496820965e-07,
      "loss": 0.028,
      "step": 7087
    },
    {
      "epoch": 1.6127417519908986,
      "grad_norm": 1.4855302834454076,
      "learning_rate": 9.557321371260675e-07,
      "loss": 0.0496,
      "step": 7088
    },
    {
      "epoch": 1.6129692832764504,
      "grad_norm": 0.614029539419703,
      "learning_rate": 9.55656317810479e-07,
      "loss": 0.0062,
      "step": 7089
    },
    {
      "epoch": 1.6131968145620021,
      "grad_norm": 1.2505747278089356,
      "learning_rate": 9.555804917368808e-07,
      "loss": 0.0318,
      "step": 7090
    },
    {
      "epoch": 1.613424345847554,
      "grad_norm": 0.48847611292584286,
      "learning_rate": 9.55504658906823e-07,
      "loss": 0.009,
      "step": 7091
    },
    {
      "epoch": 1.6136518771331056,
      "grad_norm": 1.2376858891100468,
      "learning_rate": 9.554288193218552e-07,
      "loss": 0.0173,
      "step": 7092
    },
    {
      "epoch": 1.6138794084186574,
      "grad_norm": 1.169463723123295,
      "learning_rate": 9.553529729835275e-07,
      "loss": 0.0211,
      "step": 7093
    },
    {
      "epoch": 1.6141069397042094,
      "grad_norm": 0.857640511419088,
      "learning_rate": 9.552771198933903e-07,
      "loss": 0.0288,
      "step": 7094
    },
    {
      "epoch": 1.6143344709897611,
      "grad_norm": 1.47380443051205,
      "learning_rate": 9.552012600529934e-07,
      "loss": 0.039,
      "step": 7095
    },
    {
      "epoch": 1.6145620022753129,
      "grad_norm": 0.6577467462472111,
      "learning_rate": 9.551253934638874e-07,
      "loss": 0.0161,
      "step": 7096
    },
    {
      "epoch": 1.6147895335608646,
      "grad_norm": 0.8489890237697779,
      "learning_rate": 9.550495201276231e-07,
      "loss": 0.0205,
      "step": 7097
    },
    {
      "epoch": 1.6150170648464164,
      "grad_norm": 0.9889782164153689,
      "learning_rate": 9.54973640045751e-07,
      "loss": 0.0237,
      "step": 7098
    },
    {
      "epoch": 1.6152445961319681,
      "grad_norm": 0.809890832374879,
      "learning_rate": 9.54897753219822e-07,
      "loss": 0.0291,
      "step": 7099
    },
    {
      "epoch": 1.6154721274175199,
      "grad_norm": 1.2855868112580302,
      "learning_rate": 9.548218596513871e-07,
      "loss": 0.0264,
      "step": 7100
    },
    {
      "epoch": 1.6156996587030716,
      "grad_norm": 1.212074283812276,
      "learning_rate": 9.547459593419975e-07,
      "loss": 0.025,
      "step": 7101
    },
    {
      "epoch": 1.6159271899886234,
      "grad_norm": 0.8920524690958506,
      "learning_rate": 9.546700522932042e-07,
      "loss": 0.0223,
      "step": 7102
    },
    {
      "epoch": 1.6161547212741754,
      "grad_norm": 1.4420267514584946,
      "learning_rate": 9.54594138506559e-07,
      "loss": 0.0249,
      "step": 7103
    },
    {
      "epoch": 1.6163822525597271,
      "grad_norm": 1.054995208097927,
      "learning_rate": 9.545182179836132e-07,
      "loss": 0.0195,
      "step": 7104
    },
    {
      "epoch": 1.6166097838452789,
      "grad_norm": 0.9679075593459778,
      "learning_rate": 9.544422907259185e-07,
      "loss": 0.0212,
      "step": 7105
    },
    {
      "epoch": 1.6168373151308306,
      "grad_norm": 1.397218645429241,
      "learning_rate": 9.543663567350267e-07,
      "loss": 0.0392,
      "step": 7106
    },
    {
      "epoch": 1.6170648464163824,
      "grad_norm": 1.2828433568502122,
      "learning_rate": 9.542904160124896e-07,
      "loss": 0.0249,
      "step": 7107
    },
    {
      "epoch": 1.6172923777019341,
      "grad_norm": 1.5322721263241048,
      "learning_rate": 9.542144685598598e-07,
      "loss": 0.0478,
      "step": 7108
    },
    {
      "epoch": 1.6175199089874859,
      "grad_norm": 1.2137972791553564,
      "learning_rate": 9.54138514378689e-07,
      "loss": 0.0375,
      "step": 7109
    },
    {
      "epoch": 1.6177474402730376,
      "grad_norm": 0.7890160927750831,
      "learning_rate": 9.540625534705297e-07,
      "loss": 0.0283,
      "step": 7110
    },
    {
      "epoch": 1.6179749715585894,
      "grad_norm": 0.985091132889504,
      "learning_rate": 9.539865858369347e-07,
      "loss": 0.0232,
      "step": 7111
    },
    {
      "epoch": 1.6182025028441411,
      "grad_norm": 1.9270292191386824,
      "learning_rate": 9.539106114794564e-07,
      "loss": 0.0349,
      "step": 7112
    },
    {
      "epoch": 1.6184300341296929,
      "grad_norm": 1.0921842788734426,
      "learning_rate": 9.538346303996472e-07,
      "loss": 0.0309,
      "step": 7113
    },
    {
      "epoch": 1.6186575654152446,
      "grad_norm": 1.5428410141330413,
      "learning_rate": 9.537586425990604e-07,
      "loss": 0.0376,
      "step": 7114
    },
    {
      "epoch": 1.6188850967007964,
      "grad_norm": 0.9541924274634735,
      "learning_rate": 9.536826480792493e-07,
      "loss": 0.0222,
      "step": 7115
    },
    {
      "epoch": 1.6191126279863481,
      "grad_norm": 0.6491782772626704,
      "learning_rate": 9.53606646841767e-07,
      "loss": 0.0117,
      "step": 7116
    },
    {
      "epoch": 1.6193401592719,
      "grad_norm": 0.7582067823854062,
      "learning_rate": 9.535306388881663e-07,
      "loss": 0.0115,
      "step": 7117
    },
    {
      "epoch": 1.6195676905574516,
      "grad_norm": 1.058048254765856,
      "learning_rate": 9.534546242200012e-07,
      "loss": 0.0288,
      "step": 7118
    },
    {
      "epoch": 1.6197952218430034,
      "grad_norm": 1.006371706098208,
      "learning_rate": 9.53378602838825e-07,
      "loss": 0.0251,
      "step": 7119
    },
    {
      "epoch": 1.6200227531285551,
      "grad_norm": 1.4533877255802528,
      "learning_rate": 9.533025747461916e-07,
      "loss": 0.0265,
      "step": 7120
    },
    {
      "epoch": 1.620250284414107,
      "grad_norm": 0.6692865346649786,
      "learning_rate": 9.532265399436549e-07,
      "loss": 0.0212,
      "step": 7121
    },
    {
      "epoch": 1.6204778156996587,
      "grad_norm": 0.7404732378891826,
      "learning_rate": 9.531504984327688e-07,
      "loss": 0.0244,
      "step": 7122
    },
    {
      "epoch": 1.6207053469852104,
      "grad_norm": 1.2920381960808738,
      "learning_rate": 9.530744502150874e-07,
      "loss": 0.038,
      "step": 7123
    },
    {
      "epoch": 1.6209328782707622,
      "grad_norm": 1.065581280470929,
      "learning_rate": 9.529983952921652e-07,
      "loss": 0.0133,
      "step": 7124
    },
    {
      "epoch": 1.621160409556314,
      "grad_norm": 1.0352421647641357,
      "learning_rate": 9.529223336655565e-07,
      "loss": 0.0148,
      "step": 7125
    },
    {
      "epoch": 1.6213879408418657,
      "grad_norm": 1.7347261921396229,
      "learning_rate": 9.528462653368158e-07,
      "loss": 0.0332,
      "step": 7126
    },
    {
      "epoch": 1.6216154721274174,
      "grad_norm": 3.9973664373381004,
      "learning_rate": 9.527701903074977e-07,
      "loss": 0.0268,
      "step": 7127
    },
    {
      "epoch": 1.6218430034129692,
      "grad_norm": 1.2707508978387,
      "learning_rate": 9.526941085791574e-07,
      "loss": 0.0144,
      "step": 7128
    },
    {
      "epoch": 1.622070534698521,
      "grad_norm": 1.3140699709212826,
      "learning_rate": 9.526180201533497e-07,
      "loss": 0.0481,
      "step": 7129
    },
    {
      "epoch": 1.6222980659840727,
      "grad_norm": 1.2337723938681724,
      "learning_rate": 9.525419250316295e-07,
      "loss": 0.0393,
      "step": 7130
    },
    {
      "epoch": 1.6225255972696244,
      "grad_norm": 0.9964400049733639,
      "learning_rate": 9.524658232155524e-07,
      "loss": 0.024,
      "step": 7131
    },
    {
      "epoch": 1.6227531285551762,
      "grad_norm": 1.707150409119038,
      "learning_rate": 9.523897147066735e-07,
      "loss": 0.0838,
      "step": 7132
    },
    {
      "epoch": 1.6229806598407281,
      "grad_norm": 1.8267390601418545,
      "learning_rate": 9.523135995065483e-07,
      "loss": 0.0496,
      "step": 7133
    },
    {
      "epoch": 1.62320819112628,
      "grad_norm": 1.1513425182735106,
      "learning_rate": 9.522374776167328e-07,
      "loss": 0.0184,
      "step": 7134
    },
    {
      "epoch": 1.6234357224118316,
      "grad_norm": 0.9185048401595257,
      "learning_rate": 9.521613490387824e-07,
      "loss": 0.0248,
      "step": 7135
    },
    {
      "epoch": 1.6236632536973834,
      "grad_norm": 0.8073827228532496,
      "learning_rate": 9.520852137742534e-07,
      "loss": 0.0101,
      "step": 7136
    },
    {
      "epoch": 1.6238907849829352,
      "grad_norm": 1.1657350043826,
      "learning_rate": 9.520090718247016e-07,
      "loss": 0.0196,
      "step": 7137
    },
    {
      "epoch": 1.624118316268487,
      "grad_norm": 1.216680110083079,
      "learning_rate": 9.519329231916831e-07,
      "loss": 0.0166,
      "step": 7138
    },
    {
      "epoch": 1.6243458475540387,
      "grad_norm": 0.8535660856425935,
      "learning_rate": 9.518567678767546e-07,
      "loss": 0.0175,
      "step": 7139
    },
    {
      "epoch": 1.6245733788395904,
      "grad_norm": 0.8711790356295614,
      "learning_rate": 9.51780605881472e-07,
      "loss": 0.0194,
      "step": 7140
    },
    {
      "epoch": 1.6248009101251422,
      "grad_norm": 0.9825901471843272,
      "learning_rate": 9.517044372073927e-07,
      "loss": 0.0329,
      "step": 7141
    },
    {
      "epoch": 1.6250284414106941,
      "grad_norm": 0.8661738676983474,
      "learning_rate": 9.516282618560727e-07,
      "loss": 0.0159,
      "step": 7142
    },
    {
      "epoch": 1.6252559726962459,
      "grad_norm": 1.4036723999735945,
      "learning_rate": 9.515520798290695e-07,
      "loss": 0.0349,
      "step": 7143
    },
    {
      "epoch": 1.6254835039817976,
      "grad_norm": 2.037341510060114,
      "learning_rate": 9.514758911279398e-07,
      "loss": 0.0468,
      "step": 7144
    },
    {
      "epoch": 1.6257110352673494,
      "grad_norm": 0.9408128263381598,
      "learning_rate": 9.513996957542407e-07,
      "loss": 0.0159,
      "step": 7145
    },
    {
      "epoch": 1.6259385665529011,
      "grad_norm": 0.8642664200516336,
      "learning_rate": 9.513234937095296e-07,
      "loss": 0.0171,
      "step": 7146
    },
    {
      "epoch": 1.626166097838453,
      "grad_norm": 0.6696172061661173,
      "learning_rate": 9.512472849953639e-07,
      "loss": 0.008,
      "step": 7147
    },
    {
      "epoch": 1.6263936291240046,
      "grad_norm": 1.0375083047536862,
      "learning_rate": 9.511710696133012e-07,
      "loss": 0.0301,
      "step": 7148
    },
    {
      "epoch": 1.6266211604095564,
      "grad_norm": 1.0584004449366777,
      "learning_rate": 9.510948475648993e-07,
      "loss": 0.0303,
      "step": 7149
    },
    {
      "epoch": 1.6268486916951082,
      "grad_norm": 1.3319455316451698,
      "learning_rate": 9.510186188517159e-07,
      "loss": 0.0252,
      "step": 7150
    },
    {
      "epoch": 1.62707622298066,
      "grad_norm": 0.8623683188578419,
      "learning_rate": 9.50942383475309e-07,
      "loss": 0.021,
      "step": 7151
    },
    {
      "epoch": 1.6273037542662117,
      "grad_norm": 0.8237163876539081,
      "learning_rate": 9.508661414372367e-07,
      "loss": 0.0187,
      "step": 7152
    },
    {
      "epoch": 1.6275312855517634,
      "grad_norm": 1.228002048633561,
      "learning_rate": 9.507898927390571e-07,
      "loss": 0.0269,
      "step": 7153
    },
    {
      "epoch": 1.6277588168373152,
      "grad_norm": 0.974390010062014,
      "learning_rate": 9.507136373823289e-07,
      "loss": 0.0239,
      "step": 7154
    },
    {
      "epoch": 1.627986348122867,
      "grad_norm": 0.9397725129892953,
      "learning_rate": 9.506373753686104e-07,
      "loss": 0.023,
      "step": 7155
    },
    {
      "epoch": 1.6282138794084187,
      "grad_norm": 0.7242842173267595,
      "learning_rate": 9.505611066994605e-07,
      "loss": 0.0181,
      "step": 7156
    },
    {
      "epoch": 1.6284414106939704,
      "grad_norm": 1.0535752948833204,
      "learning_rate": 9.504848313764375e-07,
      "loss": 0.0308,
      "step": 7157
    },
    {
      "epoch": 1.6286689419795222,
      "grad_norm": 0.6753903947752541,
      "learning_rate": 9.50408549401101e-07,
      "loss": 0.0213,
      "step": 7158
    },
    {
      "epoch": 1.628896473265074,
      "grad_norm": 0.6869344941943479,
      "learning_rate": 9.503322607750096e-07,
      "loss": 0.0177,
      "step": 7159
    },
    {
      "epoch": 1.6291240045506257,
      "grad_norm": 0.7178640822389143,
      "learning_rate": 9.502559654997226e-07,
      "loss": 0.0132,
      "step": 7160
    },
    {
      "epoch": 1.6293515358361774,
      "grad_norm": 0.7864947505397063,
      "learning_rate": 9.501796635767992e-07,
      "loss": 0.0173,
      "step": 7161
    },
    {
      "epoch": 1.6295790671217292,
      "grad_norm": 3.864576003209912,
      "learning_rate": 9.501033550077993e-07,
      "loss": 0.0368,
      "step": 7162
    },
    {
      "epoch": 1.629806598407281,
      "grad_norm": 1.2986294430212195,
      "learning_rate": 9.500270397942819e-07,
      "loss": 0.0478,
      "step": 7163
    },
    {
      "epoch": 1.6300341296928327,
      "grad_norm": 1.3721206434366422,
      "learning_rate": 9.499507179378072e-07,
      "loss": 0.0359,
      "step": 7164
    },
    {
      "epoch": 1.6302616609783844,
      "grad_norm": 1.316378474967445,
      "learning_rate": 9.498743894399348e-07,
      "loss": 0.0303,
      "step": 7165
    },
    {
      "epoch": 1.6304891922639362,
      "grad_norm": 0.8738319121581792,
      "learning_rate": 9.497980543022251e-07,
      "loss": 0.0132,
      "step": 7166
    },
    {
      "epoch": 1.630716723549488,
      "grad_norm": 0.608916362735345,
      "learning_rate": 9.497217125262378e-07,
      "loss": 0.0151,
      "step": 7167
    },
    {
      "epoch": 1.6309442548350397,
      "grad_norm": 2.021086411000962,
      "learning_rate": 9.496453641135337e-07,
      "loss": 0.0522,
      "step": 7168
    },
    {
      "epoch": 1.6311717861205914,
      "grad_norm": 1.1154208604186049,
      "learning_rate": 9.495690090656728e-07,
      "loss": 0.0159,
      "step": 7169
    },
    {
      "epoch": 1.6313993174061432,
      "grad_norm": 0.9846084938710009,
      "learning_rate": 9.494926473842156e-07,
      "loss": 0.0165,
      "step": 7170
    },
    {
      "epoch": 1.631626848691695,
      "grad_norm": 0.5328708135848763,
      "learning_rate": 9.494162790707232e-07,
      "loss": 0.0068,
      "step": 7171
    },
    {
      "epoch": 1.631854379977247,
      "grad_norm": 0.6702708665810064,
      "learning_rate": 9.493399041267559e-07,
      "loss": 0.0098,
      "step": 7172
    },
    {
      "epoch": 1.6320819112627987,
      "grad_norm": 1.448780085449854,
      "learning_rate": 9.492635225538751e-07,
      "loss": 0.0299,
      "step": 7173
    },
    {
      "epoch": 1.6323094425483504,
      "grad_norm": 0.7311154868873958,
      "learning_rate": 9.491871343536418e-07,
      "loss": 0.0232,
      "step": 7174
    },
    {
      "epoch": 1.6325369738339022,
      "grad_norm": 0.8439545969021113,
      "learning_rate": 9.491107395276172e-07,
      "loss": 0.0165,
      "step": 7175
    },
    {
      "epoch": 1.632764505119454,
      "grad_norm": 1.1423895358734097,
      "learning_rate": 9.490343380773629e-07,
      "loss": 0.0248,
      "step": 7176
    },
    {
      "epoch": 1.6329920364050057,
      "grad_norm": 0.9038841951011269,
      "learning_rate": 9.489579300044396e-07,
      "loss": 0.015,
      "step": 7177
    },
    {
      "epoch": 1.6332195676905574,
      "grad_norm": 0.8016153369265806,
      "learning_rate": 9.4888151531041e-07,
      "loss": 0.0187,
      "step": 7178
    },
    {
      "epoch": 1.6334470989761092,
      "grad_norm": 0.8901417826041008,
      "learning_rate": 9.488050939968351e-07,
      "loss": 0.0258,
      "step": 7179
    },
    {
      "epoch": 1.633674630261661,
      "grad_norm": 0.8252096401919968,
      "learning_rate": 9.487286660652773e-07,
      "loss": 0.0103,
      "step": 7180
    },
    {
      "epoch": 1.633902161547213,
      "grad_norm": 1.5818607668923288,
      "learning_rate": 9.486522315172983e-07,
      "loss": 0.0392,
      "step": 7181
    },
    {
      "epoch": 1.6341296928327647,
      "grad_norm": 0.843929267173253,
      "learning_rate": 9.485757903544606e-07,
      "loss": 0.0193,
      "step": 7182
    },
    {
      "epoch": 1.6343572241183164,
      "grad_norm": 0.9775290008365384,
      "learning_rate": 9.484993425783262e-07,
      "loss": 0.0249,
      "step": 7183
    },
    {
      "epoch": 1.6345847554038682,
      "grad_norm": 1.3741978732780942,
      "learning_rate": 9.484228881904577e-07,
      "loss": 0.0445,
      "step": 7184
    },
    {
      "epoch": 1.63481228668942,
      "grad_norm": 1.494963474433962,
      "learning_rate": 9.483464271924177e-07,
      "loss": 0.0309,
      "step": 7185
    },
    {
      "epoch": 1.6350398179749717,
      "grad_norm": 0.9195283327097106,
      "learning_rate": 9.482699595857689e-07,
      "loss": 0.0197,
      "step": 7186
    },
    {
      "epoch": 1.6352673492605234,
      "grad_norm": 0.819552193594169,
      "learning_rate": 9.481934853720742e-07,
      "loss": 0.0136,
      "step": 7187
    },
    {
      "epoch": 1.6354948805460752,
      "grad_norm": 1.3681250341696483,
      "learning_rate": 9.481170045528968e-07,
      "loss": 0.0309,
      "step": 7188
    },
    {
      "epoch": 1.635722411831627,
      "grad_norm": 1.0838732709720864,
      "learning_rate": 9.480405171297992e-07,
      "loss": 0.0183,
      "step": 7189
    },
    {
      "epoch": 1.6359499431171787,
      "grad_norm": 0.8589492659489036,
      "learning_rate": 9.479640231043453e-07,
      "loss": 0.0283,
      "step": 7190
    },
    {
      "epoch": 1.6361774744027304,
      "grad_norm": 0.4389346454173033,
      "learning_rate": 9.47887522478098e-07,
      "loss": 0.0107,
      "step": 7191
    },
    {
      "epoch": 1.6364050056882822,
      "grad_norm": 1.5076520306070575,
      "learning_rate": 9.478110152526212e-07,
      "loss": 0.0318,
      "step": 7192
    },
    {
      "epoch": 1.636632536973834,
      "grad_norm": 0.8666999378509156,
      "learning_rate": 9.477345014294787e-07,
      "loss": 0.0343,
      "step": 7193
    },
    {
      "epoch": 1.6368600682593857,
      "grad_norm": 1.4057039119124,
      "learning_rate": 9.476579810102337e-07,
      "loss": 0.0308,
      "step": 7194
    },
    {
      "epoch": 1.6370875995449374,
      "grad_norm": 0.9295417198301341,
      "learning_rate": 9.475814539964506e-07,
      "loss": 0.0146,
      "step": 7195
    },
    {
      "epoch": 1.6373151308304892,
      "grad_norm": 1.7038654584980972,
      "learning_rate": 9.475049203896934e-07,
      "loss": 0.0385,
      "step": 7196
    },
    {
      "epoch": 1.637542662116041,
      "grad_norm": 0.8551699103163861,
      "learning_rate": 9.474283801915262e-07,
      "loss": 0.0181,
      "step": 7197
    },
    {
      "epoch": 1.6377701934015927,
      "grad_norm": 0.733499182783048,
      "learning_rate": 9.473518334035134e-07,
      "loss": 0.0141,
      "step": 7198
    },
    {
      "epoch": 1.6379977246871444,
      "grad_norm": 1.8245741661840873,
      "learning_rate": 9.472752800272194e-07,
      "loss": 0.0502,
      "step": 7199
    },
    {
      "epoch": 1.6382252559726962,
      "grad_norm": 1.7101516018547451,
      "learning_rate": 9.471987200642093e-07,
      "loss": 0.0461,
      "step": 7200
    },
    {
      "epoch": 1.638452787258248,
      "grad_norm": 0.7019785797323762,
      "learning_rate": 9.471221535160471e-07,
      "loss": 0.013,
      "step": 7201
    },
    {
      "epoch": 1.6386803185437997,
      "grad_norm": 1.2558910970344273,
      "learning_rate": 9.470455803842982e-07,
      "loss": 0.029,
      "step": 7202
    },
    {
      "epoch": 1.6389078498293514,
      "grad_norm": 1.521689503580041,
      "learning_rate": 9.469690006705274e-07,
      "loss": 0.0158,
      "step": 7203
    },
    {
      "epoch": 1.6391353811149032,
      "grad_norm": 1.334570302964857,
      "learning_rate": 9.468924143762996e-07,
      "loss": 0.0271,
      "step": 7204
    },
    {
      "epoch": 1.639362912400455,
      "grad_norm": 0.718253019493209,
      "learning_rate": 9.468158215031805e-07,
      "loss": 0.0142,
      "step": 7205
    },
    {
      "epoch": 1.6395904436860067,
      "grad_norm": 1.066686510907017,
      "learning_rate": 9.467392220527358e-07,
      "loss": 0.0289,
      "step": 7206
    },
    {
      "epoch": 1.6398179749715585,
      "grad_norm": 1.0683098954831638,
      "learning_rate": 9.4666261602653e-07,
      "loss": 0.0362,
      "step": 7207
    },
    {
      "epoch": 1.6400455062571102,
      "grad_norm": 1.8077047968568642,
      "learning_rate": 9.465860034261298e-07,
      "loss": 0.026,
      "step": 7208
    },
    {
      "epoch": 1.640273037542662,
      "grad_norm": 1.1472184148194302,
      "learning_rate": 9.465093842531007e-07,
      "loss": 0.0234,
      "step": 7209
    },
    {
      "epoch": 1.640500568828214,
      "grad_norm": 0.8276394924288174,
      "learning_rate": 9.464327585090084e-07,
      "loss": 0.0125,
      "step": 7210
    },
    {
      "epoch": 1.6407281001137657,
      "grad_norm": 1.301523612966525,
      "learning_rate": 9.463561261954192e-07,
      "loss": 0.0398,
      "step": 7211
    },
    {
      "epoch": 1.6409556313993174,
      "grad_norm": 0.8836337063456542,
      "learning_rate": 9.462794873138995e-07,
      "loss": 0.0209,
      "step": 7212
    },
    {
      "epoch": 1.6411831626848692,
      "grad_norm": 1.0208628010333278,
      "learning_rate": 9.462028418660155e-07,
      "loss": 0.0331,
      "step": 7213
    },
    {
      "epoch": 1.641410693970421,
      "grad_norm": 0.8560244593909649,
      "learning_rate": 9.461261898533333e-07,
      "loss": 0.0181,
      "step": 7214
    },
    {
      "epoch": 1.6416382252559727,
      "grad_norm": 0.8831773567982883,
      "learning_rate": 9.460495312774203e-07,
      "loss": 0.0168,
      "step": 7215
    },
    {
      "epoch": 1.6418657565415244,
      "grad_norm": 1.4094359738334923,
      "learning_rate": 9.459728661398427e-07,
      "loss": 0.0321,
      "step": 7216
    },
    {
      "epoch": 1.6420932878270762,
      "grad_norm": 1.2053767582653305,
      "learning_rate": 9.458961944421674e-07,
      "loss": 0.026,
      "step": 7217
    },
    {
      "epoch": 1.642320819112628,
      "grad_norm": 0.9548072481260151,
      "learning_rate": 9.458195161859618e-07,
      "loss": 0.0164,
      "step": 7218
    },
    {
      "epoch": 1.6425483503981797,
      "grad_norm": 1.0144085210237268,
      "learning_rate": 9.457428313727927e-07,
      "loss": 0.0167,
      "step": 7219
    },
    {
      "epoch": 1.6427758816837317,
      "grad_norm": 0.9111105741597719,
      "learning_rate": 9.456661400042278e-07,
      "loss": 0.0232,
      "step": 7220
    },
    {
      "epoch": 1.6430034129692834,
      "grad_norm": 1.2223424085257393,
      "learning_rate": 9.455894420818339e-07,
      "loss": 0.0372,
      "step": 7221
    },
    {
      "epoch": 1.6432309442548352,
      "grad_norm": 0.9476964903869362,
      "learning_rate": 9.455127376071791e-07,
      "loss": 0.0312,
      "step": 7222
    },
    {
      "epoch": 1.643458475540387,
      "grad_norm": 1.9505254497792965,
      "learning_rate": 9.454360265818309e-07,
      "loss": 0.0434,
      "step": 7223
    },
    {
      "epoch": 1.6436860068259387,
      "grad_norm": 0.5165376819358042,
      "learning_rate": 9.453593090073571e-07,
      "loss": 0.0096,
      "step": 7224
    },
    {
      "epoch": 1.6439135381114904,
      "grad_norm": 0.9526396496896132,
      "learning_rate": 9.45282584885326e-07,
      "loss": 0.0214,
      "step": 7225
    },
    {
      "epoch": 1.6441410693970422,
      "grad_norm": 0.9178268969108745,
      "learning_rate": 9.452058542173054e-07,
      "loss": 0.0165,
      "step": 7226
    },
    {
      "epoch": 1.644368600682594,
      "grad_norm": 1.1974894992771798,
      "learning_rate": 9.451291170048632e-07,
      "loss": 0.0297,
      "step": 7227
    },
    {
      "epoch": 1.6445961319681457,
      "grad_norm": 1.5678130163290367,
      "learning_rate": 9.450523732495684e-07,
      "loss": 0.0464,
      "step": 7228
    },
    {
      "epoch": 1.6448236632536974,
      "grad_norm": 1.4468718707240071,
      "learning_rate": 9.449756229529893e-07,
      "loss": 0.0431,
      "step": 7229
    },
    {
      "epoch": 1.6450511945392492,
      "grad_norm": 1.6752861254777147,
      "learning_rate": 9.448988661166944e-07,
      "loss": 0.0437,
      "step": 7230
    },
    {
      "epoch": 1.645278725824801,
      "grad_norm": 1.17485711424526,
      "learning_rate": 9.448221027422525e-07,
      "loss": 0.0181,
      "step": 7231
    },
    {
      "epoch": 1.6455062571103527,
      "grad_norm": 1.1092110440467429,
      "learning_rate": 9.447453328312325e-07,
      "loss": 0.0199,
      "step": 7232
    },
    {
      "epoch": 1.6457337883959045,
      "grad_norm": 1.4139573034619977,
      "learning_rate": 9.446685563852037e-07,
      "loss": 0.0596,
      "step": 7233
    },
    {
      "epoch": 1.6459613196814562,
      "grad_norm": 1.2088953201451433,
      "learning_rate": 9.445917734057349e-07,
      "loss": 0.0287,
      "step": 7234
    },
    {
      "epoch": 1.646188850967008,
      "grad_norm": 0.7540427001183505,
      "learning_rate": 9.445149838943955e-07,
      "loss": 0.0102,
      "step": 7235
    },
    {
      "epoch": 1.6464163822525597,
      "grad_norm": 1.241840511545734,
      "learning_rate": 9.44438187852755e-07,
      "loss": 0.0251,
      "step": 7236
    },
    {
      "epoch": 1.6466439135381115,
      "grad_norm": 1.290037885762213,
      "learning_rate": 9.443613852823832e-07,
      "loss": 0.0261,
      "step": 7237
    },
    {
      "epoch": 1.6468714448236632,
      "grad_norm": 0.7588594864777832,
      "learning_rate": 9.442845761848493e-07,
      "loss": 0.0163,
      "step": 7238
    },
    {
      "epoch": 1.647098976109215,
      "grad_norm": 1.0104778694363679,
      "learning_rate": 9.442077605617236e-07,
      "loss": 0.026,
      "step": 7239
    },
    {
      "epoch": 1.6473265073947667,
      "grad_norm": 0.8587600369889241,
      "learning_rate": 9.441309384145758e-07,
      "loss": 0.0284,
      "step": 7240
    },
    {
      "epoch": 1.6475540386803185,
      "grad_norm": 1.9541419364169168,
      "learning_rate": 9.440541097449759e-07,
      "loss": 0.0459,
      "step": 7241
    },
    {
      "epoch": 1.6477815699658702,
      "grad_norm": 0.6698653743269256,
      "learning_rate": 9.439772745544945e-07,
      "loss": 0.0146,
      "step": 7242
    },
    {
      "epoch": 1.648009101251422,
      "grad_norm": 1.3702624178440999,
      "learning_rate": 9.439004328447019e-07,
      "loss": 0.0271,
      "step": 7243
    },
    {
      "epoch": 1.6482366325369737,
      "grad_norm": 0.9113341463128616,
      "learning_rate": 9.438235846171684e-07,
      "loss": 0.0201,
      "step": 7244
    },
    {
      "epoch": 1.6484641638225255,
      "grad_norm": 2.2293229554521536,
      "learning_rate": 9.437467298734646e-07,
      "loss": 0.0516,
      "step": 7245
    },
    {
      "epoch": 1.6486916951080772,
      "grad_norm": 1.2345944716664825,
      "learning_rate": 9.436698686151616e-07,
      "loss": 0.0433,
      "step": 7246
    },
    {
      "epoch": 1.648919226393629,
      "grad_norm": 1.026762857151891,
      "learning_rate": 9.435930008438299e-07,
      "loss": 0.0249,
      "step": 7247
    },
    {
      "epoch": 1.6491467576791807,
      "grad_norm": 2.006823130023674,
      "learning_rate": 9.435161265610407e-07,
      "loss": 0.0533,
      "step": 7248
    },
    {
      "epoch": 1.6493742889647327,
      "grad_norm": 0.9398804691129719,
      "learning_rate": 9.434392457683653e-07,
      "loss": 0.0194,
      "step": 7249
    },
    {
      "epoch": 1.6496018202502845,
      "grad_norm": 0.9516028765806651,
      "learning_rate": 9.433623584673751e-07,
      "loss": 0.0172,
      "step": 7250
    },
    {
      "epoch": 1.6498293515358362,
      "grad_norm": 1.2348276563947835,
      "learning_rate": 9.432854646596412e-07,
      "loss": 0.028,
      "step": 7251
    },
    {
      "epoch": 1.650056882821388,
      "grad_norm": 1.1645809136436136,
      "learning_rate": 9.432085643467352e-07,
      "loss": 0.0327,
      "step": 7252
    },
    {
      "epoch": 1.6502844141069397,
      "grad_norm": 0.9654120827213446,
      "learning_rate": 9.43131657530229e-07,
      "loss": 0.0279,
      "step": 7253
    },
    {
      "epoch": 1.6505119453924915,
      "grad_norm": 1.6599070006924297,
      "learning_rate": 9.430547442116944e-07,
      "loss": 0.0255,
      "step": 7254
    },
    {
      "epoch": 1.6507394766780432,
      "grad_norm": 1.3574100186809808,
      "learning_rate": 9.429778243927031e-07,
      "loss": 0.0319,
      "step": 7255
    },
    {
      "epoch": 1.650967007963595,
      "grad_norm": 0.8311506286233733,
      "learning_rate": 9.429008980748279e-07,
      "loss": 0.0174,
      "step": 7256
    },
    {
      "epoch": 1.6511945392491467,
      "grad_norm": 1.1772735396269376,
      "learning_rate": 9.428239652596402e-07,
      "loss": 0.0208,
      "step": 7257
    },
    {
      "epoch": 1.6514220705346985,
      "grad_norm": 1.2001797242913066,
      "learning_rate": 9.427470259487127e-07,
      "loss": 0.0211,
      "step": 7258
    },
    {
      "epoch": 1.6516496018202504,
      "grad_norm": 0.9282059355299311,
      "learning_rate": 9.426700801436181e-07,
      "loss": 0.0173,
      "step": 7259
    },
    {
      "epoch": 1.6518771331058022,
      "grad_norm": 1.5594692564466364,
      "learning_rate": 9.425931278459287e-07,
      "loss": 0.0325,
      "step": 7260
    },
    {
      "epoch": 1.652104664391354,
      "grad_norm": 0.7983223497891204,
      "learning_rate": 9.425161690572174e-07,
      "loss": 0.0221,
      "step": 7261
    },
    {
      "epoch": 1.6523321956769057,
      "grad_norm": 0.7429757690165122,
      "learning_rate": 9.42439203779057e-07,
      "loss": 0.0181,
      "step": 7262
    },
    {
      "epoch": 1.6525597269624575,
      "grad_norm": 0.6184275747393381,
      "learning_rate": 9.42362232013021e-07,
      "loss": 0.0101,
      "step": 7263
    },
    {
      "epoch": 1.6527872582480092,
      "grad_norm": 0.8345316604506051,
      "learning_rate": 9.422852537606819e-07,
      "loss": 0.0171,
      "step": 7264
    },
    {
      "epoch": 1.653014789533561,
      "grad_norm": 1.211708377820841,
      "learning_rate": 9.422082690236134e-07,
      "loss": 0.0391,
      "step": 7265
    },
    {
      "epoch": 1.6532423208191127,
      "grad_norm": 0.9833647618604962,
      "learning_rate": 9.421312778033889e-07,
      "loss": 0.0255,
      "step": 7266
    },
    {
      "epoch": 1.6534698521046645,
      "grad_norm": 0.6817384084220871,
      "learning_rate": 9.420542801015817e-07,
      "loss": 0.0113,
      "step": 7267
    },
    {
      "epoch": 1.6536973833902162,
      "grad_norm": 0.7215860417309474,
      "learning_rate": 9.419772759197656e-07,
      "loss": 0.0136,
      "step": 7268
    },
    {
      "epoch": 1.653924914675768,
      "grad_norm": 1.638819919411159,
      "learning_rate": 9.41900265259515e-07,
      "loss": 0.0288,
      "step": 7269
    },
    {
      "epoch": 1.6541524459613197,
      "grad_norm": 1.4490962654168988,
      "learning_rate": 9.418232481224029e-07,
      "loss": 0.0174,
      "step": 7270
    },
    {
      "epoch": 1.6543799772468715,
      "grad_norm": 0.8905064462620383,
      "learning_rate": 9.417462245100038e-07,
      "loss": 0.0317,
      "step": 7271
    },
    {
      "epoch": 1.6546075085324232,
      "grad_norm": 1.6835820308875613,
      "learning_rate": 9.416691944238922e-07,
      "loss": 0.0232,
      "step": 7272
    },
    {
      "epoch": 1.654835039817975,
      "grad_norm": 1.1362565076308901,
      "learning_rate": 9.415921578656422e-07,
      "loss": 0.0204,
      "step": 7273
    },
    {
      "epoch": 1.6550625711035267,
      "grad_norm": 0.8853854118859108,
      "learning_rate": 9.415151148368282e-07,
      "loss": 0.0178,
      "step": 7274
    },
    {
      "epoch": 1.6552901023890785,
      "grad_norm": 1.4527616622652773,
      "learning_rate": 9.41438065339025e-07,
      "loss": 0.0324,
      "step": 7275
    },
    {
      "epoch": 1.6555176336746302,
      "grad_norm": 1.7095898173356143,
      "learning_rate": 9.413610093738072e-07,
      "loss": 0.0593,
      "step": 7276
    },
    {
      "epoch": 1.655745164960182,
      "grad_norm": 1.983306213333864,
      "learning_rate": 9.412839469427498e-07,
      "loss": 0.0194,
      "step": 7277
    },
    {
      "epoch": 1.6559726962457337,
      "grad_norm": 0.8860075841323366,
      "learning_rate": 9.412068780474277e-07,
      "loss": 0.0176,
      "step": 7278
    },
    {
      "epoch": 1.6562002275312855,
      "grad_norm": 1.241390628166181,
      "learning_rate": 9.411298026894161e-07,
      "loss": 0.0347,
      "step": 7279
    },
    {
      "epoch": 1.6564277588168372,
      "grad_norm": 1.6842602052705251,
      "learning_rate": 9.410527208702905e-07,
      "loss": 0.0421,
      "step": 7280
    },
    {
      "epoch": 1.656655290102389,
      "grad_norm": 0.750562018500017,
      "learning_rate": 9.409756325916259e-07,
      "loss": 0.0059,
      "step": 7281
    },
    {
      "epoch": 1.6568828213879407,
      "grad_norm": 1.815823404100124,
      "learning_rate": 9.408985378549981e-07,
      "loss": 0.0168,
      "step": 7282
    },
    {
      "epoch": 1.6571103526734925,
      "grad_norm": 0.7503514595870169,
      "learning_rate": 9.408214366619829e-07,
      "loss": 0.0124,
      "step": 7283
    },
    {
      "epoch": 1.6573378839590442,
      "grad_norm": 2.24763699227768,
      "learning_rate": 9.407443290141557e-07,
      "loss": 0.0377,
      "step": 7284
    },
    {
      "epoch": 1.657565415244596,
      "grad_norm": 1.0037578814814987,
      "learning_rate": 9.406672149130928e-07,
      "loss": 0.0243,
      "step": 7285
    },
    {
      "epoch": 1.6577929465301477,
      "grad_norm": 1.4239295877989984,
      "learning_rate": 9.405900943603702e-07,
      "loss": 0.0293,
      "step": 7286
    },
    {
      "epoch": 1.6580204778156995,
      "grad_norm": 1.4300072807272197,
      "learning_rate": 9.405129673575639e-07,
      "loss": 0.0236,
      "step": 7287
    },
    {
      "epoch": 1.6582480091012515,
      "grad_norm": 1.1491851444657144,
      "learning_rate": 9.404358339062505e-07,
      "loss": 0.0288,
      "step": 7288
    },
    {
      "epoch": 1.6584755403868032,
      "grad_norm": 0.7516080758836439,
      "learning_rate": 9.403586940080063e-07,
      "loss": 0.0144,
      "step": 7289
    },
    {
      "epoch": 1.658703071672355,
      "grad_norm": 1.530757840694647,
      "learning_rate": 9.40281547664408e-07,
      "loss": 0.0397,
      "step": 7290
    },
    {
      "epoch": 1.6589306029579067,
      "grad_norm": 1.1643602793634618,
      "learning_rate": 9.402043948770321e-07,
      "loss": 0.028,
      "step": 7291
    },
    {
      "epoch": 1.6591581342434585,
      "grad_norm": 1.0627327234680397,
      "learning_rate": 9.401272356474557e-07,
      "loss": 0.02,
      "step": 7292
    },
    {
      "epoch": 1.6593856655290102,
      "grad_norm": 1.4900034307204348,
      "learning_rate": 9.400500699772558e-07,
      "loss": 0.0311,
      "step": 7293
    },
    {
      "epoch": 1.659613196814562,
      "grad_norm": 1.0255614989109871,
      "learning_rate": 9.399728978680094e-07,
      "loss": 0.0169,
      "step": 7294
    },
    {
      "epoch": 1.6598407281001137,
      "grad_norm": 1.9785969932383063,
      "learning_rate": 9.39895719321294e-07,
      "loss": 0.0473,
      "step": 7295
    },
    {
      "epoch": 1.6600682593856655,
      "grad_norm": 1.7615173107010964,
      "learning_rate": 9.398185343386866e-07,
      "loss": 0.0247,
      "step": 7296
    },
    {
      "epoch": 1.6602957906712172,
      "grad_norm": 1.1723530953451027,
      "learning_rate": 9.39741342921765e-07,
      "loss": 0.0554,
      "step": 7297
    },
    {
      "epoch": 1.6605233219567692,
      "grad_norm": 0.7436297955855727,
      "learning_rate": 9.396641450721067e-07,
      "loss": 0.0194,
      "step": 7298
    },
    {
      "epoch": 1.660750853242321,
      "grad_norm": 1.4081310717365318,
      "learning_rate": 9.395869407912897e-07,
      "loss": 0.0569,
      "step": 7299
    },
    {
      "epoch": 1.6609783845278727,
      "grad_norm": 0.894628689932518,
      "learning_rate": 9.395097300808916e-07,
      "loss": 0.0159,
      "step": 7300
    },
    {
      "epoch": 1.6612059158134245,
      "grad_norm": 1.372321476496526,
      "learning_rate": 9.394325129424906e-07,
      "loss": 0.0302,
      "step": 7301
    },
    {
      "epoch": 1.6614334470989762,
      "grad_norm": 1.3718759092878112,
      "learning_rate": 9.39355289377665e-07,
      "loss": 0.0438,
      "step": 7302
    },
    {
      "epoch": 1.661660978384528,
      "grad_norm": 0.7459997929216897,
      "learning_rate": 9.392780593879932e-07,
      "loss": 0.014,
      "step": 7303
    },
    {
      "epoch": 1.6618885096700797,
      "grad_norm": 1.247027911519019,
      "learning_rate": 9.392008229750533e-07,
      "loss": 0.0262,
      "step": 7304
    },
    {
      "epoch": 1.6621160409556315,
      "grad_norm": 0.7780135059733467,
      "learning_rate": 9.391235801404236e-07,
      "loss": 0.025,
      "step": 7305
    },
    {
      "epoch": 1.6623435722411832,
      "grad_norm": 1.2318547883057585,
      "learning_rate": 9.390463308856837e-07,
      "loss": 0.0457,
      "step": 7306
    },
    {
      "epoch": 1.662571103526735,
      "grad_norm": 1.0879174466911097,
      "learning_rate": 9.389690752124118e-07,
      "loss": 0.0241,
      "step": 7307
    },
    {
      "epoch": 1.6627986348122867,
      "grad_norm": 0.7759080705210638,
      "learning_rate": 9.388918131221869e-07,
      "loss": 0.0128,
      "step": 7308
    },
    {
      "epoch": 1.6630261660978385,
      "grad_norm": 0.8826852994061488,
      "learning_rate": 9.388145446165884e-07,
      "loss": 0.0182,
      "step": 7309
    },
    {
      "epoch": 1.6632536973833902,
      "grad_norm": 1.1297233571426981,
      "learning_rate": 9.387372696971952e-07,
      "loss": 0.0199,
      "step": 7310
    },
    {
      "epoch": 1.663481228668942,
      "grad_norm": 0.8544651904661695,
      "learning_rate": 9.386599883655869e-07,
      "loss": 0.0221,
      "step": 7311
    },
    {
      "epoch": 1.6637087599544937,
      "grad_norm": 0.9358707418174771,
      "learning_rate": 9.385827006233426e-07,
      "loss": 0.0198,
      "step": 7312
    },
    {
      "epoch": 1.6639362912400455,
      "grad_norm": 1.1977498117294199,
      "learning_rate": 9.385054064720425e-07,
      "loss": 0.0239,
      "step": 7313
    },
    {
      "epoch": 1.6641638225255972,
      "grad_norm": 0.7773318647986129,
      "learning_rate": 9.38428105913266e-07,
      "loss": 0.0172,
      "step": 7314
    },
    {
      "epoch": 1.664391353811149,
      "grad_norm": 0.9321658881754474,
      "learning_rate": 9.38350798948593e-07,
      "loss": 0.0264,
      "step": 7315
    },
    {
      "epoch": 1.6646188850967008,
      "grad_norm": 0.8640475281006017,
      "learning_rate": 9.382734855796036e-07,
      "loss": 0.0143,
      "step": 7316
    },
    {
      "epoch": 1.6648464163822525,
      "grad_norm": 1.0548381808425786,
      "learning_rate": 9.381961658078778e-07,
      "loss": 0.0281,
      "step": 7317
    },
    {
      "epoch": 1.6650739476678043,
      "grad_norm": 8.927252731821625,
      "learning_rate": 9.381188396349958e-07,
      "loss": 0.1749,
      "step": 7318
    },
    {
      "epoch": 1.665301478953356,
      "grad_norm": 1.344358077736511,
      "learning_rate": 9.380415070625384e-07,
      "loss": 0.0198,
      "step": 7319
    },
    {
      "epoch": 1.6655290102389078,
      "grad_norm": 0.9360614628382258,
      "learning_rate": 9.379641680920859e-07,
      "loss": 0.0305,
      "step": 7320
    },
    {
      "epoch": 1.6657565415244595,
      "grad_norm": 1.0008530296862048,
      "learning_rate": 9.378868227252188e-07,
      "loss": 0.0316,
      "step": 7321
    },
    {
      "epoch": 1.6659840728100113,
      "grad_norm": 1.0104338772741304,
      "learning_rate": 9.378094709635183e-07,
      "loss": 0.0222,
      "step": 7322
    },
    {
      "epoch": 1.666211604095563,
      "grad_norm": 1.1295029565277173,
      "learning_rate": 9.377321128085651e-07,
      "loss": 0.0216,
      "step": 7323
    },
    {
      "epoch": 1.6664391353811148,
      "grad_norm": 1.4325128459640029,
      "learning_rate": 9.3765474826194e-07,
      "loss": 0.0407,
      "step": 7324
    },
    {
      "epoch": 1.6666666666666665,
      "grad_norm": 0.8717439451003935,
      "learning_rate": 9.375773773252248e-07,
      "loss": 0.0229,
      "step": 7325
    },
    {
      "epoch": 1.6668941979522183,
      "grad_norm": 1.5212675549475514,
      "learning_rate": 9.375000000000001e-07,
      "loss": 0.0272,
      "step": 7326
    },
    {
      "epoch": 1.6671217292377702,
      "grad_norm": 0.8629482674997104,
      "learning_rate": 9.374226162878478e-07,
      "loss": 0.0384,
      "step": 7327
    },
    {
      "epoch": 1.667349260523322,
      "grad_norm": 1.1348949806800248,
      "learning_rate": 9.373452261903495e-07,
      "loss": 0.024,
      "step": 7328
    },
    {
      "epoch": 1.6675767918088737,
      "grad_norm": 0.5711114628031161,
      "learning_rate": 9.372678297090867e-07,
      "loss": 0.0079,
      "step": 7329
    },
    {
      "epoch": 1.6678043230944255,
      "grad_norm": 1.4567308110129513,
      "learning_rate": 9.371904268456415e-07,
      "loss": 0.0331,
      "step": 7330
    },
    {
      "epoch": 1.6680318543799773,
      "grad_norm": 1.9138608329728288,
      "learning_rate": 9.371130176015958e-07,
      "loss": 0.0366,
      "step": 7331
    },
    {
      "epoch": 1.668259385665529,
      "grad_norm": 2.0571902981483845,
      "learning_rate": 9.370356019785315e-07,
      "loss": 0.0592,
      "step": 7332
    },
    {
      "epoch": 1.6684869169510808,
      "grad_norm": 0.7881569099172085,
      "learning_rate": 9.369581799780308e-07,
      "loss": 0.0176,
      "step": 7333
    },
    {
      "epoch": 1.6687144482366325,
      "grad_norm": 1.1900721132902,
      "learning_rate": 9.368807516016764e-07,
      "loss": 0.0328,
      "step": 7334
    },
    {
      "epoch": 1.6689419795221843,
      "grad_norm": 0.5535988858385974,
      "learning_rate": 9.368033168510506e-07,
      "loss": 0.0104,
      "step": 7335
    },
    {
      "epoch": 1.669169510807736,
      "grad_norm": 1.6854773824574556,
      "learning_rate": 9.36725875727736e-07,
      "loss": 0.0526,
      "step": 7336
    },
    {
      "epoch": 1.669397042093288,
      "grad_norm": 3.146539161288818,
      "learning_rate": 9.366484282333155e-07,
      "loss": 0.0663,
      "step": 7337
    },
    {
      "epoch": 1.6696245733788397,
      "grad_norm": 0.7589184408986651,
      "learning_rate": 9.365709743693718e-07,
      "loss": 0.0129,
      "step": 7338
    },
    {
      "epoch": 1.6698521046643915,
      "grad_norm": 1.0190198600786413,
      "learning_rate": 9.364935141374881e-07,
      "loss": 0.0205,
      "step": 7339
    },
    {
      "epoch": 1.6700796359499432,
      "grad_norm": 1.0383142874986837,
      "learning_rate": 9.364160475392472e-07,
      "loss": 0.0219,
      "step": 7340
    },
    {
      "epoch": 1.670307167235495,
      "grad_norm": 0.8229443778154666,
      "learning_rate": 9.363385745762329e-07,
      "loss": 0.0181,
      "step": 7341
    },
    {
      "epoch": 1.6705346985210467,
      "grad_norm": 0.7971442304086289,
      "learning_rate": 9.362610952500281e-07,
      "loss": 0.0131,
      "step": 7342
    },
    {
      "epoch": 1.6707622298065985,
      "grad_norm": 1.2641570859853692,
      "learning_rate": 9.361836095622167e-07,
      "loss": 0.0268,
      "step": 7343
    },
    {
      "epoch": 1.6709897610921502,
      "grad_norm": 1.5217910567435424,
      "learning_rate": 9.361061175143823e-07,
      "loss": 0.0465,
      "step": 7344
    },
    {
      "epoch": 1.671217292377702,
      "grad_norm": 1.3028329103521379,
      "learning_rate": 9.360286191081085e-07,
      "loss": 0.0354,
      "step": 7345
    },
    {
      "epoch": 1.6714448236632538,
      "grad_norm": 1.3445524633570782,
      "learning_rate": 9.359511143449794e-07,
      "loss": 0.0476,
      "step": 7346
    },
    {
      "epoch": 1.6716723549488055,
      "grad_norm": 0.6942290235140224,
      "learning_rate": 9.358736032265788e-07,
      "loss": 0.0179,
      "step": 7347
    },
    {
      "epoch": 1.6718998862343573,
      "grad_norm": 1.4997610759007964,
      "learning_rate": 9.357960857544912e-07,
      "loss": 0.0142,
      "step": 7348
    },
    {
      "epoch": 1.672127417519909,
      "grad_norm": 1.3145764575012553,
      "learning_rate": 9.357185619303009e-07,
      "loss": 0.023,
      "step": 7349
    },
    {
      "epoch": 1.6723549488054608,
      "grad_norm": 1.079215509391157,
      "learning_rate": 9.356410317555922e-07,
      "loss": 0.0315,
      "step": 7350
    },
    {
      "epoch": 1.6725824800910125,
      "grad_norm": 1.029924502412833,
      "learning_rate": 9.355634952319498e-07,
      "loss": 0.0183,
      "step": 7351
    },
    {
      "epoch": 1.6728100113765643,
      "grad_norm": 1.2293364013970451,
      "learning_rate": 9.354859523609583e-07,
      "loss": 0.0252,
      "step": 7352
    },
    {
      "epoch": 1.673037542662116,
      "grad_norm": 0.9643322571862788,
      "learning_rate": 9.354084031442027e-07,
      "loss": 0.0226,
      "step": 7353
    },
    {
      "epoch": 1.6732650739476678,
      "grad_norm": 1.57517031227939,
      "learning_rate": 9.353308475832676e-07,
      "loss": 0.0383,
      "step": 7354
    },
    {
      "epoch": 1.6734926052332195,
      "grad_norm": 0.9929231596528064,
      "learning_rate": 9.352532856797382e-07,
      "loss": 0.0227,
      "step": 7355
    },
    {
      "epoch": 1.6737201365187713,
      "grad_norm": 0.6182995644994113,
      "learning_rate": 9.351757174352e-07,
      "loss": 0.0106,
      "step": 7356
    },
    {
      "epoch": 1.673947667804323,
      "grad_norm": 1.3596248896377752,
      "learning_rate": 9.350981428512383e-07,
      "loss": 0.0203,
      "step": 7357
    },
    {
      "epoch": 1.6741751990898748,
      "grad_norm": 1.124354179944877,
      "learning_rate": 9.350205619294383e-07,
      "loss": 0.0285,
      "step": 7358
    },
    {
      "epoch": 1.6744027303754265,
      "grad_norm": 1.3116324776794714,
      "learning_rate": 9.349429746713859e-07,
      "loss": 0.0271,
      "step": 7359
    },
    {
      "epoch": 1.6746302616609783,
      "grad_norm": 1.3279079181685614,
      "learning_rate": 9.348653810786667e-07,
      "loss": 0.0402,
      "step": 7360
    },
    {
      "epoch": 1.67485779294653,
      "grad_norm": 1.265035440034809,
      "learning_rate": 9.347877811528666e-07,
      "loss": 0.0562,
      "step": 7361
    },
    {
      "epoch": 1.6750853242320818,
      "grad_norm": 1.6657212776082273,
      "learning_rate": 9.347101748955715e-07,
      "loss": 0.0492,
      "step": 7362
    },
    {
      "epoch": 1.6753128555176335,
      "grad_norm": 1.4213794447450105,
      "learning_rate": 9.346325623083679e-07,
      "loss": 0.0293,
      "step": 7363
    },
    {
      "epoch": 1.6755403868031853,
      "grad_norm": 1.7636702749688,
      "learning_rate": 9.345549433928416e-07,
      "loss": 0.035,
      "step": 7364
    },
    {
      "epoch": 1.675767918088737,
      "grad_norm": 1.589993461555999,
      "learning_rate": 9.344773181505793e-07,
      "loss": 0.0368,
      "step": 7365
    },
    {
      "epoch": 1.675995449374289,
      "grad_norm": 0.9830121986788878,
      "learning_rate": 9.343996865831673e-07,
      "loss": 0.0286,
      "step": 7366
    },
    {
      "epoch": 1.6762229806598408,
      "grad_norm": 1.110294495379522,
      "learning_rate": 9.343220486921924e-07,
      "loss": 0.0201,
      "step": 7367
    },
    {
      "epoch": 1.6764505119453925,
      "grad_norm": 0.9378938801043143,
      "learning_rate": 9.342444044792412e-07,
      "loss": 0.0161,
      "step": 7368
    },
    {
      "epoch": 1.6766780432309443,
      "grad_norm": 1.0705872722742766,
      "learning_rate": 9.341667539459006e-07,
      "loss": 0.0288,
      "step": 7369
    },
    {
      "epoch": 1.676905574516496,
      "grad_norm": 0.9448376066075745,
      "learning_rate": 9.340890970937583e-07,
      "loss": 0.0204,
      "step": 7370
    },
    {
      "epoch": 1.6771331058020478,
      "grad_norm": 0.5664806207314872,
      "learning_rate": 9.340114339244006e-07,
      "loss": 0.0095,
      "step": 7371
    },
    {
      "epoch": 1.6773606370875995,
      "grad_norm": 0.8796169735391105,
      "learning_rate": 9.33933764439415e-07,
      "loss": 0.0151,
      "step": 7372
    },
    {
      "epoch": 1.6775881683731513,
      "grad_norm": 1.1600519212694234,
      "learning_rate": 9.338560886403891e-07,
      "loss": 0.0406,
      "step": 7373
    },
    {
      "epoch": 1.677815699658703,
      "grad_norm": 1.0289543332326065,
      "learning_rate": 9.337784065289104e-07,
      "loss": 0.0193,
      "step": 7374
    },
    {
      "epoch": 1.6780432309442548,
      "grad_norm": 1.826700733249313,
      "learning_rate": 9.337007181065667e-07,
      "loss": 0.0701,
      "step": 7375
    },
    {
      "epoch": 1.6782707622298068,
      "grad_norm": 1.2835558531886389,
      "learning_rate": 9.336230233749456e-07,
      "loss": 0.0076,
      "step": 7376
    },
    {
      "epoch": 1.6784982935153585,
      "grad_norm": 0.7659024994689081,
      "learning_rate": 9.335453223356351e-07,
      "loss": 0.0114,
      "step": 7377
    },
    {
      "epoch": 1.6787258248009103,
      "grad_norm": 0.9334888188139624,
      "learning_rate": 9.334676149902233e-07,
      "loss": 0.018,
      "step": 7378
    },
    {
      "epoch": 1.678953356086462,
      "grad_norm": 0.9494022161647141,
      "learning_rate": 9.333899013402984e-07,
      "loss": 0.0234,
      "step": 7379
    },
    {
      "epoch": 1.6791808873720138,
      "grad_norm": 1.139412640514536,
      "learning_rate": 9.333121813874486e-07,
      "loss": 0.0264,
      "step": 7380
    },
    {
      "epoch": 1.6794084186575655,
      "grad_norm": 1.3475699049472754,
      "learning_rate": 9.332344551332626e-07,
      "loss": 0.0236,
      "step": 7381
    },
    {
      "epoch": 1.6796359499431173,
      "grad_norm": 1.1919998065647228,
      "learning_rate": 9.331567225793288e-07,
      "loss": 0.034,
      "step": 7382
    },
    {
      "epoch": 1.679863481228669,
      "grad_norm": 0.8936854196890182,
      "learning_rate": 9.330789837272359e-07,
      "loss": 0.0238,
      "step": 7383
    },
    {
      "epoch": 1.6800910125142208,
      "grad_norm": 1.846647841510941,
      "learning_rate": 9.330012385785729e-07,
      "loss": 0.0722,
      "step": 7384
    },
    {
      "epoch": 1.6803185437997725,
      "grad_norm": 1.193450876221038,
      "learning_rate": 9.329234871349285e-07,
      "loss": 0.036,
      "step": 7385
    },
    {
      "epoch": 1.6805460750853243,
      "grad_norm": 0.8992689459727948,
      "learning_rate": 9.328457293978921e-07,
      "loss": 0.0089,
      "step": 7386
    },
    {
      "epoch": 1.680773606370876,
      "grad_norm": 0.9963987302000527,
      "learning_rate": 9.327679653690527e-07,
      "loss": 0.0235,
      "step": 7387
    },
    {
      "epoch": 1.6810011376564278,
      "grad_norm": 1.4603172265236832,
      "learning_rate": 9.326901950499997e-07,
      "loss": 0.0596,
      "step": 7388
    },
    {
      "epoch": 1.6812286689419795,
      "grad_norm": 1.043900668728106,
      "learning_rate": 9.326124184423228e-07,
      "loss": 0.0163,
      "step": 7389
    },
    {
      "epoch": 1.6814562002275313,
      "grad_norm": 1.0382331801469895,
      "learning_rate": 9.325346355476114e-07,
      "loss": 0.0208,
      "step": 7390
    },
    {
      "epoch": 1.681683731513083,
      "grad_norm": 0.893778247022103,
      "learning_rate": 9.324568463674552e-07,
      "loss": 0.0251,
      "step": 7391
    },
    {
      "epoch": 1.6819112627986348,
      "grad_norm": 1.3602050853170236,
      "learning_rate": 9.323790509034441e-07,
      "loss": 0.0275,
      "step": 7392
    },
    {
      "epoch": 1.6821387940841865,
      "grad_norm": 1.0857184214021618,
      "learning_rate": 9.323012491571682e-07,
      "loss": 0.0191,
      "step": 7393
    },
    {
      "epoch": 1.6823663253697383,
      "grad_norm": 0.8532593287473691,
      "learning_rate": 9.322234411302175e-07,
      "loss": 0.0121,
      "step": 7394
    },
    {
      "epoch": 1.68259385665529,
      "grad_norm": 0.8830902866673185,
      "learning_rate": 9.321456268241824e-07,
      "loss": 0.0241,
      "step": 7395
    },
    {
      "epoch": 1.6828213879408418,
      "grad_norm": 0.7559806353099447,
      "learning_rate": 9.320678062406531e-07,
      "loss": 0.0076,
      "step": 7396
    },
    {
      "epoch": 1.6830489192263935,
      "grad_norm": 0.9228198263065078,
      "learning_rate": 9.319899793812204e-07,
      "loss": 0.0145,
      "step": 7397
    },
    {
      "epoch": 1.6832764505119453,
      "grad_norm": 1.1206750032487018,
      "learning_rate": 9.319121462474745e-07,
      "loss": 0.0285,
      "step": 7398
    },
    {
      "epoch": 1.683503981797497,
      "grad_norm": 0.9495217613214355,
      "learning_rate": 9.318343068410065e-07,
      "loss": 0.0201,
      "step": 7399
    },
    {
      "epoch": 1.6837315130830488,
      "grad_norm": 1.2258328125547402,
      "learning_rate": 9.317564611634072e-07,
      "loss": 0.025,
      "step": 7400
    },
    {
      "epoch": 1.6839590443686006,
      "grad_norm": 0.8280424203512745,
      "learning_rate": 9.316786092162678e-07,
      "loss": 0.0189,
      "step": 7401
    },
    {
      "epoch": 1.6841865756541523,
      "grad_norm": 1.1336003257642784,
      "learning_rate": 9.31600751001179e-07,
      "loss": 0.0225,
      "step": 7402
    },
    {
      "epoch": 1.684414106939704,
      "grad_norm": 1.69117863085864,
      "learning_rate": 9.315228865197326e-07,
      "loss": 0.0549,
      "step": 7403
    },
    {
      "epoch": 1.6846416382252558,
      "grad_norm": 1.110342505920422,
      "learning_rate": 9.314450157735197e-07,
      "loss": 0.0196,
      "step": 7404
    },
    {
      "epoch": 1.6848691695108078,
      "grad_norm": 68.47354332095117,
      "learning_rate": 9.313671387641318e-07,
      "loss": 1.134,
      "step": 7405
    },
    {
      "epoch": 1.6850967007963595,
      "grad_norm": 1.1485468681711508,
      "learning_rate": 9.312892554931608e-07,
      "loss": 0.0197,
      "step": 7406
    },
    {
      "epoch": 1.6853242320819113,
      "grad_norm": 1.3445765903343956,
      "learning_rate": 9.312113659621984e-07,
      "loss": 0.0476,
      "step": 7407
    },
    {
      "epoch": 1.685551763367463,
      "grad_norm": 1.3255216302851847,
      "learning_rate": 9.311334701728364e-07,
      "loss": 0.027,
      "step": 7408
    },
    {
      "epoch": 1.6857792946530148,
      "grad_norm": 0.883532434592329,
      "learning_rate": 9.310555681266668e-07,
      "loss": 0.0155,
      "step": 7409
    },
    {
      "epoch": 1.6860068259385665,
      "grad_norm": 0.8561859578890467,
      "learning_rate": 9.30977659825282e-07,
      "loss": 0.0175,
      "step": 7410
    },
    {
      "epoch": 1.6862343572241183,
      "grad_norm": 0.8447963217284506,
      "learning_rate": 9.308997452702743e-07,
      "loss": 0.0196,
      "step": 7411
    },
    {
      "epoch": 1.68646188850967,
      "grad_norm": 1.0538999067728367,
      "learning_rate": 9.308218244632358e-07,
      "loss": 0.0325,
      "step": 7412
    },
    {
      "epoch": 1.6866894197952218,
      "grad_norm": 1.1841066018440165,
      "learning_rate": 9.307438974057595e-07,
      "loss": 0.032,
      "step": 7413
    },
    {
      "epoch": 1.6869169510807736,
      "grad_norm": 1.1766023443596012,
      "learning_rate": 9.306659640994381e-07,
      "loss": 0.0242,
      "step": 7414
    },
    {
      "epoch": 1.6871444823663255,
      "grad_norm": 1.1353504481993184,
      "learning_rate": 9.305880245458637e-07,
      "loss": 0.0427,
      "step": 7415
    },
    {
      "epoch": 1.6873720136518773,
      "grad_norm": 0.9494360095379312,
      "learning_rate": 9.305100787466301e-07,
      "loss": 0.0199,
      "step": 7416
    },
    {
      "epoch": 1.687599544937429,
      "grad_norm": 1.1011922337725744,
      "learning_rate": 9.304321267033298e-07,
      "loss": 0.0101,
      "step": 7417
    },
    {
      "epoch": 1.6878270762229808,
      "grad_norm": 0.7887872994259693,
      "learning_rate": 9.303541684175563e-07,
      "loss": 0.0228,
      "step": 7418
    },
    {
      "epoch": 1.6880546075085325,
      "grad_norm": 1.071159027848962,
      "learning_rate": 9.302762038909028e-07,
      "loss": 0.0248,
      "step": 7419
    },
    {
      "epoch": 1.6882821387940843,
      "grad_norm": 0.9825876397165821,
      "learning_rate": 9.301982331249629e-07,
      "loss": 0.0169,
      "step": 7420
    },
    {
      "epoch": 1.688509670079636,
      "grad_norm": 0.8027691821074338,
      "learning_rate": 9.301202561213298e-07,
      "loss": 0.0212,
      "step": 7421
    },
    {
      "epoch": 1.6887372013651878,
      "grad_norm": 1.7674718194893417,
      "learning_rate": 9.300422728815976e-07,
      "loss": 0.0408,
      "step": 7422
    },
    {
      "epoch": 1.6889647326507395,
      "grad_norm": 1.0714477140147776,
      "learning_rate": 9.2996428340736e-07,
      "loss": 0.016,
      "step": 7423
    },
    {
      "epoch": 1.6891922639362913,
      "grad_norm": 0.9724404293673335,
      "learning_rate": 9.29886287700211e-07,
      "loss": 0.0421,
      "step": 7424
    },
    {
      "epoch": 1.689419795221843,
      "grad_norm": 0.7905940535995999,
      "learning_rate": 9.298082857617446e-07,
      "loss": 0.0192,
      "step": 7425
    },
    {
      "epoch": 1.6896473265073948,
      "grad_norm": 1.026772059663101,
      "learning_rate": 9.297302775935552e-07,
      "loss": 0.0206,
      "step": 7426
    },
    {
      "epoch": 1.6898748577929465,
      "grad_norm": 2.099313726248406,
      "learning_rate": 9.296522631972368e-07,
      "loss": 0.0546,
      "step": 7427
    },
    {
      "epoch": 1.6901023890784983,
      "grad_norm": 1.1080974714503837,
      "learning_rate": 9.295742425743842e-07,
      "loss": 0.0292,
      "step": 7428
    },
    {
      "epoch": 1.69032992036405,
      "grad_norm": 0.9493726656722671,
      "learning_rate": 9.294962157265918e-07,
      "loss": 0.0203,
      "step": 7429
    },
    {
      "epoch": 1.6905574516496018,
      "grad_norm": 0.6548379026825043,
      "learning_rate": 9.294181826554547e-07,
      "loss": 0.0122,
      "step": 7430
    },
    {
      "epoch": 1.6907849829351536,
      "grad_norm": 0.7572188450490555,
      "learning_rate": 9.29340143362567e-07,
      "loss": 0.0247,
      "step": 7431
    },
    {
      "epoch": 1.6910125142207053,
      "grad_norm": 1.2752020698204618,
      "learning_rate": 9.292620978495246e-07,
      "loss": 0.0319,
      "step": 7432
    },
    {
      "epoch": 1.691240045506257,
      "grad_norm": 1.4114210445705098,
      "learning_rate": 9.291840461179219e-07,
      "loss": 0.0457,
      "step": 7433
    },
    {
      "epoch": 1.6914675767918088,
      "grad_norm": 0.816540485123266,
      "learning_rate": 9.291059881693544e-07,
      "loss": 0.0135,
      "step": 7434
    },
    {
      "epoch": 1.6916951080773606,
      "grad_norm": 1.280629632454358,
      "learning_rate": 9.290279240054176e-07,
      "loss": 0.0228,
      "step": 7435
    },
    {
      "epoch": 1.6919226393629123,
      "grad_norm": 1.4406386686448849,
      "learning_rate": 9.289498536277066e-07,
      "loss": 0.0536,
      "step": 7436
    },
    {
      "epoch": 1.692150170648464,
      "grad_norm": 1.5606918794122948,
      "learning_rate": 9.288717770378172e-07,
      "loss": 0.0312,
      "step": 7437
    },
    {
      "epoch": 1.6923777019340158,
      "grad_norm": 1.2240851387654794,
      "learning_rate": 9.287936942373454e-07,
      "loss": 0.0286,
      "step": 7438
    },
    {
      "epoch": 1.6926052332195676,
      "grad_norm": 1.2141319640587955,
      "learning_rate": 9.287156052278869e-07,
      "loss": 0.0205,
      "step": 7439
    },
    {
      "epoch": 1.6928327645051193,
      "grad_norm": 1.0529105375576406,
      "learning_rate": 9.286375100110376e-07,
      "loss": 0.0239,
      "step": 7440
    },
    {
      "epoch": 1.693060295790671,
      "grad_norm": 1.1278735283615677,
      "learning_rate": 9.285594085883937e-07,
      "loss": 0.0151,
      "step": 7441
    },
    {
      "epoch": 1.6932878270762228,
      "grad_norm": 0.9651776105310846,
      "learning_rate": 9.284813009615512e-07,
      "loss": 0.0294,
      "step": 7442
    },
    {
      "epoch": 1.6935153583617746,
      "grad_norm": 1.3435219511382133,
      "learning_rate": 9.28403187132107e-07,
      "loss": 0.0446,
      "step": 7443
    },
    {
      "epoch": 1.6937428896473266,
      "grad_norm": 1.3393538999942238,
      "learning_rate": 9.28325067101657e-07,
      "loss": 0.0313,
      "step": 7444
    },
    {
      "epoch": 1.6939704209328783,
      "grad_norm": 1.2636248804704842,
      "learning_rate": 9.282469408717984e-07,
      "loss": 0.0351,
      "step": 7445
    },
    {
      "epoch": 1.69419795221843,
      "grad_norm": 0.7870749416554453,
      "learning_rate": 9.281688084441277e-07,
      "loss": 0.0214,
      "step": 7446
    },
    {
      "epoch": 1.6944254835039818,
      "grad_norm": 1.8762587310238628,
      "learning_rate": 9.280906698202417e-07,
      "loss": 0.0515,
      "step": 7447
    },
    {
      "epoch": 1.6946530147895336,
      "grad_norm": 1.6998422130770203,
      "learning_rate": 9.280125250017375e-07,
      "loss": 0.0239,
      "step": 7448
    },
    {
      "epoch": 1.6948805460750853,
      "grad_norm": 0.9027847291427934,
      "learning_rate": 9.279343739902122e-07,
      "loss": 0.017,
      "step": 7449
    },
    {
      "epoch": 1.695108077360637,
      "grad_norm": 1.1062164148509734,
      "learning_rate": 9.278562167872632e-07,
      "loss": 0.0235,
      "step": 7450
    },
    {
      "epoch": 1.6953356086461888,
      "grad_norm": 1.812784584203428,
      "learning_rate": 9.277780533944879e-07,
      "loss": 0.0443,
      "step": 7451
    },
    {
      "epoch": 1.6955631399317406,
      "grad_norm": 0.9038757591065284,
      "learning_rate": 9.276998838134834e-07,
      "loss": 0.0233,
      "step": 7452
    },
    {
      "epoch": 1.6957906712172923,
      "grad_norm": 0.9025973802036622,
      "learning_rate": 9.276217080458478e-07,
      "loss": 0.0189,
      "step": 7453
    },
    {
      "epoch": 1.6960182025028443,
      "grad_norm": 1.081066833196723,
      "learning_rate": 9.275435260931786e-07,
      "loss": 0.0291,
      "step": 7454
    },
    {
      "epoch": 1.696245733788396,
      "grad_norm": 1.4646992378969927,
      "learning_rate": 9.274653379570739e-07,
      "loss": 0.0387,
      "step": 7455
    },
    {
      "epoch": 1.6964732650739478,
      "grad_norm": 0.9392167588281297,
      "learning_rate": 9.273871436391315e-07,
      "loss": 0.0394,
      "step": 7456
    },
    {
      "epoch": 1.6967007963594996,
      "grad_norm": 1.3086290254658115,
      "learning_rate": 9.273089431409499e-07,
      "loss": 0.0305,
      "step": 7457
    },
    {
      "epoch": 1.6969283276450513,
      "grad_norm": 0.9762633653670589,
      "learning_rate": 9.272307364641271e-07,
      "loss": 0.0202,
      "step": 7458
    },
    {
      "epoch": 1.697155858930603,
      "grad_norm": 1.5684419166838992,
      "learning_rate": 9.271525236102614e-07,
      "loss": 0.0353,
      "step": 7459
    },
    {
      "epoch": 1.6973833902161548,
      "grad_norm": 1.3066854597715047,
      "learning_rate": 9.270743045809516e-07,
      "loss": 0.0173,
      "step": 7460
    },
    {
      "epoch": 1.6976109215017066,
      "grad_norm": 1.2901196273408886,
      "learning_rate": 9.269960793777963e-07,
      "loss": 0.0259,
      "step": 7461
    },
    {
      "epoch": 1.6978384527872583,
      "grad_norm": 1.4716393793331854,
      "learning_rate": 9.26917848002394e-07,
      "loss": 0.029,
      "step": 7462
    },
    {
      "epoch": 1.69806598407281,
      "grad_norm": 2.9483009434833374,
      "learning_rate": 9.26839610456344e-07,
      "loss": 0.02,
      "step": 7463
    },
    {
      "epoch": 1.6982935153583618,
      "grad_norm": 0.780802510505116,
      "learning_rate": 9.267613667412453e-07,
      "loss": 0.0226,
      "step": 7464
    },
    {
      "epoch": 1.6985210466439136,
      "grad_norm": 1.1866004688183827,
      "learning_rate": 9.266831168586968e-07,
      "loss": 0.0265,
      "step": 7465
    },
    {
      "epoch": 1.6987485779294653,
      "grad_norm": 0.7694346635219753,
      "learning_rate": 9.266048608102978e-07,
      "loss": 0.0202,
      "step": 7466
    },
    {
      "epoch": 1.698976109215017,
      "grad_norm": 1.106780796505893,
      "learning_rate": 9.265265985976478e-07,
      "loss": 0.0175,
      "step": 7467
    },
    {
      "epoch": 1.6992036405005688,
      "grad_norm": 0.8669229898039597,
      "learning_rate": 9.264483302223464e-07,
      "loss": 0.0207,
      "step": 7468
    },
    {
      "epoch": 1.6994311717861206,
      "grad_norm": 1.5196225362339568,
      "learning_rate": 9.263700556859931e-07,
      "loss": 0.0391,
      "step": 7469
    },
    {
      "epoch": 1.6996587030716723,
      "grad_norm": 0.813734490997272,
      "learning_rate": 9.26291774990188e-07,
      "loss": 0.0277,
      "step": 7470
    },
    {
      "epoch": 1.699886234357224,
      "grad_norm": 0.7735285752321128,
      "learning_rate": 9.262134881365307e-07,
      "loss": 0.0197,
      "step": 7471
    },
    {
      "epoch": 1.7001137656427758,
      "grad_norm": 1.4230702281863177,
      "learning_rate": 9.261351951266211e-07,
      "loss": 0.0292,
      "step": 7472
    },
    {
      "epoch": 1.7003412969283276,
      "grad_norm": 2.4934340707549993,
      "learning_rate": 9.2605689596206e-07,
      "loss": 0.065,
      "step": 7473
    },
    {
      "epoch": 1.7005688282138793,
      "grad_norm": 1.265713576588293,
      "learning_rate": 9.259785906444473e-07,
      "loss": 0.0451,
      "step": 7474
    },
    {
      "epoch": 1.700796359499431,
      "grad_norm": 1.107331460710875,
      "learning_rate": 9.259002791753832e-07,
      "loss": 0.031,
      "step": 7475
    },
    {
      "epoch": 1.7010238907849828,
      "grad_norm": 0.9656516950035509,
      "learning_rate": 9.258219615564684e-07,
      "loss": 0.0157,
      "step": 7476
    },
    {
      "epoch": 1.7012514220705346,
      "grad_norm": 1.6648471363780213,
      "learning_rate": 9.25743637789304e-07,
      "loss": 0.0159,
      "step": 7477
    },
    {
      "epoch": 1.7014789533560863,
      "grad_norm": 0.8136434066676951,
      "learning_rate": 9.256653078754903e-07,
      "loss": 0.025,
      "step": 7478
    },
    {
      "epoch": 1.701706484641638,
      "grad_norm": 1.729468058214187,
      "learning_rate": 9.255869718166281e-07,
      "loss": 0.0448,
      "step": 7479
    },
    {
      "epoch": 1.7019340159271898,
      "grad_norm": 1.747237104201101,
      "learning_rate": 9.255086296143189e-07,
      "loss": 0.0421,
      "step": 7480
    },
    {
      "epoch": 1.7021615472127416,
      "grad_norm": 1.1319667627658456,
      "learning_rate": 9.254302812701636e-07,
      "loss": 0.0203,
      "step": 7481
    },
    {
      "epoch": 1.7023890784982934,
      "grad_norm": 6.34869179504398,
      "learning_rate": 9.253519267857637e-07,
      "loss": 0.1211,
      "step": 7482
    },
    {
      "epoch": 1.7026166097838453,
      "grad_norm": 1.342839250597215,
      "learning_rate": 9.252735661627204e-07,
      "loss": 0.0301,
      "step": 7483
    },
    {
      "epoch": 1.702844141069397,
      "grad_norm": 1.1568654598416463,
      "learning_rate": 9.251951994026353e-07,
      "loss": 0.0202,
      "step": 7484
    },
    {
      "epoch": 1.7030716723549488,
      "grad_norm": 0.9603437591283284,
      "learning_rate": 9.251168265071101e-07,
      "loss": 0.0276,
      "step": 7485
    },
    {
      "epoch": 1.7032992036405006,
      "grad_norm": 1.1938581829579682,
      "learning_rate": 9.250384474777465e-07,
      "loss": 0.0221,
      "step": 7486
    },
    {
      "epoch": 1.7035267349260523,
      "grad_norm": 0.7456001958336463,
      "learning_rate": 9.249600623161467e-07,
      "loss": 0.0262,
      "step": 7487
    },
    {
      "epoch": 1.703754266211604,
      "grad_norm": 1.4700522344198281,
      "learning_rate": 9.248816710239125e-07,
      "loss": 0.0395,
      "step": 7488
    },
    {
      "epoch": 1.7039817974971558,
      "grad_norm": 1.1506393584869568,
      "learning_rate": 9.248032736026463e-07,
      "loss": 0.0233,
      "step": 7489
    },
    {
      "epoch": 1.7042093287827076,
      "grad_norm": 1.1195835539109666,
      "learning_rate": 9.247248700539502e-07,
      "loss": 0.0217,
      "step": 7490
    },
    {
      "epoch": 1.7044368600682593,
      "grad_norm": 2.465921051795379,
      "learning_rate": 9.246464603794266e-07,
      "loss": 0.0532,
      "step": 7491
    },
    {
      "epoch": 1.7046643913538113,
      "grad_norm": 1.3355635126606438,
      "learning_rate": 9.245680445806782e-07,
      "loss": 0.0516,
      "step": 7492
    },
    {
      "epoch": 1.704891922639363,
      "grad_norm": 0.9992091005260795,
      "learning_rate": 9.244896226593074e-07,
      "loss": 0.0284,
      "step": 7493
    },
    {
      "epoch": 1.7051194539249148,
      "grad_norm": 0.8019991772837796,
      "learning_rate": 9.244111946169173e-07,
      "loss": 0.0132,
      "step": 7494
    },
    {
      "epoch": 1.7053469852104666,
      "grad_norm": 1.0414066656735075,
      "learning_rate": 9.243327604551109e-07,
      "loss": 0.0203,
      "step": 7495
    },
    {
      "epoch": 1.7055745164960183,
      "grad_norm": 1.0467030740861185,
      "learning_rate": 9.242543201754908e-07,
      "loss": 0.0217,
      "step": 7496
    },
    {
      "epoch": 1.70580204778157,
      "grad_norm": 1.7898345976143868,
      "learning_rate": 9.241758737796608e-07,
      "loss": 0.0366,
      "step": 7497
    },
    {
      "epoch": 1.7060295790671218,
      "grad_norm": 1.4837502642765534,
      "learning_rate": 9.240974212692235e-07,
      "loss": 0.0316,
      "step": 7498
    },
    {
      "epoch": 1.7062571103526736,
      "grad_norm": 1.09132144334733,
      "learning_rate": 9.240189626457828e-07,
      "loss": 0.0389,
      "step": 7499
    },
    {
      "epoch": 1.7064846416382253,
      "grad_norm": 1.192777370323259,
      "learning_rate": 9.239404979109422e-07,
      "loss": 0.0171,
      "step": 7500
    },
    {
      "epoch": 1.706712172923777,
      "grad_norm": 1.4297979166462436,
      "learning_rate": 9.238620270663053e-07,
      "loss": 0.0213,
      "step": 7501
    },
    {
      "epoch": 1.7069397042093288,
      "grad_norm": 0.948422819607699,
      "learning_rate": 9.237835501134759e-07,
      "loss": 0.0134,
      "step": 7502
    },
    {
      "epoch": 1.7071672354948806,
      "grad_norm": 1.2121992571283962,
      "learning_rate": 9.237050670540579e-07,
      "loss": 0.0184,
      "step": 7503
    },
    {
      "epoch": 1.7073947667804323,
      "grad_norm": 0.8297872919210153,
      "learning_rate": 9.236265778896554e-07,
      "loss": 0.0194,
      "step": 7504
    },
    {
      "epoch": 1.707622298065984,
      "grad_norm": 0.9096517797805543,
      "learning_rate": 9.235480826218726e-07,
      "loss": 0.0222,
      "step": 7505
    },
    {
      "epoch": 1.7078498293515358,
      "grad_norm": 0.9438726780665672,
      "learning_rate": 9.234695812523137e-07,
      "loss": 0.017,
      "step": 7506
    },
    {
      "epoch": 1.7080773606370876,
      "grad_norm": 1.6086698482204602,
      "learning_rate": 9.233910737825831e-07,
      "loss": 0.0424,
      "step": 7507
    },
    {
      "epoch": 1.7083048919226393,
      "grad_norm": 1.142830190893745,
      "learning_rate": 9.233125602142856e-07,
      "loss": 0.0186,
      "step": 7508
    },
    {
      "epoch": 1.708532423208191,
      "grad_norm": 1.6320142128089656,
      "learning_rate": 9.232340405490256e-07,
      "loss": 0.0347,
      "step": 7509
    },
    {
      "epoch": 1.7087599544937428,
      "grad_norm": 0.9514508167221402,
      "learning_rate": 9.23155514788408e-07,
      "loss": 0.0263,
      "step": 7510
    },
    {
      "epoch": 1.7089874857792946,
      "grad_norm": 0.6544786260782971,
      "learning_rate": 9.230769829340378e-07,
      "loss": 0.0102,
      "step": 7511
    },
    {
      "epoch": 1.7092150170648464,
      "grad_norm": 0.692110961995533,
      "learning_rate": 9.229984449875199e-07,
      "loss": 0.0136,
      "step": 7512
    },
    {
      "epoch": 1.709442548350398,
      "grad_norm": 1.9764741743177798,
      "learning_rate": 9.229199009504594e-07,
      "loss": 0.0235,
      "step": 7513
    },
    {
      "epoch": 1.7096700796359499,
      "grad_norm": 0.734190462175171,
      "learning_rate": 9.228413508244621e-07,
      "loss": 0.0082,
      "step": 7514
    },
    {
      "epoch": 1.7098976109215016,
      "grad_norm": 0.8945274189963984,
      "learning_rate": 9.227627946111328e-07,
      "loss": 0.0227,
      "step": 7515
    },
    {
      "epoch": 1.7101251422070534,
      "grad_norm": 3.5912888738002664,
      "learning_rate": 9.226842323120773e-07,
      "loss": 0.1081,
      "step": 7516
    },
    {
      "epoch": 1.7103526734926051,
      "grad_norm": 1.560919222763877,
      "learning_rate": 9.226056639289013e-07,
      "loss": 0.0337,
      "step": 7517
    },
    {
      "epoch": 1.7105802047781569,
      "grad_norm": 1.095784516414262,
      "learning_rate": 9.225270894632107e-07,
      "loss": 0.0115,
      "step": 7518
    },
    {
      "epoch": 1.7108077360637086,
      "grad_norm": 0.9859467458505337,
      "learning_rate": 9.224485089166111e-07,
      "loss": 0.0195,
      "step": 7519
    },
    {
      "epoch": 1.7110352673492604,
      "grad_norm": 0.7148271370823988,
      "learning_rate": 9.223699222907088e-07,
      "loss": 0.0149,
      "step": 7520
    },
    {
      "epoch": 1.7112627986348121,
      "grad_norm": 0.8875065071349563,
      "learning_rate": 9.222913295871101e-07,
      "loss": 0.0184,
      "step": 7521
    },
    {
      "epoch": 1.711490329920364,
      "grad_norm": 1.2178376423136983,
      "learning_rate": 9.22212730807421e-07,
      "loss": 0.0461,
      "step": 7522
    },
    {
      "epoch": 1.7117178612059158,
      "grad_norm": 1.0997910738708634,
      "learning_rate": 9.221341259532476e-07,
      "loss": 0.0255,
      "step": 7523
    },
    {
      "epoch": 1.7119453924914676,
      "grad_norm": 1.609907335399055,
      "learning_rate": 9.220555150261973e-07,
      "loss": 0.0582,
      "step": 7524
    },
    {
      "epoch": 1.7121729237770194,
      "grad_norm": 1.489916582646409,
      "learning_rate": 9.219768980278762e-07,
      "loss": 0.0361,
      "step": 7525
    },
    {
      "epoch": 1.712400455062571,
      "grad_norm": 1.3699399244111667,
      "learning_rate": 9.218982749598911e-07,
      "loss": 0.0314,
      "step": 7526
    },
    {
      "epoch": 1.7126279863481229,
      "grad_norm": 1.1259448383440134,
      "learning_rate": 9.21819645823849e-07,
      "loss": 0.0198,
      "step": 7527
    },
    {
      "epoch": 1.7128555176336746,
      "grad_norm": 1.2000322678352968,
      "learning_rate": 9.21741010621357e-07,
      "loss": 0.0237,
      "step": 7528
    },
    {
      "epoch": 1.7130830489192264,
      "grad_norm": 0.9291990030726819,
      "learning_rate": 9.216623693540222e-07,
      "loss": 0.0154,
      "step": 7529
    },
    {
      "epoch": 1.713310580204778,
      "grad_norm": 0.7752060975128637,
      "learning_rate": 9.215837220234518e-07,
      "loss": 0.0247,
      "step": 7530
    },
    {
      "epoch": 1.71353811149033,
      "grad_norm": 1.0433568166304124,
      "learning_rate": 9.215050686312534e-07,
      "loss": 0.0151,
      "step": 7531
    },
    {
      "epoch": 1.7137656427758818,
      "grad_norm": 0.9686106141353347,
      "learning_rate": 9.214264091790343e-07,
      "loss": 0.0144,
      "step": 7532
    },
    {
      "epoch": 1.7139931740614336,
      "grad_norm": 1.54478146480293,
      "learning_rate": 9.213477436684025e-07,
      "loss": 0.0379,
      "step": 7533
    },
    {
      "epoch": 1.7142207053469853,
      "grad_norm": 1.2519994785023527,
      "learning_rate": 9.212690721009654e-07,
      "loss": 0.0302,
      "step": 7534
    },
    {
      "epoch": 1.714448236632537,
      "grad_norm": 1.7383491670887976,
      "learning_rate": 9.21190394478331e-07,
      "loss": 0.068,
      "step": 7535
    },
    {
      "epoch": 1.7146757679180888,
      "grad_norm": 0.8552599017233935,
      "learning_rate": 9.211117108021074e-07,
      "loss": 0.0168,
      "step": 7536
    },
    {
      "epoch": 1.7149032992036406,
      "grad_norm": 1.788803024478606,
      "learning_rate": 9.210330210739029e-07,
      "loss": 0.0287,
      "step": 7537
    },
    {
      "epoch": 1.7151308304891923,
      "grad_norm": 0.9740522034800336,
      "learning_rate": 9.209543252953254e-07,
      "loss": 0.0234,
      "step": 7538
    },
    {
      "epoch": 1.715358361774744,
      "grad_norm": 1.0280768113489238,
      "learning_rate": 9.208756234679836e-07,
      "loss": 0.0159,
      "step": 7539
    },
    {
      "epoch": 1.7155858930602959,
      "grad_norm": 1.3489600407656455,
      "learning_rate": 9.20796915593486e-07,
      "loss": 0.0449,
      "step": 7540
    },
    {
      "epoch": 1.7158134243458476,
      "grad_norm": 1.4755745930733852,
      "learning_rate": 9.20718201673441e-07,
      "loss": 0.0434,
      "step": 7541
    },
    {
      "epoch": 1.7160409556313994,
      "grad_norm": 0.7616219435437076,
      "learning_rate": 9.206394817094577e-07,
      "loss": 0.0186,
      "step": 7542
    },
    {
      "epoch": 1.716268486916951,
      "grad_norm": 1.1468109557516537,
      "learning_rate": 9.205607557031446e-07,
      "loss": 0.0237,
      "step": 7543
    },
    {
      "epoch": 1.7164960182025029,
      "grad_norm": 0.9443214015433008,
      "learning_rate": 9.204820236561111e-07,
      "loss": 0.0258,
      "step": 7544
    },
    {
      "epoch": 1.7167235494880546,
      "grad_norm": 0.5774152629808543,
      "learning_rate": 9.204032855699663e-07,
      "loss": 0.01,
      "step": 7545
    },
    {
      "epoch": 1.7169510807736064,
      "grad_norm": 1.0276101636351294,
      "learning_rate": 9.203245414463192e-07,
      "loss": 0.0151,
      "step": 7546
    },
    {
      "epoch": 1.7171786120591581,
      "grad_norm": 0.6598829360890991,
      "learning_rate": 9.202457912867795e-07,
      "loss": 0.0089,
      "step": 7547
    },
    {
      "epoch": 1.7174061433447099,
      "grad_norm": 1.3229858553591636,
      "learning_rate": 9.201670350929564e-07,
      "loss": 0.0274,
      "step": 7548
    },
    {
      "epoch": 1.7176336746302616,
      "grad_norm": 0.7485471387835653,
      "learning_rate": 9.200882728664598e-07,
      "loss": 0.0138,
      "step": 7549
    },
    {
      "epoch": 1.7178612059158134,
      "grad_norm": 1.5299363260996441,
      "learning_rate": 9.20009504608899e-07,
      "loss": 0.0518,
      "step": 7550
    },
    {
      "epoch": 1.7180887372013651,
      "grad_norm": 0.7200038697909544,
      "learning_rate": 9.199307303218844e-07,
      "loss": 0.0145,
      "step": 7551
    },
    {
      "epoch": 1.7183162684869169,
      "grad_norm": 1.2509303912139782,
      "learning_rate": 9.198519500070261e-07,
      "loss": 0.0289,
      "step": 7552
    },
    {
      "epoch": 1.7185437997724686,
      "grad_norm": 1.1044637932726378,
      "learning_rate": 9.197731636659335e-07,
      "loss": 0.0326,
      "step": 7553
    },
    {
      "epoch": 1.7187713310580204,
      "grad_norm": 1.5182960629667408,
      "learning_rate": 9.196943713002177e-07,
      "loss": 0.0261,
      "step": 7554
    },
    {
      "epoch": 1.7189988623435721,
      "grad_norm": 0.8343994799574279,
      "learning_rate": 9.196155729114883e-07,
      "loss": 0.0171,
      "step": 7555
    },
    {
      "epoch": 1.7192263936291239,
      "grad_norm": 0.7119142890489658,
      "learning_rate": 9.195367685013564e-07,
      "loss": 0.0075,
      "step": 7556
    },
    {
      "epoch": 1.7194539249146756,
      "grad_norm": 1.2873836111171169,
      "learning_rate": 9.19457958071432e-07,
      "loss": 0.0367,
      "step": 7557
    },
    {
      "epoch": 1.7196814562002274,
      "grad_norm": 1.0237951516204418,
      "learning_rate": 9.193791416233266e-07,
      "loss": 0.0193,
      "step": 7558
    },
    {
      "epoch": 1.7199089874857791,
      "grad_norm": 1.871531878414218,
      "learning_rate": 9.193003191586507e-07,
      "loss": 0.0321,
      "step": 7559
    },
    {
      "epoch": 1.7201365187713311,
      "grad_norm": 0.9889278233426179,
      "learning_rate": 9.192214906790149e-07,
      "loss": 0.0291,
      "step": 7560
    },
    {
      "epoch": 1.7203640500568829,
      "grad_norm": 0.9873975389279321,
      "learning_rate": 9.191426561860308e-07,
      "loss": 0.0221,
      "step": 7561
    },
    {
      "epoch": 1.7205915813424346,
      "grad_norm": 0.9219114577412436,
      "learning_rate": 9.190638156813097e-07,
      "loss": 0.0136,
      "step": 7562
    },
    {
      "epoch": 1.7208191126279864,
      "grad_norm": 0.8251545811829912,
      "learning_rate": 9.189849691664626e-07,
      "loss": 0.0149,
      "step": 7563
    },
    {
      "epoch": 1.7210466439135381,
      "grad_norm": 1.0869299679841706,
      "learning_rate": 9.189061166431012e-07,
      "loss": 0.0376,
      "step": 7564
    },
    {
      "epoch": 1.7212741751990899,
      "grad_norm": 0.6003001398423039,
      "learning_rate": 9.188272581128372e-07,
      "loss": 0.0111,
      "step": 7565
    },
    {
      "epoch": 1.7215017064846416,
      "grad_norm": 1.5253865672646039,
      "learning_rate": 9.187483935772818e-07,
      "loss": 0.0242,
      "step": 7566
    },
    {
      "epoch": 1.7217292377701934,
      "grad_norm": 1.5907519445499267,
      "learning_rate": 9.186695230380474e-07,
      "loss": 0.0296,
      "step": 7567
    },
    {
      "epoch": 1.7219567690557451,
      "grad_norm": 0.9523869090994135,
      "learning_rate": 9.185906464967459e-07,
      "loss": 0.0263,
      "step": 7568
    },
    {
      "epoch": 1.7221843003412969,
      "grad_norm": 1.2465558325527257,
      "learning_rate": 9.185117639549891e-07,
      "loss": 0.0171,
      "step": 7569
    },
    {
      "epoch": 1.7224118316268489,
      "grad_norm": 0.6801386122458417,
      "learning_rate": 9.184328754143893e-07,
      "loss": 0.0104,
      "step": 7570
    },
    {
      "epoch": 1.7226393629124006,
      "grad_norm": 0.7904268653100202,
      "learning_rate": 9.183539808765591e-07,
      "loss": 0.0254,
      "step": 7571
    },
    {
      "epoch": 1.7228668941979524,
      "grad_norm": 0.7570825293218078,
      "learning_rate": 9.182750803431109e-07,
      "loss": 0.0232,
      "step": 7572
    },
    {
      "epoch": 1.723094425483504,
      "grad_norm": 1.2836056735908439,
      "learning_rate": 9.181961738156568e-07,
      "loss": 0.0475,
      "step": 7573
    },
    {
      "epoch": 1.7233219567690559,
      "grad_norm": 1.168137213634141,
      "learning_rate": 9.181172612958101e-07,
      "loss": 0.0236,
      "step": 7574
    },
    {
      "epoch": 1.7235494880546076,
      "grad_norm": 0.47684817828903375,
      "learning_rate": 9.180383427851834e-07,
      "loss": 0.0127,
      "step": 7575
    },
    {
      "epoch": 1.7237770193401594,
      "grad_norm": 1.046578127030525,
      "learning_rate": 9.179594182853898e-07,
      "loss": 0.0175,
      "step": 7576
    },
    {
      "epoch": 1.7240045506257111,
      "grad_norm": 1.1140965944976429,
      "learning_rate": 9.17880487798042e-07,
      "loss": 0.024,
      "step": 7577
    },
    {
      "epoch": 1.7242320819112629,
      "grad_norm": 1.1920191951868853,
      "learning_rate": 9.178015513247534e-07,
      "loss": 0.0184,
      "step": 7578
    },
    {
      "epoch": 1.7244596131968146,
      "grad_norm": 0.7990927056094128,
      "learning_rate": 9.177226088671375e-07,
      "loss": 0.0193,
      "step": 7579
    },
    {
      "epoch": 1.7246871444823664,
      "grad_norm": 0.7907100321498542,
      "learning_rate": 9.176436604268073e-07,
      "loss": 0.0163,
      "step": 7580
    },
    {
      "epoch": 1.7249146757679181,
      "grad_norm": 0.7239726229060026,
      "learning_rate": 9.175647060053767e-07,
      "loss": 0.0297,
      "step": 7581
    },
    {
      "epoch": 1.7251422070534699,
      "grad_norm": 1.2556927924521644,
      "learning_rate": 9.174857456044595e-07,
      "loss": 0.0177,
      "step": 7582
    },
    {
      "epoch": 1.7253697383390216,
      "grad_norm": 0.7280661687160516,
      "learning_rate": 9.174067792256693e-07,
      "loss": 0.0158,
      "step": 7583
    },
    {
      "epoch": 1.7255972696245734,
      "grad_norm": 0.8140510413495747,
      "learning_rate": 9.1732780687062e-07,
      "loss": 0.0187,
      "step": 7584
    },
    {
      "epoch": 1.7258248009101251,
      "grad_norm": 1.2618606105844503,
      "learning_rate": 9.172488285409256e-07,
      "loss": 0.0345,
      "step": 7585
    },
    {
      "epoch": 1.7260523321956769,
      "grad_norm": 1.2138797171822122,
      "learning_rate": 9.171698442382005e-07,
      "loss": 0.0233,
      "step": 7586
    },
    {
      "epoch": 1.7262798634812286,
      "grad_norm": 0.7342179818630192,
      "learning_rate": 9.170908539640587e-07,
      "loss": 0.0095,
      "step": 7587
    },
    {
      "epoch": 1.7265073947667804,
      "grad_norm": 0.8163594897829524,
      "learning_rate": 9.170118577201149e-07,
      "loss": 0.0133,
      "step": 7588
    },
    {
      "epoch": 1.7267349260523321,
      "grad_norm": 0.764870090102491,
      "learning_rate": 9.169328555079836e-07,
      "loss": 0.0148,
      "step": 7589
    },
    {
      "epoch": 1.726962457337884,
      "grad_norm": 1.0034494545193837,
      "learning_rate": 9.168538473292793e-07,
      "loss": 0.0227,
      "step": 7590
    },
    {
      "epoch": 1.7271899886234356,
      "grad_norm": 0.8061146348026804,
      "learning_rate": 9.167748331856169e-07,
      "loss": 0.0153,
      "step": 7591
    },
    {
      "epoch": 1.7274175199089874,
      "grad_norm": 0.7549426350040234,
      "learning_rate": 9.166958130786113e-07,
      "loss": 0.0158,
      "step": 7592
    },
    {
      "epoch": 1.7276450511945391,
      "grad_norm": 0.7542254964161569,
      "learning_rate": 9.166167870098773e-07,
      "loss": 0.0272,
      "step": 7593
    },
    {
      "epoch": 1.727872582480091,
      "grad_norm": 0.7889596892238906,
      "learning_rate": 9.165377549810305e-07,
      "loss": 0.0095,
      "step": 7594
    },
    {
      "epoch": 1.7281001137656427,
      "grad_norm": 1.2357332696364653,
      "learning_rate": 9.164587169936858e-07,
      "loss": 0.0223,
      "step": 7595
    },
    {
      "epoch": 1.7283276450511944,
      "grad_norm": 1.1093238855927807,
      "learning_rate": 9.163796730494587e-07,
      "loss": 0.0165,
      "step": 7596
    },
    {
      "epoch": 1.7285551763367462,
      "grad_norm": 0.8007305366299827,
      "learning_rate": 9.163006231499647e-07,
      "loss": 0.0144,
      "step": 7597
    },
    {
      "epoch": 1.728782707622298,
      "grad_norm": 0.7873052222770126,
      "learning_rate": 9.162215672968194e-07,
      "loss": 0.0131,
      "step": 7598
    },
    {
      "epoch": 1.7290102389078499,
      "grad_norm": 1.4418125843313119,
      "learning_rate": 9.161425054916388e-07,
      "loss": 0.0295,
      "step": 7599
    },
    {
      "epoch": 1.7292377701934016,
      "grad_norm": 1.376158903306575,
      "learning_rate": 9.160634377360383e-07,
      "loss": 0.0285,
      "step": 7600
    },
    {
      "epoch": 1.7294653014789534,
      "grad_norm": 1.2162050247028884,
      "learning_rate": 9.159843640316345e-07,
      "loss": 0.0342,
      "step": 7601
    },
    {
      "epoch": 1.7296928327645051,
      "grad_norm": 1.0911247413585297,
      "learning_rate": 9.159052843800431e-07,
      "loss": 0.0355,
      "step": 7602
    },
    {
      "epoch": 1.729920364050057,
      "grad_norm": 0.8596735837001528,
      "learning_rate": 9.158261987828804e-07,
      "loss": 0.0195,
      "step": 7603
    },
    {
      "epoch": 1.7301478953356086,
      "grad_norm": 1.0547684986800836,
      "learning_rate": 9.157471072417629e-07,
      "loss": 0.0328,
      "step": 7604
    },
    {
      "epoch": 1.7303754266211604,
      "grad_norm": 0.9808997952350144,
      "learning_rate": 9.156680097583071e-07,
      "loss": 0.0156,
      "step": 7605
    },
    {
      "epoch": 1.7306029579067121,
      "grad_norm": 0.9559678639554583,
      "learning_rate": 9.155889063341293e-07,
      "loss": 0.0275,
      "step": 7606
    },
    {
      "epoch": 1.730830489192264,
      "grad_norm": 0.738789350988573,
      "learning_rate": 9.155097969708464e-07,
      "loss": 0.0237,
      "step": 7607
    },
    {
      "epoch": 1.7310580204778157,
      "grad_norm": 1.6370811776774898,
      "learning_rate": 9.154306816700755e-07,
      "loss": 0.0682,
      "step": 7608
    },
    {
      "epoch": 1.7312855517633676,
      "grad_norm": 0.9051106679300437,
      "learning_rate": 9.153515604334334e-07,
      "loss": 0.0116,
      "step": 7609
    },
    {
      "epoch": 1.7315130830489194,
      "grad_norm": 0.8336307762546704,
      "learning_rate": 9.152724332625369e-07,
      "loss": 0.0275,
      "step": 7610
    },
    {
      "epoch": 1.7317406143344711,
      "grad_norm": 1.7561758401491077,
      "learning_rate": 9.151933001590035e-07,
      "loss": 0.032,
      "step": 7611
    },
    {
      "epoch": 1.7319681456200229,
      "grad_norm": 1.0973531743829115,
      "learning_rate": 9.151141611244507e-07,
      "loss": 0.029,
      "step": 7612
    },
    {
      "epoch": 1.7321956769055746,
      "grad_norm": 1.2627804391545918,
      "learning_rate": 9.150350161604957e-07,
      "loss": 0.0528,
      "step": 7613
    },
    {
      "epoch": 1.7324232081911264,
      "grad_norm": 1.6371028258031264,
      "learning_rate": 9.149558652687561e-07,
      "loss": 0.0572,
      "step": 7614
    },
    {
      "epoch": 1.7326507394766781,
      "grad_norm": 0.8989759452735959,
      "learning_rate": 9.148767084508497e-07,
      "loss": 0.0116,
      "step": 7615
    },
    {
      "epoch": 1.7328782707622299,
      "grad_norm": 2.1842929628774055,
      "learning_rate": 9.147975457083943e-07,
      "loss": 0.0329,
      "step": 7616
    },
    {
      "epoch": 1.7331058020477816,
      "grad_norm": 0.5320116937274283,
      "learning_rate": 9.147183770430076e-07,
      "loss": 0.0147,
      "step": 7617
    },
    {
      "epoch": 1.7333333333333334,
      "grad_norm": 1.8104756589418485,
      "learning_rate": 9.146392024563081e-07,
      "loss": 0.0597,
      "step": 7618
    },
    {
      "epoch": 1.7335608646188851,
      "grad_norm": 0.7544718465559345,
      "learning_rate": 9.145600219499137e-07,
      "loss": 0.0192,
      "step": 7619
    },
    {
      "epoch": 1.733788395904437,
      "grad_norm": 1.008324222694622,
      "learning_rate": 9.144808355254426e-07,
      "loss": 0.0122,
      "step": 7620
    },
    {
      "epoch": 1.7340159271899886,
      "grad_norm": 0.8300301314244892,
      "learning_rate": 9.144016431845136e-07,
      "loss": 0.0144,
      "step": 7621
    },
    {
      "epoch": 1.7342434584755404,
      "grad_norm": 0.6402850376707664,
      "learning_rate": 9.143224449287449e-07,
      "loss": 0.0097,
      "step": 7622
    },
    {
      "epoch": 1.7344709897610922,
      "grad_norm": 1.3962807580152192,
      "learning_rate": 9.142432407597552e-07,
      "loss": 0.0289,
      "step": 7623
    },
    {
      "epoch": 1.734698521046644,
      "grad_norm": 0.7192631361443328,
      "learning_rate": 9.141640306791635e-07,
      "loss": 0.0218,
      "step": 7624
    },
    {
      "epoch": 1.7349260523321957,
      "grad_norm": 1.2618423454164023,
      "learning_rate": 9.140848146885888e-07,
      "loss": 0.0185,
      "step": 7625
    },
    {
      "epoch": 1.7351535836177474,
      "grad_norm": 1.1271057373978712,
      "learning_rate": 9.140055927896497e-07,
      "loss": 0.0242,
      "step": 7626
    },
    {
      "epoch": 1.7353811149032992,
      "grad_norm": 0.8881741003317172,
      "learning_rate": 9.139263649839654e-07,
      "loss": 0.0196,
      "step": 7627
    },
    {
      "epoch": 1.735608646188851,
      "grad_norm": 0.8282851305946936,
      "learning_rate": 9.138471312731558e-07,
      "loss": 0.0118,
      "step": 7628
    },
    {
      "epoch": 1.7358361774744027,
      "grad_norm": 1.030456822140469,
      "learning_rate": 9.137678916588395e-07,
      "loss": 0.0201,
      "step": 7629
    },
    {
      "epoch": 1.7360637087599544,
      "grad_norm": 1.2098948742481863,
      "learning_rate": 9.136886461426363e-07,
      "loss": 0.0324,
      "step": 7630
    },
    {
      "epoch": 1.7362912400455062,
      "grad_norm": 0.944387144024131,
      "learning_rate": 9.136093947261659e-07,
      "loss": 0.0349,
      "step": 7631
    },
    {
      "epoch": 1.736518771331058,
      "grad_norm": 1.0435591722724238,
      "learning_rate": 9.135301374110482e-07,
      "loss": 0.0245,
      "step": 7632
    },
    {
      "epoch": 1.7367463026166097,
      "grad_norm": 1.1073705234081634,
      "learning_rate": 9.134508741989028e-07,
      "loss": 0.0149,
      "step": 7633
    },
    {
      "epoch": 1.7369738339021614,
      "grad_norm": 0.6957285837630877,
      "learning_rate": 9.133716050913499e-07,
      "loss": 0.011,
      "step": 7634
    },
    {
      "epoch": 1.7372013651877132,
      "grad_norm": 0.8438881915815134,
      "learning_rate": 9.132923300900096e-07,
      "loss": 0.0194,
      "step": 7635
    },
    {
      "epoch": 1.737428896473265,
      "grad_norm": 1.1755251192640188,
      "learning_rate": 9.132130491965019e-07,
      "loss": 0.0329,
      "step": 7636
    },
    {
      "epoch": 1.7376564277588167,
      "grad_norm": 1.9216126330504404,
      "learning_rate": 9.131337624124473e-07,
      "loss": 0.0547,
      "step": 7637
    },
    {
      "epoch": 1.7378839590443687,
      "grad_norm": 0.861009420801015,
      "learning_rate": 9.130544697394662e-07,
      "loss": 0.0215,
      "step": 7638
    },
    {
      "epoch": 1.7381114903299204,
      "grad_norm": 1.1882707354996638,
      "learning_rate": 9.129751711791796e-07,
      "loss": 0.023,
      "step": 7639
    },
    {
      "epoch": 1.7383390216154722,
      "grad_norm": 1.2239475383326814,
      "learning_rate": 9.128958667332076e-07,
      "loss": 0.0288,
      "step": 7640
    },
    {
      "epoch": 1.738566552901024,
      "grad_norm": 53.355564991987,
      "learning_rate": 9.128165564031715e-07,
      "loss": 0.2876,
      "step": 7641
    },
    {
      "epoch": 1.7387940841865757,
      "grad_norm": 0.8841269788948465,
      "learning_rate": 9.127372401906919e-07,
      "loss": 0.01,
      "step": 7642
    },
    {
      "epoch": 1.7390216154721274,
      "grad_norm": 1.2206625753180147,
      "learning_rate": 9.126579180973904e-07,
      "loss": 0.0415,
      "step": 7643
    },
    {
      "epoch": 1.7392491467576792,
      "grad_norm": 1.086603987389488,
      "learning_rate": 9.125785901248875e-07,
      "loss": 0.0173,
      "step": 7644
    },
    {
      "epoch": 1.739476678043231,
      "grad_norm": 1.1225650990057543,
      "learning_rate": 9.124992562748051e-07,
      "loss": 0.0294,
      "step": 7645
    },
    {
      "epoch": 1.7397042093287827,
      "grad_norm": 1.2175034170150192,
      "learning_rate": 9.124199165487646e-07,
      "loss": 0.0159,
      "step": 7646
    },
    {
      "epoch": 1.7399317406143344,
      "grad_norm": 2.092254610281203,
      "learning_rate": 9.12340570948387e-07,
      "loss": 0.0881,
      "step": 7647
    },
    {
      "epoch": 1.7401592718998864,
      "grad_norm": 0.8586350139414584,
      "learning_rate": 9.122612194752947e-07,
      "loss": 0.0261,
      "step": 7648
    },
    {
      "epoch": 1.7403868031854381,
      "grad_norm": 1.215639356905783,
      "learning_rate": 9.12181862131109e-07,
      "loss": 0.0163,
      "step": 7649
    },
    {
      "epoch": 1.74061433447099,
      "grad_norm": 1.725700806539854,
      "learning_rate": 9.121024989174521e-07,
      "loss": 0.0263,
      "step": 7650
    },
    {
      "epoch": 1.7408418657565417,
      "grad_norm": 1.2188163467235449,
      "learning_rate": 9.120231298359458e-07,
      "loss": 0.022,
      "step": 7651
    },
    {
      "epoch": 1.7410693970420934,
      "grad_norm": 1.1976272309936158,
      "learning_rate": 9.119437548882125e-07,
      "loss": 0.0387,
      "step": 7652
    },
    {
      "epoch": 1.7412969283276452,
      "grad_norm": 1.2953796333595993,
      "learning_rate": 9.118643740758744e-07,
      "loss": 0.0134,
      "step": 7653
    },
    {
      "epoch": 1.741524459613197,
      "grad_norm": 1.2311997698528327,
      "learning_rate": 9.117849874005537e-07,
      "loss": 0.036,
      "step": 7654
    },
    {
      "epoch": 1.7417519908987487,
      "grad_norm": 1.1628900893695522,
      "learning_rate": 9.117055948638731e-07,
      "loss": 0.0429,
      "step": 7655
    },
    {
      "epoch": 1.7419795221843004,
      "grad_norm": 1.8508559292221363,
      "learning_rate": 9.116261964674553e-07,
      "loss": 0.0517,
      "step": 7656
    },
    {
      "epoch": 1.7422070534698522,
      "grad_norm": 0.8746941733793605,
      "learning_rate": 9.115467922129229e-07,
      "loss": 0.0241,
      "step": 7657
    },
    {
      "epoch": 1.742434584755404,
      "grad_norm": 0.9077736076367343,
      "learning_rate": 9.114673821018987e-07,
      "loss": 0.0244,
      "step": 7658
    },
    {
      "epoch": 1.7426621160409557,
      "grad_norm": 1.5287989175270456,
      "learning_rate": 9.113879661360063e-07,
      "loss": 0.0234,
      "step": 7659
    },
    {
      "epoch": 1.7428896473265074,
      "grad_norm": 1.6432947458522034,
      "learning_rate": 9.11308544316868e-07,
      "loss": 0.0352,
      "step": 7660
    },
    {
      "epoch": 1.7431171786120592,
      "grad_norm": 0.8065355560534044,
      "learning_rate": 9.112291166461076e-07,
      "loss": 0.0219,
      "step": 7661
    },
    {
      "epoch": 1.743344709897611,
      "grad_norm": 2.3664864556682845,
      "learning_rate": 9.111496831253481e-07,
      "loss": 0.075,
      "step": 7662
    },
    {
      "epoch": 1.7435722411831627,
      "grad_norm": 1.5414737823776685,
      "learning_rate": 9.110702437562132e-07,
      "loss": 0.0295,
      "step": 7663
    },
    {
      "epoch": 1.7437997724687144,
      "grad_norm": 1.5196225851842906,
      "learning_rate": 9.109907985403265e-07,
      "loss": 0.0254,
      "step": 7664
    },
    {
      "epoch": 1.7440273037542662,
      "grad_norm": 1.8653557622531247,
      "learning_rate": 9.109113474793116e-07,
      "loss": 0.0577,
      "step": 7665
    },
    {
      "epoch": 1.744254835039818,
      "grad_norm": 1.5718530339793813,
      "learning_rate": 9.108318905747924e-07,
      "loss": 0.0433,
      "step": 7666
    },
    {
      "epoch": 1.7444823663253697,
      "grad_norm": 0.8753040187644087,
      "learning_rate": 9.107524278283928e-07,
      "loss": 0.0133,
      "step": 7667
    },
    {
      "epoch": 1.7447098976109214,
      "grad_norm": 0.9921757911909639,
      "learning_rate": 9.106729592417368e-07,
      "loss": 0.0148,
      "step": 7668
    },
    {
      "epoch": 1.7449374288964732,
      "grad_norm": 1.023329566714207,
      "learning_rate": 9.105934848164488e-07,
      "loss": 0.0282,
      "step": 7669
    },
    {
      "epoch": 1.745164960182025,
      "grad_norm": 1.2481670331085064,
      "learning_rate": 9.105140045541532e-07,
      "loss": 0.0427,
      "step": 7670
    },
    {
      "epoch": 1.7453924914675767,
      "grad_norm": 1.3173050639321486,
      "learning_rate": 9.10434518456474e-07,
      "loss": 0.0519,
      "step": 7671
    },
    {
      "epoch": 1.7456200227531284,
      "grad_norm": 0.9060431301408944,
      "learning_rate": 9.10355026525036e-07,
      "loss": 0.0247,
      "step": 7672
    },
    {
      "epoch": 1.7458475540386802,
      "grad_norm": 0.9275970798228872,
      "learning_rate": 9.102755287614639e-07,
      "loss": 0.0178,
      "step": 7673
    },
    {
      "epoch": 1.746075085324232,
      "grad_norm": 1.2478408678541246,
      "learning_rate": 9.101960251673825e-07,
      "loss": 0.0253,
      "step": 7674
    },
    {
      "epoch": 1.7463026166097837,
      "grad_norm": 1.0197676494224464,
      "learning_rate": 9.101165157444166e-07,
      "loss": 0.0225,
      "step": 7675
    },
    {
      "epoch": 1.7465301478953354,
      "grad_norm": 0.909094720632412,
      "learning_rate": 9.100370004941912e-07,
      "loss": 0.0188,
      "step": 7676
    },
    {
      "epoch": 1.7467576791808874,
      "grad_norm": 1.2769228356341842,
      "learning_rate": 9.099574794183317e-07,
      "loss": 0.0356,
      "step": 7677
    },
    {
      "epoch": 1.7469852104664392,
      "grad_norm": 1.0980874895229569,
      "learning_rate": 9.098779525184631e-07,
      "loss": 0.029,
      "step": 7678
    },
    {
      "epoch": 1.747212741751991,
      "grad_norm": 1.1165038442429356,
      "learning_rate": 9.097984197962109e-07,
      "loss": 0.0289,
      "step": 7679
    },
    {
      "epoch": 1.7474402730375427,
      "grad_norm": 0.96382139387914,
      "learning_rate": 9.097188812532006e-07,
      "loss": 0.0171,
      "step": 7680
    },
    {
      "epoch": 1.7476678043230944,
      "grad_norm": 1.3350100903286808,
      "learning_rate": 9.096393368910578e-07,
      "loss": 0.0345,
      "step": 7681
    },
    {
      "epoch": 1.7478953356086462,
      "grad_norm": 1.0713437039923497,
      "learning_rate": 9.095597867114082e-07,
      "loss": 0.0131,
      "step": 7682
    },
    {
      "epoch": 1.748122866894198,
      "grad_norm": 1.0460207092655962,
      "learning_rate": 9.094802307158777e-07,
      "loss": 0.0282,
      "step": 7683
    },
    {
      "epoch": 1.7483503981797497,
      "grad_norm": 1.2263445161655493,
      "learning_rate": 9.094006689060924e-07,
      "loss": 0.0336,
      "step": 7684
    },
    {
      "epoch": 1.7485779294653014,
      "grad_norm": 1.1487925099211806,
      "learning_rate": 9.093211012836782e-07,
      "loss": 0.0194,
      "step": 7685
    },
    {
      "epoch": 1.7488054607508532,
      "grad_norm": 0.9865132985502546,
      "learning_rate": 9.092415278502614e-07,
      "loss": 0.0426,
      "step": 7686
    },
    {
      "epoch": 1.7490329920364052,
      "grad_norm": 0.9010433966430745,
      "learning_rate": 9.091619486074684e-07,
      "loss": 0.0153,
      "step": 7687
    },
    {
      "epoch": 1.749260523321957,
      "grad_norm": 0.3964431214764849,
      "learning_rate": 9.090823635569254e-07,
      "loss": 0.0089,
      "step": 7688
    },
    {
      "epoch": 1.7494880546075087,
      "grad_norm": 2.1364993775182044,
      "learning_rate": 9.090027727002594e-07,
      "loss": 0.1037,
      "step": 7689
    },
    {
      "epoch": 1.7497155858930604,
      "grad_norm": 1.0142838136704868,
      "learning_rate": 9.089231760390968e-07,
      "loss": 0.0383,
      "step": 7690
    },
    {
      "epoch": 1.7499431171786122,
      "grad_norm": 0.9385889195565317,
      "learning_rate": 9.088435735750643e-07,
      "loss": 0.0216,
      "step": 7691
    },
    {
      "epoch": 1.750170648464164,
      "grad_norm": 0.8684423870809375,
      "learning_rate": 9.087639653097892e-07,
      "loss": 0.0317,
      "step": 7692
    },
    {
      "epoch": 1.7503981797497157,
      "grad_norm": 1.296162026638584,
      "learning_rate": 9.086843512448983e-07,
      "loss": 0.0442,
      "step": 7693
    },
    {
      "epoch": 1.7506257110352674,
      "grad_norm": 0.6407459373187171,
      "learning_rate": 9.086047313820186e-07,
      "loss": 0.0131,
      "step": 7694
    },
    {
      "epoch": 1.7508532423208192,
      "grad_norm": 1.5991269608707057,
      "learning_rate": 9.085251057227777e-07,
      "loss": 0.0494,
      "step": 7695
    },
    {
      "epoch": 1.751080773606371,
      "grad_norm": 1.246921304342,
      "learning_rate": 9.08445474268803e-07,
      "loss": 0.0254,
      "step": 7696
    },
    {
      "epoch": 1.7513083048919227,
      "grad_norm": 1.0011932326498272,
      "learning_rate": 9.083658370217219e-07,
      "loss": 0.0273,
      "step": 7697
    },
    {
      "epoch": 1.7515358361774744,
      "grad_norm": 1.7295845937220786,
      "learning_rate": 9.082861939831619e-07,
      "loss": 0.0849,
      "step": 7698
    },
    {
      "epoch": 1.7517633674630262,
      "grad_norm": 0.7603956433839477,
      "learning_rate": 9.08206545154751e-07,
      "loss": 0.017,
      "step": 7699
    },
    {
      "epoch": 1.751990898748578,
      "grad_norm": 1.1788808766142622,
      "learning_rate": 9.08126890538117e-07,
      "loss": 0.0504,
      "step": 7700
    },
    {
      "epoch": 1.7522184300341297,
      "grad_norm": 0.747307893601606,
      "learning_rate": 9.080472301348878e-07,
      "loss": 0.0192,
      "step": 7701
    },
    {
      "epoch": 1.7524459613196814,
      "grad_norm": 1.1518157305353764,
      "learning_rate": 9.079675639466918e-07,
      "loss": 0.045,
      "step": 7702
    },
    {
      "epoch": 1.7526734926052332,
      "grad_norm": 1.0953592367909082,
      "learning_rate": 9.078878919751569e-07,
      "loss": 0.0428,
      "step": 7703
    },
    {
      "epoch": 1.752901023890785,
      "grad_norm": 1.363606620510174,
      "learning_rate": 9.078082142219114e-07,
      "loss": 0.0349,
      "step": 7704
    },
    {
      "epoch": 1.7531285551763367,
      "grad_norm": 1.0499526817597125,
      "learning_rate": 9.077285306885842e-07,
      "loss": 0.0202,
      "step": 7705
    },
    {
      "epoch": 1.7533560864618885,
      "grad_norm": 0.7675164362932144,
      "learning_rate": 9.076488413768035e-07,
      "loss": 0.0218,
      "step": 7706
    },
    {
      "epoch": 1.7535836177474402,
      "grad_norm": 1.2618843485916398,
      "learning_rate": 9.075691462881981e-07,
      "loss": 0.0502,
      "step": 7707
    },
    {
      "epoch": 1.753811149032992,
      "grad_norm": 1.5141641473520007,
      "learning_rate": 9.074894454243968e-07,
      "loss": 0.0283,
      "step": 7708
    },
    {
      "epoch": 1.7540386803185437,
      "grad_norm": 1.1092758523541948,
      "learning_rate": 9.074097387870289e-07,
      "loss": 0.0208,
      "step": 7709
    },
    {
      "epoch": 1.7542662116040955,
      "grad_norm": 1.6035144538724415,
      "learning_rate": 9.07330026377723e-07,
      "loss": 0.0198,
      "step": 7710
    },
    {
      "epoch": 1.7544937428896472,
      "grad_norm": 1.167017853281529,
      "learning_rate": 9.072503081981081e-07,
      "loss": 0.0293,
      "step": 7711
    },
    {
      "epoch": 1.754721274175199,
      "grad_norm": 0.9184705076375725,
      "learning_rate": 9.07170584249814e-07,
      "loss": 0.0255,
      "step": 7712
    },
    {
      "epoch": 1.7549488054607507,
      "grad_norm": 1.2906245689143385,
      "learning_rate": 9.070908545344702e-07,
      "loss": 0.0323,
      "step": 7713
    },
    {
      "epoch": 1.7551763367463025,
      "grad_norm": 1.1306246357358782,
      "learning_rate": 9.070111190537057e-07,
      "loss": 0.027,
      "step": 7714
    },
    {
      "epoch": 1.7554038680318542,
      "grad_norm": 1.2717939026208809,
      "learning_rate": 9.069313778091504e-07,
      "loss": 0.0701,
      "step": 7715
    },
    {
      "epoch": 1.7556313993174062,
      "grad_norm": 0.9364844022319152,
      "learning_rate": 9.068516308024343e-07,
      "loss": 0.0179,
      "step": 7716
    },
    {
      "epoch": 1.755858930602958,
      "grad_norm": 2.120884695289572,
      "learning_rate": 9.067718780351867e-07,
      "loss": 0.0357,
      "step": 7717
    },
    {
      "epoch": 1.7560864618885097,
      "grad_norm": 1.7372318690678035,
      "learning_rate": 9.066921195090383e-07,
      "loss": 0.0533,
      "step": 7718
    },
    {
      "epoch": 1.7563139931740614,
      "grad_norm": 0.9006127396212078,
      "learning_rate": 9.066123552256187e-07,
      "loss": 0.0262,
      "step": 7719
    },
    {
      "epoch": 1.7565415244596132,
      "grad_norm": 1.1618139823950684,
      "learning_rate": 9.065325851865583e-07,
      "loss": 0.0255,
      "step": 7720
    },
    {
      "epoch": 1.756769055745165,
      "grad_norm": 1.4553349352673766,
      "learning_rate": 9.064528093934874e-07,
      "loss": 0.0231,
      "step": 7721
    },
    {
      "epoch": 1.7569965870307167,
      "grad_norm": 0.7189318772127737,
      "learning_rate": 9.063730278480368e-07,
      "loss": 0.0168,
      "step": 7722
    },
    {
      "epoch": 1.7572241183162685,
      "grad_norm": 0.7521106143967847,
      "learning_rate": 9.062932405518365e-07,
      "loss": 0.0131,
      "step": 7723
    },
    {
      "epoch": 1.7574516496018202,
      "grad_norm": 1.3566892763492646,
      "learning_rate": 9.062134475065176e-07,
      "loss": 0.0452,
      "step": 7724
    },
    {
      "epoch": 1.757679180887372,
      "grad_norm": 2.059052973839986,
      "learning_rate": 9.06133648713711e-07,
      "loss": 0.0423,
      "step": 7725
    },
    {
      "epoch": 1.757906712172924,
      "grad_norm": 1.673179110604813,
      "learning_rate": 9.060538441750475e-07,
      "loss": 0.0488,
      "step": 7726
    },
    {
      "epoch": 1.7581342434584757,
      "grad_norm": 1.190424328968161,
      "learning_rate": 9.05974033892158e-07,
      "loss": 0.021,
      "step": 7727
    },
    {
      "epoch": 1.7583617747440274,
      "grad_norm": 1.0485429519532234,
      "learning_rate": 9.058942178666738e-07,
      "loss": 0.028,
      "step": 7728
    },
    {
      "epoch": 1.7585893060295792,
      "grad_norm": 1.0357293376161343,
      "learning_rate": 9.058143961002263e-07,
      "loss": 0.0319,
      "step": 7729
    },
    {
      "epoch": 1.758816837315131,
      "grad_norm": 1.5732344809554888,
      "learning_rate": 9.05734568594447e-07,
      "loss": 0.028,
      "step": 7730
    },
    {
      "epoch": 1.7590443686006827,
      "grad_norm": 0.9379471313058495,
      "learning_rate": 9.05654735350967e-07,
      "loss": 0.012,
      "step": 7731
    },
    {
      "epoch": 1.7592718998862344,
      "grad_norm": 1.1693048482911874,
      "learning_rate": 9.055748963714183e-07,
      "loss": 0.0354,
      "step": 7732
    },
    {
      "epoch": 1.7594994311717862,
      "grad_norm": 1.1467606906199936,
      "learning_rate": 9.054950516574327e-07,
      "loss": 0.0274,
      "step": 7733
    },
    {
      "epoch": 1.759726962457338,
      "grad_norm": 1.0319823876324419,
      "learning_rate": 9.054152012106417e-07,
      "loss": 0.0325,
      "step": 7734
    },
    {
      "epoch": 1.7599544937428897,
      "grad_norm": 0.9017057064223635,
      "learning_rate": 9.053353450326777e-07,
      "loss": 0.0163,
      "step": 7735
    },
    {
      "epoch": 1.7601820250284415,
      "grad_norm": 1.3437259664925094,
      "learning_rate": 9.052554831251725e-07,
      "loss": 0.0492,
      "step": 7736
    },
    {
      "epoch": 1.7604095563139932,
      "grad_norm": 1.235304716459916,
      "learning_rate": 9.051756154897587e-07,
      "loss": 0.0264,
      "step": 7737
    },
    {
      "epoch": 1.760637087599545,
      "grad_norm": 1.3021796415012392,
      "learning_rate": 9.050957421280683e-07,
      "loss": 0.0355,
      "step": 7738
    },
    {
      "epoch": 1.7608646188850967,
      "grad_norm": 0.6730637046090184,
      "learning_rate": 9.05015863041734e-07,
      "loss": 0.01,
      "step": 7739
    },
    {
      "epoch": 1.7610921501706485,
      "grad_norm": 1.2763599910909942,
      "learning_rate": 9.049359782323881e-07,
      "loss": 0.0226,
      "step": 7740
    },
    {
      "epoch": 1.7613196814562002,
      "grad_norm": 1.3964209453463265,
      "learning_rate": 9.048560877016637e-07,
      "loss": 0.0284,
      "step": 7741
    },
    {
      "epoch": 1.761547212741752,
      "grad_norm": 1.620164064470503,
      "learning_rate": 9.047761914511933e-07,
      "loss": 0.0291,
      "step": 7742
    },
    {
      "epoch": 1.7617747440273037,
      "grad_norm": 0.3708609160081374,
      "learning_rate": 9.0469628948261e-07,
      "loss": 0.0055,
      "step": 7743
    },
    {
      "epoch": 1.7620022753128555,
      "grad_norm": 0.8799311045280214,
      "learning_rate": 9.046163817975466e-07,
      "loss": 0.0149,
      "step": 7744
    },
    {
      "epoch": 1.7622298065984072,
      "grad_norm": 1.062318942187663,
      "learning_rate": 9.045364683976366e-07,
      "loss": 0.0377,
      "step": 7745
    },
    {
      "epoch": 1.762457337883959,
      "grad_norm": 1.085127144596073,
      "learning_rate": 9.044565492845131e-07,
      "loss": 0.0471,
      "step": 7746
    },
    {
      "epoch": 1.7626848691695107,
      "grad_norm": 1.7016862939695618,
      "learning_rate": 9.043766244598096e-07,
      "loss": 0.0365,
      "step": 7747
    },
    {
      "epoch": 1.7629124004550625,
      "grad_norm": 0.7069815925067215,
      "learning_rate": 9.042966939251595e-07,
      "loss": 0.0211,
      "step": 7748
    },
    {
      "epoch": 1.7631399317406142,
      "grad_norm": 2.3165835449442826,
      "learning_rate": 9.042167576821964e-07,
      "loss": 0.0642,
      "step": 7749
    },
    {
      "epoch": 1.763367463026166,
      "grad_norm": 1.1909554351095002,
      "learning_rate": 9.041368157325543e-07,
      "loss": 0.0301,
      "step": 7750
    },
    {
      "epoch": 1.7635949943117177,
      "grad_norm": 1.460760689556158,
      "learning_rate": 9.040568680778668e-07,
      "loss": 0.0481,
      "step": 7751
    },
    {
      "epoch": 1.7638225255972695,
      "grad_norm": 1.4806066713210055,
      "learning_rate": 9.03976914719768e-07,
      "loss": 0.0397,
      "step": 7752
    },
    {
      "epoch": 1.7640500568828212,
      "grad_norm": 0.990535408952515,
      "learning_rate": 9.03896955659892e-07,
      "loss": 0.0266,
      "step": 7753
    },
    {
      "epoch": 1.764277588168373,
      "grad_norm": 1.0187827421869802,
      "learning_rate": 9.03816990899873e-07,
      "loss": 0.025,
      "step": 7754
    },
    {
      "epoch": 1.764505119453925,
      "grad_norm": 1.468842093699268,
      "learning_rate": 9.037370204413452e-07,
      "loss": 0.0308,
      "step": 7755
    },
    {
      "epoch": 1.7647326507394767,
      "grad_norm": 1.104325889984664,
      "learning_rate": 9.036570442859433e-07,
      "loss": 0.0274,
      "step": 7756
    },
    {
      "epoch": 1.7649601820250285,
      "grad_norm": 0.9609041262599042,
      "learning_rate": 9.035770624353018e-07,
      "loss": 0.0233,
      "step": 7757
    },
    {
      "epoch": 1.7651877133105802,
      "grad_norm": 1.300462656770205,
      "learning_rate": 9.034970748910552e-07,
      "loss": 0.0564,
      "step": 7758
    },
    {
      "epoch": 1.765415244596132,
      "grad_norm": 0.7966242008387683,
      "learning_rate": 9.034170816548387e-07,
      "loss": 0.0158,
      "step": 7759
    },
    {
      "epoch": 1.7656427758816837,
      "grad_norm": 1.0068645889333558,
      "learning_rate": 9.033370827282868e-07,
      "loss": 0.015,
      "step": 7760
    },
    {
      "epoch": 1.7658703071672355,
      "grad_norm": 1.84372314061819,
      "learning_rate": 9.032570781130346e-07,
      "loss": 0.0359,
      "step": 7761
    },
    {
      "epoch": 1.7660978384527872,
      "grad_norm": 0.9517555469231603,
      "learning_rate": 9.031770678107174e-07,
      "loss": 0.0221,
      "step": 7762
    },
    {
      "epoch": 1.766325369738339,
      "grad_norm": 1.008357002228374,
      "learning_rate": 9.030970518229704e-07,
      "loss": 0.0224,
      "step": 7763
    },
    {
      "epoch": 1.7665529010238907,
      "grad_norm": 0.4490169893485138,
      "learning_rate": 9.030170301514289e-07,
      "loss": 0.0051,
      "step": 7764
    },
    {
      "epoch": 1.7667804323094427,
      "grad_norm": 1.433194534836921,
      "learning_rate": 9.029370027977284e-07,
      "loss": 0.0244,
      "step": 7765
    },
    {
      "epoch": 1.7670079635949945,
      "grad_norm": 0.9253299076158662,
      "learning_rate": 9.028569697635047e-07,
      "loss": 0.016,
      "step": 7766
    },
    {
      "epoch": 1.7672354948805462,
      "grad_norm": 1.0293544978932465,
      "learning_rate": 9.027769310503935e-07,
      "loss": 0.0393,
      "step": 7767
    },
    {
      "epoch": 1.767463026166098,
      "grad_norm": 1.2746844789912744,
      "learning_rate": 9.026968866600304e-07,
      "loss": 0.0254,
      "step": 7768
    },
    {
      "epoch": 1.7676905574516497,
      "grad_norm": 0.9738684622085294,
      "learning_rate": 9.026168365940516e-07,
      "loss": 0.0353,
      "step": 7769
    },
    {
      "epoch": 1.7679180887372015,
      "grad_norm": 1.4215988518341673,
      "learning_rate": 9.02536780854093e-07,
      "loss": 0.0289,
      "step": 7770
    },
    {
      "epoch": 1.7681456200227532,
      "grad_norm": 1.9463891862128035,
      "learning_rate": 9.024567194417911e-07,
      "loss": 0.0234,
      "step": 7771
    },
    {
      "epoch": 1.768373151308305,
      "grad_norm": 0.9261059100245073,
      "learning_rate": 9.023766523587817e-07,
      "loss": 0.0121,
      "step": 7772
    },
    {
      "epoch": 1.7686006825938567,
      "grad_norm": 2.673950698193844,
      "learning_rate": 9.022965796067016e-07,
      "loss": 0.0323,
      "step": 7773
    },
    {
      "epoch": 1.7688282138794085,
      "grad_norm": 0.8834482418976045,
      "learning_rate": 9.022165011871873e-07,
      "loss": 0.0302,
      "step": 7774
    },
    {
      "epoch": 1.7690557451649602,
      "grad_norm": 1.1860901670932948,
      "learning_rate": 9.021364171018754e-07,
      "loss": 0.0311,
      "step": 7775
    },
    {
      "epoch": 1.769283276450512,
      "grad_norm": 0.9597974646507369,
      "learning_rate": 9.020563273524027e-07,
      "loss": 0.0229,
      "step": 7776
    },
    {
      "epoch": 1.7695108077360637,
      "grad_norm": 0.847588734062367,
      "learning_rate": 9.019762319404061e-07,
      "loss": 0.0195,
      "step": 7777
    },
    {
      "epoch": 1.7697383390216155,
      "grad_norm": 1.1738522512510032,
      "learning_rate": 9.018961308675225e-07,
      "loss": 0.0342,
      "step": 7778
    },
    {
      "epoch": 1.7699658703071672,
      "grad_norm": 0.7503668751477243,
      "learning_rate": 9.018160241353893e-07,
      "loss": 0.0145,
      "step": 7779
    },
    {
      "epoch": 1.770193401592719,
      "grad_norm": 1.084695064985875,
      "learning_rate": 9.017359117456434e-07,
      "loss": 0.0248,
      "step": 7780
    },
    {
      "epoch": 1.7704209328782707,
      "grad_norm": 0.8899828699614253,
      "learning_rate": 9.016557936999221e-07,
      "loss": 0.0212,
      "step": 7781
    },
    {
      "epoch": 1.7706484641638225,
      "grad_norm": 0.7597079457803091,
      "learning_rate": 9.015756699998632e-07,
      "loss": 0.032,
      "step": 7782
    },
    {
      "epoch": 1.7708759954493742,
      "grad_norm": 0.8711745869354168,
      "learning_rate": 9.014955406471041e-07,
      "loss": 0.0102,
      "step": 7783
    },
    {
      "epoch": 1.771103526734926,
      "grad_norm": 0.8186824087409144,
      "learning_rate": 9.014154056432828e-07,
      "loss": 0.0241,
      "step": 7784
    },
    {
      "epoch": 1.7713310580204777,
      "grad_norm": 0.7283696556407617,
      "learning_rate": 9.013352649900365e-07,
      "loss": 0.0164,
      "step": 7785
    },
    {
      "epoch": 1.7715585893060295,
      "grad_norm": 0.9632421077654101,
      "learning_rate": 9.012551186890037e-07,
      "loss": 0.0201,
      "step": 7786
    },
    {
      "epoch": 1.7717861205915812,
      "grad_norm": 1.2968869584635645,
      "learning_rate": 9.011749667418221e-07,
      "loss": 0.033,
      "step": 7787
    },
    {
      "epoch": 1.772013651877133,
      "grad_norm": 1.1781087189325576,
      "learning_rate": 9.010948091501298e-07,
      "loss": 0.0323,
      "step": 7788
    },
    {
      "epoch": 1.7722411831626848,
      "grad_norm": 0.6227189258752035,
      "learning_rate": 9.010146459155654e-07,
      "loss": 0.0177,
      "step": 7789
    },
    {
      "epoch": 1.7724687144482365,
      "grad_norm": 1.1525215566714462,
      "learning_rate": 9.009344770397671e-07,
      "loss": 0.0209,
      "step": 7790
    },
    {
      "epoch": 1.7726962457337883,
      "grad_norm": 0.9174352181093524,
      "learning_rate": 9.008543025243735e-07,
      "loss": 0.0191,
      "step": 7791
    },
    {
      "epoch": 1.77292377701934,
      "grad_norm": 0.7635480894250786,
      "learning_rate": 9.007741223710232e-07,
      "loss": 0.0146,
      "step": 7792
    },
    {
      "epoch": 1.7731513083048918,
      "grad_norm": 1.521151554028818,
      "learning_rate": 9.006939365813549e-07,
      "loss": 0.0471,
      "step": 7793
    },
    {
      "epoch": 1.7733788395904437,
      "grad_norm": 2.081083148922227,
      "learning_rate": 9.006137451570074e-07,
      "loss": 0.029,
      "step": 7794
    },
    {
      "epoch": 1.7736063708759955,
      "grad_norm": 1.1298450626530687,
      "learning_rate": 9.005335480996196e-07,
      "loss": 0.022,
      "step": 7795
    },
    {
      "epoch": 1.7738339021615472,
      "grad_norm": 1.2497151471052976,
      "learning_rate": 9.004533454108308e-07,
      "loss": 0.0333,
      "step": 7796
    },
    {
      "epoch": 1.774061433447099,
      "grad_norm": 1.0582139306273477,
      "learning_rate": 9.0037313709228e-07,
      "loss": 0.0199,
      "step": 7797
    },
    {
      "epoch": 1.7742889647326507,
      "grad_norm": 1.5328551731921303,
      "learning_rate": 9.002929231456067e-07,
      "loss": 0.0529,
      "step": 7798
    },
    {
      "epoch": 1.7745164960182025,
      "grad_norm": 1.2155287205090175,
      "learning_rate": 9.002127035724502e-07,
      "loss": 0.037,
      "step": 7799
    },
    {
      "epoch": 1.7747440273037542,
      "grad_norm": 1.0218347865522226,
      "learning_rate": 9.001324783744501e-07,
      "loss": 0.0417,
      "step": 7800
    },
    {
      "epoch": 1.774971558589306,
      "grad_norm": 3.390477976630992,
      "learning_rate": 9.000522475532461e-07,
      "loss": 0.0554,
      "step": 7801
    },
    {
      "epoch": 1.7751990898748577,
      "grad_norm": 0.592801892287802,
      "learning_rate": 8.999720111104776e-07,
      "loss": 0.0088,
      "step": 7802
    },
    {
      "epoch": 1.7754266211604095,
      "grad_norm": 1.4856820411102438,
      "learning_rate": 8.99891769047785e-07,
      "loss": 0.0302,
      "step": 7803
    },
    {
      "epoch": 1.7756541524459615,
      "grad_norm": 1.3825413426440172,
      "learning_rate": 8.998115213668082e-07,
      "loss": 0.0393,
      "step": 7804
    },
    {
      "epoch": 1.7758816837315132,
      "grad_norm": 1.0715853298379499,
      "learning_rate": 8.997312680691869e-07,
      "loss": 0.0251,
      "step": 7805
    },
    {
      "epoch": 1.776109215017065,
      "grad_norm": 1.086273322174223,
      "learning_rate": 8.996510091565618e-07,
      "loss": 0.0222,
      "step": 7806
    },
    {
      "epoch": 1.7763367463026167,
      "grad_norm": 1.574504492749921,
      "learning_rate": 8.99570744630573e-07,
      "loss": 0.0218,
      "step": 7807
    },
    {
      "epoch": 1.7765642775881685,
      "grad_norm": 0.7604610383602236,
      "learning_rate": 8.99490474492861e-07,
      "loss": 0.0165,
      "step": 7808
    },
    {
      "epoch": 1.7767918088737202,
      "grad_norm": 0.9545288476626476,
      "learning_rate": 8.994101987450665e-07,
      "loss": 0.0223,
      "step": 7809
    },
    {
      "epoch": 1.777019340159272,
      "grad_norm": 0.6882788064421679,
      "learning_rate": 8.993299173888302e-07,
      "loss": 0.0225,
      "step": 7810
    },
    {
      "epoch": 1.7772468714448237,
      "grad_norm": 1.7333778039577077,
      "learning_rate": 8.992496304257926e-07,
      "loss": 0.0323,
      "step": 7811
    },
    {
      "epoch": 1.7774744027303755,
      "grad_norm": 1.1051484045866493,
      "learning_rate": 8.99169337857595e-07,
      "loss": 0.0293,
      "step": 7812
    },
    {
      "epoch": 1.7777019340159272,
      "grad_norm": 0.9884789847111075,
      "learning_rate": 8.990890396858781e-07,
      "loss": 0.0259,
      "step": 7813
    },
    {
      "epoch": 1.777929465301479,
      "grad_norm": 1.2914085957558916,
      "learning_rate": 8.990087359122832e-07,
      "loss": 0.0311,
      "step": 7814
    },
    {
      "epoch": 1.7781569965870307,
      "grad_norm": 1.049883175471557,
      "learning_rate": 8.989284265384515e-07,
      "loss": 0.023,
      "step": 7815
    },
    {
      "epoch": 1.7783845278725825,
      "grad_norm": 0.6694835390188534,
      "learning_rate": 8.988481115660247e-07,
      "loss": 0.0099,
      "step": 7816
    },
    {
      "epoch": 1.7786120591581343,
      "grad_norm": 1.0588133640503645,
      "learning_rate": 8.987677909966439e-07,
      "loss": 0.0206,
      "step": 7817
    },
    {
      "epoch": 1.778839590443686,
      "grad_norm": 0.7819167306883373,
      "learning_rate": 8.986874648319507e-07,
      "loss": 0.0148,
      "step": 7818
    },
    {
      "epoch": 1.7790671217292378,
      "grad_norm": 1.4821086058928985,
      "learning_rate": 8.986071330735872e-07,
      "loss": 0.0237,
      "step": 7819
    },
    {
      "epoch": 1.7792946530147895,
      "grad_norm": 1.0958911672140499,
      "learning_rate": 8.985267957231947e-07,
      "loss": 0.0308,
      "step": 7820
    },
    {
      "epoch": 1.7795221843003413,
      "grad_norm": 0.8791299425783055,
      "learning_rate": 8.984464527824157e-07,
      "loss": 0.0218,
      "step": 7821
    },
    {
      "epoch": 1.779749715585893,
      "grad_norm": 1.1468618488025002,
      "learning_rate": 8.983661042528917e-07,
      "loss": 0.015,
      "step": 7822
    },
    {
      "epoch": 1.7799772468714448,
      "grad_norm": 0.9483905040621247,
      "learning_rate": 8.982857501362655e-07,
      "loss": 0.0161,
      "step": 7823
    },
    {
      "epoch": 1.7802047781569965,
      "grad_norm": 0.8960232478818805,
      "learning_rate": 8.982053904341789e-07,
      "loss": 0.0198,
      "step": 7824
    },
    {
      "epoch": 1.7804323094425483,
      "grad_norm": 1.4328448860754595,
      "learning_rate": 8.981250251482746e-07,
      "loss": 0.017,
      "step": 7825
    },
    {
      "epoch": 1.7806598407281,
      "grad_norm": 0.9743777428010655,
      "learning_rate": 8.980446542801947e-07,
      "loss": 0.0113,
      "step": 7826
    },
    {
      "epoch": 1.7808873720136518,
      "grad_norm": 1.2703087358314946,
      "learning_rate": 8.979642778315824e-07,
      "loss": 0.032,
      "step": 7827
    },
    {
      "epoch": 1.7811149032992035,
      "grad_norm": 1.4011343554143572,
      "learning_rate": 8.9788389580408e-07,
      "loss": 0.0324,
      "step": 7828
    },
    {
      "epoch": 1.7813424345847553,
      "grad_norm": 1.6343876573253318,
      "learning_rate": 8.978035081993307e-07,
      "loss": 0.0268,
      "step": 7829
    },
    {
      "epoch": 1.781569965870307,
      "grad_norm": 1.7645817562932788,
      "learning_rate": 8.977231150189772e-07,
      "loss": 0.0317,
      "step": 7830
    },
    {
      "epoch": 1.7817974971558588,
      "grad_norm": 1.6087529896149062,
      "learning_rate": 8.976427162646628e-07,
      "loss": 0.0449,
      "step": 7831
    },
    {
      "epoch": 1.7820250284414105,
      "grad_norm": 0.8531146257805737,
      "learning_rate": 8.975623119380304e-07,
      "loss": 0.0196,
      "step": 7832
    },
    {
      "epoch": 1.7822525597269625,
      "grad_norm": 1.206582803943983,
      "learning_rate": 8.974819020407237e-07,
      "loss": 0.02,
      "step": 7833
    },
    {
      "epoch": 1.7824800910125143,
      "grad_norm": 1.0145655803861944,
      "learning_rate": 8.974014865743859e-07,
      "loss": 0.022,
      "step": 7834
    },
    {
      "epoch": 1.782707622298066,
      "grad_norm": 1.4964526683444779,
      "learning_rate": 8.973210655406605e-07,
      "loss": 0.0404,
      "step": 7835
    },
    {
      "epoch": 1.7829351535836178,
      "grad_norm": 0.7848232581214782,
      "learning_rate": 8.972406389411915e-07,
      "loss": 0.0183,
      "step": 7836
    },
    {
      "epoch": 1.7831626848691695,
      "grad_norm": 0.9976090202029402,
      "learning_rate": 8.971602067776222e-07,
      "loss": 0.0197,
      "step": 7837
    },
    {
      "epoch": 1.7833902161547213,
      "grad_norm": 1.4889625882400916,
      "learning_rate": 8.970797690515967e-07,
      "loss": 0.0304,
      "step": 7838
    },
    {
      "epoch": 1.783617747440273,
      "grad_norm": 0.8452260647654178,
      "learning_rate": 8.969993257647591e-07,
      "loss": 0.0146,
      "step": 7839
    },
    {
      "epoch": 1.7838452787258248,
      "grad_norm": 1.4782817609899408,
      "learning_rate": 8.969188769187534e-07,
      "loss": 0.026,
      "step": 7840
    },
    {
      "epoch": 1.7840728100113765,
      "grad_norm": 1.225959473840193,
      "learning_rate": 8.96838422515224e-07,
      "loss": 0.0429,
      "step": 7841
    },
    {
      "epoch": 1.7843003412969285,
      "grad_norm": 0.8206177553996756,
      "learning_rate": 8.967579625558148e-07,
      "loss": 0.0111,
      "step": 7842
    },
    {
      "epoch": 1.7845278725824802,
      "grad_norm": 1.2525656353069259,
      "learning_rate": 8.966774970421708e-07,
      "loss": 0.0289,
      "step": 7843
    },
    {
      "epoch": 1.784755403868032,
      "grad_norm": 0.9338096531101292,
      "learning_rate": 8.965970259759363e-07,
      "loss": 0.0228,
      "step": 7844
    },
    {
      "epoch": 1.7849829351535837,
      "grad_norm": 0.7275812301839438,
      "learning_rate": 8.965165493587557e-07,
      "loss": 0.0117,
      "step": 7845
    },
    {
      "epoch": 1.7852104664391355,
      "grad_norm": 0.7969106720107036,
      "learning_rate": 8.964360671922743e-07,
      "loss": 0.0147,
      "step": 7846
    },
    {
      "epoch": 1.7854379977246873,
      "grad_norm": 1.458795393176787,
      "learning_rate": 8.963555794781369e-07,
      "loss": 0.0273,
      "step": 7847
    },
    {
      "epoch": 1.785665529010239,
      "grad_norm": 1.7699436515502187,
      "learning_rate": 8.962750862179883e-07,
      "loss": 0.06,
      "step": 7848
    },
    {
      "epoch": 1.7858930602957908,
      "grad_norm": 0.5431062763845044,
      "learning_rate": 8.961945874134738e-07,
      "loss": 0.0081,
      "step": 7849
    },
    {
      "epoch": 1.7861205915813425,
      "grad_norm": 1.2451912708323942,
      "learning_rate": 8.961140830662386e-07,
      "loss": 0.0481,
      "step": 7850
    },
    {
      "epoch": 1.7863481228668943,
      "grad_norm": 0.9236082482255349,
      "learning_rate": 8.960335731779281e-07,
      "loss": 0.0209,
      "step": 7851
    },
    {
      "epoch": 1.786575654152446,
      "grad_norm": 0.6970816456118641,
      "learning_rate": 8.959530577501875e-07,
      "loss": 0.0171,
      "step": 7852
    },
    {
      "epoch": 1.7868031854379978,
      "grad_norm": 0.9506476518379603,
      "learning_rate": 8.958725367846628e-07,
      "loss": 0.0221,
      "step": 7853
    },
    {
      "epoch": 1.7870307167235495,
      "grad_norm": 1.000808992626888,
      "learning_rate": 8.957920102829997e-07,
      "loss": 0.0155,
      "step": 7854
    },
    {
      "epoch": 1.7872582480091013,
      "grad_norm": 1.4581764598911346,
      "learning_rate": 8.957114782468436e-07,
      "loss": 0.0308,
      "step": 7855
    },
    {
      "epoch": 1.787485779294653,
      "grad_norm": 1.2562396437883345,
      "learning_rate": 8.956309406778407e-07,
      "loss": 0.0206,
      "step": 7856
    },
    {
      "epoch": 1.7877133105802048,
      "grad_norm": 1.4772297021594707,
      "learning_rate": 8.955503975776371e-07,
      "loss": 0.0396,
      "step": 7857
    },
    {
      "epoch": 1.7879408418657565,
      "grad_norm": 1.005126633250784,
      "learning_rate": 8.954698489478788e-07,
      "loss": 0.0151,
      "step": 7858
    },
    {
      "epoch": 1.7881683731513083,
      "grad_norm": 1.8209464517520468,
      "learning_rate": 8.953892947902121e-07,
      "loss": 0.047,
      "step": 7859
    },
    {
      "epoch": 1.78839590443686,
      "grad_norm": 1.4465065149603678,
      "learning_rate": 8.953087351062835e-07,
      "loss": 0.0274,
      "step": 7860
    },
    {
      "epoch": 1.7886234357224118,
      "grad_norm": 1.972798509649138,
      "learning_rate": 8.952281698977394e-07,
      "loss": 0.0326,
      "step": 7861
    },
    {
      "epoch": 1.7888509670079635,
      "grad_norm": 1.34164452934922,
      "learning_rate": 8.951475991662263e-07,
      "loss": 0.0318,
      "step": 7862
    },
    {
      "epoch": 1.7890784982935153,
      "grad_norm": 1.212238624627934,
      "learning_rate": 8.950670229133912e-07,
      "loss": 0.0398,
      "step": 7863
    },
    {
      "epoch": 1.789306029579067,
      "grad_norm": 1.7900647347687184,
      "learning_rate": 8.949864411408807e-07,
      "loss": 0.056,
      "step": 7864
    },
    {
      "epoch": 1.7895335608646188,
      "grad_norm": 1.1479883805932574,
      "learning_rate": 8.949058538503416e-07,
      "loss": 0.0272,
      "step": 7865
    },
    {
      "epoch": 1.7897610921501705,
      "grad_norm": 0.8523729761501173,
      "learning_rate": 8.948252610434213e-07,
      "loss": 0.0216,
      "step": 7866
    },
    {
      "epoch": 1.7899886234357223,
      "grad_norm": 1.53956140553133,
      "learning_rate": 8.947446627217669e-07,
      "loss": 0.0369,
      "step": 7867
    },
    {
      "epoch": 1.790216154721274,
      "grad_norm": 0.9818481136003215,
      "learning_rate": 8.946640588870254e-07,
      "loss": 0.0284,
      "step": 7868
    },
    {
      "epoch": 1.7904436860068258,
      "grad_norm": 0.7124464532670008,
      "learning_rate": 8.945834495408447e-07,
      "loss": 0.014,
      "step": 7869
    },
    {
      "epoch": 1.7906712172923775,
      "grad_norm": 0.6767071176129998,
      "learning_rate": 8.945028346848718e-07,
      "loss": 0.0204,
      "step": 7870
    },
    {
      "epoch": 1.7908987485779293,
      "grad_norm": 1.195333260881732,
      "learning_rate": 8.944222143207545e-07,
      "loss": 0.0306,
      "step": 7871
    },
    {
      "epoch": 1.7911262798634813,
      "grad_norm": 1.7612928725245607,
      "learning_rate": 8.943415884501407e-07,
      "loss": 0.0242,
      "step": 7872
    },
    {
      "epoch": 1.791353811149033,
      "grad_norm": 0.8906098462117128,
      "learning_rate": 8.942609570746781e-07,
      "loss": 0.0155,
      "step": 7873
    },
    {
      "epoch": 1.7915813424345848,
      "grad_norm": 1.6234756574496614,
      "learning_rate": 8.941803201960146e-07,
      "loss": 0.0568,
      "step": 7874
    },
    {
      "epoch": 1.7918088737201365,
      "grad_norm": 1.3753803192692626,
      "learning_rate": 8.940996778157983e-07,
      "loss": 0.0504,
      "step": 7875
    },
    {
      "epoch": 1.7920364050056883,
      "grad_norm": 1.3303981402354177,
      "learning_rate": 8.940190299356774e-07,
      "loss": 0.0348,
      "step": 7876
    },
    {
      "epoch": 1.79226393629124,
      "grad_norm": 1.1381428529489983,
      "learning_rate": 8.939383765573004e-07,
      "loss": 0.0313,
      "step": 7877
    },
    {
      "epoch": 1.7924914675767918,
      "grad_norm": 0.6313270073845239,
      "learning_rate": 8.938577176823154e-07,
      "loss": 0.0133,
      "step": 7878
    },
    {
      "epoch": 1.7927189988623435,
      "grad_norm": 1.1526738136130819,
      "learning_rate": 8.93777053312371e-07,
      "loss": 0.0462,
      "step": 7879
    },
    {
      "epoch": 1.7929465301478953,
      "grad_norm": 1.4424825876427267,
      "learning_rate": 8.936963834491161e-07,
      "loss": 0.0192,
      "step": 7880
    },
    {
      "epoch": 1.7931740614334473,
      "grad_norm": 1.0566092128588358,
      "learning_rate": 8.93615708094199e-07,
      "loss": 0.0225,
      "step": 7881
    },
    {
      "epoch": 1.793401592718999,
      "grad_norm": 1.9176983642544965,
      "learning_rate": 8.935350272492687e-07,
      "loss": 0.0644,
      "step": 7882
    },
    {
      "epoch": 1.7936291240045508,
      "grad_norm": 0.9927270092271882,
      "learning_rate": 8.934543409159743e-07,
      "loss": 0.0196,
      "step": 7883
    },
    {
      "epoch": 1.7938566552901025,
      "grad_norm": 0.8956004254398622,
      "learning_rate": 8.933736490959649e-07,
      "loss": 0.0283,
      "step": 7884
    },
    {
      "epoch": 1.7940841865756543,
      "grad_norm": 0.8771623418465615,
      "learning_rate": 8.932929517908896e-07,
      "loss": 0.0276,
      "step": 7885
    },
    {
      "epoch": 1.794311717861206,
      "grad_norm": 0.8757995564864757,
      "learning_rate": 8.932122490023977e-07,
      "loss": 0.0178,
      "step": 7886
    },
    {
      "epoch": 1.7945392491467578,
      "grad_norm": 0.9619281755992323,
      "learning_rate": 8.931315407321387e-07,
      "loss": 0.0254,
      "step": 7887
    },
    {
      "epoch": 1.7947667804323095,
      "grad_norm": 0.830607334642633,
      "learning_rate": 8.93050826981762e-07,
      "loss": 0.0145,
      "step": 7888
    },
    {
      "epoch": 1.7949943117178613,
      "grad_norm": 0.9754645372250353,
      "learning_rate": 8.929701077529173e-07,
      "loss": 0.0248,
      "step": 7889
    },
    {
      "epoch": 1.795221843003413,
      "grad_norm": 1.3446910778885603,
      "learning_rate": 8.928893830472544e-07,
      "loss": 0.0257,
      "step": 7890
    },
    {
      "epoch": 1.7954493742889648,
      "grad_norm": 1.0600421240544748,
      "learning_rate": 8.92808652866423e-07,
      "loss": 0.0229,
      "step": 7891
    },
    {
      "epoch": 1.7956769055745165,
      "grad_norm": 1.3045169917709618,
      "learning_rate": 8.927279172120734e-07,
      "loss": 0.0345,
      "step": 7892
    },
    {
      "epoch": 1.7959044368600683,
      "grad_norm": 1.3535442753365499,
      "learning_rate": 8.926471760858554e-07,
      "loss": 0.0331,
      "step": 7893
    },
    {
      "epoch": 1.79613196814562,
      "grad_norm": 1.1986153843262288,
      "learning_rate": 8.925664294894193e-07,
      "loss": 0.0228,
      "step": 7894
    },
    {
      "epoch": 1.7963594994311718,
      "grad_norm": 1.6326418452338822,
      "learning_rate": 8.924856774244154e-07,
      "loss": 0.0439,
      "step": 7895
    },
    {
      "epoch": 1.7965870307167235,
      "grad_norm": 1.274533599454996,
      "learning_rate": 8.92404919892494e-07,
      "loss": 0.0284,
      "step": 7896
    },
    {
      "epoch": 1.7968145620022753,
      "grad_norm": 0.9245558198366477,
      "learning_rate": 8.923241568953061e-07,
      "loss": 0.025,
      "step": 7897
    },
    {
      "epoch": 1.797042093287827,
      "grad_norm": 1.4659383065797449,
      "learning_rate": 8.922433884345018e-07,
      "loss": 0.0359,
      "step": 7898
    },
    {
      "epoch": 1.7972696245733788,
      "grad_norm": 1.2678360189132305,
      "learning_rate": 8.921626145117321e-07,
      "loss": 0.045,
      "step": 7899
    },
    {
      "epoch": 1.7974971558589306,
      "grad_norm": 0.807399970096153,
      "learning_rate": 8.920818351286479e-07,
      "loss": 0.0165,
      "step": 7900
    },
    {
      "epoch": 1.7977246871444823,
      "grad_norm": 1.1167806568004228,
      "learning_rate": 8.920010502868999e-07,
      "loss": 0.0248,
      "step": 7901
    },
    {
      "epoch": 1.797952218430034,
      "grad_norm": 1.1293230465683268,
      "learning_rate": 8.919202599881395e-07,
      "loss": 0.0209,
      "step": 7902
    },
    {
      "epoch": 1.7981797497155858,
      "grad_norm": 1.0832700318486221,
      "learning_rate": 8.918394642340179e-07,
      "loss": 0.0221,
      "step": 7903
    },
    {
      "epoch": 1.7984072810011376,
      "grad_norm": 0.7064646043342412,
      "learning_rate": 8.917586630261864e-07,
      "loss": 0.0206,
      "step": 7904
    },
    {
      "epoch": 1.7986348122866893,
      "grad_norm": 1.4065933340491492,
      "learning_rate": 8.916778563662963e-07,
      "loss": 0.0303,
      "step": 7905
    },
    {
      "epoch": 1.798862343572241,
      "grad_norm": 0.6552459781539552,
      "learning_rate": 8.915970442559993e-07,
      "loss": 0.0132,
      "step": 7906
    },
    {
      "epoch": 1.7990898748577928,
      "grad_norm": 0.7452927540406299,
      "learning_rate": 8.915162266969469e-07,
      "loss": 0.0207,
      "step": 7907
    },
    {
      "epoch": 1.7993174061433446,
      "grad_norm": 1.4399513026164452,
      "learning_rate": 8.91435403690791e-07,
      "loss": 0.0292,
      "step": 7908
    },
    {
      "epoch": 1.7995449374288963,
      "grad_norm": 0.758350041185595,
      "learning_rate": 8.913545752391832e-07,
      "loss": 0.0149,
      "step": 7909
    },
    {
      "epoch": 1.799772468714448,
      "grad_norm": 0.9981336405534768,
      "learning_rate": 8.912737413437758e-07,
      "loss": 0.0181,
      "step": 7910
    },
    {
      "epoch": 1.8,
      "grad_norm": 0.8948519853122613,
      "learning_rate": 8.91192902006221e-07,
      "loss": 0.0137,
      "step": 7911
    },
    {
      "epoch": 1.8002275312855518,
      "grad_norm": 1.138919667694445,
      "learning_rate": 8.911120572281705e-07,
      "loss": 0.0377,
      "step": 7912
    },
    {
      "epoch": 1.8004550625711035,
      "grad_norm": 1.2169297926213631,
      "learning_rate": 8.91031207011277e-07,
      "loss": 0.0265,
      "step": 7913
    },
    {
      "epoch": 1.8006825938566553,
      "grad_norm": 1.0046099153521237,
      "learning_rate": 8.90950351357193e-07,
      "loss": 0.0165,
      "step": 7914
    },
    {
      "epoch": 1.800910125142207,
      "grad_norm": 1.2305460947745517,
      "learning_rate": 8.908694902675706e-07,
      "loss": 0.034,
      "step": 7915
    },
    {
      "epoch": 1.8011376564277588,
      "grad_norm": 0.908175336046685,
      "learning_rate": 8.907886237440627e-07,
      "loss": 0.0311,
      "step": 7916
    },
    {
      "epoch": 1.8013651877133106,
      "grad_norm": 0.9640989084574261,
      "learning_rate": 8.907077517883225e-07,
      "loss": 0.011,
      "step": 7917
    },
    {
      "epoch": 1.8015927189988623,
      "grad_norm": 1.0611776598333045,
      "learning_rate": 8.906268744020022e-07,
      "loss": 0.0391,
      "step": 7918
    },
    {
      "epoch": 1.801820250284414,
      "grad_norm": 1.2696769190669295,
      "learning_rate": 8.905459915867551e-07,
      "loss": 0.0394,
      "step": 7919
    },
    {
      "epoch": 1.802047781569966,
      "grad_norm": 0.9864570238447806,
      "learning_rate": 8.904651033442342e-07,
      "loss": 0.0209,
      "step": 7920
    },
    {
      "epoch": 1.8022753128555178,
      "grad_norm": 0.7898526012800101,
      "learning_rate": 8.903842096760929e-07,
      "loss": 0.0295,
      "step": 7921
    },
    {
      "epoch": 1.8025028441410695,
      "grad_norm": 1.1684045824501679,
      "learning_rate": 8.903033105839842e-07,
      "loss": 0.0316,
      "step": 7922
    },
    {
      "epoch": 1.8027303754266213,
      "grad_norm": 1.718325976581944,
      "learning_rate": 8.902224060695619e-07,
      "loss": 0.0553,
      "step": 7923
    },
    {
      "epoch": 1.802957906712173,
      "grad_norm": 1.1620730732976743,
      "learning_rate": 8.901414961344792e-07,
      "loss": 0.036,
      "step": 7924
    },
    {
      "epoch": 1.8031854379977248,
      "grad_norm": 1.6500968741389468,
      "learning_rate": 8.900605807803901e-07,
      "loss": 0.0398,
      "step": 7925
    },
    {
      "epoch": 1.8034129692832765,
      "grad_norm": 1.103134518928244,
      "learning_rate": 8.89979660008948e-07,
      "loss": 0.0242,
      "step": 7926
    },
    {
      "epoch": 1.8036405005688283,
      "grad_norm": 0.977204940987202,
      "learning_rate": 8.898987338218069e-07,
      "loss": 0.0202,
      "step": 7927
    },
    {
      "epoch": 1.80386803185438,
      "grad_norm": 1.0754200353452477,
      "learning_rate": 8.89817802220621e-07,
      "loss": 0.0227,
      "step": 7928
    },
    {
      "epoch": 1.8040955631399318,
      "grad_norm": 1.2777556299967832,
      "learning_rate": 8.89736865207044e-07,
      "loss": 0.026,
      "step": 7929
    },
    {
      "epoch": 1.8043230944254836,
      "grad_norm": 1.2235154401866535,
      "learning_rate": 8.896559227827305e-07,
      "loss": 0.031,
      "step": 7930
    },
    {
      "epoch": 1.8045506257110353,
      "grad_norm": 0.6503055008256462,
      "learning_rate": 8.895749749493346e-07,
      "loss": 0.0096,
      "step": 7931
    },
    {
      "epoch": 1.804778156996587,
      "grad_norm": 0.834214057724775,
      "learning_rate": 8.894940217085106e-07,
      "loss": 0.0155,
      "step": 7932
    },
    {
      "epoch": 1.8050056882821388,
      "grad_norm": 1.0672819385337902,
      "learning_rate": 8.894130630619133e-07,
      "loss": 0.0243,
      "step": 7933
    },
    {
      "epoch": 1.8052332195676906,
      "grad_norm": 0.9146202345603645,
      "learning_rate": 8.893320990111972e-07,
      "loss": 0.0185,
      "step": 7934
    },
    {
      "epoch": 1.8054607508532423,
      "grad_norm": 1.2276574359105301,
      "learning_rate": 8.892511295580172e-07,
      "loss": 0.0265,
      "step": 7935
    },
    {
      "epoch": 1.805688282138794,
      "grad_norm": 1.3625327539993874,
      "learning_rate": 8.891701547040281e-07,
      "loss": 0.0428,
      "step": 7936
    },
    {
      "epoch": 1.8059158134243458,
      "grad_norm": 0.8436216552179677,
      "learning_rate": 8.890891744508847e-07,
      "loss": 0.0157,
      "step": 7937
    },
    {
      "epoch": 1.8061433447098976,
      "grad_norm": 2.15371282783646,
      "learning_rate": 8.890081888002424e-07,
      "loss": 0.0624,
      "step": 7938
    },
    {
      "epoch": 1.8063708759954493,
      "grad_norm": 0.8801776627933559,
      "learning_rate": 8.88927197753756e-07,
      "loss": 0.024,
      "step": 7939
    },
    {
      "epoch": 1.806598407281001,
      "grad_norm": 0.9460686718278507,
      "learning_rate": 8.888462013130811e-07,
      "loss": 0.0246,
      "step": 7940
    },
    {
      "epoch": 1.8068259385665528,
      "grad_norm": 1.1341179520118103,
      "learning_rate": 8.887651994798732e-07,
      "loss": 0.0531,
      "step": 7941
    },
    {
      "epoch": 1.8070534698521046,
      "grad_norm": 0.83323675871723,
      "learning_rate": 8.886841922557876e-07,
      "loss": 0.0107,
      "step": 7942
    },
    {
      "epoch": 1.8072810011376563,
      "grad_norm": 1.769703851324121,
      "learning_rate": 8.8860317964248e-07,
      "loss": 0.0255,
      "step": 7943
    },
    {
      "epoch": 1.807508532423208,
      "grad_norm": 1.63313720621551,
      "learning_rate": 8.885221616416063e-07,
      "loss": 0.0419,
      "step": 7944
    },
    {
      "epoch": 1.8077360637087598,
      "grad_norm": 1.135504726716167,
      "learning_rate": 8.884411382548221e-07,
      "loss": 0.03,
      "step": 7945
    },
    {
      "epoch": 1.8079635949943116,
      "grad_norm": 1.2830667286887556,
      "learning_rate": 8.883601094837835e-07,
      "loss": 0.0321,
      "step": 7946
    },
    {
      "epoch": 1.8081911262798633,
      "grad_norm": 1.3598193317394558,
      "learning_rate": 8.882790753301465e-07,
      "loss": 0.0267,
      "step": 7947
    },
    {
      "epoch": 1.808418657565415,
      "grad_norm": 1.33412820168281,
      "learning_rate": 8.881980357955676e-07,
      "loss": 0.0241,
      "step": 7948
    },
    {
      "epoch": 1.808646188850967,
      "grad_norm": 1.658987742425555,
      "learning_rate": 8.881169908817028e-07,
      "loss": 0.0349,
      "step": 7949
    },
    {
      "epoch": 1.8088737201365188,
      "grad_norm": 0.4945242506035396,
      "learning_rate": 8.880359405902085e-07,
      "loss": 0.0159,
      "step": 7950
    },
    {
      "epoch": 1.8091012514220706,
      "grad_norm": 1.5533789527331354,
      "learning_rate": 8.879548849227413e-07,
      "loss": 0.0493,
      "step": 7951
    },
    {
      "epoch": 1.8093287827076223,
      "grad_norm": 1.78859864942384,
      "learning_rate": 8.87873823880958e-07,
      "loss": 0.0371,
      "step": 7952
    },
    {
      "epoch": 1.809556313993174,
      "grad_norm": 0.9977759236749801,
      "learning_rate": 8.877927574665149e-07,
      "loss": 0.0143,
      "step": 7953
    },
    {
      "epoch": 1.8097838452787258,
      "grad_norm": 1.7399862593633308,
      "learning_rate": 8.877116856810693e-07,
      "loss": 0.0389,
      "step": 7954
    },
    {
      "epoch": 1.8100113765642776,
      "grad_norm": 1.335838427331153,
      "learning_rate": 8.876306085262781e-07,
      "loss": 0.0215,
      "step": 7955
    },
    {
      "epoch": 1.8102389078498293,
      "grad_norm": 1.1320804254900072,
      "learning_rate": 8.875495260037979e-07,
      "loss": 0.0407,
      "step": 7956
    },
    {
      "epoch": 1.810466439135381,
      "grad_norm": 1.1642449596779554,
      "learning_rate": 8.874684381152865e-07,
      "loss": 0.021,
      "step": 7957
    },
    {
      "epoch": 1.8106939704209328,
      "grad_norm": 0.8676467298246693,
      "learning_rate": 8.873873448624008e-07,
      "loss": 0.0197,
      "step": 7958
    },
    {
      "epoch": 1.8109215017064848,
      "grad_norm": 0.8532877197343411,
      "learning_rate": 8.873062462467983e-07,
      "loss": 0.0148,
      "step": 7959
    },
    {
      "epoch": 1.8111490329920366,
      "grad_norm": 1.1008892392506717,
      "learning_rate": 8.872251422701366e-07,
      "loss": 0.0229,
      "step": 7960
    },
    {
      "epoch": 1.8113765642775883,
      "grad_norm": 0.7547039153153899,
      "learning_rate": 8.871440329340733e-07,
      "loss": 0.0247,
      "step": 7961
    },
    {
      "epoch": 1.81160409556314,
      "grad_norm": 1.815452580029783,
      "learning_rate": 8.870629182402659e-07,
      "loss": 0.0669,
      "step": 7962
    },
    {
      "epoch": 1.8118316268486918,
      "grad_norm": 1.3820207820518486,
      "learning_rate": 8.869817981903725e-07,
      "loss": 0.031,
      "step": 7963
    },
    {
      "epoch": 1.8120591581342436,
      "grad_norm": 0.8467210866515458,
      "learning_rate": 8.869006727860508e-07,
      "loss": 0.0204,
      "step": 7964
    },
    {
      "epoch": 1.8122866894197953,
      "grad_norm": 0.5195820151495079,
      "learning_rate": 8.868195420289591e-07,
      "loss": 0.006,
      "step": 7965
    },
    {
      "epoch": 1.812514220705347,
      "grad_norm": 1.158209961743232,
      "learning_rate": 8.867384059207554e-07,
      "loss": 0.032,
      "step": 7966
    },
    {
      "epoch": 1.8127417519908988,
      "grad_norm": 0.731730761445067,
      "learning_rate": 8.86657264463098e-07,
      "loss": 0.0122,
      "step": 7967
    },
    {
      "epoch": 1.8129692832764506,
      "grad_norm": 1.5433000241193469,
      "learning_rate": 8.865761176576457e-07,
      "loss": 0.0358,
      "step": 7968
    },
    {
      "epoch": 1.8131968145620023,
      "grad_norm": 0.9802801658756328,
      "learning_rate": 8.864949655060562e-07,
      "loss": 0.0254,
      "step": 7969
    },
    {
      "epoch": 1.813424345847554,
      "grad_norm": 1.2893269013156747,
      "learning_rate": 8.864138080099886e-07,
      "loss": 0.0344,
      "step": 7970
    },
    {
      "epoch": 1.8136518771331058,
      "grad_norm": 1.6378973935445966,
      "learning_rate": 8.863326451711015e-07,
      "loss": 0.0346,
      "step": 7971
    },
    {
      "epoch": 1.8138794084186576,
      "grad_norm": 1.5901382545769034,
      "learning_rate": 8.862514769910538e-07,
      "loss": 0.0257,
      "step": 7972
    },
    {
      "epoch": 1.8141069397042093,
      "grad_norm": 1.1125358321048289,
      "learning_rate": 8.861703034715042e-07,
      "loss": 0.0214,
      "step": 7973
    },
    {
      "epoch": 1.814334470989761,
      "grad_norm": 0.8342772107162001,
      "learning_rate": 8.860891246141123e-07,
      "loss": 0.0117,
      "step": 7974
    },
    {
      "epoch": 1.8145620022753128,
      "grad_norm": 1.3763064707775834,
      "learning_rate": 8.860079404205367e-07,
      "loss": 0.0244,
      "step": 7975
    },
    {
      "epoch": 1.8147895335608646,
      "grad_norm": 1.2902917037907535,
      "learning_rate": 8.859267508924366e-07,
      "loss": 0.0292,
      "step": 7976
    },
    {
      "epoch": 1.8150170648464163,
      "grad_norm": 0.9417112029914969,
      "learning_rate": 8.858455560314718e-07,
      "loss": 0.0152,
      "step": 7977
    },
    {
      "epoch": 1.815244596131968,
      "grad_norm": 1.228076700663053,
      "learning_rate": 8.857643558393015e-07,
      "loss": 0.0392,
      "step": 7978
    },
    {
      "epoch": 1.8154721274175198,
      "grad_norm": 1.3554843285182212,
      "learning_rate": 8.856831503175852e-07,
      "loss": 0.0425,
      "step": 7979
    },
    {
      "epoch": 1.8156996587030716,
      "grad_norm": 1.283048051846109,
      "learning_rate": 8.856019394679828e-07,
      "loss": 0.0145,
      "step": 7980
    },
    {
      "epoch": 1.8159271899886233,
      "grad_norm": 1.2300924493354228,
      "learning_rate": 8.855207232921541e-07,
      "loss": 0.0159,
      "step": 7981
    },
    {
      "epoch": 1.816154721274175,
      "grad_norm": 1.053461009988531,
      "learning_rate": 8.854395017917588e-07,
      "loss": 0.0212,
      "step": 7982
    },
    {
      "epoch": 1.8163822525597269,
      "grad_norm": 0.8458982455151165,
      "learning_rate": 8.853582749684571e-07,
      "loss": 0.0276,
      "step": 7983
    },
    {
      "epoch": 1.8166097838452786,
      "grad_norm": 0.7110026030106191,
      "learning_rate": 8.852770428239091e-07,
      "loss": 0.012,
      "step": 7984
    },
    {
      "epoch": 1.8168373151308304,
      "grad_norm": 1.3057855515690473,
      "learning_rate": 8.851958053597751e-07,
      "loss": 0.0393,
      "step": 7985
    },
    {
      "epoch": 1.817064846416382,
      "grad_norm": 1.3452318740996148,
      "learning_rate": 8.851145625777153e-07,
      "loss": 0.0508,
      "step": 7986
    },
    {
      "epoch": 1.8172923777019339,
      "grad_norm": 0.8218495774051612,
      "learning_rate": 8.850333144793903e-07,
      "loss": 0.0179,
      "step": 7987
    },
    {
      "epoch": 1.8175199089874858,
      "grad_norm": 0.5802176343383713,
      "learning_rate": 8.849520610664605e-07,
      "loss": 0.0038,
      "step": 7988
    },
    {
      "epoch": 1.8177474402730376,
      "grad_norm": 1.9961297192685032,
      "learning_rate": 8.848708023405866e-07,
      "loss": 0.0418,
      "step": 7989
    },
    {
      "epoch": 1.8179749715585893,
      "grad_norm": 0.8797672758098891,
      "learning_rate": 8.847895383034294e-07,
      "loss": 0.0101,
      "step": 7990
    },
    {
      "epoch": 1.818202502844141,
      "grad_norm": 1.3216697039016492,
      "learning_rate": 8.847082689566499e-07,
      "loss": 0.0273,
      "step": 7991
    },
    {
      "epoch": 1.8184300341296928,
      "grad_norm": 1.3162763009494356,
      "learning_rate": 8.846269943019091e-07,
      "loss": 0.0317,
      "step": 7992
    },
    {
      "epoch": 1.8186575654152446,
      "grad_norm": 1.3398769067970908,
      "learning_rate": 8.84545714340868e-07,
      "loss": 0.043,
      "step": 7993
    },
    {
      "epoch": 1.8188850967007963,
      "grad_norm": 1.3502712513593529,
      "learning_rate": 8.844644290751877e-07,
      "loss": 0.0276,
      "step": 7994
    },
    {
      "epoch": 1.819112627986348,
      "grad_norm": 1.3898881342636042,
      "learning_rate": 8.843831385065298e-07,
      "loss": 0.0475,
      "step": 7995
    },
    {
      "epoch": 1.8193401592718998,
      "grad_norm": 1.0577003818497825,
      "learning_rate": 8.843018426365555e-07,
      "loss": 0.0289,
      "step": 7996
    },
    {
      "epoch": 1.8195676905574516,
      "grad_norm": 1.3552733005663498,
      "learning_rate": 8.842205414669264e-07,
      "loss": 0.024,
      "step": 7997
    },
    {
      "epoch": 1.8197952218430036,
      "grad_norm": 0.6563826511647947,
      "learning_rate": 8.841392349993041e-07,
      "loss": 0.01,
      "step": 7998
    },
    {
      "epoch": 1.8200227531285553,
      "grad_norm": 0.9195258352180651,
      "learning_rate": 8.840579232353506e-07,
      "loss": 0.0203,
      "step": 7999
    },
    {
      "epoch": 1.820250284414107,
      "grad_norm": 0.9859767262646522,
      "learning_rate": 8.839766061767277e-07,
      "loss": 0.0268,
      "step": 8000
    },
    {
      "epoch": 1.8204778156996588,
      "grad_norm": 1.1374828137033943,
      "learning_rate": 8.838952838250971e-07,
      "loss": 0.0412,
      "step": 8001
    },
    {
      "epoch": 1.8207053469852106,
      "grad_norm": 1.6080148893216402,
      "learning_rate": 8.83813956182121e-07,
      "loss": 0.0557,
      "step": 8002
    },
    {
      "epoch": 1.8209328782707623,
      "grad_norm": 0.9689172649902232,
      "learning_rate": 8.837326232494616e-07,
      "loss": 0.018,
      "step": 8003
    },
    {
      "epoch": 1.821160409556314,
      "grad_norm": 1.869046995653937,
      "learning_rate": 8.836512850287812e-07,
      "loss": 0.0148,
      "step": 8004
    },
    {
      "epoch": 1.8213879408418658,
      "grad_norm": 1.0872933086687537,
      "learning_rate": 8.835699415217425e-07,
      "loss": 0.0149,
      "step": 8005
    },
    {
      "epoch": 1.8216154721274176,
      "grad_norm": 1.0450905191093498,
      "learning_rate": 8.834885927300075e-07,
      "loss": 0.0373,
      "step": 8006
    },
    {
      "epoch": 1.8218430034129693,
      "grad_norm": 1.5362767889787077,
      "learning_rate": 8.834072386552392e-07,
      "loss": 0.0384,
      "step": 8007
    },
    {
      "epoch": 1.822070534698521,
      "grad_norm": 1.2374289696704717,
      "learning_rate": 8.833258792991001e-07,
      "loss": 0.0324,
      "step": 8008
    },
    {
      "epoch": 1.8222980659840728,
      "grad_norm": 0.6400280975849308,
      "learning_rate": 8.832445146632531e-07,
      "loss": 0.0109,
      "step": 8009
    },
    {
      "epoch": 1.8225255972696246,
      "grad_norm": 1.3815605316771693,
      "learning_rate": 8.831631447493612e-07,
      "loss": 0.0322,
      "step": 8010
    },
    {
      "epoch": 1.8227531285551763,
      "grad_norm": 1.5167509474073282,
      "learning_rate": 8.830817695590875e-07,
      "loss": 0.0224,
      "step": 8011
    },
    {
      "epoch": 1.822980659840728,
      "grad_norm": 0.6230151252863323,
      "learning_rate": 8.830003890940953e-07,
      "loss": 0.0129,
      "step": 8012
    },
    {
      "epoch": 1.8232081911262799,
      "grad_norm": 1.096690926973307,
      "learning_rate": 8.829190033560473e-07,
      "loss": 0.0176,
      "step": 8013
    },
    {
      "epoch": 1.8234357224118316,
      "grad_norm": 0.9261907876289526,
      "learning_rate": 8.828376123466072e-07,
      "loss": 0.02,
      "step": 8014
    },
    {
      "epoch": 1.8236632536973834,
      "grad_norm": 0.9210807078893485,
      "learning_rate": 8.827562160674387e-07,
      "loss": 0.0268,
      "step": 8015
    },
    {
      "epoch": 1.823890784982935,
      "grad_norm": 1.0197961436664829,
      "learning_rate": 8.826748145202052e-07,
      "loss": 0.0302,
      "step": 8016
    },
    {
      "epoch": 1.8241183162684869,
      "grad_norm": 0.8048295376769727,
      "learning_rate": 8.8259340770657e-07,
      "loss": 0.0254,
      "step": 8017
    },
    {
      "epoch": 1.8243458475540386,
      "grad_norm": 0.656143016454103,
      "learning_rate": 8.82511995628198e-07,
      "loss": 0.0121,
      "step": 8018
    },
    {
      "epoch": 1.8245733788395904,
      "grad_norm": 0.8966243554035213,
      "learning_rate": 8.824305782867521e-07,
      "loss": 0.0247,
      "step": 8019
    },
    {
      "epoch": 1.8248009101251421,
      "grad_norm": 1.1794697891283992,
      "learning_rate": 8.823491556838964e-07,
      "loss": 0.019,
      "step": 8020
    },
    {
      "epoch": 1.8250284414106939,
      "grad_norm": 1.1709408047651713,
      "learning_rate": 8.822677278212955e-07,
      "loss": 0.0255,
      "step": 8021
    },
    {
      "epoch": 1.8252559726962456,
      "grad_norm": 1.5148678871584642,
      "learning_rate": 8.821862947006134e-07,
      "loss": 0.0318,
      "step": 8022
    },
    {
      "epoch": 1.8254835039817974,
      "grad_norm": 1.0770996966783146,
      "learning_rate": 8.821048563235143e-07,
      "loss": 0.0223,
      "step": 8023
    },
    {
      "epoch": 1.8257110352673491,
      "grad_norm": 0.8636016370138058,
      "learning_rate": 8.820234126916631e-07,
      "loss": 0.026,
      "step": 8024
    },
    {
      "epoch": 1.8259385665529009,
      "grad_norm": 1.3682922135783069,
      "learning_rate": 8.81941963806724e-07,
      "loss": 0.0381,
      "step": 8025
    },
    {
      "epoch": 1.8261660978384526,
      "grad_norm": 0.9348631362839371,
      "learning_rate": 8.818605096703614e-07,
      "loss": 0.0209,
      "step": 8026
    },
    {
      "epoch": 1.8263936291240046,
      "grad_norm": 0.4823581924596872,
      "learning_rate": 8.817790502842406e-07,
      "loss": 0.0068,
      "step": 8027
    },
    {
      "epoch": 1.8266211604095564,
      "grad_norm": 1.0951844413599716,
      "learning_rate": 8.816975856500264e-07,
      "loss": 0.0458,
      "step": 8028
    },
    {
      "epoch": 1.826848691695108,
      "grad_norm": 0.6118985572179404,
      "learning_rate": 8.816161157693837e-07,
      "loss": 0.0093,
      "step": 8029
    },
    {
      "epoch": 1.8270762229806599,
      "grad_norm": 1.3843657752880825,
      "learning_rate": 8.815346406439773e-07,
      "loss": 0.0308,
      "step": 8030
    },
    {
      "epoch": 1.8273037542662116,
      "grad_norm": 1.4803326193372828,
      "learning_rate": 8.814531602754728e-07,
      "loss": 0.0299,
      "step": 8031
    },
    {
      "epoch": 1.8275312855517634,
      "grad_norm": 1.1534269433987345,
      "learning_rate": 8.813716746655354e-07,
      "loss": 0.0524,
      "step": 8032
    },
    {
      "epoch": 1.8277588168373151,
      "grad_norm": 1.5295917320213728,
      "learning_rate": 8.812901838158304e-07,
      "loss": 0.0294,
      "step": 8033
    },
    {
      "epoch": 1.8279863481228669,
      "grad_norm": 1.0053889970420802,
      "learning_rate": 8.812086877280234e-07,
      "loss": 0.0246,
      "step": 8034
    },
    {
      "epoch": 1.8282138794084186,
      "grad_norm": 1.3826048330122502,
      "learning_rate": 8.811271864037802e-07,
      "loss": 0.0208,
      "step": 8035
    },
    {
      "epoch": 1.8284414106939704,
      "grad_norm": 1.166250279335265,
      "learning_rate": 8.810456798447664e-07,
      "loss": 0.0259,
      "step": 8036
    },
    {
      "epoch": 1.8286689419795223,
      "grad_norm": 1.2992210361531946,
      "learning_rate": 8.809641680526477e-07,
      "loss": 0.0357,
      "step": 8037
    },
    {
      "epoch": 1.828896473265074,
      "grad_norm": 1.2611943101937177,
      "learning_rate": 8.808826510290904e-07,
      "loss": 0.025,
      "step": 8038
    },
    {
      "epoch": 1.8291240045506258,
      "grad_norm": 0.8944973994876296,
      "learning_rate": 8.808011287757601e-07,
      "loss": 0.0207,
      "step": 8039
    },
    {
      "epoch": 1.8293515358361776,
      "grad_norm": 0.6737233324041566,
      "learning_rate": 8.807196012943231e-07,
      "loss": 0.0153,
      "step": 8040
    },
    {
      "epoch": 1.8295790671217294,
      "grad_norm": 0.9209907577540047,
      "learning_rate": 8.80638068586446e-07,
      "loss": 0.0206,
      "step": 8041
    },
    {
      "epoch": 1.829806598407281,
      "grad_norm": 1.8380608111866295,
      "learning_rate": 8.805565306537949e-07,
      "loss": 0.068,
      "step": 8042
    },
    {
      "epoch": 1.8300341296928329,
      "grad_norm": 1.4638737163478475,
      "learning_rate": 8.804749874980364e-07,
      "loss": 0.051,
      "step": 8043
    },
    {
      "epoch": 1.8302616609783846,
      "grad_norm": 0.8030032240145646,
      "learning_rate": 8.80393439120837e-07,
      "loss": 0.0128,
      "step": 8044
    },
    {
      "epoch": 1.8304891922639364,
      "grad_norm": 1.479151025541696,
      "learning_rate": 8.803118855238635e-07,
      "loss": 0.0245,
      "step": 8045
    },
    {
      "epoch": 1.8307167235494881,
      "grad_norm": 1.551454930794604,
      "learning_rate": 8.802303267087825e-07,
      "loss": 0.0454,
      "step": 8046
    },
    {
      "epoch": 1.8309442548350399,
      "grad_norm": 1.0588321775780563,
      "learning_rate": 8.801487626772611e-07,
      "loss": 0.0229,
      "step": 8047
    },
    {
      "epoch": 1.8311717861205916,
      "grad_norm": 0.9880309790661209,
      "learning_rate": 8.800671934309663e-07,
      "loss": 0.0283,
      "step": 8048
    },
    {
      "epoch": 1.8313993174061434,
      "grad_norm": 1.0102582573010799,
      "learning_rate": 8.799856189715653e-07,
      "loss": 0.0159,
      "step": 8049
    },
    {
      "epoch": 1.8316268486916951,
      "grad_norm": 1.5479135317155381,
      "learning_rate": 8.79904039300725e-07,
      "loss": 0.0267,
      "step": 8050
    },
    {
      "epoch": 1.8318543799772469,
      "grad_norm": 1.2096856484161966,
      "learning_rate": 8.798224544201132e-07,
      "loss": 0.019,
      "step": 8051
    },
    {
      "epoch": 1.8320819112627986,
      "grad_norm": 0.8689124435880402,
      "learning_rate": 8.79740864331397e-07,
      "loss": 0.0149,
      "step": 8052
    },
    {
      "epoch": 1.8323094425483504,
      "grad_norm": 0.7456276512123506,
      "learning_rate": 8.796592690362439e-07,
      "loss": 0.0138,
      "step": 8053
    },
    {
      "epoch": 1.8325369738339021,
      "grad_norm": 0.7019017016852895,
      "learning_rate": 8.795776685363219e-07,
      "loss": 0.0178,
      "step": 8054
    },
    {
      "epoch": 1.8327645051194539,
      "grad_norm": 1.6867823859871856,
      "learning_rate": 8.794960628332986e-07,
      "loss": 0.0354,
      "step": 8055
    },
    {
      "epoch": 1.8329920364050056,
      "grad_norm": 0.7129105859531477,
      "learning_rate": 8.794144519288419e-07,
      "loss": 0.0178,
      "step": 8056
    },
    {
      "epoch": 1.8332195676905574,
      "grad_norm": 1.397218489443679,
      "learning_rate": 8.793328358246198e-07,
      "loss": 0.0428,
      "step": 8057
    },
    {
      "epoch": 1.8334470989761091,
      "grad_norm": 0.9872685652054035,
      "learning_rate": 8.792512145223002e-07,
      "loss": 0.0188,
      "step": 8058
    },
    {
      "epoch": 1.8336746302616609,
      "grad_norm": 0.8253338993696552,
      "learning_rate": 8.791695880235515e-07,
      "loss": 0.0254,
      "step": 8059
    },
    {
      "epoch": 1.8339021615472126,
      "grad_norm": 1.4428772827676484,
      "learning_rate": 8.790879563300416e-07,
      "loss": 0.0415,
      "step": 8060
    },
    {
      "epoch": 1.8341296928327644,
      "grad_norm": 0.4796256176224381,
      "learning_rate": 8.790063194434397e-07,
      "loss": 0.0097,
      "step": 8061
    },
    {
      "epoch": 1.8343572241183161,
      "grad_norm": 1.3614523308348847,
      "learning_rate": 8.789246773654136e-07,
      "loss": 0.0346,
      "step": 8062
    },
    {
      "epoch": 1.834584755403868,
      "grad_norm": 0.7218128847996856,
      "learning_rate": 8.788430300976321e-07,
      "loss": 0.0172,
      "step": 8063
    },
    {
      "epoch": 1.8348122866894196,
      "grad_norm": 1.126640751968136,
      "learning_rate": 8.787613776417639e-07,
      "loss": 0.0214,
      "step": 8064
    },
    {
      "epoch": 1.8350398179749714,
      "grad_norm": 0.7030643016032253,
      "learning_rate": 8.78679719999478e-07,
      "loss": 0.0161,
      "step": 8065
    },
    {
      "epoch": 1.8352673492605234,
      "grad_norm": 1.6419687984973874,
      "learning_rate": 8.785980571724433e-07,
      "loss": 0.0374,
      "step": 8066
    },
    {
      "epoch": 1.8354948805460751,
      "grad_norm": 0.8975514934459169,
      "learning_rate": 8.785163891623284e-07,
      "loss": 0.022,
      "step": 8067
    },
    {
      "epoch": 1.8357224118316269,
      "grad_norm": 0.9300496560802327,
      "learning_rate": 8.784347159708033e-07,
      "loss": 0.0199,
      "step": 8068
    },
    {
      "epoch": 1.8359499431171786,
      "grad_norm": 1.4971241023373771,
      "learning_rate": 8.783530375995364e-07,
      "loss": 0.0266,
      "step": 8069
    },
    {
      "epoch": 1.8361774744027304,
      "grad_norm": 0.9699889820724317,
      "learning_rate": 8.782713540501975e-07,
      "loss": 0.0142,
      "step": 8070
    },
    {
      "epoch": 1.8364050056882821,
      "grad_norm": 1.2128464569079203,
      "learning_rate": 8.78189665324456e-07,
      "loss": 0.0385,
      "step": 8071
    },
    {
      "epoch": 1.8366325369738339,
      "grad_norm": 1.1994951742697528,
      "learning_rate": 8.781079714239812e-07,
      "loss": 0.0376,
      "step": 8072
    },
    {
      "epoch": 1.8368600682593856,
      "grad_norm": 1.1920945512599561,
      "learning_rate": 8.780262723504434e-07,
      "loss": 0.0276,
      "step": 8073
    },
    {
      "epoch": 1.8370875995449374,
      "grad_norm": 1.3444176419784304,
      "learning_rate": 8.779445681055115e-07,
      "loss": 0.0282,
      "step": 8074
    },
    {
      "epoch": 1.8373151308304891,
      "grad_norm": 1.170851765126466,
      "learning_rate": 8.778628586908563e-07,
      "loss": 0.0314,
      "step": 8075
    },
    {
      "epoch": 1.8375426621160411,
      "grad_norm": 1.2510038912360684,
      "learning_rate": 8.777811441081475e-07,
      "loss": 0.0331,
      "step": 8076
    },
    {
      "epoch": 1.8377701934015929,
      "grad_norm": 0.9620868665247972,
      "learning_rate": 8.776994243590545e-07,
      "loss": 0.0183,
      "step": 8077
    },
    {
      "epoch": 1.8379977246871446,
      "grad_norm": 1.6724076249826632,
      "learning_rate": 8.776176994452485e-07,
      "loss": 0.0296,
      "step": 8078
    },
    {
      "epoch": 1.8382252559726964,
      "grad_norm": 1.2051672402979108,
      "learning_rate": 8.775359693683991e-07,
      "loss": 0.0271,
      "step": 8079
    },
    {
      "epoch": 1.8384527872582481,
      "grad_norm": 0.9151061745096992,
      "learning_rate": 8.77454234130177e-07,
      "loss": 0.0378,
      "step": 8080
    },
    {
      "epoch": 1.8386803185437999,
      "grad_norm": 0.7885192147031999,
      "learning_rate": 8.773724937322531e-07,
      "loss": 0.0276,
      "step": 8081
    },
    {
      "epoch": 1.8389078498293516,
      "grad_norm": 1.2112579740172957,
      "learning_rate": 8.772907481762973e-07,
      "loss": 0.0255,
      "step": 8082
    },
    {
      "epoch": 1.8391353811149034,
      "grad_norm": 1.0373152134098131,
      "learning_rate": 8.772089974639806e-07,
      "loss": 0.0411,
      "step": 8083
    },
    {
      "epoch": 1.8393629124004551,
      "grad_norm": 1.4909226278812742,
      "learning_rate": 8.77127241596974e-07,
      "loss": 0.0312,
      "step": 8084
    },
    {
      "epoch": 1.8395904436860069,
      "grad_norm": 0.745134622479357,
      "learning_rate": 8.770454805769482e-07,
      "loss": 0.0132,
      "step": 8085
    },
    {
      "epoch": 1.8398179749715586,
      "grad_norm": 0.8888480022796346,
      "learning_rate": 8.769637144055745e-07,
      "loss": 0.0184,
      "step": 8086
    },
    {
      "epoch": 1.8400455062571104,
      "grad_norm": 0.6860545499445614,
      "learning_rate": 8.768819430845241e-07,
      "loss": 0.0125,
      "step": 8087
    },
    {
      "epoch": 1.8402730375426621,
      "grad_norm": 1.072302307300318,
      "learning_rate": 8.768001666154678e-07,
      "loss": 0.0298,
      "step": 8088
    },
    {
      "epoch": 1.840500568828214,
      "grad_norm": 1.4163122617295698,
      "learning_rate": 8.767183850000774e-07,
      "loss": 0.0281,
      "step": 8089
    },
    {
      "epoch": 1.8407281001137656,
      "grad_norm": 1.4432293918074388,
      "learning_rate": 8.76636598240024e-07,
      "loss": 0.0387,
      "step": 8090
    },
    {
      "epoch": 1.8409556313993174,
      "grad_norm": 1.2795674829954617,
      "learning_rate": 8.765548063369795e-07,
      "loss": 0.0333,
      "step": 8091
    },
    {
      "epoch": 1.8411831626848691,
      "grad_norm": 1.462900154174469,
      "learning_rate": 8.764730092926156e-07,
      "loss": 0.0318,
      "step": 8092
    },
    {
      "epoch": 1.841410693970421,
      "grad_norm": 1.4294887556185045,
      "learning_rate": 8.763912071086039e-07,
      "loss": 0.0329,
      "step": 8093
    },
    {
      "epoch": 1.8416382252559726,
      "grad_norm": 0.8439752476671738,
      "learning_rate": 8.763093997866162e-07,
      "loss": 0.0135,
      "step": 8094
    },
    {
      "epoch": 1.8418657565415244,
      "grad_norm": 1.1771260310982554,
      "learning_rate": 8.762275873283248e-07,
      "loss": 0.0147,
      "step": 8095
    },
    {
      "epoch": 1.8420932878270762,
      "grad_norm": 0.9091367917683557,
      "learning_rate": 8.761457697354015e-07,
      "loss": 0.016,
      "step": 8096
    },
    {
      "epoch": 1.842320819112628,
      "grad_norm": 1.1114412202510104,
      "learning_rate": 8.760639470095185e-07,
      "loss": 0.018,
      "step": 8097
    },
    {
      "epoch": 1.8425483503981797,
      "grad_norm": 1.1552516434606968,
      "learning_rate": 8.759821191523483e-07,
      "loss": 0.0338,
      "step": 8098
    },
    {
      "epoch": 1.8427758816837314,
      "grad_norm": 1.2468801216425964,
      "learning_rate": 8.759002861655633e-07,
      "loss": 0.0448,
      "step": 8099
    },
    {
      "epoch": 1.8430034129692832,
      "grad_norm": 39.45012838572698,
      "learning_rate": 8.75818448050836e-07,
      "loss": 0.2857,
      "step": 8100
    },
    {
      "epoch": 1.843230944254835,
      "grad_norm": 1.8524240850943434,
      "learning_rate": 8.757366048098389e-07,
      "loss": 0.0466,
      "step": 8101
    },
    {
      "epoch": 1.8434584755403867,
      "grad_norm": 2.1307221074840736,
      "learning_rate": 8.756547564442449e-07,
      "loss": 0.0401,
      "step": 8102
    },
    {
      "epoch": 1.8436860068259384,
      "grad_norm": 1.6032119560353542,
      "learning_rate": 8.755729029557266e-07,
      "loss": 0.0459,
      "step": 8103
    },
    {
      "epoch": 1.8439135381114902,
      "grad_norm": 1.5350656489130663,
      "learning_rate": 8.75491044345957e-07,
      "loss": 0.0358,
      "step": 8104
    },
    {
      "epoch": 1.8441410693970421,
      "grad_norm": 1.8138288140149013,
      "learning_rate": 8.754091806166092e-07,
      "loss": 0.02,
      "step": 8105
    },
    {
      "epoch": 1.844368600682594,
      "grad_norm": 0.8666408926248154,
      "learning_rate": 8.753273117693567e-07,
      "loss": 0.0385,
      "step": 8106
    },
    {
      "epoch": 1.8445961319681456,
      "grad_norm": 0.9479917894975077,
      "learning_rate": 8.752454378058721e-07,
      "loss": 0.0207,
      "step": 8107
    },
    {
      "epoch": 1.8448236632536974,
      "grad_norm": 1.154157519846883,
      "learning_rate": 8.751635587278291e-07,
      "loss": 0.0281,
      "step": 8108
    },
    {
      "epoch": 1.8450511945392492,
      "grad_norm": 1.6293722575021,
      "learning_rate": 8.750816745369012e-07,
      "loss": 0.0635,
      "step": 8109
    },
    {
      "epoch": 1.845278725824801,
      "grad_norm": 0.8664652175714467,
      "learning_rate": 8.749997852347619e-07,
      "loss": 0.0246,
      "step": 8110
    },
    {
      "epoch": 1.8455062571103527,
      "grad_norm": 1.1023719369301574,
      "learning_rate": 8.749178908230845e-07,
      "loss": 0.0267,
      "step": 8111
    },
    {
      "epoch": 1.8457337883959044,
      "grad_norm": 1.9613190675321726,
      "learning_rate": 8.748359913035436e-07,
      "loss": 0.0594,
      "step": 8112
    },
    {
      "epoch": 1.8459613196814562,
      "grad_norm": 0.8660552025551523,
      "learning_rate": 8.747540866778124e-07,
      "loss": 0.0199,
      "step": 8113
    },
    {
      "epoch": 1.846188850967008,
      "grad_norm": 0.9647974529318043,
      "learning_rate": 8.74672176947565e-07,
      "loss": 0.0177,
      "step": 8114
    },
    {
      "epoch": 1.8464163822525599,
      "grad_norm": 0.9109302295934834,
      "learning_rate": 8.745902621144755e-07,
      "loss": 0.0103,
      "step": 8115
    },
    {
      "epoch": 1.8466439135381116,
      "grad_norm": 1.0544271239090326,
      "learning_rate": 8.745083421802183e-07,
      "loss": 0.0252,
      "step": 8116
    },
    {
      "epoch": 1.8468714448236634,
      "grad_norm": 0.9489229943769115,
      "learning_rate": 8.744264171464673e-07,
      "loss": 0.0226,
      "step": 8117
    },
    {
      "epoch": 1.8470989761092151,
      "grad_norm": 1.213440014870784,
      "learning_rate": 8.743444870148974e-07,
      "loss": 0.0437,
      "step": 8118
    },
    {
      "epoch": 1.847326507394767,
      "grad_norm": 0.7128445249897901,
      "learning_rate": 8.742625517871828e-07,
      "loss": 0.0242,
      "step": 8119
    },
    {
      "epoch": 1.8475540386803186,
      "grad_norm": 0.7222931636864752,
      "learning_rate": 8.74180611464998e-07,
      "loss": 0.0169,
      "step": 8120
    },
    {
      "epoch": 1.8477815699658704,
      "grad_norm": 0.7458819467430629,
      "learning_rate": 8.74098666050018e-07,
      "loss": 0.0142,
      "step": 8121
    },
    {
      "epoch": 1.8480091012514221,
      "grad_norm": 1.0424051331077842,
      "learning_rate": 8.740167155439173e-07,
      "loss": 0.0281,
      "step": 8122
    },
    {
      "epoch": 1.848236632536974,
      "grad_norm": 0.670541698875183,
      "learning_rate": 8.739347599483712e-07,
      "loss": 0.0174,
      "step": 8123
    },
    {
      "epoch": 1.8484641638225257,
      "grad_norm": 1.5332876404909188,
      "learning_rate": 8.738527992650542e-07,
      "loss": 0.0402,
      "step": 8124
    },
    {
      "epoch": 1.8486916951080774,
      "grad_norm": 1.3030714005810538,
      "learning_rate": 8.737708334956421e-07,
      "loss": 0.0197,
      "step": 8125
    },
    {
      "epoch": 1.8489192263936292,
      "grad_norm": 0.5780610713137618,
      "learning_rate": 8.736888626418097e-07,
      "loss": 0.0081,
      "step": 8126
    },
    {
      "epoch": 1.849146757679181,
      "grad_norm": 1.7468565482433127,
      "learning_rate": 8.736068867052322e-07,
      "loss": 0.0519,
      "step": 8127
    },
    {
      "epoch": 1.8493742889647327,
      "grad_norm": 1.328356788559963,
      "learning_rate": 8.735249056875852e-07,
      "loss": 0.0341,
      "step": 8128
    },
    {
      "epoch": 1.8496018202502844,
      "grad_norm": 1.3535479602581053,
      "learning_rate": 8.734429195905446e-07,
      "loss": 0.0312,
      "step": 8129
    },
    {
      "epoch": 1.8498293515358362,
      "grad_norm": 1.0241814023510956,
      "learning_rate": 8.733609284157855e-07,
      "loss": 0.0172,
      "step": 8130
    },
    {
      "epoch": 1.850056882821388,
      "grad_norm": 0.5782127255787763,
      "learning_rate": 8.73278932164984e-07,
      "loss": 0.0073,
      "step": 8131
    },
    {
      "epoch": 1.8502844141069397,
      "grad_norm": 1.5278820863630018,
      "learning_rate": 8.731969308398158e-07,
      "loss": 0.0238,
      "step": 8132
    },
    {
      "epoch": 1.8505119453924914,
      "grad_norm": 0.9456727006478703,
      "learning_rate": 8.731149244419568e-07,
      "loss": 0.0142,
      "step": 8133
    },
    {
      "epoch": 1.8507394766780432,
      "grad_norm": 1.1745932934806986,
      "learning_rate": 8.73032912973083e-07,
      "loss": 0.0342,
      "step": 8134
    },
    {
      "epoch": 1.850967007963595,
      "grad_norm": 1.190670680670907,
      "learning_rate": 8.72950896434871e-07,
      "loss": 0.0228,
      "step": 8135
    },
    {
      "epoch": 1.8511945392491467,
      "grad_norm": 1.1365337146814314,
      "learning_rate": 8.728688748289966e-07,
      "loss": 0.0282,
      "step": 8136
    },
    {
      "epoch": 1.8514220705346984,
      "grad_norm": 1.323317823211618,
      "learning_rate": 8.727868481571365e-07,
      "loss": 0.0402,
      "step": 8137
    },
    {
      "epoch": 1.8516496018202502,
      "grad_norm": 1.4579556474234914,
      "learning_rate": 8.72704816420967e-07,
      "loss": 0.0276,
      "step": 8138
    },
    {
      "epoch": 1.851877133105802,
      "grad_norm": 1.1704165060413627,
      "learning_rate": 8.726227796221646e-07,
      "loss": 0.0155,
      "step": 8139
    },
    {
      "epoch": 1.8521046643913537,
      "grad_norm": 1.4181298343589945,
      "learning_rate": 8.725407377624062e-07,
      "loss": 0.0167,
      "step": 8140
    },
    {
      "epoch": 1.8523321956769054,
      "grad_norm": 1.759030380654703,
      "learning_rate": 8.724586908433682e-07,
      "loss": 0.0335,
      "step": 8141
    },
    {
      "epoch": 1.8525597269624572,
      "grad_norm": 0.9073385097866469,
      "learning_rate": 8.72376638866728e-07,
      "loss": 0.0214,
      "step": 8142
    },
    {
      "epoch": 1.852787258248009,
      "grad_norm": 1.669887811223168,
      "learning_rate": 8.722945818341624e-07,
      "loss": 0.0436,
      "step": 8143
    },
    {
      "epoch": 1.853014789533561,
      "grad_norm": 0.9318869567954254,
      "learning_rate": 8.722125197473483e-07,
      "loss": 0.0272,
      "step": 8144
    },
    {
      "epoch": 1.8532423208191127,
      "grad_norm": 0.6154076945150099,
      "learning_rate": 8.721304526079631e-07,
      "loss": 0.0104,
      "step": 8145
    },
    {
      "epoch": 1.8534698521046644,
      "grad_norm": 1.016481897300851,
      "learning_rate": 8.72048380417684e-07,
      "loss": 0.0178,
      "step": 8146
    },
    {
      "epoch": 1.8536973833902162,
      "grad_norm": 1.5921420686048406,
      "learning_rate": 8.719663031781884e-07,
      "loss": 0.0401,
      "step": 8147
    },
    {
      "epoch": 1.853924914675768,
      "grad_norm": 1.1899281281195604,
      "learning_rate": 8.71884220891154e-07,
      "loss": 0.0401,
      "step": 8148
    },
    {
      "epoch": 1.8541524459613197,
      "grad_norm": 0.7938509448775539,
      "learning_rate": 8.718021335582583e-07,
      "loss": 0.0239,
      "step": 8149
    },
    {
      "epoch": 1.8543799772468714,
      "grad_norm": 1.5250257259638016,
      "learning_rate": 8.71720041181179e-07,
      "loss": 0.0596,
      "step": 8150
    },
    {
      "epoch": 1.8546075085324232,
      "grad_norm": 0.9788138895139992,
      "learning_rate": 8.716379437615937e-07,
      "loss": 0.0278,
      "step": 8151
    },
    {
      "epoch": 1.854835039817975,
      "grad_norm": 1.1219617460465487,
      "learning_rate": 8.715558413011807e-07,
      "loss": 0.0145,
      "step": 8152
    },
    {
      "epoch": 1.8550625711035267,
      "grad_norm": 1.4696274577201667,
      "learning_rate": 8.714737338016178e-07,
      "loss": 0.0299,
      "step": 8153
    },
    {
      "epoch": 1.8552901023890787,
      "grad_norm": 0.5461938735927995,
      "learning_rate": 8.713916212645832e-07,
      "loss": 0.0087,
      "step": 8154
    },
    {
      "epoch": 1.8555176336746304,
      "grad_norm": 0.8728191580515667,
      "learning_rate": 8.713095036917551e-07,
      "loss": 0.0155,
      "step": 8155
    },
    {
      "epoch": 1.8557451649601822,
      "grad_norm": 0.8112633112836988,
      "learning_rate": 8.712273810848118e-07,
      "loss": 0.0279,
      "step": 8156
    },
    {
      "epoch": 1.855972696245734,
      "grad_norm": 1.6292544002783556,
      "learning_rate": 8.711452534454318e-07,
      "loss": 0.0473,
      "step": 8157
    },
    {
      "epoch": 1.8562002275312857,
      "grad_norm": 1.0397332926990757,
      "learning_rate": 8.710631207752936e-07,
      "loss": 0.013,
      "step": 8158
    },
    {
      "epoch": 1.8564277588168374,
      "grad_norm": 1.2786813393181724,
      "learning_rate": 8.709809830760759e-07,
      "loss": 0.0391,
      "step": 8159
    },
    {
      "epoch": 1.8566552901023892,
      "grad_norm": 0.6749339564730344,
      "learning_rate": 8.708988403494572e-07,
      "loss": 0.0101,
      "step": 8160
    },
    {
      "epoch": 1.856882821387941,
      "grad_norm": 1.6928146286387011,
      "learning_rate": 8.708166925971168e-07,
      "loss": 0.0387,
      "step": 8161
    },
    {
      "epoch": 1.8571103526734927,
      "grad_norm": 1.0430859648850737,
      "learning_rate": 8.707345398207332e-07,
      "loss": 0.0219,
      "step": 8162
    },
    {
      "epoch": 1.8573378839590444,
      "grad_norm": 1.5665708380160561,
      "learning_rate": 8.706523820219858e-07,
      "loss": 0.0368,
      "step": 8163
    },
    {
      "epoch": 1.8575654152445962,
      "grad_norm": 1.2966054046900626,
      "learning_rate": 8.705702192025537e-07,
      "loss": 0.0238,
      "step": 8164
    },
    {
      "epoch": 1.857792946530148,
      "grad_norm": 0.7008382933501183,
      "learning_rate": 8.704880513641158e-07,
      "loss": 0.017,
      "step": 8165
    },
    {
      "epoch": 1.8580204778156997,
      "grad_norm": 1.7406243621129194,
      "learning_rate": 8.704058785083519e-07,
      "loss": 0.0288,
      "step": 8166
    },
    {
      "epoch": 1.8582480091012514,
      "grad_norm": 1.288425765941536,
      "learning_rate": 8.703237006369411e-07,
      "loss": 0.0242,
      "step": 8167
    },
    {
      "epoch": 1.8584755403868032,
      "grad_norm": 1.1833551322412017,
      "learning_rate": 8.702415177515633e-07,
      "loss": 0.0305,
      "step": 8168
    },
    {
      "epoch": 1.858703071672355,
      "grad_norm": 1.1465262412236228,
      "learning_rate": 8.70159329853898e-07,
      "loss": 0.0301,
      "step": 8169
    },
    {
      "epoch": 1.8589306029579067,
      "grad_norm": 0.7748327018464876,
      "learning_rate": 8.700771369456249e-07,
      "loss": 0.0209,
      "step": 8170
    },
    {
      "epoch": 1.8591581342434584,
      "grad_norm": 1.6454633052536287,
      "learning_rate": 8.69994939028424e-07,
      "loss": 0.0299,
      "step": 8171
    },
    {
      "epoch": 1.8593856655290102,
      "grad_norm": 1.5662753124634492,
      "learning_rate": 8.699127361039753e-07,
      "loss": 0.0176,
      "step": 8172
    },
    {
      "epoch": 1.859613196814562,
      "grad_norm": 1.131248911574669,
      "learning_rate": 8.698305281739589e-07,
      "loss": 0.0201,
      "step": 8173
    },
    {
      "epoch": 1.8598407281001137,
      "grad_norm": 0.694155214467786,
      "learning_rate": 8.697483152400546e-07,
      "loss": 0.0133,
      "step": 8174
    },
    {
      "epoch": 1.8600682593856654,
      "grad_norm": 1.4592860863841783,
      "learning_rate": 8.696660973039432e-07,
      "loss": 0.0262,
      "step": 8175
    },
    {
      "epoch": 1.8602957906712172,
      "grad_norm": 1.118677631149914,
      "learning_rate": 8.695838743673048e-07,
      "loss": 0.022,
      "step": 8176
    },
    {
      "epoch": 1.860523321956769,
      "grad_norm": 1.518417282459403,
      "learning_rate": 8.695016464318199e-07,
      "loss": 0.0385,
      "step": 8177
    },
    {
      "epoch": 1.8607508532423207,
      "grad_norm": 1.5865200516406028,
      "learning_rate": 8.69419413499169e-07,
      "loss": 0.0354,
      "step": 8178
    },
    {
      "epoch": 1.8609783845278725,
      "grad_norm": 1.2344679505066096,
      "learning_rate": 8.693371755710332e-07,
      "loss": 0.0222,
      "step": 8179
    },
    {
      "epoch": 1.8612059158134242,
      "grad_norm": 0.6543044122992299,
      "learning_rate": 8.692549326490929e-07,
      "loss": 0.0103,
      "step": 8180
    },
    {
      "epoch": 1.861433447098976,
      "grad_norm": 1.512390509071281,
      "learning_rate": 8.69172684735029e-07,
      "loss": 0.0322,
      "step": 8181
    },
    {
      "epoch": 1.8616609783845277,
      "grad_norm": 1.22591570641012,
      "learning_rate": 8.690904318305228e-07,
      "loss": 0.021,
      "step": 8182
    },
    {
      "epoch": 1.8618885096700797,
      "grad_norm": 1.9278847876150058,
      "learning_rate": 8.690081739372553e-07,
      "loss": 0.0516,
      "step": 8183
    },
    {
      "epoch": 1.8621160409556314,
      "grad_norm": 1.8449505870002252,
      "learning_rate": 8.689259110569072e-07,
      "loss": 0.0316,
      "step": 8184
    },
    {
      "epoch": 1.8623435722411832,
      "grad_norm": 0.8653952943404032,
      "learning_rate": 8.688436431911604e-07,
      "loss": 0.0129,
      "step": 8185
    },
    {
      "epoch": 1.862571103526735,
      "grad_norm": 1.1405663906685892,
      "learning_rate": 8.687613703416962e-07,
      "loss": 0.0256,
      "step": 8186
    },
    {
      "epoch": 1.8627986348122867,
      "grad_norm": 1.4081877293849094,
      "learning_rate": 8.686790925101959e-07,
      "loss": 0.0237,
      "step": 8187
    },
    {
      "epoch": 1.8630261660978384,
      "grad_norm": 2.5371190563306456,
      "learning_rate": 8.685968096983413e-07,
      "loss": 0.0637,
      "step": 8188
    },
    {
      "epoch": 1.8632536973833902,
      "grad_norm": 1.2527275753497469,
      "learning_rate": 8.685145219078141e-07,
      "loss": 0.028,
      "step": 8189
    },
    {
      "epoch": 1.863481228668942,
      "grad_norm": 1.119060898694386,
      "learning_rate": 8.684322291402959e-07,
      "loss": 0.0265,
      "step": 8190
    },
    {
      "epoch": 1.8637087599544937,
      "grad_norm": 1.000037810567958,
      "learning_rate": 8.683499313974687e-07,
      "loss": 0.0188,
      "step": 8191
    },
    {
      "epoch": 1.8639362912400455,
      "grad_norm": 1.076779356138403,
      "learning_rate": 8.682676286810147e-07,
      "loss": 0.0228,
      "step": 8192
    },
    {
      "epoch": 1.8641638225255974,
      "grad_norm": 0.8435905221735835,
      "learning_rate": 8.68185320992616e-07,
      "loss": 0.0176,
      "step": 8193
    },
    {
      "epoch": 1.8643913538111492,
      "grad_norm": 1.478401595078038,
      "learning_rate": 8.681030083339545e-07,
      "loss": 0.037,
      "step": 8194
    },
    {
      "epoch": 1.864618885096701,
      "grad_norm": 1.9537854826335743,
      "learning_rate": 8.680206907067129e-07,
      "loss": 0.0501,
      "step": 8195
    },
    {
      "epoch": 1.8648464163822527,
      "grad_norm": 0.9494709693436619,
      "learning_rate": 8.679383681125735e-07,
      "loss": 0.0197,
      "step": 8196
    },
    {
      "epoch": 1.8650739476678044,
      "grad_norm": 1.1114327548613774,
      "learning_rate": 8.678560405532186e-07,
      "loss": 0.0213,
      "step": 8197
    },
    {
      "epoch": 1.8653014789533562,
      "grad_norm": 1.1110146727410957,
      "learning_rate": 8.67773708030331e-07,
      "loss": 0.0184,
      "step": 8198
    },
    {
      "epoch": 1.865529010238908,
      "grad_norm": 1.029742737133803,
      "learning_rate": 8.676913705455935e-07,
      "loss": 0.0201,
      "step": 8199
    },
    {
      "epoch": 1.8657565415244597,
      "grad_norm": 1.6329280956348171,
      "learning_rate": 8.676090281006889e-07,
      "loss": 0.0317,
      "step": 8200
    },
    {
      "epoch": 1.8659840728100114,
      "grad_norm": 0.8949711512027054,
      "learning_rate": 8.675266806972999e-07,
      "loss": 0.0304,
      "step": 8201
    },
    {
      "epoch": 1.8662116040955632,
      "grad_norm": 1.256881973616814,
      "learning_rate": 8.674443283371099e-07,
      "loss": 0.0492,
      "step": 8202
    },
    {
      "epoch": 1.866439135381115,
      "grad_norm": 1.3417083032412724,
      "learning_rate": 8.673619710218018e-07,
      "loss": 0.0315,
      "step": 8203
    },
    {
      "epoch": 1.8666666666666667,
      "grad_norm": 1.0536476664103258,
      "learning_rate": 8.672796087530589e-07,
      "loss": 0.015,
      "step": 8204
    },
    {
      "epoch": 1.8668941979522184,
      "grad_norm": 1.2340572224640087,
      "learning_rate": 8.671972415325644e-07,
      "loss": 0.0425,
      "step": 8205
    },
    {
      "epoch": 1.8671217292377702,
      "grad_norm": 1.8631204208091858,
      "learning_rate": 8.671148693620019e-07,
      "loss": 0.0464,
      "step": 8206
    },
    {
      "epoch": 1.867349260523322,
      "grad_norm": 1.0989421249652223,
      "learning_rate": 8.67032492243055e-07,
      "loss": 0.0219,
      "step": 8207
    },
    {
      "epoch": 1.8675767918088737,
      "grad_norm": 1.4023003837125518,
      "learning_rate": 8.66950110177407e-07,
      "loss": 0.0276,
      "step": 8208
    },
    {
      "epoch": 1.8678043230944255,
      "grad_norm": 0.8082699835707026,
      "learning_rate": 8.668677231667422e-07,
      "loss": 0.0093,
      "step": 8209
    },
    {
      "epoch": 1.8680318543799772,
      "grad_norm": 0.7929493429012964,
      "learning_rate": 8.667853312127439e-07,
      "loss": 0.0134,
      "step": 8210
    },
    {
      "epoch": 1.868259385665529,
      "grad_norm": 0.6808371029554392,
      "learning_rate": 8.66702934317096e-07,
      "loss": 0.0117,
      "step": 8211
    },
    {
      "epoch": 1.8684869169510807,
      "grad_norm": 0.9747831617637507,
      "learning_rate": 8.66620532481483e-07,
      "loss": 0.0306,
      "step": 8212
    },
    {
      "epoch": 1.8687144482366325,
      "grad_norm": 1.0522411306849213,
      "learning_rate": 8.66538125707589e-07,
      "loss": 0.0279,
      "step": 8213
    },
    {
      "epoch": 1.8689419795221842,
      "grad_norm": 0.7338998590475417,
      "learning_rate": 8.664557139970978e-07,
      "loss": 0.0243,
      "step": 8214
    },
    {
      "epoch": 1.869169510807736,
      "grad_norm": 0.8452592387133178,
      "learning_rate": 8.663732973516942e-07,
      "loss": 0.0141,
      "step": 8215
    },
    {
      "epoch": 1.8693970420932877,
      "grad_norm": 1.367535148889206,
      "learning_rate": 8.662908757730623e-07,
      "loss": 0.0398,
      "step": 8216
    },
    {
      "epoch": 1.8696245733788395,
      "grad_norm": 1.7047697026339605,
      "learning_rate": 8.66208449262887e-07,
      "loss": 0.038,
      "step": 8217
    },
    {
      "epoch": 1.8698521046643912,
      "grad_norm": 0.785015744452762,
      "learning_rate": 8.661260178228524e-07,
      "loss": 0.0148,
      "step": 8218
    },
    {
      "epoch": 1.870079635949943,
      "grad_norm": 1.7605600844674687,
      "learning_rate": 8.660435814546439e-07,
      "loss": 0.0781,
      "step": 8219
    },
    {
      "epoch": 1.8703071672354947,
      "grad_norm": 0.9382348167426583,
      "learning_rate": 8.65961140159946e-07,
      "loss": 0.0207,
      "step": 8220
    },
    {
      "epoch": 1.8705346985210465,
      "grad_norm": 0.75663826121798,
      "learning_rate": 8.658786939404435e-07,
      "loss": 0.0169,
      "step": 8221
    },
    {
      "epoch": 1.8707622298065985,
      "grad_norm": 0.6536072701513589,
      "learning_rate": 8.657962427978219e-07,
      "loss": 0.0116,
      "step": 8222
    },
    {
      "epoch": 1.8709897610921502,
      "grad_norm": 0.9362614356572078,
      "learning_rate": 8.657137867337659e-07,
      "loss": 0.0138,
      "step": 8223
    },
    {
      "epoch": 1.871217292377702,
      "grad_norm": 1.2693273865002388,
      "learning_rate": 8.65631325749961e-07,
      "loss": 0.0262,
      "step": 8224
    },
    {
      "epoch": 1.8714448236632537,
      "grad_norm": 1.5432025950539427,
      "learning_rate": 8.655488598480925e-07,
      "loss": 0.0336,
      "step": 8225
    },
    {
      "epoch": 1.8716723549488055,
      "grad_norm": 1.0516533581879228,
      "learning_rate": 8.65466389029846e-07,
      "loss": 0.0211,
      "step": 8226
    },
    {
      "epoch": 1.8718998862343572,
      "grad_norm": 0.8833787381932218,
      "learning_rate": 8.65383913296907e-07,
      "loss": 0.0204,
      "step": 8227
    },
    {
      "epoch": 1.872127417519909,
      "grad_norm": 0.7146310943505433,
      "learning_rate": 8.653014326509605e-07,
      "loss": 0.0141,
      "step": 8228
    },
    {
      "epoch": 1.8723549488054607,
      "grad_norm": 0.9123200522206458,
      "learning_rate": 8.652189470936932e-07,
      "loss": 0.0249,
      "step": 8229
    },
    {
      "epoch": 1.8725824800910125,
      "grad_norm": 0.9608224087050663,
      "learning_rate": 8.651364566267906e-07,
      "loss": 0.0164,
      "step": 8230
    },
    {
      "epoch": 1.8728100113765644,
      "grad_norm": 0.855934814485092,
      "learning_rate": 8.650539612519385e-07,
      "loss": 0.012,
      "step": 8231
    },
    {
      "epoch": 1.8730375426621162,
      "grad_norm": 0.7144672916403165,
      "learning_rate": 8.64971460970823e-07,
      "loss": 0.0104,
      "step": 8232
    },
    {
      "epoch": 1.873265073947668,
      "grad_norm": 1.0071325611643098,
      "learning_rate": 8.648889557851306e-07,
      "loss": 0.0271,
      "step": 8233
    },
    {
      "epoch": 1.8734926052332197,
      "grad_norm": 1.2665565567523644,
      "learning_rate": 8.64806445696547e-07,
      "loss": 0.0245,
      "step": 8234
    },
    {
      "epoch": 1.8737201365187715,
      "grad_norm": 2.605935561638153,
      "learning_rate": 8.647239307067588e-07,
      "loss": 0.0448,
      "step": 8235
    },
    {
      "epoch": 1.8739476678043232,
      "grad_norm": 1.2983317147503366,
      "learning_rate": 8.646414108174527e-07,
      "loss": 0.0392,
      "step": 8236
    },
    {
      "epoch": 1.874175199089875,
      "grad_norm": 1.3224033440777954,
      "learning_rate": 8.64558886030315e-07,
      "loss": 0.0361,
      "step": 8237
    },
    {
      "epoch": 1.8744027303754267,
      "grad_norm": 2.2075340386024758,
      "learning_rate": 8.644763563470324e-07,
      "loss": 0.0875,
      "step": 8238
    },
    {
      "epoch": 1.8746302616609785,
      "grad_norm": 0.6781023089584342,
      "learning_rate": 8.643938217692916e-07,
      "loss": 0.0074,
      "step": 8239
    },
    {
      "epoch": 1.8748577929465302,
      "grad_norm": 1.4762759957767195,
      "learning_rate": 8.643112822987795e-07,
      "loss": 0.0346,
      "step": 8240
    },
    {
      "epoch": 1.875085324232082,
      "grad_norm": 1.1333048212875287,
      "learning_rate": 8.642287379371831e-07,
      "loss": 0.0188,
      "step": 8241
    },
    {
      "epoch": 1.8753128555176337,
      "grad_norm": 0.8747310853807329,
      "learning_rate": 8.641461886861893e-07,
      "loss": 0.0182,
      "step": 8242
    },
    {
      "epoch": 1.8755403868031855,
      "grad_norm": 1.5952524050355323,
      "learning_rate": 8.640636345474857e-07,
      "loss": 0.0329,
      "step": 8243
    },
    {
      "epoch": 1.8757679180887372,
      "grad_norm": 1.1585605892511983,
      "learning_rate": 8.639810755227591e-07,
      "loss": 0.0204,
      "step": 8244
    },
    {
      "epoch": 1.875995449374289,
      "grad_norm": 1.1746025644765552,
      "learning_rate": 8.638985116136968e-07,
      "loss": 0.0192,
      "step": 8245
    },
    {
      "epoch": 1.8762229806598407,
      "grad_norm": 0.8194477397027945,
      "learning_rate": 8.638159428219866e-07,
      "loss": 0.0126,
      "step": 8246
    },
    {
      "epoch": 1.8764505119453925,
      "grad_norm": 1.1857441250834007,
      "learning_rate": 8.637333691493159e-07,
      "loss": 0.0202,
      "step": 8247
    },
    {
      "epoch": 1.8766780432309442,
      "grad_norm": 0.9836177902311528,
      "learning_rate": 8.636507905973722e-07,
      "loss": 0.0154,
      "step": 8248
    },
    {
      "epoch": 1.876905574516496,
      "grad_norm": 1.47601792495945,
      "learning_rate": 8.635682071678437e-07,
      "loss": 0.0531,
      "step": 8249
    },
    {
      "epoch": 1.8771331058020477,
      "grad_norm": 2.0178451609426338,
      "learning_rate": 8.634856188624177e-07,
      "loss": 0.0489,
      "step": 8250
    },
    {
      "epoch": 1.8773606370875995,
      "grad_norm": 1.30969876612142,
      "learning_rate": 8.634030256827825e-07,
      "loss": 0.041,
      "step": 8251
    },
    {
      "epoch": 1.8775881683731512,
      "grad_norm": 2.304683562391102,
      "learning_rate": 8.633204276306261e-07,
      "loss": 0.0542,
      "step": 8252
    },
    {
      "epoch": 1.877815699658703,
      "grad_norm": 1.1487287390747398,
      "learning_rate": 8.632378247076366e-07,
      "loss": 0.0194,
      "step": 8253
    },
    {
      "epoch": 1.8780432309442547,
      "grad_norm": 1.0025919212586938,
      "learning_rate": 8.631552169155023e-07,
      "loss": 0.0164,
      "step": 8254
    },
    {
      "epoch": 1.8782707622298065,
      "grad_norm": 0.9283184333065247,
      "learning_rate": 8.630726042559115e-07,
      "loss": 0.022,
      "step": 8255
    },
    {
      "epoch": 1.8784982935153582,
      "grad_norm": 1.4582280154857454,
      "learning_rate": 8.629899867305526e-07,
      "loss": 0.0313,
      "step": 8256
    },
    {
      "epoch": 1.87872582480091,
      "grad_norm": 1.259728727435889,
      "learning_rate": 8.629073643411145e-07,
      "loss": 0.0321,
      "step": 8257
    },
    {
      "epoch": 1.8789533560864617,
      "grad_norm": 0.8530994350432433,
      "learning_rate": 8.628247370892853e-07,
      "loss": 0.0273,
      "step": 8258
    },
    {
      "epoch": 1.8791808873720135,
      "grad_norm": 0.7361257055990824,
      "learning_rate": 8.627421049767541e-07,
      "loss": 0.0112,
      "step": 8259
    },
    {
      "epoch": 1.8794084186575652,
      "grad_norm": 0.9977469415704284,
      "learning_rate": 8.626594680052097e-07,
      "loss": 0.0314,
      "step": 8260
    },
    {
      "epoch": 1.8796359499431172,
      "grad_norm": 0.8056748779789032,
      "learning_rate": 8.62576826176341e-07,
      "loss": 0.0088,
      "step": 8261
    },
    {
      "epoch": 1.879863481228669,
      "grad_norm": 1.0730418628235108,
      "learning_rate": 8.62494179491837e-07,
      "loss": 0.0293,
      "step": 8262
    },
    {
      "epoch": 1.8800910125142207,
      "grad_norm": 1.8855161438814367,
      "learning_rate": 8.624115279533872e-07,
      "loss": 0.0296,
      "step": 8263
    },
    {
      "epoch": 1.8803185437997725,
      "grad_norm": 1.5978910261364054,
      "learning_rate": 8.623288715626804e-07,
      "loss": 0.0228,
      "step": 8264
    },
    {
      "epoch": 1.8805460750853242,
      "grad_norm": 1.0656659599434526,
      "learning_rate": 8.62246210321406e-07,
      "loss": 0.0213,
      "step": 8265
    },
    {
      "epoch": 1.880773606370876,
      "grad_norm": 1.7230213852178717,
      "learning_rate": 8.621635442312537e-07,
      "loss": 0.0355,
      "step": 8266
    },
    {
      "epoch": 1.8810011376564277,
      "grad_norm": 0.9435954471011736,
      "learning_rate": 8.620808732939129e-07,
      "loss": 0.0156,
      "step": 8267
    },
    {
      "epoch": 1.8812286689419795,
      "grad_norm": 0.6950460880605197,
      "learning_rate": 8.619981975110731e-07,
      "loss": 0.0092,
      "step": 8268
    },
    {
      "epoch": 1.8814562002275312,
      "grad_norm": 1.6645334845711723,
      "learning_rate": 8.619155168844243e-07,
      "loss": 0.0618,
      "step": 8269
    },
    {
      "epoch": 1.8816837315130832,
      "grad_norm": 0.8924934641594449,
      "learning_rate": 8.618328314156564e-07,
      "loss": 0.0193,
      "step": 8270
    },
    {
      "epoch": 1.881911262798635,
      "grad_norm": 1.3551138271734262,
      "learning_rate": 8.617501411064588e-07,
      "loss": 0.0454,
      "step": 8271
    },
    {
      "epoch": 1.8821387940841867,
      "grad_norm": 1.4834150753791622,
      "learning_rate": 8.616674459585221e-07,
      "loss": 0.0335,
      "step": 8272
    },
    {
      "epoch": 1.8823663253697385,
      "grad_norm": 1.4119258619306272,
      "learning_rate": 8.615847459735363e-07,
      "loss": 0.0344,
      "step": 8273
    },
    {
      "epoch": 1.8825938566552902,
      "grad_norm": 1.4662232210629031,
      "learning_rate": 8.615020411531915e-07,
      "loss": 0.0446,
      "step": 8274
    },
    {
      "epoch": 1.882821387940842,
      "grad_norm": 1.5611191254491945,
      "learning_rate": 8.61419331499178e-07,
      "loss": 0.0465,
      "step": 8275
    },
    {
      "epoch": 1.8830489192263937,
      "grad_norm": 0.9048921640349314,
      "learning_rate": 8.613366170131867e-07,
      "loss": 0.0198,
      "step": 8276
    },
    {
      "epoch": 1.8832764505119455,
      "grad_norm": 1.035461174281126,
      "learning_rate": 8.612538976969074e-07,
      "loss": 0.0267,
      "step": 8277
    },
    {
      "epoch": 1.8835039817974972,
      "grad_norm": 0.9855906993732617,
      "learning_rate": 8.611711735520312e-07,
      "loss": 0.0218,
      "step": 8278
    },
    {
      "epoch": 1.883731513083049,
      "grad_norm": 1.6015259785217235,
      "learning_rate": 8.610884445802488e-07,
      "loss": 0.0489,
      "step": 8279
    },
    {
      "epoch": 1.8839590443686007,
      "grad_norm": 1.2880775218416267,
      "learning_rate": 8.610057107832509e-07,
      "loss": 0.0415,
      "step": 8280
    },
    {
      "epoch": 1.8841865756541525,
      "grad_norm": 1.2850990560050233,
      "learning_rate": 8.609229721627287e-07,
      "loss": 0.0473,
      "step": 8281
    },
    {
      "epoch": 1.8844141069397042,
      "grad_norm": 1.149003164747692,
      "learning_rate": 8.608402287203728e-07,
      "loss": 0.0281,
      "step": 8282
    },
    {
      "epoch": 1.884641638225256,
      "grad_norm": 1.7925247412693852,
      "learning_rate": 8.607574804578747e-07,
      "loss": 0.0551,
      "step": 8283
    },
    {
      "epoch": 1.8848691695108077,
      "grad_norm": 1.0529263745268131,
      "learning_rate": 8.606747273769253e-07,
      "loss": 0.0273,
      "step": 8284
    },
    {
      "epoch": 1.8850967007963595,
      "grad_norm": 1.2214684896707904,
      "learning_rate": 8.605919694792161e-07,
      "loss": 0.0227,
      "step": 8285
    },
    {
      "epoch": 1.8853242320819112,
      "grad_norm": 0.8930497461591164,
      "learning_rate": 8.605092067664386e-07,
      "loss": 0.0189,
      "step": 8286
    },
    {
      "epoch": 1.885551763367463,
      "grad_norm": 1.3666208480259556,
      "learning_rate": 8.604264392402842e-07,
      "loss": 0.0455,
      "step": 8287
    },
    {
      "epoch": 1.8857792946530147,
      "grad_norm": 1.042161632021271,
      "learning_rate": 8.603436669024446e-07,
      "loss": 0.0205,
      "step": 8288
    },
    {
      "epoch": 1.8860068259385665,
      "grad_norm": 1.162512156537206,
      "learning_rate": 8.602608897546115e-07,
      "loss": 0.0324,
      "step": 8289
    },
    {
      "epoch": 1.8862343572241183,
      "grad_norm": 1.0589861861099525,
      "learning_rate": 8.601781077984767e-07,
      "loss": 0.0256,
      "step": 8290
    },
    {
      "epoch": 1.88646188850967,
      "grad_norm": 1.1272093093873294,
      "learning_rate": 8.600953210357319e-07,
      "loss": 0.0232,
      "step": 8291
    },
    {
      "epoch": 1.8866894197952218,
      "grad_norm": 1.7153443676242524,
      "learning_rate": 8.600125294680692e-07,
      "loss": 0.0518,
      "step": 8292
    },
    {
      "epoch": 1.8869169510807735,
      "grad_norm": 0.9046095063866724,
      "learning_rate": 8.59929733097181e-07,
      "loss": 0.0267,
      "step": 8293
    },
    {
      "epoch": 1.8871444823663253,
      "grad_norm": 0.9024113986945366,
      "learning_rate": 8.598469319247593e-07,
      "loss": 0.0157,
      "step": 8294
    },
    {
      "epoch": 1.887372013651877,
      "grad_norm": 1.4704486910706056,
      "learning_rate": 8.597641259524965e-07,
      "loss": 0.0285,
      "step": 8295
    },
    {
      "epoch": 1.8875995449374288,
      "grad_norm": 0.9238625713309345,
      "learning_rate": 8.596813151820849e-07,
      "loss": 0.0306,
      "step": 8296
    },
    {
      "epoch": 1.8878270762229805,
      "grad_norm": 0.6184180904992845,
      "learning_rate": 8.595984996152168e-07,
      "loss": 0.011,
      "step": 8297
    },
    {
      "epoch": 1.8880546075085323,
      "grad_norm": 1.4313296623449745,
      "learning_rate": 8.595156792535852e-07,
      "loss": 0.0453,
      "step": 8298
    },
    {
      "epoch": 1.8882821387940842,
      "grad_norm": 1.9252715310610398,
      "learning_rate": 8.594328540988825e-07,
      "loss": 0.0331,
      "step": 8299
    },
    {
      "epoch": 1.888509670079636,
      "grad_norm": 0.5403949165206943,
      "learning_rate": 8.593500241528016e-07,
      "loss": 0.0075,
      "step": 8300
    },
    {
      "epoch": 1.8887372013651877,
      "grad_norm": 1.0057890593594268,
      "learning_rate": 8.592671894170356e-07,
      "loss": 0.0282,
      "step": 8301
    },
    {
      "epoch": 1.8889647326507395,
      "grad_norm": 0.9380707896915073,
      "learning_rate": 8.59184349893277e-07,
      "loss": 0.0249,
      "step": 8302
    },
    {
      "epoch": 1.8891922639362912,
      "grad_norm": 1.1957453930209534,
      "learning_rate": 8.591015055832195e-07,
      "loss": 0.0336,
      "step": 8303
    },
    {
      "epoch": 1.889419795221843,
      "grad_norm": 1.6201566102692035,
      "learning_rate": 8.590186564885557e-07,
      "loss": 0.0342,
      "step": 8304
    },
    {
      "epoch": 1.8896473265073948,
      "grad_norm": 1.242275772305289,
      "learning_rate": 8.589358026109792e-07,
      "loss": 0.04,
      "step": 8305
    },
    {
      "epoch": 1.8898748577929465,
      "grad_norm": 0.9816574572206791,
      "learning_rate": 8.588529439521834e-07,
      "loss": 0.0182,
      "step": 8306
    },
    {
      "epoch": 1.8901023890784983,
      "grad_norm": 0.5762361178743535,
      "learning_rate": 8.587700805138617e-07,
      "loss": 0.0132,
      "step": 8307
    },
    {
      "epoch": 1.89032992036405,
      "grad_norm": 2.230891945909432,
      "learning_rate": 8.586872122977076e-07,
      "loss": 0.0649,
      "step": 8308
    },
    {
      "epoch": 1.890557451649602,
      "grad_norm": 0.8923642900973059,
      "learning_rate": 8.58604339305415e-07,
      "loss": 0.0221,
      "step": 8309
    },
    {
      "epoch": 1.8907849829351537,
      "grad_norm": 1.413899468730935,
      "learning_rate": 8.585214615386773e-07,
      "loss": 0.0216,
      "step": 8310
    },
    {
      "epoch": 1.8910125142207055,
      "grad_norm": 0.8754816462406323,
      "learning_rate": 8.584385789991887e-07,
      "loss": 0.0258,
      "step": 8311
    },
    {
      "epoch": 1.8912400455062572,
      "grad_norm": 0.924318820572518,
      "learning_rate": 8.583556916886432e-07,
      "loss": 0.0162,
      "step": 8312
    },
    {
      "epoch": 1.891467576791809,
      "grad_norm": 0.7762207786564913,
      "learning_rate": 8.582727996087345e-07,
      "loss": 0.0156,
      "step": 8313
    },
    {
      "epoch": 1.8916951080773607,
      "grad_norm": 1.7197432751965054,
      "learning_rate": 8.581899027611574e-07,
      "loss": 0.0432,
      "step": 8314
    },
    {
      "epoch": 1.8919226393629125,
      "grad_norm": 0.6713325945774121,
      "learning_rate": 8.581070011476053e-07,
      "loss": 0.0079,
      "step": 8315
    },
    {
      "epoch": 1.8921501706484642,
      "grad_norm": 0.9139036984168137,
      "learning_rate": 8.580240947697732e-07,
      "loss": 0.0216,
      "step": 8316
    },
    {
      "epoch": 1.892377701934016,
      "grad_norm": 1.0287826908579287,
      "learning_rate": 8.579411836293555e-07,
      "loss": 0.0184,
      "step": 8317
    },
    {
      "epoch": 1.8926052332195678,
      "grad_norm": 1.343241857750621,
      "learning_rate": 8.578582677280464e-07,
      "loss": 0.0596,
      "step": 8318
    },
    {
      "epoch": 1.8928327645051195,
      "grad_norm": 1.0633519122682107,
      "learning_rate": 8.577753470675408e-07,
      "loss": 0.0245,
      "step": 8319
    },
    {
      "epoch": 1.8930602957906713,
      "grad_norm": 0.7693491499865697,
      "learning_rate": 8.576924216495336e-07,
      "loss": 0.0157,
      "step": 8320
    },
    {
      "epoch": 1.893287827076223,
      "grad_norm": 2.098129898190892,
      "learning_rate": 8.576094914757194e-07,
      "loss": 0.0531,
      "step": 8321
    },
    {
      "epoch": 1.8935153583617748,
      "grad_norm": 1.029612822990029,
      "learning_rate": 8.575265565477931e-07,
      "loss": 0.0225,
      "step": 8322
    },
    {
      "epoch": 1.8937428896473265,
      "grad_norm": 1.5486723274495435,
      "learning_rate": 8.574436168674498e-07,
      "loss": 0.0426,
      "step": 8323
    },
    {
      "epoch": 1.8939704209328783,
      "grad_norm": 2.937968838728311,
      "learning_rate": 8.573606724363848e-07,
      "loss": 0.0223,
      "step": 8324
    },
    {
      "epoch": 1.89419795221843,
      "grad_norm": 0.9186281289893576,
      "learning_rate": 8.572777232562929e-07,
      "loss": 0.0183,
      "step": 8325
    },
    {
      "epoch": 1.8944254835039818,
      "grad_norm": 0.7647862792866064,
      "learning_rate": 8.571947693288702e-07,
      "loss": 0.0159,
      "step": 8326
    },
    {
      "epoch": 1.8946530147895335,
      "grad_norm": 1.191823185384319,
      "learning_rate": 8.571118106558114e-07,
      "loss": 0.023,
      "step": 8327
    },
    {
      "epoch": 1.8948805460750853,
      "grad_norm": 1.3699185120387163,
      "learning_rate": 8.570288472388122e-07,
      "loss": 0.0332,
      "step": 8328
    },
    {
      "epoch": 1.895108077360637,
      "grad_norm": 1.5461111445030973,
      "learning_rate": 8.569458790795685e-07,
      "loss": 0.0352,
      "step": 8329
    },
    {
      "epoch": 1.8953356086461888,
      "grad_norm": 0.8535154134838575,
      "learning_rate": 8.568629061797757e-07,
      "loss": 0.011,
      "step": 8330
    },
    {
      "epoch": 1.8955631399317405,
      "grad_norm": 1.570117171787192,
      "learning_rate": 8.567799285411298e-07,
      "loss": 0.0326,
      "step": 8331
    },
    {
      "epoch": 1.8957906712172923,
      "grad_norm": 1.791000174693399,
      "learning_rate": 8.566969461653266e-07,
      "loss": 0.0449,
      "step": 8332
    },
    {
      "epoch": 1.896018202502844,
      "grad_norm": 1.295995913807357,
      "learning_rate": 8.566139590540622e-07,
      "loss": 0.0275,
      "step": 8333
    },
    {
      "epoch": 1.8962457337883958,
      "grad_norm": 1.394137990033564,
      "learning_rate": 8.565309672090328e-07,
      "loss": 0.0418,
      "step": 8334
    },
    {
      "epoch": 1.8964732650739475,
      "grad_norm": 1.3620290597281826,
      "learning_rate": 8.564479706319339e-07,
      "loss": 0.0296,
      "step": 8335
    },
    {
      "epoch": 1.8967007963594993,
      "grad_norm": 1.1143468857481267,
      "learning_rate": 8.563649693244629e-07,
      "loss": 0.0283,
      "step": 8336
    },
    {
      "epoch": 1.896928327645051,
      "grad_norm": 0.8195233179892975,
      "learning_rate": 8.562819632883155e-07,
      "loss": 0.0127,
      "step": 8337
    },
    {
      "epoch": 1.897155858930603,
      "grad_norm": 1.101615888325663,
      "learning_rate": 8.561989525251883e-07,
      "loss": 0.0426,
      "step": 8338
    },
    {
      "epoch": 1.8973833902161548,
      "grad_norm": 1.0049505213568453,
      "learning_rate": 8.56115937036778e-07,
      "loss": 0.0121,
      "step": 8339
    },
    {
      "epoch": 1.8976109215017065,
      "grad_norm": 20.746676374419295,
      "learning_rate": 8.560329168247812e-07,
      "loss": 0.239,
      "step": 8340
    },
    {
      "epoch": 1.8978384527872583,
      "grad_norm": 0.931717318764593,
      "learning_rate": 8.559498918908948e-07,
      "loss": 0.0375,
      "step": 8341
    },
    {
      "epoch": 1.89806598407281,
      "grad_norm": 1.3285902856176695,
      "learning_rate": 8.558668622368154e-07,
      "loss": 0.0279,
      "step": 8342
    },
    {
      "epoch": 1.8982935153583618,
      "grad_norm": 1.3731987814764735,
      "learning_rate": 8.557838278642401e-07,
      "loss": 0.0451,
      "step": 8343
    },
    {
      "epoch": 1.8985210466439135,
      "grad_norm": 0.9111979938572307,
      "learning_rate": 8.557007887748661e-07,
      "loss": 0.0135,
      "step": 8344
    },
    {
      "epoch": 1.8987485779294653,
      "grad_norm": 1.546482352872472,
      "learning_rate": 8.556177449703906e-07,
      "loss": 0.0458,
      "step": 8345
    },
    {
      "epoch": 1.898976109215017,
      "grad_norm": 1.1451460927699535,
      "learning_rate": 8.555346964525107e-07,
      "loss": 0.0227,
      "step": 8346
    },
    {
      "epoch": 1.8992036405005688,
      "grad_norm": 1.2348650674661643,
      "learning_rate": 8.554516432229238e-07,
      "loss": 0.0198,
      "step": 8347
    },
    {
      "epoch": 1.8994311717861208,
      "grad_norm": 0.8684328346312213,
      "learning_rate": 8.553685852833274e-07,
      "loss": 0.0136,
      "step": 8348
    },
    {
      "epoch": 1.8996587030716725,
      "grad_norm": 0.8459249699601199,
      "learning_rate": 8.552855226354187e-07,
      "loss": 0.0183,
      "step": 8349
    },
    {
      "epoch": 1.8998862343572243,
      "grad_norm": 0.8398092931808809,
      "learning_rate": 8.55202455280896e-07,
      "loss": 0.0169,
      "step": 8350
    },
    {
      "epoch": 1.900113765642776,
      "grad_norm": 0.6669246846137414,
      "learning_rate": 8.551193832214567e-07,
      "loss": 0.013,
      "step": 8351
    },
    {
      "epoch": 1.9003412969283278,
      "grad_norm": 1.427927029573224,
      "learning_rate": 8.550363064587985e-07,
      "loss": 0.0241,
      "step": 8352
    },
    {
      "epoch": 1.9005688282138795,
      "grad_norm": 0.8047507484768054,
      "learning_rate": 8.549532249946197e-07,
      "loss": 0.0162,
      "step": 8353
    },
    {
      "epoch": 1.9007963594994313,
      "grad_norm": 0.9422455546065365,
      "learning_rate": 8.548701388306179e-07,
      "loss": 0.0141,
      "step": 8354
    },
    {
      "epoch": 1.901023890784983,
      "grad_norm": 1.3733449199975643,
      "learning_rate": 8.547870479684916e-07,
      "loss": 0.0399,
      "step": 8355
    },
    {
      "epoch": 1.9012514220705348,
      "grad_norm": 0.7185318034874649,
      "learning_rate": 8.547039524099387e-07,
      "loss": 0.009,
      "step": 8356
    },
    {
      "epoch": 1.9014789533560865,
      "grad_norm": 0.7062440700182584,
      "learning_rate": 8.546208521566578e-07,
      "loss": 0.0199,
      "step": 8357
    },
    {
      "epoch": 1.9017064846416383,
      "grad_norm": 1.2742509728964952,
      "learning_rate": 8.545377472103474e-07,
      "loss": 0.0388,
      "step": 8358
    },
    {
      "epoch": 1.90193401592719,
      "grad_norm": 1.411456477780153,
      "learning_rate": 8.544546375727055e-07,
      "loss": 0.0144,
      "step": 8359
    },
    {
      "epoch": 1.9021615472127418,
      "grad_norm": 0.9938212599624071,
      "learning_rate": 8.543715232454311e-07,
      "loss": 0.0225,
      "step": 8360
    },
    {
      "epoch": 1.9023890784982935,
      "grad_norm": 1.5657277168399435,
      "learning_rate": 8.54288404230223e-07,
      "loss": 0.0528,
      "step": 8361
    },
    {
      "epoch": 1.9026166097838453,
      "grad_norm": 0.9284050397615987,
      "learning_rate": 8.542052805287797e-07,
      "loss": 0.0246,
      "step": 8362
    },
    {
      "epoch": 1.902844141069397,
      "grad_norm": 1.3974079836594142,
      "learning_rate": 8.541221521428003e-07,
      "loss": 0.0451,
      "step": 8363
    },
    {
      "epoch": 1.9030716723549488,
      "grad_norm": 1.04036961337048,
      "learning_rate": 8.540390190739839e-07,
      "loss": 0.0232,
      "step": 8364
    },
    {
      "epoch": 1.9032992036405005,
      "grad_norm": 1.7484807221488305,
      "learning_rate": 8.539558813240291e-07,
      "loss": 0.0293,
      "step": 8365
    },
    {
      "epoch": 1.9035267349260523,
      "grad_norm": 1.1185754396379677,
      "learning_rate": 8.538727388946356e-07,
      "loss": 0.0267,
      "step": 8366
    },
    {
      "epoch": 1.903754266211604,
      "grad_norm": 0.9078252207266309,
      "learning_rate": 8.537895917875023e-07,
      "loss": 0.0324,
      "step": 8367
    },
    {
      "epoch": 1.9039817974971558,
      "grad_norm": 0.9124967769994551,
      "learning_rate": 8.537064400043289e-07,
      "loss": 0.0186,
      "step": 8368
    },
    {
      "epoch": 1.9042093287827075,
      "grad_norm": 1.8217424208968858,
      "learning_rate": 8.536232835468145e-07,
      "loss": 0.0327,
      "step": 8369
    },
    {
      "epoch": 1.9044368600682593,
      "grad_norm": 1.061454118710923,
      "learning_rate": 8.535401224166593e-07,
      "loss": 0.0257,
      "step": 8370
    },
    {
      "epoch": 1.904664391353811,
      "grad_norm": 1.2108474848055848,
      "learning_rate": 8.534569566155623e-07,
      "loss": 0.0346,
      "step": 8371
    },
    {
      "epoch": 1.9048919226393628,
      "grad_norm": 0.9981046532618882,
      "learning_rate": 8.533737861452235e-07,
      "loss": 0.0367,
      "step": 8372
    },
    {
      "epoch": 1.9051194539249146,
      "grad_norm": 0.7137971971995981,
      "learning_rate": 8.532906110073427e-07,
      "loss": 0.022,
      "step": 8373
    },
    {
      "epoch": 1.9053469852104663,
      "grad_norm": 1.1506401527737304,
      "learning_rate": 8.5320743120362e-07,
      "loss": 0.034,
      "step": 8374
    },
    {
      "epoch": 1.905574516496018,
      "grad_norm": 1.0060448506402915,
      "learning_rate": 8.531242467357555e-07,
      "loss": 0.0176,
      "step": 8375
    },
    {
      "epoch": 1.9058020477815698,
      "grad_norm": 1.2596476896044595,
      "learning_rate": 8.530410576054489e-07,
      "loss": 0.0296,
      "step": 8376
    },
    {
      "epoch": 1.9060295790671218,
      "grad_norm": 1.270048014695048,
      "learning_rate": 8.52957863814401e-07,
      "loss": 0.0225,
      "step": 8377
    },
    {
      "epoch": 1.9062571103526735,
      "grad_norm": 0.9739005233427236,
      "learning_rate": 8.528746653643116e-07,
      "loss": 0.0219,
      "step": 8378
    },
    {
      "epoch": 1.9064846416382253,
      "grad_norm": 1.4836478796362045,
      "learning_rate": 8.527914622568814e-07,
      "loss": 0.0326,
      "step": 8379
    },
    {
      "epoch": 1.906712172923777,
      "grad_norm": 1.0636112939942484,
      "learning_rate": 8.527082544938111e-07,
      "loss": 0.0186,
      "step": 8380
    },
    {
      "epoch": 1.9069397042093288,
      "grad_norm": 0.6927545848426347,
      "learning_rate": 8.526250420768009e-07,
      "loss": 0.0138,
      "step": 8381
    },
    {
      "epoch": 1.9071672354948805,
      "grad_norm": 1.3875179087787834,
      "learning_rate": 8.525418250075518e-07,
      "loss": 0.0568,
      "step": 8382
    },
    {
      "epoch": 1.9073947667804323,
      "grad_norm": 0.9015365651542543,
      "learning_rate": 8.524586032877645e-07,
      "loss": 0.022,
      "step": 8383
    },
    {
      "epoch": 1.907622298065984,
      "grad_norm": 1.343545255745248,
      "learning_rate": 8.523753769191399e-07,
      "loss": 0.0266,
      "step": 8384
    },
    {
      "epoch": 1.9078498293515358,
      "grad_norm": 1.0970681264556976,
      "learning_rate": 8.522921459033791e-07,
      "loss": 0.0219,
      "step": 8385
    },
    {
      "epoch": 1.9080773606370875,
      "grad_norm": 0.8250759323381144,
      "learning_rate": 8.522089102421829e-07,
      "loss": 0.0177,
      "step": 8386
    },
    {
      "epoch": 1.9083048919226395,
      "grad_norm": 0.9830515935915406,
      "learning_rate": 8.52125669937253e-07,
      "loss": 0.0218,
      "step": 8387
    },
    {
      "epoch": 1.9085324232081913,
      "grad_norm": 1.4564551244698456,
      "learning_rate": 8.5204242499029e-07,
      "loss": 0.044,
      "step": 8388
    },
    {
      "epoch": 1.908759954493743,
      "grad_norm": 1.1380219813778063,
      "learning_rate": 8.51959175402996e-07,
      "loss": 0.0259,
      "step": 8389
    },
    {
      "epoch": 1.9089874857792948,
      "grad_norm": 0.8340581743448356,
      "learning_rate": 8.518759211770719e-07,
      "loss": 0.015,
      "step": 8390
    },
    {
      "epoch": 1.9092150170648465,
      "grad_norm": 1.143444324624706,
      "learning_rate": 8.517926623142196e-07,
      "loss": 0.0266,
      "step": 8391
    },
    {
      "epoch": 1.9094425483503983,
      "grad_norm": 1.4872909115352921,
      "learning_rate": 8.517093988161404e-07,
      "loss": 0.055,
      "step": 8392
    },
    {
      "epoch": 1.90967007963595,
      "grad_norm": 1.1451443339474134,
      "learning_rate": 8.516261306845365e-07,
      "loss": 0.0349,
      "step": 8393
    },
    {
      "epoch": 1.9098976109215018,
      "grad_norm": 0.964597490955224,
      "learning_rate": 8.515428579211095e-07,
      "loss": 0.0146,
      "step": 8394
    },
    {
      "epoch": 1.9101251422070535,
      "grad_norm": 1.0969553831767032,
      "learning_rate": 8.514595805275614e-07,
      "loss": 0.0241,
      "step": 8395
    },
    {
      "epoch": 1.9103526734926053,
      "grad_norm": 1.4080510598222953,
      "learning_rate": 8.513762985055942e-07,
      "loss": 0.0567,
      "step": 8396
    },
    {
      "epoch": 1.910580204778157,
      "grad_norm": 1.2786633879760345,
      "learning_rate": 8.5129301185691e-07,
      "loss": 0.0382,
      "step": 8397
    },
    {
      "epoch": 1.9108077360637088,
      "grad_norm": 0.6123509234890121,
      "learning_rate": 8.512097205832111e-07,
      "loss": 0.008,
      "step": 8398
    },
    {
      "epoch": 1.9110352673492605,
      "grad_norm": 0.6834596034514431,
      "learning_rate": 8.511264246861997e-07,
      "loss": 0.0158,
      "step": 8399
    },
    {
      "epoch": 1.9112627986348123,
      "grad_norm": 1.4469107691813436,
      "learning_rate": 8.510431241675784e-07,
      "loss": 0.0207,
      "step": 8400
    },
    {
      "epoch": 1.911490329920364,
      "grad_norm": 1.1241815055827342,
      "learning_rate": 8.509598190290497e-07,
      "loss": 0.0192,
      "step": 8401
    },
    {
      "epoch": 1.9117178612059158,
      "grad_norm": 0.8640122699671915,
      "learning_rate": 8.508765092723159e-07,
      "loss": 0.0252,
      "step": 8402
    },
    {
      "epoch": 1.9119453924914676,
      "grad_norm": 1.1837107369030004,
      "learning_rate": 8.507931948990801e-07,
      "loss": 0.0538,
      "step": 8403
    },
    {
      "epoch": 1.9121729237770193,
      "grad_norm": 1.1995895924958146,
      "learning_rate": 8.507098759110449e-07,
      "loss": 0.0453,
      "step": 8404
    },
    {
      "epoch": 1.912400455062571,
      "grad_norm": 1.619876859968541,
      "learning_rate": 8.506265523099133e-07,
      "loss": 0.0339,
      "step": 8405
    },
    {
      "epoch": 1.9126279863481228,
      "grad_norm": 0.9981047989443556,
      "learning_rate": 8.50543224097388e-07,
      "loss": 0.0231,
      "step": 8406
    },
    {
      "epoch": 1.9128555176336746,
      "grad_norm": 0.6500442811423481,
      "learning_rate": 8.504598912751722e-07,
      "loss": 0.0135,
      "step": 8407
    },
    {
      "epoch": 1.9130830489192263,
      "grad_norm": 1.7930157168186414,
      "learning_rate": 8.503765538449695e-07,
      "loss": 0.0746,
      "step": 8408
    },
    {
      "epoch": 1.913310580204778,
      "grad_norm": 1.1905903879309094,
      "learning_rate": 8.502932118084825e-07,
      "loss": 0.0274,
      "step": 8409
    },
    {
      "epoch": 1.9135381114903298,
      "grad_norm": 1.3730986996255226,
      "learning_rate": 8.502098651674148e-07,
      "loss": 0.0512,
      "step": 8410
    },
    {
      "epoch": 1.9137656427758816,
      "grad_norm": 0.7930623765153966,
      "learning_rate": 8.501265139234702e-07,
      "loss": 0.0216,
      "step": 8411
    },
    {
      "epoch": 1.9139931740614333,
      "grad_norm": 1.484429796680887,
      "learning_rate": 8.500431580783518e-07,
      "loss": 0.0213,
      "step": 8412
    },
    {
      "epoch": 1.914220705346985,
      "grad_norm": 3.26329303562541,
      "learning_rate": 8.499597976337632e-07,
      "loss": 0.0298,
      "step": 8413
    },
    {
      "epoch": 1.9144482366325368,
      "grad_norm": 0.6275582463172372,
      "learning_rate": 8.498764325914087e-07,
      "loss": 0.0092,
      "step": 8414
    },
    {
      "epoch": 1.9146757679180886,
      "grad_norm": 0.8514325784501834,
      "learning_rate": 8.497930629529916e-07,
      "loss": 0.0135,
      "step": 8415
    },
    {
      "epoch": 1.9149032992036406,
      "grad_norm": 0.7539415354286236,
      "learning_rate": 8.497096887202158e-07,
      "loss": 0.0186,
      "step": 8416
    },
    {
      "epoch": 1.9151308304891923,
      "grad_norm": 0.8168825733483965,
      "learning_rate": 8.496263098947857e-07,
      "loss": 0.0171,
      "step": 8417
    },
    {
      "epoch": 1.915358361774744,
      "grad_norm": 1.2697149138942274,
      "learning_rate": 8.495429264784051e-07,
      "loss": 0.0336,
      "step": 8418
    },
    {
      "epoch": 1.9155858930602958,
      "grad_norm": 1.3946643910966317,
      "learning_rate": 8.494595384727783e-07,
      "loss": 0.0224,
      "step": 8419
    },
    {
      "epoch": 1.9158134243458476,
      "grad_norm": 1.4038850936602918,
      "learning_rate": 8.493761458796098e-07,
      "loss": 0.057,
      "step": 8420
    },
    {
      "epoch": 1.9160409556313993,
      "grad_norm": 0.6736182648948457,
      "learning_rate": 8.492927487006037e-07,
      "loss": 0.0095,
      "step": 8421
    },
    {
      "epoch": 1.916268486916951,
      "grad_norm": 1.564923396961411,
      "learning_rate": 8.492093469374646e-07,
      "loss": 0.035,
      "step": 8422
    },
    {
      "epoch": 1.9164960182025028,
      "grad_norm": 1.1021686395110446,
      "learning_rate": 8.491259405918969e-07,
      "loss": 0.0267,
      "step": 8423
    },
    {
      "epoch": 1.9167235494880546,
      "grad_norm": 1.4012443000406984,
      "learning_rate": 8.490425296656057e-07,
      "loss": 0.0345,
      "step": 8424
    },
    {
      "epoch": 1.9169510807736063,
      "grad_norm": 0.7791471615047615,
      "learning_rate": 8.489591141602954e-07,
      "loss": 0.0167,
      "step": 8425
    },
    {
      "epoch": 1.9171786120591583,
      "grad_norm": 1.2116548450079154,
      "learning_rate": 8.488756940776709e-07,
      "loss": 0.0323,
      "step": 8426
    },
    {
      "epoch": 1.91740614334471,
      "grad_norm": 2.33324083375237,
      "learning_rate": 8.487922694194374e-07,
      "loss": 0.0579,
      "step": 8427
    },
    {
      "epoch": 1.9176336746302618,
      "grad_norm": 1.6160700922599316,
      "learning_rate": 8.487088401872997e-07,
      "loss": 0.0319,
      "step": 8428
    },
    {
      "epoch": 1.9178612059158135,
      "grad_norm": 0.797732964369814,
      "learning_rate": 8.486254063829628e-07,
      "loss": 0.013,
      "step": 8429
    },
    {
      "epoch": 1.9180887372013653,
      "grad_norm": 0.731455585098273,
      "learning_rate": 8.485419680081324e-07,
      "loss": 0.013,
      "step": 8430
    },
    {
      "epoch": 1.918316268486917,
      "grad_norm": 1.3985142165078785,
      "learning_rate": 8.484585250645134e-07,
      "loss": 0.0312,
      "step": 8431
    },
    {
      "epoch": 1.9185437997724688,
      "grad_norm": 1.0826264757291735,
      "learning_rate": 8.483750775538116e-07,
      "loss": 0.0258,
      "step": 8432
    },
    {
      "epoch": 1.9187713310580206,
      "grad_norm": 0.9737168446676101,
      "learning_rate": 8.482916254777321e-07,
      "loss": 0.0165,
      "step": 8433
    },
    {
      "epoch": 1.9189988623435723,
      "grad_norm": 1.1198156854996961,
      "learning_rate": 8.482081688379809e-07,
      "loss": 0.0185,
      "step": 8434
    },
    {
      "epoch": 1.919226393629124,
      "grad_norm": 1.2595137307081843,
      "learning_rate": 8.481247076362633e-07,
      "loss": 0.0369,
      "step": 8435
    },
    {
      "epoch": 1.9194539249146758,
      "grad_norm": 0.6339129462269499,
      "learning_rate": 8.480412418742855e-07,
      "loss": 0.0078,
      "step": 8436
    },
    {
      "epoch": 1.9196814562002276,
      "grad_norm": 0.8520737497185187,
      "learning_rate": 8.479577715537531e-07,
      "loss": 0.0236,
      "step": 8437
    },
    {
      "epoch": 1.9199089874857793,
      "grad_norm": 1.445989975288279,
      "learning_rate": 8.478742966763721e-07,
      "loss": 0.0327,
      "step": 8438
    },
    {
      "epoch": 1.920136518771331,
      "grad_norm": 1.2746227485283579,
      "learning_rate": 8.477908172438488e-07,
      "loss": 0.0469,
      "step": 8439
    },
    {
      "epoch": 1.9203640500568828,
      "grad_norm": 0.89691294519206,
      "learning_rate": 8.477073332578892e-07,
      "loss": 0.0097,
      "step": 8440
    },
    {
      "epoch": 1.9205915813424346,
      "grad_norm": 0.962306699621327,
      "learning_rate": 8.476238447201995e-07,
      "loss": 0.0291,
      "step": 8441
    },
    {
      "epoch": 1.9208191126279863,
      "grad_norm": 1.1132372477063388,
      "learning_rate": 8.475403516324863e-07,
      "loss": 0.0202,
      "step": 8442
    },
    {
      "epoch": 1.921046643913538,
      "grad_norm": 1.6547856713968876,
      "learning_rate": 8.474568539964556e-07,
      "loss": 0.056,
      "step": 8443
    },
    {
      "epoch": 1.9212741751990898,
      "grad_norm": 0.4787214241555102,
      "learning_rate": 8.473733518138145e-07,
      "loss": 0.01,
      "step": 8444
    },
    {
      "epoch": 1.9215017064846416,
      "grad_norm": 1.1801121879348935,
      "learning_rate": 8.472898450862691e-07,
      "loss": 0.028,
      "step": 8445
    },
    {
      "epoch": 1.9217292377701933,
      "grad_norm": 1.0120828980304566,
      "learning_rate": 8.472063338155265e-07,
      "loss": 0.0325,
      "step": 8446
    },
    {
      "epoch": 1.921956769055745,
      "grad_norm": 2.2691336336786536,
      "learning_rate": 8.471228180032934e-07,
      "loss": 0.0511,
      "step": 8447
    },
    {
      "epoch": 1.9221843003412968,
      "grad_norm": 0.679913941000245,
      "learning_rate": 8.470392976512768e-07,
      "loss": 0.0098,
      "step": 8448
    },
    {
      "epoch": 1.9224118316268486,
      "grad_norm": 1.2697734767948874,
      "learning_rate": 8.469557727611833e-07,
      "loss": 0.0435,
      "step": 8449
    },
    {
      "epoch": 1.9226393629124003,
      "grad_norm": 1.2670054762383771,
      "learning_rate": 8.468722433347204e-07,
      "loss": 0.0361,
      "step": 8450
    },
    {
      "epoch": 1.922866894197952,
      "grad_norm": 0.9079601191824546,
      "learning_rate": 8.467887093735953e-07,
      "loss": 0.0138,
      "step": 8451
    },
    {
      "epoch": 1.9230944254835038,
      "grad_norm": 0.9892645162047258,
      "learning_rate": 8.467051708795152e-07,
      "loss": 0.0194,
      "step": 8452
    },
    {
      "epoch": 1.9233219567690556,
      "grad_norm": 4.904769472158991,
      "learning_rate": 8.466216278541874e-07,
      "loss": 0.118,
      "step": 8453
    },
    {
      "epoch": 1.9235494880546073,
      "grad_norm": 1.0889415277846448,
      "learning_rate": 8.465380802993193e-07,
      "loss": 0.0205,
      "step": 8454
    },
    {
      "epoch": 1.9237770193401593,
      "grad_norm": 0.9092866101119246,
      "learning_rate": 8.464545282166187e-07,
      "loss": 0.0177,
      "step": 8455
    },
    {
      "epoch": 1.924004550625711,
      "grad_norm": 1.0710600040644043,
      "learning_rate": 8.463709716077929e-07,
      "loss": 0.0473,
      "step": 8456
    },
    {
      "epoch": 1.9242320819112628,
      "grad_norm": 0.7766173559355853,
      "learning_rate": 8.4628741047455e-07,
      "loss": 0.0141,
      "step": 8457
    },
    {
      "epoch": 1.9244596131968146,
      "grad_norm": 1.5075758759501967,
      "learning_rate": 8.462038448185977e-07,
      "loss": 0.0425,
      "step": 8458
    },
    {
      "epoch": 1.9246871444823663,
      "grad_norm": 1.0521484097583373,
      "learning_rate": 8.461202746416442e-07,
      "loss": 0.034,
      "step": 8459
    },
    {
      "epoch": 1.924914675767918,
      "grad_norm": 0.9737045662390044,
      "learning_rate": 8.460366999453968e-07,
      "loss": 0.0349,
      "step": 8460
    },
    {
      "epoch": 1.9251422070534698,
      "grad_norm": 1.3360371476012138,
      "learning_rate": 8.459531207315644e-07,
      "loss": 0.0309,
      "step": 8461
    },
    {
      "epoch": 1.9253697383390216,
      "grad_norm": 1.1388283401231705,
      "learning_rate": 8.458695370018546e-07,
      "loss": 0.0249,
      "step": 8462
    },
    {
      "epoch": 1.9255972696245733,
      "grad_norm": 1.543023823562457,
      "learning_rate": 8.457859487579762e-07,
      "loss": 0.0419,
      "step": 8463
    },
    {
      "epoch": 1.925824800910125,
      "grad_norm": 0.7045927127470623,
      "learning_rate": 8.457023560016371e-07,
      "loss": 0.019,
      "step": 8464
    },
    {
      "epoch": 1.926052332195677,
      "grad_norm": 1.5680014180333777,
      "learning_rate": 8.456187587345463e-07,
      "loss": 0.0377,
      "step": 8465
    },
    {
      "epoch": 1.9262798634812288,
      "grad_norm": 0.6049760071323378,
      "learning_rate": 8.455351569584119e-07,
      "loss": 0.0123,
      "step": 8466
    },
    {
      "epoch": 1.9265073947667806,
      "grad_norm": 0.6949704083990405,
      "learning_rate": 8.454515506749431e-07,
      "loss": 0.0127,
      "step": 8467
    },
    {
      "epoch": 1.9267349260523323,
      "grad_norm": 1.2166952289517832,
      "learning_rate": 8.453679398858481e-07,
      "loss": 0.0337,
      "step": 8468
    },
    {
      "epoch": 1.926962457337884,
      "grad_norm": 0.5643035292729891,
      "learning_rate": 8.452843245928359e-07,
      "loss": 0.0083,
      "step": 8469
    },
    {
      "epoch": 1.9271899886234358,
      "grad_norm": 1.268210689591597,
      "learning_rate": 8.452007047976155e-07,
      "loss": 0.0393,
      "step": 8470
    },
    {
      "epoch": 1.9274175199089876,
      "grad_norm": 1.2083646438460875,
      "learning_rate": 8.451170805018964e-07,
      "loss": 0.0436,
      "step": 8471
    },
    {
      "epoch": 1.9276450511945393,
      "grad_norm": 0.5160432674685907,
      "learning_rate": 8.45033451707387e-07,
      "loss": 0.0126,
      "step": 8472
    },
    {
      "epoch": 1.927872582480091,
      "grad_norm": 2.133760042950509,
      "learning_rate": 8.449498184157968e-07,
      "loss": 0.0464,
      "step": 8473
    },
    {
      "epoch": 1.9281001137656428,
      "grad_norm": 1.1469697318986505,
      "learning_rate": 8.448661806288352e-07,
      "loss": 0.0248,
      "step": 8474
    },
    {
      "epoch": 1.9283276450511946,
      "grad_norm": 1.1436508537006016,
      "learning_rate": 8.447825383482116e-07,
      "loss": 0.0202,
      "step": 8475
    },
    {
      "epoch": 1.9285551763367463,
      "grad_norm": 1.1291433248112117,
      "learning_rate": 8.446988915756353e-07,
      "loss": 0.0214,
      "step": 8476
    },
    {
      "epoch": 1.928782707622298,
      "grad_norm": 1.4797955722306269,
      "learning_rate": 8.446152403128161e-07,
      "loss": 0.0467,
      "step": 8477
    },
    {
      "epoch": 1.9290102389078498,
      "grad_norm": 1.0466122715039303,
      "learning_rate": 8.445315845614636e-07,
      "loss": 0.0421,
      "step": 8478
    },
    {
      "epoch": 1.9292377701934016,
      "grad_norm": 1.5338203633569327,
      "learning_rate": 8.444479243232875e-07,
      "loss": 0.0208,
      "step": 8479
    },
    {
      "epoch": 1.9294653014789533,
      "grad_norm": 1.123497381048244,
      "learning_rate": 8.443642595999977e-07,
      "loss": 0.0185,
      "step": 8480
    },
    {
      "epoch": 1.929692832764505,
      "grad_norm": 0.9233591838211316,
      "learning_rate": 8.442805903933041e-07,
      "loss": 0.013,
      "step": 8481
    },
    {
      "epoch": 1.9299203640500568,
      "grad_norm": 1.3201346964016165,
      "learning_rate": 8.441969167049171e-07,
      "loss": 0.0316,
      "step": 8482
    },
    {
      "epoch": 1.9301478953356086,
      "grad_norm": 1.332722144963646,
      "learning_rate": 8.441132385365462e-07,
      "loss": 0.0132,
      "step": 8483
    },
    {
      "epoch": 1.9303754266211604,
      "grad_norm": 0.5893018408424756,
      "learning_rate": 8.440295558899024e-07,
      "loss": 0.0095,
      "step": 8484
    },
    {
      "epoch": 1.930602957906712,
      "grad_norm": 1.3366402306240976,
      "learning_rate": 8.439458687666954e-07,
      "loss": 0.0354,
      "step": 8485
    },
    {
      "epoch": 1.9308304891922639,
      "grad_norm": 1.825240306798872,
      "learning_rate": 8.438621771686358e-07,
      "loss": 0.0419,
      "step": 8486
    },
    {
      "epoch": 1.9310580204778156,
      "grad_norm": 1.1674735397270124,
      "learning_rate": 8.437784810974343e-07,
      "loss": 0.014,
      "step": 8487
    },
    {
      "epoch": 1.9312855517633674,
      "grad_norm": 0.8491504333753751,
      "learning_rate": 8.43694780554801e-07,
      "loss": 0.019,
      "step": 8488
    },
    {
      "epoch": 1.931513083048919,
      "grad_norm": 0.9480439733246314,
      "learning_rate": 8.436110755424472e-07,
      "loss": 0.0174,
      "step": 8489
    },
    {
      "epoch": 1.9317406143344709,
      "grad_norm": 0.6821732577268869,
      "learning_rate": 8.435273660620833e-07,
      "loss": 0.0111,
      "step": 8490
    },
    {
      "epoch": 1.9319681456200226,
      "grad_norm": 1.3852655380236707,
      "learning_rate": 8.434436521154202e-07,
      "loss": 0.0317,
      "step": 8491
    },
    {
      "epoch": 1.9321956769055744,
      "grad_norm": 1.0833902156494688,
      "learning_rate": 8.43359933704169e-07,
      "loss": 0.0244,
      "step": 8492
    },
    {
      "epoch": 1.9324232081911261,
      "grad_norm": 1.0603517811346055,
      "learning_rate": 8.432762108300404e-07,
      "loss": 0.027,
      "step": 8493
    },
    {
      "epoch": 1.932650739476678,
      "grad_norm": 1.366822240460289,
      "learning_rate": 8.431924834947461e-07,
      "loss": 0.0633,
      "step": 8494
    },
    {
      "epoch": 1.9328782707622298,
      "grad_norm": 1.0761472143543775,
      "learning_rate": 8.431087516999969e-07,
      "loss": 0.0256,
      "step": 8495
    },
    {
      "epoch": 1.9331058020477816,
      "grad_norm": 1.4433162417817076,
      "learning_rate": 8.430250154475044e-07,
      "loss": 0.0476,
      "step": 8496
    },
    {
      "epoch": 1.9333333333333333,
      "grad_norm": 0.9530040965882383,
      "learning_rate": 8.429412747389798e-07,
      "loss": 0.024,
      "step": 8497
    },
    {
      "epoch": 1.933560864618885,
      "grad_norm": 0.4978173677065556,
      "learning_rate": 8.428575295761346e-07,
      "loss": 0.0067,
      "step": 8498
    },
    {
      "epoch": 1.9337883959044369,
      "grad_norm": 1.1905435081364368,
      "learning_rate": 8.427737799606806e-07,
      "loss": 0.018,
      "step": 8499
    },
    {
      "epoch": 1.9340159271899886,
      "grad_norm": 1.012957267594849,
      "learning_rate": 8.426900258943292e-07,
      "loss": 0.0195,
      "step": 8500
    },
    {
      "epoch": 1.9342434584755404,
      "grad_norm": 0.9089230595622955,
      "learning_rate": 8.426062673787926e-07,
      "loss": 0.0242,
      "step": 8501
    },
    {
      "epoch": 1.934470989761092,
      "grad_norm": 1.3717302642577704,
      "learning_rate": 8.425225044157823e-07,
      "loss": 0.0312,
      "step": 8502
    },
    {
      "epoch": 1.9346985210466439,
      "grad_norm": 0.9103155187419014,
      "learning_rate": 8.424387370070102e-07,
      "loss": 0.0226,
      "step": 8503
    },
    {
      "epoch": 1.9349260523321958,
      "grad_norm": 1.9980223373755015,
      "learning_rate": 8.423549651541889e-07,
      "loss": 0.0469,
      "step": 8504
    },
    {
      "epoch": 1.9351535836177476,
      "grad_norm": 1.2594006281057515,
      "learning_rate": 8.4227118885903e-07,
      "loss": 0.0252,
      "step": 8505
    },
    {
      "epoch": 1.9353811149032993,
      "grad_norm": 1.3049118421199715,
      "learning_rate": 8.421874081232459e-07,
      "loss": 0.0379,
      "step": 8506
    },
    {
      "epoch": 1.935608646188851,
      "grad_norm": 0.6477534084196926,
      "learning_rate": 8.421036229485489e-07,
      "loss": 0.01,
      "step": 8507
    },
    {
      "epoch": 1.9358361774744028,
      "grad_norm": 1.1696824410055644,
      "learning_rate": 8.420198333366513e-07,
      "loss": 0.0259,
      "step": 8508
    },
    {
      "epoch": 1.9360637087599546,
      "grad_norm": 0.8079709445966895,
      "learning_rate": 8.419360392892663e-07,
      "loss": 0.0248,
      "step": 8509
    },
    {
      "epoch": 1.9362912400455063,
      "grad_norm": 1.4511253525917833,
      "learning_rate": 8.418522408081054e-07,
      "loss": 0.0214,
      "step": 8510
    },
    {
      "epoch": 1.936518771331058,
      "grad_norm": 1.129206025501574,
      "learning_rate": 8.417684378948822e-07,
      "loss": 0.0222,
      "step": 8511
    },
    {
      "epoch": 1.9367463026166098,
      "grad_norm": 0.842844775664999,
      "learning_rate": 8.416846305513089e-07,
      "loss": 0.0106,
      "step": 8512
    },
    {
      "epoch": 1.9369738339021616,
      "grad_norm": 0.9685261785917861,
      "learning_rate": 8.416008187790986e-07,
      "loss": 0.0125,
      "step": 8513
    },
    {
      "epoch": 1.9372013651877134,
      "grad_norm": 1.0807816755151243,
      "learning_rate": 8.415170025799644e-07,
      "loss": 0.0302,
      "step": 8514
    },
    {
      "epoch": 1.937428896473265,
      "grad_norm": 0.745171251075455,
      "learning_rate": 8.414331819556193e-07,
      "loss": 0.0064,
      "step": 8515
    },
    {
      "epoch": 1.9376564277588169,
      "grad_norm": 1.1211889513548992,
      "learning_rate": 8.41349356907776e-07,
      "loss": 0.0439,
      "step": 8516
    },
    {
      "epoch": 1.9378839590443686,
      "grad_norm": 0.8981099859269421,
      "learning_rate": 8.412655274381481e-07,
      "loss": 0.0195,
      "step": 8517
    },
    {
      "epoch": 1.9381114903299204,
      "grad_norm": 1.671316797260959,
      "learning_rate": 8.411816935484491e-07,
      "loss": 0.0426,
      "step": 8518
    },
    {
      "epoch": 1.9383390216154721,
      "grad_norm": 1.183556123114075,
      "learning_rate": 8.410978552403923e-07,
      "loss": 0.0414,
      "step": 8519
    },
    {
      "epoch": 1.9385665529010239,
      "grad_norm": 0.716076072097655,
      "learning_rate": 8.410140125156907e-07,
      "loss": 0.0126,
      "step": 8520
    },
    {
      "epoch": 1.9387940841865756,
      "grad_norm": 1.33588142832007,
      "learning_rate": 8.409301653760587e-07,
      "loss": 0.0271,
      "step": 8521
    },
    {
      "epoch": 1.9390216154721274,
      "grad_norm": 1.0341941339252312,
      "learning_rate": 8.408463138232094e-07,
      "loss": 0.0188,
      "step": 8522
    },
    {
      "epoch": 1.9392491467576791,
      "grad_norm": 1.5332586177406462,
      "learning_rate": 8.407624578588566e-07,
      "loss": 0.0439,
      "step": 8523
    },
    {
      "epoch": 1.9394766780432309,
      "grad_norm": 2.010127945672871,
      "learning_rate": 8.406785974847145e-07,
      "loss": 0.0328,
      "step": 8524
    },
    {
      "epoch": 1.9397042093287826,
      "grad_norm": 0.6510194841437853,
      "learning_rate": 8.405947327024968e-07,
      "loss": 0.0151,
      "step": 8525
    },
    {
      "epoch": 1.9399317406143344,
      "grad_norm": 1.2048863218421133,
      "learning_rate": 8.405108635139178e-07,
      "loss": 0.0184,
      "step": 8526
    },
    {
      "epoch": 1.9401592718998861,
      "grad_norm": 1.2753069331272529,
      "learning_rate": 8.404269899206911e-07,
      "loss": 0.0286,
      "step": 8527
    },
    {
      "epoch": 1.9403868031854379,
      "grad_norm": 1.3484674371507035,
      "learning_rate": 8.403431119245316e-07,
      "loss": 0.054,
      "step": 8528
    },
    {
      "epoch": 1.9406143344709896,
      "grad_norm": 0.9252465578467342,
      "learning_rate": 8.40259229527153e-07,
      "loss": 0.0242,
      "step": 8529
    },
    {
      "epoch": 1.9408418657565414,
      "grad_norm": 0.83021150201602,
      "learning_rate": 8.4017534273027e-07,
      "loss": 0.0304,
      "step": 8530
    },
    {
      "epoch": 1.9410693970420931,
      "grad_norm": 1.925549082550425,
      "learning_rate": 8.400914515355972e-07,
      "loss": 0.0526,
      "step": 8531
    },
    {
      "epoch": 1.9412969283276449,
      "grad_norm": 1.3291950350754373,
      "learning_rate": 8.40007555944849e-07,
      "loss": 0.0215,
      "step": 8532
    },
    {
      "epoch": 1.9415244596131969,
      "grad_norm": 0.917419574940405,
      "learning_rate": 8.399236559597403e-07,
      "loss": 0.0162,
      "step": 8533
    },
    {
      "epoch": 1.9417519908987486,
      "grad_norm": 1.0211676002921655,
      "learning_rate": 8.398397515819855e-07,
      "loss": 0.0244,
      "step": 8534
    },
    {
      "epoch": 1.9419795221843004,
      "grad_norm": 0.7994766197110429,
      "learning_rate": 8.397558428132995e-07,
      "loss": 0.0185,
      "step": 8535
    },
    {
      "epoch": 1.9422070534698521,
      "grad_norm": 3.5208757922434404,
      "learning_rate": 8.396719296553976e-07,
      "loss": 0.089,
      "step": 8536
    },
    {
      "epoch": 1.9424345847554039,
      "grad_norm": 1.9415052584017878,
      "learning_rate": 8.395880121099944e-07,
      "loss": 0.0625,
      "step": 8537
    },
    {
      "epoch": 1.9426621160409556,
      "grad_norm": 1.470869728302076,
      "learning_rate": 8.395040901788056e-07,
      "loss": 0.0397,
      "step": 8538
    },
    {
      "epoch": 1.9428896473265074,
      "grad_norm": 1.21141425306834,
      "learning_rate": 8.394201638635458e-07,
      "loss": 0.0299,
      "step": 8539
    },
    {
      "epoch": 1.9431171786120591,
      "grad_norm": 1.537393692306185,
      "learning_rate": 8.393362331659305e-07,
      "loss": 0.0484,
      "step": 8540
    },
    {
      "epoch": 1.9433447098976109,
      "grad_norm": 0.6350523232851911,
      "learning_rate": 8.392522980876752e-07,
      "loss": 0.0226,
      "step": 8541
    },
    {
      "epoch": 1.9435722411831626,
      "grad_norm": 0.9508792371503724,
      "learning_rate": 8.391683586304954e-07,
      "loss": 0.0232,
      "step": 8542
    },
    {
      "epoch": 1.9437997724687146,
      "grad_norm": 0.6704249174467451,
      "learning_rate": 8.390844147961064e-07,
      "loss": 0.014,
      "step": 8543
    },
    {
      "epoch": 1.9440273037542664,
      "grad_norm": 0.8355479717869047,
      "learning_rate": 8.39000466586224e-07,
      "loss": 0.0132,
      "step": 8544
    },
    {
      "epoch": 1.944254835039818,
      "grad_norm": 1.4903130768638286,
      "learning_rate": 8.389165140025642e-07,
      "loss": 0.0557,
      "step": 8545
    },
    {
      "epoch": 1.9444823663253699,
      "grad_norm": 1.4269569411160623,
      "learning_rate": 8.388325570468425e-07,
      "loss": 0.042,
      "step": 8546
    },
    {
      "epoch": 1.9447098976109216,
      "grad_norm": 2.033630266136058,
      "learning_rate": 8.38748595720775e-07,
      "loss": 0.0447,
      "step": 8547
    },
    {
      "epoch": 1.9449374288964734,
      "grad_norm": 1.1410481102298577,
      "learning_rate": 8.386646300260777e-07,
      "loss": 0.0355,
      "step": 8548
    },
    {
      "epoch": 1.9451649601820251,
      "grad_norm": 0.6948480718809295,
      "learning_rate": 8.385806599644666e-07,
      "loss": 0.0137,
      "step": 8549
    },
    {
      "epoch": 1.9453924914675769,
      "grad_norm": 0.8140264883212044,
      "learning_rate": 8.384966855376579e-07,
      "loss": 0.0113,
      "step": 8550
    },
    {
      "epoch": 1.9456200227531286,
      "grad_norm": 33.67982669588671,
      "learning_rate": 8.384127067473681e-07,
      "loss": 0.2731,
      "step": 8551
    },
    {
      "epoch": 1.9458475540386804,
      "grad_norm": 0.5758831577423202,
      "learning_rate": 8.383287235953133e-07,
      "loss": 0.0068,
      "step": 8552
    },
    {
      "epoch": 1.9460750853242321,
      "grad_norm": 2.176471810433205,
      "learning_rate": 8.382447360832102e-07,
      "loss": 0.0658,
      "step": 8553
    },
    {
      "epoch": 1.9463026166097839,
      "grad_norm": 0.8880018882253463,
      "learning_rate": 8.381607442127753e-07,
      "loss": 0.0201,
      "step": 8554
    },
    {
      "epoch": 1.9465301478953356,
      "grad_norm": 1.199962036020287,
      "learning_rate": 8.38076747985725e-07,
      "loss": 0.0464,
      "step": 8555
    },
    {
      "epoch": 1.9467576791808874,
      "grad_norm": 0.9170057900529092,
      "learning_rate": 8.379927474037762e-07,
      "loss": 0.0362,
      "step": 8556
    },
    {
      "epoch": 1.9469852104664391,
      "grad_norm": 1.3486833781970093,
      "learning_rate": 8.379087424686458e-07,
      "loss": 0.0575,
      "step": 8557
    },
    {
      "epoch": 1.9472127417519909,
      "grad_norm": 1.0923652362515575,
      "learning_rate": 8.378247331820504e-07,
      "loss": 0.0163,
      "step": 8558
    },
    {
      "epoch": 1.9474402730375426,
      "grad_norm": 1.0406302266708418,
      "learning_rate": 8.377407195457077e-07,
      "loss": 0.0237,
      "step": 8559
    },
    {
      "epoch": 1.9476678043230944,
      "grad_norm": 1.1825169811274758,
      "learning_rate": 8.376567015613339e-07,
      "loss": 0.0275,
      "step": 8560
    },
    {
      "epoch": 1.9478953356086461,
      "grad_norm": 1.1379295942906191,
      "learning_rate": 8.375726792306467e-07,
      "loss": 0.027,
      "step": 8561
    },
    {
      "epoch": 1.948122866894198,
      "grad_norm": 0.8123191602181989,
      "learning_rate": 8.374886525553632e-07,
      "loss": 0.0194,
      "step": 8562
    },
    {
      "epoch": 1.9483503981797496,
      "grad_norm": 1.3859418636229017,
      "learning_rate": 8.374046215372011e-07,
      "loss": 0.0303,
      "step": 8563
    },
    {
      "epoch": 1.9485779294653014,
      "grad_norm": 0.8420953328721413,
      "learning_rate": 8.37320586177877e-07,
      "loss": 0.0126,
      "step": 8564
    },
    {
      "epoch": 1.9488054607508531,
      "grad_norm": 1.1485451801525324,
      "learning_rate": 8.372365464791094e-07,
      "loss": 0.0177,
      "step": 8565
    },
    {
      "epoch": 1.949032992036405,
      "grad_norm": 1.3651782750195407,
      "learning_rate": 8.371525024426153e-07,
      "loss": 0.0254,
      "step": 8566
    },
    {
      "epoch": 1.9492605233219567,
      "grad_norm": 0.9995494790984009,
      "learning_rate": 8.370684540701126e-07,
      "loss": 0.0204,
      "step": 8567
    },
    {
      "epoch": 1.9494880546075084,
      "grad_norm": 1.5531029356279562,
      "learning_rate": 8.369844013633191e-07,
      "loss": 0.0479,
      "step": 8568
    },
    {
      "epoch": 1.9497155858930602,
      "grad_norm": 1.024177847940211,
      "learning_rate": 8.369003443239528e-07,
      "loss": 0.0309,
      "step": 8569
    },
    {
      "epoch": 1.949943117178612,
      "grad_norm": 0.9818950626534545,
      "learning_rate": 8.368162829537313e-07,
      "loss": 0.0165,
      "step": 8570
    },
    {
      "epoch": 1.9501706484641637,
      "grad_norm": 0.5604927182640149,
      "learning_rate": 8.367322172543729e-07,
      "loss": 0.0117,
      "step": 8571
    },
    {
      "epoch": 1.9503981797497156,
      "grad_norm": 0.9133949942753709,
      "learning_rate": 8.366481472275961e-07,
      "loss": 0.0188,
      "step": 8572
    },
    {
      "epoch": 1.9506257110352674,
      "grad_norm": 1.2929357532911143,
      "learning_rate": 8.365640728751184e-07,
      "loss": 0.0302,
      "step": 8573
    },
    {
      "epoch": 1.9508532423208191,
      "grad_norm": 0.7383923021797731,
      "learning_rate": 8.364799941986587e-07,
      "loss": 0.0187,
      "step": 8574
    },
    {
      "epoch": 1.9510807736063709,
      "grad_norm": 1.1839304673917488,
      "learning_rate": 8.363959111999352e-07,
      "loss": 0.0377,
      "step": 8575
    },
    {
      "epoch": 1.9513083048919226,
      "grad_norm": 1.1521870732115753,
      "learning_rate": 8.363118238806663e-07,
      "loss": 0.0208,
      "step": 8576
    },
    {
      "epoch": 1.9515358361774744,
      "grad_norm": 0.8447968990656242,
      "learning_rate": 8.362277322425709e-07,
      "loss": 0.0207,
      "step": 8577
    },
    {
      "epoch": 1.9517633674630261,
      "grad_norm": 3.511151868539225,
      "learning_rate": 8.361436362873676e-07,
      "loss": 0.1474,
      "step": 8578
    },
    {
      "epoch": 1.951990898748578,
      "grad_norm": 1.0506334321174164,
      "learning_rate": 8.360595360167748e-07,
      "loss": 0.03,
      "step": 8579
    },
    {
      "epoch": 1.9522184300341296,
      "grad_norm": 1.4388344988078532,
      "learning_rate": 8.359754314325117e-07,
      "loss": 0.0428,
      "step": 8580
    },
    {
      "epoch": 1.9524459613196816,
      "grad_norm": 0.9175157387598539,
      "learning_rate": 8.358913225362972e-07,
      "loss": 0.0212,
      "step": 8581
    },
    {
      "epoch": 1.9526734926052334,
      "grad_norm": 1.1474465849879947,
      "learning_rate": 8.358072093298503e-07,
      "loss": 0.0319,
      "step": 8582
    },
    {
      "epoch": 1.9529010238907851,
      "grad_norm": 0.9606355848307595,
      "learning_rate": 8.357230918148901e-07,
      "loss": 0.0261,
      "step": 8583
    },
    {
      "epoch": 1.9531285551763369,
      "grad_norm": 0.8419564239260479,
      "learning_rate": 8.356389699931359e-07,
      "loss": 0.0295,
      "step": 8584
    },
    {
      "epoch": 1.9533560864618886,
      "grad_norm": 0.739116737145134,
      "learning_rate": 8.35554843866307e-07,
      "loss": 0.0142,
      "step": 8585
    },
    {
      "epoch": 1.9535836177474404,
      "grad_norm": 1.4018896373401493,
      "learning_rate": 8.354707134361224e-07,
      "loss": 0.0415,
      "step": 8586
    },
    {
      "epoch": 1.9538111490329921,
      "grad_norm": 1.4592585138779515,
      "learning_rate": 8.35386578704302e-07,
      "loss": 0.0287,
      "step": 8587
    },
    {
      "epoch": 1.9540386803185439,
      "grad_norm": 1.7715878188929528,
      "learning_rate": 8.353024396725653e-07,
      "loss": 0.0538,
      "step": 8588
    },
    {
      "epoch": 1.9542662116040956,
      "grad_norm": 1.3693419664466187,
      "learning_rate": 8.352182963426317e-07,
      "loss": 0.0198,
      "step": 8589
    },
    {
      "epoch": 1.9544937428896474,
      "grad_norm": 1.1544739346084487,
      "learning_rate": 8.351341487162214e-07,
      "loss": 0.0254,
      "step": 8590
    },
    {
      "epoch": 1.9547212741751991,
      "grad_norm": 2.1104083976647012,
      "learning_rate": 8.350499967950538e-07,
      "loss": 0.0492,
      "step": 8591
    },
    {
      "epoch": 1.954948805460751,
      "grad_norm": 1.1692825473703383,
      "learning_rate": 8.349658405808489e-07,
      "loss": 0.0258,
      "step": 8592
    },
    {
      "epoch": 1.9551763367463026,
      "grad_norm": 1.2419564910884628,
      "learning_rate": 8.348816800753269e-07,
      "loss": 0.0293,
      "step": 8593
    },
    {
      "epoch": 1.9554038680318544,
      "grad_norm": 1.1309042179682378,
      "learning_rate": 8.347975152802075e-07,
      "loss": 0.0264,
      "step": 8594
    },
    {
      "epoch": 1.9556313993174061,
      "grad_norm": 1.4290099146173239,
      "learning_rate": 8.347133461972112e-07,
      "loss": 0.0395,
      "step": 8595
    },
    {
      "epoch": 1.955858930602958,
      "grad_norm": 1.4936400221654453,
      "learning_rate": 8.346291728280584e-07,
      "loss": 0.0467,
      "step": 8596
    },
    {
      "epoch": 1.9560864618885097,
      "grad_norm": 0.7787161379756027,
      "learning_rate": 8.345449951744692e-07,
      "loss": 0.0117,
      "step": 8597
    },
    {
      "epoch": 1.9563139931740614,
      "grad_norm": 0.9751506057590029,
      "learning_rate": 8.344608132381637e-07,
      "loss": 0.014,
      "step": 8598
    },
    {
      "epoch": 1.9565415244596132,
      "grad_norm": 1.3679824917988233,
      "learning_rate": 8.343766270208631e-07,
      "loss": 0.0309,
      "step": 8599
    },
    {
      "epoch": 1.956769055745165,
      "grad_norm": 0.7086403075954423,
      "learning_rate": 8.342924365242878e-07,
      "loss": 0.0187,
      "step": 8600
    },
    {
      "epoch": 1.9569965870307167,
      "grad_norm": 1.5007550176295361,
      "learning_rate": 8.342082417501579e-07,
      "loss": 0.0214,
      "step": 8601
    },
    {
      "epoch": 1.9572241183162684,
      "grad_norm": 0.9485634208130852,
      "learning_rate": 8.341240427001951e-07,
      "loss": 0.024,
      "step": 8602
    },
    {
      "epoch": 1.9574516496018202,
      "grad_norm": 1.1100998316363275,
      "learning_rate": 8.340398393761199e-07,
      "loss": 0.0275,
      "step": 8603
    },
    {
      "epoch": 1.957679180887372,
      "grad_norm": 0.8641664732957542,
      "learning_rate": 8.339556317796529e-07,
      "loss": 0.0133,
      "step": 8604
    },
    {
      "epoch": 1.9579067121729237,
      "grad_norm": 0.8683423704438483,
      "learning_rate": 8.338714199125157e-07,
      "loss": 0.0192,
      "step": 8605
    },
    {
      "epoch": 1.9581342434584754,
      "grad_norm": 1.128209302155601,
      "learning_rate": 8.337872037764292e-07,
      "loss": 0.0278,
      "step": 8606
    },
    {
      "epoch": 1.9583617747440272,
      "grad_norm": 0.766978896602123,
      "learning_rate": 8.337029833731145e-07,
      "loss": 0.0173,
      "step": 8607
    },
    {
      "epoch": 1.958589306029579,
      "grad_norm": 1.2084456984055698,
      "learning_rate": 8.336187587042932e-07,
      "loss": 0.0299,
      "step": 8608
    },
    {
      "epoch": 1.9588168373151307,
      "grad_norm": 0.9962730444702378,
      "learning_rate": 8.335345297716863e-07,
      "loss": 0.0386,
      "step": 8609
    },
    {
      "epoch": 1.9590443686006824,
      "grad_norm": 1.1075213008133922,
      "learning_rate": 8.334502965770158e-07,
      "loss": 0.0402,
      "step": 8610
    },
    {
      "epoch": 1.9592718998862344,
      "grad_norm": 0.6988985648561841,
      "learning_rate": 8.33366059122003e-07,
      "loss": 0.011,
      "step": 8611
    },
    {
      "epoch": 1.9594994311717862,
      "grad_norm": 1.144705818495189,
      "learning_rate": 8.332818174083694e-07,
      "loss": 0.0297,
      "step": 8612
    },
    {
      "epoch": 1.959726962457338,
      "grad_norm": 1.058490736306249,
      "learning_rate": 8.331975714378369e-07,
      "loss": 0.0159,
      "step": 8613
    },
    {
      "epoch": 1.9599544937428897,
      "grad_norm": 1.091387700160984,
      "learning_rate": 8.331133212121273e-07,
      "loss": 0.0262,
      "step": 8614
    },
    {
      "epoch": 1.9601820250284414,
      "grad_norm": 0.9330086964337541,
      "learning_rate": 8.330290667329627e-07,
      "loss": 0.0225,
      "step": 8615
    },
    {
      "epoch": 1.9604095563139932,
      "grad_norm": 1.967442475907266,
      "learning_rate": 8.329448080020651e-07,
      "loss": 0.074,
      "step": 8616
    },
    {
      "epoch": 1.960637087599545,
      "grad_norm": 1.1163242624066385,
      "learning_rate": 8.32860545021156e-07,
      "loss": 0.0386,
      "step": 8617
    },
    {
      "epoch": 1.9608646188850967,
      "grad_norm": 0.970224699604641,
      "learning_rate": 8.327762777919585e-07,
      "loss": 0.0164,
      "step": 8618
    },
    {
      "epoch": 1.9610921501706484,
      "grad_norm": 0.5276741067783547,
      "learning_rate": 8.326920063161942e-07,
      "loss": 0.0178,
      "step": 8619
    },
    {
      "epoch": 1.9613196814562004,
      "grad_norm": 1.0009730372919852,
      "learning_rate": 8.326077305955857e-07,
      "loss": 0.0231,
      "step": 8620
    },
    {
      "epoch": 1.9615472127417521,
      "grad_norm": 1.2868902445641404,
      "learning_rate": 8.325234506318553e-07,
      "loss": 0.033,
      "step": 8621
    },
    {
      "epoch": 1.961774744027304,
      "grad_norm": 1.1246482872639023,
      "learning_rate": 8.324391664267257e-07,
      "loss": 0.0394,
      "step": 8622
    },
    {
      "epoch": 1.9620022753128556,
      "grad_norm": 1.3455231367897886,
      "learning_rate": 8.323548779819194e-07,
      "loss": 0.0298,
      "step": 8623
    },
    {
      "epoch": 1.9622298065984074,
      "grad_norm": 1.6393099584172452,
      "learning_rate": 8.322705852991592e-07,
      "loss": 0.0436,
      "step": 8624
    },
    {
      "epoch": 1.9624573378839592,
      "grad_norm": 0.7536024647751555,
      "learning_rate": 8.321862883801677e-07,
      "loss": 0.0111,
      "step": 8625
    },
    {
      "epoch": 1.962684869169511,
      "grad_norm": 0.773879477613702,
      "learning_rate": 8.321019872266682e-07,
      "loss": 0.022,
      "step": 8626
    },
    {
      "epoch": 1.9629124004550627,
      "grad_norm": 1.1594672149335956,
      "learning_rate": 8.320176818403831e-07,
      "loss": 0.0174,
      "step": 8627
    },
    {
      "epoch": 1.9631399317406144,
      "grad_norm": 1.0266232168394467,
      "learning_rate": 8.319333722230359e-07,
      "loss": 0.0328,
      "step": 8628
    },
    {
      "epoch": 1.9633674630261662,
      "grad_norm": 0.8306133501722734,
      "learning_rate": 8.318490583763494e-07,
      "loss": 0.0253,
      "step": 8629
    },
    {
      "epoch": 1.963594994311718,
      "grad_norm": 1.2649424862296168,
      "learning_rate": 8.317647403020472e-07,
      "loss": 0.0475,
      "step": 8630
    },
    {
      "epoch": 1.9638225255972697,
      "grad_norm": 2.014816072849253,
      "learning_rate": 8.316804180018522e-07,
      "loss": 0.0552,
      "step": 8631
    },
    {
      "epoch": 1.9640500568828214,
      "grad_norm": 0.49606898198662275,
      "learning_rate": 8.315960914774878e-07,
      "loss": 0.0075,
      "step": 8632
    },
    {
      "epoch": 1.9642775881683732,
      "grad_norm": 1.2199716369799536,
      "learning_rate": 8.31511760730678e-07,
      "loss": 0.0305,
      "step": 8633
    },
    {
      "epoch": 1.964505119453925,
      "grad_norm": 1.320463922808405,
      "learning_rate": 8.314274257631458e-07,
      "loss": 0.0253,
      "step": 8634
    },
    {
      "epoch": 1.9647326507394767,
      "grad_norm": 0.7885505673935275,
      "learning_rate": 8.313430865766154e-07,
      "loss": 0.014,
      "step": 8635
    },
    {
      "epoch": 1.9649601820250284,
      "grad_norm": 1.484537808236749,
      "learning_rate": 8.312587431728098e-07,
      "loss": 0.0246,
      "step": 8636
    },
    {
      "epoch": 1.9651877133105802,
      "grad_norm": 1.8047818594210032,
      "learning_rate": 8.311743955534533e-07,
      "loss": 0.0434,
      "step": 8637
    },
    {
      "epoch": 1.965415244596132,
      "grad_norm": 1.6774167551399732,
      "learning_rate": 8.310900437202699e-07,
      "loss": 0.0204,
      "step": 8638
    },
    {
      "epoch": 1.9656427758816837,
      "grad_norm": 0.7736895427296182,
      "learning_rate": 8.310056876749833e-07,
      "loss": 0.017,
      "step": 8639
    },
    {
      "epoch": 1.9658703071672354,
      "grad_norm": 1.3766383243992963,
      "learning_rate": 8.309213274193179e-07,
      "loss": 0.0243,
      "step": 8640
    },
    {
      "epoch": 1.9660978384527872,
      "grad_norm": 10.91155799511908,
      "learning_rate": 8.308369629549976e-07,
      "loss": 0.1186,
      "step": 8641
    },
    {
      "epoch": 1.966325369738339,
      "grad_norm": 0.9268482275101161,
      "learning_rate": 8.307525942837468e-07,
      "loss": 0.0327,
      "step": 8642
    },
    {
      "epoch": 1.9665529010238907,
      "grad_norm": 1.9798265565542381,
      "learning_rate": 8.306682214072897e-07,
      "loss": 0.0719,
      "step": 8643
    },
    {
      "epoch": 1.9667804323094424,
      "grad_norm": 1.1530434422527183,
      "learning_rate": 8.305838443273508e-07,
      "loss": 0.0312,
      "step": 8644
    },
    {
      "epoch": 1.9670079635949942,
      "grad_norm": 1.6632270883813656,
      "learning_rate": 8.304994630456546e-07,
      "loss": 0.0382,
      "step": 8645
    },
    {
      "epoch": 1.967235494880546,
      "grad_norm": 1.2715280679684924,
      "learning_rate": 8.304150775639258e-07,
      "loss": 0.0234,
      "step": 8646
    },
    {
      "epoch": 1.9674630261660977,
      "grad_norm": 1.3001980514362683,
      "learning_rate": 8.303306878838892e-07,
      "loss": 0.031,
      "step": 8647
    },
    {
      "epoch": 1.9676905574516494,
      "grad_norm": 0.8323495719274282,
      "learning_rate": 8.302462940072691e-07,
      "loss": 0.0167,
      "step": 8648
    },
    {
      "epoch": 1.9679180887372012,
      "grad_norm": 0.7713193628443443,
      "learning_rate": 8.301618959357908e-07,
      "loss": 0.0276,
      "step": 8649
    },
    {
      "epoch": 1.9681456200227532,
      "grad_norm": 1.1972821252281405,
      "learning_rate": 8.300774936711792e-07,
      "loss": 0.0371,
      "step": 8650
    },
    {
      "epoch": 1.968373151308305,
      "grad_norm": 1.1958620099652342,
      "learning_rate": 8.299930872151589e-07,
      "loss": 0.0206,
      "step": 8651
    },
    {
      "epoch": 1.9686006825938567,
      "grad_norm": 1.3119998832101367,
      "learning_rate": 8.299086765694554e-07,
      "loss": 0.0227,
      "step": 8652
    },
    {
      "epoch": 1.9688282138794084,
      "grad_norm": 1.1941007190599409,
      "learning_rate": 8.298242617357939e-07,
      "loss": 0.0185,
      "step": 8653
    },
    {
      "epoch": 1.9690557451649602,
      "grad_norm": 1.017322753157879,
      "learning_rate": 8.297398427158996e-07,
      "loss": 0.0255,
      "step": 8654
    },
    {
      "epoch": 1.969283276450512,
      "grad_norm": 1.073134449146722,
      "learning_rate": 8.296554195114979e-07,
      "loss": 0.0282,
      "step": 8655
    },
    {
      "epoch": 1.9695108077360637,
      "grad_norm": 0.9658448239276997,
      "learning_rate": 8.295709921243143e-07,
      "loss": 0.0227,
      "step": 8656
    },
    {
      "epoch": 1.9697383390216154,
      "grad_norm": 0.7505703251353405,
      "learning_rate": 8.294865605560743e-07,
      "loss": 0.0183,
      "step": 8657
    },
    {
      "epoch": 1.9699658703071672,
      "grad_norm": 0.8668155064414824,
      "learning_rate": 8.294021248085032e-07,
      "loss": 0.026,
      "step": 8658
    },
    {
      "epoch": 1.9701934015927192,
      "grad_norm": 0.9982979030777734,
      "learning_rate": 8.293176848833273e-07,
      "loss": 0.0234,
      "step": 8659
    },
    {
      "epoch": 1.970420932878271,
      "grad_norm": 1.051098783463964,
      "learning_rate": 8.292332407822721e-07,
      "loss": 0.0243,
      "step": 8660
    },
    {
      "epoch": 1.9706484641638227,
      "grad_norm": 0.6169455454202162,
      "learning_rate": 8.291487925070634e-07,
      "loss": 0.0102,
      "step": 8661
    },
    {
      "epoch": 1.9708759954493744,
      "grad_norm": 1.0371576668223386,
      "learning_rate": 8.290643400594273e-07,
      "loss": 0.0196,
      "step": 8662
    },
    {
      "epoch": 1.9711035267349262,
      "grad_norm": 0.8863462428028039,
      "learning_rate": 8.289798834410898e-07,
      "loss": 0.0195,
      "step": 8663
    },
    {
      "epoch": 1.971331058020478,
      "grad_norm": 0.9635487205693061,
      "learning_rate": 8.288954226537768e-07,
      "loss": 0.0267,
      "step": 8664
    },
    {
      "epoch": 1.9715585893060297,
      "grad_norm": 1.3899273814546962,
      "learning_rate": 8.288109576992151e-07,
      "loss": 0.0502,
      "step": 8665
    },
    {
      "epoch": 1.9717861205915814,
      "grad_norm": 1.2549189430443881,
      "learning_rate": 8.287264885791308e-07,
      "loss": 0.034,
      "step": 8666
    },
    {
      "epoch": 1.9720136518771332,
      "grad_norm": 1.1139192118311512,
      "learning_rate": 8.286420152952499e-07,
      "loss": 0.0306,
      "step": 8667
    },
    {
      "epoch": 1.972241183162685,
      "grad_norm": 1.794539989066656,
      "learning_rate": 8.285575378492992e-07,
      "loss": 0.0303,
      "step": 8668
    },
    {
      "epoch": 1.9724687144482367,
      "grad_norm": 1.1435782603954396,
      "learning_rate": 8.284730562430053e-07,
      "loss": 0.0307,
      "step": 8669
    },
    {
      "epoch": 1.9726962457337884,
      "grad_norm": 1.5817748755403243,
      "learning_rate": 8.283885704780947e-07,
      "loss": 0.0645,
      "step": 8670
    },
    {
      "epoch": 1.9729237770193402,
      "grad_norm": 1.1078746367367593,
      "learning_rate": 8.283040805562942e-07,
      "loss": 0.0215,
      "step": 8671
    },
    {
      "epoch": 1.973151308304892,
      "grad_norm": 1.1982443797342497,
      "learning_rate": 8.282195864793307e-07,
      "loss": 0.0429,
      "step": 8672
    },
    {
      "epoch": 1.9733788395904437,
      "grad_norm": 0.9251868863784273,
      "learning_rate": 8.281350882489311e-07,
      "loss": 0.0167,
      "step": 8673
    },
    {
      "epoch": 1.9736063708759954,
      "grad_norm": 1.6819184224447514,
      "learning_rate": 8.280505858668222e-07,
      "loss": 0.0594,
      "step": 8674
    },
    {
      "epoch": 1.9738339021615472,
      "grad_norm": 1.5761511420584504,
      "learning_rate": 8.279660793347314e-07,
      "loss": 0.0338,
      "step": 8675
    },
    {
      "epoch": 1.974061433447099,
      "grad_norm": 1.1263389265862735,
      "learning_rate": 8.278815686543854e-07,
      "loss": 0.0416,
      "step": 8676
    },
    {
      "epoch": 1.9742889647326507,
      "grad_norm": 1.3226547626551843,
      "learning_rate": 8.277970538275118e-07,
      "loss": 0.0232,
      "step": 8677
    },
    {
      "epoch": 1.9745164960182024,
      "grad_norm": 0.7605156284409775,
      "learning_rate": 8.277125348558376e-07,
      "loss": 0.0204,
      "step": 8678
    },
    {
      "epoch": 1.9747440273037542,
      "grad_norm": 1.0778152208501437,
      "learning_rate": 8.276280117410909e-07,
      "loss": 0.0315,
      "step": 8679
    },
    {
      "epoch": 1.974971558589306,
      "grad_norm": 0.843263740923669,
      "learning_rate": 8.275434844849985e-07,
      "loss": 0.0163,
      "step": 8680
    },
    {
      "epoch": 1.9751990898748577,
      "grad_norm": 0.822307963561357,
      "learning_rate": 8.274589530892881e-07,
      "loss": 0.0124,
      "step": 8681
    },
    {
      "epoch": 1.9754266211604095,
      "grad_norm": 1.6723312975706626,
      "learning_rate": 8.273744175556877e-07,
      "loss": 0.0534,
      "step": 8682
    },
    {
      "epoch": 1.9756541524459612,
      "grad_norm": 0.9524245517137393,
      "learning_rate": 8.272898778859247e-07,
      "loss": 0.0213,
      "step": 8683
    },
    {
      "epoch": 1.975881683731513,
      "grad_norm": 0.9428614388231097,
      "learning_rate": 8.272053340817271e-07,
      "loss": 0.0174,
      "step": 8684
    },
    {
      "epoch": 1.9761092150170647,
      "grad_norm": 0.982980553142515,
      "learning_rate": 8.271207861448227e-07,
      "loss": 0.0434,
      "step": 8685
    },
    {
      "epoch": 1.9763367463026165,
      "grad_norm": 1.7503072386779026,
      "learning_rate": 8.270362340769397e-07,
      "loss": 0.0701,
      "step": 8686
    },
    {
      "epoch": 1.9765642775881682,
      "grad_norm": 1.4686852590607977,
      "learning_rate": 8.269516778798062e-07,
      "loss": 0.0609,
      "step": 8687
    },
    {
      "epoch": 1.9767918088737202,
      "grad_norm": 1.1044537127744922,
      "learning_rate": 8.268671175551499e-07,
      "loss": 0.0279,
      "step": 8688
    },
    {
      "epoch": 1.977019340159272,
      "grad_norm": 0.8675072379447266,
      "learning_rate": 8.267825531046997e-07,
      "loss": 0.0162,
      "step": 8689
    },
    {
      "epoch": 1.9772468714448237,
      "grad_norm": 1.3340777286424959,
      "learning_rate": 8.266979845301837e-07,
      "loss": 0.0347,
      "step": 8690
    },
    {
      "epoch": 1.9774744027303754,
      "grad_norm": 1.862082949838379,
      "learning_rate": 8.266134118333302e-07,
      "loss": 0.0605,
      "step": 8691
    },
    {
      "epoch": 1.9777019340159272,
      "grad_norm": 1.724709349145104,
      "learning_rate": 8.265288350158677e-07,
      "loss": 0.0421,
      "step": 8692
    },
    {
      "epoch": 1.977929465301479,
      "grad_norm": 0.6617687281908288,
      "learning_rate": 8.26444254079525e-07,
      "loss": 0.0155,
      "step": 8693
    },
    {
      "epoch": 1.9781569965870307,
      "grad_norm": 0.9542020767481539,
      "learning_rate": 8.263596690260306e-07,
      "loss": 0.0283,
      "step": 8694
    },
    {
      "epoch": 1.9783845278725825,
      "grad_norm": 1.1565542136712943,
      "learning_rate": 8.262750798571134e-07,
      "loss": 0.0267,
      "step": 8695
    },
    {
      "epoch": 1.9786120591581342,
      "grad_norm": 1.0125080139878857,
      "learning_rate": 8.261904865745022e-07,
      "loss": 0.0297,
      "step": 8696
    },
    {
      "epoch": 1.978839590443686,
      "grad_norm": 1.061120595054336,
      "learning_rate": 8.26105889179926e-07,
      "loss": 0.0233,
      "step": 8697
    },
    {
      "epoch": 1.979067121729238,
      "grad_norm": 1.3367563555151707,
      "learning_rate": 8.260212876751137e-07,
      "loss": 0.0356,
      "step": 8698
    },
    {
      "epoch": 1.9792946530147897,
      "grad_norm": 1.064582252802175,
      "learning_rate": 8.259366820617943e-07,
      "loss": 0.0325,
      "step": 8699
    },
    {
      "epoch": 1.9795221843003414,
      "grad_norm": 1.0237417556631883,
      "learning_rate": 8.258520723416972e-07,
      "loss": 0.0172,
      "step": 8700
    },
    {
      "epoch": 1.9797497155858932,
      "grad_norm": 0.9079968240893457,
      "learning_rate": 8.257674585165515e-07,
      "loss": 0.0234,
      "step": 8701
    },
    {
      "epoch": 1.979977246871445,
      "grad_norm": 0.8075493458055363,
      "learning_rate": 8.256828405880868e-07,
      "loss": 0.02,
      "step": 8702
    },
    {
      "epoch": 1.9802047781569967,
      "grad_norm": 0.8383755855854361,
      "learning_rate": 8.255982185580323e-07,
      "loss": 0.0176,
      "step": 8703
    },
    {
      "epoch": 1.9804323094425484,
      "grad_norm": 0.8588452408891758,
      "learning_rate": 8.255135924281175e-07,
      "loss": 0.0156,
      "step": 8704
    },
    {
      "epoch": 1.9806598407281002,
      "grad_norm": 0.8869722078081338,
      "learning_rate": 8.254289622000724e-07,
      "loss": 0.0156,
      "step": 8705
    },
    {
      "epoch": 1.980887372013652,
      "grad_norm": 1.1750604227287018,
      "learning_rate": 8.253443278756261e-07,
      "loss": 0.0265,
      "step": 8706
    },
    {
      "epoch": 1.9811149032992037,
      "grad_norm": 1.0528660523642601,
      "learning_rate": 8.252596894565088e-07,
      "loss": 0.0287,
      "step": 8707
    },
    {
      "epoch": 1.9813424345847555,
      "grad_norm": 0.8956761378108004,
      "learning_rate": 8.251750469444498e-07,
      "loss": 0.017,
      "step": 8708
    },
    {
      "epoch": 1.9815699658703072,
      "grad_norm": 1.1554876465333057,
      "learning_rate": 8.250904003411798e-07,
      "loss": 0.0269,
      "step": 8709
    },
    {
      "epoch": 1.981797497155859,
      "grad_norm": 1.0472399011095535,
      "learning_rate": 8.250057496484285e-07,
      "loss": 0.0164,
      "step": 8710
    },
    {
      "epoch": 1.9820250284414107,
      "grad_norm": 1.0382917933149183,
      "learning_rate": 8.24921094867926e-07,
      "loss": 0.0208,
      "step": 8711
    },
    {
      "epoch": 1.9822525597269625,
      "grad_norm": 1.7066003981198556,
      "learning_rate": 8.248364360014023e-07,
      "loss": 0.0321,
      "step": 8712
    },
    {
      "epoch": 1.9824800910125142,
      "grad_norm": 1.2766339870608123,
      "learning_rate": 8.247517730505879e-07,
      "loss": 0.0211,
      "step": 8713
    },
    {
      "epoch": 1.982707622298066,
      "grad_norm": 0.9057421594594387,
      "learning_rate": 8.24667106017213e-07,
      "loss": 0.0147,
      "step": 8714
    },
    {
      "epoch": 1.9829351535836177,
      "grad_norm": 0.7963917651951765,
      "learning_rate": 8.245824349030082e-07,
      "loss": 0.0266,
      "step": 8715
    },
    {
      "epoch": 1.9831626848691695,
      "grad_norm": 1.1790616888344387,
      "learning_rate": 8.244977597097039e-07,
      "loss": 0.0467,
      "step": 8716
    },
    {
      "epoch": 1.9833902161547212,
      "grad_norm": 1.0535014935954148,
      "learning_rate": 8.244130804390311e-07,
      "loss": 0.0205,
      "step": 8717
    },
    {
      "epoch": 1.983617747440273,
      "grad_norm": 1.5597342762619004,
      "learning_rate": 8.243283970927196e-07,
      "loss": 0.0392,
      "step": 8718
    },
    {
      "epoch": 1.9838452787258247,
      "grad_norm": 2.409043909428933,
      "learning_rate": 8.242437096725009e-07,
      "loss": 0.0171,
      "step": 8719
    },
    {
      "epoch": 1.9840728100113765,
      "grad_norm": 0.9772091858306938,
      "learning_rate": 8.241590181801059e-07,
      "loss": 0.023,
      "step": 8720
    },
    {
      "epoch": 1.9843003412969282,
      "grad_norm": 1.1293407226155698,
      "learning_rate": 8.240743226172651e-07,
      "loss": 0.0207,
      "step": 8721
    },
    {
      "epoch": 1.98452787258248,
      "grad_norm": 0.8710932227462915,
      "learning_rate": 8.239896229857096e-07,
      "loss": 0.0237,
      "step": 8722
    },
    {
      "epoch": 1.9847554038680317,
      "grad_norm": 2.231175161679385,
      "learning_rate": 8.23904919287171e-07,
      "loss": 0.0975,
      "step": 8723
    },
    {
      "epoch": 1.9849829351535835,
      "grad_norm": 0.7504706692267575,
      "learning_rate": 8.2382021152338e-07,
      "loss": 0.0207,
      "step": 8724
    },
    {
      "epoch": 1.9852104664391352,
      "grad_norm": 1.2591709035268228,
      "learning_rate": 8.237354996960678e-07,
      "loss": 0.029,
      "step": 8725
    },
    {
      "epoch": 1.985437997724687,
      "grad_norm": 1.4421635680441485,
      "learning_rate": 8.23650783806966e-07,
      "loss": 0.0511,
      "step": 8726
    },
    {
      "epoch": 1.985665529010239,
      "grad_norm": 1.1065774105002377,
      "learning_rate": 8.235660638578061e-07,
      "loss": 0.0245,
      "step": 8727
    },
    {
      "epoch": 1.9858930602957907,
      "grad_norm": 0.9428658154154148,
      "learning_rate": 8.234813398503194e-07,
      "loss": 0.0186,
      "step": 8728
    },
    {
      "epoch": 1.9861205915813425,
      "grad_norm": 1.2117398012423959,
      "learning_rate": 8.233966117862378e-07,
      "loss": 0.0313,
      "step": 8729
    },
    {
      "epoch": 1.9863481228668942,
      "grad_norm": 0.8148093077827677,
      "learning_rate": 8.233118796672929e-07,
      "loss": 0.0332,
      "step": 8730
    },
    {
      "epoch": 1.986575654152446,
      "grad_norm": 1.3969644663564684,
      "learning_rate": 8.23227143495216e-07,
      "loss": 0.0206,
      "step": 8731
    },
    {
      "epoch": 1.9868031854379977,
      "grad_norm": 0.9233518202413549,
      "learning_rate": 8.231424032717395e-07,
      "loss": 0.015,
      "step": 8732
    },
    {
      "epoch": 1.9870307167235495,
      "grad_norm": 0.8688629986743365,
      "learning_rate": 8.230576589985951e-07,
      "loss": 0.0179,
      "step": 8733
    },
    {
      "epoch": 1.9872582480091012,
      "grad_norm": 1.7448892272436423,
      "learning_rate": 8.22972910677515e-07,
      "loss": 0.0364,
      "step": 8734
    },
    {
      "epoch": 1.987485779294653,
      "grad_norm": 0.9001399229517513,
      "learning_rate": 8.22888158310231e-07,
      "loss": 0.0189,
      "step": 8735
    },
    {
      "epoch": 1.9877133105802047,
      "grad_norm": 1.0684583399613707,
      "learning_rate": 8.228034018984757e-07,
      "loss": 0.0225,
      "step": 8736
    },
    {
      "epoch": 1.9879408418657567,
      "grad_norm": 1.137792725613461,
      "learning_rate": 8.227186414439812e-07,
      "loss": 0.0321,
      "step": 8737
    },
    {
      "epoch": 1.9881683731513085,
      "grad_norm": 0.4761613521913952,
      "learning_rate": 8.226338769484793e-07,
      "loss": 0.013,
      "step": 8738
    },
    {
      "epoch": 1.9883959044368602,
      "grad_norm": 1.300142438025343,
      "learning_rate": 8.225491084137032e-07,
      "loss": 0.0338,
      "step": 8739
    },
    {
      "epoch": 1.988623435722412,
      "grad_norm": 1.2848804726113578,
      "learning_rate": 8.22464335841385e-07,
      "loss": 0.0243,
      "step": 8740
    },
    {
      "epoch": 1.9888509670079637,
      "grad_norm": 0.9225434461136578,
      "learning_rate": 8.223795592332575e-07,
      "loss": 0.0165,
      "step": 8741
    },
    {
      "epoch": 1.9890784982935155,
      "grad_norm": 0.8346238495866183,
      "learning_rate": 8.222947785910534e-07,
      "loss": 0.0179,
      "step": 8742
    },
    {
      "epoch": 1.9893060295790672,
      "grad_norm": 0.9537674858060043,
      "learning_rate": 8.222099939165053e-07,
      "loss": 0.0217,
      "step": 8743
    },
    {
      "epoch": 1.989533560864619,
      "grad_norm": 1.545983536426448,
      "learning_rate": 8.221252052113461e-07,
      "loss": 0.0314,
      "step": 8744
    },
    {
      "epoch": 1.9897610921501707,
      "grad_norm": 1.2343476530361066,
      "learning_rate": 8.220404124773084e-07,
      "loss": 0.033,
      "step": 8745
    },
    {
      "epoch": 1.9899886234357225,
      "grad_norm": 1.0359221806193963,
      "learning_rate": 8.21955615716126e-07,
      "loss": 0.0308,
      "step": 8746
    },
    {
      "epoch": 1.9902161547212742,
      "grad_norm": 0.6159295211114494,
      "learning_rate": 8.218708149295312e-07,
      "loss": 0.0146,
      "step": 8747
    },
    {
      "epoch": 1.990443686006826,
      "grad_norm": 1.0030155089241657,
      "learning_rate": 8.217860101192578e-07,
      "loss": 0.0296,
      "step": 8748
    },
    {
      "epoch": 1.9906712172923777,
      "grad_norm": 1.2171836674531433,
      "learning_rate": 8.217012012870384e-07,
      "loss": 0.0332,
      "step": 8749
    },
    {
      "epoch": 1.9908987485779295,
      "grad_norm": 1.0134205901264104,
      "learning_rate": 8.21616388434607e-07,
      "loss": 0.0407,
      "step": 8750
    },
    {
      "epoch": 1.9911262798634812,
      "grad_norm": 1.2613091211202425,
      "learning_rate": 8.215315715636965e-07,
      "loss": 0.0271,
      "step": 8751
    },
    {
      "epoch": 1.991353811149033,
      "grad_norm": 0.9701824049292664,
      "learning_rate": 8.214467506760407e-07,
      "loss": 0.0188,
      "step": 8752
    },
    {
      "epoch": 1.9915813424345847,
      "grad_norm": 1.1531015931207416,
      "learning_rate": 8.213619257733729e-07,
      "loss": 0.0172,
      "step": 8753
    },
    {
      "epoch": 1.9918088737201365,
      "grad_norm": 1.223676074095881,
      "learning_rate": 8.212770968574274e-07,
      "loss": 0.0295,
      "step": 8754
    },
    {
      "epoch": 1.9920364050056882,
      "grad_norm": 0.9060703172422784,
      "learning_rate": 8.211922639299372e-07,
      "loss": 0.0197,
      "step": 8755
    },
    {
      "epoch": 1.99226393629124,
      "grad_norm": 1.1589371318449682,
      "learning_rate": 8.211074269926364e-07,
      "loss": 0.0319,
      "step": 8756
    },
    {
      "epoch": 1.9924914675767917,
      "grad_norm": 1.162550322553717,
      "learning_rate": 8.21022586047259e-07,
      "loss": 0.027,
      "step": 8757
    },
    {
      "epoch": 1.9927189988623435,
      "grad_norm": 1.483389942397735,
      "learning_rate": 8.209377410955388e-07,
      "loss": 0.058,
      "step": 8758
    },
    {
      "epoch": 1.9929465301478952,
      "grad_norm": 0.5989315482760323,
      "learning_rate": 8.208528921392101e-07,
      "loss": 0.0179,
      "step": 8759
    },
    {
      "epoch": 1.993174061433447,
      "grad_norm": 0.8224236601819628,
      "learning_rate": 8.207680391800071e-07,
      "loss": 0.0143,
      "step": 8760
    },
    {
      "epoch": 1.9934015927189987,
      "grad_norm": 1.5769681179365267,
      "learning_rate": 8.206831822196639e-07,
      "loss": 0.0389,
      "step": 8761
    },
    {
      "epoch": 1.9936291240045505,
      "grad_norm": 1.2642461984598112,
      "learning_rate": 8.205983212599147e-07,
      "loss": 0.0355,
      "step": 8762
    },
    {
      "epoch": 1.9938566552901023,
      "grad_norm": 0.8245254804736318,
      "learning_rate": 8.205134563024942e-07,
      "loss": 0.0138,
      "step": 8763
    },
    {
      "epoch": 1.994084186575654,
      "grad_norm": 1.027387146344138,
      "learning_rate": 8.204285873491366e-07,
      "loss": 0.019,
      "step": 8764
    },
    {
      "epoch": 1.9943117178612058,
      "grad_norm": 1.2011382159350898,
      "learning_rate": 8.203437144015766e-07,
      "loss": 0.026,
      "step": 8765
    },
    {
      "epoch": 1.9945392491467577,
      "grad_norm": 1.5164278338793358,
      "learning_rate": 8.202588374615489e-07,
      "loss": 0.0325,
      "step": 8766
    },
    {
      "epoch": 1.9947667804323095,
      "grad_norm": 1.9144837369745,
      "learning_rate": 8.201739565307881e-07,
      "loss": 0.0286,
      "step": 8767
    },
    {
      "epoch": 1.9949943117178612,
      "grad_norm": 1.1510459916405422,
      "learning_rate": 8.200890716110291e-07,
      "loss": 0.0348,
      "step": 8768
    },
    {
      "epoch": 1.995221843003413,
      "grad_norm": 1.0579695094334676,
      "learning_rate": 8.200041827040067e-07,
      "loss": 0.0333,
      "step": 8769
    },
    {
      "epoch": 1.9954493742889647,
      "grad_norm": 0.8783006470237726,
      "learning_rate": 8.19919289811456e-07,
      "loss": 0.0188,
      "step": 8770
    },
    {
      "epoch": 1.9956769055745165,
      "grad_norm": 1.2337651735307613,
      "learning_rate": 8.19834392935112e-07,
      "loss": 0.0351,
      "step": 8771
    },
    {
      "epoch": 1.9959044368600682,
      "grad_norm": 1.614485473246855,
      "learning_rate": 8.197494920767098e-07,
      "loss": 0.0377,
      "step": 8772
    },
    {
      "epoch": 1.99613196814562,
      "grad_norm": 1.1038033120164423,
      "learning_rate": 8.196645872379847e-07,
      "loss": 0.0364,
      "step": 8773
    },
    {
      "epoch": 1.9963594994311717,
      "grad_norm": 0.8343418233948011,
      "learning_rate": 8.19579678420672e-07,
      "loss": 0.0187,
      "step": 8774
    },
    {
      "epoch": 1.9965870307167235,
      "grad_norm": 1.8439932234882088,
      "learning_rate": 8.194947656265068e-07,
      "loss": 0.051,
      "step": 8775
    },
    {
      "epoch": 1.9968145620022755,
      "grad_norm": 0.9058793146661416,
      "learning_rate": 8.19409848857225e-07,
      "loss": 0.0203,
      "step": 8776
    },
    {
      "epoch": 1.9970420932878272,
      "grad_norm": 0.5079408054170178,
      "learning_rate": 8.193249281145618e-07,
      "loss": 0.0059,
      "step": 8777
    },
    {
      "epoch": 1.997269624573379,
      "grad_norm": 1.1010114111421319,
      "learning_rate": 8.19240003400253e-07,
      "loss": 0.0251,
      "step": 8778
    },
    {
      "epoch": 1.9974971558589307,
      "grad_norm": 1.23466091864004,
      "learning_rate": 8.191550747160343e-07,
      "loss": 0.0429,
      "step": 8779
    },
    {
      "epoch": 1.9977246871444825,
      "grad_norm": 1.0463717415347287,
      "learning_rate": 8.190701420636415e-07,
      "loss": 0.0301,
      "step": 8780
    },
    {
      "epoch": 1.9979522184300342,
      "grad_norm": 0.9950332785403146,
      "learning_rate": 8.189852054448104e-07,
      "loss": 0.0145,
      "step": 8781
    },
    {
      "epoch": 1.998179749715586,
      "grad_norm": 0.712340126659162,
      "learning_rate": 8.189002648612768e-07,
      "loss": 0.0136,
      "step": 8782
    },
    {
      "epoch": 1.9984072810011377,
      "grad_norm": 1.1073274746874324,
      "learning_rate": 8.188153203147769e-07,
      "loss": 0.0227,
      "step": 8783
    },
    {
      "epoch": 1.9986348122866895,
      "grad_norm": 1.964238477769453,
      "learning_rate": 8.18730371807047e-07,
      "loss": 0.041,
      "step": 8784
    },
    {
      "epoch": 1.9988623435722412,
      "grad_norm": 1.1408478266591537,
      "learning_rate": 8.18645419339823e-07,
      "loss": 0.0302,
      "step": 8785
    },
    {
      "epoch": 1.999089874857793,
      "grad_norm": 0.8521168973864249,
      "learning_rate": 8.185604629148413e-07,
      "loss": 0.0218,
      "step": 8786
    },
    {
      "epoch": 1.9993174061433447,
      "grad_norm": 1.1445024111773645,
      "learning_rate": 8.184755025338384e-07,
      "loss": 0.0376,
      "step": 8787
    },
    {
      "epoch": 1.9995449374288965,
      "grad_norm": 1.3001610144118216,
      "learning_rate": 8.183905381985503e-07,
      "loss": 0.0377,
      "step": 8788
    },
    {
      "epoch": 1.9997724687144482,
      "grad_norm": 1.2615958821120417,
      "learning_rate": 8.183055699107139e-07,
      "loss": 0.0319,
      "step": 8789
    },
    {
      "epoch": 2.0,
      "grad_norm": 0.6899444448695068,
      "learning_rate": 8.182205976720656e-07,
      "loss": 0.0119,
      "step": 8790
    },
    {
      "epoch": 2.0002275312855518,
      "grad_norm": 1.1709402263017634,
      "learning_rate": 8.181356214843422e-07,
      "loss": 0.0157,
      "step": 8791
    },
    {
      "epoch": 2.0004550625711035,
      "grad_norm": 0.5198887943317367,
      "learning_rate": 8.180506413492804e-07,
      "loss": 0.0071,
      "step": 8792
    },
    {
      "epoch": 2.0006825938566553,
      "grad_norm": 0.6737223248160417,
      "learning_rate": 8.17965657268617e-07,
      "loss": 0.0125,
      "step": 8793
    },
    {
      "epoch": 2.000910125142207,
      "grad_norm": 0.5483868719187942,
      "learning_rate": 8.178806692440891e-07,
      "loss": 0.0102,
      "step": 8794
    },
    {
      "epoch": 2.0011376564277588,
      "grad_norm": 0.5878907649994676,
      "learning_rate": 8.177956772774334e-07,
      "loss": 0.0081,
      "step": 8795
    },
    {
      "epoch": 2.0013651877133105,
      "grad_norm": 0.37933293187237677,
      "learning_rate": 8.177106813703872e-07,
      "loss": 0.0048,
      "step": 8796
    },
    {
      "epoch": 2.0015927189988623,
      "grad_norm": 0.4588421161999595,
      "learning_rate": 8.176256815246878e-07,
      "loss": 0.0047,
      "step": 8797
    },
    {
      "epoch": 2.001820250284414,
      "grad_norm": 0.30559275383648293,
      "learning_rate": 8.175406777420721e-07,
      "loss": 0.005,
      "step": 8798
    },
    {
      "epoch": 2.0020477815699658,
      "grad_norm": 0.5216872504697616,
      "learning_rate": 8.174556700242775e-07,
      "loss": 0.0116,
      "step": 8799
    },
    {
      "epoch": 2.0022753128555175,
      "grad_norm": 1.1268898419644335,
      "learning_rate": 8.173706583730414e-07,
      "loss": 0.0251,
      "step": 8800
    },
    {
      "epoch": 2.0025028441410693,
      "grad_norm": 0.46037136526859374,
      "learning_rate": 8.172856427901015e-07,
      "loss": 0.0116,
      "step": 8801
    },
    {
      "epoch": 2.002730375426621,
      "grad_norm": 0.724042564589398,
      "learning_rate": 8.17200623277195e-07,
      "loss": 0.0145,
      "step": 8802
    },
    {
      "epoch": 2.0029579067121728,
      "grad_norm": 0.5425702242840104,
      "learning_rate": 8.171155998360601e-07,
      "loss": 0.0145,
      "step": 8803
    },
    {
      "epoch": 2.0031854379977245,
      "grad_norm": 0.8078346383668281,
      "learning_rate": 8.170305724684341e-07,
      "loss": 0.019,
      "step": 8804
    },
    {
      "epoch": 2.0034129692832763,
      "grad_norm": 0.32320182253326146,
      "learning_rate": 8.169455411760547e-07,
      "loss": 0.0042,
      "step": 8805
    },
    {
      "epoch": 2.003640500568828,
      "grad_norm": 0.9468999792481988,
      "learning_rate": 8.168605059606601e-07,
      "loss": 0.027,
      "step": 8806
    },
    {
      "epoch": 2.00386803185438,
      "grad_norm": 0.504492378138807,
      "learning_rate": 8.167754668239883e-07,
      "loss": 0.0104,
      "step": 8807
    },
    {
      "epoch": 2.0040955631399315,
      "grad_norm": 0.5334252678232069,
      "learning_rate": 8.16690423767777e-07,
      "loss": 0.0168,
      "step": 8808
    },
    {
      "epoch": 2.0043230944254833,
      "grad_norm": 0.789594051061424,
      "learning_rate": 8.166053767937643e-07,
      "loss": 0.0209,
      "step": 8809
    },
    {
      "epoch": 2.0045506257110355,
      "grad_norm": 0.7868203944304248,
      "learning_rate": 8.165203259036888e-07,
      "loss": 0.0123,
      "step": 8810
    },
    {
      "epoch": 2.0047781569965872,
      "grad_norm": 0.93302247140185,
      "learning_rate": 8.164352710992887e-07,
      "loss": 0.0218,
      "step": 8811
    },
    {
      "epoch": 2.005005688282139,
      "grad_norm": 0.37733358623075536,
      "learning_rate": 8.163502123823021e-07,
      "loss": 0.0042,
      "step": 8812
    },
    {
      "epoch": 2.0052332195676907,
      "grad_norm": 0.9133370079801494,
      "learning_rate": 8.162651497544677e-07,
      "loss": 0.0132,
      "step": 8813
    },
    {
      "epoch": 2.0054607508532425,
      "grad_norm": 0.5344697172467202,
      "learning_rate": 8.161800832175239e-07,
      "loss": 0.0109,
      "step": 8814
    },
    {
      "epoch": 2.0056882821387942,
      "grad_norm": 0.686643323340727,
      "learning_rate": 8.160950127732093e-07,
      "loss": 0.0118,
      "step": 8815
    },
    {
      "epoch": 2.005915813424346,
      "grad_norm": 0.45484660984425673,
      "learning_rate": 8.160099384232625e-07,
      "loss": 0.0127,
      "step": 8816
    },
    {
      "epoch": 2.0061433447098977,
      "grad_norm": 0.4897150816329801,
      "learning_rate": 8.159248601694226e-07,
      "loss": 0.0083,
      "step": 8817
    },
    {
      "epoch": 2.0063708759954495,
      "grad_norm": 0.7942946127916877,
      "learning_rate": 8.158397780134281e-07,
      "loss": 0.0102,
      "step": 8818
    },
    {
      "epoch": 2.0065984072810013,
      "grad_norm": 0.9250277086108294,
      "learning_rate": 8.157546919570181e-07,
      "loss": 0.0085,
      "step": 8819
    },
    {
      "epoch": 2.006825938566553,
      "grad_norm": 1.8778433152181757,
      "learning_rate": 8.156696020019314e-07,
      "loss": 0.0244,
      "step": 8820
    },
    {
      "epoch": 2.0070534698521048,
      "grad_norm": 1.1993933794239524,
      "learning_rate": 8.155845081499074e-07,
      "loss": 0.0247,
      "step": 8821
    },
    {
      "epoch": 2.0072810011376565,
      "grad_norm": 0.8085508317141936,
      "learning_rate": 8.154994104026849e-07,
      "loss": 0.0076,
      "step": 8822
    },
    {
      "epoch": 2.0075085324232083,
      "grad_norm": 0.8970209300212904,
      "learning_rate": 8.154143087620035e-07,
      "loss": 0.0121,
      "step": 8823
    },
    {
      "epoch": 2.00773606370876,
      "grad_norm": 0.3197563006884184,
      "learning_rate": 8.153292032296025e-07,
      "loss": 0.0036,
      "step": 8824
    },
    {
      "epoch": 2.0079635949943118,
      "grad_norm": 0.9785745980402671,
      "learning_rate": 8.152440938072208e-07,
      "loss": 0.0297,
      "step": 8825
    },
    {
      "epoch": 2.0081911262798635,
      "grad_norm": 0.5102544551579844,
      "learning_rate": 8.151589804965984e-07,
      "loss": 0.0028,
      "step": 8826
    },
    {
      "epoch": 2.0084186575654153,
      "grad_norm": 0.7165375825214081,
      "learning_rate": 8.150738632994748e-07,
      "loss": 0.0091,
      "step": 8827
    },
    {
      "epoch": 2.008646188850967,
      "grad_norm": 1.7463034408814433,
      "learning_rate": 8.149887422175895e-07,
      "loss": 0.019,
      "step": 8828
    },
    {
      "epoch": 2.0088737201365188,
      "grad_norm": 0.5830956601446018,
      "learning_rate": 8.149036172526821e-07,
      "loss": 0.0134,
      "step": 8829
    },
    {
      "epoch": 2.0091012514220705,
      "grad_norm": 0.4365691139998193,
      "learning_rate": 8.148184884064928e-07,
      "loss": 0.0058,
      "step": 8830
    },
    {
      "epoch": 2.0093287827076223,
      "grad_norm": 0.4129790478139653,
      "learning_rate": 8.14733355680761e-07,
      "loss": 0.0063,
      "step": 8831
    },
    {
      "epoch": 2.009556313993174,
      "grad_norm": 0.9807885758736534,
      "learning_rate": 8.146482190772271e-07,
      "loss": 0.0191,
      "step": 8832
    },
    {
      "epoch": 2.0097838452787258,
      "grad_norm": 0.6550137634160845,
      "learning_rate": 8.145630785976307e-07,
      "loss": 0.0109,
      "step": 8833
    },
    {
      "epoch": 2.0100113765642775,
      "grad_norm": 0.1815930144575576,
      "learning_rate": 8.144779342437123e-07,
      "loss": 0.0014,
      "step": 8834
    },
    {
      "epoch": 2.0102389078498293,
      "grad_norm": 0.30779399114598865,
      "learning_rate": 8.143927860172118e-07,
      "loss": 0.0032,
      "step": 8835
    },
    {
      "epoch": 2.010466439135381,
      "grad_norm": 1.3781483552111702,
      "learning_rate": 8.143076339198698e-07,
      "loss": 0.0154,
      "step": 8836
    },
    {
      "epoch": 2.010693970420933,
      "grad_norm": 0.5163766547449539,
      "learning_rate": 8.142224779534263e-07,
      "loss": 0.0046,
      "step": 8837
    },
    {
      "epoch": 2.0109215017064845,
      "grad_norm": 0.14466268063999152,
      "learning_rate": 8.14137318119622e-07,
      "loss": 0.0012,
      "step": 8838
    },
    {
      "epoch": 2.0111490329920363,
      "grad_norm": 0.5235772934791572,
      "learning_rate": 8.140521544201972e-07,
      "loss": 0.0076,
      "step": 8839
    },
    {
      "epoch": 2.011376564277588,
      "grad_norm": 0.5237206847198979,
      "learning_rate": 8.139669868568927e-07,
      "loss": 0.0079,
      "step": 8840
    },
    {
      "epoch": 2.01160409556314,
      "grad_norm": 0.6991686108225662,
      "learning_rate": 8.13881815431449e-07,
      "loss": 0.0066,
      "step": 8841
    },
    {
      "epoch": 2.0118316268486915,
      "grad_norm": 0.6117733013500027,
      "learning_rate": 8.13796640145607e-07,
      "loss": 0.017,
      "step": 8842
    },
    {
      "epoch": 2.0120591581342433,
      "grad_norm": 0.6344895088061999,
      "learning_rate": 8.137114610011074e-07,
      "loss": 0.0093,
      "step": 8843
    },
    {
      "epoch": 2.012286689419795,
      "grad_norm": 0.3577293634720585,
      "learning_rate": 8.136262779996912e-07,
      "loss": 0.0041,
      "step": 8844
    },
    {
      "epoch": 2.012514220705347,
      "grad_norm": 1.2364309637360462,
      "learning_rate": 8.135410911430992e-07,
      "loss": 0.0251,
      "step": 8845
    },
    {
      "epoch": 2.0127417519908986,
      "grad_norm": 1.2064652484875504,
      "learning_rate": 8.134559004330725e-07,
      "loss": 0.0126,
      "step": 8846
    },
    {
      "epoch": 2.0129692832764503,
      "grad_norm": 1.055012807579558,
      "learning_rate": 8.133707058713525e-07,
      "loss": 0.006,
      "step": 8847
    },
    {
      "epoch": 2.013196814562002,
      "grad_norm": 0.44452896500004035,
      "learning_rate": 8.132855074596803e-07,
      "loss": 0.0036,
      "step": 8848
    },
    {
      "epoch": 2.0134243458475543,
      "grad_norm": 0.5888926114458435,
      "learning_rate": 8.132003051997972e-07,
      "loss": 0.0096,
      "step": 8849
    },
    {
      "epoch": 2.013651877133106,
      "grad_norm": 0.8079902367360255,
      "learning_rate": 8.131150990934445e-07,
      "loss": 0.0249,
      "step": 8850
    },
    {
      "epoch": 2.0138794084186578,
      "grad_norm": 0.8703704206567516,
      "learning_rate": 8.130298891423636e-07,
      "loss": 0.0223,
      "step": 8851
    },
    {
      "epoch": 2.0141069397042095,
      "grad_norm": 0.5693115424305861,
      "learning_rate": 8.12944675348296e-07,
      "loss": 0.0081,
      "step": 8852
    },
    {
      "epoch": 2.0143344709897613,
      "grad_norm": 0.8080455402833692,
      "learning_rate": 8.128594577129836e-07,
      "loss": 0.0121,
      "step": 8853
    },
    {
      "epoch": 2.014562002275313,
      "grad_norm": 0.5479120256962474,
      "learning_rate": 8.12774236238168e-07,
      "loss": 0.0088,
      "step": 8854
    },
    {
      "epoch": 2.0147895335608648,
      "grad_norm": 0.5649292330396924,
      "learning_rate": 8.126890109255908e-07,
      "loss": 0.0099,
      "step": 8855
    },
    {
      "epoch": 2.0150170648464165,
      "grad_norm": 0.38594711490407335,
      "learning_rate": 8.126037817769939e-07,
      "loss": 0.0084,
      "step": 8856
    },
    {
      "epoch": 2.0152445961319683,
      "grad_norm": 0.7510993902450168,
      "learning_rate": 8.125185487941195e-07,
      "loss": 0.0152,
      "step": 8857
    },
    {
      "epoch": 2.01547212741752,
      "grad_norm": 0.5835031624844149,
      "learning_rate": 8.124333119787093e-07,
      "loss": 0.0064,
      "step": 8858
    },
    {
      "epoch": 2.0156996587030718,
      "grad_norm": 0.5597660181617066,
      "learning_rate": 8.123480713325053e-07,
      "loss": 0.0087,
      "step": 8859
    },
    {
      "epoch": 2.0159271899886235,
      "grad_norm": 0.5247004899887854,
      "learning_rate": 8.122628268572499e-07,
      "loss": 0.0128,
      "step": 8860
    },
    {
      "epoch": 2.0161547212741753,
      "grad_norm": 0.566052373412455,
      "learning_rate": 8.121775785546855e-07,
      "loss": 0.0058,
      "step": 8861
    },
    {
      "epoch": 2.016382252559727,
      "grad_norm": 1.0307604808475657,
      "learning_rate": 8.120923264265539e-07,
      "loss": 0.0131,
      "step": 8862
    },
    {
      "epoch": 2.016609783845279,
      "grad_norm": 1.1342668705919563,
      "learning_rate": 8.120070704745979e-07,
      "loss": 0.012,
      "step": 8863
    },
    {
      "epoch": 2.0168373151308305,
      "grad_norm": 0.7561026503729099,
      "learning_rate": 8.1192181070056e-07,
      "loss": 0.009,
      "step": 8864
    },
    {
      "epoch": 2.0170648464163823,
      "grad_norm": 0.6203228474509735,
      "learning_rate": 8.118365471061825e-07,
      "loss": 0.008,
      "step": 8865
    },
    {
      "epoch": 2.017292377701934,
      "grad_norm": 0.22624176637311427,
      "learning_rate": 8.117512796932079e-07,
      "loss": 0.0016,
      "step": 8866
    },
    {
      "epoch": 2.017519908987486,
      "grad_norm": 0.5131236339857828,
      "learning_rate": 8.116660084633796e-07,
      "loss": 0.0039,
      "step": 8867
    },
    {
      "epoch": 2.0177474402730375,
      "grad_norm": 1.3348847929301328,
      "learning_rate": 8.115807334184398e-07,
      "loss": 0.0149,
      "step": 8868
    },
    {
      "epoch": 2.0179749715585893,
      "grad_norm": 1.1999017764200335,
      "learning_rate": 8.114954545601314e-07,
      "loss": 0.0109,
      "step": 8869
    },
    {
      "epoch": 2.018202502844141,
      "grad_norm": 0.5178327281568412,
      "learning_rate": 8.114101718901976e-07,
      "loss": 0.0032,
      "step": 8870
    },
    {
      "epoch": 2.018430034129693,
      "grad_norm": 1.1828292639159577,
      "learning_rate": 8.113248854103811e-07,
      "loss": 0.0174,
      "step": 8871
    },
    {
      "epoch": 2.0186575654152445,
      "grad_norm": 0.5401939506471521,
      "learning_rate": 8.112395951224254e-07,
      "loss": 0.007,
      "step": 8872
    },
    {
      "epoch": 2.0188850967007963,
      "grad_norm": 0.9142563228439178,
      "learning_rate": 8.111543010280733e-07,
      "loss": 0.0097,
      "step": 8873
    },
    {
      "epoch": 2.019112627986348,
      "grad_norm": 0.8487407604632405,
      "learning_rate": 8.110690031290683e-07,
      "loss": 0.0083,
      "step": 8874
    },
    {
      "epoch": 2.0193401592719,
      "grad_norm": 0.4147069374933176,
      "learning_rate": 8.109837014271536e-07,
      "loss": 0.0048,
      "step": 8875
    },
    {
      "epoch": 2.0195676905574516,
      "grad_norm": 0.9484556445049667,
      "learning_rate": 8.108983959240726e-07,
      "loss": 0.0157,
      "step": 8876
    },
    {
      "epoch": 2.0197952218430033,
      "grad_norm": 0.9879486396757432,
      "learning_rate": 8.108130866215689e-07,
      "loss": 0.0153,
      "step": 8877
    },
    {
      "epoch": 2.020022753128555,
      "grad_norm": 1.266706637316005,
      "learning_rate": 8.107277735213861e-07,
      "loss": 0.0111,
      "step": 8878
    },
    {
      "epoch": 2.020250284414107,
      "grad_norm": 1.152593355345514,
      "learning_rate": 8.106424566252675e-07,
      "loss": 0.0121,
      "step": 8879
    },
    {
      "epoch": 2.0204778156996586,
      "grad_norm": 31.972085462973816,
      "learning_rate": 8.105571359349575e-07,
      "loss": 0.1801,
      "step": 8880
    },
    {
      "epoch": 2.0207053469852103,
      "grad_norm": 0.4667549780604784,
      "learning_rate": 8.104718114521993e-07,
      "loss": 0.0066,
      "step": 8881
    },
    {
      "epoch": 2.020932878270762,
      "grad_norm": 0.45424657076751523,
      "learning_rate": 8.103864831787367e-07,
      "loss": 0.0072,
      "step": 8882
    },
    {
      "epoch": 2.021160409556314,
      "grad_norm": 1.0788646098304586,
      "learning_rate": 8.103011511163141e-07,
      "loss": 0.015,
      "step": 8883
    },
    {
      "epoch": 2.0213879408418656,
      "grad_norm": 0.8152685446198583,
      "learning_rate": 8.102158152666753e-07,
      "loss": 0.0178,
      "step": 8884
    },
    {
      "epoch": 2.0216154721274173,
      "grad_norm": 1.0690660609265563,
      "learning_rate": 8.101304756315645e-07,
      "loss": 0.0103,
      "step": 8885
    },
    {
      "epoch": 2.021843003412969,
      "grad_norm": 1.21599202432248,
      "learning_rate": 8.100451322127257e-07,
      "loss": 0.0176,
      "step": 8886
    },
    {
      "epoch": 2.022070534698521,
      "grad_norm": 1.285998257222507,
      "learning_rate": 8.099597850119035e-07,
      "loss": 0.0174,
      "step": 8887
    },
    {
      "epoch": 2.022298065984073,
      "grad_norm": 0.2775345886721884,
      "learning_rate": 8.098744340308419e-07,
      "loss": 0.0095,
      "step": 8888
    },
    {
      "epoch": 2.0225255972696248,
      "grad_norm": 1.1182380948561206,
      "learning_rate": 8.097890792712853e-07,
      "loss": 0.0151,
      "step": 8889
    },
    {
      "epoch": 2.0227531285551765,
      "grad_norm": 0.9201126822233208,
      "learning_rate": 8.097037207349785e-07,
      "loss": 0.0082,
      "step": 8890
    },
    {
      "epoch": 2.0229806598407283,
      "grad_norm": 0.5416948586082189,
      "learning_rate": 8.096183584236659e-07,
      "loss": 0.0057,
      "step": 8891
    },
    {
      "epoch": 2.02320819112628,
      "grad_norm": 0.855302718014359,
      "learning_rate": 8.095329923390924e-07,
      "loss": 0.0094,
      "step": 8892
    },
    {
      "epoch": 2.023435722411832,
      "grad_norm": 0.38112583835583014,
      "learning_rate": 8.094476224830022e-07,
      "loss": 0.0042,
      "step": 8893
    },
    {
      "epoch": 2.0236632536973835,
      "grad_norm": 0.5106426753539933,
      "learning_rate": 8.093622488571405e-07,
      "loss": 0.008,
      "step": 8894
    },
    {
      "epoch": 2.0238907849829353,
      "grad_norm": 0.7454048076191544,
      "learning_rate": 8.09276871463252e-07,
      "loss": 0.0146,
      "step": 8895
    },
    {
      "epoch": 2.024118316268487,
      "grad_norm": 0.6556405334233815,
      "learning_rate": 8.091914903030818e-07,
      "loss": 0.0026,
      "step": 8896
    },
    {
      "epoch": 2.024345847554039,
      "grad_norm": 1.1409480911971517,
      "learning_rate": 8.091061053783748e-07,
      "loss": 0.0171,
      "step": 8897
    },
    {
      "epoch": 2.0245733788395905,
      "grad_norm": 1.2140550836825563,
      "learning_rate": 8.090207166908763e-07,
      "loss": 0.0197,
      "step": 8898
    },
    {
      "epoch": 2.0248009101251423,
      "grad_norm": 0.6006794941935975,
      "learning_rate": 8.089353242423313e-07,
      "loss": 0.0069,
      "step": 8899
    },
    {
      "epoch": 2.025028441410694,
      "grad_norm": 0.42400098109544904,
      "learning_rate": 8.088499280344851e-07,
      "loss": 0.0042,
      "step": 8900
    },
    {
      "epoch": 2.025255972696246,
      "grad_norm": 1.486151136146108,
      "learning_rate": 8.087645280690831e-07,
      "loss": 0.0134,
      "step": 8901
    },
    {
      "epoch": 2.0254835039817976,
      "grad_norm": 0.4257547909991054,
      "learning_rate": 8.086791243478709e-07,
      "loss": 0.0062,
      "step": 8902
    },
    {
      "epoch": 2.0257110352673493,
      "grad_norm": 1.6558564991678728,
      "learning_rate": 8.085937168725934e-07,
      "loss": 0.0347,
      "step": 8903
    },
    {
      "epoch": 2.025938566552901,
      "grad_norm": 0.47041782103413543,
      "learning_rate": 8.085083056449968e-07,
      "loss": 0.007,
      "step": 8904
    },
    {
      "epoch": 2.026166097838453,
      "grad_norm": 0.6524053506495247,
      "learning_rate": 8.084228906668267e-07,
      "loss": 0.0108,
      "step": 8905
    },
    {
      "epoch": 2.0263936291240046,
      "grad_norm": 0.9388216844035533,
      "learning_rate": 8.083374719398282e-07,
      "loss": 0.0154,
      "step": 8906
    },
    {
      "epoch": 2.0266211604095563,
      "grad_norm": 0.5932063789932374,
      "learning_rate": 8.082520494657478e-07,
      "loss": 0.0131,
      "step": 8907
    },
    {
      "epoch": 2.026848691695108,
      "grad_norm": 1.7723010612424621,
      "learning_rate": 8.08166623246331e-07,
      "loss": 0.0551,
      "step": 8908
    },
    {
      "epoch": 2.02707622298066,
      "grad_norm": 1.112698680144584,
      "learning_rate": 8.08081193283324e-07,
      "loss": 0.0219,
      "step": 8909
    },
    {
      "epoch": 2.0273037542662116,
      "grad_norm": 1.072970555370149,
      "learning_rate": 8.079957595784727e-07,
      "loss": 0.0103,
      "step": 8910
    },
    {
      "epoch": 2.0275312855517633,
      "grad_norm": 0.5308079651026945,
      "learning_rate": 8.07910322133523e-07,
      "loss": 0.0108,
      "step": 8911
    },
    {
      "epoch": 2.027758816837315,
      "grad_norm": 0.5035444843591139,
      "learning_rate": 8.078248809502215e-07,
      "loss": 0.0063,
      "step": 8912
    },
    {
      "epoch": 2.027986348122867,
      "grad_norm": 1.6508397127082461,
      "learning_rate": 8.077394360303143e-07,
      "loss": 0.0158,
      "step": 8913
    },
    {
      "epoch": 2.0282138794084186,
      "grad_norm": 0.47561982426348653,
      "learning_rate": 8.076539873755476e-07,
      "loss": 0.0104,
      "step": 8914
    },
    {
      "epoch": 2.0284414106939703,
      "grad_norm": 0.6109694524570578,
      "learning_rate": 8.07568534987668e-07,
      "loss": 0.0055,
      "step": 8915
    },
    {
      "epoch": 2.028668941979522,
      "grad_norm": 1.0892654029210274,
      "learning_rate": 8.074830788684218e-07,
      "loss": 0.0189,
      "step": 8916
    },
    {
      "epoch": 2.028896473265074,
      "grad_norm": 1.2435519587854957,
      "learning_rate": 8.073976190195557e-07,
      "loss": 0.0252,
      "step": 8917
    },
    {
      "epoch": 2.0291240045506256,
      "grad_norm": 1.181853187713523,
      "learning_rate": 8.073121554428165e-07,
      "loss": 0.017,
      "step": 8918
    },
    {
      "epoch": 2.0293515358361773,
      "grad_norm": 0.2538018454491145,
      "learning_rate": 8.072266881399504e-07,
      "loss": 0.0021,
      "step": 8919
    },
    {
      "epoch": 2.029579067121729,
      "grad_norm": 1.1916511430599437,
      "learning_rate": 8.071412171127047e-07,
      "loss": 0.0184,
      "step": 8920
    },
    {
      "epoch": 2.029806598407281,
      "grad_norm": 1.1753077172107904,
      "learning_rate": 8.070557423628262e-07,
      "loss": 0.0122,
      "step": 8921
    },
    {
      "epoch": 2.0300341296928326,
      "grad_norm": 0.33606691616182494,
      "learning_rate": 8.069702638920615e-07,
      "loss": 0.003,
      "step": 8922
    },
    {
      "epoch": 2.0302616609783843,
      "grad_norm": 1.248770958358262,
      "learning_rate": 8.06884781702158e-07,
      "loss": 0.0054,
      "step": 8923
    },
    {
      "epoch": 2.030489192263936,
      "grad_norm": 0.6781973886658158,
      "learning_rate": 8.067992957948628e-07,
      "loss": 0.0124,
      "step": 8924
    },
    {
      "epoch": 2.030716723549488,
      "grad_norm": 0.3502584562776884,
      "learning_rate": 8.067138061719227e-07,
      "loss": 0.0037,
      "step": 8925
    },
    {
      "epoch": 2.03094425483504,
      "grad_norm": 0.718987233926625,
      "learning_rate": 8.066283128350854e-07,
      "loss": 0.008,
      "step": 8926
    },
    {
      "epoch": 2.031171786120592,
      "grad_norm": 0.8505287305354738,
      "learning_rate": 8.065428157860978e-07,
      "loss": 0.0085,
      "step": 8927
    },
    {
      "epoch": 2.0313993174061435,
      "grad_norm": 0.4810527989255863,
      "learning_rate": 8.064573150267077e-07,
      "loss": 0.0048,
      "step": 8928
    },
    {
      "epoch": 2.0316268486916953,
      "grad_norm": 1.5732589603701084,
      "learning_rate": 8.063718105586624e-07,
      "loss": 0.02,
      "step": 8929
    },
    {
      "epoch": 2.031854379977247,
      "grad_norm": 1.3526673760378833,
      "learning_rate": 8.062863023837093e-07,
      "loss": 0.0262,
      "step": 8930
    },
    {
      "epoch": 2.032081911262799,
      "grad_norm": 0.9043005415960484,
      "learning_rate": 8.062007905035965e-07,
      "loss": 0.0043,
      "step": 8931
    },
    {
      "epoch": 2.0323094425483506,
      "grad_norm": 0.9672668132294734,
      "learning_rate": 8.061152749200713e-07,
      "loss": 0.0122,
      "step": 8932
    },
    {
      "epoch": 2.0325369738339023,
      "grad_norm": 1.4872897063402382,
      "learning_rate": 8.060297556348812e-07,
      "loss": 0.0224,
      "step": 8933
    },
    {
      "epoch": 2.032764505119454,
      "grad_norm": 1.2655166086852356,
      "learning_rate": 8.059442326497748e-07,
      "loss": 0.008,
      "step": 8934
    },
    {
      "epoch": 2.032992036405006,
      "grad_norm": 0.3450765270112137,
      "learning_rate": 8.058587059664996e-07,
      "loss": 0.0021,
      "step": 8935
    },
    {
      "epoch": 2.0332195676905576,
      "grad_norm": 0.8752950335447572,
      "learning_rate": 8.057731755868036e-07,
      "loss": 0.0236,
      "step": 8936
    },
    {
      "epoch": 2.0334470989761093,
      "grad_norm": 0.3176354327883559,
      "learning_rate": 8.056876415124352e-07,
      "loss": 0.0037,
      "step": 8937
    },
    {
      "epoch": 2.033674630261661,
      "grad_norm": 0.5461737826476964,
      "learning_rate": 8.056021037451422e-07,
      "loss": 0.0067,
      "step": 8938
    },
    {
      "epoch": 2.033902161547213,
      "grad_norm": 0.17133198077613082,
      "learning_rate": 8.055165622866726e-07,
      "loss": 0.0014,
      "step": 8939
    },
    {
      "epoch": 2.0341296928327646,
      "grad_norm": 0.6883875443062736,
      "learning_rate": 8.054310171387756e-07,
      "loss": 0.0081,
      "step": 8940
    },
    {
      "epoch": 2.0343572241183163,
      "grad_norm": 0.9648755007275786,
      "learning_rate": 8.053454683031987e-07,
      "loss": 0.0071,
      "step": 8941
    },
    {
      "epoch": 2.034584755403868,
      "grad_norm": 0.8409550490400549,
      "learning_rate": 8.052599157816908e-07,
      "loss": 0.0101,
      "step": 8942
    },
    {
      "epoch": 2.03481228668942,
      "grad_norm": 0.7630549826940595,
      "learning_rate": 8.051743595760005e-07,
      "loss": 0.0102,
      "step": 8943
    },
    {
      "epoch": 2.0350398179749716,
      "grad_norm": 2.6379153822624373,
      "learning_rate": 8.050887996878761e-07,
      "loss": 0.0134,
      "step": 8944
    },
    {
      "epoch": 2.0352673492605233,
      "grad_norm": 0.8411361959733792,
      "learning_rate": 8.050032361190666e-07,
      "loss": 0.0132,
      "step": 8945
    },
    {
      "epoch": 2.035494880546075,
      "grad_norm": 0.4110485826017932,
      "learning_rate": 8.049176688713203e-07,
      "loss": 0.004,
      "step": 8946
    },
    {
      "epoch": 2.035722411831627,
      "grad_norm": 2.8164652778212638,
      "learning_rate": 8.048320979463867e-07,
      "loss": 0.0268,
      "step": 8947
    },
    {
      "epoch": 2.0359499431171786,
      "grad_norm": 1.1851892458441966,
      "learning_rate": 8.047465233460141e-07,
      "loss": 0.0285,
      "step": 8948
    },
    {
      "epoch": 2.0361774744027303,
      "grad_norm": 0.7574982419765178,
      "learning_rate": 8.04660945071952e-07,
      "loss": 0.0069,
      "step": 8949
    },
    {
      "epoch": 2.036405005688282,
      "grad_norm": 0.44412736517823015,
      "learning_rate": 8.045753631259491e-07,
      "loss": 0.004,
      "step": 8950
    },
    {
      "epoch": 2.036632536973834,
      "grad_norm": 1.117920730656277,
      "learning_rate": 8.044897775097548e-07,
      "loss": 0.0075,
      "step": 8951
    },
    {
      "epoch": 2.0368600682593856,
      "grad_norm": 0.3887980095836281,
      "learning_rate": 8.04404188225118e-07,
      "loss": 0.0022,
      "step": 8952
    },
    {
      "epoch": 2.0370875995449373,
      "grad_norm": 0.591381612688169,
      "learning_rate": 8.043185952737881e-07,
      "loss": 0.0089,
      "step": 8953
    },
    {
      "epoch": 2.037315130830489,
      "grad_norm": 1.1988230813792133,
      "learning_rate": 8.042329986575145e-07,
      "loss": 0.0046,
      "step": 8954
    },
    {
      "epoch": 2.037542662116041,
      "grad_norm": 0.8524240083915456,
      "learning_rate": 8.041473983780467e-07,
      "loss": 0.011,
      "step": 8955
    },
    {
      "epoch": 2.0377701934015926,
      "grad_norm": 0.45742066554856414,
      "learning_rate": 8.040617944371343e-07,
      "loss": 0.0068,
      "step": 8956
    },
    {
      "epoch": 2.0379977246871444,
      "grad_norm": 0.7032718802225679,
      "learning_rate": 8.039761868365267e-07,
      "loss": 0.0067,
      "step": 8957
    },
    {
      "epoch": 2.038225255972696,
      "grad_norm": 0.46332231714977534,
      "learning_rate": 8.038905755779737e-07,
      "loss": 0.0028,
      "step": 8958
    },
    {
      "epoch": 2.038452787258248,
      "grad_norm": 0.6953451418720955,
      "learning_rate": 8.038049606632248e-07,
      "loss": 0.0063,
      "step": 8959
    },
    {
      "epoch": 2.0386803185437996,
      "grad_norm": 1.5095349839763461,
      "learning_rate": 8.0371934209403e-07,
      "loss": 0.0062,
      "step": 8960
    },
    {
      "epoch": 2.0389078498293514,
      "grad_norm": 0.38559837742698944,
      "learning_rate": 8.036337198721392e-07,
      "loss": 0.0047,
      "step": 8961
    },
    {
      "epoch": 2.039135381114903,
      "grad_norm": 0.7901667788607518,
      "learning_rate": 8.035480939993025e-07,
      "loss": 0.0089,
      "step": 8962
    },
    {
      "epoch": 2.039362912400455,
      "grad_norm": 0.35187461070550274,
      "learning_rate": 8.034624644772694e-07,
      "loss": 0.0039,
      "step": 8963
    },
    {
      "epoch": 2.0395904436860066,
      "grad_norm": 0.4561879491037348,
      "learning_rate": 8.033768313077905e-07,
      "loss": 0.0029,
      "step": 8964
    },
    {
      "epoch": 2.039817974971559,
      "grad_norm": 0.9242655687617418,
      "learning_rate": 8.03291194492616e-07,
      "loss": 0.0125,
      "step": 8965
    },
    {
      "epoch": 2.0400455062571106,
      "grad_norm": 0.5578814606863747,
      "learning_rate": 8.032055540334958e-07,
      "loss": 0.0122,
      "step": 8966
    },
    {
      "epoch": 2.0402730375426623,
      "grad_norm": 0.3959293258283401,
      "learning_rate": 8.031199099321804e-07,
      "loss": 0.0044,
      "step": 8967
    },
    {
      "epoch": 2.040500568828214,
      "grad_norm": 0.7511332134836817,
      "learning_rate": 8.030342621904205e-07,
      "loss": 0.0104,
      "step": 8968
    },
    {
      "epoch": 2.040728100113766,
      "grad_norm": 0.7723017158992114,
      "learning_rate": 8.029486108099663e-07,
      "loss": 0.0092,
      "step": 8969
    },
    {
      "epoch": 2.0409556313993176,
      "grad_norm": 0.41211418590091653,
      "learning_rate": 8.028629557925683e-07,
      "loss": 0.0034,
      "step": 8970
    },
    {
      "epoch": 2.0411831626848693,
      "grad_norm": 1.0666589983958474,
      "learning_rate": 8.027772971399773e-07,
      "loss": 0.0172,
      "step": 8971
    },
    {
      "epoch": 2.041410693970421,
      "grad_norm": 0.3495874852861486,
      "learning_rate": 8.026916348539438e-07,
      "loss": 0.0027,
      "step": 8972
    },
    {
      "epoch": 2.041638225255973,
      "grad_norm": 0.7646452546610977,
      "learning_rate": 8.026059689362186e-07,
      "loss": 0.0074,
      "step": 8973
    },
    {
      "epoch": 2.0418657565415246,
      "grad_norm": 0.5174281054023835,
      "learning_rate": 8.025202993885528e-07,
      "loss": 0.0094,
      "step": 8974
    },
    {
      "epoch": 2.0420932878270763,
      "grad_norm": 0.3397256143456613,
      "learning_rate": 8.024346262126976e-07,
      "loss": 0.0033,
      "step": 8975
    },
    {
      "epoch": 2.042320819112628,
      "grad_norm": 0.7890970235106726,
      "learning_rate": 8.02348949410403e-07,
      "loss": 0.0107,
      "step": 8976
    },
    {
      "epoch": 2.04254835039818,
      "grad_norm": 0.5797109965203312,
      "learning_rate": 8.022632689834209e-07,
      "loss": 0.0085,
      "step": 8977
    },
    {
      "epoch": 2.0427758816837316,
      "grad_norm": 0.5763357319854734,
      "learning_rate": 8.021775849335023e-07,
      "loss": 0.0071,
      "step": 8978
    },
    {
      "epoch": 2.0430034129692833,
      "grad_norm": 0.3898751609908676,
      "learning_rate": 8.020918972623983e-07,
      "loss": 0.003,
      "step": 8979
    },
    {
      "epoch": 2.043230944254835,
      "grad_norm": 0.6442895083097052,
      "learning_rate": 8.0200620597186e-07,
      "loss": 0.0052,
      "step": 8980
    },
    {
      "epoch": 2.043458475540387,
      "grad_norm": 0.4535279724862919,
      "learning_rate": 8.019205110636394e-07,
      "loss": 0.0037,
      "step": 8981
    },
    {
      "epoch": 2.0436860068259386,
      "grad_norm": 0.928510455854613,
      "learning_rate": 8.018348125394874e-07,
      "loss": 0.018,
      "step": 8982
    },
    {
      "epoch": 2.0439135381114903,
      "grad_norm": 0.7955962332228361,
      "learning_rate": 8.017491104011557e-07,
      "loss": 0.0051,
      "step": 8983
    },
    {
      "epoch": 2.044141069397042,
      "grad_norm": 1.0844193492460144,
      "learning_rate": 8.016634046503958e-07,
      "loss": 0.0077,
      "step": 8984
    },
    {
      "epoch": 2.044368600682594,
      "grad_norm": 0.48679526588275923,
      "learning_rate": 8.015776952889596e-07,
      "loss": 0.0062,
      "step": 8985
    },
    {
      "epoch": 2.0445961319681456,
      "grad_norm": 0.8517562832191604,
      "learning_rate": 8.014919823185987e-07,
      "loss": 0.0082,
      "step": 8986
    },
    {
      "epoch": 2.0448236632536974,
      "grad_norm": 0.3585852442137168,
      "learning_rate": 8.014062657410648e-07,
      "loss": 0.0027,
      "step": 8987
    },
    {
      "epoch": 2.045051194539249,
      "grad_norm": 0.9666705604775362,
      "learning_rate": 8.0132054555811e-07,
      "loss": 0.008,
      "step": 8988
    },
    {
      "epoch": 2.045278725824801,
      "grad_norm": 0.2820816822059884,
      "learning_rate": 8.012348217714861e-07,
      "loss": 0.0016,
      "step": 8989
    },
    {
      "epoch": 2.0455062571103526,
      "grad_norm": 0.6974292898679078,
      "learning_rate": 8.011490943829451e-07,
      "loss": 0.0103,
      "step": 8990
    },
    {
      "epoch": 2.0457337883959044,
      "grad_norm": 1.480846150230487,
      "learning_rate": 8.010633633942394e-07,
      "loss": 0.0239,
      "step": 8991
    },
    {
      "epoch": 2.045961319681456,
      "grad_norm": 0.8994891189070041,
      "learning_rate": 8.00977628807121e-07,
      "loss": 0.0125,
      "step": 8992
    },
    {
      "epoch": 2.046188850967008,
      "grad_norm": 0.36008186965231853,
      "learning_rate": 8.008918906233421e-07,
      "loss": 0.0025,
      "step": 8993
    },
    {
      "epoch": 2.0464163822525596,
      "grad_norm": 0.8791914697678171,
      "learning_rate": 8.00806148844655e-07,
      "loss": 0.0069,
      "step": 8994
    },
    {
      "epoch": 2.0466439135381114,
      "grad_norm": 3.43520789541217,
      "learning_rate": 8.007204034728123e-07,
      "loss": 0.0131,
      "step": 8995
    },
    {
      "epoch": 2.046871444823663,
      "grad_norm": 0.292058078028329,
      "learning_rate": 8.006346545095664e-07,
      "loss": 0.0023,
      "step": 8996
    },
    {
      "epoch": 2.047098976109215,
      "grad_norm": 0.34017235538854296,
      "learning_rate": 8.005489019566697e-07,
      "loss": 0.0024,
      "step": 8997
    },
    {
      "epoch": 2.0473265073947666,
      "grad_norm": 0.4926797661406449,
      "learning_rate": 8.004631458158749e-07,
      "loss": 0.002,
      "step": 8998
    },
    {
      "epoch": 2.0475540386803184,
      "grad_norm": 0.3789956350727742,
      "learning_rate": 8.00377386088935e-07,
      "loss": 0.0026,
      "step": 8999
    },
    {
      "epoch": 2.04778156996587,
      "grad_norm": 1.730755383544156,
      "learning_rate": 8.002916227776023e-07,
      "loss": 0.0166,
      "step": 9000
    },
    {
      "epoch": 2.048009101251422,
      "grad_norm": 0.6316753185358109,
      "learning_rate": 8.002058558836298e-07,
      "loss": 0.0051,
      "step": 9001
    },
    {
      "epoch": 2.0482366325369736,
      "grad_norm": 1.9860353721035442,
      "learning_rate": 8.001200854087707e-07,
      "loss": 0.0234,
      "step": 9002
    },
    {
      "epoch": 2.0484641638225254,
      "grad_norm": 0.81057068106513,
      "learning_rate": 8.000343113547777e-07,
      "loss": 0.0103,
      "step": 9003
    },
    {
      "epoch": 2.0486916951080776,
      "grad_norm": 0.7193106660313205,
      "learning_rate": 7.999485337234038e-07,
      "loss": 0.016,
      "step": 9004
    },
    {
      "epoch": 2.0489192263936293,
      "grad_norm": 0.5644722293497418,
      "learning_rate": 7.998627525164024e-07,
      "loss": 0.005,
      "step": 9005
    },
    {
      "epoch": 2.049146757679181,
      "grad_norm": 1.0815118124610577,
      "learning_rate": 7.997769677355266e-07,
      "loss": 0.013,
      "step": 9006
    },
    {
      "epoch": 2.049374288964733,
      "grad_norm": 0.8002532530964719,
      "learning_rate": 7.996911793825296e-07,
      "loss": 0.0101,
      "step": 9007
    },
    {
      "epoch": 2.0496018202502846,
      "grad_norm": 0.12992878138807132,
      "learning_rate": 7.996053874591649e-07,
      "loss": 0.0009,
      "step": 9008
    },
    {
      "epoch": 2.0498293515358363,
      "grad_norm": 0.911223912401618,
      "learning_rate": 7.995195919671858e-07,
      "loss": 0.0116,
      "step": 9009
    },
    {
      "epoch": 2.050056882821388,
      "grad_norm": 1.0486722371360964,
      "learning_rate": 7.994337929083458e-07,
      "loss": 0.0086,
      "step": 9010
    },
    {
      "epoch": 2.05028441410694,
      "grad_norm": 1.0615929119317695,
      "learning_rate": 7.993479902843987e-07,
      "loss": 0.0172,
      "step": 9011
    },
    {
      "epoch": 2.0505119453924916,
      "grad_norm": 0.5975358571869722,
      "learning_rate": 7.99262184097098e-07,
      "loss": 0.0034,
      "step": 9012
    },
    {
      "epoch": 2.0507394766780433,
      "grad_norm": 1.64169570548523,
      "learning_rate": 7.991763743481971e-07,
      "loss": 0.0087,
      "step": 9013
    },
    {
      "epoch": 2.050967007963595,
      "grad_norm": 1.6125987888060733,
      "learning_rate": 7.990905610394503e-07,
      "loss": 0.0134,
      "step": 9014
    },
    {
      "epoch": 2.051194539249147,
      "grad_norm": 1.2438605249431998,
      "learning_rate": 7.990047441726114e-07,
      "loss": 0.0185,
      "step": 9015
    },
    {
      "epoch": 2.0514220705346986,
      "grad_norm": 1.3056193907972449,
      "learning_rate": 7.989189237494339e-07,
      "loss": 0.0207,
      "step": 9016
    },
    {
      "epoch": 2.0516496018202504,
      "grad_norm": 0.30070481424774803,
      "learning_rate": 7.988330997716723e-07,
      "loss": 0.0028,
      "step": 9017
    },
    {
      "epoch": 2.051877133105802,
      "grad_norm": 1.960063236002895,
      "learning_rate": 7.987472722410805e-07,
      "loss": 0.0159,
      "step": 9018
    },
    {
      "epoch": 2.052104664391354,
      "grad_norm": 1.9691086699401108,
      "learning_rate": 7.986614411594126e-07,
      "loss": 0.0143,
      "step": 9019
    },
    {
      "epoch": 2.0523321956769056,
      "grad_norm": 0.5811928503525017,
      "learning_rate": 7.98575606528423e-07,
      "loss": 0.0044,
      "step": 9020
    },
    {
      "epoch": 2.0525597269624574,
      "grad_norm": 0.30333396807128366,
      "learning_rate": 7.984897683498658e-07,
      "loss": 0.0089,
      "step": 9021
    },
    {
      "epoch": 2.052787258248009,
      "grad_norm": 2.612841603050968,
      "learning_rate": 7.984039266254955e-07,
      "loss": 0.0128,
      "step": 9022
    },
    {
      "epoch": 2.053014789533561,
      "grad_norm": 0.8940541192462828,
      "learning_rate": 7.983180813570665e-07,
      "loss": 0.009,
      "step": 9023
    },
    {
      "epoch": 2.0532423208191126,
      "grad_norm": 0.3836424193297725,
      "learning_rate": 7.982322325463331e-07,
      "loss": 0.0032,
      "step": 9024
    },
    {
      "epoch": 2.0534698521046644,
      "grad_norm": 1.1020748181251936,
      "learning_rate": 7.981463801950507e-07,
      "loss": 0.0162,
      "step": 9025
    },
    {
      "epoch": 2.053697383390216,
      "grad_norm": 0.3686580621761448,
      "learning_rate": 7.98060524304973e-07,
      "loss": 0.0023,
      "step": 9026
    },
    {
      "epoch": 2.053924914675768,
      "grad_norm": 0.463039486142923,
      "learning_rate": 7.97974664877855e-07,
      "loss": 0.0065,
      "step": 9027
    },
    {
      "epoch": 2.0541524459613196,
      "grad_norm": 0.9088558526550047,
      "learning_rate": 7.978888019154518e-07,
      "loss": 0.0104,
      "step": 9028
    },
    {
      "epoch": 2.0543799772468714,
      "grad_norm": 1.4091409433907256,
      "learning_rate": 7.978029354195182e-07,
      "loss": 0.0061,
      "step": 9029
    },
    {
      "epoch": 2.054607508532423,
      "grad_norm": 1.219361089372488,
      "learning_rate": 7.977170653918088e-07,
      "loss": 0.0218,
      "step": 9030
    },
    {
      "epoch": 2.054835039817975,
      "grad_norm": 1.909174154160353,
      "learning_rate": 7.976311918340792e-07,
      "loss": 0.0156,
      "step": 9031
    },
    {
      "epoch": 2.0550625711035266,
      "grad_norm": 1.1829264688338827,
      "learning_rate": 7.975453147480843e-07,
      "loss": 0.0219,
      "step": 9032
    },
    {
      "epoch": 2.0552901023890784,
      "grad_norm": 0.4502657256145903,
      "learning_rate": 7.974594341355787e-07,
      "loss": 0.0018,
      "step": 9033
    },
    {
      "epoch": 2.05551763367463,
      "grad_norm": 0.7235083335627304,
      "learning_rate": 7.973735499983185e-07,
      "loss": 0.0063,
      "step": 9034
    },
    {
      "epoch": 2.055745164960182,
      "grad_norm": 0.9685176320074955,
      "learning_rate": 7.972876623380585e-07,
      "loss": 0.0099,
      "step": 9035
    },
    {
      "epoch": 2.0559726962457336,
      "grad_norm": 1.3658844400261203,
      "learning_rate": 7.972017711565543e-07,
      "loss": 0.0175,
      "step": 9036
    },
    {
      "epoch": 2.0562002275312854,
      "grad_norm": 0.8263557490636182,
      "learning_rate": 7.971158764555611e-07,
      "loss": 0.0112,
      "step": 9037
    },
    {
      "epoch": 2.056427758816837,
      "grad_norm": 0.8469835655904605,
      "learning_rate": 7.970299782368347e-07,
      "loss": 0.012,
      "step": 9038
    },
    {
      "epoch": 2.056655290102389,
      "grad_norm": 0.7231820277605212,
      "learning_rate": 7.969440765021306e-07,
      "loss": 0.0066,
      "step": 9039
    },
    {
      "epoch": 2.0568828213879407,
      "grad_norm": 0.8362930993172825,
      "learning_rate": 7.968581712532044e-07,
      "loss": 0.021,
      "step": 9040
    },
    {
      "epoch": 2.0571103526734924,
      "grad_norm": 0.7421397318368298,
      "learning_rate": 7.96772262491812e-07,
      "loss": 0.0162,
      "step": 9041
    },
    {
      "epoch": 2.057337883959044,
      "grad_norm": 0.5653847698807809,
      "learning_rate": 7.966863502197092e-07,
      "loss": 0.004,
      "step": 9042
    },
    {
      "epoch": 2.0575654152445964,
      "grad_norm": 0.65364731752615,
      "learning_rate": 7.966004344386517e-07,
      "loss": 0.0068,
      "step": 9043
    },
    {
      "epoch": 2.057792946530148,
      "grad_norm": 0.6117471407776484,
      "learning_rate": 7.965145151503957e-07,
      "loss": 0.0065,
      "step": 9044
    },
    {
      "epoch": 2.0580204778157,
      "grad_norm": 1.0405491373096627,
      "learning_rate": 7.964285923566971e-07,
      "loss": 0.0138,
      "step": 9045
    },
    {
      "epoch": 2.0582480091012516,
      "grad_norm": 0.6073237190255709,
      "learning_rate": 7.963426660593121e-07,
      "loss": 0.0085,
      "step": 9046
    },
    {
      "epoch": 2.0584755403868034,
      "grad_norm": 0.5217280099543562,
      "learning_rate": 7.962567362599965e-07,
      "loss": 0.0054,
      "step": 9047
    },
    {
      "epoch": 2.058703071672355,
      "grad_norm": 0.6392511927922102,
      "learning_rate": 7.961708029605072e-07,
      "loss": 0.0072,
      "step": 9048
    },
    {
      "epoch": 2.058930602957907,
      "grad_norm": 1.0489995042804985,
      "learning_rate": 7.960848661626e-07,
      "loss": 0.0188,
      "step": 9049
    },
    {
      "epoch": 2.0591581342434586,
      "grad_norm": 0.8699211663352905,
      "learning_rate": 7.959989258680314e-07,
      "loss": 0.0191,
      "step": 9050
    },
    {
      "epoch": 2.0593856655290104,
      "grad_norm": 0.9463630209342634,
      "learning_rate": 7.959129820785581e-07,
      "loss": 0.0047,
      "step": 9051
    },
    {
      "epoch": 2.059613196814562,
      "grad_norm": 0.7350564549155074,
      "learning_rate": 7.958270347959365e-07,
      "loss": 0.0064,
      "step": 9052
    },
    {
      "epoch": 2.059840728100114,
      "grad_norm": 0.28592973791968845,
      "learning_rate": 7.95741084021923e-07,
      "loss": 0.0025,
      "step": 9053
    },
    {
      "epoch": 2.0600682593856656,
      "grad_norm": 1.1979986828968519,
      "learning_rate": 7.956551297582744e-07,
      "loss": 0.0253,
      "step": 9054
    },
    {
      "epoch": 2.0602957906712174,
      "grad_norm": 0.6510461072413279,
      "learning_rate": 7.955691720067476e-07,
      "loss": 0.0177,
      "step": 9055
    },
    {
      "epoch": 2.060523321956769,
      "grad_norm": 1.3235521566375663,
      "learning_rate": 7.954832107690994e-07,
      "loss": 0.0049,
      "step": 9056
    },
    {
      "epoch": 2.060750853242321,
      "grad_norm": 0.7983932906236514,
      "learning_rate": 7.953972460470865e-07,
      "loss": 0.0091,
      "step": 9057
    },
    {
      "epoch": 2.0609783845278726,
      "grad_norm": 0.6824474435128689,
      "learning_rate": 7.953112778424658e-07,
      "loss": 0.0097,
      "step": 9058
    },
    {
      "epoch": 2.0612059158134244,
      "grad_norm": 0.982846671224294,
      "learning_rate": 7.952253061569946e-07,
      "loss": 0.0144,
      "step": 9059
    },
    {
      "epoch": 2.061433447098976,
      "grad_norm": 0.7138860884269269,
      "learning_rate": 7.951393309924299e-07,
      "loss": 0.0087,
      "step": 9060
    },
    {
      "epoch": 2.061660978384528,
      "grad_norm": 0.1514271661885778,
      "learning_rate": 7.950533523505288e-07,
      "loss": 0.0007,
      "step": 9061
    },
    {
      "epoch": 2.0618885096700796,
      "grad_norm": 0.49698122778317044,
      "learning_rate": 7.949673702330487e-07,
      "loss": 0.0033,
      "step": 9062
    },
    {
      "epoch": 2.0621160409556314,
      "grad_norm": 1.011219866728441,
      "learning_rate": 7.948813846417469e-07,
      "loss": 0.015,
      "step": 9063
    },
    {
      "epoch": 2.062343572241183,
      "grad_norm": 0.94325376844481,
      "learning_rate": 7.947953955783808e-07,
      "loss": 0.0074,
      "step": 9064
    },
    {
      "epoch": 2.062571103526735,
      "grad_norm": 0.43953814787234513,
      "learning_rate": 7.947094030447077e-07,
      "loss": 0.0082,
      "step": 9065
    },
    {
      "epoch": 2.0627986348122866,
      "grad_norm": 0.770624168599187,
      "learning_rate": 7.946234070424852e-07,
      "loss": 0.0087,
      "step": 9066
    },
    {
      "epoch": 2.0630261660978384,
      "grad_norm": 0.7852652186331285,
      "learning_rate": 7.945374075734706e-07,
      "loss": 0.0066,
      "step": 9067
    },
    {
      "epoch": 2.06325369738339,
      "grad_norm": 1.8257160662731842,
      "learning_rate": 7.944514046394222e-07,
      "loss": 0.0166,
      "step": 9068
    },
    {
      "epoch": 2.063481228668942,
      "grad_norm": 0.6943659241212427,
      "learning_rate": 7.943653982420976e-07,
      "loss": 0.0091,
      "step": 9069
    },
    {
      "epoch": 2.0637087599544937,
      "grad_norm": 0.337384829969432,
      "learning_rate": 7.942793883832541e-07,
      "loss": 0.0019,
      "step": 9070
    },
    {
      "epoch": 2.0639362912400454,
      "grad_norm": 0.7104155788709781,
      "learning_rate": 7.9419337506465e-07,
      "loss": 0.0095,
      "step": 9071
    },
    {
      "epoch": 2.064163822525597,
      "grad_norm": 0.5962868363957752,
      "learning_rate": 7.941073582880431e-07,
      "loss": 0.0044,
      "step": 9072
    },
    {
      "epoch": 2.064391353811149,
      "grad_norm": 0.338270468072751,
      "learning_rate": 7.940213380551918e-07,
      "loss": 0.0035,
      "step": 9073
    },
    {
      "epoch": 2.0646188850967007,
      "grad_norm": 0.6378159954038821,
      "learning_rate": 7.939353143678535e-07,
      "loss": 0.005,
      "step": 9074
    },
    {
      "epoch": 2.0648464163822524,
      "grad_norm": 0.6056365879021587,
      "learning_rate": 7.938492872277872e-07,
      "loss": 0.0096,
      "step": 9075
    },
    {
      "epoch": 2.065073947667804,
      "grad_norm": 1.5006441244985853,
      "learning_rate": 7.937632566367505e-07,
      "loss": 0.0182,
      "step": 9076
    },
    {
      "epoch": 2.065301478953356,
      "grad_norm": 0.38261687744743117,
      "learning_rate": 7.936772225965018e-07,
      "loss": 0.0024,
      "step": 9077
    },
    {
      "epoch": 2.0655290102389077,
      "grad_norm": 0.9294557601529759,
      "learning_rate": 7.935911851087996e-07,
      "loss": 0.0072,
      "step": 9078
    },
    {
      "epoch": 2.0657565415244594,
      "grad_norm": 1.479721573100565,
      "learning_rate": 7.935051441754024e-07,
      "loss": 0.0264,
      "step": 9079
    },
    {
      "epoch": 2.065984072810011,
      "grad_norm": 0.6723776429149259,
      "learning_rate": 7.934190997980687e-07,
      "loss": 0.0172,
      "step": 9080
    },
    {
      "epoch": 2.066211604095563,
      "grad_norm": 0.9140317493359092,
      "learning_rate": 7.933330519785569e-07,
      "loss": 0.0097,
      "step": 9081
    },
    {
      "epoch": 2.066439135381115,
      "grad_norm": 0.44542992334864684,
      "learning_rate": 7.93247000718626e-07,
      "loss": 0.005,
      "step": 9082
    },
    {
      "epoch": 2.066666666666667,
      "grad_norm": 0.6753816366432505,
      "learning_rate": 7.931609460200345e-07,
      "loss": 0.0086,
      "step": 9083
    },
    {
      "epoch": 2.0668941979522186,
      "grad_norm": 0.19971118324429432,
      "learning_rate": 7.930748878845411e-07,
      "loss": 0.0016,
      "step": 9084
    },
    {
      "epoch": 2.0671217292377704,
      "grad_norm": 1.0544990680868636,
      "learning_rate": 7.929888263139049e-07,
      "loss": 0.0189,
      "step": 9085
    },
    {
      "epoch": 2.067349260523322,
      "grad_norm": 0.5973469577421473,
      "learning_rate": 7.929027613098848e-07,
      "loss": 0.0084,
      "step": 9086
    },
    {
      "epoch": 2.067576791808874,
      "grad_norm": 0.5179164240056631,
      "learning_rate": 7.928166928742398e-07,
      "loss": 0.0093,
      "step": 9087
    },
    {
      "epoch": 2.0678043230944256,
      "grad_norm": 0.8743029146459123,
      "learning_rate": 7.927306210087287e-07,
      "loss": 0.0168,
      "step": 9088
    },
    {
      "epoch": 2.0680318543799774,
      "grad_norm": 0.6036406868986361,
      "learning_rate": 7.926445457151111e-07,
      "loss": 0.0099,
      "step": 9089
    },
    {
      "epoch": 2.068259385665529,
      "grad_norm": 0.3948224738554559,
      "learning_rate": 7.925584669951459e-07,
      "loss": 0.0021,
      "step": 9090
    },
    {
      "epoch": 2.068486916951081,
      "grad_norm": 0.44922581856779004,
      "learning_rate": 7.924723848505925e-07,
      "loss": 0.0028,
      "step": 9091
    },
    {
      "epoch": 2.0687144482366326,
      "grad_norm": 1.1948292410858585,
      "learning_rate": 7.923862992832103e-07,
      "loss": 0.0217,
      "step": 9092
    },
    {
      "epoch": 2.0689419795221844,
      "grad_norm": 1.100180037321114,
      "learning_rate": 7.923002102947587e-07,
      "loss": 0.01,
      "step": 9093
    },
    {
      "epoch": 2.069169510807736,
      "grad_norm": 0.6937675534728082,
      "learning_rate": 7.922141178869973e-07,
      "loss": 0.009,
      "step": 9094
    },
    {
      "epoch": 2.069397042093288,
      "grad_norm": 0.7752258384880746,
      "learning_rate": 7.921280220616855e-07,
      "loss": 0.0152,
      "step": 9095
    },
    {
      "epoch": 2.0696245733788396,
      "grad_norm": 1.0795686730194802,
      "learning_rate": 7.920419228205829e-07,
      "loss": 0.0166,
      "step": 9096
    },
    {
      "epoch": 2.0698521046643914,
      "grad_norm": 0.607461867352133,
      "learning_rate": 7.919558201654493e-07,
      "loss": 0.0086,
      "step": 9097
    },
    {
      "epoch": 2.070079635949943,
      "grad_norm": 0.3311543160766717,
      "learning_rate": 7.918697140980447e-07,
      "loss": 0.0018,
      "step": 9098
    },
    {
      "epoch": 2.070307167235495,
      "grad_norm": 1.362904988120221,
      "learning_rate": 7.917836046201285e-07,
      "loss": 0.0195,
      "step": 9099
    },
    {
      "epoch": 2.0705346985210467,
      "grad_norm": 0.5205124460602316,
      "learning_rate": 7.916974917334612e-07,
      "loss": 0.0042,
      "step": 9100
    },
    {
      "epoch": 2.0707622298065984,
      "grad_norm": 0.8203663475474079,
      "learning_rate": 7.916113754398022e-07,
      "loss": 0.0156,
      "step": 9101
    },
    {
      "epoch": 2.07098976109215,
      "grad_norm": 1.1847893907735414,
      "learning_rate": 7.91525255740912e-07,
      "loss": 0.0231,
      "step": 9102
    },
    {
      "epoch": 2.071217292377702,
      "grad_norm": 1.242769539345523,
      "learning_rate": 7.914391326385504e-07,
      "loss": 0.0294,
      "step": 9103
    },
    {
      "epoch": 2.0714448236632537,
      "grad_norm": 0.6044568620800408,
      "learning_rate": 7.913530061344778e-07,
      "loss": 0.0086,
      "step": 9104
    },
    {
      "epoch": 2.0716723549488054,
      "grad_norm": 0.8446369320506389,
      "learning_rate": 7.912668762304544e-07,
      "loss": 0.0069,
      "step": 9105
    },
    {
      "epoch": 2.071899886234357,
      "grad_norm": 0.9634276448430097,
      "learning_rate": 7.911807429282406e-07,
      "loss": 0.0108,
      "step": 9106
    },
    {
      "epoch": 2.072127417519909,
      "grad_norm": 0.4255837431208955,
      "learning_rate": 7.910946062295967e-07,
      "loss": 0.0026,
      "step": 9107
    },
    {
      "epoch": 2.0723549488054607,
      "grad_norm": 1.0929442448124784,
      "learning_rate": 7.910084661362832e-07,
      "loss": 0.0201,
      "step": 9108
    },
    {
      "epoch": 2.0725824800910124,
      "grad_norm": 1.0093345207408708,
      "learning_rate": 7.909223226500609e-07,
      "loss": 0.0091,
      "step": 9109
    },
    {
      "epoch": 2.072810011376564,
      "grad_norm": 0.3205405530723011,
      "learning_rate": 7.908361757726901e-07,
      "loss": 0.0042,
      "step": 9110
    },
    {
      "epoch": 2.073037542662116,
      "grad_norm": 0.4914107843164597,
      "learning_rate": 7.907500255059314e-07,
      "loss": 0.0057,
      "step": 9111
    },
    {
      "epoch": 2.0732650739476677,
      "grad_norm": 2.183497169121956,
      "learning_rate": 7.906638718515458e-07,
      "loss": 0.0328,
      "step": 9112
    },
    {
      "epoch": 2.0734926052332194,
      "grad_norm": 0.6287818502880228,
      "learning_rate": 7.905777148112943e-07,
      "loss": 0.0043,
      "step": 9113
    },
    {
      "epoch": 2.073720136518771,
      "grad_norm": 1.770987962266444,
      "learning_rate": 7.904915543869372e-07,
      "loss": 0.0161,
      "step": 9114
    },
    {
      "epoch": 2.073947667804323,
      "grad_norm": 1.6020734134248478,
      "learning_rate": 7.904053905802362e-07,
      "loss": 0.0116,
      "step": 9115
    },
    {
      "epoch": 2.0741751990898747,
      "grad_norm": 0.4170883617995873,
      "learning_rate": 7.903192233929515e-07,
      "loss": 0.0026,
      "step": 9116
    },
    {
      "epoch": 2.0744027303754264,
      "grad_norm": 1.4426736839119412,
      "learning_rate": 7.90233052826845e-07,
      "loss": 0.0136,
      "step": 9117
    },
    {
      "epoch": 2.074630261660978,
      "grad_norm": 0.5364490219390096,
      "learning_rate": 7.901468788836771e-07,
      "loss": 0.0125,
      "step": 9118
    },
    {
      "epoch": 2.07485779294653,
      "grad_norm": 0.8118218292814557,
      "learning_rate": 7.900607015652101e-07,
      "loss": 0.0123,
      "step": 9119
    },
    {
      "epoch": 2.0750853242320817,
      "grad_norm": 0.4720346997920728,
      "learning_rate": 7.899745208732043e-07,
      "loss": 0.0038,
      "step": 9120
    },
    {
      "epoch": 2.075312855517634,
      "grad_norm": 1.1400139505704863,
      "learning_rate": 7.898883368094213e-07,
      "loss": 0.0143,
      "step": 9121
    },
    {
      "epoch": 2.0755403868031856,
      "grad_norm": 1.5162285525990173,
      "learning_rate": 7.898021493756228e-07,
      "loss": 0.0179,
      "step": 9122
    },
    {
      "epoch": 2.0757679180887374,
      "grad_norm": 1.2460561011233084,
      "learning_rate": 7.897159585735702e-07,
      "loss": 0.0145,
      "step": 9123
    },
    {
      "epoch": 2.075995449374289,
      "grad_norm": 0.4314378033216816,
      "learning_rate": 7.896297644050249e-07,
      "loss": 0.0031,
      "step": 9124
    },
    {
      "epoch": 2.076222980659841,
      "grad_norm": 0.5290618219543192,
      "learning_rate": 7.895435668717488e-07,
      "loss": 0.0031,
      "step": 9125
    },
    {
      "epoch": 2.0764505119453927,
      "grad_norm": 0.6912892689024679,
      "learning_rate": 7.894573659755038e-07,
      "loss": 0.0139,
      "step": 9126
    },
    {
      "epoch": 2.0766780432309444,
      "grad_norm": 0.24807223933852632,
      "learning_rate": 7.89371161718051e-07,
      "loss": 0.0014,
      "step": 9127
    },
    {
      "epoch": 2.076905574516496,
      "grad_norm": 0.5503007574869582,
      "learning_rate": 7.892849541011531e-07,
      "loss": 0.0063,
      "step": 9128
    },
    {
      "epoch": 2.077133105802048,
      "grad_norm": 1.195350434972967,
      "learning_rate": 7.891987431265714e-07,
      "loss": 0.0073,
      "step": 9129
    },
    {
      "epoch": 2.0773606370875997,
      "grad_norm": 0.379320486105582,
      "learning_rate": 7.891125287960682e-07,
      "loss": 0.0042,
      "step": 9130
    },
    {
      "epoch": 2.0775881683731514,
      "grad_norm": 1.1603595803509519,
      "learning_rate": 7.890263111114052e-07,
      "loss": 0.0126,
      "step": 9131
    },
    {
      "epoch": 2.077815699658703,
      "grad_norm": 0.46371094308488725,
      "learning_rate": 7.889400900743452e-07,
      "loss": 0.0017,
      "step": 9132
    },
    {
      "epoch": 2.078043230944255,
      "grad_norm": 1.4002252428358113,
      "learning_rate": 7.888538656866498e-07,
      "loss": 0.0115,
      "step": 9133
    },
    {
      "epoch": 2.0782707622298067,
      "grad_norm": 1.2049172434717985,
      "learning_rate": 7.887676379500814e-07,
      "loss": 0.01,
      "step": 9134
    },
    {
      "epoch": 2.0784982935153584,
      "grad_norm": 1.1518502874753174,
      "learning_rate": 7.886814068664025e-07,
      "loss": 0.0128,
      "step": 9135
    },
    {
      "epoch": 2.07872582480091,
      "grad_norm": 0.806150601896166,
      "learning_rate": 7.885951724373754e-07,
      "loss": 0.0095,
      "step": 9136
    },
    {
      "epoch": 2.078953356086462,
      "grad_norm": 0.7296766572310524,
      "learning_rate": 7.885089346647625e-07,
      "loss": 0.0052,
      "step": 9137
    },
    {
      "epoch": 2.0791808873720137,
      "grad_norm": 0.49776046586875056,
      "learning_rate": 7.884226935503267e-07,
      "loss": 0.0072,
      "step": 9138
    },
    {
      "epoch": 2.0794084186575654,
      "grad_norm": 1.029646859030014,
      "learning_rate": 7.8833644909583e-07,
      "loss": 0.0119,
      "step": 9139
    },
    {
      "epoch": 2.079635949943117,
      "grad_norm": 0.9421760038041583,
      "learning_rate": 7.882502013030358e-07,
      "loss": 0.0066,
      "step": 9140
    },
    {
      "epoch": 2.079863481228669,
      "grad_norm": 0.6011728262566824,
      "learning_rate": 7.881639501737059e-07,
      "loss": 0.006,
      "step": 9141
    },
    {
      "epoch": 2.0800910125142207,
      "grad_norm": 0.5032679788716042,
      "learning_rate": 7.880776957096041e-07,
      "loss": 0.005,
      "step": 9142
    },
    {
      "epoch": 2.0803185437997724,
      "grad_norm": 0.25041585957228457,
      "learning_rate": 7.879914379124928e-07,
      "loss": 0.002,
      "step": 9143
    },
    {
      "epoch": 2.080546075085324,
      "grad_norm": 1.3963677037027489,
      "learning_rate": 7.879051767841351e-07,
      "loss": 0.023,
      "step": 9144
    },
    {
      "epoch": 2.080773606370876,
      "grad_norm": 1.1133586768130128,
      "learning_rate": 7.878189123262937e-07,
      "loss": 0.0115,
      "step": 9145
    },
    {
      "epoch": 2.0810011376564277,
      "grad_norm": 0.4853918417871173,
      "learning_rate": 7.877326445407321e-07,
      "loss": 0.0074,
      "step": 9146
    },
    {
      "epoch": 2.0812286689419794,
      "grad_norm": 1.5958592224731931,
      "learning_rate": 7.876463734292133e-07,
      "loss": 0.041,
      "step": 9147
    },
    {
      "epoch": 2.081456200227531,
      "grad_norm": 0.5297580612008737,
      "learning_rate": 7.875600989935004e-07,
      "loss": 0.0051,
      "step": 9148
    },
    {
      "epoch": 2.081683731513083,
      "grad_norm": 0.7536803610859091,
      "learning_rate": 7.874738212353567e-07,
      "loss": 0.0045,
      "step": 9149
    },
    {
      "epoch": 2.0819112627986347,
      "grad_norm": 0.6095081040006934,
      "learning_rate": 7.873875401565458e-07,
      "loss": 0.0079,
      "step": 9150
    },
    {
      "epoch": 2.0821387940841865,
      "grad_norm": 0.6794626976572432,
      "learning_rate": 7.873012557588309e-07,
      "loss": 0.0056,
      "step": 9151
    },
    {
      "epoch": 2.082366325369738,
      "grad_norm": 2.24210983874312,
      "learning_rate": 7.872149680439756e-07,
      "loss": 0.0245,
      "step": 9152
    },
    {
      "epoch": 2.08259385665529,
      "grad_norm": 0.5269261677908896,
      "learning_rate": 7.871286770137434e-07,
      "loss": 0.0095,
      "step": 9153
    },
    {
      "epoch": 2.0828213879408417,
      "grad_norm": 2.916933169794705,
      "learning_rate": 7.870423826698981e-07,
      "loss": 0.0797,
      "step": 9154
    },
    {
      "epoch": 2.0830489192263935,
      "grad_norm": 1.0193821841518744,
      "learning_rate": 7.869560850142031e-07,
      "loss": 0.0206,
      "step": 9155
    },
    {
      "epoch": 2.083276450511945,
      "grad_norm": 1.9213408348971808,
      "learning_rate": 7.868697840484225e-07,
      "loss": 0.0402,
      "step": 9156
    },
    {
      "epoch": 2.083503981797497,
      "grad_norm": 0.7622595400971296,
      "learning_rate": 7.867834797743199e-07,
      "loss": 0.0109,
      "step": 9157
    },
    {
      "epoch": 2.0837315130830487,
      "grad_norm": 1.5515544467481235,
      "learning_rate": 7.866971721936592e-07,
      "loss": 0.0065,
      "step": 9158
    },
    {
      "epoch": 2.0839590443686005,
      "grad_norm": 1.1105111985696028,
      "learning_rate": 7.866108613082045e-07,
      "loss": 0.0066,
      "step": 9159
    },
    {
      "epoch": 2.0841865756541527,
      "grad_norm": 1.0582893467528578,
      "learning_rate": 7.865245471197198e-07,
      "loss": 0.0188,
      "step": 9160
    },
    {
      "epoch": 2.0844141069397044,
      "grad_norm": 1.0595467375317473,
      "learning_rate": 7.864382296299689e-07,
      "loss": 0.0216,
      "step": 9161
    },
    {
      "epoch": 2.084641638225256,
      "grad_norm": 0.4115376271807129,
      "learning_rate": 7.863519088407164e-07,
      "loss": 0.0031,
      "step": 9162
    },
    {
      "epoch": 2.084869169510808,
      "grad_norm": 0.8300257540667818,
      "learning_rate": 7.862655847537265e-07,
      "loss": 0.0132,
      "step": 9163
    },
    {
      "epoch": 2.0850967007963597,
      "grad_norm": 0.9064075749899173,
      "learning_rate": 7.861792573707634e-07,
      "loss": 0.0123,
      "step": 9164
    },
    {
      "epoch": 2.0853242320819114,
      "grad_norm": 0.6521758632997577,
      "learning_rate": 7.860929266935914e-07,
      "loss": 0.0052,
      "step": 9165
    },
    {
      "epoch": 2.085551763367463,
      "grad_norm": 0.27304053836354314,
      "learning_rate": 7.860065927239752e-07,
      "loss": 0.0024,
      "step": 9166
    },
    {
      "epoch": 2.085779294653015,
      "grad_norm": 0.5580775824388217,
      "learning_rate": 7.859202554636788e-07,
      "loss": 0.0093,
      "step": 9167
    },
    {
      "epoch": 2.0860068259385667,
      "grad_norm": 0.7820423391937628,
      "learning_rate": 7.858339149144671e-07,
      "loss": 0.0148,
      "step": 9168
    },
    {
      "epoch": 2.0862343572241184,
      "grad_norm": 0.6776426598903225,
      "learning_rate": 7.857475710781048e-07,
      "loss": 0.0099,
      "step": 9169
    },
    {
      "epoch": 2.08646188850967,
      "grad_norm": 0.9597468223496397,
      "learning_rate": 7.856612239563568e-07,
      "loss": 0.012,
      "step": 9170
    },
    {
      "epoch": 2.086689419795222,
      "grad_norm": 0.5301772798675423,
      "learning_rate": 7.855748735509873e-07,
      "loss": 0.004,
      "step": 9171
    },
    {
      "epoch": 2.0869169510807737,
      "grad_norm": 0.32840689959966574,
      "learning_rate": 7.854885198637615e-07,
      "loss": 0.0021,
      "step": 9172
    },
    {
      "epoch": 2.0871444823663254,
      "grad_norm": 0.8126896608713541,
      "learning_rate": 7.854021628964445e-07,
      "loss": 0.0072,
      "step": 9173
    },
    {
      "epoch": 2.087372013651877,
      "grad_norm": 0.966939750909063,
      "learning_rate": 7.853158026508009e-07,
      "loss": 0.0135,
      "step": 9174
    },
    {
      "epoch": 2.087599544937429,
      "grad_norm": 0.5501365281480977,
      "learning_rate": 7.852294391285959e-07,
      "loss": 0.0105,
      "step": 9175
    },
    {
      "epoch": 2.0878270762229807,
      "grad_norm": 0.8447495657831009,
      "learning_rate": 7.851430723315946e-07,
      "loss": 0.0088,
      "step": 9176
    },
    {
      "epoch": 2.0880546075085324,
      "grad_norm": 0.3759611000792586,
      "learning_rate": 7.850567022615623e-07,
      "loss": 0.0025,
      "step": 9177
    },
    {
      "epoch": 2.088282138794084,
      "grad_norm": 0.7947305000118688,
      "learning_rate": 7.84970328920264e-07,
      "loss": 0.0071,
      "step": 9178
    },
    {
      "epoch": 2.088509670079636,
      "grad_norm": 0.6577083486461792,
      "learning_rate": 7.848839523094651e-07,
      "loss": 0.0098,
      "step": 9179
    },
    {
      "epoch": 2.0887372013651877,
      "grad_norm": 1.7343321429669305,
      "learning_rate": 7.847975724309312e-07,
      "loss": 0.014,
      "step": 9180
    },
    {
      "epoch": 2.0889647326507395,
      "grad_norm": 0.680123524548824,
      "learning_rate": 7.847111892864275e-07,
      "loss": 0.0051,
      "step": 9181
    },
    {
      "epoch": 2.089192263936291,
      "grad_norm": 0.6703674002826029,
      "learning_rate": 7.846248028777195e-07,
      "loss": 0.0131,
      "step": 9182
    },
    {
      "epoch": 2.089419795221843,
      "grad_norm": 1.1807807672416228,
      "learning_rate": 7.845384132065731e-07,
      "loss": 0.0181,
      "step": 9183
    },
    {
      "epoch": 2.0896473265073947,
      "grad_norm": 0.703243120255615,
      "learning_rate": 7.844520202747534e-07,
      "loss": 0.0118,
      "step": 9184
    },
    {
      "epoch": 2.0898748577929465,
      "grad_norm": 1.5320837528615565,
      "learning_rate": 7.843656240840265e-07,
      "loss": 0.0247,
      "step": 9185
    },
    {
      "epoch": 2.090102389078498,
      "grad_norm": 0.5499001837311822,
      "learning_rate": 7.842792246361583e-07,
      "loss": 0.0026,
      "step": 9186
    },
    {
      "epoch": 2.09032992036405,
      "grad_norm": 1.267889632928035,
      "learning_rate": 7.841928219329144e-07,
      "loss": 0.0192,
      "step": 9187
    },
    {
      "epoch": 2.0905574516496017,
      "grad_norm": 0.8294155358531753,
      "learning_rate": 7.841064159760605e-07,
      "loss": 0.0061,
      "step": 9188
    },
    {
      "epoch": 2.0907849829351535,
      "grad_norm": 0.6126934411224282,
      "learning_rate": 7.840200067673632e-07,
      "loss": 0.0053,
      "step": 9189
    },
    {
      "epoch": 2.091012514220705,
      "grad_norm": 0.21523624283281878,
      "learning_rate": 7.839335943085882e-07,
      "loss": 0.002,
      "step": 9190
    },
    {
      "epoch": 2.091240045506257,
      "grad_norm": 0.48695567972763487,
      "learning_rate": 7.838471786015013e-07,
      "loss": 0.0095,
      "step": 9191
    },
    {
      "epoch": 2.0914675767918087,
      "grad_norm": 1.2177457934193316,
      "learning_rate": 7.837607596478691e-07,
      "loss": 0.0173,
      "step": 9192
    },
    {
      "epoch": 2.0916951080773605,
      "grad_norm": 1.0586963752613927,
      "learning_rate": 7.836743374494579e-07,
      "loss": 0.01,
      "step": 9193
    },
    {
      "epoch": 2.0919226393629122,
      "grad_norm": 0.9346759824478303,
      "learning_rate": 7.835879120080337e-07,
      "loss": 0.0224,
      "step": 9194
    },
    {
      "epoch": 2.092150170648464,
      "grad_norm": 0.4713141300371756,
      "learning_rate": 7.83501483325363e-07,
      "loss": 0.0049,
      "step": 9195
    },
    {
      "epoch": 2.0923777019340157,
      "grad_norm": 1.162350148632552,
      "learning_rate": 7.834150514032124e-07,
      "loss": 0.0139,
      "step": 9196
    },
    {
      "epoch": 2.0926052332195675,
      "grad_norm": 0.41157198927050814,
      "learning_rate": 7.833286162433481e-07,
      "loss": 0.0028,
      "step": 9197
    },
    {
      "epoch": 2.0928327645051192,
      "grad_norm": 1.1099539495640738,
      "learning_rate": 7.832421778475369e-07,
      "loss": 0.0149,
      "step": 9198
    },
    {
      "epoch": 2.0930602957906714,
      "grad_norm": 2.1675030701503224,
      "learning_rate": 7.831557362175454e-07,
      "loss": 0.0311,
      "step": 9199
    },
    {
      "epoch": 2.093287827076223,
      "grad_norm": 0.6801671638755284,
      "learning_rate": 7.830692913551403e-07,
      "loss": 0.015,
      "step": 9200
    },
    {
      "epoch": 2.093515358361775,
      "grad_norm": 0.9498273562152529,
      "learning_rate": 7.829828432620884e-07,
      "loss": 0.0139,
      "step": 9201
    },
    {
      "epoch": 2.0937428896473267,
      "grad_norm": 0.17757418969248412,
      "learning_rate": 7.828963919401567e-07,
      "loss": 0.0005,
      "step": 9202
    },
    {
      "epoch": 2.0939704209328784,
      "grad_norm": 0.9529307364560996,
      "learning_rate": 7.828099373911116e-07,
      "loss": 0.0206,
      "step": 9203
    },
    {
      "epoch": 2.09419795221843,
      "grad_norm": 0.4720678151921159,
      "learning_rate": 7.827234796167206e-07,
      "loss": 0.0045,
      "step": 9204
    },
    {
      "epoch": 2.094425483503982,
      "grad_norm": 0.5564923865534581,
      "learning_rate": 7.826370186187503e-07,
      "loss": 0.0112,
      "step": 9205
    },
    {
      "epoch": 2.0946530147895337,
      "grad_norm": 1.1258120369556415,
      "learning_rate": 7.825505543989682e-07,
      "loss": 0.0116,
      "step": 9206
    },
    {
      "epoch": 2.0948805460750854,
      "grad_norm": 1.739664201930335,
      "learning_rate": 7.824640869591412e-07,
      "loss": 0.0234,
      "step": 9207
    },
    {
      "epoch": 2.095108077360637,
      "grad_norm": 0.23151133729253148,
      "learning_rate": 7.823776163010369e-07,
      "loss": 0.0022,
      "step": 9208
    },
    {
      "epoch": 2.095335608646189,
      "grad_norm": 1.3191505337757548,
      "learning_rate": 7.822911424264222e-07,
      "loss": 0.0108,
      "step": 9209
    },
    {
      "epoch": 2.0955631399317407,
      "grad_norm": 1.0201891270344592,
      "learning_rate": 7.822046653370647e-07,
      "loss": 0.012,
      "step": 9210
    },
    {
      "epoch": 2.0957906712172925,
      "grad_norm": 1.1541827816719863,
      "learning_rate": 7.821181850347316e-07,
      "loss": 0.0055,
      "step": 9211
    },
    {
      "epoch": 2.096018202502844,
      "grad_norm": 1.3264770201849603,
      "learning_rate": 7.820317015211905e-07,
      "loss": 0.0162,
      "step": 9212
    },
    {
      "epoch": 2.096245733788396,
      "grad_norm": 0.6640103343814197,
      "learning_rate": 7.819452147982091e-07,
      "loss": 0.0064,
      "step": 9213
    },
    {
      "epoch": 2.0964732650739477,
      "grad_norm": 0.6796595134370279,
      "learning_rate": 7.81858724867555e-07,
      "loss": 0.0095,
      "step": 9214
    },
    {
      "epoch": 2.0967007963594995,
      "grad_norm": 0.5236777498501003,
      "learning_rate": 7.817722317309958e-07,
      "loss": 0.0055,
      "step": 9215
    },
    {
      "epoch": 2.096928327645051,
      "grad_norm": 1.2277357402303708,
      "learning_rate": 7.816857353902993e-07,
      "loss": 0.0171,
      "step": 9216
    },
    {
      "epoch": 2.097155858930603,
      "grad_norm": 0.628137537522774,
      "learning_rate": 7.815992358472332e-07,
      "loss": 0.0063,
      "step": 9217
    },
    {
      "epoch": 2.0973833902161547,
      "grad_norm": 0.14293525239107094,
      "learning_rate": 7.815127331035656e-07,
      "loss": 0.0013,
      "step": 9218
    },
    {
      "epoch": 2.0976109215017065,
      "grad_norm": 1.9126063856709803,
      "learning_rate": 7.814262271610645e-07,
      "loss": 0.0361,
      "step": 9219
    },
    {
      "epoch": 2.0978384527872582,
      "grad_norm": 0.5261794311393453,
      "learning_rate": 7.813397180214978e-07,
      "loss": 0.0063,
      "step": 9220
    },
    {
      "epoch": 2.09806598407281,
      "grad_norm": 0.8997752073865246,
      "learning_rate": 7.812532056866334e-07,
      "loss": 0.0157,
      "step": 9221
    },
    {
      "epoch": 2.0982935153583617,
      "grad_norm": 2.3182064053409372,
      "learning_rate": 7.811666901582397e-07,
      "loss": 0.0236,
      "step": 9222
    },
    {
      "epoch": 2.0985210466439135,
      "grad_norm": 0.6059480274929465,
      "learning_rate": 7.81080171438085e-07,
      "loss": 0.0048,
      "step": 9223
    },
    {
      "epoch": 2.0987485779294652,
      "grad_norm": 0.7420906474117391,
      "learning_rate": 7.809936495279373e-07,
      "loss": 0.0043,
      "step": 9224
    },
    {
      "epoch": 2.098976109215017,
      "grad_norm": 0.7252106301316513,
      "learning_rate": 7.80907124429565e-07,
      "loss": 0.013,
      "step": 9225
    },
    {
      "epoch": 2.0992036405005687,
      "grad_norm": 0.9002092835571386,
      "learning_rate": 7.808205961447368e-07,
      "loss": 0.0138,
      "step": 9226
    },
    {
      "epoch": 2.0994311717861205,
      "grad_norm": 1.2342077987333195,
      "learning_rate": 7.80734064675221e-07,
      "loss": 0.0174,
      "step": 9227
    },
    {
      "epoch": 2.0996587030716722,
      "grad_norm": 1.1037992668025742,
      "learning_rate": 7.806475300227859e-07,
      "loss": 0.0192,
      "step": 9228
    },
    {
      "epoch": 2.099886234357224,
      "grad_norm": 0.786939908331163,
      "learning_rate": 7.805609921892006e-07,
      "loss": 0.0056,
      "step": 9229
    },
    {
      "epoch": 2.1001137656427757,
      "grad_norm": 0.7136016687159851,
      "learning_rate": 7.804744511762334e-07,
      "loss": 0.0139,
      "step": 9230
    },
    {
      "epoch": 2.1003412969283275,
      "grad_norm": 0.2289243124919493,
      "learning_rate": 7.803879069856532e-07,
      "loss": 0.0017,
      "step": 9231
    },
    {
      "epoch": 2.1005688282138792,
      "grad_norm": 1.1413146766634172,
      "learning_rate": 7.803013596192285e-07,
      "loss": 0.0216,
      "step": 9232
    },
    {
      "epoch": 2.100796359499431,
      "grad_norm": 1.0425362067877952,
      "learning_rate": 7.802148090787289e-07,
      "loss": 0.0113,
      "step": 9233
    },
    {
      "epoch": 2.1010238907849828,
      "grad_norm": 1.1714190250184568,
      "learning_rate": 7.801282553659225e-07,
      "loss": 0.0261,
      "step": 9234
    },
    {
      "epoch": 2.1012514220705345,
      "grad_norm": 0.8367732906380727,
      "learning_rate": 7.800416984825787e-07,
      "loss": 0.0106,
      "step": 9235
    },
    {
      "epoch": 2.1014789533560863,
      "grad_norm": 0.542809043926659,
      "learning_rate": 7.799551384304666e-07,
      "loss": 0.006,
      "step": 9236
    },
    {
      "epoch": 2.101706484641638,
      "grad_norm": 0.7628267307241319,
      "learning_rate": 7.798685752113553e-07,
      "loss": 0.0055,
      "step": 9237
    },
    {
      "epoch": 2.10193401592719,
      "grad_norm": 0.955833216060939,
      "learning_rate": 7.797820088270141e-07,
      "loss": 0.0156,
      "step": 9238
    },
    {
      "epoch": 2.102161547212742,
      "grad_norm": 1.0564360349539703,
      "learning_rate": 7.796954392792119e-07,
      "loss": 0.0173,
      "step": 9239
    },
    {
      "epoch": 2.1023890784982937,
      "grad_norm": 0.9030033078460505,
      "learning_rate": 7.796088665697183e-07,
      "loss": 0.0129,
      "step": 9240
    },
    {
      "epoch": 2.1026166097838455,
      "grad_norm": 0.8232700967230857,
      "learning_rate": 7.795222907003025e-07,
      "loss": 0.0246,
      "step": 9241
    },
    {
      "epoch": 2.102844141069397,
      "grad_norm": 0.7206999137991037,
      "learning_rate": 7.794357116727342e-07,
      "loss": 0.0091,
      "step": 9242
    },
    {
      "epoch": 2.103071672354949,
      "grad_norm": 0.9851948857694026,
      "learning_rate": 7.793491294887827e-07,
      "loss": 0.0158,
      "step": 9243
    },
    {
      "epoch": 2.1032992036405007,
      "grad_norm": 2.514647738852726,
      "learning_rate": 7.792625441502178e-07,
      "loss": 0.0316,
      "step": 9244
    },
    {
      "epoch": 2.1035267349260525,
      "grad_norm": 1.0881689113865614,
      "learning_rate": 7.791759556588088e-07,
      "loss": 0.0166,
      "step": 9245
    },
    {
      "epoch": 2.103754266211604,
      "grad_norm": 1.0334434567242292,
      "learning_rate": 7.790893640163258e-07,
      "loss": 0.0106,
      "step": 9246
    },
    {
      "epoch": 2.103981797497156,
      "grad_norm": 0.6278377087522803,
      "learning_rate": 7.790027692245383e-07,
      "loss": 0.0035,
      "step": 9247
    },
    {
      "epoch": 2.1042093287827077,
      "grad_norm": 0.5021948878550339,
      "learning_rate": 7.789161712852163e-07,
      "loss": 0.0071,
      "step": 9248
    },
    {
      "epoch": 2.1044368600682595,
      "grad_norm": 0.8056033221753424,
      "learning_rate": 7.788295702001296e-07,
      "loss": 0.0117,
      "step": 9249
    },
    {
      "epoch": 2.1046643913538112,
      "grad_norm": 0.9880212959773031,
      "learning_rate": 7.787429659710483e-07,
      "loss": 0.0133,
      "step": 9250
    },
    {
      "epoch": 2.104891922639363,
      "grad_norm": 0.4645349011452211,
      "learning_rate": 7.786563585997422e-07,
      "loss": 0.0023,
      "step": 9251
    },
    {
      "epoch": 2.1051194539249147,
      "grad_norm": 1.1823375145594603,
      "learning_rate": 7.785697480879817e-07,
      "loss": 0.0082,
      "step": 9252
    },
    {
      "epoch": 2.1053469852104665,
      "grad_norm": 0.32591765668254413,
      "learning_rate": 7.784831344375368e-07,
      "loss": 0.0034,
      "step": 9253
    },
    {
      "epoch": 2.1055745164960182,
      "grad_norm": 1.2687961743078804,
      "learning_rate": 7.783965176501776e-07,
      "loss": 0.0244,
      "step": 9254
    },
    {
      "epoch": 2.10580204778157,
      "grad_norm": 1.433044307027219,
      "learning_rate": 7.783098977276746e-07,
      "loss": 0.0105,
      "step": 9255
    },
    {
      "epoch": 2.1060295790671217,
      "grad_norm": 1.3256650687936837,
      "learning_rate": 7.78223274671798e-07,
      "loss": 0.0312,
      "step": 9256
    },
    {
      "epoch": 2.1062571103526735,
      "grad_norm": 0.5832669016674589,
      "learning_rate": 7.781366484843184e-07,
      "loss": 0.005,
      "step": 9257
    },
    {
      "epoch": 2.1064846416382252,
      "grad_norm": 0.9472246850407402,
      "learning_rate": 7.78050019167006e-07,
      "loss": 0.0058,
      "step": 9258
    },
    {
      "epoch": 2.106712172923777,
      "grad_norm": 0.7969795958517575,
      "learning_rate": 7.779633867216316e-07,
      "loss": 0.0071,
      "step": 9259
    },
    {
      "epoch": 2.1069397042093287,
      "grad_norm": 0.5809977867579027,
      "learning_rate": 7.778767511499657e-07,
      "loss": 0.007,
      "step": 9260
    },
    {
      "epoch": 2.1071672354948805,
      "grad_norm": 1.376158219884132,
      "learning_rate": 7.77790112453779e-07,
      "loss": 0.0161,
      "step": 9261
    },
    {
      "epoch": 2.1073947667804322,
      "grad_norm": 0.8468462694388563,
      "learning_rate": 7.77703470634842e-07,
      "loss": 0.0171,
      "step": 9262
    },
    {
      "epoch": 2.107622298065984,
      "grad_norm": 1.0779180072248953,
      "learning_rate": 7.77616825694926e-07,
      "loss": 0.0151,
      "step": 9263
    },
    {
      "epoch": 2.1078498293515358,
      "grad_norm": 0.4770232661427211,
      "learning_rate": 7.775301776358017e-07,
      "loss": 0.0056,
      "step": 9264
    },
    {
      "epoch": 2.1080773606370875,
      "grad_norm": 0.6523197760944786,
      "learning_rate": 7.774435264592396e-07,
      "loss": 0.0129,
      "step": 9265
    },
    {
      "epoch": 2.1083048919226393,
      "grad_norm": 0.5892460233849306,
      "learning_rate": 7.773568721670111e-07,
      "loss": 0.0075,
      "step": 9266
    },
    {
      "epoch": 2.108532423208191,
      "grad_norm": 0.1882698375940475,
      "learning_rate": 7.772702147608873e-07,
      "loss": 0.0011,
      "step": 9267
    },
    {
      "epoch": 2.1087599544937428,
      "grad_norm": 0.52604619679456,
      "learning_rate": 7.771835542426389e-07,
      "loss": 0.006,
      "step": 9268
    },
    {
      "epoch": 2.1089874857792945,
      "grad_norm": 1.2801995718752124,
      "learning_rate": 7.770968906140376e-07,
      "loss": 0.0178,
      "step": 9269
    },
    {
      "epoch": 2.1092150170648463,
      "grad_norm": 0.7107149098259525,
      "learning_rate": 7.770102238768543e-07,
      "loss": 0.0055,
      "step": 9270
    },
    {
      "epoch": 2.109442548350398,
      "grad_norm": 0.7205424693820719,
      "learning_rate": 7.769235540328607e-07,
      "loss": 0.0082,
      "step": 9271
    },
    {
      "epoch": 2.1096700796359498,
      "grad_norm": 1.2027000820028442,
      "learning_rate": 7.768368810838275e-07,
      "loss": 0.0209,
      "step": 9272
    },
    {
      "epoch": 2.1098976109215015,
      "grad_norm": 0.39160046850220054,
      "learning_rate": 7.767502050315266e-07,
      "loss": 0.0033,
      "step": 9273
    },
    {
      "epoch": 2.1101251422070533,
      "grad_norm": 0.9445899721010399,
      "learning_rate": 7.766635258777296e-07,
      "loss": 0.0133,
      "step": 9274
    },
    {
      "epoch": 2.110352673492605,
      "grad_norm": 0.5626709367148773,
      "learning_rate": 7.765768436242077e-07,
      "loss": 0.002,
      "step": 9275
    },
    {
      "epoch": 2.1105802047781568,
      "grad_norm": 0.48427462422275985,
      "learning_rate": 7.764901582727328e-07,
      "loss": 0.0057,
      "step": 9276
    },
    {
      "epoch": 2.110807736063709,
      "grad_norm": 1.3310285303812137,
      "learning_rate": 7.764034698250767e-07,
      "loss": 0.0311,
      "step": 9277
    },
    {
      "epoch": 2.1110352673492607,
      "grad_norm": 1.027449064610943,
      "learning_rate": 7.763167782830107e-07,
      "loss": 0.0121,
      "step": 9278
    },
    {
      "epoch": 2.1112627986348125,
      "grad_norm": 0.435579428938178,
      "learning_rate": 7.762300836483069e-07,
      "loss": 0.005,
      "step": 9279
    },
    {
      "epoch": 2.1114903299203642,
      "grad_norm": 0.679395307172523,
      "learning_rate": 7.761433859227373e-07,
      "loss": 0.0056,
      "step": 9280
    },
    {
      "epoch": 2.111717861205916,
      "grad_norm": 0.6020128188175906,
      "learning_rate": 7.760566851080736e-07,
      "loss": 0.0032,
      "step": 9281
    },
    {
      "epoch": 2.1119453924914677,
      "grad_norm": 1.5418885483278937,
      "learning_rate": 7.759699812060878e-07,
      "loss": 0.0167,
      "step": 9282
    },
    {
      "epoch": 2.1121729237770195,
      "grad_norm": 0.44315206311217464,
      "learning_rate": 7.758832742185523e-07,
      "loss": 0.0042,
      "step": 9283
    },
    {
      "epoch": 2.1124004550625712,
      "grad_norm": 0.5147215125707401,
      "learning_rate": 7.75796564147239e-07,
      "loss": 0.0078,
      "step": 9284
    },
    {
      "epoch": 2.112627986348123,
      "grad_norm": 1.0508832902452774,
      "learning_rate": 7.757098509939198e-07,
      "loss": 0.0168,
      "step": 9285
    },
    {
      "epoch": 2.1128555176336747,
      "grad_norm": 0.3557205753864511,
      "learning_rate": 7.756231347603674e-07,
      "loss": 0.0024,
      "step": 9286
    },
    {
      "epoch": 2.1130830489192265,
      "grad_norm": 1.2853552341593915,
      "learning_rate": 7.755364154483541e-07,
      "loss": 0.0167,
      "step": 9287
    },
    {
      "epoch": 2.1133105802047782,
      "grad_norm": 0.7996360698496128,
      "learning_rate": 7.754496930596521e-07,
      "loss": 0.0148,
      "step": 9288
    },
    {
      "epoch": 2.11353811149033,
      "grad_norm": 1.2944159932286945,
      "learning_rate": 7.753629675960339e-07,
      "loss": 0.0223,
      "step": 9289
    },
    {
      "epoch": 2.1137656427758817,
      "grad_norm": 0.5226371406326719,
      "learning_rate": 7.752762390592721e-07,
      "loss": 0.0058,
      "step": 9290
    },
    {
      "epoch": 2.1139931740614335,
      "grad_norm": 1.2825230470735336,
      "learning_rate": 7.751895074511391e-07,
      "loss": 0.0235,
      "step": 9291
    },
    {
      "epoch": 2.1142207053469853,
      "grad_norm": 0.9194008058245837,
      "learning_rate": 7.751027727734075e-07,
      "loss": 0.0153,
      "step": 9292
    },
    {
      "epoch": 2.114448236632537,
      "grad_norm": 0.32574968478101046,
      "learning_rate": 7.750160350278503e-07,
      "loss": 0.004,
      "step": 9293
    },
    {
      "epoch": 2.1146757679180888,
      "grad_norm": 1.0432027991771016,
      "learning_rate": 7.7492929421624e-07,
      "loss": 0.0062,
      "step": 9294
    },
    {
      "epoch": 2.1149032992036405,
      "grad_norm": 0.9357758012635514,
      "learning_rate": 7.748425503403497e-07,
      "loss": 0.0177,
      "step": 9295
    },
    {
      "epoch": 2.1151308304891923,
      "grad_norm": 0.45931328768984875,
      "learning_rate": 7.747558034019518e-07,
      "loss": 0.0055,
      "step": 9296
    },
    {
      "epoch": 2.115358361774744,
      "grad_norm": 0.7789332237598385,
      "learning_rate": 7.746690534028198e-07,
      "loss": 0.008,
      "step": 9297
    },
    {
      "epoch": 2.1155858930602958,
      "grad_norm": 0.585121819859683,
      "learning_rate": 7.745823003447262e-07,
      "loss": 0.0028,
      "step": 9298
    },
    {
      "epoch": 2.1158134243458475,
      "grad_norm": 0.5964394322215167,
      "learning_rate": 7.744955442294444e-07,
      "loss": 0.0137,
      "step": 9299
    },
    {
      "epoch": 2.1160409556313993,
      "grad_norm": 0.3205518841094132,
      "learning_rate": 7.744087850587476e-07,
      "loss": 0.0026,
      "step": 9300
    },
    {
      "epoch": 2.116268486916951,
      "grad_norm": 0.5896827856487226,
      "learning_rate": 7.743220228344088e-07,
      "loss": 0.0086,
      "step": 9301
    },
    {
      "epoch": 2.1164960182025028,
      "grad_norm": 0.49452255381969884,
      "learning_rate": 7.742352575582014e-07,
      "loss": 0.0022,
      "step": 9302
    },
    {
      "epoch": 2.1167235494880545,
      "grad_norm": 0.6386684805696516,
      "learning_rate": 7.741484892318986e-07,
      "loss": 0.0065,
      "step": 9303
    },
    {
      "epoch": 2.1169510807736063,
      "grad_norm": 1.1712198529798117,
      "learning_rate": 7.740617178572738e-07,
      "loss": 0.0238,
      "step": 9304
    },
    {
      "epoch": 2.117178612059158,
      "grad_norm": 1.443578022133474,
      "learning_rate": 7.739749434361003e-07,
      "loss": 0.0246,
      "step": 9305
    },
    {
      "epoch": 2.11740614334471,
      "grad_norm": 1.148306127698227,
      "learning_rate": 7.738881659701521e-07,
      "loss": 0.0181,
      "step": 9306
    },
    {
      "epoch": 2.1176336746302615,
      "grad_norm": 0.5081086380563994,
      "learning_rate": 7.738013854612023e-07,
      "loss": 0.003,
      "step": 9307
    },
    {
      "epoch": 2.1178612059158133,
      "grad_norm": 1.09859413339637,
      "learning_rate": 7.737146019110247e-07,
      "loss": 0.0183,
      "step": 9308
    },
    {
      "epoch": 2.118088737201365,
      "grad_norm": 0.6865986011028383,
      "learning_rate": 7.736278153213932e-07,
      "loss": 0.0097,
      "step": 9309
    },
    {
      "epoch": 2.118316268486917,
      "grad_norm": 0.5334779464452993,
      "learning_rate": 7.735410256940812e-07,
      "loss": 0.0013,
      "step": 9310
    },
    {
      "epoch": 2.1185437997724685,
      "grad_norm": 0.8175878440237826,
      "learning_rate": 7.734542330308626e-07,
      "loss": 0.0112,
      "step": 9311
    },
    {
      "epoch": 2.1187713310580203,
      "grad_norm": 0.4008291179836023,
      "learning_rate": 7.733674373335112e-07,
      "loss": 0.006,
      "step": 9312
    },
    {
      "epoch": 2.118998862343572,
      "grad_norm": 0.7555099706421347,
      "learning_rate": 7.732806386038013e-07,
      "loss": 0.0104,
      "step": 9313
    },
    {
      "epoch": 2.119226393629124,
      "grad_norm": 1.8740139859178238,
      "learning_rate": 7.731938368435069e-07,
      "loss": 0.0277,
      "step": 9314
    },
    {
      "epoch": 2.1194539249146755,
      "grad_norm": 0.4870467351113141,
      "learning_rate": 7.731070320544017e-07,
      "loss": 0.0043,
      "step": 9315
    },
    {
      "epoch": 2.1196814562002277,
      "grad_norm": 0.8886276104915918,
      "learning_rate": 7.7302022423826e-07,
      "loss": 0.0145,
      "step": 9316
    },
    {
      "epoch": 2.1199089874857795,
      "grad_norm": 0.4939635956444076,
      "learning_rate": 7.72933413396856e-07,
      "loss": 0.0055,
      "step": 9317
    },
    {
      "epoch": 2.1201365187713312,
      "grad_norm": 0.8002772576797165,
      "learning_rate": 7.728465995319641e-07,
      "loss": 0.009,
      "step": 9318
    },
    {
      "epoch": 2.120364050056883,
      "grad_norm": 1.5504173847797766,
      "learning_rate": 7.727597826453583e-07,
      "loss": 0.0199,
      "step": 9319
    },
    {
      "epoch": 2.1205915813424348,
      "grad_norm": 0.601728472782442,
      "learning_rate": 7.726729627388132e-07,
      "loss": 0.0086,
      "step": 9320
    },
    {
      "epoch": 2.1208191126279865,
      "grad_norm": 1.2178644264360978,
      "learning_rate": 7.725861398141032e-07,
      "loss": 0.0095,
      "step": 9321
    },
    {
      "epoch": 2.1210466439135383,
      "grad_norm": 0.99613089080032,
      "learning_rate": 7.724993138730027e-07,
      "loss": 0.0163,
      "step": 9322
    },
    {
      "epoch": 2.12127417519909,
      "grad_norm": 0.7812678431584936,
      "learning_rate": 7.724124849172865e-07,
      "loss": 0.0155,
      "step": 9323
    },
    {
      "epoch": 2.1215017064846418,
      "grad_norm": 0.21368210456392514,
      "learning_rate": 7.723256529487291e-07,
      "loss": 0.0014,
      "step": 9324
    },
    {
      "epoch": 2.1217292377701935,
      "grad_norm": 0.5802091982439473,
      "learning_rate": 7.722388179691051e-07,
      "loss": 0.0075,
      "step": 9325
    },
    {
      "epoch": 2.1219567690557453,
      "grad_norm": 1.0755789840402012,
      "learning_rate": 7.721519799801892e-07,
      "loss": 0.0273,
      "step": 9326
    },
    {
      "epoch": 2.122184300341297,
      "grad_norm": 0.8044716006135214,
      "learning_rate": 7.720651389837566e-07,
      "loss": 0.0122,
      "step": 9327
    },
    {
      "epoch": 2.1224118316268488,
      "grad_norm": 0.9656814876357248,
      "learning_rate": 7.719782949815817e-07,
      "loss": 0.0107,
      "step": 9328
    },
    {
      "epoch": 2.1226393629124005,
      "grad_norm": 0.6167739226303055,
      "learning_rate": 7.718914479754395e-07,
      "loss": 0.0067,
      "step": 9329
    },
    {
      "epoch": 2.1228668941979523,
      "grad_norm": 0.6281965997313745,
      "learning_rate": 7.718045979671054e-07,
      "loss": 0.007,
      "step": 9330
    },
    {
      "epoch": 2.123094425483504,
      "grad_norm": 1.2644289472661194,
      "learning_rate": 7.717177449583541e-07,
      "loss": 0.0122,
      "step": 9331
    },
    {
      "epoch": 2.1233219567690558,
      "grad_norm": 0.7532418971080128,
      "learning_rate": 7.716308889509608e-07,
      "loss": 0.0134,
      "step": 9332
    },
    {
      "epoch": 2.1235494880546075,
      "grad_norm": 1.3621156306514859,
      "learning_rate": 7.715440299467006e-07,
      "loss": 0.0124,
      "step": 9333
    },
    {
      "epoch": 2.1237770193401593,
      "grad_norm": 0.4543555285791925,
      "learning_rate": 7.714571679473489e-07,
      "loss": 0.0058,
      "step": 9334
    },
    {
      "epoch": 2.124004550625711,
      "grad_norm": 0.7671567799229786,
      "learning_rate": 7.713703029546809e-07,
      "loss": 0.0095,
      "step": 9335
    },
    {
      "epoch": 2.124232081911263,
      "grad_norm": 0.43296493773638894,
      "learning_rate": 7.712834349704718e-07,
      "loss": 0.0038,
      "step": 9336
    },
    {
      "epoch": 2.1244596131968145,
      "grad_norm": 0.5458144248235711,
      "learning_rate": 7.711965639964971e-07,
      "loss": 0.0041,
      "step": 9337
    },
    {
      "epoch": 2.1246871444823663,
      "grad_norm": 0.42708321909161506,
      "learning_rate": 7.711096900345327e-07,
      "loss": 0.0032,
      "step": 9338
    },
    {
      "epoch": 2.124914675767918,
      "grad_norm": 0.49979807660715214,
      "learning_rate": 7.710228130863537e-07,
      "loss": 0.0043,
      "step": 9339
    },
    {
      "epoch": 2.12514220705347,
      "grad_norm": 0.7726907736551077,
      "learning_rate": 7.70935933153736e-07,
      "loss": 0.0184,
      "step": 9340
    },
    {
      "epoch": 2.1253697383390215,
      "grad_norm": 0.5235571073448476,
      "learning_rate": 7.70849050238455e-07,
      "loss": 0.0058,
      "step": 9341
    },
    {
      "epoch": 2.1255972696245733,
      "grad_norm": 0.7968690006044425,
      "learning_rate": 7.707621643422862e-07,
      "loss": 0.0054,
      "step": 9342
    },
    {
      "epoch": 2.125824800910125,
      "grad_norm": 1.1544453770762437,
      "learning_rate": 7.706752754670061e-07,
      "loss": 0.0189,
      "step": 9343
    },
    {
      "epoch": 2.126052332195677,
      "grad_norm": 1.308513866531729,
      "learning_rate": 7.7058838361439e-07,
      "loss": 0.0097,
      "step": 9344
    },
    {
      "epoch": 2.1262798634812285,
      "grad_norm": 0.6257116060187328,
      "learning_rate": 7.70501488786214e-07,
      "loss": 0.0053,
      "step": 9345
    },
    {
      "epoch": 2.1265073947667803,
      "grad_norm": 0.730755860799214,
      "learning_rate": 7.70414590984254e-07,
      "loss": 0.0078,
      "step": 9346
    },
    {
      "epoch": 2.126734926052332,
      "grad_norm": 1.922894797173844,
      "learning_rate": 7.703276902102863e-07,
      "loss": 0.0295,
      "step": 9347
    },
    {
      "epoch": 2.126962457337884,
      "grad_norm": 1.090601578418276,
      "learning_rate": 7.702407864660865e-07,
      "loss": 0.0124,
      "step": 9348
    },
    {
      "epoch": 2.1271899886234356,
      "grad_norm": 0.8431305968754933,
      "learning_rate": 7.701538797534311e-07,
      "loss": 0.0076,
      "step": 9349
    },
    {
      "epoch": 2.1274175199089873,
      "grad_norm": 0.6170441599759381,
      "learning_rate": 7.700669700740962e-07,
      "loss": 0.0038,
      "step": 9350
    },
    {
      "epoch": 2.127645051194539,
      "grad_norm": 0.6726246066713302,
      "learning_rate": 7.699800574298582e-07,
      "loss": 0.0059,
      "step": 9351
    },
    {
      "epoch": 2.127872582480091,
      "grad_norm": 0.7135448306323023,
      "learning_rate": 7.698931418224934e-07,
      "loss": 0.0032,
      "step": 9352
    },
    {
      "epoch": 2.128100113765643,
      "grad_norm": 0.40119783725346897,
      "learning_rate": 7.69806223253778e-07,
      "loss": 0.0032,
      "step": 9353
    },
    {
      "epoch": 2.1283276450511943,
      "grad_norm": 0.7506531546338718,
      "learning_rate": 7.697193017254886e-07,
      "loss": 0.0187,
      "step": 9354
    },
    {
      "epoch": 2.1285551763367465,
      "grad_norm": 0.7292817188394103,
      "learning_rate": 7.696323772394018e-07,
      "loss": 0.009,
      "step": 9355
    },
    {
      "epoch": 2.1287827076222983,
      "grad_norm": 1.2231787447525568,
      "learning_rate": 7.69545449797294e-07,
      "loss": 0.0112,
      "step": 9356
    },
    {
      "epoch": 2.12901023890785,
      "grad_norm": 1.4078722402329749,
      "learning_rate": 7.694585194009419e-07,
      "loss": 0.0248,
      "step": 9357
    },
    {
      "epoch": 2.1292377701934018,
      "grad_norm": 1.241126896304744,
      "learning_rate": 7.693715860521224e-07,
      "loss": 0.0197,
      "step": 9358
    },
    {
      "epoch": 2.1294653014789535,
      "grad_norm": 0.958743737993354,
      "learning_rate": 7.69284649752612e-07,
      "loss": 0.0047,
      "step": 9359
    },
    {
      "epoch": 2.1296928327645053,
      "grad_norm": 0.3864131477070798,
      "learning_rate": 7.691977105041876e-07,
      "loss": 0.003,
      "step": 9360
    },
    {
      "epoch": 2.129920364050057,
      "grad_norm": 0.5648643571031741,
      "learning_rate": 7.69110768308626e-07,
      "loss": 0.0068,
      "step": 9361
    },
    {
      "epoch": 2.1301478953356088,
      "grad_norm": 0.6800766992493664,
      "learning_rate": 7.690238231677045e-07,
      "loss": 0.0108,
      "step": 9362
    },
    {
      "epoch": 2.1303754266211605,
      "grad_norm": 0.44346652887854204,
      "learning_rate": 7.689368750831994e-07,
      "loss": 0.0021,
      "step": 9363
    },
    {
      "epoch": 2.1306029579067123,
      "grad_norm": 0.7125435430441509,
      "learning_rate": 7.688499240568883e-07,
      "loss": 0.0098,
      "step": 9364
    },
    {
      "epoch": 2.130830489192264,
      "grad_norm": 1.026299881027326,
      "learning_rate": 7.687629700905485e-07,
      "loss": 0.0083,
      "step": 9365
    },
    {
      "epoch": 2.131058020477816,
      "grad_norm": 0.47670964321392373,
      "learning_rate": 7.686760131859566e-07,
      "loss": 0.0047,
      "step": 9366
    },
    {
      "epoch": 2.1312855517633675,
      "grad_norm": 0.6884814697189213,
      "learning_rate": 7.685890533448901e-07,
      "loss": 0.0097,
      "step": 9367
    },
    {
      "epoch": 2.1315130830489193,
      "grad_norm": 0.7542648522616061,
      "learning_rate": 7.685020905691265e-07,
      "loss": 0.0086,
      "step": 9368
    },
    {
      "epoch": 2.131740614334471,
      "grad_norm": 0.7224628820415935,
      "learning_rate": 7.684151248604428e-07,
      "loss": 0.0112,
      "step": 9369
    },
    {
      "epoch": 2.131968145620023,
      "grad_norm": 0.6788415249222343,
      "learning_rate": 7.683281562206167e-07,
      "loss": 0.0087,
      "step": 9370
    },
    {
      "epoch": 2.1321956769055745,
      "grad_norm": 0.7639085976718167,
      "learning_rate": 7.682411846514258e-07,
      "loss": 0.0101,
      "step": 9371
    },
    {
      "epoch": 2.1324232081911263,
      "grad_norm": 1.6322301282997367,
      "learning_rate": 7.68154210154647e-07,
      "loss": 0.0231,
      "step": 9372
    },
    {
      "epoch": 2.132650739476678,
      "grad_norm": 1.2027520015285638,
      "learning_rate": 7.680672327320586e-07,
      "loss": 0.0148,
      "step": 9373
    },
    {
      "epoch": 2.13287827076223,
      "grad_norm": 0.5244073897194097,
      "learning_rate": 7.679802523854379e-07,
      "loss": 0.0063,
      "step": 9374
    },
    {
      "epoch": 2.1331058020477816,
      "grad_norm": 0.7908598612710436,
      "learning_rate": 7.678932691165628e-07,
      "loss": 0.0129,
      "step": 9375
    },
    {
      "epoch": 2.1333333333333333,
      "grad_norm": 0.6648253075564049,
      "learning_rate": 7.678062829272107e-07,
      "loss": 0.0093,
      "step": 9376
    },
    {
      "epoch": 2.133560864618885,
      "grad_norm": 0.9794199826108704,
      "learning_rate": 7.677192938191599e-07,
      "loss": 0.0111,
      "step": 9377
    },
    {
      "epoch": 2.133788395904437,
      "grad_norm": 0.6293850966699488,
      "learning_rate": 7.676323017941882e-07,
      "loss": 0.0098,
      "step": 9378
    },
    {
      "epoch": 2.1340159271899886,
      "grad_norm": 1.644736601726369,
      "learning_rate": 7.675453068540733e-07,
      "loss": 0.0348,
      "step": 9379
    },
    {
      "epoch": 2.1342434584755403,
      "grad_norm": 1.0251249555513355,
      "learning_rate": 7.674583090005934e-07,
      "loss": 0.0166,
      "step": 9380
    },
    {
      "epoch": 2.134470989761092,
      "grad_norm": 0.4773574826145558,
      "learning_rate": 7.673713082355267e-07,
      "loss": 0.0041,
      "step": 9381
    },
    {
      "epoch": 2.134698521046644,
      "grad_norm": 1.3226375035547333,
      "learning_rate": 7.672843045606511e-07,
      "loss": 0.0196,
      "step": 9382
    },
    {
      "epoch": 2.1349260523321956,
      "grad_norm": 0.4583698966991122,
      "learning_rate": 7.671972979777448e-07,
      "loss": 0.0053,
      "step": 9383
    },
    {
      "epoch": 2.1351535836177473,
      "grad_norm": 0.7776463167696871,
      "learning_rate": 7.671102884885865e-07,
      "loss": 0.0181,
      "step": 9384
    },
    {
      "epoch": 2.135381114903299,
      "grad_norm": 1.2305453054299074,
      "learning_rate": 7.67023276094954e-07,
      "loss": 0.0112,
      "step": 9385
    },
    {
      "epoch": 2.135608646188851,
      "grad_norm": 0.48346899208308736,
      "learning_rate": 7.669362607986256e-07,
      "loss": 0.0053,
      "step": 9386
    },
    {
      "epoch": 2.1358361774744026,
      "grad_norm": 0.4860523191118012,
      "learning_rate": 7.668492426013802e-07,
      "loss": 0.0053,
      "step": 9387
    },
    {
      "epoch": 2.1360637087599543,
      "grad_norm": 0.9433358046702833,
      "learning_rate": 7.667622215049959e-07,
      "loss": 0.0123,
      "step": 9388
    },
    {
      "epoch": 2.136291240045506,
      "grad_norm": 0.7274294117541075,
      "learning_rate": 7.666751975112515e-07,
      "loss": 0.0095,
      "step": 9389
    },
    {
      "epoch": 2.136518771331058,
      "grad_norm": 1.0546542084991488,
      "learning_rate": 7.665881706219256e-07,
      "loss": 0.0233,
      "step": 9390
    },
    {
      "epoch": 2.1367463026166096,
      "grad_norm": 0.882509401051961,
      "learning_rate": 7.665011408387968e-07,
      "loss": 0.0102,
      "step": 9391
    },
    {
      "epoch": 2.136973833902162,
      "grad_norm": 0.5578384978853934,
      "learning_rate": 7.664141081636438e-07,
      "loss": 0.0079,
      "step": 9392
    },
    {
      "epoch": 2.137201365187713,
      "grad_norm": 0.3448317807599905,
      "learning_rate": 7.663270725982452e-07,
      "loss": 0.0021,
      "step": 9393
    },
    {
      "epoch": 2.1374288964732653,
      "grad_norm": 3.8326467239135615,
      "learning_rate": 7.662400341443804e-07,
      "loss": 0.031,
      "step": 9394
    },
    {
      "epoch": 2.137656427758817,
      "grad_norm": 0.3455373445440017,
      "learning_rate": 7.661529928038277e-07,
      "loss": 0.0031,
      "step": 9395
    },
    {
      "epoch": 2.137883959044369,
      "grad_norm": 1.7781400920219406,
      "learning_rate": 7.660659485783667e-07,
      "loss": 0.0192,
      "step": 9396
    },
    {
      "epoch": 2.1381114903299205,
      "grad_norm": 0.31969138424410454,
      "learning_rate": 7.659789014697758e-07,
      "loss": 0.0038,
      "step": 9397
    },
    {
      "epoch": 2.1383390216154723,
      "grad_norm": 0.9760789003130663,
      "learning_rate": 7.658918514798345e-07,
      "loss": 0.007,
      "step": 9398
    },
    {
      "epoch": 2.138566552901024,
      "grad_norm": 1.201389914321695,
      "learning_rate": 7.658047986103215e-07,
      "loss": 0.0175,
      "step": 9399
    },
    {
      "epoch": 2.138794084186576,
      "grad_norm": 0.6620998699159607,
      "learning_rate": 7.657177428630164e-07,
      "loss": 0.0122,
      "step": 9400
    },
    {
      "epoch": 2.1390216154721275,
      "grad_norm": 0.5768939443316081,
      "learning_rate": 7.656306842396985e-07,
      "loss": 0.0026,
      "step": 9401
    },
    {
      "epoch": 2.1392491467576793,
      "grad_norm": 0.4747361814852938,
      "learning_rate": 7.655436227421468e-07,
      "loss": 0.0087,
      "step": 9402
    },
    {
      "epoch": 2.139476678043231,
      "grad_norm": 4.4082218686791395,
      "learning_rate": 7.65456558372141e-07,
      "loss": 0.0753,
      "step": 9403
    },
    {
      "epoch": 2.139704209328783,
      "grad_norm": 0.6337172476139666,
      "learning_rate": 7.653694911314602e-07,
      "loss": 0.006,
      "step": 9404
    },
    {
      "epoch": 2.1399317406143346,
      "grad_norm": 0.7778276823461497,
      "learning_rate": 7.652824210218842e-07,
      "loss": 0.0044,
      "step": 9405
    },
    {
      "epoch": 2.1401592718998863,
      "grad_norm": 0.742826462321095,
      "learning_rate": 7.651953480451923e-07,
      "loss": 0.0047,
      "step": 9406
    },
    {
      "epoch": 2.140386803185438,
      "grad_norm": 0.6461920198683603,
      "learning_rate": 7.651082722031644e-07,
      "loss": 0.0101,
      "step": 9407
    },
    {
      "epoch": 2.14061433447099,
      "grad_norm": 0.4830927720119314,
      "learning_rate": 7.6502119349758e-07,
      "loss": 0.0058,
      "step": 9408
    },
    {
      "epoch": 2.1408418657565416,
      "grad_norm": 0.5529198464070699,
      "learning_rate": 7.649341119302188e-07,
      "loss": 0.0019,
      "step": 9409
    },
    {
      "epoch": 2.1410693970420933,
      "grad_norm": 0.39647907236481306,
      "learning_rate": 7.648470275028607e-07,
      "loss": 0.0037,
      "step": 9410
    },
    {
      "epoch": 2.141296928327645,
      "grad_norm": 0.6657325208626473,
      "learning_rate": 7.647599402172854e-07,
      "loss": 0.0104,
      "step": 9411
    },
    {
      "epoch": 2.141524459613197,
      "grad_norm": 2.7179501593018385,
      "learning_rate": 7.646728500752729e-07,
      "loss": 0.0165,
      "step": 9412
    },
    {
      "epoch": 2.1417519908987486,
      "grad_norm": 0.32904954642977885,
      "learning_rate": 7.645857570786029e-07,
      "loss": 0.0031,
      "step": 9413
    },
    {
      "epoch": 2.1419795221843003,
      "grad_norm": 0.5212837342866012,
      "learning_rate": 7.64498661229056e-07,
      "loss": 0.0118,
      "step": 9414
    },
    {
      "epoch": 2.142207053469852,
      "grad_norm": 0.43073017368918903,
      "learning_rate": 7.644115625284119e-07,
      "loss": 0.0035,
      "step": 9415
    },
    {
      "epoch": 2.142434584755404,
      "grad_norm": 0.822263994599402,
      "learning_rate": 7.643244609784506e-07,
      "loss": 0.0079,
      "step": 9416
    },
    {
      "epoch": 2.1426621160409556,
      "grad_norm": 0.27185301200030915,
      "learning_rate": 7.642373565809527e-07,
      "loss": 0.0019,
      "step": 9417
    },
    {
      "epoch": 2.1428896473265073,
      "grad_norm": 0.3921986137982507,
      "learning_rate": 7.641502493376981e-07,
      "loss": 0.0025,
      "step": 9418
    },
    {
      "epoch": 2.143117178612059,
      "grad_norm": 0.7397915172226301,
      "learning_rate": 7.640631392504673e-07,
      "loss": 0.0083,
      "step": 9419
    },
    {
      "epoch": 2.143344709897611,
      "grad_norm": 0.4658356579199857,
      "learning_rate": 7.639760263210405e-07,
      "loss": 0.0084,
      "step": 9420
    },
    {
      "epoch": 2.1435722411831626,
      "grad_norm": 0.7949755028655057,
      "learning_rate": 7.638889105511983e-07,
      "loss": 0.0085,
      "step": 9421
    },
    {
      "epoch": 2.1437997724687143,
      "grad_norm": 0.9932434767294664,
      "learning_rate": 7.638017919427212e-07,
      "loss": 0.0153,
      "step": 9422
    },
    {
      "epoch": 2.144027303754266,
      "grad_norm": 1.0447557791765516,
      "learning_rate": 7.637146704973897e-07,
      "loss": 0.0087,
      "step": 9423
    },
    {
      "epoch": 2.144254835039818,
      "grad_norm": 1.5602769173885682,
      "learning_rate": 7.636275462169843e-07,
      "loss": 0.0107,
      "step": 9424
    },
    {
      "epoch": 2.1444823663253696,
      "grad_norm": 0.8175983751576256,
      "learning_rate": 7.635404191032858e-07,
      "loss": 0.0078,
      "step": 9425
    },
    {
      "epoch": 2.1447098976109213,
      "grad_norm": 0.7162127814674661,
      "learning_rate": 7.634532891580748e-07,
      "loss": 0.0085,
      "step": 9426
    },
    {
      "epoch": 2.144937428896473,
      "grad_norm": 0.435753980256996,
      "learning_rate": 7.633661563831321e-07,
      "loss": 0.0063,
      "step": 9427
    },
    {
      "epoch": 2.145164960182025,
      "grad_norm": 0.4886704090763667,
      "learning_rate": 7.632790207802388e-07,
      "loss": 0.0023,
      "step": 9428
    },
    {
      "epoch": 2.1453924914675766,
      "grad_norm": 0.6109087044132886,
      "learning_rate": 7.631918823511751e-07,
      "loss": 0.007,
      "step": 9429
    },
    {
      "epoch": 2.1456200227531284,
      "grad_norm": 0.3862693252752087,
      "learning_rate": 7.631047410977227e-07,
      "loss": 0.004,
      "step": 9430
    },
    {
      "epoch": 2.1458475540386805,
      "grad_norm": 0.3916971112441679,
      "learning_rate": 7.630175970216625e-07,
      "loss": 0.0053,
      "step": 9431
    },
    {
      "epoch": 2.146075085324232,
      "grad_norm": 1.3442295458084064,
      "learning_rate": 7.629304501247751e-07,
      "loss": 0.0105,
      "step": 9432
    },
    {
      "epoch": 2.146302616609784,
      "grad_norm": 1.5551974523245196,
      "learning_rate": 7.628433004088419e-07,
      "loss": 0.0226,
      "step": 9433
    },
    {
      "epoch": 2.146530147895336,
      "grad_norm": 0.6491097255442313,
      "learning_rate": 7.627561478756443e-07,
      "loss": 0.0096,
      "step": 9434
    },
    {
      "epoch": 2.1467576791808876,
      "grad_norm": 0.28904704944060683,
      "learning_rate": 7.626689925269633e-07,
      "loss": 0.0013,
      "step": 9435
    },
    {
      "epoch": 2.1469852104664393,
      "grad_norm": 0.4286314484617513,
      "learning_rate": 7.625818343645799e-07,
      "loss": 0.0022,
      "step": 9436
    },
    {
      "epoch": 2.147212741751991,
      "grad_norm": 0.9002201977623425,
      "learning_rate": 7.624946733902762e-07,
      "loss": 0.0184,
      "step": 9437
    },
    {
      "epoch": 2.147440273037543,
      "grad_norm": 1.4108767985987805,
      "learning_rate": 7.624075096058329e-07,
      "loss": 0.0055,
      "step": 9438
    },
    {
      "epoch": 2.1476678043230946,
      "grad_norm": 0.5736247065532574,
      "learning_rate": 7.623203430130319e-07,
      "loss": 0.0116,
      "step": 9439
    },
    {
      "epoch": 2.1478953356086463,
      "grad_norm": 0.5929807473323365,
      "learning_rate": 7.622331736136546e-07,
      "loss": 0.0043,
      "step": 9440
    },
    {
      "epoch": 2.148122866894198,
      "grad_norm": 0.7999228166052934,
      "learning_rate": 7.621460014094825e-07,
      "loss": 0.0071,
      "step": 9441
    },
    {
      "epoch": 2.14835039817975,
      "grad_norm": 1.4416999673958641,
      "learning_rate": 7.620588264022973e-07,
      "loss": 0.0169,
      "step": 9442
    },
    {
      "epoch": 2.1485779294653016,
      "grad_norm": 0.7268189408978283,
      "learning_rate": 7.619716485938805e-07,
      "loss": 0.0148,
      "step": 9443
    },
    {
      "epoch": 2.1488054607508533,
      "grad_norm": 1.3340670521142381,
      "learning_rate": 7.618844679860144e-07,
      "loss": 0.0115,
      "step": 9444
    },
    {
      "epoch": 2.149032992036405,
      "grad_norm": 0.8489110923475581,
      "learning_rate": 7.617972845804802e-07,
      "loss": 0.007,
      "step": 9445
    },
    {
      "epoch": 2.149260523321957,
      "grad_norm": 0.33573283808619636,
      "learning_rate": 7.617100983790603e-07,
      "loss": 0.0021,
      "step": 9446
    },
    {
      "epoch": 2.1494880546075086,
      "grad_norm": 0.9083438426522049,
      "learning_rate": 7.616229093835361e-07,
      "loss": 0.0089,
      "step": 9447
    },
    {
      "epoch": 2.1497155858930603,
      "grad_norm": 0.907124212698418,
      "learning_rate": 7.6153571759569e-07,
      "loss": 0.0131,
      "step": 9448
    },
    {
      "epoch": 2.149943117178612,
      "grad_norm": 0.7957411004808628,
      "learning_rate": 7.61448523017304e-07,
      "loss": 0.0155,
      "step": 9449
    },
    {
      "epoch": 2.150170648464164,
      "grad_norm": 1.116334684654861,
      "learning_rate": 7.613613256501598e-07,
      "loss": 0.0064,
      "step": 9450
    },
    {
      "epoch": 2.1503981797497156,
      "grad_norm": 0.781080877759642,
      "learning_rate": 7.612741254960398e-07,
      "loss": 0.0048,
      "step": 9451
    },
    {
      "epoch": 2.1506257110352673,
      "grad_norm": 0.4434069545080247,
      "learning_rate": 7.611869225567266e-07,
      "loss": 0.0088,
      "step": 9452
    },
    {
      "epoch": 2.150853242320819,
      "grad_norm": 0.7341899066169435,
      "learning_rate": 7.610997168340019e-07,
      "loss": 0.0106,
      "step": 9453
    },
    {
      "epoch": 2.151080773606371,
      "grad_norm": 0.4168473038479732,
      "learning_rate": 7.610125083296482e-07,
      "loss": 0.0028,
      "step": 9454
    },
    {
      "epoch": 2.1513083048919226,
      "grad_norm": 0.5004702961294094,
      "learning_rate": 7.609252970454481e-07,
      "loss": 0.0072,
      "step": 9455
    },
    {
      "epoch": 2.1515358361774743,
      "grad_norm": 0.7117155550645661,
      "learning_rate": 7.608380829831838e-07,
      "loss": 0.0048,
      "step": 9456
    },
    {
      "epoch": 2.151763367463026,
      "grad_norm": 0.678480481404826,
      "learning_rate": 7.607508661446378e-07,
      "loss": 0.0074,
      "step": 9457
    },
    {
      "epoch": 2.151990898748578,
      "grad_norm": 21.208307356330284,
      "learning_rate": 7.606636465315927e-07,
      "loss": 0.105,
      "step": 9458
    },
    {
      "epoch": 2.1522184300341296,
      "grad_norm": 0.3955596626341846,
      "learning_rate": 7.605764241458313e-07,
      "loss": 0.0029,
      "step": 9459
    },
    {
      "epoch": 2.1524459613196814,
      "grad_norm": 0.4107454918314581,
      "learning_rate": 7.604891989891358e-07,
      "loss": 0.0039,
      "step": 9460
    },
    {
      "epoch": 2.152673492605233,
      "grad_norm": 0.3899303997658839,
      "learning_rate": 7.604019710632895e-07,
      "loss": 0.0024,
      "step": 9461
    },
    {
      "epoch": 2.152901023890785,
      "grad_norm": 0.7710664044298258,
      "learning_rate": 7.603147403700746e-07,
      "loss": 0.0088,
      "step": 9462
    },
    {
      "epoch": 2.1531285551763366,
      "grad_norm": 0.47473637556937476,
      "learning_rate": 7.602275069112742e-07,
      "loss": 0.0027,
      "step": 9463
    },
    {
      "epoch": 2.1533560864618884,
      "grad_norm": 0.6280469293683831,
      "learning_rate": 7.601402706886716e-07,
      "loss": 0.0118,
      "step": 9464
    },
    {
      "epoch": 2.15358361774744,
      "grad_norm": 2.528224730827772,
      "learning_rate": 7.60053031704049e-07,
      "loss": 0.0279,
      "step": 9465
    },
    {
      "epoch": 2.153811149032992,
      "grad_norm": 0.8247841083927588,
      "learning_rate": 7.5996578995919e-07,
      "loss": 0.0075,
      "step": 9466
    },
    {
      "epoch": 2.1540386803185436,
      "grad_norm": 0.5704115365180658,
      "learning_rate": 7.598785454558773e-07,
      "loss": 0.0052,
      "step": 9467
    },
    {
      "epoch": 2.1542662116040954,
      "grad_norm": 2.3222450091342335,
      "learning_rate": 7.597912981958943e-07,
      "loss": 0.0413,
      "step": 9468
    },
    {
      "epoch": 2.154493742889647,
      "grad_norm": 0.5098409311036662,
      "learning_rate": 7.597040481810239e-07,
      "loss": 0.0063,
      "step": 9469
    },
    {
      "epoch": 2.1547212741751993,
      "grad_norm": 1.2227569097380357,
      "learning_rate": 7.596167954130493e-07,
      "loss": 0.0288,
      "step": 9470
    },
    {
      "epoch": 2.1549488054607506,
      "grad_norm": 0.8736975235883835,
      "learning_rate": 7.595295398937541e-07,
      "loss": 0.0079,
      "step": 9471
    },
    {
      "epoch": 2.155176336746303,
      "grad_norm": 0.4426169456378669,
      "learning_rate": 7.594422816249217e-07,
      "loss": 0.0052,
      "step": 9472
    },
    {
      "epoch": 2.1554038680318546,
      "grad_norm": 0.7438309941974948,
      "learning_rate": 7.59355020608335e-07,
      "loss": 0.0122,
      "step": 9473
    },
    {
      "epoch": 2.1556313993174063,
      "grad_norm": 0.8106342731576935,
      "learning_rate": 7.592677568457778e-07,
      "loss": 0.0131,
      "step": 9474
    },
    {
      "epoch": 2.155858930602958,
      "grad_norm": 0.49358495814785197,
      "learning_rate": 7.591804903390336e-07,
      "loss": 0.0048,
      "step": 9475
    },
    {
      "epoch": 2.15608646188851,
      "grad_norm": 0.4796019135160445,
      "learning_rate": 7.590932210898859e-07,
      "loss": 0.0077,
      "step": 9476
    },
    {
      "epoch": 2.1563139931740616,
      "grad_norm": 1.0830714330161306,
      "learning_rate": 7.590059491001183e-07,
      "loss": 0.0048,
      "step": 9477
    },
    {
      "epoch": 2.1565415244596133,
      "grad_norm": 0.438982850407974,
      "learning_rate": 7.589186743715146e-07,
      "loss": 0.005,
      "step": 9478
    },
    {
      "epoch": 2.156769055745165,
      "grad_norm": 0.9225043810448496,
      "learning_rate": 7.588313969058584e-07,
      "loss": 0.01,
      "step": 9479
    },
    {
      "epoch": 2.156996587030717,
      "grad_norm": 1.43802230176175,
      "learning_rate": 7.587441167049335e-07,
      "loss": 0.0192,
      "step": 9480
    },
    {
      "epoch": 2.1572241183162686,
      "grad_norm": 0.6073054248035636,
      "learning_rate": 7.586568337705239e-07,
      "loss": 0.0082,
      "step": 9481
    },
    {
      "epoch": 2.1574516496018203,
      "grad_norm": 0.8121646376322118,
      "learning_rate": 7.585695481044133e-07,
      "loss": 0.0098,
      "step": 9482
    },
    {
      "epoch": 2.157679180887372,
      "grad_norm": 1.2052420742230894,
      "learning_rate": 7.584822597083859e-07,
      "loss": 0.0107,
      "step": 9483
    },
    {
      "epoch": 2.157906712172924,
      "grad_norm": 0.539829338418699,
      "learning_rate": 7.583949685842253e-07,
      "loss": 0.0086,
      "step": 9484
    },
    {
      "epoch": 2.1581342434584756,
      "grad_norm": 2.1410125457119964,
      "learning_rate": 7.583076747337162e-07,
      "loss": 0.0142,
      "step": 9485
    },
    {
      "epoch": 2.1583617747440274,
      "grad_norm": 1.045962862510021,
      "learning_rate": 7.582203781586422e-07,
      "loss": 0.0165,
      "step": 9486
    },
    {
      "epoch": 2.158589306029579,
      "grad_norm": 0.8260434410402252,
      "learning_rate": 7.581330788607875e-07,
      "loss": 0.0109,
      "step": 9487
    },
    {
      "epoch": 2.158816837315131,
      "grad_norm": 0.56108076832216,
      "learning_rate": 7.580457768419367e-07,
      "loss": 0.0061,
      "step": 9488
    },
    {
      "epoch": 2.1590443686006826,
      "grad_norm": 1.2999028206144865,
      "learning_rate": 7.579584721038738e-07,
      "loss": 0.0142,
      "step": 9489
    },
    {
      "epoch": 2.1592718998862344,
      "grad_norm": 0.9635014287836223,
      "learning_rate": 7.578711646483832e-07,
      "loss": 0.0115,
      "step": 9490
    },
    {
      "epoch": 2.159499431171786,
      "grad_norm": 0.8058367882164231,
      "learning_rate": 7.577838544772495e-07,
      "loss": 0.0123,
      "step": 9491
    },
    {
      "epoch": 2.159726962457338,
      "grad_norm": 1.5356278393841314,
      "learning_rate": 7.576965415922569e-07,
      "loss": 0.0188,
      "step": 9492
    },
    {
      "epoch": 2.1599544937428896,
      "grad_norm": 0.9017611836523003,
      "learning_rate": 7.576092259951899e-07,
      "loss": 0.0068,
      "step": 9493
    },
    {
      "epoch": 2.1601820250284414,
      "grad_norm": 1.0158966242457275,
      "learning_rate": 7.575219076878332e-07,
      "loss": 0.0098,
      "step": 9494
    },
    {
      "epoch": 2.160409556313993,
      "grad_norm": 1.1609277837198713,
      "learning_rate": 7.574345866719715e-07,
      "loss": 0.0136,
      "step": 9495
    },
    {
      "epoch": 2.160637087599545,
      "grad_norm": 1.0316834767383807,
      "learning_rate": 7.573472629493894e-07,
      "loss": 0.0051,
      "step": 9496
    },
    {
      "epoch": 2.1608646188850966,
      "grad_norm": 0.7475939929931336,
      "learning_rate": 7.572599365218716e-07,
      "loss": 0.0031,
      "step": 9497
    },
    {
      "epoch": 2.1610921501706484,
      "grad_norm": 1.674687281291756,
      "learning_rate": 7.571726073912031e-07,
      "loss": 0.0347,
      "step": 9498
    },
    {
      "epoch": 2.1613196814562,
      "grad_norm": 1.7016842706971795,
      "learning_rate": 7.570852755591683e-07,
      "loss": 0.0239,
      "step": 9499
    },
    {
      "epoch": 2.161547212741752,
      "grad_norm": 2.3793445926790495,
      "learning_rate": 7.569979410275525e-07,
      "loss": 0.0543,
      "step": 9500
    },
    {
      "epoch": 2.1617747440273036,
      "grad_norm": 0.42685115386991784,
      "learning_rate": 7.569106037981403e-07,
      "loss": 0.0025,
      "step": 9501
    },
    {
      "epoch": 2.1620022753128554,
      "grad_norm": 0.25006824516898746,
      "learning_rate": 7.568232638727173e-07,
      "loss": 0.001,
      "step": 9502
    },
    {
      "epoch": 2.162229806598407,
      "grad_norm": 0.5397028423357909,
      "learning_rate": 7.56735921253068e-07,
      "loss": 0.0052,
      "step": 9503
    },
    {
      "epoch": 2.162457337883959,
      "grad_norm": 1.272527147351978,
      "learning_rate": 7.566485759409778e-07,
      "loss": 0.0226,
      "step": 9504
    },
    {
      "epoch": 2.1626848691695106,
      "grad_norm": 1.2505588806487877,
      "learning_rate": 7.565612279382318e-07,
      "loss": 0.0278,
      "step": 9505
    },
    {
      "epoch": 2.1629124004550624,
      "grad_norm": 0.860761461786727,
      "learning_rate": 7.564738772466153e-07,
      "loss": 0.0241,
      "step": 9506
    },
    {
      "epoch": 2.163139931740614,
      "grad_norm": 1.0340484413917008,
      "learning_rate": 7.563865238679133e-07,
      "loss": 0.0153,
      "step": 9507
    },
    {
      "epoch": 2.163367463026166,
      "grad_norm": 1.1514085735061357,
      "learning_rate": 7.562991678039116e-07,
      "loss": 0.0129,
      "step": 9508
    },
    {
      "epoch": 2.163594994311718,
      "grad_norm": 0.9495706834279195,
      "learning_rate": 7.562118090563953e-07,
      "loss": 0.0123,
      "step": 9509
    },
    {
      "epoch": 2.1638225255972694,
      "grad_norm": 0.8809677222109957,
      "learning_rate": 7.5612444762715e-07,
      "loss": 0.0078,
      "step": 9510
    },
    {
      "epoch": 2.1640500568828216,
      "grad_norm": 1.538436546868351,
      "learning_rate": 7.560370835179611e-07,
      "loss": 0.0144,
      "step": 9511
    },
    {
      "epoch": 2.1642775881683733,
      "grad_norm": 0.9998239799209021,
      "learning_rate": 7.559497167306141e-07,
      "loss": 0.016,
      "step": 9512
    },
    {
      "epoch": 2.164505119453925,
      "grad_norm": 0.6613338635935855,
      "learning_rate": 7.558623472668948e-07,
      "loss": 0.0049,
      "step": 9513
    },
    {
      "epoch": 2.164732650739477,
      "grad_norm": 0.6066017285747133,
      "learning_rate": 7.557749751285887e-07,
      "loss": 0.0053,
      "step": 9514
    },
    {
      "epoch": 2.1649601820250286,
      "grad_norm": 1.151736591714483,
      "learning_rate": 7.556876003174816e-07,
      "loss": 0.027,
      "step": 9515
    },
    {
      "epoch": 2.1651877133105804,
      "grad_norm": 0.49550111870942964,
      "learning_rate": 7.556002228353595e-07,
      "loss": 0.0074,
      "step": 9516
    },
    {
      "epoch": 2.165415244596132,
      "grad_norm": 0.6363948637670657,
      "learning_rate": 7.555128426840078e-07,
      "loss": 0.0058,
      "step": 9517
    },
    {
      "epoch": 2.165642775881684,
      "grad_norm": 0.4948378226484785,
      "learning_rate": 7.554254598652127e-07,
      "loss": 0.0055,
      "step": 9518
    },
    {
      "epoch": 2.1658703071672356,
      "grad_norm": 1.452716090777511,
      "learning_rate": 7.553380743807602e-07,
      "loss": 0.0192,
      "step": 9519
    },
    {
      "epoch": 2.1660978384527874,
      "grad_norm": 0.8510160858709985,
      "learning_rate": 7.552506862324358e-07,
      "loss": 0.0184,
      "step": 9520
    },
    {
      "epoch": 2.166325369738339,
      "grad_norm": 0.489365539657168,
      "learning_rate": 7.551632954220263e-07,
      "loss": 0.0035,
      "step": 9521
    },
    {
      "epoch": 2.166552901023891,
      "grad_norm": 0.5232426370605646,
      "learning_rate": 7.550759019513173e-07,
      "loss": 0.01,
      "step": 9522
    },
    {
      "epoch": 2.1667804323094426,
      "grad_norm": 0.9718750730295778,
      "learning_rate": 7.54988505822095e-07,
      "loss": 0.0196,
      "step": 9523
    },
    {
      "epoch": 2.1670079635949944,
      "grad_norm": 1.2403730448401704,
      "learning_rate": 7.549011070361459e-07,
      "loss": 0.0251,
      "step": 9524
    },
    {
      "epoch": 2.167235494880546,
      "grad_norm": 0.6693444311484058,
      "learning_rate": 7.548137055952559e-07,
      "loss": 0.0066,
      "step": 9525
    },
    {
      "epoch": 2.167463026166098,
      "grad_norm": 0.22895710946983883,
      "learning_rate": 7.547263015012116e-07,
      "loss": 0.0017,
      "step": 9526
    },
    {
      "epoch": 2.1676905574516496,
      "grad_norm": 1.164643709332353,
      "learning_rate": 7.54638894755799e-07,
      "loss": 0.0193,
      "step": 9527
    },
    {
      "epoch": 2.1679180887372014,
      "grad_norm": 1.420130355190451,
      "learning_rate": 7.54551485360805e-07,
      "loss": 0.0135,
      "step": 9528
    },
    {
      "epoch": 2.168145620022753,
      "grad_norm": 0.7162802351482405,
      "learning_rate": 7.544640733180161e-07,
      "loss": 0.0068,
      "step": 9529
    },
    {
      "epoch": 2.168373151308305,
      "grad_norm": 0.28723169717464375,
      "learning_rate": 7.543766586292185e-07,
      "loss": 0.0037,
      "step": 9530
    },
    {
      "epoch": 2.1686006825938566,
      "grad_norm": 0.566716757776061,
      "learning_rate": 7.542892412961988e-07,
      "loss": 0.0073,
      "step": 9531
    },
    {
      "epoch": 2.1688282138794084,
      "grad_norm": 0.48710835462811136,
      "learning_rate": 7.54201821320744e-07,
      "loss": 0.0046,
      "step": 9532
    },
    {
      "epoch": 2.16905574516496,
      "grad_norm": 0.5591433451852652,
      "learning_rate": 7.541143987046406e-07,
      "loss": 0.0052,
      "step": 9533
    },
    {
      "epoch": 2.169283276450512,
      "grad_norm": 0.2464541059726593,
      "learning_rate": 7.540269734496751e-07,
      "loss": 0.002,
      "step": 9534
    },
    {
      "epoch": 2.1695108077360636,
      "grad_norm": 1.543625162402517,
      "learning_rate": 7.539395455576348e-07,
      "loss": 0.0269,
      "step": 9535
    },
    {
      "epoch": 2.1697383390216154,
      "grad_norm": 1.1466098187169464,
      "learning_rate": 7.538521150303063e-07,
      "loss": 0.0252,
      "step": 9536
    },
    {
      "epoch": 2.169965870307167,
      "grad_norm": 0.9776895216291299,
      "learning_rate": 7.537646818694764e-07,
      "loss": 0.016,
      "step": 9537
    },
    {
      "epoch": 2.170193401592719,
      "grad_norm": 0.8253760441414819,
      "learning_rate": 7.536772460769324e-07,
      "loss": 0.0146,
      "step": 9538
    },
    {
      "epoch": 2.1704209328782706,
      "grad_norm": 0.7606950356821949,
      "learning_rate": 7.535898076544611e-07,
      "loss": 0.0136,
      "step": 9539
    },
    {
      "epoch": 2.1706484641638224,
      "grad_norm": 0.38894234172403025,
      "learning_rate": 7.535023666038497e-07,
      "loss": 0.0035,
      "step": 9540
    },
    {
      "epoch": 2.170875995449374,
      "grad_norm": 2.2465203039467516,
      "learning_rate": 7.534149229268852e-07,
      "loss": 0.0208,
      "step": 9541
    },
    {
      "epoch": 2.171103526734926,
      "grad_norm": 0.6776393549469663,
      "learning_rate": 7.533274766253548e-07,
      "loss": 0.0087,
      "step": 9542
    },
    {
      "epoch": 2.1713310580204777,
      "grad_norm": 2.600125250146677,
      "learning_rate": 7.532400277010458e-07,
      "loss": 0.0262,
      "step": 9543
    },
    {
      "epoch": 2.1715585893060294,
      "grad_norm": 1.0368628318608313,
      "learning_rate": 7.531525761557454e-07,
      "loss": 0.0238,
      "step": 9544
    },
    {
      "epoch": 2.171786120591581,
      "grad_norm": 0.8869741215994206,
      "learning_rate": 7.530651219912413e-07,
      "loss": 0.0126,
      "step": 9545
    },
    {
      "epoch": 2.172013651877133,
      "grad_norm": 1.5295344443470758,
      "learning_rate": 7.529776652093204e-07,
      "loss": 0.0175,
      "step": 9546
    },
    {
      "epoch": 2.1722411831626847,
      "grad_norm": 0.5789423568425607,
      "learning_rate": 7.528902058117707e-07,
      "loss": 0.0107,
      "step": 9547
    },
    {
      "epoch": 2.172468714448237,
      "grad_norm": 3.4095737366223253,
      "learning_rate": 7.528027438003792e-07,
      "loss": 0.0236,
      "step": 9548
    },
    {
      "epoch": 2.172696245733788,
      "grad_norm": 0.4689578531811793,
      "learning_rate": 7.527152791769338e-07,
      "loss": 0.0133,
      "step": 9549
    },
    {
      "epoch": 2.1729237770193404,
      "grad_norm": 0.6528946872508574,
      "learning_rate": 7.526278119432219e-07,
      "loss": 0.0065,
      "step": 9550
    },
    {
      "epoch": 2.173151308304892,
      "grad_norm": 1.2003887222435774,
      "learning_rate": 7.52540342101031e-07,
      "loss": 0.0124,
      "step": 9551
    },
    {
      "epoch": 2.173378839590444,
      "grad_norm": 1.8213001244569549,
      "learning_rate": 7.524528696521495e-07,
      "loss": 0.0184,
      "step": 9552
    },
    {
      "epoch": 2.1736063708759956,
      "grad_norm": 0.8017969114131855,
      "learning_rate": 7.523653945983645e-07,
      "loss": 0.0044,
      "step": 9553
    },
    {
      "epoch": 2.1738339021615474,
      "grad_norm": 0.8195679399688243,
      "learning_rate": 7.522779169414643e-07,
      "loss": 0.0193,
      "step": 9554
    },
    {
      "epoch": 2.174061433447099,
      "grad_norm": 0.4250071114814642,
      "learning_rate": 7.521904366832365e-07,
      "loss": 0.0041,
      "step": 9555
    },
    {
      "epoch": 2.174288964732651,
      "grad_norm": 1.5407280672030532,
      "learning_rate": 7.521029538254692e-07,
      "loss": 0.0175,
      "step": 9556
    },
    {
      "epoch": 2.1745164960182026,
      "grad_norm": 0.5890809727558881,
      "learning_rate": 7.520154683699501e-07,
      "loss": 0.0055,
      "step": 9557
    },
    {
      "epoch": 2.1747440273037544,
      "grad_norm": 0.4782126783491611,
      "learning_rate": 7.519279803184676e-07,
      "loss": 0.0048,
      "step": 9558
    },
    {
      "epoch": 2.174971558589306,
      "grad_norm": 0.9263124914046664,
      "learning_rate": 7.518404896728096e-07,
      "loss": 0.0043,
      "step": 9559
    },
    {
      "epoch": 2.175199089874858,
      "grad_norm": 0.811010556777203,
      "learning_rate": 7.517529964347642e-07,
      "loss": 0.0111,
      "step": 9560
    },
    {
      "epoch": 2.1754266211604096,
      "grad_norm": 0.43401860315958946,
      "learning_rate": 7.516655006061198e-07,
      "loss": 0.0073,
      "step": 9561
    },
    {
      "epoch": 2.1756541524459614,
      "grad_norm": 0.8439321416931025,
      "learning_rate": 7.515780021886646e-07,
      "loss": 0.0104,
      "step": 9562
    },
    {
      "epoch": 2.175881683731513,
      "grad_norm": 0.7400634559362335,
      "learning_rate": 7.514905011841867e-07,
      "loss": 0.0027,
      "step": 9563
    },
    {
      "epoch": 2.176109215017065,
      "grad_norm": 0.2752754746882654,
      "learning_rate": 7.514029975944746e-07,
      "loss": 0.0036,
      "step": 9564
    },
    {
      "epoch": 2.1763367463026166,
      "grad_norm": 0.6620459171025077,
      "learning_rate": 7.513154914213168e-07,
      "loss": 0.0097,
      "step": 9565
    },
    {
      "epoch": 2.1765642775881684,
      "grad_norm": 0.44617721651979925,
      "learning_rate": 7.512279826665018e-07,
      "loss": 0.0049,
      "step": 9566
    },
    {
      "epoch": 2.17679180887372,
      "grad_norm": 1.2066031048324617,
      "learning_rate": 7.511404713318178e-07,
      "loss": 0.004,
      "step": 9567
    },
    {
      "epoch": 2.177019340159272,
      "grad_norm": 1.1304229490260347,
      "learning_rate": 7.510529574190537e-07,
      "loss": 0.0339,
      "step": 9568
    },
    {
      "epoch": 2.1772468714448237,
      "grad_norm": 0.4524987590201581,
      "learning_rate": 7.50965440929998e-07,
      "loss": 0.0029,
      "step": 9569
    },
    {
      "epoch": 2.1774744027303754,
      "grad_norm": 0.5350983992744186,
      "learning_rate": 7.508779218664395e-07,
      "loss": 0.0039,
      "step": 9570
    },
    {
      "epoch": 2.177701934015927,
      "grad_norm": 0.9750713017031654,
      "learning_rate": 7.507904002301665e-07,
      "loss": 0.0172,
      "step": 9571
    },
    {
      "epoch": 2.177929465301479,
      "grad_norm": 0.31495024248764425,
      "learning_rate": 7.507028760229683e-07,
      "loss": 0.002,
      "step": 9572
    },
    {
      "epoch": 2.1781569965870307,
      "grad_norm": 0.9604492360124036,
      "learning_rate": 7.506153492466337e-07,
      "loss": 0.0114,
      "step": 9573
    },
    {
      "epoch": 2.1783845278725824,
      "grad_norm": 0.7212474160748571,
      "learning_rate": 7.505278199029511e-07,
      "loss": 0.0068,
      "step": 9574
    },
    {
      "epoch": 2.178612059158134,
      "grad_norm": 0.8209618033231948,
      "learning_rate": 7.504402879937098e-07,
      "loss": 0.0052,
      "step": 9575
    },
    {
      "epoch": 2.178839590443686,
      "grad_norm": 0.6558409298307322,
      "learning_rate": 7.503527535206989e-07,
      "loss": 0.0065,
      "step": 9576
    },
    {
      "epoch": 2.1790671217292377,
      "grad_norm": 1.3196233115801024,
      "learning_rate": 7.502652164857072e-07,
      "loss": 0.0186,
      "step": 9577
    },
    {
      "epoch": 2.1792946530147894,
      "grad_norm": 0.48778107650934543,
      "learning_rate": 7.501776768905238e-07,
      "loss": 0.0039,
      "step": 9578
    },
    {
      "epoch": 2.179522184300341,
      "grad_norm": 0.6238318071136563,
      "learning_rate": 7.500901347369382e-07,
      "loss": 0.0102,
      "step": 9579
    },
    {
      "epoch": 2.179749715585893,
      "grad_norm": 0.6164717050241055,
      "learning_rate": 7.500025900267391e-07,
      "loss": 0.0048,
      "step": 9580
    },
    {
      "epoch": 2.1799772468714447,
      "grad_norm": 0.5548015201688632,
      "learning_rate": 7.49915042761716e-07,
      "loss": 0.0073,
      "step": 9581
    },
    {
      "epoch": 2.1802047781569964,
      "grad_norm": 1.247205981181567,
      "learning_rate": 7.498274929436583e-07,
      "loss": 0.0169,
      "step": 9582
    },
    {
      "epoch": 2.180432309442548,
      "grad_norm": 1.0737704832054575,
      "learning_rate": 7.497399405743552e-07,
      "loss": 0.0136,
      "step": 9583
    },
    {
      "epoch": 2.1806598407281,
      "grad_norm": 1.5342282126082314,
      "learning_rate": 7.49652385655596e-07,
      "loss": 0.0097,
      "step": 9584
    },
    {
      "epoch": 2.1808873720136517,
      "grad_norm": 1.0253123674011426,
      "learning_rate": 7.495648281891707e-07,
      "loss": 0.0074,
      "step": 9585
    },
    {
      "epoch": 2.1811149032992034,
      "grad_norm": 0.3825181504641467,
      "learning_rate": 7.494772681768683e-07,
      "loss": 0.0025,
      "step": 9586
    },
    {
      "epoch": 2.1813424345847556,
      "grad_norm": 0.4871828783572268,
      "learning_rate": 7.493897056204784e-07,
      "loss": 0.0079,
      "step": 9587
    },
    {
      "epoch": 2.181569965870307,
      "grad_norm": 0.651545542290053,
      "learning_rate": 7.493021405217907e-07,
      "loss": 0.0063,
      "step": 9588
    },
    {
      "epoch": 2.181797497155859,
      "grad_norm": 0.8304489261902582,
      "learning_rate": 7.49214572882595e-07,
      "loss": 0.0112,
      "step": 9589
    },
    {
      "epoch": 2.182025028441411,
      "grad_norm": 0.39866347770388444,
      "learning_rate": 7.49127002704681e-07,
      "loss": 0.0022,
      "step": 9590
    },
    {
      "epoch": 2.1822525597269626,
      "grad_norm": 0.7037762301674517,
      "learning_rate": 7.490394299898382e-07,
      "loss": 0.0031,
      "step": 9591
    },
    {
      "epoch": 2.1824800910125144,
      "grad_norm": 1.1093378758916597,
      "learning_rate": 7.489518547398568e-07,
      "loss": 0.0162,
      "step": 9592
    },
    {
      "epoch": 2.182707622298066,
      "grad_norm": 1.1760891436295915,
      "learning_rate": 7.488642769565264e-07,
      "loss": 0.007,
      "step": 9593
    },
    {
      "epoch": 2.182935153583618,
      "grad_norm": 0.5126123392999782,
      "learning_rate": 7.48776696641637e-07,
      "loss": 0.0037,
      "step": 9594
    },
    {
      "epoch": 2.1831626848691696,
      "grad_norm": 1.2561187293283704,
      "learning_rate": 7.486891137969786e-07,
      "loss": 0.0192,
      "step": 9595
    },
    {
      "epoch": 2.1833902161547214,
      "grad_norm": 0.4428997280141245,
      "learning_rate": 7.486015284243413e-07,
      "loss": 0.0043,
      "step": 9596
    },
    {
      "epoch": 2.183617747440273,
      "grad_norm": 1.2136340385801088,
      "learning_rate": 7.485139405255151e-07,
      "loss": 0.0182,
      "step": 9597
    },
    {
      "epoch": 2.183845278725825,
      "grad_norm": 0.7423522442041067,
      "learning_rate": 7.484263501022902e-07,
      "loss": 0.0101,
      "step": 9598
    },
    {
      "epoch": 2.1840728100113767,
      "grad_norm": 1.647640497082901,
      "learning_rate": 7.483387571564567e-07,
      "loss": 0.0212,
      "step": 9599
    },
    {
      "epoch": 2.1843003412969284,
      "grad_norm": 0.8536106447079324,
      "learning_rate": 7.48251161689805e-07,
      "loss": 0.0023,
      "step": 9600
    },
    {
      "epoch": 2.18452787258248,
      "grad_norm": 0.4079155494938698,
      "learning_rate": 7.48163563704125e-07,
      "loss": 0.0045,
      "step": 9601
    },
    {
      "epoch": 2.184755403868032,
      "grad_norm": 1.1308651387228579,
      "learning_rate": 7.480759632012074e-07,
      "loss": 0.0122,
      "step": 9602
    },
    {
      "epoch": 2.1849829351535837,
      "grad_norm": 0.6120958606576665,
      "learning_rate": 7.479883601828428e-07,
      "loss": 0.0026,
      "step": 9603
    },
    {
      "epoch": 2.1852104664391354,
      "grad_norm": 1.9195083990763984,
      "learning_rate": 7.479007546508211e-07,
      "loss": 0.0925,
      "step": 9604
    },
    {
      "epoch": 2.185437997724687,
      "grad_norm": 0.9168734226255825,
      "learning_rate": 7.478131466069331e-07,
      "loss": 0.0079,
      "step": 9605
    },
    {
      "epoch": 2.185665529010239,
      "grad_norm": 1.7360235232611294,
      "learning_rate": 7.477255360529695e-07,
      "loss": 0.0516,
      "step": 9606
    },
    {
      "epoch": 2.1858930602957907,
      "grad_norm": 0.7931409946825114,
      "learning_rate": 7.476379229907205e-07,
      "loss": 0.0179,
      "step": 9607
    },
    {
      "epoch": 2.1861205915813424,
      "grad_norm": 0.5513116615728468,
      "learning_rate": 7.475503074219769e-07,
      "loss": 0.0084,
      "step": 9608
    },
    {
      "epoch": 2.186348122866894,
      "grad_norm": 1.5527290776544047,
      "learning_rate": 7.474626893485295e-07,
      "loss": 0.0244,
      "step": 9609
    },
    {
      "epoch": 2.186575654152446,
      "grad_norm": 0.9065148771736532,
      "learning_rate": 7.473750687721692e-07,
      "loss": 0.0034,
      "step": 9610
    },
    {
      "epoch": 2.1868031854379977,
      "grad_norm": 1.1387030868166388,
      "learning_rate": 7.472874456946865e-07,
      "loss": 0.018,
      "step": 9611
    },
    {
      "epoch": 2.1870307167235494,
      "grad_norm": 0.9547637556667848,
      "learning_rate": 7.471998201178724e-07,
      "loss": 0.0125,
      "step": 9612
    },
    {
      "epoch": 2.187258248009101,
      "grad_norm": 0.8066593350658329,
      "learning_rate": 7.471121920435176e-07,
      "loss": 0.0095,
      "step": 9613
    },
    {
      "epoch": 2.187485779294653,
      "grad_norm": 0.6157773503488007,
      "learning_rate": 7.470245614734132e-07,
      "loss": 0.0068,
      "step": 9614
    },
    {
      "epoch": 2.1877133105802047,
      "grad_norm": 0.5660108008538516,
      "learning_rate": 7.469369284093504e-07,
      "loss": 0.0064,
      "step": 9615
    },
    {
      "epoch": 2.1879408418657564,
      "grad_norm": 0.45605242358242326,
      "learning_rate": 7.468492928531201e-07,
      "loss": 0.0041,
      "step": 9616
    },
    {
      "epoch": 2.188168373151308,
      "grad_norm": 1.0144865167191068,
      "learning_rate": 7.467616548065134e-07,
      "loss": 0.0103,
      "step": 9617
    },
    {
      "epoch": 2.18839590443686,
      "grad_norm": 1.3562826992877761,
      "learning_rate": 7.466740142713217e-07,
      "loss": 0.0272,
      "step": 9618
    },
    {
      "epoch": 2.1886234357224117,
      "grad_norm": 0.9701025438819655,
      "learning_rate": 7.465863712493357e-07,
      "loss": 0.0135,
      "step": 9619
    },
    {
      "epoch": 2.1888509670079634,
      "grad_norm": 1.306657040530877,
      "learning_rate": 7.46498725742347e-07,
      "loss": 0.0214,
      "step": 9620
    },
    {
      "epoch": 2.189078498293515,
      "grad_norm": 0.42960866613517307,
      "learning_rate": 7.464110777521467e-07,
      "loss": 0.003,
      "step": 9621
    },
    {
      "epoch": 2.189306029579067,
      "grad_norm": 0.665244392318043,
      "learning_rate": 7.463234272805265e-07,
      "loss": 0.0114,
      "step": 9622
    },
    {
      "epoch": 2.1895335608646187,
      "grad_norm": 0.6265393107396235,
      "learning_rate": 7.462357743292778e-07,
      "loss": 0.0103,
      "step": 9623
    },
    {
      "epoch": 2.1897610921501705,
      "grad_norm": 0.7142319456732945,
      "learning_rate": 7.461481189001917e-07,
      "loss": 0.0071,
      "step": 9624
    },
    {
      "epoch": 2.189988623435722,
      "grad_norm": 1.432775949720797,
      "learning_rate": 7.460604609950599e-07,
      "loss": 0.0115,
      "step": 9625
    },
    {
      "epoch": 2.1902161547212744,
      "grad_norm": 0.2710535011303526,
      "learning_rate": 7.459728006156741e-07,
      "loss": 0.0017,
      "step": 9626
    },
    {
      "epoch": 2.1904436860068257,
      "grad_norm": 0.5450555510465152,
      "learning_rate": 7.458851377638257e-07,
      "loss": 0.0071,
      "step": 9627
    },
    {
      "epoch": 2.190671217292378,
      "grad_norm": 0.7651487622447738,
      "learning_rate": 7.457974724413065e-07,
      "loss": 0.0146,
      "step": 9628
    },
    {
      "epoch": 2.1908987485779297,
      "grad_norm": 0.40725306256207244,
      "learning_rate": 7.457098046499084e-07,
      "loss": 0.0039,
      "step": 9629
    },
    {
      "epoch": 2.1911262798634814,
      "grad_norm": 0.8897519703656316,
      "learning_rate": 7.456221343914228e-07,
      "loss": 0.0119,
      "step": 9630
    },
    {
      "epoch": 2.191353811149033,
      "grad_norm": 0.7763470111164008,
      "learning_rate": 7.455344616676415e-07,
      "loss": 0.0081,
      "step": 9631
    },
    {
      "epoch": 2.191581342434585,
      "grad_norm": 0.9182052754974886,
      "learning_rate": 7.454467864803567e-07,
      "loss": 0.021,
      "step": 9632
    },
    {
      "epoch": 2.1918088737201367,
      "grad_norm": 0.5725793457523388,
      "learning_rate": 7.453591088313603e-07,
      "loss": 0.0042,
      "step": 9633
    },
    {
      "epoch": 2.1920364050056884,
      "grad_norm": 1.0484063564453674,
      "learning_rate": 7.45271428722444e-07,
      "loss": 0.0064,
      "step": 9634
    },
    {
      "epoch": 2.19226393629124,
      "grad_norm": 1.064052456490962,
      "learning_rate": 7.451837461553998e-07,
      "loss": 0.0082,
      "step": 9635
    },
    {
      "epoch": 2.192491467576792,
      "grad_norm": 0.5759445364609324,
      "learning_rate": 7.450960611320204e-07,
      "loss": 0.0049,
      "step": 9636
    },
    {
      "epoch": 2.1927189988623437,
      "grad_norm": 1.3167249572075435,
      "learning_rate": 7.450083736540972e-07,
      "loss": 0.0097,
      "step": 9637
    },
    {
      "epoch": 2.1929465301478954,
      "grad_norm": 0.42812996497137096,
      "learning_rate": 7.449206837234224e-07,
      "loss": 0.0043,
      "step": 9638
    },
    {
      "epoch": 2.193174061433447,
      "grad_norm": 0.47389532187756966,
      "learning_rate": 7.448329913417887e-07,
      "loss": 0.0021,
      "step": 9639
    },
    {
      "epoch": 2.193401592718999,
      "grad_norm": 0.5674565541782638,
      "learning_rate": 7.447452965109882e-07,
      "loss": 0.0106,
      "step": 9640
    },
    {
      "epoch": 2.1936291240045507,
      "grad_norm": 0.6660424990854488,
      "learning_rate": 7.446575992328128e-07,
      "loss": 0.0086,
      "step": 9641
    },
    {
      "epoch": 2.1938566552901024,
      "grad_norm": 0.20386544039544735,
      "learning_rate": 7.445698995090557e-07,
      "loss": 0.0013,
      "step": 9642
    },
    {
      "epoch": 2.194084186575654,
      "grad_norm": 0.2099220833545005,
      "learning_rate": 7.444821973415086e-07,
      "loss": 0.0017,
      "step": 9643
    },
    {
      "epoch": 2.194311717861206,
      "grad_norm": 0.3596174077194479,
      "learning_rate": 7.443944927319641e-07,
      "loss": 0.0023,
      "step": 9644
    },
    {
      "epoch": 2.1945392491467577,
      "grad_norm": 0.9224216606352855,
      "learning_rate": 7.443067856822149e-07,
      "loss": 0.0129,
      "step": 9645
    },
    {
      "epoch": 2.1947667804323094,
      "grad_norm": 1.1469534965029151,
      "learning_rate": 7.442190761940535e-07,
      "loss": 0.014,
      "step": 9646
    },
    {
      "epoch": 2.194994311717861,
      "grad_norm": 0.5576731804567137,
      "learning_rate": 7.441313642692726e-07,
      "loss": 0.0046,
      "step": 9647
    },
    {
      "epoch": 2.195221843003413,
      "grad_norm": 1.0981045949315773,
      "learning_rate": 7.440436499096647e-07,
      "loss": 0.0199,
      "step": 9648
    },
    {
      "epoch": 2.1954493742889647,
      "grad_norm": 1.0420832800323911,
      "learning_rate": 7.439559331170226e-07,
      "loss": 0.0128,
      "step": 9649
    },
    {
      "epoch": 2.1956769055745164,
      "grad_norm": 0.626981774480868,
      "learning_rate": 7.438682138931393e-07,
      "loss": 0.0099,
      "step": 9650
    },
    {
      "epoch": 2.195904436860068,
      "grad_norm": 1.5521547661201331,
      "learning_rate": 7.43780492239807e-07,
      "loss": 0.0145,
      "step": 9651
    },
    {
      "epoch": 2.19613196814562,
      "grad_norm": 1.6288646101627964,
      "learning_rate": 7.436927681588192e-07,
      "loss": 0.0144,
      "step": 9652
    },
    {
      "epoch": 2.1963594994311717,
      "grad_norm": 0.8653215956544341,
      "learning_rate": 7.436050416519687e-07,
      "loss": 0.0135,
      "step": 9653
    },
    {
      "epoch": 2.1965870307167235,
      "grad_norm": 1.0809108035517665,
      "learning_rate": 7.435173127210482e-07,
      "loss": 0.013,
      "step": 9654
    },
    {
      "epoch": 2.196814562002275,
      "grad_norm": 2.1424690805717157,
      "learning_rate": 7.43429581367851e-07,
      "loss": 0.0073,
      "step": 9655
    },
    {
      "epoch": 2.197042093287827,
      "grad_norm": 0.9652735610174711,
      "learning_rate": 7.4334184759417e-07,
      "loss": 0.0211,
      "step": 9656
    },
    {
      "epoch": 2.1972696245733787,
      "grad_norm": 0.5538362197707377,
      "learning_rate": 7.432541114017984e-07,
      "loss": 0.004,
      "step": 9657
    },
    {
      "epoch": 2.1974971558589305,
      "grad_norm": 0.8759362034094246,
      "learning_rate": 7.431663727925293e-07,
      "loss": 0.0132,
      "step": 9658
    },
    {
      "epoch": 2.197724687144482,
      "grad_norm": 0.5670103598613527,
      "learning_rate": 7.430786317681559e-07,
      "loss": 0.0078,
      "step": 9659
    },
    {
      "epoch": 2.197952218430034,
      "grad_norm": 0.9198394347608914,
      "learning_rate": 7.429908883304716e-07,
      "loss": 0.0124,
      "step": 9660
    },
    {
      "epoch": 2.1981797497155857,
      "grad_norm": 0.6520971346068454,
      "learning_rate": 7.429031424812697e-07,
      "loss": 0.009,
      "step": 9661
    },
    {
      "epoch": 2.1984072810011375,
      "grad_norm": 1.0629564601389152,
      "learning_rate": 7.428153942223433e-07,
      "loss": 0.0188,
      "step": 9662
    },
    {
      "epoch": 2.198634812286689,
      "grad_norm": 1.1199232548698799,
      "learning_rate": 7.427276435554861e-07,
      "loss": 0.0206,
      "step": 9663
    },
    {
      "epoch": 2.198862343572241,
      "grad_norm": 1.0829610534274123,
      "learning_rate": 7.426398904824916e-07,
      "loss": 0.0287,
      "step": 9664
    },
    {
      "epoch": 2.199089874857793,
      "grad_norm": 0.8314551580354805,
      "learning_rate": 7.425521350051529e-07,
      "loss": 0.012,
      "step": 9665
    },
    {
      "epoch": 2.1993174061433445,
      "grad_norm": 0.8582521213620955,
      "learning_rate": 7.42464377125264e-07,
      "loss": 0.0089,
      "step": 9666
    },
    {
      "epoch": 2.1995449374288967,
      "grad_norm": 1.5212309016840493,
      "learning_rate": 7.423766168446187e-07,
      "loss": 0.0084,
      "step": 9667
    },
    {
      "epoch": 2.1997724687144484,
      "grad_norm": 0.5467873056219017,
      "learning_rate": 7.422888541650097e-07,
      "loss": 0.0082,
      "step": 9668
    },
    {
      "epoch": 2.2,
      "grad_norm": 1.1352258144425396,
      "learning_rate": 7.422010890882317e-07,
      "loss": 0.0059,
      "step": 9669
    },
    {
      "epoch": 2.200227531285552,
      "grad_norm": 0.36437846592903533,
      "learning_rate": 7.421133216160781e-07,
      "loss": 0.0022,
      "step": 9670
    },
    {
      "epoch": 2.2004550625711037,
      "grad_norm": 0.5866144111825328,
      "learning_rate": 7.420255517503424e-07,
      "loss": 0.0062,
      "step": 9671
    },
    {
      "epoch": 2.2006825938566554,
      "grad_norm": 0.8754788877292402,
      "learning_rate": 7.41937779492819e-07,
      "loss": 0.0146,
      "step": 9672
    },
    {
      "epoch": 2.200910125142207,
      "grad_norm": 0.4171831071461107,
      "learning_rate": 7.418500048453016e-07,
      "loss": 0.005,
      "step": 9673
    },
    {
      "epoch": 2.201137656427759,
      "grad_norm": 0.549269081954464,
      "learning_rate": 7.417622278095838e-07,
      "loss": 0.0139,
      "step": 9674
    },
    {
      "epoch": 2.2013651877133107,
      "grad_norm": 1.2626057795914143,
      "learning_rate": 7.416744483874602e-07,
      "loss": 0.01,
      "step": 9675
    },
    {
      "epoch": 2.2015927189988624,
      "grad_norm": 0.7629721611373987,
      "learning_rate": 7.415866665807245e-07,
      "loss": 0.0164,
      "step": 9676
    },
    {
      "epoch": 2.201820250284414,
      "grad_norm": 0.7138412826063927,
      "learning_rate": 7.414988823911708e-07,
      "loss": 0.0121,
      "step": 9677
    },
    {
      "epoch": 2.202047781569966,
      "grad_norm": 0.7896205889847766,
      "learning_rate": 7.41411095820593e-07,
      "loss": 0.0088,
      "step": 9678
    },
    {
      "epoch": 2.2022753128555177,
      "grad_norm": 0.5642157705673803,
      "learning_rate": 7.41323306870786e-07,
      "loss": 0.0063,
      "step": 9679
    },
    {
      "epoch": 2.2025028441410694,
      "grad_norm": 0.9039809924753222,
      "learning_rate": 7.412355155435437e-07,
      "loss": 0.005,
      "step": 9680
    },
    {
      "epoch": 2.202730375426621,
      "grad_norm": 1.0135787829433691,
      "learning_rate": 7.411477218406602e-07,
      "loss": 0.0086,
      "step": 9681
    },
    {
      "epoch": 2.202957906712173,
      "grad_norm": 0.744895167811461,
      "learning_rate": 7.410599257639299e-07,
      "loss": 0.0053,
      "step": 9682
    },
    {
      "epoch": 2.2031854379977247,
      "grad_norm": 1.9278466629666653,
      "learning_rate": 7.409721273151473e-07,
      "loss": 0.0215,
      "step": 9683
    },
    {
      "epoch": 2.2034129692832765,
      "grad_norm": 0.7608888631413969,
      "learning_rate": 7.408843264961068e-07,
      "loss": 0.0112,
      "step": 9684
    },
    {
      "epoch": 2.203640500568828,
      "grad_norm": 0.24319250994984543,
      "learning_rate": 7.407965233086029e-07,
      "loss": 0.0019,
      "step": 9685
    },
    {
      "epoch": 2.20386803185438,
      "grad_norm": 0.10591754608419379,
      "learning_rate": 7.407087177544304e-07,
      "loss": 0.0007,
      "step": 9686
    },
    {
      "epoch": 2.2040955631399317,
      "grad_norm": 0.49157075978222076,
      "learning_rate": 7.406209098353834e-07,
      "loss": 0.0035,
      "step": 9687
    },
    {
      "epoch": 2.2043230944254835,
      "grad_norm": 1.062474610827862,
      "learning_rate": 7.405330995532566e-07,
      "loss": 0.0206,
      "step": 9688
    },
    {
      "epoch": 2.204550625711035,
      "grad_norm": 1.5601721722500297,
      "learning_rate": 7.40445286909845e-07,
      "loss": 0.035,
      "step": 9689
    },
    {
      "epoch": 2.204778156996587,
      "grad_norm": 0.8126677089149624,
      "learning_rate": 7.403574719069431e-07,
      "loss": 0.0063,
      "step": 9690
    },
    {
      "epoch": 2.2050056882821387,
      "grad_norm": 0.31967234686748364,
      "learning_rate": 7.402696545463458e-07,
      "loss": 0.0024,
      "step": 9691
    },
    {
      "epoch": 2.2052332195676905,
      "grad_norm": 0.6296830929918673,
      "learning_rate": 7.401818348298478e-07,
      "loss": 0.0053,
      "step": 9692
    },
    {
      "epoch": 2.2054607508532422,
      "grad_norm": 2.4668394767267454,
      "learning_rate": 7.400940127592441e-07,
      "loss": 0.0176,
      "step": 9693
    },
    {
      "epoch": 2.205688282138794,
      "grad_norm": 1.1958832285426688,
      "learning_rate": 7.400061883363297e-07,
      "loss": 0.0074,
      "step": 9694
    },
    {
      "epoch": 2.2059158134243457,
      "grad_norm": 0.4318878369099174,
      "learning_rate": 7.399183615628991e-07,
      "loss": 0.0021,
      "step": 9695
    },
    {
      "epoch": 2.2061433447098975,
      "grad_norm": 1.2109798418826194,
      "learning_rate": 7.398305324407479e-07,
      "loss": 0.009,
      "step": 9696
    },
    {
      "epoch": 2.2063708759954492,
      "grad_norm": 0.8625508430982195,
      "learning_rate": 7.397427009716709e-07,
      "loss": 0.0042,
      "step": 9697
    },
    {
      "epoch": 2.206598407281001,
      "grad_norm": 0.9448879693822796,
      "learning_rate": 7.396548671574632e-07,
      "loss": 0.0156,
      "step": 9698
    },
    {
      "epoch": 2.2068259385665527,
      "grad_norm": 0.6357788236890357,
      "learning_rate": 7.395670309999201e-07,
      "loss": 0.0075,
      "step": 9699
    },
    {
      "epoch": 2.2070534698521045,
      "grad_norm": 0.4122020439740377,
      "learning_rate": 7.394791925008366e-07,
      "loss": 0.0041,
      "step": 9700
    },
    {
      "epoch": 2.2072810011376562,
      "grad_norm": 0.5639709716365757,
      "learning_rate": 7.39391351662008e-07,
      "loss": 0.0075,
      "step": 9701
    },
    {
      "epoch": 2.207508532423208,
      "grad_norm": 0.9578828855651509,
      "learning_rate": 7.393035084852296e-07,
      "loss": 0.0087,
      "step": 9702
    },
    {
      "epoch": 2.2077360637087597,
      "grad_norm": 0.8353081392030337,
      "learning_rate": 7.39215662972297e-07,
      "loss": 0.0079,
      "step": 9703
    },
    {
      "epoch": 2.207963594994312,
      "grad_norm": 0.6967088628881625,
      "learning_rate": 7.391278151250053e-07,
      "loss": 0.0071,
      "step": 9704
    },
    {
      "epoch": 2.2081911262798632,
      "grad_norm": 0.43785772640606097,
      "learning_rate": 7.390399649451501e-07,
      "loss": 0.0047,
      "step": 9705
    },
    {
      "epoch": 2.2084186575654154,
      "grad_norm": 1.2611023380849133,
      "learning_rate": 7.389521124345271e-07,
      "loss": 0.0144,
      "step": 9706
    },
    {
      "epoch": 2.208646188850967,
      "grad_norm": 0.9869060087712334,
      "learning_rate": 7.388642575949315e-07,
      "loss": 0.0128,
      "step": 9707
    },
    {
      "epoch": 2.208873720136519,
      "grad_norm": 1.1398066210971305,
      "learning_rate": 7.387764004281588e-07,
      "loss": 0.0125,
      "step": 9708
    },
    {
      "epoch": 2.2091012514220707,
      "grad_norm": 0.5568227219345118,
      "learning_rate": 7.38688540936005e-07,
      "loss": 0.0073,
      "step": 9709
    },
    {
      "epoch": 2.2093287827076225,
      "grad_norm": 0.5467974468137079,
      "learning_rate": 7.386006791202656e-07,
      "loss": 0.0048,
      "step": 9710
    },
    {
      "epoch": 2.209556313993174,
      "grad_norm": 0.9673966550728507,
      "learning_rate": 7.385128149827364e-07,
      "loss": 0.0111,
      "step": 9711
    },
    {
      "epoch": 2.209783845278726,
      "grad_norm": 2.183643309598677,
      "learning_rate": 7.384249485252132e-07,
      "loss": 0.0245,
      "step": 9712
    },
    {
      "epoch": 2.2100113765642777,
      "grad_norm": 0.36609886258629293,
      "learning_rate": 7.383370797494918e-07,
      "loss": 0.0057,
      "step": 9713
    },
    {
      "epoch": 2.2102389078498295,
      "grad_norm": 0.55612498845999,
      "learning_rate": 7.382492086573679e-07,
      "loss": 0.0044,
      "step": 9714
    },
    {
      "epoch": 2.210466439135381,
      "grad_norm": 0.8549709760846276,
      "learning_rate": 7.381613352506376e-07,
      "loss": 0.0166,
      "step": 9715
    },
    {
      "epoch": 2.210693970420933,
      "grad_norm": 1.600095099597409,
      "learning_rate": 7.380734595310969e-07,
      "loss": 0.0145,
      "step": 9716
    },
    {
      "epoch": 2.2109215017064847,
      "grad_norm": 1.6694806248049556,
      "learning_rate": 7.37985581500542e-07,
      "loss": 0.0295,
      "step": 9717
    },
    {
      "epoch": 2.2111490329920365,
      "grad_norm": 0.9519877093467191,
      "learning_rate": 7.378977011607684e-07,
      "loss": 0.0212,
      "step": 9718
    },
    {
      "epoch": 2.211376564277588,
      "grad_norm": 0.2321442487803093,
      "learning_rate": 7.378098185135727e-07,
      "loss": 0.0011,
      "step": 9719
    },
    {
      "epoch": 2.21160409556314,
      "grad_norm": 0.9669976328820756,
      "learning_rate": 7.377219335607509e-07,
      "loss": 0.0113,
      "step": 9720
    },
    {
      "epoch": 2.2118316268486917,
      "grad_norm": 0.35874159162276587,
      "learning_rate": 7.376340463040993e-07,
      "loss": 0.0035,
      "step": 9721
    },
    {
      "epoch": 2.2120591581342435,
      "grad_norm": 0.4911321717522839,
      "learning_rate": 7.375461567454138e-07,
      "loss": 0.0055,
      "step": 9722
    },
    {
      "epoch": 2.2122866894197952,
      "grad_norm": 1.6567432890342024,
      "learning_rate": 7.374582648864912e-07,
      "loss": 0.0173,
      "step": 9723
    },
    {
      "epoch": 2.212514220705347,
      "grad_norm": 0.7746999797599043,
      "learning_rate": 7.373703707291277e-07,
      "loss": 0.0141,
      "step": 9724
    },
    {
      "epoch": 2.2127417519908987,
      "grad_norm": 1.4899640213951841,
      "learning_rate": 7.372824742751194e-07,
      "loss": 0.0118,
      "step": 9725
    },
    {
      "epoch": 2.2129692832764505,
      "grad_norm": 1.1306059510750244,
      "learning_rate": 7.37194575526263e-07,
      "loss": 0.0192,
      "step": 9726
    },
    {
      "epoch": 2.2131968145620022,
      "grad_norm": 0.6094722469976073,
      "learning_rate": 7.371066744843551e-07,
      "loss": 0.0086,
      "step": 9727
    },
    {
      "epoch": 2.213424345847554,
      "grad_norm": 0.4183731135562627,
      "learning_rate": 7.37018771151192e-07,
      "loss": 0.0023,
      "step": 9728
    },
    {
      "epoch": 2.2136518771331057,
      "grad_norm": 0.775796211052962,
      "learning_rate": 7.369308655285702e-07,
      "loss": 0.007,
      "step": 9729
    },
    {
      "epoch": 2.2138794084186575,
      "grad_norm": 0.9083986254705477,
      "learning_rate": 7.368429576182869e-07,
      "loss": 0.0107,
      "step": 9730
    },
    {
      "epoch": 2.2141069397042092,
      "grad_norm": 0.5339774729572225,
      "learning_rate": 7.367550474221381e-07,
      "loss": 0.006,
      "step": 9731
    },
    {
      "epoch": 2.214334470989761,
      "grad_norm": 0.6757592444800821,
      "learning_rate": 7.366671349419207e-07,
      "loss": 0.0091,
      "step": 9732
    },
    {
      "epoch": 2.2145620022753127,
      "grad_norm": 0.7867220166681209,
      "learning_rate": 7.365792201794318e-07,
      "loss": 0.0128,
      "step": 9733
    },
    {
      "epoch": 2.2147895335608645,
      "grad_norm": 0.8067592322250232,
      "learning_rate": 7.364913031364679e-07,
      "loss": 0.0125,
      "step": 9734
    },
    {
      "epoch": 2.2150170648464163,
      "grad_norm": 0.8112378359985258,
      "learning_rate": 7.364033838148258e-07,
      "loss": 0.0166,
      "step": 9735
    },
    {
      "epoch": 2.215244596131968,
      "grad_norm": 0.9469113449795432,
      "learning_rate": 7.363154622163029e-07,
      "loss": 0.0054,
      "step": 9736
    },
    {
      "epoch": 2.2154721274175198,
      "grad_norm": 1.00490074369384,
      "learning_rate": 7.362275383426956e-07,
      "loss": 0.0174,
      "step": 9737
    },
    {
      "epoch": 2.2156996587030715,
      "grad_norm": 1.0761337878453314,
      "learning_rate": 7.361396121958011e-07,
      "loss": 0.0117,
      "step": 9738
    },
    {
      "epoch": 2.2159271899886233,
      "grad_norm": 0.5956420441176925,
      "learning_rate": 7.360516837774165e-07,
      "loss": 0.0086,
      "step": 9739
    },
    {
      "epoch": 2.216154721274175,
      "grad_norm": 0.3459081577385382,
      "learning_rate": 7.359637530893389e-07,
      "loss": 0.0025,
      "step": 9740
    },
    {
      "epoch": 2.2163822525597268,
      "grad_norm": 0.6341278133011897,
      "learning_rate": 7.358758201333654e-07,
      "loss": 0.0057,
      "step": 9741
    },
    {
      "epoch": 2.2166097838452785,
      "grad_norm": 1.338725636079328,
      "learning_rate": 7.357878849112932e-07,
      "loss": 0.019,
      "step": 9742
    },
    {
      "epoch": 2.2168373151308307,
      "grad_norm": 0.7306342018789439,
      "learning_rate": 7.356999474249196e-07,
      "loss": 0.0044,
      "step": 9743
    },
    {
      "epoch": 2.217064846416382,
      "grad_norm": 0.5703587106869538,
      "learning_rate": 7.356120076760417e-07,
      "loss": 0.0048,
      "step": 9744
    },
    {
      "epoch": 2.217292377701934,
      "grad_norm": 0.6262614362856112,
      "learning_rate": 7.35524065666457e-07,
      "loss": 0.0053,
      "step": 9745
    },
    {
      "epoch": 2.217519908987486,
      "grad_norm": 0.5066663290483394,
      "learning_rate": 7.354361213979627e-07,
      "loss": 0.0071,
      "step": 9746
    },
    {
      "epoch": 2.2177474402730377,
      "grad_norm": 0.9215848985052789,
      "learning_rate": 7.353481748723565e-07,
      "loss": 0.0221,
      "step": 9747
    },
    {
      "epoch": 2.2179749715585895,
      "grad_norm": 0.33323939733761665,
      "learning_rate": 7.352602260914356e-07,
      "loss": 0.0026,
      "step": 9748
    },
    {
      "epoch": 2.218202502844141,
      "grad_norm": 0.6033511338392381,
      "learning_rate": 7.351722750569976e-07,
      "loss": 0.0054,
      "step": 9749
    },
    {
      "epoch": 2.218430034129693,
      "grad_norm": 0.7854914660549903,
      "learning_rate": 7.350843217708402e-07,
      "loss": 0.0059,
      "step": 9750
    },
    {
      "epoch": 2.2186575654152447,
      "grad_norm": 0.9021074580889269,
      "learning_rate": 7.349963662347608e-07,
      "loss": 0.0164,
      "step": 9751
    },
    {
      "epoch": 2.2188850967007965,
      "grad_norm": 0.8693317117604361,
      "learning_rate": 7.349084084505569e-07,
      "loss": 0.0146,
      "step": 9752
    },
    {
      "epoch": 2.2191126279863482,
      "grad_norm": 0.7937928809056105,
      "learning_rate": 7.348204484200267e-07,
      "loss": 0.0129,
      "step": 9753
    },
    {
      "epoch": 2.2193401592719,
      "grad_norm": 0.30519990948993614,
      "learning_rate": 7.347324861449676e-07,
      "loss": 0.0034,
      "step": 9754
    },
    {
      "epoch": 2.2195676905574517,
      "grad_norm": 0.4962820337660195,
      "learning_rate": 7.346445216271773e-07,
      "loss": 0.0094,
      "step": 9755
    },
    {
      "epoch": 2.2197952218430035,
      "grad_norm": 0.6728902306719448,
      "learning_rate": 7.34556554868454e-07,
      "loss": 0.0063,
      "step": 9756
    },
    {
      "epoch": 2.2200227531285552,
      "grad_norm": 0.5476631883349763,
      "learning_rate": 7.344685858705952e-07,
      "loss": 0.0089,
      "step": 9757
    },
    {
      "epoch": 2.220250284414107,
      "grad_norm": 0.6789322134548239,
      "learning_rate": 7.343806146353991e-07,
      "loss": 0.0135,
      "step": 9758
    },
    {
      "epoch": 2.2204778156996587,
      "grad_norm": 1.327768648701586,
      "learning_rate": 7.342926411646634e-07,
      "loss": 0.0146,
      "step": 9759
    },
    {
      "epoch": 2.2207053469852105,
      "grad_norm": 0.7078627325170037,
      "learning_rate": 7.342046654601864e-07,
      "loss": 0.0122,
      "step": 9760
    },
    {
      "epoch": 2.2209328782707622,
      "grad_norm": 0.41115602276630314,
      "learning_rate": 7.341166875237661e-07,
      "loss": 0.0025,
      "step": 9761
    },
    {
      "epoch": 2.221160409556314,
      "grad_norm": 0.7821378111405003,
      "learning_rate": 7.340287073572004e-07,
      "loss": 0.0051,
      "step": 9762
    },
    {
      "epoch": 2.2213879408418657,
      "grad_norm": 0.7857378323477028,
      "learning_rate": 7.339407249622878e-07,
      "loss": 0.0107,
      "step": 9763
    },
    {
      "epoch": 2.2216154721274175,
      "grad_norm": 0.855086007831799,
      "learning_rate": 7.33852740340826e-07,
      "loss": 0.0071,
      "step": 9764
    },
    {
      "epoch": 2.2218430034129693,
      "grad_norm": 0.6590632272539326,
      "learning_rate": 7.337647534946137e-07,
      "loss": 0.0203,
      "step": 9765
    },
    {
      "epoch": 2.222070534698521,
      "grad_norm": 1.1095541097579438,
      "learning_rate": 7.33676764425449e-07,
      "loss": 0.0103,
      "step": 9766
    },
    {
      "epoch": 2.2222980659840728,
      "grad_norm": 0.3227748470963595,
      "learning_rate": 7.335887731351303e-07,
      "loss": 0.0016,
      "step": 9767
    },
    {
      "epoch": 2.2225255972696245,
      "grad_norm": 0.6449957560967259,
      "learning_rate": 7.33500779625456e-07,
      "loss": 0.0125,
      "step": 9768
    },
    {
      "epoch": 2.2227531285551763,
      "grad_norm": 0.5534057876566227,
      "learning_rate": 7.334127838982244e-07,
      "loss": 0.0057,
      "step": 9769
    },
    {
      "epoch": 2.222980659840728,
      "grad_norm": 1.1213144565575701,
      "learning_rate": 7.333247859552343e-07,
      "loss": 0.0237,
      "step": 9770
    },
    {
      "epoch": 2.2232081911262798,
      "grad_norm": 0.8275619816698846,
      "learning_rate": 7.332367857982836e-07,
      "loss": 0.0067,
      "step": 9771
    },
    {
      "epoch": 2.2234357224118315,
      "grad_norm": 0.8588347708509323,
      "learning_rate": 7.331487834291712e-07,
      "loss": 0.0031,
      "step": 9772
    },
    {
      "epoch": 2.2236632536973833,
      "grad_norm": 0.3094536562946835,
      "learning_rate": 7.33060778849696e-07,
      "loss": 0.0019,
      "step": 9773
    },
    {
      "epoch": 2.223890784982935,
      "grad_norm": 0.8073760403098668,
      "learning_rate": 7.329727720616564e-07,
      "loss": 0.0048,
      "step": 9774
    },
    {
      "epoch": 2.2241183162684868,
      "grad_norm": 1.2548787501600978,
      "learning_rate": 7.328847630668508e-07,
      "loss": 0.0192,
      "step": 9775
    },
    {
      "epoch": 2.2243458475540385,
      "grad_norm": 0.7128602785656115,
      "learning_rate": 7.327967518670784e-07,
      "loss": 0.0137,
      "step": 9776
    },
    {
      "epoch": 2.2245733788395903,
      "grad_norm": 0.4251278673353657,
      "learning_rate": 7.327087384641378e-07,
      "loss": 0.0047,
      "step": 9777
    },
    {
      "epoch": 2.224800910125142,
      "grad_norm": 0.46802974253945123,
      "learning_rate": 7.326207228598278e-07,
      "loss": 0.0045,
      "step": 9778
    },
    {
      "epoch": 2.225028441410694,
      "grad_norm": 0.8979014112760015,
      "learning_rate": 7.325327050559473e-07,
      "loss": 0.0112,
      "step": 9779
    },
    {
      "epoch": 2.2252559726962455,
      "grad_norm": 0.6449635573539817,
      "learning_rate": 7.324446850542954e-07,
      "loss": 0.0077,
      "step": 9780
    },
    {
      "epoch": 2.2254835039817973,
      "grad_norm": 0.8246198874106694,
      "learning_rate": 7.32356662856671e-07,
      "loss": 0.0112,
      "step": 9781
    },
    {
      "epoch": 2.2257110352673495,
      "grad_norm": 1.2500208791308247,
      "learning_rate": 7.322686384648727e-07,
      "loss": 0.0149,
      "step": 9782
    },
    {
      "epoch": 2.225938566552901,
      "grad_norm": 1.2583446061023784,
      "learning_rate": 7.321806118807e-07,
      "loss": 0.0129,
      "step": 9783
    },
    {
      "epoch": 2.226166097838453,
      "grad_norm": 0.9198656547797969,
      "learning_rate": 7.320925831059519e-07,
      "loss": 0.0076,
      "step": 9784
    },
    {
      "epoch": 2.2263936291240047,
      "grad_norm": 0.4272487565003964,
      "learning_rate": 7.320045521424277e-07,
      "loss": 0.006,
      "step": 9785
    },
    {
      "epoch": 2.2266211604095565,
      "grad_norm": 1.0344296558168884,
      "learning_rate": 7.31916518991926e-07,
      "loss": 0.0092,
      "step": 9786
    },
    {
      "epoch": 2.2268486916951082,
      "grad_norm": 1.6360792104942652,
      "learning_rate": 7.318284836562469e-07,
      "loss": 0.0207,
      "step": 9787
    },
    {
      "epoch": 2.22707622298066,
      "grad_norm": 1.0966744290252408,
      "learning_rate": 7.31740446137189e-07,
      "loss": 0.006,
      "step": 9788
    },
    {
      "epoch": 2.2273037542662117,
      "grad_norm": 1.3749354643682972,
      "learning_rate": 7.31652406436552e-07,
      "loss": 0.0258,
      "step": 9789
    },
    {
      "epoch": 2.2275312855517635,
      "grad_norm": 0.47025859751517013,
      "learning_rate": 7.315643645561351e-07,
      "loss": 0.0059,
      "step": 9790
    },
    {
      "epoch": 2.2277588168373152,
      "grad_norm": 0.651972060161316,
      "learning_rate": 7.314763204977376e-07,
      "loss": 0.007,
      "step": 9791
    },
    {
      "epoch": 2.227986348122867,
      "grad_norm": 1.700188115563667,
      "learning_rate": 7.313882742631594e-07,
      "loss": 0.0123,
      "step": 9792
    },
    {
      "epoch": 2.2282138794084188,
      "grad_norm": 0.7057590972401023,
      "learning_rate": 7.313002258541994e-07,
      "loss": 0.0051,
      "step": 9793
    },
    {
      "epoch": 2.2284414106939705,
      "grad_norm": 0.7493382724211597,
      "learning_rate": 7.312121752726577e-07,
      "loss": 0.0088,
      "step": 9794
    },
    {
      "epoch": 2.2286689419795223,
      "grad_norm": 1.2755977608794125,
      "learning_rate": 7.311241225203336e-07,
      "loss": 0.0091,
      "step": 9795
    },
    {
      "epoch": 2.228896473265074,
      "grad_norm": 0.8875300782255566,
      "learning_rate": 7.310360675990266e-07,
      "loss": 0.0127,
      "step": 9796
    },
    {
      "epoch": 2.2291240045506258,
      "grad_norm": 0.4464116210139732,
      "learning_rate": 7.309480105105368e-07,
      "loss": 0.0045,
      "step": 9797
    },
    {
      "epoch": 2.2293515358361775,
      "grad_norm": 0.41302439996092777,
      "learning_rate": 7.308599512566636e-07,
      "loss": 0.0043,
      "step": 9798
    },
    {
      "epoch": 2.2295790671217293,
      "grad_norm": 0.42317329442010415,
      "learning_rate": 7.30771889839207e-07,
      "loss": 0.0032,
      "step": 9799
    },
    {
      "epoch": 2.229806598407281,
      "grad_norm": 0.9520388291511624,
      "learning_rate": 7.306838262599666e-07,
      "loss": 0.0161,
      "step": 9800
    },
    {
      "epoch": 2.2300341296928328,
      "grad_norm": 1.0791193206115408,
      "learning_rate": 7.305957605207423e-07,
      "loss": 0.0136,
      "step": 9801
    },
    {
      "epoch": 2.2302616609783845,
      "grad_norm": 1.0202899053623313,
      "learning_rate": 7.30507692623334e-07,
      "loss": 0.0136,
      "step": 9802
    },
    {
      "epoch": 2.2304891922639363,
      "grad_norm": 1.02029317561812,
      "learning_rate": 7.304196225695417e-07,
      "loss": 0.0114,
      "step": 9803
    },
    {
      "epoch": 2.230716723549488,
      "grad_norm": 0.938555627060815,
      "learning_rate": 7.303315503611654e-07,
      "loss": 0.0179,
      "step": 9804
    },
    {
      "epoch": 2.2309442548350398,
      "grad_norm": 1.2107991875616646,
      "learning_rate": 7.302434760000053e-07,
      "loss": 0.0189,
      "step": 9805
    },
    {
      "epoch": 2.2311717861205915,
      "grad_norm": 0.7699707256139708,
      "learning_rate": 7.301553994878613e-07,
      "loss": 0.0136,
      "step": 9806
    },
    {
      "epoch": 2.2313993174061433,
      "grad_norm": 0.7647305051147182,
      "learning_rate": 7.300673208265332e-07,
      "loss": 0.0039,
      "step": 9807
    },
    {
      "epoch": 2.231626848691695,
      "grad_norm": 0.8948089189331243,
      "learning_rate": 7.299792400178219e-07,
      "loss": 0.01,
      "step": 9808
    },
    {
      "epoch": 2.231854379977247,
      "grad_norm": 0.5862296396402799,
      "learning_rate": 7.298911570635267e-07,
      "loss": 0.0046,
      "step": 9809
    },
    {
      "epoch": 2.2320819112627985,
      "grad_norm": 0.4150383305633549,
      "learning_rate": 7.298030719654487e-07,
      "loss": 0.0056,
      "step": 9810
    },
    {
      "epoch": 2.2323094425483503,
      "grad_norm": 0.3408274369550502,
      "learning_rate": 7.297149847253877e-07,
      "loss": 0.003,
      "step": 9811
    },
    {
      "epoch": 2.232536973833902,
      "grad_norm": 0.8281706591502016,
      "learning_rate": 7.296268953451443e-07,
      "loss": 0.0164,
      "step": 9812
    },
    {
      "epoch": 2.232764505119454,
      "grad_norm": 0.8156682184149247,
      "learning_rate": 7.295388038265188e-07,
      "loss": 0.0117,
      "step": 9813
    },
    {
      "epoch": 2.2329920364050055,
      "grad_norm": 0.31356092314798434,
      "learning_rate": 7.294507101713115e-07,
      "loss": 0.0026,
      "step": 9814
    },
    {
      "epoch": 2.2332195676905573,
      "grad_norm": 0.1722784825797177,
      "learning_rate": 7.29362614381323e-07,
      "loss": 0.0007,
      "step": 9815
    },
    {
      "epoch": 2.233447098976109,
      "grad_norm": 0.9121187741998779,
      "learning_rate": 7.292745164583537e-07,
      "loss": 0.0071,
      "step": 9816
    },
    {
      "epoch": 2.233674630261661,
      "grad_norm": 0.4238101403118168,
      "learning_rate": 7.291864164042042e-07,
      "loss": 0.0042,
      "step": 9817
    },
    {
      "epoch": 2.2339021615472126,
      "grad_norm": 1.3971696239193727,
      "learning_rate": 7.290983142206756e-07,
      "loss": 0.0062,
      "step": 9818
    },
    {
      "epoch": 2.2341296928327643,
      "grad_norm": 2.0004624861085603,
      "learning_rate": 7.290102099095676e-07,
      "loss": 0.0171,
      "step": 9819
    },
    {
      "epoch": 2.234357224118316,
      "grad_norm": 0.7587978882187041,
      "learning_rate": 7.289221034726816e-07,
      "loss": 0.0126,
      "step": 9820
    },
    {
      "epoch": 2.2345847554038683,
      "grad_norm": 1.5659537091049416,
      "learning_rate": 7.288339949118182e-07,
      "loss": 0.034,
      "step": 9821
    },
    {
      "epoch": 2.23481228668942,
      "grad_norm": 0.5574965430835777,
      "learning_rate": 7.287458842287781e-07,
      "loss": 0.0121,
      "step": 9822
    },
    {
      "epoch": 2.2350398179749718,
      "grad_norm": 0.8150748717339137,
      "learning_rate": 7.286577714253619e-07,
      "loss": 0.0055,
      "step": 9823
    },
    {
      "epoch": 2.2352673492605235,
      "grad_norm": 0.9893593702832831,
      "learning_rate": 7.285696565033711e-07,
      "loss": 0.013,
      "step": 9824
    },
    {
      "epoch": 2.2354948805460753,
      "grad_norm": 0.7106957623380815,
      "learning_rate": 7.284815394646058e-07,
      "loss": 0.0133,
      "step": 9825
    },
    {
      "epoch": 2.235722411831627,
      "grad_norm": 0.7503666419101801,
      "learning_rate": 7.283934203108675e-07,
      "loss": 0.0083,
      "step": 9826
    },
    {
      "epoch": 2.2359499431171788,
      "grad_norm": 1.5961350157212673,
      "learning_rate": 7.283052990439571e-07,
      "loss": 0.0203,
      "step": 9827
    },
    {
      "epoch": 2.2361774744027305,
      "grad_norm": 0.9131961315726698,
      "learning_rate": 7.282171756656756e-07,
      "loss": 0.0104,
      "step": 9828
    },
    {
      "epoch": 2.2364050056882823,
      "grad_norm": 0.7092513159299336,
      "learning_rate": 7.281290501778238e-07,
      "loss": 0.0083,
      "step": 9829
    },
    {
      "epoch": 2.236632536973834,
      "grad_norm": 0.6068866460357687,
      "learning_rate": 7.280409225822033e-07,
      "loss": 0.0045,
      "step": 9830
    },
    {
      "epoch": 2.2368600682593858,
      "grad_norm": 1.1164599055298168,
      "learning_rate": 7.279527928806152e-07,
      "loss": 0.0176,
      "step": 9831
    },
    {
      "epoch": 2.2370875995449375,
      "grad_norm": 1.7549284881303862,
      "learning_rate": 7.278646610748602e-07,
      "loss": 0.0042,
      "step": 9832
    },
    {
      "epoch": 2.2373151308304893,
      "grad_norm": 1.2772923637454094,
      "learning_rate": 7.277765271667402e-07,
      "loss": 0.0107,
      "step": 9833
    },
    {
      "epoch": 2.237542662116041,
      "grad_norm": 1.1558772857109452,
      "learning_rate": 7.276883911580561e-07,
      "loss": 0.0143,
      "step": 9834
    },
    {
      "epoch": 2.2377701934015928,
      "grad_norm": 0.13811763035516314,
      "learning_rate": 7.276002530506094e-07,
      "loss": 0.0008,
      "step": 9835
    },
    {
      "epoch": 2.2379977246871445,
      "grad_norm": 1.3838590962790442,
      "learning_rate": 7.275121128462012e-07,
      "loss": 0.0155,
      "step": 9836
    },
    {
      "epoch": 2.2382252559726963,
      "grad_norm": 0.5462770499918175,
      "learning_rate": 7.274239705466336e-07,
      "loss": 0.0032,
      "step": 9837
    },
    {
      "epoch": 2.238452787258248,
      "grad_norm": 0.4650289070672557,
      "learning_rate": 7.273358261537074e-07,
      "loss": 0.0077,
      "step": 9838
    },
    {
      "epoch": 2.2386803185438,
      "grad_norm": 0.7486677476056879,
      "learning_rate": 7.272476796692242e-07,
      "loss": 0.006,
      "step": 9839
    },
    {
      "epoch": 2.2389078498293515,
      "grad_norm": 1.240466182892842,
      "learning_rate": 7.271595310949858e-07,
      "loss": 0.0184,
      "step": 9840
    },
    {
      "epoch": 2.2391353811149033,
      "grad_norm": 1.198743237576172,
      "learning_rate": 7.270713804327937e-07,
      "loss": 0.0101,
      "step": 9841
    },
    {
      "epoch": 2.239362912400455,
      "grad_norm": 0.5698215052417472,
      "learning_rate": 7.269832276844494e-07,
      "loss": 0.0054,
      "step": 9842
    },
    {
      "epoch": 2.239590443686007,
      "grad_norm": 1.056366075940522,
      "learning_rate": 7.26895072851755e-07,
      "loss": 0.0158,
      "step": 9843
    },
    {
      "epoch": 2.2398179749715585,
      "grad_norm": 0.9334759081557071,
      "learning_rate": 7.268069159365117e-07,
      "loss": 0.0127,
      "step": 9844
    },
    {
      "epoch": 2.2400455062571103,
      "grad_norm": 0.3535598005006818,
      "learning_rate": 7.267187569405215e-07,
      "loss": 0.0006,
      "step": 9845
    },
    {
      "epoch": 2.240273037542662,
      "grad_norm": 1.6242627835730075,
      "learning_rate": 7.266305958655861e-07,
      "loss": 0.0142,
      "step": 9846
    },
    {
      "epoch": 2.240500568828214,
      "grad_norm": 0.8513253296568062,
      "learning_rate": 7.265424327135077e-07,
      "loss": 0.0057,
      "step": 9847
    },
    {
      "epoch": 2.2407281001137656,
      "grad_norm": 0.7318184998351438,
      "learning_rate": 7.264542674860878e-07,
      "loss": 0.0121,
      "step": 9848
    },
    {
      "epoch": 2.2409556313993173,
      "grad_norm": 2.563118781437132,
      "learning_rate": 7.263661001851284e-07,
      "loss": 0.0207,
      "step": 9849
    },
    {
      "epoch": 2.241183162684869,
      "grad_norm": 0.8512064646495209,
      "learning_rate": 7.262779308124317e-07,
      "loss": 0.0146,
      "step": 9850
    },
    {
      "epoch": 2.241410693970421,
      "grad_norm": 4.806450743698874,
      "learning_rate": 7.261897593697995e-07,
      "loss": 0.0623,
      "step": 9851
    },
    {
      "epoch": 2.2416382252559726,
      "grad_norm": 1.0142721854804766,
      "learning_rate": 7.26101585859034e-07,
      "loss": 0.0125,
      "step": 9852
    },
    {
      "epoch": 2.2418657565415243,
      "grad_norm": 0.8079928288463585,
      "learning_rate": 7.26013410281937e-07,
      "loss": 0.0112,
      "step": 9853
    },
    {
      "epoch": 2.242093287827076,
      "grad_norm": 0.7888005296122781,
      "learning_rate": 7.259252326403111e-07,
      "loss": 0.0047,
      "step": 9854
    },
    {
      "epoch": 2.242320819112628,
      "grad_norm": 2.24849494075429,
      "learning_rate": 7.258370529359583e-07,
      "loss": 0.0158,
      "step": 9855
    },
    {
      "epoch": 2.2425483503981796,
      "grad_norm": 0.6208958590234988,
      "learning_rate": 7.257488711706808e-07,
      "loss": 0.0079,
      "step": 9856
    },
    {
      "epoch": 2.2427758816837313,
      "grad_norm": 0.9012489777215701,
      "learning_rate": 7.256606873462808e-07,
      "loss": 0.0092,
      "step": 9857
    },
    {
      "epoch": 2.243003412969283,
      "grad_norm": 0.7591352408814183,
      "learning_rate": 7.255725014645608e-07,
      "loss": 0.0073,
      "step": 9858
    },
    {
      "epoch": 2.243230944254835,
      "grad_norm": 0.3850374073560092,
      "learning_rate": 7.254843135273229e-07,
      "loss": 0.0037,
      "step": 9859
    },
    {
      "epoch": 2.243458475540387,
      "grad_norm": 29.04233693615321,
      "learning_rate": 7.253961235363699e-07,
      "loss": 0.2345,
      "step": 9860
    },
    {
      "epoch": 2.2436860068259388,
      "grad_norm": 0.7503943638738817,
      "learning_rate": 7.253079314935038e-07,
      "loss": 0.0082,
      "step": 9861
    },
    {
      "epoch": 2.2439135381114905,
      "grad_norm": 0.6280314704325309,
      "learning_rate": 7.252197374005273e-07,
      "loss": 0.0033,
      "step": 9862
    },
    {
      "epoch": 2.2441410693970423,
      "grad_norm": 0.877531951440151,
      "learning_rate": 7.251315412592431e-07,
      "loss": 0.0146,
      "step": 9863
    },
    {
      "epoch": 2.244368600682594,
      "grad_norm": 1.9113251453059674,
      "learning_rate": 7.250433430714534e-07,
      "loss": 0.0267,
      "step": 9864
    },
    {
      "epoch": 2.244596131968146,
      "grad_norm": 1.5745364735804175,
      "learning_rate": 7.249551428389612e-07,
      "loss": 0.0246,
      "step": 9865
    },
    {
      "epoch": 2.2448236632536975,
      "grad_norm": 0.4839943673425523,
      "learning_rate": 7.248669405635686e-07,
      "loss": 0.0058,
      "step": 9866
    },
    {
      "epoch": 2.2450511945392493,
      "grad_norm": 0.7177519233243725,
      "learning_rate": 7.247787362470789e-07,
      "loss": 0.0088,
      "step": 9867
    },
    {
      "epoch": 2.245278725824801,
      "grad_norm": 1.415176869636246,
      "learning_rate": 7.246905298912945e-07,
      "loss": 0.016,
      "step": 9868
    },
    {
      "epoch": 2.245506257110353,
      "grad_norm": 0.6967365816363168,
      "learning_rate": 7.246023214980183e-07,
      "loss": 0.0134,
      "step": 9869
    },
    {
      "epoch": 2.2457337883959045,
      "grad_norm": 1.0011183121512415,
      "learning_rate": 7.245141110690529e-07,
      "loss": 0.0101,
      "step": 9870
    },
    {
      "epoch": 2.2459613196814563,
      "grad_norm": 0.9370592883686928,
      "learning_rate": 7.244258986062015e-07,
      "loss": 0.0093,
      "step": 9871
    },
    {
      "epoch": 2.246188850967008,
      "grad_norm": 0.6341452038820068,
      "learning_rate": 7.243376841112668e-07,
      "loss": 0.0164,
      "step": 9872
    },
    {
      "epoch": 2.24641638225256,
      "grad_norm": 1.3667640257675806,
      "learning_rate": 7.242494675860515e-07,
      "loss": 0.0284,
      "step": 9873
    },
    {
      "epoch": 2.2466439135381115,
      "grad_norm": 0.8021917566369299,
      "learning_rate": 7.241612490323591e-07,
      "loss": 0.0146,
      "step": 9874
    },
    {
      "epoch": 2.2468714448236633,
      "grad_norm": 0.8883651217319947,
      "learning_rate": 7.240730284519924e-07,
      "loss": 0.009,
      "step": 9875
    },
    {
      "epoch": 2.247098976109215,
      "grad_norm": 2.4725039536714735,
      "learning_rate": 7.239848058467544e-07,
      "loss": 0.0698,
      "step": 9876
    },
    {
      "epoch": 2.247326507394767,
      "grad_norm": 1.4231309129974479,
      "learning_rate": 7.238965812184482e-07,
      "loss": 0.0114,
      "step": 9877
    },
    {
      "epoch": 2.2475540386803186,
      "grad_norm": 1.3068513930464432,
      "learning_rate": 7.23808354568877e-07,
      "loss": 0.0209,
      "step": 9878
    },
    {
      "epoch": 2.2477815699658703,
      "grad_norm": 1.0823713349372646,
      "learning_rate": 7.23720125899844e-07,
      "loss": 0.0202,
      "step": 9879
    },
    {
      "epoch": 2.248009101251422,
      "grad_norm": 0.5500138347976872,
      "learning_rate": 7.236318952131524e-07,
      "loss": 0.0046,
      "step": 9880
    },
    {
      "epoch": 2.248236632536974,
      "grad_norm": 0.37886665958519683,
      "learning_rate": 7.235436625106057e-07,
      "loss": 0.005,
      "step": 9881
    },
    {
      "epoch": 2.2484641638225256,
      "grad_norm": 0.3946397254837047,
      "learning_rate": 7.234554277940067e-07,
      "loss": 0.0048,
      "step": 9882
    },
    {
      "epoch": 2.2486916951080773,
      "grad_norm": 0.3939158732248495,
      "learning_rate": 7.233671910651592e-07,
      "loss": 0.0033,
      "step": 9883
    },
    {
      "epoch": 2.248919226393629,
      "grad_norm": 0.8027707642637251,
      "learning_rate": 7.232789523258665e-07,
      "loss": 0.0154,
      "step": 9884
    },
    {
      "epoch": 2.249146757679181,
      "grad_norm": 1.937385341864222,
      "learning_rate": 7.23190711577932e-07,
      "loss": 0.0064,
      "step": 9885
    },
    {
      "epoch": 2.2493742889647326,
      "grad_norm": 0.6952305635914994,
      "learning_rate": 7.23102468823159e-07,
      "loss": 0.008,
      "step": 9886
    },
    {
      "epoch": 2.2496018202502843,
      "grad_norm": 1.6963099035620435,
      "learning_rate": 7.230142240633515e-07,
      "loss": 0.0236,
      "step": 9887
    },
    {
      "epoch": 2.249829351535836,
      "grad_norm": 1.4712782075428819,
      "learning_rate": 7.229259773003128e-07,
      "loss": 0.0067,
      "step": 9888
    },
    {
      "epoch": 2.250056882821388,
      "grad_norm": 1.4019973115143842,
      "learning_rate": 7.228377285358461e-07,
      "loss": 0.035,
      "step": 9889
    },
    {
      "epoch": 2.2502844141069396,
      "grad_norm": 0.8709636448921553,
      "learning_rate": 7.227494777717555e-07,
      "loss": 0.0216,
      "step": 9890
    },
    {
      "epoch": 2.2505119453924913,
      "grad_norm": 0.7050296491753442,
      "learning_rate": 7.226612250098449e-07,
      "loss": 0.0078,
      "step": 9891
    },
    {
      "epoch": 2.250739476678043,
      "grad_norm": 0.9119132940087451,
      "learning_rate": 7.225729702519174e-07,
      "loss": 0.0081,
      "step": 9892
    },
    {
      "epoch": 2.250967007963595,
      "grad_norm": 1.0967216519794032,
      "learning_rate": 7.224847134997772e-07,
      "loss": 0.0165,
      "step": 9893
    },
    {
      "epoch": 2.2511945392491466,
      "grad_norm": 3.341890355154279,
      "learning_rate": 7.223964547552281e-07,
      "loss": 0.042,
      "step": 9894
    },
    {
      "epoch": 2.2514220705346983,
      "grad_norm": 0.815166521141191,
      "learning_rate": 7.223081940200738e-07,
      "loss": 0.0118,
      "step": 9895
    },
    {
      "epoch": 2.25164960182025,
      "grad_norm": 0.7029474873706334,
      "learning_rate": 7.22219931296118e-07,
      "loss": 0.0091,
      "step": 9896
    },
    {
      "epoch": 2.2518771331058023,
      "grad_norm": 0.4637350032709855,
      "learning_rate": 7.22131666585165e-07,
      "loss": 0.0049,
      "step": 9897
    },
    {
      "epoch": 2.2521046643913536,
      "grad_norm": 0.8241535144472323,
      "learning_rate": 7.220433998890188e-07,
      "loss": 0.0112,
      "step": 9898
    },
    {
      "epoch": 2.252332195676906,
      "grad_norm": 0.20136236788487277,
      "learning_rate": 7.21955131209483e-07,
      "loss": 0.0011,
      "step": 9899
    },
    {
      "epoch": 2.252559726962457,
      "grad_norm": 2.2955301241086192,
      "learning_rate": 7.21866860548362e-07,
      "loss": 0.09,
      "step": 9900
    },
    {
      "epoch": 2.2527872582480093,
      "grad_norm": 0.5101507391858183,
      "learning_rate": 7.217785879074597e-07,
      "loss": 0.0075,
      "step": 9901
    },
    {
      "epoch": 2.253014789533561,
      "grad_norm": 1.111182269507313,
      "learning_rate": 7.216903132885803e-07,
      "loss": 0.0163,
      "step": 9902
    },
    {
      "epoch": 2.253242320819113,
      "grad_norm": 1.06050499482352,
      "learning_rate": 7.216020366935279e-07,
      "loss": 0.01,
      "step": 9903
    },
    {
      "epoch": 2.2534698521046646,
      "grad_norm": 1.1864788126580075,
      "learning_rate": 7.215137581241071e-07,
      "loss": 0.0178,
      "step": 9904
    },
    {
      "epoch": 2.2536973833902163,
      "grad_norm": 0.19975498662253313,
      "learning_rate": 7.214254775821216e-07,
      "loss": 0.0016,
      "step": 9905
    },
    {
      "epoch": 2.253924914675768,
      "grad_norm": 0.9805863537396342,
      "learning_rate": 7.213371950693759e-07,
      "loss": 0.016,
      "step": 9906
    },
    {
      "epoch": 2.25415244596132,
      "grad_norm": 0.5856181787788934,
      "learning_rate": 7.212489105876745e-07,
      "loss": 0.004,
      "step": 9907
    },
    {
      "epoch": 2.2543799772468716,
      "grad_norm": 0.2733027188406292,
      "learning_rate": 7.211606241388217e-07,
      "loss": 0.0018,
      "step": 9908
    },
    {
      "epoch": 2.2546075085324233,
      "grad_norm": 0.8297275694957768,
      "learning_rate": 7.210723357246216e-07,
      "loss": 0.0073,
      "step": 9909
    },
    {
      "epoch": 2.254835039817975,
      "grad_norm": 1.4004828171199504,
      "learning_rate": 7.20984045346879e-07,
      "loss": 0.0118,
      "step": 9910
    },
    {
      "epoch": 2.255062571103527,
      "grad_norm": 0.9866359770683436,
      "learning_rate": 7.208957530073983e-07,
      "loss": 0.0146,
      "step": 9911
    },
    {
      "epoch": 2.2552901023890786,
      "grad_norm": 0.4227327466029483,
      "learning_rate": 7.208074587079841e-07,
      "loss": 0.0039,
      "step": 9912
    },
    {
      "epoch": 2.2555176336746303,
      "grad_norm": 0.4973724222704189,
      "learning_rate": 7.207191624504409e-07,
      "loss": 0.0086,
      "step": 9913
    },
    {
      "epoch": 2.255745164960182,
      "grad_norm": 1.229368217025756,
      "learning_rate": 7.206308642365733e-07,
      "loss": 0.0071,
      "step": 9914
    },
    {
      "epoch": 2.255972696245734,
      "grad_norm": 0.6723357562159281,
      "learning_rate": 7.20542564068186e-07,
      "loss": 0.0085,
      "step": 9915
    },
    {
      "epoch": 2.2562002275312856,
      "grad_norm": 1.0110097193814493,
      "learning_rate": 7.204542619470837e-07,
      "loss": 0.0183,
      "step": 9916
    },
    {
      "epoch": 2.2564277588168373,
      "grad_norm": 0.8974325972903089,
      "learning_rate": 7.203659578750709e-07,
      "loss": 0.0197,
      "step": 9917
    },
    {
      "epoch": 2.256655290102389,
      "grad_norm": 0.394039393446717,
      "learning_rate": 7.202776518539527e-07,
      "loss": 0.0025,
      "step": 9918
    },
    {
      "epoch": 2.256882821387941,
      "grad_norm": 0.4378212666472128,
      "learning_rate": 7.20189343885534e-07,
      "loss": 0.0031,
      "step": 9919
    },
    {
      "epoch": 2.2571103526734926,
      "grad_norm": 1.841281636521212,
      "learning_rate": 7.201010339716191e-07,
      "loss": 0.0246,
      "step": 9920
    },
    {
      "epoch": 2.2573378839590443,
      "grad_norm": 0.39720051181356536,
      "learning_rate": 7.200127221140134e-07,
      "loss": 0.003,
      "step": 9921
    },
    {
      "epoch": 2.257565415244596,
      "grad_norm": 0.5699912064843484,
      "learning_rate": 7.199244083145217e-07,
      "loss": 0.006,
      "step": 9922
    },
    {
      "epoch": 2.257792946530148,
      "grad_norm": 1.1717119799885392,
      "learning_rate": 7.198360925749488e-07,
      "loss": 0.0049,
      "step": 9923
    },
    {
      "epoch": 2.2580204778156996,
      "grad_norm": 0.9444660239647007,
      "learning_rate": 7.197477748971e-07,
      "loss": 0.0081,
      "step": 9924
    },
    {
      "epoch": 2.2582480091012513,
      "grad_norm": 0.9901137476612132,
      "learning_rate": 7.196594552827803e-07,
      "loss": 0.0068,
      "step": 9925
    },
    {
      "epoch": 2.258475540386803,
      "grad_norm": 0.30968324441541073,
      "learning_rate": 7.195711337337943e-07,
      "loss": 0.0011,
      "step": 9926
    },
    {
      "epoch": 2.258703071672355,
      "grad_norm": 0.8325270530872885,
      "learning_rate": 7.194828102519479e-07,
      "loss": 0.0071,
      "step": 9927
    },
    {
      "epoch": 2.2589306029579066,
      "grad_norm": 0.19973341876397938,
      "learning_rate": 7.193944848390458e-07,
      "loss": 0.0011,
      "step": 9928
    },
    {
      "epoch": 2.2591581342434583,
      "grad_norm": 0.4684436667323531,
      "learning_rate": 7.193061574968932e-07,
      "loss": 0.0038,
      "step": 9929
    },
    {
      "epoch": 2.25938566552901,
      "grad_norm": 1.3140170788234653,
      "learning_rate": 7.192178282272955e-07,
      "loss": 0.0173,
      "step": 9930
    },
    {
      "epoch": 2.259613196814562,
      "grad_norm": 0.7982829634078386,
      "learning_rate": 7.191294970320581e-07,
      "loss": 0.0125,
      "step": 9931
    },
    {
      "epoch": 2.2598407281001136,
      "grad_norm": 0.4916575563622796,
      "learning_rate": 7.19041163912986e-07,
      "loss": 0.0032,
      "step": 9932
    },
    {
      "epoch": 2.2600682593856654,
      "grad_norm": 0.6790168057763974,
      "learning_rate": 7.189528288718846e-07,
      "loss": 0.0043,
      "step": 9933
    },
    {
      "epoch": 2.260295790671217,
      "grad_norm": 0.710603316218467,
      "learning_rate": 7.188644919105597e-07,
      "loss": 0.0109,
      "step": 9934
    },
    {
      "epoch": 2.260523321956769,
      "grad_norm": 0.5574884681160212,
      "learning_rate": 7.187761530308164e-07,
      "loss": 0.0056,
      "step": 9935
    },
    {
      "epoch": 2.260750853242321,
      "grad_norm": 0.15545496798330738,
      "learning_rate": 7.186878122344602e-07,
      "loss": 0.0008,
      "step": 9936
    },
    {
      "epoch": 2.2609783845278724,
      "grad_norm": 1.0354566137465173,
      "learning_rate": 7.185994695232967e-07,
      "loss": 0.0194,
      "step": 9937
    },
    {
      "epoch": 2.2612059158134246,
      "grad_norm": 1.225411334353908,
      "learning_rate": 7.185111248991318e-07,
      "loss": 0.0101,
      "step": 9938
    },
    {
      "epoch": 2.261433447098976,
      "grad_norm": 1.0853192029390262,
      "learning_rate": 7.184227783637705e-07,
      "loss": 0.0132,
      "step": 9939
    },
    {
      "epoch": 2.261660978384528,
      "grad_norm": 0.997677365905451,
      "learning_rate": 7.183344299190186e-07,
      "loss": 0.0106,
      "step": 9940
    },
    {
      "epoch": 2.26188850967008,
      "grad_norm": 1.0530305107415374,
      "learning_rate": 7.182460795666821e-07,
      "loss": 0.0058,
      "step": 9941
    },
    {
      "epoch": 2.2621160409556316,
      "grad_norm": 0.5645944290761391,
      "learning_rate": 7.181577273085663e-07,
      "loss": 0.0028,
      "step": 9942
    },
    {
      "epoch": 2.2623435722411833,
      "grad_norm": 0.9816035838856915,
      "learning_rate": 7.180693731464773e-07,
      "loss": 0.0067,
      "step": 9943
    },
    {
      "epoch": 2.262571103526735,
      "grad_norm": 1.1825829184757581,
      "learning_rate": 7.179810170822208e-07,
      "loss": 0.0172,
      "step": 9944
    },
    {
      "epoch": 2.262798634812287,
      "grad_norm": 1.3190733706115991,
      "learning_rate": 7.178926591176025e-07,
      "loss": 0.0113,
      "step": 9945
    },
    {
      "epoch": 2.2630261660978386,
      "grad_norm": 0.41854947979742385,
      "learning_rate": 7.178042992544284e-07,
      "loss": 0.0037,
      "step": 9946
    },
    {
      "epoch": 2.2632536973833903,
      "grad_norm": 1.0135387984521642,
      "learning_rate": 7.177159374945043e-07,
      "loss": 0.0149,
      "step": 9947
    },
    {
      "epoch": 2.263481228668942,
      "grad_norm": 1.3368543905703725,
      "learning_rate": 7.176275738396363e-07,
      "loss": 0.0214,
      "step": 9948
    },
    {
      "epoch": 2.263708759954494,
      "grad_norm": 0.2887938541541053,
      "learning_rate": 7.175392082916305e-07,
      "loss": 0.0023,
      "step": 9949
    },
    {
      "epoch": 2.2639362912400456,
      "grad_norm": 1.15649587646679,
      "learning_rate": 7.174508408522926e-07,
      "loss": 0.0218,
      "step": 9950
    },
    {
      "epoch": 2.2641638225255973,
      "grad_norm": 1.0800510285096194,
      "learning_rate": 7.173624715234288e-07,
      "loss": 0.0147,
      "step": 9951
    },
    {
      "epoch": 2.264391353811149,
      "grad_norm": 0.6160351854072771,
      "learning_rate": 7.172741003068454e-07,
      "loss": 0.0024,
      "step": 9952
    },
    {
      "epoch": 2.264618885096701,
      "grad_norm": 0.46827063444077116,
      "learning_rate": 7.171857272043481e-07,
      "loss": 0.0028,
      "step": 9953
    },
    {
      "epoch": 2.2648464163822526,
      "grad_norm": 0.5611835041131369,
      "learning_rate": 7.170973522177435e-07,
      "loss": 0.0025,
      "step": 9954
    },
    {
      "epoch": 2.2650739476678043,
      "grad_norm": 0.48161928840126583,
      "learning_rate": 7.170089753488378e-07,
      "loss": 0.0038,
      "step": 9955
    },
    {
      "epoch": 2.265301478953356,
      "grad_norm": 1.2060593955214094,
      "learning_rate": 7.169205965994371e-07,
      "loss": 0.0152,
      "step": 9956
    },
    {
      "epoch": 2.265529010238908,
      "grad_norm": 0.5609727284015287,
      "learning_rate": 7.168322159713477e-07,
      "loss": 0.007,
      "step": 9957
    },
    {
      "epoch": 2.2657565415244596,
      "grad_norm": 0.9866672053615072,
      "learning_rate": 7.16743833466376e-07,
      "loss": 0.0048,
      "step": 9958
    },
    {
      "epoch": 2.2659840728100114,
      "grad_norm": 0.37810180326163845,
      "learning_rate": 7.166554490863283e-07,
      "loss": 0.0039,
      "step": 9959
    },
    {
      "epoch": 2.266211604095563,
      "grad_norm": 0.1938911177720205,
      "learning_rate": 7.165670628330112e-07,
      "loss": 0.0013,
      "step": 9960
    },
    {
      "epoch": 2.266439135381115,
      "grad_norm": 0.4583894174535528,
      "learning_rate": 7.16478674708231e-07,
      "loss": 0.003,
      "step": 9961
    },
    {
      "epoch": 2.2666666666666666,
      "grad_norm": 1.3040987284773864,
      "learning_rate": 7.163902847137942e-07,
      "loss": 0.0091,
      "step": 9962
    },
    {
      "epoch": 2.2668941979522184,
      "grad_norm": 0.7712497233194445,
      "learning_rate": 7.163018928515074e-07,
      "loss": 0.0123,
      "step": 9963
    },
    {
      "epoch": 2.26712172923777,
      "grad_norm": 1.3301068153214746,
      "learning_rate": 7.16213499123177e-07,
      "loss": 0.014,
      "step": 9964
    },
    {
      "epoch": 2.267349260523322,
      "grad_norm": 2.8241366237585894,
      "learning_rate": 7.161251035306099e-07,
      "loss": 0.0132,
      "step": 9965
    },
    {
      "epoch": 2.2675767918088736,
      "grad_norm": 0.801120085816114,
      "learning_rate": 7.160367060756125e-07,
      "loss": 0.0088,
      "step": 9966
    },
    {
      "epoch": 2.2678043230944254,
      "grad_norm": 0.9044674800670061,
      "learning_rate": 7.159483067599913e-07,
      "loss": 0.0144,
      "step": 9967
    },
    {
      "epoch": 2.268031854379977,
      "grad_norm": 0.6123545477064796,
      "learning_rate": 7.158599055855536e-07,
      "loss": 0.007,
      "step": 9968
    },
    {
      "epoch": 2.268259385665529,
      "grad_norm": 0.819322434683557,
      "learning_rate": 7.157715025541059e-07,
      "loss": 0.0158,
      "step": 9969
    },
    {
      "epoch": 2.2684869169510806,
      "grad_norm": 0.7630562241053435,
      "learning_rate": 7.156830976674547e-07,
      "loss": 0.0108,
      "step": 9970
    },
    {
      "epoch": 2.2687144482366324,
      "grad_norm": 0.9506425770060405,
      "learning_rate": 7.155946909274071e-07,
      "loss": 0.0143,
      "step": 9971
    },
    {
      "epoch": 2.268941979522184,
      "grad_norm": 0.8054458430852172,
      "learning_rate": 7.1550628233577e-07,
      "loss": 0.0123,
      "step": 9972
    },
    {
      "epoch": 2.269169510807736,
      "grad_norm": 0.7467245350913407,
      "learning_rate": 7.154178718943502e-07,
      "loss": 0.0089,
      "step": 9973
    },
    {
      "epoch": 2.2693970420932876,
      "grad_norm": 0.378769776387463,
      "learning_rate": 7.153294596049547e-07,
      "loss": 0.0021,
      "step": 9974
    },
    {
      "epoch": 2.26962457337884,
      "grad_norm": 0.71241206876713,
      "learning_rate": 7.152410454693905e-07,
      "loss": 0.0035,
      "step": 9975
    },
    {
      "epoch": 2.269852104664391,
      "grad_norm": 0.7202281998843869,
      "learning_rate": 7.151526294894646e-07,
      "loss": 0.0053,
      "step": 9976
    },
    {
      "epoch": 2.2700796359499433,
      "grad_norm": 1.0591483680969398,
      "learning_rate": 7.150642116669839e-07,
      "loss": 0.0138,
      "step": 9977
    },
    {
      "epoch": 2.2703071672354946,
      "grad_norm": 1.2565068895872433,
      "learning_rate": 7.149757920037558e-07,
      "loss": 0.0067,
      "step": 9978
    },
    {
      "epoch": 2.270534698521047,
      "grad_norm": 1.9560005314266655,
      "learning_rate": 7.148873705015875e-07,
      "loss": 0.0256,
      "step": 9979
    },
    {
      "epoch": 2.2707622298065986,
      "grad_norm": 1.2639553498840532,
      "learning_rate": 7.147989471622856e-07,
      "loss": 0.0113,
      "step": 9980
    },
    {
      "epoch": 2.2709897610921503,
      "grad_norm": 0.8022394009151615,
      "learning_rate": 7.147105219876578e-07,
      "loss": 0.0106,
      "step": 9981
    },
    {
      "epoch": 2.271217292377702,
      "grad_norm": 1.279081961574699,
      "learning_rate": 7.146220949795114e-07,
      "loss": 0.0081,
      "step": 9982
    },
    {
      "epoch": 2.271444823663254,
      "grad_norm": 0.4727614962938117,
      "learning_rate": 7.145336661396532e-07,
      "loss": 0.0082,
      "step": 9983
    },
    {
      "epoch": 2.2716723549488056,
      "grad_norm": 1.579355647620252,
      "learning_rate": 7.144452354698911e-07,
      "loss": 0.0167,
      "step": 9984
    },
    {
      "epoch": 2.2718998862343573,
      "grad_norm": 0.3479168390911897,
      "learning_rate": 7.143568029720321e-07,
      "loss": 0.0033,
      "step": 9985
    },
    {
      "epoch": 2.272127417519909,
      "grad_norm": 0.38954071165498017,
      "learning_rate": 7.142683686478838e-07,
      "loss": 0.0027,
      "step": 9986
    },
    {
      "epoch": 2.272354948805461,
      "grad_norm": 1.5444954312199928,
      "learning_rate": 7.141799324992532e-07,
      "loss": 0.0369,
      "step": 9987
    },
    {
      "epoch": 2.2725824800910126,
      "grad_norm": 1.2367218560853073,
      "learning_rate": 7.140914945279486e-07,
      "loss": 0.0103,
      "step": 9988
    },
    {
      "epoch": 2.2728100113765644,
      "grad_norm": 1.8050517650851405,
      "learning_rate": 7.140030547357768e-07,
      "loss": 0.0247,
      "step": 9989
    },
    {
      "epoch": 2.273037542662116,
      "grad_norm": 1.1044644885041117,
      "learning_rate": 7.139146131245453e-07,
      "loss": 0.0082,
      "step": 9990
    }
  ],
  "logging_steps": 1,
  "max_steps": 21975,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 1110,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 32844969787392.0,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}