{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 3.5964754540550263,
  "eval_steps": 500,
  "global_step": 5000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0007192950908110052,
      "grad_norm": 64.34911621183012,
      "learning_rate": 6.000000000000001e-08,
      "loss": 4.4253,
      "step": 1
    },
    {
      "epoch": 0.0014385901816220104,
      "grad_norm": 51.27350234777956,
      "learning_rate": 1.2000000000000002e-07,
      "loss": 4.157,
      "step": 2
    },
    {
      "epoch": 0.002157885272433016,
      "grad_norm": 58.38570787634046,
      "learning_rate": 1.8e-07,
      "loss": 4.3753,
      "step": 3
    },
    {
      "epoch": 0.002877180363244021,
      "grad_norm": 60.33603591467569,
      "learning_rate": 2.4000000000000003e-07,
      "loss": 4.7219,
      "step": 4
    },
    {
      "epoch": 0.0035964754540550262,
      "grad_norm": 63.405779091424165,
      "learning_rate": 3.0000000000000004e-07,
      "loss": 4.4572,
      "step": 5
    },
    {
      "epoch": 0.004315770544866032,
      "grad_norm": 57.43191624928036,
      "learning_rate": 3.6e-07,
      "loss": 4.3722,
      "step": 6
    },
    {
      "epoch": 0.005035065635677036,
      "grad_norm": 45.80791469017583,
      "learning_rate": 4.2000000000000006e-07,
      "loss": 4.3694,
      "step": 7
    },
    {
      "epoch": 0.005754360726488042,
      "grad_norm": 52.561254618344556,
      "learning_rate": 4.800000000000001e-07,
      "loss": 4.2124,
      "step": 8
    },
    {
      "epoch": 0.006473655817299047,
      "grad_norm": 52.714545202649816,
      "learning_rate": 5.4e-07,
      "loss": 4.1141,
      "step": 9
    },
    {
      "epoch": 0.0071929509081100525,
      "grad_norm": 63.569929213232236,
      "learning_rate": 6.000000000000001e-07,
      "loss": 4.4139,
      "step": 10
    },
    {
      "epoch": 0.007912245998921057,
      "grad_norm": 52.68955747119893,
      "learning_rate": 6.6e-07,
      "loss": 4.37,
      "step": 11
    },
    {
      "epoch": 0.008631541089732063,
      "grad_norm": 44.654498976303714,
      "learning_rate": 7.2e-07,
      "loss": 3.9797,
      "step": 12
    },
    {
      "epoch": 0.009350836180543068,
      "grad_norm": 50.05528844599175,
      "learning_rate": 7.8e-07,
      "loss": 3.9772,
      "step": 13
    },
    {
      "epoch": 0.010070131271354072,
      "grad_norm": 67.73011138752118,
      "learning_rate": 8.400000000000001e-07,
      "loss": 4.2362,
      "step": 14
    },
    {
      "epoch": 0.010789426362165079,
      "grad_norm": 45.67621010960208,
      "learning_rate": 9e-07,
      "loss": 4.0604,
      "step": 15
    },
    {
      "epoch": 0.011508721452976083,
      "grad_norm": 66.74676866766224,
      "learning_rate": 9.600000000000001e-07,
      "loss": 4.1516,
      "step": 16
    },
    {
      "epoch": 0.012228016543787088,
      "grad_norm": 48.062555064438754,
      "learning_rate": 1.0200000000000002e-06,
      "loss": 4.1489,
      "step": 17
    },
    {
      "epoch": 0.012947311634598094,
      "grad_norm": 42.64480446155265,
      "learning_rate": 1.08e-06,
      "loss": 3.5589,
      "step": 18
    },
    {
      "epoch": 0.013666606725409099,
      "grad_norm": 42.07904498757481,
      "learning_rate": 1.14e-06,
      "loss": 3.9886,
      "step": 19
    },
    {
      "epoch": 0.014385901816220105,
      "grad_norm": 48.72108700022635,
      "learning_rate": 1.2000000000000002e-06,
      "loss": 3.4519,
      "step": 20
    },
    {
      "epoch": 0.01510519690703111,
      "grad_norm": 42.250033849621516,
      "learning_rate": 1.26e-06,
      "loss": 3.303,
      "step": 21
    },
    {
      "epoch": 0.015824491997842114,
      "grad_norm": 33.70223333768381,
      "learning_rate": 1.32e-06,
      "loss": 3.6903,
      "step": 22
    },
    {
      "epoch": 0.01654378708865312,
      "grad_norm": 40.15971992201178,
      "learning_rate": 1.3800000000000001e-06,
      "loss": 3.5432,
      "step": 23
    },
    {
      "epoch": 0.017263082179464127,
      "grad_norm": 34.63347885795475,
      "learning_rate": 1.44e-06,
      "loss": 3.0621,
      "step": 24
    },
    {
      "epoch": 0.01798237727027513,
      "grad_norm": 25.449455048743598,
      "learning_rate": 1.5e-06,
      "loss": 2.8903,
      "step": 25
    },
    {
      "epoch": 0.018701672361086136,
      "grad_norm": 21.57750342260305,
      "learning_rate": 1.56e-06,
      "loss": 2.8485,
      "step": 26
    },
    {
      "epoch": 0.019420967451897142,
      "grad_norm": 25.591838990527403,
      "learning_rate": 1.6200000000000002e-06,
      "loss": 2.8427,
      "step": 27
    },
    {
      "epoch": 0.020140262542708145,
      "grad_norm": 21.888134103825376,
      "learning_rate": 1.6800000000000002e-06,
      "loss": 2.5554,
      "step": 28
    },
    {
      "epoch": 0.02085955763351915,
      "grad_norm": 23.093437270571588,
      "learning_rate": 1.7399999999999999e-06,
      "loss": 2.5745,
      "step": 29
    },
    {
      "epoch": 0.021578852724330157,
      "grad_norm": 22.408129250183837,
      "learning_rate": 1.8e-06,
      "loss": 2.8245,
      "step": 30
    },
    {
      "epoch": 0.02229814781514116,
      "grad_norm": 21.62571464737288,
      "learning_rate": 1.86e-06,
      "loss": 2.7638,
      "step": 31
    },
    {
      "epoch": 0.023017442905952167,
      "grad_norm": 22.1754794247378,
      "learning_rate": 1.9200000000000003e-06,
      "loss": 2.566,
      "step": 32
    },
    {
      "epoch": 0.023736737996763173,
      "grad_norm": 17.36822178512086,
      "learning_rate": 1.98e-06,
      "loss": 2.3873,
      "step": 33
    },
    {
      "epoch": 0.024456033087574176,
      "grad_norm": 20.82250738790257,
      "learning_rate": 2.0400000000000004e-06,
      "loss": 2.4609,
      "step": 34
    },
    {
      "epoch": 0.025175328178385182,
      "grad_norm": 21.727374515734958,
      "learning_rate": 2.1e-06,
      "loss": 2.104,
      "step": 35
    },
    {
      "epoch": 0.025894623269196188,
      "grad_norm": 25.192785218202438,
      "learning_rate": 2.16e-06,
      "loss": 2.2068,
      "step": 36
    },
    {
      "epoch": 0.026613918360007194,
      "grad_norm": 22.949876065186068,
      "learning_rate": 2.22e-06,
      "loss": 2.1981,
      "step": 37
    },
    {
      "epoch": 0.027333213450818197,
      "grad_norm": 20.452749514647596,
      "learning_rate": 2.28e-06,
      "loss": 2.1726,
      "step": 38
    },
    {
      "epoch": 0.028052508541629204,
      "grad_norm": 19.783955544164957,
      "learning_rate": 2.34e-06,
      "loss": 2.0562,
      "step": 39
    },
    {
      "epoch": 0.02877180363244021,
      "grad_norm": 21.913942051708446,
      "learning_rate": 2.4000000000000003e-06,
      "loss": 2.5344,
      "step": 40
    },
    {
      "epoch": 0.029491098723251213,
      "grad_norm": 16.568586126115413,
      "learning_rate": 2.4599999999999997e-06,
      "loss": 2.1713,
      "step": 41
    },
    {
      "epoch": 0.03021039381406222,
      "grad_norm": 15.232605106865144,
      "learning_rate": 2.52e-06,
      "loss": 2.2402,
      "step": 42
    },
    {
      "epoch": 0.030929688904873225,
      "grad_norm": 15.106443009338856,
      "learning_rate": 2.58e-06,
      "loss": 2.304,
      "step": 43
    },
    {
      "epoch": 0.03164898399568423,
      "grad_norm": 15.979582692190226,
      "learning_rate": 2.64e-06,
      "loss": 2.0808,
      "step": 44
    },
    {
      "epoch": 0.032368279086495234,
      "grad_norm": 16.092017705403578,
      "learning_rate": 2.7e-06,
      "loss": 2.3539,
      "step": 45
    },
    {
      "epoch": 0.03308757417730624,
      "grad_norm": 16.06526364336813,
      "learning_rate": 2.7600000000000003e-06,
      "loss": 2.2053,
      "step": 46
    },
    {
      "epoch": 0.03380686926811725,
      "grad_norm": 14.316985257725126,
      "learning_rate": 2.82e-06,
      "loss": 1.9945,
      "step": 47
    },
    {
      "epoch": 0.03452616435892825,
      "grad_norm": 14.958090177786831,
      "learning_rate": 2.88e-06,
      "loss": 2.1573,
      "step": 48
    },
    {
      "epoch": 0.03524545944973925,
      "grad_norm": 14.026710889707836,
      "learning_rate": 2.9400000000000002e-06,
      "loss": 2.2067,
      "step": 49
    },
    {
      "epoch": 0.03596475454055026,
      "grad_norm": 14.51423023014951,
      "learning_rate": 3e-06,
      "loss": 1.9742,
      "step": 50
    },
    {
      "epoch": 0.036684049631361265,
      "grad_norm": 15.22192441616925,
      "learning_rate": 2.999999961411183e-06,
      "loss": 2.0712,
      "step": 51
    },
    {
      "epoch": 0.03740334472217227,
      "grad_norm": 14.194763449787578,
      "learning_rate": 2.9999998456447355e-06,
      "loss": 1.9204,
      "step": 52
    },
    {
      "epoch": 0.03812263981298328,
      "grad_norm": 13.667584041378175,
      "learning_rate": 2.9999996527006628e-06,
      "loss": 1.8722,
      "step": 53
    },
    {
      "epoch": 0.038841934903794284,
      "grad_norm": 14.801818205020217,
      "learning_rate": 2.9999993825789743e-06,
      "loss": 1.9964,
      "step": 54
    },
    {
      "epoch": 0.03956122999460528,
      "grad_norm": 14.031676772035228,
      "learning_rate": 2.999999035279685e-06,
      "loss": 1.7175,
      "step": 55
    },
    {
      "epoch": 0.04028052508541629,
      "grad_norm": 15.806689286588222,
      "learning_rate": 2.9999986108028114e-06,
      "loss": 1.7233,
      "step": 56
    },
    {
      "epoch": 0.040999820176227296,
      "grad_norm": 17.51440992749227,
      "learning_rate": 2.9999981091483768e-06,
      "loss": 1.9272,
      "step": 57
    },
    {
      "epoch": 0.0417191152670383,
      "grad_norm": 13.741266330932344,
      "learning_rate": 2.9999975303164056e-06,
      "loss": 1.954,
      "step": 58
    },
    {
      "epoch": 0.04243841035784931,
      "grad_norm": 18.39147673593491,
      "learning_rate": 2.999996874306929e-06,
      "loss": 1.8145,
      "step": 59
    },
    {
      "epoch": 0.043157705448660315,
      "grad_norm": 18.716766390380016,
      "learning_rate": 2.99999614111998e-06,
      "loss": 1.8562,
      "step": 60
    },
    {
      "epoch": 0.04387700053947132,
      "grad_norm": 15.030579384425387,
      "learning_rate": 2.999995330755596e-06,
      "loss": 1.7145,
      "step": 61
    },
    {
      "epoch": 0.04459629563028232,
      "grad_norm": 16.27860968362536,
      "learning_rate": 2.999994443213819e-06,
      "loss": 1.4302,
      "step": 62
    },
    {
      "epoch": 0.04531559072109333,
      "grad_norm": 18.137559091067445,
      "learning_rate": 2.999993478494695e-06,
      "loss": 2.0254,
      "step": 63
    },
    {
      "epoch": 0.04603488581190433,
      "grad_norm": 15.51057143193802,
      "learning_rate": 2.9999924365982735e-06,
      "loss": 1.5322,
      "step": 64
    },
    {
      "epoch": 0.04675418090271534,
      "grad_norm": 18.095173249301645,
      "learning_rate": 2.9999913175246076e-06,
      "loss": 1.7118,
      "step": 65
    },
    {
      "epoch": 0.047473475993526346,
      "grad_norm": 17.045312399277137,
      "learning_rate": 2.9999901212737557e-06,
      "loss": 1.642,
      "step": 66
    },
    {
      "epoch": 0.04819277108433735,
      "grad_norm": 19.706738120317652,
      "learning_rate": 2.9999888478457784e-06,
      "loss": 1.5376,
      "step": 67
    },
    {
      "epoch": 0.04891206617514835,
      "grad_norm": 32.97972115668073,
      "learning_rate": 2.9999874972407424e-06,
      "loss": 1.7039,
      "step": 68
    },
    {
      "epoch": 0.04963136126595936,
      "grad_norm": 21.563861795013054,
      "learning_rate": 2.999986069458716e-06,
      "loss": 1.4447,
      "step": 69
    },
    {
      "epoch": 0.050350656356770364,
      "grad_norm": 17.419159113718333,
      "learning_rate": 2.9999845644997736e-06,
      "loss": 1.5097,
      "step": 70
    },
    {
      "epoch": 0.05106995144758137,
      "grad_norm": 21.98894598263102,
      "learning_rate": 2.9999829823639916e-06,
      "loss": 1.4381,
      "step": 71
    },
    {
      "epoch": 0.051789246538392376,
      "grad_norm": 17.957091655713384,
      "learning_rate": 2.9999813230514524e-06,
      "loss": 1.6295,
      "step": 72
    },
    {
      "epoch": 0.05250854162920338,
      "grad_norm": 18.191843369307144,
      "learning_rate": 2.9999795865622413e-06,
      "loss": 1.394,
      "step": 73
    },
    {
      "epoch": 0.05322783672001439,
      "grad_norm": 15.974616072378568,
      "learning_rate": 2.9999777728964466e-06,
      "loss": 1.153,
      "step": 74
    },
    {
      "epoch": 0.05394713181082539,
      "grad_norm": 15.851738860415221,
      "learning_rate": 2.999975882054163e-06,
      "loss": 1.5358,
      "step": 75
    },
    {
      "epoch": 0.054666426901636395,
      "grad_norm": 18.13076452589248,
      "learning_rate": 2.999973914035487e-06,
      "loss": 1.031,
      "step": 76
    },
    {
      "epoch": 0.0553857219924474,
      "grad_norm": 20.323852306393164,
      "learning_rate": 2.9999718688405194e-06,
      "loss": 1.2072,
      "step": 77
    },
    {
      "epoch": 0.05610501708325841,
      "grad_norm": 16.390106814037097,
      "learning_rate": 2.9999697464693666e-06,
      "loss": 1.1953,
      "step": 78
    },
    {
      "epoch": 0.056824312174069413,
      "grad_norm": 13.93589642424244,
      "learning_rate": 2.9999675469221375e-06,
      "loss": 1.082,
      "step": 79
    },
    {
      "epoch": 0.05754360726488042,
      "grad_norm": 12.913261699312807,
      "learning_rate": 2.9999652701989444e-06,
      "loss": 1.0928,
      "step": 80
    },
    {
      "epoch": 0.05826290235569142,
      "grad_norm": 13.776128845237146,
      "learning_rate": 2.9999629162999054e-06,
      "loss": 1.3075,
      "step": 81
    },
    {
      "epoch": 0.058982197446502425,
      "grad_norm": 15.81637358851,
      "learning_rate": 2.999960485225141e-06,
      "loss": 1.2325,
      "step": 82
    },
    {
      "epoch": 0.05970149253731343,
      "grad_norm": 20.356411498268486,
      "learning_rate": 2.999957976974777e-06,
      "loss": 0.9464,
      "step": 83
    },
    {
      "epoch": 0.06042078762812444,
      "grad_norm": 18.885207613766585,
      "learning_rate": 2.999955391548942e-06,
      "loss": 1.1445,
      "step": 84
    },
    {
      "epoch": 0.061140082718935444,
      "grad_norm": 17.14246664289988,
      "learning_rate": 2.999952728947768e-06,
      "loss": 1.3846,
      "step": 85
    },
    {
      "epoch": 0.06185937780974645,
      "grad_norm": 17.689753014018272,
      "learning_rate": 2.999949989171394e-06,
      "loss": 1.1818,
      "step": 86
    },
    {
      "epoch": 0.06257867290055745,
      "grad_norm": 19.517032972882443,
      "learning_rate": 2.9999471722199595e-06,
      "loss": 1.0695,
      "step": 87
    },
    {
      "epoch": 0.06329796799136846,
      "grad_norm": 17.38660383027941,
      "learning_rate": 2.9999442780936104e-06,
      "loss": 1.1583,
      "step": 88
    },
    {
      "epoch": 0.06401726308217946,
      "grad_norm": 15.48079391320298,
      "learning_rate": 2.999941306792495e-06,
      "loss": 1.3267,
      "step": 89
    },
    {
      "epoch": 0.06473655817299047,
      "grad_norm": 15.360783627227772,
      "learning_rate": 2.999938258316766e-06,
      "loss": 1.0709,
      "step": 90
    },
    {
      "epoch": 0.06545585326380148,
      "grad_norm": 17.381248590193184,
      "learning_rate": 2.999935132666581e-06,
      "loss": 1.2396,
      "step": 91
    },
    {
      "epoch": 0.06617514835461248,
      "grad_norm": 15.682577761416166,
      "learning_rate": 2.9999319298420998e-06,
      "loss": 1.1373,
      "step": 92
    },
    {
      "epoch": 0.06689444344542349,
      "grad_norm": 15.080397284117504,
      "learning_rate": 2.999928649843488e-06,
      "loss": 0.9047,
      "step": 93
    },
    {
      "epoch": 0.0676137385362345,
      "grad_norm": 14.62098720233514,
      "learning_rate": 2.999925292670914e-06,
      "loss": 1.1609,
      "step": 94
    },
    {
      "epoch": 0.0683330336270455,
      "grad_norm": 13.450215274233114,
      "learning_rate": 2.999921858324551e-06,
      "loss": 0.9095,
      "step": 95
    },
    {
      "epoch": 0.0690523287178565,
      "grad_norm": 15.936356086252578,
      "learning_rate": 2.999918346804575e-06,
      "loss": 1.0574,
      "step": 96
    },
    {
      "epoch": 0.0697716238086675,
      "grad_norm": 11.361017434660084,
      "learning_rate": 2.9999147581111674e-06,
      "loss": 0.9188,
      "step": 97
    },
    {
      "epoch": 0.0704909188994785,
      "grad_norm": 16.136922711242608,
      "learning_rate": 2.9999110922445124e-06,
      "loss": 0.8518,
      "step": 98
    },
    {
      "epoch": 0.07121021399028951,
      "grad_norm": 14.479550155285834,
      "learning_rate": 2.9999073492047983e-06,
      "loss": 0.9733,
      "step": 99
    },
    {
      "epoch": 0.07192950908110052,
      "grad_norm": 14.14720055173962,
      "learning_rate": 2.9999035289922186e-06,
      "loss": 0.9136,
      "step": 100
    },
    {
      "epoch": 0.07264880417191152,
      "grad_norm": 14.460728505543798,
      "learning_rate": 2.999899631606969e-06,
      "loss": 0.7732,
      "step": 101
    },
    {
      "epoch": 0.07336809926272253,
      "grad_norm": 14.915994889708552,
      "learning_rate": 2.9998956570492504e-06,
      "loss": 1.0222,
      "step": 102
    },
    {
      "epoch": 0.07408739435353354,
      "grad_norm": 13.798774107026125,
      "learning_rate": 2.9998916053192673e-06,
      "loss": 0.9379,
      "step": 103
    },
    {
      "epoch": 0.07480668944434454,
      "grad_norm": 16.45470711743046,
      "learning_rate": 2.999887476417228e-06,
      "loss": 1.1023,
      "step": 104
    },
    {
      "epoch": 0.07552598453515555,
      "grad_norm": 16.82514571020224,
      "learning_rate": 2.9998832703433456e-06,
      "loss": 0.9055,
      "step": 105
    },
    {
      "epoch": 0.07624527962596656,
      "grad_norm": 18.355401485360265,
      "learning_rate": 2.9998789870978352e-06,
      "loss": 1.0206,
      "step": 106
    },
    {
      "epoch": 0.07696457471677756,
      "grad_norm": 17.01847252982604,
      "learning_rate": 2.9998746266809183e-06,
      "loss": 0.9563,
      "step": 107
    },
    {
      "epoch": 0.07768386980758857,
      "grad_norm": 13.709892094376332,
      "learning_rate": 2.9998701890928187e-06,
      "loss": 0.6598,
      "step": 108
    },
    {
      "epoch": 0.07840316489839957,
      "grad_norm": 17.46028171479583,
      "learning_rate": 2.9998656743337654e-06,
      "loss": 0.894,
      "step": 109
    },
    {
      "epoch": 0.07912245998921057,
      "grad_norm": 13.80409609539761,
      "learning_rate": 2.99986108240399e-06,
      "loss": 0.7213,
      "step": 110
    },
    {
      "epoch": 0.07984175508002157,
      "grad_norm": 17.175494336691322,
      "learning_rate": 2.999856413303729e-06,
      "loss": 0.9785,
      "step": 111
    },
    {
      "epoch": 0.08056105017083258,
      "grad_norm": 18.60010165596918,
      "learning_rate": 2.9998516670332227e-06,
      "loss": 1.0314,
      "step": 112
    },
    {
      "epoch": 0.08128034526164359,
      "grad_norm": 14.885433927059273,
      "learning_rate": 2.9998468435927154e-06,
      "loss": 0.8089,
      "step": 113
    },
    {
      "epoch": 0.08199964035245459,
      "grad_norm": 16.42747769780196,
      "learning_rate": 2.999841942982455e-06,
      "loss": 0.9663,
      "step": 114
    },
    {
      "epoch": 0.0827189354432656,
      "grad_norm": 15.279786015115542,
      "learning_rate": 2.999836965202693e-06,
      "loss": 0.7738,
      "step": 115
    },
    {
      "epoch": 0.0834382305340766,
      "grad_norm": 14.116665198902021,
      "learning_rate": 2.9998319102536868e-06,
      "loss": 0.9416,
      "step": 116
    },
    {
      "epoch": 0.08415752562488761,
      "grad_norm": 13.104497420126021,
      "learning_rate": 2.9998267781356964e-06,
      "loss": 0.6117,
      "step": 117
    },
    {
      "epoch": 0.08487682071569862,
      "grad_norm": 14.089846442513345,
      "learning_rate": 2.999821568848985e-06,
      "loss": 0.9485,
      "step": 118
    },
    {
      "epoch": 0.08559611580650962,
      "grad_norm": 17.090806029715722,
      "learning_rate": 2.999816282393821e-06,
      "loss": 0.9913,
      "step": 119
    },
    {
      "epoch": 0.08631541089732063,
      "grad_norm": 17.008896450934316,
      "learning_rate": 2.9998109187704765e-06,
      "loss": 0.9245,
      "step": 120
    },
    {
      "epoch": 0.08703470598813164,
      "grad_norm": 17.121880191450334,
      "learning_rate": 2.999805477979227e-06,
      "loss": 0.7626,
      "step": 121
    },
    {
      "epoch": 0.08775400107894264,
      "grad_norm": 14.606147063625393,
      "learning_rate": 2.9997999600203533e-06,
      "loss": 0.8401,
      "step": 122
    },
    {
      "epoch": 0.08847329616975363,
      "grad_norm": 17.397779605226102,
      "learning_rate": 2.999794364894139e-06,
      "loss": 0.7384,
      "step": 123
    },
    {
      "epoch": 0.08919259126056464,
      "grad_norm": 16.957065813386407,
      "learning_rate": 2.9997886926008713e-06,
      "loss": 0.8409,
      "step": 124
    },
    {
      "epoch": 0.08991188635137565,
      "grad_norm": 17.929248752097,
      "learning_rate": 2.999782943140843e-06,
      "loss": 0.8461,
      "step": 125
    },
    {
      "epoch": 0.09063118144218665,
      "grad_norm": 12.05064245598457,
      "learning_rate": 2.999777116514349e-06,
      "loss": 0.7481,
      "step": 126
    },
    {
      "epoch": 0.09135047653299766,
      "grad_norm": 15.70261153256519,
      "learning_rate": 2.99977121272169e-06,
      "loss": 0.8238,
      "step": 127
    },
    {
      "epoch": 0.09206977162380867,
      "grad_norm": 17.78830125589999,
      "learning_rate": 2.9997652317631693e-06,
      "loss": 0.7911,
      "step": 128
    },
    {
      "epoch": 0.09278906671461967,
      "grad_norm": 13.864250229628292,
      "learning_rate": 2.9997591736390945e-06,
      "loss": 0.7995,
      "step": 129
    },
    {
      "epoch": 0.09350836180543068,
      "grad_norm": 15.502336013451172,
      "learning_rate": 2.9997530383497773e-06,
      "loss": 0.8459,
      "step": 130
    },
    {
      "epoch": 0.09422765689624168,
      "grad_norm": 15.540040407710661,
      "learning_rate": 2.999746825895534e-06,
      "loss": 0.5619,
      "step": 131
    },
    {
      "epoch": 0.09494695198705269,
      "grad_norm": 14.749204467275762,
      "learning_rate": 2.9997405362766832e-06,
      "loss": 0.5666,
      "step": 132
    },
    {
      "epoch": 0.0956662470778637,
      "grad_norm": 12.622048470319367,
      "learning_rate": 2.99973416949355e-06,
      "loss": 0.8049,
      "step": 133
    },
    {
      "epoch": 0.0963855421686747,
      "grad_norm": 14.279358326559988,
      "learning_rate": 2.9997277255464603e-06,
      "loss": 0.69,
      "step": 134
    },
    {
      "epoch": 0.09710483725948571,
      "grad_norm": 14.37701622582175,
      "learning_rate": 2.999721204435747e-06,
      "loss": 0.7442,
      "step": 135
    },
    {
      "epoch": 0.0978241323502967,
      "grad_norm": 14.970198887449568,
      "learning_rate": 2.999714606161745e-06,
      "loss": 0.7065,
      "step": 136
    },
    {
      "epoch": 0.09854342744110771,
      "grad_norm": 11.087967774569819,
      "learning_rate": 2.9997079307247934e-06,
      "loss": 0.4393,
      "step": 137
    },
    {
      "epoch": 0.09926272253191872,
      "grad_norm": 14.08146398456673,
      "learning_rate": 2.999701178125236e-06,
      "loss": 0.6798,
      "step": 138
    },
    {
      "epoch": 0.09998201762272972,
      "grad_norm": 11.636590991525697,
      "learning_rate": 2.999694348363421e-06,
      "loss": 0.5427,
      "step": 139
    },
    {
      "epoch": 0.10070131271354073,
      "grad_norm": 11.641045285267525,
      "learning_rate": 2.9996874414396985e-06,
      "loss": 0.4303,
      "step": 140
    },
    {
      "epoch": 0.10142060780435173,
      "grad_norm": 15.17560296873663,
      "learning_rate": 2.999680457354425e-06,
      "loss": 0.648,
      "step": 141
    },
    {
      "epoch": 0.10213990289516274,
      "grad_norm": 14.812459173916162,
      "learning_rate": 2.999673396107959e-06,
      "loss": 0.7599,
      "step": 142
    },
    {
      "epoch": 0.10285919798597375,
      "grad_norm": 12.417508363153125,
      "learning_rate": 2.9996662577006645e-06,
      "loss": 0.5857,
      "step": 143
    },
    {
      "epoch": 0.10357849307678475,
      "grad_norm": 13.864225195088665,
      "learning_rate": 2.999659042132908e-06,
      "loss": 0.9716,
      "step": 144
    },
    {
      "epoch": 0.10429778816759576,
      "grad_norm": 12.987253142548148,
      "learning_rate": 2.9996517494050618e-06,
      "loss": 0.6616,
      "step": 145
    },
    {
      "epoch": 0.10501708325840677,
      "grad_norm": 11.110045110963247,
      "learning_rate": 2.9996443795175005e-06,
      "loss": 0.6188,
      "step": 146
    },
    {
      "epoch": 0.10573637834921777,
      "grad_norm": 16.827417071411812,
      "learning_rate": 2.999636932470603e-06,
      "loss": 0.5019,
      "step": 147
    },
    {
      "epoch": 0.10645567344002878,
      "grad_norm": 14.899423395901154,
      "learning_rate": 2.999629408264753e-06,
      "loss": 0.7334,
      "step": 148
    },
    {
      "epoch": 0.10717496853083977,
      "grad_norm": 15.329389828315854,
      "learning_rate": 2.999621806900337e-06,
      "loss": 0.7033,
      "step": 149
    },
    {
      "epoch": 0.10789426362165078,
      "grad_norm": 15.779200047974253,
      "learning_rate": 2.999614128377747e-06,
      "loss": 0.8697,
      "step": 150
    },
    {
      "epoch": 0.10861355871246178,
      "grad_norm": 18.170095046078853,
      "learning_rate": 2.9996063726973773e-06,
      "loss": 0.767,
      "step": 151
    },
    {
      "epoch": 0.10933285380327279,
      "grad_norm": 12.716753914369793,
      "learning_rate": 2.9995985398596274e-06,
      "loss": 0.5188,
      "step": 152
    },
    {
      "epoch": 0.1100521488940838,
      "grad_norm": 18.130394895813886,
      "learning_rate": 2.9995906298649e-06,
      "loss": 0.6062,
      "step": 153
    },
    {
      "epoch": 0.1107714439848948,
      "grad_norm": 12.356114027945125,
      "learning_rate": 2.9995826427136028e-06,
      "loss": 0.5099,
      "step": 154
    },
    {
      "epoch": 0.11149073907570581,
      "grad_norm": 16.007003122424482,
      "learning_rate": 2.999574578406146e-06,
      "loss": 0.5008,
      "step": 155
    },
    {
      "epoch": 0.11221003416651681,
      "grad_norm": 15.156259796857944,
      "learning_rate": 2.9995664369429444e-06,
      "loss": 0.5808,
      "step": 156
    },
    {
      "epoch": 0.11292932925732782,
      "grad_norm": 13.664453171228516,
      "learning_rate": 2.9995582183244173e-06,
      "loss": 0.7076,
      "step": 157
    },
    {
      "epoch": 0.11364862434813883,
      "grad_norm": 17.933972247998355,
      "learning_rate": 2.9995499225509876e-06,
      "loss": 0.8485,
      "step": 158
    },
    {
      "epoch": 0.11436791943894983,
      "grad_norm": 13.377851473531214,
      "learning_rate": 2.9995415496230818e-06,
      "loss": 0.5764,
      "step": 159
    },
    {
      "epoch": 0.11508721452976084,
      "grad_norm": 16.776637613444294,
      "learning_rate": 2.999533099541131e-06,
      "loss": 0.5597,
      "step": 160
    },
    {
      "epoch": 0.11580650962057185,
      "grad_norm": 10.738040442707506,
      "learning_rate": 2.99952457230557e-06,
      "loss": 0.4552,
      "step": 161
    },
    {
      "epoch": 0.11652580471138284,
      "grad_norm": 15.306648055364423,
      "learning_rate": 2.9995159679168376e-06,
      "loss": 0.656,
      "step": 162
    },
    {
      "epoch": 0.11724509980219384,
      "grad_norm": 12.17035497932848,
      "learning_rate": 2.999507286375376e-06,
      "loss": 0.5925,
      "step": 163
    },
    {
      "epoch": 0.11796439489300485,
      "grad_norm": 13.543791536121919,
      "learning_rate": 2.999498527681632e-06,
      "loss": 0.4981,
      "step": 164
    },
    {
      "epoch": 0.11868368998381586,
      "grad_norm": 13.118076682057563,
      "learning_rate": 2.999489691836057e-06,
      "loss": 0.4965,
      "step": 165
    },
    {
      "epoch": 0.11940298507462686,
      "grad_norm": 12.71381775320525,
      "learning_rate": 2.9994807788391045e-06,
      "loss": 0.471,
      "step": 166
    },
    {
      "epoch": 0.12012228016543787,
      "grad_norm": 12.399044755111824,
      "learning_rate": 2.9994717886912343e-06,
      "loss": 0.5591,
      "step": 167
    },
    {
      "epoch": 0.12084157525624888,
      "grad_norm": 15.706549334877247,
      "learning_rate": 2.999462721392908e-06,
      "loss": 0.5651,
      "step": 168
    },
    {
      "epoch": 0.12156087034705988,
      "grad_norm": 16.52354346421275,
      "learning_rate": 2.9994535769445925e-06,
      "loss": 0.8633,
      "step": 169
    },
    {
      "epoch": 0.12228016543787089,
      "grad_norm": 18.27461329136108,
      "learning_rate": 2.9994443553467584e-06,
      "loss": 0.7281,
      "step": 170
    },
    {
      "epoch": 0.1229994605286819,
      "grad_norm": 12.463490196038022,
      "learning_rate": 2.99943505659988e-06,
      "loss": 0.4946,
      "step": 171
    },
    {
      "epoch": 0.1237187556194929,
      "grad_norm": 17.214653610695475,
      "learning_rate": 2.9994256807044354e-06,
      "loss": 0.7973,
      "step": 172
    },
    {
      "epoch": 0.12443805071030391,
      "grad_norm": 14.550749335726373,
      "learning_rate": 2.9994162276609078e-06,
      "loss": 0.6072,
      "step": 173
    },
    {
      "epoch": 0.1251573458011149,
      "grad_norm": 11.27619069006964,
      "learning_rate": 2.9994066974697828e-06,
      "loss": 0.4613,
      "step": 174
    },
    {
      "epoch": 0.1258766408919259,
      "grad_norm": 16.14197403634099,
      "learning_rate": 2.9993970901315513e-06,
      "loss": 0.611,
      "step": 175
    },
    {
      "epoch": 0.1265959359827369,
      "grad_norm": 11.947668987471774,
      "learning_rate": 2.999387405646707e-06,
      "loss": 0.5964,
      "step": 176
    },
    {
      "epoch": 0.12731523107354792,
      "grad_norm": 11.917806216130602,
      "learning_rate": 2.999377644015749e-06,
      "loss": 0.5331,
      "step": 177
    },
    {
      "epoch": 0.12803452616435892,
      "grad_norm": 10.823123910733955,
      "learning_rate": 2.9993678052391797e-06,
      "loss": 0.6022,
      "step": 178
    },
    {
      "epoch": 0.12875382125516993,
      "grad_norm": 10.724683122415733,
      "learning_rate": 2.9993578893175036e-06,
      "loss": 0.3974,
      "step": 179
    },
    {
      "epoch": 0.12947311634598094,
      "grad_norm": 11.844584382631318,
      "learning_rate": 2.999347896251233e-06,
      "loss": 0.3949,
      "step": 180
    },
    {
      "epoch": 0.13019241143679194,
      "grad_norm": 11.831219301309252,
      "learning_rate": 2.9993378260408804e-06,
      "loss": 0.3115,
      "step": 181
    },
    {
      "epoch": 0.13091170652760295,
      "grad_norm": 11.061144669012819,
      "learning_rate": 2.9993276786869654e-06,
      "loss": 0.3338,
      "step": 182
    },
    {
      "epoch": 0.13163100161841396,
      "grad_norm": 14.56869049925627,
      "learning_rate": 2.9993174541900088e-06,
      "loss": 0.6595,
      "step": 183
    },
    {
      "epoch": 0.13235029670922496,
      "grad_norm": 11.654208837607412,
      "learning_rate": 2.9993071525505375e-06,
      "loss": 0.5349,
      "step": 184
    },
    {
      "epoch": 0.13306959180003597,
      "grad_norm": 15.254812193241476,
      "learning_rate": 2.9992967737690808e-06,
      "loss": 0.5788,
      "step": 185
    },
    {
      "epoch": 0.13378888689084698,
      "grad_norm": 14.96019408719912,
      "learning_rate": 2.9992863178461737e-06,
      "loss": 0.6025,
      "step": 186
    },
    {
      "epoch": 0.13450818198165798,
      "grad_norm": 15.053189519836618,
      "learning_rate": 2.999275784782353e-06,
      "loss": 0.5472,
      "step": 187
    },
    {
      "epoch": 0.135227477072469,
      "grad_norm": 20.259725486038164,
      "learning_rate": 2.9992651745781617e-06,
      "loss": 0.747,
      "step": 188
    },
    {
      "epoch": 0.13594677216328,
      "grad_norm": 11.486538136158382,
      "learning_rate": 2.9992544872341453e-06,
      "loss": 0.2998,
      "step": 189
    },
    {
      "epoch": 0.136666067254091,
      "grad_norm": 12.60224429554145,
      "learning_rate": 2.9992437227508536e-06,
      "loss": 0.5738,
      "step": 190
    },
    {
      "epoch": 0.137385362344902,
      "grad_norm": 12.630395204793585,
      "learning_rate": 2.9992328811288407e-06,
      "loss": 0.4701,
      "step": 191
    },
    {
      "epoch": 0.138104657435713,
      "grad_norm": 16.38951406253093,
      "learning_rate": 2.9992219623686637e-06,
      "loss": 0.6143,
      "step": 192
    },
    {
      "epoch": 0.13882395252652402,
      "grad_norm": 12.247087834394414,
      "learning_rate": 2.9992109664708857e-06,
      "loss": 0.5247,
      "step": 193
    },
    {
      "epoch": 0.139543247617335,
      "grad_norm": 13.261523423865471,
      "learning_rate": 2.999199893436071e-06,
      "loss": 0.4652,
      "step": 194
    },
    {
      "epoch": 0.140262542708146,
      "grad_norm": 16.34509283856271,
      "learning_rate": 2.9991887432647905e-06,
      "loss": 0.5315,
      "step": 195
    },
    {
      "epoch": 0.140981837798957,
      "grad_norm": 12.950869035317044,
      "learning_rate": 2.999177515957617e-06,
      "loss": 0.3374,
      "step": 196
    },
    {
      "epoch": 0.14170113288976802,
      "grad_norm": 12.263787315778378,
      "learning_rate": 2.999166211515129e-06,
      "loss": 0.4732,
      "step": 197
    },
    {
      "epoch": 0.14242042798057902,
      "grad_norm": 12.788931392269063,
      "learning_rate": 2.9991548299379073e-06,
      "loss": 0.5551,
      "step": 198
    },
    {
      "epoch": 0.14313972307139003,
      "grad_norm": 10.989185749610701,
      "learning_rate": 2.9991433712265382e-06,
      "loss": 0.423,
      "step": 199
    },
    {
      "epoch": 0.14385901816220104,
      "grad_norm": 14.02524180139987,
      "learning_rate": 2.999131835381611e-06,
      "loss": 0.5528,
      "step": 200
    },
    {
      "epoch": 0.14457831325301204,
      "grad_norm": 12.568161708376003,
      "learning_rate": 2.9991202224037193e-06,
      "loss": 0.5286,
      "step": 201
    },
    {
      "epoch": 0.14529760834382305,
      "grad_norm": 15.398108360402558,
      "learning_rate": 2.9991085322934607e-06,
      "loss": 0.4578,
      "step": 202
    },
    {
      "epoch": 0.14601690343463405,
      "grad_norm": 14.441341458041183,
      "learning_rate": 2.9990967650514363e-06,
      "loss": 0.5822,
      "step": 203
    },
    {
      "epoch": 0.14673619852544506,
      "grad_norm": 14.364747704730616,
      "learning_rate": 2.9990849206782516e-06,
      "loss": 0.4663,
      "step": 204
    },
    {
      "epoch": 0.14745549361625607,
      "grad_norm": 16.574475031532405,
      "learning_rate": 2.999072999174516e-06,
      "loss": 0.4073,
      "step": 205
    },
    {
      "epoch": 0.14817478870706707,
      "grad_norm": 16.82965096041825,
      "learning_rate": 2.9990610005408434e-06,
      "loss": 0.5181,
      "step": 206
    },
    {
      "epoch": 0.14889408379787808,
      "grad_norm": 8.874566029723,
      "learning_rate": 2.9990489247778507e-06,
      "loss": 0.2668,
      "step": 207
    },
    {
      "epoch": 0.14961337888868909,
      "grad_norm": 10.609333465643331,
      "learning_rate": 2.9990367718861596e-06,
      "loss": 0.4078,
      "step": 208
    },
    {
      "epoch": 0.1503326739795001,
      "grad_norm": 12.414073459052013,
      "learning_rate": 2.999024541866395e-06,
      "loss": 0.4306,
      "step": 209
    },
    {
      "epoch": 0.1510519690703111,
      "grad_norm": 12.841986715041784,
      "learning_rate": 2.9990122347191857e-06,
      "loss": 0.5474,
      "step": 210
    },
    {
      "epoch": 0.1517712641611221,
      "grad_norm": 15.865072441413076,
      "learning_rate": 2.998999850445166e-06,
      "loss": 0.6193,
      "step": 211
    },
    {
      "epoch": 0.1524905592519331,
      "grad_norm": 11.45279052450105,
      "learning_rate": 2.9989873890449723e-06,
      "loss": 0.4464,
      "step": 212
    },
    {
      "epoch": 0.15320985434274412,
      "grad_norm": 9.090892566687161,
      "learning_rate": 2.998974850519246e-06,
      "loss": 0.4324,
      "step": 213
    },
    {
      "epoch": 0.15392914943355512,
      "grad_norm": 9.674584282140811,
      "learning_rate": 2.9989622348686327e-06,
      "loss": 0.4122,
      "step": 214
    },
    {
      "epoch": 0.15464844452436613,
      "grad_norm": 6.684517331123412,
      "learning_rate": 2.998949542093781e-06,
      "loss": 0.2104,
      "step": 215
    },
    {
      "epoch": 0.15536773961517714,
      "grad_norm": 14.33919684371673,
      "learning_rate": 2.9989367721953438e-06,
      "loss": 0.2846,
      "step": 216
    },
    {
      "epoch": 0.15608703470598814,
      "grad_norm": 13.59688480883437,
      "learning_rate": 2.9989239251739785e-06,
      "loss": 0.5451,
      "step": 217
    },
    {
      "epoch": 0.15680632979679915,
      "grad_norm": 15.17634264162047,
      "learning_rate": 2.9989110010303464e-06,
      "loss": 0.7904,
      "step": 218
    },
    {
      "epoch": 0.15752562488761016,
      "grad_norm": 10.835490596827958,
      "learning_rate": 2.998897999765111e-06,
      "loss": 0.444,
      "step": 219
    },
    {
      "epoch": 0.15824491997842113,
      "grad_norm": 11.121993133593538,
      "learning_rate": 2.998884921378943e-06,
      "loss": 0.3226,
      "step": 220
    },
    {
      "epoch": 0.15896421506923214,
      "grad_norm": 12.4576180308362,
      "learning_rate": 2.9988717658725144e-06,
      "loss": 0.3672,
      "step": 221
    },
    {
      "epoch": 0.15968351016004315,
      "grad_norm": 11.500297093293284,
      "learning_rate": 2.9988585332465027e-06,
      "loss": 0.3059,
      "step": 222
    },
    {
      "epoch": 0.16040280525085415,
      "grad_norm": 11.692934284894786,
      "learning_rate": 2.9988452235015877e-06,
      "loss": 0.3584,
      "step": 223
    },
    {
      "epoch": 0.16112210034166516,
      "grad_norm": 10.92625880939289,
      "learning_rate": 2.998831836638455e-06,
      "loss": 0.1749,
      "step": 224
    },
    {
      "epoch": 0.16184139543247616,
      "grad_norm": 12.101401854465001,
      "learning_rate": 2.9988183726577932e-06,
      "loss": 0.2852,
      "step": 225
    },
    {
      "epoch": 0.16256069052328717,
      "grad_norm": 10.696809101144598,
      "learning_rate": 2.998804831560295e-06,
      "loss": 0.3363,
      "step": 226
    },
    {
      "epoch": 0.16327998561409818,
      "grad_norm": 13.259836928252481,
      "learning_rate": 2.9987912133466573e-06,
      "loss": 0.5374,
      "step": 227
    },
    {
      "epoch": 0.16399928070490918,
      "grad_norm": 15.045405140587834,
      "learning_rate": 2.9987775180175806e-06,
      "loss": 0.4377,
      "step": 228
    },
    {
      "epoch": 0.1647185757957202,
      "grad_norm": 5.533706659541429,
      "learning_rate": 2.998763745573769e-06,
      "loss": 0.096,
      "step": 229
    },
    {
      "epoch": 0.1654378708865312,
      "grad_norm": 7.837804068147167,
      "learning_rate": 2.9987498960159326e-06,
      "loss": 0.16,
      "step": 230
    },
    {
      "epoch": 0.1661571659773422,
      "grad_norm": 12.514268459194259,
      "learning_rate": 2.9987359693447828e-06,
      "loss": 0.4397,
      "step": 231
    },
    {
      "epoch": 0.1668764610681532,
      "grad_norm": 8.16973774034906,
      "learning_rate": 2.9987219655610362e-06,
      "loss": 0.2041,
      "step": 232
    },
    {
      "epoch": 0.16759575615896422,
      "grad_norm": 11.796620785675312,
      "learning_rate": 2.9987078846654134e-06,
      "loss": 0.5258,
      "step": 233
    },
    {
      "epoch": 0.16831505124977522,
      "grad_norm": 9.1366011301431,
      "learning_rate": 2.9986937266586394e-06,
      "loss": 0.4041,
      "step": 234
    },
    {
      "epoch": 0.16903434634058623,
      "grad_norm": 7.319660588016992,
      "learning_rate": 2.998679491541442e-06,
      "loss": 0.2205,
      "step": 235
    },
    {
      "epoch": 0.16975364143139723,
      "grad_norm": 9.663758952207552,
      "learning_rate": 2.9986651793145543e-06,
      "loss": 0.2728,
      "step": 236
    },
    {
      "epoch": 0.17047293652220824,
      "grad_norm": 9.432959955727869,
      "learning_rate": 2.998650789978712e-06,
      "loss": 0.2088,
      "step": 237
    },
    {
      "epoch": 0.17119223161301925,
      "grad_norm": 14.911613342507426,
      "learning_rate": 2.998636323534656e-06,
      "loss": 0.3816,
      "step": 238
    },
    {
      "epoch": 0.17191152670383025,
      "grad_norm": 9.91889765561175,
      "learning_rate": 2.99862177998313e-06,
      "loss": 0.485,
      "step": 239
    },
    {
      "epoch": 0.17263082179464126,
      "grad_norm": 7.970415448239636,
      "learning_rate": 2.998607159324883e-06,
      "loss": 0.1747,
      "step": 240
    },
    {
      "epoch": 0.17335011688545227,
      "grad_norm": 10.490791513990272,
      "learning_rate": 2.998592461560667e-06,
      "loss": 0.3607,
      "step": 241
    },
    {
      "epoch": 0.17406941197626327,
      "grad_norm": 12.008227473172994,
      "learning_rate": 2.998577686691238e-06,
      "loss": 0.2864,
      "step": 242
    },
    {
      "epoch": 0.17478870706707428,
      "grad_norm": 11.469251123315106,
      "learning_rate": 2.998562834717356e-06,
      "loss": 0.5207,
      "step": 243
    },
    {
      "epoch": 0.17550800215788528,
      "grad_norm": 8.869318644710406,
      "learning_rate": 2.9985479056397858e-06,
      "loss": 0.267,
      "step": 244
    },
    {
      "epoch": 0.1762272972486963,
      "grad_norm": 10.837545663504354,
      "learning_rate": 2.9985328994592958e-06,
      "loss": 0.3718,
      "step": 245
    },
    {
      "epoch": 0.17694659233950727,
      "grad_norm": 9.438194242279943,
      "learning_rate": 2.9985178161766568e-06,
      "loss": 0.2437,
      "step": 246
    },
    {
      "epoch": 0.17766588743031828,
      "grad_norm": 14.817412116589152,
      "learning_rate": 2.9985026557926456e-06,
      "loss": 0.5897,
      "step": 247
    },
    {
      "epoch": 0.17838518252112928,
      "grad_norm": 9.465656960834561,
      "learning_rate": 2.9984874183080427e-06,
      "loss": 0.2346,
      "step": 248
    },
    {
      "epoch": 0.1791044776119403,
      "grad_norm": 13.316057832497831,
      "learning_rate": 2.998472103723631e-06,
      "loss": 0.4892,
      "step": 249
    },
    {
      "epoch": 0.1798237727027513,
      "grad_norm": 13.335893338717915,
      "learning_rate": 2.9984567120401996e-06,
      "loss": 0.5034,
      "step": 250
    },
    {
      "epoch": 0.1805430677935623,
      "grad_norm": 12.60598126805529,
      "learning_rate": 2.99844124325854e-06,
      "loss": 0.4103,
      "step": 251
    },
    {
      "epoch": 0.1812623628843733,
      "grad_norm": 11.39934372716221,
      "learning_rate": 2.9984256973794474e-06,
      "loss": 0.4216,
      "step": 252
    },
    {
      "epoch": 0.1819816579751843,
      "grad_norm": 12.896340882610827,
      "learning_rate": 2.998410074403723e-06,
      "loss": 0.571,
      "step": 253
    },
    {
      "epoch": 0.18270095306599532,
      "grad_norm": 7.954801231901637,
      "learning_rate": 2.998394374332169e-06,
      "loss": 0.2671,
      "step": 254
    },
    {
      "epoch": 0.18342024815680633,
      "grad_norm": 13.453184658810134,
      "learning_rate": 2.9983785971655947e-06,
      "loss": 0.4554,
      "step": 255
    },
    {
      "epoch": 0.18413954324761733,
      "grad_norm": 8.15725989141266,
      "learning_rate": 2.9983627429048113e-06,
      "loss": 0.2668,
      "step": 256
    },
    {
      "epoch": 0.18485883833842834,
      "grad_norm": 12.551962320728792,
      "learning_rate": 2.998346811550634e-06,
      "loss": 0.2095,
      "step": 257
    },
    {
      "epoch": 0.18557813342923934,
      "grad_norm": 17.30908229697925,
      "learning_rate": 2.9983308031038837e-06,
      "loss": 0.5998,
      "step": 258
    },
    {
      "epoch": 0.18629742852005035,
      "grad_norm": 11.557334489143301,
      "learning_rate": 2.9983147175653827e-06,
      "loss": 0.3596,
      "step": 259
    },
    {
      "epoch": 0.18701672361086136,
      "grad_norm": 8.856220867626062,
      "learning_rate": 2.9982985549359598e-06,
      "loss": 0.2382,
      "step": 260
    },
    {
      "epoch": 0.18773601870167236,
      "grad_norm": 12.347054297084657,
      "learning_rate": 2.9982823152164458e-06,
      "loss": 0.3965,
      "step": 261
    },
    {
      "epoch": 0.18845531379248337,
      "grad_norm": 7.181096905871709,
      "learning_rate": 2.9982659984076764e-06,
      "loss": 0.1093,
      "step": 262
    },
    {
      "epoch": 0.18917460888329438,
      "grad_norm": 13.246743112939686,
      "learning_rate": 2.9982496045104914e-06,
      "loss": 0.5906,
      "step": 263
    },
    {
      "epoch": 0.18989390397410538,
      "grad_norm": 9.777738264141377,
      "learning_rate": 2.998233133525734e-06,
      "loss": 0.3573,
      "step": 264
    },
    {
      "epoch": 0.1906131990649164,
      "grad_norm": 8.247457435694521,
      "learning_rate": 2.998216585454252e-06,
      "loss": 0.2562,
      "step": 265
    },
    {
      "epoch": 0.1913324941557274,
      "grad_norm": 13.976344251379821,
      "learning_rate": 2.9981999602968967e-06,
      "loss": 0.3708,
      "step": 266
    },
    {
      "epoch": 0.1920517892465384,
      "grad_norm": 13.163142692336633,
      "learning_rate": 2.9981832580545235e-06,
      "loss": 0.4492,
      "step": 267
    },
    {
      "epoch": 0.1927710843373494,
      "grad_norm": 13.001953400210564,
      "learning_rate": 2.9981664787279915e-06,
      "loss": 0.4655,
      "step": 268
    },
    {
      "epoch": 0.19349037942816041,
      "grad_norm": 13.842506368966323,
      "learning_rate": 2.998149622318164e-06,
      "loss": 0.3648,
      "step": 269
    },
    {
      "epoch": 0.19420967451897142,
      "grad_norm": 10.906398363882099,
      "learning_rate": 2.998132688825909e-06,
      "loss": 0.3095,
      "step": 270
    },
    {
      "epoch": 0.19492896960978243,
      "grad_norm": 13.87216090717685,
      "learning_rate": 2.998115678252097e-06,
      "loss": 0.375,
      "step": 271
    },
    {
      "epoch": 0.1956482647005934,
      "grad_norm": 7.815467917214922,
      "learning_rate": 2.9980985905976037e-06,
      "loss": 0.2633,
      "step": 272
    },
    {
      "epoch": 0.1963675597914044,
      "grad_norm": 11.856422063191868,
      "learning_rate": 2.9980814258633074e-06,
      "loss": 0.2885,
      "step": 273
    },
    {
      "epoch": 0.19708685488221542,
      "grad_norm": 10.880015187107556,
      "learning_rate": 2.9980641840500925e-06,
      "loss": 0.4076,
      "step": 274
    },
    {
      "epoch": 0.19780614997302642,
      "grad_norm": 14.201705034672239,
      "learning_rate": 2.9980468651588458e-06,
      "loss": 0.5385,
      "step": 275
    },
    {
      "epoch": 0.19852544506383743,
      "grad_norm": 15.601881547790963,
      "learning_rate": 2.9980294691904575e-06,
      "loss": 0.388,
      "step": 276
    },
    {
      "epoch": 0.19924474015464844,
      "grad_norm": 15.390198877632194,
      "learning_rate": 2.9980119961458237e-06,
      "loss": 0.2607,
      "step": 277
    },
    {
      "epoch": 0.19996403524545944,
      "grad_norm": 14.371856391914728,
      "learning_rate": 2.997994446025843e-06,
      "loss": 0.4136,
      "step": 278
    },
    {
      "epoch": 0.20068333033627045,
      "grad_norm": 10.420784489540349,
      "learning_rate": 2.9979768188314182e-06,
      "loss": 0.2827,
      "step": 279
    },
    {
      "epoch": 0.20140262542708146,
      "grad_norm": 13.104961484102828,
      "learning_rate": 2.997959114563457e-06,
      "loss": 0.6195,
      "step": 280
    },
    {
      "epoch": 0.20212192051789246,
      "grad_norm": 10.38392710315252,
      "learning_rate": 2.9979413332228692e-06,
      "loss": 0.3119,
      "step": 281
    },
    {
      "epoch": 0.20284121560870347,
      "grad_norm": 6.14429188712158,
      "learning_rate": 2.9979234748105707e-06,
      "loss": 0.2586,
      "step": 282
    },
    {
      "epoch": 0.20356051069951447,
      "grad_norm": 6.487760743841203,
      "learning_rate": 2.99790553932748e-06,
      "loss": 0.182,
      "step": 283
    },
    {
      "epoch": 0.20427980579032548,
      "grad_norm": 11.743271681054466,
      "learning_rate": 2.9978875267745193e-06,
      "loss": 0.4323,
      "step": 284
    },
    {
      "epoch": 0.2049991008811365,
      "grad_norm": 9.956722592479277,
      "learning_rate": 2.997869437152616e-06,
      "loss": 0.161,
      "step": 285
    },
    {
      "epoch": 0.2057183959719475,
      "grad_norm": 10.740249919417174,
      "learning_rate": 2.997851270462701e-06,
      "loss": 0.2286,
      "step": 286
    },
    {
      "epoch": 0.2064376910627585,
      "grad_norm": 10.504140656380809,
      "learning_rate": 2.997833026705708e-06,
      "loss": 0.2957,
      "step": 287
    },
    {
      "epoch": 0.2071569861535695,
      "grad_norm": 8.5000954682574,
      "learning_rate": 2.997814705882577e-06,
      "loss": 0.1918,
      "step": 288
    },
    {
      "epoch": 0.2078762812443805,
      "grad_norm": 10.098724047024069,
      "learning_rate": 2.99779630799425e-06,
      "loss": 0.1764,
      "step": 289
    },
    {
      "epoch": 0.20859557633519152,
      "grad_norm": 10.857302085391463,
      "learning_rate": 2.997777833041674e-06,
      "loss": 0.2413,
      "step": 290
    },
    {
      "epoch": 0.20931487142600252,
      "grad_norm": 8.268792145880168,
      "learning_rate": 2.9977592810257984e-06,
      "loss": 0.1949,
      "step": 291
    },
    {
      "epoch": 0.21003416651681353,
      "grad_norm": 13.155447855701645,
      "learning_rate": 2.997740651947579e-06,
      "loss": 0.3198,
      "step": 292
    },
    {
      "epoch": 0.21075346160762454,
      "grad_norm": 10.049044718287215,
      "learning_rate": 2.997721945807974e-06,
      "loss": 0.3267,
      "step": 293
    },
    {
      "epoch": 0.21147275669843554,
      "grad_norm": 10.560958792742182,
      "learning_rate": 2.997703162607946e-06,
      "loss": 0.2936,
      "step": 294
    },
    {
      "epoch": 0.21219205178924655,
      "grad_norm": 9.690052481092012,
      "learning_rate": 2.9976843023484606e-06,
      "loss": 0.2943,
      "step": 295
    },
    {
      "epoch": 0.21291134688005756,
      "grad_norm": 11.5968414234468,
      "learning_rate": 2.997665365030489e-06,
      "loss": 0.4533,
      "step": 296
    },
    {
      "epoch": 0.21363064197086856,
      "grad_norm": 8.65941356334403,
      "learning_rate": 2.997646350655005e-06,
      "loss": 0.34,
      "step": 297
    },
    {
      "epoch": 0.21434993706167954,
      "grad_norm": 14.093680848748951,
      "learning_rate": 2.9976272592229876e-06,
      "loss": 0.3068,
      "step": 298
    },
    {
      "epoch": 0.21506923215249055,
      "grad_norm": 10.73945937673034,
      "learning_rate": 2.997608090735419e-06,
      "loss": 0.2158,
      "step": 299
    },
    {
      "epoch": 0.21578852724330155,
      "grad_norm": 10.319541571696844,
      "learning_rate": 2.997588845193284e-06,
      "loss": 0.529,
      "step": 300
    },
    {
      "epoch": 0.21650782233411256,
      "grad_norm": 8.049753202701346,
      "learning_rate": 2.997569522597575e-06,
      "loss": 0.2708,
      "step": 301
    },
    {
      "epoch": 0.21722711742492357,
      "grad_norm": 10.15704158478656,
      "learning_rate": 2.9975501229492847e-06,
      "loss": 0.3963,
      "step": 302
    },
    {
      "epoch": 0.21794641251573457,
      "grad_norm": 10.72051688835257,
      "learning_rate": 2.997530646249412e-06,
      "loss": 0.3836,
      "step": 303
    },
    {
      "epoch": 0.21866570760654558,
      "grad_norm": 10.25860994558,
      "learning_rate": 2.9975110924989584e-06,
      "loss": 0.2646,
      "step": 304
    },
    {
      "epoch": 0.21938500269735658,
      "grad_norm": 6.991686682325399,
      "learning_rate": 2.997491461698931e-06,
      "loss": 0.1934,
      "step": 305
    },
    {
      "epoch": 0.2201042977881676,
      "grad_norm": 11.001826145243419,
      "learning_rate": 2.9974717538503385e-06,
      "loss": 0.1715,
      "step": 306
    },
    {
      "epoch": 0.2208235928789786,
      "grad_norm": 12.894266580814731,
      "learning_rate": 2.9974519689541953e-06,
      "loss": 0.3524,
      "step": 307
    },
    {
      "epoch": 0.2215428879697896,
      "grad_norm": 10.243557471185888,
      "learning_rate": 2.9974321070115196e-06,
      "loss": 0.3797,
      "step": 308
    },
    {
      "epoch": 0.2222621830606006,
      "grad_norm": 11.622193991467736,
      "learning_rate": 2.997412168023334e-06,
      "loss": 0.2475,
      "step": 309
    },
    {
      "epoch": 0.22298147815141162,
      "grad_norm": 13.620762783141206,
      "learning_rate": 2.9973921519906634e-06,
      "loss": 0.4691,
      "step": 310
    },
    {
      "epoch": 0.22370077324222262,
      "grad_norm": 12.052922046994613,
      "learning_rate": 2.997372058914538e-06,
      "loss": 0.2727,
      "step": 311
    },
    {
      "epoch": 0.22442006833303363,
      "grad_norm": 16.174380113680808,
      "learning_rate": 2.997351888795992e-06,
      "loss": 0.6491,
      "step": 312
    },
    {
      "epoch": 0.22513936342384464,
      "grad_norm": 9.63241089464826,
      "learning_rate": 2.9973316416360624e-06,
      "loss": 0.2147,
      "step": 313
    },
    {
      "epoch": 0.22585865851465564,
      "grad_norm": 17.522132794774755,
      "learning_rate": 2.9973113174357908e-06,
      "loss": 0.3858,
      "step": 314
    },
    {
      "epoch": 0.22657795360546665,
      "grad_norm": 8.973491779012276,
      "learning_rate": 2.9972909161962247e-06,
      "loss": 0.1409,
      "step": 315
    },
    {
      "epoch": 0.22729724869627765,
      "grad_norm": 8.243014165895838,
      "learning_rate": 2.997270437918412e-06,
      "loss": 0.2116,
      "step": 316
    },
    {
      "epoch": 0.22801654378708866,
      "grad_norm": 7.407220332696786,
      "learning_rate": 2.997249882603406e-06,
      "loss": 0.1531,
      "step": 317
    },
    {
      "epoch": 0.22873583887789967,
      "grad_norm": 9.18431680608267,
      "learning_rate": 2.997229250252266e-06,
      "loss": 0.3942,
      "step": 318
    },
    {
      "epoch": 0.22945513396871067,
      "grad_norm": 10.355015508976939,
      "learning_rate": 2.997208540866053e-06,
      "loss": 0.2808,
      "step": 319
    },
    {
      "epoch": 0.23017442905952168,
      "grad_norm": 15.603005344421826,
      "learning_rate": 2.9971877544458325e-06,
      "loss": 0.4613,
      "step": 320
    },
    {
      "epoch": 0.23089372415033269,
      "grad_norm": 8.679251032096321,
      "learning_rate": 2.9971668909926733e-06,
      "loss": 0.322,
      "step": 321
    },
    {
      "epoch": 0.2316130192411437,
      "grad_norm": 8.052456370936467,
      "learning_rate": 2.9971459505076494e-06,
      "loss": 0.2745,
      "step": 322
    },
    {
      "epoch": 0.2323323143319547,
      "grad_norm": 8.42723801671621,
      "learning_rate": 2.997124932991838e-06,
      "loss": 0.1704,
      "step": 323
    },
    {
      "epoch": 0.23305160942276568,
      "grad_norm": 9.695100590141454,
      "learning_rate": 2.997103838446321e-06,
      "loss": 0.167,
      "step": 324
    },
    {
      "epoch": 0.23377090451357668,
      "grad_norm": 9.102937950822987,
      "learning_rate": 2.9970826668721836e-06,
      "loss": 0.28,
      "step": 325
    },
    {
      "epoch": 0.2344901996043877,
      "grad_norm": 8.471200978402717,
      "learning_rate": 2.9970614182705145e-06,
      "loss": 0.2106,
      "step": 326
    },
    {
      "epoch": 0.2352094946951987,
      "grad_norm": 11.863885987586409,
      "learning_rate": 2.9970400926424076e-06,
      "loss": 0.4534,
      "step": 327
    },
    {
      "epoch": 0.2359287897860097,
      "grad_norm": 10.935872905596742,
      "learning_rate": 2.9970186899889594e-06,
      "loss": 0.3792,
      "step": 328
    },
    {
      "epoch": 0.2366480848768207,
      "grad_norm": 11.52976338678996,
      "learning_rate": 2.9969972103112724e-06,
      "loss": 0.1784,
      "step": 329
    },
    {
      "epoch": 0.23736737996763171,
      "grad_norm": 9.062366806315843,
      "learning_rate": 2.996975653610451e-06,
      "loss": 0.2023,
      "step": 330
    },
    {
      "epoch": 0.23808667505844272,
      "grad_norm": 9.407895213960451,
      "learning_rate": 2.996954019887604e-06,
      "loss": 0.3943,
      "step": 331
    },
    {
      "epoch": 0.23880597014925373,
      "grad_norm": 15.337835908445518,
      "learning_rate": 2.9969323091438455e-06,
      "loss": 0.5976,
      "step": 332
    },
    {
      "epoch": 0.23952526524006473,
      "grad_norm": 12.85595473580567,
      "learning_rate": 2.9969105213802915e-06,
      "loss": 0.4178,
      "step": 333
    },
    {
      "epoch": 0.24024456033087574,
      "grad_norm": 9.00368307970452,
      "learning_rate": 2.9968886565980633e-06,
      "loss": 0.1701,
      "step": 334
    },
    {
      "epoch": 0.24096385542168675,
      "grad_norm": 12.983566330146337,
      "learning_rate": 2.9968667147982864e-06,
      "loss": 0.4775,
      "step": 335
    },
    {
      "epoch": 0.24168315051249775,
      "grad_norm": 12.490135154595928,
      "learning_rate": 2.996844695982089e-06,
      "loss": 0.391,
      "step": 336
    },
    {
      "epoch": 0.24240244560330876,
      "grad_norm": 10.662630389549339,
      "learning_rate": 2.9968226001506042e-06,
      "loss": 0.3546,
      "step": 337
    },
    {
      "epoch": 0.24312174069411976,
      "grad_norm": 8.438847678165793,
      "learning_rate": 2.9968004273049694e-06,
      "loss": 0.2592,
      "step": 338
    },
    {
      "epoch": 0.24384103578493077,
      "grad_norm": 9.506620260928214,
      "learning_rate": 2.996778177446325e-06,
      "loss": 0.2267,
      "step": 339
    },
    {
      "epoch": 0.24456033087574178,
      "grad_norm": 10.231121716210279,
      "learning_rate": 2.996755850575816e-06,
      "loss": 0.3227,
      "step": 340
    },
    {
      "epoch": 0.24527962596655278,
      "grad_norm": 7.227690713001611,
      "learning_rate": 2.9967334466945903e-06,
      "loss": 0.2186,
      "step": 341
    },
    {
      "epoch": 0.2459989210573638,
      "grad_norm": 9.57947279862929,
      "learning_rate": 2.996710965803802e-06,
      "loss": 0.2507,
      "step": 342
    },
    {
      "epoch": 0.2467182161481748,
      "grad_norm": 9.371660195527829,
      "learning_rate": 2.996688407904607e-06,
      "loss": 0.3166,
      "step": 343
    },
    {
      "epoch": 0.2474375112389858,
      "grad_norm": 13.460697955358041,
      "learning_rate": 2.9966657729981665e-06,
      "loss": 0.233,
      "step": 344
    },
    {
      "epoch": 0.2481568063297968,
      "grad_norm": 9.83124222393436,
      "learning_rate": 2.9966430610856436e-06,
      "loss": 0.2656,
      "step": 345
    },
    {
      "epoch": 0.24887610142060781,
      "grad_norm": 10.010909354391735,
      "learning_rate": 2.996620272168209e-06,
      "loss": 0.4054,
      "step": 346
    },
    {
      "epoch": 0.24959539651141882,
      "grad_norm": 4.824021065095197,
      "learning_rate": 2.9965974062470333e-06,
      "loss": 0.1997,
      "step": 347
    },
    {
      "epoch": 0.2503146916022298,
      "grad_norm": 8.005322435112516,
      "learning_rate": 2.9965744633232947e-06,
      "loss": 0.2646,
      "step": 348
    },
    {
      "epoch": 0.25103398669304083,
      "grad_norm": 7.343884721132704,
      "learning_rate": 2.996551443398172e-06,
      "loss": 0.1738,
      "step": 349
    },
    {
      "epoch": 0.2517532817838518,
      "grad_norm": 8.360177818344797,
      "learning_rate": 2.9965283464728508e-06,
      "loss": 0.2571,
      "step": 350
    },
    {
      "epoch": 0.25247257687466285,
      "grad_norm": 6.792580794942874,
      "learning_rate": 2.996505172548519e-06,
      "loss": 0.0952,
      "step": 351
    },
    {
      "epoch": 0.2531918719654738,
      "grad_norm": 10.696204710229608,
      "learning_rate": 2.9964819216263694e-06,
      "loss": 0.3471,
      "step": 352
    },
    {
      "epoch": 0.25391116705628486,
      "grad_norm": 9.32400869528146,
      "learning_rate": 2.996458593707598e-06,
      "loss": 0.1938,
      "step": 353
    },
    {
      "epoch": 0.25463046214709584,
      "grad_norm": 6.526513107235073,
      "learning_rate": 2.9964351887934043e-06,
      "loss": 0.1676,
      "step": 354
    },
    {
      "epoch": 0.25534975723790687,
      "grad_norm": 8.338472510715075,
      "learning_rate": 2.9964117068849937e-06,
      "loss": 0.1876,
      "step": 355
    },
    {
      "epoch": 0.25606905232871785,
      "grad_norm": 9.621677975595418,
      "learning_rate": 2.996388147983574e-06,
      "loss": 0.2056,
      "step": 356
    },
    {
      "epoch": 0.2567883474195289,
      "grad_norm": 12.484588912806705,
      "learning_rate": 2.996364512090357e-06,
      "loss": 0.2654,
      "step": 357
    },
    {
      "epoch": 0.25750764251033986,
      "grad_norm": 8.106593765931988,
      "learning_rate": 2.996340799206559e-06,
      "loss": 0.2597,
      "step": 358
    },
    {
      "epoch": 0.2582269376011509,
      "grad_norm": 14.282226452989242,
      "learning_rate": 2.9963170093334004e-06,
      "loss": 0.3816,
      "step": 359
    },
    {
      "epoch": 0.2589462326919619,
      "grad_norm": 8.368840076338213,
      "learning_rate": 2.996293142472105e-06,
      "loss": 0.1286,
      "step": 360
    },
    {
      "epoch": 0.2596655277827729,
      "grad_norm": 6.696843858149411,
      "learning_rate": 2.9962691986239e-06,
      "loss": 0.1924,
      "step": 361
    },
    {
      "epoch": 0.2603848228735839,
      "grad_norm": 7.962899969224421,
      "learning_rate": 2.9962451777900192e-06,
      "loss": 0.1765,
      "step": 362
    },
    {
      "epoch": 0.26110411796439487,
      "grad_norm": 17.73875889449904,
      "learning_rate": 2.996221079971697e-06,
      "loss": 0.6514,
      "step": 363
    },
    {
      "epoch": 0.2618234130552059,
      "grad_norm": 10.088957533953707,
      "learning_rate": 2.9961969051701735e-06,
      "loss": 0.1896,
      "step": 364
    },
    {
      "epoch": 0.2625427081460169,
      "grad_norm": 12.453133378487488,
      "learning_rate": 2.9961726533866926e-06,
      "loss": 0.3453,
      "step": 365
    },
    {
      "epoch": 0.2632620032368279,
      "grad_norm": 11.899756096310341,
      "learning_rate": 2.9961483246225027e-06,
      "loss": 0.1886,
      "step": 366
    },
    {
      "epoch": 0.2639812983276389,
      "grad_norm": 11.495582861425552,
      "learning_rate": 2.996123918878855e-06,
      "loss": 0.2784,
      "step": 367
    },
    {
      "epoch": 0.2647005934184499,
      "grad_norm": 12.54192687953523,
      "learning_rate": 2.9960994361570052e-06,
      "loss": 0.2389,
      "step": 368
    },
    {
      "epoch": 0.2654198885092609,
      "grad_norm": 15.193888755066343,
      "learning_rate": 2.9960748764582134e-06,
      "loss": 0.4148,
      "step": 369
    },
    {
      "epoch": 0.26613918360007194,
      "grad_norm": 9.503164320388725,
      "learning_rate": 2.9960502397837422e-06,
      "loss": 0.1989,
      "step": 370
    },
    {
      "epoch": 0.2668584786908829,
      "grad_norm": 8.269561546615632,
      "learning_rate": 2.9960255261348606e-06,
      "loss": 0.2405,
      "step": 371
    },
    {
      "epoch": 0.26757777378169395,
      "grad_norm": 12.020635644499855,
      "learning_rate": 2.996000735512839e-06,
      "loss": 0.2988,
      "step": 372
    },
    {
      "epoch": 0.26829706887250493,
      "grad_norm": 8.985638658238557,
      "learning_rate": 2.995975867918954e-06,
      "loss": 0.1619,
      "step": 373
    },
    {
      "epoch": 0.26901636396331596,
      "grad_norm": 7.935651727610059,
      "learning_rate": 2.9959509233544843e-06,
      "loss": 0.2772,
      "step": 374
    },
    {
      "epoch": 0.26973565905412694,
      "grad_norm": 13.423066123686887,
      "learning_rate": 2.9959259018207133e-06,
      "loss": 0.3276,
      "step": 375
    },
    {
      "epoch": 0.270454954144938,
      "grad_norm": 8.090818837580843,
      "learning_rate": 2.9959008033189287e-06,
      "loss": 0.0981,
      "step": 376
    },
    {
      "epoch": 0.27117424923574895,
      "grad_norm": 10.917968964794552,
      "learning_rate": 2.995875627850422e-06,
      "loss": 0.3252,
      "step": 377
    },
    {
      "epoch": 0.27189354432656,
      "grad_norm": 8.49562529169398,
      "learning_rate": 2.9958503754164883e-06,
      "loss": 0.2594,
      "step": 378
    },
    {
      "epoch": 0.27261283941737097,
      "grad_norm": 8.652543862175486,
      "learning_rate": 2.9958250460184265e-06,
      "loss": 0.271,
      "step": 379
    },
    {
      "epoch": 0.273332134508182,
      "grad_norm": 7.735114801134429,
      "learning_rate": 2.9957996396575405e-06,
      "loss": 0.2967,
      "step": 380
    },
    {
      "epoch": 0.274051429598993,
      "grad_norm": 7.461929682508416,
      "learning_rate": 2.995774156335137e-06,
      "loss": 0.1971,
      "step": 381
    },
    {
      "epoch": 0.274770724689804,
      "grad_norm": 8.373943213507687,
      "learning_rate": 2.995748596052528e-06,
      "loss": 0.2388,
      "step": 382
    },
    {
      "epoch": 0.275490019780615,
      "grad_norm": 7.692796650185384,
      "learning_rate": 2.9957229588110277e-06,
      "loss": 0.2689,
      "step": 383
    },
    {
      "epoch": 0.276209314871426,
      "grad_norm": 11.5353347069809,
      "learning_rate": 2.9956972446119556e-06,
      "loss": 0.1924,
      "step": 384
    },
    {
      "epoch": 0.276928609962237,
      "grad_norm": 9.236543467720582,
      "learning_rate": 2.9956714534566343e-06,
      "loss": 0.1729,
      "step": 385
    },
    {
      "epoch": 0.27764790505304804,
      "grad_norm": 9.711051511926513,
      "learning_rate": 2.9956455853463914e-06,
      "loss": 0.2198,
      "step": 386
    },
    {
      "epoch": 0.278367200143859,
      "grad_norm": 9.059058712001333,
      "learning_rate": 2.9956196402825573e-06,
      "loss": 0.3891,
      "step": 387
    },
    {
      "epoch": 0.27908649523467,
      "grad_norm": 6.277351548744565,
      "learning_rate": 2.9955936182664678e-06,
      "loss": 0.1852,
      "step": 388
    },
    {
      "epoch": 0.27980579032548103,
      "grad_norm": 7.242455927831628,
      "learning_rate": 2.9955675192994615e-06,
      "loss": 0.2527,
      "step": 389
    },
    {
      "epoch": 0.280525085416292,
      "grad_norm": 10.367919504041973,
      "learning_rate": 2.99554134338288e-06,
      "loss": 0.2319,
      "step": 390
    },
    {
      "epoch": 0.28124438050710304,
      "grad_norm": 8.790781759048409,
      "learning_rate": 2.995515090518072e-06,
      "loss": 0.2425,
      "step": 391
    },
    {
      "epoch": 0.281963675597914,
      "grad_norm": 7.180223846706552,
      "learning_rate": 2.995488760706386e-06,
      "loss": 0.228,
      "step": 392
    },
    {
      "epoch": 0.28268297068872505,
      "grad_norm": 6.964087388445663,
      "learning_rate": 2.9954623539491793e-06,
      "loss": 0.165,
      "step": 393
    },
    {
      "epoch": 0.28340226577953603,
      "grad_norm": 10.702327770754659,
      "learning_rate": 2.9954358702478084e-06,
      "loss": 0.2385,
      "step": 394
    },
    {
      "epoch": 0.28412156087034707,
      "grad_norm": 9.082516119263104,
      "learning_rate": 2.995409309603637e-06,
      "loss": 0.3171,
      "step": 395
    },
    {
      "epoch": 0.28484085596115805,
      "grad_norm": 7.707621376635846,
      "learning_rate": 2.995382672018032e-06,
      "loss": 0.1612,
      "step": 396
    },
    {
      "epoch": 0.2855601510519691,
      "grad_norm": 6.014506748872444,
      "learning_rate": 2.995355957492363e-06,
      "loss": 0.2554,
      "step": 397
    },
    {
      "epoch": 0.28627944614278006,
      "grad_norm": 7.564226651367394,
      "learning_rate": 2.9953291660280055e-06,
      "loss": 0.2188,
      "step": 398
    },
    {
      "epoch": 0.2869987412335911,
      "grad_norm": 9.981279147419581,
      "learning_rate": 2.995302297626337e-06,
      "loss": 0.2318,
      "step": 399
    },
    {
      "epoch": 0.28771803632440207,
      "grad_norm": 8.532486186989155,
      "learning_rate": 2.99527535228874e-06,
      "loss": 0.3529,
      "step": 400
    },
    {
      "epoch": 0.2884373314152131,
      "grad_norm": 7.82831385381344,
      "learning_rate": 2.995248330016602e-06,
      "loss": 0.2805,
      "step": 401
    },
    {
      "epoch": 0.2891566265060241,
      "grad_norm": 8.805743130071244,
      "learning_rate": 2.995221230811312e-06,
      "loss": 0.2915,
      "step": 402
    },
    {
      "epoch": 0.2898759215968351,
      "grad_norm": 9.094733511915015,
      "learning_rate": 2.9951940546742653e-06,
      "loss": 0.1482,
      "step": 403
    },
    {
      "epoch": 0.2905952166876461,
      "grad_norm": 8.207026219832166,
      "learning_rate": 2.9951668016068596e-06,
      "loss": 0.1724,
      "step": 404
    },
    {
      "epoch": 0.29131451177845713,
      "grad_norm": 6.867471145862766,
      "learning_rate": 2.9951394716104974e-06,
      "loss": 0.2305,
      "step": 405
    },
    {
      "epoch": 0.2920338068692681,
      "grad_norm": 9.204708353903655,
      "learning_rate": 2.9951120646865842e-06,
      "loss": 0.286,
      "step": 406
    },
    {
      "epoch": 0.29275310196007914,
      "grad_norm": 11.684321050750068,
      "learning_rate": 2.9950845808365317e-06,
      "loss": 0.2599,
      "step": 407
    },
    {
      "epoch": 0.2934723970508901,
      "grad_norm": 12.972548620267979,
      "learning_rate": 2.9950570200617524e-06,
      "loss": 0.2231,
      "step": 408
    },
    {
      "epoch": 0.29419169214170116,
      "grad_norm": 8.150918145053511,
      "learning_rate": 2.995029382363665e-06,
      "loss": 0.1222,
      "step": 409
    },
    {
      "epoch": 0.29491098723251213,
      "grad_norm": 13.31114508219458,
      "learning_rate": 2.995001667743691e-06,
      "loss": 0.2331,
      "step": 410
    },
    {
      "epoch": 0.29563028232332317,
      "grad_norm": 10.915018744772286,
      "learning_rate": 2.9949738762032576e-06,
      "loss": 0.2106,
      "step": 411
    },
    {
      "epoch": 0.29634957741413415,
      "grad_norm": 11.212739819920543,
      "learning_rate": 2.9949460077437932e-06,
      "loss": 0.1485,
      "step": 412
    },
    {
      "epoch": 0.2970688725049452,
      "grad_norm": 15.257880903781947,
      "learning_rate": 2.994918062366733e-06,
      "loss": 0.5924,
      "step": 413
    },
    {
      "epoch": 0.29778816759575616,
      "grad_norm": 8.772076814667747,
      "learning_rate": 2.9948900400735138e-06,
      "loss": 0.2593,
      "step": 414
    },
    {
      "epoch": 0.29850746268656714,
      "grad_norm": 9.238925942706313,
      "learning_rate": 2.994861940865578e-06,
      "loss": 0.2807,
      "step": 415
    },
    {
      "epoch": 0.29922675777737817,
      "grad_norm": 12.327731313364321,
      "learning_rate": 2.994833764744371e-06,
      "loss": 0.1232,
      "step": 416
    },
    {
      "epoch": 0.29994605286818915,
      "grad_norm": 9.820439632000484,
      "learning_rate": 2.994805511711343e-06,
      "loss": 0.3351,
      "step": 417
    },
    {
      "epoch": 0.3006653479590002,
      "grad_norm": 8.000892509445384,
      "learning_rate": 2.9947771817679474e-06,
      "loss": 0.266,
      "step": 418
    },
    {
      "epoch": 0.30138464304981116,
      "grad_norm": 10.352406847146282,
      "learning_rate": 2.9947487749156418e-06,
      "loss": 0.1777,
      "step": 419
    },
    {
      "epoch": 0.3021039381406222,
      "grad_norm": 7.480424892887567,
      "learning_rate": 2.9947202911558874e-06,
      "loss": 0.2517,
      "step": 420
    },
    {
      "epoch": 0.3028232332314332,
      "grad_norm": 6.623389444191224,
      "learning_rate": 2.9946917304901506e-06,
      "loss": 0.2247,
      "step": 421
    },
    {
      "epoch": 0.3035425283222442,
      "grad_norm": 7.417147665562637,
      "learning_rate": 2.9946630929199e-06,
      "loss": 0.2303,
      "step": 422
    },
    {
      "epoch": 0.3042618234130552,
      "grad_norm": 8.923072775228164,
      "learning_rate": 2.9946343784466096e-06,
      "loss": 0.226,
      "step": 423
    },
    {
      "epoch": 0.3049811185038662,
      "grad_norm": 10.476843191096348,
      "learning_rate": 2.9946055870717565e-06,
      "loss": 0.4008,
      "step": 424
    },
    {
      "epoch": 0.3057004135946772,
      "grad_norm": 8.703951018313962,
      "learning_rate": 2.9945767187968227e-06,
      "loss": 0.2258,
      "step": 425
    },
    {
      "epoch": 0.30641970868548823,
      "grad_norm": 10.476161352254241,
      "learning_rate": 2.994547773623293e-06,
      "loss": 0.4341,
      "step": 426
    },
    {
      "epoch": 0.3071390037762992,
      "grad_norm": 8.696807780643637,
      "learning_rate": 2.9945187515526566e-06,
      "loss": 0.287,
      "step": 427
    },
    {
      "epoch": 0.30785829886711025,
      "grad_norm": 10.68503471179048,
      "learning_rate": 2.994489652586407e-06,
      "loss": 0.3893,
      "step": 428
    },
    {
      "epoch": 0.3085775939579212,
      "grad_norm": 11.173799729660958,
      "learning_rate": 2.994460476726041e-06,
      "loss": 0.2456,
      "step": 429
    },
    {
      "epoch": 0.30929688904873226,
      "grad_norm": 7.304502575495571,
      "learning_rate": 2.9944312239730604e-06,
      "loss": 0.258,
      "step": 430
    },
    {
      "epoch": 0.31001618413954324,
      "grad_norm": 9.2050161333308,
      "learning_rate": 2.9944018943289698e-06,
      "loss": 0.1753,
      "step": 431
    },
    {
      "epoch": 0.3107354792303543,
      "grad_norm": 7.25244315285998,
      "learning_rate": 2.9943724877952783e-06,
      "loss": 0.1689,
      "step": 432
    },
    {
      "epoch": 0.31145477432116525,
      "grad_norm": 8.57014411753549,
      "learning_rate": 2.9943430043734988e-06,
      "loss": 0.1171,
      "step": 433
    },
    {
      "epoch": 0.3121740694119763,
      "grad_norm": 5.12479042738401,
      "learning_rate": 2.994313444065149e-06,
      "loss": 0.0794,
      "step": 434
    },
    {
      "epoch": 0.31289336450278726,
      "grad_norm": 8.167122669177642,
      "learning_rate": 2.994283806871749e-06,
      "loss": 0.1969,
      "step": 435
    },
    {
      "epoch": 0.3136126595935983,
      "grad_norm": 5.889526847070679,
      "learning_rate": 2.994254092794824e-06,
      "loss": 0.1907,
      "step": 436
    },
    {
      "epoch": 0.3143319546844093,
      "grad_norm": 7.803611222302982,
      "learning_rate": 2.994224301835903e-06,
      "loss": 0.2208,
      "step": 437
    },
    {
      "epoch": 0.3150512497752203,
      "grad_norm": 17.112049783414655,
      "learning_rate": 2.9941944339965187e-06,
      "loss": 0.1361,
      "step": 438
    },
    {
      "epoch": 0.3157705448660313,
      "grad_norm": 12.968078824689863,
      "learning_rate": 2.9941644892782074e-06,
      "loss": 0.3567,
      "step": 439
    },
    {
      "epoch": 0.31648983995684227,
      "grad_norm": 6.543873111288228,
      "learning_rate": 2.9941344676825104e-06,
      "loss": 0.0938,
      "step": 440
    },
    {
      "epoch": 0.3172091350476533,
      "grad_norm": 11.561968819083491,
      "learning_rate": 2.9941043692109725e-06,
      "loss": 0.5349,
      "step": 441
    },
    {
      "epoch": 0.3179284301384643,
      "grad_norm": 7.899621134761218,
      "learning_rate": 2.994074193865142e-06,
      "loss": 0.121,
      "step": 442
    },
    {
      "epoch": 0.3186477252292753,
      "grad_norm": 12.740350221282768,
      "learning_rate": 2.994043941646571e-06,
      "loss": 0.2477,
      "step": 443
    },
    {
      "epoch": 0.3193670203200863,
      "grad_norm": 5.348027947374008,
      "learning_rate": 2.9940136125568165e-06,
      "loss": 0.0354,
      "step": 444
    },
    {
      "epoch": 0.3200863154108973,
      "grad_norm": 11.986539406760485,
      "learning_rate": 2.993983206597439e-06,
      "loss": 0.5919,
      "step": 445
    },
    {
      "epoch": 0.3208056105017083,
      "grad_norm": 8.767824151738221,
      "learning_rate": 2.993952723770003e-06,
      "loss": 0.1385,
      "step": 446
    },
    {
      "epoch": 0.32152490559251934,
      "grad_norm": 11.490044874098949,
      "learning_rate": 2.9939221640760773e-06,
      "loss": 0.2664,
      "step": 447
    },
    {
      "epoch": 0.3222442006833303,
      "grad_norm": 6.473208496720477,
      "learning_rate": 2.993891527517233e-06,
      "loss": 0.1232,
      "step": 448
    },
    {
      "epoch": 0.32296349577414135,
      "grad_norm": 8.402492007996052,
      "learning_rate": 2.9938608140950475e-06,
      "loss": 0.3161,
      "step": 449
    },
    {
      "epoch": 0.32368279086495233,
      "grad_norm": 10.887662933472471,
      "learning_rate": 2.9938300238111007e-06,
      "loss": 0.374,
      "step": 450
    },
    {
      "epoch": 0.32440208595576336,
      "grad_norm": 7.531560746533734,
      "learning_rate": 2.9937991566669765e-06,
      "loss": 0.1566,
      "step": 451
    },
    {
      "epoch": 0.32512138104657434,
      "grad_norm": 8.045718514802468,
      "learning_rate": 2.9937682126642643e-06,
      "loss": 0.1335,
      "step": 452
    },
    {
      "epoch": 0.3258406761373854,
      "grad_norm": 7.20557052846816,
      "learning_rate": 2.993737191804555e-06,
      "loss": 0.2782,
      "step": 453
    },
    {
      "epoch": 0.32655997122819636,
      "grad_norm": 7.3314078440514105,
      "learning_rate": 2.9937060940894447e-06,
      "loss": 0.0879,
      "step": 454
    },
    {
      "epoch": 0.3272792663190074,
      "grad_norm": 12.314161812184437,
      "learning_rate": 2.9936749195205335e-06,
      "loss": 0.2937,
      "step": 455
    },
    {
      "epoch": 0.32799856140981837,
      "grad_norm": 10.013293912777787,
      "learning_rate": 2.993643668099426e-06,
      "loss": 0.2753,
      "step": 456
    },
    {
      "epoch": 0.3287178565006294,
      "grad_norm": 9.215564999455935,
      "learning_rate": 2.99361233982773e-06,
      "loss": 0.1477,
      "step": 457
    },
    {
      "epoch": 0.3294371515914404,
      "grad_norm": 11.464109463075435,
      "learning_rate": 2.9935809347070566e-06,
      "loss": 0.373,
      "step": 458
    },
    {
      "epoch": 0.3301564466822514,
      "grad_norm": 7.98721130374862,
      "learning_rate": 2.9935494527390227e-06,
      "loss": 0.0831,
      "step": 459
    },
    {
      "epoch": 0.3308757417730624,
      "grad_norm": 8.461868928839882,
      "learning_rate": 2.9935178939252478e-06,
      "loss": 0.0513,
      "step": 460
    },
    {
      "epoch": 0.3315950368638734,
      "grad_norm": 11.237735546551782,
      "learning_rate": 2.9934862582673547e-06,
      "loss": 0.3597,
      "step": 461
    },
    {
      "epoch": 0.3323143319546844,
      "grad_norm": 6.777642651412544,
      "learning_rate": 2.9934545457669722e-06,
      "loss": 0.091,
      "step": 462
    },
    {
      "epoch": 0.33303362704549544,
      "grad_norm": 6.099996550715816,
      "learning_rate": 2.993422756425732e-06,
      "loss": 0.1597,
      "step": 463
    },
    {
      "epoch": 0.3337529221363064,
      "grad_norm": 10.10252724634063,
      "learning_rate": 2.9933908902452688e-06,
      "loss": 0.2642,
      "step": 464
    },
    {
      "epoch": 0.33447221722711745,
      "grad_norm": 8.73958452919609,
      "learning_rate": 2.9933589472272228e-06,
      "loss": 0.1605,
      "step": 465
    },
    {
      "epoch": 0.33519151231792843,
      "grad_norm": 10.637020856187041,
      "learning_rate": 2.993326927373238e-06,
      "loss": 0.3334,
      "step": 466
    },
    {
      "epoch": 0.3359108074087394,
      "grad_norm": 7.948066977617699,
      "learning_rate": 2.9932948306849607e-06,
      "loss": 0.3347,
      "step": 467
    },
    {
      "epoch": 0.33663010249955044,
      "grad_norm": 11.7979073268195,
      "learning_rate": 2.9932626571640433e-06,
      "loss": 0.5206,
      "step": 468
    },
    {
      "epoch": 0.3373493975903614,
      "grad_norm": 7.492798403225308,
      "learning_rate": 2.993230406812141e-06,
      "loss": 0.1728,
      "step": 469
    },
    {
      "epoch": 0.33806869268117246,
      "grad_norm": 7.385286357099298,
      "learning_rate": 2.9931980796309132e-06,
      "loss": 0.1249,
      "step": 470
    },
    {
      "epoch": 0.33878798777198343,
      "grad_norm": 7.8909238454313995,
      "learning_rate": 2.993165675622022e-06,
      "loss": 0.2917,
      "step": 471
    },
    {
      "epoch": 0.33950728286279447,
      "grad_norm": 7.794771394916748,
      "learning_rate": 2.9931331947871364e-06,
      "loss": 0.2058,
      "step": 472
    },
    {
      "epoch": 0.34022657795360545,
      "grad_norm": 7.267841128071171,
      "learning_rate": 2.9931006371279264e-06,
      "loss": 0.1779,
      "step": 473
    },
    {
      "epoch": 0.3409458730444165,
      "grad_norm": 6.995480773011945,
      "learning_rate": 2.993068002646068e-06,
      "loss": 0.1657,
      "step": 474
    },
    {
      "epoch": 0.34166516813522746,
      "grad_norm": 6.687070292531621,
      "learning_rate": 2.99303529134324e-06,
      "loss": 0.1984,
      "step": 475
    },
    {
      "epoch": 0.3423844632260385,
      "grad_norm": 7.04320434425698,
      "learning_rate": 2.9930025032211246e-06,
      "loss": 0.1614,
      "step": 476
    },
    {
      "epoch": 0.34310375831684947,
      "grad_norm": 7.962174451472046,
      "learning_rate": 2.99296963828141e-06,
      "loss": 0.1542,
      "step": 477
    },
    {
      "epoch": 0.3438230534076605,
      "grad_norm": 7.506214315558402,
      "learning_rate": 2.9929366965257867e-06,
      "loss": 0.0898,
      "step": 478
    },
    {
      "epoch": 0.3445423484984715,
      "grad_norm": 7.815488630923755,
      "learning_rate": 2.9929036779559494e-06,
      "loss": 0.2657,
      "step": 479
    },
    {
      "epoch": 0.3452616435892825,
      "grad_norm": 12.177242440046614,
      "learning_rate": 2.9928705825735976e-06,
      "loss": 0.2595,
      "step": 480
    },
    {
      "epoch": 0.3459809386800935,
      "grad_norm": 9.561239344410755,
      "learning_rate": 2.992837410380433e-06,
      "loss": 0.2468,
      "step": 481
    },
    {
      "epoch": 0.34670023377090453,
      "grad_norm": 9.73125276390459,
      "learning_rate": 2.992804161378163e-06,
      "loss": 0.248,
      "step": 482
    },
    {
      "epoch": 0.3474195288617155,
      "grad_norm": 10.663792090176129,
      "learning_rate": 2.992770835568499e-06,
      "loss": 0.4104,
      "step": 483
    },
    {
      "epoch": 0.34813882395252654,
      "grad_norm": 6.349392040007562,
      "learning_rate": 2.9927374329531547e-06,
      "loss": 0.0644,
      "step": 484
    },
    {
      "epoch": 0.3488581190433375,
      "grad_norm": 8.458321034729055,
      "learning_rate": 2.9927039535338493e-06,
      "loss": 0.2923,
      "step": 485
    },
    {
      "epoch": 0.34957741413414856,
      "grad_norm": 5.340889505512948,
      "learning_rate": 2.9926703973123047e-06,
      "loss": 0.0805,
      "step": 486
    },
    {
      "epoch": 0.35029670922495953,
      "grad_norm": 9.371768873596796,
      "learning_rate": 2.9926367642902484e-06,
      "loss": 0.242,
      "step": 487
    },
    {
      "epoch": 0.35101600431577057,
      "grad_norm": 6.487285839408496,
      "learning_rate": 2.9926030544694098e-06,
      "loss": 0.0783,
      "step": 488
    },
    {
      "epoch": 0.35173529940658155,
      "grad_norm": 7.616234299134105,
      "learning_rate": 2.9925692678515243e-06,
      "loss": 0.0961,
      "step": 489
    },
    {
      "epoch": 0.3524545944973926,
      "grad_norm": 6.992847248226069,
      "learning_rate": 2.9925354044383292e-06,
      "loss": 0.1936,
      "step": 490
    },
    {
      "epoch": 0.35317388958820356,
      "grad_norm": 9.653437422048437,
      "learning_rate": 2.992501464231568e-06,
      "loss": 0.1853,
      "step": 491
    },
    {
      "epoch": 0.35389318467901454,
      "grad_norm": 6.1322948238121775,
      "learning_rate": 2.992467447232986e-06,
      "loss": 0.202,
      "step": 492
    },
    {
      "epoch": 0.3546124797698256,
      "grad_norm": 4.817342730839772,
      "learning_rate": 2.9924333534443345e-06,
      "loss": 0.0877,
      "step": 493
    },
    {
      "epoch": 0.35533177486063655,
      "grad_norm": 9.542291528465045,
      "learning_rate": 2.9923991828673674e-06,
      "loss": 0.345,
      "step": 494
    },
    {
      "epoch": 0.3560510699514476,
      "grad_norm": 13.503074532301907,
      "learning_rate": 2.9923649355038418e-06,
      "loss": 0.0948,
      "step": 495
    },
    {
      "epoch": 0.35677036504225856,
      "grad_norm": 8.40388429041765,
      "learning_rate": 2.992330611355521e-06,
      "loss": 0.3541,
      "step": 496
    },
    {
      "epoch": 0.3574896601330696,
      "grad_norm": 10.966619584674941,
      "learning_rate": 2.9922962104241697e-06,
      "loss": 0.3177,
      "step": 497
    },
    {
      "epoch": 0.3582089552238806,
      "grad_norm": 7.13962635679365,
      "learning_rate": 2.9922617327115596e-06,
      "loss": 0.3161,
      "step": 498
    },
    {
      "epoch": 0.3589282503146916,
      "grad_norm": 6.188551855241201,
      "learning_rate": 2.9922271782194633e-06,
      "loss": 0.3086,
      "step": 499
    },
    {
      "epoch": 0.3596475454055026,
      "grad_norm": 8.454800054539927,
      "learning_rate": 2.9921925469496592e-06,
      "loss": 0.2869,
      "step": 500
    },
    {
      "epoch": 0.3603668404963136,
      "grad_norm": 7.162322257370967,
      "learning_rate": 2.992157838903929e-06,
      "loss": 0.2941,
      "step": 501
    },
    {
      "epoch": 0.3610861355871246,
      "grad_norm": 11.809733697609971,
      "learning_rate": 2.992123054084059e-06,
      "loss": 0.2529,
      "step": 502
    },
    {
      "epoch": 0.36180543067793564,
      "grad_norm": 11.192862994065177,
      "learning_rate": 2.992088192491838e-06,
      "loss": 0.292,
      "step": 503
    },
    {
      "epoch": 0.3625247257687466,
      "grad_norm": 6.543959350421433,
      "learning_rate": 2.992053254129061e-06,
      "loss": 0.1429,
      "step": 504
    },
    {
      "epoch": 0.36324402085955765,
      "grad_norm": 6.468307908696831,
      "learning_rate": 2.992018238997524e-06,
      "loss": 0.2569,
      "step": 505
    },
    {
      "epoch": 0.3639633159503686,
      "grad_norm": 6.7613249894524,
      "learning_rate": 2.99198314709903e-06,
      "loss": 0.0466,
      "step": 506
    },
    {
      "epoch": 0.36468261104117966,
      "grad_norm": 8.295124851483795,
      "learning_rate": 2.9919479784353837e-06,
      "loss": 0.1143,
      "step": 507
    },
    {
      "epoch": 0.36540190613199064,
      "grad_norm": 7.524959168455001,
      "learning_rate": 2.991912733008395e-06,
      "loss": 0.2229,
      "step": 508
    },
    {
      "epoch": 0.3661212012228017,
      "grad_norm": 6.718934638289708,
      "learning_rate": 2.991877410819877e-06,
      "loss": 0.2365,
      "step": 509
    },
    {
      "epoch": 0.36684049631361265,
      "grad_norm": 13.189631741873894,
      "learning_rate": 2.9918420118716476e-06,
      "loss": 0.4204,
      "step": 510
    },
    {
      "epoch": 0.3675597914044237,
      "grad_norm": 10.64610812505103,
      "learning_rate": 2.991806536165528e-06,
      "loss": 0.3337,
      "step": 511
    },
    {
      "epoch": 0.36827908649523466,
      "grad_norm": 12.436063489280153,
      "learning_rate": 2.991770983703343e-06,
      "loss": 0.279,
      "step": 512
    },
    {
      "epoch": 0.3689983815860457,
      "grad_norm": 8.553716095731936,
      "learning_rate": 2.991735354486922e-06,
      "loss": 0.2034,
      "step": 513
    },
    {
      "epoch": 0.3697176766768567,
      "grad_norm": 10.794690904916838,
      "learning_rate": 2.991699648518099e-06,
      "loss": 0.3749,
      "step": 514
    },
    {
      "epoch": 0.3704369717676677,
      "grad_norm": 23.608326345542828,
      "learning_rate": 2.99166386579871e-06,
      "loss": 0.388,
      "step": 515
    },
    {
      "epoch": 0.3711562668584787,
      "grad_norm": 8.114143071314698,
      "learning_rate": 2.991628006330596e-06,
      "loss": 0.2778,
      "step": 516
    },
    {
      "epoch": 0.3718755619492897,
      "grad_norm": 7.394027502213448,
      "learning_rate": 2.991592070115604e-06,
      "loss": 0.1384,
      "step": 517
    },
    {
      "epoch": 0.3725948570401007,
      "grad_norm": 10.12722355127731,
      "learning_rate": 2.9915560571555805e-06,
      "loss": 0.2268,
      "step": 518
    },
    {
      "epoch": 0.3733141521309117,
      "grad_norm": 8.959721954921136,
      "learning_rate": 2.9915199674523797e-06,
      "loss": 0.154,
      "step": 519
    },
    {
      "epoch": 0.3740334472217227,
      "grad_norm": 11.097370858106627,
      "learning_rate": 2.991483801007859e-06,
      "loss": 0.2331,
      "step": 520
    },
    {
      "epoch": 0.3747527423125337,
      "grad_norm": 10.23729402319527,
      "learning_rate": 2.9914475578238775e-06,
      "loss": 0.2367,
      "step": 521
    },
    {
      "epoch": 0.3754720374033447,
      "grad_norm": 8.759875368003424,
      "learning_rate": 2.9914112379023017e-06,
      "loss": 0.2951,
      "step": 522
    },
    {
      "epoch": 0.3761913324941557,
      "grad_norm": 8.902771910725425,
      "learning_rate": 2.9913748412449996e-06,
      "loss": 0.3176,
      "step": 523
    },
    {
      "epoch": 0.37691062758496674,
      "grad_norm": 7.050805312115364,
      "learning_rate": 2.991338367853844e-06,
      "loss": 0.2211,
      "step": 524
    },
    {
      "epoch": 0.3776299226757777,
      "grad_norm": 7.734429974304813,
      "learning_rate": 2.9913018177307105e-06,
      "loss": 0.231,
      "step": 525
    },
    {
      "epoch": 0.37834921776658875,
      "grad_norm": 5.43757036795608,
      "learning_rate": 2.9912651908774814e-06,
      "loss": 0.1995,
      "step": 526
    },
    {
      "epoch": 0.37906851285739973,
      "grad_norm": 6.483873167419547,
      "learning_rate": 2.9912284872960397e-06,
      "loss": 0.2105,
      "step": 527
    },
    {
      "epoch": 0.37978780794821076,
      "grad_norm": 7.3010543475949925,
      "learning_rate": 2.9911917069882753e-06,
      "loss": 0.0761,
      "step": 528
    },
    {
      "epoch": 0.38050710303902174,
      "grad_norm": 8.32884538218666,
      "learning_rate": 2.9911548499560794e-06,
      "loss": 0.1122,
      "step": 529
    },
    {
      "epoch": 0.3812263981298328,
      "grad_norm": 8.480023160282716,
      "learning_rate": 2.9911179162013497e-06,
      "loss": 0.2223,
      "step": 530
    },
    {
      "epoch": 0.38194569322064376,
      "grad_norm": 12.045079861664414,
      "learning_rate": 2.9910809057259844e-06,
      "loss": 0.3155,
      "step": 531
    },
    {
      "epoch": 0.3826649883114548,
      "grad_norm": 7.271205841558662,
      "learning_rate": 2.99104381853189e-06,
      "loss": 0.1235,
      "step": 532
    },
    {
      "epoch": 0.38338428340226577,
      "grad_norm": 15.42344551275697,
      "learning_rate": 2.991006654620973e-06,
      "loss": 0.3456,
      "step": 533
    },
    {
      "epoch": 0.3841035784930768,
      "grad_norm": 8.111247993915727,
      "learning_rate": 2.9909694139951465e-06,
      "loss": 0.0458,
      "step": 534
    },
    {
      "epoch": 0.3848228735838878,
      "grad_norm": 6.181929860819091,
      "learning_rate": 2.9909320966563264e-06,
      "loss": 0.1166,
      "step": 535
    },
    {
      "epoch": 0.3855421686746988,
      "grad_norm": 4.268457502173128,
      "learning_rate": 2.9908947026064325e-06,
      "loss": 0.1057,
      "step": 536
    },
    {
      "epoch": 0.3862614637655098,
      "grad_norm": 9.901830929736837,
      "learning_rate": 2.990857231847389e-06,
      "loss": 0.2146,
      "step": 537
    },
    {
      "epoch": 0.38698075885632083,
      "grad_norm": 6.815730598847511,
      "learning_rate": 2.9908196843811242e-06,
      "loss": 0.1423,
      "step": 538
    },
    {
      "epoch": 0.3877000539471318,
      "grad_norm": 8.483742294843939,
      "learning_rate": 2.990782060209569e-06,
      "loss": 0.2209,
      "step": 539
    },
    {
      "epoch": 0.38841934903794284,
      "grad_norm": 10.974153112435175,
      "learning_rate": 2.99074435933466e-06,
      "loss": 0.5092,
      "step": 540
    },
    {
      "epoch": 0.3891386441287538,
      "grad_norm": 11.761229447582627,
      "learning_rate": 2.9907065817583372e-06,
      "loss": 0.4249,
      "step": 541
    },
    {
      "epoch": 0.38985793921956485,
      "grad_norm": 5.364895028470896,
      "learning_rate": 2.9906687274825434e-06,
      "loss": 0.2539,
      "step": 542
    },
    {
      "epoch": 0.39057723431037583,
      "grad_norm": 15.050771989918358,
      "learning_rate": 2.990630796509227e-06,
      "loss": 0.2649,
      "step": 543
    },
    {
      "epoch": 0.3912965294011868,
      "grad_norm": 9.976972697423335,
      "learning_rate": 2.9905927888403396e-06,
      "loss": 0.302,
      "step": 544
    },
    {
      "epoch": 0.39201582449199784,
      "grad_norm": 7.391920640469044,
      "learning_rate": 2.990554704477836e-06,
      "loss": 0.0516,
      "step": 545
    },
    {
      "epoch": 0.3927351195828088,
      "grad_norm": 7.041586378117951,
      "learning_rate": 2.990516543423677e-06,
      "loss": 0.1401,
      "step": 546
    },
    {
      "epoch": 0.39345441467361986,
      "grad_norm": 8.427415270807284,
      "learning_rate": 2.990478305679825e-06,
      "loss": 0.3231,
      "step": 547
    },
    {
      "epoch": 0.39417370976443084,
      "grad_norm": 9.918189223212861,
      "learning_rate": 2.9904399912482473e-06,
      "loss": 0.2854,
      "step": 548
    },
    {
      "epoch": 0.39489300485524187,
      "grad_norm": 5.0042513771867085,
      "learning_rate": 2.9904016001309162e-06,
      "loss": 0.2168,
      "step": 549
    },
    {
      "epoch": 0.39561229994605285,
      "grad_norm": 7.54059565286538,
      "learning_rate": 2.990363132329806e-06,
      "loss": 0.1821,
      "step": 550
    },
    {
      "epoch": 0.3963315950368639,
      "grad_norm": 7.863760165364239,
      "learning_rate": 2.990324587846897e-06,
      "loss": 0.2189,
      "step": 551
    },
    {
      "epoch": 0.39705089012767486,
      "grad_norm": 9.16986964019923,
      "learning_rate": 2.990285966684171e-06,
      "loss": 0.3196,
      "step": 552
    },
    {
      "epoch": 0.3977701852184859,
      "grad_norm": 9.123454530945438,
      "learning_rate": 2.9902472688436166e-06,
      "loss": 0.1538,
      "step": 553
    },
    {
      "epoch": 0.3984894803092969,
      "grad_norm": 8.8030982967322,
      "learning_rate": 2.990208494327224e-06,
      "loss": 0.1839,
      "step": 554
    },
    {
      "epoch": 0.3992087754001079,
      "grad_norm": 6.085133818971661,
      "learning_rate": 2.990169643136988e-06,
      "loss": 0.1835,
      "step": 555
    },
    {
      "epoch": 0.3999280704909189,
      "grad_norm": 5.4845724521404176,
      "learning_rate": 2.9901307152749085e-06,
      "loss": 0.1069,
      "step": 556
    },
    {
      "epoch": 0.4006473655817299,
      "grad_norm": 10.55518273666302,
      "learning_rate": 2.990091710742988e-06,
      "loss": 0.1236,
      "step": 557
    },
    {
      "epoch": 0.4013666606725409,
      "grad_norm": 6.282292647659121,
      "learning_rate": 2.9900526295432323e-06,
      "loss": 0.1235,
      "step": 558
    },
    {
      "epoch": 0.40208595576335193,
      "grad_norm": 6.778760028112376,
      "learning_rate": 2.9900134716776535e-06,
      "loss": 0.1123,
      "step": 559
    },
    {
      "epoch": 0.4028052508541629,
      "grad_norm": 10.965481679600794,
      "learning_rate": 2.989974237148266e-06,
      "loss": 0.2948,
      "step": 560
    },
    {
      "epoch": 0.40352454594497394,
      "grad_norm": 9.358817453662242,
      "learning_rate": 2.9899349259570887e-06,
      "loss": 0.3162,
      "step": 561
    },
    {
      "epoch": 0.4042438410357849,
      "grad_norm": 9.072919156506938,
      "learning_rate": 2.989895538106144e-06,
      "loss": 0.1678,
      "step": 562
    },
    {
      "epoch": 0.40496313612659596,
      "grad_norm": 7.334111730541219,
      "learning_rate": 2.989856073597458e-06,
      "loss": 0.1197,
      "step": 563
    },
    {
      "epoch": 0.40568243121740694,
      "grad_norm": 4.1093610492547,
      "learning_rate": 2.9898165324330617e-06,
      "loss": 0.0688,
      "step": 564
    },
    {
      "epoch": 0.40640172630821797,
      "grad_norm": 5.408633667840331,
      "learning_rate": 2.9897769146149896e-06,
      "loss": 0.1209,
      "step": 565
    },
    {
      "epoch": 0.40712102139902895,
      "grad_norm": 9.990975918910715,
      "learning_rate": 2.9897372201452804e-06,
      "loss": 0.2834,
      "step": 566
    },
    {
      "epoch": 0.40784031648984,
      "grad_norm": 7.2589940043717895,
      "learning_rate": 2.9896974490259754e-06,
      "loss": 0.041,
      "step": 567
    },
    {
      "epoch": 0.40855961158065096,
      "grad_norm": 7.773056940759599,
      "learning_rate": 2.989657601259122e-06,
      "loss": 0.1928,
      "step": 568
    },
    {
      "epoch": 0.409278906671462,
      "grad_norm": 6.567740277178054,
      "learning_rate": 2.9896176768467703e-06,
      "loss": 0.0657,
      "step": 569
    },
    {
      "epoch": 0.409998201762273,
      "grad_norm": 10.81902515901499,
      "learning_rate": 2.9895776757909735e-06,
      "loss": 0.1264,
      "step": 570
    },
    {
      "epoch": 0.41071749685308395,
      "grad_norm": 8.865799846970328,
      "learning_rate": 2.989537598093791e-06,
      "loss": 0.3872,
      "step": 571
    },
    {
      "epoch": 0.411436791943895,
      "grad_norm": 15.254691911822684,
      "learning_rate": 2.9894974437572835e-06,
      "loss": 0.2536,
      "step": 572
    },
    {
      "epoch": 0.41215608703470596,
      "grad_norm": 7.159124207249005,
      "learning_rate": 2.9894572127835187e-06,
      "loss": 0.1144,
      "step": 573
    },
    {
      "epoch": 0.412875382125517,
      "grad_norm": 8.79482496857362,
      "learning_rate": 2.989416905174565e-06,
      "loss": 0.1438,
      "step": 574
    },
    {
      "epoch": 0.413594677216328,
      "grad_norm": 7.027221158617834,
      "learning_rate": 2.989376520932497e-06,
      "loss": 0.2318,
      "step": 575
    },
    {
      "epoch": 0.414313972307139,
      "grad_norm": 7.251576110185878,
      "learning_rate": 2.989336060059393e-06,
      "loss": 0.1345,
      "step": 576
    },
    {
      "epoch": 0.41503326739795,
      "grad_norm": 5.7282401777393845,
      "learning_rate": 2.989295522557334e-06,
      "loss": 0.1738,
      "step": 577
    },
    {
      "epoch": 0.415752562488761,
      "grad_norm": 7.512920906389525,
      "learning_rate": 2.9892549084284058e-06,
      "loss": 0.1235,
      "step": 578
    },
    {
      "epoch": 0.416471857579572,
      "grad_norm": 5.087999073143586,
      "learning_rate": 2.989214217674698e-06,
      "loss": 0.048,
      "step": 579
    },
    {
      "epoch": 0.41719115267038304,
      "grad_norm": 11.190401308971477,
      "learning_rate": 2.9891734502983053e-06,
      "loss": 0.4374,
      "step": 580
    },
    {
      "epoch": 0.417910447761194,
      "grad_norm": 8.535846595307731,
      "learning_rate": 2.9891326063013237e-06,
      "loss": 0.3329,
      "step": 581
    },
    {
      "epoch": 0.41862974285200505,
      "grad_norm": 8.045877789828847,
      "learning_rate": 2.9890916856858554e-06,
      "loss": 0.2156,
      "step": 582
    },
    {
      "epoch": 0.419349037942816,
      "grad_norm": 6.369909292910831,
      "learning_rate": 2.9890506884540064e-06,
      "loss": 0.5132,
      "step": 583
    },
    {
      "epoch": 0.42006833303362706,
      "grad_norm": 7.818690288087174,
      "learning_rate": 2.989009614607885e-06,
      "loss": 0.1819,
      "step": 584
    },
    {
      "epoch": 0.42078762812443804,
      "grad_norm": 6.884612407988808,
      "learning_rate": 2.9889684641496055e-06,
      "loss": 0.0953,
      "step": 585
    },
    {
      "epoch": 0.4215069232152491,
      "grad_norm": 8.551756168499713,
      "learning_rate": 2.988927237081284e-06,
      "loss": 0.3049,
      "step": 586
    },
    {
      "epoch": 0.42222621830606005,
      "grad_norm": 6.796960057299801,
      "learning_rate": 2.9888859334050435e-06,
      "loss": 0.0677,
      "step": 587
    },
    {
      "epoch": 0.4229455133968711,
      "grad_norm": 8.561352029905176,
      "learning_rate": 2.9888445531230076e-06,
      "loss": 0.0882,
      "step": 588
    },
    {
      "epoch": 0.42366480848768207,
      "grad_norm": 7.6161044367457835,
      "learning_rate": 2.988803096237306e-06,
      "loss": 0.3083,
      "step": 589
    },
    {
      "epoch": 0.4243841035784931,
      "grad_norm": 13.454930616180322,
      "learning_rate": 2.988761562750071e-06,
      "loss": 0.59,
      "step": 590
    },
    {
      "epoch": 0.4251033986693041,
      "grad_norm": 8.0279068423302,
      "learning_rate": 2.988719952663441e-06,
      "loss": 0.1738,
      "step": 591
    },
    {
      "epoch": 0.4258226937601151,
      "grad_norm": 6.064140395677239,
      "learning_rate": 2.988678265979555e-06,
      "loss": 0.1167,
      "step": 592
    },
    {
      "epoch": 0.4265419888509261,
      "grad_norm": 7.566254479498275,
      "learning_rate": 2.9886365027005595e-06,
      "loss": 0.167,
      "step": 593
    },
    {
      "epoch": 0.4272612839417371,
      "grad_norm": 7.160711044708271,
      "learning_rate": 2.9885946628286028e-06,
      "loss": 0.1023,
      "step": 594
    },
    {
      "epoch": 0.4279805790325481,
      "grad_norm": 10.181651394413505,
      "learning_rate": 2.9885527463658377e-06,
      "loss": 0.257,
      "step": 595
    },
    {
      "epoch": 0.4286998741233591,
      "grad_norm": 5.118949215894016,
      "learning_rate": 2.98851075331442e-06,
      "loss": 0.0944,
      "step": 596
    },
    {
      "epoch": 0.4294191692141701,
      "grad_norm": 7.512608938043663,
      "learning_rate": 2.988468683676512e-06,
      "loss": 0.2957,
      "step": 597
    },
    {
      "epoch": 0.4301384643049811,
      "grad_norm": 6.641450299767169,
      "learning_rate": 2.9884265374542765e-06,
      "loss": 0.1542,
      "step": 598
    },
    {
      "epoch": 0.43085775939579213,
      "grad_norm": 6.652629687802819,
      "learning_rate": 2.988384314649883e-06,
      "loss": 0.252,
      "step": 599
    },
    {
      "epoch": 0.4315770544866031,
      "grad_norm": 9.917527283754659,
      "learning_rate": 2.988342015265504e-06,
      "loss": 0.322,
      "step": 600
    },
    {
      "epoch": 0.43229634957741414,
      "grad_norm": 7.346807434337759,
      "learning_rate": 2.988299639303316e-06,
      "loss": 0.3069,
      "step": 601
    },
    {
      "epoch": 0.4330156446682251,
      "grad_norm": 12.027092956868286,
      "learning_rate": 2.9882571867654986e-06,
      "loss": 0.3036,
      "step": 602
    },
    {
      "epoch": 0.43373493975903615,
      "grad_norm": 7.871003547860107,
      "learning_rate": 2.9882146576542357e-06,
      "loss": 0.1724,
      "step": 603
    },
    {
      "epoch": 0.43445423484984713,
      "grad_norm": 11.4574400805975,
      "learning_rate": 2.988172051971717e-06,
      "loss": 0.5131,
      "step": 604
    },
    {
      "epoch": 0.43517352994065817,
      "grad_norm": 6.963701130057769,
      "learning_rate": 2.9881293697201334e-06,
      "loss": 0.2412,
      "step": 605
    },
    {
      "epoch": 0.43589282503146914,
      "grad_norm": 7.798993128468614,
      "learning_rate": 2.988086610901681e-06,
      "loss": 0.1144,
      "step": 606
    },
    {
      "epoch": 0.4366121201222802,
      "grad_norm": 9.079845766536438,
      "learning_rate": 2.9880437755185613e-06,
      "loss": 0.312,
      "step": 607
    },
    {
      "epoch": 0.43733141521309116,
      "grad_norm": 8.26055961405721,
      "learning_rate": 2.9880008635729766e-06,
      "loss": 0.2598,
      "step": 608
    },
    {
      "epoch": 0.4380507103039022,
      "grad_norm": 6.518289068819356,
      "learning_rate": 2.9879578750671356e-06,
      "loss": 0.1522,
      "step": 609
    },
    {
      "epoch": 0.43877000539471317,
      "grad_norm": 10.980823267639687,
      "learning_rate": 2.987914810003249e-06,
      "loss": 0.331,
      "step": 610
    },
    {
      "epoch": 0.4394893004855242,
      "grad_norm": 8.462160580724552,
      "learning_rate": 2.9878716683835343e-06,
      "loss": 0.2036,
      "step": 611
    },
    {
      "epoch": 0.4402085955763352,
      "grad_norm": 6.504916199265367,
      "learning_rate": 2.9878284502102104e-06,
      "loss": 0.2739,
      "step": 612
    },
    {
      "epoch": 0.4409278906671462,
      "grad_norm": 9.6391017965284,
      "learning_rate": 2.9877851554855007e-06,
      "loss": 0.26,
      "step": 613
    },
    {
      "epoch": 0.4416471857579572,
      "grad_norm": 6.822877387341788,
      "learning_rate": 2.987741784211633e-06,
      "loss": 0.1603,
      "step": 614
    },
    {
      "epoch": 0.44236648084876823,
      "grad_norm": 8.354057412567972,
      "learning_rate": 2.987698336390839e-06,
      "loss": 0.2595,
      "step": 615
    },
    {
      "epoch": 0.4430857759395792,
      "grad_norm": 7.712805431473086,
      "learning_rate": 2.987654812025354e-06,
      "loss": 0.3013,
      "step": 616
    },
    {
      "epoch": 0.44380507103039024,
      "grad_norm": 6.820823276133487,
      "learning_rate": 2.9876112111174175e-06,
      "loss": 0.0679,
      "step": 617
    },
    {
      "epoch": 0.4445243661212012,
      "grad_norm": 7.3218377848106275,
      "learning_rate": 2.9875675336692728e-06,
      "loss": 0.1217,
      "step": 618
    },
    {
      "epoch": 0.44524366121201225,
      "grad_norm": 7.803188661672437,
      "learning_rate": 2.9875237796831665e-06,
      "loss": 0.2676,
      "step": 619
    },
    {
      "epoch": 0.44596295630282323,
      "grad_norm": 8.325760523497854,
      "learning_rate": 2.9874799491613515e-06,
      "loss": 0.316,
      "step": 620
    },
    {
      "epoch": 0.4466822513936342,
      "grad_norm": 6.090735806757629,
      "learning_rate": 2.987436042106081e-06,
      "loss": 0.291,
      "step": 621
    },
    {
      "epoch": 0.44740154648444524,
      "grad_norm": 3.802907758694769,
      "learning_rate": 2.9873920585196155e-06,
      "loss": 0.0147,
      "step": 622
    },
    {
      "epoch": 0.4481208415752562,
      "grad_norm": 4.945923938755192,
      "learning_rate": 2.9873479984042177e-06,
      "loss": 0.0283,
      "step": 623
    },
    {
      "epoch": 0.44884013666606726,
      "grad_norm": 6.643269050647207,
      "learning_rate": 2.987303861762154e-06,
      "loss": 0.1419,
      "step": 624
    },
    {
      "epoch": 0.44955943175687824,
      "grad_norm": 11.47672021422071,
      "learning_rate": 2.987259648595696e-06,
      "loss": 0.5138,
      "step": 625
    },
    {
      "epoch": 0.45027872684768927,
      "grad_norm": 12.200240121757396,
      "learning_rate": 2.9872153589071185e-06,
      "loss": 0.2653,
      "step": 626
    },
    {
      "epoch": 0.45099802193850025,
      "grad_norm": 5.487828060112799,
      "learning_rate": 2.9871709926987e-06,
      "loss": 0.0673,
      "step": 627
    },
    {
      "epoch": 0.4517173170293113,
      "grad_norm": 7.96134008578685,
      "learning_rate": 2.9871265499727227e-06,
      "loss": 0.0956,
      "step": 628
    },
    {
      "epoch": 0.45243661212012226,
      "grad_norm": 9.014733530641363,
      "learning_rate": 2.9870820307314743e-06,
      "loss": 0.099,
      "step": 629
    },
    {
      "epoch": 0.4531559072109333,
      "grad_norm": 5.620593454289586,
      "learning_rate": 2.9870374349772448e-06,
      "loss": 0.0739,
      "step": 630
    },
    {
      "epoch": 0.4538752023017443,
      "grad_norm": 12.261486479299116,
      "learning_rate": 2.986992762712329e-06,
      "loss": 0.4411,
      "step": 631
    },
    {
      "epoch": 0.4545944973925553,
      "grad_norm": 9.810601747113425,
      "learning_rate": 2.9869480139390247e-06,
      "loss": 0.2169,
      "step": 632
    },
    {
      "epoch": 0.4553137924833663,
      "grad_norm": 12.102061342834876,
      "learning_rate": 2.9869031886596357e-06,
      "loss": 0.1651,
      "step": 633
    },
    {
      "epoch": 0.4560330875741773,
      "grad_norm": 5.488367023817783,
      "learning_rate": 2.9868582868764667e-06,
      "loss": 0.1201,
      "step": 634
    },
    {
      "epoch": 0.4567523826649883,
      "grad_norm": 7.617074599084058,
      "learning_rate": 2.986813308591829e-06,
      "loss": 0.1553,
      "step": 635
    },
    {
      "epoch": 0.45747167775579933,
      "grad_norm": 5.5786678434242605,
      "learning_rate": 2.986768253808037e-06,
      "loss": 0.2147,
      "step": 636
    },
    {
      "epoch": 0.4581909728466103,
      "grad_norm": 8.989821213248181,
      "learning_rate": 2.9867231225274075e-06,
      "loss": 0.3028,
      "step": 637
    },
    {
      "epoch": 0.45891026793742135,
      "grad_norm": 5.091181429753673,
      "learning_rate": 2.986677914752264e-06,
      "loss": 0.0591,
      "step": 638
    },
    {
      "epoch": 0.4596295630282323,
      "grad_norm": 10.911399896249282,
      "learning_rate": 2.986632630484932e-06,
      "loss": 0.5262,
      "step": 639
    },
    {
      "epoch": 0.46034885811904336,
      "grad_norm": 6.656110126300123,
      "learning_rate": 2.9865872697277417e-06,
      "loss": 0.1292,
      "step": 640
    },
    {
      "epoch": 0.46106815320985434,
      "grad_norm": 10.0668832546612,
      "learning_rate": 2.9865418324830263e-06,
      "loss": 0.3637,
      "step": 641
    },
    {
      "epoch": 0.46178744830066537,
      "grad_norm": 8.427653507735005,
      "learning_rate": 2.986496318753124e-06,
      "loss": 0.107,
      "step": 642
    },
    {
      "epoch": 0.46250674339147635,
      "grad_norm": 4.764136536876361,
      "learning_rate": 2.986450728540377e-06,
      "loss": 0.1999,
      "step": 643
    },
    {
      "epoch": 0.4632260384822874,
      "grad_norm": 3.9011328290520995,
      "learning_rate": 2.986405061847131e-06,
      "loss": 0.0203,
      "step": 644
    },
    {
      "epoch": 0.46394533357309836,
      "grad_norm": 6.459055537304067,
      "learning_rate": 2.986359318675735e-06,
      "loss": 0.0821,
      "step": 645
    },
    {
      "epoch": 0.4646646286639094,
      "grad_norm": 7.965736761871791,
      "learning_rate": 2.9863134990285425e-06,
      "loss": 0.4077,
      "step": 646
    },
    {
      "epoch": 0.4653839237547204,
      "grad_norm": 6.2188393746309965,
      "learning_rate": 2.9862676029079114e-06,
      "loss": 0.3448,
      "step": 647
    },
    {
      "epoch": 0.46610321884553135,
      "grad_norm": 7.443160591071639,
      "learning_rate": 2.986221630316203e-06,
      "loss": 0.1343,
      "step": 648
    },
    {
      "epoch": 0.4668225139363424,
      "grad_norm": 7.108644129952765,
      "learning_rate": 2.986175581255783e-06,
      "loss": 0.2194,
      "step": 649
    },
    {
      "epoch": 0.46754180902715337,
      "grad_norm": 6.813646701011755,
      "learning_rate": 2.9861294557290204e-06,
      "loss": 0.2985,
      "step": 650
    },
    {
      "epoch": 0.4682611041179644,
      "grad_norm": 6.585814058876483,
      "learning_rate": 2.986083253738288e-06,
      "loss": 0.1636,
      "step": 651
    },
    {
      "epoch": 0.4689803992087754,
      "grad_norm": 8.545202459019087,
      "learning_rate": 2.986036975285964e-06,
      "loss": 0.3853,
      "step": 652
    },
    {
      "epoch": 0.4696996942995864,
      "grad_norm": 8.29350019433357,
      "learning_rate": 2.9859906203744294e-06,
      "loss": 0.2736,
      "step": 653
    },
    {
      "epoch": 0.4704189893903974,
      "grad_norm": 7.26657270385466,
      "learning_rate": 2.985944189006068e-06,
      "loss": 0.3296,
      "step": 654
    },
    {
      "epoch": 0.4711382844812084,
      "grad_norm": 4.716989904329773,
      "learning_rate": 2.9858976811832696e-06,
      "loss": 0.1396,
      "step": 655
    },
    {
      "epoch": 0.4718575795720194,
      "grad_norm": 5.303771176013311,
      "learning_rate": 2.9858510969084277e-06,
      "loss": 0.2398,
      "step": 656
    },
    {
      "epoch": 0.47257687466283044,
      "grad_norm": 5.264001879842815,
      "learning_rate": 2.985804436183938e-06,
      "loss": 0.2143,
      "step": 657
    },
    {
      "epoch": 0.4732961697536414,
      "grad_norm": 8.491284647304514,
      "learning_rate": 2.985757699012202e-06,
      "loss": 0.4539,
      "step": 658
    },
    {
      "epoch": 0.47401546484445245,
      "grad_norm": 9.197947922453682,
      "learning_rate": 2.9857108853956246e-06,
      "loss": 0.3073,
      "step": 659
    },
    {
      "epoch": 0.47473475993526343,
      "grad_norm": 8.184881279794077,
      "learning_rate": 2.9856639953366134e-06,
      "loss": 0.2799,
      "step": 660
    },
    {
      "epoch": 0.47545405502607446,
      "grad_norm": 6.276697037518068,
      "learning_rate": 2.9856170288375815e-06,
      "loss": 0.2185,
      "step": 661
    },
    {
      "epoch": 0.47617335011688544,
      "grad_norm": 3.5239432360839817,
      "learning_rate": 2.9855699859009463e-06,
      "loss": 0.036,
      "step": 662
    },
    {
      "epoch": 0.4768926452076965,
      "grad_norm": 7.090429537477977,
      "learning_rate": 2.9855228665291273e-06,
      "loss": 0.3242,
      "step": 663
    },
    {
      "epoch": 0.47761194029850745,
      "grad_norm": 9.354126086365557,
      "learning_rate": 2.9854756707245487e-06,
      "loss": 0.1249,
      "step": 664
    },
    {
      "epoch": 0.4783312353893185,
      "grad_norm": 8.30907469094051,
      "learning_rate": 2.9854283984896394e-06,
      "loss": 0.3307,
      "step": 665
    },
    {
      "epoch": 0.47905053048012947,
      "grad_norm": 9.96926455120718,
      "learning_rate": 2.9853810498268316e-06,
      "loss": 0.2644,
      "step": 666
    },
    {
      "epoch": 0.4797698255709405,
      "grad_norm": 10.064518275162772,
      "learning_rate": 2.9853336247385607e-06,
      "loss": 0.1975,
      "step": 667
    },
    {
      "epoch": 0.4804891206617515,
      "grad_norm": 9.843722243189204,
      "learning_rate": 2.985286123227268e-06,
      "loss": 0.1901,
      "step": 668
    },
    {
      "epoch": 0.4812084157525625,
      "grad_norm": 13.833537456986445,
      "learning_rate": 2.9852385452953966e-06,
      "loss": 0.5601,
      "step": 669
    },
    {
      "epoch": 0.4819277108433735,
      "grad_norm": 10.319994056237286,
      "learning_rate": 2.985190890945395e-06,
      "loss": 0.2334,
      "step": 670
    },
    {
      "epoch": 0.4826470059341845,
      "grad_norm": 5.913399332588977,
      "learning_rate": 2.9851431601797144e-06,
      "loss": 0.0998,
      "step": 671
    },
    {
      "epoch": 0.4833663010249955,
      "grad_norm": 9.09397873095818,
      "learning_rate": 2.9850953530008115e-06,
      "loss": 0.1238,
      "step": 672
    },
    {
      "epoch": 0.4840855961158065,
      "grad_norm": 8.559550481125624,
      "learning_rate": 2.985047469411146e-06,
      "loss": 0.22,
      "step": 673
    },
    {
      "epoch": 0.4848048912066175,
      "grad_norm": 5.346905375652378,
      "learning_rate": 2.984999509413181e-06,
      "loss": 0.2827,
      "step": 674
    },
    {
      "epoch": 0.4855241862974285,
      "grad_norm": 5.272507163621891,
      "learning_rate": 2.984951473009384e-06,
      "loss": 0.0607,
      "step": 675
    },
    {
      "epoch": 0.48624348138823953,
      "grad_norm": 8.902859741949358,
      "learning_rate": 2.9849033602022273e-06,
      "loss": 0.0788,
      "step": 676
    },
    {
      "epoch": 0.4869627764790505,
      "grad_norm": 5.137273810772606,
      "learning_rate": 2.984855170994186e-06,
      "loss": 0.1291,
      "step": 677
    },
    {
      "epoch": 0.48768207156986154,
      "grad_norm": 9.110879109386415,
      "learning_rate": 2.98480690538774e-06,
      "loss": 0.1309,
      "step": 678
    },
    {
      "epoch": 0.4884013666606725,
      "grad_norm": 7.998516210908917,
      "learning_rate": 2.984758563385372e-06,
      "loss": 0.3183,
      "step": 679
    },
    {
      "epoch": 0.48912066175148355,
      "grad_norm": 7.2025293524855485,
      "learning_rate": 2.984710144989569e-06,
      "loss": 0.3035,
      "step": 680
    },
    {
      "epoch": 0.48983995684229453,
      "grad_norm": 11.145719945762266,
      "learning_rate": 2.9846616502028233e-06,
      "loss": 0.2932,
      "step": 681
    },
    {
      "epoch": 0.49055925193310557,
      "grad_norm": 13.72270979098324,
      "learning_rate": 2.9846130790276292e-06,
      "loss": 0.4495,
      "step": 682
    },
    {
      "epoch": 0.49127854702391655,
      "grad_norm": 7.064784455853499,
      "learning_rate": 2.984564431466486e-06,
      "loss": 0.3142,
      "step": 683
    },
    {
      "epoch": 0.4919978421147276,
      "grad_norm": 7.014948623482035,
      "learning_rate": 2.984515707521897e-06,
      "loss": 0.1956,
      "step": 684
    },
    {
      "epoch": 0.49271713720553856,
      "grad_norm": 6.795210125441131,
      "learning_rate": 2.9844669071963684e-06,
      "loss": 0.1822,
      "step": 685
    },
    {
      "epoch": 0.4934364322963496,
      "grad_norm": 8.325957162522203,
      "learning_rate": 2.9844180304924117e-06,
      "loss": 0.0935,
      "step": 686
    },
    {
      "epoch": 0.49415572738716057,
      "grad_norm": 9.216902870286452,
      "learning_rate": 2.984369077412542e-06,
      "loss": 0.2075,
      "step": 687
    },
    {
      "epoch": 0.4948750224779716,
      "grad_norm": 2.841775058998902,
      "learning_rate": 2.984320047959277e-06,
      "loss": 0.0145,
      "step": 688
    },
    {
      "epoch": 0.4955943175687826,
      "grad_norm": 5.812174435888005,
      "learning_rate": 2.98427094213514e-06,
      "loss": 0.2834,
      "step": 689
    },
    {
      "epoch": 0.4963136126595936,
      "grad_norm": 8.10164249230395,
      "learning_rate": 2.9842217599426575e-06,
      "loss": 0.214,
      "step": 690
    },
    {
      "epoch": 0.4970329077504046,
      "grad_norm": 3.8883802970936796,
      "learning_rate": 2.98417250138436e-06,
      "loss": 0.0363,
      "step": 691
    },
    {
      "epoch": 0.49775220284121563,
      "grad_norm": 8.266520114913085,
      "learning_rate": 2.984123166462782e-06,
      "loss": 0.1406,
      "step": 692
    },
    {
      "epoch": 0.4984714979320266,
      "grad_norm": 6.241849146339004,
      "learning_rate": 2.9840737551804614e-06,
      "loss": 0.2336,
      "step": 693
    },
    {
      "epoch": 0.49919079302283764,
      "grad_norm": 10.290181309552962,
      "learning_rate": 2.9840242675399413e-06,
      "loss": 0.3195,
      "step": 694
    },
    {
      "epoch": 0.4999100881136486,
      "grad_norm": 9.230658413338618,
      "learning_rate": 2.9839747035437676e-06,
      "loss": 0.2065,
      "step": 695
    },
    {
      "epoch": 0.5006293832044596,
      "grad_norm": 6.484240878839433,
      "learning_rate": 2.98392506319449e-06,
      "loss": 0.1148,
      "step": 696
    },
    {
      "epoch": 0.5013486782952706,
      "grad_norm": 6.185091579068325,
      "learning_rate": 2.9838753464946635e-06,
      "loss": 0.3242,
      "step": 697
    },
    {
      "epoch": 0.5020679733860817,
      "grad_norm": 3.5617342463535513,
      "learning_rate": 2.983825553446845e-06,
      "loss": 0.1263,
      "step": 698
    },
    {
      "epoch": 0.5027872684768926,
      "grad_norm": 8.275063334715663,
      "learning_rate": 2.983775684053597e-06,
      "loss": 0.1571,
      "step": 699
    },
    {
      "epoch": 0.5035065635677036,
      "grad_norm": 11.220184159353963,
      "learning_rate": 2.9837257383174856e-06,
      "loss": 0.2024,
      "step": 700
    },
    {
      "epoch": 0.5042258586585147,
      "grad_norm": 8.877131241467795,
      "learning_rate": 2.9836757162410803e-06,
      "loss": 0.1502,
      "step": 701
    },
    {
      "epoch": 0.5049451537493257,
      "grad_norm": 7.9782318968943144,
      "learning_rate": 2.983625617826955e-06,
      "loss": 0.4078,
      "step": 702
    },
    {
      "epoch": 0.5056644488401366,
      "grad_norm": 7.498834138261986,
      "learning_rate": 2.983575443077687e-06,
      "loss": 0.1716,
      "step": 703
    },
    {
      "epoch": 0.5063837439309476,
      "grad_norm": 7.683376921578665,
      "learning_rate": 2.9835251919958582e-06,
      "loss": 0.3026,
      "step": 704
    },
    {
      "epoch": 0.5071030390217587,
      "grad_norm": 8.30032932846491,
      "learning_rate": 2.983474864584054e-06,
      "loss": 0.2691,
      "step": 705
    },
    {
      "epoch": 0.5078223341125697,
      "grad_norm": 4.6011062591079135,
      "learning_rate": 2.983424460844864e-06,
      "loss": 0.1886,
      "step": 706
    },
    {
      "epoch": 0.5085416292033806,
      "grad_norm": 7.5362654330016,
      "learning_rate": 2.983373980780881e-06,
      "loss": 0.2406,
      "step": 707
    },
    {
      "epoch": 0.5092609242941917,
      "grad_norm": 7.103149578313491,
      "learning_rate": 2.9833234243947035e-06,
      "loss": 0.2909,
      "step": 708
    },
    {
      "epoch": 0.5099802193850027,
      "grad_norm": 7.0080138150005675,
      "learning_rate": 2.9832727916889317e-06,
      "loss": 0.226,
      "step": 709
    },
    {
      "epoch": 0.5106995144758137,
      "grad_norm": 7.438000679237786,
      "learning_rate": 2.9832220826661706e-06,
      "loss": 0.2728,
      "step": 710
    },
    {
      "epoch": 0.5114188095666247,
      "grad_norm": 5.9851638874817406,
      "learning_rate": 2.98317129732903e-06,
      "loss": 0.2456,
      "step": 711
    },
    {
      "epoch": 0.5121381046574357,
      "grad_norm": 8.225734653045484,
      "learning_rate": 2.983120435680122e-06,
      "loss": 0.1945,
      "step": 712
    },
    {
      "epoch": 0.5128573997482467,
      "grad_norm": 5.482244552245788,
      "learning_rate": 2.9830694977220643e-06,
      "loss": 0.1894,
      "step": 713
    },
    {
      "epoch": 0.5135766948390578,
      "grad_norm": 10.237687884137133,
      "learning_rate": 2.9830184834574777e-06,
      "loss": 0.2621,
      "step": 714
    },
    {
      "epoch": 0.5142959899298687,
      "grad_norm": 7.576925742496242,
      "learning_rate": 2.982967392888987e-06,
      "loss": 0.3155,
      "step": 715
    },
    {
      "epoch": 0.5150152850206797,
      "grad_norm": 7.294886815052957,
      "learning_rate": 2.98291622601922e-06,
      "loss": 0.2994,
      "step": 716
    },
    {
      "epoch": 0.5157345801114908,
      "grad_norm": 7.306832655659561,
      "learning_rate": 2.9828649828508104e-06,
      "loss": 0.2122,
      "step": 717
    },
    {
      "epoch": 0.5164538752023018,
      "grad_norm": 7.26810226698831,
      "learning_rate": 2.982813663386394e-06,
      "loss": 0.1442,
      "step": 718
    },
    {
      "epoch": 0.5171731702931127,
      "grad_norm": 10.433100264454152,
      "learning_rate": 2.982762267628612e-06,
      "loss": 0.1599,
      "step": 719
    },
    {
      "epoch": 0.5178924653839238,
      "grad_norm": 9.097836449521381,
      "learning_rate": 2.982710795580108e-06,
      "loss": 0.1606,
      "step": 720
    },
    {
      "epoch": 0.5186117604747348,
      "grad_norm": 7.726615576774665,
      "learning_rate": 2.982659247243531e-06,
      "loss": 0.3383,
      "step": 721
    },
    {
      "epoch": 0.5193310555655458,
      "grad_norm": 10.375030166002839,
      "learning_rate": 2.9826076226215332e-06,
      "loss": 0.1489,
      "step": 722
    },
    {
      "epoch": 0.5200503506563567,
      "grad_norm": 4.060363415490424,
      "learning_rate": 2.9825559217167704e-06,
      "loss": 0.0198,
      "step": 723
    },
    {
      "epoch": 0.5207696457471678,
      "grad_norm": 4.569597776815956,
      "learning_rate": 2.982504144531903e-06,
      "loss": 0.0489,
      "step": 724
    },
    {
      "epoch": 0.5214889408379788,
      "grad_norm": 5.166227494881769,
      "learning_rate": 2.982452291069595e-06,
      "loss": 0.1906,
      "step": 725
    },
    {
      "epoch": 0.5222082359287897,
      "grad_norm": 6.530324433683541,
      "learning_rate": 2.9824003613325138e-06,
      "loss": 0.1675,
      "step": 726
    },
    {
      "epoch": 0.5229275310196008,
      "grad_norm": 3.5966463961911264,
      "learning_rate": 2.9823483553233324e-06,
      "loss": 0.0334,
      "step": 727
    },
    {
      "epoch": 0.5236468261104118,
      "grad_norm": 6.514705195264471,
      "learning_rate": 2.982296273044725e-06,
      "loss": 0.1166,
      "step": 728
    },
    {
      "epoch": 0.5243661212012228,
      "grad_norm": 7.176127604553889,
      "learning_rate": 2.982244114499373e-06,
      "loss": 0.2241,
      "step": 729
    },
    {
      "epoch": 0.5250854162920338,
      "grad_norm": 10.866936803204483,
      "learning_rate": 2.982191879689959e-06,
      "loss": 0.1729,
      "step": 730
    },
    {
      "epoch": 0.5258047113828448,
      "grad_norm": 9.303566206043383,
      "learning_rate": 2.982139568619171e-06,
      "loss": 0.3529,
      "step": 731
    },
    {
      "epoch": 0.5265240064736558,
      "grad_norm": 8.556510060440488,
      "learning_rate": 2.9820871812897007e-06,
      "loss": 0.3519,
      "step": 732
    },
    {
      "epoch": 0.5272433015644669,
      "grad_norm": 4.548847054868503,
      "learning_rate": 2.9820347177042427e-06,
      "loss": 0.0885,
      "step": 733
    },
    {
      "epoch": 0.5279625966552778,
      "grad_norm": 6.4970436363222355,
      "learning_rate": 2.981982177865497e-06,
      "loss": 0.227,
      "step": 734
    },
    {
      "epoch": 0.5286818917460888,
      "grad_norm": 7.420866239821348,
      "learning_rate": 2.981929561776167e-06,
      "loss": 0.2299,
      "step": 735
    },
    {
      "epoch": 0.5294011868368999,
      "grad_norm": 4.732821810252163,
      "learning_rate": 2.981876869438959e-06,
      "loss": 0.0814,
      "step": 736
    },
    {
      "epoch": 0.5301204819277109,
      "grad_norm": 7.684468584751193,
      "learning_rate": 2.981824100856585e-06,
      "loss": 0.2261,
      "step": 737
    },
    {
      "epoch": 0.5308397770185218,
      "grad_norm": 11.498093492987481,
      "learning_rate": 2.9817712560317602e-06,
      "loss": 0.2003,
      "step": 738
    },
    {
      "epoch": 0.5315590721093328,
      "grad_norm": 5.899400287135329,
      "learning_rate": 2.9817183349672028e-06,
      "loss": 0.1327,
      "step": 739
    },
    {
      "epoch": 0.5322783672001439,
      "grad_norm": 7.28054074370964,
      "learning_rate": 2.981665337665636e-06,
      "loss": 0.3803,
      "step": 740
    },
    {
      "epoch": 0.5329976622909549,
      "grad_norm": 9.39912153050433,
      "learning_rate": 2.9816122641297866e-06,
      "loss": 0.3053,
      "step": 741
    },
    {
      "epoch": 0.5337169573817658,
      "grad_norm": 5.43629824544262,
      "learning_rate": 2.981559114362385e-06,
      "loss": 0.1666,
      "step": 742
    },
    {
      "epoch": 0.5344362524725769,
      "grad_norm": 8.028951891863958,
      "learning_rate": 2.981505888366166e-06,
      "loss": 0.1775,
      "step": 743
    },
    {
      "epoch": 0.5351555475633879,
      "grad_norm": 7.897529684917291,
      "learning_rate": 2.981452586143869e-06,
      "loss": 0.172,
      "step": 744
    },
    {
      "epoch": 0.5358748426541989,
      "grad_norm": 7.422495662263877,
      "learning_rate": 2.9813992076982357e-06,
      "loss": 0.2211,
      "step": 745
    },
    {
      "epoch": 0.5365941377450099,
      "grad_norm": 4.731913158161135,
      "learning_rate": 2.9813457530320123e-06,
      "loss": 0.1039,
      "step": 746
    },
    {
      "epoch": 0.5373134328358209,
      "grad_norm": 7.63727860675573,
      "learning_rate": 2.9812922221479497e-06,
      "loss": 0.1062,
      "step": 747
    },
    {
      "epoch": 0.5380327279266319,
      "grad_norm": 8.234338080748223,
      "learning_rate": 2.981238615048802e-06,
      "loss": 0.1273,
      "step": 748
    },
    {
      "epoch": 0.538752023017443,
      "grad_norm": 8.317128213380808,
      "learning_rate": 2.9811849317373274e-06,
      "loss": 0.2148,
      "step": 749
    },
    {
      "epoch": 0.5394713181082539,
      "grad_norm": 12.143393161110994,
      "learning_rate": 2.981131172216288e-06,
      "loss": 0.2847,
      "step": 750
    },
    {
      "epoch": 0.5401906131990649,
      "grad_norm": 7.808658804256654,
      "learning_rate": 2.9810773364884493e-06,
      "loss": 0.0629,
      "step": 751
    },
    {
      "epoch": 0.540909908289876,
      "grad_norm": 7.742930271868104,
      "learning_rate": 2.981023424556582e-06,
      "loss": 0.1126,
      "step": 752
    },
    {
      "epoch": 0.5416292033806869,
      "grad_norm": 5.016080151648933,
      "learning_rate": 2.9809694364234597e-06,
      "loss": 0.1071,
      "step": 753
    },
    {
      "epoch": 0.5423484984714979,
      "grad_norm": 6.053067099455596,
      "learning_rate": 2.98091537209186e-06,
      "loss": 0.0862,
      "step": 754
    },
    {
      "epoch": 0.5430677935623089,
      "grad_norm": 9.692485088274085,
      "learning_rate": 2.980861231564565e-06,
      "loss": 0.243,
      "step": 755
    },
    {
      "epoch": 0.54378708865312,
      "grad_norm": 6.457491962704044,
      "learning_rate": 2.9808070148443594e-06,
      "loss": 0.1179,
      "step": 756
    },
    {
      "epoch": 0.5445063837439309,
      "grad_norm": 8.732374837188866,
      "learning_rate": 2.980752721934034e-06,
      "loss": 0.2488,
      "step": 757
    },
    {
      "epoch": 0.5452256788347419,
      "grad_norm": 5.924270881380956,
      "learning_rate": 2.9806983528363816e-06,
      "loss": 0.1444,
      "step": 758
    },
    {
      "epoch": 0.545944973925553,
      "grad_norm": 7.5988289184306,
      "learning_rate": 2.9806439075541993e-06,
      "loss": 0.1845,
      "step": 759
    },
    {
      "epoch": 0.546664269016364,
      "grad_norm": 11.24620708238816,
      "learning_rate": 2.980589386090289e-06,
      "loss": 0.0838,
      "step": 760
    },
    {
      "epoch": 0.5473835641071749,
      "grad_norm": 5.958027477253964,
      "learning_rate": 2.980534788447456e-06,
      "loss": 0.1508,
      "step": 761
    },
    {
      "epoch": 0.548102859197986,
      "grad_norm": 13.125376727031792,
      "learning_rate": 2.9804801146285088e-06,
      "loss": 0.2799,
      "step": 762
    },
    {
      "epoch": 0.548822154288797,
      "grad_norm": 8.536323223419584,
      "learning_rate": 2.980425364636261e-06,
      "loss": 0.1704,
      "step": 763
    },
    {
      "epoch": 0.549541449379608,
      "grad_norm": 9.20526378210383,
      "learning_rate": 2.980370538473529e-06,
      "loss": 0.2532,
      "step": 764
    },
    {
      "epoch": 0.550260744470419,
      "grad_norm": 4.356231652665598,
      "learning_rate": 2.9803156361431346e-06,
      "loss": 0.0946,
      "step": 765
    },
    {
      "epoch": 0.55098003956123,
      "grad_norm": 3.5342297318161093,
      "learning_rate": 2.980260657647902e-06,
      "loss": 0.0499,
      "step": 766
    },
    {
      "epoch": 0.551699334652041,
      "grad_norm": 5.79384531558842,
      "learning_rate": 2.9802056029906597e-06,
      "loss": 0.1506,
      "step": 767
    },
    {
      "epoch": 0.552418629742852,
      "grad_norm": 9.612019951464557,
      "learning_rate": 2.980150472174241e-06,
      "loss": 0.1162,
      "step": 768
    },
    {
      "epoch": 0.553137924833663,
      "grad_norm": 6.795021971884667,
      "learning_rate": 2.980095265201482e-06,
      "loss": 0.2568,
      "step": 769
    },
    {
      "epoch": 0.553857219924474,
      "grad_norm": 4.046837386431047,
      "learning_rate": 2.9800399820752235e-06,
      "loss": 0.0829,
      "step": 770
    },
    {
      "epoch": 0.554576515015285,
      "grad_norm": 9.140823444754814,
      "learning_rate": 2.9799846227983097e-06,
      "loss": 0.2293,
      "step": 771
    },
    {
      "epoch": 0.5552958101060961,
      "grad_norm": 4.868268078666935,
      "learning_rate": 2.979929187373589e-06,
      "loss": 0.0318,
      "step": 772
    },
    {
      "epoch": 0.556015105196907,
      "grad_norm": 9.610275083769052,
      "learning_rate": 2.979873675803914e-06,
      "loss": 0.261,
      "step": 773
    },
    {
      "epoch": 0.556734400287718,
      "grad_norm": 8.981453072200969,
      "learning_rate": 2.9798180880921405e-06,
      "loss": 0.2829,
      "step": 774
    },
    {
      "epoch": 0.5574536953785291,
      "grad_norm": 5.700400150105251,
      "learning_rate": 2.9797624242411286e-06,
      "loss": 0.2019,
      "step": 775
    },
    {
      "epoch": 0.55817299046934,
      "grad_norm": 10.113887925402263,
      "learning_rate": 2.979706684253742e-06,
      "loss": 0.2332,
      "step": 776
    },
    {
      "epoch": 0.558892285560151,
      "grad_norm": 4.808140392186802,
      "learning_rate": 2.9796508681328488e-06,
      "loss": 0.1976,
      "step": 777
    },
    {
      "epoch": 0.5596115806509621,
      "grad_norm": 7.986932477784339,
      "learning_rate": 2.9795949758813216e-06,
      "loss": 0.3967,
      "step": 778
    },
    {
      "epoch": 0.5603308757417731,
      "grad_norm": 9.85334741423081,
      "learning_rate": 2.9795390075020353e-06,
      "loss": 0.3556,
      "step": 779
    },
    {
      "epoch": 0.561050170832584,
      "grad_norm": 2.610636021212666,
      "learning_rate": 2.97948296299787e-06,
      "loss": 0.0159,
      "step": 780
    },
    {
      "epoch": 0.561769465923395,
      "grad_norm": 5.897553693290661,
      "learning_rate": 2.9794268423717085e-06,
      "loss": 0.0621,
      "step": 781
    },
    {
      "epoch": 0.5624887610142061,
      "grad_norm": 7.251537052285208,
      "learning_rate": 2.9793706456264397e-06,
      "loss": 0.1637,
      "step": 782
    },
    {
      "epoch": 0.5632080561050171,
      "grad_norm": 8.99624237501475,
      "learning_rate": 2.9793143727649534e-06,
      "loss": 0.1249,
      "step": 783
    },
    {
      "epoch": 0.563927351195828,
      "grad_norm": 6.819524668479388,
      "learning_rate": 2.9792580237901465e-06,
      "loss": 0.075,
      "step": 784
    },
    {
      "epoch": 0.5646466462866391,
      "grad_norm": 3.9832100822175374,
      "learning_rate": 2.979201598704917e-06,
      "loss": 0.1153,
      "step": 785
    },
    {
      "epoch": 0.5653659413774501,
      "grad_norm": 9.265727486230267,
      "learning_rate": 2.979145097512169e-06,
      "loss": 0.2155,
      "step": 786
    },
    {
      "epoch": 0.5660852364682611,
      "grad_norm": 5.731188520557445,
      "learning_rate": 2.979088520214809e-06,
      "loss": 0.171,
      "step": 787
    },
    {
      "epoch": 0.5668045315590721,
      "grad_norm": 6.487421516949131,
      "learning_rate": 2.9790318668157477e-06,
      "loss": 0.166,
      "step": 788
    },
    {
      "epoch": 0.5675238266498831,
      "grad_norm": 10.45535100630016,
      "learning_rate": 2.9789751373179013e-06,
      "loss": 0.2556,
      "step": 789
    },
    {
      "epoch": 0.5682431217406941,
      "grad_norm": 5.256325166345312,
      "learning_rate": 2.978918331724188e-06,
      "loss": 0.0858,
      "step": 790
    },
    {
      "epoch": 0.5689624168315052,
      "grad_norm": 5.737190551099546,
      "learning_rate": 2.9788614500375296e-06,
      "loss": 0.1384,
      "step": 791
    },
    {
      "epoch": 0.5696817119223161,
      "grad_norm": 9.911757841096794,
      "learning_rate": 2.978804492260854e-06,
      "loss": 0.1805,
      "step": 792
    },
    {
      "epoch": 0.5704010070131271,
      "grad_norm": 6.815015945738683,
      "learning_rate": 2.9787474583970907e-06,
      "loss": 0.1975,
      "step": 793
    },
    {
      "epoch": 0.5711203021039382,
      "grad_norm": 10.433350453804477,
      "learning_rate": 2.9786903484491753e-06,
      "loss": 0.1525,
      "step": 794
    },
    {
      "epoch": 0.5718395971947492,
      "grad_norm": 6.323978532782773,
      "learning_rate": 2.9786331624200453e-06,
      "loss": 0.1871,
      "step": 795
    },
    {
      "epoch": 0.5725588922855601,
      "grad_norm": 7.396082338969181,
      "learning_rate": 2.978575900312644e-06,
      "loss": 0.2419,
      "step": 796
    },
    {
      "epoch": 0.5732781873763712,
      "grad_norm": 8.032791141097622,
      "learning_rate": 2.978518562129917e-06,
      "loss": 0.1349,
      "step": 797
    },
    {
      "epoch": 0.5739974824671822,
      "grad_norm": 6.510710205003883,
      "learning_rate": 2.978461147874814e-06,
      "loss": 0.0453,
      "step": 798
    },
    {
      "epoch": 0.5747167775579932,
      "grad_norm": 6.8503162322583115,
      "learning_rate": 2.97840365755029e-06,
      "loss": 0.2867,
      "step": 799
    },
    {
      "epoch": 0.5754360726488041,
      "grad_norm": 2.70073630363353,
      "learning_rate": 2.9783460911593023e-06,
      "loss": 0.0702,
      "step": 800
    },
    {
      "epoch": 0.5761553677396152,
      "grad_norm": 2.9048919548146626,
      "learning_rate": 2.9782884487048135e-06,
      "loss": 0.0469,
      "step": 801
    },
    {
      "epoch": 0.5768746628304262,
      "grad_norm": 5.166261801389186,
      "learning_rate": 2.9782307301897888e-06,
      "loss": 0.1059,
      "step": 802
    },
    {
      "epoch": 0.5775939579212371,
      "grad_norm": 7.376611675269217,
      "learning_rate": 2.978172935617198e-06,
      "loss": 0.1832,
      "step": 803
    },
    {
      "epoch": 0.5783132530120482,
      "grad_norm": 5.681383195694462,
      "learning_rate": 2.9781150649900146e-06,
      "loss": 0.1817,
      "step": 804
    },
    {
      "epoch": 0.5790325481028592,
      "grad_norm": 3.723639290013787,
      "learning_rate": 2.9780571183112164e-06,
      "loss": 0.0189,
      "step": 805
    },
    {
      "epoch": 0.5797518431936702,
      "grad_norm": 10.700512122340205,
      "learning_rate": 2.977999095583785e-06,
      "loss": 0.3174,
      "step": 806
    },
    {
      "epoch": 0.5804711382844812,
      "grad_norm": 5.529048279432231,
      "learning_rate": 2.9779409968107058e-06,
      "loss": 0.1629,
      "step": 807
    },
    {
      "epoch": 0.5811904333752922,
      "grad_norm": 5.842538322821677,
      "learning_rate": 2.9778828219949674e-06,
      "loss": 0.06,
      "step": 808
    },
    {
      "epoch": 0.5819097284661032,
      "grad_norm": 9.259137428455585,
      "learning_rate": 2.9778245711395643e-06,
      "loss": 0.4163,
      "step": 809
    },
    {
      "epoch": 0.5826290235569143,
      "grad_norm": 10.802316835435011,
      "learning_rate": 2.977766244247492e-06,
      "loss": 0.1261,
      "step": 810
    },
    {
      "epoch": 0.5833483186477252,
      "grad_norm": 10.72991981273546,
      "learning_rate": 2.9777078413217525e-06,
      "loss": 0.2708,
      "step": 811
    },
    {
      "epoch": 0.5840676137385362,
      "grad_norm": 7.103068474529098,
      "learning_rate": 2.9776493623653505e-06,
      "loss": 0.2065,
      "step": 812
    },
    {
      "epoch": 0.5847869088293473,
      "grad_norm": 8.449802954978395,
      "learning_rate": 2.9775908073812947e-06,
      "loss": 0.2483,
      "step": 813
    },
    {
      "epoch": 0.5855062039201583,
      "grad_norm": 7.945204474271803,
      "learning_rate": 2.977532176372598e-06,
      "loss": 0.1614,
      "step": 814
    },
    {
      "epoch": 0.5862254990109692,
      "grad_norm": 6.46974024027023,
      "learning_rate": 2.9774734693422776e-06,
      "loss": 0.1636,
      "step": 815
    },
    {
      "epoch": 0.5869447941017802,
      "grad_norm": 2.5219976288419357,
      "learning_rate": 2.9774146862933535e-06,
      "loss": 0.011,
      "step": 816
    },
    {
      "epoch": 0.5876640891925913,
      "grad_norm": 7.828295205251189,
      "learning_rate": 2.9773558272288497e-06,
      "loss": 0.1442,
      "step": 817
    },
    {
      "epoch": 0.5883833842834023,
      "grad_norm": 3.81562387033852,
      "learning_rate": 2.977296892151796e-06,
      "loss": 0.0766,
      "step": 818
    },
    {
      "epoch": 0.5891026793742132,
      "grad_norm": 9.429459668067874,
      "learning_rate": 2.9772378810652234e-06,
      "loss": 0.1673,
      "step": 819
    },
    {
      "epoch": 0.5898219744650243,
      "grad_norm": 5.713974117643566,
      "learning_rate": 2.977178793972168e-06,
      "loss": 0.0452,
      "step": 820
    },
    {
      "epoch": 0.5905412695558353,
      "grad_norm": 6.732103452438126,
      "learning_rate": 2.9771196308756717e-06,
      "loss": 0.1426,
      "step": 821
    },
    {
      "epoch": 0.5912605646466463,
      "grad_norm": 4.709989179152442,
      "learning_rate": 2.9770603917787763e-06,
      "loss": 0.1111,
      "step": 822
    },
    {
      "epoch": 0.5919798597374573,
      "grad_norm": 6.1052474311868155,
      "learning_rate": 2.9770010766845315e-06,
      "loss": 0.0889,
      "step": 823
    },
    {
      "epoch": 0.5926991548282683,
      "grad_norm": 5.044198680883185,
      "learning_rate": 2.9769416855959886e-06,
      "loss": 0.1284,
      "step": 824
    },
    {
      "epoch": 0.5934184499190793,
      "grad_norm": 4.620031748227179,
      "learning_rate": 2.9768822185162034e-06,
      "loss": 0.1813,
      "step": 825
    },
    {
      "epoch": 0.5941377450098904,
      "grad_norm": 4.955822576738682,
      "learning_rate": 2.9768226754482352e-06,
      "loss": 0.0611,
      "step": 826
    },
    {
      "epoch": 0.5948570401007013,
      "grad_norm": 7.06242278617012,
      "learning_rate": 2.9767630563951475e-06,
      "loss": 0.1843,
      "step": 827
    },
    {
      "epoch": 0.5955763351915123,
      "grad_norm": 5.004874387211047,
      "learning_rate": 2.9767033613600085e-06,
      "loss": 0.1357,
      "step": 828
    },
    {
      "epoch": 0.5962956302823234,
      "grad_norm": 7.795250779728666,
      "learning_rate": 2.9766435903458897e-06,
      "loss": 0.2441,
      "step": 829
    },
    {
      "epoch": 0.5970149253731343,
      "grad_norm": 4.433821476306185,
      "learning_rate": 2.9765837433558652e-06,
      "loss": 0.1733,
      "step": 830
    },
    {
      "epoch": 0.5977342204639453,
      "grad_norm": 3.720381732391177,
      "learning_rate": 2.9765238203930155e-06,
      "loss": 0.1126,
      "step": 831
    },
    {
      "epoch": 0.5984535155547563,
      "grad_norm": 8.256773045292633,
      "learning_rate": 2.976463821460423e-06,
      "loss": 0.1267,
      "step": 832
    },
    {
      "epoch": 0.5991728106455674,
      "grad_norm": 6.335327053294731,
      "learning_rate": 2.9764037465611757e-06,
      "loss": 0.1256,
      "step": 833
    },
    {
      "epoch": 0.5998921057363783,
      "grad_norm": 5.176784166666724,
      "learning_rate": 2.9763435956983635e-06,
      "loss": 0.028,
      "step": 834
    },
    {
      "epoch": 0.6006114008271893,
      "grad_norm": 5.120191727793925,
      "learning_rate": 2.9762833688750816e-06,
      "loss": 0.0464,
      "step": 835
    },
    {
      "epoch": 0.6013306959180004,
      "grad_norm": 6.275983829145228,
      "learning_rate": 2.976223066094429e-06,
      "loss": 0.1378,
      "step": 836
    },
    {
      "epoch": 0.6020499910088114,
      "grad_norm": 8.702513434315492,
      "learning_rate": 2.976162687359508e-06,
      "loss": 0.1712,
      "step": 837
    },
    {
      "epoch": 0.6027692860996223,
      "grad_norm": 8.700678119525492,
      "learning_rate": 2.9761022326734253e-06,
      "loss": 0.0744,
      "step": 838
    },
    {
      "epoch": 0.6034885811904334,
      "grad_norm": 9.153608113492444,
      "learning_rate": 2.976041702039292e-06,
      "loss": 0.3031,
      "step": 839
    },
    {
      "epoch": 0.6042078762812444,
      "grad_norm": 10.418573599689886,
      "learning_rate": 2.975981095460222e-06,
      "loss": 0.4991,
      "step": 840
    },
    {
      "epoch": 0.6049271713720554,
      "grad_norm": 7.871958835957954,
      "learning_rate": 2.975920412939333e-06,
      "loss": 0.357,
      "step": 841
    },
    {
      "epoch": 0.6056464664628664,
      "grad_norm": 5.115200490467494,
      "learning_rate": 2.975859654479748e-06,
      "loss": 0.2311,
      "step": 842
    },
    {
      "epoch": 0.6063657615536774,
      "grad_norm": 5.9669621928778565,
      "learning_rate": 2.9757988200845935e-06,
      "loss": 0.1251,
      "step": 843
    },
    {
      "epoch": 0.6070850566444884,
      "grad_norm": 6.396034729264032,
      "learning_rate": 2.9757379097569988e-06,
      "loss": 0.1222,
      "step": 844
    },
    {
      "epoch": 0.6078043517352995,
      "grad_norm": 2.6562509712514792,
      "learning_rate": 2.975676923500098e-06,
      "loss": 0.0217,
      "step": 845
    },
    {
      "epoch": 0.6085236468261104,
      "grad_norm": 7.02411881784209,
      "learning_rate": 2.975615861317029e-06,
      "loss": 0.2655,
      "step": 846
    },
    {
      "epoch": 0.6092429419169214,
      "grad_norm": 4.529085862499067,
      "learning_rate": 2.9755547232109334e-06,
      "loss": 0.0962,
      "step": 847
    },
    {
      "epoch": 0.6099622370077324,
      "grad_norm": 4.750791297610307,
      "learning_rate": 2.9754935091849575e-06,
      "loss": 0.0283,
      "step": 848
    },
    {
      "epoch": 0.6106815320985435,
      "grad_norm": 8.73875925573561,
      "learning_rate": 2.9754322192422497e-06,
      "loss": 0.281,
      "step": 849
    },
    {
      "epoch": 0.6114008271893544,
      "grad_norm": 1.627604198611702,
      "learning_rate": 2.9753708533859654e-06,
      "loss": 0.0255,
      "step": 850
    },
    {
      "epoch": 0.6121201222801654,
      "grad_norm": 3.921026272107377,
      "learning_rate": 2.9753094116192595e-06,
      "loss": 0.0459,
      "step": 851
    },
    {
      "epoch": 0.6128394173709765,
      "grad_norm": 5.204394312262345,
      "learning_rate": 2.975247893945295e-06,
      "loss": 0.2747,
      "step": 852
    },
    {
      "epoch": 0.6135587124617875,
      "grad_norm": 5.117453766528016,
      "learning_rate": 2.9751863003672364e-06,
      "loss": 0.1247,
      "step": 853
    },
    {
      "epoch": 0.6142780075525984,
      "grad_norm": 2.2165642479184364,
      "learning_rate": 2.9751246308882537e-06,
      "loss": 0.0085,
      "step": 854
    },
    {
      "epoch": 0.6149973026434095,
      "grad_norm": 9.728527315951023,
      "learning_rate": 2.975062885511519e-06,
      "loss": 0.3202,
      "step": 855
    },
    {
      "epoch": 0.6157165977342205,
      "grad_norm": 7.023117301343883,
      "learning_rate": 2.975001064240209e-06,
      "loss": 0.0763,
      "step": 856
    },
    {
      "epoch": 0.6164358928250314,
      "grad_norm": 11.218005745567654,
      "learning_rate": 2.9749391670775056e-06,
      "loss": 0.5647,
      "step": 857
    },
    {
      "epoch": 0.6171551879158425,
      "grad_norm": 6.146475892297033,
      "learning_rate": 2.9748771940265923e-06,
      "loss": 0.1852,
      "step": 858
    },
    {
      "epoch": 0.6178744830066535,
      "grad_norm": 6.063987943425242,
      "learning_rate": 2.9748151450906584e-06,
      "loss": 0.0667,
      "step": 859
    },
    {
      "epoch": 0.6185937780974645,
      "grad_norm": 12.564102885214416,
      "learning_rate": 2.9747530202728968e-06,
      "loss": 0.1499,
      "step": 860
    },
    {
      "epoch": 0.6193130731882754,
      "grad_norm": 5.217187746869808,
      "learning_rate": 2.974690819576503e-06,
      "loss": 0.2194,
      "step": 861
    },
    {
      "epoch": 0.6200323682790865,
      "grad_norm": 8.542969665607426,
      "learning_rate": 2.974628543004678e-06,
      "loss": 0.2519,
      "step": 862
    },
    {
      "epoch": 0.6207516633698975,
      "grad_norm": 6.71096918047829,
      "learning_rate": 2.9745661905606255e-06,
      "loss": 0.1985,
      "step": 863
    },
    {
      "epoch": 0.6214709584607085,
      "grad_norm": 8.426627127116994,
      "learning_rate": 2.9745037622475542e-06,
      "loss": 0.2507,
      "step": 864
    },
    {
      "epoch": 0.6221902535515195,
      "grad_norm": 9.730786077324774,
      "learning_rate": 2.9744412580686764e-06,
      "loss": 0.3127,
      "step": 865
    },
    {
      "epoch": 0.6229095486423305,
      "grad_norm": 5.119390609090124,
      "learning_rate": 2.974378678027207e-06,
      "loss": 0.1108,
      "step": 866
    },
    {
      "epoch": 0.6236288437331415,
      "grad_norm": 8.018984599126975,
      "learning_rate": 2.9743160221263667e-06,
      "loss": 0.1288,
      "step": 867
    },
    {
      "epoch": 0.6243481388239526,
      "grad_norm": 4.840556373181076,
      "learning_rate": 2.974253290369379e-06,
      "loss": 0.108,
      "step": 868
    },
    {
      "epoch": 0.6250674339147635,
      "grad_norm": 6.134456768729974,
      "learning_rate": 2.974190482759471e-06,
      "loss": 0.1538,
      "step": 869
    },
    {
      "epoch": 0.6257867290055745,
      "grad_norm": 4.498009589164801,
      "learning_rate": 2.974127599299875e-06,
      "loss": 0.185,
      "step": 870
    },
    {
      "epoch": 0.6265060240963856,
      "grad_norm": 4.333757459896507,
      "learning_rate": 2.9740646399938265e-06,
      "loss": 0.0424,
      "step": 871
    },
    {
      "epoch": 0.6272253191871966,
      "grad_norm": 9.4075492372385,
      "learning_rate": 2.974001604844565e-06,
      "loss": 0.3262,
      "step": 872
    },
    {
      "epoch": 0.6279446142780075,
      "grad_norm": 7.915385965207062,
      "learning_rate": 2.973938493855333e-06,
      "loss": 0.279,
      "step": 873
    },
    {
      "epoch": 0.6286639093688186,
      "grad_norm": 7.425017485693498,
      "learning_rate": 2.973875307029378e-06,
      "loss": 0.1751,
      "step": 874
    },
    {
      "epoch": 0.6293832044596296,
      "grad_norm": 5.25227791466232,
      "learning_rate": 2.973812044369951e-06,
      "loss": 0.1113,
      "step": 875
    },
    {
      "epoch": 0.6301024995504406,
      "grad_norm": 8.690057578540404,
      "learning_rate": 2.9737487058803073e-06,
      "loss": 0.1286,
      "step": 876
    },
    {
      "epoch": 0.6308217946412515,
      "grad_norm": 3.788961434337117,
      "learning_rate": 2.9736852915637056e-06,
      "loss": 0.1133,
      "step": 877
    },
    {
      "epoch": 0.6315410897320626,
      "grad_norm": 6.688914066306362,
      "learning_rate": 2.973621801423409e-06,
      "loss": 0.1359,
      "step": 878
    },
    {
      "epoch": 0.6322603848228736,
      "grad_norm": 5.650706561607885,
      "learning_rate": 2.9735582354626833e-06,
      "loss": 0.1359,
      "step": 879
    },
    {
      "epoch": 0.6329796799136845,
      "grad_norm": 5.536085481886316,
      "learning_rate": 2.9734945936848e-06,
      "loss": 0.1386,
      "step": 880
    },
    {
      "epoch": 0.6336989750044956,
      "grad_norm": 6.869733995664741,
      "learning_rate": 2.9734308760930334e-06,
      "loss": 0.1221,
      "step": 881
    },
    {
      "epoch": 0.6344182700953066,
      "grad_norm": 6.164015336665599,
      "learning_rate": 2.973367082690661e-06,
      "loss": 0.2132,
      "step": 882
    },
    {
      "epoch": 0.6351375651861176,
      "grad_norm": 4.285198217814613,
      "learning_rate": 2.973303213480966e-06,
      "loss": 0.0337,
      "step": 883
    },
    {
      "epoch": 0.6358568602769286,
      "grad_norm": 13.904838648691733,
      "learning_rate": 2.973239268467235e-06,
      "loss": 0.0781,
      "step": 884
    },
    {
      "epoch": 0.6365761553677396,
      "grad_norm": 6.666734623160479,
      "learning_rate": 2.973175247652757e-06,
      "loss": 0.2278,
      "step": 885
    },
    {
      "epoch": 0.6372954504585506,
      "grad_norm": 6.62265366931964,
      "learning_rate": 2.9731111510408264e-06,
      "loss": 0.1156,
      "step": 886
    },
    {
      "epoch": 0.6380147455493617,
      "grad_norm": 9.471533073267807,
      "learning_rate": 2.973046978634741e-06,
      "loss": 0.2777,
      "step": 887
    },
    {
      "epoch": 0.6387340406401726,
      "grad_norm": 4.824796222940127,
      "learning_rate": 2.972982730437803e-06,
      "loss": 0.1629,
      "step": 888
    },
    {
      "epoch": 0.6394533357309836,
      "grad_norm": 5.367577098199019,
      "learning_rate": 2.9729184064533176e-06,
      "loss": 0.2324,
      "step": 889
    },
    {
      "epoch": 0.6401726308217947,
      "grad_norm": 10.36031445648042,
      "learning_rate": 2.9728540066845947e-06,
      "loss": 0.1892,
      "step": 890
    },
    {
      "epoch": 0.6408919259126057,
      "grad_norm": 3.9358579549312864,
      "learning_rate": 2.9727895311349476e-06,
      "loss": 0.1133,
      "step": 891
    },
    {
      "epoch": 0.6416112210034166,
      "grad_norm": 6.990521432262684,
      "learning_rate": 2.9727249798076934e-06,
      "loss": 0.2543,
      "step": 892
    },
    {
      "epoch": 0.6423305160942276,
      "grad_norm": 7.700750845499684,
      "learning_rate": 2.9726603527061537e-06,
      "loss": 0.1537,
      "step": 893
    },
    {
      "epoch": 0.6430498111850387,
      "grad_norm": 7.9271304139400955,
      "learning_rate": 2.9725956498336544e-06,
      "loss": 0.2247,
      "step": 894
    },
    {
      "epoch": 0.6437691062758497,
      "grad_norm": 5.8354026064268085,
      "learning_rate": 2.972530871193523e-06,
      "loss": 0.1517,
      "step": 895
    },
    {
      "epoch": 0.6444884013666606,
      "grad_norm": 6.865431154496454,
      "learning_rate": 2.9724660167890937e-06,
      "loss": 0.1779,
      "step": 896
    },
    {
      "epoch": 0.6452076964574717,
      "grad_norm": 6.300216577933797,
      "learning_rate": 2.9724010866237034e-06,
      "loss": 0.0863,
      "step": 897
    },
    {
      "epoch": 0.6459269915482827,
      "grad_norm": 8.92952838354427,
      "learning_rate": 2.972336080700692e-06,
      "loss": 0.3282,
      "step": 898
    },
    {
      "epoch": 0.6466462866390937,
      "grad_norm": 6.962271718972027,
      "learning_rate": 2.972270999023405e-06,
      "loss": 0.1949,
      "step": 899
    },
    {
      "epoch": 0.6473655817299047,
      "grad_norm": 8.933220531831731,
      "learning_rate": 2.97220584159519e-06,
      "loss": 0.1558,
      "step": 900
    },
    {
      "epoch": 0.6480848768207157,
      "grad_norm": 5.447335503944896,
      "learning_rate": 2.9721406084194e-06,
      "loss": 0.0331,
      "step": 901
    },
    {
      "epoch": 0.6488041719115267,
      "grad_norm": 6.005076152896434,
      "learning_rate": 2.972075299499392e-06,
      "loss": 0.1277,
      "step": 902
    },
    {
      "epoch": 0.6495234670023378,
      "grad_norm": 4.884670744944036,
      "learning_rate": 2.972009914838526e-06,
      "loss": 0.135,
      "step": 903
    },
    {
      "epoch": 0.6502427620931487,
      "grad_norm": 7.780318510269028,
      "learning_rate": 2.9719444544401653e-06,
      "loss": 0.2154,
      "step": 904
    },
    {
      "epoch": 0.6509620571839597,
      "grad_norm": 6.667906910236401,
      "learning_rate": 2.9718789183076786e-06,
      "loss": 0.2764,
      "step": 905
    },
    {
      "epoch": 0.6516813522747708,
      "grad_norm": 3.5288210181500297,
      "learning_rate": 2.971813306444438e-06,
      "loss": 0.1276,
      "step": 906
    },
    {
      "epoch": 0.6524006473655817,
      "grad_norm": 4.6994708361789534,
      "learning_rate": 2.9717476188538185e-06,
      "loss": 0.2257,
      "step": 907
    },
    {
      "epoch": 0.6531199424563927,
      "grad_norm": 8.552437912521276,
      "learning_rate": 2.971681855539201e-06,
      "loss": 0.2624,
      "step": 908
    },
    {
      "epoch": 0.6538392375472037,
      "grad_norm": 5.324840877686902,
      "learning_rate": 2.9716160165039682e-06,
      "loss": 0.1073,
      "step": 909
    },
    {
      "epoch": 0.6545585326380148,
      "grad_norm": 6.083879695968263,
      "learning_rate": 2.971550101751508e-06,
      "loss": 0.0718,
      "step": 910
    },
    {
      "epoch": 0.6552778277288257,
      "grad_norm": 6.576718341399474,
      "learning_rate": 2.9714841112852124e-06,
      "loss": 0.3142,
      "step": 911
    },
    {
      "epoch": 0.6559971228196367,
      "grad_norm": 7.884330445452147,
      "learning_rate": 2.971418045108476e-06,
      "loss": 0.0916,
      "step": 912
    },
    {
      "epoch": 0.6567164179104478,
      "grad_norm": 4.9547657609957225,
      "learning_rate": 2.9713519032246977e-06,
      "loss": 0.0619,
      "step": 913
    },
    {
      "epoch": 0.6574357130012588,
      "grad_norm": 7.268345932188917,
      "learning_rate": 2.9712856856372812e-06,
      "loss": 0.2041,
      "step": 914
    },
    {
      "epoch": 0.6581550080920697,
      "grad_norm": 9.412585766618037,
      "learning_rate": 2.971219392349634e-06,
      "loss": 0.1887,
      "step": 915
    },
    {
      "epoch": 0.6588743031828808,
      "grad_norm": 5.810528225012237,
      "learning_rate": 2.9711530233651656e-06,
      "loss": 0.1326,
      "step": 916
    },
    {
      "epoch": 0.6595935982736918,
      "grad_norm": 4.490857380383742,
      "learning_rate": 2.971086578687292e-06,
      "loss": 0.1755,
      "step": 917
    },
    {
      "epoch": 0.6603128933645028,
      "grad_norm": 8.427661227436248,
      "learning_rate": 2.9710200583194315e-06,
      "loss": 0.2563,
      "step": 918
    },
    {
      "epoch": 0.6610321884553138,
      "grad_norm": 6.352343960941773,
      "learning_rate": 2.970953462265007e-06,
      "loss": 0.2035,
      "step": 919
    },
    {
      "epoch": 0.6617514835461248,
      "grad_norm": 5.8647878017999275,
      "learning_rate": 2.9708867905274444e-06,
      "loss": 0.2122,
      "step": 920
    },
    {
      "epoch": 0.6624707786369358,
      "grad_norm": 6.084389482201497,
      "learning_rate": 2.9708200431101744e-06,
      "loss": 0.213,
      "step": 921
    },
    {
      "epoch": 0.6631900737277469,
      "grad_norm": 6.170068691394691,
      "learning_rate": 2.970753220016631e-06,
      "loss": 0.2141,
      "step": 922
    },
    {
      "epoch": 0.6639093688185578,
      "grad_norm": 4.002915223438293,
      "learning_rate": 2.970686321250253e-06,
      "loss": 0.2609,
      "step": 923
    },
    {
      "epoch": 0.6646286639093688,
      "grad_norm": 7.4291739362946645,
      "learning_rate": 2.970619346814482e-06,
      "loss": 0.1457,
      "step": 924
    },
    {
      "epoch": 0.6653479590001798,
      "grad_norm": 10.923704001605786,
      "learning_rate": 2.970552296712764e-06,
      "loss": 0.3077,
      "step": 925
    },
    {
      "epoch": 0.6660672540909909,
      "grad_norm": 4.567310973115804,
      "learning_rate": 2.970485170948549e-06,
      "loss": 0.1155,
      "step": 926
    },
    {
      "epoch": 0.6667865491818018,
      "grad_norm": 15.33365099139537,
      "learning_rate": 2.9704179695252903e-06,
      "loss": 0.0732,
      "step": 927
    },
    {
      "epoch": 0.6675058442726128,
      "grad_norm": 4.055209818056239,
      "learning_rate": 2.970350692446446e-06,
      "loss": 0.1111,
      "step": 928
    },
    {
      "epoch": 0.6682251393634239,
      "grad_norm": 8.32518136007484,
      "learning_rate": 2.9702833397154773e-06,
      "loss": 0.186,
      "step": 929
    },
    {
      "epoch": 0.6689444344542349,
      "grad_norm": 6.070507782562587,
      "learning_rate": 2.9702159113358496e-06,
      "loss": 0.3945,
      "step": 930
    },
    {
      "epoch": 0.6696637295450458,
      "grad_norm": 10.56763916855245,
      "learning_rate": 2.970148407311033e-06,
      "loss": 0.1321,
      "step": 931
    },
    {
      "epoch": 0.6703830246358569,
      "grad_norm": 7.362035488523589,
      "learning_rate": 2.9700808276444995e-06,
      "loss": 0.0956,
      "step": 932
    },
    {
      "epoch": 0.6711023197266679,
      "grad_norm": 2.405992336768348,
      "learning_rate": 2.970013172339727e-06,
      "loss": 0.0675,
      "step": 933
    },
    {
      "epoch": 0.6718216148174788,
      "grad_norm": 6.233934300851908,
      "learning_rate": 2.969945441400196e-06,
      "loss": 0.156,
      "step": 934
    },
    {
      "epoch": 0.6725409099082899,
      "grad_norm": 5.287259405651,
      "learning_rate": 2.969877634829392e-06,
      "loss": 0.0951,
      "step": 935
    },
    {
      "epoch": 0.6732602049991009,
      "grad_norm": 8.737951734122264,
      "learning_rate": 2.9698097526308034e-06,
      "loss": 0.4278,
      "step": 936
    },
    {
      "epoch": 0.6739795000899119,
      "grad_norm": 11.565358945609637,
      "learning_rate": 2.969741794807923e-06,
      "loss": 0.1419,
      "step": 937
    },
    {
      "epoch": 0.6746987951807228,
      "grad_norm": 13.482488887119393,
      "learning_rate": 2.9696737613642467e-06,
      "loss": 0.1875,
      "step": 938
    },
    {
      "epoch": 0.6754180902715339,
      "grad_norm": 4.447649668678088,
      "learning_rate": 2.9696056523032756e-06,
      "loss": 0.0627,
      "step": 939
    },
    {
      "epoch": 0.6761373853623449,
      "grad_norm": 8.016031755063839,
      "learning_rate": 2.969537467628514e-06,
      "loss": 0.1417,
      "step": 940
    },
    {
      "epoch": 0.676856680453156,
      "grad_norm": 7.756605673502039,
      "learning_rate": 2.9694692073434696e-06,
      "loss": 0.0826,
      "step": 941
    },
    {
      "epoch": 0.6775759755439669,
      "grad_norm": 9.046690224668627,
      "learning_rate": 2.9694008714516555e-06,
      "loss": 0.3833,
      "step": 942
    },
    {
      "epoch": 0.6782952706347779,
      "grad_norm": 7.484253094813111,
      "learning_rate": 2.9693324599565867e-06,
      "loss": 0.0641,
      "step": 943
    },
    {
      "epoch": 0.6790145657255889,
      "grad_norm": 6.483536334863294,
      "learning_rate": 2.969263972861784e-06,
      "loss": 0.2174,
      "step": 944
    },
    {
      "epoch": 0.6797338608164,
      "grad_norm": 7.901059115931261,
      "learning_rate": 2.9691954101707703e-06,
      "loss": 0.208,
      "step": 945
    },
    {
      "epoch": 0.6804531559072109,
      "grad_norm": 8.175716995140329,
      "learning_rate": 2.9691267718870738e-06,
      "loss": 0.2317,
      "step": 946
    },
    {
      "epoch": 0.6811724509980219,
      "grad_norm": 7.685723437258521,
      "learning_rate": 2.9690580580142257e-06,
      "loss": 0.1976,
      "step": 947
    },
    {
      "epoch": 0.681891746088833,
      "grad_norm": 4.707952007348748,
      "learning_rate": 2.9689892685557615e-06,
      "loss": 0.2099,
      "step": 948
    },
    {
      "epoch": 0.682611041179644,
      "grad_norm": 8.336041941938515,
      "learning_rate": 2.9689204035152214e-06,
      "loss": 0.1955,
      "step": 949
    },
    {
      "epoch": 0.6833303362704549,
      "grad_norm": 7.2925717327929,
      "learning_rate": 2.9688514628961473e-06,
      "loss": 0.1584,
      "step": 950
    },
    {
      "epoch": 0.684049631361266,
      "grad_norm": 6.065612648113627,
      "learning_rate": 2.9687824467020873e-06,
      "loss": 0.0913,
      "step": 951
    },
    {
      "epoch": 0.684768926452077,
      "grad_norm": 6.032179881947198,
      "learning_rate": 2.9687133549365923e-06,
      "loss": 0.1195,
      "step": 952
    },
    {
      "epoch": 0.685488221542888,
      "grad_norm": 6.738343429863798,
      "learning_rate": 2.9686441876032163e-06,
      "loss": 0.3167,
      "step": 953
    },
    {
      "epoch": 0.6862075166336989,
      "grad_norm": 5.740668653914783,
      "learning_rate": 2.968574944705519e-06,
      "loss": 0.1593,
      "step": 954
    },
    {
      "epoch": 0.68692681172451,
      "grad_norm": 4.059127673729532,
      "learning_rate": 2.968505626247063e-06,
      "loss": 0.0734,
      "step": 955
    },
    {
      "epoch": 0.687646106815321,
      "grad_norm": 8.467989583057847,
      "learning_rate": 2.968436232231414e-06,
      "loss": 0.217,
      "step": 956
    },
    {
      "epoch": 0.688365401906132,
      "grad_norm": 4.754709716730067,
      "learning_rate": 2.9683667626621436e-06,
      "loss": 0.1607,
      "step": 957
    },
    {
      "epoch": 0.689084696996943,
      "grad_norm": 6.492539802730643,
      "learning_rate": 2.9682972175428257e-06,
      "loss": 0.1135,
      "step": 958
    },
    {
      "epoch": 0.689803992087754,
      "grad_norm": 7.037692384877635,
      "learning_rate": 2.9682275968770386e-06,
      "loss": 0.2422,
      "step": 959
    },
    {
      "epoch": 0.690523287178565,
      "grad_norm": 6.151695768339403,
      "learning_rate": 2.9681579006683636e-06,
      "loss": 0.1268,
      "step": 960
    },
    {
      "epoch": 0.691242582269376,
      "grad_norm": 7.652421229038361,
      "learning_rate": 2.9680881289203874e-06,
      "loss": 0.3431,
      "step": 961
    },
    {
      "epoch": 0.691961877360187,
      "grad_norm": 3.764330521601031,
      "learning_rate": 2.9680182816367006e-06,
      "loss": 0.1311,
      "step": 962
    },
    {
      "epoch": 0.692681172450998,
      "grad_norm": 4.229019754615401,
      "learning_rate": 2.9679483588208955e-06,
      "loss": 0.1396,
      "step": 963
    },
    {
      "epoch": 0.6934004675418091,
      "grad_norm": 5.224071629996026,
      "learning_rate": 2.9678783604765707e-06,
      "loss": 0.1206,
      "step": 964
    },
    {
      "epoch": 0.69411976263262,
      "grad_norm": 4.011431670303693,
      "learning_rate": 2.9678082866073273e-06,
      "loss": 0.0713,
      "step": 965
    },
    {
      "epoch": 0.694839057723431,
      "grad_norm": 6.126105544714853,
      "learning_rate": 2.9677381372167707e-06,
      "loss": 0.1258,
      "step": 966
    },
    {
      "epoch": 0.695558352814242,
      "grad_norm": 7.805957238731541,
      "learning_rate": 2.967667912308511e-06,
      "loss": 0.1286,
      "step": 967
    },
    {
      "epoch": 0.6962776479050531,
      "grad_norm": 6.460562531551328,
      "learning_rate": 2.9675976118861605e-06,
      "loss": 0.0887,
      "step": 968
    },
    {
      "epoch": 0.696996942995864,
      "grad_norm": 8.118452439188117,
      "learning_rate": 2.967527235953337e-06,
      "loss": 0.1177,
      "step": 969
    },
    {
      "epoch": 0.697716238086675,
      "grad_norm": 5.769810171818604,
      "learning_rate": 2.96745678451366e-06,
      "loss": 0.1082,
      "step": 970
    },
    {
      "epoch": 0.6984355331774861,
      "grad_norm": 4.401102495960162,
      "learning_rate": 2.9673862575707564e-06,
      "loss": 0.0795,
      "step": 971
    },
    {
      "epoch": 0.6991548282682971,
      "grad_norm": 8.63568154024474,
      "learning_rate": 2.9673156551282536e-06,
      "loss": 0.127,
      "step": 972
    },
    {
      "epoch": 0.699874123359108,
      "grad_norm": 6.455892407809294,
      "learning_rate": 2.9672449771897842e-06,
      "loss": 0.2006,
      "step": 973
    },
    {
      "epoch": 0.7005934184499191,
      "grad_norm": 3.5765667942005503,
      "learning_rate": 2.967174223758986e-06,
      "loss": 0.1018,
      "step": 974
    },
    {
      "epoch": 0.7013127135407301,
      "grad_norm": 3.334368291701813,
      "learning_rate": 2.9671033948394975e-06,
      "loss": 0.0491,
      "step": 975
    },
    {
      "epoch": 0.7020320086315411,
      "grad_norm": 8.253089659471735,
      "learning_rate": 2.967032490434964e-06,
      "loss": 0.2462,
      "step": 976
    },
    {
      "epoch": 0.7027513037223521,
      "grad_norm": 2.8823983697196263,
      "learning_rate": 2.966961510549034e-06,
      "loss": 0.0637,
      "step": 977
    },
    {
      "epoch": 0.7034705988131631,
      "grad_norm": 34.893804146111194,
      "learning_rate": 2.966890455185359e-06,
      "loss": 0.3012,
      "step": 978
    },
    {
      "epoch": 0.7041898939039741,
      "grad_norm": 19.958986078167154,
      "learning_rate": 2.966819324347595e-06,
      "loss": 0.1786,
      "step": 979
    },
    {
      "epoch": 0.7049091889947852,
      "grad_norm": 8.434388695810744,
      "learning_rate": 2.966748118039402e-06,
      "loss": 0.1356,
      "step": 980
    },
    {
      "epoch": 0.7056284840855961,
      "grad_norm": 7.3825319122220785,
      "learning_rate": 2.966676836264443e-06,
      "loss": 0.2168,
      "step": 981
    },
    {
      "epoch": 0.7063477791764071,
      "grad_norm": 6.90429604963317,
      "learning_rate": 2.9666054790263865e-06,
      "loss": 0.118,
      "step": 982
    },
    {
      "epoch": 0.7070670742672182,
      "grad_norm": 6.50748962531561,
      "learning_rate": 2.9665340463289037e-06,
      "loss": 0.3254,
      "step": 983
    },
    {
      "epoch": 0.7077863693580291,
      "grad_norm": 4.983237971721256,
      "learning_rate": 2.9664625381756693e-06,
      "loss": 0.2475,
      "step": 984
    },
    {
      "epoch": 0.7085056644488401,
      "grad_norm": 9.890964593091509,
      "learning_rate": 2.9663909545703633e-06,
      "loss": 0.226,
      "step": 985
    },
    {
      "epoch": 0.7092249595396511,
      "grad_norm": 4.4264168456726605,
      "learning_rate": 2.966319295516669e-06,
      "loss": 0.1635,
      "step": 986
    },
    {
      "epoch": 0.7099442546304622,
      "grad_norm": 5.838681578080727,
      "learning_rate": 2.9662475610182723e-06,
      "loss": 0.0493,
      "step": 987
    },
    {
      "epoch": 0.7106635497212731,
      "grad_norm": 7.884719276335242,
      "learning_rate": 2.9661757510788646e-06,
      "loss": 0.1754,
      "step": 988
    },
    {
      "epoch": 0.7113828448120841,
      "grad_norm": 6.507672641984555,
      "learning_rate": 2.966103865702141e-06,
      "loss": 0.0645,
      "step": 989
    },
    {
      "epoch": 0.7121021399028952,
      "grad_norm": 7.225249366538003,
      "learning_rate": 2.9660319048917997e-06,
      "loss": 0.1235,
      "step": 990
    },
    {
      "epoch": 0.7128214349937062,
      "grad_norm": 3.7333508759405913,
      "learning_rate": 2.9659598686515434e-06,
      "loss": 0.122,
      "step": 991
    },
    {
      "epoch": 0.7135407300845171,
      "grad_norm": 10.042644778962964,
      "learning_rate": 2.9658877569850785e-06,
      "loss": 0.1271,
      "step": 992
    },
    {
      "epoch": 0.7142600251753282,
      "grad_norm": 7.766648915957607,
      "learning_rate": 2.965815569896115e-06,
      "loss": 0.0961,
      "step": 993
    },
    {
      "epoch": 0.7149793202661392,
      "grad_norm": 5.279417321782498,
      "learning_rate": 2.965743307388368e-06,
      "loss": 0.1893,
      "step": 994
    },
    {
      "epoch": 0.7156986153569502,
      "grad_norm": 7.560644437226907,
      "learning_rate": 2.965670969465554e-06,
      "loss": 0.2037,
      "step": 995
    },
    {
      "epoch": 0.7164179104477612,
      "grad_norm": 6.114245524146761,
      "learning_rate": 2.965598556131396e-06,
      "loss": 0.0396,
      "step": 996
    },
    {
      "epoch": 0.7171372055385722,
      "grad_norm": 6.2985475339173895,
      "learning_rate": 2.9655260673896195e-06,
      "loss": 0.2074,
      "step": 997
    },
    {
      "epoch": 0.7178565006293832,
      "grad_norm": 5.282943824069809,
      "learning_rate": 2.965453503243954e-06,
      "loss": 0.1521,
      "step": 998
    },
    {
      "epoch": 0.7185757957201943,
      "grad_norm": 8.014710357070566,
      "learning_rate": 2.9653808636981336e-06,
      "loss": 0.2785,
      "step": 999
    },
    {
      "epoch": 0.7192950908110052,
      "grad_norm": 8.650382345832208,
      "learning_rate": 2.9653081487558953e-06,
      "loss": 0.3086,
      "step": 1000
    },
    {
      "epoch": 0.7200143859018162,
      "grad_norm": 5.003521045734504,
      "learning_rate": 2.96523535842098e-06,
      "loss": 0.0893,
      "step": 1001
    },
    {
      "epoch": 0.7207336809926272,
      "grad_norm": 5.751385819014117,
      "learning_rate": 2.965162492697134e-06,
      "loss": 0.2563,
      "step": 1002
    },
    {
      "epoch": 0.7214529760834383,
      "grad_norm": 7.441038980787459,
      "learning_rate": 2.9650895515881057e-06,
      "loss": 0.1702,
      "step": 1003
    },
    {
      "epoch": 0.7221722711742492,
      "grad_norm": 4.836643831737609,
      "learning_rate": 2.9650165350976476e-06,
      "loss": 0.1199,
      "step": 1004
    },
    {
      "epoch": 0.7228915662650602,
      "grad_norm": 4.146165859090649,
      "learning_rate": 2.9649434432295175e-06,
      "loss": 0.0687,
      "step": 1005
    },
    {
      "epoch": 0.7236108613558713,
      "grad_norm": 7.924222140472695,
      "learning_rate": 2.9648702759874753e-06,
      "loss": 0.1596,
      "step": 1006
    },
    {
      "epoch": 0.7243301564466823,
      "grad_norm": 5.4462806625069975,
      "learning_rate": 2.964797033375286e-06,
      "loss": 0.2539,
      "step": 1007
    },
    {
      "epoch": 0.7250494515374932,
      "grad_norm": 6.551134818408238,
      "learning_rate": 2.9647237153967175e-06,
      "loss": 0.1955,
      "step": 1008
    },
    {
      "epoch": 0.7257687466283043,
      "grad_norm": 8.860054031426118,
      "learning_rate": 2.964650322055543e-06,
      "loss": 0.3214,
      "step": 1009
    },
    {
      "epoch": 0.7264880417191153,
      "grad_norm": 6.168108824487009,
      "learning_rate": 2.964576853355539e-06,
      "loss": 0.0882,
      "step": 1010
    },
    {
      "epoch": 0.7272073368099262,
      "grad_norm": 5.965482796743988,
      "learning_rate": 2.964503309300484e-06,
      "loss": 0.1507,
      "step": 1011
    },
    {
      "epoch": 0.7279266319007373,
      "grad_norm": 6.853415871889261,
      "learning_rate": 2.964429689894163e-06,
      "loss": 0.1678,
      "step": 1012
    },
    {
      "epoch": 0.7286459269915483,
      "grad_norm": 7.8826983989593975,
      "learning_rate": 2.964355995140364e-06,
      "loss": 0.2033,
      "step": 1013
    },
    {
      "epoch": 0.7293652220823593,
      "grad_norm": 2.310130006174262,
      "learning_rate": 2.9642822250428784e-06,
      "loss": 0.0551,
      "step": 1014
    },
    {
      "epoch": 0.7300845171731702,
      "grad_norm": 6.319449571703531,
      "learning_rate": 2.964208379605502e-06,
      "loss": 0.3185,
      "step": 1015
    },
    {
      "epoch": 0.7308038122639813,
      "grad_norm": 3.9879871398239364,
      "learning_rate": 2.9641344588320337e-06,
      "loss": 0.132,
      "step": 1016
    },
    {
      "epoch": 0.7315231073547923,
      "grad_norm": 4.5887037458182105,
      "learning_rate": 2.964060462726278e-06,
      "loss": 0.0406,
      "step": 1017
    },
    {
      "epoch": 0.7322424024456033,
      "grad_norm": 9.666589232054042,
      "learning_rate": 2.9639863912920406e-06,
      "loss": 0.1545,
      "step": 1018
    },
    {
      "epoch": 0.7329616975364143,
      "grad_norm": 11.836243335873432,
      "learning_rate": 2.9639122445331336e-06,
      "loss": 0.135,
      "step": 1019
    },
    {
      "epoch": 0.7336809926272253,
      "grad_norm": 5.42392486236328,
      "learning_rate": 2.963838022453372e-06,
      "loss": 0.164,
      "step": 1020
    },
    {
      "epoch": 0.7344002877180363,
      "grad_norm": 5.274419931289659,
      "learning_rate": 2.9637637250565745e-06,
      "loss": 0.2822,
      "step": 1021
    },
    {
      "epoch": 0.7351195828088474,
      "grad_norm": 9.797947072139278,
      "learning_rate": 2.963689352346564e-06,
      "loss": 0.5928,
      "step": 1022
    },
    {
      "epoch": 0.7358388778996583,
      "grad_norm": 8.378388821768572,
      "learning_rate": 2.9636149043271666e-06,
      "loss": 0.2254,
      "step": 1023
    },
    {
      "epoch": 0.7365581729904693,
      "grad_norm": 8.304864011786217,
      "learning_rate": 2.963540381002213e-06,
      "loss": 0.2131,
      "step": 1024
    },
    {
      "epoch": 0.7372774680812804,
      "grad_norm": 5.377043117794858,
      "learning_rate": 2.9634657823755374e-06,
      "loss": 0.1024,
      "step": 1025
    },
    {
      "epoch": 0.7379967631720914,
      "grad_norm": 6.220054664975505,
      "learning_rate": 2.9633911084509788e-06,
      "loss": 0.0535,
      "step": 1026
    },
    {
      "epoch": 0.7387160582629023,
      "grad_norm": 0.9637168646863399,
      "learning_rate": 2.9633163592323786e-06,
      "loss": 0.0065,
      "step": 1027
    },
    {
      "epoch": 0.7394353533537134,
      "grad_norm": 8.054302386051065,
      "learning_rate": 2.9632415347235825e-06,
      "loss": 0.1028,
      "step": 1028
    },
    {
      "epoch": 0.7401546484445244,
      "grad_norm": 3.0726704388718353,
      "learning_rate": 2.9631666349284414e-06,
      "loss": 0.0907,
      "step": 1029
    },
    {
      "epoch": 0.7408739435353354,
      "grad_norm": 8.51683505148813,
      "learning_rate": 2.963091659850808e-06,
      "loss": 0.2076,
      "step": 1030
    },
    {
      "epoch": 0.7415932386261463,
      "grad_norm": 6.667369843402183,
      "learning_rate": 2.963016609494541e-06,
      "loss": 0.1198,
      "step": 1031
    },
    {
      "epoch": 0.7423125337169574,
      "grad_norm": 5.317912112099469,
      "learning_rate": 2.9629414838635004e-06,
      "loss": 0.0828,
      "step": 1032
    },
    {
      "epoch": 0.7430318288077684,
      "grad_norm": 2.645375849464506,
      "learning_rate": 2.962866282961553e-06,
      "loss": 0.0802,
      "step": 1033
    },
    {
      "epoch": 0.7437511238985794,
      "grad_norm": 8.688885817653823,
      "learning_rate": 2.9627910067925666e-06,
      "loss": 0.1758,
      "step": 1034
    },
    {
      "epoch": 0.7444704189893904,
      "grad_norm": 8.254817497594031,
      "learning_rate": 2.9627156553604157e-06,
      "loss": 0.1529,
      "step": 1035
    },
    {
      "epoch": 0.7451897140802014,
      "grad_norm": 6.204186959347255,
      "learning_rate": 2.9626402286689765e-06,
      "loss": 0.1675,
      "step": 1036
    },
    {
      "epoch": 0.7459090091710124,
      "grad_norm": 4.408973375756506,
      "learning_rate": 2.9625647267221294e-06,
      "loss": 0.0916,
      "step": 1037
    },
    {
      "epoch": 0.7466283042618234,
      "grad_norm": 6.056189613910105,
      "learning_rate": 2.9624891495237606e-06,
      "loss": 0.2593,
      "step": 1038
    },
    {
      "epoch": 0.7473475993526344,
      "grad_norm": 6.390137114374384,
      "learning_rate": 2.962413497077757e-06,
      "loss": 0.2372,
      "step": 1039
    },
    {
      "epoch": 0.7480668944434454,
      "grad_norm": 6.641398191765912,
      "learning_rate": 2.9623377693880123e-06,
      "loss": 0.1913,
      "step": 1040
    },
    {
      "epoch": 0.7487861895342565,
      "grad_norm": 6.969302015391716,
      "learning_rate": 2.962261966458422e-06,
      "loss": 0.1709,
      "step": 1041
    },
    {
      "epoch": 0.7495054846250674,
      "grad_norm": 4.949634384482189,
      "learning_rate": 2.9621860882928868e-06,
      "loss": 0.1111,
      "step": 1042
    },
    {
      "epoch": 0.7502247797158784,
      "grad_norm": 8.44134191394853,
      "learning_rate": 2.9621101348953105e-06,
      "loss": 0.1495,
      "step": 1043
    },
    {
      "epoch": 0.7509440748066895,
      "grad_norm": 8.032308034360867,
      "learning_rate": 2.962034106269601e-06,
      "loss": 0.1796,
      "step": 1044
    },
    {
      "epoch": 0.7516633698975005,
      "grad_norm": 5.1075318898939015,
      "learning_rate": 2.96195800241967e-06,
      "loss": 0.1875,
      "step": 1045
    },
    {
      "epoch": 0.7523826649883114,
      "grad_norm": 9.949516066867346,
      "learning_rate": 2.9618818233494335e-06,
      "loss": 0.1622,
      "step": 1046
    },
    {
      "epoch": 0.7531019600791224,
      "grad_norm": 7.077940367846719,
      "learning_rate": 2.9618055690628112e-06,
      "loss": 0.101,
      "step": 1047
    },
    {
      "epoch": 0.7538212551699335,
      "grad_norm": 7.770866220795333,
      "learning_rate": 2.9617292395637265e-06,
      "loss": 0.3442,
      "step": 1048
    },
    {
      "epoch": 0.7545405502607445,
      "grad_norm": 4.687233301268084,
      "learning_rate": 2.9616528348561057e-06,
      "loss": 0.2032,
      "step": 1049
    },
    {
      "epoch": 0.7552598453515554,
      "grad_norm": 5.3693367091754665,
      "learning_rate": 2.961576354943881e-06,
      "loss": 0.2438,
      "step": 1050
    },
    {
      "epoch": 0.7559791404423665,
      "grad_norm": 5.873247532550017,
      "learning_rate": 2.9614997998309878e-06,
      "loss": 0.2456,
      "step": 1051
    },
    {
      "epoch": 0.7566984355331775,
      "grad_norm": 6.4936719305034405,
      "learning_rate": 2.961423169521363e-06,
      "loss": 0.3128,
      "step": 1052
    },
    {
      "epoch": 0.7574177306239885,
      "grad_norm": 4.460512181817813,
      "learning_rate": 2.961346464018952e-06,
      "loss": 0.094,
      "step": 1053
    },
    {
      "epoch": 0.7581370257147995,
      "grad_norm": 6.5616676589913006,
      "learning_rate": 2.961269683327699e-06,
      "loss": 0.1021,
      "step": 1054
    },
    {
      "epoch": 0.7588563208056105,
      "grad_norm": 7.666176125891536,
      "learning_rate": 2.9611928274515567e-06,
      "loss": 0.1959,
      "step": 1055
    },
    {
      "epoch": 0.7595756158964215,
      "grad_norm": 4.8175082478123485,
      "learning_rate": 2.9611158963944776e-06,
      "loss": 0.0594,
      "step": 1056
    },
    {
      "epoch": 0.7602949109872326,
      "grad_norm": 4.1431375213170405,
      "learning_rate": 2.961038890160421e-06,
      "loss": 0.1013,
      "step": 1057
    },
    {
      "epoch": 0.7610142060780435,
      "grad_norm": 4.763195705949494,
      "learning_rate": 2.9609618087533486e-06,
      "loss": 0.2288,
      "step": 1058
    },
    {
      "epoch": 0.7617335011688545,
      "grad_norm": 5.651407027708578,
      "learning_rate": 2.9608846521772265e-06,
      "loss": 0.1345,
      "step": 1059
    },
    {
      "epoch": 0.7624527962596656,
      "grad_norm": 7.988657506027217,
      "learning_rate": 2.9608074204360245e-06,
      "loss": 0.0894,
      "step": 1060
    },
    {
      "epoch": 0.7631720913504765,
      "grad_norm": 4.799091978005488,
      "learning_rate": 2.9607301135337165e-06,
      "loss": 0.0689,
      "step": 1061
    },
    {
      "epoch": 0.7638913864412875,
      "grad_norm": 7.254078601061779,
      "learning_rate": 2.96065273147428e-06,
      "loss": 0.1273,
      "step": 1062
    },
    {
      "epoch": 0.7646106815320985,
      "grad_norm": 5.505255175721506,
      "learning_rate": 2.9605752742616963e-06,
      "loss": 0.1091,
      "step": 1063
    },
    {
      "epoch": 0.7653299766229096,
      "grad_norm": 9.390324692548925,
      "learning_rate": 2.960497741899951e-06,
      "loss": 0.2214,
      "step": 1064
    },
    {
      "epoch": 0.7660492717137205,
      "grad_norm": 8.265178676861693,
      "learning_rate": 2.9604201343930326e-06,
      "loss": 0.2469,
      "step": 1065
    },
    {
      "epoch": 0.7667685668045315,
      "grad_norm": 5.013709944651262,
      "learning_rate": 2.9603424517449343e-06,
      "loss": 0.0646,
      "step": 1066
    },
    {
      "epoch": 0.7674878618953426,
      "grad_norm": 8.230599844197473,
      "learning_rate": 2.960264693959654e-06,
      "loss": 0.2249,
      "step": 1067
    },
    {
      "epoch": 0.7682071569861536,
      "grad_norm": 4.821036551476318,
      "learning_rate": 2.9601868610411914e-06,
      "loss": 0.1896,
      "step": 1068
    },
    {
      "epoch": 0.7689264520769645,
      "grad_norm": 6.817500977639897,
      "learning_rate": 2.960108952993552e-06,
      "loss": 0.1728,
      "step": 1069
    },
    {
      "epoch": 0.7696457471677756,
      "grad_norm": 5.993676325974895,
      "learning_rate": 2.960030969820743e-06,
      "loss": 0.1427,
      "step": 1070
    },
    {
      "epoch": 0.7703650422585866,
      "grad_norm": 5.1225152323778715,
      "learning_rate": 2.9599529115267787e-06,
      "loss": 0.2073,
      "step": 1071
    },
    {
      "epoch": 0.7710843373493976,
      "grad_norm": 4.727738733597606,
      "learning_rate": 2.9598747781156734e-06,
      "loss": 0.1028,
      "step": 1072
    },
    {
      "epoch": 0.7718036324402086,
      "grad_norm": 4.8080381627304805,
      "learning_rate": 2.9597965695914486e-06,
      "loss": 0.1531,
      "step": 1073
    },
    {
      "epoch": 0.7725229275310196,
      "grad_norm": 7.1750051303221545,
      "learning_rate": 2.959718285958127e-06,
      "loss": 0.0973,
      "step": 1074
    },
    {
      "epoch": 0.7732422226218306,
      "grad_norm": 7.560782095923482,
      "learning_rate": 2.959639927219738e-06,
      "loss": 0.3099,
      "step": 1075
    },
    {
      "epoch": 0.7739615177126417,
      "grad_norm": 6.081471630530151,
      "learning_rate": 2.959561493380312e-06,
      "loss": 0.2704,
      "step": 1076
    },
    {
      "epoch": 0.7746808128034526,
      "grad_norm": 4.5573471916570005,
      "learning_rate": 2.959482984443885e-06,
      "loss": 0.1634,
      "step": 1077
    },
    {
      "epoch": 0.7754001078942636,
      "grad_norm": 8.024352829564638,
      "learning_rate": 2.959404400414496e-06,
      "loss": 0.121,
      "step": 1078
    },
    {
      "epoch": 0.7761194029850746,
      "grad_norm": 6.826364976830265,
      "learning_rate": 2.959325741296189e-06,
      "loss": 0.213,
      "step": 1079
    },
    {
      "epoch": 0.7768386980758857,
      "grad_norm": 4.616172305121968,
      "learning_rate": 2.959247007093011e-06,
      "loss": 0.0142,
      "step": 1080
    },
    {
      "epoch": 0.7775579931666966,
      "grad_norm": 3.903262585727129,
      "learning_rate": 2.9591681978090127e-06,
      "loss": 0.1081,
      "step": 1081
    },
    {
      "epoch": 0.7782772882575076,
      "grad_norm": 7.339701277038209,
      "learning_rate": 2.9590893134482493e-06,
      "loss": 0.1958,
      "step": 1082
    },
    {
      "epoch": 0.7789965833483187,
      "grad_norm": 7.0135713771378425,
      "learning_rate": 2.959010354014779e-06,
      "loss": 0.0795,
      "step": 1083
    },
    {
      "epoch": 0.7797158784391297,
      "grad_norm": 3.777432169863555,
      "learning_rate": 2.9589313195126654e-06,
      "loss": 0.1546,
      "step": 1084
    },
    {
      "epoch": 0.7804351735299406,
      "grad_norm": 4.856435890238801,
      "learning_rate": 2.958852209945974e-06,
      "loss": 0.1952,
      "step": 1085
    },
    {
      "epoch": 0.7811544686207517,
      "grad_norm": 8.20527872966151,
      "learning_rate": 2.958773025318775e-06,
      "loss": 0.0835,
      "step": 1086
    },
    {
      "epoch": 0.7818737637115627,
      "grad_norm": 5.514808434154136,
      "learning_rate": 2.9586937656351437e-06,
      "loss": 0.1738,
      "step": 1087
    },
    {
      "epoch": 0.7825930588023736,
      "grad_norm": 4.659218147016898,
      "learning_rate": 2.958614430899157e-06,
      "loss": 0.0753,
      "step": 1088
    },
    {
      "epoch": 0.7833123538931847,
      "grad_norm": 2.847833195278624,
      "learning_rate": 2.958535021114898e-06,
      "loss": 0.0229,
      "step": 1089
    },
    {
      "epoch": 0.7840316489839957,
      "grad_norm": 4.964037366847555,
      "learning_rate": 2.958455536286451e-06,
      "loss": 0.1589,
      "step": 1090
    },
    {
      "epoch": 0.7847509440748067,
      "grad_norm": 4.055532056707014,
      "learning_rate": 2.958375976417907e-06,
      "loss": 0.0392,
      "step": 1091
    },
    {
      "epoch": 0.7854702391656176,
      "grad_norm": 7.985326591241292,
      "learning_rate": 2.958296341513359e-06,
      "loss": 0.1535,
      "step": 1092
    },
    {
      "epoch": 0.7861895342564287,
      "grad_norm": 3.0336896592445393,
      "learning_rate": 2.9582166315769037e-06,
      "loss": 0.038,
      "step": 1093
    },
    {
      "epoch": 0.7869088293472397,
      "grad_norm": 4.4161714083243355,
      "learning_rate": 2.9581368466126434e-06,
      "loss": 0.0506,
      "step": 1094
    },
    {
      "epoch": 0.7876281244380507,
      "grad_norm": 6.301016191414249,
      "learning_rate": 2.9580569866246827e-06,
      "loss": 0.1047,
      "step": 1095
    },
    {
      "epoch": 0.7883474195288617,
      "grad_norm": 5.4218925181388835,
      "learning_rate": 2.95797705161713e-06,
      "loss": 0.1682,
      "step": 1096
    },
    {
      "epoch": 0.7890667146196727,
      "grad_norm": 5.861239821465975,
      "learning_rate": 2.9578970415940986e-06,
      "loss": 0.1167,
      "step": 1097
    },
    {
      "epoch": 0.7897860097104837,
      "grad_norm": 7.475721342617539,
      "learning_rate": 2.9578169565597058e-06,
      "loss": 0.0846,
      "step": 1098
    },
    {
      "epoch": 0.7905053048012948,
      "grad_norm": 7.172766655699195,
      "learning_rate": 2.9577367965180705e-06,
      "loss": 0.0873,
      "step": 1099
    },
    {
      "epoch": 0.7912245998921057,
      "grad_norm": 8.105874256968812,
      "learning_rate": 2.957656561473319e-06,
      "loss": 0.0662,
      "step": 1100
    },
    {
      "epoch": 0.7919438949829167,
      "grad_norm": 6.239076982398492,
      "learning_rate": 2.957576251429578e-06,
      "loss": 0.1863,
      "step": 1101
    },
    {
      "epoch": 0.7926631900737278,
      "grad_norm": 5.69335042861376,
      "learning_rate": 2.9574958663909803e-06,
      "loss": 0.1408,
      "step": 1102
    },
    {
      "epoch": 0.7933824851645388,
      "grad_norm": 5.003234054276349,
      "learning_rate": 2.957415406361662e-06,
      "loss": 0.028,
      "step": 1103
    },
    {
      "epoch": 0.7941017802553497,
      "grad_norm": 4.592670187945757,
      "learning_rate": 2.9573348713457623e-06,
      "loss": 0.2002,
      "step": 1104
    },
    {
      "epoch": 0.7948210753461608,
      "grad_norm": 6.293272077818915,
      "learning_rate": 2.9572542613474254e-06,
      "loss": 0.2168,
      "step": 1105
    },
    {
      "epoch": 0.7955403704369718,
      "grad_norm": 12.009341416844594,
      "learning_rate": 2.9571735763707985e-06,
      "loss": 0.3254,
      "step": 1106
    },
    {
      "epoch": 0.7962596655277828,
      "grad_norm": 7.263073206419407,
      "learning_rate": 2.9570928164200333e-06,
      "loss": 0.0756,
      "step": 1107
    },
    {
      "epoch": 0.7969789606185937,
      "grad_norm": 6.098330065257,
      "learning_rate": 2.9570119814992844e-06,
      "loss": 0.3136,
      "step": 1108
    },
    {
      "epoch": 0.7976982557094048,
      "grad_norm": 7.046862916564739,
      "learning_rate": 2.9569310716127116e-06,
      "loss": 0.2601,
      "step": 1109
    },
    {
      "epoch": 0.7984175508002158,
      "grad_norm": 5.88888767432492,
      "learning_rate": 2.9568500867644777e-06,
      "loss": 0.2498,
      "step": 1110
    },
    {
      "epoch": 0.7991368458910268,
      "grad_norm": 6.403418222612225,
      "learning_rate": 2.9567690269587493e-06,
      "loss": 0.161,
      "step": 1111
    },
    {
      "epoch": 0.7998561409818378,
      "grad_norm": 7.526621169944265,
      "learning_rate": 2.956687892199697e-06,
      "loss": 0.2115,
      "step": 1112
    },
    {
      "epoch": 0.8005754360726488,
      "grad_norm": 8.018745664550192,
      "learning_rate": 2.956606682491496e-06,
      "loss": 0.1093,
      "step": 1113
    },
    {
      "epoch": 0.8012947311634598,
      "grad_norm": 6.200905213411397,
      "learning_rate": 2.956525397838324e-06,
      "loss": 0.1934,
      "step": 1114
    },
    {
      "epoch": 0.8020140262542708,
      "grad_norm": 7.932886285925514,
      "learning_rate": 2.9564440382443636e-06,
      "loss": 0.2301,
      "step": 1115
    },
    {
      "epoch": 0.8027333213450818,
      "grad_norm": 4.949344856984922,
      "learning_rate": 2.9563626037138003e-06,
      "loss": 0.3011,
      "step": 1116
    },
    {
      "epoch": 0.8034526164358928,
      "grad_norm": 8.274999836950926,
      "learning_rate": 2.956281094250825e-06,
      "loss": 0.3443,
      "step": 1117
    },
    {
      "epoch": 0.8041719115267039,
      "grad_norm": 5.756304489457235,
      "learning_rate": 2.9561995098596304e-06,
      "loss": 0.1179,
      "step": 1118
    },
    {
      "epoch": 0.8048912066175148,
      "grad_norm": 6.890620747454537,
      "learning_rate": 2.956117850544415e-06,
      "loss": 0.1811,
      "step": 1119
    },
    {
      "epoch": 0.8056105017083258,
      "grad_norm": 9.648966240120986,
      "learning_rate": 2.95603611630938e-06,
      "loss": 0.3437,
      "step": 1120
    },
    {
      "epoch": 0.8063297967991369,
      "grad_norm": 6.604232863540325,
      "learning_rate": 2.955954307158731e-06,
      "loss": 0.1327,
      "step": 1121
    },
    {
      "epoch": 0.8070490918899479,
      "grad_norm": 4.577272025294506,
      "learning_rate": 2.955872423096677e-06,
      "loss": 0.0532,
      "step": 1122
    },
    {
      "epoch": 0.8077683869807588,
      "grad_norm": 5.682910705275785,
      "learning_rate": 2.955790464127431e-06,
      "loss": 0.2226,
      "step": 1123
    },
    {
      "epoch": 0.8084876820715698,
      "grad_norm": 6.64412933635135,
      "learning_rate": 2.9557084302552104e-06,
      "loss": 0.1609,
      "step": 1124
    },
    {
      "epoch": 0.8092069771623809,
      "grad_norm": 6.443648576287447,
      "learning_rate": 2.955626321484235e-06,
      "loss": 0.0856,
      "step": 1125
    },
    {
      "epoch": 0.8099262722531919,
      "grad_norm": 7.326989682730623,
      "learning_rate": 2.9555441378187304e-06,
      "loss": 0.1617,
      "step": 1126
    },
    {
      "epoch": 0.8106455673440028,
      "grad_norm": 7.9020294740056585,
      "learning_rate": 2.9554618792629248e-06,
      "loss": 0.4006,
      "step": 1127
    },
    {
      "epoch": 0.8113648624348139,
      "grad_norm": 4.302728084841077,
      "learning_rate": 2.9553795458210506e-06,
      "loss": 0.1214,
      "step": 1128
    },
    {
      "epoch": 0.8120841575256249,
      "grad_norm": 7.649938144231672,
      "learning_rate": 2.9552971374973443e-06,
      "loss": 0.1833,
      "step": 1129
    },
    {
      "epoch": 0.8128034526164359,
      "grad_norm": 6.8285741943298675,
      "learning_rate": 2.955214654296045e-06,
      "loss": 0.2469,
      "step": 1130
    },
    {
      "epoch": 0.8135227477072469,
      "grad_norm": 8.6710081624358,
      "learning_rate": 2.955132096221397e-06,
      "loss": 0.0963,
      "step": 1131
    },
    {
      "epoch": 0.8142420427980579,
      "grad_norm": 8.997418489820435,
      "learning_rate": 2.955049463277649e-06,
      "loss": 0.2333,
      "step": 1132
    },
    {
      "epoch": 0.8149613378888689,
      "grad_norm": 7.300476127066846,
      "learning_rate": 2.9549667554690507e-06,
      "loss": 0.1037,
      "step": 1133
    },
    {
      "epoch": 0.81568063297968,
      "grad_norm": 6.147210142800407,
      "learning_rate": 2.95488397279986e-06,
      "loss": 0.1754,
      "step": 1134
    },
    {
      "epoch": 0.8163999280704909,
      "grad_norm": 6.078805507443698,
      "learning_rate": 2.9548011152743345e-06,
      "loss": 0.2416,
      "step": 1135
    },
    {
      "epoch": 0.8171192231613019,
      "grad_norm": 7.486754463637567,
      "learning_rate": 2.954718182896737e-06,
      "loss": 0.3225,
      "step": 1136
    },
    {
      "epoch": 0.817838518252113,
      "grad_norm": 4.44992158630488,
      "learning_rate": 2.954635175671337e-06,
      "loss": 0.0599,
      "step": 1137
    },
    {
      "epoch": 0.818557813342924,
      "grad_norm": 9.187427134684947,
      "learning_rate": 2.954552093602402e-06,
      "loss": 0.2059,
      "step": 1138
    },
    {
      "epoch": 0.8192771084337349,
      "grad_norm": 3.794189057153781,
      "learning_rate": 2.9544689366942094e-06,
      "loss": 0.0773,
      "step": 1139
    },
    {
      "epoch": 0.819996403524546,
      "grad_norm": 7.126594883887073,
      "learning_rate": 2.9543857049510366e-06,
      "loss": 0.1345,
      "step": 1140
    },
    {
      "epoch": 0.820715698615357,
      "grad_norm": 6.5230481639395315,
      "learning_rate": 2.954302398377166e-06,
      "loss": 0.2994,
      "step": 1141
    },
    {
      "epoch": 0.8214349937061679,
      "grad_norm": 5.126007158127512,
      "learning_rate": 2.954219016976884e-06,
      "loss": 0.162,
      "step": 1142
    },
    {
      "epoch": 0.8221542887969789,
      "grad_norm": 3.7723544701317153,
      "learning_rate": 2.9541355607544807e-06,
      "loss": 0.19,
      "step": 1143
    },
    {
      "epoch": 0.82287358388779,
      "grad_norm": 6.178592525153852,
      "learning_rate": 2.9540520297142505e-06,
      "loss": 0.0789,
      "step": 1144
    },
    {
      "epoch": 0.823592878978601,
      "grad_norm": 3.672396832378457,
      "learning_rate": 2.9539684238604905e-06,
      "loss": 0.0887,
      "step": 1145
    },
    {
      "epoch": 0.8243121740694119,
      "grad_norm": 6.795273015308563,
      "learning_rate": 2.953884743197503e-06,
      "loss": 0.1575,
      "step": 1146
    },
    {
      "epoch": 0.825031469160223,
      "grad_norm": 6.8400877172759715,
      "learning_rate": 2.9538009877295934e-06,
      "loss": 0.2467,
      "step": 1147
    },
    {
      "epoch": 0.825750764251034,
      "grad_norm": 7.9529661138274435,
      "learning_rate": 2.9537171574610706e-06,
      "loss": 0.2735,
      "step": 1148
    },
    {
      "epoch": 0.826470059341845,
      "grad_norm": 6.646315973163267,
      "learning_rate": 2.9536332523962483e-06,
      "loss": 0.2352,
      "step": 1149
    },
    {
      "epoch": 0.827189354432656,
      "grad_norm": 5.4355372209105095,
      "learning_rate": 2.953549272539443e-06,
      "loss": 0.2303,
      "step": 1150
    },
    {
      "epoch": 0.827908649523467,
      "grad_norm": 4.649327710734051,
      "learning_rate": 2.9534652178949764e-06,
      "loss": 0.1088,
      "step": 1151
    },
    {
      "epoch": 0.828627944614278,
      "grad_norm": 7.301757477076145,
      "learning_rate": 2.953381088467173e-06,
      "loss": 0.1146,
      "step": 1152
    },
    {
      "epoch": 0.8293472397050891,
      "grad_norm": 6.484628002096242,
      "learning_rate": 2.953296884260361e-06,
      "loss": 0.1852,
      "step": 1153
    },
    {
      "epoch": 0.8300665347959,
      "grad_norm": 6.809996187292308,
      "learning_rate": 2.953212605278873e-06,
      "loss": 0.051,
      "step": 1154
    },
    {
      "epoch": 0.830785829886711,
      "grad_norm": 4.737229044265182,
      "learning_rate": 2.9531282515270457e-06,
      "loss": 0.1251,
      "step": 1155
    },
    {
      "epoch": 0.831505124977522,
      "grad_norm": 6.21572016099917,
      "learning_rate": 2.953043823009219e-06,
      "loss": 0.1774,
      "step": 1156
    },
    {
      "epoch": 0.8322244200683331,
      "grad_norm": 3.982063537099724,
      "learning_rate": 2.952959319729737e-06,
      "loss": 0.1102,
      "step": 1157
    },
    {
      "epoch": 0.832943715159144,
      "grad_norm": 4.725101822974682,
      "learning_rate": 2.9528747416929465e-06,
      "loss": 0.105,
      "step": 1158
    },
    {
      "epoch": 0.833663010249955,
      "grad_norm": 9.06050821331901,
      "learning_rate": 2.952790088903201e-06,
      "loss": 0.3101,
      "step": 1159
    },
    {
      "epoch": 0.8343823053407661,
      "grad_norm": 3.292327735901539,
      "learning_rate": 2.9527053613648548e-06,
      "loss": 0.097,
      "step": 1160
    },
    {
      "epoch": 0.8351016004315771,
      "grad_norm": 5.202336927832246,
      "learning_rate": 2.952620559082268e-06,
      "loss": 0.2143,
      "step": 1161
    },
    {
      "epoch": 0.835820895522388,
      "grad_norm": 6.689117450786752,
      "learning_rate": 2.952535682059803e-06,
      "loss": 0.1148,
      "step": 1162
    },
    {
      "epoch": 0.8365401906131991,
      "grad_norm": 3.639434170382141,
      "learning_rate": 2.9524507303018274e-06,
      "loss": 0.1436,
      "step": 1163
    },
    {
      "epoch": 0.8372594857040101,
      "grad_norm": 5.337383021001455,
      "learning_rate": 2.9523657038127117e-06,
      "loss": 0.1328,
      "step": 1164
    },
    {
      "epoch": 0.837978780794821,
      "grad_norm": 10.150920536661177,
      "learning_rate": 2.9522806025968315e-06,
      "loss": 0.1365,
      "step": 1165
    },
    {
      "epoch": 0.838698075885632,
      "grad_norm": 7.101209791133696,
      "learning_rate": 2.952195426658565e-06,
      "loss": 0.1619,
      "step": 1166
    },
    {
      "epoch": 0.8394173709764431,
      "grad_norm": 7.04230801618488,
      "learning_rate": 2.9521101760022943e-06,
      "loss": 0.2199,
      "step": 1167
    },
    {
      "epoch": 0.8401366660672541,
      "grad_norm": 9.453457447970827,
      "learning_rate": 2.952024850632406e-06,
      "loss": 0.3058,
      "step": 1168
    },
    {
      "epoch": 0.840855961158065,
      "grad_norm": 4.428357603479249,
      "learning_rate": 2.95193945055329e-06,
      "loss": 0.0438,
      "step": 1169
    },
    {
      "epoch": 0.8415752562488761,
      "grad_norm": 4.189763361018842,
      "learning_rate": 2.9518539757693406e-06,
      "loss": 0.1537,
      "step": 1170
    },
    {
      "epoch": 0.8422945513396871,
      "grad_norm": 3.3079856837768338,
      "learning_rate": 2.9517684262849557e-06,
      "loss": 0.0736,
      "step": 1171
    },
    {
      "epoch": 0.8430138464304981,
      "grad_norm": 2.643169964527249,
      "learning_rate": 2.9516828021045363e-06,
      "loss": 0.0104,
      "step": 1172
    },
    {
      "epoch": 0.8437331415213091,
      "grad_norm": 6.765432324989906,
      "learning_rate": 2.951597103232489e-06,
      "loss": 0.131,
      "step": 1173
    },
    {
      "epoch": 0.8444524366121201,
      "grad_norm": 3.418371881465293,
      "learning_rate": 2.951511329673222e-06,
      "loss": 0.0769,
      "step": 1174
    },
    {
      "epoch": 0.8451717317029311,
      "grad_norm": 4.966100371899219,
      "learning_rate": 2.9514254814311493e-06,
      "loss": 0.1442,
      "step": 1175
    },
    {
      "epoch": 0.8458910267937422,
      "grad_norm": 5.023802103418847,
      "learning_rate": 2.9513395585106873e-06,
      "loss": 0.0917,
      "step": 1176
    },
    {
      "epoch": 0.8466103218845531,
      "grad_norm": 4.664124893028925,
      "learning_rate": 2.9512535609162577e-06,
      "loss": 0.1302,
      "step": 1177
    },
    {
      "epoch": 0.8473296169753641,
      "grad_norm": 6.295702829080911,
      "learning_rate": 2.951167488652285e-06,
      "loss": 0.0948,
      "step": 1178
    },
    {
      "epoch": 0.8480489120661752,
      "grad_norm": 7.916290328605307,
      "learning_rate": 2.951081341723197e-06,
      "loss": 0.1991,
      "step": 1179
    },
    {
      "epoch": 0.8487682071569862,
      "grad_norm": 7.6712222152259715,
      "learning_rate": 2.950995120133427e-06,
      "loss": 0.1496,
      "step": 1180
    },
    {
      "epoch": 0.8494875022477971,
      "grad_norm": 9.276953839431325,
      "learning_rate": 2.950908823887411e-06,
      "loss": 0.2795,
      "step": 1181
    },
    {
      "epoch": 0.8502067973386082,
      "grad_norm": 1.9217269220028907,
      "learning_rate": 2.950822452989589e-06,
      "loss": 0.0253,
      "step": 1182
    },
    {
      "epoch": 0.8509260924294192,
      "grad_norm": 8.14480139577424,
      "learning_rate": 2.950736007444404e-06,
      "loss": 0.1772,
      "step": 1183
    },
    {
      "epoch": 0.8516453875202302,
      "grad_norm": 8.885296371982806,
      "learning_rate": 2.950649487256306e-06,
      "loss": 0.114,
      "step": 1184
    },
    {
      "epoch": 0.8523646826110411,
      "grad_norm": 8.971215417365684,
      "learning_rate": 2.950562892429745e-06,
      "loss": 0.0933,
      "step": 1185
    },
    {
      "epoch": 0.8530839777018522,
      "grad_norm": 2.573547917683082,
      "learning_rate": 2.9504762229691767e-06,
      "loss": 0.0371,
      "step": 1186
    },
    {
      "epoch": 0.8538032727926632,
      "grad_norm": 3.9643645332769712,
      "learning_rate": 2.9503894788790604e-06,
      "loss": 0.0691,
      "step": 1187
    },
    {
      "epoch": 0.8545225678834742,
      "grad_norm": 5.7179828786626175,
      "learning_rate": 2.9503026601638592e-06,
      "loss": 0.1094,
      "step": 1188
    },
    {
      "epoch": 0.8552418629742852,
      "grad_norm": 1.2052738335468935,
      "learning_rate": 2.9502157668280407e-06,
      "loss": 0.003,
      "step": 1189
    },
    {
      "epoch": 0.8559611580650962,
      "grad_norm": 3.6236871979980614,
      "learning_rate": 2.9501287988760747e-06,
      "loss": 0.0241,
      "step": 1190
    },
    {
      "epoch": 0.8566804531559072,
      "grad_norm": 7.552550722934375,
      "learning_rate": 2.950041756312437e-06,
      "loss": 0.2191,
      "step": 1191
    },
    {
      "epoch": 0.8573997482467182,
      "grad_norm": 8.600707780290314,
      "learning_rate": 2.9499546391416053e-06,
      "loss": 0.2345,
      "step": 1192
    },
    {
      "epoch": 0.8581190433375292,
      "grad_norm": 5.803646814690306,
      "learning_rate": 2.9498674473680617e-06,
      "loss": 0.0759,
      "step": 1193
    },
    {
      "epoch": 0.8588383384283402,
      "grad_norm": 6.604878421124933,
      "learning_rate": 2.9497801809962927e-06,
      "loss": 0.0672,
      "step": 1194
    },
    {
      "epoch": 0.8595576335191513,
      "grad_norm": 6.378136126701569,
      "learning_rate": 2.9496928400307886e-06,
      "loss": 0.196,
      "step": 1195
    },
    {
      "epoch": 0.8602769286099622,
      "grad_norm": 5.707386542098853,
      "learning_rate": 2.9496054244760434e-06,
      "loss": 0.0935,
      "step": 1196
    },
    {
      "epoch": 0.8609962237007732,
      "grad_norm": 6.677928575334813,
      "learning_rate": 2.949517934336554e-06,
      "loss": 0.1042,
      "step": 1197
    },
    {
      "epoch": 0.8617155187915843,
      "grad_norm": 5.3079440530726885,
      "learning_rate": 2.949430369616822e-06,
      "loss": 0.3231,
      "step": 1198
    },
    {
      "epoch": 0.8624348138823953,
      "grad_norm": 4.190781666623576,
      "learning_rate": 2.949342730321354e-06,
      "loss": 0.1149,
      "step": 1199
    },
    {
      "epoch": 0.8631541089732062,
      "grad_norm": 6.1539370694544475,
      "learning_rate": 2.949255016454658e-06,
      "loss": 0.0918,
      "step": 1200
    },
    {
      "epoch": 0.8638734040640172,
      "grad_norm": 3.4290825650816505,
      "learning_rate": 2.9491672280212467e-06,
      "loss": 0.1636,
      "step": 1201
    },
    {
      "epoch": 0.8645926991548283,
      "grad_norm": 9.99921442925207,
      "learning_rate": 2.9490793650256383e-06,
      "loss": 0.1481,
      "step": 1202
    },
    {
      "epoch": 0.8653119942456393,
      "grad_norm": 5.949960649467944,
      "learning_rate": 2.9489914274723526e-06,
      "loss": 0.0723,
      "step": 1203
    },
    {
      "epoch": 0.8660312893364502,
      "grad_norm": 7.583766986501179,
      "learning_rate": 2.948903415365914e-06,
      "loss": 0.1765,
      "step": 1204
    },
    {
      "epoch": 0.8667505844272613,
      "grad_norm": 9.160160172140433,
      "learning_rate": 2.948815328710852e-06,
      "loss": 0.0937,
      "step": 1205
    },
    {
      "epoch": 0.8674698795180723,
      "grad_norm": 5.684521603683679,
      "learning_rate": 2.948727167511697e-06,
      "loss": 0.2544,
      "step": 1206
    },
    {
      "epoch": 0.8681891746088833,
      "grad_norm": 4.213944374192409,
      "learning_rate": 2.9486389317729865e-06,
      "loss": 0.0375,
      "step": 1207
    },
    {
      "epoch": 0.8689084696996943,
      "grad_norm": 5.973771362392137,
      "learning_rate": 2.9485506214992602e-06,
      "loss": 0.1503,
      "step": 1208
    },
    {
      "epoch": 0.8696277647905053,
      "grad_norm": 6.228585309902353,
      "learning_rate": 2.9484622366950613e-06,
      "loss": 0.0494,
      "step": 1209
    },
    {
      "epoch": 0.8703470598813163,
      "grad_norm": 6.940746207053184,
      "learning_rate": 2.948373777364938e-06,
      "loss": 0.0685,
      "step": 1210
    },
    {
      "epoch": 0.8710663549721274,
      "grad_norm": 4.388336320038271,
      "learning_rate": 2.9482852435134406e-06,
      "loss": 0.0836,
      "step": 1211
    },
    {
      "epoch": 0.8717856500629383,
      "grad_norm": 2.5279300724029734,
      "learning_rate": 2.9481966351451257e-06,
      "loss": 0.0068,
      "step": 1212
    },
    {
      "epoch": 0.8725049451537493,
      "grad_norm": 4.259963930781713,
      "learning_rate": 2.9481079522645515e-06,
      "loss": 0.0721,
      "step": 1213
    },
    {
      "epoch": 0.8732242402445604,
      "grad_norm": 6.216823784857963,
      "learning_rate": 2.948019194876281e-06,
      "loss": 0.3805,
      "step": 1214
    },
    {
      "epoch": 0.8739435353353714,
      "grad_norm": 6.62381001420388,
      "learning_rate": 2.9479303629848808e-06,
      "loss": 0.0861,
      "step": 1215
    },
    {
      "epoch": 0.8746628304261823,
      "grad_norm": 4.2549517734651,
      "learning_rate": 2.9478414565949216e-06,
      "loss": 0.0706,
      "step": 1216
    },
    {
      "epoch": 0.8753821255169933,
      "grad_norm": 5.935522314523385,
      "learning_rate": 2.9477524757109783e-06,
      "loss": 0.1294,
      "step": 1217
    },
    {
      "epoch": 0.8761014206078044,
      "grad_norm": 6.954651229777788,
      "learning_rate": 2.9476634203376286e-06,
      "loss": 0.2319,
      "step": 1218
    },
    {
      "epoch": 0.8768207156986153,
      "grad_norm": 4.200440348501926,
      "learning_rate": 2.9475742904794546e-06,
      "loss": 0.0946,
      "step": 1219
    },
    {
      "epoch": 0.8775400107894263,
      "grad_norm": 6.456132926050342,
      "learning_rate": 2.947485086141042e-06,
      "loss": 0.2028,
      "step": 1220
    },
    {
      "epoch": 0.8782593058802374,
      "grad_norm": 5.677931268460433,
      "learning_rate": 2.947395807326981e-06,
      "loss": 0.2854,
      "step": 1221
    },
    {
      "epoch": 0.8789786009710484,
      "grad_norm": 5.203662927823084,
      "learning_rate": 2.9473064540418647e-06,
      "loss": 0.2042,
      "step": 1222
    },
    {
      "epoch": 0.8796978960618593,
      "grad_norm": 4.362549025321659,
      "learning_rate": 2.947217026290291e-06,
      "loss": 0.1679,
      "step": 1223
    },
    {
      "epoch": 0.8804171911526704,
      "grad_norm": 6.652739407814444,
      "learning_rate": 2.94712752407686e-06,
      "loss": 0.0691,
      "step": 1224
    },
    {
      "epoch": 0.8811364862434814,
      "grad_norm": 5.045016718414958,
      "learning_rate": 2.9470379474061786e-06,
      "loss": 0.0467,
      "step": 1225
    },
    {
      "epoch": 0.8818557813342924,
      "grad_norm": 10.990865894028701,
      "learning_rate": 2.946948296282854e-06,
      "loss": 0.2178,
      "step": 1226
    },
    {
      "epoch": 0.8825750764251034,
      "grad_norm": 5.601601655540781,
      "learning_rate": 2.9468585707114996e-06,
      "loss": 0.2552,
      "step": 1227
    },
    {
      "epoch": 0.8832943715159144,
      "grad_norm": 6.521488628112908,
      "learning_rate": 2.946768770696732e-06,
      "loss": 0.1975,
      "step": 1228
    },
    {
      "epoch": 0.8840136666067254,
      "grad_norm": 6.806559308235417,
      "learning_rate": 2.946678896243171e-06,
      "loss": 0.1087,
      "step": 1229
    },
    {
      "epoch": 0.8847329616975365,
      "grad_norm": 3.8790748262950028,
      "learning_rate": 2.946588947355442e-06,
      "loss": 0.0668,
      "step": 1230
    },
    {
      "epoch": 0.8854522567883474,
      "grad_norm": 4.823969467566996,
      "learning_rate": 2.9464989240381715e-06,
      "loss": 0.0816,
      "step": 1231
    },
    {
      "epoch": 0.8861715518791584,
      "grad_norm": 5.323535724191767,
      "learning_rate": 2.946408826295992e-06,
      "loss": 0.2467,
      "step": 1232
    },
    {
      "epoch": 0.8868908469699694,
      "grad_norm": 7.013989091142057,
      "learning_rate": 2.94631865413354e-06,
      "loss": 0.2513,
      "step": 1233
    },
    {
      "epoch": 0.8876101420607805,
      "grad_norm": 5.465247838719756,
      "learning_rate": 2.946228407555454e-06,
      "loss": 0.1457,
      "step": 1234
    },
    {
      "epoch": 0.8883294371515914,
      "grad_norm": 4.679006903274349,
      "learning_rate": 2.9461380865663774e-06,
      "loss": 0.1078,
      "step": 1235
    },
    {
      "epoch": 0.8890487322424024,
      "grad_norm": 5.483062868460178,
      "learning_rate": 2.946047691170958e-06,
      "loss": 0.1565,
      "step": 1236
    },
    {
      "epoch": 0.8897680273332135,
      "grad_norm": 5.406812191515783,
      "learning_rate": 2.9459572213738464e-06,
      "loss": 0.035,
      "step": 1237
    },
    {
      "epoch": 0.8904873224240245,
      "grad_norm": 2.5888970798615603,
      "learning_rate": 2.945866677179698e-06,
      "loss": 0.0136,
      "step": 1238
    },
    {
      "epoch": 0.8912066175148354,
      "grad_norm": 3.9216443470909277,
      "learning_rate": 2.9457760585931695e-06,
      "loss": 0.1292,
      "step": 1239
    },
    {
      "epoch": 0.8919259126056465,
      "grad_norm": 1.0890010511649675,
      "learning_rate": 2.945685365618926e-06,
      "loss": 0.0035,
      "step": 1240
    },
    {
      "epoch": 0.8926452076964575,
      "grad_norm": 5.750388507468079,
      "learning_rate": 2.9455945982616323e-06,
      "loss": 0.2842,
      "step": 1241
    },
    {
      "epoch": 0.8933645027872684,
      "grad_norm": 9.726848843463296,
      "learning_rate": 2.9455037565259584e-06,
      "loss": 0.0856,
      "step": 1242
    },
    {
      "epoch": 0.8940837978780795,
      "grad_norm": 4.062250826333195,
      "learning_rate": 2.945412840416579e-06,
      "loss": 0.0195,
      "step": 1243
    },
    {
      "epoch": 0.8948030929688905,
      "grad_norm": 8.346903018705726,
      "learning_rate": 2.945321849938172e-06,
      "loss": 0.2444,
      "step": 1244
    },
    {
      "epoch": 0.8955223880597015,
      "grad_norm": 6.2177094945650095,
      "learning_rate": 2.9452307850954182e-06,
      "loss": 0.1567,
      "step": 1245
    },
    {
      "epoch": 0.8962416831505124,
      "grad_norm": 6.6402104096089065,
      "learning_rate": 2.9451396458930037e-06,
      "loss": 0.191,
      "step": 1246
    },
    {
      "epoch": 0.8969609782413235,
      "grad_norm": 3.4509943019419875,
      "learning_rate": 2.9450484323356174e-06,
      "loss": 0.0691,
      "step": 1247
    },
    {
      "epoch": 0.8976802733321345,
      "grad_norm": 5.251082501208291,
      "learning_rate": 2.9449571444279525e-06,
      "loss": 0.1794,
      "step": 1248
    },
    {
      "epoch": 0.8983995684229455,
      "grad_norm": 6.132516370067139,
      "learning_rate": 2.944865782174706e-06,
      "loss": 0.2095,
      "step": 1249
    },
    {
      "epoch": 0.8991188635137565,
      "grad_norm": 7.915475872403965,
      "learning_rate": 2.9447743455805794e-06,
      "loss": 0.1762,
      "step": 1250
    },
    {
      "epoch": 0.8998381586045675,
      "grad_norm": 6.770286576991967,
      "learning_rate": 2.9446828346502754e-06,
      "loss": 0.1382,
      "step": 1251
    },
    {
      "epoch": 0.9005574536953785,
      "grad_norm": 6.971956049358906,
      "learning_rate": 2.944591249388504e-06,
      "loss": 0.2458,
      "step": 1252
    },
    {
      "epoch": 0.9012767487861896,
      "grad_norm": 4.784110524046328,
      "learning_rate": 2.944499589799977e-06,
      "loss": 0.2185,
      "step": 1253
    },
    {
      "epoch": 0.9019960438770005,
      "grad_norm": 5.236202692358694,
      "learning_rate": 2.9444078558894104e-06,
      "loss": 0.0482,
      "step": 1254
    },
    {
      "epoch": 0.9027153389678115,
      "grad_norm": 4.625465142948313,
      "learning_rate": 2.944316047661524e-06,
      "loss": 0.1741,
      "step": 1255
    },
    {
      "epoch": 0.9034346340586226,
      "grad_norm": 4.24548194353502,
      "learning_rate": 2.944224165121041e-06,
      "loss": 0.1101,
      "step": 1256
    },
    {
      "epoch": 0.9041539291494336,
      "grad_norm": 8.385866965279517,
      "learning_rate": 2.94413220827269e-06,
      "loss": 0.1371,
      "step": 1257
    },
    {
      "epoch": 0.9048732242402445,
      "grad_norm": 8.72447130184926,
      "learning_rate": 2.9440401771212016e-06,
      "loss": 0.1705,
      "step": 1258
    },
    {
      "epoch": 0.9055925193310556,
      "grad_norm": 9.930171442491735,
      "learning_rate": 2.943948071671311e-06,
      "loss": 0.1952,
      "step": 1259
    },
    {
      "epoch": 0.9063118144218666,
      "grad_norm": 6.506973826894642,
      "learning_rate": 2.943855891927758e-06,
      "loss": 0.3228,
      "step": 1260
    },
    {
      "epoch": 0.9070311095126776,
      "grad_norm": 8.808876841120776,
      "learning_rate": 2.943763637895284e-06,
      "loss": 0.3505,
      "step": 1261
    },
    {
      "epoch": 0.9077504046034885,
      "grad_norm": 6.216960179004871,
      "learning_rate": 2.9436713095786365e-06,
      "loss": 0.1968,
      "step": 1262
    },
    {
      "epoch": 0.9084696996942996,
      "grad_norm": 4.13135181764248,
      "learning_rate": 2.9435789069825663e-06,
      "loss": 0.0744,
      "step": 1263
    },
    {
      "epoch": 0.9091889947851106,
      "grad_norm": 5.358020926579018,
      "learning_rate": 2.943486430111827e-06,
      "loss": 0.053,
      "step": 1264
    },
    {
      "epoch": 0.9099082898759216,
      "grad_norm": 7.161167042971359,
      "learning_rate": 2.9433938789711763e-06,
      "loss": 0.2794,
      "step": 1265
    },
    {
      "epoch": 0.9106275849667326,
      "grad_norm": 8.033414353489725,
      "learning_rate": 2.943301253565378e-06,
      "loss": 0.1243,
      "step": 1266
    },
    {
      "epoch": 0.9113468800575436,
      "grad_norm": 3.9047823003403646,
      "learning_rate": 2.9432085538991954e-06,
      "loss": 0.0507,
      "step": 1267
    },
    {
      "epoch": 0.9120661751483546,
      "grad_norm": 5.256928859374807,
      "learning_rate": 2.9431157799773998e-06,
      "loss": 0.1042,
      "step": 1268
    },
    {
      "epoch": 0.9127854702391656,
      "grad_norm": 4.162360614440319,
      "learning_rate": 2.9430229318047638e-06,
      "loss": 0.0951,
      "step": 1269
    },
    {
      "epoch": 0.9135047653299766,
      "grad_norm": 3.6772985401229925,
      "learning_rate": 2.9429300093860646e-06,
      "loss": 0.0914,
      "step": 1270
    },
    {
      "epoch": 0.9142240604207876,
      "grad_norm": 5.549020067543673,
      "learning_rate": 2.942837012726084e-06,
      "loss": 0.0981,
      "step": 1271
    },
    {
      "epoch": 0.9149433555115987,
      "grad_norm": 5.554176913634114,
      "learning_rate": 2.942743941829606e-06,
      "loss": 0.1003,
      "step": 1272
    },
    {
      "epoch": 0.9156626506024096,
      "grad_norm": 5.843817959520465,
      "learning_rate": 2.9426507967014195e-06,
      "loss": 0.2313,
      "step": 1273
    },
    {
      "epoch": 0.9163819456932206,
      "grad_norm": 2.45356412532244,
      "learning_rate": 2.942557577346317e-06,
      "loss": 0.0304,
      "step": 1274
    },
    {
      "epoch": 0.9171012407840317,
      "grad_norm": 6.255491047619371,
      "learning_rate": 2.942464283769095e-06,
      "loss": 0.2095,
      "step": 1275
    },
    {
      "epoch": 0.9178205358748427,
      "grad_norm": 3.467736850193894,
      "learning_rate": 2.942370915974553e-06,
      "loss": 0.0842,
      "step": 1276
    },
    {
      "epoch": 0.9185398309656536,
      "grad_norm": 8.9891249190695,
      "learning_rate": 2.942277473967496e-06,
      "loss": 0.1419,
      "step": 1277
    },
    {
      "epoch": 0.9192591260564646,
      "grad_norm": 6.123182947956828,
      "learning_rate": 2.9421839577527303e-06,
      "loss": 0.264,
      "step": 1278
    },
    {
      "epoch": 0.9199784211472757,
      "grad_norm": 4.592559339279738,
      "learning_rate": 2.942090367335069e-06,
      "loss": 0.1805,
      "step": 1279
    },
    {
      "epoch": 0.9206977162380867,
      "grad_norm": 3.202767606028297,
      "learning_rate": 2.9419967027193267e-06,
      "loss": 0.0861,
      "step": 1280
    },
    {
      "epoch": 0.9214170113288976,
      "grad_norm": 3.990044003740237,
      "learning_rate": 2.9419029639103224e-06,
      "loss": 0.0984,
      "step": 1281
    },
    {
      "epoch": 0.9221363064197087,
      "grad_norm": 6.3767280819076175,
      "learning_rate": 2.94180915091288e-06,
      "loss": 0.2484,
      "step": 1282
    },
    {
      "epoch": 0.9228556015105197,
      "grad_norm": 2.395642048952248,
      "learning_rate": 2.941715263731825e-06,
      "loss": 0.0637,
      "step": 1283
    },
    {
      "epoch": 0.9235748966013307,
      "grad_norm": 4.72161436848947,
      "learning_rate": 2.9416213023719897e-06,
      "loss": 0.1272,
      "step": 1284
    },
    {
      "epoch": 0.9242941916921417,
      "grad_norm": 5.226652822577646,
      "learning_rate": 2.941527266838207e-06,
      "loss": 0.1075,
      "step": 1285
    },
    {
      "epoch": 0.9250134867829527,
      "grad_norm": 9.29996368039424,
      "learning_rate": 2.941433157135316e-06,
      "loss": 0.2204,
      "step": 1286
    },
    {
      "epoch": 0.9257327818737637,
      "grad_norm": 3.7379044946969633,
      "learning_rate": 2.9413389732681587e-06,
      "loss": 0.0442,
      "step": 1287
    },
    {
      "epoch": 0.9264520769645748,
      "grad_norm": 4.56554867619894,
      "learning_rate": 2.9412447152415814e-06,
      "loss": 0.104,
      "step": 1288
    },
    {
      "epoch": 0.9271713720553857,
      "grad_norm": 5.042743762018386,
      "learning_rate": 2.941150383060433e-06,
      "loss": 0.1619,
      "step": 1289
    },
    {
      "epoch": 0.9278906671461967,
      "grad_norm": 4.87798369899283,
      "learning_rate": 2.9410559767295678e-06,
      "loss": 0.2154,
      "step": 1290
    },
    {
      "epoch": 0.9286099622370078,
      "grad_norm": 3.575254223545318,
      "learning_rate": 2.9409614962538426e-06,
      "loss": 0.1993,
      "step": 1291
    },
    {
      "epoch": 0.9293292573278188,
      "grad_norm": 5.75267092343592,
      "learning_rate": 2.9408669416381194e-06,
      "loss": 0.2261,
      "step": 1292
    },
    {
      "epoch": 0.9300485524186297,
      "grad_norm": 5.991164848122601,
      "learning_rate": 2.9407723128872626e-06,
      "loss": 0.2486,
      "step": 1293
    },
    {
      "epoch": 0.9307678475094407,
      "grad_norm": 7.90291872679686,
      "learning_rate": 2.940677610006141e-06,
      "loss": 0.0533,
      "step": 1294
    },
    {
      "epoch": 0.9314871426002518,
      "grad_norm": 5.052033192637415,
      "learning_rate": 2.940582832999627e-06,
      "loss": 0.1146,
      "step": 1295
    },
    {
      "epoch": 0.9322064376910627,
      "grad_norm": 4.6764772861453165,
      "learning_rate": 2.9404879818725976e-06,
      "loss": 0.2488,
      "step": 1296
    },
    {
      "epoch": 0.9329257327818737,
      "grad_norm": 3.854389510193265,
      "learning_rate": 2.940393056629933e-06,
      "loss": 0.1619,
      "step": 1297
    },
    {
      "epoch": 0.9336450278726848,
      "grad_norm": 5.171191248763344,
      "learning_rate": 2.9402980572765174e-06,
      "loss": 0.1611,
      "step": 1298
    },
    {
      "epoch": 0.9343643229634958,
      "grad_norm": 4.043876190598902,
      "learning_rate": 2.9402029838172375e-06,
      "loss": 0.0957,
      "step": 1299
    },
    {
      "epoch": 0.9350836180543067,
      "grad_norm": 4.263483569515798,
      "learning_rate": 2.9401078362569865e-06,
      "loss": 0.0244,
      "step": 1300
    },
    {
      "epoch": 0.9358029131451178,
      "grad_norm": 8.249234951837312,
      "learning_rate": 2.9400126146006595e-06,
      "loss": 0.2394,
      "step": 1301
    },
    {
      "epoch": 0.9365222082359288,
      "grad_norm": 7.578433066199969,
      "learning_rate": 2.939917318853155e-06,
      "loss": 0.1882,
      "step": 1302
    },
    {
      "epoch": 0.9372415033267398,
      "grad_norm": 7.045618904804449,
      "learning_rate": 2.9398219490193773e-06,
      "loss": 0.0711,
      "step": 1303
    },
    {
      "epoch": 0.9379607984175508,
      "grad_norm": 6.621477538424895,
      "learning_rate": 2.939726505104232e-06,
      "loss": 0.2995,
      "step": 1304
    },
    {
      "epoch": 0.9386800935083618,
      "grad_norm": 4.87428226046337,
      "learning_rate": 2.9396309871126317e-06,
      "loss": 0.1381,
      "step": 1305
    },
    {
      "epoch": 0.9393993885991728,
      "grad_norm": 3.9012964794409446,
      "learning_rate": 2.9395353950494893e-06,
      "loss": 0.0381,
      "step": 1306
    },
    {
      "epoch": 0.9401186836899839,
      "grad_norm": 4.528116812568083,
      "learning_rate": 2.939439728919724e-06,
      "loss": 0.107,
      "step": 1307
    },
    {
      "epoch": 0.9408379787807948,
      "grad_norm": 8.494676761221033,
      "learning_rate": 2.939343988728257e-06,
      "loss": 0.3946,
      "step": 1308
    },
    {
      "epoch": 0.9415572738716058,
      "grad_norm": 3.663528762661194,
      "learning_rate": 2.9392481744800163e-06,
      "loss": 0.0892,
      "step": 1309
    },
    {
      "epoch": 0.9422765689624168,
      "grad_norm": 6.238754791163149,
      "learning_rate": 2.93915228617993e-06,
      "loss": 0.0526,
      "step": 1310
    },
    {
      "epoch": 0.9429958640532279,
      "grad_norm": 4.424049831857157,
      "learning_rate": 2.9390563238329324e-06,
      "loss": 0.0841,
      "step": 1311
    },
    {
      "epoch": 0.9437151591440388,
      "grad_norm": 0.6020758718317281,
      "learning_rate": 2.9389602874439607e-06,
      "loss": 0.0019,
      "step": 1312
    },
    {
      "epoch": 0.9444344542348498,
      "grad_norm": 6.779203769438188,
      "learning_rate": 2.9388641770179557e-06,
      "loss": 0.2314,
      "step": 1313
    },
    {
      "epoch": 0.9451537493256609,
      "grad_norm": 6.937282478794683,
      "learning_rate": 2.938767992559863e-06,
      "loss": 0.1591,
      "step": 1314
    },
    {
      "epoch": 0.9458730444164719,
      "grad_norm": 6.234079077051618,
      "learning_rate": 2.938671734074632e-06,
      "loss": 0.1834,
      "step": 1315
    },
    {
      "epoch": 0.9465923395072828,
      "grad_norm": 6.841309775424993,
      "learning_rate": 2.938575401567215e-06,
      "loss": 0.2178,
      "step": 1316
    },
    {
      "epoch": 0.9473116345980939,
      "grad_norm": 9.602528586623908,
      "learning_rate": 2.9384789950425673e-06,
      "loss": 0.2306,
      "step": 1317
    },
    {
      "epoch": 0.9480309296889049,
      "grad_norm": 6.005471809326496,
      "learning_rate": 2.9383825145056503e-06,
      "loss": 0.0187,
      "step": 1318
    },
    {
      "epoch": 0.9487502247797159,
      "grad_norm": 2.657277897645906,
      "learning_rate": 2.9382859599614284e-06,
      "loss": 0.0502,
      "step": 1319
    },
    {
      "epoch": 0.9494695198705269,
      "grad_norm": 8.775551327657592,
      "learning_rate": 2.9381893314148687e-06,
      "loss": 0.2498,
      "step": 1320
    },
    {
      "epoch": 0.9501888149613379,
      "grad_norm": 5.698308629174225,
      "learning_rate": 2.9380926288709437e-06,
      "loss": 0.1002,
      "step": 1321
    },
    {
      "epoch": 0.9509081100521489,
      "grad_norm": 5.1383003657534765,
      "learning_rate": 2.9379958523346283e-06,
      "loss": 0.1909,
      "step": 1322
    },
    {
      "epoch": 0.9516274051429598,
      "grad_norm": 8.798398440645089,
      "learning_rate": 2.9378990018109014e-06,
      "loss": 0.0802,
      "step": 1323
    },
    {
      "epoch": 0.9523467002337709,
      "grad_norm": 4.914679586563055,
      "learning_rate": 2.937802077304747e-06,
      "loss": 0.1796,
      "step": 1324
    },
    {
      "epoch": 0.9530659953245819,
      "grad_norm": 8.312203314162511,
      "learning_rate": 2.937705078821152e-06,
      "loss": 0.095,
      "step": 1325
    },
    {
      "epoch": 0.953785290415393,
      "grad_norm": 3.6172836222284883,
      "learning_rate": 2.937608006365107e-06,
      "loss": 0.0435,
      "step": 1326
    },
    {
      "epoch": 0.9545045855062039,
      "grad_norm": 5.043999173881439,
      "learning_rate": 2.9375108599416062e-06,
      "loss": 0.1614,
      "step": 1327
    },
    {
      "epoch": 0.9552238805970149,
      "grad_norm": 6.349023434348644,
      "learning_rate": 2.9374136395556477e-06,
      "loss": 0.103,
      "step": 1328
    },
    {
      "epoch": 0.9559431756878259,
      "grad_norm": 5.276008530527816,
      "learning_rate": 2.937316345212235e-06,
      "loss": 0.1666,
      "step": 1329
    },
    {
      "epoch": 0.956662470778637,
      "grad_norm": 6.163055949919098,
      "learning_rate": 2.9372189769163726e-06,
      "loss": 0.1866,
      "step": 1330
    },
    {
      "epoch": 0.9573817658694479,
      "grad_norm": 3.929252611917299,
      "learning_rate": 2.937121534673071e-06,
      "loss": 0.0842,
      "step": 1331
    },
    {
      "epoch": 0.9581010609602589,
      "grad_norm": 9.238766847364214,
      "learning_rate": 2.937024018487344e-06,
      "loss": 0.2874,
      "step": 1332
    },
    {
      "epoch": 0.95882035605107,
      "grad_norm": 4.538855438518856,
      "learning_rate": 2.936926428364208e-06,
      "loss": 0.1823,
      "step": 1333
    },
    {
      "epoch": 0.959539651141881,
      "grad_norm": 2.543646867943277,
      "learning_rate": 2.936828764308685e-06,
      "loss": 0.1089,
      "step": 1334
    },
    {
      "epoch": 0.9602589462326919,
      "grad_norm": 4.442979294694012,
      "learning_rate": 2.9367310263258e-06,
      "loss": 0.1321,
      "step": 1335
    },
    {
      "epoch": 0.960978241323503,
      "grad_norm": 5.115615296348857,
      "learning_rate": 2.936633214420582e-06,
      "loss": 0.0692,
      "step": 1336
    },
    {
      "epoch": 0.961697536414314,
      "grad_norm": 4.844818887072438,
      "learning_rate": 2.936535328598063e-06,
      "loss": 0.2071,
      "step": 1337
    },
    {
      "epoch": 0.962416831505125,
      "grad_norm": 5.909470102807963,
      "learning_rate": 2.936437368863279e-06,
      "loss": 0.1876,
      "step": 1338
    },
    {
      "epoch": 0.963136126595936,
      "grad_norm": 8.972473629729977,
      "learning_rate": 2.9363393352212715e-06,
      "loss": 0.3448,
      "step": 1339
    },
    {
      "epoch": 0.963855421686747,
      "grad_norm": 5.185373699215868,
      "learning_rate": 2.9362412276770835e-06,
      "loss": 0.1129,
      "step": 1340
    },
    {
      "epoch": 0.964574716777558,
      "grad_norm": 2.1134832222454696,
      "learning_rate": 2.9361430462357625e-06,
      "loss": 0.0257,
      "step": 1341
    },
    {
      "epoch": 0.965294011868369,
      "grad_norm": 3.2378638520482026,
      "learning_rate": 2.9360447909023615e-06,
      "loss": 0.0285,
      "step": 1342
    },
    {
      "epoch": 0.96601330695918,
      "grad_norm": 3.2743449072117907,
      "learning_rate": 2.935946461681935e-06,
      "loss": 0.0225,
      "step": 1343
    },
    {
      "epoch": 0.966732602049991,
      "grad_norm": 4.9159760654342435,
      "learning_rate": 2.9358480585795416e-06,
      "loss": 0.1496,
      "step": 1344
    },
    {
      "epoch": 0.967451897140802,
      "grad_norm": 1.8003457505524643,
      "learning_rate": 2.935749581600246e-06,
      "loss": 0.0152,
      "step": 1345
    },
    {
      "epoch": 0.968171192231613,
      "grad_norm": 3.3679058981093135,
      "learning_rate": 2.9356510307491135e-06,
      "loss": 0.0885,
      "step": 1346
    },
    {
      "epoch": 0.968890487322424,
      "grad_norm": 3.2417301776507714,
      "learning_rate": 2.9355524060312155e-06,
      "loss": 0.0298,
      "step": 1347
    },
    {
      "epoch": 0.969609782413235,
      "grad_norm": 6.817446565799031,
      "learning_rate": 2.935453707451626e-06,
      "loss": 0.2995,
      "step": 1348
    },
    {
      "epoch": 0.9703290775040461,
      "grad_norm": 6.2293619177312225,
      "learning_rate": 2.9353549350154236e-06,
      "loss": 0.1859,
      "step": 1349
    },
    {
      "epoch": 0.971048372594857,
      "grad_norm": 4.873471336724789,
      "learning_rate": 2.93525608872769e-06,
      "loss": 0.1807,
      "step": 1350
    },
    {
      "epoch": 0.971767667685668,
      "grad_norm": 7.62690882822264,
      "learning_rate": 2.935157168593511e-06,
      "loss": 0.034,
      "step": 1351
    },
    {
      "epoch": 0.9724869627764791,
      "grad_norm": 5.999300008855233,
      "learning_rate": 2.9350581746179764e-06,
      "loss": 0.1917,
      "step": 1352
    },
    {
      "epoch": 0.9732062578672901,
      "grad_norm": 4.680982964250305,
      "learning_rate": 2.9349591068061797e-06,
      "loss": 0.0774,
      "step": 1353
    },
    {
      "epoch": 0.973925552958101,
      "grad_norm": 5.22320571231796,
      "learning_rate": 2.9348599651632177e-06,
      "loss": 0.157,
      "step": 1354
    },
    {
      "epoch": 0.974644848048912,
      "grad_norm": 5.632876424360375,
      "learning_rate": 2.934760749694192e-06,
      "loss": 0.2553,
      "step": 1355
    },
    {
      "epoch": 0.9753641431397231,
      "grad_norm": 6.096409888625927,
      "learning_rate": 2.934661460404207e-06,
      "loss": 0.1513,
      "step": 1356
    },
    {
      "epoch": 0.9760834382305341,
      "grad_norm": 4.187139932755058,
      "learning_rate": 2.9345620972983715e-06,
      "loss": 0.0236,
      "step": 1357
    },
    {
      "epoch": 0.976802733321345,
      "grad_norm": 5.64497313491945,
      "learning_rate": 2.9344626603817972e-06,
      "loss": 0.1223,
      "step": 1358
    },
    {
      "epoch": 0.9775220284121561,
      "grad_norm": 11.614650134794717,
      "learning_rate": 2.9343631496596014e-06,
      "loss": 0.2071,
      "step": 1359
    },
    {
      "epoch": 0.9782413235029671,
      "grad_norm": 5.196556178766472,
      "learning_rate": 2.9342635651369033e-06,
      "loss": 0.1035,
      "step": 1360
    },
    {
      "epoch": 0.9789606185937781,
      "grad_norm": 8.2711162132498,
      "learning_rate": 2.9341639068188275e-06,
      "loss": 0.1057,
      "step": 1361
    },
    {
      "epoch": 0.9796799136845891,
      "grad_norm": 5.906295648441473,
      "learning_rate": 2.934064174710501e-06,
      "loss": 0.1331,
      "step": 1362
    },
    {
      "epoch": 0.9803992087754001,
      "grad_norm": 5.165080378088079,
      "learning_rate": 2.933964368817055e-06,
      "loss": 0.04,
      "step": 1363
    },
    {
      "epoch": 0.9811185038662111,
      "grad_norm": 3.4052673277614267,
      "learning_rate": 2.933864489143625e-06,
      "loss": 0.0135,
      "step": 1364
    },
    {
      "epoch": 0.9818377989570222,
      "grad_norm": 4.38275133226788,
      "learning_rate": 2.9337645356953496e-06,
      "loss": 0.124,
      "step": 1365
    },
    {
      "epoch": 0.9825570940478331,
      "grad_norm": 5.412869994564405,
      "learning_rate": 2.9336645084773725e-06,
      "loss": 0.1408,
      "step": 1366
    },
    {
      "epoch": 0.9832763891386441,
      "grad_norm": 5.028367265133365,
      "learning_rate": 2.93356440749484e-06,
      "loss": 0.2062,
      "step": 1367
    },
    {
      "epoch": 0.9839956842294552,
      "grad_norm": 1.3784656980872196,
      "learning_rate": 2.9334642327529016e-06,
      "loss": 0.0093,
      "step": 1368
    },
    {
      "epoch": 0.9847149793202662,
      "grad_norm": 5.78242422254892,
      "learning_rate": 2.9333639842567123e-06,
      "loss": 0.1425,
      "step": 1369
    },
    {
      "epoch": 0.9854342744110771,
      "grad_norm": 7.063427648312734,
      "learning_rate": 2.93326366201143e-06,
      "loss": 0.2509,
      "step": 1370
    },
    {
      "epoch": 0.9861535695018881,
      "grad_norm": 5.499604929505757,
      "learning_rate": 2.9331632660222156e-06,
      "loss": 0.0416,
      "step": 1371
    },
    {
      "epoch": 0.9868728645926992,
      "grad_norm": 5.263638780634471,
      "learning_rate": 2.933062796294236e-06,
      "loss": 0.1209,
      "step": 1372
    },
    {
      "epoch": 0.9875921596835101,
      "grad_norm": 2.6796228443468735,
      "learning_rate": 2.9329622528326596e-06,
      "loss": 0.0206,
      "step": 1373
    },
    {
      "epoch": 0.9883114547743211,
      "grad_norm": 4.00930218762538,
      "learning_rate": 2.9328616356426597e-06,
      "loss": 0.1217,
      "step": 1374
    },
    {
      "epoch": 0.9890307498651322,
      "grad_norm": 6.193822314405915,
      "learning_rate": 2.932760944729414e-06,
      "loss": 0.2707,
      "step": 1375
    },
    {
      "epoch": 0.9897500449559432,
      "grad_norm": 6.603153096357186,
      "learning_rate": 2.9326601800981015e-06,
      "loss": 0.1886,
      "step": 1376
    },
    {
      "epoch": 0.9904693400467541,
      "grad_norm": 5.059592451445477,
      "learning_rate": 2.9325593417539088e-06,
      "loss": 0.1363,
      "step": 1377
    },
    {
      "epoch": 0.9911886351375652,
      "grad_norm": 4.818312944074029,
      "learning_rate": 2.9324584297020228e-06,
      "loss": 0.176,
      "step": 1378
    },
    {
      "epoch": 0.9919079302283762,
      "grad_norm": 4.182412138337019,
      "learning_rate": 2.9323574439476354e-06,
      "loss": 0.1295,
      "step": 1379
    },
    {
      "epoch": 0.9926272253191872,
      "grad_norm": 5.4954889159296805,
      "learning_rate": 2.9322563844959438e-06,
      "loss": 0.0343,
      "step": 1380
    },
    {
      "epoch": 0.9933465204099982,
      "grad_norm": 4.712358215324916,
      "learning_rate": 2.9321552513521472e-06,
      "loss": 0.0926,
      "step": 1381
    },
    {
      "epoch": 0.9940658155008092,
      "grad_norm": 7.396615357648573,
      "learning_rate": 2.9320540445214483e-06,
      "loss": 0.1132,
      "step": 1382
    },
    {
      "epoch": 0.9947851105916202,
      "grad_norm": 7.973474165631346,
      "learning_rate": 2.931952764009055e-06,
      "loss": 0.2409,
      "step": 1383
    },
    {
      "epoch": 0.9955044056824313,
      "grad_norm": 8.781700307611532,
      "learning_rate": 2.9318514098201783e-06,
      "loss": 0.0689,
      "step": 1384
    },
    {
      "epoch": 0.9962237007732422,
      "grad_norm": 8.515470094618632,
      "learning_rate": 2.9317499819600332e-06,
      "loss": 0.1517,
      "step": 1385
    },
    {
      "epoch": 0.9969429958640532,
      "grad_norm": 4.958434198085519,
      "learning_rate": 2.9316484804338377e-06,
      "loss": 0.0607,
      "step": 1386
    },
    {
      "epoch": 0.9976622909548643,
      "grad_norm": 7.187995371338799,
      "learning_rate": 2.931546905246815e-06,
      "loss": 0.0821,
      "step": 1387
    },
    {
      "epoch": 0.9983815860456753,
      "grad_norm": 4.784326353445957,
      "learning_rate": 2.9314452564041913e-06,
      "loss": 0.0671,
      "step": 1388
    },
    {
      "epoch": 0.9991008811364862,
      "grad_norm": 7.122956004731571,
      "learning_rate": 2.9313435339111957e-06,
      "loss": 0.0955,
      "step": 1389
    },
    {
      "epoch": 0.9998201762272972,
      "grad_norm": 4.884861312124538,
      "learning_rate": 2.9312417377730633e-06,
      "loss": 0.1574,
      "step": 1390
    },
    {
      "epoch": 1.0005394713181082,
      "grad_norm": 2.3247312465883416,
      "learning_rate": 2.9311398679950304e-06,
      "loss": 0.0058,
      "step": 1391
    },
    {
      "epoch": 1.0012587664089192,
      "grad_norm": 4.477142552949364,
      "learning_rate": 2.9310379245823397e-06,
      "loss": 0.0387,
      "step": 1392
    },
    {
      "epoch": 1.0019780614997302,
      "grad_norm": 5.628018056361989,
      "learning_rate": 2.9309359075402347e-06,
      "loss": 0.1208,
      "step": 1393
    },
    {
      "epoch": 1.0026973565905413,
      "grad_norm": 7.118604766685235,
      "learning_rate": 2.9308338168739663e-06,
      "loss": 0.1335,
      "step": 1394
    },
    {
      "epoch": 1.0034166516813523,
      "grad_norm": 5.515477812340682,
      "learning_rate": 2.9307316525887855e-06,
      "loss": 0.0396,
      "step": 1395
    },
    {
      "epoch": 1.0041359467721633,
      "grad_norm": 2.789903061554602,
      "learning_rate": 2.9306294146899504e-06,
      "loss": 0.0532,
      "step": 1396
    },
    {
      "epoch": 1.0048552418629744,
      "grad_norm": 5.143654984850947,
      "learning_rate": 2.93052710318272e-06,
      "loss": 0.1297,
      "step": 1397
    },
    {
      "epoch": 1.0055745369537852,
      "grad_norm": 5.494299138391224,
      "learning_rate": 2.9304247180723595e-06,
      "loss": 0.1874,
      "step": 1398
    },
    {
      "epoch": 1.0062938320445962,
      "grad_norm": 2.4251399029339287,
      "learning_rate": 2.9303222593641357e-06,
      "loss": 0.0112,
      "step": 1399
    },
    {
      "epoch": 1.0070131271354072,
      "grad_norm": 3.6563007601426576,
      "learning_rate": 2.930219727063321e-06,
      "loss": 0.0721,
      "step": 1400
    },
    {
      "epoch": 1.0077324222262183,
      "grad_norm": 3.487920433125351,
      "learning_rate": 2.9301171211751904e-06,
      "loss": 0.1469,
      "step": 1401
    },
    {
      "epoch": 1.0084517173170293,
      "grad_norm": 7.97052461925232,
      "learning_rate": 2.9300144417050237e-06,
      "loss": 0.1024,
      "step": 1402
    },
    {
      "epoch": 1.0091710124078404,
      "grad_norm": 6.148267670307876,
      "learning_rate": 2.9299116886581032e-06,
      "loss": 0.0484,
      "step": 1403
    },
    {
      "epoch": 1.0098903074986514,
      "grad_norm": 6.740993747586648,
      "learning_rate": 2.9298088620397166e-06,
      "loss": 0.1861,
      "step": 1404
    },
    {
      "epoch": 1.0106096025894624,
      "grad_norm": 6.533819187784662,
      "learning_rate": 2.929705961855154e-06,
      "loss": 0.1007,
      "step": 1405
    },
    {
      "epoch": 1.0113288976802732,
      "grad_norm": 3.6454979506074268,
      "learning_rate": 2.9296029881097104e-06,
      "loss": 0.1467,
      "step": 1406
    },
    {
      "epoch": 1.0120481927710843,
      "grad_norm": 3.0240831331690945,
      "learning_rate": 2.9294999408086825e-06,
      "loss": 0.0781,
      "step": 1407
    },
    {
      "epoch": 1.0127674878618953,
      "grad_norm": 5.737698595041257,
      "learning_rate": 2.929396819957374e-06,
      "loss": 0.1157,
      "step": 1408
    },
    {
      "epoch": 1.0134867829527063,
      "grad_norm": 7.86037923693671,
      "learning_rate": 2.92929362556109e-06,
      "loss": 0.0637,
      "step": 1409
    },
    {
      "epoch": 1.0142060780435174,
      "grad_norm": 1.3225909553719413,
      "learning_rate": 2.9291903576251394e-06,
      "loss": 0.0065,
      "step": 1410
    },
    {
      "epoch": 1.0149253731343284,
      "grad_norm": 10.01505367850342,
      "learning_rate": 2.929087016154836e-06,
      "loss": 0.2248,
      "step": 1411
    },
    {
      "epoch": 1.0156446682251394,
      "grad_norm": 3.3701173192710914,
      "learning_rate": 2.928983601155497e-06,
      "loss": 0.0091,
      "step": 1412
    },
    {
      "epoch": 1.0163639633159505,
      "grad_norm": 4.647333162567701,
      "learning_rate": 2.928880112632443e-06,
      "loss": 0.069,
      "step": 1413
    },
    {
      "epoch": 1.0170832584067613,
      "grad_norm": 2.105368285009397,
      "learning_rate": 2.9287765505909995e-06,
      "loss": 0.0105,
      "step": 1414
    },
    {
      "epoch": 1.0178025534975723,
      "grad_norm": 7.194997295623576,
      "learning_rate": 2.928672915036494e-06,
      "loss": 0.2147,
      "step": 1415
    },
    {
      "epoch": 1.0185218485883833,
      "grad_norm": 3.9545142095715553,
      "learning_rate": 2.928569205974259e-06,
      "loss": 0.1553,
      "step": 1416
    },
    {
      "epoch": 1.0192411436791944,
      "grad_norm": 2.6531451827293253,
      "learning_rate": 2.928465423409631e-06,
      "loss": 0.0178,
      "step": 1417
    },
    {
      "epoch": 1.0199604387700054,
      "grad_norm": 6.907830394938768,
      "learning_rate": 2.928361567347949e-06,
      "loss": 0.2328,
      "step": 1418
    },
    {
      "epoch": 1.0206797338608165,
      "grad_norm": 4.547012339078066,
      "learning_rate": 2.928257637794557e-06,
      "loss": 0.1655,
      "step": 1419
    },
    {
      "epoch": 1.0213990289516275,
      "grad_norm": 10.334031782391865,
      "learning_rate": 2.9281536347548024e-06,
      "loss": 0.098,
      "step": 1420
    },
    {
      "epoch": 1.0221183240424385,
      "grad_norm": 7.75808816296915,
      "learning_rate": 2.928049558234036e-06,
      "loss": 0.1627,
      "step": 1421
    },
    {
      "epoch": 1.0228376191332493,
      "grad_norm": 6.071793791720597,
      "learning_rate": 2.9279454082376134e-06,
      "loss": 0.1057,
      "step": 1422
    },
    {
      "epoch": 1.0235569142240604,
      "grad_norm": 4.817435916806376,
      "learning_rate": 2.9278411847708928e-06,
      "loss": 0.1751,
      "step": 1423
    },
    {
      "epoch": 1.0242762093148714,
      "grad_norm": 1.3726063175249814,
      "learning_rate": 2.9277368878392365e-06,
      "loss": 0.0248,
      "step": 1424
    },
    {
      "epoch": 1.0249955044056824,
      "grad_norm": 5.6378240333896565,
      "learning_rate": 2.927632517448011e-06,
      "loss": 0.134,
      "step": 1425
    },
    {
      "epoch": 1.0257147994964935,
      "grad_norm": 5.4055841455839255,
      "learning_rate": 2.9275280736025864e-06,
      "loss": 0.119,
      "step": 1426
    },
    {
      "epoch": 1.0264340945873045,
      "grad_norm": 4.493895074694661,
      "learning_rate": 2.9274235563083365e-06,
      "loss": 0.2053,
      "step": 1427
    },
    {
      "epoch": 1.0271533896781155,
      "grad_norm": 4.274873372164417,
      "learning_rate": 2.927318965570639e-06,
      "loss": 0.0457,
      "step": 1428
    },
    {
      "epoch": 1.0278726847689263,
      "grad_norm": 7.900905239687231,
      "learning_rate": 2.9272143013948747e-06,
      "loss": 0.2692,
      "step": 1429
    },
    {
      "epoch": 1.0285919798597374,
      "grad_norm": 3.793530770988621,
      "learning_rate": 2.9271095637864294e-06,
      "loss": 0.0985,
      "step": 1430
    },
    {
      "epoch": 1.0293112749505484,
      "grad_norm": 8.240289582051897,
      "learning_rate": 2.9270047527506922e-06,
      "loss": 0.2147,
      "step": 1431
    },
    {
      "epoch": 1.0300305700413595,
      "grad_norm": 4.551649957537649,
      "learning_rate": 2.926899868293055e-06,
      "loss": 0.0983,
      "step": 1432
    },
    {
      "epoch": 1.0307498651321705,
      "grad_norm": 7.346732655405621,
      "learning_rate": 2.926794910418915e-06,
      "loss": 0.0999,
      "step": 1433
    },
    {
      "epoch": 1.0314691602229815,
      "grad_norm": 6.0936634645168475,
      "learning_rate": 2.926689879133672e-06,
      "loss": 0.2177,
      "step": 1434
    },
    {
      "epoch": 1.0321884553137926,
      "grad_norm": 6.091105478119522,
      "learning_rate": 2.9265847744427307e-06,
      "loss": 0.1846,
      "step": 1435
    },
    {
      "epoch": 1.0329077504046036,
      "grad_norm": 4.057924498885607,
      "learning_rate": 2.9264795963514983e-06,
      "loss": 0.1894,
      "step": 1436
    },
    {
      "epoch": 1.0336270454954144,
      "grad_norm": 4.9917784186657235,
      "learning_rate": 2.926374344865386e-06,
      "loss": 0.189,
      "step": 1437
    },
    {
      "epoch": 1.0343463405862254,
      "grad_norm": 7.027690058970331,
      "learning_rate": 2.92626901998981e-06,
      "loss": 0.1647,
      "step": 1438
    },
    {
      "epoch": 1.0350656356770365,
      "grad_norm": 12.398962184859455,
      "learning_rate": 2.9261636217301895e-06,
      "loss": 0.232,
      "step": 1439
    },
    {
      "epoch": 1.0357849307678475,
      "grad_norm": 6.077292551731999,
      "learning_rate": 2.9260581500919465e-06,
      "loss": 0.0736,
      "step": 1440
    },
    {
      "epoch": 1.0365042258586585,
      "grad_norm": 7.571167475639318,
      "learning_rate": 2.9259526050805087e-06,
      "loss": 0.0668,
      "step": 1441
    },
    {
      "epoch": 1.0372235209494696,
      "grad_norm": 4.388295734597306,
      "learning_rate": 2.9258469867013064e-06,
      "loss": 0.0775,
      "step": 1442
    },
    {
      "epoch": 1.0379428160402806,
      "grad_norm": 7.09977324931201,
      "learning_rate": 2.9257412949597733e-06,
      "loss": 0.3884,
      "step": 1443
    },
    {
      "epoch": 1.0386621111310916,
      "grad_norm": 6.287338747096999,
      "learning_rate": 2.925635529861348e-06,
      "loss": 0.3921,
      "step": 1444
    },
    {
      "epoch": 1.0393814062219024,
      "grad_norm": 5.847406918683498,
      "learning_rate": 2.925529691411472e-06,
      "loss": 0.2147,
      "step": 1445
    },
    {
      "epoch": 1.0401007013127135,
      "grad_norm": 6.180789413120241,
      "learning_rate": 2.925423779615591e-06,
      "loss": 0.0817,
      "step": 1446
    },
    {
      "epoch": 1.0408199964035245,
      "grad_norm": 6.7466227065199815,
      "learning_rate": 2.925317794479154e-06,
      "loss": 0.0836,
      "step": 1447
    },
    {
      "epoch": 1.0415392914943356,
      "grad_norm": 7.155673514386755,
      "learning_rate": 2.9252117360076146e-06,
      "loss": 0.0498,
      "step": 1448
    },
    {
      "epoch": 1.0422585865851466,
      "grad_norm": 5.530935444079844,
      "learning_rate": 2.9251056042064297e-06,
      "loss": 0.1739,
      "step": 1449
    },
    {
      "epoch": 1.0429778816759576,
      "grad_norm": 4.001087623476624,
      "learning_rate": 2.9249993990810598e-06,
      "loss": 0.1431,
      "step": 1450
    },
    {
      "epoch": 1.0436971767667687,
      "grad_norm": 1.0575289350855142,
      "learning_rate": 2.924893120636969e-06,
      "loss": 0.0061,
      "step": 1451
    },
    {
      "epoch": 1.0444164718575795,
      "grad_norm": 3.468307518976385,
      "learning_rate": 2.9247867688796257e-06,
      "loss": 0.1044,
      "step": 1452
    },
    {
      "epoch": 1.0451357669483905,
      "grad_norm": 6.469066752340097,
      "learning_rate": 2.9246803438145023e-06,
      "loss": 0.0339,
      "step": 1453
    },
    {
      "epoch": 1.0458550620392015,
      "grad_norm": 6.6049055395458955,
      "learning_rate": 2.9245738454470744e-06,
      "loss": 0.0855,
      "step": 1454
    },
    {
      "epoch": 1.0465743571300126,
      "grad_norm": 2.1504829050347958,
      "learning_rate": 2.924467273782821e-06,
      "loss": 0.0749,
      "step": 1455
    },
    {
      "epoch": 1.0472936522208236,
      "grad_norm": 3.8041312945012264,
      "learning_rate": 2.9243606288272257e-06,
      "loss": 0.1165,
      "step": 1456
    },
    {
      "epoch": 1.0480129473116346,
      "grad_norm": 7.742287880125306,
      "learning_rate": 2.924253910585776e-06,
      "loss": 0.1575,
      "step": 1457
    },
    {
      "epoch": 1.0487322424024457,
      "grad_norm": 0.7523721604658536,
      "learning_rate": 2.9241471190639624e-06,
      "loss": 0.0028,
      "step": 1458
    },
    {
      "epoch": 1.0494515374932567,
      "grad_norm": 3.5971517352750255,
      "learning_rate": 2.924040254267279e-06,
      "loss": 0.0175,
      "step": 1459
    },
    {
      "epoch": 1.0501708325840675,
      "grad_norm": 4.0900087429273855,
      "learning_rate": 2.9239333162012256e-06,
      "loss": 0.0356,
      "step": 1460
    },
    {
      "epoch": 1.0508901276748785,
      "grad_norm": 5.512019872798274,
      "learning_rate": 2.923826304871303e-06,
      "loss": 0.2396,
      "step": 1461
    },
    {
      "epoch": 1.0516094227656896,
      "grad_norm": 4.543217051628713,
      "learning_rate": 2.9237192202830173e-06,
      "loss": 0.0946,
      "step": 1462
    },
    {
      "epoch": 1.0523287178565006,
      "grad_norm": 5.630736402167831,
      "learning_rate": 2.9236120624418786e-06,
      "loss": 0.1906,
      "step": 1463
    },
    {
      "epoch": 1.0530480129473117,
      "grad_norm": 4.3672323352831794,
      "learning_rate": 2.9235048313534007e-06,
      "loss": 0.0378,
      "step": 1464
    },
    {
      "epoch": 1.0537673080381227,
      "grad_norm": 5.25257814040437,
      "learning_rate": 2.9233975270230997e-06,
      "loss": 0.2043,
      "step": 1465
    },
    {
      "epoch": 1.0544866031289337,
      "grad_norm": 3.5113107174612366,
      "learning_rate": 2.9232901494564976e-06,
      "loss": 0.0094,
      "step": 1466
    },
    {
      "epoch": 1.0552058982197448,
      "grad_norm": 3.717395206264605,
      "learning_rate": 2.9231826986591186e-06,
      "loss": 0.0953,
      "step": 1467
    },
    {
      "epoch": 1.0559251933105556,
      "grad_norm": 4.042386265159079,
      "learning_rate": 2.923075174636492e-06,
      "loss": 0.1294,
      "step": 1468
    },
    {
      "epoch": 1.0566444884013666,
      "grad_norm": 4.6509046800507825,
      "learning_rate": 2.9229675773941484e-06,
      "loss": 0.0603,
      "step": 1469
    },
    {
      "epoch": 1.0573637834921776,
      "grad_norm": 0.7541291078948386,
      "learning_rate": 2.9228599069376257e-06,
      "loss": 0.0055,
      "step": 1470
    },
    {
      "epoch": 1.0580830785829887,
      "grad_norm": 5.113995253185113,
      "learning_rate": 2.922752163272463e-06,
      "loss": 0.1465,
      "step": 1471
    },
    {
      "epoch": 1.0588023736737997,
      "grad_norm": 5.425080566314265,
      "learning_rate": 2.922644346404204e-06,
      "loss": 0.1132,
      "step": 1472
    },
    {
      "epoch": 1.0595216687646107,
      "grad_norm": 3.1178686089635117,
      "learning_rate": 2.922536456338396e-06,
      "loss": 0.0899,
      "step": 1473
    },
    {
      "epoch": 1.0602409638554218,
      "grad_norm": 4.261433285723393,
      "learning_rate": 2.92242849308059e-06,
      "loss": 0.1209,
      "step": 1474
    },
    {
      "epoch": 1.0609602589462326,
      "grad_norm": 5.058830564921074,
      "learning_rate": 2.9223204566363416e-06,
      "loss": 0.2113,
      "step": 1475
    },
    {
      "epoch": 1.0616795540370436,
      "grad_norm": 4.372576214669597,
      "learning_rate": 2.922212347011208e-06,
      "loss": 0.0579,
      "step": 1476
    },
    {
      "epoch": 1.0623988491278546,
      "grad_norm": 2.6226259871011286,
      "learning_rate": 2.922104164210753e-06,
      "loss": 0.0617,
      "step": 1477
    },
    {
      "epoch": 1.0631181442186657,
      "grad_norm": 5.23400879076255,
      "learning_rate": 2.9219959082405422e-06,
      "loss": 0.1924,
      "step": 1478
    },
    {
      "epoch": 1.0638374393094767,
      "grad_norm": 1.9934632681874618,
      "learning_rate": 2.921887579106146e-06,
      "loss": 0.0348,
      "step": 1479
    },
    {
      "epoch": 1.0645567344002878,
      "grad_norm": 5.656383427397238,
      "learning_rate": 2.921779176813137e-06,
      "loss": 0.2216,
      "step": 1480
    },
    {
      "epoch": 1.0652760294910988,
      "grad_norm": 5.449406362596943,
      "learning_rate": 2.9216707013670946e-06,
      "loss": 0.2164,
      "step": 1481
    },
    {
      "epoch": 1.0659953245819098,
      "grad_norm": 4.958180971987771,
      "learning_rate": 2.9215621527735986e-06,
      "loss": 0.038,
      "step": 1482
    },
    {
      "epoch": 1.0667146196727206,
      "grad_norm": 4.064581350276271,
      "learning_rate": 2.921453531038234e-06,
      "loss": 0.2205,
      "step": 1483
    },
    {
      "epoch": 1.0674339147635317,
      "grad_norm": 4.031473562279539,
      "learning_rate": 2.92134483616659e-06,
      "loss": 0.1879,
      "step": 1484
    },
    {
      "epoch": 1.0681532098543427,
      "grad_norm": 6.274310679738218,
      "learning_rate": 2.9212360681642592e-06,
      "loss": 0.2501,
      "step": 1485
    },
    {
      "epoch": 1.0688725049451537,
      "grad_norm": 6.095003922119308,
      "learning_rate": 2.921127227036838e-06,
      "loss": 0.0938,
      "step": 1486
    },
    {
      "epoch": 1.0695918000359648,
      "grad_norm": 5.232846560734211,
      "learning_rate": 2.921018312789926e-06,
      "loss": 0.1341,
      "step": 1487
    },
    {
      "epoch": 1.0703110951267758,
      "grad_norm": 5.001592695540452,
      "learning_rate": 2.9209093254291273e-06,
      "loss": 0.1873,
      "step": 1488
    },
    {
      "epoch": 1.0710303902175868,
      "grad_norm": 4.035141328939517,
      "learning_rate": 2.9208002649600493e-06,
      "loss": 0.1061,
      "step": 1489
    },
    {
      "epoch": 1.0717496853083979,
      "grad_norm": 4.649097168360496,
      "learning_rate": 2.920691131388304e-06,
      "loss": 0.0712,
      "step": 1490
    },
    {
      "epoch": 1.0724689803992087,
      "grad_norm": 4.481316157897854,
      "learning_rate": 2.920581924719506e-06,
      "loss": 0.0394,
      "step": 1491
    },
    {
      "epoch": 1.0731882754900197,
      "grad_norm": 6.06525938989988,
      "learning_rate": 2.920472644959274e-06,
      "loss": 0.0963,
      "step": 1492
    },
    {
      "epoch": 1.0739075705808308,
      "grad_norm": 7.427434280609504,
      "learning_rate": 2.920363292113231e-06,
      "loss": 0.3338,
      "step": 1493
    },
    {
      "epoch": 1.0746268656716418,
      "grad_norm": 3.4303618319189217,
      "learning_rate": 2.9202538661870032e-06,
      "loss": 0.0399,
      "step": 1494
    },
    {
      "epoch": 1.0753461607624528,
      "grad_norm": 2.731194643778459,
      "learning_rate": 2.9201443671862205e-06,
      "loss": 0.0376,
      "step": 1495
    },
    {
      "epoch": 1.0760654558532639,
      "grad_norm": 3.306613701716555,
      "learning_rate": 2.920034795116517e-06,
      "loss": 0.0546,
      "step": 1496
    },
    {
      "epoch": 1.0767847509440749,
      "grad_norm": 2.8839141253524576,
      "learning_rate": 2.9199251499835313e-06,
      "loss": 0.0267,
      "step": 1497
    },
    {
      "epoch": 1.077504046034886,
      "grad_norm": 6.925996889433103,
      "learning_rate": 2.9198154317929035e-06,
      "loss": 0.2802,
      "step": 1498
    },
    {
      "epoch": 1.0782233411256967,
      "grad_norm": 3.489081656265809,
      "learning_rate": 2.9197056405502795e-06,
      "loss": 0.0969,
      "step": 1499
    },
    {
      "epoch": 1.0789426362165078,
      "grad_norm": 5.419614479280475,
      "learning_rate": 2.9195957762613083e-06,
      "loss": 0.0788,
      "step": 1500
    },
    {
      "epoch": 1.0796619313073188,
      "grad_norm": 5.577798786356071,
      "learning_rate": 2.9194858389316416e-06,
      "loss": 0.1749,
      "step": 1501
    },
    {
      "epoch": 1.0803812263981298,
      "grad_norm": 4.539753135639085,
      "learning_rate": 2.9193758285669373e-06,
      "loss": 0.0986,
      "step": 1502
    },
    {
      "epoch": 1.0811005214889409,
      "grad_norm": 4.9256584195434066,
      "learning_rate": 2.9192657451728547e-06,
      "loss": 0.1503,
      "step": 1503
    },
    {
      "epoch": 1.081819816579752,
      "grad_norm": 3.3280021763159007,
      "learning_rate": 2.9191555887550585e-06,
      "loss": 0.1057,
      "step": 1504
    },
    {
      "epoch": 1.082539111670563,
      "grad_norm": 3.263113229654212,
      "learning_rate": 2.9190453593192156e-06,
      "loss": 0.0222,
      "step": 1505
    },
    {
      "epoch": 1.0832584067613737,
      "grad_norm": 3.839295795800084,
      "learning_rate": 2.918935056870998e-06,
      "loss": 0.131,
      "step": 1506
    },
    {
      "epoch": 1.0839777018521848,
      "grad_norm": 3.4267481683413163,
      "learning_rate": 2.9188246814160808e-06,
      "loss": 0.0918,
      "step": 1507
    },
    {
      "epoch": 1.0846969969429958,
      "grad_norm": 8.627027513030217,
      "learning_rate": 2.9187142329601437e-06,
      "loss": 0.3319,
      "step": 1508
    },
    {
      "epoch": 1.0854162920338069,
      "grad_norm": 5.16670900444957,
      "learning_rate": 2.918603711508868e-06,
      "loss": 0.1224,
      "step": 1509
    },
    {
      "epoch": 1.0861355871246179,
      "grad_norm": 6.5169930251932655,
      "learning_rate": 2.9184931170679414e-06,
      "loss": 0.0295,
      "step": 1510
    },
    {
      "epoch": 1.086854882215429,
      "grad_norm": 7.975271216831541,
      "learning_rate": 2.9183824496430535e-06,
      "loss": 0.0333,
      "step": 1511
    },
    {
      "epoch": 1.08757417730624,
      "grad_norm": 3.6150544552242723,
      "learning_rate": 2.918271709239899e-06,
      "loss": 0.1581,
      "step": 1512
    },
    {
      "epoch": 1.088293472397051,
      "grad_norm": 3.736523644745436,
      "learning_rate": 2.9181608958641756e-06,
      "loss": 0.0496,
      "step": 1513
    },
    {
      "epoch": 1.0890127674878618,
      "grad_norm": 3.429287413290873,
      "learning_rate": 2.918050009521584e-06,
      "loss": 0.0189,
      "step": 1514
    },
    {
      "epoch": 1.0897320625786728,
      "grad_norm": 5.575831110927186,
      "learning_rate": 2.9179390502178307e-06,
      "loss": 0.2109,
      "step": 1515
    },
    {
      "epoch": 1.0904513576694839,
      "grad_norm": 7.013230445251674,
      "learning_rate": 2.917828017958624e-06,
      "loss": 0.0752,
      "step": 1516
    },
    {
      "epoch": 1.091170652760295,
      "grad_norm": 2.9801877393944856,
      "learning_rate": 2.9177169127496766e-06,
      "loss": 0.1064,
      "step": 1517
    },
    {
      "epoch": 1.091889947851106,
      "grad_norm": 3.174139546542005,
      "learning_rate": 2.917605734596705e-06,
      "loss": 0.021,
      "step": 1518
    },
    {
      "epoch": 1.092609242941917,
      "grad_norm": 4.479997482667637,
      "learning_rate": 2.9174944835054305e-06,
      "loss": 0.1909,
      "step": 1519
    },
    {
      "epoch": 1.093328538032728,
      "grad_norm": 5.009655545194303,
      "learning_rate": 2.9173831594815766e-06,
      "loss": 0.0836,
      "step": 1520
    },
    {
      "epoch": 1.0940478331235388,
      "grad_norm": 4.451020554627839,
      "learning_rate": 2.917271762530871e-06,
      "loss": 0.0985,
      "step": 1521
    },
    {
      "epoch": 1.0947671282143498,
      "grad_norm": 6.125687542248357,
      "learning_rate": 2.9171602926590447e-06,
      "loss": 0.1761,
      "step": 1522
    },
    {
      "epoch": 1.0954864233051609,
      "grad_norm": 2.894633391894206,
      "learning_rate": 2.917048749871834e-06,
      "loss": 0.0549,
      "step": 1523
    },
    {
      "epoch": 1.096205718395972,
      "grad_norm": 5.311210867302218,
      "learning_rate": 2.9169371341749777e-06,
      "loss": 0.2164,
      "step": 1524
    },
    {
      "epoch": 1.096925013486783,
      "grad_norm": 2.7861491703189896,
      "learning_rate": 2.916825445574218e-06,
      "loss": 0.1039,
      "step": 1525
    },
    {
      "epoch": 1.097644308577594,
      "grad_norm": 4.814114699571843,
      "learning_rate": 2.916713684075302e-06,
      "loss": 0.2066,
      "step": 1526
    },
    {
      "epoch": 1.098363603668405,
      "grad_norm": 5.544663014767644,
      "learning_rate": 2.9166018496839804e-06,
      "loss": 0.3047,
      "step": 1527
    },
    {
      "epoch": 1.099082898759216,
      "grad_norm": 4.202365550843018,
      "learning_rate": 2.9164899424060063e-06,
      "loss": 0.1997,
      "step": 1528
    },
    {
      "epoch": 1.099802193850027,
      "grad_norm": 3.2860409072834855,
      "learning_rate": 2.916377962247139e-06,
      "loss": 0.0212,
      "step": 1529
    },
    {
      "epoch": 1.100521488940838,
      "grad_norm": 5.201999634671689,
      "learning_rate": 2.9162659092131387e-06,
      "loss": 0.0553,
      "step": 1530
    },
    {
      "epoch": 1.101240784031649,
      "grad_norm": 4.6061959025854735,
      "learning_rate": 2.916153783309771e-06,
      "loss": 0.0182,
      "step": 1531
    },
    {
      "epoch": 1.10196007912246,
      "grad_norm": 6.418190686183614,
      "learning_rate": 2.916041584542805e-06,
      "loss": 0.1723,
      "step": 1532
    },
    {
      "epoch": 1.102679374213271,
      "grad_norm": 4.424650205317266,
      "learning_rate": 2.9159293129180142e-06,
      "loss": 0.2243,
      "step": 1533
    },
    {
      "epoch": 1.103398669304082,
      "grad_norm": 3.819294914050668,
      "learning_rate": 2.9158169684411744e-06,
      "loss": 0.1249,
      "step": 1534
    },
    {
      "epoch": 1.104117964394893,
      "grad_norm": 3.609413402564569,
      "learning_rate": 2.915704551118066e-06,
      "loss": 0.1149,
      "step": 1535
    },
    {
      "epoch": 1.104837259485704,
      "grad_norm": 4.902049742689961,
      "learning_rate": 2.9155920609544737e-06,
      "loss": 0.1057,
      "step": 1536
    },
    {
      "epoch": 1.105556554576515,
      "grad_norm": 4.975100067862964,
      "learning_rate": 2.915479497956185e-06,
      "loss": 0.1452,
      "step": 1537
    },
    {
      "epoch": 1.106275849667326,
      "grad_norm": 2.0901599900076078,
      "learning_rate": 2.9153668621289903e-06,
      "loss": 0.0066,
      "step": 1538
    },
    {
      "epoch": 1.106995144758137,
      "grad_norm": 5.892322293288568,
      "learning_rate": 2.9152541534786866e-06,
      "loss": 0.0551,
      "step": 1539
    },
    {
      "epoch": 1.107714439848948,
      "grad_norm": 6.262369229039697,
      "learning_rate": 2.9151413720110724e-06,
      "loss": 0.1587,
      "step": 1540
    },
    {
      "epoch": 1.108433734939759,
      "grad_norm": 5.7664315663265135,
      "learning_rate": 2.915028517731951e-06,
      "loss": 0.1796,
      "step": 1541
    },
    {
      "epoch": 1.10915303003057,
      "grad_norm": 7.455060344039463,
      "learning_rate": 2.9149155906471275e-06,
      "loss": 0.316,
      "step": 1542
    },
    {
      "epoch": 1.1098723251213811,
      "grad_norm": 7.1115107586720665,
      "learning_rate": 2.914802590762413e-06,
      "loss": 0.2119,
      "step": 1543
    },
    {
      "epoch": 1.1105916202121922,
      "grad_norm": 5.886648515140201,
      "learning_rate": 2.9146895180836217e-06,
      "loss": 0.1851,
      "step": 1544
    },
    {
      "epoch": 1.111310915303003,
      "grad_norm": 2.813077492182065,
      "learning_rate": 2.9145763726165717e-06,
      "loss": 0.0108,
      "step": 1545
    },
    {
      "epoch": 1.112030210393814,
      "grad_norm": 2.622905487123166,
      "learning_rate": 2.914463154367084e-06,
      "loss": 0.0102,
      "step": 1546
    },
    {
      "epoch": 1.112749505484625,
      "grad_norm": 4.475410008524123,
      "learning_rate": 2.9143498633409836e-06,
      "loss": 0.2229,
      "step": 1547
    },
    {
      "epoch": 1.113468800575436,
      "grad_norm": 3.86739818321325,
      "learning_rate": 2.9142364995441e-06,
      "loss": 0.1164,
      "step": 1548
    },
    {
      "epoch": 1.114188095666247,
      "grad_norm": 9.08395473574008,
      "learning_rate": 2.914123062982266e-06,
      "loss": 0.3142,
      "step": 1549
    },
    {
      "epoch": 1.1149073907570581,
      "grad_norm": 5.945113354008717,
      "learning_rate": 2.9140095536613182e-06,
      "loss": 0.1653,
      "step": 1550
    },
    {
      "epoch": 1.1156266858478692,
      "grad_norm": 4.910511869490129,
      "learning_rate": 2.913895971587097e-06,
      "loss": 0.0962,
      "step": 1551
    },
    {
      "epoch": 1.11634598093868,
      "grad_norm": 3.6194086687598443,
      "learning_rate": 2.913782316765445e-06,
      "loss": 0.1011,
      "step": 1552
    },
    {
      "epoch": 1.117065276029491,
      "grad_norm": 4.9102983992662566,
      "learning_rate": 2.9136685892022118e-06,
      "loss": 0.1588,
      "step": 1553
    },
    {
      "epoch": 1.117784571120302,
      "grad_norm": 11.319662486227994,
      "learning_rate": 2.913554788903248e-06,
      "loss": 0.1685,
      "step": 1554
    },
    {
      "epoch": 1.118503866211113,
      "grad_norm": 10.794114707844374,
      "learning_rate": 2.913440915874408e-06,
      "loss": 0.1582,
      "step": 1555
    },
    {
      "epoch": 1.1192231613019241,
      "grad_norm": 5.640046685728013,
      "learning_rate": 2.9133269701215525e-06,
      "loss": 0.1814,
      "step": 1556
    },
    {
      "epoch": 1.1199424563927352,
      "grad_norm": 1.8783785292977917,
      "learning_rate": 2.9132129516505437e-06,
      "loss": 0.0466,
      "step": 1557
    },
    {
      "epoch": 1.1206617514835462,
      "grad_norm": 6.649623046313859,
      "learning_rate": 2.913098860467247e-06,
      "loss": 0.2591,
      "step": 1558
    },
    {
      "epoch": 1.1213810465743572,
      "grad_norm": 2.4541511087383463,
      "learning_rate": 2.9129846965775336e-06,
      "loss": 0.0437,
      "step": 1559
    },
    {
      "epoch": 1.122100341665168,
      "grad_norm": 4.196139146157071,
      "learning_rate": 2.9128704599872772e-06,
      "loss": 0.1225,
      "step": 1560
    },
    {
      "epoch": 1.122819636755979,
      "grad_norm": 5.6710068660214725,
      "learning_rate": 2.9127561507023553e-06,
      "loss": 0.1585,
      "step": 1561
    },
    {
      "epoch": 1.12353893184679,
      "grad_norm": 3.7293700063475366,
      "learning_rate": 2.9126417687286497e-06,
      "loss": 0.0693,
      "step": 1562
    },
    {
      "epoch": 1.1242582269376011,
      "grad_norm": 7.142216721266976,
      "learning_rate": 2.9125273140720446e-06,
      "loss": 0.2059,
      "step": 1563
    },
    {
      "epoch": 1.1249775220284122,
      "grad_norm": 1.502967925286762,
      "learning_rate": 2.91241278673843e-06,
      "loss": 0.0116,
      "step": 1564
    },
    {
      "epoch": 1.1256968171192232,
      "grad_norm": 7.837880717222229,
      "learning_rate": 2.9122981867336983e-06,
      "loss": 0.1559,
      "step": 1565
    },
    {
      "epoch": 1.1264161122100342,
      "grad_norm": 1.7459042863174874,
      "learning_rate": 2.9121835140637455e-06,
      "loss": 0.0386,
      "step": 1566
    },
    {
      "epoch": 1.127135407300845,
      "grad_norm": 8.00085869248901,
      "learning_rate": 2.912068768734472e-06,
      "loss": 0.2524,
      "step": 1567
    },
    {
      "epoch": 1.127854702391656,
      "grad_norm": 2.338797998044746,
      "learning_rate": 2.911953950751781e-06,
      "loss": 0.0073,
      "step": 1568
    },
    {
      "epoch": 1.1285739974824671,
      "grad_norm": 3.3131262445769454,
      "learning_rate": 2.9118390601215815e-06,
      "loss": 0.1398,
      "step": 1569
    },
    {
      "epoch": 1.1292932925732782,
      "grad_norm": 4.161046435348234,
      "learning_rate": 2.9117240968497834e-06,
      "loss": 0.1839,
      "step": 1570
    },
    {
      "epoch": 1.1300125876640892,
      "grad_norm": 2.831063301192099,
      "learning_rate": 2.911609060942302e-06,
      "loss": 0.0244,
      "step": 1571
    },
    {
      "epoch": 1.1307318827549002,
      "grad_norm": 8.83505975602995,
      "learning_rate": 2.911493952405057e-06,
      "loss": 0.1729,
      "step": 1572
    },
    {
      "epoch": 1.1314511778457113,
      "grad_norm": 5.834108132069269,
      "learning_rate": 2.9113787712439704e-06,
      "loss": 0.0456,
      "step": 1573
    },
    {
      "epoch": 1.1321704729365223,
      "grad_norm": 3.897330672700195,
      "learning_rate": 2.9112635174649682e-06,
      "loss": 0.1276,
      "step": 1574
    },
    {
      "epoch": 1.1328897680273333,
      "grad_norm": 5.150524497442603,
      "learning_rate": 2.9111481910739807e-06,
      "loss": 0.1107,
      "step": 1575
    },
    {
      "epoch": 1.1336090631181441,
      "grad_norm": 5.365956942451597,
      "learning_rate": 2.9110327920769416e-06,
      "loss": 0.2568,
      "step": 1576
    },
    {
      "epoch": 1.1343283582089552,
      "grad_norm": 10.071509649899602,
      "learning_rate": 2.9109173204797884e-06,
      "loss": 0.2087,
      "step": 1577
    },
    {
      "epoch": 1.1350476532997662,
      "grad_norm": 9.254387644492711,
      "learning_rate": 2.9108017762884624e-06,
      "loss": 0.2935,
      "step": 1578
    },
    {
      "epoch": 1.1357669483905772,
      "grad_norm": 8.097262911525858,
      "learning_rate": 2.910686159508908e-06,
      "loss": 0.1981,
      "step": 1579
    },
    {
      "epoch": 1.1364862434813883,
      "grad_norm": 2.787547442070557,
      "learning_rate": 2.9105704701470745e-06,
      "loss": 0.0113,
      "step": 1580
    },
    {
      "epoch": 1.1372055385721993,
      "grad_norm": 2.557056387541306,
      "learning_rate": 2.910454708208914e-06,
      "loss": 0.0226,
      "step": 1581
    },
    {
      "epoch": 1.1379248336630103,
      "grad_norm": 4.709660866454045,
      "learning_rate": 2.9103388737003825e-06,
      "loss": 0.1623,
      "step": 1582
    },
    {
      "epoch": 1.1386441287538211,
      "grad_norm": 5.3638953598016474,
      "learning_rate": 2.910222966627441e-06,
      "loss": 0.2392,
      "step": 1583
    },
    {
      "epoch": 1.1393634238446322,
      "grad_norm": 5.617800416822645,
      "learning_rate": 2.910106986996052e-06,
      "loss": 0.0265,
      "step": 1584
    },
    {
      "epoch": 1.1400827189354432,
      "grad_norm": 7.406853221393165,
      "learning_rate": 2.909990934812183e-06,
      "loss": 0.0512,
      "step": 1585
    },
    {
      "epoch": 1.1408020140262543,
      "grad_norm": 5.258026027066331,
      "learning_rate": 2.9098748100818054e-06,
      "loss": 0.1145,
      "step": 1586
    },
    {
      "epoch": 1.1415213091170653,
      "grad_norm": 5.309413136417267,
      "learning_rate": 2.9097586128108933e-06,
      "loss": 0.158,
      "step": 1587
    },
    {
      "epoch": 1.1422406042078763,
      "grad_norm": 4.8261048926627605,
      "learning_rate": 2.9096423430054265e-06,
      "loss": 0.2778,
      "step": 1588
    },
    {
      "epoch": 1.1429598992986874,
      "grad_norm": 7.625622314728041,
      "learning_rate": 2.9095260006713864e-06,
      "loss": 0.2095,
      "step": 1589
    },
    {
      "epoch": 1.1436791943894984,
      "grad_norm": 5.149380534623682,
      "learning_rate": 2.9094095858147593e-06,
      "loss": 0.0574,
      "step": 1590
    },
    {
      "epoch": 1.1443984894803092,
      "grad_norm": 4.041501218959745,
      "learning_rate": 2.9092930984415345e-06,
      "loss": 0.0429,
      "step": 1591
    },
    {
      "epoch": 1.1451177845711202,
      "grad_norm": 8.74005471457248,
      "learning_rate": 2.9091765385577062e-06,
      "loss": 0.2577,
      "step": 1592
    },
    {
      "epoch": 1.1458370796619313,
      "grad_norm": 6.140724726940976,
      "learning_rate": 2.9090599061692713e-06,
      "loss": 0.0325,
      "step": 1593
    },
    {
      "epoch": 1.1465563747527423,
      "grad_norm": 8.44877172666737,
      "learning_rate": 2.9089432012822302e-06,
      "loss": 0.2595,
      "step": 1594
    },
    {
      "epoch": 1.1472756698435533,
      "grad_norm": 3.3033147905789986,
      "learning_rate": 2.908826423902589e-06,
      "loss": 0.0529,
      "step": 1595
    },
    {
      "epoch": 1.1479949649343644,
      "grad_norm": 3.8281522604427707,
      "learning_rate": 2.9087095740363546e-06,
      "loss": 0.0403,
      "step": 1596
    },
    {
      "epoch": 1.1487142600251754,
      "grad_norm": 3.3425254006305978,
      "learning_rate": 2.9085926516895396e-06,
      "loss": 0.07,
      "step": 1597
    },
    {
      "epoch": 1.1494335551159862,
      "grad_norm": 7.303284480632762,
      "learning_rate": 2.90847565686816e-06,
      "loss": 0.1798,
      "step": 1598
    },
    {
      "epoch": 1.1501528502067973,
      "grad_norm": 5.195745851298197,
      "learning_rate": 2.908358589578235e-06,
      "loss": 0.1652,
      "step": 1599
    },
    {
      "epoch": 1.1508721452976083,
      "grad_norm": 4.128539758666629,
      "learning_rate": 2.9082414498257892e-06,
      "loss": 0.2128,
      "step": 1600
    },
    {
      "epoch": 1.1515914403884193,
      "grad_norm": 1.2779722479038704,
      "learning_rate": 2.9081242376168482e-06,
      "loss": 0.0285,
      "step": 1601
    },
    {
      "epoch": 1.1523107354792304,
      "grad_norm": 9.234725656767678,
      "learning_rate": 2.9080069529574434e-06,
      "loss": 0.2444,
      "step": 1602
    },
    {
      "epoch": 1.1530300305700414,
      "grad_norm": 3.0457630925252355,
      "learning_rate": 2.907889595853609e-06,
      "loss": 0.0849,
      "step": 1603
    },
    {
      "epoch": 1.1537493256608524,
      "grad_norm": 6.307294480679254,
      "learning_rate": 2.9077721663113837e-06,
      "loss": 0.2584,
      "step": 1604
    },
    {
      "epoch": 1.1544686207516635,
      "grad_norm": 3.7467935991572134,
      "learning_rate": 2.9076546643368087e-06,
      "loss": 0.0898,
      "step": 1605
    },
    {
      "epoch": 1.1551879158424745,
      "grad_norm": 7.174854215821884,
      "learning_rate": 2.9075370899359306e-06,
      "loss": 0.2132,
      "step": 1606
    },
    {
      "epoch": 1.1559072109332853,
      "grad_norm": 5.258392544012779,
      "learning_rate": 2.907419443114798e-06,
      "loss": 0.1903,
      "step": 1607
    },
    {
      "epoch": 1.1566265060240963,
      "grad_norm": 4.88702945614562,
      "learning_rate": 2.9073017238794647e-06,
      "loss": 0.1104,
      "step": 1608
    },
    {
      "epoch": 1.1573458011149074,
      "grad_norm": 5.9137560242249005,
      "learning_rate": 2.9071839322359872e-06,
      "loss": 0.0195,
      "step": 1609
    },
    {
      "epoch": 1.1580650962057184,
      "grad_norm": 5.631158851072675,
      "learning_rate": 2.9070660681904264e-06,
      "loss": 0.1812,
      "step": 1610
    },
    {
      "epoch": 1.1587843912965294,
      "grad_norm": 4.279208120920692,
      "learning_rate": 2.906948131748846e-06,
      "loss": 0.1262,
      "step": 1611
    },
    {
      "epoch": 1.1595036863873405,
      "grad_norm": 6.257222694408051,
      "learning_rate": 2.9068301229173147e-06,
      "loss": 0.0703,
      "step": 1612
    },
    {
      "epoch": 1.1602229814781515,
      "grad_norm": 0.2032081337469878,
      "learning_rate": 2.9067120417019036e-06,
      "loss": 0.0008,
      "step": 1613
    },
    {
      "epoch": 1.1609422765689623,
      "grad_norm": 3.1377080755271,
      "learning_rate": 2.906593888108689e-06,
      "loss": 0.023,
      "step": 1614
    },
    {
      "epoch": 1.1616615716597734,
      "grad_norm": 5.02787057365801,
      "learning_rate": 2.906475662143749e-06,
      "loss": 0.1807,
      "step": 1615
    },
    {
      "epoch": 1.1623808667505844,
      "grad_norm": 5.467161484946802,
      "learning_rate": 2.906357363813168e-06,
      "loss": 0.2285,
      "step": 1616
    },
    {
      "epoch": 1.1631001618413954,
      "grad_norm": 7.3490352375187245,
      "learning_rate": 2.9062389931230312e-06,
      "loss": 0.1174,
      "step": 1617
    },
    {
      "epoch": 1.1638194569322065,
      "grad_norm": 6.908168157345872,
      "learning_rate": 2.90612055007943e-06,
      "loss": 0.2132,
      "step": 1618
    },
    {
      "epoch": 1.1645387520230175,
      "grad_norm": 7.707772252927093,
      "learning_rate": 2.906002034688458e-06,
      "loss": 0.0638,
      "step": 1619
    },
    {
      "epoch": 1.1652580471138285,
      "grad_norm": 4.193829636563466,
      "learning_rate": 2.9058834469562132e-06,
      "loss": 0.0434,
      "step": 1620
    },
    {
      "epoch": 1.1659773422046396,
      "grad_norm": 5.047942454099802,
      "learning_rate": 2.905764786888797e-06,
      "loss": 0.192,
      "step": 1621
    },
    {
      "epoch": 1.1666966372954504,
      "grad_norm": 2.1721732650697443,
      "learning_rate": 2.9056460544923148e-06,
      "loss": 0.061,
      "step": 1622
    },
    {
      "epoch": 1.1674159323862614,
      "grad_norm": 4.716460384335506,
      "learning_rate": 2.9055272497728752e-06,
      "loss": 0.0689,
      "step": 1623
    },
    {
      "epoch": 1.1681352274770724,
      "grad_norm": 6.096225078287153,
      "learning_rate": 2.9054083727365918e-06,
      "loss": 0.1649,
      "step": 1624
    },
    {
      "epoch": 1.1688545225678835,
      "grad_norm": 2.728843418451714,
      "learning_rate": 2.9052894233895803e-06,
      "loss": 0.0826,
      "step": 1625
    },
    {
      "epoch": 1.1695738176586945,
      "grad_norm": 3.5483582819149637,
      "learning_rate": 2.9051704017379616e-06,
      "loss": 0.1121,
      "step": 1626
    },
    {
      "epoch": 1.1702931127495055,
      "grad_norm": 2.82822980608982,
      "learning_rate": 2.9050513077878584e-06,
      "loss": 0.0654,
      "step": 1627
    },
    {
      "epoch": 1.1710124078403166,
      "grad_norm": 6.554517102944529,
      "learning_rate": 2.904932141545399e-06,
      "loss": 0.3161,
      "step": 1628
    },
    {
      "epoch": 1.1717317029311274,
      "grad_norm": 4.840126659486563,
      "learning_rate": 2.904812903016715e-06,
      "loss": 0.0688,
      "step": 1629
    },
    {
      "epoch": 1.1724509980219384,
      "grad_norm": 6.832470797988187,
      "learning_rate": 2.9046935922079406e-06,
      "loss": 0.3091,
      "step": 1630
    },
    {
      "epoch": 1.1731702931127495,
      "grad_norm": 3.863683949245649,
      "learning_rate": 2.9045742091252152e-06,
      "loss": 0.0981,
      "step": 1631
    },
    {
      "epoch": 1.1738895882035605,
      "grad_norm": 3.9288208803046483,
      "learning_rate": 2.904454753774681e-06,
      "loss": 0.0989,
      "step": 1632
    },
    {
      "epoch": 1.1746088832943715,
      "grad_norm": 3.7116885465434852,
      "learning_rate": 2.9043352261624846e-06,
      "loss": 0.0658,
      "step": 1633
    },
    {
      "epoch": 1.1753281783851826,
      "grad_norm": 4.45888215501642,
      "learning_rate": 2.9042156262947753e-06,
      "loss": 0.1439,
      "step": 1634
    },
    {
      "epoch": 1.1760474734759936,
      "grad_norm": 5.3579535942791265,
      "learning_rate": 2.9040959541777073e-06,
      "loss": 0.084,
      "step": 1635
    },
    {
      "epoch": 1.1767667685668046,
      "grad_norm": 5.018282903143844,
      "learning_rate": 2.9039762098174372e-06,
      "loss": 0.1068,
      "step": 1636
    },
    {
      "epoch": 1.1774860636576157,
      "grad_norm": 2.7803505311432417,
      "learning_rate": 2.903856393220127e-06,
      "loss": 0.0379,
      "step": 1637
    },
    {
      "epoch": 1.1782053587484265,
      "grad_norm": 2.0191149948054914,
      "learning_rate": 2.9037365043919405e-06,
      "loss": 0.0148,
      "step": 1638
    },
    {
      "epoch": 1.1789246538392375,
      "grad_norm": 7.588252475581475,
      "learning_rate": 2.9036165433390465e-06,
      "loss": 0.2215,
      "step": 1639
    },
    {
      "epoch": 1.1796439489300485,
      "grad_norm": 7.867204675293377,
      "learning_rate": 2.903496510067618e-06,
      "loss": 0.1564,
      "step": 1640
    },
    {
      "epoch": 1.1803632440208596,
      "grad_norm": 4.055991528870527,
      "learning_rate": 2.90337640458383e-06,
      "loss": 0.0736,
      "step": 1641
    },
    {
      "epoch": 1.1810825391116706,
      "grad_norm": 8.103835864067179,
      "learning_rate": 2.9032562268938625e-06,
      "loss": 0.2698,
      "step": 1642
    },
    {
      "epoch": 1.1818018342024816,
      "grad_norm": 4.384752734563741,
      "learning_rate": 2.9031359770038986e-06,
      "loss": 0.0974,
      "step": 1643
    },
    {
      "epoch": 1.1825211292932925,
      "grad_norm": 5.261961242773679,
      "learning_rate": 2.9030156549201258e-06,
      "loss": 0.1479,
      "step": 1644
    },
    {
      "epoch": 1.1832404243841035,
      "grad_norm": 5.305132852860056,
      "learning_rate": 2.902895260648735e-06,
      "loss": 0.1416,
      "step": 1645
    },
    {
      "epoch": 1.1839597194749145,
      "grad_norm": 3.4968084235009202,
      "learning_rate": 2.9027747941959194e-06,
      "loss": 0.0847,
      "step": 1646
    },
    {
      "epoch": 1.1846790145657256,
      "grad_norm": 4.627624467589781,
      "learning_rate": 2.9026542555678784e-06,
      "loss": 0.1882,
      "step": 1647
    },
    {
      "epoch": 1.1853983096565366,
      "grad_norm": 4.225286606857522,
      "learning_rate": 2.902533644770814e-06,
      "loss": 0.0803,
      "step": 1648
    },
    {
      "epoch": 1.1861176047473476,
      "grad_norm": 7.82031764626809,
      "learning_rate": 2.9024129618109314e-06,
      "loss": 0.2254,
      "step": 1649
    },
    {
      "epoch": 1.1868368998381587,
      "grad_norm": 1.7268987748539237,
      "learning_rate": 2.9022922066944404e-06,
      "loss": 0.0064,
      "step": 1650
    },
    {
      "epoch": 1.1875561949289697,
      "grad_norm": 6.649548549476015,
      "learning_rate": 2.9021713794275534e-06,
      "loss": 0.2012,
      "step": 1651
    },
    {
      "epoch": 1.1882754900197807,
      "grad_norm": 4.181289615962555,
      "learning_rate": 2.9020504800164876e-06,
      "loss": 0.0563,
      "step": 1652
    },
    {
      "epoch": 1.1889947851105915,
      "grad_norm": 4.3707111153336875,
      "learning_rate": 2.901929508467463e-06,
      "loss": 0.0199,
      "step": 1653
    },
    {
      "epoch": 1.1897140802014026,
      "grad_norm": 3.2651002102996634,
      "learning_rate": 2.9018084647867044e-06,
      "loss": 0.0697,
      "step": 1654
    },
    {
      "epoch": 1.1904333752922136,
      "grad_norm": 3.792175290629736,
      "learning_rate": 2.9016873489804397e-06,
      "loss": 0.1297,
      "step": 1655
    },
    {
      "epoch": 1.1911526703830246,
      "grad_norm": 5.088332277924237,
      "learning_rate": 2.9015661610549e-06,
      "loss": 0.1366,
      "step": 1656
    },
    {
      "epoch": 1.1918719654738357,
      "grad_norm": 5.444267509245161,
      "learning_rate": 2.9014449010163215e-06,
      "loss": 0.1073,
      "step": 1657
    },
    {
      "epoch": 1.1925912605646467,
      "grad_norm": 6.90261058553411,
      "learning_rate": 2.9013235688709417e-06,
      "loss": 0.1746,
      "step": 1658
    },
    {
      "epoch": 1.1933105556554577,
      "grad_norm": 7.2030895963938715,
      "learning_rate": 2.901202164625005e-06,
      "loss": 0.0453,
      "step": 1659
    },
    {
      "epoch": 1.1940298507462686,
      "grad_norm": 0.295911356265283,
      "learning_rate": 2.901080688284757e-06,
      "loss": 0.0011,
      "step": 1660
    },
    {
      "epoch": 1.1947491458370796,
      "grad_norm": 5.592379349397983,
      "learning_rate": 2.9009591398564485e-06,
      "loss": 0.2711,
      "step": 1661
    },
    {
      "epoch": 1.1954684409278906,
      "grad_norm": 3.1551209109348144,
      "learning_rate": 2.9008375193463324e-06,
      "loss": 0.0928,
      "step": 1662
    },
    {
      "epoch": 1.1961877360187017,
      "grad_norm": 3.6384103828381433,
      "learning_rate": 2.900715826760667e-06,
      "loss": 0.0153,
      "step": 1663
    },
    {
      "epoch": 1.1969070311095127,
      "grad_norm": 6.72422480277707,
      "learning_rate": 2.9005940621057136e-06,
      "loss": 0.1839,
      "step": 1664
    },
    {
      "epoch": 1.1976263262003237,
      "grad_norm": 4.500247125539293,
      "learning_rate": 2.900472225387737e-06,
      "loss": 0.0792,
      "step": 1665
    },
    {
      "epoch": 1.1983456212911348,
      "grad_norm": 4.2434099752318994,
      "learning_rate": 2.9003503166130058e-06,
      "loss": 0.0755,
      "step": 1666
    },
    {
      "epoch": 1.1990649163819458,
      "grad_norm": 4.460320268967298,
      "learning_rate": 2.9002283357877925e-06,
      "loss": 0.0816,
      "step": 1667
    },
    {
      "epoch": 1.1997842114727568,
      "grad_norm": 5.604233029443637,
      "learning_rate": 2.900106282918373e-06,
      "loss": 0.3211,
      "step": 1668
    },
    {
      "epoch": 1.2005035065635676,
      "grad_norm": 4.807050427072252,
      "learning_rate": 2.8999841580110285e-06,
      "loss": 0.1609,
      "step": 1669
    },
    {
      "epoch": 1.2012228016543787,
      "grad_norm": 4.0434982571244875,
      "learning_rate": 2.899861961072041e-06,
      "loss": 0.0429,
      "step": 1670
    },
    {
      "epoch": 1.2019420967451897,
      "grad_norm": 0.2619300813377373,
      "learning_rate": 2.8997396921076976e-06,
      "loss": 0.0007,
      "step": 1671
    },
    {
      "epoch": 1.2026613918360007,
      "grad_norm": 6.572932135924672,
      "learning_rate": 2.8996173511242903e-06,
      "loss": 0.073,
      "step": 1672
    },
    {
      "epoch": 1.2033806869268118,
      "grad_norm": 6.528714417366292,
      "learning_rate": 2.8994949381281133e-06,
      "loss": 0.1391,
      "step": 1673
    },
    {
      "epoch": 1.2040999820176228,
      "grad_norm": 3.315380520396503,
      "learning_rate": 2.8993724531254655e-06,
      "loss": 0.0428,
      "step": 1674
    },
    {
      "epoch": 1.2048192771084336,
      "grad_norm": 4.1644101093246855,
      "learning_rate": 2.899249896122648e-06,
      "loss": 0.1788,
      "step": 1675
    },
    {
      "epoch": 1.2055385721992447,
      "grad_norm": 3.8480625075497272,
      "learning_rate": 2.899127267125967e-06,
      "loss": 0.077,
      "step": 1676
    },
    {
      "epoch": 1.2062578672900557,
      "grad_norm": 4.588933061483007,
      "learning_rate": 2.8990045661417323e-06,
      "loss": 0.2078,
      "step": 1677
    },
    {
      "epoch": 1.2069771623808667,
      "grad_norm": 3.8666833205546607,
      "learning_rate": 2.8988817931762568e-06,
      "loss": 0.0929,
      "step": 1678
    },
    {
      "epoch": 1.2076964574716778,
      "grad_norm": 5.676310045283477,
      "learning_rate": 2.898758948235857e-06,
      "loss": 0.2461,
      "step": 1679
    },
    {
      "epoch": 1.2084157525624888,
      "grad_norm": 5.637635799687778,
      "learning_rate": 2.898636031326854e-06,
      "loss": 0.2119,
      "step": 1680
    },
    {
      "epoch": 1.2091350476532998,
      "grad_norm": 4.460716115876083,
      "learning_rate": 2.898513042455572e-06,
      "loss": 0.2893,
      "step": 1681
    },
    {
      "epoch": 1.2098543427441109,
      "grad_norm": 3.600088991992017,
      "learning_rate": 2.898389981628339e-06,
      "loss": 0.0759,
      "step": 1682
    },
    {
      "epoch": 1.210573637834922,
      "grad_norm": 2.6545082260260515,
      "learning_rate": 2.898266848851486e-06,
      "loss": 0.065,
      "step": 1683
    },
    {
      "epoch": 1.2112929329257327,
      "grad_norm": 6.046021847605433,
      "learning_rate": 2.8981436441313504e-06,
      "loss": 0.1142,
      "step": 1684
    },
    {
      "epoch": 1.2120122280165437,
      "grad_norm": 2.3560799867125826,
      "learning_rate": 2.8980203674742693e-06,
      "loss": 0.0451,
      "step": 1685
    },
    {
      "epoch": 1.2127315231073548,
      "grad_norm": 5.070365155320128,
      "learning_rate": 2.897897018886586e-06,
      "loss": 0.1977,
      "step": 1686
    },
    {
      "epoch": 1.2134508181981658,
      "grad_norm": 6.724096286192235,
      "learning_rate": 2.8977735983746467e-06,
      "loss": 0.2726,
      "step": 1687
    },
    {
      "epoch": 1.2141701132889768,
      "grad_norm": 5.338627161316427,
      "learning_rate": 2.8976501059448025e-06,
      "loss": 0.0556,
      "step": 1688
    },
    {
      "epoch": 1.2148894083797879,
      "grad_norm": 4.0449252993987885,
      "learning_rate": 2.897526541603407e-06,
      "loss": 0.0289,
      "step": 1689
    },
    {
      "epoch": 1.215608703470599,
      "grad_norm": 4.4576782854327535,
      "learning_rate": 2.897402905356818e-06,
      "loss": 0.0652,
      "step": 1690
    },
    {
      "epoch": 1.2163279985614097,
      "grad_norm": 4.602794051716955,
      "learning_rate": 2.8972791972113957e-06,
      "loss": 0.0341,
      "step": 1691
    },
    {
      "epoch": 1.2170472936522208,
      "grad_norm": 3.724446979806658,
      "learning_rate": 2.897155417173506e-06,
      "loss": 0.099,
      "step": 1692
    },
    {
      "epoch": 1.2177665887430318,
      "grad_norm": 4.986198057836993,
      "learning_rate": 2.897031565249518e-06,
      "loss": 0.1343,
      "step": 1693
    },
    {
      "epoch": 1.2184858838338428,
      "grad_norm": 5.981542866327953,
      "learning_rate": 2.896907641445803e-06,
      "loss": 0.1653,
      "step": 1694
    },
    {
      "epoch": 1.2192051789246539,
      "grad_norm": 5.443099139719888,
      "learning_rate": 2.8967836457687377e-06,
      "loss": 0.101,
      "step": 1695
    },
    {
      "epoch": 1.2199244740154649,
      "grad_norm": 7.664009635039305,
      "learning_rate": 2.896659578224702e-06,
      "loss": 0.1482,
      "step": 1696
    },
    {
      "epoch": 1.220643769106276,
      "grad_norm": 3.3137968050544084,
      "learning_rate": 2.8965354388200792e-06,
      "loss": 0.0754,
      "step": 1697
    },
    {
      "epoch": 1.221363064197087,
      "grad_norm": 5.193651429800962,
      "learning_rate": 2.896411227561257e-06,
      "loss": 0.1697,
      "step": 1698
    },
    {
      "epoch": 1.2220823592878978,
      "grad_norm": 5.331218352793793,
      "learning_rate": 2.8962869444546252e-06,
      "loss": 0.1576,
      "step": 1699
    },
    {
      "epoch": 1.2228016543787088,
      "grad_norm": 5.747028241965266,
      "learning_rate": 2.896162589506579e-06,
      "loss": 0.0535,
      "step": 1700
    },
    {
      "epoch": 1.2235209494695198,
      "grad_norm": 10.478785866039598,
      "learning_rate": 2.896038162723517e-06,
      "loss": 0.1801,
      "step": 1701
    },
    {
      "epoch": 1.2242402445603309,
      "grad_norm": 4.430856437026723,
      "learning_rate": 2.895913664111841e-06,
      "loss": 0.1079,
      "step": 1702
    },
    {
      "epoch": 1.224959539651142,
      "grad_norm": 4.38930395769076,
      "learning_rate": 2.8957890936779563e-06,
      "loss": 0.1367,
      "step": 1703
    },
    {
      "epoch": 1.225678834741953,
      "grad_norm": 7.706914834039335,
      "learning_rate": 2.895664451428272e-06,
      "loss": 0.292,
      "step": 1704
    },
    {
      "epoch": 1.226398129832764,
      "grad_norm": 7.317569259186702,
      "learning_rate": 2.8955397373692023e-06,
      "loss": 0.1357,
      "step": 1705
    },
    {
      "epoch": 1.2271174249235748,
      "grad_norm": 5.101754142556382,
      "learning_rate": 2.8954149515071635e-06,
      "loss": 0.2589,
      "step": 1706
    },
    {
      "epoch": 1.2278367200143858,
      "grad_norm": 2.6797096333382493,
      "learning_rate": 2.895290093848575e-06,
      "loss": 0.0541,
      "step": 1707
    },
    {
      "epoch": 1.2285560151051969,
      "grad_norm": 7.141590482261668,
      "learning_rate": 2.8951651643998623e-06,
      "loss": 0.1327,
      "step": 1708
    },
    {
      "epoch": 1.2292753101960079,
      "grad_norm": 3.5902350157397236,
      "learning_rate": 2.895040163167453e-06,
      "loss": 0.0376,
      "step": 1709
    },
    {
      "epoch": 1.229994605286819,
      "grad_norm": 3.947371518643215,
      "learning_rate": 2.8949150901577784e-06,
      "loss": 0.0526,
      "step": 1710
    },
    {
      "epoch": 1.23071390037763,
      "grad_norm": 4.036717449798842,
      "learning_rate": 2.894789945377273e-06,
      "loss": 0.0835,
      "step": 1711
    },
    {
      "epoch": 1.231433195468441,
      "grad_norm": 5.2361268322036665,
      "learning_rate": 2.894664728832377e-06,
      "loss": 0.0801,
      "step": 1712
    },
    {
      "epoch": 1.232152490559252,
      "grad_norm": 7.4235292974507665,
      "learning_rate": 2.8945394405295327e-06,
      "loss": 0.1383,
      "step": 1713
    },
    {
      "epoch": 1.232871785650063,
      "grad_norm": 6.6864392813312525,
      "learning_rate": 2.8944140804751855e-06,
      "loss": 0.1541,
      "step": 1714
    },
    {
      "epoch": 1.2335910807408739,
      "grad_norm": 5.6049479536392575,
      "learning_rate": 2.8942886486757865e-06,
      "loss": 0.0993,
      "step": 1715
    },
    {
      "epoch": 1.234310375831685,
      "grad_norm": 6.597159116914805,
      "learning_rate": 2.894163145137789e-06,
      "loss": 0.2109,
      "step": 1716
    },
    {
      "epoch": 1.235029670922496,
      "grad_norm": 8.59232018090716,
      "learning_rate": 2.89403756986765e-06,
      "loss": 0.1359,
      "step": 1717
    },
    {
      "epoch": 1.235748966013307,
      "grad_norm": 6.9354133161352065,
      "learning_rate": 2.893911922871831e-06,
      "loss": 0.0575,
      "step": 1718
    },
    {
      "epoch": 1.236468261104118,
      "grad_norm": 6.398491529279321,
      "learning_rate": 2.8937862041567965e-06,
      "loss": 0.3083,
      "step": 1719
    },
    {
      "epoch": 1.237187556194929,
      "grad_norm": 7.600526697434066,
      "learning_rate": 2.8936604137290152e-06,
      "loss": 0.158,
      "step": 1720
    },
    {
      "epoch": 1.2379068512857399,
      "grad_norm": 6.121493554226204,
      "learning_rate": 2.893534551594959e-06,
      "loss": 0.1952,
      "step": 1721
    },
    {
      "epoch": 1.2386261463765509,
      "grad_norm": 4.091168035155126,
      "learning_rate": 2.893408617761104e-06,
      "loss": 0.0896,
      "step": 1722
    },
    {
      "epoch": 1.239345441467362,
      "grad_norm": 5.762578515654643,
      "learning_rate": 2.8932826122339297e-06,
      "loss": 0.1906,
      "step": 1723
    },
    {
      "epoch": 1.240064736558173,
      "grad_norm": 8.26043874681714,
      "learning_rate": 2.893156535019919e-06,
      "loss": 0.1216,
      "step": 1724
    },
    {
      "epoch": 1.240784031648984,
      "grad_norm": 6.749779803615349,
      "learning_rate": 2.893030386125559e-06,
      "loss": 0.2953,
      "step": 1725
    },
    {
      "epoch": 1.241503326739795,
      "grad_norm": 3.848139590984846,
      "learning_rate": 2.89290416555734e-06,
      "loss": 0.1744,
      "step": 1726
    },
    {
      "epoch": 1.242222621830606,
      "grad_norm": 5.583602290170202,
      "learning_rate": 2.892777873321757e-06,
      "loss": 0.041,
      "step": 1727
    },
    {
      "epoch": 1.242941916921417,
      "grad_norm": 4.676847080727266,
      "learning_rate": 2.892651509425307e-06,
      "loss": 0.0645,
      "step": 1728
    },
    {
      "epoch": 1.2436612120122281,
      "grad_norm": 6.433773964091845,
      "learning_rate": 2.892525073874492e-06,
      "loss": 0.1588,
      "step": 1729
    },
    {
      "epoch": 1.244380507103039,
      "grad_norm": 6.444075682049979,
      "learning_rate": 2.892398566675818e-06,
      "loss": 0.0945,
      "step": 1730
    },
    {
      "epoch": 1.24509980219385,
      "grad_norm": 6.680322537514869,
      "learning_rate": 2.892271987835793e-06,
      "loss": 0.1826,
      "step": 1731
    },
    {
      "epoch": 1.245819097284661,
      "grad_norm": 0.3296360838826887,
      "learning_rate": 2.8921453373609307e-06,
      "loss": 0.0019,
      "step": 1732
    },
    {
      "epoch": 1.246538392375472,
      "grad_norm": 2.1011532810299163,
      "learning_rate": 2.8920186152577465e-06,
      "loss": 0.0274,
      "step": 1733
    },
    {
      "epoch": 1.247257687466283,
      "grad_norm": 6.976378875433132,
      "learning_rate": 2.8918918215327614e-06,
      "loss": 0.3516,
      "step": 1734
    },
    {
      "epoch": 1.247976982557094,
      "grad_norm": 6.291583287655501,
      "learning_rate": 2.8917649561924983e-06,
      "loss": 0.2794,
      "step": 1735
    },
    {
      "epoch": 1.2486962776479051,
      "grad_norm": 5.361044298982202,
      "learning_rate": 2.8916380192434846e-06,
      "loss": 0.1136,
      "step": 1736
    },
    {
      "epoch": 1.249415572738716,
      "grad_norm": 4.922418236009025,
      "learning_rate": 2.8915110106922526e-06,
      "loss": 0.1511,
      "step": 1737
    },
    {
      "epoch": 1.250134867829527,
      "grad_norm": 5.538833389035142,
      "learning_rate": 2.8913839305453367e-06,
      "loss": 0.085,
      "step": 1738
    },
    {
      "epoch": 1.250854162920338,
      "grad_norm": 2.4748071806339125,
      "learning_rate": 2.891256778809274e-06,
      "loss": 0.0171,
      "step": 1739
    },
    {
      "epoch": 1.251573458011149,
      "grad_norm": 3.3505386432754416,
      "learning_rate": 2.891129555490608e-06,
      "loss": 0.0495,
      "step": 1740
    },
    {
      "epoch": 1.25229275310196,
      "grad_norm": 6.994173889253759,
      "learning_rate": 2.891002260595885e-06,
      "loss": 0.2899,
      "step": 1741
    },
    {
      "epoch": 1.2530120481927711,
      "grad_norm": 6.602909639058268,
      "learning_rate": 2.8908748941316534e-06,
      "loss": 0.1698,
      "step": 1742
    },
    {
      "epoch": 1.2537313432835822,
      "grad_norm": 6.379299761467531,
      "learning_rate": 2.890747456104467e-06,
      "loss": 0.1662,
      "step": 1743
    },
    {
      "epoch": 1.2544506383743932,
      "grad_norm": 4.356813554599697,
      "learning_rate": 2.8906199465208825e-06,
      "loss": 0.0419,
      "step": 1744
    },
    {
      "epoch": 1.2551699334652042,
      "grad_norm": 8.794103807862856,
      "learning_rate": 2.8904923653874607e-06,
      "loss": 0.3719,
      "step": 1745
    },
    {
      "epoch": 1.255889228556015,
      "grad_norm": 8.530073217884308,
      "learning_rate": 2.890364712710766e-06,
      "loss": 0.1047,
      "step": 1746
    },
    {
      "epoch": 1.256608523646826,
      "grad_norm": 2.0734435101327624,
      "learning_rate": 2.8902369884973657e-06,
      "loss": 0.057,
      "step": 1747
    },
    {
      "epoch": 1.257327818737637,
      "grad_norm": 4.430281238214375,
      "learning_rate": 2.890109192753832e-06,
      "loss": 0.0498,
      "step": 1748
    },
    {
      "epoch": 1.2580471138284481,
      "grad_norm": 3.641779769524025,
      "learning_rate": 2.88998132548674e-06,
      "loss": 0.1163,
      "step": 1749
    },
    {
      "epoch": 1.2587664089192592,
      "grad_norm": 1.6740554612817447,
      "learning_rate": 2.889853386702669e-06,
      "loss": 0.0385,
      "step": 1750
    },
    {
      "epoch": 1.2594857040100702,
      "grad_norm": 2.500383046995515,
      "learning_rate": 2.889725376408201e-06,
      "loss": 0.0159,
      "step": 1751
    },
    {
      "epoch": 1.260204999100881,
      "grad_norm": 4.620359802476714,
      "learning_rate": 2.8895972946099234e-06,
      "loss": 0.1521,
      "step": 1752
    },
    {
      "epoch": 1.260924294191692,
      "grad_norm": 3.90366105756383,
      "learning_rate": 2.889469141314425e-06,
      "loss": 0.025,
      "step": 1753
    },
    {
      "epoch": 1.261643589282503,
      "grad_norm": 3.0382449177771913,
      "learning_rate": 2.8893409165283006e-06,
      "loss": 0.0666,
      "step": 1754
    },
    {
      "epoch": 1.2623628843733141,
      "grad_norm": 3.7165377644985442,
      "learning_rate": 2.889212620258147e-06,
      "loss": 0.107,
      "step": 1755
    },
    {
      "epoch": 1.2630821794641252,
      "grad_norm": 5.03734432241681,
      "learning_rate": 2.8890842525105657e-06,
      "loss": 0.2066,
      "step": 1756
    },
    {
      "epoch": 1.2638014745549362,
      "grad_norm": 4.060745336182474,
      "learning_rate": 2.888955813292161e-06,
      "loss": 0.0557,
      "step": 1757
    },
    {
      "epoch": 1.2645207696457472,
      "grad_norm": 9.626269648195246,
      "learning_rate": 2.888827302609541e-06,
      "loss": 0.1153,
      "step": 1758
    },
    {
      "epoch": 1.2652400647365583,
      "grad_norm": 5.0309212296301355,
      "learning_rate": 2.8886987204693185e-06,
      "loss": 0.0466,
      "step": 1759
    },
    {
      "epoch": 1.2659593598273693,
      "grad_norm": 3.8823097303680516,
      "learning_rate": 2.888570066878109e-06,
      "loss": 0.0715,
      "step": 1760
    },
    {
      "epoch": 1.26667865491818,
      "grad_norm": 4.899662830415322,
      "learning_rate": 2.8884413418425323e-06,
      "loss": 0.1671,
      "step": 1761
    },
    {
      "epoch": 1.2673979500089911,
      "grad_norm": 1.4556095411675072,
      "learning_rate": 2.888312545369211e-06,
      "loss": 0.0096,
      "step": 1762
    },
    {
      "epoch": 1.2681172450998022,
      "grad_norm": 4.863421250658491,
      "learning_rate": 2.888183677464772e-06,
      "loss": 0.1088,
      "step": 1763
    },
    {
      "epoch": 1.2688365401906132,
      "grad_norm": 5.069363586327778,
      "learning_rate": 2.8880547381358466e-06,
      "loss": 0.1343,
      "step": 1764
    },
    {
      "epoch": 1.2695558352814242,
      "grad_norm": 6.596713193494357,
      "learning_rate": 2.8879257273890674e-06,
      "loss": 0.2512,
      "step": 1765
    },
    {
      "epoch": 1.2702751303722353,
      "grad_norm": 3.6578834557298676,
      "learning_rate": 2.8877966452310737e-06,
      "loss": 0.0197,
      "step": 1766
    },
    {
      "epoch": 1.270994425463046,
      "grad_norm": 7.139498237139241,
      "learning_rate": 2.8876674916685066e-06,
      "loss": 0.1187,
      "step": 1767
    },
    {
      "epoch": 1.2717137205538571,
      "grad_norm": 5.442985747142592,
      "learning_rate": 2.8875382667080103e-06,
      "loss": 0.2379,
      "step": 1768
    },
    {
      "epoch": 1.2724330156446682,
      "grad_norm": 2.815477592454708,
      "learning_rate": 2.887408970356235e-06,
      "loss": 0.0099,
      "step": 1769
    },
    {
      "epoch": 1.2731523107354792,
      "grad_norm": 3.5539231872341364,
      "learning_rate": 2.8872796026198322e-06,
      "loss": 0.1148,
      "step": 1770
    },
    {
      "epoch": 1.2738716058262902,
      "grad_norm": 5.956974322358411,
      "learning_rate": 2.887150163505459e-06,
      "loss": 0.1346,
      "step": 1771
    },
    {
      "epoch": 1.2745909009171013,
      "grad_norm": 5.778114535080318,
      "learning_rate": 2.8870206530197747e-06,
      "loss": 0.0432,
      "step": 1772
    },
    {
      "epoch": 1.2753101960079123,
      "grad_norm": 7.749029376734017,
      "learning_rate": 2.8868910711694428e-06,
      "loss": 0.499,
      "step": 1773
    },
    {
      "epoch": 1.2760294910987233,
      "grad_norm": 8.817791650529825,
      "learning_rate": 2.886761417961131e-06,
      "loss": 0.3223,
      "step": 1774
    },
    {
      "epoch": 1.2767487861895344,
      "grad_norm": 5.582186615519434,
      "learning_rate": 2.8866316934015094e-06,
      "loss": 0.0394,
      "step": 1775
    },
    {
      "epoch": 1.2774680812803454,
      "grad_norm": 3.724161404478559,
      "learning_rate": 2.886501897497253e-06,
      "loss": 0.1146,
      "step": 1776
    },
    {
      "epoch": 1.2781873763711562,
      "grad_norm": 4.598854300745457,
      "learning_rate": 2.8863720302550403e-06,
      "loss": 0.0926,
      "step": 1777
    },
    {
      "epoch": 1.2789066714619672,
      "grad_norm": 1.443410755650339,
      "learning_rate": 2.8862420916815535e-06,
      "loss": 0.0036,
      "step": 1778
    },
    {
      "epoch": 1.2796259665527783,
      "grad_norm": 6.324532017095005,
      "learning_rate": 2.886112081783477e-06,
      "loss": 0.1501,
      "step": 1779
    },
    {
      "epoch": 1.2803452616435893,
      "grad_norm": 2.037291547362322,
      "learning_rate": 2.885982000567501e-06,
      "loss": 0.0426,
      "step": 1780
    },
    {
      "epoch": 1.2810645567344003,
      "grad_norm": 4.535852923569214,
      "learning_rate": 2.8858518480403175e-06,
      "loss": 0.122,
      "step": 1781
    },
    {
      "epoch": 1.2817838518252114,
      "grad_norm": 2.390959142873699,
      "learning_rate": 2.8857216242086236e-06,
      "loss": 0.0165,
      "step": 1782
    },
    {
      "epoch": 1.2825031469160222,
      "grad_norm": 6.136026621783262,
      "learning_rate": 2.8855913290791202e-06,
      "loss": 0.2305,
      "step": 1783
    },
    {
      "epoch": 1.2832224420068332,
      "grad_norm": 6.910578009349899,
      "learning_rate": 2.88546096265851e-06,
      "loss": 0.1542,
      "step": 1784
    },
    {
      "epoch": 1.2839417370976443,
      "grad_norm": 6.4265249107022955,
      "learning_rate": 2.8853305249535015e-06,
      "loss": 0.1086,
      "step": 1785
    },
    {
      "epoch": 1.2846610321884553,
      "grad_norm": 8.33119472210341,
      "learning_rate": 2.8852000159708057e-06,
      "loss": 0.1584,
      "step": 1786
    },
    {
      "epoch": 1.2853803272792663,
      "grad_norm": 4.704317466052146,
      "learning_rate": 2.8850694357171374e-06,
      "loss": 0.142,
      "step": 1787
    },
    {
      "epoch": 1.2860996223700774,
      "grad_norm": 6.978055033832254,
      "learning_rate": 2.884938784199215e-06,
      "loss": 0.1079,
      "step": 1788
    },
    {
      "epoch": 1.2868189174608884,
      "grad_norm": 4.60709267559893,
      "learning_rate": 2.8848080614237615e-06,
      "loss": 0.0396,
      "step": 1789
    },
    {
      "epoch": 1.2875382125516994,
      "grad_norm": 1.2713493964235183,
      "learning_rate": 2.884677267397502e-06,
      "loss": 0.0031,
      "step": 1790
    },
    {
      "epoch": 1.2882575076425105,
      "grad_norm": 3.0437346723017846,
      "learning_rate": 2.884546402127166e-06,
      "loss": 0.0179,
      "step": 1791
    },
    {
      "epoch": 1.2889768027333213,
      "grad_norm": 2.369345739160241,
      "learning_rate": 2.8844154656194872e-06,
      "loss": 0.0723,
      "step": 1792
    },
    {
      "epoch": 1.2896960978241323,
      "grad_norm": 6.1815280427151045,
      "learning_rate": 2.8842844578812026e-06,
      "loss": 0.2611,
      "step": 1793
    },
    {
      "epoch": 1.2904153929149433,
      "grad_norm": 3.369798320848323,
      "learning_rate": 2.8841533789190523e-06,
      "loss": 0.0337,
      "step": 1794
    },
    {
      "epoch": 1.2911346880057544,
      "grad_norm": 4.75350687169893,
      "learning_rate": 2.884022228739781e-06,
      "loss": 0.1725,
      "step": 1795
    },
    {
      "epoch": 1.2918539830965654,
      "grad_norm": 3.9552176641061934,
      "learning_rate": 2.883891007350137e-06,
      "loss": 0.0274,
      "step": 1796
    },
    {
      "epoch": 1.2925732781873764,
      "grad_norm": 3.6310465645044543,
      "learning_rate": 2.883759714756871e-06,
      "loss": 0.1276,
      "step": 1797
    },
    {
      "epoch": 1.2932925732781873,
      "grad_norm": 4.689954191449377,
      "learning_rate": 2.8836283509667385e-06,
      "loss": 0.0084,
      "step": 1798
    },
    {
      "epoch": 1.2940118683689983,
      "grad_norm": 5.860750940850004,
      "learning_rate": 2.8834969159864985e-06,
      "loss": 0.2997,
      "step": 1799
    },
    {
      "epoch": 1.2947311634598093,
      "grad_norm": 8.771977581882343,
      "learning_rate": 2.8833654098229132e-06,
      "loss": 0.3005,
      "step": 1800
    },
    {
      "epoch": 1.2954504585506204,
      "grad_norm": 7.472293046286825,
      "learning_rate": 2.8832338324827496e-06,
      "loss": 0.1178,
      "step": 1801
    },
    {
      "epoch": 1.2961697536414314,
      "grad_norm": 4.310086642271168,
      "learning_rate": 2.883102183972777e-06,
      "loss": 0.2254,
      "step": 1802
    },
    {
      "epoch": 1.2968890487322424,
      "grad_norm": 4.707417766061669,
      "learning_rate": 2.8829704642997685e-06,
      "loss": 0.1275,
      "step": 1803
    },
    {
      "epoch": 1.2976083438230535,
      "grad_norm": 10.22764751000119,
      "learning_rate": 2.882838673470503e-06,
      "loss": 0.2069,
      "step": 1804
    },
    {
      "epoch": 1.2983276389138645,
      "grad_norm": 5.686348961327223,
      "learning_rate": 2.8827068114917596e-06,
      "loss": 0.1678,
      "step": 1805
    },
    {
      "epoch": 1.2990469340046755,
      "grad_norm": 2.0740265001200275,
      "learning_rate": 2.882574878370323e-06,
      "loss": 0.0726,
      "step": 1806
    },
    {
      "epoch": 1.2997662290954866,
      "grad_norm": 4.024038524883801,
      "learning_rate": 2.8824428741129828e-06,
      "loss": 0.1014,
      "step": 1807
    },
    {
      "epoch": 1.3004855241862974,
      "grad_norm": 2.939439582226724,
      "learning_rate": 2.8823107987265297e-06,
      "loss": 0.0511,
      "step": 1808
    },
    {
      "epoch": 1.3012048192771084,
      "grad_norm": 3.567666260928984,
      "learning_rate": 2.8821786522177592e-06,
      "loss": 0.0651,
      "step": 1809
    },
    {
      "epoch": 1.3019241143679194,
      "grad_norm": 0.2020885875817831,
      "learning_rate": 2.882046434593471e-06,
      "loss": 0.0008,
      "step": 1810
    },
    {
      "epoch": 1.3026434094587305,
      "grad_norm": 5.3511253225263475,
      "learning_rate": 2.881914145860467e-06,
      "loss": 0.1655,
      "step": 1811
    },
    {
      "epoch": 1.3033627045495415,
      "grad_norm": 7.059966386638408,
      "learning_rate": 2.8817817860255553e-06,
      "loss": 0.2073,
      "step": 1812
    },
    {
      "epoch": 1.3040819996403523,
      "grad_norm": 4.115207631963609,
      "learning_rate": 2.881649355095544e-06,
      "loss": 0.0305,
      "step": 1813
    },
    {
      "epoch": 1.3048012947311634,
      "grad_norm": 5.305701484618844,
      "learning_rate": 2.8815168530772485e-06,
      "loss": 0.109,
      "step": 1814
    },
    {
      "epoch": 1.3055205898219744,
      "grad_norm": 7.543386207758745,
      "learning_rate": 2.8813842799774857e-06,
      "loss": 0.3113,
      "step": 1815
    },
    {
      "epoch": 1.3062398849127854,
      "grad_norm": 5.235520833490784,
      "learning_rate": 2.881251635803077e-06,
      "loss": 0.1324,
      "step": 1816
    },
    {
      "epoch": 1.3069591800035965,
      "grad_norm": 3.283920929196038,
      "learning_rate": 2.8811189205608466e-06,
      "loss": 0.0564,
      "step": 1817
    },
    {
      "epoch": 1.3076784750944075,
      "grad_norm": 3.5219511617821206,
      "learning_rate": 2.8809861342576233e-06,
      "loss": 0.1593,
      "step": 1818
    },
    {
      "epoch": 1.3083977701852185,
      "grad_norm": 8.377887136886393,
      "learning_rate": 2.8808532769002395e-06,
      "loss": 0.3513,
      "step": 1819
    },
    {
      "epoch": 1.3091170652760296,
      "grad_norm": 2.095217812133882,
      "learning_rate": 2.88072034849553e-06,
      "loss": 0.0101,
      "step": 1820
    },
    {
      "epoch": 1.3098363603668406,
      "grad_norm": 0.12494302192704931,
      "learning_rate": 2.880587349050335e-06,
      "loss": 0.0004,
      "step": 1821
    },
    {
      "epoch": 1.3105556554576516,
      "grad_norm": 5.062425812515262,
      "learning_rate": 2.8804542785714975e-06,
      "loss": 0.1723,
      "step": 1822
    },
    {
      "epoch": 1.3112749505484624,
      "grad_norm": 2.8671584954488165,
      "learning_rate": 2.880321137065864e-06,
      "loss": 0.0644,
      "step": 1823
    },
    {
      "epoch": 1.3119942456392735,
      "grad_norm": 7.190457460575838,
      "learning_rate": 2.8801879245402853e-06,
      "loss": 0.1453,
      "step": 1824
    },
    {
      "epoch": 1.3127135407300845,
      "grad_norm": 4.555336227310286,
      "learning_rate": 2.8800546410016143e-06,
      "loss": 0.1951,
      "step": 1825
    },
    {
      "epoch": 1.3134328358208955,
      "grad_norm": 2.8697308459652904,
      "learning_rate": 2.87992128645671e-06,
      "loss": 0.0344,
      "step": 1826
    },
    {
      "epoch": 1.3141521309117066,
      "grad_norm": 4.465652176100414,
      "learning_rate": 2.8797878609124323e-06,
      "loss": 0.1618,
      "step": 1827
    },
    {
      "epoch": 1.3148714260025176,
      "grad_norm": 1.8426529178084399,
      "learning_rate": 2.879654364375648e-06,
      "loss": 0.0037,
      "step": 1828
    },
    {
      "epoch": 1.3155907210933284,
      "grad_norm": 5.041210844442453,
      "learning_rate": 2.879520796853224e-06,
      "loss": 0.2127,
      "step": 1829
    },
    {
      "epoch": 1.3163100161841395,
      "grad_norm": 4.17112896426112,
      "learning_rate": 2.879387158352034e-06,
      "loss": 0.1656,
      "step": 1830
    },
    {
      "epoch": 1.3170293112749505,
      "grad_norm": 5.417089385255273,
      "learning_rate": 2.8792534488789528e-06,
      "loss": 0.1785,
      "step": 1831
    },
    {
      "epoch": 1.3177486063657615,
      "grad_norm": 6.594442214193899,
      "learning_rate": 2.8791196684408606e-06,
      "loss": 0.3024,
      "step": 1832
    },
    {
      "epoch": 1.3184679014565726,
      "grad_norm": 2.4721537178047086,
      "learning_rate": 2.8789858170446407e-06,
      "loss": 0.0193,
      "step": 1833
    },
    {
      "epoch": 1.3191871965473836,
      "grad_norm": 5.675765175033997,
      "learning_rate": 2.8788518946971795e-06,
      "loss": 0.1626,
      "step": 1834
    },
    {
      "epoch": 1.3199064916381946,
      "grad_norm": 3.992369985089034,
      "learning_rate": 2.878717901405368e-06,
      "loss": 0.0839,
      "step": 1835
    },
    {
      "epoch": 1.3206257867290057,
      "grad_norm": 6.138903582265613,
      "learning_rate": 2.8785838371761e-06,
      "loss": 0.2443,
      "step": 1836
    },
    {
      "epoch": 1.3213450818198167,
      "grad_norm": 3.5018656836216384,
      "learning_rate": 2.8784497020162735e-06,
      "loss": 0.0231,
      "step": 1837
    },
    {
      "epoch": 1.3220643769106275,
      "grad_norm": 5.687174189057592,
      "learning_rate": 2.8783154959327904e-06,
      "loss": 0.2432,
      "step": 1838
    },
    {
      "epoch": 1.3227836720014385,
      "grad_norm": 5.2718872033794435,
      "learning_rate": 2.878181218932555e-06,
      "loss": 0.1147,
      "step": 1839
    },
    {
      "epoch": 1.3235029670922496,
      "grad_norm": 6.607650048238397,
      "learning_rate": 2.878046871022477e-06,
      "loss": 0.2408,
      "step": 1840
    },
    {
      "epoch": 1.3242222621830606,
      "grad_norm": 5.059278886193367,
      "learning_rate": 2.877912452209468e-06,
      "loss": 0.158,
      "step": 1841
    },
    {
      "epoch": 1.3249415572738716,
      "grad_norm": 5.742009902032591,
      "learning_rate": 2.877777962500445e-06,
      "loss": 0.2005,
      "step": 1842
    },
    {
      "epoch": 1.3256608523646827,
      "grad_norm": 2.883964825618812,
      "learning_rate": 2.877643401902327e-06,
      "loss": 0.0677,
      "step": 1843
    },
    {
      "epoch": 1.3263801474554935,
      "grad_norm": 0.9963092949581706,
      "learning_rate": 2.8775087704220374e-06,
      "loss": 0.007,
      "step": 1844
    },
    {
      "epoch": 1.3270994425463045,
      "grad_norm": 2.112298784312431,
      "learning_rate": 2.8773740680665037e-06,
      "loss": 0.054,
      "step": 1845
    },
    {
      "epoch": 1.3278187376371156,
      "grad_norm": 2.374280353637511,
      "learning_rate": 2.877239294842656e-06,
      "loss": 0.0071,
      "step": 1846
    },
    {
      "epoch": 1.3285380327279266,
      "grad_norm": 7.016661202511424,
      "learning_rate": 2.877104450757429e-06,
      "loss": 0.1912,
      "step": 1847
    },
    {
      "epoch": 1.3292573278187376,
      "grad_norm": 3.237087173742285,
      "learning_rate": 2.8769695358177606e-06,
      "loss": 0.0782,
      "step": 1848
    },
    {
      "epoch": 1.3299766229095487,
      "grad_norm": 6.190447468857955,
      "learning_rate": 2.8768345500305926e-06,
      "loss": 0.1052,
      "step": 1849
    },
    {
      "epoch": 1.3306959180003597,
      "grad_norm": 4.884468546707741,
      "learning_rate": 2.8766994934028697e-06,
      "loss": 0.0965,
      "step": 1850
    },
    {
      "epoch": 1.3314152130911707,
      "grad_norm": 5.051099855875771,
      "learning_rate": 2.8765643659415414e-06,
      "loss": 0.1169,
      "step": 1851
    },
    {
      "epoch": 1.3321345081819818,
      "grad_norm": 6.455809895305699,
      "learning_rate": 2.8764291676535597e-06,
      "loss": 0.0151,
      "step": 1852
    },
    {
      "epoch": 1.3328538032727928,
      "grad_norm": 5.785708648703478,
      "learning_rate": 2.8762938985458814e-06,
      "loss": 0.1337,
      "step": 1853
    },
    {
      "epoch": 1.3335730983636036,
      "grad_norm": 3.931423117882026,
      "learning_rate": 2.8761585586254655e-06,
      "loss": 0.1127,
      "step": 1854
    },
    {
      "epoch": 1.3342923934544146,
      "grad_norm": 4.761794607288105,
      "learning_rate": 2.876023147899277e-06,
      "loss": 0.1858,
      "step": 1855
    },
    {
      "epoch": 1.3350116885452257,
      "grad_norm": 3.006424913906359,
      "learning_rate": 2.875887666374281e-06,
      "loss": 0.0475,
      "step": 1856
    },
    {
      "epoch": 1.3357309836360367,
      "grad_norm": 7.130340436290732,
      "learning_rate": 2.87575211405745e-06,
      "loss": 0.0653,
      "step": 1857
    },
    {
      "epoch": 1.3364502787268477,
      "grad_norm": 5.813711605396539,
      "learning_rate": 2.875616490955757e-06,
      "loss": 0.1385,
      "step": 1858
    },
    {
      "epoch": 1.3371695738176588,
      "grad_norm": 8.33374339637154,
      "learning_rate": 2.8754807970761812e-06,
      "loss": 0.1908,
      "step": 1859
    },
    {
      "epoch": 1.3378888689084696,
      "grad_norm": 6.39867126613376,
      "learning_rate": 2.8753450324257036e-06,
      "loss": 0.1393,
      "step": 1860
    },
    {
      "epoch": 1.3386081639992806,
      "grad_norm": 3.4348843090382966,
      "learning_rate": 2.87520919701131e-06,
      "loss": 0.1019,
      "step": 1861
    },
    {
      "epoch": 1.3393274590900917,
      "grad_norm": 4.684351452048262,
      "learning_rate": 2.8750732908399887e-06,
      "loss": 0.0786,
      "step": 1862
    },
    {
      "epoch": 1.3400467541809027,
      "grad_norm": 6.398053898938119,
      "learning_rate": 2.874937313918733e-06,
      "loss": 0.0547,
      "step": 1863
    },
    {
      "epoch": 1.3407660492717137,
      "grad_norm": 3.3814315374114825,
      "learning_rate": 2.874801266254539e-06,
      "loss": 0.0197,
      "step": 1864
    },
    {
      "epoch": 1.3414853443625248,
      "grad_norm": 6.8620285275705,
      "learning_rate": 2.8746651478544064e-06,
      "loss": 0.2157,
      "step": 1865
    },
    {
      "epoch": 1.3422046394533358,
      "grad_norm": 7.384899978037323,
      "learning_rate": 2.8745289587253384e-06,
      "loss": 0.168,
      "step": 1866
    },
    {
      "epoch": 1.3429239345441468,
      "grad_norm": 6.078623593013814,
      "learning_rate": 2.874392698874343e-06,
      "loss": 0.1552,
      "step": 1867
    },
    {
      "epoch": 1.3436432296349579,
      "grad_norm": 4.592180192317263,
      "learning_rate": 2.8742563683084304e-06,
      "loss": 0.1839,
      "step": 1868
    },
    {
      "epoch": 1.3443625247257687,
      "grad_norm": 7.5951922177785045,
      "learning_rate": 2.874119967034615e-06,
      "loss": 0.3214,
      "step": 1869
    },
    {
      "epoch": 1.3450818198165797,
      "grad_norm": 10.287606520018116,
      "learning_rate": 2.873983495059915e-06,
      "loss": 0.0433,
      "step": 1870
    },
    {
      "epoch": 1.3458011149073907,
      "grad_norm": 3.396971632359829,
      "learning_rate": 2.873846952391353e-06,
      "loss": 0.0855,
      "step": 1871
    },
    {
      "epoch": 1.3465204099982018,
      "grad_norm": 2.5977139927443296,
      "learning_rate": 2.873710339035953e-06,
      "loss": 0.0748,
      "step": 1872
    },
    {
      "epoch": 1.3472397050890128,
      "grad_norm": 9.532789292087031,
      "learning_rate": 2.8735736550007447e-06,
      "loss": 0.1301,
      "step": 1873
    },
    {
      "epoch": 1.3479590001798238,
      "grad_norm": 0.5868089599182268,
      "learning_rate": 2.8734369002927607e-06,
      "loss": 0.0027,
      "step": 1874
    },
    {
      "epoch": 1.3486782952706347,
      "grad_norm": 5.32084968799656,
      "learning_rate": 2.873300074919037e-06,
      "loss": 0.0328,
      "step": 1875
    },
    {
      "epoch": 1.3493975903614457,
      "grad_norm": 3.453961300947311,
      "learning_rate": 2.873163178886614e-06,
      "loss": 0.0584,
      "step": 1876
    },
    {
      "epoch": 1.3501168854522567,
      "grad_norm": 4.282843496149778,
      "learning_rate": 2.873026212202535e-06,
      "loss": 0.0776,
      "step": 1877
    },
    {
      "epoch": 1.3508361805430678,
      "grad_norm": 3.671254299140581,
      "learning_rate": 2.8728891748738466e-06,
      "loss": 0.0674,
      "step": 1878
    },
    {
      "epoch": 1.3515554756338788,
      "grad_norm": 5.896455441705231,
      "learning_rate": 2.8727520669076007e-06,
      "loss": 0.2183,
      "step": 1879
    },
    {
      "epoch": 1.3522747707246898,
      "grad_norm": 4.721397643763593,
      "learning_rate": 2.8726148883108505e-06,
      "loss": 0.1232,
      "step": 1880
    },
    {
      "epoch": 1.3529940658155009,
      "grad_norm": 4.928749330687315,
      "learning_rate": 2.8724776390906553e-06,
      "loss": 0.2478,
      "step": 1881
    },
    {
      "epoch": 1.353713360906312,
      "grad_norm": 6.17180554171681,
      "learning_rate": 2.8723403192540763e-06,
      "loss": 0.2045,
      "step": 1882
    },
    {
      "epoch": 1.354432655997123,
      "grad_norm": 6.8157629871334935,
      "learning_rate": 2.872202928808178e-06,
      "loss": 0.0328,
      "step": 1883
    },
    {
      "epoch": 1.355151951087934,
      "grad_norm": 4.703999473831349,
      "learning_rate": 2.8720654677600314e-06,
      "loss": 0.0672,
      "step": 1884
    },
    {
      "epoch": 1.3558712461787448,
      "grad_norm": 5.759290786253367,
      "learning_rate": 2.871927936116707e-06,
      "loss": 0.1364,
      "step": 1885
    },
    {
      "epoch": 1.3565905412695558,
      "grad_norm": 4.724110816705259,
      "learning_rate": 2.871790333885282e-06,
      "loss": 0.0973,
      "step": 1886
    },
    {
      "epoch": 1.3573098363603668,
      "grad_norm": 5.829353646123308,
      "learning_rate": 2.8716526610728364e-06,
      "loss": 0.21,
      "step": 1887
    },
    {
      "epoch": 1.3580291314511779,
      "grad_norm": 4.067270789819697,
      "learning_rate": 2.871514917686454e-06,
      "loss": 0.2588,
      "step": 1888
    },
    {
      "epoch": 1.358748426541989,
      "grad_norm": 2.4918752137717415,
      "learning_rate": 2.8713771037332207e-06,
      "loss": 0.0774,
      "step": 1889
    },
    {
      "epoch": 1.3594677216327997,
      "grad_norm": 3.8950955116659096,
      "learning_rate": 2.8712392192202284e-06,
      "loss": 0.2138,
      "step": 1890
    },
    {
      "epoch": 1.3601870167236108,
      "grad_norm": 4.90310513012652,
      "learning_rate": 2.8711012641545715e-06,
      "loss": 0.1013,
      "step": 1891
    },
    {
      "epoch": 1.3609063118144218,
      "grad_norm": 5.761451843950282,
      "learning_rate": 2.870963238543347e-06,
      "loss": 0.1667,
      "step": 1892
    },
    {
      "epoch": 1.3616256069052328,
      "grad_norm": 6.030882256630764,
      "learning_rate": 2.8708251423936573e-06,
      "loss": 0.2053,
      "step": 1893
    },
    {
      "epoch": 1.3623449019960439,
      "grad_norm": 6.522936530240776,
      "learning_rate": 2.8706869757126077e-06,
      "loss": 0.0359,
      "step": 1894
    },
    {
      "epoch": 1.363064197086855,
      "grad_norm": 6.601411931507219,
      "learning_rate": 2.870548738507307e-06,
      "loss": 0.2296,
      "step": 1895
    },
    {
      "epoch": 1.363783492177666,
      "grad_norm": 5.691795061079199,
      "learning_rate": 2.8704104307848684e-06,
      "loss": 0.3567,
      "step": 1896
    },
    {
      "epoch": 1.364502787268477,
      "grad_norm": 1.283835894083121,
      "learning_rate": 2.870272052552407e-06,
      "loss": 0.0189,
      "step": 1897
    },
    {
      "epoch": 1.365222082359288,
      "grad_norm": 4.106284504778463,
      "learning_rate": 2.8701336038170428e-06,
      "loss": 0.0741,
      "step": 1898
    },
    {
      "epoch": 1.365941377450099,
      "grad_norm": 3.5059526807046923,
      "learning_rate": 2.8699950845858995e-06,
      "loss": 0.0234,
      "step": 1899
    },
    {
      "epoch": 1.3666606725409098,
      "grad_norm": 3.779700276545603,
      "learning_rate": 2.8698564948661045e-06,
      "loss": 0.0147,
      "step": 1900
    },
    {
      "epoch": 1.3673799676317209,
      "grad_norm": 6.783809531972101,
      "learning_rate": 2.8697178346647882e-06,
      "loss": 0.2371,
      "step": 1901
    },
    {
      "epoch": 1.368099262722532,
      "grad_norm": 0.9677415429901468,
      "learning_rate": 2.8695791039890843e-06,
      "loss": 0.0115,
      "step": 1902
    },
    {
      "epoch": 1.368818557813343,
      "grad_norm": 6.727792863141242,
      "learning_rate": 2.869440302846132e-06,
      "loss": 0.2896,
      "step": 1903
    },
    {
      "epoch": 1.369537852904154,
      "grad_norm": 4.147112617344063,
      "learning_rate": 2.8693014312430718e-06,
      "loss": 0.0431,
      "step": 1904
    },
    {
      "epoch": 1.370257147994965,
      "grad_norm": 4.241363176441661,
      "learning_rate": 2.869162489187049e-06,
      "loss": 0.1397,
      "step": 1905
    },
    {
      "epoch": 1.3709764430857758,
      "grad_norm": 6.687088618579141,
      "learning_rate": 2.8690234766852132e-06,
      "loss": 0.1491,
      "step": 1906
    },
    {
      "epoch": 1.3716957381765869,
      "grad_norm": 3.7765268032910586,
      "learning_rate": 2.868884393744716e-06,
      "loss": 0.1477,
      "step": 1907
    },
    {
      "epoch": 1.3724150332673979,
      "grad_norm": 4.139740864851264,
      "learning_rate": 2.868745240372713e-06,
      "loss": 0.0792,
      "step": 1908
    },
    {
      "epoch": 1.373134328358209,
      "grad_norm": 3.9534063648908515,
      "learning_rate": 2.868606016576366e-06,
      "loss": 0.0556,
      "step": 1909
    },
    {
      "epoch": 1.37385362344902,
      "grad_norm": 7.8093498646775545,
      "learning_rate": 2.8684667223628362e-06,
      "loss": 0.2441,
      "step": 1910
    },
    {
      "epoch": 1.374572918539831,
      "grad_norm": 3.716601863584651,
      "learning_rate": 2.868327357739291e-06,
      "loss": 0.1437,
      "step": 1911
    },
    {
      "epoch": 1.375292213630642,
      "grad_norm": 2.91209494429562,
      "learning_rate": 2.8681879227129014e-06,
      "loss": 0.0644,
      "step": 1912
    },
    {
      "epoch": 1.376011508721453,
      "grad_norm": 8.512505309921577,
      "learning_rate": 2.8680484172908416e-06,
      "loss": 0.0947,
      "step": 1913
    },
    {
      "epoch": 1.376730803812264,
      "grad_norm": 4.730720273817467,
      "learning_rate": 2.867908841480289e-06,
      "loss": 0.1724,
      "step": 1914
    },
    {
      "epoch": 1.377450098903075,
      "grad_norm": 4.575301320873031,
      "learning_rate": 2.8677691952884254e-06,
      "loss": 0.0307,
      "step": 1915
    },
    {
      "epoch": 1.378169393993886,
      "grad_norm": 2.7056726784039946,
      "learning_rate": 2.8676294787224355e-06,
      "loss": 0.0763,
      "step": 1916
    },
    {
      "epoch": 1.378888689084697,
      "grad_norm": 5.9917559907685005,
      "learning_rate": 2.867489691789508e-06,
      "loss": 0.0435,
      "step": 1917
    },
    {
      "epoch": 1.379607984175508,
      "grad_norm": 2.2920247880790816,
      "learning_rate": 2.867349834496836e-06,
      "loss": 0.0218,
      "step": 1918
    },
    {
      "epoch": 1.380327279266319,
      "grad_norm": 4.274463707796197,
      "learning_rate": 2.8672099068516137e-06,
      "loss": 0.1012,
      "step": 1919
    },
    {
      "epoch": 1.38104657435713,
      "grad_norm": 3.3508697117175243,
      "learning_rate": 2.867069908861042e-06,
      "loss": 0.1919,
      "step": 1920
    },
    {
      "epoch": 1.3817658694479409,
      "grad_norm": 9.257193476715038,
      "learning_rate": 2.866929840532324e-06,
      "loss": 0.2183,
      "step": 1921
    },
    {
      "epoch": 1.382485164538752,
      "grad_norm": 4.9685247566317985,
      "learning_rate": 2.8667897018726663e-06,
      "loss": 0.128,
      "step": 1922
    },
    {
      "epoch": 1.383204459629563,
      "grad_norm": 0.1546847182092274,
      "learning_rate": 2.866649492889279e-06,
      "loss": 0.0009,
      "step": 1923
    },
    {
      "epoch": 1.383923754720374,
      "grad_norm": 5.234305306662609,
      "learning_rate": 2.8665092135893755e-06,
      "loss": 0.0833,
      "step": 1924
    },
    {
      "epoch": 1.384643049811185,
      "grad_norm": 4.213856902876313,
      "learning_rate": 2.8663688639801747e-06,
      "loss": 0.1818,
      "step": 1925
    },
    {
      "epoch": 1.385362344901996,
      "grad_norm": 5.7795682626638305,
      "learning_rate": 2.8662284440688973e-06,
      "loss": 0.1123,
      "step": 1926
    },
    {
      "epoch": 1.386081639992807,
      "grad_norm": 3.0411290605332306,
      "learning_rate": 2.8660879538627676e-06,
      "loss": 0.0811,
      "step": 1927
    },
    {
      "epoch": 1.3868009350836181,
      "grad_norm": 1.4826076423309897,
      "learning_rate": 2.8659473933690157e-06,
      "loss": 0.0117,
      "step": 1928
    },
    {
      "epoch": 1.3875202301744292,
      "grad_norm": 5.216213811121435,
      "learning_rate": 2.8658067625948716e-06,
      "loss": 0.1762,
      "step": 1929
    },
    {
      "epoch": 1.3882395252652402,
      "grad_norm": 4.634746753642027,
      "learning_rate": 2.865666061547572e-06,
      "loss": 0.1135,
      "step": 1930
    },
    {
      "epoch": 1.388958820356051,
      "grad_norm": 4.5587033326342095,
      "learning_rate": 2.865525290234356e-06,
      "loss": 0.086,
      "step": 1931
    },
    {
      "epoch": 1.389678115446862,
      "grad_norm": 2.290336732356614,
      "learning_rate": 2.8653844486624675e-06,
      "loss": 0.0473,
      "step": 1932
    },
    {
      "epoch": 1.390397410537673,
      "grad_norm": 7.910976969174364,
      "learning_rate": 2.8652435368391522e-06,
      "loss": 0.2336,
      "step": 1933
    },
    {
      "epoch": 1.391116705628484,
      "grad_norm": 2.9833598994318775,
      "learning_rate": 2.8651025547716598e-06,
      "loss": 0.0678,
      "step": 1934
    },
    {
      "epoch": 1.3918360007192951,
      "grad_norm": 3.9513792992593246,
      "learning_rate": 2.8649615024672445e-06,
      "loss": 0.125,
      "step": 1935
    },
    {
      "epoch": 1.3925552958101062,
      "grad_norm": 3.6054297150420154,
      "learning_rate": 2.864820379933164e-06,
      "loss": 0.1097,
      "step": 1936
    },
    {
      "epoch": 1.393274590900917,
      "grad_norm": 8.399979399115058,
      "learning_rate": 2.8646791871766795e-06,
      "loss": 0.1244,
      "step": 1937
    },
    {
      "epoch": 1.393993885991728,
      "grad_norm": 4.452107154972758,
      "learning_rate": 2.8645379242050546e-06,
      "loss": 0.0801,
      "step": 1938
    },
    {
      "epoch": 1.394713181082539,
      "grad_norm": 2.0151878951527493,
      "learning_rate": 2.8643965910255584e-06,
      "loss": 0.0397,
      "step": 1939
    },
    {
      "epoch": 1.39543247617335,
      "grad_norm": 5.397892435682733,
      "learning_rate": 2.8642551876454624e-06,
      "loss": 0.1883,
      "step": 1940
    },
    {
      "epoch": 1.3961517712641611,
      "grad_norm": 4.773125167591133,
      "learning_rate": 2.8641137140720425e-06,
      "loss": 0.026,
      "step": 1941
    },
    {
      "epoch": 1.3968710663549722,
      "grad_norm": 4.9317297230557875,
      "learning_rate": 2.863972170312577e-06,
      "loss": 0.1169,
      "step": 1942
    },
    {
      "epoch": 1.3975903614457832,
      "grad_norm": 7.644149189830625,
      "learning_rate": 2.863830556374349e-06,
      "loss": 0.1355,
      "step": 1943
    },
    {
      "epoch": 1.3983096565365942,
      "grad_norm": 5.867470273592033,
      "learning_rate": 2.8636888722646445e-06,
      "loss": 0.1344,
      "step": 1944
    },
    {
      "epoch": 1.3990289516274053,
      "grad_norm": 7.036349942610618,
      "learning_rate": 2.863547117990754e-06,
      "loss": 0.0937,
      "step": 1945
    },
    {
      "epoch": 1.399748246718216,
      "grad_norm": 6.743646316357328,
      "learning_rate": 2.863405293559971e-06,
      "loss": 0.2231,
      "step": 1946
    },
    {
      "epoch": 1.400467541809027,
      "grad_norm": 6.694266047154306,
      "learning_rate": 2.8632633989795912e-06,
      "loss": 0.1484,
      "step": 1947
    },
    {
      "epoch": 1.4011868368998381,
      "grad_norm": 5.865500304378709,
      "learning_rate": 2.863121434256917e-06,
      "loss": 0.1529,
      "step": 1948
    },
    {
      "epoch": 1.4019061319906492,
      "grad_norm": 3.2047204858265337,
      "learning_rate": 2.8629793993992527e-06,
      "loss": 0.0637,
      "step": 1949
    },
    {
      "epoch": 1.4026254270814602,
      "grad_norm": 4.871879921497448,
      "learning_rate": 2.862837294413905e-06,
      "loss": 0.1437,
      "step": 1950
    },
    {
      "epoch": 1.4033447221722712,
      "grad_norm": 7.193840915806507,
      "learning_rate": 2.8626951193081856e-06,
      "loss": 0.1608,
      "step": 1951
    },
    {
      "epoch": 1.404064017263082,
      "grad_norm": 5.911742055638088,
      "learning_rate": 2.8625528740894114e-06,
      "loss": 0.0722,
      "step": 1952
    },
    {
      "epoch": 1.404783312353893,
      "grad_norm": 8.623124966845783,
      "learning_rate": 2.862410558764899e-06,
      "loss": 0.3501,
      "step": 1953
    },
    {
      "epoch": 1.4055026074447041,
      "grad_norm": 5.022809816515642,
      "learning_rate": 2.8622681733419724e-06,
      "loss": 0.1108,
      "step": 1954
    },
    {
      "epoch": 1.4062219025355152,
      "grad_norm": 2.080315819619389,
      "learning_rate": 2.862125717827956e-06,
      "loss": 0.0042,
      "step": 1955
    },
    {
      "epoch": 1.4069411976263262,
      "grad_norm": 6.969493607240946,
      "learning_rate": 2.861983192230181e-06,
      "loss": 0.222,
      "step": 1956
    },
    {
      "epoch": 1.4076604927171372,
      "grad_norm": 7.1644543279506445,
      "learning_rate": 2.86184059655598e-06,
      "loss": 0.0592,
      "step": 1957
    },
    {
      "epoch": 1.4083797878079483,
      "grad_norm": 4.117975356153995,
      "learning_rate": 2.861697930812689e-06,
      "loss": 0.1629,
      "step": 1958
    },
    {
      "epoch": 1.4090990828987593,
      "grad_norm": 3.9833930394152643,
      "learning_rate": 2.8615551950076496e-06,
      "loss": 0.1811,
      "step": 1959
    },
    {
      "epoch": 1.4098183779895703,
      "grad_norm": 2.1241573575202835,
      "learning_rate": 2.861412389148205e-06,
      "loss": 0.0064,
      "step": 1960
    },
    {
      "epoch": 1.4105376730803814,
      "grad_norm": 6.982257172465749,
      "learning_rate": 2.8612695132417037e-06,
      "loss": 0.088,
      "step": 1961
    },
    {
      "epoch": 1.4112569681711922,
      "grad_norm": 2.871719751941688,
      "learning_rate": 2.8611265672954954e-06,
      "loss": 0.08,
      "step": 1962
    },
    {
      "epoch": 1.4119762632620032,
      "grad_norm": 4.6275268980567335,
      "learning_rate": 2.8609835513169364e-06,
      "loss": 0.1588,
      "step": 1963
    },
    {
      "epoch": 1.4126955583528142,
      "grad_norm": 3.7635037556838276,
      "learning_rate": 2.8608404653133843e-06,
      "loss": 0.0447,
      "step": 1964
    },
    {
      "epoch": 1.4134148534436253,
      "grad_norm": 4.483019355120998,
      "learning_rate": 2.8606973092922017e-06,
      "loss": 0.1542,
      "step": 1965
    },
    {
      "epoch": 1.4141341485344363,
      "grad_norm": 5.338802657060233,
      "learning_rate": 2.8605540832607533e-06,
      "loss": 0.1126,
      "step": 1966
    },
    {
      "epoch": 1.4148534436252471,
      "grad_norm": 3.7953944018148422,
      "learning_rate": 2.8604107872264094e-06,
      "loss": 0.109,
      "step": 1967
    },
    {
      "epoch": 1.4155727387160582,
      "grad_norm": 4.2957269082654665,
      "learning_rate": 2.8602674211965424e-06,
      "loss": 0.1856,
      "step": 1968
    },
    {
      "epoch": 1.4162920338068692,
      "grad_norm": 8.252263302831576,
      "learning_rate": 2.8601239851785287e-06,
      "loss": 0.0896,
      "step": 1969
    },
    {
      "epoch": 1.4170113288976802,
      "grad_norm": 1.4726144052907457,
      "learning_rate": 2.859980479179748e-06,
      "loss": 0.0347,
      "step": 1970
    },
    {
      "epoch": 1.4177306239884913,
      "grad_norm": 4.323203799427797,
      "learning_rate": 2.8598369032075848e-06,
      "loss": 0.1692,
      "step": 1971
    },
    {
      "epoch": 1.4184499190793023,
      "grad_norm": 2.5610834467759713,
      "learning_rate": 2.859693257269426e-06,
      "loss": 0.0318,
      "step": 1972
    },
    {
      "epoch": 1.4191692141701133,
      "grad_norm": 5.7596271294171695,
      "learning_rate": 2.859549541372661e-06,
      "loss": 0.0675,
      "step": 1973
    },
    {
      "epoch": 1.4198885092609244,
      "grad_norm": 6.049257903513594,
      "learning_rate": 2.8594057555246864e-06,
      "loss": 0.1191,
      "step": 1974
    },
    {
      "epoch": 1.4206078043517354,
      "grad_norm": 3.3476359232453765,
      "learning_rate": 2.8592618997328993e-06,
      "loss": 0.0815,
      "step": 1975
    },
    {
      "epoch": 1.4213270994425464,
      "grad_norm": 5.203874376363021,
      "learning_rate": 2.859117974004701e-06,
      "loss": 0.1282,
      "step": 1976
    },
    {
      "epoch": 1.4220463945333572,
      "grad_norm": 6.927045448765321,
      "learning_rate": 2.8589739783474977e-06,
      "loss": 0.2118,
      "step": 1977
    },
    {
      "epoch": 1.4227656896241683,
      "grad_norm": 6.792837192376061,
      "learning_rate": 2.858829912768697e-06,
      "loss": 0.1443,
      "step": 1978
    },
    {
      "epoch": 1.4234849847149793,
      "grad_norm": 5.64689845317184,
      "learning_rate": 2.858685777275712e-06,
      "loss": 0.1448,
      "step": 1979
    },
    {
      "epoch": 1.4242042798057903,
      "grad_norm": 4.357875012857252,
      "learning_rate": 2.858541571875959e-06,
      "loss": 0.1041,
      "step": 1980
    },
    {
      "epoch": 1.4249235748966014,
      "grad_norm": 4.810267602299373,
      "learning_rate": 2.858397296576857e-06,
      "loss": 0.0146,
      "step": 1981
    },
    {
      "epoch": 1.4256428699874124,
      "grad_norm": 4.061077942725784,
      "learning_rate": 2.8582529513858298e-06,
      "loss": 0.0727,
      "step": 1982
    },
    {
      "epoch": 1.4263621650782232,
      "grad_norm": 3.1351744387694906,
      "learning_rate": 2.8581085363103036e-06,
      "loss": 0.0809,
      "step": 1983
    },
    {
      "epoch": 1.4270814601690343,
      "grad_norm": 3.1452340296216574,
      "learning_rate": 2.8579640513577094e-06,
      "loss": 0.0736,
      "step": 1984
    },
    {
      "epoch": 1.4278007552598453,
      "grad_norm": 5.15105769766964,
      "learning_rate": 2.8578194965354807e-06,
      "loss": 0.0563,
      "step": 1985
    },
    {
      "epoch": 1.4285200503506563,
      "grad_norm": 7.113597312031152,
      "learning_rate": 2.8576748718510553e-06,
      "loss": 0.278,
      "step": 1986
    },
    {
      "epoch": 1.4292393454414674,
      "grad_norm": 6.395007254471104,
      "learning_rate": 2.8575301773118744e-06,
      "loss": 0.1952,
      "step": 1987
    },
    {
      "epoch": 1.4299586405322784,
      "grad_norm": 6.705746236761664,
      "learning_rate": 2.857385412925383e-06,
      "loss": 0.2438,
      "step": 1988
    },
    {
      "epoch": 1.4306779356230894,
      "grad_norm": 8.292862561319234,
      "learning_rate": 2.8572405786990296e-06,
      "loss": 0.2797,
      "step": 1989
    },
    {
      "epoch": 1.4313972307139005,
      "grad_norm": 2.0034687573800594,
      "learning_rate": 2.8570956746402653e-06,
      "loss": 0.0477,
      "step": 1990
    },
    {
      "epoch": 1.4321165258047115,
      "grad_norm": 5.050066848276019,
      "learning_rate": 2.8569507007565463e-06,
      "loss": 0.0882,
      "step": 1991
    },
    {
      "epoch": 1.4328358208955223,
      "grad_norm": 2.437126982427539,
      "learning_rate": 2.856805657055332e-06,
      "loss": 0.0069,
      "step": 1992
    },
    {
      "epoch": 1.4335551159863333,
      "grad_norm": 5.794620779332187,
      "learning_rate": 2.8566605435440846e-06,
      "loss": 0.0938,
      "step": 1993
    },
    {
      "epoch": 1.4342744110771444,
      "grad_norm": 5.756558951752547,
      "learning_rate": 2.856515360230271e-06,
      "loss": 0.1184,
      "step": 1994
    },
    {
      "epoch": 1.4349937061679554,
      "grad_norm": 5.573712647649542,
      "learning_rate": 2.8563701071213603e-06,
      "loss": 0.2207,
      "step": 1995
    },
    {
      "epoch": 1.4357130012587664,
      "grad_norm": 3.857616752790406,
      "learning_rate": 2.856224784224827e-06,
      "loss": 0.1051,
      "step": 1996
    },
    {
      "epoch": 1.4364322963495775,
      "grad_norm": 4.030817745454483,
      "learning_rate": 2.8560793915481476e-06,
      "loss": 0.1789,
      "step": 1997
    },
    {
      "epoch": 1.4371515914403883,
      "grad_norm": 3.3241275062078977,
      "learning_rate": 2.8559339290988028e-06,
      "loss": 0.0269,
      "step": 1998
    },
    {
      "epoch": 1.4378708865311993,
      "grad_norm": 2.2123817156316297,
      "learning_rate": 2.855788396884277e-06,
      "loss": 0.0075,
      "step": 1999
    },
    {
      "epoch": 1.4385901816220104,
      "grad_norm": 6.6883924643440995,
      "learning_rate": 2.8556427949120587e-06,
      "loss": 0.2912,
      "step": 2000
    },
    {
      "epoch": 1.4393094767128214,
      "grad_norm": 6.6418425700081425,
      "learning_rate": 2.8554971231896387e-06,
      "loss": 0.1646,
      "step": 2001
    },
    {
      "epoch": 1.4400287718036324,
      "grad_norm": 4.106760965615277,
      "learning_rate": 2.8553513817245117e-06,
      "loss": 0.0397,
      "step": 2002
    },
    {
      "epoch": 1.4407480668944435,
      "grad_norm": 6.027272729648282,
      "learning_rate": 2.8552055705241777e-06,
      "loss": 0.0673,
      "step": 2003
    },
    {
      "epoch": 1.4414673619852545,
      "grad_norm": 5.069987811777948,
      "learning_rate": 2.8550596895961373e-06,
      "loss": 0.1126,
      "step": 2004
    },
    {
      "epoch": 1.4421866570760655,
      "grad_norm": 4.201847354434991,
      "learning_rate": 2.8549137389478977e-06,
      "loss": 0.1438,
      "step": 2005
    },
    {
      "epoch": 1.4429059521668766,
      "grad_norm": 3.975616438255497,
      "learning_rate": 2.8547677185869673e-06,
      "loss": 0.1163,
      "step": 2006
    },
    {
      "epoch": 1.4436252472576876,
      "grad_norm": 3.5032386694636113,
      "learning_rate": 2.8546216285208603e-06,
      "loss": 0.184,
      "step": 2007
    },
    {
      "epoch": 1.4443445423484984,
      "grad_norm": 8.388021678809483,
      "learning_rate": 2.854475468757092e-06,
      "loss": 0.2439,
      "step": 2008
    },
    {
      "epoch": 1.4450638374393094,
      "grad_norm": 4.663240467198657,
      "learning_rate": 2.854329239303183e-06,
      "loss": 0.0992,
      "step": 2009
    },
    {
      "epoch": 1.4457831325301205,
      "grad_norm": 6.709242882617589,
      "learning_rate": 2.8541829401666576e-06,
      "loss": 0.1276,
      "step": 2010
    },
    {
      "epoch": 1.4465024276209315,
      "grad_norm": 1.0214979926387269,
      "learning_rate": 2.8540365713550427e-06,
      "loss": 0.0244,
      "step": 2011
    },
    {
      "epoch": 1.4472217227117425,
      "grad_norm": 5.602976734843479,
      "learning_rate": 2.8538901328758686e-06,
      "loss": 0.2921,
      "step": 2012
    },
    {
      "epoch": 1.4479410178025536,
      "grad_norm": 5.639849342658185,
      "learning_rate": 2.853743624736671e-06,
      "loss": 0.1666,
      "step": 2013
    },
    {
      "epoch": 1.4486603128933644,
      "grad_norm": 4.030356849308093,
      "learning_rate": 2.853597046944988e-06,
      "loss": 0.0789,
      "step": 2014
    },
    {
      "epoch": 1.4493796079841754,
      "grad_norm": 1.9121352050018268,
      "learning_rate": 2.85345039950836e-06,
      "loss": 0.02,
      "step": 2015
    },
    {
      "epoch": 1.4500989030749865,
      "grad_norm": 4.922742816632193,
      "learning_rate": 2.853303682434333e-06,
      "loss": 0.0197,
      "step": 2016
    },
    {
      "epoch": 1.4508181981657975,
      "grad_norm": 3.6686333502637885,
      "learning_rate": 2.8531568957304567e-06,
      "loss": 0.1278,
      "step": 2017
    },
    {
      "epoch": 1.4515374932566085,
      "grad_norm": 7.560223797392064,
      "learning_rate": 2.8530100394042818e-06,
      "loss": 0.365,
      "step": 2018
    },
    {
      "epoch": 1.4522567883474196,
      "grad_norm": 5.954246705072298,
      "learning_rate": 2.8528631134633657e-06,
      "loss": 0.0735,
      "step": 2019
    },
    {
      "epoch": 1.4529760834382306,
      "grad_norm": 4.8185372937658295,
      "learning_rate": 2.8527161179152675e-06,
      "loss": 0.0334,
      "step": 2020
    },
    {
      "epoch": 1.4536953785290416,
      "grad_norm": 3.8525377363520157,
      "learning_rate": 2.8525690527675503e-06,
      "loss": 0.0583,
      "step": 2021
    },
    {
      "epoch": 1.4544146736198527,
      "grad_norm": 2.8681137010864366,
      "learning_rate": 2.852421918027781e-06,
      "loss": 0.0701,
      "step": 2022
    },
    {
      "epoch": 1.4551339687106635,
      "grad_norm": 1.4605679622233498,
      "learning_rate": 2.85227471370353e-06,
      "loss": 0.0138,
      "step": 2023
    },
    {
      "epoch": 1.4558532638014745,
      "grad_norm": 2.2306526101979873,
      "learning_rate": 2.852127439802371e-06,
      "loss": 0.0083,
      "step": 2024
    },
    {
      "epoch": 1.4565725588922855,
      "grad_norm": 6.412544721770911,
      "learning_rate": 2.8519800963318817e-06,
      "loss": 0.142,
      "step": 2025
    },
    {
      "epoch": 1.4572918539830966,
      "grad_norm": 2.2109584020261424,
      "learning_rate": 2.851832683299643e-06,
      "loss": 0.0693,
      "step": 2026
    },
    {
      "epoch": 1.4580111490739076,
      "grad_norm": 3.100725321871934,
      "learning_rate": 2.85168520071324e-06,
      "loss": 0.0323,
      "step": 2027
    },
    {
      "epoch": 1.4587304441647186,
      "grad_norm": 4.2648413371460565,
      "learning_rate": 2.8515376485802603e-06,
      "loss": 0.2209,
      "step": 2028
    },
    {
      "epoch": 1.4594497392555295,
      "grad_norm": 5.760164744133177,
      "learning_rate": 2.8513900269082963e-06,
      "loss": 0.0416,
      "step": 2029
    },
    {
      "epoch": 1.4601690343463405,
      "grad_norm": 1.8832845216831633,
      "learning_rate": 2.851242335704943e-06,
      "loss": 0.0745,
      "step": 2030
    },
    {
      "epoch": 1.4608883294371515,
      "grad_norm": 3.3336648414078622,
      "learning_rate": 2.8510945749777997e-06,
      "loss": 0.0488,
      "step": 2031
    },
    {
      "epoch": 1.4616076245279626,
      "grad_norm": 2.8156889918014176,
      "learning_rate": 2.8509467447344685e-06,
      "loss": 0.0635,
      "step": 2032
    },
    {
      "epoch": 1.4623269196187736,
      "grad_norm": 2.7023495191631186,
      "learning_rate": 2.8507988449825556e-06,
      "loss": 0.0592,
      "step": 2033
    },
    {
      "epoch": 1.4630462147095846,
      "grad_norm": 3.2573707948615764,
      "learning_rate": 2.850650875729671e-06,
      "loss": 0.03,
      "step": 2034
    },
    {
      "epoch": 1.4637655098003957,
      "grad_norm": 6.956052131891015,
      "learning_rate": 2.850502836983428e-06,
      "loss": 0.1123,
      "step": 2035
    },
    {
      "epoch": 1.4644848048912067,
      "grad_norm": 3.7428585578782525,
      "learning_rate": 2.850354728751443e-06,
      "loss": 0.0842,
      "step": 2036
    },
    {
      "epoch": 1.4652040999820177,
      "grad_norm": 3.805828719010263,
      "learning_rate": 2.8502065510413373e-06,
      "loss": 0.0932,
      "step": 2037
    },
    {
      "epoch": 1.4659233950728288,
      "grad_norm": 4.28252119918998,
      "learning_rate": 2.8500583038607338e-06,
      "loss": 0.1393,
      "step": 2038
    },
    {
      "epoch": 1.4666426901636396,
      "grad_norm": 7.973858689388834,
      "learning_rate": 2.8499099872172613e-06,
      "loss": 0.1917,
      "step": 2039
    },
    {
      "epoch": 1.4673619852544506,
      "grad_norm": 8.223178114991434,
      "learning_rate": 2.8497616011185496e-06,
      "loss": 0.0656,
      "step": 2040
    },
    {
      "epoch": 1.4680812803452616,
      "grad_norm": 5.168299555372078,
      "learning_rate": 2.8496131455722342e-06,
      "loss": 0.1017,
      "step": 2041
    },
    {
      "epoch": 1.4688005754360727,
      "grad_norm": 3.559538930193632,
      "learning_rate": 2.8494646205859537e-06,
      "loss": 0.1353,
      "step": 2042
    },
    {
      "epoch": 1.4695198705268837,
      "grad_norm": 5.266724370704678,
      "learning_rate": 2.849316026167349e-06,
      "loss": 0.0966,
      "step": 2043
    },
    {
      "epoch": 1.4702391656176945,
      "grad_norm": 2.809598015565207,
      "learning_rate": 2.849167362324066e-06,
      "loss": 0.0221,
      "step": 2044
    },
    {
      "epoch": 1.4709584607085056,
      "grad_norm": 4.1877156425923125,
      "learning_rate": 2.8490186290637544e-06,
      "loss": 0.0822,
      "step": 2045
    },
    {
      "epoch": 1.4716777557993166,
      "grad_norm": 4.088972091034248,
      "learning_rate": 2.8488698263940664e-06,
      "loss": 0.1396,
      "step": 2046
    },
    {
      "epoch": 1.4723970508901276,
      "grad_norm": 2.109749846862112,
      "learning_rate": 2.8487209543226572e-06,
      "loss": 0.0369,
      "step": 2047
    },
    {
      "epoch": 1.4731163459809387,
      "grad_norm": 2.508024449228336,
      "learning_rate": 2.8485720128571874e-06,
      "loss": 0.0647,
      "step": 2048
    },
    {
      "epoch": 1.4738356410717497,
      "grad_norm": 2.776322273845714,
      "learning_rate": 2.8484230020053206e-06,
      "loss": 0.0856,
      "step": 2049
    },
    {
      "epoch": 1.4745549361625607,
      "grad_norm": 7.516425647591088,
      "learning_rate": 2.848273921774723e-06,
      "loss": 0.094,
      "step": 2050
    },
    {
      "epoch": 1.4752742312533718,
      "grad_norm": 2.093822245357266,
      "learning_rate": 2.8481247721730657e-06,
      "loss": 0.0419,
      "step": 2051
    },
    {
      "epoch": 1.4759935263441828,
      "grad_norm": 6.340528833023023,
      "learning_rate": 2.847975553208022e-06,
      "loss": 0.2125,
      "step": 2052
    },
    {
      "epoch": 1.4767128214349938,
      "grad_norm": 7.041347015291274,
      "learning_rate": 2.8478262648872696e-06,
      "loss": 0.2156,
      "step": 2053
    },
    {
      "epoch": 1.4774321165258046,
      "grad_norm": 3.432607818547082,
      "learning_rate": 2.8476769072184897e-06,
      "loss": 0.0273,
      "step": 2054
    },
    {
      "epoch": 1.4781514116166157,
      "grad_norm": 4.094829800066383,
      "learning_rate": 2.8475274802093675e-06,
      "loss": 0.1143,
      "step": 2055
    },
    {
      "epoch": 1.4788707067074267,
      "grad_norm": 4.5262466889442585,
      "learning_rate": 2.847377983867591e-06,
      "loss": 0.1708,
      "step": 2056
    },
    {
      "epoch": 1.4795900017982377,
      "grad_norm": 2.0823664496833936,
      "learning_rate": 2.8472284182008523e-06,
      "loss": 0.0622,
      "step": 2057
    },
    {
      "epoch": 1.4803092968890488,
      "grad_norm": 4.501304531567463,
      "learning_rate": 2.847078783216846e-06,
      "loss": 0.1086,
      "step": 2058
    },
    {
      "epoch": 1.4810285919798598,
      "grad_norm": 2.037806446765926,
      "learning_rate": 2.8469290789232715e-06,
      "loss": 0.0504,
      "step": 2059
    },
    {
      "epoch": 1.4817478870706706,
      "grad_norm": 2.4295263506985254,
      "learning_rate": 2.8467793053278316e-06,
      "loss": 0.0509,
      "step": 2060
    },
    {
      "epoch": 1.4824671821614817,
      "grad_norm": 4.652872837521685,
      "learning_rate": 2.8466294624382323e-06,
      "loss": 0.1332,
      "step": 2061
    },
    {
      "epoch": 1.4831864772522927,
      "grad_norm": 6.830314820488075,
      "learning_rate": 2.8464795502621833e-06,
      "loss": 0.2922,
      "step": 2062
    },
    {
      "epoch": 1.4839057723431037,
      "grad_norm": 4.253866968653982,
      "learning_rate": 2.8463295688073976e-06,
      "loss": 0.1309,
      "step": 2063
    },
    {
      "epoch": 1.4846250674339148,
      "grad_norm": 5.088782440817938,
      "learning_rate": 2.8461795180815926e-06,
      "loss": 0.0897,
      "step": 2064
    },
    {
      "epoch": 1.4853443625247258,
      "grad_norm": 3.3649566755197364,
      "learning_rate": 2.8460293980924876e-06,
      "loss": 0.0984,
      "step": 2065
    },
    {
      "epoch": 1.4860636576155368,
      "grad_norm": 6.0273737748079075,
      "learning_rate": 2.8458792088478072e-06,
      "loss": 0.1303,
      "step": 2066
    },
    {
      "epoch": 1.4867829527063479,
      "grad_norm": 8.346312372214113,
      "learning_rate": 2.8457289503552794e-06,
      "loss": 0.3255,
      "step": 2067
    },
    {
      "epoch": 1.487502247797159,
      "grad_norm": 7.722930486457627,
      "learning_rate": 2.8455786226226345e-06,
      "loss": 0.2197,
      "step": 2068
    },
    {
      "epoch": 1.48822154288797,
      "grad_norm": 4.1457954984655085,
      "learning_rate": 2.8454282256576073e-06,
      "loss": 0.1395,
      "step": 2069
    },
    {
      "epoch": 1.4889408379787807,
      "grad_norm": 2.992790058022124,
      "learning_rate": 2.845277759467936e-06,
      "loss": 0.0806,
      "step": 2070
    },
    {
      "epoch": 1.4896601330695918,
      "grad_norm": 3.988386463574146,
      "learning_rate": 2.8451272240613626e-06,
      "loss": 0.0759,
      "step": 2071
    },
    {
      "epoch": 1.4903794281604028,
      "grad_norm": 4.860492825399819,
      "learning_rate": 2.8449766194456324e-06,
      "loss": 0.1433,
      "step": 2072
    },
    {
      "epoch": 1.4910987232512138,
      "grad_norm": 3.730774308350756,
      "learning_rate": 2.8448259456284936e-06,
      "loss": 0.1141,
      "step": 2073
    },
    {
      "epoch": 1.4918180183420249,
      "grad_norm": 3.162546996326205,
      "learning_rate": 2.8446752026176997e-06,
      "loss": 0.0965,
      "step": 2074
    },
    {
      "epoch": 1.4925373134328357,
      "grad_norm": 3.013932178510901,
      "learning_rate": 2.8445243904210052e-06,
      "loss": 0.0767,
      "step": 2075
    },
    {
      "epoch": 1.4932566085236467,
      "grad_norm": 8.156451224532342,
      "learning_rate": 2.8443735090461715e-06,
      "loss": 0.2668,
      "step": 2076
    },
    {
      "epoch": 1.4939759036144578,
      "grad_norm": 7.42329868406929,
      "learning_rate": 2.8442225585009604e-06,
      "loss": 0.0384,
      "step": 2077
    },
    {
      "epoch": 1.4946951987052688,
      "grad_norm": 6.91169662433428,
      "learning_rate": 2.844071538793139e-06,
      "loss": 0.0886,
      "step": 2078
    },
    {
      "epoch": 1.4954144937960798,
      "grad_norm": 5.8427580964208286,
      "learning_rate": 2.843920449930478e-06,
      "loss": 0.3789,
      "step": 2079
    },
    {
      "epoch": 1.4961337888868909,
      "grad_norm": 0.8940149760610763,
      "learning_rate": 2.8437692919207497e-06,
      "loss": 0.0159,
      "step": 2080
    },
    {
      "epoch": 1.496853083977702,
      "grad_norm": 4.320881760156023,
      "learning_rate": 2.843618064771733e-06,
      "loss": 0.094,
      "step": 2081
    },
    {
      "epoch": 1.497572379068513,
      "grad_norm": 3.0063002444265416,
      "learning_rate": 2.8434667684912085e-06,
      "loss": 0.011,
      "step": 2082
    },
    {
      "epoch": 1.498291674159324,
      "grad_norm": 5.59085829591869,
      "learning_rate": 2.84331540308696e-06,
      "loss": 0.2387,
      "step": 2083
    },
    {
      "epoch": 1.499010969250135,
      "grad_norm": 6.100495860841695,
      "learning_rate": 2.843163968566776e-06,
      "loss": 0.1665,
      "step": 2084
    },
    {
      "epoch": 1.4997302643409458,
      "grad_norm": 3.0958914749470394,
      "learning_rate": 2.8430124649384477e-06,
      "loss": 0.0429,
      "step": 2085
    },
    {
      "epoch": 1.5004495594317568,
      "grad_norm": 4.280542763861561,
      "learning_rate": 2.842860892209771e-06,
      "loss": 0.0997,
      "step": 2086
    },
    {
      "epoch": 1.5011688545225679,
      "grad_norm": 4.056064159836871,
      "learning_rate": 2.8427092503885435e-06,
      "loss": 0.1334,
      "step": 2087
    },
    {
      "epoch": 1.501888149613379,
      "grad_norm": 6.841318298812241,
      "learning_rate": 2.842557539482568e-06,
      "loss": 0.1717,
      "step": 2088
    },
    {
      "epoch": 1.50260744470419,
      "grad_norm": 0.3478902733757561,
      "learning_rate": 2.842405759499651e-06,
      "loss": 0.0014,
      "step": 2089
    },
    {
      "epoch": 1.5033267397950008,
      "grad_norm": 5.5537515948259575,
      "learning_rate": 2.8422539104476008e-06,
      "loss": 0.0518,
      "step": 2090
    },
    {
      "epoch": 1.5040460348858118,
      "grad_norm": 4.4363030448350385,
      "learning_rate": 2.84210199233423e-06,
      "loss": 0.0504,
      "step": 2091
    },
    {
      "epoch": 1.5047653299766228,
      "grad_norm": 4.944684416211211,
      "learning_rate": 2.841950005167357e-06,
      "loss": 0.1242,
      "step": 2092
    },
    {
      "epoch": 1.5054846250674339,
      "grad_norm": 5.592434384718533,
      "learning_rate": 2.8417979489548e-06,
      "loss": 0.2782,
      "step": 2093
    },
    {
      "epoch": 1.506203920158245,
      "grad_norm": 5.1179840008794475,
      "learning_rate": 2.8416458237043824e-06,
      "loss": 0.2164,
      "step": 2094
    },
    {
      "epoch": 1.506923215249056,
      "grad_norm": 3.3254603002561804,
      "learning_rate": 2.8414936294239327e-06,
      "loss": 0.0649,
      "step": 2095
    },
    {
      "epoch": 1.507642510339867,
      "grad_norm": 5.490782884881453,
      "learning_rate": 2.8413413661212804e-06,
      "loss": 0.1622,
      "step": 2096
    },
    {
      "epoch": 1.508361805430678,
      "grad_norm": 5.602155450846729,
      "learning_rate": 2.8411890338042607e-06,
      "loss": 0.1257,
      "step": 2097
    },
    {
      "epoch": 1.509081100521489,
      "grad_norm": 4.74006037373041,
      "learning_rate": 2.8410366324807106e-06,
      "loss": 0.1273,
      "step": 2098
    },
    {
      "epoch": 1.5098003956123,
      "grad_norm": 4.810121949424923,
      "learning_rate": 2.8408841621584715e-06,
      "loss": 0.3123,
      "step": 2099
    },
    {
      "epoch": 1.510519690703111,
      "grad_norm": 3.081775619299007,
      "learning_rate": 2.8407316228453883e-06,
      "loss": 0.0718,
      "step": 2100
    },
    {
      "epoch": 1.511238985793922,
      "grad_norm": 2.262127417717525,
      "learning_rate": 2.8405790145493096e-06,
      "loss": 0.0666,
      "step": 2101
    },
    {
      "epoch": 1.511958280884733,
      "grad_norm": 3.7019262391922627,
      "learning_rate": 2.8404263372780874e-06,
      "loss": 0.1146,
      "step": 2102
    },
    {
      "epoch": 1.512677575975544,
      "grad_norm": 2.4618931060105673,
      "learning_rate": 2.8402735910395774e-06,
      "loss": 0.0159,
      "step": 2103
    },
    {
      "epoch": 1.513396871066355,
      "grad_norm": 3.6972846953421024,
      "learning_rate": 2.8401207758416376e-06,
      "loss": 0.1453,
      "step": 2104
    },
    {
      "epoch": 1.5141161661571658,
      "grad_norm": 2.796573496131828,
      "learning_rate": 2.839967891692132e-06,
      "loss": 0.0229,
      "step": 2105
    },
    {
      "epoch": 1.5148354612479769,
      "grad_norm": 3.491710770460096,
      "learning_rate": 2.839814938598926e-06,
      "loss": 0.122,
      "step": 2106
    },
    {
      "epoch": 1.515554756338788,
      "grad_norm": 2.9846426227939897,
      "learning_rate": 2.8396619165698897e-06,
      "loss": 0.0535,
      "step": 2107
    },
    {
      "epoch": 1.516274051429599,
      "grad_norm": 4.27477578987767,
      "learning_rate": 2.8395088256128955e-06,
      "loss": 0.0136,
      "step": 2108
    },
    {
      "epoch": 1.51699334652041,
      "grad_norm": 4.233778688584056,
      "learning_rate": 2.839355665735821e-06,
      "loss": 0.0798,
      "step": 2109
    },
    {
      "epoch": 1.517712641611221,
      "grad_norm": 3.401993069318298,
      "learning_rate": 2.8392024369465464e-06,
      "loss": 0.1459,
      "step": 2110
    },
    {
      "epoch": 1.518431936702032,
      "grad_norm": 6.029752369915126,
      "learning_rate": 2.8390491392529557e-06,
      "loss": 0.1322,
      "step": 2111
    },
    {
      "epoch": 1.519151231792843,
      "grad_norm": 4.32490982016153,
      "learning_rate": 2.8388957726629362e-06,
      "loss": 0.0405,
      "step": 2112
    },
    {
      "epoch": 1.519870526883654,
      "grad_norm": 4.280474133523757,
      "learning_rate": 2.838742337184378e-06,
      "loss": 0.1767,
      "step": 2113
    },
    {
      "epoch": 1.5205898219744651,
      "grad_norm": 7.766998333039946,
      "learning_rate": 2.8385888328251778e-06,
      "loss": 0.0744,
      "step": 2114
    },
    {
      "epoch": 1.5213091170652762,
      "grad_norm": 3.926046147946892,
      "learning_rate": 2.8384352595932313e-06,
      "loss": 0.0517,
      "step": 2115
    },
    {
      "epoch": 1.5220284121560872,
      "grad_norm": 3.2565401292161242,
      "learning_rate": 2.8382816174964415e-06,
      "loss": 0.0967,
      "step": 2116
    },
    {
      "epoch": 1.522747707246898,
      "grad_norm": 3.4892653674603062,
      "learning_rate": 2.8381279065427133e-06,
      "loss": 0.0671,
      "step": 2117
    },
    {
      "epoch": 1.523467002337709,
      "grad_norm": 9.527947159507624,
      "learning_rate": 2.837974126739955e-06,
      "loss": 0.3026,
      "step": 2118
    },
    {
      "epoch": 1.52418629742852,
      "grad_norm": 5.393193564746958,
      "learning_rate": 2.8378202780960796e-06,
      "loss": 0.1253,
      "step": 2119
    },
    {
      "epoch": 1.524905592519331,
      "grad_norm": 5.206110148094954,
      "learning_rate": 2.837666360619002e-06,
      "loss": 0.1697,
      "step": 2120
    },
    {
      "epoch": 1.525624887610142,
      "grad_norm": 4.556115272105652,
      "learning_rate": 2.837512374316642e-06,
      "loss": 0.1253,
      "step": 2121
    },
    {
      "epoch": 1.526344182700953,
      "grad_norm": 7.753247368191384,
      "learning_rate": 2.8373583191969223e-06,
      "loss": 0.1275,
      "step": 2122
    },
    {
      "epoch": 1.527063477791764,
      "grad_norm": 2.8192905021359613,
      "learning_rate": 2.8372041952677698e-06,
      "loss": 0.0379,
      "step": 2123
    },
    {
      "epoch": 1.527782772882575,
      "grad_norm": 4.6326478281778485,
      "learning_rate": 2.8370500025371138e-06,
      "loss": 0.0361,
      "step": 2124
    },
    {
      "epoch": 1.528502067973386,
      "grad_norm": 5.233258207349289,
      "learning_rate": 2.8368957410128885e-06,
      "loss": 0.0796,
      "step": 2125
    },
    {
      "epoch": 1.529221363064197,
      "grad_norm": 6.6001008617478245,
      "learning_rate": 2.83674141070303e-06,
      "loss": 0.045,
      "step": 2126
    },
    {
      "epoch": 1.5299406581550081,
      "grad_norm": 4.955776387563965,
      "learning_rate": 2.8365870116154796e-06,
      "loss": 0.1265,
      "step": 2127
    },
    {
      "epoch": 1.5306599532458192,
      "grad_norm": 4.485594805529013,
      "learning_rate": 2.8364325437581805e-06,
      "loss": 0.1343,
      "step": 2128
    },
    {
      "epoch": 1.5313792483366302,
      "grad_norm": 2.7781981584041135,
      "learning_rate": 2.8362780071390818e-06,
      "loss": 0.0713,
      "step": 2129
    },
    {
      "epoch": 1.5320985434274412,
      "grad_norm": 4.578639959764174,
      "learning_rate": 2.8361234017661336e-06,
      "loss": 0.0697,
      "step": 2130
    },
    {
      "epoch": 1.5328178385182523,
      "grad_norm": 1.5602621036185553,
      "learning_rate": 2.835968727647291e-06,
      "loss": 0.0046,
      "step": 2131
    },
    {
      "epoch": 1.533537133609063,
      "grad_norm": 5.95512753043447,
      "learning_rate": 2.8358139847905116e-06,
      "loss": 0.1301,
      "step": 2132
    },
    {
      "epoch": 1.534256428699874,
      "grad_norm": 4.202836573840528,
      "learning_rate": 2.8356591732037583e-06,
      "loss": 0.2462,
      "step": 2133
    },
    {
      "epoch": 1.5349757237906851,
      "grad_norm": 4.508608078966251,
      "learning_rate": 2.835504292894996e-06,
      "loss": 0.1211,
      "step": 2134
    },
    {
      "epoch": 1.5356950188814962,
      "grad_norm": 3.145227810248748,
      "learning_rate": 2.835349343872193e-06,
      "loss": 0.1013,
      "step": 2135
    },
    {
      "epoch": 1.536414313972307,
      "grad_norm": 2.3072490212915695,
      "learning_rate": 2.835194326143322e-06,
      "loss": 0.0485,
      "step": 2136
    },
    {
      "epoch": 1.537133609063118,
      "grad_norm": 5.600000829856032,
      "learning_rate": 2.835039239716359e-06,
      "loss": 0.1174,
      "step": 2137
    },
    {
      "epoch": 1.537852904153929,
      "grad_norm": 7.163622283089642,
      "learning_rate": 2.8348840845992843e-06,
      "loss": 0.2072,
      "step": 2138
    },
    {
      "epoch": 1.53857219924474,
      "grad_norm": 5.641013139238723,
      "learning_rate": 2.8347288608000796e-06,
      "loss": 0.1043,
      "step": 2139
    },
    {
      "epoch": 1.5392914943355511,
      "grad_norm": 5.000928066496515,
      "learning_rate": 2.834573568326732e-06,
      "loss": 0.1205,
      "step": 2140
    },
    {
      "epoch": 1.5400107894263622,
      "grad_norm": 5.483672257956804,
      "learning_rate": 2.8344182071872316e-06,
      "loss": 0.0393,
      "step": 2141
    },
    {
      "epoch": 1.5407300845171732,
      "grad_norm": 8.45431669748343,
      "learning_rate": 2.8342627773895717e-06,
      "loss": 0.14,
      "step": 2142
    },
    {
      "epoch": 1.5414493796079842,
      "grad_norm": 5.670442439429874,
      "learning_rate": 2.83410727894175e-06,
      "loss": 0.2208,
      "step": 2143
    },
    {
      "epoch": 1.5421686746987953,
      "grad_norm": 2.172276367616782,
      "learning_rate": 2.833951711851767e-06,
      "loss": 0.1047,
      "step": 2144
    },
    {
      "epoch": 1.5428879697896063,
      "grad_norm": 4.652407188300584,
      "learning_rate": 2.8337960761276263e-06,
      "loss": 0.1199,
      "step": 2145
    },
    {
      "epoch": 1.5436072648804173,
      "grad_norm": 3.3518755937829177,
      "learning_rate": 2.8336403717773364e-06,
      "loss": 0.087,
      "step": 2146
    },
    {
      "epoch": 1.5443265599712281,
      "grad_norm": 5.78383195192292,
      "learning_rate": 2.833484598808908e-06,
      "loss": 0.3425,
      "step": 2147
    },
    {
      "epoch": 1.5450458550620392,
      "grad_norm": 1.9750756454812182,
      "learning_rate": 2.833328757230356e-06,
      "loss": 0.0176,
      "step": 2148
    },
    {
      "epoch": 1.5457651501528502,
      "grad_norm": 1.3809507169742352,
      "learning_rate": 2.8331728470496998e-06,
      "loss": 0.0079,
      "step": 2149
    },
    {
      "epoch": 1.5464844452436612,
      "grad_norm": 5.144868372888545,
      "learning_rate": 2.833016868274959e-06,
      "loss": 0.0664,
      "step": 2150
    },
    {
      "epoch": 1.547203740334472,
      "grad_norm": 7.6631992327283545,
      "learning_rate": 2.832860820914161e-06,
      "loss": 0.2146,
      "step": 2151
    },
    {
      "epoch": 1.547923035425283,
      "grad_norm": 3.8504844262217985,
      "learning_rate": 2.832704704975334e-06,
      "loss": 0.1732,
      "step": 2152
    },
    {
      "epoch": 1.5486423305160941,
      "grad_norm": 2.2987732676595836,
      "learning_rate": 2.8325485204665104e-06,
      "loss": 0.0248,
      "step": 2153
    },
    {
      "epoch": 1.5493616256069052,
      "grad_norm": 3.1349656365080882,
      "learning_rate": 2.8323922673957264e-06,
      "loss": 0.0342,
      "step": 2154
    },
    {
      "epoch": 1.5500809206977162,
      "grad_norm": 4.032634076190705,
      "learning_rate": 2.832235945771021e-06,
      "loss": 0.1517,
      "step": 2155
    },
    {
      "epoch": 1.5508002157885272,
      "grad_norm": 2.9285883897014675,
      "learning_rate": 2.832079555600437e-06,
      "loss": 0.0608,
      "step": 2156
    },
    {
      "epoch": 1.5515195108793383,
      "grad_norm": 4.342767791337391,
      "learning_rate": 2.831923096892022e-06,
      "loss": 0.0206,
      "step": 2157
    },
    {
      "epoch": 1.5522388059701493,
      "grad_norm": 4.963240834884954,
      "learning_rate": 2.831766569653826e-06,
      "loss": 0.0904,
      "step": 2158
    },
    {
      "epoch": 1.5529581010609603,
      "grad_norm": 5.293310723694916,
      "learning_rate": 2.8316099738939016e-06,
      "loss": 0.1143,
      "step": 2159
    },
    {
      "epoch": 1.5536773961517714,
      "grad_norm": 2.4109550114761547,
      "learning_rate": 2.8314533096203066e-06,
      "loss": 0.0669,
      "step": 2160
    },
    {
      "epoch": 1.5543966912425824,
      "grad_norm": 5.225078460235985,
      "learning_rate": 2.8312965768411018e-06,
      "loss": 0.2022,
      "step": 2161
    },
    {
      "epoch": 1.5551159863333934,
      "grad_norm": 8.052550713583928,
      "learning_rate": 2.831139775564351e-06,
      "loss": 0.3126,
      "step": 2162
    },
    {
      "epoch": 1.5558352814242042,
      "grad_norm": 4.311607885095645,
      "learning_rate": 2.8309829057981213e-06,
      "loss": 0.1268,
      "step": 2163
    },
    {
      "epoch": 1.5565545765150153,
      "grad_norm": 5.597379773741944,
      "learning_rate": 2.8308259675504854e-06,
      "loss": 0.1181,
      "step": 2164
    },
    {
      "epoch": 1.5572738716058263,
      "grad_norm": 3.469146886917397,
      "learning_rate": 2.830668960829517e-06,
      "loss": 0.1413,
      "step": 2165
    },
    {
      "epoch": 1.5579931666966373,
      "grad_norm": 5.873459104704315,
      "learning_rate": 2.830511885643295e-06,
      "loss": 0.0917,
      "step": 2166
    },
    {
      "epoch": 1.5587124617874482,
      "grad_norm": 4.351247191572507,
      "learning_rate": 2.8303547419999005e-06,
      "loss": 0.0438,
      "step": 2167
    },
    {
      "epoch": 1.5594317568782592,
      "grad_norm": 4.579715624408742,
      "learning_rate": 2.830197529907419e-06,
      "loss": 0.0301,
      "step": 2168
    },
    {
      "epoch": 1.5601510519690702,
      "grad_norm": 3.561731954684281,
      "learning_rate": 2.83004024937394e-06,
      "loss": 0.0966,
      "step": 2169
    },
    {
      "epoch": 1.5608703470598813,
      "grad_norm": 6.0243790476827,
      "learning_rate": 2.829882900407555e-06,
      "loss": 0.1343,
      "step": 2170
    },
    {
      "epoch": 1.5615896421506923,
      "grad_norm": 3.2786841480857594,
      "learning_rate": 2.8297254830163604e-06,
      "loss": 0.1283,
      "step": 2171
    },
    {
      "epoch": 1.5623089372415033,
      "grad_norm": 9.864595709499882,
      "learning_rate": 2.8295679972084555e-06,
      "loss": 0.4016,
      "step": 2172
    },
    {
      "epoch": 1.5630282323323144,
      "grad_norm": 4.863301507240836,
      "learning_rate": 2.829410442991943e-06,
      "loss": 0.2683,
      "step": 2173
    },
    {
      "epoch": 1.5637475274231254,
      "grad_norm": 10.295005455750127,
      "learning_rate": 2.829252820374929e-06,
      "loss": 0.3946,
      "step": 2174
    },
    {
      "epoch": 1.5644668225139364,
      "grad_norm": 4.392501607213936,
      "learning_rate": 2.8290951293655245e-06,
      "loss": 0.1145,
      "step": 2175
    },
    {
      "epoch": 1.5651861176047475,
      "grad_norm": 3.773643525555693,
      "learning_rate": 2.8289373699718425e-06,
      "loss": 0.0928,
      "step": 2176
    },
    {
      "epoch": 1.5659054126955585,
      "grad_norm": 3.0179265026340807,
      "learning_rate": 2.828779542202e-06,
      "loss": 0.0537,
      "step": 2177
    },
    {
      "epoch": 1.5666247077863693,
      "grad_norm": 3.5954299563091676,
      "learning_rate": 2.8286216460641173e-06,
      "loss": 0.142,
      "step": 2178
    },
    {
      "epoch": 1.5673440028771803,
      "grad_norm": 6.858052538285908,
      "learning_rate": 2.8284636815663186e-06,
      "loss": 0.2409,
      "step": 2179
    },
    {
      "epoch": 1.5680632979679914,
      "grad_norm": 5.430650453224053,
      "learning_rate": 2.828305648716731e-06,
      "loss": 0.1071,
      "step": 2180
    },
    {
      "epoch": 1.5687825930588024,
      "grad_norm": 3.248055123298909,
      "learning_rate": 2.828147547523487e-06,
      "loss": 0.079,
      "step": 2181
    },
    {
      "epoch": 1.5695018881496132,
      "grad_norm": 2.0344750643084364,
      "learning_rate": 2.8279893779947197e-06,
      "loss": 0.0473,
      "step": 2182
    },
    {
      "epoch": 1.5702211832404243,
      "grad_norm": 3.8886855686785635,
      "learning_rate": 2.8278311401385675e-06,
      "loss": 0.0261,
      "step": 2183
    },
    {
      "epoch": 1.5709404783312353,
      "grad_norm": 4.521234663506471,
      "learning_rate": 2.8276728339631722e-06,
      "loss": 0.1006,
      "step": 2184
    },
    {
      "epoch": 1.5716597734220463,
      "grad_norm": 5.85412494818436,
      "learning_rate": 2.827514459476679e-06,
      "loss": 0.1395,
      "step": 2185
    },
    {
      "epoch": 1.5723790685128574,
      "grad_norm": 3.3530034789998604,
      "learning_rate": 2.8273560166872367e-06,
      "loss": 0.1225,
      "step": 2186
    },
    {
      "epoch": 1.5730983636036684,
      "grad_norm": 5.210246229598176,
      "learning_rate": 2.8271975056029968e-06,
      "loss": 0.0473,
      "step": 2187
    },
    {
      "epoch": 1.5738176586944794,
      "grad_norm": 5.026905690085372,
      "learning_rate": 2.827038926232116e-06,
      "loss": 0.054,
      "step": 2188
    },
    {
      "epoch": 1.5745369537852905,
      "grad_norm": 3.5326371526764517,
      "learning_rate": 2.8268802785827525e-06,
      "loss": 0.0485,
      "step": 2189
    },
    {
      "epoch": 1.5752562488761015,
      "grad_norm": 2.6430518956004003,
      "learning_rate": 2.8267215626630697e-06,
      "loss": 0.0354,
      "step": 2190
    },
    {
      "epoch": 1.5759755439669125,
      "grad_norm": 4.005273130577424,
      "learning_rate": 2.8265627784812333e-06,
      "loss": 0.1501,
      "step": 2191
    },
    {
      "epoch": 1.5766948390577236,
      "grad_norm": 5.806301907848369,
      "learning_rate": 2.8264039260454135e-06,
      "loss": 0.1681,
      "step": 2192
    },
    {
      "epoch": 1.5774141341485346,
      "grad_norm": 5.590764946339853,
      "learning_rate": 2.8262450053637827e-06,
      "loss": 0.2601,
      "step": 2193
    },
    {
      "epoch": 1.5781334292393454,
      "grad_norm": 4.194100201745286,
      "learning_rate": 2.8260860164445183e-06,
      "loss": 0.1805,
      "step": 2194
    },
    {
      "epoch": 1.5788527243301564,
      "grad_norm": 2.791708114715783,
      "learning_rate": 2.825926959295801e-06,
      "loss": 0.0332,
      "step": 2195
    },
    {
      "epoch": 1.5795720194209675,
      "grad_norm": 3.89971520548335,
      "learning_rate": 2.8257678339258136e-06,
      "loss": 0.0895,
      "step": 2196
    },
    {
      "epoch": 1.5802913145117785,
      "grad_norm": 2.926517474868108,
      "learning_rate": 2.8256086403427444e-06,
      "loss": 0.0092,
      "step": 2197
    },
    {
      "epoch": 1.5810106096025893,
      "grad_norm": 3.910635926420911,
      "learning_rate": 2.825449378554783e-06,
      "loss": 0.0686,
      "step": 2198
    },
    {
      "epoch": 1.5817299046934004,
      "grad_norm": 5.32665345577694,
      "learning_rate": 2.8252900485701245e-06,
      "loss": 0.1869,
      "step": 2199
    },
    {
      "epoch": 1.5824491997842114,
      "grad_norm": 5.924374069458105,
      "learning_rate": 2.8251306503969665e-06,
      "loss": 0.1249,
      "step": 2200
    },
    {
      "epoch": 1.5831684948750224,
      "grad_norm": 4.296047295252244,
      "learning_rate": 2.8249711840435105e-06,
      "loss": 0.1586,
      "step": 2201
    },
    {
      "epoch": 1.5838877899658335,
      "grad_norm": 3.996958827891937,
      "learning_rate": 2.8248116495179613e-06,
      "loss": 0.1272,
      "step": 2202
    },
    {
      "epoch": 1.5846070850566445,
      "grad_norm": 4.970200546125578,
      "learning_rate": 2.824652046828527e-06,
      "loss": 0.1521,
      "step": 2203
    },
    {
      "epoch": 1.5853263801474555,
      "grad_norm": 4.689615966087016,
      "learning_rate": 2.8244923759834194e-06,
      "loss": 0.0576,
      "step": 2204
    },
    {
      "epoch": 1.5860456752382666,
      "grad_norm": 6.0204970378752,
      "learning_rate": 2.824332636990854e-06,
      "loss": 0.0783,
      "step": 2205
    },
    {
      "epoch": 1.5867649703290776,
      "grad_norm": 8.514657860316047,
      "learning_rate": 2.82417282985905e-06,
      "loss": 0.2059,
      "step": 2206
    },
    {
      "epoch": 1.5874842654198886,
      "grad_norm": 4.1644252119388385,
      "learning_rate": 2.8240129545962284e-06,
      "loss": 0.1088,
      "step": 2207
    },
    {
      "epoch": 1.5882035605106997,
      "grad_norm": 3.6262486693879663,
      "learning_rate": 2.823853011210617e-06,
      "loss": 0.1078,
      "step": 2208
    },
    {
      "epoch": 1.5889228556015105,
      "grad_norm": 2.959788960634419,
      "learning_rate": 2.823692999710444e-06,
      "loss": 0.0832,
      "step": 2209
    },
    {
      "epoch": 1.5896421506923215,
      "grad_norm": 3.762232847064055,
      "learning_rate": 2.8235329201039425e-06,
      "loss": 0.1442,
      "step": 2210
    },
    {
      "epoch": 1.5903614457831325,
      "grad_norm": 6.115826253694773,
      "learning_rate": 2.8233727723993488e-06,
      "loss": 0.0253,
      "step": 2211
    },
    {
      "epoch": 1.5910807408739436,
      "grad_norm": 0.34769324865897927,
      "learning_rate": 2.823212556604903e-06,
      "loss": 0.0012,
      "step": 2212
    },
    {
      "epoch": 1.5918000359647544,
      "grad_norm": 2.547732307064375,
      "learning_rate": 2.823052272728848e-06,
      "loss": 0.1076,
      "step": 2213
    },
    {
      "epoch": 1.5925193310555654,
      "grad_norm": 3.0890038567971003,
      "learning_rate": 2.822891920779431e-06,
      "loss": 0.0732,
      "step": 2214
    },
    {
      "epoch": 1.5932386261463765,
      "grad_norm": 7.256156531818067,
      "learning_rate": 2.822731500764903e-06,
      "loss": 0.3677,
      "step": 2215
    },
    {
      "epoch": 1.5939579212371875,
      "grad_norm": 3.471913647728683,
      "learning_rate": 2.822571012693517e-06,
      "loss": 0.1305,
      "step": 2216
    },
    {
      "epoch": 1.5946772163279985,
      "grad_norm": 6.163097763007754,
      "learning_rate": 2.8224104565735303e-06,
      "loss": 0.0803,
      "step": 2217
    },
    {
      "epoch": 1.5953965114188096,
      "grad_norm": 3.3098696649244452,
      "learning_rate": 2.8222498324132045e-06,
      "loss": 0.0445,
      "step": 2218
    },
    {
      "epoch": 1.5961158065096206,
      "grad_norm": 6.246364002162796,
      "learning_rate": 2.8220891402208037e-06,
      "loss": 0.1418,
      "step": 2219
    },
    {
      "epoch": 1.5968351016004316,
      "grad_norm": 4.306632523253456,
      "learning_rate": 2.821928380004596e-06,
      "loss": 0.1117,
      "step": 2220
    },
    {
      "epoch": 1.5975543966912427,
      "grad_norm": 3.4714969927413044,
      "learning_rate": 2.8217675517728523e-06,
      "loss": 0.0765,
      "step": 2221
    },
    {
      "epoch": 1.5982736917820537,
      "grad_norm": 3.8205470990773076,
      "learning_rate": 2.821606655533848e-06,
      "loss": 0.1076,
      "step": 2222
    },
    {
      "epoch": 1.5989929868728647,
      "grad_norm": 2.8841160104454735,
      "learning_rate": 2.821445691295861e-06,
      "loss": 0.1278,
      "step": 2223
    },
    {
      "epoch": 1.5997122819636758,
      "grad_norm": 3.6395173762428796,
      "learning_rate": 2.821284659067174e-06,
      "loss": 0.1872,
      "step": 2224
    },
    {
      "epoch": 1.6004315770544866,
      "grad_norm": 2.8901669681731494,
      "learning_rate": 2.821123558856071e-06,
      "loss": 0.0316,
      "step": 2225
    },
    {
      "epoch": 1.6011508721452976,
      "grad_norm": 2.5862905311458317,
      "learning_rate": 2.820962390670842e-06,
      "loss": 0.1112,
      "step": 2226
    },
    {
      "epoch": 1.6018701672361086,
      "grad_norm": 5.803221851571076,
      "learning_rate": 2.820801154519779e-06,
      "loss": 0.264,
      "step": 2227
    },
    {
      "epoch": 1.6025894623269195,
      "grad_norm": 3.11843554181106,
      "learning_rate": 2.8206398504111787e-06,
      "loss": 0.039,
      "step": 2228
    },
    {
      "epoch": 1.6033087574177305,
      "grad_norm": 6.86113282444148,
      "learning_rate": 2.820478478353339e-06,
      "loss": 0.2357,
      "step": 2229
    },
    {
      "epoch": 1.6040280525085415,
      "grad_norm": 6.941501109239274,
      "learning_rate": 2.8203170383545643e-06,
      "loss": 0.179,
      "step": 2230
    },
    {
      "epoch": 1.6047473475993526,
      "grad_norm": 2.240716476669486,
      "learning_rate": 2.82015553042316e-06,
      "loss": 0.0272,
      "step": 2231
    },
    {
      "epoch": 1.6054666426901636,
      "grad_norm": 3.111928094951867,
      "learning_rate": 2.819993954567436e-06,
      "loss": 0.1592,
      "step": 2232
    },
    {
      "epoch": 1.6061859377809746,
      "grad_norm": 3.0023741082243434,
      "learning_rate": 2.8198323107957055e-06,
      "loss": 0.0648,
      "step": 2233
    },
    {
      "epoch": 1.6069052328717857,
      "grad_norm": 5.8812597294182005,
      "learning_rate": 2.8196705991162862e-06,
      "loss": 0.0343,
      "step": 2234
    },
    {
      "epoch": 1.6076245279625967,
      "grad_norm": 2.7439908128283856,
      "learning_rate": 2.819508819537498e-06,
      "loss": 0.0889,
      "step": 2235
    },
    {
      "epoch": 1.6083438230534077,
      "grad_norm": 2.923020489316437,
      "learning_rate": 2.8193469720676648e-06,
      "loss": 0.0244,
      "step": 2236
    },
    {
      "epoch": 1.6090631181442188,
      "grad_norm": 3.3689334171131167,
      "learning_rate": 2.8191850567151135e-06,
      "loss": 0.1249,
      "step": 2237
    },
    {
      "epoch": 1.6097824132350298,
      "grad_norm": 4.561428389612156,
      "learning_rate": 2.8190230734881753e-06,
      "loss": 0.0173,
      "step": 2238
    },
    {
      "epoch": 1.6105017083258408,
      "grad_norm": 5.528875668548176,
      "learning_rate": 2.818861022395185e-06,
      "loss": 0.1071,
      "step": 2239
    },
    {
      "epoch": 1.6112210034166516,
      "grad_norm": 5.414652000744565,
      "learning_rate": 2.8186989034444794e-06,
      "loss": 0.2672,
      "step": 2240
    },
    {
      "epoch": 1.6119402985074627,
      "grad_norm": 0.1915798511413906,
      "learning_rate": 2.8185367166444e-06,
      "loss": 0.0006,
      "step": 2241
    },
    {
      "epoch": 1.6126595935982737,
      "grad_norm": 1.7896864197806093,
      "learning_rate": 2.8183744620032927e-06,
      "loss": 0.0489,
      "step": 2242
    },
    {
      "epoch": 1.6133788886890847,
      "grad_norm": 5.360176939414118,
      "learning_rate": 2.818212139529505e-06,
      "loss": 0.2154,
      "step": 2243
    },
    {
      "epoch": 1.6140981837798956,
      "grad_norm": 3.614753660243484,
      "learning_rate": 2.818049749231388e-06,
      "loss": 0.0255,
      "step": 2244
    },
    {
      "epoch": 1.6148174788707066,
      "grad_norm": 0.8295565308390117,
      "learning_rate": 2.817887291117298e-06,
      "loss": 0.0016,
      "step": 2245
    },
    {
      "epoch": 1.6155367739615176,
      "grad_norm": 5.539456900748424,
      "learning_rate": 2.8177247651955935e-06,
      "loss": 0.1398,
      "step": 2246
    },
    {
      "epoch": 1.6162560690523287,
      "grad_norm": 4.704211833566174,
      "learning_rate": 2.817562171474636e-06,
      "loss": 0.0343,
      "step": 2247
    },
    {
      "epoch": 1.6169753641431397,
      "grad_norm": 5.250055237749541,
      "learning_rate": 2.817399509962793e-06,
      "loss": 0.2029,
      "step": 2248
    },
    {
      "epoch": 1.6176946592339507,
      "grad_norm": 3.6006835441705145,
      "learning_rate": 2.8172367806684317e-06,
      "loss": 0.0214,
      "step": 2249
    },
    {
      "epoch": 1.6184139543247618,
      "grad_norm": 2.011895212746637,
      "learning_rate": 2.8170739835999258e-06,
      "loss": 0.0094,
      "step": 2250
    },
    {
      "epoch": 1.6191332494155728,
      "grad_norm": 2.8655939483148813,
      "learning_rate": 2.8169111187656517e-06,
      "loss": 0.0773,
      "step": 2251
    },
    {
      "epoch": 1.6198525445063838,
      "grad_norm": 11.696558726027765,
      "learning_rate": 2.8167481861739884e-06,
      "loss": 0.131,
      "step": 2252
    },
    {
      "epoch": 1.6205718395971949,
      "grad_norm": 1.549969438440941,
      "learning_rate": 2.81658518583332e-06,
      "loss": 0.0493,
      "step": 2253
    },
    {
      "epoch": 1.621291134688006,
      "grad_norm": 5.285909841275818,
      "learning_rate": 2.816422117752032e-06,
      "loss": 0.0803,
      "step": 2254
    },
    {
      "epoch": 1.6220104297788167,
      "grad_norm": 2.9645018350910806,
      "learning_rate": 2.8162589819385157e-06,
      "loss": 0.057,
      "step": 2255
    },
    {
      "epoch": 1.6227297248696277,
      "grad_norm": 5.523642795654883,
      "learning_rate": 2.816095778401164e-06,
      "loss": 0.1699,
      "step": 2256
    },
    {
      "epoch": 1.6234490199604388,
      "grad_norm": 4.265287292529384,
      "learning_rate": 2.815932507148374e-06,
      "loss": 0.1583,
      "step": 2257
    },
    {
      "epoch": 1.6241683150512498,
      "grad_norm": 6.164838547940256,
      "learning_rate": 2.8157691681885463e-06,
      "loss": 0.1556,
      "step": 2258
    },
    {
      "epoch": 1.6248876101420606,
      "grad_norm": 6.47143839920109,
      "learning_rate": 2.8156057615300853e-06,
      "loss": 0.2227,
      "step": 2259
    },
    {
      "epoch": 1.6256069052328717,
      "grad_norm": 4.936288979930599,
      "learning_rate": 2.815442287181399e-06,
      "loss": 0.1467,
      "step": 2260
    },
    {
      "epoch": 1.6263262003236827,
      "grad_norm": 2.452710234812695,
      "learning_rate": 2.815278745150897e-06,
      "loss": 0.0479,
      "step": 2261
    },
    {
      "epoch": 1.6270454954144937,
      "grad_norm": 5.164236589206694,
      "learning_rate": 2.815115135446995e-06,
      "loss": 0.0265,
      "step": 2262
    },
    {
      "epoch": 1.6277647905053048,
      "grad_norm": 2.7093230626051636,
      "learning_rate": 2.8149514580781105e-06,
      "loss": 0.0158,
      "step": 2263
    },
    {
      "epoch": 1.6284840855961158,
      "grad_norm": 4.363892670433591,
      "learning_rate": 2.814787713052665e-06,
      "loss": 0.1273,
      "step": 2264
    },
    {
      "epoch": 1.6292033806869268,
      "grad_norm": 1.3594032623016834,
      "learning_rate": 2.814623900379084e-06,
      "loss": 0.0279,
      "step": 2265
    },
    {
      "epoch": 1.6299226757777379,
      "grad_norm": 2.884470539701012,
      "learning_rate": 2.814460020065795e-06,
      "loss": 0.0159,
      "step": 2266
    },
    {
      "epoch": 1.630641970868549,
      "grad_norm": 4.018579597539334,
      "learning_rate": 2.8142960721212316e-06,
      "loss": 0.0776,
      "step": 2267
    },
    {
      "epoch": 1.63136126595936,
      "grad_norm": 1.9747994372467361,
      "learning_rate": 2.8141320565538275e-06,
      "loss": 0.0089,
      "step": 2268
    },
    {
      "epoch": 1.632080561050171,
      "grad_norm": 6.1296844227917715,
      "learning_rate": 2.813967973372022e-06,
      "loss": 0.0764,
      "step": 2269
    },
    {
      "epoch": 1.632799856140982,
      "grad_norm": 4.987269399869254,
      "learning_rate": 2.8138038225842577e-06,
      "loss": 0.1961,
      "step": 2270
    },
    {
      "epoch": 1.6335191512317928,
      "grad_norm": 2.2096645390981897,
      "learning_rate": 2.813639604198981e-06,
      "loss": 0.062,
      "step": 2271
    },
    {
      "epoch": 1.6342384463226038,
      "grad_norm": 4.4752041455864,
      "learning_rate": 2.8134753182246398e-06,
      "loss": 0.1245,
      "step": 2272
    },
    {
      "epoch": 1.6349577414134149,
      "grad_norm": 5.2397614527439,
      "learning_rate": 2.813310964669688e-06,
      "loss": 0.1182,
      "step": 2273
    },
    {
      "epoch": 1.635677036504226,
      "grad_norm": 3.070377812357022,
      "learning_rate": 2.813146543542582e-06,
      "loss": 0.072,
      "step": 2274
    },
    {
      "epoch": 1.6363963315950367,
      "grad_norm": 7.470500337100392,
      "learning_rate": 2.8129820548517813e-06,
      "loss": 0.1969,
      "step": 2275
    },
    {
      "epoch": 1.6371156266858478,
      "grad_norm": 5.249217968348018,
      "learning_rate": 2.8128174986057486e-06,
      "loss": 0.1027,
      "step": 2276
    },
    {
      "epoch": 1.6378349217766588,
      "grad_norm": 5.7018996050692605,
      "learning_rate": 2.8126528748129506e-06,
      "loss": 0.0388,
      "step": 2277
    },
    {
      "epoch": 1.6385542168674698,
      "grad_norm": 1.8024741017662713,
      "learning_rate": 2.8124881834818586e-06,
      "loss": 0.0147,
      "step": 2278
    },
    {
      "epoch": 1.6392735119582809,
      "grad_norm": 4.54133779935897,
      "learning_rate": 2.812323424620946e-06,
      "loss": 0.1609,
      "step": 2279
    },
    {
      "epoch": 1.639992807049092,
      "grad_norm": 3.45566076563589,
      "learning_rate": 2.8121585982386883e-06,
      "loss": 0.0885,
      "step": 2280
    },
    {
      "epoch": 1.640712102139903,
      "grad_norm": 5.113097389723223,
      "learning_rate": 2.811993704343568e-06,
      "loss": 0.0934,
      "step": 2281
    },
    {
      "epoch": 1.641431397230714,
      "grad_norm": 2.083635302303951,
      "learning_rate": 2.8118287429440684e-06,
      "loss": 0.0355,
      "step": 2282
    },
    {
      "epoch": 1.642150692321525,
      "grad_norm": 7.046809877712022,
      "learning_rate": 2.811663714048677e-06,
      "loss": 0.2638,
      "step": 2283
    },
    {
      "epoch": 1.642869987412336,
      "grad_norm": 4.662899348259462,
      "learning_rate": 2.811498617665885e-06,
      "loss": 0.0872,
      "step": 2284
    },
    {
      "epoch": 1.643589282503147,
      "grad_norm": 6.789058395980352,
      "learning_rate": 2.811333453804187e-06,
      "loss": 0.1588,
      "step": 2285
    },
    {
      "epoch": 1.6443085775939579,
      "grad_norm": 5.349274469126305,
      "learning_rate": 2.8111682224720807e-06,
      "loss": 0.1744,
      "step": 2286
    },
    {
      "epoch": 1.645027872684769,
      "grad_norm": 1.6239266166418995,
      "learning_rate": 2.8110029236780672e-06,
      "loss": 0.0235,
      "step": 2287
    },
    {
      "epoch": 1.64574716777558,
      "grad_norm": 4.6050908785026285,
      "learning_rate": 2.8108375574306523e-06,
      "loss": 0.0719,
      "step": 2288
    },
    {
      "epoch": 1.646466462866391,
      "grad_norm": 2.215111682739865,
      "learning_rate": 2.8106721237383445e-06,
      "loss": 0.0406,
      "step": 2289
    },
    {
      "epoch": 1.6471857579572018,
      "grad_norm": 3.0358773674490767,
      "learning_rate": 2.810506622609654e-06,
      "loss": 0.117,
      "step": 2290
    },
    {
      "epoch": 1.6479050530480128,
      "grad_norm": 1.409113504727841,
      "learning_rate": 2.8103410540530984e-06,
      "loss": 0.0159,
      "step": 2291
    },
    {
      "epoch": 1.6486243481388239,
      "grad_norm": 3.046594539798307,
      "learning_rate": 2.810175418077195e-06,
      "loss": 0.0257,
      "step": 2292
    },
    {
      "epoch": 1.649343643229635,
      "grad_norm": 5.131974119578082,
      "learning_rate": 2.810009714690466e-06,
      "loss": 0.2435,
      "step": 2293
    },
    {
      "epoch": 1.650062938320446,
      "grad_norm": 6.4958482363193895,
      "learning_rate": 2.809843943901438e-06,
      "loss": 0.2056,
      "step": 2294
    },
    {
      "epoch": 1.650782233411257,
      "grad_norm": 3.564856109982279,
      "learning_rate": 2.8096781057186393e-06,
      "loss": 0.1636,
      "step": 2295
    },
    {
      "epoch": 1.651501528502068,
      "grad_norm": 0.8193251687915041,
      "learning_rate": 2.809512200150603e-06,
      "loss": 0.0083,
      "step": 2296
    },
    {
      "epoch": 1.652220823592879,
      "grad_norm": 1.795011701523758,
      "learning_rate": 2.8093462272058654e-06,
      "loss": 0.0293,
      "step": 2297
    },
    {
      "epoch": 1.65294011868369,
      "grad_norm": 3.3611128272617856,
      "learning_rate": 2.809180186892966e-06,
      "loss": 0.087,
      "step": 2298
    },
    {
      "epoch": 1.653659413774501,
      "grad_norm": 3.2815134778480903,
      "learning_rate": 2.809014079220448e-06,
      "loss": 0.1097,
      "step": 2299
    },
    {
      "epoch": 1.6543787088653121,
      "grad_norm": 4.351753010329881,
      "learning_rate": 2.808847904196857e-06,
      "loss": 0.0863,
      "step": 2300
    },
    {
      "epoch": 1.6550980039561232,
      "grad_norm": 4.096582758129234,
      "learning_rate": 2.8086816618307447e-06,
      "loss": 0.2683,
      "step": 2301
    },
    {
      "epoch": 1.655817299046934,
      "grad_norm": 6.01764618605431,
      "learning_rate": 2.808515352130663e-06,
      "loss": 0.1444,
      "step": 2302
    },
    {
      "epoch": 1.656536594137745,
      "grad_norm": 3.543681806265652,
      "learning_rate": 2.808348975105169e-06,
      "loss": 0.056,
      "step": 2303
    },
    {
      "epoch": 1.657255889228556,
      "grad_norm": 6.168429797203481,
      "learning_rate": 2.8081825307628243e-06,
      "loss": 0.1253,
      "step": 2304
    },
    {
      "epoch": 1.6579751843193669,
      "grad_norm": 2.1512686325368375,
      "learning_rate": 2.8080160191121915e-06,
      "loss": 0.0458,
      "step": 2305
    },
    {
      "epoch": 1.658694479410178,
      "grad_norm": 4.939568498325234,
      "learning_rate": 2.8078494401618386e-06,
      "loss": 0.0663,
      "step": 2306
    },
    {
      "epoch": 1.659413774500989,
      "grad_norm": 4.034034535770205,
      "learning_rate": 2.807682793920336e-06,
      "loss": 0.1563,
      "step": 2307
    },
    {
      "epoch": 1.6601330695918,
      "grad_norm": 2.9118870979833247,
      "learning_rate": 2.807516080396258e-06,
      "loss": 0.012,
      "step": 2308
    },
    {
      "epoch": 1.660852364682611,
      "grad_norm": 0.5425354631322455,
      "learning_rate": 2.8073492995981823e-06,
      "loss": 0.0008,
      "step": 2309
    },
    {
      "epoch": 1.661571659773422,
      "grad_norm": 7.007609927392306,
      "learning_rate": 2.8071824515346904e-06,
      "loss": 0.2073,
      "step": 2310
    },
    {
      "epoch": 1.662290954864233,
      "grad_norm": 4.882534919897073,
      "learning_rate": 2.807015536214367e-06,
      "loss": 0.1861,
      "step": 2311
    },
    {
      "epoch": 1.663010249955044,
      "grad_norm": 5.001806483729345,
      "learning_rate": 2.8068485536457993e-06,
      "loss": 0.1051,
      "step": 2312
    },
    {
      "epoch": 1.6637295450458551,
      "grad_norm": 4.005848121328785,
      "learning_rate": 2.8066815038375797e-06,
      "loss": 0.1536,
      "step": 2313
    },
    {
      "epoch": 1.6644488401366662,
      "grad_norm": 6.363799871490677,
      "learning_rate": 2.8065143867983027e-06,
      "loss": 0.1323,
      "step": 2314
    },
    {
      "epoch": 1.6651681352274772,
      "grad_norm": 4.58530015470701,
      "learning_rate": 2.806347202536567e-06,
      "loss": 0.0796,
      "step": 2315
    },
    {
      "epoch": 1.6658874303182882,
      "grad_norm": 2.350720584461565,
      "learning_rate": 2.8061799510609747e-06,
      "loss": 0.043,
      "step": 2316
    },
    {
      "epoch": 1.666606725409099,
      "grad_norm": 6.480705472133568,
      "learning_rate": 2.806012632380131e-06,
      "loss": 0.1831,
      "step": 2317
    },
    {
      "epoch": 1.66732602049991,
      "grad_norm": 2.2798640447567857,
      "learning_rate": 2.805845246502645e-06,
      "loss": 0.0802,
      "step": 2318
    },
    {
      "epoch": 1.6680453155907211,
      "grad_norm": 5.085479480582992,
      "learning_rate": 2.805677793437128e-06,
      "loss": 0.1542,
      "step": 2319
    },
    {
      "epoch": 1.6687646106815321,
      "grad_norm": 3.018508368209533,
      "learning_rate": 2.8055102731921972e-06,
      "loss": 0.0969,
      "step": 2320
    },
    {
      "epoch": 1.669483905772343,
      "grad_norm": 0.34720257861225523,
      "learning_rate": 2.8053426857764702e-06,
      "loss": 0.0012,
      "step": 2321
    },
    {
      "epoch": 1.670203200863154,
      "grad_norm": 3.736993532295286,
      "learning_rate": 2.8051750311985716e-06,
      "loss": 0.1521,
      "step": 2322
    },
    {
      "epoch": 1.670922495953965,
      "grad_norm": 6.2985218066676225,
      "learning_rate": 2.805007309467126e-06,
      "loss": 0.1019,
      "step": 2323
    },
    {
      "epoch": 1.671641791044776,
      "grad_norm": 2.9203803049116113,
      "learning_rate": 2.804839520590763e-06,
      "loss": 0.0615,
      "step": 2324
    },
    {
      "epoch": 1.672361086135587,
      "grad_norm": 6.425577303323589,
      "learning_rate": 2.8046716645781166e-06,
      "loss": 0.187,
      "step": 2325
    },
    {
      "epoch": 1.6730803812263981,
      "grad_norm": 1.697400462742743,
      "learning_rate": 2.804503741437823e-06,
      "loss": 0.0481,
      "step": 2326
    },
    {
      "epoch": 1.6737996763172092,
      "grad_norm": 2.187582140444186,
      "learning_rate": 2.8043357511785217e-06,
      "loss": 0.0138,
      "step": 2327
    },
    {
      "epoch": 1.6745189714080202,
      "grad_norm": 5.07818170563932,
      "learning_rate": 2.804167693808856e-06,
      "loss": 0.1954,
      "step": 2328
    },
    {
      "epoch": 1.6752382664988312,
      "grad_norm": 5.310186154045098,
      "learning_rate": 2.8039995693374734e-06,
      "loss": 0.1073,
      "step": 2329
    },
    {
      "epoch": 1.6759575615896423,
      "grad_norm": 3.3889899030592825,
      "learning_rate": 2.803831377773024e-06,
      "loss": 0.1176,
      "step": 2330
    },
    {
      "epoch": 1.6766768566804533,
      "grad_norm": 2.49210587642558,
      "learning_rate": 2.803663119124161e-06,
      "loss": 0.0678,
      "step": 2331
    },
    {
      "epoch": 1.677396151771264,
      "grad_norm": 4.344678116227538,
      "learning_rate": 2.8034947933995425e-06,
      "loss": 0.1543,
      "step": 2332
    },
    {
      "epoch": 1.6781154468620751,
      "grad_norm": 2.3696161097073363,
      "learning_rate": 2.8033264006078283e-06,
      "loss": 0.0483,
      "step": 2333
    },
    {
      "epoch": 1.6788347419528862,
      "grad_norm": 3.5993557221384536,
      "learning_rate": 2.803157940757683e-06,
      "loss": 0.0836,
      "step": 2334
    },
    {
      "epoch": 1.6795540370436972,
      "grad_norm": 7.841194297845523,
      "learning_rate": 2.8029894138577735e-06,
      "loss": 0.0462,
      "step": 2335
    },
    {
      "epoch": 1.680273332134508,
      "grad_norm": 4.784775815494183,
      "learning_rate": 2.802820819916772e-06,
      "loss": 0.0759,
      "step": 2336
    },
    {
      "epoch": 1.680992627225319,
      "grad_norm": 4.729570994216389,
      "learning_rate": 2.802652158943352e-06,
      "loss": 0.1444,
      "step": 2337
    },
    {
      "epoch": 1.68171192231613,
      "grad_norm": 4.215344439412829,
      "learning_rate": 2.802483430946192e-06,
      "loss": 0.0302,
      "step": 2338
    },
    {
      "epoch": 1.6824312174069411,
      "grad_norm": 6.690240133242299,
      "learning_rate": 2.8023146359339725e-06,
      "loss": 0.1967,
      "step": 2339
    },
    {
      "epoch": 1.6831505124977522,
      "grad_norm": 3.6220979254508556,
      "learning_rate": 2.8021457739153793e-06,
      "loss": 0.1321,
      "step": 2340
    },
    {
      "epoch": 1.6838698075885632,
      "grad_norm": 5.424669130094212,
      "learning_rate": 2.8019768448991e-06,
      "loss": 0.0929,
      "step": 2341
    },
    {
      "epoch": 1.6845891026793742,
      "grad_norm": 1.7144708037615792,
      "learning_rate": 2.801807848893827e-06,
      "loss": 0.0096,
      "step": 2342
    },
    {
      "epoch": 1.6853083977701853,
      "grad_norm": 4.06530826171108,
      "learning_rate": 2.801638785908254e-06,
      "loss": 0.0444,
      "step": 2343
    },
    {
      "epoch": 1.6860276928609963,
      "grad_norm": 4.721120526321798,
      "learning_rate": 2.801469655951081e-06,
      "loss": 0.0998,
      "step": 2344
    },
    {
      "epoch": 1.6867469879518073,
      "grad_norm": 5.8860746771055386,
      "learning_rate": 2.8013004590310097e-06,
      "loss": 0.078,
      "step": 2345
    },
    {
      "epoch": 1.6874662830426184,
      "grad_norm": 6.064701007299643,
      "learning_rate": 2.801131195156745e-06,
      "loss": 0.2046,
      "step": 2346
    },
    {
      "epoch": 1.6881855781334294,
      "grad_norm": 6.173058455453307,
      "learning_rate": 2.8009618643369964e-06,
      "loss": 0.3731,
      "step": 2347
    },
    {
      "epoch": 1.6889048732242402,
      "grad_norm": 1.9379741238649226,
      "learning_rate": 2.8007924665804767e-06,
      "loss": 0.0068,
      "step": 2348
    },
    {
      "epoch": 1.6896241683150512,
      "grad_norm": 3.7071785250223943,
      "learning_rate": 2.8006230018959e-06,
      "loss": 0.1862,
      "step": 2349
    },
    {
      "epoch": 1.6903434634058623,
      "grad_norm": 1.7909371751100964,
      "learning_rate": 2.8004534702919876e-06,
      "loss": 0.0073,
      "step": 2350
    },
    {
      "epoch": 1.6910627584966733,
      "grad_norm": 2.372855211559006,
      "learning_rate": 2.8002838717774608e-06,
      "loss": 0.0636,
      "step": 2351
    },
    {
      "epoch": 1.6917820535874841,
      "grad_norm": 7.358328824425426,
      "learning_rate": 2.8001142063610463e-06,
      "loss": 0.4972,
      "step": 2352
    },
    {
      "epoch": 1.6925013486782952,
      "grad_norm": 6.74744800508788,
      "learning_rate": 2.7999444740514736e-06,
      "loss": 0.2217,
      "step": 2353
    },
    {
      "epoch": 1.6932206437691062,
      "grad_norm": 7.429661107300888,
      "learning_rate": 2.799774674857476e-06,
      "loss": 0.3916,
      "step": 2354
    },
    {
      "epoch": 1.6939399388599172,
      "grad_norm": 2.3350017624151396,
      "learning_rate": 2.7996048087877893e-06,
      "loss": 0.0829,
      "step": 2355
    },
    {
      "epoch": 1.6946592339507283,
      "grad_norm": 3.3803922707648595,
      "learning_rate": 2.799434875851154e-06,
      "loss": 0.1471,
      "step": 2356
    },
    {
      "epoch": 1.6953785290415393,
      "grad_norm": 3.7978222134421586,
      "learning_rate": 2.7992648760563134e-06,
      "loss": 0.053,
      "step": 2357
    },
    {
      "epoch": 1.6960978241323503,
      "grad_norm": 4.815996112642946,
      "learning_rate": 2.7990948094120143e-06,
      "loss": 0.1971,
      "step": 2358
    },
    {
      "epoch": 1.6968171192231614,
      "grad_norm": 2.439127198144136,
      "learning_rate": 2.7989246759270062e-06,
      "loss": 0.0577,
      "step": 2359
    },
    {
      "epoch": 1.6975364143139724,
      "grad_norm": 2.8930296653214094,
      "learning_rate": 2.7987544756100436e-06,
      "loss": 0.1225,
      "step": 2360
    },
    {
      "epoch": 1.6982557094047834,
      "grad_norm": 5.017556773666212,
      "learning_rate": 2.7985842084698836e-06,
      "loss": 0.0505,
      "step": 2361
    },
    {
      "epoch": 1.6989750044955945,
      "grad_norm": 4.7436929412514885,
      "learning_rate": 2.7984138745152865e-06,
      "loss": 0.0374,
      "step": 2362
    },
    {
      "epoch": 1.6996942995864053,
      "grad_norm": 4.041335827226105,
      "learning_rate": 2.7982434737550157e-06,
      "loss": 0.0388,
      "step": 2363
    },
    {
      "epoch": 1.7004135946772163,
      "grad_norm": 2.9873480818832894,
      "learning_rate": 2.7980730061978394e-06,
      "loss": 0.143,
      "step": 2364
    },
    {
      "epoch": 1.7011328897680273,
      "grad_norm": 1.9307763350469296,
      "learning_rate": 2.797902471852528e-06,
      "loss": 0.0321,
      "step": 2365
    },
    {
      "epoch": 1.7018521848588384,
      "grad_norm": 6.812876028175807,
      "learning_rate": 2.7977318707278564e-06,
      "loss": 0.1261,
      "step": 2366
    },
    {
      "epoch": 1.7025714799496492,
      "grad_norm": 5.680763272515375,
      "learning_rate": 2.7975612028326022e-06,
      "loss": 0.1719,
      "step": 2367
    },
    {
      "epoch": 1.7032907750404602,
      "grad_norm": 3.264300707776835,
      "learning_rate": 2.797390468175546e-06,
      "loss": 0.111,
      "step": 2368
    },
    {
      "epoch": 1.7040100701312713,
      "grad_norm": 1.8210095092883605,
      "learning_rate": 2.7972196667654726e-06,
      "loss": 0.0282,
      "step": 2369
    },
    {
      "epoch": 1.7047293652220823,
      "grad_norm": 4.463694700132819,
      "learning_rate": 2.7970487986111705e-06,
      "loss": 0.0541,
      "step": 2370
    },
    {
      "epoch": 1.7054486603128933,
      "grad_norm": 1.8725822761990196,
      "learning_rate": 2.7968778637214304e-06,
      "loss": 0.0048,
      "step": 2371
    },
    {
      "epoch": 1.7061679554037044,
      "grad_norm": 1.9843340567547916,
      "learning_rate": 2.796706862105048e-06,
      "loss": 0.0092,
      "step": 2372
    },
    {
      "epoch": 1.7068872504945154,
      "grad_norm": 3.875985553398004,
      "learning_rate": 2.7965357937708207e-06,
      "loss": 0.1433,
      "step": 2373
    },
    {
      "epoch": 1.7076065455853264,
      "grad_norm": 3.7845446624688823,
      "learning_rate": 2.7963646587275513e-06,
      "loss": 0.1838,
      "step": 2374
    },
    {
      "epoch": 1.7083258406761375,
      "grad_norm": 2.8864845486353286,
      "learning_rate": 2.7961934569840443e-06,
      "loss": 0.0611,
      "step": 2375
    },
    {
      "epoch": 1.7090451357669485,
      "grad_norm": 3.0131955885034047,
      "learning_rate": 2.7960221885491086e-06,
      "loss": 0.0423,
      "step": 2376
    },
    {
      "epoch": 1.7097644308577595,
      "grad_norm": 4.7533239840069195,
      "learning_rate": 2.795850853431556e-06,
      "loss": 0.1347,
      "step": 2377
    },
    {
      "epoch": 1.7104837259485706,
      "grad_norm": 4.092464808013002,
      "learning_rate": 2.795679451640203e-06,
      "loss": 0.2292,
      "step": 2378
    },
    {
      "epoch": 1.7112030210393814,
      "grad_norm": 1.995013038112466,
      "learning_rate": 2.7955079831838667e-06,
      "loss": 0.0116,
      "step": 2379
    },
    {
      "epoch": 1.7119223161301924,
      "grad_norm": 6.157404191331068,
      "learning_rate": 2.795336448071371e-06,
      "loss": 0.1794,
      "step": 2380
    },
    {
      "epoch": 1.7126416112210034,
      "grad_norm": 5.011503741469248,
      "learning_rate": 2.795164846311541e-06,
      "loss": 0.0262,
      "step": 2381
    },
    {
      "epoch": 1.7133609063118143,
      "grad_norm": 3.534925477162323,
      "learning_rate": 2.7949931779132066e-06,
      "loss": 0.1281,
      "step": 2382
    },
    {
      "epoch": 1.7140802014026253,
      "grad_norm": 7.017899273204855,
      "learning_rate": 2.794821442885199e-06,
      "loss": 0.334,
      "step": 2383
    },
    {
      "epoch": 1.7147994964934363,
      "grad_norm": 6.3196514031515285,
      "learning_rate": 2.7946496412363555e-06,
      "loss": 0.0319,
      "step": 2384
    },
    {
      "epoch": 1.7155187915842474,
      "grad_norm": 1.8931560590820344,
      "learning_rate": 2.794477772975516e-06,
      "loss": 0.0558,
      "step": 2385
    },
    {
      "epoch": 1.7162380866750584,
      "grad_norm": 4.286760853514821,
      "learning_rate": 2.7943058381115215e-06,
      "loss": 0.109,
      "step": 2386
    },
    {
      "epoch": 1.7169573817658694,
      "grad_norm": 3.636483978192362,
      "learning_rate": 2.7941338366532204e-06,
      "loss": 0.0916,
      "step": 2387
    },
    {
      "epoch": 1.7176766768566805,
      "grad_norm": 5.498435541224947,
      "learning_rate": 2.7939617686094615e-06,
      "loss": 0.0458,
      "step": 2388
    },
    {
      "epoch": 1.7183959719474915,
      "grad_norm": 2.417437411202421,
      "learning_rate": 2.793789633989098e-06,
      "loss": 0.0068,
      "step": 2389
    },
    {
      "epoch": 1.7191152670383025,
      "grad_norm": 5.543263452435766,
      "learning_rate": 2.793617432800986e-06,
      "loss": 0.1707,
      "step": 2390
    },
    {
      "epoch": 1.7198345621291136,
      "grad_norm": 4.396643626007494,
      "learning_rate": 2.7934451650539874e-06,
      "loss": 0.0112,
      "step": 2391
    },
    {
      "epoch": 1.7205538572199246,
      "grad_norm": 4.964203598794387,
      "learning_rate": 2.793272830756964e-06,
      "loss": 0.1877,
      "step": 2392
    },
    {
      "epoch": 1.7212731523107356,
      "grad_norm": 5.04100974186584,
      "learning_rate": 2.7931004299187827e-06,
      "loss": 0.0753,
      "step": 2393
    },
    {
      "epoch": 1.7219924474015464,
      "grad_norm": 7.147974844653717,
      "learning_rate": 2.792927962548315e-06,
      "loss": 0.3401,
      "step": 2394
    },
    {
      "epoch": 1.7227117424923575,
      "grad_norm": 5.050882737588076,
      "learning_rate": 2.792755428654434e-06,
      "loss": 0.1955,
      "step": 2395
    },
    {
      "epoch": 1.7234310375831685,
      "grad_norm": 3.722476540875743,
      "learning_rate": 2.792582828246016e-06,
      "loss": 0.0071,
      "step": 2396
    },
    {
      "epoch": 1.7241503326739795,
      "grad_norm": 2.978173772916557,
      "learning_rate": 2.792410161331943e-06,
      "loss": 0.0212,
      "step": 2397
    },
    {
      "epoch": 1.7248696277647904,
      "grad_norm": 4.731260403746607,
      "learning_rate": 2.792237427921099e-06,
      "loss": 0.1153,
      "step": 2398
    },
    {
      "epoch": 1.7255889228556014,
      "grad_norm": 7.267363709414449,
      "learning_rate": 2.79206462802237e-06,
      "loss": 0.2066,
      "step": 2399
    },
    {
      "epoch": 1.7263082179464124,
      "grad_norm": 4.306413977625975,
      "learning_rate": 2.791891761644648e-06,
      "loss": 0.1158,
      "step": 2400
    },
    {
      "epoch": 1.7270275130372235,
      "grad_norm": 2.7527140148638307,
      "learning_rate": 2.7917188287968268e-06,
      "loss": 0.0775,
      "step": 2401
    },
    {
      "epoch": 1.7277468081280345,
      "grad_norm": 5.892912500185107,
      "learning_rate": 2.791545829487805e-06,
      "loss": 0.16,
      "step": 2402
    },
    {
      "epoch": 1.7284661032188455,
      "grad_norm": 8.155263083235745,
      "learning_rate": 2.7913727637264824e-06,
      "loss": 0.1022,
      "step": 2403
    },
    {
      "epoch": 1.7291853983096566,
      "grad_norm": 2.999315306442573,
      "learning_rate": 2.7911996315217643e-06,
      "loss": 0.0848,
      "step": 2404
    },
    {
      "epoch": 1.7299046934004676,
      "grad_norm": 3.5327156826921424,
      "learning_rate": 2.7910264328825586e-06,
      "loss": 0.1383,
      "step": 2405
    },
    {
      "epoch": 1.7306239884912786,
      "grad_norm": 5.699458381892293,
      "learning_rate": 2.7908531678177767e-06,
      "loss": 0.0878,
      "step": 2406
    },
    {
      "epoch": 1.7313432835820897,
      "grad_norm": 1.7807204861908321,
      "learning_rate": 2.790679836336333e-06,
      "loss": 0.006,
      "step": 2407
    },
    {
      "epoch": 1.7320625786729007,
      "grad_norm": 2.197058617885282,
      "learning_rate": 2.7905064384471462e-06,
      "loss": 0.0478,
      "step": 2408
    },
    {
      "epoch": 1.7327818737637115,
      "grad_norm": 3.579967652177571,
      "learning_rate": 2.7903329741591378e-06,
      "loss": 0.0559,
      "step": 2409
    },
    {
      "epoch": 1.7335011688545225,
      "grad_norm": 2.665072432924307,
      "learning_rate": 2.7901594434812324e-06,
      "loss": 0.0534,
      "step": 2410
    },
    {
      "epoch": 1.7342204639453336,
      "grad_norm": 1.8579175465103006,
      "learning_rate": 2.7899858464223595e-06,
      "loss": 0.0495,
      "step": 2411
    },
    {
      "epoch": 1.7349397590361446,
      "grad_norm": 4.540021825871725,
      "learning_rate": 2.78981218299145e-06,
      "loss": 0.0163,
      "step": 2412
    },
    {
      "epoch": 1.7356590541269554,
      "grad_norm": 2.9008181430926667,
      "learning_rate": 2.7896384531974396e-06,
      "loss": 0.055,
      "step": 2413
    },
    {
      "epoch": 1.7363783492177665,
      "grad_norm": 4.175275968242346,
      "learning_rate": 2.7894646570492666e-06,
      "loss": 0.1024,
      "step": 2414
    },
    {
      "epoch": 1.7370976443085775,
      "grad_norm": 6.8995704900251065,
      "learning_rate": 2.7892907945558734e-06,
      "loss": 0.1833,
      "step": 2415
    },
    {
      "epoch": 1.7378169393993885,
      "grad_norm": 2.6365483682168978,
      "learning_rate": 2.7891168657262056e-06,
      "loss": 0.0117,
      "step": 2416
    },
    {
      "epoch": 1.7385362344901996,
      "grad_norm": 4.811859968335681,
      "learning_rate": 2.7889428705692125e-06,
      "loss": 0.1356,
      "step": 2417
    },
    {
      "epoch": 1.7392555295810106,
      "grad_norm": 8.103032461889205,
      "learning_rate": 2.7887688090938455e-06,
      "loss": 0.2083,
      "step": 2418
    },
    {
      "epoch": 1.7399748246718216,
      "grad_norm": 5.687230353382354,
      "learning_rate": 2.788594681309061e-06,
      "loss": 0.203,
      "step": 2419
    },
    {
      "epoch": 1.7406941197626327,
      "grad_norm": 6.527378717586143,
      "learning_rate": 2.788420487223818e-06,
      "loss": 0.2212,
      "step": 2420
    },
    {
      "epoch": 1.7414134148534437,
      "grad_norm": 3.5944677575595545,
      "learning_rate": 2.7882462268470796e-06,
      "loss": 0.0949,
      "step": 2421
    },
    {
      "epoch": 1.7421327099442547,
      "grad_norm": 3.900858930748474,
      "learning_rate": 2.788071900187811e-06,
      "loss": 0.1667,
      "step": 2422
    },
    {
      "epoch": 1.7428520050350658,
      "grad_norm": 4.584469738180817,
      "learning_rate": 2.7878975072549823e-06,
      "loss": 0.1476,
      "step": 2423
    },
    {
      "epoch": 1.7435713001258768,
      "grad_norm": 2.523693751667623,
      "learning_rate": 2.787723048057566e-06,
      "loss": 0.0772,
      "step": 2424
    },
    {
      "epoch": 1.7442905952166876,
      "grad_norm": 6.124401260014997,
      "learning_rate": 2.7875485226045385e-06,
      "loss": 0.2145,
      "step": 2425
    },
    {
      "epoch": 1.7450098903074986,
      "grad_norm": 4.7632250768252575,
      "learning_rate": 2.787373930904879e-06,
      "loss": 0.2092,
      "step": 2426
    },
    {
      "epoch": 1.7457291853983097,
      "grad_norm": 3.851149311424704,
      "learning_rate": 2.7871992729675705e-06,
      "loss": 0.1175,
      "step": 2427
    },
    {
      "epoch": 1.7464484804891207,
      "grad_norm": 3.1676414628856917,
      "learning_rate": 2.7870245488016005e-06,
      "loss": 0.0257,
      "step": 2428
    },
    {
      "epoch": 1.7471677755799315,
      "grad_norm": 2.5536292631549937,
      "learning_rate": 2.786849758415958e-06,
      "loss": 0.0149,
      "step": 2429
    },
    {
      "epoch": 1.7478870706707426,
      "grad_norm": 4.859296970292306,
      "learning_rate": 2.786674901819636e-06,
      "loss": 0.2743,
      "step": 2430
    },
    {
      "epoch": 1.7486063657615536,
      "grad_norm": 6.884180404294966,
      "learning_rate": 2.786499979021632e-06,
      "loss": 0.0488,
      "step": 2431
    },
    {
      "epoch": 1.7493256608523646,
      "grad_norm": 5.287736487607871,
      "learning_rate": 2.7863249900309456e-06,
      "loss": 0.1154,
      "step": 2432
    },
    {
      "epoch": 1.7500449559431757,
      "grad_norm": 6.156031457626568,
      "learning_rate": 2.78614993485658e-06,
      "loss": 0.256,
      "step": 2433
    },
    {
      "epoch": 1.7507642510339867,
      "grad_norm": 2.9101570063235265,
      "learning_rate": 2.785974813507543e-06,
      "loss": 0.0436,
      "step": 2434
    },
    {
      "epoch": 1.7514835461247977,
      "grad_norm": 1.0490987630399833,
      "learning_rate": 2.7857996259928445e-06,
      "loss": 0.006,
      "step": 2435
    },
    {
      "epoch": 1.7522028412156088,
      "grad_norm": 1.6146349516856136,
      "learning_rate": 2.7856243723214972e-06,
      "loss": 0.0211,
      "step": 2436
    },
    {
      "epoch": 1.7529221363064198,
      "grad_norm": 2.5100020503515794,
      "learning_rate": 2.78544905250252e-06,
      "loss": 0.0279,
      "step": 2437
    },
    {
      "epoch": 1.7536414313972308,
      "grad_norm": 0.5825906204160238,
      "learning_rate": 2.7852736665449325e-06,
      "loss": 0.0029,
      "step": 2438
    },
    {
      "epoch": 1.7543607264880419,
      "grad_norm": 2.820026517255655,
      "learning_rate": 2.7850982144577585e-06,
      "loss": 0.0271,
      "step": 2439
    },
    {
      "epoch": 1.7550800215788527,
      "grad_norm": 1.776986012527359,
      "learning_rate": 2.784922696250025e-06,
      "loss": 0.004,
      "step": 2440
    },
    {
      "epoch": 1.7557993166696637,
      "grad_norm": 1.6706054156054357,
      "learning_rate": 2.7847471119307635e-06,
      "loss": 0.0256,
      "step": 2441
    },
    {
      "epoch": 1.7565186117604747,
      "grad_norm": 4.0572610573536885,
      "learning_rate": 2.784571461509008e-06,
      "loss": 0.0942,
      "step": 2442
    },
    {
      "epoch": 1.7572379068512858,
      "grad_norm": 3.681397781999924,
      "learning_rate": 2.7843957449937956e-06,
      "loss": 0.0812,
      "step": 2443
    },
    {
      "epoch": 1.7579572019420966,
      "grad_norm": 2.813872836830971,
      "learning_rate": 2.7842199623941674e-06,
      "loss": 0.1138,
      "step": 2444
    },
    {
      "epoch": 1.7586764970329076,
      "grad_norm": 4.527030593476702,
      "learning_rate": 2.7840441137191676e-06,
      "loss": 0.13,
      "step": 2445
    },
    {
      "epoch": 1.7593957921237187,
      "grad_norm": 5.193605920645928,
      "learning_rate": 2.7838681989778442e-06,
      "loss": 0.0815,
      "step": 2446
    },
    {
      "epoch": 1.7601150872145297,
      "grad_norm": 0.8159936995068852,
      "learning_rate": 2.7836922181792483e-06,
      "loss": 0.0028,
      "step": 2447
    },
    {
      "epoch": 1.7608343823053407,
      "grad_norm": 5.553293228334078,
      "learning_rate": 2.7835161713324346e-06,
      "loss": 0.155,
      "step": 2448
    },
    {
      "epoch": 1.7615536773961518,
      "grad_norm": 1.8438614964271234,
      "learning_rate": 2.7833400584464604e-06,
      "loss": 0.0408,
      "step": 2449
    },
    {
      "epoch": 1.7622729724869628,
      "grad_norm": 5.0434768747216046,
      "learning_rate": 2.7831638795303873e-06,
      "loss": 0.1741,
      "step": 2450
    },
    {
      "epoch": 1.7629922675777738,
      "grad_norm": 3.008907579746472,
      "learning_rate": 2.78298763459328e-06,
      "loss": 0.1955,
      "step": 2451
    },
    {
      "epoch": 1.7637115626685849,
      "grad_norm": 3.9620030366647105,
      "learning_rate": 2.782811323644207e-06,
      "loss": 0.0852,
      "step": 2452
    },
    {
      "epoch": 1.764430857759396,
      "grad_norm": 3.316202661520531,
      "learning_rate": 2.7826349466922396e-06,
      "loss": 0.0862,
      "step": 2453
    },
    {
      "epoch": 1.765150152850207,
      "grad_norm": 2.2031411658279576,
      "learning_rate": 2.782458503746452e-06,
      "loss": 0.0337,
      "step": 2454
    },
    {
      "epoch": 1.765869447941018,
      "grad_norm": 3.2166319287286638,
      "learning_rate": 2.7822819948159237e-06,
      "loss": 0.0957,
      "step": 2455
    },
    {
      "epoch": 1.7665887430318288,
      "grad_norm": 3.240275325137214,
      "learning_rate": 2.7821054199097357e-06,
      "loss": 0.0836,
      "step": 2456
    },
    {
      "epoch": 1.7673080381226398,
      "grad_norm": 5.202324080627204,
      "learning_rate": 2.781928779036973e-06,
      "loss": 0.1207,
      "step": 2457
    },
    {
      "epoch": 1.7680273332134508,
      "grad_norm": 7.2558041039963115,
      "learning_rate": 2.781752072206724e-06,
      "loss": 0.1287,
      "step": 2458
    },
    {
      "epoch": 1.7687466283042617,
      "grad_norm": 5.53981910154158,
      "learning_rate": 2.781575299428081e-06,
      "loss": 0.1373,
      "step": 2459
    },
    {
      "epoch": 1.7694659233950727,
      "grad_norm": 4.104216394763029,
      "learning_rate": 2.7813984607101394e-06,
      "loss": 0.1158,
      "step": 2460
    },
    {
      "epoch": 1.7701852184858837,
      "grad_norm": 1.4470698433642928,
      "learning_rate": 2.7812215560619977e-06,
      "loss": 0.0294,
      "step": 2461
    },
    {
      "epoch": 1.7709045135766948,
      "grad_norm": 5.669550095457154,
      "learning_rate": 2.781044585492757e-06,
      "loss": 0.2916,
      "step": 2462
    },
    {
      "epoch": 1.7716238086675058,
      "grad_norm": 3.1666442293662467,
      "learning_rate": 2.7808675490115246e-06,
      "loss": 0.0894,
      "step": 2463
    },
    {
      "epoch": 1.7723431037583168,
      "grad_norm": 2.712345202276612,
      "learning_rate": 2.780690446627408e-06,
      "loss": 0.0557,
      "step": 2464
    },
    {
      "epoch": 1.7730623988491279,
      "grad_norm": 5.505570748437723,
      "learning_rate": 2.7805132783495192e-06,
      "loss": 0.2554,
      "step": 2465
    },
    {
      "epoch": 1.773781693939939,
      "grad_norm": 5.542460629541906,
      "learning_rate": 2.7803360441869747e-06,
      "loss": 0.0789,
      "step": 2466
    },
    {
      "epoch": 1.77450098903075,
      "grad_norm": 4.138194696914727,
      "learning_rate": 2.780158744148893e-06,
      "loss": 0.1718,
      "step": 2467
    },
    {
      "epoch": 1.775220284121561,
      "grad_norm": 3.203864931136845,
      "learning_rate": 2.779981378244397e-06,
      "loss": 0.1018,
      "step": 2468
    },
    {
      "epoch": 1.775939579212372,
      "grad_norm": 4.099174749866319,
      "learning_rate": 2.7798039464826124e-06,
      "loss": 0.0531,
      "step": 2469
    },
    {
      "epoch": 1.776658874303183,
      "grad_norm": 6.047535212216035,
      "learning_rate": 2.7796264488726674e-06,
      "loss": 0.2787,
      "step": 2470
    },
    {
      "epoch": 1.7773781693939938,
      "grad_norm": 5.314497018473563,
      "learning_rate": 2.779448885423696e-06,
      "loss": 0.0702,
      "step": 2471
    },
    {
      "epoch": 1.7780974644848049,
      "grad_norm": 2.278598676722101,
      "learning_rate": 2.779271256144833e-06,
      "loss": 0.0968,
      "step": 2472
    },
    {
      "epoch": 1.778816759575616,
      "grad_norm": 0.05148420749332451,
      "learning_rate": 2.779093561045218e-06,
      "loss": 0.0003,
      "step": 2473
    },
    {
      "epoch": 1.779536054666427,
      "grad_norm": 6.310569639532884,
      "learning_rate": 2.778915800133994e-06,
      "loss": 0.4003,
      "step": 2474
    },
    {
      "epoch": 1.7802553497572378,
      "grad_norm": 6.283635654690987,
      "learning_rate": 2.7787379734203072e-06,
      "loss": 0.0724,
      "step": 2475
    },
    {
      "epoch": 1.7809746448480488,
      "grad_norm": 3.9531614800696007,
      "learning_rate": 2.778560080913307e-06,
      "loss": 0.1231,
      "step": 2476
    },
    {
      "epoch": 1.7816939399388598,
      "grad_norm": 6.43125342617877,
      "learning_rate": 2.7783821226221462e-06,
      "loss": 0.0852,
      "step": 2477
    },
    {
      "epoch": 1.7824132350296709,
      "grad_norm": 7.028478314159626,
      "learning_rate": 2.778204098555981e-06,
      "loss": 0.3916,
      "step": 2478
    },
    {
      "epoch": 1.783132530120482,
      "grad_norm": 4.739546740935013,
      "learning_rate": 2.778026008723971e-06,
      "loss": 0.0915,
      "step": 2479
    },
    {
      "epoch": 1.783851825211293,
      "grad_norm": 3.5839932374655885,
      "learning_rate": 2.7778478531352794e-06,
      "loss": 0.1126,
      "step": 2480
    },
    {
      "epoch": 1.784571120302104,
      "grad_norm": 2.3778317157586706,
      "learning_rate": 2.777669631799073e-06,
      "loss": 0.0692,
      "step": 2481
    },
    {
      "epoch": 1.785290415392915,
      "grad_norm": 3.334496721064537,
      "learning_rate": 2.7774913447245202e-06,
      "loss": 0.1262,
      "step": 2482
    },
    {
      "epoch": 1.786009710483726,
      "grad_norm": 3.2463333086306787,
      "learning_rate": 2.777312991920796e-06,
      "loss": 0.046,
      "step": 2483
    },
    {
      "epoch": 1.786729005574537,
      "grad_norm": 3.505916217151454,
      "learning_rate": 2.777134573397076e-06,
      "loss": 0.082,
      "step": 2484
    },
    {
      "epoch": 1.787448300665348,
      "grad_norm": 6.085618892510015,
      "learning_rate": 2.77695608916254e-06,
      "loss": 0.2499,
      "step": 2485
    },
    {
      "epoch": 1.788167595756159,
      "grad_norm": 3.48603526679261,
      "learning_rate": 2.7767775392263714e-06,
      "loss": 0.0787,
      "step": 2486
    },
    {
      "epoch": 1.78888689084697,
      "grad_norm": 2.1435328531321587,
      "learning_rate": 2.7765989235977573e-06,
      "loss": 0.0138,
      "step": 2487
    },
    {
      "epoch": 1.789606185937781,
      "grad_norm": 5.289632312079522,
      "learning_rate": 2.7764202422858873e-06,
      "loss": 0.2931,
      "step": 2488
    },
    {
      "epoch": 1.790325481028592,
      "grad_norm": 5.343247603244428,
      "learning_rate": 2.776241495299956e-06,
      "loss": 0.146,
      "step": 2489
    },
    {
      "epoch": 1.7910447761194028,
      "grad_norm": 4.656668544117325,
      "learning_rate": 2.7760626826491588e-06,
      "loss": 0.1656,
      "step": 2490
    },
    {
      "epoch": 1.7917640712102139,
      "grad_norm": 6.237523832262931,
      "learning_rate": 2.7758838043426965e-06,
      "loss": 0.0507,
      "step": 2491
    },
    {
      "epoch": 1.792483366301025,
      "grad_norm": 4.491730845376762,
      "learning_rate": 2.7757048603897726e-06,
      "loss": 0.1342,
      "step": 2492
    },
    {
      "epoch": 1.793202661391836,
      "grad_norm": 6.478613923551439,
      "learning_rate": 2.7755258507995943e-06,
      "loss": 0.1437,
      "step": 2493
    },
    {
      "epoch": 1.793921956482647,
      "grad_norm": 4.411105366060626,
      "learning_rate": 2.7753467755813724e-06,
      "loss": 0.0242,
      "step": 2494
    },
    {
      "epoch": 1.794641251573458,
      "grad_norm": 0.8858500767843286,
      "learning_rate": 2.7751676347443194e-06,
      "loss": 0.0043,
      "step": 2495
    },
    {
      "epoch": 1.795360546664269,
      "grad_norm": 2.6120016270015123,
      "learning_rate": 2.7749884282976534e-06,
      "loss": 0.0141,
      "step": 2496
    },
    {
      "epoch": 1.79607984175508,
      "grad_norm": 1.1829032619641606,
      "learning_rate": 2.7748091562505946e-06,
      "loss": 0.0025,
      "step": 2497
    },
    {
      "epoch": 1.796799136845891,
      "grad_norm": 5.9738657468641705,
      "learning_rate": 2.7746298186123667e-06,
      "loss": 0.1216,
      "step": 2498
    },
    {
      "epoch": 1.7975184319367021,
      "grad_norm": 3.551417411720854,
      "learning_rate": 2.774450415392197e-06,
      "loss": 0.1287,
      "step": 2499
    },
    {
      "epoch": 1.7982377270275132,
      "grad_norm": 5.570595282951437,
      "learning_rate": 2.774270946599317e-06,
      "loss": 0.1204,
      "step": 2500
    },
    {
      "epoch": 1.7989570221183242,
      "grad_norm": 6.634960975916419,
      "learning_rate": 2.7740914122429596e-06,
      "loss": 0.1355,
      "step": 2501
    },
    {
      "epoch": 1.799676317209135,
      "grad_norm": 1.370184468491438,
      "learning_rate": 2.773911812332362e-06,
      "loss": 0.0271,
      "step": 2502
    },
    {
      "epoch": 1.800395612299946,
      "grad_norm": 3.719098128948436,
      "learning_rate": 2.7737321468767657e-06,
      "loss": 0.026,
      "step": 2503
    },
    {
      "epoch": 1.801114907390757,
      "grad_norm": 4.413730591800512,
      "learning_rate": 2.7735524158854136e-06,
      "loss": 0.1071,
      "step": 2504
    },
    {
      "epoch": 1.8018342024815681,
      "grad_norm": 4.423628648379886,
      "learning_rate": 2.773372619367555e-06,
      "loss": 0.1083,
      "step": 2505
    },
    {
      "epoch": 1.802553497572379,
      "grad_norm": 6.1161384519081965,
      "learning_rate": 2.7731927573324395e-06,
      "loss": 0.3374,
      "step": 2506
    },
    {
      "epoch": 1.80327279266319,
      "grad_norm": 3.3458772925524523,
      "learning_rate": 2.773012829789322e-06,
      "loss": 0.1452,
      "step": 2507
    },
    {
      "epoch": 1.803992087754001,
      "grad_norm": 3.042766699375507,
      "learning_rate": 2.7728328367474588e-06,
      "loss": 0.0402,
      "step": 2508
    },
    {
      "epoch": 1.804711382844812,
      "grad_norm": 4.947010356396303,
      "learning_rate": 2.7726527782161126e-06,
      "loss": 0.099,
      "step": 2509
    },
    {
      "epoch": 1.805430677935623,
      "grad_norm": 5.7295888327825155,
      "learning_rate": 2.7724726542045464e-06,
      "loss": 0.1001,
      "step": 2510
    },
    {
      "epoch": 1.806149973026434,
      "grad_norm": 4.448403931657433,
      "learning_rate": 2.772292464722029e-06,
      "loss": 0.1106,
      "step": 2511
    },
    {
      "epoch": 1.8068692681172451,
      "grad_norm": 3.1343087476722085,
      "learning_rate": 2.77211220977783e-06,
      "loss": 0.0629,
      "step": 2512
    },
    {
      "epoch": 1.8075885632080562,
      "grad_norm": 4.799081920086391,
      "learning_rate": 2.7719318893812254e-06,
      "loss": 0.0605,
      "step": 2513
    },
    {
      "epoch": 1.8083078582988672,
      "grad_norm": 4.8838820525245294,
      "learning_rate": 2.7717515035414923e-06,
      "loss": 0.0532,
      "step": 2514
    },
    {
      "epoch": 1.8090271533896782,
      "grad_norm": 1.9114660253876055,
      "learning_rate": 2.7715710522679113e-06,
      "loss": 0.0454,
      "step": 2515
    },
    {
      "epoch": 1.8097464484804893,
      "grad_norm": 3.678612753068636,
      "learning_rate": 2.7713905355697676e-06,
      "loss": 0.108,
      "step": 2516
    },
    {
      "epoch": 1.8104657435713,
      "grad_norm": 3.132743077380941,
      "learning_rate": 2.7712099534563494e-06,
      "loss": 0.0211,
      "step": 2517
    },
    {
      "epoch": 1.8111850386621111,
      "grad_norm": 7.180254446978263,
      "learning_rate": 2.7710293059369473e-06,
      "loss": 0.0688,
      "step": 2518
    },
    {
      "epoch": 1.8119043337529221,
      "grad_norm": 3.3402755016222083,
      "learning_rate": 2.7708485930208563e-06,
      "loss": 0.0518,
      "step": 2519
    },
    {
      "epoch": 1.8126236288437332,
      "grad_norm": 5.8339881130520626,
      "learning_rate": 2.770667814717374e-06,
      "loss": 0.1752,
      "step": 2520
    },
    {
      "epoch": 1.813342923934544,
      "grad_norm": 3.3025676992552544,
      "learning_rate": 2.7704869710358027e-06,
      "loss": 0.0311,
      "step": 2521
    },
    {
      "epoch": 1.814062219025355,
      "grad_norm": 0.6364832336183773,
      "learning_rate": 2.7703060619854456e-06,
      "loss": 0.0015,
      "step": 2522
    },
    {
      "epoch": 1.814781514116166,
      "grad_norm": 6.372236187824131,
      "learning_rate": 2.7701250875756123e-06,
      "loss": 0.4278,
      "step": 2523
    },
    {
      "epoch": 1.815500809206977,
      "grad_norm": 3.03470470010061,
      "learning_rate": 2.769944047815613e-06,
      "loss": 0.098,
      "step": 2524
    },
    {
      "epoch": 1.8162201042977881,
      "grad_norm": 2.490473441477687,
      "learning_rate": 2.769762942714764e-06,
      "loss": 0.0312,
      "step": 2525
    },
    {
      "epoch": 1.8169393993885992,
      "grad_norm": 6.379695479011435,
      "learning_rate": 2.769581772282382e-06,
      "loss": 0.2005,
      "step": 2526
    },
    {
      "epoch": 1.8176586944794102,
      "grad_norm": 0.5678288856838097,
      "learning_rate": 2.769400536527789e-06,
      "loss": 0.0043,
      "step": 2527
    },
    {
      "epoch": 1.8183779895702212,
      "grad_norm": 5.136506839394631,
      "learning_rate": 2.76921923546031e-06,
      "loss": 0.0227,
      "step": 2528
    },
    {
      "epoch": 1.8190972846610323,
      "grad_norm": 4.59435729111555,
      "learning_rate": 2.769037869089274e-06,
      "loss": 0.1467,
      "step": 2529
    },
    {
      "epoch": 1.8198165797518433,
      "grad_norm": 6.142522539934962,
      "learning_rate": 2.7688564374240113e-06,
      "loss": 0.2131,
      "step": 2530
    },
    {
      "epoch": 1.8205358748426543,
      "grad_norm": 2.7401917562809595,
      "learning_rate": 2.7686749404738578e-06,
      "loss": 0.042,
      "step": 2531
    },
    {
      "epoch": 1.8212551699334654,
      "grad_norm": 4.484183712917439,
      "learning_rate": 2.7684933782481516e-06,
      "loss": 0.1757,
      "step": 2532
    },
    {
      "epoch": 1.8219744650242762,
      "grad_norm": 4.321577310364181,
      "learning_rate": 2.768311750756234e-06,
      "loss": 0.0577,
      "step": 2533
    },
    {
      "epoch": 1.8226937601150872,
      "grad_norm": 5.6410512813533655,
      "learning_rate": 2.7681300580074503e-06,
      "loss": 0.1218,
      "step": 2534
    },
    {
      "epoch": 1.8234130552058982,
      "grad_norm": 1.4061629497049661,
      "learning_rate": 2.767948300011149e-06,
      "loss": 0.0082,
      "step": 2535
    },
    {
      "epoch": 1.824132350296709,
      "grad_norm": 5.666716902630463,
      "learning_rate": 2.7677664767766826e-06,
      "loss": 0.214,
      "step": 2536
    },
    {
      "epoch": 1.82485164538752,
      "grad_norm": 5.501311365360597,
      "learning_rate": 2.7675845883134043e-06,
      "loss": 0.1412,
      "step": 2537
    },
    {
      "epoch": 1.8255709404783311,
      "grad_norm": 6.143742811163902,
      "learning_rate": 2.7674026346306745e-06,
      "loss": 0.2483,
      "step": 2538
    },
    {
      "epoch": 1.8262902355691422,
      "grad_norm": 3.8972748751103556,
      "learning_rate": 2.767220615737854e-06,
      "loss": 0.1379,
      "step": 2539
    },
    {
      "epoch": 1.8270095306599532,
      "grad_norm": 4.082049239194885,
      "learning_rate": 2.7670385316443084e-06,
      "loss": 0.2196,
      "step": 2540
    },
    {
      "epoch": 1.8277288257507642,
      "grad_norm": 0.8622557730121388,
      "learning_rate": 2.7668563823594063e-06,
      "loss": 0.0042,
      "step": 2541
    },
    {
      "epoch": 1.8284481208415753,
      "grad_norm": 5.33404811561849,
      "learning_rate": 2.766674167892519e-06,
      "loss": 0.1734,
      "step": 2542
    },
    {
      "epoch": 1.8291674159323863,
      "grad_norm": 9.41902966258388,
      "learning_rate": 2.7664918882530226e-06,
      "loss": 0.0514,
      "step": 2543
    },
    {
      "epoch": 1.8298867110231973,
      "grad_norm": 2.108167626441434,
      "learning_rate": 2.766309543450295e-06,
      "loss": 0.0567,
      "step": 2544
    },
    {
      "epoch": 1.8306060061140084,
      "grad_norm": 3.797098679091282,
      "learning_rate": 2.766127133493719e-06,
      "loss": 0.0969,
      "step": 2545
    },
    {
      "epoch": 1.8313253012048194,
      "grad_norm": 5.9798385156995275,
      "learning_rate": 2.7659446583926786e-06,
      "loss": 0.2701,
      "step": 2546
    },
    {
      "epoch": 1.8320445962956304,
      "grad_norm": 1.7870722112789903,
      "learning_rate": 2.7657621181565637e-06,
      "loss": 0.029,
      "step": 2547
    },
    {
      "epoch": 1.8327638913864412,
      "grad_norm": 7.096455078234722,
      "learning_rate": 2.765579512794766e-06,
      "loss": 0.248,
      "step": 2548
    },
    {
      "epoch": 1.8334831864772523,
      "grad_norm": 2.8736836913931745,
      "learning_rate": 2.7653968423166806e-06,
      "loss": 0.0865,
      "step": 2549
    },
    {
      "epoch": 1.8342024815680633,
      "grad_norm": 2.703632678442727,
      "learning_rate": 2.765214106731706e-06,
      "loss": 0.1338,
      "step": 2550
    },
    {
      "epoch": 1.8349217766588743,
      "grad_norm": 4.787599189402569,
      "learning_rate": 2.765031306049245e-06,
      "loss": 0.2232,
      "step": 2551
    },
    {
      "epoch": 1.8356410717496852,
      "grad_norm": 2.9099792337197896,
      "learning_rate": 2.7648484402787024e-06,
      "loss": 0.0306,
      "step": 2552
    },
    {
      "epoch": 1.8363603668404962,
      "grad_norm": 3.7077688443027346,
      "learning_rate": 2.764665509429487e-06,
      "loss": 0.158,
      "step": 2553
    },
    {
      "epoch": 1.8370796619313072,
      "grad_norm": 6.2501465658507565,
      "learning_rate": 2.764482513511011e-06,
      "loss": 0.1424,
      "step": 2554
    },
    {
      "epoch": 1.8377989570221183,
      "grad_norm": 4.593488218805693,
      "learning_rate": 2.7642994525326906e-06,
      "loss": 0.097,
      "step": 2555
    },
    {
      "epoch": 1.8385182521129293,
      "grad_norm": 4.865436683626331,
      "learning_rate": 2.7641163265039436e-06,
      "loss": 0.1259,
      "step": 2556
    },
    {
      "epoch": 1.8392375472037403,
      "grad_norm": 3.4111713371569348,
      "learning_rate": 2.7639331354341924e-06,
      "loss": 0.1456,
      "step": 2557
    },
    {
      "epoch": 1.8399568422945514,
      "grad_norm": 6.077197315162484,
      "learning_rate": 2.763749879332863e-06,
      "loss": 0.1722,
      "step": 2558
    },
    {
      "epoch": 1.8406761373853624,
      "grad_norm": 4.350936199232469,
      "learning_rate": 2.7635665582093833e-06,
      "loss": 0.1583,
      "step": 2559
    },
    {
      "epoch": 1.8413954324761734,
      "grad_norm": 5.393039107024693,
      "learning_rate": 2.7633831720731862e-06,
      "loss": 0.1353,
      "step": 2560
    },
    {
      "epoch": 1.8421147275669845,
      "grad_norm": 3.724004833307669,
      "learning_rate": 2.7631997209337076e-06,
      "loss": 0.0764,
      "step": 2561
    },
    {
      "epoch": 1.8428340226577955,
      "grad_norm": 3.39537330649432,
      "learning_rate": 2.763016204800385e-06,
      "loss": 0.1358,
      "step": 2562
    },
    {
      "epoch": 1.8435533177486063,
      "grad_norm": 4.16350961159215,
      "learning_rate": 2.7628326236826623e-06,
      "loss": 0.1239,
      "step": 2563
    },
    {
      "epoch": 1.8442726128394173,
      "grad_norm": 2.3934462367152713,
      "learning_rate": 2.762648977589984e-06,
      "loss": 0.0467,
      "step": 2564
    },
    {
      "epoch": 1.8449919079302284,
      "grad_norm": 4.079145847815187,
      "learning_rate": 2.7624652665317993e-06,
      "loss": 0.0292,
      "step": 2565
    },
    {
      "epoch": 1.8457112030210394,
      "grad_norm": 3.829549074675402,
      "learning_rate": 2.76228149051756e-06,
      "loss": 0.1173,
      "step": 2566
    },
    {
      "epoch": 1.8464304981118502,
      "grad_norm": 2.8405438174038697,
      "learning_rate": 2.762097649556723e-06,
      "loss": 0.1238,
      "step": 2567
    },
    {
      "epoch": 1.8471497932026613,
      "grad_norm": 0.8928345250244636,
      "learning_rate": 2.761913743658746e-06,
      "loss": 0.0017,
      "step": 2568
    },
    {
      "epoch": 1.8478690882934723,
      "grad_norm": 1.351053113380072,
      "learning_rate": 2.7617297728330914e-06,
      "loss": 0.0224,
      "step": 2569
    },
    {
      "epoch": 1.8485883833842833,
      "grad_norm": 3.9917059487893614,
      "learning_rate": 2.7615457370892255e-06,
      "loss": 0.1113,
      "step": 2570
    },
    {
      "epoch": 1.8493076784750944,
      "grad_norm": 5.1258846233141595,
      "learning_rate": 2.7613616364366164e-06,
      "loss": 0.1161,
      "step": 2571
    },
    {
      "epoch": 1.8500269735659054,
      "grad_norm": 5.3875538424950395,
      "learning_rate": 2.7611774708847375e-06,
      "loss": 0.1207,
      "step": 2572
    },
    {
      "epoch": 1.8507462686567164,
      "grad_norm": 6.870508253130083,
      "learning_rate": 2.7609932404430634e-06,
      "loss": 0.134,
      "step": 2573
    },
    {
      "epoch": 1.8514655637475275,
      "grad_norm": 3.103572294299757,
      "learning_rate": 2.7608089451210737e-06,
      "loss": 0.0105,
      "step": 2574
    },
    {
      "epoch": 1.8521848588383385,
      "grad_norm": 4.7453135593811435,
      "learning_rate": 2.76062458492825e-06,
      "loss": 0.1542,
      "step": 2575
    },
    {
      "epoch": 1.8529041539291495,
      "grad_norm": 2.735687693644433,
      "learning_rate": 2.7604401598740787e-06,
      "loss": 0.087,
      "step": 2576
    },
    {
      "epoch": 1.8536234490199606,
      "grad_norm": 4.79297985610655,
      "learning_rate": 2.7602556699680486e-06,
      "loss": 0.105,
      "step": 2577
    },
    {
      "epoch": 1.8543427441107716,
      "grad_norm": 9.7597237804412,
      "learning_rate": 2.760071115219652e-06,
      "loss": 0.341,
      "step": 2578
    },
    {
      "epoch": 1.8550620392015824,
      "grad_norm": 1.5567371165359478,
      "learning_rate": 2.7598864956383844e-06,
      "loss": 0.0037,
      "step": 2579
    },
    {
      "epoch": 1.8557813342923934,
      "grad_norm": 7.088457252028718,
      "learning_rate": 2.759701811233745e-06,
      "loss": 0.1921,
      "step": 2580
    },
    {
      "epoch": 1.8565006293832045,
      "grad_norm": 4.789658381825956,
      "learning_rate": 2.7595170620152367e-06,
      "loss": 0.0824,
      "step": 2581
    },
    {
      "epoch": 1.8572199244740155,
      "grad_norm": 7.203203258850551,
      "learning_rate": 2.7593322479923637e-06,
      "loss": 0.3552,
      "step": 2582
    },
    {
      "epoch": 1.8579392195648263,
      "grad_norm": 3.8164995313332657,
      "learning_rate": 2.759147369174636e-06,
      "loss": 0.0987,
      "step": 2583
    },
    {
      "epoch": 1.8586585146556374,
      "grad_norm": 4.145443626045281,
      "learning_rate": 2.758962425571566e-06,
      "loss": 0.1554,
      "step": 2584
    },
    {
      "epoch": 1.8593778097464484,
      "grad_norm": 2.859034568344092,
      "learning_rate": 2.7587774171926694e-06,
      "loss": 0.0499,
      "step": 2585
    },
    {
      "epoch": 1.8600971048372594,
      "grad_norm": 4.499078457035729,
      "learning_rate": 2.7585923440474642e-06,
      "loss": 0.0215,
      "step": 2586
    },
    {
      "epoch": 1.8608163999280705,
      "grad_norm": 3.5980214138060136,
      "learning_rate": 2.758407206145474e-06,
      "loss": 0.1175,
      "step": 2587
    },
    {
      "epoch": 1.8615356950188815,
      "grad_norm": 5.321695178615645,
      "learning_rate": 2.7582220034962237e-06,
      "loss": 0.2354,
      "step": 2588
    },
    {
      "epoch": 1.8622549901096925,
      "grad_norm": 1.6756558997411046,
      "learning_rate": 2.758036736109243e-06,
      "loss": 0.0204,
      "step": 2589
    },
    {
      "epoch": 1.8629742852005036,
      "grad_norm": 2.113439479447713,
      "learning_rate": 2.7578514039940634e-06,
      "loss": 0.0151,
      "step": 2590
    },
    {
      "epoch": 1.8636935802913146,
      "grad_norm": 4.0202960158140355,
      "learning_rate": 2.7576660071602217e-06,
      "loss": 0.16,
      "step": 2591
    },
    {
      "epoch": 1.8644128753821256,
      "grad_norm": 7.580735602536769,
      "learning_rate": 2.7574805456172553e-06,
      "loss": 0.2725,
      "step": 2592
    },
    {
      "epoch": 1.8651321704729367,
      "grad_norm": 6.020124808693399,
      "learning_rate": 2.757295019374708e-06,
      "loss": 0.2323,
      "step": 2593
    },
    {
      "epoch": 1.8658514655637475,
      "grad_norm": 1.9967106864927242,
      "learning_rate": 2.7571094284421247e-06,
      "loss": 0.0081,
      "step": 2594
    },
    {
      "epoch": 1.8665707606545585,
      "grad_norm": 7.655655997374509,
      "learning_rate": 2.7569237728290543e-06,
      "loss": 0.3506,
      "step": 2595
    },
    {
      "epoch": 1.8672900557453695,
      "grad_norm": 4.758963619368994,
      "learning_rate": 2.7567380525450494e-06,
      "loss": 0.1633,
      "step": 2596
    },
    {
      "epoch": 1.8680093508361806,
      "grad_norm": 2.1736695132744055,
      "learning_rate": 2.7565522675996656e-06,
      "loss": 0.0791,
      "step": 2597
    },
    {
      "epoch": 1.8687286459269914,
      "grad_norm": 2.6852785671423516,
      "learning_rate": 2.7563664180024623e-06,
      "loss": 0.0302,
      "step": 2598
    },
    {
      "epoch": 1.8694479410178024,
      "grad_norm": 4.44922070142439,
      "learning_rate": 2.7561805037630006e-06,
      "loss": 0.0816,
      "step": 2599
    },
    {
      "epoch": 1.8701672361086135,
      "grad_norm": 5.138668862679516,
      "learning_rate": 2.755994524890847e-06,
      "loss": 0.0263,
      "step": 2600
    },
    {
      "epoch": 1.8708865311994245,
      "grad_norm": 5.33058579020469,
      "learning_rate": 2.75580848139557e-06,
      "loss": 0.108,
      "step": 2601
    },
    {
      "epoch": 1.8716058262902355,
      "grad_norm": 4.620534925820611,
      "learning_rate": 2.7556223732867428e-06,
      "loss": 0.0601,
      "step": 2602
    },
    {
      "epoch": 1.8723251213810466,
      "grad_norm": 1.4127467525521846,
      "learning_rate": 2.75543620057394e-06,
      "loss": 0.0293,
      "step": 2603
    },
    {
      "epoch": 1.8730444164718576,
      "grad_norm": 5.735407022147661,
      "learning_rate": 2.755249963266741e-06,
      "loss": 0.1163,
      "step": 2604
    },
    {
      "epoch": 1.8737637115626686,
      "grad_norm": 2.6254787174382495,
      "learning_rate": 2.7550636613747277e-06,
      "loss": 0.045,
      "step": 2605
    },
    {
      "epoch": 1.8744830066534797,
      "grad_norm": 5.262904717803658,
      "learning_rate": 2.7548772949074858e-06,
      "loss": 0.1409,
      "step": 2606
    },
    {
      "epoch": 1.8752023017442907,
      "grad_norm": 3.313369242155628,
      "learning_rate": 2.7546908638746044e-06,
      "loss": 0.0835,
      "step": 2607
    },
    {
      "epoch": 1.8759215968351017,
      "grad_norm": 5.690519549358206,
      "learning_rate": 2.754504368285675e-06,
      "loss": 0.1302,
      "step": 2608
    },
    {
      "epoch": 1.8766408919259128,
      "grad_norm": 3.8559148294499064,
      "learning_rate": 2.7543178081502944e-06,
      "loss": 0.2305,
      "step": 2609
    },
    {
      "epoch": 1.8773601870167236,
      "grad_norm": 4.165036960653529,
      "learning_rate": 2.75413118347806e-06,
      "loss": 0.1524,
      "step": 2610
    },
    {
      "epoch": 1.8780794821075346,
      "grad_norm": 7.982903612558595,
      "learning_rate": 2.7539444942785746e-06,
      "loss": 0.2369,
      "step": 2611
    },
    {
      "epoch": 1.8787987771983456,
      "grad_norm": 3.932219405413893,
      "learning_rate": 2.753757740561444e-06,
      "loss": 0.2065,
      "step": 2612
    },
    {
      "epoch": 1.8795180722891565,
      "grad_norm": 3.8268151438795592,
      "learning_rate": 2.753570922336277e-06,
      "loss": 0.017,
      "step": 2613
    },
    {
      "epoch": 1.8802373673799675,
      "grad_norm": 2.4966641738816078,
      "learning_rate": 2.7533840396126856e-06,
      "loss": 0.0936,
      "step": 2614
    },
    {
      "epoch": 1.8809566624707785,
      "grad_norm": 2.907683058504994,
      "learning_rate": 2.7531970924002847e-06,
      "loss": 0.0721,
      "step": 2615
    },
    {
      "epoch": 1.8816759575615896,
      "grad_norm": 5.254665989150888,
      "learning_rate": 2.7530100807086933e-06,
      "loss": 0.0943,
      "step": 2616
    },
    {
      "epoch": 1.8823952526524006,
      "grad_norm": 3.4759941344314944,
      "learning_rate": 2.752823004547534e-06,
      "loss": 0.0914,
      "step": 2617
    },
    {
      "epoch": 1.8831145477432116,
      "grad_norm": 6.3961238833266485,
      "learning_rate": 2.752635863926432e-06,
      "loss": 0.0936,
      "step": 2618
    },
    {
      "epoch": 1.8838338428340227,
      "grad_norm": 3.7628225317620707,
      "learning_rate": 2.752448658855015e-06,
      "loss": 0.2094,
      "step": 2619
    },
    {
      "epoch": 1.8845531379248337,
      "grad_norm": 1.5933574618098518,
      "learning_rate": 2.752261389342917e-06,
      "loss": 0.0059,
      "step": 2620
    },
    {
      "epoch": 1.8852724330156447,
      "grad_norm": 5.573897749077213,
      "learning_rate": 2.752074055399772e-06,
      "loss": 0.1308,
      "step": 2621
    },
    {
      "epoch": 1.8859917281064558,
      "grad_norm": 5.210401869334891,
      "learning_rate": 2.7518866570352184e-06,
      "loss": 0.1528,
      "step": 2622
    },
    {
      "epoch": 1.8867110231972668,
      "grad_norm": 4.6923379508666025,
      "learning_rate": 2.751699194258899e-06,
      "loss": 0.1212,
      "step": 2623
    },
    {
      "epoch": 1.8874303182880778,
      "grad_norm": 2.525731022066795,
      "learning_rate": 2.7515116670804587e-06,
      "loss": 0.1162,
      "step": 2624
    },
    {
      "epoch": 1.8881496133788886,
      "grad_norm": 3.53999169828396,
      "learning_rate": 2.7513240755095462e-06,
      "loss": 0.0713,
      "step": 2625
    },
    {
      "epoch": 1.8888689084696997,
      "grad_norm": 4.513193890063883,
      "learning_rate": 2.751136419555813e-06,
      "loss": 0.1254,
      "step": 2626
    },
    {
      "epoch": 1.8895882035605107,
      "grad_norm": 5.751532062687417,
      "learning_rate": 2.7509486992289157e-06,
      "loss": 0.1817,
      "step": 2627
    },
    {
      "epoch": 1.8903074986513217,
      "grad_norm": 4.685866684721937,
      "learning_rate": 2.7507609145385112e-06,
      "loss": 0.128,
      "step": 2628
    },
    {
      "epoch": 1.8910267937421326,
      "grad_norm": 2.0287089955374547,
      "learning_rate": 2.750573065494262e-06,
      "loss": 0.0492,
      "step": 2629
    },
    {
      "epoch": 1.8917460888329436,
      "grad_norm": 5.949195719536943,
      "learning_rate": 2.7503851521058333e-06,
      "loss": 0.1583,
      "step": 2630
    },
    {
      "epoch": 1.8924653839237546,
      "grad_norm": 3.2664639232048986,
      "learning_rate": 2.750197174382894e-06,
      "loss": 0.1058,
      "step": 2631
    },
    {
      "epoch": 1.8931846790145657,
      "grad_norm": 5.8958012880129775,
      "learning_rate": 2.7500091323351146e-06,
      "loss": 0.0533,
      "step": 2632
    },
    {
      "epoch": 1.8939039741053767,
      "grad_norm": 3.0500755610443875,
      "learning_rate": 2.7498210259721717e-06,
      "loss": 0.0786,
      "step": 2633
    },
    {
      "epoch": 1.8946232691961877,
      "grad_norm": 0.5781434341025135,
      "learning_rate": 2.749632855303743e-06,
      "loss": 0.001,
      "step": 2634
    },
    {
      "epoch": 1.8953425642869988,
      "grad_norm": 4.320845989212069,
      "learning_rate": 2.7494446203395094e-06,
      "loss": 0.0582,
      "step": 2635
    },
    {
      "epoch": 1.8960618593778098,
      "grad_norm": 5.099250674385128,
      "learning_rate": 2.7492563210891573e-06,
      "loss": 0.1537,
      "step": 2636
    },
    {
      "epoch": 1.8967811544686208,
      "grad_norm": 5.657414802029139,
      "learning_rate": 2.7490679575623746e-06,
      "loss": 0.1388,
      "step": 2637
    },
    {
      "epoch": 1.8975004495594319,
      "grad_norm": 3.6585869069263803,
      "learning_rate": 2.748879529768853e-06,
      "loss": 0.198,
      "step": 2638
    },
    {
      "epoch": 1.898219744650243,
      "grad_norm": 4.718944659374927,
      "learning_rate": 2.748691037718287e-06,
      "loss": 0.0789,
      "step": 2639
    },
    {
      "epoch": 1.8989390397410537,
      "grad_norm": 7.6545431678876295,
      "learning_rate": 2.748502481420375e-06,
      "loss": 0.0659,
      "step": 2640
    },
    {
      "epoch": 1.8996583348318647,
      "grad_norm": 2.9068554596982046,
      "learning_rate": 2.7483138608848186e-06,
      "loss": 0.0932,
      "step": 2641
    },
    {
      "epoch": 1.9003776299226758,
      "grad_norm": 6.576430349344039,
      "learning_rate": 2.748125176121323e-06,
      "loss": 0.1231,
      "step": 2642
    },
    {
      "epoch": 1.9010969250134868,
      "grad_norm": 3.473443227864142,
      "learning_rate": 2.7479364271395958e-06,
      "loss": 0.0188,
      "step": 2643
    },
    {
      "epoch": 1.9018162201042976,
      "grad_norm": 3.166504488870473,
      "learning_rate": 2.7477476139493492e-06,
      "loss": 0.057,
      "step": 2644
    },
    {
      "epoch": 1.9025355151951087,
      "grad_norm": 3.9595257531034527,
      "learning_rate": 2.7475587365602966e-06,
      "loss": 0.086,
      "step": 2645
    },
    {
      "epoch": 1.9032548102859197,
      "grad_norm": 10.51806716799473,
      "learning_rate": 2.7473697949821576e-06,
      "loss": 0.0866,
      "step": 2646
    },
    {
      "epoch": 1.9039741053767307,
      "grad_norm": 0.824083774413045,
      "learning_rate": 2.7471807892246533e-06,
      "loss": 0.0054,
      "step": 2647
    },
    {
      "epoch": 1.9046934004675418,
      "grad_norm": 8.661154600812727,
      "learning_rate": 2.7469917192975075e-06,
      "loss": 0.2572,
      "step": 2648
    },
    {
      "epoch": 1.9054126955583528,
      "grad_norm": 3.1450275233360907,
      "learning_rate": 2.7468025852104492e-06,
      "loss": 0.0199,
      "step": 2649
    },
    {
      "epoch": 1.9061319906491638,
      "grad_norm": 5.35192237307085,
      "learning_rate": 2.746613386973209e-06,
      "loss": 0.0682,
      "step": 2650
    },
    {
      "epoch": 1.9068512857399749,
      "grad_norm": 4.055016494275361,
      "learning_rate": 2.7464241245955213e-06,
      "loss": 0.1067,
      "step": 2651
    },
    {
      "epoch": 1.907570580830786,
      "grad_norm": 3.711703605062127,
      "learning_rate": 2.7462347980871244e-06,
      "loss": 0.1793,
      "step": 2652
    },
    {
      "epoch": 1.908289875921597,
      "grad_norm": 1.4010358917622097,
      "learning_rate": 2.74604540745776e-06,
      "loss": 0.0201,
      "step": 2653
    },
    {
      "epoch": 1.909009171012408,
      "grad_norm": 2.327665862942012,
      "learning_rate": 2.745855952717171e-06,
      "loss": 0.0588,
      "step": 2654
    },
    {
      "epoch": 1.909728466103219,
      "grad_norm": 8.620272490399229,
      "learning_rate": 2.7456664338751074e-06,
      "loss": 0.3683,
      "step": 2655
    },
    {
      "epoch": 1.9104477611940298,
      "grad_norm": 2.514204228254778,
      "learning_rate": 2.745476850941318e-06,
      "loss": 0.0156,
      "step": 2656
    },
    {
      "epoch": 1.9111670562848408,
      "grad_norm": 3.194223407839609,
      "learning_rate": 2.7452872039255593e-06,
      "loss": 0.0932,
      "step": 2657
    },
    {
      "epoch": 1.9118863513756519,
      "grad_norm": 3.0134651312825143,
      "learning_rate": 2.7450974928375873e-06,
      "loss": 0.0702,
      "step": 2658
    },
    {
      "epoch": 1.912605646466463,
      "grad_norm": 5.905898751964238,
      "learning_rate": 2.7449077176871636e-06,
      "loss": 0.2473,
      "step": 2659
    },
    {
      "epoch": 1.9133249415572737,
      "grad_norm": 3.5573901440104736,
      "learning_rate": 2.7447178784840527e-06,
      "loss": 0.1384,
      "step": 2660
    },
    {
      "epoch": 1.9140442366480848,
      "grad_norm": 1.955068255783858,
      "learning_rate": 2.744527975238022e-06,
      "loss": 0.0273,
      "step": 2661
    },
    {
      "epoch": 1.9147635317388958,
      "grad_norm": 5.929418601396656,
      "learning_rate": 2.7443380079588418e-06,
      "loss": 0.0575,
      "step": 2662
    },
    {
      "epoch": 1.9154828268297068,
      "grad_norm": 4.742786296076306,
      "learning_rate": 2.744147976656287e-06,
      "loss": 0.1081,
      "step": 2663
    },
    {
      "epoch": 1.9162021219205179,
      "grad_norm": 0.9801979342671292,
      "learning_rate": 2.743957881340135e-06,
      "loss": 0.0021,
      "step": 2664
    },
    {
      "epoch": 1.916921417011329,
      "grad_norm": 4.0643078107752295,
      "learning_rate": 2.743767722020166e-06,
      "loss": 0.0646,
      "step": 2665
    },
    {
      "epoch": 1.91764071210214,
      "grad_norm": 4.589616284196056,
      "learning_rate": 2.743577498706164e-06,
      "loss": 0.0579,
      "step": 2666
    },
    {
      "epoch": 1.918360007192951,
      "grad_norm": 2.877835691001051,
      "learning_rate": 2.7433872114079177e-06,
      "loss": 0.021,
      "step": 2667
    },
    {
      "epoch": 1.919079302283762,
      "grad_norm": 5.184181287637767,
      "learning_rate": 2.743196860135216e-06,
      "loss": 0.173,
      "step": 2668
    },
    {
      "epoch": 1.919798597374573,
      "grad_norm": 3.140217264246085,
      "learning_rate": 2.7430064448978536e-06,
      "loss": 0.0595,
      "step": 2669
    },
    {
      "epoch": 1.920517892465384,
      "grad_norm": 9.49980406971857,
      "learning_rate": 2.742815965705627e-06,
      "loss": 0.1151,
      "step": 2670
    },
    {
      "epoch": 1.9212371875561949,
      "grad_norm": 7.643895606210381,
      "learning_rate": 2.7426254225683383e-06,
      "loss": 0.1632,
      "step": 2671
    },
    {
      "epoch": 1.921956482647006,
      "grad_norm": 2.1327788422579403,
      "learning_rate": 2.7424348154957895e-06,
      "loss": 0.0511,
      "step": 2672
    },
    {
      "epoch": 1.922675777737817,
      "grad_norm": 3.8966712982055776,
      "learning_rate": 2.742244144497789e-06,
      "loss": 0.0598,
      "step": 2673
    },
    {
      "epoch": 1.923395072828628,
      "grad_norm": 4.010080813656915,
      "learning_rate": 2.742053409584146e-06,
      "loss": 0.2432,
      "step": 2674
    },
    {
      "epoch": 1.9241143679194388,
      "grad_norm": 3.805906449783639,
      "learning_rate": 2.7418626107646747e-06,
      "loss": 0.0423,
      "step": 2675
    },
    {
      "epoch": 1.9248336630102498,
      "grad_norm": 8.633212801204584,
      "learning_rate": 2.7416717480491922e-06,
      "loss": 0.1525,
      "step": 2676
    },
    {
      "epoch": 1.9255529581010609,
      "grad_norm": 4.549930590222354,
      "learning_rate": 2.741480821447519e-06,
      "loss": 0.0995,
      "step": 2677
    },
    {
      "epoch": 1.926272253191872,
      "grad_norm": 4.193951764747331,
      "learning_rate": 2.7412898309694776e-06,
      "loss": 0.1787,
      "step": 2678
    },
    {
      "epoch": 1.926991548282683,
      "grad_norm": 3.4876358720109097,
      "learning_rate": 2.741098776624896e-06,
      "loss": 0.1067,
      "step": 2679
    },
    {
      "epoch": 1.927710843373494,
      "grad_norm": 1.5812705732780352,
      "learning_rate": 2.7409076584236032e-06,
      "loss": 0.0281,
      "step": 2680
    },
    {
      "epoch": 1.928430138464305,
      "grad_norm": 6.508487758473994,
      "learning_rate": 2.740716476375433e-06,
      "loss": 0.0684,
      "step": 2681
    },
    {
      "epoch": 1.929149433555116,
      "grad_norm": 7.090071138435471,
      "learning_rate": 2.7405252304902225e-06,
      "loss": 0.2076,
      "step": 2682
    },
    {
      "epoch": 1.929868728645927,
      "grad_norm": 9.593967799666707,
      "learning_rate": 2.7403339207778105e-06,
      "loss": 0.3193,
      "step": 2683
    },
    {
      "epoch": 1.930588023736738,
      "grad_norm": 8.763926929831355,
      "learning_rate": 2.7401425472480414e-06,
      "loss": 0.4113,
      "step": 2684
    },
    {
      "epoch": 1.9313073188275491,
      "grad_norm": 2.8955824580313805,
      "learning_rate": 2.739951109910761e-06,
      "loss": 0.1063,
      "step": 2685
    },
    {
      "epoch": 1.9320266139183602,
      "grad_norm": 0.44832818791809065,
      "learning_rate": 2.73975960877582e-06,
      "loss": 0.0024,
      "step": 2686
    },
    {
      "epoch": 1.932745909009171,
      "grad_norm": 4.736195941011606,
      "learning_rate": 2.73956804385307e-06,
      "loss": 0.232,
      "step": 2687
    },
    {
      "epoch": 1.933465204099982,
      "grad_norm": 0.9406893099653586,
      "learning_rate": 2.7393764151523686e-06,
      "loss": 0.003,
      "step": 2688
    },
    {
      "epoch": 1.934184499190793,
      "grad_norm": 4.651353938025924,
      "learning_rate": 2.739184722683575e-06,
      "loss": 0.1075,
      "step": 2689
    },
    {
      "epoch": 1.9349037942816039,
      "grad_norm": 3.26245172926038,
      "learning_rate": 2.738992966456552e-06,
      "loss": 0.159,
      "step": 2690
    },
    {
      "epoch": 1.935623089372415,
      "grad_norm": 4.901951626246722,
      "learning_rate": 2.738801146481166e-06,
      "loss": 0.0288,
      "step": 2691
    },
    {
      "epoch": 1.936342384463226,
      "grad_norm": 8.296549801205924,
      "learning_rate": 2.738609262767286e-06,
      "loss": 0.1843,
      "step": 2692
    },
    {
      "epoch": 1.937061679554037,
      "grad_norm": 3.739199107182429,
      "learning_rate": 2.7384173153247856e-06,
      "loss": 0.1029,
      "step": 2693
    },
    {
      "epoch": 1.937780974644848,
      "grad_norm": 7.113141510342706,
      "learning_rate": 2.7382253041635402e-06,
      "loss": 0.0925,
      "step": 2694
    },
    {
      "epoch": 1.938500269735659,
      "grad_norm": 2.8530266938887974,
      "learning_rate": 2.738033229293429e-06,
      "loss": 0.0777,
      "step": 2695
    },
    {
      "epoch": 1.93921956482647,
      "grad_norm": 3.2066562835307817,
      "learning_rate": 2.737841090724335e-06,
      "loss": 0.0451,
      "step": 2696
    },
    {
      "epoch": 1.939938859917281,
      "grad_norm": 5.885262123716507,
      "learning_rate": 2.737648888466144e-06,
      "loss": 0.0783,
      "step": 2697
    },
    {
      "epoch": 1.9406581550080921,
      "grad_norm": 4.174928963202971,
      "learning_rate": 2.7374566225287447e-06,
      "loss": 0.2098,
      "step": 2698
    },
    {
      "epoch": 1.9413774500989032,
      "grad_norm": 2.413024916685747,
      "learning_rate": 2.7372642929220306e-06,
      "loss": 0.0104,
      "step": 2699
    },
    {
      "epoch": 1.9420967451897142,
      "grad_norm": 3.432784003167446,
      "learning_rate": 2.737071899655896e-06,
      "loss": 0.0233,
      "step": 2700
    },
    {
      "epoch": 1.9428160402805252,
      "grad_norm": 1.1351395875892847,
      "learning_rate": 2.736879442740241e-06,
      "loss": 0.0122,
      "step": 2701
    },
    {
      "epoch": 1.943535335371336,
      "grad_norm": 3.989611278976564,
      "learning_rate": 2.736686922184967e-06,
      "loss": 0.0803,
      "step": 2702
    },
    {
      "epoch": 1.944254630462147,
      "grad_norm": 0.875615250929831,
      "learning_rate": 2.73649433799998e-06,
      "loss": 0.0217,
      "step": 2703
    },
    {
      "epoch": 1.9449739255529581,
      "grad_norm": 2.7863064472305306,
      "learning_rate": 2.7363016901951894e-06,
      "loss": 0.0119,
      "step": 2704
    },
    {
      "epoch": 1.9456932206437692,
      "grad_norm": 4.530716452232525,
      "learning_rate": 2.736108978780506e-06,
      "loss": 0.1577,
      "step": 2705
    },
    {
      "epoch": 1.94641251573458,
      "grad_norm": 4.632707994491346,
      "learning_rate": 2.7359162037658456e-06,
      "loss": 0.1425,
      "step": 2706
    },
    {
      "epoch": 1.947131810825391,
      "grad_norm": 2.108005426848352,
      "learning_rate": 2.735723365161127e-06,
      "loss": 0.0605,
      "step": 2707
    },
    {
      "epoch": 1.947851105916202,
      "grad_norm": 3.829371895094149,
      "learning_rate": 2.735530462976272e-06,
      "loss": 0.1068,
      "step": 2708
    },
    {
      "epoch": 1.948570401007013,
      "grad_norm": 7.11910574314732,
      "learning_rate": 2.735337497221206e-06,
      "loss": 0.0367,
      "step": 2709
    },
    {
      "epoch": 1.949289696097824,
      "grad_norm": 0.1332685536008957,
      "learning_rate": 2.735144467905857e-06,
      "loss": 0.0006,
      "step": 2710
    },
    {
      "epoch": 1.9500089911886351,
      "grad_norm": 3.9452746036807538,
      "learning_rate": 2.7349513750401574e-06,
      "loss": 0.071,
      "step": 2711
    },
    {
      "epoch": 1.9507282862794462,
      "grad_norm": 1.879051716352009,
      "learning_rate": 2.734758218634041e-06,
      "loss": 0.0093,
      "step": 2712
    },
    {
      "epoch": 1.9514475813702572,
      "grad_norm": 5.049579027171884,
      "learning_rate": 2.734564998697447e-06,
      "loss": 0.1577,
      "step": 2713
    },
    {
      "epoch": 1.9521668764610682,
      "grad_norm": 1.6533844928468322,
      "learning_rate": 2.7343717152403164e-06,
      "loss": 0.0645,
      "step": 2714
    },
    {
      "epoch": 1.9528861715518793,
      "grad_norm": 3.422795443349315,
      "learning_rate": 2.7341783682725946e-06,
      "loss": 0.099,
      "step": 2715
    },
    {
      "epoch": 1.9536054666426903,
      "grad_norm": 3.5319280093702177,
      "learning_rate": 2.733984957804229e-06,
      "loss": 0.0884,
      "step": 2716
    },
    {
      "epoch": 1.9543247617335011,
      "grad_norm": 3.324243580628092,
      "learning_rate": 2.733791483845171e-06,
      "loss": 0.0941,
      "step": 2717
    },
    {
      "epoch": 1.9550440568243121,
      "grad_norm": 3.338605218176604,
      "learning_rate": 2.7335979464053755e-06,
      "loss": 0.0484,
      "step": 2718
    },
    {
      "epoch": 1.9557633519151232,
      "grad_norm": 3.7113958770767446,
      "learning_rate": 2.7334043454948e-06,
      "loss": 0.0239,
      "step": 2719
    },
    {
      "epoch": 1.9564826470059342,
      "grad_norm": 3.779995322362371,
      "learning_rate": 2.733210681123406e-06,
      "loss": 0.038,
      "step": 2720
    },
    {
      "epoch": 1.957201942096745,
      "grad_norm": 5.2104804683943,
      "learning_rate": 2.733016953301157e-06,
      "loss": 0.1087,
      "step": 2721
    },
    {
      "epoch": 1.957921237187556,
      "grad_norm": 5.8591328096040876,
      "learning_rate": 2.732823162038022e-06,
      "loss": 0.035,
      "step": 2722
    },
    {
      "epoch": 1.958640532278367,
      "grad_norm": 2.569942583080411,
      "learning_rate": 2.7326293073439703e-06,
      "loss": 0.0538,
      "step": 2723
    },
    {
      "epoch": 1.9593598273691781,
      "grad_norm": 5.764905833702417,
      "learning_rate": 2.732435389228978e-06,
      "loss": 0.132,
      "step": 2724
    },
    {
      "epoch": 1.9600791224599892,
      "grad_norm": 2.434867160531595,
      "learning_rate": 2.7322414077030207e-06,
      "loss": 0.0145,
      "step": 2725
    },
    {
      "epoch": 1.9607984175508002,
      "grad_norm": 3.100274728637934,
      "learning_rate": 2.73204736277608e-06,
      "loss": 0.044,
      "step": 2726
    },
    {
      "epoch": 1.9615177126416112,
      "grad_norm": 4.48620272563212,
      "learning_rate": 2.7318532544581395e-06,
      "loss": 0.1496,
      "step": 2727
    },
    {
      "epoch": 1.9622370077324223,
      "grad_norm": 2.800612765615511,
      "learning_rate": 2.731659082759187e-06,
      "loss": 0.0753,
      "step": 2728
    },
    {
      "epoch": 1.9629563028232333,
      "grad_norm": 2.932305866401869,
      "learning_rate": 2.7314648476892126e-06,
      "loss": 0.0116,
      "step": 2729
    },
    {
      "epoch": 1.9636755979140443,
      "grad_norm": 6.408723630327014,
      "learning_rate": 2.7312705492582097e-06,
      "loss": 0.1096,
      "step": 2730
    },
    {
      "epoch": 1.9643948930048554,
      "grad_norm": 4.255124917951107,
      "learning_rate": 2.7310761874761758e-06,
      "loss": 0.1877,
      "step": 2731
    },
    {
      "epoch": 1.9651141880956664,
      "grad_norm": 3.2247639669575157,
      "learning_rate": 2.730881762353111e-06,
      "loss": 0.0845,
      "step": 2732
    },
    {
      "epoch": 1.9658334831864772,
      "grad_norm": 3.9895061821278,
      "learning_rate": 2.7306872738990186e-06,
      "loss": 0.0611,
      "step": 2733
    },
    {
      "epoch": 1.9665527782772882,
      "grad_norm": 3.089948161774254,
      "learning_rate": 2.730492722123906e-06,
      "loss": 0.1164,
      "step": 2734
    },
    {
      "epoch": 1.9672720733680993,
      "grad_norm": 4.254484432369005,
      "learning_rate": 2.7302981070377823e-06,
      "loss": 0.0134,
      "step": 2735
    },
    {
      "epoch": 1.9679913684589103,
      "grad_norm": 3.6114213810200524,
      "learning_rate": 2.7301034286506616e-06,
      "loss": 0.1131,
      "step": 2736
    },
    {
      "epoch": 1.9687106635497211,
      "grad_norm": 2.9160693307549526,
      "learning_rate": 2.7299086869725597e-06,
      "loss": 0.056,
      "step": 2737
    },
    {
      "epoch": 1.9694299586405322,
      "grad_norm": 3.7748429195859763,
      "learning_rate": 2.729713882013497e-06,
      "loss": 0.0821,
      "step": 2738
    },
    {
      "epoch": 1.9701492537313432,
      "grad_norm": 2.5399610012448925,
      "learning_rate": 2.7295190137834967e-06,
      "loss": 0.0503,
      "step": 2739
    },
    {
      "epoch": 1.9708685488221542,
      "grad_norm": 5.202703122086643,
      "learning_rate": 2.729324082292585e-06,
      "loss": 0.1298,
      "step": 2740
    },
    {
      "epoch": 1.9715878439129653,
      "grad_norm": 2.4220772728369924,
      "learning_rate": 2.7291290875507904e-06,
      "loss": 0.0243,
      "step": 2741
    },
    {
      "epoch": 1.9723071390037763,
      "grad_norm": 0.6595968108214497,
      "learning_rate": 2.728934029568147e-06,
      "loss": 0.0025,
      "step": 2742
    },
    {
      "epoch": 1.9730264340945873,
      "grad_norm": 4.125301787334915,
      "learning_rate": 2.72873890835469e-06,
      "loss": 0.0697,
      "step": 2743
    },
    {
      "epoch": 1.9737457291853984,
      "grad_norm": 2.7893690605429655,
      "learning_rate": 2.7285437239204594e-06,
      "loss": 0.0174,
      "step": 2744
    },
    {
      "epoch": 1.9744650242762094,
      "grad_norm": 4.842528841922146,
      "learning_rate": 2.7283484762754986e-06,
      "loss": 0.0658,
      "step": 2745
    },
    {
      "epoch": 1.9751843193670204,
      "grad_norm": 1.624597495959097,
      "learning_rate": 2.7281531654298512e-06,
      "loss": 0.0072,
      "step": 2746
    },
    {
      "epoch": 1.9759036144578315,
      "grad_norm": 3.007118944458473,
      "learning_rate": 2.727957791393568e-06,
      "loss": 0.0979,
      "step": 2747
    },
    {
      "epoch": 1.9766229095486423,
      "grad_norm": 7.194295187142598,
      "learning_rate": 2.7277623541767004e-06,
      "loss": 0.2966,
      "step": 2748
    },
    {
      "epoch": 1.9773422046394533,
      "grad_norm": 3.6850712655862625,
      "learning_rate": 2.7275668537893046e-06,
      "loss": 0.0186,
      "step": 2749
    },
    {
      "epoch": 1.9780614997302643,
      "grad_norm": 9.269940023444025,
      "learning_rate": 2.7273712902414397e-06,
      "loss": 0.1964,
      "step": 2750
    },
    {
      "epoch": 1.9787807948210754,
      "grad_norm": 1.9688814697774948,
      "learning_rate": 2.727175663543167e-06,
      "loss": 0.0509,
      "step": 2751
    },
    {
      "epoch": 1.9795000899118862,
      "grad_norm": 4.955911007828335,
      "learning_rate": 2.726979973704552e-06,
      "loss": 0.0855,
      "step": 2752
    },
    {
      "epoch": 1.9802193850026972,
      "grad_norm": 5.2816354592967345,
      "learning_rate": 2.7267842207356636e-06,
      "loss": 0.0273,
      "step": 2753
    },
    {
      "epoch": 1.9809386800935083,
      "grad_norm": 3.956057839838247,
      "learning_rate": 2.7265884046465734e-06,
      "loss": 0.1666,
      "step": 2754
    },
    {
      "epoch": 1.9816579751843193,
      "grad_norm": 1.6072058821572746,
      "learning_rate": 2.726392525447357e-06,
      "loss": 0.0213,
      "step": 2755
    },
    {
      "epoch": 1.9823772702751303,
      "grad_norm": 6.4775060702831695,
      "learning_rate": 2.7261965831480917e-06,
      "loss": 0.1527,
      "step": 2756
    },
    {
      "epoch": 1.9830965653659414,
      "grad_norm": 4.192038209219215,
      "learning_rate": 2.72600057775886e-06,
      "loss": 0.0996,
      "step": 2757
    },
    {
      "epoch": 1.9838158604567524,
      "grad_norm": 4.860116427408447,
      "learning_rate": 2.725804509289746e-06,
      "loss": 0.1071,
      "step": 2758
    },
    {
      "epoch": 1.9845351555475634,
      "grad_norm": 2.3987438825732754,
      "learning_rate": 2.725608377750839e-06,
      "loss": 0.0711,
      "step": 2759
    },
    {
      "epoch": 1.9852544506383745,
      "grad_norm": 1.7101575941302474,
      "learning_rate": 2.725412183152229e-06,
      "loss": 0.0082,
      "step": 2760
    },
    {
      "epoch": 1.9859737457291855,
      "grad_norm": 2.3136104429305875,
      "learning_rate": 2.7252159255040107e-06,
      "loss": 0.0102,
      "step": 2761
    },
    {
      "epoch": 1.9866930408199965,
      "grad_norm": 2.989112624623687,
      "learning_rate": 2.7250196048162824e-06,
      "loss": 0.095,
      "step": 2762
    },
    {
      "epoch": 1.9874123359108076,
      "grad_norm": 4.803505387877456,
      "learning_rate": 2.724823221099145e-06,
      "loss": 0.1737,
      "step": 2763
    },
    {
      "epoch": 1.9881316310016184,
      "grad_norm": 3.5616184297518543,
      "learning_rate": 2.724626774362703e-06,
      "loss": 0.1122,
      "step": 2764
    },
    {
      "epoch": 1.9888509260924294,
      "grad_norm": 3.99910606016285,
      "learning_rate": 2.7244302646170637e-06,
      "loss": 0.0485,
      "step": 2765
    },
    {
      "epoch": 1.9895702211832405,
      "grad_norm": 5.244362464451042,
      "learning_rate": 2.7242336918723377e-06,
      "loss": 0.0535,
      "step": 2766
    },
    {
      "epoch": 1.9902895162740515,
      "grad_norm": 3.1386017639220856,
      "learning_rate": 2.7240370561386392e-06,
      "loss": 0.0881,
      "step": 2767
    },
    {
      "epoch": 1.9910088113648623,
      "grad_norm": 3.2222745840776916,
      "learning_rate": 2.7238403574260854e-06,
      "loss": 0.0747,
      "step": 2768
    },
    {
      "epoch": 1.9917281064556733,
      "grad_norm": 3.691132173757284,
      "learning_rate": 2.723643595744797e-06,
      "loss": 0.2281,
      "step": 2769
    },
    {
      "epoch": 1.9924474015464844,
      "grad_norm": 3.003509620936156,
      "learning_rate": 2.7234467711048974e-06,
      "loss": 0.0048,
      "step": 2770
    },
    {
      "epoch": 1.9931666966372954,
      "grad_norm": 3.9822283598442154,
      "learning_rate": 2.7232498835165137e-06,
      "loss": 0.0562,
      "step": 2771
    },
    {
      "epoch": 1.9938859917281064,
      "grad_norm": 5.5085108815195944,
      "learning_rate": 2.723052932989776e-06,
      "loss": 0.0914,
      "step": 2772
    },
    {
      "epoch": 1.9946052868189175,
      "grad_norm": 4.905355923487537,
      "learning_rate": 2.7228559195348184e-06,
      "loss": 0.193,
      "step": 2773
    },
    {
      "epoch": 1.9953245819097285,
      "grad_norm": 2.6997842587859164,
      "learning_rate": 2.7226588431617766e-06,
      "loss": 0.1181,
      "step": 2774
    },
    {
      "epoch": 1.9960438770005395,
      "grad_norm": 3.955298846710558,
      "learning_rate": 2.7224617038807913e-06,
      "loss": 0.1458,
      "step": 2775
    },
    {
      "epoch": 1.9967631720913506,
      "grad_norm": 3.2238497763550464,
      "learning_rate": 2.7222645017020053e-06,
      "loss": 0.1708,
      "step": 2776
    },
    {
      "epoch": 1.9974824671821616,
      "grad_norm": 7.317171371890231,
      "learning_rate": 2.722067236635565e-06,
      "loss": 0.074,
      "step": 2777
    },
    {
      "epoch": 1.9982017622729726,
      "grad_norm": 4.054485450254262,
      "learning_rate": 2.7218699086916204e-06,
      "loss": 0.0595,
      "step": 2778
    },
    {
      "epoch": 1.9989210573637834,
      "grad_norm": 2.029654451210026,
      "learning_rate": 2.7216725178803233e-06,
      "loss": 0.0104,
      "step": 2779
    },
    {
      "epoch": 1.9996403524545945,
      "grad_norm": 4.55743599142024,
      "learning_rate": 2.7214750642118314e-06,
      "loss": 0.1304,
      "step": 2780
    },
    {
      "epoch": 2.0003596475454053,
      "grad_norm": 0.6247991836602562,
      "learning_rate": 2.721277547696303e-06,
      "loss": 0.0013,
      "step": 2781
    },
    {
      "epoch": 2.0010789426362163,
      "grad_norm": 2.219010924610205,
      "learning_rate": 2.721079968343901e-06,
      "loss": 0.0119,
      "step": 2782
    },
    {
      "epoch": 2.0017982377270274,
      "grad_norm": 0.9012996985022022,
      "learning_rate": 2.720882326164791e-06,
      "loss": 0.0011,
      "step": 2783
    },
    {
      "epoch": 2.0025175328178384,
      "grad_norm": 4.030547548310015,
      "learning_rate": 2.7206846211691415e-06,
      "loss": 0.0817,
      "step": 2784
    },
    {
      "epoch": 2.0032368279086494,
      "grad_norm": 5.619457071778356,
      "learning_rate": 2.7204868533671258e-06,
      "loss": 0.0245,
      "step": 2785
    },
    {
      "epoch": 2.0039561229994605,
      "grad_norm": 4.47418191124176,
      "learning_rate": 2.7202890227689194e-06,
      "loss": 0.0662,
      "step": 2786
    },
    {
      "epoch": 2.0046754180902715,
      "grad_norm": 2.9616917963504137,
      "learning_rate": 2.7200911293847007e-06,
      "loss": 0.0271,
      "step": 2787
    },
    {
      "epoch": 2.0053947131810825,
      "grad_norm": 2.066912038355341,
      "learning_rate": 2.719893173224651e-06,
      "loss": 0.0604,
      "step": 2788
    },
    {
      "epoch": 2.0061140082718936,
      "grad_norm": 4.43492987418814,
      "learning_rate": 2.719695154298956e-06,
      "loss": 0.1186,
      "step": 2789
    },
    {
      "epoch": 2.0068333033627046,
      "grad_norm": 3.814897479505931,
      "learning_rate": 2.7194970726178046e-06,
      "loss": 0.1258,
      "step": 2790
    },
    {
      "epoch": 2.0075525984535156,
      "grad_norm": 3.0465531715652094,
      "learning_rate": 2.7192989281913882e-06,
      "loss": 0.0639,
      "step": 2791
    },
    {
      "epoch": 2.0082718935443267,
      "grad_norm": 4.155592607552763,
      "learning_rate": 2.7191007210299014e-06,
      "loss": 0.018,
      "step": 2792
    },
    {
      "epoch": 2.0089911886351377,
      "grad_norm": 3.6127933444386837,
      "learning_rate": 2.718902451143542e-06,
      "loss": 0.114,
      "step": 2793
    },
    {
      "epoch": 2.0097104837259487,
      "grad_norm": 3.052211247299576,
      "learning_rate": 2.718704118542512e-06,
      "loss": 0.1391,
      "step": 2794
    },
    {
      "epoch": 2.0104297788167598,
      "grad_norm": 1.9941958659627657,
      "learning_rate": 2.718505723237016e-06,
      "loss": 0.0746,
      "step": 2795
    },
    {
      "epoch": 2.0111490739075704,
      "grad_norm": 6.271733071387418,
      "learning_rate": 2.7183072652372614e-06,
      "loss": 0.241,
      "step": 2796
    },
    {
      "epoch": 2.0118683689983814,
      "grad_norm": 4.5760518934568,
      "learning_rate": 2.7181087445534596e-06,
      "loss": 0.0366,
      "step": 2797
    },
    {
      "epoch": 2.0125876640891924,
      "grad_norm": 4.454022890322071,
      "learning_rate": 2.717910161195824e-06,
      "loss": 0.076,
      "step": 2798
    },
    {
      "epoch": 2.0133069591800035,
      "grad_norm": 0.37003568008588267,
      "learning_rate": 2.717711515174573e-06,
      "loss": 0.0011,
      "step": 2799
    },
    {
      "epoch": 2.0140262542708145,
      "grad_norm": 3.085489732709743,
      "learning_rate": 2.717512806499927e-06,
      "loss": 0.1056,
      "step": 2800
    },
    {
      "epoch": 2.0147455493616255,
      "grad_norm": 5.079153035692809,
      "learning_rate": 2.7173140351821095e-06,
      "loss": 0.2155,
      "step": 2801
    },
    {
      "epoch": 2.0154648444524366,
      "grad_norm": 1.619421015483736,
      "learning_rate": 2.717115201231348e-06,
      "loss": 0.0067,
      "step": 2802
    },
    {
      "epoch": 2.0161841395432476,
      "grad_norm": 3.1639939847466043,
      "learning_rate": 2.716916304657873e-06,
      "loss": 0.0518,
      "step": 2803
    },
    {
      "epoch": 2.0169034346340586,
      "grad_norm": 4.318803062561228,
      "learning_rate": 2.716717345471918e-06,
      "loss": 0.0399,
      "step": 2804
    },
    {
      "epoch": 2.0176227297248697,
      "grad_norm": 2.268858526030059,
      "learning_rate": 2.7165183236837197e-06,
      "loss": 0.0135,
      "step": 2805
    },
    {
      "epoch": 2.0183420248156807,
      "grad_norm": 1.546237046231724,
      "learning_rate": 2.7163192393035175e-06,
      "loss": 0.0062,
      "step": 2806
    },
    {
      "epoch": 2.0190613199064917,
      "grad_norm": 1.9073589745711095,
      "learning_rate": 2.716120092341556e-06,
      "loss": 0.0271,
      "step": 2807
    },
    {
      "epoch": 2.0197806149973028,
      "grad_norm": 8.484870302892674,
      "learning_rate": 2.7159208828080803e-06,
      "loss": 0.1617,
      "step": 2808
    },
    {
      "epoch": 2.020499910088114,
      "grad_norm": 3.8386481781464195,
      "learning_rate": 2.7157216107133414e-06,
      "loss": 0.0886,
      "step": 2809
    },
    {
      "epoch": 2.021219205178925,
      "grad_norm": 4.467874712096741,
      "learning_rate": 2.715522276067591e-06,
      "loss": 0.0824,
      "step": 2810
    },
    {
      "epoch": 2.021938500269736,
      "grad_norm": 3.659386606493036,
      "learning_rate": 2.7153228788810856e-06,
      "loss": 0.0457,
      "step": 2811
    },
    {
      "epoch": 2.0226577953605465,
      "grad_norm": 5.398311847914288,
      "learning_rate": 2.715123419164085e-06,
      "loss": 0.0655,
      "step": 2812
    },
    {
      "epoch": 2.0233770904513575,
      "grad_norm": 7.430266630975111,
      "learning_rate": 2.7149238969268513e-06,
      "loss": 0.0211,
      "step": 2813
    },
    {
      "epoch": 2.0240963855421685,
      "grad_norm": 6.480968913700203,
      "learning_rate": 2.7147243121796503e-06,
      "loss": 0.2189,
      "step": 2814
    },
    {
      "epoch": 2.0248156806329796,
      "grad_norm": 5.836683650324344,
      "learning_rate": 2.714524664932751e-06,
      "loss": 0.17,
      "step": 2815
    },
    {
      "epoch": 2.0255349757237906,
      "grad_norm": 3.9349897314803925,
      "learning_rate": 2.714324955196426e-06,
      "loss": 0.1472,
      "step": 2816
    },
    {
      "epoch": 2.0262542708146016,
      "grad_norm": 2.26787486431004,
      "learning_rate": 2.71412518298095e-06,
      "loss": 0.0173,
      "step": 2817
    },
    {
      "epoch": 2.0269735659054127,
      "grad_norm": 1.8047415051231606,
      "learning_rate": 2.713925348296602e-06,
      "loss": 0.0304,
      "step": 2818
    },
    {
      "epoch": 2.0276928609962237,
      "grad_norm": 4.461285807801138,
      "learning_rate": 2.713725451153664e-06,
      "loss": 0.1124,
      "step": 2819
    },
    {
      "epoch": 2.0284121560870347,
      "grad_norm": 2.7188520446413444,
      "learning_rate": 2.713525491562421e-06,
      "loss": 0.0862,
      "step": 2820
    },
    {
      "epoch": 2.0291314511778458,
      "grad_norm": 1.0976615378646268,
      "learning_rate": 2.7133254695331614e-06,
      "loss": 0.0028,
      "step": 2821
    },
    {
      "epoch": 2.029850746268657,
      "grad_norm": 4.15485904843733,
      "learning_rate": 2.713125385076176e-06,
      "loss": 0.1266,
      "step": 2822
    },
    {
      "epoch": 2.030570041359468,
      "grad_norm": 1.5880528038831108,
      "learning_rate": 2.71292523820176e-06,
      "loss": 0.0241,
      "step": 2823
    },
    {
      "epoch": 2.031289336450279,
      "grad_norm": 4.33536997293647,
      "learning_rate": 2.7127250289202115e-06,
      "loss": 0.0895,
      "step": 2824
    },
    {
      "epoch": 2.03200863154109,
      "grad_norm": 2.06688137916887,
      "learning_rate": 2.7125247572418315e-06,
      "loss": 0.0084,
      "step": 2825
    },
    {
      "epoch": 2.032727926631901,
      "grad_norm": 2.2942832983499306,
      "learning_rate": 2.712324423176924e-06,
      "loss": 0.1034,
      "step": 2826
    },
    {
      "epoch": 2.0334472217227115,
      "grad_norm": 5.819125177481963,
      "learning_rate": 2.7121240267357967e-06,
      "loss": 0.1182,
      "step": 2827
    },
    {
      "epoch": 2.0341665168135226,
      "grad_norm": 1.4806506937539476,
      "learning_rate": 2.7119235679287605e-06,
      "loss": 0.0106,
      "step": 2828
    },
    {
      "epoch": 2.0348858119043336,
      "grad_norm": 3.448025879752963,
      "learning_rate": 2.711723046766129e-06,
      "loss": 0.1303,
      "step": 2829
    },
    {
      "epoch": 2.0356051069951446,
      "grad_norm": 1.3952158591504202,
      "learning_rate": 2.71152246325822e-06,
      "loss": 0.0207,
      "step": 2830
    },
    {
      "epoch": 2.0363244020859557,
      "grad_norm": 5.444168632996188,
      "learning_rate": 2.7113218174153537e-06,
      "loss": 0.3073,
      "step": 2831
    },
    {
      "epoch": 2.0370436971767667,
      "grad_norm": 0.14657342091201225,
      "learning_rate": 2.711121109247853e-06,
      "loss": 0.0005,
      "step": 2832
    },
    {
      "epoch": 2.0377629922675777,
      "grad_norm": 2.1574770295331027,
      "learning_rate": 2.710920338766045e-06,
      "loss": 0.0504,
      "step": 2833
    },
    {
      "epoch": 2.0384822873583888,
      "grad_norm": 3.115014121264926,
      "learning_rate": 2.7107195059802603e-06,
      "loss": 0.0424,
      "step": 2834
    },
    {
      "epoch": 2.0392015824492,
      "grad_norm": 3.994109904266133,
      "learning_rate": 2.710518610900832e-06,
      "loss": 0.0948,
      "step": 2835
    },
    {
      "epoch": 2.039920877540011,
      "grad_norm": 7.230308004929196,
      "learning_rate": 2.710317653538095e-06,
      "loss": 0.0622,
      "step": 2836
    },
    {
      "epoch": 2.040640172630822,
      "grad_norm": 7.943485615827747,
      "learning_rate": 2.710116633902391e-06,
      "loss": 0.1657,
      "step": 2837
    },
    {
      "epoch": 2.041359467721633,
      "grad_norm": 2.4278652976648223,
      "learning_rate": 2.709915552004062e-06,
      "loss": 0.0511,
      "step": 2838
    },
    {
      "epoch": 2.042078762812444,
      "grad_norm": 3.966996139087217,
      "learning_rate": 2.709714407853454e-06,
      "loss": 0.1563,
      "step": 2839
    },
    {
      "epoch": 2.042798057903255,
      "grad_norm": 4.267452681601579,
      "learning_rate": 2.709513201460915e-06,
      "loss": 0.0514,
      "step": 2840
    },
    {
      "epoch": 2.043517352994066,
      "grad_norm": 2.0192790206568336,
      "learning_rate": 2.7093119328367997e-06,
      "loss": 0.0546,
      "step": 2841
    },
    {
      "epoch": 2.044236648084877,
      "grad_norm": 2.912353301984279,
      "learning_rate": 2.709110601991462e-06,
      "loss": 0.0456,
      "step": 2842
    },
    {
      "epoch": 2.0449559431756876,
      "grad_norm": 4.978384529183961,
      "learning_rate": 2.708909208935262e-06,
      "loss": 0.1326,
      "step": 2843
    },
    {
      "epoch": 2.0456752382664987,
      "grad_norm": 3.331634584450944,
      "learning_rate": 2.70870775367856e-06,
      "loss": 0.1528,
      "step": 2844
    },
    {
      "epoch": 2.0463945333573097,
      "grad_norm": 4.449390741492378,
      "learning_rate": 2.708506236231723e-06,
      "loss": 0.1946,
      "step": 2845
    },
    {
      "epoch": 2.0471138284481207,
      "grad_norm": 5.246598072230538,
      "learning_rate": 2.7083046566051186e-06,
      "loss": 0.1447,
      "step": 2846
    },
    {
      "epoch": 2.0478331235389318,
      "grad_norm": 2.644773198980077,
      "learning_rate": 2.7081030148091177e-06,
      "loss": 0.076,
      "step": 2847
    },
    {
      "epoch": 2.048552418629743,
      "grad_norm": 7.170148536840793,
      "learning_rate": 2.707901310854097e-06,
      "loss": 0.069,
      "step": 2848
    },
    {
      "epoch": 2.049271713720554,
      "grad_norm": 6.145849778851321,
      "learning_rate": 2.707699544750433e-06,
      "loss": 0.132,
      "step": 2849
    },
    {
      "epoch": 2.049991008811365,
      "grad_norm": 2.543005988948136,
      "learning_rate": 2.7074977165085074e-06,
      "loss": 0.0803,
      "step": 2850
    },
    {
      "epoch": 2.050710303902176,
      "grad_norm": 1.735885549214208,
      "learning_rate": 2.7072958261387044e-06,
      "loss": 0.0249,
      "step": 2851
    },
    {
      "epoch": 2.051429598992987,
      "grad_norm": 2.588109804804989,
      "learning_rate": 2.707093873651412e-06,
      "loss": 0.0765,
      "step": 2852
    },
    {
      "epoch": 2.052148894083798,
      "grad_norm": 2.663709577313606,
      "learning_rate": 2.706891859057021e-06,
      "loss": 0.0889,
      "step": 2853
    },
    {
      "epoch": 2.052868189174609,
      "grad_norm": 5.4433859508191516,
      "learning_rate": 2.706689782365925e-06,
      "loss": 0.1637,
      "step": 2854
    },
    {
      "epoch": 2.05358748426542,
      "grad_norm": 6.236158105254232,
      "learning_rate": 2.7064876435885215e-06,
      "loss": 0.0482,
      "step": 2855
    },
    {
      "epoch": 2.054306779356231,
      "grad_norm": 6.520916861143226,
      "learning_rate": 2.7062854427352108e-06,
      "loss": 0.0675,
      "step": 2856
    },
    {
      "epoch": 2.055026074447042,
      "grad_norm": 2.9841237110257652,
      "learning_rate": 2.7060831798163965e-06,
      "loss": 0.0272,
      "step": 2857
    },
    {
      "epoch": 2.0557453695378527,
      "grad_norm": 3.461535408529019,
      "learning_rate": 2.7058808548424852e-06,
      "loss": 0.1067,
      "step": 2858
    },
    {
      "epoch": 2.0564646646286637,
      "grad_norm": 7.1922968820156745,
      "learning_rate": 2.705678467823887e-06,
      "loss": 0.16,
      "step": 2859
    },
    {
      "epoch": 2.0571839597194748,
      "grad_norm": 2.802629225891705,
      "learning_rate": 2.7054760187710158e-06,
      "loss": 0.0514,
      "step": 2860
    },
    {
      "epoch": 2.057903254810286,
      "grad_norm": 2.645477453293368,
      "learning_rate": 2.7052735076942865e-06,
      "loss": 0.0393,
      "step": 2861
    },
    {
      "epoch": 2.058622549901097,
      "grad_norm": 3.9603260287374034,
      "learning_rate": 2.70507093460412e-06,
      "loss": 0.0168,
      "step": 2862
    },
    {
      "epoch": 2.059341844991908,
      "grad_norm": 6.664075058381859,
      "learning_rate": 2.7048682995109383e-06,
      "loss": 0.1383,
      "step": 2863
    },
    {
      "epoch": 2.060061140082719,
      "grad_norm": 4.179636360368768,
      "learning_rate": 2.7046656024251674e-06,
      "loss": 0.0505,
      "step": 2864
    },
    {
      "epoch": 2.06078043517353,
      "grad_norm": 1.6382320499211176,
      "learning_rate": 2.7044628433572367e-06,
      "loss": 0.0414,
      "step": 2865
    },
    {
      "epoch": 2.061499730264341,
      "grad_norm": 4.204107340462218,
      "learning_rate": 2.704260022317578e-06,
      "loss": 0.0476,
      "step": 2866
    },
    {
      "epoch": 2.062219025355152,
      "grad_norm": 2.9783997258046764,
      "learning_rate": 2.7040571393166274e-06,
      "loss": 0.0469,
      "step": 2867
    },
    {
      "epoch": 2.062938320445963,
      "grad_norm": 6.045178140926529,
      "learning_rate": 2.703854194364823e-06,
      "loss": 0.1211,
      "step": 2868
    },
    {
      "epoch": 2.063657615536774,
      "grad_norm": 0.20845555049375436,
      "learning_rate": 2.7036511874726077e-06,
      "loss": 0.0003,
      "step": 2869
    },
    {
      "epoch": 2.064376910627585,
      "grad_norm": 5.056599188113428,
      "learning_rate": 2.7034481186504253e-06,
      "loss": 0.0934,
      "step": 2870
    },
    {
      "epoch": 2.065096205718396,
      "grad_norm": 5.7650915892466745,
      "learning_rate": 2.703244987908725e-06,
      "loss": 0.1872,
      "step": 2871
    },
    {
      "epoch": 2.065815500809207,
      "grad_norm": 2.3393772680989757,
      "learning_rate": 2.7030417952579574e-06,
      "loss": 0.0714,
      "step": 2872
    },
    {
      "epoch": 2.0665347959000178,
      "grad_norm": 3.6000945563863676,
      "learning_rate": 2.7028385407085776e-06,
      "loss": 0.0805,
      "step": 2873
    },
    {
      "epoch": 2.067254090990829,
      "grad_norm": 5.535896920535353,
      "learning_rate": 2.7026352242710434e-06,
      "loss": 0.0992,
      "step": 2874
    },
    {
      "epoch": 2.06797338608164,
      "grad_norm": 1.7338699960326092,
      "learning_rate": 2.702431845955816e-06,
      "loss": 0.012,
      "step": 2875
    },
    {
      "epoch": 2.068692681172451,
      "grad_norm": 6.501831485576869,
      "learning_rate": 2.70222840577336e-06,
      "loss": 0.2113,
      "step": 2876
    },
    {
      "epoch": 2.069411976263262,
      "grad_norm": 7.530651501944735,
      "learning_rate": 2.7020249037341408e-06,
      "loss": 0.0757,
      "step": 2877
    },
    {
      "epoch": 2.070131271354073,
      "grad_norm": 5.358410037982092,
      "learning_rate": 2.7018213398486306e-06,
      "loss": 0.1405,
      "step": 2878
    },
    {
      "epoch": 2.070850566444884,
      "grad_norm": 2.5619082870532446,
      "learning_rate": 2.7016177141273025e-06,
      "loss": 0.091,
      "step": 2879
    },
    {
      "epoch": 2.071569861535695,
      "grad_norm": 0.05927888327756036,
      "learning_rate": 2.7014140265806347e-06,
      "loss": 0.0003,
      "step": 2880
    },
    {
      "epoch": 2.072289156626506,
      "grad_norm": 2.657744763327062,
      "learning_rate": 2.701210277219105e-06,
      "loss": 0.1127,
      "step": 2881
    },
    {
      "epoch": 2.073008451717317,
      "grad_norm": 4.855462734285935,
      "learning_rate": 2.7010064660531988e-06,
      "loss": 0.1165,
      "step": 2882
    },
    {
      "epoch": 2.073727746808128,
      "grad_norm": 3.299968811696358,
      "learning_rate": 2.700802593093401e-06,
      "loss": 0.1322,
      "step": 2883
    },
    {
      "epoch": 2.074447041898939,
      "grad_norm": 2.713412676740052,
      "learning_rate": 2.700598658350202e-06,
      "loss": 0.0076,
      "step": 2884
    },
    {
      "epoch": 2.07516633698975,
      "grad_norm": 4.071712260648528,
      "learning_rate": 2.7003946618340945e-06,
      "loss": 0.1469,
      "step": 2885
    },
    {
      "epoch": 2.075885632080561,
      "grad_norm": 1.448805845567579,
      "learning_rate": 2.7001906035555747e-06,
      "loss": 0.0037,
      "step": 2886
    },
    {
      "epoch": 2.0766049271713722,
      "grad_norm": 4.710089331706898,
      "learning_rate": 2.6999864835251407e-06,
      "loss": 0.0261,
      "step": 2887
    },
    {
      "epoch": 2.0773242222621833,
      "grad_norm": 3.3906585116952757,
      "learning_rate": 2.6997823017532965e-06,
      "loss": 0.0049,
      "step": 2888
    },
    {
      "epoch": 2.078043517352994,
      "grad_norm": 4.974308262357309,
      "learning_rate": 2.699578058250546e-06,
      "loss": 0.1814,
      "step": 2889
    },
    {
      "epoch": 2.078762812443805,
      "grad_norm": 1.9709754085523008,
      "learning_rate": 2.6993737530273993e-06,
      "loss": 0.0159,
      "step": 2890
    },
    {
      "epoch": 2.079482107534616,
      "grad_norm": 3.7408653619798278,
      "learning_rate": 2.6991693860943667e-06,
      "loss": 0.1133,
      "step": 2891
    },
    {
      "epoch": 2.080201402625427,
      "grad_norm": 2.935382699602907,
      "learning_rate": 2.6989649574619647e-06,
      "loss": 0.0218,
      "step": 2892
    },
    {
      "epoch": 2.080920697716238,
      "grad_norm": 4.0281844640731705,
      "learning_rate": 2.698760467140711e-06,
      "loss": 0.0483,
      "step": 2893
    },
    {
      "epoch": 2.081639992807049,
      "grad_norm": 4.064628009452115,
      "learning_rate": 2.6985559151411264e-06,
      "loss": 0.0795,
      "step": 2894
    },
    {
      "epoch": 2.08235928789786,
      "grad_norm": 2.501698238278197,
      "learning_rate": 2.6983513014737363e-06,
      "loss": 0.0288,
      "step": 2895
    },
    {
      "epoch": 2.083078582988671,
      "grad_norm": 2.6206935825263833,
      "learning_rate": 2.6981466261490685e-06,
      "loss": 0.0106,
      "step": 2896
    },
    {
      "epoch": 2.083797878079482,
      "grad_norm": 2.0805132327356044,
      "learning_rate": 2.697941889177653e-06,
      "loss": 0.0581,
      "step": 2897
    },
    {
      "epoch": 2.084517173170293,
      "grad_norm": 0.7818793583257235,
      "learning_rate": 2.6977370905700245e-06,
      "loss": 0.0117,
      "step": 2898
    },
    {
      "epoch": 2.085236468261104,
      "grad_norm": 4.311638089214096,
      "learning_rate": 2.6975322303367204e-06,
      "loss": 0.1123,
      "step": 2899
    },
    {
      "epoch": 2.0859557633519152,
      "grad_norm": 4.139080905419962,
      "learning_rate": 2.69732730848828e-06,
      "loss": 0.0907,
      "step": 2900
    },
    {
      "epoch": 2.0866750584427263,
      "grad_norm": 2.214659225866576,
      "learning_rate": 2.6971223250352484e-06,
      "loss": 0.0411,
      "step": 2901
    },
    {
      "epoch": 2.0873943535335373,
      "grad_norm": 2.413139243340381,
      "learning_rate": 2.696917279988172e-06,
      "loss": 0.0625,
      "step": 2902
    },
    {
      "epoch": 2.0881136486243483,
      "grad_norm": 4.498754842960439,
      "learning_rate": 2.6967121733576e-06,
      "loss": 0.1444,
      "step": 2903
    },
    {
      "epoch": 2.088832943715159,
      "grad_norm": 1.036526231725358,
      "learning_rate": 2.6965070051540857e-06,
      "loss": 0.0033,
      "step": 2904
    },
    {
      "epoch": 2.08955223880597,
      "grad_norm": 4.776313354805104,
      "learning_rate": 2.696301775388186e-06,
      "loss": 0.2467,
      "step": 2905
    },
    {
      "epoch": 2.090271533896781,
      "grad_norm": 2.52415644235071,
      "learning_rate": 2.6960964840704594e-06,
      "loss": 0.0637,
      "step": 2906
    },
    {
      "epoch": 2.090990828987592,
      "grad_norm": 2.436682772121266,
      "learning_rate": 2.6958911312114694e-06,
      "loss": 0.0255,
      "step": 2907
    },
    {
      "epoch": 2.091710124078403,
      "grad_norm": 1.4429705263965487,
      "learning_rate": 2.6956857168217814e-06,
      "loss": 0.0049,
      "step": 2908
    },
    {
      "epoch": 2.092429419169214,
      "grad_norm": 2.859501605564474,
      "learning_rate": 2.695480240911964e-06,
      "loss": 0.021,
      "step": 2909
    },
    {
      "epoch": 2.093148714260025,
      "grad_norm": 5.1537380153314665,
      "learning_rate": 2.6952747034925904e-06,
      "loss": 0.2112,
      "step": 2910
    },
    {
      "epoch": 2.093868009350836,
      "grad_norm": 5.855738569645706,
      "learning_rate": 2.695069104574234e-06,
      "loss": 0.1484,
      "step": 2911
    },
    {
      "epoch": 2.094587304441647,
      "grad_norm": 0.25332351452389795,
      "learning_rate": 2.694863444167475e-06,
      "loss": 0.0013,
      "step": 2912
    },
    {
      "epoch": 2.0953065995324582,
      "grad_norm": 3.434302414182239,
      "learning_rate": 2.6946577222828937e-06,
      "loss": 0.0876,
      "step": 2913
    },
    {
      "epoch": 2.0960258946232693,
      "grad_norm": 4.086544533580528,
      "learning_rate": 2.6944519389310765e-06,
      "loss": 0.1986,
      "step": 2914
    },
    {
      "epoch": 2.0967451897140803,
      "grad_norm": 3.8506262753492093,
      "learning_rate": 2.6942460941226094e-06,
      "loss": 0.2298,
      "step": 2915
    },
    {
      "epoch": 2.0974644848048913,
      "grad_norm": 3.3132662908177566,
      "learning_rate": 2.6940401878680844e-06,
      "loss": 0.0827,
      "step": 2916
    },
    {
      "epoch": 2.0981837798957024,
      "grad_norm": 2.4893188167297944,
      "learning_rate": 2.693834220178096e-06,
      "loss": 0.0741,
      "step": 2917
    },
    {
      "epoch": 2.0989030749865134,
      "grad_norm": 2.5766126043141147,
      "learning_rate": 2.693628191063241e-06,
      "loss": 0.0988,
      "step": 2918
    },
    {
      "epoch": 2.0996223700773244,
      "grad_norm": 4.93133336756768,
      "learning_rate": 2.69342210053412e-06,
      "loss": 0.0295,
      "step": 2919
    },
    {
      "epoch": 2.100341665168135,
      "grad_norm": 1.4705158230720918,
      "learning_rate": 2.6932159486013372e-06,
      "loss": 0.0377,
      "step": 2920
    },
    {
      "epoch": 2.101060960258946,
      "grad_norm": 2.93231780719383,
      "learning_rate": 2.693009735275499e-06,
      "loss": 0.1103,
      "step": 2921
    },
    {
      "epoch": 2.101780255349757,
      "grad_norm": 1.6751038900492348,
      "learning_rate": 2.692803460567216e-06,
      "loss": 0.008,
      "step": 2922
    },
    {
      "epoch": 2.102499550440568,
      "grad_norm": 3.541810755257078,
      "learning_rate": 2.6925971244871007e-06,
      "loss": 0.2007,
      "step": 2923
    },
    {
      "epoch": 2.103218845531379,
      "grad_norm": 6.288832017446734,
      "learning_rate": 2.69239072704577e-06,
      "loss": 0.2127,
      "step": 2924
    },
    {
      "epoch": 2.10393814062219,
      "grad_norm": 5.576877420818762,
      "learning_rate": 2.6921842682538434e-06,
      "loss": 0.0836,
      "step": 2925
    },
    {
      "epoch": 2.1046574357130012,
      "grad_norm": 3.670234693775757,
      "learning_rate": 2.691977748121943e-06,
      "loss": 0.0174,
      "step": 2926
    },
    {
      "epoch": 2.1053767308038123,
      "grad_norm": 5.309282768027773,
      "learning_rate": 2.6917711666606955e-06,
      "loss": 0.1346,
      "step": 2927
    },
    {
      "epoch": 2.1060960258946233,
      "grad_norm": 4.252061231370029,
      "learning_rate": 2.691564523880729e-06,
      "loss": 0.0807,
      "step": 2928
    },
    {
      "epoch": 2.1068153209854343,
      "grad_norm": 1.4268997965102455,
      "learning_rate": 2.6913578197926765e-06,
      "loss": 0.0402,
      "step": 2929
    },
    {
      "epoch": 2.1075346160762454,
      "grad_norm": 5.973370139577655,
      "learning_rate": 2.691151054407172e-06,
      "loss": 0.1377,
      "step": 2930
    },
    {
      "epoch": 2.1082539111670564,
      "grad_norm": 0.8930848275859191,
      "learning_rate": 2.6909442277348553e-06,
      "loss": 0.003,
      "step": 2931
    },
    {
      "epoch": 2.1089732062578674,
      "grad_norm": 4.657529562237361,
      "learning_rate": 2.6907373397863675e-06,
      "loss": 0.1874,
      "step": 2932
    },
    {
      "epoch": 2.1096925013486785,
      "grad_norm": 2.073690725354036,
      "learning_rate": 2.690530390572353e-06,
      "loss": 0.0582,
      "step": 2933
    },
    {
      "epoch": 2.1104117964394895,
      "grad_norm": 2.996825676239332,
      "learning_rate": 2.69032338010346e-06,
      "loss": 0.0522,
      "step": 2934
    },
    {
      "epoch": 2.1111310915303,
      "grad_norm": 6.869905188009063,
      "learning_rate": 2.6901163083903396e-06,
      "loss": 0.1191,
      "step": 2935
    },
    {
      "epoch": 2.111850386621111,
      "grad_norm": 3.2832506329399034,
      "learning_rate": 2.6899091754436464e-06,
      "loss": 0.057,
      "step": 2936
    },
    {
      "epoch": 2.112569681711922,
      "grad_norm": 1.9787105990508114,
      "learning_rate": 2.689701981274037e-06,
      "loss": 0.0085,
      "step": 2937
    },
    {
      "epoch": 2.113288976802733,
      "grad_norm": 1.7328058463020288,
      "learning_rate": 2.689494725892172e-06,
      "loss": 0.0075,
      "step": 2938
    },
    {
      "epoch": 2.1140082718935442,
      "grad_norm": 2.8948611029461935,
      "learning_rate": 2.6892874093087153e-06,
      "loss": 0.1409,
      "step": 2939
    },
    {
      "epoch": 2.1147275669843553,
      "grad_norm": 2.525295282714169,
      "learning_rate": 2.689080031534334e-06,
      "loss": 0.0769,
      "step": 2940
    },
    {
      "epoch": 2.1154468620751663,
      "grad_norm": 2.3096384712137965,
      "learning_rate": 2.6888725925796974e-06,
      "loss": 0.0807,
      "step": 2941
    },
    {
      "epoch": 2.1161661571659773,
      "grad_norm": 3.4804000809233635,
      "learning_rate": 2.6886650924554786e-06,
      "loss": 0.1043,
      "step": 2942
    },
    {
      "epoch": 2.1168854522567884,
      "grad_norm": 1.476342280066496,
      "learning_rate": 2.688457531172355e-06,
      "loss": 0.0109,
      "step": 2943
    },
    {
      "epoch": 2.1176047473475994,
      "grad_norm": 2.520978701311248,
      "learning_rate": 2.688249908741005e-06,
      "loss": 0.0507,
      "step": 2944
    },
    {
      "epoch": 2.1183240424384104,
      "grad_norm": 2.189522237803197,
      "learning_rate": 2.6880422251721104e-06,
      "loss": 0.0458,
      "step": 2945
    },
    {
      "epoch": 2.1190433375292215,
      "grad_norm": 0.9308178347925022,
      "learning_rate": 2.6878344804763587e-06,
      "loss": 0.0026,
      "step": 2946
    },
    {
      "epoch": 2.1197626326200325,
      "grad_norm": 6.920025212568776,
      "learning_rate": 2.687626674664437e-06,
      "loss": 0.1925,
      "step": 2947
    },
    {
      "epoch": 2.1204819277108435,
      "grad_norm": 4.017538052314308,
      "learning_rate": 2.687418807747039e-06,
      "loss": 0.1184,
      "step": 2948
    },
    {
      "epoch": 2.1212012228016546,
      "grad_norm": 0.6984131718619034,
      "learning_rate": 2.6872108797348583e-06,
      "loss": 0.0073,
      "step": 2949
    },
    {
      "epoch": 2.121920517892465,
      "grad_norm": 6.485046510514252,
      "learning_rate": 2.6870028906385938e-06,
      "loss": 0.0652,
      "step": 2950
    },
    {
      "epoch": 2.122639812983276,
      "grad_norm": 3.9208748495925905,
      "learning_rate": 2.686794840468947e-06,
      "loss": 0.1427,
      "step": 2951
    },
    {
      "epoch": 2.1233591080740872,
      "grad_norm": 1.018764436705655,
      "learning_rate": 2.6865867292366223e-06,
      "loss": 0.0015,
      "step": 2952
    },
    {
      "epoch": 2.1240784031648983,
      "grad_norm": 4.228581355661939,
      "learning_rate": 2.686378556952327e-06,
      "loss": 0.1115,
      "step": 2953
    },
    {
      "epoch": 2.1247976982557093,
      "grad_norm": 4.077305115319506,
      "learning_rate": 2.686170323626773e-06,
      "loss": 0.0834,
      "step": 2954
    },
    {
      "epoch": 2.1255169933465203,
      "grad_norm": 5.037579660156506,
      "learning_rate": 2.685962029270673e-06,
      "loss": 0.0277,
      "step": 2955
    },
    {
      "epoch": 2.1262362884373314,
      "grad_norm": 6.254952615196554,
      "learning_rate": 2.685753673894745e-06,
      "loss": 0.2789,
      "step": 2956
    },
    {
      "epoch": 2.1269555835281424,
      "grad_norm": 5.266516029647939,
      "learning_rate": 2.6855452575097094e-06,
      "loss": 0.0862,
      "step": 2957
    },
    {
      "epoch": 2.1276748786189534,
      "grad_norm": 5.512030801731916,
      "learning_rate": 2.6853367801262883e-06,
      "loss": 0.0492,
      "step": 2958
    },
    {
      "epoch": 2.1283941737097645,
      "grad_norm": 3.7268551741164258,
      "learning_rate": 2.6851282417552093e-06,
      "loss": 0.1145,
      "step": 2959
    },
    {
      "epoch": 2.1291134688005755,
      "grad_norm": 3.8324970526459134,
      "learning_rate": 2.6849196424072024e-06,
      "loss": 0.0984,
      "step": 2960
    },
    {
      "epoch": 2.1298327638913865,
      "grad_norm": 3.495163608649748,
      "learning_rate": 2.6847109820929993e-06,
      "loss": 0.0758,
      "step": 2961
    },
    {
      "epoch": 2.1305520589821976,
      "grad_norm": 3.259665676266259,
      "learning_rate": 2.6845022608233364e-06,
      "loss": 0.1327,
      "step": 2962
    },
    {
      "epoch": 2.1312713540730086,
      "grad_norm": 10.163836273929322,
      "learning_rate": 2.684293478608953e-06,
      "loss": 0.0334,
      "step": 2963
    },
    {
      "epoch": 2.1319906491638196,
      "grad_norm": 2.3310650358435003,
      "learning_rate": 2.6840846354605917e-06,
      "loss": 0.063,
      "step": 2964
    },
    {
      "epoch": 2.1327099442546302,
      "grad_norm": 5.0634793593735115,
      "learning_rate": 2.683875731388996e-06,
      "loss": 0.0674,
      "step": 2965
    },
    {
      "epoch": 2.1334292393454413,
      "grad_norm": 6.464794535905222,
      "learning_rate": 2.683666766404917e-06,
      "loss": 0.0299,
      "step": 2966
    },
    {
      "epoch": 2.1341485344362523,
      "grad_norm": 5.249986535858385,
      "learning_rate": 2.6834577405191044e-06,
      "loss": 0.1801,
      "step": 2967
    },
    {
      "epoch": 2.1348678295270633,
      "grad_norm": 4.074269595258422,
      "learning_rate": 2.683248653742313e-06,
      "loss": 0.0764,
      "step": 2968
    },
    {
      "epoch": 2.1355871246178744,
      "grad_norm": 4.282602739228891,
      "learning_rate": 2.6830395060853025e-06,
      "loss": 0.1766,
      "step": 2969
    },
    {
      "epoch": 2.1363064197086854,
      "grad_norm": 4.918205245332558,
      "learning_rate": 2.682830297558832e-06,
      "loss": 0.0471,
      "step": 2970
    },
    {
      "epoch": 2.1370257147994964,
      "grad_norm": 4.4657728485742725,
      "learning_rate": 2.6826210281736663e-06,
      "loss": 0.1888,
      "step": 2971
    },
    {
      "epoch": 2.1377450098903075,
      "grad_norm": 2.846559633436882,
      "learning_rate": 2.682411697940573e-06,
      "loss": 0.0919,
      "step": 2972
    },
    {
      "epoch": 2.1384643049811185,
      "grad_norm": 5.699903429233865,
      "learning_rate": 2.6822023068703217e-06,
      "loss": 0.1778,
      "step": 2973
    },
    {
      "epoch": 2.1391836000719295,
      "grad_norm": 7.0843882964182585,
      "learning_rate": 2.6819928549736864e-06,
      "loss": 0.0849,
      "step": 2974
    },
    {
      "epoch": 2.1399028951627406,
      "grad_norm": 0.6260016601724464,
      "learning_rate": 2.681783342261444e-06,
      "loss": 0.0018,
      "step": 2975
    },
    {
      "epoch": 2.1406221902535516,
      "grad_norm": 3.889841875582308,
      "learning_rate": 2.6815737687443743e-06,
      "loss": 0.1055,
      "step": 2976
    },
    {
      "epoch": 2.1413414853443626,
      "grad_norm": 2.6137391227573405,
      "learning_rate": 2.68136413443326e-06,
      "loss": 0.0577,
      "step": 2977
    },
    {
      "epoch": 2.1420607804351737,
      "grad_norm": 2.612490229406799,
      "learning_rate": 2.6811544393388867e-06,
      "loss": 0.0384,
      "step": 2978
    },
    {
      "epoch": 2.1427800755259847,
      "grad_norm": 4.29215554513666,
      "learning_rate": 2.6809446834720443e-06,
      "loss": 0.1313,
      "step": 2979
    },
    {
      "epoch": 2.1434993706167957,
      "grad_norm": 2.17004453287081,
      "learning_rate": 2.680734866843525e-06,
      "loss": 0.0502,
      "step": 2980
    },
    {
      "epoch": 2.1442186657076068,
      "grad_norm": 3.3945073452656906,
      "learning_rate": 2.680524989464124e-06,
      "loss": 0.1423,
      "step": 2981
    },
    {
      "epoch": 2.1449379607984174,
      "grad_norm": 5.202709582701237,
      "learning_rate": 2.68031505134464e-06,
      "loss": 0.2498,
      "step": 2982
    },
    {
      "epoch": 2.1456572558892284,
      "grad_norm": 8.065446627017817,
      "learning_rate": 2.6801050524958743e-06,
      "loss": 0.0606,
      "step": 2983
    },
    {
      "epoch": 2.1463765509800394,
      "grad_norm": 2.4804218657275574,
      "learning_rate": 2.6798949929286323e-06,
      "loss": 0.0246,
      "step": 2984
    },
    {
      "epoch": 2.1470958460708505,
      "grad_norm": 10.265412244776828,
      "learning_rate": 2.679684872653722e-06,
      "loss": 0.1096,
      "step": 2985
    },
    {
      "epoch": 2.1478151411616615,
      "grad_norm": 0.738843136816604,
      "learning_rate": 2.6794746916819536e-06,
      "loss": 0.0026,
      "step": 2986
    },
    {
      "epoch": 2.1485344362524725,
      "grad_norm": 5.243622357570583,
      "learning_rate": 2.6792644500241416e-06,
      "loss": 0.1345,
      "step": 2987
    },
    {
      "epoch": 2.1492537313432836,
      "grad_norm": 10.341664595677095,
      "learning_rate": 2.679054147691104e-06,
      "loss": 0.1395,
      "step": 2988
    },
    {
      "epoch": 2.1499730264340946,
      "grad_norm": 3.1635200937070285,
      "learning_rate": 2.6788437846936608e-06,
      "loss": 0.1319,
      "step": 2989
    },
    {
      "epoch": 2.1506923215249056,
      "grad_norm": 1.5905810658491417,
      "learning_rate": 2.6786333610426353e-06,
      "loss": 0.0523,
      "step": 2990
    },
    {
      "epoch": 2.1514116166157167,
      "grad_norm": 4.24920358932737,
      "learning_rate": 2.6784228767488542e-06,
      "loss": 0.0663,
      "step": 2991
    },
    {
      "epoch": 2.1521309117065277,
      "grad_norm": 0.11900852508119229,
      "learning_rate": 2.6782123318231474e-06,
      "loss": 0.0004,
      "step": 2992
    },
    {
      "epoch": 2.1528502067973387,
      "grad_norm": 3.999618463647315,
      "learning_rate": 2.678001726276348e-06,
      "loss": 0.153,
      "step": 2993
    },
    {
      "epoch": 2.1535695018881498,
      "grad_norm": 2.490362701481361,
      "learning_rate": 2.677791060119292e-06,
      "loss": 0.0341,
      "step": 2994
    },
    {
      "epoch": 2.154288796978961,
      "grad_norm": 1.6611576826425836,
      "learning_rate": 2.6775803333628186e-06,
      "loss": 0.0213,
      "step": 2995
    },
    {
      "epoch": 2.155008092069772,
      "grad_norm": 5.226548229925101,
      "learning_rate": 2.6773695460177697e-06,
      "loss": 0.0671,
      "step": 2996
    },
    {
      "epoch": 2.1557273871605824,
      "grad_norm": 5.062231585762362,
      "learning_rate": 2.677158698094991e-06,
      "loss": 0.2191,
      "step": 2997
    },
    {
      "epoch": 2.1564466822513935,
      "grad_norm": 4.864395184930188,
      "learning_rate": 2.67694778960533e-06,
      "loss": 0.0969,
      "step": 2998
    },
    {
      "epoch": 2.1571659773422045,
      "grad_norm": 3.5275070482483373,
      "learning_rate": 2.67673682055964e-06,
      "loss": 0.0154,
      "step": 2999
    },
    {
      "epoch": 2.1578852724330155,
      "grad_norm": 0.8059943215117032,
      "learning_rate": 2.6765257909687743e-06,
      "loss": 0.0085,
      "step": 3000
    },
    {
      "epoch": 2.1586045675238266,
      "grad_norm": 0.29667075068835635,
      "learning_rate": 2.676314700843592e-06,
      "loss": 0.0011,
      "step": 3001
    },
    {
      "epoch": 2.1593238626146376,
      "grad_norm": 3.5364574232061226,
      "learning_rate": 2.676103550194953e-06,
      "loss": 0.0973,
      "step": 3002
    },
    {
      "epoch": 2.1600431577054486,
      "grad_norm": 4.584638837921622,
      "learning_rate": 2.6758923390337216e-06,
      "loss": 0.0784,
      "step": 3003
    },
    {
      "epoch": 2.1607624527962597,
      "grad_norm": 3.3382879608375604,
      "learning_rate": 2.6756810673707655e-06,
      "loss": 0.0122,
      "step": 3004
    },
    {
      "epoch": 2.1614817478870707,
      "grad_norm": 0.07690135658984816,
      "learning_rate": 2.6754697352169544e-06,
      "loss": 0.0003,
      "step": 3005
    },
    {
      "epoch": 2.1622010429778817,
      "grad_norm": 2.276349708864349,
      "learning_rate": 2.675258342583162e-06,
      "loss": 0.0591,
      "step": 3006
    },
    {
      "epoch": 2.1629203380686928,
      "grad_norm": 2.167555930442845,
      "learning_rate": 2.675046889480265e-06,
      "loss": 0.0063,
      "step": 3007
    },
    {
      "epoch": 2.163639633159504,
      "grad_norm": 1.8130281675141482,
      "learning_rate": 2.6748353759191424e-06,
      "loss": 0.0403,
      "step": 3008
    },
    {
      "epoch": 2.164358928250315,
      "grad_norm": 3.878861015105422,
      "learning_rate": 2.6746238019106778e-06,
      "loss": 0.146,
      "step": 3009
    },
    {
      "epoch": 2.165078223341126,
      "grad_norm": 4.150130225843815,
      "learning_rate": 2.6744121674657563e-06,
      "loss": 0.0733,
      "step": 3010
    },
    {
      "epoch": 2.165797518431937,
      "grad_norm": 4.496257504023216,
      "learning_rate": 2.6742004725952674e-06,
      "loss": 0.1194,
      "step": 3011
    },
    {
      "epoch": 2.1665168135227475,
      "grad_norm": 5.020480889325239,
      "learning_rate": 2.6739887173101028e-06,
      "loss": 0.1292,
      "step": 3012
    },
    {
      "epoch": 2.1672361086135585,
      "grad_norm": 3.675056734361526,
      "learning_rate": 2.6737769016211577e-06,
      "loss": 0.0726,
      "step": 3013
    },
    {
      "epoch": 2.1679554037043696,
      "grad_norm": 4.212027705118157,
      "learning_rate": 2.6735650255393306e-06,
      "loss": 0.1671,
      "step": 3014
    },
    {
      "epoch": 2.1686746987951806,
      "grad_norm": 4.114358040075331,
      "learning_rate": 2.673353089075523e-06,
      "loss": 0.0312,
      "step": 3015
    },
    {
      "epoch": 2.1693939938859916,
      "grad_norm": 4.180423186928452,
      "learning_rate": 2.673141092240639e-06,
      "loss": 0.0428,
      "step": 3016
    },
    {
      "epoch": 2.1701132889768027,
      "grad_norm": 4.085663632187638,
      "learning_rate": 2.6729290350455863e-06,
      "loss": 0.1238,
      "step": 3017
    },
    {
      "epoch": 2.1708325840676137,
      "grad_norm": 4.640466564508529,
      "learning_rate": 2.672716917501276e-06,
      "loss": 0.1065,
      "step": 3018
    },
    {
      "epoch": 2.1715518791584247,
      "grad_norm": 2.5048164408670575,
      "learning_rate": 2.6725047396186214e-06,
      "loss": 0.0539,
      "step": 3019
    },
    {
      "epoch": 2.1722711742492358,
      "grad_norm": 3.509772459152732,
      "learning_rate": 2.6722925014085397e-06,
      "loss": 0.0838,
      "step": 3020
    },
    {
      "epoch": 2.172990469340047,
      "grad_norm": 4.819312111793526,
      "learning_rate": 2.6720802028819512e-06,
      "loss": 0.194,
      "step": 3021
    },
    {
      "epoch": 2.173709764430858,
      "grad_norm": 4.124789055153528,
      "learning_rate": 2.6718678440497785e-06,
      "loss": 0.0679,
      "step": 3022
    },
    {
      "epoch": 2.174429059521669,
      "grad_norm": 3.0456272847013452,
      "learning_rate": 2.671655424922948e-06,
      "loss": 0.062,
      "step": 3023
    },
    {
      "epoch": 2.17514835461248,
      "grad_norm": 2.45044484575865,
      "learning_rate": 2.6714429455123895e-06,
      "loss": 0.1028,
      "step": 3024
    },
    {
      "epoch": 2.175867649703291,
      "grad_norm": 4.288011935264229,
      "learning_rate": 2.6712304058290348e-06,
      "loss": 0.1486,
      "step": 3025
    },
    {
      "epoch": 2.176586944794102,
      "grad_norm": 5.760993852396619,
      "learning_rate": 2.6710178058838196e-06,
      "loss": 0.1088,
      "step": 3026
    },
    {
      "epoch": 2.1773062398849126,
      "grad_norm": 7.444748149105423,
      "learning_rate": 2.670805145687683e-06,
      "loss": 0.1549,
      "step": 3027
    },
    {
      "epoch": 2.1780255349757236,
      "grad_norm": 3.172223258702454,
      "learning_rate": 2.670592425251566e-06,
      "loss": 0.095,
      "step": 3028
    },
    {
      "epoch": 2.1787448300665346,
      "grad_norm": 4.424579500391891,
      "learning_rate": 2.6703796445864138e-06,
      "loss": 0.09,
      "step": 3029
    },
    {
      "epoch": 2.1794641251573457,
      "grad_norm": 4.079890933420054,
      "learning_rate": 2.6701668037031744e-06,
      "loss": 0.1316,
      "step": 3030
    },
    {
      "epoch": 2.1801834202481567,
      "grad_norm": 3.690180223968812,
      "learning_rate": 2.6699539026127992e-06,
      "loss": 0.0598,
      "step": 3031
    },
    {
      "epoch": 2.1809027153389677,
      "grad_norm": 2.137619410962381,
      "learning_rate": 2.669740941326241e-06,
      "loss": 0.0059,
      "step": 3032
    },
    {
      "epoch": 2.1816220104297788,
      "grad_norm": 2.269622753530593,
      "learning_rate": 2.669527919854459e-06,
      "loss": 0.0062,
      "step": 3033
    },
    {
      "epoch": 2.18234130552059,
      "grad_norm": 4.202944797807136,
      "learning_rate": 2.669314838208412e-06,
      "loss": 0.1886,
      "step": 3034
    },
    {
      "epoch": 2.183060600611401,
      "grad_norm": 4.811381399924753,
      "learning_rate": 2.6691016963990637e-06,
      "loss": 0.2848,
      "step": 3035
    },
    {
      "epoch": 2.183779895702212,
      "grad_norm": 0.7006711844653877,
      "learning_rate": 2.668888494437381e-06,
      "loss": 0.0038,
      "step": 3036
    },
    {
      "epoch": 2.184499190793023,
      "grad_norm": 2.08816362262734,
      "learning_rate": 2.6686752323343335e-06,
      "loss": 0.0536,
      "step": 3037
    },
    {
      "epoch": 2.185218485883834,
      "grad_norm": 3.292789418425388,
      "learning_rate": 2.6684619101008935e-06,
      "loss": 0.1126,
      "step": 3038
    },
    {
      "epoch": 2.185937780974645,
      "grad_norm": 1.6968916278192299,
      "learning_rate": 2.6682485277480373e-06,
      "loss": 0.0497,
      "step": 3039
    },
    {
      "epoch": 2.186657076065456,
      "grad_norm": 4.034804436079525,
      "learning_rate": 2.6680350852867433e-06,
      "loss": 0.096,
      "step": 3040
    },
    {
      "epoch": 2.187376371156267,
      "grad_norm": 4.226381220734043,
      "learning_rate": 2.6678215827279937e-06,
      "loss": 0.0218,
      "step": 3041
    },
    {
      "epoch": 2.1880956662470776,
      "grad_norm": 4.35835483225906,
      "learning_rate": 2.667608020082774e-06,
      "loss": 0.0774,
      "step": 3042
    },
    {
      "epoch": 2.1888149613378887,
      "grad_norm": 5.590528732088101,
      "learning_rate": 2.667394397362072e-06,
      "loss": 0.0916,
      "step": 3043
    },
    {
      "epoch": 2.1895342564286997,
      "grad_norm": 1.4990879729942823,
      "learning_rate": 2.6671807145768783e-06,
      "loss": 0.0036,
      "step": 3044
    },
    {
      "epoch": 2.1902535515195107,
      "grad_norm": 1.8407408914777634,
      "learning_rate": 2.6669669717381886e-06,
      "loss": 0.0091,
      "step": 3045
    },
    {
      "epoch": 2.1909728466103218,
      "grad_norm": 4.632695370374956,
      "learning_rate": 2.666753168856999e-06,
      "loss": 0.089,
      "step": 3046
    },
    {
      "epoch": 2.191692141701133,
      "grad_norm": 3.699219177846792,
      "learning_rate": 2.6665393059443117e-06,
      "loss": 0.1049,
      "step": 3047
    },
    {
      "epoch": 2.192411436791944,
      "grad_norm": 3.358600634471962,
      "learning_rate": 2.6663253830111284e-06,
      "loss": 0.0207,
      "step": 3048
    },
    {
      "epoch": 2.193130731882755,
      "grad_norm": 4.742007947914259,
      "learning_rate": 2.6661114000684572e-06,
      "loss": 0.0329,
      "step": 3049
    },
    {
      "epoch": 2.193850026973566,
      "grad_norm": 5.744341087408164,
      "learning_rate": 2.6658973571273076e-06,
      "loss": 0.1555,
      "step": 3050
    },
    {
      "epoch": 2.194569322064377,
      "grad_norm": 3.0015122433218164,
      "learning_rate": 2.6656832541986917e-06,
      "loss": 0.0614,
      "step": 3051
    },
    {
      "epoch": 2.195288617155188,
      "grad_norm": 2.8691462886200503,
      "learning_rate": 2.6654690912936264e-06,
      "loss": 0.1052,
      "step": 3052
    },
    {
      "epoch": 2.196007912245999,
      "grad_norm": 1.1360029414327333,
      "learning_rate": 2.6652548684231305e-06,
      "loss": 0.0065,
      "step": 3053
    },
    {
      "epoch": 2.19672720733681,
      "grad_norm": 3.603605197657369,
      "learning_rate": 2.665040585598226e-06,
      "loss": 0.0831,
      "step": 3054
    },
    {
      "epoch": 2.197446502427621,
      "grad_norm": 2.88806921822241,
      "learning_rate": 2.664826242829938e-06,
      "loss": 0.0119,
      "step": 3055
    },
    {
      "epoch": 2.198165797518432,
      "grad_norm": 2.1464758097691026,
      "learning_rate": 2.6646118401292954e-06,
      "loss": 0.0504,
      "step": 3056
    },
    {
      "epoch": 2.198885092609243,
      "grad_norm": 0.8742601408280847,
      "learning_rate": 2.6643973775073287e-06,
      "loss": 0.0024,
      "step": 3057
    },
    {
      "epoch": 2.199604387700054,
      "grad_norm": 3.1768981905761366,
      "learning_rate": 2.664182854975073e-06,
      "loss": 0.0191,
      "step": 3058
    },
    {
      "epoch": 2.2003236827908648,
      "grad_norm": 3.2981551100920825,
      "learning_rate": 2.6639682725435664e-06,
      "loss": 0.0743,
      "step": 3059
    },
    {
      "epoch": 2.201042977881676,
      "grad_norm": 5.2537328892294655,
      "learning_rate": 2.6637536302238483e-06,
      "loss": 0.2043,
      "step": 3060
    },
    {
      "epoch": 2.201762272972487,
      "grad_norm": 4.145852215231934,
      "learning_rate": 2.6635389280269634e-06,
      "loss": 0.1833,
      "step": 3061
    },
    {
      "epoch": 2.202481568063298,
      "grad_norm": 6.971997181248326,
      "learning_rate": 2.6633241659639577e-06,
      "loss": 0.0389,
      "step": 3062
    },
    {
      "epoch": 2.203200863154109,
      "grad_norm": 3.8223544162447047,
      "learning_rate": 2.6631093440458816e-06,
      "loss": 0.0119,
      "step": 3063
    },
    {
      "epoch": 2.20392015824492,
      "grad_norm": 0.6589239233339064,
      "learning_rate": 2.662894462283788e-06,
      "loss": 0.0013,
      "step": 3064
    },
    {
      "epoch": 2.204639453335731,
      "grad_norm": 4.302088979137138,
      "learning_rate": 2.6626795206887333e-06,
      "loss": 0.1031,
      "step": 3065
    },
    {
      "epoch": 2.205358748426542,
      "grad_norm": 3.447668234683055,
      "learning_rate": 2.6624645192717757e-06,
      "loss": 0.0361,
      "step": 3066
    },
    {
      "epoch": 2.206078043517353,
      "grad_norm": 2.8259633678433076,
      "learning_rate": 2.662249458043978e-06,
      "loss": 0.0498,
      "step": 3067
    },
    {
      "epoch": 2.206797338608164,
      "grad_norm": 3.1920687745161582,
      "learning_rate": 2.6620343370164062e-06,
      "loss": 0.0827,
      "step": 3068
    },
    {
      "epoch": 2.207516633698975,
      "grad_norm": 4.923688877228495,
      "learning_rate": 2.6618191562001273e-06,
      "loss": 0.0839,
      "step": 3069
    },
    {
      "epoch": 2.208235928789786,
      "grad_norm": 3.841778209790039,
      "learning_rate": 2.6616039156062134e-06,
      "loss": 0.1114,
      "step": 3070
    },
    {
      "epoch": 2.208955223880597,
      "grad_norm": 4.16621976504901,
      "learning_rate": 2.6613886152457386e-06,
      "loss": 0.1253,
      "step": 3071
    },
    {
      "epoch": 2.209674518971408,
      "grad_norm": 4.419159339269173,
      "learning_rate": 2.661173255129781e-06,
      "loss": 0.1242,
      "step": 3072
    },
    {
      "epoch": 2.2103938140622192,
      "grad_norm": 7.488103182362443,
      "learning_rate": 2.660957835269421e-06,
      "loss": 0.0641,
      "step": 3073
    },
    {
      "epoch": 2.21111310915303,
      "grad_norm": 5.214825466260846,
      "learning_rate": 2.6607423556757424e-06,
      "loss": 0.128,
      "step": 3074
    },
    {
      "epoch": 2.211832404243841,
      "grad_norm": 3.105561969429023,
      "learning_rate": 2.6605268163598324e-06,
      "loss": 0.0448,
      "step": 3075
    },
    {
      "epoch": 2.212551699334652,
      "grad_norm": 2.0604305855261233,
      "learning_rate": 2.6603112173327794e-06,
      "loss": 0.0519,
      "step": 3076
    },
    {
      "epoch": 2.213270994425463,
      "grad_norm": 1.6548731989231549,
      "learning_rate": 2.660095558605678e-06,
      "loss": 0.0163,
      "step": 3077
    },
    {
      "epoch": 2.213990289516274,
      "grad_norm": 3.4541999606719394,
      "learning_rate": 2.659879840189624e-06,
      "loss": 0.1401,
      "step": 3078
    },
    {
      "epoch": 2.214709584607085,
      "grad_norm": 2.824920314336351,
      "learning_rate": 2.6596640620957156e-06,
      "loss": 0.0244,
      "step": 3079
    },
    {
      "epoch": 2.215428879697896,
      "grad_norm": 2.7697266329639545,
      "learning_rate": 2.6594482243350557e-06,
      "loss": 0.0707,
      "step": 3080
    },
    {
      "epoch": 2.216148174788707,
      "grad_norm": 2.3062724635950773,
      "learning_rate": 2.6592323269187496e-06,
      "loss": 0.0069,
      "step": 3081
    },
    {
      "epoch": 2.216867469879518,
      "grad_norm": 6.030789082073365,
      "learning_rate": 2.659016369857905e-06,
      "loss": 0.1482,
      "step": 3082
    },
    {
      "epoch": 2.217586764970329,
      "grad_norm": 4.282264507857953,
      "learning_rate": 2.6588003531636334e-06,
      "loss": 0.0476,
      "step": 3083
    },
    {
      "epoch": 2.21830606006114,
      "grad_norm": 1.346627419906323,
      "learning_rate": 2.65858427684705e-06,
      "loss": 0.0108,
      "step": 3084
    },
    {
      "epoch": 2.219025355151951,
      "grad_norm": 3.611694539938345,
      "learning_rate": 2.658368140919271e-06,
      "loss": 0.0321,
      "step": 3085
    },
    {
      "epoch": 2.2197446502427622,
      "grad_norm": 4.206972357074755,
      "learning_rate": 2.658151945391418e-06,
      "loss": 0.1496,
      "step": 3086
    },
    {
      "epoch": 2.2204639453335733,
      "grad_norm": 4.308843457335962,
      "learning_rate": 2.6579356902746145e-06,
      "loss": 0.0401,
      "step": 3087
    },
    {
      "epoch": 2.2211832404243843,
      "grad_norm": 3.8243727547483046,
      "learning_rate": 2.657719375579987e-06,
      "loss": 0.0375,
      "step": 3088
    },
    {
      "epoch": 2.221902535515195,
      "grad_norm": 5.928819220465108,
      "learning_rate": 2.6575030013186654e-06,
      "loss": 0.0901,
      "step": 3089
    },
    {
      "epoch": 2.222621830606006,
      "grad_norm": 4.55346488043891,
      "learning_rate": 2.657286567501782e-06,
      "loss": 0.0099,
      "step": 3090
    },
    {
      "epoch": 2.223341125696817,
      "grad_norm": 4.499254796889424,
      "learning_rate": 2.6570700741404732e-06,
      "loss": 0.2184,
      "step": 3091
    },
    {
      "epoch": 2.224060420787628,
      "grad_norm": 2.6206496319399064,
      "learning_rate": 2.6568535212458786e-06,
      "loss": 0.108,
      "step": 3092
    },
    {
      "epoch": 2.224779715878439,
      "grad_norm": 4.782337007279096,
      "learning_rate": 2.6566369088291387e-06,
      "loss": 0.1914,
      "step": 3093
    },
    {
      "epoch": 2.22549901096925,
      "grad_norm": 1.1555587658614224,
      "learning_rate": 2.6564202369014e-06,
      "loss": 0.0161,
      "step": 3094
    },
    {
      "epoch": 2.226218306060061,
      "grad_norm": 2.9691056961547186,
      "learning_rate": 2.65620350547381e-06,
      "loss": 0.0614,
      "step": 3095
    },
    {
      "epoch": 2.226937601150872,
      "grad_norm": 0.9616482974102715,
      "learning_rate": 2.65598671455752e-06,
      "loss": 0.0186,
      "step": 3096
    },
    {
      "epoch": 2.227656896241683,
      "grad_norm": 1.8839980332932773,
      "learning_rate": 2.655769864163684e-06,
      "loss": 0.0245,
      "step": 3097
    },
    {
      "epoch": 2.228376191332494,
      "grad_norm": 2.4385583247630684,
      "learning_rate": 2.6555529543034596e-06,
      "loss": 0.0552,
      "step": 3098
    },
    {
      "epoch": 2.2290954864233052,
      "grad_norm": 6.735281310814383,
      "learning_rate": 2.6553359849880075e-06,
      "loss": 0.1087,
      "step": 3099
    },
    {
      "epoch": 2.2298147815141163,
      "grad_norm": 2.548770476578207,
      "learning_rate": 2.6551189562284905e-06,
      "loss": 0.0893,
      "step": 3100
    },
    {
      "epoch": 2.2305340766049273,
      "grad_norm": 3.155652151145516,
      "learning_rate": 2.6549018680360758e-06,
      "loss": 0.0677,
      "step": 3101
    },
    {
      "epoch": 2.2312533716957383,
      "grad_norm": 2.1949546512118174,
      "learning_rate": 2.6546847204219323e-06,
      "loss": 0.0368,
      "step": 3102
    },
    {
      "epoch": 2.2319726667865494,
      "grad_norm": 4.9440348416183575,
      "learning_rate": 2.6544675133972334e-06,
      "loss": 0.1895,
      "step": 3103
    },
    {
      "epoch": 2.23269196187736,
      "grad_norm": 5.463020706055418,
      "learning_rate": 2.654250246973154e-06,
      "loss": 0.1279,
      "step": 3104
    },
    {
      "epoch": 2.233411256968171,
      "grad_norm": 1.436102030721381,
      "learning_rate": 2.654032921160873e-06,
      "loss": 0.0059,
      "step": 3105
    },
    {
      "epoch": 2.234130552058982,
      "grad_norm": 2.829540367633825,
      "learning_rate": 2.6538155359715727e-06,
      "loss": 0.1505,
      "step": 3106
    },
    {
      "epoch": 2.234849847149793,
      "grad_norm": 3.928851933506906,
      "learning_rate": 2.6535980914164377e-06,
      "loss": 0.1376,
      "step": 3107
    },
    {
      "epoch": 2.235569142240604,
      "grad_norm": 2.770296962778117,
      "learning_rate": 2.6533805875066556e-06,
      "loss": 0.0936,
      "step": 3108
    },
    {
      "epoch": 2.236288437331415,
      "grad_norm": 1.0681266078493197,
      "learning_rate": 2.6531630242534174e-06,
      "loss": 0.0254,
      "step": 3109
    },
    {
      "epoch": 2.237007732422226,
      "grad_norm": 0.030510104157994884,
      "learning_rate": 2.6529454016679175e-06,
      "loss": 0.0002,
      "step": 3110
    },
    {
      "epoch": 2.237727027513037,
      "grad_norm": 5.551385749463073,
      "learning_rate": 2.6527277197613525e-06,
      "loss": 0.3379,
      "step": 3111
    },
    {
      "epoch": 2.2384463226038482,
      "grad_norm": 1.3553843636496001,
      "learning_rate": 2.6525099785449234e-06,
      "loss": 0.0035,
      "step": 3112
    },
    {
      "epoch": 2.2391656176946593,
      "grad_norm": 5.718011817443443,
      "learning_rate": 2.6522921780298323e-06,
      "loss": 0.1444,
      "step": 3113
    },
    {
      "epoch": 2.2398849127854703,
      "grad_norm": 3.593441842493296,
      "learning_rate": 2.652074318227286e-06,
      "loss": 0.0498,
      "step": 3114
    },
    {
      "epoch": 2.2406042078762813,
      "grad_norm": 3.111707617281264,
      "learning_rate": 2.6518563991484932e-06,
      "loss": 0.012,
      "step": 3115
    },
    {
      "epoch": 2.2413235029670924,
      "grad_norm": 5.802986567880055,
      "learning_rate": 2.651638420804667e-06,
      "loss": 0.2772,
      "step": 3116
    },
    {
      "epoch": 2.2420427980579034,
      "grad_norm": 4.121308886450126,
      "learning_rate": 2.6514203832070225e-06,
      "loss": 0.1063,
      "step": 3117
    },
    {
      "epoch": 2.2427620931487144,
      "grad_norm": 5.8871115749295875,
      "learning_rate": 2.6512022863667777e-06,
      "loss": 0.0834,
      "step": 3118
    },
    {
      "epoch": 2.243481388239525,
      "grad_norm": 0.9379958287580851,
      "learning_rate": 2.6509841302951545e-06,
      "loss": 0.0026,
      "step": 3119
    },
    {
      "epoch": 2.244200683330336,
      "grad_norm": 1.8625252842730533,
      "learning_rate": 2.6507659150033773e-06,
      "loss": 0.0103,
      "step": 3120
    },
    {
      "epoch": 2.244919978421147,
      "grad_norm": 4.693076547652265,
      "learning_rate": 2.650547640502674e-06,
      "loss": 0.0383,
      "step": 3121
    },
    {
      "epoch": 2.245639273511958,
      "grad_norm": 1.9409617699373858,
      "learning_rate": 2.650329306804275e-06,
      "loss": 0.0322,
      "step": 3122
    },
    {
      "epoch": 2.246358568602769,
      "grad_norm": 5.047323420218156,
      "learning_rate": 2.650110913919413e-06,
      "loss": 0.1851,
      "step": 3123
    },
    {
      "epoch": 2.24707786369358,
      "grad_norm": 4.152964517058767,
      "learning_rate": 2.6498924618593263e-06,
      "loss": 0.0232,
      "step": 3124
    },
    {
      "epoch": 2.2477971587843912,
      "grad_norm": 4.181388212459385,
      "learning_rate": 2.6496739506352535e-06,
      "loss": 0.04,
      "step": 3125
    },
    {
      "epoch": 2.2485164538752023,
      "grad_norm": 5.541747333116619,
      "learning_rate": 2.649455380258438e-06,
      "loss": 0.1298,
      "step": 3126
    },
    {
      "epoch": 2.2492357489660133,
      "grad_norm": 3.4677824536010395,
      "learning_rate": 2.6492367507401248e-06,
      "loss": 0.0331,
      "step": 3127
    },
    {
      "epoch": 2.2499550440568243,
      "grad_norm": 4.108935045626365,
      "learning_rate": 2.649018062091564e-06,
      "loss": 0.2064,
      "step": 3128
    },
    {
      "epoch": 2.2506743391476354,
      "grad_norm": 3.3692509220688476,
      "learning_rate": 2.6487993143240064e-06,
      "loss": 0.1277,
      "step": 3129
    },
    {
      "epoch": 2.2513936342384464,
      "grad_norm": 3.8016912674631658,
      "learning_rate": 2.6485805074487077e-06,
      "loss": 0.0555,
      "step": 3130
    },
    {
      "epoch": 2.2521129293292574,
      "grad_norm": 4.840451003937193,
      "learning_rate": 2.6483616414769257e-06,
      "loss": 0.1553,
      "step": 3131
    },
    {
      "epoch": 2.2528322244200685,
      "grad_norm": 5.281670339443802,
      "learning_rate": 2.648142716419921e-06,
      "loss": 0.1385,
      "step": 3132
    },
    {
      "epoch": 2.2535515195108795,
      "grad_norm": 5.755838631696636,
      "learning_rate": 2.6479237322889583e-06,
      "loss": 0.0462,
      "step": 3133
    },
    {
      "epoch": 2.25427081460169,
      "grad_norm": 3.132203960316398,
      "learning_rate": 2.6477046890953044e-06,
      "loss": 0.1352,
      "step": 3134
    },
    {
      "epoch": 2.2549901096925016,
      "grad_norm": 4.320073845384806,
      "learning_rate": 2.6474855868502297e-06,
      "loss": 0.0982,
      "step": 3135
    },
    {
      "epoch": 2.255709404783312,
      "grad_norm": 3.920408125103883,
      "learning_rate": 2.6472664255650067e-06,
      "loss": 0.0715,
      "step": 3136
    },
    {
      "epoch": 2.256428699874123,
      "grad_norm": 1.860282120490665,
      "learning_rate": 2.647047205250913e-06,
      "loss": 0.0487,
      "step": 3137
    },
    {
      "epoch": 2.2571479949649342,
      "grad_norm": 4.011568537652182,
      "learning_rate": 2.646827925919226e-06,
      "loss": 0.1121,
      "step": 3138
    },
    {
      "epoch": 2.2578672900557453,
      "grad_norm": 3.5299060011450956,
      "learning_rate": 2.64660858758123e-06,
      "loss": 0.1042,
      "step": 3139
    },
    {
      "epoch": 2.2585865851465563,
      "grad_norm": 1.8406257536718567,
      "learning_rate": 2.6463891902482088e-06,
      "loss": 0.0371,
      "step": 3140
    },
    {
      "epoch": 2.2593058802373673,
      "grad_norm": 3.2687682073787903,
      "learning_rate": 2.646169733931451e-06,
      "loss": 0.0236,
      "step": 3141
    },
    {
      "epoch": 2.2600251753281784,
      "grad_norm": 2.5703984038313465,
      "learning_rate": 2.645950218642249e-06,
      "loss": 0.0568,
      "step": 3142
    },
    {
      "epoch": 2.2607444704189894,
      "grad_norm": 7.178317791973445,
      "learning_rate": 2.6457306443918964e-06,
      "loss": 0.1419,
      "step": 3143
    },
    {
      "epoch": 2.2614637655098004,
      "grad_norm": 2.5026856690020227,
      "learning_rate": 2.645511011191691e-06,
      "loss": 0.0351,
      "step": 3144
    },
    {
      "epoch": 2.2621830606006115,
      "grad_norm": 2.1657039791045896,
      "learning_rate": 2.6452913190529326e-06,
      "loss": 0.0088,
      "step": 3145
    },
    {
      "epoch": 2.2629023556914225,
      "grad_norm": 2.279016402288655,
      "learning_rate": 2.645071567986926e-06,
      "loss": 0.0176,
      "step": 3146
    },
    {
      "epoch": 2.2636216507822335,
      "grad_norm": 6.099628841196772,
      "learning_rate": 2.6448517580049767e-06,
      "loss": 0.2778,
      "step": 3147
    },
    {
      "epoch": 2.2643409458730446,
      "grad_norm": 2.0492876487644014,
      "learning_rate": 2.6446318891183953e-06,
      "loss": 0.0078,
      "step": 3148
    },
    {
      "epoch": 2.2650602409638556,
      "grad_norm": 1.4951885996037297,
      "learning_rate": 2.644411961338493e-06,
      "loss": 0.0211,
      "step": 3149
    },
    {
      "epoch": 2.2657795360546666,
      "grad_norm": 0.33223813321767576,
      "learning_rate": 2.644191974676587e-06,
      "loss": 0.0012,
      "step": 3150
    },
    {
      "epoch": 2.2664988311454772,
      "grad_norm": 6.302396631023723,
      "learning_rate": 2.6439719291439953e-06,
      "loss": 0.2991,
      "step": 3151
    },
    {
      "epoch": 2.2672181262362883,
      "grad_norm": 3.8098503758096727,
      "learning_rate": 2.6437518247520396e-06,
      "loss": 0.1413,
      "step": 3152
    },
    {
      "epoch": 2.2679374213270993,
      "grad_norm": 2.9614581694930684,
      "learning_rate": 2.6435316615120447e-06,
      "loss": 0.0704,
      "step": 3153
    },
    {
      "epoch": 2.2686567164179103,
      "grad_norm": 3.567948743864831,
      "learning_rate": 2.6433114394353384e-06,
      "loss": 0.0931,
      "step": 3154
    },
    {
      "epoch": 2.2693760115087214,
      "grad_norm": 2.877146070664812,
      "learning_rate": 2.6430911585332513e-06,
      "loss": 0.0179,
      "step": 3155
    },
    {
      "epoch": 2.2700953065995324,
      "grad_norm": 4.404098845967638,
      "learning_rate": 2.6428708188171174e-06,
      "loss": 0.0993,
      "step": 3156
    },
    {
      "epoch": 2.2708146016903434,
      "grad_norm": 5.952296052235995,
      "learning_rate": 2.642650420298274e-06,
      "loss": 0.2137,
      "step": 3157
    },
    {
      "epoch": 2.2715338967811545,
      "grad_norm": 4.68130613818375,
      "learning_rate": 2.6424299629880605e-06,
      "loss": 0.0838,
      "step": 3158
    },
    {
      "epoch": 2.2722531918719655,
      "grad_norm": 2.9327875125044818,
      "learning_rate": 2.6422094468978202e-06,
      "loss": 0.1198,
      "step": 3159
    },
    {
      "epoch": 2.2729724869627765,
      "grad_norm": 7.118918166578994,
      "learning_rate": 2.6419888720388984e-06,
      "loss": 0.4315,
      "step": 3160
    },
    {
      "epoch": 2.2736917820535876,
      "grad_norm": 5.517152386786748,
      "learning_rate": 2.6417682384226445e-06,
      "loss": 0.3461,
      "step": 3161
    },
    {
      "epoch": 2.2744110771443986,
      "grad_norm": 3.3752909775236026,
      "learning_rate": 2.6415475460604103e-06,
      "loss": 0.0737,
      "step": 3162
    },
    {
      "epoch": 2.2751303722352096,
      "grad_norm": 5.323040915013503,
      "learning_rate": 2.641326794963551e-06,
      "loss": 0.0896,
      "step": 3163
    },
    {
      "epoch": 2.2758496673260207,
      "grad_norm": 2.977056064877686,
      "learning_rate": 2.6411059851434243e-06,
      "loss": 0.0589,
      "step": 3164
    },
    {
      "epoch": 2.2765689624168317,
      "grad_norm": 2.4402915535172323,
      "learning_rate": 2.640885116611392e-06,
      "loss": 0.0343,
      "step": 3165
    },
    {
      "epoch": 2.2772882575076423,
      "grad_norm": 3.294882762963217,
      "learning_rate": 2.640664189378818e-06,
      "loss": 0.0895,
      "step": 3166
    },
    {
      "epoch": 2.2780075525984533,
      "grad_norm": 2.5086126833391162,
      "learning_rate": 2.640443203457068e-06,
      "loss": 0.0462,
      "step": 3167
    },
    {
      "epoch": 2.2787268476892644,
      "grad_norm": 5.712148283429059,
      "learning_rate": 2.6402221588575144e-06,
      "loss": 0.2806,
      "step": 3168
    },
    {
      "epoch": 2.2794461427800754,
      "grad_norm": 2.2443941728901726,
      "learning_rate": 2.6400010555915287e-06,
      "loss": 0.0807,
      "step": 3169
    },
    {
      "epoch": 2.2801654378708864,
      "grad_norm": 4.121774184077429,
      "learning_rate": 2.639779893670487e-06,
      "loss": 0.1267,
      "step": 3170
    },
    {
      "epoch": 2.2808847329616975,
      "grad_norm": 5.768451589776608,
      "learning_rate": 2.6395586731057694e-06,
      "loss": 0.0703,
      "step": 3171
    },
    {
      "epoch": 2.2816040280525085,
      "grad_norm": 2.685928919688494,
      "learning_rate": 2.6393373939087574e-06,
      "loss": 0.0883,
      "step": 3172
    },
    {
      "epoch": 2.2823233231433195,
      "grad_norm": 6.696330085918585,
      "learning_rate": 2.6391160560908367e-06,
      "loss": 0.2338,
      "step": 3173
    },
    {
      "epoch": 2.2830426182341306,
      "grad_norm": 6.611166296543465,
      "learning_rate": 2.638894659663395e-06,
      "loss": 0.203,
      "step": 3174
    },
    {
      "epoch": 2.2837619133249416,
      "grad_norm": 0.3734308692080473,
      "learning_rate": 2.6386732046378243e-06,
      "loss": 0.0014,
      "step": 3175
    },
    {
      "epoch": 2.2844812084157526,
      "grad_norm": 3.2628901400469457,
      "learning_rate": 2.6384516910255177e-06,
      "loss": 0.1223,
      "step": 3176
    },
    {
      "epoch": 2.2852005035065637,
      "grad_norm": 2.0397670262804404,
      "learning_rate": 2.6382301188378733e-06,
      "loss": 0.0135,
      "step": 3177
    },
    {
      "epoch": 2.2859197985973747,
      "grad_norm": 1.9947481532087334,
      "learning_rate": 2.6380084880862913e-06,
      "loss": 0.0349,
      "step": 3178
    },
    {
      "epoch": 2.2866390936881857,
      "grad_norm": 3.6195942903952294,
      "learning_rate": 2.6377867987821745e-06,
      "loss": 0.1501,
      "step": 3179
    },
    {
      "epoch": 2.2873583887789968,
      "grad_norm": 4.563441919501048,
      "learning_rate": 2.6375650509369303e-06,
      "loss": 0.1943,
      "step": 3180
    },
    {
      "epoch": 2.2880776838698074,
      "grad_norm": 4.6351640395391795,
      "learning_rate": 2.637343244561967e-06,
      "loss": 0.0193,
      "step": 3181
    },
    {
      "epoch": 2.2887969789606184,
      "grad_norm": 4.14082647467695,
      "learning_rate": 2.637121379668697e-06,
      "loss": 0.2096,
      "step": 3182
    },
    {
      "epoch": 2.2895162740514294,
      "grad_norm": 2.4403530330068044,
      "learning_rate": 2.6368994562685355e-06,
      "loss": 0.0921,
      "step": 3183
    },
    {
      "epoch": 2.2902355691422405,
      "grad_norm": 2.8270799803446063,
      "learning_rate": 2.6366774743729024e-06,
      "loss": 0.089,
      "step": 3184
    },
    {
      "epoch": 2.2909548642330515,
      "grad_norm": 3.633335232082756,
      "learning_rate": 2.636455433993217e-06,
      "loss": 0.1557,
      "step": 3185
    },
    {
      "epoch": 2.2916741593238625,
      "grad_norm": 4.766040928579625,
      "learning_rate": 2.6362333351409045e-06,
      "loss": 0.05,
      "step": 3186
    },
    {
      "epoch": 2.2923934544146736,
      "grad_norm": 2.901915924214587,
      "learning_rate": 2.6360111778273924e-06,
      "loss": 0.0211,
      "step": 3187
    },
    {
      "epoch": 2.2931127495054846,
      "grad_norm": 3.19360889581358,
      "learning_rate": 2.6357889620641117e-06,
      "loss": 0.154,
      "step": 3188
    },
    {
      "epoch": 2.2938320445962956,
      "grad_norm": 4.927885647196908,
      "learning_rate": 2.635566687862494e-06,
      "loss": 0.1897,
      "step": 3189
    },
    {
      "epoch": 2.2945513396871067,
      "grad_norm": 3.826894403536538,
      "learning_rate": 2.6353443552339772e-06,
      "loss": 0.0127,
      "step": 3190
    },
    {
      "epoch": 2.2952706347779177,
      "grad_norm": 4.488003219315735,
      "learning_rate": 2.6351219641900005e-06,
      "loss": 0.1046,
      "step": 3191
    },
    {
      "epoch": 2.2959899298687287,
      "grad_norm": 2.6056650237929584,
      "learning_rate": 2.634899514742006e-06,
      "loss": 0.1008,
      "step": 3192
    },
    {
      "epoch": 2.2967092249595398,
      "grad_norm": 6.445082830139398,
      "learning_rate": 2.6346770069014393e-06,
      "loss": 0.0784,
      "step": 3193
    },
    {
      "epoch": 2.297428520050351,
      "grad_norm": 3.0478827675390434,
      "learning_rate": 2.6344544406797484e-06,
      "loss": 0.0672,
      "step": 3194
    },
    {
      "epoch": 2.298147815141162,
      "grad_norm": 4.392440685929836,
      "learning_rate": 2.6342318160883857e-06,
      "loss": 0.0822,
      "step": 3195
    },
    {
      "epoch": 2.2988671102319724,
      "grad_norm": 3.8900908966153644,
      "learning_rate": 2.6340091331388042e-06,
      "loss": 0.0228,
      "step": 3196
    },
    {
      "epoch": 2.299586405322784,
      "grad_norm": 4.511217985765185,
      "learning_rate": 2.6337863918424627e-06,
      "loss": 0.1393,
      "step": 3197
    },
    {
      "epoch": 2.3003057004135945,
      "grad_norm": 2.707253810269414,
      "learning_rate": 2.6335635922108207e-06,
      "loss": 0.057,
      "step": 3198
    },
    {
      "epoch": 2.3010249955044055,
      "grad_norm": 7.328774868536007,
      "learning_rate": 2.633340734255342e-06,
      "loss": 0.3008,
      "step": 3199
    },
    {
      "epoch": 2.3017442905952166,
      "grad_norm": 4.227621068655837,
      "learning_rate": 2.6331178179874934e-06,
      "loss": 0.137,
      "step": 3200
    },
    {
      "epoch": 2.3024635856860276,
      "grad_norm": 2.229177161741265,
      "learning_rate": 2.6328948434187433e-06,
      "loss": 0.047,
      "step": 3201
    },
    {
      "epoch": 2.3031828807768386,
      "grad_norm": 0.9093004970368942,
      "learning_rate": 2.632671810560565e-06,
      "loss": 0.0095,
      "step": 3202
    },
    {
      "epoch": 2.3039021758676497,
      "grad_norm": 1.7377163845615304,
      "learning_rate": 2.632448719424434e-06,
      "loss": 0.0449,
      "step": 3203
    },
    {
      "epoch": 2.3046214709584607,
      "grad_norm": 5.742766020731803,
      "learning_rate": 2.6322255700218282e-06,
      "loss": 0.1254,
      "step": 3204
    },
    {
      "epoch": 2.3053407660492717,
      "grad_norm": 2.3689766354974426,
      "learning_rate": 2.6320023623642295e-06,
      "loss": 0.0697,
      "step": 3205
    },
    {
      "epoch": 2.3060600611400828,
      "grad_norm": 4.684064289612578,
      "learning_rate": 2.631779096463122e-06,
      "loss": 0.1798,
      "step": 3206
    },
    {
      "epoch": 2.306779356230894,
      "grad_norm": 3.0546339029182907,
      "learning_rate": 2.631555772329993e-06,
      "loss": 0.0891,
      "step": 3207
    },
    {
      "epoch": 2.307498651321705,
      "grad_norm": 6.055255042058,
      "learning_rate": 2.631332389976333e-06,
      "loss": 0.0968,
      "step": 3208
    },
    {
      "epoch": 2.308217946412516,
      "grad_norm": 5.505575914633913,
      "learning_rate": 2.6311089494136357e-06,
      "loss": 0.0783,
      "step": 3209
    },
    {
      "epoch": 2.308937241503327,
      "grad_norm": 3.6646172010398823,
      "learning_rate": 2.6308854506533974e-06,
      "loss": 0.1522,
      "step": 3210
    },
    {
      "epoch": 2.3096565365941375,
      "grad_norm": 5.002773504550886,
      "learning_rate": 2.6306618937071177e-06,
      "loss": 0.2355,
      "step": 3211
    },
    {
      "epoch": 2.310375831684949,
      "grad_norm": 5.986526501043334,
      "learning_rate": 2.630438278586298e-06,
      "loss": 0.0623,
      "step": 3212
    },
    {
      "epoch": 2.3110951267757596,
      "grad_norm": 5.714904457199305,
      "learning_rate": 2.6302146053024456e-06,
      "loss": 0.2362,
      "step": 3213
    },
    {
      "epoch": 2.3118144218665706,
      "grad_norm": 4.022492054063511,
      "learning_rate": 2.629990873867067e-06,
      "loss": 0.1714,
      "step": 3214
    },
    {
      "epoch": 2.3125337169573816,
      "grad_norm": 2.5832028401638722,
      "learning_rate": 2.6297670842916748e-06,
      "loss": 0.0263,
      "step": 3215
    },
    {
      "epoch": 2.3132530120481927,
      "grad_norm": 2.6269669518033707,
      "learning_rate": 2.629543236587782e-06,
      "loss": 0.0252,
      "step": 3216
    },
    {
      "epoch": 2.3139723071390037,
      "grad_norm": 4.417787754433668,
      "learning_rate": 2.6293193307669074e-06,
      "loss": 0.1439,
      "step": 3217
    },
    {
      "epoch": 2.3146916022298147,
      "grad_norm": 4.479064879562671,
      "learning_rate": 2.629095366840571e-06,
      "loss": 0.1103,
      "step": 3218
    },
    {
      "epoch": 2.3154108973206258,
      "grad_norm": 1.3272260207938729,
      "learning_rate": 2.6288713448202953e-06,
      "loss": 0.021,
      "step": 3219
    },
    {
      "epoch": 2.316130192411437,
      "grad_norm": 1.7860871088357648,
      "learning_rate": 2.628647264717607e-06,
      "loss": 0.0331,
      "step": 3220
    },
    {
      "epoch": 2.316849487502248,
      "grad_norm": 3.16881770218607,
      "learning_rate": 2.628423126544037e-06,
      "loss": 0.0282,
      "step": 3221
    },
    {
      "epoch": 2.317568782593059,
      "grad_norm": 5.569281482560819,
      "learning_rate": 2.6281989303111154e-06,
      "loss": 0.2545,
      "step": 3222
    },
    {
      "epoch": 2.31828807768387,
      "grad_norm": 3.937085780683137,
      "learning_rate": 2.627974676030378e-06,
      "loss": 0.0219,
      "step": 3223
    },
    {
      "epoch": 2.319007372774681,
      "grad_norm": 4.681059642585563,
      "learning_rate": 2.627750363713364e-06,
      "loss": 0.0379,
      "step": 3224
    },
    {
      "epoch": 2.319726667865492,
      "grad_norm": 3.5556408175114065,
      "learning_rate": 2.6275259933716137e-06,
      "loss": 0.0806,
      "step": 3225
    },
    {
      "epoch": 2.320445962956303,
      "grad_norm": 4.914838878091459,
      "learning_rate": 2.627301565016672e-06,
      "loss": 0.2153,
      "step": 3226
    },
    {
      "epoch": 2.321165258047114,
      "grad_norm": 4.421490232990445,
      "learning_rate": 2.627077078660086e-06,
      "loss": 0.1786,
      "step": 3227
    },
    {
      "epoch": 2.3218845531379246,
      "grad_norm": 3.8499002494768932,
      "learning_rate": 2.6268525343134053e-06,
      "loss": 0.1013,
      "step": 3228
    },
    {
      "epoch": 2.3226038482287357,
      "grad_norm": 3.084380894997996,
      "learning_rate": 2.626627931988184e-06,
      "loss": 0.1139,
      "step": 3229
    },
    {
      "epoch": 2.3233231433195467,
      "grad_norm": 2.314498156322512,
      "learning_rate": 2.626403271695978e-06,
      "loss": 0.0636,
      "step": 3230
    },
    {
      "epoch": 2.3240424384103577,
      "grad_norm": 3.651956815139289,
      "learning_rate": 2.626178553448346e-06,
      "loss": 0.1136,
      "step": 3231
    },
    {
      "epoch": 2.3247617335011688,
      "grad_norm": 4.452654780777023,
      "learning_rate": 2.6259537772568503e-06,
      "loss": 0.1856,
      "step": 3232
    },
    {
      "epoch": 2.32548102859198,
      "grad_norm": 0.8384514621235604,
      "learning_rate": 2.6257289431330564e-06,
      "loss": 0.0038,
      "step": 3233
    },
    {
      "epoch": 2.326200323682791,
      "grad_norm": 0.5008624707146139,
      "learning_rate": 2.6255040510885324e-06,
      "loss": 0.0016,
      "step": 3234
    },
    {
      "epoch": 2.326919618773602,
      "grad_norm": 3.668534297032654,
      "learning_rate": 2.625279101134849e-06,
      "loss": 0.0634,
      "step": 3235
    },
    {
      "epoch": 2.327638913864413,
      "grad_norm": 7.115826003727372,
      "learning_rate": 2.6250540932835806e-06,
      "loss": 0.1129,
      "step": 3236
    },
    {
      "epoch": 2.328358208955224,
      "grad_norm": 1.8504405875231962,
      "learning_rate": 2.624829027546304e-06,
      "loss": 0.0088,
      "step": 3237
    },
    {
      "epoch": 2.329077504046035,
      "grad_norm": 6.654540441922057,
      "learning_rate": 2.6246039039345997e-06,
      "loss": 0.1282,
      "step": 3238
    },
    {
      "epoch": 2.329796799136846,
      "grad_norm": 3.385123564832717,
      "learning_rate": 2.62437872246005e-06,
      "loss": 0.1041,
      "step": 3239
    },
    {
      "epoch": 2.330516094227657,
      "grad_norm": 3.133796092188657,
      "learning_rate": 2.6241534831342413e-06,
      "loss": 0.1467,
      "step": 3240
    },
    {
      "epoch": 2.331235389318468,
      "grad_norm": 3.4958061330003942,
      "learning_rate": 2.6239281859687626e-06,
      "loss": 0.0235,
      "step": 3241
    },
    {
      "epoch": 2.331954684409279,
      "grad_norm": 4.6336141186220345,
      "learning_rate": 2.623702830975206e-06,
      "loss": 0.0945,
      "step": 3242
    },
    {
      "epoch": 2.3326739795000897,
      "grad_norm": 3.740908834709217,
      "learning_rate": 2.6234774181651655e-06,
      "loss": 0.0151,
      "step": 3243
    },
    {
      "epoch": 2.3333932745909007,
      "grad_norm": 3.2491759287648345,
      "learning_rate": 2.6232519475502396e-06,
      "loss": 0.111,
      "step": 3244
    },
    {
      "epoch": 2.3341125696817118,
      "grad_norm": 2.8293464501125016,
      "learning_rate": 2.6230264191420296e-06,
      "loss": 0.1011,
      "step": 3245
    },
    {
      "epoch": 2.334831864772523,
      "grad_norm": 0.8665390429828626,
      "learning_rate": 2.6228008329521393e-06,
      "loss": 0.0055,
      "step": 3246
    },
    {
      "epoch": 2.335551159863334,
      "grad_norm": 4.305871716659971,
      "learning_rate": 2.622575188992175e-06,
      "loss": 0.1415,
      "step": 3247
    },
    {
      "epoch": 2.336270454954145,
      "grad_norm": 1.7828022797847378,
      "learning_rate": 2.6223494872737457e-06,
      "loss": 0.0088,
      "step": 3248
    },
    {
      "epoch": 2.336989750044956,
      "grad_norm": 3.8370622995810537,
      "learning_rate": 2.622123727808466e-06,
      "loss": 0.0508,
      "step": 3249
    },
    {
      "epoch": 2.337709045135767,
      "grad_norm": 3.915717824877781,
      "learning_rate": 2.6218979106079504e-06,
      "loss": 0.1283,
      "step": 3250
    },
    {
      "epoch": 2.338428340226578,
      "grad_norm": 0.21989529950977305,
      "learning_rate": 2.621672035683818e-06,
      "loss": 0.0005,
      "step": 3251
    },
    {
      "epoch": 2.339147635317389,
      "grad_norm": 3.632246475177959,
      "learning_rate": 2.62144610304769e-06,
      "loss": 0.0683,
      "step": 3252
    },
    {
      "epoch": 2.3398669304082,
      "grad_norm": 5.102368009167012,
      "learning_rate": 2.621220112711192e-06,
      "loss": 0.0857,
      "step": 3253
    },
    {
      "epoch": 2.340586225499011,
      "grad_norm": 7.120913357059576,
      "learning_rate": 2.6209940646859498e-06,
      "loss": 0.1972,
      "step": 3254
    },
    {
      "epoch": 2.341305520589822,
      "grad_norm": 2.819765217164804,
      "learning_rate": 2.620767958983596e-06,
      "loss": 0.0719,
      "step": 3255
    },
    {
      "epoch": 2.342024815680633,
      "grad_norm": 4.521278162963943,
      "learning_rate": 2.6205417956157627e-06,
      "loss": 0.1091,
      "step": 3256
    },
    {
      "epoch": 2.342744110771444,
      "grad_norm": 3.6339082644166805,
      "learning_rate": 2.620315574594088e-06,
      "loss": 0.1275,
      "step": 3257
    },
    {
      "epoch": 2.3434634058622548,
      "grad_norm": 3.9336499752771803,
      "learning_rate": 2.6200892959302094e-06,
      "loss": 0.0788,
      "step": 3258
    },
    {
      "epoch": 2.3441827009530662,
      "grad_norm": 3.4050869661139007,
      "learning_rate": 2.6198629596357703e-06,
      "loss": 0.0965,
      "step": 3259
    },
    {
      "epoch": 2.344901996043877,
      "grad_norm": 2.9182447108112517,
      "learning_rate": 2.6196365657224165e-06,
      "loss": 0.0143,
      "step": 3260
    },
    {
      "epoch": 2.345621291134688,
      "grad_norm": 4.367304625559985,
      "learning_rate": 2.6194101142017958e-06,
      "loss": 0.1405,
      "step": 3261
    },
    {
      "epoch": 2.346340586225499,
      "grad_norm": 3.300698634372535,
      "learning_rate": 2.6191836050855596e-06,
      "loss": 0.0446,
      "step": 3262
    },
    {
      "epoch": 2.34705988131631,
      "grad_norm": 2.126951173850292,
      "learning_rate": 2.6189570383853623e-06,
      "loss": 0.0523,
      "step": 3263
    },
    {
      "epoch": 2.347779176407121,
      "grad_norm": 2.753151965244601,
      "learning_rate": 2.618730414112861e-06,
      "loss": 0.037,
      "step": 3264
    },
    {
      "epoch": 2.348498471497932,
      "grad_norm": 3.4713069535858985,
      "learning_rate": 2.618503732279716e-06,
      "loss": 0.0494,
      "step": 3265
    },
    {
      "epoch": 2.349217766588743,
      "grad_norm": 2.933278348690487,
      "learning_rate": 2.618276992897591e-06,
      "loss": 0.0883,
      "step": 3266
    },
    {
      "epoch": 2.349937061679554,
      "grad_norm": 1.6307076004087933,
      "learning_rate": 2.618050195978151e-06,
      "loss": 0.0258,
      "step": 3267
    },
    {
      "epoch": 2.350656356770365,
      "grad_norm": 2.605336933895664,
      "learning_rate": 2.617823341533066e-06,
      "loss": 0.0554,
      "step": 3268
    },
    {
      "epoch": 2.351375651861176,
      "grad_norm": 0.6912758079659216,
      "learning_rate": 2.617596429574008e-06,
      "loss": 0.012,
      "step": 3269
    },
    {
      "epoch": 2.352094946951987,
      "grad_norm": 6.249300910941087,
      "learning_rate": 2.617369460112651e-06,
      "loss": 0.0425,
      "step": 3270
    },
    {
      "epoch": 2.352814242042798,
      "grad_norm": 1.922700386099548,
      "learning_rate": 2.617142433160675e-06,
      "loss": 0.0418,
      "step": 3271
    },
    {
      "epoch": 2.3535335371336092,
      "grad_norm": 4.465866071095441,
      "learning_rate": 2.616915348729759e-06,
      "loss": 0.1555,
      "step": 3272
    },
    {
      "epoch": 2.35425283222442,
      "grad_norm": 2.0089191688740478,
      "learning_rate": 2.616688206831588e-06,
      "loss": 0.0066,
      "step": 3273
    },
    {
      "epoch": 2.3549721273152313,
      "grad_norm": 0.6821764097499252,
      "learning_rate": 2.616461007477848e-06,
      "loss": 0.0027,
      "step": 3274
    },
    {
      "epoch": 2.355691422406042,
      "grad_norm": 2.3125492349411987,
      "learning_rate": 2.6162337506802296e-06,
      "loss": 0.0468,
      "step": 3275
    },
    {
      "epoch": 2.356410717496853,
      "grad_norm": 1.7133712405744657,
      "learning_rate": 2.616006436450425e-06,
      "loss": 0.0164,
      "step": 3276
    },
    {
      "epoch": 2.357130012587664,
      "grad_norm": 3.6468154805373465,
      "learning_rate": 2.6157790648001305e-06,
      "loss": 0.1108,
      "step": 3277
    },
    {
      "epoch": 2.357849307678475,
      "grad_norm": 2.630860690600742,
      "learning_rate": 2.615551635741044e-06,
      "loss": 0.0744,
      "step": 3278
    },
    {
      "epoch": 2.358568602769286,
      "grad_norm": 4.205270685035885,
      "learning_rate": 2.6153241492848678e-06,
      "loss": 0.1699,
      "step": 3279
    },
    {
      "epoch": 2.359287897860097,
      "grad_norm": 5.15763402108436,
      "learning_rate": 2.6150966054433066e-06,
      "loss": 0.1655,
      "step": 3280
    },
    {
      "epoch": 2.360007192950908,
      "grad_norm": 6.6261376506575855,
      "learning_rate": 2.614869004228067e-06,
      "loss": 0.1239,
      "step": 3281
    },
    {
      "epoch": 2.360726488041719,
      "grad_norm": 8.706499400256206,
      "learning_rate": 2.6146413456508604e-06,
      "loss": 0.2258,
      "step": 3282
    },
    {
      "epoch": 2.36144578313253,
      "grad_norm": 5.074909894744741,
      "learning_rate": 2.6144136297233993e-06,
      "loss": 0.1163,
      "step": 3283
    },
    {
      "epoch": 2.362165078223341,
      "grad_norm": 7.135554117990332,
      "learning_rate": 2.6141858564574014e-06,
      "loss": 0.1388,
      "step": 3284
    },
    {
      "epoch": 2.3628843733141522,
      "grad_norm": 4.576786876407508,
      "learning_rate": 2.6139580258645847e-06,
      "loss": 0.1137,
      "step": 3285
    },
    {
      "epoch": 2.3636036684049633,
      "grad_norm": 6.097897237511379,
      "learning_rate": 2.6137301379566725e-06,
      "loss": 0.1427,
      "step": 3286
    },
    {
      "epoch": 2.3643229634957743,
      "grad_norm": 2.8765828585811084,
      "learning_rate": 2.6135021927453894e-06,
      "loss": 0.1441,
      "step": 3287
    },
    {
      "epoch": 2.365042258586585,
      "grad_norm": 1.7587624042284395,
      "learning_rate": 2.613274190242464e-06,
      "loss": 0.0624,
      "step": 3288
    },
    {
      "epoch": 2.3657615536773964,
      "grad_norm": 2.2563846892607775,
      "learning_rate": 2.613046130459627e-06,
      "loss": 0.0068,
      "step": 3289
    },
    {
      "epoch": 2.366480848768207,
      "grad_norm": 7.365683069539245,
      "learning_rate": 2.6128180134086127e-06,
      "loss": 0.3169,
      "step": 3290
    },
    {
      "epoch": 2.367200143859018,
      "grad_norm": 7.742375105819887,
      "learning_rate": 2.612589839101158e-06,
      "loss": 0.0208,
      "step": 3291
    },
    {
      "epoch": 2.367919438949829,
      "grad_norm": 3.44450696687367,
      "learning_rate": 2.612361607549003e-06,
      "loss": 0.0654,
      "step": 3292
    },
    {
      "epoch": 2.36863873404064,
      "grad_norm": 1.8758272923381216,
      "learning_rate": 2.612133318763891e-06,
      "loss": 0.0077,
      "step": 3293
    },
    {
      "epoch": 2.369358029131451,
      "grad_norm": 0.31217239936916535,
      "learning_rate": 2.6119049727575674e-06,
      "loss": 0.001,
      "step": 3294
    },
    {
      "epoch": 2.370077324222262,
      "grad_norm": 5.398066594081044,
      "learning_rate": 2.611676569541781e-06,
      "loss": 0.2323,
      "step": 3295
    },
    {
      "epoch": 2.370796619313073,
      "grad_norm": 2.733079687555914,
      "learning_rate": 2.611448109128283e-06,
      "loss": 0.0251,
      "step": 3296
    },
    {
      "epoch": 2.371515914403884,
      "grad_norm": 4.673694763889029,
      "learning_rate": 2.6112195915288295e-06,
      "loss": 0.1463,
      "step": 3297
    },
    {
      "epoch": 2.3722352094946952,
      "grad_norm": 2.0619641737898617,
      "learning_rate": 2.610991016755177e-06,
      "loss": 0.07,
      "step": 3298
    },
    {
      "epoch": 2.3729545045855063,
      "grad_norm": 4.895604290524185,
      "learning_rate": 2.6107623848190868e-06,
      "loss": 0.1705,
      "step": 3299
    },
    {
      "epoch": 2.3736737996763173,
      "grad_norm": 0.02379951344047758,
      "learning_rate": 2.6105336957323216e-06,
      "loss": 0.0002,
      "step": 3300
    },
    {
      "epoch": 2.3743930947671283,
      "grad_norm": 2.718621997906582,
      "learning_rate": 2.6103049495066484e-06,
      "loss": 0.0589,
      "step": 3301
    },
    {
      "epoch": 2.3751123898579394,
      "grad_norm": 0.20847892315946664,
      "learning_rate": 2.610076146153836e-06,
      "loss": 0.0005,
      "step": 3302
    },
    {
      "epoch": 2.3758316849487504,
      "grad_norm": 5.962183935914835,
      "learning_rate": 2.609847285685658e-06,
      "loss": 0.1788,
      "step": 3303
    },
    {
      "epoch": 2.3765509800395614,
      "grad_norm": 4.2281524682648985,
      "learning_rate": 2.6096183681138883e-06,
      "loss": 0.0638,
      "step": 3304
    },
    {
      "epoch": 2.377270275130372,
      "grad_norm": 4.477807691422038,
      "learning_rate": 2.609389393450306e-06,
      "loss": 0.022,
      "step": 3305
    },
    {
      "epoch": 2.377989570221183,
      "grad_norm": 6.325700217130086,
      "learning_rate": 2.6091603617066915e-06,
      "loss": 0.1787,
      "step": 3306
    },
    {
      "epoch": 2.378708865311994,
      "grad_norm": 6.746682756191801,
      "learning_rate": 2.6089312728948294e-06,
      "loss": 0.1382,
      "step": 3307
    },
    {
      "epoch": 2.379428160402805,
      "grad_norm": 4.268627621717964,
      "learning_rate": 2.6087021270265074e-06,
      "loss": 0.1834,
      "step": 3308
    },
    {
      "epoch": 2.380147455493616,
      "grad_norm": 2.559970540991954,
      "learning_rate": 2.608472924113514e-06,
      "loss": 0.0432,
      "step": 3309
    },
    {
      "epoch": 2.380866750584427,
      "grad_norm": 1.284566262720924,
      "learning_rate": 2.608243664167643e-06,
      "loss": 0.0153,
      "step": 3310
    },
    {
      "epoch": 2.3815860456752382,
      "grad_norm": 8.963321732078805,
      "learning_rate": 2.60801434720069e-06,
      "loss": 0.2047,
      "step": 3311
    },
    {
      "epoch": 2.3823053407660493,
      "grad_norm": 1.4640861417792845,
      "learning_rate": 2.607784973224454e-06,
      "loss": 0.0189,
      "step": 3312
    },
    {
      "epoch": 2.3830246358568603,
      "grad_norm": 5.548997010812094,
      "learning_rate": 2.607555542250736e-06,
      "loss": 0.0788,
      "step": 3313
    },
    {
      "epoch": 2.3837439309476713,
      "grad_norm": 4.609212106189434,
      "learning_rate": 2.607326054291341e-06,
      "loss": 0.0742,
      "step": 3314
    },
    {
      "epoch": 2.3844632260384824,
      "grad_norm": 3.3537050038921765,
      "learning_rate": 2.607096509358077e-06,
      "loss": 0.0666,
      "step": 3315
    },
    {
      "epoch": 2.3851825211292934,
      "grad_norm": 4.735711732501709,
      "learning_rate": 2.6068669074627543e-06,
      "loss": 0.1777,
      "step": 3316
    },
    {
      "epoch": 2.3859018162201044,
      "grad_norm": 0.6845861451879028,
      "learning_rate": 2.606637248617186e-06,
      "loss": 0.0023,
      "step": 3317
    },
    {
      "epoch": 2.3866211113109155,
      "grad_norm": 5.202610760805279,
      "learning_rate": 2.6064075328331886e-06,
      "loss": 0.221,
      "step": 3318
    },
    {
      "epoch": 2.3873404064017265,
      "grad_norm": 1.4539529324271563,
      "learning_rate": 2.6061777601225813e-06,
      "loss": 0.0039,
      "step": 3319
    },
    {
      "epoch": 2.388059701492537,
      "grad_norm": 1.7351750997833395,
      "learning_rate": 2.6059479304971867e-06,
      "loss": 0.0039,
      "step": 3320
    },
    {
      "epoch": 2.388778996583348,
      "grad_norm": 5.693676322666301,
      "learning_rate": 2.6057180439688293e-06,
      "loss": 0.1728,
      "step": 3321
    },
    {
      "epoch": 2.389498291674159,
      "grad_norm": 5.880913972371311,
      "learning_rate": 2.605488100549338e-06,
      "loss": 0.0984,
      "step": 3322
    },
    {
      "epoch": 2.39021758676497,
      "grad_norm": 5.657369583615648,
      "learning_rate": 2.605258100250543e-06,
      "loss": 0.2198,
      "step": 3323
    },
    {
      "epoch": 2.3909368818557812,
      "grad_norm": 4.291443872787153,
      "learning_rate": 2.6050280430842783e-06,
      "loss": 0.0188,
      "step": 3324
    },
    {
      "epoch": 2.3916561769465923,
      "grad_norm": 0.21068462410677244,
      "learning_rate": 2.604797929062381e-06,
      "loss": 0.0004,
      "step": 3325
    },
    {
      "epoch": 2.3923754720374033,
      "grad_norm": 1.9287491125814664,
      "learning_rate": 2.604567758196691e-06,
      "loss": 0.036,
      "step": 3326
    },
    {
      "epoch": 2.3930947671282143,
      "grad_norm": 3.5285771990173767,
      "learning_rate": 2.604337530499051e-06,
      "loss": 0.1178,
      "step": 3327
    },
    {
      "epoch": 2.3938140622190254,
      "grad_norm": 3.8284554381985654,
      "learning_rate": 2.6041072459813064e-06,
      "loss": 0.0799,
      "step": 3328
    },
    {
      "epoch": 2.3945333573098364,
      "grad_norm": 3.98823087275605,
      "learning_rate": 2.6038769046553056e-06,
      "loss": 0.1615,
      "step": 3329
    },
    {
      "epoch": 2.3952526524006474,
      "grad_norm": 2.0777798272628756,
      "learning_rate": 2.6036465065329005e-06,
      "loss": 0.0334,
      "step": 3330
    },
    {
      "epoch": 2.3959719474914585,
      "grad_norm": 2.2956077869831657,
      "learning_rate": 2.603416051625945e-06,
      "loss": 0.1066,
      "step": 3331
    },
    {
      "epoch": 2.3966912425822695,
      "grad_norm": 3.910657981411721,
      "learning_rate": 2.603185539946297e-06,
      "loss": 0.1121,
      "step": 3332
    },
    {
      "epoch": 2.3974105376730805,
      "grad_norm": 4.969600182242484,
      "learning_rate": 2.602954971505816e-06,
      "loss": 0.1736,
      "step": 3333
    },
    {
      "epoch": 2.3981298327638916,
      "grad_norm": 4.619373398717188,
      "learning_rate": 2.6027243463163656e-06,
      "loss": 0.178,
      "step": 3334
    },
    {
      "epoch": 2.398849127854702,
      "grad_norm": 4.0176905539229,
      "learning_rate": 2.6024936643898117e-06,
      "loss": 0.1219,
      "step": 3335
    },
    {
      "epoch": 2.3995684229455136,
      "grad_norm": 2.911164907699844,
      "learning_rate": 2.6022629257380238e-06,
      "loss": 0.1077,
      "step": 3336
    },
    {
      "epoch": 2.4002877180363242,
      "grad_norm": 0.05520278714887086,
      "learning_rate": 2.6020321303728733e-06,
      "loss": 0.0002,
      "step": 3337
    },
    {
      "epoch": 2.4010070131271353,
      "grad_norm": 3.700502145048883,
      "learning_rate": 2.601801278306235e-06,
      "loss": 0.1374,
      "step": 3338
    },
    {
      "epoch": 2.4017263082179463,
      "grad_norm": 4.375445460056996,
      "learning_rate": 2.6015703695499867e-06,
      "loss": 0.0575,
      "step": 3339
    },
    {
      "epoch": 2.4024456033087573,
      "grad_norm": 2.5373263776325343,
      "learning_rate": 2.6013394041160094e-06,
      "loss": 0.0529,
      "step": 3340
    },
    {
      "epoch": 2.4031648983995684,
      "grad_norm": 3.969533419084874,
      "learning_rate": 2.601108382016186e-06,
      "loss": 0.0874,
      "step": 3341
    },
    {
      "epoch": 2.4038841934903794,
      "grad_norm": 3.004498988729512,
      "learning_rate": 2.600877303262404e-06,
      "loss": 0.1078,
      "step": 3342
    },
    {
      "epoch": 2.4046034885811904,
      "grad_norm": 1.196434756477466,
      "learning_rate": 2.600646167866552e-06,
      "loss": 0.0032,
      "step": 3343
    },
    {
      "epoch": 2.4053227836720015,
      "grad_norm": 2.9129363951437566,
      "learning_rate": 2.6004149758405226e-06,
      "loss": 0.0223,
      "step": 3344
    },
    {
      "epoch": 2.4060420787628125,
      "grad_norm": 2.1768567043038445,
      "learning_rate": 2.60018372719621e-06,
      "loss": 0.0426,
      "step": 3345
    },
    {
      "epoch": 2.4067613738536235,
      "grad_norm": 1.3644177442622671,
      "learning_rate": 2.5999524219455147e-06,
      "loss": 0.0443,
      "step": 3346
    },
    {
      "epoch": 2.4074806689444346,
      "grad_norm": 4.841823896802837,
      "learning_rate": 2.5997210601003357e-06,
      "loss": 0.0668,
      "step": 3347
    },
    {
      "epoch": 2.4081999640352456,
      "grad_norm": 1.986431198879765,
      "learning_rate": 2.599489641672578e-06,
      "loss": 0.0057,
      "step": 3348
    },
    {
      "epoch": 2.4089192591260566,
      "grad_norm": 0.10347057383346972,
      "learning_rate": 2.599258166674148e-06,
      "loss": 0.0008,
      "step": 3349
    },
    {
      "epoch": 2.4096385542168672,
      "grad_norm": 3.195216799238045,
      "learning_rate": 2.5990266351169555e-06,
      "loss": 0.0861,
      "step": 3350
    },
    {
      "epoch": 2.4103578493076787,
      "grad_norm": 3.543421611520562,
      "learning_rate": 2.598795047012913e-06,
      "loss": 0.1782,
      "step": 3351
    },
    {
      "epoch": 2.4110771443984893,
      "grad_norm": 2.5107469882176856,
      "learning_rate": 2.598563402373937e-06,
      "loss": 0.0421,
      "step": 3352
    },
    {
      "epoch": 2.4117964394893003,
      "grad_norm": 1.8087245629594786,
      "learning_rate": 2.5983317012119457e-06,
      "loss": 0.0421,
      "step": 3353
    },
    {
      "epoch": 2.4125157345801114,
      "grad_norm": 0.8696523989624212,
      "learning_rate": 2.59809994353886e-06,
      "loss": 0.0074,
      "step": 3354
    },
    {
      "epoch": 2.4132350296709224,
      "grad_norm": 5.185922568234041,
      "learning_rate": 2.5978681293666044e-06,
      "loss": 0.1517,
      "step": 3355
    },
    {
      "epoch": 2.4139543247617334,
      "grad_norm": 3.288360429377135,
      "learning_rate": 2.5976362587071064e-06,
      "loss": 0.1172,
      "step": 3356
    },
    {
      "epoch": 2.4146736198525445,
      "grad_norm": 5.316650332879753,
      "learning_rate": 2.597404331572296e-06,
      "loss": 0.087,
      "step": 3357
    },
    {
      "epoch": 2.4153929149433555,
      "grad_norm": 3.4457847770267724,
      "learning_rate": 2.5971723479741066e-06,
      "loss": 0.087,
      "step": 3358
    },
    {
      "epoch": 2.4161122100341665,
      "grad_norm": 3.91358756568543,
      "learning_rate": 2.5969403079244732e-06,
      "loss": 0.0395,
      "step": 3359
    },
    {
      "epoch": 2.4168315051249776,
      "grad_norm": 2.3142443316994696,
      "learning_rate": 2.5967082114353363e-06,
      "loss": 0.0491,
      "step": 3360
    },
    {
      "epoch": 2.4175508002157886,
      "grad_norm": 3.9008660445679424,
      "learning_rate": 2.596476058518636e-06,
      "loss": 0.1718,
      "step": 3361
    },
    {
      "epoch": 2.4182700953065996,
      "grad_norm": 3.5717885099215683,
      "learning_rate": 2.596243849186318e-06,
      "loss": 0.0752,
      "step": 3362
    },
    {
      "epoch": 2.4189893903974107,
      "grad_norm": 9.375595050626663,
      "learning_rate": 2.5960115834503293e-06,
      "loss": 0.0237,
      "step": 3363
    },
    {
      "epoch": 2.4197086854882217,
      "grad_norm": 2.87364536482396,
      "learning_rate": 2.5957792613226207e-06,
      "loss": 0.0441,
      "step": 3364
    },
    {
      "epoch": 2.4204279805790323,
      "grad_norm": 3.288896594087049,
      "learning_rate": 2.5955468828151455e-06,
      "loss": 0.109,
      "step": 3365
    },
    {
      "epoch": 2.421147275669844,
      "grad_norm": 2.423450768463097,
      "learning_rate": 2.5953144479398605e-06,
      "loss": 0.0359,
      "step": 3366
    },
    {
      "epoch": 2.4218665707606544,
      "grad_norm": 4.743707451739019,
      "learning_rate": 2.5950819567087236e-06,
      "loss": 0.1116,
      "step": 3367
    },
    {
      "epoch": 2.4225858658514654,
      "grad_norm": 3.2977940980205696,
      "learning_rate": 2.5948494091336983e-06,
      "loss": 0.0195,
      "step": 3368
    },
    {
      "epoch": 2.4233051609422764,
      "grad_norm": 6.405287817605597,
      "learning_rate": 2.594616805226749e-06,
      "loss": 0.0533,
      "step": 3369
    },
    {
      "epoch": 2.4240244560330875,
      "grad_norm": 4.082338278555258,
      "learning_rate": 2.594384144999843e-06,
      "loss": 0.1438,
      "step": 3370
    },
    {
      "epoch": 2.4247437511238985,
      "grad_norm": 2.1280398269173126,
      "learning_rate": 2.5941514284649517e-06,
      "loss": 0.0128,
      "step": 3371
    },
    {
      "epoch": 2.4254630462147095,
      "grad_norm": 1.7504449581928783,
      "learning_rate": 2.593918655634049e-06,
      "loss": 0.0314,
      "step": 3372
    },
    {
      "epoch": 2.4261823413055206,
      "grad_norm": 5.407934457696997,
      "learning_rate": 2.5936858265191112e-06,
      "loss": 0.1324,
      "step": 3373
    },
    {
      "epoch": 2.4269016363963316,
      "grad_norm": 6.1129463466402685,
      "learning_rate": 2.5934529411321173e-06,
      "loss": 0.0471,
      "step": 3374
    },
    {
      "epoch": 2.4276209314871426,
      "grad_norm": 1.3093890673171242,
      "learning_rate": 2.5932199994850503e-06,
      "loss": 0.015,
      "step": 3375
    },
    {
      "epoch": 2.4283402265779537,
      "grad_norm": 2.750444920834696,
      "learning_rate": 2.5929870015898955e-06,
      "loss": 0.0895,
      "step": 3376
    },
    {
      "epoch": 2.4290595216687647,
      "grad_norm": 3.4624809219481567,
      "learning_rate": 2.5927539474586407e-06,
      "loss": 0.0205,
      "step": 3377
    },
    {
      "epoch": 2.4297788167595757,
      "grad_norm": 6.362491284606588,
      "learning_rate": 2.5925208371032764e-06,
      "loss": 0.0884,
      "step": 3378
    },
    {
      "epoch": 2.4304981118503868,
      "grad_norm": 8.514175262125423,
      "learning_rate": 2.592287670535798e-06,
      "loss": 0.2373,
      "step": 3379
    },
    {
      "epoch": 2.431217406941198,
      "grad_norm": 2.3367082217991233,
      "learning_rate": 2.592054447768201e-06,
      "loss": 0.0362,
      "step": 3380
    },
    {
      "epoch": 2.431936702032009,
      "grad_norm": 7.921066853596047,
      "learning_rate": 2.591821168812486e-06,
      "loss": 0.0742,
      "step": 3381
    },
    {
      "epoch": 2.4326559971228194,
      "grad_norm": 4.259824305418047,
      "learning_rate": 2.5915878336806546e-06,
      "loss": 0.0615,
      "step": 3382
    },
    {
      "epoch": 2.4333752922136305,
      "grad_norm": 3.7321459764353127,
      "learning_rate": 2.5913544423847135e-06,
      "loss": 0.0887,
      "step": 3383
    },
    {
      "epoch": 2.4340945873044415,
      "grad_norm": 3.062307487019876,
      "learning_rate": 2.5911209949366704e-06,
      "loss": 0.1347,
      "step": 3384
    },
    {
      "epoch": 2.4348138823952525,
      "grad_norm": 1.6440660304177186,
      "learning_rate": 2.5908874913485364e-06,
      "loss": 0.0516,
      "step": 3385
    },
    {
      "epoch": 2.4355331774860636,
      "grad_norm": 3.0905183111129633,
      "learning_rate": 2.5906539316323265e-06,
      "loss": 0.0684,
      "step": 3386
    },
    {
      "epoch": 2.4362524725768746,
      "grad_norm": 3.850520551711745,
      "learning_rate": 2.590420315800057e-06,
      "loss": 0.0264,
      "step": 3387
    },
    {
      "epoch": 2.4369717676676856,
      "grad_norm": 4.065710165088571,
      "learning_rate": 2.5901866438637484e-06,
      "loss": 0.0985,
      "step": 3388
    },
    {
      "epoch": 2.4376910627584967,
      "grad_norm": 6.097092239218359,
      "learning_rate": 2.589952915835423e-06,
      "loss": 0.0885,
      "step": 3389
    },
    {
      "epoch": 2.4384103578493077,
      "grad_norm": 2.5285060496203147,
      "learning_rate": 2.5897191317271062e-06,
      "loss": 0.0565,
      "step": 3390
    },
    {
      "epoch": 2.4391296529401187,
      "grad_norm": 5.327229582029605,
      "learning_rate": 2.5894852915508277e-06,
      "loss": 0.1358,
      "step": 3391
    },
    {
      "epoch": 2.4398489480309298,
      "grad_norm": 3.789754346722222,
      "learning_rate": 2.589251395318618e-06,
      "loss": 0.0603,
      "step": 3392
    },
    {
      "epoch": 2.440568243121741,
      "grad_norm": 4.372608287586437,
      "learning_rate": 2.5890174430425126e-06,
      "loss": 0.0304,
      "step": 3393
    },
    {
      "epoch": 2.441287538212552,
      "grad_norm": 5.267081450191482,
      "learning_rate": 2.5887834347345473e-06,
      "loss": 0.0488,
      "step": 3394
    },
    {
      "epoch": 2.442006833303363,
      "grad_norm": 5.036738375049488,
      "learning_rate": 2.5885493704067635e-06,
      "loss": 0.1051,
      "step": 3395
    },
    {
      "epoch": 2.442726128394174,
      "grad_norm": 6.614851470908532,
      "learning_rate": 2.588315250071203e-06,
      "loss": 0.182,
      "step": 3396
    },
    {
      "epoch": 2.4434454234849845,
      "grad_norm": 3.506279227999809,
      "learning_rate": 2.588081073739913e-06,
      "loss": 0.0799,
      "step": 3397
    },
    {
      "epoch": 2.4441647185757955,
      "grad_norm": 0.8487522398469651,
      "learning_rate": 2.5878468414249416e-06,
      "loss": 0.0022,
      "step": 3398
    },
    {
      "epoch": 2.4448840136666066,
      "grad_norm": 4.75268044529781,
      "learning_rate": 2.5876125531383405e-06,
      "loss": 0.0647,
      "step": 3399
    },
    {
      "epoch": 2.4456033087574176,
      "grad_norm": 1.9494331968068976,
      "learning_rate": 2.5873782088921647e-06,
      "loss": 0.0365,
      "step": 3400
    },
    {
      "epoch": 2.4463226038482286,
      "grad_norm": 3.5028073114148173,
      "learning_rate": 2.5871438086984704e-06,
      "loss": 0.1169,
      "step": 3401
    },
    {
      "epoch": 2.4470418989390397,
      "grad_norm": 7.151831254856234,
      "learning_rate": 2.5869093525693194e-06,
      "loss": 0.1337,
      "step": 3402
    },
    {
      "epoch": 2.4477611940298507,
      "grad_norm": 2.1325537963916337,
      "learning_rate": 2.5866748405167743e-06,
      "loss": 0.0062,
      "step": 3403
    },
    {
      "epoch": 2.4484804891206617,
      "grad_norm": 3.1658495566572937,
      "learning_rate": 2.5864402725529002e-06,
      "loss": 0.1524,
      "step": 3404
    },
    {
      "epoch": 2.4491997842114728,
      "grad_norm": 2.449317908627163,
      "learning_rate": 2.586205648689768e-06,
      "loss": 0.0784,
      "step": 3405
    },
    {
      "epoch": 2.449919079302284,
      "grad_norm": 2.8773663106903467,
      "learning_rate": 2.5859709689394477e-06,
      "loss": 0.0751,
      "step": 3406
    },
    {
      "epoch": 2.450638374393095,
      "grad_norm": 4.412699307329153,
      "learning_rate": 2.5857362333140147e-06,
      "loss": 0.0961,
      "step": 3407
    },
    {
      "epoch": 2.451357669483906,
      "grad_norm": 0.9175684358436597,
      "learning_rate": 2.585501441825547e-06,
      "loss": 0.0016,
      "step": 3408
    },
    {
      "epoch": 2.452076964574717,
      "grad_norm": 3.7052084265086234,
      "learning_rate": 2.585266594486124e-06,
      "loss": 0.0676,
      "step": 3409
    },
    {
      "epoch": 2.452796259665528,
      "grad_norm": 3.71279159033265,
      "learning_rate": 2.5850316913078297e-06,
      "loss": 0.02,
      "step": 3410
    },
    {
      "epoch": 2.453515554756339,
      "grad_norm": 2.2857494777406204,
      "learning_rate": 2.58479673230275e-06,
      "loss": 0.0719,
      "step": 3411
    },
    {
      "epoch": 2.4542348498471496,
      "grad_norm": 4.196155481203113,
      "learning_rate": 2.584561717482975e-06,
      "loss": 0.1438,
      "step": 3412
    },
    {
      "epoch": 2.454954144937961,
      "grad_norm": 4.694625183649557,
      "learning_rate": 2.5843266468605946e-06,
      "loss": 0.1425,
      "step": 3413
    },
    {
      "epoch": 2.4556734400287716,
      "grad_norm": 4.401462365748412,
      "learning_rate": 2.5840915204477057e-06,
      "loss": 0.1802,
      "step": 3414
    },
    {
      "epoch": 2.4563927351195827,
      "grad_norm": 5.637995206520162,
      "learning_rate": 2.5838563382564042e-06,
      "loss": 0.0715,
      "step": 3415
    },
    {
      "epoch": 2.4571120302103937,
      "grad_norm": 0.2422150793303434,
      "learning_rate": 2.583621100298792e-06,
      "loss": 0.0009,
      "step": 3416
    },
    {
      "epoch": 2.4578313253012047,
      "grad_norm": 4.403015170691564,
      "learning_rate": 2.583385806586972e-06,
      "loss": 0.1129,
      "step": 3417
    },
    {
      "epoch": 2.4585506203920158,
      "grad_norm": 1.161711401840374,
      "learning_rate": 2.58315045713305e-06,
      "loss": 0.0022,
      "step": 3418
    },
    {
      "epoch": 2.459269915482827,
      "grad_norm": 4.644936655558416,
      "learning_rate": 2.582915051949136e-06,
      "loss": 0.0829,
      "step": 3419
    },
    {
      "epoch": 2.459989210573638,
      "grad_norm": 4.809733004354658,
      "learning_rate": 2.5826795910473416e-06,
      "loss": 0.0883,
      "step": 3420
    },
    {
      "epoch": 2.460708505664449,
      "grad_norm": 3.253446473659127,
      "learning_rate": 2.5824440744397812e-06,
      "loss": 0.0664,
      "step": 3421
    },
    {
      "epoch": 2.46142780075526,
      "grad_norm": 7.668360686525356,
      "learning_rate": 2.5822085021385735e-06,
      "loss": 0.221,
      "step": 3422
    },
    {
      "epoch": 2.462147095846071,
      "grad_norm": 7.039733166596783,
      "learning_rate": 2.581972874155838e-06,
      "loss": 0.0302,
      "step": 3423
    },
    {
      "epoch": 2.462866390936882,
      "grad_norm": 1.3166476076117712,
      "learning_rate": 2.5817371905036994e-06,
      "loss": 0.0042,
      "step": 3424
    },
    {
      "epoch": 2.463585686027693,
      "grad_norm": 9.798752074651636,
      "learning_rate": 2.5815014511942827e-06,
      "loss": 0.1244,
      "step": 3425
    },
    {
      "epoch": 2.464304981118504,
      "grad_norm": 7.067742758794416,
      "learning_rate": 2.5812656562397185e-06,
      "loss": 0.1939,
      "step": 3426
    },
    {
      "epoch": 2.4650242762093146,
      "grad_norm": 5.559221270898104,
      "learning_rate": 2.581029805652138e-06,
      "loss": 0.2307,
      "step": 3427
    },
    {
      "epoch": 2.465743571300126,
      "grad_norm": 3.596524679933892,
      "learning_rate": 2.580793899443676e-06,
      "loss": 0.1125,
      "step": 3428
    },
    {
      "epoch": 2.4664628663909367,
      "grad_norm": 5.028566565099954,
      "learning_rate": 2.5805579376264716e-06,
      "loss": 0.1277,
      "step": 3429
    },
    {
      "epoch": 2.4671821614817477,
      "grad_norm": 4.961126121193562,
      "learning_rate": 2.5803219202126633e-06,
      "loss": 0.2435,
      "step": 3430
    },
    {
      "epoch": 2.4679014565725588,
      "grad_norm": 4.2849133828681385,
      "learning_rate": 2.5800858472143964e-06,
      "loss": 0.1018,
      "step": 3431
    },
    {
      "epoch": 2.46862075166337,
      "grad_norm": 4.244787306340676,
      "learning_rate": 2.579849718643816e-06,
      "loss": 0.1088,
      "step": 3432
    },
    {
      "epoch": 2.469340046754181,
      "grad_norm": 1.6101625533924997,
      "learning_rate": 2.579613534513073e-06,
      "loss": 0.0209,
      "step": 3433
    },
    {
      "epoch": 2.470059341844992,
      "grad_norm": 1.5374963902010577,
      "learning_rate": 2.5793772948343175e-06,
      "loss": 0.0025,
      "step": 3434
    },
    {
      "epoch": 2.470778636935803,
      "grad_norm": 4.73898260240712,
      "learning_rate": 2.5791409996197056e-06,
      "loss": 0.0694,
      "step": 3435
    },
    {
      "epoch": 2.471497932026614,
      "grad_norm": 3.4592227554468855,
      "learning_rate": 2.578904648881395e-06,
      "loss": 0.0908,
      "step": 3436
    },
    {
      "epoch": 2.472217227117425,
      "grad_norm": 2.872259587967067,
      "learning_rate": 2.5786682426315463e-06,
      "loss": 0.0576,
      "step": 3437
    },
    {
      "epoch": 2.472936522208236,
      "grad_norm": 3.8494832459425736,
      "learning_rate": 2.578431780882323e-06,
      "loss": 0.0843,
      "step": 3438
    },
    {
      "epoch": 2.473655817299047,
      "grad_norm": 7.9149608087806,
      "learning_rate": 2.5781952636458914e-06,
      "loss": 0.1545,
      "step": 3439
    },
    {
      "epoch": 2.474375112389858,
      "grad_norm": 3.767159575672662,
      "learning_rate": 2.577958690934421e-06,
      "loss": 0.1623,
      "step": 3440
    },
    {
      "epoch": 2.475094407480669,
      "grad_norm": 2.568868154022677,
      "learning_rate": 2.577722062760083e-06,
      "loss": 0.0649,
      "step": 3441
    },
    {
      "epoch": 2.4758137025714797,
      "grad_norm": 2.89387852458236,
      "learning_rate": 2.577485379135054e-06,
      "loss": 0.0783,
      "step": 3442
    },
    {
      "epoch": 2.476532997662291,
      "grad_norm": 2.8730866278137426,
      "learning_rate": 2.57724864007151e-06,
      "loss": 0.0619,
      "step": 3443
    },
    {
      "epoch": 2.4772522927531018,
      "grad_norm": 7.879335102715933,
      "learning_rate": 2.5770118455816323e-06,
      "loss": 0.0585,
      "step": 3444
    },
    {
      "epoch": 2.477971587843913,
      "grad_norm": 3.8749099118810038,
      "learning_rate": 2.576774995677605e-06,
      "loss": 0.0965,
      "step": 3445
    },
    {
      "epoch": 2.478690882934724,
      "grad_norm": 4.033621101909721,
      "learning_rate": 2.5765380903716134e-06,
      "loss": 0.1124,
      "step": 3446
    },
    {
      "epoch": 2.479410178025535,
      "grad_norm": 4.765948368994288,
      "learning_rate": 2.576301129675848e-06,
      "loss": 0.1809,
      "step": 3447
    },
    {
      "epoch": 2.480129473116346,
      "grad_norm": 3.689093766643328,
      "learning_rate": 2.5760641136024996e-06,
      "loss": 0.1232,
      "step": 3448
    },
    {
      "epoch": 2.480848768207157,
      "grad_norm": 3.3345309188009136,
      "learning_rate": 2.5758270421637637e-06,
      "loss": 0.0344,
      "step": 3449
    },
    {
      "epoch": 2.481568063297968,
      "grad_norm": 5.93760728144874,
      "learning_rate": 2.575589915371838e-06,
      "loss": 0.1048,
      "step": 3450
    },
    {
      "epoch": 2.482287358388779,
      "grad_norm": 4.8695160974900435,
      "learning_rate": 2.575352733238923e-06,
      "loss": 0.2506,
      "step": 3451
    },
    {
      "epoch": 2.48300665347959,
      "grad_norm": 1.622523380034065,
      "learning_rate": 2.575115495777222e-06,
      "loss": 0.0275,
      "step": 3452
    },
    {
      "epoch": 2.483725948570401,
      "grad_norm": 0.08692547003772613,
      "learning_rate": 2.5748782029989416e-06,
      "loss": 0.0003,
      "step": 3453
    },
    {
      "epoch": 2.484445243661212,
      "grad_norm": 7.743798592859323,
      "learning_rate": 2.5746408549162907e-06,
      "loss": 0.2036,
      "step": 3454
    },
    {
      "epoch": 2.485164538752023,
      "grad_norm": 2.971061650563225,
      "learning_rate": 2.574403451541481e-06,
      "loss": 0.0563,
      "step": 3455
    },
    {
      "epoch": 2.485883833842834,
      "grad_norm": 2.948099268300286,
      "learning_rate": 2.5741659928867285e-06,
      "loss": 0.0156,
      "step": 3456
    },
    {
      "epoch": 2.486603128933645,
      "grad_norm": 0.48934136727654826,
      "learning_rate": 2.5739284789642494e-06,
      "loss": 0.0016,
      "step": 3457
    },
    {
      "epoch": 2.4873224240244562,
      "grad_norm": 0.2350650483100412,
      "learning_rate": 2.5736909097862647e-06,
      "loss": 0.0007,
      "step": 3458
    },
    {
      "epoch": 2.488041719115267,
      "grad_norm": 8.544466973534586,
      "learning_rate": 2.573453285364998e-06,
      "loss": 0.3311,
      "step": 3459
    },
    {
      "epoch": 2.488761014206078,
      "grad_norm": 1.0104081536986096,
      "learning_rate": 2.5732156057126757e-06,
      "loss": 0.0119,
      "step": 3460
    },
    {
      "epoch": 2.489480309296889,
      "grad_norm": 3.692569602083198,
      "learning_rate": 2.572977870841526e-06,
      "loss": 0.1327,
      "step": 3461
    },
    {
      "epoch": 2.4901996043877,
      "grad_norm": 5.525408530482962,
      "learning_rate": 2.572740080763782e-06,
      "loss": 0.2189,
      "step": 3462
    },
    {
      "epoch": 2.490918899478511,
      "grad_norm": 2.956617739630439,
      "learning_rate": 2.5725022354916773e-06,
      "loss": 0.0761,
      "step": 3463
    },
    {
      "epoch": 2.491638194569322,
      "grad_norm": 4.020900199064354,
      "learning_rate": 2.57226433503745e-06,
      "loss": 0.1782,
      "step": 3464
    },
    {
      "epoch": 2.492357489660133,
      "grad_norm": 2.5759569122013466,
      "learning_rate": 2.5720263794133404e-06,
      "loss": 0.054,
      "step": 3465
    },
    {
      "epoch": 2.493076784750944,
      "grad_norm": 2.789070835502873,
      "learning_rate": 2.5717883686315916e-06,
      "loss": 0.0527,
      "step": 3466
    },
    {
      "epoch": 2.493796079841755,
      "grad_norm": 3.625469161888554,
      "learning_rate": 2.5715503027044494e-06,
      "loss": 0.0585,
      "step": 3467
    },
    {
      "epoch": 2.494515374932566,
      "grad_norm": 1.34999479950769,
      "learning_rate": 2.5713121816441634e-06,
      "loss": 0.0042,
      "step": 3468
    },
    {
      "epoch": 2.495234670023377,
      "grad_norm": 3.7664437012124705,
      "learning_rate": 2.571074005462985e-06,
      "loss": 0.1147,
      "step": 3469
    },
    {
      "epoch": 2.495953965114188,
      "grad_norm": 2.311293013704647,
      "learning_rate": 2.5708357741731687e-06,
      "loss": 0.0706,
      "step": 3470
    },
    {
      "epoch": 2.4966732602049992,
      "grad_norm": 2.36359780466598,
      "learning_rate": 2.5705974877869723e-06,
      "loss": 0.0592,
      "step": 3471
    },
    {
      "epoch": 2.4973925552958103,
      "grad_norm": 4.831511177319388,
      "learning_rate": 2.570359146316656e-06,
      "loss": 0.1229,
      "step": 3472
    },
    {
      "epoch": 2.4981118503866213,
      "grad_norm": 4.685474798416406,
      "learning_rate": 2.570120749774482e-06,
      "loss": 0.1808,
      "step": 3473
    },
    {
      "epoch": 2.498831145477432,
      "grad_norm": 2.964450738747615,
      "learning_rate": 2.5698822981727173e-06,
      "loss": 0.1165,
      "step": 3474
    },
    {
      "epoch": 2.499550440568243,
      "grad_norm": 1.8199873753847282,
      "learning_rate": 2.5696437915236303e-06,
      "loss": 0.0557,
      "step": 3475
    },
    {
      "epoch": 2.500269735659054,
      "grad_norm": 3.637782794830298,
      "learning_rate": 2.5694052298394924e-06,
      "loss": 0.0701,
      "step": 3476
    },
    {
      "epoch": 2.500989030749865,
      "grad_norm": 5.478374569787154,
      "learning_rate": 2.569166613132578e-06,
      "loss": 0.0733,
      "step": 3477
    },
    {
      "epoch": 2.501708325840676,
      "grad_norm": 2.090721205101083,
      "learning_rate": 2.5689279414151644e-06,
      "loss": 0.0494,
      "step": 3478
    },
    {
      "epoch": 2.502427620931487,
      "grad_norm": 2.428853089569013,
      "learning_rate": 2.568689214699532e-06,
      "loss": 0.0583,
      "step": 3479
    },
    {
      "epoch": 2.503146916022298,
      "grad_norm": 5.760256976194934,
      "learning_rate": 2.5684504329979635e-06,
      "loss": 0.0966,
      "step": 3480
    },
    {
      "epoch": 2.503866211113109,
      "grad_norm": 5.131107741427995,
      "learning_rate": 2.5682115963227445e-06,
      "loss": 0.0426,
      "step": 3481
    },
    {
      "epoch": 2.50458550620392,
      "grad_norm": 2.831223196978667,
      "learning_rate": 2.567972704686164e-06,
      "loss": 0.0337,
      "step": 3482
    },
    {
      "epoch": 2.505304801294731,
      "grad_norm": 3.4787747516993868,
      "learning_rate": 2.5677337581005126e-06,
      "loss": 0.1199,
      "step": 3483
    },
    {
      "epoch": 2.5060240963855422,
      "grad_norm": 2.7148804193958016,
      "learning_rate": 2.5674947565780856e-06,
      "loss": 0.091,
      "step": 3484
    },
    {
      "epoch": 2.5067433914763533,
      "grad_norm": 4.723520996312077,
      "learning_rate": 2.567255700131179e-06,
      "loss": 0.2543,
      "step": 3485
    },
    {
      "epoch": 2.5074626865671643,
      "grad_norm": 4.047217775691797,
      "learning_rate": 2.567016588772093e-06,
      "loss": 0.0724,
      "step": 3486
    },
    {
      "epoch": 2.5081819816579753,
      "grad_norm": 5.086265238121583,
      "learning_rate": 2.56677742251313e-06,
      "loss": 0.1988,
      "step": 3487
    },
    {
      "epoch": 2.5089012767487864,
      "grad_norm": 1.4415772066784551,
      "learning_rate": 2.566538201366597e-06,
      "loss": 0.048,
      "step": 3488
    },
    {
      "epoch": 2.509620571839597,
      "grad_norm": 3.5788341161110004,
      "learning_rate": 2.566298925344801e-06,
      "loss": 0.0518,
      "step": 3489
    },
    {
      "epoch": 2.5103398669304084,
      "grad_norm": 2.4690328057732307,
      "learning_rate": 2.5660595944600533e-06,
      "loss": 0.106,
      "step": 3490
    },
    {
      "epoch": 2.511059162021219,
      "grad_norm": 2.7122239813943496,
      "learning_rate": 2.565820208724668e-06,
      "loss": 0.0858,
      "step": 3491
    },
    {
      "epoch": 2.51177845711203,
      "grad_norm": 1.432113821623591,
      "learning_rate": 2.565580768150962e-06,
      "loss": 0.0092,
      "step": 3492
    },
    {
      "epoch": 2.512497752202841,
      "grad_norm": 5.811480143422191,
      "learning_rate": 2.565341272751255e-06,
      "loss": 0.2093,
      "step": 3493
    },
    {
      "epoch": 2.513217047293652,
      "grad_norm": 1.8972828138963855,
      "learning_rate": 2.5651017225378692e-06,
      "loss": 0.0043,
      "step": 3494
    },
    {
      "epoch": 2.513936342384463,
      "grad_norm": 8.60955408438331,
      "learning_rate": 2.5648621175231304e-06,
      "loss": 0.1093,
      "step": 3495
    },
    {
      "epoch": 2.514655637475274,
      "grad_norm": 5.175124785751448,
      "learning_rate": 2.564622457719366e-06,
      "loss": 0.1054,
      "step": 3496
    },
    {
      "epoch": 2.5153749325660852,
      "grad_norm": 4.668449988014636,
      "learning_rate": 2.564382743138907e-06,
      "loss": 0.24,
      "step": 3497
    },
    {
      "epoch": 2.5160942276568963,
      "grad_norm": 5.170177622199164,
      "learning_rate": 2.564142973794088e-06,
      "loss": 0.2063,
      "step": 3498
    },
    {
      "epoch": 2.5168135227477073,
      "grad_norm": 2.9388821711037716,
      "learning_rate": 2.563903149697245e-06,
      "loss": 0.0764,
      "step": 3499
    },
    {
      "epoch": 2.5175328178385183,
      "grad_norm": 2.9821945593432058,
      "learning_rate": 2.563663270860717e-06,
      "loss": 0.0143,
      "step": 3500
    },
    {
      "epoch": 2.5182521129293294,
      "grad_norm": 2.98017025394871,
      "learning_rate": 2.5634233372968467e-06,
      "loss": 0.1017,
      "step": 3501
    },
    {
      "epoch": 2.5189714080201404,
      "grad_norm": 2.1879599259132387,
      "learning_rate": 2.5631833490179786e-06,
      "loss": 0.0238,
      "step": 3502
    },
    {
      "epoch": 2.5196907031109514,
      "grad_norm": 2.193676767115623,
      "learning_rate": 2.562943306036461e-06,
      "loss": 0.034,
      "step": 3503
    },
    {
      "epoch": 2.520409998201762,
      "grad_norm": 2.531547326602505,
      "learning_rate": 2.562703208364644e-06,
      "loss": 0.0473,
      "step": 3504
    },
    {
      "epoch": 2.5211292932925735,
      "grad_norm": 4.256083829310361,
      "learning_rate": 2.5624630560148826e-06,
      "loss": 0.0634,
      "step": 3505
    },
    {
      "epoch": 2.521848588383384,
      "grad_norm": 3.824612074522197,
      "learning_rate": 2.5622228489995307e-06,
      "loss": 0.1111,
      "step": 3506
    },
    {
      "epoch": 2.522567883474195,
      "grad_norm": 3.566015620133227,
      "learning_rate": 2.561982587330949e-06,
      "loss": 0.1917,
      "step": 3507
    },
    {
      "epoch": 2.523287178565006,
      "grad_norm": 4.52746314624229,
      "learning_rate": 2.5617422710214986e-06,
      "loss": 0.054,
      "step": 3508
    },
    {
      "epoch": 2.524006473655817,
      "grad_norm": 3.674946033888536,
      "learning_rate": 2.561501900083545e-06,
      "loss": 0.0318,
      "step": 3509
    },
    {
      "epoch": 2.5247257687466282,
      "grad_norm": 2.9951051769960966,
      "learning_rate": 2.561261474529455e-06,
      "loss": 0.0351,
      "step": 3510
    },
    {
      "epoch": 2.5254450638374393,
      "grad_norm": 3.301985966737383,
      "learning_rate": 2.561020994371599e-06,
      "loss": 0.0582,
      "step": 3511
    },
    {
      "epoch": 2.5261643589282503,
      "grad_norm": 2.4413480614112237,
      "learning_rate": 2.5607804596223503e-06,
      "loss": 0.044,
      "step": 3512
    },
    {
      "epoch": 2.5268836540190613,
      "grad_norm": 2.1213398118727937,
      "learning_rate": 2.5605398702940855e-06,
      "loss": 0.0701,
      "step": 3513
    },
    {
      "epoch": 2.5276029491098724,
      "grad_norm": 1.4305166774404456,
      "learning_rate": 2.560299226399182e-06,
      "loss": 0.0021,
      "step": 3514
    },
    {
      "epoch": 2.5283222442006834,
      "grad_norm": 2.2873225424740293,
      "learning_rate": 2.560058527950022e-06,
      "loss": 0.0548,
      "step": 3515
    },
    {
      "epoch": 2.5290415392914944,
      "grad_norm": 4.070064242254187,
      "learning_rate": 2.55981777495899e-06,
      "loss": 0.0789,
      "step": 3516
    },
    {
      "epoch": 2.5297608343823055,
      "grad_norm": 0.3847651824752757,
      "learning_rate": 2.5595769674384725e-06,
      "loss": 0.0015,
      "step": 3517
    },
    {
      "epoch": 2.5304801294731165,
      "grad_norm": 5.730510961371827,
      "learning_rate": 2.559336105400861e-06,
      "loss": 0.2363,
      "step": 3518
    },
    {
      "epoch": 2.531199424563927,
      "grad_norm": 8.84048825219756,
      "learning_rate": 2.5590951888585468e-06,
      "loss": 0.1629,
      "step": 3519
    },
    {
      "epoch": 2.5319187196547386,
      "grad_norm": 0.7057526693714911,
      "learning_rate": 2.558854217823926e-06,
      "loss": 0.0015,
      "step": 3520
    },
    {
      "epoch": 2.532638014745549,
      "grad_norm": 2.8979907444052038,
      "learning_rate": 2.5586131923093965e-06,
      "loss": 0.0778,
      "step": 3521
    },
    {
      "epoch": 2.53335730983636,
      "grad_norm": 5.293105278693575,
      "learning_rate": 2.5583721123273604e-06,
      "loss": 0.0422,
      "step": 3522
    },
    {
      "epoch": 2.5340766049271712,
      "grad_norm": 3.8822705868467535,
      "learning_rate": 2.558130977890221e-06,
      "loss": 0.0692,
      "step": 3523
    },
    {
      "epoch": 2.5347959000179823,
      "grad_norm": 4.027518531992648,
      "learning_rate": 2.5578897890103857e-06,
      "loss": 0.2075,
      "step": 3524
    },
    {
      "epoch": 2.5355151951087933,
      "grad_norm": 5.611138962024131,
      "learning_rate": 2.557648545700263e-06,
      "loss": 0.1925,
      "step": 3525
    },
    {
      "epoch": 2.5362344901996043,
      "grad_norm": 5.283192825036582,
      "learning_rate": 2.5574072479722663e-06,
      "loss": 0.1808,
      "step": 3526
    },
    {
      "epoch": 2.5369537852904154,
      "grad_norm": 3.2139419240967677,
      "learning_rate": 2.5571658958388103e-06,
      "loss": 0.086,
      "step": 3527
    },
    {
      "epoch": 2.5376730803812264,
      "grad_norm": 1.60526938420732,
      "learning_rate": 2.5569244893123136e-06,
      "loss": 0.04,
      "step": 3528
    },
    {
      "epoch": 2.5383923754720374,
      "grad_norm": 3.818629682478443,
      "learning_rate": 2.5566830284051963e-06,
      "loss": 0.0781,
      "step": 3529
    },
    {
      "epoch": 2.5391116705628485,
      "grad_norm": 12.924028996405493,
      "learning_rate": 2.5564415131298823e-06,
      "loss": 0.207,
      "step": 3530
    },
    {
      "epoch": 2.5398309656536595,
      "grad_norm": 1.4297078160043157,
      "learning_rate": 2.556199943498798e-06,
      "loss": 0.027,
      "step": 3531
    },
    {
      "epoch": 2.5405502607444705,
      "grad_norm": 3.0880962668288987,
      "learning_rate": 2.5559583195243725e-06,
      "loss": 0.1166,
      "step": 3532
    },
    {
      "epoch": 2.5412695558352816,
      "grad_norm": 7.185202767540575,
      "learning_rate": 2.5557166412190373e-06,
      "loss": 0.1743,
      "step": 3533
    },
    {
      "epoch": 2.541988850926092,
      "grad_norm": 3.743089599588881,
      "learning_rate": 2.5554749085952286e-06,
      "loss": 0.1209,
      "step": 3534
    },
    {
      "epoch": 2.5427081460169036,
      "grad_norm": 5.855815763094652,
      "learning_rate": 2.555233121665382e-06,
      "loss": 0.1971,
      "step": 3535
    },
    {
      "epoch": 2.5434274411077142,
      "grad_norm": 0.8108980233085022,
      "learning_rate": 2.5549912804419397e-06,
      "loss": 0.0077,
      "step": 3536
    },
    {
      "epoch": 2.5441467361985257,
      "grad_norm": 6.867518261027968,
      "learning_rate": 2.5547493849373435e-06,
      "loss": 0.021,
      "step": 3537
    },
    {
      "epoch": 2.5448660312893363,
      "grad_norm": 2.1523704520124536,
      "learning_rate": 2.55450743516404e-06,
      "loss": 0.0782,
      "step": 3538
    },
    {
      "epoch": 2.5455853263801473,
      "grad_norm": 4.555569683926216,
      "learning_rate": 2.554265431134478e-06,
      "loss": 0.1947,
      "step": 3539
    },
    {
      "epoch": 2.5463046214709584,
      "grad_norm": 3.025986828250512,
      "learning_rate": 2.5540233728611086e-06,
      "loss": 0.0392,
      "step": 3540
    },
    {
      "epoch": 2.5470239165617694,
      "grad_norm": 4.721736906888111,
      "learning_rate": 2.553781260356387e-06,
      "loss": 0.0864,
      "step": 3541
    },
    {
      "epoch": 2.5477432116525804,
      "grad_norm": 2.898133230771604,
      "learning_rate": 2.553539093632769e-06,
      "loss": 0.0931,
      "step": 3542
    },
    {
      "epoch": 2.5484625067433915,
      "grad_norm": 4.790008065763163,
      "learning_rate": 2.5532968727027153e-06,
      "loss": 0.206,
      "step": 3543
    },
    {
      "epoch": 2.5491818018342025,
      "grad_norm": 1.969083972083599,
      "learning_rate": 2.5530545975786888e-06,
      "loss": 0.007,
      "step": 3544
    },
    {
      "epoch": 2.5499010969250135,
      "grad_norm": 2.1324070330305482,
      "learning_rate": 2.5528122682731545e-06,
      "loss": 0.0083,
      "step": 3545
    },
    {
      "epoch": 2.5506203920158246,
      "grad_norm": 1.9699174578661331,
      "learning_rate": 2.552569884798581e-06,
      "loss": 0.037,
      "step": 3546
    },
    {
      "epoch": 2.5513396871066356,
      "grad_norm": 4.775038969293058,
      "learning_rate": 2.5523274471674388e-06,
      "loss": 0.2031,
      "step": 3547
    },
    {
      "epoch": 2.5520589821974466,
      "grad_norm": 1.5812838241336613,
      "learning_rate": 2.5520849553922023e-06,
      "loss": 0.0329,
      "step": 3548
    },
    {
      "epoch": 2.5527782772882572,
      "grad_norm": 6.37634727568618,
      "learning_rate": 2.5518424094853485e-06,
      "loss": 0.1274,
      "step": 3549
    },
    {
      "epoch": 2.5534975723790687,
      "grad_norm": 6.306381648339605,
      "learning_rate": 2.5515998094593556e-06,
      "loss": 0.2527,
      "step": 3550
    },
    {
      "epoch": 2.5542168674698793,
      "grad_norm": 2.061940813255708,
      "learning_rate": 2.551357155326707e-06,
      "loss": 0.0251,
      "step": 3551
    },
    {
      "epoch": 2.554936162560691,
      "grad_norm": 3.3143704393288136,
      "learning_rate": 2.551114447099887e-06,
      "loss": 0.0125,
      "step": 3552
    },
    {
      "epoch": 2.5556554576515014,
      "grad_norm": 2.5576226238416546,
      "learning_rate": 2.550871684791383e-06,
      "loss": 0.0625,
      "step": 3553
    },
    {
      "epoch": 2.5563747527423124,
      "grad_norm": 2.324275910227882,
      "learning_rate": 2.5506288684136865e-06,
      "loss": 0.0491,
      "step": 3554
    },
    {
      "epoch": 2.5570940478331234,
      "grad_norm": 2.1751372224957803,
      "learning_rate": 2.550385997979291e-06,
      "loss": 0.0637,
      "step": 3555
    },
    {
      "epoch": 2.5578133429239345,
      "grad_norm": 5.3770796886050425,
      "learning_rate": 2.5501430735006907e-06,
      "loss": 0.1733,
      "step": 3556
    },
    {
      "epoch": 2.5585326380147455,
      "grad_norm": 6.289112856011835,
      "learning_rate": 2.5499000949903866e-06,
      "loss": 0.1315,
      "step": 3557
    },
    {
      "epoch": 2.5592519331055565,
      "grad_norm": 3.4819626801396617,
      "learning_rate": 2.5496570624608793e-06,
      "loss": 0.0344,
      "step": 3558
    },
    {
      "epoch": 2.5599712281963676,
      "grad_norm": 2.673207582988735,
      "learning_rate": 2.5494139759246738e-06,
      "loss": 0.0975,
      "step": 3559
    },
    {
      "epoch": 2.5606905232871786,
      "grad_norm": 4.6567109007560905,
      "learning_rate": 2.549170835394277e-06,
      "loss": 0.165,
      "step": 3560
    },
    {
      "epoch": 2.5614098183779896,
      "grad_norm": 2.3764473821705425,
      "learning_rate": 2.548927640882199e-06,
      "loss": 0.0521,
      "step": 3561
    },
    {
      "epoch": 2.5621291134688007,
      "grad_norm": 3.401520100863981,
      "learning_rate": 2.5486843924009528e-06,
      "loss": 0.1057,
      "step": 3562
    },
    {
      "epoch": 2.5628484085596117,
      "grad_norm": 3.580039724768081,
      "learning_rate": 2.548441089963053e-06,
      "loss": 0.0448,
      "step": 3563
    },
    {
      "epoch": 2.5635677036504227,
      "grad_norm": 5.1496752627595095,
      "learning_rate": 2.548197733581019e-06,
      "loss": 0.1372,
      "step": 3564
    },
    {
      "epoch": 2.564286998741234,
      "grad_norm": 3.0066521239283763,
      "learning_rate": 2.5479543232673715e-06,
      "loss": 0.0194,
      "step": 3565
    },
    {
      "epoch": 2.5650062938320444,
      "grad_norm": 2.6907702309665376,
      "learning_rate": 2.5477108590346346e-06,
      "loss": 0.0163,
      "step": 3566
    },
    {
      "epoch": 2.565725588922856,
      "grad_norm": 2.952590666270628,
      "learning_rate": 2.5474673408953344e-06,
      "loss": 0.049,
      "step": 3567
    },
    {
      "epoch": 2.5664448840136664,
      "grad_norm": 1.5528305246308567,
      "learning_rate": 2.547223768862001e-06,
      "loss": 0.0258,
      "step": 3568
    },
    {
      "epoch": 2.5671641791044775,
      "grad_norm": 3.2861827407112476,
      "learning_rate": 2.5469801429471667e-06,
      "loss": 0.0542,
      "step": 3569
    },
    {
      "epoch": 2.5678834741952885,
      "grad_norm": 3.987059589041141,
      "learning_rate": 2.546736463163366e-06,
      "loss": 0.1344,
      "step": 3570
    },
    {
      "epoch": 2.5686027692860995,
      "grad_norm": 4.0469800427662985,
      "learning_rate": 2.5464927295231367e-06,
      "loss": 0.0384,
      "step": 3571
    },
    {
      "epoch": 2.5693220643769106,
      "grad_norm": 3.687294745418768,
      "learning_rate": 2.5462489420390187e-06,
      "loss": 0.0492,
      "step": 3572
    },
    {
      "epoch": 2.5700413594677216,
      "grad_norm": 3.350669057530878,
      "learning_rate": 2.546005100723557e-06,
      "loss": 0.0305,
      "step": 3573
    },
    {
      "epoch": 2.5707606545585326,
      "grad_norm": 3.0989712046710984,
      "learning_rate": 2.545761205589296e-06,
      "loss": 0.0598,
      "step": 3574
    },
    {
      "epoch": 2.5714799496493437,
      "grad_norm": 3.1113822826441266,
      "learning_rate": 2.5455172566487857e-06,
      "loss": 0.0717,
      "step": 3575
    },
    {
      "epoch": 2.5721992447401547,
      "grad_norm": 4.247450222425722,
      "learning_rate": 2.545273253914577e-06,
      "loss": 0.0451,
      "step": 3576
    },
    {
      "epoch": 2.5729185398309657,
      "grad_norm": 4.344623508312981,
      "learning_rate": 2.5450291973992245e-06,
      "loss": 0.0803,
      "step": 3577
    },
    {
      "epoch": 2.573637834921777,
      "grad_norm": 6.0443315182528865,
      "learning_rate": 2.544785087115285e-06,
      "loss": 0.2302,
      "step": 3578
    },
    {
      "epoch": 2.574357130012588,
      "grad_norm": 6.532146424687256,
      "learning_rate": 2.5445409230753193e-06,
      "loss": 0.3271,
      "step": 3579
    },
    {
      "epoch": 2.575076425103399,
      "grad_norm": 3.3577906606487242,
      "learning_rate": 2.544296705291889e-06,
      "loss": 0.1773,
      "step": 3580
    },
    {
      "epoch": 2.5757957201942094,
      "grad_norm": 2.2909592260368346,
      "learning_rate": 2.54405243377756e-06,
      "loss": 0.0353,
      "step": 3581
    },
    {
      "epoch": 2.576515015285021,
      "grad_norm": 2.5766040408158126,
      "learning_rate": 2.5438081085449004e-06,
      "loss": 0.0467,
      "step": 3582
    },
    {
      "epoch": 2.5772343103758315,
      "grad_norm": 5.207479191201509,
      "learning_rate": 2.5435637296064816e-06,
      "loss": 0.1597,
      "step": 3583
    },
    {
      "epoch": 2.5779536054666425,
      "grad_norm": 4.418553524417093,
      "learning_rate": 2.543319296974877e-06,
      "loss": 0.1587,
      "step": 3584
    },
    {
      "epoch": 2.5786729005574536,
      "grad_norm": 4.0761017841284595,
      "learning_rate": 2.5430748106626633e-06,
      "loss": 0.1064,
      "step": 3585
    },
    {
      "epoch": 2.5793921956482646,
      "grad_norm": 3.1550199749966246,
      "learning_rate": 2.5428302706824192e-06,
      "loss": 0.0458,
      "step": 3586
    },
    {
      "epoch": 2.5801114907390756,
      "grad_norm": 5.958745985124681,
      "learning_rate": 2.542585677046727e-06,
      "loss": 0.0234,
      "step": 3587
    },
    {
      "epoch": 2.5808307858298867,
      "grad_norm": 2.20615071082052,
      "learning_rate": 2.5423410297681715e-06,
      "loss": 0.0382,
      "step": 3588
    },
    {
      "epoch": 2.5815500809206977,
      "grad_norm": 5.511574152943852,
      "learning_rate": 2.5420963288593403e-06,
      "loss": 0.2206,
      "step": 3589
    },
    {
      "epoch": 2.5822693760115087,
      "grad_norm": 2.4973371450672865,
      "learning_rate": 2.5418515743328234e-06,
      "loss": 0.0186,
      "step": 3590
    },
    {
      "epoch": 2.5829886711023198,
      "grad_norm": 4.51868552850043,
      "learning_rate": 2.5416067662012145e-06,
      "loss": 0.104,
      "step": 3591
    },
    {
      "epoch": 2.583707966193131,
      "grad_norm": 2.3391649051777574,
      "learning_rate": 2.541361904477109e-06,
      "loss": 0.0168,
      "step": 3592
    },
    {
      "epoch": 2.584427261283942,
      "grad_norm": 4.156181341218098,
      "learning_rate": 2.541116989173105e-06,
      "loss": 0.1084,
      "step": 3593
    },
    {
      "epoch": 2.585146556374753,
      "grad_norm": 4.701650207067254,
      "learning_rate": 2.5408720203018044e-06,
      "loss": 0.1242,
      "step": 3594
    },
    {
      "epoch": 2.585865851465564,
      "grad_norm": 1.9496321604718572,
      "learning_rate": 2.5406269978758116e-06,
      "loss": 0.0444,
      "step": 3595
    },
    {
      "epoch": 2.5865851465563745,
      "grad_norm": 3.642812884068933,
      "learning_rate": 2.5403819219077324e-06,
      "loss": 0.0284,
      "step": 3596
    },
    {
      "epoch": 2.587304441647186,
      "grad_norm": 3.717199735116792,
      "learning_rate": 2.5401367924101775e-06,
      "loss": 0.0954,
      "step": 3597
    },
    {
      "epoch": 2.5880237367379966,
      "grad_norm": 3.8250986849617754,
      "learning_rate": 2.5398916093957588e-06,
      "loss": 0.1652,
      "step": 3598
    },
    {
      "epoch": 2.5887430318288076,
      "grad_norm": 3.3550735046669327,
      "learning_rate": 2.539646372877091e-06,
      "loss": 0.1015,
      "step": 3599
    },
    {
      "epoch": 2.5894623269196186,
      "grad_norm": 7.403908730681601,
      "learning_rate": 2.539401082866792e-06,
      "loss": 0.0419,
      "step": 3600
    },
    {
      "epoch": 2.5901816220104297,
      "grad_norm": 4.458507419027894,
      "learning_rate": 2.539155739377483e-06,
      "loss": 0.1317,
      "step": 3601
    },
    {
      "epoch": 2.5909009171012407,
      "grad_norm": 5.411549982522412,
      "learning_rate": 2.5389103424217872e-06,
      "loss": 0.1254,
      "step": 3602
    },
    {
      "epoch": 2.5916202121920517,
      "grad_norm": 5.551914161242035,
      "learning_rate": 2.5386648920123307e-06,
      "loss": 0.1345,
      "step": 3603
    },
    {
      "epoch": 2.5923395072828628,
      "grad_norm": 3.8350715945936757,
      "learning_rate": 2.5384193881617415e-06,
      "loss": 0.0654,
      "step": 3604
    },
    {
      "epoch": 2.593058802373674,
      "grad_norm": 0.7387453956564594,
      "learning_rate": 2.5381738308826526e-06,
      "loss": 0.0078,
      "step": 3605
    },
    {
      "epoch": 2.593778097464485,
      "grad_norm": 4.686956035330403,
      "learning_rate": 2.537928220187697e-06,
      "loss": 0.1908,
      "step": 3606
    },
    {
      "epoch": 2.594497392555296,
      "grad_norm": 3.1634346554714035,
      "learning_rate": 2.537682556089513e-06,
      "loss": 0.1345,
      "step": 3607
    },
    {
      "epoch": 2.595216687646107,
      "grad_norm": 4.155984034833928,
      "learning_rate": 2.5374368386007395e-06,
      "loss": 0.1816,
      "step": 3608
    },
    {
      "epoch": 2.595935982736918,
      "grad_norm": 4.035613536214,
      "learning_rate": 2.5371910677340196e-06,
      "loss": 0.1298,
      "step": 3609
    },
    {
      "epoch": 2.596655277827729,
      "grad_norm": 4.410615262037329,
      "learning_rate": 2.5369452435019984e-06,
      "loss": 0.0967,
      "step": 3610
    },
    {
      "epoch": 2.5973745729185396,
      "grad_norm": 4.398167551036084,
      "learning_rate": 2.5366993659173243e-06,
      "loss": 0.1624,
      "step": 3611
    },
    {
      "epoch": 2.598093868009351,
      "grad_norm": 4.094728965884214,
      "learning_rate": 2.5364534349926477e-06,
      "loss": 0.0854,
      "step": 3612
    },
    {
      "epoch": 2.5988131631001616,
      "grad_norm": 4.6881330585398455,
      "learning_rate": 2.536207450740623e-06,
      "loss": 0.1006,
      "step": 3613
    },
    {
      "epoch": 2.599532458190973,
      "grad_norm": 4.999955757286984,
      "learning_rate": 2.5359614131739057e-06,
      "loss": 0.0947,
      "step": 3614
    },
    {
      "epoch": 2.6002517532817837,
      "grad_norm": 2.2982670107156236,
      "learning_rate": 2.535715322305155e-06,
      "loss": 0.0807,
      "step": 3615
    },
    {
      "epoch": 2.6009710483725947,
      "grad_norm": 3.2221983666217793,
      "learning_rate": 2.535469178147033e-06,
      "loss": 0.1313,
      "step": 3616
    },
    {
      "epoch": 2.6016903434634058,
      "grad_norm": 3.61813131024914,
      "learning_rate": 2.5352229807122037e-06,
      "loss": 0.1815,
      "step": 3617
    },
    {
      "epoch": 2.602409638554217,
      "grad_norm": 3.7769570401644152,
      "learning_rate": 2.5349767300133352e-06,
      "loss": 0.0803,
      "step": 3618
    },
    {
      "epoch": 2.603128933645028,
      "grad_norm": 5.318290462611663,
      "learning_rate": 2.534730426063097e-06,
      "loss": 0.0611,
      "step": 3619
    },
    {
      "epoch": 2.603848228735839,
      "grad_norm": 5.459847473252826,
      "learning_rate": 2.534484068874162e-06,
      "loss": 0.0873,
      "step": 3620
    },
    {
      "epoch": 2.60456752382665,
      "grad_norm": 0.6823841030323171,
      "learning_rate": 2.5342376584592056e-06,
      "loss": 0.0056,
      "step": 3621
    },
    {
      "epoch": 2.605286818917461,
      "grad_norm": 2.684371990325371,
      "learning_rate": 2.5339911948309057e-06,
      "loss": 0.1071,
      "step": 3622
    },
    {
      "epoch": 2.606006114008272,
      "grad_norm": 3.6864551386607904,
      "learning_rate": 2.533744678001944e-06,
      "loss": 0.0622,
      "step": 3623
    },
    {
      "epoch": 2.606725409099083,
      "grad_norm": 1.6205701101000574,
      "learning_rate": 2.533498107985004e-06,
      "loss": 0.0269,
      "step": 3624
    },
    {
      "epoch": 2.607444704189894,
      "grad_norm": 6.042361537084525,
      "learning_rate": 2.5332514847927726e-06,
      "loss": 0.1588,
      "step": 3625
    },
    {
      "epoch": 2.6081639992807046,
      "grad_norm": 3.3956691276029956,
      "learning_rate": 2.533004808437938e-06,
      "loss": 0.0215,
      "step": 3626
    },
    {
      "epoch": 2.608883294371516,
      "grad_norm": 1.8695545031916554,
      "learning_rate": 2.532758078933192e-06,
      "loss": 0.0139,
      "step": 3627
    },
    {
      "epoch": 2.6096025894623267,
      "grad_norm": 5.4890198792463964,
      "learning_rate": 2.532511296291231e-06,
      "loss": 0.1388,
      "step": 3628
    },
    {
      "epoch": 2.610321884553138,
      "grad_norm": 2.54397434864001,
      "learning_rate": 2.532264460524751e-06,
      "loss": 0.0354,
      "step": 3629
    },
    {
      "epoch": 2.6110411796439488,
      "grad_norm": 2.8251835019619747,
      "learning_rate": 2.532017571646452e-06,
      "loss": 0.0334,
      "step": 3630
    },
    {
      "epoch": 2.61176047473476,
      "grad_norm": 6.623189264211329,
      "learning_rate": 2.5317706296690376e-06,
      "loss": 0.0528,
      "step": 3631
    },
    {
      "epoch": 2.612479769825571,
      "grad_norm": 6.746329951477701,
      "learning_rate": 2.531523634605213e-06,
      "loss": 0.1999,
      "step": 3632
    },
    {
      "epoch": 2.613199064916382,
      "grad_norm": 5.7608315074917495,
      "learning_rate": 2.531276586467687e-06,
      "loss": 0.169,
      "step": 3633
    },
    {
      "epoch": 2.613918360007193,
      "grad_norm": 4.979693590182001,
      "learning_rate": 2.53102948526917e-06,
      "loss": 0.2363,
      "step": 3634
    },
    {
      "epoch": 2.614637655098004,
      "grad_norm": 2.9257103235188624,
      "learning_rate": 2.5307823310223756e-06,
      "loss": 0.0452,
      "step": 3635
    },
    {
      "epoch": 2.615356950188815,
      "grad_norm": 2.6623679283048407,
      "learning_rate": 2.530535123740021e-06,
      "loss": 0.0491,
      "step": 3636
    },
    {
      "epoch": 2.616076245279626,
      "grad_norm": 3.7123501371943477,
      "learning_rate": 2.5302878634348255e-06,
      "loss": 0.0608,
      "step": 3637
    },
    {
      "epoch": 2.616795540370437,
      "grad_norm": 3.5322104196983006,
      "learning_rate": 2.5300405501195107e-06,
      "loss": 0.019,
      "step": 3638
    },
    {
      "epoch": 2.617514835461248,
      "grad_norm": 4.9593428058442255,
      "learning_rate": 2.5297931838068013e-06,
      "loss": 0.4764,
      "step": 3639
    },
    {
      "epoch": 2.618234130552059,
      "grad_norm": 5.775091458388362,
      "learning_rate": 2.529545764509425e-06,
      "loss": 0.0493,
      "step": 3640
    },
    {
      "epoch": 2.61895342564287,
      "grad_norm": 5.51211232126548,
      "learning_rate": 2.5292982922401116e-06,
      "loss": 0.1193,
      "step": 3641
    },
    {
      "epoch": 2.619672720733681,
      "grad_norm": 4.183424246739544,
      "learning_rate": 2.5290507670115945e-06,
      "loss": 0.0862,
      "step": 3642
    },
    {
      "epoch": 2.6203920158244918,
      "grad_norm": 6.2689336496148425,
      "learning_rate": 2.528803188836609e-06,
      "loss": 0.2448,
      "step": 3643
    },
    {
      "epoch": 2.6211113109153032,
      "grad_norm": 4.683188815343918,
      "learning_rate": 2.528555557727894e-06,
      "loss": 0.1217,
      "step": 3644
    },
    {
      "epoch": 2.621830606006114,
      "grad_norm": 3.779198395109637,
      "learning_rate": 2.528307873698189e-06,
      "loss": 0.1743,
      "step": 3645
    },
    {
      "epoch": 2.622549901096925,
      "grad_norm": 4.749523536449059,
      "learning_rate": 2.528060136760239e-06,
      "loss": 0.1401,
      "step": 3646
    },
    {
      "epoch": 2.623269196187736,
      "grad_norm": 4.608089140342495,
      "learning_rate": 2.5278123469267903e-06,
      "loss": 0.0204,
      "step": 3647
    },
    {
      "epoch": 2.623988491278547,
      "grad_norm": 0.8244730202703748,
      "learning_rate": 2.5275645042105916e-06,
      "loss": 0.0013,
      "step": 3648
    },
    {
      "epoch": 2.624707786369358,
      "grad_norm": 4.554578329634901,
      "learning_rate": 2.527316608624396e-06,
      "loss": 0.1309,
      "step": 3649
    },
    {
      "epoch": 2.625427081460169,
      "grad_norm": 1.8678742867034965,
      "learning_rate": 2.5270686601809577e-06,
      "loss": 0.0478,
      "step": 3650
    },
    {
      "epoch": 2.62614637655098,
      "grad_norm": 4.25204586476136,
      "learning_rate": 2.526820658893033e-06,
      "loss": 0.0722,
      "step": 3651
    },
    {
      "epoch": 2.626865671641791,
      "grad_norm": 0.7705422900817839,
      "learning_rate": 2.5265726047733837e-06,
      "loss": 0.0067,
      "step": 3652
    },
    {
      "epoch": 2.627584966732602,
      "grad_norm": 4.4015151038859734,
      "learning_rate": 2.526324497834771e-06,
      "loss": 0.2269,
      "step": 3653
    },
    {
      "epoch": 2.628304261823413,
      "grad_norm": 3.428253962330576,
      "learning_rate": 2.5260763380899614e-06,
      "loss": 0.1341,
      "step": 3654
    },
    {
      "epoch": 2.629023556914224,
      "grad_norm": 2.745567761914923,
      "learning_rate": 2.525828125551724e-06,
      "loss": 0.0807,
      "step": 3655
    },
    {
      "epoch": 2.629742852005035,
      "grad_norm": 2.09935528166114,
      "learning_rate": 2.5255798602328276e-06,
      "loss": 0.0307,
      "step": 3656
    },
    {
      "epoch": 2.6304621470958462,
      "grad_norm": 5.762636952307489,
      "learning_rate": 2.5253315421460476e-06,
      "loss": 0.2204,
      "step": 3657
    },
    {
      "epoch": 2.631181442186657,
      "grad_norm": 4.301575920146319,
      "learning_rate": 2.5250831713041596e-06,
      "loss": 0.1691,
      "step": 3658
    },
    {
      "epoch": 2.6319007372774683,
      "grad_norm": 0.165754111986202,
      "learning_rate": 2.524834747719943e-06,
      "loss": 0.0005,
      "step": 3659
    },
    {
      "epoch": 2.632620032368279,
      "grad_norm": 4.2900920314099675,
      "learning_rate": 2.52458627140618e-06,
      "loss": 0.0677,
      "step": 3660
    },
    {
      "epoch": 2.63333932745909,
      "grad_norm": 4.390895562820762,
      "learning_rate": 2.524337742375654e-06,
      "loss": 0.1087,
      "step": 3661
    },
    {
      "epoch": 2.634058622549901,
      "grad_norm": 1.2567750183294795,
      "learning_rate": 2.5240891606411533e-06,
      "loss": 0.0053,
      "step": 3662
    },
    {
      "epoch": 2.634777917640712,
      "grad_norm": 3.5118046317122094,
      "learning_rate": 2.523840526215468e-06,
      "loss": 0.1538,
      "step": 3663
    },
    {
      "epoch": 2.635497212731523,
      "grad_norm": 3.812563811661294,
      "learning_rate": 2.5235918391113895e-06,
      "loss": 0.1311,
      "step": 3664
    },
    {
      "epoch": 2.636216507822334,
      "grad_norm": 2.3299051593230944,
      "learning_rate": 2.523343099341715e-06,
      "loss": 0.0372,
      "step": 3665
    },
    {
      "epoch": 2.636935802913145,
      "grad_norm": 0.5680373022103768,
      "learning_rate": 2.5230943069192407e-06,
      "loss": 0.0047,
      "step": 3666
    },
    {
      "epoch": 2.637655098003956,
      "grad_norm": 0.99127957395232,
      "learning_rate": 2.522845461856769e-06,
      "loss": 0.007,
      "step": 3667
    },
    {
      "epoch": 2.638374393094767,
      "grad_norm": 2.4659282685610586,
      "learning_rate": 2.5225965641671027e-06,
      "loss": 0.0308,
      "step": 3668
    },
    {
      "epoch": 2.639093688185578,
      "grad_norm": 2.5579028925216583,
      "learning_rate": 2.522347613863047e-06,
      "loss": 0.0541,
      "step": 3669
    },
    {
      "epoch": 2.6398129832763892,
      "grad_norm": 4.180844752048521,
      "learning_rate": 2.522098610957413e-06,
      "loss": 0.1575,
      "step": 3670
    },
    {
      "epoch": 2.6405322783672003,
      "grad_norm": 0.7525115756979937,
      "learning_rate": 2.521849555463011e-06,
      "loss": 0.007,
      "step": 3671
    },
    {
      "epoch": 2.6412515734580113,
      "grad_norm": 5.527869819047108,
      "learning_rate": 2.521600447392656e-06,
      "loss": 0.1649,
      "step": 3672
    },
    {
      "epoch": 2.641970868548822,
      "grad_norm": 5.021716658896101,
      "learning_rate": 2.5213512867591637e-06,
      "loss": 0.1074,
      "step": 3673
    },
    {
      "epoch": 2.6426901636396334,
      "grad_norm": 7.113543250843125,
      "learning_rate": 2.521102073575355e-06,
      "loss": 0.1661,
      "step": 3674
    },
    {
      "epoch": 2.643409458730444,
      "grad_norm": 5.486427128211821,
      "learning_rate": 2.5208528078540526e-06,
      "loss": 0.144,
      "step": 3675
    },
    {
      "epoch": 2.644128753821255,
      "grad_norm": 3.7938829008850474,
      "learning_rate": 2.5206034896080806e-06,
      "loss": 0.1485,
      "step": 3676
    },
    {
      "epoch": 2.644848048912066,
      "grad_norm": 3.3037568657850587,
      "learning_rate": 2.5203541188502676e-06,
      "loss": 0.186,
      "step": 3677
    },
    {
      "epoch": 2.645567344002877,
      "grad_norm": 1.5458791820632884,
      "learning_rate": 2.5201046955934444e-06,
      "loss": 0.047,
      "step": 3678
    },
    {
      "epoch": 2.646286639093688,
      "grad_norm": 4.0669746862308465,
      "learning_rate": 2.519855219850444e-06,
      "loss": 0.2249,
      "step": 3679
    },
    {
      "epoch": 2.647005934184499,
      "grad_norm": 3.4681655703951138,
      "learning_rate": 2.5196056916341016e-06,
      "loss": 0.0835,
      "step": 3680
    },
    {
      "epoch": 2.64772522927531,
      "grad_norm": 3.36461425629639,
      "learning_rate": 2.519356110957257e-06,
      "loss": 0.1012,
      "step": 3681
    },
    {
      "epoch": 2.648444524366121,
      "grad_norm": 3.9286969089688384,
      "learning_rate": 2.5191064778327505e-06,
      "loss": 0.1335,
      "step": 3682
    },
    {
      "epoch": 2.6491638194569322,
      "grad_norm": 3.80027168525941,
      "learning_rate": 2.518856792273427e-06,
      "loss": 0.1119,
      "step": 3683
    },
    {
      "epoch": 2.6498831145477433,
      "grad_norm": 6.36085198822989,
      "learning_rate": 2.5186070542921335e-06,
      "loss": 0.0961,
      "step": 3684
    },
    {
      "epoch": 2.6506024096385543,
      "grad_norm": 3.109014723056777,
      "learning_rate": 2.5183572639017182e-06,
      "loss": 0.0108,
      "step": 3685
    },
    {
      "epoch": 2.6513217047293653,
      "grad_norm": 4.080856582961475,
      "learning_rate": 2.5181074211150345e-06,
      "loss": 0.0788,
      "step": 3686
    },
    {
      "epoch": 2.6520409998201764,
      "grad_norm": 4.368595693091915,
      "learning_rate": 2.5178575259449364e-06,
      "loss": 0.1507,
      "step": 3687
    },
    {
      "epoch": 2.652760294910987,
      "grad_norm": 2.069468663756614,
      "learning_rate": 2.517607578404282e-06,
      "loss": 0.0476,
      "step": 3688
    },
    {
      "epoch": 2.6534795900017984,
      "grad_norm": 3.670353653272536,
      "learning_rate": 2.5173575785059314e-06,
      "loss": 0.135,
      "step": 3689
    },
    {
      "epoch": 2.654198885092609,
      "grad_norm": 1.913866513974025,
      "learning_rate": 2.5171075262627475e-06,
      "loss": 0.0452,
      "step": 3690
    },
    {
      "epoch": 2.6549181801834205,
      "grad_norm": 5.38354193408671,
      "learning_rate": 2.5168574216875955e-06,
      "loss": 0.1878,
      "step": 3691
    },
    {
      "epoch": 2.655637475274231,
      "grad_norm": 3.8498479946055157,
      "learning_rate": 2.5166072647933445e-06,
      "loss": 0.1365,
      "step": 3692
    },
    {
      "epoch": 2.656356770365042,
      "grad_norm": 7.240928825153363,
      "learning_rate": 2.516357055592865e-06,
      "loss": 0.1269,
      "step": 3693
    },
    {
      "epoch": 2.657076065455853,
      "grad_norm": 2.8701860407827517,
      "learning_rate": 2.5161067940990307e-06,
      "loss": 0.0602,
      "step": 3694
    },
    {
      "epoch": 2.657795360546664,
      "grad_norm": 0.8349571543988786,
      "learning_rate": 2.515856480324718e-06,
      "loss": 0.0029,
      "step": 3695
    },
    {
      "epoch": 2.6585146556374752,
      "grad_norm": 3.352004991881754,
      "learning_rate": 2.515606114282807e-06,
      "loss": 0.0565,
      "step": 3696
    },
    {
      "epoch": 2.6592339507282863,
      "grad_norm": 3.8020763122160806,
      "learning_rate": 2.515355695986178e-06,
      "loss": 0.1456,
      "step": 3697
    },
    {
      "epoch": 2.6599532458190973,
      "grad_norm": 3.1413841286057083,
      "learning_rate": 2.515105225447716e-06,
      "loss": 0.0573,
      "step": 3698
    },
    {
      "epoch": 2.6606725409099083,
      "grad_norm": 3.3504557985257133,
      "learning_rate": 2.514854702680308e-06,
      "loss": 0.0397,
      "step": 3699
    },
    {
      "epoch": 2.6613918360007194,
      "grad_norm": 3.8842973516463855,
      "learning_rate": 2.514604127696844e-06,
      "loss": 0.0605,
      "step": 3700
    },
    {
      "epoch": 2.6621111310915304,
      "grad_norm": 3.4559127078038396,
      "learning_rate": 2.514353500510217e-06,
      "loss": 0.1296,
      "step": 3701
    },
    {
      "epoch": 2.6628304261823414,
      "grad_norm": 3.4128165104304133,
      "learning_rate": 2.5141028211333215e-06,
      "loss": 0.0677,
      "step": 3702
    },
    {
      "epoch": 2.663549721273152,
      "grad_norm": 1.4948083931353333,
      "learning_rate": 2.513852089579056e-06,
      "loss": 0.0332,
      "step": 3703
    },
    {
      "epoch": 2.6642690163639635,
      "grad_norm": 0.3681565698425834,
      "learning_rate": 2.5136013058603203e-06,
      "loss": 0.0011,
      "step": 3704
    },
    {
      "epoch": 2.664988311454774,
      "grad_norm": 3.0782569704485168,
      "learning_rate": 2.513350469990018e-06,
      "loss": 0.0464,
      "step": 3705
    },
    {
      "epoch": 2.6657076065455856,
      "grad_norm": 4.508129414508531,
      "learning_rate": 2.5130995819810554e-06,
      "loss": 0.0997,
      "step": 3706
    },
    {
      "epoch": 2.666426901636396,
      "grad_norm": 4.439008375032999,
      "learning_rate": 2.5128486418463405e-06,
      "loss": 0.1724,
      "step": 3707
    },
    {
      "epoch": 2.667146196727207,
      "grad_norm": 6.448391018258753,
      "learning_rate": 2.5125976495987855e-06,
      "loss": 0.1329,
      "step": 3708
    },
    {
      "epoch": 2.6678654918180182,
      "grad_norm": 1.5862208292542819,
      "learning_rate": 2.5123466052513037e-06,
      "loss": 0.0509,
      "step": 3709
    },
    {
      "epoch": 2.6685847869088293,
      "grad_norm": 6.006311399033885,
      "learning_rate": 2.512095508816812e-06,
      "loss": 0.2098,
      "step": 3710
    },
    {
      "epoch": 2.6693040819996403,
      "grad_norm": 2.0973880586514104,
      "learning_rate": 2.511844360308229e-06,
      "loss": 0.0429,
      "step": 3711
    },
    {
      "epoch": 2.6700233770904513,
      "grad_norm": 2.200696129453983,
      "learning_rate": 2.5115931597384786e-06,
      "loss": 0.0564,
      "step": 3712
    },
    {
      "epoch": 2.6707426721812624,
      "grad_norm": 5.1460394661064885,
      "learning_rate": 2.5113419071204837e-06,
      "loss": 0.0496,
      "step": 3713
    },
    {
      "epoch": 2.6714619672720734,
      "grad_norm": 3.0807263481660847,
      "learning_rate": 2.511090602467172e-06,
      "loss": 0.0882,
      "step": 3714
    },
    {
      "epoch": 2.6721812623628844,
      "grad_norm": 2.767989077439596,
      "learning_rate": 2.5108392457914745e-06,
      "loss": 0.035,
      "step": 3715
    },
    {
      "epoch": 2.6729005574536955,
      "grad_norm": 3.22985158347298,
      "learning_rate": 2.510587837106323e-06,
      "loss": 0.0918,
      "step": 3716
    },
    {
      "epoch": 2.6736198525445065,
      "grad_norm": 5.685990673290136,
      "learning_rate": 2.5103363764246534e-06,
      "loss": 0.02,
      "step": 3717
    },
    {
      "epoch": 2.6743391476353175,
      "grad_norm": 1.3526489239489268,
      "learning_rate": 2.5100848637594037e-06,
      "loss": 0.0276,
      "step": 3718
    },
    {
      "epoch": 2.6750584427261286,
      "grad_norm": 5.080821916705485,
      "learning_rate": 2.5098332991235146e-06,
      "loss": 0.0773,
      "step": 3719
    },
    {
      "epoch": 2.675777737816939,
      "grad_norm": 4.825549932184532,
      "learning_rate": 2.5095816825299296e-06,
      "loss": 0.0155,
      "step": 3720
    },
    {
      "epoch": 2.6764970329077507,
      "grad_norm": 3.339456065210547,
      "learning_rate": 2.509330013991594e-06,
      "loss": 0.0647,
      "step": 3721
    },
    {
      "epoch": 2.6772163279985612,
      "grad_norm": 1.9561245021690106,
      "learning_rate": 2.5090782935214582e-06,
      "loss": 0.0125,
      "step": 3722
    },
    {
      "epoch": 2.6779356230893723,
      "grad_norm": 6.473213261711899,
      "learning_rate": 2.508826521132473e-06,
      "loss": 0.0417,
      "step": 3723
    },
    {
      "epoch": 2.6786549181801833,
      "grad_norm": 2.838375328362837,
      "learning_rate": 2.5085746968375916e-06,
      "loss": 0.0686,
      "step": 3724
    },
    {
      "epoch": 2.6793742132709943,
      "grad_norm": 2.328340876556262,
      "learning_rate": 2.508322820649772e-06,
      "loss": 0.0062,
      "step": 3725
    },
    {
      "epoch": 2.6800935083618054,
      "grad_norm": 4.180222374359697,
      "learning_rate": 2.508070892581973e-06,
      "loss": 0.0846,
      "step": 3726
    },
    {
      "epoch": 2.6808128034526164,
      "grad_norm": 2.0105111378374856,
      "learning_rate": 2.5078189126471572e-06,
      "loss": 0.0299,
      "step": 3727
    },
    {
      "epoch": 2.6815320985434274,
      "grad_norm": 1.8372937207366125,
      "learning_rate": 2.5075668808582892e-06,
      "loss": 0.0405,
      "step": 3728
    },
    {
      "epoch": 2.6822513936342385,
      "grad_norm": 3.9709861439457126,
      "learning_rate": 2.507314797228336e-06,
      "loss": 0.172,
      "step": 3729
    },
    {
      "epoch": 2.6829706887250495,
      "grad_norm": 2.9393880250293907,
      "learning_rate": 2.5070626617702685e-06,
      "loss": 0.1039,
      "step": 3730
    },
    {
      "epoch": 2.6836899838158605,
      "grad_norm": 3.582140188522379,
      "learning_rate": 2.5068104744970596e-06,
      "loss": 0.0234,
      "step": 3731
    },
    {
      "epoch": 2.6844092789066716,
      "grad_norm": 4.536372856698448,
      "learning_rate": 2.5065582354216836e-06,
      "loss": 0.2384,
      "step": 3732
    },
    {
      "epoch": 2.6851285739974826,
      "grad_norm": 4.683097014370957,
      "learning_rate": 2.50630594455712e-06,
      "loss": 0.1698,
      "step": 3733
    },
    {
      "epoch": 2.6858478690882936,
      "grad_norm": 7.623351361379616,
      "learning_rate": 2.5060536019163488e-06,
      "loss": 0.0651,
      "step": 3734
    },
    {
      "epoch": 2.6865671641791042,
      "grad_norm": 3.436159723634687,
      "learning_rate": 2.505801207512354e-06,
      "loss": 0.0673,
      "step": 3735
    },
    {
      "epoch": 2.6872864592699157,
      "grad_norm": 3.3445653716355954,
      "learning_rate": 2.505548761358121e-06,
      "loss": 0.0258,
      "step": 3736
    },
    {
      "epoch": 2.6880057543607263,
      "grad_norm": 5.9275934808945,
      "learning_rate": 2.5052962634666394e-06,
      "loss": 0.1,
      "step": 3737
    },
    {
      "epoch": 2.6887250494515373,
      "grad_norm": 3.612022371542458,
      "learning_rate": 2.5050437138509e-06,
      "loss": 0.0311,
      "step": 3738
    },
    {
      "epoch": 2.6894443445423484,
      "grad_norm": 5.101380180247387,
      "learning_rate": 2.5047911125238976e-06,
      "loss": 0.1947,
      "step": 3739
    },
    {
      "epoch": 2.6901636396331594,
      "grad_norm": 3.324910546004371,
      "learning_rate": 2.5045384594986286e-06,
      "loss": 0.0614,
      "step": 3740
    },
    {
      "epoch": 2.6908829347239704,
      "grad_norm": 3.563723911119825,
      "learning_rate": 2.504285754788092e-06,
      "loss": 0.0999,
      "step": 3741
    },
    {
      "epoch": 2.6916022298147815,
      "grad_norm": 1.791112471627173,
      "learning_rate": 2.50403299840529e-06,
      "loss": 0.0234,
      "step": 3742
    },
    {
      "epoch": 2.6923215249055925,
      "grad_norm": 3.7782141473859157,
      "learning_rate": 2.503780190363229e-06,
      "loss": 0.1059,
      "step": 3743
    },
    {
      "epoch": 2.6930408199964035,
      "grad_norm": 1.8366305740591597,
      "learning_rate": 2.5035273306749142e-06,
      "loss": 0.0245,
      "step": 3744
    },
    {
      "epoch": 2.6937601150872146,
      "grad_norm": 2.5378193881548032,
      "learning_rate": 2.503274419353357e-06,
      "loss": 0.0858,
      "step": 3745
    },
    {
      "epoch": 2.6944794101780256,
      "grad_norm": 4.097122388215333,
      "learning_rate": 2.50302145641157e-06,
      "loss": 0.0849,
      "step": 3746
    },
    {
      "epoch": 2.6951987052688366,
      "grad_norm": 2.0825888205759426,
      "learning_rate": 2.5027684418625677e-06,
      "loss": 0.0236,
      "step": 3747
    },
    {
      "epoch": 2.6959180003596477,
      "grad_norm": 1.715268006172488,
      "learning_rate": 2.5025153757193687e-06,
      "loss": 0.0174,
      "step": 3748
    },
    {
      "epoch": 2.6966372954504587,
      "grad_norm": 4.008159243437323,
      "learning_rate": 2.502262257994994e-06,
      "loss": 0.1017,
      "step": 3749
    },
    {
      "epoch": 2.6973565905412693,
      "grad_norm": 3.910186110703431,
      "learning_rate": 2.5020090887024663e-06,
      "loss": 0.0258,
      "step": 3750
    },
    {
      "epoch": 2.698075885632081,
      "grad_norm": 2.6885361033008324,
      "learning_rate": 2.5017558678548125e-06,
      "loss": 0.081,
      "step": 3751
    },
    {
      "epoch": 2.6987951807228914,
      "grad_norm": 5.532676240066749,
      "learning_rate": 2.501502595465061e-06,
      "loss": 0.1759,
      "step": 3752
    },
    {
      "epoch": 2.6995144758137024,
      "grad_norm": 3.003324394368566,
      "learning_rate": 2.5012492715462422e-06,
      "loss": 0.0656,
      "step": 3753
    },
    {
      "epoch": 2.7002337709045134,
      "grad_norm": 1.3094946917644321,
      "learning_rate": 2.5009958961113914e-06,
      "loss": 0.029,
      "step": 3754
    },
    {
      "epoch": 2.7009530659953245,
      "grad_norm": 6.562976193825908,
      "learning_rate": 2.500742469173544e-06,
      "loss": 0.157,
      "step": 3755
    },
    {
      "epoch": 2.7016723610861355,
      "grad_norm": 4.3720726567237005,
      "learning_rate": 2.50048899074574e-06,
      "loss": 0.1266,
      "step": 3756
    },
    {
      "epoch": 2.7023916561769465,
      "grad_norm": 2.3029017667465963,
      "learning_rate": 2.5002354608410212e-06,
      "loss": 0.0175,
      "step": 3757
    },
    {
      "epoch": 2.7031109512677576,
      "grad_norm": 2.6149334590859934,
      "learning_rate": 2.4999818794724324e-06,
      "loss": 0.0622,
      "step": 3758
    },
    {
      "epoch": 2.7038302463585686,
      "grad_norm": 1.2967474991398897,
      "learning_rate": 2.4997282466530206e-06,
      "loss": 0.0275,
      "step": 3759
    },
    {
      "epoch": 2.7045495414493796,
      "grad_norm": 4.003735820358936,
      "learning_rate": 2.499474562395835e-06,
      "loss": 0.0986,
      "step": 3760
    },
    {
      "epoch": 2.7052688365401907,
      "grad_norm": 3.1107490529969914,
      "learning_rate": 2.499220826713928e-06,
      "loss": 0.1539,
      "step": 3761
    },
    {
      "epoch": 2.7059881316310017,
      "grad_norm": 1.5447893298975588,
      "learning_rate": 2.498967039620357e-06,
      "loss": 0.0393,
      "step": 3762
    },
    {
      "epoch": 2.7067074267218127,
      "grad_norm": 1.430421493773461,
      "learning_rate": 2.498713201128177e-06,
      "loss": 0.0312,
      "step": 3763
    },
    {
      "epoch": 2.707426721812624,
      "grad_norm": 0.7147655038678005,
      "learning_rate": 2.4984593112504496e-06,
      "loss": 0.0028,
      "step": 3764
    },
    {
      "epoch": 2.7081460169034344,
      "grad_norm": 2.6163023900954605,
      "learning_rate": 2.4982053700002387e-06,
      "loss": 0.0455,
      "step": 3765
    },
    {
      "epoch": 2.708865311994246,
      "grad_norm": 2.8367525629082797,
      "learning_rate": 2.4979513773906086e-06,
      "loss": 0.039,
      "step": 3766
    },
    {
      "epoch": 2.7095846070850564,
      "grad_norm": 4.418308938583768,
      "learning_rate": 2.4976973334346285e-06,
      "loss": 0.1885,
      "step": 3767
    },
    {
      "epoch": 2.710303902175868,
      "grad_norm": 2.2718286888193684,
      "learning_rate": 2.4974432381453694e-06,
      "loss": 0.0262,
      "step": 3768
    },
    {
      "epoch": 2.7110231972666785,
      "grad_norm": 2.1181053858894985,
      "learning_rate": 2.497189091535904e-06,
      "loss": 0.0287,
      "step": 3769
    },
    {
      "epoch": 2.7117424923574895,
      "grad_norm": 5.550152053541698,
      "learning_rate": 2.4969348936193103e-06,
      "loss": 0.0358,
      "step": 3770
    },
    {
      "epoch": 2.7124617874483006,
      "grad_norm": 7.826237432316334,
      "learning_rate": 2.4966806444086653e-06,
      "loss": 0.2648,
      "step": 3771
    },
    {
      "epoch": 2.7131810825391116,
      "grad_norm": 2.2515669585947684,
      "learning_rate": 2.496426343917052e-06,
      "loss": 0.0363,
      "step": 3772
    },
    {
      "epoch": 2.7139003776299226,
      "grad_norm": 3.033278647540971,
      "learning_rate": 2.4961719921575544e-06,
      "loss": 0.0723,
      "step": 3773
    },
    {
      "epoch": 2.7146196727207337,
      "grad_norm": 5.0513808588723474,
      "learning_rate": 2.495917589143259e-06,
      "loss": 0.1542,
      "step": 3774
    },
    {
      "epoch": 2.7153389678115447,
      "grad_norm": 1.236200234155994,
      "learning_rate": 2.4956631348872547e-06,
      "loss": 0.0011,
      "step": 3775
    },
    {
      "epoch": 2.7160582629023557,
      "grad_norm": 1.794058497036737,
      "learning_rate": 2.4954086294026347e-06,
      "loss": 0.0425,
      "step": 3776
    },
    {
      "epoch": 2.716777557993167,
      "grad_norm": 0.7885792545823976,
      "learning_rate": 2.4951540727024937e-06,
      "loss": 0.0113,
      "step": 3777
    },
    {
      "epoch": 2.717496853083978,
      "grad_norm": 3.9786612271396318,
      "learning_rate": 2.4948994647999284e-06,
      "loss": 0.1299,
      "step": 3778
    },
    {
      "epoch": 2.718216148174789,
      "grad_norm": 3.800782343998083,
      "learning_rate": 2.4946448057080387e-06,
      "loss": 0.0195,
      "step": 3779
    },
    {
      "epoch": 2.7189354432655994,
      "grad_norm": 5.517397828314818,
      "learning_rate": 2.4943900954399285e-06,
      "loss": 0.1374,
      "step": 3780
    },
    {
      "epoch": 2.719654738356411,
      "grad_norm": 4.749014827544546,
      "learning_rate": 2.494135334008702e-06,
      "loss": 0.0712,
      "step": 3781
    },
    {
      "epoch": 2.7203740334472215,
      "grad_norm": 2.079495561781943,
      "learning_rate": 2.4938805214274674e-06,
      "loss": 0.0589,
      "step": 3782
    },
    {
      "epoch": 2.721093328538033,
      "grad_norm": 3.7865581853634414,
      "learning_rate": 2.493625657709335e-06,
      "loss": 0.0141,
      "step": 3783
    },
    {
      "epoch": 2.7218126236288436,
      "grad_norm": 2.7866566240133426,
      "learning_rate": 2.4933707428674186e-06,
      "loss": 0.0629,
      "step": 3784
    },
    {
      "epoch": 2.7225319187196546,
      "grad_norm": 6.317690132660262,
      "learning_rate": 2.4931157769148337e-06,
      "loss": 0.1965,
      "step": 3785
    },
    {
      "epoch": 2.7232512138104656,
      "grad_norm": 4.884710160667021,
      "learning_rate": 2.4928607598646986e-06,
      "loss": 0.1197,
      "step": 3786
    },
    {
      "epoch": 2.7239705089012767,
      "grad_norm": 2.4681905263098405,
      "learning_rate": 2.4926056917301344e-06,
      "loss": 0.0571,
      "step": 3787
    },
    {
      "epoch": 2.7246898039920877,
      "grad_norm": 4.1517503717633,
      "learning_rate": 2.492350572524265e-06,
      "loss": 0.22,
      "step": 3788
    },
    {
      "epoch": 2.7254090990828987,
      "grad_norm": 2.5431151256523674,
      "learning_rate": 2.4920954022602164e-06,
      "loss": 0.0204,
      "step": 3789
    },
    {
      "epoch": 2.72612839417371,
      "grad_norm": 4.02233241232701,
      "learning_rate": 2.491840180951118e-06,
      "loss": 0.1906,
      "step": 3790
    },
    {
      "epoch": 2.726847689264521,
      "grad_norm": 1.8758054620885445,
      "learning_rate": 2.491584908610101e-06,
      "loss": 0.0162,
      "step": 3791
    },
    {
      "epoch": 2.727566984355332,
      "grad_norm": 4.11083243647948,
      "learning_rate": 2.4913295852503e-06,
      "loss": 0.0949,
      "step": 3792
    },
    {
      "epoch": 2.728286279446143,
      "grad_norm": 4.023825793882214,
      "learning_rate": 2.4910742108848517e-06,
      "loss": 0.0804,
      "step": 3793
    },
    {
      "epoch": 2.729005574536954,
      "grad_norm": 2.093990926581722,
      "learning_rate": 2.4908187855268954e-06,
      "loss": 0.0066,
      "step": 3794
    },
    {
      "epoch": 2.729724869627765,
      "grad_norm": 1.7491648990941016,
      "learning_rate": 2.490563309189573e-06,
      "loss": 0.0126,
      "step": 3795
    },
    {
      "epoch": 2.730444164718576,
      "grad_norm": 2.399989557005225,
      "learning_rate": 2.4903077818860298e-06,
      "loss": 0.045,
      "step": 3796
    },
    {
      "epoch": 2.7311634598093866,
      "grad_norm": 7.659149651078495,
      "learning_rate": 2.490052203629412e-06,
      "loss": 0.2433,
      "step": 3797
    },
    {
      "epoch": 2.731882754900198,
      "grad_norm": 5.279722924681732,
      "learning_rate": 2.489796574432872e-06,
      "loss": 0.0777,
      "step": 3798
    },
    {
      "epoch": 2.7326020499910086,
      "grad_norm": 2.779167424058132,
      "learning_rate": 2.489540894309559e-06,
      "loss": 0.0724,
      "step": 3799
    },
    {
      "epoch": 2.7333213450818197,
      "grad_norm": 3.9183631881981826,
      "learning_rate": 2.489285163272631e-06,
      "loss": 0.1064,
      "step": 3800
    },
    {
      "epoch": 2.7340406401726307,
      "grad_norm": 1.7996846817942445,
      "learning_rate": 2.489029381335244e-06,
      "loss": 0.0526,
      "step": 3801
    },
    {
      "epoch": 2.7347599352634417,
      "grad_norm": 2.0556973503417004,
      "learning_rate": 2.4887735485105597e-06,
      "loss": 0.0331,
      "step": 3802
    },
    {
      "epoch": 2.7354792303542528,
      "grad_norm": 4.095523898768103,
      "learning_rate": 2.48851766481174e-06,
      "loss": 0.1218,
      "step": 3803
    },
    {
      "epoch": 2.736198525445064,
      "grad_norm": 1.4382698777803322,
      "learning_rate": 2.488261730251952e-06,
      "loss": 0.037,
      "step": 3804
    },
    {
      "epoch": 2.736917820535875,
      "grad_norm": 2.913472064449454,
      "learning_rate": 2.4880057448443625e-06,
      "loss": 0.0171,
      "step": 3805
    },
    {
      "epoch": 2.737637115626686,
      "grad_norm": 5.497036022929975,
      "learning_rate": 2.487749708602143e-06,
      "loss": 0.2356,
      "step": 3806
    },
    {
      "epoch": 2.738356410717497,
      "grad_norm": 2.5011258367401026,
      "learning_rate": 2.4874936215384673e-06,
      "loss": 0.0705,
      "step": 3807
    },
    {
      "epoch": 2.739075705808308,
      "grad_norm": 4.92438581028327,
      "learning_rate": 2.4872374836665114e-06,
      "loss": 0.0954,
      "step": 3808
    },
    {
      "epoch": 2.739795000899119,
      "grad_norm": 2.786830921258804,
      "learning_rate": 2.4869812949994533e-06,
      "loss": 0.0749,
      "step": 3809
    },
    {
      "epoch": 2.74051429598993,
      "grad_norm": 1.8534674508542008,
      "learning_rate": 2.486725055550476e-06,
      "loss": 0.0069,
      "step": 3810
    },
    {
      "epoch": 2.741233591080741,
      "grad_norm": 6.822405150654127,
      "learning_rate": 2.4864687653327615e-06,
      "loss": 0.2362,
      "step": 3811
    },
    {
      "epoch": 2.7419528861715516,
      "grad_norm": 3.241730363571939,
      "learning_rate": 2.486212424359498e-06,
      "loss": 0.1017,
      "step": 3812
    },
    {
      "epoch": 2.742672181262363,
      "grad_norm": 0.9117142688657816,
      "learning_rate": 2.4859560326438735e-06,
      "loss": 0.0023,
      "step": 3813
    },
    {
      "epoch": 2.7433914763531737,
      "grad_norm": 2.1969584255683046,
      "learning_rate": 2.48569959019908e-06,
      "loss": 0.0518,
      "step": 3814
    },
    {
      "epoch": 2.7441107714439847,
      "grad_norm": 7.07143618197524,
      "learning_rate": 2.4854430970383133e-06,
      "loss": 0.1294,
      "step": 3815
    },
    {
      "epoch": 2.7448300665347958,
      "grad_norm": 4.424225818464861,
      "learning_rate": 2.485186553174769e-06,
      "loss": 0.0107,
      "step": 3816
    },
    {
      "epoch": 2.745549361625607,
      "grad_norm": 3.60659689273812,
      "learning_rate": 2.484929958621647e-06,
      "loss": 0.0483,
      "step": 3817
    },
    {
      "epoch": 2.746268656716418,
      "grad_norm": 5.526072670512086,
      "learning_rate": 2.4846733133921493e-06,
      "loss": 0.0575,
      "step": 3818
    },
    {
      "epoch": 2.746987951807229,
      "grad_norm": 2.5986371436460836,
      "learning_rate": 2.4844166174994813e-06,
      "loss": 0.0567,
      "step": 3819
    },
    {
      "epoch": 2.74770724689804,
      "grad_norm": 6.72525545727892,
      "learning_rate": 2.4841598709568505e-06,
      "loss": 0.2309,
      "step": 3820
    },
    {
      "epoch": 2.748426541988851,
      "grad_norm": 5.790728760969494,
      "learning_rate": 2.4839030737774662e-06,
      "loss": 0.1239,
      "step": 3821
    },
    {
      "epoch": 2.749145837079662,
      "grad_norm": 2.932769247696104,
      "learning_rate": 2.483646225974542e-06,
      "loss": 0.0964,
      "step": 3822
    },
    {
      "epoch": 2.749865132170473,
      "grad_norm": 2.5459656140521436,
      "learning_rate": 2.483389327561293e-06,
      "loss": 0.0117,
      "step": 3823
    },
    {
      "epoch": 2.750584427261284,
      "grad_norm": 2.0235417859285096,
      "learning_rate": 2.483132378550936e-06,
      "loss": 0.052,
      "step": 3824
    },
    {
      "epoch": 2.751303722352095,
      "grad_norm": 4.49608334341545,
      "learning_rate": 2.482875378956693e-06,
      "loss": 0.1671,
      "step": 3825
    },
    {
      "epoch": 2.752023017442906,
      "grad_norm": 5.001709918313664,
      "learning_rate": 2.4826183287917862e-06,
      "loss": 0.1057,
      "step": 3826
    },
    {
      "epoch": 2.7527423125337167,
      "grad_norm": 0.625589154422185,
      "learning_rate": 2.482361228069441e-06,
      "loss": 0.0084,
      "step": 3827
    },
    {
      "epoch": 2.753461607624528,
      "grad_norm": 1.6003918171345752,
      "learning_rate": 2.4821040768028867e-06,
      "loss": 0.0061,
      "step": 3828
    },
    {
      "epoch": 2.7541809027153388,
      "grad_norm": 5.221914451576344,
      "learning_rate": 2.4818468750053532e-06,
      "loss": 0.2122,
      "step": 3829
    },
    {
      "epoch": 2.75490019780615,
      "grad_norm": 3.8509367562719,
      "learning_rate": 2.4815896226900747e-06,
      "loss": 0.0365,
      "step": 3830
    },
    {
      "epoch": 2.755619492896961,
      "grad_norm": 1.6401478360540251,
      "learning_rate": 2.481332319870287e-06,
      "loss": 0.0175,
      "step": 3831
    },
    {
      "epoch": 2.756338787987772,
      "grad_norm": 2.3623839972385743,
      "learning_rate": 2.481074966559229e-06,
      "loss": 0.0492,
      "step": 3832
    },
    {
      "epoch": 2.757058083078583,
      "grad_norm": 3.2357350982296254,
      "learning_rate": 2.480817562770141e-06,
      "loss": 0.1159,
      "step": 3833
    },
    {
      "epoch": 2.757777378169394,
      "grad_norm": 2.8424872505661654,
      "learning_rate": 2.480560108516268e-06,
      "loss": 0.1208,
      "step": 3834
    },
    {
      "epoch": 2.758496673260205,
      "grad_norm": 3.342064015514164,
      "learning_rate": 2.480302603810856e-06,
      "loss": 0.1189,
      "step": 3835
    },
    {
      "epoch": 2.759215968351016,
      "grad_norm": 5.102169524804131,
      "learning_rate": 2.480045048667155e-06,
      "loss": 0.128,
      "step": 3836
    },
    {
      "epoch": 2.759935263441827,
      "grad_norm": 3.4758142104540886,
      "learning_rate": 2.4797874430984145e-06,
      "loss": 0.0851,
      "step": 3837
    },
    {
      "epoch": 2.760654558532638,
      "grad_norm": 3.803944360324067,
      "learning_rate": 2.479529787117891e-06,
      "loss": 0.145,
      "step": 3838
    },
    {
      "epoch": 2.761373853623449,
      "grad_norm": 0.053240153791519355,
      "learning_rate": 2.47927208073884e-06,
      "loss": 0.0002,
      "step": 3839
    },
    {
      "epoch": 2.76209314871426,
      "grad_norm": 4.682782318988433,
      "learning_rate": 2.479014323974522e-06,
      "loss": 0.1588,
      "step": 3840
    },
    {
      "epoch": 2.762812443805071,
      "grad_norm": 5.049956238321539,
      "learning_rate": 2.4787565168381977e-06,
      "loss": 0.1748,
      "step": 3841
    },
    {
      "epoch": 2.7635317388958818,
      "grad_norm": 1.8721860693308074,
      "learning_rate": 2.4784986593431326e-06,
      "loss": 0.0042,
      "step": 3842
    },
    {
      "epoch": 2.7642510339866933,
      "grad_norm": 7.490756552189824,
      "learning_rate": 2.478240751502594e-06,
      "loss": 0.1896,
      "step": 3843
    },
    {
      "epoch": 2.764970329077504,
      "grad_norm": 3.874699148544077,
      "learning_rate": 2.4779827933298515e-06,
      "loss": 0.0935,
      "step": 3844
    },
    {
      "epoch": 2.7656896241683153,
      "grad_norm": 3.364981346670917,
      "learning_rate": 2.477724784838177e-06,
      "loss": 0.0555,
      "step": 3845
    },
    {
      "epoch": 2.766408919259126,
      "grad_norm": 2.460734601641832,
      "learning_rate": 2.4774667260408467e-06,
      "loss": 0.0404,
      "step": 3846
    },
    {
      "epoch": 2.767128214349937,
      "grad_norm": 3.726257485263743,
      "learning_rate": 2.4772086169511368e-06,
      "loss": 0.0478,
      "step": 3847
    },
    {
      "epoch": 2.767847509440748,
      "grad_norm": 4.982709621343737,
      "learning_rate": 2.4769504575823287e-06,
      "loss": 0.1154,
      "step": 3848
    },
    {
      "epoch": 2.768566804531559,
      "grad_norm": 4.482995244067132,
      "learning_rate": 2.4766922479477044e-06,
      "loss": 0.1304,
      "step": 3849
    },
    {
      "epoch": 2.76928609962237,
      "grad_norm": 2.468381460186908,
      "learning_rate": 2.476433988060549e-06,
      "loss": 0.0441,
      "step": 3850
    },
    {
      "epoch": 2.770005394713181,
      "grad_norm": 2.6766598459936906,
      "learning_rate": 2.4761756779341514e-06,
      "loss": 0.0742,
      "step": 3851
    },
    {
      "epoch": 2.770724689803992,
      "grad_norm": 7.454960120124403,
      "learning_rate": 2.475917317581801e-06,
      "loss": 0.1368,
      "step": 3852
    },
    {
      "epoch": 2.771443984894803,
      "grad_norm": 5.028421935334111,
      "learning_rate": 2.4756589070167924e-06,
      "loss": 0.0116,
      "step": 3853
    },
    {
      "epoch": 2.772163279985614,
      "grad_norm": 1.81474267889335,
      "learning_rate": 2.4754004462524194e-06,
      "loss": 0.0026,
      "step": 3854
    },
    {
      "epoch": 2.772882575076425,
      "grad_norm": 3.4418421952476472,
      "learning_rate": 2.4751419353019816e-06,
      "loss": 0.0774,
      "step": 3855
    },
    {
      "epoch": 2.7736018701672362,
      "grad_norm": 3.722658637863967,
      "learning_rate": 2.474883374178779e-06,
      "loss": 0.1103,
      "step": 3856
    },
    {
      "epoch": 2.774321165258047,
      "grad_norm": 2.5819807755254662,
      "learning_rate": 2.4746247628961158e-06,
      "loss": 0.092,
      "step": 3857
    },
    {
      "epoch": 2.7750404603488583,
      "grad_norm": 1.5906114731577325,
      "learning_rate": 2.4743661014672976e-06,
      "loss": 0.014,
      "step": 3858
    },
    {
      "epoch": 2.775759755439669,
      "grad_norm": 1.4831716114139566,
      "learning_rate": 2.474107389905633e-06,
      "loss": 0.0396,
      "step": 3859
    },
    {
      "epoch": 2.7764790505304804,
      "grad_norm": 3.783161035595401,
      "learning_rate": 2.4738486282244334e-06,
      "loss": 0.162,
      "step": 3860
    },
    {
      "epoch": 2.777198345621291,
      "grad_norm": 4.850337177229191,
      "learning_rate": 2.473589816437012e-06,
      "loss": 0.2067,
      "step": 3861
    },
    {
      "epoch": 2.777917640712102,
      "grad_norm": 3.2986567958396296,
      "learning_rate": 2.4733309545566857e-06,
      "loss": 0.0812,
      "step": 3862
    },
    {
      "epoch": 2.778636935802913,
      "grad_norm": 7.190863343943745,
      "learning_rate": 2.473072042596773e-06,
      "loss": 0.221,
      "step": 3863
    },
    {
      "epoch": 2.779356230893724,
      "grad_norm": 4.647389894934905,
      "learning_rate": 2.4728130805705956e-06,
      "loss": 0.1893,
      "step": 3864
    },
    {
      "epoch": 2.780075525984535,
      "grad_norm": 0.6471652289611521,
      "learning_rate": 2.4725540684914774e-06,
      "loss": 0.0062,
      "step": 3865
    },
    {
      "epoch": 2.780794821075346,
      "grad_norm": 2.2653525905085967,
      "learning_rate": 2.4722950063727452e-06,
      "loss": 0.0132,
      "step": 3866
    },
    {
      "epoch": 2.781514116166157,
      "grad_norm": 3.94115588378899,
      "learning_rate": 2.4720358942277284e-06,
      "loss": 0.1455,
      "step": 3867
    },
    {
      "epoch": 2.782233411256968,
      "grad_norm": 1.4484118116893585,
      "learning_rate": 2.471776732069758e-06,
      "loss": 0.0312,
      "step": 3868
    },
    {
      "epoch": 2.7829527063477792,
      "grad_norm": 5.164363175950282,
      "learning_rate": 2.471517519912169e-06,
      "loss": 0.1202,
      "step": 3869
    },
    {
      "epoch": 2.7836720014385903,
      "grad_norm": 4.085297370762143,
      "learning_rate": 2.4712582577682983e-06,
      "loss": 0.1215,
      "step": 3870
    },
    {
      "epoch": 2.7843912965294013,
      "grad_norm": 2.1976435355200423,
      "learning_rate": 2.4709989456514852e-06,
      "loss": 0.023,
      "step": 3871
    },
    {
      "epoch": 2.7851105916202124,
      "grad_norm": 3.795978867425413,
      "learning_rate": 2.4707395835750715e-06,
      "loss": 0.0536,
      "step": 3872
    },
    {
      "epoch": 2.7858298867110234,
      "grad_norm": 0.49831318191236784,
      "learning_rate": 2.4704801715524023e-06,
      "loss": 0.0016,
      "step": 3873
    },
    {
      "epoch": 2.786549181801834,
      "grad_norm": 2.735670696013772,
      "learning_rate": 2.4702207095968246e-06,
      "loss": 0.0972,
      "step": 3874
    },
    {
      "epoch": 2.7872684768926455,
      "grad_norm": 1.5306207695904406,
      "learning_rate": 2.469961197721689e-06,
      "loss": 0.0094,
      "step": 3875
    },
    {
      "epoch": 2.787987771983456,
      "grad_norm": 7.485317038192657,
      "learning_rate": 2.469701635940346e-06,
      "loss": 0.0977,
      "step": 3876
    },
    {
      "epoch": 2.788707067074267,
      "grad_norm": 0.8017605506729603,
      "learning_rate": 2.469442024266152e-06,
      "loss": 0.0075,
      "step": 3877
    },
    {
      "epoch": 2.789426362165078,
      "grad_norm": 1.5315999796239386,
      "learning_rate": 2.469182362712464e-06,
      "loss": 0.0486,
      "step": 3878
    },
    {
      "epoch": 2.790145657255889,
      "grad_norm": 3.0094171232032885,
      "learning_rate": 2.468922651292642e-06,
      "loss": 0.0973,
      "step": 3879
    },
    {
      "epoch": 2.7908649523467,
      "grad_norm": 5.495940362829175,
      "learning_rate": 2.468662890020049e-06,
      "loss": 0.2341,
      "step": 3880
    },
    {
      "epoch": 2.791584247437511,
      "grad_norm": 0.03222792807784245,
      "learning_rate": 2.4684030789080497e-06,
      "loss": 0.0002,
      "step": 3881
    },
    {
      "epoch": 2.7923035425283222,
      "grad_norm": 1.6522192177849573,
      "learning_rate": 2.468143217970012e-06,
      "loss": 0.0216,
      "step": 3882
    },
    {
      "epoch": 2.7930228376191333,
      "grad_norm": 4.278584963512131,
      "learning_rate": 2.4678833072193064e-06,
      "loss": 0.0904,
      "step": 3883
    },
    {
      "epoch": 2.7937421327099443,
      "grad_norm": 1.8658198901715342,
      "learning_rate": 2.467623346669305e-06,
      "loss": 0.0646,
      "step": 3884
    },
    {
      "epoch": 2.7944614278007553,
      "grad_norm": 4.165368900570307,
      "learning_rate": 2.467363336333385e-06,
      "loss": 0.0574,
      "step": 3885
    },
    {
      "epoch": 2.7951807228915664,
      "grad_norm": 6.042793593636978,
      "learning_rate": 2.467103276224922e-06,
      "loss": 0.1367,
      "step": 3886
    },
    {
      "epoch": 2.7959000179823774,
      "grad_norm": 2.5061249024464836,
      "learning_rate": 2.4668431663572983e-06,
      "loss": 0.0082,
      "step": 3887
    },
    {
      "epoch": 2.7966193130731885,
      "grad_norm": 3.5663650094448576,
      "learning_rate": 2.4665830067438963e-06,
      "loss": 0.1184,
      "step": 3888
    },
    {
      "epoch": 2.797338608163999,
      "grad_norm": 5.498811650309177,
      "learning_rate": 2.4663227973981017e-06,
      "loss": 0.0741,
      "step": 3889
    },
    {
      "epoch": 2.7980579032548105,
      "grad_norm": 3.1247564428907095,
      "learning_rate": 2.466062538333303e-06,
      "loss": 0.1138,
      "step": 3890
    },
    {
      "epoch": 2.798777198345621,
      "grad_norm": 1.8035930668096747,
      "learning_rate": 2.4658022295628908e-06,
      "loss": 0.0081,
      "step": 3891
    },
    {
      "epoch": 2.799496493436432,
      "grad_norm": 2.3399103368230056,
      "learning_rate": 2.4655418711002585e-06,
      "loss": 0.0818,
      "step": 3892
    },
    {
      "epoch": 2.800215788527243,
      "grad_norm": 3.7871763473950506,
      "learning_rate": 2.465281462958802e-06,
      "loss": 0.046,
      "step": 3893
    },
    {
      "epoch": 2.800935083618054,
      "grad_norm": 2.826320122319046,
      "learning_rate": 2.4650210051519196e-06,
      "loss": 0.0279,
      "step": 3894
    },
    {
      "epoch": 2.8016543787088652,
      "grad_norm": 2.7598554463993406,
      "learning_rate": 2.4647604976930126e-06,
      "loss": 0.1152,
      "step": 3895
    },
    {
      "epoch": 2.8023736737996763,
      "grad_norm": 4.017108707566826,
      "learning_rate": 2.4644999405954846e-06,
      "loss": 0.0171,
      "step": 3896
    },
    {
      "epoch": 2.8030929688904873,
      "grad_norm": 2.5880644865828164,
      "learning_rate": 2.464239333872741e-06,
      "loss": 0.0483,
      "step": 3897
    },
    {
      "epoch": 2.8038122639812983,
      "grad_norm": 0.020342216381505664,
      "learning_rate": 2.4639786775381917e-06,
      "loss": 0.0001,
      "step": 3898
    },
    {
      "epoch": 2.8045315590721094,
      "grad_norm": 3.279351052693282,
      "learning_rate": 2.4637179716052464e-06,
      "loss": 0.0659,
      "step": 3899
    },
    {
      "epoch": 2.8052508541629204,
      "grad_norm": 3.747689614132019,
      "learning_rate": 2.4634572160873202e-06,
      "loss": 0.1245,
      "step": 3900
    },
    {
      "epoch": 2.8059701492537314,
      "grad_norm": 2.556847198583854,
      "learning_rate": 2.4631964109978293e-06,
      "loss": 0.0805,
      "step": 3901
    },
    {
      "epoch": 2.8066894443445425,
      "grad_norm": 1.487752215800543,
      "learning_rate": 2.4629355563501913e-06,
      "loss": 0.0379,
      "step": 3902
    },
    {
      "epoch": 2.8074087394353535,
      "grad_norm": 3.5541552559427996,
      "learning_rate": 2.462674652157829e-06,
      "loss": 0.1612,
      "step": 3903
    },
    {
      "epoch": 2.808128034526164,
      "grad_norm": 4.497174420593064,
      "learning_rate": 2.4624136984341665e-06,
      "loss": 0.0715,
      "step": 3904
    },
    {
      "epoch": 2.8088473296169756,
      "grad_norm": 5.354153560445609,
      "learning_rate": 2.4621526951926288e-06,
      "loss": 0.2761,
      "step": 3905
    },
    {
      "epoch": 2.809566624707786,
      "grad_norm": 0.16774901689798485,
      "learning_rate": 2.4618916424466465e-06,
      "loss": 0.0007,
      "step": 3906
    },
    {
      "epoch": 2.810285919798597,
      "grad_norm": 1.3001457227868953,
      "learning_rate": 2.46163054020965e-06,
      "loss": 0.0353,
      "step": 3907
    },
    {
      "epoch": 2.8110052148894082,
      "grad_norm": 3.36958623910022,
      "learning_rate": 2.461369388495075e-06,
      "loss": 0.1269,
      "step": 3908
    },
    {
      "epoch": 2.8117245099802193,
      "grad_norm": 3.7189422319004803,
      "learning_rate": 2.461108187316357e-06,
      "loss": 0.0856,
      "step": 3909
    },
    {
      "epoch": 2.8124438050710303,
      "grad_norm": 2.1581560593632,
      "learning_rate": 2.4608469366869355e-06,
      "loss": 0.0534,
      "step": 3910
    },
    {
      "epoch": 2.8131631001618413,
      "grad_norm": 4.074745164373334,
      "learning_rate": 2.4605856366202524e-06,
      "loss": 0.0265,
      "step": 3911
    },
    {
      "epoch": 2.8138823952526524,
      "grad_norm": 4.702900835399429,
      "learning_rate": 2.4603242871297524e-06,
      "loss": 0.1406,
      "step": 3912
    },
    {
      "epoch": 2.8146016903434634,
      "grad_norm": 3.7609312800917847,
      "learning_rate": 2.460062888228881e-06,
      "loss": 0.1355,
      "step": 3913
    },
    {
      "epoch": 2.8153209854342744,
      "grad_norm": 1.9643504726316963,
      "learning_rate": 2.4598014399310895e-06,
      "loss": 0.029,
      "step": 3914
    },
    {
      "epoch": 2.8160402805250855,
      "grad_norm": 3.899946623151346,
      "learning_rate": 2.4595399422498287e-06,
      "loss": 0.2234,
      "step": 3915
    },
    {
      "epoch": 2.8167595756158965,
      "grad_norm": 4.158502327403848,
      "learning_rate": 2.459278395198554e-06,
      "loss": 0.1442,
      "step": 3916
    },
    {
      "epoch": 2.8174788707067075,
      "grad_norm": 0.3814063460861912,
      "learning_rate": 2.4590167987907216e-06,
      "loss": 0.0008,
      "step": 3917
    },
    {
      "epoch": 2.8181981657975186,
      "grad_norm": 3.792684106102148,
      "learning_rate": 2.458755153039791e-06,
      "loss": 0.105,
      "step": 3918
    },
    {
      "epoch": 2.818917460888329,
      "grad_norm": 3.037930935556303,
      "learning_rate": 2.458493457959225e-06,
      "loss": 0.0505,
      "step": 3919
    },
    {
      "epoch": 2.8196367559791407,
      "grad_norm": 4.051596450862193,
      "learning_rate": 2.4582317135624885e-06,
      "loss": 0.0599,
      "step": 3920
    },
    {
      "epoch": 2.8203560510699512,
      "grad_norm": 0.6171836207977903,
      "learning_rate": 2.4579699198630472e-06,
      "loss": 0.0019,
      "step": 3921
    },
    {
      "epoch": 2.8210753461607627,
      "grad_norm": 0.13068336295729466,
      "learning_rate": 2.457708076874373e-06,
      "loss": 0.0006,
      "step": 3922
    },
    {
      "epoch": 2.8217946412515733,
      "grad_norm": 4.043825817052798,
      "learning_rate": 2.4574461846099364e-06,
      "loss": 0.1228,
      "step": 3923
    },
    {
      "epoch": 2.8225139363423843,
      "grad_norm": 2.822966298703546,
      "learning_rate": 2.457184243083213e-06,
      "loss": 0.0971,
      "step": 3924
    },
    {
      "epoch": 2.8232332314331954,
      "grad_norm": 3.629541350056613,
      "learning_rate": 2.4569222523076796e-06,
      "loss": 0.1121,
      "step": 3925
    },
    {
      "epoch": 2.8239525265240064,
      "grad_norm": 3.074149761417383,
      "learning_rate": 2.456660212296817e-06,
      "loss": 0.0718,
      "step": 3926
    },
    {
      "epoch": 2.8246718216148174,
      "grad_norm": 0.5343892705709685,
      "learning_rate": 2.4563981230641072e-06,
      "loss": 0.0031,
      "step": 3927
    },
    {
      "epoch": 2.8253911167056285,
      "grad_norm": 3.4434637464799733,
      "learning_rate": 2.456135984623035e-06,
      "loss": 0.0941,
      "step": 3928
    },
    {
      "epoch": 2.8261104117964395,
      "grad_norm": 0.07857673604378634,
      "learning_rate": 2.455873796987087e-06,
      "loss": 0.0003,
      "step": 3929
    },
    {
      "epoch": 2.8268297068872505,
      "grad_norm": 3.7174050141573796,
      "learning_rate": 2.4556115601697554e-06,
      "loss": 0.11,
      "step": 3930
    },
    {
      "epoch": 2.8275490019780616,
      "grad_norm": 1.2891632747551618,
      "learning_rate": 2.455349274184531e-06,
      "loss": 0.0018,
      "step": 3931
    },
    {
      "epoch": 2.8282682970688726,
      "grad_norm": 5.876042409452094,
      "learning_rate": 2.4550869390449094e-06,
      "loss": 0.0384,
      "step": 3932
    },
    {
      "epoch": 2.8289875921596837,
      "grad_norm": 4.036109615461958,
      "learning_rate": 2.4548245547643885e-06,
      "loss": 0.1664,
      "step": 3933
    },
    {
      "epoch": 2.8297068872504942,
      "grad_norm": 4.02423685266137,
      "learning_rate": 2.4545621213564678e-06,
      "loss": 0.1349,
      "step": 3934
    },
    {
      "epoch": 2.8304261823413057,
      "grad_norm": 7.121058934795864,
      "learning_rate": 2.4542996388346503e-06,
      "loss": 0.0881,
      "step": 3935
    },
    {
      "epoch": 2.8311454774321163,
      "grad_norm": 3.4641058028701472,
      "learning_rate": 2.454037107212441e-06,
      "loss": 0.0375,
      "step": 3936
    },
    {
      "epoch": 2.831864772522928,
      "grad_norm": 2.165451777711609,
      "learning_rate": 2.4537745265033483e-06,
      "loss": 0.0453,
      "step": 3937
    },
    {
      "epoch": 2.8325840676137384,
      "grad_norm": 2.5579313152664014,
      "learning_rate": 2.453511896720881e-06,
      "loss": 0.0986,
      "step": 3938
    },
    {
      "epoch": 2.8333033627045494,
      "grad_norm": 4.266215110747072,
      "learning_rate": 2.4532492178785534e-06,
      "loss": 0.1384,
      "step": 3939
    },
    {
      "epoch": 2.8340226577953604,
      "grad_norm": 5.1746847339273625,
      "learning_rate": 2.4529864899898805e-06,
      "loss": 0.171,
      "step": 3940
    },
    {
      "epoch": 2.8347419528861715,
      "grad_norm": 5.762053673650539,
      "learning_rate": 2.4527237130683795e-06,
      "loss": 0.1993,
      "step": 3941
    },
    {
      "epoch": 2.8354612479769825,
      "grad_norm": 4.186070567031017,
      "learning_rate": 2.4524608871275703e-06,
      "loss": 0.1241,
      "step": 3942
    },
    {
      "epoch": 2.8361805430677935,
      "grad_norm": 3.1773694056042703,
      "learning_rate": 2.452198012180977e-06,
      "loss": 0.0856,
      "step": 3943
    },
    {
      "epoch": 2.8368998381586046,
      "grad_norm": 1.5064016810534515,
      "learning_rate": 2.4519350882421248e-06,
      "loss": 0.0609,
      "step": 3944
    },
    {
      "epoch": 2.8376191332494156,
      "grad_norm": 1.750157835421884,
      "learning_rate": 2.451672115324541e-06,
      "loss": 0.0434,
      "step": 3945
    },
    {
      "epoch": 2.8383384283402266,
      "grad_norm": 3.1831157950776383,
      "learning_rate": 2.4514090934417563e-06,
      "loss": 0.0419,
      "step": 3946
    },
    {
      "epoch": 2.8390577234310377,
      "grad_norm": 1.0870018209080667,
      "learning_rate": 2.451146022607303e-06,
      "loss": 0.01,
      "step": 3947
    },
    {
      "epoch": 2.8397770185218487,
      "grad_norm": 7.362506528824175,
      "learning_rate": 2.4508829028347176e-06,
      "loss": 0.0892,
      "step": 3948
    },
    {
      "epoch": 2.8404963136126598,
      "grad_norm": 2.6623095274022117,
      "learning_rate": 2.450619734137538e-06,
      "loss": 0.0925,
      "step": 3949
    },
    {
      "epoch": 2.841215608703471,
      "grad_norm": 5.4033203665308545,
      "learning_rate": 2.450356516529304e-06,
      "loss": 0.0848,
      "step": 3950
    },
    {
      "epoch": 2.8419349037942814,
      "grad_norm": 4.701511893088685,
      "learning_rate": 2.450093250023559e-06,
      "loss": 0.1343,
      "step": 3951
    },
    {
      "epoch": 2.842654198885093,
      "grad_norm": 6.710058632805847,
      "learning_rate": 2.449829934633848e-06,
      "loss": 0.214,
      "step": 3952
    },
    {
      "epoch": 2.8433734939759034,
      "grad_norm": 0.5164955730981341,
      "learning_rate": 2.44956657037372e-06,
      "loss": 0.0013,
      "step": 3953
    },
    {
      "epoch": 2.8440927890667145,
      "grad_norm": 2.2719414017304467,
      "learning_rate": 2.449303157256725e-06,
      "loss": 0.0545,
      "step": 3954
    },
    {
      "epoch": 2.8448120841575255,
      "grad_norm": 3.3097190162063135,
      "learning_rate": 2.449039695296416e-06,
      "loss": 0.0617,
      "step": 3955
    },
    {
      "epoch": 2.8455313792483365,
      "grad_norm": 3.661703921111953,
      "learning_rate": 2.448776184506349e-06,
      "loss": 0.1348,
      "step": 3956
    },
    {
      "epoch": 2.8462506743391476,
      "grad_norm": 3.9434513878931234,
      "learning_rate": 2.448512624900081e-06,
      "loss": 0.0482,
      "step": 3957
    },
    {
      "epoch": 2.8469699694299586,
      "grad_norm": 2.061810728510171,
      "learning_rate": 2.448249016491174e-06,
      "loss": 0.0449,
      "step": 3958
    },
    {
      "epoch": 2.8476892645207696,
      "grad_norm": 2.9805563445526033,
      "learning_rate": 2.4479853592931908e-06,
      "loss": 0.0717,
      "step": 3959
    },
    {
      "epoch": 2.8484085596115807,
      "grad_norm": 3.79019769136605,
      "learning_rate": 2.4477216533196956e-06,
      "loss": 0.102,
      "step": 3960
    },
    {
      "epoch": 2.8491278547023917,
      "grad_norm": 2.931977244647889,
      "learning_rate": 2.4474578985842586e-06,
      "loss": 0.0402,
      "step": 3961
    },
    {
      "epoch": 2.8498471497932027,
      "grad_norm": 3.6379791922182214,
      "learning_rate": 2.447194095100449e-06,
      "loss": 0.1069,
      "step": 3962
    },
    {
      "epoch": 2.850566444884014,
      "grad_norm": 4.267252625106105,
      "learning_rate": 2.4469302428818414e-06,
      "loss": 0.1547,
      "step": 3963
    },
    {
      "epoch": 2.851285739974825,
      "grad_norm": 0.05056332357411624,
      "learning_rate": 2.4466663419420095e-06,
      "loss": 0.0002,
      "step": 3964
    },
    {
      "epoch": 2.852005035065636,
      "grad_norm": 3.288978529667222,
      "learning_rate": 2.446402392294533e-06,
      "loss": 0.1552,
      "step": 3965
    },
    {
      "epoch": 2.8527243301564464,
      "grad_norm": 1.8203500233120882,
      "learning_rate": 2.4461383939529917e-06,
      "loss": 0.0107,
      "step": 3966
    },
    {
      "epoch": 2.853443625247258,
      "grad_norm": 3.7618450611360825,
      "learning_rate": 2.4458743469309693e-06,
      "loss": 0.0988,
      "step": 3967
    },
    {
      "epoch": 2.8541629203380685,
      "grad_norm": 4.304674200966189,
      "learning_rate": 2.445610251242051e-06,
      "loss": 0.1306,
      "step": 3968
    },
    {
      "epoch": 2.8548822154288795,
      "grad_norm": 4.316998971394456,
      "learning_rate": 2.445346106899826e-06,
      "loss": 0.1338,
      "step": 3969
    },
    {
      "epoch": 2.8556015105196906,
      "grad_norm": 6.114940909869115,
      "learning_rate": 2.445081913917884e-06,
      "loss": 0.1813,
      "step": 3970
    },
    {
      "epoch": 2.8563208056105016,
      "grad_norm": 5.57808243356584,
      "learning_rate": 2.444817672309818e-06,
      "loss": 0.1616,
      "step": 3971
    },
    {
      "epoch": 2.8570401007013126,
      "grad_norm": 1.1535881143341444,
      "learning_rate": 2.444553382089225e-06,
      "loss": 0.0237,
      "step": 3972
    },
    {
      "epoch": 2.8577593957921237,
      "grad_norm": 7.024083904122868,
      "learning_rate": 2.444289043269702e-06,
      "loss": 0.2559,
      "step": 3973
    },
    {
      "epoch": 2.8584786908829347,
      "grad_norm": 3.469631292689222,
      "learning_rate": 2.4440246558648503e-06,
      "loss": 0.0238,
      "step": 3974
    },
    {
      "epoch": 2.8591979859737457,
      "grad_norm": 2.472121495050298,
      "learning_rate": 2.443760219888273e-06,
      "loss": 0.0513,
      "step": 3975
    },
    {
      "epoch": 2.859917281064557,
      "grad_norm": 2.5089600059955406,
      "learning_rate": 2.4434957353535747e-06,
      "loss": 0.0438,
      "step": 3976
    },
    {
      "epoch": 2.860636576155368,
      "grad_norm": 4.500106520565204,
      "learning_rate": 2.4432312022743655e-06,
      "loss": 0.0984,
      "step": 3977
    },
    {
      "epoch": 2.861355871246179,
      "grad_norm": 4.2500380728354346,
      "learning_rate": 2.442966620664255e-06,
      "loss": 0.0946,
      "step": 3978
    },
    {
      "epoch": 2.86207516633699,
      "grad_norm": 2.3293505078068515,
      "learning_rate": 2.442701990536856e-06,
      "loss": 0.0509,
      "step": 3979
    },
    {
      "epoch": 2.862794461427801,
      "grad_norm": 4.326611013798776,
      "learning_rate": 2.4424373119057853e-06,
      "loss": 0.0682,
      "step": 3980
    },
    {
      "epoch": 2.8635137565186115,
      "grad_norm": 3.086132242994011,
      "learning_rate": 2.44217258478466e-06,
      "loss": 0.1492,
      "step": 3981
    },
    {
      "epoch": 2.864233051609423,
      "grad_norm": 3.410951034652264,
      "learning_rate": 2.441907809187102e-06,
      "loss": 0.0952,
      "step": 3982
    },
    {
      "epoch": 2.8649523467002336,
      "grad_norm": 7.174790656874843,
      "learning_rate": 2.441642985126733e-06,
      "loss": 0.1243,
      "step": 3983
    },
    {
      "epoch": 2.8656716417910446,
      "grad_norm": 2.978051351629623,
      "learning_rate": 2.44137811261718e-06,
      "loss": 0.0929,
      "step": 3984
    },
    {
      "epoch": 2.8663909368818556,
      "grad_norm": 4.6915648873351,
      "learning_rate": 2.4411131916720706e-06,
      "loss": 0.1022,
      "step": 3985
    },
    {
      "epoch": 2.8671102319726667,
      "grad_norm": 3.4057457818581196,
      "learning_rate": 2.440848222305035e-06,
      "loss": 0.0313,
      "step": 3986
    },
    {
      "epoch": 2.8678295270634777,
      "grad_norm": 2.6204260333164884,
      "learning_rate": 2.4405832045297065e-06,
      "loss": 0.0688,
      "step": 3987
    },
    {
      "epoch": 2.8685488221542887,
      "grad_norm": 3.997649918011928,
      "learning_rate": 2.4403181383597214e-06,
      "loss": 0.1027,
      "step": 3988
    },
    {
      "epoch": 2.8692681172451,
      "grad_norm": 0.40027512507999896,
      "learning_rate": 2.4400530238087175e-06,
      "loss": 0.0012,
      "step": 3989
    },
    {
      "epoch": 2.869987412335911,
      "grad_norm": 2.927866164123594,
      "learning_rate": 2.439787860890335e-06,
      "loss": 0.0696,
      "step": 3990
    },
    {
      "epoch": 2.870706707426722,
      "grad_norm": 7.310923557898916,
      "learning_rate": 2.4395226496182176e-06,
      "loss": 0.2647,
      "step": 3991
    },
    {
      "epoch": 2.871426002517533,
      "grad_norm": 5.384865722855087,
      "learning_rate": 2.439257390006011e-06,
      "loss": 0.1457,
      "step": 3992
    },
    {
      "epoch": 2.872145297608344,
      "grad_norm": 1.9490482562957219,
      "learning_rate": 2.438992082067362e-06,
      "loss": 0.0288,
      "step": 3993
    },
    {
      "epoch": 2.872864592699155,
      "grad_norm": 0.6213338602229772,
      "learning_rate": 2.438726725815922e-06,
      "loss": 0.003,
      "step": 3994
    },
    {
      "epoch": 2.873583887789966,
      "grad_norm": 1.1257479061043243,
      "learning_rate": 2.4384613212653447e-06,
      "loss": 0.0047,
      "step": 3995
    },
    {
      "epoch": 2.8743031828807766,
      "grad_norm": 1.5047134781980294,
      "learning_rate": 2.4381958684292852e-06,
      "loss": 0.0232,
      "step": 3996
    },
    {
      "epoch": 2.875022477971588,
      "grad_norm": 2.8033836154383045,
      "learning_rate": 2.437930367321401e-06,
      "loss": 0.0474,
      "step": 3997
    },
    {
      "epoch": 2.8757417730623986,
      "grad_norm": 2.191107533495528,
      "learning_rate": 2.4376648179553525e-06,
      "loss": 0.0501,
      "step": 3998
    },
    {
      "epoch": 2.87646106815321,
      "grad_norm": 4.106771730558472,
      "learning_rate": 2.4373992203448033e-06,
      "loss": 0.093,
      "step": 3999
    },
    {
      "epoch": 2.8771803632440207,
      "grad_norm": 4.079746737069633,
      "learning_rate": 2.4371335745034187e-06,
      "loss": 0.1954,
      "step": 4000
    },
    {
      "epoch": 2.8778996583348317,
      "grad_norm": 5.605464788763568,
      "learning_rate": 2.436867880444867e-06,
      "loss": 0.082,
      "step": 4001
    },
    {
      "epoch": 2.878618953425643,
      "grad_norm": 1.871823648892638,
      "learning_rate": 2.4366021381828177e-06,
      "loss": 0.0428,
      "step": 4002
    },
    {
      "epoch": 2.879338248516454,
      "grad_norm": 0.06567784626870671,
      "learning_rate": 2.4363363477309446e-06,
      "loss": 0.0002,
      "step": 4003
    },
    {
      "epoch": 2.880057543607265,
      "grad_norm": 2.8194330758604726,
      "learning_rate": 2.4360705091029225e-06,
      "loss": 0.0395,
      "step": 4004
    },
    {
      "epoch": 2.880776838698076,
      "grad_norm": 2.703882846444028,
      "learning_rate": 2.43580462231243e-06,
      "loss": 0.0175,
      "step": 4005
    },
    {
      "epoch": 2.881496133788887,
      "grad_norm": 4.932939455022974,
      "learning_rate": 2.4355386873731465e-06,
      "loss": 0.1408,
      "step": 4006
    },
    {
      "epoch": 2.882215428879698,
      "grad_norm": 3.0557873329567626,
      "learning_rate": 2.435272704298755e-06,
      "loss": 0.1338,
      "step": 4007
    },
    {
      "epoch": 2.882934723970509,
      "grad_norm": 6.306904775007282,
      "learning_rate": 2.435006673102941e-06,
      "loss": 0.1457,
      "step": 4008
    },
    {
      "epoch": 2.88365401906132,
      "grad_norm": 0.8317877104009143,
      "learning_rate": 2.4347405937993928e-06,
      "loss": 0.0089,
      "step": 4009
    },
    {
      "epoch": 2.884373314152131,
      "grad_norm": 4.8348194107271985,
      "learning_rate": 2.4344744664018e-06,
      "loss": 0.2038,
      "step": 4010
    },
    {
      "epoch": 2.8850926092429416,
      "grad_norm": 4.047458302412651,
      "learning_rate": 2.434208290923855e-06,
      "loss": 0.2075,
      "step": 4011
    },
    {
      "epoch": 2.885811904333753,
      "grad_norm": 9.477334823381563,
      "learning_rate": 2.4339420673792547e-06,
      "loss": 0.2474,
      "step": 4012
    },
    {
      "epoch": 2.8865311994245637,
      "grad_norm": 5.367446599454089,
      "learning_rate": 2.4336757957816946e-06,
      "loss": 0.2452,
      "step": 4013
    },
    {
      "epoch": 2.887250494515375,
      "grad_norm": 2.3864766391157275,
      "learning_rate": 2.433409476144876e-06,
      "loss": 0.0419,
      "step": 4014
    },
    {
      "epoch": 2.8879697896061858,
      "grad_norm": 3.6073024638919846,
      "learning_rate": 2.433143108482501e-06,
      "loss": 0.047,
      "step": 4015
    },
    {
      "epoch": 2.888689084696997,
      "grad_norm": 3.2261438181962165,
      "learning_rate": 2.4328766928082754e-06,
      "loss": 0.1055,
      "step": 4016
    },
    {
      "epoch": 2.889408379787808,
      "grad_norm": 3.953972455532349,
      "learning_rate": 2.4326102291359064e-06,
      "loss": 0.0472,
      "step": 4017
    },
    {
      "epoch": 2.890127674878619,
      "grad_norm": 4.349582275879489,
      "learning_rate": 2.4323437174791035e-06,
      "loss": 0.1768,
      "step": 4018
    },
    {
      "epoch": 2.89084696996943,
      "grad_norm": 0.23608956034537712,
      "learning_rate": 2.43207715785158e-06,
      "loss": 0.0009,
      "step": 4019
    },
    {
      "epoch": 2.891566265060241,
      "grad_norm": 4.629453085014154,
      "learning_rate": 2.4318105502670505e-06,
      "loss": 0.2065,
      "step": 4020
    },
    {
      "epoch": 2.892285560151052,
      "grad_norm": 2.8321606790857743,
      "learning_rate": 2.431543894739232e-06,
      "loss": 0.0741,
      "step": 4021
    },
    {
      "epoch": 2.893004855241863,
      "grad_norm": 3.857402773719244,
      "learning_rate": 2.431277191281846e-06,
      "loss": 0.0169,
      "step": 4022
    },
    {
      "epoch": 2.893724150332674,
      "grad_norm": 2.949516060112828,
      "learning_rate": 2.431010439908612e-06,
      "loss": 0.0266,
      "step": 4023
    },
    {
      "epoch": 2.894443445423485,
      "grad_norm": 1.7563058374528309,
      "learning_rate": 2.430743640633258e-06,
      "loss": 0.0319,
      "step": 4024
    },
    {
      "epoch": 2.895162740514296,
      "grad_norm": 2.13374950469749,
      "learning_rate": 2.4304767934695094e-06,
      "loss": 0.0803,
      "step": 4025
    },
    {
      "epoch": 2.895882035605107,
      "grad_norm": 5.829500039613066,
      "learning_rate": 2.4302098984310963e-06,
      "loss": 0.102,
      "step": 4026
    },
    {
      "epoch": 2.896601330695918,
      "grad_norm": 1.000359921258487,
      "learning_rate": 2.4299429555317506e-06,
      "loss": 0.0095,
      "step": 4027
    },
    {
      "epoch": 2.8973206257867288,
      "grad_norm": 2.273987849521096,
      "learning_rate": 2.429675964785208e-06,
      "loss": 0.0982,
      "step": 4028
    },
    {
      "epoch": 2.8980399208775403,
      "grad_norm": 4.377851940181564,
      "learning_rate": 2.4294089262052047e-06,
      "loss": 0.1295,
      "step": 4029
    },
    {
      "epoch": 2.898759215968351,
      "grad_norm": 1.2304655173003995,
      "learning_rate": 2.4291418398054805e-06,
      "loss": 0.0252,
      "step": 4030
    },
    {
      "epoch": 2.899478511059162,
      "grad_norm": 7.747474029538773,
      "learning_rate": 2.428874705599778e-06,
      "loss": 0.2892,
      "step": 4031
    },
    {
      "epoch": 2.900197806149973,
      "grad_norm": 3.5166445876424732,
      "learning_rate": 2.4286075236018413e-06,
      "loss": 0.0571,
      "step": 4032
    },
    {
      "epoch": 2.900917101240784,
      "grad_norm": 7.080811210758189,
      "learning_rate": 2.4283402938254167e-06,
      "loss": 0.1347,
      "step": 4033
    },
    {
      "epoch": 2.901636396331595,
      "grad_norm": 3.6410186993151767,
      "learning_rate": 2.428073016284255e-06,
      "loss": 0.0872,
      "step": 4034
    },
    {
      "epoch": 2.902355691422406,
      "grad_norm": 0.5075037744799848,
      "learning_rate": 2.427805690992107e-06,
      "loss": 0.0009,
      "step": 4035
    },
    {
      "epoch": 2.903074986513217,
      "grad_norm": 4.944652242433811,
      "learning_rate": 2.427538317962728e-06,
      "loss": 0.1087,
      "step": 4036
    },
    {
      "epoch": 2.903794281604028,
      "grad_norm": 2.71077036535615,
      "learning_rate": 2.4272708972098743e-06,
      "loss": 0.0961,
      "step": 4037
    },
    {
      "epoch": 2.904513576694839,
      "grad_norm": 3.177434253173833,
      "learning_rate": 2.4270034287473045e-06,
      "loss": 0.0752,
      "step": 4038
    },
    {
      "epoch": 2.90523287178565,
      "grad_norm": 5.699450624542296,
      "learning_rate": 2.4267359125887813e-06,
      "loss": 0.2963,
      "step": 4039
    },
    {
      "epoch": 2.905952166876461,
      "grad_norm": 3.3445526437650117,
      "learning_rate": 2.4264683487480685e-06,
      "loss": 0.0884,
      "step": 4040
    },
    {
      "epoch": 2.906671461967272,
      "grad_norm": 9.414000578171455,
      "learning_rate": 2.426200737238933e-06,
      "loss": 0.1655,
      "step": 4041
    },
    {
      "epoch": 2.9073907570580833,
      "grad_norm": 5.457660715804781,
      "learning_rate": 2.4259330780751435e-06,
      "loss": 0.0939,
      "step": 4042
    },
    {
      "epoch": 2.908110052148894,
      "grad_norm": 5.06964764595076,
      "learning_rate": 2.425665371270472e-06,
      "loss": 0.0987,
      "step": 4043
    },
    {
      "epoch": 2.9088293472397053,
      "grad_norm": 1.8295566895353454,
      "learning_rate": 2.4253976168386917e-06,
      "loss": 0.0274,
      "step": 4044
    },
    {
      "epoch": 2.909548642330516,
      "grad_norm": 2.3415930447779005,
      "learning_rate": 2.42512981479358e-06,
      "loss": 0.0749,
      "step": 4045
    },
    {
      "epoch": 2.910267937421327,
      "grad_norm": 2.6951652858207837,
      "learning_rate": 2.424861965148915e-06,
      "loss": 0.0604,
      "step": 4046
    },
    {
      "epoch": 2.910987232512138,
      "grad_norm": 2.109786192311443,
      "learning_rate": 2.4245940679184787e-06,
      "loss": 0.046,
      "step": 4047
    },
    {
      "epoch": 2.911706527602949,
      "grad_norm": 6.558879078971595,
      "learning_rate": 2.4243261231160543e-06,
      "loss": 0.2115,
      "step": 4048
    },
    {
      "epoch": 2.91242582269376,
      "grad_norm": 3.884864128916162,
      "learning_rate": 2.424058130755428e-06,
      "loss": 0.1546,
      "step": 4049
    },
    {
      "epoch": 2.913145117784571,
      "grad_norm": 4.692880484306014,
      "learning_rate": 2.4237900908503896e-06,
      "loss": 0.1718,
      "step": 4050
    },
    {
      "epoch": 2.913864412875382,
      "grad_norm": 3.838108473194209,
      "learning_rate": 2.4235220034147286e-06,
      "loss": 0.1605,
      "step": 4051
    },
    {
      "epoch": 2.914583707966193,
      "grad_norm": 3.7062481520301174,
      "learning_rate": 2.423253868462239e-06,
      "loss": 0.1137,
      "step": 4052
    },
    {
      "epoch": 2.915303003057004,
      "grad_norm": 2.3533508807442103,
      "learning_rate": 2.422985686006718e-06,
      "loss": 0.0316,
      "step": 4053
    },
    {
      "epoch": 2.916022298147815,
      "grad_norm": 4.785612029882363,
      "learning_rate": 2.422717456061963e-06,
      "loss": 0.1106,
      "step": 4054
    },
    {
      "epoch": 2.9167415932386263,
      "grad_norm": 2.436704157359387,
      "learning_rate": 2.422449178641775e-06,
      "loss": 0.0337,
      "step": 4055
    },
    {
      "epoch": 2.9174608883294373,
      "grad_norm": 3.5728247008084706,
      "learning_rate": 2.4221808537599577e-06,
      "loss": 0.1106,
      "step": 4056
    },
    {
      "epoch": 2.9181801834202483,
      "grad_norm": 5.389854919641401,
      "learning_rate": 2.421912481430316e-06,
      "loss": 0.2431,
      "step": 4057
    },
    {
      "epoch": 2.918899478511059,
      "grad_norm": 3.687793091354124,
      "learning_rate": 2.4216440616666593e-06,
      "loss": 0.0967,
      "step": 4058
    },
    {
      "epoch": 2.9196187736018704,
      "grad_norm": 5.042936774376561,
      "learning_rate": 2.4213755944827978e-06,
      "loss": 0.1393,
      "step": 4059
    },
    {
      "epoch": 2.920338068692681,
      "grad_norm": 2.916097533936283,
      "learning_rate": 2.421107079892544e-06,
      "loss": 0.0247,
      "step": 4060
    },
    {
      "epoch": 2.921057363783492,
      "grad_norm": 2.4429610804838493,
      "learning_rate": 2.4208385179097143e-06,
      "loss": 0.0585,
      "step": 4061
    },
    {
      "epoch": 2.921776658874303,
      "grad_norm": 3.999142528298892,
      "learning_rate": 2.4205699085481263e-06,
      "loss": 0.1089,
      "step": 4062
    },
    {
      "epoch": 2.922495953965114,
      "grad_norm": 1.2621150856212513,
      "learning_rate": 2.4203012518216003e-06,
      "loss": 0.0026,
      "step": 4063
    },
    {
      "epoch": 2.923215249055925,
      "grad_norm": 4.286648128816217,
      "learning_rate": 2.42003254774396e-06,
      "loss": 0.176,
      "step": 4064
    },
    {
      "epoch": 2.923934544146736,
      "grad_norm": 1.404824649710889,
      "learning_rate": 2.4197637963290296e-06,
      "loss": 0.0032,
      "step": 4065
    },
    {
      "epoch": 2.924653839237547,
      "grad_norm": 2.7698107332444546,
      "learning_rate": 2.419494997590637e-06,
      "loss": 0.0968,
      "step": 4066
    },
    {
      "epoch": 2.925373134328358,
      "grad_norm": 2.3330046691836754,
      "learning_rate": 2.4192261515426137e-06,
      "loss": 0.074,
      "step": 4067
    },
    {
      "epoch": 2.9260924294191692,
      "grad_norm": 5.4132219419039895,
      "learning_rate": 2.4189572581987904e-06,
      "loss": 0.0865,
      "step": 4068
    },
    {
      "epoch": 2.9268117245099803,
      "grad_norm": 4.36362305725598,
      "learning_rate": 2.418688317573003e-06,
      "loss": 0.1926,
      "step": 4069
    },
    {
      "epoch": 2.9275310196007913,
      "grad_norm": 3.9338084921022762,
      "learning_rate": 2.418419329679089e-06,
      "loss": 0.1855,
      "step": 4070
    },
    {
      "epoch": 2.9282503146916024,
      "grad_norm": 5.427174204663504,
      "learning_rate": 2.4181502945308884e-06,
      "loss": 0.2282,
      "step": 4071
    },
    {
      "epoch": 2.9289696097824134,
      "grad_norm": 6.510650692511353,
      "learning_rate": 2.4178812121422433e-06,
      "loss": 0.209,
      "step": 4072
    },
    {
      "epoch": 2.929688904873224,
      "grad_norm": 3.3937190120295377,
      "learning_rate": 2.4176120825269984e-06,
      "loss": 0.132,
      "step": 4073
    },
    {
      "epoch": 2.9304081999640355,
      "grad_norm": 4.374561865419099,
      "learning_rate": 2.4173429056990015e-06,
      "loss": 0.0982,
      "step": 4074
    },
    {
      "epoch": 2.931127495054846,
      "grad_norm": 6.1255230612342135,
      "learning_rate": 2.4170736816721014e-06,
      "loss": 0.0456,
      "step": 4075
    },
    {
      "epoch": 2.9318467901456575,
      "grad_norm": 3.9525807601560303,
      "learning_rate": 2.4168044104601506e-06,
      "loss": 0.1562,
      "step": 4076
    },
    {
      "epoch": 2.932566085236468,
      "grad_norm": 2.1090112562012178,
      "learning_rate": 2.4165350920770032e-06,
      "loss": 0.061,
      "step": 4077
    },
    {
      "epoch": 2.933285380327279,
      "grad_norm": 3.8221194495351583,
      "learning_rate": 2.416265726536517e-06,
      "loss": 0.1289,
      "step": 4078
    },
    {
      "epoch": 2.93400467541809,
      "grad_norm": 5.441863010607799,
      "learning_rate": 2.4159963138525504e-06,
      "loss": 0.0359,
      "step": 4079
    },
    {
      "epoch": 2.934723970508901,
      "grad_norm": 3.5601061736795105,
      "learning_rate": 2.415726854038965e-06,
      "loss": 0.0479,
      "step": 4080
    },
    {
      "epoch": 2.9354432655997122,
      "grad_norm": 1.6543937266672544,
      "learning_rate": 2.4154573471096264e-06,
      "loss": 0.0463,
      "step": 4081
    },
    {
      "epoch": 2.9361625606905233,
      "grad_norm": 4.292833439786673,
      "learning_rate": 2.4151877930783995e-06,
      "loss": 0.11,
      "step": 4082
    },
    {
      "epoch": 2.9368818557813343,
      "grad_norm": 0.10876797326340275,
      "learning_rate": 2.4149181919591543e-06,
      "loss": 0.0004,
      "step": 4083
    },
    {
      "epoch": 2.9376011508721454,
      "grad_norm": 7.594886590458589,
      "learning_rate": 2.414648543765762e-06,
      "loss": 0.2957,
      "step": 4084
    },
    {
      "epoch": 2.9383204459629564,
      "grad_norm": 4.326872976514883,
      "learning_rate": 2.4143788485120967e-06,
      "loss": 0.0915,
      "step": 4085
    },
    {
      "epoch": 2.9390397410537674,
      "grad_norm": 1.7994498309696922,
      "learning_rate": 2.4141091062120345e-06,
      "loss": 0.0382,
      "step": 4086
    },
    {
      "epoch": 2.9397590361445785,
      "grad_norm": 0.015075957072079137,
      "learning_rate": 2.413839316879454e-06,
      "loss": 0.0001,
      "step": 4087
    },
    {
      "epoch": 2.940478331235389,
      "grad_norm": 4.7113750136763155,
      "learning_rate": 2.4135694805282363e-06,
      "loss": 0.1854,
      "step": 4088
    },
    {
      "epoch": 2.9411976263262005,
      "grad_norm": 5.076628862837255,
      "learning_rate": 2.4132995971722657e-06,
      "loss": 0.0923,
      "step": 4089
    },
    {
      "epoch": 2.941916921417011,
      "grad_norm": 5.084939676213631,
      "learning_rate": 2.413029666825427e-06,
      "loss": 0.1131,
      "step": 4090
    },
    {
      "epoch": 2.9426362165078226,
      "grad_norm": 1.483487642108858,
      "learning_rate": 2.412759689501609e-06,
      "loss": 0.0252,
      "step": 4091
    },
    {
      "epoch": 2.943355511598633,
      "grad_norm": 0.5106460693582094,
      "learning_rate": 2.412489665214703e-06,
      "loss": 0.0009,
      "step": 4092
    },
    {
      "epoch": 2.944074806689444,
      "grad_norm": 1.9776705996474475,
      "learning_rate": 2.412219593978602e-06,
      "loss": 0.009,
      "step": 4093
    },
    {
      "epoch": 2.9447941017802552,
      "grad_norm": 2.5431989767215004,
      "learning_rate": 2.4119494758072014e-06,
      "loss": 0.0105,
      "step": 4094
    },
    {
      "epoch": 2.9455133968710663,
      "grad_norm": 0.13345312251002414,
      "learning_rate": 2.4116793107143997e-06,
      "loss": 0.0003,
      "step": 4095
    },
    {
      "epoch": 2.9462326919618773,
      "grad_norm": 2.277061590681093,
      "learning_rate": 2.4114090987140963e-06,
      "loss": 0.0198,
      "step": 4096
    },
    {
      "epoch": 2.9469519870526883,
      "grad_norm": 1.9406600013171222,
      "learning_rate": 2.411138839820195e-06,
      "loss": 0.0568,
      "step": 4097
    },
    {
      "epoch": 2.9476712821434994,
      "grad_norm": 4.042199765571634,
      "learning_rate": 2.4108685340466013e-06,
      "loss": 0.179,
      "step": 4098
    },
    {
      "epoch": 2.9483905772343104,
      "grad_norm": 3.1920518577664327,
      "learning_rate": 2.410598181407223e-06,
      "loss": 0.0625,
      "step": 4099
    },
    {
      "epoch": 2.9491098723251215,
      "grad_norm": 2.06695238138367,
      "learning_rate": 2.4103277819159686e-06,
      "loss": 0.0603,
      "step": 4100
    },
    {
      "epoch": 2.9498291674159325,
      "grad_norm": 1.662737001137136,
      "learning_rate": 2.410057335586753e-06,
      "loss": 0.0036,
      "step": 4101
    },
    {
      "epoch": 2.9505484625067435,
      "grad_norm": 3.2805203402024303,
      "learning_rate": 2.4097868424334896e-06,
      "loss": 0.1487,
      "step": 4102
    },
    {
      "epoch": 2.9512677575975546,
      "grad_norm": 4.528886246289247,
      "learning_rate": 2.4095163024700957e-06,
      "loss": 0.0906,
      "step": 4103
    },
    {
      "epoch": 2.9519870526883656,
      "grad_norm": 0.8292465952872653,
      "learning_rate": 2.4092457157104913e-06,
      "loss": 0.002,
      "step": 4104
    },
    {
      "epoch": 2.952706347779176,
      "grad_norm": 2.8809842182877103,
      "learning_rate": 2.408975082168599e-06,
      "loss": 0.0827,
      "step": 4105
    },
    {
      "epoch": 2.9534256428699877,
      "grad_norm": 5.588086998950786,
      "learning_rate": 2.408704401858343e-06,
      "loss": 0.126,
      "step": 4106
    },
    {
      "epoch": 2.9541449379607982,
      "grad_norm": 1.8061928646134922,
      "learning_rate": 2.4084336747936505e-06,
      "loss": 0.0644,
      "step": 4107
    },
    {
      "epoch": 2.9548642330516093,
      "grad_norm": 3.5734074429837968,
      "learning_rate": 2.4081629009884504e-06,
      "loss": 0.1727,
      "step": 4108
    },
    {
      "epoch": 2.9555835281424203,
      "grad_norm": 2.178123590626089,
      "learning_rate": 2.407892080456675e-06,
      "loss": 0.0556,
      "step": 4109
    },
    {
      "epoch": 2.9563028232332313,
      "grad_norm": 2.6174168141438092,
      "learning_rate": 2.4076212132122586e-06,
      "loss": 0.1022,
      "step": 4110
    },
    {
      "epoch": 2.9570221183240424,
      "grad_norm": 4.69771767275535,
      "learning_rate": 2.4073502992691374e-06,
      "loss": 0.1327,
      "step": 4111
    },
    {
      "epoch": 2.9577414134148534,
      "grad_norm": 2.220444234192975,
      "learning_rate": 2.40707933864125e-06,
      "loss": 0.0328,
      "step": 4112
    },
    {
      "epoch": 2.9584607085056644,
      "grad_norm": 2.6102616753250474,
      "learning_rate": 2.4068083313425387e-06,
      "loss": 0.0048,
      "step": 4113
    },
    {
      "epoch": 2.9591800035964755,
      "grad_norm": 6.103492458754118,
      "learning_rate": 2.406537277386947e-06,
      "loss": 0.1393,
      "step": 4114
    },
    {
      "epoch": 2.9598992986872865,
      "grad_norm": 4.016951525446478,
      "learning_rate": 2.4062661767884207e-06,
      "loss": 0.0847,
      "step": 4115
    },
    {
      "epoch": 2.9606185937780976,
      "grad_norm": 3.433703290909037,
      "learning_rate": 2.405995029560909e-06,
      "loss": 0.0888,
      "step": 4116
    },
    {
      "epoch": 2.9613378888689086,
      "grad_norm": 3.5172940546188016,
      "learning_rate": 2.405723835718362e-06,
      "loss": 0.0991,
      "step": 4117
    },
    {
      "epoch": 2.9620571839597196,
      "grad_norm": 3.728253007345405,
      "learning_rate": 2.4054525952747345e-06,
      "loss": 0.1529,
      "step": 4118
    },
    {
      "epoch": 2.9627764790505307,
      "grad_norm": 3.5878716793670042,
      "learning_rate": 2.4051813082439813e-06,
      "loss": 0.1431,
      "step": 4119
    },
    {
      "epoch": 2.9634957741413412,
      "grad_norm": 2.0671486323487356,
      "learning_rate": 2.40490997464006e-06,
      "loss": 0.0621,
      "step": 4120
    },
    {
      "epoch": 2.9642150692321527,
      "grad_norm": 4.068389413588118,
      "learning_rate": 2.404638594476933e-06,
      "loss": 0.0798,
      "step": 4121
    },
    {
      "epoch": 2.9649343643229633,
      "grad_norm": 1.6161380640706662,
      "learning_rate": 2.404367167768562e-06,
      "loss": 0.0346,
      "step": 4122
    },
    {
      "epoch": 2.9656536594137743,
      "grad_norm": 3.357229265397604,
      "learning_rate": 2.4040956945289126e-06,
      "loss": 0.0074,
      "step": 4123
    },
    {
      "epoch": 2.9663729545045854,
      "grad_norm": 5.914602916531723,
      "learning_rate": 2.4038241747719528e-06,
      "loss": 0.3287,
      "step": 4124
    },
    {
      "epoch": 2.9670922495953964,
      "grad_norm": 1.9470666319214112,
      "learning_rate": 2.403552608511652e-06,
      "loss": 0.0156,
      "step": 4125
    },
    {
      "epoch": 2.9678115446862074,
      "grad_norm": 12.440589763005336,
      "learning_rate": 2.4032809957619838e-06,
      "loss": 0.1262,
      "step": 4126
    },
    {
      "epoch": 2.9685308397770185,
      "grad_norm": 3.663332747719266,
      "learning_rate": 2.403009336536923e-06,
      "loss": 0.1644,
      "step": 4127
    },
    {
      "epoch": 2.9692501348678295,
      "grad_norm": 2.646464936106486,
      "learning_rate": 2.402737630850446e-06,
      "loss": 0.0884,
      "step": 4128
    },
    {
      "epoch": 2.9699694299586405,
      "grad_norm": 4.9178212709685685,
      "learning_rate": 2.4024658787165335e-06,
      "loss": 0.1932,
      "step": 4129
    },
    {
      "epoch": 2.9706887250494516,
      "grad_norm": 3.7730786708412256,
      "learning_rate": 2.402194080149167e-06,
      "loss": 0.0667,
      "step": 4130
    },
    {
      "epoch": 2.9714080201402626,
      "grad_norm": 4.759544024134562,
      "learning_rate": 2.401922235162332e-06,
      "loss": 0.1953,
      "step": 4131
    },
    {
      "epoch": 2.9721273152310737,
      "grad_norm": 3.1101232914634624,
      "learning_rate": 2.401650343770014e-06,
      "loss": 0.0312,
      "step": 4132
    },
    {
      "epoch": 2.9728466103218847,
      "grad_norm": 2.3170645809851673,
      "learning_rate": 2.4013784059862038e-06,
      "loss": 0.075,
      "step": 4133
    },
    {
      "epoch": 2.9735659054126957,
      "grad_norm": 2.056849305860185,
      "learning_rate": 2.401106421824892e-06,
      "loss": 0.08,
      "step": 4134
    },
    {
      "epoch": 2.9742852005035063,
      "grad_norm": 6.622587287512343,
      "learning_rate": 2.4008343913000727e-06,
      "loss": 0.1385,
      "step": 4135
    },
    {
      "epoch": 2.975004495594318,
      "grad_norm": 1.9454634426209558,
      "learning_rate": 2.4005623144257423e-06,
      "loss": 0.046,
      "step": 4136
    },
    {
      "epoch": 2.9757237906851284,
      "grad_norm": 2.5591183213031656,
      "learning_rate": 2.4002901912159005e-06,
      "loss": 0.0302,
      "step": 4137
    },
    {
      "epoch": 2.97644308577594,
      "grad_norm": 3.3934713244465393,
      "learning_rate": 2.400018021684548e-06,
      "loss": 0.1026,
      "step": 4138
    },
    {
      "epoch": 2.9771623808667504,
      "grad_norm": 3.7655848725890886,
      "learning_rate": 2.399745805845688e-06,
      "loss": 0.1699,
      "step": 4139
    },
    {
      "epoch": 2.9778816759575615,
      "grad_norm": 2.6937700686001893,
      "learning_rate": 2.399473543713327e-06,
      "loss": 0.0716,
      "step": 4140
    },
    {
      "epoch": 2.9786009710483725,
      "grad_norm": 13.1724771866324,
      "learning_rate": 2.399201235301473e-06,
      "loss": 0.2103,
      "step": 4141
    },
    {
      "epoch": 2.9793202661391835,
      "grad_norm": 3.174580183326379,
      "learning_rate": 2.3989288806241372e-06,
      "loss": 0.0949,
      "step": 4142
    },
    {
      "epoch": 2.9800395612299946,
      "grad_norm": 2.368162761869619,
      "learning_rate": 2.398656479695332e-06,
      "loss": 0.0288,
      "step": 4143
    },
    {
      "epoch": 2.9807588563208056,
      "grad_norm": 4.826055368228739,
      "learning_rate": 2.3983840325290744e-06,
      "loss": 0.0422,
      "step": 4144
    },
    {
      "epoch": 2.9814781514116167,
      "grad_norm": 1.4393542045677177,
      "learning_rate": 2.3981115391393805e-06,
      "loss": 0.0049,
      "step": 4145
    },
    {
      "epoch": 2.9821974465024277,
      "grad_norm": 2.8331877704508943,
      "learning_rate": 2.3978389995402718e-06,
      "loss": 0.0906,
      "step": 4146
    },
    {
      "epoch": 2.9829167415932387,
      "grad_norm": 3.698643699422827,
      "learning_rate": 2.39756641374577e-06,
      "loss": 0.1112,
      "step": 4147
    },
    {
      "epoch": 2.9836360366840498,
      "grad_norm": 4.043395873149081,
      "learning_rate": 2.397293781769901e-06,
      "loss": 0.157,
      "step": 4148
    },
    {
      "epoch": 2.984355331774861,
      "grad_norm": 7.529882959656265,
      "learning_rate": 2.397021103626691e-06,
      "loss": 0.1253,
      "step": 4149
    },
    {
      "epoch": 2.9850746268656714,
      "grad_norm": 5.1465982499934375,
      "learning_rate": 2.396748379330171e-06,
      "loss": 0.0571,
      "step": 4150
    },
    {
      "epoch": 2.985793921956483,
      "grad_norm": 4.1784428736381,
      "learning_rate": 2.3964756088943726e-06,
      "loss": 0.1348,
      "step": 4151
    },
    {
      "epoch": 2.9865132170472934,
      "grad_norm": 4.2778857605226825,
      "learning_rate": 2.3962027923333308e-06,
      "loss": 0.0214,
      "step": 4152
    },
    {
      "epoch": 2.987232512138105,
      "grad_norm": 2.0542372568580376,
      "learning_rate": 2.395929929661082e-06,
      "loss": 0.0288,
      "step": 4153
    },
    {
      "epoch": 2.9879518072289155,
      "grad_norm": 3.854234059046007,
      "learning_rate": 2.395657020891665e-06,
      "loss": 0.1617,
      "step": 4154
    },
    {
      "epoch": 2.9886711023197265,
      "grad_norm": 2.8655471471596616,
      "learning_rate": 2.3953840660391227e-06,
      "loss": 0.1316,
      "step": 4155
    },
    {
      "epoch": 2.9893903974105376,
      "grad_norm": 4.3688363590003165,
      "learning_rate": 2.395111065117498e-06,
      "loss": 0.1842,
      "step": 4156
    },
    {
      "epoch": 2.9901096925013486,
      "grad_norm": 3.1108634363265653,
      "learning_rate": 2.394838018140838e-06,
      "loss": 0.0865,
      "step": 4157
    },
    {
      "epoch": 2.9908289875921596,
      "grad_norm": 4.690860341645325,
      "learning_rate": 2.3945649251231912e-06,
      "loss": 0.0129,
      "step": 4158
    },
    {
      "epoch": 2.9915482826829707,
      "grad_norm": 2.745800892612085,
      "learning_rate": 2.3942917860786078e-06,
      "loss": 0.0111,
      "step": 4159
    },
    {
      "epoch": 2.9922675777737817,
      "grad_norm": 1.3408737233608998,
      "learning_rate": 2.3940186010211427e-06,
      "loss": 0.0077,
      "step": 4160
    },
    {
      "epoch": 2.9929868728645928,
      "grad_norm": 1.9225141314379461,
      "learning_rate": 2.393745369964851e-06,
      "loss": 0.0089,
      "step": 4161
    },
    {
      "epoch": 2.993706167955404,
      "grad_norm": 1.652740339238617,
      "learning_rate": 2.3934720929237915e-06,
      "loss": 0.0123,
      "step": 4162
    },
    {
      "epoch": 2.994425463046215,
      "grad_norm": 3.463711466785171,
      "learning_rate": 2.3931987699120244e-06,
      "loss": 0.0234,
      "step": 4163
    },
    {
      "epoch": 2.995144758137026,
      "grad_norm": 5.3191335017786985,
      "learning_rate": 2.3929254009436123e-06,
      "loss": 0.1326,
      "step": 4164
    },
    {
      "epoch": 2.9958640532278364,
      "grad_norm": 2.639773402631255,
      "learning_rate": 2.392651986032621e-06,
      "loss": 0.0689,
      "step": 4165
    },
    {
      "epoch": 2.996583348318648,
      "grad_norm": 0.2142367408113003,
      "learning_rate": 2.392378525193118e-06,
      "loss": 0.0005,
      "step": 4166
    },
    {
      "epoch": 2.9973026434094585,
      "grad_norm": 2.5295504667030615,
      "learning_rate": 2.392105018439173e-06,
      "loss": 0.0459,
      "step": 4167
    },
    {
      "epoch": 2.99802193850027,
      "grad_norm": 4.44897061246736,
      "learning_rate": 2.3918314657848593e-06,
      "loss": 0.0974,
      "step": 4168
    },
    {
      "epoch": 2.9987412335910806,
      "grad_norm": 3.359496912409121,
      "learning_rate": 2.3915578672442507e-06,
      "loss": 0.0505,
      "step": 4169
    },
    {
      "epoch": 2.9994605286818916,
      "grad_norm": 2.470114387979346,
      "learning_rate": 2.3912842228314247e-06,
      "loss": 0.0072,
      "step": 4170
    },
    {
      "epoch": 3.0001798237727026,
      "grad_norm": 3.59122860591066,
      "learning_rate": 2.3910105325604615e-06,
      "loss": 0.0332,
      "step": 4171
    },
    {
      "epoch": 3.0008991188635137,
      "grad_norm": 0.6169503159338277,
      "learning_rate": 2.3907367964454414e-06,
      "loss": 0.0023,
      "step": 4172
    },
    {
      "epoch": 3.0016184139543247,
      "grad_norm": 0.9037409611200752,
      "learning_rate": 2.39046301450045e-06,
      "loss": 0.0036,
      "step": 4173
    },
    {
      "epoch": 3.0023377090451357,
      "grad_norm": 3.9201807625746623,
      "learning_rate": 2.3901891867395736e-06,
      "loss": 0.2522,
      "step": 4174
    },
    {
      "epoch": 3.003057004135947,
      "grad_norm": 1.7476328669061456,
      "learning_rate": 2.3899153131769004e-06,
      "loss": 0.0408,
      "step": 4175
    },
    {
      "epoch": 3.003776299226758,
      "grad_norm": 5.552091647776159,
      "learning_rate": 2.3896413938265225e-06,
      "loss": 0.243,
      "step": 4176
    },
    {
      "epoch": 3.004495594317569,
      "grad_norm": 1.8745584996485445,
      "learning_rate": 2.389367428702533e-06,
      "loss": 0.0279,
      "step": 4177
    },
    {
      "epoch": 3.00521488940838,
      "grad_norm": 5.520559230484063,
      "learning_rate": 2.3890934178190278e-06,
      "loss": 0.1771,
      "step": 4178
    },
    {
      "epoch": 3.005934184499191,
      "grad_norm": 2.2436343713028415,
      "learning_rate": 2.388819361190106e-06,
      "loss": 0.0779,
      "step": 4179
    },
    {
      "epoch": 3.006653479590002,
      "grad_norm": 3.941050956450359,
      "learning_rate": 2.388545258829867e-06,
      "loss": 0.1301,
      "step": 4180
    },
    {
      "epoch": 3.007372774680813,
      "grad_norm": 1.2230568613484318,
      "learning_rate": 2.3882711107524152e-06,
      "loss": 0.0191,
      "step": 4181
    },
    {
      "epoch": 3.008092069771624,
      "grad_norm": 2.134468316102497,
      "learning_rate": 2.387996916971855e-06,
      "loss": 0.0313,
      "step": 4182
    },
    {
      "epoch": 3.0088113648624346,
      "grad_norm": 2.6230265454737904,
      "learning_rate": 2.3877226775022956e-06,
      "loss": 0.0465,
      "step": 4183
    },
    {
      "epoch": 3.0095306599532456,
      "grad_norm": 2.6253203439982187,
      "learning_rate": 2.3874483923578456e-06,
      "loss": 0.0632,
      "step": 4184
    },
    {
      "epoch": 3.0102499550440567,
      "grad_norm": 3.860021219245699,
      "learning_rate": 2.3871740615526174e-06,
      "loss": 0.0858,
      "step": 4185
    },
    {
      "epoch": 3.0109692501348677,
      "grad_norm": 4.717155763948642,
      "learning_rate": 2.386899685100727e-06,
      "loss": 0.1256,
      "step": 4186
    },
    {
      "epoch": 3.0116885452256787,
      "grad_norm": 1.1979552935091249,
      "learning_rate": 2.3866252630162906e-06,
      "loss": 0.0015,
      "step": 4187
    },
    {
      "epoch": 3.01240784031649,
      "grad_norm": 1.3444719069735305,
      "learning_rate": 2.3863507953134277e-06,
      "loss": 0.032,
      "step": 4188
    },
    {
      "epoch": 3.013127135407301,
      "grad_norm": 3.0218792562505294,
      "learning_rate": 2.3860762820062613e-06,
      "loss": 0.0957,
      "step": 4189
    },
    {
      "epoch": 3.013846430498112,
      "grad_norm": 5.539820067894367,
      "learning_rate": 2.385801723108914e-06,
      "loss": 0.1122,
      "step": 4190
    },
    {
      "epoch": 3.014565725588923,
      "grad_norm": 2.839439274299694,
      "learning_rate": 2.3855271186355133e-06,
      "loss": 0.14,
      "step": 4191
    },
    {
      "epoch": 3.015285020679734,
      "grad_norm": 6.025425588496945,
      "learning_rate": 2.385252468600188e-06,
      "loss": 0.1919,
      "step": 4192
    },
    {
      "epoch": 3.016004315770545,
      "grad_norm": 2.651902982370502,
      "learning_rate": 2.384977773017069e-06,
      "loss": 0.0765,
      "step": 4193
    },
    {
      "epoch": 3.016723610861356,
      "grad_norm": 5.584723435995463,
      "learning_rate": 2.38470303190029e-06,
      "loss": 0.243,
      "step": 4194
    },
    {
      "epoch": 3.017442905952167,
      "grad_norm": 0.21702919331759238,
      "learning_rate": 2.384428245263987e-06,
      "loss": 0.0007,
      "step": 4195
    },
    {
      "epoch": 3.018162201042978,
      "grad_norm": 3.959544896838745,
      "learning_rate": 2.3841534131222983e-06,
      "loss": 0.0944,
      "step": 4196
    },
    {
      "epoch": 3.018881496133789,
      "grad_norm": 8.091757801544954,
      "learning_rate": 2.3838785354893643e-06,
      "loss": 0.0883,
      "step": 4197
    },
    {
      "epoch": 3.0196007912245997,
      "grad_norm": 5.334334448935455,
      "learning_rate": 2.3836036123793284e-06,
      "loss": 0.1804,
      "step": 4198
    },
    {
      "epoch": 3.0203200863154107,
      "grad_norm": 2.4739603545097917,
      "learning_rate": 2.3833286438063347e-06,
      "loss": 0.0344,
      "step": 4199
    },
    {
      "epoch": 3.0210393814062217,
      "grad_norm": 1.8935032969526147,
      "learning_rate": 2.383053629784533e-06,
      "loss": 0.0413,
      "step": 4200
    },
    {
      "epoch": 3.021758676497033,
      "grad_norm": 1.7693898253045113,
      "learning_rate": 2.3827785703280705e-06,
      "loss": 0.049,
      "step": 4201
    },
    {
      "epoch": 3.022477971587844,
      "grad_norm": 0.9341041953345984,
      "learning_rate": 2.3825034654511016e-06,
      "loss": 0.0051,
      "step": 4202
    },
    {
      "epoch": 3.023197266678655,
      "grad_norm": 3.376126724265184,
      "learning_rate": 2.3822283151677804e-06,
      "loss": 0.044,
      "step": 4203
    },
    {
      "epoch": 3.023916561769466,
      "grad_norm": 6.430254996723841,
      "learning_rate": 2.381953119492263e-06,
      "loss": 0.0397,
      "step": 4204
    },
    {
      "epoch": 3.024635856860277,
      "grad_norm": 3.146090713379502,
      "learning_rate": 2.3816778784387097e-06,
      "loss": 0.1337,
      "step": 4205
    },
    {
      "epoch": 3.025355151951088,
      "grad_norm": 0.9404201372790487,
      "learning_rate": 2.381402592021282e-06,
      "loss": 0.0013,
      "step": 4206
    },
    {
      "epoch": 3.026074447041899,
      "grad_norm": 3.040531749485178,
      "learning_rate": 2.3811272602541435e-06,
      "loss": 0.1072,
      "step": 4207
    },
    {
      "epoch": 3.02679374213271,
      "grad_norm": 2.7331294212449406,
      "learning_rate": 2.380851883151461e-06,
      "loss": 0.1086,
      "step": 4208
    },
    {
      "epoch": 3.027513037223521,
      "grad_norm": 0.7813001789305579,
      "learning_rate": 2.3805764607274025e-06,
      "loss": 0.0107,
      "step": 4209
    },
    {
      "epoch": 3.028232332314332,
      "grad_norm": 0.8467279756161453,
      "learning_rate": 2.3803009929961392e-06,
      "loss": 0.0107,
      "step": 4210
    },
    {
      "epoch": 3.028951627405143,
      "grad_norm": 1.9139865378183534,
      "learning_rate": 2.380025479971845e-06,
      "loss": 0.0684,
      "step": 4211
    },
    {
      "epoch": 3.029670922495954,
      "grad_norm": 4.271522356206117,
      "learning_rate": 2.3797499216686944e-06,
      "loss": 0.0632,
      "step": 4212
    },
    {
      "epoch": 3.030390217586765,
      "grad_norm": 3.2269966076346774,
      "learning_rate": 2.379474318100866e-06,
      "loss": 0.0057,
      "step": 4213
    },
    {
      "epoch": 3.031109512677576,
      "grad_norm": 4.38574000228918,
      "learning_rate": 2.3791986692825403e-06,
      "loss": 0.0705,
      "step": 4214
    },
    {
      "epoch": 3.031828807768387,
      "grad_norm": 4.599974964173051,
      "learning_rate": 2.3789229752279e-06,
      "loss": 0.0479,
      "step": 4215
    },
    {
      "epoch": 3.032548102859198,
      "grad_norm": 3.428312100683051,
      "learning_rate": 2.3786472359511294e-06,
      "loss": 0.0975,
      "step": 4216
    },
    {
      "epoch": 3.033267397950009,
      "grad_norm": 2.890842723546386,
      "learning_rate": 2.378371451466416e-06,
      "loss": 0.0647,
      "step": 4217
    },
    {
      "epoch": 3.03398669304082,
      "grad_norm": 0.5021541169120519,
      "learning_rate": 2.3780956217879497e-06,
      "loss": 0.0036,
      "step": 4218
    },
    {
      "epoch": 3.034705988131631,
      "grad_norm": 0.9431453515040041,
      "learning_rate": 2.377819746929922e-06,
      "loss": 0.004,
      "step": 4219
    },
    {
      "epoch": 3.035425283222442,
      "grad_norm": 2.80218662382666,
      "learning_rate": 2.3775438269065277e-06,
      "loss": 0.085,
      "step": 4220
    },
    {
      "epoch": 3.036144578313253,
      "grad_norm": 2.194163801698843,
      "learning_rate": 2.3772678617319625e-06,
      "loss": 0.0053,
      "step": 4221
    },
    {
      "epoch": 3.036863873404064,
      "grad_norm": 0.943174449177465,
      "learning_rate": 2.376991851420426e-06,
      "loss": 0.0035,
      "step": 4222
    },
    {
      "epoch": 3.037583168494875,
      "grad_norm": 1.1345296273371606,
      "learning_rate": 2.3767157959861194e-06,
      "loss": 0.0135,
      "step": 4223
    },
    {
      "epoch": 3.038302463585686,
      "grad_norm": 0.947695332204924,
      "learning_rate": 2.3764396954432463e-06,
      "loss": 0.0026,
      "step": 4224
    },
    {
      "epoch": 3.039021758676497,
      "grad_norm": 2.085304078368595,
      "learning_rate": 2.3761635498060116e-06,
      "loss": 0.0468,
      "step": 4225
    },
    {
      "epoch": 3.039741053767308,
      "grad_norm": 2.747362085775327,
      "learning_rate": 2.375887359088625e-06,
      "loss": 0.0409,
      "step": 4226
    },
    {
      "epoch": 3.040460348858119,
      "grad_norm": 3.118929709318799,
      "learning_rate": 2.3756111233052956e-06,
      "loss": 0.0739,
      "step": 4227
    },
    {
      "epoch": 3.0411796439489303,
      "grad_norm": 5.440501693345429,
      "learning_rate": 2.3753348424702375e-06,
      "loss": 0.1631,
      "step": 4228
    },
    {
      "epoch": 3.041898939039741,
      "grad_norm": 4.482552556565129,
      "learning_rate": 2.3750585165976645e-06,
      "loss": 0.0973,
      "step": 4229
    },
    {
      "epoch": 3.042618234130552,
      "grad_norm": 2.864436996747955,
      "learning_rate": 2.374782145701795e-06,
      "loss": 0.0919,
      "step": 4230
    },
    {
      "epoch": 3.043337529221363,
      "grad_norm": 2.461399628082142,
      "learning_rate": 2.374505729796848e-06,
      "loss": 0.0846,
      "step": 4231
    },
    {
      "epoch": 3.044056824312174,
      "grad_norm": 2.2924828328619595,
      "learning_rate": 2.374229268897047e-06,
      "loss": 0.0085,
      "step": 4232
    },
    {
      "epoch": 3.044776119402985,
      "grad_norm": 2.618717050607476,
      "learning_rate": 2.373952763016615e-06,
      "loss": 0.0666,
      "step": 4233
    },
    {
      "epoch": 3.045495414493796,
      "grad_norm": 1.14162621878143,
      "learning_rate": 2.373676212169779e-06,
      "loss": 0.0029,
      "step": 4234
    },
    {
      "epoch": 3.046214709584607,
      "grad_norm": 2.7276185291154365,
      "learning_rate": 2.3733996163707682e-06,
      "loss": 0.0464,
      "step": 4235
    },
    {
      "epoch": 3.046934004675418,
      "grad_norm": 3.208819015070432,
      "learning_rate": 2.373122975633814e-06,
      "loss": 0.1133,
      "step": 4236
    },
    {
      "epoch": 3.047653299766229,
      "grad_norm": 3.697283228708233,
      "learning_rate": 2.3728462899731503e-06,
      "loss": 0.1071,
      "step": 4237
    },
    {
      "epoch": 3.04837259485704,
      "grad_norm": 4.019303212686577,
      "learning_rate": 2.372569559403012e-06,
      "loss": 0.1478,
      "step": 4238
    },
    {
      "epoch": 3.049091889947851,
      "grad_norm": 2.9207523777158064,
      "learning_rate": 2.372292783937639e-06,
      "loss": 0.0204,
      "step": 4239
    },
    {
      "epoch": 3.049811185038662,
      "grad_norm": 2.9299556598537184,
      "learning_rate": 2.37201596359127e-06,
      "loss": 0.0083,
      "step": 4240
    },
    {
      "epoch": 3.0505304801294733,
      "grad_norm": 2.9216239530004393,
      "learning_rate": 2.3717390983781496e-06,
      "loss": 0.0317,
      "step": 4241
    },
    {
      "epoch": 3.0512497752202843,
      "grad_norm": 2.608884176859807,
      "learning_rate": 2.3714621883125223e-06,
      "loss": 0.0697,
      "step": 4242
    },
    {
      "epoch": 3.0519690703110953,
      "grad_norm": 1.2258199667983332,
      "learning_rate": 2.3711852334086353e-06,
      "loss": 0.0158,
      "step": 4243
    },
    {
      "epoch": 3.052688365401906,
      "grad_norm": 2.4133042469503865,
      "learning_rate": 2.3709082336807394e-06,
      "loss": 0.0997,
      "step": 4244
    },
    {
      "epoch": 3.053407660492717,
      "grad_norm": 0.37056642016267966,
      "learning_rate": 2.3706311891430852e-06,
      "loss": 0.0025,
      "step": 4245
    },
    {
      "epoch": 3.054126955583528,
      "grad_norm": 2.076519312693626,
      "learning_rate": 2.3703540998099285e-06,
      "loss": 0.0499,
      "step": 4246
    },
    {
      "epoch": 3.054846250674339,
      "grad_norm": 3.0397354064361455,
      "learning_rate": 2.370076965695525e-06,
      "loss": 0.0074,
      "step": 4247
    },
    {
      "epoch": 3.05556554576515,
      "grad_norm": 4.750150026330861,
      "learning_rate": 2.3697997868141346e-06,
      "loss": 0.0148,
      "step": 4248
    },
    {
      "epoch": 3.056284840855961,
      "grad_norm": 0.7002074750284241,
      "learning_rate": 2.3695225631800188e-06,
      "loss": 0.0018,
      "step": 4249
    },
    {
      "epoch": 3.057004135946772,
      "grad_norm": 3.513798130996827,
      "learning_rate": 2.3692452948074395e-06,
      "loss": 0.1049,
      "step": 4250
    },
    {
      "epoch": 3.057723431037583,
      "grad_norm": 1.5761236159852587,
      "learning_rate": 2.3689679817106648e-06,
      "loss": 0.0227,
      "step": 4251
    },
    {
      "epoch": 3.058442726128394,
      "grad_norm": 1.8652839386456317,
      "learning_rate": 2.368690623903962e-06,
      "loss": 0.0555,
      "step": 4252
    },
    {
      "epoch": 3.059162021219205,
      "grad_norm": 1.2630611348212153,
      "learning_rate": 2.3684132214016012e-06,
      "loss": 0.0208,
      "step": 4253
    },
    {
      "epoch": 3.0598813163100163,
      "grad_norm": 4.613969455167519,
      "learning_rate": 2.368135774217856e-06,
      "loss": 0.1331,
      "step": 4254
    },
    {
      "epoch": 3.0606006114008273,
      "grad_norm": 7.9408672917409655,
      "learning_rate": 2.367858282367001e-06,
      "loss": 0.2847,
      "step": 4255
    },
    {
      "epoch": 3.0613199064916383,
      "grad_norm": 2.611169267826939,
      "learning_rate": 2.367580745863314e-06,
      "loss": 0.0473,
      "step": 4256
    },
    {
      "epoch": 3.0620392015824494,
      "grad_norm": 4.657516864019772,
      "learning_rate": 2.367303164721075e-06,
      "loss": 0.1567,
      "step": 4257
    },
    {
      "epoch": 3.0627584966732604,
      "grad_norm": 5.592713514327501,
      "learning_rate": 2.367025538954565e-06,
      "loss": 0.1874,
      "step": 4258
    },
    {
      "epoch": 3.0634777917640714,
      "grad_norm": 3.344231730713646,
      "learning_rate": 2.3667478685780696e-06,
      "loss": 0.0883,
      "step": 4259
    },
    {
      "epoch": 3.064197086854882,
      "grad_norm": 4.46359162643568,
      "learning_rate": 2.366470153605875e-06,
      "loss": 0.111,
      "step": 4260
    },
    {
      "epoch": 3.064916381945693,
      "grad_norm": 3.4810248485826074,
      "learning_rate": 2.3661923940522688e-06,
      "loss": 0.0874,
      "step": 4261
    },
    {
      "epoch": 3.065635677036504,
      "grad_norm": 3.5521754248150783,
      "learning_rate": 2.3659145899315443e-06,
      "loss": 0.1066,
      "step": 4262
    },
    {
      "epoch": 3.066354972127315,
      "grad_norm": 2.1156323313741363,
      "learning_rate": 2.3656367412579944e-06,
      "loss": 0.0219,
      "step": 4263
    },
    {
      "epoch": 3.067074267218126,
      "grad_norm": 2.8901210133463264,
      "learning_rate": 2.365358848045914e-06,
      "loss": 0.1019,
      "step": 4264
    },
    {
      "epoch": 3.067793562308937,
      "grad_norm": 0.2997566649560629,
      "learning_rate": 2.365080910309602e-06,
      "loss": 0.0006,
      "step": 4265
    },
    {
      "epoch": 3.068512857399748,
      "grad_norm": 4.486524192683793,
      "learning_rate": 2.3648029280633583e-06,
      "loss": 0.0114,
      "step": 4266
    },
    {
      "epoch": 3.0692321524905593,
      "grad_norm": 6.182350158815735,
      "learning_rate": 2.3645249013214863e-06,
      "loss": 0.028,
      "step": 4267
    },
    {
      "epoch": 3.0699514475813703,
      "grad_norm": 4.254354420486257,
      "learning_rate": 2.36424683009829e-06,
      "loss": 0.1259,
      "step": 4268
    },
    {
      "epoch": 3.0706707426721813,
      "grad_norm": 1.9498966249832805,
      "learning_rate": 2.363968714408077e-06,
      "loss": 0.0569,
      "step": 4269
    },
    {
      "epoch": 3.0713900377629924,
      "grad_norm": 4.125931196586699,
      "learning_rate": 2.3636905542651575e-06,
      "loss": 0.143,
      "step": 4270
    },
    {
      "epoch": 3.0721093328538034,
      "grad_norm": 5.334595462075598,
      "learning_rate": 2.3634123496838426e-06,
      "loss": 0.1689,
      "step": 4271
    },
    {
      "epoch": 3.0728286279446144,
      "grad_norm": 1.5484097879141911,
      "learning_rate": 2.363134100678447e-06,
      "loss": 0.0023,
      "step": 4272
    },
    {
      "epoch": 3.0735479230354255,
      "grad_norm": 3.7753831885329063,
      "learning_rate": 2.3628558072632864e-06,
      "loss": 0.0657,
      "step": 4273
    },
    {
      "epoch": 3.0742672181262365,
      "grad_norm": 2.839757893740574,
      "learning_rate": 2.3625774694526798e-06,
      "loss": 0.0154,
      "step": 4274
    },
    {
      "epoch": 3.074986513217047,
      "grad_norm": 3.802846975250464,
      "learning_rate": 2.3622990872609484e-06,
      "loss": 0.036,
      "step": 4275
    },
    {
      "epoch": 3.075705808307858,
      "grad_norm": 1.0863605541048587,
      "learning_rate": 2.362020660702415e-06,
      "loss": 0.0125,
      "step": 4276
    },
    {
      "epoch": 3.076425103398669,
      "grad_norm": 1.3829114362419619,
      "learning_rate": 2.3617421897914056e-06,
      "loss": 0.0178,
      "step": 4277
    },
    {
      "epoch": 3.07714439848948,
      "grad_norm": 2.1706024767608665,
      "learning_rate": 2.3614636745422475e-06,
      "loss": 0.0454,
      "step": 4278
    },
    {
      "epoch": 3.077863693580291,
      "grad_norm": 4.155052721344708,
      "learning_rate": 2.3611851149692716e-06,
      "loss": 0.0134,
      "step": 4279
    },
    {
      "epoch": 3.0785829886711022,
      "grad_norm": 4.3679023295653865,
      "learning_rate": 2.3609065110868087e-06,
      "loss": 0.1243,
      "step": 4280
    },
    {
      "epoch": 3.0793022837619133,
      "grad_norm": 3.2743271295106804,
      "learning_rate": 2.360627862909196e-06,
      "loss": 0.1273,
      "step": 4281
    },
    {
      "epoch": 3.0800215788527243,
      "grad_norm": 3.0484755149030303,
      "learning_rate": 2.360349170450768e-06,
      "loss": 0.0696,
      "step": 4282
    },
    {
      "epoch": 3.0807408739435354,
      "grad_norm": 3.7898927680087153,
      "learning_rate": 2.3600704337258647e-06,
      "loss": 0.1295,
      "step": 4283
    },
    {
      "epoch": 3.0814601690343464,
      "grad_norm": 5.614572298420928,
      "learning_rate": 2.359791652748829e-06,
      "loss": 0.0927,
      "step": 4284
    },
    {
      "epoch": 3.0821794641251574,
      "grad_norm": 3.055719870226799,
      "learning_rate": 2.3595128275340023e-06,
      "loss": 0.0787,
      "step": 4285
    },
    {
      "epoch": 3.0828987592159685,
      "grad_norm": 3.0982478577483175,
      "learning_rate": 2.359233958095732e-06,
      "loss": 0.0227,
      "step": 4286
    },
    {
      "epoch": 3.0836180543067795,
      "grad_norm": 1.3530381316982012,
      "learning_rate": 2.358955044448367e-06,
      "loss": 0.0042,
      "step": 4287
    },
    {
      "epoch": 3.0843373493975905,
      "grad_norm": 2.3582436814956123,
      "learning_rate": 2.358676086606256e-06,
      "loss": 0.059,
      "step": 4288
    },
    {
      "epoch": 3.0850566444884016,
      "grad_norm": 4.760299677220257,
      "learning_rate": 2.3583970845837532e-06,
      "loss": 0.1986,
      "step": 4289
    },
    {
      "epoch": 3.0857759395792126,
      "grad_norm": 3.2049452017410696,
      "learning_rate": 2.3581180383952136e-06,
      "loss": 0.0372,
      "step": 4290
    },
    {
      "epoch": 3.086495234670023,
      "grad_norm": 2.417617986286057,
      "learning_rate": 2.357838948054995e-06,
      "loss": 0.0622,
      "step": 4291
    },
    {
      "epoch": 3.087214529760834,
      "grad_norm": 3.997759165906392,
      "learning_rate": 2.3575598135774566e-06,
      "loss": 0.2331,
      "step": 4292
    },
    {
      "epoch": 3.0879338248516452,
      "grad_norm": 2.8098320071384792,
      "learning_rate": 2.35728063497696e-06,
      "loss": 0.0641,
      "step": 4293
    },
    {
      "epoch": 3.0886531199424563,
      "grad_norm": 3.3666155757235465,
      "learning_rate": 2.3570014122678704e-06,
      "loss": 0.0883,
      "step": 4294
    },
    {
      "epoch": 3.0893724150332673,
      "grad_norm": 3.025132373950285,
      "learning_rate": 2.3567221454645527e-06,
      "loss": 0.1054,
      "step": 4295
    },
    {
      "epoch": 3.0900917101240784,
      "grad_norm": 2.5349951182898525,
      "learning_rate": 2.3564428345813777e-06,
      "loss": 0.0416,
      "step": 4296
    },
    {
      "epoch": 3.0908110052148894,
      "grad_norm": 3.175242898859685,
      "learning_rate": 2.3561634796327147e-06,
      "loss": 0.1184,
      "step": 4297
    },
    {
      "epoch": 3.0915303003057004,
      "grad_norm": 3.264749656039814,
      "learning_rate": 2.355884080632938e-06,
      "loss": 0.0854,
      "step": 4298
    },
    {
      "epoch": 3.0922495953965115,
      "grad_norm": 4.211797112789473,
      "learning_rate": 2.355604637596423e-06,
      "loss": 0.0912,
      "step": 4299
    },
    {
      "epoch": 3.0929688904873225,
      "grad_norm": 1.9526795931431653,
      "learning_rate": 2.3553251505375475e-06,
      "loss": 0.0346,
      "step": 4300
    },
    {
      "epoch": 3.0936881855781335,
      "grad_norm": 2.891672190616186,
      "learning_rate": 2.3550456194706914e-06,
      "loss": 0.0583,
      "step": 4301
    },
    {
      "epoch": 3.0944074806689446,
      "grad_norm": 2.1798682428435843,
      "learning_rate": 2.354766044410237e-06,
      "loss": 0.0691,
      "step": 4302
    },
    {
      "epoch": 3.0951267757597556,
      "grad_norm": 2.8837444574227895,
      "learning_rate": 2.354486425370569e-06,
      "loss": 0.0173,
      "step": 4303
    },
    {
      "epoch": 3.0958460708505666,
      "grad_norm": 3.996177179250047,
      "learning_rate": 2.354206762366075e-06,
      "loss": 0.1788,
      "step": 4304
    },
    {
      "epoch": 3.0965653659413777,
      "grad_norm": 0.3590702367727158,
      "learning_rate": 2.3539270554111435e-06,
      "loss": 0.0029,
      "step": 4305
    },
    {
      "epoch": 3.0972846610321882,
      "grad_norm": 2.0487003853320527,
      "learning_rate": 2.3536473045201658e-06,
      "loss": 0.0401,
      "step": 4306
    },
    {
      "epoch": 3.0980039561229993,
      "grad_norm": 1.546319294010759,
      "learning_rate": 2.3533675097075355e-06,
      "loss": 0.0259,
      "step": 4307
    },
    {
      "epoch": 3.0987232512138103,
      "grad_norm": 1.077268305263352,
      "learning_rate": 2.3530876709876492e-06,
      "loss": 0.0073,
      "step": 4308
    },
    {
      "epoch": 3.0994425463046213,
      "grad_norm": 3.038848824000709,
      "learning_rate": 2.3528077883749044e-06,
      "loss": 0.1274,
      "step": 4309
    },
    {
      "epoch": 3.1001618413954324,
      "grad_norm": 1.3542669452799583,
      "learning_rate": 2.352527861883702e-06,
      "loss": 0.031,
      "step": 4310
    },
    {
      "epoch": 3.1008811364862434,
      "grad_norm": 2.630013177638445,
      "learning_rate": 2.3522478915284443e-06,
      "loss": 0.023,
      "step": 4311
    },
    {
      "epoch": 3.1016004315770545,
      "grad_norm": 3.393761631108634,
      "learning_rate": 2.351967877323537e-06,
      "loss": 0.0583,
      "step": 4312
    },
    {
      "epoch": 3.1023197266678655,
      "grad_norm": 0.9049014106597464,
      "learning_rate": 2.351687819283386e-06,
      "loss": 0.0044,
      "step": 4313
    },
    {
      "epoch": 3.1030390217586765,
      "grad_norm": 6.652411489218541,
      "learning_rate": 2.351407717422402e-06,
      "loss": 0.1481,
      "step": 4314
    },
    {
      "epoch": 3.1037583168494876,
      "grad_norm": 0.3874367078146442,
      "learning_rate": 2.351127571754997e-06,
      "loss": 0.0012,
      "step": 4315
    },
    {
      "epoch": 3.1044776119402986,
      "grad_norm": 3.737806608133162,
      "learning_rate": 2.3508473822955833e-06,
      "loss": 0.0637,
      "step": 4316
    },
    {
      "epoch": 3.1051969070311096,
      "grad_norm": 4.028230804251777,
      "learning_rate": 2.350567149058579e-06,
      "loss": 0.0655,
      "step": 4317
    },
    {
      "epoch": 3.1059162021219207,
      "grad_norm": 4.619850208644465,
      "learning_rate": 2.3502868720584012e-06,
      "loss": 0.1221,
      "step": 4318
    },
    {
      "epoch": 3.1066354972127317,
      "grad_norm": 3.1554251194268703,
      "learning_rate": 2.350006551309471e-06,
      "loss": 0.0741,
      "step": 4319
    },
    {
      "epoch": 3.1073547923035427,
      "grad_norm": 2.8972149475938838,
      "learning_rate": 2.3497261868262125e-06,
      "loss": 0.0521,
      "step": 4320
    },
    {
      "epoch": 3.1080740873943533,
      "grad_norm": 4.270402225725747,
      "learning_rate": 2.3494457786230496e-06,
      "loss": 0.198,
      "step": 4321
    },
    {
      "epoch": 3.1087933824851643,
      "grad_norm": 2.3661587186136623,
      "learning_rate": 2.3491653267144097e-06,
      "loss": 0.0751,
      "step": 4322
    },
    {
      "epoch": 3.1095126775759754,
      "grad_norm": 1.6209350706745373,
      "learning_rate": 2.348884831114724e-06,
      "loss": 0.0394,
      "step": 4323
    },
    {
      "epoch": 3.1102319726667864,
      "grad_norm": 3.401284027484061,
      "learning_rate": 2.3486042918384227e-06,
      "loss": 0.1487,
      "step": 4324
    },
    {
      "epoch": 3.1109512677575974,
      "grad_norm": 2.5718903872531147,
      "learning_rate": 2.3483237088999416e-06,
      "loss": 0.0495,
      "step": 4325
    },
    {
      "epoch": 3.1116705628484085,
      "grad_norm": 4.056533512076855,
      "learning_rate": 2.3480430823137164e-06,
      "loss": 0.0835,
      "step": 4326
    },
    {
      "epoch": 3.1123898579392195,
      "grad_norm": 4.296967812275655,
      "learning_rate": 2.347762412094185e-06,
      "loss": 0.0833,
      "step": 4327
    },
    {
      "epoch": 3.1131091530300306,
      "grad_norm": 3.0100353057211184,
      "learning_rate": 2.3474816982557907e-06,
      "loss": 0.0452,
      "step": 4328
    },
    {
      "epoch": 3.1138284481208416,
      "grad_norm": 4.925011069746328,
      "learning_rate": 2.3472009408129744e-06,
      "loss": 0.0975,
      "step": 4329
    },
    {
      "epoch": 3.1145477432116526,
      "grad_norm": 2.8372176835274674,
      "learning_rate": 2.346920139780183e-06,
      "loss": 0.0761,
      "step": 4330
    },
    {
      "epoch": 3.1152670383024637,
      "grad_norm": 5.141305497682005,
      "learning_rate": 2.346639295171863e-06,
      "loss": 0.1165,
      "step": 4331
    },
    {
      "epoch": 3.1159863333932747,
      "grad_norm": 2.6174047495789323,
      "learning_rate": 2.3463584070024654e-06,
      "loss": 0.0171,
      "step": 4332
    },
    {
      "epoch": 3.1167056284840857,
      "grad_norm": 4.752582155482094,
      "learning_rate": 2.3460774752864424e-06,
      "loss": 0.0586,
      "step": 4333
    },
    {
      "epoch": 3.1174249235748968,
      "grad_norm": 3.049226264561142,
      "learning_rate": 2.345796500038248e-06,
      "loss": 0.0458,
      "step": 4334
    },
    {
      "epoch": 3.118144218665708,
      "grad_norm": 0.38172098568531765,
      "learning_rate": 2.3455154812723384e-06,
      "loss": 0.0009,
      "step": 4335
    },
    {
      "epoch": 3.118863513756519,
      "grad_norm": 2.1839789865994983,
      "learning_rate": 2.345234419003173e-06,
      "loss": 0.0073,
      "step": 4336
    },
    {
      "epoch": 3.1195828088473294,
      "grad_norm": 5.145329011124619,
      "learning_rate": 2.3449533132452134e-06,
      "loss": 0.1399,
      "step": 4337
    },
    {
      "epoch": 3.1203021039381404,
      "grad_norm": 2.912096667122734,
      "learning_rate": 2.3446721640129223e-06,
      "loss": 0.0631,
      "step": 4338
    },
    {
      "epoch": 3.1210213990289515,
      "grad_norm": 3.0175501768625073,
      "learning_rate": 2.344390971320766e-06,
      "loss": 0.0729,
      "step": 4339
    },
    {
      "epoch": 3.1217406941197625,
      "grad_norm": 4.234504107732753,
      "learning_rate": 2.344109735183211e-06,
      "loss": 0.0736,
      "step": 4340
    },
    {
      "epoch": 3.1224599892105735,
      "grad_norm": 3.084957838339434,
      "learning_rate": 2.3438284556147294e-06,
      "loss": 0.0761,
      "step": 4341
    },
    {
      "epoch": 3.1231792843013846,
      "grad_norm": 3.2653839732704286,
      "learning_rate": 2.3435471326297923e-06,
      "loss": 0.0202,
      "step": 4342
    },
    {
      "epoch": 3.1238985793921956,
      "grad_norm": 1.4975803740093616,
      "learning_rate": 2.343265766242874e-06,
      "loss": 0.0055,
      "step": 4343
    },
    {
      "epoch": 3.1246178744830067,
      "grad_norm": 5.944314430720115,
      "learning_rate": 2.342984356468452e-06,
      "loss": 0.1112,
      "step": 4344
    },
    {
      "epoch": 3.1253371695738177,
      "grad_norm": 2.115896244822855,
      "learning_rate": 2.342702903321005e-06,
      "loss": 0.0578,
      "step": 4345
    },
    {
      "epoch": 3.1260564646646287,
      "grad_norm": 2.6737930817274354,
      "learning_rate": 2.342421406815014e-06,
      "loss": 0.0674,
      "step": 4346
    },
    {
      "epoch": 3.1267757597554398,
      "grad_norm": 5.484989361988779,
      "learning_rate": 2.342139866964963e-06,
      "loss": 0.1758,
      "step": 4347
    },
    {
      "epoch": 3.127495054846251,
      "grad_norm": 4.471772851325045,
      "learning_rate": 2.3418582837853376e-06,
      "loss": 0.0179,
      "step": 4348
    },
    {
      "epoch": 3.128214349937062,
      "grad_norm": 1.3428885128036792,
      "learning_rate": 2.341576657290626e-06,
      "loss": 0.0156,
      "step": 4349
    },
    {
      "epoch": 3.128933645027873,
      "grad_norm": 2.791134095875226,
      "learning_rate": 2.3412949874953174e-06,
      "loss": 0.0832,
      "step": 4350
    },
    {
      "epoch": 3.129652940118684,
      "grad_norm": 1.542315512294783,
      "learning_rate": 2.341013274413905e-06,
      "loss": 0.004,
      "step": 4351
    },
    {
      "epoch": 3.130372235209495,
      "grad_norm": 4.910016549068694,
      "learning_rate": 2.3407315180608835e-06,
      "loss": 0.1427,
      "step": 4352
    },
    {
      "epoch": 3.1310915303003055,
      "grad_norm": 4.93607016673045,
      "learning_rate": 2.3404497184507492e-06,
      "loss": 0.1069,
      "step": 4353
    },
    {
      "epoch": 3.1318108253911165,
      "grad_norm": 2.9150605732507944,
      "learning_rate": 2.340167875598002e-06,
      "loss": 0.059,
      "step": 4354
    },
    {
      "epoch": 3.1325301204819276,
      "grad_norm": 3.991647708689319,
      "learning_rate": 2.339885989517142e-06,
      "loss": 0.2022,
      "step": 4355
    },
    {
      "epoch": 3.1332494155727386,
      "grad_norm": 5.586382933642049,
      "learning_rate": 2.3396040602226743e-06,
      "loss": 0.1204,
      "step": 4356
    },
    {
      "epoch": 3.1339687106635497,
      "grad_norm": 5.914467360284644,
      "learning_rate": 2.3393220877291035e-06,
      "loss": 0.2169,
      "step": 4357
    },
    {
      "epoch": 3.1346880057543607,
      "grad_norm": 3.0540154205588688,
      "learning_rate": 2.3390400720509377e-06,
      "loss": 0.0445,
      "step": 4358
    },
    {
      "epoch": 3.1354073008451717,
      "grad_norm": 1.3041172786915693,
      "learning_rate": 2.3387580132026882e-06,
      "loss": 0.0183,
      "step": 4359
    },
    {
      "epoch": 3.1361265959359828,
      "grad_norm": 3.2906619112387876,
      "learning_rate": 2.3384759111988657e-06,
      "loss": 0.0948,
      "step": 4360
    },
    {
      "epoch": 3.136845891026794,
      "grad_norm": 2.3115473671685565,
      "learning_rate": 2.338193766053985e-06,
      "loss": 0.0688,
      "step": 4361
    },
    {
      "epoch": 3.137565186117605,
      "grad_norm": 2.6231877080535253,
      "learning_rate": 2.337911577782565e-06,
      "loss": 0.0862,
      "step": 4362
    },
    {
      "epoch": 3.138284481208416,
      "grad_norm": 3.759699379289737,
      "learning_rate": 2.337629346399123e-06,
      "loss": 0.0673,
      "step": 4363
    },
    {
      "epoch": 3.139003776299227,
      "grad_norm": 2.5690403586303616,
      "learning_rate": 2.33734707191818e-06,
      "loss": 0.0613,
      "step": 4364
    },
    {
      "epoch": 3.139723071390038,
      "grad_norm": 2.7753052910416987,
      "learning_rate": 2.3370647543542612e-06,
      "loss": 0.0267,
      "step": 4365
    },
    {
      "epoch": 3.140442366480849,
      "grad_norm": 2.7417495295000798,
      "learning_rate": 2.336782393721891e-06,
      "loss": 0.0842,
      "step": 4366
    },
    {
      "epoch": 3.14116166157166,
      "grad_norm": 3.954977435689569,
      "learning_rate": 2.336499990035597e-06,
      "loss": 0.0303,
      "step": 4367
    },
    {
      "epoch": 3.1418809566624706,
      "grad_norm": 2.6735883113450196,
      "learning_rate": 2.3362175433099115e-06,
      "loss": 0.0181,
      "step": 4368
    },
    {
      "epoch": 3.1426002517532816,
      "grad_norm": 2.783632486656075,
      "learning_rate": 2.3359350535593644e-06,
      "loss": 0.0243,
      "step": 4369
    },
    {
      "epoch": 3.1433195468440926,
      "grad_norm": 4.187529340339971,
      "learning_rate": 2.3356525207984917e-06,
      "loss": 0.1168,
      "step": 4370
    },
    {
      "epoch": 3.1440388419349037,
      "grad_norm": 2.488807835596682,
      "learning_rate": 2.3353699450418297e-06,
      "loss": 0.0391,
      "step": 4371
    },
    {
      "epoch": 3.1447581370257147,
      "grad_norm": 2.275658666436562,
      "learning_rate": 2.335087326303918e-06,
      "loss": 0.0438,
      "step": 4372
    },
    {
      "epoch": 3.1454774321165258,
      "grad_norm": 4.183793661469565,
      "learning_rate": 2.334804664599297e-06,
      "loss": 0.1694,
      "step": 4373
    },
    {
      "epoch": 3.146196727207337,
      "grad_norm": 0.6376790921047801,
      "learning_rate": 2.3345219599425105e-06,
      "loss": 0.0027,
      "step": 4374
    },
    {
      "epoch": 3.146916022298148,
      "grad_norm": 2.3987108888760424,
      "learning_rate": 2.3342392123481047e-06,
      "loss": 0.0507,
      "step": 4375
    },
    {
      "epoch": 3.147635317388959,
      "grad_norm": 0.17687590556014357,
      "learning_rate": 2.3339564218306266e-06,
      "loss": 0.0008,
      "step": 4376
    },
    {
      "epoch": 3.14835461247977,
      "grad_norm": 2.364436737289925,
      "learning_rate": 2.333673588404627e-06,
      "loss": 0.0047,
      "step": 4377
    },
    {
      "epoch": 3.149073907570581,
      "grad_norm": 3.785599503105051,
      "learning_rate": 2.333390712084657e-06,
      "loss": 0.0402,
      "step": 4378
    },
    {
      "epoch": 3.149793202661392,
      "grad_norm": 4.104086656378021,
      "learning_rate": 2.3331077928852723e-06,
      "loss": 0.033,
      "step": 4379
    },
    {
      "epoch": 3.150512497752203,
      "grad_norm": 2.2335032453804646,
      "learning_rate": 2.3328248308210295e-06,
      "loss": 0.0317,
      "step": 4380
    },
    {
      "epoch": 3.151231792843014,
      "grad_norm": 6.85313650262855,
      "learning_rate": 2.3325418259064867e-06,
      "loss": 0.0987,
      "step": 4381
    },
    {
      "epoch": 3.151951087933825,
      "grad_norm": 4.338090647555055,
      "learning_rate": 2.332258778156206e-06,
      "loss": 0.0687,
      "step": 4382
    },
    {
      "epoch": 3.1526703830246356,
      "grad_norm": 4.947901042791069,
      "learning_rate": 2.3319756875847495e-06,
      "loss": 0.223,
      "step": 4383
    },
    {
      "epoch": 3.1533896781154467,
      "grad_norm": 2.094919411961174,
      "learning_rate": 2.331692554206684e-06,
      "loss": 0.0449,
      "step": 4384
    },
    {
      "epoch": 3.1541089732062577,
      "grad_norm": 5.742208622773294,
      "learning_rate": 2.3314093780365763e-06,
      "loss": 0.0373,
      "step": 4385
    },
    {
      "epoch": 3.1548282682970687,
      "grad_norm": 3.617450356904854,
      "learning_rate": 2.3311261590889966e-06,
      "loss": 0.0782,
      "step": 4386
    },
    {
      "epoch": 3.15554756338788,
      "grad_norm": 4.714092859564531,
      "learning_rate": 2.3308428973785173e-06,
      "loss": 0.1354,
      "step": 4387
    },
    {
      "epoch": 3.156266858478691,
      "grad_norm": 2.043927054606574,
      "learning_rate": 2.330559592919712e-06,
      "loss": 0.0324,
      "step": 4388
    },
    {
      "epoch": 3.156986153569502,
      "grad_norm": 3.354761684526233,
      "learning_rate": 2.330276245727158e-06,
      "loss": 0.0628,
      "step": 4389
    },
    {
      "epoch": 3.157705448660313,
      "grad_norm": 2.8181267759493362,
      "learning_rate": 2.3299928558154333e-06,
      "loss": 0.0306,
      "step": 4390
    },
    {
      "epoch": 3.158424743751124,
      "grad_norm": 1.189782749982172,
      "learning_rate": 2.3297094231991195e-06,
      "loss": 0.027,
      "step": 4391
    },
    {
      "epoch": 3.159144038841935,
      "grad_norm": 10.402515550770174,
      "learning_rate": 2.3294259478927993e-06,
      "loss": 0.1305,
      "step": 4392
    },
    {
      "epoch": 3.159863333932746,
      "grad_norm": 5.9075749703713045,
      "learning_rate": 2.329142429911058e-06,
      "loss": 0.0738,
      "step": 4393
    },
    {
      "epoch": 3.160582629023557,
      "grad_norm": 2.470905455932841,
      "learning_rate": 2.3288588692684832e-06,
      "loss": 0.0617,
      "step": 4394
    },
    {
      "epoch": 3.161301924114368,
      "grad_norm": 5.257654640936029,
      "learning_rate": 2.3285752659796647e-06,
      "loss": 0.0687,
      "step": 4395
    },
    {
      "epoch": 3.162021219205179,
      "grad_norm": 1.072471334251246,
      "learning_rate": 2.3282916200591944e-06,
      "loss": 0.0213,
      "step": 4396
    },
    {
      "epoch": 3.16274051429599,
      "grad_norm": 3.0844900738070473,
      "learning_rate": 2.3280079315216656e-06,
      "loss": 0.0098,
      "step": 4397
    },
    {
      "epoch": 3.1634598093868007,
      "grad_norm": 3.3694254660500405,
      "learning_rate": 2.3277242003816755e-06,
      "loss": 0.08,
      "step": 4398
    },
    {
      "epoch": 3.1641791044776117,
      "grad_norm": 1.0663286297831176,
      "learning_rate": 2.327440426653823e-06,
      "loss": 0.0091,
      "step": 4399
    },
    {
      "epoch": 3.164898399568423,
      "grad_norm": 4.808314710642797,
      "learning_rate": 2.3271566103527065e-06,
      "loss": 0.1251,
      "step": 4400
    },
    {
      "epoch": 3.165617694659234,
      "grad_norm": 1.6913797189511743,
      "learning_rate": 2.3268727514929317e-06,
      "loss": 0.0039,
      "step": 4401
    },
    {
      "epoch": 3.166336989750045,
      "grad_norm": 0.5088426229237254,
      "learning_rate": 2.326588850089102e-06,
      "loss": 0.0056,
      "step": 4402
    },
    {
      "epoch": 3.167056284840856,
      "grad_norm": 1.00465384953645,
      "learning_rate": 2.3263049061558243e-06,
      "loss": 0.0174,
      "step": 4403
    },
    {
      "epoch": 3.167775579931667,
      "grad_norm": 3.146771588246694,
      "learning_rate": 2.326020919707709e-06,
      "loss": 0.0736,
      "step": 4404
    },
    {
      "epoch": 3.168494875022478,
      "grad_norm": 6.461581125668157,
      "learning_rate": 2.3257368907593674e-06,
      "loss": 0.1716,
      "step": 4405
    },
    {
      "epoch": 3.169214170113289,
      "grad_norm": 3.9975187908012844,
      "learning_rate": 2.325452819325413e-06,
      "loss": 0.0562,
      "step": 4406
    },
    {
      "epoch": 3.1699334652041,
      "grad_norm": 4.842077619584255,
      "learning_rate": 2.3251687054204626e-06,
      "loss": 0.076,
      "step": 4407
    },
    {
      "epoch": 3.170652760294911,
      "grad_norm": 1.6013897776608923,
      "learning_rate": 2.324884549059133e-06,
      "loss": 0.0528,
      "step": 4408
    },
    {
      "epoch": 3.171372055385722,
      "grad_norm": 7.85392116682029,
      "learning_rate": 2.324600350256046e-06,
      "loss": 0.3062,
      "step": 4409
    },
    {
      "epoch": 3.172091350476533,
      "grad_norm": 1.3042384112023762,
      "learning_rate": 2.3243161090258233e-06,
      "loss": 0.0039,
      "step": 4410
    },
    {
      "epoch": 3.172810645567344,
      "grad_norm": 2.278091582036591,
      "learning_rate": 2.3240318253830895e-06,
      "loss": 0.0567,
      "step": 4411
    },
    {
      "epoch": 3.173529940658155,
      "grad_norm": 3.211394369583977,
      "learning_rate": 2.323747499342472e-06,
      "loss": 0.0613,
      "step": 4412
    },
    {
      "epoch": 3.174249235748966,
      "grad_norm": 2.8824392541265307,
      "learning_rate": 2.3234631309185993e-06,
      "loss": 0.0418,
      "step": 4413
    },
    {
      "epoch": 3.174968530839777,
      "grad_norm": 1.6915780414511505,
      "learning_rate": 2.3231787201261037e-06,
      "loss": 0.0084,
      "step": 4414
    },
    {
      "epoch": 3.175687825930588,
      "grad_norm": 5.258206607465898,
      "learning_rate": 2.322894266979617e-06,
      "loss": 0.1322,
      "step": 4415
    },
    {
      "epoch": 3.176407121021399,
      "grad_norm": 3.223558723258728,
      "learning_rate": 2.322609771493776e-06,
      "loss": 0.09,
      "step": 4416
    },
    {
      "epoch": 3.17712641611221,
      "grad_norm": 2.4456818557466606,
      "learning_rate": 2.322325233683218e-06,
      "loss": 0.028,
      "step": 4417
    },
    {
      "epoch": 3.177845711203021,
      "grad_norm": 0.5984818678664977,
      "learning_rate": 2.322040653562584e-06,
      "loss": 0.0027,
      "step": 4418
    },
    {
      "epoch": 3.178565006293832,
      "grad_norm": 3.2196869190183617,
      "learning_rate": 2.321756031146515e-06,
      "loss": 0.0867,
      "step": 4419
    },
    {
      "epoch": 3.179284301384643,
      "grad_norm": 2.0150142720916913,
      "learning_rate": 2.3214713664496553e-06,
      "loss": 0.0229,
      "step": 4420
    },
    {
      "epoch": 3.180003596475454,
      "grad_norm": 0.23136971926811617,
      "learning_rate": 2.3211866594866517e-06,
      "loss": 0.0005,
      "step": 4421
    },
    {
      "epoch": 3.180722891566265,
      "grad_norm": 1.8908780644179548,
      "learning_rate": 2.3209019102721534e-06,
      "loss": 0.035,
      "step": 4422
    },
    {
      "epoch": 3.181442186657076,
      "grad_norm": 4.551876932182956,
      "learning_rate": 2.3206171188208104e-06,
      "loss": 0.0171,
      "step": 4423
    },
    {
      "epoch": 3.182161481747887,
      "grad_norm": 0.21177469250078276,
      "learning_rate": 2.320332285147276e-06,
      "loss": 0.0008,
      "step": 4424
    },
    {
      "epoch": 3.182880776838698,
      "grad_norm": 2.736776600710655,
      "learning_rate": 2.3200474092662057e-06,
      "loss": 0.0612,
      "step": 4425
    },
    {
      "epoch": 3.1836000719295092,
      "grad_norm": 5.346517619869282,
      "learning_rate": 2.3197624911922568e-06,
      "loss": 0.0711,
      "step": 4426
    },
    {
      "epoch": 3.1843193670203203,
      "grad_norm": 2.523345662587272,
      "learning_rate": 2.3194775309400884e-06,
      "loss": 0.0308,
      "step": 4427
    },
    {
      "epoch": 3.1850386621111313,
      "grad_norm": 0.31850409570219446,
      "learning_rate": 2.319192528524363e-06,
      "loss": 0.0018,
      "step": 4428
    },
    {
      "epoch": 3.1857579572019423,
      "grad_norm": 6.595269192651395,
      "learning_rate": 2.318907483959743e-06,
      "loss": 0.0984,
      "step": 4429
    },
    {
      "epoch": 3.186477252292753,
      "grad_norm": 4.045564026113133,
      "learning_rate": 2.3186223972608963e-06,
      "loss": 0.1256,
      "step": 4430
    },
    {
      "epoch": 3.187196547383564,
      "grad_norm": 1.807476724480253,
      "learning_rate": 2.3183372684424896e-06,
      "loss": 0.0333,
      "step": 4431
    },
    {
      "epoch": 3.187915842474375,
      "grad_norm": 2.752883557467025,
      "learning_rate": 2.3180520975191944e-06,
      "loss": 0.0585,
      "step": 4432
    },
    {
      "epoch": 3.188635137565186,
      "grad_norm": 3.2346090928380984,
      "learning_rate": 2.3177668845056827e-06,
      "loss": 0.0783,
      "step": 4433
    },
    {
      "epoch": 3.189354432655997,
      "grad_norm": 3.0956889653819144,
      "learning_rate": 2.317481629416629e-06,
      "loss": 0.0867,
      "step": 4434
    },
    {
      "epoch": 3.190073727746808,
      "grad_norm": 1.4527593343471032,
      "learning_rate": 2.31719633226671e-06,
      "loss": 0.0391,
      "step": 4435
    },
    {
      "epoch": 3.190793022837619,
      "grad_norm": 0.06607369291953118,
      "learning_rate": 2.316910993070606e-06,
      "loss": 0.0003,
      "step": 4436
    },
    {
      "epoch": 3.19151231792843,
      "grad_norm": 0.2958334965677924,
      "learning_rate": 2.3166256118429966e-06,
      "loss": 0.0023,
      "step": 4437
    },
    {
      "epoch": 3.192231613019241,
      "grad_norm": 2.8391639476371613,
      "learning_rate": 2.3163401885985663e-06,
      "loss": 0.0784,
      "step": 4438
    },
    {
      "epoch": 3.192950908110052,
      "grad_norm": 3.8482289008831425,
      "learning_rate": 2.3160547233520006e-06,
      "loss": 0.0589,
      "step": 4439
    },
    {
      "epoch": 3.1936702032008633,
      "grad_norm": 2.9866771657659994,
      "learning_rate": 2.315769216117986e-06,
      "loss": 0.094,
      "step": 4440
    },
    {
      "epoch": 3.1943894982916743,
      "grad_norm": 1.9922775740843797,
      "learning_rate": 2.315483666911214e-06,
      "loss": 0.0321,
      "step": 4441
    },
    {
      "epoch": 3.1951087933824853,
      "grad_norm": 3.2256003173956,
      "learning_rate": 2.3151980757463756e-06,
      "loss": 0.1114,
      "step": 4442
    },
    {
      "epoch": 3.1958280884732964,
      "grad_norm": 1.8775875489575369,
      "learning_rate": 2.314912442638165e-06,
      "loss": 0.0552,
      "step": 4443
    },
    {
      "epoch": 3.1965473835641074,
      "grad_norm": 4.578461062774031,
      "learning_rate": 2.3146267676012787e-06,
      "loss": 0.0781,
      "step": 4444
    },
    {
      "epoch": 3.197266678654918,
      "grad_norm": 3.5597560614703454,
      "learning_rate": 2.314341050650415e-06,
      "loss": 0.1009,
      "step": 4445
    },
    {
      "epoch": 3.197985973745729,
      "grad_norm": 3.475688725970508,
      "learning_rate": 2.314055291800275e-06,
      "loss": 0.0742,
      "step": 4446
    },
    {
      "epoch": 3.19870526883654,
      "grad_norm": 0.4769903742144726,
      "learning_rate": 2.3137694910655614e-06,
      "loss": 0.0014,
      "step": 4447
    },
    {
      "epoch": 3.199424563927351,
      "grad_norm": 3.0777961065453936,
      "learning_rate": 2.3134836484609786e-06,
      "loss": 0.0576,
      "step": 4448
    },
    {
      "epoch": 3.200143859018162,
      "grad_norm": 3.5262820400683257,
      "learning_rate": 2.313197764001234e-06,
      "loss": 0.0686,
      "step": 4449
    },
    {
      "epoch": 3.200863154108973,
      "grad_norm": 6.804225922747237,
      "learning_rate": 2.312911837701037e-06,
      "loss": 0.1149,
      "step": 4450
    },
    {
      "epoch": 3.201582449199784,
      "grad_norm": 2.3001825882476306,
      "learning_rate": 2.312625869575099e-06,
      "loss": 0.0091,
      "step": 4451
    },
    {
      "epoch": 3.202301744290595,
      "grad_norm": 0.7355766387388128,
      "learning_rate": 2.312339859638134e-06,
      "loss": 0.0028,
      "step": 4452
    },
    {
      "epoch": 3.2030210393814063,
      "grad_norm": 2.914060450606174,
      "learning_rate": 2.3120538079048566e-06,
      "loss": 0.0847,
      "step": 4453
    },
    {
      "epoch": 3.2037403344722173,
      "grad_norm": 3.1174857545574137,
      "learning_rate": 2.3117677143899854e-06,
      "loss": 0.1301,
      "step": 4454
    },
    {
      "epoch": 3.2044596295630283,
      "grad_norm": 2.068425161934889,
      "learning_rate": 2.31148157910824e-06,
      "loss": 0.0559,
      "step": 4455
    },
    {
      "epoch": 3.2051789246538394,
      "grad_norm": 3.6944000626291995,
      "learning_rate": 2.3111954020743433e-06,
      "loss": 0.015,
      "step": 4456
    },
    {
      "epoch": 3.2058982197446504,
      "grad_norm": 4.724724367187219,
      "learning_rate": 2.3109091833030194e-06,
      "loss": 0.1656,
      "step": 4457
    },
    {
      "epoch": 3.2066175148354614,
      "grad_norm": 1.4975532713284245,
      "learning_rate": 2.310622922808994e-06,
      "loss": 0.0175,
      "step": 4458
    },
    {
      "epoch": 3.2073368099262725,
      "grad_norm": 2.565549422071911,
      "learning_rate": 2.310336620606996e-06,
      "loss": 0.0842,
      "step": 4459
    },
    {
      "epoch": 3.208056105017083,
      "grad_norm": 3.538279043331068,
      "learning_rate": 2.3100502767117564e-06,
      "loss": 0.0488,
      "step": 4460
    },
    {
      "epoch": 3.208775400107894,
      "grad_norm": 3.7405877425216785,
      "learning_rate": 2.3097638911380082e-06,
      "loss": 0.087,
      "step": 4461
    },
    {
      "epoch": 3.209494695198705,
      "grad_norm": 4.413247130370564,
      "learning_rate": 2.309477463900487e-06,
      "loss": 0.094,
      "step": 4462
    },
    {
      "epoch": 3.210213990289516,
      "grad_norm": 3.137204079538125,
      "learning_rate": 2.3091909950139285e-06,
      "loss": 0.0584,
      "step": 4463
    },
    {
      "epoch": 3.210933285380327,
      "grad_norm": 0.33025369181503383,
      "learning_rate": 2.308904484493073e-06,
      "loss": 0.0011,
      "step": 4464
    },
    {
      "epoch": 3.211652580471138,
      "grad_norm": 4.331290380990014,
      "learning_rate": 2.3086179323526622e-06,
      "loss": 0.1194,
      "step": 4465
    },
    {
      "epoch": 3.2123718755619493,
      "grad_norm": 4.480447079152821,
      "learning_rate": 2.3083313386074386e-06,
      "loss": 0.133,
      "step": 4466
    },
    {
      "epoch": 3.2130911706527603,
      "grad_norm": 2.075713118990339,
      "learning_rate": 2.308044703272149e-06,
      "loss": 0.049,
      "step": 4467
    },
    {
      "epoch": 3.2138104657435713,
      "grad_norm": 1.3081725493856065,
      "learning_rate": 2.307758026361541e-06,
      "loss": 0.0268,
      "step": 4468
    },
    {
      "epoch": 3.2145297608343824,
      "grad_norm": 2.798791305170333,
      "learning_rate": 2.3074713078903647e-06,
      "loss": 0.0724,
      "step": 4469
    },
    {
      "epoch": 3.2152490559251934,
      "grad_norm": 5.410810379485853,
      "learning_rate": 2.3071845478733724e-06,
      "loss": 0.0911,
      "step": 4470
    },
    {
      "epoch": 3.2159683510160044,
      "grad_norm": 2.0910150816236084,
      "learning_rate": 2.306897746325317e-06,
      "loss": 0.0423,
      "step": 4471
    },
    {
      "epoch": 3.2166876461068155,
      "grad_norm": 0.6258747737965367,
      "learning_rate": 2.306610903260957e-06,
      "loss": 0.0014,
      "step": 4472
    },
    {
      "epoch": 3.2174069411976265,
      "grad_norm": 0.7972727761803204,
      "learning_rate": 2.3063240186950498e-06,
      "loss": 0.0023,
      "step": 4473
    },
    {
      "epoch": 3.2181262362884375,
      "grad_norm": 3.3915993658670294,
      "learning_rate": 2.3060370926423562e-06,
      "loss": 0.0958,
      "step": 4474
    },
    {
      "epoch": 3.218845531379248,
      "grad_norm": 3.94959226914782,
      "learning_rate": 2.3057501251176396e-06,
      "loss": 0.0748,
      "step": 4475
    },
    {
      "epoch": 3.219564826470059,
      "grad_norm": 3.883366501246911,
      "learning_rate": 2.3054631161356644e-06,
      "loss": 0.0979,
      "step": 4476
    },
    {
      "epoch": 3.22028412156087,
      "grad_norm": 3.1993830122167055,
      "learning_rate": 2.305176065711198e-06,
      "loss": 0.0478,
      "step": 4477
    },
    {
      "epoch": 3.221003416651681,
      "grad_norm": 2.1813816984212995,
      "learning_rate": 2.30488897385901e-06,
      "loss": 0.0077,
      "step": 4478
    },
    {
      "epoch": 3.2217227117424923,
      "grad_norm": 0.3713677942767126,
      "learning_rate": 2.3046018405938703e-06,
      "loss": 0.0015,
      "step": 4479
    },
    {
      "epoch": 3.2224420068333033,
      "grad_norm": 3.771772288692133,
      "learning_rate": 2.304314665930554e-06,
      "loss": 0.0433,
      "step": 4480
    },
    {
      "epoch": 3.2231613019241143,
      "grad_norm": 5.365918324831739,
      "learning_rate": 2.3040274498838368e-06,
      "loss": 0.1204,
      "step": 4481
    },
    {
      "epoch": 3.2238805970149254,
      "grad_norm": 0.03214763778544739,
      "learning_rate": 2.303740192468495e-06,
      "loss": 0.0002,
      "step": 4482
    },
    {
      "epoch": 3.2245998921057364,
      "grad_norm": 4.317335105175406,
      "learning_rate": 2.3034528936993097e-06,
      "loss": 0.0212,
      "step": 4483
    },
    {
      "epoch": 3.2253191871965474,
      "grad_norm": 4.584782371559871,
      "learning_rate": 2.303165553591063e-06,
      "loss": 0.0604,
      "step": 4484
    },
    {
      "epoch": 3.2260384822873585,
      "grad_norm": 4.475533312134028,
      "learning_rate": 2.302878172158538e-06,
      "loss": 0.1119,
      "step": 4485
    },
    {
      "epoch": 3.2267577773781695,
      "grad_norm": 2.838723315491718,
      "learning_rate": 2.302590749416522e-06,
      "loss": 0.0079,
      "step": 4486
    },
    {
      "epoch": 3.2274770724689805,
      "grad_norm": 3.0813757998138915,
      "learning_rate": 2.3023032853798027e-06,
      "loss": 0.0764,
      "step": 4487
    },
    {
      "epoch": 3.2281963675597916,
      "grad_norm": 3.1761811516910146,
      "learning_rate": 2.3020157800631715e-06,
      "loss": 0.0858,
      "step": 4488
    },
    {
      "epoch": 3.2289156626506026,
      "grad_norm": 4.015051287524155,
      "learning_rate": 2.30172823348142e-06,
      "loss": 0.0715,
      "step": 4489
    },
    {
      "epoch": 3.229634957741413,
      "grad_norm": 1.6266441911200107,
      "learning_rate": 2.301440645649344e-06,
      "loss": 0.0056,
      "step": 4490
    },
    {
      "epoch": 3.230354252832224,
      "grad_norm": 5.880698040183779,
      "learning_rate": 2.3011530165817396e-06,
      "loss": 0.1065,
      "step": 4491
    },
    {
      "epoch": 3.2310735479230352,
      "grad_norm": 3.950368788874015,
      "learning_rate": 2.300865346293406e-06,
      "loss": 0.0552,
      "step": 4492
    },
    {
      "epoch": 3.2317928430138463,
      "grad_norm": 2.0590123928379023,
      "learning_rate": 2.3005776347991447e-06,
      "loss": 0.043,
      "step": 4493
    },
    {
      "epoch": 3.2325121381046573,
      "grad_norm": 1.58608227069417,
      "learning_rate": 2.300289882113759e-06,
      "loss": 0.0394,
      "step": 4494
    },
    {
      "epoch": 3.2332314331954684,
      "grad_norm": 7.333014031893844,
      "learning_rate": 2.3000020882520532e-06,
      "loss": 0.13,
      "step": 4495
    },
    {
      "epoch": 3.2339507282862794,
      "grad_norm": 5.6578992083860635,
      "learning_rate": 2.2997142532288362e-06,
      "loss": 0.159,
      "step": 4496
    },
    {
      "epoch": 3.2346700233770904,
      "grad_norm": 3.0283983287122624,
      "learning_rate": 2.2994263770589165e-06,
      "loss": 0.1066,
      "step": 4497
    },
    {
      "epoch": 3.2353893184679015,
      "grad_norm": 1.9122741314876381,
      "learning_rate": 2.299138459757107e-06,
      "loss": 0.0181,
      "step": 4498
    },
    {
      "epoch": 3.2361086135587125,
      "grad_norm": 2.1077813933946623,
      "learning_rate": 2.298850501338221e-06,
      "loss": 0.0619,
      "step": 4499
    },
    {
      "epoch": 3.2368279086495235,
      "grad_norm": 2.8305511425373227,
      "learning_rate": 2.298562501817074e-06,
      "loss": 0.0378,
      "step": 4500
    },
    {
      "epoch": 3.2375472037403346,
      "grad_norm": 1.6833399133911264,
      "learning_rate": 2.2982744612084845e-06,
      "loss": 0.0031,
      "step": 4501
    },
    {
      "epoch": 3.2382664988311456,
      "grad_norm": 1.5574718042363924,
      "learning_rate": 2.297986379527272e-06,
      "loss": 0.031,
      "step": 4502
    },
    {
      "epoch": 3.2389857939219566,
      "grad_norm": 5.587475936995155,
      "learning_rate": 2.2976982567882604e-06,
      "loss": 0.0804,
      "step": 4503
    },
    {
      "epoch": 3.2397050890127677,
      "grad_norm": 3.9502177006772508,
      "learning_rate": 2.297410093006273e-06,
      "loss": 0.0142,
      "step": 4504
    },
    {
      "epoch": 3.2404243841035787,
      "grad_norm": 3.10472429272831,
      "learning_rate": 2.297121888196136e-06,
      "loss": 0.0842,
      "step": 4505
    },
    {
      "epoch": 3.2411436791943897,
      "grad_norm": 4.399228698845738,
      "learning_rate": 2.2968336423726793e-06,
      "loss": 0.1163,
      "step": 4506
    },
    {
      "epoch": 3.2418629742852003,
      "grad_norm": 3.613437795295939,
      "learning_rate": 2.296545355550732e-06,
      "loss": 0.0516,
      "step": 4507
    },
    {
      "epoch": 3.2425822693760114,
      "grad_norm": 1.3604763218491365,
      "learning_rate": 2.2962570277451285e-06,
      "loss": 0.0032,
      "step": 4508
    },
    {
      "epoch": 3.2433015644668224,
      "grad_norm": 5.82123442505743,
      "learning_rate": 2.2959686589707033e-06,
      "loss": 0.2941,
      "step": 4509
    },
    {
      "epoch": 3.2440208595576334,
      "grad_norm": 1.240475221497884,
      "learning_rate": 2.2956802492422922e-06,
      "loss": 0.0056,
      "step": 4510
    },
    {
      "epoch": 3.2447401546484445,
      "grad_norm": 2.482300430094419,
      "learning_rate": 2.2953917985747366e-06,
      "loss": 0.0809,
      "step": 4511
    },
    {
      "epoch": 3.2454594497392555,
      "grad_norm": 1.1974042821033857,
      "learning_rate": 2.2951033069828765e-06,
      "loss": 0.0043,
      "step": 4512
    },
    {
      "epoch": 3.2461787448300665,
      "grad_norm": 4.178142093388856,
      "learning_rate": 2.2948147744815552e-06,
      "loss": 0.0921,
      "step": 4513
    },
    {
      "epoch": 3.2468980399208776,
      "grad_norm": 1.6573401438851636,
      "learning_rate": 2.294526201085619e-06,
      "loss": 0.0244,
      "step": 4514
    },
    {
      "epoch": 3.2476173350116886,
      "grad_norm": 1.6868882414163784,
      "learning_rate": 2.294237586809915e-06,
      "loss": 0.0226,
      "step": 4515
    },
    {
      "epoch": 3.2483366301024996,
      "grad_norm": 2.5941880716242203,
      "learning_rate": 2.2939489316692923e-06,
      "loss": 0.0872,
      "step": 4516
    },
    {
      "epoch": 3.2490559251933107,
      "grad_norm": 2.3213252904537636,
      "learning_rate": 2.2936602356786034e-06,
      "loss": 0.0516,
      "step": 4517
    },
    {
      "epoch": 3.2497752202841217,
      "grad_norm": 1.7414896510303517,
      "learning_rate": 2.2933714988527022e-06,
      "loss": 0.0349,
      "step": 4518
    },
    {
      "epoch": 3.2504945153749327,
      "grad_norm": 0.5481272617545307,
      "learning_rate": 2.293082721206445e-06,
      "loss": 0.0012,
      "step": 4519
    },
    {
      "epoch": 3.2512138104657438,
      "grad_norm": 2.2856843161152884,
      "learning_rate": 2.2927939027546894e-06,
      "loss": 0.063,
      "step": 4520
    },
    {
      "epoch": 3.251933105556555,
      "grad_norm": 3.8500958724591814,
      "learning_rate": 2.292505043512296e-06,
      "loss": 0.0929,
      "step": 4521
    },
    {
      "epoch": 3.2526524006473654,
      "grad_norm": 4.01258132567913,
      "learning_rate": 2.292216143494127e-06,
      "loss": 0.0796,
      "step": 4522
    },
    {
      "epoch": 3.2533716957381764,
      "grad_norm": 1.9024944093843816,
      "learning_rate": 2.291927202715046e-06,
      "loss": 0.038,
      "step": 4523
    },
    {
      "epoch": 3.2540909908289875,
      "grad_norm": 4.1204685252936795,
      "learning_rate": 2.2916382211899207e-06,
      "loss": 0.172,
      "step": 4524
    },
    {
      "epoch": 3.2548102859197985,
      "grad_norm": 3.9832920083420627,
      "learning_rate": 2.291349198933619e-06,
      "loss": 0.0191,
      "step": 4525
    },
    {
      "epoch": 3.2555295810106095,
      "grad_norm": 4.723811135501269,
      "learning_rate": 2.2910601359610125e-06,
      "loss": 0.1843,
      "step": 4526
    },
    {
      "epoch": 3.2562488761014206,
      "grad_norm": 3.1517702708389836,
      "learning_rate": 2.2907710322869728e-06,
      "loss": 0.0552,
      "step": 4527
    },
    {
      "epoch": 3.2569681711922316,
      "grad_norm": 3.010441660491754,
      "learning_rate": 2.290481887926376e-06,
      "loss": 0.049,
      "step": 4528
    },
    {
      "epoch": 3.2576874662830426,
      "grad_norm": 4.882684002253346,
      "learning_rate": 2.2901927028940976e-06,
      "loss": 0.0976,
      "step": 4529
    },
    {
      "epoch": 3.2584067613738537,
      "grad_norm": 3.0209006867492363,
      "learning_rate": 2.2899034772050177e-06,
      "loss": 0.031,
      "step": 4530
    },
    {
      "epoch": 3.2591260564646647,
      "grad_norm": 4.482030342568125,
      "learning_rate": 2.289614210874017e-06,
      "loss": 0.0085,
      "step": 4531
    },
    {
      "epoch": 3.2598453515554757,
      "grad_norm": 4.693993702004803,
      "learning_rate": 2.28932490391598e-06,
      "loss": 0.1204,
      "step": 4532
    },
    {
      "epoch": 3.2605646466462868,
      "grad_norm": 4.065221188719599,
      "learning_rate": 2.28903555634579e-06,
      "loss": 0.0289,
      "step": 4533
    },
    {
      "epoch": 3.261283941737098,
      "grad_norm": 2.6341416286948567,
      "learning_rate": 2.288746168178336e-06,
      "loss": 0.0955,
      "step": 4534
    },
    {
      "epoch": 3.262003236827909,
      "grad_norm": 3.239883654137051,
      "learning_rate": 2.2884567394285073e-06,
      "loss": 0.113,
      "step": 4535
    },
    {
      "epoch": 3.26272253191872,
      "grad_norm": 4.264913439965863,
      "learning_rate": 2.288167270111195e-06,
      "loss": 0.0286,
      "step": 4536
    },
    {
      "epoch": 3.2634418270095304,
      "grad_norm": 4.938021074297976,
      "learning_rate": 2.287877760241293e-06,
      "loss": 0.1536,
      "step": 4537
    },
    {
      "epoch": 3.2641611221003415,
      "grad_norm": 10.212806225955248,
      "learning_rate": 2.2875882098336975e-06,
      "loss": 0.1728,
      "step": 4538
    },
    {
      "epoch": 3.2648804171911525,
      "grad_norm": 3.230838492279424,
      "learning_rate": 2.287298618903305e-06,
      "loss": 0.0409,
      "step": 4539
    },
    {
      "epoch": 3.2655997122819636,
      "grad_norm": 3.0017179474407647,
      "learning_rate": 2.2870089874650177e-06,
      "loss": 0.0897,
      "step": 4540
    },
    {
      "epoch": 3.2663190073727746,
      "grad_norm": 1.9895188313574754,
      "learning_rate": 2.2867193155337357e-06,
      "loss": 0.0096,
      "step": 4541
    },
    {
      "epoch": 3.2670383024635856,
      "grad_norm": 0.03845037147874532,
      "learning_rate": 2.2864296031243645e-06,
      "loss": 0.0001,
      "step": 4542
    },
    {
      "epoch": 3.2677575975543967,
      "grad_norm": 0.8479546687788646,
      "learning_rate": 2.2861398502518093e-06,
      "loss": 0.01,
      "step": 4543
    },
    {
      "epoch": 3.2684768926452077,
      "grad_norm": 6.196471453208916,
      "learning_rate": 2.2858500569309787e-06,
      "loss": 0.1443,
      "step": 4544
    },
    {
      "epoch": 3.2691961877360187,
      "grad_norm": 0.6438104350712269,
      "learning_rate": 2.2855602231767834e-06,
      "loss": 0.0058,
      "step": 4545
    },
    {
      "epoch": 3.2699154828268298,
      "grad_norm": 0.12485292211004564,
      "learning_rate": 2.2852703490041355e-06,
      "loss": 0.0002,
      "step": 4546
    },
    {
      "epoch": 3.270634777917641,
      "grad_norm": 7.715480287665266,
      "learning_rate": 2.2849804344279495e-06,
      "loss": 0.3101,
      "step": 4547
    },
    {
      "epoch": 3.271354073008452,
      "grad_norm": 2.899992539018861,
      "learning_rate": 2.284690479463142e-06,
      "loss": 0.0613,
      "step": 4548
    },
    {
      "epoch": 3.272073368099263,
      "grad_norm": 1.10308044019799,
      "learning_rate": 2.2844004841246326e-06,
      "loss": 0.0028,
      "step": 4549
    },
    {
      "epoch": 3.272792663190074,
      "grad_norm": 4.677775241500674,
      "learning_rate": 2.284110448427341e-06,
      "loss": 0.1668,
      "step": 4550
    },
    {
      "epoch": 3.273511958280885,
      "grad_norm": 2.1997057213962234,
      "learning_rate": 2.2838203723861905e-06,
      "loss": 0.0633,
      "step": 4551
    },
    {
      "epoch": 3.2742312533716955,
      "grad_norm": 2.551868283747727,
      "learning_rate": 2.2835302560161054e-06,
      "loss": 0.11,
      "step": 4552
    },
    {
      "epoch": 3.2749505484625065,
      "grad_norm": 1.3950783421460902,
      "learning_rate": 2.283240099332013e-06,
      "loss": 0.0227,
      "step": 4553
    },
    {
      "epoch": 3.2756698435533176,
      "grad_norm": 1.960847236190856,
      "learning_rate": 2.282949902348843e-06,
      "loss": 0.0313,
      "step": 4554
    },
    {
      "epoch": 3.2763891386441286,
      "grad_norm": 1.6767291378098843,
      "learning_rate": 2.282659665081526e-06,
      "loss": 0.0573,
      "step": 4555
    },
    {
      "epoch": 3.2771084337349397,
      "grad_norm": 1.7117983309855933,
      "learning_rate": 2.2823693875449954e-06,
      "loss": 0.0399,
      "step": 4556
    },
    {
      "epoch": 3.2778277288257507,
      "grad_norm": 0.049293936119734004,
      "learning_rate": 2.2820790697541866e-06,
      "loss": 0.0002,
      "step": 4557
    },
    {
      "epoch": 3.2785470239165617,
      "grad_norm": 3.528735185834412,
      "learning_rate": 2.2817887117240367e-06,
      "loss": 0.0749,
      "step": 4558
    },
    {
      "epoch": 3.2792663190073728,
      "grad_norm": 2.3572186886854807,
      "learning_rate": 2.281498313469485e-06,
      "loss": 0.0973,
      "step": 4559
    },
    {
      "epoch": 3.279985614098184,
      "grad_norm": 3.158029135814789,
      "learning_rate": 2.281207875005473e-06,
      "loss": 0.0851,
      "step": 4560
    },
    {
      "epoch": 3.280704909188995,
      "grad_norm": 0.7366220091713963,
      "learning_rate": 2.280917396346945e-06,
      "loss": 0.0016,
      "step": 4561
    },
    {
      "epoch": 3.281424204279806,
      "grad_norm": 7.647067954876685,
      "learning_rate": 2.280626877508846e-06,
      "loss": 0.1104,
      "step": 4562
    },
    {
      "epoch": 3.282143499370617,
      "grad_norm": 1.4819066623474257,
      "learning_rate": 2.280336318506123e-06,
      "loss": 0.03,
      "step": 4563
    },
    {
      "epoch": 3.282862794461428,
      "grad_norm": 6.302592744530469,
      "learning_rate": 2.2800457193537274e-06,
      "loss": 0.1614,
      "step": 4564
    },
    {
      "epoch": 3.283582089552239,
      "grad_norm": 1.9464195660864163,
      "learning_rate": 2.2797550800666097e-06,
      "loss": 0.028,
      "step": 4565
    },
    {
      "epoch": 3.28430138464305,
      "grad_norm": 2.3357279928713117,
      "learning_rate": 2.279464400659725e-06,
      "loss": 0.0788,
      "step": 4566
    },
    {
      "epoch": 3.2850206797338606,
      "grad_norm": 1.4252714385344964,
      "learning_rate": 2.279173681148028e-06,
      "loss": 0.0279,
      "step": 4567
    },
    {
      "epoch": 3.285739974824672,
      "grad_norm": 9.437745735072891,
      "learning_rate": 2.2788829215464774e-06,
      "loss": 0.0497,
      "step": 4568
    },
    {
      "epoch": 3.2864592699154827,
      "grad_norm": 2.070819588304671,
      "learning_rate": 2.2785921218700335e-06,
      "loss": 0.0281,
      "step": 4569
    },
    {
      "epoch": 3.2871785650062937,
      "grad_norm": 3.296193370164543,
      "learning_rate": 2.278301282133658e-06,
      "loss": 0.1054,
      "step": 4570
    },
    {
      "epoch": 3.2878978600971047,
      "grad_norm": 2.2553582316870675,
      "learning_rate": 2.278010402352315e-06,
      "loss": 0.0572,
      "step": 4571
    },
    {
      "epoch": 3.2886171551879158,
      "grad_norm": 1.6190806394079351,
      "learning_rate": 2.2777194825409715e-06,
      "loss": 0.0277,
      "step": 4572
    },
    {
      "epoch": 3.289336450278727,
      "grad_norm": 4.375624257877794,
      "learning_rate": 2.277428522714595e-06,
      "loss": 0.1327,
      "step": 4573
    },
    {
      "epoch": 3.290055745369538,
      "grad_norm": 4.60772729249138,
      "learning_rate": 2.2771375228881565e-06,
      "loss": 0.0091,
      "step": 4574
    },
    {
      "epoch": 3.290775040460349,
      "grad_norm": 3.808832312377162,
      "learning_rate": 2.2768464830766282e-06,
      "loss": 0.0983,
      "step": 4575
    },
    {
      "epoch": 3.29149433555116,
      "grad_norm": 3.9519047824282008,
      "learning_rate": 2.2765554032949844e-06,
      "loss": 0.0751,
      "step": 4576
    },
    {
      "epoch": 3.292213630641971,
      "grad_norm": 5.251271528843381,
      "learning_rate": 2.276264283558202e-06,
      "loss": 0.1656,
      "step": 4577
    },
    {
      "epoch": 3.292932925732782,
      "grad_norm": 1.6159554904377285,
      "learning_rate": 2.27597312388126e-06,
      "loss": 0.0363,
      "step": 4578
    },
    {
      "epoch": 3.293652220823593,
      "grad_norm": 5.314410382343385,
      "learning_rate": 2.275681924279138e-06,
      "loss": 0.1606,
      "step": 4579
    },
    {
      "epoch": 3.294371515914404,
      "grad_norm": 3.712655228468354,
      "learning_rate": 2.2753906847668197e-06,
      "loss": 0.0505,
      "step": 4580
    },
    {
      "epoch": 3.295090811005215,
      "grad_norm": 1.4819984412072205,
      "learning_rate": 2.2750994053592892e-06,
      "loss": 0.0262,
      "step": 4581
    },
    {
      "epoch": 3.2958101060960256,
      "grad_norm": 1.5768488565343495,
      "learning_rate": 2.2748080860715338e-06,
      "loss": 0.02,
      "step": 4582
    },
    {
      "epoch": 3.296529401186837,
      "grad_norm": 4.143716206358008,
      "learning_rate": 2.2745167269185423e-06,
      "loss": 0.0174,
      "step": 4583
    },
    {
      "epoch": 3.2972486962776477,
      "grad_norm": 3.8417429639463854,
      "learning_rate": 2.2742253279153053e-06,
      "loss": 0.1345,
      "step": 4584
    },
    {
      "epoch": 3.2979679913684588,
      "grad_norm": 3.9513148302901424,
      "learning_rate": 2.273933889076816e-06,
      "loss": 0.1042,
      "step": 4585
    },
    {
      "epoch": 3.29868728645927,
      "grad_norm": 0.3357264799130588,
      "learning_rate": 2.27364241041807e-06,
      "loss": 0.0013,
      "step": 4586
    },
    {
      "epoch": 3.299406581550081,
      "grad_norm": 2.5913857987668356,
      "learning_rate": 2.273350891954064e-06,
      "loss": 0.0202,
      "step": 4587
    },
    {
      "epoch": 3.300125876640892,
      "grad_norm": 3.5526145432577168,
      "learning_rate": 2.273059333699797e-06,
      "loss": 0.0914,
      "step": 4588
    },
    {
      "epoch": 3.300845171731703,
      "grad_norm": 4.027765050839108,
      "learning_rate": 2.2727677356702695e-06,
      "loss": 0.1174,
      "step": 4589
    },
    {
      "epoch": 3.301564466822514,
      "grad_norm": 2.6971976355039846,
      "learning_rate": 2.272476097880486e-06,
      "loss": 0.0114,
      "step": 4590
    },
    {
      "epoch": 3.302283761913325,
      "grad_norm": 3.7877920073943425,
      "learning_rate": 2.2721844203454512e-06,
      "loss": 0.1523,
      "step": 4591
    },
    {
      "epoch": 3.303003057004136,
      "grad_norm": 1.4624130923754577,
      "learning_rate": 2.2718927030801718e-06,
      "loss": 0.0375,
      "step": 4592
    },
    {
      "epoch": 3.303722352094947,
      "grad_norm": 2.601114071889803,
      "learning_rate": 2.2716009460996587e-06,
      "loss": 0.0113,
      "step": 4593
    },
    {
      "epoch": 3.304441647185758,
      "grad_norm": 1.658452112315142,
      "learning_rate": 2.2713091494189224e-06,
      "loss": 0.0442,
      "step": 4594
    },
    {
      "epoch": 3.305160942276569,
      "grad_norm": 4.687579377825402,
      "learning_rate": 2.271017313052976e-06,
      "loss": 0.1974,
      "step": 4595
    },
    {
      "epoch": 3.30588023736738,
      "grad_norm": 8.066413590243029,
      "learning_rate": 2.270725437016835e-06,
      "loss": 0.2049,
      "step": 4596
    },
    {
      "epoch": 3.306599532458191,
      "grad_norm": 1.0743605143054202,
      "learning_rate": 2.270433521325518e-06,
      "loss": 0.0247,
      "step": 4597
    },
    {
      "epoch": 3.307318827549002,
      "grad_norm": 7.126794679566996,
      "learning_rate": 2.270141565994044e-06,
      "loss": 0.1819,
      "step": 4598
    },
    {
      "epoch": 3.308038122639813,
      "grad_norm": 1.4065200007617749,
      "learning_rate": 2.2698495710374336e-06,
      "loss": 0.0335,
      "step": 4599
    },
    {
      "epoch": 3.308757417730624,
      "grad_norm": 5.219751576680809,
      "learning_rate": 2.2695575364707123e-06,
      "loss": 0.1905,
      "step": 4600
    },
    {
      "epoch": 3.309476712821435,
      "grad_norm": 2.011498576996309,
      "learning_rate": 2.2692654623089043e-06,
      "loss": 0.0451,
      "step": 4601
    },
    {
      "epoch": 3.310196007912246,
      "grad_norm": 5.471526910226561,
      "learning_rate": 2.2689733485670374e-06,
      "loss": 0.1386,
      "step": 4602
    },
    {
      "epoch": 3.310915303003057,
      "grad_norm": 2.6324609005193302,
      "learning_rate": 2.268681195260143e-06,
      "loss": 0.026,
      "step": 4603
    },
    {
      "epoch": 3.311634598093868,
      "grad_norm": 2.8379221293866235,
      "learning_rate": 2.2683890024032507e-06,
      "loss": 0.0397,
      "step": 4604
    },
    {
      "epoch": 3.312353893184679,
      "grad_norm": 2.401384847474412,
      "learning_rate": 2.2680967700113954e-06,
      "loss": 0.0414,
      "step": 4605
    },
    {
      "epoch": 3.31307318827549,
      "grad_norm": 4.437594829064407,
      "learning_rate": 2.2678044980996138e-06,
      "loss": 0.108,
      "step": 4606
    },
    {
      "epoch": 3.313792483366301,
      "grad_norm": 2.7898312208528577,
      "learning_rate": 2.267512186682942e-06,
      "loss": 0.0615,
      "step": 4607
    },
    {
      "epoch": 3.314511778457112,
      "grad_norm": 3.1202331093819042,
      "learning_rate": 2.2672198357764215e-06,
      "loss": 0.1272,
      "step": 4608
    },
    {
      "epoch": 3.315231073547923,
      "grad_norm": 4.005148568976252,
      "learning_rate": 2.2669274453950936e-06,
      "loss": 0.0549,
      "step": 4609
    },
    {
      "epoch": 3.315950368638734,
      "grad_norm": 3.3979289183777737,
      "learning_rate": 2.266635015554002e-06,
      "loss": 0.0478,
      "step": 4610
    },
    {
      "epoch": 3.316669663729545,
      "grad_norm": 4.255760231466779,
      "learning_rate": 2.2663425462681937e-06,
      "loss": 0.0402,
      "step": 4611
    },
    {
      "epoch": 3.3173889588203562,
      "grad_norm": 0.14566386006638457,
      "learning_rate": 2.2660500375527154e-06,
      "loss": 0.0004,
      "step": 4612
    },
    {
      "epoch": 3.3181082539111673,
      "grad_norm": 3.377909714033796,
      "learning_rate": 2.2657574894226176e-06,
      "loss": 0.0246,
      "step": 4613
    },
    {
      "epoch": 3.318827549001978,
      "grad_norm": 2.7798196755299425,
      "learning_rate": 2.2654649018929538e-06,
      "loss": 0.1077,
      "step": 4614
    },
    {
      "epoch": 3.319546844092789,
      "grad_norm": 3.4333059745254584,
      "learning_rate": 2.265172274978776e-06,
      "loss": 0.1422,
      "step": 4615
    },
    {
      "epoch": 3.3202661391836,
      "grad_norm": 4.149765173269704,
      "learning_rate": 2.2648796086951424e-06,
      "loss": 0.0956,
      "step": 4616
    },
    {
      "epoch": 3.320985434274411,
      "grad_norm": 3.4253001642208973,
      "learning_rate": 2.26458690305711e-06,
      "loss": 0.1475,
      "step": 4617
    },
    {
      "epoch": 3.321704729365222,
      "grad_norm": 2.411090217356525,
      "learning_rate": 2.264294158079739e-06,
      "loss": 0.0919,
      "step": 4618
    },
    {
      "epoch": 3.322424024456033,
      "grad_norm": 1.716393759176561,
      "learning_rate": 2.2640013737780917e-06,
      "loss": 0.0279,
      "step": 4619
    },
    {
      "epoch": 3.323143319546844,
      "grad_norm": 3.5579797535563813,
      "learning_rate": 2.2637085501672326e-06,
      "loss": 0.08,
      "step": 4620
    },
    {
      "epoch": 3.323862614637655,
      "grad_norm": 5.852410579839579,
      "learning_rate": 2.2634156872622283e-06,
      "loss": 0.0247,
      "step": 4621
    },
    {
      "epoch": 3.324581909728466,
      "grad_norm": 3.3410967195786423,
      "learning_rate": 2.2631227850781464e-06,
      "loss": 0.1077,
      "step": 4622
    },
    {
      "epoch": 3.325301204819277,
      "grad_norm": 3.4875794784119467,
      "learning_rate": 2.262829843630058e-06,
      "loss": 0.0496,
      "step": 4623
    },
    {
      "epoch": 3.326020499910088,
      "grad_norm": 2.9862350985219055,
      "learning_rate": 2.2625368629330348e-06,
      "loss": 0.112,
      "step": 4624
    },
    {
      "epoch": 3.3267397950008992,
      "grad_norm": 3.5766844424611715,
      "learning_rate": 2.2622438430021515e-06,
      "loss": 0.0549,
      "step": 4625
    },
    {
      "epoch": 3.3274590900917103,
      "grad_norm": 4.145329247767296,
      "learning_rate": 2.2619507838524844e-06,
      "loss": 0.1037,
      "step": 4626
    },
    {
      "epoch": 3.3281783851825213,
      "grad_norm": 2.374786616840892,
      "learning_rate": 2.2616576854991116e-06,
      "loss": 0.0979,
      "step": 4627
    },
    {
      "epoch": 3.3288976802733323,
      "grad_norm": 3.43815562239229,
      "learning_rate": 2.261364547957114e-06,
      "loss": 0.0667,
      "step": 4628
    },
    {
      "epoch": 3.329616975364143,
      "grad_norm": 3.5825413319629664,
      "learning_rate": 2.261071371241574e-06,
      "loss": 0.0599,
      "step": 4629
    },
    {
      "epoch": 3.330336270454954,
      "grad_norm": 0.9785888096761388,
      "learning_rate": 2.2607781553675764e-06,
      "loss": 0.003,
      "step": 4630
    },
    {
      "epoch": 3.331055565545765,
      "grad_norm": 2.7211478599190375,
      "learning_rate": 2.2604849003502063e-06,
      "loss": 0.0665,
      "step": 4631
    },
    {
      "epoch": 3.331774860636576,
      "grad_norm": 2.5696218567258176,
      "learning_rate": 2.2601916062045536e-06,
      "loss": 0.078,
      "step": 4632
    },
    {
      "epoch": 3.332494155727387,
      "grad_norm": 1.0607272301614792,
      "learning_rate": 2.259898272945708e-06,
      "loss": 0.0282,
      "step": 4633
    },
    {
      "epoch": 3.333213450818198,
      "grad_norm": 2.1483468583017515,
      "learning_rate": 2.2596049005887624e-06,
      "loss": 0.0304,
      "step": 4634
    },
    {
      "epoch": 3.333932745909009,
      "grad_norm": 2.7495025918075258,
      "learning_rate": 2.259311489148811e-06,
      "loss": 0.0699,
      "step": 4635
    },
    {
      "epoch": 3.33465204099982,
      "grad_norm": 6.8697577740125,
      "learning_rate": 2.2590180386409514e-06,
      "loss": 0.1266,
      "step": 4636
    },
    {
      "epoch": 3.335371336090631,
      "grad_norm": 2.451761324228775,
      "learning_rate": 2.258724549080281e-06,
      "loss": 0.0632,
      "step": 4637
    },
    {
      "epoch": 3.3360906311814422,
      "grad_norm": 0.7727714464516373,
      "learning_rate": 2.2584310204819005e-06,
      "loss": 0.0011,
      "step": 4638
    },
    {
      "epoch": 3.3368099262722533,
      "grad_norm": 1.5251708462637936,
      "learning_rate": 2.258137452860912e-06,
      "loss": 0.0058,
      "step": 4639
    },
    {
      "epoch": 3.3375292213630643,
      "grad_norm": 2.038861291381106,
      "learning_rate": 2.2578438462324214e-06,
      "loss": 0.0137,
      "step": 4640
    },
    {
      "epoch": 3.3382485164538753,
      "grad_norm": 4.041237973999525,
      "learning_rate": 2.257550200611534e-06,
      "loss": 0.0567,
      "step": 4641
    },
    {
      "epoch": 3.3389678115446864,
      "grad_norm": 3.3431803805449647,
      "learning_rate": 2.257256516013359e-06,
      "loss": 0.0643,
      "step": 4642
    },
    {
      "epoch": 3.3396871066354974,
      "grad_norm": 4.086610789954164,
      "learning_rate": 2.256962792453008e-06,
      "loss": 0.0952,
      "step": 4643
    },
    {
      "epoch": 3.340406401726308,
      "grad_norm": 2.2747009156079625,
      "learning_rate": 2.2566690299455917e-06,
      "loss": 0.0477,
      "step": 4644
    },
    {
      "epoch": 3.3411256968171195,
      "grad_norm": 1.7163701090440489,
      "learning_rate": 2.256375228506225e-06,
      "loss": 0.0114,
      "step": 4645
    },
    {
      "epoch": 3.34184499190793,
      "grad_norm": 2.4754866696384252,
      "learning_rate": 2.256081388150026e-06,
      "loss": 0.0439,
      "step": 4646
    },
    {
      "epoch": 3.342564286998741,
      "grad_norm": 0.03531110884529071,
      "learning_rate": 2.2557875088921117e-06,
      "loss": 0.0002,
      "step": 4647
    },
    {
      "epoch": 3.343283582089552,
      "grad_norm": 2.2581208739683722,
      "learning_rate": 2.2554935907476036e-06,
      "loss": 0.0259,
      "step": 4648
    },
    {
      "epoch": 3.344002877180363,
      "grad_norm": 3.503500988454564,
      "learning_rate": 2.2551996337316237e-06,
      "loss": 0.0903,
      "step": 4649
    },
    {
      "epoch": 3.344722172271174,
      "grad_norm": 3.9290873109363837,
      "learning_rate": 2.2549056378592974e-06,
      "loss": 0.1438,
      "step": 4650
    },
    {
      "epoch": 3.345441467361985,
      "grad_norm": 1.6002283641932478,
      "learning_rate": 2.2546116031457505e-06,
      "loss": 0.0326,
      "step": 4651
    },
    {
      "epoch": 3.3461607624527963,
      "grad_norm": 3.5557990694248285,
      "learning_rate": 2.2543175296061123e-06,
      "loss": 0.0906,
      "step": 4652
    },
    {
      "epoch": 3.3468800575436073,
      "grad_norm": 6.145418868372815,
      "learning_rate": 2.2540234172555126e-06,
      "loss": 0.2325,
      "step": 4653
    },
    {
      "epoch": 3.3475993526344183,
      "grad_norm": 3.263152845867906,
      "learning_rate": 2.2537292661090845e-06,
      "loss": 0.0913,
      "step": 4654
    },
    {
      "epoch": 3.3483186477252294,
      "grad_norm": 1.6448441310083617,
      "learning_rate": 2.2534350761819627e-06,
      "loss": 0.0277,
      "step": 4655
    },
    {
      "epoch": 3.3490379428160404,
      "grad_norm": 0.8600227807255956,
      "learning_rate": 2.2531408474892836e-06,
      "loss": 0.0081,
      "step": 4656
    },
    {
      "epoch": 3.3497572379068514,
      "grad_norm": 4.091182238194695,
      "learning_rate": 2.2528465800461856e-06,
      "loss": 0.0444,
      "step": 4657
    },
    {
      "epoch": 3.3504765329976625,
      "grad_norm": 4.467972655570028,
      "learning_rate": 2.2525522738678096e-06,
      "loss": 0.1393,
      "step": 4658
    },
    {
      "epoch": 3.351195828088473,
      "grad_norm": 2.885702680797472,
      "learning_rate": 2.2522579289692986e-06,
      "loss": 0.0129,
      "step": 4659
    },
    {
      "epoch": 3.3519151231792845,
      "grad_norm": 4.4838451862595905,
      "learning_rate": 2.251963545365796e-06,
      "loss": 0.0588,
      "step": 4660
    },
    {
      "epoch": 3.352634418270095,
      "grad_norm": 5.723855746996082,
      "learning_rate": 2.251669123072449e-06,
      "loss": 0.114,
      "step": 4661
    },
    {
      "epoch": 3.353353713360906,
      "grad_norm": 5.322550712106414,
      "learning_rate": 2.251374662104406e-06,
      "loss": 0.1994,
      "step": 4662
    },
    {
      "epoch": 3.354073008451717,
      "grad_norm": 1.7644032474537261,
      "learning_rate": 2.251080162476818e-06,
      "loss": 0.0064,
      "step": 4663
    },
    {
      "epoch": 3.354792303542528,
      "grad_norm": 2.5606621465875317,
      "learning_rate": 2.250785624204837e-06,
      "loss": 0.0458,
      "step": 4664
    },
    {
      "epoch": 3.3555115986333393,
      "grad_norm": 1.4876610701893838,
      "learning_rate": 2.250491047303618e-06,
      "loss": 0.0195,
      "step": 4665
    },
    {
      "epoch": 3.3562308937241503,
      "grad_norm": 7.727388536219961,
      "learning_rate": 2.2501964317883166e-06,
      "loss": 0.1523,
      "step": 4666
    },
    {
      "epoch": 3.3569501888149613,
      "grad_norm": 2.166299620552467,
      "learning_rate": 2.249901777674092e-06,
      "loss": 0.0309,
      "step": 4667
    },
    {
      "epoch": 3.3576694839057724,
      "grad_norm": 3.62008146747624,
      "learning_rate": 2.2496070849761046e-06,
      "loss": 0.1292,
      "step": 4668
    },
    {
      "epoch": 3.3583887789965834,
      "grad_norm": 2.653985730380036,
      "learning_rate": 2.2493123537095168e-06,
      "loss": 0.0665,
      "step": 4669
    },
    {
      "epoch": 3.3591080740873944,
      "grad_norm": 3.7695583004826085,
      "learning_rate": 2.2490175838894925e-06,
      "loss": 0.0767,
      "step": 4670
    },
    {
      "epoch": 3.3598273691782055,
      "grad_norm": 0.8552403061605411,
      "learning_rate": 2.2487227755311992e-06,
      "loss": 0.0089,
      "step": 4671
    },
    {
      "epoch": 3.3605466642690165,
      "grad_norm": 5.442625327055703,
      "learning_rate": 2.248427928649805e-06,
      "loss": 0.3339,
      "step": 4672
    },
    {
      "epoch": 3.3612659593598275,
      "grad_norm": 3.7944016774693776,
      "learning_rate": 2.24813304326048e-06,
      "loss": 0.042,
      "step": 4673
    },
    {
      "epoch": 3.3619852544506386,
      "grad_norm": 2.9805827335521,
      "learning_rate": 2.247838119378396e-06,
      "loss": 0.049,
      "step": 4674
    },
    {
      "epoch": 3.3627045495414496,
      "grad_norm": 2.5559681234181997,
      "learning_rate": 2.2475431570187286e-06,
      "loss": 0.0656,
      "step": 4675
    },
    {
      "epoch": 3.36342384463226,
      "grad_norm": 3.600913993243652,
      "learning_rate": 2.247248156196653e-06,
      "loss": 0.0677,
      "step": 4676
    },
    {
      "epoch": 3.364143139723071,
      "grad_norm": 0.9568398140694035,
      "learning_rate": 2.246953116927348e-06,
      "loss": 0.013,
      "step": 4677
    },
    {
      "epoch": 3.3648624348138823,
      "grad_norm": 5.4087340214151975,
      "learning_rate": 2.2466580392259943e-06,
      "loss": 0.109,
      "step": 4678
    },
    {
      "epoch": 3.3655817299046933,
      "grad_norm": 4.776792629506669,
      "learning_rate": 2.2463629231077736e-06,
      "loss": 0.0768,
      "step": 4679
    },
    {
      "epoch": 3.3663010249955043,
      "grad_norm": 0.7862005452467916,
      "learning_rate": 2.2460677685878705e-06,
      "loss": 0.0026,
      "step": 4680
    },
    {
      "epoch": 3.3670203200863154,
      "grad_norm": 1.2502530924420738,
      "learning_rate": 2.245772575681471e-06,
      "loss": 0.015,
      "step": 4681
    },
    {
      "epoch": 3.3677396151771264,
      "grad_norm": 2.7579716090563773,
      "learning_rate": 2.245477344403763e-06,
      "loss": 0.0533,
      "step": 4682
    },
    {
      "epoch": 3.3684589102679374,
      "grad_norm": 2.3366873385121605,
      "learning_rate": 2.245182074769937e-06,
      "loss": 0.0079,
      "step": 4683
    },
    {
      "epoch": 3.3691782053587485,
      "grad_norm": 2.975293668753321,
      "learning_rate": 2.244886766795185e-06,
      "loss": 0.0584,
      "step": 4684
    },
    {
      "epoch": 3.3698975004495595,
      "grad_norm": 0.8520672451743246,
      "learning_rate": 2.2445914204947023e-06,
      "loss": 0.009,
      "step": 4685
    },
    {
      "epoch": 3.3706167955403705,
      "grad_norm": 3.168689809464066,
      "learning_rate": 2.2442960358836828e-06,
      "loss": 0.0222,
      "step": 4686
    },
    {
      "epoch": 3.3713360906311816,
      "grad_norm": 3.4564929491761203,
      "learning_rate": 2.244000612977326e-06,
      "loss": 0.0573,
      "step": 4687
    },
    {
      "epoch": 3.3720553857219926,
      "grad_norm": 1.756055278236517,
      "learning_rate": 2.243705151790832e-06,
      "loss": 0.0457,
      "step": 4688
    },
    {
      "epoch": 3.3727746808128036,
      "grad_norm": 3.67772882765112,
      "learning_rate": 2.2434096523394017e-06,
      "loss": 0.0645,
      "step": 4689
    },
    {
      "epoch": 3.3734939759036147,
      "grad_norm": 1.62152638725986,
      "learning_rate": 2.2431141146382405e-06,
      "loss": 0.0038,
      "step": 4690
    },
    {
      "epoch": 3.3742132709944253,
      "grad_norm": 1.2338867089516974,
      "learning_rate": 2.242818538702553e-06,
      "loss": 0.0188,
      "step": 4691
    },
    {
      "epoch": 3.3749325660852363,
      "grad_norm": 1.2105834463216467,
      "learning_rate": 2.2425229245475478e-06,
      "loss": 0.0124,
      "step": 4692
    },
    {
      "epoch": 3.3756518611760473,
      "grad_norm": 4.782924634672067,
      "learning_rate": 2.242227272188435e-06,
      "loss": 0.1057,
      "step": 4693
    },
    {
      "epoch": 3.3763711562668584,
      "grad_norm": 6.56223586524317,
      "learning_rate": 2.2419315816404257e-06,
      "loss": 0.1561,
      "step": 4694
    },
    {
      "epoch": 3.3770904513576694,
      "grad_norm": 0.874906315979422,
      "learning_rate": 2.2416358529187346e-06,
      "loss": 0.0081,
      "step": 4695
    },
    {
      "epoch": 3.3778097464484804,
      "grad_norm": 5.931676174794837,
      "learning_rate": 2.2413400860385764e-06,
      "loss": 0.2462,
      "step": 4696
    },
    {
      "epoch": 3.3785290415392915,
      "grad_norm": 5.629994826764464,
      "learning_rate": 2.2410442810151698e-06,
      "loss": 0.1874,
      "step": 4697
    },
    {
      "epoch": 3.3792483366301025,
      "grad_norm": 5.238026553500715,
      "learning_rate": 2.240748437863734e-06,
      "loss": 0.0823,
      "step": 4698
    },
    {
      "epoch": 3.3799676317209135,
      "grad_norm": 4.131621153131269,
      "learning_rate": 2.2404525565994904e-06,
      "loss": 0.0354,
      "step": 4699
    },
    {
      "epoch": 3.3806869268117246,
      "grad_norm": 3.4533131216732538,
      "learning_rate": 2.2401566372376638e-06,
      "loss": 0.0553,
      "step": 4700
    },
    {
      "epoch": 3.3814062219025356,
      "grad_norm": 6.023716375523928,
      "learning_rate": 2.239860679793478e-06,
      "loss": 0.08,
      "step": 4701
    },
    {
      "epoch": 3.3821255169933466,
      "grad_norm": 1.1710310757796447,
      "learning_rate": 2.239564684282162e-06,
      "loss": 0.0106,
      "step": 4702
    },
    {
      "epoch": 3.3828448120841577,
      "grad_norm": 2.700397450236177,
      "learning_rate": 2.2392686507189445e-06,
      "loss": 0.0581,
      "step": 4703
    },
    {
      "epoch": 3.3835641071749687,
      "grad_norm": 3.049607382976599,
      "learning_rate": 2.238972579119057e-06,
      "loss": 0.0396,
      "step": 4704
    },
    {
      "epoch": 3.3842834022657797,
      "grad_norm": 0.5042966511924528,
      "learning_rate": 2.2386764694977334e-06,
      "loss": 0.0012,
      "step": 4705
    },
    {
      "epoch": 3.3850026973565903,
      "grad_norm": 0.5870613057463729,
      "learning_rate": 2.2383803218702087e-06,
      "loss": 0.0061,
      "step": 4706
    },
    {
      "epoch": 3.385721992447402,
      "grad_norm": 2.7772328619387063,
      "learning_rate": 2.23808413625172e-06,
      "loss": 0.0754,
      "step": 4707
    },
    {
      "epoch": 3.3864412875382124,
      "grad_norm": 3.8853509258692345,
      "learning_rate": 2.237787912657507e-06,
      "loss": 0.101,
      "step": 4708
    },
    {
      "epoch": 3.3871605826290234,
      "grad_norm": 1.2301877074023235,
      "learning_rate": 2.2374916511028104e-06,
      "loss": 0.019,
      "step": 4709
    },
    {
      "epoch": 3.3878798777198345,
      "grad_norm": 0.06698512511063469,
      "learning_rate": 2.237195351602874e-06,
      "loss": 0.0002,
      "step": 4710
    },
    {
      "epoch": 3.3885991728106455,
      "grad_norm": 2.1117887089124165,
      "learning_rate": 2.236899014172943e-06,
      "loss": 0.0193,
      "step": 4711
    },
    {
      "epoch": 3.3893184679014565,
      "grad_norm": 2.3698672501265694,
      "learning_rate": 2.236602638828264e-06,
      "loss": 0.0588,
      "step": 4712
    },
    {
      "epoch": 3.3900377629922676,
      "grad_norm": 1.477754247353774,
      "learning_rate": 2.2363062255840857e-06,
      "loss": 0.0325,
      "step": 4713
    },
    {
      "epoch": 3.3907570580830786,
      "grad_norm": 4.545704100593133,
      "learning_rate": 2.23600977445566e-06,
      "loss": 0.158,
      "step": 4714
    },
    {
      "epoch": 3.3914763531738896,
      "grad_norm": 3.5984078643191704,
      "learning_rate": 2.235713285458239e-06,
      "loss": 0.1288,
      "step": 4715
    },
    {
      "epoch": 3.3921956482647007,
      "grad_norm": 2.328364900180709,
      "learning_rate": 2.2354167586070785e-06,
      "loss": 0.0508,
      "step": 4716
    },
    {
      "epoch": 3.3929149433555117,
      "grad_norm": 1.1801729186809167,
      "learning_rate": 2.235120193917435e-06,
      "loss": 0.023,
      "step": 4717
    },
    {
      "epoch": 3.3936342384463227,
      "grad_norm": 2.9665168857646584,
      "learning_rate": 2.234823591404566e-06,
      "loss": 0.0739,
      "step": 4718
    },
    {
      "epoch": 3.3943535335371338,
      "grad_norm": 1.6817976416840759,
      "learning_rate": 2.234526951083734e-06,
      "loss": 0.0232,
      "step": 4719
    },
    {
      "epoch": 3.395072828627945,
      "grad_norm": 5.154914619240526,
      "learning_rate": 2.2342302729702003e-06,
      "loss": 0.1776,
      "step": 4720
    },
    {
      "epoch": 3.3957921237187554,
      "grad_norm": 0.4536112945459199,
      "learning_rate": 2.233933557079231e-06,
      "loss": 0.001,
      "step": 4721
    },
    {
      "epoch": 3.396511418809567,
      "grad_norm": 6.357817633434567,
      "learning_rate": 2.2336368034260918e-06,
      "loss": 0.1256,
      "step": 4722
    },
    {
      "epoch": 3.3972307139003775,
      "grad_norm": 4.672749765779479,
      "learning_rate": 2.2333400120260502e-06,
      "loss": 0.149,
      "step": 4723
    },
    {
      "epoch": 3.3979500089911885,
      "grad_norm": 3.8458885065188166,
      "learning_rate": 2.233043182894379e-06,
      "loss": 0.0974,
      "step": 4724
    },
    {
      "epoch": 3.3986693040819995,
      "grad_norm": 1.3360643800555752,
      "learning_rate": 2.232746316046348e-06,
      "loss": 0.0176,
      "step": 4725
    },
    {
      "epoch": 3.3993885991728106,
      "grad_norm": 3.417572831543109,
      "learning_rate": 2.2324494114972335e-06,
      "loss": 0.0443,
      "step": 4726
    },
    {
      "epoch": 3.4001078942636216,
      "grad_norm": 1.2847969351838116,
      "learning_rate": 2.2321524692623107e-06,
      "loss": 0.0186,
      "step": 4727
    },
    {
      "epoch": 3.4008271893544326,
      "grad_norm": 7.080391959934435,
      "learning_rate": 2.231855489356858e-06,
      "loss": 0.146,
      "step": 4728
    },
    {
      "epoch": 3.4015464844452437,
      "grad_norm": 0.08117420920412446,
      "learning_rate": 2.2315584717961557e-06,
      "loss": 0.0004,
      "step": 4729
    },
    {
      "epoch": 3.4022657795360547,
      "grad_norm": 3.2496910956414493,
      "learning_rate": 2.231261416595486e-06,
      "loss": 0.1129,
      "step": 4730
    },
    {
      "epoch": 3.4029850746268657,
      "grad_norm": 2.062150696300699,
      "learning_rate": 2.2309643237701327e-06,
      "loss": 0.0553,
      "step": 4731
    },
    {
      "epoch": 3.4037043697176768,
      "grad_norm": 2.9589660811934393,
      "learning_rate": 2.230667193335382e-06,
      "loss": 0.0203,
      "step": 4732
    },
    {
      "epoch": 3.404423664808488,
      "grad_norm": 2.3570763722052503,
      "learning_rate": 2.2303700253065208e-06,
      "loss": 0.0507,
      "step": 4733
    },
    {
      "epoch": 3.405142959899299,
      "grad_norm": 5.094271613780984,
      "learning_rate": 2.2300728196988405e-06,
      "loss": 0.1658,
      "step": 4734
    },
    {
      "epoch": 3.40586225499011,
      "grad_norm": 2.452345117430938,
      "learning_rate": 2.229775576527632e-06,
      "loss": 0.0371,
      "step": 4735
    },
    {
      "epoch": 3.4065815500809205,
      "grad_norm": 0.9341596809003695,
      "learning_rate": 2.229478295808188e-06,
      "loss": 0.0215,
      "step": 4736
    },
    {
      "epoch": 3.407300845171732,
      "grad_norm": 2.753016232623573,
      "learning_rate": 2.2291809775558058e-06,
      "loss": 0.0567,
      "step": 4737
    },
    {
      "epoch": 3.4080201402625425,
      "grad_norm": 7.266014821988594,
      "learning_rate": 2.2288836217857825e-06,
      "loss": 0.0234,
      "step": 4738
    },
    {
      "epoch": 3.4087394353533536,
      "grad_norm": 3.2188200848077315,
      "learning_rate": 2.2285862285134173e-06,
      "loss": 0.0188,
      "step": 4739
    },
    {
      "epoch": 3.4094587304441646,
      "grad_norm": 2.3016824937207137,
      "learning_rate": 2.228288797754011e-06,
      "loss": 0.0458,
      "step": 4740
    },
    {
      "epoch": 3.4101780255349756,
      "grad_norm": 2.072983380105102,
      "learning_rate": 2.2279913295228682e-06,
      "loss": 0.0338,
      "step": 4741
    },
    {
      "epoch": 3.4108973206257867,
      "grad_norm": 3.8986320642007835,
      "learning_rate": 2.2276938238352935e-06,
      "loss": 0.1289,
      "step": 4742
    },
    {
      "epoch": 3.4116166157165977,
      "grad_norm": 1.9355951217390621,
      "learning_rate": 2.227396280706594e-06,
      "loss": 0.0637,
      "step": 4743
    },
    {
      "epoch": 3.4123359108074087,
      "grad_norm": 1.7729114699246107,
      "learning_rate": 2.227098700152079e-06,
      "loss": 0.0447,
      "step": 4744
    },
    {
      "epoch": 3.4130552058982198,
      "grad_norm": 2.938018749472116,
      "learning_rate": 2.226801082187059e-06,
      "loss": 0.0841,
      "step": 4745
    },
    {
      "epoch": 3.413774500989031,
      "grad_norm": 4.51382066051197,
      "learning_rate": 2.2265034268268483e-06,
      "loss": 0.1777,
      "step": 4746
    },
    {
      "epoch": 3.414493796079842,
      "grad_norm": 4.276834181330756,
      "learning_rate": 2.226205734086761e-06,
      "loss": 0.065,
      "step": 4747
    },
    {
      "epoch": 3.415213091170653,
      "grad_norm": 3.1470600200877135,
      "learning_rate": 2.2259080039821136e-06,
      "loss": 0.0296,
      "step": 4748
    },
    {
      "epoch": 3.415932386261464,
      "grad_norm": 2.3592435874716364,
      "learning_rate": 2.2256102365282243e-06,
      "loss": 0.0127,
      "step": 4749
    },
    {
      "epoch": 3.416651681352275,
      "grad_norm": 1.5945228046146487,
      "learning_rate": 2.2253124317404156e-06,
      "loss": 0.0398,
      "step": 4750
    },
    {
      "epoch": 3.417370976443086,
      "grad_norm": 4.126296426157023,
      "learning_rate": 2.2250145896340086e-06,
      "loss": 0.0123,
      "step": 4751
    },
    {
      "epoch": 3.418090271533897,
      "grad_norm": 4.164000556128089,
      "learning_rate": 2.224716710224328e-06,
      "loss": 0.0986,
      "step": 4752
    },
    {
      "epoch": 3.4188095666247076,
      "grad_norm": 3.4981189654928304,
      "learning_rate": 2.224418793526701e-06,
      "loss": 0.0717,
      "step": 4753
    },
    {
      "epoch": 3.4195288617155186,
      "grad_norm": 3.431808144491488,
      "learning_rate": 2.2241208395564552e-06,
      "loss": 0.1565,
      "step": 4754
    },
    {
      "epoch": 3.4202481568063297,
      "grad_norm": 2.461926186203359,
      "learning_rate": 2.223822848328921e-06,
      "loss": 0.0601,
      "step": 4755
    },
    {
      "epoch": 3.4209674518971407,
      "grad_norm": 0.8505056629769038,
      "learning_rate": 2.223524819859431e-06,
      "loss": 0.0015,
      "step": 4756
    },
    {
      "epoch": 3.4216867469879517,
      "grad_norm": 4.657436120293943,
      "learning_rate": 2.223226754163318e-06,
      "loss": 0.0262,
      "step": 4757
    },
    {
      "epoch": 3.4224060420787628,
      "grad_norm": 2.0413337749692633,
      "learning_rate": 2.2229286512559198e-06,
      "loss": 0.0589,
      "step": 4758
    },
    {
      "epoch": 3.423125337169574,
      "grad_norm": 3.1019435858687157,
      "learning_rate": 2.222630511152573e-06,
      "loss": 0.0572,
      "step": 4759
    },
    {
      "epoch": 3.423844632260385,
      "grad_norm": 8.692928929918308,
      "learning_rate": 2.222332333868618e-06,
      "loss": 0.1414,
      "step": 4760
    },
    {
      "epoch": 3.424563927351196,
      "grad_norm": 2.312905081810137,
      "learning_rate": 2.2220341194193968e-06,
      "loss": 0.0516,
      "step": 4761
    },
    {
      "epoch": 3.425283222442007,
      "grad_norm": 4.827060618384299,
      "learning_rate": 2.221735867820252e-06,
      "loss": 0.1791,
      "step": 4762
    },
    {
      "epoch": 3.426002517532818,
      "grad_norm": 1.0573534294508333,
      "learning_rate": 2.2214375790865296e-06,
      "loss": 0.013,
      "step": 4763
    },
    {
      "epoch": 3.426721812623629,
      "grad_norm": 4.310965306608159,
      "learning_rate": 2.221139253233578e-06,
      "loss": 0.1026,
      "step": 4764
    },
    {
      "epoch": 3.42744110771444,
      "grad_norm": 3.3568607263317216,
      "learning_rate": 2.2208408902767454e-06,
      "loss": 0.0919,
      "step": 4765
    },
    {
      "epoch": 3.428160402805251,
      "grad_norm": 1.7000930435793433,
      "learning_rate": 2.220542490231384e-06,
      "loss": 0.0448,
      "step": 4766
    },
    {
      "epoch": 3.428879697896062,
      "grad_norm": 0.8983436418572041,
      "learning_rate": 2.2202440531128468e-06,
      "loss": 0.01,
      "step": 4767
    },
    {
      "epoch": 3.4295989929868727,
      "grad_norm": 0.5820506919658236,
      "learning_rate": 2.2199455789364886e-06,
      "loss": 0.0051,
      "step": 4768
    },
    {
      "epoch": 3.4303182880776837,
      "grad_norm": 1.844664191878256,
      "learning_rate": 2.219647067717666e-06,
      "loss": 0.0201,
      "step": 4769
    },
    {
      "epoch": 3.4310375831684947,
      "grad_norm": 2.6896506284078194,
      "learning_rate": 2.219348519471739e-06,
      "loss": 0.0693,
      "step": 4770
    },
    {
      "epoch": 3.4317568782593058,
      "grad_norm": 0.8393228314382527,
      "learning_rate": 2.2190499342140676e-06,
      "loss": 0.0042,
      "step": 4771
    },
    {
      "epoch": 3.432476173350117,
      "grad_norm": 1.0645244329435726,
      "learning_rate": 2.2187513119600148e-06,
      "loss": 0.0018,
      "step": 4772
    },
    {
      "epoch": 3.433195468440928,
      "grad_norm": 2.5849355096930937,
      "learning_rate": 2.2184526527249456e-06,
      "loss": 0.1126,
      "step": 4773
    },
    {
      "epoch": 3.433914763531739,
      "grad_norm": 3.6215160325631706,
      "learning_rate": 2.2181539565242258e-06,
      "loss": 0.0437,
      "step": 4774
    },
    {
      "epoch": 3.43463405862255,
      "grad_norm": 0.27295305371903983,
      "learning_rate": 2.2178552233732246e-06,
      "loss": 0.0009,
      "step": 4775
    },
    {
      "epoch": 3.435353353713361,
      "grad_norm": 2.0767110673034637,
      "learning_rate": 2.217556453287312e-06,
      "loss": 0.0473,
      "step": 4776
    },
    {
      "epoch": 3.436072648804172,
      "grad_norm": 3.623912487586658,
      "learning_rate": 2.21725764628186e-06,
      "loss": 0.0642,
      "step": 4777
    },
    {
      "epoch": 3.436791943894983,
      "grad_norm": 0.2156963663171608,
      "learning_rate": 2.216958802372243e-06,
      "loss": 0.0007,
      "step": 4778
    },
    {
      "epoch": 3.437511238985794,
      "grad_norm": 1.407256471520192,
      "learning_rate": 2.2166599215738374e-06,
      "loss": 0.0029,
      "step": 4779
    },
    {
      "epoch": 3.438230534076605,
      "grad_norm": 2.2997588575231753,
      "learning_rate": 2.2163610039020204e-06,
      "loss": 0.0072,
      "step": 4780
    },
    {
      "epoch": 3.438949829167416,
      "grad_norm": 4.880387905896978,
      "learning_rate": 2.216062049372172e-06,
      "loss": 0.2318,
      "step": 4781
    },
    {
      "epoch": 3.439669124258227,
      "grad_norm": 4.383579327297662,
      "learning_rate": 2.2157630579996752e-06,
      "loss": 0.1023,
      "step": 4782
    },
    {
      "epoch": 3.4403884193490377,
      "grad_norm": 2.269023261897153,
      "learning_rate": 2.2154640297999117e-06,
      "loss": 0.0372,
      "step": 4783
    },
    {
      "epoch": 3.441107714439849,
      "grad_norm": 5.8591300479251665,
      "learning_rate": 2.2151649647882683e-06,
      "loss": 0.089,
      "step": 4784
    },
    {
      "epoch": 3.44182700953066,
      "grad_norm": 1.8839948307234988,
      "learning_rate": 2.2148658629801317e-06,
      "loss": 0.0062,
      "step": 4785
    },
    {
      "epoch": 3.442546304621471,
      "grad_norm": 1.5117026637128765,
      "learning_rate": 2.2145667243908917e-06,
      "loss": 0.003,
      "step": 4786
    },
    {
      "epoch": 3.443265599712282,
      "grad_norm": 2.0335287134665556,
      "learning_rate": 2.2142675490359392e-06,
      "loss": 0.0353,
      "step": 4787
    },
    {
      "epoch": 3.443984894803093,
      "grad_norm": 3.9371115547260778,
      "learning_rate": 2.213968336930667e-06,
      "loss": 0.04,
      "step": 4788
    },
    {
      "epoch": 3.444704189893904,
      "grad_norm": 3.218963408983558,
      "learning_rate": 2.2136690880904713e-06,
      "loss": 0.099,
      "step": 4789
    },
    {
      "epoch": 3.445423484984715,
      "grad_norm": 2.425191887586878,
      "learning_rate": 2.2133698025307485e-06,
      "loss": 0.0455,
      "step": 4790
    },
    {
      "epoch": 3.446142780075526,
      "grad_norm": 2.4118511727463092,
      "learning_rate": 2.2130704802668964e-06,
      "loss": 0.0374,
      "step": 4791
    },
    {
      "epoch": 3.446862075166337,
      "grad_norm": 7.404532667329372,
      "learning_rate": 2.2127711213143165e-06,
      "loss": 0.1302,
      "step": 4792
    },
    {
      "epoch": 3.447581370257148,
      "grad_norm": 1.497443467854853,
      "learning_rate": 2.212471725688411e-06,
      "loss": 0.042,
      "step": 4793
    },
    {
      "epoch": 3.448300665347959,
      "grad_norm": 3.2975377265124606,
      "learning_rate": 2.212172293404585e-06,
      "loss": 0.1353,
      "step": 4794
    },
    {
      "epoch": 3.44901996043877,
      "grad_norm": 0.5311910963585446,
      "learning_rate": 2.211872824478244e-06,
      "loss": 0.0033,
      "step": 4795
    },
    {
      "epoch": 3.449739255529581,
      "grad_norm": 2.6612211662663072,
      "learning_rate": 2.2115733189247966e-06,
      "loss": 0.0444,
      "step": 4796
    },
    {
      "epoch": 3.450458550620392,
      "grad_norm": 5.314415983328357,
      "learning_rate": 2.2112737767596527e-06,
      "loss": 0.1169,
      "step": 4797
    },
    {
      "epoch": 3.451177845711203,
      "grad_norm": 1.1604427158550368,
      "learning_rate": 2.2109741979982245e-06,
      "loss": 0.0165,
      "step": 4798
    },
    {
      "epoch": 3.4518971408020143,
      "grad_norm": 2.397253116568816,
      "learning_rate": 2.2106745826559256e-06,
      "loss": 0.0107,
      "step": 4799
    },
    {
      "epoch": 3.452616435892825,
      "grad_norm": 2.995297121283954,
      "learning_rate": 2.210374930748172e-06,
      "loss": 0.0398,
      "step": 4800
    },
    {
      "epoch": 3.453335730983636,
      "grad_norm": 5.240625330951832,
      "learning_rate": 2.210075242290381e-06,
      "loss": 0.0503,
      "step": 4801
    },
    {
      "epoch": 3.454055026074447,
      "grad_norm": 2.337034381877142,
      "learning_rate": 2.2097755172979722e-06,
      "loss": 0.0053,
      "step": 4802
    },
    {
      "epoch": 3.454774321165258,
      "grad_norm": 2.9441691203761993,
      "learning_rate": 2.209475755786368e-06,
      "loss": 0.0769,
      "step": 4803
    },
    {
      "epoch": 3.455493616256069,
      "grad_norm": 0.33313153810628077,
      "learning_rate": 2.20917595777099e-06,
      "loss": 0.0008,
      "step": 4804
    },
    {
      "epoch": 3.45621291134688,
      "grad_norm": 4.820762431462426,
      "learning_rate": 2.208876123267264e-06,
      "loss": 0.051,
      "step": 4805
    },
    {
      "epoch": 3.456932206437691,
      "grad_norm": 3.1215357916527964,
      "learning_rate": 2.208576252290617e-06,
      "loss": 0.0915,
      "step": 4806
    },
    {
      "epoch": 3.457651501528502,
      "grad_norm": 5.115781000789495,
      "learning_rate": 2.208276344856478e-06,
      "loss": 0.1369,
      "step": 4807
    },
    {
      "epoch": 3.458370796619313,
      "grad_norm": 4.257051483880606,
      "learning_rate": 2.207976400980278e-06,
      "loss": 0.0842,
      "step": 4808
    },
    {
      "epoch": 3.459090091710124,
      "grad_norm": 5.732060199041762,
      "learning_rate": 2.207676420677449e-06,
      "loss": 0.0917,
      "step": 4809
    },
    {
      "epoch": 3.459809386800935,
      "grad_norm": 3.8551372652742573,
      "learning_rate": 2.207376403963426e-06,
      "loss": 0.0795,
      "step": 4810
    },
    {
      "epoch": 3.4605286818917462,
      "grad_norm": 5.766331900216892,
      "learning_rate": 2.2070763508536457e-06,
      "loss": 0.1073,
      "step": 4811
    },
    {
      "epoch": 3.4612479769825573,
      "grad_norm": 1.768230523377009,
      "learning_rate": 2.2067762613635452e-06,
      "loss": 0.0055,
      "step": 4812
    },
    {
      "epoch": 3.461967272073368,
      "grad_norm": 0.3718786956399877,
      "learning_rate": 2.2064761355085657e-06,
      "loss": 0.0022,
      "step": 4813
    },
    {
      "epoch": 3.4626865671641793,
      "grad_norm": 0.8282951884961748,
      "learning_rate": 2.2061759733041485e-06,
      "loss": 0.0017,
      "step": 4814
    },
    {
      "epoch": 3.46340586225499,
      "grad_norm": 1.603258333835655,
      "learning_rate": 2.2058757747657385e-06,
      "loss": 0.0193,
      "step": 4815
    },
    {
      "epoch": 3.464125157345801,
      "grad_norm": 0.34204251628380355,
      "learning_rate": 2.2055755399087804e-06,
      "loss": 0.0026,
      "step": 4816
    },
    {
      "epoch": 3.464844452436612,
      "grad_norm": 3.8597933053417743,
      "learning_rate": 2.205275268748722e-06,
      "loss": 0.0734,
      "step": 4817
    },
    {
      "epoch": 3.465563747527423,
      "grad_norm": 6.299047268610459,
      "learning_rate": 2.2049749613010132e-06,
      "loss": 0.1062,
      "step": 4818
    },
    {
      "epoch": 3.466283042618234,
      "grad_norm": 3.7208291349345775,
      "learning_rate": 2.2046746175811052e-06,
      "loss": 0.0733,
      "step": 4819
    },
    {
      "epoch": 3.467002337709045,
      "grad_norm": 1.1055361911179837,
      "learning_rate": 2.204374237604451e-06,
      "loss": 0.0025,
      "step": 4820
    },
    {
      "epoch": 3.467721632799856,
      "grad_norm": 5.443168655841614,
      "learning_rate": 2.2040738213865054e-06,
      "loss": 0.1361,
      "step": 4821
    },
    {
      "epoch": 3.468440927890667,
      "grad_norm": 5.0597320958529055,
      "learning_rate": 2.203773368942726e-06,
      "loss": 0.0979,
      "step": 4822
    },
    {
      "epoch": 3.469160222981478,
      "grad_norm": 5.356266334212832,
      "learning_rate": 2.2034728802885715e-06,
      "loss": 0.0908,
      "step": 4823
    },
    {
      "epoch": 3.4698795180722892,
      "grad_norm": 5.114737848695762,
      "learning_rate": 2.2031723554395022e-06,
      "loss": 0.1271,
      "step": 4824
    },
    {
      "epoch": 3.4705988131631003,
      "grad_norm": 3.0881994600138327,
      "learning_rate": 2.202871794410981e-06,
      "loss": 0.0576,
      "step": 4825
    },
    {
      "epoch": 3.4713181082539113,
      "grad_norm": 4.7809283534040485,
      "learning_rate": 2.202571197218472e-06,
      "loss": 0.1225,
      "step": 4826
    },
    {
      "epoch": 3.4720374033447223,
      "grad_norm": 1.8495115894811514,
      "learning_rate": 2.2022705638774416e-06,
      "loss": 0.0597,
      "step": 4827
    },
    {
      "epoch": 3.4727566984355334,
      "grad_norm": 4.455856983776783,
      "learning_rate": 2.2019698944033576e-06,
      "loss": 0.0893,
      "step": 4828
    },
    {
      "epoch": 3.4734759935263444,
      "grad_norm": 7.28770115351474,
      "learning_rate": 2.201669188811691e-06,
      "loss": 0.0199,
      "step": 4829
    },
    {
      "epoch": 3.474195288617155,
      "grad_norm": 3.4951082900396933,
      "learning_rate": 2.2013684471179123e-06,
      "loss": 0.0092,
      "step": 4830
    },
    {
      "epoch": 3.474914583707966,
      "grad_norm": 5.740298402600414,
      "learning_rate": 2.201067669337496e-06,
      "loss": 0.2041,
      "step": 4831
    },
    {
      "epoch": 3.475633878798777,
      "grad_norm": 4.843091644427837,
      "learning_rate": 2.2007668554859173e-06,
      "loss": 0.1439,
      "step": 4832
    },
    {
      "epoch": 3.476353173889588,
      "grad_norm": 2.814245508710047,
      "learning_rate": 2.2004660055786537e-06,
      "loss": 0.0458,
      "step": 4833
    },
    {
      "epoch": 3.477072468980399,
      "grad_norm": 4.0456574482737455,
      "learning_rate": 2.200165119631185e-06,
      "loss": 0.0715,
      "step": 4834
    },
    {
      "epoch": 3.47779176407121,
      "grad_norm": 4.0440437231253314,
      "learning_rate": 2.199864197658991e-06,
      "loss": 0.0666,
      "step": 4835
    },
    {
      "epoch": 3.478511059162021,
      "grad_norm": 1.902982336394222,
      "learning_rate": 2.199563239677556e-06,
      "loss": 0.0056,
      "step": 4836
    },
    {
      "epoch": 3.4792303542528322,
      "grad_norm": 6.2582767766186285,
      "learning_rate": 2.199262245702364e-06,
      "loss": 0.1685,
      "step": 4837
    },
    {
      "epoch": 3.4799496493436433,
      "grad_norm": 3.1754417759809566,
      "learning_rate": 2.1989612157489017e-06,
      "loss": 0.0316,
      "step": 4838
    },
    {
      "epoch": 3.4806689444344543,
      "grad_norm": 6.567200345995638,
      "learning_rate": 2.1986601498326584e-06,
      "loss": 0.0496,
      "step": 4839
    },
    {
      "epoch": 3.4813882395252653,
      "grad_norm": 4.70674847353393,
      "learning_rate": 2.1983590479691236e-06,
      "loss": 0.1979,
      "step": 4840
    },
    {
      "epoch": 3.4821075346160764,
      "grad_norm": 5.48478163915878,
      "learning_rate": 2.19805791017379e-06,
      "loss": 0.0166,
      "step": 4841
    },
    {
      "epoch": 3.4828268297068874,
      "grad_norm": 0.5247275126491133,
      "learning_rate": 2.197756736462152e-06,
      "loss": 0.001,
      "step": 4842
    },
    {
      "epoch": 3.4835461247976984,
      "grad_norm": 2.7112718389133836,
      "learning_rate": 2.197455526849704e-06,
      "loss": 0.0525,
      "step": 4843
    },
    {
      "epoch": 3.4842654198885095,
      "grad_norm": 2.221012068173011,
      "learning_rate": 2.1971542813519455e-06,
      "loss": 0.0301,
      "step": 4844
    },
    {
      "epoch": 3.48498471497932,
      "grad_norm": 4.475693317034441,
      "learning_rate": 2.196852999984375e-06,
      "loss": 0.1899,
      "step": 4845
    },
    {
      "epoch": 3.485704010070131,
      "grad_norm": 1.7437235918549527,
      "learning_rate": 2.196551682762495e-06,
      "loss": 0.0462,
      "step": 4846
    },
    {
      "epoch": 3.486423305160942,
      "grad_norm": 0.16984235785490032,
      "learning_rate": 2.1962503297018076e-06,
      "loss": 0.0004,
      "step": 4847
    },
    {
      "epoch": 3.487142600251753,
      "grad_norm": 7.079868497042358,
      "learning_rate": 2.195948940817819e-06,
      "loss": 0.07,
      "step": 4848
    },
    {
      "epoch": 3.487861895342564,
      "grad_norm": 3.7555040124173957,
      "learning_rate": 2.1956475161260353e-06,
      "loss": 0.1158,
      "step": 4849
    },
    {
      "epoch": 3.4885811904333752,
      "grad_norm": 4.491133366888093,
      "learning_rate": 2.1953460556419657e-06,
      "loss": 0.0928,
      "step": 4850
    },
    {
      "epoch": 3.4893004855241863,
      "grad_norm": 4.724994717066071,
      "learning_rate": 2.195044559381121e-06,
      "loss": 0.1828,
      "step": 4851
    },
    {
      "epoch": 3.4900197806149973,
      "grad_norm": 3.9067909934061693,
      "learning_rate": 2.1947430273590135e-06,
      "loss": 0.098,
      "step": 4852
    },
    {
      "epoch": 3.4907390757058083,
      "grad_norm": 4.2038001135107885,
      "learning_rate": 2.1944414595911577e-06,
      "loss": 0.0802,
      "step": 4853
    },
    {
      "epoch": 3.4914583707966194,
      "grad_norm": 3.432742509832068,
      "learning_rate": 2.19413985609307e-06,
      "loss": 0.0218,
      "step": 4854
    },
    {
      "epoch": 3.4921776658874304,
      "grad_norm": 0.7951939351877209,
      "learning_rate": 2.193838216880268e-06,
      "loss": 0.0021,
      "step": 4855
    },
    {
      "epoch": 3.4928969609782414,
      "grad_norm": 6.390214057799085,
      "learning_rate": 2.1935365419682715e-06,
      "loss": 0.1049,
      "step": 4856
    },
    {
      "epoch": 3.4936162560690525,
      "grad_norm": 2.1418844754843813,
      "learning_rate": 2.193234831372603e-06,
      "loss": 0.0399,
      "step": 4857
    },
    {
      "epoch": 3.4943355511598635,
      "grad_norm": 1.6271091114694236,
      "learning_rate": 2.192933085108785e-06,
      "loss": 0.0308,
      "step": 4858
    },
    {
      "epoch": 3.4950548462506745,
      "grad_norm": 3.817326141273365,
      "learning_rate": 2.1926313031923435e-06,
      "loss": 0.1025,
      "step": 4859
    },
    {
      "epoch": 3.495774141341485,
      "grad_norm": 4.220084374943218,
      "learning_rate": 2.1923294856388054e-06,
      "loss": 0.0271,
      "step": 4860
    },
    {
      "epoch": 3.4964934364322966,
      "grad_norm": 3.7373019386303117,
      "learning_rate": 2.1920276324637008e-06,
      "loss": 0.1036,
      "step": 4861
    },
    {
      "epoch": 3.497212731523107,
      "grad_norm": 1.5690691261621614,
      "learning_rate": 2.1917257436825586e-06,
      "loss": 0.0101,
      "step": 4862
    },
    {
      "epoch": 3.497932026613918,
      "grad_norm": 3.6064158289155746,
      "learning_rate": 2.1914238193109136e-06,
      "loss": 0.1491,
      "step": 4863
    },
    {
      "epoch": 3.4986513217047293,
      "grad_norm": 3.743077382579993,
      "learning_rate": 2.1911218593642986e-06,
      "loss": 0.1132,
      "step": 4864
    },
    {
      "epoch": 3.4993706167955403,
      "grad_norm": 3.187245347990246,
      "learning_rate": 2.1908198638582506e-06,
      "loss": 0.0114,
      "step": 4865
    },
    {
      "epoch": 3.5000899118863513,
      "grad_norm": 2.1084257743348456,
      "learning_rate": 2.1905178328083082e-06,
      "loss": 0.0174,
      "step": 4866
    },
    {
      "epoch": 3.5008092069771624,
      "grad_norm": 2.763042267534291,
      "learning_rate": 2.190215766230011e-06,
      "loss": 0.0686,
      "step": 4867
    },
    {
      "epoch": 3.5015285020679734,
      "grad_norm": 5.438437336498403,
      "learning_rate": 2.1899136641389015e-06,
      "loss": 0.0419,
      "step": 4868
    },
    {
      "epoch": 3.5022477971587844,
      "grad_norm": 2.6532191757757873,
      "learning_rate": 2.1896115265505226e-06,
      "loss": 0.0629,
      "step": 4869
    },
    {
      "epoch": 3.5029670922495955,
      "grad_norm": 4.757314616140639,
      "learning_rate": 2.1893093534804196e-06,
      "loss": 0.1382,
      "step": 4870
    },
    {
      "epoch": 3.5036863873404065,
      "grad_norm": 3.552614355791362,
      "learning_rate": 2.1890071449441413e-06,
      "loss": 0.0981,
      "step": 4871
    },
    {
      "epoch": 3.5044056824312175,
      "grad_norm": 1.8651862808933177,
      "learning_rate": 2.1887049009572348e-06,
      "loss": 0.0368,
      "step": 4872
    },
    {
      "epoch": 3.5051249775220286,
      "grad_norm": 1.3662920651085406,
      "learning_rate": 2.1884026215352526e-06,
      "loss": 0.0251,
      "step": 4873
    },
    {
      "epoch": 3.5058442726128396,
      "grad_norm": 1.1599515501537767,
      "learning_rate": 2.1881003066937474e-06,
      "loss": 0.0206,
      "step": 4874
    },
    {
      "epoch": 3.50656356770365,
      "grad_norm": 1.7465249054354721,
      "learning_rate": 2.1877979564482725e-06,
      "loss": 0.004,
      "step": 4875
    },
    {
      "epoch": 3.5072828627944617,
      "grad_norm": 2.6156387384693174,
      "learning_rate": 2.1874955708143866e-06,
      "loss": 0.0718,
      "step": 4876
    },
    {
      "epoch": 3.5080021578852723,
      "grad_norm": 2.413137535371077,
      "learning_rate": 2.1871931498076464e-06,
      "loss": 0.024,
      "step": 4877
    },
    {
      "epoch": 3.5087214529760833,
      "grad_norm": 3.629325701602865,
      "learning_rate": 2.1868906934436117e-06,
      "loss": 0.1032,
      "step": 4878
    },
    {
      "epoch": 3.5094407480668943,
      "grad_norm": 3.2838490134698843,
      "learning_rate": 2.186588201737846e-06,
      "loss": 0.0643,
      "step": 4879
    },
    {
      "epoch": 3.5101600431577054,
      "grad_norm": 2.212070957134834,
      "learning_rate": 2.186285674705911e-06,
      "loss": 0.0312,
      "step": 4880
    },
    {
      "epoch": 3.5108793382485164,
      "grad_norm": 3.2144864553877786,
      "learning_rate": 2.185983112363374e-06,
      "loss": 0.1228,
      "step": 4881
    },
    {
      "epoch": 3.5115986333393274,
      "grad_norm": 3.1966301414430887,
      "learning_rate": 2.1856805147258013e-06,
      "loss": 0.0033,
      "step": 4882
    },
    {
      "epoch": 3.5123179284301385,
      "grad_norm": 0.04427591360824512,
      "learning_rate": 2.1853778818087623e-06,
      "loss": 0.0003,
      "step": 4883
    },
    {
      "epoch": 3.5130372235209495,
      "grad_norm": 4.143326777970669,
      "learning_rate": 2.1850752136278283e-06,
      "loss": 0.0796,
      "step": 4884
    },
    {
      "epoch": 3.5137565186117605,
      "grad_norm": 2.872072409903535,
      "learning_rate": 2.1847725101985723e-06,
      "loss": 0.0825,
      "step": 4885
    },
    {
      "epoch": 3.5144758137025716,
      "grad_norm": 1.5345612104279909,
      "learning_rate": 2.1844697715365684e-06,
      "loss": 0.0179,
      "step": 4886
    },
    {
      "epoch": 3.5151951087933826,
      "grad_norm": 0.5790961448218425,
      "learning_rate": 2.184166997657393e-06,
      "loss": 0.0019,
      "step": 4887
    },
    {
      "epoch": 3.5159144038841936,
      "grad_norm": 2.9251428085817133,
      "learning_rate": 2.1838641885766245e-06,
      "loss": 0.0693,
      "step": 4888
    },
    {
      "epoch": 3.5166336989750047,
      "grad_norm": 2.985852853842518,
      "learning_rate": 2.1835613443098435e-06,
      "loss": 0.0822,
      "step": 4889
    },
    {
      "epoch": 3.5173529940658153,
      "grad_norm": 1.2479712757385204,
      "learning_rate": 2.1832584648726307e-06,
      "loss": 0.0034,
      "step": 4890
    },
    {
      "epoch": 3.5180722891566267,
      "grad_norm": 2.20104165452996,
      "learning_rate": 2.1829555502805716e-06,
      "loss": 0.0135,
      "step": 4891
    },
    {
      "epoch": 3.5187915842474373,
      "grad_norm": 1.5618470854524142,
      "learning_rate": 2.18265260054925e-06,
      "loss": 0.0203,
      "step": 4892
    },
    {
      "epoch": 3.5195108793382484,
      "grad_norm": 5.055012622752106,
      "learning_rate": 2.182349615694253e-06,
      "loss": 0.1166,
      "step": 4893
    },
    {
      "epoch": 3.5202301744290594,
      "grad_norm": 1.2706035843690742,
      "learning_rate": 2.182046595731171e-06,
      "loss": 0.0422,
      "step": 4894
    },
    {
      "epoch": 3.5209494695198704,
      "grad_norm": 7.4233894590197735,
      "learning_rate": 2.1817435406755945e-06,
      "loss": 0.1161,
      "step": 4895
    },
    {
      "epoch": 3.5216687646106815,
      "grad_norm": 4.319686697014445,
      "learning_rate": 2.181440450543116e-06,
      "loss": 0.1826,
      "step": 4896
    },
    {
      "epoch": 3.5223880597014925,
      "grad_norm": 0.6589927516925955,
      "learning_rate": 2.1811373253493295e-06,
      "loss": 0.0013,
      "step": 4897
    },
    {
      "epoch": 3.5231073547923035,
      "grad_norm": 1.6402064615708674,
      "learning_rate": 2.1808341651098327e-06,
      "loss": 0.0289,
      "step": 4898
    },
    {
      "epoch": 3.5238266498831146,
      "grad_norm": 3.521533270867982,
      "learning_rate": 2.1805309698402225e-06,
      "loss": 0.0377,
      "step": 4899
    },
    {
      "epoch": 3.5245459449739256,
      "grad_norm": 2.1475470750666843,
      "learning_rate": 2.1802277395561e-06,
      "loss": 0.0092,
      "step": 4900
    },
    {
      "epoch": 3.5252652400647366,
      "grad_norm": 1.3894314856570087,
      "learning_rate": 2.1799244742730653e-06,
      "loss": 0.0269,
      "step": 4901
    },
    {
      "epoch": 3.5259845351555477,
      "grad_norm": 0.1949933081185339,
      "learning_rate": 2.179621174006723e-06,
      "loss": 0.0005,
      "step": 4902
    },
    {
      "epoch": 3.5267038302463587,
      "grad_norm": 2.9381673242638064,
      "learning_rate": 2.1793178387726784e-06,
      "loss": 0.0094,
      "step": 4903
    },
    {
      "epoch": 3.5274231253371697,
      "grad_norm": 3.092016306548222,
      "learning_rate": 2.1790144685865383e-06,
      "loss": 0.1498,
      "step": 4904
    },
    {
      "epoch": 3.5281424204279803,
      "grad_norm": 1.6294322519560263,
      "learning_rate": 2.1787110634639123e-06,
      "loss": 0.0055,
      "step": 4905
    },
    {
      "epoch": 3.528861715518792,
      "grad_norm": 5.463797396451713,
      "learning_rate": 2.1784076234204104e-06,
      "loss": 0.1025,
      "step": 4906
    },
    {
      "epoch": 3.5295810106096024,
      "grad_norm": 4.952837610595327,
      "learning_rate": 2.178104148471645e-06,
      "loss": 0.1765,
      "step": 4907
    },
    {
      "epoch": 3.530300305700414,
      "grad_norm": 6.774843612396574,
      "learning_rate": 2.177800638633232e-06,
      "loss": 0.1159,
      "step": 4908
    },
    {
      "epoch": 3.5310196007912245,
      "grad_norm": 3.4840700682112224,
      "learning_rate": 2.177497093920785e-06,
      "loss": 0.1628,
      "step": 4909
    },
    {
      "epoch": 3.5317388958820355,
      "grad_norm": 6.199866654305057,
      "learning_rate": 2.1771935143499234e-06,
      "loss": 0.1818,
      "step": 4910
    },
    {
      "epoch": 3.5324581909728465,
      "grad_norm": 2.046759083356113,
      "learning_rate": 2.1768898999362668e-06,
      "loss": 0.0241,
      "step": 4911
    },
    {
      "epoch": 3.5331774860636576,
      "grad_norm": 1.9130583771886223,
      "learning_rate": 2.1765862506954364e-06,
      "loss": 0.0272,
      "step": 4912
    },
    {
      "epoch": 3.5338967811544686,
      "grad_norm": 3.385717255185832,
      "learning_rate": 2.1762825666430563e-06,
      "loss": 0.0777,
      "step": 4913
    },
    {
      "epoch": 3.5346160762452796,
      "grad_norm": 7.0984841772019385,
      "learning_rate": 2.1759788477947507e-06,
      "loss": 0.2091,
      "step": 4914
    },
    {
      "epoch": 3.5353353713360907,
      "grad_norm": 0.5230066272528906,
      "learning_rate": 2.175675094166147e-06,
      "loss": 0.0027,
      "step": 4915
    },
    {
      "epoch": 3.5360546664269017,
      "grad_norm": 3.0556022336705726,
      "learning_rate": 2.175371305772873e-06,
      "loss": 0.0162,
      "step": 4916
    },
    {
      "epoch": 3.5367739615177127,
      "grad_norm": 2.514462587495062,
      "learning_rate": 2.1750674826305603e-06,
      "loss": 0.0474,
      "step": 4917
    },
    {
      "epoch": 3.5374932566085238,
      "grad_norm": 3.219340654847328,
      "learning_rate": 2.1747636247548403e-06,
      "loss": 0.0701,
      "step": 4918
    },
    {
      "epoch": 3.538212551699335,
      "grad_norm": 1.6700729865669701,
      "learning_rate": 2.1744597321613472e-06,
      "loss": 0.0433,
      "step": 4919
    },
    {
      "epoch": 3.5389318467901454,
      "grad_norm": 1.2812045924955413,
      "learning_rate": 2.1741558048657173e-06,
      "loss": 0.0149,
      "step": 4920
    },
    {
      "epoch": 3.539651141880957,
      "grad_norm": 2.997424556815624,
      "learning_rate": 2.173851842883588e-06,
      "loss": 0.0611,
      "step": 4921
    },
    {
      "epoch": 3.5403704369717675,
      "grad_norm": 2.165117561067377,
      "learning_rate": 2.1735478462305973e-06,
      "loss": 0.0498,
      "step": 4922
    },
    {
      "epoch": 3.541089732062579,
      "grad_norm": 5.220080729522584,
      "learning_rate": 2.1732438149223885e-06,
      "loss": 0.0575,
      "step": 4923
    },
    {
      "epoch": 3.5418090271533895,
      "grad_norm": 1.818055944976054,
      "learning_rate": 2.172939748974603e-06,
      "loss": 0.0445,
      "step": 4924
    },
    {
      "epoch": 3.5425283222442006,
      "grad_norm": 3.934929578568546,
      "learning_rate": 2.1726356484028865e-06,
      "loss": 0.0846,
      "step": 4925
    },
    {
      "epoch": 3.5432476173350116,
      "grad_norm": 2.7358353990192277,
      "learning_rate": 2.172331513222885e-06,
      "loss": 0.0046,
      "step": 4926
    },
    {
      "epoch": 3.5439669124258226,
      "grad_norm": 3.1842762373118276,
      "learning_rate": 2.1720273434502475e-06,
      "loss": 0.1191,
      "step": 4927
    },
    {
      "epoch": 3.5446862075166337,
      "grad_norm": 2.8212096207384234,
      "learning_rate": 2.171723139100623e-06,
      "loss": 0.0542,
      "step": 4928
    },
    {
      "epoch": 3.5454055026074447,
      "grad_norm": 4.130780804685353,
      "learning_rate": 2.171418900189664e-06,
      "loss": 0.1254,
      "step": 4929
    },
    {
      "epoch": 3.5461247976982557,
      "grad_norm": 4.766454496182892,
      "learning_rate": 2.1711146267330233e-06,
      "loss": 0.0377,
      "step": 4930
    },
    {
      "epoch": 3.5468440927890668,
      "grad_norm": 2.2615499055780535,
      "learning_rate": 2.170810318746357e-06,
      "loss": 0.0511,
      "step": 4931
    },
    {
      "epoch": 3.547563387879878,
      "grad_norm": 3.201066184630596,
      "learning_rate": 2.1705059762453226e-06,
      "loss": 0.0541,
      "step": 4932
    },
    {
      "epoch": 3.548282682970689,
      "grad_norm": 0.1240405240899677,
      "learning_rate": 2.170201599245579e-06,
      "loss": 0.0004,
      "step": 4933
    },
    {
      "epoch": 3.5490019780615,
      "grad_norm": 0.4175299488632816,
      "learning_rate": 2.169897187762786e-06,
      "loss": 0.0031,
      "step": 4934
    },
    {
      "epoch": 3.549721273152311,
      "grad_norm": 2.898932980286494,
      "learning_rate": 2.169592741812607e-06,
      "loss": 0.088,
      "step": 4935
    },
    {
      "epoch": 3.550440568243122,
      "grad_norm": 2.708537208093538,
      "learning_rate": 2.169288261410706e-06,
      "loss": 0.0062,
      "step": 4936
    },
    {
      "epoch": 3.5511598633339325,
      "grad_norm": 2.1083417651944876,
      "learning_rate": 2.1689837465727493e-06,
      "loss": 0.0551,
      "step": 4937
    },
    {
      "epoch": 3.551879158424744,
      "grad_norm": 1.37034943613225,
      "learning_rate": 2.168679197314404e-06,
      "loss": 0.0343,
      "step": 4938
    },
    {
      "epoch": 3.5525984535155546,
      "grad_norm": 7.961506818593601,
      "learning_rate": 2.1683746136513404e-06,
      "loss": 0.191,
      "step": 4939
    },
    {
      "epoch": 3.5533177486063656,
      "grad_norm": 4.359091854811136,
      "learning_rate": 2.1680699955992294e-06,
      "loss": 0.0928,
      "step": 4940
    },
    {
      "epoch": 3.5540370436971767,
      "grad_norm": 3.1692972284042558,
      "learning_rate": 2.167765343173745e-06,
      "loss": 0.0459,
      "step": 4941
    },
    {
      "epoch": 3.5547563387879877,
      "grad_norm": 2.490250825925234,
      "learning_rate": 2.167460656390561e-06,
      "loss": 0.0741,
      "step": 4942
    },
    {
      "epoch": 3.5554756338787987,
      "grad_norm": 1.257446423892046,
      "learning_rate": 2.1671559352653548e-06,
      "loss": 0.0286,
      "step": 4943
    },
    {
      "epoch": 3.5561949289696098,
      "grad_norm": 3.2391543776489216,
      "learning_rate": 2.1668511798138043e-06,
      "loss": 0.1269,
      "step": 4944
    },
    {
      "epoch": 3.556914224060421,
      "grad_norm": 2.312964952252786,
      "learning_rate": 2.1665463900515897e-06,
      "loss": 0.0426,
      "step": 4945
    },
    {
      "epoch": 3.557633519151232,
      "grad_norm": 1.4233288164345326,
      "learning_rate": 2.166241565994394e-06,
      "loss": 0.0039,
      "step": 4946
    },
    {
      "epoch": 3.558352814242043,
      "grad_norm": 3.589166672979967,
      "learning_rate": 2.1659367076578995e-06,
      "loss": 0.1473,
      "step": 4947
    },
    {
      "epoch": 3.559072109332854,
      "grad_norm": 0.5492191115568714,
      "learning_rate": 2.1656318150577926e-06,
      "loss": 0.0023,
      "step": 4948
    },
    {
      "epoch": 3.559791404423665,
      "grad_norm": 3.9694371695879047,
      "learning_rate": 2.1653268882097605e-06,
      "loss": 0.0083,
      "step": 4949
    },
    {
      "epoch": 3.560510699514476,
      "grad_norm": 2.6560628024035977,
      "learning_rate": 2.1650219271294923e-06,
      "loss": 0.0747,
      "step": 4950
    },
    {
      "epoch": 3.561229994605287,
      "grad_norm": 0.9980010998787516,
      "learning_rate": 2.164716931832678e-06,
      "loss": 0.0244,
      "step": 4951
    },
    {
      "epoch": 3.5619492896960976,
      "grad_norm": 5.574308064625232,
      "learning_rate": 2.164411902335011e-06,
      "loss": 0.1874,
      "step": 4952
    },
    {
      "epoch": 3.562668584786909,
      "grad_norm": 0.608369724425219,
      "learning_rate": 2.1641068386521846e-06,
      "loss": 0.0018,
      "step": 4953
    },
    {
      "epoch": 3.5633878798777197,
      "grad_norm": 6.274084201836541,
      "learning_rate": 2.1638017407998963e-06,
      "loss": 0.1306,
      "step": 4954
    },
    {
      "epoch": 3.5641071749685307,
      "grad_norm": 2.6271276905631185,
      "learning_rate": 2.163496608793843e-06,
      "loss": 0.0718,
      "step": 4955
    },
    {
      "epoch": 3.5648264700593417,
      "grad_norm": 2.7970882868535365,
      "learning_rate": 2.1631914426497246e-06,
      "loss": 0.1004,
      "step": 4956
    },
    {
      "epoch": 3.5655457651501528,
      "grad_norm": 8.269943985785455,
      "learning_rate": 2.162886242383242e-06,
      "loss": 0.0712,
      "step": 4957
    },
    {
      "epoch": 3.566265060240964,
      "grad_norm": 3.734004450995905,
      "learning_rate": 2.162581008010099e-06,
      "loss": 0.1203,
      "step": 4958
    },
    {
      "epoch": 3.566984355331775,
      "grad_norm": 1.56706991384246,
      "learning_rate": 2.1622757395459994e-06,
      "loss": 0.0183,
      "step": 4959
    },
    {
      "epoch": 3.567703650422586,
      "grad_norm": 2.4348180965066692,
      "learning_rate": 2.161970437006651e-06,
      "loss": 0.0463,
      "step": 4960
    },
    {
      "epoch": 3.568422945513397,
      "grad_norm": 1.5796661406566637,
      "learning_rate": 2.161665100407761e-06,
      "loss": 0.0039,
      "step": 4961
    },
    {
      "epoch": 3.569142240604208,
      "grad_norm": 3.160879949138244,
      "learning_rate": 2.1613597297650407e-06,
      "loss": 0.0932,
      "step": 4962
    },
    {
      "epoch": 3.569861535695019,
      "grad_norm": 2.209695827724106,
      "learning_rate": 2.161054325094201e-06,
      "loss": 0.0591,
      "step": 4963
    },
    {
      "epoch": 3.57058083078583,
      "grad_norm": 3.882947644563503,
      "learning_rate": 2.1607488864109554e-06,
      "loss": 0.0334,
      "step": 4964
    },
    {
      "epoch": 3.571300125876641,
      "grad_norm": 5.619410526692964,
      "learning_rate": 2.1604434137310203e-06,
      "loss": 0.2277,
      "step": 4965
    },
    {
      "epoch": 3.572019420967452,
      "grad_norm": 1.0372653389807698,
      "learning_rate": 2.1601379070701117e-06,
      "loss": 0.0129,
      "step": 4966
    },
    {
      "epoch": 3.5727387160582627,
      "grad_norm": 4.3503687872688666,
      "learning_rate": 2.1598323664439494e-06,
      "loss": 0.1102,
      "step": 4967
    },
    {
      "epoch": 3.573458011149074,
      "grad_norm": 2.7827456548223606,
      "learning_rate": 2.1595267918682534e-06,
      "loss": 0.039,
      "step": 4968
    },
    {
      "epoch": 3.5741773062398847,
      "grad_norm": 1.2651431851387254,
      "learning_rate": 2.159221183358746e-06,
      "loss": 0.0104,
      "step": 4969
    },
    {
      "epoch": 3.5748966013306958,
      "grad_norm": 2.592730312191458,
      "learning_rate": 2.1589155409311516e-06,
      "loss": 0.0731,
      "step": 4970
    },
    {
      "epoch": 3.575615896421507,
      "grad_norm": 6.741021911897331,
      "learning_rate": 2.158609864601196e-06,
      "loss": 0.07,
      "step": 4971
    },
    {
      "epoch": 3.576335191512318,
      "grad_norm": 3.6197214238346893,
      "learning_rate": 2.158304154384607e-06,
      "loss": 0.1352,
      "step": 4972
    },
    {
      "epoch": 3.577054486603129,
      "grad_norm": 2.9266824049036297,
      "learning_rate": 2.157998410297113e-06,
      "loss": 0.0555,
      "step": 4973
    },
    {
      "epoch": 3.57777378169394,
      "grad_norm": 2.539900770485992,
      "learning_rate": 2.157692632354446e-06,
      "loss": 0.0594,
      "step": 4974
    },
    {
      "epoch": 3.578493076784751,
      "grad_norm": 5.19123338157212,
      "learning_rate": 2.157386820572339e-06,
      "loss": 0.1712,
      "step": 4975
    },
    {
      "epoch": 3.579212371875562,
      "grad_norm": 4.18260445525137,
      "learning_rate": 2.1570809749665254e-06,
      "loss": 0.0818,
      "step": 4976
    },
    {
      "epoch": 3.579931666966373,
      "grad_norm": 5.303174616590207,
      "learning_rate": 2.156775095552742e-06,
      "loss": 0.2442,
      "step": 4977
    },
    {
      "epoch": 3.580650962057184,
      "grad_norm": 7.615785954899499,
      "learning_rate": 2.1564691823467277e-06,
      "loss": 0.1377,
      "step": 4978
    },
    {
      "epoch": 3.581370257147995,
      "grad_norm": 5.916285664751357,
      "learning_rate": 2.156163235364221e-06,
      "loss": 0.095,
      "step": 4979
    },
    {
      "epoch": 3.582089552238806,
      "grad_norm": 4.852688109404041,
      "learning_rate": 2.1558572546209646e-06,
      "loss": 0.055,
      "step": 4980
    },
    {
      "epoch": 3.582808847329617,
      "grad_norm": 4.047262610326514,
      "learning_rate": 2.1555512401327007e-06,
      "loss": 0.1325,
      "step": 4981
    },
    {
      "epoch": 3.5835281424204277,
      "grad_norm": 0.16520524597641148,
      "learning_rate": 2.1552451919151745e-06,
      "loss": 0.0005,
      "step": 4982
    },
    {
      "epoch": 3.584247437511239,
      "grad_norm": 3.8031209495730964,
      "learning_rate": 2.1549391099841335e-06,
      "loss": 0.0448,
      "step": 4983
    },
    {
      "epoch": 3.58496673260205,
      "grad_norm": 0.5955508347828341,
      "learning_rate": 2.1546329943553254e-06,
      "loss": 0.0046,
      "step": 4984
    },
    {
      "epoch": 3.5856860276928613,
      "grad_norm": 1.8398059563277926,
      "learning_rate": 2.1543268450445003e-06,
      "loss": 0.0272,
      "step": 4985
    },
    {
      "epoch": 3.586405322783672,
      "grad_norm": 0.07530066741538657,
      "learning_rate": 2.154020662067411e-06,
      "loss": 0.0003,
      "step": 4986
    },
    {
      "epoch": 3.587124617874483,
      "grad_norm": 2.715835225097343,
      "learning_rate": 2.1537144454398096e-06,
      "loss": 0.0883,
      "step": 4987
    },
    {
      "epoch": 3.587843912965294,
      "grad_norm": 2.6481001221907152,
      "learning_rate": 2.153408195177453e-06,
      "loss": 0.0738,
      "step": 4988
    },
    {
      "epoch": 3.588563208056105,
      "grad_norm": 3.292607766905006,
      "learning_rate": 2.153101911296098e-06,
      "loss": 0.0175,
      "step": 4989
    },
    {
      "epoch": 3.589282503146916,
      "grad_norm": 1.9499554566716863,
      "learning_rate": 2.152795593811503e-06,
      "loss": 0.0464,
      "step": 4990
    },
    {
      "epoch": 3.590001798237727,
      "grad_norm": 3.3520262255954476,
      "learning_rate": 2.1524892427394283e-06,
      "loss": 0.115,
      "step": 4991
    },
    {
      "epoch": 3.590721093328538,
      "grad_norm": 1.9539607069264007,
      "learning_rate": 2.152182858095637e-06,
      "loss": 0.0027,
      "step": 4992
    },
    {
      "epoch": 3.591440388419349,
      "grad_norm": 2.1188639801532383,
      "learning_rate": 2.1518764398958926e-06,
      "loss": 0.0526,
      "step": 4993
    },
    {
      "epoch": 3.59215968351016,
      "grad_norm": 3.3221023159324172,
      "learning_rate": 2.1515699881559615e-06,
      "loss": 0.1019,
      "step": 4994
    },
    {
      "epoch": 3.592878978600971,
      "grad_norm": 2.556026840208387,
      "learning_rate": 2.15126350289161e-06,
      "loss": 0.0145,
      "step": 4995
    },
    {
      "epoch": 3.593598273691782,
      "grad_norm": 1.9304543270871037,
      "learning_rate": 2.1509569841186084e-06,
      "loss": 0.004,
      "step": 4996
    },
    {
      "epoch": 3.594317568782593,
      "grad_norm": 3.437969830642564,
      "learning_rate": 2.1506504318527274e-06,
      "loss": 0.0788,
      "step": 4997
    },
    {
      "epoch": 3.5950368638734043,
      "grad_norm": 0.41351942765961686,
      "learning_rate": 2.1503438461097395e-06,
      "loss": 0.0028,
      "step": 4998
    },
    {
      "epoch": 3.595756158964215,
      "grad_norm": 2.37687191271257,
      "learning_rate": 2.150037226905419e-06,
      "loss": 0.062,
      "step": 4999
    },
    {
      "epoch": 3.5964754540550263,
      "grad_norm": 4.995834331651179,
      "learning_rate": 2.149730574255542e-06,
      "loss": 0.1319,
      "step": 5000
    }
  ],
  "logging_steps": 1,
  "max_steps": 13900,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 10,
  "save_steps": 100,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 2298279037108224.0,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}