{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 1.0,
  "eval_steps": 500,
  "global_step": 2917,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.003428914320003429,
      "grad_norm": 5.610459804534912,
      "learning_rate": 2.054794520547945e-06,
      "loss": 8.609027862548828,
      "step": 10
    },
    {
      "epoch": 0.006857828640006858,
      "grad_norm": 5.640138626098633,
      "learning_rate": 4.337899543378996e-06,
      "loss": 8.658802032470703,
      "step": 20
    },
    {
      "epoch": 0.010286742960010286,
      "grad_norm": 5.976888179779053,
      "learning_rate": 6.621004566210046e-06,
      "loss": 8.27346420288086,
      "step": 30
    },
    {
      "epoch": 0.013715657280013716,
      "grad_norm": 7.228434085845947,
      "learning_rate": 8.904109589041095e-06,
      "loss": 7.799014282226563,
      "step": 40
    },
    {
      "epoch": 0.017144571600017144,
      "grad_norm": 5.19713830947876,
      "learning_rate": 1.1187214611872146e-05,
      "loss": 6.894306945800781,
      "step": 50
    },
    {
      "epoch": 0.020573485920020573,
      "grad_norm": 3.126099109649658,
      "learning_rate": 1.3470319634703196e-05,
      "loss": 5.792304611206054,
      "step": 60
    },
    {
      "epoch": 0.024002400240024,
      "grad_norm": 1.902426838874817,
      "learning_rate": 1.5753424657534248e-05,
      "loss": 4.926207733154297,
      "step": 70
    },
    {
      "epoch": 0.027431314560027432,
      "grad_norm": 1.4311621189117432,
      "learning_rate": 1.80365296803653e-05,
      "loss": 4.423136520385742,
      "step": 80
    },
    {
      "epoch": 0.03086022888003086,
      "grad_norm": 1.5511047840118408,
      "learning_rate": 2.0319634703196345e-05,
      "loss": 4.315621185302734,
      "step": 90
    },
    {
      "epoch": 0.03428914320003429,
      "grad_norm": 1.028840184211731,
      "learning_rate": 2.2602739726027396e-05,
      "loss": 4.081908798217773,
      "step": 100
    },
    {
      "epoch": 0.03771805752003772,
      "grad_norm": 1.351972222328186,
      "learning_rate": 2.4885844748858446e-05,
      "loss": 3.812961196899414,
      "step": 110
    },
    {
      "epoch": 0.041146971840041145,
      "grad_norm": 1.2235620021820068,
      "learning_rate": 2.71689497716895e-05,
      "loss": 3.6152915954589844,
      "step": 120
    },
    {
      "epoch": 0.04457588616004458,
      "grad_norm": 1.178669810295105,
      "learning_rate": 2.945205479452055e-05,
      "loss": 3.3623470306396483,
      "step": 130
    },
    {
      "epoch": 0.048004800480048,
      "grad_norm": 1.3753740787506104,
      "learning_rate": 3.17351598173516e-05,
      "loss": 3.269461441040039,
      "step": 140
    },
    {
      "epoch": 0.05143371480005143,
      "grad_norm": 1.5192193984985352,
      "learning_rate": 3.4018264840182654e-05,
      "loss": 3.1576816558837892,
      "step": 150
    },
    {
      "epoch": 0.054862629120054865,
      "grad_norm": 1.185823917388916,
      "learning_rate": 3.63013698630137e-05,
      "loss": 2.989401435852051,
      "step": 160
    },
    {
      "epoch": 0.05829154344005829,
      "grad_norm": 1.2177551984786987,
      "learning_rate": 3.8584474885844754e-05,
      "loss": 2.9551776885986327,
      "step": 170
    },
    {
      "epoch": 0.06172045776006172,
      "grad_norm": 1.6961393356323242,
      "learning_rate": 4.08675799086758e-05,
      "loss": 2.888522148132324,
      "step": 180
    },
    {
      "epoch": 0.06514937208006515,
      "grad_norm": 1.9519675970077515,
      "learning_rate": 4.3150684931506855e-05,
      "loss": 2.8754018783569335,
      "step": 190
    },
    {
      "epoch": 0.06857828640006858,
      "grad_norm": 1.8900648355484009,
      "learning_rate": 4.54337899543379e-05,
      "loss": 2.728044319152832,
      "step": 200
    },
    {
      "epoch": 0.07200720072007201,
      "grad_norm": 2.05349063873291,
      "learning_rate": 4.7716894977168955e-05,
      "loss": 2.7611804962158204,
      "step": 210
    },
    {
      "epoch": 0.07543611504007544,
      "grad_norm": 2.1214232444763184,
      "learning_rate": 5e-05,
      "loss": 2.6431552886962892,
      "step": 220
    },
    {
      "epoch": 0.07886502936007886,
      "grad_norm": 2.1663858890533447,
      "learning_rate": 5.2283105022831056e-05,
      "loss": 2.6501447677612306,
      "step": 230
    },
    {
      "epoch": 0.08229394368008229,
      "grad_norm": 2.0235824584960938,
      "learning_rate": 5.45662100456621e-05,
      "loss": 2.6541574478149412,
      "step": 240
    },
    {
      "epoch": 0.08572285800008572,
      "grad_norm": 1.7880959510803223,
      "learning_rate": 5.684931506849316e-05,
      "loss": 2.5670385360717773,
      "step": 250
    },
    {
      "epoch": 0.08915177232008915,
      "grad_norm": 2.3312928676605225,
      "learning_rate": 5.9132420091324204e-05,
      "loss": 2.587135887145996,
      "step": 260
    },
    {
      "epoch": 0.09258068664009259,
      "grad_norm": 2.3413031101226807,
      "learning_rate": 6.141552511415526e-05,
      "loss": 2.5546056747436525,
      "step": 270
    },
    {
      "epoch": 0.096009600960096,
      "grad_norm": 1.8639769554138184,
      "learning_rate": 6.36986301369863e-05,
      "loss": 2.482296180725098,
      "step": 280
    },
    {
      "epoch": 0.09943851528009943,
      "grad_norm": 2.093104362487793,
      "learning_rate": 6.598173515981736e-05,
      "loss": 2.3930648803710937,
      "step": 290
    },
    {
      "epoch": 0.10286742960010287,
      "grad_norm": 2.4355857372283936,
      "learning_rate": 6.82648401826484e-05,
      "loss": 2.401523399353027,
      "step": 300
    },
    {
      "epoch": 0.1062963439201063,
      "grad_norm": 1.8734594583511353,
      "learning_rate": 7.054794520547947e-05,
      "loss": 2.343214416503906,
      "step": 310
    },
    {
      "epoch": 0.10972525824010973,
      "grad_norm": 2.372744560241699,
      "learning_rate": 7.28310502283105e-05,
      "loss": 2.3665439605712892,
      "step": 320
    },
    {
      "epoch": 0.11315417256011315,
      "grad_norm": 2.343730926513672,
      "learning_rate": 7.511415525114156e-05,
      "loss": 2.4043195724487303,
      "step": 330
    },
    {
      "epoch": 0.11658308688011658,
      "grad_norm": 2.2000231742858887,
      "learning_rate": 7.73972602739726e-05,
      "loss": 2.3719856262207033,
      "step": 340
    },
    {
      "epoch": 0.12001200120012001,
      "grad_norm": 2.6558656692504883,
      "learning_rate": 7.968036529680367e-05,
      "loss": 2.3660871505737306,
      "step": 350
    },
    {
      "epoch": 0.12344091552012344,
      "grad_norm": 2.3702239990234375,
      "learning_rate": 8.19634703196347e-05,
      "loss": 2.328384780883789,
      "step": 360
    },
    {
      "epoch": 0.12686982984012687,
      "grad_norm": 2.633316993713379,
      "learning_rate": 8.424657534246576e-05,
      "loss": 2.2853065490722657,
      "step": 370
    },
    {
      "epoch": 0.1302987441601303,
      "grad_norm": 2.089054584503174,
      "learning_rate": 8.65296803652968e-05,
      "loss": 2.3018625259399412,
      "step": 380
    },
    {
      "epoch": 0.13372765848013374,
      "grad_norm": 2.5298097133636475,
      "learning_rate": 8.881278538812787e-05,
      "loss": 2.2930763244628904,
      "step": 390
    },
    {
      "epoch": 0.13715657280013716,
      "grad_norm": 2.35923171043396,
      "learning_rate": 9.109589041095891e-05,
      "loss": 2.254222869873047,
      "step": 400
    },
    {
      "epoch": 0.14058548712014057,
      "grad_norm": 2.32989239692688,
      "learning_rate": 9.337899543378996e-05,
      "loss": 2.2330236434936523,
      "step": 410
    },
    {
      "epoch": 0.14401440144014402,
      "grad_norm": 2.4885056018829346,
      "learning_rate": 9.5662100456621e-05,
      "loss": 2.282236099243164,
      "step": 420
    },
    {
      "epoch": 0.14744331576014744,
      "grad_norm": 2.455707550048828,
      "learning_rate": 9.794520547945207e-05,
      "loss": 2.3149810791015626,
      "step": 430
    },
    {
      "epoch": 0.15087223008015088,
      "grad_norm": 2.834005355834961,
      "learning_rate": 0.00010022831050228312,
      "loss": 2.2503019332885743,
      "step": 440
    },
    {
      "epoch": 0.1543011444001543,
      "grad_norm": 2.496020555496216,
      "learning_rate": 0.00010251141552511416,
      "loss": 2.300754165649414,
      "step": 450
    },
    {
      "epoch": 0.15773005872015772,
      "grad_norm": 2.560879707336426,
      "learning_rate": 0.0001047945205479452,
      "loss": 2.168670082092285,
      "step": 460
    },
    {
      "epoch": 0.16115897304016116,
      "grad_norm": 2.4285600185394287,
      "learning_rate": 0.00010707762557077626,
      "loss": 2.327123832702637,
      "step": 470
    },
    {
      "epoch": 0.16458788736016458,
      "grad_norm": 2.9308815002441406,
      "learning_rate": 0.00010936073059360732,
      "loss": 2.2077674865722656,
      "step": 480
    },
    {
      "epoch": 0.16801680168016803,
      "grad_norm": 2.1128101348876953,
      "learning_rate": 0.00011164383561643836,
      "loss": 2.1291229248046877,
      "step": 490
    },
    {
      "epoch": 0.17144571600017144,
      "grad_norm": 3.1350789070129395,
      "learning_rate": 0.0001139269406392694,
      "loss": 2.269939422607422,
      "step": 500
    },
    {
      "epoch": 0.17487463032017486,
      "grad_norm": 3.754878520965576,
      "learning_rate": 0.00011621004566210046,
      "loss": 2.1504594802856447,
      "step": 510
    },
    {
      "epoch": 0.1783035446401783,
      "grad_norm": 3.5980262756347656,
      "learning_rate": 0.00011849315068493152,
      "loss": 2.18060188293457,
      "step": 520
    },
    {
      "epoch": 0.18173245896018173,
      "grad_norm": 2.6357414722442627,
      "learning_rate": 0.00012077625570776256,
      "loss": 2.2333398818969727,
      "step": 530
    },
    {
      "epoch": 0.18516137328018517,
      "grad_norm": 2.370551824569702,
      "learning_rate": 0.0001230593607305936,
      "loss": 2.083410453796387,
      "step": 540
    },
    {
      "epoch": 0.1885902876001886,
      "grad_norm": 2.782283306121826,
      "learning_rate": 0.00012534246575342465,
      "loss": 2.1948259353637694,
      "step": 550
    },
    {
      "epoch": 0.192019201920192,
      "grad_norm": 2.1626694202423096,
      "learning_rate": 0.0001276255707762557,
      "loss": 2.159800148010254,
      "step": 560
    },
    {
      "epoch": 0.19544811624019545,
      "grad_norm": 2.3545384407043457,
      "learning_rate": 0.00012990867579908675,
      "loss": 2.095806694030762,
      "step": 570
    },
    {
      "epoch": 0.19887703056019887,
      "grad_norm": 2.0062921047210693,
      "learning_rate": 0.00013219178082191782,
      "loss": 2.074744987487793,
      "step": 580
    },
    {
      "epoch": 0.20230594488020232,
      "grad_norm": 1.8789395093917847,
      "learning_rate": 0.00013447488584474886,
      "loss": 2.105234718322754,
      "step": 590
    },
    {
      "epoch": 0.20573485920020573,
      "grad_norm": 3.183367967605591,
      "learning_rate": 0.00013675799086757993,
      "loss": 2.092378616333008,
      "step": 600
    },
    {
      "epoch": 0.20916377352020915,
      "grad_norm": 2.670680284500122,
      "learning_rate": 0.00013904109589041097,
      "loss": 2.107949638366699,
      "step": 610
    },
    {
      "epoch": 0.2125926878402126,
      "grad_norm": 2.535616636276245,
      "learning_rate": 0.000141324200913242,
      "loss": 2.104890251159668,
      "step": 620
    },
    {
      "epoch": 0.21602160216021601,
      "grad_norm": 2.1831581592559814,
      "learning_rate": 0.00014360730593607305,
      "loss": 2.068290901184082,
      "step": 630
    },
    {
      "epoch": 0.21945051648021946,
      "grad_norm": 2.19616961479187,
      "learning_rate": 0.00014589041095890411,
      "loss": 2.072966766357422,
      "step": 640
    },
    {
      "epoch": 0.22287943080022288,
      "grad_norm": 1.7544476985931396,
      "learning_rate": 0.00014817351598173515,
      "loss": 2.0328533172607424,
      "step": 650
    },
    {
      "epoch": 0.2263083451202263,
      "grad_norm": 2.0088233947753906,
      "learning_rate": 0.00015045662100456622,
      "loss": 2.054640007019043,
      "step": 660
    },
    {
      "epoch": 0.22973725944022974,
      "grad_norm": 1.7762000560760498,
      "learning_rate": 0.0001527397260273973,
      "loss": 2.1484561920166017,
      "step": 670
    },
    {
      "epoch": 0.23316617376023316,
      "grad_norm": 1.9017152786254883,
      "learning_rate": 0.00015502283105022833,
      "loss": 2.135583686828613,
      "step": 680
    },
    {
      "epoch": 0.2365950880802366,
      "grad_norm": 2.2522757053375244,
      "learning_rate": 0.00015730593607305937,
      "loss": 2.091486167907715,
      "step": 690
    },
    {
      "epoch": 0.24002400240024002,
      "grad_norm": 1.6416990756988525,
      "learning_rate": 0.0001595890410958904,
      "loss": 2.0411283493041994,
      "step": 700
    },
    {
      "epoch": 0.24345291672024344,
      "grad_norm": 2.146657943725586,
      "learning_rate": 0.00016187214611872148,
      "loss": 2.074742889404297,
      "step": 710
    },
    {
      "epoch": 0.24688183104024689,
      "grad_norm": 2.341554641723633,
      "learning_rate": 0.00016415525114155252,
      "loss": 2.025459861755371,
      "step": 720
    },
    {
      "epoch": 0.25031074536025033,
      "grad_norm": 2.311725616455078,
      "learning_rate": 0.00016643835616438356,
      "loss": 2.084691619873047,
      "step": 730
    },
    {
      "epoch": 0.25373965968025375,
      "grad_norm": 3.090297222137451,
      "learning_rate": 0.00016872146118721462,
      "loss": 2.0962242126464843,
      "step": 740
    },
    {
      "epoch": 0.25716857400025717,
      "grad_norm": 2.4631097316741943,
      "learning_rate": 0.0001710045662100457,
      "loss": 1.9135856628417969,
      "step": 750
    },
    {
      "epoch": 0.2605974883202606,
      "grad_norm": 1.8587676286697388,
      "learning_rate": 0.00017328767123287673,
      "loss": 1.9986955642700195,
      "step": 760
    },
    {
      "epoch": 0.264026402640264,
      "grad_norm": 2.0810563564300537,
      "learning_rate": 0.00017557077625570777,
      "loss": 2.079091262817383,
      "step": 770
    },
    {
      "epoch": 0.2674553169602675,
      "grad_norm": 1.857906699180603,
      "learning_rate": 0.0001778538812785388,
      "loss": 1.8792608261108399,
      "step": 780
    },
    {
      "epoch": 0.2708842312802709,
      "grad_norm": 1.5612140893936157,
      "learning_rate": 0.00018013698630136988,
      "loss": 2.057716178894043,
      "step": 790
    },
    {
      "epoch": 0.2743131456002743,
      "grad_norm": 1.810215950012207,
      "learning_rate": 0.00018242009132420092,
      "loss": 1.9703859329223632,
      "step": 800
    },
    {
      "epoch": 0.27774205992027773,
      "grad_norm": 2.8802878856658936,
      "learning_rate": 0.00018470319634703196,
      "loss": 1.9415620803833007,
      "step": 810
    },
    {
      "epoch": 0.28117097424028115,
      "grad_norm": 1.5730887651443481,
      "learning_rate": 0.00018698630136986303,
      "loss": 1.961550521850586,
      "step": 820
    },
    {
      "epoch": 0.2845998885602846,
      "grad_norm": 1.6987899541854858,
      "learning_rate": 0.0001892694063926941,
      "loss": 1.9590007781982421,
      "step": 830
    },
    {
      "epoch": 0.28802880288028804,
      "grad_norm": 1.5258079767227173,
      "learning_rate": 0.00019155251141552513,
      "loss": 1.9020195007324219,
      "step": 840
    },
    {
      "epoch": 0.29145771720029146,
      "grad_norm": 1.5485594272613525,
      "learning_rate": 0.00019383561643835617,
      "loss": 1.9389507293701171,
      "step": 850
    },
    {
      "epoch": 0.2948866315202949,
      "grad_norm": 1.7738769054412842,
      "learning_rate": 0.00019611872146118721,
      "loss": 1.9449392318725587,
      "step": 860
    },
    {
      "epoch": 0.2983155458402983,
      "grad_norm": 1.8433096408843994,
      "learning_rate": 0.00019840182648401828,
      "loss": 1.9001651763916017,
      "step": 870
    },
    {
      "epoch": 0.30174446016030176,
      "grad_norm": 1.5525556802749634,
      "learning_rate": 0.00019999999445216476,
      "loss": 1.9523231506347656,
      "step": 880
    },
    {
      "epoch": 0.3051733744803052,
      "grad_norm": 1.9127219915390015,
      "learning_rate": 0.0001999998958239994,
      "loss": 1.9692331314086915,
      "step": 890
    },
    {
      "epoch": 0.3086022888003086,
      "grad_norm": 1.4665924310684204,
      "learning_rate": 0.0001999996739107459,
      "loss": 1.9772893905639648,
      "step": 900
    },
    {
      "epoch": 0.312031203120312,
      "grad_norm": 1.2458423376083374,
      "learning_rate": 0.00019999932871267792,
      "loss": 1.8345914840698243,
      "step": 910
    },
    {
      "epoch": 0.31546011744031544,
      "grad_norm": 1.2821522951126099,
      "learning_rate": 0.0001999988602302209,
      "loss": 1.8470048904418945,
      "step": 920
    },
    {
      "epoch": 0.3188890317603189,
      "grad_norm": 1.4526731967926025,
      "learning_rate": 0.00019999826846395254,
      "loss": 1.86633243560791,
      "step": 930
    },
    {
      "epoch": 0.3223179460803223,
      "grad_norm": 2.0247552394866943,
      "learning_rate": 0.00019999755341460236,
      "loss": 1.9277801513671875,
      "step": 940
    },
    {
      "epoch": 0.32574686040032574,
      "grad_norm": 1.3679293394088745,
      "learning_rate": 0.00019999671508305185,
      "loss": 1.8985067367553712,
      "step": 950
    },
    {
      "epoch": 0.32917577472032916,
      "grad_norm": 1.4078823328018188,
      "learning_rate": 0.0001999957534703346,
      "loss": 1.869325828552246,
      "step": 960
    },
    {
      "epoch": 0.3326046890403326,
      "grad_norm": 1.3597145080566406,
      "learning_rate": 0.00019999466857763615,
      "loss": 1.8647089004516602,
      "step": 970
    },
    {
      "epoch": 0.33603360336033605,
      "grad_norm": 1.2973030805587769,
      "learning_rate": 0.000199993460406294,
      "loss": 1.9011524200439454,
      "step": 980
    },
    {
      "epoch": 0.33946251768033947,
      "grad_norm": 1.2790284156799316,
      "learning_rate": 0.0001999921289577976,
      "loss": 1.846287155151367,
      "step": 990
    },
    {
      "epoch": 0.3428914320003429,
      "grad_norm": 1.3097280263900757,
      "learning_rate": 0.00019999067423378853,
      "loss": 1.8515178680419921,
      "step": 1000
    },
    {
      "epoch": 0.3463203463203463,
      "grad_norm": 1.507128119468689,
      "learning_rate": 0.00019998909623606014,
      "loss": 1.88543758392334,
      "step": 1010
    },
    {
      "epoch": 0.3497492606403497,
      "grad_norm": 1.1927403211593628,
      "learning_rate": 0.0001999873949665579,
      "loss": 1.6981697082519531,
      "step": 1020
    },
    {
      "epoch": 0.3531781749603532,
      "grad_norm": 1.331699013710022,
      "learning_rate": 0.0001999855704273793,
      "loss": 1.8030847549438476,
      "step": 1030
    },
    {
      "epoch": 0.3566070892803566,
      "grad_norm": 1.2758996486663818,
      "learning_rate": 0.00019998362262077364,
      "loss": 1.8041196823120118,
      "step": 1040
    },
    {
      "epoch": 0.36003600360036003,
      "grad_norm": 1.010694980621338,
      "learning_rate": 0.0001999815515491423,
      "loss": 1.7889204025268555,
      "step": 1050
    },
    {
      "epoch": 0.36346491792036345,
      "grad_norm": 1.0782943964004517,
      "learning_rate": 0.0001999793572150386,
      "loss": 1.8582679748535156,
      "step": 1060
    },
    {
      "epoch": 0.36689383224036687,
      "grad_norm": 1.6891676187515259,
      "learning_rate": 0.0001999770396211679,
      "loss": 1.838225746154785,
      "step": 1070
    },
    {
      "epoch": 0.37032274656037034,
      "grad_norm": 1.369584560394287,
      "learning_rate": 0.00019997459877038734,
      "loss": 1.7001020431518554,
      "step": 1080
    },
    {
      "epoch": 0.37375166088037376,
      "grad_norm": 1.8637924194335938,
      "learning_rate": 0.00019997203466570618,
      "loss": 1.854163360595703,
      "step": 1090
    },
    {
      "epoch": 0.3771805752003772,
      "grad_norm": 1.3408452272415161,
      "learning_rate": 0.00019996934731028558,
      "loss": 1.7165304183959962,
      "step": 1100
    },
    {
      "epoch": 0.3806094895203806,
      "grad_norm": 1.0390794277191162,
      "learning_rate": 0.00019996653670743868,
      "loss": 1.767896842956543,
      "step": 1110
    },
    {
      "epoch": 0.384038403840384,
      "grad_norm": 1.5660971403121948,
      "learning_rate": 0.00019996360286063047,
      "loss": 1.780000877380371,
      "step": 1120
    },
    {
      "epoch": 0.3874673181603875,
      "grad_norm": 1.0876163244247437,
      "learning_rate": 0.00019996054577347804,
      "loss": 1.6469404220581054,
      "step": 1130
    },
    {
      "epoch": 0.3908962324803909,
      "grad_norm": 1.1568710803985596,
      "learning_rate": 0.00019995736544975022,
      "loss": 1.6766242980957031,
      "step": 1140
    },
    {
      "epoch": 0.3943251468003943,
      "grad_norm": 1.6297674179077148,
      "learning_rate": 0.00019995406189336797,
      "loss": 1.6692434310913087,
      "step": 1150
    },
    {
      "epoch": 0.39775406112039774,
      "grad_norm": 1.1923457384109497,
      "learning_rate": 0.00019995063510840403,
      "loss": 1.6861326217651367,
      "step": 1160
    },
    {
      "epoch": 0.40118297544040116,
      "grad_norm": 1.114096999168396,
      "learning_rate": 0.00013619420538871178,
      "loss": 1.8907966613769531,
      "step": 1170
    },
    {
      "epoch": 0.40461188976040463,
      "grad_norm": 1.0683165788650513,
      "learning_rate": 0.00013515679009466856,
      "loss": 1.7222326278686524,
      "step": 1180
    },
    {
      "epoch": 0.40804080408040805,
      "grad_norm": 1.0893193483352661,
      "learning_rate": 0.00013411503930648705,
      "loss": 1.7620574951171875,
      "step": 1190
    },
    {
      "epoch": 0.41146971840041147,
      "grad_norm": 1.0386360883712769,
      "learning_rate": 0.0001330690814916532,
      "loss": 1.6898015975952148,
      "step": 1200
    },
    {
      "epoch": 0.4148986327204149,
      "grad_norm": 0.9659198522567749,
      "learning_rate": 0.00013201904563645852,
      "loss": 1.7476688385009767,
      "step": 1210
    },
    {
      "epoch": 0.4183275470404183,
      "grad_norm": 1.0083379745483398,
      "learning_rate": 0.00013096506123009367,
      "loss": 1.751401710510254,
      "step": 1220
    },
    {
      "epoch": 0.4217564613604218,
      "grad_norm": 0.9516507983207703,
      "learning_rate": 0.00012990725824867994,
      "loss": 1.714729118347168,
      "step": 1230
    },
    {
      "epoch": 0.4251853756804252,
      "grad_norm": 1.1555304527282715,
      "learning_rate": 0.00012884576713924093,
      "loss": 1.6349376678466796,
      "step": 1240
    },
    {
      "epoch": 0.4286142900004286,
      "grad_norm": 1.0956013202667236,
      "learning_rate": 0.00012778071880361578,
      "loss": 1.7309770584106445,
      "step": 1250
    },
    {
      "epoch": 0.43204320432043203,
      "grad_norm": 0.9085822105407715,
      "learning_rate": 0.00012671224458231663,
      "loss": 1.6261857986450194,
      "step": 1260
    },
    {
      "epoch": 0.43547211864043545,
      "grad_norm": 1.1097203493118286,
      "learning_rate": 0.00012564047623833186,
      "loss": 1.7695898056030273,
      "step": 1270
    },
    {
      "epoch": 0.4389010329604389,
      "grad_norm": 1.0959359407424927,
      "learning_rate": 0.00012456554594087707,
      "loss": 1.563797950744629,
      "step": 1280
    },
    {
      "epoch": 0.44232994728044234,
      "grad_norm": 1.0197439193725586,
      "learning_rate": 0.00012348758624909645,
      "loss": 1.7713394165039062,
      "step": 1290
    },
    {
      "epoch": 0.44575886160044575,
      "grad_norm": 0.9670558571815491,
      "learning_rate": 0.00012240673009571534,
      "loss": 1.7140020370483398,
      "step": 1300
    },
    {
      "epoch": 0.4491877759204492,
      "grad_norm": 1.0650691986083984,
      "learning_rate": 0.00012132311077064749,
      "loss": 1.5492257118225097,
      "step": 1310
    },
    {
      "epoch": 0.4526166902404526,
      "grad_norm": 1.0070433616638184,
      "learning_rate": 0.00012023686190455765,
      "loss": 1.6611621856689454,
      "step": 1320
    },
    {
      "epoch": 0.45604560456045606,
      "grad_norm": 1.268326997756958,
      "learning_rate": 0.00011914811745238254,
      "loss": 1.5741493225097656,
      "step": 1330
    },
    {
      "epoch": 0.4594745188804595,
      "grad_norm": 0.9612389206886292,
      "learning_rate": 0.0001180570116768116,
      "loss": 1.6689472198486328,
      "step": 1340
    },
    {
      "epoch": 0.4629034332004629,
      "grad_norm": 1.0205965042114258,
      "learning_rate": 0.0001169636791317299,
      "loss": 1.6650468826293945,
      "step": 1350
    },
    {
      "epoch": 0.4663323475204663,
      "grad_norm": 1.0054283142089844,
      "learning_rate": 0.00011586825464562514,
      "loss": 1.6489713668823243,
      "step": 1360
    },
    {
      "epoch": 0.46976126184046973,
      "grad_norm": 0.9264296293258667,
      "learning_rate": 0.0001147708733049607,
      "loss": 1.5504850387573241,
      "step": 1370
    },
    {
      "epoch": 0.4731901761604732,
      "grad_norm": 0.9366150498390198,
      "learning_rate": 0.000113671670437517,
      "loss": 1.602787399291992,
      "step": 1380
    },
    {
      "epoch": 0.4766190904804766,
      "grad_norm": 1.2649458646774292,
      "learning_rate": 0.00011257078159570302,
      "loss": 1.6327119827270509,
      "step": 1390
    },
    {
      "epoch": 0.48004800480048004,
      "grad_norm": 1.1246364116668701,
      "learning_rate": 0.00011146834253984006,
      "loss": 1.5650274276733398,
      "step": 1400
    },
    {
      "epoch": 0.48347691912048346,
      "grad_norm": 1.1438333988189697,
      "learning_rate": 0.00011036448922142004,
      "loss": 1.6341632843017577,
      "step": 1410
    },
    {
      "epoch": 0.4869058334404869,
      "grad_norm": 0.959320068359375,
      "learning_rate": 0.00010925935776634013,
      "loss": 1.5222739219665526,
      "step": 1420
    },
    {
      "epoch": 0.49033474776049035,
      "grad_norm": 1.1040620803833008,
      "learning_rate": 0.00010815308445811569,
      "loss": 1.5879670143127442,
      "step": 1430
    },
    {
      "epoch": 0.49376366208049377,
      "grad_norm": 1.0933845043182373,
      "learning_rate": 0.00010704580572107423,
      "loss": 1.6102508544921874,
      "step": 1440
    },
    {
      "epoch": 0.4971925764004972,
      "grad_norm": 0.9115415215492249,
      "learning_rate": 0.00010593765810353143,
      "loss": 1.5763679504394532,
      "step": 1450
    },
    {
      "epoch": 0.5006214907205007,
      "grad_norm": 1.0321201086044312,
      "learning_rate": 0.00010482877826095233,
      "loss": 1.7495067596435547,
      "step": 1460
    },
    {
      "epoch": 0.504050405040504,
      "grad_norm": 0.8646981120109558,
      "learning_rate": 0.0001037193029390991,
      "loss": 1.5509690284729003,
      "step": 1470
    },
    {
      "epoch": 0.5074793193605075,
      "grad_norm": 1.0316283702850342,
      "learning_rate": 0.00010260936895716782,
      "loss": 1.6160917282104492,
      "step": 1480
    },
    {
      "epoch": 0.5109082336805109,
      "grad_norm": 0.9568183422088623,
      "learning_rate": 0.00010149911319091583,
      "loss": 1.5358777046203613,
      "step": 1490
    },
    {
      "epoch": 0.5143371480005143,
      "grad_norm": 0.9523828029632568,
      "learning_rate": 0.00010038867255578262,
      "loss": 1.523128128051758,
      "step": 1500
    },
    {
      "epoch": 0.5177660623205178,
      "grad_norm": 0.8689149022102356,
      "learning_rate": 9.927818399000549e-05,
      "loss": 1.5746896743774415,
      "step": 1510
    },
    {
      "epoch": 0.5211949766405212,
      "grad_norm": 0.9068400859832764,
      "learning_rate": 9.816778443773253e-05,
      "loss": 1.5674661636352538,
      "step": 1520
    },
    {
      "epoch": 0.5246238909605246,
      "grad_norm": 1.2628779411315918,
      "learning_rate": 9.705761083213463e-05,
      "loss": 1.5925458908081054,
      "step": 1530
    },
    {
      "epoch": 0.528052805280528,
      "grad_norm": 0.988800585269928,
      "learning_rate": 9.594780007851946e-05,
      "loss": 1.4770522117614746,
      "step": 1540
    },
    {
      "epoch": 0.5314817196005315,
      "grad_norm": 0.9503228664398193,
      "learning_rate": 9.483848903744794e-05,
      "loss": 1.4404266357421875,
      "step": 1550
    },
    {
      "epoch": 0.534910633920535,
      "grad_norm": 0.8870835900306702,
      "learning_rate": 9.372981450785722e-05,
      "loss": 1.6078582763671876,
      "step": 1560
    },
    {
      "epoch": 0.5383395482405383,
      "grad_norm": 0.9531223773956299,
      "learning_rate": 9.26219132101905e-05,
      "loss": 1.4260448455810546,
      "step": 1570
    },
    {
      "epoch": 0.5417684625605418,
      "grad_norm": 0.7870782017707825,
      "learning_rate": 9.151492176953697e-05,
      "loss": 1.5140536308288575,
      "step": 1580
    },
    {
      "epoch": 0.5451973768805451,
      "grad_norm": 1.0012214183807373,
      "learning_rate": 9.040897669878327e-05,
      "loss": 1.4889466285705566,
      "step": 1590
    },
    {
      "epoch": 0.5486262912005486,
      "grad_norm": 0.9016415476799011,
      "learning_rate": 8.9304214381779e-05,
      "loss": 1.520207977294922,
      "step": 1600
    },
    {
      "epoch": 0.5520552055205521,
      "grad_norm": 0.9925042986869812,
      "learning_rate": 8.820077105651794e-05,
      "loss": 1.4937515258789062,
      "step": 1610
    },
    {
      "epoch": 0.5554841198405555,
      "grad_norm": 1.084972620010376,
      "learning_rate": 8.709878279833726e-05,
      "loss": 1.4967594146728516,
      "step": 1620
    },
    {
      "epoch": 0.5589130341605589,
      "grad_norm": 0.896564781665802,
      "learning_rate": 8.599838550313713e-05,
      "loss": 1.58676176071167,
      "step": 1630
    },
    {
      "epoch": 0.5623419484805623,
      "grad_norm": 0.8893474340438843,
      "learning_rate": 8.489971487062183e-05,
      "loss": 1.5984265327453613,
      "step": 1640
    },
    {
      "epoch": 0.5657708628005658,
      "grad_norm": 1.103757381439209,
      "learning_rate": 8.380290638756584e-05,
      "loss": 1.500683307647705,
      "step": 1650
    },
    {
      "epoch": 0.5691997771205692,
      "grad_norm": 0.8493750095367432,
      "learning_rate": 8.270809531110534e-05,
      "loss": 1.4846227645874024,
      "step": 1660
    },
    {
      "epoch": 0.5726286914405726,
      "grad_norm": 0.9126749038696289,
      "learning_rate": 8.161541665205885e-05,
      "loss": 1.4371955871582032,
      "step": 1670
    },
    {
      "epoch": 0.5760576057605761,
      "grad_norm": 0.821764349937439,
      "learning_rate": 8.052500515827758e-05,
      "loss": 1.5160919189453126,
      "step": 1680
    },
    {
      "epoch": 0.5794865200805794,
      "grad_norm": 0.914444625377655,
      "learning_rate": 7.943699529802855e-05,
      "loss": 1.479448413848877,
      "step": 1690
    },
    {
      "epoch": 0.5829154344005829,
      "grad_norm": 1.22026789188385,
      "learning_rate": 7.835152124341228e-05,
      "loss": 1.4975048065185548,
      "step": 1700
    },
    {
      "epoch": 0.5863443487205864,
      "grad_norm": 0.9239258766174316,
      "learning_rate": 7.726871685381652e-05,
      "loss": 1.610418701171875,
      "step": 1710
    },
    {
      "epoch": 0.5897732630405897,
      "grad_norm": 1.1614937782287598,
      "learning_rate": 7.618871565940921e-05,
      "loss": 1.5355528831481933,
      "step": 1720
    },
    {
      "epoch": 0.5932021773605932,
      "grad_norm": 1.099278450012207,
      "learning_rate": 7.511165084467151e-05,
      "loss": 1.477409839630127,
      "step": 1730
    },
    {
      "epoch": 0.5966310916805966,
      "grad_norm": 0.9028745293617249,
      "learning_rate": 7.403765523197364e-05,
      "loss": 1.4444008827209474,
      "step": 1740
    },
    {
      "epoch": 0.6000600060006,
      "grad_norm": 1.182869791984558,
      "learning_rate": 7.296686126519552e-05,
      "loss": 1.475886058807373,
      "step": 1750
    },
    {
      "epoch": 0.6034889203206035,
      "grad_norm": 0.9319450855255127,
      "learning_rate": 7.189940099339369e-05,
      "loss": 1.524347686767578,
      "step": 1760
    },
    {
      "epoch": 0.6069178346406069,
      "grad_norm": 0.9652882814407349,
      "learning_rate": 7.08354060545175e-05,
      "loss": 1.456563377380371,
      "step": 1770
    },
    {
      "epoch": 0.6103467489606104,
      "grad_norm": 0.9450862407684326,
      "learning_rate": 6.977500765917539e-05,
      "loss": 1.4752133369445801,
      "step": 1780
    },
    {
      "epoch": 0.6137756632806137,
      "grad_norm": 1.0317661762237549,
      "learning_rate": 6.871833657445438e-05,
      "loss": 1.4760221481323241,
      "step": 1790
    },
    {
      "epoch": 0.6172045776006172,
      "grad_norm": 0.9987633228302002,
      "learning_rate": 6.766552310779373e-05,
      "loss": 1.462003803253174,
      "step": 1800
    },
    {
      "epoch": 0.6206334919206207,
      "grad_norm": 1.3961414098739624,
      "learning_rate": 6.661669709091597e-05,
      "loss": 1.4372685432434082,
      "step": 1810
    },
    {
      "epoch": 0.624062406240624,
      "grad_norm": 1.0918911695480347,
      "learning_rate": 6.557198786381584e-05,
      "loss": 1.5176566123962403,
      "step": 1820
    },
    {
      "epoch": 0.6274913205606275,
      "grad_norm": 1.1604210138320923,
      "learning_rate": 6.45315242588105e-05,
      "loss": 1.498598003387451,
      "step": 1830
    },
    {
      "epoch": 0.6309202348806309,
      "grad_norm": 0.8426704406738281,
      "learning_rate": 6.349543458465193e-05,
      "loss": 1.5005523681640625,
      "step": 1840
    },
    {
      "epoch": 0.6343491492006343,
      "grad_norm": 1.0768263339996338,
      "learning_rate": 6.246384661070404e-05,
      "loss": 1.5108792304992675,
      "step": 1850
    },
    {
      "epoch": 0.6377780635206378,
      "grad_norm": 1.007082223892212,
      "learning_rate": 6.143688755118646e-05,
      "loss": 1.4070115089416504,
      "step": 1860
    },
    {
      "epoch": 0.6412069778406412,
      "grad_norm": 1.0561329126358032,
      "learning_rate": 6.0414684049486334e-05,
      "loss": 1.467573356628418,
      "step": 1870
    },
    {
      "epoch": 0.6446358921606447,
      "grad_norm": 0.999627947807312,
      "learning_rate": 5.9397362162541256e-05,
      "loss": 1.4690093994140625,
      "step": 1880
    },
    {
      "epoch": 0.648064806480648,
      "grad_norm": 1.119313359260559,
      "learning_rate": 5.838504734529353e-05,
      "loss": 1.5171260833740234,
      "step": 1890
    },
    {
      "epoch": 0.6514937208006515,
      "grad_norm": 1.227597713470459,
      "learning_rate": 5.7377864435219684e-05,
      "loss": 1.5152154922485352,
      "step": 1900
    },
    {
      "epoch": 0.654922635120655,
      "grad_norm": 1.2129477262496948,
      "learning_rate": 5.637593763693545e-05,
      "loss": 1.4743117332458495,
      "step": 1910
    },
    {
      "epoch": 0.6583515494406583,
      "grad_norm": 1.1016993522644043,
      "learning_rate": 5.537939050687886e-05,
      "loss": 1.4259710311889648,
      "step": 1920
    },
    {
      "epoch": 0.6617804637606618,
      "grad_norm": 1.0503939390182495,
      "learning_rate": 5.438834593807382e-05,
      "loss": 1.3683046340942382,
      "step": 1930
    },
    {
      "epoch": 0.6652093780806652,
      "grad_norm": 1.0326424837112427,
      "learning_rate": 5.3402926144974616e-05,
      "loss": 1.3892087936401367,
      "step": 1940
    },
    {
      "epoch": 0.6686382924006686,
      "grad_norm": 0.9721679091453552,
      "learning_rate": 5.242325264839494e-05,
      "loss": 1.5494998931884765,
      "step": 1950
    },
    {
      "epoch": 0.6720672067206721,
      "grad_norm": 1.0285075902938843,
      "learning_rate": 5.144944626052177e-05,
      "loss": 1.453069305419922,
      "step": 1960
    },
    {
      "epoch": 0.6754961210406755,
      "grad_norm": 0.9177806973457336,
      "learning_rate": 5.048162707001727e-05,
      "loss": 1.3458268165588378,
      "step": 1970
    },
    {
      "epoch": 0.6789250353606789,
      "grad_norm": 1.1880685091018677,
      "learning_rate": 4.9519914427209366e-05,
      "loss": 1.4461503982543946,
      "step": 1980
    },
    {
      "epoch": 0.6823539496806823,
      "grad_norm": 0.8908017873764038,
      "learning_rate": 4.856442692937372e-05,
      "loss": 1.4234731674194336,
      "step": 1990
    },
    {
      "epoch": 0.6857828640006858,
      "grad_norm": 1.0302280187606812,
      "learning_rate": 4.761528240610842e-05,
      "loss": 1.481229591369629,
      "step": 2000
    },
    {
      "epoch": 0.6892117783206893,
      "grad_norm": 1.0356521606445312,
      "learning_rate": 4.6672597904803274e-05,
      "loss": 1.4873802185058593,
      "step": 2010
    },
    {
      "epoch": 0.6926406926406926,
      "grad_norm": 1.0063520669937134,
      "learning_rate": 4.5736489676205885e-05,
      "loss": 1.4728717803955078,
      "step": 2020
    },
    {
      "epoch": 0.6960696069606961,
      "grad_norm": 1.245090365409851,
      "learning_rate": 4.4807073160085486e-05,
      "loss": 1.3752614974975585,
      "step": 2030
    },
    {
      "epoch": 0.6994985212806994,
      "grad_norm": 1.1579086780548096,
      "learning_rate": 4.3884462970997275e-05,
      "loss": 1.4613791465759278,
      "step": 2040
    },
    {
      "epoch": 0.7029274356007029,
      "grad_norm": 1.0563075542449951,
      "learning_rate": 4.2968772884148147e-05,
      "loss": 1.279030704498291,
      "step": 2050
    },
    {
      "epoch": 0.7063563499207064,
      "grad_norm": 1.2080659866333008,
      "learning_rate": 4.206011582136609e-05,
      "loss": 1.4714184761047364,
      "step": 2060
    },
    {
      "epoch": 0.7097852642407098,
      "grad_norm": 0.9796544909477234,
      "learning_rate": 4.115860383717486e-05,
      "loss": 1.3989374160766601,
      "step": 2070
    },
    {
      "epoch": 0.7132141785607132,
      "grad_norm": 1.0658836364746094,
      "learning_rate": 4.026434810497538e-05,
      "loss": 1.4178956985473632,
      "step": 2080
    },
    {
      "epoch": 0.7166430928807166,
      "grad_norm": 1.1085129976272583,
      "learning_rate": 3.937745890333623e-05,
      "loss": 1.4281699180603027,
      "step": 2090
    },
    {
      "epoch": 0.7200720072007201,
      "grad_norm": 0.9898178577423096,
      "learning_rate": 3.849804560239394e-05,
      "loss": 1.296155834197998,
      "step": 2100
    },
    {
      "epoch": 0.7235009215207235,
      "grad_norm": 1.182127594947815,
      "learning_rate": 3.762621665036583e-05,
      "loss": 1.4685094833374024,
      "step": 2110
    },
    {
      "epoch": 0.7269298358407269,
      "grad_norm": 0.9919553399085999,
      "learning_rate": 3.6762079560176174e-05,
      "loss": 1.351000690460205,
      "step": 2120
    },
    {
      "epoch": 0.7303587501607304,
      "grad_norm": 1.1257010698318481,
      "learning_rate": 3.590574089619786e-05,
      "loss": 1.3968747138977051,
      "step": 2130
    },
    {
      "epoch": 0.7337876644807337,
      "grad_norm": 1.0049985647201538,
      "learning_rate": 3.505730626111102e-05,
      "loss": 1.2970483779907227,
      "step": 2140
    },
    {
      "epoch": 0.7372165788007372,
      "grad_norm": 0.8539199829101562,
      "learning_rate": 3.4216880282880124e-05,
      "loss": 1.4288148880004883,
      "step": 2150
    },
    {
      "epoch": 0.7406454931207407,
      "grad_norm": 1.125787377357483,
      "learning_rate": 3.3384566601851576e-05,
      "loss": 1.4017534255981445,
      "step": 2160
    },
    {
      "epoch": 0.744074407440744,
      "grad_norm": 1.0176100730895996,
      "learning_rate": 3.256046785797274e-05,
      "loss": 1.3627896308898926,
      "step": 2170
    },
    {
      "epoch": 0.7475033217607475,
      "grad_norm": 0.8985896706581116,
      "learning_rate": 3.174468567813461e-05,
      "loss": 1.3177756309509276,
      "step": 2180
    },
    {
      "epoch": 0.7509322360807509,
      "grad_norm": 0.9561920166015625,
      "learning_rate": 3.093732066363915e-05,
      "loss": 1.3332663536071778,
      "step": 2190
    },
    {
      "epoch": 0.7543611504007544,
      "grad_norm": 1.1088588237762451,
      "learning_rate": 3.013847237779346e-05,
      "loss": 1.3807819366455079,
      "step": 2200
    },
    {
      "epoch": 0.7577900647207578,
      "grad_norm": 0.9919373393058777,
      "learning_rate": 2.9348239333631656e-05,
      "loss": 1.3635038375854491,
      "step": 2210
    },
    {
      "epoch": 0.7612189790407612,
      "grad_norm": 0.9647587537765503,
      "learning_rate": 2.8566718981766238e-05,
      "loss": 1.4107337951660157,
      "step": 2220
    },
    {
      "epoch": 0.7646478933607647,
      "grad_norm": 1.059625506401062,
      "learning_rate": 2.7794007698370826e-05,
      "loss": 1.386619758605957,
      "step": 2230
    },
    {
      "epoch": 0.768076807680768,
      "grad_norm": 1.1716138124465942,
      "learning_rate": 2.703020077329498e-05,
      "loss": 1.3893098831176758,
      "step": 2240
    },
    {
      "epoch": 0.7715057220007715,
      "grad_norm": 1.1476608514785767,
      "learning_rate": 2.627539239831328e-05,
      "loss": 1.3669297218322753,
      "step": 2250
    },
    {
      "epoch": 0.774934636320775,
      "grad_norm": 1.0528924465179443,
      "learning_rate": 2.5529675655509567e-05,
      "loss": 1.3363636016845704,
      "step": 2260
    },
    {
      "epoch": 0.7783635506407783,
      "grad_norm": 0.8927570581436157,
      "learning_rate": 2.4793142505798362e-05,
      "loss": 1.438697338104248,
      "step": 2270
    },
    {
      "epoch": 0.7817924649607818,
      "grad_norm": 1.102528691291809,
      "learning_rate": 2.4065883777584207e-05,
      "loss": 1.3745312690734863,
      "step": 2280
    },
    {
      "epoch": 0.7852213792807852,
      "grad_norm": 0.9203447103500366,
      "learning_rate": 2.3347989155560834e-05,
      "loss": 1.402297019958496,
      "step": 2290
    },
    {
      "epoch": 0.7886502936007886,
      "grad_norm": 0.9557979702949524,
      "learning_rate": 2.263954716965142e-05,
      "loss": 1.4507503509521484,
      "step": 2300
    },
    {
      "epoch": 0.7920792079207921,
      "grad_norm": 0.9851720929145813,
      "learning_rate": 2.1940645184091113e-05,
      "loss": 1.4763115882873534,
      "step": 2310
    },
    {
      "epoch": 0.7955081222407955,
      "grad_norm": 1.058793544769287,
      "learning_rate": 2.1251369386653454e-05,
      "loss": 1.3218843460083007,
      "step": 2320
    },
    {
      "epoch": 0.798937036560799,
      "grad_norm": 0.9868523478507996,
      "learning_rate": 2.0571804778021642e-05,
      "loss": 1.4029590606689453,
      "step": 2330
    },
    {
      "epoch": 0.8023659508808023,
      "grad_norm": 1.0275932550430298,
      "learning_rate": 1.9902035161306575e-05,
      "loss": 1.481428623199463,
      "step": 2340
    },
    {
      "epoch": 0.8057948652008058,
      "grad_norm": 1.0626776218414307,
      "learning_rate": 1.924214313171211e-05,
      "loss": 1.4031143188476562,
      "step": 2350
    },
    {
      "epoch": 0.8092237795208093,
      "grad_norm": 1.1311109066009521,
      "learning_rate": 1.859221006634978e-05,
      "loss": 1.3093020439147949,
      "step": 2360
    },
    {
      "epoch": 0.8126526938408126,
      "grad_norm": 1.0533804893493652,
      "learning_rate": 1.7952316114203248e-05,
      "loss": 1.296053123474121,
      "step": 2370
    },
    {
      "epoch": 0.8160816081608161,
      "grad_norm": 1.1000430583953857,
      "learning_rate": 1.732254018624446e-05,
      "loss": 1.367686653137207,
      "step": 2380
    },
    {
      "epoch": 0.8195105224808195,
      "grad_norm": 0.9357230067253113,
      "learning_rate": 1.67029599457026e-05,
      "loss": 1.3232050895690919,
      "step": 2390
    },
    {
      "epoch": 0.8229394368008229,
      "grad_norm": 1.1516122817993164,
      "learning_rate": 1.6093651798486488e-05,
      "loss": 1.4300651550292969,
      "step": 2400
    },
    {
      "epoch": 0.8263683511208264,
      "grad_norm": 1.0197898149490356,
      "learning_rate": 1.5494690883762554e-05,
      "loss": 1.3701356887817382,
      "step": 2410
    },
    {
      "epoch": 0.8297972654408298,
      "grad_norm": 1.1041163206100464,
      "learning_rate": 1.49061510646886e-05,
      "loss": 1.413942241668701,
      "step": 2420
    },
    {
      "epoch": 0.8332261797608332,
      "grad_norm": 0.9526892304420471,
      "learning_rate": 1.4328104919305141e-05,
      "loss": 1.3763864517211915,
      "step": 2430
    },
    {
      "epoch": 0.8366550940808366,
      "grad_norm": 1.0712953805923462,
      "learning_rate": 1.3760623731585165e-05,
      "loss": 1.3769660949707032,
      "step": 2440
    },
    {
      "epoch": 0.8400840084008401,
      "grad_norm": 1.0556769371032715,
      "learning_rate": 1.320377748264341e-05,
      "loss": 1.3898455619812011,
      "step": 2450
    },
    {
      "epoch": 0.8435129227208435,
      "grad_norm": 1.0892301797866821,
      "learning_rate": 1.2657634842106524e-05,
      "loss": 1.370183563232422,
      "step": 2460
    },
    {
      "epoch": 0.8469418370408469,
      "grad_norm": 1.1025022268295288,
      "learning_rate": 1.212226315964462e-05,
      "loss": 1.3115644454956055,
      "step": 2470
    },
    {
      "epoch": 0.8503707513608504,
      "grad_norm": 1.1549314260482788,
      "learning_rate": 1.1597728456665957e-05,
      "loss": 1.3754231452941894,
      "step": 2480
    },
    {
      "epoch": 0.8537996656808537,
      "grad_norm": 0.991187572479248,
      "learning_rate": 1.1084095418175156e-05,
      "loss": 1.4104815483093263,
      "step": 2490
    },
    {
      "epoch": 0.8572285800008572,
      "grad_norm": 1.0332754850387573,
      "learning_rate": 1.0581427384796372e-05,
      "loss": 1.2882612228393555,
      "step": 2500
    },
    {
      "epoch": 0.8606574943208607,
      "grad_norm": 1.2193644046783447,
      "learning_rate": 1.0089786344962194e-05,
      "loss": 1.355900192260742,
      "step": 2510
    },
    {
      "epoch": 0.8640864086408641,
      "grad_norm": 0.994361400604248,
      "learning_rate": 9.60923292726923e-06,
      "loss": 1.4017061233520507,
      "step": 2520
    },
    {
      "epoch": 0.8675153229608675,
      "grad_norm": 0.9575983285903931,
      "learning_rate": 9.139826393001617e-06,
      "loss": 1.2158626556396483,
      "step": 2530
    },
    {
      "epoch": 0.8709442372808709,
      "grad_norm": 1.0644499063491821,
      "learning_rate": 8.681624628822793e-06,
      "loss": 1.455751323699951,
      "step": 2540
    },
    {
      "epoch": 0.8743731516008744,
      "grad_norm": 0.9530572295188904,
      "learning_rate": 8.234684139637205e-06,
      "loss": 1.3413010597229005,
      "step": 2550
    },
    {
      "epoch": 0.8778020659208778,
      "grad_norm": 1.0808656215667725,
      "learning_rate": 7.799060041621974e-06,
      "loss": 1.2076227188110351,
      "step": 2560
    },
    {
      "epoch": 0.8812309802408812,
      "grad_norm": 1.0733431577682495,
      "learning_rate": 7.37480605543015e-06,
      "loss": 1.3099120140075684,
      "step": 2570
    },
    {
      "epoch": 0.8846598945608847,
      "grad_norm": 0.9678606390953064,
      "learning_rate": 6.961974499565982e-06,
      "loss": 1.2811833381652833,
      "step": 2580
    },
    {
      "epoch": 0.888088808880888,
      "grad_norm": 0.9264373779296875,
      "learning_rate": 6.560616283932897e-06,
      "loss": 1.3343655586242675,
      "step": 2590
    },
    {
      "epoch": 0.8915177232008915,
      "grad_norm": 1.028125286102295,
      "learning_rate": 6.170780903555528e-06,
      "loss": 1.3404034614562987,
      "step": 2600
    },
    {
      "epoch": 0.894946637520895,
      "grad_norm": 1.072121500968933,
      "learning_rate": 5.79251643247598e-06,
      "loss": 1.2757418632507325,
      "step": 2610
    },
    {
      "epoch": 0.8983755518408983,
      "grad_norm": 0.8474506139755249,
      "learning_rate": 5.425869517825366e-06,
      "loss": 1.4403800010681151,
      "step": 2620
    },
    {
      "epoch": 0.9018044661609018,
      "grad_norm": 1.1511244773864746,
      "learning_rate": 5.0708853740713216e-06,
      "loss": 1.4375917434692382,
      "step": 2630
    },
    {
      "epoch": 0.9052333804809052,
      "grad_norm": 1.081178069114685,
      "learning_rate": 4.727607777442333e-06,
      "loss": 1.4061691284179687,
      "step": 2640
    },
    {
      "epoch": 0.9086622948009087,
      "grad_norm": 1.0403251647949219,
      "learning_rate": 4.396079060529146e-06,
      "loss": 1.4513935089111327,
      "step": 2650
    },
    {
      "epoch": 0.9120912091209121,
      "grad_norm": 1.096602439880371,
      "learning_rate": 4.076340107064458e-06,
      "loss": 1.4470197677612304,
      "step": 2660
    },
    {
      "epoch": 0.9155201234409155,
      "grad_norm": 0.8697181344032288,
      "learning_rate": 3.7684303468811688e-06,
      "loss": 1.3405644416809082,
      "step": 2670
    },
    {
      "epoch": 0.918949037760919,
      "grad_norm": 1.0498508214950562,
      "learning_rate": 3.4723877510499437e-06,
      "loss": 1.4038556098937989,
      "step": 2680
    },
    {
      "epoch": 0.9223779520809223,
      "grad_norm": 0.9626743793487549,
      "learning_rate": 3.188248827196616e-06,
      "loss": 1.327705478668213,
      "step": 2690
    },
    {
      "epoch": 0.9258068664009258,
      "grad_norm": 1.0272961854934692,
      "learning_rate": 2.9160486150001554e-06,
      "loss": 1.3231115341186523,
      "step": 2700
    },
    {
      "epoch": 0.9292357807209293,
      "grad_norm": 0.9614903330802917,
      "learning_rate": 2.6558206818716346e-06,
      "loss": 1.3979063034057617,
      "step": 2710
    },
    {
      "epoch": 0.9326646950409326,
      "grad_norm": 1.1483595371246338,
      "learning_rate": 2.407597118814675e-06,
      "loss": 1.372023868560791,
      "step": 2720
    },
    {
      "epoch": 0.9360936093609361,
      "grad_norm": 0.9721912145614624,
      "learning_rate": 2.171408536468067e-06,
      "loss": 1.3513710021972656,
      "step": 2730
    },
    {
      "epoch": 0.9395225236809395,
      "grad_norm": 1.0480808019638062,
      "learning_rate": 1.9472840613308787e-06,
      "loss": 1.452831745147705,
      "step": 2740
    },
    {
      "epoch": 0.9429514380009429,
      "grad_norm": 0.7381024360656738,
      "learning_rate": 1.735251332170662e-06,
      "loss": 1.4015365600585938,
      "step": 2750
    },
    {
      "epoch": 0.9463803523209464,
      "grad_norm": 1.2480961084365845,
      "learning_rate": 1.5353364966149696e-06,
      "loss": 1.3289525985717774,
      "step": 2760
    },
    {
      "epoch": 0.9498092666409498,
      "grad_norm": 1.1523001194000244,
      "learning_rate": 1.3475642079269656e-06,
      "loss": 1.4086174011230468,
      "step": 2770
    },
    {
      "epoch": 0.9532381809609533,
      "grad_norm": 1.03007173538208,
      "learning_rate": 1.1719576219651585e-06,
      "loss": 1.3973102569580078,
      "step": 2780
    },
    {
      "epoch": 0.9566670952809566,
      "grad_norm": 0.9748601317405701,
      "learning_rate": 1.0085383943278292e-06,
      "loss": 1.4070877075195312,
      "step": 2790
    },
    {
      "epoch": 0.9600960096009601,
      "grad_norm": 1.0782294273376465,
      "learning_rate": 8.573266776825684e-07,
      "loss": 1.4141860961914063,
      "step": 2800
    },
    {
      "epoch": 0.9635249239209636,
      "grad_norm": 1.007416844367981,
      "learning_rate": 7.183411192810075e-07,
      "loss": 1.4470108985900878,
      "step": 2810
    },
    {
      "epoch": 0.9669538382409669,
      "grad_norm": 0.8667322397232056,
      "learning_rate": 5.91598858659248e-07,
      "loss": 1.3368803024291993,
      "step": 2820
    },
    {
      "epoch": 0.9703827525609704,
      "grad_norm": 1.0433310270309448,
      "learning_rate": 4.771155255242854e-07,
      "loss": 1.3557028770446777,
      "step": 2830
    },
    {
      "epoch": 0.9738116668809738,
      "grad_norm": 1.104323387145996,
      "learning_rate": 3.749052378265505e-07,
      "loss": 1.3755488395690918,
      "step": 2840
    },
    {
      "epoch": 0.9772405812009772,
      "grad_norm": 1.060232162475586,
      "learning_rate": 2.849806000189026e-07,
      "loss": 1.343965721130371,
      "step": 2850
    },
    {
      "epoch": 0.9806694955209807,
      "grad_norm": 0.9373921155929565,
      "learning_rate": 2.0735270150223918e-07,
      "loss": 1.4640989303588867,
      "step": 2860
    },
    {
      "epoch": 0.9840984098409841,
      "grad_norm": 1.392088770866394,
      "learning_rate": 1.4203111525801227e-07,
      "loss": 1.281282329559326,
      "step": 2870
    },
    {
      "epoch": 0.9875273241609875,
      "grad_norm": 1.1889876127243042,
      "learning_rate": 8.902389666765042e-08,
      "loss": 1.4277688980102539,
      "step": 2880
    },
    {
      "epoch": 0.9909562384809909,
      "grad_norm": 1.1139051914215088,
      "learning_rate": 4.833758251919784e-08,
      "loss": 1.295989418029785,
      "step": 2890
    },
    {
      "epoch": 0.9943851528009944,
      "grad_norm": 0.8890063762664795,
      "learning_rate": 1.9977190201225794e-08,
      "loss": 1.4465449333190918,
      "step": 2900
    },
    {
      "epoch": 0.9978140671209978,
      "grad_norm": 1.0152575969696045,
      "learning_rate": 3.9462170840720566e-09,
      "loss": 1.3812876701354981,
      "step": 2910
    }
  ],
  "logging_steps": 10,
  "max_steps": 2917,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 1.8483614067586253e+18,
  "train_batch_size": 4,
  "trial_name": null,
  "trial_params": null
}