{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.5001406645619081,
  "eval_steps": 500,
  "global_step": 500,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.001000281329123816,
      "grad_norm": 7.503077030181885,
      "learning_rate": 0.0001999999450590425,
      "loss": 3.6513,
      "step": 1
    },
    {
      "epoch": 0.002000562658247632,
      "grad_norm": 3.1191189289093018,
      "learning_rate": 0.00019999978023623033,
      "loss": 2.8683,
      "step": 2
    },
    {
      "epoch": 0.003000843987371448,
      "grad_norm": 1.9282511472702026,
      "learning_rate": 0.0001999995055317446,
      "loss": 2.7882,
      "step": 3
    },
    {
      "epoch": 0.004001125316495264,
      "grad_norm": 1.726026177406311,
      "learning_rate": 0.00019999912094588717,
      "loss": 2.7716,
      "step": 4
    },
    {
      "epoch": 0.005001406645619081,
      "grad_norm": 1.4632996320724487,
      "learning_rate": 0.00019999862647908064,
      "loss": 2.6869,
      "step": 5
    },
    {
      "epoch": 0.006001687974742896,
      "grad_norm": 1.5544543266296387,
      "learning_rate": 0.00019999802213186834,
      "loss": 2.6952,
      "step": 6
    },
    {
      "epoch": 0.007001969303866712,
      "grad_norm": 1.5888980627059937,
      "learning_rate": 0.0001999973079049143,
      "loss": 2.1237,
      "step": 7
    },
    {
      "epoch": 0.008002250632990529,
      "grad_norm": 1.8750641345977783,
      "learning_rate": 0.00019999648379900338,
      "loss": 2.3376,
      "step": 8
    },
    {
      "epoch": 0.009002531962114344,
      "grad_norm": 1.0540648698806763,
      "learning_rate": 0.0001999955498150411,
      "loss": 2.4896,
      "step": 9
    },
    {
      "epoch": 0.010002813291238161,
      "grad_norm": 1.0269274711608887,
      "learning_rate": 0.00019999450595405374,
      "loss": 2.1365,
      "step": 10
    },
    {
      "epoch": 0.011003094620361977,
      "grad_norm": 1.0851730108261108,
      "learning_rate": 0.0001999933522171883,
      "loss": 2.235,
      "step": 11
    },
    {
      "epoch": 0.012003375949485792,
      "grad_norm": 0.927042543888092,
      "learning_rate": 0.00019999208860571255,
      "loss": 2.2438,
      "step": 12
    },
    {
      "epoch": 0.01300365727860961,
      "grad_norm": 1.3729208707809448,
      "learning_rate": 0.00019999071512101496,
      "loss": 2.0845,
      "step": 13
    },
    {
      "epoch": 0.014003938607733425,
      "grad_norm": 1.1325910091400146,
      "learning_rate": 0.00019998923176460474,
      "loss": 2.0668,
      "step": 14
    },
    {
      "epoch": 0.01500421993685724,
      "grad_norm": 0.9290457367897034,
      "learning_rate": 0.00019998763853811184,
      "loss": 2.0227,
      "step": 15
    },
    {
      "epoch": 0.016004501265981057,
      "grad_norm": 0.942140519618988,
      "learning_rate": 0.00019998593544328692,
      "loss": 2.1598,
      "step": 16
    },
    {
      "epoch": 0.017004782595104875,
      "grad_norm": 1.096635103225708,
      "learning_rate": 0.00019998412248200138,
      "loss": 2.1897,
      "step": 17
    },
    {
      "epoch": 0.01800506392422869,
      "grad_norm": 1.1107186079025269,
      "learning_rate": 0.00019998219965624734,
      "loss": 2.0546,
      "step": 18
    },
    {
      "epoch": 0.019005345253352506,
      "grad_norm": 0.9696593880653381,
      "learning_rate": 0.0001999801669681376,
      "loss": 2.0317,
      "step": 19
    },
    {
      "epoch": 0.020005626582476323,
      "grad_norm": 0.9394300580024719,
      "learning_rate": 0.00019997802441990573,
      "loss": 2.2883,
      "step": 20
    },
    {
      "epoch": 0.021005907911600136,
      "grad_norm": 1.08865225315094,
      "learning_rate": 0.00019997577201390606,
      "loss": 1.9838,
      "step": 21
    },
    {
      "epoch": 0.022006189240723954,
      "grad_norm": 1.0712405443191528,
      "learning_rate": 0.00019997340975261353,
      "loss": 2.1177,
      "step": 22
    },
    {
      "epoch": 0.02300647056984777,
      "grad_norm": 1.3190314769744873,
      "learning_rate": 0.00019997093763862383,
      "loss": 1.9755,
      "step": 23
    },
    {
      "epoch": 0.024006751898971584,
      "grad_norm": 1.0659812688827515,
      "learning_rate": 0.0001999683556746534,
      "loss": 1.9829,
      "step": 24
    },
    {
      "epoch": 0.0250070332280954,
      "grad_norm": 1.1824345588684082,
      "learning_rate": 0.0001999656638635393,
      "loss": 2.4219,
      "step": 25
    },
    {
      "epoch": 0.02600731455721922,
      "grad_norm": 1.3446214199066162,
      "learning_rate": 0.0001999628622082394,
      "loss": 1.9644,
      "step": 26
    },
    {
      "epoch": 0.027007595886343033,
      "grad_norm": 1.2527475357055664,
      "learning_rate": 0.0001999599507118322,
      "loss": 2.1889,
      "step": 27
    },
    {
      "epoch": 0.02800787721546685,
      "grad_norm": 1.4738999605178833,
      "learning_rate": 0.00019995692937751683,
      "loss": 2.1949,
      "step": 28
    },
    {
      "epoch": 0.029008158544590667,
      "grad_norm": 1.0533576011657715,
      "learning_rate": 0.0001999537982086133,
      "loss": 2.1034,
      "step": 29
    },
    {
      "epoch": 0.03000843987371448,
      "grad_norm": 1.0343223810195923,
      "learning_rate": 0.00019995055720856218,
      "loss": 1.9561,
      "step": 30
    },
    {
      "epoch": 0.031008721202838298,
      "grad_norm": 1.1149976253509521,
      "learning_rate": 0.00019994720638092468,
      "loss": 2.0981,
      "step": 31
    },
    {
      "epoch": 0.032009002531962115,
      "grad_norm": 1.197178840637207,
      "learning_rate": 0.00019994374572938277,
      "loss": 2.1587,
      "step": 32
    },
    {
      "epoch": 0.03300928386108593,
      "grad_norm": 0.9382303953170776,
      "learning_rate": 0.00019994017525773913,
      "loss": 1.869,
      "step": 33
    },
    {
      "epoch": 0.03400956519020975,
      "grad_norm": 1.0526461601257324,
      "learning_rate": 0.00019993649496991705,
      "loss": 1.9045,
      "step": 34
    },
    {
      "epoch": 0.03500984651933356,
      "grad_norm": 0.8510498404502869,
      "learning_rate": 0.00019993270486996046,
      "loss": 2.1005,
      "step": 35
    },
    {
      "epoch": 0.03601012784845738,
      "grad_norm": 0.9990401268005371,
      "learning_rate": 0.000199928804962034,
      "loss": 1.8569,
      "step": 36
    },
    {
      "epoch": 0.037010409177581194,
      "grad_norm": 0.9243854284286499,
      "learning_rate": 0.00019992479525042303,
      "loss": 1.9666,
      "step": 37
    },
    {
      "epoch": 0.03801069050670501,
      "grad_norm": 0.7774227261543274,
      "learning_rate": 0.00019992067573953342,
      "loss": 2.0376,
      "step": 38
    },
    {
      "epoch": 0.03901097183582883,
      "grad_norm": 0.8114833235740662,
      "learning_rate": 0.0001999164464338918,
      "loss": 2.1608,
      "step": 39
    },
    {
      "epoch": 0.040011253164952645,
      "grad_norm": 0.8716320395469666,
      "learning_rate": 0.0001999121073381454,
      "loss": 2.0743,
      "step": 40
    },
    {
      "epoch": 0.041011534494076456,
      "grad_norm": 0.9571239948272705,
      "learning_rate": 0.0001999076584570621,
      "loss": 2.0128,
      "step": 41
    },
    {
      "epoch": 0.04201181582320027,
      "grad_norm": 1.038691520690918,
      "learning_rate": 0.00019990309979553045,
      "loss": 1.976,
      "step": 42
    },
    {
      "epoch": 0.04301209715232409,
      "grad_norm": 1.0576292276382446,
      "learning_rate": 0.00019989843135855958,
      "loss": 1.94,
      "step": 43
    },
    {
      "epoch": 0.04401237848144791,
      "grad_norm": 1.0991204977035522,
      "learning_rate": 0.00019989365315127922,
      "loss": 1.9397,
      "step": 44
    },
    {
      "epoch": 0.045012659810571724,
      "grad_norm": 0.9268686175346375,
      "learning_rate": 0.0001998887651789398,
      "loss": 1.9305,
      "step": 45
    },
    {
      "epoch": 0.04601294113969554,
      "grad_norm": 0.8459104299545288,
      "learning_rate": 0.0001998837674469123,
      "loss": 1.7941,
      "step": 46
    },
    {
      "epoch": 0.04701322246881936,
      "grad_norm": 0.9260527491569519,
      "learning_rate": 0.00019987865996068833,
      "loss": 1.8843,
      "step": 47
    },
    {
      "epoch": 0.04801350379794317,
      "grad_norm": 0.8370497226715088,
      "learning_rate": 0.00019987344272588006,
      "loss": 1.8779,
      "step": 48
    },
    {
      "epoch": 0.049013785127066986,
      "grad_norm": 0.9228008389472961,
      "learning_rate": 0.00019986811574822033,
      "loss": 2.1713,
      "step": 49
    },
    {
      "epoch": 0.0500140664561908,
      "grad_norm": 1.013746738433838,
      "learning_rate": 0.00019986267903356254,
      "loss": 2.1443,
      "step": 50
    },
    {
      "epoch": 0.05101434778531462,
      "grad_norm": 1.0155737400054932,
      "learning_rate": 0.0001998571325878806,
      "loss": 1.9679,
      "step": 51
    },
    {
      "epoch": 0.05201462911443844,
      "grad_norm": 0.9591345191001892,
      "learning_rate": 0.0001998514764172691,
      "loss": 2.0611,
      "step": 52
    },
    {
      "epoch": 0.053014910443562255,
      "grad_norm": 0.9030050039291382,
      "learning_rate": 0.00019984571052794313,
      "loss": 1.9698,
      "step": 53
    },
    {
      "epoch": 0.054015191772686065,
      "grad_norm": 0.7697799205780029,
      "learning_rate": 0.00019983983492623833,
      "loss": 2.0609,
      "step": 54
    },
    {
      "epoch": 0.05501547310180988,
      "grad_norm": 0.8806005716323853,
      "learning_rate": 0.00019983384961861096,
      "loss": 1.9756,
      "step": 55
    },
    {
      "epoch": 0.0560157544309337,
      "grad_norm": 0.9424449801445007,
      "learning_rate": 0.0001998277546116378,
      "loss": 2.0913,
      "step": 56
    },
    {
      "epoch": 0.05701603576005752,
      "grad_norm": 1.139495849609375,
      "learning_rate": 0.00019982154991201608,
      "loss": 2.2524,
      "step": 57
    },
    {
      "epoch": 0.058016317089181334,
      "grad_norm": 1.094347357749939,
      "learning_rate": 0.00019981523552656377,
      "loss": 1.8501,
      "step": 58
    },
    {
      "epoch": 0.05901659841830515,
      "grad_norm": 1.1519278287887573,
      "learning_rate": 0.00019980881146221914,
      "loss": 1.9866,
      "step": 59
    },
    {
      "epoch": 0.06001687974742896,
      "grad_norm": 1.2018250226974487,
      "learning_rate": 0.00019980227772604112,
      "loss": 1.8226,
      "step": 60
    },
    {
      "epoch": 0.06101716107655278,
      "grad_norm": 0.9565753936767578,
      "learning_rate": 0.0001997956343252091,
      "loss": 1.8434,
      "step": 61
    },
    {
      "epoch": 0.062017442405676595,
      "grad_norm": 1.0832768678665161,
      "learning_rate": 0.00019978888126702296,
      "loss": 2.1271,
      "step": 62
    },
    {
      "epoch": 0.06301772373480041,
      "grad_norm": 0.8973837494850159,
      "learning_rate": 0.00019978201855890308,
      "loss": 1.8331,
      "step": 63
    },
    {
      "epoch": 0.06401800506392423,
      "grad_norm": 0.8754604458808899,
      "learning_rate": 0.00019977504620839035,
      "loss": 2.1379,
      "step": 64
    },
    {
      "epoch": 0.06501828639304805,
      "grad_norm": 0.8244839310646057,
      "learning_rate": 0.00019976796422314615,
      "loss": 1.8431,
      "step": 65
    },
    {
      "epoch": 0.06601856772217186,
      "grad_norm": 0.8213551044464111,
      "learning_rate": 0.00019976077261095226,
      "loss": 1.9155,
      "step": 66
    },
    {
      "epoch": 0.06701884905129568,
      "grad_norm": 0.9140985608100891,
      "learning_rate": 0.00019975347137971098,
      "loss": 2.0651,
      "step": 67
    },
    {
      "epoch": 0.0680191303804195,
      "grad_norm": 0.8518921732902527,
      "learning_rate": 0.00019974606053744503,
      "loss": 1.8197,
      "step": 68
    },
    {
      "epoch": 0.06901941170954332,
      "grad_norm": 0.8397145867347717,
      "learning_rate": 0.00019973854009229763,
      "loss": 1.8621,
      "step": 69
    },
    {
      "epoch": 0.07001969303866712,
      "grad_norm": 0.8727964162826538,
      "learning_rate": 0.00019973091005253232,
      "loss": 1.762,
      "step": 70
    },
    {
      "epoch": 0.07101997436779094,
      "grad_norm": 0.9284623265266418,
      "learning_rate": 0.0001997231704265332,
      "loss": 1.8675,
      "step": 71
    },
    {
      "epoch": 0.07202025569691475,
      "grad_norm": 0.8280015587806702,
      "learning_rate": 0.00019971532122280464,
      "loss": 1.931,
      "step": 72
    },
    {
      "epoch": 0.07302053702603857,
      "grad_norm": 0.7591394186019897,
      "learning_rate": 0.0001997073624499716,
      "loss": 1.8485,
      "step": 73
    },
    {
      "epoch": 0.07402081835516239,
      "grad_norm": 0.975128710269928,
      "learning_rate": 0.0001996992941167792,
      "loss": 2.0784,
      "step": 74
    },
    {
      "epoch": 0.0750210996842862,
      "grad_norm": 0.8034948110580444,
      "learning_rate": 0.00019969111623209323,
      "loss": 1.9849,
      "step": 75
    },
    {
      "epoch": 0.07602138101341002,
      "grad_norm": 0.8540483713150024,
      "learning_rate": 0.00019968282880489957,
      "loss": 1.7832,
      "step": 76
    },
    {
      "epoch": 0.07702166234253384,
      "grad_norm": 0.8181695342063904,
      "learning_rate": 0.00019967443184430467,
      "loss": 1.944,
      "step": 77
    },
    {
      "epoch": 0.07802194367165766,
      "grad_norm": 0.8446747064590454,
      "learning_rate": 0.0001996659253595353,
      "loss": 1.8508,
      "step": 78
    },
    {
      "epoch": 0.07902222500078147,
      "grad_norm": 0.8280364871025085,
      "learning_rate": 0.0001996573093599385,
      "loss": 1.843,
      "step": 79
    },
    {
      "epoch": 0.08002250632990529,
      "grad_norm": 0.8016006350517273,
      "learning_rate": 0.00019964858385498172,
      "loss": 1.9368,
      "step": 80
    },
    {
      "epoch": 0.08102278765902911,
      "grad_norm": 0.8450536131858826,
      "learning_rate": 0.00019963974885425266,
      "loss": 1.9736,
      "step": 81
    },
    {
      "epoch": 0.08202306898815291,
      "grad_norm": 0.9172171950340271,
      "learning_rate": 0.00019963080436745945,
      "loss": 1.9382,
      "step": 82
    },
    {
      "epoch": 0.08302335031727673,
      "grad_norm": 0.8581916689872742,
      "learning_rate": 0.00019962175040443044,
      "loss": 2.224,
      "step": 83
    },
    {
      "epoch": 0.08402363164640055,
      "grad_norm": 0.9350367188453674,
      "learning_rate": 0.0001996125869751143,
      "loss": 2.0519,
      "step": 84
    },
    {
      "epoch": 0.08502391297552436,
      "grad_norm": 0.9276247620582581,
      "learning_rate": 0.00019960331408957997,
      "loss": 1.9657,
      "step": 85
    },
    {
      "epoch": 0.08602419430464818,
      "grad_norm": 0.871574342250824,
      "learning_rate": 0.00019959393175801671,
      "loss": 1.9399,
      "step": 86
    },
    {
      "epoch": 0.087024475633772,
      "grad_norm": 1.0662888288497925,
      "learning_rate": 0.00019958443999073397,
      "loss": 1.9089,
      "step": 87
    },
    {
      "epoch": 0.08802475696289581,
      "grad_norm": 0.8258713483810425,
      "learning_rate": 0.00019957483879816151,
      "loss": 1.839,
      "step": 88
    },
    {
      "epoch": 0.08902503829201963,
      "grad_norm": 0.8154664039611816,
      "learning_rate": 0.00019956512819084928,
      "loss": 1.8409,
      "step": 89
    },
    {
      "epoch": 0.09002531962114345,
      "grad_norm": 0.8584638833999634,
      "learning_rate": 0.00019955530817946748,
      "loss": 1.9521,
      "step": 90
    },
    {
      "epoch": 0.09102560095026727,
      "grad_norm": 0.7917523384094238,
      "learning_rate": 0.00019954537877480655,
      "loss": 1.9495,
      "step": 91
    },
    {
      "epoch": 0.09202588227939108,
      "grad_norm": 1.0129039287567139,
      "learning_rate": 0.00019953533998777706,
      "loss": 1.949,
      "step": 92
    },
    {
      "epoch": 0.0930261636085149,
      "grad_norm": 0.8677986264228821,
      "learning_rate": 0.00019952519182940993,
      "loss": 1.7875,
      "step": 93
    },
    {
      "epoch": 0.09402644493763872,
      "grad_norm": 0.8848614692687988,
      "learning_rate": 0.00019951493431085603,
      "loss": 2.0675,
      "step": 94
    },
    {
      "epoch": 0.09502672626676252,
      "grad_norm": 0.9936463832855225,
      "learning_rate": 0.00019950456744338658,
      "loss": 1.6761,
      "step": 95
    },
    {
      "epoch": 0.09602700759588634,
      "grad_norm": 1.0520148277282715,
      "learning_rate": 0.00019949409123839288,
      "loss": 2.081,
      "step": 96
    },
    {
      "epoch": 0.09702728892501016,
      "grad_norm": 0.8061773180961609,
      "learning_rate": 0.00019948350570738642,
      "loss": 1.7281,
      "step": 97
    },
    {
      "epoch": 0.09802757025413397,
      "grad_norm": 0.7642756104469299,
      "learning_rate": 0.0001994728108619987,
      "loss": 2.0032,
      "step": 98
    },
    {
      "epoch": 0.09902785158325779,
      "grad_norm": 0.8541550040245056,
      "learning_rate": 0.0001994620067139815,
      "loss": 2.1136,
      "step": 99
    },
    {
      "epoch": 0.1000281329123816,
      "grad_norm": 0.7868679761886597,
      "learning_rate": 0.00019945109327520658,
      "loss": 1.8695,
      "step": 100
    },
    {
      "epoch": 0.10102841424150542,
      "grad_norm": 0.8776901364326477,
      "learning_rate": 0.00019944007055766586,
      "loss": 1.9786,
      "step": 101
    },
    {
      "epoch": 0.10202869557062924,
      "grad_norm": 0.9013833999633789,
      "learning_rate": 0.00019942893857347128,
      "loss": 2.1466,
      "step": 102
    },
    {
      "epoch": 0.10302897689975306,
      "grad_norm": 0.957558274269104,
      "learning_rate": 0.00019941769733485494,
      "loss": 2.0473,
      "step": 103
    },
    {
      "epoch": 0.10402925822887688,
      "grad_norm": 0.8921108841896057,
      "learning_rate": 0.00019940634685416888,
      "loss": 1.7882,
      "step": 104
    },
    {
      "epoch": 0.10502953955800069,
      "grad_norm": 0.896019697189331,
      "learning_rate": 0.00019939488714388524,
      "loss": 1.8811,
      "step": 105
    },
    {
      "epoch": 0.10602982088712451,
      "grad_norm": 0.8792067766189575,
      "learning_rate": 0.00019938331821659614,
      "loss": 1.8624,
      "step": 106
    },
    {
      "epoch": 0.10703010221624833,
      "grad_norm": 0.8739930391311646,
      "learning_rate": 0.0001993716400850138,
      "loss": 1.8105,
      "step": 107
    },
    {
      "epoch": 0.10803038354537213,
      "grad_norm": 0.7678424715995789,
      "learning_rate": 0.0001993598527619703,
      "loss": 1.8772,
      "step": 108
    },
    {
      "epoch": 0.10903066487449595,
      "grad_norm": 0.8718745112419128,
      "learning_rate": 0.00019934795626041783,
      "loss": 1.8236,
      "step": 109
    },
    {
      "epoch": 0.11003094620361976,
      "grad_norm": 0.8467247486114502,
      "learning_rate": 0.0001993359505934285,
      "loss": 1.8188,
      "step": 110
    },
    {
      "epoch": 0.11103122753274358,
      "grad_norm": 0.8685783743858337,
      "learning_rate": 0.00019932383577419432,
      "loss": 2.0775,
      "step": 111
    },
    {
      "epoch": 0.1120315088618674,
      "grad_norm": 0.7799698710441589,
      "learning_rate": 0.0001993116118160273,
      "loss": 1.6489,
      "step": 112
    },
    {
      "epoch": 0.11303179019099122,
      "grad_norm": 0.7900094389915466,
      "learning_rate": 0.00019929927873235938,
      "loss": 1.8332,
      "step": 113
    },
    {
      "epoch": 0.11403207152011503,
      "grad_norm": 0.9433258771896362,
      "learning_rate": 0.00019928683653674237,
      "loss": 1.9331,
      "step": 114
    },
    {
      "epoch": 0.11503235284923885,
      "grad_norm": 0.8861056566238403,
      "learning_rate": 0.00019927428524284805,
      "loss": 1.9135,
      "step": 115
    },
    {
      "epoch": 0.11603263417836267,
      "grad_norm": 0.8566756844520569,
      "learning_rate": 0.00019926162486446792,
      "loss": 1.9874,
      "step": 116
    },
    {
      "epoch": 0.11703291550748648,
      "grad_norm": 0.6897929310798645,
      "learning_rate": 0.0001992488554155135,
      "loss": 1.946,
      "step": 117
    },
    {
      "epoch": 0.1180331968366103,
      "grad_norm": 0.7807729244232178,
      "learning_rate": 0.00019923597691001615,
      "loss": 1.8127,
      "step": 118
    },
    {
      "epoch": 0.11903347816573412,
      "grad_norm": 0.7572523355484009,
      "learning_rate": 0.0001992229893621269,
      "loss": 1.7768,
      "step": 119
    },
    {
      "epoch": 0.12003375949485792,
      "grad_norm": 0.7393172979354858,
      "learning_rate": 0.00019920989278611687,
      "loss": 1.894,
      "step": 120
    },
    {
      "epoch": 0.12103404082398174,
      "grad_norm": 0.866576611995697,
      "learning_rate": 0.0001991966871963767,
      "loss": 1.9285,
      "step": 121
    },
    {
      "epoch": 0.12203432215310556,
      "grad_norm": 0.7326533794403076,
      "learning_rate": 0.000199183372607417,
      "loss": 1.9309,
      "step": 122
    },
    {
      "epoch": 0.12303460348222937,
      "grad_norm": 0.7655537724494934,
      "learning_rate": 0.0001991699490338681,
      "loss": 2.1145,
      "step": 123
    },
    {
      "epoch": 0.12403488481135319,
      "grad_norm": 0.9714633226394653,
      "learning_rate": 0.00019915641649048005,
      "loss": 2.0341,
      "step": 124
    },
    {
      "epoch": 0.12503516614047702,
      "grad_norm": 0.8542420864105225,
      "learning_rate": 0.0001991427749921227,
      "loss": 2.1426,
      "step": 125
    },
    {
      "epoch": 0.12603544746960083,
      "grad_norm": 0.8286274671554565,
      "learning_rate": 0.00019912902455378556,
      "loss": 1.8452,
      "step": 126
    },
    {
      "epoch": 0.12703572879872463,
      "grad_norm": 0.8823768496513367,
      "learning_rate": 0.00019911516519057788,
      "loss": 1.8651,
      "step": 127
    },
    {
      "epoch": 0.12803601012784846,
      "grad_norm": 0.7301567196846008,
      "learning_rate": 0.00019910119691772863,
      "loss": 1.7776,
      "step": 128
    },
    {
      "epoch": 0.12903629145697226,
      "grad_norm": 0.8402552604675293,
      "learning_rate": 0.00019908711975058637,
      "loss": 1.8617,
      "step": 129
    },
    {
      "epoch": 0.1300365727860961,
      "grad_norm": 0.814500093460083,
      "learning_rate": 0.0001990729337046194,
      "loss": 1.9156,
      "step": 130
    },
    {
      "epoch": 0.1310368541152199,
      "grad_norm": 0.8262699246406555,
      "learning_rate": 0.0001990586387954156,
      "loss": 1.8659,
      "step": 131
    },
    {
      "epoch": 0.13203713544434373,
      "grad_norm": 0.8846324682235718,
      "learning_rate": 0.00019904423503868247,
      "loss": 2.043,
      "step": 132
    },
    {
      "epoch": 0.13303741677346753,
      "grad_norm": 0.8757227659225464,
      "learning_rate": 0.00019902972245024715,
      "loss": 1.9217,
      "step": 133
    },
    {
      "epoch": 0.13403769810259136,
      "grad_norm": 0.8476879596710205,
      "learning_rate": 0.00019901510104605637,
      "loss": 1.8892,
      "step": 134
    },
    {
      "epoch": 0.13503797943171517,
      "grad_norm": 0.7707583904266357,
      "learning_rate": 0.00019900037084217637,
      "loss": 1.787,
      "step": 135
    },
    {
      "epoch": 0.136038260760839,
      "grad_norm": 0.7389562129974365,
      "learning_rate": 0.00019898553185479303,
      "loss": 1.5854,
      "step": 136
    },
    {
      "epoch": 0.1370385420899628,
      "grad_norm": 0.7331375479698181,
      "learning_rate": 0.00019897058410021167,
      "loss": 1.997,
      "step": 137
    },
    {
      "epoch": 0.13803882341908663,
      "grad_norm": 0.7219388484954834,
      "learning_rate": 0.00019895552759485722,
      "loss": 1.8337,
      "step": 138
    },
    {
      "epoch": 0.13903910474821043,
      "grad_norm": 0.8535702228546143,
      "learning_rate": 0.00019894036235527395,
      "loss": 1.7818,
      "step": 139
    },
    {
      "epoch": 0.14003938607733424,
      "grad_norm": 0.7627841830253601,
      "learning_rate": 0.00019892508839812584,
      "loss": 1.8173,
      "step": 140
    },
    {
      "epoch": 0.14103966740645807,
      "grad_norm": 0.8397619724273682,
      "learning_rate": 0.00019890970574019617,
      "loss": 1.8735,
      "step": 141
    },
    {
      "epoch": 0.14203994873558187,
      "grad_norm": 0.8093482851982117,
      "learning_rate": 0.00019889421439838763,
      "loss": 1.9918,
      "step": 142
    },
    {
      "epoch": 0.1430402300647057,
      "grad_norm": 0.8853684067726135,
      "learning_rate": 0.00019887861438972246,
      "loss": 1.5825,
      "step": 143
    },
    {
      "epoch": 0.1440405113938295,
      "grad_norm": 0.7413788437843323,
      "learning_rate": 0.00019886290573134228,
      "loss": 1.9068,
      "step": 144
    },
    {
      "epoch": 0.14504079272295334,
      "grad_norm": 0.7924477458000183,
      "learning_rate": 0.000198847088440508,
      "loss": 1.8582,
      "step": 145
    },
    {
      "epoch": 0.14604107405207714,
      "grad_norm": 0.8679131865501404,
      "learning_rate": 0.0001988311625346,
      "loss": 1.7104,
      "step": 146
    },
    {
      "epoch": 0.14704135538120097,
      "grad_norm": 0.7480150461196899,
      "learning_rate": 0.00019881512803111796,
      "loss": 1.7288,
      "step": 147
    },
    {
      "epoch": 0.14804163671032478,
      "grad_norm": 0.8382390737533569,
      "learning_rate": 0.00019879898494768093,
      "loss": 1.8004,
      "step": 148
    },
    {
      "epoch": 0.1490419180394486,
      "grad_norm": 0.7360037565231323,
      "learning_rate": 0.00019878273330202717,
      "loss": 1.85,
      "step": 149
    },
    {
      "epoch": 0.1500421993685724,
      "grad_norm": 0.9644019603729248,
      "learning_rate": 0.00019876637311201433,
      "loss": 2.065,
      "step": 150
    },
    {
      "epoch": 0.15104248069769624,
      "grad_norm": 0.8116248250007629,
      "learning_rate": 0.00019874990439561934,
      "loss": 1.702,
      "step": 151
    },
    {
      "epoch": 0.15204276202682004,
      "grad_norm": 0.9301722645759583,
      "learning_rate": 0.0001987333271709383,
      "loss": 1.8089,
      "step": 152
    },
    {
      "epoch": 0.15304304335594385,
      "grad_norm": 0.7991555333137512,
      "learning_rate": 0.00019871664145618657,
      "loss": 1.8227,
      "step": 153
    },
    {
      "epoch": 0.15404332468506768,
      "grad_norm": 0.8676092028617859,
      "learning_rate": 0.00019869984726969878,
      "loss": 1.7253,
      "step": 154
    },
    {
      "epoch": 0.15504360601419148,
      "grad_norm": 0.8022972941398621,
      "learning_rate": 0.00019868294462992866,
      "loss": 1.8766,
      "step": 155
    },
    {
      "epoch": 0.1560438873433153,
      "grad_norm": 1.128886103630066,
      "learning_rate": 0.00019866593355544922,
      "loss": 2.0197,
      "step": 156
    },
    {
      "epoch": 0.15704416867243912,
      "grad_norm": 0.7420483827590942,
      "learning_rate": 0.00019864881406495246,
      "loss": 1.8825,
      "step": 157
    },
    {
      "epoch": 0.15804445000156295,
      "grad_norm": 0.7797536849975586,
      "learning_rate": 0.00019863158617724967,
      "loss": 1.8892,
      "step": 158
    },
    {
      "epoch": 0.15904473133068675,
      "grad_norm": 0.6859965324401855,
      "learning_rate": 0.00019861424991127115,
      "loss": 1.8424,
      "step": 159
    },
    {
      "epoch": 0.16004501265981058,
      "grad_norm": 0.8115108609199524,
      "learning_rate": 0.00019859680528606637,
      "loss": 1.8394,
      "step": 160
    },
    {
      "epoch": 0.16104529398893438,
      "grad_norm": 0.9756322503089905,
      "learning_rate": 0.00019857925232080373,
      "loss": 1.726,
      "step": 161
    },
    {
      "epoch": 0.16204557531805822,
      "grad_norm": 0.8894350528717041,
      "learning_rate": 0.00019856159103477086,
      "loss": 1.8893,
      "step": 162
    },
    {
      "epoch": 0.16304585664718202,
      "grad_norm": 0.8075819611549377,
      "learning_rate": 0.00019854382144737426,
      "loss": 1.6596,
      "step": 163
    },
    {
      "epoch": 0.16404613797630582,
      "grad_norm": 0.8861923813819885,
      "learning_rate": 0.00019852594357813952,
      "loss": 1.9352,
      "step": 164
    },
    {
      "epoch": 0.16504641930542965,
      "grad_norm": 0.8511936068534851,
      "learning_rate": 0.00019850795744671116,
      "loss": 1.9416,
      "step": 165
    },
    {
      "epoch": 0.16604670063455346,
      "grad_norm": 0.9425658583641052,
      "learning_rate": 0.0001984898630728527,
      "loss": 1.9081,
      "step": 166
    },
    {
      "epoch": 0.1670469819636773,
      "grad_norm": 0.7502055168151855,
      "learning_rate": 0.0001984716604764466,
      "loss": 1.703,
      "step": 167
    },
    {
      "epoch": 0.1680472632928011,
      "grad_norm": 0.9135978817939758,
      "learning_rate": 0.0001984533496774942,
      "loss": 1.7641,
      "step": 168
    },
    {
      "epoch": 0.16904754462192492,
      "grad_norm": 0.7768126726150513,
      "learning_rate": 0.0001984349306961158,
      "loss": 1.7053,
      "step": 169
    },
    {
      "epoch": 0.17004782595104873,
      "grad_norm": 0.8106538653373718,
      "learning_rate": 0.00019841640355255043,
      "loss": 1.8646,
      "step": 170
    },
    {
      "epoch": 0.17104810728017256,
      "grad_norm": 0.7872330546379089,
      "learning_rate": 0.00019839776826715614,
      "loss": 1.7814,
      "step": 171
    },
    {
      "epoch": 0.17204838860929636,
      "grad_norm": 0.869532585144043,
      "learning_rate": 0.00019837902486040978,
      "loss": 1.7812,
      "step": 172
    },
    {
      "epoch": 0.1730486699384202,
      "grad_norm": 1.015028715133667,
      "learning_rate": 0.0001983601733529069,
      "loss": 1.9432,
      "step": 173
    },
    {
      "epoch": 0.174048951267544,
      "grad_norm": 0.800183117389679,
      "learning_rate": 0.00019834121376536187,
      "loss": 1.758,
      "step": 174
    },
    {
      "epoch": 0.17504923259666783,
      "grad_norm": 0.7427104711532593,
      "learning_rate": 0.00019832214611860793,
      "loss": 1.6476,
      "step": 175
    },
    {
      "epoch": 0.17604951392579163,
      "grad_norm": 0.8289130926132202,
      "learning_rate": 0.00019830297043359692,
      "loss": 1.7702,
      "step": 176
    },
    {
      "epoch": 0.17704979525491543,
      "grad_norm": 0.8298771977424622,
      "learning_rate": 0.00019828368673139947,
      "loss": 1.7515,
      "step": 177
    },
    {
      "epoch": 0.17805007658403926,
      "grad_norm": 0.7602815628051758,
      "learning_rate": 0.0001982642950332049,
      "loss": 1.7545,
      "step": 178
    },
    {
      "epoch": 0.17905035791316307,
      "grad_norm": 0.8110321164131165,
      "learning_rate": 0.00019824479536032112,
      "loss": 2.2604,
      "step": 179
    },
    {
      "epoch": 0.1800506392422869,
      "grad_norm": 0.882273256778717,
      "learning_rate": 0.0001982251877341748,
      "loss": 1.8133,
      "step": 180
    },
    {
      "epoch": 0.1810509205714107,
      "grad_norm": 0.9015639424324036,
      "learning_rate": 0.00019820547217631117,
      "loss": 1.7282,
      "step": 181
    },
    {
      "epoch": 0.18205120190053453,
      "grad_norm": 0.9231659173965454,
      "learning_rate": 0.00019818564870839405,
      "loss": 1.9094,
      "step": 182
    },
    {
      "epoch": 0.18305148322965833,
      "grad_norm": 0.8110967874526978,
      "learning_rate": 0.00019816571735220583,
      "loss": 1.886,
      "step": 183
    },
    {
      "epoch": 0.18405176455878217,
      "grad_norm": 0.7670036554336548,
      "learning_rate": 0.00019814567812964748,
      "loss": 1.9895,
      "step": 184
    },
    {
      "epoch": 0.18505204588790597,
      "grad_norm": 0.7955975532531738,
      "learning_rate": 0.00019812553106273847,
      "loss": 1.8127,
      "step": 185
    },
    {
      "epoch": 0.1860523272170298,
      "grad_norm": 0.8790062665939331,
      "learning_rate": 0.00019810527617361681,
      "loss": 1.8899,
      "step": 186
    },
    {
      "epoch": 0.1870526085461536,
      "grad_norm": 0.8818586468696594,
      "learning_rate": 0.00019808491348453894,
      "loss": 1.7707,
      "step": 187
    },
    {
      "epoch": 0.18805288987527743,
      "grad_norm": 0.746442437171936,
      "learning_rate": 0.00019806444301787978,
      "loss": 1.7281,
      "step": 188
    },
    {
      "epoch": 0.18905317120440124,
      "grad_norm": 0.7786905169487,
      "learning_rate": 0.0001980438647961327,
      "loss": 1.7317,
      "step": 189
    },
    {
      "epoch": 0.19005345253352504,
      "grad_norm": 0.9338862299919128,
      "learning_rate": 0.00019802317884190935,
      "loss": 1.9548,
      "step": 190
    },
    {
      "epoch": 0.19105373386264887,
      "grad_norm": 0.7416581511497498,
      "learning_rate": 0.00019800238517793996,
      "loss": 1.8601,
      "step": 191
    },
    {
      "epoch": 0.19205401519177268,
      "grad_norm": 0.6782898902893066,
      "learning_rate": 0.00019798148382707296,
      "loss": 1.8477,
      "step": 192
    },
    {
      "epoch": 0.1930542965208965,
      "grad_norm": 0.7389237880706787,
      "learning_rate": 0.00019796047481227515,
      "loss": 1.7749,
      "step": 193
    },
    {
      "epoch": 0.1940545778500203,
      "grad_norm": 0.9711095094680786,
      "learning_rate": 0.00019793935815663163,
      "loss": 2.0899,
      "step": 194
    },
    {
      "epoch": 0.19505485917914414,
      "grad_norm": 0.7949391007423401,
      "learning_rate": 0.00019791813388334581,
      "loss": 1.8778,
      "step": 195
    },
    {
      "epoch": 0.19605514050826794,
      "grad_norm": 0.8871057033538818,
      "learning_rate": 0.00019789680201573933,
      "loss": 1.7511,
      "step": 196
    },
    {
      "epoch": 0.19705542183739178,
      "grad_norm": 0.8664624094963074,
      "learning_rate": 0.00019787536257725202,
      "loss": 1.7232,
      "step": 197
    },
    {
      "epoch": 0.19805570316651558,
      "grad_norm": 0.871658980846405,
      "learning_rate": 0.00019785381559144196,
      "loss": 1.7987,
      "step": 198
    },
    {
      "epoch": 0.1990559844956394,
      "grad_norm": 0.7748361229896545,
      "learning_rate": 0.00019783216108198542,
      "loss": 1.9239,
      "step": 199
    },
    {
      "epoch": 0.2000562658247632,
      "grad_norm": 0.9393408298492432,
      "learning_rate": 0.00019781039907267677,
      "loss": 2.0936,
      "step": 200
    },
    {
      "epoch": 0.20105654715388704,
      "grad_norm": 0.8519601225852966,
      "learning_rate": 0.00019778852958742853,
      "loss": 1.9108,
      "step": 201
    },
    {
      "epoch": 0.20205682848301085,
      "grad_norm": 0.8464863300323486,
      "learning_rate": 0.00019776655265027127,
      "loss": 1.897,
      "step": 202
    },
    {
      "epoch": 0.20305710981213465,
      "grad_norm": 0.8933351635932922,
      "learning_rate": 0.00019774446828535371,
      "loss": 1.8204,
      "step": 203
    },
    {
      "epoch": 0.20405739114125848,
      "grad_norm": 0.8305785059928894,
      "learning_rate": 0.00019772227651694256,
      "loss": 1.9135,
      "step": 204
    },
    {
      "epoch": 0.20505767247038229,
      "grad_norm": 0.8117037415504456,
      "learning_rate": 0.00019769997736942258,
      "loss": 1.7585,
      "step": 205
    },
    {
      "epoch": 0.20605795379950612,
      "grad_norm": 0.7570348381996155,
      "learning_rate": 0.00019767757086729647,
      "loss": 1.8373,
      "step": 206
    },
    {
      "epoch": 0.20705823512862992,
      "grad_norm": 0.9291234016418457,
      "learning_rate": 0.00019765505703518496,
      "loss": 1.7774,
      "step": 207
    },
    {
      "epoch": 0.20805851645775375,
      "grad_norm": 0.8211004137992859,
      "learning_rate": 0.00019763243589782662,
      "loss": 1.8766,
      "step": 208
    },
    {
      "epoch": 0.20905879778687755,
      "grad_norm": 0.6625431180000305,
      "learning_rate": 0.00019760970748007803,
      "loss": 1.628,
      "step": 209
    },
    {
      "epoch": 0.21005907911600138,
      "grad_norm": 0.7974782586097717,
      "learning_rate": 0.0001975868718069136,
      "loss": 1.6896,
      "step": 210
    },
    {
      "epoch": 0.2110593604451252,
      "grad_norm": 0.8364912867546082,
      "learning_rate": 0.00019756392890342563,
      "loss": 1.7492,
      "step": 211
    },
    {
      "epoch": 0.21205964177424902,
      "grad_norm": 0.8730652332305908,
      "learning_rate": 0.00019754087879482422,
      "loss": 1.8295,
      "step": 212
    },
    {
      "epoch": 0.21305992310337282,
      "grad_norm": 0.7532863020896912,
      "learning_rate": 0.00019751772150643722,
      "loss": 1.8309,
      "step": 213
    },
    {
      "epoch": 0.21406020443249665,
      "grad_norm": 0.7375178933143616,
      "learning_rate": 0.00019749445706371038,
      "loss": 1.7854,
      "step": 214
    },
    {
      "epoch": 0.21506048576162046,
      "grad_norm": 0.7524377703666687,
      "learning_rate": 0.00019747108549220702,
      "loss": 1.7683,
      "step": 215
    },
    {
      "epoch": 0.21606076709074426,
      "grad_norm": 0.7331809997558594,
      "learning_rate": 0.00019744760681760832,
      "loss": 1.7103,
      "step": 216
    },
    {
      "epoch": 0.2170610484198681,
      "grad_norm": 0.8083691596984863,
      "learning_rate": 0.00019742402106571314,
      "loss": 1.674,
      "step": 217
    },
    {
      "epoch": 0.2180613297489919,
      "grad_norm": 0.8524570465087891,
      "learning_rate": 0.00019740032826243788,
      "loss": 1.7227,
      "step": 218
    },
    {
      "epoch": 0.21906161107811573,
      "grad_norm": 0.7676658630371094,
      "learning_rate": 0.0001973765284338167,
      "loss": 1.8561,
      "step": 219
    },
    {
      "epoch": 0.22006189240723953,
      "grad_norm": 0.7858710289001465,
      "learning_rate": 0.00019735262160600127,
      "loss": 1.7796,
      "step": 220
    },
    {
      "epoch": 0.22106217373636336,
      "grad_norm": 0.7587497234344482,
      "learning_rate": 0.00019732860780526088,
      "loss": 1.9271,
      "step": 221
    },
    {
      "epoch": 0.22206245506548716,
      "grad_norm": 0.8084688186645508,
      "learning_rate": 0.00019730448705798239,
      "loss": 1.8176,
      "step": 222
    },
    {
      "epoch": 0.223062736394611,
      "grad_norm": 0.6736906170845032,
      "learning_rate": 0.00019728025939067008,
      "loss": 1.6288,
      "step": 223
    },
    {
      "epoch": 0.2240630177237348,
      "grad_norm": 0.7483925819396973,
      "learning_rate": 0.00019725592482994583,
      "loss": 1.8363,
      "step": 224
    },
    {
      "epoch": 0.22506329905285863,
      "grad_norm": 1.7995796203613281,
      "learning_rate": 0.00019723148340254892,
      "loss": 1.9072,
      "step": 225
    },
    {
      "epoch": 0.22606358038198243,
      "grad_norm": 0.8028881549835205,
      "learning_rate": 0.00019720693513533598,
      "loss": 1.9021,
      "step": 226
    },
    {
      "epoch": 0.22706386171110624,
      "grad_norm": 0.9853909015655518,
      "learning_rate": 0.00019718228005528122,
      "loss": 2.0159,
      "step": 227
    },
    {
      "epoch": 0.22806414304023007,
      "grad_norm": 0.7784947156906128,
      "learning_rate": 0.00019715751818947603,
      "loss": 1.7816,
      "step": 228
    },
    {
      "epoch": 0.22906442436935387,
      "grad_norm": 0.7447614669799805,
      "learning_rate": 0.0001971326495651293,
      "loss": 1.654,
      "step": 229
    },
    {
      "epoch": 0.2300647056984777,
      "grad_norm": 0.8673064112663269,
      "learning_rate": 0.00019710767420956705,
      "loss": 2.0049,
      "step": 230
    },
    {
      "epoch": 0.2310649870276015,
      "grad_norm": 0.8207747936248779,
      "learning_rate": 0.0001970825921502328,
      "loss": 1.9388,
      "step": 231
    },
    {
      "epoch": 0.23206526835672533,
      "grad_norm": 0.742266058921814,
      "learning_rate": 0.0001970574034146871,
      "loss": 1.7658,
      "step": 232
    },
    {
      "epoch": 0.23306554968584914,
      "grad_norm": 0.9097973704338074,
      "learning_rate": 0.00019703210803060782,
      "loss": 1.8023,
      "step": 233
    },
    {
      "epoch": 0.23406583101497297,
      "grad_norm": 0.7512438297271729,
      "learning_rate": 0.00019700670602579008,
      "loss": 1.8551,
      "step": 234
    },
    {
      "epoch": 0.23506611234409677,
      "grad_norm": 0.8303943872451782,
      "learning_rate": 0.00019698119742814606,
      "loss": 1.7723,
      "step": 235
    },
    {
      "epoch": 0.2360663936732206,
      "grad_norm": 0.9195139408111572,
      "learning_rate": 0.00019695558226570507,
      "loss": 1.6426,
      "step": 236
    },
    {
      "epoch": 0.2370666750023444,
      "grad_norm": 0.7734714150428772,
      "learning_rate": 0.00019692986056661356,
      "loss": 1.7798,
      "step": 237
    },
    {
      "epoch": 0.23806695633146824,
      "grad_norm": 0.8759648203849792,
      "learning_rate": 0.00019690403235913504,
      "loss": 1.6465,
      "step": 238
    },
    {
      "epoch": 0.23906723766059204,
      "grad_norm": 0.7688003778457642,
      "learning_rate": 0.00019687809767165,
      "loss": 2.0092,
      "step": 239
    },
    {
      "epoch": 0.24006751898971584,
      "grad_norm": 0.7398790121078491,
      "learning_rate": 0.000196852056532656,
      "loss": 1.8176,
      "step": 240
    },
    {
      "epoch": 0.24106780031883968,
      "grad_norm": 0.8921257853507996,
      "learning_rate": 0.00019682590897076752,
      "loss": 1.7387,
      "step": 241
    },
    {
      "epoch": 0.24206808164796348,
      "grad_norm": 0.7939002513885498,
      "learning_rate": 0.00019679965501471608,
      "loss": 1.9417,
      "step": 242
    },
    {
      "epoch": 0.2430683629770873,
      "grad_norm": 0.7798025608062744,
      "learning_rate": 0.0001967732946933499,
      "loss": 1.7134,
      "step": 243
    },
    {
      "epoch": 0.2440686443062111,
      "grad_norm": 0.8007254600524902,
      "learning_rate": 0.00019674682803563428,
      "loss": 1.7387,
      "step": 244
    },
    {
      "epoch": 0.24506892563533494,
      "grad_norm": 0.6257696151733398,
      "learning_rate": 0.00019672025507065131,
      "loss": 1.767,
      "step": 245
    },
    {
      "epoch": 0.24606920696445875,
      "grad_norm": 0.7942785620689392,
      "learning_rate": 0.00019669357582759983,
      "loss": 1.8801,
      "step": 246
    },
    {
      "epoch": 0.24706948829358258,
      "grad_norm": 0.7933829426765442,
      "learning_rate": 0.00019666679033579552,
      "loss": 1.9711,
      "step": 247
    },
    {
      "epoch": 0.24806976962270638,
      "grad_norm": 0.7489326596260071,
      "learning_rate": 0.00019663989862467082,
      "loss": 1.8038,
      "step": 248
    },
    {
      "epoch": 0.2490700509518302,
      "grad_norm": 0.7279101014137268,
      "learning_rate": 0.00019661290072377482,
      "loss": 1.66,
      "step": 249
    },
    {
      "epoch": 0.25007033228095404,
      "grad_norm": 0.6823874115943909,
      "learning_rate": 0.00019658579666277334,
      "loss": 1.8064,
      "step": 250
    },
    {
      "epoch": 0.2510706136100778,
      "grad_norm": 0.6561273336410522,
      "learning_rate": 0.0001965585864714488,
      "loss": 1.6874,
      "step": 251
    },
    {
      "epoch": 0.25207089493920165,
      "grad_norm": 0.6457573175430298,
      "learning_rate": 0.00019653127017970034,
      "loss": 1.4587,
      "step": 252
    },
    {
      "epoch": 0.2530711762683255,
      "grad_norm": 0.7649476528167725,
      "learning_rate": 0.0001965038478175436,
      "loss": 1.9811,
      "step": 253
    },
    {
      "epoch": 0.25407145759744926,
      "grad_norm": 0.8786829710006714,
      "learning_rate": 0.00019647631941511082,
      "loss": 1.8629,
      "step": 254
    },
    {
      "epoch": 0.2550717389265731,
      "grad_norm": 0.7038159966468811,
      "learning_rate": 0.0001964486850026507,
      "loss": 1.6885,
      "step": 255
    },
    {
      "epoch": 0.2560720202556969,
      "grad_norm": 0.7255909442901611,
      "learning_rate": 0.00019642094461052852,
      "loss": 1.7335,
      "step": 256
    },
    {
      "epoch": 0.25707230158482075,
      "grad_norm": 0.7780727744102478,
      "learning_rate": 0.00019639309826922585,
      "loss": 1.899,
      "step": 257
    },
    {
      "epoch": 0.2580725829139445,
      "grad_norm": 0.8533650040626526,
      "learning_rate": 0.0001963651460093409,
      "loss": 1.7711,
      "step": 258
    },
    {
      "epoch": 0.25907286424306836,
      "grad_norm": 0.6440068483352661,
      "learning_rate": 0.00019633708786158806,
      "loss": 1.6685,
      "step": 259
    },
    {
      "epoch": 0.2600731455721922,
      "grad_norm": 0.6873877048492432,
      "learning_rate": 0.00019630892385679818,
      "loss": 1.7502,
      "step": 260
    },
    {
      "epoch": 0.261073426901316,
      "grad_norm": 0.7100672721862793,
      "learning_rate": 0.00019628065402591845,
      "loss": 1.7789,
      "step": 261
    },
    {
      "epoch": 0.2620737082304398,
      "grad_norm": 0.8447420001029968,
      "learning_rate": 0.00019625227840001225,
      "loss": 1.8577,
      "step": 262
    },
    {
      "epoch": 0.2630739895595636,
      "grad_norm": 0.767888605594635,
      "learning_rate": 0.0001962237970102593,
      "loss": 1.5936,
      "step": 263
    },
    {
      "epoch": 0.26407427088868746,
      "grad_norm": 0.6955805420875549,
      "learning_rate": 0.0001961952098879555,
      "loss": 1.7733,
      "step": 264
    },
    {
      "epoch": 0.26507455221781123,
      "grad_norm": 0.777740478515625,
      "learning_rate": 0.00019616651706451287,
      "loss": 1.6027,
      "step": 265
    },
    {
      "epoch": 0.26607483354693506,
      "grad_norm": 0.7691099047660828,
      "learning_rate": 0.0001961377185714597,
      "loss": 1.7457,
      "step": 266
    },
    {
      "epoch": 0.2670751148760589,
      "grad_norm": 0.6778420805931091,
      "learning_rate": 0.0001961088144404403,
      "loss": 1.7704,
      "step": 267
    },
    {
      "epoch": 0.2680753962051827,
      "grad_norm": 0.7943267226219177,
      "learning_rate": 0.00019607980470321505,
      "loss": 1.9775,
      "step": 268
    },
    {
      "epoch": 0.2690756775343065,
      "grad_norm": 0.6660135388374329,
      "learning_rate": 0.00019605068939166045,
      "loss": 1.6556,
      "step": 269
    },
    {
      "epoch": 0.27007595886343033,
      "grad_norm": 0.8664935827255249,
      "learning_rate": 0.00019602146853776894,
      "loss": 2.03,
      "step": 270
    },
    {
      "epoch": 0.27107624019255416,
      "grad_norm": 0.7783074975013733,
      "learning_rate": 0.000195992142173649,
      "loss": 1.7426,
      "step": 271
    },
    {
      "epoch": 0.272076521521678,
      "grad_norm": 0.7470223903656006,
      "learning_rate": 0.0001959627103315249,
      "loss": 1.7284,
      "step": 272
    },
    {
      "epoch": 0.27307680285080177,
      "grad_norm": 0.7284931540489197,
      "learning_rate": 0.00019593317304373705,
      "loss": 1.6977,
      "step": 273
    },
    {
      "epoch": 0.2740770841799256,
      "grad_norm": 0.7201762795448303,
      "learning_rate": 0.00019590353034274144,
      "loss": 1.7184,
      "step": 274
    },
    {
      "epoch": 0.27507736550904943,
      "grad_norm": 0.6756151914596558,
      "learning_rate": 0.00019587378226111014,
      "loss": 1.7276,
      "step": 275
    },
    {
      "epoch": 0.27607764683817326,
      "grad_norm": 0.6784201860427856,
      "learning_rate": 0.00019584392883153088,
      "loss": 1.642,
      "step": 276
    },
    {
      "epoch": 0.27707792816729704,
      "grad_norm": 0.7387176752090454,
      "learning_rate": 0.00019581397008680717,
      "loss": 1.7911,
      "step": 277
    },
    {
      "epoch": 0.27807820949642087,
      "grad_norm": 0.9367021918296814,
      "learning_rate": 0.00019578390605985826,
      "loss": 2.0034,
      "step": 278
    },
    {
      "epoch": 0.2790784908255447,
      "grad_norm": 0.803698718547821,
      "learning_rate": 0.00019575373678371909,
      "loss": 1.7907,
      "step": 279
    },
    {
      "epoch": 0.2800787721546685,
      "grad_norm": 0.7324479818344116,
      "learning_rate": 0.00019572346229154025,
      "loss": 1.5539,
      "step": 280
    },
    {
      "epoch": 0.2810790534837923,
      "grad_norm": 0.7107382416725159,
      "learning_rate": 0.00019569308261658787,
      "loss": 1.838,
      "step": 281
    },
    {
      "epoch": 0.28207933481291614,
      "grad_norm": 0.8698626756668091,
      "learning_rate": 0.00019566259779224378,
      "loss": 1.7433,
      "step": 282
    },
    {
      "epoch": 0.28307961614203997,
      "grad_norm": 0.7804028391838074,
      "learning_rate": 0.00019563200785200526,
      "loss": 1.7161,
      "step": 283
    },
    {
      "epoch": 0.28407989747116374,
      "grad_norm": 0.8762909173965454,
      "learning_rate": 0.00019560131282948516,
      "loss": 1.8031,
      "step": 284
    },
    {
      "epoch": 0.2850801788002876,
      "grad_norm": 0.8252436518669128,
      "learning_rate": 0.0001955705127584117,
      "loss": 1.6434,
      "step": 285
    },
    {
      "epoch": 0.2860804601294114,
      "grad_norm": 0.8220797181129456,
      "learning_rate": 0.00019553960767262863,
      "loss": 1.8522,
      "step": 286
    },
    {
      "epoch": 0.28708074145853524,
      "grad_norm": 0.7883003950119019,
      "learning_rate": 0.00019550859760609503,
      "loss": 1.8245,
      "step": 287
    },
    {
      "epoch": 0.288081022787659,
      "grad_norm": 0.9208703637123108,
      "learning_rate": 0.00019547748259288536,
      "loss": 1.8877,
      "step": 288
    },
    {
      "epoch": 0.28908130411678284,
      "grad_norm": 0.8452202677726746,
      "learning_rate": 0.0001954462626671894,
      "loss": 1.554,
      "step": 289
    },
    {
      "epoch": 0.2900815854459067,
      "grad_norm": 0.82865971326828,
      "learning_rate": 0.0001954149378633122,
      "loss": 1.655,
      "step": 290
    },
    {
      "epoch": 0.29108186677503045,
      "grad_norm": 0.7871205806732178,
      "learning_rate": 0.00019538350821567404,
      "loss": 1.621,
      "step": 291
    },
    {
      "epoch": 0.2920821481041543,
      "grad_norm": 0.8288848996162415,
      "learning_rate": 0.00019535197375881045,
      "loss": 1.9277,
      "step": 292
    },
    {
      "epoch": 0.2930824294332781,
      "grad_norm": 0.7275516986846924,
      "learning_rate": 0.00019532033452737205,
      "loss": 1.7949,
      "step": 293
    },
    {
      "epoch": 0.29408271076240194,
      "grad_norm": 0.7424570322036743,
      "learning_rate": 0.00019528859055612468,
      "loss": 1.6407,
      "step": 294
    },
    {
      "epoch": 0.2950829920915257,
      "grad_norm": 0.7031363248825073,
      "learning_rate": 0.0001952567418799492,
      "loss": 1.8793,
      "step": 295
    },
    {
      "epoch": 0.29608327342064955,
      "grad_norm": 0.7190185189247131,
      "learning_rate": 0.00019522478853384155,
      "loss": 1.6759,
      "step": 296
    },
    {
      "epoch": 0.2970835547497734,
      "grad_norm": 0.7270736694335938,
      "learning_rate": 0.00019519273055291266,
      "loss": 1.6351,
      "step": 297
    },
    {
      "epoch": 0.2980838360788972,
      "grad_norm": 0.8894152641296387,
      "learning_rate": 0.00019516056797238846,
      "loss": 1.7908,
      "step": 298
    },
    {
      "epoch": 0.299084117408021,
      "grad_norm": 0.9089106321334839,
      "learning_rate": 0.00019512830082760987,
      "loss": 1.6018,
      "step": 299
    },
    {
      "epoch": 0.3000843987371448,
      "grad_norm": 0.8772429823875427,
      "learning_rate": 0.00019509592915403255,
      "loss": 1.8474,
      "step": 300
    },
    {
      "epoch": 0.30108468006626865,
      "grad_norm": 0.8244933485984802,
      "learning_rate": 0.00019506345298722717,
      "loss": 1.4324,
      "step": 301
    },
    {
      "epoch": 0.3020849613953925,
      "grad_norm": 0.7283012866973877,
      "learning_rate": 0.00019503087236287913,
      "loss": 1.5115,
      "step": 302
    },
    {
      "epoch": 0.30308524272451626,
      "grad_norm": 0.7721333503723145,
      "learning_rate": 0.00019499818731678873,
      "loss": 1.6728,
      "step": 303
    },
    {
      "epoch": 0.3040855240536401,
      "grad_norm": 0.7579306960105896,
      "learning_rate": 0.00019496539788487082,
      "loss": 1.5927,
      "step": 304
    },
    {
      "epoch": 0.3050858053827639,
      "grad_norm": 0.9054704308509827,
      "learning_rate": 0.0001949325041031551,
      "loss": 1.9027,
      "step": 305
    },
    {
      "epoch": 0.3060860867118877,
      "grad_norm": 0.7023262977600098,
      "learning_rate": 0.0001948995060077859,
      "loss": 1.7705,
      "step": 306
    },
    {
      "epoch": 0.3070863680410115,
      "grad_norm": 0.7942065000534058,
      "learning_rate": 0.0001948664036350221,
      "loss": 1.8269,
      "step": 307
    },
    {
      "epoch": 0.30808664937013536,
      "grad_norm": 0.9305068850517273,
      "learning_rate": 0.00019483319702123732,
      "loss": 1.8247,
      "step": 308
    },
    {
      "epoch": 0.3090869306992592,
      "grad_norm": 0.814664900302887,
      "learning_rate": 0.00019479988620291956,
      "loss": 1.9179,
      "step": 309
    },
    {
      "epoch": 0.31008721202838296,
      "grad_norm": 0.6418014764785767,
      "learning_rate": 0.00019476647121667137,
      "loss": 1.5011,
      "step": 310
    },
    {
      "epoch": 0.3110874933575068,
      "grad_norm": 0.7911447882652283,
      "learning_rate": 0.00019473295209920983,
      "loss": 1.857,
      "step": 311
    },
    {
      "epoch": 0.3120877746866306,
      "grad_norm": 0.7792949676513672,
      "learning_rate": 0.00019469932888736632,
      "loss": 1.7279,
      "step": 312
    },
    {
      "epoch": 0.31308805601575446,
      "grad_norm": 0.7579171657562256,
      "learning_rate": 0.00019466560161808674,
      "loss": 1.6902,
      "step": 313
    },
    {
      "epoch": 0.31408833734487823,
      "grad_norm": 0.7052372694015503,
      "learning_rate": 0.00019463177032843124,
      "loss": 1.7302,
      "step": 314
    },
    {
      "epoch": 0.31508861867400206,
      "grad_norm": 0.7188624143600464,
      "learning_rate": 0.00019459783505557424,
      "loss": 1.7338,
      "step": 315
    },
    {
      "epoch": 0.3160889000031259,
      "grad_norm": 0.6057978272438049,
      "learning_rate": 0.00019456379583680452,
      "loss": 1.6123,
      "step": 316
    },
    {
      "epoch": 0.31708918133224967,
      "grad_norm": 0.8339365720748901,
      "learning_rate": 0.000194529652709525,
      "loss": 1.9765,
      "step": 317
    },
    {
      "epoch": 0.3180894626613735,
      "grad_norm": 0.8524260520935059,
      "learning_rate": 0.00019449540571125286,
      "loss": 1.6803,
      "step": 318
    },
    {
      "epoch": 0.31908974399049733,
      "grad_norm": 0.7035975456237793,
      "learning_rate": 0.00019446105487961926,
      "loss": 1.5792,
      "step": 319
    },
    {
      "epoch": 0.32009002531962116,
      "grad_norm": 0.7894249558448792,
      "learning_rate": 0.0001944266002523696,
      "loss": 1.6326,
      "step": 320
    },
    {
      "epoch": 0.32109030664874494,
      "grad_norm": 0.7716989517211914,
      "learning_rate": 0.0001943920418673633,
      "loss": 1.6871,
      "step": 321
    },
    {
      "epoch": 0.32209058797786877,
      "grad_norm": 0.7914933562278748,
      "learning_rate": 0.00019435737976257377,
      "loss": 1.7148,
      "step": 322
    },
    {
      "epoch": 0.3230908693069926,
      "grad_norm": 0.7113205790519714,
      "learning_rate": 0.00019432261397608834,
      "loss": 1.5236,
      "step": 323
    },
    {
      "epoch": 0.32409115063611643,
      "grad_norm": 0.8609917163848877,
      "learning_rate": 0.00019428774454610843,
      "loss": 1.8101,
      "step": 324
    },
    {
      "epoch": 0.3250914319652402,
      "grad_norm": 0.7319685220718384,
      "learning_rate": 0.00019425277151094913,
      "loss": 1.7712,
      "step": 325
    },
    {
      "epoch": 0.32609171329436404,
      "grad_norm": 0.6478747725486755,
      "learning_rate": 0.00019421769490903957,
      "loss": 1.8535,
      "step": 326
    },
    {
      "epoch": 0.32709199462348787,
      "grad_norm": 0.7025763392448425,
      "learning_rate": 0.0001941825147789225,
      "loss": 1.9213,
      "step": 327
    },
    {
      "epoch": 0.32809227595261165,
      "grad_norm": 0.7595239877700806,
      "learning_rate": 0.00019414723115925456,
      "loss": 1.7449,
      "step": 328
    },
    {
      "epoch": 0.3290925572817355,
      "grad_norm": 0.7728105783462524,
      "learning_rate": 0.0001941118440888061,
      "loss": 1.8821,
      "step": 329
    },
    {
      "epoch": 0.3300928386108593,
      "grad_norm": 0.7430977821350098,
      "learning_rate": 0.0001940763536064611,
      "loss": 1.6904,
      "step": 330
    },
    {
      "epoch": 0.33109311993998314,
      "grad_norm": 0.7909367680549622,
      "learning_rate": 0.00019404075975121716,
      "loss": 1.7899,
      "step": 331
    },
    {
      "epoch": 0.3320934012691069,
      "grad_norm": 0.7561226487159729,
      "learning_rate": 0.0001940050625621855,
      "loss": 1.7746,
      "step": 332
    },
    {
      "epoch": 0.33309368259823074,
      "grad_norm": 0.7602452635765076,
      "learning_rate": 0.00019396926207859084,
      "loss": 1.7909,
      "step": 333
    },
    {
      "epoch": 0.3340939639273546,
      "grad_norm": 0.8194379806518555,
      "learning_rate": 0.0001939333583397715,
      "loss": 1.7039,
      "step": 334
    },
    {
      "epoch": 0.3350942452564784,
      "grad_norm": 0.7036342024803162,
      "learning_rate": 0.00019389735138517915,
      "loss": 1.6663,
      "step": 335
    },
    {
      "epoch": 0.3360945265856022,
      "grad_norm": 0.8429521918296814,
      "learning_rate": 0.00019386124125437895,
      "loss": 1.589,
      "step": 336
    },
    {
      "epoch": 0.337094807914726,
      "grad_norm": 0.7271071076393127,
      "learning_rate": 0.00019382502798704935,
      "loss": 1.646,
      "step": 337
    },
    {
      "epoch": 0.33809508924384984,
      "grad_norm": 0.7862086892127991,
      "learning_rate": 0.00019378871162298227,
      "loss": 1.6085,
      "step": 338
    },
    {
      "epoch": 0.3390953705729737,
      "grad_norm": 0.676815390586853,
      "learning_rate": 0.00019375229220208276,
      "loss": 1.7335,
      "step": 339
    },
    {
      "epoch": 0.34009565190209745,
      "grad_norm": 0.8916042447090149,
      "learning_rate": 0.00019371576976436917,
      "loss": 1.7914,
      "step": 340
    },
    {
      "epoch": 0.3410959332312213,
      "grad_norm": 0.7913751006126404,
      "learning_rate": 0.00019367914434997312,
      "loss": 1.6031,
      "step": 341
    },
    {
      "epoch": 0.3420962145603451,
      "grad_norm": 0.7409866452217102,
      "learning_rate": 0.00019364241599913924,
      "loss": 1.6525,
      "step": 342
    },
    {
      "epoch": 0.3430964958894689,
      "grad_norm": 0.7472705841064453,
      "learning_rate": 0.0001936055847522254,
      "loss": 1.6716,
      "step": 343
    },
    {
      "epoch": 0.3440967772185927,
      "grad_norm": 0.7030773758888245,
      "learning_rate": 0.00019356865064970244,
      "loss": 1.7134,
      "step": 344
    },
    {
      "epoch": 0.34509705854771655,
      "grad_norm": 0.6609564423561096,
      "learning_rate": 0.0001935316137321543,
      "loss": 1.7127,
      "step": 345
    },
    {
      "epoch": 0.3460973398768404,
      "grad_norm": 0.7811393141746521,
      "learning_rate": 0.00019349447404027782,
      "loss": 1.75,
      "step": 346
    },
    {
      "epoch": 0.34709762120596416,
      "grad_norm": 0.6980521082878113,
      "learning_rate": 0.00019345723161488283,
      "loss": 1.82,
      "step": 347
    },
    {
      "epoch": 0.348097902535088,
      "grad_norm": 0.749796986579895,
      "learning_rate": 0.000193419886496892,
      "loss": 1.9755,
      "step": 348
    },
    {
      "epoch": 0.3490981838642118,
      "grad_norm": 0.9486667513847351,
      "learning_rate": 0.00019338243872734086,
      "loss": 1.7047,
      "step": 349
    },
    {
      "epoch": 0.35009846519333565,
      "grad_norm": 0.8086081147193909,
      "learning_rate": 0.00019334488834737775,
      "loss": 1.661,
      "step": 350
    },
    {
      "epoch": 0.3510987465224594,
      "grad_norm": 0.700549840927124,
      "learning_rate": 0.00019330723539826375,
      "loss": 1.8696,
      "step": 351
    },
    {
      "epoch": 0.35209902785158326,
      "grad_norm": 0.7465476393699646,
      "learning_rate": 0.00019326947992137262,
      "loss": 1.5444,
      "step": 352
    },
    {
      "epoch": 0.3530993091807071,
      "grad_norm": 0.7370999455451965,
      "learning_rate": 0.00019323162195819082,
      "loss": 1.8805,
      "step": 353
    },
    {
      "epoch": 0.35409959050983086,
      "grad_norm": 0.719359278678894,
      "learning_rate": 0.0001931936615503174,
      "loss": 1.8022,
      "step": 354
    },
    {
      "epoch": 0.3550998718389547,
      "grad_norm": 0.7301434278488159,
      "learning_rate": 0.000193155598739464,
      "loss": 1.6984,
      "step": 355
    },
    {
      "epoch": 0.3561001531680785,
      "grad_norm": 0.7191399335861206,
      "learning_rate": 0.0001931174335674547,
      "loss": 1.7229,
      "step": 356
    },
    {
      "epoch": 0.35710043449720236,
      "grad_norm": 0.7471932768821716,
      "learning_rate": 0.0001930791660762262,
      "loss": 1.7408,
      "step": 357
    },
    {
      "epoch": 0.35810071582632613,
      "grad_norm": 0.8197934031486511,
      "learning_rate": 0.00019304079630782752,
      "loss": 1.6938,
      "step": 358
    },
    {
      "epoch": 0.35910099715544996,
      "grad_norm": 0.7408166527748108,
      "learning_rate": 0.0001930023243044201,
      "loss": 1.7798,
      "step": 359
    },
    {
      "epoch": 0.3601012784845738,
      "grad_norm": 0.7525373101234436,
      "learning_rate": 0.00019296375010827773,
      "loss": 1.711,
      "step": 360
    },
    {
      "epoch": 0.3611015598136976,
      "grad_norm": 0.6712046265602112,
      "learning_rate": 0.00019292507376178643,
      "loss": 1.8157,
      "step": 361
    },
    {
      "epoch": 0.3621018411428214,
      "grad_norm": 0.6712916493415833,
      "learning_rate": 0.00019288629530744454,
      "loss": 1.8707,
      "step": 362
    },
    {
      "epoch": 0.36310212247194523,
      "grad_norm": 0.6127772331237793,
      "learning_rate": 0.0001928474147878626,
      "loss": 1.4743,
      "step": 363
    },
    {
      "epoch": 0.36410240380106906,
      "grad_norm": 0.910310685634613,
      "learning_rate": 0.0001928084322457632,
      "loss": 1.7956,
      "step": 364
    },
    {
      "epoch": 0.3651026851301929,
      "grad_norm": 0.6267688870429993,
      "learning_rate": 0.00019276934772398114,
      "loss": 1.4664,
      "step": 365
    },
    {
      "epoch": 0.36610296645931667,
      "grad_norm": 0.8317943811416626,
      "learning_rate": 0.00019273016126546323,
      "loss": 1.853,
      "step": 366
    },
    {
      "epoch": 0.3671032477884405,
      "grad_norm": 0.7581344842910767,
      "learning_rate": 0.00019269087291326833,
      "loss": 1.9236,
      "step": 367
    },
    {
      "epoch": 0.36810352911756433,
      "grad_norm": 0.9311390519142151,
      "learning_rate": 0.00019265148271056722,
      "loss": 1.7019,
      "step": 368
    },
    {
      "epoch": 0.3691038104466881,
      "grad_norm": 0.9513958096504211,
      "learning_rate": 0.0001926119907006426,
      "loss": 1.7617,
      "step": 369
    },
    {
      "epoch": 0.37010409177581194,
      "grad_norm": 0.7407613396644592,
      "learning_rate": 0.00019257239692688907,
      "loss": 2.1057,
      "step": 370
    },
    {
      "epoch": 0.37110437310493577,
      "grad_norm": 0.7530227899551392,
      "learning_rate": 0.00019253270143281296,
      "loss": 1.9844,
      "step": 371
    },
    {
      "epoch": 0.3721046544340596,
      "grad_norm": 0.6733037233352661,
      "learning_rate": 0.00019249290426203252,
      "loss": 1.8939,
      "step": 372
    },
    {
      "epoch": 0.3731049357631834,
      "grad_norm": 0.7037007808685303,
      "learning_rate": 0.0001924530054582776,
      "loss": 1.6897,
      "step": 373
    },
    {
      "epoch": 0.3741052170923072,
      "grad_norm": 0.7689145803451538,
      "learning_rate": 0.0001924130050653898,
      "loss": 1.6544,
      "step": 374
    },
    {
      "epoch": 0.37510549842143104,
      "grad_norm": 0.7100968360900879,
      "learning_rate": 0.00019237290312732226,
      "loss": 1.7932,
      "step": 375
    },
    {
      "epoch": 0.37610577975055487,
      "grad_norm": 0.7645193338394165,
      "learning_rate": 0.00019233269968813984,
      "loss": 1.6576,
      "step": 376
    },
    {
      "epoch": 0.37710606107967864,
      "grad_norm": 0.6831678152084351,
      "learning_rate": 0.00019229239479201876,
      "loss": 1.6675,
      "step": 377
    },
    {
      "epoch": 0.3781063424088025,
      "grad_norm": 0.8677794337272644,
      "learning_rate": 0.0001922519884832469,
      "loss": 1.6064,
      "step": 378
    },
    {
      "epoch": 0.3791066237379263,
      "grad_norm": 0.6727691888809204,
      "learning_rate": 0.0001922114808062234,
      "loss": 1.6249,
      "step": 379
    },
    {
      "epoch": 0.3801069050670501,
      "grad_norm": 0.6588670611381531,
      "learning_rate": 0.00019217087180545893,
      "loss": 1.5541,
      "step": 380
    },
    {
      "epoch": 0.3811071863961739,
      "grad_norm": 0.7666369080543518,
      "learning_rate": 0.0001921301615255754,
      "loss": 1.7806,
      "step": 381
    },
    {
      "epoch": 0.38210746772529774,
      "grad_norm": 0.6465156078338623,
      "learning_rate": 0.0001920893500113061,
      "loss": 1.512,
      "step": 382
    },
    {
      "epoch": 0.3831077490544216,
      "grad_norm": 0.7854346632957458,
      "learning_rate": 0.00019204843730749547,
      "loss": 1.6857,
      "step": 383
    },
    {
      "epoch": 0.38410803038354535,
      "grad_norm": 0.6625111103057861,
      "learning_rate": 0.00019200742345909915,
      "loss": 1.7033,
      "step": 384
    },
    {
      "epoch": 0.3851083117126692,
      "grad_norm": 0.7273709177970886,
      "learning_rate": 0.00019196630851118398,
      "loss": 1.665,
      "step": 385
    },
    {
      "epoch": 0.386108593041793,
      "grad_norm": 0.6861465573310852,
      "learning_rate": 0.0001919250925089278,
      "loss": 1.5028,
      "step": 386
    },
    {
      "epoch": 0.38710887437091684,
      "grad_norm": 0.845456063747406,
      "learning_rate": 0.00019188377549761963,
      "loss": 1.967,
      "step": 387
    },
    {
      "epoch": 0.3881091557000406,
      "grad_norm": 0.6481165289878845,
      "learning_rate": 0.00019184235752265928,
      "loss": 1.6053,
      "step": 388
    },
    {
      "epoch": 0.38910943702916445,
      "grad_norm": 0.6312947273254395,
      "learning_rate": 0.00019180083862955772,
      "loss": 1.4427,
      "step": 389
    },
    {
      "epoch": 0.3901097183582883,
      "grad_norm": 0.7874154448509216,
      "learning_rate": 0.00019175921886393666,
      "loss": 1.6099,
      "step": 390
    },
    {
      "epoch": 0.39110999968741206,
      "grad_norm": 0.6839481592178345,
      "learning_rate": 0.00019171749827152869,
      "loss": 1.7004,
      "step": 391
    },
    {
      "epoch": 0.3921102810165359,
      "grad_norm": 0.7239277362823486,
      "learning_rate": 0.0001916756768981772,
      "loss": 1.8813,
      "step": 392
    },
    {
      "epoch": 0.3931105623456597,
      "grad_norm": 0.8241100311279297,
      "learning_rate": 0.00019163375478983632,
      "loss": 1.9443,
      "step": 393
    },
    {
      "epoch": 0.39411084367478355,
      "grad_norm": 0.7401999235153198,
      "learning_rate": 0.00019159173199257085,
      "loss": 1.6663,
      "step": 394
    },
    {
      "epoch": 0.3951111250039073,
      "grad_norm": 0.8297036290168762,
      "learning_rate": 0.00019154960855255628,
      "loss": 1.8012,
      "step": 395
    },
    {
      "epoch": 0.39611140633303116,
      "grad_norm": 0.9661216735839844,
      "learning_rate": 0.0001915073845160786,
      "loss": 1.7007,
      "step": 396
    },
    {
      "epoch": 0.397111687662155,
      "grad_norm": 1.4041926860809326,
      "learning_rate": 0.00019146505992953446,
      "loss": 1.7232,
      "step": 397
    },
    {
      "epoch": 0.3981119689912788,
      "grad_norm": 0.8469036221504211,
      "learning_rate": 0.00019142263483943085,
      "loss": 1.4479,
      "step": 398
    },
    {
      "epoch": 0.3991122503204026,
      "grad_norm": 0.9476561546325684,
      "learning_rate": 0.00019138010929238534,
      "loss": 1.8572,
      "step": 399
    },
    {
      "epoch": 0.4001125316495264,
      "grad_norm": 0.7196705937385559,
      "learning_rate": 0.00019133748333512575,
      "loss": 1.6184,
      "step": 400
    },
    {
      "epoch": 0.40111281297865026,
      "grad_norm": 0.8957480192184448,
      "learning_rate": 0.00019129475701449035,
      "loss": 1.762,
      "step": 401
    },
    {
      "epoch": 0.4021130943077741,
      "grad_norm": 0.7850635647773743,
      "learning_rate": 0.0001912519303774276,
      "loss": 1.6764,
      "step": 402
    },
    {
      "epoch": 0.40311337563689786,
      "grad_norm": 0.7579814791679382,
      "learning_rate": 0.0001912090034709963,
      "loss": 1.6231,
      "step": 403
    },
    {
      "epoch": 0.4041136569660217,
      "grad_norm": 0.7173107266426086,
      "learning_rate": 0.00019116597634236525,
      "loss": 1.7107,
      "step": 404
    },
    {
      "epoch": 0.4051139382951455,
      "grad_norm": 0.7832950353622437,
      "learning_rate": 0.0001911228490388136,
      "loss": 1.8608,
      "step": 405
    },
    {
      "epoch": 0.4061142196242693,
      "grad_norm": 0.716299295425415,
      "learning_rate": 0.00019107962160773035,
      "loss": 1.652,
      "step": 406
    },
    {
      "epoch": 0.40711450095339313,
      "grad_norm": 0.6675253510475159,
      "learning_rate": 0.0001910362940966147,
      "loss": 1.5963,
      "step": 407
    },
    {
      "epoch": 0.40811478228251696,
      "grad_norm": 0.6555336713790894,
      "learning_rate": 0.00019099286655307568,
      "loss": 1.4991,
      "step": 408
    },
    {
      "epoch": 0.4091150636116408,
      "grad_norm": 0.7307867407798767,
      "learning_rate": 0.0001909493390248324,
      "loss": 1.8221,
      "step": 409
    },
    {
      "epoch": 0.41011534494076457,
      "grad_norm": 0.6557430624961853,
      "learning_rate": 0.00019090571155971366,
      "loss": 1.6484,
      "step": 410
    },
    {
      "epoch": 0.4111156262698884,
      "grad_norm": 0.6816605925559998,
      "learning_rate": 0.00019086198420565823,
      "loss": 1.5052,
      "step": 411
    },
    {
      "epoch": 0.41211590759901223,
      "grad_norm": 0.66513592004776,
      "learning_rate": 0.00019081815701071445,
      "loss": 1.8818,
      "step": 412
    },
    {
      "epoch": 0.41311618892813606,
      "grad_norm": 0.6807469129562378,
      "learning_rate": 0.0001907742300230406,
      "loss": 1.5997,
      "step": 413
    },
    {
      "epoch": 0.41411647025725984,
      "grad_norm": 0.8060654401779175,
      "learning_rate": 0.00019073020329090444,
      "loss": 1.8099,
      "step": 414
    },
    {
      "epoch": 0.41511675158638367,
      "grad_norm": 0.7440110445022583,
      "learning_rate": 0.0001906860768626834,
      "loss": 1.4876,
      "step": 415
    },
    {
      "epoch": 0.4161170329155075,
      "grad_norm": 0.7675415277481079,
      "learning_rate": 0.00019064185078686443,
      "loss": 1.4722,
      "step": 416
    },
    {
      "epoch": 0.4171173142446313,
      "grad_norm": 0.6656553149223328,
      "learning_rate": 0.000190597525112044,
      "loss": 1.6453,
      "step": 417
    },
    {
      "epoch": 0.4181175955737551,
      "grad_norm": 0.730689287185669,
      "learning_rate": 0.000190553099886928,
      "loss": 1.6584,
      "step": 418
    },
    {
      "epoch": 0.41911787690287894,
      "grad_norm": 0.8425858616828918,
      "learning_rate": 0.00019050857516033173,
      "loss": 1.6249,
      "step": 419
    },
    {
      "epoch": 0.42011815823200277,
      "grad_norm": 0.7816892266273499,
      "learning_rate": 0.00019046395098117983,
      "loss": 1.7532,
      "step": 420
    },
    {
      "epoch": 0.42111843956112655,
      "grad_norm": 0.7324026823043823,
      "learning_rate": 0.00019041922739850616,
      "loss": 1.8523,
      "step": 421
    },
    {
      "epoch": 0.4221187208902504,
      "grad_norm": 0.7473389506340027,
      "learning_rate": 0.00019037440446145385,
      "loss": 1.583,
      "step": 422
    },
    {
      "epoch": 0.4231190022193742,
      "grad_norm": 0.8720895051956177,
      "learning_rate": 0.00019032948221927524,
      "loss": 1.6806,
      "step": 423
    },
    {
      "epoch": 0.42411928354849804,
      "grad_norm": 0.728528618812561,
      "learning_rate": 0.00019028446072133175,
      "loss": 1.7283,
      "step": 424
    },
    {
      "epoch": 0.4251195648776218,
      "grad_norm": 0.739930272102356,
      "learning_rate": 0.00019023934001709383,
      "loss": 1.7244,
      "step": 425
    },
    {
      "epoch": 0.42611984620674564,
      "grad_norm": 0.7825399041175842,
      "learning_rate": 0.00019019412015614098,
      "loss": 1.7871,
      "step": 426
    },
    {
      "epoch": 0.4271201275358695,
      "grad_norm": 0.8878734707832336,
      "learning_rate": 0.00019014880118816164,
      "loss": 1.6018,
      "step": 427
    },
    {
      "epoch": 0.4281204088649933,
      "grad_norm": 0.726259708404541,
      "learning_rate": 0.0001901033831629532,
      "loss": 1.7732,
      "step": 428
    },
    {
      "epoch": 0.4291206901941171,
      "grad_norm": 0.7620319724082947,
      "learning_rate": 0.00019005786613042185,
      "loss": 1.6466,
      "step": 429
    },
    {
      "epoch": 0.4301209715232409,
      "grad_norm": 0.7295501828193665,
      "learning_rate": 0.00019001225014058255,
      "loss": 1.8708,
      "step": 430
    },
    {
      "epoch": 0.43112125285236474,
      "grad_norm": 0.7419458031654358,
      "learning_rate": 0.00018996653524355902,
      "loss": 1.6583,
      "step": 431
    },
    {
      "epoch": 0.4321215341814885,
      "grad_norm": 0.7701705098152161,
      "learning_rate": 0.00018992072148958368,
      "loss": 1.4421,
      "step": 432
    },
    {
      "epoch": 0.43312181551061235,
      "grad_norm": 0.8237659931182861,
      "learning_rate": 0.00018987480892899758,
      "loss": 1.844,
      "step": 433
    },
    {
      "epoch": 0.4341220968397362,
      "grad_norm": 0.6167672276496887,
      "learning_rate": 0.00018982879761225027,
      "loss": 1.6193,
      "step": 434
    },
    {
      "epoch": 0.43512237816886,
      "grad_norm": 0.7565534710884094,
      "learning_rate": 0.00018978268758989991,
      "loss": 1.7655,
      "step": 435
    },
    {
      "epoch": 0.4361226594979838,
      "grad_norm": 0.8333333730697632,
      "learning_rate": 0.00018973647891261307,
      "loss": 1.5764,
      "step": 436
    },
    {
      "epoch": 0.4371229408271076,
      "grad_norm": 0.7404434084892273,
      "learning_rate": 0.00018969017163116472,
      "loss": 1.7922,
      "step": 437
    },
    {
      "epoch": 0.43812322215623145,
      "grad_norm": 0.7129400372505188,
      "learning_rate": 0.0001896437657964382,
      "loss": 1.6925,
      "step": 438
    },
    {
      "epoch": 0.4391235034853553,
      "grad_norm": 0.7750307321548462,
      "learning_rate": 0.00018959726145942508,
      "loss": 1.8133,
      "step": 439
    },
    {
      "epoch": 0.44012378481447906,
      "grad_norm": 0.7244920134544373,
      "learning_rate": 0.00018955065867122528,
      "loss": 1.6425,
      "step": 440
    },
    {
      "epoch": 0.4411240661436029,
      "grad_norm": 0.7624644637107849,
      "learning_rate": 0.00018950395748304678,
      "loss": 1.886,
      "step": 441
    },
    {
      "epoch": 0.4421243474727267,
      "grad_norm": 0.7016286849975586,
      "learning_rate": 0.0001894571579462058,
      "loss": 1.7308,
      "step": 442
    },
    {
      "epoch": 0.4431246288018505,
      "grad_norm": 0.6965353488922119,
      "learning_rate": 0.00018941026011212654,
      "loss": 1.5872,
      "step": 443
    },
    {
      "epoch": 0.4441249101309743,
      "grad_norm": 0.7479360699653625,
      "learning_rate": 0.00018936326403234125,
      "loss": 1.8129,
      "step": 444
    },
    {
      "epoch": 0.44512519146009816,
      "grad_norm": 0.7027686834335327,
      "learning_rate": 0.00018931616975849006,
      "loss": 1.6433,
      "step": 445
    },
    {
      "epoch": 0.446125472789222,
      "grad_norm": 0.7771592140197754,
      "learning_rate": 0.00018926897734232115,
      "loss": 1.4645,
      "step": 446
    },
    {
      "epoch": 0.44712575411834576,
      "grad_norm": 0.7766458988189697,
      "learning_rate": 0.0001892216868356904,
      "loss": 1.7873,
      "step": 447
    },
    {
      "epoch": 0.4481260354474696,
      "grad_norm": 0.8146182298660278,
      "learning_rate": 0.0001891742982905615,
      "loss": 1.7935,
      "step": 448
    },
    {
      "epoch": 0.4491263167765934,
      "grad_norm": 0.6744781136512756,
      "learning_rate": 0.00018912681175900598,
      "loss": 1.8916,
      "step": 449
    },
    {
      "epoch": 0.45012659810571726,
      "grad_norm": 0.6259024143218994,
      "learning_rate": 0.00018907922729320285,
      "loss": 1.6616,
      "step": 450
    },
    {
      "epoch": 0.45112687943484103,
      "grad_norm": 0.7717494368553162,
      "learning_rate": 0.00018903154494543889,
      "loss": 1.817,
      "step": 451
    },
    {
      "epoch": 0.45212716076396486,
      "grad_norm": 0.648040771484375,
      "learning_rate": 0.00018898376476810834,
      "loss": 1.6309,
      "step": 452
    },
    {
      "epoch": 0.4531274420930887,
      "grad_norm": 0.7560017704963684,
      "learning_rate": 0.00018893588681371303,
      "loss": 1.8016,
      "step": 453
    },
    {
      "epoch": 0.45412772342221247,
      "grad_norm": 0.8778628706932068,
      "learning_rate": 0.00018888791113486213,
      "loss": 1.7797,
      "step": 454
    },
    {
      "epoch": 0.4551280047513363,
      "grad_norm": 0.6795655488967896,
      "learning_rate": 0.00018883983778427227,
      "loss": 1.6343,
      "step": 455
    },
    {
      "epoch": 0.45612828608046013,
      "grad_norm": 0.6690213084220886,
      "learning_rate": 0.0001887916668147673,
      "loss": 1.6224,
      "step": 456
    },
    {
      "epoch": 0.45712856740958396,
      "grad_norm": 0.7529327869415283,
      "learning_rate": 0.00018874339827927846,
      "loss": 1.6396,
      "step": 457
    },
    {
      "epoch": 0.45812884873870774,
      "grad_norm": 0.8393098711967468,
      "learning_rate": 0.00018869503223084414,
      "loss": 1.8374,
      "step": 458
    },
    {
      "epoch": 0.45912913006783157,
      "grad_norm": 0.7435776591300964,
      "learning_rate": 0.00018864656872260985,
      "loss": 1.7363,
      "step": 459
    },
    {
      "epoch": 0.4601294113969554,
      "grad_norm": 0.6737055778503418,
      "learning_rate": 0.00018859800780782828,
      "loss": 1.6661,
      "step": 460
    },
    {
      "epoch": 0.46112969272607923,
      "grad_norm": 0.7674340605735779,
      "learning_rate": 0.000188549349539859,
      "loss": 1.6269,
      "step": 461
    },
    {
      "epoch": 0.462129974055203,
      "grad_norm": 0.7329950928688049,
      "learning_rate": 0.00018850059397216876,
      "loss": 1.6989,
      "step": 462
    },
    {
      "epoch": 0.46313025538432684,
      "grad_norm": 0.7075778841972351,
      "learning_rate": 0.00018845174115833099,
      "loss": 1.7286,
      "step": 463
    },
    {
      "epoch": 0.46413053671345067,
      "grad_norm": 0.7973611354827881,
      "learning_rate": 0.0001884027911520262,
      "loss": 1.7478,
      "step": 464
    },
    {
      "epoch": 0.4651308180425745,
      "grad_norm": 0.7790057063102722,
      "learning_rate": 0.00018835374400704154,
      "loss": 1.6659,
      "step": 465
    },
    {
      "epoch": 0.4661310993716983,
      "grad_norm": 0.8505310416221619,
      "learning_rate": 0.00018830459977727096,
      "loss": 1.6879,
      "step": 466
    },
    {
      "epoch": 0.4671313807008221,
      "grad_norm": 0.7616267800331116,
      "learning_rate": 0.0001882553585167151,
      "loss": 1.6525,
      "step": 467
    },
    {
      "epoch": 0.46813166202994594,
      "grad_norm": 0.8038133978843689,
      "learning_rate": 0.00018820602027948114,
      "loss": 1.7929,
      "step": 468
    },
    {
      "epoch": 0.4691319433590697,
      "grad_norm": 0.6762365698814392,
      "learning_rate": 0.00018815658511978298,
      "loss": 1.6926,
      "step": 469
    },
    {
      "epoch": 0.47013222468819355,
      "grad_norm": 0.6515015959739685,
      "learning_rate": 0.00018810705309194083,
      "loss": 1.7817,
      "step": 470
    },
    {
      "epoch": 0.4711325060173174,
      "grad_norm": 0.696675181388855,
      "learning_rate": 0.00018805742425038145,
      "loss": 1.7195,
      "step": 471
    },
    {
      "epoch": 0.4721327873464412,
      "grad_norm": 0.7929533123970032,
      "learning_rate": 0.00018800769864963802,
      "loss": 2.0165,
      "step": 472
    },
    {
      "epoch": 0.473133068675565,
      "grad_norm": 0.7223886251449585,
      "learning_rate": 0.00018795787634434994,
      "loss": 1.6708,
      "step": 473
    },
    {
      "epoch": 0.4741333500046888,
      "grad_norm": 0.7982028126716614,
      "learning_rate": 0.0001879079573892629,
      "loss": 1.628,
      "step": 474
    },
    {
      "epoch": 0.47513363133381264,
      "grad_norm": 0.6962152123451233,
      "learning_rate": 0.00018785794183922883,
      "loss": 1.6429,
      "step": 475
    },
    {
      "epoch": 0.4761339126629365,
      "grad_norm": 0.687489926815033,
      "learning_rate": 0.00018780782974920572,
      "loss": 1.4546,
      "step": 476
    },
    {
      "epoch": 0.47713419399206025,
      "grad_norm": 0.7260375022888184,
      "learning_rate": 0.00018775762117425777,
      "loss": 1.739,
      "step": 477
    },
    {
      "epoch": 0.4781344753211841,
      "grad_norm": 0.759400486946106,
      "learning_rate": 0.0001877073161695551,
      "loss": 1.6465,
      "step": 478
    },
    {
      "epoch": 0.4791347566503079,
      "grad_norm": 0.7412364482879639,
      "learning_rate": 0.00018765691479037376,
      "loss": 1.7333,
      "step": 479
    },
    {
      "epoch": 0.4801350379794317,
      "grad_norm": 0.6909674406051636,
      "learning_rate": 0.00018760641709209583,
      "loss": 1.6936,
      "step": 480
    },
    {
      "epoch": 0.4811353193085555,
      "grad_norm": 0.6478050947189331,
      "learning_rate": 0.0001875558231302091,
      "loss": 1.4435,
      "step": 481
    },
    {
      "epoch": 0.48213560063767935,
      "grad_norm": 0.6662638187408447,
      "learning_rate": 0.00018750513296030718,
      "loss": 1.5567,
      "step": 482
    },
    {
      "epoch": 0.4831358819668032,
      "grad_norm": 0.6973413825035095,
      "learning_rate": 0.00018745434663808942,
      "loss": 1.6434,
      "step": 483
    },
    {
      "epoch": 0.48413616329592696,
      "grad_norm": 0.8381956815719604,
      "learning_rate": 0.0001874034642193608,
      "loss": 1.8568,
      "step": 484
    },
    {
      "epoch": 0.4851364446250508,
      "grad_norm": 0.6522262096405029,
      "learning_rate": 0.0001873524857600319,
      "loss": 1.4265,
      "step": 485
    },
    {
      "epoch": 0.4861367259541746,
      "grad_norm": 0.7018440961837769,
      "learning_rate": 0.00018730141131611882,
      "loss": 1.6914,
      "step": 486
    },
    {
      "epoch": 0.48713700728329845,
      "grad_norm": 0.8237236142158508,
      "learning_rate": 0.00018725024094374315,
      "loss": 1.4462,
      "step": 487
    },
    {
      "epoch": 0.4881372886124222,
      "grad_norm": 0.6507928967475891,
      "learning_rate": 0.00018719897469913184,
      "loss": 1.5802,
      "step": 488
    },
    {
      "epoch": 0.48913756994154606,
      "grad_norm": 0.8120100498199463,
      "learning_rate": 0.00018714761263861728,
      "loss": 1.7819,
      "step": 489
    },
    {
      "epoch": 0.4901378512706699,
      "grad_norm": 0.759722888469696,
      "learning_rate": 0.000187096154818637,
      "loss": 1.7481,
      "step": 490
    },
    {
      "epoch": 0.4911381325997937,
      "grad_norm": 0.7146822214126587,
      "learning_rate": 0.00018704460129573391,
      "loss": 1.6217,
      "step": 491
    },
    {
      "epoch": 0.4921384139289175,
      "grad_norm": 0.7138429284095764,
      "learning_rate": 0.00018699295212655596,
      "loss": 1.7242,
      "step": 492
    },
    {
      "epoch": 0.4931386952580413,
      "grad_norm": 0.8145613670349121,
      "learning_rate": 0.00018694120736785632,
      "loss": 1.8759,
      "step": 493
    },
    {
      "epoch": 0.49413897658716516,
      "grad_norm": 0.6624435186386108,
      "learning_rate": 0.00018688936707649304,
      "loss": 1.8632,
      "step": 494
    },
    {
      "epoch": 0.49513925791628893,
      "grad_norm": 0.6550843119621277,
      "learning_rate": 0.00018683743130942928,
      "loss": 1.7645,
      "step": 495
    },
    {
      "epoch": 0.49613953924541276,
      "grad_norm": 0.6931488513946533,
      "learning_rate": 0.00018678540012373302,
      "loss": 1.5899,
      "step": 496
    },
    {
      "epoch": 0.4971398205745366,
      "grad_norm": 0.684229850769043,
      "learning_rate": 0.00018673327357657715,
      "loss": 1.6902,
      "step": 497
    },
    {
      "epoch": 0.4981401019036604,
      "grad_norm": 0.7380666136741638,
      "learning_rate": 0.0001866810517252393,
      "loss": 1.7115,
      "step": 498
    },
    {
      "epoch": 0.4991403832327842,
      "grad_norm": 0.6782827973365784,
      "learning_rate": 0.00018662873462710184,
      "loss": 1.495,
      "step": 499
    },
    {
      "epoch": 0.5001406645619081,
      "grad_norm": 0.730248749256134,
      "learning_rate": 0.0001865763223396518,
      "loss": 1.5147,
      "step": 500
    }
  ],
  "logging_steps": 1,
  "max_steps": 2997,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 4045941965881344.0,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}